removed hparams req

2019-07-25 12:08:00 -04:00 · 2019-07-25 12:08:00 -04:00 · 20227b1382
parent d0d5653b06
commit 20227b1382
1 changed files with 2 additions and 2 deletions
--- 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