Varun Wadhwa commited on
Commit
2293530
·
unverified ·
1 Parent(s): 9d4c2df
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -128,11 +128,11 @@ def evaluate_model(model, dataloader, device):
128
  attention_mask = batch['attention_mask'].to(device)
129
  labels = batch['labels'].to(device).cpu().numpy()
130
  x = len(labels[0])
131
- print(labels[0])
 
132
  for l in labels:
133
  if len(l) != x:
134
  print(len(l))
135
- print(l)
136
  break
137
 
138
 
 
128
  attention_mask = batch['attention_mask'].to(device)
129
  labels = batch['labels'].to(device).cpu().numpy()
130
  x = len(labels[0])
131
+ print(x)
132
+ print("OTHERS:")
133
  for l in labels:
134
  if len(l) != x:
135
  print(len(l))
 
136
  break
137
 
138