diff --git a/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb b/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb index 8419a6a6df37c57786c2e8d7e58e5c486a9ec7e6..d828bc6133cbf21ca2b1083b25d079b8701c38d5 100644 --- a/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb +++ b/datascienceintro/solutions/Solution_PyTorch_MNIST.ipynb @@ -209,6 +209,7 @@ "***Hint*** \\\n", "One of the tricky things to evaluate is the output size after pooling layers.\n", "This [blog](https://androidkt.com/calculate-output-size-convolutional-pooling-layers-cnn/) has a nice illustrated intuition behind the formulae for convolutional and maxpool layers. This [repository](https://github.com/tuttelikz/conv_output_size) has a collection of helper functions, there is a pre-print [A guide to convolution arithmetic for deep learning](https://arxiv.org/pdf/1603.07285.pdf) to discuss this in more detail.\n", + "You can also use this handy (web calculator)[http://layer-calc.com/] that gives you the size after all convolutional and maxpool layers.\n", "\n", "For convolutional layers: \\\n", "```#output = Floor( ( #input - #filter + 2* #padding)/stride + 1)```\n",