From ffdf161830b7528251e609aa6f41b1970d27f388 Mon Sep 17 00:00:00 2001 From: fanlu Date: Thu, 2 Jun 2022 20:56:34 +0800 Subject: [PATCH] Update decode.py fix bug ```TypeError: greedy_search_batch() missing 1 required positional argument: 'encoder_out_lens'``` --- 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 be3d01f6a..97aa9a5cd 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]])