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 1ba8ebb1..65fb303f 100644 --- a/pytorch_lightning/examples/new_project_templates/lightning_module_template.py +++ b/pytorch_lightning/examples/new_project_templates/lightning_module_template.py @@ -25,7 +25,7 @@ class LightningTemplateModel(LightningModule): :param hparams: """ # init superclass - super(LightningTemplateModel, self).__init__(hparams) + super(LightningTemplateModel, self).__init__() 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 080d8fed..8cae1c16 100644 --- a/pytorch_lightning/testing_models/lm_test_module.py +++ b/pytorch_lightning/testing_models/lm_test_module.py @@ -25,7 +25,7 @@ class LightningTestModel(LightningModule): :param hparams: """ # init superclass - super(LightningTestModel, self).__init__(hparams) + super(LightningTestModel, self).__init__() self.batch_size = hparams.batch_size