🍻 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: 1.2503
+- Accuracy: 1.0
 
 ## Model description
 

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 1.3101392984390259,
-    "eval_runtime": 26.5465,
-    "eval_samples_per_second": 3.767,
-    "eval_steps_per_second": 0.49,
-    "train_loss": 1.9514953068324499,
-    "train_runtime": 286.5779,
-    "train_samples_per_second": 1.396,
-    "train_steps_per_second": 0.098
+    "eval_loss": 1.2503035068511963,
+    "eval_runtime": 25.1213,
+    "eval_samples_per_second": 3.981,
+    "eval_steps_per_second": 0.517,
+    "train_loss": 1.8649438108716692,
+    "train_runtime": 290.6251,
+    "train_samples_per_second": 1.376,
+    "train_steps_per_second": 0.096
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
     "eval_accuracy": 1.0,
-    "eval_loss": 1.3101392984390259,
-    "eval_runtime": 26.5465,
-    "eval_samples_per_second": 3.767,
-    "eval_steps_per_second": 0.49
+    "eval_loss": 1.2503035068511963,
+    "eval_runtime": 25.1213,
+    "eval_samples_per_second": 3.981,
+    "eval_steps_per_second": 0.517
 }

runs/Dec06_20-15-17_LAPTOP-PH299HA6/events.out.tfevents.1701869119.LAPTOP-PH299HA6.5764.1

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

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 4.0,
-    "train_loss": 1.9514953068324499,
-    "train_runtime": 286.5779,
-    "train_samples_per_second": 1.396,
-    "train_steps_per_second": 0.098
+    "train_loss": 1.8649438108716692,
+    "train_runtime": 290.6251,
+    "train_samples_per_second": 1.376,
+    "train_steps_per_second": 0.096
 }

trainer_state.json

@@ -11,23 +11,23 @@
     {
       "epoch": 1.43,
       "learning_rate": 0.00012857142857142858,
-      "loss": 2.85,
+      "loss": 2.7406,
       "step": 10
     },
     {
       "epoch": 2.86,
       "learning_rate": 5.714285714285714e-05,
-      "loss": 1.5431,
+      "loss": 1.4589,
       "step": 20
     },
     {
       "epoch": 4.0,
       "step": 28,
       "total_flos": 3.102429947904e+16,
-      "train_loss": 1.9514953068324499,
-      "train_runtime": 286.5779,
-      "train_samples_per_second": 1.396,
-      "train_steps_per_second": 0.098
+      "train_loss": 1.8649438108716692,
+      "train_runtime": 290.6251,
+      "train_samples_per_second": 1.376,
+      "train_steps_per_second": 0.096
     }
   ],
   "logging_steps": 10,