removed hparams req

Lightning-AI · Jul 25, 2019 · 20227b1 · 20227b1
1 parent d0d5653
commit 20227b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/root_module/root_module.py b/pytorch_lightning/root_module/root_module.py
@@ -8,8 +8,8 @@
 
 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