|
{ |
|
"_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": "apple", |
|
"1": "aquarium_fish", |
|
"2": "baby", |
|
"3": "bear", |
|
"4": "beaver", |
|
"5": "bed", |
|
"6": "bee", |
|
"7": "beetle", |
|
"8": "bicycle", |
|
"9": "bottle", |
|
"10": "bowl", |
|
"11": "boy", |
|
"12": "bridge", |
|
"13": "bus", |
|
"14": "butterfly", |
|
"15": "camel", |
|
"16": "can", |
|
"17": "castle", |
|
"18": "caterpillar", |
|
"19": "cattle", |
|
"20": "chair", |
|
"21": "chimpanzee", |
|
"22": "clock", |
|
"23": "cloud", |
|
"24": "cockroach", |
|
"25": "couch", |
|
"26": "cra", |
|
"27": "crocodile", |
|
"28": "cup", |
|
"29": "dinosaur", |
|
"30": "dolphin", |
|
"31": "elephant", |
|
"32": "flatfish", |
|
"33": "forest", |
|
"34": "fox", |
|
"35": "girl", |
|
"36": "hamster", |
|
"37": "house", |
|
"38": "kangaroo", |
|
"39": "keyboard", |
|
"40": "lamp", |
|
"41": "lawn_mower", |
|
"42": "leopard", |
|
"43": "lion", |
|
"44": "lizard", |
|
"45": "lobster", |
|
"46": "man", |
|
"47": "maple_tree", |
|
"48": "motorcycle", |
|
"49": "mountain", |
|
"50": "mouse", |
|
"51": "mushroom", |
|
"52": "oak_tree", |
|
"53": "orange", |
|
"54": "orchid", |
|
"55": "otter", |
|
"56": "palm_tree", |
|
"57": "pear", |
|
"58": "pickup_truck", |
|
"59": "pine_tree", |
|
"60": "plain", |
|
"61": "plate", |
|
"62": "poppy", |
|
"63": "porcupine", |
|
"64": "possum", |
|
"65": "rabbit", |
|
"66": "raccoon", |
|
"67": "ray", |
|
"68": "road", |
|
"69": "rocket", |
|
"70": "rose", |
|
"71": "sea", |
|
"72": "seal", |
|
"73": "shark", |
|
"74": "shrew", |
|
"75": "skunk", |
|
"76": "skyscraper", |
|
"77": "snail", |
|
"78": "snake", |
|
"79": "spider", |
|
"80": "squirrel", |
|
"81": "streetcar", |
|
"82": "sunflower", |
|
"83": "sweet_pepper", |
|
"84": "table", |
|
"85": "tank", |
|
"86": "telephone", |
|
"87": "television", |
|
"88": "tiger", |
|
"89": "tractor", |
|
"90": "train", |
|
"91": "trout", |
|
"92": "tulip", |
|
"93": "turtle", |
|
"94": "wardrobe", |
|
"95": "whale", |
|
"96": "willow_tree", |
|
"97": "wolf", |
|
"98": "woman", |
|
"99": "worm" |
|
}, |
|
"image_size": 224, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"label2id": { |
|
"apple": "0", |
|
"aquarium_fish": "1", |
|
"baby": "2", |
|
"bear": "3", |
|
"beaver": "4", |
|
"bed": "5", |
|
"bee": "6", |
|
"beetle": "7", |
|
"bicycle": "8", |
|
"bottle": "9", |
|
"bowl": "10", |
|
"boy": "11", |
|
"bridge": "12", |
|
"bus": "13", |
|
"butterfly": "14", |
|
"camel": "15", |
|
"can": "16", |
|
"castle": "17", |
|
"caterpillar": "18", |
|
"cattle": "19", |
|
"chair": "20", |
|
"chimpanzee": "21", |
|
"clock": "22", |
|
"cloud": "23", |
|
"cockroach": "24", |
|
"couch": "25", |
|
"cra": "26", |
|
"crocodile": "27", |
|
"cup": "28", |
|
"dinosaur": "29", |
|
"dolphin": "30", |
|
"elephant": "31", |
|
"flatfish": "32", |
|
"forest": "33", |
|
"fox": "34", |
|
"girl": "35", |
|
"hamster": "36", |
|
"house": "37", |
|
"kangaroo": "38", |
|
"keyboard": "39", |
|
"lamp": "40", |
|
"lawn_mower": "41", |
|
"leopard": "42", |
|
"lion": "43", |
|
"lizard": "44", |
|
"lobster": "45", |
|
"man": "46", |
|
"maple_tree": "47", |
|
"motorcycle": "48", |
|
"mountain": "49", |
|
"mouse": "50", |
|
"mushroom": "51", |
|
"oak_tree": "52", |
|
"orange": "53", |
|
"orchid": "54", |
|
"otter": "55", |
|
"palm_tree": "56", |
|
"pear": "57", |
|
"pickup_truck": "58", |
|
"pine_tree": "59", |
|
"plain": "60", |
|
"plate": "61", |
|
"poppy": "62", |
|
"porcupine": "63", |
|
"possum": "64", |
|
"rabbit": "65", |
|
"raccoon": "66", |
|
"ray": "67", |
|
"road": "68", |
|
"rocket": "69", |
|
"rose": "70", |
|
"sea": "71", |
|
"seal": "72", |
|
"shark": "73", |
|
"shrew": "74", |
|
"skunk": "75", |
|
"skyscraper": "76", |
|
"snail": "77", |
|
"snake": "78", |
|
"spider": "79", |
|
"squirrel": "80", |
|
"streetcar": "81", |
|
"sunflower": "82", |
|
"sweet_pepper": "83", |
|
"table": "84", |
|
"tank": "85", |
|
"telephone": "86", |
|
"television": "87", |
|
"tiger": "88", |
|
"tractor": "89", |
|
"train": "90", |
|
"trout": "91", |
|
"tulip": "92", |
|
"turtle": "93", |
|
"wardrobe": "94", |
|
"whale": "95", |
|
"willow_tree": "96", |
|
"wolf": "97", |
|
"woman": "98", |
|
"worm": "99" |
|
}, |
|
"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.33.3" |
|
} |
|
|