{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 4.32152117545376,
  "eval_steps": 500,
  "global_step": 5000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.00864304235090752,
      "grad_norm": 3.292635679244995,
      "learning_rate": 9e-06,
      "loss": 5.8856,
      "step": 10
    },
    {
      "epoch": 0.01728608470181504,
      "grad_norm": 0.9546366333961487,
      "learning_rate": 1.9e-05,
      "loss": 5.8159,
      "step": 20
    },
    {
      "epoch": 0.02592912705272256,
      "grad_norm": 0.6935359835624695,
      "learning_rate": 2.9e-05,
      "loss": 5.5635,
      "step": 30
    },
    {
      "epoch": 0.03457216940363008,
      "grad_norm": 0.3259704113006592,
      "learning_rate": 3.9000000000000006e-05,
      "loss": 5.2746,
      "step": 40
    },
    {
      "epoch": 0.043215211754537596,
      "grad_norm": 0.2702903747558594,
      "learning_rate": 4.9e-05,
      "loss": 5.0768,
      "step": 50
    },
    {
      "epoch": 0.05185825410544512,
      "grad_norm": 0.2707994878292084,
      "learning_rate": 5.9e-05,
      "loss": 5.0055,
      "step": 60
    },
    {
      "epoch": 0.06050129645635264,
      "grad_norm": 0.22631309926509857,
      "learning_rate": 6.9e-05,
      "loss": 4.9783,
      "step": 70
    },
    {
      "epoch": 0.06914433880726016,
      "grad_norm": 0.2907443046569824,
      "learning_rate": 7.900000000000001e-05,
      "loss": 4.985,
      "step": 80
    },
    {
      "epoch": 0.07778738115816768,
      "grad_norm": 0.19670741260051727,
      "learning_rate": 8.900000000000001e-05,
      "loss": 4.9292,
      "step": 90
    },
    {
      "epoch": 0.08643042350907519,
      "grad_norm": 0.2479913830757141,
      "learning_rate": 9.900000000000001e-05,
      "loss": 4.845,
      "step": 100
    },
    {
      "epoch": 0.09507346585998272,
      "grad_norm": 0.2541202902793884,
      "learning_rate": 9.999999799740194e-05,
      "loss": 4.8112,
      "step": 110
    },
    {
      "epoch": 0.10371650821089023,
      "grad_norm": 0.23621784150600433,
      "learning_rate": 9.99999910748409e-05,
      "loss": 4.7817,
      "step": 120
    },
    {
      "epoch": 0.11235955056179775,
      "grad_norm": 0.2523561716079712,
      "learning_rate": 9.999997920759416e-05,
      "loss": 4.7711,
      "step": 130
    },
    {
      "epoch": 0.12100259291270528,
      "grad_norm": 0.28017476201057434,
      "learning_rate": 9.999996239566291e-05,
      "loss": 4.7316,
      "step": 140
    },
    {
      "epoch": 0.12964563526361278,
      "grad_norm": 0.34279248118400574,
      "learning_rate": 9.999994063904876e-05,
      "loss": 4.7239,
      "step": 150
    },
    {
      "epoch": 0.13828867761452032,
      "grad_norm": 0.3247782588005066,
      "learning_rate": 9.999991393775392e-05,
      "loss": 4.7047,
      "step": 160
    },
    {
      "epoch": 0.14693171996542784,
      "grad_norm": 0.35700860619544983,
      "learning_rate": 9.999988229178099e-05,
      "loss": 4.6752,
      "step": 170
    },
    {
      "epoch": 0.15557476231633535,
      "grad_norm": 0.4021146595478058,
      "learning_rate": 9.99998457011331e-05,
      "loss": 4.6578,
      "step": 180
    },
    {
      "epoch": 0.16421780466724287,
      "grad_norm": 0.37335410714149475,
      "learning_rate": 9.99998041658139e-05,
      "loss": 4.6256,
      "step": 190
    },
    {
      "epoch": 0.17286084701815038,
      "grad_norm": 0.4010736644268036,
      "learning_rate": 9.999975768582747e-05,
      "loss": 4.6342,
      "step": 200
    },
    {
      "epoch": 0.1815038893690579,
      "grad_norm": 0.4114508032798767,
      "learning_rate": 9.999970626117841e-05,
      "loss": 4.6065,
      "step": 210
    },
    {
      "epoch": 0.19014693171996544,
      "grad_norm": 0.4122427701950073,
      "learning_rate": 9.999964989187182e-05,
      "loss": 4.5984,
      "step": 220
    },
    {
      "epoch": 0.19878997407087295,
      "grad_norm": 0.43556344509124756,
      "learning_rate": 9.999958857791324e-05,
      "loss": 4.5916,
      "step": 230
    },
    {
      "epoch": 0.20743301642178047,
      "grad_norm": 0.3875134289264679,
      "learning_rate": 9.99995223193088e-05,
      "loss": 4.5649,
      "step": 240
    },
    {
      "epoch": 0.21607605877268798,
      "grad_norm": 0.38185903429985046,
      "learning_rate": 9.999945111606498e-05,
      "loss": 4.5544,
      "step": 250
    },
    {
      "epoch": 0.2247191011235955,
      "grad_norm": 0.3720795512199402,
      "learning_rate": 9.999937496818888e-05,
      "loss": 4.5316,
      "step": 260
    },
    {
      "epoch": 0.23336214347450301,
      "grad_norm": 0.3817805349826813,
      "learning_rate": 9.999929387568799e-05,
      "loss": 4.5399,
      "step": 270
    },
    {
      "epoch": 0.24200518582541056,
      "grad_norm": 0.3755202293395996,
      "learning_rate": 9.999920783857033e-05,
      "loss": 4.5317,
      "step": 280
    },
    {
      "epoch": 0.25064822817631804,
      "grad_norm": 0.40243858098983765,
      "learning_rate": 9.999911685684443e-05,
      "loss": 4.5072,
      "step": 290
    },
    {
      "epoch": 0.25929127052722556,
      "grad_norm": 0.4177573621273041,
      "learning_rate": 9.999902093051928e-05,
      "loss": 4.53,
      "step": 300
    },
    {
      "epoch": 0.26793431287813313,
      "grad_norm": 0.398754358291626,
      "learning_rate": 9.999892005960436e-05,
      "loss": 4.511,
      "step": 310
    },
    {
      "epoch": 0.27657735522904064,
      "grad_norm": 0.4280771017074585,
      "learning_rate": 9.999881424410966e-05,
      "loss": 4.4996,
      "step": 320
    },
    {
      "epoch": 0.28522039757994816,
      "grad_norm": 0.361464262008667,
      "learning_rate": 9.999870348404563e-05,
      "loss": 4.4959,
      "step": 330
    },
    {
      "epoch": 0.2938634399308557,
      "grad_norm": 0.4270114004611969,
      "learning_rate": 9.999858777942324e-05,
      "loss": 4.512,
      "step": 340
    },
    {
      "epoch": 0.3025064822817632,
      "grad_norm": 0.4303843677043915,
      "learning_rate": 9.999846713025391e-05,
      "loss": 4.4887,
      "step": 350
    },
    {
      "epoch": 0.3111495246326707,
      "grad_norm": 0.4300330877304077,
      "learning_rate": 9.999834153654958e-05,
      "loss": 4.4755,
      "step": 360
    },
    {
      "epoch": 0.3197925669835782,
      "grad_norm": 0.4024096727371216,
      "learning_rate": 9.999821099832268e-05,
      "loss": 4.4441,
      "step": 370
    },
    {
      "epoch": 0.32843560933448573,
      "grad_norm": 0.4054619371891022,
      "learning_rate": 9.999807551558611e-05,
      "loss": 4.465,
      "step": 380
    },
    {
      "epoch": 0.33707865168539325,
      "grad_norm": 0.42903706431388855,
      "learning_rate": 9.999793508835328e-05,
      "loss": 4.461,
      "step": 390
    },
    {
      "epoch": 0.34572169403630076,
      "grad_norm": 0.45155981183052063,
      "learning_rate": 9.999778971663806e-05,
      "loss": 4.4619,
      "step": 400
    },
    {
      "epoch": 0.3543647363872083,
      "grad_norm": 0.4230276644229889,
      "learning_rate": 9.999763940045484e-05,
      "loss": 4.4362,
      "step": 410
    },
    {
      "epoch": 0.3630077787381158,
      "grad_norm": 0.4837290644645691,
      "learning_rate": 9.999748413981847e-05,
      "loss": 4.4586,
      "step": 420
    },
    {
      "epoch": 0.3716508210890233,
      "grad_norm": 0.40158677101135254,
      "learning_rate": 9.999732393474433e-05,
      "loss": 4.4213,
      "step": 430
    },
    {
      "epoch": 0.3802938634399309,
      "grad_norm": 0.4637954533100128,
      "learning_rate": 9.999715878524823e-05,
      "loss": 4.4292,
      "step": 440
    },
    {
      "epoch": 0.3889369057908384,
      "grad_norm": 0.4022907018661499,
      "learning_rate": 9.999698869134652e-05,
      "loss": 4.423,
      "step": 450
    },
    {
      "epoch": 0.3975799481417459,
      "grad_norm": 0.5074650049209595,
      "learning_rate": 9.999681365305603e-05,
      "loss": 4.4158,
      "step": 460
    },
    {
      "epoch": 0.4062229904926534,
      "grad_norm": 0.40068456530570984,
      "learning_rate": 9.999663367039407e-05,
      "loss": 4.4286,
      "step": 470
    },
    {
      "epoch": 0.41486603284356094,
      "grad_norm": 0.46993207931518555,
      "learning_rate": 9.999644874337842e-05,
      "loss": 4.4149,
      "step": 480
    },
    {
      "epoch": 0.42350907519446845,
      "grad_norm": 0.4461494982242584,
      "learning_rate": 9.999625887202737e-05,
      "loss": 4.4359,
      "step": 490
    },
    {
      "epoch": 0.43215211754537597,
      "grad_norm": 0.4538363218307495,
      "learning_rate": 9.999606405635973e-05,
      "loss": 4.4131,
      "step": 500
    },
    {
      "epoch": 0.4407951598962835,
      "grad_norm": 0.43857431411743164,
      "learning_rate": 9.999586429639471e-05,
      "loss": 4.4162,
      "step": 510
    },
    {
      "epoch": 0.449438202247191,
      "grad_norm": 0.45296424627304077,
      "learning_rate": 9.999565959215212e-05,
      "loss": 4.3884,
      "step": 520
    },
    {
      "epoch": 0.4580812445980985,
      "grad_norm": 0.4565926194190979,
      "learning_rate": 9.999544994365218e-05,
      "loss": 4.3909,
      "step": 530
    },
    {
      "epoch": 0.46672428694900603,
      "grad_norm": 0.4178192615509033,
      "learning_rate": 9.999523535091562e-05,
      "loss": 4.4095,
      "step": 540
    },
    {
      "epoch": 0.47536732929991354,
      "grad_norm": 0.41680389642715454,
      "learning_rate": 9.999501581396367e-05,
      "loss": 4.3951,
      "step": 550
    },
    {
      "epoch": 0.4840103716508211,
      "grad_norm": 0.44991809129714966,
      "learning_rate": 9.999479133281802e-05,
      "loss": 4.384,
      "step": 560
    },
    {
      "epoch": 0.49265341400172863,
      "grad_norm": 0.4085325002670288,
      "learning_rate": 9.99945619075009e-05,
      "loss": 4.3801,
      "step": 570
    },
    {
      "epoch": 0.5012964563526361,
      "grad_norm": 0.4458463191986084,
      "learning_rate": 9.999432753803498e-05,
      "loss": 4.3939,
      "step": 580
    },
    {
      "epoch": 0.5099394987035436,
      "grad_norm": 0.44879746437072754,
      "learning_rate": 9.999408822444344e-05,
      "loss": 4.3756,
      "step": 590
    },
    {
      "epoch": 0.5185825410544511,
      "grad_norm": 0.48292091488838196,
      "learning_rate": 9.999384396674995e-05,
      "loss": 4.3777,
      "step": 600
    },
    {
      "epoch": 0.5272255834053586,
      "grad_norm": 0.4579383432865143,
      "learning_rate": 9.999359476497867e-05,
      "loss": 4.39,
      "step": 610
    },
    {
      "epoch": 0.5358686257562663,
      "grad_norm": 0.46980705857276917,
      "learning_rate": 9.999334061915424e-05,
      "loss": 4.382,
      "step": 620
    },
    {
      "epoch": 0.5445116681071738,
      "grad_norm": 0.48752474784851074,
      "learning_rate": 9.999308152930179e-05,
      "loss": 4.342,
      "step": 630
    },
    {
      "epoch": 0.5531547104580813,
      "grad_norm": 0.3954947888851166,
      "learning_rate": 9.999281749544695e-05,
      "loss": 4.3455,
      "step": 640
    },
    {
      "epoch": 0.5617977528089888,
      "grad_norm": 0.4067750871181488,
      "learning_rate": 9.999254851761581e-05,
      "loss": 4.3459,
      "step": 650
    },
    {
      "epoch": 0.5704407951598963,
      "grad_norm": 0.43236544728279114,
      "learning_rate": 9.9992274595835e-05,
      "loss": 4.3276,
      "step": 660
    },
    {
      "epoch": 0.5790838375108038,
      "grad_norm": 0.3900803029537201,
      "learning_rate": 9.999199573013159e-05,
      "loss": 4.3591,
      "step": 670
    },
    {
      "epoch": 0.5877268798617113,
      "grad_norm": 0.3870255947113037,
      "learning_rate": 9.999171192053317e-05,
      "loss": 4.3508,
      "step": 680
    },
    {
      "epoch": 0.5963699222126189,
      "grad_norm": 0.379078209400177,
      "learning_rate": 9.99914231670678e-05,
      "loss": 4.3236,
      "step": 690
    },
    {
      "epoch": 0.6050129645635264,
      "grad_norm": 0.41333916783332825,
      "learning_rate": 9.999112946976404e-05,
      "loss": 4.3206,
      "step": 700
    },
    {
      "epoch": 0.6136560069144339,
      "grad_norm": 0.45452338457107544,
      "learning_rate": 9.999083082865092e-05,
      "loss": 4.3337,
      "step": 710
    },
    {
      "epoch": 0.6222990492653414,
      "grad_norm": 0.41121405363082886,
      "learning_rate": 9.9990527243758e-05,
      "loss": 4.317,
      "step": 720
    },
    {
      "epoch": 0.6309420916162489,
      "grad_norm": 0.41415151953697205,
      "learning_rate": 9.999021871511527e-05,
      "loss": 4.33,
      "step": 730
    },
    {
      "epoch": 0.6395851339671564,
      "grad_norm": 0.39316877722740173,
      "learning_rate": 9.998990524275327e-05,
      "loss": 4.3324,
      "step": 740
    },
    {
      "epoch": 0.648228176318064,
      "grad_norm": 0.4139920175075531,
      "learning_rate": 9.998958682670299e-05,
      "loss": 4.3067,
      "step": 750
    },
    {
      "epoch": 0.6568712186689715,
      "grad_norm": 0.44398072361946106,
      "learning_rate": 9.998926346699591e-05,
      "loss": 4.3371,
      "step": 760
    },
    {
      "epoch": 0.665514261019879,
      "grad_norm": 0.4074004590511322,
      "learning_rate": 9.998893516366403e-05,
      "loss": 4.3289,
      "step": 770
    },
    {
      "epoch": 0.6741573033707865,
      "grad_norm": 0.38742855191230774,
      "learning_rate": 9.99886019167398e-05,
      "loss": 4.3128,
      "step": 780
    },
    {
      "epoch": 0.682800345721694,
      "grad_norm": 0.385664165019989,
      "learning_rate": 9.998826372625618e-05,
      "loss": 4.3027,
      "step": 790
    },
    {
      "epoch": 0.6914433880726015,
      "grad_norm": 0.46065405011177063,
      "learning_rate": 9.998792059224662e-05,
      "loss": 4.2908,
      "step": 800
    },
    {
      "epoch": 0.700086430423509,
      "grad_norm": 0.3861907422542572,
      "learning_rate": 9.998757251474505e-05,
      "loss": 4.296,
      "step": 810
    },
    {
      "epoch": 0.7087294727744166,
      "grad_norm": 0.4229384958744049,
      "learning_rate": 9.998721949378588e-05,
      "loss": 4.334,
      "step": 820
    },
    {
      "epoch": 0.7173725151253241,
      "grad_norm": 0.3762718737125397,
      "learning_rate": 9.998686152940404e-05,
      "loss": 4.3185,
      "step": 830
    },
    {
      "epoch": 0.7260155574762316,
      "grad_norm": 0.4024622142314911,
      "learning_rate": 9.998649862163492e-05,
      "loss": 4.2898,
      "step": 840
    },
    {
      "epoch": 0.7346585998271391,
      "grad_norm": 0.4279251992702484,
      "learning_rate": 9.998613077051441e-05,
      "loss": 4.3158,
      "step": 850
    },
    {
      "epoch": 0.7433016421780466,
      "grad_norm": 0.43628787994384766,
      "learning_rate": 9.998575797607891e-05,
      "loss": 4.3093,
      "step": 860
    },
    {
      "epoch": 0.7519446845289542,
      "grad_norm": 0.4247160851955414,
      "learning_rate": 9.998538023836526e-05,
      "loss": 4.2912,
      "step": 870
    },
    {
      "epoch": 0.7605877268798618,
      "grad_norm": 0.3851368725299835,
      "learning_rate": 9.998499755741082e-05,
      "loss": 4.2972,
      "step": 880
    },
    {
      "epoch": 0.7692307692307693,
      "grad_norm": 0.4201742708683014,
      "learning_rate": 9.998460993325345e-05,
      "loss": 4.2941,
      "step": 890
    },
    {
      "epoch": 0.7778738115816768,
      "grad_norm": 0.3850812315940857,
      "learning_rate": 9.998421736593147e-05,
      "loss": 4.276,
      "step": 900
    },
    {
      "epoch": 0.7865168539325843,
      "grad_norm": 0.4028345048427582,
      "learning_rate": 9.998381985548368e-05,
      "loss": 4.2888,
      "step": 910
    },
    {
      "epoch": 0.7951598962834918,
      "grad_norm": 0.4184126555919647,
      "learning_rate": 9.998341740194944e-05,
      "loss": 4.2836,
      "step": 920
    },
    {
      "epoch": 0.8038029386343993,
      "grad_norm": 0.514551043510437,
      "learning_rate": 9.998301000536853e-05,
      "loss": 4.2734,
      "step": 930
    },
    {
      "epoch": 0.8124459809853068,
      "grad_norm": 0.41385290026664734,
      "learning_rate": 9.998259766578122e-05,
      "loss": 4.2936,
      "step": 940
    },
    {
      "epoch": 0.8210890233362144,
      "grad_norm": 0.41763439774513245,
      "learning_rate": 9.998218038322831e-05,
      "loss": 4.2656,
      "step": 950
    },
    {
      "epoch": 0.8297320656871219,
      "grad_norm": 0.40201616287231445,
      "learning_rate": 9.998175815775106e-05,
      "loss": 4.2702,
      "step": 960
    },
    {
      "epoch": 0.8383751080380294,
      "grad_norm": 0.3937337100505829,
      "learning_rate": 9.998133098939124e-05,
      "loss": 4.269,
      "step": 970
    },
    {
      "epoch": 0.8470181503889369,
      "grad_norm": 0.41643446683883667,
      "learning_rate": 9.998089887819107e-05,
      "loss": 4.2625,
      "step": 980
    },
    {
      "epoch": 0.8556611927398444,
      "grad_norm": 0.34548649191856384,
      "learning_rate": 9.998046182419328e-05,
      "loss": 4.2662,
      "step": 990
    },
    {
      "epoch": 0.8643042350907519,
      "grad_norm": 0.38553744554519653,
      "learning_rate": 9.998001982744112e-05,
      "loss": 4.2846,
      "step": 1000
    },
    {
      "epoch": 0.8729472774416595,
      "grad_norm": 0.41422292590141296,
      "learning_rate": 9.997957288797828e-05,
      "loss": 4.2718,
      "step": 1010
    },
    {
      "epoch": 0.881590319792567,
      "grad_norm": 0.4107045829296112,
      "learning_rate": 9.997912100584896e-05,
      "loss": 4.2718,
      "step": 1020
    },
    {
      "epoch": 0.8902333621434745,
      "grad_norm": 0.41860222816467285,
      "learning_rate": 9.997866418109785e-05,
      "loss": 4.2405,
      "step": 1030
    },
    {
      "epoch": 0.898876404494382,
      "grad_norm": 0.39858201146125793,
      "learning_rate": 9.997820241377015e-05,
      "loss": 4.2581,
      "step": 1040
    },
    {
      "epoch": 0.9075194468452895,
      "grad_norm": 0.4497794806957245,
      "learning_rate": 9.99777357039115e-05,
      "loss": 4.2374,
      "step": 1050
    },
    {
      "epoch": 0.916162489196197,
      "grad_norm": 0.43440842628479004,
      "learning_rate": 9.997726405156804e-05,
      "loss": 4.2748,
      "step": 1060
    },
    {
      "epoch": 0.9248055315471045,
      "grad_norm": 0.3732091188430786,
      "learning_rate": 9.997678745678646e-05,
      "loss": 4.2616,
      "step": 1070
    },
    {
      "epoch": 0.9334485738980121,
      "grad_norm": 0.3860764503479004,
      "learning_rate": 9.997630591961385e-05,
      "loss": 4.2604,
      "step": 1080
    },
    {
      "epoch": 0.9420916162489196,
      "grad_norm": 0.388324111700058,
      "learning_rate": 9.997581944009787e-05,
      "loss": 4.2557,
      "step": 1090
    },
    {
      "epoch": 0.9507346585998271,
      "grad_norm": 0.3761633038520813,
      "learning_rate": 9.997532801828658e-05,
      "loss": 4.2452,
      "step": 1100
    },
    {
      "epoch": 0.9593777009507347,
      "grad_norm": 0.3888528048992157,
      "learning_rate": 9.997483165422863e-05,
      "loss": 4.2324,
      "step": 1110
    },
    {
      "epoch": 0.9680207433016422,
      "grad_norm": 0.3964410722255707,
      "learning_rate": 9.997433034797306e-05,
      "loss": 4.2575,
      "step": 1120
    },
    {
      "epoch": 0.9766637856525497,
      "grad_norm": 0.3878832161426544,
      "learning_rate": 9.997382409956948e-05,
      "loss": 4.2287,
      "step": 1130
    },
    {
      "epoch": 0.9853068280034573,
      "grad_norm": 0.40653976798057556,
      "learning_rate": 9.997331290906795e-05,
      "loss": 4.2419,
      "step": 1140
    },
    {
      "epoch": 0.9939498703543648,
      "grad_norm": 0.4196073114871979,
      "learning_rate": 9.9972796776519e-05,
      "loss": 4.2637,
      "step": 1150
    },
    {
      "epoch": 1.0025929127052722,
      "grad_norm": 0.4322204291820526,
      "learning_rate": 9.997227570197371e-05,
      "loss": 4.241,
      "step": 1160
    },
    {
      "epoch": 1.0112359550561798,
      "grad_norm": 0.43909183144569397,
      "learning_rate": 9.997174968548358e-05,
      "loss": 4.223,
      "step": 1170
    },
    {
      "epoch": 1.0198789974070872,
      "grad_norm": 0.4085228443145752,
      "learning_rate": 9.997121872710065e-05,
      "loss": 4.2065,
      "step": 1180
    },
    {
      "epoch": 1.0285220397579948,
      "grad_norm": 0.43573814630508423,
      "learning_rate": 9.99706828268774e-05,
      "loss": 4.2157,
      "step": 1190
    },
    {
      "epoch": 1.0371650821089022,
      "grad_norm": 0.43408551812171936,
      "learning_rate": 9.997014198486688e-05,
      "loss": 4.2243,
      "step": 1200
    },
    {
      "epoch": 1.0458081244598099,
      "grad_norm": 0.4013933837413788,
      "learning_rate": 9.996959620112252e-05,
      "loss": 4.2049,
      "step": 1210
    },
    {
      "epoch": 1.0544511668107173,
      "grad_norm": 0.39446181058883667,
      "learning_rate": 9.996904547569833e-05,
      "loss": 4.2275,
      "step": 1220
    },
    {
      "epoch": 1.063094209161625,
      "grad_norm": 0.3860018849372864,
      "learning_rate": 9.996848980864872e-05,
      "loss": 4.2294,
      "step": 1230
    },
    {
      "epoch": 1.0717372515125323,
      "grad_norm": 0.41339194774627686,
      "learning_rate": 9.996792920002872e-05,
      "loss": 4.2181,
      "step": 1240
    },
    {
      "epoch": 1.08038029386344,
      "grad_norm": 0.3952074944972992,
      "learning_rate": 9.996736364989371e-05,
      "loss": 4.2021,
      "step": 1250
    },
    {
      "epoch": 1.0890233362143475,
      "grad_norm": 0.38102981448173523,
      "learning_rate": 9.996679315829967e-05,
      "loss": 4.1976,
      "step": 1260
    },
    {
      "epoch": 1.097666378565255,
      "grad_norm": 0.4206709861755371,
      "learning_rate": 9.996621772530297e-05,
      "loss": 4.2038,
      "step": 1270
    },
    {
      "epoch": 1.1063094209161626,
      "grad_norm": 0.426329642534256,
      "learning_rate": 9.996563735096054e-05,
      "loss": 4.2013,
      "step": 1280
    },
    {
      "epoch": 1.11495246326707,
      "grad_norm": 0.41012808680534363,
      "learning_rate": 9.996505203532978e-05,
      "loss": 4.2102,
      "step": 1290
    },
    {
      "epoch": 1.1235955056179776,
      "grad_norm": 0.44611433148384094,
      "learning_rate": 9.996446177846857e-05,
      "loss": 4.2184,
      "step": 1300
    },
    {
      "epoch": 1.132238547968885,
      "grad_norm": 0.39608779549598694,
      "learning_rate": 9.996386658043526e-05,
      "loss": 4.2013,
      "step": 1310
    },
    {
      "epoch": 1.1408815903197926,
      "grad_norm": 0.3952358067035675,
      "learning_rate": 9.996326644128874e-05,
      "loss": 4.2146,
      "step": 1320
    },
    {
      "epoch": 1.1495246326707,
      "grad_norm": 0.40919986367225647,
      "learning_rate": 9.996266136108835e-05,
      "loss": 4.2076,
      "step": 1330
    },
    {
      "epoch": 1.1581676750216077,
      "grad_norm": 0.4090837836265564,
      "learning_rate": 9.996205133989394e-05,
      "loss": 4.2036,
      "step": 1340
    },
    {
      "epoch": 1.166810717372515,
      "grad_norm": 0.4050755500793457,
      "learning_rate": 9.99614363777658e-05,
      "loss": 4.2,
      "step": 1350
    },
    {
      "epoch": 1.1754537597234227,
      "grad_norm": 0.4159175455570221,
      "learning_rate": 9.996081647476479e-05,
      "loss": 4.186,
      "step": 1360
    },
    {
      "epoch": 1.18409680207433,
      "grad_norm": 0.38426005840301514,
      "learning_rate": 9.996019163095221e-05,
      "loss": 4.2028,
      "step": 1370
    },
    {
      "epoch": 1.1927398444252377,
      "grad_norm": 0.37985771894454956,
      "learning_rate": 9.995956184638982e-05,
      "loss": 4.1962,
      "step": 1380
    },
    {
      "epoch": 1.2013828867761451,
      "grad_norm": 0.4341757595539093,
      "learning_rate": 9.995892712113992e-05,
      "loss": 4.178,
      "step": 1390
    },
    {
      "epoch": 1.2100259291270528,
      "grad_norm": 0.41969048976898193,
      "learning_rate": 9.995828745526529e-05,
      "loss": 4.1717,
      "step": 1400
    },
    {
      "epoch": 1.2186689714779602,
      "grad_norm": 0.3623236417770386,
      "learning_rate": 9.995764284882919e-05,
      "loss": 4.1822,
      "step": 1410
    },
    {
      "epoch": 1.2273120138288678,
      "grad_norm": 0.41732436418533325,
      "learning_rate": 9.995699330189533e-05,
      "loss": 4.1949,
      "step": 1420
    },
    {
      "epoch": 1.2359550561797752,
      "grad_norm": 0.419674813747406,
      "learning_rate": 9.995633881452799e-05,
      "loss": 4.1858,
      "step": 1430
    },
    {
      "epoch": 1.2445980985306828,
      "grad_norm": 0.3526209890842438,
      "learning_rate": 9.995567938679189e-05,
      "loss": 4.1839,
      "step": 1440
    },
    {
      "epoch": 1.2532411408815904,
      "grad_norm": 0.4514230191707611,
      "learning_rate": 9.995501501875222e-05,
      "loss": 4.1851,
      "step": 1450
    },
    {
      "epoch": 1.2618841832324978,
      "grad_norm": 0.36050572991371155,
      "learning_rate": 9.995434571047469e-05,
      "loss": 4.1824,
      "step": 1460
    },
    {
      "epoch": 1.2705272255834053,
      "grad_norm": 0.381134569644928,
      "learning_rate": 9.99536714620255e-05,
      "loss": 4.1699,
      "step": 1470
    },
    {
      "epoch": 1.2791702679343129,
      "grad_norm": 0.4092234969139099,
      "learning_rate": 9.995299227347132e-05,
      "loss": 4.1783,
      "step": 1480
    },
    {
      "epoch": 1.2878133102852205,
      "grad_norm": 0.3862142264842987,
      "learning_rate": 9.995230814487933e-05,
      "loss": 4.1754,
      "step": 1490
    },
    {
      "epoch": 1.296456352636128,
      "grad_norm": 0.4242073893547058,
      "learning_rate": 9.995161907631718e-05,
      "loss": 4.1723,
      "step": 1500
    },
    {
      "epoch": 1.3050993949870353,
      "grad_norm": 0.37873271107673645,
      "learning_rate": 9.995092506785301e-05,
      "loss": 4.1884,
      "step": 1510
    },
    {
      "epoch": 1.313742437337943,
      "grad_norm": 0.4228824973106384,
      "learning_rate": 9.995022611955544e-05,
      "loss": 4.1715,
      "step": 1520
    },
    {
      "epoch": 1.3223854796888506,
      "grad_norm": 0.40594765543937683,
      "learning_rate": 9.994952223149363e-05,
      "loss": 4.1761,
      "step": 1530
    },
    {
      "epoch": 1.331028522039758,
      "grad_norm": 0.41995537281036377,
      "learning_rate": 9.994881340373715e-05,
      "loss": 4.1791,
      "step": 1540
    },
    {
      "epoch": 1.3396715643906656,
      "grad_norm": 0.4363441467285156,
      "learning_rate": 9.994809963635612e-05,
      "loss": 4.1794,
      "step": 1550
    },
    {
      "epoch": 1.348314606741573,
      "grad_norm": 0.42417559027671814,
      "learning_rate": 9.994738092942111e-05,
      "loss": 4.1789,
      "step": 1560
    },
    {
      "epoch": 1.3569576490924806,
      "grad_norm": 0.35726356506347656,
      "learning_rate": 9.994665728300323e-05,
      "loss": 4.1471,
      "step": 1570
    },
    {
      "epoch": 1.365600691443388,
      "grad_norm": 0.5013107657432556,
      "learning_rate": 9.9945928697174e-05,
      "loss": 4.1752,
      "step": 1580
    },
    {
      "epoch": 1.3742437337942957,
      "grad_norm": 0.40358176827430725,
      "learning_rate": 9.994519517200552e-05,
      "loss": 4.1757,
      "step": 1590
    },
    {
      "epoch": 1.382886776145203,
      "grad_norm": 0.40083205699920654,
      "learning_rate": 9.994445670757029e-05,
      "loss": 4.1556,
      "step": 1600
    },
    {
      "epoch": 1.3915298184961107,
      "grad_norm": 0.3931852877140045,
      "learning_rate": 9.994371330394135e-05,
      "loss": 4.1877,
      "step": 1610
    },
    {
      "epoch": 1.400172860847018,
      "grad_norm": 0.40225785970687866,
      "learning_rate": 9.994296496119222e-05,
      "loss": 4.1645,
      "step": 1620
    },
    {
      "epoch": 1.4088159031979257,
      "grad_norm": 0.3956712782382965,
      "learning_rate": 9.994221167939692e-05,
      "loss": 4.1584,
      "step": 1630
    },
    {
      "epoch": 1.4174589455488331,
      "grad_norm": 0.48266497254371643,
      "learning_rate": 9.994145345862994e-05,
      "loss": 4.1584,
      "step": 1640
    },
    {
      "epoch": 1.4261019878997407,
      "grad_norm": 0.4280920624732971,
      "learning_rate": 9.994069029896624e-05,
      "loss": 4.1832,
      "step": 1650
    },
    {
      "epoch": 1.4347450302506481,
      "grad_norm": 0.3954409062862396,
      "learning_rate": 9.993992220048132e-05,
      "loss": 4.1604,
      "step": 1660
    },
    {
      "epoch": 1.4433880726015558,
      "grad_norm": 0.3857767879962921,
      "learning_rate": 9.993914916325113e-05,
      "loss": 4.1445,
      "step": 1670
    },
    {
      "epoch": 1.4520311149524634,
      "grad_norm": 0.37384387850761414,
      "learning_rate": 9.993837118735213e-05,
      "loss": 4.1619,
      "step": 1680
    },
    {
      "epoch": 1.4606741573033708,
      "grad_norm": 0.46884164214134216,
      "learning_rate": 9.993758827286123e-05,
      "loss": 4.1609,
      "step": 1690
    },
    {
      "epoch": 1.4693171996542782,
      "grad_norm": 0.41062942147254944,
      "learning_rate": 9.993680041985587e-05,
      "loss": 4.1574,
      "step": 1700
    },
    {
      "epoch": 1.4779602420051858,
      "grad_norm": 0.3853404223918915,
      "learning_rate": 9.993600762841396e-05,
      "loss": 4.1521,
      "step": 1710
    },
    {
      "epoch": 1.4866032843560935,
      "grad_norm": 0.41848281025886536,
      "learning_rate": 9.993520989861391e-05,
      "loss": 4.1478,
      "step": 1720
    },
    {
      "epoch": 1.4952463267070009,
      "grad_norm": 0.46264174580574036,
      "learning_rate": 9.993440723053462e-05,
      "loss": 4.1473,
      "step": 1730
    },
    {
      "epoch": 1.5038893690579083,
      "grad_norm": 0.40048494935035706,
      "learning_rate": 9.993359962425543e-05,
      "loss": 4.1574,
      "step": 1740
    },
    {
      "epoch": 1.512532411408816,
      "grad_norm": 0.4058494567871094,
      "learning_rate": 9.993278707985625e-05,
      "loss": 4.1594,
      "step": 1750
    },
    {
      "epoch": 1.5211754537597235,
      "grad_norm": 0.43248727917671204,
      "learning_rate": 9.993196959741741e-05,
      "loss": 4.1507,
      "step": 1760
    },
    {
      "epoch": 1.529818496110631,
      "grad_norm": 0.3807108402252197,
      "learning_rate": 9.993114717701977e-05,
      "loss": 4.1287,
      "step": 1770
    },
    {
      "epoch": 1.5384615384615383,
      "grad_norm": 0.4381742477416992,
      "learning_rate": 9.993031981874466e-05,
      "loss": 4.1517,
      "step": 1780
    },
    {
      "epoch": 1.547104580812446,
      "grad_norm": 0.43384110927581787,
      "learning_rate": 9.992948752267388e-05,
      "loss": 4.1385,
      "step": 1790
    },
    {
      "epoch": 1.5557476231633536,
      "grad_norm": 0.40165552496910095,
      "learning_rate": 9.992865028888976e-05,
      "loss": 4.1444,
      "step": 1800
    },
    {
      "epoch": 1.564390665514261,
      "grad_norm": 0.3919483721256256,
      "learning_rate": 9.99278081174751e-05,
      "loss": 4.1494,
      "step": 1810
    },
    {
      "epoch": 1.5730337078651684,
      "grad_norm": 0.3615829646587372,
      "learning_rate": 9.992696100851317e-05,
      "loss": 4.1446,
      "step": 1820
    },
    {
      "epoch": 1.581676750216076,
      "grad_norm": 0.3679444193840027,
      "learning_rate": 9.992610896208774e-05,
      "loss": 4.134,
      "step": 1830
    },
    {
      "epoch": 1.5903197925669836,
      "grad_norm": 0.37700727581977844,
      "learning_rate": 9.992525197828309e-05,
      "loss": 4.1507,
      "step": 1840
    },
    {
      "epoch": 1.598962834917891,
      "grad_norm": 0.38863468170166016,
      "learning_rate": 9.992439005718395e-05,
      "loss": 4.1301,
      "step": 1850
    },
    {
      "epoch": 1.6076058772687987,
      "grad_norm": 0.4527703821659088,
      "learning_rate": 9.99235231988756e-05,
      "loss": 4.1374,
      "step": 1860
    },
    {
      "epoch": 1.6162489196197063,
      "grad_norm": 0.39162564277648926,
      "learning_rate": 9.992265140344371e-05,
      "loss": 4.1011,
      "step": 1870
    },
    {
      "epoch": 1.6248919619706137,
      "grad_norm": 0.37502825260162354,
      "learning_rate": 9.992177467097454e-05,
      "loss": 4.1603,
      "step": 1880
    },
    {
      "epoch": 1.633535004321521,
      "grad_norm": 0.3959217965602875,
      "learning_rate": 9.992089300155476e-05,
      "loss": 4.1594,
      "step": 1890
    },
    {
      "epoch": 1.6421780466724287,
      "grad_norm": 0.37182241678237915,
      "learning_rate": 9.992000639527159e-05,
      "loss": 4.1269,
      "step": 1900
    },
    {
      "epoch": 1.6508210890233364,
      "grad_norm": 0.37193068861961365,
      "learning_rate": 9.99191148522127e-05,
      "loss": 4.1209,
      "step": 1910
    },
    {
      "epoch": 1.6594641313742438,
      "grad_norm": 0.45231059193611145,
      "learning_rate": 9.991821837246624e-05,
      "loss": 4.126,
      "step": 1920
    },
    {
      "epoch": 1.6681071737251512,
      "grad_norm": 0.3941323161125183,
      "learning_rate": 9.99173169561209e-05,
      "loss": 4.1337,
      "step": 1930
    },
    {
      "epoch": 1.6767502160760588,
      "grad_norm": 0.3815472722053528,
      "learning_rate": 9.99164106032658e-05,
      "loss": 4.1436,
      "step": 1940
    },
    {
      "epoch": 1.6853932584269664,
      "grad_norm": 0.3901166319847107,
      "learning_rate": 9.991549931399058e-05,
      "loss": 4.1479,
      "step": 1950
    },
    {
      "epoch": 1.6940363007778738,
      "grad_norm": 0.3755541741847992,
      "learning_rate": 9.991458308838535e-05,
      "loss": 4.119,
      "step": 1960
    },
    {
      "epoch": 1.7026793431287812,
      "grad_norm": 0.38016921281814575,
      "learning_rate": 9.991366192654075e-05,
      "loss": 4.1265,
      "step": 1970
    },
    {
      "epoch": 1.7113223854796888,
      "grad_norm": 0.38255974650382996,
      "learning_rate": 9.991273582854783e-05,
      "loss": 4.1142,
      "step": 1980
    },
    {
      "epoch": 1.7199654278305965,
      "grad_norm": 0.40945836901664734,
      "learning_rate": 9.991180479449821e-05,
      "loss": 4.1435,
      "step": 1990
    },
    {
      "epoch": 1.7286084701815039,
      "grad_norm": 0.4141862690448761,
      "learning_rate": 9.991086882448397e-05,
      "loss": 4.1401,
      "step": 2000
    },
    {
      "epoch": 1.7372515125324113,
      "grad_norm": 0.4165342152118683,
      "learning_rate": 9.990992791859764e-05,
      "loss": 4.1169,
      "step": 2010
    },
    {
      "epoch": 1.745894554883319,
      "grad_norm": 0.4032461941242218,
      "learning_rate": 9.990898207693229e-05,
      "loss": 4.1384,
      "step": 2020
    },
    {
      "epoch": 1.7545375972342265,
      "grad_norm": 0.3860340714454651,
      "learning_rate": 9.990803129958146e-05,
      "loss": 4.1267,
      "step": 2030
    },
    {
      "epoch": 1.763180639585134,
      "grad_norm": 0.37838783860206604,
      "learning_rate": 9.990707558663917e-05,
      "loss": 4.1285,
      "step": 2040
    },
    {
      "epoch": 1.7718236819360413,
      "grad_norm": 0.378642737865448,
      "learning_rate": 9.990611493819992e-05,
      "loss": 4.1082,
      "step": 2050
    },
    {
      "epoch": 1.780466724286949,
      "grad_norm": 0.3989695906639099,
      "learning_rate": 9.990514935435873e-05,
      "loss": 4.1193,
      "step": 2060
    },
    {
      "epoch": 1.7891097666378566,
      "grad_norm": 0.3969743847846985,
      "learning_rate": 9.99041788352111e-05,
      "loss": 4.1043,
      "step": 2070
    },
    {
      "epoch": 1.797752808988764,
      "grad_norm": 0.36269089579582214,
      "learning_rate": 9.990320338085297e-05,
      "loss": 4.1011,
      "step": 2080
    },
    {
      "epoch": 1.8063958513396714,
      "grad_norm": 0.36973994970321655,
      "learning_rate": 9.990222299138086e-05,
      "loss": 4.1152,
      "step": 2090
    },
    {
      "epoch": 1.8150388936905792,
      "grad_norm": 0.4093567728996277,
      "learning_rate": 9.990123766689168e-05,
      "loss": 4.1212,
      "step": 2100
    },
    {
      "epoch": 1.8236819360414867,
      "grad_norm": 0.3885112404823303,
      "learning_rate": 9.990024740748288e-05,
      "loss": 4.121,
      "step": 2110
    },
    {
      "epoch": 1.832324978392394,
      "grad_norm": 0.390177845954895,
      "learning_rate": 9.98992522132524e-05,
      "loss": 4.1315,
      "step": 2120
    },
    {
      "epoch": 1.8409680207433017,
      "grad_norm": 0.37001585960388184,
      "learning_rate": 9.989825208429866e-05,
      "loss": 4.1075,
      "step": 2130
    },
    {
      "epoch": 1.8496110630942093,
      "grad_norm": 0.4177457392215729,
      "learning_rate": 9.989724702072057e-05,
      "loss": 4.1119,
      "step": 2140
    },
    {
      "epoch": 1.8582541054451167,
      "grad_norm": 0.3780882954597473,
      "learning_rate": 9.98962370226175e-05,
      "loss": 4.1153,
      "step": 2150
    },
    {
      "epoch": 1.8668971477960241,
      "grad_norm": 0.3910919725894928,
      "learning_rate": 9.989522209008937e-05,
      "loss": 4.1165,
      "step": 2160
    },
    {
      "epoch": 1.8755401901469317,
      "grad_norm": 0.42845630645751953,
      "learning_rate": 9.98942022232365e-05,
      "loss": 4.111,
      "step": 2170
    },
    {
      "epoch": 1.8841832324978394,
      "grad_norm": 0.42277225852012634,
      "learning_rate": 9.989317742215981e-05,
      "loss": 4.1065,
      "step": 2180
    },
    {
      "epoch": 1.8928262748487468,
      "grad_norm": 0.3859688341617584,
      "learning_rate": 9.98921476869606e-05,
      "loss": 4.0743,
      "step": 2190
    },
    {
      "epoch": 1.9014693171996542,
      "grad_norm": 0.4149601459503174,
      "learning_rate": 9.989111301774072e-05,
      "loss": 4.1142,
      "step": 2200
    },
    {
      "epoch": 1.9101123595505618,
      "grad_norm": 0.407088965177536,
      "learning_rate": 9.98900734146025e-05,
      "loss": 4.1036,
      "step": 2210
    },
    {
      "epoch": 1.9187554019014694,
      "grad_norm": 0.3807941973209381,
      "learning_rate": 9.988902887764874e-05,
      "loss": 4.0997,
      "step": 2220
    },
    {
      "epoch": 1.9273984442523768,
      "grad_norm": 0.3798404932022095,
      "learning_rate": 9.988797940698274e-05,
      "loss": 4.1078,
      "step": 2230
    },
    {
      "epoch": 1.9360414866032842,
      "grad_norm": 0.3978034555912018,
      "learning_rate": 9.988692500270829e-05,
      "loss": 4.1001,
      "step": 2240
    },
    {
      "epoch": 1.9446845289541919,
      "grad_norm": 0.4104568064212799,
      "learning_rate": 9.988586566492964e-05,
      "loss": 4.0929,
      "step": 2250
    },
    {
      "epoch": 1.9533275713050995,
      "grad_norm": 0.42732730507850647,
      "learning_rate": 9.988480139375158e-05,
      "loss": 4.0953,
      "step": 2260
    },
    {
      "epoch": 1.961970613656007,
      "grad_norm": 0.37609344720840454,
      "learning_rate": 9.988373218927937e-05,
      "loss": 4.1026,
      "step": 2270
    },
    {
      "epoch": 1.9706136560069143,
      "grad_norm": 0.3761756718158722,
      "learning_rate": 9.988265805161872e-05,
      "loss": 4.0989,
      "step": 2280
    },
    {
      "epoch": 1.979256698357822,
      "grad_norm": 0.3553387224674225,
      "learning_rate": 9.988157898087584e-05,
      "loss": 4.0934,
      "step": 2290
    },
    {
      "epoch": 1.9878997407087295,
      "grad_norm": 0.3820377588272095,
      "learning_rate": 9.98804949771575e-05,
      "loss": 4.0862,
      "step": 2300
    },
    {
      "epoch": 1.996542783059637,
      "grad_norm": 0.42904117703437805,
      "learning_rate": 9.987940604057086e-05,
      "loss": 4.0965,
      "step": 2310
    },
    {
      "epoch": 2.0051858254105444,
      "grad_norm": 0.3741513788700104,
      "learning_rate": 9.987831217122361e-05,
      "loss": 4.0732,
      "step": 2320
    },
    {
      "epoch": 2.013828867761452,
      "grad_norm": 0.39713436365127563,
      "learning_rate": 9.987721336922395e-05,
      "loss": 4.0809,
      "step": 2330
    },
    {
      "epoch": 2.0224719101123596,
      "grad_norm": 0.4682977497577667,
      "learning_rate": 9.987610963468052e-05,
      "loss": 4.0707,
      "step": 2340
    },
    {
      "epoch": 2.031114952463267,
      "grad_norm": 0.41846728324890137,
      "learning_rate": 9.987500096770248e-05,
      "loss": 4.0701,
      "step": 2350
    },
    {
      "epoch": 2.0397579948141744,
      "grad_norm": 0.41535961627960205,
      "learning_rate": 9.987388736839947e-05,
      "loss": 4.0614,
      "step": 2360
    },
    {
      "epoch": 2.0484010371650823,
      "grad_norm": 0.4282785654067993,
      "learning_rate": 9.987276883688163e-05,
      "loss": 4.0585,
      "step": 2370
    },
    {
      "epoch": 2.0570440795159897,
      "grad_norm": 0.3872493505477905,
      "learning_rate": 9.987164537325957e-05,
      "loss": 4.0676,
      "step": 2380
    },
    {
      "epoch": 2.065687121866897,
      "grad_norm": 0.42589569091796875,
      "learning_rate": 9.987051697764437e-05,
      "loss": 4.0547,
      "step": 2390
    },
    {
      "epoch": 2.0743301642178045,
      "grad_norm": 0.4496053159236908,
      "learning_rate": 9.986938365014766e-05,
      "loss": 4.0768,
      "step": 2400
    },
    {
      "epoch": 2.0829732065687123,
      "grad_norm": 0.4225335717201233,
      "learning_rate": 9.986824539088148e-05,
      "loss": 4.0752,
      "step": 2410
    },
    {
      "epoch": 2.0916162489196197,
      "grad_norm": 0.41624388098716736,
      "learning_rate": 9.986710219995844e-05,
      "loss": 4.0746,
      "step": 2420
    },
    {
      "epoch": 2.100259291270527,
      "grad_norm": 0.38141313195228577,
      "learning_rate": 9.986595407749156e-05,
      "loss": 4.0824,
      "step": 2430
    },
    {
      "epoch": 2.1089023336214345,
      "grad_norm": 0.4378717839717865,
      "learning_rate": 9.986480102359438e-05,
      "loss": 4.0538,
      "step": 2440
    },
    {
      "epoch": 2.1175453759723424,
      "grad_norm": 0.4250941872596741,
      "learning_rate": 9.986364303838097e-05,
      "loss": 4.0847,
      "step": 2450
    },
    {
      "epoch": 2.12618841832325,
      "grad_norm": 0.3991968631744385,
      "learning_rate": 9.986248012196581e-05,
      "loss": 4.0638,
      "step": 2460
    },
    {
      "epoch": 2.134831460674157,
      "grad_norm": 0.40770578384399414,
      "learning_rate": 9.986131227446391e-05,
      "loss": 4.0701,
      "step": 2470
    },
    {
      "epoch": 2.1434745030250646,
      "grad_norm": 0.40430259704589844,
      "learning_rate": 9.986013949599078e-05,
      "loss": 4.0793,
      "step": 2480
    },
    {
      "epoch": 2.1521175453759724,
      "grad_norm": 0.35163435339927673,
      "learning_rate": 9.985896178666238e-05,
      "loss": 4.0672,
      "step": 2490
    },
    {
      "epoch": 2.16076058772688,
      "grad_norm": 0.41108131408691406,
      "learning_rate": 9.985777914659517e-05,
      "loss": 4.0491,
      "step": 2500
    },
    {
      "epoch": 2.1694036300777872,
      "grad_norm": 0.3857698440551758,
      "learning_rate": 9.985659157590614e-05,
      "loss": 4.051,
      "step": 2510
    },
    {
      "epoch": 2.178046672428695,
      "grad_norm": 0.36890825629234314,
      "learning_rate": 9.985539907471273e-05,
      "loss": 4.0638,
      "step": 2520
    },
    {
      "epoch": 2.1866897147796025,
      "grad_norm": 0.42200028896331787,
      "learning_rate": 9.985420164313284e-05,
      "loss": 4.0626,
      "step": 2530
    },
    {
      "epoch": 2.19533275713051,
      "grad_norm": 0.40555793046951294,
      "learning_rate": 9.985299928128491e-05,
      "loss": 4.0548,
      "step": 2540
    },
    {
      "epoch": 2.2039757994814173,
      "grad_norm": 0.37539124488830566,
      "learning_rate": 9.985179198928782e-05,
      "loss": 4.0602,
      "step": 2550
    },
    {
      "epoch": 2.212618841832325,
      "grad_norm": 0.3607882261276245,
      "learning_rate": 9.985057976726102e-05,
      "loss": 4.0685,
      "step": 2560
    },
    {
      "epoch": 2.2212618841832326,
      "grad_norm": 0.41010382771492004,
      "learning_rate": 9.984936261532434e-05,
      "loss": 4.0756,
      "step": 2570
    },
    {
      "epoch": 2.22990492653414,
      "grad_norm": 0.4026075601577759,
      "learning_rate": 9.984814053359816e-05,
      "loss": 4.0678,
      "step": 2580
    },
    {
      "epoch": 2.2385479688850474,
      "grad_norm": 0.4083262085914612,
      "learning_rate": 9.984691352220335e-05,
      "loss": 4.0702,
      "step": 2590
    },
    {
      "epoch": 2.247191011235955,
      "grad_norm": 0.3730342388153076,
      "learning_rate": 9.984568158126123e-05,
      "loss": 4.0778,
      "step": 2600
    },
    {
      "epoch": 2.2558340535868626,
      "grad_norm": 0.3822701871395111,
      "learning_rate": 9.984444471089365e-05,
      "loss": 4.0681,
      "step": 2610
    },
    {
      "epoch": 2.26447709593777,
      "grad_norm": 0.42851561307907104,
      "learning_rate": 9.984320291122294e-05,
      "loss": 4.0671,
      "step": 2620
    },
    {
      "epoch": 2.2731201382886774,
      "grad_norm": 0.38593462109565735,
      "learning_rate": 9.984195618237188e-05,
      "loss": 4.0772,
      "step": 2630
    },
    {
      "epoch": 2.2817631806395853,
      "grad_norm": 0.4122885763645172,
      "learning_rate": 9.984070452446377e-05,
      "loss": 4.0506,
      "step": 2640
    },
    {
      "epoch": 2.2904062229904927,
      "grad_norm": 0.36611250042915344,
      "learning_rate": 9.983944793762241e-05,
      "loss": 4.041,
      "step": 2650
    },
    {
      "epoch": 2.2990492653414,
      "grad_norm": 0.3792266547679901,
      "learning_rate": 9.983818642197205e-05,
      "loss": 4.0487,
      "step": 2660
    },
    {
      "epoch": 2.3076923076923075,
      "grad_norm": 0.37189576029777527,
      "learning_rate": 9.983691997763744e-05,
      "loss": 4.0715,
      "step": 2670
    },
    {
      "epoch": 2.3163353500432153,
      "grad_norm": 0.439643919467926,
      "learning_rate": 9.983564860474384e-05,
      "loss": 4.0565,
      "step": 2680
    },
    {
      "epoch": 2.3249783923941227,
      "grad_norm": 0.40879160165786743,
      "learning_rate": 9.983437230341697e-05,
      "loss": 4.0594,
      "step": 2690
    },
    {
      "epoch": 2.33362143474503,
      "grad_norm": 0.5908119678497314,
      "learning_rate": 9.983309107378305e-05,
      "loss": 4.0808,
      "step": 2700
    },
    {
      "epoch": 2.342264477095938,
      "grad_norm": 0.4059031903743744,
      "learning_rate": 9.98318049159688e-05,
      "loss": 4.05,
      "step": 2710
    },
    {
      "epoch": 2.3509075194468454,
      "grad_norm": 0.40800032019615173,
      "learning_rate": 9.98305138301014e-05,
      "loss": 4.0567,
      "step": 2720
    },
    {
      "epoch": 2.359550561797753,
      "grad_norm": 0.3834645450115204,
      "learning_rate": 9.982921781630852e-05,
      "loss": 4.0568,
      "step": 2730
    },
    {
      "epoch": 2.36819360414866,
      "grad_norm": 0.38579824566841125,
      "learning_rate": 9.982791687471835e-05,
      "loss": 4.0539,
      "step": 2740
    },
    {
      "epoch": 2.3768366464995676,
      "grad_norm": 0.41873669624328613,
      "learning_rate": 9.982661100545954e-05,
      "loss": 4.0337,
      "step": 2750
    },
    {
      "epoch": 2.3854796888504755,
      "grad_norm": 0.4565282166004181,
      "learning_rate": 9.982530020866121e-05,
      "loss": 4.0617,
      "step": 2760
    },
    {
      "epoch": 2.394122731201383,
      "grad_norm": 0.3846672773361206,
      "learning_rate": 9.982398448445301e-05,
      "loss": 4.0333,
      "step": 2770
    },
    {
      "epoch": 2.4027657735522903,
      "grad_norm": 0.38927486538887024,
      "learning_rate": 9.982266383296506e-05,
      "loss": 4.0538,
      "step": 2780
    },
    {
      "epoch": 2.411408815903198,
      "grad_norm": 0.4223096966743469,
      "learning_rate": 9.982133825432794e-05,
      "loss": 4.0583,
      "step": 2790
    },
    {
      "epoch": 2.4200518582541055,
      "grad_norm": 0.39263612031936646,
      "learning_rate": 9.982000774867277e-05,
      "loss": 4.0439,
      "step": 2800
    },
    {
      "epoch": 2.428694900605013,
      "grad_norm": 0.3869222104549408,
      "learning_rate": 9.981867231613114e-05,
      "loss": 4.0278,
      "step": 2810
    },
    {
      "epoch": 2.4373379429559203,
      "grad_norm": 0.4026440382003784,
      "learning_rate": 9.981733195683506e-05,
      "loss": 4.0305,
      "step": 2820
    },
    {
      "epoch": 2.445980985306828,
      "grad_norm": 0.39806249737739563,
      "learning_rate": 9.981598667091712e-05,
      "loss": 4.0547,
      "step": 2830
    },
    {
      "epoch": 2.4546240276577356,
      "grad_norm": 0.4208745062351227,
      "learning_rate": 9.981463645851037e-05,
      "loss": 4.0619,
      "step": 2840
    },
    {
      "epoch": 2.463267070008643,
      "grad_norm": 0.38155847787857056,
      "learning_rate": 9.981328131974831e-05,
      "loss": 4.0498,
      "step": 2850
    },
    {
      "epoch": 2.4719101123595504,
      "grad_norm": 0.39878010749816895,
      "learning_rate": 9.981192125476499e-05,
      "loss": 4.0355,
      "step": 2860
    },
    {
      "epoch": 2.4805531547104582,
      "grad_norm": 0.42514941096305847,
      "learning_rate": 9.981055626369486e-05,
      "loss": 4.0421,
      "step": 2870
    },
    {
      "epoch": 2.4891961970613656,
      "grad_norm": 0.38501569628715515,
      "learning_rate": 9.980918634667296e-05,
      "loss": 4.0238,
      "step": 2880
    },
    {
      "epoch": 2.497839239412273,
      "grad_norm": 0.39104270935058594,
      "learning_rate": 9.980781150383475e-05,
      "loss": 4.0295,
      "step": 2890
    },
    {
      "epoch": 2.506482281763181,
      "grad_norm": 0.3947426378726959,
      "learning_rate": 9.980643173531618e-05,
      "loss": 4.0342,
      "step": 2900
    },
    {
      "epoch": 2.5151253241140883,
      "grad_norm": 0.4183313548564911,
      "learning_rate": 9.98050470412537e-05,
      "loss": 4.0226,
      "step": 2910
    },
    {
      "epoch": 2.5237683664649957,
      "grad_norm": 0.4077540338039398,
      "learning_rate": 9.980365742178428e-05,
      "loss": 4.0581,
      "step": 2920
    },
    {
      "epoch": 2.532411408815903,
      "grad_norm": 0.4163062870502472,
      "learning_rate": 9.98022628770453e-05,
      "loss": 4.0519,
      "step": 2930
    },
    {
      "epoch": 2.5410544511668105,
      "grad_norm": 0.4286973178386688,
      "learning_rate": 9.980086340717472e-05,
      "loss": 4.0412,
      "step": 2940
    },
    {
      "epoch": 2.5496974935177183,
      "grad_norm": 0.39301538467407227,
      "learning_rate": 9.979945901231089e-05,
      "loss": 4.0528,
      "step": 2950
    },
    {
      "epoch": 2.5583405358686258,
      "grad_norm": 0.3591897189617157,
      "learning_rate": 9.979804969259271e-05,
      "loss": 4.0473,
      "step": 2960
    },
    {
      "epoch": 2.566983578219533,
      "grad_norm": 0.3480299413204193,
      "learning_rate": 9.979663544815959e-05,
      "loss": 4.026,
      "step": 2970
    },
    {
      "epoch": 2.575626620570441,
      "grad_norm": 0.4198789596557617,
      "learning_rate": 9.979521627915134e-05,
      "loss": 4.0218,
      "step": 2980
    },
    {
      "epoch": 2.5842696629213484,
      "grad_norm": 0.4237462282180786,
      "learning_rate": 9.979379218570835e-05,
      "loss": 4.032,
      "step": 2990
    },
    {
      "epoch": 2.592912705272256,
      "grad_norm": 0.3961845338344574,
      "learning_rate": 9.97923631679714e-05,
      "loss": 4.0338,
      "step": 3000
    },
    {
      "epoch": 2.601555747623163,
      "grad_norm": 0.37998783588409424,
      "learning_rate": 9.979092922608187e-05,
      "loss": 4.0343,
      "step": 3010
    },
    {
      "epoch": 2.6101987899740706,
      "grad_norm": 0.3984253704547882,
      "learning_rate": 9.978949036018153e-05,
      "loss": 4.0317,
      "step": 3020
    },
    {
      "epoch": 2.6188418323249785,
      "grad_norm": 0.39773324131965637,
      "learning_rate": 9.978804657041269e-05,
      "loss": 4.022,
      "step": 3030
    },
    {
      "epoch": 2.627484874675886,
      "grad_norm": 0.374600887298584,
      "learning_rate": 9.978659785691812e-05,
      "loss": 4.0421,
      "step": 3040
    },
    {
      "epoch": 2.6361279170267933,
      "grad_norm": 0.4022236466407776,
      "learning_rate": 9.97851442198411e-05,
      "loss": 4.0526,
      "step": 3050
    },
    {
      "epoch": 2.644770959377701,
      "grad_norm": 0.42483973503112793,
      "learning_rate": 9.978368565932538e-05,
      "loss": 4.0351,
      "step": 3060
    },
    {
      "epoch": 2.6534140017286085,
      "grad_norm": 0.39662912487983704,
      "learning_rate": 9.978222217551521e-05,
      "loss": 4.0228,
      "step": 3070
    },
    {
      "epoch": 2.662057044079516,
      "grad_norm": 0.3777247369289398,
      "learning_rate": 9.97807537685553e-05,
      "loss": 4.0304,
      "step": 3080
    },
    {
      "epoch": 2.670700086430424,
      "grad_norm": 0.38182365894317627,
      "learning_rate": 9.977928043859088e-05,
      "loss": 4.0461,
      "step": 3090
    },
    {
      "epoch": 2.679343128781331,
      "grad_norm": 0.4112408459186554,
      "learning_rate": 9.977780218576767e-05,
      "loss": 4.0291,
      "step": 3100
    },
    {
      "epoch": 2.6879861711322386,
      "grad_norm": 0.42202165722846985,
      "learning_rate": 9.977631901023183e-05,
      "loss": 4.0138,
      "step": 3110
    },
    {
      "epoch": 2.696629213483146,
      "grad_norm": 0.408734530210495,
      "learning_rate": 9.977483091213005e-05,
      "loss": 4.0286,
      "step": 3120
    },
    {
      "epoch": 2.7052722558340534,
      "grad_norm": 0.38142406940460205,
      "learning_rate": 9.97733378916095e-05,
      "loss": 4.0152,
      "step": 3130
    },
    {
      "epoch": 2.7139152981849612,
      "grad_norm": 0.38112136721611023,
      "learning_rate": 9.977183994881782e-05,
      "loss": 4.0427,
      "step": 3140
    },
    {
      "epoch": 2.7225583405358686,
      "grad_norm": 0.38315171003341675,
      "learning_rate": 9.977033708390315e-05,
      "loss": 4.023,
      "step": 3150
    },
    {
      "epoch": 2.731201382886776,
      "grad_norm": 0.4059940278530121,
      "learning_rate": 9.976882929701411e-05,
      "loss": 4.0224,
      "step": 3160
    },
    {
      "epoch": 2.739844425237684,
      "grad_norm": 0.362513929605484,
      "learning_rate": 9.976731658829981e-05,
      "loss": 4.0091,
      "step": 3170
    },
    {
      "epoch": 2.7484874675885913,
      "grad_norm": 0.3792708218097687,
      "learning_rate": 9.976579895790987e-05,
      "loss": 4.0506,
      "step": 3180
    },
    {
      "epoch": 2.7571305099394987,
      "grad_norm": 0.43076249957084656,
      "learning_rate": 9.976427640599435e-05,
      "loss": 4.0359,
      "step": 3190
    },
    {
      "epoch": 2.765773552290406,
      "grad_norm": 0.40846163034439087,
      "learning_rate": 9.976274893270383e-05,
      "loss": 4.0205,
      "step": 3200
    },
    {
      "epoch": 2.7744165946413135,
      "grad_norm": 0.4179258346557617,
      "learning_rate": 9.976121653818935e-05,
      "loss": 4.0356,
      "step": 3210
    },
    {
      "epoch": 2.7830596369922214,
      "grad_norm": 0.4077315330505371,
      "learning_rate": 9.975967922260248e-05,
      "loss": 4.0255,
      "step": 3220
    },
    {
      "epoch": 2.7917026793431288,
      "grad_norm": 0.43419283628463745,
      "learning_rate": 9.975813698609524e-05,
      "loss": 4.027,
      "step": 3230
    },
    {
      "epoch": 2.800345721694036,
      "grad_norm": 0.4041370749473572,
      "learning_rate": 9.975658982882014e-05,
      "loss": 4.0067,
      "step": 3240
    },
    {
      "epoch": 2.808988764044944,
      "grad_norm": 0.4162820279598236,
      "learning_rate": 9.975503775093021e-05,
      "loss": 4.0069,
      "step": 3250
    },
    {
      "epoch": 2.8176318063958514,
      "grad_norm": 0.38512253761291504,
      "learning_rate": 9.97534807525789e-05,
      "loss": 4.0121,
      "step": 3260
    },
    {
      "epoch": 2.826274848746759,
      "grad_norm": 0.3714209794998169,
      "learning_rate": 9.975191883392023e-05,
      "loss": 4.0197,
      "step": 3270
    },
    {
      "epoch": 2.8349178910976662,
      "grad_norm": 0.3607481122016907,
      "learning_rate": 9.975035199510865e-05,
      "loss": 4.0279,
      "step": 3280
    },
    {
      "epoch": 2.8435609334485736,
      "grad_norm": 0.4277598559856415,
      "learning_rate": 9.974878023629908e-05,
      "loss": 4.0153,
      "step": 3290
    },
    {
      "epoch": 2.8522039757994815,
      "grad_norm": 0.5314668416976929,
      "learning_rate": 9.9747203557647e-05,
      "loss": 4.0446,
      "step": 3300
    },
    {
      "epoch": 2.860847018150389,
      "grad_norm": 0.3989110291004181,
      "learning_rate": 9.97456219593083e-05,
      "loss": 4.0293,
      "step": 3310
    },
    {
      "epoch": 2.8694900605012963,
      "grad_norm": 0.3987487256526947,
      "learning_rate": 9.974403544143941e-05,
      "loss": 4.0332,
      "step": 3320
    },
    {
      "epoch": 2.878133102852204,
      "grad_norm": 0.37416449189186096,
      "learning_rate": 9.974244400419723e-05,
      "loss": 4.0267,
      "step": 3330
    },
    {
      "epoch": 2.8867761452031115,
      "grad_norm": 0.39734718203544617,
      "learning_rate": 9.974084764773913e-05,
      "loss": 4.0127,
      "step": 3340
    },
    {
      "epoch": 2.895419187554019,
      "grad_norm": 0.40905460715293884,
      "learning_rate": 9.973924637222299e-05,
      "loss": 4.0062,
      "step": 3350
    },
    {
      "epoch": 2.904062229904927,
      "grad_norm": 0.39355310797691345,
      "learning_rate": 9.973764017780715e-05,
      "loss": 4.0152,
      "step": 3360
    },
    {
      "epoch": 2.912705272255834,
      "grad_norm": 0.43402382731437683,
      "learning_rate": 9.973602906465048e-05,
      "loss": 4.0089,
      "step": 3370
    },
    {
      "epoch": 2.9213483146067416,
      "grad_norm": 0.3763706684112549,
      "learning_rate": 9.973441303291227e-05,
      "loss": 4.0132,
      "step": 3380
    },
    {
      "epoch": 2.929991356957649,
      "grad_norm": 0.3907529413700104,
      "learning_rate": 9.973279208275237e-05,
      "loss": 4.0251,
      "step": 3390
    },
    {
      "epoch": 2.9386343993085564,
      "grad_norm": 0.35746288299560547,
      "learning_rate": 9.973116621433107e-05,
      "loss": 4.0142,
      "step": 3400
    },
    {
      "epoch": 2.9472774416594643,
      "grad_norm": 0.37593522667884827,
      "learning_rate": 9.972953542780917e-05,
      "loss": 4.043,
      "step": 3410
    },
    {
      "epoch": 2.9559204840103717,
      "grad_norm": 0.391845703125,
      "learning_rate": 9.972789972334791e-05,
      "loss": 4.0254,
      "step": 3420
    },
    {
      "epoch": 2.964563526361279,
      "grad_norm": 0.42428308725357056,
      "learning_rate": 9.972625910110907e-05,
      "loss": 4.0275,
      "step": 3430
    },
    {
      "epoch": 2.973206568712187,
      "grad_norm": 0.41555356979370117,
      "learning_rate": 9.972461356125493e-05,
      "loss": 4.0108,
      "step": 3440
    },
    {
      "epoch": 2.9818496110630943,
      "grad_norm": 0.410685658454895,
      "learning_rate": 9.972296310394817e-05,
      "loss": 4.0132,
      "step": 3450
    },
    {
      "epoch": 2.9904926534140017,
      "grad_norm": 0.42894941568374634,
      "learning_rate": 9.972130772935204e-05,
      "loss": 4.031,
      "step": 3460
    },
    {
      "epoch": 2.999135695764909,
      "grad_norm": 0.41685834527015686,
      "learning_rate": 9.971964743763025e-05,
      "loss": 4.0433,
      "step": 3470
    },
    {
      "epoch": 3.007778738115817,
      "grad_norm": 0.4265305995941162,
      "learning_rate": 9.971798222894698e-05,
      "loss": 3.9885,
      "step": 3480
    },
    {
      "epoch": 3.0164217804667244,
      "grad_norm": 0.4140925705432892,
      "learning_rate": 9.97163121034669e-05,
      "loss": 3.9851,
      "step": 3490
    },
    {
      "epoch": 3.025064822817632,
      "grad_norm": 0.4080989360809326,
      "learning_rate": 9.971463706135518e-05,
      "loss": 4.001,
      "step": 3500
    },
    {
      "epoch": 3.033707865168539,
      "grad_norm": 0.3796727657318115,
      "learning_rate": 9.971295710277751e-05,
      "loss": 3.9844,
      "step": 3510
    },
    {
      "epoch": 3.042350907519447,
      "grad_norm": 0.3994463086128235,
      "learning_rate": 9.971127222789997e-05,
      "loss": 3.984,
      "step": 3520
    },
    {
      "epoch": 3.0509939498703544,
      "grad_norm": 0.37408745288848877,
      "learning_rate": 9.970958243688921e-05,
      "loss": 3.9793,
      "step": 3530
    },
    {
      "epoch": 3.059636992221262,
      "grad_norm": 0.4110405445098877,
      "learning_rate": 9.970788772991234e-05,
      "loss": 3.995,
      "step": 3540
    },
    {
      "epoch": 3.0682800345721692,
      "grad_norm": 0.4115905463695526,
      "learning_rate": 9.970618810713696e-05,
      "loss": 3.9781,
      "step": 3550
    },
    {
      "epoch": 3.076923076923077,
      "grad_norm": 0.4134933650493622,
      "learning_rate": 9.970448356873114e-05,
      "loss": 3.9761,
      "step": 3560
    },
    {
      "epoch": 3.0855661192739845,
      "grad_norm": 0.5576270818710327,
      "learning_rate": 9.970277411486347e-05,
      "loss": 3.9831,
      "step": 3570
    },
    {
      "epoch": 3.094209161624892,
      "grad_norm": 0.4009261727333069,
      "learning_rate": 9.970105974570296e-05,
      "loss": 3.9924,
      "step": 3580
    },
    {
      "epoch": 3.1028522039757993,
      "grad_norm": 0.37966781854629517,
      "learning_rate": 9.96993404614192e-05,
      "loss": 3.9732,
      "step": 3590
    },
    {
      "epoch": 3.111495246326707,
      "grad_norm": 0.4052596390247345,
      "learning_rate": 9.969761626218219e-05,
      "loss": 3.9736,
      "step": 3600
    },
    {
      "epoch": 3.1201382886776146,
      "grad_norm": 0.4077953100204468,
      "learning_rate": 9.969588714816245e-05,
      "loss": 3.9945,
      "step": 3610
    },
    {
      "epoch": 3.128781331028522,
      "grad_norm": 0.3810597360134125,
      "learning_rate": 9.969415311953097e-05,
      "loss": 3.9801,
      "step": 3620
    },
    {
      "epoch": 3.1374243733794294,
      "grad_norm": 0.40926823019981384,
      "learning_rate": 9.969241417645925e-05,
      "loss": 3.9873,
      "step": 3630
    },
    {
      "epoch": 3.146067415730337,
      "grad_norm": 0.41089189052581787,
      "learning_rate": 9.969067031911924e-05,
      "loss": 3.9869,
      "step": 3640
    },
    {
      "epoch": 3.1547104580812446,
      "grad_norm": 0.4003468453884125,
      "learning_rate": 9.968892154768342e-05,
      "loss": 3.9736,
      "step": 3650
    },
    {
      "epoch": 3.163353500432152,
      "grad_norm": 0.41555094718933105,
      "learning_rate": 9.96871678623247e-05,
      "loss": 3.9983,
      "step": 3660
    },
    {
      "epoch": 3.1719965427830594,
      "grad_norm": 0.39583298563957214,
      "learning_rate": 9.968540926321656e-05,
      "loss": 3.9836,
      "step": 3670
    },
    {
      "epoch": 3.1806395851339673,
      "grad_norm": 0.3939920663833618,
      "learning_rate": 9.968364575053287e-05,
      "loss": 3.9812,
      "step": 3680
    },
    {
      "epoch": 3.1892826274848747,
      "grad_norm": 0.3791988790035248,
      "learning_rate": 9.968187732444803e-05,
      "loss": 4.0007,
      "step": 3690
    },
    {
      "epoch": 3.197925669835782,
      "grad_norm": 0.4620395600795746,
      "learning_rate": 9.968010398513697e-05,
      "loss": 3.9846,
      "step": 3700
    },
    {
      "epoch": 3.20656871218669,
      "grad_norm": 0.3994518220424652,
      "learning_rate": 9.9678325732775e-05,
      "loss": 3.9847,
      "step": 3710
    },
    {
      "epoch": 3.2152117545375973,
      "grad_norm": 0.3929818272590637,
      "learning_rate": 9.967654256753802e-05,
      "loss": 3.9681,
      "step": 3720
    },
    {
      "epoch": 3.2238547968885047,
      "grad_norm": 0.4671798646450043,
      "learning_rate": 9.967475448960237e-05,
      "loss": 3.9615,
      "step": 3730
    },
    {
      "epoch": 3.232497839239412,
      "grad_norm": 0.3958970904350281,
      "learning_rate": 9.967296149914487e-05,
      "loss": 3.9808,
      "step": 3740
    },
    {
      "epoch": 3.24114088159032,
      "grad_norm": 0.3957637846469879,
      "learning_rate": 9.967116359634285e-05,
      "loss": 3.9802,
      "step": 3750
    },
    {
      "epoch": 3.2497839239412274,
      "grad_norm": 0.40409496426582336,
      "learning_rate": 9.966936078137409e-05,
      "loss": 3.9744,
      "step": 3760
    },
    {
      "epoch": 3.258426966292135,
      "grad_norm": 0.39653480052948,
      "learning_rate": 9.96675530544169e-05,
      "loss": 3.977,
      "step": 3770
    },
    {
      "epoch": 3.267070008643042,
      "grad_norm": 0.3906710743904114,
      "learning_rate": 9.966574041565002e-05,
      "loss": 3.9944,
      "step": 3780
    },
    {
      "epoch": 3.27571305099395,
      "grad_norm": 0.4071696698665619,
      "learning_rate": 9.966392286525275e-05,
      "loss": 3.9693,
      "step": 3790
    },
    {
      "epoch": 3.2843560933448575,
      "grad_norm": 0.40187910199165344,
      "learning_rate": 9.96621004034048e-05,
      "loss": 3.9705,
      "step": 3800
    },
    {
      "epoch": 3.292999135695765,
      "grad_norm": 0.39552152156829834,
      "learning_rate": 9.966027303028643e-05,
      "loss": 3.9831,
      "step": 3810
    },
    {
      "epoch": 3.3016421780466723,
      "grad_norm": 0.41403430700302124,
      "learning_rate": 9.965844074607832e-05,
      "loss": 3.9815,
      "step": 3820
    },
    {
      "epoch": 3.31028522039758,
      "grad_norm": 0.36947181820869446,
      "learning_rate": 9.965660355096169e-05,
      "loss": 3.9812,
      "step": 3830
    },
    {
      "epoch": 3.3189282627484875,
      "grad_norm": 0.3820069134235382,
      "learning_rate": 9.965476144511825e-05,
      "loss": 3.9935,
      "step": 3840
    },
    {
      "epoch": 3.327571305099395,
      "grad_norm": 0.38387468457221985,
      "learning_rate": 9.965291442873012e-05,
      "loss": 3.9591,
      "step": 3850
    },
    {
      "epoch": 3.3362143474503023,
      "grad_norm": 0.39455530047416687,
      "learning_rate": 9.965106250198002e-05,
      "loss": 3.983,
      "step": 3860
    },
    {
      "epoch": 3.34485738980121,
      "grad_norm": 0.41662219166755676,
      "learning_rate": 9.964920566505105e-05,
      "loss": 3.9726,
      "step": 3870
    },
    {
      "epoch": 3.3535004321521176,
      "grad_norm": 0.4535958170890808,
      "learning_rate": 9.964734391812685e-05,
      "loss": 3.9751,
      "step": 3880
    },
    {
      "epoch": 3.362143474503025,
      "grad_norm": 0.4376087784767151,
      "learning_rate": 9.964547726139153e-05,
      "loss": 3.9763,
      "step": 3890
    },
    {
      "epoch": 3.370786516853933,
      "grad_norm": 0.45594897866249084,
      "learning_rate": 9.96436056950297e-05,
      "loss": 3.9859,
      "step": 3900
    },
    {
      "epoch": 3.3794295592048402,
      "grad_norm": 0.39022424817085266,
      "learning_rate": 9.964172921922647e-05,
      "loss": 3.9643,
      "step": 3910
    },
    {
      "epoch": 3.3880726015557476,
      "grad_norm": 0.4227944612503052,
      "learning_rate": 9.963984783416736e-05,
      "loss": 3.9546,
      "step": 3920
    },
    {
      "epoch": 3.396715643906655,
      "grad_norm": 0.4053288698196411,
      "learning_rate": 9.963796154003846e-05,
      "loss": 3.9806,
      "step": 3930
    },
    {
      "epoch": 3.4053586862575624,
      "grad_norm": 0.42025065422058105,
      "learning_rate": 9.963607033702632e-05,
      "loss": 3.9638,
      "step": 3940
    },
    {
      "epoch": 3.4140017286084703,
      "grad_norm": 0.42370736598968506,
      "learning_rate": 9.963417422531793e-05,
      "loss": 3.9706,
      "step": 3950
    },
    {
      "epoch": 3.4226447709593777,
      "grad_norm": 0.4178035855293274,
      "learning_rate": 9.963227320510085e-05,
      "loss": 3.9757,
      "step": 3960
    },
    {
      "epoch": 3.431287813310285,
      "grad_norm": 0.4232766628265381,
      "learning_rate": 9.963036727656302e-05,
      "loss": 3.9743,
      "step": 3970
    },
    {
      "epoch": 3.439930855661193,
      "grad_norm": 0.4033137857913971,
      "learning_rate": 9.9628456439893e-05,
      "loss": 3.9601,
      "step": 3980
    },
    {
      "epoch": 3.4485738980121003,
      "grad_norm": 0.38971027731895447,
      "learning_rate": 9.962654069527971e-05,
      "loss": 3.9504,
      "step": 3990
    },
    {
      "epoch": 3.4572169403630078,
      "grad_norm": 0.41488805413246155,
      "learning_rate": 9.962462004291262e-05,
      "loss": 3.9739,
      "step": 4000
    },
    {
      "epoch": 3.465859982713915,
      "grad_norm": 0.4420943558216095,
      "learning_rate": 9.962269448298165e-05,
      "loss": 3.9626,
      "step": 4010
    },
    {
      "epoch": 3.474503025064823,
      "grad_norm": 0.38567495346069336,
      "learning_rate": 9.962076401567726e-05,
      "loss": 3.9649,
      "step": 4020
    },
    {
      "epoch": 3.4831460674157304,
      "grad_norm": 0.38574954867362976,
      "learning_rate": 9.961882864119033e-05,
      "loss": 3.9604,
      "step": 4030
    },
    {
      "epoch": 3.491789109766638,
      "grad_norm": 0.38442280888557434,
      "learning_rate": 9.961688835971228e-05,
      "loss": 3.9836,
      "step": 4040
    },
    {
      "epoch": 3.500432152117545,
      "grad_norm": 0.37078383564949036,
      "learning_rate": 9.961494317143498e-05,
      "loss": 3.9614,
      "step": 4050
    },
    {
      "epoch": 3.509075194468453,
      "grad_norm": 0.3818207383155823,
      "learning_rate": 9.961299307655078e-05,
      "loss": 3.9792,
      "step": 4060
    },
    {
      "epoch": 3.5177182368193605,
      "grad_norm": 0.4358178377151489,
      "learning_rate": 9.961103807525258e-05,
      "loss": 3.955,
      "step": 4070
    },
    {
      "epoch": 3.526361279170268,
      "grad_norm": 0.41927260160446167,
      "learning_rate": 9.960907816773367e-05,
      "loss": 3.9561,
      "step": 4080
    },
    {
      "epoch": 3.5350043215211757,
      "grad_norm": 0.39505141973495483,
      "learning_rate": 9.960711335418789e-05,
      "loss": 3.9673,
      "step": 4090
    },
    {
      "epoch": 3.543647363872083,
      "grad_norm": 0.3742333948612213,
      "learning_rate": 9.960514363480955e-05,
      "loss": 3.9682,
      "step": 4100
    },
    {
      "epoch": 3.5522904062229905,
      "grad_norm": 0.4105224013328552,
      "learning_rate": 9.960316900979345e-05,
      "loss": 3.9791,
      "step": 4110
    },
    {
      "epoch": 3.560933448573898,
      "grad_norm": 0.412748783826828,
      "learning_rate": 9.960118947933486e-05,
      "loss": 3.9727,
      "step": 4120
    },
    {
      "epoch": 3.5695764909248053,
      "grad_norm": 0.42374593019485474,
      "learning_rate": 9.959920504362953e-05,
      "loss": 3.999,
      "step": 4130
    },
    {
      "epoch": 3.578219533275713,
      "grad_norm": 0.4130748212337494,
      "learning_rate": 9.959721570287372e-05,
      "loss": 3.9716,
      "step": 4140
    },
    {
      "epoch": 3.5868625756266206,
      "grad_norm": 0.41365188360214233,
      "learning_rate": 9.959522145726416e-05,
      "loss": 3.9589,
      "step": 4150
    },
    {
      "epoch": 3.595505617977528,
      "grad_norm": 0.3764607012271881,
      "learning_rate": 9.95932223069981e-05,
      "loss": 3.9633,
      "step": 4160
    },
    {
      "epoch": 3.604148660328436,
      "grad_norm": 0.39547720551490784,
      "learning_rate": 9.959121825227319e-05,
      "loss": 3.9798,
      "step": 4170
    },
    {
      "epoch": 3.6127917026793432,
      "grad_norm": 0.39446717500686646,
      "learning_rate": 9.958920929328765e-05,
      "loss": 3.982,
      "step": 4180
    },
    {
      "epoch": 3.6214347450302506,
      "grad_norm": 0.3937778174877167,
      "learning_rate": 9.958719543024015e-05,
      "loss": 3.9588,
      "step": 4190
    },
    {
      "epoch": 3.630077787381158,
      "grad_norm": 0.4112344980239868,
      "learning_rate": 9.958517666332986e-05,
      "loss": 3.9606,
      "step": 4200
    },
    {
      "epoch": 3.6387208297320655,
      "grad_norm": 0.39207276701927185,
      "learning_rate": 9.958315299275637e-05,
      "loss": 3.9529,
      "step": 4210
    },
    {
      "epoch": 3.6473638720829733,
      "grad_norm": 0.4045730531215668,
      "learning_rate": 9.958112441871987e-05,
      "loss": 3.9666,
      "step": 4220
    },
    {
      "epoch": 3.6560069144338807,
      "grad_norm": 0.38452064990997314,
      "learning_rate": 9.957909094142097e-05,
      "loss": 3.9672,
      "step": 4230
    },
    {
      "epoch": 3.664649956784788,
      "grad_norm": 0.42553505301475525,
      "learning_rate": 9.957705256106073e-05,
      "loss": 3.967,
      "step": 4240
    },
    {
      "epoch": 3.673292999135696,
      "grad_norm": 0.393327921628952,
      "learning_rate": 9.957500927784076e-05,
      "loss": 3.987,
      "step": 4250
    },
    {
      "epoch": 3.6819360414866034,
      "grad_norm": 0.3970630466938019,
      "learning_rate": 9.95729610919631e-05,
      "loss": 3.9638,
      "step": 4260
    },
    {
      "epoch": 3.6905790838375108,
      "grad_norm": 0.3931358754634857,
      "learning_rate": 9.957090800363033e-05,
      "loss": 3.9708,
      "step": 4270
    },
    {
      "epoch": 3.699222126188418,
      "grad_norm": 0.40673935413360596,
      "learning_rate": 9.95688500130455e-05,
      "loss": 3.977,
      "step": 4280
    },
    {
      "epoch": 3.7078651685393256,
      "grad_norm": 0.43467679619789124,
      "learning_rate": 9.956678712041209e-05,
      "loss": 3.9657,
      "step": 4290
    },
    {
      "epoch": 3.7165082108902334,
      "grad_norm": 0.39805540442466736,
      "learning_rate": 9.956471932593416e-05,
      "loss": 3.9583,
      "step": 4300
    },
    {
      "epoch": 3.725151253241141,
      "grad_norm": 0.41366034746170044,
      "learning_rate": 9.956264662981613e-05,
      "loss": 3.9722,
      "step": 4310
    },
    {
      "epoch": 3.7337942955920482,
      "grad_norm": 0.42467692494392395,
      "learning_rate": 9.956056903226306e-05,
      "loss": 3.9656,
      "step": 4320
    },
    {
      "epoch": 3.742437337942956,
      "grad_norm": 0.39933547377586365,
      "learning_rate": 9.955848653348034e-05,
      "loss": 3.9683,
      "step": 4330
    },
    {
      "epoch": 3.7510803802938635,
      "grad_norm": 0.3879496157169342,
      "learning_rate": 9.955639913367397e-05,
      "loss": 3.9773,
      "step": 4340
    },
    {
      "epoch": 3.759723422644771,
      "grad_norm": 0.37019243836402893,
      "learning_rate": 9.955430683305033e-05,
      "loss": 3.9524,
      "step": 4350
    },
    {
      "epoch": 3.7683664649956787,
      "grad_norm": 0.4178643822669983,
      "learning_rate": 9.955220963181638e-05,
      "loss": 3.9485,
      "step": 4360
    },
    {
      "epoch": 3.777009507346586,
      "grad_norm": 0.41122493147850037,
      "learning_rate": 9.95501075301795e-05,
      "loss": 3.9685,
      "step": 4370
    },
    {
      "epoch": 3.7856525496974935,
      "grad_norm": 0.41016119718551636,
      "learning_rate": 9.954800052834758e-05,
      "loss": 3.9673,
      "step": 4380
    },
    {
      "epoch": 3.794295592048401,
      "grad_norm": 0.45394212007522583,
      "learning_rate": 9.954588862652898e-05,
      "loss": 3.9604,
      "step": 4390
    },
    {
      "epoch": 3.8029386343993083,
      "grad_norm": 0.3679504096508026,
      "learning_rate": 9.954377182493256e-05,
      "loss": 3.963,
      "step": 4400
    },
    {
      "epoch": 3.811581676750216,
      "grad_norm": 0.3887917101383209,
      "learning_rate": 9.954165012376766e-05,
      "loss": 3.9563,
      "step": 4410
    },
    {
      "epoch": 3.8202247191011236,
      "grad_norm": 0.39349594712257385,
      "learning_rate": 9.95395235232441e-05,
      "loss": 3.9755,
      "step": 4420
    },
    {
      "epoch": 3.828867761452031,
      "grad_norm": 0.4100673198699951,
      "learning_rate": 9.953739202357218e-05,
      "loss": 3.963,
      "step": 4430
    },
    {
      "epoch": 3.837510803802939,
      "grad_norm": 0.4095550775527954,
      "learning_rate": 9.953525562496271e-05,
      "loss": 3.9566,
      "step": 4440
    },
    {
      "epoch": 3.8461538461538463,
      "grad_norm": 0.4018843173980713,
      "learning_rate": 9.953311432762697e-05,
      "loss": 3.9928,
      "step": 4450
    },
    {
      "epoch": 3.8547968885047537,
      "grad_norm": 0.417694628238678,
      "learning_rate": 9.95309681317767e-05,
      "loss": 3.9688,
      "step": 4460
    },
    {
      "epoch": 3.863439930855661,
      "grad_norm": 0.37409707903862,
      "learning_rate": 9.952881703762413e-05,
      "loss": 3.968,
      "step": 4470
    },
    {
      "epoch": 3.8720829732065685,
      "grad_norm": 0.4068198800086975,
      "learning_rate": 9.952666104538204e-05,
      "loss": 3.9465,
      "step": 4480
    },
    {
      "epoch": 3.8807260155574763,
      "grad_norm": 0.3868768513202667,
      "learning_rate": 9.95245001552636e-05,
      "loss": 3.9768,
      "step": 4490
    },
    {
      "epoch": 3.8893690579083837,
      "grad_norm": 0.3920934498310089,
      "learning_rate": 9.952233436748255e-05,
      "loss": 3.9508,
      "step": 4500
    },
    {
      "epoch": 3.898012100259291,
      "grad_norm": 0.3930341601371765,
      "learning_rate": 9.952016368225304e-05,
      "loss": 3.9583,
      "step": 4510
    },
    {
      "epoch": 3.906655142610199,
      "grad_norm": 0.38753554224967957,
      "learning_rate": 9.951798809978972e-05,
      "loss": 3.9619,
      "step": 4520
    },
    {
      "epoch": 3.9152981849611064,
      "grad_norm": 0.3948099613189697,
      "learning_rate": 9.951580762030779e-05,
      "loss": 3.9462,
      "step": 4530
    },
    {
      "epoch": 3.923941227312014,
      "grad_norm": 0.38144612312316895,
      "learning_rate": 9.951362224402286e-05,
      "loss": 3.9389,
      "step": 4540
    },
    {
      "epoch": 3.932584269662921,
      "grad_norm": 0.4239857494831085,
      "learning_rate": 9.951143197115105e-05,
      "loss": 3.979,
      "step": 4550
    },
    {
      "epoch": 3.941227312013829,
      "grad_norm": 0.38614705204963684,
      "learning_rate": 9.950923680190897e-05,
      "loss": 3.9719,
      "step": 4560
    },
    {
      "epoch": 3.9498703543647364,
      "grad_norm": 0.40395599603652954,
      "learning_rate": 9.950703673651371e-05,
      "loss": 3.9602,
      "step": 4570
    },
    {
      "epoch": 3.958513396715644,
      "grad_norm": 0.38685745000839233,
      "learning_rate": 9.950483177518284e-05,
      "loss": 3.9595,
      "step": 4580
    },
    {
      "epoch": 3.9671564390665512,
      "grad_norm": 0.4111729562282562,
      "learning_rate": 9.950262191813439e-05,
      "loss": 3.9515,
      "step": 4590
    },
    {
      "epoch": 3.975799481417459,
      "grad_norm": 0.389687180519104,
      "learning_rate": 9.950040716558695e-05,
      "loss": 3.9467,
      "step": 4600
    },
    {
      "epoch": 3.9844425237683665,
      "grad_norm": 0.3774176836013794,
      "learning_rate": 9.949818751775953e-05,
      "loss": 3.953,
      "step": 4610
    },
    {
      "epoch": 3.993085566119274,
      "grad_norm": 0.4081401228904724,
      "learning_rate": 9.94959629748716e-05,
      "loss": 3.9703,
      "step": 4620
    },
    {
      "epoch": 4.001728608470182,
      "grad_norm": 0.5200892090797424,
      "learning_rate": 9.949373353714322e-05,
      "loss": 3.9673,
      "step": 4630
    },
    {
      "epoch": 4.010371650821089,
      "grad_norm": 0.43167346715927124,
      "learning_rate": 9.94914992047948e-05,
      "loss": 3.9162,
      "step": 4640
    },
    {
      "epoch": 4.0190146931719966,
      "grad_norm": 0.4425268769264221,
      "learning_rate": 9.948925997804735e-05,
      "loss": 3.9323,
      "step": 4650
    },
    {
      "epoch": 4.027657735522904,
      "grad_norm": 0.3791431188583374,
      "learning_rate": 9.94870158571223e-05,
      "loss": 3.92,
      "step": 4660
    },
    {
      "epoch": 4.036300777873811,
      "grad_norm": 0.38714367151260376,
      "learning_rate": 9.948476684224157e-05,
      "loss": 3.9103,
      "step": 4670
    },
    {
      "epoch": 4.044943820224719,
      "grad_norm": 0.44668495655059814,
      "learning_rate": 9.948251293362759e-05,
      "loss": 3.9365,
      "step": 4680
    },
    {
      "epoch": 4.053586862575627,
      "grad_norm": 0.3981260061264038,
      "learning_rate": 9.948025413150324e-05,
      "loss": 3.9073,
      "step": 4690
    },
    {
      "epoch": 4.062229904926534,
      "grad_norm": 0.40968841314315796,
      "learning_rate": 9.94779904360919e-05,
      "loss": 3.9182,
      "step": 4700
    },
    {
      "epoch": 4.070872947277442,
      "grad_norm": 0.40504834055900574,
      "learning_rate": 9.947572184761747e-05,
      "loss": 3.9191,
      "step": 4710
    },
    {
      "epoch": 4.079515989628349,
      "grad_norm": 0.4014981985092163,
      "learning_rate": 9.947344836630425e-05,
      "loss": 3.9082,
      "step": 4720
    },
    {
      "epoch": 4.088159031979257,
      "grad_norm": 0.40645283460617065,
      "learning_rate": 9.94711699923771e-05,
      "loss": 3.9224,
      "step": 4730
    },
    {
      "epoch": 4.0968020743301645,
      "grad_norm": 0.514344334602356,
      "learning_rate": 9.946888672606136e-05,
      "loss": 3.9206,
      "step": 4740
    },
    {
      "epoch": 4.1054451166810715,
      "grad_norm": 0.39526522159576416,
      "learning_rate": 9.946659856758279e-05,
      "loss": 3.9295,
      "step": 4750
    },
    {
      "epoch": 4.114088159031979,
      "grad_norm": 0.4328455924987793,
      "learning_rate": 9.946430551716767e-05,
      "loss": 3.9237,
      "step": 4760
    },
    {
      "epoch": 4.122731201382887,
      "grad_norm": 0.4382619261741638,
      "learning_rate": 9.946200757504282e-05,
      "loss": 3.9386,
      "step": 4770
    },
    {
      "epoch": 4.131374243733794,
      "grad_norm": 0.3957505524158478,
      "learning_rate": 9.945970474143544e-05,
      "loss": 3.8995,
      "step": 4780
    },
    {
      "epoch": 4.140017286084702,
      "grad_norm": 0.4167569577693939,
      "learning_rate": 9.945739701657331e-05,
      "loss": 3.9122,
      "step": 4790
    },
    {
      "epoch": 4.148660328435609,
      "grad_norm": 0.39202484488487244,
      "learning_rate": 9.94550844006846e-05,
      "loss": 3.9125,
      "step": 4800
    },
    {
      "epoch": 4.157303370786517,
      "grad_norm": 0.40711510181427,
      "learning_rate": 9.945276689399805e-05,
      "loss": 3.9124,
      "step": 4810
    },
    {
      "epoch": 4.165946413137425,
      "grad_norm": 0.45867350697517395,
      "learning_rate": 9.945044449674283e-05,
      "loss": 3.9302,
      "step": 4820
    },
    {
      "epoch": 4.174589455488332,
      "grad_norm": 0.4911673367023468,
      "learning_rate": 9.944811720914861e-05,
      "loss": 3.917,
      "step": 4830
    },
    {
      "epoch": 4.1832324978392394,
      "grad_norm": 0.4624628722667694,
      "learning_rate": 9.944578503144556e-05,
      "loss": 3.9294,
      "step": 4840
    },
    {
      "epoch": 4.191875540190147,
      "grad_norm": 0.4343968331813812,
      "learning_rate": 9.944344796386432e-05,
      "loss": 3.9165,
      "step": 4850
    },
    {
      "epoch": 4.200518582541054,
      "grad_norm": 0.4132777750492096,
      "learning_rate": 9.944110600663599e-05,
      "loss": 3.9292,
      "step": 4860
    },
    {
      "epoch": 4.209161624891962,
      "grad_norm": 0.4056624174118042,
      "learning_rate": 9.943875915999219e-05,
      "loss": 3.928,
      "step": 4870
    },
    {
      "epoch": 4.217804667242869,
      "grad_norm": 0.41375166177749634,
      "learning_rate": 9.943640742416498e-05,
      "loss": 3.913,
      "step": 4880
    },
    {
      "epoch": 4.226447709593777,
      "grad_norm": 0.3951173424720764,
      "learning_rate": 9.943405079938696e-05,
      "loss": 3.9164,
      "step": 4890
    },
    {
      "epoch": 4.235090751944685,
      "grad_norm": 0.46749618649482727,
      "learning_rate": 9.94316892858912e-05,
      "loss": 3.9088,
      "step": 4900
    },
    {
      "epoch": 4.243733794295592,
      "grad_norm": 0.4152389466762543,
      "learning_rate": 9.942932288391122e-05,
      "loss": 3.9215,
      "step": 4910
    },
    {
      "epoch": 4.2523768366465,
      "grad_norm": 0.41738367080688477,
      "learning_rate": 9.942695159368103e-05,
      "loss": 3.9275,
      "step": 4920
    },
    {
      "epoch": 4.261019878997407,
      "grad_norm": 0.39906904101371765,
      "learning_rate": 9.942457541543514e-05,
      "loss": 3.9347,
      "step": 4930
    },
    {
      "epoch": 4.269662921348314,
      "grad_norm": 0.393771231174469,
      "learning_rate": 9.942219434940855e-05,
      "loss": 3.9201,
      "step": 4940
    },
    {
      "epoch": 4.278305963699222,
      "grad_norm": 0.4133332073688507,
      "learning_rate": 9.941980839583674e-05,
      "loss": 3.934,
      "step": 4950
    },
    {
      "epoch": 4.286949006050129,
      "grad_norm": 0.3788248300552368,
      "learning_rate": 9.941741755495563e-05,
      "loss": 3.9211,
      "step": 4960
    },
    {
      "epoch": 4.295592048401037,
      "grad_norm": 0.40453118085861206,
      "learning_rate": 9.941502182700171e-05,
      "loss": 3.916,
      "step": 4970
    },
    {
      "epoch": 4.304235090751945,
      "grad_norm": 0.4149778485298157,
      "learning_rate": 9.941262121221187e-05,
      "loss": 3.9263,
      "step": 4980
    },
    {
      "epoch": 4.312878133102852,
      "grad_norm": 0.41949522495269775,
      "learning_rate": 9.941021571082352e-05,
      "loss": 3.9193,
      "step": 4990
    },
    {
      "epoch": 4.32152117545376,
      "grad_norm": 0.4075440466403961,
      "learning_rate": 9.940780532307455e-05,
      "loss": 3.9093,
      "step": 5000
    }
  ],
  "logging_steps": 10,
  "max_steps": 100000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 87,
  "save_steps": 5000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 7.0595601838482e+18,
  "train_batch_size": 64,
  "trial_name": null,
  "trial_params": null
}