From 9b99a02061b0fae447a37e808615775e88b2ee05 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jul 2019 12:09:09 -0400 Subject: [PATCH] removed hparams req --- .../examples/new_project_templates/lightning_module_template.py | 1 + pytorch_lightning/testing_models/lm_test_module.py | 1 + 2 files 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 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