from local

This commit is contained in:
dohe0342 2023-02-02 15:04:31 +09:00
parent 9ef29629ab
commit c3c62a87fd
2 changed files with 5 additions and 2 deletions

View File

@ -305,8 +305,11 @@ def save_checkpoint(
best_train_filename = params.exp_dir / "best-train-loss.pt"
copyfile(src=filename, dst=best_train_filename)
if params.best_valid_epoch == params.cur_epoch:
best_valid_filename = params.exp_dir / "best-valid-loss.pt"
#if params.best_valid_epoch == params.cur_epoch:
# best_valid_filename = params.exp_dir / "best-valid-loss.pt"
# copyfile(src=filename, dst=best_valid_filename)
if max(params.best_valid_losses.keys()) == params.cur_epoch:
best_valid_filename = params.exp_dir / f"best-valid-loss_{params.cur_epoch).pt"
copyfile(src=filename, dst=best_valid_filename)