diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 08736a62..ffa400a2 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -399,7 +399,7 @@ class Trainer(TrainerIO): # when DP, we need to aggregate the scalars we received as outputs # use mean as the reduce function if self.data_parallel: - output = reduce_distributed_output(output, len(self.gpus)) + output = reduce_distributed_output(output, len(self.data_parallel_device_ids)) model_specific_tqdm_metrics_dic = output['tqdm_metrics'] loss = output['loss']