Fix dataloaders in TPU example (#1174)

This commit is contained in:
Teddy Koker 2020-03-24 14:53:15 -04:00 committed by GitHub
parent 60b8246bc3
commit 6a9bbbc886
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -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...).