From 28a7dfde9c7d3d7439562358be9aa5ae0bbe7b5f Mon Sep 17 00:00:00 2001
From: Ulrich <ulrich.kerzel@rwth-aachen.de>
Date: Fri, 22 Mar 2024 10:47:02 +0100
Subject: [PATCH] add comment on size after pooling layer
---
datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb | 2 ++
1 file changed, 2 insertions(+)
diff --git a/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb b/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb
index 79e3bcf..8419a6a 100644
--- a/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb
+++ b/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb
@@ -294,6 +294,8 @@
" # #output = Floor ( (#input - #filter)/stride +1)\n",
" #\n",
" # here: 1* 64 * ( (26-2)/2) * ( (26-2)/2) = 9216\n",
+ " # ( after two convolutional layers, with 64 filters in the second, and the down-sizing with the CNN layers and pooling, we\n",
+ " # to from a 28x28 pixel image (one colour channel) to a size of 12x12, for 64 filters )\n",
" self.fc1 = nn.Linear(9216, 128)\n",
" self.fc2 = nn.Linear(128, 10)\n",
" self.dropout1 = nn.Dropout2d(0.25)\n",
--
GitLab