removed hparams req

williamFalcon · williamFalcon · commit 9b99a02061b0 · 2019-07-25T12:09:09.000-04:00
diff --git 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 @@ def __init__(self, hparams):
         """
         # 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
@@ -26,6 +26,7 @@ def __init__(self, hparams, force_remove_distributed_sampler=False):
         """
         # init superclass
         super(LightningTestModel, self).__init__()
+        self.hparams = hparams
 
         self.batch_size = hparams.batch_size
 
