@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 https://github.com/kuangliu/pytorch-cifar/blob/master/models/resnet.py 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?