from local

This commit is contained in:
dohe0342 2023-02-14 15:35:33 +09:00
parent b4f4274693
commit 49e80c3713
2 changed files with 8 additions and 2 deletions

View File

@ -821,8 +821,14 @@ def main() -> None:
wer_dict[epoch] = wer
wer_dict = sorted(wer_dict.items(), key=lambda x:x[1])
for k, v in wer_dict:
print(k, v)
import shutil
import os
if not os.makedir(
for i, (k, v) in enumerate(wer_dict):
shutil.copy(f"{params.exp_dir}/epoch-{k}.pt", f"{params.exp_dir}/sorted/epoch-{100-i}.pt")
if i > 20:
break
torch.set_num_threads(1)
# when we import add_model_arguments from train.py