From cbd267f9325b37251b1b82d3fccc812032c23aac Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Sat, 26 Aug 2023 23:35:33 +0800
Subject: [PATCH] =?UTF-8?q?dur=E8=AE=AD=E7=BB=83=E5=A4=9A=E5=8D=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train_ms.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/train_ms.py b/train_ms.py
index e2c7149..d2a4dd8 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -152,6 +152,8 @@ def run(rank, n_gpus, hps):
         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)
+    if net_dur_disc is not None:
+        net_dur_disc = DDP(net_dur_disc, device_ids=[rank], find_unused_parameters=True)
 
     pretrain_dir = None
     if pretrain_dir is None: