diff --git a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py index db065deff5..59d7210acc 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -54,10 +54,6 @@ class LightningTemplateModel(LightningModule): :param x: :return: """ - print('-'*100) - print('x: ', x.device) - print('model: ', self.c_d1.weight.device) - print('-'*100) x = self.c_d1(x) x = torch.tanh(x) diff --git a/tests/debug.py b/tests/debug.py index 095a060ef5..876075c51e 100644 --- a/tests/debug.py +++ b/tests/debug.py @@ -107,7 +107,7 @@ def main(): trainer = Trainer( experiment=exp, checkpoint_callback=checkpoint, - progress_bar=False, + progress_bar=True, max_nb_epochs=1, gpus=[0, 1], distributed_backend='dp',