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.