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