From 797b7488512f40924dc333d4f303977e6db94680 Mon Sep 17 00:00:00 2001 From: Mike Walmsley Date: Mon, 26 Feb 2024 22:36:57 -0500 Subject: [PATCH] typo --- zoobot/pytorch/training/finetune.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/zoobot/pytorch/training/finetune.py b/zoobot/pytorch/training/finetune.py index 144ce79b..7e963a99 100644 --- a/zoobot/pytorch/training/finetune.py +++ b/zoobot/pytorch/training/finetune.py @@ -555,8 +555,7 @@ def __init__(self, input_dim, output_dim, dropout_prob=0.5, activation=None): self.output_dim = output_dim self.dropout = torch.nn.Dropout(p=dropout_prob) self.linear = torch.nn.Linear(input_dim, output_dim) - if activation is not None: - self.activation = activation + self.activation = activation def forward(self, x): # returns logits, as recommended for CrossEntropy loss