fchazuoj / checkpoint-1000 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
017e084 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": 1.1280501924455166,
"epoch": 0.016,
"grad_norm": 161.0,
"learning_rate": 6.000000000000001e-07,
"loss": 36.0968,
"mean_token_accuracy": 0.6165979273617268,
"num_tokens": 280941.0,
"step": 10
},
{
"entropy": 1.1705728020519017,
"epoch": 0.032,
"grad_norm": 146.0,
"learning_rate": 1.2666666666666669e-06,
"loss": 36.5572,
"mean_token_accuracy": 0.6121777599677444,
"num_tokens": 558056.0,
"step": 20
},
{
"entropy": 1.1892314109951259,
"epoch": 0.048,
"grad_norm": 135.0,
"learning_rate": 1.9333333333333336e-06,
"loss": 35.5667,
"mean_token_accuracy": 0.6162483215332031,
"num_tokens": 836753.0,
"step": 30
},
{
"entropy": 1.2923907738178968,
"epoch": 0.064,
"grad_norm": 106.0,
"learning_rate": 2.6e-06,
"loss": 34.4747,
"mean_token_accuracy": 0.6145394820719957,
"num_tokens": 1126446.0,
"step": 40
},
{
"entropy": 1.4465828079730272,
"epoch": 0.08,
"grad_norm": 86.5,
"learning_rate": 3.266666666666667e-06,
"loss": 32.2616,
"mean_token_accuracy": 0.6272102266550064,
"num_tokens": 1413596.0,
"step": 50
},
{
"entropy": 1.5344239331781864,
"epoch": 0.096,
"grad_norm": 54.25,
"learning_rate": 3.9333333333333335e-06,
"loss": 28.8152,
"mean_token_accuracy": 0.6551170501857996,
"num_tokens": 1701193.0,
"step": 60
},
{
"entropy": 1.5569457441568375,
"epoch": 0.112,
"grad_norm": 45.75,
"learning_rate": 4.600000000000001e-06,
"loss": 26.6094,
"mean_token_accuracy": 0.6667167004197836,
"num_tokens": 1979232.0,
"step": 70
},
{
"entropy": 1.4980619736015797,
"epoch": 0.128,
"grad_norm": 34.0,
"learning_rate": 5.2666666666666665e-06,
"loss": 25.0735,
"mean_token_accuracy": 0.6823939502239227,
"num_tokens": 2274177.0,
"step": 80
},
{
"entropy": 1.4427384316921235,
"epoch": 0.144,
"grad_norm": 29.75,
"learning_rate": 5.933333333333335e-06,
"loss": 23.2103,
"mean_token_accuracy": 0.6962548345327377,
"num_tokens": 2548445.0,
"step": 90
},
{
"entropy": 1.3565625827759504,
"epoch": 0.16,
"grad_norm": 30.875,
"learning_rate": 6.600000000000001e-06,
"loss": 21.4845,
"mean_token_accuracy": 0.7108539894223214,
"num_tokens": 2824418.0,
"step": 100
},
{
"epoch": 0.16,
"eval_biology_entropy": 3.535919818878174,
"eval_biology_loss": 5.02457332611084,
"eval_biology_mean_token_accuracy": 0.3542800180912018,
"eval_biology_num_tokens": 2824418.0,
"eval_biology_runtime": 49.6437,
"eval_biology_samples_per_second": 10.072,
"eval_biology_steps_per_second": 2.518,
"step": 100
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 2.5280555896759034,
"eval_chemistry_loss": 2.8887903690338135,
"eval_chemistry_mean_token_accuracy": 0.5300401592254639,
"eval_chemistry_num_tokens": 2824418.0,
"eval_chemistry_runtime": 61.5471,
"eval_chemistry_samples_per_second": 8.124,
"eval_chemistry_steps_per_second": 2.031,
"step": 100
},
{
"epoch": 0.16,
"eval_math_entropy": 1.3736235461235047,
"eval_math_loss": 1.3470827341079712,
"eval_math_mean_token_accuracy": 0.7081305732727051,
"eval_math_num_tokens": 2824418.0,
"eval_math_runtime": 63.1282,
"eval_math_samples_per_second": 7.92,
"eval_math_steps_per_second": 1.98,
"step": 100
},
{
"epoch": 0.16,
"eval_physics_entropy": 2.064271595954895,
"eval_physics_loss": 2.1881864070892334,
"eval_physics_mean_token_accuracy": 0.6055651969909668,
"eval_physics_num_tokens": 2824418.0,
"eval_physics_runtime": 71.8769,
"eval_physics_samples_per_second": 6.956,
"eval_physics_steps_per_second": 1.739,
"step": 100
},
{
"entropy": 1.3322576396167278,
"epoch": 0.176,
"grad_norm": 23.25,
"learning_rate": 7.266666666666668e-06,
"loss": 20.9108,
"mean_token_accuracy": 0.7151085637509823,
"num_tokens": 3110313.0,
"step": 110
},
{
"entropy": 1.3247066088020802,
"epoch": 0.192,
"grad_norm": 23.625,
"learning_rate": 7.933333333333334e-06,
"loss": 20.6356,
"mean_token_accuracy": 0.7142969533801079,
"num_tokens": 3394170.0,
"step": 120
},
{
"entropy": 1.256456857919693,
"epoch": 0.208,
"grad_norm": 21.125,
"learning_rate": 8.6e-06,
"loss": 19.9583,
"mean_token_accuracy": 0.7226813279092312,
"num_tokens": 3673600.0,
"step": 130
},
{
"entropy": 1.2233942866325378,
"epoch": 0.224,
"grad_norm": 23.5,
"learning_rate": 9.266666666666667e-06,
"loss": 19.4829,
"mean_token_accuracy": 0.7258789885789156,
"num_tokens": 3953732.0,
"step": 140
},
{
"entropy": 1.1788023415952922,
"epoch": 0.24,
"grad_norm": 22.625,
"learning_rate": 9.933333333333334e-06,
"loss": 18.6617,
"mean_token_accuracy": 0.735758152604103,
"num_tokens": 4243655.0,
"step": 150
},
{
"entropy": 1.1358939699828625,
"epoch": 0.256,
"grad_norm": 21.25,
"learning_rate": 1.0600000000000002e-05,
"loss": 18.067,
"mean_token_accuracy": 0.7414251279085875,
"num_tokens": 4531471.0,
"step": 160
},
{
"entropy": 1.1413200695067645,
"epoch": 0.272,
"grad_norm": 18.75,
"learning_rate": 1.1266666666666668e-05,
"loss": 18.2054,
"mean_token_accuracy": 0.740831508487463,
"num_tokens": 4810284.0,
"step": 170
},
{
"entropy": 1.125382575765252,
"epoch": 0.288,
"grad_norm": 17.875,
"learning_rate": 1.1933333333333335e-05,
"loss": 17.7696,
"mean_token_accuracy": 0.7443546559661627,
"num_tokens": 5095104.0,
"step": 180
},
{
"entropy": 1.1154642656445504,
"epoch": 0.304,
"grad_norm": 17.125,
"learning_rate": 1.2600000000000001e-05,
"loss": 17.7511,
"mean_token_accuracy": 0.7442465547472239,
"num_tokens": 5383732.0,
"step": 190
},
{
"entropy": 1.0828186847269534,
"epoch": 0.32,
"grad_norm": 19.875,
"learning_rate": 1.3266666666666668e-05,
"loss": 17.2671,
"mean_token_accuracy": 0.7503975994884968,
"num_tokens": 5676334.0,
"step": 200
},
{
"epoch": 0.32,
"eval_biology_entropy": 3.6005834312438965,
"eval_biology_loss": 4.470179557800293,
"eval_biology_mean_token_accuracy": 0.3823731031417847,
"eval_biology_num_tokens": 5676334.0,
"eval_biology_runtime": 49.5338,
"eval_biology_samples_per_second": 10.094,
"eval_biology_steps_per_second": 2.524,
"step": 200
},
{
"epoch": 0.32,
"eval_chemistry_entropy": 2.351666664123535,
"eval_chemistry_loss": 2.5015459060668945,
"eval_chemistry_mean_token_accuracy": 0.5612157773971558,
"eval_chemistry_num_tokens": 5676334.0,
"eval_chemistry_runtime": 61.4417,
"eval_chemistry_samples_per_second": 8.138,
"eval_chemistry_steps_per_second": 2.034,
"step": 200
},
{
"epoch": 0.32,
"eval_math_entropy": 1.1107734112739562,
"eval_math_loss": 1.1035394668579102,
"eval_math_mean_token_accuracy": 0.744808388710022,
"eval_math_num_tokens": 5676334.0,
"eval_math_runtime": 63.0675,
"eval_math_samples_per_second": 7.928,
"eval_math_steps_per_second": 1.982,
"step": 200
},
{
"epoch": 0.32,
"eval_physics_entropy": 1.8196128549575805,
"eval_physics_loss": 1.8889656066894531,
"eval_physics_mean_token_accuracy": 0.6360980491638184,
"eval_physics_num_tokens": 5676334.0,
"eval_physics_runtime": 71.7937,
"eval_physics_samples_per_second": 6.964,
"eval_physics_steps_per_second": 1.741,
"step": 200
},
{
"entropy": 1.0973670542240144,
"epoch": 0.336,
"grad_norm": 19.0,
"learning_rate": 1.3933333333333334e-05,
"loss": 17.6111,
"mean_token_accuracy": 0.7442069798707962,
"num_tokens": 5958480.0,
"step": 210
},
{
"entropy": 1.064722668007016,
"epoch": 0.352,
"grad_norm": 18.25,
"learning_rate": 1.46e-05,
"loss": 16.8683,
"mean_token_accuracy": 0.752158772200346,
"num_tokens": 6242161.0,
"step": 220
},
{
"entropy": 1.0617528311908244,
"epoch": 0.368,
"grad_norm": 20.5,
"learning_rate": 1.5266666666666667e-05,
"loss": 16.9997,
"mean_token_accuracy": 0.751146937161684,
"num_tokens": 6523679.0,
"step": 230
},
{
"entropy": 1.0443669583648443,
"epoch": 0.384,
"grad_norm": 19.625,
"learning_rate": 1.5933333333333336e-05,
"loss": 16.6591,
"mean_token_accuracy": 0.7551476296037436,
"num_tokens": 6810978.0,
"step": 240
},
{
"entropy": 1.0753562293946743,
"epoch": 0.4,
"grad_norm": 17.875,
"learning_rate": 1.66e-05,
"loss": 17.4353,
"mean_token_accuracy": 0.7485508523881436,
"num_tokens": 7096903.0,
"step": 250
},
{
"entropy": 1.0416356481611728,
"epoch": 0.416,
"grad_norm": 17.375,
"learning_rate": 1.726666666666667e-05,
"loss": 16.623,
"mean_token_accuracy": 0.7539484884589911,
"num_tokens": 7377181.0,
"step": 260
},
{
"entropy": 1.0091398146003485,
"epoch": 0.432,
"grad_norm": 20.375,
"learning_rate": 1.7933333333333333e-05,
"loss": 16.1553,
"mean_token_accuracy": 0.7619809154421091,
"num_tokens": 7650523.0,
"step": 270
},
{
"entropy": 0.9927664395421744,
"epoch": 0.448,
"grad_norm": 20.75,
"learning_rate": 1.86e-05,
"loss": 15.9549,
"mean_token_accuracy": 0.7623589888215065,
"num_tokens": 7936788.0,
"step": 280
},
{
"entropy": 0.9981700140982867,
"epoch": 0.464,
"grad_norm": 21.75,
"learning_rate": 1.926666666666667e-05,
"loss": 16.1264,
"mean_token_accuracy": 0.761608887091279,
"num_tokens": 8223147.0,
"step": 290
},
{
"entropy": 1.0235895983874799,
"epoch": 0.48,
"grad_norm": 19.375,
"learning_rate": 1.9933333333333334e-05,
"loss": 16.2896,
"mean_token_accuracy": 0.7603169117122889,
"num_tokens": 8506030.0,
"step": 300
},
{
"epoch": 0.48,
"eval_biology_entropy": 3.66986682510376,
"eval_biology_loss": 4.148224353790283,
"eval_biology_mean_token_accuracy": 0.3987880692481995,
"eval_biology_num_tokens": 8506030.0,
"eval_biology_runtime": 49.5301,
"eval_biology_samples_per_second": 10.095,
"eval_biology_steps_per_second": 2.524,
"step": 300
},
{
"epoch": 0.48,
"eval_chemistry_entropy": 2.309468500137329,
"eval_chemistry_loss": 2.3434462547302246,
"eval_chemistry_mean_token_accuracy": 0.5782731935977936,
"eval_chemistry_num_tokens": 8506030.0,
"eval_chemistry_runtime": 61.4332,
"eval_chemistry_samples_per_second": 8.139,
"eval_chemistry_steps_per_second": 2.035,
"step": 300
},
{
"epoch": 0.48,
"eval_math_entropy": 1.007653223514557,
"eval_math_loss": 1.0127896070480347,
"eval_math_mean_token_accuracy": 0.7593745794296265,
"eval_math_num_tokens": 8506030.0,
"eval_math_runtime": 63.0217,
"eval_math_samples_per_second": 7.934,
"eval_math_steps_per_second": 1.983,
"step": 300
},
{
"epoch": 0.48,
"eval_physics_entropy": 1.7608964786529542,
"eval_physics_loss": 1.7697628736495972,
"eval_physics_mean_token_accuracy": 0.6516191956996917,
"eval_physics_num_tokens": 8506030.0,
"eval_physics_runtime": 71.8045,
"eval_physics_samples_per_second": 6.963,
"eval_physics_steps_per_second": 1.741,
"step": 300
},
{
"entropy": 0.9751624643802643,
"epoch": 0.496,
"grad_norm": 27.0,
"learning_rate": 1.9933333333333334e-05,
"loss": 15.6472,
"mean_token_accuracy": 0.7660828057676554,
"num_tokens": 8788726.0,
"step": 310
},
{
"entropy": 0.9841579314321279,
"epoch": 0.512,
"grad_norm": 19.5,
"learning_rate": 1.985925925925926e-05,
"loss": 15.7341,
"mean_token_accuracy": 0.765475382655859,
"num_tokens": 9078039.0,
"step": 320
},
{
"entropy": 0.9567954104393721,
"epoch": 0.528,
"grad_norm": 22.875,
"learning_rate": 1.9785185185185187e-05,
"loss": 15.3963,
"mean_token_accuracy": 0.7705613534897566,
"num_tokens": 9373860.0,
"step": 330
},
{
"entropy": 0.9622254882007837,
"epoch": 0.544,
"grad_norm": 20.625,
"learning_rate": 1.971111111111111e-05,
"loss": 15.6284,
"mean_token_accuracy": 0.7682077161967754,
"num_tokens": 9660940.0,
"step": 340
},
{
"entropy": 1.013891428336501,
"epoch": 0.56,
"grad_norm": 28.875,
"learning_rate": 1.963703703703704e-05,
"loss": 16.1123,
"mean_token_accuracy": 0.758896803110838,
"num_tokens": 9932302.0,
"step": 350
},
{
"entropy": 0.9643709316849709,
"epoch": 0.576,
"grad_norm": 20.0,
"learning_rate": 1.9562962962962964e-05,
"loss": 15.4925,
"mean_token_accuracy": 0.7669644717127084,
"num_tokens": 10215462.0,
"step": 360
},
{
"entropy": 0.9496656563133001,
"epoch": 0.592,
"grad_norm": 18.5,
"learning_rate": 1.948888888888889e-05,
"loss": 15.3772,
"mean_token_accuracy": 0.7689070332795381,
"num_tokens": 10504396.0,
"step": 370
},
{
"entropy": 0.9635139308869839,
"epoch": 0.608,
"grad_norm": 17.5,
"learning_rate": 1.9414814814814817e-05,
"loss": 15.4665,
"mean_token_accuracy": 0.768926914036274,
"num_tokens": 10793126.0,
"step": 380
},
{
"entropy": 0.9394328493624926,
"epoch": 0.624,
"grad_norm": 19.375,
"learning_rate": 1.9340740740740743e-05,
"loss": 15.1417,
"mean_token_accuracy": 0.7703750677406788,
"num_tokens": 11081768.0,
"step": 390
},
{
"entropy": 0.930184380710125,
"epoch": 0.64,
"grad_norm": 18.25,
"learning_rate": 1.926666666666667e-05,
"loss": 14.8657,
"mean_token_accuracy": 0.7726314965635538,
"num_tokens": 11370320.0,
"step": 400
},
{
"epoch": 0.64,
"eval_biology_entropy": 3.578768089294434,
"eval_biology_loss": 4.099727630615234,
"eval_biology_mean_token_accuracy": 0.4042117977142334,
"eval_biology_num_tokens": 11370320.0,
"eval_biology_runtime": 49.5083,
"eval_biology_samples_per_second": 10.099,
"eval_biology_steps_per_second": 2.525,
"step": 400
},
{
"epoch": 0.64,
"eval_chemistry_entropy": 2.1963028135299685,
"eval_chemistry_loss": 2.250903367996216,
"eval_chemistry_mean_token_accuracy": 0.588011109828949,
"eval_chemistry_num_tokens": 11370320.0,
"eval_chemistry_runtime": 61.4213,
"eval_chemistry_samples_per_second": 8.14,
"eval_chemistry_steps_per_second": 2.035,
"step": 400
},
{
"epoch": 0.64,
"eval_math_entropy": 0.9574988222122193,
"eval_math_loss": 0.9642173051834106,
"eval_math_mean_token_accuracy": 0.7676870808601379,
"eval_math_num_tokens": 11370320.0,
"eval_math_runtime": 62.9918,
"eval_math_samples_per_second": 7.938,
"eval_math_steps_per_second": 1.984,
"step": 400
},
{
"epoch": 0.64,
"eval_physics_entropy": 1.6637140741348266,
"eval_physics_loss": 1.7104088068008423,
"eval_physics_mean_token_accuracy": 0.6602483534812927,
"eval_physics_num_tokens": 11370320.0,
"eval_physics_runtime": 71.7644,
"eval_physics_samples_per_second": 6.967,
"eval_physics_steps_per_second": 1.742,
"step": 400
},
{
"entropy": 0.908730499073863,
"epoch": 0.656,
"grad_norm": 16.625,
"learning_rate": 1.9192592592592593e-05,
"loss": 14.6477,
"mean_token_accuracy": 0.7755995105952025,
"num_tokens": 11657835.0,
"step": 410
},
{
"entropy": 0.9500968560576439,
"epoch": 0.672,
"grad_norm": 18.25,
"learning_rate": 1.911851851851852e-05,
"loss": 15.2383,
"mean_token_accuracy": 0.769153194501996,
"num_tokens": 11949262.0,
"step": 420
},
{
"entropy": 0.9080810055136681,
"epoch": 0.688,
"grad_norm": 15.625,
"learning_rate": 1.9044444444444446e-05,
"loss": 14.6139,
"mean_token_accuracy": 0.7764004003256559,
"num_tokens": 12227640.0,
"step": 430
},
{
"entropy": 0.9170140367001295,
"epoch": 0.704,
"grad_norm": 17.625,
"learning_rate": 1.8970370370370372e-05,
"loss": 14.7655,
"mean_token_accuracy": 0.7752966023981571,
"num_tokens": 12516641.0,
"step": 440
},
{
"entropy": 0.9368526700884103,
"epoch": 0.72,
"grad_norm": 19.25,
"learning_rate": 1.8896296296296295e-05,
"loss": 15.134,
"mean_token_accuracy": 0.7706776697188615,
"num_tokens": 12793343.0,
"step": 450
},
{
"entropy": 0.9261821806430817,
"epoch": 0.736,
"grad_norm": 17.5,
"learning_rate": 1.8822222222222225e-05,
"loss": 14.833,
"mean_token_accuracy": 0.7745088890194893,
"num_tokens": 13077981.0,
"step": 460
},
{
"entropy": 0.9040410481393337,
"epoch": 0.752,
"grad_norm": 19.75,
"learning_rate": 1.874814814814815e-05,
"loss": 14.4907,
"mean_token_accuracy": 0.7791126936674118,
"num_tokens": 13358957.0,
"step": 470
},
{
"entropy": 0.8739457067102194,
"epoch": 0.768,
"grad_norm": 15.8125,
"learning_rate": 1.8674074074074075e-05,
"loss": 14.112,
"mean_token_accuracy": 0.7853138707578182,
"num_tokens": 13653412.0,
"step": 480
},
{
"entropy": 0.946398203074932,
"epoch": 0.784,
"grad_norm": 19.875,
"learning_rate": 1.86e-05,
"loss": 15.145,
"mean_token_accuracy": 0.7723897945135831,
"num_tokens": 13940856.0,
"step": 490
},
{
"entropy": 0.9361972361803055,
"epoch": 0.8,
"grad_norm": 19.625,
"learning_rate": 1.8525925925925928e-05,
"loss": 15.2105,
"mean_token_accuracy": 0.7712519209831953,
"num_tokens": 14230754.0,
"step": 500
},
{
"epoch": 0.8,
"eval_biology_entropy": 3.714933115005493,
"eval_biology_loss": 3.970184564590454,
"eval_biology_mean_token_accuracy": 0.41206827068328855,
"eval_biology_num_tokens": 14230754.0,
"eval_biology_runtime": 49.5281,
"eval_biology_samples_per_second": 10.095,
"eval_biology_steps_per_second": 2.524,
"step": 500
},
{
"epoch": 0.8,
"eval_chemistry_entropy": 2.237494807243347,
"eval_chemistry_loss": 2.1822235584259033,
"eval_chemistry_mean_token_accuracy": 0.5956678519248962,
"eval_chemistry_num_tokens": 14230754.0,
"eval_chemistry_runtime": 61.0757,
"eval_chemistry_samples_per_second": 8.187,
"eval_chemistry_steps_per_second": 2.047,
"step": 500
},
{
"epoch": 0.8,
"eval_math_entropy": 0.9503358464241027,
"eval_math_loss": 0.9351838827133179,
"eval_math_mean_token_accuracy": 0.7728546581268311,
"eval_math_num_tokens": 14230754.0,
"eval_math_runtime": 63.0753,
"eval_math_samples_per_second": 7.927,
"eval_math_steps_per_second": 1.982,
"step": 500
},
{
"epoch": 0.8,
"eval_physics_entropy": 1.6920014672279358,
"eval_physics_loss": 1.671562671661377,
"eval_physics_mean_token_accuracy": 0.6658579094409942,
"eval_physics_num_tokens": 14230754.0,
"eval_physics_runtime": 71.8921,
"eval_physics_samples_per_second": 6.955,
"eval_physics_steps_per_second": 1.739,
"step": 500
},
{
"entropy": 0.9170173428952694,
"epoch": 0.816,
"grad_norm": 18.625,
"learning_rate": 1.8451851851851855e-05,
"loss": 14.5639,
"mean_token_accuracy": 0.7766096990555524,
"num_tokens": 14519893.0,
"step": 510
},
{
"entropy": 0.9109945900738239,
"epoch": 0.832,
"grad_norm": 18.5,
"learning_rate": 1.8377777777777778e-05,
"loss": 14.6409,
"mean_token_accuracy": 0.7777223017066717,
"num_tokens": 14805088.0,
"step": 520
},
{
"entropy": 0.9337830897420645,
"epoch": 0.848,
"grad_norm": 19.0,
"learning_rate": 1.8303703703703704e-05,
"loss": 15.0412,
"mean_token_accuracy": 0.7716363571584225,
"num_tokens": 15086306.0,
"step": 530
},
{
"entropy": 0.9221564382314682,
"epoch": 0.864,
"grad_norm": 15.8125,
"learning_rate": 1.822962962962963e-05,
"loss": 14.7588,
"mean_token_accuracy": 0.7748291842639446,
"num_tokens": 15370985.0,
"step": 540
},
{
"entropy": 0.9273673612624407,
"epoch": 0.88,
"grad_norm": 21.0,
"learning_rate": 1.8155555555555557e-05,
"loss": 14.9233,
"mean_token_accuracy": 0.7736149989068508,
"num_tokens": 15653836.0,
"step": 550
},
{
"entropy": 0.9161856021732092,
"epoch": 0.896,
"grad_norm": 21.75,
"learning_rate": 1.8081481481481484e-05,
"loss": 14.7618,
"mean_token_accuracy": 0.7746457736939192,
"num_tokens": 15932179.0,
"step": 560
},
{
"entropy": 0.8957702405750751,
"epoch": 0.912,
"grad_norm": 18.75,
"learning_rate": 1.800740740740741e-05,
"loss": 14.2902,
"mean_token_accuracy": 0.7794793792068958,
"num_tokens": 16219842.0,
"step": 570
},
{
"entropy": 0.8888734232634306,
"epoch": 0.928,
"grad_norm": 20.875,
"learning_rate": 1.7933333333333333e-05,
"loss": 14.2919,
"mean_token_accuracy": 0.7802723813802004,
"num_tokens": 16499842.0,
"step": 580
},
{
"entropy": 0.9097447639331222,
"epoch": 0.944,
"grad_norm": 35.0,
"learning_rate": 1.785925925925926e-05,
"loss": 14.7214,
"mean_token_accuracy": 0.7760318968445062,
"num_tokens": 16781882.0,
"step": 590
},
{
"entropy": 0.9282805230468512,
"epoch": 0.96,
"grad_norm": 16.125,
"learning_rate": 1.7785185185185186e-05,
"loss": 14.8274,
"mean_token_accuracy": 0.7749283302575349,
"num_tokens": 17067407.0,
"step": 600
},
{
"epoch": 0.96,
"eval_biology_entropy": 3.4800376834869384,
"eval_biology_loss": 3.9260165691375732,
"eval_biology_mean_token_accuracy": 0.41479419231414794,
"eval_biology_num_tokens": 17067407.0,
"eval_biology_runtime": 49.593,
"eval_biology_samples_per_second": 10.082,
"eval_biology_steps_per_second": 2.521,
"step": 600
},
{
"epoch": 0.96,
"eval_chemistry_entropy": 2.062796443462372,
"eval_chemistry_loss": 2.1318087577819824,
"eval_chemistry_mean_token_accuracy": 0.6000087871551514,
"eval_chemistry_num_tokens": 17067407.0,
"eval_chemistry_runtime": 61.4719,
"eval_chemistry_samples_per_second": 8.134,
"eval_chemistry_steps_per_second": 2.033,
"step": 600
},
{
"epoch": 0.96,
"eval_math_entropy": 0.9119843735694885,
"eval_math_loss": 0.9157347679138184,
"eval_math_mean_token_accuracy": 0.7754620661735535,
"eval_math_num_tokens": 17067407.0,
"eval_math_runtime": 63.0502,
"eval_math_samples_per_second": 7.93,
"eval_math_steps_per_second": 1.983,
"step": 600
},
{
"epoch": 0.96,
"eval_physics_entropy": 1.5833718237876893,
"eval_physics_loss": 1.6547900438308716,
"eval_physics_mean_token_accuracy": 0.6688490805625915,
"eval_physics_num_tokens": 17067407.0,
"eval_physics_runtime": 72.0729,
"eval_physics_samples_per_second": 6.937,
"eval_physics_steps_per_second": 1.734,
"step": 600
},
{
"entropy": 0.8985319763422013,
"epoch": 0.976,
"grad_norm": 21.625,
"learning_rate": 1.7711111111111113e-05,
"loss": 14.4151,
"mean_token_accuracy": 0.7804601702839136,
"num_tokens": 17350994.0,
"step": 610
},
{
"entropy": 0.9126239087432623,
"epoch": 0.992,
"grad_norm": 18.0,
"learning_rate": 1.763703703703704e-05,
"loss": 14.6522,
"mean_token_accuracy": 0.7754363786429167,
"num_tokens": 17637514.0,
"step": 620
},
{
"entropy": 0.8935651954263448,
"epoch": 1.008,
"grad_norm": 17.375,
"learning_rate": 1.7562962962962962e-05,
"loss": 14.2584,
"mean_token_accuracy": 0.7789396952837706,
"num_tokens": 17926570.0,
"step": 630
},
{
"entropy": 0.8810664782300591,
"epoch": 1.024,
"grad_norm": 17.625,
"learning_rate": 1.7488888888888892e-05,
"loss": 14.0978,
"mean_token_accuracy": 0.7811610117554665,
"num_tokens": 18207652.0,
"step": 640
},
{
"entropy": 0.8790556967258454,
"epoch": 1.04,
"grad_norm": 20.375,
"learning_rate": 1.7414814814814815e-05,
"loss": 14.0815,
"mean_token_accuracy": 0.782483147084713,
"num_tokens": 18484931.0,
"step": 650
},
{
"entropy": 0.872769825719297,
"epoch": 1.056,
"grad_norm": 18.625,
"learning_rate": 1.7340740740740742e-05,
"loss": 14.2329,
"mean_token_accuracy": 0.7814408276230097,
"num_tokens": 18773457.0,
"step": 660
},
{
"entropy": 0.8854838136583567,
"epoch": 1.072,
"grad_norm": 19.125,
"learning_rate": 1.726666666666667e-05,
"loss": 14.1351,
"mean_token_accuracy": 0.7804106209427119,
"num_tokens": 19055365.0,
"step": 670
},
{
"entropy": 0.8737944336608052,
"epoch": 1.088,
"grad_norm": 16.625,
"learning_rate": 1.7192592592592595e-05,
"loss": 14.0953,
"mean_token_accuracy": 0.7833730541169643,
"num_tokens": 19345730.0,
"step": 680
},
{
"entropy": 0.8823883978649973,
"epoch": 1.104,
"grad_norm": 15.1875,
"learning_rate": 1.711851851851852e-05,
"loss": 14.1044,
"mean_token_accuracy": 0.7833204790949821,
"num_tokens": 19637390.0,
"step": 690
},
{
"entropy": 0.8538309209048748,
"epoch": 1.12,
"grad_norm": 19.5,
"learning_rate": 1.7044444444444445e-05,
"loss": 13.6987,
"mean_token_accuracy": 0.7868878465145827,
"num_tokens": 19923914.0,
"step": 700
},
{
"epoch": 1.12,
"eval_biology_entropy": 3.443337703704834,
"eval_biology_loss": 3.951155662536621,
"eval_biology_mean_token_accuracy": 0.4163366384506226,
"eval_biology_num_tokens": 19923914.0,
"eval_biology_runtime": 49.651,
"eval_biology_samples_per_second": 10.07,
"eval_biology_steps_per_second": 2.518,
"step": 700
},
{
"epoch": 1.12,
"eval_chemistry_entropy": 2.0116386432647704,
"eval_chemistry_loss": 2.1324052810668945,
"eval_chemistry_mean_token_accuracy": 0.6028851909637452,
"eval_chemistry_num_tokens": 19923914.0,
"eval_chemistry_runtime": 61.5705,
"eval_chemistry_samples_per_second": 8.121,
"eval_chemistry_steps_per_second": 2.03,
"step": 700
},
{
"epoch": 1.12,
"eval_math_entropy": 0.8889193925857544,
"eval_math_loss": 0.9035109877586365,
"eval_math_mean_token_accuracy": 0.7773573837280273,
"eval_math_num_tokens": 19923914.0,
"eval_math_runtime": 63.1467,
"eval_math_samples_per_second": 7.918,
"eval_math_steps_per_second": 1.98,
"step": 700
},
{
"epoch": 1.12,
"eval_physics_entropy": 1.5386076803207398,
"eval_physics_loss": 1.6331955194473267,
"eval_physics_mean_token_accuracy": 0.6719050664901733,
"eval_physics_num_tokens": 19923914.0,
"eval_physics_runtime": 72.2366,
"eval_physics_samples_per_second": 6.922,
"eval_physics_steps_per_second": 1.73,
"step": 700
},
{
"entropy": 0.8486799146980047,
"epoch": 1.1360000000000001,
"grad_norm": 18.375,
"learning_rate": 1.697037037037037e-05,
"loss": 13.6728,
"mean_token_accuracy": 0.7864403679966927,
"num_tokens": 20201892.0,
"step": 710
},
{
"entropy": 0.8742417544126511,
"epoch": 1.152,
"grad_norm": 20.0,
"learning_rate": 1.6896296296296298e-05,
"loss": 14.0206,
"mean_token_accuracy": 0.7843449763953686,
"num_tokens": 20490282.0,
"step": 720
},
{
"entropy": 0.8681927429512143,
"epoch": 1.168,
"grad_norm": 16.625,
"learning_rate": 1.6822222222222224e-05,
"loss": 13.9888,
"mean_token_accuracy": 0.7844074387103319,
"num_tokens": 20785786.0,
"step": 730
},
{
"entropy": 0.8594187637791038,
"epoch": 1.184,
"grad_norm": 20.875,
"learning_rate": 1.6748148148148147e-05,
"loss": 13.8341,
"mean_token_accuracy": 0.7861495897173881,
"num_tokens": 21074205.0,
"step": 740
},
{
"entropy": 0.853638318926096,
"epoch": 1.2,
"grad_norm": 20.25,
"learning_rate": 1.6674074074074077e-05,
"loss": 13.7126,
"mean_token_accuracy": 0.7865742921829224,
"num_tokens": 21369159.0,
"step": 750
},
{
"entropy": 0.8770630281418562,
"epoch": 1.216,
"grad_norm": 23.25,
"learning_rate": 1.66e-05,
"loss": 14.0894,
"mean_token_accuracy": 0.7835288234055042,
"num_tokens": 21649178.0,
"step": 760
},
{
"entropy": 0.8497887661680579,
"epoch": 1.232,
"grad_norm": 19.25,
"learning_rate": 1.6525925925925927e-05,
"loss": 13.5638,
"mean_token_accuracy": 0.7882835403084755,
"num_tokens": 21930239.0,
"step": 770
},
{
"entropy": 0.8405620498582721,
"epoch": 1.248,
"grad_norm": 17.875,
"learning_rate": 1.6451851851851853e-05,
"loss": 13.5058,
"mean_token_accuracy": 0.7887706879526377,
"num_tokens": 22216387.0,
"step": 780
},
{
"entropy": 0.8788484372198582,
"epoch": 1.264,
"grad_norm": 20.125,
"learning_rate": 1.637777777777778e-05,
"loss": 14.209,
"mean_token_accuracy": 0.7804543025791645,
"num_tokens": 22501002.0,
"step": 790
},
{
"entropy": 0.8771140918135643,
"epoch": 1.28,
"grad_norm": 22.25,
"learning_rate": 1.6303703703703706e-05,
"loss": 13.9869,
"mean_token_accuracy": 0.7842618364840745,
"num_tokens": 22779682.0,
"step": 800
},
{
"epoch": 1.28,
"eval_biology_entropy": 3.436663423538208,
"eval_biology_loss": 3.9039478302001953,
"eval_biology_mean_token_accuracy": 0.41795943784713746,
"eval_biology_num_tokens": 22779682.0,
"eval_biology_runtime": 49.6843,
"eval_biology_samples_per_second": 10.064,
"eval_biology_steps_per_second": 2.516,
"step": 800
},
{
"epoch": 1.28,
"eval_chemistry_entropy": 1.97736017036438,
"eval_chemistry_loss": 2.080076217651367,
"eval_chemistry_mean_token_accuracy": 0.606924372434616,
"eval_chemistry_num_tokens": 22779682.0,
"eval_chemistry_runtime": 61.5504,
"eval_chemistry_samples_per_second": 8.123,
"eval_chemistry_steps_per_second": 2.031,
"step": 800
},
{
"epoch": 1.28,
"eval_math_entropy": 0.8625033555030823,
"eval_math_loss": 0.892142653465271,
"eval_math_mean_token_accuracy": 0.7796395468711853,
"eval_math_num_tokens": 22779682.0,
"eval_math_runtime": 63.1583,
"eval_math_samples_per_second": 7.917,
"eval_math_steps_per_second": 1.979,
"step": 800
},
{
"epoch": 1.28,
"eval_physics_entropy": 1.5225397615432739,
"eval_physics_loss": 1.612709879875183,
"eval_physics_mean_token_accuracy": 0.674011738061905,
"eval_physics_num_tokens": 22779682.0,
"eval_physics_runtime": 72.1546,
"eval_physics_samples_per_second": 6.93,
"eval_physics_steps_per_second": 1.732,
"step": 800
},
{
"entropy": 0.8474511569365859,
"epoch": 1.296,
"grad_norm": 17.75,
"learning_rate": 1.622962962962963e-05,
"loss": 13.6822,
"mean_token_accuracy": 0.7866408120840788,
"num_tokens": 23057744.0,
"step": 810
},
{
"entropy": 0.8580924984067678,
"epoch": 1.312,
"grad_norm": 21.5,
"learning_rate": 1.6155555555555556e-05,
"loss": 13.9676,
"mean_token_accuracy": 0.7842281598597765,
"num_tokens": 23344644.0,
"step": 820
},
{
"entropy": 0.8850870780646801,
"epoch": 1.328,
"grad_norm": 21.5,
"learning_rate": 1.6081481481481482e-05,
"loss": 14.0615,
"mean_token_accuracy": 0.781786335259676,
"num_tokens": 23622405.0,
"step": 830
},
{
"entropy": 0.8474001951515675,
"epoch": 1.3439999999999999,
"grad_norm": 18.25,
"learning_rate": 1.600740740740741e-05,
"loss": 13.6058,
"mean_token_accuracy": 0.7873330093920231,
"num_tokens": 23899771.0,
"step": 840
},
{
"entropy": 0.8435361562296748,
"epoch": 1.3599999999999999,
"grad_norm": 16.0,
"learning_rate": 1.5933333333333336e-05,
"loss": 13.6319,
"mean_token_accuracy": 0.7882646590471267,
"num_tokens": 24187023.0,
"step": 850
},
{
"entropy": 0.9156765401363373,
"epoch": 1.376,
"grad_norm": 21.5,
"learning_rate": 1.5859259259259262e-05,
"loss": 14.6843,
"mean_token_accuracy": 0.7743376530706882,
"num_tokens": 24466132.0,
"step": 860
},
{
"entropy": 0.8523435527458787,
"epoch": 1.392,
"grad_norm": 20.75,
"learning_rate": 1.5785185185185185e-05,
"loss": 13.6998,
"mean_token_accuracy": 0.7863496944308281,
"num_tokens": 24748043.0,
"step": 870
},
{
"entropy": 0.8375731509178876,
"epoch": 1.408,
"grad_norm": 21.125,
"learning_rate": 1.571111111111111e-05,
"loss": 13.5188,
"mean_token_accuracy": 0.7898116644471884,
"num_tokens": 25036674.0,
"step": 880
},
{
"entropy": 0.8376417780295015,
"epoch": 1.424,
"grad_norm": 15.25,
"learning_rate": 1.5637037037037038e-05,
"loss": 13.4567,
"mean_token_accuracy": 0.7908716473728419,
"num_tokens": 25324579.0,
"step": 890
},
{
"entropy": 0.8603908697143197,
"epoch": 1.44,
"grad_norm": 19.125,
"learning_rate": 1.5562962962962965e-05,
"loss": 13.9612,
"mean_token_accuracy": 0.7841609816998243,
"num_tokens": 25606824.0,
"step": 900
},
{
"epoch": 1.44,
"eval_biology_entropy": 3.45110001373291,
"eval_biology_loss": 3.8754560947418213,
"eval_biology_mean_token_accuracy": 0.4207558875083923,
"eval_biology_num_tokens": 25606824.0,
"eval_biology_runtime": 49.6588,
"eval_biology_samples_per_second": 10.069,
"eval_biology_steps_per_second": 2.517,
"step": 900
},
{
"epoch": 1.44,
"eval_chemistry_entropy": 1.9984806547164917,
"eval_chemistry_loss": 2.088613748550415,
"eval_chemistry_mean_token_accuracy": 0.607667599439621,
"eval_chemistry_num_tokens": 25606824.0,
"eval_chemistry_runtime": 61.604,
"eval_chemistry_samples_per_second": 8.116,
"eval_chemistry_steps_per_second": 2.029,
"step": 900
},
{
"epoch": 1.44,
"eval_math_entropy": 0.8740571973323822,
"eval_math_loss": 0.8836535811424255,
"eval_math_mean_token_accuracy": 0.7810649185180664,
"eval_math_num_tokens": 25606824.0,
"eval_math_runtime": 63.1568,
"eval_math_samples_per_second": 7.917,
"eval_math_steps_per_second": 1.979,
"step": 900
},
{
"epoch": 1.44,
"eval_physics_entropy": 1.533125587463379,
"eval_physics_loss": 1.6004778146743774,
"eval_physics_mean_token_accuracy": 0.6762587761878968,
"eval_physics_num_tokens": 25606824.0,
"eval_physics_runtime": 71.9331,
"eval_physics_samples_per_second": 6.951,
"eval_physics_steps_per_second": 1.738,
"step": 900
},
{
"entropy": 0.8511543165892362,
"epoch": 1.456,
"grad_norm": 16.125,
"learning_rate": 1.548888888888889e-05,
"loss": 13.5133,
"mean_token_accuracy": 0.7876119159162045,
"num_tokens": 25886396.0,
"step": 910
},
{
"entropy": 0.8463271845132112,
"epoch": 1.472,
"grad_norm": 17.75,
"learning_rate": 1.5414814814814814e-05,
"loss": 13.5862,
"mean_token_accuracy": 0.787867683917284,
"num_tokens": 26163618.0,
"step": 920
},
{
"entropy": 0.8713837819173932,
"epoch": 1.488,
"grad_norm": 20.125,
"learning_rate": 1.5340740740740744e-05,
"loss": 14.1889,
"mean_token_accuracy": 0.7813853111118079,
"num_tokens": 26438338.0,
"step": 930
},
{
"entropy": 0.834681292809546,
"epoch": 1.504,
"grad_norm": 21.875,
"learning_rate": 1.5266666666666667e-05,
"loss": 13.4447,
"mean_token_accuracy": 0.7905492588877678,
"num_tokens": 26729255.0,
"step": 940
},
{
"entropy": 0.8771929260343313,
"epoch": 1.52,
"grad_norm": 18.0,
"learning_rate": 1.5192592592592594e-05,
"loss": 14.0064,
"mean_token_accuracy": 0.7831294402480126,
"num_tokens": 27017935.0,
"step": 950
},
{
"entropy": 0.8460531679913401,
"epoch": 1.536,
"grad_norm": 16.125,
"learning_rate": 1.5118518518518519e-05,
"loss": 13.6594,
"mean_token_accuracy": 0.786826192215085,
"num_tokens": 27306339.0,
"step": 960
},
{
"entropy": 0.8540312614291906,
"epoch": 1.552,
"grad_norm": 18.75,
"learning_rate": 1.5044444444444445e-05,
"loss": 13.6638,
"mean_token_accuracy": 0.7870731830596924,
"num_tokens": 27591959.0,
"step": 970
},
{
"entropy": 0.8264467475935817,
"epoch": 1.568,
"grad_norm": 16.125,
"learning_rate": 1.497037037037037e-05,
"loss": 13.3238,
"mean_token_accuracy": 0.7920118164271116,
"num_tokens": 27884398.0,
"step": 980
},
{
"entropy": 0.8386921418830753,
"epoch": 1.584,
"grad_norm": 20.25,
"learning_rate": 1.4896296296296298e-05,
"loss": 13.5938,
"mean_token_accuracy": 0.7884967364370823,
"num_tokens": 28171274.0,
"step": 990
},
{
"entropy": 0.8454421229660511,
"epoch": 1.6,
"grad_norm": 21.125,
"learning_rate": 1.4822222222222225e-05,
"loss": 13.4115,
"mean_token_accuracy": 0.7884562328457833,
"num_tokens": 28457624.0,
"step": 1000
},
{
"epoch": 1.6,
"eval_biology_entropy": 3.4279630908966063,
"eval_biology_loss": 3.9252524375915527,
"eval_biology_mean_token_accuracy": 0.42100021028518675,
"eval_biology_num_tokens": 28457624.0,
"eval_biology_runtime": 49.5484,
"eval_biology_samples_per_second": 10.091,
"eval_biology_steps_per_second": 2.523,
"step": 1000
},
{
"epoch": 1.6,
"eval_chemistry_entropy": 1.9675587797164917,
"eval_chemistry_loss": 2.0829849243164062,
"eval_chemistry_mean_token_accuracy": 0.6087198803424835,
"eval_chemistry_num_tokens": 28457624.0,
"eval_chemistry_runtime": 60.8881,
"eval_chemistry_samples_per_second": 8.212,
"eval_chemistry_steps_per_second": 2.053,
"step": 1000
},
{
"epoch": 1.6,
"eval_math_entropy": 0.858300416469574,
"eval_math_loss": 0.8771592974662781,
"eval_math_mean_token_accuracy": 0.7824492840766907,
"eval_math_num_tokens": 28457624.0,
"eval_math_runtime": 62.7471,
"eval_math_samples_per_second": 7.968,
"eval_math_steps_per_second": 1.992,
"step": 1000
},
{
"epoch": 1.6,
"eval_physics_entropy": 1.5079545302391053,
"eval_physics_loss": 1.5952943563461304,
"eval_physics_mean_token_accuracy": 0.6772087650299072,
"eval_physics_num_tokens": 28457624.0,
"eval_physics_runtime": 71.6735,
"eval_physics_samples_per_second": 6.976,
"eval_physics_steps_per_second": 1.744,
"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": 2.897870312538204e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}