RaLo4:

7*32, num_classes)
It seams like this network was made to run with RGB images of size 28x28. But since you provide it with images of size 224x224 the feature maps that get get passed into the fc layer are not of size 7x7x32 but 56x56x32 (if my calculations are correct).
So to make it work again change:


Hi @RaLo4,
Thanks for your help.
Noted on the code part.
May I know how did (7732) derived from?
or from my case (565632)?
How did from conv2(16,32,5) become (565632).
From my understanding the output of conv2 became input of fc.
Or am I wrong?