tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update train_ms.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-23 17:21:43 +08:00
committed by GitHub
parent e9223872c2
commit 356efb724f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_ms.py
View File

@@ -141,7 +141,7 @@ def run(rank, n_gpus, hps):
betas=hps.train.betas, betas=hps.train.betas,
eps=hps.train.eps) eps=hps.train.eps)
if net_dur_disc is not None: if net_dur_disc is not None:
optim_dur_disc = torch.optim.AdanW( optim_dur_disc = torch.optim.AdamW(
net_dur_disc.parameters(), net_dur_disc.parameters(),
hps.train.learning_rate, hps.train.learning_rate,
betas=hps.train.betas, betas=hps.train.betas,