I am trying to use a pretrained ResNet model but I get this error:
RuntimeError: size mismatch, m1: [30 x 2048], m2: [9216 x 1024] at /pytorch/aten/src/THC/generic/THCTensorMathBlas.cu:283
model.fc = nn.Sequential(nn.Linear(9216,1024), nn.ReLU(),nn.Dropout(0.4),nn.Linear(1024,10))
Full Notebook can be found here: https://colab.research.google.com/drive/1k40NNulSIS6ANagopSPBH4Xty_Cw39qC#scrollTo=9J6VCm2y8tXn 2
Please I would really appreciate a bit of help with this issue since I am a new practitioner. Thanks in advanced.