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

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-11-11 19:35:53 +08:00
parent 3bcdecda44
commit 940aedf7ce
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/conformer_ctc/decode.py
View File

@ -479,14 +479,13 @@ def decode_dataset(
assert ( assert (
len(results) > 0 len(results) > 0
), "It should not decode to empty in the first batch!" ), "It should not decode to empty in the first batch!"
keys = list(results.keys())
for lm_scale in keys:
this_batch = [] this_batch = []
hyp_words = [] hyp_words = []
for ref_text in texts: for ref_text in texts:
ref_words = ref_text.split() ref_words = ref_text.split()
this_batch.append((ref_words, hyp_words)) this_batch.append((ref_words, hyp_words))
for lm_scale in results.keys():
results[lm_scale].extend(this_batch) results[lm_scale].extend(this_batch)
num_cuts += len(texts) num_cuts += len(texts)