b5e92jps / checkpoint-500 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
059f48b verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.22255605630668224,
"eval_steps": 100,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"entropy": 0.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
}
],
"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": 2.2280979081801686e+17,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}