config.json

{
    "module": "keras_hub.src.models.mit.mit_backbone",
    "class_name": "MiTBackbone",
    "config": {
        "name": "mi_t_backbone",
        "trainable": true,
        "layerwise_depths": [
            2,
            2,
            2,
            2
        ],
        "hidden_dims": [
            64,
            128,
            320,
            512
        ],
        "image_shape": [
            224,
            224,
            3
        ],
        "num_layers": 4,
        "layerwise_num_heads": [
            1,
            2,
            5,
            8
        ],
        "layerwise_sr_ratios": [
            8,
            4,
            2,
            1
        ],
        "max_drop_path_rate": 0.1,
        "layerwise_patch_sizes": [
            7,
            3,
            3,
            3
        ],
        "layerwise_strides": [
            4,
            2,
            2,
            2
        ]
    },
    "registered_name": "keras_hub>MiTBackbone"
}