ResNet Running problem

@justusschock I think the older version of torchvision model is using the standard average pooling instead of adaptive. @Niki can you double check your version? If it’s hsing the normal average pooling just change it do adaptive pooling

1 Like

@justusschock it seems it has problem with nonsquare input, it shows the error for 448x224.
@unkn0wnxx The version of my torchvision is 0.2.2

But does it work with square images?

Yes, it works for square images.

If I want to use ResNet how can I get two outputs from two parallel linear layers at the end+ softmax?

You could use the same approach as described using the torchvision ResNet model.
Besides some other used names I don’t see any differences between both codes.
Where are you stuck?