config.json

{
  "_name_or_path": "dima806/mammals_45_types_image_classification",
  "architectures": [
    "ViTForImageClassification"
  ],
  "attention_probs_dropout_prob": 0.0,
  "encoder_stride": 16,
  "hidden_act": "gelu",
  "hidden_dropout_prob": 0.0,
  "hidden_size": 768,
  "id2label": {
    "0": "african_elephant",
    "1": "alpaca",
    "2": "american_bison",
    "3": "anteater",
    "4": "arctic_fox",
    "5": "armadillo",
    "6": "baboon",
    "7": "badger",
    "8": "blue_whale",
    "9": "brown_bear",
    "10": "camel",
    "11": "dolphin",
    "12": "giraffe",
    "13": "groundhog",
    "14": "highland_cattle",
    "15": "horse",
    "16": "jackal",
    "17": "kangaroo",
    "18": "koala",
    "19": "manatee",
    "20": "mongoose",
    "21": "mountain_goat",
    "22": "opossum",
    "23": "orangutan",
    "24": "otter",
    "25": "polar_bear",
    "26": "porcupine",
    "27": "red_panda",
    "28": "rhinoceros",
    "29": "sea_lion",
    "30": "seal",
    "31": "snow_leopard",
    "32": "squirrel",
    "33": "sugar_glider",
    "34": "tapir",
    "35": "vampire_bat",
    "36": "vicuna",
    "37": "walrus",
    "38": "warthog",
    "39": "water_buffalo",
    "40": "weasel",
    "41": "wildebeest",
    "42": "wombat",
    "43": "yak",
    "44": "zebra"
  },
  "image_size": 224,
  "initializer_range": 0.02,
  "intermediate_size": 3072,
  "label2id": {
    "african_elephant": 0,
    "alpaca": 1,
    "american_bison": 2,
    "anteater": 3,
    "arctic_fox": 4,
    "armadillo": 5,
    "baboon": 6,
    "badger": 7,
    "blue_whale": 8,
    "brown_bear": 9,
    "camel": 10,
    "dolphin": 11,
    "giraffe": 12,
    "groundhog": 13,
    "highland_cattle": 14,
    "horse": 15,
    "jackal": 16,
    "kangaroo": 17,
    "koala": 18,
    "manatee": 19,
    "mongoose": 20,
    "mountain_goat": 21,
    "opossum": 22,
    "orangutan": 23,
    "otter": 24,
    "polar_bear": 25,
    "porcupine": 26,
    "red_panda": 27,
    "rhinoceros": 28,
    "sea_lion": 29,
    "seal": 30,
    "snow_leopard": 31,
    "squirrel": 32,
    "sugar_glider": 33,
    "tapir": 34,
    "vampire_bat": 35,
    "vicuna": 36,
    "walrus": 37,
    "warthog": 38,
    "water_buffalo": 39,
    "weasel": 40,
    "wildebeest": 41,
    "wombat": 42,
    "yak": 43,
    "zebra": 44
  },
  "layer_norm_eps": 1e-12,
  "model_type": "vit",
  "num_attention_heads": 12,
  "num_channels": 3,
  "num_hidden_layers": 12,
  "patch_size": 16,
  "problem_type": "single_label_classification",
  "qkv_bias": true,
  "torch_dtype": "float32",
  "transformers_version": "4.38.2"
}