remove a little more

mwalmsley · Mar 5, 2024 · 477c075 · 477c075
1 parent c14637e
commit 477c075
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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: