mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-28 19:24:17 +00:00
Minor fixes.
This commit is contained in:
parent
3bcdecda44
commit
940aedf7ce
@ -479,14 +479,13 @@ def decode_dataset(
|
||||
assert (
|
||||
len(results) > 0
|
||||
), "It should not decode to empty in the first batch!"
|
||||
keys = list(results.keys())
|
||||
for lm_scale in keys:
|
||||
this_batch = []
|
||||
hyp_words = []
|
||||
for ref_text in texts:
|
||||
ref_words = ref_text.split()
|
||||
this_batch.append((ref_words, hyp_words))
|
||||
|
||||
for lm_scale in results.keys():
|
||||
results[lm_scale].extend(this_batch)
|
||||
|
||||
num_cuts += len(texts)
|
||||
|
Loading…
x
Reference in New Issue
Block a user