From 031a543150d0829359db084a0f1e7258dd8acead Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com> Date: Wed, 23 Aug 2023 17:17:47 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_ms.py b/train_ms.py index 7299116..4e6fad9 100644 --- a/train_ms.py +++ b/train_ms.py @@ -140,7 +140,7 @@ def run(rank, n_gpus, hps): hps.train.learning_rate, betas=hps.train.betas, eps=hps.train.eps) - if net_dur_disc is not None: + if net_dur_disc is not None: optim_dur_disc = torch.optim.AdanW( net_dur_disc.parameters(), hps.train.learning_rate,