Update config.json

config.json

@@ -1,14 +1,15 @@
 {
-  "_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": {
+  "model_type": "resnet50",
+  "model_name": "my_resnet_model",
+  "description": "A custom ResNet-50 based model for image classification",
+  
+  "data": {
+    "input_shape": [250, 250, 3],
+    "num_classes": 30,
+    "data_augmentation": true
+  },
+  
+  "class_names": {
     "0": "Alpinia Galanga (Rasna)",
     "1": "Amaranthus Viridis (Arive-Dantu)",
     "2": "Artocarpus Heterophyllus (Jackfruit)",
@@ -40,48 +41,30 @@
     "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"
+
+  "architecture": {
+    "base_model": "ResNet50",
+    "include_top": false,
+    "weights": "imagenet",
+    "trainable": false
+  },
+
+  "layers": {
+    "global_average_pooling": true,
+    "dropout_rate": 0.2,
+    "dense_units": 512
+  },
+  
+  "training": {
+    "optimizer": {
+      "optimizer_type": "Adam",
+      "learning_rate": 0.0001,
+      "weight_decay": 1e-4
+    },
+    "loss_function": "SparseCategoricalCrossentropy",
+    "metrics": ["accuracy"],
+    "num_epochs": 25,
+    "batch_size": 32
   },
-  "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
-}
+}