🍻 cheers

README.md

@@ -2,7 +2,10 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
 - generated_from_trainer
+metrics:
+- accuracy
 model-index:
 - name: food-recognition
   results: []
@@ -13,7 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # food-recognition
 
-This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown 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 food-recognition dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.6212
+- Accuracy: 1.0
 
 ## Model description
 

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 1.1791681051254272,
-    "eval_runtime": 27.1307,
-    "eval_samples_per_second": 3.686,
-    "eval_steps_per_second": 0.479,
-    "train_loss": 1.8178349222455705,
-    "train_runtime": 294.5114,
-    "train_samples_per_second": 1.358,
-    "train_steps_per_second": 0.095
+    "eval_loss": 0.6212310194969177,
+    "eval_runtime": 26.6398,
+    "eval_samples_per_second": 3.754,
+    "eval_steps_per_second": 0.488,
+    "train_loss": 0.8097262382507324,
+    "train_runtime": 301.5804,
+    "train_samples_per_second": 1.326,
+    "train_steps_per_second": 0.093
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 1.1791681051254272,
-    "eval_runtime": 27.1307,
-    "eval_samples_per_second": 3.686,
-    "eval_steps_per_second": 0.479
+    "eval_loss": 0.6212310194969177,
+    "eval_runtime": 26.6398,
+    "eval_samples_per_second": 3.754,
+    "eval_steps_per_second": 0.488
 }

runs/Dec06_21-27-18_LAPTOP-PH299HA6/events.out.tfevents.1701877053.LAPTOP-PH299HA6.21920.2

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

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 4.0,
-    "train_loss": 1.8178349222455705,
-    "train_runtime": 294.5114,
-    "train_samples_per_second": 1.358,
-    "train_steps_per_second": 0.095
+    "train_loss": 0.8097262382507324,
+    "train_runtime": 301.5804,
+    "train_samples_per_second": 1.326,
+    "train_steps_per_second": 0.093
 }

trainer_state.json

@@ -11,23 +11,23 @@
     {
       "epoch": 1.43,
       "learning_rate": 0.00012857142857142858,
-      "loss": 2.7199,
+      "loss": 1.0099,
       "step": 10
     },
     {
       "epoch": 2.86,
       "learning_rate": 5.714285714285714e-05,
-      "loss": 1.4042,
+      "loss": 0.7465,
       "step": 20
     },
     {
       "epoch": 4.0,
       "step": 28,
       "total_flos": 3.102429947904e+16,
-      "train_loss": 1.8178349222455705,
-      "train_runtime": 294.5114,
-      "train_samples_per_second": 1.358,
-      "train_steps_per_second": 0.095
+      "train_loss": 0.8097262382507324,
+      "train_runtime": 301.5804,
+      "train_samples_per_second": 1.326,
+      "train_steps_per_second": 0.093
     }
   ],
   "logging_steps": 10,