Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
For image classification models, ([ViTForImageClassification]), the model expects a tensor of dimension
(batch_size) with each value of the batch corresponding to the expected label of each individual image.