Error in trying to use for the first time

[gpleiss]

Error in trying to use for the first time

[gpleiss]

The "size mismatch" error looks like its coming from the final classification layer. Did you set the num_classes option to be the correct number of classes?

[chmaz]

Hi,

Yes, I have set the num_classes. I managed to have a training working with my images and resnet (adapting the resnet to the size of my images as described in pytorch/vision#155).
So my guess is that I get this error due to the custom size (80 x 80 pixels) of my images and I don't know how to adapt the densenet model for this image size.

I will keep looking,

Thanks,

Christophe

[gpleiss]

My other guess is that its a pooling issue. The default settings (3 pooling blocks) will reduce a 32x32 image into 1x1 feature maps by the last layer. An 80x80 image might instead be reduced to 3x3 feature maps, which cannot directly be fed into the classifier without further pooling (or increasing the size of the layer).

[chmaz]

By trying different values, I found that setting the avgpool_size to 10 solves the problem.

Thanks for the assistance,

Christophe

[taineleau-zz]

@chmaz Hi, I haven't seen this issue until now.

If you are going to test different size of images, I would recommend you using AdaptiveAvgPool2d or AdaptiveMaxPool2d instead of manually setting up the avgpooling size. I guess this would be much more convenient for you. :D