From dbfe8fbb1a6cebee48b0000916a752770d4b95ec Mon Sep 17 00:00:00 2001 From: pkufool Date: Wed, 16 Mar 2022 11:53:03 +0800 Subject: [PATCH] Minor fixes --- egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py index c97359778..e5fad74c4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless/beam_search.py @@ -90,7 +90,7 @@ def fast_beam_search( logits = logits.squeeze(1).squeeze(1) log_probs = logits.log_softmax(dim=-1) decoding_streams.advance(log_probs) - decoding_streams.detach() + decoding_streams.terminate_and_flush_to_atreams() lattice = decoding_streams.format_output(encoder_out_lens.tolist()) best_path = one_best_decoding(lattice)