Unknown error related to channel dimensions

#12
by vigneshwar472 - opened

I was training the VIT for image classification of skin cancer images.

When I just started training the model, I was getting this error.

ValueError: Make sure that the channel dimension of the pixel values match with the one set in the configuration. Expected 3 but got 224.

But my images are as desired.

ISIC_holdout['train'][0]['pixel_values'].shape
(224, 224, 3)

Please help me how to resolve the error. I should meet the deadline of my competition and resolving this error is very crucial.

Sign up or log in to comment