End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224-in21k
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 metrics:
 - f1
@@ -17,7 +19,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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0446
+- Loss: 0.0422
 - F1: 0.9705
 
 ## Model description

all_results.json

@@ -1,13 +1,13 @@
 {
     "epoch": 5.0,
     "eval_f1": 0.9705180789481339,
-    "eval_loss": 0.04290741682052612,
-    "eval_runtime": 939.7613,
-    "eval_samples_per_second": 219.483,
-    "eval_steps_per_second": 3.43,
-    "total_flos": 9.82001462664467e+19,
-    "train_loss": 0.0,
-    "train_runtime": 0.0662,
-    "train_samples_per_second": 19152190.372,
-    "train_steps_per_second": 299304.924
+    "eval_loss": 0.04224640876054764,
+    "eval_runtime": 901.493,
+    "eval_samples_per_second": 228.8,
+    "eval_steps_per_second": 3.575,
+    "total_flos": 9.820471825285631e+19,
+    "train_loss": 0.06684475449172272,
+    "train_runtime": 18247.0343,
+    "train_samples_per_second": 69.452,
+    "train_steps_per_second": 1.085
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 5.0,
     "eval_f1": 0.9705180789481339,
-    "eval_loss": 0.04290741682052612,
-    "eval_runtime": 939.7613,
-    "eval_samples_per_second": 219.483,
-    "eval_steps_per_second": 3.43
+    "eval_loss": 0.04224640876054764,
+    "eval_runtime": 901.493,
+    "eval_samples_per_second": 228.8,
+    "eval_steps_per_second": 3.575
 }

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 5.0,
-    "total_flos": 9.82001462664467e+19,
-    "train_loss": 0.0,
-    "train_runtime": 0.0662,
-    "train_samples_per_second": 19152190.372,
-    "train_steps_per_second": 299304.924
+    "total_flos": 9.820471825285631e+19,
+    "train_loss": 0.06684475449172272,
+    "train_runtime": 18247.0343,
+    "train_samples_per_second": 69.452,
+    "train_steps_per_second": 1.085
 }