🍻 cheers

README.md

@@ -23,9 +23,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 1
-pipeline_tag: image-classification
-inference: false
+      value: 1.0
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -35,7 +33,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 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.
 It achieves the following results on the evaluation set:
-- Loss: 0.0594
+- Loss: 0.0524
 - Accuracy: 1.0
 
 ## Model description
@@ -72,4 +70,4 @@ The following hyperparameters were used during training:
 - Transformers 4.38.2
 - Pytorch 2.2.1
 - Datasets 2.18.0
-- Tokenizers 0.15.2
+- Tokenizers 0.15.2

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 0.05941523611545563,
-    "eval_runtime": 2.2052,
-    "eval_samples_per_second": 12.244,
-    "eval_steps_per_second": 1.814,
-    "train_loss": 0.25243260501883924,
-    "train_runtime": 85.4792,
-    "train_samples_per_second": 5.288,
-    "train_steps_per_second": 0.374
+    "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_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 0.05941523611545563,
-    "eval_runtime": 2.2052,
-    "eval_samples_per_second": 12.244,
-    "eval_steps_per_second": 1.814
+    "eval_loss": 0.052425507456064224,
+    "eval_runtime": 2.2674,
+    "eval_samples_per_second": 11.908,
+    "eval_steps_per_second": 1.764
 }

model.safetensors

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

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 4.0,
-    "train_loss": 0.25243260501883924,
-    "train_runtime": 85.4792,
-    "train_samples_per_second": 5.288,
-    "train_steps_per_second": 0.374
+    "train_loss": 0.23122912645339966,
+    "train_runtime": 88.6604,
+    "train_samples_per_second": 5.098,
+    "train_steps_per_second": 0.361
 }

trainer_state.json

@@ -10,33 +10,33 @@
   "log_history": [
     {
       "epoch": 1.25,
-      "grad_norm": 0.9073283672332764,
+      "grad_norm": 0.7662187218666077,
       "learning_rate": 0.0001375,
-      "loss": 0.6222,
+      "loss": 0.5713,
       "step": 10
     },
     {
       "epoch": 2.5,
-      "grad_norm": 0.34812086820602417,
+      "grad_norm": 0.32940855622291565,
       "learning_rate": 7.500000000000001e-05,
-      "loss": 0.1155,
+      "loss": 0.1034,
       "step": 20
     },
     {
       "epoch": 3.75,
-      "grad_norm": 0.27614542841911316,
+      "grad_norm": 0.25528982281684875,
       "learning_rate": 1.25e-05,
-      "loss": 0.0591,
+      "loss": 0.055,
       "step": 30
     },
     {
       "epoch": 4.0,
       "step": 32,
       "total_flos": 3.502669323615437e+16,
-      "train_loss": 0.25243260501883924,
-      "train_runtime": 85.4792,
-      "train_samples_per_second": 5.288,
-      "train_steps_per_second": 0.374
+      "train_loss": 0.23122912645339966,
+      "train_runtime": 88.6604,
+      "train_samples_per_second": 5.098,
+      "train_steps_per_second": 0.361
     }
   ],
   "logging_steps": 10,

training_args.bin

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