{ "_name_or_path": "google/vit-base-patch16-224-in21k", "architectures": [ "ViTForImageClassification" ], "attention_probs_dropout_prob": 0.0, "encoder_stride": 16, "finetuning_task": "image-classification", "hidden_act": "gelu", "hidden_dropout_prob": 0.0, "hidden_size": 768, "id2label": { "0": "apple", "1": "aquarium_fish", "10": "bowl", "11": "boy", "12": "bridge", "13": "bus", "14": "butterfly", "15": "camel", "16": "can", "17": "castle", "18": "caterpillar", "19": "cattle", "2": "baby", "20": "chair", "21": "chimpanzee", "22": "clock", "23": "cloud", "24": "cockroach", "25": "couch", "26": "cra", "27": "crocodile", "28": "cup", "29": "dinosaur", "3": "bear", "30": "dolphin", "31": "elephant", "32": "flatfish", "33": "forest", "34": "fox", "35": "girl", "36": "hamster", "37": "house", "38": "kangaroo", "39": "keyboard", "4": "beaver", "40": "lamp", "41": "lawn_mower", "42": "leopard", "43": "lion", "44": "lizard", "45": "lobster", "46": "man", "47": "maple_tree", "48": "motorcycle", "49": "mountain", "5": "bed", "50": "mouse", "51": "mushroom", "52": "oak_tree", "53": "orange", "54": "orchid", "55": "otter", "56": "palm_tree", "57": "pear", "58": "pickup_truck", "59": "pine_tree", "6": "bee", "60": "plain", "61": "plate", "62": "poppy", "63": "porcupine", "64": "possum", "65": "rabbit", "66": "raccoon", "67": "ray", "68": "road", "69": "rocket", "7": "beetle", "70": "rose", "71": "sea", "72": "seal", "73": "shark", "74": "shrew", "75": "skunk", "76": "skyscraper", "77": "snail", "78": "snake", "79": "spider", "8": "bicycle", "80": "squirrel", "81": "streetcar", "82": "sunflower", "83": "sweet_pepper", "84": "table", "85": "tank", "86": "telephone", "87": "television", "88": "tiger", "89": "tractor", "9": "bottle", "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.44.2" }