from local

This commit is contained in:
dohe0342 2023-02-02 15:11:04 +09:00
parent 987c8a3911
commit a6d4ce1256
2 changed files with 1 additions and 0 deletions

View File

@ -453,6 +453,7 @@ def compute_validation_loss(
if v == max(params.best_valid_losses.values()):
remove_filename = params.exp_dir / f"best-valid-loss_{k}.pt"
if loss.device == 0: os.remove(remove_filename)
del params.best_valid_losses[k]
params.best_valid_losses[params.cur_epoch] = loss_value
return tot_loss