metadata
library_name: keras-hub
This is a ViT model 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 for more information.