From 4085b3fa692f929e6a428a512e3e7c24ec90899a Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 11:57:39 -0400 Subject: [PATCH] set dp as default backend --- 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 b371fb33..feea0bed 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -29,7 +29,7 @@ except ModuleNotFoundError: def reduce_distributed_output(output, nb_gpus): - if nb_gpus <= 1: + if nb_gpus <= 1 or type(output) is torch.Tensor: return output for k, v in output.items():