Update train_ms.py

2023-08-20 16:48:39 +08:00
parent 2d04c4dd5b
commit 7cffaf2f08
1 changed files with 2 additions and 2 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -109,8 +109,8 @@ def run(rank, n_gpus, hps):
        hps.train.learning_rate,
        betas=hps.train.betas,
        eps=hps.train.eps)
-    net_g = DDP(net_g, device_ids=[rank])
-    net_d = DDP(net_d, device_ids=[rank])
+    net_g = DDP(net_g, device_ids=[rank],find_unused_parameters=True)
+    net_d = DDP(net_d, device_ids=[rank],find_unused_parameters=True)

    pretrain_dir = None
    if pretrain_dir is None: