add alternative line

Author: rasbt

ch14/ch14_part1.ipynb

@@ -1154,7 +1154,7 @@
     "    ax = fig.add_subplot(2, 6, i+1)\n",
     "    ax.set_xticks([]); ax.set_yticks([])\n",
     "    img = mnist_test_dataset[i][0][0, :, :]\n",
-    "    pred = model(img.unsqueeze(0).unsqueeze(1))\n",
+    "    pred = model(img.unsqueeze(0).unsqueeze(1)) # alternatively: mnist_test_dataset.data[i].float()\n",
     "    y_pred = torch.argmax(pred)\n",
     "    ax.imshow(img, cmap='gray_r')\n",
     "    ax.text(0.9, 0.1, y_pred.item(), \n",
@@ -1235,7 +1235,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.7"
+   "version": "3.10.12"
   }
  },
  "nbformat": 4,