from local

This commit is contained in:
dohe0342 2023-02-02 15:01:32 +09:00
parent 44baa3ef47
commit 4dc44cc4d0
2 changed files with 4 additions and 2 deletions

View File

@ -446,7 +446,9 @@ def compute_validation_loss(
# params.best_valid_epoch = params.cur_epoch # params.best_valid_epoch = params.cur_epoch
# params.best_valid_loss = loss_value # params.best_valid_loss = loss_value
if params.cur_epoch >= 10 and loss_value < max(params.best_valid_losses.values()): if params.cur_epoch >= 10 and loss_value < max(params.best_valid_losses.values()):
for k, v in params.best_valid_losses.items():
if v == max(params.best_valid_losses.values()):
params.best_valid_losses[k] = v
return tot_loss return tot_loss