config.json

{
  "_name_or_path": "MBZUAI/swiftformer-xs",
  "architectures": [
    "SwiftFormerForImageClassification"
  ],
  "batch_norm_eps": 1e-05,
  "depths": [
    3,
    3,
    6,
    4
  ],
  "down_pad": 1,
  "down_patch_size": 3,
  "down_stride": 2,
  "downsamples": [
    true,
    true,
    true,
    true
  ],
  "drop_path_rate": 0.0,
  "embed_dims": [
    48,
    56,
    112,
    220
  ],
  "hidden_act": "gelu",
  "id2label": {
    "0": "Mild",
    "1": "Moderate",
    "2": "No_DR",
    "3": "Proliferate_DR",
    "4": "Severe"
  },
  "label2id": {
    "Mild": 0,
    "Moderate": 1,
    "No_DR": 2,
    "Proliferate_DR": 3,
    "Severe": 4
  },
  "layer_scale_init_value": 1e-05,
  "mlp_ratio": 4,
  "model_type": "swiftformer",
  "num_channels": 3,
  "problem_type": "single_label_classification",
  "torch_dtype": "float32",
  "transformers_version": "4.36.2",
  "use_layer_scale": true
}