From aba7006fc28170bdc0a9087bff8593fdf5c86f8e Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:46:32 -0400 Subject: [PATCH] refactored model tests --- .../examples/new_project_templates/lightning_module_template.py | 2 ++ 1 file changed, 2 insertions(+) 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 6248ebcf..41f53518 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -98,8 +98,10 @@ class LightningTemplateModel(LightningModule): """ x, y = data_batch x = x.view(x.size(0), -1) + print('-'*100) print('x: ', x.device) print('model: ', self.c_d1.weight.device) + print('-'*100) y_hat = self.forward(x) loss_val = self.loss(y, y_hat)