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)