Created using Colaboratory

Notebooks/Chap10/10_2_Convolution_for_MNIST_1D.ipynb

@@ -4,7 +4,7 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "authorship_tag": "ABX9TyN1v/yg9PtdSVOWlYJ7bgkz",
+      "authorship_tag": "ABX9TyOL1aX3C4928zwYWnmdoN+P",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -141,6 +141,9 @@
         "# https://pytorch.org/docs/stable/generated/torch.nn.Flatten.html\n",
         "# https://pytorch.org/docs/1.13/generated/torch.nn.Linear.html?highlight=linear#torch.nn.Linear\n",
         "\n",
+        "# NOTE THAT THE CONVOLUTIONAL LAYERS DO NOT NEED TO TAKE THE NUMBER OF INPUT CHANNELS AS A PARAMETER\n",
+        "# AND NOT THE INPUT SIZE.\n",
+        "\n",
         "# Replace the following function:\n",
         "model = nn.Sequential(\n",
         "nn.Flatten(),\n",