Skip to content

Commit f53ea4d

Browse files
fix typo
1 parent 75d645a commit f53ea4d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onmt/trainer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -323,7 +323,7 @@ def validate(self, valid_iter, moving_average=None):
323323
with_align=self.with_align,
324324
encode_tgt=self.encode_tgt)
325325
else:
326-
output, attns = valid_model(
326+
outputs, attns = valid_model(
327327
src, tgt, src_lengths,
328328
with_align=self.with_align)
329329
enc_src, enc_tgt = None, None
@@ -379,7 +379,7 @@ def _gradient_accumulation(self, true_batches, normalization, total_stats,
379379
src, tgt, src_lengths, bptt=bptt,
380380
with_align=self.with_align, encode_tgt=self.encode_tgt)
381381
else:
382-
output, attns = self.model(
382+
outputs, attns = self.model(
383383
src, tgt, src_lengths, bptt=bptt,
384384
with_align=self.with_align)
385385
enc_src, enc_tgt = None, None

0 commit comments

Comments
 (0)