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