vit-cifar100-cifar100 / config.json
pkr7098's picture
End of training
8e29bb5 verified
raw
history blame
4.73 kB
{
"_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"
}