From 9eb0d89408eab28eb0ddd1b2eed72011fb769563 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Tue, 27 Feb 2024 13:20:27 -0500 Subject: [PATCH] fix typo --- zoobot/pytorch/training/finetune.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/zoobot/pytorch/training/finetune.py b/zoobot/pytorch/training/finetune.py index 7e963a99..7902db7e 100644 --- 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