Update train_ms.py

2023-09-07 11:09:55 +08:00
parent 63ea12459d
commit 1069066390
1 changed files with 1 additions and 2 deletions
--- 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(