Add model

config.json

@@ -1,24 +1,34 @@
 {
   "architecture": "vit_base_patch14_dinov2",
-  "as_regression": true,
-  "ema": {
-    "decay": 0.999
-  },
-  "img_size": [
-    512,
-    512
-  ],
-  "lr": 3e-05,
-  "num_classes": 5,
-  "optimizer": {
-    "betas": [
-      0.9,
-      0.999
+  "num_classes": 1,
+  "num_features": 768,
+  "global_pool": "token",
+  "pretrained_cfg": {
+    "tag": "lvd142m",
+    "custom_load": false,
+    "input_size": [
+      3,
+      518,
+      518
     ],
-    "weight_decay": 0.0005
-  },
-  "pretrained": true,
-  "swa": {
-    "swa_lrs": 3e-05
+    "fixed_input_size": true,
+    "interpolation": "bicubic",
+    "crop_pct": 1.0,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 0,
+    "pool_size": null,
+    "first_conv": "patch_embed.proj",
+    "classifier": "head",
+    "license": "apache-2.0"
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ee1443adc0fdb9978471245c8d18cfb55acb2c2b87353193b06b55471dae893
-size 346114908
+oid sha256:50312b1e8ab3d3b4d1805b187be614f55ca6953371ec5cdd4bdc5861f85e3b6d
+size 346113852