blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
RVC-Boss 2024-08-06 12:03:05 +08:00 committed by GitHub
parent a4fbd65ecb
commit 32a46d9494
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
GPT_SoVITS/s2_train.py
View File

@ -119,7 +119,7 @@ def run(rank, n_gpus, hps):
collate_fn=collate_fn, collate_fn=collate_fn,
batch_sampler=train_sampler, batch_sampler=train_sampler,
persistent_workers=True, persistent_workers=True,
prefetch_factor=16, prefetch_factor=4,
) )
# if rank == 0: # if rank == 0:
# eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data, val=True) # eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data, val=True)
@ -211,7 +211,7 @@ def run(rank, n_gpus, hps):
# traceback.print_exc() # traceback.print_exc()
epoch_str = 1 epoch_str = 1
global_step = 0 global_step = 0
if hps.train.pretrained_s2G != "": if hps.train.pretrained_s2G != ""and hps.train.pretrained_s2G != None and os.path.exists(hps.train.pretrained_s2G):
if rank == 0: if rank == 0:
logger.info("loaded pretrained %s" % hps.train.pretrained_s2G) logger.info("loaded pretrained %s" % hps.train.pretrained_s2G)
print( print(
@ -223,7 +223,7 @@ def run(rank, n_gpus, hps):
strict=False, strict=False,
) )
) ##测试不加载优化器 ) ##测试不加载优化器
if hps.train.pretrained_s2D != "": if hps.train.pretrained_s2D != ""and hps.train.pretrained_s2D != None and os.path.exists(hps.train.pretrained_s2D):
if rank == 0: if rank == 0:
logger.info("loaded pretrained %s" % hps.train.pretrained_s2D) logger.info("loaded pretrained %s" % hps.train.pretrained_s2D)
print( print(