Halp!  I am trying to feed 512x512 and higher res images to ResNet models in the TorchVision zoo but it keeps throwing size_mismatch errors at me.
Can someone please explain what needs to be changed in following model definition to feed it 512x512 or 1024x1024 dimensions?  I am new from Keras and a bit lost here.  Thank you!
ResNet model definition:
The error seems to be occurring when executing line 151, in forward x = self.fc(x)

https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py 118