diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-23 16:06:45 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-23 16:06:45 +0200 |
commit | 524bf4351ac295bd4ff9914bb1f32eda7f7ff855 (patch) | |
tree | 5ddf3b6e52da67186294deceae0e67820149a30c /text_recognizer/models | |
parent | 63a69eb4015f203ff53f9fd0cbed10abbf3bbd87 (diff) |
Remove dino
Diffstat (limited to 'text_recognizer/models')
-rw-r--r-- | text_recognizer/models/dino.py | 28 |
1 files changed, 0 insertions, 28 deletions
diff --git a/text_recognizer/models/dino.py b/text_recognizer/models/dino.py deleted file mode 100644 index dca954c..0000000 --- a/text_recognizer/models/dino.py +++ /dev/null @@ -1,28 +0,0 @@ -"""Dino: pretraining of models with self supervision.""" -import copy -from functools import wraps, partial - -import torch -from torch import nn -import torch.nn.funtional as F -import torchvision.transforms as T -import wandb - -from text_recognizer.models.base import LitBaseModel - - -def singleton(cache_key): - def inner_fn(fn): - @wraps(fn) - def wrapper(self, *args, **kwargs): - instance = getattr(self, cache_key) - if instance is not None: - return instance - - instance = fn(self, *args, **kwargs) - setattr(self, cache_key, instance) - return instance - - return wrapper - - return inner_fn |