From b54d9a256d936f71d751ff7d5b4b4c02216bcf2a Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Sun, 24 Apr 2022 15:25:34 +0800 Subject: [PATCH] Minor fixes. --- .../ASR/pruned_transducer_stateless2/beam_search.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py b/egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py index 10bd7bf7e..ad492aaa5 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py @@ -146,12 +146,7 @@ def fast_beam_search_nbest_oracle( nbest_scale=nbest_scale, ) - # We assume the labels of nbest.fsa are token IDs and the aux_labels - # are word IDs. - word_fsa = k2.invert(nbest.fsa) - word_ids = get_texts(word_fsa, return_ragged=True) - - hyps = k2.levenshtein_graph(word_ids) + hyps = nbest.build_levenshtein_graphs() refs = k2.levenshtein_graph(ref_texts, device=hyps.device) levenshtein_alignment = k2.levenshtein_alignment(