hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[flaky] fix mnist ptl data cache (#15344)

Browse source 
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Richard Liaw 2021-04-15 16:24:17 -07:00 • committed by GitHub
parent a54d69f535
commit dc80d9f42a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
python/ray/util/sgd/torch/examples/pytorch-lightning/mnist-ptl.py
View file

@ -60,7 +60,10 @@ class LitMNIST(LightningModule):
# prepare transforms standard to MNIST
mnist_train = MNIST(
os.getcwd(), train=True, download=True, transform=transform)
os.path.expanduser("~/data"),
train=True,
download=True,
transform=transform)
self.mnist_train, self.mnist_val = random_split(
mnist_train, [55000, 5000])