1aurent
/

vit_small_patch16_224.lunit_dino

Image Classification

Model card Files Files and versions

1aurent commited on Nov 26, 2023

Commit

a2c3f3a

·

1 Parent(s): 27ed6d6

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
   "num_features": 384,
   "global_pool": "token",
   "pretrained_cfg": {
-    "tag": "augreg_in21k",
     "custom_load": true,
     "input_size": [
       3,
@@ -25,7 +25,7 @@
       0.5,
       0.5
     ],
-    "num_classes": 21843,
     "pool_size": null,
     "first_conv": "patch_embed.proj",
     "classifier": "head"

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