set dp as default backend

2019-07-18 11:40:00 -04:00 · 2019-07-18 11:40:00 -04:00 · c67c84b443
parent 4db32984c6
commit c67c84b443
1 changed files with 3 additions and 0 deletions
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@ -29,6 +29,9 @@ except ModuleNotFoundError:


 def reduce_distributed_output(output, nb_gpus):
+    if nb_gpus <= 1:
+        return output
+
    for k, v in output.items():
        # recurse on nested dics
        if isinstance(output[k], dict):