---
library_name: keras-hub
---
This is a [`ViT` model](https://keras.io/api/keras_hub/models/vi_t) uploaded using the KerasHub library and can be used with JAX, TensorFlow, and PyTorch backends.
This model is related to a `ImageClassifier` task.

Model config:
* **name:** vi_t_backbone
* **trainable:** True
* **image_shape:** [384, 384, 3]
* **patch_size:** 32
* **num_layers:** 24
* **num_heads:** 16
* **hidden_dim:** 1024
* **mlp_dim:** 4096
* **dropout_rate:** 0.0
* **attention_dropout:** 0.0
* **layer_norm_epsilon:** 1e-06
* **use_mha_bias:** True
* **use_mlp_bias:** True

This model card has been generated automatically and should be completed by the model author. See [Model Cards documentation](https://huggingface.co/docs/hub/model-cards) for more information.