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 65fb303f..0a4dab26 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -26,6 +26,7 @@ class LightningTemplateModel(LightningModule): """ # init superclass super(LightningTemplateModel, self).__init__() + self.hparams = hparams self.batch_size = hparams.batch_size diff --git a/pytorch_lightning/testing_models/lm_test_module.py b/pytorch_lightning/testing_models/lm_test_module.py index 8cae1c16..e33ee53e 100644 --- a/pytorch_lightning/testing_models/lm_test_module.py +++ b/pytorch_lightning/testing_models/lm_test_module.py @@ -26,6 +26,7 @@ class LightningTestModel(LightningModule): """ # init superclass super(LightningTestModel, self).__init__() + self.hparams = hparams self.batch_size = hparams.batch_size