We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d0d5653 commit 20227b1Copy full SHA for 20227b1
pytorch_lightning/root_module/root_module.py
@@ -8,8 +8,8 @@
8
9
class LightningModule(GradInformation, ModelIO, ModelHooks):
10
11
- def __init__(self):
12
- super(LightningModule, self).__init__()
+ def __init__(self, *args, **kwargs):
+ super(LightningModule, self).__init__(*args, **kwargs)
13
14
self.dtype = torch.FloatTensor
15
self.exp_save_path = None
0 commit comments