wu5onsx2 / checkpoint-500 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
aad4197 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.8,
"eval_steps": 100,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"entropy": 0.7081545192748309,
"epoch": 0.016,
"grad_norm": 119.5,
"learning_rate": 6.000000000000001e-07,
"loss": 30.4211,
"mean_token_accuracy": 0.7030085142701864,
"num_tokens": 338352.0,
"step": 10
},
{
"entropy": 0.7043411549180746,
"epoch": 0.032,
"grad_norm": 111.5,
"learning_rate": 1.2666666666666669e-06,
"loss": 30.1214,
"mean_token_accuracy": 0.7080705553293228,
"num_tokens": 671193.0,
"step": 20
},
{
"entropy": 0.7044402219355106,
"epoch": 0.048,
"grad_norm": 106.0,
"learning_rate": 1.9333333333333336e-06,
"loss": 29.7341,
"mean_token_accuracy": 0.7064134150743484,
"num_tokens": 1001008.0,
"step": 30
},
{
"entropy": 0.733870861120522,
"epoch": 0.064,
"grad_norm": 96.0,
"learning_rate": 2.6e-06,
"loss": 30.1494,
"mean_token_accuracy": 0.7005880128592252,
"num_tokens": 1331208.0,
"step": 40
},
{
"entropy": 0.7398215295746923,
"epoch": 0.08,
"grad_norm": 90.5,
"learning_rate": 3.266666666666667e-06,
"loss": 28.7202,
"mean_token_accuracy": 0.7051243670284748,
"num_tokens": 1667532.0,
"step": 50
},
{
"entropy": 0.738157893717289,
"epoch": 0.096,
"grad_norm": 76.5,
"learning_rate": 3.9333333333333335e-06,
"loss": 27.1633,
"mean_token_accuracy": 0.71303277797997,
"num_tokens": 2007176.0,
"step": 60
},
{
"entropy": 0.8131938450038433,
"epoch": 0.112,
"grad_norm": 57.5,
"learning_rate": 4.600000000000001e-06,
"loss": 26.4428,
"mean_token_accuracy": 0.7120848346501589,
"num_tokens": 2342259.0,
"step": 70
},
{
"entropy": 0.9618701986968518,
"epoch": 0.128,
"grad_norm": 43.75,
"learning_rate": 5.2666666666666665e-06,
"loss": 25.9908,
"mean_token_accuracy": 0.7077770136296749,
"num_tokens": 2663987.0,
"step": 80
},
{
"entropy": 1.0031472396105527,
"epoch": 0.144,
"grad_norm": 30.125,
"learning_rate": 5.933333333333335e-06,
"loss": 23.3524,
"mean_token_accuracy": 0.7268724206835031,
"num_tokens": 2997824.0,
"step": 90
},
{
"entropy": 1.0786659345030785,
"epoch": 0.16,
"grad_norm": 20.75,
"learning_rate": 6.600000000000001e-06,
"loss": 21.8746,
"mean_token_accuracy": 0.7358953565359115,
"num_tokens": 3330597.0,
"step": 100
},
{
"epoch": 0.16,
"eval_biology_entropy": 3.260982286453247,
"eval_biology_loss": 3.7093346118927,
"eval_biology_mean_token_accuracy": 0.46263515305519104,
"eval_biology_num_tokens": 3330597.0,
"eval_biology_runtime": 57.0058,
"eval_biology_samples_per_second": 8.771,
"eval_biology_steps_per_second": 2.193,
"step": 100
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 1.6407185397148132,
"eval_chemistry_loss": 1.7927597761154175,
"eval_chemistry_mean_token_accuracy": 0.677183347940445,
"eval_chemistry_num_tokens": 3330597.0,
"eval_chemistry_runtime": 71.6705,
"eval_chemistry_samples_per_second": 6.976,
"eval_chemistry_steps_per_second": 1.744,
"step": 100
},
{
"epoch": 0.16,
"eval_math_entropy": 0.9620108480453491,
"eval_math_loss": 1.4342334270477295,
"eval_math_mean_token_accuracy": 0.7288489332199096,
"eval_math_num_tokens": 3330597.0,
"eval_math_runtime": 74.2141,
"eval_math_samples_per_second": 6.737,
"eval_math_steps_per_second": 1.684,
"step": 100
},
{
"epoch": 0.16,
"eval_physics_entropy": 1.1280243759155273,
"eval_physics_loss": 1.3446351289749146,
"eval_physics_mean_token_accuracy": 0.7332401766777038,
"eval_physics_num_tokens": 3330597.0,
"eval_physics_runtime": 84.9827,
"eval_physics_samples_per_second": 5.884,
"eval_physics_steps_per_second": 1.471,
"step": 100
},
{
"entropy": 1.1533773414790631,
"epoch": 0.176,
"grad_norm": 19.75,
"learning_rate": 7.266666666666668e-06,
"loss": 21.3285,
"mean_token_accuracy": 0.7356826655566693,
"num_tokens": 3658264.0,
"step": 110
},
{
"entropy": 1.112282781302929,
"epoch": 0.192,
"grad_norm": 12.625,
"learning_rate": 7.933333333333334e-06,
"loss": 20.4084,
"mean_token_accuracy": 0.7426700782030821,
"num_tokens": 3995568.0,
"step": 120
},
{
"entropy": 1.086438114568591,
"epoch": 0.208,
"grad_norm": 13.875,
"learning_rate": 8.6e-06,
"loss": 19.3962,
"mean_token_accuracy": 0.7525120593607426,
"num_tokens": 4321436.0,
"step": 130
},
{
"entropy": 1.089601282030344,
"epoch": 0.224,
"grad_norm": 11.8125,
"learning_rate": 9.266666666666667e-06,
"loss": 19.0472,
"mean_token_accuracy": 0.7530231148004531,
"num_tokens": 4648491.0,
"step": 140
},
{
"entropy": 1.0804125092923642,
"epoch": 0.24,
"grad_norm": 11.8125,
"learning_rate": 9.933333333333334e-06,
"loss": 18.7739,
"mean_token_accuracy": 0.7554640270769596,
"num_tokens": 4986175.0,
"step": 150
},
{
"entropy": 1.0836318269371987,
"epoch": 0.256,
"grad_norm": 11.9375,
"learning_rate": 1.0600000000000002e-05,
"loss": 18.4704,
"mean_token_accuracy": 0.7584241010248661,
"num_tokens": 5329320.0,
"step": 160
},
{
"entropy": 1.0882270745933056,
"epoch": 0.272,
"grad_norm": 13.375,
"learning_rate": 1.1266666666666668e-05,
"loss": 18.6237,
"mean_token_accuracy": 0.754904105886817,
"num_tokens": 5658796.0,
"step": 170
},
{
"entropy": 1.076054659858346,
"epoch": 0.288,
"grad_norm": 11.0625,
"learning_rate": 1.1933333333333335e-05,
"loss": 18.4139,
"mean_token_accuracy": 0.7581049785017967,
"num_tokens": 5980201.0,
"step": 180
},
{
"entropy": 1.072243456915021,
"epoch": 0.304,
"grad_norm": 15.5,
"learning_rate": 1.2600000000000001e-05,
"loss": 18.1532,
"mean_token_accuracy": 0.7589787419885397,
"num_tokens": 6298858.0,
"step": 190
},
{
"entropy": 1.0509168311953545,
"epoch": 0.32,
"grad_norm": 12.0,
"learning_rate": 1.3266666666666668e-05,
"loss": 17.8979,
"mean_token_accuracy": 0.7612847778946161,
"num_tokens": 6622798.0,
"step": 200
},
{
"epoch": 0.32,
"eval_biology_entropy": 2.7866374349594114,
"eval_biology_loss": 3.2049155235290527,
"eval_biology_mean_token_accuracy": 0.502942953824997,
"eval_biology_num_tokens": 6622798.0,
"eval_biology_runtime": 56.7587,
"eval_biology_samples_per_second": 8.809,
"eval_biology_steps_per_second": 2.202,
"step": 200
},
{
"epoch": 0.32,
"eval_chemistry_entropy": 1.41602649641037,
"eval_chemistry_loss": 1.4916778802871704,
"eval_chemistry_mean_token_accuracy": 0.7079435839653015,
"eval_chemistry_num_tokens": 6622798.0,
"eval_chemistry_runtime": 71.0228,
"eval_chemistry_samples_per_second": 7.04,
"eval_chemistry_steps_per_second": 1.76,
"step": 200
},
{
"epoch": 0.32,
"eval_math_entropy": 0.9821910600662231,
"eval_math_loss": 1.256189227104187,
"eval_math_mean_token_accuracy": 0.7425830450057983,
"eval_math_num_tokens": 6622798.0,
"eval_math_runtime": 75.4477,
"eval_math_samples_per_second": 6.627,
"eval_math_steps_per_second": 1.657,
"step": 200
},
{
"epoch": 0.32,
"eval_physics_entropy": 1.0466554942131043,
"eval_physics_loss": 1.1129223108291626,
"eval_physics_mean_token_accuracy": 0.7617223796844482,
"eval_physics_num_tokens": 6622798.0,
"eval_physics_runtime": 85.3386,
"eval_physics_samples_per_second": 5.859,
"eval_physics_steps_per_second": 1.465,
"step": 200
},
{
"entropy": 1.0462193094193935,
"epoch": 0.336,
"grad_norm": 11.875,
"learning_rate": 1.3933333333333334e-05,
"loss": 17.7296,
"mean_token_accuracy": 0.7623197592794895,
"num_tokens": 6953611.0,
"step": 210
},
{
"entropy": 1.0164767142385245,
"epoch": 0.352,
"grad_norm": 11.25,
"learning_rate": 1.46e-05,
"loss": 17.1525,
"mean_token_accuracy": 0.7695928748697043,
"num_tokens": 7290378.0,
"step": 220
},
{
"entropy": 0.9987690806388855,
"epoch": 0.368,
"grad_norm": 11.4375,
"learning_rate": 1.5266666666666667e-05,
"loss": 16.9228,
"mean_token_accuracy": 0.7714470315724611,
"num_tokens": 7621082.0,
"step": 230
},
{
"entropy": 1.0506686847656965,
"epoch": 0.384,
"grad_norm": 11.3125,
"learning_rate": 1.5933333333333336e-05,
"loss": 17.6416,
"mean_token_accuracy": 0.7599896423518657,
"num_tokens": 7955570.0,
"step": 240
},
{
"entropy": 1.0135670874267817,
"epoch": 0.4,
"grad_norm": 14.5,
"learning_rate": 1.66e-05,
"loss": 16.9698,
"mean_token_accuracy": 0.7687441952526569,
"num_tokens": 8291049.0,
"step": 250
},
{
"entropy": 0.9876859273761511,
"epoch": 0.416,
"grad_norm": 13.75,
"learning_rate": 1.726666666666667e-05,
"loss": 16.6909,
"mean_token_accuracy": 0.7725014701485634,
"num_tokens": 8627310.0,
"step": 260
},
{
"entropy": 1.0216444052755833,
"epoch": 0.432,
"grad_norm": 12.125,
"learning_rate": 1.7933333333333333e-05,
"loss": 17.011,
"mean_token_accuracy": 0.7689431305974722,
"num_tokens": 8958371.0,
"step": 270
},
{
"entropy": 0.9649551097303629,
"epoch": 0.448,
"grad_norm": 12.1875,
"learning_rate": 1.86e-05,
"loss": 16.2252,
"mean_token_accuracy": 0.7780237648636102,
"num_tokens": 9294388.0,
"step": 280
},
{
"entropy": 0.9653072291985154,
"epoch": 0.464,
"grad_norm": 12.5,
"learning_rate": 1.926666666666667e-05,
"loss": 16.1144,
"mean_token_accuracy": 0.7772431872785092,
"num_tokens": 9620366.0,
"step": 290
},
{
"entropy": 0.9695413928478956,
"epoch": 0.48,
"grad_norm": 11.8125,
"learning_rate": 1.9933333333333334e-05,
"loss": 16.4447,
"mean_token_accuracy": 0.7733453687280416,
"num_tokens": 9955431.0,
"step": 300
},
{
"epoch": 0.48,
"eval_biology_entropy": 2.7462952222824097,
"eval_biology_loss": 3.060053825378418,
"eval_biology_mean_token_accuracy": 0.5114872539043427,
"eval_biology_num_tokens": 9955431.0,
"eval_biology_runtime": 56.9805,
"eval_biology_samples_per_second": 8.775,
"eval_biology_steps_per_second": 2.194,
"step": 300
},
{
"epoch": 0.48,
"eval_chemistry_entropy": 1.3710836601257324,
"eval_chemistry_loss": 1.4066872596740723,
"eval_chemistry_mean_token_accuracy": 0.7168692064285278,
"eval_chemistry_num_tokens": 9955431.0,
"eval_chemistry_runtime": 71.3717,
"eval_chemistry_samples_per_second": 7.006,
"eval_chemistry_steps_per_second": 1.751,
"step": 300
},
{
"epoch": 0.48,
"eval_math_entropy": 0.9784291224479675,
"eval_math_loss": 1.2082456350326538,
"eval_math_mean_token_accuracy": 0.7478420066833497,
"eval_math_num_tokens": 9955431.0,
"eval_math_runtime": 74.3706,
"eval_math_samples_per_second": 6.723,
"eval_math_steps_per_second": 1.681,
"step": 300
},
{
"epoch": 0.48,
"eval_physics_entropy": 1.007008915424347,
"eval_physics_loss": 1.0401662588119507,
"eval_physics_mean_token_accuracy": 0.7707204632759094,
"eval_physics_num_tokens": 9955431.0,
"eval_physics_runtime": 84.6939,
"eval_physics_samples_per_second": 5.904,
"eval_physics_steps_per_second": 1.476,
"step": 300
},
{
"entropy": 1.007104156538844,
"epoch": 0.496,
"grad_norm": 10.5,
"learning_rate": 1.9933333333333334e-05,
"loss": 16.6639,
"mean_token_accuracy": 0.771253039687872,
"num_tokens": 10289883.0,
"step": 310
},
{
"entropy": 0.9723599992692471,
"epoch": 0.512,
"grad_norm": 12.4375,
"learning_rate": 1.985925925925926e-05,
"loss": 16.4387,
"mean_token_accuracy": 0.7748051699250936,
"num_tokens": 10619468.0,
"step": 320
},
{
"entropy": 0.9979474730789661,
"epoch": 0.528,
"grad_norm": 10.8125,
"learning_rate": 1.9785185185185187e-05,
"loss": 16.6398,
"mean_token_accuracy": 0.7712166555225849,
"num_tokens": 10944904.0,
"step": 330
},
{
"entropy": 0.9800913408398628,
"epoch": 0.544,
"grad_norm": 11.625,
"learning_rate": 1.971111111111111e-05,
"loss": 16.374,
"mean_token_accuracy": 0.7737775973975658,
"num_tokens": 11266409.0,
"step": 340
},
{
"entropy": 0.9599094491451978,
"epoch": 0.56,
"grad_norm": 10.625,
"learning_rate": 1.963703703703704e-05,
"loss": 16.0194,
"mean_token_accuracy": 0.778597004711628,
"num_tokens": 11605544.0,
"step": 350
},
{
"entropy": 0.948445113003254,
"epoch": 0.576,
"grad_norm": 11.75,
"learning_rate": 1.9562962962962964e-05,
"loss": 15.9791,
"mean_token_accuracy": 0.7805619373917579,
"num_tokens": 11931715.0,
"step": 360
},
{
"entropy": 0.9618662863969802,
"epoch": 0.592,
"grad_norm": 13.0625,
"learning_rate": 1.948888888888889e-05,
"loss": 16.0713,
"mean_token_accuracy": 0.776887471601367,
"num_tokens": 12257243.0,
"step": 370
},
{
"entropy": 0.9480277199298144,
"epoch": 0.608,
"grad_norm": 10.8125,
"learning_rate": 1.9414814814814817e-05,
"loss": 15.8656,
"mean_token_accuracy": 0.7797395702451467,
"num_tokens": 12583154.0,
"step": 380
},
{
"entropy": 0.9716805059462785,
"epoch": 0.624,
"grad_norm": 10.625,
"learning_rate": 1.9340740740740743e-05,
"loss": 16.1886,
"mean_token_accuracy": 0.7762768242508173,
"num_tokens": 12905392.0,
"step": 390
},
{
"entropy": 0.9691856369376183,
"epoch": 0.64,
"grad_norm": 11.8125,
"learning_rate": 1.926666666666667e-05,
"loss": 16.2314,
"mean_token_accuracy": 0.7761619180440903,
"num_tokens": 13232198.0,
"step": 400
},
{
"epoch": 0.64,
"eval_biology_entropy": 2.67273916721344,
"eval_biology_loss": 2.96358585357666,
"eval_biology_mean_token_accuracy": 0.5216054661273957,
"eval_biology_num_tokens": 13232198.0,
"eval_biology_runtime": 57.8851,
"eval_biology_samples_per_second": 8.638,
"eval_biology_steps_per_second": 2.159,
"step": 400
},
{
"epoch": 0.64,
"eval_chemistry_entropy": 1.3173069462776184,
"eval_chemistry_loss": 1.36112380027771,
"eval_chemistry_mean_token_accuracy": 0.7232093935012818,
"eval_chemistry_num_tokens": 13232198.0,
"eval_chemistry_runtime": 71.8687,
"eval_chemistry_samples_per_second": 6.957,
"eval_chemistry_steps_per_second": 1.739,
"step": 400
},
{
"epoch": 0.64,
"eval_math_entropy": 0.9528153438568115,
"eval_math_loss": 1.1841319799423218,
"eval_math_mean_token_accuracy": 0.7507999067306519,
"eval_math_num_tokens": 13232198.0,
"eval_math_runtime": 74.8756,
"eval_math_samples_per_second": 6.678,
"eval_math_steps_per_second": 1.669,
"step": 400
},
{
"epoch": 0.64,
"eval_physics_entropy": 0.9614342203140259,
"eval_physics_loss": 1.0016775131225586,
"eval_physics_mean_token_accuracy": 0.7765112986564636,
"eval_physics_num_tokens": 13232198.0,
"eval_physics_runtime": 85.1175,
"eval_physics_samples_per_second": 5.874,
"eval_physics_steps_per_second": 1.469,
"step": 400
},
{
"entropy": 0.9530647125095129,
"epoch": 0.656,
"grad_norm": 11.375,
"learning_rate": 1.9192592592592593e-05,
"loss": 15.9429,
"mean_token_accuracy": 0.7780249908566474,
"num_tokens": 13575902.0,
"step": 410
},
{
"entropy": 0.9399624351412058,
"epoch": 0.672,
"grad_norm": 12.0625,
"learning_rate": 1.911851851851852e-05,
"loss": 15.7284,
"mean_token_accuracy": 0.7817466016858816,
"num_tokens": 13895997.0,
"step": 420
},
{
"entropy": 0.9540142439305782,
"epoch": 0.688,
"grad_norm": 10.6875,
"learning_rate": 1.9044444444444446e-05,
"loss": 15.7763,
"mean_token_accuracy": 0.7817316662520171,
"num_tokens": 14234888.0,
"step": 430
},
{
"entropy": 0.9538515221327544,
"epoch": 0.704,
"grad_norm": 11.625,
"learning_rate": 1.8970370370370372e-05,
"loss": 16.0838,
"mean_token_accuracy": 0.775813952833414,
"num_tokens": 14567908.0,
"step": 440
},
{
"entropy": 0.9464580094441771,
"epoch": 0.72,
"grad_norm": 12.375,
"learning_rate": 1.8896296296296295e-05,
"loss": 15.704,
"mean_token_accuracy": 0.7821726374328136,
"num_tokens": 14882927.0,
"step": 450
},
{
"entropy": 0.9363573594018817,
"epoch": 0.736,
"grad_norm": 13.0625,
"learning_rate": 1.8822222222222225e-05,
"loss": 15.7075,
"mean_token_accuracy": 0.7809599131345749,
"num_tokens": 15217342.0,
"step": 460
},
{
"entropy": 0.9193624388426542,
"epoch": 0.752,
"grad_norm": 10.4375,
"learning_rate": 1.874814814814815e-05,
"loss": 15.3677,
"mean_token_accuracy": 0.7846659943461418,
"num_tokens": 15547710.0,
"step": 470
},
{
"entropy": 0.9613647162914276,
"epoch": 0.768,
"grad_norm": 11.1875,
"learning_rate": 1.8674074074074075e-05,
"loss": 16.0934,
"mean_token_accuracy": 0.7759233452379704,
"num_tokens": 15877177.0,
"step": 480
},
{
"entropy": 0.9139260027557612,
"epoch": 0.784,
"grad_norm": 10.1875,
"learning_rate": 1.86e-05,
"loss": 15.3216,
"mean_token_accuracy": 0.786144433543086,
"num_tokens": 16219640.0,
"step": 490
},
{
"entropy": 0.9349881060421467,
"epoch": 0.8,
"grad_norm": 12.0,
"learning_rate": 1.8525925925925928e-05,
"loss": 15.6534,
"mean_token_accuracy": 0.7819662269204855,
"num_tokens": 16548261.0,
"step": 500
},
{
"epoch": 0.8,
"eval_biology_entropy": 2.617759398460388,
"eval_biology_loss": 2.9445557594299316,
"eval_biology_mean_token_accuracy": 0.5238966343402862,
"eval_biology_num_tokens": 16548261.0,
"eval_biology_runtime": 57.6905,
"eval_biology_samples_per_second": 8.667,
"eval_biology_steps_per_second": 2.167,
"step": 500
},
{
"epoch": 0.8,
"eval_chemistry_entropy": 1.2937732858657838,
"eval_chemistry_loss": 1.3414490222930908,
"eval_chemistry_mean_token_accuracy": 0.7248484263420105,
"eval_chemistry_num_tokens": 16548261.0,
"eval_chemistry_runtime": 71.1659,
"eval_chemistry_samples_per_second": 7.026,
"eval_chemistry_steps_per_second": 1.756,
"step": 500
},
{
"epoch": 0.8,
"eval_math_entropy": 0.93889684009552,
"eval_math_loss": 1.17385995388031,
"eval_math_mean_token_accuracy": 0.7522515163421631,
"eval_math_num_tokens": 16548261.0,
"eval_math_runtime": 74.7142,
"eval_math_samples_per_second": 6.692,
"eval_math_steps_per_second": 1.673,
"step": 500
},
{
"epoch": 0.8,
"eval_physics_entropy": 0.9420124840736389,
"eval_physics_loss": 0.9809063673019409,
"eval_physics_mean_token_accuracy": 0.7797514395713806,
"eval_physics_num_tokens": 16548261.0,
"eval_physics_runtime": 84.5204,
"eval_physics_samples_per_second": 5.916,
"eval_physics_steps_per_second": 1.479,
"step": 500
}
],
"logging_steps": 10,
"max_steps": 3000,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 1.652179086470234e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}