From 4e6c7f80e50116b88562f644065bd87bc5dc7857 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 24 Jul 2019 13:47:37 -0400 Subject: [PATCH] refactored model tests --- .../new_project_templates/lightning_module_template.py | 6 ++++++ 1 file changed, 6 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 41f53518..bf736227 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -78,6 +78,11 @@ class LightningTemplateModel(LightningModule): # forward pass x, y = data_batch x = x.view(x.size(0), -1) + print('-'*100) + print('TRAIN') + print('x: ', x.device) + print('model: ', self.c_d1.weight.device) + print('-'*100) y_hat = self.forward(x) # calculate loss @@ -99,6 +104,7 @@ class LightningTemplateModel(LightningModule): x, y = data_batch x = x.view(x.size(0), -1) print('-'*100) + print('VAL') print('x: ', x.device) print('model: ', self.c_d1.weight.device) print('-'*100)