From 67f4d08520e002d75c30498c629a8da11842a11b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Thu, 7 Sep 2023 22:15:15 +0800
Subject: [PATCH] fix

---
 train_ms.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train_ms.py b/train_ms.py
index 1a5660777..11fffced0 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -72,8 +72,6 @@ def run():
         num_replicas=n_gpus,
         rank=rank,
         shuffle=True,
-        persistent_workers=True,
-        prefetch_factor=4,
     )
     collate_fn = TextAudioSpeakerCollate()
     train_loader = DataLoader(
@@ -83,6 +81,8 @@ def run():
         pin_memory=True,
         collate_fn=collate_fn,
         batch_sampler=train_sampler,
+        persistent_workers=True,
+        prefetch_factor=4,
     )  # DataLoader config could be adjusted.
     if rank == 0:
         eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data)