From d0d5653b06a6f4acba4073c2da1f236251f4cc22 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jul 2019 12:04:20 -0400 Subject: [PATCH] removed hparams req --- .../examples/new_project_templates/lightning_module_template.py | 2 +- pytorch_lightning/testing_models/lm_test_module.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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