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

updated readme

Browse Source
This commit is contained in:
William Falcon
2019-07-25 10:33:35 -04:00
parent d23d25646a
commit c6da6eb46c
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+9 -6
View File
@@ -66,18 +66,21 @@ class CoolModel(LightningModule):
@property
def tng_dataloader(self):
mnist = MNIST('path/to/save', train=True)
return DataLoader(mnist, batch_size=32)
if not self._tng_dataloader:
self._tng_dataloader = DataLoader(MNIST('path/to/save', train=True), batch_size=32)
return self._tng_dataloader
@property
def val_dataloader(self):
mnist = MNIST('path/to/save', train=False)
return DataLoader(mnist, batch_size=32)
if not self._val_dataloader:
self._val_dataloader = DataLoader(MNIST('path/to/save', train=False), batch_size=32)
return self._val_dataloader
@property
def test_dataloader(self):
mnist = MNIST('sam/as/val/for/simplicity', train=False)
return DataLoader(mnist, batch_size=32)
if not self._test_dataloader:
self._test_dataloader = DataLoader(MNIST('path/to/save', train=False), batch_size=32)
return self._test_dataloader
```
2. Fit with a [trainer](https://williamfalcon.github.io/pytorch-lightning/Trainer/)