lpywp18n / checkpoint-1000 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
87e63d7 verified
{
"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": 0.5395457262173295,
"epoch": 0.016,
"grad_norm": 3200.0,
"learning_rate": 4.5000000000000003e-07,
"loss": 92.5745,
"mean_token_accuracy": 0.4849457062780857,
"num_tokens": 195524.0,
"step": 10
},
{
"entropy": 0.5515131525695324,
"epoch": 0.032,
"grad_norm": 2912.0,
"learning_rate": 9.500000000000001e-07,
"loss": 88.931,
"mean_token_accuracy": 0.4874983370304108,
"num_tokens": 390903.0,
"step": 20
},
{
"entropy": 0.6169477274641395,
"epoch": 0.048,
"grad_norm": 2912.0,
"learning_rate": 1.45e-06,
"loss": 78.3695,
"mean_token_accuracy": 0.4966813003644347,
"num_tokens": 589868.0,
"step": 30
},
{
"entropy": 0.8348326332867145,
"epoch": 0.064,
"grad_norm": 1160.0,
"learning_rate": 1.9500000000000004e-06,
"loss": 60.9248,
"mean_token_accuracy": 0.505222667939961,
"num_tokens": 791190.0,
"step": 40
},
{
"entropy": 1.3063700653612613,
"epoch": 0.08,
"grad_norm": 247.0,
"learning_rate": 2.4500000000000003e-06,
"loss": 46.7906,
"mean_token_accuracy": 0.5185849336907268,
"num_tokens": 989860.0,
"step": 50
},
{
"entropy": 1.959995786100626,
"epoch": 0.096,
"grad_norm": 81.0,
"learning_rate": 2.95e-06,
"loss": 39.2759,
"mean_token_accuracy": 0.5406044816598297,
"num_tokens": 1181777.0,
"step": 60
},
{
"entropy": 2.191839651763439,
"epoch": 0.112,
"grad_norm": 57.0,
"learning_rate": 3.45e-06,
"loss": 34.17,
"mean_token_accuracy": 0.5669999368488788,
"num_tokens": 1385513.0,
"step": 70
},
{
"entropy": 1.863330227136612,
"epoch": 0.128,
"grad_norm": 48.25,
"learning_rate": 3.95e-06,
"loss": 31.3942,
"mean_token_accuracy": 0.5891917455941439,
"num_tokens": 1582368.0,
"step": 80
},
{
"entropy": 1.8903757609426974,
"epoch": 0.144,
"grad_norm": 39.25,
"learning_rate": 4.450000000000001e-06,
"loss": 29.9275,
"mean_token_accuracy": 0.6045342192053795,
"num_tokens": 1773764.0,
"step": 90
},
{
"entropy": 1.8238266378641128,
"epoch": 0.16,
"grad_norm": 36.25,
"learning_rate": 4.95e-06,
"loss": 29.3315,
"mean_token_accuracy": 0.6066708620637655,
"num_tokens": 1970077.0,
"step": 100
},
{
"epoch": 0.16,
"eval_biology_entropy": 1.7597194662094116,
"eval_biology_loss": 1.708297848701477,
"eval_biology_mean_token_accuracy": 0.6260299117565155,
"eval_biology_num_tokens": 1970077.0,
"eval_biology_runtime": 36.35,
"eval_biology_samples_per_second": 13.755,
"eval_biology_steps_per_second": 3.439,
"step": 100
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 1.278447434425354,
"eval_chemistry_loss": 1.24350905418396,
"eval_chemistry_mean_token_accuracy": 0.717205593585968,
"eval_chemistry_num_tokens": 1970077.0,
"eval_chemistry_runtime": 44.9852,
"eval_chemistry_samples_per_second": 11.115,
"eval_chemistry_steps_per_second": 2.779,
"step": 100
},
{
"epoch": 0.16,
"eval_math_entropy": 0.9187946987152099,
"eval_math_loss": 1.206222414970398,
"eval_math_mean_token_accuracy": 0.7349825048446655,
"eval_math_num_tokens": 1970077.0,
"eval_math_runtime": 46.1658,
"eval_math_samples_per_second": 10.831,
"eval_math_steps_per_second": 2.708,
"step": 100
},
{
"epoch": 0.16,
"eval_physics_entropy": 1.2150983595848084,
"eval_physics_loss": 1.2188719511032104,
"eval_physics_mean_token_accuracy": 0.727249002456665,
"eval_physics_num_tokens": 1970077.0,
"eval_physics_runtime": 52.9985,
"eval_physics_samples_per_second": 9.434,
"eval_physics_steps_per_second": 2.359,
"step": 100
},
{
"entropy": 1.7616284050047397,
"epoch": 0.176,
"grad_norm": 33.25,
"learning_rate": 5.450000000000001e-06,
"loss": 27.9906,
"mean_token_accuracy": 0.6186339979991317,
"num_tokens": 2168354.0,
"step": 110
},
{
"entropy": 1.7145348988473414,
"epoch": 0.192,
"grad_norm": 36.25,
"learning_rate": 5.950000000000001e-06,
"loss": 27.271,
"mean_token_accuracy": 0.6236628353595733,
"num_tokens": 2365822.0,
"step": 120
},
{
"entropy": 1.7000788144767285,
"epoch": 0.208,
"grad_norm": 34.75,
"learning_rate": 6.450000000000001e-06,
"loss": 26.7483,
"mean_token_accuracy": 0.6299795724451542,
"num_tokens": 2558762.0,
"step": 130
},
{
"entropy": 1.6474159061908722,
"epoch": 0.224,
"grad_norm": 30.875,
"learning_rate": 6.95e-06,
"loss": 25.9891,
"mean_token_accuracy": 0.6365062229335308,
"num_tokens": 2755347.0,
"step": 140
},
{
"entropy": 1.6314032576978206,
"epoch": 0.24,
"grad_norm": 31.125,
"learning_rate": 7.450000000000001e-06,
"loss": 25.1799,
"mean_token_accuracy": 0.64104891307652,
"num_tokens": 2947346.0,
"step": 150
},
{
"entropy": 1.609092117100954,
"epoch": 0.256,
"grad_norm": 28.5,
"learning_rate": 7.950000000000002e-06,
"loss": 24.976,
"mean_token_accuracy": 0.6415446903556585,
"num_tokens": 3139957.0,
"step": 160
},
{
"entropy": 1.5253454238176345,
"epoch": 0.272,
"grad_norm": 28.875,
"learning_rate": 8.45e-06,
"loss": 24.4431,
"mean_token_accuracy": 0.6438875071704387,
"num_tokens": 3335951.0,
"step": 170
},
{
"entropy": 1.5003480702638625,
"epoch": 0.288,
"grad_norm": 31.5,
"learning_rate": 8.95e-06,
"loss": 23.9226,
"mean_token_accuracy": 0.6480991121381521,
"num_tokens": 3539731.0,
"step": 180
},
{
"entropy": 1.4835937578231095,
"epoch": 0.304,
"grad_norm": 29.375,
"learning_rate": 9.450000000000001e-06,
"loss": 23.7553,
"mean_token_accuracy": 0.6533681858330965,
"num_tokens": 3733488.0,
"step": 190
},
{
"entropy": 1.501522469520569,
"epoch": 0.32,
"grad_norm": 31.625,
"learning_rate": 9.950000000000001e-06,
"loss": 24.0263,
"mean_token_accuracy": 0.6483922835439444,
"num_tokens": 3920545.0,
"step": 200
},
{
"epoch": 0.32,
"eval_biology_entropy": 1.430639804840088,
"eval_biology_loss": 1.4240669012069702,
"eval_biology_mean_token_accuracy": 0.6643721928596497,
"eval_biology_num_tokens": 3920545.0,
"eval_biology_runtime": 36.631,
"eval_biology_samples_per_second": 13.65,
"eval_biology_steps_per_second": 3.412,
"step": 200
},
{
"epoch": 0.32,
"eval_chemistry_entropy": 1.068233790397644,
"eval_chemistry_loss": 1.0446134805679321,
"eval_chemistry_mean_token_accuracy": 0.7419922127723694,
"eval_chemistry_num_tokens": 3920545.0,
"eval_chemistry_runtime": 45.0089,
"eval_chemistry_samples_per_second": 11.109,
"eval_chemistry_steps_per_second": 2.777,
"step": 200
},
{
"epoch": 0.32,
"eval_math_entropy": 0.8016578803062439,
"eval_math_loss": 1.064239740371704,
"eval_math_mean_token_accuracy": 0.7532512550354004,
"eval_math_num_tokens": 3920545.0,
"eval_math_runtime": 46.1998,
"eval_math_samples_per_second": 10.823,
"eval_math_steps_per_second": 2.706,
"step": 200
},
{
"epoch": 0.32,
"eval_physics_entropy": 1.0390429463386535,
"eval_physics_loss": 1.033566951751709,
"eval_physics_mean_token_accuracy": 0.751745885848999,
"eval_physics_num_tokens": 3920545.0,
"eval_physics_runtime": 53.0382,
"eval_physics_samples_per_second": 9.427,
"eval_physics_steps_per_second": 2.357,
"step": 200
},
{
"entropy": 1.4409266009926796,
"epoch": 0.336,
"grad_norm": 30.5,
"learning_rate": 1.045e-05,
"loss": 23.1469,
"mean_token_accuracy": 0.6563692390918732,
"num_tokens": 4114077.0,
"step": 210
},
{
"entropy": 1.4634413696825503,
"epoch": 0.352,
"grad_norm": 29.875,
"learning_rate": 1.095e-05,
"loss": 23.4319,
"mean_token_accuracy": 0.6559967070817947,
"num_tokens": 4306949.0,
"step": 220
},
{
"entropy": 1.4390382423996926,
"epoch": 0.368,
"grad_norm": 29.5,
"learning_rate": 1.145e-05,
"loss": 23.1058,
"mean_token_accuracy": 0.6596628129482269,
"num_tokens": 4504001.0,
"step": 230
},
{
"entropy": 1.4329517252743245,
"epoch": 0.384,
"grad_norm": 30.625,
"learning_rate": 1.195e-05,
"loss": 23.0101,
"mean_token_accuracy": 0.6596732661128044,
"num_tokens": 4693812.0,
"step": 240
},
{
"entropy": 1.4356985121965409,
"epoch": 0.4,
"grad_norm": 26.0,
"learning_rate": 1.2450000000000003e-05,
"loss": 22.9403,
"mean_token_accuracy": 0.6609152596443891,
"num_tokens": 4887094.0,
"step": 250
},
{
"entropy": 1.445911981910467,
"epoch": 0.416,
"grad_norm": 28.0,
"learning_rate": 1.295e-05,
"loss": 23.2032,
"mean_token_accuracy": 0.6575265824794769,
"num_tokens": 5085369.0,
"step": 260
},
{
"entropy": 1.4276192136108876,
"epoch": 0.432,
"grad_norm": 30.25,
"learning_rate": 1.3450000000000002e-05,
"loss": 22.9807,
"mean_token_accuracy": 0.6605408634990454,
"num_tokens": 5271275.0,
"step": 270
},
{
"entropy": 1.4254673644900322,
"epoch": 0.448,
"grad_norm": 32.0,
"learning_rate": 1.3950000000000002e-05,
"loss": 22.9563,
"mean_token_accuracy": 0.6602664466947317,
"num_tokens": 5460559.0,
"step": 280
},
{
"entropy": 1.3786304764449597,
"epoch": 0.464,
"grad_norm": 28.375,
"learning_rate": 1.4450000000000002e-05,
"loss": 22.0498,
"mean_token_accuracy": 0.6699710045009851,
"num_tokens": 5653809.0,
"step": 290
},
{
"entropy": 1.3980159699916839,
"epoch": 0.48,
"grad_norm": 26.0,
"learning_rate": 1.4950000000000003e-05,
"loss": 22.5067,
"mean_token_accuracy": 0.6643361520022154,
"num_tokens": 5850176.0,
"step": 300
},
{
"epoch": 0.48,
"eval_biology_entropy": 1.378262495994568,
"eval_biology_loss": 1.34170663356781,
"eval_biology_mean_token_accuracy": 0.6778094511032104,
"eval_biology_num_tokens": 5850176.0,
"eval_biology_runtime": 36.5741,
"eval_biology_samples_per_second": 13.671,
"eval_biology_steps_per_second": 3.418,
"step": 300
},
{
"epoch": 0.48,
"eval_chemistry_entropy": 1.0265604996681212,
"eval_chemistry_loss": 0.9883458018302917,
"eval_chemistry_mean_token_accuracy": 0.7520135464668274,
"eval_chemistry_num_tokens": 5850176.0,
"eval_chemistry_runtime": 44.9771,
"eval_chemistry_samples_per_second": 11.117,
"eval_chemistry_steps_per_second": 2.779,
"step": 300
},
{
"epoch": 0.48,
"eval_math_entropy": 0.7964297127723694,
"eval_math_loss": 1.0260037183761597,
"eval_math_mean_token_accuracy": 0.759199161529541,
"eval_math_num_tokens": 5850176.0,
"eval_math_runtime": 46.2056,
"eval_math_samples_per_second": 10.821,
"eval_math_steps_per_second": 2.705,
"step": 300
},
{
"epoch": 0.48,
"eval_physics_entropy": 1.0048614959716797,
"eval_physics_loss": 0.9760251045227051,
"eval_physics_mean_token_accuracy": 0.761619930267334,
"eval_physics_num_tokens": 5850176.0,
"eval_physics_runtime": 53.0249,
"eval_physics_samples_per_second": 9.43,
"eval_physics_steps_per_second": 2.357,
"step": 300
},
{
"entropy": 1.3288711309432983,
"epoch": 0.496,
"grad_norm": 26.375,
"learning_rate": 1.545e-05,
"loss": 21.2235,
"mean_token_accuracy": 0.6774784430861474,
"num_tokens": 6046503.0,
"step": 310
},
{
"entropy": 1.385315388813615,
"epoch": 0.512,
"grad_norm": 27.125,
"learning_rate": 1.595e-05,
"loss": 22.2247,
"mean_token_accuracy": 0.6680108610540628,
"num_tokens": 6240456.0,
"step": 320
},
{
"entropy": 1.3709059737622737,
"epoch": 0.528,
"grad_norm": 24.75,
"learning_rate": 1.645e-05,
"loss": 22.1079,
"mean_token_accuracy": 0.6704635102301836,
"num_tokens": 6430555.0,
"step": 330
},
{
"entropy": 1.3751415818929673,
"epoch": 0.544,
"grad_norm": 26.25,
"learning_rate": 1.6950000000000002e-05,
"loss": 21.961,
"mean_token_accuracy": 0.6710947025567293,
"num_tokens": 6626006.0,
"step": 340
},
{
"entropy": 1.3602795064449311,
"epoch": 0.56,
"grad_norm": 30.625,
"learning_rate": 1.7450000000000004e-05,
"loss": 21.9474,
"mean_token_accuracy": 0.6700355879962444,
"num_tokens": 6820754.0,
"step": 350
},
{
"entropy": 1.344116109609604,
"epoch": 0.576,
"grad_norm": 26.375,
"learning_rate": 1.795e-05,
"loss": 21.5572,
"mean_token_accuracy": 0.6739088613539934,
"num_tokens": 7021844.0,
"step": 360
},
{
"entropy": 1.3490931887179614,
"epoch": 0.592,
"grad_norm": 26.125,
"learning_rate": 1.845e-05,
"loss": 21.7554,
"mean_token_accuracy": 0.6715496938675642,
"num_tokens": 7213951.0,
"step": 370
},
{
"entropy": 1.3264936488121748,
"epoch": 0.608,
"grad_norm": 27.25,
"learning_rate": 1.8950000000000003e-05,
"loss": 21.3421,
"mean_token_accuracy": 0.6760978367179632,
"num_tokens": 7416773.0,
"step": 380
},
{
"entropy": 1.348380234092474,
"epoch": 0.624,
"grad_norm": 28.125,
"learning_rate": 1.9450000000000002e-05,
"loss": 21.6293,
"mean_token_accuracy": 0.6746726848185063,
"num_tokens": 7612843.0,
"step": 390
},
{
"entropy": 1.3465636018663645,
"epoch": 0.64,
"grad_norm": 26.25,
"learning_rate": 1.9950000000000004e-05,
"loss": 21.6324,
"mean_token_accuracy": 0.6738044265657663,
"num_tokens": 7801633.0,
"step": 400
},
{
"epoch": 0.64,
"eval_biology_entropy": 1.3016406693458558,
"eval_biology_loss": 1.2880380153656006,
"eval_biology_mean_token_accuracy": 0.6870926094055175,
"eval_biology_num_tokens": 7801633.0,
"eval_biology_runtime": 36.6216,
"eval_biology_samples_per_second": 13.653,
"eval_biology_steps_per_second": 3.413,
"step": 400
},
{
"epoch": 0.64,
"eval_chemistry_entropy": 0.9742186889648438,
"eval_chemistry_loss": 0.954098641872406,
"eval_chemistry_mean_token_accuracy": 0.7588451662063599,
"eval_chemistry_num_tokens": 7801633.0,
"eval_chemistry_runtime": 44.9361,
"eval_chemistry_samples_per_second": 11.127,
"eval_chemistry_steps_per_second": 2.782,
"step": 400
},
{
"epoch": 0.64,
"eval_math_entropy": 0.7798485422134399,
"eval_math_loss": 1.0040466785430908,
"eval_math_mean_token_accuracy": 0.7620619239807129,
"eval_math_num_tokens": 7801633.0,
"eval_math_runtime": 46.2072,
"eval_math_samples_per_second": 10.821,
"eval_math_steps_per_second": 2.705,
"step": 400
},
{
"epoch": 0.64,
"eval_physics_entropy": 0.9594037117958069,
"eval_physics_loss": 0.9417626261711121,
"eval_physics_mean_token_accuracy": 0.7674873929023742,
"eval_physics_num_tokens": 7801633.0,
"eval_physics_runtime": 53.0463,
"eval_physics_samples_per_second": 9.426,
"eval_physics_steps_per_second": 2.356,
"step": 400
},
{
"entropy": 1.3527625199407338,
"epoch": 0.656,
"grad_norm": 27.875,
"learning_rate": 1.9950000000000004e-05,
"loss": 21.651,
"mean_token_accuracy": 0.6724541377276182,
"num_tokens": 7995843.0,
"step": 410
},
{
"entropy": 1.3500655848532914,
"epoch": 0.672,
"grad_norm": 27.125,
"learning_rate": 1.9894444444444446e-05,
"loss": 21.6896,
"mean_token_accuracy": 0.673105026781559,
"num_tokens": 8183103.0,
"step": 420
},
{
"entropy": 1.3231504444032907,
"epoch": 0.688,
"grad_norm": 23.5,
"learning_rate": 1.983888888888889e-05,
"loss": 21.3727,
"mean_token_accuracy": 0.6781029276549816,
"num_tokens": 8385976.0,
"step": 430
},
{
"entropy": 1.3333053421229124,
"epoch": 0.704,
"grad_norm": 23.5,
"learning_rate": 1.9783333333333334e-05,
"loss": 21.2722,
"mean_token_accuracy": 0.6781596153974533,
"num_tokens": 8578431.0,
"step": 440
},
{
"entropy": 1.2669135373085738,
"epoch": 0.72,
"grad_norm": 23.375,
"learning_rate": 1.972777777777778e-05,
"loss": 20.3907,
"mean_token_accuracy": 0.6872118715196848,
"num_tokens": 8781342.0,
"step": 450
},
{
"entropy": 1.2877357687801123,
"epoch": 0.736,
"grad_norm": 23.875,
"learning_rate": 1.9672222222222222e-05,
"loss": 20.6999,
"mean_token_accuracy": 0.6825123306363821,
"num_tokens": 8977918.0,
"step": 460
},
{
"entropy": 1.3120017491281033,
"epoch": 0.752,
"grad_norm": 24.5,
"learning_rate": 1.9616666666666668e-05,
"loss": 21.1222,
"mean_token_accuracy": 0.6796863108873368,
"num_tokens": 9169322.0,
"step": 470
},
{
"entropy": 1.263747826963663,
"epoch": 0.768,
"grad_norm": 26.375,
"learning_rate": 1.9561111111111114e-05,
"loss": 20.3726,
"mean_token_accuracy": 0.6869186259806156,
"num_tokens": 9368141.0,
"step": 480
},
{
"entropy": 1.3065229691565037,
"epoch": 0.784,
"grad_norm": 26.5,
"learning_rate": 1.9505555555555556e-05,
"loss": 20.8536,
"mean_token_accuracy": 0.683413552865386,
"num_tokens": 9565236.0,
"step": 490
},
{
"entropy": 1.3189398009330033,
"epoch": 0.8,
"grad_norm": 25.5,
"learning_rate": 1.9450000000000002e-05,
"loss": 21.2576,
"mean_token_accuracy": 0.6789898939430714,
"num_tokens": 9761227.0,
"step": 500
},
{
"epoch": 0.8,
"eval_biology_entropy": 1.2551557726860045,
"eval_biology_loss": 1.2525163888931274,
"eval_biology_mean_token_accuracy": 0.6931379227638245,
"eval_biology_num_tokens": 9761227.0,
"eval_biology_runtime": 36.6065,
"eval_biology_samples_per_second": 13.659,
"eval_biology_steps_per_second": 3.415,
"step": 500
},
{
"epoch": 0.8,
"eval_chemistry_entropy": 0.9375595455169677,
"eval_chemistry_loss": 0.9276740550994873,
"eval_chemistry_mean_token_accuracy": 0.7639822545051574,
"eval_chemistry_num_tokens": 9761227.0,
"eval_chemistry_runtime": 44.5343,
"eval_chemistry_samples_per_second": 11.227,
"eval_chemistry_steps_per_second": 2.807,
"step": 500
},
{
"epoch": 0.8,
"eval_math_entropy": 0.7748257055282592,
"eval_math_loss": 0.9847215414047241,
"eval_math_mean_token_accuracy": 0.7653367323875427,
"eval_math_num_tokens": 9761227.0,
"eval_math_runtime": 45.9419,
"eval_math_samples_per_second": 10.883,
"eval_math_steps_per_second": 2.721,
"step": 500
},
{
"epoch": 0.8,
"eval_physics_entropy": 0.9384980492591858,
"eval_physics_loss": 0.9171136021614075,
"eval_physics_mean_token_accuracy": 0.7720685515403748,
"eval_physics_num_tokens": 9761227.0,
"eval_physics_runtime": 52.888,
"eval_physics_samples_per_second": 9.454,
"eval_physics_steps_per_second": 2.363,
"step": 500
},
{
"entropy": 1.2767818640917539,
"epoch": 0.816,
"grad_norm": 25.5,
"learning_rate": 1.9394444444444448e-05,
"loss": 20.4527,
"mean_token_accuracy": 0.6852765031158924,
"num_tokens": 9958727.0,
"step": 510
},
{
"entropy": 1.2603609178215265,
"epoch": 0.832,
"grad_norm": 23.0,
"learning_rate": 1.933888888888889e-05,
"loss": 20.3411,
"mean_token_accuracy": 0.6862579733133316,
"num_tokens": 10155771.0,
"step": 520
},
{
"entropy": 1.2630852952599525,
"epoch": 0.848,
"grad_norm": 26.375,
"learning_rate": 1.9283333333333332e-05,
"loss": 20.316,
"mean_token_accuracy": 0.6886717110872269,
"num_tokens": 10357721.0,
"step": 530
},
{
"entropy": 1.2910415723919868,
"epoch": 0.864,
"grad_norm": 25.0,
"learning_rate": 1.9227777777777778e-05,
"loss": 20.7324,
"mean_token_accuracy": 0.6839874558150768,
"num_tokens": 10552495.0,
"step": 540
},
{
"entropy": 1.2793519891798497,
"epoch": 0.88,
"grad_norm": 26.625,
"learning_rate": 1.9172222222222224e-05,
"loss": 20.4504,
"mean_token_accuracy": 0.6864676021039486,
"num_tokens": 10748749.0,
"step": 550
},
{
"entropy": 1.2906367748975753,
"epoch": 0.896,
"grad_norm": 24.875,
"learning_rate": 1.911666666666667e-05,
"loss": 20.7703,
"mean_token_accuracy": 0.6835371732711792,
"num_tokens": 10943319.0,
"step": 560
},
{
"entropy": 1.2809050466865302,
"epoch": 0.912,
"grad_norm": 26.0,
"learning_rate": 1.9061111111111112e-05,
"loss": 20.5319,
"mean_token_accuracy": 0.6867790374904871,
"num_tokens": 11136935.0,
"step": 570
},
{
"entropy": 1.283418832719326,
"epoch": 0.928,
"grad_norm": 23.875,
"learning_rate": 1.9005555555555557e-05,
"loss": 20.6537,
"mean_token_accuracy": 0.6832854464650154,
"num_tokens": 11331098.0,
"step": 580
},
{
"entropy": 1.2303649988025427,
"epoch": 0.944,
"grad_norm": 21.5,
"learning_rate": 1.8950000000000003e-05,
"loss": 19.7901,
"mean_token_accuracy": 0.6927386663854123,
"num_tokens": 11530550.0,
"step": 590
},
{
"entropy": 1.2655960384756326,
"epoch": 0.96,
"grad_norm": 26.125,
"learning_rate": 1.8894444444444446e-05,
"loss": 20.362,
"mean_token_accuracy": 0.6886723652482033,
"num_tokens": 11729645.0,
"step": 600
},
{
"epoch": 0.96,
"eval_biology_entropy": 1.2511886906623841,
"eval_biology_loss": 1.2273433208465576,
"eval_biology_mean_token_accuracy": 0.6978591928482055,
"eval_biology_num_tokens": 11729645.0,
"eval_biology_runtime": 36.8245,
"eval_biology_samples_per_second": 13.578,
"eval_biology_steps_per_second": 3.394,
"step": 600
},
{
"epoch": 0.96,
"eval_chemistry_entropy": 0.9398184275627136,
"eval_chemistry_loss": 0.9139939546585083,
"eval_chemistry_mean_token_accuracy": 0.7667500405311585,
"eval_chemistry_num_tokens": 11729645.0,
"eval_chemistry_runtime": 45.002,
"eval_chemistry_samples_per_second": 11.111,
"eval_chemistry_steps_per_second": 2.778,
"step": 600
},
{
"epoch": 0.96,
"eval_math_entropy": 0.7878352689743042,
"eval_math_loss": 0.9694741368293762,
"eval_math_mean_token_accuracy": 0.767361653804779,
"eval_math_num_tokens": 11729645.0,
"eval_math_runtime": 46.2016,
"eval_math_samples_per_second": 10.822,
"eval_math_steps_per_second": 2.706,
"step": 600
},
{
"epoch": 0.96,
"eval_physics_entropy": 0.9437616720199585,
"eval_physics_loss": 0.9044761061668396,
"eval_physics_mean_token_accuracy": 0.7742767133712769,
"eval_physics_num_tokens": 11729645.0,
"eval_physics_runtime": 53.0008,
"eval_physics_samples_per_second": 9.434,
"eval_physics_steps_per_second": 2.358,
"step": 600
},
{
"entropy": 1.2491024732589722,
"epoch": 0.976,
"grad_norm": 25.875,
"learning_rate": 1.883888888888889e-05,
"loss": 19.9555,
"mean_token_accuracy": 0.6900846060365439,
"num_tokens": 11924644.0,
"step": 610
},
{
"entropy": 1.2545549362897872,
"epoch": 0.992,
"grad_norm": 23.0,
"learning_rate": 1.8783333333333337e-05,
"loss": 20.317,
"mean_token_accuracy": 0.686628095433116,
"num_tokens": 12123059.0,
"step": 620
},
{
"entropy": 1.2396365478634834,
"epoch": 1.008,
"grad_norm": 22.5,
"learning_rate": 1.872777777777778e-05,
"loss": 19.6736,
"mean_token_accuracy": 0.6955156117677689,
"num_tokens": 12319366.0,
"step": 630
},
{
"entropy": 1.143751036375761,
"epoch": 1.024,
"grad_norm": 24.5,
"learning_rate": 1.8672222222222222e-05,
"loss": 18.51,
"mean_token_accuracy": 0.7079542689025402,
"num_tokens": 12524183.0,
"step": 640
},
{
"entropy": 1.1911311563104392,
"epoch": 1.04,
"grad_norm": 25.25,
"learning_rate": 1.8616666666666667e-05,
"loss": 19.0515,
"mean_token_accuracy": 0.6993662625551224,
"num_tokens": 12718593.0,
"step": 650
},
{
"entropy": 1.1548987552523613,
"epoch": 1.056,
"grad_norm": 25.0,
"learning_rate": 1.8561111111111113e-05,
"loss": 18.5231,
"mean_token_accuracy": 0.707820488885045,
"num_tokens": 12917803.0,
"step": 660
},
{
"entropy": 1.1886027652770281,
"epoch": 1.072,
"grad_norm": 26.0,
"learning_rate": 1.8505555555555556e-05,
"loss": 18.9314,
"mean_token_accuracy": 0.7034936733543873,
"num_tokens": 13105826.0,
"step": 670
},
{
"entropy": 1.1603233687579633,
"epoch": 1.088,
"grad_norm": 24.5,
"learning_rate": 1.845e-05,
"loss": 18.681,
"mean_token_accuracy": 0.7059972729533911,
"num_tokens": 13298619.0,
"step": 680
},
{
"entropy": 1.1554178480058908,
"epoch": 1.104,
"grad_norm": 24.0,
"learning_rate": 1.8394444444444447e-05,
"loss": 18.6001,
"mean_token_accuracy": 0.7077722396701575,
"num_tokens": 13491486.0,
"step": 690
},
{
"entropy": 1.1779750995337963,
"epoch": 1.12,
"grad_norm": 26.125,
"learning_rate": 1.833888888888889e-05,
"loss": 18.886,
"mean_token_accuracy": 0.7039439618587494,
"num_tokens": 13674663.0,
"step": 700
},
{
"epoch": 1.12,
"eval_biology_entropy": 1.115936914920807,
"eval_biology_loss": 1.2197552919387817,
"eval_biology_mean_token_accuracy": 0.6998812022209168,
"eval_biology_num_tokens": 13674663.0,
"eval_biology_runtime": 36.6382,
"eval_biology_samples_per_second": 13.647,
"eval_biology_steps_per_second": 3.412,
"step": 700
},
{
"epoch": 1.12,
"eval_chemistry_entropy": 0.8371734113693238,
"eval_chemistry_loss": 0.9145113229751587,
"eval_chemistry_mean_token_accuracy": 0.7663986759185791,
"eval_chemistry_num_tokens": 13674663.0,
"eval_chemistry_runtime": 45.0064,
"eval_chemistry_samples_per_second": 11.11,
"eval_chemistry_steps_per_second": 2.777,
"step": 700
},
{
"epoch": 1.12,
"eval_math_entropy": 0.7305963957309722,
"eval_math_loss": 0.9764987826347351,
"eval_math_mean_token_accuracy": 0.766656831741333,
"eval_math_num_tokens": 13674663.0,
"eval_math_runtime": 46.2254,
"eval_math_samples_per_second": 10.817,
"eval_math_steps_per_second": 2.704,
"step": 700
},
{
"epoch": 1.12,
"eval_physics_entropy": 0.852522144317627,
"eval_physics_loss": 0.9007405638694763,
"eval_physics_mean_token_accuracy": 0.7750241093635559,
"eval_physics_num_tokens": 13674663.0,
"eval_physics_runtime": 53.0213,
"eval_physics_samples_per_second": 9.43,
"eval_physics_steps_per_second": 2.358,
"step": 700
},
{
"entropy": 1.1570473805069923,
"epoch": 1.1360000000000001,
"grad_norm": 30.875,
"learning_rate": 1.8283333333333335e-05,
"loss": 18.7252,
"mean_token_accuracy": 0.7053195346146822,
"num_tokens": 13869134.0,
"step": 710
},
{
"entropy": 1.153097653761506,
"epoch": 1.152,
"grad_norm": 23.125,
"learning_rate": 1.822777777777778e-05,
"loss": 18.3474,
"mean_token_accuracy": 0.7079035714268684,
"num_tokens": 14078365.0,
"step": 720
},
{
"entropy": 1.1558022141456603,
"epoch": 1.168,
"grad_norm": 25.375,
"learning_rate": 1.8172222222222223e-05,
"loss": 18.6158,
"mean_token_accuracy": 0.706475806236267,
"num_tokens": 14266831.0,
"step": 730
},
{
"entropy": 1.1689991034567355,
"epoch": 1.184,
"grad_norm": 24.625,
"learning_rate": 1.811666666666667e-05,
"loss": 18.7482,
"mean_token_accuracy": 0.7053946666419506,
"num_tokens": 14465660.0,
"step": 740
},
{
"entropy": 1.1589057791978121,
"epoch": 1.2,
"grad_norm": 26.125,
"learning_rate": 1.806111111111111e-05,
"loss": 18.4761,
"mean_token_accuracy": 0.7090217288583517,
"num_tokens": 14653228.0,
"step": 750
},
{
"entropy": 1.1286497369408608,
"epoch": 1.216,
"grad_norm": 24.625,
"learning_rate": 1.8005555555555557e-05,
"loss": 18.0937,
"mean_token_accuracy": 0.710760310664773,
"num_tokens": 14857782.0,
"step": 760
},
{
"entropy": 1.1454383581876755,
"epoch": 1.232,
"grad_norm": 23.0,
"learning_rate": 1.795e-05,
"loss": 18.3666,
"mean_token_accuracy": 0.7111793085932732,
"num_tokens": 15047356.0,
"step": 770
},
{
"entropy": 1.1489330865442753,
"epoch": 1.248,
"grad_norm": 24.5,
"learning_rate": 1.7894444444444445e-05,
"loss": 18.4807,
"mean_token_accuracy": 0.7070541866123676,
"num_tokens": 15241098.0,
"step": 780
},
{
"entropy": 1.1616758443415165,
"epoch": 1.264,
"grad_norm": 23.0,
"learning_rate": 1.783888888888889e-05,
"loss": 18.6418,
"mean_token_accuracy": 0.7069107063114644,
"num_tokens": 15437657.0,
"step": 790
},
{
"entropy": 1.1660578336566687,
"epoch": 1.28,
"grad_norm": 26.875,
"learning_rate": 1.7783333333333333e-05,
"loss": 18.8491,
"mean_token_accuracy": 0.7026976983994245,
"num_tokens": 15630795.0,
"step": 800
},
{
"epoch": 1.28,
"eval_biology_entropy": 1.1199735798835755,
"eval_biology_loss": 1.2088388204574585,
"eval_biology_mean_token_accuracy": 0.7015871267318725,
"eval_biology_num_tokens": 15630795.0,
"eval_biology_runtime": 36.9629,
"eval_biology_samples_per_second": 13.527,
"eval_biology_steps_per_second": 3.382,
"step": 800
},
{
"epoch": 1.28,
"eval_chemistry_entropy": 0.8543037304878235,
"eval_chemistry_loss": 0.9098371863365173,
"eval_chemistry_mean_token_accuracy": 0.7672378678321838,
"eval_chemistry_num_tokens": 15630795.0,
"eval_chemistry_runtime": 44.9998,
"eval_chemistry_samples_per_second": 11.111,
"eval_chemistry_steps_per_second": 2.778,
"step": 800
},
{
"epoch": 1.28,
"eval_math_entropy": 0.7368179533481598,
"eval_math_loss": 0.9706032872200012,
"eval_math_mean_token_accuracy": 0.7682779974937439,
"eval_math_num_tokens": 15630795.0,
"eval_math_runtime": 46.2307,
"eval_math_samples_per_second": 10.815,
"eval_math_steps_per_second": 2.704,
"step": 800
},
{
"epoch": 1.28,
"eval_physics_entropy": 0.8641948227882386,
"eval_physics_loss": 0.8950499296188354,
"eval_physics_mean_token_accuracy": 0.7758663225173951,
"eval_physics_num_tokens": 15630795.0,
"eval_physics_runtime": 53.0442,
"eval_physics_samples_per_second": 9.426,
"eval_physics_steps_per_second": 2.357,
"step": 800
},
{
"entropy": 1.1806952260434627,
"epoch": 1.296,
"grad_norm": 26.125,
"learning_rate": 1.772777777777778e-05,
"loss": 18.9295,
"mean_token_accuracy": 0.70291638225317,
"num_tokens": 15827105.0,
"step": 810
},
{
"entropy": 1.1602427676320075,
"epoch": 1.312,
"grad_norm": 27.625,
"learning_rate": 1.7672222222222224e-05,
"loss": 18.7911,
"mean_token_accuracy": 0.7041201893240213,
"num_tokens": 16019645.0,
"step": 820
},
{
"entropy": 1.1715985681861638,
"epoch": 1.328,
"grad_norm": 24.375,
"learning_rate": 1.761666666666667e-05,
"loss": 18.8262,
"mean_token_accuracy": 0.7034816723316908,
"num_tokens": 16221726.0,
"step": 830
},
{
"entropy": 1.1461844369769096,
"epoch": 1.3439999999999999,
"grad_norm": 25.5,
"learning_rate": 1.7561111111111113e-05,
"loss": 18.2642,
"mean_token_accuracy": 0.7110818449407816,
"num_tokens": 16427594.0,
"step": 840
},
{
"entropy": 1.1582913786172866,
"epoch": 1.3599999999999999,
"grad_norm": 25.25,
"learning_rate": 1.7505555555555558e-05,
"loss": 18.5772,
"mean_token_accuracy": 0.7078793905675411,
"num_tokens": 16621605.0,
"step": 850
},
{
"entropy": 1.1942507456988096,
"epoch": 1.376,
"grad_norm": 27.875,
"learning_rate": 1.7450000000000004e-05,
"loss": 19.0764,
"mean_token_accuracy": 0.701082956790924,
"num_tokens": 16813444.0,
"step": 860
},
{
"entropy": 1.1594038642942905,
"epoch": 1.392,
"grad_norm": 24.125,
"learning_rate": 1.7394444444444446e-05,
"loss": 18.7411,
"mean_token_accuracy": 0.7061580993235111,
"num_tokens": 17006509.0,
"step": 870
},
{
"entropy": 1.1641225919127465,
"epoch": 1.408,
"grad_norm": 21.5,
"learning_rate": 1.733888888888889e-05,
"loss": 18.5028,
"mean_token_accuracy": 0.7065019059926272,
"num_tokens": 17197870.0,
"step": 880
},
{
"entropy": 1.153414449468255,
"epoch": 1.424,
"grad_norm": 24.125,
"learning_rate": 1.7283333333333334e-05,
"loss": 18.5571,
"mean_token_accuracy": 0.7072413295507431,
"num_tokens": 17394390.0,
"step": 890
},
{
"entropy": 1.1581655487418174,
"epoch": 1.44,
"grad_norm": 28.5,
"learning_rate": 1.722777777777778e-05,
"loss": 18.5762,
"mean_token_accuracy": 0.7053147312253714,
"num_tokens": 17587777.0,
"step": 900
},
{
"epoch": 1.44,
"eval_biology_entropy": 1.1328148250579835,
"eval_biology_loss": 1.1959257125854492,
"eval_biology_mean_token_accuracy": 0.7035633249282837,
"eval_biology_num_tokens": 17587777.0,
"eval_biology_runtime": 38.7737,
"eval_biology_samples_per_second": 12.895,
"eval_biology_steps_per_second": 3.224,
"step": 900
},
{
"epoch": 1.44,
"eval_chemistry_entropy": 0.8504280371665954,
"eval_chemistry_loss": 0.9063042402267456,
"eval_chemistry_mean_token_accuracy": 0.7678593263626099,
"eval_chemistry_num_tokens": 17587777.0,
"eval_chemistry_runtime": 44.9861,
"eval_chemistry_samples_per_second": 11.115,
"eval_chemistry_steps_per_second": 2.779,
"step": 900
},
{
"epoch": 1.44,
"eval_math_entropy": 0.7459685401916504,
"eval_math_loss": 0.9690774083137512,
"eval_math_mean_token_accuracy": 0.7676572284698486,
"eval_math_num_tokens": 17587777.0,
"eval_math_runtime": 82.6785,
"eval_math_samples_per_second": 6.048,
"eval_math_steps_per_second": 1.512,
"step": 900
},
{
"epoch": 1.44,
"eval_physics_entropy": 0.8569722015857697,
"eval_physics_loss": 0.8932082653045654,
"eval_physics_mean_token_accuracy": 0.7766451034545898,
"eval_physics_num_tokens": 17587777.0,
"eval_physics_runtime": 52.7335,
"eval_physics_samples_per_second": 9.482,
"eval_physics_steps_per_second": 2.37,
"step": 900
},
{
"entropy": 1.1505886979401112,
"epoch": 1.456,
"grad_norm": 24.25,
"learning_rate": 1.7172222222222223e-05,
"loss": 18.5378,
"mean_token_accuracy": 0.7084831334650517,
"num_tokens": 17788456.0,
"step": 910
},
{
"entropy": 1.1548077858984471,
"epoch": 1.472,
"grad_norm": 24.125,
"learning_rate": 1.7116666666666668e-05,
"loss": 18.4342,
"mean_token_accuracy": 0.7085961502045393,
"num_tokens": 17984063.0,
"step": 920
},
{
"entropy": 1.1482487738132476,
"epoch": 1.488,
"grad_norm": 24.0,
"learning_rate": 1.7061111111111114e-05,
"loss": 18.4515,
"mean_token_accuracy": 0.7099530510604382,
"num_tokens": 18175640.0,
"step": 930
},
{
"entropy": 1.1528287250548601,
"epoch": 1.504,
"grad_norm": 23.375,
"learning_rate": 1.7005555555555556e-05,
"loss": 18.5572,
"mean_token_accuracy": 0.7075925685465336,
"num_tokens": 18367857.0,
"step": 940
},
{
"entropy": 1.1533676456660031,
"epoch": 1.52,
"grad_norm": 23.5,
"learning_rate": 1.6950000000000002e-05,
"loss": 18.4716,
"mean_token_accuracy": 0.7079793553799391,
"num_tokens": 18569146.0,
"step": 950
},
{
"entropy": 1.1715267714112998,
"epoch": 1.536,
"grad_norm": 26.25,
"learning_rate": 1.6894444444444448e-05,
"loss": 18.754,
"mean_token_accuracy": 0.7050810724496841,
"num_tokens": 18755079.0,
"step": 960
},
{
"entropy": 1.1350098505616188,
"epoch": 1.552,
"grad_norm": 26.0,
"learning_rate": 1.683888888888889e-05,
"loss": 18.4393,
"mean_token_accuracy": 0.7091891258955002,
"num_tokens": 18956248.0,
"step": 970
},
{
"entropy": 1.1361894980072975,
"epoch": 1.568,
"grad_norm": 25.625,
"learning_rate": 1.6783333333333332e-05,
"loss": 18.2272,
"mean_token_accuracy": 0.711289395019412,
"num_tokens": 19150315.0,
"step": 980
},
{
"entropy": 1.1439239967614412,
"epoch": 1.584,
"grad_norm": 26.125,
"learning_rate": 1.6727777777777778e-05,
"loss": 18.3302,
"mean_token_accuracy": 0.7097273994237184,
"num_tokens": 19344260.0,
"step": 990
},
{
"entropy": 1.1727315995842218,
"epoch": 1.6,
"grad_norm": 23.625,
"learning_rate": 1.6672222222222224e-05,
"loss": 18.7555,
"mean_token_accuracy": 0.7035728722810746,
"num_tokens": 19532552.0,
"step": 1000
},
{
"epoch": 1.6,
"eval_biology_entropy": 1.0895025281906128,
"eval_biology_loss": 1.1883716583251953,
"eval_biology_mean_token_accuracy": 0.705087408065796,
"eval_biology_num_tokens": 19532552.0,
"eval_biology_runtime": 36.662,
"eval_biology_samples_per_second": 13.638,
"eval_biology_steps_per_second": 3.41,
"step": 1000
},
{
"epoch": 1.6,
"eval_chemistry_entropy": 0.8066284563541413,
"eval_chemistry_loss": 0.9020781517028809,
"eval_chemistry_mean_token_accuracy": 0.7695949592590332,
"eval_chemistry_num_tokens": 19532552.0,
"eval_chemistry_runtime": 44.602,
"eval_chemistry_samples_per_second": 11.21,
"eval_chemistry_steps_per_second": 2.803,
"step": 1000
},
{
"epoch": 1.6,
"eval_math_entropy": 0.6928402831554413,
"eval_math_loss": 0.9739019870758057,
"eval_math_mean_token_accuracy": 0.7690589771270752,
"eval_math_num_tokens": 19532552.0,
"eval_math_runtime": 46.1135,
"eval_math_samples_per_second": 10.843,
"eval_math_steps_per_second": 2.711,
"step": 1000
},
{
"epoch": 1.6,
"eval_physics_entropy": 0.8003968651294708,
"eval_physics_loss": 0.889771044254303,
"eval_physics_mean_token_accuracy": 0.7777439093589783,
"eval_physics_num_tokens": 19532552.0,
"eval_physics_runtime": 53.0769,
"eval_physics_samples_per_second": 9.42,
"eval_physics_steps_per_second": 2.355,
"step": 1000
}
],
"logging_steps": 10,
"max_steps": 4000,
"num_input_tokens_seen": 0,
"num_train_epochs": 7,
"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": 2.2020838985302374e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}