From 1835ed244269b263fff31ac51ec30e576673fbcc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Sun, 3 Sep 2023 12:35:58 +0800 Subject: [PATCH] add drop_last --- train_ms.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/train_ms.py b/train_ms.py index 93846c2..4669f3b 100644 --- a/train_ms.py +++ b/train_ms.py @@ -78,7 +78,8 @@ def run(rank, n_gpus, hps): shuffle=True) collate_fn = TextAudioSpeakerCollate() train_loader = DataLoader(train_dataset, num_workers=24, shuffle=False, pin_memory=True, - collate_fn=collate_fn, batch_sampler=train_sampler, persistent_workers=True,prefetch_factor=4) + collate_fn=collate_fn, batch_sampler=train_sampler, persistent_workers=True, + prefetch_factor=4,drop_last=True) #256G Memory suitable loader. if rank == 0: eval_dataset = TextAudioSpeakerLoader(hps.data.validation_files, hps.data) eval_loader = DataLoader(eval_dataset, num_workers=0, shuffle=False,