From 8a3068ead8da3c28baf3ac2ff6d1dc9eca90bcfe Mon Sep 17 00:00:00 2001 From: fanlu Date: Sat, 4 Jun 2022 19:08:17 +0800 Subject: [PATCH] Update decode.py (#392) * Update decode.py fix bug ```TypeError: greedy_search_batch() missing 1 required positional argument: 'encoder_out_lens'``` * fix modified_beam_search Co-authored-by: fanlu3 --- egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py b/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py index be3d01f6a..f9a03f336 100755 --- a/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py +++ b/egs/wenetspeech/ASR/pruned_transducer_stateless2/decode.py @@ -274,6 +274,7 @@ def decode_one_batch( hyp_tokens = greedy_search_batch( model=model, encoder_out=encoder_out, + 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]]) @@ -282,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]])