From e0fd96d0bc127b40156eb1d7209fa44d750aaafa Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 19 Mar 2024 18:15:31 -0400 Subject: [PATCH] typo --- zoobot/pytorch/training/representations.py | 24 +--------------------- 1 file changed, 1 insertion(+), 23 deletions(-) diff --git a/zoobot/pytorch/training/representations.py b/zoobot/pytorch/training/representations.py index 1c5fab19..09b379a4 100644 --- a/zoobot/pytorch/training/representations.py +++ b/zoobot/pytorch/training/representations.py @@ -28,26 +28,4 @@ def load_from_name(cls, name: str): """ timm_model = create_model(name) return cls(timm_model) - - - - - -class ZoobotEncoder(pl.LightningModule): - # very simple wrapper to turn pytorch model into lightning module - # useful when we want to use lightning to make predictions with our encoder - # (i.e. to get representations) - - # pretrained_cfg, pretrained_cfg_overlay=timm_kwargs - def __init__(self, architecture_name=None, channels=None, timm_kwargs={}) -> None: - super().__init__() - - logging.info('ZoobotEncoder: using timm encoder') - self.encoder = - - # if pyramid: - # raise NotImplementedError('Will eventually support resetting timm classifier to get FPN features') - - -# def save_timm_encoder(): - + \ No newline at end of file