{ "best_global_step": null, "best_metric": null, "best_model_checkpoint": null, "epoch": 1.6, "eval_steps": 100, "global_step": 1000, "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 } ], "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": 3.3005113980506685e+18, "train_batch_size": 4, "trial_name": null, "trial_params": null }