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. | |