Test different config

config.json

@@ -1,5 +1,14 @@
 {
     "architecture": "vit_base_patch16_384",
-    "image_size": 384,
+    "pretrained_cfg": {
+      "custom_load": false,
+      "input_size": [
+        3,
+        384,
+        384
+      ],
+      "fixed_input_size": true,
+      "first_conv": "patch_embed.proj"
+    },
     "num_classes": 182
 }