From da6736fbf7403c2feb71cc05e3ea05ad87a006ef Mon Sep 17 00:00:00 2001 From: Plachta Date: Thu, 16 Feb 2023 20:20:06 +0800 Subject: [PATCH] upload files --- finetune_speaker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune_speaker.py b/finetune_speaker.py index 5bd185e..1679741 100644 --- a/finetune_speaker.py +++ b/finetune_speaker.py @@ -78,7 +78,7 @@ def run(rank, n_gpus, hps): rank=rank, shuffle=True) collate_fn = TextAudioSpeakerCollate() - train_loader = DataLoader(train_dataset, num_workers=0, shuffle=False, pin_memory=True, + train_loader = DataLoader(train_dataset, num_workers=8, shuffle=False, pin_memory=True, collate_fn=collate_fn, batch_sampler=train_sampler) # train_loader = DataLoader(train_dataset, batch_size=hps.train.batch_size, num_workers=0, shuffle=False, pin_memory=True, # collate_fn=collate_fn)