{ "_name_or_path": "apple/deeplabv3-mobilevit-small", "architectures": [ "MobileViTForSemanticSegmentation" ], "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": 4.0, "hidden_act": "silu", "hidden_dropout_prob": 0.1, "hidden_sizes": [ 144, 192, 240 ], "id2label": { "0": "background", "1": "aeroplane", "2": "bicycle", "3": "bird", "4": "boat", "5": "bottle", "6": "bus", "7": "car", "8": "cat", "9": "chair", "10": "cow", "11": "diningtable", "12": "dog", "13": "horse", "14": "motorbike", "15": "person", "16": "pottedplant", "17": "sheep", "18": "sofa", "19": "train", "20": "tvmonitor" }, "image_size": 512, "initializer_range": 0.02, "label2id": { "aeroplane": 1, "background": 0, "bicycle": 2, "bird": 3, "boat": 4, "bottle": 5, "bus": 6, "car": 7, "cat": 8, "chair": 9, "cow": 10, "diningtable": 11, "dog": 12, "horse": 13, "motorbike": 14, "person": 15, "pottedplant": 16, "sheep": 17, "sofa": 18, "train": 19, "tvmonitor": 20 }, "layer_norm_eps": 1e-05, "mlp_ratio": 2.0, "model_type": "mobilevit", "neck_hidden_sizes": [ 16, 32, 64, 96, 128, 160, 640 ], "neuron": { "auto_cast": null, "auto_cast_type": null, "compiler_type": "neuronx-cc", "compiler_version": "2.13.66.0+6dfecc895", "dynamic_batch_size": false, "inline_weights_to_neff": false, "input_names": [ "pixel_values" ], "model_type": "mobilevit", "optlevel": "2", "output_attentions": false, "output_hidden_states": false, "output_names": [ "logits" ], "static_batch_size": 1, "static_image_size": 512, "static_num_channels": 3, "static_patch_size": 2 }, "num_attention_heads": 4, "num_channels": 3, "output_stride": 16, "patch_size": 2, "qkv_bias": true, "semantic_loss_ignore_index": 255, "task": "semantic-segmentation", "torchscript": true, "transformers_version": "4.40.2" }