Test new size format

config.json

@@ -1,9 +1,5 @@
 {
     "architecture": "vit_base_patch16_384",
-    "input_size": [
-        3,
-        384,
-        384
-    ],
+    "image_size": 384,
     "num_classes": 182
 }