Add model

config.json

@@ -1,24 +1,41 @@
 {
   "architecture": "convnext_base",
-  "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": 1024,
+  "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:37f86b78f9963e96f404f35ea34d206c60b584b8f5969fb66543cd82daeac8d1
-size 350306372
+oid sha256:91d83cd59569fad5afc077af0f060d9d999bafe3f53a2bdad40d325f6f48d538
+size 350304308