brand_identification / config.json
mstatt's picture
Upload ViTForImageClassification
fec95aa verified
raw
history blame
3.26 kB
{
"_name_or_path": "./model/vit_model/",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "abu_dhabi_islamic_bank",
"1": "acer",
"2": "adidas",
"3": "adnoc",
"4": "aldar",
"5": "alienware",
"6": "amazon",
"7": "amd",
"8": "apple",
"9": "asus",
"10": "beats_by_dre",
"11": "blackberry",
"12": "bose",
"13": "careem",
"14": "cisco_systems",
"15": "coke",
"16": "d-link",
"17": "dell",
"18": "delonghi",
"19": "dp_world",
"20": "du",
"21": "e&",
"22": "emaar",
"23": "emirates",
"24": "emirates_nbd",
"25": "etisalat",
"26": "falcons.ai",
"27": "first_abu_dhabi_bank",
"28": "fujitsu",
"29": "google",
"30": "gopro",
"31": "hec",
"32": "hewlett_packard",
"33": "hilti",
"34": "hisense",
"35": "huawei",
"36": "ibm",
"37": "khaleej_times",
"38": "l'or\u00e9al",
"39": "lenovo",
"40": "lg",
"41": "linkedin",
"42": "louis_vuitton",
"43": "majid_al_futtaim",
"44": "mashreq",
"45": "maybelline",
"46": "mcdonalds",
"47": "mercedes",
"48": "meta",
"49": "microsoft",
"50": "msi",
"51": "nike",
"52": "nvidia",
"53": "openai",
"54": "puma",
"55": "rakez",
"56": "samsung",
"57": "snapdragon",
"58": "tesla",
"59": "ubuntu",
"60": "virgin",
"61": "zwag"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"abu_dhabi_islamic_bank": "0",
"acer": "1",
"adidas": "2",
"adnoc": "3",
"aldar": "4",
"alienware": "5",
"amazon": "6",
"amd": "7",
"apple": "8",
"asus": "9",
"beats_by_dre": "10",
"blackberry": "11",
"bose": "12",
"careem": "13",
"cisco_systems": "14",
"coke": "15",
"d-link": "16",
"dell": "17",
"delonghi": "18",
"dp_world": "19",
"du": "20",
"e&": "21",
"emaar": "22",
"emirates": "23",
"emirates_nbd": "24",
"etisalat": "25",
"falcons.ai": "26",
"first_abu_dhabi_bank": "27",
"fujitsu": "28",
"google": "29",
"gopro": "30",
"hec": "31",
"hewlett_packard": "32",
"hilti": "33",
"hisense": "34",
"huawei": "35",
"ibm": "36",
"khaleej_times": "37",
"l'or\u00e9al": "38",
"lenovo": "39",
"lg": "40",
"linkedin": "41",
"louis_vuitton": "42",
"majid_al_futtaim": "43",
"mashreq": "44",
"maybelline": "45",
"mcdonalds": "46",
"mercedes": "47",
"meta": "48",
"microsoft": "49",
"msi": "50",
"nike": "51",
"nvidia": "52",
"openai": "53",
"puma": "54",
"rakez": "55",
"samsung": "56",
"snapdragon": "57",
"tesla": "58",
"ubuntu": "59",
"virgin": "60",
"zwag": "61"
},
"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.41.2"
}