From 3095b5a5c625b6671fb3e51e0aa4ae29e8b7e2c8 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 6 Feb 2024 15:06:39 -0500 Subject: [PATCH] maxvit fix --- zoobot/pytorch/training/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/zoobot/pytorch/training/finetune.py b/zoobot/pytorch/training/finetune.py index b3d7dbe9..0a846cb9 100644 --- a/zoobot/pytorch/training/finetune.py +++ b/zoobot/pytorch/training/finetune.py @@ -175,7 +175,7 @@ def configure_optimizers(self): self.encoder.layer4 ] elif isinstance(self.encoder, timm.models.MaxxVit): - blocks_to_tune = self.encoder.stem + [stage for stage in self.encoder.stages] + blocks_to_tune = [self.encoder.stem] + [stage for stage in self.encoder.stages] else: raise ValueError(f'Encoder architecture not automatically recognised: {type(self.encoder)}')