{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 1623,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0030807147258163892,
      "grad_norm": 0.5649558305740356,
      "learning_rate": 9.756097560975611e-06,
      "loss": 1.1107,
      "step": 5
    },
    {
      "epoch": 0.0061614294516327784,
      "grad_norm": 0.5505046844482422,
      "learning_rate": 2.1951219512195124e-05,
      "loss": 1.1729,
      "step": 10
    },
    {
      "epoch": 0.009242144177449169,
      "grad_norm": 0.32379722595214844,
      "learning_rate": 3.414634146341464e-05,
      "loss": 1.2201,
      "step": 15
    },
    {
      "epoch": 0.012322858903265557,
      "grad_norm": 0.519706130027771,
      "learning_rate": 4.634146341463415e-05,
      "loss": 1.067,
      "step": 20
    },
    {
      "epoch": 0.015403573629081947,
      "grad_norm": 0.862956702709198,
      "learning_rate": 5.853658536585366e-05,
      "loss": 0.9211,
      "step": 25
    },
    {
      "epoch": 0.018484288354898338,
      "grad_norm": 0.370582640171051,
      "learning_rate": 7.073170731707317e-05,
      "loss": 1.0194,
      "step": 30
    },
    {
      "epoch": 0.021565003080714726,
      "grad_norm": 0.28057190775871277,
      "learning_rate": 8.292682926829268e-05,
      "loss": 1.0605,
      "step": 35
    },
    {
      "epoch": 0.024645717806531114,
      "grad_norm": 0.25552278757095337,
      "learning_rate": 9.51219512195122e-05,
      "loss": 1.0577,
      "step": 40
    },
    {
      "epoch": 0.027726432532347505,
      "grad_norm": 0.653907299041748,
      "learning_rate": 0.00010731707317073172,
      "loss": 0.947,
      "step": 45
    },
    {
      "epoch": 0.030807147258163893,
      "grad_norm": 0.5950673222541809,
      "learning_rate": 0.00011951219512195122,
      "loss": 0.8823,
      "step": 50
    },
    {
      "epoch": 0.033887861983980284,
      "grad_norm": 0.30638569593429565,
      "learning_rate": 0.00013170731707317076,
      "loss": 0.9989,
      "step": 55
    },
    {
      "epoch": 0.036968576709796676,
      "grad_norm": 0.2737618386745453,
      "learning_rate": 0.00014390243902439025,
      "loss": 1.0177,
      "step": 60
    },
    {
      "epoch": 0.04004929143561306,
      "grad_norm": 0.214647114276886,
      "learning_rate": 0.00015609756097560978,
      "loss": 0.9521,
      "step": 65
    },
    {
      "epoch": 0.04313000616142945,
      "grad_norm": 0.46437978744506836,
      "learning_rate": 0.00016829268292682927,
      "loss": 0.9445,
      "step": 70
    },
    {
      "epoch": 0.04621072088724584,
      "grad_norm": 0.6505705118179321,
      "learning_rate": 0.0001804878048780488,
      "loss": 0.8239,
      "step": 75
    },
    {
      "epoch": 0.04929143561306223,
      "grad_norm": 0.30413898825645447,
      "learning_rate": 0.0001926829268292683,
      "loss": 0.9014,
      "step": 80
    },
    {
      "epoch": 0.05237215033887862,
      "grad_norm": 0.2977719306945801,
      "learning_rate": 0.0001999991687649223,
      "loss": 0.9701,
      "step": 85
    },
    {
      "epoch": 0.05545286506469501,
      "grad_norm": 0.20709441602230072,
      "learning_rate": 0.00019998981752900036,
      "loss": 0.9756,
      "step": 90
    },
    {
      "epoch": 0.0585335797905114,
      "grad_norm": 0.4442475140094757,
      "learning_rate": 0.00019997007698817557,
      "loss": 0.9496,
      "step": 95
    },
    {
      "epoch": 0.061614294516327786,
      "grad_norm": 0.545454204082489,
      "learning_rate": 0.00019993994919356167,
      "loss": 0.7949,
      "step": 100
    },
    {
      "epoch": 0.06469500924214418,
      "grad_norm": 0.26897159218788147,
      "learning_rate": 0.00019989943727554598,
      "loss": 0.867,
      "step": 105
    },
    {
      "epoch": 0.06777572396796057,
      "grad_norm": 0.26484453678131104,
      "learning_rate": 0.00019984854544346367,
      "loss": 0.9259,
      "step": 110
    },
    {
      "epoch": 0.07085643869377696,
      "grad_norm": 0.2327209860086441,
      "learning_rate": 0.00019978727898516086,
      "loss": 0.9823,
      "step": 115
    },
    {
      "epoch": 0.07393715341959335,
      "grad_norm": 0.4446864426136017,
      "learning_rate": 0.0001997156442664449,
      "loss": 0.7825,
      "step": 120
    },
    {
      "epoch": 0.07701786814540973,
      "grad_norm": 0.5609371066093445,
      "learning_rate": 0.00019963364873042298,
      "loss": 0.7495,
      "step": 125
    },
    {
      "epoch": 0.08009858287122612,
      "grad_norm": 0.22691862285137177,
      "learning_rate": 0.0001995413008967289,
      "loss": 0.9486,
      "step": 130
    },
    {
      "epoch": 0.08317929759704251,
      "grad_norm": 0.24267292022705078,
      "learning_rate": 0.00019943861036063768,
      "loss": 0.9331,
      "step": 135
    },
    {
      "epoch": 0.0862600123228589,
      "grad_norm": 0.21934683620929718,
      "learning_rate": 0.00019932558779206874,
      "loss": 0.9401,
      "step": 140
    },
    {
      "epoch": 0.0893407270486753,
      "grad_norm": 0.42394694685935974,
      "learning_rate": 0.00019920224493447702,
      "loss": 0.8971,
      "step": 145
    },
    {
      "epoch": 0.09242144177449169,
      "grad_norm": 0.5011906027793884,
      "learning_rate": 0.00019906859460363307,
      "loss": 0.7387,
      "step": 150
    },
    {
      "epoch": 0.09550215650030808,
      "grad_norm": 0.25576251745224,
      "learning_rate": 0.00019892465068629131,
      "loss": 0.8509,
      "step": 155
    },
    {
      "epoch": 0.09858287122612445,
      "grad_norm": 0.23253919184207916,
      "learning_rate": 0.0001987704281387471,
      "loss": 0.9972,
      "step": 160
    },
    {
      "epoch": 0.10166358595194085,
      "grad_norm": 0.21661245822906494,
      "learning_rate": 0.00019860594298528282,
      "loss": 0.9319,
      "step": 165
    },
    {
      "epoch": 0.10474430067775724,
      "grad_norm": 0.45695972442626953,
      "learning_rate": 0.0001984312123165028,
      "loss": 0.9361,
      "step": 170
    },
    {
      "epoch": 0.10782501540357363,
      "grad_norm": 0.41159898042678833,
      "learning_rate": 0.0001982462542875576,
      "loss": 0.7077,
      "step": 175
    },
    {
      "epoch": 0.11090573012939002,
      "grad_norm": 0.2505335509777069,
      "learning_rate": 0.00019805108811625773,
      "loss": 0.8675,
      "step": 180
    },
    {
      "epoch": 0.11398644485520641,
      "grad_norm": 0.23700909316539764,
      "learning_rate": 0.00019784573408107657,
      "loss": 0.9005,
      "step": 185
    },
    {
      "epoch": 0.1170671595810228,
      "grad_norm": 0.21003198623657227,
      "learning_rate": 0.00019763021351904358,
      "loss": 0.8991,
      "step": 190
    },
    {
      "epoch": 0.12014787430683918,
      "grad_norm": 0.40320080518722534,
      "learning_rate": 0.00019740454882352732,
      "loss": 0.8521,
      "step": 195
    },
    {
      "epoch": 0.12322858903265557,
      "grad_norm": 0.46959367394447327,
      "learning_rate": 0.0001971687634419086,
      "loss": 0.7645,
      "step": 200
    },
    {
      "epoch": 0.12630930375847196,
      "grad_norm": 0.27376580238342285,
      "learning_rate": 0.0001969228818731442,
      "loss": 0.8629,
      "step": 205
    },
    {
      "epoch": 0.12939001848428835,
      "grad_norm": 0.22045589983463287,
      "learning_rate": 0.00019666692966522145,
      "loss": 0.9351,
      "step": 210
    },
    {
      "epoch": 0.13247073321010475,
      "grad_norm": 0.23090502619743347,
      "learning_rate": 0.00019640093341250357,
      "loss": 0.8712,
      "step": 215
    },
    {
      "epoch": 0.13555144793592114,
      "grad_norm": 0.4026584327220917,
      "learning_rate": 0.0001961249207529665,
      "loss": 0.8206,
      "step": 220
    },
    {
      "epoch": 0.13863216266173753,
      "grad_norm": 0.5381972789764404,
      "learning_rate": 0.00019583892036532726,
      "loss": 0.7596,
      "step": 225
    },
    {
      "epoch": 0.14171287738755392,
      "grad_norm": 0.24759496748447418,
      "learning_rate": 0.00019554296196606395,
      "loss": 0.885,
      "step": 230
    },
    {
      "epoch": 0.1447935921133703,
      "grad_norm": 0.2439427524805069,
      "learning_rate": 0.00019523707630632835,
      "loss": 0.8783,
      "step": 235
    },
    {
      "epoch": 0.1478743068391867,
      "grad_norm": 0.22079786658287048,
      "learning_rate": 0.00019492129516875055,
      "loss": 0.9023,
      "step": 240
    },
    {
      "epoch": 0.15095502156500307,
      "grad_norm": 0.38100528717041016,
      "learning_rate": 0.00019459565136413666,
      "loss": 0.9026,
      "step": 245
    },
    {
      "epoch": 0.15403573629081946,
      "grad_norm": 0.4669598937034607,
      "learning_rate": 0.0001942601787280598,
      "loss": 0.7342,
      "step": 250
    },
    {
      "epoch": 0.15711645101663585,
      "grad_norm": 0.27741730213165283,
      "learning_rate": 0.00019391491211734425,
      "loss": 0.9125,
      "step": 255
    },
    {
      "epoch": 0.16019716574245224,
      "grad_norm": 0.2022334635257721,
      "learning_rate": 0.0001935598874064438,
      "loss": 0.8812,
      "step": 260
    },
    {
      "epoch": 0.16327788046826863,
      "grad_norm": 0.19794999063014984,
      "learning_rate": 0.00019319514148371435,
      "loss": 0.8242,
      "step": 265
    },
    {
      "epoch": 0.16635859519408502,
      "grad_norm": 0.40853703022003174,
      "learning_rate": 0.00019282071224758091,
      "loss": 0.8201,
      "step": 270
    },
    {
      "epoch": 0.16943930991990142,
      "grad_norm": 0.488699734210968,
      "learning_rate": 0.00019243663860259993,
      "loss": 0.7571,
      "step": 275
    },
    {
      "epoch": 0.1725200246457178,
      "grad_norm": 0.23293153941631317,
      "learning_rate": 0.00019204296045541685,
      "loss": 0.879,
      "step": 280
    },
    {
      "epoch": 0.1756007393715342,
      "grad_norm": 0.24481086432933807,
      "learning_rate": 0.0001916397187106199,
      "loss": 0.8304,
      "step": 285
    },
    {
      "epoch": 0.1786814540973506,
      "grad_norm": 0.1973239630460739,
      "learning_rate": 0.00019122695526648968,
      "loss": 0.869,
      "step": 290
    },
    {
      "epoch": 0.18176216882316698,
      "grad_norm": 0.3842983543872833,
      "learning_rate": 0.00019080471301064598,
      "loss": 0.8815,
      "step": 295
    },
    {
      "epoch": 0.18484288354898337,
      "grad_norm": 0.484453946352005,
      "learning_rate": 0.00019037303581559143,
      "loss": 0.7238,
      "step": 300
    },
    {
      "epoch": 0.18792359827479976,
      "grad_norm": 0.2231549322605133,
      "learning_rate": 0.00018993196853415317,
      "loss": 0.8462,
      "step": 305
    },
    {
      "epoch": 0.19100431300061615,
      "grad_norm": 0.262131005525589,
      "learning_rate": 0.00018948155699482244,
      "loss": 0.8284,
      "step": 310
    },
    {
      "epoch": 0.19408502772643252,
      "grad_norm": 0.23521290719509125,
      "learning_rate": 0.00018902184799699263,
      "loss": 0.9654,
      "step": 315
    },
    {
      "epoch": 0.1971657424522489,
      "grad_norm": 0.3194115459918976,
      "learning_rate": 0.00018855288930609692,
      "loss": 0.7911,
      "step": 320
    },
    {
      "epoch": 0.2002464571780653,
      "grad_norm": 0.38318154215812683,
      "learning_rate": 0.00018807472964864515,
      "loss": 0.7259,
      "step": 325
    },
    {
      "epoch": 0.2033271719038817,
      "grad_norm": 0.24597209692001343,
      "learning_rate": 0.00018758741870716092,
      "loss": 0.9146,
      "step": 330
    },
    {
      "epoch": 0.20640788662969808,
      "grad_norm": 0.21656718850135803,
      "learning_rate": 0.00018709100711501955,
      "loss": 0.9038,
      "step": 335
    },
    {
      "epoch": 0.20948860135551448,
      "grad_norm": 0.2119424194097519,
      "learning_rate": 0.0001865855464511869,
      "loss": 0.8179,
      "step": 340
    },
    {
      "epoch": 0.21256931608133087,
      "grad_norm": 0.3952539563179016,
      "learning_rate": 0.00018607108923486025,
      "loss": 0.7233,
      "step": 345
    },
    {
      "epoch": 0.21565003080714726,
      "grad_norm": 0.4704679846763611,
      "learning_rate": 0.00018554768892001136,
      "loss": 0.7141,
      "step": 350
    },
    {
      "epoch": 0.21873074553296365,
      "grad_norm": 0.24604864418506622,
      "learning_rate": 0.00018501539988983234,
      "loss": 0.7551,
      "step": 355
    },
    {
      "epoch": 0.22181146025878004,
      "grad_norm": 0.23363694548606873,
      "learning_rate": 0.0001844742774510851,
      "loss": 0.8763,
      "step": 360
    },
    {
      "epoch": 0.22489217498459643,
      "grad_norm": 0.17385488748550415,
      "learning_rate": 0.00018392437782835475,
      "loss": 0.8606,
      "step": 365
    },
    {
      "epoch": 0.22797288971041282,
      "grad_norm": 0.7904961109161377,
      "learning_rate": 0.00018336575815820766,
      "loss": 0.8528,
      "step": 370
    },
    {
      "epoch": 0.23105360443622922,
      "grad_norm": 0.4967353940010071,
      "learning_rate": 0.00018279847648325478,
      "loss": 0.7325,
      "step": 375
    },
    {
      "epoch": 0.2341343191620456,
      "grad_norm": 0.29136228561401367,
      "learning_rate": 0.0001822225917461208,
      "loss": 0.7879,
      "step": 380
    },
    {
      "epoch": 0.23721503388786197,
      "grad_norm": 0.2662856876850128,
      "learning_rate": 0.0001816381637833198,
      "loss": 0.8703,
      "step": 385
    },
    {
      "epoch": 0.24029574861367836,
      "grad_norm": 0.22255033254623413,
      "learning_rate": 0.00018104525331903799,
      "loss": 0.9028,
      "step": 390
    },
    {
      "epoch": 0.24337646333949475,
      "grad_norm": 0.3649587035179138,
      "learning_rate": 0.00018044392195882427,
      "loss": 0.9442,
      "step": 395
    },
    {
      "epoch": 0.24645717806531114,
      "grad_norm": 0.4449125826358795,
      "learning_rate": 0.00017983423218318918,
      "loss": 0.7348,
      "step": 400
    },
    {
      "epoch": 0.24953789279112754,
      "grad_norm": 0.309292197227478,
      "learning_rate": 0.00017921624734111292,
      "loss": 0.8436,
      "step": 405
    },
    {
      "epoch": 0.2526186075169439,
      "grad_norm": 0.23456443846225739,
      "learning_rate": 0.00017859003164346336,
      "loss": 0.8975,
      "step": 410
    },
    {
      "epoch": 0.2556993222427603,
      "grad_norm": 0.18259836733341217,
      "learning_rate": 0.0001779556501563239,
      "loss": 0.8344,
      "step": 415
    },
    {
      "epoch": 0.2587800369685767,
      "grad_norm": 0.3429722785949707,
      "learning_rate": 0.00017731316879423327,
      "loss": 0.8817,
      "step": 420
    },
    {
      "epoch": 0.2618607516943931,
      "grad_norm": 0.43789297342300415,
      "learning_rate": 0.00017666265431333654,
      "loss": 0.7864,
      "step": 425
    },
    {
      "epoch": 0.2649414664202095,
      "grad_norm": 0.23547668755054474,
      "learning_rate": 0.000176004174304449,
      "loss": 0.8631,
      "step": 430
    },
    {
      "epoch": 0.2680221811460259,
      "grad_norm": 0.243401899933815,
      "learning_rate": 0.00017533779718603313,
      "loss": 0.9021,
      "step": 435
    },
    {
      "epoch": 0.2711028958718423,
      "grad_norm": 0.19024285674095154,
      "learning_rate": 0.00017466359219708985,
      "loss": 0.8052,
      "step": 440
    },
    {
      "epoch": 0.27418361059765867,
      "grad_norm": 0.3085165321826935,
      "learning_rate": 0.00017398162938996422,
      "loss": 0.8126,
      "step": 445
    },
    {
      "epoch": 0.27726432532347506,
      "grad_norm": 0.32612961530685425,
      "learning_rate": 0.00017329197962306664,
      "loss": 0.689,
      "step": 450
    },
    {
      "epoch": 0.28034504004929145,
      "grad_norm": 0.2392513006925583,
      "learning_rate": 0.00017259471455351072,
      "loss": 0.8309,
      "step": 455
    },
    {
      "epoch": 0.28342575477510784,
      "grad_norm": 0.2373919039964676,
      "learning_rate": 0.0001718899066296675,
      "loss": 0.9067,
      "step": 460
    },
    {
      "epoch": 0.28650646950092423,
      "grad_norm": 0.21918949484825134,
      "learning_rate": 0.000171177629083638,
      "loss": 0.8411,
      "step": 465
    },
    {
      "epoch": 0.2895871842267406,
      "grad_norm": 0.35443195700645447,
      "learning_rate": 0.0001704579559236441,
      "loss": 0.8656,
      "step": 470
    },
    {
      "epoch": 0.292667898952557,
      "grad_norm": 0.5146204829216003,
      "learning_rate": 0.00016973096192633884,
      "loss": 0.6823,
      "step": 475
    },
    {
      "epoch": 0.2957486136783734,
      "grad_norm": 0.2694316804409027,
      "learning_rate": 0.00016899672262903677,
      "loss": 0.7831,
      "step": 480
    },
    {
      "epoch": 0.2988293284041898,
      "grad_norm": 0.25592705607414246,
      "learning_rate": 0.00016825531432186543,
      "loss": 0.9099,
      "step": 485
    },
    {
      "epoch": 0.30191004313000613,
      "grad_norm": 0.1862918585538864,
      "learning_rate": 0.00016750681403983846,
      "loss": 0.9212,
      "step": 490
    },
    {
      "epoch": 0.3049907578558225,
      "grad_norm": 0.3766527473926544,
      "learning_rate": 0.00016675129955485152,
      "loss": 0.8217,
      "step": 495
    },
    {
      "epoch": 0.3080714725816389,
      "grad_norm": 0.3869279623031616,
      "learning_rate": 0.00016598884936760131,
      "loss": 0.7405,
      "step": 500
    },
    {
      "epoch": 0.3111521873074553,
      "grad_norm": 0.228221595287323,
      "learning_rate": 0.00016521954269942918,
      "loss": 0.8923,
      "step": 505
    },
    {
      "epoch": 0.3142329020332717,
      "grad_norm": 0.24637649953365326,
      "learning_rate": 0.00016444345948408984,
      "loss": 0.8654,
      "step": 510
    },
    {
      "epoch": 0.3173136167590881,
      "grad_norm": 0.21193628013134003,
      "learning_rate": 0.0001636606803594457,
      "loss": 0.9014,
      "step": 515
    },
    {
      "epoch": 0.3203943314849045,
      "grad_norm": 0.388420969247818,
      "learning_rate": 0.0001628712866590885,
      "loss": 0.8323,
      "step": 520
    },
    {
      "epoch": 0.3234750462107209,
      "grad_norm": 0.43453922867774963,
      "learning_rate": 0.00016207536040388845,
      "loss": 0.7208,
      "step": 525
    },
    {
      "epoch": 0.32655576093653726,
      "grad_norm": 0.2307548075914383,
      "learning_rate": 0.0001612729842934718,
      "loss": 0.8748,
      "step": 530
    },
    {
      "epoch": 0.32963647566235366,
      "grad_norm": 0.2415218949317932,
      "learning_rate": 0.00016046424169762827,
      "loss": 0.9362,
      "step": 535
    },
    {
      "epoch": 0.33271719038817005,
      "grad_norm": 0.20099782943725586,
      "learning_rate": 0.0001596492166476485,
      "loss": 0.8561,
      "step": 540
    },
    {
      "epoch": 0.33579790511398644,
      "grad_norm": 0.3241688311100006,
      "learning_rate": 0.0001588279938275929,
      "loss": 0.8262,
      "step": 545
    },
    {
      "epoch": 0.33887861983980283,
      "grad_norm": 0.47993138432502747,
      "learning_rate": 0.00015800065856549269,
      "loss": 0.6732,
      "step": 550
    },
    {
      "epoch": 0.3419593345656192,
      "grad_norm": 0.2494264543056488,
      "learning_rate": 0.00015716729682448393,
      "loss": 0.8533,
      "step": 555
    },
    {
      "epoch": 0.3450400492914356,
      "grad_norm": 0.22302919626235962,
      "learning_rate": 0.0001563279951938758,
      "loss": 0.8869,
      "step": 560
    },
    {
      "epoch": 0.348120764017252,
      "grad_norm": 0.1880647987127304,
      "learning_rate": 0.00015548284088015354,
      "loss": 0.8315,
      "step": 565
    },
    {
      "epoch": 0.3512014787430684,
      "grad_norm": 0.38539907336235046,
      "learning_rate": 0.00015463192169791741,
      "loss": 0.8108,
      "step": 570
    },
    {
      "epoch": 0.3542821934688848,
      "grad_norm": 0.5778854489326477,
      "learning_rate": 0.0001537753260607584,
      "loss": 0.7275,
      "step": 575
    },
    {
      "epoch": 0.3573629081947012,
      "grad_norm": 0.22120217978954315,
      "learning_rate": 0.00015291314297207175,
      "loss": 0.8384,
      "step": 580
    },
    {
      "epoch": 0.36044362292051757,
      "grad_norm": 0.19677136838436127,
      "learning_rate": 0.0001520454620158093,
      "loss": 0.8899,
      "step": 585
    },
    {
      "epoch": 0.36352433764633396,
      "grad_norm": 0.1967296153306961,
      "learning_rate": 0.00015117237334717117,
      "loss": 0.8315,
      "step": 590
    },
    {
      "epoch": 0.36660505237215035,
      "grad_norm": 0.3274117410182953,
      "learning_rate": 0.00015029396768323846,
      "loss": 0.9155,
      "step": 595
    },
    {
      "epoch": 0.36968576709796674,
      "grad_norm": 0.4024091064929962,
      "learning_rate": 0.00014941033629354734,
      "loss": 0.7731,
      "step": 600
    },
    {
      "epoch": 0.37276648182378314,
      "grad_norm": 0.25141751766204834,
      "learning_rate": 0.00014852157099060596,
      "loss": 0.8884,
      "step": 605
    },
    {
      "epoch": 0.3758471965495995,
      "grad_norm": 0.22811076045036316,
      "learning_rate": 0.00014762776412035456,
      "loss": 0.9339,
      "step": 610
    },
    {
      "epoch": 0.3789279112754159,
      "grad_norm": 0.20071439445018768,
      "learning_rate": 0.00014672900855257056,
      "loss": 0.8493,
      "step": 615
    },
    {
      "epoch": 0.3820086260012323,
      "grad_norm": 0.38140973448753357,
      "learning_rate": 0.00014582539767121904,
      "loss": 0.8539,
      "step": 620
    },
    {
      "epoch": 0.3850893407270487,
      "grad_norm": 0.3909998834133148,
      "learning_rate": 0.0001449170253647498,
      "loss": 0.648,
      "step": 625
    },
    {
      "epoch": 0.38817005545286504,
      "grad_norm": 0.2962195873260498,
      "learning_rate": 0.0001440039860163419,
      "loss": 0.8375,
      "step": 630
    },
    {
      "epoch": 0.39125077017868143,
      "grad_norm": 0.24794480204582214,
      "learning_rate": 0.00014308637449409706,
      "loss": 0.8591,
      "step": 635
    },
    {
      "epoch": 0.3943314849044978,
      "grad_norm": 0.20156452059745789,
      "learning_rate": 0.00014216428614118243,
      "loss": 0.895,
      "step": 640
    },
    {
      "epoch": 0.3974121996303142,
      "grad_norm": 0.3954678773880005,
      "learning_rate": 0.00014123781676592418,
      "loss": 0.8753,
      "step": 645
    },
    {
      "epoch": 0.4004929143561306,
      "grad_norm": 0.48697060346603394,
      "learning_rate": 0.00014030706263185247,
      "loss": 0.749,
      "step": 650
    },
    {
      "epoch": 0.403573629081947,
      "grad_norm": 0.243213951587677,
      "learning_rate": 0.00013937212044769955,
      "loss": 0.8205,
      "step": 655
    },
    {
      "epoch": 0.4066543438077634,
      "grad_norm": 0.2900526225566864,
      "learning_rate": 0.0001384330873573513,
      "loss": 0.8888,
      "step": 660
    },
    {
      "epoch": 0.4097350585335798,
      "grad_norm": 0.19511188566684723,
      "learning_rate": 0.00013749006092975347,
      "loss": 0.8602,
      "step": 665
    },
    {
      "epoch": 0.41281577325939617,
      "grad_norm": 0.3556269705295563,
      "learning_rate": 0.00013654313914877414,
      "loss": 0.8346,
      "step": 670
    },
    {
      "epoch": 0.41589648798521256,
      "grad_norm": 0.34523579478263855,
      "learning_rate": 0.00013559242040302272,
      "loss": 0.6822,
      "step": 675
    },
    {
      "epoch": 0.41897720271102895,
      "grad_norm": 0.2661334276199341,
      "learning_rate": 0.00013463800347562706,
      "loss": 0.8797,
      "step": 680
    },
    {
      "epoch": 0.42205791743684534,
      "grad_norm": 0.1797327697277069,
      "learning_rate": 0.00013367998753396944,
      "loss": 0.7698,
      "step": 685
    },
    {
      "epoch": 0.42513863216266173,
      "grad_norm": 0.19273746013641357,
      "learning_rate": 0.00013271847211938285,
      "loss": 0.8585,
      "step": 690
    },
    {
      "epoch": 0.4282193468884781,
      "grad_norm": 0.3688190281391144,
      "learning_rate": 0.0001317535571368082,
      "loss": 0.8312,
      "step": 695
    },
    {
      "epoch": 0.4313000616142945,
      "grad_norm": 0.37159615755081177,
      "learning_rate": 0.00013078534284441382,
      "loss": 0.7529,
      "step": 700
    },
    {
      "epoch": 0.4343807763401109,
      "grad_norm": 0.19623301923274994,
      "learning_rate": 0.00012981392984317834,
      "loss": 0.7814,
      "step": 705
    },
    {
      "epoch": 0.4374614910659273,
      "grad_norm": 0.2310985028743744,
      "learning_rate": 0.00012883941906643786,
      "loss": 0.8723,
      "step": 710
    },
    {
      "epoch": 0.4405422057917437,
      "grad_norm": 0.21924515068531036,
      "learning_rate": 0.00012786191176939848,
      "loss": 0.8321,
      "step": 715
    },
    {
      "epoch": 0.4436229205175601,
      "grad_norm": 0.3389222323894501,
      "learning_rate": 0.00012688150951861582,
      "loss": 0.732,
      "step": 720
    },
    {
      "epoch": 0.4467036352433765,
      "grad_norm": 0.5648886561393738,
      "learning_rate": 0.00012589831418144154,
      "loss": 0.7315,
      "step": 725
    },
    {
      "epoch": 0.44978434996919286,
      "grad_norm": 0.24998104572296143,
      "learning_rate": 0.00012491242791543922,
      "loss": 0.8625,
      "step": 730
    },
    {
      "epoch": 0.45286506469500926,
      "grad_norm": 0.21635833382606506,
      "learning_rate": 0.00012392395315776963,
      "loss": 0.8461,
      "step": 735
    },
    {
      "epoch": 0.45594577942082565,
      "grad_norm": 0.20510129630565643,
      "learning_rate": 0.00012293299261454725,
      "loss": 0.8047,
      "step": 740
    },
    {
      "epoch": 0.45902649414664204,
      "grad_norm": 0.3469765782356262,
      "learning_rate": 0.00012193964925016872,
      "loss": 0.8065,
      "step": 745
    },
    {
      "epoch": 0.46210720887245843,
      "grad_norm": 0.5244903564453125,
      "learning_rate": 0.00012094402627661447,
      "loss": 0.6974,
      "step": 750
    },
    {
      "epoch": 0.4651879235982748,
      "grad_norm": 0.24346120655536652,
      "learning_rate": 0.00011994622714272448,
      "loss": 0.8694,
      "step": 755
    },
    {
      "epoch": 0.4682686383240912,
      "grad_norm": 0.2859845459461212,
      "learning_rate": 0.00011894635552344975,
      "loss": 0.8711,
      "step": 760
    },
    {
      "epoch": 0.4713493530499076,
      "grad_norm": 0.16937606036663055,
      "learning_rate": 0.00011794451530908011,
      "loss": 0.8107,
      "step": 765
    },
    {
      "epoch": 0.47443006777572394,
      "grad_norm": 0.35386890172958374,
      "learning_rate": 0.00011694081059444946,
      "loss": 0.8089,
      "step": 770
    },
    {
      "epoch": 0.47751078250154033,
      "grad_norm": 0.43906474113464355,
      "learning_rate": 0.0001159353456681201,
      "loss": 0.7098,
      "step": 775
    },
    {
      "epoch": 0.4805914972273567,
      "grad_norm": 0.23767797648906708,
      "learning_rate": 0.00011492822500154667,
      "loss": 0.8025,
      "step": 780
    },
    {
      "epoch": 0.4836722119531731,
      "grad_norm": 0.22543567419052124,
      "learning_rate": 0.00011391955323822126,
      "loss": 0.8193,
      "step": 785
    },
    {
      "epoch": 0.4867529266789895,
      "grad_norm": 0.18470603227615356,
      "learning_rate": 0.00011290943518280057,
      "loss": 0.9265,
      "step": 790
    },
    {
      "epoch": 0.4898336414048059,
      "grad_norm": 0.3396364450454712,
      "learning_rate": 0.0001118979757902162,
      "loss": 0.789,
      "step": 795
    },
    {
      "epoch": 0.4929143561306223,
      "grad_norm": 0.41464564204216003,
      "learning_rate": 0.00011088528015476964,
      "loss": 0.7778,
      "step": 800
    },
    {
      "epoch": 0.4959950708564387,
      "grad_norm": 0.20553241670131683,
      "learning_rate": 0.00010987145349921251,
      "loss": 0.8102,
      "step": 805
    },
    {
      "epoch": 0.49907578558225507,
      "grad_norm": 0.20871993899345398,
      "learning_rate": 0.0001088566011638134,
      "loss": 0.8346,
      "step": 810
    },
    {
      "epoch": 0.5021565003080715,
      "grad_norm": 0.19382049143314362,
      "learning_rate": 0.00010784082859541292,
      "loss": 0.8303,
      "step": 815
    },
    {
      "epoch": 0.5052372150338879,
      "grad_norm": 0.34537407755851746,
      "learning_rate": 0.0001068242413364671,
      "loss": 0.8067,
      "step": 820
    },
    {
      "epoch": 0.5083179297597042,
      "grad_norm": 0.5449668765068054,
      "learning_rate": 0.00010580694501408138,
      "loss": 0.7138,
      "step": 825
    },
    {
      "epoch": 0.5113986444855206,
      "grad_norm": 0.22544096410274506,
      "learning_rate": 0.00010478904532903535,
      "loss": 0.7688,
      "step": 830
    },
    {
      "epoch": 0.514479359211337,
      "grad_norm": 0.21812790632247925,
      "learning_rate": 0.00010377064804480025,
      "loss": 0.9282,
      "step": 835
    },
    {
      "epoch": 0.5175600739371534,
      "grad_norm": 0.189622700214386,
      "learning_rate": 0.00010275185897654971,
      "loss": 0.791,
      "step": 840
    },
    {
      "epoch": 0.5206407886629698,
      "grad_norm": 0.3448941707611084,
      "learning_rate": 0.00010173278398016501,
      "loss": 0.7811,
      "step": 845
    },
    {
      "epoch": 0.5237215033887862,
      "grad_norm": 0.35987991094589233,
      "learning_rate": 0.00010071352894123654,
      "loss": 0.7112,
      "step": 850
    },
    {
      "epoch": 0.5268022181146026,
      "grad_norm": 0.2988508939743042,
      "learning_rate": 9.969419976406165e-05,
      "loss": 0.8362,
      "step": 855
    },
    {
      "epoch": 0.529882932840419,
      "grad_norm": 0.24420854449272156,
      "learning_rate": 9.867490236064108e-05,
      "loss": 0.8405,
      "step": 860
    },
    {
      "epoch": 0.5329636475662354,
      "grad_norm": 0.22778154909610748,
      "learning_rate": 9.765574263967396e-05,
      "loss": 0.9235,
      "step": 865
    },
    {
      "epoch": 0.5360443622920518,
      "grad_norm": 0.3345077335834503,
      "learning_rate": 9.66368264955539e-05,
      "loss": 0.7999,
      "step": 870
    },
    {
      "epoch": 0.5391250770178682,
      "grad_norm": 0.39567625522613525,
      "learning_rate": 9.56182597973658e-05,
      "loss": 0.7299,
      "step": 875
    },
    {
      "epoch": 0.5422057917436846,
      "grad_norm": 0.19459989666938782,
      "learning_rate": 9.460014837788605e-05,
      "loss": 0.8329,
      "step": 880
    },
    {
      "epoch": 0.5452865064695009,
      "grad_norm": 0.2300824522972107,
      "learning_rate": 9.358259802258581e-05,
      "loss": 0.8484,
      "step": 885
    },
    {
      "epoch": 0.5483672211953173,
      "grad_norm": 0.1935604363679886,
      "learning_rate": 9.256571445863972e-05,
      "loss": 0.8014,
      "step": 890
    },
    {
      "epoch": 0.5514479359211337,
      "grad_norm": 0.36507704854011536,
      "learning_rate": 9.154960334394027e-05,
      "loss": 0.8372,
      "step": 895
    },
    {
      "epoch": 0.5545286506469501,
      "grad_norm": 0.3593067228794098,
      "learning_rate": 9.053437025611973e-05,
      "loss": 0.6882,
      "step": 900
    },
    {
      "epoch": 0.5576093653727665,
      "grad_norm": 0.23021775484085083,
      "learning_rate": 8.952012068158027e-05,
      "loss": 0.8409,
      "step": 905
    },
    {
      "epoch": 0.5606900800985829,
      "grad_norm": 0.189047172665596,
      "learning_rate": 8.850696000453326e-05,
      "loss": 0.8741,
      "step": 910
    },
    {
      "epoch": 0.5637707948243993,
      "grad_norm": 0.18688207864761353,
      "learning_rate": 8.749499349604993e-05,
      "loss": 0.8321,
      "step": 915
    },
    {
      "epoch": 0.5668515095502157,
      "grad_norm": 0.3526299297809601,
      "learning_rate": 8.64843263031228e-05,
      "loss": 0.7838,
      "step": 920
    },
    {
      "epoch": 0.5699322242760321,
      "grad_norm": 0.4248710870742798,
      "learning_rate": 8.547506343774097e-05,
      "loss": 0.6726,
      "step": 925
    },
    {
      "epoch": 0.5730129390018485,
      "grad_norm": 0.2310338169336319,
      "learning_rate": 8.446730976597878e-05,
      "loss": 0.8931,
      "step": 930
    },
    {
      "epoch": 0.5760936537276649,
      "grad_norm": 0.26518529653549194,
      "learning_rate": 8.346116999709975e-05,
      "loss": 0.8133,
      "step": 935
    },
    {
      "epoch": 0.5791743684534812,
      "grad_norm": 0.20033273100852966,
      "learning_rate": 8.245674867267724e-05,
      "loss": 0.7912,
      "step": 940
    },
    {
      "epoch": 0.5822550831792976,
      "grad_norm": 0.41144856810569763,
      "learning_rate": 8.145415015573183e-05,
      "loss": 0.8607,
      "step": 945
    },
    {
      "epoch": 0.585335797905114,
      "grad_norm": 0.3872707188129425,
      "learning_rate": 8.045347861988789e-05,
      "loss": 0.6516,
      "step": 950
    },
    {
      "epoch": 0.5884165126309304,
      "grad_norm": 0.28106680512428284,
      "learning_rate": 7.945483803854936e-05,
      "loss": 0.8349,
      "step": 955
    },
    {
      "epoch": 0.5914972273567468,
      "grad_norm": 0.2388414740562439,
      "learning_rate": 7.845833217409675e-05,
      "loss": 0.9233,
      "step": 960
    },
    {
      "epoch": 0.5945779420825632,
      "grad_norm": 0.18663132190704346,
      "learning_rate": 7.746406456710564e-05,
      "loss": 0.8001,
      "step": 965
    },
    {
      "epoch": 0.5976586568083796,
      "grad_norm": 0.3416728675365448,
      "learning_rate": 7.64721385255886e-05,
      "loss": 0.8076,
      "step": 970
    },
    {
      "epoch": 0.600739371534196,
      "grad_norm": 0.400666743516922,
      "learning_rate": 7.548265711426104e-05,
      "loss": 0.6584,
      "step": 975
    },
    {
      "epoch": 0.6038200862600123,
      "grad_norm": 0.2638736963272095,
      "learning_rate": 7.449572314383237e-05,
      "loss": 0.8991,
      "step": 980
    },
    {
      "epoch": 0.6069008009858287,
      "grad_norm": 0.22274282574653625,
      "learning_rate": 7.351143916032374e-05,
      "loss": 0.9054,
      "step": 985
    },
    {
      "epoch": 0.609981515711645,
      "grad_norm": 0.1962013691663742,
      "learning_rate": 7.252990743441293e-05,
      "loss": 0.8379,
      "step": 990
    },
    {
      "epoch": 0.6130622304374614,
      "grad_norm": 0.35286328196525574,
      "learning_rate": 7.155122995080827e-05,
      "loss": 0.8635,
      "step": 995
    },
    {
      "epoch": 0.6161429451632778,
      "grad_norm": 0.48713040351867676,
      "learning_rate": 7.057550839765188e-05,
      "loss": 0.7351,
      "step": 1000
    },
    {
      "epoch": 0.6192236598890942,
      "grad_norm": 0.2801983952522278,
      "learning_rate": 6.960284415595407e-05,
      "loss": 0.8368,
      "step": 1005
    },
    {
      "epoch": 0.6223043746149106,
      "grad_norm": 0.22647203505039215,
      "learning_rate": 6.863333828905929e-05,
      "loss": 0.9548,
      "step": 1010
    },
    {
      "epoch": 0.625385089340727,
      "grad_norm": 0.1952330321073532,
      "learning_rate": 6.766709153214542e-05,
      "loss": 0.8465,
      "step": 1015
    },
    {
      "epoch": 0.6284658040665434,
      "grad_norm": 0.3249083161354065,
      "learning_rate": 6.670420428175705e-05,
      "loss": 0.8318,
      "step": 1020
    },
    {
      "epoch": 0.6315465187923598,
      "grad_norm": 0.3647523522377014,
      "learning_rate": 6.574477658537375e-05,
      "loss": 0.7159,
      "step": 1025
    },
    {
      "epoch": 0.6346272335181762,
      "grad_norm": 0.27877968549728394,
      "learning_rate": 6.4788908131015e-05,
      "loss": 0.7783,
      "step": 1030
    },
    {
      "epoch": 0.6377079482439926,
      "grad_norm": 0.21505144238471985,
      "learning_rate": 6.38366982368819e-05,
      "loss": 0.755,
      "step": 1035
    },
    {
      "epoch": 0.640788662969809,
      "grad_norm": 0.19141019880771637,
      "learning_rate": 6.288824584103816e-05,
      "loss": 0.828,
      "step": 1040
    },
    {
      "epoch": 0.6438693776956254,
      "grad_norm": 0.36781400442123413,
      "learning_rate": 6.194364949112953e-05,
      "loss": 0.7687,
      "step": 1045
    },
    {
      "epoch": 0.6469500924214417,
      "grad_norm": 0.34893959760665894,
      "learning_rate": 6.100300733414474e-05,
      "loss": 0.6824,
      "step": 1050
    },
    {
      "epoch": 0.6500308071472581,
      "grad_norm": 0.2389339655637741,
      "learning_rate": 6.0066417106217455e-05,
      "loss": 0.8163,
      "step": 1055
    },
    {
      "epoch": 0.6531115218730745,
      "grad_norm": 0.23985756933689117,
      "learning_rate": 5.9133976122471214e-05,
      "loss": 0.8292,
      "step": 1060
    },
    {
      "epoch": 0.6561922365988909,
      "grad_norm": 0.1888720840215683,
      "learning_rate": 5.82057812669081e-05,
      "loss": 0.8482,
      "step": 1065
    },
    {
      "epoch": 0.6592729513247073,
      "grad_norm": 0.31835633516311646,
      "learning_rate": 5.728192898234195e-05,
      "loss": 0.754,
      "step": 1070
    },
    {
      "epoch": 0.6623536660505237,
      "grad_norm": 0.4028412103652954,
      "learning_rate": 5.6362515260377835e-05,
      "loss": 0.6716,
      "step": 1075
    },
    {
      "epoch": 0.6654343807763401,
      "grad_norm": 0.23141424357891083,
      "learning_rate": 5.544763563143793e-05,
      "loss": 0.8604,
      "step": 1080
    },
    {
      "epoch": 0.6685150955021565,
      "grad_norm": 0.2335502952337265,
      "learning_rate": 5.4537385154835864e-05,
      "loss": 0.8513,
      "step": 1085
    },
    {
      "epoch": 0.6715958102279729,
      "grad_norm": 0.1902121901512146,
      "learning_rate": 5.363185840889935e-05,
      "loss": 0.793,
      "step": 1090
    },
    {
      "epoch": 0.6746765249537893,
      "grad_norm": 0.3410889804363251,
      "learning_rate": 5.273114948114346e-05,
      "loss": 0.8378,
      "step": 1095
    },
    {
      "epoch": 0.6777572396796057,
      "grad_norm": 0.41296476125717163,
      "learning_rate": 5.1835351958494515e-05,
      "loss": 0.678,
      "step": 1100
    },
    {
      "epoch": 0.680837954405422,
      "grad_norm": 0.25026175379753113,
      "learning_rate": 5.094455891756587e-05,
      "loss": 0.8418,
      "step": 1105
    },
    {
      "epoch": 0.6839186691312384,
      "grad_norm": 0.22610455751419067,
      "learning_rate": 5.00588629149872e-05,
      "loss": 0.8793,
      "step": 1110
    },
    {
      "epoch": 0.6869993838570548,
      "grad_norm": 0.19563516974449158,
      "learning_rate": 4.91783559777873e-05,
      "loss": 0.8106,
      "step": 1115
    },
    {
      "epoch": 0.6900800985828712,
      "grad_norm": 0.3695812225341797,
      "learning_rate": 4.830312959383238e-05,
      "loss": 0.7451,
      "step": 1120
    },
    {
      "epoch": 0.6931608133086876,
      "grad_norm": 0.3287087082862854,
      "learning_rate": 4.7433274702319815e-05,
      "loss": 0.6435,
      "step": 1125
    },
    {
      "epoch": 0.696241528034504,
      "grad_norm": 0.22749105095863342,
      "learning_rate": 4.656888168432962e-05,
      "loss": 0.8142,
      "step": 1130
    },
    {
      "epoch": 0.6993222427603204,
      "grad_norm": 0.227505624294281,
      "learning_rate": 4.571004035343315e-05,
      "loss": 0.8842,
      "step": 1135
    },
    {
      "epoch": 0.7024029574861368,
      "grad_norm": 0.18932074308395386,
      "learning_rate": 4.485683994636144e-05,
      "loss": 0.7862,
      "step": 1140
    },
    {
      "epoch": 0.7054836722119532,
      "grad_norm": 0.3436351716518402,
      "learning_rate": 4.400936911373308e-05,
      "loss": 0.8462,
      "step": 1145
    },
    {
      "epoch": 0.7085643869377696,
      "grad_norm": 0.3784382939338684,
      "learning_rate": 4.3167715910842966e-05,
      "loss": 0.7086,
      "step": 1150
    },
    {
      "epoch": 0.711645101663586,
      "grad_norm": 0.2360491007566452,
      "learning_rate": 4.2331967788513295e-05,
      "loss": 0.8275,
      "step": 1155
    },
    {
      "epoch": 0.7147258163894024,
      "grad_norm": 0.2204228937625885,
      "learning_rate": 4.1502211584006836e-05,
      "loss": 0.8698,
      "step": 1160
    },
    {
      "epoch": 0.7178065311152187,
      "grad_norm": 0.1819538027048111,
      "learning_rate": 4.067853351200446e-05,
      "loss": 0.8826,
      "step": 1165
    },
    {
      "epoch": 0.7208872458410351,
      "grad_norm": 0.3149859607219696,
      "learning_rate": 3.986101915564695e-05,
      "loss": 0.7709,
      "step": 1170
    },
    {
      "epoch": 0.7239679605668515,
      "grad_norm": 0.5513156652450562,
      "learning_rate": 3.904975345764262e-05,
      "loss": 0.6964,
      "step": 1175
    },
    {
      "epoch": 0.7270486752926679,
      "grad_norm": 0.2228625863790512,
      "learning_rate": 3.824482071144163e-05,
      "loss": 0.8452,
      "step": 1180
    },
    {
      "epoch": 0.7301293900184843,
      "grad_norm": 0.24210941791534424,
      "learning_rate": 3.744630455247739e-05,
      "loss": 0.8381,
      "step": 1185
    },
    {
      "epoch": 0.7332101047443007,
      "grad_norm": 0.18489736318588257,
      "learning_rate": 3.6654287949476626e-05,
      "loss": 0.8177,
      "step": 1190
    },
    {
      "epoch": 0.7362908194701171,
      "grad_norm": 0.328374981880188,
      "learning_rate": 3.586885319583858e-05,
      "loss": 0.8,
      "step": 1195
    },
    {
      "epoch": 0.7393715341959335,
      "grad_norm": 0.4047057330608368,
      "learning_rate": 3.5090081901084525e-05,
      "loss": 0.7031,
      "step": 1200
    },
    {
      "epoch": 0.7424522489217499,
      "grad_norm": 0.25028735399246216,
      "learning_rate": 3.431805498237808e-05,
      "loss": 0.8824,
      "step": 1205
    },
    {
      "epoch": 0.7455329636475663,
      "grad_norm": 0.268685519695282,
      "learning_rate": 3.355285265611784e-05,
      "loss": 0.8874,
      "step": 1210
    },
    {
      "epoch": 0.7486136783733827,
      "grad_norm": 0.17774948477745056,
      "learning_rate": 3.279455442960238e-05,
      "loss": 0.8467,
      "step": 1215
    },
    {
      "epoch": 0.751694393099199,
      "grad_norm": 0.3131676912307739,
      "learning_rate": 3.204323909276924e-05,
      "loss": 0.8565,
      "step": 1220
    },
    {
      "epoch": 0.7547751078250154,
      "grad_norm": 0.32786959409713745,
      "learning_rate": 3.1298984710008484e-05,
      "loss": 0.6674,
      "step": 1225
    },
    {
      "epoch": 0.7578558225508318,
      "grad_norm": 0.31701722741127014,
      "learning_rate": 3.056186861205136e-05,
      "loss": 0.7799,
      "step": 1230
    },
    {
      "epoch": 0.7609365372766482,
      "grad_norm": 0.2162780612707138,
      "learning_rate": 2.9831967387935467e-05,
      "loss": 0.8906,
      "step": 1235
    },
    {
      "epoch": 0.7640172520024646,
      "grad_norm": 0.1988825500011444,
      "learning_rate": 2.9109356877046712e-05,
      "loss": 0.7644,
      "step": 1240
    },
    {
      "epoch": 0.767097966728281,
      "grad_norm": 0.35476982593536377,
      "learning_rate": 2.8394112161239605e-05,
      "loss": 0.8323,
      "step": 1245
    },
    {
      "epoch": 0.7701786814540974,
      "grad_norm": 0.4047326147556305,
      "learning_rate": 2.7686307557035685e-05,
      "loss": 0.6564,
      "step": 1250
    },
    {
      "epoch": 0.7732593961799138,
      "grad_norm": 0.20286810398101807,
      "learning_rate": 2.6986016607901908e-05,
      "loss": 0.7925,
      "step": 1255
    },
    {
      "epoch": 0.7763401109057301,
      "grad_norm": 0.23300860822200775,
      "learning_rate": 2.629331207660931e-05,
      "loss": 0.7996,
      "step": 1260
    },
    {
      "epoch": 0.7794208256315465,
      "grad_norm": 0.19397461414337158,
      "learning_rate": 2.5608265937672436e-05,
      "loss": 0.817,
      "step": 1265
    },
    {
      "epoch": 0.7825015403573629,
      "grad_norm": 0.32662224769592285,
      "learning_rate": 2.4930949369871203e-05,
      "loss": 0.8455,
      "step": 1270
    },
    {
      "epoch": 0.7855822550831792,
      "grad_norm": 0.4699757993221283,
      "learning_rate": 2.426143274885493e-05,
      "loss": 0.6576,
      "step": 1275
    },
    {
      "epoch": 0.7886629698089956,
      "grad_norm": 0.27602919936180115,
      "learning_rate": 2.359978563983022e-05,
      "loss": 0.8137,
      "step": 1280
    },
    {
      "epoch": 0.791743684534812,
      "grad_norm": 0.20884881913661957,
      "learning_rate": 2.2946076790332827e-05,
      "loss": 0.7759,
      "step": 1285
    },
    {
      "epoch": 0.7948243992606284,
      "grad_norm": 0.20058295130729675,
      "learning_rate": 2.2300374123084522e-05,
      "loss": 0.7794,
      "step": 1290
    },
    {
      "epoch": 0.7979051139864448,
      "grad_norm": 0.38759925961494446,
      "learning_rate": 2.166274472893567e-05,
      "loss": 0.801,
      "step": 1295
    },
    {
      "epoch": 0.8009858287122612,
      "grad_norm": 0.3663594722747803,
      "learning_rate": 2.1033254859894226e-05,
      "loss": 0.6458,
      "step": 1300
    },
    {
      "epoch": 0.8040665434380776,
      "grad_norm": 0.20937398076057434,
      "learning_rate": 2.041196992224206e-05,
      "loss": 0.8004,
      "step": 1305
    },
    {
      "epoch": 0.807147258163894,
      "grad_norm": 0.23166590929031372,
      "learning_rate": 1.9798954469738762e-05,
      "loss": 0.7889,
      "step": 1310
    },
    {
      "epoch": 0.8102279728897104,
      "grad_norm": 0.1858280897140503,
      "learning_rate": 1.919427219691453e-05,
      "loss": 0.811,
      "step": 1315
    },
    {
      "epoch": 0.8133086876155268,
      "grad_norm": 0.3292345702648163,
      "learning_rate": 1.8597985932451856e-05,
      "loss": 0.7721,
      "step": 1320
    },
    {
      "epoch": 0.8163894023413432,
      "grad_norm": 0.41900116205215454,
      "learning_rate": 1.8010157632657543e-05,
      "loss": 0.6585,
      "step": 1325
    },
    {
      "epoch": 0.8194701170671596,
      "grad_norm": 0.2675899565219879,
      "learning_rate": 1.7430848375025176e-05,
      "loss": 0.8278,
      "step": 1330
    },
    {
      "epoch": 0.822550831792976,
      "grad_norm": 0.2361748367547989,
      "learning_rate": 1.686011835188891e-05,
      "loss": 0.828,
      "step": 1335
    },
    {
      "epoch": 0.8256315465187923,
      "grad_norm": 0.17088614404201508,
      "learning_rate": 1.6298026864169335e-05,
      "loss": 0.8391,
      "step": 1340
    },
    {
      "epoch": 0.8287122612446087,
      "grad_norm": 0.3461218476295471,
      "learning_rate": 1.5744632315211815e-05,
      "loss": 0.8286,
      "step": 1345
    },
    {
      "epoch": 0.8317929759704251,
      "grad_norm": 0.37459784746170044,
      "learning_rate": 1.5199992204718294e-05,
      "loss": 0.719,
      "step": 1350
    },
    {
      "epoch": 0.8348736906962415,
      "grad_norm": 0.24230735003948212,
      "learning_rate": 1.4664163122772689e-05,
      "loss": 0.8367,
      "step": 1355
    },
    {
      "epoch": 0.8379544054220579,
      "grad_norm": 0.25588348507881165,
      "learning_rate": 1.4137200743961188e-05,
      "loss": 0.8658,
      "step": 1360
    },
    {
      "epoch": 0.8410351201478743,
      "grad_norm": 0.17835567891597748,
      "learning_rate": 1.3619159821587235e-05,
      "loss": 0.8104,
      "step": 1365
    },
    {
      "epoch": 0.8441158348736907,
      "grad_norm": 0.3383285105228424,
      "learning_rate": 1.3110094181982657e-05,
      "loss": 0.8084,
      "step": 1370
    },
    {
      "epoch": 0.8471965495995071,
      "grad_norm": 0.4140583872795105,
      "learning_rate": 1.261005671891482e-05,
      "loss": 0.6785,
      "step": 1375
    },
    {
      "epoch": 0.8502772643253235,
      "grad_norm": 0.20381665229797363,
      "learning_rate": 1.2119099388090716e-05,
      "loss": 0.8666,
      "step": 1380
    },
    {
      "epoch": 0.8533579790511399,
      "grad_norm": 0.2234826385974884,
      "learning_rate": 1.1637273201758748e-05,
      "loss": 0.8612,
      "step": 1385
    },
    {
      "epoch": 0.8564386937769563,
      "grad_norm": 0.19456957280635834,
      "learning_rate": 1.1164628223408168e-05,
      "loss": 0.7868,
      "step": 1390
    },
    {
      "epoch": 0.8595194085027726,
      "grad_norm": 0.3582867383956909,
      "learning_rate": 1.0701213562567492e-05,
      "loss": 0.7432,
      "step": 1395
    },
    {
      "epoch": 0.862600123228589,
      "grad_norm": 0.5242052674293518,
      "learning_rate": 1.0247077369701653e-05,
      "loss": 0.6805,
      "step": 1400
    },
    {
      "epoch": 0.8656808379544054,
      "grad_norm": 0.2359609752893448,
      "learning_rate": 9.802266831209206e-06,
      "loss": 0.769,
      "step": 1405
    },
    {
      "epoch": 0.8687615526802218,
      "grad_norm": 0.2003209888935089,
      "learning_rate": 9.366828164519258e-06,
      "loss": 0.7757,
      "step": 1410
    },
    {
      "epoch": 0.8718422674060382,
      "grad_norm": 0.19699274003505707,
      "learning_rate": 8.940806613289498e-06,
      "loss": 0.8391,
      "step": 1415
    },
    {
      "epoch": 0.8749229821318546,
      "grad_norm": 0.3794746994972229,
      "learning_rate": 8.524246442705153e-06,
      "loss": 0.7833,
      "step": 1420
    },
    {
      "epoch": 0.878003696857671,
      "grad_norm": 0.44309327006340027,
      "learning_rate": 8.117190934879593e-06,
      "loss": 0.6823,
      "step": 1425
    },
    {
      "epoch": 0.8810844115834874,
      "grad_norm": 0.2537841498851776,
      "learning_rate": 7.719682384357308e-06,
      "loss": 0.8217,
      "step": 1430
    },
    {
      "epoch": 0.8841651263093038,
      "grad_norm": 0.24696236848831177,
      "learning_rate": 7.33176209371923e-06,
      "loss": 0.8548,
      "step": 1435
    },
    {
      "epoch": 0.8872458410351202,
      "grad_norm": 0.21302761137485504,
      "learning_rate": 6.953470369291348e-06,
      "loss": 0.8104,
      "step": 1440
    },
    {
      "epoch": 0.8903265557609366,
      "grad_norm": 0.34981197118759155,
      "learning_rate": 6.5848465169566e-06,
      "loss": 0.8252,
      "step": 1445
    },
    {
      "epoch": 0.893407270486753,
      "grad_norm": 0.3871757686138153,
      "learning_rate": 6.225928838071016e-06,
      "loss": 0.6776,
      "step": 1450
    },
    {
      "epoch": 0.8964879852125693,
      "grad_norm": 0.2748248875141144,
      "learning_rate": 5.876754625483904e-06,
      "loss": 0.7302,
      "step": 1455
    },
    {
      "epoch": 0.8995686999383857,
      "grad_norm": 0.2139190435409546,
      "learning_rate": 5.537360159663108e-06,
      "loss": 0.8257,
      "step": 1460
    },
    {
      "epoch": 0.9026494146642021,
      "grad_norm": 0.20311224460601807,
      "learning_rate": 5.207780704925314e-06,
      "loss": 0.7711,
      "step": 1465
    },
    {
      "epoch": 0.9057301293900185,
      "grad_norm": 0.3428252041339874,
      "learning_rate": 4.888050505771868e-06,
      "loss": 0.7913,
      "step": 1470
    },
    {
      "epoch": 0.9088108441158349,
      "grad_norm": 0.3909055292606354,
      "learning_rate": 4.578202783330799e-06,
      "loss": 0.6941,
      "step": 1475
    },
    {
      "epoch": 0.9118915588416513,
      "grad_norm": 0.24520839750766754,
      "learning_rate": 4.2782697319048605e-06,
      "loss": 0.8022,
      "step": 1480
    },
    {
      "epoch": 0.9149722735674677,
      "grad_norm": 0.22214862704277039,
      "learning_rate": 3.988282515626585e-06,
      "loss": 0.7689,
      "step": 1485
    },
    {
      "epoch": 0.9180529882932841,
      "grad_norm": 0.19475002586841583,
      "learning_rate": 3.7082712652200867e-06,
      "loss": 0.8022,
      "step": 1490
    },
    {
      "epoch": 0.9211337030191005,
      "grad_norm": 0.33562320470809937,
      "learning_rate": 3.438265074870417e-06,
      "loss": 0.8355,
      "step": 1495
    },
    {
      "epoch": 0.9242144177449169,
      "grad_norm": 0.40620696544647217,
      "learning_rate": 3.1782919992006333e-06,
      "loss": 0.6802,
      "step": 1500
    },
    {
      "epoch": 0.9272951324707333,
      "grad_norm": 0.21948112547397614,
      "learning_rate": 2.9283790503567222e-06,
      "loss": 0.8226,
      "step": 1505
    },
    {
      "epoch": 0.9303758471965496,
      "grad_norm": 0.23767168819904327,
      "learning_rate": 2.6885521952010105e-06,
      "loss": 0.8635,
      "step": 1510
    },
    {
      "epoch": 0.933456561922366,
      "grad_norm": 0.20257094502449036,
      "learning_rate": 2.458836352614069e-06,
      "loss": 0.769,
      "step": 1515
    },
    {
      "epoch": 0.9365372766481824,
      "grad_norm": 0.35244113206863403,
      "learning_rate": 2.239255390905581e-06,
      "loss": 0.7843,
      "step": 1520
    },
    {
      "epoch": 0.9396179913739988,
      "grad_norm": 0.4661599099636078,
      "learning_rate": 2.029832125334319e-06,
      "loss": 0.6962,
      "step": 1525
    },
    {
      "epoch": 0.9426987060998152,
      "grad_norm": 0.22016648948192596,
      "learning_rate": 1.8305883157375804e-06,
      "loss": 0.7648,
      "step": 1530
    },
    {
      "epoch": 0.9457794208256316,
      "grad_norm": 0.22861087322235107,
      "learning_rate": 1.6415446642702337e-06,
      "loss": 0.9004,
      "step": 1535
    },
    {
      "epoch": 0.9488601355514479,
      "grad_norm": 0.20447802543640137,
      "learning_rate": 1.462720813253682e-06,
      "loss": 0.8064,
      "step": 1540
    },
    {
      "epoch": 0.9519408502772643,
      "grad_norm": 0.36860471963882446,
      "learning_rate": 1.2941353431350056e-06,
      "loss": 0.7823,
      "step": 1545
    },
    {
      "epoch": 0.9550215650030807,
      "grad_norm": 0.42686179280281067,
      "learning_rate": 1.135805770556364e-06,
      "loss": 0.7038,
      "step": 1550
    },
    {
      "epoch": 0.958102279728897,
      "grad_norm": 0.2485632747411728,
      "learning_rate": 9.877485465349058e-07,
      "loss": 0.8623,
      "step": 1555
    },
    {
      "epoch": 0.9611829944547134,
      "grad_norm": 0.2371508926153183,
      "learning_rate": 8.499790547535025e-07,
      "loss": 0.803,
      "step": 1560
    },
    {
      "epoch": 0.9642637091805298,
      "grad_norm": 0.17529284954071045,
      "learning_rate": 7.225116099623286e-07,
      "loss": 0.7638,
      "step": 1565
    },
    {
      "epoch": 0.9673444239063462,
      "grad_norm": 0.31856223940849304,
      "learning_rate": 6.053594564914611e-07,
      "loss": 0.8051,
      "step": 1570
    },
    {
      "epoch": 0.9704251386321626,
      "grad_norm": 0.41933473944664,
      "learning_rate": 4.985347668747809e-07,
      "loss": 0.6794,
      "step": 1575
    },
    {
      "epoch": 0.973505853357979,
      "grad_norm": 0.23226027190685272,
      "learning_rate": 4.0204864058522864e-07,
      "loss": 0.8806,
      "step": 1580
    },
    {
      "epoch": 0.9765865680837954,
      "grad_norm": 0.1822706013917923,
      "learning_rate": 3.15911102881461e-07,
      "loss": 0.9161,
      "step": 1585
    },
    {
      "epoch": 0.9796672828096118,
      "grad_norm": 0.20591747760772705,
      "learning_rate": 2.40131103766239e-07,
      "loss": 0.7944,
      "step": 1590
    },
    {
      "epoch": 0.9827479975354282,
      "grad_norm": 0.3590210974216461,
      "learning_rate": 1.747165170564724e-07,
      "loss": 0.8582,
      "step": 1595
    },
    {
      "epoch": 0.9858287122612446,
      "grad_norm": 0.4347827732563019,
      "learning_rate": 1.1967413956510686e-07,
      "loss": 0.7153,
      "step": 1600
    },
    {
      "epoch": 0.988909426987061,
      "grad_norm": 0.29511308670043945,
      "learning_rate": 7.500969039491157e-08,
      "loss": 0.7962,
      "step": 1605
    },
    {
      "epoch": 0.9919901417128774,
      "grad_norm": 0.27846843004226685,
      "learning_rate": 4.0727810344254325e-08,
      "loss": 0.8787,
      "step": 1610
    },
    {
      "epoch": 0.9950708564386938,
      "grad_norm": 0.18349671363830566,
      "learning_rate": 1.6832061424865153e-08,
      "loss": 0.784,
      "step": 1615
    },
    {
      "epoch": 0.9981515711645101,
      "grad_norm": 0.2996051609516144,
      "learning_rate": 3.3249264917878387e-09,
      "loss": 0.7315,
      "step": 1620
    },
    {
      "epoch": 1.0,
      "step": 1623,
      "total_flos": 889480391426048.0,
      "train_loss": 0.8250481778284098,
      "train_runtime": 5390.1779,
      "train_samples_per_second": 9.634,
      "train_steps_per_second": 0.301
    }
  ],
  "logging_steps": 5,
  "max_steps": 1623,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 889480391426048.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}