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
 datasets:
 - imagefolder
@@ -22,7 +24,7 @@ model-index:
     metrics:
     - name: Accuracy
       type: accuracy
-      value: 0.9303030303030303
+      value: 0.9318181818181818
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +34,8 @@ 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 imagefolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1497
-- Accuracy: 0.9303
+- Loss: 0.1496
+- Accuracy: 0.9318
 
 ## Model description
 

all_results.json

@@ -0,0 +1,12 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.9318181818181818,
+    "eval_loss": 0.1496073603630066,
+    "eval_runtime": 2.3983,
+    "eval_samples_per_second": 275.193,
+    "eval_steps_per_second": 17.512,
+    "train_loss": 0.21498717446612498,
+    "train_runtime": 220.3587,
+    "train_samples_per_second": 84.862,
+    "train_steps_per_second": 5.31
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.9318181818181818,
+    "eval_loss": 0.1496073603630066,
+    "eval_runtime": 2.3983,
+    "eval_samples_per_second": 275.193,
+    "eval_steps_per_second": 17.512
+}

train_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 5.0,
+    "train_loss": 0.21498717446612498,
+    "train_runtime": 220.3587,
+    "train_samples_per_second": 84.862,
+    "train_steps_per_second": 5.31
+}

trainer_state.json

@@ -0,0 +1,85 @@
+{
+  "best_metric": 0.1496073603630066,
+  "best_model_checkpoint": "./outputs/50_50/checkpoint-702",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 1170,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9303030303030303,
+      "eval_loss": 0.17232811450958252,
+      "eval_runtime": 2.4494,
+      "eval_samples_per_second": 269.458,
+      "eval_steps_per_second": 17.147,
+      "step": 234
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9242424242424242,
+      "eval_loss": 0.21031387150287628,
+      "eval_runtime": 2.5804,
+      "eval_samples_per_second": 255.774,
+      "eval_steps_per_second": 16.277,
+      "step": 468
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 1.1452991452991454e-05,
+      "loss": 0.2573,
+      "step": 500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9318181818181818,
+      "eval_loss": 0.1496073603630066,
+      "eval_runtime": 2.4748,
+      "eval_samples_per_second": 266.687,
+      "eval_steps_per_second": 16.971,
+      "step": 702
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9363636363636364,
+      "eval_loss": 0.15435752272605896,
+      "eval_runtime": 2.405,
+      "eval_samples_per_second": 274.432,
+      "eval_steps_per_second": 17.464,
+      "step": 936
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 2.9059829059829063e-06,
+      "loss": 0.1879,
+      "step": 1000
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9303030303030303,
+      "eval_loss": 0.14966832101345062,
+      "eval_runtime": 2.4823,
+      "eval_samples_per_second": 265.887,
+      "eval_steps_per_second": 16.92,
+      "step": 1170
+    },
+    {
+      "epoch": 5.0,
+      "step": 1170,
+      "total_flos": 1.4491002057928704e+18,
+      "train_loss": 0.21498717446612498,
+      "train_runtime": 220.3587,
+      "train_samples_per_second": 84.862,
+      "train_steps_per_second": 5.31
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 1170,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 1.4491002057928704e+18,
+  "trial_name": null,
+  "trial_params": null
+}