from local

This commit is contained in:
dohe0342 2023-02-03 02:25:59 +09:00
parent 18fd718501
commit 4d82da2317
2 changed files with 2 additions and 2 deletions

View File

@ -470,8 +470,8 @@ 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 loss.device == 0: if wb is not None:
wb.log({"valid/loss": loss_value}) wb.log({"valid/loss": loss_value})
if params.cur_epoch < 10: if params.cur_epoch < 10: