fix

2023-09-03 13:14:55 +08:00
parent d9a5696cf4
commit c18837481d
1 changed files with 2 additions and 2 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -78,8 +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,drop_last=True)  #256G Memory suitable loader.
+                              collate_fn=collate_fn, batch_sampler=train_sampler,
+                              persistent_workers=True,prefetch_factor=4)  #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,