cmendezc

Ejemplo NPUs

...@@ -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)
......