typo

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