from local

This commit is contained in:
dohe0342 2023-02-02 15:48:28 +09:00
parent f0537dbeaf
commit d8bfb903db
2 changed files with 2 additions and 1 deletions

View File

@ -450,10 +450,11 @@ def compute_validation_loss(
# params.best_valid_loss = loss_value
if params.cur_epoch >= 2 and loss_value < max(params.best_valid_losses.values()):
remove_epoch = -1
params.best_valid_losses[params.cur_epoch] = loss_value
for k, v in params.best_valid_losses.items():
if v == max(params.best_valid_losses.values()):
remove_epoch = k
params.best_valid_losses[params.cur_epoch] = loss_value
remove_filename = params.exp_dir / f"best-valid-loss_{remove_epoch}.pt"
if loss.device == 0: os.remove(remove_filename)