fix iter next

Author: YoniChechik

c_10_neural_networks_basics/fully_connected.ipynb

@@ -23,7 +23,9 @@
         "import matplotlib.pyplot as plt\n",
         "\n",
         "import torch.nn as nn\n",
-        "import torch.optim as optim\n"
+        "import torch.optim as optim\n",
+        "\n",
+        "torch.manual_seed(0)"
       ]
     },
     {
@@ -669,8 +671,7 @@
         }
       ],
       "source": [
-        "dataiter = iter(test_loader)\n",
-        "images, labels = dataiter.next()\n",
+        "images, labels = next(iter(test_loader))\n",
         "\n",
         "outputs = model(images.to(device))\n",
         "\n",
@@ -679,12 +680,12 @@
         "figure = plt.figure()\n",
         "num_of_images = 20\n",
         "for index in range(num_of_images):\n",
-        "    plt.subplot(4, 5, index+1)\n",
-        "    plt.axis('off')\n",
-        "    plt.imshow(images[index,0,:,:], cmap='gray')\n",
+        "    plt.subplot(4, 5, index + 1)\n",
+        "    plt.axis(\"off\")\n",
+        "    plt.imshow(images[index, 0, :, :], cmap=\"gray\")\n",
         "    prd = int(predicted[index])\n",
         "    gt = int(labels[index])\n",
-        "    plt.title(f\"pred:{prd}\\n gt:{gt}\" +\"\\nwrong!\"*(prd!=gt))\n",
+        "    plt.title(f\"pred:{prd}\\n gt:{gt}\" + \"\\nwrong!\" * (prd != gt))\n",
         "plt.tight_layout()"
       ]
     }