| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 3.0, | |
| "global_step": 76095, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 4.9342926604901765e-05, | |
| "loss": 2.907, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 4.868585320980354e-05, | |
| "loss": 2.7863, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.12, | |
| "learning_rate": 4.8028779814705304e-05, | |
| "loss": 2.7315, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.16, | |
| "learning_rate": 4.737170641960707e-05, | |
| "loss": 2.7056, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.2, | |
| "learning_rate": 4.671463302450884e-05, | |
| "loss": 2.6726, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.24, | |
| "learning_rate": 4.6057559629410605e-05, | |
| "loss": 2.6651, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.28, | |
| "learning_rate": 4.5400486234312375e-05, | |
| "loss": 2.625, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.4743412839214144e-05, | |
| "loss": 2.6331, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.35, | |
| "learning_rate": 4.408633944411591e-05, | |
| "loss": 2.6084, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "learning_rate": 4.3429266049017676e-05, | |
| "loss": 2.5925, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.39, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.45664119720459, | |
| "eval_rouge1": 17.8432, | |
| "eval_rouge2": 6.6779, | |
| "eval_rougeL": 14.2303, | |
| "eval_rougeLsum": 16.1952, | |
| "eval_runtime": 446.6007, | |
| "eval_samples_per_second": 14.411, | |
| "eval_steps_per_second": 1.803, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.43, | |
| "learning_rate": 4.2772192653919446e-05, | |
| "loss": 2.5905, | |
| "step": 11000 | |
| }, | |
| { | |
| "epoch": 0.47, | |
| "learning_rate": 4.2115119258821215e-05, | |
| "loss": 2.5838, | |
| "step": 12000 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 4.145804586372298e-05, | |
| "loss": 2.565, | |
| "step": 13000 | |
| }, | |
| { | |
| "epoch": 0.55, | |
| "learning_rate": 4.080097246862475e-05, | |
| "loss": 2.5687, | |
| "step": 14000 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 4.0143899073526516e-05, | |
| "loss": 2.5542, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 3.948682567842828e-05, | |
| "loss": 2.5402, | |
| "step": 16000 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 3.882975228333005e-05, | |
| "loss": 2.5368, | |
| "step": 17000 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 3.817267888823182e-05, | |
| "loss": 2.5261, | |
| "step": 18000 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 3.751560549313358e-05, | |
| "loss": 2.5257, | |
| "step": 19000 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "learning_rate": 3.6858532098035357e-05, | |
| "loss": 2.518, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 0.79, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.3868377208709717, | |
| "eval_rouge1": 18.0354, | |
| "eval_rouge2": 6.8565, | |
| "eval_rougeL": 14.3552, | |
| "eval_rougeLsum": 16.3664, | |
| "eval_runtime": 450.4559, | |
| "eval_samples_per_second": 14.288, | |
| "eval_steps_per_second": 1.787, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 3.620145870293712e-05, | |
| "loss": 2.5159, | |
| "step": 21000 | |
| }, | |
| { | |
| "epoch": 0.87, | |
| "learning_rate": 3.554438530783889e-05, | |
| "loss": 2.4994, | |
| "step": 22000 | |
| }, | |
| { | |
| "epoch": 0.91, | |
| "learning_rate": 3.488731191274066e-05, | |
| "loss": 2.4993, | |
| "step": 23000 | |
| }, | |
| { | |
| "epoch": 0.95, | |
| "learning_rate": 3.423023851764242e-05, | |
| "loss": 2.5103, | |
| "step": 24000 | |
| }, | |
| { | |
| "epoch": 0.99, | |
| "learning_rate": 3.357316512254419e-05, | |
| "loss": 2.4925, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 1.03, | |
| "learning_rate": 3.291609172744596e-05, | |
| "loss": 2.4812, | |
| "step": 26000 | |
| }, | |
| { | |
| "epoch": 1.06, | |
| "learning_rate": 3.225901833234772e-05, | |
| "loss": 2.4806, | |
| "step": 27000 | |
| }, | |
| { | |
| "epoch": 1.1, | |
| "learning_rate": 3.160194493724949e-05, | |
| "loss": 2.4638, | |
| "step": 28000 | |
| }, | |
| { | |
| "epoch": 1.14, | |
| "learning_rate": 3.094487154215126e-05, | |
| "loss": 2.4676, | |
| "step": 29000 | |
| }, | |
| { | |
| "epoch": 1.18, | |
| "learning_rate": 3.0287798147053027e-05, | |
| "loss": 2.4587, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 1.18, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.3600151538848877, | |
| "eval_rouge1": 18.2076, | |
| "eval_rouge2": 6.9618, | |
| "eval_rougeL": 14.5349, | |
| "eval_rougeLsum": 16.5626, | |
| "eval_runtime": 450.2674, | |
| "eval_samples_per_second": 14.294, | |
| "eval_steps_per_second": 1.788, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 1.22, | |
| "learning_rate": 2.9630724751954796e-05, | |
| "loss": 2.453, | |
| "step": 31000 | |
| }, | |
| { | |
| "epoch": 1.26, | |
| "learning_rate": 2.8973651356856562e-05, | |
| "loss": 2.452, | |
| "step": 32000 | |
| }, | |
| { | |
| "epoch": 1.3, | |
| "learning_rate": 2.831657796175833e-05, | |
| "loss": 2.4553, | |
| "step": 33000 | |
| }, | |
| { | |
| "epoch": 1.34, | |
| "learning_rate": 2.7659504566660098e-05, | |
| "loss": 2.4527, | |
| "step": 34000 | |
| }, | |
| { | |
| "epoch": 1.38, | |
| "learning_rate": 2.7002431171561864e-05, | |
| "loss": 2.4478, | |
| "step": 35000 | |
| }, | |
| { | |
| "epoch": 1.42, | |
| "learning_rate": 2.634535777646363e-05, | |
| "loss": 2.4563, | |
| "step": 36000 | |
| }, | |
| { | |
| "epoch": 1.46, | |
| "learning_rate": 2.5688284381365403e-05, | |
| "loss": 2.4402, | |
| "step": 37000 | |
| }, | |
| { | |
| "epoch": 1.5, | |
| "learning_rate": 2.503121098626717e-05, | |
| "loss": 2.447, | |
| "step": 38000 | |
| }, | |
| { | |
| "epoch": 1.54, | |
| "learning_rate": 2.4374137591168935e-05, | |
| "loss": 2.4436, | |
| "step": 39000 | |
| }, | |
| { | |
| "epoch": 1.58, | |
| "learning_rate": 2.37170641960707e-05, | |
| "loss": 2.4365, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 1.58, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.329491138458252, | |
| "eval_rouge1": 18.3579, | |
| "eval_rouge2": 7.0312, | |
| "eval_rougeL": 14.6145, | |
| "eval_rougeLsum": 16.6845, | |
| "eval_runtime": 446.3631, | |
| "eval_samples_per_second": 14.419, | |
| "eval_steps_per_second": 1.803, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 1.62, | |
| "learning_rate": 2.305999080097247e-05, | |
| "loss": 2.4393, | |
| "step": 41000 | |
| }, | |
| { | |
| "epoch": 1.66, | |
| "learning_rate": 2.240291740587424e-05, | |
| "loss": 2.4266, | |
| "step": 42000 | |
| }, | |
| { | |
| "epoch": 1.7, | |
| "learning_rate": 2.1745844010776002e-05, | |
| "loss": 2.4337, | |
| "step": 43000 | |
| }, | |
| { | |
| "epoch": 1.73, | |
| "learning_rate": 2.108877061567777e-05, | |
| "loss": 2.4238, | |
| "step": 44000 | |
| }, | |
| { | |
| "epoch": 1.77, | |
| "learning_rate": 2.043169722057954e-05, | |
| "loss": 2.4304, | |
| "step": 45000 | |
| }, | |
| { | |
| "epoch": 1.81, | |
| "learning_rate": 1.9774623825481307e-05, | |
| "loss": 2.4267, | |
| "step": 46000 | |
| }, | |
| { | |
| "epoch": 1.85, | |
| "learning_rate": 1.9117550430383073e-05, | |
| "loss": 2.4263, | |
| "step": 47000 | |
| }, | |
| { | |
| "epoch": 1.89, | |
| "learning_rate": 1.8460477035284842e-05, | |
| "loss": 2.4328, | |
| "step": 48000 | |
| }, | |
| { | |
| "epoch": 1.93, | |
| "learning_rate": 1.7803403640186608e-05, | |
| "loss": 2.4215, | |
| "step": 49000 | |
| }, | |
| { | |
| "epoch": 1.97, | |
| "learning_rate": 1.7146330245088378e-05, | |
| "loss": 2.4306, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 1.97, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.3189666271209717, | |
| "eval_rouge1": 18.4551, | |
| "eval_rouge2": 7.0861, | |
| "eval_rougeL": 14.6879, | |
| "eval_rougeLsum": 16.7627, | |
| "eval_runtime": 447.1376, | |
| "eval_samples_per_second": 14.394, | |
| "eval_steps_per_second": 1.8, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 2.01, | |
| "learning_rate": 1.6489256849990144e-05, | |
| "loss": 2.4224, | |
| "step": 51000 | |
| }, | |
| { | |
| "epoch": 2.05, | |
| "learning_rate": 1.5832183454891913e-05, | |
| "loss": 2.414, | |
| "step": 52000 | |
| }, | |
| { | |
| "epoch": 2.09, | |
| "learning_rate": 1.5175110059793679e-05, | |
| "loss": 2.4029, | |
| "step": 53000 | |
| }, | |
| { | |
| "epoch": 2.13, | |
| "learning_rate": 1.4518036664695447e-05, | |
| "loss": 2.4104, | |
| "step": 54000 | |
| }, | |
| { | |
| "epoch": 2.17, | |
| "learning_rate": 1.3860963269597216e-05, | |
| "loss": 2.4117, | |
| "step": 55000 | |
| }, | |
| { | |
| "epoch": 2.21, | |
| "learning_rate": 1.320388987449898e-05, | |
| "loss": 2.4096, | |
| "step": 56000 | |
| }, | |
| { | |
| "epoch": 2.25, | |
| "learning_rate": 1.254681647940075e-05, | |
| "loss": 2.4072, | |
| "step": 57000 | |
| }, | |
| { | |
| "epoch": 2.29, | |
| "learning_rate": 1.1889743084302516e-05, | |
| "loss": 2.4062, | |
| "step": 58000 | |
| }, | |
| { | |
| "epoch": 2.33, | |
| "learning_rate": 1.1232669689204285e-05, | |
| "loss": 2.4014, | |
| "step": 59000 | |
| }, | |
| { | |
| "epoch": 2.37, | |
| "learning_rate": 1.0575596294106051e-05, | |
| "loss": 2.4005, | |
| "step": 60000 | |
| }, | |
| { | |
| "epoch": 2.37, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.3056259155273438, | |
| "eval_rouge1": 18.3521, | |
| "eval_rouge2": 7.0496, | |
| "eval_rougeL": 14.6413, | |
| "eval_rougeLsum": 16.6832, | |
| "eval_runtime": 444.8978, | |
| "eval_samples_per_second": 14.466, | |
| "eval_steps_per_second": 1.809, | |
| "step": 60000 | |
| }, | |
| { | |
| "epoch": 2.4, | |
| "learning_rate": 9.918522899007819e-06, | |
| "loss": 2.4054, | |
| "step": 61000 | |
| }, | |
| { | |
| "epoch": 2.44, | |
| "learning_rate": 9.261449503909587e-06, | |
| "loss": 2.4074, | |
| "step": 62000 | |
| }, | |
| { | |
| "epoch": 2.48, | |
| "learning_rate": 8.604376108811355e-06, | |
| "loss": 2.3965, | |
| "step": 63000 | |
| }, | |
| { | |
| "epoch": 2.52, | |
| "learning_rate": 7.947302713713122e-06, | |
| "loss": 2.3964, | |
| "step": 64000 | |
| }, | |
| { | |
| "epoch": 2.56, | |
| "learning_rate": 7.29022931861489e-06, | |
| "loss": 2.3965, | |
| "step": 65000 | |
| }, | |
| { | |
| "epoch": 2.6, | |
| "learning_rate": 6.633155923516657e-06, | |
| "loss": 2.3934, | |
| "step": 66000 | |
| }, | |
| { | |
| "epoch": 2.64, | |
| "learning_rate": 5.9760825284184245e-06, | |
| "loss": 2.4052, | |
| "step": 67000 | |
| }, | |
| { | |
| "epoch": 2.68, | |
| "learning_rate": 5.319009133320192e-06, | |
| "loss": 2.397, | |
| "step": 68000 | |
| }, | |
| { | |
| "epoch": 2.72, | |
| "learning_rate": 4.66193573822196e-06, | |
| "loss": 2.3875, | |
| "step": 69000 | |
| }, | |
| { | |
| "epoch": 2.76, | |
| "learning_rate": 4.004862343123727e-06, | |
| "loss": 2.396, | |
| "step": 70000 | |
| }, | |
| { | |
| "epoch": 2.76, | |
| "eval_gen_len": 19.0, | |
| "eval_loss": 2.3012354373931885, | |
| "eval_rouge1": 18.348, | |
| "eval_rouge2": 7.0439, | |
| "eval_rougeL": 14.6509, | |
| "eval_rougeLsum": 16.6994, | |
| "eval_runtime": 450.1278, | |
| "eval_samples_per_second": 14.298, | |
| "eval_steps_per_second": 1.788, | |
| "step": 70000 | |
| }, | |
| { | |
| "epoch": 2.8, | |
| "learning_rate": 3.3477889480254945e-06, | |
| "loss": 2.3949, | |
| "step": 71000 | |
| }, | |
| { | |
| "epoch": 2.84, | |
| "learning_rate": 2.6907155529272622e-06, | |
| "loss": 2.4087, | |
| "step": 72000 | |
| }, | |
| { | |
| "epoch": 2.88, | |
| "learning_rate": 2.0336421578290295e-06, | |
| "loss": 2.3957, | |
| "step": 73000 | |
| }, | |
| { | |
| "epoch": 2.92, | |
| "learning_rate": 1.376568762730797e-06, | |
| "loss": 2.3948, | |
| "step": 74000 | |
| }, | |
| { | |
| "epoch": 2.96, | |
| "learning_rate": 7.194953676325646e-07, | |
| "loss": 2.3925, | |
| "step": 75000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "learning_rate": 6.242197253433209e-08, | |
| "loss": 2.3883, | |
| "step": 76000 | |
| }, | |
| { | |
| "epoch": 3.0, | |
| "step": 76095, | |
| "total_flos": 1.6477647782333645e+17, | |
| "train_loss": 2.480459571265385, | |
| "train_runtime": 37543.9323, | |
| "train_samples_per_second": 16.214, | |
| "train_steps_per_second": 2.027 | |
| } | |
| ], | |
| "max_steps": 76095, | |
| "num_train_epochs": 3, | |
| "total_flos": 1.6477647782333645e+17, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |