From 4a3c9de85708bb7f4e7564b4f63b3b1e5b6eeb4c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 26 Jun 2019 17:53:05 -0400 Subject: [PATCH] updated args --- pytorch_lightning/models/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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']