diff options
Diffstat (limited to 'mnist.py')
-rw-r--r-- | mnist.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -8,14 +8,14 @@ import matplotlib.pyplot as plt training_data = datasets.MNIST( - root="data", + root=".data", train=True, download=True, transform=ToTensor(), ) test_data = datasets.MNIST( - root="data", + root=".data", train=False, download=True, transform=ToTensor(), |