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

train.py typo

Browse Source
This commit is contained in:
Raul Puri
2018-05-04 06:26:51 -07:00
parent 0d4218fb20
commit 4d5826e894
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

@@ -194,7 +194,7 @@ def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus,
epoch_offset = max(0, int(iteration / len(train_loader))) epoch_offset = max(0, int(iteration / len(train_loader)))
model.train() model.train()
if distributed_run or torch.cuda.device_count() > 1: if hparams.distributed_run or torch.cuda.device_count() > 1:
batch_parser = model.module.parse_batch batch_parser = model.module.parse_batch
else: else:
batch_parser = model.parse_batch batch_parser = model.parse_batch