performance improve

2023-11-07 21:03:50 +08:00
parent 03b2ec09e1
commit af7b0ea0ff
1 changed files with 2 additions and 2 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -206,8 +206,8 @@ def run():
        )
    else:
        optim_dur_disc = None
-    net_g = DDP(net_g, device_ids=[rank], find_unused_parameters=True)
-    net_d = DDP(net_d, device_ids=[rank], find_unused_parameters=True)
+    net_g = DDP(net_g, device_ids=[rank])
+    net_d = DDP(net_d, device_ids=[rank])
    dur_resume_lr = None
    if net_dur_disc is not None:
        net_dur_disc = DDP(net_dur_disc, device_ids=[rank], find_unused_parameters=True)