b5e92jps / checkpoint-1000 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
facfa87 verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.4451121126133645,
"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.7622424783185124,
"epoch": 0.004451121126133645,
"grad_norm": 1888.0,
"learning_rate": 1.8e-07,
"loss": 36.8185,
"mean_token_accuracy": 0.6663544788956642,
"num_tokens": 63133.0,
"step": 10
},
{
"entropy": 0.8160548411309719,
"epoch": 0.00890224225226729,
"grad_norm": 1728.0,
"learning_rate": 3.8e-07,
"loss": 36.6381,
"mean_token_accuracy": 0.649208546616137,
"num_tokens": 128968.0,
"step": 20
},
{
"entropy": 0.8332255953922868,
"epoch": 0.013353363378400934,
"grad_norm": 948.0,
"learning_rate": 5.800000000000001e-07,
"loss": 34.7797,
"mean_token_accuracy": 0.6578839531168341,
"num_tokens": 193314.0,
"step": 30
},
{
"entropy": 0.9456521481275558,
"epoch": 0.01780448450453458,
"grad_norm": 716.0,
"learning_rate": 7.8e-07,
"loss": 34.1076,
"mean_token_accuracy": 0.6613732609897852,
"num_tokens": 257189.0,
"step": 40
},
{
"entropy": 0.9749895730987191,
"epoch": 0.022255605630668224,
"grad_norm": 660.0,
"learning_rate": 9.800000000000001e-07,
"loss": 29.937,
"mean_token_accuracy": 0.6729008939117194,
"num_tokens": 324224.0,
"step": 50
},
{
"entropy": 1.0305172327905894,
"epoch": 0.026706726756801868,
"grad_norm": 492.0,
"learning_rate": 1.1800000000000001e-06,
"loss": 28.3561,
"mean_token_accuracy": 0.680633837915957,
"num_tokens": 394654.0,
"step": 60
},
{
"entropy": 1.0123192116618156,
"epoch": 0.031157847882935515,
"grad_norm": 404.0,
"learning_rate": 1.3800000000000001e-06,
"loss": 26.0449,
"mean_token_accuracy": 0.6903208963572979,
"num_tokens": 461658.0,
"step": 70
},
{
"entropy": 1.024911842495203,
"epoch": 0.03560896900906916,
"grad_norm": 250.0,
"learning_rate": 1.5800000000000001e-06,
"loss": 23.6986,
"mean_token_accuracy": 0.6996766375377774,
"num_tokens": 528285.0,
"step": 80
},
{
"entropy": 1.0939506959170102,
"epoch": 0.04006009013520281,
"grad_norm": 227.0,
"learning_rate": 1.7800000000000001e-06,
"loss": 23.2949,
"mean_token_accuracy": 0.6969541085883975,
"num_tokens": 595336.0,
"step": 90
},
{
"entropy": 1.0676697574555873,
"epoch": 0.04451121126133645,
"grad_norm": 278.0,
"learning_rate": 1.98e-06,
"loss": 21.5533,
"mean_token_accuracy": 0.7063678905367852,
"num_tokens": 663126.0,
"step": 100
},
{
"epoch": 0.04451121126133645,
"eval_biology_entropy": 1.5063953498601914,
"eval_biology_loss": 1.9373619556427002,
"eval_biology_mean_token_accuracy": 0.6030384682416916,
"eval_biology_num_tokens": 663126.0,
"eval_biology_runtime": 37.2317,
"eval_biology_samples_per_second": 13.429,
"eval_biology_steps_per_second": 13.429,
"step": 100
},
{
"epoch": 0.04451121126133645,
"eval_chemistry_entropy": 1.0538504299521447,
"eval_chemistry_loss": 1.2949438095092773,
"eval_chemistry_mean_token_accuracy": 0.7114702309370041,
"eval_chemistry_num_tokens": 663126.0,
"eval_chemistry_runtime": 41.1049,
"eval_chemistry_samples_per_second": 12.164,
"eval_chemistry_steps_per_second": 12.164,
"step": 100
},
{
"epoch": 0.04451121126133645,
"eval_math_entropy": 0.8524410739541054,
"eval_math_loss": 1.2029013633728027,
"eval_math_mean_token_accuracy": 0.7322385207414627,
"eval_math_num_tokens": 663126.0,
"eval_math_runtime": 41.5152,
"eval_math_samples_per_second": 12.044,
"eval_math_steps_per_second": 12.044,
"step": 100
},
{
"epoch": 0.04451121126133645,
"eval_physics_entropy": 0.9442882173061371,
"eval_physics_loss": 1.1572656631469727,
"eval_physics_mean_token_accuracy": 0.7343679416179657,
"eval_physics_num_tokens": 663126.0,
"eval_physics_runtime": 45.8024,
"eval_physics_samples_per_second": 10.916,
"eval_physics_steps_per_second": 10.916,
"step": 100
},
{
"entropy": 1.0567705381661654,
"epoch": 0.048962332387470095,
"grad_norm": 202.0,
"learning_rate": 2.1800000000000003e-06,
"loss": 20.0858,
"mean_token_accuracy": 0.7180810730904341,
"num_tokens": 731060.0,
"step": 110
},
{
"entropy": 1.0101547280326486,
"epoch": 0.053413453513603736,
"grad_norm": 202.0,
"learning_rate": 2.38e-06,
"loss": 18.8763,
"mean_token_accuracy": 0.7310481067746878,
"num_tokens": 801635.0,
"step": 120
},
{
"entropy": 1.0049639470875262,
"epoch": 0.05786457463973738,
"grad_norm": 154.0,
"learning_rate": 2.5800000000000003e-06,
"loss": 19.1729,
"mean_token_accuracy": 0.7293503576889634,
"num_tokens": 867260.0,
"step": 130
},
{
"entropy": 1.0176106113940477,
"epoch": 0.06231569576587103,
"grad_norm": 144.0,
"learning_rate": 2.7800000000000005e-06,
"loss": 18.97,
"mean_token_accuracy": 0.7298020845279097,
"num_tokens": 931344.0,
"step": 140
},
{
"entropy": 1.0698495015501976,
"epoch": 0.06676681689200467,
"grad_norm": 169.0,
"learning_rate": 2.9800000000000003e-06,
"loss": 17.4943,
"mean_token_accuracy": 0.7370716020464897,
"num_tokens": 993586.0,
"step": 150
},
{
"entropy": 1.0634983357042074,
"epoch": 0.07121793801813832,
"grad_norm": 123.0,
"learning_rate": 3.1800000000000005e-06,
"loss": 16.9373,
"mean_token_accuracy": 0.7367568526417017,
"num_tokens": 1060812.0,
"step": 160
},
{
"entropy": 1.0108168195933103,
"epoch": 0.07566905914427197,
"grad_norm": 125.0,
"learning_rate": 3.3800000000000007e-06,
"loss": 15.569,
"mean_token_accuracy": 0.7519758876413107,
"num_tokens": 1131832.0,
"step": 170
},
{
"entropy": 0.9829567475244403,
"epoch": 0.08012018027040561,
"grad_norm": 99.0,
"learning_rate": 3.58e-06,
"loss": 15.2082,
"mean_token_accuracy": 0.7475447814911604,
"num_tokens": 1200650.0,
"step": 180
},
{
"entropy": 0.9538513226434588,
"epoch": 0.08457130139653925,
"grad_norm": 119.0,
"learning_rate": 3.7800000000000002e-06,
"loss": 15.0014,
"mean_token_accuracy": 0.7515964422374963,
"num_tokens": 1265107.0,
"step": 190
},
{
"entropy": 0.9125228714197874,
"epoch": 0.0890224225226729,
"grad_norm": 117.0,
"learning_rate": 3.980000000000001e-06,
"loss": 14.4833,
"mean_token_accuracy": 0.7538308618590236,
"num_tokens": 1327380.0,
"step": 200
},
{
"epoch": 0.0890224225226729,
"eval_biology_entropy": 1.280277446448803,
"eval_biology_loss": 1.453066349029541,
"eval_biology_mean_token_accuracy": 0.6518876985907555,
"eval_biology_num_tokens": 1327380.0,
"eval_biology_runtime": 36.821,
"eval_biology_samples_per_second": 13.579,
"eval_biology_steps_per_second": 13.579,
"step": 200
},
{
"epoch": 0.0890224225226729,
"eval_chemistry_entropy": 0.8954760466217995,
"eval_chemistry_loss": 0.9106767177581787,
"eval_chemistry_mean_token_accuracy": 0.7551076735854149,
"eval_chemistry_num_tokens": 1327380.0,
"eval_chemistry_runtime": 40.4689,
"eval_chemistry_samples_per_second": 12.355,
"eval_chemistry_steps_per_second": 12.355,
"step": 200
},
{
"epoch": 0.0890224225226729,
"eval_math_entropy": 0.7658050524294376,
"eval_math_loss": 0.9040815234184265,
"eval_math_mean_token_accuracy": 0.7642563463449478,
"eval_math_num_tokens": 1327380.0,
"eval_math_runtime": 41.4556,
"eval_math_samples_per_second": 12.061,
"eval_math_steps_per_second": 12.061,
"step": 200
},
{
"epoch": 0.0890224225226729,
"eval_physics_entropy": 0.8174435417056084,
"eval_physics_loss": 0.8484614491462708,
"eval_physics_mean_token_accuracy": 0.7708570960760116,
"eval_physics_num_tokens": 1327380.0,
"eval_physics_runtime": 45.875,
"eval_physics_samples_per_second": 10.899,
"eval_physics_steps_per_second": 10.899,
"step": 200
},
{
"entropy": 0.9408239619806409,
"epoch": 0.09347354364880654,
"grad_norm": 126.5,
"learning_rate": 4.18e-06,
"loss": 15.2387,
"mean_token_accuracy": 0.7457781180739402,
"num_tokens": 1393379.0,
"step": 210
},
{
"entropy": 0.8950148992240429,
"epoch": 0.09792466477494019,
"grad_norm": 95.0,
"learning_rate": 4.38e-06,
"loss": 14.2337,
"mean_token_accuracy": 0.7587446294724941,
"num_tokens": 1460130.0,
"step": 220
},
{
"entropy": 0.859116705879569,
"epoch": 0.10237578590107384,
"grad_norm": 94.0,
"learning_rate": 4.58e-06,
"loss": 13.7082,
"mean_token_accuracy": 0.76347241923213,
"num_tokens": 1529183.0,
"step": 230
},
{
"entropy": 0.8980951353907585,
"epoch": 0.10682690702720747,
"grad_norm": 118.5,
"learning_rate": 4.78e-06,
"loss": 14.5897,
"mean_token_accuracy": 0.7553087200969457,
"num_tokens": 1597405.0,
"step": 240
},
{
"entropy": 0.9019628280773759,
"epoch": 0.11127802815334112,
"grad_norm": 98.5,
"learning_rate": 4.980000000000001e-06,
"loss": 14.5049,
"mean_token_accuracy": 0.7539074700325727,
"num_tokens": 1661767.0,
"step": 250
},
{
"entropy": 0.9095736019313335,
"epoch": 0.11572914927947477,
"grad_norm": 80.0,
"learning_rate": 5.18e-06,
"loss": 14.553,
"mean_token_accuracy": 0.7567780796438456,
"num_tokens": 1728207.0,
"step": 260
},
{
"entropy": 0.855735239200294,
"epoch": 0.12018027040560841,
"grad_norm": 99.0,
"learning_rate": 5.380000000000001e-06,
"loss": 13.731,
"mean_token_accuracy": 0.7652106903493404,
"num_tokens": 1796055.0,
"step": 270
},
{
"entropy": 0.830021676234901,
"epoch": 0.12463139153174206,
"grad_norm": 97.5,
"learning_rate": 5.580000000000001e-06,
"loss": 13.2554,
"mean_token_accuracy": 0.7721017152070999,
"num_tokens": 1859684.0,
"step": 280
},
{
"entropy": 0.8921441739425063,
"epoch": 0.1290825126578757,
"grad_norm": 106.5,
"learning_rate": 5.78e-06,
"loss": 14.4012,
"mean_token_accuracy": 0.7561684042215348,
"num_tokens": 1921704.0,
"step": 290
},
{
"entropy": 0.8731818461790681,
"epoch": 0.13353363378400934,
"grad_norm": 87.5,
"learning_rate": 5.98e-06,
"loss": 13.9502,
"mean_token_accuracy": 0.7628704030066729,
"num_tokens": 1985766.0,
"step": 300
},
{
"epoch": 0.13353363378400934,
"eval_biology_entropy": 1.2637426958680154,
"eval_biology_loss": 1.3796980381011963,
"eval_biology_mean_token_accuracy": 0.6648430502414704,
"eval_biology_num_tokens": 1985766.0,
"eval_biology_runtime": 37.0397,
"eval_biology_samples_per_second": 13.499,
"eval_biology_steps_per_second": 13.499,
"step": 300
},
{
"epoch": 0.13353363378400934,
"eval_chemistry_entropy": 0.8489444836974144,
"eval_chemistry_loss": 0.8385273814201355,
"eval_chemistry_mean_token_accuracy": 0.7699487572312355,
"eval_chemistry_num_tokens": 1985766.0,
"eval_chemistry_runtime": 40.8665,
"eval_chemistry_samples_per_second": 12.235,
"eval_chemistry_steps_per_second": 12.235,
"step": 300
},
{
"epoch": 0.13353363378400934,
"eval_math_entropy": 0.7357360773980618,
"eval_math_loss": 0.8767910599708557,
"eval_math_mean_token_accuracy": 0.7708948514461518,
"eval_math_num_tokens": 1985766.0,
"eval_math_runtime": 41.8192,
"eval_math_samples_per_second": 11.956,
"eval_math_steps_per_second": 11.956,
"step": 300
},
{
"epoch": 0.13353363378400934,
"eval_physics_entropy": 0.7905060887336731,
"eval_physics_loss": 0.7890318632125854,
"eval_physics_mean_token_accuracy": 0.7825853993296623,
"eval_physics_num_tokens": 1985766.0,
"eval_physics_runtime": 46.1418,
"eval_physics_samples_per_second": 10.836,
"eval_physics_steps_per_second": 10.836,
"step": 300
},
{
"entropy": 0.8452363438904286,
"epoch": 0.137984754910143,
"grad_norm": 96.0,
"learning_rate": 6.18e-06,
"loss": 13.634,
"mean_token_accuracy": 0.7645608998835087,
"num_tokens": 2049041.0,
"step": 310
},
{
"entropy": 0.8076417898759246,
"epoch": 0.14243587603627664,
"grad_norm": 80.5,
"learning_rate": 6.380000000000001e-06,
"loss": 12.8943,
"mean_token_accuracy": 0.7760006580501795,
"num_tokens": 2118068.0,
"step": 320
},
{
"entropy": 0.8085115062072873,
"epoch": 0.14688699716241027,
"grad_norm": 106.5,
"learning_rate": 6.5800000000000005e-06,
"loss": 12.7928,
"mean_token_accuracy": 0.7790838234126568,
"num_tokens": 2186387.0,
"step": 330
},
{
"entropy": 0.824213713593781,
"epoch": 0.15133811828854393,
"grad_norm": 82.5,
"learning_rate": 6.780000000000001e-06,
"loss": 13.389,
"mean_token_accuracy": 0.7689453203231096,
"num_tokens": 2252650.0,
"step": 340
},
{
"entropy": 0.797634300775826,
"epoch": 0.15578923941467757,
"grad_norm": 85.5,
"learning_rate": 6.98e-06,
"loss": 12.8088,
"mean_token_accuracy": 0.7797435287386179,
"num_tokens": 2320358.0,
"step": 350
},
{
"entropy": 0.7920836627483367,
"epoch": 0.16024036054081123,
"grad_norm": 88.0,
"learning_rate": 7.180000000000001e-06,
"loss": 12.6336,
"mean_token_accuracy": 0.7806293956935406,
"num_tokens": 2388824.0,
"step": 360
},
{
"entropy": 0.7749077206477523,
"epoch": 0.16469148166694486,
"grad_norm": 80.5,
"learning_rate": 7.3800000000000005e-06,
"loss": 12.3995,
"mean_token_accuracy": 0.7864189695566892,
"num_tokens": 2456144.0,
"step": 370
},
{
"entropy": 0.8578483302146196,
"epoch": 0.1691426027930785,
"grad_norm": 81.0,
"learning_rate": 7.58e-06,
"loss": 13.8496,
"mean_token_accuracy": 0.7662820883095265,
"num_tokens": 2515325.0,
"step": 380
},
{
"entropy": 0.7971056224778295,
"epoch": 0.17359372391921216,
"grad_norm": 68.0,
"learning_rate": 7.78e-06,
"loss": 12.6922,
"mean_token_accuracy": 0.7778837148100137,
"num_tokens": 2580490.0,
"step": 390
},
{
"entropy": 0.7984419580549001,
"epoch": 0.1780448450453458,
"grad_norm": 70.5,
"learning_rate": 7.980000000000002e-06,
"loss": 12.8255,
"mean_token_accuracy": 0.7773921977728605,
"num_tokens": 2644330.0,
"step": 400
},
{
"epoch": 0.1780448450453458,
"eval_biology_entropy": 1.2305693982243537,
"eval_biology_loss": 1.3490314483642578,
"eval_biology_mean_token_accuracy": 0.6716532117724419,
"eval_biology_num_tokens": 2644330.0,
"eval_biology_runtime": 36.6001,
"eval_biology_samples_per_second": 13.661,
"eval_biology_steps_per_second": 13.661,
"step": 400
},
{
"epoch": 0.1780448450453458,
"eval_chemistry_entropy": 0.7882854393720626,
"eval_chemistry_loss": 0.7991188764572144,
"eval_chemistry_mean_token_accuracy": 0.778610565006733,
"eval_chemistry_num_tokens": 2644330.0,
"eval_chemistry_runtime": 40.3843,
"eval_chemistry_samples_per_second": 12.381,
"eval_chemistry_steps_per_second": 12.381,
"step": 400
},
{
"epoch": 0.1780448450453458,
"eval_math_entropy": 0.6974474709928036,
"eval_math_loss": 0.8647986054420471,
"eval_math_mean_token_accuracy": 0.7742947788238526,
"eval_math_num_tokens": 2644330.0,
"eval_math_runtime": 41.8502,
"eval_math_samples_per_second": 11.947,
"eval_math_steps_per_second": 11.947,
"step": 400
},
{
"epoch": 0.1780448450453458,
"eval_physics_entropy": 0.740912606716156,
"eval_physics_loss": 0.7630434632301331,
"eval_physics_mean_token_accuracy": 0.7882855938673019,
"eval_physics_num_tokens": 2644330.0,
"eval_physics_runtime": 46.1492,
"eval_physics_samples_per_second": 10.834,
"eval_physics_steps_per_second": 10.834,
"step": 400
},
{
"entropy": 0.7443430663086474,
"epoch": 0.18249596617147945,
"grad_norm": 72.5,
"learning_rate": 8.18e-06,
"loss": 11.7994,
"mean_token_accuracy": 0.7938185524195432,
"num_tokens": 2713234.0,
"step": 410
},
{
"entropy": 0.7856309018097818,
"epoch": 0.18694708729761308,
"grad_norm": 78.0,
"learning_rate": 8.380000000000001e-06,
"loss": 12.9567,
"mean_token_accuracy": 0.7781453486531973,
"num_tokens": 2783261.0,
"step": 420
},
{
"entropy": 0.799421458132565,
"epoch": 0.19139820842374672,
"grad_norm": 61.25,
"learning_rate": 8.580000000000001e-06,
"loss": 12.6392,
"mean_token_accuracy": 0.7806750666350126,
"num_tokens": 2850170.0,
"step": 430
},
{
"entropy": 0.7786322007887065,
"epoch": 0.19584932954988038,
"grad_norm": 80.5,
"learning_rate": 8.78e-06,
"loss": 12.4988,
"mean_token_accuracy": 0.7827117882668972,
"num_tokens": 2913700.0,
"step": 440
},
{
"entropy": 0.763565490860492,
"epoch": 0.200300450676014,
"grad_norm": 81.0,
"learning_rate": 8.98e-06,
"loss": 12.4779,
"mean_token_accuracy": 0.7831043265759945,
"num_tokens": 2981212.0,
"step": 450
},
{
"entropy": 0.7708647299557925,
"epoch": 0.20475157180214767,
"grad_norm": 85.0,
"learning_rate": 9.180000000000002e-06,
"loss": 12.1503,
"mean_token_accuracy": 0.7912418570369482,
"num_tokens": 3048973.0,
"step": 460
},
{
"entropy": 0.769937212113291,
"epoch": 0.2092026929282813,
"grad_norm": 62.0,
"learning_rate": 9.38e-06,
"loss": 12.6137,
"mean_token_accuracy": 0.7829986553639173,
"num_tokens": 3117033.0,
"step": 470
},
{
"entropy": 0.7863288868218661,
"epoch": 0.21365381405441494,
"grad_norm": 74.5,
"learning_rate": 9.58e-06,
"loss": 12.3857,
"mean_token_accuracy": 0.7842021461576223,
"num_tokens": 3185255.0,
"step": 480
},
{
"entropy": 0.7699232887476682,
"epoch": 0.2181049351805486,
"grad_norm": 74.5,
"learning_rate": 9.780000000000001e-06,
"loss": 12.4166,
"mean_token_accuracy": 0.7846909243613481,
"num_tokens": 3253121.0,
"step": 490
},
{
"entropy": 0.7251296367496252,
"epoch": 0.22255605630668224,
"grad_norm": 75.0,
"learning_rate": 9.980000000000001e-06,
"loss": 11.4498,
"mean_token_accuracy": 0.7972834445536137,
"num_tokens": 3322823.0,
"step": 500
},
{
"epoch": 0.22255605630668224,
"eval_biology_entropy": 1.1898637573719024,
"eval_biology_loss": 1.338150978088379,
"eval_biology_mean_token_accuracy": 0.6746688461303711,
"eval_biology_num_tokens": 3322823.0,
"eval_biology_runtime": 37.4484,
"eval_biology_samples_per_second": 13.352,
"eval_biology_steps_per_second": 13.352,
"step": 500
},
{
"epoch": 0.22255605630668224,
"eval_chemistry_entropy": 0.7506220785379409,
"eval_chemistry_loss": 0.7758934497833252,
"eval_chemistry_mean_token_accuracy": 0.7828538768291473,
"eval_chemistry_num_tokens": 3322823.0,
"eval_chemistry_runtime": 40.7375,
"eval_chemistry_samples_per_second": 12.274,
"eval_chemistry_steps_per_second": 12.274,
"step": 500
},
{
"epoch": 0.22255605630668224,
"eval_math_entropy": 0.6889803909361363,
"eval_math_loss": 0.8622512817382812,
"eval_math_mean_token_accuracy": 0.7754237843751908,
"eval_math_num_tokens": 3322823.0,
"eval_math_runtime": 41.9238,
"eval_math_samples_per_second": 11.926,
"eval_math_steps_per_second": 11.926,
"step": 500
},
{
"epoch": 0.22255605630668224,
"eval_physics_entropy": 0.7242196025848389,
"eval_physics_loss": 0.7506964802742004,
"eval_physics_mean_token_accuracy": 0.7921671935915947,
"eval_physics_num_tokens": 3322823.0,
"eval_physics_runtime": 45.9692,
"eval_physics_samples_per_second": 10.877,
"eval_physics_steps_per_second": 10.877,
"step": 500
},
{
"entropy": 0.7351194931194186,
"epoch": 0.2270071774328159,
"grad_norm": 75.0,
"learning_rate": 1.018e-05,
"loss": 11.9225,
"mean_token_accuracy": 0.7910898551344872,
"num_tokens": 3393393.0,
"step": 510
},
{
"entropy": 0.8338562373071909,
"epoch": 0.23145829855894953,
"grad_norm": 79.0,
"learning_rate": 1.038e-05,
"loss": 13.1291,
"mean_token_accuracy": 0.774738060683012,
"num_tokens": 3454750.0,
"step": 520
},
{
"entropy": 0.7108834735117853,
"epoch": 0.2359094196850832,
"grad_norm": 70.5,
"learning_rate": 1.0580000000000002e-05,
"loss": 11.7207,
"mean_token_accuracy": 0.7975892208516597,
"num_tokens": 3526914.0,
"step": 530
},
{
"entropy": 0.8373067150823772,
"epoch": 0.24036054081121683,
"grad_norm": 72.0,
"learning_rate": 1.0780000000000002e-05,
"loss": 13.0512,
"mean_token_accuracy": 0.7754946660250426,
"num_tokens": 3589449.0,
"step": 540
},
{
"entropy": 0.7569006633013486,
"epoch": 0.24481166193735046,
"grad_norm": 79.0,
"learning_rate": 1.0980000000000002e-05,
"loss": 12.267,
"mean_token_accuracy": 0.786787087097764,
"num_tokens": 3655092.0,
"step": 550
},
{
"entropy": 0.7805684333667159,
"epoch": 0.24926278306348412,
"grad_norm": 93.5,
"learning_rate": 1.1180000000000001e-05,
"loss": 12.5142,
"mean_token_accuracy": 0.7839910417795182,
"num_tokens": 3719113.0,
"step": 560
},
{
"entropy": 0.7685065409168601,
"epoch": 0.25371390418961776,
"grad_norm": 65.5,
"learning_rate": 1.138e-05,
"loss": 12.4558,
"mean_token_accuracy": 0.7818987771868706,
"num_tokens": 3785282.0,
"step": 570
},
{
"entropy": 0.774918237887323,
"epoch": 0.2581650253157514,
"grad_norm": 61.25,
"learning_rate": 1.1580000000000001e-05,
"loss": 12.415,
"mean_token_accuracy": 0.7842379476875067,
"num_tokens": 3852276.0,
"step": 580
},
{
"entropy": 0.7105797030963004,
"epoch": 0.262616146441885,
"grad_norm": 67.5,
"learning_rate": 1.178e-05,
"loss": 11.3721,
"mean_token_accuracy": 0.7995343253016471,
"num_tokens": 3925649.0,
"step": 590
},
{
"entropy": 0.7462131816893816,
"epoch": 0.2670672675680187,
"grad_norm": 74.5,
"learning_rate": 1.198e-05,
"loss": 11.8831,
"mean_token_accuracy": 0.7898255579173565,
"num_tokens": 3990505.0,
"step": 600
},
{
"epoch": 0.2670672675680187,
"eval_biology_entropy": 1.2392896317243576,
"eval_biology_loss": 1.3273310661315918,
"eval_biology_mean_token_accuracy": 0.6752943544387817,
"eval_biology_num_tokens": 3990505.0,
"eval_biology_runtime": 36.4668,
"eval_biology_samples_per_second": 13.711,
"eval_biology_steps_per_second": 13.711,
"step": 600
},
{
"epoch": 0.2670672675680187,
"eval_chemistry_entropy": 0.7596916342377663,
"eval_chemistry_loss": 0.7626407742500305,
"eval_chemistry_mean_token_accuracy": 0.7862690081000328,
"eval_chemistry_num_tokens": 3990505.0,
"eval_chemistry_runtime": 40.266,
"eval_chemistry_samples_per_second": 12.417,
"eval_chemistry_steps_per_second": 12.417,
"step": 600
},
{
"epoch": 0.2670672675680187,
"eval_math_entropy": 0.6872916958034039,
"eval_math_loss": 0.8660696148872375,
"eval_math_mean_token_accuracy": 0.7750280216932297,
"eval_math_num_tokens": 3990505.0,
"eval_math_runtime": 41.4933,
"eval_math_samples_per_second": 12.05,
"eval_math_steps_per_second": 12.05,
"step": 600
},
{
"epoch": 0.2670672675680187,
"eval_physics_entropy": 0.7289359270334244,
"eval_physics_loss": 0.744917631149292,
"eval_physics_mean_token_accuracy": 0.7920673425793647,
"eval_physics_num_tokens": 3990505.0,
"eval_physics_runtime": 46.0067,
"eval_physics_samples_per_second": 10.868,
"eval_physics_steps_per_second": 10.868,
"step": 600
},
{
"entropy": 0.7445939108729362,
"epoch": 0.27151838869415235,
"grad_norm": 65.5,
"learning_rate": 1.218e-05,
"loss": 11.9377,
"mean_token_accuracy": 0.790003751963377,
"num_tokens": 4060657.0,
"step": 610
},
{
"entropy": 0.7278808873146773,
"epoch": 0.275969509820286,
"grad_norm": 65.0,
"learning_rate": 1.2380000000000002e-05,
"loss": 11.5778,
"mean_token_accuracy": 0.7942668478935957,
"num_tokens": 4133119.0,
"step": 620
},
{
"entropy": 0.7475641055032611,
"epoch": 0.2804206309464196,
"grad_norm": 70.5,
"learning_rate": 1.2580000000000002e-05,
"loss": 11.9803,
"mean_token_accuracy": 0.7913637235760689,
"num_tokens": 4200051.0,
"step": 630
},
{
"entropy": 0.7986388958990573,
"epoch": 0.2848717520725533,
"grad_norm": 71.0,
"learning_rate": 1.2780000000000001e-05,
"loss": 12.767,
"mean_token_accuracy": 0.782176585495472,
"num_tokens": 4267403.0,
"step": 640
},
{
"entropy": 0.7370933200232684,
"epoch": 0.28932287319868694,
"grad_norm": 72.0,
"learning_rate": 1.2980000000000001e-05,
"loss": 11.8884,
"mean_token_accuracy": 0.7921638786792755,
"num_tokens": 4333268.0,
"step": 650
},
{
"entropy": 0.737369327340275,
"epoch": 0.29377399432482054,
"grad_norm": 79.0,
"learning_rate": 1.3180000000000001e-05,
"loss": 11.7388,
"mean_token_accuracy": 0.7942005816847086,
"num_tokens": 4400242.0,
"step": 660
},
{
"entropy": 0.7550807458348572,
"epoch": 0.2982251154509542,
"grad_norm": 74.5,
"learning_rate": 1.3380000000000002e-05,
"loss": 12.0296,
"mean_token_accuracy": 0.7894790366291999,
"num_tokens": 4464714.0,
"step": 670
},
{
"entropy": 0.7167205691337586,
"epoch": 0.30267623657708786,
"grad_norm": 63.75,
"learning_rate": 1.3580000000000002e-05,
"loss": 11.5706,
"mean_token_accuracy": 0.796867698058486,
"num_tokens": 4535458.0,
"step": 680
},
{
"entropy": 0.7481121812015772,
"epoch": 0.3071273577032215,
"grad_norm": 67.0,
"learning_rate": 1.378e-05,
"loss": 11.985,
"mean_token_accuracy": 0.7914165269583464,
"num_tokens": 4599099.0,
"step": 690
},
{
"entropy": 0.7162985533475876,
"epoch": 0.31157847882935513,
"grad_norm": 56.25,
"learning_rate": 1.398e-05,
"loss": 11.4683,
"mean_token_accuracy": 0.7971325770020485,
"num_tokens": 4662903.0,
"step": 700
},
{
"epoch": 0.31157847882935513,
"eval_biology_entropy": 1.2363531575202942,
"eval_biology_loss": 1.32443106174469,
"eval_biology_mean_token_accuracy": 0.6762348954081535,
"eval_biology_num_tokens": 4662903.0,
"eval_biology_runtime": 36.3542,
"eval_biology_samples_per_second": 13.754,
"eval_biology_steps_per_second": 13.754,
"step": 700
},
{
"epoch": 0.31157847882935513,
"eval_chemistry_entropy": 0.7484616157412529,
"eval_chemistry_loss": 0.7520119547843933,
"eval_chemistry_mean_token_accuracy": 0.7884139549732209,
"eval_chemistry_num_tokens": 4662903.0,
"eval_chemistry_runtime": 41.3205,
"eval_chemistry_samples_per_second": 12.101,
"eval_chemistry_steps_per_second": 12.101,
"step": 700
},
{
"epoch": 0.31157847882935513,
"eval_math_entropy": 0.6818331519067288,
"eval_math_loss": 0.862537145614624,
"eval_math_mean_token_accuracy": 0.7769467064142227,
"eval_math_num_tokens": 4662903.0,
"eval_math_runtime": 42.2089,
"eval_math_samples_per_second": 11.846,
"eval_math_steps_per_second": 11.846,
"step": 700
},
{
"epoch": 0.31157847882935513,
"eval_physics_entropy": 0.7252178380489349,
"eval_physics_loss": 0.7445859313011169,
"eval_physics_mean_token_accuracy": 0.7919469431638718,
"eval_physics_num_tokens": 4662903.0,
"eval_physics_runtime": 45.6964,
"eval_physics_samples_per_second": 10.942,
"eval_physics_steps_per_second": 10.942,
"step": 700
},
{
"entropy": 0.7590530560351908,
"epoch": 0.3160295999554888,
"grad_norm": 83.5,
"learning_rate": 1.418e-05,
"loss": 12.3935,
"mean_token_accuracy": 0.7850905049592256,
"num_tokens": 4729166.0,
"step": 710
},
{
"entropy": 0.800558901950717,
"epoch": 0.32048072108162245,
"grad_norm": 79.0,
"learning_rate": 1.4380000000000001e-05,
"loss": 12.7943,
"mean_token_accuracy": 0.7798430301249027,
"num_tokens": 4796815.0,
"step": 720
},
{
"entropy": 0.6950356176123023,
"epoch": 0.32493184220775606,
"grad_norm": 69.0,
"learning_rate": 1.4580000000000001e-05,
"loss": 11.0797,
"mean_token_accuracy": 0.803941647708416,
"num_tokens": 4865172.0,
"step": 730
},
{
"entropy": 0.7522321011871099,
"epoch": 0.3293829633338897,
"grad_norm": 64.0,
"learning_rate": 1.478e-05,
"loss": 12.1663,
"mean_token_accuracy": 0.7838000770658254,
"num_tokens": 4932671.0,
"step": 740
},
{
"entropy": 0.7758943744935095,
"epoch": 0.3338340844600234,
"grad_norm": 65.0,
"learning_rate": 1.498e-05,
"loss": 12.4442,
"mean_token_accuracy": 0.7852539490908385,
"num_tokens": 4998710.0,
"step": 750
},
{
"entropy": 0.7163919297046959,
"epoch": 0.338285205586157,
"grad_norm": 67.5,
"learning_rate": 1.5180000000000002e-05,
"loss": 11.5154,
"mean_token_accuracy": 0.7952407207340002,
"num_tokens": 5066948.0,
"step": 760
},
{
"entropy": 0.753566917218268,
"epoch": 0.34273632671229065,
"grad_norm": 67.0,
"learning_rate": 1.5380000000000002e-05,
"loss": 12.0632,
"mean_token_accuracy": 0.7887947488576174,
"num_tokens": 5129950.0,
"step": 770
},
{
"entropy": 0.7747300613671542,
"epoch": 0.3471874478384243,
"grad_norm": 64.5,
"learning_rate": 1.5580000000000003e-05,
"loss": 12.4883,
"mean_token_accuracy": 0.7800568122416734,
"num_tokens": 5193519.0,
"step": 780
},
{
"entropy": 0.7476665185764432,
"epoch": 0.351638568964558,
"grad_norm": 91.0,
"learning_rate": 1.578e-05,
"loss": 11.9465,
"mean_token_accuracy": 0.7881375091150403,
"num_tokens": 5256503.0,
"step": 790
},
{
"entropy": 0.8204591780900955,
"epoch": 0.3560896900906916,
"grad_norm": 68.0,
"learning_rate": 1.5980000000000003e-05,
"loss": 13.1278,
"mean_token_accuracy": 0.7780214017257094,
"num_tokens": 5324751.0,
"step": 800
},
{
"epoch": 0.3560896900906916,
"eval_biology_entropy": 1.3236136644482612,
"eval_biology_loss": 1.3220669031143188,
"eval_biology_mean_token_accuracy": 0.6748943754434585,
"eval_biology_num_tokens": 5324751.0,
"eval_biology_runtime": 36.5537,
"eval_biology_samples_per_second": 13.679,
"eval_biology_steps_per_second": 13.679,
"step": 800
},
{
"epoch": 0.3560896900906916,
"eval_chemistry_entropy": 0.7881391143500804,
"eval_chemistry_loss": 0.7549666166305542,
"eval_chemistry_mean_token_accuracy": 0.7878072342276573,
"eval_chemistry_num_tokens": 5324751.0,
"eval_chemistry_runtime": 40.4688,
"eval_chemistry_samples_per_second": 12.355,
"eval_chemistry_steps_per_second": 12.355,
"step": 800
},
{
"epoch": 0.3560896900906916,
"eval_math_entropy": 0.7137112567722798,
"eval_math_loss": 0.8679834008216858,
"eval_math_mean_token_accuracy": 0.774067041516304,
"eval_math_num_tokens": 5324751.0,
"eval_math_runtime": 41.3922,
"eval_math_samples_per_second": 12.08,
"eval_math_steps_per_second": 12.08,
"step": 800
},
{
"epoch": 0.3560896900906916,
"eval_physics_entropy": 0.764995231449604,
"eval_physics_loss": 0.750312864780426,
"eval_physics_mean_token_accuracy": 0.7906691903471946,
"eval_physics_num_tokens": 5324751.0,
"eval_physics_runtime": 45.8484,
"eval_physics_samples_per_second": 10.906,
"eval_physics_steps_per_second": 10.906,
"step": 800
},
{
"entropy": 0.6771990330889821,
"epoch": 0.36054081121682524,
"grad_norm": 65.0,
"learning_rate": 1.618e-05,
"loss": 11.1158,
"mean_token_accuracy": 0.8028291787952184,
"num_tokens": 5391664.0,
"step": 810
},
{
"entropy": 0.7129247286356986,
"epoch": 0.3649919323429589,
"grad_norm": 70.5,
"learning_rate": 1.638e-05,
"loss": 11.1981,
"mean_token_accuracy": 0.8007186874747276,
"num_tokens": 5458519.0,
"step": 820
},
{
"entropy": 0.7549328833818436,
"epoch": 0.3694430534690925,
"grad_norm": 65.0,
"learning_rate": 1.658e-05,
"loss": 12.1723,
"mean_token_accuracy": 0.7901175323873758,
"num_tokens": 5523391.0,
"step": 830
},
{
"entropy": 0.7134035120718181,
"epoch": 0.37389417459522617,
"grad_norm": 59.0,
"learning_rate": 1.6780000000000002e-05,
"loss": 11.7326,
"mean_token_accuracy": 0.795387938991189,
"num_tokens": 5585508.0,
"step": 840
},
{
"entropy": 0.7443888144567609,
"epoch": 0.37834529572135983,
"grad_norm": 71.0,
"learning_rate": 1.698e-05,
"loss": 11.8848,
"mean_token_accuracy": 0.7947730336338281,
"num_tokens": 5650755.0,
"step": 850
},
{
"entropy": 0.7494110313244164,
"epoch": 0.38279641684749344,
"grad_norm": 61.5,
"learning_rate": 1.718e-05,
"loss": 11.969,
"mean_token_accuracy": 0.788362193852663,
"num_tokens": 5716321.0,
"step": 860
},
{
"entropy": 0.7754998824559152,
"epoch": 0.3872475379736271,
"grad_norm": 67.0,
"learning_rate": 1.7380000000000003e-05,
"loss": 12.4655,
"mean_token_accuracy": 0.7815641265362501,
"num_tokens": 5782698.0,
"step": 870
},
{
"entropy": 0.7258908217772841,
"epoch": 0.39169865909976076,
"grad_norm": 54.5,
"learning_rate": 1.758e-05,
"loss": 11.662,
"mean_token_accuracy": 0.7951392225921154,
"num_tokens": 5848889.0,
"step": 880
},
{
"entropy": 0.7699500862509012,
"epoch": 0.3961497802258944,
"grad_norm": 65.5,
"learning_rate": 1.7780000000000003e-05,
"loss": 12.5712,
"mean_token_accuracy": 0.7857454361394047,
"num_tokens": 5912919.0,
"step": 890
},
{
"entropy": 0.7454446164891124,
"epoch": 0.400600901352028,
"grad_norm": 60.75,
"learning_rate": 1.798e-05,
"loss": 11.8793,
"mean_token_accuracy": 0.793294046446681,
"num_tokens": 5980229.0,
"step": 900
},
{
"epoch": 0.400600901352028,
"eval_biology_entropy": 1.2526325694918632,
"eval_biology_loss": 1.3328229188919067,
"eval_biology_mean_token_accuracy": 0.6739730868339538,
"eval_biology_num_tokens": 5980229.0,
"eval_biology_runtime": 37.1628,
"eval_biology_samples_per_second": 13.454,
"eval_biology_steps_per_second": 13.454,
"step": 900
},
{
"epoch": 0.400600901352028,
"eval_chemistry_entropy": 0.7478999142050743,
"eval_chemistry_loss": 0.7579869627952576,
"eval_chemistry_mean_token_accuracy": 0.7872751998305321,
"eval_chemistry_num_tokens": 5980229.0,
"eval_chemistry_runtime": 41.0567,
"eval_chemistry_samples_per_second": 12.178,
"eval_chemistry_steps_per_second": 12.178,
"step": 900
},
{
"epoch": 0.400600901352028,
"eval_math_entropy": 0.691623161137104,
"eval_math_loss": 0.8888975977897644,
"eval_math_mean_token_accuracy": 0.7720628465414048,
"eval_math_num_tokens": 5980229.0,
"eval_math_runtime": 41.3967,
"eval_math_samples_per_second": 12.078,
"eval_math_steps_per_second": 12.078,
"step": 900
},
{
"epoch": 0.400600901352028,
"eval_physics_entropy": 0.7361411595344544,
"eval_physics_loss": 0.7520632743835449,
"eval_physics_mean_token_accuracy": 0.7907493528723717,
"eval_physics_num_tokens": 5980229.0,
"eval_physics_runtime": 46.0024,
"eval_physics_samples_per_second": 10.869,
"eval_physics_steps_per_second": 10.869,
"step": 900
},
{
"entropy": 0.7022204802371561,
"epoch": 0.4050520224781617,
"grad_norm": 61.0,
"learning_rate": 1.8180000000000002e-05,
"loss": 11.4405,
"mean_token_accuracy": 0.798212106898427,
"num_tokens": 6048508.0,
"step": 910
},
{
"entropy": 0.7361409039236605,
"epoch": 0.40950314360429535,
"grad_norm": 60.25,
"learning_rate": 1.8380000000000004e-05,
"loss": 11.7212,
"mean_token_accuracy": 0.7943942189216614,
"num_tokens": 6114855.0,
"step": 920
},
{
"entropy": 0.7555323796346783,
"epoch": 0.41395426473042896,
"grad_norm": 54.0,
"learning_rate": 1.858e-05,
"loss": 12.1222,
"mean_token_accuracy": 0.7887333419173956,
"num_tokens": 6179050.0,
"step": 930
},
{
"entropy": 0.7449197143316268,
"epoch": 0.4184053858565626,
"grad_norm": 59.0,
"learning_rate": 1.878e-05,
"loss": 12.1501,
"mean_token_accuracy": 0.7886579774320126,
"num_tokens": 6244987.0,
"step": 940
},
{
"entropy": 0.7183629649691283,
"epoch": 0.4228565069826963,
"grad_norm": 66.0,
"learning_rate": 1.898e-05,
"loss": 11.6027,
"mean_token_accuracy": 0.794747618958354,
"num_tokens": 6310799.0,
"step": 950
},
{
"entropy": 0.7170003782957792,
"epoch": 0.4273076281088299,
"grad_norm": 63.5,
"learning_rate": 1.918e-05,
"loss": 11.5812,
"mean_token_accuracy": 0.7965250793844462,
"num_tokens": 6378152.0,
"step": 960
},
{
"entropy": 0.7422669138759375,
"epoch": 0.43175874923496355,
"grad_norm": 54.0,
"learning_rate": 1.938e-05,
"loss": 11.7787,
"mean_token_accuracy": 0.7933820992708206,
"num_tokens": 6442491.0,
"step": 970
},
{
"entropy": 0.7739317396655678,
"epoch": 0.4362098703610972,
"grad_norm": 57.75,
"learning_rate": 1.9580000000000002e-05,
"loss": 12.5927,
"mean_token_accuracy": 0.7841954406350851,
"num_tokens": 6510231.0,
"step": 980
},
{
"entropy": 0.745883577503264,
"epoch": 0.44066099148723087,
"grad_norm": 57.25,
"learning_rate": 1.978e-05,
"loss": 12.0418,
"mean_token_accuracy": 0.790126183629036,
"num_tokens": 6572858.0,
"step": 990
},
{
"entropy": 0.7626472780480981,
"epoch": 0.4451121126133645,
"grad_norm": 48.0,
"learning_rate": 1.9980000000000002e-05,
"loss": 12.3648,
"mean_token_accuracy": 0.7834480810910464,
"num_tokens": 6637273.0,
"step": 1000
},
{
"epoch": 0.4451121126133645,
"eval_biology_entropy": 1.2697618773579598,
"eval_biology_loss": 1.3405715227127075,
"eval_biology_mean_token_accuracy": 0.6724427245259285,
"eval_biology_num_tokens": 6637273.0,
"eval_biology_runtime": 34.6717,
"eval_biology_samples_per_second": 14.421,
"eval_biology_steps_per_second": 14.421,
"step": 1000
},
{
"epoch": 0.4451121126133645,
"eval_chemistry_entropy": 0.7615428499877452,
"eval_chemistry_loss": 0.7553035616874695,
"eval_chemistry_mean_token_accuracy": 0.7880414368510247,
"eval_chemistry_num_tokens": 6637273.0,
"eval_chemistry_runtime": 38.9404,
"eval_chemistry_samples_per_second": 12.84,
"eval_chemistry_steps_per_second": 12.84,
"step": 1000
},
{
"epoch": 0.4451121126133645,
"eval_math_entropy": 0.7116467788815498,
"eval_math_loss": 0.8722440004348755,
"eval_math_mean_token_accuracy": 0.7725971038341523,
"eval_math_num_tokens": 6637273.0,
"eval_math_runtime": 39.9329,
"eval_math_samples_per_second": 12.521,
"eval_math_steps_per_second": 12.521,
"step": 1000
},
{
"epoch": 0.4451121126133645,
"eval_physics_entropy": 0.7675083922743797,
"eval_physics_loss": 0.7605612874031067,
"eval_physics_mean_token_accuracy": 0.7887651996016503,
"eval_physics_num_tokens": 6637273.0,
"eval_physics_runtime": 44.6453,
"eval_physics_samples_per_second": 11.199,
"eval_physics_steps_per_second": 11.199,
"step": 1000
}
],
"logging_steps": 10,
"max_steps": 10000,
"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.450581354264347e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}