Upload deit_base_patch16_224 fine-tuned for document classification

README.md

@@ -0,0 +1,8 @@
+---
+tags:
+- image-classification
+- timm
+library_name: timm
+license: apache-2.0
+---
+# Model card for document-classifier-deit_base_patch16_224

config.json

@@ -0,0 +1,49 @@
+{
+  "architecture": "deit_base_patch16_224",
+  "num_classes": 3,
+  "num_features": 768,
+  "global_pool": "token",
+  "label_names": [
+    "contains_figure",
+    "contains_table",
+    "contains_text"
+  ],
+  "pretrained_cfg": {
+    "tag": "fb_in1k",
+    "custom_load": false,
+    "input_size": [
+      3,
+      224,
+      224
+    ],
+    "fixed_input_size": true,
+    "interpolation": "bicubic",
+    "crop_pct": 0.9,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 1000,
+    "pool_size": null,
+    "first_conv": "patch_embed.proj",
+    "classifier": "head"
+  },
+  "base_model": "deit_base_patch16_224",
+  "best_f1_score": 0.8557077625570777,
+  "hyperparameters": {
+    "model_name": "deit_base_patch16_224",
+    "learning_rate": 3.8234752246751835e-05,
+    "batch_size": 8.0,
+    "dropout_rate": 0.2465447373174767,
+    "hidden_size": 512.0,
+    "weight_decay": 0.00010677482709481354
+  },
+  "input_size": 224
+}

model.safetensors

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

pytorch_model.bin

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