Skip to content

Commit

Permalink
typo
Browse files Browse the repository at this point in the history
  • Loading branch information
mwalmsley committed Mar 19, 2024
1 parent 0fe5cec commit e0fd96d
Showing 1 changed file with 1 addition and 23 deletions.
24 changes: 1 addition & 23 deletions zoobot/pytorch/training/representations.py
Original file line number Diff line number Diff line change
Expand Up @@ -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():


0 comments on commit e0fd96d

Please sign in to comment.