From 0630c258ea5574c1e47fa38100a6543a76145e82 Mon Sep 17 00:00:00 2001 From: Plachta Date: Thu, 9 Mar 2023 13:22:27 +0800 Subject: [PATCH] changed num_workers --- finetune_speaker_v2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune_speaker_v2.py b/finetune_speaker_v2.py index 90b1b7c..85fa044 100644 --- a/finetune_speaker_v2.py +++ b/finetune_speaker_v2.py @@ -79,7 +79,7 @@ def run(rank, n_gpus, hps): rank=rank, shuffle=True) collate_fn = TextAudioSpeakerCollate() - train_loader = DataLoader(train_dataset, num_workers=8, shuffle=False, pin_memory=True, + train_loader = DataLoader(train_dataset, num_workers=2, 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=2, shuffle=False, pin_memory=True, # collate_fn=collate_fn)