corrected device assignment for non-cuda

IAHispano · Jan 25, 2025 · 9cfa0ae · 9cfa0ae
1 parent 8d6398b
commit 9cfa0ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -420,8 +420,8 @@ def run(
         net_g = net_g.cuda(device_id)
         net_d = net_d.cuda(device_id)
     else:
-        net_g.to(device)
-        net_d.to(device)
+        net_g = net_g.to(device)
+        net_d = net_d.to(device)
 
     if optimizer == "AdamW":
         optimizer = torch.optim.AdamW