Gemma-3-Earthen-v0.2-4B-QLoRA / trainer_state.json
xzuyn's picture
Upload Step 160/2504
3ba332e verified
{
"best_global_step": null,
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.06388500698742264,
"eval_steps": 10,
"global_step": 160,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0003992812936713915,
"grad_norm": 11.861096382141113,
"learning_rate": 2.4999051176541087e-07,
"loss": 3.5277,
"step": 1
},
{
"epoch": 0.0003992812936713915,
"eval_loss": 3.8839635848999023,
"eval_runtime": 287.9309,
"eval_samples_per_second": 0.889,
"eval_steps_per_second": 0.222,
"step": 1
},
{
"epoch": 0.000798562587342783,
"grad_norm": 11.42729377746582,
"learning_rate": 2.4998101670529935e-07,
"loss": 3.4112,
"step": 2
},
{
"epoch": 0.0011978438810141745,
"grad_norm": 9.308838844299316,
"learning_rate": 2.499715148122976e-07,
"loss": 3.8903,
"step": 3
},
{
"epoch": 0.001597125174685566,
"grad_norm": 5.668716907501221,
"learning_rate": 2.499620060790273e-07,
"loss": 3.3354,
"step": 4
},
{
"epoch": 0.0019964064683569574,
"grad_norm": 7.672369003295898,
"learning_rate": 2.499524904980996e-07,
"loss": 3.6227,
"step": 5
},
{
"epoch": 0.002395687762028349,
"grad_norm": 4.239442825317383,
"learning_rate": 2.4994296806211477e-07,
"loss": 3.105,
"step": 6
},
{
"epoch": 0.0027949690556997404,
"grad_norm": 3.5887157917022705,
"learning_rate": 2.4993343876366256e-07,
"loss": 2.9517,
"step": 7
},
{
"epoch": 0.003194250349371132,
"grad_norm": 3.6663546562194824,
"learning_rate": 2.4992390259532204e-07,
"loss": 2.9232,
"step": 8
},
{
"epoch": 0.0035935316430425234,
"grad_norm": 1.8050780296325684,
"learning_rate": 2.4991435954966147e-07,
"loss": 2.7544,
"step": 9
},
{
"epoch": 0.003992812936713915,
"grad_norm": 1.7229609489440918,
"learning_rate": 2.499048096192385e-07,
"loss": 2.5745,
"step": 10
},
{
"epoch": 0.003992812936713915,
"eval_loss": 2.8815834522247314,
"eval_runtime": 290.9736,
"eval_samples_per_second": 0.88,
"eval_steps_per_second": 0.22,
"step": 10
},
{
"epoch": 0.004392094230385306,
"grad_norm": 2.123410701751709,
"learning_rate": 2.4989525279659993e-07,
"loss": 3.0969,
"step": 11
},
{
"epoch": 0.004791375524056698,
"grad_norm": 1.229800820350647,
"learning_rate": 2.4988568907428204e-07,
"loss": 2.8336,
"step": 12
},
{
"epoch": 0.005190656817728089,
"grad_norm": 0.9092569351196289,
"learning_rate": 2.4987611844481e-07,
"loss": 2.7336,
"step": 13
},
{
"epoch": 0.005589938111399481,
"grad_norm": 1.8590848445892334,
"learning_rate": 2.498665409006984e-07,
"loss": 2.8651,
"step": 14
},
{
"epoch": 0.005989219405070872,
"grad_norm": 1.1895430088043213,
"learning_rate": 2.4985695643445094e-07,
"loss": 2.5926,
"step": 15
},
{
"epoch": 0.006388500698742264,
"grad_norm": 0.8829777836799622,
"learning_rate": 2.4984736503856043e-07,
"loss": 2.5937,
"step": 16
},
{
"epoch": 0.006787781992413655,
"grad_norm": 0.7874141335487366,
"learning_rate": 2.498377667055089e-07,
"loss": 2.7383,
"step": 17
},
{
"epoch": 0.007187063286085047,
"grad_norm": 0.8703321814537048,
"learning_rate": 2.498281614277675e-07,
"loss": 2.7594,
"step": 18
},
{
"epoch": 0.007586344579756438,
"grad_norm": 1.0839616060256958,
"learning_rate": 2.4981854919779646e-07,
"loss": 2.5339,
"step": 19
},
{
"epoch": 0.00798562587342783,
"grad_norm": 0.9549999237060547,
"learning_rate": 2.4980893000804503e-07,
"loss": 2.8402,
"step": 20
},
{
"epoch": 0.00798562587342783,
"eval_loss": 2.659167766571045,
"eval_runtime": 290.5049,
"eval_samples_per_second": 0.881,
"eval_steps_per_second": 0.22,
"step": 20
},
{
"epoch": 0.008384907167099222,
"grad_norm": 1.1994329690933228,
"learning_rate": 2.497993038509517e-07,
"loss": 2.7734,
"step": 21
},
{
"epoch": 0.008784188460770613,
"grad_norm": 1.2531729936599731,
"learning_rate": 2.497896707189437e-07,
"loss": 2.4779,
"step": 22
},
{
"epoch": 0.009183469754442005,
"grad_norm": 0.8631401658058167,
"learning_rate": 2.4978003060443764e-07,
"loss": 2.7183,
"step": 23
},
{
"epoch": 0.009582751048113396,
"grad_norm": 0.6393347978591919,
"learning_rate": 2.4977038349983886e-07,
"loss": 2.5175,
"step": 24
},
{
"epoch": 0.009982032341784788,
"grad_norm": 0.7909420728683472,
"learning_rate": 2.497607293975418e-07,
"loss": 2.7034,
"step": 25
},
{
"epoch": 0.010381313635456179,
"grad_norm": 0.8058650493621826,
"learning_rate": 2.4975106828992983e-07,
"loss": 2.688,
"step": 26
},
{
"epoch": 0.010780594929127571,
"grad_norm": 0.7194178700447083,
"learning_rate": 2.497414001693753e-07,
"loss": 2.415,
"step": 27
},
{
"epoch": 0.011179876222798962,
"grad_norm": 0.7787066698074341,
"learning_rate": 2.4973172502823945e-07,
"loss": 2.4489,
"step": 28
},
{
"epoch": 0.011579157516470354,
"grad_norm": 0.7317251563072205,
"learning_rate": 2.497220428588724e-07,
"loss": 2.6261,
"step": 29
},
{
"epoch": 0.011978438810141745,
"grad_norm": 0.643670916557312,
"learning_rate": 2.4971235365361323e-07,
"loss": 2.3531,
"step": 30
},
{
"epoch": 0.011978438810141745,
"eval_loss": 2.5479750633239746,
"eval_runtime": 288.2263,
"eval_samples_per_second": 0.888,
"eval_steps_per_second": 0.222,
"step": 30
},
{
"epoch": 0.012377720103813137,
"grad_norm": 1.17843496799469,
"learning_rate": 2.4970265740478976e-07,
"loss": 2.6157,
"step": 31
},
{
"epoch": 0.012777001397484528,
"grad_norm": 0.7883862257003784,
"learning_rate": 2.496929541047188e-07,
"loss": 2.5571,
"step": 32
},
{
"epoch": 0.01317628269115592,
"grad_norm": 0.7334747314453125,
"learning_rate": 2.4968324374570584e-07,
"loss": 2.6712,
"step": 33
},
{
"epoch": 0.01357556398482731,
"grad_norm": 0.8014112114906311,
"learning_rate": 2.496735263200453e-07,
"loss": 2.774,
"step": 34
},
{
"epoch": 0.013974845278498703,
"grad_norm": 0.6167514324188232,
"learning_rate": 2.496638018200202e-07,
"loss": 2.6011,
"step": 35
},
{
"epoch": 0.014374126572170094,
"grad_norm": 0.5952684879302979,
"learning_rate": 2.4965407023790255e-07,
"loss": 2.325,
"step": 36
},
{
"epoch": 0.014773407865841486,
"grad_norm": 0.56275874376297,
"learning_rate": 2.4964433156595294e-07,
"loss": 2.4661,
"step": 37
},
{
"epoch": 0.015172689159512876,
"grad_norm": 0.6017836332321167,
"learning_rate": 2.496345857964207e-07,
"loss": 2.4421,
"step": 38
},
{
"epoch": 0.015571970453184269,
"grad_norm": 0.5634634494781494,
"learning_rate": 2.4962483292154404e-07,
"loss": 2.7153,
"step": 39
},
{
"epoch": 0.01597125174685566,
"grad_norm": 0.6215035915374756,
"learning_rate": 2.496150729335494e-07,
"loss": 2.325,
"step": 40
},
{
"epoch": 0.01597125174685566,
"eval_loss": 2.4711108207702637,
"eval_runtime": 289.8366,
"eval_samples_per_second": 0.883,
"eval_steps_per_second": 0.221,
"step": 40
},
{
"epoch": 0.01637053304052705,
"grad_norm": 0.5762554407119751,
"learning_rate": 2.4960530582465247e-07,
"loss": 2.5452,
"step": 41
},
{
"epoch": 0.016769814334198444,
"grad_norm": 0.5260322690010071,
"learning_rate": 2.49595531587057e-07,
"loss": 2.5862,
"step": 42
},
{
"epoch": 0.017169095627869835,
"grad_norm": 0.5681716799736023,
"learning_rate": 2.495857502129558e-07,
"loss": 2.7202,
"step": 43
},
{
"epoch": 0.017568376921541225,
"grad_norm": 0.6019343733787537,
"learning_rate": 2.4957596169453007e-07,
"loss": 2.5932,
"step": 44
},
{
"epoch": 0.017967658215212616,
"grad_norm": 0.5380938053131104,
"learning_rate": 2.495661660239497e-07,
"loss": 2.5838,
"step": 45
},
{
"epoch": 0.01836693950888401,
"grad_norm": 0.5226567983627319,
"learning_rate": 2.4955636319337287e-07,
"loss": 2.6499,
"step": 46
},
{
"epoch": 0.0187662208025554,
"grad_norm": 0.5076204538345337,
"learning_rate": 2.495465531949466e-07,
"loss": 2.4434,
"step": 47
},
{
"epoch": 0.01916550209622679,
"grad_norm": 0.8190023303031921,
"learning_rate": 2.495367360208062e-07,
"loss": 2.4931,
"step": 48
},
{
"epoch": 0.019564783389898182,
"grad_norm": 0.6802453994750977,
"learning_rate": 2.495269116630757e-07,
"loss": 2.5744,
"step": 49
},
{
"epoch": 0.019964064683569576,
"grad_norm": 0.4862814247608185,
"learning_rate": 2.495170801138674e-07,
"loss": 2.2362,
"step": 50
},
{
"epoch": 0.019964064683569576,
"eval_loss": 2.414149522781372,
"eval_runtime": 289.0292,
"eval_samples_per_second": 0.886,
"eval_steps_per_second": 0.221,
"step": 50
},
{
"epoch": 0.020363345977240967,
"grad_norm": 0.5404508113861084,
"learning_rate": 2.495072413652821e-07,
"loss": 2.75,
"step": 51
},
{
"epoch": 0.020762627270912357,
"grad_norm": 0.4755939543247223,
"learning_rate": 2.4949739540940906e-07,
"loss": 2.2318,
"step": 52
},
{
"epoch": 0.021161908564583748,
"grad_norm": 0.7830485701560974,
"learning_rate": 2.494875422383259e-07,
"loss": 2.6674,
"step": 53
},
{
"epoch": 0.021561189858255142,
"grad_norm": 0.5113694667816162,
"learning_rate": 2.494776818440987e-07,
"loss": 2.5452,
"step": 54
},
{
"epoch": 0.021960471151926533,
"grad_norm": 0.597758412361145,
"learning_rate": 2.494678142187818e-07,
"loss": 2.116,
"step": 55
},
{
"epoch": 0.022359752445597923,
"grad_norm": 0.5865931510925293,
"learning_rate": 2.49457939354418e-07,
"loss": 2.3269,
"step": 56
},
{
"epoch": 0.022759033739269314,
"grad_norm": 0.6509243845939636,
"learning_rate": 2.4944805724303826e-07,
"loss": 2.4119,
"step": 57
},
{
"epoch": 0.023158315032940708,
"grad_norm": 0.6174994111061096,
"learning_rate": 2.4943816787666206e-07,
"loss": 2.3669,
"step": 58
},
{
"epoch": 0.0235575963266121,
"grad_norm": 0.5243375301361084,
"learning_rate": 2.494282712472969e-07,
"loss": 2.4973,
"step": 59
},
{
"epoch": 0.02395687762028349,
"grad_norm": 0.5543302893638611,
"learning_rate": 2.494183673469388e-07,
"loss": 2.471,
"step": 60
},
{
"epoch": 0.02395687762028349,
"eval_loss": 2.3702573776245117,
"eval_runtime": 289.9081,
"eval_samples_per_second": 0.883,
"eval_steps_per_second": 0.221,
"step": 60
},
{
"epoch": 0.02435615891395488,
"grad_norm": 0.5695878863334656,
"learning_rate": 2.494084561675717e-07,
"loss": 2.5227,
"step": 61
},
{
"epoch": 0.024755440207626274,
"grad_norm": 0.48718804121017456,
"learning_rate": 2.493985377011682e-07,
"loss": 2.4296,
"step": 62
},
{
"epoch": 0.025154721501297665,
"grad_norm": 0.5036266446113586,
"learning_rate": 2.4938861193968864e-07,
"loss": 2.4357,
"step": 63
},
{
"epoch": 0.025554002794969055,
"grad_norm": 0.47645705938339233,
"learning_rate": 2.4937867887508175e-07,
"loss": 2.1584,
"step": 64
},
{
"epoch": 0.025953284088640446,
"grad_norm": 0.4842926263809204,
"learning_rate": 2.493687384992844e-07,
"loss": 2.3637,
"step": 65
},
{
"epoch": 0.02635256538231184,
"grad_norm": 0.6649777889251709,
"learning_rate": 2.493587908042216e-07,
"loss": 2.3034,
"step": 66
},
{
"epoch": 0.02675184667598323,
"grad_norm": 0.49681320786476135,
"learning_rate": 2.493488357818062e-07,
"loss": 2.4476,
"step": 67
},
{
"epoch": 0.02715112796965462,
"grad_norm": 0.514624297618866,
"learning_rate": 2.4933887342393975e-07,
"loss": 2.457,
"step": 68
},
{
"epoch": 0.02755040926332601,
"grad_norm": 0.662382185459137,
"learning_rate": 2.493289037225111e-07,
"loss": 2.646,
"step": 69
},
{
"epoch": 0.027949690556997406,
"grad_norm": 0.4702717661857605,
"learning_rate": 2.493189266693978e-07,
"loss": 2.3054,
"step": 70
},
{
"epoch": 0.027949690556997406,
"eval_loss": 2.331839084625244,
"eval_runtime": 290.0915,
"eval_samples_per_second": 0.882,
"eval_steps_per_second": 0.221,
"step": 70
},
{
"epoch": 0.028348971850668796,
"grad_norm": 0.5980565547943115,
"learning_rate": 2.493089422564649e-07,
"loss": 2.1874,
"step": 71
},
{
"epoch": 0.028748253144340187,
"grad_norm": 0.5449456572532654,
"learning_rate": 2.4929895047556574e-07,
"loss": 2.3876,
"step": 72
},
{
"epoch": 0.029147534438011578,
"grad_norm": 0.4739711582660675,
"learning_rate": 2.492889513185416e-07,
"loss": 2.3522,
"step": 73
},
{
"epoch": 0.02954681573168297,
"grad_norm": 0.5783733129501343,
"learning_rate": 2.492789447772216e-07,
"loss": 2.4153,
"step": 74
},
{
"epoch": 0.029946097025354362,
"grad_norm": 0.429364949464798,
"learning_rate": 2.4926893084342294e-07,
"loss": 2.3046,
"step": 75
},
{
"epoch": 0.030345378319025753,
"grad_norm": 0.48500293493270874,
"learning_rate": 2.4925890950895056e-07,
"loss": 2.2062,
"step": 76
},
{
"epoch": 0.030744659612697144,
"grad_norm": 0.4081466495990753,
"learning_rate": 2.492488807655974e-07,
"loss": 2.2804,
"step": 77
},
{
"epoch": 0.031143940906368538,
"grad_norm": 0.5915509462356567,
"learning_rate": 2.4923884460514423e-07,
"loss": 2.304,
"step": 78
},
{
"epoch": 0.031543222200039925,
"grad_norm": 0.5798116326332092,
"learning_rate": 2.4922880101935963e-07,
"loss": 2.3379,
"step": 79
},
{
"epoch": 0.03194250349371132,
"grad_norm": 0.49052485823631287,
"learning_rate": 2.4921874999999997e-07,
"loss": 2.3184,
"step": 80
},
{
"epoch": 0.03194250349371132,
"eval_loss": 2.2982735633850098,
"eval_runtime": 290.0312,
"eval_samples_per_second": 0.883,
"eval_steps_per_second": 0.221,
"step": 80
},
{
"epoch": 0.03234178478738271,
"grad_norm": 0.5482540726661682,
"learning_rate": 2.492086915388095e-07,
"loss": 2.2594,
"step": 81
},
{
"epoch": 0.0327410660810541,
"grad_norm": 0.44985291361808777,
"learning_rate": 2.4919862562752036e-07,
"loss": 2.3427,
"step": 82
},
{
"epoch": 0.033140347374725494,
"grad_norm": 0.6427104473114014,
"learning_rate": 2.4918855225785203e-07,
"loss": 2.128,
"step": 83
},
{
"epoch": 0.03353962866839689,
"grad_norm": 0.5451573133468628,
"learning_rate": 2.491784714215121e-07,
"loss": 2.3032,
"step": 84
},
{
"epoch": 0.033938909962068275,
"grad_norm": 0.5538087487220764,
"learning_rate": 2.4916838311019566e-07,
"loss": 2.161,
"step": 85
},
{
"epoch": 0.03433819125573967,
"grad_norm": 0.6961225271224976,
"learning_rate": 2.4915828731558555e-07,
"loss": 2.3407,
"step": 86
},
{
"epoch": 0.03473747254941106,
"grad_norm": 0.5596035718917847,
"learning_rate": 2.4914818402935235e-07,
"loss": 2.4305,
"step": 87
},
{
"epoch": 0.03513675384308245,
"grad_norm": 0.5701755881309509,
"learning_rate": 2.4913807324315406e-07,
"loss": 2.2315,
"step": 88
},
{
"epoch": 0.035536035136753845,
"grad_norm": 0.5389382839202881,
"learning_rate": 2.491279549486365e-07,
"loss": 2.3222,
"step": 89
},
{
"epoch": 0.03593531643042523,
"grad_norm": 0.4433860778808594,
"learning_rate": 2.491178291374329e-07,
"loss": 2.0449,
"step": 90
},
{
"epoch": 0.03593531643042523,
"eval_loss": 2.2697765827178955,
"eval_runtime": 290.8528,
"eval_samples_per_second": 0.88,
"eval_steps_per_second": 0.22,
"step": 90
},
{
"epoch": 0.036334597724096626,
"grad_norm": 0.4071255028247833,
"learning_rate": 2.491076958011643e-07,
"loss": 2.391,
"step": 91
},
{
"epoch": 0.03673387901776802,
"grad_norm": 0.6126638054847717,
"learning_rate": 2.49097554931439e-07,
"loss": 2.3567,
"step": 92
},
{
"epoch": 0.03713316031143941,
"grad_norm": 0.4549248516559601,
"learning_rate": 2.490874065198529e-07,
"loss": 2.1967,
"step": 93
},
{
"epoch": 0.0375324416051108,
"grad_norm": 0.6232178807258606,
"learning_rate": 2.4907725055798955e-07,
"loss": 2.337,
"step": 94
},
{
"epoch": 0.03793172289878219,
"grad_norm": 0.5109763145446777,
"learning_rate": 2.4906708703741986e-07,
"loss": 2.3171,
"step": 95
},
{
"epoch": 0.03833100419245358,
"grad_norm": 0.584500789642334,
"learning_rate": 2.490569159497022e-07,
"loss": 2.5082,
"step": 96
},
{
"epoch": 0.03873028548612498,
"grad_norm": 0.5159786343574524,
"learning_rate": 2.4904673728638226e-07,
"loss": 2.1303,
"step": 97
},
{
"epoch": 0.039129566779796364,
"grad_norm": 0.6698859930038452,
"learning_rate": 2.490365510389933e-07,
"loss": 2.274,
"step": 98
},
{
"epoch": 0.03952884807346776,
"grad_norm": 0.4932086765766144,
"learning_rate": 2.4902635719905583e-07,
"loss": 2.3279,
"step": 99
},
{
"epoch": 0.03992812936713915,
"grad_norm": 0.5625030994415283,
"learning_rate": 2.490161557580779e-07,
"loss": 2.2174,
"step": 100
},
{
"epoch": 0.03992812936713915,
"eval_loss": 2.243732213973999,
"eval_runtime": 290.3533,
"eval_samples_per_second": 0.882,
"eval_steps_per_second": 0.22,
"step": 100
},
{
"epoch": 0.04032741066081054,
"grad_norm": 0.5488671660423279,
"learning_rate": 2.490059467075546e-07,
"loss": 2.3836,
"step": 101
},
{
"epoch": 0.04072669195448193,
"grad_norm": 0.5322253108024597,
"learning_rate": 2.489957300389686e-07,
"loss": 2.2279,
"step": 102
},
{
"epoch": 0.04112597324815333,
"grad_norm": 0.568695604801178,
"learning_rate": 2.4898550574378964e-07,
"loss": 2.2101,
"step": 103
},
{
"epoch": 0.041525254541824715,
"grad_norm": 0.5796780586242676,
"learning_rate": 2.4897527381347493e-07,
"loss": 2.2868,
"step": 104
},
{
"epoch": 0.04192453583549611,
"grad_norm": 0.5519862174987793,
"learning_rate": 2.4896503423946873e-07,
"loss": 2.3682,
"step": 105
},
{
"epoch": 0.042323817129167496,
"grad_norm": 0.4374459385871887,
"learning_rate": 2.4895478701320267e-07,
"loss": 2.2602,
"step": 106
},
{
"epoch": 0.04272309842283889,
"grad_norm": 0.47316691279411316,
"learning_rate": 2.4894453212609545e-07,
"loss": 2.2186,
"step": 107
},
{
"epoch": 0.043122379716510284,
"grad_norm": 0.5060014128684998,
"learning_rate": 2.489342695695529e-07,
"loss": 2.2943,
"step": 108
},
{
"epoch": 0.04352166101018167,
"grad_norm": 0.4985349774360657,
"learning_rate": 2.4892399933496816e-07,
"loss": 2.2959,
"step": 109
},
{
"epoch": 0.043920942303853065,
"grad_norm": 0.5927018523216248,
"learning_rate": 2.489137214137214e-07,
"loss": 1.9577,
"step": 110
},
{
"epoch": 0.043920942303853065,
"eval_loss": 2.221006155014038,
"eval_runtime": 291.9802,
"eval_samples_per_second": 0.877,
"eval_steps_per_second": 0.219,
"step": 110
},
{
"epoch": 0.04432022359752446,
"grad_norm": 0.5408579707145691,
"learning_rate": 2.489034357971798e-07,
"loss": 2.3129,
"step": 111
},
{
"epoch": 0.044719504891195846,
"grad_norm": 0.600794792175293,
"learning_rate": 2.488931424766977e-07,
"loss": 2.2035,
"step": 112
},
{
"epoch": 0.04511878618486724,
"grad_norm": 0.5417996644973755,
"learning_rate": 2.4888284144361654e-07,
"loss": 2.1448,
"step": 113
},
{
"epoch": 0.04551806747853863,
"grad_norm": 0.5807819962501526,
"learning_rate": 2.488725326892646e-07,
"loss": 2.1125,
"step": 114
},
{
"epoch": 0.04591734877221002,
"grad_norm": 0.6197684407234192,
"learning_rate": 2.4886221620495726e-07,
"loss": 2.2536,
"step": 115
},
{
"epoch": 0.046316630065881416,
"grad_norm": 0.49753668904304504,
"learning_rate": 2.48851891981997e-07,
"loss": 2.1006,
"step": 116
},
{
"epoch": 0.0467159113595528,
"grad_norm": 0.5194461345672607,
"learning_rate": 2.48841560011673e-07,
"loss": 2.3214,
"step": 117
},
{
"epoch": 0.0471151926532242,
"grad_norm": 0.48744580149650574,
"learning_rate": 2.4883122028526144e-07,
"loss": 2.3342,
"step": 118
},
{
"epoch": 0.04751447394689559,
"grad_norm": 0.6764684319496155,
"learning_rate": 2.488208727940256e-07,
"loss": 2.23,
"step": 119
},
{
"epoch": 0.04791375524056698,
"grad_norm": 0.513804018497467,
"learning_rate": 2.4881051752921537e-07,
"loss": 2.4721,
"step": 120
},
{
"epoch": 0.04791375524056698,
"eval_loss": 2.200222969055176,
"eval_runtime": 291.0217,
"eval_samples_per_second": 0.88,
"eval_steps_per_second": 0.22,
"step": 120
},
{
"epoch": 0.04831303653423837,
"grad_norm": 0.5214951038360596,
"learning_rate": 2.4880015448206755e-07,
"loss": 2.2337,
"step": 121
},
{
"epoch": 0.04871231782790976,
"grad_norm": 0.644791305065155,
"learning_rate": 2.4878978364380587e-07,
"loss": 2.1174,
"step": 122
},
{
"epoch": 0.049111599121581154,
"grad_norm": 0.59427410364151,
"learning_rate": 2.4877940500564076e-07,
"loss": 2.0378,
"step": 123
},
{
"epoch": 0.04951088041525255,
"grad_norm": 0.44396963715553284,
"learning_rate": 2.4876901855876943e-07,
"loss": 2.0257,
"step": 124
},
{
"epoch": 0.049910161708923935,
"grad_norm": 0.5068044662475586,
"learning_rate": 2.4875862429437587e-07,
"loss": 2.2572,
"step": 125
},
{
"epoch": 0.05030944300259533,
"grad_norm": 0.5593441128730774,
"learning_rate": 2.4874822220363086e-07,
"loss": 2.1756,
"step": 126
},
{
"epoch": 0.05070872429626672,
"grad_norm": 0.5590158700942993,
"learning_rate": 2.4873781227769175e-07,
"loss": 2.159,
"step": 127
},
{
"epoch": 0.05110800558993811,
"grad_norm": 0.44792988896369934,
"learning_rate": 2.487273945077026e-07,
"loss": 2.3303,
"step": 128
},
{
"epoch": 0.051507286883609504,
"grad_norm": 0.4920126497745514,
"learning_rate": 2.4871696888479413e-07,
"loss": 2.2013,
"step": 129
},
{
"epoch": 0.05190656817728089,
"grad_norm": 0.6910445094108582,
"learning_rate": 2.4870653540008377e-07,
"loss": 2.4384,
"step": 130
},
{
"epoch": 0.05190656817728089,
"eval_loss": 2.1825649738311768,
"eval_runtime": 292.6432,
"eval_samples_per_second": 0.875,
"eval_steps_per_second": 0.219,
"step": 130
},
{
"epoch": 0.052305849470952286,
"grad_norm": 0.5733334422111511,
"learning_rate": 2.486960940446754e-07,
"loss": 2.3224,
"step": 131
},
{
"epoch": 0.05270513076462368,
"grad_norm": 0.5770164132118225,
"learning_rate": 2.4868564480965954e-07,
"loss": 2.1093,
"step": 132
},
{
"epoch": 0.05310441205829507,
"grad_norm": 0.5555194616317749,
"learning_rate": 2.486751876861133e-07,
"loss": 2.2621,
"step": 133
},
{
"epoch": 0.05350369335196646,
"grad_norm": 0.5883967876434326,
"learning_rate": 2.4866472266510026e-07,
"loss": 2.1643,
"step": 134
},
{
"epoch": 0.053902974645637855,
"grad_norm": 0.5379590392112732,
"learning_rate": 2.486542497376705e-07,
"loss": 2.0828,
"step": 135
},
{
"epoch": 0.05430225593930924,
"grad_norm": 0.6013762354850769,
"learning_rate": 2.486437688948606e-07,
"loss": 2.1612,
"step": 136
},
{
"epoch": 0.054701537232980636,
"grad_norm": 0.53562992811203,
"learning_rate": 2.486332801276935e-07,
"loss": 2.1625,
"step": 137
},
{
"epoch": 0.05510081852665202,
"grad_norm": 0.508752167224884,
"learning_rate": 2.486227834271787e-07,
"loss": 2.1188,
"step": 138
},
{
"epoch": 0.05550009982032342,
"grad_norm": 0.558559775352478,
"learning_rate": 2.48612278784312e-07,
"loss": 2.0337,
"step": 139
},
{
"epoch": 0.05589938111399481,
"grad_norm": 0.4346345365047455,
"learning_rate": 2.4860176619007567e-07,
"loss": 2.0459,
"step": 140
},
{
"epoch": 0.05589938111399481,
"eval_loss": 2.166564702987671,
"eval_runtime": 292.2803,
"eval_samples_per_second": 0.876,
"eval_steps_per_second": 0.219,
"step": 140
},
{
"epoch": 0.0562986624076662,
"grad_norm": 0.49492570757865906,
"learning_rate": 2.4859124563543813e-07,
"loss": 2.2877,
"step": 141
},
{
"epoch": 0.05669794370133759,
"grad_norm": 0.4642382860183716,
"learning_rate": 2.4858071711135426e-07,
"loss": 2.1271,
"step": 142
},
{
"epoch": 0.05709722499500899,
"grad_norm": 0.46632635593414307,
"learning_rate": 2.485701806087652e-07,
"loss": 2.1594,
"step": 143
},
{
"epoch": 0.057496506288680374,
"grad_norm": 0.4801916480064392,
"learning_rate": 2.485596361185984e-07,
"loss": 2.1931,
"step": 144
},
{
"epoch": 0.05789578758235177,
"grad_norm": 0.501196026802063,
"learning_rate": 2.485490836317674e-07,
"loss": 2.2264,
"step": 145
},
{
"epoch": 0.058295068876023155,
"grad_norm": 0.47007688879966736,
"learning_rate": 2.4853852313917224e-07,
"loss": 2.0938,
"step": 146
},
{
"epoch": 0.05869435016969455,
"grad_norm": 0.5533556342124939,
"learning_rate": 2.485279546316988e-07,
"loss": 2.0474,
"step": 147
},
{
"epoch": 0.05909363146336594,
"grad_norm": 0.5677098035812378,
"learning_rate": 2.485173781002193e-07,
"loss": 2.1814,
"step": 148
},
{
"epoch": 0.05949291275703733,
"grad_norm": 0.4659578800201416,
"learning_rate": 2.4850679353559223e-07,
"loss": 2.1346,
"step": 149
},
{
"epoch": 0.059892194050708725,
"grad_norm": 0.4257633090019226,
"learning_rate": 2.4849620092866184e-07,
"loss": 2.2614,
"step": 150
},
{
"epoch": 0.059892194050708725,
"eval_loss": 2.1524667739868164,
"eval_runtime": 292.1713,
"eval_samples_per_second": 0.876,
"eval_steps_per_second": 0.219,
"step": 150
},
{
"epoch": 0.06029147534438012,
"grad_norm": 0.463015079498291,
"learning_rate": 2.4848560027025883e-07,
"loss": 2.0948,
"step": 151
},
{
"epoch": 0.060690756638051506,
"grad_norm": 0.4769456386566162,
"learning_rate": 2.484749915511997e-07,
"loss": 2.1417,
"step": 152
},
{
"epoch": 0.0610900379317229,
"grad_norm": 0.4819183647632599,
"learning_rate": 2.484643747622871e-07,
"loss": 2.2244,
"step": 153
},
{
"epoch": 0.06148931922539429,
"grad_norm": 0.4990222454071045,
"learning_rate": 2.4845374989430964e-07,
"loss": 2.3024,
"step": 154
},
{
"epoch": 0.06188860051906568,
"grad_norm": 0.553051233291626,
"learning_rate": 2.4844311693804187e-07,
"loss": 2.1907,
"step": 155
},
{
"epoch": 0.062287881812737075,
"grad_norm": 0.483012318611145,
"learning_rate": 2.4843247588424436e-07,
"loss": 2.2427,
"step": 156
},
{
"epoch": 0.06268716310640847,
"grad_norm": 0.5101610422134399,
"learning_rate": 2.4842182672366355e-07,
"loss": 2.0001,
"step": 157
},
{
"epoch": 0.06308644440007985,
"grad_norm": 0.47115758061408997,
"learning_rate": 2.484111694470319e-07,
"loss": 2.2248,
"step": 158
},
{
"epoch": 0.06348572569375124,
"grad_norm": 0.46976238489151,
"learning_rate": 2.4840050404506756e-07,
"loss": 2.0394,
"step": 159
},
{
"epoch": 0.06388500698742264,
"grad_norm": 0.4438592791557312,
"learning_rate": 2.4838983050847455e-07,
"loss": 2.0788,
"step": 160
},
{
"epoch": 0.06388500698742264,
"eval_loss": 2.1408231258392334,
"eval_runtime": 292.5781,
"eval_samples_per_second": 0.875,
"eval_steps_per_second": 0.219,
"step": 160
}
],
"logging_steps": 1,
"max_steps": 2504,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 10,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 5.160187202843443e+17,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}