Add model

config.json

@@ -1,24 +1,43 @@
 {
   "architecture": "resnet18",
-  "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": 512,
+  "pretrained_cfg": {
+    "tag": "a1_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.95,
+    "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": "conv1",
+    "classifier": "fc",
+    "origin_url": "https://github.com/huggingface/pytorch-image-models",
+    "paper_ids": "arXiv:2110.00476"
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:da586f3c2ee1ca6fef6366ae8bd0a858a0610d640a37216c6a7265ee6faad134
-size 44758148
+oid sha256:16c14a32771743dcfd3e36eaca308ff332a8fefc2e6a0c30104859983983a9c1
+size 44757412