Input size of fc layer in tutorial? vision
  
  
    The input of a Pytorch Neural Network is of type  [BATCH_SIZE] * [CHANNEL_NUMBER] * [HEIGHT] * [WIDTH]. 
Example  : So lets assume you image is of dimension 1×3×32×32 meaning that you have 1 image with 3 channels (RGB) with height 32 and width 32. So using the formular of convolution which is  ((W - F + 2P)/ S )+1 
and  ((H - F + 2P)/ S )+1 . The first one is for the  Width and the second one is for the height 
NOTE : Delete the the 5th line of the code because you already have pooling in the 12…
  


This might help