optimum
/

vit-base-patch16-224

regisss HF staff commited on Dec 14, 2022

Commit

f83e016

1 Parent(s): 95b5a98

Specify the head of the model in the model card

Files changed (1) hide show

README.md CHANGED Viewed

@@ -17,7 +17,7 @@ widget:
 # ONNX convert of ViT (base-sized model)
-# Conversion of [ViT-base](https://huggingface.co/google/vit-base-patch16-224)
 # Vision Transformer (base-sized model)

 # ONNX convert of ViT (base-sized model)
+Conversion of [ViT-base](https://huggingface.co/google/vit-base-patch16-224), which has a classification head to perform **image classification**.
 # Vision Transformer (base-sized model)