from local

This commit is contained in:
dohe0342 2023-02-03 08:47:49 +09:00
parent 1e4f9abd9d
commit b2ca1fda53
2 changed files with 1 additions and 1 deletions

View File

@ -472,7 +472,7 @@ def compute_validation_loss(
loss_value = tot_loss["loss"] / tot_loss["frames"]
if wb is not None:
wb.log({"valid/loss": loss_value.item()})
wb.log({"valid/loss": loss_value})
if params.cur_epoch < 10:
params.best_valid_losses[params.cur_epoch] = loss_value