From 0e9e07835cfc457384a48a4aa5f26af44ffac202 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:53:34 -0400 Subject: [PATCH] refactored model tests --- .../examples/new_project_templates/lightning_module_template.py | 1 + 1 file changed, 1 insertion(+) 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 5507341b..16f588ae 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -55,6 +55,7 @@ class LightningTemplateModel(LightningModule): :return: """ + print(x.device) x = self.c_d1(x) x = torch.tanh(x) x = self.c_d1_bn(x)