config.json

{
    "module": "keras_hub.src.models.deit.deit_backbone",
    "class_name": "DeiTBackbone",
    "config": {
        "name": "dei_t_backbone",
        "trainable": true,
        "image_shape": [
            224,
            224,
            3
        ],
        "patch_size": [
            16,
            16
        ],
        "num_layers": 12,
        "num_heads": 3,
        "hidden_dim": 192,
        "intermediate_dim": 768,
        "dropout_rate": 0.0,
        "attention_dropout": 0.0,
        "layer_norm_epsilon": 1e-12,
        "use_mha_bias": true
    },
    "registered_name": "keras_hub>DeiTBackbone"
}