From fce5e36cb0970db2a7392357a9abb03ac7767005 Mon Sep 17 00:00:00 2001 From: fanlu3 Date: Thu, 2 Jun 2022 21:44:18 +0800 Subject: [PATCH] fix modified_beam_search --- egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py | 1 + 1 file changed, 1 insertion(+) diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py b/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py index 97aa9a5cd..f9a03f336 100755 --- a/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py +++ b/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py @@ -283,6 +283,7 @@ def decode_one_batch( model=model, encoder_out=encoder_out, beam=params.beam_size, + encoder_out_lens=encoder_out_lens, ) for i in range(encoder_out.size(0)): hyps.append([lexicon.token_table[idx] for idx in hyp_tokens[i]])