From 524bf4351ac295bd4ff9914bb1f32eda7f7ff855 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Fri, 23 Jul 2021 16:06:45 +0200 Subject: Remove dino --- text_recognizer/models/dino.py | 28 ---------------------------- 1 file changed, 28 deletions(-) delete mode 100644 text_recognizer/models/dino.py (limited to 'text_recognizer/models') 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 -- cgit v1.2.3-70-g09d2