text2arch-deepseek / trainer_state.json
shivank21's picture
Upload folder using huggingface_hub
65b5d02 verified
{
"best_global_step": 3750,
"best_metric": 0.5079798102378845,
"best_model_checkpoint": "./results/checkpoint-3500",
"epoch": 4.997752808988764,
"eval_steps": 250,
"global_step": 9455,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.026437541308658295,
"grad_norm": 0.2475886046886444,
"learning_rate": 0.00034625958983852136,
"loss": 0.8762,
"mean_token_accuracy": 0.8128470236063003,
"num_tokens": 1638400.0,
"step": 50
},
{
"epoch": 0.05287508261731659,
"grad_norm": 0.25265493988990784,
"learning_rate": 0.000407611186724682,
"loss": 0.6461,
"mean_token_accuracy": 0.8552572166919709,
"num_tokens": 3276800.0,
"step": 100
},
{
"epoch": 0.07931262392597488,
"grad_norm": 0.21018819510936737,
"learning_rate": 0.0004434995702624468,
"loss": 0.635,
"mean_token_accuracy": 0.8568304070830345,
"num_tokens": 4915200.0,
"step": 150
},
{
"epoch": 0.10575016523463318,
"grad_norm": 0.23957596719264984,
"learning_rate": 0.0004689627836108426,
"loss": 0.6235,
"mean_token_accuracy": 0.8593035244941711,
"num_tokens": 6553600.0,
"step": 200
},
{
"epoch": 0.13218770654329148,
"grad_norm": 0.192116379737854,
"learning_rate": 0.0004887135863147016,
"loss": 0.6217,
"step": 250
},
{
"epoch": 0.13218770654329148,
"eval_loss": 0.630805492401123,
"eval_mean_token_accuracy": 0.8574222893971973,
"eval_num_tokens": 8192000.0,
"eval_runtime": 1444.1199,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 250
},
{
"epoch": 0.15862524785194976,
"grad_norm": 0.2664429545402527,
"learning_rate": 0.0004991822047759241,
"loss": 0.6239,
"mean_token_accuracy": 0.859162701368332,
"num_tokens": 9830400.0,
"step": 300
},
{
"epoch": 0.18506278916060806,
"grad_norm": 0.5154497027397156,
"learning_rate": 0.0004964562206956711,
"loss": 0.6973,
"mean_token_accuracy": 0.8490096586942673,
"num_tokens": 11468800.0,
"step": 350
},
{
"epoch": 0.21150033046926636,
"grad_norm": 0.2535574436187744,
"learning_rate": 0.0004937302366154182,
"loss": 0.6079,
"mean_token_accuracy": 0.8621737584471703,
"num_tokens": 13107200.0,
"step": 400
},
{
"epoch": 0.23793787177792466,
"grad_norm": 0.16894356906414032,
"learning_rate": 0.0004910042525351653,
"loss": 0.6082,
"mean_token_accuracy": 0.8616935542225838,
"num_tokens": 14745600.0,
"step": 450
},
{
"epoch": 0.26437541308658297,
"grad_norm": 0.1515253186225891,
"learning_rate": 0.00048827826845491225,
"loss": 0.6043,
"step": 500
},
{
"epoch": 0.26437541308658297,
"eval_loss": 0.6044251918792725,
"eval_mean_token_accuracy": 0.8618915110637472,
"eval_num_tokens": 16384000.0,
"eval_runtime": 1444.8043,
"eval_samples_per_second": 5.236,
"eval_steps_per_second": 0.655,
"step": 500
},
{
"epoch": 0.29081295439524124,
"grad_norm": 0.15358857810497284,
"learning_rate": 0.0004855522843746593,
"loss": 0.6137,
"mean_token_accuracy": 0.8617021070420742,
"num_tokens": 18022400.0,
"step": 550
},
{
"epoch": 0.3172504957038995,
"grad_norm": 0.34029558300971985,
"learning_rate": 0.00048282630029440626,
"loss": 0.5973,
"mean_token_accuracy": 0.8634732463955879,
"num_tokens": 19660800.0,
"step": 600
},
{
"epoch": 0.34368803701255785,
"grad_norm": 0.22880446910858154,
"learning_rate": 0.00048010031621415335,
"loss": 0.5984,
"mean_token_accuracy": 0.8633871012926102,
"num_tokens": 21299200.0,
"step": 650
},
{
"epoch": 0.3701255783212161,
"grad_norm": 0.17699383199214935,
"learning_rate": 0.0004773743321339004,
"loss": 0.5871,
"mean_token_accuracy": 0.8653806248307228,
"num_tokens": 22937600.0,
"step": 700
},
{
"epoch": 0.3965631196298744,
"grad_norm": 0.18763592839241028,
"learning_rate": 0.00047464834805364736,
"loss": 0.5889,
"step": 750
},
{
"epoch": 0.3965631196298744,
"eval_loss": 0.5848008394241333,
"eval_mean_token_accuracy": 0.865356788007704,
"eval_num_tokens": 24576000.0,
"eval_runtime": 1443.8073,
"eval_samples_per_second": 5.24,
"eval_steps_per_second": 0.655,
"step": 750
},
{
"epoch": 0.4230006609385327,
"grad_norm": 0.1546117663383484,
"learning_rate": 0.0004719223639733944,
"loss": 0.5821,
"mean_token_accuracy": 0.8651845099031925,
"num_tokens": 26214400.0,
"step": 800
},
{
"epoch": 0.449438202247191,
"grad_norm": 0.13760164380073547,
"learning_rate": 0.0004691963798931415,
"loss": 0.5824,
"mean_token_accuracy": 0.8658082884550095,
"num_tokens": 27852800.0,
"step": 850
},
{
"epoch": 0.47587574355584933,
"grad_norm": 0.1396537572145462,
"learning_rate": 0.00046647039581288846,
"loss": 0.5707,
"mean_token_accuracy": 0.8684304764866829,
"num_tokens": 29491200.0,
"step": 900
},
{
"epoch": 0.5023132848645075,
"grad_norm": 0.13558807969093323,
"learning_rate": 0.0004637444117326355,
"loss": 0.58,
"mean_token_accuracy": 0.8662952136993408,
"num_tokens": 31129600.0,
"step": 950
},
{
"epoch": 0.5287508261731659,
"grad_norm": 0.12448519468307495,
"learning_rate": 0.0004610184276523825,
"loss": 0.5769,
"step": 1000
},
{
"epoch": 0.5287508261731659,
"eval_loss": 0.5692248940467834,
"eval_mean_token_accuracy": 0.8682624975546218,
"eval_num_tokens": 32768000.0,
"eval_runtime": 1443.7648,
"eval_samples_per_second": 5.24,
"eval_steps_per_second": 0.655,
"step": 1000
},
{
"epoch": 0.5551883674818242,
"grad_norm": 0.12668457627296448,
"learning_rate": 0.00045829244357212956,
"loss": 0.5683,
"mean_token_accuracy": 0.8675177201628685,
"num_tokens": 34406400.0,
"step": 1050
},
{
"epoch": 0.5816259087904825,
"grad_norm": 0.1283566802740097,
"learning_rate": 0.0004555664594918766,
"loss": 0.5686,
"mean_token_accuracy": 0.8686253690719604,
"num_tokens": 36044800.0,
"step": 1100
},
{
"epoch": 0.6080634500991408,
"grad_norm": 0.15960244834423065,
"learning_rate": 0.0004528404754116236,
"loss": 0.5644,
"mean_token_accuracy": 0.8688807463645936,
"num_tokens": 37683200.0,
"step": 1150
},
{
"epoch": 0.634500991407799,
"grad_norm": 0.11330343037843704,
"learning_rate": 0.0004501144913313706,
"loss": 0.5649,
"mean_token_accuracy": 0.8686217039823532,
"num_tokens": 39321600.0,
"step": 1200
},
{
"epoch": 0.6609385327164574,
"grad_norm": 0.12131626158952713,
"learning_rate": 0.0004473885072511177,
"loss": 0.5638,
"step": 1250
},
{
"epoch": 0.6609385327164574,
"eval_loss": 0.5589728951454163,
"eval_mean_token_accuracy": 0.8698779833493223,
"eval_num_tokens": 40960000.0,
"eval_runtime": 1443.9339,
"eval_samples_per_second": 5.239,
"eval_steps_per_second": 0.655,
"step": 1250
},
{
"epoch": 0.6873760740251157,
"grad_norm": 0.135975182056427,
"learning_rate": 0.0004446625231708647,
"loss": 0.5606,
"mean_token_accuracy": 0.8690710559487342,
"num_tokens": 42598400.0,
"step": 1300
},
{
"epoch": 0.713813615333774,
"grad_norm": 0.15456999838352203,
"learning_rate": 0.0004419365390906117,
"loss": 0.5495,
"mean_token_accuracy": 0.8715310388803482,
"num_tokens": 44236800.0,
"step": 1350
},
{
"epoch": 0.7402511566424322,
"grad_norm": 0.17465010285377502,
"learning_rate": 0.00043921055501035873,
"loss": 0.5424,
"mean_token_accuracy": 0.873205646276474,
"num_tokens": 45875200.0,
"step": 1400
},
{
"epoch": 0.7666886979510905,
"grad_norm": 0.11691431701183319,
"learning_rate": 0.0004364845709301058,
"loss": 0.5407,
"mean_token_accuracy": 0.8731714335083961,
"num_tokens": 47513600.0,
"step": 1450
},
{
"epoch": 0.7931262392597488,
"grad_norm": 0.1413394808769226,
"learning_rate": 0.0004337585868498528,
"loss": 0.5477,
"step": 1500
},
{
"epoch": 0.7931262392597488,
"eval_loss": 0.5501642227172852,
"eval_mean_token_accuracy": 0.8714489584619348,
"eval_num_tokens": 49152000.0,
"eval_runtime": 1443.5916,
"eval_samples_per_second": 5.24,
"eval_steps_per_second": 0.655,
"step": 1500
},
{
"epoch": 0.8195637805684072,
"grad_norm": 0.10578064620494843,
"learning_rate": 0.00043103260276959983,
"loss": 0.5508,
"mean_token_accuracy": 0.8713572224974633,
"num_tokens": 50790400.0,
"step": 1550
},
{
"epoch": 0.8460013218770654,
"grad_norm": 0.10736281424760818,
"learning_rate": 0.0004283066186893468,
"loss": 0.5481,
"mean_token_accuracy": 0.8717375376820564,
"num_tokens": 52428800.0,
"step": 1600
},
{
"epoch": 0.8724388631857237,
"grad_norm": 0.12125309556722641,
"learning_rate": 0.0004255806346090939,
"loss": 0.5537,
"mean_token_accuracy": 0.86990652769804,
"num_tokens": 54067200.0,
"step": 1650
},
{
"epoch": 0.898876404494382,
"grad_norm": 0.10367817431688309,
"learning_rate": 0.00042285465052884093,
"loss": 0.5446,
"mean_token_accuracy": 0.8718084079027176,
"num_tokens": 55705600.0,
"step": 1700
},
{
"epoch": 0.9253139458030403,
"grad_norm": 0.19373266398906708,
"learning_rate": 0.0004201286664485879,
"loss": 0.5486,
"step": 1750
},
{
"epoch": 0.9253139458030403,
"eval_loss": 0.5393346548080444,
"eval_mean_token_accuracy": 0.8730530525236755,
"eval_num_tokens": 57344000.0,
"eval_runtime": 1444.4931,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 1750
},
{
"epoch": 0.9517514871116987,
"grad_norm": 0.10371921956539154,
"learning_rate": 0.00041740268236833495,
"loss": 0.5381,
"mean_token_accuracy": 0.8720436228811741,
"num_tokens": 58982400.0,
"step": 1800
},
{
"epoch": 0.9781890284203569,
"grad_norm": 0.09864738583564758,
"learning_rate": 0.00041467669828808203,
"loss": 0.5321,
"mean_token_accuracy": 0.8747250729799271,
"num_tokens": 60620800.0,
"step": 1850
},
{
"epoch": 1.0042300066093852,
"grad_norm": 0.12164577841758728,
"learning_rate": 0.00041200523388943414,
"loss": 0.519,
"mean_token_accuracy": 0.8759235533966025,
"num_tokens": 62234624.0,
"step": 1900
},
{
"epoch": 1.0306675479180436,
"grad_norm": 0.16176024079322815,
"learning_rate": 0.0004092792498091811,
"loss": 0.4182,
"mean_token_accuracy": 0.8925769734382629,
"num_tokens": 63873024.0,
"step": 1950
},
{
"epoch": 1.057105089226702,
"grad_norm": 0.09831411391496658,
"learning_rate": 0.00040655326572892816,
"loss": 0.426,
"step": 2000
},
{
"epoch": 1.057105089226702,
"eval_loss": 0.5495159029960632,
"eval_mean_token_accuracy": 0.8728643428321621,
"eval_num_tokens": 65511424.0,
"eval_runtime": 1443.7886,
"eval_samples_per_second": 5.24,
"eval_steps_per_second": 0.655,
"step": 2000
},
{
"epoch": 1.0835426305353602,
"grad_norm": 0.11064837127923965,
"learning_rate": 0.00040382728164867513,
"loss": 0.4203,
"mean_token_accuracy": 0.891697817593813,
"num_tokens": 67149824.0,
"step": 2050
},
{
"epoch": 1.1099801718440185,
"grad_norm": 0.12444479018449783,
"learning_rate": 0.0004011012975684222,
"loss": 0.4324,
"mean_token_accuracy": 0.8894703006744384,
"num_tokens": 68788224.0,
"step": 2100
},
{
"epoch": 1.1364177131526767,
"grad_norm": 0.10909954458475113,
"learning_rate": 0.00039837531348816925,
"loss": 0.4243,
"mean_token_accuracy": 0.8913825675845146,
"num_tokens": 70426624.0,
"step": 2150
},
{
"epoch": 1.162855254461335,
"grad_norm": 0.09556654095649719,
"learning_rate": 0.00039564932940791623,
"loss": 0.426,
"mean_token_accuracy": 0.8912683191895485,
"num_tokens": 72065024.0,
"step": 2200
},
{
"epoch": 1.1892927957699935,
"grad_norm": 0.11516989767551422,
"learning_rate": 0.00039292334532766327,
"loss": 0.4313,
"step": 2250
},
{
"epoch": 1.1892927957699935,
"eval_loss": 0.5482946634292603,
"eval_mean_token_accuracy": 0.873437058408971,
"eval_num_tokens": 73703424.0,
"eval_runtime": 1445.2243,
"eval_samples_per_second": 5.234,
"eval_steps_per_second": 0.655,
"step": 2250
},
{
"epoch": 1.2157303370786516,
"grad_norm": 0.11027960479259491,
"learning_rate": 0.00039019736124741035,
"loss": 0.4393,
"mean_token_accuracy": 0.889146803766489,
"num_tokens": 75341824.0,
"step": 2300
},
{
"epoch": 1.24216787838731,
"grad_norm": 0.118904247879982,
"learning_rate": 0.00038747137716715733,
"loss": 0.4408,
"mean_token_accuracy": 0.8877504804730415,
"num_tokens": 76980224.0,
"step": 2350
},
{
"epoch": 1.2686054196959682,
"grad_norm": 0.10859134048223495,
"learning_rate": 0.00038474539308690437,
"loss": 0.4349,
"mean_token_accuracy": 0.8895466673374176,
"num_tokens": 78618624.0,
"step": 2400
},
{
"epoch": 1.2950429610046266,
"grad_norm": 0.09974760562181473,
"learning_rate": 0.0003820194090066514,
"loss": 0.4375,
"mean_token_accuracy": 0.8887909257411957,
"num_tokens": 80257024.0,
"step": 2450
},
{
"epoch": 1.321480502313285,
"grad_norm": 0.10627176612615585,
"learning_rate": 0.00037929342492639843,
"loss": 0.4354,
"step": 2500
},
{
"epoch": 1.321480502313285,
"eval_loss": 0.5409240126609802,
"eval_mean_token_accuracy": 0.8742006784639701,
"eval_num_tokens": 81895424.0,
"eval_runtime": 1443.9561,
"eval_samples_per_second": 5.239,
"eval_steps_per_second": 0.655,
"step": 2500
},
{
"epoch": 1.3479180436219431,
"grad_norm": 0.10294341295957565,
"learning_rate": 0.00037656744084614547,
"loss": 0.4372,
"mean_token_accuracy": 0.8892436383664608,
"num_tokens": 83533824.0,
"step": 2550
},
{
"epoch": 1.3743555849306015,
"grad_norm": 0.12393230944871902,
"learning_rate": 0.0003738414567658925,
"loss": 0.4386,
"mean_token_accuracy": 0.8886467415094376,
"num_tokens": 85172224.0,
"step": 2600
},
{
"epoch": 1.4007931262392597,
"grad_norm": 0.09495224803686142,
"learning_rate": 0.0003711154726856395,
"loss": 0.4302,
"mean_token_accuracy": 0.8906915840506554,
"num_tokens": 86810624.0,
"step": 2650
},
{
"epoch": 1.427230667547918,
"grad_norm": 0.10494447499513626,
"learning_rate": 0.00036838948860538656,
"loss": 0.4312,
"mean_token_accuracy": 0.8900836914777756,
"num_tokens": 88449024.0,
"step": 2700
},
{
"epoch": 1.4536682088565764,
"grad_norm": 0.12757033109664917,
"learning_rate": 0.0003656635045251336,
"loss": 0.437,
"step": 2750
},
{
"epoch": 1.4536682088565764,
"eval_loss": 0.53493332862854,
"eval_mean_token_accuracy": 0.8759168862162382,
"eval_num_tokens": 90087424.0,
"eval_runtime": 1443.9928,
"eval_samples_per_second": 5.239,
"eval_steps_per_second": 0.655,
"step": 2750
},
{
"epoch": 1.4801057501652346,
"grad_norm": 0.12719060480594635,
"learning_rate": 0.0003629375204448806,
"loss": 0.4384,
"mean_token_accuracy": 0.8888401076197624,
"num_tokens": 91725824.0,
"step": 2800
},
{
"epoch": 1.5065432914738928,
"grad_norm": 0.1001153439283371,
"learning_rate": 0.0003602115363646276,
"loss": 0.443,
"mean_token_accuracy": 0.8874511170387268,
"num_tokens": 93364224.0,
"step": 2850
},
{
"epoch": 1.5329808327825512,
"grad_norm": 0.10144038498401642,
"learning_rate": 0.0003574855522843747,
"loss": 0.4439,
"mean_token_accuracy": 0.8875879681110382,
"num_tokens": 95002624.0,
"step": 2900
},
{
"epoch": 1.5594183740912095,
"grad_norm": 0.10192844271659851,
"learning_rate": 0.0003547595682041217,
"loss": 0.4345,
"mean_token_accuracy": 0.8896059301495552,
"num_tokens": 96641024.0,
"step": 2950
},
{
"epoch": 1.585855915399868,
"grad_norm": 0.11319652944803238,
"learning_rate": 0.0003520335841238687,
"loss": 0.4378,
"step": 3000
},
{
"epoch": 1.585855915399868,
"eval_loss": 0.5288907289505005,
"eval_mean_token_accuracy": 0.8770441555951665,
"eval_num_tokens": 98279424.0,
"eval_runtime": 1443.9606,
"eval_samples_per_second": 5.239,
"eval_steps_per_second": 0.655,
"step": 3000
},
{
"epoch": 1.612293456708526,
"grad_norm": 0.10589715093374252,
"learning_rate": 0.00034930760004361574,
"loss": 0.4334,
"mean_token_accuracy": 0.8895524734258652,
"num_tokens": 99917824.0,
"step": 3050
},
{
"epoch": 1.6387309980171842,
"grad_norm": 0.09860669076442719,
"learning_rate": 0.0003465816159633628,
"loss": 0.4377,
"mean_token_accuracy": 0.8887499925494194,
"num_tokens": 101556224.0,
"step": 3100
},
{
"epoch": 1.6651685393258426,
"grad_norm": 0.11345923691987991,
"learning_rate": 0.0003438556318831098,
"loss": 0.4413,
"mean_token_accuracy": 0.8883473780751229,
"num_tokens": 103194624.0,
"step": 3150
},
{
"epoch": 1.691606080634501,
"grad_norm": 0.10729037970304489,
"learning_rate": 0.00034112964780285684,
"loss": 0.4365,
"mean_token_accuracy": 0.8893462771177292,
"num_tokens": 104833024.0,
"step": 3200
},
{
"epoch": 1.7180436219431594,
"grad_norm": 0.0929078608751297,
"learning_rate": 0.0003384036637226039,
"loss": 0.4339,
"step": 3250
},
{
"epoch": 1.7180436219431594,
"eval_loss": 0.522659182548523,
"eval_mean_token_accuracy": 0.8778351331740051,
"eval_num_tokens": 106471424.0,
"eval_runtime": 1444.1286,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 3250
},
{
"epoch": 1.7444811632518176,
"grad_norm": 0.09583564847707748,
"learning_rate": 0.0003356776796423509,
"loss": 0.4292,
"mean_token_accuracy": 0.8906918911635876,
"num_tokens": 108109824.0,
"step": 3300
},
{
"epoch": 1.7709187045604757,
"grad_norm": 0.09396171569824219,
"learning_rate": 0.00033295169556209794,
"loss": 0.4259,
"mean_token_accuracy": 0.8916892632842064,
"num_tokens": 109748224.0,
"step": 3350
},
{
"epoch": 1.7973562458691341,
"grad_norm": 0.08548162132501602,
"learning_rate": 0.0003302257114818449,
"loss": 0.4327,
"mean_token_accuracy": 0.8902602550387383,
"num_tokens": 111386624.0,
"step": 3400
},
{
"epoch": 1.8237937871777925,
"grad_norm": 0.1594037264585495,
"learning_rate": 0.000327499727401592,
"loss": 0.4279,
"mean_token_accuracy": 0.8916428324580192,
"num_tokens": 113025024.0,
"step": 3450
},
{
"epoch": 1.8502313284864509,
"grad_norm": 0.09137943387031555,
"learning_rate": 0.00032477374332133904,
"loss": 0.4259,
"step": 3500
},
{
"epoch": 1.8502313284864509,
"eval_loss": 0.5174182653427124,
"eval_mean_token_accuracy": 0.878870385834833,
"eval_num_tokens": 114663424.0,
"eval_runtime": 1444.5837,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 3500
},
{
"epoch": 1.876668869795109,
"grad_norm": 0.13611243665218353,
"learning_rate": 0.000322047759241086,
"loss": 0.4334,
"mean_token_accuracy": 0.8908357685804367,
"num_tokens": 116301824.0,
"step": 3550
},
{
"epoch": 1.9031064111037672,
"grad_norm": 0.10409994423389435,
"learning_rate": 0.00031932177516083305,
"loss": 0.4396,
"mean_token_accuracy": 0.8884219127893448,
"num_tokens": 117940224.0,
"step": 3600
},
{
"epoch": 1.9295439524124256,
"grad_norm": 0.08799432218074799,
"learning_rate": 0.00031659579108058014,
"loss": 0.4238,
"mean_token_accuracy": 0.892235452234745,
"num_tokens": 119578624.0,
"step": 3650
},
{
"epoch": 1.955981493721084,
"grad_norm": 0.0893888846039772,
"learning_rate": 0.0003138698070003271,
"loss": 0.4292,
"mean_token_accuracy": 0.8910862573981285,
"num_tokens": 121217024.0,
"step": 3700
},
{
"epoch": 1.9824190350297424,
"grad_norm": 0.08708135783672333,
"learning_rate": 0.00031114382292007415,
"loss": 0.4319,
"step": 3750
},
{
"epoch": 1.9824190350297424,
"eval_loss": 0.5079798102378845,
"eval_mean_token_accuracy": 0.880161566348711,
"eval_num_tokens": 122855424.0,
"eval_runtime": 1444.3228,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 3750
},
{
"epoch": 2.0084600132187704,
"grad_norm": 0.10443489998579025,
"learning_rate": 0.00030847235852142626,
"loss": 0.3692,
"mean_token_accuracy": 0.89680361027081,
"num_tokens": 124469248.0,
"step": 3800
},
{
"epoch": 2.034897554527429,
"grad_norm": 0.09454915672540665,
"learning_rate": 0.00030574637444117324,
"loss": 0.2389,
"mean_token_accuracy": 0.9304979285597801,
"num_tokens": 126107648.0,
"step": 3850
},
{
"epoch": 2.061335095836087,
"grad_norm": 0.10566703230142593,
"learning_rate": 0.0003030203903609203,
"loss": 0.2421,
"mean_token_accuracy": 0.9295185798406601,
"num_tokens": 127746048.0,
"step": 3900
},
{
"epoch": 2.0877726371447456,
"grad_norm": 0.10303833335638046,
"learning_rate": 0.00030029440628066736,
"loss": 0.2398,
"mean_token_accuracy": 0.9298442155122757,
"num_tokens": 129384448.0,
"step": 3950
},
{
"epoch": 2.114210178453404,
"grad_norm": 0.0934186801314354,
"learning_rate": 0.00029756842220041434,
"loss": 0.2444,
"step": 4000
},
{
"epoch": 2.114210178453404,
"eval_loss": 0.5857627391815186,
"eval_mean_token_accuracy": 0.8763663001342757,
"eval_num_tokens": 131022848.0,
"eval_runtime": 1444.4937,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 4000
},
{
"epoch": 2.140647719762062,
"grad_norm": 0.11742925643920898,
"learning_rate": 0.0002948424381201614,
"loss": 0.2433,
"mean_token_accuracy": 0.9287463417649269,
"num_tokens": 132661248.0,
"step": 4050
},
{
"epoch": 2.1670852610707203,
"grad_norm": 0.09310949593782425,
"learning_rate": 0.0002921164540399084,
"loss": 0.2502,
"mean_token_accuracy": 0.9270796775817871,
"num_tokens": 134299648.0,
"step": 4100
},
{
"epoch": 2.1935228023793787,
"grad_norm": 0.10311378538608551,
"learning_rate": 0.00028939046995965544,
"loss": 0.2487,
"mean_token_accuracy": 0.9274065324664116,
"num_tokens": 135938048.0,
"step": 4150
},
{
"epoch": 2.219960343688037,
"grad_norm": 0.10651703178882599,
"learning_rate": 0.00028666448587940247,
"loss": 0.2481,
"mean_token_accuracy": 0.9273692658543586,
"num_tokens": 137576448.0,
"step": 4200
},
{
"epoch": 2.2463978849966955,
"grad_norm": 0.11140276491641998,
"learning_rate": 0.0002839385017991495,
"loss": 0.2513,
"step": 4250
},
{
"epoch": 2.2463978849966955,
"eval_loss": 0.5914037823677063,
"eval_mean_token_accuracy": 0.8769333329815693,
"eval_num_tokens": 139214848.0,
"eval_runtime": 1445.0132,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 4250
},
{
"epoch": 2.2728354263053534,
"grad_norm": 0.10259566456079483,
"learning_rate": 0.00028121251771889654,
"loss": 0.253,
"mean_token_accuracy": 0.9264763659238815,
"num_tokens": 140853248.0,
"step": 4300
},
{
"epoch": 2.299272967614012,
"grad_norm": 0.10440666228532791,
"learning_rate": 0.00027848653363864357,
"loss": 0.25,
"mean_token_accuracy": 0.9270870080590248,
"num_tokens": 142491648.0,
"step": 4350
},
{
"epoch": 2.32571050892267,
"grad_norm": 0.11085245013237,
"learning_rate": 0.0002757605495583906,
"loss": 0.2507,
"mean_token_accuracy": 0.9270876184105873,
"num_tokens": 144130048.0,
"step": 4400
},
{
"epoch": 2.3521480502313286,
"grad_norm": 0.11133650690317154,
"learning_rate": 0.0002730345654781376,
"loss": 0.2509,
"mean_token_accuracy": 0.9270026966929436,
"num_tokens": 145768448.0,
"step": 4450
},
{
"epoch": 2.378585591539987,
"grad_norm": 0.09678333252668381,
"learning_rate": 0.00027030858139788467,
"loss": 0.252,
"step": 4500
},
{
"epoch": 2.378585591539987,
"eval_loss": 0.588071882724762,
"eval_mean_token_accuracy": 0.8775507902242164,
"eval_num_tokens": 147406848.0,
"eval_runtime": 1444.9835,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 4500
},
{
"epoch": 2.405023132848645,
"grad_norm": 0.09919407218694687,
"learning_rate": 0.0002675825973176317,
"loss": 0.2597,
"mean_token_accuracy": 0.9257050442695618,
"num_tokens": 149045248.0,
"step": 4550
},
{
"epoch": 2.4314606741573033,
"grad_norm": 0.09900713711977005,
"learning_rate": 0.0002648566132373787,
"loss": 0.2594,
"mean_token_accuracy": 0.9249798482656479,
"num_tokens": 150683648.0,
"step": 4600
},
{
"epoch": 2.4578982154659617,
"grad_norm": 0.11208795011043549,
"learning_rate": 0.0002621306291571257,
"loss": 0.2512,
"mean_token_accuracy": 0.926915942132473,
"num_tokens": 152322048.0,
"step": 4650
},
{
"epoch": 2.48433575677462,
"grad_norm": 0.1047317236661911,
"learning_rate": 0.0002594046450768728,
"loss": 0.2575,
"mean_token_accuracy": 0.9254233974218369,
"num_tokens": 153960448.0,
"step": 4700
},
{
"epoch": 2.5107732980832784,
"grad_norm": 0.10790830850601196,
"learning_rate": 0.0002566786609966198,
"loss": 0.2597,
"step": 4750
},
{
"epoch": 2.5107732980832784,
"eval_loss": 0.5828577876091003,
"eval_mean_token_accuracy": 0.8786447106153717,
"eval_num_tokens": 155598848.0,
"eval_runtime": 1445.085,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 4750
},
{
"epoch": 2.5372108393919364,
"grad_norm": 0.1063418760895729,
"learning_rate": 0.0002539526769163668,
"loss": 0.2532,
"mean_token_accuracy": 0.9256470040977001,
"num_tokens": 157237248.0,
"step": 4800
},
{
"epoch": 2.5636483807005948,
"grad_norm": 0.10517441481351852,
"learning_rate": 0.00025122669283611385,
"loss": 0.257,
"mean_token_accuracy": 0.92532870054245,
"num_tokens": 158875648.0,
"step": 4850
},
{
"epoch": 2.590085922009253,
"grad_norm": 0.10175232589244843,
"learning_rate": 0.0002485007087558609,
"loss": 0.2578,
"mean_token_accuracy": 0.9254624971747398,
"num_tokens": 160514048.0,
"step": 4900
},
{
"epoch": 2.6165234633179115,
"grad_norm": 0.10122060030698776,
"learning_rate": 0.0002457747246756079,
"loss": 0.2585,
"mean_token_accuracy": 0.925160077214241,
"num_tokens": 162152448.0,
"step": 4950
},
{
"epoch": 2.64296100462657,
"grad_norm": 0.17676186561584473,
"learning_rate": 0.00024304874059535492,
"loss": 0.2549,
"step": 5000
},
{
"epoch": 2.64296100462657,
"eval_loss": 0.5834685564041138,
"eval_mean_token_accuracy": 0.8788562039618261,
"eval_num_tokens": 163790848.0,
"eval_runtime": 1444.3995,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 5000
},
{
"epoch": 2.669398545935228,
"grad_norm": 0.09294166415929794,
"learning_rate": 0.00024032275651510195,
"loss": 0.2578,
"mean_token_accuracy": 0.9256011824309825,
"num_tokens": 165429248.0,
"step": 5050
},
{
"epoch": 2.6958360872438862,
"grad_norm": 0.10612129420042038,
"learning_rate": 0.000237596772434849,
"loss": 0.2525,
"mean_token_accuracy": 0.9266947796940803,
"num_tokens": 167067648.0,
"step": 5100
},
{
"epoch": 2.7222736285525446,
"grad_norm": 0.09773480892181396,
"learning_rate": 0.00023487078835459602,
"loss": 0.2574,
"mean_token_accuracy": 0.9255730789899826,
"num_tokens": 168706048.0,
"step": 5150
},
{
"epoch": 2.748711169861203,
"grad_norm": 0.10722821205854416,
"learning_rate": 0.00023214480427434303,
"loss": 0.255,
"mean_token_accuracy": 0.9261156010627747,
"num_tokens": 170344448.0,
"step": 5200
},
{
"epoch": 2.7751487111698614,
"grad_norm": 0.10466789454221725,
"learning_rate": 0.00022941882019409009,
"loss": 0.2579,
"step": 5250
},
{
"epoch": 2.7751487111698614,
"eval_loss": 0.5760974287986755,
"eval_mean_token_accuracy": 0.8801544366628876,
"eval_num_tokens": 171982848.0,
"eval_runtime": 1445.5403,
"eval_samples_per_second": 5.233,
"eval_steps_per_second": 0.654,
"step": 5250
},
{
"epoch": 2.8015862524785193,
"grad_norm": 0.1087755486369133,
"learning_rate": 0.0002266928361138371,
"loss": 0.2609,
"mean_token_accuracy": 0.9251267817616463,
"num_tokens": 173621248.0,
"step": 5300
},
{
"epoch": 2.8280237937871777,
"grad_norm": 0.09775590896606445,
"learning_rate": 0.00022396685203358413,
"loss": 0.2557,
"mean_token_accuracy": 0.9260593932867051,
"num_tokens": 175259648.0,
"step": 5350
},
{
"epoch": 2.854461335095836,
"grad_norm": 0.09694032371044159,
"learning_rate": 0.00022124086795333116,
"loss": 0.2575,
"mean_token_accuracy": 0.9255865186452865,
"num_tokens": 176898048.0,
"step": 5400
},
{
"epoch": 2.8808988764044945,
"grad_norm": 0.09860485792160034,
"learning_rate": 0.0002185148838730782,
"loss": 0.2567,
"mean_token_accuracy": 0.9257099312543869,
"num_tokens": 178536448.0,
"step": 5450
},
{
"epoch": 2.907336417713153,
"grad_norm": 0.10467279702425003,
"learning_rate": 0.0002157888997928252,
"loss": 0.2537,
"step": 5500
},
{
"epoch": 2.907336417713153,
"eval_loss": 0.5691426396369934,
"eval_mean_token_accuracy": 0.8808775775402343,
"eval_num_tokens": 180174848.0,
"eval_runtime": 1445.1515,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 5500
},
{
"epoch": 2.933773959021811,
"grad_norm": 0.13350491225719452,
"learning_rate": 0.00021306291571257226,
"loss": 0.2548,
"mean_token_accuracy": 0.9265344049036502,
"num_tokens": 181813248.0,
"step": 5550
},
{
"epoch": 2.960211500330469,
"grad_norm": 0.10815941542387009,
"learning_rate": 0.00021033693163231926,
"loss": 0.2529,
"mean_token_accuracy": 0.9266226869821549,
"num_tokens": 183451648.0,
"step": 5600
},
{
"epoch": 2.9866490416391276,
"grad_norm": 0.09711950272321701,
"learning_rate": 0.0002076109475520663,
"loss": 0.2477,
"mean_token_accuracy": 0.928189767897129,
"num_tokens": 185090048.0,
"step": 5650
},
{
"epoch": 3.012690019828156,
"grad_norm": 0.07202441990375519,
"learning_rate": 0.0002049394831534184,
"loss": 0.1737,
"mean_token_accuracy": 0.9494730455016122,
"num_tokens": 186703872.0,
"step": 5700
},
{
"epoch": 3.0391275611368145,
"grad_norm": 0.09105918556451797,
"learning_rate": 0.0002022134990731654,
"loss": 0.0857,
"step": 5750
},
{
"epoch": 3.0391275611368145,
"eval_loss": 0.7057962417602539,
"eval_mean_token_accuracy": 0.8797050480736739,
"eval_num_tokens": 188342272.0,
"eval_runtime": 1445.2714,
"eval_samples_per_second": 5.234,
"eval_steps_per_second": 0.655,
"step": 5750
},
{
"epoch": 3.0655651024454724,
"grad_norm": 0.08339951187372208,
"learning_rate": 0.00019948751499291245,
"loss": 0.0836,
"mean_token_accuracy": 0.9740020345151424,
"num_tokens": 189980672.0,
"step": 5800
},
{
"epoch": 3.092002643754131,
"grad_norm": 0.07610207796096802,
"learning_rate": 0.00019676153091265948,
"loss": 0.0835,
"mean_token_accuracy": 0.9742461076378822,
"num_tokens": 191619072.0,
"step": 5850
},
{
"epoch": 3.118440185062789,
"grad_norm": 0.08562301844358444,
"learning_rate": 0.0001940355468324065,
"loss": 0.0841,
"mean_token_accuracy": 0.9739308601617813,
"num_tokens": 193257472.0,
"step": 5900
},
{
"epoch": 3.1448777263714476,
"grad_norm": 0.08178895711898804,
"learning_rate": 0.00019130956275215352,
"loss": 0.0824,
"mean_token_accuracy": 0.9746059566736222,
"num_tokens": 194895872.0,
"step": 5950
},
{
"epoch": 3.1713152676801055,
"grad_norm": 0.08008107542991638,
"learning_rate": 0.00018858357867190058,
"loss": 0.0826,
"step": 6000
},
{
"epoch": 3.1713152676801055,
"eval_loss": 0.718048095703125,
"eval_mean_token_accuracy": 0.8796352478842907,
"eval_num_tokens": 196534272.0,
"eval_runtime": 1444.6019,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 6000
},
{
"epoch": 3.197752808988764,
"grad_norm": 0.08383160084486008,
"learning_rate": 0.00018585759459164758,
"loss": 0.0833,
"mean_token_accuracy": 0.9743710477650166,
"num_tokens": 198172672.0,
"step": 6050
},
{
"epoch": 3.2241903502974223,
"grad_norm": 0.08407159149646759,
"learning_rate": 0.00018313161051139462,
"loss": 0.0843,
"mean_token_accuracy": 0.9739204734563828,
"num_tokens": 199811072.0,
"step": 6100
},
{
"epoch": 3.2506278916060807,
"grad_norm": 0.0758429542183876,
"learning_rate": 0.00018040562643114165,
"loss": 0.0825,
"mean_token_accuracy": 0.9741593536734581,
"num_tokens": 201449472.0,
"step": 6150
},
{
"epoch": 3.277065432914739,
"grad_norm": 0.08588337153196335,
"learning_rate": 0.00017767964235088868,
"loss": 0.0841,
"mean_token_accuracy": 0.9740646561980247,
"num_tokens": 203087872.0,
"step": 6200
},
{
"epoch": 3.303502974223397,
"grad_norm": 0.08413270860910416,
"learning_rate": 0.0001749536582706357,
"loss": 0.0859,
"step": 6250
},
{
"epoch": 3.303502974223397,
"eval_loss": 0.7220771908760071,
"eval_mean_token_accuracy": 0.8795976611700925,
"eval_num_tokens": 204726272.0,
"eval_runtime": 1445.7274,
"eval_samples_per_second": 5.233,
"eval_steps_per_second": 0.654,
"step": 6250
},
{
"epoch": 3.3299405155320554,
"grad_norm": 0.0801534354686737,
"learning_rate": 0.00017222767419038275,
"loss": 0.0856,
"mean_token_accuracy": 0.9733324344456196,
"num_tokens": 206364672.0,
"step": 6300
},
{
"epoch": 3.3563780568407138,
"grad_norm": 0.08935518562793732,
"learning_rate": 0.00016950169011012976,
"loss": 0.0851,
"mean_token_accuracy": 0.9735490143299103,
"num_tokens": 208003072.0,
"step": 6350
},
{
"epoch": 3.382815598149372,
"grad_norm": 0.08593747019767761,
"learning_rate": 0.0001667757060298768,
"loss": 0.0898,
"mean_token_accuracy": 0.9727987706661224,
"num_tokens": 209641472.0,
"step": 6400
},
{
"epoch": 3.4092531394580305,
"grad_norm": 0.08515259623527527,
"learning_rate": 0.00016404972194962382,
"loss": 0.0862,
"mean_token_accuracy": 0.9732246008515358,
"num_tokens": 211279872.0,
"step": 6450
},
{
"epoch": 3.4356906807666885,
"grad_norm": 0.08400440216064453,
"learning_rate": 0.00016132373786937086,
"loss": 0.0848,
"step": 6500
},
{
"epoch": 3.4356906807666885,
"eval_loss": 0.7208439111709595,
"eval_mean_token_accuracy": 0.8801698074154319,
"eval_num_tokens": 212918272.0,
"eval_runtime": 1445.2121,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 6500
},
{
"epoch": 3.462128222075347,
"grad_norm": 0.09465966373682022,
"learning_rate": 0.00015859775378911786,
"loss": 0.0845,
"mean_token_accuracy": 0.9737170253694057,
"num_tokens": 214556672.0,
"step": 6550
},
{
"epoch": 3.4885657633840053,
"grad_norm": 0.08350127190351486,
"learning_rate": 0.0001558717697088649,
"loss": 0.0845,
"mean_token_accuracy": 0.9738056141138077,
"num_tokens": 216195072.0,
"step": 6600
},
{
"epoch": 3.5150033046926636,
"grad_norm": 0.09258560091257095,
"learning_rate": 0.00015314578562861193,
"loss": 0.0867,
"mean_token_accuracy": 0.973144568502903,
"num_tokens": 217833472.0,
"step": 6650
},
{
"epoch": 3.541440846001322,
"grad_norm": 0.08615857362747192,
"learning_rate": 0.00015041980154835896,
"loss": 0.0836,
"mean_token_accuracy": 0.9740035626292228,
"num_tokens": 219471872.0,
"step": 6700
},
{
"epoch": 3.56787838730998,
"grad_norm": 0.08243568986654282,
"learning_rate": 0.00014769381746810597,
"loss": 0.0847,
"step": 6750
},
{
"epoch": 3.56787838730998,
"eval_loss": 0.7248117923736572,
"eval_mean_token_accuracy": 0.8803664739489807,
"eval_num_tokens": 221110272.0,
"eval_runtime": 1445.5737,
"eval_samples_per_second": 5.233,
"eval_steps_per_second": 0.654,
"step": 6750
},
{
"epoch": 3.5943159286186384,
"grad_norm": 0.08806952089071274,
"learning_rate": 0.00014496783338785303,
"loss": 0.0861,
"mean_token_accuracy": 0.9733803936839104,
"num_tokens": 222748672.0,
"step": 6800
},
{
"epoch": 3.6207534699272967,
"grad_norm": 0.08953431993722916,
"learning_rate": 0.00014224184930760003,
"loss": 0.0856,
"mean_token_accuracy": 0.9733651205897331,
"num_tokens": 224387072.0,
"step": 6850
},
{
"epoch": 3.647191011235955,
"grad_norm": 0.08528242260217667,
"learning_rate": 0.00013951586522734707,
"loss": 0.0852,
"mean_token_accuracy": 0.9733651199936867,
"num_tokens": 226025472.0,
"step": 6900
},
{
"epoch": 3.6736285525446135,
"grad_norm": 0.08363524079322815,
"learning_rate": 0.0001367898811470941,
"loss": 0.0845,
"mean_token_accuracy": 0.9736650982499122,
"num_tokens": 227663872.0,
"step": 6950
},
{
"epoch": 3.7000660938532715,
"grad_norm": 0.07602488994598389,
"learning_rate": 0.00013406389706684113,
"loss": 0.0845,
"step": 7000
},
{
"epoch": 3.7000660938532715,
"eval_loss": 0.7213518023490906,
"eval_mean_token_accuracy": 0.8806469935695415,
"eval_num_tokens": 229302272.0,
"eval_runtime": 1444.415,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 7000
},
{
"epoch": 3.72650363516193,
"grad_norm": 0.08517055958509445,
"learning_rate": 0.00013133791298658814,
"loss": 0.0847,
"mean_token_accuracy": 0.9737151943147182,
"num_tokens": 230940672.0,
"step": 7050
},
{
"epoch": 3.7529411764705882,
"grad_norm": 0.0767749696969986,
"learning_rate": 0.0001286119289063352,
"loss": 0.0828,
"mean_token_accuracy": 0.9742509955167771,
"num_tokens": 232579072.0,
"step": 7100
},
{
"epoch": 3.7793787177792466,
"grad_norm": 0.08359658718109131,
"learning_rate": 0.0001258859448260822,
"loss": 0.0849,
"mean_token_accuracy": 0.973515414595604,
"num_tokens": 234217472.0,
"step": 7150
},
{
"epoch": 3.805816259087905,
"grad_norm": 0.07397591322660446,
"learning_rate": 0.00012315996074582924,
"loss": 0.0835,
"mean_token_accuracy": 0.9739962309598923,
"num_tokens": 235855872.0,
"step": 7200
},
{
"epoch": 3.832253800396563,
"grad_norm": 0.08941724896430969,
"learning_rate": 0.00012043397666557627,
"loss": 0.0821,
"step": 7250
},
{
"epoch": 3.832253800396563,
"eval_loss": 0.727186381816864,
"eval_mean_token_accuracy": 0.881018961017782,
"eval_num_tokens": 237494272.0,
"eval_runtime": 1445.0501,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 7250
},
{
"epoch": 3.8586913417052213,
"grad_norm": 0.08002132922410965,
"learning_rate": 0.0001177079925853233,
"loss": 0.0824,
"mean_token_accuracy": 0.9744101479649544,
"num_tokens": 239132672.0,
"step": 7300
},
{
"epoch": 3.8851288830138797,
"grad_norm": 0.08109039068222046,
"learning_rate": 0.00011498200850507034,
"loss": 0.0842,
"mean_token_accuracy": 0.9738893151283264,
"num_tokens": 240771072.0,
"step": 7350
},
{
"epoch": 3.911566424322538,
"grad_norm": 0.08812084048986435,
"learning_rate": 0.00011225602442481736,
"loss": 0.0808,
"mean_token_accuracy": 0.9748161256313324,
"num_tokens": 242409472.0,
"step": 7400
},
{
"epoch": 3.9380039656311965,
"grad_norm": 0.08132945001125336,
"learning_rate": 0.00010953004034456439,
"loss": 0.0824,
"mean_token_accuracy": 0.9742913186550141,
"num_tokens": 244047872.0,
"step": 7450
},
{
"epoch": 3.9644415069398544,
"grad_norm": 0.08432092517614365,
"learning_rate": 0.00010680405626431142,
"loss": 0.0815,
"step": 7500
},
{
"epoch": 3.9644415069398544,
"eval_loss": 0.727695643901825,
"eval_mean_token_accuracy": 0.8814008755109275,
"eval_num_tokens": 245686272.0,
"eval_runtime": 1444.3382,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 7500
},
{
"epoch": 3.990879048248513,
"grad_norm": 0.08465278148651123,
"learning_rate": 0.00010407807218405844,
"loss": 0.0823,
"mean_token_accuracy": 0.9746123726665974,
"num_tokens": 247324672.0,
"step": 7550
},
{
"epoch": 4.016920026437541,
"grad_norm": 0.04739232733845711,
"learning_rate": 0.00010135208810380548,
"loss": 0.0478,
"mean_token_accuracy": 0.9862868610372398,
"num_tokens": 248938496.0,
"step": 7600
},
{
"epoch": 4.0433575677462,
"grad_norm": 0.05006248131394386,
"learning_rate": 9.86261040235525e-05,
"loss": 0.0278,
"mean_token_accuracy": 0.9926484674215317,
"num_tokens": 250576896.0,
"step": 7650
},
{
"epoch": 4.069795109054858,
"grad_norm": 0.048755794763565063,
"learning_rate": 9.590011994329953e-05,
"loss": 0.0275,
"mean_token_accuracy": 0.9927639365196228,
"num_tokens": 252215296.0,
"step": 7700
},
{
"epoch": 4.0962326503635165,
"grad_norm": 0.04973224923014641,
"learning_rate": 9.317413586304656e-05,
"loss": 0.0269,
"step": 7750
},
{
"epoch": 4.0962326503635165,
"eval_loss": 0.8192626237869263,
"eval_mean_token_accuracy": 0.88220195217062,
"eval_num_tokens": 253853696.0,
"eval_runtime": 1444.8032,
"eval_samples_per_second": 5.236,
"eval_steps_per_second": 0.655,
"step": 7750
},
{
"epoch": 4.122670191672174,
"grad_norm": 0.05098351836204529,
"learning_rate": 9.044815178279358e-05,
"loss": 0.0268,
"mean_token_accuracy": 0.9928580224514008,
"num_tokens": 255492096.0,
"step": 7800
},
{
"epoch": 4.149107732980832,
"grad_norm": 0.04829578474164009,
"learning_rate": 8.772216770254061e-05,
"loss": 0.0272,
"mean_token_accuracy": 0.9927712675929069,
"num_tokens": 257130496.0,
"step": 7850
},
{
"epoch": 4.175545274289491,
"grad_norm": 0.04151781275868416,
"learning_rate": 8.499618362228765e-05,
"loss": 0.0269,
"mean_token_accuracy": 0.9928335842490196,
"num_tokens": 258768896.0,
"step": 7900
},
{
"epoch": 4.201982815598149,
"grad_norm": 0.05445162579417229,
"learning_rate": 8.227019954203467e-05,
"loss": 0.0265,
"mean_token_accuracy": 0.9929741024971008,
"num_tokens": 260407296.0,
"step": 7950
},
{
"epoch": 4.228420356906808,
"grad_norm": 0.03888251632452011,
"learning_rate": 7.95442154617817e-05,
"loss": 0.0273,
"step": 8000
},
{
"epoch": 4.228420356906808,
"eval_loss": 0.8266236782073975,
"eval_mean_token_accuracy": 0.8823872005712659,
"eval_num_tokens": 262045696.0,
"eval_runtime": 1444.6622,
"eval_samples_per_second": 5.237,
"eval_steps_per_second": 0.655,
"step": 8000
},
{
"epoch": 4.254857898215466,
"grad_norm": 0.049200285226106644,
"learning_rate": 7.681823138152873e-05,
"loss": 0.0265,
"mean_token_accuracy": 0.9928293077647686,
"num_tokens": 263684096.0,
"step": 8050
},
{
"epoch": 4.281295439524124,
"grad_norm": 0.04635064676403999,
"learning_rate": 7.409224730127575e-05,
"loss": 0.0259,
"mean_token_accuracy": 0.9931225627660751,
"num_tokens": 265322496.0,
"step": 8100
},
{
"epoch": 4.307732980832783,
"grad_norm": 0.047765497118234634,
"learning_rate": 7.136626322102279e-05,
"loss": 0.0269,
"mean_token_accuracy": 0.992902010679245,
"num_tokens": 266960896.0,
"step": 8150
},
{
"epoch": 4.334170522141441,
"grad_norm": 0.040082089602947235,
"learning_rate": 6.864027914076983e-05,
"loss": 0.0327,
"mean_token_accuracy": 0.9925085598230362,
"num_tokens": 268599296.0,
"step": 8200
},
{
"epoch": 4.360608063450099,
"grad_norm": 0.040357593446969986,
"learning_rate": 6.591429506051685e-05,
"loss": 0.0262,
"step": 8250
},
{
"epoch": 4.360608063450099,
"eval_loss": 0.8343076109886169,
"eval_mean_token_accuracy": 0.8827618814169983,
"eval_num_tokens": 270237696.0,
"eval_runtime": 1444.2593,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 8250
},
{
"epoch": 4.387045604758757,
"grad_norm": 0.04214048385620117,
"learning_rate": 6.318831098026388e-05,
"loss": 0.0258,
"mean_token_accuracy": 0.9930578023195267,
"num_tokens": 271876096.0,
"step": 8300
},
{
"epoch": 4.413483146067415,
"grad_norm": 0.0495334193110466,
"learning_rate": 6.0462326900010904e-05,
"loss": 0.0259,
"mean_token_accuracy": 0.9930883496999741,
"num_tokens": 273514496.0,
"step": 8350
},
{
"epoch": 4.439920687376074,
"grad_norm": 0.04878053814172745,
"learning_rate": 5.773634281975793e-05,
"loss": 0.0256,
"mean_token_accuracy": 0.9932361990213394,
"num_tokens": 275152896.0,
"step": 8400
},
{
"epoch": 4.466358228684732,
"grad_norm": 0.04534803330898285,
"learning_rate": 5.5010358739504963e-05,
"loss": 0.0257,
"mean_token_accuracy": 0.9931818246841431,
"num_tokens": 276791296.0,
"step": 8450
},
{
"epoch": 4.492795769993391,
"grad_norm": 0.04720517247915268,
"learning_rate": 5.228437465925199e-05,
"loss": 0.0255,
"step": 8500
},
{
"epoch": 4.492795769993391,
"eval_loss": 0.8357202410697937,
"eval_mean_token_accuracy": 0.8828653151338751,
"eval_num_tokens": 278429696.0,
"eval_runtime": 1445.1121,
"eval_samples_per_second": 5.235,
"eval_steps_per_second": 0.655,
"step": 8500
},
{
"epoch": 4.519233311302049,
"grad_norm": 0.04248838871717453,
"learning_rate": 4.9558390578999016e-05,
"loss": 0.0252,
"mean_token_accuracy": 0.9932594150304794,
"num_tokens": 280068096.0,
"step": 8550
},
{
"epoch": 4.545670852610707,
"grad_norm": 0.04879326745867729,
"learning_rate": 4.683240649874604e-05,
"loss": 0.0256,
"mean_token_accuracy": 0.9932215362787247,
"num_tokens": 281706496.0,
"step": 8600
},
{
"epoch": 4.572108393919366,
"grad_norm": 0.04033661261200905,
"learning_rate": 4.4106422418493076e-05,
"loss": 0.0256,
"mean_token_accuracy": 0.993228867650032,
"num_tokens": 283344896.0,
"step": 8650
},
{
"epoch": 4.598545935228024,
"grad_norm": 0.043814148753881454,
"learning_rate": 4.138043833824011e-05,
"loss": 0.0251,
"mean_token_accuracy": 0.9933846592903137,
"num_tokens": 284983296.0,
"step": 8700
},
{
"epoch": 4.624983476536682,
"grad_norm": 0.04637971520423889,
"learning_rate": 3.8654454257987135e-05,
"loss": 0.0256,
"step": 8750
},
{
"epoch": 4.624983476536682,
"eval_loss": 0.8386735916137695,
"eval_mean_token_accuracy": 0.8830419348485878,
"eval_num_tokens": 286621696.0,
"eval_runtime": 1444.7416,
"eval_samples_per_second": 5.236,
"eval_steps_per_second": 0.655,
"step": 8750
},
{
"epoch": 4.65142101784534,
"grad_norm": 0.04898098483681679,
"learning_rate": 3.592847017773417e-05,
"loss": 0.025,
"mean_token_accuracy": 0.9933046251535416,
"num_tokens": 288260096.0,
"step": 8800
},
{
"epoch": 4.677858559153998,
"grad_norm": 0.04658009856939316,
"learning_rate": 3.3202486097481194e-05,
"loss": 0.025,
"mean_token_accuracy": 0.9934017658233643,
"num_tokens": 289898496.0,
"step": 8850
},
{
"epoch": 4.704296100462657,
"grad_norm": 0.04260571300983429,
"learning_rate": 3.0476502017228217e-05,
"loss": 0.0246,
"mean_token_accuracy": 0.9935636669397354,
"num_tokens": 291536896.0,
"step": 8900
},
{
"epoch": 4.730733641771315,
"grad_norm": 0.04221141338348389,
"learning_rate": 2.775051793697525e-05,
"loss": 0.0249,
"mean_token_accuracy": 0.9934304803609848,
"num_tokens": 293175296.0,
"step": 8950
},
{
"epoch": 4.757171183079974,
"grad_norm": 0.046503596007823944,
"learning_rate": 2.502453385672228e-05,
"loss": 0.0247,
"step": 9000
},
{
"epoch": 4.757171183079974,
"eval_loss": 0.8432579636573792,
"eval_mean_token_accuracy": 0.8832672491008081,
"eval_num_tokens": 294813696.0,
"eval_runtime": 1444.873,
"eval_samples_per_second": 5.236,
"eval_steps_per_second": 0.655,
"step": 9000
},
{
"epoch": 4.783608724388632,
"grad_norm": 0.049112509936094284,
"learning_rate": 2.2298549776469306e-05,
"loss": 0.0244,
"mean_token_accuracy": 0.9935187622904778,
"num_tokens": 296452096.0,
"step": 9050
},
{
"epoch": 4.81004626569729,
"grad_norm": 0.049860984086990356,
"learning_rate": 1.9572565696216336e-05,
"loss": 0.0248,
"mean_token_accuracy": 0.9935349524021149,
"num_tokens": 298090496.0,
"step": 9100
},
{
"epoch": 4.836483807005949,
"grad_norm": 0.041883405297994614,
"learning_rate": 1.6846581615963362e-05,
"loss": 0.0243,
"mean_token_accuracy": 0.9936204850673676,
"num_tokens": 299728896.0,
"step": 9150
},
{
"epoch": 4.8629213483146065,
"grad_norm": 0.043799083679914474,
"learning_rate": 1.4120597535710392e-05,
"loss": 0.024,
"mean_token_accuracy": 0.9937127381563187,
"num_tokens": 301367296.0,
"step": 9200
},
{
"epoch": 4.889358889623265,
"grad_norm": 0.04291240870952606,
"learning_rate": 1.139461345545742e-05,
"loss": 0.0238,
"step": 9250
},
{
"epoch": 4.889358889623265,
"eval_loss": 0.8444077372550964,
"eval_mean_token_accuracy": 0.88355980766248,
"eval_num_tokens": 303005696.0,
"eval_runtime": 1444.2008,
"eval_samples_per_second": 5.238,
"eval_steps_per_second": 0.655,
"step": 9250
},
{
"epoch": 4.915796430931923,
"grad_norm": 0.04179864004254341,
"learning_rate": 8.668629375204448e-06,
"loss": 0.0237,
"mean_token_accuracy": 0.9937710836529732,
"num_tokens": 304644096.0,
"step": 9300
},
{
"epoch": 4.942233972240581,
"grad_norm": 0.04021856561303139,
"learning_rate": 5.942645294951477e-06,
"loss": 0.0239,
"mean_token_accuracy": 0.993884414434433,
"num_tokens": 306282496.0,
"step": 9350
},
{
"epoch": 4.96867151354924,
"grad_norm": 0.04214942827820778,
"learning_rate": 3.2166612146985063e-06,
"loss": 0.0243,
"mean_token_accuracy": 0.9936669170856476,
"num_tokens": 307920896.0,
"step": 9400
},
{
"epoch": 4.995109054857898,
"grad_norm": 0.04387414827942848,
"learning_rate": 4.906771344455349e-07,
"loss": 0.0242,
"mean_token_accuracy": 0.9936302602291107,
"num_tokens": 309559296.0,
"step": 9450
}
],
"logging_steps": 50,
"max_steps": 9455,
"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": true
},
"attributes": {}
}
},
"total_flos": 464304822288384.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}