test-1_5b / trainer_state.json
Alexandre-Numind's picture
Upload 8 files
7399534 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.820250284414107,
"eval_steps": 200,
"global_step": 5000,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"grad_norm": 1.7517368793487549,
"learning_rate": 9.999e-07,
"loss": 0.2253,
"step": 100
},
{
"epoch": 0.07,
"grad_norm": 1.4760167598724365,
"learning_rate": 9.998e-07,
"loss": 0.1588,
"step": 200
},
{
"epoch": 0.07,
"eval_validation_loss": 0.1540723592042923,
"eval_validation_runtime": 386.0092,
"eval_validation_samples_per_second": 3.0,
"eval_validation_steps_per_second": 0.376,
"step": 200
},
{
"epoch": 0.07,
"eval_benchmark_loss": 0.3658876121044159,
"eval_benchmark_runtime": 9.4993,
"eval_benchmark_samples_per_second": 2.737,
"eval_benchmark_steps_per_second": 0.421,
"step": 200
},
{
"epoch": 0.11,
"grad_norm": 1.4102065563201904,
"learning_rate": 9.997e-07,
"loss": 0.1423,
"step": 300
},
{
"epoch": 0.15,
"grad_norm": 1.657212495803833,
"learning_rate": 9.996e-07,
"loss": 0.1345,
"step": 400
},
{
"epoch": 0.15,
"eval_validation_loss": 0.13775885105133057,
"eval_validation_runtime": 383.1365,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 400
},
{
"epoch": 0.15,
"eval_benchmark_loss": 0.3596523106098175,
"eval_benchmark_runtime": 9.4551,
"eval_benchmark_samples_per_second": 2.75,
"eval_benchmark_steps_per_second": 0.423,
"step": 400
},
{
"epoch": 0.18,
"grad_norm": 1.1109451055526733,
"learning_rate": 9.995e-07,
"loss": 0.1278,
"step": 500
},
{
"epoch": 0.22,
"grad_norm": 1.219743013381958,
"learning_rate": 9.994e-07,
"loss": 0.128,
"step": 600
},
{
"epoch": 0.22,
"eval_validation_loss": 0.1310659945011139,
"eval_validation_runtime": 382.987,
"eval_validation_samples_per_second": 3.024,
"eval_validation_steps_per_second": 0.379,
"step": 600
},
{
"epoch": 0.22,
"eval_benchmark_loss": 0.35413601994514465,
"eval_benchmark_runtime": 9.452,
"eval_benchmark_samples_per_second": 2.751,
"eval_benchmark_steps_per_second": 0.423,
"step": 600
},
{
"epoch": 0.25,
"grad_norm": 1.3631513118743896,
"learning_rate": 9.993e-07,
"loss": 0.1251,
"step": 700
},
{
"epoch": 0.29,
"grad_norm": 1.6779521703720093,
"learning_rate": 9.992e-07,
"loss": 0.1198,
"step": 800
},
{
"epoch": 0.29,
"eval_validation_loss": 0.12629443407058716,
"eval_validation_runtime": 383.2491,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 800
},
{
"epoch": 0.29,
"eval_benchmark_loss": 0.35233384370803833,
"eval_benchmark_runtime": 9.4696,
"eval_benchmark_samples_per_second": 2.746,
"eval_benchmark_steps_per_second": 0.422,
"step": 800
},
{
"epoch": 0.33,
"grad_norm": 2.0499491691589355,
"learning_rate": 9.990999999999999e-07,
"loss": 0.1127,
"step": 900
},
{
"epoch": 0.36,
"grad_norm": 2.0875236988067627,
"learning_rate": 9.989999999999999e-07,
"loss": 0.1227,
"step": 1000
},
{
"epoch": 0.36,
"eval_validation_loss": 0.12267088890075684,
"eval_validation_runtime": 383.4228,
"eval_validation_samples_per_second": 3.02,
"eval_validation_steps_per_second": 0.378,
"step": 1000
},
{
"epoch": 0.36,
"eval_benchmark_loss": 0.35131120681762695,
"eval_benchmark_runtime": 9.4614,
"eval_benchmark_samples_per_second": 2.748,
"eval_benchmark_steps_per_second": 0.423,
"step": 1000
},
{
"epoch": 0.4,
"grad_norm": 2.061170816421509,
"learning_rate": 9.988999999999999e-07,
"loss": 0.1178,
"step": 1100
},
{
"epoch": 0.44,
"grad_norm": 1.5514030456542969,
"learning_rate": 9.988e-07,
"loss": 0.1157,
"step": 1200
},
{
"epoch": 0.44,
"eval_validation_loss": 0.12000161409378052,
"eval_validation_runtime": 383.1507,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 1200
},
{
"epoch": 0.44,
"eval_benchmark_loss": 0.3518659174442291,
"eval_benchmark_runtime": 9.4808,
"eval_benchmark_samples_per_second": 2.742,
"eval_benchmark_steps_per_second": 0.422,
"step": 1200
},
{
"epoch": 0.47,
"grad_norm": 1.9566336870193481,
"learning_rate": 9.987e-07,
"loss": 0.1114,
"step": 1300
},
{
"epoch": 0.51,
"grad_norm": 1.365792155265808,
"learning_rate": 9.986e-07,
"loss": 0.115,
"step": 1400
},
{
"epoch": 0.51,
"eval_validation_loss": 0.117983378469944,
"eval_validation_runtime": 383.1539,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 1400
},
{
"epoch": 0.51,
"eval_benchmark_loss": 0.3509116470813751,
"eval_benchmark_runtime": 9.4708,
"eval_benchmark_samples_per_second": 2.745,
"eval_benchmark_steps_per_second": 0.422,
"step": 1400
},
{
"epoch": 0.55,
"grad_norm": 1.6281362771987915,
"learning_rate": 9.985e-07,
"loss": 0.1192,
"step": 1500
},
{
"epoch": 0.58,
"grad_norm": 1.7361140251159668,
"learning_rate": 9.983999999999998e-07,
"loss": 0.1114,
"step": 1600
},
{
"epoch": 0.58,
"eval_validation_loss": 0.11662287265062332,
"eval_validation_runtime": 383.2086,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 1600
},
{
"epoch": 0.58,
"eval_benchmark_loss": 0.3516782522201538,
"eval_benchmark_runtime": 9.4592,
"eval_benchmark_samples_per_second": 2.749,
"eval_benchmark_steps_per_second": 0.423,
"step": 1600
},
{
"epoch": 0.62,
"grad_norm": 2.332932710647583,
"learning_rate": 9.982999999999998e-07,
"loss": 0.1128,
"step": 1700
},
{
"epoch": 0.66,
"grad_norm": 1.4797985553741455,
"learning_rate": 9.982e-07,
"loss": 0.1036,
"step": 1800
},
{
"epoch": 0.66,
"eval_validation_loss": 0.11438746005296707,
"eval_validation_runtime": 383.1904,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 1800
},
{
"epoch": 0.66,
"eval_benchmark_loss": 0.34705713391304016,
"eval_benchmark_runtime": 9.4646,
"eval_benchmark_samples_per_second": 2.747,
"eval_benchmark_steps_per_second": 0.423,
"step": 1800
},
{
"epoch": 0.69,
"grad_norm": 1.546353816986084,
"learning_rate": 9.981e-07,
"loss": 0.1132,
"step": 1900
},
{
"epoch": 0.73,
"grad_norm": 2.1084446907043457,
"learning_rate": 9.98e-07,
"loss": 0.1125,
"step": 2000
},
{
"epoch": 0.73,
"eval_validation_loss": 0.11280551552772522,
"eval_validation_runtime": 383.2179,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 2000
},
{
"epoch": 0.73,
"eval_benchmark_loss": 0.34782856702804565,
"eval_benchmark_runtime": 9.4577,
"eval_benchmark_samples_per_second": 2.749,
"eval_benchmark_steps_per_second": 0.423,
"step": 2000
},
{
"epoch": 0.76,
"grad_norm": 1.3358793258666992,
"learning_rate": 9.979e-07,
"loss": 0.1155,
"step": 2100
},
{
"epoch": 0.8,
"grad_norm": 1.6584503650665283,
"learning_rate": 9.978e-07,
"loss": 0.1034,
"step": 2200
},
{
"epoch": 0.8,
"eval_validation_loss": 0.11192985624074936,
"eval_validation_runtime": 383.2409,
"eval_validation_samples_per_second": 3.022,
"eval_validation_steps_per_second": 0.378,
"step": 2200
},
{
"epoch": 0.8,
"eval_benchmark_loss": 0.3532721698284149,
"eval_benchmark_runtime": 9.4718,
"eval_benchmark_samples_per_second": 2.745,
"eval_benchmark_steps_per_second": 0.422,
"step": 2200
},
{
"epoch": 0.84,
"grad_norm": 1.2788842916488647,
"learning_rate": 9.977e-07,
"loss": 0.1091,
"step": 2300
},
{
"epoch": 0.87,
"grad_norm": 1.3015437126159668,
"learning_rate": 9.976e-07,
"loss": 0.1024,
"step": 2400
},
{
"epoch": 0.87,
"eval_validation_loss": 0.11032804101705551,
"eval_validation_runtime": 383.3623,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 2400
},
{
"epoch": 0.87,
"eval_benchmark_loss": 0.34818902611732483,
"eval_benchmark_runtime": 9.4633,
"eval_benchmark_samples_per_second": 2.747,
"eval_benchmark_steps_per_second": 0.423,
"step": 2400
},
{
"epoch": 0.91,
"grad_norm": 1.2690401077270508,
"learning_rate": 9.975e-07,
"loss": 0.0981,
"step": 2500
},
{
"epoch": 0.95,
"grad_norm": 1.4752624034881592,
"learning_rate": 9.974e-07,
"loss": 0.1036,
"step": 2600
},
{
"epoch": 0.95,
"eval_validation_loss": 0.10945109277963638,
"eval_validation_runtime": 383.3638,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 2600
},
{
"epoch": 0.95,
"eval_benchmark_loss": 0.3477522134780884,
"eval_benchmark_runtime": 9.4688,
"eval_benchmark_samples_per_second": 2.746,
"eval_benchmark_steps_per_second": 0.422,
"step": 2600
},
{
"epoch": 0.98,
"grad_norm": 1.502576231956482,
"learning_rate": 9.973e-07,
"loss": 0.11,
"step": 2700
},
{
"epoch": 1.02,
"grad_norm": 0.9190167784690857,
"learning_rate": 9.972e-07,
"loss": 0.086,
"step": 2800
},
{
"epoch": 1.02,
"eval_validation_loss": 0.11021959781646729,
"eval_validation_runtime": 383.4523,
"eval_validation_samples_per_second": 3.02,
"eval_validation_steps_per_second": 0.378,
"step": 2800
},
{
"epoch": 1.02,
"eval_benchmark_loss": 0.35620927810668945,
"eval_benchmark_runtime": 9.4715,
"eval_benchmark_samples_per_second": 2.745,
"eval_benchmark_steps_per_second": 0.422,
"step": 2800
},
{
"epoch": 1.06,
"grad_norm": 2.0933382511138916,
"learning_rate": 9.971e-07,
"loss": 0.0849,
"step": 2900
},
{
"epoch": 1.09,
"grad_norm": 1.020569920539856,
"learning_rate": 9.97e-07,
"loss": 0.0858,
"step": 3000
},
{
"epoch": 1.09,
"eval_validation_loss": 0.10909596085548401,
"eval_validation_runtime": 383.3001,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 3000
},
{
"epoch": 1.09,
"eval_benchmark_loss": 0.3547273278236389,
"eval_benchmark_runtime": 9.4578,
"eval_benchmark_samples_per_second": 2.749,
"eval_benchmark_steps_per_second": 0.423,
"step": 3000
},
{
"epoch": 1.13,
"grad_norm": 1.2632180452346802,
"learning_rate": 9.969e-07,
"loss": 0.0834,
"step": 3100
},
{
"epoch": 1.16,
"grad_norm": 1.0163979530334473,
"learning_rate": 9.968e-07,
"loss": 0.0786,
"step": 3200
},
{
"epoch": 1.16,
"eval_validation_loss": 0.11020828038454056,
"eval_validation_runtime": 383.4418,
"eval_validation_samples_per_second": 3.02,
"eval_validation_steps_per_second": 0.378,
"step": 3200
},
{
"epoch": 1.16,
"eval_benchmark_loss": 0.35691550374031067,
"eval_benchmark_runtime": 9.4772,
"eval_benchmark_samples_per_second": 2.743,
"eval_benchmark_steps_per_second": 0.422,
"step": 3200
},
{
"epoch": 1.2,
"grad_norm": 1.0731770992279053,
"learning_rate": 9.967e-07,
"loss": 0.0817,
"step": 3300
},
{
"epoch": 1.24,
"grad_norm": 1.4652243852615356,
"learning_rate": 9.966e-07,
"loss": 0.0885,
"step": 3400
},
{
"epoch": 1.24,
"eval_validation_loss": 0.10934263467788696,
"eval_validation_runtime": 383.3702,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 3400
},
{
"epoch": 1.24,
"eval_benchmark_loss": 0.35489001870155334,
"eval_benchmark_runtime": 9.4613,
"eval_benchmark_samples_per_second": 2.748,
"eval_benchmark_steps_per_second": 0.423,
"step": 3400
},
{
"epoch": 1.27,
"grad_norm": 1.1992924213409424,
"learning_rate": 9.965e-07,
"loss": 0.078,
"step": 3500
},
{
"epoch": 1.31,
"grad_norm": 1.3917715549468994,
"learning_rate": 9.964e-07,
"loss": 0.0857,
"step": 3600
},
{
"epoch": 1.31,
"eval_validation_loss": 0.10760623216629028,
"eval_validation_runtime": 383.3793,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 3600
},
{
"epoch": 1.31,
"eval_benchmark_loss": 0.35664084553718567,
"eval_benchmark_runtime": 9.4529,
"eval_benchmark_samples_per_second": 2.75,
"eval_benchmark_steps_per_second": 0.423,
"step": 3600
},
{
"epoch": 1.35,
"grad_norm": 0.9880099892616272,
"learning_rate": 9.962999999999999e-07,
"loss": 0.084,
"step": 3700
},
{
"epoch": 1.38,
"grad_norm": 1.15889310836792,
"learning_rate": 9.961999999999999e-07,
"loss": 0.0833,
"step": 3800
},
{
"epoch": 1.38,
"eval_validation_loss": 0.10785996913909912,
"eval_validation_runtime": 383.3503,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 3800
},
{
"epoch": 1.38,
"eval_benchmark_loss": 0.3560557961463928,
"eval_benchmark_runtime": 9.457,
"eval_benchmark_samples_per_second": 2.749,
"eval_benchmark_steps_per_second": 0.423,
"step": 3800
},
{
"epoch": 1.42,
"grad_norm": 1.4265196323394775,
"learning_rate": 9.960999999999999e-07,
"loss": 0.0831,
"step": 3900
},
{
"epoch": 1.46,
"grad_norm": 1.4567248821258545,
"learning_rate": 9.959999999999999e-07,
"loss": 0.084,
"step": 4000
},
{
"epoch": 1.46,
"eval_validation_loss": 0.1076769158244133,
"eval_validation_runtime": 383.3556,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 4000
},
{
"epoch": 1.46,
"eval_benchmark_loss": 0.3554157614707947,
"eval_benchmark_runtime": 9.4598,
"eval_benchmark_samples_per_second": 2.748,
"eval_benchmark_steps_per_second": 0.423,
"step": 4000
},
{
"epoch": 1.49,
"grad_norm": 1.7788864374160767,
"learning_rate": 9.958999999999999e-07,
"loss": 0.0813,
"step": 4100
},
{
"epoch": 1.53,
"grad_norm": 1.4983494281768799,
"learning_rate": 9.958e-07,
"loss": 0.0815,
"step": 4200
},
{
"epoch": 1.53,
"eval_validation_loss": 0.10664806514978409,
"eval_validation_runtime": 383.2552,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 4200
},
{
"epoch": 1.53,
"eval_benchmark_loss": 0.35523322224617004,
"eval_benchmark_runtime": 9.4557,
"eval_benchmark_samples_per_second": 2.75,
"eval_benchmark_steps_per_second": 0.423,
"step": 4200
},
{
"epoch": 1.57,
"grad_norm": 0.7813511490821838,
"learning_rate": 9.957e-07,
"loss": 0.0877,
"step": 4300
},
{
"epoch": 1.6,
"grad_norm": 1.3383861780166626,
"learning_rate": 9.956e-07,
"loss": 0.0812,
"step": 4400
},
{
"epoch": 1.6,
"eval_validation_loss": 0.10593847930431366,
"eval_validation_runtime": 383.4244,
"eval_validation_samples_per_second": 3.02,
"eval_validation_steps_per_second": 0.378,
"step": 4400
},
{
"epoch": 1.6,
"eval_benchmark_loss": 0.3536509871482849,
"eval_benchmark_runtime": 9.4751,
"eval_benchmark_samples_per_second": 2.744,
"eval_benchmark_steps_per_second": 0.422,
"step": 4400
},
{
"epoch": 1.64,
"grad_norm": 1.5067020654678345,
"learning_rate": 9.955e-07,
"loss": 0.0821,
"step": 4500
},
{
"epoch": 1.67,
"grad_norm": 1.46514093875885,
"learning_rate": 9.953999999999998e-07,
"loss": 0.0841,
"step": 4600
},
{
"epoch": 1.67,
"eval_validation_loss": 0.10573519766330719,
"eval_validation_runtime": 383.3608,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 4600
},
{
"epoch": 1.67,
"eval_benchmark_loss": 0.35240989923477173,
"eval_benchmark_runtime": 9.4682,
"eval_benchmark_samples_per_second": 2.746,
"eval_benchmark_steps_per_second": 0.422,
"step": 4600
},
{
"epoch": 1.71,
"grad_norm": 1.124515175819397,
"learning_rate": 9.952999999999998e-07,
"loss": 0.0833,
"step": 4700
},
{
"epoch": 1.75,
"grad_norm": 1.2566176652908325,
"learning_rate": 9.952e-07,
"loss": 0.0822,
"step": 4800
},
{
"epoch": 1.75,
"eval_validation_loss": 0.10490844398736954,
"eval_validation_runtime": 383.3754,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 4800
},
{
"epoch": 1.75,
"eval_benchmark_loss": 0.3501981496810913,
"eval_benchmark_runtime": 9.4492,
"eval_benchmark_samples_per_second": 2.752,
"eval_benchmark_steps_per_second": 0.423,
"step": 4800
},
{
"epoch": 1.78,
"grad_norm": 1.4075120687484741,
"learning_rate": 9.951e-07,
"loss": 0.0824,
"step": 4900
},
{
"epoch": 1.82,
"grad_norm": 0.9538173079490662,
"learning_rate": 9.95e-07,
"loss": 0.0809,
"step": 5000
},
{
"epoch": 1.82,
"eval_validation_loss": 0.10531202703714371,
"eval_validation_runtime": 383.2794,
"eval_validation_samples_per_second": 3.021,
"eval_validation_steps_per_second": 0.378,
"step": 5000
},
{
"epoch": 1.82,
"eval_benchmark_loss": 0.3513197898864746,
"eval_benchmark_runtime": 9.4616,
"eval_benchmark_samples_per_second": 2.748,
"eval_benchmark_steps_per_second": 0.423,
"step": 5000
}
],
"logging_steps": 100,
"max_steps": 1000000,
"num_input_tokens_seen": 0,
"num_train_epochs": 365,
"save_steps": 200,
"total_flos": 1.617395756875776e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}