From 4d5826e8949f44aab79016b0f1596859ab0c9e62 Mon Sep 17 00:00:00 2001 From: Raul Puri Date: Fri, 4 May 2018 06:26:51 -0700 Subject: [PATCH] train.py typo --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c64ea1c..4a83921 100644 --- a/train.py +++ b/train.py @@ -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))) 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 else: batch_parser = model.parse_batch