diff --git a/train_network.py b/train_network.py index 387b94b1..a4125e9f 100644 --- a/train_network.py +++ b/train_network.py @@ -930,7 +930,6 @@ class NetworkTrainer: if args.logging_dir is not None: - # logs = {"loss/epoch": loss_recorder.moving_average} logs = {"loss/epoch_average": loss_recorder.moving_average} accelerator.log(logs, step=epoch + 1)