2sybub3b / checkpoint-1000 /trainer_state.json
roonbug's picture
Upload folder using huggingface_hub
a9db18d 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.9036113740876317,
"epoch": 0.016,
"grad_norm": 128.0,
"learning_rate": 6.000000000000001e-07,
"loss": 34.907,
"mean_token_accuracy": 0.6652832090854645,
"num_tokens": 338352.0,
"step": 10
},
{
"entropy": 0.9087308555841446,
"epoch": 0.032,
"grad_norm": 119.5,
"learning_rate": 1.2666666666666669e-06,
"loss": 34.745,
"mean_token_accuracy": 0.6691140927374363,
"num_tokens": 671193.0,
"step": 20
},
{
"entropy": 0.9059555266052485,
"epoch": 0.048,
"grad_norm": 109.5,
"learning_rate": 1.9333333333333336e-06,
"loss": 34.1928,
"mean_token_accuracy": 0.6685802672058344,
"num_tokens": 1001008.0,
"step": 30
},
{
"entropy": 0.9450886320322752,
"epoch": 0.064,
"grad_norm": 102.0,
"learning_rate": 2.6e-06,
"loss": 34.8515,
"mean_token_accuracy": 0.6614982625469565,
"num_tokens": 1331208.0,
"step": 40
},
{
"entropy": 0.9593638770282269,
"epoch": 0.08,
"grad_norm": 94.5,
"learning_rate": 3.266666666666667e-06,
"loss": 33.1046,
"mean_token_accuracy": 0.6680185409262777,
"num_tokens": 1667532.0,
"step": 50
},
{
"entropy": 0.9606023650616408,
"epoch": 0.096,
"grad_norm": 80.0,
"learning_rate": 3.9333333333333335e-06,
"loss": 31.2876,
"mean_token_accuracy": 0.678196857497096,
"num_tokens": 2007176.0,
"step": 60
},
{
"entropy": 1.069316839799285,
"epoch": 0.112,
"grad_norm": 63.25,
"learning_rate": 4.600000000000001e-06,
"loss": 30.7512,
"mean_token_accuracy": 0.6743977405130863,
"num_tokens": 2342259.0,
"step": 70
},
{
"entropy": 1.2628379821777345,
"epoch": 0.128,
"grad_norm": 48.0,
"learning_rate": 5.2666666666666665e-06,
"loss": 30.3935,
"mean_token_accuracy": 0.6708028431981802,
"num_tokens": 2663987.0,
"step": 80
},
{
"entropy": 1.3055987607687711,
"epoch": 0.144,
"grad_norm": 32.25,
"learning_rate": 5.933333333333335e-06,
"loss": 27.3167,
"mean_token_accuracy": 0.6920903529971838,
"num_tokens": 2997824.0,
"step": 90
},
{
"entropy": 1.3579398691654205,
"epoch": 0.16,
"grad_norm": 24.625,
"learning_rate": 6.600000000000001e-06,
"loss": 25.5362,
"mean_token_accuracy": 0.7021430663764476,
"num_tokens": 3330597.0,
"step": 100
},
{
"epoch": 0.16,
"eval_biology_entropy": 3.549320913314819,
"eval_biology_loss": 4.242035388946533,
"eval_biology_mean_token_accuracy": 0.4143077299594879,
"eval_biology_num_tokens": 3330597.0,
"eval_biology_runtime": 55.9019,
"eval_biology_samples_per_second": 8.944,
"eval_biology_steps_per_second": 2.236,
"step": 100
},
{
"epoch": 0.16,
"eval_chemistry_entropy": 1.9644288024902343,
"eval_chemistry_loss": 2.1523141860961914,
"eval_chemistry_mean_token_accuracy": 0.635781112909317,
"eval_chemistry_num_tokens": 3330597.0,
"eval_chemistry_runtime": 69.6126,
"eval_chemistry_samples_per_second": 7.183,
"eval_chemistry_steps_per_second": 1.796,
"step": 100
},
{
"epoch": 0.16,
"eval_math_entropy": 1.1708643288612366,
"eval_math_loss": 1.6400102376937866,
"eval_math_mean_token_accuracy": 0.7028316712379455,
"eval_math_num_tokens": 3330597.0,
"eval_math_runtime": 71.6855,
"eval_math_samples_per_second": 6.975,
"eval_math_steps_per_second": 1.744,
"step": 100
},
{
"epoch": 0.16,
"eval_physics_entropy": 1.388730420589447,
"eval_physics_loss": 1.5701876878738403,
"eval_physics_mean_token_accuracy": 0.7025446329116821,
"eval_physics_num_tokens": 3330597.0,
"eval_physics_runtime": 83.1014,
"eval_physics_samples_per_second": 6.017,
"eval_physics_steps_per_second": 1.504,
"step": 100
},
{
"entropy": 1.3951868526637554,
"epoch": 0.176,
"grad_norm": 23.0,
"learning_rate": 7.266666666666668e-06,
"loss": 24.797,
"mean_token_accuracy": 0.7061628632247448,
"num_tokens": 3658264.0,
"step": 110
},
{
"entropy": 1.3387951213866471,
"epoch": 0.192,
"grad_norm": 15.375,
"learning_rate": 7.933333333333334e-06,
"loss": 23.7842,
"mean_token_accuracy": 0.713041927665472,
"num_tokens": 3995568.0,
"step": 120
},
{
"entropy": 1.3034498687833547,
"epoch": 0.208,
"grad_norm": 17.5,
"learning_rate": 8.6e-06,
"loss": 22.4839,
"mean_token_accuracy": 0.7252091735601425,
"num_tokens": 4321436.0,
"step": 130
},
{
"entropy": 1.2866257604211568,
"epoch": 0.224,
"grad_norm": 14.6875,
"learning_rate": 9.266666666666667e-06,
"loss": 22.0837,
"mean_token_accuracy": 0.7270880017429591,
"num_tokens": 4648491.0,
"step": 140
},
{
"entropy": 1.268964756652713,
"epoch": 0.24,
"grad_norm": 14.25,
"learning_rate": 9.933333333333334e-06,
"loss": 21.6129,
"mean_token_accuracy": 0.7301765695214272,
"num_tokens": 4986175.0,
"step": 150
},
{
"entropy": 1.2642716094851494,
"epoch": 0.256,
"grad_norm": 15.5625,
"learning_rate": 1.0600000000000002e-05,
"loss": 21.2618,
"mean_token_accuracy": 0.734415826946497,
"num_tokens": 5329320.0,
"step": 160
},
{
"entropy": 1.2717196103185415,
"epoch": 0.272,
"grad_norm": 17.5,
"learning_rate": 1.1266666666666668e-05,
"loss": 21.4982,
"mean_token_accuracy": 0.7297716859728098,
"num_tokens": 5658796.0,
"step": 170
},
{
"entropy": 1.2510493908077478,
"epoch": 0.288,
"grad_norm": 14.5625,
"learning_rate": 1.1933333333333335e-05,
"loss": 21.0954,
"mean_token_accuracy": 0.7358391582965851,
"num_tokens": 5980201.0,
"step": 180
},
{
"entropy": 1.2463122855871915,
"epoch": 0.304,
"grad_norm": 19.5,
"learning_rate": 1.2600000000000001e-05,
"loss": 20.8303,
"mean_token_accuracy": 0.7361553307622671,
"num_tokens": 6298858.0,
"step": 190
},
{
"entropy": 1.215079977735877,
"epoch": 0.32,
"grad_norm": 15.875,
"learning_rate": 1.3266666666666668e-05,
"loss": 20.4326,
"mean_token_accuracy": 0.7394854992628097,
"num_tokens": 6622798.0,
"step": 200
},
{
"epoch": 0.32,
"eval_biology_entropy": 3.1598935775756836,
"eval_biology_loss": 3.6490557193756104,
"eval_biology_mean_token_accuracy": 0.45875698828697203,
"eval_biology_num_tokens": 6622798.0,
"eval_biology_runtime": 56.3027,
"eval_biology_samples_per_second": 8.881,
"eval_biology_steps_per_second": 2.22,
"step": 200
},
{
"epoch": 0.32,
"eval_chemistry_entropy": 1.6644244351387023,
"eval_chemistry_loss": 1.768973469734192,
"eval_chemistry_mean_token_accuracy": 0.6738141541481018,
"eval_chemistry_num_tokens": 6622798.0,
"eval_chemistry_runtime": 69.6913,
"eval_chemistry_samples_per_second": 7.174,
"eval_chemistry_steps_per_second": 1.794,
"step": 200
},
{
"epoch": 0.32,
"eval_math_entropy": 1.124363118648529,
"eval_math_loss": 1.4134961366653442,
"eval_math_mean_token_accuracy": 0.7205373592376709,
"eval_math_num_tokens": 6622798.0,
"eval_math_runtime": 71.6804,
"eval_math_samples_per_second": 6.975,
"eval_math_steps_per_second": 1.744,
"step": 200
},
{
"epoch": 0.32,
"eval_physics_entropy": 1.2113610072135925,
"eval_physics_loss": 1.2715808153152466,
"eval_physics_mean_token_accuracy": 0.7393231744766235,
"eval_physics_num_tokens": 6622798.0,
"eval_physics_runtime": 83.0035,
"eval_physics_samples_per_second": 6.024,
"eval_physics_steps_per_second": 1.506,
"step": 200
},
{
"entropy": 1.2057966101914643,
"epoch": 0.336,
"grad_norm": 13.875,
"learning_rate": 1.3933333333333334e-05,
"loss": 20.1637,
"mean_token_accuracy": 0.740947175398469,
"num_tokens": 6953611.0,
"step": 210
},
{
"entropy": 1.1682135988026858,
"epoch": 0.352,
"grad_norm": 14.625,
"learning_rate": 1.46e-05,
"loss": 19.5323,
"mean_token_accuracy": 0.748641949146986,
"num_tokens": 7290378.0,
"step": 220
},
{
"entropy": 1.1533767595887183,
"epoch": 0.368,
"grad_norm": 14.75,
"learning_rate": 1.5266666666666667e-05,
"loss": 19.3393,
"mean_token_accuracy": 0.7496714886277914,
"num_tokens": 7621082.0,
"step": 230
},
{
"entropy": 1.2154120814055205,
"epoch": 0.384,
"grad_norm": 13.875,
"learning_rate": 1.5933333333333336e-05,
"loss": 20.1519,
"mean_token_accuracy": 0.7383838955312967,
"num_tokens": 7955570.0,
"step": 240
},
{
"entropy": 1.162219974398613,
"epoch": 0.4,
"grad_norm": 17.0,
"learning_rate": 1.66e-05,
"loss": 19.3544,
"mean_token_accuracy": 0.7475786007940769,
"num_tokens": 8291049.0,
"step": 250
},
{
"entropy": 1.1398306384682655,
"epoch": 0.416,
"grad_norm": 17.0,
"learning_rate": 1.726666666666667e-05,
"loss": 18.9335,
"mean_token_accuracy": 0.7522856764495373,
"num_tokens": 8627310.0,
"step": 260
},
{
"entropy": 1.160587527230382,
"epoch": 0.432,
"grad_norm": 14.5,
"learning_rate": 1.7933333333333333e-05,
"loss": 19.2248,
"mean_token_accuracy": 0.7491097755730152,
"num_tokens": 8958371.0,
"step": 270
},
{
"entropy": 1.1003841817378999,
"epoch": 0.448,
"grad_norm": 14.0,
"learning_rate": 1.86e-05,
"loss": 18.3143,
"mean_token_accuracy": 0.7592806451022625,
"num_tokens": 9294388.0,
"step": 280
},
{
"entropy": 1.1066032487899065,
"epoch": 0.464,
"grad_norm": 15.375,
"learning_rate": 1.926666666666667e-05,
"loss": 18.3092,
"mean_token_accuracy": 0.7574895836412907,
"num_tokens": 9620366.0,
"step": 290
},
{
"entropy": 1.1134729530662297,
"epoch": 0.48,
"grad_norm": 14.875,
"learning_rate": 1.9933333333333334e-05,
"loss": 18.5931,
"mean_token_accuracy": 0.7540634006261826,
"num_tokens": 9955431.0,
"step": 300
},
{
"epoch": 0.48,
"eval_biology_entropy": 3.058082123756409,
"eval_biology_loss": 3.4944381713867188,
"eval_biology_mean_token_accuracy": 0.4677197203636169,
"eval_biology_num_tokens": 9955431.0,
"eval_biology_runtime": 56.129,
"eval_biology_samples_per_second": 8.908,
"eval_biology_steps_per_second": 2.227,
"step": 300
},
{
"epoch": 0.48,
"eval_chemistry_entropy": 1.584096737384796,
"eval_chemistry_loss": 1.6560814380645752,
"eval_chemistry_mean_token_accuracy": 0.6847201838493348,
"eval_chemistry_num_tokens": 9955431.0,
"eval_chemistry_runtime": 69.5947,
"eval_chemistry_samples_per_second": 7.184,
"eval_chemistry_steps_per_second": 1.796,
"step": 300
},
{
"epoch": 0.48,
"eval_math_entropy": 1.1037719421386718,
"eval_math_loss": 1.3507343530654907,
"eval_math_mean_token_accuracy": 0.7274009571075439,
"eval_math_num_tokens": 9955431.0,
"eval_math_runtime": 71.8212,
"eval_math_samples_per_second": 6.962,
"eval_math_steps_per_second": 1.74,
"step": 300
},
{
"epoch": 0.48,
"eval_physics_entropy": 1.1398421788215638,
"eval_physics_loss": 1.1733136177062988,
"eval_physics_mean_token_accuracy": 0.7520353999137879,
"eval_physics_num_tokens": 9955431.0,
"eval_physics_runtime": 83.0575,
"eval_physics_samples_per_second": 6.02,
"eval_physics_steps_per_second": 1.505,
"step": 300
},
{
"entropy": 1.1404459465295076,
"epoch": 0.496,
"grad_norm": 14.25,
"learning_rate": 1.9933333333333334e-05,
"loss": 18.8326,
"mean_token_accuracy": 0.7513993989676238,
"num_tokens": 10289883.0,
"step": 310
},
{
"entropy": 1.114009590819478,
"epoch": 0.512,
"grad_norm": 15.4375,
"learning_rate": 1.985925925925926e-05,
"loss": 18.526,
"mean_token_accuracy": 0.7558605384081603,
"num_tokens": 10619468.0,
"step": 320
},
{
"entropy": 1.126872155070305,
"epoch": 0.528,
"grad_norm": 13.5625,
"learning_rate": 1.9785185185185187e-05,
"loss": 18.6944,
"mean_token_accuracy": 0.75301427654922,
"num_tokens": 10944904.0,
"step": 330
},
{
"entropy": 1.11179842017591,
"epoch": 0.544,
"grad_norm": 14.625,
"learning_rate": 1.971111111111111e-05,
"loss": 18.3887,
"mean_token_accuracy": 0.7559159055352211,
"num_tokens": 11266409.0,
"step": 340
},
{
"entropy": 1.0923905447125435,
"epoch": 0.56,
"grad_norm": 13.5,
"learning_rate": 1.963703703703704e-05,
"loss": 18.0431,
"mean_token_accuracy": 0.7601441144943237,
"num_tokens": 11605544.0,
"step": 350
},
{
"entropy": 1.075535924360156,
"epoch": 0.576,
"grad_norm": 15.375,
"learning_rate": 1.9562962962962964e-05,
"loss": 17.9134,
"mean_token_accuracy": 0.762933773174882,
"num_tokens": 11931715.0,
"step": 360
},
{
"entropy": 1.0857708267867565,
"epoch": 0.592,
"grad_norm": 16.5,
"learning_rate": 1.948888888888889e-05,
"loss": 17.9806,
"mean_token_accuracy": 0.7590868677943945,
"num_tokens": 12257243.0,
"step": 370
},
{
"entropy": 1.077372272312641,
"epoch": 0.608,
"grad_norm": 14.25,
"learning_rate": 1.9414814814814817e-05,
"loss": 17.8473,
"mean_token_accuracy": 0.7623418286442757,
"num_tokens": 12583154.0,
"step": 380
},
{
"entropy": 1.101408040151,
"epoch": 0.624,
"grad_norm": 13.4375,
"learning_rate": 1.9340740740740743e-05,
"loss": 18.1987,
"mean_token_accuracy": 0.7578636299818754,
"num_tokens": 12905392.0,
"step": 390
},
{
"entropy": 1.0936459716409446,
"epoch": 0.64,
"grad_norm": 15.0,
"learning_rate": 1.926666666666667e-05,
"loss": 18.2164,
"mean_token_accuracy": 0.7581360317766667,
"num_tokens": 13232198.0,
"step": 400
},
{
"epoch": 0.64,
"eval_biology_entropy": 2.985183084487915,
"eval_biology_loss": 3.3934555053710938,
"eval_biology_mean_token_accuracy": 0.47690209436416625,
"eval_biology_num_tokens": 13232198.0,
"eval_biology_runtime": 56.2992,
"eval_biology_samples_per_second": 8.881,
"eval_biology_steps_per_second": 2.22,
"step": 400
},
{
"epoch": 0.64,
"eval_chemistry_entropy": 1.5230218110084535,
"eval_chemistry_loss": 1.5974879264831543,
"eval_chemistry_mean_token_accuracy": 0.6924252939224244,
"eval_chemistry_num_tokens": 13232198.0,
"eval_chemistry_runtime": 69.6914,
"eval_chemistry_samples_per_second": 7.174,
"eval_chemistry_steps_per_second": 1.794,
"step": 400
},
{
"epoch": 0.64,
"eval_math_entropy": 1.0690260066986084,
"eval_math_loss": 1.3214365243911743,
"eval_math_mean_token_accuracy": 0.7313346996307373,
"eval_math_num_tokens": 13232198.0,
"eval_math_runtime": 71.8154,
"eval_math_samples_per_second": 6.962,
"eval_math_steps_per_second": 1.741,
"step": 400
},
{
"epoch": 0.64,
"eval_physics_entropy": 1.079647916316986,
"eval_physics_loss": 1.122850775718689,
"eval_physics_mean_token_accuracy": 0.7587474570274353,
"eval_physics_num_tokens": 13232198.0,
"eval_physics_runtime": 83.3676,
"eval_physics_samples_per_second": 5.998,
"eval_physics_steps_per_second": 1.499,
"step": 400
},
{
"entropy": 1.0861666969954968,
"epoch": 0.656,
"grad_norm": 14.1875,
"learning_rate": 1.9192592592592593e-05,
"loss": 17.9599,
"mean_token_accuracy": 0.7603602163493633,
"num_tokens": 13575902.0,
"step": 410
},
{
"entropy": 1.0595566183328629,
"epoch": 0.672,
"grad_norm": 15.5625,
"learning_rate": 1.911851851851852e-05,
"loss": 17.6133,
"mean_token_accuracy": 0.7644100178033113,
"num_tokens": 13895997.0,
"step": 420
},
{
"entropy": 1.0779153060168027,
"epoch": 0.688,
"grad_norm": 13.625,
"learning_rate": 1.9044444444444446e-05,
"loss": 17.708,
"mean_token_accuracy": 0.7641124926507473,
"num_tokens": 14234888.0,
"step": 430
},
{
"entropy": 1.0761012937873602,
"epoch": 0.704,
"grad_norm": 14.4375,
"learning_rate": 1.8970370370370372e-05,
"loss": 17.9412,
"mean_token_accuracy": 0.7593970015645027,
"num_tokens": 14567908.0,
"step": 440
},
{
"entropy": 1.0635488711297512,
"epoch": 0.72,
"grad_norm": 15.5625,
"learning_rate": 1.8896296296296295e-05,
"loss": 17.5409,
"mean_token_accuracy": 0.7655271884053946,
"num_tokens": 14882927.0,
"step": 450
},
{
"entropy": 1.0563195772469043,
"epoch": 0.736,
"grad_norm": 15.8125,
"learning_rate": 1.8822222222222225e-05,
"loss": 17.5759,
"mean_token_accuracy": 0.7641396790742874,
"num_tokens": 15217342.0,
"step": 460
},
{
"entropy": 1.037678936868906,
"epoch": 0.752,
"grad_norm": 13.125,
"learning_rate": 1.874814814814815e-05,
"loss": 17.1107,
"mean_token_accuracy": 0.7690521724522114,
"num_tokens": 15547710.0,
"step": 470
},
{
"entropy": 1.0838343646377324,
"epoch": 0.768,
"grad_norm": 16.75,
"learning_rate": 1.8674074074074075e-05,
"loss": 18.0446,
"mean_token_accuracy": 0.759132880717516,
"num_tokens": 15877177.0,
"step": 480
},
{
"entropy": 1.0296639803797007,
"epoch": 0.784,
"grad_norm": 12.5,
"learning_rate": 1.86e-05,
"loss": 17.1015,
"mean_token_accuracy": 0.7697366833686828,
"num_tokens": 16219640.0,
"step": 490
},
{
"entropy": 1.0563473679125308,
"epoch": 0.8,
"grad_norm": 15.5,
"learning_rate": 1.8525925925925928e-05,
"loss": 17.5054,
"mean_token_accuracy": 0.7657136749476194,
"num_tokens": 16548261.0,
"step": 500
},
{
"epoch": 0.8,
"eval_biology_entropy": 2.9737762069702147,
"eval_biology_loss": 3.3795254230499268,
"eval_biology_mean_token_accuracy": 0.48070894145965576,
"eval_biology_num_tokens": 16548261.0,
"eval_biology_runtime": 56.18,
"eval_biology_samples_per_second": 8.9,
"eval_biology_steps_per_second": 2.225,
"step": 500
},
{
"epoch": 0.8,
"eval_chemistry_entropy": 1.5115246725082399,
"eval_chemistry_loss": 1.573546051979065,
"eval_chemistry_mean_token_accuracy": 0.6963776540756226,
"eval_chemistry_num_tokens": 16548261.0,
"eval_chemistry_runtime": 69.0701,
"eval_chemistry_samples_per_second": 7.239,
"eval_chemistry_steps_per_second": 1.81,
"step": 500
},
{
"epoch": 0.8,
"eval_math_entropy": 1.0560337038040162,
"eval_math_loss": 1.306509017944336,
"eval_math_mean_token_accuracy": 0.7335917987823486,
"eval_math_num_tokens": 16548261.0,
"eval_math_runtime": 71.2835,
"eval_math_samples_per_second": 7.014,
"eval_math_steps_per_second": 1.754,
"step": 500
},
{
"epoch": 0.8,
"eval_physics_entropy": 1.0595522990226747,
"eval_physics_loss": 1.0949721336364746,
"eval_physics_mean_token_accuracy": 0.7634064397811889,
"eval_physics_num_tokens": 16548261.0,
"eval_physics_runtime": 83.0981,
"eval_physics_samples_per_second": 6.017,
"eval_physics_steps_per_second": 1.504,
"step": 500
},
{
"entropy": 1.042676868662238,
"epoch": 0.816,
"grad_norm": 14.1875,
"learning_rate": 1.8451851851851855e-05,
"loss": 17.3757,
"mean_token_accuracy": 0.7653235342353583,
"num_tokens": 16870533.0,
"step": 510
},
{
"entropy": 1.060330757498741,
"epoch": 0.832,
"grad_norm": 17.875,
"learning_rate": 1.8377777777777778e-05,
"loss": 17.5577,
"mean_token_accuracy": 0.7640024948865175,
"num_tokens": 17196371.0,
"step": 520
},
{
"entropy": 1.045187360420823,
"epoch": 0.848,
"grad_norm": 12.75,
"learning_rate": 1.8303703703703704e-05,
"loss": 17.2392,
"mean_token_accuracy": 0.7683587469160557,
"num_tokens": 17527066.0,
"step": 530
},
{
"entropy": 1.0516619730740786,
"epoch": 0.864,
"grad_norm": 13.9375,
"learning_rate": 1.822962962962963e-05,
"loss": 17.4228,
"mean_token_accuracy": 0.7653108265250921,
"num_tokens": 17859605.0,
"step": 540
},
{
"entropy": 1.0745812084525823,
"epoch": 0.88,
"grad_norm": 16.0,
"learning_rate": 1.8155555555555557e-05,
"loss": 17.7266,
"mean_token_accuracy": 0.7631923343986273,
"num_tokens": 18187315.0,
"step": 550
},
{
"entropy": 0.9957344826310873,
"epoch": 0.896,
"grad_norm": 12.625,
"learning_rate": 1.8081481481481484e-05,
"loss": 16.4857,
"mean_token_accuracy": 0.7752151470631361,
"num_tokens": 18536790.0,
"step": 560
},
{
"entropy": 1.0575346298515798,
"epoch": 0.912,
"grad_norm": 14.9375,
"learning_rate": 1.800740740740741e-05,
"loss": 17.4739,
"mean_token_accuracy": 0.7637861587107182,
"num_tokens": 18869344.0,
"step": 570
},
{
"entropy": 1.0460248701274395,
"epoch": 0.928,
"grad_norm": 16.125,
"learning_rate": 1.7933333333333333e-05,
"loss": 17.3717,
"mean_token_accuracy": 0.7649744093418122,
"num_tokens": 19199834.0,
"step": 580
},
{
"entropy": 1.0241859197616576,
"epoch": 0.944,
"grad_norm": 15.5,
"learning_rate": 1.785925925925926e-05,
"loss": 16.9257,
"mean_token_accuracy": 0.7700011480599642,
"num_tokens": 19525385.0,
"step": 590
},
{
"entropy": 1.0272901140153408,
"epoch": 0.96,
"grad_norm": 17.125,
"learning_rate": 1.7785185185185186e-05,
"loss": 17.0788,
"mean_token_accuracy": 0.7674089819192886,
"num_tokens": 19856973.0,
"step": 600
},
{
"epoch": 0.96,
"eval_biology_entropy": 2.8927044954299928,
"eval_biology_loss": 3.3549082279205322,
"eval_biology_mean_token_accuracy": 0.48264122939109805,
"eval_biology_num_tokens": 19856973.0,
"eval_biology_runtime": 55.8568,
"eval_biology_samples_per_second": 8.951,
"eval_biology_steps_per_second": 2.238,
"step": 600
},
{
"epoch": 0.96,
"eval_chemistry_entropy": 1.464129506111145,
"eval_chemistry_loss": 1.5512925386428833,
"eval_chemistry_mean_token_accuracy": 0.6986791820526123,
"eval_chemistry_num_tokens": 19856973.0,
"eval_chemistry_runtime": 69.6627,
"eval_chemistry_samples_per_second": 7.177,
"eval_chemistry_steps_per_second": 1.794,
"step": 600
},
{
"epoch": 0.96,
"eval_math_entropy": 1.0420140266418456,
"eval_math_loss": 1.297760248184204,
"eval_math_mean_token_accuracy": 0.7350094408988953,
"eval_math_num_tokens": 19856973.0,
"eval_math_runtime": 71.5855,
"eval_math_samples_per_second": 6.985,
"eval_math_steps_per_second": 1.746,
"step": 600
},
{
"epoch": 0.96,
"eval_physics_entropy": 1.0300544242858887,
"eval_physics_loss": 1.076904058456421,
"eval_physics_mean_token_accuracy": 0.7661643705368042,
"eval_physics_num_tokens": 19856973.0,
"eval_physics_runtime": 83.7599,
"eval_physics_samples_per_second": 5.969,
"eval_physics_steps_per_second": 1.492,
"step": 600
},
{
"entropy": 1.029942188039422,
"epoch": 0.976,
"grad_norm": 14.5625,
"learning_rate": 1.7711111111111113e-05,
"loss": 17.0833,
"mean_token_accuracy": 0.7702321965247393,
"num_tokens": 20190600.0,
"step": 610
},
{
"entropy": 1.043160342797637,
"epoch": 0.992,
"grad_norm": 15.0,
"learning_rate": 1.763703703703704e-05,
"loss": 17.2354,
"mean_token_accuracy": 0.7673168186098337,
"num_tokens": 20518014.0,
"step": 620
},
{
"entropy": 1.0153347484767437,
"epoch": 1.008,
"grad_norm": 14.8125,
"learning_rate": 1.7562962962962962e-05,
"loss": 16.634,
"mean_token_accuracy": 0.7728456899523735,
"num_tokens": 20859215.0,
"step": 630
},
{
"entropy": 1.010590048506856,
"epoch": 1.024,
"grad_norm": 15.25,
"learning_rate": 1.7488888888888892e-05,
"loss": 16.8635,
"mean_token_accuracy": 0.770306408405304,
"num_tokens": 21186658.0,
"step": 640
},
{
"entropy": 1.0216664642095565,
"epoch": 1.04,
"grad_norm": 13.4375,
"learning_rate": 1.7414814814814815e-05,
"loss": 16.8228,
"mean_token_accuracy": 0.771704015135765,
"num_tokens": 21517237.0,
"step": 650
},
{
"entropy": 0.9916718799620867,
"epoch": 1.056,
"grad_norm": 13.9375,
"learning_rate": 1.7340740740740742e-05,
"loss": 16.4977,
"mean_token_accuracy": 0.7741733457893133,
"num_tokens": 21852528.0,
"step": 660
},
{
"entropy": 1.0193660650402308,
"epoch": 1.072,
"grad_norm": 12.6875,
"learning_rate": 1.726666666666667e-05,
"loss": 16.7942,
"mean_token_accuracy": 0.7709830049425364,
"num_tokens": 22192380.0,
"step": 670
},
{
"entropy": 1.017938245087862,
"epoch": 1.088,
"grad_norm": 17.5,
"learning_rate": 1.7192592592592595e-05,
"loss": 17.004,
"mean_token_accuracy": 0.7690784070640803,
"num_tokens": 22517971.0,
"step": 680
},
{
"entropy": 1.009276295825839,
"epoch": 1.104,
"grad_norm": 13.5625,
"learning_rate": 1.711851851851852e-05,
"loss": 16.5871,
"mean_token_accuracy": 0.773436525464058,
"num_tokens": 22848694.0,
"step": 690
},
{
"entropy": 1.0195886611938476,
"epoch": 1.12,
"grad_norm": 14.125,
"learning_rate": 1.7044444444444445e-05,
"loss": 17.0053,
"mean_token_accuracy": 0.7680860497057438,
"num_tokens": 23183889.0,
"step": 700
},
{
"epoch": 1.12,
"eval_biology_entropy": 2.837118970870972,
"eval_biology_loss": 3.3420016765594482,
"eval_biology_mean_token_accuracy": 0.48326786851882936,
"eval_biology_num_tokens": 23183889.0,
"eval_biology_runtime": 56.3022,
"eval_biology_samples_per_second": 8.881,
"eval_biology_steps_per_second": 2.22,
"step": 700
},
{
"epoch": 1.12,
"eval_chemistry_entropy": 1.433508418083191,
"eval_chemistry_loss": 1.5409785509109497,
"eval_chemistry_mean_token_accuracy": 0.6995256505012513,
"eval_chemistry_num_tokens": 23183889.0,
"eval_chemistry_runtime": 69.4682,
"eval_chemistry_samples_per_second": 7.198,
"eval_chemistry_steps_per_second": 1.799,
"step": 700
},
{
"epoch": 1.12,
"eval_math_entropy": 1.0230936131477355,
"eval_math_loss": 1.2898101806640625,
"eval_math_mean_token_accuracy": 0.736490508556366,
"eval_math_num_tokens": 23183889.0,
"eval_math_runtime": 71.817,
"eval_math_samples_per_second": 6.962,
"eval_math_steps_per_second": 1.741,
"step": 700
},
{
"epoch": 1.12,
"eval_physics_entropy": 1.0090678911209106,
"eval_physics_loss": 1.062973976135254,
"eval_physics_mean_token_accuracy": 0.7677445983886719,
"eval_physics_num_tokens": 23183889.0,
"eval_physics_runtime": 82.8728,
"eval_physics_samples_per_second": 6.033,
"eval_physics_steps_per_second": 1.508,
"step": 700
},
{
"entropy": 0.9851093973964453,
"epoch": 1.1360000000000001,
"grad_norm": 13.0625,
"learning_rate": 1.697037037037037e-05,
"loss": 16.2335,
"mean_token_accuracy": 0.7770712476223707,
"num_tokens": 23523903.0,
"step": 710
},
{
"entropy": 1.0010987296700478,
"epoch": 1.152,
"grad_norm": 16.375,
"learning_rate": 1.6896296296296298e-05,
"loss": 16.6749,
"mean_token_accuracy": 0.7745790995657444,
"num_tokens": 23848848.0,
"step": 720
},
{
"entropy": 1.0347776237875224,
"epoch": 1.168,
"grad_norm": 14.375,
"learning_rate": 1.6822222222222224e-05,
"loss": 16.9856,
"mean_token_accuracy": 0.7672174524515867,
"num_tokens": 24174589.0,
"step": 730
},
{
"entropy": 1.0073780696839094,
"epoch": 1.184,
"grad_norm": 16.875,
"learning_rate": 1.6748148148148147e-05,
"loss": 16.7297,
"mean_token_accuracy": 0.7723128654062748,
"num_tokens": 24495981.0,
"step": 740
},
{
"entropy": 0.9867725741118193,
"epoch": 1.2,
"grad_norm": 14.25,
"learning_rate": 1.6674074074074077e-05,
"loss": 16.3695,
"mean_token_accuracy": 0.7768875520676375,
"num_tokens": 24826245.0,
"step": 750
},
{
"entropy": 1.018533943966031,
"epoch": 1.216,
"grad_norm": 17.875,
"learning_rate": 1.66e-05,
"loss": 16.7892,
"mean_token_accuracy": 0.7702058039605617,
"num_tokens": 25147775.0,
"step": 760
},
{
"entropy": 0.9930577903985978,
"epoch": 1.232,
"grad_norm": 16.125,
"learning_rate": 1.6525925925925927e-05,
"loss": 16.5313,
"mean_token_accuracy": 0.7744447194039822,
"num_tokens": 25480033.0,
"step": 770
},
{
"entropy": 1.0064157444983721,
"epoch": 1.248,
"grad_norm": 15.8125,
"learning_rate": 1.6451851851851853e-05,
"loss": 16.6421,
"mean_token_accuracy": 0.773406020924449,
"num_tokens": 25811978.0,
"step": 780
},
{
"entropy": 0.9985887616872787,
"epoch": 1.264,
"grad_norm": 16.5,
"learning_rate": 1.637777777777778e-05,
"loss": 16.5153,
"mean_token_accuracy": 0.774219336360693,
"num_tokens": 26146299.0,
"step": 790
},
{
"entropy": 1.0295063827186823,
"epoch": 1.28,
"grad_norm": 16.125,
"learning_rate": 1.6303703703703706e-05,
"loss": 17.0668,
"mean_token_accuracy": 0.7661171369254589,
"num_tokens": 26478207.0,
"step": 800
},
{
"epoch": 1.28,
"eval_biology_entropy": 2.898805745124817,
"eval_biology_loss": 3.341296911239624,
"eval_biology_mean_token_accuracy": 0.48686809062957764,
"eval_biology_num_tokens": 26478207.0,
"eval_biology_runtime": 56.3495,
"eval_biology_samples_per_second": 8.873,
"eval_biology_steps_per_second": 2.218,
"step": 800
},
{
"epoch": 1.28,
"eval_chemistry_entropy": 1.4466705160140991,
"eval_chemistry_loss": 1.5262336730957031,
"eval_chemistry_mean_token_accuracy": 0.7025133318901062,
"eval_chemistry_num_tokens": 26478207.0,
"eval_chemistry_runtime": 69.3655,
"eval_chemistry_samples_per_second": 7.208,
"eval_chemistry_steps_per_second": 1.802,
"step": 800
},
{
"epoch": 1.28,
"eval_math_entropy": 1.024560465335846,
"eval_math_loss": 1.2816721200942993,
"eval_math_mean_token_accuracy": 0.737667652130127,
"eval_math_num_tokens": 26478207.0,
"eval_math_runtime": 71.7104,
"eval_math_samples_per_second": 6.972,
"eval_math_steps_per_second": 1.743,
"step": 800
},
{
"epoch": 1.28,
"eval_physics_entropy": 1.0090858311653137,
"eval_physics_loss": 1.0530790090560913,
"eval_physics_mean_token_accuracy": 0.7696638231277466,
"eval_physics_num_tokens": 26478207.0,
"eval_physics_runtime": 83.3567,
"eval_physics_samples_per_second": 5.998,
"eval_physics_steps_per_second": 1.5,
"step": 800
},
{
"entropy": 0.9791497606784105,
"epoch": 1.296,
"grad_norm": 15.9375,
"learning_rate": 1.622962962962963e-05,
"loss": 16.2738,
"mean_token_accuracy": 0.777652832493186,
"num_tokens": 26806477.0,
"step": 810
},
{
"entropy": 1.0318988490849734,
"epoch": 1.312,
"grad_norm": 17.125,
"learning_rate": 1.6155555555555556e-05,
"loss": 16.9474,
"mean_token_accuracy": 0.7686588883399963,
"num_tokens": 27137528.0,
"step": 820
},
{
"entropy": 1.007080302760005,
"epoch": 1.328,
"grad_norm": 17.875,
"learning_rate": 1.6081481481481482e-05,
"loss": 16.6844,
"mean_token_accuracy": 0.771523043885827,
"num_tokens": 27465859.0,
"step": 830
},
{
"entropy": 1.0098473474383354,
"epoch": 1.3439999999999999,
"grad_norm": 16.5,
"learning_rate": 1.600740740740741e-05,
"loss": 16.8087,
"mean_token_accuracy": 0.7707158491015434,
"num_tokens": 27795568.0,
"step": 840
},
{
"entropy": 1.0220563992857934,
"epoch": 1.3599999999999999,
"grad_norm": 15.3125,
"learning_rate": 1.5933333333333336e-05,
"loss": 16.768,
"mean_token_accuracy": 0.770363200828433,
"num_tokens": 28129988.0,
"step": 850
},
{
"entropy": 0.9859919909387826,
"epoch": 1.376,
"grad_norm": 17.875,
"learning_rate": 1.5859259259259262e-05,
"loss": 16.4135,
"mean_token_accuracy": 0.7744484882801771,
"num_tokens": 28465903.0,
"step": 860
},
{
"entropy": 0.988647548854351,
"epoch": 1.392,
"grad_norm": 17.125,
"learning_rate": 1.5785185185185185e-05,
"loss": 16.2043,
"mean_token_accuracy": 0.7775332607328892,
"num_tokens": 28801554.0,
"step": 870
},
{
"entropy": 0.9736344590783119,
"epoch": 1.408,
"grad_norm": 15.25,
"learning_rate": 1.571111111111111e-05,
"loss": 16.332,
"mean_token_accuracy": 0.7759838547557593,
"num_tokens": 29125114.0,
"step": 880
},
{
"entropy": 1.024445422179997,
"epoch": 1.424,
"grad_norm": 14.125,
"learning_rate": 1.5637037037037038e-05,
"loss": 16.9175,
"mean_token_accuracy": 0.7679786011576653,
"num_tokens": 29454334.0,
"step": 890
},
{
"entropy": 0.9635057615116238,
"epoch": 1.44,
"grad_norm": 15.4375,
"learning_rate": 1.5562962962962965e-05,
"loss": 15.9192,
"mean_token_accuracy": 0.7802809584885836,
"num_tokens": 29789253.0,
"step": 900
},
{
"epoch": 1.44,
"eval_biology_entropy": 2.8437174711227415,
"eval_biology_loss": 3.318575143814087,
"eval_biology_mean_token_accuracy": 0.4861623268127441,
"eval_biology_num_tokens": 29789253.0,
"eval_biology_runtime": 56.4644,
"eval_biology_samples_per_second": 8.855,
"eval_biology_steps_per_second": 2.214,
"step": 900
},
{
"epoch": 1.44,
"eval_chemistry_entropy": 1.420595220565796,
"eval_chemistry_loss": 1.5160387754440308,
"eval_chemistry_mean_token_accuracy": 0.7028642320632934,
"eval_chemistry_num_tokens": 29789253.0,
"eval_chemistry_runtime": 69.7177,
"eval_chemistry_samples_per_second": 7.172,
"eval_chemistry_steps_per_second": 1.793,
"step": 900
},
{
"epoch": 1.44,
"eval_math_entropy": 1.0116905975341797,
"eval_math_loss": 1.2779794931411743,
"eval_math_mean_token_accuracy": 0.7384266419410705,
"eval_math_num_tokens": 29789253.0,
"eval_math_runtime": 71.6683,
"eval_math_samples_per_second": 6.977,
"eval_math_steps_per_second": 1.744,
"step": 900
},
{
"epoch": 1.44,
"eval_physics_entropy": 0.9905482964515686,
"eval_physics_loss": 1.0435165166854858,
"eval_physics_mean_token_accuracy": 0.770782280921936,
"eval_physics_num_tokens": 29789253.0,
"eval_physics_runtime": 83.2228,
"eval_physics_samples_per_second": 6.008,
"eval_physics_steps_per_second": 1.502,
"step": 900
},
{
"entropy": 0.9581953696906567,
"epoch": 1.456,
"grad_norm": 13.1875,
"learning_rate": 1.548888888888889e-05,
"loss": 15.9515,
"mean_token_accuracy": 0.7805543716996908,
"num_tokens": 30128576.0,
"step": 910
},
{
"entropy": 1.0345965854823589,
"epoch": 1.472,
"grad_norm": 17.125,
"learning_rate": 1.5414814814814814e-05,
"loss": 17.1026,
"mean_token_accuracy": 0.7668049655854702,
"num_tokens": 30452719.0,
"step": 920
},
{
"entropy": 1.0023833878338337,
"epoch": 1.488,
"grad_norm": 13.8125,
"learning_rate": 1.5340740740740744e-05,
"loss": 16.5595,
"mean_token_accuracy": 0.7728984329849482,
"num_tokens": 30782343.0,
"step": 930
},
{
"entropy": 0.9856885317713022,
"epoch": 1.504,
"grad_norm": 14.1875,
"learning_rate": 1.5266666666666667e-05,
"loss": 16.267,
"mean_token_accuracy": 0.7769489102065563,
"num_tokens": 31113238.0,
"step": 940
},
{
"entropy": 1.0002378012984992,
"epoch": 1.52,
"grad_norm": 15.4375,
"learning_rate": 1.5192592592592594e-05,
"loss": 16.5956,
"mean_token_accuracy": 0.7721422206610441,
"num_tokens": 31444589.0,
"step": 950
},
{
"entropy": 1.0214108165353537,
"epoch": 1.536,
"grad_norm": 17.625,
"learning_rate": 1.5118518518518519e-05,
"loss": 16.8896,
"mean_token_accuracy": 0.7694170981645584,
"num_tokens": 31774137.0,
"step": 960
},
{
"entropy": 0.9722279075533151,
"epoch": 1.552,
"grad_norm": 16.375,
"learning_rate": 1.5044444444444445e-05,
"loss": 16.0224,
"mean_token_accuracy": 0.7783945810049773,
"num_tokens": 32101776.0,
"step": 970
},
{
"entropy": 0.9741225987672806,
"epoch": 1.568,
"grad_norm": 15.25,
"learning_rate": 1.497037037037037e-05,
"loss": 16.145,
"mean_token_accuracy": 0.778332832083106,
"num_tokens": 32428021.0,
"step": 980
},
{
"entropy": 1.0168811423704027,
"epoch": 1.584,
"grad_norm": 16.5,
"learning_rate": 1.4896296296296298e-05,
"loss": 16.8197,
"mean_token_accuracy": 0.7688806146383286,
"num_tokens": 32752735.0,
"step": 990
},
{
"entropy": 1.0012462731450795,
"epoch": 1.6,
"grad_norm": 14.625,
"learning_rate": 1.4822222222222225e-05,
"loss": 16.6785,
"mean_token_accuracy": 0.7718310851603747,
"num_tokens": 33075822.0,
"step": 1000
},
{
"epoch": 1.6,
"eval_biology_entropy": 2.871693316459656,
"eval_biology_loss": 3.2952377796173096,
"eval_biology_mean_token_accuracy": 0.48806494069099426,
"eval_biology_num_tokens": 33075822.0,
"eval_biology_runtime": 56.2343,
"eval_biology_samples_per_second": 8.891,
"eval_biology_steps_per_second": 2.223,
"step": 1000
},
{
"epoch": 1.6,
"eval_chemistry_entropy": 1.435364501476288,
"eval_chemistry_loss": 1.5101022720336914,
"eval_chemistry_mean_token_accuracy": 0.7037839899063111,
"eval_chemistry_num_tokens": 33075822.0,
"eval_chemistry_runtime": 69.2005,
"eval_chemistry_samples_per_second": 7.225,
"eval_chemistry_steps_per_second": 1.806,
"step": 1000
},
{
"epoch": 1.6,
"eval_math_entropy": 1.0202706332206726,
"eval_math_loss": 1.2738072872161865,
"eval_math_mean_token_accuracy": 0.7389369897842407,
"eval_math_num_tokens": 33075822.0,
"eval_math_runtime": 71.9142,
"eval_math_samples_per_second": 6.953,
"eval_math_steps_per_second": 1.738,
"step": 1000
},
{
"epoch": 1.6,
"eval_physics_entropy": 0.9962421131134033,
"eval_physics_loss": 1.0371240377426147,
"eval_physics_mean_token_accuracy": 0.7717792973518371,
"eval_physics_num_tokens": 33075822.0,
"eval_physics_runtime": 83.5174,
"eval_physics_samples_per_second": 5.987,
"eval_physics_steps_per_second": 1.497,
"step": 1000
}
],
"logging_steps": 10,
"max_steps": 3000,
"num_input_tokens_seen": 0,
"num_train_epochs": 5,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 3.3005113980506685e+18,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}