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",