From 5689a4cfc1eacf8c4ee3f7f072883f90da4e4079 Mon Sep 17 00:00:00 2001 From: jinzr Date: Thu, 11 Jan 2024 09:55:28 +0800 Subject: [PATCH] Update decode.py --- egs/aishell/ASR/zipformer_bbpe/decode.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/egs/aishell/ASR/zipformer_bbpe/decode.py b/egs/aishell/ASR/zipformer_bbpe/decode.py index 543ab84c8..14b67d5e0 100755 --- a/egs/aishell/ASR/zipformer_bbpe/decode.py +++ b/egs/aishell/ASR/zipformer_bbpe/decode.py @@ -585,9 +585,14 @@ def save_results( errs_filename = ( 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: 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