polish-gpt2-small-instruct / trainer_state.json
Aspik101's picture
Hubert commit
15d7bf2
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 20.0,
"global_step": 10000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.5,
"learning_rate": 0.0002954545454545454,
"loss": 0.71,
"step": 250
},
{
"epoch": 1.0,
"learning_rate": 0.00028787878787878786,
"loss": 0.5325,
"step": 500
},
{
"epoch": 1.0,
"eval_loss": 0.5033329725265503,
"eval_runtime": 9.0003,
"eval_samples_per_second": 888.861,
"eval_steps_per_second": 27.777,
"step": 500
},
{
"epoch": 1.5,
"learning_rate": 0.0002803030303030303,
"loss": 0.4322,
"step": 750
},
{
"epoch": 2.0,
"learning_rate": 0.0002727272727272727,
"loss": 0.441,
"step": 1000
},
{
"epoch": 2.0,
"eval_loss": 0.4973089098930359,
"eval_runtime": 9.0771,
"eval_samples_per_second": 881.341,
"eval_steps_per_second": 27.542,
"step": 1000
},
{
"epoch": 2.5,
"learning_rate": 0.0002651515151515151,
"loss": 0.352,
"step": 1250
},
{
"epoch": 3.0,
"learning_rate": 0.00025757575757575756,
"loss": 0.372,
"step": 1500
},
{
"epoch": 3.0,
"eval_loss": 0.5114700198173523,
"eval_runtime": 9.3118,
"eval_samples_per_second": 859.12,
"eval_steps_per_second": 26.848,
"step": 1500
},
{
"epoch": 3.5,
"learning_rate": 0.00025,
"loss": 0.2921,
"step": 1750
},
{
"epoch": 4.0,
"learning_rate": 0.0002424242424242424,
"loss": 0.3107,
"step": 2000
},
{
"epoch": 4.0,
"eval_loss": 0.5385135412216187,
"eval_runtime": 9.7573,
"eval_samples_per_second": 819.896,
"eval_steps_per_second": 25.622,
"step": 2000
},
{
"epoch": 4.5,
"learning_rate": 0.00023484848484848483,
"loss": 0.2439,
"step": 2250
},
{
"epoch": 5.0,
"learning_rate": 0.00022727272727272725,
"loss": 0.2619,
"step": 2500
},
{
"epoch": 5.0,
"eval_loss": 0.5607308149337769,
"eval_runtime": 9.0142,
"eval_samples_per_second": 887.492,
"eval_steps_per_second": 27.734,
"step": 2500
},
{
"epoch": 5.5,
"learning_rate": 0.00021969696969696969,
"loss": 0.2105,
"step": 2750
},
{
"epoch": 6.0,
"learning_rate": 0.0002121212121212121,
"loss": 0.2257,
"step": 3000
},
{
"epoch": 6.0,
"eval_loss": 0.5820582509040833,
"eval_runtime": 22.0785,
"eval_samples_per_second": 362.343,
"eval_steps_per_second": 11.323,
"step": 3000
},
{
"epoch": 6.5,
"learning_rate": 0.0002045454545454545,
"loss": 0.1812,
"step": 3250
},
{
"epoch": 7.0,
"learning_rate": 0.00019696969696969695,
"loss": 0.1947,
"step": 3500
},
{
"epoch": 7.0,
"eval_loss": 0.6063645482063293,
"eval_runtime": 22.1706,
"eval_samples_per_second": 360.838,
"eval_steps_per_second": 11.276,
"step": 3500
},
{
"epoch": 7.5,
"learning_rate": 0.00018939393939393937,
"loss": 0.1597,
"step": 3750
},
{
"epoch": 8.0,
"learning_rate": 0.0001818181818181818,
"loss": 0.1715,
"step": 4000
},
{
"epoch": 8.0,
"eval_loss": 0.6192964911460876,
"eval_runtime": 22.493,
"eval_samples_per_second": 355.666,
"eval_steps_per_second": 11.115,
"step": 4000
},
{
"epoch": 8.5,
"learning_rate": 0.00017424242424242422,
"loss": 0.1439,
"step": 4250
},
{
"epoch": 9.0,
"learning_rate": 0.00016666666666666666,
"loss": 0.1535,
"step": 4500
},
{
"epoch": 9.0,
"eval_loss": 0.6339118480682373,
"eval_runtime": 17.7948,
"eval_samples_per_second": 449.57,
"eval_steps_per_second": 14.049,
"step": 4500
},
{
"epoch": 9.5,
"learning_rate": 0.00015909090909090907,
"loss": 0.1322,
"step": 4750
},
{
"epoch": 10.0,
"learning_rate": 0.00015151515151515152,
"loss": 0.1403,
"step": 5000
},
{
"epoch": 10.0,
"eval_loss": 0.648122251033783,
"eval_runtime": 22.0949,
"eval_samples_per_second": 362.074,
"eval_steps_per_second": 11.315,
"step": 5000
},
{
"epoch": 10.5,
"learning_rate": 0.00014393939393939393,
"loss": 0.1229,
"step": 5250
},
{
"epoch": 11.0,
"learning_rate": 0.00013636363636363634,
"loss": 0.1297,
"step": 5500
},
{
"epoch": 11.0,
"eval_loss": 0.6578399538993835,
"eval_runtime": 22.4161,
"eval_samples_per_second": 356.887,
"eval_steps_per_second": 11.153,
"step": 5500
},
{
"epoch": 11.5,
"learning_rate": 0.00012878787878787878,
"loss": 0.1158,
"step": 5750
},
{
"epoch": 12.0,
"learning_rate": 0.0001212121212121212,
"loss": 0.1217,
"step": 6000
},
{
"epoch": 12.0,
"eval_loss": 0.6666226983070374,
"eval_runtime": 17.8264,
"eval_samples_per_second": 448.773,
"eval_steps_per_second": 14.024,
"step": 6000
},
{
"epoch": 12.5,
"learning_rate": 0.00011363636363636362,
"loss": 0.1106,
"step": 6250
},
{
"epoch": 13.0,
"learning_rate": 0.00010606060606060605,
"loss": 0.1152,
"step": 6500
},
{
"epoch": 13.0,
"eval_loss": 0.6757141351699829,
"eval_runtime": 15.3201,
"eval_samples_per_second": 522.191,
"eval_steps_per_second": 16.318,
"step": 6500
},
{
"epoch": 13.5,
"learning_rate": 9.848484848484848e-05,
"loss": 0.1061,
"step": 6750
},
{
"epoch": 14.0,
"learning_rate": 9.09090909090909e-05,
"loss": 0.1101,
"step": 7000
},
{
"epoch": 14.0,
"eval_loss": 0.687734842300415,
"eval_runtime": 21.8339,
"eval_samples_per_second": 366.402,
"eval_steps_per_second": 11.45,
"step": 7000
},
{
"epoch": 14.5,
"learning_rate": 8.333333333333333e-05,
"loss": 0.1024,
"step": 7250
},
{
"epoch": 15.0,
"learning_rate": 7.575757575757576e-05,
"loss": 0.1059,
"step": 7500
},
{
"epoch": 15.0,
"eval_loss": 0.6898844242095947,
"eval_runtime": 22.4671,
"eval_samples_per_second": 356.076,
"eval_steps_per_second": 11.127,
"step": 7500
},
{
"epoch": 15.5,
"learning_rate": 6.818181818181817e-05,
"loss": 0.0991,
"step": 7750
},
{
"epoch": 16.0,
"learning_rate": 6.06060606060606e-05,
"loss": 0.1022,
"step": 8000
},
{
"epoch": 16.0,
"eval_loss": 0.6962941884994507,
"eval_runtime": 17.8764,
"eval_samples_per_second": 447.518,
"eval_steps_per_second": 13.985,
"step": 8000
},
{
"epoch": 16.5,
"learning_rate": 5.3030303030303025e-05,
"loss": 0.0963,
"step": 8250
},
{
"epoch": 17.0,
"learning_rate": 4.545454545454545e-05,
"loss": 0.0991,
"step": 8500
},
{
"epoch": 17.0,
"eval_loss": 0.702376127243042,
"eval_runtime": 21.7707,
"eval_samples_per_second": 367.466,
"eval_steps_per_second": 11.483,
"step": 8500
},
{
"epoch": 17.5,
"learning_rate": 3.790909090909091e-05,
"loss": 0.0939,
"step": 8750
},
{
"epoch": 18.0,
"learning_rate": 3.033333333333333e-05,
"loss": 0.0961,
"step": 9000
},
{
"epoch": 18.0,
"eval_loss": 0.7053780555725098,
"eval_runtime": 22.1712,
"eval_samples_per_second": 360.829,
"eval_steps_per_second": 11.276,
"step": 9000
},
{
"epoch": 18.5,
"learning_rate": 2.2757575757575757e-05,
"loss": 0.0914,
"step": 9250
},
{
"epoch": 19.0,
"learning_rate": 1.518181818181818e-05,
"loss": 0.0933,
"step": 9500
},
{
"epoch": 19.0,
"eval_loss": 0.7104560732841492,
"eval_runtime": 22.4548,
"eval_samples_per_second": 356.271,
"eval_steps_per_second": 11.133,
"step": 9500
},
{
"epoch": 19.5,
"learning_rate": 7.6060606060606056e-06,
"loss": 0.0893,
"step": 9750
},
{
"epoch": 20.0,
"learning_rate": 3.03030303030303e-08,
"loss": 0.0901,
"step": 10000
},
{
"epoch": 20.0,
"eval_loss": 0.7170030474662781,
"eval_runtime": 17.8822,
"eval_samples_per_second": 447.372,
"eval_steps_per_second": 13.98,
"step": 10000
}
],
"max_steps": 10000,
"num_train_epochs": 20,
"total_flos": 4.180672512e+16,
"trial_name": null,
"trial_params": null
}