From 20227b13825f5767775571a65ac7496f91f8a7e2 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 25 Jul 2019 12:08:00 -0400 Subject: [PATCH] removed hparams req --- pytorch_lightning/root_module/root_module.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py index 4f55ed37..d6d740f0 100644 --- a/pytorch_lightning/root_module/root_module.py +++ b/pytorch_lightning/root_module/root_module.py @@ -8,8 +8,8 @@ from pytorch_lightning.root_module.decorators import data_loader class LightningModule(GradInformation, ModelIO, ModelHooks): - def __init__(self): - super(LightningModule, self).__init__() + def __init__(self, *args, **kwargs): + super(LightningModule, self).__init__(*args, **kwargs) self.dtype = torch.FloatTensor self.exp_save_path = None