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

Update

Browse Source
This commit is contained in:
litagin02
2024-02-02 21:11:23 +09:00
parent 7a8477b9bc
commit e566ac62f3
17 changed files with 1535 additions and 1162 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
train_ms.py
View File

@@ -1,6 +1,5 @@
import argparse
import datetime
import gc
import os
import platform
@@ -299,9 +298,9 @@ def run():
utils.latest_checkpoint_path(model_dir, "DUR_*.pth"),
net_dur_disc,
optim_dur_disc,
skip_optimizer=hps.train.skip_optimizer
if "skip_optimizer" in hps.train
else True,
skip_optimizer=(
hps.train.skip_optimizer if "skip_optimizer" in hps.train else True
),
)
if not optim_dur_disc.param_groups[0].get("initial_lr"):
optim_dur_disc.param_groups[0]["initial_lr"] = dur_resume_lr
@@ -309,17 +308,17 @@ def run():
utils.latest_checkpoint_path(model_dir, "G_*.pth"),
net_g,
optim_g,
skip_optimizer=hps.train.skip_optimizer
if "skip_optimizer" in hps.train
else True,
skip_optimizer=(
hps.train.skip_optimizer if "skip_optimizer" in hps.train else True
),
)
_, optim_d, d_resume_lr, epoch_str = utils.load_checkpoint(
utils.latest_checkpoint_path(model_dir, "D_*.pth"),
net_d,
optim_d,
skip_optimizer=hps.train.skip_optimizer
if "skip_optimizer" in hps.train
else True,
skip_optimizer=(
hps.train.skip_optimizer if "skip_optimizer" in hps.train else True
),
)
if not optim_g.param_groups[0].get("initial_lr"):
optim_g.param_groups[0]["initial_lr"] = g_resume_lr