{ "best_metric": null, "best_model_checkpoint": null, "epoch": 4.0, "eval_steps": 500, "global_step": 60000, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 4.9833333333333336e-05, "loss": 0.4822, "step": 200 }, { "epoch": 0.03, "learning_rate": 4.966666666666667e-05, "loss": 0.4862, "step": 400 }, { "epoch": 0.04, "learning_rate": 4.9500000000000004e-05, "loss": 0.4869, "step": 600 }, { "epoch": 0.05, "learning_rate": 4.933333333333334e-05, "loss": 0.4921, "step": 800 }, { "epoch": 0.07, "learning_rate": 4.9166666666666665e-05, "loss": 0.4922, "step": 1000 }, { "epoch": 0.08, "learning_rate": 4.9e-05, "loss": 0.5054, "step": 1200 }, { "epoch": 0.09, "learning_rate": 4.883333333333334e-05, "loss": 0.4965, "step": 1400 }, { "epoch": 0.11, "learning_rate": 4.866666666666667e-05, "loss": 0.499, "step": 1600 }, { "epoch": 0.12, "learning_rate": 4.85e-05, "loss": 0.5007, "step": 1800 }, { "epoch": 0.13, "learning_rate": 4.8333333333333334e-05, "loss": 0.5055, "step": 2000 }, { "epoch": 0.15, "learning_rate": 4.8166666666666674e-05, "loss": 0.502, "step": 2200 }, { "epoch": 0.16, "learning_rate": 4.8e-05, "loss": 0.503, "step": 2400 }, { "epoch": 0.17, "learning_rate": 4.7833333333333335e-05, "loss": 0.5074, "step": 2600 }, { "epoch": 0.19, "learning_rate": 4.766666666666667e-05, "loss": 0.4997, "step": 2800 }, { "epoch": 0.2, "learning_rate": 4.75e-05, "loss": 0.5062, "step": 3000 }, { "epoch": 0.21, "learning_rate": 4.7333333333333336e-05, "loss": 0.5107, "step": 3200 }, { "epoch": 0.23, "learning_rate": 4.716666666666667e-05, "loss": 0.5003, "step": 3400 }, { "epoch": 0.24, "learning_rate": 4.7e-05, "loss": 0.5005, "step": 3600 }, { "epoch": 0.25, "learning_rate": 4.683333333333334e-05, "loss": 0.5086, "step": 3800 }, { "epoch": 0.27, "learning_rate": 4.666666666666667e-05, "loss": 0.5066, "step": 4000 }, { "epoch": 0.28, "learning_rate": 4.6500000000000005e-05, "loss": 0.5108, "step": 4200 }, { "epoch": 0.29, "learning_rate": 4.633333333333333e-05, "loss": 0.4954, "step": 4400 }, { "epoch": 0.31, "learning_rate": 4.6166666666666666e-05, "loss": 0.5032, "step": 4600 }, { "epoch": 0.32, "learning_rate": 4.600000000000001e-05, "loss": 0.5038, "step": 4800 }, { "epoch": 0.33, "learning_rate": 4.5833333333333334e-05, "loss": 0.494, "step": 5000 }, { "epoch": 0.35, "learning_rate": 4.566666666666667e-05, "loss": 0.5092, "step": 5200 }, { "epoch": 0.36, "learning_rate": 4.55e-05, "loss": 0.5075, "step": 5400 }, { "epoch": 0.37, "learning_rate": 4.5333333333333335e-05, "loss": 0.501, "step": 5600 }, { "epoch": 0.39, "learning_rate": 4.516666666666667e-05, "loss": 0.4951, "step": 5800 }, { "epoch": 0.4, "learning_rate": 4.5e-05, "loss": 0.5019, "step": 6000 }, { "epoch": 0.41, "learning_rate": 4.483333333333333e-05, "loss": 0.5036, "step": 6200 }, { "epoch": 0.43, "learning_rate": 4.466666666666667e-05, "loss": 0.4893, "step": 6400 }, { "epoch": 0.44, "learning_rate": 4.4500000000000004e-05, "loss": 0.5014, "step": 6600 }, { "epoch": 0.45, "learning_rate": 4.433333333333334e-05, "loss": 0.5085, "step": 6800 }, { "epoch": 0.47, "learning_rate": 4.4166666666666665e-05, "loss": 0.5084, "step": 7000 }, { "epoch": 0.48, "learning_rate": 4.4000000000000006e-05, "loss": 0.4965, "step": 7200 }, { "epoch": 0.49, "learning_rate": 4.383333333333334e-05, "loss": 0.5047, "step": 7400 }, { "epoch": 0.51, "learning_rate": 4.3666666666666666e-05, "loss": 0.5116, "step": 7600 }, { "epoch": 0.52, "learning_rate": 4.35e-05, "loss": 0.5123, "step": 7800 }, { "epoch": 0.53, "learning_rate": 4.3333333333333334e-05, "loss": 0.5085, "step": 8000 }, { "epoch": 0.55, "learning_rate": 4.316666666666667e-05, "loss": 0.5017, "step": 8200 }, { "epoch": 0.56, "learning_rate": 4.3e-05, "loss": 0.5076, "step": 8400 }, { "epoch": 0.57, "learning_rate": 4.2833333333333335e-05, "loss": 0.4979, "step": 8600 }, { "epoch": 0.59, "learning_rate": 4.266666666666667e-05, "loss": 0.5106, "step": 8800 }, { "epoch": 0.6, "learning_rate": 4.25e-05, "loss": 0.5041, "step": 9000 }, { "epoch": 0.61, "learning_rate": 4.233333333333334e-05, "loss": 0.4982, "step": 9200 }, { "epoch": 0.63, "learning_rate": 4.216666666666667e-05, "loss": 0.5019, "step": 9400 }, { "epoch": 0.64, "learning_rate": 4.2e-05, "loss": 0.5079, "step": 9600 }, { "epoch": 0.65, "learning_rate": 4.183333333333334e-05, "loss": 0.5023, "step": 9800 }, { "epoch": 0.67, "learning_rate": 4.166666666666667e-05, "loss": 0.5066, "step": 10000 }, { "epoch": 0.68, "learning_rate": 4.15e-05, "loss": 0.4997, "step": 10200 }, { "epoch": 0.69, "learning_rate": 4.133333333333333e-05, "loss": 0.5038, "step": 10400 }, { "epoch": 0.71, "learning_rate": 4.116666666666667e-05, "loss": 0.5081, "step": 10600 }, { "epoch": 0.72, "learning_rate": 4.1e-05, "loss": 0.4978, "step": 10800 }, { "epoch": 0.73, "learning_rate": 4.0833333333333334e-05, "loss": 0.4943, "step": 11000 }, { "epoch": 0.75, "learning_rate": 4.066666666666667e-05, "loss": 0.5108, "step": 11200 }, { "epoch": 0.76, "learning_rate": 4.05e-05, "loss": 0.5028, "step": 11400 }, { "epoch": 0.77, "learning_rate": 4.0333333333333336e-05, "loss": 0.4967, "step": 11600 }, { "epoch": 0.79, "learning_rate": 4.016666666666667e-05, "loss": 0.5051, "step": 11800 }, { "epoch": 0.8, "learning_rate": 4e-05, "loss": 0.5008, "step": 12000 }, { "epoch": 0.81, "learning_rate": 3.983333333333333e-05, "loss": 0.5015, "step": 12200 }, { "epoch": 0.83, "learning_rate": 3.966666666666667e-05, "loss": 0.5001, "step": 12400 }, { "epoch": 0.84, "learning_rate": 3.9500000000000005e-05, "loss": 0.4999, "step": 12600 }, { "epoch": 0.85, "learning_rate": 3.933333333333333e-05, "loss": 0.506, "step": 12800 }, { "epoch": 0.87, "learning_rate": 3.9166666666666665e-05, "loss": 0.5071, "step": 13000 }, { "epoch": 0.88, "learning_rate": 3.9000000000000006e-05, "loss": 0.499, "step": 13200 }, { "epoch": 0.89, "learning_rate": 3.883333333333333e-05, "loss": 0.5013, "step": 13400 }, { "epoch": 0.91, "learning_rate": 3.866666666666667e-05, "loss": 0.5068, "step": 13600 }, { "epoch": 0.92, "learning_rate": 3.85e-05, "loss": 0.5012, "step": 13800 }, { "epoch": 0.93, "learning_rate": 3.8333333333333334e-05, "loss": 0.4999, "step": 14000 }, { "epoch": 0.95, "learning_rate": 3.816666666666667e-05, "loss": 0.4964, "step": 14200 }, { "epoch": 0.96, "learning_rate": 3.8e-05, "loss": 0.5054, "step": 14400 }, { "epoch": 0.97, "learning_rate": 3.7833333333333336e-05, "loss": 0.4953, "step": 14600 }, { "epoch": 0.99, "learning_rate": 3.766666666666667e-05, "loss": 0.4969, "step": 14800 }, { "epoch": 1.0, "learning_rate": 3.7500000000000003e-05, "loss": 0.4963, "step": 15000 }, { "epoch": 1.0, "eval_gen_len": 17.499259198287923, "eval_loss": 3.158352851867676, "eval_rouge1": 34.8762, "eval_rouge2": 14.5541, "eval_rougeL": 30.9109, "eval_rougeLsum": 30.9733, "eval_runtime": 3289.9782, "eval_samples_per_second": 7.385, "eval_steps_per_second": 1.847, "step": 15000 }, { "epoch": 1.01, "learning_rate": 3.733333333333334e-05, "loss": 0.3748, "step": 15200 }, { "epoch": 1.03, "learning_rate": 3.7166666666666664e-05, "loss": 0.3788, "step": 15400 }, { "epoch": 1.04, "learning_rate": 3.7e-05, "loss": 0.3805, "step": 15600 }, { "epoch": 1.05, "learning_rate": 3.683333333333334e-05, "loss": 0.3788, "step": 15800 }, { "epoch": 1.07, "learning_rate": 3.6666666666666666e-05, "loss": 0.3818, "step": 16000 }, { "epoch": 1.08, "learning_rate": 3.65e-05, "loss": 0.3776, "step": 16200 }, { "epoch": 1.09, "learning_rate": 3.633333333333333e-05, "loss": 0.3799, "step": 16400 }, { "epoch": 1.11, "learning_rate": 3.6166666666666674e-05, "loss": 0.384, "step": 16600 }, { "epoch": 1.12, "learning_rate": 3.6e-05, "loss": 0.3881, "step": 16800 }, { "epoch": 1.13, "learning_rate": 3.5833333333333335e-05, "loss": 0.3829, "step": 17000 }, { "epoch": 1.15, "learning_rate": 3.566666666666667e-05, "loss": 0.3864, "step": 17200 }, { "epoch": 1.16, "learning_rate": 3.55e-05, "loss": 0.3854, "step": 17400 }, { "epoch": 1.17, "learning_rate": 3.5333333333333336e-05, "loss": 0.3908, "step": 17600 }, { "epoch": 1.19, "learning_rate": 3.516666666666667e-05, "loss": 0.3914, "step": 17800 }, { "epoch": 1.2, "learning_rate": 3.5e-05, "loss": 0.3902, "step": 18000 }, { "epoch": 1.21, "learning_rate": 3.483333333333334e-05, "loss": 0.3918, "step": 18200 }, { "epoch": 1.23, "learning_rate": 3.466666666666667e-05, "loss": 0.392, "step": 18400 }, { "epoch": 1.24, "learning_rate": 3.45e-05, "loss": 0.3896, "step": 18600 }, { "epoch": 1.25, "learning_rate": 3.433333333333333e-05, "loss": 0.391, "step": 18800 }, { "epoch": 1.27, "learning_rate": 3.4166666666666666e-05, "loss": 0.3925, "step": 19000 }, { "epoch": 1.28, "learning_rate": 3.4000000000000007e-05, "loss": 0.3819, "step": 19200 }, { "epoch": 1.29, "learning_rate": 3.3833333333333334e-05, "loss": 0.391, "step": 19400 }, { "epoch": 1.31, "learning_rate": 3.366666666666667e-05, "loss": 0.386, "step": 19600 }, { "epoch": 1.32, "learning_rate": 3.35e-05, "loss": 0.3919, "step": 19800 }, { "epoch": 1.33, "learning_rate": 3.3333333333333335e-05, "loss": 0.3869, "step": 20000 }, { "epoch": 1.35, "learning_rate": 3.316666666666667e-05, "loss": 0.3882, "step": 20200 }, { "epoch": 1.36, "learning_rate": 3.3e-05, "loss": 0.3877, "step": 20400 }, { "epoch": 1.37, "learning_rate": 3.283333333333333e-05, "loss": 0.3893, "step": 20600 }, { "epoch": 1.39, "learning_rate": 3.266666666666667e-05, "loss": 0.3932, "step": 20800 }, { "epoch": 1.4, "learning_rate": 3.2500000000000004e-05, "loss": 0.3897, "step": 21000 }, { "epoch": 1.41, "learning_rate": 3.233333333333333e-05, "loss": 0.3869, "step": 21200 }, { "epoch": 1.43, "learning_rate": 3.2166666666666665e-05, "loss": 0.3895, "step": 21400 }, { "epoch": 1.44, "learning_rate": 3.2000000000000005e-05, "loss": 0.3873, "step": 21600 }, { "epoch": 1.45, "learning_rate": 3.183333333333334e-05, "loss": 0.3906, "step": 21800 }, { "epoch": 1.47, "learning_rate": 3.1666666666666666e-05, "loss": 0.393, "step": 22000 }, { "epoch": 1.48, "learning_rate": 3.15e-05, "loss": 0.3879, "step": 22200 }, { "epoch": 1.49, "learning_rate": 3.1333333333333334e-05, "loss": 0.3859, "step": 22400 }, { "epoch": 1.51, "learning_rate": 3.116666666666667e-05, "loss": 0.3908, "step": 22600 }, { "epoch": 1.52, "learning_rate": 3.1e-05, "loss": 0.3891, "step": 22800 }, { "epoch": 1.53, "learning_rate": 3.0833333333333335e-05, "loss": 0.3956, "step": 23000 }, { "epoch": 1.55, "learning_rate": 3.066666666666667e-05, "loss": 0.3912, "step": 23200 }, { "epoch": 1.56, "learning_rate": 3.05e-05, "loss": 0.3875, "step": 23400 }, { "epoch": 1.57, "learning_rate": 3.0333333333333337e-05, "loss": 0.3839, "step": 23600 }, { "epoch": 1.59, "learning_rate": 3.016666666666667e-05, "loss": 0.3894, "step": 23800 }, { "epoch": 1.6, "learning_rate": 3e-05, "loss": 0.3894, "step": 24000 }, { "epoch": 1.61, "learning_rate": 2.9833333333333335e-05, "loss": 0.3923, "step": 24200 }, { "epoch": 1.63, "learning_rate": 2.9666666666666672e-05, "loss": 0.394, "step": 24400 }, { "epoch": 1.64, "learning_rate": 2.95e-05, "loss": 0.3928, "step": 24600 }, { "epoch": 1.65, "learning_rate": 2.9333333333333336e-05, "loss": 0.3931, "step": 24800 }, { "epoch": 1.67, "learning_rate": 2.916666666666667e-05, "loss": 0.3915, "step": 25000 }, { "epoch": 1.68, "learning_rate": 2.9e-05, "loss": 0.3886, "step": 25200 }, { "epoch": 1.69, "learning_rate": 2.8833333333333334e-05, "loss": 0.3934, "step": 25400 }, { "epoch": 1.71, "learning_rate": 2.8666666666666668e-05, "loss": 0.3905, "step": 25600 }, { "epoch": 1.72, "learning_rate": 2.8499999999999998e-05, "loss": 0.3937, "step": 25800 }, { "epoch": 1.73, "learning_rate": 2.8333333333333335e-05, "loss": 0.3879, "step": 26000 }, { "epoch": 1.75, "learning_rate": 2.816666666666667e-05, "loss": 0.3916, "step": 26200 }, { "epoch": 1.76, "learning_rate": 2.8000000000000003e-05, "loss": 0.3914, "step": 26400 }, { "epoch": 1.77, "learning_rate": 2.7833333333333333e-05, "loss": 0.3861, "step": 26600 }, { "epoch": 1.79, "learning_rate": 2.7666666666666667e-05, "loss": 0.3866, "step": 26800 }, { "epoch": 1.8, "learning_rate": 2.7500000000000004e-05, "loss": 0.3859, "step": 27000 }, { "epoch": 1.81, "learning_rate": 2.733333333333333e-05, "loss": 0.39, "step": 27200 }, { "epoch": 1.83, "learning_rate": 2.716666666666667e-05, "loss": 0.3966, "step": 27400 }, { "epoch": 1.84, "learning_rate": 2.7000000000000002e-05, "loss": 0.3868, "step": 27600 }, { "epoch": 1.85, "learning_rate": 2.6833333333333333e-05, "loss": 0.3922, "step": 27800 }, { "epoch": 1.87, "learning_rate": 2.6666666666666667e-05, "loss": 0.3867, "step": 28000 }, { "epoch": 1.88, "learning_rate": 2.6500000000000004e-05, "loss": 0.3941, "step": 28200 }, { "epoch": 1.89, "learning_rate": 2.633333333333333e-05, "loss": 0.3849, "step": 28400 }, { "epoch": 1.91, "learning_rate": 2.6166666666666668e-05, "loss": 0.3891, "step": 28600 }, { "epoch": 1.92, "learning_rate": 2.6000000000000002e-05, "loss": 0.393, "step": 28800 }, { "epoch": 1.93, "learning_rate": 2.5833333333333336e-05, "loss": 0.3892, "step": 29000 }, { "epoch": 1.95, "learning_rate": 2.5666666666666666e-05, "loss": 0.3902, "step": 29200 }, { "epoch": 1.96, "learning_rate": 2.5500000000000003e-05, "loss": 0.3889, "step": 29400 }, { "epoch": 1.97, "learning_rate": 2.5333333333333337e-05, "loss": 0.3902, "step": 29600 }, { "epoch": 1.99, "learning_rate": 2.5166666666666667e-05, "loss": 0.3864, "step": 29800 }, { "epoch": 2.0, "learning_rate": 2.5e-05, "loss": 0.3924, "step": 30000 }, { "epoch": 2.0, "eval_gen_len": 16.967157790764674, "eval_loss": 3.3668367862701416, "eval_rouge1": 34.9558, "eval_rouge2": 14.444, "eval_rougeL": 30.9878, "eval_rougeLsum": 31.0487, "eval_runtime": 3218.069, "eval_samples_per_second": 7.55, "eval_steps_per_second": 1.888, "step": 30000 }, { "epoch": 2.01, "learning_rate": 2.4833333333333335e-05, "loss": 0.3016, "step": 30200 }, { "epoch": 2.03, "learning_rate": 2.466666666666667e-05, "loss": 0.296, "step": 30400 }, { "epoch": 2.04, "learning_rate": 2.45e-05, "loss": 0.3032, "step": 30600 }, { "epoch": 2.05, "learning_rate": 2.4333333333333336e-05, "loss": 0.2978, "step": 30800 }, { "epoch": 2.07, "learning_rate": 2.4166666666666667e-05, "loss": 0.3062, "step": 31000 }, { "epoch": 2.08, "learning_rate": 2.4e-05, "loss": 0.3025, "step": 31200 }, { "epoch": 2.09, "learning_rate": 2.3833333333333334e-05, "loss": 0.3083, "step": 31400 }, { "epoch": 2.11, "learning_rate": 2.3666666666666668e-05, "loss": 0.3027, "step": 31600 }, { "epoch": 2.12, "learning_rate": 2.35e-05, "loss": 0.3031, "step": 31800 }, { "epoch": 2.13, "learning_rate": 2.3333333333333336e-05, "loss": 0.3011, "step": 32000 }, { "epoch": 2.15, "learning_rate": 2.3166666666666666e-05, "loss": 0.3032, "step": 32200 }, { "epoch": 2.16, "learning_rate": 2.3000000000000003e-05, "loss": 0.3012, "step": 32400 }, { "epoch": 2.17, "learning_rate": 2.2833333333333334e-05, "loss": 0.3039, "step": 32600 }, { "epoch": 2.19, "learning_rate": 2.2666666666666668e-05, "loss": 0.3047, "step": 32800 }, { "epoch": 2.2, "learning_rate": 2.25e-05, "loss": 0.305, "step": 33000 }, { "epoch": 2.21, "learning_rate": 2.2333333333333335e-05, "loss": 0.306, "step": 33200 }, { "epoch": 2.23, "learning_rate": 2.216666666666667e-05, "loss": 0.2989, "step": 33400 }, { "epoch": 2.24, "learning_rate": 2.2000000000000003e-05, "loss": 0.3099, "step": 33600 }, { "epoch": 2.25, "learning_rate": 2.1833333333333333e-05, "loss": 0.3028, "step": 33800 }, { "epoch": 2.27, "learning_rate": 2.1666666666666667e-05, "loss": 0.3064, "step": 34000 }, { "epoch": 2.28, "learning_rate": 2.15e-05, "loss": 0.3092, "step": 34200 }, { "epoch": 2.29, "learning_rate": 2.1333333333333335e-05, "loss": 0.3044, "step": 34400 }, { "epoch": 2.31, "learning_rate": 2.116666666666667e-05, "loss": 0.3059, "step": 34600 }, { "epoch": 2.32, "learning_rate": 2.1e-05, "loss": 0.311, "step": 34800 }, { "epoch": 2.33, "learning_rate": 2.0833333333333336e-05, "loss": 0.3086, "step": 35000 }, { "epoch": 2.35, "learning_rate": 2.0666666666666666e-05, "loss": 0.3114, "step": 35200 }, { "epoch": 2.36, "learning_rate": 2.05e-05, "loss": 0.3085, "step": 35400 }, { "epoch": 2.37, "learning_rate": 2.0333333333333334e-05, "loss": 0.3103, "step": 35600 }, { "epoch": 2.39, "learning_rate": 2.0166666666666668e-05, "loss": 0.3072, "step": 35800 }, { "epoch": 2.4, "learning_rate": 2e-05, "loss": 0.3061, "step": 36000 }, { "epoch": 2.41, "learning_rate": 1.9833333333333335e-05, "loss": 0.3078, "step": 36200 }, { "epoch": 2.43, "learning_rate": 1.9666666666666666e-05, "loss": 0.3053, "step": 36400 }, { "epoch": 2.44, "learning_rate": 1.9500000000000003e-05, "loss": 0.3042, "step": 36600 }, { "epoch": 2.45, "learning_rate": 1.9333333333333333e-05, "loss": 0.3047, "step": 36800 }, { "epoch": 2.47, "learning_rate": 1.9166666666666667e-05, "loss": 0.3051, "step": 37000 }, { "epoch": 2.48, "learning_rate": 1.9e-05, "loss": 0.3012, "step": 37200 }, { "epoch": 2.49, "learning_rate": 1.8833333333333335e-05, "loss": 0.3054, "step": 37400 }, { "epoch": 2.51, "learning_rate": 1.866666666666667e-05, "loss": 0.3058, "step": 37600 }, { "epoch": 2.52, "learning_rate": 1.85e-05, "loss": 0.3061, "step": 37800 }, { "epoch": 2.53, "learning_rate": 1.8333333333333333e-05, "loss": 0.312, "step": 38000 }, { "epoch": 2.55, "learning_rate": 1.8166666666666667e-05, "loss": 0.3102, "step": 38200 }, { "epoch": 2.56, "learning_rate": 1.8e-05, "loss": 0.3058, "step": 38400 }, { "epoch": 2.57, "learning_rate": 1.7833333333333334e-05, "loss": 0.3077, "step": 38600 }, { "epoch": 2.59, "learning_rate": 1.7666666666666668e-05, "loss": 0.3043, "step": 38800 }, { "epoch": 2.6, "learning_rate": 1.75e-05, "loss": 0.3082, "step": 39000 }, { "epoch": 2.61, "learning_rate": 1.7333333333333336e-05, "loss": 0.3051, "step": 39200 }, { "epoch": 2.63, "learning_rate": 1.7166666666666666e-05, "loss": 0.3109, "step": 39400 }, { "epoch": 2.64, "learning_rate": 1.7000000000000003e-05, "loss": 0.3058, "step": 39600 }, { "epoch": 2.65, "learning_rate": 1.6833333333333334e-05, "loss": 0.3055, "step": 39800 }, { "epoch": 2.67, "learning_rate": 1.6666666666666667e-05, "loss": 0.3052, "step": 40000 }, { "epoch": 2.68, "learning_rate": 1.65e-05, "loss": 0.3074, "step": 40200 }, { "epoch": 2.69, "learning_rate": 1.6333333333333335e-05, "loss": 0.3029, "step": 40400 }, { "epoch": 2.71, "learning_rate": 1.6166666666666665e-05, "loss": 0.3076, "step": 40600 }, { "epoch": 2.72, "learning_rate": 1.6000000000000003e-05, "loss": 0.303, "step": 40800 }, { "epoch": 2.73, "learning_rate": 1.5833333333333333e-05, "loss": 0.3078, "step": 41000 }, { "epoch": 2.75, "learning_rate": 1.5666666666666667e-05, "loss": 0.307, "step": 41200 }, { "epoch": 2.76, "learning_rate": 1.55e-05, "loss": 0.3104, "step": 41400 }, { "epoch": 2.77, "learning_rate": 1.5333333333333334e-05, "loss": 0.3022, "step": 41600 }, { "epoch": 2.79, "learning_rate": 1.5166666666666668e-05, "loss": 0.304, "step": 41800 }, { "epoch": 2.8, "learning_rate": 1.5e-05, "loss": 0.3093, "step": 42000 }, { "epoch": 2.81, "learning_rate": 1.4833333333333336e-05, "loss": 0.3074, "step": 42200 }, { "epoch": 2.83, "learning_rate": 1.4666666666666668e-05, "loss": 0.2986, "step": 42400 }, { "epoch": 2.84, "learning_rate": 1.45e-05, "loss": 0.3095, "step": 42600 }, { "epoch": 2.85, "learning_rate": 1.4333333333333334e-05, "loss": 0.3096, "step": 42800 }, { "epoch": 2.87, "learning_rate": 1.4166666666666668e-05, "loss": 0.3104, "step": 43000 }, { "epoch": 2.88, "learning_rate": 1.4000000000000001e-05, "loss": 0.306, "step": 43200 }, { "epoch": 2.89, "learning_rate": 1.3833333333333334e-05, "loss": 0.3044, "step": 43400 }, { "epoch": 2.91, "learning_rate": 1.3666666666666666e-05, "loss": 0.3038, "step": 43600 }, { "epoch": 2.92, "learning_rate": 1.3500000000000001e-05, "loss": 0.3043, "step": 43800 }, { "epoch": 2.93, "learning_rate": 1.3333333333333333e-05, "loss": 0.303, "step": 44000 }, { "epoch": 2.95, "learning_rate": 1.3166666666666665e-05, "loss": 0.3071, "step": 44200 }, { "epoch": 2.96, "learning_rate": 1.3000000000000001e-05, "loss": 0.2934, "step": 44400 }, { "epoch": 2.97, "learning_rate": 1.2833333333333333e-05, "loss": 0.3034, "step": 44600 }, { "epoch": 2.99, "learning_rate": 1.2666666666666668e-05, "loss": 0.3111, "step": 44800 }, { "epoch": 3.0, "learning_rate": 1.25e-05, "loss": 0.3005, "step": 45000 }, { "epoch": 3.0, "eval_gen_len": 17.190715285208658, "eval_loss": 3.592427968978882, "eval_rouge1": 35.1731, "eval_rouge2": 14.5281, "eval_rougeL": 31.1176, "eval_rougeLsum": 31.1876, "eval_runtime": 3269.7836, "eval_samples_per_second": 7.431, "eval_steps_per_second": 1.858, "step": 45000 }, { "epoch": 3.01, "learning_rate": 1.2333333333333334e-05, "loss": 0.2488, "step": 45200 }, { "epoch": 3.03, "learning_rate": 1.2166666666666668e-05, "loss": 0.2485, "step": 45400 }, { "epoch": 3.04, "learning_rate": 1.2e-05, "loss": 0.2482, "step": 45600 }, { "epoch": 3.05, "learning_rate": 1.1833333333333334e-05, "loss": 0.243, "step": 45800 }, { "epoch": 3.07, "learning_rate": 1.1666666666666668e-05, "loss": 0.2503, "step": 46000 }, { "epoch": 3.08, "learning_rate": 1.1500000000000002e-05, "loss": 0.249, "step": 46200 }, { "epoch": 3.09, "learning_rate": 1.1333333333333334e-05, "loss": 0.2444, "step": 46400 }, { "epoch": 3.11, "learning_rate": 1.1166666666666668e-05, "loss": 0.2505, "step": 46600 }, { "epoch": 3.12, "learning_rate": 1.1000000000000001e-05, "loss": 0.2491, "step": 46800 }, { "epoch": 3.13, "learning_rate": 1.0833333333333334e-05, "loss": 0.251, "step": 47000 }, { "epoch": 3.15, "learning_rate": 1.0666666666666667e-05, "loss": 0.2501, "step": 47200 }, { "epoch": 3.16, "learning_rate": 1.05e-05, "loss": 0.2465, "step": 47400 }, { "epoch": 3.17, "learning_rate": 1.0333333333333333e-05, "loss": 0.2505, "step": 47600 }, { "epoch": 3.19, "learning_rate": 1.0166666666666667e-05, "loss": 0.2437, "step": 47800 }, { "epoch": 3.2, "learning_rate": 1e-05, "loss": 0.2498, "step": 48000 }, { "epoch": 3.21, "learning_rate": 9.833333333333333e-06, "loss": 0.2511, "step": 48200 }, { "epoch": 3.23, "learning_rate": 9.666666666666667e-06, "loss": 0.248, "step": 48400 }, { "epoch": 3.24, "learning_rate": 9.5e-06, "loss": 0.2442, "step": 48600 }, { "epoch": 3.25, "learning_rate": 9.333333333333334e-06, "loss": 0.249, "step": 48800 }, { "epoch": 3.27, "learning_rate": 9.166666666666666e-06, "loss": 0.2486, "step": 49000 }, { "epoch": 3.28, "learning_rate": 9e-06, "loss": 0.2476, "step": 49200 }, { "epoch": 3.29, "learning_rate": 8.833333333333334e-06, "loss": 0.2485, "step": 49400 }, { "epoch": 3.31, "learning_rate": 8.666666666666668e-06, "loss": 0.2489, "step": 49600 }, { "epoch": 3.32, "learning_rate": 8.500000000000002e-06, "loss": 0.2471, "step": 49800 }, { "epoch": 3.33, "learning_rate": 8.333333333333334e-06, "loss": 0.2478, "step": 50000 }, { "epoch": 3.35, "learning_rate": 8.166666666666668e-06, "loss": 0.2491, "step": 50200 }, { "epoch": 3.36, "learning_rate": 8.000000000000001e-06, "loss": 0.2464, "step": 50400 }, { "epoch": 3.37, "learning_rate": 7.833333333333333e-06, "loss": 0.244, "step": 50600 }, { "epoch": 3.39, "learning_rate": 7.666666666666667e-06, "loss": 0.2441, "step": 50800 }, { "epoch": 3.4, "learning_rate": 7.5e-06, "loss": 0.2485, "step": 51000 }, { "epoch": 3.41, "learning_rate": 7.333333333333334e-06, "loss": 0.2457, "step": 51200 }, { "epoch": 3.43, "learning_rate": 7.166666666666667e-06, "loss": 0.2433, "step": 51400 }, { "epoch": 3.44, "learning_rate": 7.000000000000001e-06, "loss": 0.2454, "step": 51600 }, { "epoch": 3.45, "learning_rate": 6.833333333333333e-06, "loss": 0.2502, "step": 51800 }, { "epoch": 3.47, "learning_rate": 6.666666666666667e-06, "loss": 0.247, "step": 52000 }, { "epoch": 3.48, "learning_rate": 6.5000000000000004e-06, "loss": 0.2466, "step": 52200 }, { "epoch": 3.49, "learning_rate": 6.333333333333334e-06, "loss": 0.253, "step": 52400 }, { "epoch": 3.51, "learning_rate": 6.166666666666667e-06, "loss": 0.245, "step": 52600 }, { "epoch": 3.52, "learning_rate": 6e-06, "loss": 0.2438, "step": 52800 }, { "epoch": 3.53, "learning_rate": 5.833333333333334e-06, "loss": 0.2475, "step": 53000 }, { "epoch": 3.55, "learning_rate": 5.666666666666667e-06, "loss": 0.2516, "step": 53200 }, { "epoch": 3.56, "learning_rate": 5.500000000000001e-06, "loss": 0.2497, "step": 53400 }, { "epoch": 3.57, "learning_rate": 5.333333333333334e-06, "loss": 0.2453, "step": 53600 }, { "epoch": 3.59, "learning_rate": 5.166666666666667e-06, "loss": 0.249, "step": 53800 }, { "epoch": 3.6, "learning_rate": 5e-06, "loss": 0.2468, "step": 54000 }, { "epoch": 3.61, "learning_rate": 4.833333333333333e-06, "loss": 0.2488, "step": 54200 }, { "epoch": 3.63, "learning_rate": 4.666666666666667e-06, "loss": 0.2452, "step": 54400 }, { "epoch": 3.64, "learning_rate": 4.5e-06, "loss": 0.2456, "step": 54600 }, { "epoch": 3.65, "learning_rate": 4.333333333333334e-06, "loss": 0.2499, "step": 54800 }, { "epoch": 3.67, "learning_rate": 4.166666666666667e-06, "loss": 0.2463, "step": 55000 }, { "epoch": 3.68, "learning_rate": 4.000000000000001e-06, "loss": 0.2485, "step": 55200 }, { "epoch": 3.69, "learning_rate": 3.833333333333334e-06, "loss": 0.2519, "step": 55400 }, { "epoch": 3.71, "learning_rate": 3.666666666666667e-06, "loss": 0.2502, "step": 55600 }, { "epoch": 3.72, "learning_rate": 3.5000000000000004e-06, "loss": 0.2533, "step": 55800 }, { "epoch": 3.73, "learning_rate": 3.3333333333333333e-06, "loss": 0.2471, "step": 56000 }, { "epoch": 3.75, "learning_rate": 3.166666666666667e-06, "loss": 0.2498, "step": 56200 }, { "epoch": 3.76, "learning_rate": 3e-06, "loss": 0.246, "step": 56400 }, { "epoch": 3.77, "learning_rate": 2.8333333333333335e-06, "loss": 0.2469, "step": 56600 }, { "epoch": 3.79, "learning_rate": 2.666666666666667e-06, "loss": 0.2493, "step": 56800 }, { "epoch": 3.8, "learning_rate": 2.5e-06, "loss": 0.249, "step": 57000 }, { "epoch": 3.81, "learning_rate": 2.3333333333333336e-06, "loss": 0.2436, "step": 57200 }, { "epoch": 3.83, "learning_rate": 2.166666666666667e-06, "loss": 0.2471, "step": 57400 }, { "epoch": 3.84, "learning_rate": 2.0000000000000003e-06, "loss": 0.2486, "step": 57600 }, { "epoch": 3.85, "learning_rate": 1.8333333333333335e-06, "loss": 0.2507, "step": 57800 }, { "epoch": 3.87, "learning_rate": 1.6666666666666667e-06, "loss": 0.2434, "step": 58000 }, { "epoch": 3.88, "learning_rate": 1.5e-06, "loss": 0.2499, "step": 58200 }, { "epoch": 3.89, "learning_rate": 1.3333333333333334e-06, "loss": 0.2477, "step": 58400 }, { "epoch": 3.91, "learning_rate": 1.1666666666666668e-06, "loss": 0.2494, "step": 58600 }, { "epoch": 3.92, "learning_rate": 1.0000000000000002e-06, "loss": 0.2427, "step": 58800 }, { "epoch": 3.93, "learning_rate": 8.333333333333333e-07, "loss": 0.2403, "step": 59000 }, { "epoch": 3.95, "learning_rate": 6.666666666666667e-07, "loss": 0.246, "step": 59200 }, { "epoch": 3.96, "learning_rate": 5.000000000000001e-07, "loss": 0.2467, "step": 59400 }, { "epoch": 3.97, "learning_rate": 3.3333333333333335e-07, "loss": 0.2455, "step": 59600 }, { "epoch": 3.99, "learning_rate": 1.6666666666666668e-07, "loss": 0.2488, "step": 59800 }, { "epoch": 4.0, "learning_rate": 0.0, "loss": 0.251, "step": 60000 }, { "epoch": 4.0, "eval_gen_len": 17.042061074985597, "eval_loss": 3.7253761291503906, "eval_rouge1": 35.3058, "eval_rouge2": 14.5137, "eval_rougeL": 31.2044, "eval_rougeLsum": 31.2698, "eval_runtime": 3366.7596, "eval_samples_per_second": 7.217, "eval_steps_per_second": 1.804, "step": 60000 } ], "logging_steps": 200, "max_steps": 60000, "num_input_tokens_seen": 0, "num_train_epochs": 4, "save_steps": 500, "total_flos": 5.064159661468877e+17, "train_batch_size": 4, "trial_name": null, "trial_params": null }