Merge pull request #5055 from pyf98/fix-maxlen

Fix the condition for maxlen warning in beam search
espnet · Mar 24, 2023 · c4d7063 · c4d7063
2 parents b579d70 + 7f4f0bd
commit c4d7063
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/espnet/nets/beam_search.py b/espnet/nets/beam_search.py
@@ -419,7 +419,7 @@ def forward(
                 + "".join([self.token_list[x] for x in best.yseq[1:-1]])
                 + "\n"
             )
-        if best.yseq[1:-1].shape[0] == x.shape[0]:
+        if best.yseq[1:-1].shape[0] == maxlen:
             logging.warning(
                 "best hypo length: {} == max output length: {}".format(
                     best.yseq[1:-1].shape[0], maxlen