From 1069066390443ee856865f5d652ef6517dfbff2a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Thu, 7 Sep 2023 11:09:55 +0800 Subject: [PATCH] Update train_ms.py --- train_ms.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train_ms.py b/train_ms.py index 23f0252..37f9087 100644 --- 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(