| { | |
| "_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", | |
| "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.19.2" | |
| } | |