{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 456,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.006593406593406593,
      "grad_norm": 0.4668516516685486,
      "learning_rate": 0.0,
      "loss": 0.7034,
      "step": 1
    },
    {
      "epoch": 0.013186813186813187,
      "grad_norm": 0.44846728444099426,
      "learning_rate": 4e-05,
      "loss": 0.6371,
      "step": 2
    },
    {
      "epoch": 0.01978021978021978,
      "grad_norm": 0.4179936647415161,
      "learning_rate": 8e-05,
      "loss": 0.6892,
      "step": 3
    },
    {
      "epoch": 0.026373626373626374,
      "grad_norm": 0.5183476209640503,
      "learning_rate": 0.00012,
      "loss": 0.6733,
      "step": 4
    },
    {
      "epoch": 0.03296703296703297,
      "grad_norm": 0.4141606390476227,
      "learning_rate": 0.00016,
      "loss": 0.6744,
      "step": 5
    },
    {
      "epoch": 0.03956043956043956,
      "grad_norm": 0.41007187962532043,
      "learning_rate": 0.0002,
      "loss": 0.6728,
      "step": 6
    },
    {
      "epoch": 0.046153846153846156,
      "grad_norm": 0.6870153546333313,
      "learning_rate": 0.00019955654101995565,
      "loss": 0.6332,
      "step": 7
    },
    {
      "epoch": 0.05274725274725275,
      "grad_norm": 0.5108512043952942,
      "learning_rate": 0.00019911308203991133,
      "loss": 0.691,
      "step": 8
    },
    {
      "epoch": 0.05934065934065934,
      "grad_norm": 0.4644394814968109,
      "learning_rate": 0.00019866962305986697,
      "loss": 0.6771,
      "step": 9
    },
    {
      "epoch": 0.06593406593406594,
      "grad_norm": 0.28286677598953247,
      "learning_rate": 0.00019822616407982261,
      "loss": 0.5937,
      "step": 10
    },
    {
      "epoch": 0.07252747252747253,
      "grad_norm": 0.2530657649040222,
      "learning_rate": 0.00019778270509977829,
      "loss": 0.6188,
      "step": 11
    },
    {
      "epoch": 0.07912087912087912,
      "grad_norm": 0.1987961381673813,
      "learning_rate": 0.00019733924611973393,
      "loss": 0.6398,
      "step": 12
    },
    {
      "epoch": 0.08571428571428572,
      "grad_norm": 0.18104024231433868,
      "learning_rate": 0.0001968957871396896,
      "loss": 0.6224,
      "step": 13
    },
    {
      "epoch": 0.09230769230769231,
      "grad_norm": 0.18297415971755981,
      "learning_rate": 0.00019645232815964525,
      "loss": 0.5222,
      "step": 14
    },
    {
      "epoch": 0.0989010989010989,
      "grad_norm": 0.29442763328552246,
      "learning_rate": 0.00019600886917960092,
      "loss": 0.5976,
      "step": 15
    },
    {
      "epoch": 0.1054945054945055,
      "grad_norm": 0.34058958292007446,
      "learning_rate": 0.00019556541019955653,
      "loss": 0.6337,
      "step": 16
    },
    {
      "epoch": 0.11208791208791209,
      "grad_norm": 0.29069840908050537,
      "learning_rate": 0.0001951219512195122,
      "loss": 0.6109,
      "step": 17
    },
    {
      "epoch": 0.11868131868131868,
      "grad_norm": 0.2199062705039978,
      "learning_rate": 0.00019467849223946785,
      "loss": 0.6149,
      "step": 18
    },
    {
      "epoch": 0.12527472527472527,
      "grad_norm": 0.23966625332832336,
      "learning_rate": 0.00019423503325942352,
      "loss": 0.6908,
      "step": 19
    },
    {
      "epoch": 0.13186813186813187,
      "grad_norm": 0.29024893045425415,
      "learning_rate": 0.00019379157427937917,
      "loss": 0.603,
      "step": 20
    },
    {
      "epoch": 0.13846153846153847,
      "grad_norm": 0.19527003169059753,
      "learning_rate": 0.00019334811529933484,
      "loss": 0.4587,
      "step": 21
    },
    {
      "epoch": 0.14505494505494507,
      "grad_norm": 0.22737528383731842,
      "learning_rate": 0.00019290465631929045,
      "loss": 0.6066,
      "step": 22
    },
    {
      "epoch": 0.15164835164835164,
      "grad_norm": 0.19743725657463074,
      "learning_rate": 0.00019246119733924613,
      "loss": 0.5668,
      "step": 23
    },
    {
      "epoch": 0.15824175824175823,
      "grad_norm": 0.2503700852394104,
      "learning_rate": 0.00019201773835920177,
      "loss": 0.619,
      "step": 24
    },
    {
      "epoch": 0.16483516483516483,
      "grad_norm": 0.19420984387397766,
      "learning_rate": 0.00019157427937915744,
      "loss": 0.5819,
      "step": 25
    },
    {
      "epoch": 0.17142857142857143,
      "grad_norm": 0.21643847227096558,
      "learning_rate": 0.00019113082039911309,
      "loss": 0.5903,
      "step": 26
    },
    {
      "epoch": 0.17802197802197803,
      "grad_norm": 0.16861997544765472,
      "learning_rate": 0.00019068736141906876,
      "loss": 0.5422,
      "step": 27
    },
    {
      "epoch": 0.18461538461538463,
      "grad_norm": 0.22646990418434143,
      "learning_rate": 0.0001902439024390244,
      "loss": 0.5798,
      "step": 28
    },
    {
      "epoch": 0.1912087912087912,
      "grad_norm": 0.1941661536693573,
      "learning_rate": 0.00018980044345898005,
      "loss": 0.654,
      "step": 29
    },
    {
      "epoch": 0.1978021978021978,
      "grad_norm": 0.2271111011505127,
      "learning_rate": 0.00018935698447893572,
      "loss": 0.6043,
      "step": 30
    },
    {
      "epoch": 0.2043956043956044,
      "grad_norm": 0.23530516028404236,
      "learning_rate": 0.00018891352549889136,
      "loss": 0.6002,
      "step": 31
    },
    {
      "epoch": 0.210989010989011,
      "grad_norm": 0.2832397222518921,
      "learning_rate": 0.00018847006651884703,
      "loss": 0.607,
      "step": 32
    },
    {
      "epoch": 0.2175824175824176,
      "grad_norm": 0.27622371912002563,
      "learning_rate": 0.00018802660753880268,
      "loss": 0.6409,
      "step": 33
    },
    {
      "epoch": 0.22417582417582418,
      "grad_norm": 0.23992006480693817,
      "learning_rate": 0.00018758314855875832,
      "loss": 0.5801,
      "step": 34
    },
    {
      "epoch": 0.23076923076923078,
      "grad_norm": 0.1933351457118988,
      "learning_rate": 0.00018713968957871397,
      "loss": 0.5257,
      "step": 35
    },
    {
      "epoch": 0.23736263736263735,
      "grad_norm": 0.19885019958019257,
      "learning_rate": 0.00018669623059866964,
      "loss": 0.5203,
      "step": 36
    },
    {
      "epoch": 0.24395604395604395,
      "grad_norm": 0.19667015969753265,
      "learning_rate": 0.00018625277161862528,
      "loss": 0.5367,
      "step": 37
    },
    {
      "epoch": 0.25054945054945055,
      "grad_norm": 0.1923234760761261,
      "learning_rate": 0.00018580931263858095,
      "loss": 0.5901,
      "step": 38
    },
    {
      "epoch": 0.2571428571428571,
      "grad_norm": 0.22012682259082794,
      "learning_rate": 0.0001853658536585366,
      "loss": 0.6576,
      "step": 39
    },
    {
      "epoch": 0.26373626373626374,
      "grad_norm": 0.1744205802679062,
      "learning_rate": 0.00018492239467849224,
      "loss": 0.5573,
      "step": 40
    },
    {
      "epoch": 0.2703296703296703,
      "grad_norm": 0.17654304206371307,
      "learning_rate": 0.0001844789356984479,
      "loss": 0.6044,
      "step": 41
    },
    {
      "epoch": 0.27692307692307694,
      "grad_norm": 0.20053048431873322,
      "learning_rate": 0.00018403547671840356,
      "loss": 0.5933,
      "step": 42
    },
    {
      "epoch": 0.2835164835164835,
      "grad_norm": 0.17334218323230743,
      "learning_rate": 0.0001835920177383592,
      "loss": 0.5344,
      "step": 43
    },
    {
      "epoch": 0.29010989010989013,
      "grad_norm": 0.18245984613895416,
      "learning_rate": 0.00018314855875831487,
      "loss": 0.5802,
      "step": 44
    },
    {
      "epoch": 0.2967032967032967,
      "grad_norm": 0.20091219246387482,
      "learning_rate": 0.00018270509977827052,
      "loss": 0.6782,
      "step": 45
    },
    {
      "epoch": 0.3032967032967033,
      "grad_norm": 0.20907099545001984,
      "learning_rate": 0.00018226164079822616,
      "loss": 0.6289,
      "step": 46
    },
    {
      "epoch": 0.3098901098901099,
      "grad_norm": 0.2046046257019043,
      "learning_rate": 0.00018181818181818183,
      "loss": 0.5809,
      "step": 47
    },
    {
      "epoch": 0.31648351648351647,
      "grad_norm": 0.1860591471195221,
      "learning_rate": 0.00018137472283813748,
      "loss": 0.5933,
      "step": 48
    },
    {
      "epoch": 0.3230769230769231,
      "grad_norm": 0.19137340784072876,
      "learning_rate": 0.00018093126385809312,
      "loss": 0.5735,
      "step": 49
    },
    {
      "epoch": 0.32967032967032966,
      "grad_norm": 0.17774581909179688,
      "learning_rate": 0.0001804878048780488,
      "loss": 0.4735,
      "step": 50
    },
    {
      "epoch": 0.3362637362637363,
      "grad_norm": 0.18080784380435944,
      "learning_rate": 0.00018004434589800444,
      "loss": 0.4651,
      "step": 51
    },
    {
      "epoch": 0.34285714285714286,
      "grad_norm": 0.16998295485973358,
      "learning_rate": 0.00017960088691796008,
      "loss": 0.5958,
      "step": 52
    },
    {
      "epoch": 0.34945054945054943,
      "grad_norm": 0.16915321350097656,
      "learning_rate": 0.00017915742793791575,
      "loss": 0.5615,
      "step": 53
    },
    {
      "epoch": 0.35604395604395606,
      "grad_norm": 0.20877854526042938,
      "learning_rate": 0.0001787139689578714,
      "loss": 0.6373,
      "step": 54
    },
    {
      "epoch": 0.3626373626373626,
      "grad_norm": 0.19421787559986115,
      "learning_rate": 0.00017827050997782707,
      "loss": 0.6046,
      "step": 55
    },
    {
      "epoch": 0.36923076923076925,
      "grad_norm": 0.18675541877746582,
      "learning_rate": 0.00017782705099778271,
      "loss": 0.5858,
      "step": 56
    },
    {
      "epoch": 0.3758241758241758,
      "grad_norm": 0.197981059551239,
      "learning_rate": 0.00017738359201773839,
      "loss": 0.5122,
      "step": 57
    },
    {
      "epoch": 0.3824175824175824,
      "grad_norm": 0.20023414492607117,
      "learning_rate": 0.000176940133037694,
      "loss": 0.6593,
      "step": 58
    },
    {
      "epoch": 0.389010989010989,
      "grad_norm": 0.1908295899629593,
      "learning_rate": 0.00017649667405764967,
      "loss": 0.6208,
      "step": 59
    },
    {
      "epoch": 0.3956043956043956,
      "grad_norm": 0.18750755488872528,
      "learning_rate": 0.00017605321507760532,
      "loss": 0.5986,
      "step": 60
    },
    {
      "epoch": 0.4021978021978022,
      "grad_norm": 0.20169922709465027,
      "learning_rate": 0.000175609756097561,
      "loss": 0.6745,
      "step": 61
    },
    {
      "epoch": 0.4087912087912088,
      "grad_norm": 0.18614576756954193,
      "learning_rate": 0.00017516629711751663,
      "loss": 0.5526,
      "step": 62
    },
    {
      "epoch": 0.4153846153846154,
      "grad_norm": 0.2054520845413208,
      "learning_rate": 0.0001747228381374723,
      "loss": 0.6307,
      "step": 63
    },
    {
      "epoch": 0.421978021978022,
      "grad_norm": 0.20815221965312958,
      "learning_rate": 0.00017427937915742792,
      "loss": 0.599,
      "step": 64
    },
    {
      "epoch": 0.42857142857142855,
      "grad_norm": 0.21276892721652985,
      "learning_rate": 0.0001738359201773836,
      "loss": 0.6557,
      "step": 65
    },
    {
      "epoch": 0.4351648351648352,
      "grad_norm": 0.19046878814697266,
      "learning_rate": 0.00017339246119733924,
      "loss": 0.6373,
      "step": 66
    },
    {
      "epoch": 0.44175824175824174,
      "grad_norm": 0.2046242356300354,
      "learning_rate": 0.0001729490022172949,
      "loss": 0.6064,
      "step": 67
    },
    {
      "epoch": 0.44835164835164837,
      "grad_norm": 0.19272306561470032,
      "learning_rate": 0.00017250554323725056,
      "loss": 0.5962,
      "step": 68
    },
    {
      "epoch": 0.45494505494505494,
      "grad_norm": 0.18096354603767395,
      "learning_rate": 0.00017206208425720623,
      "loss": 0.6135,
      "step": 69
    },
    {
      "epoch": 0.46153846153846156,
      "grad_norm": 0.18638895452022552,
      "learning_rate": 0.00017161862527716187,
      "loss": 0.5805,
      "step": 70
    },
    {
      "epoch": 0.46813186813186813,
      "grad_norm": 0.19761188328266144,
      "learning_rate": 0.00017117516629711752,
      "loss": 0.5362,
      "step": 71
    },
    {
      "epoch": 0.4747252747252747,
      "grad_norm": 0.19810283184051514,
      "learning_rate": 0.0001707317073170732,
      "loss": 0.6037,
      "step": 72
    },
    {
      "epoch": 0.48131868131868133,
      "grad_norm": 0.20009560883045197,
      "learning_rate": 0.00017028824833702883,
      "loss": 0.5641,
      "step": 73
    },
    {
      "epoch": 0.4879120879120879,
      "grad_norm": 0.18263758718967438,
      "learning_rate": 0.0001698447893569845,
      "loss": 0.5701,
      "step": 74
    },
    {
      "epoch": 0.4945054945054945,
      "grad_norm": 0.1989409327507019,
      "learning_rate": 0.00016940133037694015,
      "loss": 0.5227,
      "step": 75
    },
    {
      "epoch": 0.5010989010989011,
      "grad_norm": 0.18149833381175995,
      "learning_rate": 0.00016895787139689582,
      "loss": 0.5354,
      "step": 76
    },
    {
      "epoch": 0.5076923076923077,
      "grad_norm": 0.19985748827457428,
      "learning_rate": 0.00016851441241685144,
      "loss": 0.5741,
      "step": 77
    },
    {
      "epoch": 0.5142857142857142,
      "grad_norm": 0.2370215803384781,
      "learning_rate": 0.0001680709534368071,
      "loss": 0.6566,
      "step": 78
    },
    {
      "epoch": 0.5208791208791209,
      "grad_norm": 0.20534105598926544,
      "learning_rate": 0.00016762749445676275,
      "loss": 0.5815,
      "step": 79
    },
    {
      "epoch": 0.5274725274725275,
      "grad_norm": 0.21887195110321045,
      "learning_rate": 0.00016718403547671842,
      "loss": 0.5734,
      "step": 80
    },
    {
      "epoch": 0.5340659340659341,
      "grad_norm": 0.20859766006469727,
      "learning_rate": 0.00016674057649667407,
      "loss": 0.5537,
      "step": 81
    },
    {
      "epoch": 0.5406593406593406,
      "grad_norm": 0.20775769650936127,
      "learning_rate": 0.00016629711751662974,
      "loss": 0.5972,
      "step": 82
    },
    {
      "epoch": 0.5472527472527473,
      "grad_norm": 0.2022627741098404,
      "learning_rate": 0.00016585365853658536,
      "loss": 0.5717,
      "step": 83
    },
    {
      "epoch": 0.5538461538461539,
      "grad_norm": 0.1799938827753067,
      "learning_rate": 0.00016541019955654103,
      "loss": 0.5577,
      "step": 84
    },
    {
      "epoch": 0.5604395604395604,
      "grad_norm": 0.15892469882965088,
      "learning_rate": 0.00016496674057649667,
      "loss": 0.54,
      "step": 85
    },
    {
      "epoch": 0.567032967032967,
      "grad_norm": 0.2420973926782608,
      "learning_rate": 0.00016452328159645234,
      "loss": 0.5373,
      "step": 86
    },
    {
      "epoch": 0.5736263736263736,
      "grad_norm": 0.20307506620883942,
      "learning_rate": 0.000164079822616408,
      "loss": 0.6048,
      "step": 87
    },
    {
      "epoch": 0.5802197802197803,
      "grad_norm": 0.18105755746364594,
      "learning_rate": 0.00016363636363636366,
      "loss": 0.6019,
      "step": 88
    },
    {
      "epoch": 0.5868131868131868,
      "grad_norm": 0.1656375676393509,
      "learning_rate": 0.0001631929046563193,
      "loss": 0.6014,
      "step": 89
    },
    {
      "epoch": 0.5934065934065934,
      "grad_norm": 0.22959865629673004,
      "learning_rate": 0.00016274944567627495,
      "loss": 0.6606,
      "step": 90
    },
    {
      "epoch": 0.6,
      "grad_norm": 0.1757660061120987,
      "learning_rate": 0.0001623059866962306,
      "loss": 0.5918,
      "step": 91
    },
    {
      "epoch": 0.6065934065934065,
      "grad_norm": 0.19297438859939575,
      "learning_rate": 0.00016186252771618626,
      "loss": 0.5693,
      "step": 92
    },
    {
      "epoch": 0.6131868131868132,
      "grad_norm": 0.201291024684906,
      "learning_rate": 0.0001614190687361419,
      "loss": 0.6721,
      "step": 93
    },
    {
      "epoch": 0.6197802197802198,
      "grad_norm": 0.19932299852371216,
      "learning_rate": 0.00016097560975609758,
      "loss": 0.547,
      "step": 94
    },
    {
      "epoch": 0.6263736263736264,
      "grad_norm": 0.19901031255722046,
      "learning_rate": 0.00016053215077605322,
      "loss": 0.5778,
      "step": 95
    },
    {
      "epoch": 0.6329670329670329,
      "grad_norm": 0.21911878883838654,
      "learning_rate": 0.00016008869179600887,
      "loss": 0.5779,
      "step": 96
    },
    {
      "epoch": 0.6395604395604395,
      "grad_norm": 0.18315036594867706,
      "learning_rate": 0.00015964523281596454,
      "loss": 0.5338,
      "step": 97
    },
    {
      "epoch": 0.6461538461538462,
      "grad_norm": 0.2002074420452118,
      "learning_rate": 0.00015920177383592018,
      "loss": 0.6329,
      "step": 98
    },
    {
      "epoch": 0.6527472527472528,
      "grad_norm": 0.19552098214626312,
      "learning_rate": 0.00015875831485587586,
      "loss": 0.6187,
      "step": 99
    },
    {
      "epoch": 0.6593406593406593,
      "grad_norm": 0.19236156344413757,
      "learning_rate": 0.0001583148558758315,
      "loss": 0.5954,
      "step": 100
    },
    {
      "epoch": 0.6659340659340659,
      "grad_norm": 0.20366628468036652,
      "learning_rate": 0.00015787139689578714,
      "loss": 0.4912,
      "step": 101
    },
    {
      "epoch": 0.6725274725274726,
      "grad_norm": 0.17797768115997314,
      "learning_rate": 0.0001574279379157428,
      "loss": 0.5333,
      "step": 102
    },
    {
      "epoch": 0.6791208791208792,
      "grad_norm": 0.20740211009979248,
      "learning_rate": 0.00015698447893569846,
      "loss": 0.5482,
      "step": 103
    },
    {
      "epoch": 0.6857142857142857,
      "grad_norm": 0.1916942596435547,
      "learning_rate": 0.0001565410199556541,
      "loss": 0.5307,
      "step": 104
    },
    {
      "epoch": 0.6923076923076923,
      "grad_norm": 0.21760819852352142,
      "learning_rate": 0.00015609756097560978,
      "loss": 0.5656,
      "step": 105
    },
    {
      "epoch": 0.6989010989010989,
      "grad_norm": 0.19037440419197083,
      "learning_rate": 0.00015565410199556542,
      "loss": 0.5681,
      "step": 106
    },
    {
      "epoch": 0.7054945054945055,
      "grad_norm": 0.19906817376613617,
      "learning_rate": 0.00015521064301552106,
      "loss": 0.6516,
      "step": 107
    },
    {
      "epoch": 0.7120879120879121,
      "grad_norm": 0.20514219999313354,
      "learning_rate": 0.0001547671840354767,
      "loss": 0.5991,
      "step": 108
    },
    {
      "epoch": 0.7186813186813187,
      "grad_norm": 0.1973261535167694,
      "learning_rate": 0.00015432372505543238,
      "loss": 0.5141,
      "step": 109
    },
    {
      "epoch": 0.7252747252747253,
      "grad_norm": 0.2035258412361145,
      "learning_rate": 0.00015388026607538802,
      "loss": 0.6216,
      "step": 110
    },
    {
      "epoch": 0.7318681318681318,
      "grad_norm": 0.1799962818622589,
      "learning_rate": 0.0001534368070953437,
      "loss": 0.5462,
      "step": 111
    },
    {
      "epoch": 0.7384615384615385,
      "grad_norm": 0.17428919672966003,
      "learning_rate": 0.00015299334811529934,
      "loss": 0.5401,
      "step": 112
    },
    {
      "epoch": 0.7450549450549451,
      "grad_norm": 0.1946234405040741,
      "learning_rate": 0.00015254988913525498,
      "loss": 0.5302,
      "step": 113
    },
    {
      "epoch": 0.7516483516483516,
      "grad_norm": 0.19590261578559875,
      "learning_rate": 0.00015210643015521066,
      "loss": 0.6181,
      "step": 114
    },
    {
      "epoch": 0.7582417582417582,
      "grad_norm": 0.21549439430236816,
      "learning_rate": 0.0001516629711751663,
      "loss": 0.6081,
      "step": 115
    },
    {
      "epoch": 0.7648351648351648,
      "grad_norm": 0.21377599239349365,
      "learning_rate": 0.00015121951219512197,
      "loss": 0.6327,
      "step": 116
    },
    {
      "epoch": 0.7714285714285715,
      "grad_norm": 0.19143231213092804,
      "learning_rate": 0.00015077605321507762,
      "loss": 0.648,
      "step": 117
    },
    {
      "epoch": 0.778021978021978,
      "grad_norm": 0.20007622241973877,
      "learning_rate": 0.0001503325942350333,
      "loss": 0.5642,
      "step": 118
    },
    {
      "epoch": 0.7846153846153846,
      "grad_norm": 0.20515254139900208,
      "learning_rate": 0.0001498891352549889,
      "loss": 0.6323,
      "step": 119
    },
    {
      "epoch": 0.7912087912087912,
      "grad_norm": 0.18151244521141052,
      "learning_rate": 0.00014944567627494458,
      "loss": 0.5582,
      "step": 120
    },
    {
      "epoch": 0.7978021978021979,
      "grad_norm": 0.1993722766637802,
      "learning_rate": 0.00014900221729490022,
      "loss": 0.5812,
      "step": 121
    },
    {
      "epoch": 0.8043956043956044,
      "grad_norm": 0.22069858014583588,
      "learning_rate": 0.0001485587583148559,
      "loss": 0.5814,
      "step": 122
    },
    {
      "epoch": 0.810989010989011,
      "grad_norm": 0.22465714812278748,
      "learning_rate": 0.00014811529933481154,
      "loss": 0.5655,
      "step": 123
    },
    {
      "epoch": 0.8175824175824176,
      "grad_norm": 0.1979009062051773,
      "learning_rate": 0.0001476718403547672,
      "loss": 0.665,
      "step": 124
    },
    {
      "epoch": 0.8241758241758241,
      "grad_norm": 0.21059945225715637,
      "learning_rate": 0.00014722838137472282,
      "loss": 0.6013,
      "step": 125
    },
    {
      "epoch": 0.8307692307692308,
      "grad_norm": 0.19477960467338562,
      "learning_rate": 0.0001467849223946785,
      "loss": 0.5173,
      "step": 126
    },
    {
      "epoch": 0.8373626373626374,
      "grad_norm": 0.2379969358444214,
      "learning_rate": 0.00014634146341463414,
      "loss": 0.6315,
      "step": 127
    },
    {
      "epoch": 0.843956043956044,
      "grad_norm": 0.2084502875804901,
      "learning_rate": 0.0001458980044345898,
      "loss": 0.5454,
      "step": 128
    },
    {
      "epoch": 0.8505494505494505,
      "grad_norm": 0.21561095118522644,
      "learning_rate": 0.00014545454545454546,
      "loss": 0.5782,
      "step": 129
    },
    {
      "epoch": 0.8571428571428571,
      "grad_norm": 0.2167816311120987,
      "learning_rate": 0.00014501108647450113,
      "loss": 0.5507,
      "step": 130
    },
    {
      "epoch": 0.8637362637362638,
      "grad_norm": 0.20814435184001923,
      "learning_rate": 0.00014456762749445675,
      "loss": 0.5465,
      "step": 131
    },
    {
      "epoch": 0.8703296703296703,
      "grad_norm": 0.21132875978946686,
      "learning_rate": 0.00014412416851441242,
      "loss": 0.5413,
      "step": 132
    },
    {
      "epoch": 0.8769230769230769,
      "grad_norm": 0.1970619410276413,
      "learning_rate": 0.00014368070953436806,
      "loss": 0.6501,
      "step": 133
    },
    {
      "epoch": 0.8835164835164835,
      "grad_norm": 0.2053031474351883,
      "learning_rate": 0.00014323725055432373,
      "loss": 0.5629,
      "step": 134
    },
    {
      "epoch": 0.8901098901098901,
      "grad_norm": 0.20264855027198792,
      "learning_rate": 0.00014279379157427938,
      "loss": 0.6491,
      "step": 135
    },
    {
      "epoch": 0.8967032967032967,
      "grad_norm": 0.19277967512607574,
      "learning_rate": 0.00014235033259423505,
      "loss": 0.5228,
      "step": 136
    },
    {
      "epoch": 0.9032967032967033,
      "grad_norm": 0.2253330647945404,
      "learning_rate": 0.0001419068736141907,
      "loss": 0.6045,
      "step": 137
    },
    {
      "epoch": 0.9098901098901099,
      "grad_norm": 0.18566767871379852,
      "learning_rate": 0.00014146341463414634,
      "loss": 0.6159,
      "step": 138
    },
    {
      "epoch": 0.9164835164835164,
      "grad_norm": 0.20604389905929565,
      "learning_rate": 0.000141019955654102,
      "loss": 0.6079,
      "step": 139
    },
    {
      "epoch": 0.9230769230769231,
      "grad_norm": 0.19309717416763306,
      "learning_rate": 0.00014057649667405765,
      "loss": 0.5808,
      "step": 140
    },
    {
      "epoch": 0.9296703296703297,
      "grad_norm": 0.18021467328071594,
      "learning_rate": 0.00014013303769401332,
      "loss": 0.6203,
      "step": 141
    },
    {
      "epoch": 0.9362637362637363,
      "grad_norm": 0.20142363011837006,
      "learning_rate": 0.00013968957871396897,
      "loss": 0.558,
      "step": 142
    },
    {
      "epoch": 0.9428571428571428,
      "grad_norm": 0.1704377382993698,
      "learning_rate": 0.00013924611973392464,
      "loss": 0.5199,
      "step": 143
    },
    {
      "epoch": 0.9494505494505494,
      "grad_norm": 0.19239981472492218,
      "learning_rate": 0.00013880266075388026,
      "loss": 0.6452,
      "step": 144
    },
    {
      "epoch": 0.9560439560439561,
      "grad_norm": 0.21208760142326355,
      "learning_rate": 0.00013835920177383593,
      "loss": 0.586,
      "step": 145
    },
    {
      "epoch": 0.9626373626373627,
      "grad_norm": 0.21088305115699768,
      "learning_rate": 0.00013791574279379157,
      "loss": 0.6415,
      "step": 146
    },
    {
      "epoch": 0.9692307692307692,
      "grad_norm": 0.19738735258579254,
      "learning_rate": 0.00013747228381374724,
      "loss": 0.6011,
      "step": 147
    },
    {
      "epoch": 0.9758241758241758,
      "grad_norm": 0.22359523177146912,
      "learning_rate": 0.0001370288248337029,
      "loss": 0.5396,
      "step": 148
    },
    {
      "epoch": 0.9824175824175824,
      "grad_norm": 0.202518031001091,
      "learning_rate": 0.00013658536585365856,
      "loss": 0.5611,
      "step": 149
    },
    {
      "epoch": 0.989010989010989,
      "grad_norm": 0.208979070186615,
      "learning_rate": 0.00013614190687361418,
      "loss": 0.6149,
      "step": 150
    },
    {
      "epoch": 0.9956043956043956,
      "grad_norm": 0.20856443047523499,
      "learning_rate": 0.00013569844789356985,
      "loss": 0.6232,
      "step": 151
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.2518608272075653,
      "learning_rate": 0.0001352549889135255,
      "loss": 0.6112,
      "step": 152
    },
    {
      "epoch": 1.0065934065934066,
      "grad_norm": 0.19041535258293152,
      "learning_rate": 0.00013481152993348116,
      "loss": 0.6114,
      "step": 153
    },
    {
      "epoch": 1.0131868131868131,
      "grad_norm": 0.19032245874404907,
      "learning_rate": 0.0001343680709534368,
      "loss": 0.6139,
      "step": 154
    },
    {
      "epoch": 1.0197802197802197,
      "grad_norm": 0.19202658534049988,
      "learning_rate": 0.00013392461197339248,
      "loss": 0.5161,
      "step": 155
    },
    {
      "epoch": 1.0263736263736263,
      "grad_norm": 0.19449295103549957,
      "learning_rate": 0.00013348115299334812,
      "loss": 0.5353,
      "step": 156
    },
    {
      "epoch": 1.032967032967033,
      "grad_norm": 0.17760977149009705,
      "learning_rate": 0.00013303769401330377,
      "loss": 0.5781,
      "step": 157
    },
    {
      "epoch": 1.0395604395604396,
      "grad_norm": 0.17056520283222198,
      "learning_rate": 0.00013259423503325944,
      "loss": 0.551,
      "step": 158
    },
    {
      "epoch": 1.0461538461538462,
      "grad_norm": 0.22371503710746765,
      "learning_rate": 0.00013215077605321509,
      "loss": 0.6728,
      "step": 159
    },
    {
      "epoch": 1.0527472527472528,
      "grad_norm": 0.20783597230911255,
      "learning_rate": 0.00013170731707317076,
      "loss": 0.5807,
      "step": 160
    },
    {
      "epoch": 1.0593406593406594,
      "grad_norm": 0.19965267181396484,
      "learning_rate": 0.0001312638580931264,
      "loss": 0.574,
      "step": 161
    },
    {
      "epoch": 1.065934065934066,
      "grad_norm": 0.24759921431541443,
      "learning_rate": 0.00013082039911308205,
      "loss": 0.6132,
      "step": 162
    },
    {
      "epoch": 1.0725274725274725,
      "grad_norm": 0.20820704102516174,
      "learning_rate": 0.0001303769401330377,
      "loss": 0.559,
      "step": 163
    },
    {
      "epoch": 1.079120879120879,
      "grad_norm": 0.20800600945949554,
      "learning_rate": 0.00012993348115299336,
      "loss": 0.5324,
      "step": 164
    },
    {
      "epoch": 1.0857142857142856,
      "grad_norm": 0.19648154079914093,
      "learning_rate": 0.000129490022172949,
      "loss": 0.5294,
      "step": 165
    },
    {
      "epoch": 1.0923076923076924,
      "grad_norm": 0.21711808443069458,
      "learning_rate": 0.00012904656319290468,
      "loss": 0.542,
      "step": 166
    },
    {
      "epoch": 1.098901098901099,
      "grad_norm": 0.1998845785856247,
      "learning_rate": 0.00012860310421286032,
      "loss": 0.5932,
      "step": 167
    },
    {
      "epoch": 1.1054945054945056,
      "grad_norm": 0.18798790872097015,
      "learning_rate": 0.00012815964523281597,
      "loss": 0.5556,
      "step": 168
    },
    {
      "epoch": 1.1120879120879121,
      "grad_norm": 0.18659773468971252,
      "learning_rate": 0.0001277161862527716,
      "loss": 0.5359,
      "step": 169
    },
    {
      "epoch": 1.1186813186813187,
      "grad_norm": 0.21169960498809814,
      "learning_rate": 0.00012727272727272728,
      "loss": 0.5069,
      "step": 170
    },
    {
      "epoch": 1.1252747252747253,
      "grad_norm": 0.19881480932235718,
      "learning_rate": 0.00012682926829268293,
      "loss": 0.5862,
      "step": 171
    },
    {
      "epoch": 1.1318681318681318,
      "grad_norm": 0.2178238034248352,
      "learning_rate": 0.0001263858093126386,
      "loss": 0.5985,
      "step": 172
    },
    {
      "epoch": 1.1384615384615384,
      "grad_norm": 0.20887236297130585,
      "learning_rate": 0.00012594235033259424,
      "loss": 0.587,
      "step": 173
    },
    {
      "epoch": 1.145054945054945,
      "grad_norm": 0.2046881765127182,
      "learning_rate": 0.00012549889135254989,
      "loss": 0.5798,
      "step": 174
    },
    {
      "epoch": 1.1516483516483516,
      "grad_norm": 0.22640396654605865,
      "learning_rate": 0.00012505543237250553,
      "loss": 0.6371,
      "step": 175
    },
    {
      "epoch": 1.1582417582417581,
      "grad_norm": 0.22060760855674744,
      "learning_rate": 0.0001246119733924612,
      "loss": 0.5894,
      "step": 176
    },
    {
      "epoch": 1.164835164835165,
      "grad_norm": 0.24441444873809814,
      "learning_rate": 0.00012416851441241685,
      "loss": 0.5472,
      "step": 177
    },
    {
      "epoch": 1.1714285714285715,
      "grad_norm": 0.23831522464752197,
      "learning_rate": 0.00012372505543237252,
      "loss": 0.5634,
      "step": 178
    },
    {
      "epoch": 1.178021978021978,
      "grad_norm": 0.22292114794254303,
      "learning_rate": 0.00012328159645232816,
      "loss": 0.5541,
      "step": 179
    },
    {
      "epoch": 1.1846153846153846,
      "grad_norm": 0.2267928421497345,
      "learning_rate": 0.0001228381374722838,
      "loss": 0.593,
      "step": 180
    },
    {
      "epoch": 1.1912087912087912,
      "grad_norm": 0.22938065230846405,
      "learning_rate": 0.00012239467849223948,
      "loss": 0.4905,
      "step": 181
    },
    {
      "epoch": 1.1978021978021978,
      "grad_norm": 0.25057607889175415,
      "learning_rate": 0.00012195121951219512,
      "loss": 0.5709,
      "step": 182
    },
    {
      "epoch": 1.2043956043956043,
      "grad_norm": 0.23950348794460297,
      "learning_rate": 0.00012150776053215078,
      "loss": 0.5048,
      "step": 183
    },
    {
      "epoch": 1.210989010989011,
      "grad_norm": 0.23767712712287903,
      "learning_rate": 0.00012106430155210644,
      "loss": 0.6033,
      "step": 184
    },
    {
      "epoch": 1.2175824175824177,
      "grad_norm": 0.24015109241008759,
      "learning_rate": 0.0001206208425720621,
      "loss": 0.6378,
      "step": 185
    },
    {
      "epoch": 1.2241758241758243,
      "grad_norm": 0.2524959146976471,
      "learning_rate": 0.00012017738359201774,
      "loss": 0.5245,
      "step": 186
    },
    {
      "epoch": 1.2307692307692308,
      "grad_norm": 0.2228759527206421,
      "learning_rate": 0.0001197339246119734,
      "loss": 0.525,
      "step": 187
    },
    {
      "epoch": 1.2373626373626374,
      "grad_norm": 0.22547198832035065,
      "learning_rate": 0.00011929046563192906,
      "loss": 0.5857,
      "step": 188
    },
    {
      "epoch": 1.243956043956044,
      "grad_norm": 0.2189064919948578,
      "learning_rate": 0.00011884700665188471,
      "loss": 0.5362,
      "step": 189
    },
    {
      "epoch": 1.2505494505494505,
      "grad_norm": 0.23731443285942078,
      "learning_rate": 0.00011840354767184036,
      "loss": 0.5029,
      "step": 190
    },
    {
      "epoch": 1.2571428571428571,
      "grad_norm": 0.2407902181148529,
      "learning_rate": 0.00011796008869179602,
      "loss": 0.5564,
      "step": 191
    },
    {
      "epoch": 1.2637362637362637,
      "grad_norm": 0.23271149396896362,
      "learning_rate": 0.00011751662971175166,
      "loss": 0.5396,
      "step": 192
    },
    {
      "epoch": 1.2703296703296703,
      "grad_norm": 0.24813254177570343,
      "learning_rate": 0.00011707317073170732,
      "loss": 0.5528,
      "step": 193
    },
    {
      "epoch": 1.2769230769230768,
      "grad_norm": 0.22270256280899048,
      "learning_rate": 0.00011662971175166298,
      "loss": 0.513,
      "step": 194
    },
    {
      "epoch": 1.2835164835164834,
      "grad_norm": 0.2717154622077942,
      "learning_rate": 0.00011618625277161863,
      "loss": 0.5382,
      "step": 195
    },
    {
      "epoch": 1.2901098901098902,
      "grad_norm": 0.23264969885349274,
      "learning_rate": 0.00011574279379157429,
      "loss": 0.5423,
      "step": 196
    },
    {
      "epoch": 1.2967032967032968,
      "grad_norm": 0.26586130261421204,
      "learning_rate": 0.00011529933481152995,
      "loss": 0.5519,
      "step": 197
    },
    {
      "epoch": 1.3032967032967033,
      "grad_norm": 0.2638515532016754,
      "learning_rate": 0.00011485587583148558,
      "loss": 0.5129,
      "step": 198
    },
    {
      "epoch": 1.30989010989011,
      "grad_norm": 0.29879090189933777,
      "learning_rate": 0.00011441241685144124,
      "loss": 0.5341,
      "step": 199
    },
    {
      "epoch": 1.3164835164835165,
      "grad_norm": 0.27044677734375,
      "learning_rate": 0.0001139689578713969,
      "loss": 0.5217,
      "step": 200
    },
    {
      "epoch": 1.323076923076923,
      "grad_norm": 0.28926023840904236,
      "learning_rate": 0.00011352549889135255,
      "loss": 0.5893,
      "step": 201
    },
    {
      "epoch": 1.3296703296703296,
      "grad_norm": 0.2590689957141876,
      "learning_rate": 0.00011308203991130821,
      "loss": 0.5879,
      "step": 202
    },
    {
      "epoch": 1.3362637362637364,
      "grad_norm": 0.2358494997024536,
      "learning_rate": 0.00011263858093126387,
      "loss": 0.4914,
      "step": 203
    },
    {
      "epoch": 1.342857142857143,
      "grad_norm": 0.25238221883773804,
      "learning_rate": 0.00011219512195121953,
      "loss": 0.4975,
      "step": 204
    },
    {
      "epoch": 1.3494505494505495,
      "grad_norm": 0.2814512252807617,
      "learning_rate": 0.00011175166297117516,
      "loss": 0.587,
      "step": 205
    },
    {
      "epoch": 1.3560439560439561,
      "grad_norm": 0.23827160894870758,
      "learning_rate": 0.00011130820399113082,
      "loss": 0.553,
      "step": 206
    },
    {
      "epoch": 1.3626373626373627,
      "grad_norm": 0.25040608644485474,
      "learning_rate": 0.00011086474501108647,
      "loss": 0.5724,
      "step": 207
    },
    {
      "epoch": 1.3692307692307693,
      "grad_norm": 0.23351642489433289,
      "learning_rate": 0.00011042128603104213,
      "loss": 0.4819,
      "step": 208
    },
    {
      "epoch": 1.3758241758241758,
      "grad_norm": 0.2473171055316925,
      "learning_rate": 0.00010997782705099779,
      "loss": 0.5327,
      "step": 209
    },
    {
      "epoch": 1.3824175824175824,
      "grad_norm": 0.2546323537826538,
      "learning_rate": 0.00010953436807095345,
      "loss": 0.5328,
      "step": 210
    },
    {
      "epoch": 1.389010989010989,
      "grad_norm": 0.2620166838169098,
      "learning_rate": 0.00010909090909090909,
      "loss": 0.6299,
      "step": 211
    },
    {
      "epoch": 1.3956043956043955,
      "grad_norm": 0.2592707872390747,
      "learning_rate": 0.00010864745011086475,
      "loss": 0.5123,
      "step": 212
    },
    {
      "epoch": 1.402197802197802,
      "grad_norm": 0.2427063286304474,
      "learning_rate": 0.00010820399113082041,
      "loss": 0.5527,
      "step": 213
    },
    {
      "epoch": 1.4087912087912087,
      "grad_norm": 0.26192405819892883,
      "learning_rate": 0.00010776053215077607,
      "loss": 0.6507,
      "step": 214
    },
    {
      "epoch": 1.4153846153846155,
      "grad_norm": 0.24521975219249725,
      "learning_rate": 0.00010731707317073172,
      "loss": 0.481,
      "step": 215
    },
    {
      "epoch": 1.421978021978022,
      "grad_norm": 0.2793915271759033,
      "learning_rate": 0.00010687361419068738,
      "loss": 0.5946,
      "step": 216
    },
    {
      "epoch": 1.4285714285714286,
      "grad_norm": 0.25919869542121887,
      "learning_rate": 0.00010643015521064301,
      "loss": 0.5686,
      "step": 217
    },
    {
      "epoch": 1.4351648351648352,
      "grad_norm": 0.24244147539138794,
      "learning_rate": 0.00010598669623059867,
      "loss": 0.5271,
      "step": 218
    },
    {
      "epoch": 1.4417582417582417,
      "grad_norm": 0.2661493122577667,
      "learning_rate": 0.00010554323725055433,
      "loss": 0.554,
      "step": 219
    },
    {
      "epoch": 1.4483516483516483,
      "grad_norm": 0.25211963057518005,
      "learning_rate": 0.00010509977827050999,
      "loss": 0.5353,
      "step": 220
    },
    {
      "epoch": 1.4549450549450549,
      "grad_norm": 0.27057012915611267,
      "learning_rate": 0.00010465631929046564,
      "loss": 0.5511,
      "step": 221
    },
    {
      "epoch": 1.4615384615384617,
      "grad_norm": 0.256792813539505,
      "learning_rate": 0.0001042128603104213,
      "loss": 0.545,
      "step": 222
    },
    {
      "epoch": 1.4681318681318682,
      "grad_norm": 0.2717509865760803,
      "learning_rate": 0.00010376940133037693,
      "loss": 0.5877,
      "step": 223
    },
    {
      "epoch": 1.4747252747252748,
      "grad_norm": 0.2530151605606079,
      "learning_rate": 0.00010332594235033259,
      "loss": 0.5627,
      "step": 224
    },
    {
      "epoch": 1.4813186813186814,
      "grad_norm": 0.2564440071582794,
      "learning_rate": 0.00010288248337028825,
      "loss": 0.4988,
      "step": 225
    },
    {
      "epoch": 1.487912087912088,
      "grad_norm": 0.2507382333278656,
      "learning_rate": 0.0001024390243902439,
      "loss": 0.5769,
      "step": 226
    },
    {
      "epoch": 1.4945054945054945,
      "grad_norm": 0.23935091495513916,
      "learning_rate": 0.00010199556541019956,
      "loss": 0.5913,
      "step": 227
    },
    {
      "epoch": 1.501098901098901,
      "grad_norm": 0.24732615053653717,
      "learning_rate": 0.00010155210643015522,
      "loss": 0.5503,
      "step": 228
    },
    {
      "epoch": 1.5076923076923077,
      "grad_norm": 0.29727789759635925,
      "learning_rate": 0.00010110864745011087,
      "loss": 0.6148,
      "step": 229
    },
    {
      "epoch": 1.5142857142857142,
      "grad_norm": 0.22885286808013916,
      "learning_rate": 0.00010066518847006652,
      "loss": 0.5338,
      "step": 230
    },
    {
      "epoch": 1.5208791208791208,
      "grad_norm": 0.23787544667720795,
      "learning_rate": 0.00010022172949002218,
      "loss": 0.4716,
      "step": 231
    },
    {
      "epoch": 1.5274725274725274,
      "grad_norm": 0.22866325080394745,
      "learning_rate": 9.977827050997783e-05,
      "loss": 0.5416,
      "step": 232
    },
    {
      "epoch": 1.534065934065934,
      "grad_norm": 0.25525131821632385,
      "learning_rate": 9.933481152993348e-05,
      "loss": 0.5804,
      "step": 233
    },
    {
      "epoch": 1.5406593406593405,
      "grad_norm": 0.26150748133659363,
      "learning_rate": 9.889135254988914e-05,
      "loss": 0.5684,
      "step": 234
    },
    {
      "epoch": 1.5472527472527473,
      "grad_norm": 0.23591382801532745,
      "learning_rate": 9.84478935698448e-05,
      "loss": 0.5718,
      "step": 235
    },
    {
      "epoch": 1.5538461538461539,
      "grad_norm": 0.23938985168933868,
      "learning_rate": 9.800443458980046e-05,
      "loss": 0.641,
      "step": 236
    },
    {
      "epoch": 1.5604395604395604,
      "grad_norm": 0.24983075261116028,
      "learning_rate": 9.75609756097561e-05,
      "loss": 0.5362,
      "step": 237
    },
    {
      "epoch": 1.567032967032967,
      "grad_norm": 0.2679073214530945,
      "learning_rate": 9.711751662971176e-05,
      "loss": 0.5194,
      "step": 238
    },
    {
      "epoch": 1.5736263736263736,
      "grad_norm": 0.25713440775871277,
      "learning_rate": 9.667405764966742e-05,
      "loss": 0.6079,
      "step": 239
    },
    {
      "epoch": 1.5802197802197804,
      "grad_norm": 0.2716706693172455,
      "learning_rate": 9.623059866962306e-05,
      "loss": 0.55,
      "step": 240
    },
    {
      "epoch": 1.586813186813187,
      "grad_norm": 0.2506992816925049,
      "learning_rate": 9.578713968957872e-05,
      "loss": 0.4743,
      "step": 241
    },
    {
      "epoch": 1.5934065934065935,
      "grad_norm": 0.2671050429344177,
      "learning_rate": 9.534368070953438e-05,
      "loss": 0.6252,
      "step": 242
    },
    {
      "epoch": 1.6,
      "grad_norm": 0.24889801442623138,
      "learning_rate": 9.490022172949002e-05,
      "loss": 0.5859,
      "step": 243
    },
    {
      "epoch": 1.6065934065934067,
      "grad_norm": 0.2506890296936035,
      "learning_rate": 9.445676274944568e-05,
      "loss": 0.5369,
      "step": 244
    },
    {
      "epoch": 1.6131868131868132,
      "grad_norm": 0.2503523528575897,
      "learning_rate": 9.401330376940134e-05,
      "loss": 0.4974,
      "step": 245
    },
    {
      "epoch": 1.6197802197802198,
      "grad_norm": 0.23670326173305511,
      "learning_rate": 9.356984478935698e-05,
      "loss": 0.5457,
      "step": 246
    },
    {
      "epoch": 1.6263736263736264,
      "grad_norm": 0.2631224989891052,
      "learning_rate": 9.312638580931264e-05,
      "loss": 0.4962,
      "step": 247
    },
    {
      "epoch": 1.632967032967033,
      "grad_norm": 0.273565411567688,
      "learning_rate": 9.26829268292683e-05,
      "loss": 0.5554,
      "step": 248
    },
    {
      "epoch": 1.6395604395604395,
      "grad_norm": 0.2311021387577057,
      "learning_rate": 9.223946784922394e-05,
      "loss": 0.5629,
      "step": 249
    },
    {
      "epoch": 1.646153846153846,
      "grad_norm": 0.2513691782951355,
      "learning_rate": 9.17960088691796e-05,
      "loss": 0.574,
      "step": 250
    },
    {
      "epoch": 1.6527472527472526,
      "grad_norm": 0.23119987547397614,
      "learning_rate": 9.135254988913526e-05,
      "loss": 0.5356,
      "step": 251
    },
    {
      "epoch": 1.6593406593406592,
      "grad_norm": 0.25572118163108826,
      "learning_rate": 9.090909090909092e-05,
      "loss": 0.5817,
      "step": 252
    },
    {
      "epoch": 1.6659340659340658,
      "grad_norm": 0.2269141674041748,
      "learning_rate": 9.046563192904656e-05,
      "loss": 0.5671,
      "step": 253
    },
    {
      "epoch": 1.6725274725274726,
      "grad_norm": 0.24535146355628967,
      "learning_rate": 9.002217294900222e-05,
      "loss": 0.5797,
      "step": 254
    },
    {
      "epoch": 1.6791208791208792,
      "grad_norm": 0.23961450159549713,
      "learning_rate": 8.957871396895788e-05,
      "loss": 0.5728,
      "step": 255
    },
    {
      "epoch": 1.6857142857142857,
      "grad_norm": 0.22332210838794708,
      "learning_rate": 8.913525498891354e-05,
      "loss": 0.4827,
      "step": 256
    },
    {
      "epoch": 1.6923076923076923,
      "grad_norm": 0.25941914319992065,
      "learning_rate": 8.869179600886919e-05,
      "loss": 0.5238,
      "step": 257
    },
    {
      "epoch": 1.6989010989010989,
      "grad_norm": 0.20226909220218658,
      "learning_rate": 8.824833702882484e-05,
      "loss": 0.5208,
      "step": 258
    },
    {
      "epoch": 1.7054945054945057,
      "grad_norm": 0.22303876280784607,
      "learning_rate": 8.78048780487805e-05,
      "loss": 0.5185,
      "step": 259
    },
    {
      "epoch": 1.7120879120879122,
      "grad_norm": 0.2268039435148239,
      "learning_rate": 8.736141906873615e-05,
      "loss": 0.5678,
      "step": 260
    },
    {
      "epoch": 1.7186813186813188,
      "grad_norm": 0.2711758315563202,
      "learning_rate": 8.69179600886918e-05,
      "loss": 0.5524,
      "step": 261
    },
    {
      "epoch": 1.7252747252747254,
      "grad_norm": 0.25414571166038513,
      "learning_rate": 8.647450110864746e-05,
      "loss": 0.6547,
      "step": 262
    },
    {
      "epoch": 1.731868131868132,
      "grad_norm": 0.25818225741386414,
      "learning_rate": 8.603104212860311e-05,
      "loss": 0.4972,
      "step": 263
    },
    {
      "epoch": 1.7384615384615385,
      "grad_norm": 0.2654488980770111,
      "learning_rate": 8.558758314855876e-05,
      "loss": 0.6119,
      "step": 264
    },
    {
      "epoch": 1.745054945054945,
      "grad_norm": 0.23497673869132996,
      "learning_rate": 8.514412416851442e-05,
      "loss": 0.5512,
      "step": 265
    },
    {
      "epoch": 1.7516483516483516,
      "grad_norm": 0.25148245692253113,
      "learning_rate": 8.470066518847007e-05,
      "loss": 0.6089,
      "step": 266
    },
    {
      "epoch": 1.7582417582417582,
      "grad_norm": 0.2609499990940094,
      "learning_rate": 8.425720620842572e-05,
      "loss": 0.6188,
      "step": 267
    },
    {
      "epoch": 1.7648351648351648,
      "grad_norm": 0.2524459958076477,
      "learning_rate": 8.381374722838138e-05,
      "loss": 0.4777,
      "step": 268
    },
    {
      "epoch": 1.7714285714285714,
      "grad_norm": 0.2375807762145996,
      "learning_rate": 8.337028824833703e-05,
      "loss": 0.4984,
      "step": 269
    },
    {
      "epoch": 1.778021978021978,
      "grad_norm": 0.2728983759880066,
      "learning_rate": 8.292682926829268e-05,
      "loss": 0.6246,
      "step": 270
    },
    {
      "epoch": 1.7846153846153845,
      "grad_norm": 0.2366923987865448,
      "learning_rate": 8.248337028824834e-05,
      "loss": 0.4934,
      "step": 271
    },
    {
      "epoch": 1.791208791208791,
      "grad_norm": 0.24922962486743927,
      "learning_rate": 8.2039911308204e-05,
      "loss": 0.5613,
      "step": 272
    },
    {
      "epoch": 1.7978021978021979,
      "grad_norm": 0.28369754552841187,
      "learning_rate": 8.159645232815965e-05,
      "loss": 0.4915,
      "step": 273
    },
    {
      "epoch": 1.8043956043956044,
      "grad_norm": 0.24084211885929108,
      "learning_rate": 8.11529933481153e-05,
      "loss": 0.5704,
      "step": 274
    },
    {
      "epoch": 1.810989010989011,
      "grad_norm": 0.23829355835914612,
      "learning_rate": 8.070953436807095e-05,
      "loss": 0.4847,
      "step": 275
    },
    {
      "epoch": 1.8175824175824176,
      "grad_norm": 0.23963624238967896,
      "learning_rate": 8.026607538802661e-05,
      "loss": 0.5722,
      "step": 276
    },
    {
      "epoch": 1.8241758241758241,
      "grad_norm": 0.23471926152706146,
      "learning_rate": 7.982261640798227e-05,
      "loss": 0.5895,
      "step": 277
    },
    {
      "epoch": 1.830769230769231,
      "grad_norm": 0.2617950141429901,
      "learning_rate": 7.937915742793793e-05,
      "loss": 0.5754,
      "step": 278
    },
    {
      "epoch": 1.8373626373626375,
      "grad_norm": 0.2485426366329193,
      "learning_rate": 7.893569844789357e-05,
      "loss": 0.5738,
      "step": 279
    },
    {
      "epoch": 1.843956043956044,
      "grad_norm": 0.231650248169899,
      "learning_rate": 7.849223946784923e-05,
      "loss": 0.615,
      "step": 280
    },
    {
      "epoch": 1.8505494505494506,
      "grad_norm": 0.24098870158195496,
      "learning_rate": 7.804878048780489e-05,
      "loss": 0.5661,
      "step": 281
    },
    {
      "epoch": 1.8571428571428572,
      "grad_norm": 0.24551211297512054,
      "learning_rate": 7.760532150776053e-05,
      "loss": 0.5369,
      "step": 282
    },
    {
      "epoch": 1.8637362637362638,
      "grad_norm": 0.24718725681304932,
      "learning_rate": 7.716186252771619e-05,
      "loss": 0.5061,
      "step": 283
    },
    {
      "epoch": 1.8703296703296703,
      "grad_norm": 0.2683561146259308,
      "learning_rate": 7.671840354767185e-05,
      "loss": 0.6182,
      "step": 284
    },
    {
      "epoch": 1.876923076923077,
      "grad_norm": 0.2571820914745331,
      "learning_rate": 7.627494456762749e-05,
      "loss": 0.4957,
      "step": 285
    },
    {
      "epoch": 1.8835164835164835,
      "grad_norm": 0.26046374440193176,
      "learning_rate": 7.583148558758315e-05,
      "loss": 0.5826,
      "step": 286
    },
    {
      "epoch": 1.89010989010989,
      "grad_norm": 0.26960140466690063,
      "learning_rate": 7.538802660753881e-05,
      "loss": 0.6233,
      "step": 287
    },
    {
      "epoch": 1.8967032967032966,
      "grad_norm": 0.24943213164806366,
      "learning_rate": 7.494456762749445e-05,
      "loss": 0.505,
      "step": 288
    },
    {
      "epoch": 1.9032967032967032,
      "grad_norm": 0.24754472076892853,
      "learning_rate": 7.450110864745011e-05,
      "loss": 0.6231,
      "step": 289
    },
    {
      "epoch": 1.9098901098901098,
      "grad_norm": 0.26239249110221863,
      "learning_rate": 7.405764966740577e-05,
      "loss": 0.4991,
      "step": 290
    },
    {
      "epoch": 1.9164835164835163,
      "grad_norm": 0.24958840012550354,
      "learning_rate": 7.361419068736141e-05,
      "loss": 0.5379,
      "step": 291
    },
    {
      "epoch": 1.9230769230769231,
      "grad_norm": 0.23770323395729065,
      "learning_rate": 7.317073170731707e-05,
      "loss": 0.4819,
      "step": 292
    },
    {
      "epoch": 1.9296703296703297,
      "grad_norm": 0.2546253800392151,
      "learning_rate": 7.272727272727273e-05,
      "loss": 0.495,
      "step": 293
    },
    {
      "epoch": 1.9362637362637363,
      "grad_norm": 0.23957209289073944,
      "learning_rate": 7.228381374722837e-05,
      "loss": 0.5178,
      "step": 294
    },
    {
      "epoch": 1.9428571428571428,
      "grad_norm": 0.2620808780193329,
      "learning_rate": 7.184035476718403e-05,
      "loss": 0.5295,
      "step": 295
    },
    {
      "epoch": 1.9494505494505494,
      "grad_norm": 0.2699475884437561,
      "learning_rate": 7.139689578713969e-05,
      "loss": 0.5135,
      "step": 296
    },
    {
      "epoch": 1.9560439560439562,
      "grad_norm": 0.25868889689445496,
      "learning_rate": 7.095343680709535e-05,
      "loss": 0.5477,
      "step": 297
    },
    {
      "epoch": 1.9626373626373628,
      "grad_norm": 0.24537156522274017,
      "learning_rate": 7.0509977827051e-05,
      "loss": 0.5128,
      "step": 298
    },
    {
      "epoch": 1.9692307692307693,
      "grad_norm": 0.2538277506828308,
      "learning_rate": 7.006651884700666e-05,
      "loss": 0.5034,
      "step": 299
    },
    {
      "epoch": 1.975824175824176,
      "grad_norm": 0.2525295615196228,
      "learning_rate": 6.962305986696232e-05,
      "loss": 0.5351,
      "step": 300
    },
    {
      "epoch": 1.9824175824175825,
      "grad_norm": 0.24347041547298431,
      "learning_rate": 6.917960088691796e-05,
      "loss": 0.4846,
      "step": 301
    },
    {
      "epoch": 1.989010989010989,
      "grad_norm": 0.24847328662872314,
      "learning_rate": 6.873614190687362e-05,
      "loss": 0.5887,
      "step": 302
    },
    {
      "epoch": 1.9956043956043956,
      "grad_norm": 0.2484360784292221,
      "learning_rate": 6.829268292682928e-05,
      "loss": 0.5117,
      "step": 303
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.39548951387405396,
      "learning_rate": 6.784922394678492e-05,
      "loss": 0.5732,
      "step": 304
    },
    {
      "epoch": 2.0065934065934066,
      "grad_norm": 0.24039976298809052,
      "learning_rate": 6.740576496674058e-05,
      "loss": 0.4617,
      "step": 305
    },
    {
      "epoch": 2.013186813186813,
      "grad_norm": 0.2721473276615143,
      "learning_rate": 6.696230598669624e-05,
      "loss": 0.6003,
      "step": 306
    },
    {
      "epoch": 2.0197802197802197,
      "grad_norm": 0.22328545153141022,
      "learning_rate": 6.651884700665188e-05,
      "loss": 0.554,
      "step": 307
    },
    {
      "epoch": 2.0263736263736263,
      "grad_norm": 0.25350096821784973,
      "learning_rate": 6.607538802660754e-05,
      "loss": 0.5775,
      "step": 308
    },
    {
      "epoch": 2.032967032967033,
      "grad_norm": 0.2438873052597046,
      "learning_rate": 6.56319290465632e-05,
      "loss": 0.5163,
      "step": 309
    },
    {
      "epoch": 2.0395604395604394,
      "grad_norm": 0.25133413076400757,
      "learning_rate": 6.518847006651884e-05,
      "loss": 0.4577,
      "step": 310
    },
    {
      "epoch": 2.046153846153846,
      "grad_norm": 0.26965633034706116,
      "learning_rate": 6.47450110864745e-05,
      "loss": 0.5352,
      "step": 311
    },
    {
      "epoch": 2.0527472527472526,
      "grad_norm": 0.26012006402015686,
      "learning_rate": 6.430155210643016e-05,
      "loss": 0.5261,
      "step": 312
    },
    {
      "epoch": 2.059340659340659,
      "grad_norm": 0.2945905029773712,
      "learning_rate": 6.38580931263858e-05,
      "loss": 0.5248,
      "step": 313
    },
    {
      "epoch": 2.065934065934066,
      "grad_norm": 0.29531049728393555,
      "learning_rate": 6.341463414634146e-05,
      "loss": 0.4746,
      "step": 314
    },
    {
      "epoch": 2.0725274725274727,
      "grad_norm": 0.30486467480659485,
      "learning_rate": 6.297117516629712e-05,
      "loss": 0.5198,
      "step": 315
    },
    {
      "epoch": 2.0791208791208793,
      "grad_norm": 0.2699388563632965,
      "learning_rate": 6.252771618625277e-05,
      "loss": 0.4603,
      "step": 316
    },
    {
      "epoch": 2.085714285714286,
      "grad_norm": 0.301487535238266,
      "learning_rate": 6.208425720620842e-05,
      "loss": 0.4827,
      "step": 317
    },
    {
      "epoch": 2.0923076923076924,
      "grad_norm": 0.2875276207923889,
      "learning_rate": 6.164079822616408e-05,
      "loss": 0.5228,
      "step": 318
    },
    {
      "epoch": 2.098901098901099,
      "grad_norm": 0.28546640276908875,
      "learning_rate": 6.119733924611974e-05,
      "loss": 0.6257,
      "step": 319
    },
    {
      "epoch": 2.1054945054945056,
      "grad_norm": 0.27191463112831116,
      "learning_rate": 6.075388026607539e-05,
      "loss": 0.4091,
      "step": 320
    },
    {
      "epoch": 2.112087912087912,
      "grad_norm": 0.28798726201057434,
      "learning_rate": 6.031042128603105e-05,
      "loss": 0.4868,
      "step": 321
    },
    {
      "epoch": 2.1186813186813187,
      "grad_norm": 0.2725013494491577,
      "learning_rate": 5.98669623059867e-05,
      "loss": 0.4831,
      "step": 322
    },
    {
      "epoch": 2.1252747252747253,
      "grad_norm": 0.2926606833934784,
      "learning_rate": 5.942350332594236e-05,
      "loss": 0.5173,
      "step": 323
    },
    {
      "epoch": 2.131868131868132,
      "grad_norm": 0.2967899739742279,
      "learning_rate": 5.898004434589801e-05,
      "loss": 0.5197,
      "step": 324
    },
    {
      "epoch": 2.1384615384615384,
      "grad_norm": 0.29469364881515503,
      "learning_rate": 5.853658536585366e-05,
      "loss": 0.5077,
      "step": 325
    },
    {
      "epoch": 2.145054945054945,
      "grad_norm": 0.2641926109790802,
      "learning_rate": 5.809312638580932e-05,
      "loss": 0.4249,
      "step": 326
    },
    {
      "epoch": 2.1516483516483516,
      "grad_norm": 0.3154348134994507,
      "learning_rate": 5.7649667405764975e-05,
      "loss": 0.5044,
      "step": 327
    },
    {
      "epoch": 2.158241758241758,
      "grad_norm": 0.2879049777984619,
      "learning_rate": 5.720620842572062e-05,
      "loss": 0.5238,
      "step": 328
    },
    {
      "epoch": 2.1648351648351647,
      "grad_norm": 0.3066406548023224,
      "learning_rate": 5.676274944567628e-05,
      "loss": 0.5485,
      "step": 329
    },
    {
      "epoch": 2.1714285714285713,
      "grad_norm": 0.2888032793998718,
      "learning_rate": 5.6319290465631935e-05,
      "loss": 0.5272,
      "step": 330
    },
    {
      "epoch": 2.178021978021978,
      "grad_norm": 0.34023240208625793,
      "learning_rate": 5.587583148558758e-05,
      "loss": 0.5321,
      "step": 331
    },
    {
      "epoch": 2.184615384615385,
      "grad_norm": 0.2888537049293518,
      "learning_rate": 5.543237250554324e-05,
      "loss": 0.4675,
      "step": 332
    },
    {
      "epoch": 2.1912087912087914,
      "grad_norm": 0.30242598056793213,
      "learning_rate": 5.4988913525498895e-05,
      "loss": 0.4411,
      "step": 333
    },
    {
      "epoch": 2.197802197802198,
      "grad_norm": 0.3289373517036438,
      "learning_rate": 5.4545454545454546e-05,
      "loss": 0.5105,
      "step": 334
    },
    {
      "epoch": 2.2043956043956046,
      "grad_norm": 0.3340095281600952,
      "learning_rate": 5.4101995565410204e-05,
      "loss": 0.5742,
      "step": 335
    },
    {
      "epoch": 2.210989010989011,
      "grad_norm": 0.3358578681945801,
      "learning_rate": 5.365853658536586e-05,
      "loss": 0.5127,
      "step": 336
    },
    {
      "epoch": 2.2175824175824177,
      "grad_norm": 0.3217174708843231,
      "learning_rate": 5.3215077605321506e-05,
      "loss": 0.5071,
      "step": 337
    },
    {
      "epoch": 2.2241758241758243,
      "grad_norm": 0.348807156085968,
      "learning_rate": 5.2771618625277164e-05,
      "loss": 0.4915,
      "step": 338
    },
    {
      "epoch": 2.230769230769231,
      "grad_norm": 0.3451012670993805,
      "learning_rate": 5.232815964523282e-05,
      "loss": 0.5484,
      "step": 339
    },
    {
      "epoch": 2.2373626373626374,
      "grad_norm": 0.3735233545303345,
      "learning_rate": 5.1884700665188466e-05,
      "loss": 0.5364,
      "step": 340
    },
    {
      "epoch": 2.243956043956044,
      "grad_norm": 0.37038934230804443,
      "learning_rate": 5.1441241685144124e-05,
      "loss": 0.4936,
      "step": 341
    },
    {
      "epoch": 2.2505494505494505,
      "grad_norm": 0.31901484727859497,
      "learning_rate": 5.099778270509978e-05,
      "loss": 0.466,
      "step": 342
    },
    {
      "epoch": 2.257142857142857,
      "grad_norm": 0.3197035789489746,
      "learning_rate": 5.0554323725055433e-05,
      "loss": 0.5121,
      "step": 343
    },
    {
      "epoch": 2.2637362637362637,
      "grad_norm": 0.3947635889053345,
      "learning_rate": 5.011086474501109e-05,
      "loss": 0.5322,
      "step": 344
    },
    {
      "epoch": 2.2703296703296703,
      "grad_norm": 0.34863051772117615,
      "learning_rate": 4.966740576496674e-05,
      "loss": 0.4848,
      "step": 345
    },
    {
      "epoch": 2.276923076923077,
      "grad_norm": 0.3473830223083496,
      "learning_rate": 4.92239467849224e-05,
      "loss": 0.5907,
      "step": 346
    },
    {
      "epoch": 2.2835164835164834,
      "grad_norm": 0.315640389919281,
      "learning_rate": 4.878048780487805e-05,
      "loss": 0.4617,
      "step": 347
    },
    {
      "epoch": 2.29010989010989,
      "grad_norm": 0.34681639075279236,
      "learning_rate": 4.833702882483371e-05,
      "loss": 0.4827,
      "step": 348
    },
    {
      "epoch": 2.2967032967032965,
      "grad_norm": 0.35968664288520813,
      "learning_rate": 4.789356984478936e-05,
      "loss": 0.5576,
      "step": 349
    },
    {
      "epoch": 2.303296703296703,
      "grad_norm": 0.3568693995475769,
      "learning_rate": 4.745011086474501e-05,
      "loss": 0.5064,
      "step": 350
    },
    {
      "epoch": 2.3098901098901097,
      "grad_norm": 0.3719553053379059,
      "learning_rate": 4.700665188470067e-05,
      "loss": 0.5869,
      "step": 351
    },
    {
      "epoch": 2.3164835164835162,
      "grad_norm": 0.34675082564353943,
      "learning_rate": 4.656319290465632e-05,
      "loss": 0.5017,
      "step": 352
    },
    {
      "epoch": 2.3230769230769233,
      "grad_norm": 0.3885175585746765,
      "learning_rate": 4.611973392461197e-05,
      "loss": 0.5589,
      "step": 353
    },
    {
      "epoch": 2.32967032967033,
      "grad_norm": 0.3569480776786804,
      "learning_rate": 4.567627494456763e-05,
      "loss": 0.4651,
      "step": 354
    },
    {
      "epoch": 2.3362637362637364,
      "grad_norm": 0.3270711600780487,
      "learning_rate": 4.523281596452328e-05,
      "loss": 0.3714,
      "step": 355
    },
    {
      "epoch": 2.342857142857143,
      "grad_norm": 0.38800808787345886,
      "learning_rate": 4.478935698447894e-05,
      "loss": 0.4679,
      "step": 356
    },
    {
      "epoch": 2.3494505494505495,
      "grad_norm": 0.4136335849761963,
      "learning_rate": 4.4345898004434597e-05,
      "loss": 0.5295,
      "step": 357
    },
    {
      "epoch": 2.356043956043956,
      "grad_norm": 0.3837147355079651,
      "learning_rate": 4.390243902439025e-05,
      "loss": 0.5464,
      "step": 358
    },
    {
      "epoch": 2.3626373626373627,
      "grad_norm": 0.40014463663101196,
      "learning_rate": 4.34589800443459e-05,
      "loss": 0.5272,
      "step": 359
    },
    {
      "epoch": 2.3692307692307693,
      "grad_norm": 0.39902275800704956,
      "learning_rate": 4.301552106430156e-05,
      "loss": 0.5156,
      "step": 360
    },
    {
      "epoch": 2.375824175824176,
      "grad_norm": 0.3863047957420349,
      "learning_rate": 4.257206208425721e-05,
      "loss": 0.4728,
      "step": 361
    },
    {
      "epoch": 2.3824175824175824,
      "grad_norm": 0.4098444879055023,
      "learning_rate": 4.212860310421286e-05,
      "loss": 0.4699,
      "step": 362
    },
    {
      "epoch": 2.389010989010989,
      "grad_norm": 0.3766367435455322,
      "learning_rate": 4.168514412416852e-05,
      "loss": 0.4752,
      "step": 363
    },
    {
      "epoch": 2.3956043956043955,
      "grad_norm": 0.3870093524456024,
      "learning_rate": 4.124168514412417e-05,
      "loss": 0.5683,
      "step": 364
    },
    {
      "epoch": 2.402197802197802,
      "grad_norm": 0.371272474527359,
      "learning_rate": 4.0798226164079826e-05,
      "loss": 0.4843,
      "step": 365
    },
    {
      "epoch": 2.4087912087912087,
      "grad_norm": 0.37084656953811646,
      "learning_rate": 4.035476718403548e-05,
      "loss": 0.5266,
      "step": 366
    },
    {
      "epoch": 2.4153846153846152,
      "grad_norm": 0.4056682884693146,
      "learning_rate": 3.9911308203991135e-05,
      "loss": 0.4813,
      "step": 367
    },
    {
      "epoch": 2.421978021978022,
      "grad_norm": 0.40132272243499756,
      "learning_rate": 3.9467849223946786e-05,
      "loss": 0.5264,
      "step": 368
    },
    {
      "epoch": 2.4285714285714284,
      "grad_norm": 0.3832414150238037,
      "learning_rate": 3.9024390243902444e-05,
      "loss": 0.467,
      "step": 369
    },
    {
      "epoch": 2.4351648351648354,
      "grad_norm": 0.4004274010658264,
      "learning_rate": 3.8580931263858095e-05,
      "loss": 0.5295,
      "step": 370
    },
    {
      "epoch": 2.441758241758242,
      "grad_norm": 0.4040567874908447,
      "learning_rate": 3.8137472283813746e-05,
      "loss": 0.4744,
      "step": 371
    },
    {
      "epoch": 2.4483516483516485,
      "grad_norm": 0.40446919202804565,
      "learning_rate": 3.7694013303769404e-05,
      "loss": 0.4696,
      "step": 372
    },
    {
      "epoch": 2.454945054945055,
      "grad_norm": 0.3889092803001404,
      "learning_rate": 3.7250554323725055e-05,
      "loss": 0.5314,
      "step": 373
    },
    {
      "epoch": 2.4615384615384617,
      "grad_norm": 0.3994397521018982,
      "learning_rate": 3.6807095343680706e-05,
      "loss": 0.5183,
      "step": 374
    },
    {
      "epoch": 2.4681318681318682,
      "grad_norm": 0.3825172185897827,
      "learning_rate": 3.6363636363636364e-05,
      "loss": 0.4498,
      "step": 375
    },
    {
      "epoch": 2.474725274725275,
      "grad_norm": 0.40210065245628357,
      "learning_rate": 3.5920177383592015e-05,
      "loss": 0.521,
      "step": 376
    },
    {
      "epoch": 2.4813186813186814,
      "grad_norm": 0.4190777540206909,
      "learning_rate": 3.547671840354767e-05,
      "loss": 0.5146,
      "step": 377
    },
    {
      "epoch": 2.487912087912088,
      "grad_norm": 0.40617266297340393,
      "learning_rate": 3.503325942350333e-05,
      "loss": 0.5556,
      "step": 378
    },
    {
      "epoch": 2.4945054945054945,
      "grad_norm": 0.47587987780570984,
      "learning_rate": 3.458980044345898e-05,
      "loss": 0.5083,
      "step": 379
    },
    {
      "epoch": 2.501098901098901,
      "grad_norm": 0.3952488303184509,
      "learning_rate": 3.414634146341464e-05,
      "loss": 0.4811,
      "step": 380
    },
    {
      "epoch": 2.5076923076923077,
      "grad_norm": 0.37684306502342224,
      "learning_rate": 3.370288248337029e-05,
      "loss": 0.4736,
      "step": 381
    },
    {
      "epoch": 2.5142857142857142,
      "grad_norm": 0.4320245087146759,
      "learning_rate": 3.325942350332594e-05,
      "loss": 0.4627,
      "step": 382
    },
    {
      "epoch": 2.520879120879121,
      "grad_norm": 0.4025600552558899,
      "learning_rate": 3.28159645232816e-05,
      "loss": 0.4486,
      "step": 383
    },
    {
      "epoch": 2.5274725274725274,
      "grad_norm": 0.41375601291656494,
      "learning_rate": 3.237250554323725e-05,
      "loss": 0.5214,
      "step": 384
    },
    {
      "epoch": 2.534065934065934,
      "grad_norm": 0.3857056498527527,
      "learning_rate": 3.19290465631929e-05,
      "loss": 0.4795,
      "step": 385
    },
    {
      "epoch": 2.5406593406593405,
      "grad_norm": 0.43289706110954285,
      "learning_rate": 3.148558758314856e-05,
      "loss": 0.615,
      "step": 386
    },
    {
      "epoch": 2.547252747252747,
      "grad_norm": 0.40202274918556213,
      "learning_rate": 3.104212860310421e-05,
      "loss": 0.4896,
      "step": 387
    },
    {
      "epoch": 2.5538461538461537,
      "grad_norm": 0.4154258370399475,
      "learning_rate": 3.059866962305987e-05,
      "loss": 0.4521,
      "step": 388
    },
    {
      "epoch": 2.5604395604395602,
      "grad_norm": 0.3912176191806793,
      "learning_rate": 3.0155210643015524e-05,
      "loss": 0.5269,
      "step": 389
    },
    {
      "epoch": 2.567032967032967,
      "grad_norm": 0.40165093541145325,
      "learning_rate": 2.971175166297118e-05,
      "loss": 0.5014,
      "step": 390
    },
    {
      "epoch": 2.5736263736263734,
      "grad_norm": 0.41760513186454773,
      "learning_rate": 2.926829268292683e-05,
      "loss": 0.5493,
      "step": 391
    },
    {
      "epoch": 2.5802197802197804,
      "grad_norm": 0.4197694957256317,
      "learning_rate": 2.8824833702882487e-05,
      "loss": 0.4624,
      "step": 392
    },
    {
      "epoch": 2.586813186813187,
      "grad_norm": 0.41370582580566406,
      "learning_rate": 2.838137472283814e-05,
      "loss": 0.5441,
      "step": 393
    },
    {
      "epoch": 2.5934065934065935,
      "grad_norm": 0.38814762234687805,
      "learning_rate": 2.793791574279379e-05,
      "loss": 0.4914,
      "step": 394
    },
    {
      "epoch": 2.6,
      "grad_norm": 0.41799941658973694,
      "learning_rate": 2.7494456762749448e-05,
      "loss": 0.441,
      "step": 395
    },
    {
      "epoch": 2.6065934065934067,
      "grad_norm": 0.43591222167015076,
      "learning_rate": 2.7050997782705102e-05,
      "loss": 0.4997,
      "step": 396
    },
    {
      "epoch": 2.6131868131868132,
      "grad_norm": 0.4181253910064697,
      "learning_rate": 2.6607538802660753e-05,
      "loss": 0.5668,
      "step": 397
    },
    {
      "epoch": 2.61978021978022,
      "grad_norm": 0.44083935022354126,
      "learning_rate": 2.616407982261641e-05,
      "loss": 0.4634,
      "step": 398
    },
    {
      "epoch": 2.6263736263736264,
      "grad_norm": 0.42532244324684143,
      "learning_rate": 2.5720620842572062e-05,
      "loss": 0.4866,
      "step": 399
    },
    {
      "epoch": 2.632967032967033,
      "grad_norm": 0.3996051251888275,
      "learning_rate": 2.5277161862527717e-05,
      "loss": 0.42,
      "step": 400
    },
    {
      "epoch": 2.6395604395604395,
      "grad_norm": 0.46346697211265564,
      "learning_rate": 2.483370288248337e-05,
      "loss": 0.5144,
      "step": 401
    },
    {
      "epoch": 2.646153846153846,
      "grad_norm": 0.3908272087574005,
      "learning_rate": 2.4390243902439026e-05,
      "loss": 0.4302,
      "step": 402
    },
    {
      "epoch": 2.6527472527472526,
      "grad_norm": 0.41894593834877014,
      "learning_rate": 2.394678492239468e-05,
      "loss": 0.5297,
      "step": 403
    },
    {
      "epoch": 2.659340659340659,
      "grad_norm": 0.43422892689704895,
      "learning_rate": 2.3503325942350335e-05,
      "loss": 0.5781,
      "step": 404
    },
    {
      "epoch": 2.665934065934066,
      "grad_norm": 0.4160975515842438,
      "learning_rate": 2.3059866962305986e-05,
      "loss": 0.4642,
      "step": 405
    },
    {
      "epoch": 2.672527472527473,
      "grad_norm": 0.47415420413017273,
      "learning_rate": 2.261640798226164e-05,
      "loss": 0.4725,
      "step": 406
    },
    {
      "epoch": 2.6791208791208794,
      "grad_norm": 0.38781166076660156,
      "learning_rate": 2.2172949002217298e-05,
      "loss": 0.4155,
      "step": 407
    },
    {
      "epoch": 2.685714285714286,
      "grad_norm": 0.4265604317188263,
      "learning_rate": 2.172949002217295e-05,
      "loss": 0.5561,
      "step": 408
    },
    {
      "epoch": 2.6923076923076925,
      "grad_norm": 0.4570036232471466,
      "learning_rate": 2.1286031042128604e-05,
      "loss": 0.5064,
      "step": 409
    },
    {
      "epoch": 2.698901098901099,
      "grad_norm": 0.4613281488418579,
      "learning_rate": 2.084257206208426e-05,
      "loss": 0.4695,
      "step": 410
    },
    {
      "epoch": 2.7054945054945057,
      "grad_norm": 0.44290849566459656,
      "learning_rate": 2.0399113082039913e-05,
      "loss": 0.5039,
      "step": 411
    },
    {
      "epoch": 2.7120879120879122,
      "grad_norm": 0.40200501680374146,
      "learning_rate": 1.9955654101995567e-05,
      "loss": 0.4959,
      "step": 412
    },
    {
      "epoch": 2.718681318681319,
      "grad_norm": 0.4261188209056854,
      "learning_rate": 1.9512195121951222e-05,
      "loss": 0.4811,
      "step": 413
    },
    {
      "epoch": 2.7252747252747254,
      "grad_norm": 0.43007826805114746,
      "learning_rate": 1.9068736141906873e-05,
      "loss": 0.5047,
      "step": 414
    },
    {
      "epoch": 2.731868131868132,
      "grad_norm": 0.4427857995033264,
      "learning_rate": 1.8625277161862528e-05,
      "loss": 0.5019,
      "step": 415
    },
    {
      "epoch": 2.7384615384615385,
      "grad_norm": 0.4366683065891266,
      "learning_rate": 1.8181818181818182e-05,
      "loss": 0.4928,
      "step": 416
    },
    {
      "epoch": 2.745054945054945,
      "grad_norm": 0.42620736360549927,
      "learning_rate": 1.7738359201773837e-05,
      "loss": 0.4846,
      "step": 417
    },
    {
      "epoch": 2.7516483516483516,
      "grad_norm": 0.4484277665615082,
      "learning_rate": 1.729490022172949e-05,
      "loss": 0.5129,
      "step": 418
    },
    {
      "epoch": 2.758241758241758,
      "grad_norm": 0.40524691343307495,
      "learning_rate": 1.6851441241685146e-05,
      "loss": 0.4999,
      "step": 419
    },
    {
      "epoch": 2.764835164835165,
      "grad_norm": 0.42339208722114563,
      "learning_rate": 1.64079822616408e-05,
      "loss": 0.4774,
      "step": 420
    },
    {
      "epoch": 2.7714285714285714,
      "grad_norm": 0.43162041902542114,
      "learning_rate": 1.596452328159645e-05,
      "loss": 0.5131,
      "step": 421
    },
    {
      "epoch": 2.778021978021978,
      "grad_norm": 0.4126211106777191,
      "learning_rate": 1.5521064301552106e-05,
      "loss": 0.496,
      "step": 422
    },
    {
      "epoch": 2.7846153846153845,
      "grad_norm": 0.4301091134548187,
      "learning_rate": 1.5077605321507762e-05,
      "loss": 0.6165,
      "step": 423
    },
    {
      "epoch": 2.791208791208791,
      "grad_norm": 0.4416787624359131,
      "learning_rate": 1.4634146341463415e-05,
      "loss": 0.5538,
      "step": 424
    },
    {
      "epoch": 2.7978021978021976,
      "grad_norm": 0.4086042046546936,
      "learning_rate": 1.419068736141907e-05,
      "loss": 0.4986,
      "step": 425
    },
    {
      "epoch": 2.804395604395604,
      "grad_norm": 0.41502517461776733,
      "learning_rate": 1.3747228381374724e-05,
      "loss": 0.5108,
      "step": 426
    },
    {
      "epoch": 2.8109890109890108,
      "grad_norm": 0.4186524450778961,
      "learning_rate": 1.3303769401330377e-05,
      "loss": 0.4517,
      "step": 427
    },
    {
      "epoch": 2.8175824175824173,
      "grad_norm": 0.4349028170108795,
      "learning_rate": 1.2860310421286031e-05,
      "loss": 0.4835,
      "step": 428
    },
    {
      "epoch": 2.824175824175824,
      "grad_norm": 0.47845181822776794,
      "learning_rate": 1.2416851441241686e-05,
      "loss": 0.4832,
      "step": 429
    },
    {
      "epoch": 2.830769230769231,
      "grad_norm": 0.42505326867103577,
      "learning_rate": 1.197339246119734e-05,
      "loss": 0.4882,
      "step": 430
    },
    {
      "epoch": 2.8373626373626375,
      "grad_norm": 0.401644766330719,
      "learning_rate": 1.1529933481152993e-05,
      "loss": 0.4971,
      "step": 431
    },
    {
      "epoch": 2.843956043956044,
      "grad_norm": 0.4014747738838196,
      "learning_rate": 1.1086474501108649e-05,
      "loss": 0.477,
      "step": 432
    },
    {
      "epoch": 2.8505494505494506,
      "grad_norm": 0.4275830388069153,
      "learning_rate": 1.0643015521064302e-05,
      "loss": 0.439,
      "step": 433
    },
    {
      "epoch": 2.857142857142857,
      "grad_norm": 0.4436344504356384,
      "learning_rate": 1.0199556541019956e-05,
      "loss": 0.5578,
      "step": 434
    },
    {
      "epoch": 2.8637362637362638,
      "grad_norm": 0.43587949872016907,
      "learning_rate": 9.756097560975611e-06,
      "loss": 0.4519,
      "step": 435
    },
    {
      "epoch": 2.8703296703296703,
      "grad_norm": 0.43619510531425476,
      "learning_rate": 9.312638580931264e-06,
      "loss": 0.4746,
      "step": 436
    },
    {
      "epoch": 2.876923076923077,
      "grad_norm": 0.4361239969730377,
      "learning_rate": 8.869179600886918e-06,
      "loss": 0.5122,
      "step": 437
    },
    {
      "epoch": 2.8835164835164835,
      "grad_norm": 0.4136911630630493,
      "learning_rate": 8.425720620842573e-06,
      "loss": 0.4655,
      "step": 438
    },
    {
      "epoch": 2.89010989010989,
      "grad_norm": 0.41434943675994873,
      "learning_rate": 7.982261640798226e-06,
      "loss": 0.5026,
      "step": 439
    },
    {
      "epoch": 2.8967032967032966,
      "grad_norm": 0.450967013835907,
      "learning_rate": 7.538802660753881e-06,
      "loss": 0.4688,
      "step": 440
    },
    {
      "epoch": 2.903296703296703,
      "grad_norm": 0.4313754439353943,
      "learning_rate": 7.095343680709535e-06,
      "loss": 0.4447,
      "step": 441
    },
    {
      "epoch": 2.9098901098901098,
      "grad_norm": 0.4619598984718323,
      "learning_rate": 6.651884700665188e-06,
      "loss": 0.5356,
      "step": 442
    },
    {
      "epoch": 2.9164835164835163,
      "grad_norm": 0.4337940812110901,
      "learning_rate": 6.208425720620843e-06,
      "loss": 0.5465,
      "step": 443
    },
    {
      "epoch": 2.9230769230769234,
      "grad_norm": 0.4400572180747986,
      "learning_rate": 5.7649667405764965e-06,
      "loss": 0.4949,
      "step": 444
    },
    {
      "epoch": 2.92967032967033,
      "grad_norm": 0.4876385033130646,
      "learning_rate": 5.321507760532151e-06,
      "loss": 0.5579,
      "step": 445
    },
    {
      "epoch": 2.9362637362637365,
      "grad_norm": 0.4508369565010071,
      "learning_rate": 4.8780487804878055e-06,
      "loss": 0.501,
      "step": 446
    },
    {
      "epoch": 2.942857142857143,
      "grad_norm": 0.4498026371002197,
      "learning_rate": 4.434589800443459e-06,
      "loss": 0.5065,
      "step": 447
    },
    {
      "epoch": 2.9494505494505496,
      "grad_norm": 0.4527037739753723,
      "learning_rate": 3.991130820399113e-06,
      "loss": 0.5215,
      "step": 448
    },
    {
      "epoch": 2.956043956043956,
      "grad_norm": 0.44119498133659363,
      "learning_rate": 3.5476718403547673e-06,
      "loss": 0.4705,
      "step": 449
    },
    {
      "epoch": 2.9626373626373628,
      "grad_norm": 0.45794641971588135,
      "learning_rate": 3.1042128603104214e-06,
      "loss": 0.5159,
      "step": 450
    },
    {
      "epoch": 2.9692307692307693,
      "grad_norm": 0.48152127861976624,
      "learning_rate": 2.6607538802660755e-06,
      "loss": 0.5344,
      "step": 451
    },
    {
      "epoch": 2.975824175824176,
      "grad_norm": 0.43193677067756653,
      "learning_rate": 2.2172949002217296e-06,
      "loss": 0.4668,
      "step": 452
    },
    {
      "epoch": 2.9824175824175825,
      "grad_norm": 0.495225191116333,
      "learning_rate": 1.7738359201773837e-06,
      "loss": 0.5442,
      "step": 453
    },
    {
      "epoch": 2.989010989010989,
      "grad_norm": 0.4606660008430481,
      "learning_rate": 1.3303769401330377e-06,
      "loss": 0.4854,
      "step": 454
    },
    {
      "epoch": 2.9956043956043956,
      "grad_norm": 0.39847251772880554,
      "learning_rate": 8.869179600886918e-07,
      "loss": 0.4871,
      "step": 455
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.617318868637085,
      "learning_rate": 4.434589800443459e-07,
      "loss": 0.4362,
      "step": 456
    }
  ],
  "logging_steps": 1,
  "max_steps": 456,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 15,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 2.1181451835414528e+17,
  "train_batch_size": 22,
  "trial_name": null,
  "trial_params": null
}