Update GANsImagesMNST.py

fullarray · web-flow · commit 42b95fee0502 · 2021-06-22T22:16:38.000-04:00
diff --git a/handCode/GANsImagesMNST.py b/handCode/GANsImagesMNST.py
@@ -1,7 +1,10 @@
 def mnist_data():
     compose = transforms.Compose(
+        
         [transforms.ToTensor(),
+         
          transforms.Normalize((.5, .5, .5), (.5, .5, .5))
+         
         ])
     out_dir = './dataset'
     return datasets.MNIST(root=out_dir, train=True, transform=compose, download=True)
@@ -12,4 +15,4 @@ def mnist_data():
 # Create loader with data, so that we can iterate over it
 data_loader = torch.utils.data.DataLoader(data, batch_size=100, shuffle=True)
 # Num batches
-num_batches = len(data_loader)
+num_batches = len(data_loader)
