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
+2 -2
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...).