End of training

trainer_state.json

@@ -1,8 +1,8 @@
 {
-  "best_metric": 1.710868239402771,
-  "best_model_checkpoint": "detr-resnet50-leuk/checkpoint-4235",
-  "epoch": 35.0,
-  "global_step": 4235,
+  "best_metric": 1.696058988571167,
+  "best_model_checkpoint": "detr-resnet50-leuk/checkpoint-4840",
+  "epoch": 40.0,
+  "global_step": 4840,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -498,18 +498,88 @@
       "step": 4235
     },
     {
-      "epoch": 35.0,
-      "step": 4235,
-      "total_flos": 1.61575431042816e+19,
-      "train_loss": 0.30673924685931964,
-      "train_runtime": 1314.4927,
-      "train_samples_per_second": 25.721,
-      "train_steps_per_second": 3.222
+      "epoch": 36.0,
+      "learning_rate": 1e-05,
+      "loss": 2.1215,
+      "step": 4356
+    },
+    {
+      "epoch": 36.0,
+      "eval_loss": 1.7227232456207275,
+      "eval_runtime": 16.643,
+      "eval_samples_per_second": 6.009,
+      "eval_steps_per_second": 0.781,
+      "step": 4356
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 7.5e-06,
+      "loss": 2.1425,
+      "step": 4477
+    },
+    {
+      "epoch": 37.0,
+      "eval_loss": 1.7108997106552124,
+      "eval_runtime": 16.8607,
+      "eval_samples_per_second": 5.931,
+      "eval_steps_per_second": 0.771,
+      "step": 4477
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 5e-06,
+      "loss": 2.1106,
+      "step": 4598
+    },
+    {
+      "epoch": 38.0,
+      "eval_loss": 1.6993489265441895,
+      "eval_runtime": 16.7213,
+      "eval_samples_per_second": 5.98,
+      "eval_steps_per_second": 0.777,
+      "step": 4598
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 2.5e-06,
+      "loss": 2.0987,
+      "step": 4719
+    },
+    {
+      "epoch": 39.0,
+      "eval_loss": 1.6982134580612183,
+      "eval_runtime": 17.0485,
+      "eval_samples_per_second": 5.866,
+      "eval_steps_per_second": 0.763,
+      "step": 4719
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 0.0,
+      "loss": 2.1259,
+      "step": 4840
+    },
+    {
+      "epoch": 40.0,
+      "eval_loss": 1.696058988571167,
+      "eval_runtime": 17.3421,
+      "eval_samples_per_second": 5.766,
+      "eval_steps_per_second": 0.75,
+      "step": 4840
+    },
+    {
+      "epoch": 40.0,
+      "step": 4840,
+      "total_flos": 1.84657635477504e+19,
+      "train_loss": 0.2649796856336357,
+      "train_runtime": 1315.7115,
+      "train_samples_per_second": 29.368,
+      "train_steps_per_second": 3.679
     }
   ],
-  "max_steps": 4235,
-  "num_train_epochs": 35,
-  "total_flos": 1.61575431042816e+19,
+  "max_steps": 4840,
+  "num_train_epochs": 40,
+  "total_flos": 1.84657635477504e+19,
   "trial_name": null,
   "trial_params": null
 }