From 63ea12459d4c6b962224d1798c6422c039c061e2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Thu, 7 Sep 2023 01:46:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20train=5Fms.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 train_ms.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/train_ms.py b/train_ms.py
index 1277eae..23f0252 100644
--- a/train_ms.py
+++ b/train_ms.py
@@ -32,7 +32,6 @@ from losses import generator_loss, discriminator_loss, feature_loss, kl_loss
 from mel_processing import mel_spectrogram_torch, spec_to_mel_torch
 from text.symbols import symbols
 
-os.environ["NCCL_P2P_DISABLE"] = 1  # Avoid training problem.
 torch.backends.cuda.matmul.allow_tf32 = True
 torch.backends.cudnn.allow_tf32 = (
     True  # If encontered training problem,please try to disable TF32.
@@ -76,12 +75,11 @@ def run():
     collate_fn = TextAudioSpeakerCollate()
     train_loader = DataLoader(
         train_dataset,
-        num_workers=16,
+        num_workers=0,
         shuffle=False,
         pin_memory=True,
         collate_fn=collate_fn,
         batch_sampler=train_sampler,
-        persistent_workers=True,
         prefetch_factor=4,
     )  # 128G Memory suitable loader.
     if rank == 0: