code_longest_science_longest / trainer_state.json
ryanmarten's picture
End of training
15d5af7 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 4.94557270511779,
"eval_steps": 500,
"global_step": 380,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.012997562956945572,
"grad_norm": 6.099783948625711,
"learning_rate": 2.105263157894737e-06,
"loss": 1.0219,
"step": 1
},
{
"epoch": 0.025995125913891144,
"grad_norm": 6.068915599090855,
"learning_rate": 4.210526315789474e-06,
"loss": 1.0148,
"step": 2
},
{
"epoch": 0.03899268887083672,
"grad_norm": 5.594712813142529,
"learning_rate": 6.31578947368421e-06,
"loss": 0.9975,
"step": 3
},
{
"epoch": 0.05199025182778229,
"grad_norm": 4.038714211131839,
"learning_rate": 8.421052631578948e-06,
"loss": 0.9696,
"step": 4
},
{
"epoch": 0.06498781478472786,
"grad_norm": 2.365897221739178,
"learning_rate": 1.0526315789473684e-05,
"loss": 0.9322,
"step": 5
},
{
"epoch": 0.07798537774167344,
"grad_norm": 4.808874284957715,
"learning_rate": 1.263157894736842e-05,
"loss": 0.9538,
"step": 6
},
{
"epoch": 0.09098294069861901,
"grad_norm": 4.840119194557309,
"learning_rate": 1.4736842105263159e-05,
"loss": 0.9456,
"step": 7
},
{
"epoch": 0.10398050365556458,
"grad_norm": 5.7169851588504645,
"learning_rate": 1.6842105263157896e-05,
"loss": 0.9349,
"step": 8
},
{
"epoch": 0.11697806661251016,
"grad_norm": 4.120707061579036,
"learning_rate": 1.894736842105263e-05,
"loss": 0.9107,
"step": 9
},
{
"epoch": 0.12997562956945571,
"grad_norm": 2.7497360309830476,
"learning_rate": 2.105263157894737e-05,
"loss": 0.8669,
"step": 10
},
{
"epoch": 0.1429731925264013,
"grad_norm": 1.8093148555009293,
"learning_rate": 2.3157894736842107e-05,
"loss": 0.8289,
"step": 11
},
{
"epoch": 0.15597075548334688,
"grad_norm": 1.617836052019221,
"learning_rate": 2.526315789473684e-05,
"loss": 0.8096,
"step": 12
},
{
"epoch": 0.16896831844029245,
"grad_norm": 1.0744851220126015,
"learning_rate": 2.7368421052631583e-05,
"loss": 0.7877,
"step": 13
},
{
"epoch": 0.18196588139723802,
"grad_norm": 1.2705200346488472,
"learning_rate": 2.9473684210526317e-05,
"loss": 0.78,
"step": 14
},
{
"epoch": 0.19496344435418358,
"grad_norm": 1.4130680513032683,
"learning_rate": 3.157894736842106e-05,
"loss": 0.7769,
"step": 15
},
{
"epoch": 0.20796100731112915,
"grad_norm": 1.324813747526517,
"learning_rate": 3.368421052631579e-05,
"loss": 0.7705,
"step": 16
},
{
"epoch": 0.22095857026807472,
"grad_norm": 1.2590819742707091,
"learning_rate": 3.578947368421053e-05,
"loss": 0.7458,
"step": 17
},
{
"epoch": 0.23395613322502032,
"grad_norm": 1.2135489907048478,
"learning_rate": 3.789473684210526e-05,
"loss": 0.7415,
"step": 18
},
{
"epoch": 0.2469536961819659,
"grad_norm": 0.955476297443492,
"learning_rate": 4e-05,
"loss": 0.7311,
"step": 19
},
{
"epoch": 0.25995125913891143,
"grad_norm": 1.6971617462330089,
"learning_rate": 4.210526315789474e-05,
"loss": 0.7413,
"step": 20
},
{
"epoch": 0.272948822095857,
"grad_norm": 1.4395879957659508,
"learning_rate": 4.421052631578948e-05,
"loss": 0.7295,
"step": 21
},
{
"epoch": 0.2859463850528026,
"grad_norm": 1.171007731697555,
"learning_rate": 4.6315789473684214e-05,
"loss": 0.7153,
"step": 22
},
{
"epoch": 0.29894394800974816,
"grad_norm": 1.931104084566943,
"learning_rate": 4.842105263157895e-05,
"loss": 0.7249,
"step": 23
},
{
"epoch": 0.31194151096669376,
"grad_norm": 1.1062671298809976,
"learning_rate": 5.052631578947368e-05,
"loss": 0.7178,
"step": 24
},
{
"epoch": 0.3249390739236393,
"grad_norm": 2.0237024200236986,
"learning_rate": 5.263157894736843e-05,
"loss": 0.7214,
"step": 25
},
{
"epoch": 0.3379366368805849,
"grad_norm": 1.6595941751589156,
"learning_rate": 5.4736842105263165e-05,
"loss": 0.702,
"step": 26
},
{
"epoch": 0.35093419983753044,
"grad_norm": 1.643241943721994,
"learning_rate": 5.68421052631579e-05,
"loss": 0.7054,
"step": 27
},
{
"epoch": 0.36393176279447603,
"grad_norm": 1.9552851038948404,
"learning_rate": 5.8947368421052634e-05,
"loss": 0.7013,
"step": 28
},
{
"epoch": 0.37692932575142163,
"grad_norm": 1.1207858673487592,
"learning_rate": 6.105263157894738e-05,
"loss": 0.6901,
"step": 29
},
{
"epoch": 0.38992688870836717,
"grad_norm": 2.1963019920042757,
"learning_rate": 6.315789473684212e-05,
"loss": 0.6868,
"step": 30
},
{
"epoch": 0.40292445166531277,
"grad_norm": 1.4781504300412824,
"learning_rate": 6.526315789473685e-05,
"loss": 0.6903,
"step": 31
},
{
"epoch": 0.4159220146222583,
"grad_norm": 2.122635652457263,
"learning_rate": 6.736842105263159e-05,
"loss": 0.6858,
"step": 32
},
{
"epoch": 0.4289195775792039,
"grad_norm": 1.650492771547578,
"learning_rate": 6.947368421052632e-05,
"loss": 0.6898,
"step": 33
},
{
"epoch": 0.44191714053614944,
"grad_norm": 1.5236371764651064,
"learning_rate": 7.157894736842105e-05,
"loss": 0.6813,
"step": 34
},
{
"epoch": 0.45491470349309504,
"grad_norm": 1.185803755084194,
"learning_rate": 7.368421052631579e-05,
"loss": 0.6829,
"step": 35
},
{
"epoch": 0.46791226645004064,
"grad_norm": 1.9405604072257883,
"learning_rate": 7.578947368421052e-05,
"loss": 0.6738,
"step": 36
},
{
"epoch": 0.4809098294069862,
"grad_norm": 1.363411287783038,
"learning_rate": 7.789473684210527e-05,
"loss": 0.6696,
"step": 37
},
{
"epoch": 0.4939073923639318,
"grad_norm": 2.0874162228533115,
"learning_rate": 8e-05,
"loss": 0.6753,
"step": 38
},
{
"epoch": 0.5069049553208773,
"grad_norm": 1.8491550544598394,
"learning_rate": 7.99983123807325e-05,
"loss": 0.6723,
"step": 39
},
{
"epoch": 0.5199025182778229,
"grad_norm": 1.2473648539637896,
"learning_rate": 7.999324966533291e-05,
"loss": 0.66,
"step": 40
},
{
"epoch": 0.5329000812347685,
"grad_norm": 3.3198725973160084,
"learning_rate": 7.998481228099806e-05,
"loss": 0.675,
"step": 41
},
{
"epoch": 0.545897644191714,
"grad_norm": 2.604690756502679,
"learning_rate": 7.997300093968255e-05,
"loss": 0.6824,
"step": 42
},
{
"epoch": 0.5588952071486596,
"grad_norm": 2.02050667371959,
"learning_rate": 7.995781663803876e-05,
"loss": 0.6667,
"step": 43
},
{
"epoch": 0.5718927701056052,
"grad_norm": 1.1560247057038098,
"learning_rate": 7.993926065733265e-05,
"loss": 0.6619,
"step": 44
},
{
"epoch": 0.5848903330625508,
"grad_norm": 1.5367530120814719,
"learning_rate": 7.991733456333579e-05,
"loss": 0.6664,
"step": 45
},
{
"epoch": 0.5978878960194963,
"grad_norm": 1.1634220605553418,
"learning_rate": 7.98920402061931e-05,
"loss": 0.6493,
"step": 46
},
{
"epoch": 0.6108854589764419,
"grad_norm": 1.015353240524831,
"learning_rate": 7.98633797202668e-05,
"loss": 0.659,
"step": 47
},
{
"epoch": 0.6238830219333875,
"grad_norm": 1.4952317667466968,
"learning_rate": 7.98313555239563e-05,
"loss": 0.6609,
"step": 48
},
{
"epoch": 0.6368805848903331,
"grad_norm": 0.8908692378681831,
"learning_rate": 7.979597031949415e-05,
"loss": 0.6638,
"step": 49
},
{
"epoch": 0.6498781478472786,
"grad_norm": 1.7121745994414552,
"learning_rate": 7.975722709271799e-05,
"loss": 0.6636,
"step": 50
},
{
"epoch": 0.6628757108042242,
"grad_norm": 1.2827332360089716,
"learning_rate": 7.97151291128186e-05,
"loss": 0.6562,
"step": 51
},
{
"epoch": 0.6758732737611698,
"grad_norm": 1.1190183435245546,
"learning_rate": 7.96696799320641e-05,
"loss": 0.6467,
"step": 52
},
{
"epoch": 0.6888708367181153,
"grad_norm": 1.5796475425952419,
"learning_rate": 7.962088338550013e-05,
"loss": 0.6549,
"step": 53
},
{
"epoch": 0.7018683996750609,
"grad_norm": 1.0356021868877745,
"learning_rate": 7.956874359062632e-05,
"loss": 0.6486,
"step": 54
},
{
"epoch": 0.7148659626320065,
"grad_norm": 1.689157245211179,
"learning_rate": 7.951326494704878e-05,
"loss": 0.6427,
"step": 55
},
{
"epoch": 0.7278635255889521,
"grad_norm": 1.3036787434531885,
"learning_rate": 7.94544521361089e-05,
"loss": 0.6451,
"step": 56
},
{
"epoch": 0.7408610885458976,
"grad_norm": 1.411604156835551,
"learning_rate": 7.939231012048833e-05,
"loss": 0.6434,
"step": 57
},
{
"epoch": 0.7538586515028433,
"grad_norm": 1.073329384486516,
"learning_rate": 7.932684414379021e-05,
"loss": 0.6417,
"step": 58
},
{
"epoch": 0.7668562144597888,
"grad_norm": 1.1364966234135832,
"learning_rate": 7.925805973009672e-05,
"loss": 0.6348,
"step": 59
},
{
"epoch": 0.7798537774167343,
"grad_norm": 0.9594628748904098,
"learning_rate": 7.918596268350296e-05,
"loss": 0.6399,
"step": 60
},
{
"epoch": 0.7928513403736799,
"grad_norm": 1.3124733925076593,
"learning_rate": 7.911055908762718e-05,
"loss": 0.6445,
"step": 61
},
{
"epoch": 0.8058489033306255,
"grad_norm": 0.8150161261530253,
"learning_rate": 7.903185530509743e-05,
"loss": 0.6333,
"step": 62
},
{
"epoch": 0.8188464662875711,
"grad_norm": 0.8675788545202524,
"learning_rate": 7.894985797701472e-05,
"loss": 0.6342,
"step": 63
},
{
"epoch": 0.8318440292445166,
"grad_norm": 0.8701303058033523,
"learning_rate": 7.886457402239256e-05,
"loss": 0.627,
"step": 64
},
{
"epoch": 0.8448415922014623,
"grad_norm": 1.1211379390083465,
"learning_rate": 7.877601063757323e-05,
"loss": 0.6346,
"step": 65
},
{
"epoch": 0.8578391551584078,
"grad_norm": 1.0237659842661562,
"learning_rate": 7.868417529562043e-05,
"loss": 0.6244,
"step": 66
},
{
"epoch": 0.8708367181153533,
"grad_norm": 0.8724494077448317,
"learning_rate": 7.858907574568882e-05,
"loss": 0.6279,
"step": 67
},
{
"epoch": 0.8838342810722989,
"grad_norm": 0.8890596583749399,
"learning_rate": 7.849072001237001e-05,
"loss": 0.6212,
"step": 68
},
{
"epoch": 0.8968318440292445,
"grad_norm": 0.9491308234698868,
"learning_rate": 7.838911639501557e-05,
"loss": 0.6164,
"step": 69
},
{
"epoch": 0.9098294069861901,
"grad_norm": 0.9144632388033332,
"learning_rate": 7.828427346703657e-05,
"loss": 0.6141,
"step": 70
},
{
"epoch": 0.9228269699431356,
"grad_norm": 1.0369164699296847,
"learning_rate": 7.81762000751803e-05,
"loss": 0.6227,
"step": 71
},
{
"epoch": 0.9358245329000813,
"grad_norm": 1.0824992143715033,
"learning_rate": 7.806490533878368e-05,
"loss": 0.6219,
"step": 72
},
{
"epoch": 0.9488220958570268,
"grad_norm": 0.8841365960283825,
"learning_rate": 7.795039864900378e-05,
"loss": 0.6134,
"step": 73
},
{
"epoch": 0.9618196588139724,
"grad_norm": 0.6911046945305666,
"learning_rate": 7.783268966802539e-05,
"loss": 0.6263,
"step": 74
},
{
"epoch": 0.974817221770918,
"grad_norm": 0.7831186784164668,
"learning_rate": 7.771178832824573e-05,
"loss": 0.6133,
"step": 75
},
{
"epoch": 0.9878147847278635,
"grad_norm": 0.6782592588604553,
"learning_rate": 7.758770483143634e-05,
"loss": 0.6102,
"step": 76
},
{
"epoch": 1.0024370430544274,
"grad_norm": 0.6336474343134532,
"learning_rate": 7.74604496478822e-05,
"loss": 0.6116,
"step": 77
},
{
"epoch": 1.0154346060113728,
"grad_norm": 0.6278277714102893,
"learning_rate": 7.733003351549829e-05,
"loss": 0.5828,
"step": 78
},
{
"epoch": 1.0284321689683185,
"grad_norm": 0.6209449077089193,
"learning_rate": 7.719646743892352e-05,
"loss": 0.584,
"step": 79
},
{
"epoch": 1.0414297319252641,
"grad_norm": 0.73979662934484,
"learning_rate": 7.705976268859207e-05,
"loss": 0.5892,
"step": 80
},
{
"epoch": 1.0544272948822095,
"grad_norm": 0.8782533057922014,
"learning_rate": 7.691993079978252e-05,
"loss": 0.5809,
"step": 81
},
{
"epoch": 1.0674248578391552,
"grad_norm": 0.9723345888414255,
"learning_rate": 7.677698357164431e-05,
"loss": 0.5745,
"step": 82
},
{
"epoch": 1.0804224207961006,
"grad_norm": 0.9060339939969984,
"learning_rate": 7.663093306620231e-05,
"loss": 0.5768,
"step": 83
},
{
"epoch": 1.0934199837530463,
"grad_norm": 0.6635148814822602,
"learning_rate": 7.648179160733883e-05,
"loss": 0.5735,
"step": 84
},
{
"epoch": 1.106417546709992,
"grad_norm": 0.626645192938145,
"learning_rate": 7.632957177975387e-05,
"loss": 0.5774,
"step": 85
},
{
"epoch": 1.1194151096669374,
"grad_norm": 0.5880549876089324,
"learning_rate": 7.61742864279031e-05,
"loss": 0.5806,
"step": 86
},
{
"epoch": 1.132412672623883,
"grad_norm": 0.7256826177123274,
"learning_rate": 7.601594865491414e-05,
"loss": 0.5725,
"step": 87
},
{
"epoch": 1.1454102355808287,
"grad_norm": 1.0226725048413887,
"learning_rate": 7.585457182148081e-05,
"loss": 0.5766,
"step": 88
},
{
"epoch": 1.158407798537774,
"grad_norm": 1.053827253244498,
"learning_rate": 7.569016954473577e-05,
"loss": 0.5781,
"step": 89
},
{
"epoch": 1.1714053614947197,
"grad_norm": 0.9313744365954294,
"learning_rate": 7.552275569710152e-05,
"loss": 0.5826,
"step": 90
},
{
"epoch": 1.1844029244516654,
"grad_norm": 1.054632648598746,
"learning_rate": 7.535234440511979e-05,
"loss": 0.581,
"step": 91
},
{
"epoch": 1.1974004874086108,
"grad_norm": 0.8836759025467023,
"learning_rate": 7.517895004825956e-05,
"loss": 0.5821,
"step": 92
},
{
"epoch": 1.2103980503655565,
"grad_norm": 1.127228102294561,
"learning_rate": 7.500258725770375e-05,
"loss": 0.5761,
"step": 93
},
{
"epoch": 1.2233956133225021,
"grad_norm": 0.8941981850633482,
"learning_rate": 7.48232709151145e-05,
"loss": 0.5695,
"step": 94
},
{
"epoch": 1.2363931762794476,
"grad_norm": 0.820917113713941,
"learning_rate": 7.464101615137756e-05,
"loss": 0.5805,
"step": 95
},
{
"epoch": 1.2493907392363932,
"grad_norm": 0.9174411331061607,
"learning_rate": 7.445583834532546e-05,
"loss": 0.5649,
"step": 96
},
{
"epoch": 1.2623883021933389,
"grad_norm": 0.6500765370181957,
"learning_rate": 7.426775312243986e-05,
"loss": 0.5758,
"step": 97
},
{
"epoch": 1.2753858651502843,
"grad_norm": 0.763425847678293,
"learning_rate": 7.407677635353308e-05,
"loss": 0.574,
"step": 98
},
{
"epoch": 1.28838342810723,
"grad_norm": 0.8352567885058773,
"learning_rate": 7.388292415340888e-05,
"loss": 0.5685,
"step": 99
},
{
"epoch": 1.3013809910641756,
"grad_norm": 0.4826298499702842,
"learning_rate": 7.368621287950264e-05,
"loss": 0.5645,
"step": 100
},
{
"epoch": 1.314378554021121,
"grad_norm": 0.5903830880665961,
"learning_rate": 7.348665913050115e-05,
"loss": 0.5602,
"step": 101
},
{
"epoch": 1.3273761169780667,
"grad_norm": 0.790277966015901,
"learning_rate": 7.328427974494201e-05,
"loss": 0.5676,
"step": 102
},
{
"epoch": 1.340373679935012,
"grad_norm": 0.7512617541518178,
"learning_rate": 7.307909179979274e-05,
"loss": 0.5648,
"step": 103
},
{
"epoch": 1.3533712428919578,
"grad_norm": 0.5851295705571637,
"learning_rate": 7.28711126090098e-05,
"loss": 0.5651,
"step": 104
},
{
"epoch": 1.3663688058489034,
"grad_norm": 0.41913986382054075,
"learning_rate": 7.266035972207773e-05,
"loss": 0.5572,
"step": 105
},
{
"epoch": 1.3793663688058488,
"grad_norm": 0.3907177567251715,
"learning_rate": 7.24468509225281e-05,
"loss": 0.5689,
"step": 106
},
{
"epoch": 1.3923639317627945,
"grad_norm": 0.510204926248653,
"learning_rate": 7.223060422643914e-05,
"loss": 0.5685,
"step": 107
},
{
"epoch": 1.40536149471974,
"grad_norm": 0.4592923316364611,
"learning_rate": 7.201163788091536e-05,
"loss": 0.5669,
"step": 108
},
{
"epoch": 1.4183590576766856,
"grad_norm": 0.6513129221318326,
"learning_rate": 7.178997036254799e-05,
"loss": 0.5681,
"step": 109
},
{
"epoch": 1.4313566206336312,
"grad_norm": 0.7772542110813966,
"learning_rate": 7.156562037585576e-05,
"loss": 0.5661,
"step": 110
},
{
"epoch": 1.4443541835905767,
"grad_norm": 0.6513876710701023,
"learning_rate": 7.133860685170665e-05,
"loss": 0.5736,
"step": 111
},
{
"epoch": 1.4573517465475223,
"grad_norm": 0.576315717747176,
"learning_rate": 7.110894894572056e-05,
"loss": 0.5641,
"step": 112
},
{
"epoch": 1.470349309504468,
"grad_norm": 0.5234871451279655,
"learning_rate": 7.087666603665284e-05,
"loss": 0.5606,
"step": 113
},
{
"epoch": 1.4833468724614134,
"grad_norm": 0.5433209657397478,
"learning_rate": 7.064177772475912e-05,
"loss": 0.5677,
"step": 114
},
{
"epoch": 1.496344435418359,
"grad_norm": 0.5010015862310137,
"learning_rate": 7.040430383014146e-05,
"loss": 0.5653,
"step": 115
},
{
"epoch": 1.5093419983753047,
"grad_norm": 0.48832775679905743,
"learning_rate": 7.016426439107586e-05,
"loss": 0.5644,
"step": 116
},
{
"epoch": 1.5223395613322501,
"grad_norm": 0.6320240467713527,
"learning_rate": 6.992167966232143e-05,
"loss": 0.5567,
"step": 117
},
{
"epoch": 1.5353371242891958,
"grad_norm": 0.6482838878203654,
"learning_rate": 6.967657011341126e-05,
"loss": 0.5749,
"step": 118
},
{
"epoch": 1.5483346872461414,
"grad_norm": 0.4549530011344001,
"learning_rate": 6.942895642692527e-05,
"loss": 0.561,
"step": 119
},
{
"epoch": 1.5613322502030869,
"grad_norm": 0.32795847922028915,
"learning_rate": 6.917885949674483e-05,
"loss": 0.565,
"step": 120
},
{
"epoch": 1.5743298131600325,
"grad_norm": 0.4131817811198266,
"learning_rate": 6.892630042628988e-05,
"loss": 0.5559,
"step": 121
},
{
"epoch": 1.5873273761169782,
"grad_norm": 0.34923043241455753,
"learning_rate": 6.867130052673806e-05,
"loss": 0.5627,
"step": 122
},
{
"epoch": 1.6003249390739236,
"grad_norm": 0.4012122766550806,
"learning_rate": 6.841388131522656e-05,
"loss": 0.561,
"step": 123
},
{
"epoch": 1.6133225020308692,
"grad_norm": 0.45749696954329366,
"learning_rate": 6.815406451303647e-05,
"loss": 0.5619,
"step": 124
},
{
"epoch": 1.626320064987815,
"grad_norm": 0.3572180927283862,
"learning_rate": 6.789187204375981e-05,
"loss": 0.5637,
"step": 125
},
{
"epoch": 1.6393176279447603,
"grad_norm": 0.37728781845068143,
"learning_rate": 6.762732603144978e-05,
"loss": 0.5605,
"step": 126
},
{
"epoch": 1.652315190901706,
"grad_norm": 0.4270428246291779,
"learning_rate": 6.736044879875373e-05,
"loss": 0.5536,
"step": 127
},
{
"epoch": 1.6653127538586516,
"grad_norm": 0.299705291536536,
"learning_rate": 6.709126286502965e-05,
"loss": 0.5623,
"step": 128
},
{
"epoch": 1.678310316815597,
"grad_norm": 0.36624782812777956,
"learning_rate": 6.681979094444596e-05,
"loss": 0.5569,
"step": 129
},
{
"epoch": 1.6913078797725425,
"grad_norm": 0.33387655040953185,
"learning_rate": 6.654605594406486e-05,
"loss": 0.5575,
"step": 130
},
{
"epoch": 1.7043054427294884,
"grad_norm": 0.38850086520685223,
"learning_rate": 6.627008096190938e-05,
"loss": 0.5573,
"step": 131
},
{
"epoch": 1.7173030056864338,
"grad_norm": 0.412039261196286,
"learning_rate": 6.59918892850144e-05,
"loss": 0.5598,
"step": 132
},
{
"epoch": 1.7303005686433792,
"grad_norm": 0.4639064578164096,
"learning_rate": 6.571150438746157e-05,
"loss": 0.5573,
"step": 133
},
{
"epoch": 1.743298131600325,
"grad_norm": 0.569316518837733,
"learning_rate": 6.542894992839873e-05,
"loss": 0.5525,
"step": 134
},
{
"epoch": 1.7562956945572705,
"grad_norm": 0.7768980117231334,
"learning_rate": 6.514424975004329e-05,
"loss": 0.5569,
"step": 135
},
{
"epoch": 1.769293257514216,
"grad_norm": 0.8690174213866712,
"learning_rate": 6.48574278756706e-05,
"loss": 0.5581,
"step": 136
},
{
"epoch": 1.7822908204711616,
"grad_norm": 0.7189834805856271,
"learning_rate": 6.456850850758673e-05,
"loss": 0.5523,
"step": 137
},
{
"epoch": 1.7952883834281073,
"grad_norm": 0.42461801914103414,
"learning_rate": 6.427751602508628e-05,
"loss": 0.5513,
"step": 138
},
{
"epoch": 1.8082859463850527,
"grad_norm": 0.3638089421301393,
"learning_rate": 6.398447498239527e-05,
"loss": 0.5597,
"step": 139
},
{
"epoch": 1.8212835093419983,
"grad_norm": 0.4699533771325944,
"learning_rate": 6.368941010659921e-05,
"loss": 0.5582,
"step": 140
},
{
"epoch": 1.834281072298944,
"grad_norm": 0.4419813323602383,
"learning_rate": 6.339234629555655e-05,
"loss": 0.5561,
"step": 141
},
{
"epoch": 1.8472786352558894,
"grad_norm": 0.36271241045794583,
"learning_rate": 6.309330861579786e-05,
"loss": 0.5584,
"step": 142
},
{
"epoch": 1.860276198212835,
"grad_norm": 0.40965930158743397,
"learning_rate": 6.279232230041065e-05,
"loss": 0.5536,
"step": 143
},
{
"epoch": 1.8732737611697807,
"grad_norm": 0.40198532643300333,
"learning_rate": 6.248941274691017e-05,
"loss": 0.5511,
"step": 144
},
{
"epoch": 1.8862713241267262,
"grad_norm": 0.46299430742363373,
"learning_rate": 6.218460551509636e-05,
"loss": 0.5471,
"step": 145
},
{
"epoch": 1.8992688870836718,
"grad_norm": 0.5215127675687683,
"learning_rate": 6.18779263248971e-05,
"loss": 0.5553,
"step": 146
},
{
"epoch": 1.9122664500406175,
"grad_norm": 0.45729570755510884,
"learning_rate": 6.156940105419785e-05,
"loss": 0.5491,
"step": 147
},
{
"epoch": 1.925264012997563,
"grad_norm": 0.3015515408877481,
"learning_rate": 6.125905573665824e-05,
"loss": 0.5512,
"step": 148
},
{
"epoch": 1.9382615759545085,
"grad_norm": 0.4496510050196559,
"learning_rate": 6.094691655951512e-05,
"loss": 0.5537,
"step": 149
},
{
"epoch": 1.9512591389114542,
"grad_norm": 0.5782505006225216,
"learning_rate": 6.063300986137297e-05,
"loss": 0.5519,
"step": 150
},
{
"epoch": 1.9642567018683996,
"grad_norm": 0.48811843212065303,
"learning_rate": 6.0317362129981375e-05,
"loss": 0.5519,
"step": 151
},
{
"epoch": 1.9772542648253453,
"grad_norm": 0.3524549635448169,
"learning_rate": 6.000000000000001e-05,
"loss": 0.5555,
"step": 152
},
{
"epoch": 1.990251827782291,
"grad_norm": 0.3499895937576908,
"learning_rate": 5.968095025075114e-05,
"loss": 0.5571,
"step": 153
},
{
"epoch": 2.0048740861088548,
"grad_norm": 0.354088344538403,
"learning_rate": 5.936023980395997e-05,
"loss": 0.5401,
"step": 154
},
{
"epoch": 2.0178716490658,
"grad_norm": 0.4330890417017954,
"learning_rate": 5.903789572148295e-05,
"loss": 0.5114,
"step": 155
},
{
"epoch": 2.0308692120227456,
"grad_norm": 0.45299756805326713,
"learning_rate": 5.871394520302432e-05,
"loss": 0.5027,
"step": 156
},
{
"epoch": 2.0438667749796915,
"grad_norm": 0.36185553048035746,
"learning_rate": 5.838841558384091e-05,
"loss": 0.5004,
"step": 157
},
{
"epoch": 2.056864337936637,
"grad_norm": 0.4094243535630425,
"learning_rate": 5.806133433243558e-05,
"loss": 0.5089,
"step": 158
},
{
"epoch": 2.0698619008935824,
"grad_norm": 0.4119895854921856,
"learning_rate": 5.7732729048239444e-05,
"loss": 0.5041,
"step": 159
},
{
"epoch": 2.0828594638505282,
"grad_norm": 0.49951764317825964,
"learning_rate": 5.740262745928293e-05,
"loss": 0.5083,
"step": 160
},
{
"epoch": 2.0958570268074737,
"grad_norm": 0.6492588353532962,
"learning_rate": 5.707105741985615e-05,
"loss": 0.5089,
"step": 161
},
{
"epoch": 2.108854589764419,
"grad_norm": 0.8172546815280193,
"learning_rate": 5.673804690815845e-05,
"loss": 0.5125,
"step": 162
},
{
"epoch": 2.121852152721365,
"grad_norm": 0.8291933191610376,
"learning_rate": 5.6403624023937614e-05,
"loss": 0.5042,
"step": 163
},
{
"epoch": 2.1348497156783104,
"grad_norm": 0.6400143821799831,
"learning_rate": 5.606781698611879e-05,
"loss": 0.5019,
"step": 164
},
{
"epoch": 2.147847278635256,
"grad_norm": 0.5436979640183732,
"learning_rate": 5.573065413042333e-05,
"loss": 0.5045,
"step": 165
},
{
"epoch": 2.1608448415922012,
"grad_norm": 0.4492438740170881,
"learning_rate": 5.5392163906977835e-05,
"loss": 0.5067,
"step": 166
},
{
"epoch": 2.173842404549147,
"grad_norm": 0.40317590644042245,
"learning_rate": 5.505237487791343e-05,
"loss": 0.5041,
"step": 167
},
{
"epoch": 2.1868399675060926,
"grad_norm": 0.4137259663946232,
"learning_rate": 5.471131571495574e-05,
"loss": 0.5008,
"step": 168
},
{
"epoch": 2.199837530463038,
"grad_norm": 0.39271569786675564,
"learning_rate": 5.4369015197005506e-05,
"loss": 0.5109,
"step": 169
},
{
"epoch": 2.212835093419984,
"grad_norm": 0.3609357892178772,
"learning_rate": 5.4025502207710184e-05,
"loss": 0.5056,
"step": 170
},
{
"epoch": 2.2258326563769293,
"grad_norm": 0.32273394848970144,
"learning_rate": 5.368080573302676e-05,
"loss": 0.5004,
"step": 171
},
{
"epoch": 2.2388302193338747,
"grad_norm": 0.3902918387294514,
"learning_rate": 5.333495485877583e-05,
"loss": 0.5015,
"step": 172
},
{
"epoch": 2.2518277822908206,
"grad_norm": 0.3615272576113914,
"learning_rate": 5.298797876818735e-05,
"loss": 0.501,
"step": 173
},
{
"epoch": 2.264825345247766,
"grad_norm": 0.3092837009560793,
"learning_rate": 5.263990673943811e-05,
"loss": 0.5013,
"step": 174
},
{
"epoch": 2.2778229082047114,
"grad_norm": 0.3221337815028307,
"learning_rate": 5.229076814318122e-05,
"loss": 0.5023,
"step": 175
},
{
"epoch": 2.2908204711616573,
"grad_norm": 0.31876856314125945,
"learning_rate": 5.194059244006779e-05,
"loss": 0.4954,
"step": 176
},
{
"epoch": 2.3038180341186028,
"grad_norm": 0.3471396479864812,
"learning_rate": 5.158940917826099e-05,
"loss": 0.5041,
"step": 177
},
{
"epoch": 2.316815597075548,
"grad_norm": 0.30244403952200244,
"learning_rate": 5.123724799094279e-05,
"loss": 0.4968,
"step": 178
},
{
"epoch": 2.329813160032494,
"grad_norm": 0.27081780367373653,
"learning_rate": 5.088413859381341e-05,
"loss": 0.495,
"step": 179
},
{
"epoch": 2.3428107229894395,
"grad_norm": 0.26698926546704715,
"learning_rate": 5.053011078258397e-05,
"loss": 0.4986,
"step": 180
},
{
"epoch": 2.355808285946385,
"grad_norm": 0.2434254540157736,
"learning_rate": 5.017519443046226e-05,
"loss": 0.5059,
"step": 181
},
{
"epoch": 2.368805848903331,
"grad_norm": 0.2649836381754834,
"learning_rate": 4.981941948563197e-05,
"loss": 0.4939,
"step": 182
},
{
"epoch": 2.381803411860276,
"grad_norm": 0.24587524499861643,
"learning_rate": 4.94628159687257e-05,
"loss": 0.4955,
"step": 183
},
{
"epoch": 2.3948009748172217,
"grad_norm": 0.2099952601611583,
"learning_rate": 4.9105413970291747e-05,
"loss": 0.4998,
"step": 184
},
{
"epoch": 2.4077985377741675,
"grad_norm": 0.30733517181039643,
"learning_rate": 4.874724364825504e-05,
"loss": 0.5051,
"step": 185
},
{
"epoch": 2.420796100731113,
"grad_norm": 0.23936641127353223,
"learning_rate": 4.8388335225372416e-05,
"loss": 0.5042,
"step": 186
},
{
"epoch": 2.4337936636880584,
"grad_norm": 0.24807141319926213,
"learning_rate": 4.802871898668237e-05,
"loss": 0.5015,
"step": 187
},
{
"epoch": 2.4467912266450043,
"grad_norm": 0.23764312316041677,
"learning_rate": 4.7668425276949546e-05,
"loss": 0.5047,
"step": 188
},
{
"epoch": 2.4597887896019497,
"grad_norm": 0.19839288524791437,
"learning_rate": 4.730748449810429e-05,
"loss": 0.4945,
"step": 189
},
{
"epoch": 2.472786352558895,
"grad_norm": 0.21343457304874502,
"learning_rate": 4.694592710667723e-05,
"loss": 0.4948,
"step": 190
},
{
"epoch": 2.4857839155158405,
"grad_norm": 0.23293583325724027,
"learning_rate": 4.658378361122936e-05,
"loss": 0.4986,
"step": 191
},
{
"epoch": 2.4987814784727864,
"grad_norm": 0.21561162546636445,
"learning_rate": 4.622108456977773e-05,
"loss": 0.5027,
"step": 192
},
{
"epoch": 2.511779041429732,
"grad_norm": 0.20220078631632762,
"learning_rate": 4.585786058721687e-05,
"loss": 0.4992,
"step": 193
},
{
"epoch": 2.5247766043866777,
"grad_norm": 0.19015117259585296,
"learning_rate": 4.549414231273633e-05,
"loss": 0.5019,
"step": 194
},
{
"epoch": 2.537774167343623,
"grad_norm": 0.17888404685442738,
"learning_rate": 4.512996043723453e-05,
"loss": 0.5009,
"step": 195
},
{
"epoch": 2.5507717303005686,
"grad_norm": 0.1891627608715842,
"learning_rate": 4.476534569072895e-05,
"loss": 0.5005,
"step": 196
},
{
"epoch": 2.563769293257514,
"grad_norm": 0.1960672578073265,
"learning_rate": 4.440032883976318e-05,
"loss": 0.4979,
"step": 197
},
{
"epoch": 2.57676685621446,
"grad_norm": 0.1907240431557997,
"learning_rate": 4.403494068481074e-05,
"loss": 0.4965,
"step": 198
},
{
"epoch": 2.5897644191714053,
"grad_norm": 0.21281767640684754,
"learning_rate": 4.3669212057676145e-05,
"loss": 0.5035,
"step": 199
},
{
"epoch": 2.602761982128351,
"grad_norm": 0.20034978185083457,
"learning_rate": 4.33031738188933e-05,
"loss": 0.493,
"step": 200
},
{
"epoch": 2.6157595450852966,
"grad_norm": 0.18482405886366973,
"learning_rate": 4.293685685512142e-05,
"loss": 0.4979,
"step": 201
},
{
"epoch": 2.628757108042242,
"grad_norm": 0.20689788018303523,
"learning_rate": 4.257029207653881e-05,
"loss": 0.4968,
"step": 202
},
{
"epoch": 2.6417546709991875,
"grad_norm": 0.20468709904511792,
"learning_rate": 4.220351041423462e-05,
"loss": 0.496,
"step": 203
},
{
"epoch": 2.6547522339561334,
"grad_norm": 0.19708900779722158,
"learning_rate": 4.183654281759888e-05,
"loss": 0.5059,
"step": 204
},
{
"epoch": 2.667749796913079,
"grad_norm": 0.1694897534384111,
"learning_rate": 4.1469420251710905e-05,
"loss": 0.5001,
"step": 205
},
{
"epoch": 2.680747359870024,
"grad_norm": 0.2060507236034366,
"learning_rate": 4.110217369472649e-05,
"loss": 0.5042,
"step": 206
},
{
"epoch": 2.69374492282697,
"grad_norm": 0.21981117761095914,
"learning_rate": 4.07348341352639e-05,
"loss": 0.4925,
"step": 207
},
{
"epoch": 2.7067424857839155,
"grad_norm": 0.15869661704467267,
"learning_rate": 4.0367432569789065e-05,
"loss": 0.4925,
"step": 208
},
{
"epoch": 2.719740048740861,
"grad_norm": 0.17554453102907516,
"learning_rate": 4e-05,
"loss": 0.4898,
"step": 209
},
{
"epoch": 2.732737611697807,
"grad_norm": 0.20109162996025604,
"learning_rate": 3.963256743021095e-05,
"loss": 0.4935,
"step": 210
},
{
"epoch": 2.7457351746547523,
"grad_norm": 0.16965164550202852,
"learning_rate": 3.92651658647361e-05,
"loss": 0.4926,
"step": 211
},
{
"epoch": 2.7587327376116977,
"grad_norm": 0.19107787557510944,
"learning_rate": 3.889782630527353e-05,
"loss": 0.5003,
"step": 212
},
{
"epoch": 2.7717303005686436,
"grad_norm": 0.23600064349601493,
"learning_rate": 3.853057974828911e-05,
"loss": 0.5025,
"step": 213
},
{
"epoch": 2.784727863525589,
"grad_norm": 0.1938736742926907,
"learning_rate": 3.816345718240113e-05,
"loss": 0.496,
"step": 214
},
{
"epoch": 2.7977254264825344,
"grad_norm": 0.18674334250731434,
"learning_rate": 3.779648958576538e-05,
"loss": 0.4964,
"step": 215
},
{
"epoch": 2.81072298943948,
"grad_norm": 0.17897442021815202,
"learning_rate": 3.74297079234612e-05,
"loss": 0.4993,
"step": 216
},
{
"epoch": 2.8237205523964257,
"grad_norm": 0.1454966971656852,
"learning_rate": 3.706314314487859e-05,
"loss": 0.5024,
"step": 217
},
{
"epoch": 2.836718115353371,
"grad_norm": 0.17254506937932373,
"learning_rate": 3.669682618110671e-05,
"loss": 0.496,
"step": 218
},
{
"epoch": 2.849715678310317,
"grad_norm": 0.17898500506822662,
"learning_rate": 3.6330787942323855e-05,
"loss": 0.4971,
"step": 219
},
{
"epoch": 2.8627132412672625,
"grad_norm": 0.1775497414815859,
"learning_rate": 3.5965059315189274e-05,
"loss": 0.4994,
"step": 220
},
{
"epoch": 2.875710804224208,
"grad_norm": 0.15173916947706884,
"learning_rate": 3.559967116023683e-05,
"loss": 0.501,
"step": 221
},
{
"epoch": 2.8887083671811533,
"grad_norm": 0.2019676520697475,
"learning_rate": 3.523465430927106e-05,
"loss": 0.4975,
"step": 222
},
{
"epoch": 2.901705930138099,
"grad_norm": 0.17820583073365817,
"learning_rate": 3.4870039562765475e-05,
"loss": 0.497,
"step": 223
},
{
"epoch": 2.9147034930950446,
"grad_norm": 0.17230973288348897,
"learning_rate": 3.4505857687263675e-05,
"loss": 0.4918,
"step": 224
},
{
"epoch": 2.9277010560519905,
"grad_norm": 0.17781379922544957,
"learning_rate": 3.414213941278314e-05,
"loss": 0.5009,
"step": 225
},
{
"epoch": 2.940698619008936,
"grad_norm": 0.17434301641204888,
"learning_rate": 3.377891543022229e-05,
"loss": 0.4967,
"step": 226
},
{
"epoch": 2.9536961819658814,
"grad_norm": 0.15878938609241405,
"learning_rate": 3.341621638877064e-05,
"loss": 0.5022,
"step": 227
},
{
"epoch": 2.966693744922827,
"grad_norm": 0.1877350498093743,
"learning_rate": 3.305407289332279e-05,
"loss": 0.4943,
"step": 228
},
{
"epoch": 2.9796913078797727,
"grad_norm": 0.16153562862339216,
"learning_rate": 3.269251550189573e-05,
"loss": 0.4952,
"step": 229
},
{
"epoch": 2.992688870836718,
"grad_norm": 0.1636698530309882,
"learning_rate": 3.2331574723050474e-05,
"loss": 0.5036,
"step": 230
},
{
"epoch": 3.007311129163282,
"grad_norm": 0.24083959413053807,
"learning_rate": 3.197128101331764e-05,
"loss": 0.4787,
"step": 231
},
{
"epoch": 3.0203086921202273,
"grad_norm": 0.20882743227454145,
"learning_rate": 3.161166477462759e-05,
"loss": 0.4527,
"step": 232
},
{
"epoch": 3.033306255077173,
"grad_norm": 0.2562598902235666,
"learning_rate": 3.125275635174497e-05,
"loss": 0.4566,
"step": 233
},
{
"epoch": 3.0463038180341186,
"grad_norm": 0.29932911926131295,
"learning_rate": 3.089458602970828e-05,
"loss": 0.4492,
"step": 234
},
{
"epoch": 3.059301380991064,
"grad_norm": 0.22296216539718972,
"learning_rate": 3.0537184031274306e-05,
"loss": 0.452,
"step": 235
},
{
"epoch": 3.07229894394801,
"grad_norm": 0.265839442067331,
"learning_rate": 3.0180580514368037e-05,
"loss": 0.4591,
"step": 236
},
{
"epoch": 3.0852965069049554,
"grad_norm": 0.26317399079157505,
"learning_rate": 2.9824805569537747e-05,
"loss": 0.4485,
"step": 237
},
{
"epoch": 3.098294069861901,
"grad_norm": 0.21663346321522212,
"learning_rate": 2.9469889217416045e-05,
"loss": 0.4556,
"step": 238
},
{
"epoch": 3.1112916328188467,
"grad_norm": 0.2650899931767454,
"learning_rate": 2.9115861406186593e-05,
"loss": 0.4537,
"step": 239
},
{
"epoch": 3.124289195775792,
"grad_norm": 0.2506262975062407,
"learning_rate": 2.8762752009057232e-05,
"loss": 0.4535,
"step": 240
},
{
"epoch": 3.1372867587327375,
"grad_norm": 0.22119457110883312,
"learning_rate": 2.841059082173902e-05,
"loss": 0.4502,
"step": 241
},
{
"epoch": 3.150284321689683,
"grad_norm": 0.23862545493558096,
"learning_rate": 2.805940755993223e-05,
"loss": 0.4537,
"step": 242
},
{
"epoch": 3.163281884646629,
"grad_norm": 0.18800051564766848,
"learning_rate": 2.770923185681878e-05,
"loss": 0.4534,
"step": 243
},
{
"epoch": 3.1762794476035743,
"grad_norm": 0.21073908239980146,
"learning_rate": 2.7360093260561904e-05,
"loss": 0.4537,
"step": 244
},
{
"epoch": 3.1892770105605197,
"grad_norm": 0.19564030664878582,
"learning_rate": 2.7012021231812666e-05,
"loss": 0.4533,
"step": 245
},
{
"epoch": 3.2022745735174656,
"grad_norm": 0.1689855239948926,
"learning_rate": 2.6665045141224193e-05,
"loss": 0.451,
"step": 246
},
{
"epoch": 3.215272136474411,
"grad_norm": 0.20951707110954032,
"learning_rate": 2.6319194266973256e-05,
"loss": 0.4557,
"step": 247
},
{
"epoch": 3.2282696994313564,
"grad_norm": 0.16999772257768134,
"learning_rate": 2.597449779228983e-05,
"loss": 0.4468,
"step": 248
},
{
"epoch": 3.2412672623883023,
"grad_norm": 0.19094834892704782,
"learning_rate": 2.563098480299451e-05,
"loss": 0.4554,
"step": 249
},
{
"epoch": 3.2542648253452477,
"grad_norm": 0.16246873328527864,
"learning_rate": 2.5288684285044283e-05,
"loss": 0.4549,
"step": 250
},
{
"epoch": 3.267262388302193,
"grad_norm": 0.1825793103224904,
"learning_rate": 2.4947625122086585e-05,
"loss": 0.4543,
"step": 251
},
{
"epoch": 3.280259951259139,
"grad_norm": 0.15892372670398716,
"learning_rate": 2.460783609302218e-05,
"loss": 0.4575,
"step": 252
},
{
"epoch": 3.2932575142160845,
"grad_norm": 0.17673710311378757,
"learning_rate": 2.4269345869576676e-05,
"loss": 0.452,
"step": 253
},
{
"epoch": 3.30625507717303,
"grad_norm": 0.15259667013543174,
"learning_rate": 2.393218301388123e-05,
"loss": 0.4507,
"step": 254
},
{
"epoch": 3.319252640129976,
"grad_norm": 0.15618013871442282,
"learning_rate": 2.35963759760624e-05,
"loss": 0.4483,
"step": 255
},
{
"epoch": 3.332250203086921,
"grad_norm": 0.1383646033070645,
"learning_rate": 2.3261953091841553e-05,
"loss": 0.4466,
"step": 256
},
{
"epoch": 3.3452477660438666,
"grad_norm": 0.15025307639697422,
"learning_rate": 2.2928942580143855e-05,
"loss": 0.457,
"step": 257
},
{
"epoch": 3.3582453290008125,
"grad_norm": 0.12942060560157578,
"learning_rate": 2.2597372540717083e-05,
"loss": 0.4494,
"step": 258
},
{
"epoch": 3.371242891957758,
"grad_norm": 0.13761592476330478,
"learning_rate": 2.226727095176057e-05,
"loss": 0.4577,
"step": 259
},
{
"epoch": 3.3842404549147034,
"grad_norm": 0.12368831090048717,
"learning_rate": 2.1938665667564435e-05,
"loss": 0.45,
"step": 260
},
{
"epoch": 3.3972380178716493,
"grad_norm": 0.13980977768107444,
"learning_rate": 2.1611584416159106e-05,
"loss": 0.4601,
"step": 261
},
{
"epoch": 3.4102355808285947,
"grad_norm": 0.1292981782906244,
"learning_rate": 2.1286054796975696e-05,
"loss": 0.4544,
"step": 262
},
{
"epoch": 3.42323314378554,
"grad_norm": 0.13081878013901876,
"learning_rate": 2.096210427851706e-05,
"loss": 0.4499,
"step": 263
},
{
"epoch": 3.436230706742486,
"grad_norm": 0.13234583952511578,
"learning_rate": 2.063976019604006e-05,
"loss": 0.4492,
"step": 264
},
{
"epoch": 3.4492282696994314,
"grad_norm": 0.13707720449565938,
"learning_rate": 2.0319049749248876e-05,
"loss": 0.4576,
"step": 265
},
{
"epoch": 3.462225832656377,
"grad_norm": 0.1275609686612589,
"learning_rate": 2.0000000000000012e-05,
"loss": 0.4478,
"step": 266
},
{
"epoch": 3.4752233956133223,
"grad_norm": 0.13858936276545247,
"learning_rate": 1.9682637870018638e-05,
"loss": 0.4503,
"step": 267
},
{
"epoch": 3.488220958570268,
"grad_norm": 0.13526985297489078,
"learning_rate": 1.9366990138627054e-05,
"loss": 0.451,
"step": 268
},
{
"epoch": 3.5012185215272136,
"grad_norm": 0.13143298791548755,
"learning_rate": 1.9053083440484887e-05,
"loss": 0.4469,
"step": 269
},
{
"epoch": 3.5142160844841595,
"grad_norm": 0.13651218753852146,
"learning_rate": 1.8740944263341773e-05,
"loss": 0.4538,
"step": 270
},
{
"epoch": 3.527213647441105,
"grad_norm": 0.13559937030356303,
"learning_rate": 1.8430598945802156e-05,
"loss": 0.4578,
"step": 271
},
{
"epoch": 3.5402112103980503,
"grad_norm": 0.13345629350619237,
"learning_rate": 1.8122073675102935e-05,
"loss": 0.4501,
"step": 272
},
{
"epoch": 3.5532087733549957,
"grad_norm": 0.13156084596113124,
"learning_rate": 1.781539448490365e-05,
"loss": 0.4508,
"step": 273
},
{
"epoch": 3.5662063363119416,
"grad_norm": 0.12230522417142756,
"learning_rate": 1.7510587253089842e-05,
"loss": 0.4564,
"step": 274
},
{
"epoch": 3.579203899268887,
"grad_norm": 0.12733451660742465,
"learning_rate": 1.7207677699589355e-05,
"loss": 0.4562,
"step": 275
},
{
"epoch": 3.592201462225833,
"grad_norm": 0.13918819865169338,
"learning_rate": 1.690669138420215e-05,
"loss": 0.4539,
"step": 276
},
{
"epoch": 3.6051990251827783,
"grad_norm": 0.12910395860504983,
"learning_rate": 1.6607653704443457e-05,
"loss": 0.4549,
"step": 277
},
{
"epoch": 3.618196588139724,
"grad_norm": 0.11652161500078773,
"learning_rate": 1.6310589893400804e-05,
"loss": 0.4495,
"step": 278
},
{
"epoch": 3.631194151096669,
"grad_norm": 0.1435144989657529,
"learning_rate": 1.601552501760473e-05,
"loss": 0.4615,
"step": 279
},
{
"epoch": 3.644191714053615,
"grad_norm": 0.11768536430982104,
"learning_rate": 1.5722483974913737e-05,
"loss": 0.452,
"step": 280
},
{
"epoch": 3.6571892770105605,
"grad_norm": 0.12124441944302719,
"learning_rate": 1.5431491492413288e-05,
"loss": 0.4509,
"step": 281
},
{
"epoch": 3.670186839967506,
"grad_norm": 0.11219311820617218,
"learning_rate": 1.5142572124329418e-05,
"loss": 0.4536,
"step": 282
},
{
"epoch": 3.683184402924452,
"grad_norm": 0.11088572665376577,
"learning_rate": 1.4855750249956718e-05,
"loss": 0.4533,
"step": 283
},
{
"epoch": 3.6961819658813972,
"grad_norm": 0.1114181991679099,
"learning_rate": 1.457105007160129e-05,
"loss": 0.4583,
"step": 284
},
{
"epoch": 3.7091795288383427,
"grad_norm": 0.1150092625579184,
"learning_rate": 1.4288495612538427e-05,
"loss": 0.4546,
"step": 285
},
{
"epoch": 3.722177091795288,
"grad_norm": 0.11256641012147124,
"learning_rate": 1.4008110714985623e-05,
"loss": 0.4562,
"step": 286
},
{
"epoch": 3.735174654752234,
"grad_norm": 0.1121146716270174,
"learning_rate": 1.3729919038090627e-05,
"loss": 0.4531,
"step": 287
},
{
"epoch": 3.7481722177091794,
"grad_norm": 0.12240062452155943,
"learning_rate": 1.3453944055935151e-05,
"loss": 0.4554,
"step": 288
},
{
"epoch": 3.7611697806661253,
"grad_norm": 0.10662890450993669,
"learning_rate": 1.3180209055554043e-05,
"loss": 0.4557,
"step": 289
},
{
"epoch": 3.7741673436230707,
"grad_norm": 0.11472028819426605,
"learning_rate": 1.2908737134970367e-05,
"loss": 0.4568,
"step": 290
},
{
"epoch": 3.787164906580016,
"grad_norm": 0.11298609292901812,
"learning_rate": 1.2639551201246278e-05,
"loss": 0.4514,
"step": 291
},
{
"epoch": 3.8001624695369616,
"grad_norm": 0.10976020682705631,
"learning_rate": 1.2372673968550229e-05,
"loss": 0.4476,
"step": 292
},
{
"epoch": 3.8131600324939074,
"grad_norm": 0.10531475963763727,
"learning_rate": 1.2108127956240186e-05,
"loss": 0.4471,
"step": 293
},
{
"epoch": 3.826157595450853,
"grad_norm": 0.10623689348961035,
"learning_rate": 1.1845935486963546e-05,
"loss": 0.4515,
"step": 294
},
{
"epoch": 3.8391551584077988,
"grad_norm": 0.10834285056205656,
"learning_rate": 1.158611868477344e-05,
"loss": 0.4484,
"step": 295
},
{
"epoch": 3.852152721364744,
"grad_norm": 0.10958894542102338,
"learning_rate": 1.1328699473261957e-05,
"loss": 0.446,
"step": 296
},
{
"epoch": 3.8651502843216896,
"grad_norm": 0.10719668132712344,
"learning_rate": 1.107369957371013e-05,
"loss": 0.4587,
"step": 297
},
{
"epoch": 3.878147847278635,
"grad_norm": 0.10865202402459116,
"learning_rate": 1.0821140503255174e-05,
"loss": 0.4478,
"step": 298
},
{
"epoch": 3.891145410235581,
"grad_norm": 0.1075674174283554,
"learning_rate": 1.0571043573074737e-05,
"loss": 0.4474,
"step": 299
},
{
"epoch": 3.9041429731925263,
"grad_norm": 0.10420773620004607,
"learning_rate": 1.0323429886588743e-05,
"loss": 0.4558,
"step": 300
},
{
"epoch": 3.917140536149472,
"grad_norm": 0.10602592166217871,
"learning_rate": 1.0078320337678584e-05,
"loss": 0.4472,
"step": 301
},
{
"epoch": 3.9301380991064176,
"grad_norm": 0.11214585082716343,
"learning_rate": 9.835735608924155e-06,
"loss": 0.4532,
"step": 302
},
{
"epoch": 3.943135662063363,
"grad_norm": 0.10680829021211037,
"learning_rate": 9.595696169858542e-06,
"loss": 0.452,
"step": 303
},
{
"epoch": 3.9561332250203085,
"grad_norm": 0.10193849061969358,
"learning_rate": 9.358222275240884e-06,
"loss": 0.4548,
"step": 304
},
{
"epoch": 3.9691307879772544,
"grad_norm": 0.10621117615673921,
"learning_rate": 9.123333963347166e-06,
"loss": 0.4568,
"step": 305
},
{
"epoch": 3.9821283509342,
"grad_norm": 0.10611037774239646,
"learning_rate": 8.89105105427945e-06,
"loss": 0.4503,
"step": 306
},
{
"epoch": 3.9951259138911457,
"grad_norm": 0.10224681565036052,
"learning_rate": 8.661393148293355e-06,
"loss": 0.4514,
"step": 307
},
{
"epoch": 4.0097481722177095,
"grad_norm": 0.1629992310965872,
"learning_rate": 8.434379624144261e-06,
"loss": 0.4384,
"step": 308
},
{
"epoch": 4.022745735174655,
"grad_norm": 0.16305297766108903,
"learning_rate": 8.210029637452016e-06,
"loss": 0.4219,
"step": 309
},
{
"epoch": 4.0357432981316,
"grad_norm": 0.1152257478387255,
"learning_rate": 7.988362119084642e-06,
"loss": 0.423,
"step": 310
},
{
"epoch": 4.048740861088546,
"grad_norm": 0.12619317639889938,
"learning_rate": 7.769395773560874e-06,
"loss": 0.4265,
"step": 311
},
{
"epoch": 4.061738424045491,
"grad_norm": 0.1605893684483608,
"learning_rate": 7.553149077471915e-06,
"loss": 0.4293,
"step": 312
},
{
"epoch": 4.074735987002437,
"grad_norm": 0.16787469942852876,
"learning_rate": 7.3396402779222845e-06,
"loss": 0.4269,
"step": 313
},
{
"epoch": 4.087733549959383,
"grad_norm": 0.13639739623886823,
"learning_rate": 7.128887390990198e-06,
"loss": 0.4203,
"step": 314
},
{
"epoch": 4.100731112916328,
"grad_norm": 0.12896797685517175,
"learning_rate": 6.9209082002072725e-06,
"loss": 0.4237,
"step": 315
},
{
"epoch": 4.113728675873274,
"grad_norm": 0.13126789348210258,
"learning_rate": 6.715720255058e-06,
"loss": 0.4323,
"step": 316
},
{
"epoch": 4.126726238830219,
"grad_norm": 0.13882914474015806,
"learning_rate": 6.513340869498859e-06,
"loss": 0.4232,
"step": 317
},
{
"epoch": 4.139723801787165,
"grad_norm": 0.13742788716133153,
"learning_rate": 6.313787120497376e-06,
"loss": 0.427,
"step": 318
},
{
"epoch": 4.15272136474411,
"grad_norm": 0.12658794759546255,
"learning_rate": 6.117075846591123e-06,
"loss": 0.422,
"step": 319
},
{
"epoch": 4.1657189277010565,
"grad_norm": 0.11656720107903619,
"learning_rate": 5.923223646466923e-06,
"loss": 0.424,
"step": 320
},
{
"epoch": 4.178716490658002,
"grad_norm": 0.12017679947679329,
"learning_rate": 5.732246877560146e-06,
"loss": 0.4239,
"step": 321
},
{
"epoch": 4.191714053614947,
"grad_norm": 0.1248953898447142,
"learning_rate": 5.5441616546745646e-06,
"loss": 0.4247,
"step": 322
},
{
"epoch": 4.204711616571893,
"grad_norm": 0.11492392711495635,
"learning_rate": 5.358983848622452e-06,
"loss": 0.423,
"step": 323
},
{
"epoch": 4.217709179528838,
"grad_norm": 0.10979437544809782,
"learning_rate": 5.176729084885508e-06,
"loss": 0.426,
"step": 324
},
{
"epoch": 4.230706742485784,
"grad_norm": 0.10996774248170614,
"learning_rate": 4.99741274229625e-06,
"loss": 0.424,
"step": 325
},
{
"epoch": 4.24370430544273,
"grad_norm": 0.11864930806086794,
"learning_rate": 4.821049951740442e-06,
"loss": 0.4236,
"step": 326
},
{
"epoch": 4.256701868399675,
"grad_norm": 0.11614244217733552,
"learning_rate": 4.647655594880225e-06,
"loss": 0.4264,
"step": 327
},
{
"epoch": 4.269699431356621,
"grad_norm": 0.10489433778288049,
"learning_rate": 4.4772443028985004e-06,
"loss": 0.4306,
"step": 328
},
{
"epoch": 4.282696994313566,
"grad_norm": 0.1045337845950868,
"learning_rate": 4.3098304552642385e-06,
"loss": 0.4279,
"step": 329
},
{
"epoch": 4.295694557270512,
"grad_norm": 0.10425269207053385,
"learning_rate": 4.1454281785191995e-06,
"loss": 0.4219,
"step": 330
},
{
"epoch": 4.308692120227457,
"grad_norm": 0.10799333991055605,
"learning_rate": 3.984051345085855e-06,
"loss": 0.4239,
"step": 331
},
{
"epoch": 4.3216896831844025,
"grad_norm": 0.11546875761746016,
"learning_rate": 3.825713572096903e-06,
"loss": 0.4188,
"step": 332
},
{
"epoch": 4.334687246141349,
"grad_norm": 0.10158288013636599,
"learning_rate": 3.6704282202461515e-06,
"loss": 0.4243,
"step": 333
},
{
"epoch": 4.347684809098294,
"grad_norm": 0.09998970634420433,
"learning_rate": 3.518208392661184e-06,
"loss": 0.4221,
"step": 334
},
{
"epoch": 4.36068237205524,
"grad_norm": 0.1072651386764742,
"learning_rate": 3.3690669337977e-06,
"loss": 0.4204,
"step": 335
},
{
"epoch": 4.373679935012185,
"grad_norm": 0.10189852488748347,
"learning_rate": 3.2230164283556918e-06,
"loss": 0.4218,
"step": 336
},
{
"epoch": 4.3866774979691305,
"grad_norm": 0.09855467315288223,
"learning_rate": 3.080069200217497e-06,
"loss": 0.424,
"step": 337
},
{
"epoch": 4.399675060926076,
"grad_norm": 0.09678471475059303,
"learning_rate": 2.9402373114079295e-06,
"loss": 0.4237,
"step": 338
},
{
"epoch": 4.412672623883022,
"grad_norm": 0.09641330522520603,
"learning_rate": 2.803532561076492e-06,
"loss": 0.4294,
"step": 339
},
{
"epoch": 4.425670186839968,
"grad_norm": 0.09993981323574717,
"learning_rate": 2.669966484501716e-06,
"loss": 0.4291,
"step": 340
},
{
"epoch": 4.438667749796913,
"grad_norm": 0.09491135655205166,
"learning_rate": 2.5395503521178143e-06,
"loss": 0.425,
"step": 341
},
{
"epoch": 4.451665312753859,
"grad_norm": 0.10027652165084279,
"learning_rate": 2.4122951685636674e-06,
"loss": 0.4287,
"step": 342
},
{
"epoch": 4.464662875710804,
"grad_norm": 0.09768195784704334,
"learning_rate": 2.2882116717542634e-06,
"loss": 0.4297,
"step": 343
},
{
"epoch": 4.477660438667749,
"grad_norm": 0.09320352715058745,
"learning_rate": 2.1673103319746146e-06,
"loss": 0.422,
"step": 344
},
{
"epoch": 4.490658001624696,
"grad_norm": 0.09297954183991879,
"learning_rate": 2.049601350996233e-06,
"loss": 0.4216,
"step": 345
},
{
"epoch": 4.503655564581641,
"grad_norm": 0.09580363639852763,
"learning_rate": 1.93509466121633e-06,
"loss": 0.4221,
"step": 346
},
{
"epoch": 4.516653127538587,
"grad_norm": 0.09572555821764618,
"learning_rate": 1.8237999248197002e-06,
"loss": 0.4261,
"step": 347
},
{
"epoch": 4.529650690495532,
"grad_norm": 0.09729341717065641,
"learning_rate": 1.7157265329634354e-06,
"loss": 0.422,
"step": 348
},
{
"epoch": 4.5426482534524775,
"grad_norm": 0.09262430982141642,
"learning_rate": 1.6108836049844434e-06,
"loss": 0.4275,
"step": 349
},
{
"epoch": 4.555645816409423,
"grad_norm": 0.09643151222343373,
"learning_rate": 1.5092799876299835e-06,
"loss": 0.4318,
"step": 350
},
{
"epoch": 4.568643379366369,
"grad_norm": 0.09552392662018869,
"learning_rate": 1.4109242543111834e-06,
"loss": 0.4262,
"step": 351
},
{
"epoch": 4.581640942323315,
"grad_norm": 0.0934634073485968,
"learning_rate": 1.3158247043795735e-06,
"loss": 0.4191,
"step": 352
},
{
"epoch": 4.59463850528026,
"grad_norm": 0.09051362955312056,
"learning_rate": 1.2239893624267852e-06,
"loss": 0.4249,
"step": 353
},
{
"epoch": 4.6076360682372055,
"grad_norm": 0.10311548227178086,
"learning_rate": 1.1354259776074472e-06,
"loss": 0.421,
"step": 354
},
{
"epoch": 4.620633631194151,
"grad_norm": 0.09034380750409249,
"learning_rate": 1.050142022985292e-06,
"loss": 0.4234,
"step": 355
},
{
"epoch": 4.633631194151096,
"grad_norm": 0.0912218320639531,
"learning_rate": 9.681446949025752e-07,
"loss": 0.4231,
"step": 356
},
{
"epoch": 4.646628757108042,
"grad_norm": 0.0907014935264679,
"learning_rate": 8.89440912372832e-07,
"loss": 0.4162,
"step": 357
},
{
"epoch": 4.659626320064988,
"grad_norm": 0.08963535898951595,
"learning_rate": 8.140373164970428e-07,
"loss": 0.4286,
"step": 358
},
{
"epoch": 4.6726238830219335,
"grad_norm": 0.09182311609539749,
"learning_rate": 7.419402699032852e-07,
"loss": 0.4213,
"step": 359
},
{
"epoch": 4.685621445978879,
"grad_norm": 0.08836898405911321,
"learning_rate": 6.731558562097995e-07,
"loss": 0.4247,
"step": 360
},
{
"epoch": 4.698619008935824,
"grad_norm": 0.09028610838105593,
"learning_rate": 6.076898795116792e-07,
"loss": 0.4253,
"step": 361
},
{
"epoch": 4.71161657189277,
"grad_norm": 0.09100823747148952,
"learning_rate": 5.455478638911071e-07,
"loss": 0.4251,
"step": 362
},
{
"epoch": 4.724614134849716,
"grad_norm": 0.09029025365560213,
"learning_rate": 4.867350529512261e-07,
"loss": 0.4187,
"step": 363
},
{
"epoch": 4.737611697806662,
"grad_norm": 0.08743205850314692,
"learning_rate": 4.3125640937368373e-07,
"loss": 0.4254,
"step": 364
},
{
"epoch": 4.750609260763607,
"grad_norm": 0.0922555367394928,
"learning_rate": 3.791166144998704e-07,
"loss": 0.423,
"step": 365
},
{
"epoch": 4.763606823720552,
"grad_norm": 0.08845599991505329,
"learning_rate": 3.3032006793590977e-07,
"loss": 0.4224,
"step": 366
},
{
"epoch": 4.776604386677498,
"grad_norm": 0.0915393101514765,
"learning_rate": 2.848708871814054e-07,
"loss": 0.4255,
"step": 367
},
{
"epoch": 4.789601949634443,
"grad_norm": 0.0888246620842322,
"learning_rate": 2.4277290728202063e-07,
"loss": 0.4202,
"step": 368
},
{
"epoch": 4.802599512591389,
"grad_norm": 0.0912092493664706,
"learning_rate": 2.040296805058528e-07,
"loss": 0.4316,
"step": 369
},
{
"epoch": 4.815597075548335,
"grad_norm": 0.09110426289157889,
"learning_rate": 1.6864447604370004e-07,
"loss": 0.43,
"step": 370
},
{
"epoch": 4.8285946385052805,
"grad_norm": 0.08633872492794921,
"learning_rate": 1.3662027973320614e-07,
"loss": 0.4311,
"step": 371
},
{
"epoch": 4.841592201462226,
"grad_norm": 0.08888780145070413,
"learning_rate": 1.0795979380690657e-07,
"loss": 0.4182,
"step": 372
},
{
"epoch": 4.854589764419171,
"grad_norm": 0.08984098123092447,
"learning_rate": 8.266543666421544e-08,
"loss": 0.4172,
"step": 373
},
{
"epoch": 4.867587327376117,
"grad_norm": 0.08852973851348196,
"learning_rate": 6.073934266735303e-08,
"loss": 0.4234,
"step": 374
},
{
"epoch": 4.880584890333062,
"grad_norm": 0.08776688037991925,
"learning_rate": 4.218336196125439e-08,
"loss": 0.4268,
"step": 375
},
{
"epoch": 4.8935824532900085,
"grad_norm": 0.08708562125303575,
"learning_rate": 2.699906031745414e-08,
"loss": 0.4234,
"step": 376
},
{
"epoch": 4.906580016246954,
"grad_norm": 0.08866044810066966,
"learning_rate": 1.5187719001943378e-08,
"loss": 0.4217,
"step": 377
},
{
"epoch": 4.919577579203899,
"grad_norm": 0.08852375482148439,
"learning_rate": 6.750334667091629e-09,
"loss": 0.4235,
"step": 378
},
{
"epoch": 4.932575142160845,
"grad_norm": 0.08832697038072729,
"learning_rate": 1.6876192675052695e-09,
"loss": 0.4274,
"step": 379
},
{
"epoch": 4.94557270511779,
"grad_norm": 0.08596221358605577,
"learning_rate": 0.0,
"loss": 0.4228,
"step": 380
},
{
"epoch": 4.94557270511779,
"step": 380,
"total_flos": 9.778530867524665e+18,
"train_loss": 0.5321214740997866,
"train_runtime": 38508.3509,
"train_samples_per_second": 5.112,
"train_steps_per_second": 0.01
}
],
"logging_steps": 1,
"max_steps": 380,
"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": 9.778530867524665e+18,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}