Update train_ms.py

fishaudio · Sep 7, 2023 · 1069066 · 1069066
1 parent 63ea124
commit 1069066
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train_ms.py b/train_ms.py
@@ -80,8 +80,7 @@ def run():
         pin_memory=True,
         collate_fn=collate_fn,
         batch_sampler=train_sampler,
-        prefetch_factor=4,
-    )  # 128G Memory suitable loader.
+    )  # 0 worker
     if rank == 0:
         eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data)
         eval_loader = DataLoader(