Removed/cleanup a line

This commit is contained in:
rockerBOO
2023-11-05 16:56:36 -05:00
parent c89252101e
commit e545fdfd9a

View File

@@ -930,7 +930,6 @@ class NetworkTrainer:
if args.logging_dir is not None: if args.logging_dir is not None:
# logs = {"loss/epoch": loss_recorder.moving_average}
logs = {"loss/epoch_average": loss_recorder.moving_average} logs = {"loss/epoch_average": loss_recorder.moving_average}
accelerator.log(logs, step=epoch + 1) accelerator.log(logs, step=epoch + 1)