Remove progress bar avg norms

2026-04-08 22:35:09 +00:00 · 2025-03-27 03:28:33 -04:00
parent 182544dcce
commit 0181b7a042
1 changed files with 1 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -1432,7 +1432,7 @@ class NetworkTrainer:
                        weight_norms = network.weight_norms()
                        maximum_norm = weight_norms.max().item() if weight_norms.numel() > 0 else None
                        keys_scaled = None
-                        max_mean_logs = {"avg weight norm": mean_norm, "avg grad norm": mean_grad_norm, "avg comb norm": mean_combined_norm}
+                        max_mean_logs = {}
                    else:
                        keys_scaled, mean_norm, maximum_norm = None, None, None
                        mean_grad_norm = None