🍻 cheers

README.md

@@ -2,7 +2,6 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
 - generated_from_trainer
 datasets:
 - imagefolder
@@ -15,10 +14,10 @@ model-index:
       name: Image Classification
       type: image-classification
     dataset:
-      name: customize
+      name: imagefolder
       type: imagefolder
       config: default
-      split: validation
+      split: test
       args: default
     metrics:
     - name: Accuracy
@@ -31,9 +30,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-id-card
 
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the customize dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0524
+- Loss: 0.0297
 - Accuracy: 1.0
 
 ## Model description

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 0.052425507456064224,
-    "eval_runtime": 2.2674,
-    "eval_samples_per_second": 11.908,
-    "eval_steps_per_second": 1.764,
-    "train_loss": 0.23122912645339966,
-    "train_runtime": 88.6604,
-    "train_samples_per_second": 5.098,
-    "train_steps_per_second": 0.361
+    "eval_loss": 0.02969413250684738,
+    "eval_runtime": 1.9654,
+    "eval_samples_per_second": 13.737,
+    "eval_steps_per_second": 2.035,
+    "train_loss": 0.05574157414957881,
+    "train_runtime": 78.7822,
+    "train_samples_per_second": 5.737,
+    "train_steps_per_second": 0.406
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 0.052425507456064224,
-    "eval_runtime": 2.2674,
-    "eval_samples_per_second": 11.908,
-    "eval_steps_per_second": 1.764
+    "eval_loss": 0.02969413250684738,
+    "eval_runtime": 1.9654,
+    "eval_samples_per_second": 13.737,
+    "eval_steps_per_second": 2.035
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3def77414fcb47b065227793279426367016697945afcdef6074883647ce2bb9
+oid sha256:80ce1343622493b270fe96e7a466ec2330a9c0103beb2c0774eb2b47f89c0d44
 size 343227052

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 4.0,
-    "train_loss": 0.23122912645339966,
-    "train_runtime": 88.6604,
-    "train_samples_per_second": 5.098,
-    "train_steps_per_second": 0.361
+    "train_loss": 0.05574157414957881,
+    "train_runtime": 78.7822,
+    "train_samples_per_second": 5.737,
+    "train_steps_per_second": 0.406
 }

trainer_state.json

@@ -10,33 +10,33 @@
   "log_history": [
     {
       "epoch": 1.25,
-      "grad_norm": 0.7662187218666077,
+      "grad_norm": 0.2023887187242508,
       "learning_rate": 0.0001375,
-      "loss": 0.5713,
+      "loss": 0.0718,
       "step": 10
     },
     {
       "epoch": 2.5,
-      "grad_norm": 0.32940855622291565,
+      "grad_norm": 0.1942940056324005,
       "learning_rate": 7.500000000000001e-05,
-      "loss": 0.1034,
+      "loss": 0.0592,
       "step": 20
     },
     {
       "epoch": 3.75,
-      "grad_norm": 0.25528982281684875,
+      "grad_norm": 0.1428154855966568,
       "learning_rate": 1.25e-05,
-      "loss": 0.055,
+      "loss": 0.0313,
       "step": 30
     },
     {
       "epoch": 4.0,
       "step": 32,
       "total_flos": 3.502669323615437e+16,
-      "train_loss": 0.23122912645339966,
-      "train_runtime": 88.6604,
-      "train_samples_per_second": 5.098,
-      "train_steps_per_second": 0.361
+      "train_loss": 0.05574157414957881,
+      "train_runtime": 78.7822,
+      "train_samples_per_second": 5.737,
+      "train_steps_per_second": 0.406
     }
   ],
   "logging_steps": 10,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:87ac2670551e95d5bde07e9c403f7cbba7c9ffcf81dab783ba52c1d3cca31d6d
+oid sha256:130b5a565fc363951f57a65b4544d73a3de88d8f52c7bff42ffc52f54f6af3d1
 size 4920