upload files

2023-02-16 20:20:06 +08:00
parent cbd32eb8b5
commit da6736fbf7
1 changed files with 1 additions and 1 deletions
@@ -78,7 +78,7 @@ def run(rank, n_gpus, hps):
      rank=rank,
      shuffle=True)
  collate_fn = TextAudioSpeakerCollate()
-  train_loader = DataLoader(train_dataset, num_workers=0, shuffle=False, pin_memory=True,
+  train_loader = DataLoader(train_dataset, num_workers=8, shuffle=False, pin_memory=True,
      collate_fn=collate_fn, batch_sampler=train_sampler)
  # train_loader = DataLoader(train_dataset, batch_size=hps.train.batch_size, num_workers=0, shuffle=False, pin_memory=True,
  #                           collate_fn=collate_fn)