huyentls1114's picture
Training in progress, epoch 0
2c8ca31 verified
raw
history blame contribute delete
No virus
3.1 kB
{
"_name_or_path": "apple/mobilevitv2-1.0-imagenet1k-256",
"architectures": [
"MobileViTV2ForImageClassification"
],
"aspp_dropout_prob": 0.1,
"aspp_out_channels": 512,
"atrous_rates": [
6,
12,
18
],
"attn_dropout": 0.0,
"base_attn_unit_dims": [
128,
192,
256
],
"classifier_dropout_prob": 0.1,
"conv_kernel_size": 3,
"expand_ratio": 2.0,
"ffn_dropout": 0.0,
"ffn_multiplier": 2,
"hidden_act": "swish",
"id2label": {
"0": "attic_door",
"1": "bi-fold_door",
"2": "bi-fold_door_double",
"3": "bi-fold_glass_door",
"4": "bi-fold_glass_door_double",
"5": "casement_diamond_window",
"6": "casement_round_window",
"7": "casement_window",
"8": "concern",
"9": "door_opening",
"10": "door_opening_arch",
"11": "double_hung_window",
"12": "elevator_door",
"13": "exterior_door",
"14": "exterior_door_double",
"15": "exterior_glass_door",
"16": "exterior_glass_door_double",
"17": "fixed_arch_window",
"18": "fixed_diamond_window",
"19": "fixed_round_window",
"20": "fixed_sloping_left_window",
"21": "fixed_sloping_right_window",
"22": "fixed_windows",
"23": "french_door",
"24": "french_glass_door",
"25": "garage_door",
"26": "garage_door_double",
"27": "interior_door",
"28": "interior_glass_door",
"29": "interior_glass_door_double",
"30": "patio_door",
"31": "pocket_door",
"32": "pocket_door_double",
"33": "sauna_door",
"34": "sauna_glass_door",
"35": "sliding_door_double",
"36": "sliding_window"
},
"image_size": 256,
"initializer_range": 0.02,
"label2id": {
"attic_door": 0,
"bi-fold_door": 1,
"bi-fold_door_double": 2,
"bi-fold_glass_door": 3,
"bi-fold_glass_door_double": 4,
"casement_diamond_window": 5,
"casement_round_window": 6,
"casement_window": 7,
"concern": 8,
"door_opening": 9,
"door_opening_arch": 10,
"double_hung_window": 11,
"elevator_door": 12,
"exterior_door": 13,
"exterior_door_double": 14,
"exterior_glass_door": 15,
"exterior_glass_door_double": 16,
"fixed_arch_window": 17,
"fixed_diamond_window": 18,
"fixed_round_window": 19,
"fixed_sloping_left_window": 20,
"fixed_sloping_right_window": 21,
"fixed_windows": 22,
"french_door": 23,
"french_glass_door": 24,
"garage_door": 25,
"garage_door_double": 26,
"interior_door": 27,
"interior_glass_door": 28,
"interior_glass_door_double": 29,
"patio_door": 30,
"pocket_door": 31,
"pocket_door_double": 32,
"sauna_door": 33,
"sauna_glass_door": 34,
"sliding_door_double": 35,
"sliding_window": 36
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 2.0,
"model_type": "mobilevitv2",
"n_attn_blocks": [
2,
4,
3
],
"num_channels": 3,
"output_stride": 32,
"patch_size": 2,
"problem_type": "single_label_classification",
"semantic_loss_ignore_index": 255,
"torch_dtype": "float32",
"transformers_version": "4.37.0",
"width_multiplier": 1.0
}