BVRA
/

vit_base_patch16_384.ft_df20_384

Image Classification

Model card Files Files and versions Community

MHanzl commited on Feb 12

Commit

86f0d43

•

1 Parent(s): 9c18579

Fix config

Files changed (1) hide show

config.json +7 -2

config.json CHANGED Viewed

@@ -1,9 +1,14 @@
 {
     "architecture": "vit_base_patch16_384",
-    "input_size": [
         3,
         384,
         384
-    ],
     "num_classes": 1604
 }

 {
     "architecture": "vit_base_patch16_384",
+    "pretrained_cfg": {
+      "custom_load": false,
+      "input_size": [
         3,
         384,
         384
+      ],
+      "fixed_input_size": true,
+      "first_conv": "patch_embed.proj"
+    },
     "num_classes": 1604
 }