Add n_exs to single-sentence forward
@@ -428,6 +428,7 @@ class MultiTaskModel(nn.Module):
classifier = getattr(self, "%s_mdl" % task.name)
logits = classifier(sent_embs, sent_mask)
out['logits'] = logits
+ out['n_exs'] = get_batch_size_from_field(batch['input1'])
if 'labels' in batch: # means we should compute loss
labels = batch['labels'].squeeze(-1)
