fix typo

mwalmsley · Feb 27, 2024 · 9eb0d89 · 9eb0d89
1 parent 797b748
commit 9eb0d89
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/zoobot/pytorch/training/finetune.py b/zoobot/pytorch/training/finetune.py
@@ -108,12 +108,13 @@ def __init__(
         else:
             self.encoder_dim = define_model.get_encoder_dim(self.encoder)
         self.n_blocks = n_blocks
+        logging.info('Blocks to finetune: {}'.format(n_layers))
 
         # for backwards compat.
         if n_layers:
             logging.warning('FinetuneableZoobot(n_layers) is now renamed to n_blocks, please update to pass n_blocks instead! For now, setting n_blocks=n_layers')
             self.n_blocks = n_layers
-        logging.info('Layers to finetune: {}'.format(n_layers))
+            logging.info('Layers to finetune: {}'.format(n_layers))
 
         self.learning_rate = learning_rate
         self.lr_decay = lr_decay