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,