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

Docs2 (#1069)

Browse Source 
* new image

* new image
This commit is contained in:
William Falcon
2020-03-06 06:33:28 -05:00
committed by GitHub
parent f996c2892f
commit d80215ed9d
2 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/introduction_guide.rst
+3 -3
View File
@@ -494,13 +494,13 @@ In this method we do all the preparation we need to do once (instead of on every
self.test_dataset = mnist_test
def train_dataloader(self):
return DataLoader(train_dataset, batch_size=64)
return DataLoader(self.train_dataset, batch_size=64)
def val_dataloader(self):
return DataLoader(mnist_val, batch_size=64)
return DataLoader(self.mnist_val, batch_size=64)
def test_dataloader(self):
return DataLoader(mnist_test, batch_size=64)
return DataLoader(self.mnist_test, batch_size=64)
The `prepare_data` method is also a good place to do any data processing that needs to be done only
once (ie: download or tokenize, etc...).