anurag629's picture
Training in progress, epoch 1
092f26f
{
"_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
"architectures": [
"SwinForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
6,
2
],
"drop_path_rate": 0.1,
"embed_dim": 96,
"encoder_stride": 32,
"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": 224,
"initializer_range": 0.02,
"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-05,
"mlp_ratio": 4.0,
"model_type": "swin",
"num_channels": 3,
"num_heads": [
3,
6,
12,
24
],
"num_layers": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"path_norm": true,
"problem_type": "single_label_classification",
"qkv_bias": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.28.0",
"use_absolute_embeddings": false,
"window_size": 7
}