added arg docs

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:

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