Fix some errors on code

Part 2 - Neural Networks in PyTorch.ipynb

@@ -44,7 +44,7 @@
    "source": [
     "# 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('MNIST_data/', download=True, train=True, transform=transform)\n",

Part 3 - Training Neural Networks.ipynb

@@ -218,7 +218,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('MNIST_data/', download=True, train=True, transform=transform)\n",

Part 4 - Fashion-MNIST Exercise.ipynb

@@ -27,7 +27,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",
     "# Download and load the training data\n",
     "trainset = datasets.FashionMNIST('F_MNIST_data/', download=True, train=True, transform=transform)\n",
     "trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)\n",

Part 5 - Inference and Validation.ipynb

@@ -43,7 +43,7 @@
    "source": [
     "# 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",
     "# Download and load the training data\n",
     "trainset = datasets.FashionMNIST('F_MNIST_data/', download=True, train=True, transform=transform)\n",
     "trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)\n",

Part 6 - Saving and Loading Models.ipynb

@@ -38,7 +38,7 @@
    "source": [
     "# 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",
     "# Download and load the training data\n",
     "trainset = datasets.FashionMNIST('F_MNIST_data/', download=True, train=True, transform=transform)\n",
     "trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)\n",

Part 7 - Loading Image Data.ipynb

@@ -120,7 +120,7 @@
    "source": [
     "If you loaded the data correctly, you should see something like this (your image will be different):\n",
     "\n",
-    "<img src='assets/cat_cropped.png', width=244>"
+    "<img src='assets/cat_cropped.png' width=244>"
    ]
   },
   {