From 1b854e5c44c03a5bea48015d9718fefacf394b52 Mon Sep 17 00:00:00 2001 From: Mingshuang Luo <37799481+luomingshuang@users.noreply.github.com> Date: Mon, 11 Apr 2022 21:14:02 +0800 Subject: [PATCH] Update beam_search.py --- egs/tedlium3/ASR/pruned_transducer_stateless/beam_search.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/egs/tedlium3/ASR/pruned_transducer_stateless/beam_search.py b/egs/tedlium3/ASR/pruned_transducer_stateless/beam_search.py index 0ae001d3f..3a08b100d 100644 --- a/egs/tedlium3/ASR/pruned_transducer_stateless/beam_search.py +++ b/egs/tedlium3/ASR/pruned_transducer_stateless/beam_search.py @@ -486,7 +486,9 @@ def modified_beam_search( for i in range(batch_size): topk_log_probs, topk_indexes = ragged_log_probs[i].topk(beam) - topk_hyp_indexes = torch.div(topk_indexes, vocab_size, rounding_mode="trunc") + topk_hyp_indexes = torch.div( + topk_indexes, vocab_size, rounding_mode="trunc" + ) topk_hyp_indexes = topk_hyp_indexes.tolist() topk_token_indexes = (topk_indexes % vocab_size).tolist()