mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-14 15:18:18 +09:00
parent 4527fabfe2
commit a568e37484
2 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.decode_multi.py.swp
View File

Binary file not shown.

14
egs/tedlium2/ASR/conformer_ctc3/decode_multi.py
View File

@ -627,15 +627,16 @@ def save_results(
print("settings\tWER", file=f)
for key, val in test_set_wers:
print("{}\t{}".format(key, val), file=f)
wer = None
s = "\nFor {}, WER of different settings are:\n".format(test_set_name)
note = "\tbest for {}".format(test_set_name)
for key, val in test_set_wers:
s += "{}\t{}{}\n".format(key, val, note)
note = ""
print(key)
print(val)
wer = val
logging.info(s)
return val
@torch.no_grad()
@ -816,7 +817,12 @@ def main() -> None:
eos_id=eos_id,
)
save_results(params=params, test_set_name=test_set, results_dict=results_dict)
wer = save_results(params=params, test_set_name=test_set, results_dict=results_dict)
wer_dict[epoch] = wer
wer_dict = sorted(wer_dict.items(), key=lambda x:x[1])
for k, v in wer_dict:
print(k, v)
torch.set_num_threads(1)
# when we import add_model_arguments from train.py