{ | |
"_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": "Residential", | |
"1": "SeaLake", | |
"2": "AnnualCrop", | |
"3": "Forest", | |
"4": "Pasture", | |
"5": "HerbaceousVegetation", | |
"6": "Highway", | |
"7": "PermanentCrop", | |
"8": "Industrial", | |
"9": "River" | |
}, | |
"image_size": 256, | |
"initializer_range": 0.02, | |
"label2id": { | |
"AnnualCrop": 2, | |
"Forest": 3, | |
"HerbaceousVegetation": 5, | |
"Highway": 6, | |
"Industrial": 8, | |
"Pasture": 4, | |
"PermanentCrop": 7, | |
"Residential": 0, | |
"River": 9, | |
"SeaLake": 1 | |
}, | |
"layer_norm_eps": 1e-05, | |
"mlp_ratio": 2.0, | |
"model_type": "mobilevitv2", | |
"n_attn_blocks": [ | |
2, | |
4, | |
3 | |
], | |
"num_channels": 13, | |
"output_stride": 32, | |
"patch_size": 2, | |
"problem_type": "single_label_classification", | |
"semantic_loss_ignore_index": 255, | |
"torch_dtype": "float32", | |
"transformers_version": "4.38.2", | |
"width_multiplier": 1.0 | |
} | |