From e184e09b280774919244c629584055236eeba1f1 Mon Sep 17 00:00:00 2001 From: udlbook <110402648+udlbook@users.noreply.github.com> Date: Thu, 7 Dec 2023 17:01:48 +0000 Subject: [PATCH] Created using Colaboratory --- Notebooks/Chap10/10_2_Convolution_for_MNIST_1D.ipynb | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/Notebooks/Chap10/10_2_Convolution_for_MNIST_1D.ipynb b/Notebooks/Chap10/10_2_Convolution_for_MNIST_1D.ipynb index c777d7e..2de008b 100644 --- a/Notebooks/Chap10/10_2_Convolution_for_MNIST_1D.ipynb +++ b/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",