Change data download path

2021-09-03 09:08:06 -05:00 · 2021-09-03 09:08:06 -05:00 · 20c8739f99
commit 20c8739f99
parent 1944c4a961
1 changed files with 2 additions and 2 deletions
--- a/mnist.py
+++ b/mnist.py
@ -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(),