From 356efb724f845d898b21b6b3fcdb59cd02032a1c 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:21:43 +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 f2bf1a3..ad57ef1 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -141,7 +141,7 @@ def run(rank, n_gpus, hps):
         betas=hps.train.betas,
         eps=hps.train.eps)
     if net_dur_disc is not None:
-        optim_dur_disc = torch.optim.AdanW(
+        optim_dur_disc = torch.optim.AdamW(
         net_dur_disc.parameters(),
         hps.train.learning_rate,
         betas=hps.train.betas,