botaniscan-models / config.json
anurag629's picture
Update config.json
6118e8b
raw
history blame
3.11 kB
{
"_name_or_path": "google/vit-base-patch16-224-in21k",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"finetuning_task": "image-classification",
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Alpinia Galanga (Rasna)",
"1": "Amaranthus Viridis (Arive-Dantu)",
"2": "Artocarpus Heterophyllus (Jackfruit)",
"3": "Azadirachta Indica (Neem)",
"4": "Basella Alba (Basale)",
"5": "Brassica Juncea (Indian Mustard)",
"6": "Carissa Carandas (Karanda)",
"7": "Citrus Limon (Lemon)",
"8": "Ficus Auriculata (Roxburgh fig)",
"9": "Ficus Religiosa (Peepal Tree)",
"10": "Hibiscus Rosa-sinensis",
"11": "Jasminum (Jasmine)",
"12": "Mangifera Indica (Mango)",
"13": "Mentha (Mint)",
"14": "Moringa Oleifera (Drumstick)",
"15": "Muntingia Calabura (Jamaica Cherry-Gasagase)",
"16": "Murraya Koenigii (Curry)",
"17": "Nerium Oleander (Oleander)",
"18": "Nyctanthes Arbor-tristis (Parijata)",
"19": "Ocimum Tenuiflorum (Tulsi)",
"20": "Piper Betle (Betel)",
"21": "Plectranthus Amboinicus (Mexican Mint)",
"22": "Pongamia Pinnata (Indian Beech)",
"23": "Psidium Guajava (Guava)",
"24": "Punica Granatum (Pomegranate)",
"25": "Santalum Album (Sandalwood)",
"26": "Syzygium Cumini (Jamun)",
"27": "Syzygium Jambos (Rose Apple)",
"28": "Tabernaemontana Divaricata (Crape Jasmine)",
"29": "Trigonella Foenum-graecum (Fenugreek)"
},
"image_size": 250,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Alpinia Galanga (Rasna)": "0",
"Amaranthus Viridis (Arive-Dantu)": "1",
"Artocarpus Heterophyllus (Jackfruit)": "2",
"Azadirachta Indica (Neem)": "3",
"Basella Alba (Basale)": "4",
"Brassica Juncea (Indian Mustard)": "5",
"Carissa Carandas (Karanda)": "6",
"Citrus Limon (Lemon)": "7",
"Ficus Auriculata (Roxburgh fig)": "8",
"Ficus Religiosa (Peepal Tree)": "9",
"Hibiscus Rosa-sinensis": "10",
"Jasminum (Jasmine)": "11",
"Mangifera Indica (Mango)": "12",
"Mentha (Mint)": "13",
"Moringa Oleifera (Drumstick)": "14",
"Muntingia Calabura (Jamaica Cherry-Gasagase)": "15",
"Murraya Koenigii (Curry)": "16",
"Nerium Oleander (Oleander)": "17",
"Nyctanthes Arbor-tristis (Parijata)": "18",
"Ocimum Tenuiflorum (Tulsi)": "19",
"Piper Betle (Betel)": "20",
"Plectranthus Amboinicus (Mexican Mint)": "21",
"Pongamia Pinnata (Indian Beech)": "22",
"Psidium Guajava (Guava)": "23",
"Punica Granatum (Pomegranate)": "24",
"Santalum Album (Sandalwood)": "25",
"Syzygium Cumini (Jamun)": "26",
"Syzygium Jambos (Rose Apple)": "27",
"Tabernaemontana Divaricata (Crape Jasmine)": "28",
"Trigonella Foenum-graecum (Fenugreek)": "29"
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"torch_dtype": "float32",
"transformers_version": "4.8.1",
"from_tf": True
}