Merge pull request #844 from ShiromiyaG/fix-multi-gpu-again

Fix Multi GPU train
IAHispano · Oct 27, 2024 · 206b6a2 · 206b6a2
2 parents eb74578 + 3069c1c
commit 206b6a2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -732,7 +732,10 @@ def train_and_evaluate(
         }
 
         with torch.no_grad():
-            o, *_ = net_g.infer(*reference)
+            if hasattr(net_g, "module"):
+                o, *_ = net_g.module.infer(*reference)
+            else:
+                o, *_ = net_g.infer(*reference)
         audio_dict = {f"gen/audio_{global_step:07d}": o[0, :, :]}
 
         summarize(