Logs

app.py

@@ -156,6 +156,7 @@ def evaluate_model(model, dataloader, device):
                 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())