From 4b31f3d4bf705160e0b6647e5475fa42b458e2bf Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Jul 2019 16:47:39 -0400 Subject: [PATCH] clean up dead code --- pytorch_lightning/models/trainer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 428d9ca5..0298e6a7 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -483,11 +483,11 @@ class Trainer(TrainerIO): # forward pass # return a scalar value and a dic with tqdm metrics - if self.data_parallel: - output = self.model(data_batch, batch_nb) - output = reduce_distributed_output(output, len(self.data_parallel_device_ids)) - else: - output = self.model.training_step(data_batch, batch_nb) + # if self.data_parallel: + # output = self.model(data_batch, batch_nb) + # output = reduce_distributed_output(output, len(self.data_parallel_device_ids)) + # else: + output = self.model.training_step(data_batch, batch_nb) model_specific_tqdm_metrics_dic = output['tqdm_metrics'] loss = output['loss']