{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.31061914481603603,
  "eval_steps": 500,
  "global_step": 5935,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0005233683990160674,
      "grad_norm": 1.1370724439620972,
      "learning_rate": 2.910112359550562e-06,
      "loss": 0.8003,
      "step": 10
    },
    {
      "epoch": 0.0010467367980321349,
      "grad_norm": 0.9877656102180481,
      "learning_rate": 3.921348314606742e-06,
      "loss": 0.7587,
      "step": 20
    },
    {
      "epoch": 0.0015701051970482022,
      "grad_norm": 0.8628589510917664,
      "learning_rate": 4.932584269662922e-06,
      "loss": 0.7476,
      "step": 30
    },
    {
      "epoch": 0.0020934735960642698,
      "grad_norm": 0.7837583422660828,
      "learning_rate": 5.943820224719102e-06,
      "loss": 0.7334,
      "step": 40
    },
    {
      "epoch": 0.002616841995080337,
      "grad_norm": 0.7342932224273682,
      "learning_rate": 6.955056179775282e-06,
      "loss": 0.7397,
      "step": 50
    },
    {
      "epoch": 0.0031402103940964044,
      "grad_norm": 0.9271304607391357,
      "learning_rate": 7.966292134831462e-06,
      "loss": 0.7296,
      "step": 60
    },
    {
      "epoch": 0.0036635787931124718,
      "grad_norm": 0.8066033720970154,
      "learning_rate": 8.977528089887641e-06,
      "loss": 0.6864,
      "step": 70
    },
    {
      "epoch": 0.0041869471921285395,
      "grad_norm": 0.8114457726478577,
      "learning_rate": 9.988764044943822e-06,
      "loss": 0.6792,
      "step": 80
    },
    {
      "epoch": 0.004710315591144607,
      "grad_norm": 0.8880093097686768,
      "learning_rate": 1.1000000000000001e-05,
      "loss": 0.7508,
      "step": 90
    },
    {
      "epoch": 0.005233683990160674,
      "grad_norm": 1.1273611783981323,
      "learning_rate": 1.2011235955056182e-05,
      "loss": 0.6849,
      "step": 100
    },
    {
      "epoch": 0.0057570523891767415,
      "grad_norm": 0.7791317701339722,
      "learning_rate": 1.3022471910112362e-05,
      "loss": 0.698,
      "step": 110
    },
    {
      "epoch": 0.006280420788192809,
      "grad_norm": 0.8140974044799805,
      "learning_rate": 1.403370786516854e-05,
      "loss": 0.6947,
      "step": 120
    },
    {
      "epoch": 0.006803789187208876,
      "grad_norm": 0.8656443357467651,
      "learning_rate": 1.504494382022472e-05,
      "loss": 0.7158,
      "step": 130
    },
    {
      "epoch": 0.0073271575862249435,
      "grad_norm": 0.744526743888855,
      "learning_rate": 1.60561797752809e-05,
      "loss": 0.6701,
      "step": 140
    },
    {
      "epoch": 0.00785052598524101,
      "grad_norm": 0.7715036273002625,
      "learning_rate": 1.706741573033708e-05,
      "loss": 0.6794,
      "step": 150
    },
    {
      "epoch": 0.008373894384257079,
      "grad_norm": 0.7816882729530334,
      "learning_rate": 1.8078651685393256e-05,
      "loss": 0.7059,
      "step": 160
    },
    {
      "epoch": 0.008897262783273146,
      "grad_norm": 0.7964625954627991,
      "learning_rate": 1.908988764044944e-05,
      "loss": 0.6954,
      "step": 170
    },
    {
      "epoch": 0.009420631182289214,
      "grad_norm": 1.0829215049743652,
      "learning_rate": 2e-05,
      "loss": 0.6888,
      "step": 180
    },
    {
      "epoch": 0.00994399958130528,
      "grad_norm": 0.6891264319419861,
      "learning_rate": 2e-05,
      "loss": 0.6539,
      "step": 190
    },
    {
      "epoch": 0.010467367980321348,
      "grad_norm": 0.7293897867202759,
      "learning_rate": 2e-05,
      "loss": 0.7164,
      "step": 200
    },
    {
      "epoch": 0.010990736379337415,
      "grad_norm": 0.6318921446800232,
      "learning_rate": 2e-05,
      "loss": 0.7084,
      "step": 210
    },
    {
      "epoch": 0.011514104778353483,
      "grad_norm": 0.6405470371246338,
      "learning_rate": 2e-05,
      "loss": 0.6812,
      "step": 220
    },
    {
      "epoch": 0.012037473177369551,
      "grad_norm": 0.6632219552993774,
      "learning_rate": 2e-05,
      "loss": 0.6733,
      "step": 230
    },
    {
      "epoch": 0.012560841576385618,
      "grad_norm": 0.7196585536003113,
      "learning_rate": 2e-05,
      "loss": 0.6835,
      "step": 240
    },
    {
      "epoch": 0.013084209975401686,
      "grad_norm": 0.5942214131355286,
      "learning_rate": 2e-05,
      "loss": 0.6613,
      "step": 250
    },
    {
      "epoch": 0.013607578374417752,
      "grad_norm": 0.5716996788978577,
      "learning_rate": 2e-05,
      "loss": 0.6759,
      "step": 260
    },
    {
      "epoch": 0.01413094677343382,
      "grad_norm": 0.5713549256324768,
      "learning_rate": 2e-05,
      "loss": 0.7041,
      "step": 270
    },
    {
      "epoch": 0.014654315172449887,
      "grad_norm": 0.5658433437347412,
      "learning_rate": 2e-05,
      "loss": 0.6866,
      "step": 280
    },
    {
      "epoch": 0.015177683571465955,
      "grad_norm": 0.5381125211715698,
      "learning_rate": 2e-05,
      "loss": 0.7024,
      "step": 290
    },
    {
      "epoch": 0.01570105197048202,
      "grad_norm": 0.5079479217529297,
      "learning_rate": 2e-05,
      "loss": 0.686,
      "step": 300
    },
    {
      "epoch": 0.016224420369498088,
      "grad_norm": 0.5530030727386475,
      "learning_rate": 2e-05,
      "loss": 0.6912,
      "step": 310
    },
    {
      "epoch": 0.016747788768514158,
      "grad_norm": 0.5396811962127686,
      "learning_rate": 2e-05,
      "loss": 0.6751,
      "step": 320
    },
    {
      "epoch": 0.017271157167530225,
      "grad_norm": 0.4894005060195923,
      "learning_rate": 2e-05,
      "loss": 0.6649,
      "step": 330
    },
    {
      "epoch": 0.01779452556654629,
      "grad_norm": 0.5230134129524231,
      "learning_rate": 2e-05,
      "loss": 0.684,
      "step": 340
    },
    {
      "epoch": 0.01831789396556236,
      "grad_norm": 0.5332402586936951,
      "learning_rate": 2e-05,
      "loss": 0.6767,
      "step": 350
    },
    {
      "epoch": 0.018841262364578427,
      "grad_norm": 0.5048417448997498,
      "learning_rate": 2e-05,
      "loss": 0.6922,
      "step": 360
    },
    {
      "epoch": 0.019364630763594494,
      "grad_norm": 0.5345087051391602,
      "learning_rate": 2e-05,
      "loss": 0.6976,
      "step": 370
    },
    {
      "epoch": 0.01988799916261056,
      "grad_norm": 0.511880099773407,
      "learning_rate": 2e-05,
      "loss": 0.6814,
      "step": 380
    },
    {
      "epoch": 0.02041136756162663,
      "grad_norm": 0.46280574798583984,
      "learning_rate": 2e-05,
      "loss": 0.6566,
      "step": 390
    },
    {
      "epoch": 0.020934735960642697,
      "grad_norm": 0.464054137468338,
      "learning_rate": 2e-05,
      "loss": 0.6483,
      "step": 400
    },
    {
      "epoch": 0.021458104359658763,
      "grad_norm": 0.5449008941650391,
      "learning_rate": 2e-05,
      "loss": 0.6841,
      "step": 410
    },
    {
      "epoch": 0.02198147275867483,
      "grad_norm": 0.5352079272270203,
      "learning_rate": 2e-05,
      "loss": 0.6604,
      "step": 420
    },
    {
      "epoch": 0.0225048411576909,
      "grad_norm": 0.5088271498680115,
      "learning_rate": 2e-05,
      "loss": 0.6683,
      "step": 430
    },
    {
      "epoch": 0.023028209556706966,
      "grad_norm": 0.5593430399894714,
      "learning_rate": 2e-05,
      "loss": 0.6795,
      "step": 440
    },
    {
      "epoch": 0.023551577955723033,
      "grad_norm": 0.507481575012207,
      "learning_rate": 2e-05,
      "loss": 0.6654,
      "step": 450
    },
    {
      "epoch": 0.024074946354739103,
      "grad_norm": 0.4815616011619568,
      "learning_rate": 2e-05,
      "loss": 0.6843,
      "step": 460
    },
    {
      "epoch": 0.02459831475375517,
      "grad_norm": 0.5703353881835938,
      "learning_rate": 2e-05,
      "loss": 0.6598,
      "step": 470
    },
    {
      "epoch": 0.025121683152771235,
      "grad_norm": 0.48705339431762695,
      "learning_rate": 2e-05,
      "loss": 0.6614,
      "step": 480
    },
    {
      "epoch": 0.025645051551787302,
      "grad_norm": 0.547533392906189,
      "learning_rate": 2e-05,
      "loss": 0.6585,
      "step": 490
    },
    {
      "epoch": 0.026168419950803372,
      "grad_norm": 0.6569129228591919,
      "learning_rate": 2e-05,
      "loss": 0.6716,
      "step": 500
    },
    {
      "epoch": 0.02669178834981944,
      "grad_norm": 0.5733374357223511,
      "learning_rate": 2e-05,
      "loss": 0.6465,
      "step": 510
    },
    {
      "epoch": 0.027215156748835505,
      "grad_norm": 0.4885842502117157,
      "learning_rate": 2e-05,
      "loss": 0.682,
      "step": 520
    },
    {
      "epoch": 0.02773852514785157,
      "grad_norm": 0.42785435914993286,
      "learning_rate": 2e-05,
      "loss": 0.6594,
      "step": 530
    },
    {
      "epoch": 0.02826189354686764,
      "grad_norm": 0.4955022931098938,
      "learning_rate": 2e-05,
      "loss": 0.6674,
      "step": 540
    },
    {
      "epoch": 0.028785261945883708,
      "grad_norm": 0.48021453619003296,
      "learning_rate": 2e-05,
      "loss": 0.6796,
      "step": 550
    },
    {
      "epoch": 0.029308630344899774,
      "grad_norm": 0.4767313301563263,
      "learning_rate": 2e-05,
      "loss": 0.6865,
      "step": 560
    },
    {
      "epoch": 0.029831998743915844,
      "grad_norm": 0.5284311771392822,
      "learning_rate": 2e-05,
      "loss": 0.6489,
      "step": 570
    },
    {
      "epoch": 0.03035536714293191,
      "grad_norm": 0.4461924731731415,
      "learning_rate": 2e-05,
      "loss": 0.686,
      "step": 580
    },
    {
      "epoch": 0.030878735541947977,
      "grad_norm": 0.45115983486175537,
      "learning_rate": 2e-05,
      "loss": 0.6539,
      "step": 590
    },
    {
      "epoch": 0.03140210394096404,
      "grad_norm": 0.5064119100570679,
      "learning_rate": 2e-05,
      "loss": 0.6946,
      "step": 600
    },
    {
      "epoch": 0.03192547233998011,
      "grad_norm": 0.5319550037384033,
      "learning_rate": 2e-05,
      "loss": 0.645,
      "step": 610
    },
    {
      "epoch": 0.032448840738996176,
      "grad_norm": 0.4419703781604767,
      "learning_rate": 2e-05,
      "loss": 0.7003,
      "step": 620
    },
    {
      "epoch": 0.03297220913801225,
      "grad_norm": 0.42887452244758606,
      "learning_rate": 2e-05,
      "loss": 0.6783,
      "step": 630
    },
    {
      "epoch": 0.033495577537028316,
      "grad_norm": 0.46661144495010376,
      "learning_rate": 2e-05,
      "loss": 0.6686,
      "step": 640
    },
    {
      "epoch": 0.03401894593604438,
      "grad_norm": 0.5283986330032349,
      "learning_rate": 2e-05,
      "loss": 0.6679,
      "step": 650
    },
    {
      "epoch": 0.03454231433506045,
      "grad_norm": 0.5113454461097717,
      "learning_rate": 2e-05,
      "loss": 0.6506,
      "step": 660
    },
    {
      "epoch": 0.035065682734076516,
      "grad_norm": 0.4744948148727417,
      "learning_rate": 2e-05,
      "loss": 0.6724,
      "step": 670
    },
    {
      "epoch": 0.03558905113309258,
      "grad_norm": 0.49696066975593567,
      "learning_rate": 2e-05,
      "loss": 0.6652,
      "step": 680
    },
    {
      "epoch": 0.03611241953210865,
      "grad_norm": 0.7813011407852173,
      "learning_rate": 2e-05,
      "loss": 0.6621,
      "step": 690
    },
    {
      "epoch": 0.03663578793112472,
      "grad_norm": 0.49424248933792114,
      "learning_rate": 2e-05,
      "loss": 0.6727,
      "step": 700
    },
    {
      "epoch": 0.03715915633014079,
      "grad_norm": 0.4447154104709625,
      "learning_rate": 2e-05,
      "loss": 0.6659,
      "step": 710
    },
    {
      "epoch": 0.037682524729156855,
      "grad_norm": 0.44892066717147827,
      "learning_rate": 2e-05,
      "loss": 0.6554,
      "step": 720
    },
    {
      "epoch": 0.03820589312817292,
      "grad_norm": 0.49734535813331604,
      "learning_rate": 2e-05,
      "loss": 0.6779,
      "step": 730
    },
    {
      "epoch": 0.03872926152718899,
      "grad_norm": 0.5591990947723389,
      "learning_rate": 2e-05,
      "loss": 0.6927,
      "step": 740
    },
    {
      "epoch": 0.039252629926205054,
      "grad_norm": 0.45224255323410034,
      "learning_rate": 2e-05,
      "loss": 0.6398,
      "step": 750
    },
    {
      "epoch": 0.03977599832522112,
      "grad_norm": 0.41793182492256165,
      "learning_rate": 2e-05,
      "loss": 0.6668,
      "step": 760
    },
    {
      "epoch": 0.04029936672423719,
      "grad_norm": 0.43993815779685974,
      "learning_rate": 2e-05,
      "loss": 0.6456,
      "step": 770
    },
    {
      "epoch": 0.04082273512325326,
      "grad_norm": 0.44372230768203735,
      "learning_rate": 2e-05,
      "loss": 0.6669,
      "step": 780
    },
    {
      "epoch": 0.04134610352226933,
      "grad_norm": 0.4827944040298462,
      "learning_rate": 2e-05,
      "loss": 0.657,
      "step": 790
    },
    {
      "epoch": 0.041869471921285394,
      "grad_norm": 0.4187488257884979,
      "learning_rate": 2e-05,
      "loss": 0.6456,
      "step": 800
    },
    {
      "epoch": 0.04239284032030146,
      "grad_norm": 0.47213008999824524,
      "learning_rate": 2e-05,
      "loss": 0.6555,
      "step": 810
    },
    {
      "epoch": 0.042916208719317527,
      "grad_norm": 0.43557852506637573,
      "learning_rate": 2e-05,
      "loss": 0.6545,
      "step": 820
    },
    {
      "epoch": 0.04343957711833359,
      "grad_norm": 0.45871806144714355,
      "learning_rate": 2e-05,
      "loss": 0.6538,
      "step": 830
    },
    {
      "epoch": 0.04396294551734966,
      "grad_norm": 0.46028947830200195,
      "learning_rate": 2e-05,
      "loss": 0.6356,
      "step": 840
    },
    {
      "epoch": 0.04448631391636573,
      "grad_norm": 0.548116147518158,
      "learning_rate": 2e-05,
      "loss": 0.6872,
      "step": 850
    },
    {
      "epoch": 0.0450096823153818,
      "grad_norm": 0.49427151679992676,
      "learning_rate": 2e-05,
      "loss": 0.6372,
      "step": 860
    },
    {
      "epoch": 0.045533050714397866,
      "grad_norm": 0.46103739738464355,
      "learning_rate": 2e-05,
      "loss": 0.6991,
      "step": 870
    },
    {
      "epoch": 0.04605641911341393,
      "grad_norm": 0.4400480091571808,
      "learning_rate": 2e-05,
      "loss": 0.667,
      "step": 880
    },
    {
      "epoch": 0.04657978751243,
      "grad_norm": 0.45002153515815735,
      "learning_rate": 2e-05,
      "loss": 0.6582,
      "step": 890
    },
    {
      "epoch": 0.047103155911446065,
      "grad_norm": 0.44472604990005493,
      "learning_rate": 2e-05,
      "loss": 0.6706,
      "step": 900
    },
    {
      "epoch": 0.04762652431046213,
      "grad_norm": 0.4121211767196655,
      "learning_rate": 2e-05,
      "loss": 0.6602,
      "step": 910
    },
    {
      "epoch": 0.048149892709478205,
      "grad_norm": 0.5010108351707458,
      "learning_rate": 2e-05,
      "loss": 0.6761,
      "step": 920
    },
    {
      "epoch": 0.04867326110849427,
      "grad_norm": 0.45920342206954956,
      "learning_rate": 2e-05,
      "loss": 0.6671,
      "step": 930
    },
    {
      "epoch": 0.04919662950751034,
      "grad_norm": 0.5177173018455505,
      "learning_rate": 2e-05,
      "loss": 0.652,
      "step": 940
    },
    {
      "epoch": 0.049719997906526404,
      "grad_norm": 0.49086499214172363,
      "learning_rate": 2e-05,
      "loss": 0.6795,
      "step": 950
    },
    {
      "epoch": 0.05024336630554247,
      "grad_norm": 0.582841157913208,
      "learning_rate": 2e-05,
      "loss": 0.6376,
      "step": 960
    },
    {
      "epoch": 0.05076673470455854,
      "grad_norm": 0.4690440595149994,
      "learning_rate": 2e-05,
      "loss": 0.6465,
      "step": 970
    },
    {
      "epoch": 0.051290103103574604,
      "grad_norm": 0.45363402366638184,
      "learning_rate": 2e-05,
      "loss": 0.6511,
      "step": 980
    },
    {
      "epoch": 0.05181347150259067,
      "grad_norm": 0.43920376896858215,
      "learning_rate": 2e-05,
      "loss": 0.6943,
      "step": 990
    },
    {
      "epoch": 0.052336839901606744,
      "grad_norm": 0.421144038438797,
      "learning_rate": 2e-05,
      "loss": 0.6434,
      "step": 1000
    },
    {
      "epoch": 0.05286020830062281,
      "grad_norm": 0.45188188552856445,
      "learning_rate": 2e-05,
      "loss": 0.6467,
      "step": 1010
    },
    {
      "epoch": 0.05338357669963888,
      "grad_norm": 0.4339853823184967,
      "learning_rate": 2e-05,
      "loss": 0.6596,
      "step": 1020
    },
    {
      "epoch": 0.05390694509865494,
      "grad_norm": 0.4234607517719269,
      "learning_rate": 2e-05,
      "loss": 0.674,
      "step": 1030
    },
    {
      "epoch": 0.05443031349767101,
      "grad_norm": 0.46552717685699463,
      "learning_rate": 2e-05,
      "loss": 0.6649,
      "step": 1040
    },
    {
      "epoch": 0.054953681896687076,
      "grad_norm": 0.4405843913555145,
      "learning_rate": 2e-05,
      "loss": 0.6577,
      "step": 1050
    },
    {
      "epoch": 0.05547705029570314,
      "grad_norm": 0.5044958591461182,
      "learning_rate": 2e-05,
      "loss": 0.6152,
      "step": 1060
    },
    {
      "epoch": 0.056000418694719216,
      "grad_norm": 0.43265852332115173,
      "learning_rate": 2e-05,
      "loss": 0.6317,
      "step": 1070
    },
    {
      "epoch": 0.05652378709373528,
      "grad_norm": 0.41526952385902405,
      "learning_rate": 2e-05,
      "loss": 0.6501,
      "step": 1080
    },
    {
      "epoch": 0.05704715549275135,
      "grad_norm": 0.43947693705558777,
      "learning_rate": 2e-05,
      "loss": 0.6542,
      "step": 1090
    },
    {
      "epoch": 0.057570523891767415,
      "grad_norm": 0.4375130534172058,
      "learning_rate": 2e-05,
      "loss": 0.671,
      "step": 1100
    },
    {
      "epoch": 0.05809389229078348,
      "grad_norm": 0.4554004371166229,
      "learning_rate": 2e-05,
      "loss": 0.6507,
      "step": 1110
    },
    {
      "epoch": 0.05861726068979955,
      "grad_norm": 0.5064834356307983,
      "learning_rate": 2e-05,
      "loss": 0.6467,
      "step": 1120
    },
    {
      "epoch": 0.059140629088815615,
      "grad_norm": 0.446809321641922,
      "learning_rate": 2e-05,
      "loss": 0.6751,
      "step": 1130
    },
    {
      "epoch": 0.05966399748783169,
      "grad_norm": 0.4745899736881256,
      "learning_rate": 2e-05,
      "loss": 0.6564,
      "step": 1140
    },
    {
      "epoch": 0.060187365886847755,
      "grad_norm": 0.5119066834449768,
      "learning_rate": 2e-05,
      "loss": 0.6618,
      "step": 1150
    },
    {
      "epoch": 0.06071073428586382,
      "grad_norm": 0.41898113489151,
      "learning_rate": 2e-05,
      "loss": 0.6622,
      "step": 1160
    },
    {
      "epoch": 0.06123410268487989,
      "grad_norm": 0.6409226059913635,
      "learning_rate": 2e-05,
      "loss": 0.6708,
      "step": 1170
    },
    {
      "epoch": 0.061757471083895954,
      "grad_norm": 0.439377099275589,
      "learning_rate": 2e-05,
      "loss": 0.6473,
      "step": 1180
    },
    {
      "epoch": 0.06228083948291202,
      "grad_norm": 0.4160749316215515,
      "learning_rate": 2e-05,
      "loss": 0.6798,
      "step": 1190
    },
    {
      "epoch": 0.06280420788192809,
      "grad_norm": 0.5204553604125977,
      "learning_rate": 2e-05,
      "loss": 0.6866,
      "step": 1200
    },
    {
      "epoch": 0.06332757628094415,
      "grad_norm": 0.48897305130958557,
      "learning_rate": 2e-05,
      "loss": 0.6618,
      "step": 1210
    },
    {
      "epoch": 0.06385094467996022,
      "grad_norm": 0.45014750957489014,
      "learning_rate": 2e-05,
      "loss": 0.6647,
      "step": 1220
    },
    {
      "epoch": 0.06437431307897629,
      "grad_norm": 0.4595189690589905,
      "learning_rate": 2e-05,
      "loss": 0.6625,
      "step": 1230
    },
    {
      "epoch": 0.06489768147799235,
      "grad_norm": 0.49671754240989685,
      "learning_rate": 2e-05,
      "loss": 0.663,
      "step": 1240
    },
    {
      "epoch": 0.06542104987700843,
      "grad_norm": 0.44070637226104736,
      "learning_rate": 2e-05,
      "loss": 0.6613,
      "step": 1250
    },
    {
      "epoch": 0.0659444182760245,
      "grad_norm": 0.45897576212882996,
      "learning_rate": 2e-05,
      "loss": 0.6516,
      "step": 1260
    },
    {
      "epoch": 0.06646778667504057,
      "grad_norm": 0.4450547695159912,
      "learning_rate": 2e-05,
      "loss": 0.641,
      "step": 1270
    },
    {
      "epoch": 0.06699115507405663,
      "grad_norm": 0.4288750886917114,
      "learning_rate": 2e-05,
      "loss": 0.6656,
      "step": 1280
    },
    {
      "epoch": 0.0675145234730727,
      "grad_norm": 0.47525206208229065,
      "learning_rate": 2e-05,
      "loss": 0.6915,
      "step": 1290
    },
    {
      "epoch": 0.06803789187208877,
      "grad_norm": 0.41285258531570435,
      "learning_rate": 2e-05,
      "loss": 0.6401,
      "step": 1300
    },
    {
      "epoch": 0.06856126027110483,
      "grad_norm": 0.46445295214653015,
      "learning_rate": 2e-05,
      "loss": 0.6692,
      "step": 1310
    },
    {
      "epoch": 0.0690846286701209,
      "grad_norm": 0.44153061509132385,
      "learning_rate": 2e-05,
      "loss": 0.6291,
      "step": 1320
    },
    {
      "epoch": 0.06960799706913696,
      "grad_norm": 0.7016943693161011,
      "learning_rate": 2e-05,
      "loss": 0.6405,
      "step": 1330
    },
    {
      "epoch": 0.07013136546815303,
      "grad_norm": 0.498710572719574,
      "learning_rate": 2e-05,
      "loss": 0.6433,
      "step": 1340
    },
    {
      "epoch": 0.0706547338671691,
      "grad_norm": 0.4415338337421417,
      "learning_rate": 2e-05,
      "loss": 0.673,
      "step": 1350
    },
    {
      "epoch": 0.07117810226618516,
      "grad_norm": 0.5377183556556702,
      "learning_rate": 2e-05,
      "loss": 0.6556,
      "step": 1360
    },
    {
      "epoch": 0.07170147066520123,
      "grad_norm": 0.40461525321006775,
      "learning_rate": 2e-05,
      "loss": 0.6384,
      "step": 1370
    },
    {
      "epoch": 0.0722248390642173,
      "grad_norm": 0.4240797758102417,
      "learning_rate": 2e-05,
      "loss": 0.6467,
      "step": 1380
    },
    {
      "epoch": 0.07274820746323336,
      "grad_norm": 0.4052056074142456,
      "learning_rate": 2e-05,
      "loss": 0.6448,
      "step": 1390
    },
    {
      "epoch": 0.07327157586224944,
      "grad_norm": 0.41825127601623535,
      "learning_rate": 2e-05,
      "loss": 0.6465,
      "step": 1400
    },
    {
      "epoch": 0.07379494426126551,
      "grad_norm": 0.41024652123451233,
      "learning_rate": 2e-05,
      "loss": 0.6428,
      "step": 1410
    },
    {
      "epoch": 0.07431831266028158,
      "grad_norm": 0.38469892740249634,
      "learning_rate": 2e-05,
      "loss": 0.6652,
      "step": 1420
    },
    {
      "epoch": 0.07484168105929764,
      "grad_norm": 0.4639175236225128,
      "learning_rate": 2e-05,
      "loss": 0.6503,
      "step": 1430
    },
    {
      "epoch": 0.07536504945831371,
      "grad_norm": 0.4508153200149536,
      "learning_rate": 2e-05,
      "loss": 0.6464,
      "step": 1440
    },
    {
      "epoch": 0.07588841785732978,
      "grad_norm": 0.5173973441123962,
      "learning_rate": 2e-05,
      "loss": 0.6475,
      "step": 1450
    },
    {
      "epoch": 0.07641178625634584,
      "grad_norm": 0.47181883454322815,
      "learning_rate": 2e-05,
      "loss": 0.6662,
      "step": 1460
    },
    {
      "epoch": 0.07693515465536191,
      "grad_norm": 0.43690988421440125,
      "learning_rate": 2e-05,
      "loss": 0.6459,
      "step": 1470
    },
    {
      "epoch": 0.07745852305437798,
      "grad_norm": 0.40047961473464966,
      "learning_rate": 2e-05,
      "loss": 0.6808,
      "step": 1480
    },
    {
      "epoch": 0.07798189145339404,
      "grad_norm": 0.3962308466434479,
      "learning_rate": 2e-05,
      "loss": 0.6249,
      "step": 1490
    },
    {
      "epoch": 0.07850525985241011,
      "grad_norm": 0.5075916051864624,
      "learning_rate": 2e-05,
      "loss": 0.6562,
      "step": 1500
    },
    {
      "epoch": 0.07902862825142618,
      "grad_norm": 0.44439756870269775,
      "learning_rate": 2e-05,
      "loss": 0.6744,
      "step": 1510
    },
    {
      "epoch": 0.07955199665044224,
      "grad_norm": 0.4172062575817108,
      "learning_rate": 2e-05,
      "loss": 0.6328,
      "step": 1520
    },
    {
      "epoch": 0.08007536504945831,
      "grad_norm": 0.9272112250328064,
      "learning_rate": 2e-05,
      "loss": 0.6554,
      "step": 1530
    },
    {
      "epoch": 0.08059873344847437,
      "grad_norm": 0.44365498423576355,
      "learning_rate": 2e-05,
      "loss": 0.6567,
      "step": 1540
    },
    {
      "epoch": 0.08112210184749045,
      "grad_norm": 0.5808505415916443,
      "learning_rate": 2e-05,
      "loss": 0.6671,
      "step": 1550
    },
    {
      "epoch": 0.08164547024650652,
      "grad_norm": 0.4238007068634033,
      "learning_rate": 2e-05,
      "loss": 0.6655,
      "step": 1560
    },
    {
      "epoch": 0.08216883864552259,
      "grad_norm": 0.4158059358596802,
      "learning_rate": 2e-05,
      "loss": 0.6502,
      "step": 1570
    },
    {
      "epoch": 0.08269220704453865,
      "grad_norm": 0.4266833961009979,
      "learning_rate": 2e-05,
      "loss": 0.6389,
      "step": 1580
    },
    {
      "epoch": 0.08321557544355472,
      "grad_norm": 0.4183817505836487,
      "learning_rate": 2e-05,
      "loss": 0.6297,
      "step": 1590
    },
    {
      "epoch": 0.08373894384257079,
      "grad_norm": 0.4242572784423828,
      "learning_rate": 2e-05,
      "loss": 0.6575,
      "step": 1600
    },
    {
      "epoch": 0.08426231224158685,
      "grad_norm": 0.42805418372154236,
      "learning_rate": 2e-05,
      "loss": 0.6316,
      "step": 1610
    },
    {
      "epoch": 0.08478568064060292,
      "grad_norm": 0.441866010427475,
      "learning_rate": 2e-05,
      "loss": 0.6702,
      "step": 1620
    },
    {
      "epoch": 0.08530904903961899,
      "grad_norm": 0.4371151626110077,
      "learning_rate": 2e-05,
      "loss": 0.6623,
      "step": 1630
    },
    {
      "epoch": 0.08583241743863505,
      "grad_norm": 0.43131211400032043,
      "learning_rate": 2e-05,
      "loss": 0.6448,
      "step": 1640
    },
    {
      "epoch": 0.08635578583765112,
      "grad_norm": 0.4316512644290924,
      "learning_rate": 2e-05,
      "loss": 0.6374,
      "step": 1650
    },
    {
      "epoch": 0.08687915423666719,
      "grad_norm": 0.48419904708862305,
      "learning_rate": 2e-05,
      "loss": 0.6224,
      "step": 1660
    },
    {
      "epoch": 0.08740252263568325,
      "grad_norm": 0.3933804929256439,
      "learning_rate": 2e-05,
      "loss": 0.6522,
      "step": 1670
    },
    {
      "epoch": 0.08792589103469932,
      "grad_norm": 0.4437639117240906,
      "learning_rate": 2e-05,
      "loss": 0.6237,
      "step": 1680
    },
    {
      "epoch": 0.08844925943371539,
      "grad_norm": 0.4899040460586548,
      "learning_rate": 2e-05,
      "loss": 0.6321,
      "step": 1690
    },
    {
      "epoch": 0.08897262783273147,
      "grad_norm": 0.4382249712944031,
      "learning_rate": 2e-05,
      "loss": 0.6469,
      "step": 1700
    },
    {
      "epoch": 0.08949599623174753,
      "grad_norm": 0.4095931053161621,
      "learning_rate": 2e-05,
      "loss": 0.6588,
      "step": 1710
    },
    {
      "epoch": 0.0900193646307636,
      "grad_norm": 0.43513184785842896,
      "learning_rate": 2e-05,
      "loss": 0.6569,
      "step": 1720
    },
    {
      "epoch": 0.09054273302977967,
      "grad_norm": 0.46793821454048157,
      "learning_rate": 2e-05,
      "loss": 0.6322,
      "step": 1730
    },
    {
      "epoch": 0.09106610142879573,
      "grad_norm": 0.430164098739624,
      "learning_rate": 2e-05,
      "loss": 0.6224,
      "step": 1740
    },
    {
      "epoch": 0.0915894698278118,
      "grad_norm": 0.3761891722679138,
      "learning_rate": 2e-05,
      "loss": 0.606,
      "step": 1750
    },
    {
      "epoch": 0.09211283822682786,
      "grad_norm": 0.373146116733551,
      "learning_rate": 2e-05,
      "loss": 0.6329,
      "step": 1760
    },
    {
      "epoch": 0.09263620662584393,
      "grad_norm": 0.42644017934799194,
      "learning_rate": 2e-05,
      "loss": 0.6466,
      "step": 1770
    },
    {
      "epoch": 0.09315957502486,
      "grad_norm": 0.40171223878860474,
      "learning_rate": 2e-05,
      "loss": 0.6567,
      "step": 1780
    },
    {
      "epoch": 0.09368294342387606,
      "grad_norm": 0.5002732872962952,
      "learning_rate": 2e-05,
      "loss": 0.6645,
      "step": 1790
    },
    {
      "epoch": 0.09420631182289213,
      "grad_norm": 0.45148345828056335,
      "learning_rate": 2e-05,
      "loss": 0.6735,
      "step": 1800
    },
    {
      "epoch": 0.0947296802219082,
      "grad_norm": 0.4408950209617615,
      "learning_rate": 2e-05,
      "loss": 0.64,
      "step": 1810
    },
    {
      "epoch": 0.09525304862092426,
      "grad_norm": 0.41224032640457153,
      "learning_rate": 2e-05,
      "loss": 0.6483,
      "step": 1820
    },
    {
      "epoch": 0.09577641701994033,
      "grad_norm": 0.3857000768184662,
      "learning_rate": 2e-05,
      "loss": 0.6591,
      "step": 1830
    },
    {
      "epoch": 0.09629978541895641,
      "grad_norm": 0.5142344832420349,
      "learning_rate": 2e-05,
      "loss": 0.6613,
      "step": 1840
    },
    {
      "epoch": 0.09682315381797248,
      "grad_norm": 0.43707653880119324,
      "learning_rate": 2e-05,
      "loss": 0.6463,
      "step": 1850
    },
    {
      "epoch": 0.09734652221698854,
      "grad_norm": 0.5951546430587769,
      "learning_rate": 2e-05,
      "loss": 0.6634,
      "step": 1860
    },
    {
      "epoch": 0.09786989061600461,
      "grad_norm": 0.451498419046402,
      "learning_rate": 2e-05,
      "loss": 0.6228,
      "step": 1870
    },
    {
      "epoch": 0.09839325901502068,
      "grad_norm": 0.43872302770614624,
      "learning_rate": 2e-05,
      "loss": 0.6417,
      "step": 1880
    },
    {
      "epoch": 0.09891662741403674,
      "grad_norm": 0.3905869722366333,
      "learning_rate": 2e-05,
      "loss": 0.6195,
      "step": 1890
    },
    {
      "epoch": 0.09943999581305281,
      "grad_norm": 0.46226704120635986,
      "learning_rate": 2e-05,
      "loss": 0.652,
      "step": 1900
    },
    {
      "epoch": 0.09996336421206888,
      "grad_norm": 0.4350249469280243,
      "learning_rate": 2e-05,
      "loss": 0.6814,
      "step": 1910
    },
    {
      "epoch": 0.10048673261108494,
      "grad_norm": 0.547003448009491,
      "learning_rate": 2e-05,
      "loss": 0.6376,
      "step": 1920
    },
    {
      "epoch": 0.10101010101010101,
      "grad_norm": 0.40520697832107544,
      "learning_rate": 2e-05,
      "loss": 0.6281,
      "step": 1930
    },
    {
      "epoch": 0.10153346940911707,
      "grad_norm": 0.3886966109275818,
      "learning_rate": 2e-05,
      "loss": 0.6349,
      "step": 1940
    },
    {
      "epoch": 0.10205683780813314,
      "grad_norm": 0.4416722059249878,
      "learning_rate": 2e-05,
      "loss": 0.6626,
      "step": 1950
    },
    {
      "epoch": 0.10258020620714921,
      "grad_norm": 0.4272310137748718,
      "learning_rate": 2e-05,
      "loss": 0.627,
      "step": 1960
    },
    {
      "epoch": 0.10310357460616527,
      "grad_norm": 0.4342178404331207,
      "learning_rate": 2e-05,
      "loss": 0.6616,
      "step": 1970
    },
    {
      "epoch": 0.10362694300518134,
      "grad_norm": 0.4413588047027588,
      "learning_rate": 2e-05,
      "loss": 0.6602,
      "step": 1980
    },
    {
      "epoch": 0.10415031140419742,
      "grad_norm": 0.39807382225990295,
      "learning_rate": 2e-05,
      "loss": 0.6184,
      "step": 1990
    },
    {
      "epoch": 0.10467367980321349,
      "grad_norm": 0.39922189712524414,
      "learning_rate": 2e-05,
      "loss": 0.6348,
      "step": 2000
    },
    {
      "epoch": 0.10519704820222955,
      "grad_norm": 0.4405508041381836,
      "learning_rate": 2e-05,
      "loss": 0.6527,
      "step": 2010
    },
    {
      "epoch": 0.10572041660124562,
      "grad_norm": 0.4154644310474396,
      "learning_rate": 2e-05,
      "loss": 0.6383,
      "step": 2020
    },
    {
      "epoch": 0.10624378500026169,
      "grad_norm": 0.4121659994125366,
      "learning_rate": 2e-05,
      "loss": 0.6337,
      "step": 2030
    },
    {
      "epoch": 0.10676715339927775,
      "grad_norm": 0.4231645166873932,
      "learning_rate": 2e-05,
      "loss": 0.6648,
      "step": 2040
    },
    {
      "epoch": 0.10729052179829382,
      "grad_norm": 0.5495308041572571,
      "learning_rate": 2e-05,
      "loss": 0.6655,
      "step": 2050
    },
    {
      "epoch": 0.10781389019730989,
      "grad_norm": 0.4083961546421051,
      "learning_rate": 2e-05,
      "loss": 0.6091,
      "step": 2060
    },
    {
      "epoch": 0.10833725859632595,
      "grad_norm": 0.394467830657959,
      "learning_rate": 2e-05,
      "loss": 0.6472,
      "step": 2070
    },
    {
      "epoch": 0.10886062699534202,
      "grad_norm": 0.4051300585269928,
      "learning_rate": 2e-05,
      "loss": 0.6295,
      "step": 2080
    },
    {
      "epoch": 0.10938399539435809,
      "grad_norm": 0.44103720784187317,
      "learning_rate": 2e-05,
      "loss": 0.6316,
      "step": 2090
    },
    {
      "epoch": 0.10990736379337415,
      "grad_norm": 0.4452396333217621,
      "learning_rate": 2e-05,
      "loss": 0.6367,
      "step": 2100
    },
    {
      "epoch": 0.11043073219239022,
      "grad_norm": 0.5510708689689636,
      "learning_rate": 2e-05,
      "loss": 0.6258,
      "step": 2110
    },
    {
      "epoch": 0.11095410059140629,
      "grad_norm": 0.5372933149337769,
      "learning_rate": 2e-05,
      "loss": 0.6476,
      "step": 2120
    },
    {
      "epoch": 0.11147746899042235,
      "grad_norm": 0.4383063316345215,
      "learning_rate": 2e-05,
      "loss": 0.6605,
      "step": 2130
    },
    {
      "epoch": 0.11200083738943843,
      "grad_norm": 0.4241836667060852,
      "learning_rate": 2e-05,
      "loss": 0.6362,
      "step": 2140
    },
    {
      "epoch": 0.1125242057884545,
      "grad_norm": 0.442451149225235,
      "learning_rate": 2e-05,
      "loss": 0.6433,
      "step": 2150
    },
    {
      "epoch": 0.11304757418747056,
      "grad_norm": 0.41505861282348633,
      "learning_rate": 2e-05,
      "loss": 0.6239,
      "step": 2160
    },
    {
      "epoch": 0.11357094258648663,
      "grad_norm": 0.43439918756484985,
      "learning_rate": 2e-05,
      "loss": 0.6274,
      "step": 2170
    },
    {
      "epoch": 0.1140943109855027,
      "grad_norm": 0.43937069177627563,
      "learning_rate": 2e-05,
      "loss": 0.6414,
      "step": 2180
    },
    {
      "epoch": 0.11461767938451876,
      "grad_norm": 0.43210020661354065,
      "learning_rate": 2e-05,
      "loss": 0.6257,
      "step": 2190
    },
    {
      "epoch": 0.11514104778353483,
      "grad_norm": 0.49416911602020264,
      "learning_rate": 2e-05,
      "loss": 0.6442,
      "step": 2200
    },
    {
      "epoch": 0.1156644161825509,
      "grad_norm": 0.40978503227233887,
      "learning_rate": 2e-05,
      "loss": 0.6329,
      "step": 2210
    },
    {
      "epoch": 0.11618778458156696,
      "grad_norm": 0.4340236485004425,
      "learning_rate": 2e-05,
      "loss": 0.6347,
      "step": 2220
    },
    {
      "epoch": 0.11671115298058303,
      "grad_norm": 0.41441839933395386,
      "learning_rate": 2e-05,
      "loss": 0.6205,
      "step": 2230
    },
    {
      "epoch": 0.1172345213795991,
      "grad_norm": 0.4365447461605072,
      "learning_rate": 2e-05,
      "loss": 0.6329,
      "step": 2240
    },
    {
      "epoch": 0.11775788977861516,
      "grad_norm": 0.44118615984916687,
      "learning_rate": 2e-05,
      "loss": 0.6549,
      "step": 2250
    },
    {
      "epoch": 0.11828125817763123,
      "grad_norm": 0.40321072936058044,
      "learning_rate": 2e-05,
      "loss": 0.6284,
      "step": 2260
    },
    {
      "epoch": 0.1188046265766473,
      "grad_norm": 0.4358213543891907,
      "learning_rate": 2e-05,
      "loss": 0.6375,
      "step": 2270
    },
    {
      "epoch": 0.11932799497566338,
      "grad_norm": 0.42884060740470886,
      "learning_rate": 2e-05,
      "loss": 0.6452,
      "step": 2280
    },
    {
      "epoch": 0.11985136337467944,
      "grad_norm": 0.43688151240348816,
      "learning_rate": 2e-05,
      "loss": 0.6433,
      "step": 2290
    },
    {
      "epoch": 0.12037473177369551,
      "grad_norm": 0.42834022641181946,
      "learning_rate": 2e-05,
      "loss": 0.6461,
      "step": 2300
    },
    {
      "epoch": 0.12089810017271158,
      "grad_norm": 0.44969525933265686,
      "learning_rate": 2e-05,
      "loss": 0.6704,
      "step": 2310
    },
    {
      "epoch": 0.12142146857172764,
      "grad_norm": 0.4053316116333008,
      "learning_rate": 2e-05,
      "loss": 0.6188,
      "step": 2320
    },
    {
      "epoch": 0.12194483697074371,
      "grad_norm": 0.3716852366924286,
      "learning_rate": 2e-05,
      "loss": 0.6249,
      "step": 2330
    },
    {
      "epoch": 0.12246820536975978,
      "grad_norm": 0.4359619617462158,
      "learning_rate": 2e-05,
      "loss": 0.6571,
      "step": 2340
    },
    {
      "epoch": 0.12299157376877584,
      "grad_norm": 0.38862133026123047,
      "learning_rate": 2e-05,
      "loss": 0.641,
      "step": 2350
    },
    {
      "epoch": 0.12351494216779191,
      "grad_norm": 0.4286133646965027,
      "learning_rate": 2e-05,
      "loss": 0.6514,
      "step": 2360
    },
    {
      "epoch": 0.12403831056680797,
      "grad_norm": 0.4758356809616089,
      "learning_rate": 2e-05,
      "loss": 0.628,
      "step": 2370
    },
    {
      "epoch": 0.12456167896582404,
      "grad_norm": 0.4074471592903137,
      "learning_rate": 2e-05,
      "loss": 0.6161,
      "step": 2380
    },
    {
      "epoch": 0.1250850473648401,
      "grad_norm": 0.4395975172519684,
      "learning_rate": 2e-05,
      "loss": 0.6514,
      "step": 2390
    },
    {
      "epoch": 0.12560841576385617,
      "grad_norm": 0.4212956130504608,
      "learning_rate": 2e-05,
      "loss": 0.6459,
      "step": 2400
    },
    {
      "epoch": 0.12613178416287224,
      "grad_norm": 0.43518325686454773,
      "learning_rate": 2e-05,
      "loss": 0.6631,
      "step": 2410
    },
    {
      "epoch": 0.1266551525618883,
      "grad_norm": 0.41669315099716187,
      "learning_rate": 2e-05,
      "loss": 0.6433,
      "step": 2420
    },
    {
      "epoch": 0.12717852096090437,
      "grad_norm": 0.47521886229515076,
      "learning_rate": 2e-05,
      "loss": 0.6008,
      "step": 2430
    },
    {
      "epoch": 0.12770188935992044,
      "grad_norm": 0.5306794047355652,
      "learning_rate": 2e-05,
      "loss": 0.6499,
      "step": 2440
    },
    {
      "epoch": 0.1282252577589365,
      "grad_norm": 0.4078957438468933,
      "learning_rate": 2e-05,
      "loss": 0.632,
      "step": 2450
    },
    {
      "epoch": 0.12874862615795257,
      "grad_norm": 0.4263654351234436,
      "learning_rate": 2e-05,
      "loss": 0.6367,
      "step": 2460
    },
    {
      "epoch": 0.12927199455696864,
      "grad_norm": 0.4139572083950043,
      "learning_rate": 2e-05,
      "loss": 0.642,
      "step": 2470
    },
    {
      "epoch": 0.1297953629559847,
      "grad_norm": 0.38871797919273376,
      "learning_rate": 2e-05,
      "loss": 0.6342,
      "step": 2480
    },
    {
      "epoch": 0.13031873135500077,
      "grad_norm": 0.429225891828537,
      "learning_rate": 2e-05,
      "loss": 0.6249,
      "step": 2490
    },
    {
      "epoch": 0.13084209975401687,
      "grad_norm": 0.45410633087158203,
      "learning_rate": 2e-05,
      "loss": 0.6395,
      "step": 2500
    },
    {
      "epoch": 0.13136546815303293,
      "grad_norm": 0.45400431752204895,
      "learning_rate": 2e-05,
      "loss": 0.6244,
      "step": 2510
    },
    {
      "epoch": 0.131888836552049,
      "grad_norm": 0.4065890312194824,
      "learning_rate": 2e-05,
      "loss": 0.6394,
      "step": 2520
    },
    {
      "epoch": 0.13241220495106507,
      "grad_norm": 0.4191737771034241,
      "learning_rate": 2e-05,
      "loss": 0.6409,
      "step": 2530
    },
    {
      "epoch": 0.13293557335008113,
      "grad_norm": 0.45096731185913086,
      "learning_rate": 2e-05,
      "loss": 0.6208,
      "step": 2540
    },
    {
      "epoch": 0.1334589417490972,
      "grad_norm": 0.46258044242858887,
      "learning_rate": 2e-05,
      "loss": 0.6395,
      "step": 2550
    },
    {
      "epoch": 0.13398231014811327,
      "grad_norm": 0.4282276928424835,
      "learning_rate": 2e-05,
      "loss": 0.601,
      "step": 2560
    },
    {
      "epoch": 0.13450567854712933,
      "grad_norm": 0.41605663299560547,
      "learning_rate": 2e-05,
      "loss": 0.6501,
      "step": 2570
    },
    {
      "epoch": 0.1350290469461454,
      "grad_norm": 0.41660454869270325,
      "learning_rate": 2e-05,
      "loss": 0.6546,
      "step": 2580
    },
    {
      "epoch": 0.13555241534516146,
      "grad_norm": 0.4275190234184265,
      "learning_rate": 2e-05,
      "loss": 0.6458,
      "step": 2590
    },
    {
      "epoch": 0.13607578374417753,
      "grad_norm": 0.4305119514465332,
      "learning_rate": 2e-05,
      "loss": 0.6403,
      "step": 2600
    },
    {
      "epoch": 0.1365991521431936,
      "grad_norm": 0.42454832792282104,
      "learning_rate": 2e-05,
      "loss": 0.6368,
      "step": 2610
    },
    {
      "epoch": 0.13712252054220966,
      "grad_norm": 0.44257181882858276,
      "learning_rate": 2e-05,
      "loss": 0.6343,
      "step": 2620
    },
    {
      "epoch": 0.13764588894122573,
      "grad_norm": 0.3995290696620941,
      "learning_rate": 2e-05,
      "loss": 0.6616,
      "step": 2630
    },
    {
      "epoch": 0.1381692573402418,
      "grad_norm": 0.4199754297733307,
      "learning_rate": 2e-05,
      "loss": 0.6286,
      "step": 2640
    },
    {
      "epoch": 0.13869262573925786,
      "grad_norm": 0.42020726203918457,
      "learning_rate": 2e-05,
      "loss": 0.6371,
      "step": 2650
    },
    {
      "epoch": 0.13921599413827393,
      "grad_norm": 0.39368361234664917,
      "learning_rate": 2e-05,
      "loss": 0.6517,
      "step": 2660
    },
    {
      "epoch": 0.13973936253729,
      "grad_norm": 0.4302023947238922,
      "learning_rate": 2e-05,
      "loss": 0.6139,
      "step": 2670
    },
    {
      "epoch": 0.14026273093630606,
      "grad_norm": 0.49658092856407166,
      "learning_rate": 2e-05,
      "loss": 0.6565,
      "step": 2680
    },
    {
      "epoch": 0.14078609933532213,
      "grad_norm": 0.3941155672073364,
      "learning_rate": 2e-05,
      "loss": 0.6527,
      "step": 2690
    },
    {
      "epoch": 0.1413094677343382,
      "grad_norm": 0.43608132004737854,
      "learning_rate": 2e-05,
      "loss": 0.6309,
      "step": 2700
    },
    {
      "epoch": 0.14183283613335426,
      "grad_norm": 0.4120804965496063,
      "learning_rate": 2e-05,
      "loss": 0.6326,
      "step": 2710
    },
    {
      "epoch": 0.14235620453237033,
      "grad_norm": 0.4378657639026642,
      "learning_rate": 2e-05,
      "loss": 0.6479,
      "step": 2720
    },
    {
      "epoch": 0.1428795729313864,
      "grad_norm": 0.39626598358154297,
      "learning_rate": 2e-05,
      "loss": 0.6537,
      "step": 2730
    },
    {
      "epoch": 0.14340294133040246,
      "grad_norm": 0.40195828676223755,
      "learning_rate": 2e-05,
      "loss": 0.6075,
      "step": 2740
    },
    {
      "epoch": 0.14392630972941853,
      "grad_norm": 0.3953752815723419,
      "learning_rate": 2e-05,
      "loss": 0.6179,
      "step": 2750
    },
    {
      "epoch": 0.1444496781284346,
      "grad_norm": 0.4433000385761261,
      "learning_rate": 2e-05,
      "loss": 0.6743,
      "step": 2760
    },
    {
      "epoch": 0.14497304652745066,
      "grad_norm": 0.42893922328948975,
      "learning_rate": 2e-05,
      "loss": 0.6272,
      "step": 2770
    },
    {
      "epoch": 0.14549641492646673,
      "grad_norm": 0.4065708518028259,
      "learning_rate": 2e-05,
      "loss": 0.6417,
      "step": 2780
    },
    {
      "epoch": 0.1460197833254828,
      "grad_norm": 0.4181337058544159,
      "learning_rate": 2e-05,
      "loss": 0.6392,
      "step": 2790
    },
    {
      "epoch": 0.1465431517244989,
      "grad_norm": 0.4179243743419647,
      "learning_rate": 2e-05,
      "loss": 0.6026,
      "step": 2800
    },
    {
      "epoch": 0.14706652012351495,
      "grad_norm": 0.46388933062553406,
      "learning_rate": 2e-05,
      "loss": 0.6335,
      "step": 2810
    },
    {
      "epoch": 0.14758988852253102,
      "grad_norm": 0.4350208640098572,
      "learning_rate": 2e-05,
      "loss": 0.6349,
      "step": 2820
    },
    {
      "epoch": 0.1481132569215471,
      "grad_norm": 0.4281371831893921,
      "learning_rate": 2e-05,
      "loss": 0.5927,
      "step": 2830
    },
    {
      "epoch": 0.14863662532056315,
      "grad_norm": 0.3796765208244324,
      "learning_rate": 2e-05,
      "loss": 0.6244,
      "step": 2840
    },
    {
      "epoch": 0.14915999371957922,
      "grad_norm": 0.400057315826416,
      "learning_rate": 2e-05,
      "loss": 0.6083,
      "step": 2850
    },
    {
      "epoch": 0.1496833621185953,
      "grad_norm": 0.5580277442932129,
      "learning_rate": 2e-05,
      "loss": 0.6088,
      "step": 2860
    },
    {
      "epoch": 0.15020673051761135,
      "grad_norm": 0.4220695197582245,
      "learning_rate": 2e-05,
      "loss": 0.6339,
      "step": 2870
    },
    {
      "epoch": 0.15073009891662742,
      "grad_norm": 0.4196612536907196,
      "learning_rate": 2e-05,
      "loss": 0.6415,
      "step": 2880
    },
    {
      "epoch": 0.1512534673156435,
      "grad_norm": 0.4167044162750244,
      "learning_rate": 2e-05,
      "loss": 0.6276,
      "step": 2890
    },
    {
      "epoch": 0.15177683571465955,
      "grad_norm": 0.4121061861515045,
      "learning_rate": 2e-05,
      "loss": 0.6311,
      "step": 2900
    },
    {
      "epoch": 0.15230020411367562,
      "grad_norm": 0.4583412706851959,
      "learning_rate": 2e-05,
      "loss": 0.6415,
      "step": 2910
    },
    {
      "epoch": 0.15282357251269169,
      "grad_norm": 0.4203738272190094,
      "learning_rate": 2e-05,
      "loss": 0.657,
      "step": 2920
    },
    {
      "epoch": 0.15334694091170775,
      "grad_norm": 0.4278572201728821,
      "learning_rate": 2e-05,
      "loss": 0.6457,
      "step": 2930
    },
    {
      "epoch": 0.15387030931072382,
      "grad_norm": 0.41517725586891174,
      "learning_rate": 2e-05,
      "loss": 0.6506,
      "step": 2940
    },
    {
      "epoch": 0.15439367770973988,
      "grad_norm": 0.39440619945526123,
      "learning_rate": 2e-05,
      "loss": 0.6247,
      "step": 2950
    },
    {
      "epoch": 0.15491704610875595,
      "grad_norm": 0.3762326240539551,
      "learning_rate": 2e-05,
      "loss": 0.6464,
      "step": 2960
    },
    {
      "epoch": 0.15544041450777202,
      "grad_norm": 0.43617796897888184,
      "learning_rate": 2e-05,
      "loss": 0.6162,
      "step": 2970
    },
    {
      "epoch": 0.15596378290678808,
      "grad_norm": 0.41636911034584045,
      "learning_rate": 2e-05,
      "loss": 0.6327,
      "step": 2980
    },
    {
      "epoch": 0.15648715130580415,
      "grad_norm": 0.4136853814125061,
      "learning_rate": 2e-05,
      "loss": 0.6292,
      "step": 2990
    },
    {
      "epoch": 0.15701051970482022,
      "grad_norm": 0.44505634903907776,
      "learning_rate": 2e-05,
      "loss": 0.646,
      "step": 3000
    },
    {
      "epoch": 0.15753388810383628,
      "grad_norm": 0.42681798338890076,
      "learning_rate": 2e-05,
      "loss": 0.6274,
      "step": 3010
    },
    {
      "epoch": 0.15805725650285235,
      "grad_norm": 0.3740960657596588,
      "learning_rate": 2e-05,
      "loss": 0.6233,
      "step": 3020
    },
    {
      "epoch": 0.15858062490186842,
      "grad_norm": 0.4044531285762787,
      "learning_rate": 2e-05,
      "loss": 0.655,
      "step": 3030
    },
    {
      "epoch": 0.15910399330088448,
      "grad_norm": 0.38664835691452026,
      "learning_rate": 2e-05,
      "loss": 0.6201,
      "step": 3040
    },
    {
      "epoch": 0.15962736169990055,
      "grad_norm": 0.5274628400802612,
      "learning_rate": 2e-05,
      "loss": 0.6374,
      "step": 3050
    },
    {
      "epoch": 0.16015073009891662,
      "grad_norm": 0.4279114305973053,
      "learning_rate": 2e-05,
      "loss": 0.6433,
      "step": 3060
    },
    {
      "epoch": 0.16067409849793268,
      "grad_norm": 0.396634578704834,
      "learning_rate": 2e-05,
      "loss": 0.6155,
      "step": 3070
    },
    {
      "epoch": 0.16119746689694875,
      "grad_norm": 0.41256171464920044,
      "learning_rate": 2e-05,
      "loss": 0.6274,
      "step": 3080
    },
    {
      "epoch": 0.16172083529596484,
      "grad_norm": 0.47806742787361145,
      "learning_rate": 2e-05,
      "loss": 0.6288,
      "step": 3090
    },
    {
      "epoch": 0.1622442036949809,
      "grad_norm": 0.41482001543045044,
      "learning_rate": 2e-05,
      "loss": 0.6097,
      "step": 3100
    },
    {
      "epoch": 0.16276757209399698,
      "grad_norm": 0.4124285876750946,
      "learning_rate": 2e-05,
      "loss": 0.6381,
      "step": 3110
    },
    {
      "epoch": 0.16329094049301304,
      "grad_norm": 0.39308762550354004,
      "learning_rate": 2e-05,
      "loss": 0.6418,
      "step": 3120
    },
    {
      "epoch": 0.1638143088920291,
      "grad_norm": 0.5976954698562622,
      "learning_rate": 2e-05,
      "loss": 0.6403,
      "step": 3130
    },
    {
      "epoch": 0.16433767729104518,
      "grad_norm": 0.4556901752948761,
      "learning_rate": 2e-05,
      "loss": 0.6415,
      "step": 3140
    },
    {
      "epoch": 0.16486104569006124,
      "grad_norm": 0.4634074866771698,
      "learning_rate": 2e-05,
      "loss": 0.6142,
      "step": 3150
    },
    {
      "epoch": 0.1653844140890773,
      "grad_norm": 0.4238816797733307,
      "learning_rate": 2e-05,
      "loss": 0.6528,
      "step": 3160
    },
    {
      "epoch": 0.16590778248809337,
      "grad_norm": 0.5388637185096741,
      "learning_rate": 2e-05,
      "loss": 0.6723,
      "step": 3170
    },
    {
      "epoch": 0.16643115088710944,
      "grad_norm": 0.3934621512889862,
      "learning_rate": 2e-05,
      "loss": 0.6347,
      "step": 3180
    },
    {
      "epoch": 0.1669545192861255,
      "grad_norm": 0.3961109519004822,
      "learning_rate": 2e-05,
      "loss": 0.6345,
      "step": 3190
    },
    {
      "epoch": 0.16747788768514157,
      "grad_norm": 0.41134336590766907,
      "learning_rate": 2e-05,
      "loss": 0.6227,
      "step": 3200
    },
    {
      "epoch": 0.16800125608415764,
      "grad_norm": 0.4064910411834717,
      "learning_rate": 2e-05,
      "loss": 0.6087,
      "step": 3210
    },
    {
      "epoch": 0.1685246244831737,
      "grad_norm": 0.411873459815979,
      "learning_rate": 2e-05,
      "loss": 0.6291,
      "step": 3220
    },
    {
      "epoch": 0.16904799288218977,
      "grad_norm": 0.41122299432754517,
      "learning_rate": 2e-05,
      "loss": 0.6217,
      "step": 3230
    },
    {
      "epoch": 0.16957136128120584,
      "grad_norm": 0.4552735686302185,
      "learning_rate": 2e-05,
      "loss": 0.6334,
      "step": 3240
    },
    {
      "epoch": 0.1700947296802219,
      "grad_norm": 0.457132488489151,
      "learning_rate": 2e-05,
      "loss": 0.6213,
      "step": 3250
    },
    {
      "epoch": 0.17061809807923797,
      "grad_norm": 0.38603243231773376,
      "learning_rate": 2e-05,
      "loss": 0.6363,
      "step": 3260
    },
    {
      "epoch": 0.17114146647825404,
      "grad_norm": 0.4321597218513489,
      "learning_rate": 2e-05,
      "loss": 0.6194,
      "step": 3270
    },
    {
      "epoch": 0.1716648348772701,
      "grad_norm": 0.43034639954566956,
      "learning_rate": 2e-05,
      "loss": 0.6409,
      "step": 3280
    },
    {
      "epoch": 0.17218820327628617,
      "grad_norm": 0.42103973031044006,
      "learning_rate": 2e-05,
      "loss": 0.6154,
      "step": 3290
    },
    {
      "epoch": 0.17271157167530224,
      "grad_norm": 0.3927100896835327,
      "learning_rate": 2e-05,
      "loss": 0.6313,
      "step": 3300
    },
    {
      "epoch": 0.1732349400743183,
      "grad_norm": 0.4339587688446045,
      "learning_rate": 2e-05,
      "loss": 0.614,
      "step": 3310
    },
    {
      "epoch": 0.17375830847333437,
      "grad_norm": 0.40991151332855225,
      "learning_rate": 2e-05,
      "loss": 0.6456,
      "step": 3320
    },
    {
      "epoch": 0.17428167687235044,
      "grad_norm": 0.4325578808784485,
      "learning_rate": 2e-05,
      "loss": 0.6041,
      "step": 3330
    },
    {
      "epoch": 0.1748050452713665,
      "grad_norm": 0.41213691234588623,
      "learning_rate": 2e-05,
      "loss": 0.6349,
      "step": 3340
    },
    {
      "epoch": 0.17532841367038257,
      "grad_norm": 0.47782137989997864,
      "learning_rate": 2e-05,
      "loss": 0.636,
      "step": 3350
    },
    {
      "epoch": 0.17585178206939864,
      "grad_norm": 0.42506250739097595,
      "learning_rate": 2e-05,
      "loss": 0.626,
      "step": 3360
    },
    {
      "epoch": 0.1763751504684147,
      "grad_norm": 0.3940761983394623,
      "learning_rate": 2e-05,
      "loss": 0.6193,
      "step": 3370
    },
    {
      "epoch": 0.17689851886743077,
      "grad_norm": 0.4069708287715912,
      "learning_rate": 2e-05,
      "loss": 0.6365,
      "step": 3380
    },
    {
      "epoch": 0.17742188726644686,
      "grad_norm": 0.504355788230896,
      "learning_rate": 2e-05,
      "loss": 0.5985,
      "step": 3390
    },
    {
      "epoch": 0.17794525566546293,
      "grad_norm": 0.3881029486656189,
      "learning_rate": 2e-05,
      "loss": 0.6328,
      "step": 3400
    },
    {
      "epoch": 0.178468624064479,
      "grad_norm": 2.036409854888916,
      "learning_rate": 2e-05,
      "loss": 0.6237,
      "step": 3410
    },
    {
      "epoch": 0.17899199246349506,
      "grad_norm": 0.41256919503211975,
      "learning_rate": 2e-05,
      "loss": 0.6066,
      "step": 3420
    },
    {
      "epoch": 0.17951536086251113,
      "grad_norm": 0.4488495886325836,
      "learning_rate": 2e-05,
      "loss": 0.6247,
      "step": 3430
    },
    {
      "epoch": 0.1800387292615272,
      "grad_norm": 0.4205232560634613,
      "learning_rate": 2e-05,
      "loss": 0.6094,
      "step": 3440
    },
    {
      "epoch": 0.18056209766054326,
      "grad_norm": 0.4089616537094116,
      "learning_rate": 2e-05,
      "loss": 0.6264,
      "step": 3450
    },
    {
      "epoch": 0.18108546605955933,
      "grad_norm": 0.37557995319366455,
      "learning_rate": 2e-05,
      "loss": 0.6322,
      "step": 3460
    },
    {
      "epoch": 0.1816088344585754,
      "grad_norm": 0.39397501945495605,
      "learning_rate": 2e-05,
      "loss": 0.6647,
      "step": 3470
    },
    {
      "epoch": 0.18213220285759146,
      "grad_norm": 0.4009655714035034,
      "learning_rate": 2e-05,
      "loss": 0.6203,
      "step": 3480
    },
    {
      "epoch": 0.18265557125660753,
      "grad_norm": 0.40983593463897705,
      "learning_rate": 2e-05,
      "loss": 0.6684,
      "step": 3490
    },
    {
      "epoch": 0.1831789396556236,
      "grad_norm": 0.41632992029190063,
      "learning_rate": 2e-05,
      "loss": 0.6055,
      "step": 3500
    },
    {
      "epoch": 0.18370230805463966,
      "grad_norm": 0.42826464772224426,
      "learning_rate": 2e-05,
      "loss": 0.6114,
      "step": 3510
    },
    {
      "epoch": 0.18422567645365573,
      "grad_norm": 0.4451156556606293,
      "learning_rate": 2e-05,
      "loss": 0.6164,
      "step": 3520
    },
    {
      "epoch": 0.1847490448526718,
      "grad_norm": 0.7244383692741394,
      "learning_rate": 2e-05,
      "loss": 0.643,
      "step": 3530
    },
    {
      "epoch": 0.18527241325168786,
      "grad_norm": 0.41192492842674255,
      "learning_rate": 2e-05,
      "loss": 0.6389,
      "step": 3540
    },
    {
      "epoch": 0.18579578165070393,
      "grad_norm": 0.42937004566192627,
      "learning_rate": 2e-05,
      "loss": 0.6449,
      "step": 3550
    },
    {
      "epoch": 0.18631915004972,
      "grad_norm": 0.4111078977584839,
      "learning_rate": 2e-05,
      "loss": 0.6202,
      "step": 3560
    },
    {
      "epoch": 0.18684251844873606,
      "grad_norm": 0.4172969460487366,
      "learning_rate": 2e-05,
      "loss": 0.6188,
      "step": 3570
    },
    {
      "epoch": 0.18736588684775213,
      "grad_norm": 0.38246381282806396,
      "learning_rate": 2e-05,
      "loss": 0.609,
      "step": 3580
    },
    {
      "epoch": 0.1878892552467682,
      "grad_norm": 0.467737078666687,
      "learning_rate": 2e-05,
      "loss": 0.6177,
      "step": 3590
    },
    {
      "epoch": 0.18841262364578426,
      "grad_norm": 0.3787391185760498,
      "learning_rate": 2e-05,
      "loss": 0.6156,
      "step": 3600
    },
    {
      "epoch": 0.18893599204480033,
      "grad_norm": 0.39074644446372986,
      "learning_rate": 2e-05,
      "loss": 0.6275,
      "step": 3610
    },
    {
      "epoch": 0.1894593604438164,
      "grad_norm": 0.4353400468826294,
      "learning_rate": 2e-05,
      "loss": 0.613,
      "step": 3620
    },
    {
      "epoch": 0.18998272884283246,
      "grad_norm": 0.4572504758834839,
      "learning_rate": 2e-05,
      "loss": 0.6243,
      "step": 3630
    },
    {
      "epoch": 0.19050609724184853,
      "grad_norm": 0.3868942856788635,
      "learning_rate": 2e-05,
      "loss": 0.6303,
      "step": 3640
    },
    {
      "epoch": 0.1910294656408646,
      "grad_norm": 0.41320866346359253,
      "learning_rate": 2e-05,
      "loss": 0.6255,
      "step": 3650
    },
    {
      "epoch": 0.19155283403988066,
      "grad_norm": 0.39754143357276917,
      "learning_rate": 2e-05,
      "loss": 0.6064,
      "step": 3660
    },
    {
      "epoch": 0.19207620243889673,
      "grad_norm": 0.45954880118370056,
      "learning_rate": 2e-05,
      "loss": 0.6355,
      "step": 3670
    },
    {
      "epoch": 0.19259957083791282,
      "grad_norm": 0.42518895864486694,
      "learning_rate": 2e-05,
      "loss": 0.6339,
      "step": 3680
    },
    {
      "epoch": 0.1931229392369289,
      "grad_norm": 0.40126708149909973,
      "learning_rate": 2e-05,
      "loss": 0.6226,
      "step": 3690
    },
    {
      "epoch": 0.19364630763594495,
      "grad_norm": 0.5110933184623718,
      "learning_rate": 2e-05,
      "loss": 0.6236,
      "step": 3700
    },
    {
      "epoch": 0.19416967603496102,
      "grad_norm": 0.4530631899833679,
      "learning_rate": 2e-05,
      "loss": 0.6132,
      "step": 3710
    },
    {
      "epoch": 0.19469304443397709,
      "grad_norm": 0.40570759773254395,
      "learning_rate": 2e-05,
      "loss": 0.6205,
      "step": 3720
    },
    {
      "epoch": 0.19521641283299315,
      "grad_norm": 0.4424304962158203,
      "learning_rate": 2e-05,
      "loss": 0.6318,
      "step": 3730
    },
    {
      "epoch": 0.19573978123200922,
      "grad_norm": 0.4104612171649933,
      "learning_rate": 2e-05,
      "loss": 0.6112,
      "step": 3740
    },
    {
      "epoch": 0.19626314963102529,
      "grad_norm": 0.4605982303619385,
      "learning_rate": 2e-05,
      "loss": 0.63,
      "step": 3750
    },
    {
      "epoch": 0.19678651803004135,
      "grad_norm": 0.41476568579673767,
      "learning_rate": 2e-05,
      "loss": 0.6301,
      "step": 3760
    },
    {
      "epoch": 0.19730988642905742,
      "grad_norm": 0.4510325491428375,
      "learning_rate": 2e-05,
      "loss": 0.6311,
      "step": 3770
    },
    {
      "epoch": 0.19783325482807348,
      "grad_norm": 0.3622564673423767,
      "learning_rate": 2e-05,
      "loss": 0.6028,
      "step": 3780
    },
    {
      "epoch": 0.19835662322708955,
      "grad_norm": 0.40758174657821655,
      "learning_rate": 2e-05,
      "loss": 0.6328,
      "step": 3790
    },
    {
      "epoch": 0.19887999162610562,
      "grad_norm": 0.42536821961402893,
      "learning_rate": 2e-05,
      "loss": 0.6343,
      "step": 3800
    },
    {
      "epoch": 0.19940336002512168,
      "grad_norm": 0.4246145486831665,
      "learning_rate": 2e-05,
      "loss": 0.617,
      "step": 3810
    },
    {
      "epoch": 0.19992672842413775,
      "grad_norm": 0.4054490327835083,
      "learning_rate": 2e-05,
      "loss": 0.6374,
      "step": 3820
    },
    {
      "epoch": 0.20045009682315382,
      "grad_norm": 0.4975704848766327,
      "learning_rate": 2e-05,
      "loss": 0.6006,
      "step": 3830
    },
    {
      "epoch": 0.20097346522216988,
      "grad_norm": 0.5053114891052246,
      "learning_rate": 2e-05,
      "loss": 0.6319,
      "step": 3840
    },
    {
      "epoch": 0.20149683362118595,
      "grad_norm": 0.41113340854644775,
      "learning_rate": 2e-05,
      "loss": 0.6342,
      "step": 3850
    },
    {
      "epoch": 0.20202020202020202,
      "grad_norm": 0.3836223781108856,
      "learning_rate": 2e-05,
      "loss": 0.6196,
      "step": 3860
    },
    {
      "epoch": 0.20254357041921808,
      "grad_norm": 0.3937544822692871,
      "learning_rate": 2e-05,
      "loss": 0.641,
      "step": 3870
    },
    {
      "epoch": 0.20306693881823415,
      "grad_norm": 0.4237481951713562,
      "learning_rate": 2e-05,
      "loss": 0.6125,
      "step": 3880
    },
    {
      "epoch": 0.20359030721725022,
      "grad_norm": 0.3751864731311798,
      "learning_rate": 2e-05,
      "loss": 0.6197,
      "step": 3890
    },
    {
      "epoch": 0.20411367561626628,
      "grad_norm": 0.5081929564476013,
      "learning_rate": 2e-05,
      "loss": 0.6203,
      "step": 3900
    },
    {
      "epoch": 0.20463704401528235,
      "grad_norm": 0.4540235698223114,
      "learning_rate": 2e-05,
      "loss": 0.6168,
      "step": 3910
    },
    {
      "epoch": 0.20516041241429842,
      "grad_norm": 0.41905754804611206,
      "learning_rate": 2e-05,
      "loss": 0.616,
      "step": 3920
    },
    {
      "epoch": 0.20568378081331448,
      "grad_norm": 0.5326279997825623,
      "learning_rate": 2e-05,
      "loss": 0.6211,
      "step": 3930
    },
    {
      "epoch": 0.20620714921233055,
      "grad_norm": 0.41054078936576843,
      "learning_rate": 2e-05,
      "loss": 0.6113,
      "step": 3940
    },
    {
      "epoch": 0.20673051761134661,
      "grad_norm": 0.4346046447753906,
      "learning_rate": 2e-05,
      "loss": 0.6495,
      "step": 3950
    },
    {
      "epoch": 0.20725388601036268,
      "grad_norm": 0.410095751285553,
      "learning_rate": 2e-05,
      "loss": 0.62,
      "step": 3960
    },
    {
      "epoch": 0.20777725440937878,
      "grad_norm": 0.40288132429122925,
      "learning_rate": 2e-05,
      "loss": 0.6214,
      "step": 3970
    },
    {
      "epoch": 0.20830062280839484,
      "grad_norm": 0.40218082070350647,
      "learning_rate": 2e-05,
      "loss": 0.6318,
      "step": 3980
    },
    {
      "epoch": 0.2088239912074109,
      "grad_norm": 0.43279343843460083,
      "learning_rate": 2e-05,
      "loss": 0.6341,
      "step": 3990
    },
    {
      "epoch": 0.20934735960642697,
      "grad_norm": 0.39592084288597107,
      "learning_rate": 2e-05,
      "loss": 0.6042,
      "step": 4000
    },
    {
      "epoch": 0.20987072800544304,
      "grad_norm": 0.4071950912475586,
      "learning_rate": 2e-05,
      "loss": 0.5843,
      "step": 4010
    },
    {
      "epoch": 0.2103940964044591,
      "grad_norm": 0.3897649347782135,
      "learning_rate": 2e-05,
      "loss": 0.6269,
      "step": 4020
    },
    {
      "epoch": 0.21091746480347517,
      "grad_norm": 0.7160203456878662,
      "learning_rate": 2e-05,
      "loss": 0.6499,
      "step": 4030
    },
    {
      "epoch": 0.21144083320249124,
      "grad_norm": 0.38851386308670044,
      "learning_rate": 2e-05,
      "loss": 0.6142,
      "step": 4040
    },
    {
      "epoch": 0.2119642016015073,
      "grad_norm": 0.3943970799446106,
      "learning_rate": 2e-05,
      "loss": 0.5983,
      "step": 4050
    },
    {
      "epoch": 0.21248757000052337,
      "grad_norm": 0.41590937972068787,
      "learning_rate": 2e-05,
      "loss": 0.6363,
      "step": 4060
    },
    {
      "epoch": 0.21301093839953944,
      "grad_norm": 0.38856077194213867,
      "learning_rate": 2e-05,
      "loss": 0.611,
      "step": 4070
    },
    {
      "epoch": 0.2135343067985555,
      "grad_norm": 0.4386836886405945,
      "learning_rate": 2e-05,
      "loss": 0.6145,
      "step": 4080
    },
    {
      "epoch": 0.21405767519757157,
      "grad_norm": 0.46340975165367126,
      "learning_rate": 2e-05,
      "loss": 0.5949,
      "step": 4090
    },
    {
      "epoch": 0.21458104359658764,
      "grad_norm": 0.4160849153995514,
      "learning_rate": 2e-05,
      "loss": 0.6446,
      "step": 4100
    },
    {
      "epoch": 0.2151044119956037,
      "grad_norm": 0.4251480996608734,
      "learning_rate": 2e-05,
      "loss": 0.6031,
      "step": 4110
    },
    {
      "epoch": 0.21562778039461977,
      "grad_norm": 0.41039806604385376,
      "learning_rate": 2e-05,
      "loss": 0.6165,
      "step": 4120
    },
    {
      "epoch": 0.21615114879363584,
      "grad_norm": 0.4157627522945404,
      "learning_rate": 2e-05,
      "loss": 0.6219,
      "step": 4130
    },
    {
      "epoch": 0.2166745171926519,
      "grad_norm": 0.3525838553905487,
      "learning_rate": 2e-05,
      "loss": 0.6062,
      "step": 4140
    },
    {
      "epoch": 0.21719788559166797,
      "grad_norm": 0.4129687249660492,
      "learning_rate": 2e-05,
      "loss": 0.6104,
      "step": 4150
    },
    {
      "epoch": 0.21772125399068404,
      "grad_norm": 0.4055493175983429,
      "learning_rate": 2e-05,
      "loss": 0.5947,
      "step": 4160
    },
    {
      "epoch": 0.2182446223897001,
      "grad_norm": 0.42468446493148804,
      "learning_rate": 2e-05,
      "loss": 0.6507,
      "step": 4170
    },
    {
      "epoch": 0.21876799078871617,
      "grad_norm": 0.4344474673271179,
      "learning_rate": 2e-05,
      "loss": 0.6062,
      "step": 4180
    },
    {
      "epoch": 0.21929135918773224,
      "grad_norm": 0.4484303295612335,
      "learning_rate": 2e-05,
      "loss": 0.6128,
      "step": 4190
    },
    {
      "epoch": 0.2198147275867483,
      "grad_norm": 0.38424041867256165,
      "learning_rate": 2e-05,
      "loss": 0.611,
      "step": 4200
    },
    {
      "epoch": 0.22033809598576437,
      "grad_norm": 0.40774214267730713,
      "learning_rate": 2e-05,
      "loss": 0.6128,
      "step": 4210
    },
    {
      "epoch": 0.22086146438478044,
      "grad_norm": 0.39545294642448425,
      "learning_rate": 2e-05,
      "loss": 0.6196,
      "step": 4220
    },
    {
      "epoch": 0.2213848327837965,
      "grad_norm": 0.42367228865623474,
      "learning_rate": 2e-05,
      "loss": 0.6324,
      "step": 4230
    },
    {
      "epoch": 0.22190820118281257,
      "grad_norm": 0.4583156406879425,
      "learning_rate": 2e-05,
      "loss": 0.6011,
      "step": 4240
    },
    {
      "epoch": 0.22243156958182864,
      "grad_norm": 0.36509275436401367,
      "learning_rate": 2e-05,
      "loss": 0.6227,
      "step": 4250
    },
    {
      "epoch": 0.2229549379808447,
      "grad_norm": 0.41607242822647095,
      "learning_rate": 2e-05,
      "loss": 0.625,
      "step": 4260
    },
    {
      "epoch": 0.2234783063798608,
      "grad_norm": 0.41028332710266113,
      "learning_rate": 2e-05,
      "loss": 0.5968,
      "step": 4270
    },
    {
      "epoch": 0.22400167477887686,
      "grad_norm": 0.4013286530971527,
      "learning_rate": 2e-05,
      "loss": 0.6143,
      "step": 4280
    },
    {
      "epoch": 0.22452504317789293,
      "grad_norm": 0.42334598302841187,
      "learning_rate": 2e-05,
      "loss": 0.6036,
      "step": 4290
    },
    {
      "epoch": 0.225048411576909,
      "grad_norm": 0.47744619846343994,
      "learning_rate": 2e-05,
      "loss": 0.6329,
      "step": 4300
    },
    {
      "epoch": 0.22557177997592506,
      "grad_norm": 0.378061443567276,
      "learning_rate": 2e-05,
      "loss": 0.6191,
      "step": 4310
    },
    {
      "epoch": 0.22609514837494113,
      "grad_norm": 0.4232480823993683,
      "learning_rate": 2e-05,
      "loss": 0.6167,
      "step": 4320
    },
    {
      "epoch": 0.2266185167739572,
      "grad_norm": 0.3921084702014923,
      "learning_rate": 2e-05,
      "loss": 0.6363,
      "step": 4330
    },
    {
      "epoch": 0.22714188517297326,
      "grad_norm": 0.3987734913825989,
      "learning_rate": 2e-05,
      "loss": 0.611,
      "step": 4340
    },
    {
      "epoch": 0.22766525357198933,
      "grad_norm": 0.4001140296459198,
      "learning_rate": 2e-05,
      "loss": 0.6025,
      "step": 4350
    },
    {
      "epoch": 0.2281886219710054,
      "grad_norm": 0.4162245988845825,
      "learning_rate": 2e-05,
      "loss": 0.6151,
      "step": 4360
    },
    {
      "epoch": 0.22871199037002146,
      "grad_norm": 0.4258664548397064,
      "learning_rate": 2e-05,
      "loss": 0.6107,
      "step": 4370
    },
    {
      "epoch": 0.22923535876903753,
      "grad_norm": 0.39507806301116943,
      "learning_rate": 2e-05,
      "loss": 0.6155,
      "step": 4380
    },
    {
      "epoch": 0.2297587271680536,
      "grad_norm": 0.49936944246292114,
      "learning_rate": 2e-05,
      "loss": 0.6367,
      "step": 4390
    },
    {
      "epoch": 0.23028209556706966,
      "grad_norm": 0.4109865128993988,
      "learning_rate": 2e-05,
      "loss": 0.6163,
      "step": 4400
    },
    {
      "epoch": 0.23080546396608573,
      "grad_norm": 0.4059872031211853,
      "learning_rate": 2e-05,
      "loss": 0.6087,
      "step": 4410
    },
    {
      "epoch": 0.2313288323651018,
      "grad_norm": 0.41509491205215454,
      "learning_rate": 2e-05,
      "loss": 0.6034,
      "step": 4420
    },
    {
      "epoch": 0.23185220076411786,
      "grad_norm": 0.4001082181930542,
      "learning_rate": 2e-05,
      "loss": 0.5977,
      "step": 4430
    },
    {
      "epoch": 0.23237556916313393,
      "grad_norm": 0.4086252748966217,
      "learning_rate": 2e-05,
      "loss": 0.606,
      "step": 4440
    },
    {
      "epoch": 0.23289893756215,
      "grad_norm": 0.3729661703109741,
      "learning_rate": 2e-05,
      "loss": 0.6201,
      "step": 4450
    },
    {
      "epoch": 0.23342230596116606,
      "grad_norm": 0.39218300580978394,
      "learning_rate": 2e-05,
      "loss": 0.6174,
      "step": 4460
    },
    {
      "epoch": 0.23394567436018213,
      "grad_norm": 0.3822868764400482,
      "learning_rate": 2e-05,
      "loss": 0.6395,
      "step": 4470
    },
    {
      "epoch": 0.2344690427591982,
      "grad_norm": 0.4116714596748352,
      "learning_rate": 2e-05,
      "loss": 0.6101,
      "step": 4480
    },
    {
      "epoch": 0.23499241115821426,
      "grad_norm": 0.4040875732898712,
      "learning_rate": 2e-05,
      "loss": 0.626,
      "step": 4490
    },
    {
      "epoch": 0.23551577955723033,
      "grad_norm": 0.3950350880622864,
      "learning_rate": 2e-05,
      "loss": 0.5975,
      "step": 4500
    },
    {
      "epoch": 0.2360391479562464,
      "grad_norm": 0.39547741413116455,
      "learning_rate": 2e-05,
      "loss": 0.6092,
      "step": 4510
    },
    {
      "epoch": 0.23656251635526246,
      "grad_norm": 0.4103192985057831,
      "learning_rate": 2e-05,
      "loss": 0.5902,
      "step": 4520
    },
    {
      "epoch": 0.23708588475427853,
      "grad_norm": 0.4051879048347473,
      "learning_rate": 2e-05,
      "loss": 0.6064,
      "step": 4530
    },
    {
      "epoch": 0.2376092531532946,
      "grad_norm": 0.4002724885940552,
      "learning_rate": 2e-05,
      "loss": 0.618,
      "step": 4540
    },
    {
      "epoch": 0.23813262155231066,
      "grad_norm": 0.4022727310657501,
      "learning_rate": 2e-05,
      "loss": 0.6145,
      "step": 4550
    },
    {
      "epoch": 0.23865598995132675,
      "grad_norm": 0.4159553050994873,
      "learning_rate": 2e-05,
      "loss": 0.5922,
      "step": 4560
    },
    {
      "epoch": 0.23917935835034282,
      "grad_norm": 0.3999551236629486,
      "learning_rate": 2e-05,
      "loss": 0.5905,
      "step": 4570
    },
    {
      "epoch": 0.23970272674935889,
      "grad_norm": 0.4089633822441101,
      "learning_rate": 2e-05,
      "loss": 0.6132,
      "step": 4580
    },
    {
      "epoch": 0.24022609514837495,
      "grad_norm": 0.37393105030059814,
      "learning_rate": 2e-05,
      "loss": 0.6094,
      "step": 4590
    },
    {
      "epoch": 0.24074946354739102,
      "grad_norm": 0.6190505623817444,
      "learning_rate": 2e-05,
      "loss": 0.6066,
      "step": 4600
    },
    {
      "epoch": 0.24127283194640708,
      "grad_norm": 0.42357298731803894,
      "learning_rate": 2e-05,
      "loss": 0.63,
      "step": 4610
    },
    {
      "epoch": 0.24179620034542315,
      "grad_norm": 0.7327708601951599,
      "learning_rate": 2e-05,
      "loss": 0.5942,
      "step": 4620
    },
    {
      "epoch": 0.24231956874443922,
      "grad_norm": 0.4425099790096283,
      "learning_rate": 2e-05,
      "loss": 0.6078,
      "step": 4630
    },
    {
      "epoch": 0.24284293714345528,
      "grad_norm": 0.3585580289363861,
      "learning_rate": 2e-05,
      "loss": 0.5886,
      "step": 4640
    },
    {
      "epoch": 0.24336630554247135,
      "grad_norm": 1.4051371812820435,
      "learning_rate": 2e-05,
      "loss": 0.6031,
      "step": 4650
    },
    {
      "epoch": 0.24388967394148742,
      "grad_norm": 0.4339183270931244,
      "learning_rate": 2e-05,
      "loss": 0.6104,
      "step": 4660
    },
    {
      "epoch": 0.24441304234050348,
      "grad_norm": 0.40716227889060974,
      "learning_rate": 2e-05,
      "loss": 0.6532,
      "step": 4670
    },
    {
      "epoch": 0.24493641073951955,
      "grad_norm": 0.3990173041820526,
      "learning_rate": 2e-05,
      "loss": 0.6159,
      "step": 4680
    },
    {
      "epoch": 0.24545977913853562,
      "grad_norm": 0.5576123595237732,
      "learning_rate": 2e-05,
      "loss": 0.6279,
      "step": 4690
    },
    {
      "epoch": 0.24598314753755168,
      "grad_norm": 0.3955235183238983,
      "learning_rate": 2e-05,
      "loss": 0.6232,
      "step": 4700
    },
    {
      "epoch": 0.24650651593656775,
      "grad_norm": 0.4591265320777893,
      "learning_rate": 2e-05,
      "loss": 0.6311,
      "step": 4710
    },
    {
      "epoch": 0.24702988433558382,
      "grad_norm": 0.3967943787574768,
      "learning_rate": 2e-05,
      "loss": 0.6285,
      "step": 4720
    },
    {
      "epoch": 0.24755325273459988,
      "grad_norm": 0.5398264527320862,
      "learning_rate": 2e-05,
      "loss": 0.6331,
      "step": 4730
    },
    {
      "epoch": 0.24807662113361595,
      "grad_norm": 0.41231897473335266,
      "learning_rate": 2e-05,
      "loss": 0.6024,
      "step": 4740
    },
    {
      "epoch": 0.24859998953263202,
      "grad_norm": 0.40629324316978455,
      "learning_rate": 2e-05,
      "loss": 0.6271,
      "step": 4750
    },
    {
      "epoch": 0.24912335793164808,
      "grad_norm": 0.42986491322517395,
      "learning_rate": 2e-05,
      "loss": 0.5869,
      "step": 4760
    },
    {
      "epoch": 0.24964672633066415,
      "grad_norm": 0.45604678988456726,
      "learning_rate": 2e-05,
      "loss": 0.632,
      "step": 4770
    },
    {
      "epoch": 0.2501700947296802,
      "grad_norm": 0.4645533859729767,
      "learning_rate": 2e-05,
      "loss": 0.6167,
      "step": 4780
    },
    {
      "epoch": 0.2506934631286963,
      "grad_norm": 0.3783343732357025,
      "learning_rate": 2e-05,
      "loss": 0.6157,
      "step": 4790
    },
    {
      "epoch": 0.25121683152771235,
      "grad_norm": 0.38526099920272827,
      "learning_rate": 2e-05,
      "loss": 0.5896,
      "step": 4800
    },
    {
      "epoch": 0.2517401999267284,
      "grad_norm": 0.3877682685852051,
      "learning_rate": 2e-05,
      "loss": 0.6102,
      "step": 4810
    },
    {
      "epoch": 0.2522635683257445,
      "grad_norm": 0.39466989040374756,
      "learning_rate": 2e-05,
      "loss": 0.5981,
      "step": 4820
    },
    {
      "epoch": 0.25278693672476055,
      "grad_norm": 0.4054771065711975,
      "learning_rate": 2e-05,
      "loss": 0.5967,
      "step": 4830
    },
    {
      "epoch": 0.2533103051237766,
      "grad_norm": 0.37173131108283997,
      "learning_rate": 2e-05,
      "loss": 0.5909,
      "step": 4840
    },
    {
      "epoch": 0.2538336735227927,
      "grad_norm": 0.41072943806648254,
      "learning_rate": 2e-05,
      "loss": 0.6111,
      "step": 4850
    },
    {
      "epoch": 0.25435704192180875,
      "grad_norm": 0.41547060012817383,
      "learning_rate": 2e-05,
      "loss": 0.6222,
      "step": 4860
    },
    {
      "epoch": 0.2548804103208248,
      "grad_norm": 0.409974604845047,
      "learning_rate": 2e-05,
      "loss": 0.6307,
      "step": 4870
    },
    {
      "epoch": 0.2554037787198409,
      "grad_norm": 0.38040807843208313,
      "learning_rate": 2e-05,
      "loss": 0.6125,
      "step": 4880
    },
    {
      "epoch": 0.25592714711885695,
      "grad_norm": 0.4161960482597351,
      "learning_rate": 2e-05,
      "loss": 0.6311,
      "step": 4890
    },
    {
      "epoch": 0.256450515517873,
      "grad_norm": 0.41063639521598816,
      "learning_rate": 2e-05,
      "loss": 0.6335,
      "step": 4900
    },
    {
      "epoch": 0.2569738839168891,
      "grad_norm": 0.4094699025154114,
      "learning_rate": 2e-05,
      "loss": 0.6166,
      "step": 4910
    },
    {
      "epoch": 0.25749725231590515,
      "grad_norm": 0.4404178559780121,
      "learning_rate": 2e-05,
      "loss": 0.6109,
      "step": 4920
    },
    {
      "epoch": 0.2580206207149212,
      "grad_norm": 0.42314210534095764,
      "learning_rate": 2e-05,
      "loss": 0.6089,
      "step": 4930
    },
    {
      "epoch": 0.2585439891139373,
      "grad_norm": 0.4365096688270569,
      "learning_rate": 2e-05,
      "loss": 0.6193,
      "step": 4940
    },
    {
      "epoch": 0.25906735751295334,
      "grad_norm": 0.4479396939277649,
      "learning_rate": 2e-05,
      "loss": 0.6254,
      "step": 4950
    },
    {
      "epoch": 0.2595907259119694,
      "grad_norm": 0.3912433385848999,
      "learning_rate": 2e-05,
      "loss": 0.6169,
      "step": 4960
    },
    {
      "epoch": 0.2601140943109855,
      "grad_norm": 0.3662976622581482,
      "learning_rate": 2e-05,
      "loss": 0.6037,
      "step": 4970
    },
    {
      "epoch": 0.26063746271000154,
      "grad_norm": 0.4002629518508911,
      "learning_rate": 2e-05,
      "loss": 0.5935,
      "step": 4980
    },
    {
      "epoch": 0.2611608311090176,
      "grad_norm": 0.39754584431648254,
      "learning_rate": 2e-05,
      "loss": 0.6045,
      "step": 4990
    },
    {
      "epoch": 0.26168419950803373,
      "grad_norm": 0.44900912046432495,
      "learning_rate": 2e-05,
      "loss": 0.6234,
      "step": 5000
    },
    {
      "epoch": 0.2622075679070498,
      "grad_norm": 0.46902185678482056,
      "learning_rate": 2e-05,
      "loss": 0.5961,
      "step": 5010
    },
    {
      "epoch": 0.26273093630606587,
      "grad_norm": 0.41309309005737305,
      "learning_rate": 2e-05,
      "loss": 0.6275,
      "step": 5020
    },
    {
      "epoch": 0.26325430470508193,
      "grad_norm": 0.396864652633667,
      "learning_rate": 2e-05,
      "loss": 0.638,
      "step": 5030
    },
    {
      "epoch": 0.263777673104098,
      "grad_norm": 0.39725908637046814,
      "learning_rate": 2e-05,
      "loss": 0.638,
      "step": 5040
    },
    {
      "epoch": 0.26430104150311406,
      "grad_norm": 0.40473729372024536,
      "learning_rate": 2e-05,
      "loss": 0.6166,
      "step": 5050
    },
    {
      "epoch": 0.26482440990213013,
      "grad_norm": 0.45297154784202576,
      "learning_rate": 2e-05,
      "loss": 0.6167,
      "step": 5060
    },
    {
      "epoch": 0.2653477783011462,
      "grad_norm": 0.3637174367904663,
      "learning_rate": 2e-05,
      "loss": 0.59,
      "step": 5070
    },
    {
      "epoch": 0.26587114670016226,
      "grad_norm": 0.3888295590877533,
      "learning_rate": 2e-05,
      "loss": 0.5946,
      "step": 5080
    },
    {
      "epoch": 0.26639451509917833,
      "grad_norm": 0.3925114870071411,
      "learning_rate": 2e-05,
      "loss": 0.6454,
      "step": 5090
    },
    {
      "epoch": 0.2669178834981944,
      "grad_norm": 0.42430034279823303,
      "learning_rate": 2e-05,
      "loss": 0.598,
      "step": 5100
    },
    {
      "epoch": 0.26744125189721046,
      "grad_norm": 0.4356481432914734,
      "learning_rate": 2e-05,
      "loss": 0.5841,
      "step": 5110
    },
    {
      "epoch": 0.26796462029622653,
      "grad_norm": 0.40845903754234314,
      "learning_rate": 2e-05,
      "loss": 0.6214,
      "step": 5120
    },
    {
      "epoch": 0.2684879886952426,
      "grad_norm": 0.37573105096817017,
      "learning_rate": 2e-05,
      "loss": 0.6283,
      "step": 5130
    },
    {
      "epoch": 0.26901135709425866,
      "grad_norm": 0.40981823205947876,
      "learning_rate": 2e-05,
      "loss": 0.5933,
      "step": 5140
    },
    {
      "epoch": 0.26953472549327473,
      "grad_norm": 0.39033955335617065,
      "learning_rate": 2e-05,
      "loss": 0.5921,
      "step": 5150
    },
    {
      "epoch": 0.2700580938922908,
      "grad_norm": 0.4205704927444458,
      "learning_rate": 2e-05,
      "loss": 0.6128,
      "step": 5160
    },
    {
      "epoch": 0.27058146229130686,
      "grad_norm": 0.37669286131858826,
      "learning_rate": 2e-05,
      "loss": 0.6054,
      "step": 5170
    },
    {
      "epoch": 0.27110483069032293,
      "grad_norm": 0.40791818499565125,
      "learning_rate": 2e-05,
      "loss": 0.6309,
      "step": 5180
    },
    {
      "epoch": 0.271628199089339,
      "grad_norm": 0.42482030391693115,
      "learning_rate": 2e-05,
      "loss": 0.6297,
      "step": 5190
    },
    {
      "epoch": 0.27215156748835506,
      "grad_norm": 0.4918680787086487,
      "learning_rate": 2e-05,
      "loss": 0.5974,
      "step": 5200
    },
    {
      "epoch": 0.27267493588737113,
      "grad_norm": 0.4212997257709503,
      "learning_rate": 2e-05,
      "loss": 0.6251,
      "step": 5210
    },
    {
      "epoch": 0.2731983042863872,
      "grad_norm": 0.40279462933540344,
      "learning_rate": 2e-05,
      "loss": 0.5907,
      "step": 5220
    },
    {
      "epoch": 0.27372167268540326,
      "grad_norm": 0.4512042999267578,
      "learning_rate": 2e-05,
      "loss": 0.6082,
      "step": 5230
    },
    {
      "epoch": 0.2742450410844193,
      "grad_norm": 0.4068576693534851,
      "learning_rate": 2e-05,
      "loss": 0.6062,
      "step": 5240
    },
    {
      "epoch": 0.2747684094834354,
      "grad_norm": 0.4271692931652069,
      "learning_rate": 2e-05,
      "loss": 0.6326,
      "step": 5250
    },
    {
      "epoch": 0.27529177788245146,
      "grad_norm": 0.41401100158691406,
      "learning_rate": 2e-05,
      "loss": 0.6105,
      "step": 5260
    },
    {
      "epoch": 0.2758151462814675,
      "grad_norm": 0.4413628578186035,
      "learning_rate": 2e-05,
      "loss": 0.5898,
      "step": 5270
    },
    {
      "epoch": 0.2763385146804836,
      "grad_norm": 0.41365042328834534,
      "learning_rate": 2e-05,
      "loss": 0.5958,
      "step": 5280
    },
    {
      "epoch": 0.27686188307949966,
      "grad_norm": 0.45695561170578003,
      "learning_rate": 2e-05,
      "loss": 0.6283,
      "step": 5290
    },
    {
      "epoch": 0.2773852514785157,
      "grad_norm": 0.4205687940120697,
      "learning_rate": 2e-05,
      "loss": 0.6257,
      "step": 5300
    },
    {
      "epoch": 0.2779086198775318,
      "grad_norm": 0.3920809030532837,
      "learning_rate": 2e-05,
      "loss": 0.6518,
      "step": 5310
    },
    {
      "epoch": 0.27843198827654786,
      "grad_norm": 0.34857314825057983,
      "learning_rate": 2e-05,
      "loss": 0.5915,
      "step": 5320
    },
    {
      "epoch": 0.2789553566755639,
      "grad_norm": 0.47157859802246094,
      "learning_rate": 2e-05,
      "loss": 0.6315,
      "step": 5330
    },
    {
      "epoch": 0.27947872507458,
      "grad_norm": 0.4518488943576813,
      "learning_rate": 2e-05,
      "loss": 0.6172,
      "step": 5340
    },
    {
      "epoch": 0.28000209347359606,
      "grad_norm": 0.44076552987098694,
      "learning_rate": 1.999194520388485e-05,
      "loss": 0.6251,
      "step": 5350
    },
    {
      "epoch": 0.2805254618726121,
      "grad_norm": 0.38890475034713745,
      "learning_rate": 1.9959247303157763e-05,
      "loss": 0.6194,
      "step": 5360
    },
    {
      "epoch": 0.2810488302716282,
      "grad_norm": 0.3902517855167389,
      "learning_rate": 1.990149424459817e-05,
      "loss": 0.6009,
      "step": 5370
    },
    {
      "epoch": 0.28157219867064426,
      "grad_norm": 0.39967110753059387,
      "learning_rate": 1.9818847538648688e-05,
      "loss": 0.6081,
      "step": 5380
    },
    {
      "epoch": 0.2820955670696603,
      "grad_norm": 0.3647516965866089,
      "learning_rate": 1.971153831256721e-05,
      "loss": 0.5785,
      "step": 5390
    },
    {
      "epoch": 0.2826189354686764,
      "grad_norm": 0.4686444401741028,
      "learning_rate": 1.9579866664063435e-05,
      "loss": 0.635,
      "step": 5400
    },
    {
      "epoch": 0.28314230386769246,
      "grad_norm": 0.4206537902355194,
      "learning_rate": 1.9424200822054782e-05,
      "loss": 0.628,
      "step": 5410
    },
    {
      "epoch": 0.2836656722667085,
      "grad_norm": 0.43939080834388733,
      "learning_rate": 1.9244976116888626e-05,
      "loss": 0.6335,
      "step": 5420
    },
    {
      "epoch": 0.2841890406657246,
      "grad_norm": 0.3906099796295166,
      "learning_rate": 1.904269376291071e-05,
      "loss": 0.5958,
      "step": 5430
    },
    {
      "epoch": 0.28471240906474066,
      "grad_norm": 0.3985750377178192,
      "learning_rate": 1.88179194567844e-05,
      "loss": 0.6213,
      "step": 5440
    },
    {
      "epoch": 0.2852357774637567,
      "grad_norm": 0.4336455762386322,
      "learning_rate": 1.8571281795480632e-05,
      "loss": 0.604,
      "step": 5450
    },
    {
      "epoch": 0.2857591458627728,
      "grad_norm": 0.4126925468444824,
      "learning_rate": 1.8303470518362808e-05,
      "loss": 0.5991,
      "step": 5460
    },
    {
      "epoch": 0.28628251426178886,
      "grad_norm": 0.4074404835700989,
      "learning_rate": 1.801523457828271e-05,
      "loss": 0.6045,
      "step": 5470
    },
    {
      "epoch": 0.2868058826608049,
      "grad_norm": 0.40034711360931396,
      "learning_rate": 1.7707380047081796e-05,
      "loss": 0.6052,
      "step": 5480
    },
    {
      "epoch": 0.287329251059821,
      "grad_norm": 0.41930365562438965,
      "learning_rate": 1.7380767861355252e-05,
      "loss": 0.6316,
      "step": 5490
    },
    {
      "epoch": 0.28785261945883706,
      "grad_norm": 0.42970430850982666,
      "learning_rate": 1.703631141478299e-05,
      "loss": 0.643,
      "step": 5500
    },
    {
      "epoch": 0.2883759878578531,
      "grad_norm": 0.6285122632980347,
      "learning_rate": 1.6674974003760705e-05,
      "loss": 0.5952,
      "step": 5510
    },
    {
      "epoch": 0.2888993562568692,
      "grad_norm": 0.4414973556995392,
      "learning_rate": 1.6297766133474556e-05,
      "loss": 0.5934,
      "step": 5520
    },
    {
      "epoch": 0.28942272465588526,
      "grad_norm": 0.3714768886566162,
      "learning_rate": 1.5905742691953186e-05,
      "loss": 0.5846,
      "step": 5530
    },
    {
      "epoch": 0.2899460930549013,
      "grad_norm": 0.3532920181751251,
      "learning_rate": 1.55e-05,
      "loss": 0.5995,
      "step": 5540
    },
    {
      "epoch": 0.2904694614539174,
      "grad_norm": 0.4924233853816986,
      "learning_rate": 1.5081672745255862e-05,
      "loss": 0.5818,
      "step": 5550
    },
    {
      "epoch": 0.29099282985293345,
      "grad_norm": 0.4407850503921509,
      "learning_rate": 1.4651930808966206e-05,
      "loss": 0.6016,
      "step": 5560
    },
    {
      "epoch": 0.2915161982519495,
      "grad_norm": 0.42552539706230164,
      "learning_rate": 1.4211975994326848e-05,
      "loss": 0.603,
      "step": 5570
    },
    {
      "epoch": 0.2920395666509656,
      "grad_norm": 0.36129358410835266,
      "learning_rate": 1.376303866555779e-05,
      "loss": 0.6101,
      "step": 5580
    },
    {
      "epoch": 0.2925629350499817,
      "grad_norm": 0.406896710395813,
      "learning_rate": 1.3306374307104164e-05,
      "loss": 0.6115,
      "step": 5590
    },
    {
      "epoch": 0.2930863034489978,
      "grad_norm": 0.43325328826904297,
      "learning_rate": 1.2843260012586718e-05,
      "loss": 0.6105,
      "step": 5600
    },
    {
      "epoch": 0.29360967184801384,
      "grad_norm": 0.40178170800209045,
      "learning_rate": 1.2374990913320726e-05,
      "loss": 0.6258,
      "step": 5610
    },
    {
      "epoch": 0.2941330402470299,
      "grad_norm": 0.37656369805336,
      "learning_rate": 1.1902876556391247e-05,
      "loss": 0.6063,
      "step": 5620
    },
    {
      "epoch": 0.294656408646046,
      "grad_norm": 0.4082496166229248,
      "learning_rate": 1.1428237242413685e-05,
      "loss": 0.579,
      "step": 5630
    },
    {
      "epoch": 0.29517977704506204,
      "grad_norm": 0.40694573521614075,
      "learning_rate": 1.095240033322132e-05,
      "loss": 0.5992,
      "step": 5640
    },
    {
      "epoch": 0.2957031454440781,
      "grad_norm": 0.4091090261936188,
      "learning_rate": 1.047669653980572e-05,
      "loss": 0.5843,
      "step": 5650
    },
    {
      "epoch": 0.2962265138430942,
      "grad_norm": 0.44866329431533813,
      "learning_rate": 1.0002456200890902e-05,
      "loss": 0.6118,
      "step": 5660
    },
    {
      "epoch": 0.29674988224211024,
      "grad_norm": 0.4145185649394989,
      "learning_rate": 9.531005562548668e-06,
      "loss": 0.5998,
      "step": 5670
    },
    {
      "epoch": 0.2972732506411263,
      "grad_norm": 0.4247947037220001,
      "learning_rate": 9.06366306925925e-06,
      "loss": 0.5885,
      "step": 5680
    },
    {
      "epoch": 0.2977966190401424,
      "grad_norm": 0.3585066795349121,
      "learning_rate": 8.601735676789686e-06,
      "loss": 0.6015,
      "step": 5690
    },
    {
      "epoch": 0.29831998743915844,
      "grad_norm": 0.3997299373149872,
      "learning_rate": 8.14651519720118e-06,
      "loss": 0.6087,
      "step": 5700
    },
    {
      "epoch": 0.2988433558381745,
      "grad_norm": 0.4697336256504059,
      "learning_rate": 7.699274686206866e-06,
      "loss": 0.6262,
      "step": 5710
    },
    {
      "epoch": 0.2993667242371906,
      "grad_norm": 0.3803713023662567,
      "learning_rate": 7.261264882983024e-06,
      "loss": 0.6104,
      "step": 5720
    },
    {
      "epoch": 0.29989009263620664,
      "grad_norm": 0.382207989692688,
      "learning_rate": 6.8337107123900386e-06,
      "loss": 0.5738,
      "step": 5730
    },
    {
      "epoch": 0.3004134610352227,
      "grad_norm": 0.39442846179008484,
      "learning_rate": 6.417807859384881e-06,
      "loss": 0.6163,
      "step": 5740
    },
    {
      "epoch": 0.3009368294342388,
      "grad_norm": 0.3684907555580139,
      "learning_rate": 6.014719425205009e-06,
      "loss": 0.6046,
      "step": 5750
    },
    {
      "epoch": 0.30146019783325484,
      "grad_norm": 0.37866348028182983,
      "learning_rate": 5.6255726746749275e-06,
      "loss": 0.5828,
      "step": 5760
    },
    {
      "epoch": 0.3019835662322709,
      "grad_norm": 0.38500234484672546,
      "learning_rate": 5.251455883731728e-06,
      "loss": 0.6056,
      "step": 5770
    },
    {
      "epoch": 0.302506934631287,
      "grad_norm": 0.4282638430595398,
      "learning_rate": 4.893415295985812e-06,
      "loss": 0.6071,
      "step": 5780
    },
    {
      "epoch": 0.30303030303030304,
      "grad_norm": 0.4080576002597809,
      "learning_rate": 4.552452196827885e-06,
      "loss": 0.5835,
      "step": 5790
    },
    {
      "epoch": 0.3035536714293191,
      "grad_norm": 0.3698746860027313,
      "learning_rate": 4.229520113264785e-06,
      "loss": 0.6342,
      "step": 5800
    },
    {
      "epoch": 0.30407703982833517,
      "grad_norm": 0.388528972864151,
      "learning_rate": 3.925522147314915e-06,
      "loss": 0.6107,
      "step": 5810
    },
    {
      "epoch": 0.30460040822735124,
      "grad_norm": 0.4334981441497803,
      "learning_rate": 3.641308450420675e-06,
      "loss": 0.5793,
      "step": 5820
    },
    {
      "epoch": 0.3051237766263673,
      "grad_norm": 0.4103776514530182,
      "learning_rate": 3.37767384594087e-06,
      "loss": 0.6098,
      "step": 5830
    },
    {
      "epoch": 0.30564714502538337,
      "grad_norm": 0.386342316865921,
      "learning_rate": 3.1353556063719357e-06,
      "loss": 0.5895,
      "step": 5840
    },
    {
      "epoch": 0.30617051342439944,
      "grad_norm": 0.39483997225761414,
      "learning_rate": 2.9150313915141474e-06,
      "loss": 0.5973,
      "step": 5850
    },
    {
      "epoch": 0.3066938818234155,
      "grad_norm": 0.3758963942527771,
      "learning_rate": 2.717317353348891e-06,
      "loss": 0.5851,
      "step": 5860
    },
    {
      "epoch": 0.30721725022243157,
      "grad_norm": 0.37651148438453674,
      "learning_rate": 2.5427664129268253e-06,
      "loss": 0.5944,
      "step": 5870
    },
    {
      "epoch": 0.30774061862144764,
      "grad_norm": 0.3909638226032257,
      "learning_rate": 2.391866714085717e-06,
      "loss": 0.5946,
      "step": 5880
    },
    {
      "epoch": 0.3082639870204637,
      "grad_norm": 0.3788841664791107,
      "learning_rate": 2.2650402583222753e-06,
      "loss": 0.6036,
      "step": 5890
    },
    {
      "epoch": 0.30878735541947977,
      "grad_norm": 0.3904835283756256,
      "learning_rate": 2.1626417246356398e-06,
      "loss": 0.5993,
      "step": 5900
    },
    {
      "epoch": 0.30931072381849584,
      "grad_norm": 0.4416792690753937,
      "learning_rate": 2.084957477642894e-06,
      "loss": 0.5919,
      "step": 5910
    },
    {
      "epoch": 0.3098340922175119,
      "grad_norm": 0.4406580626964569,
      "learning_rate": 2.0322047667405147e-06,
      "loss": 0.5997,
      "step": 5920
    },
    {
      "epoch": 0.31035746061652797,
      "grad_norm": 0.39945074915885925,
      "learning_rate": 2.0045311185513342e-06,
      "loss": 0.6132,
      "step": 5930
    }
  ],
  "logging_steps": 10,
  "max_steps": 5935,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 2035,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 2.6494000762504348e+19,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}