{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.984375,
  "eval_steps": 500,
  "global_step": 342,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01171875,
      "grad_norm": 0.9319692850112915,
      "learning_rate": 1.1764705882352942e-07,
      "loss": 0.7028,
      "step": 1
    },
    {
      "epoch": 0.0234375,
      "grad_norm": 0.9757155179977417,
      "learning_rate": 2.3529411764705883e-07,
      "loss": 0.7416,
      "step": 2
    },
    {
      "epoch": 0.03515625,
      "grad_norm": 1.0889487266540527,
      "learning_rate": 3.529411764705883e-07,
      "loss": 0.8392,
      "step": 3
    },
    {
      "epoch": 0.046875,
      "grad_norm": 1.0020272731781006,
      "learning_rate": 4.7058823529411767e-07,
      "loss": 0.7549,
      "step": 4
    },
    {
      "epoch": 0.05859375,
      "grad_norm": 1.0064201354980469,
      "learning_rate": 5.882352941176471e-07,
      "loss": 0.802,
      "step": 5
    },
    {
      "epoch": 0.0703125,
      "grad_norm": 0.9806166291236877,
      "learning_rate": 7.058823529411766e-07,
      "loss": 0.7754,
      "step": 6
    },
    {
      "epoch": 0.08203125,
      "grad_norm": 0.9506519436836243,
      "learning_rate": 8.235294117647059e-07,
      "loss": 0.7591,
      "step": 7
    },
    {
      "epoch": 0.09375,
      "grad_norm": 0.9138185977935791,
      "learning_rate": 9.411764705882353e-07,
      "loss": 0.7681,
      "step": 8
    },
    {
      "epoch": 0.10546875,
      "grad_norm": 0.872790515422821,
      "learning_rate": 1.0588235294117648e-06,
      "loss": 0.7332,
      "step": 9
    },
    {
      "epoch": 0.1171875,
      "grad_norm": 0.8308555483818054,
      "learning_rate": 1.1764705882352942e-06,
      "loss": 0.7991,
      "step": 10
    },
    {
      "epoch": 0.12890625,
      "grad_norm": 0.7250374555587769,
      "learning_rate": 1.2941176470588237e-06,
      "loss": 0.6812,
      "step": 11
    },
    {
      "epoch": 0.140625,
      "grad_norm": 0.6780915856361389,
      "learning_rate": 1.4117647058823531e-06,
      "loss": 0.7178,
      "step": 12
    },
    {
      "epoch": 0.15234375,
      "grad_norm": 0.6684752702713013,
      "learning_rate": 1.5294117647058826e-06,
      "loss": 0.7335,
      "step": 13
    },
    {
      "epoch": 0.1640625,
      "grad_norm": 0.48017001152038574,
      "learning_rate": 1.6470588235294118e-06,
      "loss": 0.7293,
      "step": 14
    },
    {
      "epoch": 0.17578125,
      "grad_norm": 0.4176006317138672,
      "learning_rate": 1.7647058823529414e-06,
      "loss": 0.6453,
      "step": 15
    },
    {
      "epoch": 0.1875,
      "grad_norm": 0.4581579566001892,
      "learning_rate": 1.8823529411764707e-06,
      "loss": 0.709,
      "step": 16
    },
    {
      "epoch": 0.19921875,
      "grad_norm": 0.41040804982185364,
      "learning_rate": 2.0000000000000003e-06,
      "loss": 0.6965,
      "step": 17
    },
    {
      "epoch": 0.2109375,
      "grad_norm": 0.38801896572113037,
      "learning_rate": 2.1176470588235296e-06,
      "loss": 0.7142,
      "step": 18
    },
    {
      "epoch": 0.22265625,
      "grad_norm": 0.5587085485458374,
      "learning_rate": 2.2352941176470592e-06,
      "loss": 0.6624,
      "step": 19
    },
    {
      "epoch": 0.234375,
      "grad_norm": 0.6821652054786682,
      "learning_rate": 2.3529411764705885e-06,
      "loss": 0.6708,
      "step": 20
    },
    {
      "epoch": 0.24609375,
      "grad_norm": 0.7130780816078186,
      "learning_rate": 2.470588235294118e-06,
      "loss": 0.68,
      "step": 21
    },
    {
      "epoch": 0.2578125,
      "grad_norm": 0.6799156069755554,
      "learning_rate": 2.5882352941176473e-06,
      "loss": 0.6069,
      "step": 22
    },
    {
      "epoch": 0.26953125,
      "grad_norm": 0.6155521869659424,
      "learning_rate": 2.7058823529411766e-06,
      "loss": 0.6541,
      "step": 23
    },
    {
      "epoch": 0.28125,
      "grad_norm": 0.5290796160697937,
      "learning_rate": 2.8235294117647062e-06,
      "loss": 0.6276,
      "step": 24
    },
    {
      "epoch": 0.29296875,
      "grad_norm": 0.460702121257782,
      "learning_rate": 2.9411764705882355e-06,
      "loss": 0.6172,
      "step": 25
    },
    {
      "epoch": 0.3046875,
      "grad_norm": 0.4080013334751129,
      "learning_rate": 3.058823529411765e-06,
      "loss": 0.609,
      "step": 26
    },
    {
      "epoch": 0.31640625,
      "grad_norm": 0.32690924406051636,
      "learning_rate": 3.1764705882352943e-06,
      "loss": 0.585,
      "step": 27
    },
    {
      "epoch": 0.328125,
      "grad_norm": 0.3010313808917999,
      "learning_rate": 3.2941176470588236e-06,
      "loss": 0.5684,
      "step": 28
    },
    {
      "epoch": 0.33984375,
      "grad_norm": 0.30681392550468445,
      "learning_rate": 3.4117647058823532e-06,
      "loss": 0.6241,
      "step": 29
    },
    {
      "epoch": 0.3515625,
      "grad_norm": 0.2821861505508423,
      "learning_rate": 3.529411764705883e-06,
      "loss": 0.5612,
      "step": 30
    },
    {
      "epoch": 0.36328125,
      "grad_norm": 0.3114885985851288,
      "learning_rate": 3.6470588235294117e-06,
      "loss": 0.5777,
      "step": 31
    },
    {
      "epoch": 0.375,
      "grad_norm": 0.2961554527282715,
      "learning_rate": 3.7647058823529414e-06,
      "loss": 0.6118,
      "step": 32
    },
    {
      "epoch": 0.38671875,
      "grad_norm": 0.2794322371482849,
      "learning_rate": 3.882352941176471e-06,
      "loss": 0.5221,
      "step": 33
    },
    {
      "epoch": 0.3984375,
      "grad_norm": 0.33694833517074585,
      "learning_rate": 4.000000000000001e-06,
      "loss": 0.5857,
      "step": 34
    },
    {
      "epoch": 0.41015625,
      "grad_norm": 0.2691885530948639,
      "learning_rate": 4.11764705882353e-06,
      "loss": 0.6122,
      "step": 35
    },
    {
      "epoch": 0.421875,
      "grad_norm": 0.2636789381504059,
      "learning_rate": 4.235294117647059e-06,
      "loss": 0.5355,
      "step": 36
    },
    {
      "epoch": 0.43359375,
      "grad_norm": 0.25277385115623474,
      "learning_rate": 4.352941176470588e-06,
      "loss": 0.5373,
      "step": 37
    },
    {
      "epoch": 0.4453125,
      "grad_norm": 0.2117689847946167,
      "learning_rate": 4.4705882352941184e-06,
      "loss": 0.5308,
      "step": 38
    },
    {
      "epoch": 0.45703125,
      "grad_norm": 0.19506287574768066,
      "learning_rate": 4.588235294117647e-06,
      "loss": 0.5372,
      "step": 39
    },
    {
      "epoch": 0.46875,
      "grad_norm": 0.21475745737552643,
      "learning_rate": 4.705882352941177e-06,
      "loss": 0.565,
      "step": 40
    },
    {
      "epoch": 0.48046875,
      "grad_norm": 0.20359668135643005,
      "learning_rate": 4.823529411764706e-06,
      "loss": 0.5349,
      "step": 41
    },
    {
      "epoch": 0.4921875,
      "grad_norm": 0.2028248906135559,
      "learning_rate": 4.941176470588236e-06,
      "loss": 0.5587,
      "step": 42
    },
    {
      "epoch": 0.50390625,
      "grad_norm": 0.21315933763980865,
      "learning_rate": 5.058823529411765e-06,
      "loss": 0.542,
      "step": 43
    },
    {
      "epoch": 0.515625,
      "grad_norm": 0.21315297484397888,
      "learning_rate": 5.176470588235295e-06,
      "loss": 0.5878,
      "step": 44
    },
    {
      "epoch": 0.52734375,
      "grad_norm": 0.19848909974098206,
      "learning_rate": 5.294117647058824e-06,
      "loss": 0.5238,
      "step": 45
    },
    {
      "epoch": 0.5390625,
      "grad_norm": 0.20311777293682098,
      "learning_rate": 5.411764705882353e-06,
      "loss": 0.548,
      "step": 46
    },
    {
      "epoch": 0.55078125,
      "grad_norm": 0.18243664503097534,
      "learning_rate": 5.529411764705883e-06,
      "loss": 0.5282,
      "step": 47
    },
    {
      "epoch": 0.5625,
      "grad_norm": 0.16974785923957825,
      "learning_rate": 5.6470588235294125e-06,
      "loss": 0.4836,
      "step": 48
    },
    {
      "epoch": 0.57421875,
      "grad_norm": 0.1762179285287857,
      "learning_rate": 5.764705882352941e-06,
      "loss": 0.5304,
      "step": 49
    },
    {
      "epoch": 0.5859375,
      "grad_norm": 0.20087581872940063,
      "learning_rate": 5.882352941176471e-06,
      "loss": 0.5317,
      "step": 50
    },
    {
      "epoch": 0.59765625,
      "grad_norm": 0.1862439662218094,
      "learning_rate": 6e-06,
      "loss": 0.5536,
      "step": 51
    },
    {
      "epoch": 0.609375,
      "grad_norm": 0.1928662210702896,
      "learning_rate": 6.11764705882353e-06,
      "loss": 0.4844,
      "step": 52
    },
    {
      "epoch": 0.62109375,
      "grad_norm": 0.20256248116493225,
      "learning_rate": 6.2352941176470595e-06,
      "loss": 0.5428,
      "step": 53
    },
    {
      "epoch": 0.6328125,
      "grad_norm": 0.2007371336221695,
      "learning_rate": 6.352941176470589e-06,
      "loss": 0.5225,
      "step": 54
    },
    {
      "epoch": 0.64453125,
      "grad_norm": 0.16175827383995056,
      "learning_rate": 6.470588235294119e-06,
      "loss": 0.5137,
      "step": 55
    },
    {
      "epoch": 0.65625,
      "grad_norm": 0.17586955428123474,
      "learning_rate": 6.588235294117647e-06,
      "loss": 0.5034,
      "step": 56
    },
    {
      "epoch": 0.66796875,
      "grad_norm": 0.17159290611743927,
      "learning_rate": 6.705882352941176e-06,
      "loss": 0.5267,
      "step": 57
    },
    {
      "epoch": 0.6796875,
      "grad_norm": 0.17129066586494446,
      "learning_rate": 6.8235294117647065e-06,
      "loss": 0.4634,
      "step": 58
    },
    {
      "epoch": 0.69140625,
      "grad_norm": 0.14943340420722961,
      "learning_rate": 6.941176470588236e-06,
      "loss": 0.4749,
      "step": 59
    },
    {
      "epoch": 0.703125,
      "grad_norm": 0.17984403669834137,
      "learning_rate": 7.058823529411766e-06,
      "loss": 0.5254,
      "step": 60
    },
    {
      "epoch": 0.71484375,
      "grad_norm": 0.15619614720344543,
      "learning_rate": 7.176470588235295e-06,
      "loss": 0.4941,
      "step": 61
    },
    {
      "epoch": 0.7265625,
      "grad_norm": 0.14417926967144012,
      "learning_rate": 7.294117647058823e-06,
      "loss": 0.4234,
      "step": 62
    },
    {
      "epoch": 0.73828125,
      "grad_norm": 0.16936203837394714,
      "learning_rate": 7.4117647058823535e-06,
      "loss": 0.5016,
      "step": 63
    },
    {
      "epoch": 0.75,
      "grad_norm": 0.17044682800769806,
      "learning_rate": 7.529411764705883e-06,
      "loss": 0.4817,
      "step": 64
    },
    {
      "epoch": 0.76171875,
      "grad_norm": 0.1539342850446701,
      "learning_rate": 7.647058823529411e-06,
      "loss": 0.4804,
      "step": 65
    },
    {
      "epoch": 0.7734375,
      "grad_norm": 0.15344035625457764,
      "learning_rate": 7.764705882352941e-06,
      "loss": 0.4875,
      "step": 66
    },
    {
      "epoch": 0.78515625,
      "grad_norm": 0.15734320878982544,
      "learning_rate": 7.882352941176471e-06,
      "loss": 0.436,
      "step": 67
    },
    {
      "epoch": 0.796875,
      "grad_norm": 0.17874149978160858,
      "learning_rate": 8.000000000000001e-06,
      "loss": 0.5315,
      "step": 68
    },
    {
      "epoch": 0.80859375,
      "grad_norm": 0.18438568711280823,
      "learning_rate": 8.11764705882353e-06,
      "loss": 0.5206,
      "step": 69
    },
    {
      "epoch": 0.8203125,
      "grad_norm": 0.16773667931556702,
      "learning_rate": 8.23529411764706e-06,
      "loss": 0.4604,
      "step": 70
    },
    {
      "epoch": 0.83203125,
      "grad_norm": 0.146653413772583,
      "learning_rate": 8.35294117647059e-06,
      "loss": 0.4445,
      "step": 71
    },
    {
      "epoch": 0.84375,
      "grad_norm": 0.1633101850748062,
      "learning_rate": 8.470588235294118e-06,
      "loss": 0.4741,
      "step": 72
    },
    {
      "epoch": 0.85546875,
      "grad_norm": 0.1426013708114624,
      "learning_rate": 8.588235294117647e-06,
      "loss": 0.4447,
      "step": 73
    },
    {
      "epoch": 0.8671875,
      "grad_norm": 0.19708958268165588,
      "learning_rate": 8.705882352941177e-06,
      "loss": 0.4915,
      "step": 74
    },
    {
      "epoch": 0.87890625,
      "grad_norm": 0.13479125499725342,
      "learning_rate": 8.823529411764707e-06,
      "loss": 0.4302,
      "step": 75
    },
    {
      "epoch": 0.890625,
      "grad_norm": 0.16112269461154938,
      "learning_rate": 8.941176470588237e-06,
      "loss": 0.4457,
      "step": 76
    },
    {
      "epoch": 0.90234375,
      "grad_norm": 0.1683078110218048,
      "learning_rate": 9.058823529411765e-06,
      "loss": 0.4617,
      "step": 77
    },
    {
      "epoch": 0.9140625,
      "grad_norm": 0.17289473116397858,
      "learning_rate": 9.176470588235294e-06,
      "loss": 0.4885,
      "step": 78
    },
    {
      "epoch": 0.92578125,
      "grad_norm": 0.14186030626296997,
      "learning_rate": 9.294117647058824e-06,
      "loss": 0.4063,
      "step": 79
    },
    {
      "epoch": 0.9375,
      "grad_norm": 0.17006702721118927,
      "learning_rate": 9.411764705882354e-06,
      "loss": 0.4684,
      "step": 80
    },
    {
      "epoch": 0.94921875,
      "grad_norm": 0.16714246571063995,
      "learning_rate": 9.529411764705882e-06,
      "loss": 0.4416,
      "step": 81
    },
    {
      "epoch": 0.9609375,
      "grad_norm": 0.1373886913061142,
      "learning_rate": 9.647058823529412e-06,
      "loss": 0.4677,
      "step": 82
    },
    {
      "epoch": 0.97265625,
      "grad_norm": 0.15918415784835815,
      "learning_rate": 9.764705882352942e-06,
      "loss": 0.4479,
      "step": 83
    },
    {
      "epoch": 0.984375,
      "grad_norm": 0.12358763813972473,
      "learning_rate": 9.882352941176472e-06,
      "loss": 0.4033,
      "step": 84
    },
    {
      "epoch": 0.99609375,
      "grad_norm": 0.19624371826648712,
      "learning_rate": 1e-05,
      "loss": 0.4579,
      "step": 85
    },
    {
      "epoch": 1.0,
      "grad_norm": 0.19624371826648712,
      "learning_rate": 9.99995783847866e-06,
      "loss": 0.4138,
      "step": 86
    },
    {
      "epoch": 1.01171875,
      "grad_norm": 0.2939195930957794,
      "learning_rate": 9.999831354625678e-06,
      "loss": 0.3971,
      "step": 87
    },
    {
      "epoch": 1.0234375,
      "grad_norm": 0.14151132106781006,
      "learning_rate": 9.999620550574155e-06,
      "loss": 0.4168,
      "step": 88
    },
    {
      "epoch": 1.03515625,
      "grad_norm": 0.174483984708786,
      "learning_rate": 9.999325429879215e-06,
      "loss": 0.3952,
      "step": 89
    },
    {
      "epoch": 1.046875,
      "grad_norm": 0.16392698884010315,
      "learning_rate": 9.998945997517957e-06,
      "loss": 0.4031,
      "step": 90
    },
    {
      "epoch": 1.05859375,
      "grad_norm": 0.13462132215499878,
      "learning_rate": 9.99848225988936e-06,
      "loss": 0.3974,
      "step": 91
    },
    {
      "epoch": 1.0703125,
      "grad_norm": 0.20017389953136444,
      "learning_rate": 9.997934224814173e-06,
      "loss": 0.4054,
      "step": 92
    },
    {
      "epoch": 1.08203125,
      "grad_norm": 0.15260270237922668,
      "learning_rate": 9.997301901534797e-06,
      "loss": 0.43,
      "step": 93
    },
    {
      "epoch": 1.09375,
      "grad_norm": 0.15193504095077515,
      "learning_rate": 9.996585300715117e-06,
      "loss": 0.3885,
      "step": 94
    },
    {
      "epoch": 1.10546875,
      "grad_norm": 0.16993655264377594,
      "learning_rate": 9.99578443444032e-06,
      "loss": 0.4191,
      "step": 95
    },
    {
      "epoch": 1.1171875,
      "grad_norm": 0.15632706880569458,
      "learning_rate": 9.994899316216709e-06,
      "loss": 0.3439,
      "step": 96
    },
    {
      "epoch": 1.12890625,
      "grad_norm": 0.1436368227005005,
      "learning_rate": 9.99392996097145e-06,
      "loss": 0.3609,
      "step": 97
    },
    {
      "epoch": 1.140625,
      "grad_norm": 0.14202651381492615,
      "learning_rate": 9.992876385052346e-06,
      "loss": 0.4133,
      "step": 98
    },
    {
      "epoch": 1.15234375,
      "grad_norm": 0.17068006098270416,
      "learning_rate": 9.991738606227537e-06,
      "loss": 0.3775,
      "step": 99
    },
    {
      "epoch": 1.1640625,
      "grad_norm": 0.15222977101802826,
      "learning_rate": 9.990516643685222e-06,
      "loss": 0.4191,
      "step": 100
    },
    {
      "epoch": 1.17578125,
      "grad_norm": 0.1581757664680481,
      "learning_rate": 9.989210518033316e-06,
      "loss": 0.4098,
      "step": 101
    },
    {
      "epoch": 1.1875,
      "grad_norm": 0.14198894798755646,
      "learning_rate": 9.987820251299121e-06,
      "loss": 0.4031,
      "step": 102
    },
    {
      "epoch": 1.19921875,
      "grad_norm": 0.1587005853652954,
      "learning_rate": 9.98634586692894e-06,
      "loss": 0.3632,
      "step": 103
    },
    {
      "epoch": 1.2109375,
      "grad_norm": 0.1380324512720108,
      "learning_rate": 9.984787389787689e-06,
      "loss": 0.4229,
      "step": 104
    },
    {
      "epoch": 1.22265625,
      "grad_norm": 0.1664944589138031,
      "learning_rate": 9.983144846158472e-06,
      "loss": 0.3952,
      "step": 105
    },
    {
      "epoch": 1.234375,
      "grad_norm": 0.15593458712100983,
      "learning_rate": 9.981418263742148e-06,
      "loss": 0.3657,
      "step": 106
    },
    {
      "epoch": 1.24609375,
      "grad_norm": 0.14692318439483643,
      "learning_rate": 9.979607671656852e-06,
      "loss": 0.3777,
      "step": 107
    },
    {
      "epoch": 1.2578125,
      "grad_norm": 0.19152411818504333,
      "learning_rate": 9.97771310043751e-06,
      "loss": 0.378,
      "step": 108
    },
    {
      "epoch": 1.26953125,
      "grad_norm": 0.1495964378118515,
      "learning_rate": 9.975734582035323e-06,
      "loss": 0.3662,
      "step": 109
    },
    {
      "epoch": 1.28125,
      "grad_norm": 0.14938651025295258,
      "learning_rate": 9.973672149817232e-06,
      "loss": 0.4197,
      "step": 110
    },
    {
      "epoch": 1.29296875,
      "grad_norm": 0.15581446886062622,
      "learning_rate": 9.971525838565348e-06,
      "loss": 0.3804,
      "step": 111
    },
    {
      "epoch": 1.3046875,
      "grad_norm": 0.14621272683143616,
      "learning_rate": 9.96929568447637e-06,
      "loss": 0.3372,
      "step": 112
    },
    {
      "epoch": 1.31640625,
      "grad_norm": 0.14326448738574982,
      "learning_rate": 9.966981725160972e-06,
      "loss": 0.4092,
      "step": 113
    },
    {
      "epoch": 1.328125,
      "grad_norm": 0.1629864126443863,
      "learning_rate": 9.964583999643174e-06,
      "loss": 0.3829,
      "step": 114
    },
    {
      "epoch": 1.33984375,
      "grad_norm": 0.16448885202407837,
      "learning_rate": 9.96210254835968e-06,
      "loss": 0.3952,
      "step": 115
    },
    {
      "epoch": 1.3515625,
      "grad_norm": 0.13906875252723694,
      "learning_rate": 9.95953741315919e-06,
      "loss": 0.3501,
      "step": 116
    },
    {
      "epoch": 1.36328125,
      "grad_norm": 0.1344955563545227,
      "learning_rate": 9.95688863730171e-06,
      "loss": 0.4212,
      "step": 117
    },
    {
      "epoch": 1.375,
      "grad_norm": 0.2054166942834854,
      "learning_rate": 9.954156265457801e-06,
      "loss": 0.4155,
      "step": 118
    },
    {
      "epoch": 1.38671875,
      "grad_norm": 0.14791074395179749,
      "learning_rate": 9.951340343707852e-06,
      "loss": 0.4104,
      "step": 119
    },
    {
      "epoch": 1.3984375,
      "grad_norm": 0.17642416059970856,
      "learning_rate": 9.948440919541277e-06,
      "loss": 0.3502,
      "step": 120
    },
    {
      "epoch": 1.41015625,
      "grad_norm": 0.14525847136974335,
      "learning_rate": 9.945458041855732e-06,
      "loss": 0.383,
      "step": 121
    },
    {
      "epoch": 1.421875,
      "grad_norm": 0.15956953167915344,
      "learning_rate": 9.942391760956277e-06,
      "loss": 0.3864,
      "step": 122
    },
    {
      "epoch": 1.43359375,
      "grad_norm": 0.16518352925777435,
      "learning_rate": 9.939242128554542e-06,
      "loss": 0.4374,
      "step": 123
    },
    {
      "epoch": 1.4453125,
      "grad_norm": 0.15952655673027039,
      "learning_rate": 9.936009197767847e-06,
      "loss": 0.3719,
      "step": 124
    },
    {
      "epoch": 1.45703125,
      "grad_norm": 0.13431760668754578,
      "learning_rate": 9.932693023118299e-06,
      "loss": 0.3693,
      "step": 125
    },
    {
      "epoch": 1.46875,
      "grad_norm": 0.15348534286022186,
      "learning_rate": 9.929293660531889e-06,
      "loss": 0.3708,
      "step": 126
    },
    {
      "epoch": 1.48046875,
      "grad_norm": 0.1592601090669632,
      "learning_rate": 9.925811167337533e-06,
      "loss": 0.3598,
      "step": 127
    },
    {
      "epoch": 1.4921875,
      "grad_norm": 0.16625793278217316,
      "learning_rate": 9.922245602266119e-06,
      "loss": 0.3548,
      "step": 128
    },
    {
      "epoch": 1.50390625,
      "grad_norm": 0.16050715744495392,
      "learning_rate": 9.918597025449505e-06,
      "loss": 0.3732,
      "step": 129
    },
    {
      "epoch": 1.515625,
      "grad_norm": 0.1567108929157257,
      "learning_rate": 9.91486549841951e-06,
      "loss": 0.3875,
      "step": 130
    },
    {
      "epoch": 1.52734375,
      "grad_norm": 0.13900204002857208,
      "learning_rate": 9.911051084106877e-06,
      "loss": 0.3134,
      "step": 131
    },
    {
      "epoch": 1.5390625,
      "grad_norm": 0.17397968471050262,
      "learning_rate": 9.90715384684021e-06,
      "loss": 0.3283,
      "step": 132
    },
    {
      "epoch": 1.55078125,
      "grad_norm": 0.13944688439369202,
      "learning_rate": 9.903173852344889e-06,
      "loss": 0.3801,
      "step": 133
    },
    {
      "epoch": 1.5625,
      "grad_norm": 0.1763840615749359,
      "learning_rate": 9.899111167741966e-06,
      "loss": 0.3426,
      "step": 134
    },
    {
      "epoch": 1.57421875,
      "grad_norm": 0.15428921580314636,
      "learning_rate": 9.894965861547023e-06,
      "loss": 0.4046,
      "step": 135
    },
    {
      "epoch": 1.5859375,
      "grad_norm": 0.18792587518692017,
      "learning_rate": 9.890738003669029e-06,
      "loss": 0.3948,
      "step": 136
    },
    {
      "epoch": 1.59765625,
      "grad_norm": 0.14246441423892975,
      "learning_rate": 9.88642766540915e-06,
      "loss": 0.343,
      "step": 137
    },
    {
      "epoch": 1.609375,
      "grad_norm": 0.14254647493362427,
      "learning_rate": 9.882034919459556e-06,
      "loss": 0.3908,
      "step": 138
    },
    {
      "epoch": 1.62109375,
      "grad_norm": 0.15610483288764954,
      "learning_rate": 9.877559839902185e-06,
      "loss": 0.4127,
      "step": 139
    },
    {
      "epoch": 1.6328125,
      "grad_norm": 0.14824527502059937,
      "learning_rate": 9.873002502207502e-06,
      "loss": 0.3469,
      "step": 140
    },
    {
      "epoch": 1.64453125,
      "grad_norm": 0.13025063276290894,
      "learning_rate": 9.868362983233226e-06,
      "loss": 0.3467,
      "step": 141
    },
    {
      "epoch": 1.65625,
      "grad_norm": 0.12863893806934357,
      "learning_rate": 9.863641361223025e-06,
      "loss": 0.3441,
      "step": 142
    },
    {
      "epoch": 1.66796875,
      "grad_norm": 0.1754492223262787,
      "learning_rate": 9.858837715805207e-06,
      "loss": 0.3893,
      "step": 143
    },
    {
      "epoch": 1.6796875,
      "grad_norm": 0.14538809657096863,
      "learning_rate": 9.853952127991374e-06,
      "loss": 0.361,
      "step": 144
    },
    {
      "epoch": 1.69140625,
      "grad_norm": 0.13839296996593475,
      "learning_rate": 9.848984680175049e-06,
      "loss": 0.4422,
      "step": 145
    },
    {
      "epoch": 1.703125,
      "grad_norm": 0.16836000978946686,
      "learning_rate": 9.843935456130295e-06,
      "loss": 0.3741,
      "step": 146
    },
    {
      "epoch": 1.71484375,
      "grad_norm": 0.1621960699558258,
      "learning_rate": 9.8388045410103e-06,
      "loss": 0.4299,
      "step": 147
    },
    {
      "epoch": 1.7265625,
      "grad_norm": 0.15016399323940277,
      "learning_rate": 9.833592021345938e-06,
      "loss": 0.3613,
      "step": 148
    },
    {
      "epoch": 1.73828125,
      "grad_norm": 0.1778838336467743,
      "learning_rate": 9.828297985044314e-06,
      "loss": 0.4127,
      "step": 149
    },
    {
      "epoch": 1.75,
      "grad_norm": 0.20492997765541077,
      "learning_rate": 9.822922521387277e-06,
      "loss": 0.4566,
      "step": 150
    },
    {
      "epoch": 1.76171875,
      "grad_norm": 0.142822265625,
      "learning_rate": 9.817465721029916e-06,
      "loss": 0.3561,
      "step": 151
    },
    {
      "epoch": 1.7734375,
      "grad_norm": 0.15397801995277405,
      "learning_rate": 9.811927675999035e-06,
      "loss": 0.3854,
      "step": 152
    },
    {
      "epoch": 1.78515625,
      "grad_norm": 0.13618412613868713,
      "learning_rate": 9.806308479691595e-06,
      "loss": 0.3933,
      "step": 153
    },
    {
      "epoch": 1.796875,
      "grad_norm": 0.19063451886177063,
      "learning_rate": 9.800608226873143e-06,
      "loss": 0.382,
      "step": 154
    },
    {
      "epoch": 1.80859375,
      "grad_norm": 0.16610917448997498,
      "learning_rate": 9.794827013676206e-06,
      "loss": 0.402,
      "step": 155
    },
    {
      "epoch": 1.8203125,
      "grad_norm": 0.14735649526119232,
      "learning_rate": 9.788964937598688e-06,
      "loss": 0.3646,
      "step": 156
    },
    {
      "epoch": 1.83203125,
      "grad_norm": 0.1583123356103897,
      "learning_rate": 9.783022097502204e-06,
      "loss": 0.3519,
      "step": 157
    },
    {
      "epoch": 1.84375,
      "grad_norm": 0.14677587151527405,
      "learning_rate": 9.776998593610428e-06,
      "loss": 0.3739,
      "step": 158
    },
    {
      "epoch": 1.85546875,
      "grad_norm": 0.15498070418834686,
      "learning_rate": 9.770894527507393e-06,
      "loss": 0.344,
      "step": 159
    },
    {
      "epoch": 1.8671875,
      "grad_norm": 0.164178267121315,
      "learning_rate": 9.764710002135784e-06,
      "loss": 0.3701,
      "step": 160
    },
    {
      "epoch": 1.87890625,
      "grad_norm": 0.12638989090919495,
      "learning_rate": 9.7584451217952e-06,
      "loss": 0.3985,
      "step": 161
    },
    {
      "epoch": 1.890625,
      "grad_norm": 0.17551939189434052,
      "learning_rate": 9.752099992140401e-06,
      "loss": 0.367,
      "step": 162
    },
    {
      "epoch": 1.90234375,
      "grad_norm": 0.13494940102100372,
      "learning_rate": 9.745674720179507e-06,
      "loss": 0.345,
      "step": 163
    },
    {
      "epoch": 1.9140625,
      "grad_norm": 0.15139806270599365,
      "learning_rate": 9.739169414272219e-06,
      "loss": 0.3767,
      "step": 164
    },
    {
      "epoch": 1.92578125,
      "grad_norm": 0.16584157943725586,
      "learning_rate": 9.732584184127973e-06,
      "loss": 0.4002,
      "step": 165
    },
    {
      "epoch": 1.9375,
      "grad_norm": 0.14180031418800354,
      "learning_rate": 9.7259191408041e-06,
      "loss": 0.3494,
      "step": 166
    },
    {
      "epoch": 1.94921875,
      "grad_norm": 0.15869130194187164,
      "learning_rate": 9.719174396703941e-06,
      "loss": 0.3527,
      "step": 167
    },
    {
      "epoch": 1.9609375,
      "grad_norm": 0.15733729302883148,
      "learning_rate": 9.71235006557497e-06,
      "loss": 0.4277,
      "step": 168
    },
    {
      "epoch": 1.97265625,
      "grad_norm": 0.16619561612606049,
      "learning_rate": 9.705446262506858e-06,
      "loss": 0.398,
      "step": 169
    },
    {
      "epoch": 1.984375,
      "grad_norm": 0.156788632273674,
      "learning_rate": 9.698463103929542e-06,
      "loss": 0.4272,
      "step": 170
    },
    {
      "epoch": 1.99609375,
      "grad_norm": 0.17090734839439392,
      "learning_rate": 9.691400707611258e-06,
      "loss": 0.3683,
      "step": 171
    },
    {
      "epoch": 2.0,
      "grad_norm": 0.17090734839439392,
      "learning_rate": 9.684259192656554e-06,
      "loss": 0.3783,
      "step": 172
    },
    {
      "epoch": 2.01171875,
      "grad_norm": 0.269039511680603,
      "learning_rate": 9.677038679504285e-06,
      "loss": 0.3033,
      "step": 173
    },
    {
      "epoch": 2.0234375,
      "grad_norm": 0.18037453293800354,
      "learning_rate": 9.669739289925578e-06,
      "loss": 0.2647,
      "step": 174
    },
    {
      "epoch": 2.03515625,
      "grad_norm": 0.16502933204174042,
      "learning_rate": 9.66236114702178e-06,
      "loss": 0.2689,
      "step": 175
    },
    {
      "epoch": 2.046875,
      "grad_norm": 0.13461141288280487,
      "learning_rate": 9.654904375222384e-06,
      "loss": 0.299,
      "step": 176
    },
    {
      "epoch": 2.05859375,
      "grad_norm": 0.15441769361495972,
      "learning_rate": 9.647369100282928e-06,
      "loss": 0.2987,
      "step": 177
    },
    {
      "epoch": 2.0703125,
      "grad_norm": 0.22800779342651367,
      "learning_rate": 9.639755449282874e-06,
      "loss": 0.2801,
      "step": 178
    },
    {
      "epoch": 2.08203125,
      "grad_norm": 0.1886795461177826,
      "learning_rate": 9.632063550623465e-06,
      "loss": 0.264,
      "step": 179
    },
    {
      "epoch": 2.09375,
      "grad_norm": 0.16359379887580872,
      "learning_rate": 9.62429353402556e-06,
      "loss": 0.278,
      "step": 180
    },
    {
      "epoch": 2.10546875,
      "grad_norm": 0.1788463145494461,
      "learning_rate": 9.616445530527448e-06,
      "loss": 0.2813,
      "step": 181
    },
    {
      "epoch": 2.1171875,
      "grad_norm": 0.12653109431266785,
      "learning_rate": 9.608519672482635e-06,
      "loss": 0.2997,
      "step": 182
    },
    {
      "epoch": 2.12890625,
      "grad_norm": 0.20340153574943542,
      "learning_rate": 9.600516093557618e-06,
      "loss": 0.3049,
      "step": 183
    },
    {
      "epoch": 2.140625,
      "grad_norm": 0.1629520207643509,
      "learning_rate": 9.592434928729617e-06,
      "loss": 0.2765,
      "step": 184
    },
    {
      "epoch": 2.15234375,
      "grad_norm": 0.17957361042499542,
      "learning_rate": 9.584276314284316e-06,
      "loss": 0.281,
      "step": 185
    },
    {
      "epoch": 2.1640625,
      "grad_norm": 0.13229969143867493,
      "learning_rate": 9.576040387813553e-06,
      "loss": 0.2668,
      "step": 186
    },
    {
      "epoch": 2.17578125,
      "grad_norm": 0.19403038918972015,
      "learning_rate": 9.567727288213005e-06,
      "loss": 0.2529,
      "step": 187
    },
    {
      "epoch": 2.1875,
      "grad_norm": 0.1554100513458252,
      "learning_rate": 9.559337155679843e-06,
      "loss": 0.2756,
      "step": 188
    },
    {
      "epoch": 2.19921875,
      "grad_norm": 0.1644996702671051,
      "learning_rate": 9.550870131710366e-06,
      "loss": 0.2765,
      "step": 189
    },
    {
      "epoch": 2.2109375,
      "grad_norm": 0.16561852395534515,
      "learning_rate": 9.542326359097619e-06,
      "loss": 0.2834,
      "step": 190
    },
    {
      "epoch": 2.22265625,
      "grad_norm": 0.15680573880672455,
      "learning_rate": 9.533705981928984e-06,
      "loss": 0.2611,
      "step": 191
    },
    {
      "epoch": 2.234375,
      "grad_norm": 0.14902271330356598,
      "learning_rate": 9.525009145583746e-06,
      "loss": 0.2759,
      "step": 192
    },
    {
      "epoch": 2.24609375,
      "grad_norm": 0.1453092247247696,
      "learning_rate": 9.516235996730645e-06,
      "loss": 0.2453,
      "step": 193
    },
    {
      "epoch": 2.2578125,
      "grad_norm": 0.15511390566825867,
      "learning_rate": 9.507386683325404e-06,
      "loss": 0.2878,
      "step": 194
    },
    {
      "epoch": 2.26953125,
      "grad_norm": 0.146243616938591,
      "learning_rate": 9.498461354608228e-06,
      "loss": 0.2813,
      "step": 195
    },
    {
      "epoch": 2.28125,
      "grad_norm": 0.13351713120937347,
      "learning_rate": 9.489460161101291e-06,
      "loss": 0.2464,
      "step": 196
    },
    {
      "epoch": 2.29296875,
      "grad_norm": 0.12981724739074707,
      "learning_rate": 9.4803832546062e-06,
      "loss": 0.2364,
      "step": 197
    },
    {
      "epoch": 2.3046875,
      "grad_norm": 0.1395965814590454,
      "learning_rate": 9.471230788201429e-06,
      "loss": 0.2699,
      "step": 198
    },
    {
      "epoch": 2.31640625,
      "grad_norm": 0.13987243175506592,
      "learning_rate": 9.46200291623974e-06,
      "loss": 0.2634,
      "step": 199
    },
    {
      "epoch": 2.328125,
      "grad_norm": 0.17181117832660675,
      "learning_rate": 9.452699794345583e-06,
      "loss": 0.2501,
      "step": 200
    },
    {
      "epoch": 2.33984375,
      "grad_norm": 0.12983955442905426,
      "learning_rate": 9.443321579412465e-06,
      "loss": 0.2383,
      "step": 201
    },
    {
      "epoch": 2.3515625,
      "grad_norm": 0.17716065049171448,
      "learning_rate": 9.43386842960031e-06,
      "loss": 0.3096,
      "step": 202
    },
    {
      "epoch": 2.36328125,
      "grad_norm": 0.16192656755447388,
      "learning_rate": 9.42434050433279e-06,
      "loss": 0.2825,
      "step": 203
    },
    {
      "epoch": 2.375,
      "grad_norm": 0.1350512057542801,
      "learning_rate": 9.414737964294636e-06,
      "loss": 0.2301,
      "step": 204
    },
    {
      "epoch": 2.38671875,
      "grad_norm": 0.1251179575920105,
      "learning_rate": 9.405060971428924e-06,
      "loss": 0.2682,
      "step": 205
    },
    {
      "epoch": 2.3984375,
      "grad_norm": 0.13068172335624695,
      "learning_rate": 9.39530968893435e-06,
      "loss": 0.3303,
      "step": 206
    },
    {
      "epoch": 2.41015625,
      "grad_norm": 0.16019192337989807,
      "learning_rate": 9.38548428126248e-06,
      "loss": 0.2651,
      "step": 207
    },
    {
      "epoch": 2.421875,
      "grad_norm": 0.14367878437042236,
      "learning_rate": 9.375584914114963e-06,
      "loss": 0.2712,
      "step": 208
    },
    {
      "epoch": 2.43359375,
      "grad_norm": 0.14105841517448425,
      "learning_rate": 9.365611754440756e-06,
      "loss": 0.2541,
      "step": 209
    },
    {
      "epoch": 2.4453125,
      "grad_norm": 0.14219817519187927,
      "learning_rate": 9.355564970433288e-06,
      "loss": 0.2874,
      "step": 210
    },
    {
      "epoch": 2.45703125,
      "grad_norm": 0.14144977927207947,
      "learning_rate": 9.345444731527642e-06,
      "loss": 0.255,
      "step": 211
    },
    {
      "epoch": 2.46875,
      "grad_norm": 0.16394628584384918,
      "learning_rate": 9.335251208397684e-06,
      "loss": 0.2946,
      "step": 212
    },
    {
      "epoch": 2.48046875,
      "grad_norm": 0.1422003209590912,
      "learning_rate": 9.32498457295319e-06,
      "loss": 0.275,
      "step": 213
    },
    {
      "epoch": 2.4921875,
      "grad_norm": 0.153361514210701,
      "learning_rate": 9.31464499833695e-06,
      "loss": 0.2695,
      "step": 214
    },
    {
      "epoch": 2.50390625,
      "grad_norm": 0.13666994869709015,
      "learning_rate": 9.30423265892184e-06,
      "loss": 0.2708,
      "step": 215
    },
    {
      "epoch": 2.515625,
      "grad_norm": 0.13980181515216827,
      "learning_rate": 9.29374773030789e-06,
      "loss": 0.3132,
      "step": 216
    },
    {
      "epoch": 2.52734375,
      "grad_norm": 0.1572592854499817,
      "learning_rate": 9.283190389319315e-06,
      "loss": 0.2802,
      "step": 217
    },
    {
      "epoch": 2.5390625,
      "grad_norm": 0.12849898636341095,
      "learning_rate": 9.27256081400154e-06,
      "loss": 0.2813,
      "step": 218
    },
    {
      "epoch": 2.55078125,
      "grad_norm": 0.14952799677848816,
      "learning_rate": 9.26185918361819e-06,
      "loss": 0.2662,
      "step": 219
    },
    {
      "epoch": 2.5625,
      "grad_norm": 0.1414070427417755,
      "learning_rate": 9.251085678648072e-06,
      "loss": 0.2833,
      "step": 220
    },
    {
      "epoch": 2.57421875,
      "grad_norm": 0.14631874859333038,
      "learning_rate": 9.24024048078213e-06,
      "loss": 0.2645,
      "step": 221
    },
    {
      "epoch": 2.5859375,
      "grad_norm": 0.16351047158241272,
      "learning_rate": 9.229323772920383e-06,
      "loss": 0.2563,
      "step": 222
    },
    {
      "epoch": 2.59765625,
      "grad_norm": 0.1420244574546814,
      "learning_rate": 9.218335739168833e-06,
      "loss": 0.3132,
      "step": 223
    },
    {
      "epoch": 2.609375,
      "grad_norm": 0.16125503182411194,
      "learning_rate": 9.207276564836367e-06,
      "loss": 0.2533,
      "step": 224
    },
    {
      "epoch": 2.62109375,
      "grad_norm": 0.16347342729568481,
      "learning_rate": 9.196146436431635e-06,
      "loss": 0.2884,
      "step": 225
    },
    {
      "epoch": 2.6328125,
      "grad_norm": 0.15693005919456482,
      "learning_rate": 9.18494554165989e-06,
      "loss": 0.2644,
      "step": 226
    },
    {
      "epoch": 2.64453125,
      "grad_norm": 0.15239854156970978,
      "learning_rate": 9.173674069419843e-06,
      "loss": 0.259,
      "step": 227
    },
    {
      "epoch": 2.65625,
      "grad_norm": 0.14987824857234955,
      "learning_rate": 9.162332209800455e-06,
      "loss": 0.2792,
      "step": 228
    },
    {
      "epoch": 2.66796875,
      "grad_norm": 0.14547227323055267,
      "learning_rate": 9.150920154077753e-06,
      "loss": 0.2663,
      "step": 229
    },
    {
      "epoch": 2.6796875,
      "grad_norm": 0.15145206451416016,
      "learning_rate": 9.13943809471159e-06,
      "loss": 0.2885,
      "step": 230
    },
    {
      "epoch": 2.69140625,
      "grad_norm": 0.17260988056659698,
      "learning_rate": 9.1278862253424e-06,
      "loss": 0.3377,
      "step": 231
    },
    {
      "epoch": 2.703125,
      "grad_norm": 0.1483244150876999,
      "learning_rate": 9.116264740787937e-06,
      "loss": 0.2742,
      "step": 232
    },
    {
      "epoch": 2.71484375,
      "grad_norm": 0.13184918463230133,
      "learning_rate": 9.104573837039992e-06,
      "loss": 0.2746,
      "step": 233
    },
    {
      "epoch": 2.7265625,
      "grad_norm": 0.14674793183803558,
      "learning_rate": 9.092813711261075e-06,
      "loss": 0.2372,
      "step": 234
    },
    {
      "epoch": 2.73828125,
      "grad_norm": 0.13162650167942047,
      "learning_rate": 9.08098456178111e-06,
      "loss": 0.287,
      "step": 235
    },
    {
      "epoch": 2.75,
      "grad_norm": 0.14429397881031036,
      "learning_rate": 9.069086588094067e-06,
      "loss": 0.2561,
      "step": 236
    },
    {
      "epoch": 2.76171875,
      "grad_norm": 0.1422855108976364,
      "learning_rate": 9.057119990854617e-06,
      "loss": 0.3037,
      "step": 237
    },
    {
      "epoch": 2.7734375,
      "grad_norm": 0.15853242576122284,
      "learning_rate": 9.045084971874738e-06,
      "loss": 0.2888,
      "step": 238
    },
    {
      "epoch": 2.78515625,
      "grad_norm": 0.12172834575176239,
      "learning_rate": 9.032981734120312e-06,
      "loss": 0.2477,
      "step": 239
    },
    {
      "epoch": 2.796875,
      "grad_norm": 0.1423560231924057,
      "learning_rate": 9.020810481707709e-06,
      "loss": 0.2867,
      "step": 240
    },
    {
      "epoch": 2.80859375,
      "grad_norm": 0.1702727973461151,
      "learning_rate": 9.008571419900334e-06,
      "loss": 0.3015,
      "step": 241
    },
    {
      "epoch": 2.8203125,
      "grad_norm": 0.14154493808746338,
      "learning_rate": 8.996264755105173e-06,
      "loss": 0.2771,
      "step": 242
    },
    {
      "epoch": 2.83203125,
      "grad_norm": 0.14596149325370789,
      "learning_rate": 8.983890694869312e-06,
      "loss": 0.3017,
      "step": 243
    },
    {
      "epoch": 2.84375,
      "grad_norm": 0.13621939718723297,
      "learning_rate": 8.97144944787643e-06,
      "loss": 0.2496,
      "step": 244
    },
    {
      "epoch": 2.85546875,
      "grad_norm": 0.18428286910057068,
      "learning_rate": 8.958941223943292e-06,
      "loss": 0.2751,
      "step": 245
    },
    {
      "epoch": 2.8671875,
      "grad_norm": 0.1273455023765564,
      "learning_rate": 8.946366234016192e-06,
      "loss": 0.2678,
      "step": 246
    },
    {
      "epoch": 2.87890625,
      "grad_norm": 0.1442483514547348,
      "learning_rate": 8.933724690167417e-06,
      "loss": 0.2574,
      "step": 247
    },
    {
      "epoch": 2.890625,
      "grad_norm": 0.12419985979795456,
      "learning_rate": 8.921016805591654e-06,
      "loss": 0.2512,
      "step": 248
    },
    {
      "epoch": 2.90234375,
      "grad_norm": 0.13012221455574036,
      "learning_rate": 8.908242794602401e-06,
      "loss": 0.2674,
      "step": 249
    },
    {
      "epoch": 2.9140625,
      "grad_norm": 0.13028892874717712,
      "learning_rate": 8.895402872628352e-06,
      "loss": 0.2566,
      "step": 250
    },
    {
      "epoch": 2.92578125,
      "grad_norm": 0.1266089677810669,
      "learning_rate": 8.882497256209767e-06,
      "loss": 0.2671,
      "step": 251
    },
    {
      "epoch": 2.9375,
      "grad_norm": 0.15992388129234314,
      "learning_rate": 8.869526162994814e-06,
      "loss": 0.276,
      "step": 252
    },
    {
      "epoch": 2.94921875,
      "grad_norm": 0.13113290071487427,
      "learning_rate": 8.856489811735904e-06,
      "loss": 0.3073,
      "step": 253
    },
    {
      "epoch": 2.9609375,
      "grad_norm": 0.15523600578308105,
      "learning_rate": 8.843388422285995e-06,
      "loss": 0.2704,
      "step": 254
    },
    {
      "epoch": 2.97265625,
      "grad_norm": 0.13755646347999573,
      "learning_rate": 8.83022221559489e-06,
      "loss": 0.3174,
      "step": 255
    },
    {
      "epoch": 2.984375,
      "grad_norm": 0.15444748103618622,
      "learning_rate": 8.816991413705515e-06,
      "loss": 0.2955,
      "step": 256
    },
    {
      "epoch": 2.99609375,
      "grad_norm": 0.13223521411418915,
      "learning_rate": 8.803696239750163e-06,
      "loss": 0.2563,
      "step": 257
    },
    {
      "epoch": 3.0,
      "grad_norm": 0.28410595655441284,
      "learning_rate": 8.790336917946737e-06,
      "loss": 0.242,
      "step": 258
    },
    {
      "epoch": 3.01171875,
      "grad_norm": 0.16228830814361572,
      "learning_rate": 8.776913673594968e-06,
      "loss": 0.1626,
      "step": 259
    },
    {
      "epoch": 3.0234375,
      "grad_norm": 0.14956367015838623,
      "learning_rate": 8.763426733072624e-06,
      "loss": 0.1751,
      "step": 260
    },
    {
      "epoch": 3.03515625,
      "grad_norm": 0.15641574561595917,
      "learning_rate": 8.74987632383167e-06,
      "loss": 0.1932,
      "step": 261
    },
    {
      "epoch": 3.046875,
      "grad_norm": 0.18344224989414215,
      "learning_rate": 8.736262674394455e-06,
      "loss": 0.1646,
      "step": 262
    },
    {
      "epoch": 3.05859375,
      "grad_norm": 0.23181785643100739,
      "learning_rate": 8.722586014349851e-06,
      "loss": 0.1855,
      "step": 263
    },
    {
      "epoch": 3.0703125,
      "grad_norm": 0.1877327412366867,
      "learning_rate": 8.708846574349372e-06,
      "loss": 0.1779,
      "step": 264
    },
    {
      "epoch": 3.08203125,
      "grad_norm": 0.1676245629787445,
      "learning_rate": 8.695044586103297e-06,
      "loss": 0.1621,
      "step": 265
    },
    {
      "epoch": 3.09375,
      "grad_norm": 0.16094529628753662,
      "learning_rate": 8.681180282376754e-06,
      "loss": 0.1732,
      "step": 266
    },
    {
      "epoch": 3.10546875,
      "grad_norm": 0.14831165969371796,
      "learning_rate": 8.667253896985796e-06,
      "loss": 0.1678,
      "step": 267
    },
    {
      "epoch": 3.1171875,
      "grad_norm": 0.1500885933637619,
      "learning_rate": 8.653265664793466e-06,
      "loss": 0.1597,
      "step": 268
    },
    {
      "epoch": 3.12890625,
      "grad_norm": 0.13528233766555786,
      "learning_rate": 8.639215821705821e-06,
      "loss": 0.2005,
      "step": 269
    },
    {
      "epoch": 3.140625,
      "grad_norm": 0.1544928252696991,
      "learning_rate": 8.625104604667965e-06,
      "loss": 0.1658,
      "step": 270
    },
    {
      "epoch": 3.15234375,
      "grad_norm": 0.1660798341035843,
      "learning_rate": 8.610932251660046e-06,
      "loss": 0.2022,
      "step": 271
    },
    {
      "epoch": 3.1640625,
      "grad_norm": 0.17334631085395813,
      "learning_rate": 8.596699001693257e-06,
      "loss": 0.1821,
      "step": 272
    },
    {
      "epoch": 3.17578125,
      "grad_norm": 0.23266500234603882,
      "learning_rate": 8.58240509480578e-06,
      "loss": 0.1905,
      "step": 273
    },
    {
      "epoch": 3.1875,
      "grad_norm": 0.14414437115192413,
      "learning_rate": 8.568050772058763e-06,
      "loss": 0.1792,
      "step": 274
    },
    {
      "epoch": 3.19921875,
      "grad_norm": 0.1535835564136505,
      "learning_rate": 8.553636275532236e-06,
      "loss": 0.1699,
      "step": 275
    },
    {
      "epoch": 3.2109375,
      "grad_norm": 0.17544692754745483,
      "learning_rate": 8.539161848321047e-06,
      "loss": 0.216,
      "step": 276
    },
    {
      "epoch": 3.22265625,
      "grad_norm": 0.14662209153175354,
      "learning_rate": 8.524627734530738e-06,
      "loss": 0.1655,
      "step": 277
    },
    {
      "epoch": 3.234375,
      "grad_norm": 0.14544612169265747,
      "learning_rate": 8.51003417927345e-06,
      "loss": 0.1764,
      "step": 278
    },
    {
      "epoch": 3.24609375,
      "grad_norm": 0.16317211091518402,
      "learning_rate": 8.49538142866378e-06,
      "loss": 0.1791,
      "step": 279
    },
    {
      "epoch": 3.2578125,
      "grad_norm": 0.1408390998840332,
      "learning_rate": 8.480669729814635e-06,
      "loss": 0.1571,
      "step": 280
    },
    {
      "epoch": 3.26953125,
      "grad_norm": 0.14905256032943726,
      "learning_rate": 8.465899330833051e-06,
      "loss": 0.1513,
      "step": 281
    },
    {
      "epoch": 3.28125,
      "grad_norm": 0.13584789633750916,
      "learning_rate": 8.451070480816027e-06,
      "loss": 0.1677,
      "step": 282
    },
    {
      "epoch": 3.29296875,
      "grad_norm": 0.14696229994297028,
      "learning_rate": 8.436183429846314e-06,
      "loss": 0.1812,
      "step": 283
    },
    {
      "epoch": 3.3046875,
      "grad_norm": 0.21364903450012207,
      "learning_rate": 8.421238428988199e-06,
      "loss": 0.1875,
      "step": 284
    },
    {
      "epoch": 3.31640625,
      "grad_norm": 0.15450192987918854,
      "learning_rate": 8.40623573028327e-06,
      "loss": 0.1763,
      "step": 285
    },
    {
      "epoch": 3.328125,
      "grad_norm": 0.14626288414001465,
      "learning_rate": 8.39117558674617e-06,
      "loss": 0.1625,
      "step": 286
    },
    {
      "epoch": 3.33984375,
      "grad_norm": 0.15964478254318237,
      "learning_rate": 8.376058252360322e-06,
      "loss": 0.1896,
      "step": 287
    },
    {
      "epoch": 3.3515625,
      "grad_norm": 0.14050042629241943,
      "learning_rate": 8.360883982073653e-06,
      "loss": 0.1695,
      "step": 288
    },
    {
      "epoch": 3.36328125,
      "grad_norm": 0.16250889003276825,
      "learning_rate": 8.345653031794292e-06,
      "loss": 0.1638,
      "step": 289
    },
    {
      "epoch": 3.375,
      "grad_norm": 0.14831937849521637,
      "learning_rate": 8.330365658386252e-06,
      "loss": 0.1714,
      "step": 290
    },
    {
      "epoch": 3.38671875,
      "grad_norm": 0.15942546725273132,
      "learning_rate": 8.3150221196651e-06,
      "loss": 0.1744,
      "step": 291
    },
    {
      "epoch": 3.3984375,
      "grad_norm": 0.14313259720802307,
      "learning_rate": 8.299622674393615e-06,
      "loss": 0.1578,
      "step": 292
    },
    {
      "epoch": 3.41015625,
      "grad_norm": 0.1886913925409317,
      "learning_rate": 8.284167582277406e-06,
      "loss": 0.2234,
      "step": 293
    },
    {
      "epoch": 3.421875,
      "grad_norm": 0.16603605449199677,
      "learning_rate": 8.268657103960558e-06,
      "loss": 0.1891,
      "step": 294
    },
    {
      "epoch": 3.43359375,
      "grad_norm": 0.18278054893016815,
      "learning_rate": 8.25309150102121e-06,
      "loss": 0.1847,
      "step": 295
    },
    {
      "epoch": 3.4453125,
      "grad_norm": 0.15222598612308502,
      "learning_rate": 8.237471035967168e-06,
      "loss": 0.181,
      "step": 296
    },
    {
      "epoch": 3.45703125,
      "grad_norm": 0.16404354572296143,
      "learning_rate": 8.221795972231459e-06,
      "loss": 0.1682,
      "step": 297
    },
    {
      "epoch": 3.46875,
      "grad_norm": 0.15001647174358368,
      "learning_rate": 8.206066574167893e-06,
      "loss": 0.1589,
      "step": 298
    },
    {
      "epoch": 3.48046875,
      "grad_norm": 0.1620359718799591,
      "learning_rate": 8.190283107046613e-06,
      "loss": 0.1914,
      "step": 299
    },
    {
      "epoch": 3.4921875,
      "grad_norm": 0.17575767636299133,
      "learning_rate": 8.174445837049614e-06,
      "loss": 0.1713,
      "step": 300
    },
    {
      "epoch": 3.50390625,
      "grad_norm": 0.16479137539863586,
      "learning_rate": 8.158555031266255e-06,
      "loss": 0.1958,
      "step": 301
    },
    {
      "epoch": 3.515625,
      "grad_norm": 0.14474551379680634,
      "learning_rate": 8.142610957688755e-06,
      "loss": 0.1741,
      "step": 302
    },
    {
      "epoch": 3.52734375,
      "grad_norm": 0.15629877150058746,
      "learning_rate": 8.12661388520767e-06,
      "loss": 0.1925,
      "step": 303
    },
    {
      "epoch": 3.5390625,
      "grad_norm": 0.21056914329528809,
      "learning_rate": 8.110564083607371e-06,
      "loss": 0.1776,
      "step": 304
    },
    {
      "epoch": 3.55078125,
      "grad_norm": 0.15697474777698517,
      "learning_rate": 8.094461823561473e-06,
      "loss": 0.16,
      "step": 305
    },
    {
      "epoch": 3.5625,
      "grad_norm": 0.15944577753543854,
      "learning_rate": 8.078307376628292e-06,
      "loss": 0.1784,
      "step": 306
    },
    {
      "epoch": 3.57421875,
      "grad_norm": 0.1611972153186798,
      "learning_rate": 8.06210101524625e-06,
      "loss": 0.1914,
      "step": 307
    },
    {
      "epoch": 3.5859375,
      "grad_norm": 0.18928834795951843,
      "learning_rate": 8.045843012729288e-06,
      "loss": 0.2047,
      "step": 308
    },
    {
      "epoch": 3.59765625,
      "grad_norm": 0.17320488393306732,
      "learning_rate": 8.029533643262257e-06,
      "loss": 0.1882,
      "step": 309
    },
    {
      "epoch": 3.609375,
      "grad_norm": 0.1613771617412567,
      "learning_rate": 8.013173181896283e-06,
      "loss": 0.1814,
      "step": 310
    },
    {
      "epoch": 3.62109375,
      "grad_norm": 0.15916645526885986,
      "learning_rate": 7.996761904544146e-06,
      "loss": 0.1905,
      "step": 311
    },
    {
      "epoch": 3.6328125,
      "grad_norm": 0.1529555767774582,
      "learning_rate": 7.980300087975612e-06,
      "loss": 0.1835,
      "step": 312
    },
    {
      "epoch": 3.64453125,
      "grad_norm": 0.16544155776500702,
      "learning_rate": 7.963788009812775e-06,
      "loss": 0.1791,
      "step": 313
    },
    {
      "epoch": 3.65625,
      "grad_norm": 0.1579754799604416,
      "learning_rate": 7.94722594852537e-06,
      "loss": 0.1943,
      "step": 314
    },
    {
      "epoch": 3.66796875,
      "grad_norm": 0.16517029702663422,
      "learning_rate": 7.930614183426074e-06,
      "loss": 0.1949,
      "step": 315
    },
    {
      "epoch": 3.6796875,
      "grad_norm": 0.19699627161026,
      "learning_rate": 7.913952994665805e-06,
      "loss": 0.1816,
      "step": 316
    },
    {
      "epoch": 3.69140625,
      "grad_norm": 0.15589186549186707,
      "learning_rate": 7.89724266322899e-06,
      "loss": 0.1857,
      "step": 317
    },
    {
      "epoch": 3.703125,
      "grad_norm": 0.15042664110660553,
      "learning_rate": 7.880483470928823e-06,
      "loss": 0.1659,
      "step": 318
    },
    {
      "epoch": 3.71484375,
      "grad_norm": 0.1443316787481308,
      "learning_rate": 7.863675700402527e-06,
      "loss": 0.1741,
      "step": 319
    },
    {
      "epoch": 3.7265625,
      "grad_norm": 0.14981609582901,
      "learning_rate": 7.846819635106569e-06,
      "loss": 0.1998,
      "step": 320
    },
    {
      "epoch": 3.73828125,
      "grad_norm": 0.13699378073215485,
      "learning_rate": 7.829915559311892e-06,
      "loss": 0.1636,
      "step": 321
    },
    {
      "epoch": 3.75,
      "grad_norm": 0.14549602568149567,
      "learning_rate": 7.812963758099118e-06,
      "loss": 0.1732,
      "step": 322
    },
    {
      "epoch": 3.76171875,
      "grad_norm": 0.15110832452774048,
      "learning_rate": 7.795964517353734e-06,
      "loss": 0.1816,
      "step": 323
    },
    {
      "epoch": 3.7734375,
      "grad_norm": 0.1504000872373581,
      "learning_rate": 7.778918123761287e-06,
      "loss": 0.1656,
      "step": 324
    },
    {
      "epoch": 3.78515625,
      "grad_norm": 0.15452761948108673,
      "learning_rate": 7.76182486480253e-06,
      "loss": 0.1764,
      "step": 325
    },
    {
      "epoch": 3.796875,
      "grad_norm": 0.14827081561088562,
      "learning_rate": 7.744685028748582e-06,
      "loss": 0.159,
      "step": 326
    },
    {
      "epoch": 3.80859375,
      "grad_norm": 0.140132874250412,
      "learning_rate": 7.72749890465607e-06,
      "loss": 0.1923,
      "step": 327
    },
    {
      "epoch": 3.8203125,
      "grad_norm": 0.1527477502822876,
      "learning_rate": 7.710266782362248e-06,
      "loss": 0.1586,
      "step": 328
    },
    {
      "epoch": 3.83203125,
      "grad_norm": 0.14466483891010284,
      "learning_rate": 7.692988952480114e-06,
      "loss": 0.1669,
      "step": 329
    },
    {
      "epoch": 3.84375,
      "grad_norm": 0.15228520333766937,
      "learning_rate": 7.675665706393502e-06,
      "loss": 0.167,
      "step": 330
    },
    {
      "epoch": 3.85546875,
      "grad_norm": 0.18122923374176025,
      "learning_rate": 7.658297336252181e-06,
      "loss": 0.1981,
      "step": 331
    },
    {
      "epoch": 3.8671875,
      "grad_norm": 0.1492803692817688,
      "learning_rate": 7.64088413496691e-06,
      "loss": 0.1802,
      "step": 332
    },
    {
      "epoch": 3.87890625,
      "grad_norm": 0.15203335881233215,
      "learning_rate": 7.623426396204516e-06,
      "loss": 0.1935,
      "step": 333
    },
    {
      "epoch": 3.890625,
      "grad_norm": 0.15901945531368256,
      "learning_rate": 7.605924414382926e-06,
      "loss": 0.1711,
      "step": 334
    },
    {
      "epoch": 3.90234375,
      "grad_norm": 0.14871086180210114,
      "learning_rate": 7.588378484666214e-06,
      "loss": 0.1658,
      "step": 335
    },
    {
      "epoch": 3.9140625,
      "grad_norm": 0.16054388880729675,
      "learning_rate": 7.570788902959612e-06,
      "loss": 0.1915,
      "step": 336
    },
    {
      "epoch": 3.92578125,
      "grad_norm": 0.14552587270736694,
      "learning_rate": 7.553155965904535e-06,
      "loss": 0.1901,
      "step": 337
    },
    {
      "epoch": 3.9375,
      "grad_norm": 0.15138490498065948,
      "learning_rate": 7.535479970873563e-06,
      "loss": 0.2054,
      "step": 338
    },
    {
      "epoch": 3.94921875,
      "grad_norm": 0.15584439039230347,
      "learning_rate": 7.517761215965429e-06,
      "loss": 0.1863,
      "step": 339
    },
    {
      "epoch": 3.9609375,
      "grad_norm": 0.15878607332706451,
      "learning_rate": 7.500000000000001e-06,
      "loss": 0.2298,
      "step": 340
    },
    {
      "epoch": 3.97265625,
      "grad_norm": 0.14984527230262756,
      "learning_rate": 7.482196622513233e-06,
      "loss": 0.1862,
      "step": 341
    },
    {
      "epoch": 3.984375,
      "grad_norm": 0.1370866894721985,
      "learning_rate": 7.464351383752117e-06,
      "loss": 0.1711,
      "step": 342
    }
  ],
  "logging_steps": 1,
  "max_steps": 850,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 171,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 73223969505280.0,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}