Update config.json

config.json

@@ -1,22 +1,15 @@
 {
-  "architectures": [
-    "ViTForImageClassification"
-  ],
-  "attention_probs_dropout_prob": 0.0,
-  "encoder_stride": 16,
-  "hidden_act": "gelu",
-  "hidden_dropout_prob": 0.0,
-  "hidden_size": 768,
-  "image_size": 224,
-  "initializer_range": 0.02,
-  "intermediate_size": 3072,
-  "layer_norm_eps": 1e-12,
-  "model_type": "vit",
-  "num_attention_heads": 12,
-  "num_channels": 3,
-  "num_hidden_layers": 12,
-  "patch_size": 16,
-  "qkv_bias": true,
-  "torch_dtype": "float32",
-  "transformers_version": "4.38.1"
-}
+    "model_name": "your_model_name",
+    "architecture": "resnet18",
+    "num_classes": 10,
+    "input_size": [3, 224, 224],
+    "batch_size": 32,
+    "learning_rate": 0.001,
+    "epochs": 10,
+    "optimizer": "Adam",
+    "loss_function": "CrossEntropyLoss",
+    "pretrained": true,
+    "dropout": 0.5,
+    "seed": 42,
+    "device": "cuda"  // or "cpu"
+}