I have been trying to develop an image classification model using ViT and done with it till model building phase. when I'm trying to train the model it's showing an error of this kind this is the error And I'm attaching an full code regarding my model Model Notebook
Thanks & Regards Satwik Sunnam