{
  "model_type": "retvim",
  "architectures": [
    "RetViM"
  ],
  "framework": "pytorch",
  "format": "safetensors",
  "backbone": "vit_base_patch16_224.augreg_in21k_ft_in1k",
  "hybrid_blocks": 2,
  "ssm_backend": "native-mamba-1-reference",
  "feature_dim": 768,
  "d_state": 16,
  "d_conv": 4,
  "expand": 2,
  "classifier_dims": [
    768,
    512,
    256,
    4
  ],
  "num_labels": 4,
  "id2label": {
    "0": "CNV",
    "1": "DME",
    "2": "DRUSEN",
    "3": "NORMAL"
  },
  "label2id": {
    "CNV": 0,
    "DME": 1,
    "DRUSEN": 2,
    "NORMAL": 3
  },
  "parameter_count": 101223557,
  "torch_dtype": "float32",
  "note": "Custom PyTorch implementation from the linked GitHub repository; not a Transformers AutoModel checkpoint."
}