1aurent
/

vit_small_patch16_224.transpath_mocov3

Feature Extraction

image-classification

Model card Files Files and versions Community

1aurent commited on Nov 1, 2023

Commit

e3bf90f

•

1 Parent(s): e208c7f

Update config.json

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -2,9 +2,10 @@
   "architecture": "vit_small_patch16_224",
   "num_classes": 0,
   "num_features": 384,
   "global_pool": "token",
   "pretrained_cfg": {
-    "tag": "augreg_in21k_ft_in1k",
     "custom_load": true,
     "input_size": [
       3,
@@ -25,7 +26,7 @@
       0.5,
       0.5
     ],
-    "num_classes": 1000,
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"

   "architecture": "vit_small_patch16_224",
   "num_classes": 0,
   "num_features": 384,
+  "num_heads": 12,
   "global_pool": "token",
   "pretrained_cfg": {
+    "tag": "mocov3",
     "custom_load": true,
     "input_size": [
       3,
       0.5,
       0.5
     ],
+    "num_classes": 0,
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"