{ "_name_or_path": "apple/mobilevit-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": "Acantharia", "1": "Calanoida", "10": "Rhizosolenia", "11": "Thalassiosira", "12": "nauplii_Copepoda", "2": "Chaetoceros_Chaetocerotales", "3": "Ciliophora", "4": "Eutintinnus", "5": "Navicula", "6": "Neoceratium", "7": "Planktoniella_sol", "8": "Pyrophacus", "9": "Rhizaria" }, "image_size": 256, "initializer_range": 0.02, "label2id": { "Acantharia": "0", "Calanoida": "1", "Chaetoceros_Chaetocerotales": "2", "Ciliophora": "3", "Eutintinnus": "4", "Navicula": "5", "Neoceratium": "6", "Planktoniella_sol": "7", "Pyrophacus": "8", "Rhizaria": "9", "Rhizosolenia": "10", "Thalassiosira": "11", "nauplii_Copepoda": "12" }, "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.35.2" }