Merge pull request ArkanDash#67 from onxlmao/fix/grad-norm-g-bug

BF667 · web-flow · commit 97fb420f4aaf · 2026-05-03T23:08:20.000+07:00
Fix UnboundLocalError: grad_norm_g undefined during gradient accumulation
diff --git a/advanced_rvc_inference/rvc/train/training/train.py b/advanced_rvc_inference/rvc/train/training/train.py
@@ -831,6 +831,9 @@ def train_and_evaluate(rank, epoch, hps, nets, optims, scaler, train_loader, wri
         # Optimization: scale loss by accumulation steps for correct gradient averaging
         loss_gen_all_scaled = loss_gen_all * loss_scale_factor
 
+        # Initialize grad_norm_g so it's always defined (needed for gradient accumulation)
+        grad_norm_g = 0.0
+
         # Optimization: set_to_none=True is faster than zeroing gradients
         optim_g.zero_grad(set_to_none=True)
         if autocast_enabled:
