From 772660d5a7f7518c93610da5aca93a01e2cecd16 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 27 Feb 2024 16:37:30 -0500 Subject: [PATCH] docs --- 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 4b461555..6e9136fa 100644 --- a/zoobot/pytorch/training/finetune.py +++ b/zoobot/pytorch/training/finetune.py @@ -177,7 +177,7 @@ def configure_optimizers(self): ] elif isinstance(self.encoder, timm.models.MaxxVit): blocks_to_tune = [self.encoder.stem] + [stage for stage in self.encoder.stages] - elif isinstance(self.encoder, timm.models.ConvNeXt): # stem + 3 blocks, for all sizes + elif isinstance(self.encoder, timm.models.ConvNeXt): # stem + 4 blocks, for all sizes # https://github.com/huggingface/pytorch-image-models/blob/main/timm/models/convnext.py#L264 blocks_to_tune = [self.encoder.stem] + [stage for stage in self.encoder.stages] else: