funkepal's picture
Upload ViTForImageClassification
b44f9a6 verified
raw
history blame
2.39 kB
{
"_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"
}