config.json

{
  "act_cfg": {
    "type": "GELU"
  },
  "architectures": [
    "SOLIDERModel"
  ],
  "attn_drop_rate": 0.0,
  "auto_map": {
    "AutoConfig": "configuration_solider.SOLIDERConfig",
    "AutoModel": "modeling_solider.SOLIDERModel"
  },
  "convert_weights": false,
  "depths": [
    2,
    2,
    18,
    2
  ],
  "drop_path_rate": 0.0,
  "drop_rate": 0.0,
  "embed_dims": 96,
  "frozen_stages": -1,
  "hidden_size": 96,
  "img_size": [
    224,
    224
  ],
  "in_channels": 3,
  "init_cfg": null,
  "mlp_ratio": 4,
  "model_type": "swin_transformer",
  "name": "solider_small",
  "norm_cfg": {
    "type": "LN"
  },
  "num_heads": [
    3,
    6,
    12,
    24
  ],
  "out_indices": [
    0,
    1,
    2,
    3
  ],
  "patch_norm": true,
  "patch_size": 4,
  "pretrain_img_size": [
    224,
    224
  ],
  "pretrained": null,
  "qk_scale": null,
  "qkv_bias": true,
  "semantic_weight": 0.5,
  "strides": [
    4,
    2,
    2,
    2
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.44.2",
  "use_abs_pos_embed": false,
  "vision_width": 768,
  "window_size": 7,
  "with_cp": false
}