RVC/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-01-31 08:20:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format code (#275)

Browse Source 
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
2023-05-14 07:52:36 +00:00
committed by GitHub
parent 32437314b8
commit 6a3eaef090
8 changed files with 295 additions and 158 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
train_nsf_sim_cache_sid_load_pretrain.py
View File

@@ -31,7 +31,8 @@ from data_utils import (
TextAudioCollate,
DistributedBucketSampler,
)
if(hps.version=="v1"):
if hps.version == "v1":
from infer_pack.models import (
SynthesizerTrnMs256NSFsid as RVC_Model_f0,
SynthesizerTrnMs256NSFsid_nono as RVC_Model_nof0,
@@ -519,14 +520,25 @@ def train_and_evaluate(
epoch,
os.path.join(hps.model_dir, "D_{}.pth".format(2333333)),
)
if(rank==0 and hps.save_every_weights=="1"):
if rank == 0 and hps.save_every_weights == "1":
if hasattr(net_g, "module"):
ckpt = net_g.module.state_dict()
else:
ckpt = net_g.state_dict()
logger.info(
"saving ckpt %s_e%s:%s"
% (hps.name,epoch,savee(ckpt, hps.sample_rate, hps.if_f0, hps.name+"_e%s"%epoch, epoch,hps.version))
% (
hps.name,
epoch,
savee(
ckpt,
hps.sample_rate,
hps.if_f0,
hps.name + "_e%s" % epoch,
epoch,
hps.version,
),
)
)
if rank == 0:
@@ -540,7 +552,7 @@ def train_and_evaluate(
ckpt = net_g.state_dict()
logger.info(
"saving final ckpt:%s"
% (savee(ckpt, hps.sample_rate, hps.if_f0, hps.name, epoch,hps.version))
% (savee(ckpt, hps.sample_rate, hps.if_f0, hps.name, epoch, hps.version))
)
sleep(1)
os._exit(2333333)