malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

train.py: shuffling at every epoch

Browse Source
This commit is contained in:
rafaelvalle
2019-03-15 17:49:27 -07:00
parent bff304f432
commit f37998c59d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@@ -59,7 +59,7 @@ def prepare_dataloaders(hparams):
train_sampler = DistributedSampler(trainset) \ train_sampler = DistributedSampler(trainset) \
if hparams.distributed_run else None if hparams.distributed_run else None
train_loader = DataLoader(trainset, num_workers=1, shuffle=False, train_loader = DataLoader(trainset, num_workers=1, shuffle=True,
sampler=train_sampler, sampler=train_sampler,
batch_size=hparams.batch_size, pin_memory=False, batch_size=hparams.batch_size, pin_memory=False,
drop_last=True, collate_fn=collate_fn) drop_last=True, collate_fn=collate_fn)