from local

This commit is contained in:
dohe0342 2023-02-14 15:15:22 +09:00
parent 771848696f
commit a941fb80e6
2 changed files with 2 additions and 2 deletions

View File

@ -789,6 +789,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