Image Classification

when I use pytorch, resnet50 to classify my images, which have 3 classed, there is a problem,


is there anyone know what happened? please help me!

It seems your dataset class does not resize all the images to same size?
The width (dimension 2) of images do not match for all images.

yes,I found the issue, thanks