End of training

all_results.json

@@ -1,18 +1,28 @@
 {
     "epoch": 10.0,
-    "eval_gen_len": 52.897333333333336,
-    "eval_loss": 2.143275499343872,
-    "eval_rouge1": 27.2333,
-    "eval_rouge2": 8.451,
-    "eval_rougeL": 18.8946,
-    "eval_rougeLsum": 23.575,
-    "eval_runtime": 250.9824,
+    "eval_gen_len": 52.72533333333333,
+    "eval_loss": 2.102524995803833,
+    "eval_rouge1": 27.5357,
+    "eval_rouge2": 8.5623,
+    "eval_rougeL": 19.1508,
+    "eval_rougeLsum": 23.9029,
+    "eval_runtime": 251.4747,
     "eval_samples": 1500,
-    "eval_samples_per_second": 5.977,
-    "eval_steps_per_second": 0.749,
-    "train_loss": 2.0453601527622767,
-    "train_runtime": 6220.6473,
+    "eval_samples_per_second": 5.965,
+    "eval_steps_per_second": 0.748,
+    "predict_gen_len": 52.516,
+    "predict_loss": 2.131106376647949,
+    "predict_rouge1": 26.7775,
+    "predict_rouge2": 8.2523,
+    "predict_rougeL": 18.6904,
+    "predict_rougeLsum": 23.2921,
+    "predict_runtime": 251.4989,
+    "predict_samples": 1500,
+    "predict_samples_per_second": 5.964,
+    "predict_steps_per_second": 0.748,
+    "train_loss": 2.0145415248325893,
+    "train_runtime": 6283.472,
     "train_samples": 7000,
-    "train_samples_per_second": 11.253,
-    "train_steps_per_second": 1.407
+    "train_samples_per_second": 11.14,
+    "train_steps_per_second": 1.393
 }

eval_results.json

@@ -1,13 +1,13 @@
 {
     "epoch": 10.0,
-    "eval_gen_len": 52.897333333333336,
-    "eval_loss": 2.143275499343872,
-    "eval_rouge1": 27.2333,
-    "eval_rouge2": 8.451,
-    "eval_rougeL": 18.8946,
-    "eval_rougeLsum": 23.575,
-    "eval_runtime": 250.9824,
+    "eval_gen_len": 52.72533333333333,
+    "eval_loss": 2.102524995803833,
+    "eval_rouge1": 27.5357,
+    "eval_rouge2": 8.5623,
+    "eval_rougeL": 19.1508,
+    "eval_rougeLsum": 23.9029,
+    "eval_runtime": 251.4747,
     "eval_samples": 1500,
-    "eval_samples_per_second": 5.977,
-    "eval_steps_per_second": 0.749
+    "eval_samples_per_second": 5.965,
+    "eval_steps_per_second": 0.748
 }

predict_results.json

@@ -0,0 +1,12 @@
+{
+    "predict_gen_len": 52.516,
+    "predict_loss": 2.131106376647949,
+    "predict_rouge1": 26.7775,
+    "predict_rouge2": 8.2523,
+    "predict_rougeL": 18.6904,
+    "predict_rougeLsum": 23.2921,
+    "predict_runtime": 251.4989,
+    "predict_samples": 1500,
+    "predict_samples_per_second": 5.964,
+    "predict_steps_per_second": 0.748
+}

train_results.json

@@ -1,8 +1,8 @@
 {
     "epoch": 10.0,
-    "train_loss": 2.0453601527622767,
-    "train_runtime": 6220.6473,
+    "train_loss": 2.0145415248325893,
+    "train_runtime": 6283.472,
     "train_samples": 7000,
-    "train_samples_per_second": 11.253,
-    "train_steps_per_second": 1.407
+    "train_samples_per_second": 11.14,
+    "train_steps_per_second": 1.393
 }

trainer_state.json

