{ | |
"_name_or_path": "./.model/vit-medicinal-plant-finetune-v2/checkpoint-4800", | |
"architectures": [ | |
"ViTForImageClassification" | |
], | |
"attention_probs_dropout_prob": 0.0, | |
"encoder_stride": 16, | |
"hidden_act": "gelu", | |
"hidden_dropout_prob": 0.0, | |
"hidden_size": 768, | |
"id2label": { | |
"0": "Aloevera", | |
"1": "Amla", | |
"2": "Amruta_Balli", | |
"3": "Arali", | |
"4": "Ashoka", | |
"5": "Ashwagandha", | |
"6": "Avacado", | |
"7": "Bamboo", | |
"8": "Basale", | |
"9": "Betel", | |
"10": "Betel_Nut", | |
"11": "Brahmi", | |
"12": "Castor", | |
"13": "Curry_Leaf", | |
"14": "Doddapatre", | |
"15": "Ekka", | |
"16": "Ganike", | |
"17": "Gauva", | |
"18": "Geranium", | |
"19": "Henna", | |
"20": "Hibiscus", | |
"21": "Honge", | |
"22": "Insulin", | |
"23": "Jasmine", | |
"24": "Lemon", | |
"25": "Lemon_grass", | |
"26": "Mango", | |
"27": "Mint", | |
"28": "Nagadali", | |
"29": "Neem", | |
"30": "Nithyapushpa", | |
"31": "Nooni", | |
"32": "Not_a_plant", | |
"33": "Pappaya", | |
"34": "Pepper", | |
"35": "Pomegranate", | |
"36": "Raktachandini", | |
"37": "Rose", | |
"38": "Sapota", | |
"39": "Tulasi", | |
"40": "Wood_sorel" | |
}, | |
"image_size": 224, | |
"initializer_range": 0.02, | |
"intermediate_size": 3072, | |
"label2id": { | |
"Aloevera": "0", | |
"Amla": "1", | |
"Amruta_Balli": "2", | |
"Arali": "3", | |
"Ashoka": "4", | |
"Ashwagandha": "5", | |
"Avacado": "6", | |
"Bamboo": "7", | |
"Basale": "8", | |
"Betel": "9", | |
"Betel_Nut": "10", | |
"Brahmi": "11", | |
"Castor": "12", | |
"Curry_Leaf": "13", | |
"Doddapatre": "14", | |
"Ekka": "15", | |
"Ganike": "16", | |
"Gauva": "17", | |
"Geranium": "18", | |
"Henna": "19", | |
"Hibiscus": "20", | |
"Honge": "21", | |
"Insulin": "22", | |
"Jasmine": "23", | |
"Lemon": "24", | |
"Lemon_grass": "25", | |
"Mango": "26", | |
"Mint": "27", | |
"Nagadali": "28", | |
"Neem": "29", | |
"Nithyapushpa": "30", | |
"Nooni": "31", | |
"Not_a_plant": "32", | |
"Pappaya": "33", | |
"Pepper": "34", | |
"Pomegranate": "35", | |
"Raktachandini": "36", | |
"Rose": "37", | |
"Sapota": "38", | |
"Tulasi": "39", | |
"Wood_sorel": "40" | |
}, | |
"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.41.2" | |
} | |