From 3521051877a55e2a3128eb57e2b9a95f31b80790 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:51:12 -0400 Subject: [PATCH] refactored model tests --- .../examples/new_project_templates/lightning_module_template.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 bf736227..6c6ba68d 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -106,7 +106,7 @@ class LightningTemplateModel(LightningModule): print('-'*100) print('VAL') print('x: ', x.device) - print('model: ', self.c_d1.weight.device) + print('model: ', self.c_d1.weight.device, self.c_d1.bias.device) print('-'*100) y_hat = self.forward(x)