Update config.json

config.json

@@ -1,9 +1,46 @@
 {
+  "model_type": "custom_resnet",
+  "auto_map": {
+    "AutoConfig": "modeling_resnet.ResNetConfig",
+    "AutoModel": "modeling_resnet.ResNet"
+  },
   "architectures": [
     "ResNet"
   ],
-  "dtype": "float32",
-  "model_type": "custom_resnet",
   "num_classes": 10,
-  "transformers_version": "4.57.1"
-}
+  "input_channels": 3,
+  "image_size": 64,
+  "activation_function": "relu",
+  "block_type": "resnet",
+  "layers": [3, 4, 6, 3],
+  "hidden_sizes": [64, 128, 256, 512],
+  "classifier_dropout": 0.0,
+  "label_smoothing": 0.1,
+  "problem_type": "image_classification",
+  "id2label": {
+    "0": "0",
+    "1": "1",
+    "2": "2",
+    "3": "3",
+    "4": "4",
+    "5": "5",
+    "6": "6",
+    "7": "7",
+    "8": "8",
+    "9": "9"
+  },
+  "label2id": {
+    "0": 0,
+    "1": 1,
+    "2": 2,
+    "3": 3,
+    "4": 4,
+    "5": 5,
+    "6": 6,
+    "7": 7,
+    "8": 8,
+    "9": 9
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.47.1"
+}