neuralhaven's picture
neuralhaven/tinyvit2mobilevitxxs_FT
9deadf1 verified
raw
history blame
3.04 kB
{
"_name_or_path": "neuralhaven/KDRSSC_TinyViT2MobileViT-xx-small",
"architectures": [
"MobileViTForImageClassification"
],
"aspp_dropout_prob": 0.1,
"aspp_out_channels": 256,
"atrous_rates": [
6,
12,
18
],
"attention_probs_dropout_prob": 0.0,
"classifier_dropout_prob": 0.1,
"conv_kernel_size": 3,
"expand_ratio": 2.0,
"hidden_act": "silu",
"hidden_dropout_prob": 0.05,
"hidden_sizes": [
64,
80,
96
],
"id2label": {
"0": "airplane",
"1": "airport",
"2": "baseball_diamond",
"3": "basketball_court",
"4": "beach",
"5": "bridge",
"6": "chaparral",
"7": "church",
"8": "circular_farmland",
"9": "cloud",
"10": "commercial_area",
"11": "dense_residential",
"12": "desert",
"13": "forest",
"14": "freeway",
"15": "golf_course",
"16": "ground_track_field",
"17": "harbor",
"18": "industrial_area",
"19": "intersection",
"20": "island",
"21": "lake",
"22": "meadow",
"23": "medium_residential",
"24": "mobile_home_park",
"25": "mountain",
"26": "overpass",
"27": "palace",
"28": "parking_lot",
"29": "railway",
"30": "railway_station",
"31": "rectangular_farmland",
"32": "river",
"33": "roundabout",
"34": "runway",
"35": "sea_ice",
"36": "ship",
"37": "snowberg",
"38": "sparse_residential",
"39": "stadium",
"40": "storage_tank",
"41": "tennis_court",
"42": "terrace",
"43": "thermal_power_station",
"44": "wetland"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"airplane": 0,
"airport": 1,
"baseball_diamond": 2,
"basketball_court": 3,
"beach": 4,
"bridge": 5,
"chaparral": 6,
"church": 7,
"circular_farmland": 8,
"cloud": 9,
"commercial_area": 10,
"dense_residential": 11,
"desert": 12,
"forest": 13,
"freeway": 14,
"golf_course": 15,
"ground_track_field": 16,
"harbor": 17,
"industrial_area": 18,
"intersection": 19,
"island": 20,
"lake": 21,
"meadow": 22,
"medium_residential": 23,
"mobile_home_park": 24,
"mountain": 25,
"overpass": 26,
"palace": 27,
"parking_lot": 28,
"railway": 29,
"railway_station": 30,
"rectangular_farmland": 31,
"river": 32,
"roundabout": 33,
"runway": 34,
"sea_ice": 35,
"ship": 36,
"snowberg": 37,
"sparse_residential": 38,
"stadium": 39,
"storage_tank": 40,
"tennis_court": 41,
"terrace": 42,
"thermal_power_station": 43,
"wetland": 44
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 2.0,
"model_type": "mobilevit",
"neck_hidden_sizes": [
16,
16,
24,
48,
64,
80,
320
],
"num_attention_heads": 4,
"num_channels": 3,
"output_stride": 32,
"patch_size": 2,
"problem_type": "single_label_classification",
"qkv_bias": true,
"semantic_loss_ignore_index": 255,
"torch_dtype": "float32",
"transformers_version": "4.44.0"
}