From c26d200c417602b391d4acdb74bde4a9165905d1 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:57:34 -0400 Subject: [PATCH] refactored model tests --- .../new_project_templates/lightning_module_template.py | 4 ---- tests/debug.py | 2 +- 2 files changed, 1 insertion(+), 5 deletions(-) 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',