Varun Wadhwa commited on
Commit
ae83b51
·
unverified ·
1 Parent(s): 5f8a11b
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -151,6 +151,11 @@ def evaluate_model(model, dataloader, device):
151
  # Process each sequence in the batch
152
  for i in range(current_batch_size):
153
  valid_mask = (labels[i] != -100) & attention_mask[i]
 
 
 
 
 
154
  valid_preds = preds[i][valid_mask[i]].flatten()
155
  valid_labels = labels[i][valid_mask[i]].flatten()
156
  all_preds.extend(valid_preds.tolist())
 
151
  # Process each sequence in the batch
152
  for i in range(current_batch_size):
153
  valid_mask = (labels[i] != -100) & attention_mask[i]
154
+ print(valid_mask.dtype)
155
+ print(labels[i].shape)
156
+ print(attention_mask[i].shape)
157
+ print(valid_mask.shape)
158
+ print(valid_labels)
159
  valid_preds = preds[i][valid_mask[i]].flatten()
160
  valid_labels = labels[i][valid_mask[i]].flatten()
161
  all_preds.extend(valid_preds.tolist())