From 970d1609e144b8fbd439909fd843360479551467 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Jul 2019 15:25:33 -0400 Subject: [PATCH] added single node distdataparallel --- 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 ceb5e397..8594f731 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -30,7 +30,7 @@ def reduce_distributed_output(output, nb_gpus): return output def dummy(gpu, a): - print(gpu) + print(a) class Trainer(TrainerIO):