[easy] another n_exs fix for MT validation
@@ -611,7 +611,7 @@ class MultiTaskModel(nn.Module):
if isinstance(task, MTTask):
decoder = getattr(self, "%s_decoder" % task.name)
- out = decoder.forward(sent, sent_mask, batch['targs'])
+ out.update(decoder.forward(sent, sent_mask, batch['targs']))
task.scorer1(math.exp(out['loss'].item()))
return out
