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

Merge branch 'dev/aishell-zipformer-bbpe' of https://github.com/jinzr/icefall into dev/aishell-zipformer-bbpe

Browse Source
This commit is contained in:
JinZr 2024-01-11 09:56:01 +08:00
commit bce81394a4
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/aishell/ASR/zipformer_bbpe/decode.py
View File

@ -584,9 +584,14 @@ def save_results(
errs_filename = ( errs_filename = (
params.res_dir / f"errs-{test_set_name}-{key}-{params.suffix}.txt" params.res_dir / f"errs-{test_set_name}-{key}-{params.suffix}.txt"
) )
results_char = []
for res in results:
results_char.append((res[0], list("".join(res[1])), list("".join(res[2]))))
with open(errs_filename, "w") as f: with open(errs_filename, "w") as f:
wer = write_error_stats( wer = write_error_stats(
f, f"{test_set_name}-{key}", results, enable_log=True f, f"{test_set_name}-{key}", results_char, enable_log=True
) )
test_set_wers[key] = wer test_set_wers[key] = wer