wellCh4n's picture
Upload model
0ee3bcf verified
{
"_name_or_path": "/Users/chenweihao/Desktop/\u8bba\u6587/classification/google:vit-base-patch16-224-in21k",
"architectures": [
"ViTModel"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"finetuning_task": "image-classification",
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "A healthy tomato leaf",
"1": "A tomato leaf with Leaf Mold",
"2": "A tomato leaf with Target Spot",
"3": "A tomato leaf with Late Blight",
"4": "A tomato leaf with Early Blight",
"5": "A tomato leaf with Bacterial Spot",
"6": "A tomato leaf with Septoria Leaf Spot",
"7": "A tomato leaf with Tomato Mosaic Virus",
"8": "A tomato leaf with Tomato Yellow Leaf Curl Virus",
"9": "A tomato leaf with Spider Mites Two-spotted Spider Mite"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"A healthy tomato leaf": "0",
"A tomato leaf with Bacterial Spot": "5",
"A tomato leaf with Early Blight": "4",
"A tomato leaf with Late Blight": "3",
"A tomato leaf with Leaf Mold": "1",
"A tomato leaf with Septoria Leaf Spot": "6",
"A tomato leaf with Spider Mites Two-spotted Spider Mite": "9",
"A tomato leaf with Target Spot": "2",
"A tomato leaf with Tomato Mosaic Virus": "7",
"A tomato leaf with Tomato Yellow Leaf Curl Virus": "8"
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.47.1"
}