from local

This commit is contained in:
dohe0342 2023-02-03 01:30:14 +09:00
parent 2bd4227889
commit a6dcaff260
2 changed files with 1 additions and 1 deletions

View File

@ -480,7 +480,7 @@ def compute_validation_loss(
try: os.remove(remove_filename) try: os.remove(remove_filename)
except: pass except: pass
del params.best_valid_losses[k] del params.best_valid_losses[remove_epoch]
params.best_valid_losses[params.cur_epoch] = loss_value params.best_valid_losses[params.cur_epoch] = loss_value
return tot_loss return tot_loss