Update Part 3 - Training Neural Networks (Solution).ipynb

Author: rrrrrr8

intro-to-pytorch/Part 3 - Training Neural Networks (Solution).ipynb

@@ -86,7 +86,7 @@
     "\n",
     "# Define a transform to normalize the data\n",
     "transform = transforms.Compose([transforms.ToTensor(),\n",
-    "                                transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),\n",
+    "                                transforms.Normalize((0.5,), (0.5,)),\n",
     "                              ])\n",
     "# Download and load the training data\n",
     "trainset = datasets.MNIST('~/.pytorch/MNIST_data/', download=True, train=True, transform=transform)\n",