from local

This commit is contained in:
dohe0342 2023-02-02 14:56:34 +09:00
parent 3308d26968
commit c7c56dba09
2 changed files with 4 additions and 2 deletions

View File

@ -440,13 +440,15 @@ def compute_validation_loss(
loss_value = tot_loss["loss"] / tot_loss["frames"] loss_value = tot_loss["loss"] / tot_loss["frames"]
if params.cur_epoch < 10: if params.cur_epoch < 10:
params.best_valid_losses.append(loss_value) #params.best_valid_losses.append(loss_value)
params.best_valid_epochs.append(params.cur_epoch) #params.best_valid_epochs.append(params.cur_epoch)
params.best_valid_losses[params.cur_epoch] = loss_value
#if loss_value < params.best_valid_loss: #if loss_value < params.best_valid_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): if params.cur_epoch >= 10 and loss_value < max(params.best_valid_losses):
params.best_valid
return tot_loss return tot_loss