{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.7312614259597806,
  "eval_steps": 500,
  "global_step": 400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0018281535648994515,
      "grad_norm": 2.089374507496642,
      "learning_rate": 0.0,
      "loss": 0.5957,
      "step": 1
    },
    {
      "epoch": 0.003656307129798903,
      "grad_norm": 2.2205513668741044,
      "learning_rate": 1.0714285714285714e-06,
      "loss": 0.5994,
      "step": 2
    },
    {
      "epoch": 0.005484460694698354,
      "grad_norm": 2.545945852200423,
      "learning_rate": 2.1428571428571427e-06,
      "loss": 0.5948,
      "step": 3
    },
    {
      "epoch": 0.007312614259597806,
      "grad_norm": 2.2330035148383947,
      "learning_rate": 3.2142857142857143e-06,
      "loss": 0.6305,
      "step": 4
    },
    {
      "epoch": 0.009140767824497258,
      "grad_norm": 1.8952582634630641,
      "learning_rate": 4.2857142857142855e-06,
      "loss": 0.5998,
      "step": 5
    },
    {
      "epoch": 0.010968921389396709,
      "grad_norm": 1.570960207475644,
      "learning_rate": 5.357142857142857e-06,
      "loss": 0.5898,
      "step": 6
    },
    {
      "epoch": 0.012797074954296161,
      "grad_norm": 1.2522662311648174,
      "learning_rate": 6.428571428571429e-06,
      "loss": 0.6013,
      "step": 7
    },
    {
      "epoch": 0.014625228519195612,
      "grad_norm": 1.065080174164971,
      "learning_rate": 7.5e-06,
      "loss": 0.5682,
      "step": 8
    },
    {
      "epoch": 0.016453382084095063,
      "grad_norm": 0.9448479764184926,
      "learning_rate": 8.571428571428571e-06,
      "loss": 0.5452,
      "step": 9
    },
    {
      "epoch": 0.018281535648994516,
      "grad_norm": 1.7141233153619826,
      "learning_rate": 9.642857142857144e-06,
      "loss": 0.59,
      "step": 10
    },
    {
      "epoch": 0.02010968921389397,
      "grad_norm": 2.017633207813829,
      "learning_rate": 1.0714285714285714e-05,
      "loss": 0.575,
      "step": 11
    },
    {
      "epoch": 0.021937842778793418,
      "grad_norm": 1.8293000392166587,
      "learning_rate": 1.1785714285714286e-05,
      "loss": 0.5598,
      "step": 12
    },
    {
      "epoch": 0.02376599634369287,
      "grad_norm": 1.4816869854216612,
      "learning_rate": 1.2857142857142857e-05,
      "loss": 0.5495,
      "step": 13
    },
    {
      "epoch": 0.025594149908592323,
      "grad_norm": 1.5033597863132906,
      "learning_rate": 1.3928571428571429e-05,
      "loss": 0.5372,
      "step": 14
    },
    {
      "epoch": 0.027422303473491772,
      "grad_norm": 1.3141514496532496,
      "learning_rate": 1.5e-05,
      "loss": 0.5733,
      "step": 15
    },
    {
      "epoch": 0.029250457038391225,
      "grad_norm": 2.7794219916654233,
      "learning_rate": 1.6071428571428572e-05,
      "loss": 0.5347,
      "step": 16
    },
    {
      "epoch": 0.031078610603290677,
      "grad_norm": 0.9928713947921902,
      "learning_rate": 1.7142857142857142e-05,
      "loss": 0.4908,
      "step": 17
    },
    {
      "epoch": 0.03290676416819013,
      "grad_norm": 0.8852252619423812,
      "learning_rate": 1.8214285714285712e-05,
      "loss": 0.5307,
      "step": 18
    },
    {
      "epoch": 0.03473491773308958,
      "grad_norm": 0.7832318403464806,
      "learning_rate": 1.928571428571429e-05,
      "loss": 0.5025,
      "step": 19
    },
    {
      "epoch": 0.03656307129798903,
      "grad_norm": 0.9796265282565124,
      "learning_rate": 2.0357142857142858e-05,
      "loss": 0.5095,
      "step": 20
    },
    {
      "epoch": 0.038391224862888484,
      "grad_norm": 0.7218139820603727,
      "learning_rate": 2.1428571428571428e-05,
      "loss": 0.4834,
      "step": 21
    },
    {
      "epoch": 0.04021937842778794,
      "grad_norm": 0.6506357909260763,
      "learning_rate": 2.25e-05,
      "loss": 0.4947,
      "step": 22
    },
    {
      "epoch": 0.04204753199268738,
      "grad_norm": 0.5616806424289195,
      "learning_rate": 2.357142857142857e-05,
      "loss": 0.4866,
      "step": 23
    },
    {
      "epoch": 0.043875685557586835,
      "grad_norm": 0.566744323961197,
      "learning_rate": 2.464285714285714e-05,
      "loss": 0.515,
      "step": 24
    },
    {
      "epoch": 0.04570383912248629,
      "grad_norm": 0.6913117297678986,
      "learning_rate": 2.5714285714285714e-05,
      "loss": 0.5227,
      "step": 25
    },
    {
      "epoch": 0.04753199268738574,
      "grad_norm": 0.5881117480751623,
      "learning_rate": 2.6785714285714288e-05,
      "loss": 0.478,
      "step": 26
    },
    {
      "epoch": 0.04936014625228519,
      "grad_norm": 0.5399101198373484,
      "learning_rate": 2.7857142857142858e-05,
      "loss": 0.5039,
      "step": 27
    },
    {
      "epoch": 0.051188299817184646,
      "grad_norm": 0.5359845255880198,
      "learning_rate": 2.892857142857143e-05,
      "loss": 0.5015,
      "step": 28
    },
    {
      "epoch": 0.05301645338208409,
      "grad_norm": 0.4718417677524385,
      "learning_rate": 3e-05,
      "loss": 0.4738,
      "step": 29
    },
    {
      "epoch": 0.054844606946983544,
      "grad_norm": 0.47359316021290165,
      "learning_rate": 2.999972519478696e-05,
      "loss": 0.4993,
      "step": 30
    },
    {
      "epoch": 0.056672760511883,
      "grad_norm": 0.4922109148988566,
      "learning_rate": 2.99989007892169e-05,
      "loss": 0.4888,
      "step": 31
    },
    {
      "epoch": 0.05850091407678245,
      "grad_norm": 0.3579397510509835,
      "learning_rate": 2.9997526813496602e-05,
      "loss": 0.4792,
      "step": 32
    },
    {
      "epoch": 0.0603290676416819,
      "grad_norm": 0.37626393648031903,
      "learning_rate": 2.9995603317969497e-05,
      "loss": 0.4873,
      "step": 33
    },
    {
      "epoch": 0.062157221206581355,
      "grad_norm": 0.41211122381414894,
      "learning_rate": 2.99931303731138e-05,
      "loss": 0.4939,
      "step": 34
    },
    {
      "epoch": 0.06398537477148081,
      "grad_norm": 0.3770284364229165,
      "learning_rate": 2.9990108069539932e-05,
      "loss": 0.479,
      "step": 35
    },
    {
      "epoch": 0.06581352833638025,
      "grad_norm": 0.3709162789872314,
      "learning_rate": 2.9986536517987187e-05,
      "loss": 0.4864,
      "step": 36
    },
    {
      "epoch": 0.06764168190127971,
      "grad_norm": 0.36989388190185846,
      "learning_rate": 2.998241584931971e-05,
      "loss": 0.4727,
      "step": 37
    },
    {
      "epoch": 0.06946983546617916,
      "grad_norm": 0.36339103914080506,
      "learning_rate": 2.9977746214521646e-05,
      "loss": 0.472,
      "step": 38
    },
    {
      "epoch": 0.0712979890310786,
      "grad_norm": 0.3378220732996809,
      "learning_rate": 2.997252778469168e-05,
      "loss": 0.4581,
      "step": 39
    },
    {
      "epoch": 0.07312614259597806,
      "grad_norm": 0.33242480046792755,
      "learning_rate": 2.9966760751036697e-05,
      "loss": 0.4567,
      "step": 40
    },
    {
      "epoch": 0.07495429616087751,
      "grad_norm": 0.3517734032444255,
      "learning_rate": 2.9960445324864815e-05,
      "loss": 0.4674,
      "step": 41
    },
    {
      "epoch": 0.07678244972577697,
      "grad_norm": 0.3558065174543996,
      "learning_rate": 2.995358173757765e-05,
      "loss": 0.4718,
      "step": 42
    },
    {
      "epoch": 0.07861060329067641,
      "grad_norm": 0.34290523065110684,
      "learning_rate": 2.994617024066181e-05,
      "loss": 0.4659,
      "step": 43
    },
    {
      "epoch": 0.08043875685557587,
      "grad_norm": 0.3028572716833668,
      "learning_rate": 2.9938211105679677e-05,
      "loss": 0.4543,
      "step": 44
    },
    {
      "epoch": 0.08226691042047532,
      "grad_norm": 0.36216688731700014,
      "learning_rate": 2.9929704624259508e-05,
      "loss": 0.4724,
      "step": 45
    },
    {
      "epoch": 0.08409506398537477,
      "grad_norm": 0.34546125994172217,
      "learning_rate": 2.992065110808469e-05,
      "loss": 0.4717,
      "step": 46
    },
    {
      "epoch": 0.08592321755027423,
      "grad_norm": 0.3173199655699041,
      "learning_rate": 2.991105088888234e-05,
      "loss": 0.4893,
      "step": 47
    },
    {
      "epoch": 0.08775137111517367,
      "grad_norm": 0.27327168658587364,
      "learning_rate": 2.990090431841117e-05,
      "loss": 0.4877,
      "step": 48
    },
    {
      "epoch": 0.08957952468007313,
      "grad_norm": 0.3207165383004196,
      "learning_rate": 2.9890211768448572e-05,
      "loss": 0.4666,
      "step": 49
    },
    {
      "epoch": 0.09140767824497258,
      "grad_norm": 0.29795651824929764,
      "learning_rate": 2.9878973630777012e-05,
      "loss": 0.4764,
      "step": 50
    },
    {
      "epoch": 0.09323583180987204,
      "grad_norm": 0.3066207186556079,
      "learning_rate": 2.9867190317169665e-05,
      "loss": 0.4793,
      "step": 51
    },
    {
      "epoch": 0.09506398537477148,
      "grad_norm": 0.30152926589089146,
      "learning_rate": 2.9854862259375326e-05,
      "loss": 0.48,
      "step": 52
    },
    {
      "epoch": 0.09689213893967093,
      "grad_norm": 0.31982845164198154,
      "learning_rate": 2.9841989909102607e-05,
      "loss": 0.4827,
      "step": 53
    },
    {
      "epoch": 0.09872029250457039,
      "grad_norm": 0.3226249549846725,
      "learning_rate": 2.982857373800337e-05,
      "loss": 0.4815,
      "step": 54
    },
    {
      "epoch": 0.10054844606946983,
      "grad_norm": 0.29313323454641665,
      "learning_rate": 2.9814614237655445e-05,
      "loss": 0.4624,
      "step": 55
    },
    {
      "epoch": 0.10237659963436929,
      "grad_norm": 0.30450893112709426,
      "learning_rate": 2.9800111919544632e-05,
      "loss": 0.4856,
      "step": 56
    },
    {
      "epoch": 0.10420475319926874,
      "grad_norm": 0.31077892246708605,
      "learning_rate": 2.9785067315045943e-05,
      "loss": 0.4801,
      "step": 57
    },
    {
      "epoch": 0.10603290676416818,
      "grad_norm": 0.30352760576382176,
      "learning_rate": 2.9769480975404143e-05,
      "loss": 0.4462,
      "step": 58
    },
    {
      "epoch": 0.10786106032906764,
      "grad_norm": 0.2820427097517702,
      "learning_rate": 2.975335347171356e-05,
      "loss": 0.4275,
      "step": 59
    },
    {
      "epoch": 0.10968921389396709,
      "grad_norm": 0.3193688120609219,
      "learning_rate": 2.9736685394897123e-05,
      "loss": 0.4585,
      "step": 60
    },
    {
      "epoch": 0.11151736745886655,
      "grad_norm": 0.29309453499656224,
      "learning_rate": 2.9719477355684767e-05,
      "loss": 0.4563,
      "step": 61
    },
    {
      "epoch": 0.113345521023766,
      "grad_norm": 0.291897548964064,
      "learning_rate": 2.9701729984591003e-05,
      "loss": 0.4402,
      "step": 62
    },
    {
      "epoch": 0.11517367458866545,
      "grad_norm": 0.3386425303422288,
      "learning_rate": 2.9683443931891837e-05,
      "loss": 0.4586,
      "step": 63
    },
    {
      "epoch": 0.1170018281535649,
      "grad_norm": 0.28801030329886385,
      "learning_rate": 2.966461986760096e-05,
      "loss": 0.4452,
      "step": 64
    },
    {
      "epoch": 0.11882998171846434,
      "grad_norm": 0.27225571075726074,
      "learning_rate": 2.964525848144517e-05,
      "loss": 0.4219,
      "step": 65
    },
    {
      "epoch": 0.1206581352833638,
      "grad_norm": 0.3150014322578861,
      "learning_rate": 2.9625360482839114e-05,
      "loss": 0.4833,
      "step": 66
    },
    {
      "epoch": 0.12248628884826325,
      "grad_norm": 0.28747617110884083,
      "learning_rate": 2.9604926600859287e-05,
      "loss": 0.4673,
      "step": 67
    },
    {
      "epoch": 0.12431444241316271,
      "grad_norm": 0.2935455145106659,
      "learning_rate": 2.958395758421733e-05,
      "loss": 0.4492,
      "step": 68
    },
    {
      "epoch": 0.12614259597806216,
      "grad_norm": 0.2748773662739781,
      "learning_rate": 2.9562454201232583e-05,
      "loss": 0.4763,
      "step": 69
    },
    {
      "epoch": 0.12797074954296161,
      "grad_norm": 0.3040072222347735,
      "learning_rate": 2.9540417239803954e-05,
      "loss": 0.4658,
      "step": 70
    },
    {
      "epoch": 0.12979890310786105,
      "grad_norm": 0.3401190893101851,
      "learning_rate": 2.9517847507381022e-05,
      "loss": 0.4619,
      "step": 71
    },
    {
      "epoch": 0.1316270566727605,
      "grad_norm": 0.2832439537823879,
      "learning_rate": 2.9494745830934473e-05,
      "loss": 0.4594,
      "step": 72
    },
    {
      "epoch": 0.13345521023765997,
      "grad_norm": 0.4216765622686639,
      "learning_rate": 2.9471113056925786e-05,
      "loss": 0.4939,
      "step": 73
    },
    {
      "epoch": 0.13528336380255943,
      "grad_norm": 0.2884581020061622,
      "learning_rate": 2.944695005127623e-05,
      "loss": 0.458,
      "step": 74
    },
    {
      "epoch": 0.13711151736745886,
      "grad_norm": 0.28039598694507106,
      "learning_rate": 2.942225769933512e-05,
      "loss": 0.4361,
      "step": 75
    },
    {
      "epoch": 0.13893967093235832,
      "grad_norm": 0.27724633932247766,
      "learning_rate": 2.939703690584741e-05,
      "loss": 0.4557,
      "step": 76
    },
    {
      "epoch": 0.14076782449725778,
      "grad_norm": 0.2694165757512015,
      "learning_rate": 2.9371288594920484e-05,
      "loss": 0.4332,
      "step": 77
    },
    {
      "epoch": 0.1425959780621572,
      "grad_norm": 0.25780899472474833,
      "learning_rate": 2.934501370999037e-05,
      "loss": 0.4242,
      "step": 78
    },
    {
      "epoch": 0.14442413162705667,
      "grad_norm": 0.2967403245642137,
      "learning_rate": 2.93182132137871e-05,
      "loss": 0.446,
      "step": 79
    },
    {
      "epoch": 0.14625228519195613,
      "grad_norm": 0.34690556417034824,
      "learning_rate": 2.9290888088299486e-05,
      "loss": 0.4818,
      "step": 80
    },
    {
      "epoch": 0.1480804387568556,
      "grad_norm": 0.3003683124367745,
      "learning_rate": 2.9263039334739127e-05,
      "loss": 0.462,
      "step": 81
    },
    {
      "epoch": 0.14990859232175502,
      "grad_norm": 0.302108544256545,
      "learning_rate": 2.92346679735037e-05,
      "loss": 0.4561,
      "step": 82
    },
    {
      "epoch": 0.15173674588665448,
      "grad_norm": 0.2841043066219529,
      "learning_rate": 2.9205775044139608e-05,
      "loss": 0.4684,
      "step": 83
    },
    {
      "epoch": 0.15356489945155394,
      "grad_norm": 0.2952668737797339,
      "learning_rate": 2.9176361605303867e-05,
      "loss": 0.4693,
      "step": 84
    },
    {
      "epoch": 0.15539305301645337,
      "grad_norm": 0.28847383193289927,
      "learning_rate": 2.914642873472531e-05,
      "loss": 0.452,
      "step": 85
    },
    {
      "epoch": 0.15722120658135283,
      "grad_norm": 0.31050885319522226,
      "learning_rate": 2.9115977529165132e-05,
      "loss": 0.4621,
      "step": 86
    },
    {
      "epoch": 0.1590493601462523,
      "grad_norm": 0.3143842313652269,
      "learning_rate": 2.9085009104376663e-05,
      "loss": 0.4563,
      "step": 87
    },
    {
      "epoch": 0.16087751371115175,
      "grad_norm": 0.33281281256892775,
      "learning_rate": 2.905352459506452e-05,
      "loss": 0.437,
      "step": 88
    },
    {
      "epoch": 0.16270566727605118,
      "grad_norm": 0.30726176715777354,
      "learning_rate": 2.9021525154842998e-05,
      "loss": 0.4614,
      "step": 89
    },
    {
      "epoch": 0.16453382084095064,
      "grad_norm": 0.31519069611911227,
      "learning_rate": 2.8989011956193834e-05,
      "loss": 0.4584,
      "step": 90
    },
    {
      "epoch": 0.1663619744058501,
      "grad_norm": 0.3142796251849267,
      "learning_rate": 2.8955986190423225e-05,
      "loss": 0.4848,
      "step": 91
    },
    {
      "epoch": 0.16819012797074953,
      "grad_norm": 0.2831585707772231,
      "learning_rate": 2.892244906761819e-05,
      "loss": 0.4463,
      "step": 92
    },
    {
      "epoch": 0.170018281535649,
      "grad_norm": 0.3271624604307319,
      "learning_rate": 2.8888401816602207e-05,
      "loss": 0.4496,
      "step": 93
    },
    {
      "epoch": 0.17184643510054845,
      "grad_norm": 0.3265076622565222,
      "learning_rate": 2.885384568489023e-05,
      "loss": 0.434,
      "step": 94
    },
    {
      "epoch": 0.1736745886654479,
      "grad_norm": 0.3384143662294581,
      "learning_rate": 2.881878193864294e-05,
      "loss": 0.4662,
      "step": 95
    },
    {
      "epoch": 0.17550274223034734,
      "grad_norm": 0.29519544921435537,
      "learning_rate": 2.878321186262037e-05,
      "loss": 0.4415,
      "step": 96
    },
    {
      "epoch": 0.1773308957952468,
      "grad_norm": 0.3152573413188835,
      "learning_rate": 2.8747136760134827e-05,
      "loss": 0.4525,
      "step": 97
    },
    {
      "epoch": 0.17915904936014626,
      "grad_norm": 0.2911315525075031,
      "learning_rate": 2.871055795300315e-05,
      "loss": 0.4393,
      "step": 98
    },
    {
      "epoch": 0.1809872029250457,
      "grad_norm": 0.28890139736154113,
      "learning_rate": 2.8673476781498242e-05,
      "loss": 0.4301,
      "step": 99
    },
    {
      "epoch": 0.18281535648994515,
      "grad_norm": 0.29566233008031895,
      "learning_rate": 2.8635894604300018e-05,
      "loss": 0.4477,
      "step": 100
    },
    {
      "epoch": 0.1846435100548446,
      "grad_norm": 0.2759848384654238,
      "learning_rate": 2.859781279844556e-05,
      "loss": 0.4477,
      "step": 101
    },
    {
      "epoch": 0.18647166361974407,
      "grad_norm": 0.3231166719493857,
      "learning_rate": 2.855923275927871e-05,
      "loss": 0.4732,
      "step": 102
    },
    {
      "epoch": 0.1882998171846435,
      "grad_norm": 0.27671665616727253,
      "learning_rate": 2.8520155900398922e-05,
      "loss": 0.4348,
      "step": 103
    },
    {
      "epoch": 0.19012797074954296,
      "grad_norm": 0.3373290606091851,
      "learning_rate": 2.8480583653609457e-05,
      "loss": 0.4772,
      "step": 104
    },
    {
      "epoch": 0.19195612431444242,
      "grad_norm": 0.270804966443481,
      "learning_rate": 2.844051746886495e-05,
      "loss": 0.4522,
      "step": 105
    },
    {
      "epoch": 0.19378427787934185,
      "grad_norm": 0.28232177001780473,
      "learning_rate": 2.8399958814218258e-05,
      "loss": 0.465,
      "step": 106
    },
    {
      "epoch": 0.1956124314442413,
      "grad_norm": 0.32735140894867865,
      "learning_rate": 2.8358909175766674e-05,
      "loss": 0.4665,
      "step": 107
    },
    {
      "epoch": 0.19744058500914077,
      "grad_norm": 0.3535841008181332,
      "learning_rate": 2.831737005759749e-05,
      "loss": 0.4817,
      "step": 108
    },
    {
      "epoch": 0.19926873857404023,
      "grad_norm": 0.31153302714419157,
      "learning_rate": 2.8275342981732868e-05,
      "loss": 0.4774,
      "step": 109
    },
    {
      "epoch": 0.20109689213893966,
      "grad_norm": 0.3414165680532852,
      "learning_rate": 2.8232829488074077e-05,
      "loss": 0.4354,
      "step": 110
    },
    {
      "epoch": 0.20292504570383912,
      "grad_norm": 0.2657136503074252,
      "learning_rate": 2.8189831134345074e-05,
      "loss": 0.4403,
      "step": 111
    },
    {
      "epoch": 0.20475319926873858,
      "grad_norm": 0.2963856605986798,
      "learning_rate": 2.8146349496035426e-05,
      "loss": 0.4262,
      "step": 112
    },
    {
      "epoch": 0.20658135283363802,
      "grad_norm": 0.3119283296440065,
      "learning_rate": 2.8102386166342582e-05,
      "loss": 0.4418,
      "step": 113
    },
    {
      "epoch": 0.20840950639853748,
      "grad_norm": 0.37109957781553604,
      "learning_rate": 2.8057942756113504e-05,
      "loss": 0.4765,
      "step": 114
    },
    {
      "epoch": 0.21023765996343693,
      "grad_norm": 0.2999662023442714,
      "learning_rate": 2.8013020893785635e-05,
      "loss": 0.4539,
      "step": 115
    },
    {
      "epoch": 0.21206581352833637,
      "grad_norm": 0.28970493967104405,
      "learning_rate": 2.796762222532723e-05,
      "loss": 0.4289,
      "step": 116
    },
    {
      "epoch": 0.21389396709323583,
      "grad_norm": 0.2939203180452282,
      "learning_rate": 2.7921748414177063e-05,
      "loss": 0.4689,
      "step": 117
    },
    {
      "epoch": 0.21572212065813529,
      "grad_norm": 0.3201929927655123,
      "learning_rate": 2.787540114118345e-05,
      "loss": 0.4338,
      "step": 118
    },
    {
      "epoch": 0.21755027422303475,
      "grad_norm": 0.2864175962342283,
      "learning_rate": 2.7828582104542694e-05,
      "loss": 0.4464,
      "step": 119
    },
    {
      "epoch": 0.21937842778793418,
      "grad_norm": 0.28592410203121005,
      "learning_rate": 2.7781293019736845e-05,
      "loss": 0.4397,
      "step": 120
    },
    {
      "epoch": 0.22120658135283364,
      "grad_norm": 0.3304180569852307,
      "learning_rate": 2.7733535619470835e-05,
      "loss": 0.4481,
      "step": 121
    },
    {
      "epoch": 0.2230347349177331,
      "grad_norm": 0.3390393526626585,
      "learning_rate": 2.7685311653609004e-05,
      "loss": 0.4432,
      "step": 122
    },
    {
      "epoch": 0.22486288848263253,
      "grad_norm": 0.2940159306891994,
      "learning_rate": 2.7636622889110975e-05,
      "loss": 0.475,
      "step": 123
    },
    {
      "epoch": 0.226691042047532,
      "grad_norm": 0.31229638977483115,
      "learning_rate": 2.758747110996693e-05,
      "loss": 0.4643,
      "step": 124
    },
    {
      "epoch": 0.22851919561243145,
      "grad_norm": 0.285967777532962,
      "learning_rate": 2.7537858117132217e-05,
      "loss": 0.4505,
      "step": 125
    },
    {
      "epoch": 0.2303473491773309,
      "grad_norm": 0.3096207093074659,
      "learning_rate": 2.7487785728461383e-05,
      "loss": 0.4606,
      "step": 126
    },
    {
      "epoch": 0.23217550274223034,
      "grad_norm": 0.3196007759195916,
      "learning_rate": 2.7437255778641548e-05,
      "loss": 0.4688,
      "step": 127
    },
    {
      "epoch": 0.2340036563071298,
      "grad_norm": 0.278093004843679,
      "learning_rate": 2.7386270119125193e-05,
      "loss": 0.4476,
      "step": 128
    },
    {
      "epoch": 0.23583180987202926,
      "grad_norm": 0.28238850493053447,
      "learning_rate": 2.7334830618062327e-05,
      "loss": 0.4328,
      "step": 129
    },
    {
      "epoch": 0.2376599634369287,
      "grad_norm": 0.31893781223089274,
      "learning_rate": 2.728293916023202e-05,
      "loss": 0.4368,
      "step": 130
    },
    {
      "epoch": 0.23948811700182815,
      "grad_norm": 0.3226432084535994,
      "learning_rate": 2.7230597646973355e-05,
      "loss": 0.4693,
      "step": 131
    },
    {
      "epoch": 0.2413162705667276,
      "grad_norm": 0.3098641582546802,
      "learning_rate": 2.717780799611576e-05,
      "loss": 0.4677,
      "step": 132
    },
    {
      "epoch": 0.24314442413162707,
      "grad_norm": 0.316005877064317,
      "learning_rate": 2.7124572141908737e-05,
      "loss": 0.4641,
      "step": 133
    },
    {
      "epoch": 0.2449725776965265,
      "grad_norm": 0.29116768124986503,
      "learning_rate": 2.707089203495098e-05,
      "loss": 0.4504,
      "step": 134
    },
    {
      "epoch": 0.24680073126142596,
      "grad_norm": 0.3312565101776482,
      "learning_rate": 2.701676964211893e-05,
      "loss": 0.4869,
      "step": 135
    },
    {
      "epoch": 0.24862888482632542,
      "grad_norm": 0.3307114085092842,
      "learning_rate": 2.696220694649467e-05,
      "loss": 0.4307,
      "step": 136
    },
    {
      "epoch": 0.25045703839122485,
      "grad_norm": 0.3139451066311321,
      "learning_rate": 2.69072059472933e-05,
      "loss": 0.4593,
      "step": 137
    },
    {
      "epoch": 0.2522851919561243,
      "grad_norm": 0.3298447274769771,
      "learning_rate": 2.685176865978965e-05,
      "loss": 0.4544,
      "step": 138
    },
    {
      "epoch": 0.25411334552102377,
      "grad_norm": 0.3032493627313141,
      "learning_rate": 2.6795897115244478e-05,
      "loss": 0.4279,
      "step": 139
    },
    {
      "epoch": 0.25594149908592323,
      "grad_norm": 0.2658733658787214,
      "learning_rate": 2.6739593360830006e-05,
      "loss": 0.4106,
      "step": 140
    },
    {
      "epoch": 0.2577696526508227,
      "grad_norm": 0.3190884630430669,
      "learning_rate": 2.668285945955493e-05,
      "loss": 0.4662,
      "step": 141
    },
    {
      "epoch": 0.2595978062157221,
      "grad_norm": 0.34411371393720197,
      "learning_rate": 2.6625697490188832e-05,
      "loss": 0.4499,
      "step": 142
    },
    {
      "epoch": 0.26142595978062155,
      "grad_norm": 0.32823365391601855,
      "learning_rate": 2.6568109547185996e-05,
      "loss": 0.4783,
      "step": 143
    },
    {
      "epoch": 0.263254113345521,
      "grad_norm": 0.30268317080382134,
      "learning_rate": 2.651009774060868e-05,
      "loss": 0.4677,
      "step": 144
    },
    {
      "epoch": 0.26508226691042047,
      "grad_norm": 0.290999041829207,
      "learning_rate": 2.6451664196049802e-05,
      "loss": 0.4461,
      "step": 145
    },
    {
      "epoch": 0.26691042047531993,
      "grad_norm": 0.3169422559498593,
      "learning_rate": 2.639281105455505e-05,
      "loss": 0.4503,
      "step": 146
    },
    {
      "epoch": 0.2687385740402194,
      "grad_norm": 0.3189364221479283,
      "learning_rate": 2.6333540472544442e-05,
      "loss": 0.4335,
      "step": 147
    },
    {
      "epoch": 0.27056672760511885,
      "grad_norm": 0.30758294437768796,
      "learning_rate": 2.6273854621733286e-05,
      "loss": 0.4504,
      "step": 148
    },
    {
      "epoch": 0.27239488117001825,
      "grad_norm": 0.2849217195325153,
      "learning_rate": 2.621375568905266e-05,
      "loss": 0.4487,
      "step": 149
    },
    {
      "epoch": 0.2742230347349177,
      "grad_norm": 0.344396845889534,
      "learning_rate": 2.615324587656921e-05,
      "loss": 0.4821,
      "step": 150
    },
    {
      "epoch": 0.2760511882998172,
      "grad_norm": 0.2696584787135129,
      "learning_rate": 2.6092327401404538e-05,
      "loss": 0.475,
      "step": 151
    },
    {
      "epoch": 0.27787934186471663,
      "grad_norm": 0.30241018017889437,
      "learning_rate": 2.6031002495653913e-05,
      "loss": 0.4342,
      "step": 152
    },
    {
      "epoch": 0.2797074954296161,
      "grad_norm": 0.28693150625713587,
      "learning_rate": 2.596927340630451e-05,
      "loss": 0.4359,
      "step": 153
    },
    {
      "epoch": 0.28153564899451555,
      "grad_norm": 0.29754809227022666,
      "learning_rate": 2.590714239515306e-05,
      "loss": 0.4438,
      "step": 154
    },
    {
      "epoch": 0.283363802559415,
      "grad_norm": 0.299567522325594,
      "learning_rate": 2.584461173872301e-05,
      "loss": 0.4532,
      "step": 155
    },
    {
      "epoch": 0.2851919561243144,
      "grad_norm": 0.2813266176963191,
      "learning_rate": 2.5781683728181066e-05,
      "loss": 0.4405,
      "step": 156
    },
    {
      "epoch": 0.2870201096892139,
      "grad_norm": 0.3171717930616808,
      "learning_rate": 2.5718360669253276e-05,
      "loss": 0.454,
      "step": 157
    },
    {
      "epoch": 0.28884826325411334,
      "grad_norm": 0.30159353684035844,
      "learning_rate": 2.565464488214053e-05,
      "loss": 0.4375,
      "step": 158
    },
    {
      "epoch": 0.2906764168190128,
      "grad_norm": 0.2692705044539712,
      "learning_rate": 2.559053870143356e-05,
      "loss": 0.4276,
      "step": 159
    },
    {
      "epoch": 0.29250457038391225,
      "grad_norm": 0.30691595589677334,
      "learning_rate": 2.5526044476027383e-05,
      "loss": 0.4606,
      "step": 160
    },
    {
      "epoch": 0.2943327239488117,
      "grad_norm": 0.31630931258544637,
      "learning_rate": 2.5461164569035244e-05,
      "loss": 0.4406,
      "step": 161
    },
    {
      "epoch": 0.2961608775137112,
      "grad_norm": 0.2440944204924941,
      "learning_rate": 2.5395901357702032e-05,
      "loss": 0.4511,
      "step": 162
    },
    {
      "epoch": 0.2979890310786106,
      "grad_norm": 0.31355225894665767,
      "learning_rate": 2.533025723331718e-05,
      "loss": 0.4384,
      "step": 163
    },
    {
      "epoch": 0.29981718464351004,
      "grad_norm": 0.2828232991554723,
      "learning_rate": 2.526423460112703e-05,
      "loss": 0.4318,
      "step": 164
    },
    {
      "epoch": 0.3016453382084095,
      "grad_norm": 0.2872196769388632,
      "learning_rate": 2.5197835880246702e-05,
      "loss": 0.4508,
      "step": 165
    },
    {
      "epoch": 0.30347349177330896,
      "grad_norm": 0.27755470022103107,
      "learning_rate": 2.51310635035715e-05,
      "loss": 0.4443,
      "step": 166
    },
    {
      "epoch": 0.3053016453382084,
      "grad_norm": 0.29288725811791394,
      "learning_rate": 2.506391991768771e-05,
      "loss": 0.4437,
      "step": 167
    },
    {
      "epoch": 0.3071297989031079,
      "grad_norm": 0.28756847297478805,
      "learning_rate": 2.4996407582782987e-05,
      "loss": 0.4306,
      "step": 168
    },
    {
      "epoch": 0.30895795246800734,
      "grad_norm": 0.3062510638432968,
      "learning_rate": 2.4928528972556207e-05,
      "loss": 0.4401,
      "step": 169
    },
    {
      "epoch": 0.31078610603290674,
      "grad_norm": 0.28844799158530887,
      "learning_rate": 2.486028657412683e-05,
      "loss": 0.465,
      "step": 170
    },
    {
      "epoch": 0.3126142595978062,
      "grad_norm": 0.30254982711927847,
      "learning_rate": 2.479168288794377e-05,
      "loss": 0.4257,
      "step": 171
    },
    {
      "epoch": 0.31444241316270566,
      "grad_norm": 0.29617551569459377,
      "learning_rate": 2.472272042769377e-05,
      "loss": 0.4413,
      "step": 172
    },
    {
      "epoch": 0.3162705667276051,
      "grad_norm": 0.2894388824454487,
      "learning_rate": 2.4653401720209316e-05,
      "loss": 0.4339,
      "step": 173
    },
    {
      "epoch": 0.3180987202925046,
      "grad_norm": 0.31278541819511707,
      "learning_rate": 2.4583729305376014e-05,
      "loss": 0.4433,
      "step": 174
    },
    {
      "epoch": 0.31992687385740404,
      "grad_norm": 0.29889716089103313,
      "learning_rate": 2.451370573603959e-05,
      "loss": 0.4465,
      "step": 175
    },
    {
      "epoch": 0.3217550274223035,
      "grad_norm": 0.30650603689771794,
      "learning_rate": 2.4443333577912285e-05,
      "loss": 0.4497,
      "step": 176
    },
    {
      "epoch": 0.3235831809872029,
      "grad_norm": 0.27739674261767966,
      "learning_rate": 2.437261540947889e-05,
      "loss": 0.4417,
      "step": 177
    },
    {
      "epoch": 0.32541133455210236,
      "grad_norm": 0.29186196118144964,
      "learning_rate": 2.430155382190225e-05,
      "loss": 0.46,
      "step": 178
    },
    {
      "epoch": 0.3272394881170018,
      "grad_norm": 0.28342780349608687,
      "learning_rate": 2.4230151418928326e-05,
      "loss": 0.4414,
      "step": 179
    },
    {
      "epoch": 0.3290676416819013,
      "grad_norm": 0.2751850949398033,
      "learning_rate": 2.415841081679079e-05,
      "loss": 0.4569,
      "step": 180
    },
    {
      "epoch": 0.33089579524680074,
      "grad_norm": 0.2848165003785897,
      "learning_rate": 2.4086334644115176e-05,
      "loss": 0.4167,
      "step": 181
    },
    {
      "epoch": 0.3327239488117002,
      "grad_norm": 0.26869428432353776,
      "learning_rate": 2.4013925541822543e-05,
      "loss": 0.4393,
      "step": 182
    },
    {
      "epoch": 0.33455210237659966,
      "grad_norm": 0.3041074292613331,
      "learning_rate": 2.3941186163032736e-05,
      "loss": 0.443,
      "step": 183
    },
    {
      "epoch": 0.33638025594149906,
      "grad_norm": 0.3054719627730434,
      "learning_rate": 2.3868119172967145e-05,
      "loss": 0.4711,
      "step": 184
    },
    {
      "epoch": 0.3382084095063985,
      "grad_norm": 0.294864775838586,
      "learning_rate": 2.379472724885108e-05,
      "loss": 0.4208,
      "step": 185
    },
    {
      "epoch": 0.340036563071298,
      "grad_norm": 0.2815964015764938,
      "learning_rate": 2.3721013079815645e-05,
      "loss": 0.445,
      "step": 186
    },
    {
      "epoch": 0.34186471663619744,
      "grad_norm": 0.31854040107909304,
      "learning_rate": 2.3646979366799234e-05,
      "loss": 0.4601,
      "step": 187
    },
    {
      "epoch": 0.3436928702010969,
      "grad_norm": 0.28167691281400475,
      "learning_rate": 2.3572628822448546e-05,
      "loss": 0.4506,
      "step": 188
    },
    {
      "epoch": 0.34552102376599636,
      "grad_norm": 0.2771446663943378,
      "learning_rate": 2.3497964171019214e-05,
      "loss": 0.4358,
      "step": 189
    },
    {
      "epoch": 0.3473491773308958,
      "grad_norm": 0.27745556919254083,
      "learning_rate": 2.3422988148275963e-05,
      "loss": 0.4461,
      "step": 190
    },
    {
      "epoch": 0.3491773308957952,
      "grad_norm": 0.26029043603341223,
      "learning_rate": 2.3347703501392373e-05,
      "loss": 0.4538,
      "step": 191
    },
    {
      "epoch": 0.3510054844606947,
      "grad_norm": 0.2744751742780149,
      "learning_rate": 2.3272112988850237e-05,
      "loss": 0.4343,
      "step": 192
    },
    {
      "epoch": 0.35283363802559414,
      "grad_norm": 0.3148709000115808,
      "learning_rate": 2.319621938033848e-05,
      "loss": 0.4421,
      "step": 193
    },
    {
      "epoch": 0.3546617915904936,
      "grad_norm": 0.2802326000313149,
      "learning_rate": 2.3120025456651658e-05,
      "loss": 0.4528,
      "step": 194
    },
    {
      "epoch": 0.35648994515539306,
      "grad_norm": 0.25928215844810953,
      "learning_rate": 2.3043534009588108e-05,
      "loss": 0.4405,
      "step": 195
    },
    {
      "epoch": 0.3583180987202925,
      "grad_norm": 0.2732416188380195,
      "learning_rate": 2.296674784184761e-05,
      "loss": 0.4123,
      "step": 196
    },
    {
      "epoch": 0.360146252285192,
      "grad_norm": 0.2616002327630866,
      "learning_rate": 2.288966976692873e-05,
      "loss": 0.4211,
      "step": 197
    },
    {
      "epoch": 0.3619744058500914,
      "grad_norm": 0.25552504229923495,
      "learning_rate": 2.2812302609025692e-05,
      "loss": 0.4219,
      "step": 198
    },
    {
      "epoch": 0.36380255941499084,
      "grad_norm": 0.26460690701785383,
      "learning_rate": 2.2734649202924955e-05,
      "loss": 0.4519,
      "step": 199
    },
    {
      "epoch": 0.3656307129798903,
      "grad_norm": 0.2703925981362186,
      "learning_rate": 2.265671239390128e-05,
      "loss": 0.4525,
      "step": 200
    },
    {
      "epoch": 0.36745886654478976,
      "grad_norm": 0.2728820059523706,
      "learning_rate": 2.2578495037613532e-05,
      "loss": 0.4429,
      "step": 201
    },
    {
      "epoch": 0.3692870201096892,
      "grad_norm": 0.2800868117281377,
      "learning_rate": 2.25e-05,
      "loss": 0.4554,
      "step": 202
    },
    {
      "epoch": 0.3711151736745887,
      "grad_norm": 0.2568829228472073,
      "learning_rate": 2.242123015717343e-05,
      "loss": 0.435,
      "step": 203
    },
    {
      "epoch": 0.37294332723948814,
      "grad_norm": 0.30387336120764,
      "learning_rate": 2.2342188395315614e-05,
      "loss": 0.4429,
      "step": 204
    },
    {
      "epoch": 0.37477148080438755,
      "grad_norm": 0.26064248998635736,
      "learning_rate": 2.226287761057165e-05,
      "loss": 0.4183,
      "step": 205
    },
    {
      "epoch": 0.376599634369287,
      "grad_norm": 0.2935373472617377,
      "learning_rate": 2.2183300708943803e-05,
      "loss": 0.4484,
      "step": 206
    },
    {
      "epoch": 0.37842778793418647,
      "grad_norm": 0.2647610943635089,
      "learning_rate": 2.2103460606185075e-05,
      "loss": 0.4615,
      "step": 207
    },
    {
      "epoch": 0.3802559414990859,
      "grad_norm": 0.26540150703748694,
      "learning_rate": 2.2023360227692323e-05,
      "loss": 0.4258,
      "step": 208
    },
    {
      "epoch": 0.3820840950639854,
      "grad_norm": 0.2930063038825291,
      "learning_rate": 2.194300250839908e-05,
      "loss": 0.4197,
      "step": 209
    },
    {
      "epoch": 0.38391224862888484,
      "grad_norm": 0.27789168717626905,
      "learning_rate": 2.1862390392668047e-05,
      "loss": 0.4343,
      "step": 210
    },
    {
      "epoch": 0.3857404021937843,
      "grad_norm": 0.2822656286867699,
      "learning_rate": 2.1781526834183172e-05,
      "loss": 0.4347,
      "step": 211
    },
    {
      "epoch": 0.3875685557586837,
      "grad_norm": 0.2890932572365719,
      "learning_rate": 2.1700414795841442e-05,
      "loss": 0.427,
      "step": 212
    },
    {
      "epoch": 0.38939670932358317,
      "grad_norm": 0.2696340314880456,
      "learning_rate": 2.1619057249644323e-05,
      "loss": 0.4144,
      "step": 213
    },
    {
      "epoch": 0.3912248628884826,
      "grad_norm": 0.2884764861291316,
      "learning_rate": 2.1537457176588854e-05,
      "loss": 0.4549,
      "step": 214
    },
    {
      "epoch": 0.3930530164533821,
      "grad_norm": 0.31287235950401526,
      "learning_rate": 2.145561756655844e-05,
      "loss": 0.4674,
      "step": 215
    },
    {
      "epoch": 0.39488117001828155,
      "grad_norm": 0.25161370397978683,
      "learning_rate": 2.137354141821326e-05,
      "loss": 0.4607,
      "step": 216
    },
    {
      "epoch": 0.396709323583181,
      "grad_norm": 0.27422251582421303,
      "learning_rate": 2.129123173888045e-05,
      "loss": 0.4261,
      "step": 217
    },
    {
      "epoch": 0.39853747714808047,
      "grad_norm": 0.29345259835682536,
      "learning_rate": 2.120869154444387e-05,
      "loss": 0.4454,
      "step": 218
    },
    {
      "epoch": 0.40036563071297987,
      "grad_norm": 0.2945645835125509,
      "learning_rate": 2.1125923859233613e-05,
      "loss": 0.4443,
      "step": 219
    },
    {
      "epoch": 0.40219378427787933,
      "grad_norm": 0.29318208612363506,
      "learning_rate": 2.1042931715915194e-05,
      "loss": 0.4547,
      "step": 220
    },
    {
      "epoch": 0.4040219378427788,
      "grad_norm": 0.27568759464008996,
      "learning_rate": 2.0959718155378434e-05,
      "loss": 0.4202,
      "step": 221
    },
    {
      "epoch": 0.40585009140767825,
      "grad_norm": 0.2573640811765329,
      "learning_rate": 2.0876286226626027e-05,
      "loss": 0.4397,
      "step": 222
    },
    {
      "epoch": 0.4076782449725777,
      "grad_norm": 0.266449263808659,
      "learning_rate": 2.0792638986661835e-05,
      "loss": 0.4367,
      "step": 223
    },
    {
      "epoch": 0.40950639853747717,
      "grad_norm": 0.2797833012185273,
      "learning_rate": 2.0708779500378876e-05,
      "loss": 0.4223,
      "step": 224
    },
    {
      "epoch": 0.4113345521023766,
      "grad_norm": 0.2704739515055686,
      "learning_rate": 2.0624710840447002e-05,
      "loss": 0.4703,
      "step": 225
    },
    {
      "epoch": 0.41316270566727603,
      "grad_norm": 0.2737555236350162,
      "learning_rate": 2.054043608720036e-05,
      "loss": 0.4386,
      "step": 226
    },
    {
      "epoch": 0.4149908592321755,
      "grad_norm": 0.26496890308771476,
      "learning_rate": 2.0455958328524484e-05,
      "loss": 0.42,
      "step": 227
    },
    {
      "epoch": 0.41681901279707495,
      "grad_norm": 0.2684436681945271,
      "learning_rate": 2.0371280659743166e-05,
      "loss": 0.4552,
      "step": 228
    },
    {
      "epoch": 0.4186471663619744,
      "grad_norm": 0.26353955168575416,
      "learning_rate": 2.0286406183505042e-05,
      "loss": 0.4463,
      "step": 229
    },
    {
      "epoch": 0.42047531992687387,
      "grad_norm": 0.2686887218089528,
      "learning_rate": 2.0201338009669928e-05,
      "loss": 0.4345,
      "step": 230
    },
    {
      "epoch": 0.42230347349177333,
      "grad_norm": 0.2481187566684283,
      "learning_rate": 2.0116079255194836e-05,
      "loss": 0.4083,
      "step": 231
    },
    {
      "epoch": 0.42413162705667273,
      "grad_norm": 0.25425363232581627,
      "learning_rate": 2.0030633044019785e-05,
      "loss": 0.4296,
      "step": 232
    },
    {
      "epoch": 0.4259597806215722,
      "grad_norm": 0.2553643382715429,
      "learning_rate": 1.9945002506953354e-05,
      "loss": 0.4405,
      "step": 233
    },
    {
      "epoch": 0.42778793418647165,
      "grad_norm": 0.23260100238414075,
      "learning_rate": 1.9859190781557938e-05,
      "loss": 0.4422,
      "step": 234
    },
    {
      "epoch": 0.4296160877513711,
      "grad_norm": 0.24053523204871977,
      "learning_rate": 1.9773201012034795e-05,
      "loss": 0.4212,
      "step": 235
    },
    {
      "epoch": 0.43144424131627057,
      "grad_norm": 0.23817148238959823,
      "learning_rate": 1.9687036349108852e-05,
      "loss": 0.4155,
      "step": 236
    },
    {
      "epoch": 0.43327239488117003,
      "grad_norm": 0.25879751895835235,
      "learning_rate": 1.9600699949913258e-05,
      "loss": 0.4433,
      "step": 237
    },
    {
      "epoch": 0.4351005484460695,
      "grad_norm": 0.2765030352462927,
      "learning_rate": 1.951419497787367e-05,
      "loss": 0.4463,
      "step": 238
    },
    {
      "epoch": 0.4369287020109689,
      "grad_norm": 0.23933280142645652,
      "learning_rate": 1.942752460259241e-05,
      "loss": 0.4352,
      "step": 239
    },
    {
      "epoch": 0.43875685557586835,
      "grad_norm": 0.2647575567604074,
      "learning_rate": 1.934069199973226e-05,
      "loss": 0.4514,
      "step": 240
    },
    {
      "epoch": 0.4405850091407678,
      "grad_norm": 0.29034825255703517,
      "learning_rate": 1.9253700350900144e-05,
      "loss": 0.4773,
      "step": 241
    },
    {
      "epoch": 0.4424131627056673,
      "grad_norm": 0.25882085427584095,
      "learning_rate": 1.916655284353055e-05,
      "loss": 0.4114,
      "step": 242
    },
    {
      "epoch": 0.44424131627056673,
      "grad_norm": 0.2567252272831029,
      "learning_rate": 1.9079252670768712e-05,
      "loss": 0.4164,
      "step": 243
    },
    {
      "epoch": 0.4460694698354662,
      "grad_norm": 0.30411306075561684,
      "learning_rate": 1.8991803031353648e-05,
      "loss": 0.4664,
      "step": 244
    },
    {
      "epoch": 0.44789762340036565,
      "grad_norm": 0.25636163158093184,
      "learning_rate": 1.8904207129500916e-05,
      "loss": 0.451,
      "step": 245
    },
    {
      "epoch": 0.44972577696526506,
      "grad_norm": 0.24831540800868265,
      "learning_rate": 1.881646817478526e-05,
      "loss": 0.4334,
      "step": 246
    },
    {
      "epoch": 0.4515539305301645,
      "grad_norm": 0.34433155752462036,
      "learning_rate": 1.8728589382022958e-05,
      "loss": 0.4656,
      "step": 247
    },
    {
      "epoch": 0.453382084095064,
      "grad_norm": 0.2840133269260318,
      "learning_rate": 1.864057397115406e-05,
      "loss": 0.4042,
      "step": 248
    },
    {
      "epoch": 0.45521023765996343,
      "grad_norm": 0.2412445664865069,
      "learning_rate": 1.8552425167124402e-05,
      "loss": 0.446,
      "step": 249
    },
    {
      "epoch": 0.4570383912248629,
      "grad_norm": 0.28608559782605364,
      "learning_rate": 1.8464146199767426e-05,
      "loss": 0.4476,
      "step": 250
    },
    {
      "epoch": 0.45886654478976235,
      "grad_norm": 0.30276416725864513,
      "learning_rate": 1.837574030368586e-05,
      "loss": 0.4317,
      "step": 251
    },
    {
      "epoch": 0.4606946983546618,
      "grad_norm": 0.2566520510938809,
      "learning_rate": 1.8287210718133185e-05,
      "loss": 0.4474,
      "step": 252
    },
    {
      "epoch": 0.4625228519195612,
      "grad_norm": 0.29023247848731565,
      "learning_rate": 1.819856068689495e-05,
      "loss": 0.4277,
      "step": 253
    },
    {
      "epoch": 0.4643510054844607,
      "grad_norm": 0.29650005092192266,
      "learning_rate": 1.8109793458169914e-05,
      "loss": 0.4128,
      "step": 254
    },
    {
      "epoch": 0.46617915904936014,
      "grad_norm": 0.26917175921222486,
      "learning_rate": 1.8020912284451047e-05,
      "loss": 0.4267,
      "step": 255
    },
    {
      "epoch": 0.4680073126142596,
      "grad_norm": 0.29145888646716905,
      "learning_rate": 1.7931920422406323e-05,
      "loss": 0.4261,
      "step": 256
    },
    {
      "epoch": 0.46983546617915906,
      "grad_norm": 0.2565415482843328,
      "learning_rate": 1.7842821132759427e-05,
      "loss": 0.4514,
      "step": 257
    },
    {
      "epoch": 0.4716636197440585,
      "grad_norm": 0.2971366072720857,
      "learning_rate": 1.775361768017026e-05,
      "loss": 0.4614,
      "step": 258
    },
    {
      "epoch": 0.473491773308958,
      "grad_norm": 0.2873986209641355,
      "learning_rate": 1.766431333311533e-05,
      "loss": 0.4425,
      "step": 259
    },
    {
      "epoch": 0.4753199268738574,
      "grad_norm": 0.24135740387274107,
      "learning_rate": 1.757491136376799e-05,
      "loss": 0.4385,
      "step": 260
    },
    {
      "epoch": 0.47714808043875684,
      "grad_norm": 0.26215574513189993,
      "learning_rate": 1.7485415047878525e-05,
      "loss": 0.4371,
      "step": 261
    },
    {
      "epoch": 0.4789762340036563,
      "grad_norm": 0.27160526260976153,
      "learning_rate": 1.7395827664654168e-05,
      "loss": 0.4041,
      "step": 262
    },
    {
      "epoch": 0.48080438756855576,
      "grad_norm": 0.2584166075498547,
      "learning_rate": 1.7306152496638906e-05,
      "loss": 0.4459,
      "step": 263
    },
    {
      "epoch": 0.4826325411334552,
      "grad_norm": 0.44741417832356484,
      "learning_rate": 1.7216392829593225e-05,
      "loss": 0.4459,
      "step": 264
    },
    {
      "epoch": 0.4844606946983547,
      "grad_norm": 0.25272710854107394,
      "learning_rate": 1.712655195237371e-05,
      "loss": 0.4363,
      "step": 265
    },
    {
      "epoch": 0.48628884826325414,
      "grad_norm": 0.26321144440276856,
      "learning_rate": 1.703663315681256e-05,
      "loss": 0.4361,
      "step": 266
    },
    {
      "epoch": 0.48811700182815354,
      "grad_norm": 0.26092944626135084,
      "learning_rate": 1.6946639737596936e-05,
      "loss": 0.455,
      "step": 267
    },
    {
      "epoch": 0.489945155393053,
      "grad_norm": 0.25963159576066375,
      "learning_rate": 1.685657499214827e-05,
      "loss": 0.4219,
      "step": 268
    },
    {
      "epoch": 0.49177330895795246,
      "grad_norm": 0.28567170577220685,
      "learning_rate": 1.676644222050144e-05,
      "loss": 0.4702,
      "step": 269
    },
    {
      "epoch": 0.4936014625228519,
      "grad_norm": 0.33362201657492396,
      "learning_rate": 1.6676244725183848e-05,
      "loss": 0.4316,
      "step": 270
    },
    {
      "epoch": 0.4954296160877514,
      "grad_norm": 0.2318177887728292,
      "learning_rate": 1.6585985811094412e-05,
      "loss": 0.4261,
      "step": 271
    },
    {
      "epoch": 0.49725776965265084,
      "grad_norm": 0.26439068415936395,
      "learning_rate": 1.6495668785382482e-05,
      "loss": 0.4379,
      "step": 272
    },
    {
      "epoch": 0.4990859232175503,
      "grad_norm": 0.2689860082042829,
      "learning_rate": 1.6405296957326665e-05,
      "loss": 0.4449,
      "step": 273
    },
    {
      "epoch": 0.5009140767824497,
      "grad_norm": 0.24934324053938617,
      "learning_rate": 1.631487363821354e-05,
      "loss": 0.4499,
      "step": 274
    },
    {
      "epoch": 0.5027422303473492,
      "grad_norm": 0.2556314468775107,
      "learning_rate": 1.6224402141216387e-05,
      "loss": 0.4308,
      "step": 275
    },
    {
      "epoch": 0.5045703839122486,
      "grad_norm": 0.23996383832876597,
      "learning_rate": 1.6133885781273724e-05,
      "loss": 0.445,
      "step": 276
    },
    {
      "epoch": 0.506398537477148,
      "grad_norm": 0.2409486532934403,
      "learning_rate": 1.6043327874967907e-05,
      "loss": 0.4267,
      "step": 277
    },
    {
      "epoch": 0.5082266910420475,
      "grad_norm": 0.2440582300231897,
      "learning_rate": 1.595273174040357e-05,
      "loss": 0.4551,
      "step": 278
    },
    {
      "epoch": 0.5100548446069469,
      "grad_norm": 0.2531767967223094,
      "learning_rate": 1.5862100697086037e-05,
      "loss": 0.433,
      "step": 279
    },
    {
      "epoch": 0.5118829981718465,
      "grad_norm": 0.27180812225430423,
      "learning_rate": 1.5771438065799745e-05,
      "loss": 0.4469,
      "step": 280
    },
    {
      "epoch": 0.5137111517367459,
      "grad_norm": 0.3856477837916029,
      "learning_rate": 1.568074716848652e-05,
      "loss": 0.4342,
      "step": 281
    },
    {
      "epoch": 0.5155393053016454,
      "grad_norm": 0.2412031468925041,
      "learning_rate": 1.559003132812387e-05,
      "loss": 0.4265,
      "step": 282
    },
    {
      "epoch": 0.5173674588665448,
      "grad_norm": 0.2708731472554568,
      "learning_rate": 1.5499293868603244e-05,
      "loss": 0.4449,
      "step": 283
    },
    {
      "epoch": 0.5191956124314442,
      "grad_norm": 0.238845552792791,
      "learning_rate": 1.5408538114608227e-05,
      "loss": 0.4713,
      "step": 284
    },
    {
      "epoch": 0.5210237659963437,
      "grad_norm": 0.23807280669571856,
      "learning_rate": 1.531776739149273e-05,
      "loss": 0.4629,
      "step": 285
    },
    {
      "epoch": 0.5228519195612431,
      "grad_norm": 0.26318324936651355,
      "learning_rate": 1.522698502515913e-05,
      "loss": 0.4265,
      "step": 286
    },
    {
      "epoch": 0.5246800731261426,
      "grad_norm": 0.26366427329382985,
      "learning_rate": 1.513619434193645e-05,
      "loss": 0.4369,
      "step": 287
    },
    {
      "epoch": 0.526508226691042,
      "grad_norm": 0.29440645333058113,
      "learning_rate": 1.5045398668458405e-05,
      "loss": 0.4092,
      "step": 288
    },
    {
      "epoch": 0.5283363802559415,
      "grad_norm": 0.27525587616540875,
      "learning_rate": 1.4954601331541598e-05,
      "loss": 0.4227,
      "step": 289
    },
    {
      "epoch": 0.5301645338208409,
      "grad_norm": 0.2410473434834418,
      "learning_rate": 1.4863805658063553e-05,
      "loss": 0.4489,
      "step": 290
    },
    {
      "epoch": 0.5319926873857403,
      "grad_norm": 0.2504271854549055,
      "learning_rate": 1.4773014974840868e-05,
      "loss": 0.4035,
      "step": 291
    },
    {
      "epoch": 0.5338208409506399,
      "grad_norm": 0.24694462096865133,
      "learning_rate": 1.4682232608507276e-05,
      "loss": 0.4237,
      "step": 292
    },
    {
      "epoch": 0.5356489945155393,
      "grad_norm": 0.2496405063277231,
      "learning_rate": 1.4591461885391774e-05,
      "loss": 0.4708,
      "step": 293
    },
    {
      "epoch": 0.5374771480804388,
      "grad_norm": 0.24558976709050298,
      "learning_rate": 1.4500706131396758e-05,
      "loss": 0.456,
      "step": 294
    },
    {
      "epoch": 0.5393053016453382,
      "grad_norm": 0.23547536499363428,
      "learning_rate": 1.440996867187613e-05,
      "loss": 0.4212,
      "step": 295
    },
    {
      "epoch": 0.5411334552102377,
      "grad_norm": 0.2576372310111232,
      "learning_rate": 1.4319252831513483e-05,
      "loss": 0.4534,
      "step": 296
    },
    {
      "epoch": 0.5429616087751371,
      "grad_norm": 0.2701654340457875,
      "learning_rate": 1.4228561934200252e-05,
      "loss": 0.4298,
      "step": 297
    },
    {
      "epoch": 0.5447897623400365,
      "grad_norm": 0.22197880621904825,
      "learning_rate": 1.4137899302913967e-05,
      "loss": 0.4091,
      "step": 298
    },
    {
      "epoch": 0.546617915904936,
      "grad_norm": 0.26073305048640977,
      "learning_rate": 1.4047268259596438e-05,
      "loss": 0.4401,
      "step": 299
    },
    {
      "epoch": 0.5484460694698354,
      "grad_norm": 0.2500444755831797,
      "learning_rate": 1.3956672125032095e-05,
      "loss": 0.43,
      "step": 300
    },
    {
      "epoch": 0.5502742230347349,
      "grad_norm": 0.2594460218106809,
      "learning_rate": 1.3866114218726276e-05,
      "loss": 0.4399,
      "step": 301
    },
    {
      "epoch": 0.5521023765996343,
      "grad_norm": 0.2239743874994638,
      "learning_rate": 1.3775597858783618e-05,
      "loss": 0.4045,
      "step": 302
    },
    {
      "epoch": 0.5539305301645339,
      "grad_norm": 0.25621586458900986,
      "learning_rate": 1.3685126361786459e-05,
      "loss": 0.465,
      "step": 303
    },
    {
      "epoch": 0.5557586837294333,
      "grad_norm": 0.2416564181734943,
      "learning_rate": 1.359470304267334e-05,
      "loss": 0.4294,
      "step": 304
    },
    {
      "epoch": 0.5575868372943327,
      "grad_norm": 0.28991744031002165,
      "learning_rate": 1.3504331214617522e-05,
      "loss": 0.4042,
      "step": 305
    },
    {
      "epoch": 0.5594149908592322,
      "grad_norm": 0.24905920744460144,
      "learning_rate": 1.3414014188905592e-05,
      "loss": 0.4276,
      "step": 306
    },
    {
      "epoch": 0.5612431444241316,
      "grad_norm": 0.24076167455030184,
      "learning_rate": 1.3323755274816156e-05,
      "loss": 0.4216,
      "step": 307
    },
    {
      "epoch": 0.5630712979890311,
      "grad_norm": 0.2179929608935108,
      "learning_rate": 1.323355777949856e-05,
      "loss": 0.4348,
      "step": 308
    },
    {
      "epoch": 0.5648994515539305,
      "grad_norm": 0.2432958183127804,
      "learning_rate": 1.3143425007851731e-05,
      "loss": 0.403,
      "step": 309
    },
    {
      "epoch": 0.56672760511883,
      "grad_norm": 0.22214885973663206,
      "learning_rate": 1.3053360262403065e-05,
      "loss": 0.428,
      "step": 310
    },
    {
      "epoch": 0.5685557586837294,
      "grad_norm": 0.23778389557885163,
      "learning_rate": 1.2963366843187444e-05,
      "loss": 0.4574,
      "step": 311
    },
    {
      "epoch": 0.5703839122486288,
      "grad_norm": 0.22719896435699155,
      "learning_rate": 1.2873448047626292e-05,
      "loss": 0.4256,
      "step": 312
    },
    {
      "epoch": 0.5722120658135283,
      "grad_norm": 0.22414999604986005,
      "learning_rate": 1.278360717040678e-05,
      "loss": 0.4414,
      "step": 313
    },
    {
      "epoch": 0.5740402193784278,
      "grad_norm": 0.2282449074945814,
      "learning_rate": 1.2693847503361098e-05,
      "loss": 0.4399,
      "step": 314
    },
    {
      "epoch": 0.5758683729433273,
      "grad_norm": 0.23794931799275706,
      "learning_rate": 1.2604172335345831e-05,
      "loss": 0.4288,
      "step": 315
    },
    {
      "epoch": 0.5776965265082267,
      "grad_norm": 0.23558540234348077,
      "learning_rate": 1.2514584952121476e-05,
      "loss": 0.431,
      "step": 316
    },
    {
      "epoch": 0.5795246800731262,
      "grad_norm": 0.21855965885160233,
      "learning_rate": 1.2425088636232018e-05,
      "loss": 0.4244,
      "step": 317
    },
    {
      "epoch": 0.5813528336380256,
      "grad_norm": 0.24100093478254633,
      "learning_rate": 1.233568666688467e-05,
      "loss": 0.4251,
      "step": 318
    },
    {
      "epoch": 0.583180987202925,
      "grad_norm": 0.22716191212818615,
      "learning_rate": 1.2246382319829742e-05,
      "loss": 0.4321,
      "step": 319
    },
    {
      "epoch": 0.5850091407678245,
      "grad_norm": 0.2154876750241009,
      "learning_rate": 1.2157178867240578e-05,
      "loss": 0.4126,
      "step": 320
    },
    {
      "epoch": 0.5868372943327239,
      "grad_norm": 0.21785385521233888,
      "learning_rate": 1.2068079577593677e-05,
      "loss": 0.4345,
      "step": 321
    },
    {
      "epoch": 0.5886654478976234,
      "grad_norm": 0.23106294766490781,
      "learning_rate": 1.1979087715548956e-05,
      "loss": 0.4265,
      "step": 322
    },
    {
      "epoch": 0.5904936014625228,
      "grad_norm": 0.245939334251153,
      "learning_rate": 1.1890206541830081e-05,
      "loss": 0.4366,
      "step": 323
    },
    {
      "epoch": 0.5923217550274223,
      "grad_norm": 0.23149020974733392,
      "learning_rate": 1.1801439313105053e-05,
      "loss": 0.414,
      "step": 324
    },
    {
      "epoch": 0.5941499085923218,
      "grad_norm": 0.245887110196614,
      "learning_rate": 1.171278928186682e-05,
      "loss": 0.4324,
      "step": 325
    },
    {
      "epoch": 0.5959780621572212,
      "grad_norm": 0.22527628578658415,
      "learning_rate": 1.1624259696314142e-05,
      "loss": 0.4452,
      "step": 326
    },
    {
      "epoch": 0.5978062157221207,
      "grad_norm": 0.22158971411757605,
      "learning_rate": 1.1535853800232578e-05,
      "loss": 0.422,
      "step": 327
    },
    {
      "epoch": 0.5996343692870201,
      "grad_norm": 0.20068045774387033,
      "learning_rate": 1.1447574832875602e-05,
      "loss": 0.4152,
      "step": 328
    },
    {
      "epoch": 0.6014625228519196,
      "grad_norm": 0.2882378294076923,
      "learning_rate": 1.1359426028845942e-05,
      "loss": 0.4392,
      "step": 329
    },
    {
      "epoch": 0.603290676416819,
      "grad_norm": 0.2547910430893769,
      "learning_rate": 1.1271410617977048e-05,
      "loss": 0.4316,
      "step": 330
    },
    {
      "epoch": 0.6051188299817185,
      "grad_norm": 0.23632189451355984,
      "learning_rate": 1.1183531825214744e-05,
      "loss": 0.4513,
      "step": 331
    },
    {
      "epoch": 0.6069469835466179,
      "grad_norm": 0.23822493582632703,
      "learning_rate": 1.1095792870499087e-05,
      "loss": 0.437,
      "step": 332
    },
    {
      "epoch": 0.6087751371115173,
      "grad_norm": 0.2758107484432963,
      "learning_rate": 1.1008196968646356e-05,
      "loss": 0.4717,
      "step": 333
    },
    {
      "epoch": 0.6106032906764168,
      "grad_norm": 0.2775006198175233,
      "learning_rate": 1.0920747329231289e-05,
      "loss": 0.4553,
      "step": 334
    },
    {
      "epoch": 0.6124314442413162,
      "grad_norm": 0.28823062194781823,
      "learning_rate": 1.0833447156469448e-05,
      "loss": 0.4126,
      "step": 335
    },
    {
      "epoch": 0.6142595978062158,
      "grad_norm": 0.26030439707102543,
      "learning_rate": 1.0746299649099855e-05,
      "loss": 0.4545,
      "step": 336
    },
    {
      "epoch": 0.6160877513711152,
      "grad_norm": 0.23051900183913224,
      "learning_rate": 1.0659308000267744e-05,
      "loss": 0.4303,
      "step": 337
    },
    {
      "epoch": 0.6179159049360147,
      "grad_norm": 0.2654378985826582,
      "learning_rate": 1.0572475397407594e-05,
      "loss": 0.4363,
      "step": 338
    },
    {
      "epoch": 0.6197440585009141,
      "grad_norm": 0.2601478752634972,
      "learning_rate": 1.0485805022126331e-05,
      "loss": 0.4311,
      "step": 339
    },
    {
      "epoch": 0.6215722120658135,
      "grad_norm": 0.22937423133347384,
      "learning_rate": 1.0399300050086748e-05,
      "loss": 0.4131,
      "step": 340
    },
    {
      "epoch": 0.623400365630713,
      "grad_norm": 0.23316785559897982,
      "learning_rate": 1.0312963650891147e-05,
      "loss": 0.4249,
      "step": 341
    },
    {
      "epoch": 0.6252285191956124,
      "grad_norm": 0.21170925446852945,
      "learning_rate": 1.0226798987965207e-05,
      "loss": 0.397,
      "step": 342
    },
    {
      "epoch": 0.6270566727605119,
      "grad_norm": 0.2765847344469056,
      "learning_rate": 1.014080921844207e-05,
      "loss": 0.4192,
      "step": 343
    },
    {
      "epoch": 0.6288848263254113,
      "grad_norm": 0.24840999968309166,
      "learning_rate": 1.0054997493046647e-05,
      "loss": 0.4253,
      "step": 344
    },
    {
      "epoch": 0.6307129798903108,
      "grad_norm": 0.23384641108626905,
      "learning_rate": 9.969366955980217e-06,
      "loss": 0.4418,
      "step": 345
    },
    {
      "epoch": 0.6325411334552102,
      "grad_norm": 0.21288227201402393,
      "learning_rate": 9.883920744805168e-06,
      "loss": 0.4269,
      "step": 346
    },
    {
      "epoch": 0.6343692870201096,
      "grad_norm": 0.2556800166390414,
      "learning_rate": 9.798661990330073e-06,
      "loss": 0.4493,
      "step": 347
    },
    {
      "epoch": 0.6361974405850092,
      "grad_norm": 0.23939265719244443,
      "learning_rate": 9.713593816494955e-06,
      "loss": 0.4285,
      "step": 348
    },
    {
      "epoch": 0.6380255941499086,
      "grad_norm": 0.2584881278947507,
      "learning_rate": 9.628719340256842e-06,
      "loss": 0.421,
      "step": 349
    },
    {
      "epoch": 0.6398537477148081,
      "grad_norm": 0.32586444845498075,
      "learning_rate": 9.544041671475522e-06,
      "loss": 0.4305,
      "step": 350
    },
    {
      "epoch": 0.6416819012797075,
      "grad_norm": 0.2194129722742284,
      "learning_rate": 9.45956391279964e-06,
      "loss": 0.4419,
      "step": 351
    },
    {
      "epoch": 0.643510054844607,
      "grad_norm": 0.2305484748288966,
      "learning_rate": 9.375289159552999e-06,
      "loss": 0.4378,
      "step": 352
    },
    {
      "epoch": 0.6453382084095064,
      "grad_norm": 0.24009228021122206,
      "learning_rate": 9.291220499621128e-06,
      "loss": 0.4293,
      "step": 353
    },
    {
      "epoch": 0.6471663619744058,
      "grad_norm": 0.24244845862851294,
      "learning_rate": 9.207361013338166e-06,
      "loss": 0.441,
      "step": 354
    },
    {
      "epoch": 0.6489945155393053,
      "grad_norm": 0.2469640042659817,
      "learning_rate": 9.123713773373977e-06,
      "loss": 0.4443,
      "step": 355
    },
    {
      "epoch": 0.6508226691042047,
      "grad_norm": 0.23826650781723333,
      "learning_rate": 9.040281844621568e-06,
      "loss": 0.4488,
      "step": 356
    },
    {
      "epoch": 0.6526508226691042,
      "grad_norm": 0.23960497327776573,
      "learning_rate": 8.95706828408481e-06,
      "loss": 0.4658,
      "step": 357
    },
    {
      "epoch": 0.6544789762340036,
      "grad_norm": 0.2339333378633527,
      "learning_rate": 8.874076140766391e-06,
      "loss": 0.4323,
      "step": 358
    },
    {
      "epoch": 0.6563071297989032,
      "grad_norm": 0.2722614454921542,
      "learning_rate": 8.791308455556134e-06,
      "loss": 0.4644,
      "step": 359
    },
    {
      "epoch": 0.6581352833638026,
      "grad_norm": 0.22016373930511474,
      "learning_rate": 8.708768261119549e-06,
      "loss": 0.404,
      "step": 360
    },
    {
      "epoch": 0.659963436928702,
      "grad_norm": 0.21854922398945179,
      "learning_rate": 8.626458581786741e-06,
      "loss": 0.4258,
      "step": 361
    },
    {
      "epoch": 0.6617915904936015,
      "grad_norm": 0.23149884953038544,
      "learning_rate": 8.54438243344157e-06,
      "loss": 0.4297,
      "step": 362
    },
    {
      "epoch": 0.6636197440585009,
      "grad_norm": 0.23821733038217283,
      "learning_rate": 8.462542823411145e-06,
      "loss": 0.4661,
      "step": 363
    },
    {
      "epoch": 0.6654478976234004,
      "grad_norm": 0.2169203574834392,
      "learning_rate": 8.380942750355678e-06,
      "loss": 0.4093,
      "step": 364
    },
    {
      "epoch": 0.6672760511882998,
      "grad_norm": 0.2297206779373042,
      "learning_rate": 8.29958520415856e-06,
      "loss": 0.434,
      "step": 365
    },
    {
      "epoch": 0.6691042047531993,
      "grad_norm": 0.23935881694189906,
      "learning_rate": 8.218473165816832e-06,
      "loss": 0.4265,
      "step": 366
    },
    {
      "epoch": 0.6709323583180987,
      "grad_norm": 0.23933928175540814,
      "learning_rate": 8.13760960733195e-06,
      "loss": 0.4293,
      "step": 367
    },
    {
      "epoch": 0.6727605118829981,
      "grad_norm": 0.23725621557658283,
      "learning_rate": 8.056997491600925e-06,
      "loss": 0.429,
      "step": 368
    },
    {
      "epoch": 0.6745886654478976,
      "grad_norm": 0.22714040033493302,
      "learning_rate": 7.976639772307686e-06,
      "loss": 0.4455,
      "step": 369
    },
    {
      "epoch": 0.676416819012797,
      "grad_norm": 0.24340512967756703,
      "learning_rate": 7.896539393814926e-06,
      "loss": 0.4347,
      "step": 370
    },
    {
      "epoch": 0.6782449725776966,
      "grad_norm": 0.2369924118603579,
      "learning_rate": 7.816699291056196e-06,
      "loss": 0.4279,
      "step": 371
    },
    {
      "epoch": 0.680073126142596,
      "grad_norm": 0.23385503687850978,
      "learning_rate": 7.737122389428354e-06,
      "loss": 0.4534,
      "step": 372
    },
    {
      "epoch": 0.6819012797074955,
      "grad_norm": 0.2239009026683857,
      "learning_rate": 7.657811604684385e-06,
      "loss": 0.44,
      "step": 373
    },
    {
      "epoch": 0.6837294332723949,
      "grad_norm": 0.23755170361829636,
      "learning_rate": 7.578769842826572e-06,
      "loss": 0.4169,
      "step": 374
    },
    {
      "epoch": 0.6855575868372943,
      "grad_norm": 0.26782081651902645,
      "learning_rate": 7.500000000000004e-06,
      "loss": 0.4394,
      "step": 375
    },
    {
      "epoch": 0.6873857404021938,
      "grad_norm": 0.23795613379536035,
      "learning_rate": 7.421504962386476e-06,
      "loss": 0.4198,
      "step": 376
    },
    {
      "epoch": 0.6892138939670932,
      "grad_norm": 0.21501096415396395,
      "learning_rate": 7.343287606098719e-06,
      "loss": 0.4363,
      "step": 377
    },
    {
      "epoch": 0.6910420475319927,
      "grad_norm": 0.2379437479294898,
      "learning_rate": 7.265350797075046e-06,
      "loss": 0.427,
      "step": 378
    },
    {
      "epoch": 0.6928702010968921,
      "grad_norm": 0.252686726735923,
      "learning_rate": 7.187697390974307e-06,
      "loss": 0.4398,
      "step": 379
    },
    {
      "epoch": 0.6946983546617916,
      "grad_norm": 0.2363724881907177,
      "learning_rate": 7.110330233071275e-06,
      "loss": 0.422,
      "step": 380
    },
    {
      "epoch": 0.696526508226691,
      "grad_norm": 0.2340508372928031,
      "learning_rate": 7.03325215815239e-06,
      "loss": 0.4674,
      "step": 381
    },
    {
      "epoch": 0.6983546617915904,
      "grad_norm": 0.23902036680650365,
      "learning_rate": 6.9564659904118925e-06,
      "loss": 0.4473,
      "step": 382
    },
    {
      "epoch": 0.70018281535649,
      "grad_norm": 0.23127217539471345,
      "learning_rate": 6.879974543348344e-06,
      "loss": 0.453,
      "step": 383
    },
    {
      "epoch": 0.7020109689213894,
      "grad_norm": 0.22708026022373418,
      "learning_rate": 6.803780619661527e-06,
      "loss": 0.4337,
      "step": 384
    },
    {
      "epoch": 0.7038391224862889,
      "grad_norm": 0.22322159823736962,
      "learning_rate": 6.727887011149762e-06,
      "loss": 0.4105,
      "step": 385
    },
    {
      "epoch": 0.7056672760511883,
      "grad_norm": 0.20597007119882674,
      "learning_rate": 6.652296498607628e-06,
      "loss": 0.4242,
      "step": 386
    },
    {
      "epoch": 0.7074954296160878,
      "grad_norm": 0.23128280054642733,
      "learning_rate": 6.577011851724044e-06,
      "loss": 0.4284,
      "step": 387
    },
    {
      "epoch": 0.7093235831809872,
      "grad_norm": 0.21734808921800447,
      "learning_rate": 6.502035828980786e-06,
      "loss": 0.4431,
      "step": 388
    },
    {
      "epoch": 0.7111517367458866,
      "grad_norm": 0.23383675393733833,
      "learning_rate": 6.4273711775514545e-06,
      "loss": 0.426,
      "step": 389
    },
    {
      "epoch": 0.7129798903107861,
      "grad_norm": 0.2295390524861705,
      "learning_rate": 6.353020633200773e-06,
      "loss": 0.4328,
      "step": 390
    },
    {
      "epoch": 0.7148080438756855,
      "grad_norm": 0.2156524952069054,
      "learning_rate": 6.278986920184362e-06,
      "loss": 0.4403,
      "step": 391
    },
    {
      "epoch": 0.716636197440585,
      "grad_norm": 0.23782474572747483,
      "learning_rate": 6.2052727511489224e-06,
      "loss": 0.435,
      "step": 392
    },
    {
      "epoch": 0.7184643510054844,
      "grad_norm": 0.20877179068075571,
      "learning_rate": 6.1318808270328595e-06,
      "loss": 0.4393,
      "step": 393
    },
    {
      "epoch": 0.720292504570384,
      "grad_norm": 0.19114849803588224,
      "learning_rate": 6.058813836967268e-06,
      "loss": 0.4239,
      "step": 394
    },
    {
      "epoch": 0.7221206581352834,
      "grad_norm": 0.19973243518529102,
      "learning_rate": 5.986074458177454e-06,
      "loss": 0.4028,
      "step": 395
    },
    {
      "epoch": 0.7239488117001828,
      "grad_norm": 0.21689931021168976,
      "learning_rate": 5.913665355884823e-06,
      "loss": 0.4538,
      "step": 396
    },
    {
      "epoch": 0.7257769652650823,
      "grad_norm": 0.29077463019229743,
      "learning_rate": 5.841589183209211e-06,
      "loss": 0.4193,
      "step": 397
    },
    {
      "epoch": 0.7276051188299817,
      "grad_norm": 0.22761140202109129,
      "learning_rate": 5.769848581071677e-06,
      "loss": 0.4384,
      "step": 398
    },
    {
      "epoch": 0.7294332723948812,
      "grad_norm": 0.22190022752222624,
      "learning_rate": 5.698446178097748e-06,
      "loss": 0.4157,
      "step": 399
    },
    {
      "epoch": 0.7312614259597806,
      "grad_norm": 0.21563520954942644,
      "learning_rate": 5.627384590521113e-06,
      "loss": 0.4397,
      "step": 400
    }
  ],
  "logging_steps": 1,
  "max_steps": 547,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 200,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 263845278842880.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}