mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
771848696f
commit
a941fb80e6
Binary file not shown.
@ -788,6 +788,8 @@ def main() -> None:
|
|||||||
|
|
||||||
test_sets = ["dev"]
|
test_sets = ["dev"]
|
||||||
test_dls = [valid_dl]
|
test_dls = [valid_dl]
|
||||||
|
|
||||||
|
wer_dict = {}
|
||||||
|
|
||||||
for epoch in range(params.start, params.end+1):
|
for epoch in range(params.start, params.end+1):
|
||||||
load_checkpoint(f"{params.exp_dir}/epoch-{epoch}.pt", model)
|
load_checkpoint(f"{params.exp_dir}/epoch-{epoch}.pt", model)
|
||||||
@ -814,8 +816,6 @@ def main() -> None:
|
|||||||
|
|
||||||
save_results(params=params, test_set_name=test_set, results_dict=results_dict)
|
save_results(params=params, test_set_name=test_set, results_dict=results_dict)
|
||||||
|
|
||||||
logging.info("Done!")
|
|
||||||
|
|
||||||
|
|
||||||
torch.set_num_threads(1)
|
torch.set_num_threads(1)
|
||||||
# when we import add_model_arguments from train.py
|
# when we import add_model_arguments from train.py
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user