End of training

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 base_model: google/vit-base-patch16-224
 tags:
+- image-classification
+- vision
 - generated_from_trainer
 datasets:
 - imagefolder

all_results.json

@@ -0,0 +1,12 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.996,
+    "eval_loss": 0.016295962035655975,
+    "eval_runtime": 4.5017,
+    "eval_samples_per_second": 111.069,
+    "eval_steps_per_second": 1.777,
+    "train_loss": 0.07546138893812895,
+    "train_runtime": 293.9975,
+    "train_samples_per_second": 34.014,
+    "train_steps_per_second": 0.544
+}

eval_results.json

@@ -0,0 +1,8 @@
+{
+    "epoch": 5.0,
+    "eval_accuracy": 0.996,
+    "eval_loss": 0.016295962035655975,
+    "eval_runtime": 4.5017,
+    "eval_samples_per_second": 111.069,
+    "eval_steps_per_second": 1.777
+}

train_results.json

@@ -0,0 +1,7 @@
+{
+    "epoch": 5.0,
+    "train_loss": 0.07546138893812895,
+    "train_runtime": 293.9975,
+    "train_samples_per_second": 34.014,
+    "train_steps_per_second": 0.544
+}

trainer_state.json

@@ -0,0 +1,187 @@
+{
+  "best_metric": 0.016295962035655975,
+  "best_model_checkpoint": "./outputs/checkpoint-160",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 160,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.31,
+      "grad_norm": 3.219058036804199,
+      "learning_rate": 1.8750000000000002e-05,
+      "loss": 0.4776,
+      "step": 10
+    },
+    {
+      "epoch": 0.62,
+      "grad_norm": 1.2438404560089111,
+      "learning_rate": 1.7500000000000002e-05,
+      "loss": 0.1885,
+      "step": 20
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 0.8410812616348267,
+      "learning_rate": 1.6250000000000002e-05,
+      "loss": 0.0948,
+      "step": 30
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.994,
+      "eval_loss": 0.038248706609010696,
+      "eval_runtime": 27.2538,
+      "eval_samples_per_second": 18.346,
+      "eval_steps_per_second": 0.294,
+      "step": 32
+    },
+    {
+      "epoch": 1.25,
+      "grad_norm": 0.9565186500549316,
+      "learning_rate": 1.5000000000000002e-05,
+      "loss": 0.0511,
+      "step": 40
+    },
+    {
+      "epoch": 1.56,
+      "grad_norm": 1.1085598468780518,
+      "learning_rate": 1.375e-05,
+      "loss": 0.045,
+      "step": 50
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 1.4812817573547363,
+      "learning_rate": 1.25e-05,
+      "loss": 0.045,
+      "step": 60
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.996,
+      "eval_loss": 0.020908161997795105,
+      "eval_runtime": 4.4414,
+      "eval_samples_per_second": 112.577,
+      "eval_steps_per_second": 1.801,
+      "step": 64
+    },
+    {
+      "epoch": 2.19,
+      "grad_norm": 2.213463068008423,
+      "learning_rate": 1.125e-05,
+      "loss": 0.0408,
+      "step": 70
+    },
+    {
+      "epoch": 2.5,
+      "grad_norm": 1.0551693439483643,
+      "learning_rate": 1e-05,
+      "loss": 0.0311,
+      "step": 80
+    },
+    {
+      "epoch": 2.81,
+      "grad_norm": 1.5963162183761597,
+      "learning_rate": 8.750000000000001e-06,
+      "loss": 0.0421,
+      "step": 90
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.996,
+      "eval_loss": 0.017463386058807373,
+      "eval_runtime": 4.5818,
+      "eval_samples_per_second": 109.128,
+      "eval_steps_per_second": 1.746,
+      "step": 96
+    },
+    {
+      "epoch": 3.12,
+      "grad_norm": 0.7329137921333313,
+      "learning_rate": 7.500000000000001e-06,
+      "loss": 0.0219,
+      "step": 100
+    },
+    {
+      "epoch": 3.44,
+      "grad_norm": 1.8856861591339111,
+      "learning_rate": 6.25e-06,
+      "loss": 0.0257,
+      "step": 110
+    },
+    {
+      "epoch": 3.75,
+      "grad_norm": 0.23129290342330933,
+      "learning_rate": 5e-06,
+      "loss": 0.0223,
+      "step": 120
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.996,
+      "eval_loss": 0.016901519149541855,
+      "eval_runtime": 4.3629,
+      "eval_samples_per_second": 114.603,
+      "eval_steps_per_second": 1.834,
+      "step": 128
+    },
+    {
+      "epoch": 4.06,
+      "grad_norm": 1.9281848669052124,
+      "learning_rate": 3.7500000000000005e-06,
+      "loss": 0.0396,
+      "step": 130
+    },
+    {
+      "epoch": 4.38,
+      "grad_norm": 0.3660925030708313,
+      "learning_rate": 2.5e-06,
+      "loss": 0.0267,
+      "step": 140
+    },
+    {
+      "epoch": 4.69,
+      "grad_norm": 0.5336365699768066,
+      "learning_rate": 1.25e-06,
+      "loss": 0.0304,
+      "step": 150
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 3.7920114994049072,
+      "learning_rate": 0.0,
+      "loss": 0.025,
+      "step": 160
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.996,
+      "eval_loss": 0.016295962035655975,
+      "eval_runtime": 4.1331,
+      "eval_samples_per_second": 120.976,
+      "eval_steps_per_second": 1.936,
+      "step": 160
+    },
+    {
+      "epoch": 5.0,
+      "step": 160,
+      "total_flos": 7.7491989614592e+17,
+      "train_loss": 0.07546138893812895,
+      "train_runtime": 293.9975,
+      "train_samples_per_second": 34.014,
+      "train_steps_per_second": 0.544
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 160,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 7.7491989614592e+17,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}