car_identified_model_7 / config.json
EstherSan's picture
Training in progress, epoch 0
0c64e10
raw
history blame
2.26 kB
{
"_name_or_path": "apple/mobilevitv2-1.0-imagenet1k-256",
"architectures": [
"MobileViTForImageClassification"
],
"aspp_dropout_prob": 0.1,
"aspp_out_channels": 512,
"atrous_rates": [
6,
12,
18
],
"attention_probs_dropout_prob": 0.0,
"attn_dropout": 0.0,
"classifier_dropout_prob": 0.1,
"conv_kernel_size": 3,
"expand_ratio": 2.0,
"ffn_dropout": 0.0,
"hidden_act": "swish",
"hidden_dropout_prob": 0.1,
"hidden_sizes": [
144,
192,
240
],
"id2label": {
"0": "Toyota_Corolla",
"1": "Hyundai_Ioniq",
"10": "Ford_Mondeo",
"11": "Mazda2_Hybrid",
"12": "Mercedes-Benz_E-Class",
"13": "Volkswagen_Polo",
"14": "Renault_Kangoo",
"15": "Ford_Focus",
"16": "Mercedes-Benz_C-Class",
"17": "Volkswagen_Golf",
"18": "SEAT_Leon",
"19": "Dacia_Dokker",
"2": "SEAT_IBIZA",
"20": "black",
"21": "silver",
"22": "white",
"23": "red",
"24": "blue",
"3": "Fiat_Egea",
"4": "Renault_Clio",
"5": "Audi_A3",
"6": "Toyota_Prius",
"7": "Kia_Rio",
"8": "Fiat_Tipo",
"9": "Citroen_C3"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"Audi_A3": "5",
"Citroen_C3": "9",
"Dacia_Dokker": "19",
"Fiat_Egea": "3",
"Fiat_Tipo": "8",
"Ford_Focus": "15",
"Ford_Mondeo": "10",
"Hyundai_Ioniq": "1",
"Kia_Rio": "7",
"Mazda2_Hybrid": "11",
"Mercedes-Benz_C-Class": "16",
"Mercedes-Benz_E-Class": "12",
"Renault_Clio": "4",
"Renault_Kangoo": "14",
"SEAT_IBIZA": "2",
"SEAT_Leon": "18",
"Toyota_Corolla": "0",
"Toyota_Prius": "6",
"Volkswagen_Golf": "17",
"Volkswagen_Polo": "13",
"black": "20",
"blue": "24",
"red": "23",
"silver": "21",
"white": "22"
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 2.0,
"model_type": "mobilevit",
"neck_hidden_sizes": [
16,
32,
64,
96,
128,
160,
640
],
"num_attention_heads": 4,
"num_channels": 3,
"output_stride": 32,
"patch_size": 2,
"problem_type": "multi_label_classification",
"qkv_bias": true,
"semantic_loss_ignore_index": 255,
"torch_dtype": "float32",
"transformers_version": "4.35.0",
"width_multiplier": 1.0
}