Update train_ms.py

2023-08-23 17:21:43 +08:00
parent e9223872c2
commit 356efb724f
1 changed files with 1 additions and 1 deletions
--- 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,