ViTuned_buildings / config.json
Alphonsce's picture
changed labels to russian
511ed4d verified
{
"_name_or_path":"google/vit-base-patch16-224-in21k",
"architectures":[
"ViTForImageClassification"
],
"attention_probs_dropout_prob":0.0,
"encoder_stride":16,
"hidden_act":"gelu",
"hidden_dropout_prob":0.0,
"hidden_size":768,
"id2label":{
"0":"осенняя_городская_улица",
"1":"кафе",
"2":"здание_кафе",
"3":"автомобиль_в_городе",
"4":"машины_на_дороге",
"5":"церковное_здание",
"6":"центр_города",
"7":"небоскребы_города",
"8":"строительство",
"9":"толпа_людей",
"10":"фонтан",
"11":"сад",
"12":"здание_сада",
"13":"готическое_здание",
"14":"шоссе",
"15":"отель",
"16":"озеро",
"17":"особняк",
"18":"рынок",
"19":"современный_мост",
"20":"современное_здание",
"21":"современная_городская_улица",
"22":"монастырь",
"23":"памятник",
"24":"музей",
"25":"ночная_городская_улица",
"26":"ночная_зимняя_улица",
"27":"старый_мост",
"28":"старая_городская_улица",
"29":"дворец",
"30":"панельное_здание",
"31":"парковка",
"32":"пешеходная_улица",
"33":"река_в_городе",
"34":"дорога",
"35":"русская_церковь",
"36":"торговый_центр",
"37":"вывеска",
"38":"каток",
"39":"стадион",
"40":"городское_искусство",
"41":"метро",
"42":"летняя_городская_улица",
"43":"высокое_здание",
"44":"театр",
"45":"театральное_здание",
"46":"туристический_замок",
"47":"башня",
"48":"железнодорожный_вокзал",
"49":"деревья",
"50":"деревенские_дома",
"51":"зимний_город",
"52":"зимняя_городская_улица"
},
"image_size":224,
"initializer_range":0.02,
"intermediate_size":3072,
"label2id":{
"autumn_city_street":0,
"cafe":1,
"cafe_building":2,
"car_in_city":3,
"cars_on_road":4,
"church_building":5,
"city_center":6,
"city_skyscrapers":7,
"construction":8,
"crowd_of_people":9,
"fountain":10,
"garden":11,
"garden_building":12,
"gothic_building":13,
"highway":14,
"hotel":15,
"lake":16,
"mansion":17,
"market":18,
"modern_bridge":19,
"modern_building":20,
"modern_city_street":21,
"monastery":22,
"monument":23,
"museum":24,
"night_city_street":25,
"night_winter_street":26,
"old_bridge":27,
"old_city_street":28,
"palace":29,
"panel_building":30,
"parking":31,
"pedestrian_street":32,
"river_in_city":33,
"road":34,
"russian_church":35,
"shopping_mall":36,
"signboard":37,
"skating_ring":38,
"stadium":39,
"street_art":40,
"subway":41,
"summer_city_street":42,
"tall_building":43,
"theatre":44,
"theatre_building":45,
"touristic_castle":46,
"tower":47,
"train_station":48,
"trees":49,
"village_houses":50,
"winter_city":51,
"winter_city_street":52
},
"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.38.2"
}