Add model

config.json

@@ -1,24 +1,41 @@
 {
   "architecture": "convnext_large",
-  "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": 1536,
+  "pretrained_cfg": {
+    "tag": "fb_in22k_ft_in1k",
+    "custom_load": false,
+    "input_size": [
+      3,
+      224,
+      224
     ],
-    "weight_decay": 0.0005
-  },
-  "pretrained": true,
-  "swa": {
-    "swa_lrs": 3e-05
+    "test_input_size": [
+      3,
+      288,
+      288
+    ],
+    "fixed_input_size": false,
+    "interpolation": "bicubic",
+    "crop_pct": 0.875,
+    "test_crop_pct": 1.0,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 1000,
+    "pool_size": [
+      7,
+      7
+    ],
+    "first_conv": "stem.0",
+    "classifier": "head.fc"
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2d068685d5b688dc10d9fbffc9e873f8cb63476276a677f63d056e6c22226e4a
-size 784964084
+oid sha256:fd78ad74f6800d6f7393e5adc1104e8a552f42172524495b11348dc5e98b536b
+size 784962020