Add model

README.md

@@ -0,0 +1,14 @@
+---
+tags:
+- image-classification
+- timm
+library_name: timm
+license: apache-2.0
+---
+# Model card for vit_small_patch16_reg1_224.st_safebooru_1k
+
+## Model Details
+- **metrics:** 
+|Precision|Recall|F1-score|
+|-|-|-|
+|0.7777145824180677|0.428464250130301|0.5269828911507399|

config.json

@@ -0,0 +1,36 @@
+{
+  "architecture": "vit_small_patch16_224",
+  "num_classes": 1000,
+  "num_features": 384,
+  "global_pool": "token",
+  "model_args": {
+    "reg_tokens": 1
+  },
+  "pretrained_cfg": {
+    "tag": "st_safebooru_1k",
+    "custom_load": true,
+    "input_size": [
+      3,
+      224,
+      224
+    ],
+    "fixed_input_size": true,
+    "interpolation": "bicubic",
+    "crop_pct": 0.9,
+    "crop_mode": "center",
+    "mean": [
+      0.6886,
+      0.6415,
+      0.6443
+    ],
+    "std": [
+      0.183,
+      0.1852,
+      0.1849
+    ],
+    "num_classes": 1000,
+    "pool_size": null,
+    "first_conv": "patch_embed.proj",
+    "classifier": "head"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:029bf1f702ca246077b8e9e90484b52746f34cb6a425259ea14e389bf57048d4
+size 88219656

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e647789bd33c1fc95edac18f93f7c65afc58068e9b8699523f1e13d3214d147
+size 88265070