37c1mxud / checkpoint-1500 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
51aee51 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.4,
"eval_steps": 100,
"global_step": 1500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"entropy": 1.2553194094449283,
"epoch": 0.016,
"grad_norm": 270.0,
"learning_rate": 6.000000000000001e-07,
"loss": 41.3914,
"mean_token_accuracy": 0.5671123253181577,
"num_tokens": 338352.0,
"step": 10
},
{
"entropy": 1.266689160466194,
"epoch": 0.032,
"grad_norm": 264.0,
"learning_rate": 1.2666666666666669e-06,
"loss": 41.0628,
"mean_token_accuracy": 0.5705239847302437,
"num_tokens": 671193.0,
"step": 20
},
{
"entropy": 1.3143046092242003,
"epoch": 0.048,
"grad_norm": 210.0,
"learning_rate": 1.9333333333333336e-06,
"loss": 39.4998,
"mean_token_accuracy": 0.5745313983410597,
"num_tokens": 1001008.0,
"step": 30
},
{
"entropy": 1.4753582045435905,
"epoch": 0.064,
"grad_norm": 155.0,
"learning_rate": 2.6e-06,
"loss": 38.0568,
"mean_token_accuracy": 0.5745376290753483,
"num_tokens": 1331208.0,
"step": 40
},
{
"entropy": 1.6769672967493534,
"epoch": 0.08,
"grad_norm": 99.5,
"learning_rate": 3.266666666666667e-06,
"loss": 34.3871,
"mean_token_accuracy": 0.5940070981159806,
"num_tokens": 1667532.0,
"step": 50
},
{
"entropy": 1.7719040885567665,
"epoch": 0.096,
"grad_norm": 55.25,
"learning_rate": 3.9333333333333335e-06,
"loss": 30.2468,
"mean_token_accuracy": 0.6289831601083279,
"num_tokens": 2007176.0,
"step": 60
},
{
"entropy": 1.7813668169081212,
"epoch": 0.112,
"grad_norm": 45.0,
"learning_rate": 4.600000000000001e-06,
"loss": 28.6054,
"mean_token_accuracy": 0.6419326957315207,
"num_tokens": 2342259.0,
"step": 70
},
{
"entropy": 1.7141680859029294,
"epoch": 0.128,
"grad_norm": 36.75,
"learning_rate": 5.2666666666666665e-06,
"loss": 27.4799,
"mean_token_accuracy": 0.6465466242283583,
"num_tokens": 2663987.0,
"step": 80
},
{
"entropy": 1.5760046038776636,
"epoch": 0.144,
"grad_norm": 26.875,
"learning_rate": 5.933333333333335e-06,
"loss": 24.8601,
"mean_token_accuracy": 0.6713848907500506,
"num_tokens": 2997824.0,
"step": 90
},
{
"entropy": 1.51325889416039,
"epoch": 0.16,
"grad_norm": 24.625,
"learning_rate": 6.600000000000001e-06,
"loss": 23.6713,
"mean_token_accuracy": 0.6833302665501833,
"num_tokens": 3330597.0,
"step": 100
},
{
"epoch": 0.16,
"eval_biology_entropy": 3.1857855548858645,
"eval_biology_loss": 4.065364837646484,
"eval_biology_mean_token_accuracy": 0.4020989770889282,
"eval_biology_num_tokens": 3330597.0,
"eval_biology_runtime": 49.9004,
"eval_biology_samples_per_second": 10.02,
"eval_biology_steps_per_second": 2.505,
"step": 100
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 1.8894153318405151,
"eval_chemistry_loss": 1.931850790977478,
"eval_chemistry_mean_token_accuracy": 0.6162000632286072,
"eval_chemistry_num_tokens": 3330597.0,
"eval_chemistry_runtime": 61.8293,
"eval_chemistry_samples_per_second": 8.087,
"eval_chemistry_steps_per_second": 2.022,
"step": 100
},
{
"epoch": 0.16,
"eval_math_entropy": 1.3163333106040955,
"eval_math_loss": 1.5677636861801147,
"eval_math_mean_token_accuracy": 0.6788510384559632,
"eval_math_num_tokens": 3330597.0,
"eval_math_runtime": 63.4108,
"eval_math_samples_per_second": 7.885,
"eval_math_steps_per_second": 1.971,
"step": 100
},
{
"epoch": 0.16,
"eval_physics_entropy": 1.495229031085968,
"eval_physics_loss": 1.4545849561691284,
"eval_physics_mean_token_accuracy": 0.6851646456718444,
"eval_physics_num_tokens": 3330597.0,
"eval_physics_runtime": 72.2177,
"eval_physics_samples_per_second": 6.924,
"eval_physics_steps_per_second": 1.731,
"step": 100
},
{
"entropy": 1.473308515548706,
"epoch": 0.176,
"grad_norm": 25.375,
"learning_rate": 7.266666666666668e-06,
"loss": 22.9019,
"mean_token_accuracy": 0.6873474020510912,
"num_tokens": 3658264.0,
"step": 110
},
{
"entropy": 1.397306866571307,
"epoch": 0.192,
"grad_norm": 19.625,
"learning_rate": 7.933333333333334e-06,
"loss": 21.9472,
"mean_token_accuracy": 0.6935743600130081,
"num_tokens": 3995568.0,
"step": 120
},
{
"entropy": 1.3264490943402052,
"epoch": 0.208,
"grad_norm": 20.5,
"learning_rate": 8.6e-06,
"loss": 20.7584,
"mean_token_accuracy": 0.7065585050731897,
"num_tokens": 4321436.0,
"step": 130
},
{
"entropy": 1.2879054959863425,
"epoch": 0.224,
"grad_norm": 18.625,
"learning_rate": 9.266666666666667e-06,
"loss": 20.3558,
"mean_token_accuracy": 0.7094652783125639,
"num_tokens": 4648491.0,
"step": 140
},
{
"entropy": 1.262782633677125,
"epoch": 0.24,
"grad_norm": 20.875,
"learning_rate": 9.933333333333334e-06,
"loss": 20.125,
"mean_token_accuracy": 0.7121465969830751,
"num_tokens": 4986175.0,
"step": 150
},
{
"entropy": 1.249500698968768,
"epoch": 0.256,
"grad_norm": 20.75,
"learning_rate": 1.0600000000000002e-05,
"loss": 19.7318,
"mean_token_accuracy": 0.716720923408866,
"num_tokens": 5329320.0,
"step": 160
},
{
"entropy": 1.2423948224633932,
"epoch": 0.272,
"grad_norm": 21.125,
"learning_rate": 1.1266666666666668e-05,
"loss": 19.6803,
"mean_token_accuracy": 0.7159549534320832,
"num_tokens": 5658796.0,
"step": 170
},
{
"entropy": 1.2081249222159385,
"epoch": 0.288,
"grad_norm": 18.5,
"learning_rate": 1.1933333333333335e-05,
"loss": 19.318,
"mean_token_accuracy": 0.7217235699295997,
"num_tokens": 5980201.0,
"step": 180
},
{
"entropy": 1.2010240752249957,
"epoch": 0.304,
"grad_norm": 23.125,
"learning_rate": 1.2600000000000001e-05,
"loss": 19.0924,
"mean_token_accuracy": 0.723386499658227,
"num_tokens": 6298858.0,
"step": 190
},
{
"entropy": 1.1805048868060113,
"epoch": 0.32,
"grad_norm": 21.5,
"learning_rate": 1.3266666666666668e-05,
"loss": 18.8302,
"mean_token_accuracy": 0.7257850173860788,
"num_tokens": 6622798.0,
"step": 200
},
{
"epoch": 0.32,
"eval_biology_entropy": 3.0335595417022705,
"eval_biology_loss": 3.5063252449035645,
"eval_biology_mean_token_accuracy": 0.43639786744117737,
"eval_biology_num_tokens": 6622798.0,
"eval_biology_runtime": 49.8291,
"eval_biology_samples_per_second": 10.034,
"eval_biology_steps_per_second": 2.509,
"step": 200
},
{
"epoch": 0.32,
"eval_chemistry_entropy": 1.5769986052513123,
"eval_chemistry_loss": 1.6105988025665283,
"eval_chemistry_mean_token_accuracy": 0.6568134782314301,
"eval_chemistry_num_tokens": 6622798.0,
"eval_chemistry_runtime": 61.7607,
"eval_chemistry_samples_per_second": 8.096,
"eval_chemistry_steps_per_second": 2.024,
"step": 200
},
{
"epoch": 0.32,
"eval_math_entropy": 1.1430701270103454,
"eval_math_loss": 1.3579970598220825,
"eval_math_mean_token_accuracy": 0.7044133353233337,
"eval_math_num_tokens": 6622798.0,
"eval_math_runtime": 63.3226,
"eval_math_samples_per_second": 7.896,
"eval_math_steps_per_second": 1.974,
"step": 200
},
{
"epoch": 0.32,
"eval_physics_entropy": 1.1658131113052368,
"eval_physics_loss": 1.1756293773651123,
"eval_physics_mean_token_accuracy": 0.7273708815574647,
"eval_physics_num_tokens": 6622798.0,
"eval_physics_runtime": 72.1537,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.732,
"step": 200
},
{
"entropy": 1.1631717231124639,
"epoch": 0.336,
"grad_norm": 15.8125,
"learning_rate": 1.3933333333333334e-05,
"loss": 18.617,
"mean_token_accuracy": 0.7290044378489255,
"num_tokens": 6953611.0,
"step": 210
},
{
"entropy": 1.1292029924690723,
"epoch": 0.352,
"grad_norm": 19.625,
"learning_rate": 1.46e-05,
"loss": 17.9563,
"mean_token_accuracy": 0.7365154311060905,
"num_tokens": 7290378.0,
"step": 220
},
{
"entropy": 1.1049898941069842,
"epoch": 0.368,
"grad_norm": 17.75,
"learning_rate": 1.5266666666666667e-05,
"loss": 17.6986,
"mean_token_accuracy": 0.7393099017441273,
"num_tokens": 7621082.0,
"step": 230
},
{
"entropy": 1.1595781400799752,
"epoch": 0.384,
"grad_norm": 19.375,
"learning_rate": 1.5933333333333336e-05,
"loss": 18.4615,
"mean_token_accuracy": 0.727893678098917,
"num_tokens": 7955570.0,
"step": 240
},
{
"entropy": 1.0945355758070945,
"epoch": 0.4,
"grad_norm": 21.0,
"learning_rate": 1.66e-05,
"loss": 17.5388,
"mean_token_accuracy": 0.7387776046991348,
"num_tokens": 8291049.0,
"step": 250
},
{
"entropy": 1.0819710556417703,
"epoch": 0.416,
"grad_norm": 18.375,
"learning_rate": 1.726666666666667e-05,
"loss": 17.251,
"mean_token_accuracy": 0.7430898856371642,
"num_tokens": 8627310.0,
"step": 260
},
{
"entropy": 1.089613201841712,
"epoch": 0.432,
"grad_norm": 20.75,
"learning_rate": 1.7933333333333333e-05,
"loss": 17.4707,
"mean_token_accuracy": 0.7406438160687685,
"num_tokens": 8958371.0,
"step": 270
},
{
"entropy": 1.0421824473887682,
"epoch": 0.448,
"grad_norm": 18.25,
"learning_rate": 1.86e-05,
"loss": 16.7755,
"mean_token_accuracy": 0.7496135134249926,
"num_tokens": 9294388.0,
"step": 280
},
{
"entropy": 1.0451348841190338,
"epoch": 0.464,
"grad_norm": 19.625,
"learning_rate": 1.926666666666667e-05,
"loss": 16.5784,
"mean_token_accuracy": 0.7497908163815736,
"num_tokens": 9620366.0,
"step": 290
},
{
"entropy": 1.0491984207183123,
"epoch": 0.48,
"grad_norm": 21.0,
"learning_rate": 1.9933333333333334e-05,
"loss": 16.8506,
"mean_token_accuracy": 0.7469458010047674,
"num_tokens": 9955431.0,
"step": 300
},
{
"epoch": 0.48,
"eval_biology_entropy": 2.9388448543548584,
"eval_biology_loss": 3.3656885623931885,
"eval_biology_mean_token_accuracy": 0.45114623188972475,
"eval_biology_num_tokens": 9955431.0,
"eval_biology_runtime": 49.8383,
"eval_biology_samples_per_second": 10.032,
"eval_biology_steps_per_second": 2.508,
"step": 300
},
{
"epoch": 0.48,
"eval_chemistry_entropy": 1.491808822631836,
"eval_chemistry_loss": 1.5070267915725708,
"eval_chemistry_mean_token_accuracy": 0.6716848533153534,
"eval_chemistry_num_tokens": 9955431.0,
"eval_chemistry_runtime": 61.771,
"eval_chemistry_samples_per_second": 8.094,
"eval_chemistry_steps_per_second": 2.024,
"step": 300
},
{
"epoch": 0.48,
"eval_math_entropy": 1.1062884669303894,
"eval_math_loss": 1.285144567489624,
"eval_math_mean_token_accuracy": 0.7163918890953064,
"eval_math_num_tokens": 9955431.0,
"eval_math_runtime": 63.3221,
"eval_math_samples_per_second": 7.896,
"eval_math_steps_per_second": 1.974,
"step": 300
},
{
"epoch": 0.48,
"eval_physics_entropy": 1.0871692457199096,
"eval_physics_loss": 1.0718116760253906,
"eval_physics_mean_token_accuracy": 0.7436052026748657,
"eval_physics_num_tokens": 9955431.0,
"eval_physics_runtime": 72.1598,
"eval_physics_samples_per_second": 6.929,
"eval_physics_steps_per_second": 1.732,
"step": 300
},
{
"entropy": 1.0622025161981583,
"epoch": 0.496,
"grad_norm": 17.625,
"learning_rate": 1.9933333333333334e-05,
"loss": 17.0457,
"mean_token_accuracy": 0.7451069477945567,
"num_tokens": 10289883.0,
"step": 310
},
{
"entropy": 1.0407135926187039,
"epoch": 0.512,
"grad_norm": 21.375,
"learning_rate": 1.985925925925926e-05,
"loss": 16.7612,
"mean_token_accuracy": 0.7487055193632841,
"num_tokens": 10619468.0,
"step": 320
},
{
"entropy": 1.060495626181364,
"epoch": 0.528,
"grad_norm": 19.625,
"learning_rate": 1.9785185185185187e-05,
"loss": 16.9124,
"mean_token_accuracy": 0.7458847668021917,
"num_tokens": 10944904.0,
"step": 330
},
{
"entropy": 1.0363870609551669,
"epoch": 0.544,
"grad_norm": 19.125,
"learning_rate": 1.971111111111111e-05,
"loss": 16.6786,
"mean_token_accuracy": 0.7487504351884127,
"num_tokens": 11266409.0,
"step": 340
},
{
"entropy": 1.01834645755589,
"epoch": 0.56,
"grad_norm": 17.125,
"learning_rate": 1.963703703703704e-05,
"loss": 16.224,
"mean_token_accuracy": 0.7543694581836462,
"num_tokens": 11605544.0,
"step": 350
},
{
"entropy": 1.0103376515209674,
"epoch": 0.576,
"grad_norm": 21.75,
"learning_rate": 1.9562962962962964e-05,
"loss": 16.1635,
"mean_token_accuracy": 0.7562580808997155,
"num_tokens": 11931715.0,
"step": 360
},
{
"entropy": 1.0192645259201527,
"epoch": 0.592,
"grad_norm": 19.625,
"learning_rate": 1.948888888888889e-05,
"loss": 16.3501,
"mean_token_accuracy": 0.7522288095206022,
"num_tokens": 12257243.0,
"step": 370
},
{
"entropy": 0.9979648280888795,
"epoch": 0.608,
"grad_norm": 17.25,
"learning_rate": 1.9414814814814817e-05,
"loss": 16.0193,
"mean_token_accuracy": 0.7565512597560883,
"num_tokens": 12583154.0,
"step": 380
},
{
"entropy": 1.0146858640015126,
"epoch": 0.624,
"grad_norm": 18.375,
"learning_rate": 1.9340740740740743e-05,
"loss": 16.2917,
"mean_token_accuracy": 0.7530687287449837,
"num_tokens": 12905392.0,
"step": 390
},
{
"entropy": 1.0172333989292384,
"epoch": 0.64,
"grad_norm": 18.0,
"learning_rate": 1.926666666666667e-05,
"loss": 16.311,
"mean_token_accuracy": 0.753251226246357,
"num_tokens": 13232198.0,
"step": 400
},
{
"epoch": 0.64,
"eval_biology_entropy": 2.8556682786941527,
"eval_biology_loss": 3.2405552864074707,
"eval_biology_mean_token_accuracy": 0.46047434616088867,
"eval_biology_num_tokens": 13232198.0,
"eval_biology_runtime": 49.8137,
"eval_biology_samples_per_second": 10.037,
"eval_biology_steps_per_second": 2.509,
"step": 400
},
{
"epoch": 0.64,
"eval_chemistry_entropy": 1.4023492636680603,
"eval_chemistry_loss": 1.4446464776992798,
"eval_chemistry_mean_token_accuracy": 0.6815885419845581,
"eval_chemistry_num_tokens": 13232198.0,
"eval_chemistry_runtime": 61.7651,
"eval_chemistry_samples_per_second": 8.095,
"eval_chemistry_steps_per_second": 2.024,
"step": 400
},
{
"epoch": 0.64,
"eval_math_entropy": 1.0369325885772704,
"eval_math_loss": 1.2500724792480469,
"eval_math_mean_token_accuracy": 0.7222203068733215,
"eval_math_num_tokens": 13232198.0,
"eval_math_runtime": 63.311,
"eval_math_samples_per_second": 7.898,
"eval_math_steps_per_second": 1.974,
"step": 400
},
{
"epoch": 0.64,
"eval_physics_entropy": 1.0061357097625732,
"eval_physics_loss": 1.0139094591140747,
"eval_physics_mean_token_accuracy": 0.7539046273231507,
"eval_physics_num_tokens": 13232198.0,
"eval_physics_runtime": 72.128,
"eval_physics_samples_per_second": 6.932,
"eval_physics_steps_per_second": 1.733,
"step": 400
},
{
"entropy": 1.0025192126631737,
"epoch": 0.656,
"grad_norm": 19.0,
"learning_rate": 1.9192592592592593e-05,
"loss": 16.0757,
"mean_token_accuracy": 0.7558728314936161,
"num_tokens": 13575902.0,
"step": 410
},
{
"entropy": 0.9805797968059778,
"epoch": 0.672,
"grad_norm": 20.75,
"learning_rate": 1.911851851851852e-05,
"loss": 15.6938,
"mean_token_accuracy": 0.7607984621077776,
"num_tokens": 13895997.0,
"step": 420
},
{
"entropy": 0.98455760627985,
"epoch": 0.688,
"grad_norm": 17.5,
"learning_rate": 1.9044444444444446e-05,
"loss": 15.8079,
"mean_token_accuracy": 0.7591990135610104,
"num_tokens": 14234888.0,
"step": 430
},
{
"entropy": 0.995212909206748,
"epoch": 0.704,
"grad_norm": 19.5,
"learning_rate": 1.8970370370370372e-05,
"loss": 16.0276,
"mean_token_accuracy": 0.7551241502165794,
"num_tokens": 14567908.0,
"step": 440
},
{
"entropy": 0.9859623953700065,
"epoch": 0.72,
"grad_norm": 19.875,
"learning_rate": 1.8896296296296295e-05,
"loss": 15.7158,
"mean_token_accuracy": 0.759292322024703,
"num_tokens": 14882927.0,
"step": 450
},
{
"entropy": 0.9759141772985458,
"epoch": 0.736,
"grad_norm": 20.625,
"learning_rate": 1.8822222222222225e-05,
"loss": 15.6771,
"mean_token_accuracy": 0.759360334649682,
"num_tokens": 15217342.0,
"step": 460
},
{
"entropy": 0.9576828479766846,
"epoch": 0.752,
"grad_norm": 19.5,
"learning_rate": 1.874814814814815e-05,
"loss": 15.2707,
"mean_token_accuracy": 0.764180276170373,
"num_tokens": 15547710.0,
"step": 470
},
{
"entropy": 0.9969257492572069,
"epoch": 0.768,
"grad_norm": 19.625,
"learning_rate": 1.8674074074074075e-05,
"loss": 16.0499,
"mean_token_accuracy": 0.7543311327695846,
"num_tokens": 15877177.0,
"step": 480
},
{
"entropy": 0.9526452720165253,
"epoch": 0.784,
"grad_norm": 15.8125,
"learning_rate": 1.86e-05,
"loss": 15.3172,
"mean_token_accuracy": 0.7652558576315641,
"num_tokens": 16219640.0,
"step": 490
},
{
"entropy": 0.9730141185224056,
"epoch": 0.8,
"grad_norm": 19.25,
"learning_rate": 1.8525925925925928e-05,
"loss": 15.5783,
"mean_token_accuracy": 0.7610898818820715,
"num_tokens": 16548261.0,
"step": 500
},
{
"epoch": 0.8,
"eval_biology_entropy": 2.8879401025772093,
"eval_biology_loss": 3.2465980052948,
"eval_biology_mean_token_accuracy": 0.4656795439720154,
"eval_biology_num_tokens": 16548261.0,
"eval_biology_runtime": 50.0039,
"eval_biology_samples_per_second": 9.999,
"eval_biology_steps_per_second": 2.5,
"step": 500
},
{
"epoch": 0.8,
"eval_chemistry_entropy": 1.4028268342018126,
"eval_chemistry_loss": 1.4141976833343506,
"eval_chemistry_mean_token_accuracy": 0.6879124054908753,
"eval_chemistry_num_tokens": 16548261.0,
"eval_chemistry_runtime": 61.4753,
"eval_chemistry_samples_per_second": 8.133,
"eval_chemistry_steps_per_second": 2.033,
"step": 500
},
{
"epoch": 0.8,
"eval_math_entropy": 1.036446961402893,
"eval_math_loss": 1.2320657968521118,
"eval_math_mean_token_accuracy": 0.7248759384155273,
"eval_math_num_tokens": 16548261.0,
"eval_math_runtime": 63.2484,
"eval_math_samples_per_second": 7.905,
"eval_math_steps_per_second": 1.976,
"step": 500
},
{
"epoch": 0.8,
"eval_physics_entropy": 0.9860591797828674,
"eval_physics_loss": 0.9807776808738708,
"eval_physics_mean_token_accuracy": 0.7594618821144103,
"eval_physics_num_tokens": 16548261.0,
"eval_physics_runtime": 72.1552,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.732,
"step": 500
},
{
"entropy": 0.9596989471465349,
"epoch": 0.816,
"grad_norm": 17.5,
"learning_rate": 1.8451851851851855e-05,
"loss": 15.4334,
"mean_token_accuracy": 0.7610802616924047,
"num_tokens": 16870533.0,
"step": 510
},
{
"entropy": 0.9716402690857648,
"epoch": 0.832,
"grad_norm": 18.25,
"learning_rate": 1.8377777777777778e-05,
"loss": 15.5823,
"mean_token_accuracy": 0.7609996516257524,
"num_tokens": 17196371.0,
"step": 520
},
{
"entropy": 0.9541628547012806,
"epoch": 0.848,
"grad_norm": 16.5,
"learning_rate": 1.8303703703703704e-05,
"loss": 15.3143,
"mean_token_accuracy": 0.7634800903499126,
"num_tokens": 17527066.0,
"step": 530
},
{
"entropy": 0.9640029605478049,
"epoch": 0.864,
"grad_norm": 17.375,
"learning_rate": 1.822962962962963e-05,
"loss": 15.4415,
"mean_token_accuracy": 0.7621862847357989,
"num_tokens": 17859605.0,
"step": 540
},
{
"entropy": 0.9721696741878987,
"epoch": 0.88,
"grad_norm": 20.25,
"learning_rate": 1.8155555555555557e-05,
"loss": 15.653,
"mean_token_accuracy": 0.7603420093655586,
"num_tokens": 18187315.0,
"step": 550
},
{
"entropy": 0.9153296016156673,
"epoch": 0.896,
"grad_norm": 17.25,
"learning_rate": 1.8081481481481484e-05,
"loss": 14.6671,
"mean_token_accuracy": 0.7723869703710079,
"num_tokens": 18536790.0,
"step": 560
},
{
"entropy": 0.958526523038745,
"epoch": 0.912,
"grad_norm": 18.75,
"learning_rate": 1.800740740740741e-05,
"loss": 15.4361,
"mean_token_accuracy": 0.7610937509685755,
"num_tokens": 18869344.0,
"step": 570
},
{
"entropy": 0.9475870430469513,
"epoch": 0.928,
"grad_norm": 19.125,
"learning_rate": 1.7933333333333333e-05,
"loss": 15.2185,
"mean_token_accuracy": 0.7640813227742911,
"num_tokens": 19199834.0,
"step": 580
},
{
"entropy": 0.9297015912830829,
"epoch": 0.944,
"grad_norm": 18.75,
"learning_rate": 1.785925925925926e-05,
"loss": 14.8769,
"mean_token_accuracy": 0.7691585179418325,
"num_tokens": 19525385.0,
"step": 590
},
{
"entropy": 0.9348739337176084,
"epoch": 0.96,
"grad_norm": 19.0,
"learning_rate": 1.7785185185185186e-05,
"loss": 15.0455,
"mean_token_accuracy": 0.7664772134274245,
"num_tokens": 19856973.0,
"step": 600
},
{
"epoch": 0.96,
"eval_biology_entropy": 2.7933437023162844,
"eval_biology_loss": 3.194451332092285,
"eval_biology_mean_token_accuracy": 0.46872161483764646,
"eval_biology_num_tokens": 19856973.0,
"eval_biology_runtime": 49.8208,
"eval_biology_samples_per_second": 10.036,
"eval_biology_steps_per_second": 2.509,
"step": 600
},
{
"epoch": 0.96,
"eval_chemistry_entropy": 1.3487913212776184,
"eval_chemistry_loss": 1.3904978036880493,
"eval_chemistry_mean_token_accuracy": 0.6907124133110046,
"eval_chemistry_num_tokens": 19856973.0,
"eval_chemistry_runtime": 62.2399,
"eval_chemistry_samples_per_second": 8.033,
"eval_chemistry_steps_per_second": 2.008,
"step": 600
},
{
"epoch": 0.96,
"eval_math_entropy": 1.0215094347000122,
"eval_math_loss": 1.2187507152557373,
"eval_math_mean_token_accuracy": 0.727313805103302,
"eval_math_num_tokens": 19856973.0,
"eval_math_runtime": 63.311,
"eval_math_samples_per_second": 7.898,
"eval_math_steps_per_second": 1.974,
"step": 600
},
{
"epoch": 0.96,
"eval_physics_entropy": 0.9578838763237,
"eval_physics_loss": 0.9591855406761169,
"eval_physics_mean_token_accuracy": 0.7633539743423462,
"eval_physics_num_tokens": 19856973.0,
"eval_physics_runtime": 72.6851,
"eval_physics_samples_per_second": 6.879,
"eval_physics_steps_per_second": 1.72,
"step": 600
},
{
"entropy": 0.9349439710378646,
"epoch": 0.976,
"grad_norm": 19.125,
"learning_rate": 1.7711111111111113e-05,
"loss": 14.9645,
"mean_token_accuracy": 0.7687592633068562,
"num_tokens": 20190600.0,
"step": 610
},
{
"entropy": 0.9438869334757328,
"epoch": 0.992,
"grad_norm": 18.625,
"learning_rate": 1.763703703703704e-05,
"loss": 15.1663,
"mean_token_accuracy": 0.765958558395505,
"num_tokens": 20518014.0,
"step": 620
},
{
"entropy": 0.9152817223221064,
"epoch": 1.008,
"grad_norm": 17.0,
"learning_rate": 1.7562962962962962e-05,
"loss": 14.5949,
"mean_token_accuracy": 0.7715727590024472,
"num_tokens": 20859215.0,
"step": 630
},
{
"entropy": 0.9109054185450077,
"epoch": 1.024,
"grad_norm": 18.0,
"learning_rate": 1.7488888888888892e-05,
"loss": 14.683,
"mean_token_accuracy": 0.7712803717702628,
"num_tokens": 21186658.0,
"step": 640
},
{
"entropy": 0.9192747503519059,
"epoch": 1.04,
"grad_norm": 17.375,
"learning_rate": 1.7414814814814815e-05,
"loss": 14.7149,
"mean_token_accuracy": 0.7704985976219177,
"num_tokens": 21517237.0,
"step": 650
},
{
"entropy": 0.8997716940939426,
"epoch": 1.056,
"grad_norm": 18.5,
"learning_rate": 1.7340740740740742e-05,
"loss": 14.4443,
"mean_token_accuracy": 0.7745653819292784,
"num_tokens": 21852528.0,
"step": 660
},
{
"entropy": 0.9201500467956066,
"epoch": 1.072,
"grad_norm": 16.375,
"learning_rate": 1.726666666666667e-05,
"loss": 14.7404,
"mean_token_accuracy": 0.7707031384110451,
"num_tokens": 22192380.0,
"step": 670
},
{
"entropy": 0.9155960086733103,
"epoch": 1.088,
"grad_norm": 20.0,
"learning_rate": 1.7192592592592595e-05,
"loss": 14.6895,
"mean_token_accuracy": 0.771490203589201,
"num_tokens": 22517971.0,
"step": 680
},
{
"entropy": 0.8976538371294737,
"epoch": 1.104,
"grad_norm": 16.625,
"learning_rate": 1.711851851851852e-05,
"loss": 14.4464,
"mean_token_accuracy": 0.7735383935272694,
"num_tokens": 22848694.0,
"step": 690
},
{
"entropy": 0.9162428434938192,
"epoch": 1.12,
"grad_norm": 17.5,
"learning_rate": 1.7044444444444445e-05,
"loss": 14.7282,
"mean_token_accuracy": 0.7701425358653069,
"num_tokens": 23183889.0,
"step": 700
},
{
"epoch": 1.12,
"eval_biology_entropy": 2.7417425537109374,
"eval_biology_loss": 3.198380947113037,
"eval_biology_mean_token_accuracy": 0.46923186993598937,
"eval_biology_num_tokens": 23183889.0,
"eval_biology_runtime": 49.9576,
"eval_biology_samples_per_second": 10.008,
"eval_biology_steps_per_second": 2.502,
"step": 700
},
{
"epoch": 1.12,
"eval_chemistry_entropy": 1.3224085516929627,
"eval_chemistry_loss": 1.3783864974975586,
"eval_chemistry_mean_token_accuracy": 0.6928294057846069,
"eval_chemistry_num_tokens": 23183889.0,
"eval_chemistry_runtime": 62.3468,
"eval_chemistry_samples_per_second": 8.02,
"eval_chemistry_steps_per_second": 2.005,
"step": 700
},
{
"epoch": 1.12,
"eval_math_entropy": 0.9974914565086365,
"eval_math_loss": 1.2118620872497559,
"eval_math_mean_token_accuracy": 0.7289316186904907,
"eval_math_num_tokens": 23183889.0,
"eval_math_runtime": 63.342,
"eval_math_samples_per_second": 7.894,
"eval_math_steps_per_second": 1.973,
"step": 700
},
{
"epoch": 1.12,
"eval_physics_entropy": 0.9341713042259217,
"eval_physics_loss": 0.9430394172668457,
"eval_physics_mean_token_accuracy": 0.7666752815246582,
"eval_physics_num_tokens": 23183889.0,
"eval_physics_runtime": 72.1566,
"eval_physics_samples_per_second": 6.929,
"eval_physics_steps_per_second": 1.732,
"step": 700
},
{
"entropy": 0.8717782899737359,
"epoch": 1.1360000000000001,
"grad_norm": 15.9375,
"learning_rate": 1.697037037037037e-05,
"loss": 14.027,
"mean_token_accuracy": 0.7789868891239167,
"num_tokens": 23523903.0,
"step": 710
},
{
"entropy": 0.9066614583134651,
"epoch": 1.152,
"grad_norm": 20.0,
"learning_rate": 1.6896296296296298e-05,
"loss": 14.5252,
"mean_token_accuracy": 0.7747723516076803,
"num_tokens": 23848848.0,
"step": 720
},
{
"entropy": 0.9192185271531343,
"epoch": 1.168,
"grad_norm": 19.0,
"learning_rate": 1.6822222222222224e-05,
"loss": 14.6797,
"mean_token_accuracy": 0.7702812351286411,
"num_tokens": 24174589.0,
"step": 730
},
{
"entropy": 0.8968648243695497,
"epoch": 1.184,
"grad_norm": 19.25,
"learning_rate": 1.6748148148148147e-05,
"loss": 14.4441,
"mean_token_accuracy": 0.7734088286757469,
"num_tokens": 24495981.0,
"step": 740
},
{
"entropy": 0.8798596655949951,
"epoch": 1.2,
"grad_norm": 18.625,
"learning_rate": 1.6674074074074077e-05,
"loss": 14.2088,
"mean_token_accuracy": 0.7773797914385796,
"num_tokens": 24826245.0,
"step": 750
},
{
"entropy": 0.9085365436971188,
"epoch": 1.216,
"grad_norm": 23.625,
"learning_rate": 1.66e-05,
"loss": 14.5158,
"mean_token_accuracy": 0.7717528346925974,
"num_tokens": 25147775.0,
"step": 760
},
{
"entropy": 0.8942177969962358,
"epoch": 1.232,
"grad_norm": 19.375,
"learning_rate": 1.6525925925925927e-05,
"loss": 14.35,
"mean_token_accuracy": 0.7760623764246702,
"num_tokens": 25480033.0,
"step": 770
},
{
"entropy": 0.8957445651292801,
"epoch": 1.248,
"grad_norm": 19.625,
"learning_rate": 1.6451851851851853e-05,
"loss": 14.3489,
"mean_token_accuracy": 0.7746611826121808,
"num_tokens": 25811978.0,
"step": 780
},
{
"entropy": 0.8850473642349244,
"epoch": 1.264,
"grad_norm": 19.375,
"learning_rate": 1.637777777777778e-05,
"loss": 14.2639,
"mean_token_accuracy": 0.7760811734944582,
"num_tokens": 26146299.0,
"step": 790
},
{
"entropy": 0.9187131529673934,
"epoch": 1.28,
"grad_norm": 18.875,
"learning_rate": 1.6303703703703706e-05,
"loss": 14.7971,
"mean_token_accuracy": 0.7688343476504087,
"num_tokens": 26478207.0,
"step": 800
},
{
"epoch": 1.28,
"eval_biology_entropy": 2.7373362789154054,
"eval_biology_loss": 3.2188053131103516,
"eval_biology_mean_token_accuracy": 0.4715359935760498,
"eval_biology_num_tokens": 26478207.0,
"eval_biology_runtime": 49.8242,
"eval_biology_samples_per_second": 10.035,
"eval_biology_steps_per_second": 2.509,
"step": 800
},
{
"epoch": 1.28,
"eval_chemistry_entropy": 1.2954678382873535,
"eval_chemistry_loss": 1.3689188957214355,
"eval_chemistry_mean_token_accuracy": 0.6956489958763122,
"eval_chemistry_num_tokens": 26478207.0,
"eval_chemistry_runtime": 61.7443,
"eval_chemistry_samples_per_second": 8.098,
"eval_chemistry_steps_per_second": 2.024,
"step": 800
},
{
"epoch": 1.28,
"eval_math_entropy": 0.9856116833686829,
"eval_math_loss": 1.2060915231704712,
"eval_math_mean_token_accuracy": 0.7302432260513305,
"eval_math_num_tokens": 26478207.0,
"eval_math_runtime": 63.311,
"eval_math_samples_per_second": 7.898,
"eval_math_steps_per_second": 1.974,
"step": 800
},
{
"epoch": 1.28,
"eval_physics_entropy": 0.9129857659339905,
"eval_physics_loss": 0.9318634271621704,
"eval_physics_mean_token_accuracy": 0.7681499900817871,
"eval_physics_num_tokens": 26478207.0,
"eval_physics_runtime": 72.1104,
"eval_physics_samples_per_second": 6.934,
"eval_physics_steps_per_second": 1.733,
"step": 800
},
{
"entropy": 0.8775971475988626,
"epoch": 1.296,
"grad_norm": 19.625,
"learning_rate": 1.622962962962963e-05,
"loss": 14.0996,
"mean_token_accuracy": 0.7783409550786018,
"num_tokens": 26806477.0,
"step": 810
},
{
"entropy": 0.9151829794049263,
"epoch": 1.312,
"grad_norm": 20.125,
"learning_rate": 1.6155555555555556e-05,
"loss": 14.6154,
"mean_token_accuracy": 0.7715053271502257,
"num_tokens": 27137528.0,
"step": 820
},
{
"entropy": 0.8893604565411806,
"epoch": 1.328,
"grad_norm": 21.375,
"learning_rate": 1.6081481481481482e-05,
"loss": 14.3336,
"mean_token_accuracy": 0.7749421369284392,
"num_tokens": 27465859.0,
"step": 830
},
{
"entropy": 0.9043792076408863,
"epoch": 1.3439999999999999,
"grad_norm": 20.25,
"learning_rate": 1.600740740740741e-05,
"loss": 14.5396,
"mean_token_accuracy": 0.7718299012631178,
"num_tokens": 27795568.0,
"step": 840
},
{
"entropy": 0.8975318446755409,
"epoch": 1.3599999999999999,
"grad_norm": 18.75,
"learning_rate": 1.5933333333333336e-05,
"loss": 14.4516,
"mean_token_accuracy": 0.7730446238070726,
"num_tokens": 28129988.0,
"step": 850
},
{
"entropy": 0.894953028857708,
"epoch": 1.376,
"grad_norm": 20.625,
"learning_rate": 1.5859259259259262e-05,
"loss": 14.2654,
"mean_token_accuracy": 0.7754011388868094,
"num_tokens": 28465903.0,
"step": 860
},
{
"entropy": 0.8700921148061752,
"epoch": 1.392,
"grad_norm": 19.875,
"learning_rate": 1.5785185185185185e-05,
"loss": 13.9603,
"mean_token_accuracy": 0.7800745751708746,
"num_tokens": 28801554.0,
"step": 870
},
{
"entropy": 0.8798988396301866,
"epoch": 1.408,
"grad_norm": 18.75,
"learning_rate": 1.571111111111111e-05,
"loss": 14.1608,
"mean_token_accuracy": 0.7768289878964424,
"num_tokens": 29125114.0,
"step": 880
},
{
"entropy": 0.9143678206950426,
"epoch": 1.424,
"grad_norm": 17.875,
"learning_rate": 1.5637037037037038e-05,
"loss": 14.7235,
"mean_token_accuracy": 0.7692738968878985,
"num_tokens": 29454334.0,
"step": 890
},
{
"entropy": 0.8575869327411055,
"epoch": 1.44,
"grad_norm": 17.75,
"learning_rate": 1.5562962962962965e-05,
"loss": 13.7841,
"mean_token_accuracy": 0.7813448339700699,
"num_tokens": 29789253.0,
"step": 900
},
{
"epoch": 1.44,
"eval_biology_entropy": 2.765932272911072,
"eval_biology_loss": 3.186239242553711,
"eval_biology_mean_token_accuracy": 0.4713315246105194,
"eval_biology_num_tokens": 29789253.0,
"eval_biology_runtime": 49.9865,
"eval_biology_samples_per_second": 10.003,
"eval_biology_steps_per_second": 2.501,
"step": 900
},
{
"epoch": 1.44,
"eval_chemistry_entropy": 1.293758502960205,
"eval_chemistry_loss": 1.3550570011138916,
"eval_chemistry_mean_token_accuracy": 0.6970094337463378,
"eval_chemistry_num_tokens": 29789253.0,
"eval_chemistry_runtime": 61.9517,
"eval_chemistry_samples_per_second": 8.071,
"eval_chemistry_steps_per_second": 2.018,
"step": 900
},
{
"epoch": 1.44,
"eval_math_entropy": 0.9867032241821289,
"eval_math_loss": 1.201257348060608,
"eval_math_mean_token_accuracy": 0.7303560256958008,
"eval_math_num_tokens": 29789253.0,
"eval_math_runtime": 63.5245,
"eval_math_samples_per_second": 7.871,
"eval_math_steps_per_second": 1.968,
"step": 900
},
{
"epoch": 1.44,
"eval_physics_entropy": 0.9099465346336365,
"eval_physics_loss": 0.9209814071655273,
"eval_physics_mean_token_accuracy": 0.7701499495506287,
"eval_physics_num_tokens": 29789253.0,
"eval_physics_runtime": 72.3521,
"eval_physics_samples_per_second": 6.911,
"eval_physics_steps_per_second": 1.728,
"step": 900
},
{
"entropy": 0.8655804365873336,
"epoch": 1.456,
"grad_norm": 15.875,
"learning_rate": 1.548888888888889e-05,
"loss": 13.9105,
"mean_token_accuracy": 0.7815111551433802,
"num_tokens": 30128576.0,
"step": 910
},
{
"entropy": 0.9171438716351986,
"epoch": 1.472,
"grad_norm": 20.375,
"learning_rate": 1.5414814814814814e-05,
"loss": 14.7757,
"mean_token_accuracy": 0.7698528375476599,
"num_tokens": 30452719.0,
"step": 920
},
{
"entropy": 0.8968168456107378,
"epoch": 1.488,
"grad_norm": 18.375,
"learning_rate": 1.5340740740740744e-05,
"loss": 14.385,
"mean_token_accuracy": 0.7741949297487736,
"num_tokens": 30782343.0,
"step": 930
},
{
"entropy": 0.8784723855555058,
"epoch": 1.504,
"grad_norm": 16.875,
"learning_rate": 1.5266666666666667e-05,
"loss": 14.0301,
"mean_token_accuracy": 0.7789344146847725,
"num_tokens": 31113238.0,
"step": 940
},
{
"entropy": 0.8974394641816617,
"epoch": 1.52,
"grad_norm": 18.875,
"learning_rate": 1.5192592592592594e-05,
"loss": 14.3838,
"mean_token_accuracy": 0.7739370841532945,
"num_tokens": 31444589.0,
"step": 950
},
{
"entropy": 0.8973223175853491,
"epoch": 1.536,
"grad_norm": 20.625,
"learning_rate": 1.5118518518518519e-05,
"loss": 14.4894,
"mean_token_accuracy": 0.7721214085817337,
"num_tokens": 31774137.0,
"step": 960
},
{
"entropy": 0.8697954054921866,
"epoch": 1.552,
"grad_norm": 19.375,
"learning_rate": 1.5044444444444445e-05,
"loss": 13.8974,
"mean_token_accuracy": 0.78069927431643,
"num_tokens": 32101776.0,
"step": 970
},
{
"entropy": 0.8705245990306139,
"epoch": 1.568,
"grad_norm": 19.125,
"learning_rate": 1.497037037037037e-05,
"loss": 13.979,
"mean_token_accuracy": 0.7797447606921196,
"num_tokens": 32428021.0,
"step": 980
},
{
"entropy": 0.8989604346454143,
"epoch": 1.584,
"grad_norm": 18.375,
"learning_rate": 1.4896296296296298e-05,
"loss": 14.4829,
"mean_token_accuracy": 0.7704301360994578,
"num_tokens": 32752735.0,
"step": 990
},
{
"entropy": 0.8908115413039923,
"epoch": 1.6,
"grad_norm": 17.75,
"learning_rate": 1.4822222222222225e-05,
"loss": 14.2986,
"mean_token_accuracy": 0.7751044664531946,
"num_tokens": 33075822.0,
"step": 1000
},
{
"epoch": 1.6,
"eval_biology_entropy": 2.702871771812439,
"eval_biology_loss": 3.161149501800537,
"eval_biology_mean_token_accuracy": 0.47477258658409116,
"eval_biology_num_tokens": 33075822.0,
"eval_biology_runtime": 50.0794,
"eval_biology_samples_per_second": 9.984,
"eval_biology_steps_per_second": 2.496,
"step": 1000
},
{
"epoch": 1.6,
"eval_chemistry_entropy": 1.265580331325531,
"eval_chemistry_loss": 1.3501311540603638,
"eval_chemistry_mean_token_accuracy": 0.6981884994506836,
"eval_chemistry_num_tokens": 33075822.0,
"eval_chemistry_runtime": 61.4474,
"eval_chemistry_samples_per_second": 8.137,
"eval_chemistry_steps_per_second": 2.034,
"step": 1000
},
{
"epoch": 1.6,
"eval_math_entropy": 0.9701343817710877,
"eval_math_loss": 1.2001285552978516,
"eval_math_mean_token_accuracy": 0.7312138648033142,
"eval_math_num_tokens": 33075822.0,
"eval_math_runtime": 63.2376,
"eval_math_samples_per_second": 7.907,
"eval_math_steps_per_second": 1.977,
"step": 1000
},
{
"epoch": 1.6,
"eval_physics_entropy": 0.8882069087028504,
"eval_physics_loss": 0.9141198992729187,
"eval_physics_mean_token_accuracy": 0.7712050895690918,
"eval_physics_num_tokens": 33075822.0,
"eval_physics_runtime": 72.1374,
"eval_physics_samples_per_second": 6.931,
"eval_physics_steps_per_second": 1.733,
"step": 1000
},
{
"entropy": 0.852231327816844,
"epoch": 1.616,
"grad_norm": 19.5,
"learning_rate": 1.474814814814815e-05,
"loss": 13.7455,
"mean_token_accuracy": 0.7828571233898401,
"num_tokens": 33411850.0,
"step": 1010
},
{
"entropy": 0.8694836173206568,
"epoch": 1.6320000000000001,
"grad_norm": 19.375,
"learning_rate": 1.4674074074074076e-05,
"loss": 13.9574,
"mean_token_accuracy": 0.7792858675122261,
"num_tokens": 33739456.0,
"step": 1020
},
{
"entropy": 0.8416479125618934,
"epoch": 1.6480000000000001,
"grad_norm": 18.375,
"learning_rate": 1.46e-05,
"loss": 13.4841,
"mean_token_accuracy": 0.7848905235528946,
"num_tokens": 34072022.0,
"step": 1030
},
{
"entropy": 0.8727968029677868,
"epoch": 1.6640000000000001,
"grad_norm": 20.875,
"learning_rate": 1.4525925925925927e-05,
"loss": 13.9919,
"mean_token_accuracy": 0.7798548869788646,
"num_tokens": 34398959.0,
"step": 1040
},
{
"entropy": 0.8499273512512445,
"epoch": 1.6800000000000002,
"grad_norm": 17.875,
"learning_rate": 1.4451851851851852e-05,
"loss": 13.7693,
"mean_token_accuracy": 0.7817482702434063,
"num_tokens": 34732284.0,
"step": 1050
},
{
"entropy": 0.8752633452415466,
"epoch": 1.696,
"grad_norm": 19.25,
"learning_rate": 1.4377777777777779e-05,
"loss": 13.9838,
"mean_token_accuracy": 0.7789037089794875,
"num_tokens": 35053712.0,
"step": 1060
},
{
"entropy": 0.8853001151233911,
"epoch": 1.712,
"grad_norm": 20.0,
"learning_rate": 1.4303703703703703e-05,
"loss": 14.2107,
"mean_token_accuracy": 0.7755775731056929,
"num_tokens": 35393088.0,
"step": 1070
},
{
"entropy": 0.8599673572927713,
"epoch": 1.728,
"grad_norm": 18.5,
"learning_rate": 1.4229629629629632e-05,
"loss": 13.8371,
"mean_token_accuracy": 0.7809670843183995,
"num_tokens": 35729007.0,
"step": 1080
},
{
"entropy": 0.867702866345644,
"epoch": 1.744,
"grad_norm": 19.125,
"learning_rate": 1.4155555555555556e-05,
"loss": 13.9299,
"mean_token_accuracy": 0.7801329270005226,
"num_tokens": 36072489.0,
"step": 1090
},
{
"entropy": 0.8642092518508434,
"epoch": 1.76,
"grad_norm": 18.625,
"learning_rate": 1.4081481481481483e-05,
"loss": 13.8054,
"mean_token_accuracy": 0.7801033087074757,
"num_tokens": 36398285.0,
"step": 1100
},
{
"epoch": 1.76,
"eval_biology_entropy": 2.676761179924011,
"eval_biology_loss": 3.160200834274292,
"eval_biology_mean_token_accuracy": 0.47685695719718935,
"eval_biology_num_tokens": 36398285.0,
"eval_biology_runtime": 49.8309,
"eval_biology_samples_per_second": 10.034,
"eval_biology_steps_per_second": 2.508,
"step": 1100
},
{
"epoch": 1.76,
"eval_chemistry_entropy": 1.250497200012207,
"eval_chemistry_loss": 1.3435908555984497,
"eval_chemistry_mean_token_accuracy": 0.7001057581901551,
"eval_chemistry_num_tokens": 36398285.0,
"eval_chemistry_runtime": 62.0199,
"eval_chemistry_samples_per_second": 8.062,
"eval_chemistry_steps_per_second": 2.015,
"step": 1100
},
{
"epoch": 1.76,
"eval_math_entropy": 0.9594085035324097,
"eval_math_loss": 1.1948144435882568,
"eval_math_mean_token_accuracy": 0.7322173299789428,
"eval_math_num_tokens": 36398285.0,
"eval_math_runtime": 63.3342,
"eval_math_samples_per_second": 7.895,
"eval_math_steps_per_second": 1.974,
"step": 1100
},
{
"epoch": 1.76,
"eval_physics_entropy": 0.8774992027282715,
"eval_physics_loss": 0.9080738425254822,
"eval_physics_mean_token_accuracy": 0.7721537561416626,
"eval_physics_num_tokens": 36398285.0,
"eval_physics_runtime": 72.1564,
"eval_physics_samples_per_second": 6.929,
"eval_physics_steps_per_second": 1.732,
"step": 1100
},
{
"entropy": 0.895181929320097,
"epoch": 1.776,
"grad_norm": 19.0,
"learning_rate": 1.400740740740741e-05,
"loss": 14.4407,
"mean_token_accuracy": 0.7728559862822294,
"num_tokens": 36730479.0,
"step": 1110
},
{
"entropy": 0.8563918903470039,
"epoch": 1.792,
"grad_norm": 23.5,
"learning_rate": 1.3933333333333334e-05,
"loss": 13.803,
"mean_token_accuracy": 0.7809405270963907,
"num_tokens": 37057425.0,
"step": 1120
},
{
"entropy": 0.8510978795588017,
"epoch": 1.808,
"grad_norm": 18.875,
"learning_rate": 1.385925925925926e-05,
"loss": 13.6333,
"mean_token_accuracy": 0.784480345249176,
"num_tokens": 37395449.0,
"step": 1130
},
{
"entropy": 0.8645633378997445,
"epoch": 1.8239999999999998,
"grad_norm": 18.875,
"learning_rate": 1.3785185185185186e-05,
"loss": 13.872,
"mean_token_accuracy": 0.7808197818696498,
"num_tokens": 37729535.0,
"step": 1140
},
{
"entropy": 0.8631477665156126,
"epoch": 1.8399999999999999,
"grad_norm": 19.625,
"learning_rate": 1.3711111111111112e-05,
"loss": 13.8345,
"mean_token_accuracy": 0.7815310817211867,
"num_tokens": 38057839.0,
"step": 1150
},
{
"entropy": 0.8587682455778122,
"epoch": 1.8559999999999999,
"grad_norm": 18.875,
"learning_rate": 1.3637037037037037e-05,
"loss": 13.8867,
"mean_token_accuracy": 0.780883165821433,
"num_tokens": 38398033.0,
"step": 1160
},
{
"entropy": 0.859000613167882,
"epoch": 1.8719999999999999,
"grad_norm": 17.875,
"learning_rate": 1.3562962962962965e-05,
"loss": 13.7667,
"mean_token_accuracy": 0.7798742011189461,
"num_tokens": 38729203.0,
"step": 1170
},
{
"entropy": 0.8777886133641004,
"epoch": 1.888,
"grad_norm": 19.625,
"learning_rate": 1.3488888888888888e-05,
"loss": 14.1636,
"mean_token_accuracy": 0.7765467241406441,
"num_tokens": 39058791.0,
"step": 1180
},
{
"entropy": 0.8881994623690843,
"epoch": 1.904,
"grad_norm": 20.125,
"learning_rate": 1.3414814814814817e-05,
"loss": 14.3781,
"mean_token_accuracy": 0.7752389151602983,
"num_tokens": 39386170.0,
"step": 1190
},
{
"entropy": 0.9030840078368783,
"epoch": 1.92,
"grad_norm": 19.5,
"learning_rate": 1.3340740740740741e-05,
"loss": 14.3564,
"mean_token_accuracy": 0.7748420361429453,
"num_tokens": 39710381.0,
"step": 1200
},
{
"epoch": 1.92,
"eval_biology_entropy": 2.697428282737732,
"eval_biology_loss": 3.138005256652832,
"eval_biology_mean_token_accuracy": 0.47676516604423524,
"eval_biology_num_tokens": 39710381.0,
"eval_biology_runtime": 49.822,
"eval_biology_samples_per_second": 10.036,
"eval_biology_steps_per_second": 2.509,
"step": 1200
},
{
"epoch": 1.92,
"eval_chemistry_entropy": 1.2482068204879762,
"eval_chemistry_loss": 1.3385791778564453,
"eval_chemistry_mean_token_accuracy": 0.7002091021537781,
"eval_chemistry_num_tokens": 39710381.0,
"eval_chemistry_runtime": 61.7709,
"eval_chemistry_samples_per_second": 8.094,
"eval_chemistry_steps_per_second": 2.024,
"step": 1200
},
{
"epoch": 1.92,
"eval_math_entropy": 0.9588226528167725,
"eval_math_loss": 1.190582036972046,
"eval_math_mean_token_accuracy": 0.7330412902832031,
"eval_math_num_tokens": 39710381.0,
"eval_math_runtime": 63.3698,
"eval_math_samples_per_second": 7.89,
"eval_math_steps_per_second": 1.973,
"step": 1200
},
{
"epoch": 1.92,
"eval_physics_entropy": 0.8757374310493469,
"eval_physics_loss": 0.9024468064308167,
"eval_physics_mean_token_accuracy": 0.7736027889251709,
"eval_physics_num_tokens": 39710381.0,
"eval_physics_runtime": 72.1456,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.733,
"step": 1200
},
{
"entropy": 0.8624950293451548,
"epoch": 1.936,
"grad_norm": 19.375,
"learning_rate": 1.3266666666666668e-05,
"loss": 13.9438,
"mean_token_accuracy": 0.7779798842966557,
"num_tokens": 40043354.0,
"step": 1210
},
{
"entropy": 0.8265730097889901,
"epoch": 1.952,
"grad_norm": 18.75,
"learning_rate": 1.3192592592592594e-05,
"loss": 13.1879,
"mean_token_accuracy": 0.7890967357903719,
"num_tokens": 40375790.0,
"step": 1220
},
{
"entropy": 0.8566216921433807,
"epoch": 1.968,
"grad_norm": 18.375,
"learning_rate": 1.311851851851852e-05,
"loss": 13.8742,
"mean_token_accuracy": 0.780952601134777,
"num_tokens": 40705451.0,
"step": 1230
},
{
"entropy": 0.8784732647240162,
"epoch": 1.984,
"grad_norm": 20.625,
"learning_rate": 1.3044444444444446e-05,
"loss": 14.1064,
"mean_token_accuracy": 0.776585515961051,
"num_tokens": 41040313.0,
"step": 1240
},
{
"entropy": 0.8804466765373945,
"epoch": 2.0,
"grad_norm": 18.5,
"learning_rate": 1.297037037037037e-05,
"loss": 14.0925,
"mean_token_accuracy": 0.7788621835410595,
"num_tokens": 41373608.0,
"step": 1250
},
{
"entropy": 0.8347923129796981,
"epoch": 2.016,
"grad_norm": 16.125,
"learning_rate": 1.2896296296296299e-05,
"loss": 13.3304,
"mean_token_accuracy": 0.7866924706846475,
"num_tokens": 41708226.0,
"step": 1260
},
{
"entropy": 0.8343301571905613,
"epoch": 2.032,
"grad_norm": 18.5,
"learning_rate": 1.2822222222222222e-05,
"loss": 13.426,
"mean_token_accuracy": 0.7863655511289835,
"num_tokens": 42045433.0,
"step": 1270
},
{
"entropy": 0.8626088196411729,
"epoch": 2.048,
"grad_norm": 19.75,
"learning_rate": 1.274814814814815e-05,
"loss": 13.9137,
"mean_token_accuracy": 0.7789722084999084,
"num_tokens": 42372694.0,
"step": 1280
},
{
"entropy": 0.8391540687531233,
"epoch": 2.064,
"grad_norm": 17.75,
"learning_rate": 1.2674074074074075e-05,
"loss": 13.4434,
"mean_token_accuracy": 0.7854840617626906,
"num_tokens": 42710240.0,
"step": 1290
},
{
"entropy": 0.8561890006065369,
"epoch": 2.08,
"grad_norm": 17.5,
"learning_rate": 1.2600000000000001e-05,
"loss": 13.7956,
"mean_token_accuracy": 0.7818153217434883,
"num_tokens": 43037844.0,
"step": 1300
},
{
"epoch": 2.08,
"eval_biology_entropy": 2.680125617027283,
"eval_biology_loss": 3.154866933822632,
"eval_biology_mean_token_accuracy": 0.4771236026287079,
"eval_biology_num_tokens": 43037844.0,
"eval_biology_runtime": 49.8062,
"eval_biology_samples_per_second": 10.039,
"eval_biology_steps_per_second": 2.51,
"step": 1300
},
{
"epoch": 2.08,
"eval_chemistry_entropy": 1.23085289478302,
"eval_chemistry_loss": 1.339719533920288,
"eval_chemistry_mean_token_accuracy": 0.700504720211029,
"eval_chemistry_num_tokens": 43037844.0,
"eval_chemistry_runtime": 61.7767,
"eval_chemistry_samples_per_second": 8.094,
"eval_chemistry_steps_per_second": 2.023,
"step": 1300
},
{
"epoch": 2.08,
"eval_math_entropy": 0.9530718994140625,
"eval_math_loss": 1.1914963722229004,
"eval_math_mean_token_accuracy": 0.732671552658081,
"eval_math_num_tokens": 43037844.0,
"eval_math_runtime": 63.3384,
"eval_math_samples_per_second": 7.894,
"eval_math_steps_per_second": 1.974,
"step": 1300
},
{
"epoch": 2.08,
"eval_physics_entropy": 0.863683602809906,
"eval_physics_loss": 0.898903489112854,
"eval_physics_mean_token_accuracy": 0.7743104872703552,
"eval_physics_num_tokens": 43037844.0,
"eval_physics_runtime": 72.1486,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.733,
"step": 1300
},
{
"entropy": 0.8574293557554483,
"epoch": 2.096,
"grad_norm": 19.5,
"learning_rate": 1.2525925925925928e-05,
"loss": 13.697,
"mean_token_accuracy": 0.7824351858347655,
"num_tokens": 43368675.0,
"step": 1310
},
{
"entropy": 0.8351046226918697,
"epoch": 2.112,
"grad_norm": 18.375,
"learning_rate": 1.2451851851851853e-05,
"loss": 13.452,
"mean_token_accuracy": 0.7861107390373945,
"num_tokens": 43699544.0,
"step": 1320
},
{
"entropy": 0.848756181821227,
"epoch": 2.128,
"grad_norm": 19.375,
"learning_rate": 1.237777777777778e-05,
"loss": 13.6014,
"mean_token_accuracy": 0.7825081821531057,
"num_tokens": 44012632.0,
"step": 1330
},
{
"entropy": 0.8376241635531188,
"epoch": 2.144,
"grad_norm": 17.875,
"learning_rate": 1.2303703703703704e-05,
"loss": 13.5012,
"mean_token_accuracy": 0.7842539783567191,
"num_tokens": 44349732.0,
"step": 1340
},
{
"entropy": 0.8786815196275711,
"epoch": 2.16,
"grad_norm": 19.75,
"learning_rate": 1.222962962962963e-05,
"loss": 14.1221,
"mean_token_accuracy": 0.7767074994742871,
"num_tokens": 44680410.0,
"step": 1350
},
{
"entropy": 0.8388923730701208,
"epoch": 2.176,
"grad_norm": 17.75,
"learning_rate": 1.2155555555555555e-05,
"loss": 13.4611,
"mean_token_accuracy": 0.7856471933424473,
"num_tokens": 45017922.0,
"step": 1360
},
{
"entropy": 0.8349485833197832,
"epoch": 2.192,
"grad_norm": 18.625,
"learning_rate": 1.2081481481481484e-05,
"loss": 13.3623,
"mean_token_accuracy": 0.7861445616930723,
"num_tokens": 45348179.0,
"step": 1370
},
{
"entropy": 0.8328462485224009,
"epoch": 2.208,
"grad_norm": 21.0,
"learning_rate": 1.2007407407407408e-05,
"loss": 13.3785,
"mean_token_accuracy": 0.7866237808018923,
"num_tokens": 45678397.0,
"step": 1380
},
{
"entropy": 0.8291925746947527,
"epoch": 2.224,
"grad_norm": 17.75,
"learning_rate": 1.1933333333333335e-05,
"loss": 13.328,
"mean_token_accuracy": 0.7863113507628441,
"num_tokens": 46010052.0,
"step": 1390
},
{
"entropy": 0.8287136562168598,
"epoch": 2.24,
"grad_norm": 18.0,
"learning_rate": 1.185925925925926e-05,
"loss": 13.3318,
"mean_token_accuracy": 0.7863692469894886,
"num_tokens": 46347116.0,
"step": 1400
},
{
"epoch": 2.24,
"eval_biology_entropy": 2.6612108364105223,
"eval_biology_loss": 3.153256416320801,
"eval_biology_mean_token_accuracy": 0.4783942770957947,
"eval_biology_num_tokens": 46347116.0,
"eval_biology_runtime": 49.9149,
"eval_biology_samples_per_second": 10.017,
"eval_biology_steps_per_second": 2.504,
"step": 1400
},
{
"epoch": 2.24,
"eval_chemistry_entropy": 1.2294214334487914,
"eval_chemistry_loss": 1.3362380266189575,
"eval_chemistry_mean_token_accuracy": 0.701572069644928,
"eval_chemistry_num_tokens": 46347116.0,
"eval_chemistry_runtime": 61.8559,
"eval_chemistry_samples_per_second": 8.083,
"eval_chemistry_steps_per_second": 2.021,
"step": 1400
},
{
"epoch": 2.24,
"eval_math_entropy": 0.9489249901771546,
"eval_math_loss": 1.1918082237243652,
"eval_math_mean_token_accuracy": 0.7329039392471314,
"eval_math_num_tokens": 46347116.0,
"eval_math_runtime": 63.4236,
"eval_math_samples_per_second": 7.884,
"eval_math_steps_per_second": 1.971,
"step": 1400
},
{
"epoch": 2.24,
"eval_physics_entropy": 0.8618579044342041,
"eval_physics_loss": 0.8958038687705994,
"eval_physics_mean_token_accuracy": 0.7750196180343628,
"eval_physics_num_tokens": 46347116.0,
"eval_physics_runtime": 72.3632,
"eval_physics_samples_per_second": 6.91,
"eval_physics_steps_per_second": 1.727,
"step": 1400
},
{
"entropy": 0.8701908178627491,
"epoch": 2.2560000000000002,
"grad_norm": 19.25,
"learning_rate": 1.1785185185185186e-05,
"loss": 13.9577,
"mean_token_accuracy": 0.7795143604278565,
"num_tokens": 46673074.0,
"step": 1410
},
{
"entropy": 0.8190190486609936,
"epoch": 2.2720000000000002,
"grad_norm": 18.875,
"learning_rate": 1.1711111111111113e-05,
"loss": 13.1852,
"mean_token_accuracy": 0.7890499792993069,
"num_tokens": 47009386.0,
"step": 1420
},
{
"entropy": 0.8609435971826315,
"epoch": 2.288,
"grad_norm": 19.75,
"learning_rate": 1.1637037037037037e-05,
"loss": 13.786,
"mean_token_accuracy": 0.7809262648224831,
"num_tokens": 47344231.0,
"step": 1430
},
{
"entropy": 0.8457425959408283,
"epoch": 2.304,
"grad_norm": 20.125,
"learning_rate": 1.1562962962962964e-05,
"loss": 13.6528,
"mean_token_accuracy": 0.7828251335769891,
"num_tokens": 47658392.0,
"step": 1440
},
{
"entropy": 0.8437560629099607,
"epoch": 2.32,
"grad_norm": 17.25,
"learning_rate": 1.1488888888888889e-05,
"loss": 13.4697,
"mean_token_accuracy": 0.7838904622942209,
"num_tokens": 47995015.0,
"step": 1450
},
{
"entropy": 0.8140388157218694,
"epoch": 2.336,
"grad_norm": 17.875,
"learning_rate": 1.1414814814814817e-05,
"loss": 13.1535,
"mean_token_accuracy": 0.7884997557848692,
"num_tokens": 48336708.0,
"step": 1460
},
{
"entropy": 0.8194112915545702,
"epoch": 2.352,
"grad_norm": 17.5,
"learning_rate": 1.1340740740740742e-05,
"loss": 13.0636,
"mean_token_accuracy": 0.7896034978330135,
"num_tokens": 48665311.0,
"step": 1470
},
{
"entropy": 0.8433867402374744,
"epoch": 2.368,
"grad_norm": 18.625,
"learning_rate": 1.1266666666666668e-05,
"loss": 13.5528,
"mean_token_accuracy": 0.7844488836824894,
"num_tokens": 49005334.0,
"step": 1480
},
{
"entropy": 0.8540257982909679,
"epoch": 2.384,
"grad_norm": 20.625,
"learning_rate": 1.1192592592592593e-05,
"loss": 13.7212,
"mean_token_accuracy": 0.781413323059678,
"num_tokens": 49331936.0,
"step": 1490
},
{
"entropy": 0.8498314812779426,
"epoch": 2.4,
"grad_norm": 19.25,
"learning_rate": 1.111851851851852e-05,
"loss": 13.604,
"mean_token_accuracy": 0.7828326590359211,
"num_tokens": 49671048.0,
"step": 1500
},
{
"epoch": 2.4,
"eval_biology_entropy": 2.621306725502014,
"eval_biology_loss": 3.1416707038879395,
"eval_biology_mean_token_accuracy": 0.4782486002445221,
"eval_biology_num_tokens": 49671048.0,
"eval_biology_runtime": 49.9245,
"eval_biology_samples_per_second": 10.015,
"eval_biology_steps_per_second": 2.504,
"step": 1500
},
{
"epoch": 2.4,
"eval_chemistry_entropy": 1.223756856918335,
"eval_chemistry_loss": 1.3346976041793823,
"eval_chemistry_mean_token_accuracy": 0.7012901954650879,
"eval_chemistry_num_tokens": 49671048.0,
"eval_chemistry_runtime": 61.475,
"eval_chemistry_samples_per_second": 8.133,
"eval_chemistry_steps_per_second": 2.033,
"step": 1500
},
{
"epoch": 2.4,
"eval_math_entropy": 0.9483405375480651,
"eval_math_loss": 1.19205641746521,
"eval_math_mean_token_accuracy": 0.7328525404930115,
"eval_math_num_tokens": 49671048.0,
"eval_math_runtime": 63.2408,
"eval_math_samples_per_second": 7.906,
"eval_math_steps_per_second": 1.977,
"step": 1500
},
{
"epoch": 2.4,
"eval_physics_entropy": 0.8615480012893677,
"eval_physics_loss": 0.8934000134468079,
"eval_physics_mean_token_accuracy": 0.7753945760726929,
"eval_physics_num_tokens": 49671048.0,
"eval_physics_runtime": 72.1452,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.733,
"step": 1500
}
],
"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": 4.962874969902339e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}