from local

This commit is contained in:
dohe0342 2023-02-03 02:30:02 +09:00
parent 4d82da2317
commit 1e4f9abd9d
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"] loss_value = tot_loss["loss"] / tot_loss["frames"]
if wb is not None: if wb is not None:
wb.log({"valid/loss": loss_value}) wb.log({"valid/loss": loss_value.item()})
if params.cur_epoch < 10: if params.cur_epoch < 10:
params.best_valid_losses[params.cur_epoch] = loss_value params.best_valid_losses[params.cur_epoch] = loss_value