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

Fix an error.

Browse Source
This commit is contained in:
Fangjun Kuang 2022-05-16 22:31:36 +08:00
parent 3d833d9430
commit 8301faeb12
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -361,9 +361,10 @@ def fast_beam_search_with_nbest_rescoring(
oov_word_id = word_table[oov_word] oov_word_id = word_table[oov_word]
word_ids_list: List[List[int]] = [] word_ids_list: List[List[int]] = []
for words in word_list: for words in word_list:
this_word_ids = [] this_word_ids = []
for w in words: for w in words.split():
if w in word_table: if w in word_table:
this_word_ids.append(word_table[w]) this_word_ids.append(word_table[w])
else: else: