from local

This commit is contained in:
dohe0342 2023-02-03 01:27:07 +09:00
parent dd0eb1a3cb
commit 46ec2c8312
2 changed files with 1 additions and 0 deletions

View File

@ -550,6 +550,7 @@ def train_one_epoch(
f"batch {batch_idx}, loss[{loss_info}], "
f"tot_loss[{tot_loss}], batch size: {batch_size}"
)
wb.log({"train/loss": tot_loss})
if batch_idx % params.log_interval == 0: