From 95cfacc76c13eb2fdd0cf491130516a63bbbacbc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sun, 3 Sep 2023 12:36:53 +0800 Subject: [PATCH] try --- train_ms.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train_ms.py b/train_ms.py index 4669f3b..9afea31 100644 --- a/train_ms.py +++ b/train_ms.py @@ -147,10 +147,10 @@ def run(rank, n_gpus, hps): eps=hps.train.eps) 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]) if net_dur_disc is not None: - net_dur_disc = DDP(net_dur_disc, device_ids=[rank], find_unused_parameters=True) + net_dur_disc = DDP(net_dur_disc, device_ids=[rank]) pretrain_dir = None if pretrain_dir is None: