Showing
1 changed file
with
1 additions
and
1 deletions
... | @@ -118,7 +118,7 @@ def main(): | ... | @@ -118,7 +118,7 @@ def main(): |
118 | transforms.ToTensor(), | 118 | transforms.ToTensor(), |
119 | transforms.Normalize((0.1307,), (0.3081,)), | 119 | transforms.Normalize((0.1307,), (0.3081,)), |
120 | ]) | 120 | ]) |
121 | - dataset1 = datasets.MNIST('../data', train=True, download=True, | 121 | + dataset1 = datasets.MNIST('../data', train=True, download=False, |
122 | transform=transform) | 122 | transform=transform) |
123 | dataset2 = datasets.MNIST('../data', train=False, | 123 | dataset2 = datasets.MNIST('../data', train=False, |
124 | transform=transform) | 124 | transform=transform) | ... | ... |
-
Please register or login to post a comment