malarinv
/
tacotron2
mirror of https://github.com/malarinv/tacotron2
1
0
Fork 0
Code Issues Releases Wiki Activity

Merge branch 'master' of https://github.com/NVIDIA/tacotron2

experiments
rafaelvalle 2019-04-03 13:51:59 -08:00
parent 821bfeba5d ece7d3f568
commit eb2a171690
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
train.py
View File

@ -45,10 +45,14 @@ def prepare_dataloaders(hparams):
valset = TextMelLoader(hparams.validation_files, hparams) valset = TextMelLoader(hparams.validation_files, hparams)
collate_fn = TextMelCollate(hparams.n_frames_per_step) collate_fn = TextMelCollate(hparams.n_frames_per_step)
train_sampler = DistributedSampler(trainset) \ if hparams.distributed_run:
if hparams.distributed_run else None train_sampler = DistributedSampler(trainset)
shuffle = False
else:
train_sampler = None
shuffle = True
train_loader = DataLoader(trainset, num_workers=1, shuffle=True, train_loader = DataLoader(trainset, num_workers=1, shuffle=shuffle,
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)