from local

This commit is contained in:
dohe0342 2023-02-02 14:52:27 +09:00
parent 7f6bb0e051
commit 6db861e89d
2 changed files with 3 additions and 0 deletions

View File

@ -439,6 +439,9 @@ def compute_validation_loss(
tot_loss.reduce(loss.device) tot_loss.reduce(loss.device)
loss_value = tot_loss["loss"] / tot_loss["frames"] loss_value = tot_loss["loss"] / tot_loss["frames"]
if params.cur_epoch < 10:
params.best_valid_losses.append(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