Merge branch 'main' into dev

Vitalya-79 · Jun 6, 2023 · 98635eb · 98635eb
2 parents 24823b0 + 0fe1afd
commit 98635eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_network.py b/train_network.py
@@ -724,7 +724,7 @@ def remove_model(old_ckpt_name):
             progress_bar.set_postfix(**logs)
 
             if args.scale_weight_norms:
-                progress_bar.set_postfix(**max_mean_logs)
+                progress_bar.set_postfix(**{**max_mean_logs,**logs})
 
             if args.logging_dir is not None:
                 logs = generate_step_logs(args, current_loss, avr_loss, lr_scheduler, keys_scaled, mean_norm, maximum_norm)