removed hparams req

2019-07-25 12:09:09 -04:00 · 2019-07-25 12:09:09 -04:00 · 9b99a02061
parent 20227b1382
commit 9b99a02061
2 changed files with 2 additions and 0 deletions
--- 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

--- 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