mirror of
https://github.com/malarinv/tacotron2
synced 2026-03-08 09:42:34 +00:00
train.py: layout changes
This commit is contained in:
1
train.py
1
train.py
@@ -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)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user