optimum

/

vit-base-patch16-224

like 0

Better specify it in the readme, there is a contributor using it for semantic segmentation task.

Hi @Jingya , it is the ONNX conversion of google/vit-base-patch16-224 that has a classification head.
You're absolutely right, I'm going to update the model card.

Here is a PR to update this: https://huggingface.co/optimum/vit-base-patch16-224/discussions/2

Closing since the PR got merged.