custom / config.json
arpanl's picture
Model save
ecd0007
{
"_name_or_path": "google/vit-base-patch16-224",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Faces",
"1": "Faces_easy",
"10": "beaver",
"100": "yin_yang",
"11": "binocular",
"12": "bonsai",
"13": "brain",
"14": "brontosaurus",
"15": "buddha",
"16": "butterfly",
"17": "camera",
"18": "cannon",
"19": "car_side",
"2": "Leopards",
"20": "ceiling_fan",
"21": "cellphone",
"22": "chair",
"23": "chandelier",
"24": "cougar_body",
"25": "cougar_face",
"26": "crab",
"27": "crayfish",
"28": "crocodile",
"29": "crocodile_head",
"3": "Motorbikes",
"30": "cup",
"31": "dalmatian",
"32": "dollar_bill",
"33": "dolphin",
"34": "dragonfly",
"35": "electric_guitar",
"36": "elephant",
"37": "emu",
"38": "euphonium",
"39": "ewer",
"4": "accordion",
"40": "ferry",
"41": "flamingo",
"42": "flamingo_head",
"43": "garfield",
"44": "gerenuk",
"45": "gramophone",
"46": "grand_piano",
"47": "hawksbill",
"48": "headphone",
"49": "hedgehog",
"5": "airplanes",
"50": "helicopter",
"51": "ibis",
"52": "inline_skate",
"53": "joshua_tree",
"54": "kangaroo",
"55": "ketch",
"56": "lamp",
"57": "laptop",
"58": "llama",
"59": "lobster",
"6": "anchor",
"60": "lotus",
"61": "mandolin",
"62": "mayfly",
"63": "menorah",
"64": "metronome",
"65": "minaret",
"66": "nautilus",
"67": "octopus",
"68": "okapi",
"69": "pagoda",
"7": "ant",
"70": "panda",
"71": "pigeon",
"72": "pizza",
"73": "platypus",
"74": "pyramid",
"75": "revolver",
"76": "rhino",
"77": "rooster",
"78": "saxophone",
"79": "schooner",
"8": "barrel",
"80": "scissors",
"81": "scorpion",
"82": "sea_horse",
"83": "snoopy",
"84": "soccer_ball",
"85": "stapler",
"86": "starfish",
"87": "stegosaurus",
"88": "stop_sign",
"89": "strawberry",
"9": "bass",
"90": "sunflower",
"91": "tick",
"92": "trilobite",
"93": "umbrella",
"94": "watch",
"95": "water_lilly",
"96": "wheelchair",
"97": "wild_cat",
"98": "windsor_chair",
"99": "wrench"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Faces": "0",
"Faces_easy": "1",
"Leopards": "2",
"Motorbikes": "3",
"accordion": "4",
"airplanes": "5",
"anchor": "6",
"ant": "7",
"barrel": "8",
"bass": "9",
"beaver": "10",
"binocular": "11",
"bonsai": "12",
"brain": "13",
"brontosaurus": "14",
"buddha": "15",
"butterfly": "16",
"camera": "17",
"cannon": "18",
"car_side": "19",
"ceiling_fan": "20",
"cellphone": "21",
"chair": "22",
"chandelier": "23",
"cougar_body": "24",
"cougar_face": "25",
"crab": "26",
"crayfish": "27",
"crocodile": "28",
"crocodile_head": "29",
"cup": "30",
"dalmatian": "31",
"dollar_bill": "32",
"dolphin": "33",
"dragonfly": "34",
"electric_guitar": "35",
"elephant": "36",
"emu": "37",
"euphonium": "38",
"ewer": "39",
"ferry": "40",
"flamingo": "41",
"flamingo_head": "42",
"garfield": "43",
"gerenuk": "44",
"gramophone": "45",
"grand_piano": "46",
"hawksbill": "47",
"headphone": "48",
"hedgehog": "49",
"helicopter": "50",
"ibis": "51",
"inline_skate": "52",
"joshua_tree": "53",
"kangaroo": "54",
"ketch": "55",
"lamp": "56",
"laptop": "57",
"llama": "58",
"lobster": "59",
"lotus": "60",
"mandolin": "61",
"mayfly": "62",
"menorah": "63",
"metronome": "64",
"minaret": "65",
"nautilus": "66",
"octopus": "67",
"okapi": "68",
"pagoda": "69",
"panda": "70",
"pigeon": "71",
"pizza": "72",
"platypus": "73",
"pyramid": "74",
"revolver": "75",
"rhino": "76",
"rooster": "77",
"saxophone": "78",
"schooner": "79",
"scissors": "80",
"scorpion": "81",
"sea_horse": "82",
"snoopy": "83",
"soccer_ball": "84",
"stapler": "85",
"starfish": "86",
"stegosaurus": "87",
"stop_sign": "88",
"strawberry": "89",
"sunflower": "90",
"tick": "91",
"trilobite": "92",
"umbrella": "93",
"watch": "94",
"water_lilly": "95",
"wheelchair": "96",
"wild_cat": "97",
"windsor_chair": "98",
"wrench": "99",
"yin_yang": "100"
},
"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.0"
}