BUG: Fix reference to undefined att_mask -> mask

bioscan-ml · Feb 9, 2025 · cb6a88a · cb6a88a
1 parent b684071
commit cb6a88a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/baselines/finetuning.py b/baselines/finetuning.py
@@ -55,7 +55,7 @@ def forward(self, sequences=None, mask=None, labels=None):
             out = self.base_model(sequences, attention_mask=mask)[0]
 
         elif self.backbone == "BarcodeBERT":
-            out = self.base_model(sequences, att_mask).hidden_states[-1]
+            out = self.base_model(sequences, mask).hidden_states[-1]
 
         # if backbone != "BarcodeBERT":
         # print(out.shape)