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

cleared spaces

Browse Source
This commit is contained in:
William Falcon
2020-01-16 16:25:25 -05:00
parent 0c38bd6fc2
commit b4bf33852c
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/core/lightning.py
+9 -7
View File
@@ -898,15 +898,13 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks):
@data_loader
def test_dataloader(self):
"""Implement a PyTorch DataLoader.
:return: PyTorch DataLoader
If you don't need a test dataset and a test_step, you don't need to implement this method.
r"""
Called by lightning during test loop. Make sure to use the @pl.data_loader decorator,
this ensures not calling this function until the data are needed.
If you want to change the data during every epoch DON'T use the data_loader decorator.
this ensures not calling this function until the data are needed.
Return:
PyTorch DataLoader
Example
-------
@@ -925,6 +923,10 @@ class LightningModule(ABC, GradInformation, ModelIO, ModelHooks):
return loader
.. note:: If you don't need a test dataset and a test_step, you don't need to implement this method.
.. note:: If you want to change the data during every epoch DON'T use the data_loader decorator.
"""
return None