Merge pull request L1aoXingyu#36 from NjtechPrinceling/master

Author: L1aoXingyu

08-AutoEncoder/simple_autoencoder.py

@@ -28,7 +28,7 @@ def to_img(x):
 
 img_transform = transforms.Compose([
     transforms.ToTensor(),
-    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
+    transforms.Normalize([0.5], [0.5])
 ])
 
 dataset = MNIST('./data', transform=img_transform)