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-09-07 11:09:55 +08:00
committed by GitHub
parent 63ea12459d
commit 1069066390
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
train_ms.py
View File

@@ -80,8 +80,7 @@ def run():
pin_memory=True, pin_memory=True,
collate_fn=collate_fn, collate_fn=collate_fn,
batch_sampler=train_sampler, batch_sampler=train_sampler,
prefetch_factor=4, ) # 0 worker
) # 128G Memory suitable loader.
if rank == 0: if rank == 0:
eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data) eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data)
eval_loader = DataLoader( eval_loader = DataLoader(