Logs

app.py

@@ -151,14 +151,14 @@ def evaluate_model(model, dataloader, device):
             # Process each sequence in the batch
             for i in range(current_batch_size):
                 valid_mask = (labels[i] != -100) & attention_mask[i]
+                valid_preds = preds[i][valid_mask[i]].flatten()
+                valid_labels = labels[i][valid_mask[i]].flatten()
                 print(valid_mask.dtype)
                 print(labels[i].shape)
                 print(attention_mask[i].shape)
                 print(valid_mask.shape)
                 print(valid_labels)
                 print(valid_mask)
-                valid_preds = preds[i][valid_mask[i]].flatten()
-                valid_labels = labels[i][valid_mask[i]].flatten()
                 all_preds.extend(valid_preds.tolist())
                 all_labels.extend(valid_labels.tolist())