From 477c0757578788d69749ee4101b031b263383640 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 5 Mar 2024 15:41:10 -0500 Subject: [PATCH] remove a little more --- zoobot/pytorch/training/finetune.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/zoobot/pytorch/training/finetune.py b/zoobot/pytorch/training/finetune.py index 098cd3ee..13b0e223 100644 --- a/zoobot/pytorch/training/finetune.py +++ b/zoobot/pytorch/training/finetune.py @@ -104,11 +104,11 @@ def __init__( self.encoder = encoder # TODO read as encoder property - if isinstance(self.encoder, CustomMAEEncoder): - self.encoder_dim = 256 # vit hidden dim, assuming average pool over seq dim - # self.encoder_dim = 9216 - else: - self.encoder_dim = define_model.get_encoder_dim(self.encoder) + # if isinstance(self.encoder, CustomMAEEncoder): + # self.encoder_dim = 256 # vit hidden dim, assuming average pool over seq dim + # # self.encoder_dim = 9216 + # else: + self.encoder_dim = define_model.get_encoder_dim(self.encoder) # for backwards compat. if n_layers: