End of training

all_results.json

@@ -1,13 +1,13 @@
 {
-    "epoch": 3.0,
-    "eval_accuracy": 0.7696709585121603,
-    "eval_loss": 0.5915560722351074,
-    "eval_runtime": 7.0343,
-    "eval_samples_per_second": 99.37,
-    "eval_steps_per_second": 3.128,
-    "total_flos": 2.085095129220956e+17,
-    "train_loss": 0.8958317330389312,
-    "train_runtime": 194.9461,
-    "train_samples_per_second": 43.027,
-    "train_steps_per_second": 0.339
+    "epoch": 10.0,
+    "eval_accuracy": 0.882689556509299,
+    "eval_loss": 0.337208092212677,
+    "eval_runtime": 7.2237,
+    "eval_samples_per_second": 96.765,
+    "eval_steps_per_second": 3.046,
+    "total_flos": 6.950317097403187e+17,
+    "train_loss": 0.30874538096514614,
+    "train_runtime": 658.9789,
+    "train_samples_per_second": 42.429,
+    "train_steps_per_second": 0.334
 }

eval_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 3.0,
-    "eval_accuracy": 0.7696709585121603,
-    "eval_loss": 0.5915560722351074,
-    "eval_runtime": 7.0343,
-    "eval_samples_per_second": 99.37,
-    "eval_steps_per_second": 3.128
+    "epoch": 10.0,
+    "eval_accuracy": 0.882689556509299,
+    "eval_loss": 0.337208092212677,
+    "eval_runtime": 7.2237,
+    "eval_samples_per_second": 96.765,
+    "eval_steps_per_second": 3.046
 }

runs/Jan03_16-42-55_527f2771f15a/events.out.tfevents.1672764998.527f2771f15a.148.5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b02f7d41284bdebba3ad062cd92051291c0a5cdc77b5e816366a209f2fe52087
+size 363

train_results.json

@@ -1,8 +1,8 @@
 {
-    "epoch": 3.0,
-    "total_flos": 2.085095129220956e+17,
-    "train_loss": 0.8958317330389312,
-    "train_runtime": 194.9461,
-    "train_samples_per_second": 43.027,
-    "train_steps_per_second": 0.339
+    "epoch": 10.0,
+    "total_flos": 6.950317097403187e+17,
+    "train_loss": 0.30874538096514614,
+    "train_runtime": 658.9789,
+    "train_samples_per_second": 42.429,
+    "train_steps_per_second": 0.334
 }

trainer_state.json

@@ -1,88 +1,247 @@
 {
-  "best_metric": 0.7696709585121603,
-  "best_model_checkpoint": "swin-tiny-patch4-window7-224-finetuned-trash_classification/checkpoint-66",
-  "epoch": 3.0,
-  "global_step": 66,
+  "best_metric": 0.882689556509299,
+  "best_model_checkpoint": "swin-tiny-patch4-window7-224-finetuned-trash_classification/checkpoint-198",
+  "epoch": 10.0,
+  "global_step": 220,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
   "log_history": [
     {
       "epoch": 0.45,
-      "learning_rate": 4.745762711864407e-05,
-      "loss": 1.5462,
+      "learning_rate": 2.272727272727273e-05,
+      "loss": 0.531,
       "step": 10
     },
     {
       "epoch": 0.91,
-      "learning_rate": 3.898305084745763e-05,
-      "loss": 1.1855,
+      "learning_rate": 4.545454545454546e-05,
+      "loss": 0.4991,
       "step": 20
     },
     {
       "epoch": 1.0,
-      "eval_accuracy": 0.6809728183118741,
-      "eval_loss": 0.9103820323944092,
-      "eval_runtime": 6.9474,
-      "eval_samples_per_second": 100.613,
-      "eval_steps_per_second": 3.167,
+      "eval_accuracy": 0.7911301859799714,
+      "eval_loss": 0.5482170581817627,
+      "eval_runtime": 7.0774,
+      "eval_samples_per_second": 98.766,
+      "eval_steps_per_second": 3.109,
       "step": 22
     },
     {
       "epoch": 1.36,
-      "learning_rate": 3.050847457627119e-05,
-      "loss": 0.8813,
+      "learning_rate": 4.797979797979798e-05,
+      "loss": 0.4159,
       "step": 30
     },
     {
       "epoch": 1.82,
-      "learning_rate": 2.2033898305084748e-05,
-      "loss": 0.7209,
+      "learning_rate": 4.545454545454546e-05,
+      "loss": 0.4008,
       "step": 40
     },
     {
       "epoch": 2.0,
-      "eval_accuracy": 0.7510729613733905,
-      "eval_loss": 0.6361371278762817,
-      "eval_runtime": 7.3972,
-      "eval_samples_per_second": 94.496,
-      "eval_steps_per_second": 2.974,
+      "eval_accuracy": 0.7954220314735336,
+      "eval_loss": 0.5192908644676208,
+      "eval_runtime": 7.6454,
+      "eval_samples_per_second": 91.427,
+      "eval_steps_per_second": 2.878,
       "step": 44
     },
     {
       "epoch": 2.27,
-      "learning_rate": 1.3559322033898305e-05,
-      "loss": 0.6267,
+      "learning_rate": 4.292929292929293e-05,
+      "loss": 0.3623,
       "step": 50
     },
     {
       "epoch": 2.73,
-      "learning_rate": 5.084745762711865e-06,
-      "loss": 0.6043,
+      "learning_rate": 4.0404040404040405e-05,
+      "loss": 0.3659,
       "step": 60
     },
     {
       "epoch": 3.0,
-      "eval_accuracy": 0.7696709585121603,
-      "eval_loss": 0.5915560722351074,
-      "eval_runtime": 6.9079,
-      "eval_samples_per_second": 101.188,
-      "eval_steps_per_second": 3.185,
+      "eval_accuracy": 0.8397711015736766,
+      "eval_loss": 0.44643908739089966,
+      "eval_runtime": 7.1094,
+      "eval_samples_per_second": 98.321,
+      "eval_steps_per_second": 3.095,
       "step": 66
     },
     {
-      "epoch": 3.0,
-      "step": 66,
-      "total_flos": 2.085095129220956e+17,
-      "train_loss": 0.8958317330389312,
-      "train_runtime": 194.9461,
-      "train_samples_per_second": 43.027,
-      "train_steps_per_second": 0.339
+      "epoch": 3.18,
+      "learning_rate": 3.787878787878788e-05,
+      "loss": 0.3551,
+      "step": 70
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 3.535353535353535e-05,
+      "loss": 0.372,
+      "step": 80
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.8397711015736766,
+      "eval_loss": 0.4384276270866394,
+      "eval_runtime": 7.6735,
+      "eval_samples_per_second": 91.093,
+      "eval_steps_per_second": 2.867,
+      "step": 88
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 3.282828282828283e-05,
+      "loss": 0.3277,
+      "step": 90
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 3.0303030303030306e-05,
+      "loss": 0.3252,
+      "step": 100
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 2.777777777777778e-05,
+      "loss": 0.3388,
+      "step": 110
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.8454935622317596,
+      "eval_loss": 0.42808040976524353,
+      "eval_runtime": 7.1513,
+      "eval_samples_per_second": 97.745,
+      "eval_steps_per_second": 3.076,
+      "step": 110
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 2.5252525252525256e-05,
+      "loss": 0.2767,
+      "step": 120
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 2.272727272727273e-05,
+      "loss": 0.2654,
+      "step": 130
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy": 0.871244635193133,
+      "eval_loss": 0.3618273138999939,
+      "eval_runtime": 7.6567,
+      "eval_samples_per_second": 91.293,
+      "eval_steps_per_second": 2.873,
+      "step": 132
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 2.0202020202020203e-05,
+      "loss": 0.235,
+      "step": 140
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 1.7676767676767676e-05,
+      "loss": 0.2326,
+      "step": 150
+    },
+    {
+      "epoch": 7.0,
+      "eval_accuracy": 0.8755364806866953,
+      "eval_loss": 0.3550046682357788,
+      "eval_runtime": 7.1895,
+      "eval_samples_per_second": 97.226,
+      "eval_steps_per_second": 3.06,
+      "step": 154
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 1.5151515151515153e-05,
+      "loss": 0.219,
+      "step": 160
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 1.2626262626262628e-05,
+      "loss": 0.2354,
+      "step": 170
+    },
+    {
+      "epoch": 8.0,
+      "eval_accuracy": 0.8798283261802575,
+      "eval_loss": 0.3400857448577881,
+      "eval_runtime": 7.5656,
+      "eval_samples_per_second": 92.392,
+      "eval_steps_per_second": 2.908,
+      "step": 176
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 1.0101010101010101e-05,
+      "loss": 0.2455,
+      "step": 180
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 7.5757575757575764e-06,
+      "loss": 0.1774,
+      "step": 190
+    },
+    {
+      "epoch": 9.0,
+      "eval_accuracy": 0.882689556509299,
+      "eval_loss": 0.337208092212677,
+      "eval_runtime": 7.0903,
+      "eval_samples_per_second": 98.586,
+      "eval_steps_per_second": 3.103,
+      "step": 198
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 5.050505050505051e-06,
+      "loss": 0.2007,
+      "step": 200
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 2.5252525252525253e-06,
+      "loss": 0.226,
+      "step": 210
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 0.0,
+      "loss": 0.1849,
+      "step": 220
+    },
+    {
+      "epoch": 10.0,
+      "eval_accuracy": 0.882689556509299,
+      "eval_loss": 0.33802154660224915,
+      "eval_runtime": 7.6016,
+      "eval_samples_per_second": 91.954,
+      "eval_steps_per_second": 2.894,
+      "step": 220
+    },
+    {
+      "epoch": 10.0,
+      "step": 220,
+      "total_flos": 6.950317097403187e+17,
+      "train_loss": 0.30874538096514614,
+      "train_runtime": 658.9789,
+      "train_samples_per_second": 42.429,
+      "train_steps_per_second": 0.334
     }
   ],
-  "max_steps": 66,
-  "num_train_epochs": 3,
-  "total_flos": 2.085095129220956e+17,
+  "max_steps": 220,
+  "num_train_epochs": 10,
+  "total_flos": 6.950317097403187e+17,
   "trial_name": null,
   "trial_params": null
 }