{ "best_metric": 1.503156304359436, "best_model_checkpoint": "/exports/eddie/scratch/s1970716/models/summarization/longt5_xl_sfd_bp_20/checkpoint-57", "epoch": 19.47826086956522, "eval_steps": 500, "global_step": 280, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.14, "learning_rate": 0.001, "loss": 2.9672, "step": 2 }, { "epoch": 0.28, "learning_rate": 0.001, "loss": 3.0162, "step": 4 }, { "epoch": 0.42, "learning_rate": 0.001, "loss": 3.1689, "step": 6 }, { "epoch": 0.56, "learning_rate": 0.001, "loss": 2.902, "step": 8 }, { "epoch": 0.7, "learning_rate": 0.001, "loss": 2.4891, "step": 10 }, { "epoch": 0.83, "learning_rate": 0.001, "loss": 2.8498, "step": 12 }, { "epoch": 0.97, "learning_rate": 0.001, "loss": 2.3973, "step": 14 }, { "epoch": 0.97, "eval_gen_len": 511.0, "eval_loss": 1.9073989391326904, "eval_rouge1": 10.6164, "eval_rouge2": 2.4585, "eval_rougeL": 10.4856, "eval_rougeLsum": 9.8193, "eval_runtime": 1794.269, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 14 }, { "epoch": 1.11, "learning_rate": 0.001, "loss": 2.1921, "step": 16 }, { "epoch": 1.25, "learning_rate": 0.001, "loss": 2.0091, "step": 18 }, { "epoch": 1.39, "learning_rate": 0.001, "loss": 1.8884, "step": 20 }, { "epoch": 1.53, "learning_rate": 0.001, "loss": 1.7955, "step": 22 }, { "epoch": 1.67, "learning_rate": 0.001, "loss": 1.7023, "step": 24 }, { "epoch": 1.81, "learning_rate": 0.001, "loss": 1.8178, "step": 26 }, { "epoch": 1.95, "learning_rate": 0.001, "loss": 1.9188, "step": 28 }, { "epoch": 1.95, "eval_gen_len": 511.0, "eval_loss": 1.7082059383392334, "eval_rouge1": 17.4258, "eval_rouge2": 4.2128, "eval_rougeL": 16.5213, "eval_rougeLsum": 15.8377, "eval_runtime": 1795.1498, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 28 }, { "epoch": 2.09, "learning_rate": 0.001, "loss": 1.6461, "step": 30 }, { "epoch": 2.23, "learning_rate": 0.001, "loss": 1.552, "step": 32 }, { "epoch": 2.37, "learning_rate": 0.001, "loss": 1.4914, "step": 34 }, { "epoch": 2.5, "learning_rate": 0.001, "loss": 1.457, "step": 36 }, { "epoch": 2.64, "learning_rate": 0.001, "loss": 1.4499, "step": 38 }, { "epoch": 2.78, "learning_rate": 0.001, "loss": 1.4868, "step": 40 }, { "epoch": 2.92, "learning_rate": 0.001, "loss": 1.4297, "step": 42 }, { "epoch": 2.99, "eval_gen_len": 506.77448071216617, "eval_loss": 1.5072847604751587, "eval_rouge1": 18.6504, "eval_rouge2": 5.4242, "eval_rougeL": 17.2648, "eval_rougeLsum": 17.0203, "eval_runtime": 1796.521, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 43 }, { "epoch": 3.06, "learning_rate": 0.001, "loss": 1.3757, "step": 44 }, { "epoch": 3.2, "learning_rate": 0.001, "loss": 1.2701, "step": 46 }, { "epoch": 3.34, "learning_rate": 0.001, "loss": 1.2826, "step": 48 }, { "epoch": 3.48, "learning_rate": 0.001, "loss": 1.2945, "step": 50 }, { "epoch": 3.62, "learning_rate": 0.001, "loss": 1.2963, "step": 52 }, { "epoch": 3.76, "learning_rate": 0.001, "loss": 1.2933, "step": 54 }, { "epoch": 3.9, "learning_rate": 0.001, "loss": 1.2759, "step": 56 }, { "epoch": 3.97, "eval_gen_len": 497.8783382789318, "eval_loss": 1.503156304359436, "eval_rouge1": 22.11, "eval_rouge2": 7.544, "eval_rougeL": 19.7035, "eval_rougeLsum": 20.2813, "eval_runtime": 1797.0423, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 57 }, { "epoch": 4.03, "learning_rate": 0.001, "loss": 1.2446, "step": 58 }, { "epoch": 4.17, "learning_rate": 0.001, "loss": 1.0992, "step": 60 }, { "epoch": 4.31, "learning_rate": 0.001, "loss": 1.0771, "step": 62 }, { "epoch": 4.45, "learning_rate": 0.001, "loss": 1.1254, "step": 64 }, { "epoch": 4.59, "learning_rate": 0.001, "loss": 1.1644, "step": 66 }, { "epoch": 4.73, "learning_rate": 0.001, "loss": 1.1485, "step": 68 }, { "epoch": 4.87, "learning_rate": 0.001, "loss": 1.1421, "step": 70 }, { "epoch": 4.94, "eval_gen_len": 503.60237388724033, "eval_loss": 1.5462485551834106, "eval_rouge1": 20.6049, "eval_rouge2": 6.7146, "eval_rougeL": 18.5084, "eval_rougeLsum": 19.0876, "eval_runtime": 1793.6239, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 71 }, { "epoch": 5.01, "learning_rate": 0.001, "loss": 1.1233, "step": 72 }, { "epoch": 5.15, "learning_rate": 0.001, "loss": 0.8919, "step": 74 }, { "epoch": 5.29, "learning_rate": 0.001, "loss": 0.9349, "step": 76 }, { "epoch": 5.43, "learning_rate": 0.001, "loss": 0.9363, "step": 78 }, { "epoch": 5.57, "learning_rate": 0.001, "loss": 0.9203, "step": 80 }, { "epoch": 5.7, "learning_rate": 0.001, "loss": 0.9429, "step": 82 }, { "epoch": 5.84, "learning_rate": 0.001, "loss": 0.9495, "step": 84 }, { "epoch": 5.98, "learning_rate": 0.001, "loss": 0.9605, "step": 86 }, { "epoch": 5.98, "eval_gen_len": 510.27299703264094, "eval_loss": 1.6233196258544922, "eval_rouge1": 22.6777, "eval_rouge2": 7.9362, "eval_rougeL": 18.7936, "eval_rougeLsum": 21.41, "eval_runtime": 1793.8682, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 86 }, { "epoch": 6.12, "learning_rate": 0.001, "loss": 0.7747, "step": 88 }, { "epoch": 6.26, "learning_rate": 0.001, "loss": 0.7664, "step": 90 }, { "epoch": 6.4, "learning_rate": 0.001, "loss": 0.7998, "step": 92 }, { "epoch": 6.54, "learning_rate": 0.001, "loss": 0.7715, "step": 94 }, { "epoch": 6.68, "learning_rate": 0.001, "loss": 0.8038, "step": 96 }, { "epoch": 6.82, "learning_rate": 0.001, "loss": 0.8059, "step": 98 }, { "epoch": 6.96, "learning_rate": 0.001, "loss": 0.8082, "step": 100 }, { "epoch": 6.96, "eval_gen_len": 511.0, "eval_loss": 1.7575491666793823, "eval_rouge1": 26.5338, "eval_rouge2": 9.9474, "eval_rougeL": 20.3789, "eval_rougeLsum": 25.0767, "eval_runtime": 1794.6298, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 100 }, { "epoch": 7.1, "learning_rate": 0.001, "loss": 0.6708, "step": 102 }, { "epoch": 7.23, "learning_rate": 0.001, "loss": 0.6186, "step": 104 }, { "epoch": 7.37, "learning_rate": 0.001, "loss": 0.6101, "step": 106 }, { "epoch": 7.51, "learning_rate": 0.001, "loss": 0.6328, "step": 108 }, { "epoch": 7.65, "learning_rate": 0.001, "loss": 0.6529, "step": 110 }, { "epoch": 7.79, "learning_rate": 0.001, "loss": 0.6312, "step": 112 }, { "epoch": 7.93, "learning_rate": 0.001, "loss": 0.664, "step": 114 }, { "epoch": 8.0, "eval_gen_len": 329.7151335311573, "eval_loss": 1.7701919078826904, "eval_rouge1": 35.1918, "eval_rouge2": 13.7223, "eval_rougeL": 26.1763, "eval_rougeLsum": 33.3997, "eval_runtime": 1795.1755, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 115 }, { "epoch": 8.07, "learning_rate": 0.001, "loss": 0.6177, "step": 116 }, { "epoch": 8.21, "learning_rate": 0.001, "loss": 0.5241, "step": 118 }, { "epoch": 8.35, "learning_rate": 0.001, "loss": 0.5173, "step": 120 }, { "epoch": 8.49, "learning_rate": 0.001, "loss": 0.5241, "step": 122 }, { "epoch": 8.63, "learning_rate": 0.001, "loss": 0.5546, "step": 124 }, { "epoch": 8.77, "learning_rate": 0.001, "loss": 0.5401, "step": 126 }, { "epoch": 8.9, "learning_rate": 0.001, "loss": 0.5471, "step": 128 }, { "epoch": 8.97, "eval_gen_len": 506.82789317507417, "eval_loss": 1.938284993171692, "eval_rouge1": 27.0414, "eval_rouge2": 10.4166, "eval_rougeL": 20.1803, "eval_rougeLsum": 25.6283, "eval_runtime": 1797.9806, "eval_samples_per_second": 0.187, "eval_steps_per_second": 0.024, "step": 129 }, { "epoch": 9.04, "learning_rate": 0.001, "loss": 0.526, "step": 130 }, { "epoch": 9.18, "learning_rate": 0.001, "loss": 0.409, "step": 132 }, { "epoch": 9.32, "learning_rate": 0.001, "loss": 0.4163, "step": 134 }, { "epoch": 9.46, "learning_rate": 0.001, "loss": 0.4304, "step": 136 }, { "epoch": 9.6, "learning_rate": 0.001, "loss": 0.4512, "step": 138 }, { "epoch": 9.74, "learning_rate": 0.001, "loss": 0.4396, "step": 140 }, { "epoch": 9.88, "learning_rate": 0.001, "loss": 0.4349, "step": 142 }, { "epoch": 9.95, "eval_gen_len": 454.7032640949555, "eval_loss": 1.9607620239257812, "eval_rouge1": 29.5613, "eval_rouge2": 11.7633, "eval_rougeL": 22.7176, "eval_rougeLsum": 27.9563, "eval_runtime": 1794.1152, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 143 }, { "epoch": 10.02, "learning_rate": 0.001, "loss": 0.4077, "step": 144 }, { "epoch": 10.16, "learning_rate": 0.001, "loss": 0.3439, "step": 146 }, { "epoch": 10.3, "learning_rate": 0.001, "loss": 0.3503, "step": 148 }, { "epoch": 10.43, "learning_rate": 0.001, "loss": 0.3572, "step": 150 }, { "epoch": 10.57, "learning_rate": 0.001, "loss": 0.3643, "step": 152 }, { "epoch": 10.71, "learning_rate": 0.001, "loss": 0.3516, "step": 154 }, { "epoch": 10.85, "learning_rate": 0.001, "loss": 0.377, "step": 156 }, { "epoch": 10.99, "learning_rate": 0.001, "loss": 0.4338, "step": 158 }, { "epoch": 10.99, "eval_gen_len": 493.3234421364985, "eval_loss": 2.1197292804718018, "eval_rouge1": 31.2004, "eval_rouge2": 12.8569, "eval_rougeL": 22.1282, "eval_rougeLsum": 29.8827, "eval_runtime": 1793.5921, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 158 }, { "epoch": 11.13, "learning_rate": 0.001, "loss": 0.2715, "step": 160 }, { "epoch": 11.27, "learning_rate": 0.001, "loss": 0.2391, "step": 162 }, { "epoch": 11.41, "learning_rate": 0.001, "loss": 0.2958, "step": 164 }, { "epoch": 11.55, "learning_rate": 0.001, "loss": 0.3101, "step": 166 }, { "epoch": 11.69, "learning_rate": 0.001, "loss": 0.3417, "step": 168 }, { "epoch": 11.83, "learning_rate": 0.001, "loss": 0.3292, "step": 170 }, { "epoch": 11.97, "learning_rate": 0.001, "loss": 0.2887, "step": 172 }, { "epoch": 11.97, "eval_gen_len": 381.35905044510383, "eval_loss": 2.1204545497894287, "eval_rouge1": 34.9566, "eval_rouge2": 13.8574, "eval_rougeL": 25.1764, "eval_rougeLsum": 33.2914, "eval_runtime": 1792.804, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 172 }, { "epoch": 12.1, "learning_rate": 0.001, "loss": 0.2532, "step": 174 }, { "epoch": 12.24, "learning_rate": 0.001, "loss": 0.2565, "step": 176 }, { "epoch": 12.38, "learning_rate": 0.001, "loss": 0.2791, "step": 178 }, { "epoch": 12.52, "learning_rate": 0.001, "loss": 0.2803, "step": 180 }, { "epoch": 12.66, "learning_rate": 0.001, "loss": 0.3015, "step": 182 }, { "epoch": 12.8, "learning_rate": 0.001, "loss": 0.2764, "step": 184 }, { "epoch": 12.94, "learning_rate": 0.001, "loss": 0.2753, "step": 186 }, { "epoch": 12.94, "eval_gen_len": 338.7240356083086, "eval_loss": 2.429886817932129, "eval_rouge1": 36.3877, "eval_rouge2": 13.8584, "eval_rougeL": 25.7829, "eval_rougeLsum": 34.8601, "eval_runtime": 1789.0093, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 186 }, { "epoch": 13.08, "learning_rate": 0.001, "loss": 0.2563, "step": 188 }, { "epoch": 13.22, "learning_rate": 0.001, "loss": 0.2024, "step": 190 }, { "epoch": 13.36, "learning_rate": 0.001, "loss": 0.2252, "step": 192 }, { "epoch": 13.5, "learning_rate": 0.001, "loss": 0.2487, "step": 194 }, { "epoch": 13.63, "learning_rate": 0.001, "loss": 0.2086, "step": 196 }, { "epoch": 13.77, "learning_rate": 0.001, "loss": 0.2181, "step": 198 }, { "epoch": 13.91, "learning_rate": 0.001, "loss": 0.2114, "step": 200 }, { "epoch": 13.98, "eval_gen_len": 302.48367952522256, "eval_loss": 2.5798637866973877, "eval_rouge1": 39.7535, "eval_rouge2": 16.1209, "eval_rougeL": 27.8512, "eval_rougeLsum": 37.8553, "eval_runtime": 1773.2158, "eval_samples_per_second": 0.19, "eval_steps_per_second": 0.024, "step": 201 }, { "epoch": 14.05, "learning_rate": 0.001, "loss": 0.1828, "step": 202 }, { "epoch": 14.19, "learning_rate": 0.001, "loss": 0.2025, "step": 204 }, { "epoch": 14.33, "learning_rate": 0.001, "loss": 0.1991, "step": 206 }, { "epoch": 14.47, "learning_rate": 0.001, "loss": 0.1844, "step": 208 }, { "epoch": 14.61, "learning_rate": 0.001, "loss": 0.1934, "step": 210 }, { "epoch": 14.75, "learning_rate": 0.001, "loss": 0.2, "step": 212 }, { "epoch": 14.89, "learning_rate": 0.001, "loss": 0.1805, "step": 214 }, { "epoch": 14.96, "eval_gen_len": 442.9258160237389, "eval_loss": 2.612277030944824, "eval_rouge1": 33.3254, "eval_rouge2": 13.0868, "eval_rougeL": 23.3214, "eval_rougeLsum": 31.7901, "eval_runtime": 1794.4617, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 215 }, { "epoch": 15.03, "learning_rate": 0.001, "loss": 0.1739, "step": 216 }, { "epoch": 15.17, "learning_rate": 0.001, "loss": 0.1504, "step": 218 }, { "epoch": 15.3, "learning_rate": 0.001, "loss": 0.1431, "step": 220 }, { "epoch": 15.44, "learning_rate": 0.001, "loss": 0.152, "step": 222 }, { "epoch": 15.58, "learning_rate": 0.001, "loss": 0.142, "step": 224 }, { "epoch": 15.72, "learning_rate": 0.001, "loss": 0.145, "step": 226 }, { "epoch": 15.86, "learning_rate": 0.001, "loss": 0.1476, "step": 228 }, { "epoch": 16.0, "learning_rate": 0.001, "loss": 0.1543, "step": 230 }, { "epoch": 16.0, "eval_gen_len": 463.080118694362, "eval_loss": 2.563481330871582, "eval_rouge1": 31.7816, "eval_rouge2": 13.1085, "eval_rougeL": 22.9117, "eval_rougeLsum": 30.2286, "eval_runtime": 1793.3728, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 230 }, { "epoch": 16.14, "learning_rate": 0.001, "loss": 0.1245, "step": 232 }, { "epoch": 16.28, "learning_rate": 0.001, "loss": 0.1204, "step": 234 }, { "epoch": 16.42, "learning_rate": 0.001, "loss": 0.2117, "step": 236 }, { "epoch": 16.56, "learning_rate": 0.001, "loss": 0.6894, "step": 238 }, { "epoch": 16.7, "learning_rate": 0.001, "loss": 0.35, "step": 240 }, { "epoch": 16.83, "learning_rate": 0.001, "loss": 0.8395, "step": 242 }, { "epoch": 16.97, "learning_rate": 0.001, "loss": 0.5166, "step": 244 }, { "epoch": 16.97, "eval_gen_len": 511.0, "eval_loss": 2.513441562652588, "eval_rouge1": 30.3969, "eval_rouge2": 12.1295, "eval_rougeL": 21.6616, "eval_rougeLsum": 28.7606, "eval_runtime": 1793.8473, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 244 }, { "epoch": 17.11, "learning_rate": 0.001, "loss": 0.202, "step": 246 }, { "epoch": 17.25, "learning_rate": 0.001, "loss": 0.1247, "step": 248 }, { "epoch": 17.39, "learning_rate": 0.001, "loss": 0.1368, "step": 250 }, { "epoch": 17.53, "learning_rate": 0.001, "loss": 0.1096, "step": 252 }, { "epoch": 17.67, "learning_rate": 0.001, "loss": 0.1066, "step": 254 }, { "epoch": 17.81, "learning_rate": 0.001, "loss": 0.1078, "step": 256 }, { "epoch": 17.95, "learning_rate": 0.001, "loss": 0.1117, "step": 258 }, { "epoch": 17.95, "eval_gen_len": 431.11572700296733, "eval_loss": 2.8108906745910645, "eval_rouge1": 35.336, "eval_rouge2": 14.9492, "eval_rougeL": 24.1938, "eval_rougeLsum": 33.822, "eval_runtime": 1792.5735, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 258 }, { "epoch": 18.09, "learning_rate": 0.001, "loss": 0.0934, "step": 260 }, { "epoch": 18.23, "learning_rate": 0.001, "loss": 0.0793, "step": 262 }, { "epoch": 18.37, "learning_rate": 0.001, "loss": 0.0887, "step": 264 }, { "epoch": 18.5, "learning_rate": 0.001, "loss": 0.103, "step": 266 }, { "epoch": 18.64, "learning_rate": 0.001, "loss": 0.0847, "step": 268 }, { "epoch": 18.78, "learning_rate": 0.001, "loss": 0.0869, "step": 270 }, { "epoch": 18.92, "learning_rate": 0.001, "loss": 0.0895, "step": 272 }, { "epoch": 18.99, "eval_gen_len": 240.13649851632047, "eval_loss": 2.7576751708984375, "eval_rouge1": 41.0982, "eval_rouge2": 16.3935, "eval_rougeL": 28.1073, "eval_rougeLsum": 39.1641, "eval_runtime": 1702.7984, "eval_samples_per_second": 0.198, "eval_steps_per_second": 0.025, "step": 273 }, { "epoch": 19.06, "learning_rate": 0.001, "loss": 0.0884, "step": 274 }, { "epoch": 19.2, "learning_rate": 0.001, "loss": 0.0838, "step": 276 }, { "epoch": 19.34, "learning_rate": 0.001, "loss": 0.0731, "step": 278 }, { "epoch": 19.48, "learning_rate": 0.001, "loss": 0.0779, "step": 280 }, { "epoch": 19.48, "eval_gen_len": 488.513353115727, "eval_loss": 2.8927440643310547, "eval_rouge1": 32.7788, "eval_rouge2": 13.9352, "eval_rougeL": 22.5175, "eval_rougeLsum": 31.548, "eval_runtime": 1795.5446, "eval_samples_per_second": 0.188, "eval_steps_per_second": 0.024, "step": 280 }, { "epoch": 19.48, "step": 280, "total_flos": 4.895224157149471e+18, "train_loss": 0.722327525381531, "train_runtime": 104579.6075, "train_samples_per_second": 0.702, "train_steps_per_second": 0.003 } ], "logging_steps": 2, "max_steps": 280, "num_train_epochs": 20, "save_steps": 500, "total_flos": 4.895224157149471e+18, "trial_name": null, "trial_params": null }