upload files
This commit is contained in:
+1
-1
@@ -78,7 +78,7 @@ def run(rank, n_gpus, hps):
|
|||||||
rank=rank,
|
rank=rank,
|
||||||
shuffle=True)
|
shuffle=True)
|
||||||
collate_fn = TextAudioSpeakerCollate()
|
collate_fn = TextAudioSpeakerCollate()
|
||||||
train_loader = DataLoader(train_dataset, num_workers=8, shuffle=False, pin_memory=True,
|
train_loader = DataLoader(train_dataset, num_workers=0, shuffle=False, pin_memory=True,
|
||||||
collate_fn=collate_fn, batch_sampler=train_sampler)
|
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,
|
# train_loader = DataLoader(train_dataset, batch_size=hps.train.batch_size, num_workers=0, shuffle=False, pin_memory=True,
|
||||||
# collate_fn=collate_fn)
|
# collate_fn=collate_fn)
|
||||||
|
|||||||
Reference in New Issue
Block a user