From 5eb264540d7a4fb03795ddc5de805602ca10a3f8 Mon Sep 17 00:00:00 2001
From: udlbook <110402648+udlbook@users.noreply.github.com>
Date: Thu, 24 Oct 2024 16:40:27 -0400
Subject: [PATCH] Created using Colab

---
 .../Chap10/10_5_Convolution_For_MNIST.ipynb   | 19 +++++++------------
 1 file changed, 7 insertions(+), 12 deletions(-)

diff --git a/Notebooks/Chap10/10_5_Convolution_For_MNIST.ipynb b/Notebooks/Chap10/10_5_Convolution_For_MNIST.ipynb
index 33322a0d..fb5b32b2 100644
--- a/Notebooks/Chap10/10_5_Convolution_For_MNIST.ipynb
+++ b/Notebooks/Chap10/10_5_Convolution_For_MNIST.ipynb
@@ -4,7 +4,7 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "authorship_tag": "ABX9TyNELb86uz5qbhEKH81UqFKT",
+      "authorship_tag": "ABX9TyORZF8xy4X1yf4oRhRq8Rtm",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -72,8 +72,12 @@
         "# even before you make changes.\n",
         "batch_size_train = 64\n",
         "batch_size_test = 1000\n",
+        "\n",
+        "# TODO Change this directory to point towards an existing directory\n",
+        "myDir = '/files/'\n",
+        "\n",
         "train_loader = torch.utils.data.DataLoader(\n",
-        "  torchvision.datasets.MNIST('/files/', train=True, download=True,\n",
+        "  torchvision.datasets.MNIST(myDir, train=True, download=True,\n",
         "                             transform=torchvision.transforms.Compose([\n",
         "                               torchvision.transforms.ToTensor(),\n",
         "                               torchvision.transforms.Normalize(\n",
@@ -82,7 +86,7 @@
         "  batch_size=batch_size_train, shuffle=True)\n",
         "\n",
         "test_loader = torch.utils.data.DataLoader(\n",
-        "  torchvision.datasets.MNIST('/files/', train=False, download=True,\n",
+        "  torchvision.datasets.MNIST(myDir, train=False, download=True,\n",
         "                             transform=torchvision.transforms.Compose([\n",
         "                               torchvision.transforms.ToTensor(),\n",
         "                               torchvision.transforms.Normalize(\n",
@@ -96,15 +100,6 @@
       "execution_count": null,
       "outputs": []
     },
-    {
-      "cell_type": "code",
-      "source": [],
-      "metadata": {
-        "id": "YGwbxJDEm88i"
-      },
-      "execution_count": null,
-      "outputs": []
-    },
     {
       "cell_type": "code",
       "source": [