mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-27 13:12:20 +08:00
Fix dataloaders in TPU example (#1174)
This commit is contained in:
1 parent
60b8246bc3
commit
6a9bbbc886
1 file changed
+2
-2
@@ -547,10 +547,10 @@ In this method we do all the preparation we need to do once (instead of on every
|
||||
return DataLoader(self.train_dataset, batch_size=64)
|
||||
|
||||
def val_dataloader(self):
|
||||
return DataLoader(self.mnist_val, batch_size=64)
|
||||
return DataLoader(self.val_dataset, batch_size=64)
|
||||
|
||||
def test_dataloader(self):
|
||||
return DataLoader(self.mnist_test, batch_size=64)
|
||||
return DataLoader(self.test_dataset, 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...).
|
||||
|
||||
Reference in new issue
Block a user