{ | |
"_name_or_path": "google/vit-base-patch16-224-in21k", | |
"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.35.2" | |
} | |