config.json

{
    "module": "keras_hub.src.models.swin_transformer.swin_transformer_backbone",
    "class_name": "SwinTransformerBackbone",
    "config": {
        "name": "swin_transformer_backbone",
        "trainable": true,
        "dtype": {
            "module": "keras",
            "class_name": "DTypePolicy",
            "config": {
                "name": "float32"
            },
            "registered_name": null
        },
        "image_shape": [
            224,
            224,
            3
        ],
        "patch_size": 4,
        "embed_dim": 96,
        "depths": [
            2,
            2,
            6,
            2
        ],
        "num_heads": [
            3,
            6,
            12,
            24
        ],
        "window_size": 7,
        "mlp_ratio": 4.0,
        "qkv_bias": true,
        "dropout_rate": 0.0,
        "attention_dropout": 0.0,
        "drop_path": 0.1,
        "patch_norm": true,
        "data_format": "channels_last"
    },
    "registered_name": "keras_hub>SwinTransformerBackbone"
}