LLM_Teached_Pegasus_50k / trainer_state.json
GlycerinLOL's picture
End of training
f6d1d9d verified
raw
history blame
11.5 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 19.990403071017276,
"eval_steps": 500,
"global_step": 7800,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 1.0,
"eval_f1": 0.9034,
"eval_gen_len": 26.296727272727274,
"eval_loss": 1.8257849216461182,
"eval_precision": 0.9049,
"eval_recall": 0.9023,
"eval_rouge1": 0.4338,
"eval_rouge2": 0.1906,
"eval_rougeL": 0.3496,
"eval_rougeLsum": 0.3498,
"eval_runtime": 513.5292,
"eval_samples_per_second": 5.355,
"eval_steps_per_second": 0.335,
"step": 390
},
{
"epoch": 1.28,
"learning_rate": 1.3589743589743592e-05,
"loss": 2.1621,
"step": 500
},
{
"epoch": 2.0,
"eval_f1": 0.9054,
"eval_gen_len": 26.272727272727273,
"eval_loss": 1.7537195682525635,
"eval_precision": 0.9068,
"eval_recall": 0.9044,
"eval_rouge1": 0.4449,
"eval_rouge2": 0.2005,
"eval_rougeL": 0.3633,
"eval_rougeLsum": 0.3633,
"eval_runtime": 505.2109,
"eval_samples_per_second": 5.443,
"eval_steps_per_second": 0.34,
"step": 781
},
{
"epoch": 2.56,
"learning_rate": 7.17948717948718e-06,
"loss": 1.8794,
"step": 1000
},
{
"epoch": 3.0,
"eval_f1": 0.9066,
"eval_gen_len": 26.434545454545454,
"eval_loss": 1.726783275604248,
"eval_precision": 0.9078,
"eval_recall": 0.9058,
"eval_rouge1": 0.4518,
"eval_rouge2": 0.2061,
"eval_rougeL": 0.3696,
"eval_rougeLsum": 0.3695,
"eval_runtime": 507.6007,
"eval_samples_per_second": 5.418,
"eval_steps_per_second": 0.339,
"step": 1172
},
{
"epoch": 3.84,
"learning_rate": 7.692307692307694e-07,
"loss": 1.8271,
"step": 1500
},
{
"epoch": 4.0,
"eval_f1": 0.9069,
"eval_gen_len": 26.39709090909091,
"eval_loss": 1.7157036066055298,
"eval_precision": 0.9082,
"eval_recall": 0.906,
"eval_rouge1": 0.4539,
"eval_rouge2": 0.2075,
"eval_rougeL": 0.3716,
"eval_rougeLsum": 0.3714,
"eval_runtime": 505.7397,
"eval_samples_per_second": 5.438,
"eval_steps_per_second": 0.34,
"step": 1560
},
{
"epoch": 5.0,
"eval_f1": 0.9074,
"eval_gen_len": 26.301454545454547,
"eval_loss": 1.703265905380249,
"eval_precision": 0.9087,
"eval_recall": 0.9065,
"eval_rouge1": 0.4561,
"eval_rouge2": 0.2098,
"eval_rougeL": 0.3735,
"eval_rougeLsum": 0.3734,
"eval_runtime": 505.7439,
"eval_samples_per_second": 5.438,
"eval_steps_per_second": 0.34,
"step": 1951
},
{
"epoch": 5.12,
"learning_rate": 2.9059829059829063e-06,
"loss": 1.8067,
"step": 2000
},
{
"epoch": 6.0,
"eval_f1": 0.9077,
"eval_gen_len": 26.438909090909092,
"eval_loss": 1.6896997690200806,
"eval_precision": 0.9089,
"eval_recall": 0.9069,
"eval_rouge1": 0.4592,
"eval_rouge2": 0.2114,
"eval_rougeL": 0.3762,
"eval_rougeLsum": 0.3759,
"eval_runtime": 506.0324,
"eval_samples_per_second": 5.434,
"eval_steps_per_second": 0.34,
"step": 2340
},
{
"epoch": 6.41,
"learning_rate": 3.974358974358974e-06,
"loss": 1.7833,
"step": 2500
},
{
"epoch": 7.0,
"eval_f1": 0.9079,
"eval_gen_len": 26.374545454545455,
"eval_loss": 1.6819266080856323,
"eval_precision": 0.9092,
"eval_recall": 0.9071,
"eval_rouge1": 0.4598,
"eval_rouge2": 0.2115,
"eval_rougeL": 0.3764,
"eval_rougeLsum": 0.376,
"eval_runtime": 506.9661,
"eval_samples_per_second": 5.424,
"eval_steps_per_second": 0.339,
"step": 2731
},
{
"epoch": 7.69,
"learning_rate": 7.692307692307694e-07,
"loss": 1.7683,
"step": 3000
},
{
"epoch": 8.0,
"eval_f1": 0.9083,
"eval_gen_len": 26.620363636363635,
"eval_loss": 1.676251769065857,
"eval_precision": 0.9094,
"eval_recall": 0.9076,
"eval_rouge1": 0.4621,
"eval_rouge2": 0.2133,
"eval_rougeL": 0.3791,
"eval_rougeLsum": 0.3789,
"eval_runtime": 506.5847,
"eval_samples_per_second": 5.429,
"eval_steps_per_second": 0.34,
"step": 3120
},
{
"epoch": 8.97,
"learning_rate": 5.042735042735043e-06,
"loss": 1.7559,
"step": 3500
},
{
"epoch": 9.0,
"eval_f1": 0.9086,
"eval_gen_len": 26.424,
"eval_loss": 1.666217565536499,
"eval_precision": 0.9098,
"eval_recall": 0.9078,
"eval_rouge1": 0.4632,
"eval_rouge2": 0.215,
"eval_rougeL": 0.38,
"eval_rougeLsum": 0.3799,
"eval_runtime": 507.5176,
"eval_samples_per_second": 5.419,
"eval_steps_per_second": 0.339,
"step": 3511
},
{
"epoch": 10.0,
"eval_f1": 0.9089,
"eval_gen_len": 26.542545454545454,
"eval_loss": 1.6593670845031738,
"eval_precision": 0.9099,
"eval_recall": 0.9082,
"eval_rouge1": 0.4651,
"eval_rouge2": 0.2168,
"eval_rougeL": 0.3812,
"eval_rougeLsum": 0.3812,
"eval_runtime": 501.1134,
"eval_samples_per_second": 5.488,
"eval_steps_per_second": 0.343,
"step": 3902
},
{
"epoch": 10.25,
"learning_rate": 2.9059829059829063e-06,
"loss": 1.7357,
"step": 4000
},
{
"epoch": 11.0,
"eval_f1": 0.9091,
"eval_gen_len": 26.605090909090908,
"eval_loss": 1.6554700136184692,
"eval_precision": 0.91,
"eval_recall": 0.9086,
"eval_rouge1": 0.4663,
"eval_rouge2": 0.2178,
"eval_rougeL": 0.3824,
"eval_rougeLsum": 0.3823,
"eval_runtime": 504.1076,
"eval_samples_per_second": 5.455,
"eval_steps_per_second": 0.341,
"step": 4293
},
{
"epoch": 11.53,
"learning_rate": 7.692307692307694e-07,
"loss": 1.7297,
"step": 4500
},
{
"epoch": 12.0,
"eval_f1": 0.9092,
"eval_gen_len": 26.439272727272726,
"eval_loss": 1.6508440971374512,
"eval_precision": 0.9103,
"eval_recall": 0.9084,
"eval_rouge1": 0.4668,
"eval_rouge2": 0.2175,
"eval_rougeL": 0.3823,
"eval_rougeLsum": 0.3822,
"eval_runtime": 506.6299,
"eval_samples_per_second": 5.428,
"eval_steps_per_second": 0.339,
"step": 4680
},
{
"epoch": 12.82,
"learning_rate": 3.974358974358974e-06,
"loss": 1.7165,
"step": 5000
},
{
"epoch": 13.0,
"eval_f1": 0.9094,
"eval_gen_len": 26.638545454545454,
"eval_loss": 1.6451035737991333,
"eval_precision": 0.9103,
"eval_recall": 0.9089,
"eval_rouge1": 0.4687,
"eval_rouge2": 0.2191,
"eval_rougeL": 0.3834,
"eval_rougeLsum": 0.3834,
"eval_runtime": 509.0115,
"eval_samples_per_second": 5.403,
"eval_steps_per_second": 0.338,
"step": 5071
},
{
"epoch": 14.0,
"eval_f1": 0.9095,
"eval_gen_len": 26.415636363636363,
"eval_loss": 1.6405242681503296,
"eval_precision": 0.9106,
"eval_recall": 0.9087,
"eval_rouge1": 0.4691,
"eval_rouge2": 0.2193,
"eval_rougeL": 0.3845,
"eval_rougeLsum": 0.3844,
"eval_runtime": 502.7756,
"eval_samples_per_second": 5.47,
"eval_steps_per_second": 0.342,
"step": 5462
},
{
"epoch": 14.1,
"learning_rate": 2.371794871794872e-06,
"loss": 1.7068,
"step": 5500
},
{
"epoch": 15.0,
"eval_f1": 0.9097,
"eval_gen_len": 26.45709090909091,
"eval_loss": 1.6383482217788696,
"eval_precision": 0.9108,
"eval_recall": 0.9089,
"eval_rouge1": 0.4699,
"eval_rouge2": 0.2204,
"eval_rougeL": 0.3853,
"eval_rougeLsum": 0.3853,
"eval_runtime": 506.608,
"eval_samples_per_second": 5.428,
"eval_steps_per_second": 0.34,
"step": 5853
},
{
"epoch": 15.38,
"learning_rate": 7.692307692307694e-07,
"loss": 1.7004,
"step": 6000
},
{
"epoch": 16.0,
"eval_f1": 0.9097,
"eval_gen_len": 26.424727272727274,
"eval_loss": 1.6346381902694702,
"eval_precision": 0.9108,
"eval_recall": 0.9089,
"eval_rouge1": 0.4703,
"eval_rouge2": 0.2204,
"eval_rougeL": 0.385,
"eval_rougeLsum": 0.385,
"eval_runtime": 506.5023,
"eval_samples_per_second": 5.429,
"eval_steps_per_second": 0.34,
"step": 6240
},
{
"epoch": 16.66,
"learning_rate": 3.3333333333333333e-06,
"loss": 1.6923,
"step": 6500
},
{
"epoch": 17.0,
"eval_f1": 0.9099,
"eval_gen_len": 26.443636363636365,
"eval_loss": 1.6305155754089355,
"eval_precision": 0.911,
"eval_recall": 0.9091,
"eval_rouge1": 0.4706,
"eval_rouge2": 0.221,
"eval_rougeL": 0.3855,
"eval_rougeLsum": 0.3856,
"eval_runtime": 506.0984,
"eval_samples_per_second": 5.434,
"eval_steps_per_second": 0.34,
"step": 6631
},
{
"epoch": 17.94,
"learning_rate": 2.0512820512820513e-06,
"loss": 1.6839,
"step": 7000
},
{
"epoch": 18.0,
"eval_f1": 0.9098,
"eval_gen_len": 26.612,
"eval_loss": 1.6285477876663208,
"eval_precision": 0.9106,
"eval_recall": 0.9094,
"eval_rouge1": 0.4712,
"eval_rouge2": 0.2215,
"eval_rougeL": 0.3862,
"eval_rougeLsum": 0.3864,
"eval_runtime": 509.5093,
"eval_samples_per_second": 5.397,
"eval_steps_per_second": 0.338,
"step": 7022
},
{
"epoch": 19.0,
"eval_f1": 0.9099,
"eval_gen_len": 26.529090909090908,
"eval_loss": 1.6263303756713867,
"eval_precision": 0.9108,
"eval_recall": 0.9093,
"eval_rouge1": 0.4709,
"eval_rouge2": 0.2217,
"eval_rougeL": 0.3862,
"eval_rougeLsum": 0.3864,
"eval_runtime": 510.6771,
"eval_samples_per_second": 5.385,
"eval_steps_per_second": 0.337,
"step": 7413
},
{
"epoch": 19.22,
"learning_rate": 7.692307692307694e-07,
"loss": 1.6743,
"step": 7500
},
{
"epoch": 19.99,
"eval_f1": 0.9099,
"eval_gen_len": 26.541090909090908,
"eval_loss": 1.6258454322814941,
"eval_precision": 0.9108,
"eval_recall": 0.9093,
"eval_rouge1": 0.4708,
"eval_rouge2": 0.2214,
"eval_rougeL": 0.3861,
"eval_rougeLsum": 0.3863,
"eval_runtime": 510.6936,
"eval_samples_per_second": 5.385,
"eval_steps_per_second": 0.337,
"step": 7800
},
{
"epoch": 19.99,
"step": 7800,
"total_flos": 1.441981436981674e+18,
"train_loss": 0.3883497776129307,
"train_runtime": 10238.8851,
"train_samples_per_second": 97.667,
"train_steps_per_second": 0.762
}
],
"logging_steps": 500,
"max_steps": 7800,
"num_input_tokens_seen": 0,
"num_train_epochs": 20,
"save_steps": 500,
"total_flos": 1.441981436981674e+18,
"train_batch_size": 32,
"trial_name": null,
"trial_params": null
}