Hi everyone,

I just went through the 60min Blitz tutorial and I have a question. How do they reach a value of 16 * 5 * 5 for the first fully connected layer in their network (in the last part of the tutorial)? 16 is the number of out_channels from the conv2 layer but I can’t figure out the 5 * 5. When I use the formula to calculate it starting with a value of 32 * 32 I reach a value of 10 * 10 and not 5 * 5.

Can anyone help me?

My calculations:

- input size (32) - kernel size (5) + 1 = 28
- ((28 - kernel size (2)) / 2) + 1 = 14
- 14 - kernel size (5) + 1 = 10

no more layers before the fully connected one so my guess would be 10…