@@ -10,118 +10,118 @@
     {
       "epoch": 0.57,
       "learning_rate": 4.714285714285714e-05,
-      "loss": 2.3409,
+      "loss": 2.2874,
       "step": 500
     },
     {
       "epoch": 1.14,
       "learning_rate": 4.428571428571428e-05,
-      "loss": 2.3049,
+      "loss": 2.2768,
       "step": 1000
     },
     {
       "epoch": 1.71,
       "learning_rate": 4.1428571428571437e-05,
-      "loss": 2.2384,
+      "loss": 2.1969,
       "step": 1500
     },
     {
       "epoch": 2.29,
       "learning_rate": 3.857142857142858e-05,
-      "loss": 2.1886,
+      "loss": 2.1459,
       "step": 2000
     },
     {
       "epoch": 2.86,
       "learning_rate": 3.571428571428572e-05,
-      "loss": 2.1368,
+      "loss": 2.1127,
       "step": 2500
     },
     {
       "epoch": 3.43,
       "learning_rate": 3.285714285714286e-05,
-      "loss": 2.1029,
+      "loss": 2.0616,
       "step": 3000
     },
     {
       "epoch": 4.0,
       "learning_rate": 3e-05,
-      "loss": 2.0797,
+      "loss": 2.0542,
       "step": 3500
     },
     {
       "epoch": 4.57,
       "learning_rate": 2.714285714285714e-05,
-      "loss": 2.0263,
+      "loss": 1.9924,
       "step": 4000
     },
     {
       "epoch": 5.14,
       "learning_rate": 2.4285714285714288e-05,
-      "loss": 2.0152,
+      "loss": 1.9893,
       "step": 4500
     },
     {
       "epoch": 5.71,
       "learning_rate": 2.1428571428571428e-05,
-      "loss": 1.9843,
+      "loss": 1.9576,
       "step": 5000
     },
     {
       "epoch": 6.29,
       "learning_rate": 1.8571428571428572e-05,
-      "loss": 1.9578,
+      "loss": 1.9438,
       "step": 5500
     },
     {
       "epoch": 6.86,
       "learning_rate": 1.5714285714285715e-05,
-      "loss": 1.9462,
+      "loss": 1.9185,
       "step": 6000
     },
     {
       "epoch": 7.43,
       "learning_rate": 1.2857142857142857e-05,
-      "loss": 1.9366,
+      "loss": 1.8899,
       "step": 6500
     },
     {
       "epoch": 8.0,
       "learning_rate": 1e-05,
-      "loss": 1.9146,
+      "loss": 1.8952,
       "step": 7000
     },
     {
       "epoch": 8.57,
       "learning_rate": 7.142857142857143e-06,
-      "loss": 1.8964,
+      "loss": 1.8678,
       "step": 7500
     },
     {
       "epoch": 9.14,
       "learning_rate": 4.285714285714286e-06,
-      "loss": 1.8958,
+      "loss": 1.881,
       "step": 8000
     },
     {
       "epoch": 9.71,
       "learning_rate": 1.4285714285714286e-06,
-      "loss": 1.8854,
+      "loss": 1.8538,
       "step": 8500
     },
     {
       "epoch": 10.0,
       "step": 8750,
-      "total_flos": 4.261889801244672e+16,
-      "train_loss": 2.0453601527622767,
-      "train_runtime": 6220.6473,
-      "train_samples_per_second": 11.253,
-      "train_steps_per_second": 1.407
+      "total_flos": 4.261976387444736e+16,
+      "train_loss": 2.0145415248325893,
+      "train_runtime": 6283.472,
+      "train_samples_per_second": 11.14,
+      "train_steps_per_second": 1.393
     }
   ],
   "max_steps": 8750,
   "num_train_epochs": 10,
-  "total_flos": 4.261889801244672e+16,
+  "total_flos": 4.261976387444736e+16,
   "trial_name": null,
   "trial_params": null
 }