End of training

README.md

@@ -5,6 +5,8 @@ license: apache-2.0
 metrics:
 - accuracy
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 model-index:
 - name: vit-base-beans
@@ -16,7 +18,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-beans
 
-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 beans dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.8347
 - Accuracy: 0.8045

all_results.json

@@ -2,12 +2,12 @@
     "epoch": 5.0,
     "eval_accuracy": 0.8045112781954887,
     "eval_loss": 0.8347006440162659,
-    "eval_runtime": 8.7879,
-    "eval_samples_per_second": 15.134,
-    "eval_steps_per_second": 1.934,
+    "eval_runtime": 4.3276,
+    "eval_samples_per_second": 30.733,
+    "eval_steps_per_second": 3.928,
     "total_flos": 4.0477911662555136e+17,
     "train_loss": 0.9380592258159931,
-    "train_runtime": 305.9476,
-    "train_samples_per_second": 16.898,
-    "train_steps_per_second": 2.125
+    "train_runtime": 422.3657,
+    "train_samples_per_second": 12.241,
+    "train_steps_per_second": 1.539
 }

eval_results.json

@@ -2,7 +2,7 @@
     "epoch": 5.0,
     "eval_accuracy": 0.8045112781954887,
     "eval_loss": 0.8347006440162659,
-    "eval_runtime": 8.7879,
-    "eval_samples_per_second": 15.134,
-    "eval_steps_per_second": 1.934
+    "eval_runtime": 4.3276,
+    "eval_samples_per_second": 30.733,
+    "eval_steps_per_second": 3.928
 }

train_results.json

@@ -2,7 +2,7 @@
     "epoch": 5.0,
     "total_flos": 4.0477911662555136e+17,
     "train_loss": 0.9380592258159931,
-    "train_runtime": 305.9476,
-    "train_samples_per_second": 16.898,
-    "train_steps_per_second": 2.125
+    "train_runtime": 422.3657,
+    "train_samples_per_second": 12.241,
+    "train_steps_per_second": 1.539
 }

trainer_state.json

@@ -103,9 +103,9 @@
       "epoch": 1.0,
       "eval_accuracy": 0.6240601503759399,
       "eval_loss": 1.0291587114334106,
-      "eval_runtime": 2.9581,
-      "eval_samples_per_second": 44.962,
-      "eval_steps_per_second": 5.747,
+      "eval_runtime": 5.0639,
+      "eval_samples_per_second": 26.264,
+      "eval_steps_per_second": 3.357,
       "step": 130
     },
     {
@@ -203,9 +203,9 @@
       "epoch": 2.0,
       "eval_accuracy": 0.7443609022556391,
       "eval_loss": 0.9589495062828064,
-      "eval_runtime": 3.0861,
-      "eval_samples_per_second": 43.096,
-      "eval_steps_per_second": 5.509,
+      "eval_runtime": 4.3534,
+      "eval_samples_per_second": 30.551,
+      "eval_steps_per_second": 3.905,
       "step": 260
     },
     {
@@ -303,9 +303,9 @@
       "epoch": 3.0,
       "eval_accuracy": 0.7894736842105263,
       "eval_loss": 0.8891889452934265,
-      "eval_runtime": 6.4976,
-      "eval_samples_per_second": 20.469,
-      "eval_steps_per_second": 2.616,
+      "eval_runtime": 17.6473,
+      "eval_samples_per_second": 7.537,
+      "eval_steps_per_second": 0.963,
       "step": 390
     },
     {
@@ -403,9 +403,9 @@
       "epoch": 4.0,
       "eval_accuracy": 0.8045112781954887,
       "eval_loss": 0.8480122685432434,
-      "eval_runtime": 6.6166,
-      "eval_samples_per_second": 20.101,
-      "eval_steps_per_second": 2.569,
+      "eval_runtime": 5.8399,
+      "eval_samples_per_second": 22.774,
+      "eval_steps_per_second": 2.911,
       "step": 520
     },
     {
@@ -503,9 +503,9 @@
       "epoch": 5.0,
       "eval_accuracy": 0.8045112781954887,
       "eval_loss": 0.8347006440162659,
-      "eval_runtime": 4.205,
-      "eval_samples_per_second": 31.629,
-      "eval_steps_per_second": 4.043,
+      "eval_runtime": 8.0546,
+      "eval_samples_per_second": 16.512,
+      "eval_steps_per_second": 2.111,
       "step": 650
     },
     {
@@ -513,9 +513,9 @@
       "step": 650,
       "total_flos": 4.0477911662555136e+17,
       "train_loss": 0.9380592258159931,
-      "train_runtime": 305.9476,
-      "train_samples_per_second": 16.898,
-      "train_steps_per_second": 2.125
+      "train_runtime": 422.3657,
+      "train_samples_per_second": 12.241,
+      "train_steps_per_second": 1.539
     }
   ],
   "logging_steps": 10,