From 9bf2856eb366fbf9623590c2222c14d1af7972ce Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20Hernandez?= Date: Mon, 17 Feb 2020 18:41:13 +0100 Subject: [PATCH] fix typo --- onmt/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onmt/trainer.py b/onmt/trainer.py index 973a5e1848..0ba703abfc 100644 --- a/onmt/trainer.py +++ b/onmt/trainer.py @@ -379,7 +379,7 @@ def _gradient_accumulation(self, true_batches, normalization, total_stats, src, tgt, src_lengths, bptt=bptt, with_align=self.with_align, encode_tgt=self.encode_tgt) else: - output, attns = self.model( + outputs, attns = self.model( src, tgt, src_lengths, bptt=bptt, with_align=self.with_align) enc_src, enc_tgt = None, None