fix lm out[n_exs]
@@ -652,7 +652,7 @@ class MultiTaskModel(nn.Module):
out = {}
b_size, seq_len = batch['targs']['words'].size()
sent_encoder = self.sent_encoder
- out['n_exs'] = get_batch_size(batch['input'])
+ out['n_exs'] = b_size #get_batch_size(batch['input'])
if not isinstance(sent_encoder, BiLMEncoder):
sent, mask = sent_encoder(batch['input'])
