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