STEM-AI-mtl's picture
Upload ViTForImageClassification
c20912e verified
{
"_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": "Ottawa",
"1": "Seoul",
"2": "Philadelphie",
"3": "Dakar",
"4": "Paris",
"5": "Sydney",
"6": "Toronto",
"7": "Montreal",
"8": "Istanbul",
"9": "Venise",
"10": "Regina",
"11": "Mumbai",
"12": "Dubai",
"13": "Seattle",
"14": "Fredericton",
"15": "St-John",
"16": "Edmonton",
"17": "Los_Angeles",
"18": "Winnipeg",
"19": "Chicago",
"20": "Singapore",
"21": "Stockholm",
"22": "Halifax",
"23": "Barcelone",
"24": "Londres",
"25": "Vancouver",
"26": "Las_Vegas",
"27": "New_York",
"28": "Prague",
"29": "Amsterdam",
"30": "Calgary",
"31": "Miami",
"32": "Washington",
"33": "Dublin",
"34": "Charlottetown",
"35": "Budapest",
"36": "Boston",
"37": "Oslo",
"38": "Zurich",
"39": "Cairo",
"40": "Quebec",
"41": "Tokyo",
"42": "Melbourne",
"43": "Hong_Kong",
"44": "San_Francisco"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Amsterdam": 29,
"Barcelone": 23,
"Boston": 36,
"Budapest": 35,
"Cairo": 39,
"Calgary": 30,
"Charlottetown": 34,
"Chicago": 19,
"Dakar": 3,
"Dubai": 12,
"Dublin": 33,
"Edmonton": 16,
"Fredericton": 14,
"Halifax": 22,
"Hong_Kong": 43,
"Istanbul": 8,
"Las_Vegas": 26,
"Londres": 24,
"Los_Angeles": 17,
"Melbourne": 42,
"Miami": 31,
"Montreal": 7,
"Mumbai": 11,
"New_York": 27,
"Oslo": 37,
"Ottawa": 0,
"Paris": 4,
"Philadelphie": 2,
"Prague": 28,
"Quebec": 40,
"Regina": 10,
"San_Francisco": 44,
"Seattle": 13,
"Seoul": 1,
"Singapore": 20,
"St-John": 15,
"Stockholm": 21,
"Sydney": 5,
"Tokyo": 41,
"Toronto": 6,
"Vancouver": 25,
"Venise": 9,
"Washington": 32,
"Winnipeg": 18,
"Zurich": 38
},
"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.39.2"
}