End of training

README.md

@@ -17,13 +17,13 @@ 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 food101 dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 3.2252
-- eval_accuracy: 0.6515
-- eval_runtime: 154.6911
-- eval_samples_per_second: 97.937
+- eval_loss: 2.8059
+- eval_accuracy: 0.6871
+- eval_runtime: 154.6826
+- eval_samples_per_second: 97.943
 - eval_steps_per_second: 6.122
-- epoch: 5.28
-- step: 5000
+- epoch: 6.34
+- step: 6000
 
 ## Model description
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:481b50b254fc5d798ca7841a37b28fd52550a3d5a27d8a4fe672d076c3a48678
+oid sha256:034a1f3bc2ce314a3d7d7773432b9af74be9fc41caa59c65b361ffc673496e07
 size 343528508

runs/Feb05_03-08-58_983d148b451b/events.out.tfevents.1707102539.983d148b451b.26.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b5123372ad017b6345bb526b80b72debcaf26727d68ee0bad53e6b242b1f331d
-size 11701
+oid sha256:6f61bbc9f2b675aaecd568e9983f6eb21df3839daec5d48b42aef1e3adceafe4
+size 12181

trainer_state.json

@@ -1,9 +1,9 @@
 {
-  "best_metric": 0.6111551155115511,
-  "best_model_checkpoint": "food-image-classification/checkpoint-4000",
-  "epoch": 5.279831045406547,
+  "best_metric": 0.6514851485148515,
+  "best_model_checkpoint": "food-image-classification/checkpoint-5000",
+  "epoch": 6.335797254487856,
   "eval_steps": 1000,
-  "global_step": 5000,
+  "global_step": 6000,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -82,6 +82,21 @@
       "eval_samples_per_second": 97.937,
       "eval_steps_per_second": 6.122,
       "step": 5000
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 6.335797254487856e-06,
+      "loss": 2.9962,
+      "step": 6000
+    },
+    {
+      "epoch": 6.34,
+      "eval_accuracy": 0.687062706270627,
+      "eval_loss": 2.805878162384033,
+      "eval_runtime": 154.6826,
+      "eval_samples_per_second": 97.943,
+      "eval_steps_per_second": 6.122,
+      "step": 6000
     }
   ],
   "logging_steps": 1000,
@@ -89,7 +104,7 @@
   "num_input_tokens_seen": 0,
   "num_train_epochs": 500,
   "save_steps": 1000,
-  "total_flos": 2.4816337153284096e+19,
+  "total_flos": 2.9779604583940915e+19,
   "train_batch_size": 16,
   "trial_name": null,
   "trial_params": null