Merge pull request #806 from ShiromiyaG/fix-multi-gpu-train

Fix multi GPU train
IAHispano · Oct 11, 2024 · b1e9ef2 · b1e9ef2
2 parents 8e02e1c + ca73f08
commit b1e9ef2
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -795,7 +795,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(