config.json

{
    "module": "keras_hub.src.models.mix_transformer.mix_transformer_backbone",
    "class_name": "MiTBackbone",
    "config": {
        "name": "mi_t_backbone",
        "trainable": true,
        "depths": [
            3,
            4,
            18,
            3
        ],
        "hidden_dims": [
            64,
            128,
            320,
            512
        ],
        "image_shape": [
            224,
            224,
            3
        ],
        "num_layers": 4,
        "blockwise_num_heads": [
            1,
            2,
            5,
            8
        ],
        "blockwise_sr_ratios": [
            8,
            4,
            2,
            1
        ],
        "max_drop_path_rate": 0.1,
        "patch_sizes": [
            7,
            3,
            3,
            3
        ],
        "strides": [
            4,
            2,
            2,
            2
        ]
    },
    "registered_name": "keras_hub>MiTBackbone"
}