From be5fdd6d20e6bd5e4353993aa9de93306f1acc1e Mon Sep 17 00:00:00 2001 From: Rafael Valle Date: Fri, 4 May 2018 09:10:53 -0700 Subject: [PATCH] train.py: layout changes --- train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/train.py b/train.py index 4a83921..ee01b07 100644 --- a/train.py +++ b/train.py @@ -132,7 +132,7 @@ def validate(model, criterion, valset, iteration, batch_size, n_gpus, batch_parser = model.module.parse_batch else: batch_parser = model.parse_batch - + for i, batch in enumerate(val_loader): x, y = batch_parser(batch) y_pred = model(x) @@ -162,7 +162,6 @@ def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus, if hparams.distributed_run: init_distributed(hparams, n_gpus, rank, group_name) - torch.manual_seed(hparams.seed) torch.cuda.manual_seed(hparams.seed)