From 2ca0864ce8e74cbd7a052231b775e2b348e0a9d0 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Thu, 18 Jul 2019 12:07:11 -0400
Subject: [PATCH] added arg docs

---
 examples/new_project_templates/lightning_module_template.py | 1 -
 pytorch_lightning/models/trainer.py                         | 1 +
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/new_project_templates/lightning_module_template.py b/examples/new_project_templates/lightning_module_template.py
index 810ee1e8..c3ecaff5 100644
--- a/examples/new_project_templates/lightning_module_template.py
+++ b/examples/new_project_templates/lightning_module_template.py
@@ -116,7 +116,6 @@ class LightningTemplateModel(LightningModule):
         :param outputs: list of individual outputs of each validation step
         :return:
         """
-        print(outputs)
         val_loss_mean = 0
         val_acc_mean = 0
         for output in outputs:
diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py
index 660a31b1..cc44fa39 100644
--- a/pytorch_lightning/models/trainer.py
+++ b/pytorch_lightning/models/trainer.py
@@ -29,6 +29,7 @@ except ModuleNotFoundError:
 
 
 def reduce_distributed_output(output, nb_gpus):
+    pdb.set_trace()
     if nb_gpus <= 1 or type(output) is torch.Tensor:
         return output