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

train.py: layout changes

single-gpu-and-0.4
Rafael Valle 2018-05-04 09:10:53 -07:00
parent a0097732c0
commit be5fdd6d20
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train.py
View File

@ -162,7 +162,6 @@ def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus,
if hparams.distributed_run: if hparams.distributed_run:
init_distributed(hparams, n_gpus, rank, group_name) init_distributed(hparams, n_gpus, rank, group_name)
torch.manual_seed(hparams.seed) torch.manual_seed(hparams.seed)
torch.cuda.manual_seed(hparams.seed) torch.cuda.manual_seed(hparams.seed)