Update config.json

config.json

@@ -1,18 +1,15 @@
 {
-  "model_type": "vision_transformer",
+  "model_type": "resnet",
   "config": {
-    "image_size": 224,
-    "patch_size": 16,
+    "layers": 18,
     "num_classes": 1000,
-    "hidden_size": 768,
-    "num_attention_heads": 12,
-    "num_hidden_layers": 12,
-    "intermediate_size": 3072,
-    "hidden_dropout_prob": 0.1,
-    "attention_probs_dropout_prob": 0.1,
-    "initializer_range": 0.02,
-    "layer_norm_eps": 1e-12,
-    "output_attentions": false,
-    "output_hidden_states": false
+    "input_size": 224,
+    "batch_norm": true,
+    "pretrained": true,
+    "dropout_rate": 0.5,
+    "optimizer": {
+      "type": "Adam",
+      "learning_rate": 0.001
+    }
   }
 }