wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated test models with lazy decorators

Browse Source
This commit is contained in:
William Falcon
2019-07-25 11:01:08 -04:00
parent 42a45bb273
commit 09dba13cde
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/root_module/root_module.py
+3 -3
View File
@@ -84,7 +84,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
for param in self.parameters():
param.requires_grad = True
@ptl.data_loader
@data_loader
def tng_dataloader(self):
"""
Implement a function to load an h5py of this data
@@ -92,7 +92,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
"""
raise NotImplementedError
@ptl.data_loader
@data_loader
def test_dataloader(self):
"""
Implement a function to load an h5py of this data
@@ -100,7 +100,7 @@ class LightningModule(GradInformation, ModelIO, ModelHooks):
"""
raise NotImplementedError
@ptl.data_loader
@data_loader
def val_dataloader(self):
"""
Implement a function to load an h5py of this data