mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 22:54:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-04-24 15:25:34 +08:00
parent b1c3705fbe
commit b54d9a256d
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/librispeech/ASR/pruned_transducer_stateless2/beam_search.py
View File

@ -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(