fix step number

vince62s · Dec 18, 2023 · 05132fb · 05132fb
1 parent 383ea06
commit 05132fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onmt/translate/translator.py b/onmt/translate/translator.py
@@ -1112,7 +1112,7 @@ def _translate_batch_with_strategy(self, batch, decode_strategy, left_pad=True):
                 batch,
                 src_len=decode_strategy.src_len,
                 src_map=src_map,
-                step=step if step == 0 else step + max(src_len.tolist()),
+                step=step if step == 0 else step + max(src_len.tolist()) - 1,
                 batch_offset=decode_strategy.batch_offset,
             )