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

all_results.json

@@ -1,12 +1,12 @@
 {
     "epoch": 4.0,
-    "eval_accuracy": 0.5,
-    "eval_loss": 3.424851179122925,
-    "eval_runtime": 126.9168,
-    "eval_samples_per_second": 3.94,
-    "eval_steps_per_second": 0.496,
-    "train_loss": 0.5848965970799327,
-    "train_runtime": 1816.5789,
-    "train_samples_per_second": 1.101,
-    "train_steps_per_second": 0.07
+    "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_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 4.0,
-    "eval_accuracy": 0.5,
-    "eval_loss": 3.424851179122925,
-    "eval_runtime": 126.9168,
-    "eval_samples_per_second": 3.94,
-    "eval_steps_per_second": 0.496
+    "eval_accuracy": 1.0,
+    "eval_loss": 1.3101392984390259,
+    "eval_runtime": 26.5465,
+    "eval_samples_per_second": 3.767,
+    "eval_steps_per_second": 0.49
 }

runs/Dec06_19-06-57_LAPTOP-PH299HA6/events.out.tfevents.1701864930.LAPTOP-PH299HA6.2064.1

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

train_results.json

@@ -1,7 +1,7 @@
 {
     "epoch": 4.0,
-    "train_loss": 0.5848965970799327,
-    "train_runtime": 1816.5789,
-    "train_samples_per_second": 1.101,
-    "train_steps_per_second": 0.07
+    "train_loss": 1.9514953068324499,
+    "train_runtime": 286.5779,
+    "train_samples_per_second": 1.396,
+    "train_steps_per_second": 0.098
 }

trainer_state.json

@@ -1,109 +1,40 @@
 {
-  "best_metric": 3.424851179122925,
-  "best_model_checkpoint": "food-recognition\\checkpoint-100",
+  "best_metric": null,
+  "best_model_checkpoint": null,
   "epoch": 4.0,
   "eval_steps": 100,
-  "global_step": 128,
+  "global_step": 28,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
-      "epoch": 0.31,
-      "learning_rate": 0.000184375,
-      "loss": 2.7978,
+      "epoch": 1.43,
+      "learning_rate": 0.00012857142857142858,
+      "loss": 2.85,
       "step": 10
     },
     {
-      "epoch": 0.62,
-      "learning_rate": 0.00016875,
-      "loss": 1.3702,
+      "epoch": 2.86,
+      "learning_rate": 5.714285714285714e-05,
+      "loss": 1.5431,
       "step": 20
     },
-    {
-      "epoch": 0.94,
-      "learning_rate": 0.000153125,
-      "loss": 0.9744,
-      "step": 30
-    },
-    {
-      "epoch": 1.25,
-      "learning_rate": 0.0001375,
-      "loss": 0.6878,
-      "step": 40
-    },
-    {
-      "epoch": 1.56,
-      "learning_rate": 0.00012187500000000001,
-      "loss": 0.4768,
-      "step": 50
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 0.00010625000000000001,
-      "loss": 0.3292,
-      "step": 60
-    },
-    {
-      "epoch": 2.19,
-      "learning_rate": 9.062500000000001e-05,
-      "loss": 0.2301,
-      "step": 70
-    },
-    {
-      "epoch": 2.5,
-      "learning_rate": 7.500000000000001e-05,
-      "loss": 0.1653,
-      "step": 80
-    },
-    {
-      "epoch": 2.81,
-      "learning_rate": 5.9375e-05,
-      "loss": 0.1245,
-      "step": 90
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 4.375e-05,
-      "loss": 0.1007,
-      "step": 100
-    },
-    {
-      "epoch": 3.12,
-      "eval_accuracy": 0.5,
-      "eval_loss": 3.424851179122925,
-      "eval_runtime": 124.3023,
-      "eval_samples_per_second": 4.022,
-      "eval_steps_per_second": 0.507,
-      "step": 100
-    },
-    {
-      "epoch": 3.44,
-      "learning_rate": 2.8125000000000003e-05,
-      "loss": 0.0875,
-      "step": 110
-    },
-    {
-      "epoch": 3.75,
-      "learning_rate": 1.25e-05,
-      "loss": 0.0804,
-      "step": 120
-    },
     {
       "epoch": 4.0,
-      "step": 128,
-      "total_flos": 1.551214973952e+17,
-      "train_loss": 0.5848965970799327,
-      "train_runtime": 1816.5789,
-      "train_samples_per_second": 1.101,
-      "train_steps_per_second": 0.07
+      "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
     }
   ],
   "logging_steps": 10,
-  "max_steps": 128,
+  "max_steps": 28,
   "num_train_epochs": 4,
   "save_steps": 100,
-  "total_flos": 1.551214973952e+17,
+  "total_flos": 3.102429947904e+16,
   "trial_name": null,
   "trial_params": null
 }