config.json

{
  "_name_or_path": "nvidia/MambaVision-B-1K",
  "architectures": [
    "MambaVisionModelForImageClassification"
  ],
  "auto_map": {
    "AutoConfig": "nvidia/MambaVision-B-1K--configuration_mambavision.MambaVisionConfig",
    "AutoModel": "nvidia/MambaVision-B-1K--modeling_mambavision.MambaVisionModel",
    "AutoModelForImageClassification": "nvidia/MambaVision-B-1K--modeling_mambavision.MambaVisionModelForImageClassification"
  },
  "crop_mode": "center",
  "crop_pct": 1.0,
  "depths": [
    3,
    3,
    10,
    5
  ],
  "dim": 128,
  "drop_path_rate": 0.3,
  "id2label": {
    "0": "0",
    "1": "1",
    "2": "2",
    "3": "3",
    "4": "4",
    "5": "5",
    "6": "6",
    "7": "7",
    "8": "8",
    "9": "9",
    "10": "10",
    "11": "11",
    "12": "12",
    "13": "13",
    "14": "14",
    "15": "15",
    "16": "16",
    "17": "17",
    "18": "18",
    "19": "19",
    "20": "20",
    "21": "21",
    "22": "22",
    "23": "23",
    "24": "24",
    "25": "25",
    "26": "26",
    "27": "27",
    "28": "28",
    "29": "29",
    "30": "30",
    "31": "31",
    "32": "32",
    "33": "33",
    "34": "34",
    "35": "35",
    "36": "36",
    "37": "37",
    "38": "38",
    "39": "39",
    "40": "40",
    "41": "41",
    "42": "42",
    "43": "43",
    "44": "44",
    "45": "45",
    "46": "46",
    "47": "47",
    "48": "48",
    "49": "49",
    "50": "50",
    "51": "51",
    "52": "52",
    "53": "53",
    "54": "54",
    "55": "55",
    "56": "56",
    "57": "57",
    "58": "58",
    "59": "59",
    "60": "60",
    "61": "61",
    "62": "62",
    "63": "63",
    "64": "64",
    "65": "65",
    "66": "66",
    "67": "67",
    "68": "68",
    "69": "69",
    "70": "70",
    "71": "71",
    "72": "72",
    "73": "73",
    "74": "74",
    "75": "75",
    "76": "76",
    "77": "77",
    "78": "78",
    "79": "79",
    "80": "80",
    "81": "81",
    "82": "82",
    "83": "83",
    "84": "84",
    "85": "85",
    "86": "86",
    "87": "87",
    "88": "88",
    "89": "89",
    "90": "90",
    "91": "91",
    "92": "92",
    "93": "93",
    "94": "94",
    "95": "95",
    "96": "96",
    "97": "97",
    "98": "98",
    "99": "99"
  },
  "in_dim": 64,
  "label2id": {
    "0": 0,
    "1": 1,
    "10": 10,
    "11": 11,
    "12": 12,
    "13": 13,
    "14": 14,
    "15": 15,
    "16": 16,
    "17": 17,
    "18": 18,
    "19": 19,
    "2": 2,
    "20": 20,
    "21": 21,
    "22": 22,
    "23": 23,
    "24": 24,
    "25": 25,
    "26": 26,
    "27": 27,
    "28": 28,
    "29": 29,
    "3": 3,
    "30": 30,
    "31": 31,
    "32": 32,
    "33": 33,
    "34": 34,
    "35": 35,
    "36": 36,
    "37": 37,
    "38": 38,
    "39": 39,
    "4": 4,
    "40": 40,
    "41": 41,
    "42": 42,
    "43": 43,
    "44": 44,
    "45": 45,
    "46": 46,
    "47": 47,
    "48": 48,
    "49": 49,
    "5": 5,
    "50": 50,
    "51": 51,
    "52": 52,
    "53": 53,
    "54": 54,
    "55": 55,
    "56": 56,
    "57": 57,
    "58": 58,
    "59": 59,
    "6": 6,
    "60": 60,
    "61": 61,
    "62": 62,
    "63": 63,
    "64": 64,
    "65": 65,
    "66": 66,
    "67": 67,
    "68": 68,
    "69": 69,
    "7": 7,
    "70": 70,
    "71": 71,
    "72": 72,
    "73": 73,
    "74": 74,
    "75": 75,
    "76": 76,
    "77": 77,
    "78": 78,
    "79": 79,
    "8": 8,
    "80": 80,
    "81": 81,
    "82": 82,
    "83": 83,
    "84": 84,
    "85": 85,
    "86": 86,
    "87": 87,
    "88": 88,
    "89": 89,
    "9": 9,
    "90": 90,
    "91": 91,
    "92": 92,
    "93": 93,
    "94": 94,
    "95": 95,
    "96": 96,
    "97": 97,
    "98": 98,
    "99": 99
  },
  "layer_scale": 1e-05,
  "layer_scale_conv": null,
  "mean": [
    0.485,
    0.456,
    0.406
  ],
  "mlp_ratio": 4,
  "model_type": "mambavision",
  "num_heads": [
    2,
    4,
    8,
    16
  ],
  "std": [
    0.229,
    0.224,
    0.225
  ],
  "torch_dtype": "float32",
  "transformers_version": "4.46.3",
  "window_size": [
    8,
    8,
    14,
    7
  ]
}