from local

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

View File

@ -462,7 +462,7 @@ def compute_validation_loss(
loss_value = tot_loss["loss"] / tot_loss["frames"]
if loss.device == 0:
wb.log({"valid/loss": loss_value)
wb.log({"valid/loss": loss_value})
if params.cur_epoch < 10:
params.best_valid_losses[params.cur_epoch] = loss_value