{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 5.994876571960876,
  "eval_steps": 500,
  "global_step": 6438,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.004657661853749418,
      "grad_norm": 7.316119501061427,
      "learning_rate": 3.8819875776397516e-07,
      "loss": 1.3166,
      "step": 5
    },
    {
      "epoch": 0.009315323707498836,
      "grad_norm": 7.205695829555083,
      "learning_rate": 7.763975155279503e-07,
      "loss": 1.3011,
      "step": 10
    },
    {
      "epoch": 0.013972985561248253,
      "grad_norm": 6.7877234577760275,
      "learning_rate": 1.1645962732919254e-06,
      "loss": 1.2836,
      "step": 15
    },
    {
      "epoch": 0.018630647414997672,
      "grad_norm": 5.1979422766233405,
      "learning_rate": 1.5527950310559006e-06,
      "loss": 1.2395,
      "step": 20
    },
    {
      "epoch": 0.02328830926874709,
      "grad_norm": 2.9164864071984806,
      "learning_rate": 1.940993788819876e-06,
      "loss": 1.1531,
      "step": 25
    },
    {
      "epoch": 0.027945971122496506,
      "grad_norm": 1.797444815824453,
      "learning_rate": 2.329192546583851e-06,
      "loss": 1.0999,
      "step": 30
    },
    {
      "epoch": 0.032603632976245925,
      "grad_norm": 1.7216499673175123,
      "learning_rate": 2.7173913043478263e-06,
      "loss": 1.0346,
      "step": 35
    },
    {
      "epoch": 0.037261294829995344,
      "grad_norm": 1.2958826673540167,
      "learning_rate": 3.1055900621118013e-06,
      "loss": 0.9875,
      "step": 40
    },
    {
      "epoch": 0.04191895668374476,
      "grad_norm": 0.8071295960793549,
      "learning_rate": 3.4937888198757763e-06,
      "loss": 0.8896,
      "step": 45
    },
    {
      "epoch": 0.04657661853749418,
      "grad_norm": 0.7485172775167197,
      "learning_rate": 3.881987577639752e-06,
      "loss": 0.8678,
      "step": 50
    },
    {
      "epoch": 0.05123428039124359,
      "grad_norm": 0.554726583009313,
      "learning_rate": 4.270186335403727e-06,
      "loss": 0.8284,
      "step": 55
    },
    {
      "epoch": 0.05589194224499301,
      "grad_norm": 0.5034661299280798,
      "learning_rate": 4.658385093167702e-06,
      "loss": 0.8054,
      "step": 60
    },
    {
      "epoch": 0.06054960409874243,
      "grad_norm": 0.4523685186926289,
      "learning_rate": 5.046583850931677e-06,
      "loss": 0.7796,
      "step": 65
    },
    {
      "epoch": 0.06520726595249185,
      "grad_norm": 0.41946665247072396,
      "learning_rate": 5.4347826086956525e-06,
      "loss": 0.7426,
      "step": 70
    },
    {
      "epoch": 0.06986492780624126,
      "grad_norm": 0.38996703013472583,
      "learning_rate": 5.8229813664596275e-06,
      "loss": 0.7344,
      "step": 75
    },
    {
      "epoch": 0.07452258965999069,
      "grad_norm": 0.4334686982341064,
      "learning_rate": 6.2111801242236025e-06,
      "loss": 0.7412,
      "step": 80
    },
    {
      "epoch": 0.0791802515137401,
      "grad_norm": 0.37086473549666105,
      "learning_rate": 6.5993788819875775e-06,
      "loss": 0.7061,
      "step": 85
    },
    {
      "epoch": 0.08383791336748952,
      "grad_norm": 0.3762145216343164,
      "learning_rate": 6.9875776397515525e-06,
      "loss": 0.708,
      "step": 90
    },
    {
      "epoch": 0.08849557522123894,
      "grad_norm": 0.3900837575897009,
      "learning_rate": 7.375776397515528e-06,
      "loss": 0.7118,
      "step": 95
    },
    {
      "epoch": 0.09315323707498836,
      "grad_norm": 0.4162744908945877,
      "learning_rate": 7.763975155279503e-06,
      "loss": 0.6936,
      "step": 100
    },
    {
      "epoch": 0.09781089892873777,
      "grad_norm": 0.36212861683111336,
      "learning_rate": 8.15217391304348e-06,
      "loss": 0.6951,
      "step": 105
    },
    {
      "epoch": 0.10246856078248719,
      "grad_norm": 0.372162946877714,
      "learning_rate": 8.540372670807453e-06,
      "loss": 0.6762,
      "step": 110
    },
    {
      "epoch": 0.10712622263623661,
      "grad_norm": 0.3709604764884615,
      "learning_rate": 8.92857142857143e-06,
      "loss": 0.6722,
      "step": 115
    },
    {
      "epoch": 0.11178388448998602,
      "grad_norm": 0.4076537865962986,
      "learning_rate": 9.316770186335403e-06,
      "loss": 0.6829,
      "step": 120
    },
    {
      "epoch": 0.11644154634373545,
      "grad_norm": 0.4103270175358574,
      "learning_rate": 9.70496894409938e-06,
      "loss": 0.6543,
      "step": 125
    },
    {
      "epoch": 0.12109920819748486,
      "grad_norm": 0.3853312352357345,
      "learning_rate": 1.0093167701863353e-05,
      "loss": 0.656,
      "step": 130
    },
    {
      "epoch": 0.1257568700512343,
      "grad_norm": 0.4279086843532676,
      "learning_rate": 1.048136645962733e-05,
      "loss": 0.6632,
      "step": 135
    },
    {
      "epoch": 0.1304145319049837,
      "grad_norm": 0.44803399637948954,
      "learning_rate": 1.0869565217391305e-05,
      "loss": 0.6521,
      "step": 140
    },
    {
      "epoch": 0.1350721937587331,
      "grad_norm": 0.4318949555258277,
      "learning_rate": 1.1257763975155281e-05,
      "loss": 0.6633,
      "step": 145
    },
    {
      "epoch": 0.13972985561248252,
      "grad_norm": 0.47395151207140046,
      "learning_rate": 1.1645962732919255e-05,
      "loss": 0.653,
      "step": 150
    },
    {
      "epoch": 0.14438751746623196,
      "grad_norm": 0.4612165533509867,
      "learning_rate": 1.2034161490683231e-05,
      "loss": 0.6289,
      "step": 155
    },
    {
      "epoch": 0.14904517931998137,
      "grad_norm": 0.48544782141567877,
      "learning_rate": 1.2422360248447205e-05,
      "loss": 0.6353,
      "step": 160
    },
    {
      "epoch": 0.1537028411737308,
      "grad_norm": 0.44421285478890205,
      "learning_rate": 1.2810559006211181e-05,
      "loss": 0.6354,
      "step": 165
    },
    {
      "epoch": 0.1583605030274802,
      "grad_norm": 0.5212944614649248,
      "learning_rate": 1.3198757763975155e-05,
      "loss": 0.6318,
      "step": 170
    },
    {
      "epoch": 0.1630181648812296,
      "grad_norm": 0.5125255393571805,
      "learning_rate": 1.3586956521739131e-05,
      "loss": 0.643,
      "step": 175
    },
    {
      "epoch": 0.16767582673497905,
      "grad_norm": 0.47654781034618465,
      "learning_rate": 1.3975155279503105e-05,
      "loss": 0.6354,
      "step": 180
    },
    {
      "epoch": 0.17233348858872846,
      "grad_norm": 0.5163976441296857,
      "learning_rate": 1.4363354037267083e-05,
      "loss": 0.6317,
      "step": 185
    },
    {
      "epoch": 0.17699115044247787,
      "grad_norm": 0.4227705180370943,
      "learning_rate": 1.4751552795031057e-05,
      "loss": 0.6241,
      "step": 190
    },
    {
      "epoch": 0.18164881229622729,
      "grad_norm": 0.5306979617290407,
      "learning_rate": 1.5139751552795031e-05,
      "loss": 0.6292,
      "step": 195
    },
    {
      "epoch": 0.18630647414997673,
      "grad_norm": 0.5299149914911258,
      "learning_rate": 1.5527950310559007e-05,
      "loss": 0.5958,
      "step": 200
    },
    {
      "epoch": 0.19096413600372614,
      "grad_norm": 0.5338483365491724,
      "learning_rate": 1.5916149068322984e-05,
      "loss": 0.62,
      "step": 205
    },
    {
      "epoch": 0.19562179785747555,
      "grad_norm": 0.4731753222663508,
      "learning_rate": 1.630434782608696e-05,
      "loss": 0.6074,
      "step": 210
    },
    {
      "epoch": 0.20027945971122496,
      "grad_norm": 0.4357226972893284,
      "learning_rate": 1.6692546583850933e-05,
      "loss": 0.6195,
      "step": 215
    },
    {
      "epoch": 0.20493712156497437,
      "grad_norm": 0.46560395525394177,
      "learning_rate": 1.7080745341614907e-05,
      "loss": 0.6121,
      "step": 220
    },
    {
      "epoch": 0.2095947834187238,
      "grad_norm": 0.4645425643230276,
      "learning_rate": 1.7468944099378884e-05,
      "loss": 0.6153,
      "step": 225
    },
    {
      "epoch": 0.21425244527247322,
      "grad_norm": 0.4972545278987897,
      "learning_rate": 1.785714285714286e-05,
      "loss": 0.6121,
      "step": 230
    },
    {
      "epoch": 0.21891010712622264,
      "grad_norm": 0.4727989423912621,
      "learning_rate": 1.8245341614906833e-05,
      "loss": 0.6057,
      "step": 235
    },
    {
      "epoch": 0.22356776897997205,
      "grad_norm": 0.542703617419881,
      "learning_rate": 1.8633540372670807e-05,
      "loss": 0.6041,
      "step": 240
    },
    {
      "epoch": 0.22822543083372146,
      "grad_norm": 0.5364494253910931,
      "learning_rate": 1.9021739130434784e-05,
      "loss": 0.5911,
      "step": 245
    },
    {
      "epoch": 0.2328830926874709,
      "grad_norm": 0.449761619186817,
      "learning_rate": 1.940993788819876e-05,
      "loss": 0.5876,
      "step": 250
    },
    {
      "epoch": 0.2375407545412203,
      "grad_norm": 0.490407296357397,
      "learning_rate": 1.9798136645962733e-05,
      "loss": 0.5994,
      "step": 255
    },
    {
      "epoch": 0.24219841639496972,
      "grad_norm": 0.5109623564885958,
      "learning_rate": 2.0186335403726707e-05,
      "loss": 0.6046,
      "step": 260
    },
    {
      "epoch": 0.24685607824871914,
      "grad_norm": 0.5200882381667634,
      "learning_rate": 2.0574534161490684e-05,
      "loss": 0.5956,
      "step": 265
    },
    {
      "epoch": 0.2515137401024686,
      "grad_norm": 0.5211850145580095,
      "learning_rate": 2.096273291925466e-05,
      "loss": 0.5821,
      "step": 270
    },
    {
      "epoch": 0.25617140195621796,
      "grad_norm": 0.5058744951138296,
      "learning_rate": 2.1350931677018636e-05,
      "loss": 0.6012,
      "step": 275
    },
    {
      "epoch": 0.2608290638099674,
      "grad_norm": 0.49640380606884477,
      "learning_rate": 2.173913043478261e-05,
      "loss": 0.588,
      "step": 280
    },
    {
      "epoch": 0.26548672566371684,
      "grad_norm": 0.5127274867579876,
      "learning_rate": 2.2127329192546584e-05,
      "loss": 0.5839,
      "step": 285
    },
    {
      "epoch": 0.2701443875174662,
      "grad_norm": 0.4813945241254604,
      "learning_rate": 2.2515527950310562e-05,
      "loss": 0.5824,
      "step": 290
    },
    {
      "epoch": 0.27480204937121566,
      "grad_norm": 0.49133782459852365,
      "learning_rate": 2.2903726708074536e-05,
      "loss": 0.5741,
      "step": 295
    },
    {
      "epoch": 0.27945971122496505,
      "grad_norm": 0.6009816382381177,
      "learning_rate": 2.329192546583851e-05,
      "loss": 0.5863,
      "step": 300
    },
    {
      "epoch": 0.2841173730787145,
      "grad_norm": 0.5421108268784873,
      "learning_rate": 2.3680124223602484e-05,
      "loss": 0.5943,
      "step": 305
    },
    {
      "epoch": 0.2887750349324639,
      "grad_norm": 0.5529878819232127,
      "learning_rate": 2.4068322981366462e-05,
      "loss": 0.5883,
      "step": 310
    },
    {
      "epoch": 0.2934326967862133,
      "grad_norm": 0.5628633182049463,
      "learning_rate": 2.4456521739130436e-05,
      "loss": 0.575,
      "step": 315
    },
    {
      "epoch": 0.29809035863996275,
      "grad_norm": 0.5689989808425255,
      "learning_rate": 2.484472049689441e-05,
      "loss": 0.5845,
      "step": 320
    },
    {
      "epoch": 0.30274802049371213,
      "grad_norm": 0.47838702912042674,
      "learning_rate": 2.5232919254658388e-05,
      "loss": 0.5865,
      "step": 325
    },
    {
      "epoch": 0.3074056823474616,
      "grad_norm": 0.512356213276674,
      "learning_rate": 2.5621118012422362e-05,
      "loss": 0.5847,
      "step": 330
    },
    {
      "epoch": 0.312063344201211,
      "grad_norm": 0.5081234053651373,
      "learning_rate": 2.6009316770186336e-05,
      "loss": 0.564,
      "step": 335
    },
    {
      "epoch": 0.3167210060549604,
      "grad_norm": 0.5210036317950753,
      "learning_rate": 2.639751552795031e-05,
      "loss": 0.5793,
      "step": 340
    },
    {
      "epoch": 0.32137866790870984,
      "grad_norm": 0.46087825267279176,
      "learning_rate": 2.6785714285714288e-05,
      "loss": 0.5672,
      "step": 345
    },
    {
      "epoch": 0.3260363297624592,
      "grad_norm": 0.5424549525795376,
      "learning_rate": 2.7173913043478262e-05,
      "loss": 0.57,
      "step": 350
    },
    {
      "epoch": 0.33069399161620866,
      "grad_norm": 0.5639014679882649,
      "learning_rate": 2.7562111801242236e-05,
      "loss": 0.5658,
      "step": 355
    },
    {
      "epoch": 0.3353516534699581,
      "grad_norm": 0.5168865188998769,
      "learning_rate": 2.795031055900621e-05,
      "loss": 0.5771,
      "step": 360
    },
    {
      "epoch": 0.3400093153237075,
      "grad_norm": 0.5298542806804847,
      "learning_rate": 2.833850931677019e-05,
      "loss": 0.5766,
      "step": 365
    },
    {
      "epoch": 0.3446669771774569,
      "grad_norm": 0.5902954997946329,
      "learning_rate": 2.8726708074534165e-05,
      "loss": 0.5874,
      "step": 370
    },
    {
      "epoch": 0.3493246390312063,
      "grad_norm": 0.5058778120818872,
      "learning_rate": 2.911490683229814e-05,
      "loss": 0.58,
      "step": 375
    },
    {
      "epoch": 0.35398230088495575,
      "grad_norm": 0.5739079801875265,
      "learning_rate": 2.9503105590062114e-05,
      "loss": 0.5652,
      "step": 380
    },
    {
      "epoch": 0.3586399627387052,
      "grad_norm": 0.49817862139478225,
      "learning_rate": 2.9891304347826088e-05,
      "loss": 0.5578,
      "step": 385
    },
    {
      "epoch": 0.36329762459245457,
      "grad_norm": 0.5617597273635523,
      "learning_rate": 3.0279503105590062e-05,
      "loss": 0.5645,
      "step": 390
    },
    {
      "epoch": 0.367955286446204,
      "grad_norm": 0.5620538139815059,
      "learning_rate": 3.066770186335404e-05,
      "loss": 0.557,
      "step": 395
    },
    {
      "epoch": 0.37261294829995345,
      "grad_norm": 0.5824549154650988,
      "learning_rate": 3.1055900621118014e-05,
      "loss": 0.5544,
      "step": 400
    },
    {
      "epoch": 0.37727061015370283,
      "grad_norm": 0.526045149297124,
      "learning_rate": 3.1444099378881995e-05,
      "loss": 0.5579,
      "step": 405
    },
    {
      "epoch": 0.3819282720074523,
      "grad_norm": 0.47899250943430915,
      "learning_rate": 3.183229813664597e-05,
      "loss": 0.5644,
      "step": 410
    },
    {
      "epoch": 0.38658593386120166,
      "grad_norm": 0.5257043673829901,
      "learning_rate": 3.222049689440994e-05,
      "loss": 0.5643,
      "step": 415
    },
    {
      "epoch": 0.3912435957149511,
      "grad_norm": 0.4396015024205851,
      "learning_rate": 3.260869565217392e-05,
      "loss": 0.5517,
      "step": 420
    },
    {
      "epoch": 0.39590125756870054,
      "grad_norm": 0.48217446312509704,
      "learning_rate": 3.299689440993789e-05,
      "loss": 0.5704,
      "step": 425
    },
    {
      "epoch": 0.4005589194224499,
      "grad_norm": 0.5360547479149677,
      "learning_rate": 3.3385093167701865e-05,
      "loss": 0.5636,
      "step": 430
    },
    {
      "epoch": 0.40521658127619936,
      "grad_norm": 0.547348763718714,
      "learning_rate": 3.377329192546584e-05,
      "loss": 0.5656,
      "step": 435
    },
    {
      "epoch": 0.40987424312994875,
      "grad_norm": 0.49113542592359566,
      "learning_rate": 3.4161490683229814e-05,
      "loss": 0.5767,
      "step": 440
    },
    {
      "epoch": 0.4145319049836982,
      "grad_norm": 0.44506021490675857,
      "learning_rate": 3.4549689440993795e-05,
      "loss": 0.5494,
      "step": 445
    },
    {
      "epoch": 0.4191895668374476,
      "grad_norm": 0.5766640009936426,
      "learning_rate": 3.493788819875777e-05,
      "loss": 0.5545,
      "step": 450
    },
    {
      "epoch": 0.423847228691197,
      "grad_norm": 0.5932763378135886,
      "learning_rate": 3.532608695652174e-05,
      "loss": 0.5754,
      "step": 455
    },
    {
      "epoch": 0.42850489054494645,
      "grad_norm": 0.5456602245870262,
      "learning_rate": 3.571428571428572e-05,
      "loss": 0.554,
      "step": 460
    },
    {
      "epoch": 0.43316255239869583,
      "grad_norm": 0.4855348959581594,
      "learning_rate": 3.610248447204969e-05,
      "loss": 0.5632,
      "step": 465
    },
    {
      "epoch": 0.43782021425244527,
      "grad_norm": 0.4564403103865344,
      "learning_rate": 3.6490683229813665e-05,
      "loss": 0.5541,
      "step": 470
    },
    {
      "epoch": 0.4424778761061947,
      "grad_norm": 0.5051373402673379,
      "learning_rate": 3.687888198757764e-05,
      "loss": 0.5646,
      "step": 475
    },
    {
      "epoch": 0.4471355379599441,
      "grad_norm": 0.4828732044548432,
      "learning_rate": 3.7267080745341614e-05,
      "loss": 0.5488,
      "step": 480
    },
    {
      "epoch": 0.45179319981369354,
      "grad_norm": 0.49057254898365754,
      "learning_rate": 3.765527950310559e-05,
      "loss": 0.5518,
      "step": 485
    },
    {
      "epoch": 0.4564508616674429,
      "grad_norm": 0.5110566501516359,
      "learning_rate": 3.804347826086957e-05,
      "loss": 0.5556,
      "step": 490
    },
    {
      "epoch": 0.46110852352119236,
      "grad_norm": 0.5295464007987085,
      "learning_rate": 3.843167701863354e-05,
      "loss": 0.546,
      "step": 495
    },
    {
      "epoch": 0.4657661853749418,
      "grad_norm": 0.4361855393946779,
      "learning_rate": 3.881987577639752e-05,
      "loss": 0.5546,
      "step": 500
    },
    {
      "epoch": 0.4704238472286912,
      "grad_norm": 0.47154619647524604,
      "learning_rate": 3.920807453416149e-05,
      "loss": 0.5469,
      "step": 505
    },
    {
      "epoch": 0.4750815090824406,
      "grad_norm": 0.5278649356127125,
      "learning_rate": 3.9596273291925465e-05,
      "loss": 0.5429,
      "step": 510
    },
    {
      "epoch": 0.47973917093619,
      "grad_norm": 0.5030318419723568,
      "learning_rate": 3.998447204968944e-05,
      "loss": 0.5568,
      "step": 515
    },
    {
      "epoch": 0.48439683278993945,
      "grad_norm": 0.5960755161869581,
      "learning_rate": 4.0372670807453414e-05,
      "loss": 0.552,
      "step": 520
    },
    {
      "epoch": 0.4890544946436889,
      "grad_norm": 0.6320982107333845,
      "learning_rate": 4.076086956521739e-05,
      "loss": 0.5631,
      "step": 525
    },
    {
      "epoch": 0.49371215649743827,
      "grad_norm": 0.5861268615367173,
      "learning_rate": 4.114906832298137e-05,
      "loss": 0.5459,
      "step": 530
    },
    {
      "epoch": 0.4983698183511877,
      "grad_norm": 0.5609433316676956,
      "learning_rate": 4.153726708074534e-05,
      "loss": 0.5533,
      "step": 535
    },
    {
      "epoch": 0.5030274802049371,
      "grad_norm": 0.5504851197985072,
      "learning_rate": 4.192546583850932e-05,
      "loss": 0.5574,
      "step": 540
    },
    {
      "epoch": 0.5076851420586865,
      "grad_norm": 0.523186312068571,
      "learning_rate": 4.23136645962733e-05,
      "loss": 0.5446,
      "step": 545
    },
    {
      "epoch": 0.5123428039124359,
      "grad_norm": 0.5324068250744057,
      "learning_rate": 4.270186335403727e-05,
      "loss": 0.5611,
      "step": 550
    },
    {
      "epoch": 0.5170004657661854,
      "grad_norm": 0.47483528240802575,
      "learning_rate": 4.3090062111801246e-05,
      "loss": 0.5477,
      "step": 555
    },
    {
      "epoch": 0.5216581276199348,
      "grad_norm": 0.5782353748860246,
      "learning_rate": 4.347826086956522e-05,
      "loss": 0.5531,
      "step": 560
    },
    {
      "epoch": 0.5263157894736842,
      "grad_norm": 0.515431367086659,
      "learning_rate": 4.3866459627329195e-05,
      "loss": 0.5503,
      "step": 565
    },
    {
      "epoch": 0.5309734513274337,
      "grad_norm": 0.46435290662891776,
      "learning_rate": 4.425465838509317e-05,
      "loss": 0.5495,
      "step": 570
    },
    {
      "epoch": 0.5356311131811831,
      "grad_norm": 0.45648027202702574,
      "learning_rate": 4.464285714285715e-05,
      "loss": 0.5458,
      "step": 575
    },
    {
      "epoch": 0.5402887750349324,
      "grad_norm": 0.4611578014783051,
      "learning_rate": 4.5031055900621124e-05,
      "loss": 0.5464,
      "step": 580
    },
    {
      "epoch": 0.5449464368886818,
      "grad_norm": 0.4949473797007459,
      "learning_rate": 4.54192546583851e-05,
      "loss": 0.5413,
      "step": 585
    },
    {
      "epoch": 0.5496040987424313,
      "grad_norm": 0.48619314011665365,
      "learning_rate": 4.580745341614907e-05,
      "loss": 0.56,
      "step": 590
    },
    {
      "epoch": 0.5542617605961807,
      "grad_norm": 0.47060805503246933,
      "learning_rate": 4.6195652173913046e-05,
      "loss": 0.5493,
      "step": 595
    },
    {
      "epoch": 0.5589194224499301,
      "grad_norm": 0.5033620925837115,
      "learning_rate": 4.658385093167702e-05,
      "loss": 0.5554,
      "step": 600
    },
    {
      "epoch": 0.5635770843036796,
      "grad_norm": 0.4727436353619463,
      "learning_rate": 4.6972049689440995e-05,
      "loss": 0.5253,
      "step": 605
    },
    {
      "epoch": 0.568234746157429,
      "grad_norm": 0.5197889877850108,
      "learning_rate": 4.736024844720497e-05,
      "loss": 0.5481,
      "step": 610
    },
    {
      "epoch": 0.5728924080111784,
      "grad_norm": 0.5029913741481368,
      "learning_rate": 4.774844720496895e-05,
      "loss": 0.5392,
      "step": 615
    },
    {
      "epoch": 0.5775500698649279,
      "grad_norm": 0.5009388967364741,
      "learning_rate": 4.8136645962732924e-05,
      "loss": 0.5426,
      "step": 620
    },
    {
      "epoch": 0.5822077317186772,
      "grad_norm": 0.45158400666590753,
      "learning_rate": 4.85248447204969e-05,
      "loss": 0.5546,
      "step": 625
    },
    {
      "epoch": 0.5868653935724266,
      "grad_norm": 0.5080779925861048,
      "learning_rate": 4.891304347826087e-05,
      "loss": 0.5394,
      "step": 630
    },
    {
      "epoch": 0.5915230554261761,
      "grad_norm": 0.5019074093792454,
      "learning_rate": 4.9301242236024846e-05,
      "loss": 0.539,
      "step": 635
    },
    {
      "epoch": 0.5961807172799255,
      "grad_norm": 0.4881685145821943,
      "learning_rate": 4.968944099378882e-05,
      "loss": 0.5447,
      "step": 640
    },
    {
      "epoch": 0.6008383791336749,
      "grad_norm": 0.4729315267644058,
      "learning_rate": 4.999137038315499e-05,
      "loss": 0.5445,
      "step": 645
    },
    {
      "epoch": 0.6054960409874243,
      "grad_norm": 0.5097738569010467,
      "learning_rate": 4.994822229892993e-05,
      "loss": 0.5403,
      "step": 650
    },
    {
      "epoch": 0.6101537028411738,
      "grad_norm": 0.49025735715149743,
      "learning_rate": 4.9905074214704875e-05,
      "loss": 0.5463,
      "step": 655
    },
    {
      "epoch": 0.6148113646949231,
      "grad_norm": 0.4358431190457931,
      "learning_rate": 4.986192613047981e-05,
      "loss": 0.5498,
      "step": 660
    },
    {
      "epoch": 0.6194690265486725,
      "grad_norm": 0.5027322377793131,
      "learning_rate": 4.9818778046254746e-05,
      "loss": 0.5431,
      "step": 665
    },
    {
      "epoch": 0.624126688402422,
      "grad_norm": 0.47706653710036584,
      "learning_rate": 4.977562996202969e-05,
      "loss": 0.5459,
      "step": 670
    },
    {
      "epoch": 0.6287843502561714,
      "grad_norm": 0.4614645736551657,
      "learning_rate": 4.9732481877804624e-05,
      "loss": 0.5353,
      "step": 675
    },
    {
      "epoch": 0.6334420121099208,
      "grad_norm": 0.4127504433580963,
      "learning_rate": 4.968933379357957e-05,
      "loss": 0.5398,
      "step": 680
    },
    {
      "epoch": 0.6380996739636703,
      "grad_norm": 0.48562111374906797,
      "learning_rate": 4.964618570935451e-05,
      "loss": 0.5536,
      "step": 685
    },
    {
      "epoch": 0.6427573358174197,
      "grad_norm": 0.4565460747029453,
      "learning_rate": 4.9603037625129445e-05,
      "loss": 0.5404,
      "step": 690
    },
    {
      "epoch": 0.6474149976711691,
      "grad_norm": 0.4261245682130229,
      "learning_rate": 4.955988954090439e-05,
      "loss": 0.5399,
      "step": 695
    },
    {
      "epoch": 0.6520726595249184,
      "grad_norm": 0.47406125045773095,
      "learning_rate": 4.951674145667933e-05,
      "loss": 0.5366,
      "step": 700
    },
    {
      "epoch": 0.6567303213786679,
      "grad_norm": 0.4686101193047825,
      "learning_rate": 4.9473593372454266e-05,
      "loss": 0.5326,
      "step": 705
    },
    {
      "epoch": 0.6613879832324173,
      "grad_norm": 0.4405250233709894,
      "learning_rate": 4.94304452882292e-05,
      "loss": 0.5457,
      "step": 710
    },
    {
      "epoch": 0.6660456450861667,
      "grad_norm": 0.4678036683381308,
      "learning_rate": 4.9387297204004144e-05,
      "loss": 0.5401,
      "step": 715
    },
    {
      "epoch": 0.6707033069399162,
      "grad_norm": 0.4583560142839113,
      "learning_rate": 4.934414911977908e-05,
      "loss": 0.5286,
      "step": 720
    },
    {
      "epoch": 0.6753609687936656,
      "grad_norm": 0.4278511454276169,
      "learning_rate": 4.930100103555402e-05,
      "loss": 0.5336,
      "step": 725
    },
    {
      "epoch": 0.680018630647415,
      "grad_norm": 0.47457517613794836,
      "learning_rate": 4.9257852951328965e-05,
      "loss": 0.5416,
      "step": 730
    },
    {
      "epoch": 0.6846762925011645,
      "grad_norm": 0.4961544514872818,
      "learning_rate": 4.921470486710391e-05,
      "loss": 0.529,
      "step": 735
    },
    {
      "epoch": 0.6893339543549138,
      "grad_norm": 0.40191770399507454,
      "learning_rate": 4.917155678287884e-05,
      "loss": 0.5357,
      "step": 740
    },
    {
      "epoch": 0.6939916162086632,
      "grad_norm": 0.4779072114072958,
      "learning_rate": 4.9128408698653785e-05,
      "loss": 0.5313,
      "step": 745
    },
    {
      "epoch": 0.6986492780624126,
      "grad_norm": 0.5562861494291095,
      "learning_rate": 4.908526061442872e-05,
      "loss": 0.5356,
      "step": 750
    },
    {
      "epoch": 0.7033069399161621,
      "grad_norm": 0.45220052232693286,
      "learning_rate": 4.904211253020366e-05,
      "loss": 0.5249,
      "step": 755
    },
    {
      "epoch": 0.7079646017699115,
      "grad_norm": 0.4590238020564062,
      "learning_rate": 4.89989644459786e-05,
      "loss": 0.5275,
      "step": 760
    },
    {
      "epoch": 0.7126222636236609,
      "grad_norm": 0.38800064117015004,
      "learning_rate": 4.895581636175354e-05,
      "loss": 0.5385,
      "step": 765
    },
    {
      "epoch": 0.7172799254774104,
      "grad_norm": 0.4721638285008215,
      "learning_rate": 4.891266827752848e-05,
      "loss": 0.5371,
      "step": 770
    },
    {
      "epoch": 0.7219375873311598,
      "grad_norm": 0.451000579844508,
      "learning_rate": 4.886952019330342e-05,
      "loss": 0.5234,
      "step": 775
    },
    {
      "epoch": 0.7265952491849091,
      "grad_norm": 0.4335286688393536,
      "learning_rate": 4.882637210907836e-05,
      "loss": 0.5421,
      "step": 780
    },
    {
      "epoch": 0.7312529110386586,
      "grad_norm": 0.3766770026295968,
      "learning_rate": 4.87832240248533e-05,
      "loss": 0.5348,
      "step": 785
    },
    {
      "epoch": 0.735910572892408,
      "grad_norm": 0.43733454307114944,
      "learning_rate": 4.874007594062824e-05,
      "loss": 0.5258,
      "step": 790
    },
    {
      "epoch": 0.7405682347461574,
      "grad_norm": 0.38251559841437066,
      "learning_rate": 4.8696927856403176e-05,
      "loss": 0.5327,
      "step": 795
    },
    {
      "epoch": 0.7452258965999069,
      "grad_norm": 0.36006570689046885,
      "learning_rate": 4.865377977217811e-05,
      "loss": 0.5286,
      "step": 800
    },
    {
      "epoch": 0.7498835584536563,
      "grad_norm": 0.38934384136785405,
      "learning_rate": 4.8610631687953055e-05,
      "loss": 0.5446,
      "step": 805
    },
    {
      "epoch": 0.7545412203074057,
      "grad_norm": 0.39990179552304195,
      "learning_rate": 4.8567483603728e-05,
      "loss": 0.5281,
      "step": 810
    },
    {
      "epoch": 0.759198882161155,
      "grad_norm": 0.4079109596632673,
      "learning_rate": 4.852433551950293e-05,
      "loss": 0.5323,
      "step": 815
    },
    {
      "epoch": 0.7638565440149045,
      "grad_norm": 0.355926856787488,
      "learning_rate": 4.8481187435277875e-05,
      "loss": 0.5318,
      "step": 820
    },
    {
      "epoch": 0.7685142058686539,
      "grad_norm": 0.4085427915458698,
      "learning_rate": 4.843803935105282e-05,
      "loss": 0.5173,
      "step": 825
    },
    {
      "epoch": 0.7731718677224033,
      "grad_norm": 0.3801129092095776,
      "learning_rate": 4.839489126682776e-05,
      "loss": 0.5266,
      "step": 830
    },
    {
      "epoch": 0.7778295295761528,
      "grad_norm": 0.380365264030816,
      "learning_rate": 4.8351743182602696e-05,
      "loss": 0.5339,
      "step": 835
    },
    {
      "epoch": 0.7824871914299022,
      "grad_norm": 0.40184569313393786,
      "learning_rate": 4.830859509837763e-05,
      "loss": 0.5177,
      "step": 840
    },
    {
      "epoch": 0.7871448532836516,
      "grad_norm": 0.4185794054830395,
      "learning_rate": 4.8265447014152574e-05,
      "loss": 0.5247,
      "step": 845
    },
    {
      "epoch": 0.7918025151374011,
      "grad_norm": 0.429974623729282,
      "learning_rate": 4.822229892992751e-05,
      "loss": 0.5268,
      "step": 850
    },
    {
      "epoch": 0.7964601769911505,
      "grad_norm": 0.380106867440683,
      "learning_rate": 4.817915084570245e-05,
      "loss": 0.5163,
      "step": 855
    },
    {
      "epoch": 0.8011178388448998,
      "grad_norm": 0.3752988924826398,
      "learning_rate": 4.8136002761477395e-05,
      "loss": 0.5211,
      "step": 860
    },
    {
      "epoch": 0.8057755006986492,
      "grad_norm": 0.4193829736647745,
      "learning_rate": 4.809285467725233e-05,
      "loss": 0.5234,
      "step": 865
    },
    {
      "epoch": 0.8104331625523987,
      "grad_norm": 0.3994087746114865,
      "learning_rate": 4.804970659302727e-05,
      "loss": 0.5161,
      "step": 870
    },
    {
      "epoch": 0.8150908244061481,
      "grad_norm": 0.4583594691171294,
      "learning_rate": 4.8006558508802216e-05,
      "loss": 0.5294,
      "step": 875
    },
    {
      "epoch": 0.8197484862598975,
      "grad_norm": 0.3722654245488076,
      "learning_rate": 4.796341042457715e-05,
      "loss": 0.5087,
      "step": 880
    },
    {
      "epoch": 0.824406148113647,
      "grad_norm": 0.4029130142450783,
      "learning_rate": 4.792026234035209e-05,
      "loss": 0.529,
      "step": 885
    },
    {
      "epoch": 0.8290638099673964,
      "grad_norm": 0.38424752610359497,
      "learning_rate": 4.787711425612703e-05,
      "loss": 0.5267,
      "step": 890
    },
    {
      "epoch": 0.8337214718211458,
      "grad_norm": 0.34261057420486174,
      "learning_rate": 4.7833966171901966e-05,
      "loss": 0.5267,
      "step": 895
    },
    {
      "epoch": 0.8383791336748952,
      "grad_norm": 0.3743153105453807,
      "learning_rate": 4.779081808767691e-05,
      "loss": 0.5203,
      "step": 900
    },
    {
      "epoch": 0.8430367955286446,
      "grad_norm": 0.3805691691922901,
      "learning_rate": 4.774767000345185e-05,
      "loss": 0.524,
      "step": 905
    },
    {
      "epoch": 0.847694457382394,
      "grad_norm": 0.38950472030002864,
      "learning_rate": 4.770452191922679e-05,
      "loss": 0.5324,
      "step": 910
    },
    {
      "epoch": 0.8523521192361434,
      "grad_norm": 0.4156968662878521,
      "learning_rate": 4.766137383500173e-05,
      "loss": 0.5205,
      "step": 915
    },
    {
      "epoch": 0.8570097810898929,
      "grad_norm": 0.40559914730591284,
      "learning_rate": 4.761822575077667e-05,
      "loss": 0.5309,
      "step": 920
    },
    {
      "epoch": 0.8616674429436423,
      "grad_norm": 0.40380600942552064,
      "learning_rate": 4.757507766655161e-05,
      "loss": 0.5234,
      "step": 925
    },
    {
      "epoch": 0.8663251047973917,
      "grad_norm": 0.3853955104044911,
      "learning_rate": 4.753192958232654e-05,
      "loss": 0.5255,
      "step": 930
    },
    {
      "epoch": 0.8709827666511412,
      "grad_norm": 0.37260776656945166,
      "learning_rate": 4.7488781498101485e-05,
      "loss": 0.5109,
      "step": 935
    },
    {
      "epoch": 0.8756404285048905,
      "grad_norm": 0.4028527034528855,
      "learning_rate": 4.744563341387643e-05,
      "loss": 0.5095,
      "step": 940
    },
    {
      "epoch": 0.8802980903586399,
      "grad_norm": 0.3785896877796604,
      "learning_rate": 4.7402485329651363e-05,
      "loss": 0.516,
      "step": 945
    },
    {
      "epoch": 0.8849557522123894,
      "grad_norm": 0.345795711764979,
      "learning_rate": 4.7359337245426306e-05,
      "loss": 0.5029,
      "step": 950
    },
    {
      "epoch": 0.8896134140661388,
      "grad_norm": 0.3564957798741615,
      "learning_rate": 4.731618916120125e-05,
      "loss": 0.5194,
      "step": 955
    },
    {
      "epoch": 0.8942710759198882,
      "grad_norm": 0.3698585419884843,
      "learning_rate": 4.7273041076976184e-05,
      "loss": 0.5251,
      "step": 960
    },
    {
      "epoch": 0.8989287377736377,
      "grad_norm": 0.3791918650222602,
      "learning_rate": 4.722989299275113e-05,
      "loss": 0.519,
      "step": 965
    },
    {
      "epoch": 0.9035863996273871,
      "grad_norm": 0.38216126716499343,
      "learning_rate": 4.718674490852606e-05,
      "loss": 0.5223,
      "step": 970
    },
    {
      "epoch": 0.9082440614811365,
      "grad_norm": 0.39451833529829045,
      "learning_rate": 4.7143596824301e-05,
      "loss": 0.5161,
      "step": 975
    },
    {
      "epoch": 0.9129017233348858,
      "grad_norm": 0.3783926625898741,
      "learning_rate": 4.710044874007594e-05,
      "loss": 0.5194,
      "step": 980
    },
    {
      "epoch": 0.9175593851886353,
      "grad_norm": 0.3556316671837827,
      "learning_rate": 4.705730065585088e-05,
      "loss": 0.525,
      "step": 985
    },
    {
      "epoch": 0.9222170470423847,
      "grad_norm": 0.4004133612924368,
      "learning_rate": 4.7014152571625826e-05,
      "loss": 0.4931,
      "step": 990
    },
    {
      "epoch": 0.9268747088961341,
      "grad_norm": 0.361122186880852,
      "learning_rate": 4.697100448740076e-05,
      "loss": 0.5112,
      "step": 995
    },
    {
      "epoch": 0.9315323707498836,
      "grad_norm": 0.4088139036868105,
      "learning_rate": 4.6927856403175704e-05,
      "loss": 0.5147,
      "step": 1000
    },
    {
      "epoch": 0.936190032603633,
      "grad_norm": 0.3658393846950928,
      "learning_rate": 4.6884708318950646e-05,
      "loss": 0.5258,
      "step": 1005
    },
    {
      "epoch": 0.9408476944573824,
      "grad_norm": 0.37929557356125604,
      "learning_rate": 4.684156023472558e-05,
      "loss": 0.5121,
      "step": 1010
    },
    {
      "epoch": 0.9455053563111319,
      "grad_norm": 0.38439507972608544,
      "learning_rate": 4.679841215050052e-05,
      "loss": 0.5151,
      "step": 1015
    },
    {
      "epoch": 0.9501630181648812,
      "grad_norm": 0.3522414026232054,
      "learning_rate": 4.675526406627546e-05,
      "loss": 0.525,
      "step": 1020
    },
    {
      "epoch": 0.9548206800186306,
      "grad_norm": 0.46755450703731466,
      "learning_rate": 4.6712115982050396e-05,
      "loss": 0.5157,
      "step": 1025
    },
    {
      "epoch": 0.95947834187238,
      "grad_norm": 0.377169912947567,
      "learning_rate": 4.666896789782534e-05,
      "loss": 0.5062,
      "step": 1030
    },
    {
      "epoch": 0.9641360037261295,
      "grad_norm": 0.3652807029310775,
      "learning_rate": 4.662581981360028e-05,
      "loss": 0.5107,
      "step": 1035
    },
    {
      "epoch": 0.9687936655798789,
      "grad_norm": 0.4221038522703394,
      "learning_rate": 4.658267172937522e-05,
      "loss": 0.5078,
      "step": 1040
    },
    {
      "epoch": 0.9734513274336283,
      "grad_norm": 0.35700579090080875,
      "learning_rate": 4.653952364515016e-05,
      "loss": 0.5099,
      "step": 1045
    },
    {
      "epoch": 0.9781089892873778,
      "grad_norm": 0.34354596053874803,
      "learning_rate": 4.64963755609251e-05,
      "loss": 0.509,
      "step": 1050
    },
    {
      "epoch": 0.9827666511411272,
      "grad_norm": 0.3779917422908632,
      "learning_rate": 4.645322747670004e-05,
      "loss": 0.5115,
      "step": 1055
    },
    {
      "epoch": 0.9874243129948765,
      "grad_norm": 0.38410134560623915,
      "learning_rate": 4.641007939247497e-05,
      "loss": 0.5153,
      "step": 1060
    },
    {
      "epoch": 0.992081974848626,
      "grad_norm": 0.3511495295941889,
      "learning_rate": 4.6366931308249916e-05,
      "loss": 0.5126,
      "step": 1065
    },
    {
      "epoch": 0.9967396367023754,
      "grad_norm": 0.3376607964011747,
      "learning_rate": 4.632378322402486e-05,
      "loss": 0.5022,
      "step": 1070
    },
    {
      "epoch": 1.00093153237075,
      "grad_norm": 0.5826337472778964,
      "learning_rate": 4.6280635139799794e-05,
      "loss": 0.5216,
      "step": 1075
    },
    {
      "epoch": 1.0055891942244992,
      "grad_norm": 0.40189887041509037,
      "learning_rate": 4.6237487055574736e-05,
      "loss": 0.4863,
      "step": 1080
    },
    {
      "epoch": 1.0102468560782487,
      "grad_norm": 0.3568971280329747,
      "learning_rate": 4.619433897134968e-05,
      "loss": 0.471,
      "step": 1085
    },
    {
      "epoch": 1.0149045179319982,
      "grad_norm": 0.39782765819393584,
      "learning_rate": 4.6151190887124615e-05,
      "loss": 0.491,
      "step": 1090
    },
    {
      "epoch": 1.0195621797857475,
      "grad_norm": 0.39251480400567884,
      "learning_rate": 4.610804280289955e-05,
      "loss": 0.4875,
      "step": 1095
    },
    {
      "epoch": 1.024219841639497,
      "grad_norm": 0.3957860380124049,
      "learning_rate": 4.606489471867449e-05,
      "loss": 0.4892,
      "step": 1100
    },
    {
      "epoch": 1.0288775034932465,
      "grad_norm": 0.3617773073107849,
      "learning_rate": 4.602174663444943e-05,
      "loss": 0.4866,
      "step": 1105
    },
    {
      "epoch": 1.0335351653469957,
      "grad_norm": 0.3774322768513962,
      "learning_rate": 4.597859855022437e-05,
      "loss": 0.4833,
      "step": 1110
    },
    {
      "epoch": 1.0381928272007452,
      "grad_norm": 0.3482318914799298,
      "learning_rate": 4.5935450465999314e-05,
      "loss": 0.4753,
      "step": 1115
    },
    {
      "epoch": 1.0428504890544947,
      "grad_norm": 0.3990402322688351,
      "learning_rate": 4.589230238177425e-05,
      "loss": 0.4856,
      "step": 1120
    },
    {
      "epoch": 1.047508150908244,
      "grad_norm": 0.32492143047085587,
      "learning_rate": 4.584915429754919e-05,
      "loss": 0.4915,
      "step": 1125
    },
    {
      "epoch": 1.0521658127619935,
      "grad_norm": 0.37066200896167156,
      "learning_rate": 4.5806006213324134e-05,
      "loss": 0.4787,
      "step": 1130
    },
    {
      "epoch": 1.056823474615743,
      "grad_norm": 0.3753960071489754,
      "learning_rate": 4.576285812909907e-05,
      "loss": 0.4856,
      "step": 1135
    },
    {
      "epoch": 1.0614811364694923,
      "grad_norm": 0.3889455837851045,
      "learning_rate": 4.5719710044874006e-05,
      "loss": 0.486,
      "step": 1140
    },
    {
      "epoch": 1.0661387983232418,
      "grad_norm": 0.40341207215908437,
      "learning_rate": 4.567656196064895e-05,
      "loss": 0.485,
      "step": 1145
    },
    {
      "epoch": 1.0707964601769913,
      "grad_norm": 0.38600084172973353,
      "learning_rate": 4.563341387642389e-05,
      "loss": 0.4841,
      "step": 1150
    },
    {
      "epoch": 1.0754541220307405,
      "grad_norm": 0.44581386446158155,
      "learning_rate": 4.5590265792198827e-05,
      "loss": 0.4869,
      "step": 1155
    },
    {
      "epoch": 1.08011178388449,
      "grad_norm": 0.40702018103004495,
      "learning_rate": 4.554711770797377e-05,
      "loss": 0.4941,
      "step": 1160
    },
    {
      "epoch": 1.0847694457382393,
      "grad_norm": 0.36806267352724337,
      "learning_rate": 4.550396962374871e-05,
      "loss": 0.489,
      "step": 1165
    },
    {
      "epoch": 1.0894271075919888,
      "grad_norm": 0.3480373642743722,
      "learning_rate": 4.546082153952365e-05,
      "loss": 0.4876,
      "step": 1170
    },
    {
      "epoch": 1.0940847694457383,
      "grad_norm": 0.3576667629074542,
      "learning_rate": 4.541767345529859e-05,
      "loss": 0.4892,
      "step": 1175
    },
    {
      "epoch": 1.0987424312994876,
      "grad_norm": 0.38344312925355145,
      "learning_rate": 4.5374525371073526e-05,
      "loss": 0.4808,
      "step": 1180
    },
    {
      "epoch": 1.103400093153237,
      "grad_norm": 0.39434389233406764,
      "learning_rate": 4.533137728684846e-05,
      "loss": 0.4784,
      "step": 1185
    },
    {
      "epoch": 1.1080577550069866,
      "grad_norm": 0.34843797547352795,
      "learning_rate": 4.5288229202623404e-05,
      "loss": 0.4612,
      "step": 1190
    },
    {
      "epoch": 1.1127154168607358,
      "grad_norm": 0.3664150315506526,
      "learning_rate": 4.5245081118398346e-05,
      "loss": 0.4728,
      "step": 1195
    },
    {
      "epoch": 1.1173730787144853,
      "grad_norm": 0.35614071325495805,
      "learning_rate": 4.520193303417328e-05,
      "loss": 0.4796,
      "step": 1200
    },
    {
      "epoch": 1.1220307405682348,
      "grad_norm": 0.38520194816763753,
      "learning_rate": 4.5158784949948224e-05,
      "loss": 0.4921,
      "step": 1205
    },
    {
      "epoch": 1.126688402421984,
      "grad_norm": 0.3839130909983971,
      "learning_rate": 4.511563686572317e-05,
      "loss": 0.487,
      "step": 1210
    },
    {
      "epoch": 1.1313460642757336,
      "grad_norm": 0.37963115301884154,
      "learning_rate": 4.50724887814981e-05,
      "loss": 0.4811,
      "step": 1215
    },
    {
      "epoch": 1.136003726129483,
      "grad_norm": 0.3493643901615747,
      "learning_rate": 4.5029340697273045e-05,
      "loss": 0.4884,
      "step": 1220
    },
    {
      "epoch": 1.1406613879832324,
      "grad_norm": 0.3595348614034336,
      "learning_rate": 4.498619261304798e-05,
      "loss": 0.4793,
      "step": 1225
    },
    {
      "epoch": 1.1453190498369819,
      "grad_norm": 0.363405444358194,
      "learning_rate": 4.4943044528822923e-05,
      "loss": 0.4789,
      "step": 1230
    },
    {
      "epoch": 1.1499767116907313,
      "grad_norm": 0.3584064837608838,
      "learning_rate": 4.489989644459786e-05,
      "loss": 0.4832,
      "step": 1235
    },
    {
      "epoch": 1.1546343735444806,
      "grad_norm": 0.31631345829294577,
      "learning_rate": 4.48567483603728e-05,
      "loss": 0.4922,
      "step": 1240
    },
    {
      "epoch": 1.1592920353982301,
      "grad_norm": 0.344377025957259,
      "learning_rate": 4.4813600276147744e-05,
      "loss": 0.4865,
      "step": 1245
    },
    {
      "epoch": 1.1639496972519794,
      "grad_norm": 0.37896151858346383,
      "learning_rate": 4.477045219192268e-05,
      "loss": 0.4973,
      "step": 1250
    },
    {
      "epoch": 1.1686073591057289,
      "grad_norm": 0.3811284732813557,
      "learning_rate": 4.472730410769762e-05,
      "loss": 0.484,
      "step": 1255
    },
    {
      "epoch": 1.1732650209594784,
      "grad_norm": 0.32182932963989064,
      "learning_rate": 4.4684156023472565e-05,
      "loss": 0.4922,
      "step": 1260
    },
    {
      "epoch": 1.1779226828132279,
      "grad_norm": 0.349311871750818,
      "learning_rate": 4.46410079392475e-05,
      "loss": 0.4784,
      "step": 1265
    },
    {
      "epoch": 1.1825803446669771,
      "grad_norm": 0.3495710230430097,
      "learning_rate": 4.4597859855022436e-05,
      "loss": 0.471,
      "step": 1270
    },
    {
      "epoch": 1.1872380065207266,
      "grad_norm": 0.36153328259694295,
      "learning_rate": 4.455471177079738e-05,
      "loss": 0.4777,
      "step": 1275
    },
    {
      "epoch": 1.191895668374476,
      "grad_norm": 0.4327287980793124,
      "learning_rate": 4.4511563686572315e-05,
      "loss": 0.4874,
      "step": 1280
    },
    {
      "epoch": 1.1965533302282254,
      "grad_norm": 0.369786728558311,
      "learning_rate": 4.446841560234726e-05,
      "loss": 0.4883,
      "step": 1285
    },
    {
      "epoch": 1.201210992081975,
      "grad_norm": 0.3740436451225295,
      "learning_rate": 4.44252675181222e-05,
      "loss": 0.4753,
      "step": 1290
    },
    {
      "epoch": 1.2058686539357242,
      "grad_norm": 0.42600752846505047,
      "learning_rate": 4.4382119433897135e-05,
      "loss": 0.4811,
      "step": 1295
    },
    {
      "epoch": 1.2105263157894737,
      "grad_norm": 0.4019267005106993,
      "learning_rate": 4.433897134967208e-05,
      "loss": 0.4765,
      "step": 1300
    },
    {
      "epoch": 1.2151839776432232,
      "grad_norm": 0.3918096155745144,
      "learning_rate": 4.429582326544702e-05,
      "loss": 0.4875,
      "step": 1305
    },
    {
      "epoch": 1.2198416394969724,
      "grad_norm": 0.36321333821851454,
      "learning_rate": 4.4252675181221956e-05,
      "loss": 0.4909,
      "step": 1310
    },
    {
      "epoch": 1.224499301350722,
      "grad_norm": 0.31678932093299844,
      "learning_rate": 4.420952709699689e-05,
      "loss": 0.4806,
      "step": 1315
    },
    {
      "epoch": 1.2291569632044714,
      "grad_norm": 0.349058772542194,
      "learning_rate": 4.4166379012771834e-05,
      "loss": 0.4912,
      "step": 1320
    },
    {
      "epoch": 1.2338146250582207,
      "grad_norm": 0.3249570833550026,
      "learning_rate": 4.412323092854678e-05,
      "loss": 0.4764,
      "step": 1325
    },
    {
      "epoch": 1.2384722869119702,
      "grad_norm": 0.370352128941829,
      "learning_rate": 4.408008284432171e-05,
      "loss": 0.4888,
      "step": 1330
    },
    {
      "epoch": 1.2431299487657197,
      "grad_norm": 0.3886105196811443,
      "learning_rate": 4.4036934760096655e-05,
      "loss": 0.4818,
      "step": 1335
    },
    {
      "epoch": 1.247787610619469,
      "grad_norm": 0.39933169849947475,
      "learning_rate": 4.39937866758716e-05,
      "loss": 0.4796,
      "step": 1340
    },
    {
      "epoch": 1.2524452724732185,
      "grad_norm": 0.32324855251514784,
      "learning_rate": 4.395063859164653e-05,
      "loss": 0.4695,
      "step": 1345
    },
    {
      "epoch": 1.257102934326968,
      "grad_norm": 0.3522414459969437,
      "learning_rate": 4.3907490507421476e-05,
      "loss": 0.4802,
      "step": 1350
    },
    {
      "epoch": 1.2617605961807172,
      "grad_norm": 0.37456981934915634,
      "learning_rate": 4.386434242319641e-05,
      "loss": 0.4728,
      "step": 1355
    },
    {
      "epoch": 1.2664182580344667,
      "grad_norm": 0.36258896140518604,
      "learning_rate": 4.382119433897135e-05,
      "loss": 0.4663,
      "step": 1360
    },
    {
      "epoch": 1.271075919888216,
      "grad_norm": 0.3305202934214233,
      "learning_rate": 4.377804625474629e-05,
      "loss": 0.4848,
      "step": 1365
    },
    {
      "epoch": 1.2757335817419655,
      "grad_norm": 0.3678123692750788,
      "learning_rate": 4.373489817052123e-05,
      "loss": 0.4809,
      "step": 1370
    },
    {
      "epoch": 1.280391243595715,
      "grad_norm": 0.3374306046011543,
      "learning_rate": 4.369175008629617e-05,
      "loss": 0.4801,
      "step": 1375
    },
    {
      "epoch": 1.2850489054494645,
      "grad_norm": 0.31751308567996994,
      "learning_rate": 4.364860200207111e-05,
      "loss": 0.4798,
      "step": 1380
    },
    {
      "epoch": 1.2897065673032138,
      "grad_norm": 0.3542199584247798,
      "learning_rate": 4.360545391784605e-05,
      "loss": 0.4804,
      "step": 1385
    },
    {
      "epoch": 1.2943642291569633,
      "grad_norm": 0.3750008093134569,
      "learning_rate": 4.356230583362099e-05,
      "loss": 0.4886,
      "step": 1390
    },
    {
      "epoch": 1.2990218910107125,
      "grad_norm": 0.35983295778029717,
      "learning_rate": 4.351915774939593e-05,
      "loss": 0.4728,
      "step": 1395
    },
    {
      "epoch": 1.303679552864462,
      "grad_norm": 0.37703340915429806,
      "learning_rate": 4.347600966517087e-05,
      "loss": 0.488,
      "step": 1400
    },
    {
      "epoch": 1.3083372147182115,
      "grad_norm": 0.33100732980317954,
      "learning_rate": 4.343286158094581e-05,
      "loss": 0.4763,
      "step": 1405
    },
    {
      "epoch": 1.312994876571961,
      "grad_norm": 0.3484010027873586,
      "learning_rate": 4.3389713496720745e-05,
      "loss": 0.4726,
      "step": 1410
    },
    {
      "epoch": 1.3176525384257103,
      "grad_norm": 0.3527394392337278,
      "learning_rate": 4.334656541249569e-05,
      "loss": 0.4938,
      "step": 1415
    },
    {
      "epoch": 1.3223102002794598,
      "grad_norm": 0.3628371235800552,
      "learning_rate": 4.330341732827063e-05,
      "loss": 0.4818,
      "step": 1420
    },
    {
      "epoch": 1.326967862133209,
      "grad_norm": 0.33213560967026967,
      "learning_rate": 4.3260269244045566e-05,
      "loss": 0.4756,
      "step": 1425
    },
    {
      "epoch": 1.3316255239869585,
      "grad_norm": 0.3735703743172359,
      "learning_rate": 4.321712115982051e-05,
      "loss": 0.4742,
      "step": 1430
    },
    {
      "epoch": 1.336283185840708,
      "grad_norm": 0.360058769782219,
      "learning_rate": 4.317397307559545e-05,
      "loss": 0.4793,
      "step": 1435
    },
    {
      "epoch": 1.3409408476944573,
      "grad_norm": 0.33726550763915253,
      "learning_rate": 4.3130824991370387e-05,
      "loss": 0.4817,
      "step": 1440
    },
    {
      "epoch": 1.3455985095482068,
      "grad_norm": 0.3619091797549457,
      "learning_rate": 4.308767690714532e-05,
      "loss": 0.4906,
      "step": 1445
    },
    {
      "epoch": 1.350256171401956,
      "grad_norm": 0.37818275927830114,
      "learning_rate": 4.3044528822920265e-05,
      "loss": 0.4776,
      "step": 1450
    },
    {
      "epoch": 1.3549138332557056,
      "grad_norm": 0.37324457523490834,
      "learning_rate": 4.30013807386952e-05,
      "loss": 0.4881,
      "step": 1455
    },
    {
      "epoch": 1.359571495109455,
      "grad_norm": 0.3510889164222137,
      "learning_rate": 4.295823265447014e-05,
      "loss": 0.4684,
      "step": 1460
    },
    {
      "epoch": 1.3642291569632046,
      "grad_norm": 0.34059012808163075,
      "learning_rate": 4.2915084570245085e-05,
      "loss": 0.4764,
      "step": 1465
    },
    {
      "epoch": 1.3688868188169538,
      "grad_norm": 0.36370936555575334,
      "learning_rate": 4.287193648602002e-05,
      "loss": 0.4799,
      "step": 1470
    },
    {
      "epoch": 1.3735444806707033,
      "grad_norm": 0.38687362284561483,
      "learning_rate": 4.2828788401794964e-05,
      "loss": 0.472,
      "step": 1475
    },
    {
      "epoch": 1.3782021425244526,
      "grad_norm": 0.3453458464242573,
      "learning_rate": 4.27856403175699e-05,
      "loss": 0.4721,
      "step": 1480
    },
    {
      "epoch": 1.382859804378202,
      "grad_norm": 0.3573619712130518,
      "learning_rate": 4.274249223334484e-05,
      "loss": 0.479,
      "step": 1485
    },
    {
      "epoch": 1.3875174662319516,
      "grad_norm": 0.39866876376582483,
      "learning_rate": 4.269934414911978e-05,
      "loss": 0.4828,
      "step": 1490
    },
    {
      "epoch": 1.392175128085701,
      "grad_norm": 0.40462952205131425,
      "learning_rate": 4.265619606489472e-05,
      "loss": 0.4837,
      "step": 1495
    },
    {
      "epoch": 1.3968327899394504,
      "grad_norm": 0.3224443511494758,
      "learning_rate": 4.261304798066966e-05,
      "loss": 0.4705,
      "step": 1500
    },
    {
      "epoch": 1.4014904517931999,
      "grad_norm": 0.32680147592250536,
      "learning_rate": 4.25698998964446e-05,
      "loss": 0.475,
      "step": 1505
    },
    {
      "epoch": 1.4061481136469491,
      "grad_norm": 0.39652400797953574,
      "learning_rate": 4.252675181221954e-05,
      "loss": 0.4744,
      "step": 1510
    },
    {
      "epoch": 1.4108057755006986,
      "grad_norm": 0.3235231585370482,
      "learning_rate": 4.248360372799448e-05,
      "loss": 0.4862,
      "step": 1515
    },
    {
      "epoch": 1.4154634373544481,
      "grad_norm": 0.3382194467998487,
      "learning_rate": 4.244045564376942e-05,
      "loss": 0.4741,
      "step": 1520
    },
    {
      "epoch": 1.4201210992081974,
      "grad_norm": 0.37971760229961427,
      "learning_rate": 4.2397307559544355e-05,
      "loss": 0.4748,
      "step": 1525
    },
    {
      "epoch": 1.424778761061947,
      "grad_norm": 0.341259018896989,
      "learning_rate": 4.23541594753193e-05,
      "loss": 0.4678,
      "step": 1530
    },
    {
      "epoch": 1.4294364229156964,
      "grad_norm": 0.3360262192796755,
      "learning_rate": 4.231101139109423e-05,
      "loss": 0.483,
      "step": 1535
    },
    {
      "epoch": 1.4340940847694457,
      "grad_norm": 0.3229008090026825,
      "learning_rate": 4.2267863306869176e-05,
      "loss": 0.4726,
      "step": 1540
    },
    {
      "epoch": 1.4387517466231952,
      "grad_norm": 0.3516505526843625,
      "learning_rate": 4.222471522264412e-05,
      "loss": 0.4746,
      "step": 1545
    },
    {
      "epoch": 1.4434094084769447,
      "grad_norm": 0.37272166207985946,
      "learning_rate": 4.2181567138419054e-05,
      "loss": 0.4695,
      "step": 1550
    },
    {
      "epoch": 1.448067070330694,
      "grad_norm": 0.3153394702959596,
      "learning_rate": 4.2138419054193996e-05,
      "loss": 0.4655,
      "step": 1555
    },
    {
      "epoch": 1.4527247321844434,
      "grad_norm": 0.32283794445463454,
      "learning_rate": 4.209527096996894e-05,
      "loss": 0.4727,
      "step": 1560
    },
    {
      "epoch": 1.4573823940381927,
      "grad_norm": 0.336101060139209,
      "learning_rate": 4.2052122885743875e-05,
      "loss": 0.4827,
      "step": 1565
    },
    {
      "epoch": 1.4620400558919422,
      "grad_norm": 0.3309370873491488,
      "learning_rate": 4.200897480151881e-05,
      "loss": 0.4987,
      "step": 1570
    },
    {
      "epoch": 1.4666977177456917,
      "grad_norm": 0.3264764614496392,
      "learning_rate": 4.196582671729375e-05,
      "loss": 0.4763,
      "step": 1575
    },
    {
      "epoch": 1.4713553795994412,
      "grad_norm": 0.34009954849905466,
      "learning_rate": 4.1922678633068695e-05,
      "loss": 0.4646,
      "step": 1580
    },
    {
      "epoch": 1.4760130414531905,
      "grad_norm": 0.32847120805588387,
      "learning_rate": 4.187953054884363e-05,
      "loss": 0.474,
      "step": 1585
    },
    {
      "epoch": 1.48067070330694,
      "grad_norm": 0.3430731869580369,
      "learning_rate": 4.1836382464618573e-05,
      "loss": 0.4739,
      "step": 1590
    },
    {
      "epoch": 1.4853283651606892,
      "grad_norm": 0.3436434463557566,
      "learning_rate": 4.1793234380393516e-05,
      "loss": 0.4696,
      "step": 1595
    },
    {
      "epoch": 1.4899860270144387,
      "grad_norm": 0.34265063576553617,
      "learning_rate": 4.175008629616845e-05,
      "loss": 0.4776,
      "step": 1600
    },
    {
      "epoch": 1.4946436888681882,
      "grad_norm": 0.33525175617156117,
      "learning_rate": 4.1706938211943394e-05,
      "loss": 0.4686,
      "step": 1605
    },
    {
      "epoch": 1.4993013507219377,
      "grad_norm": 0.33574627227154613,
      "learning_rate": 4.166379012771833e-05,
      "loss": 0.4746,
      "step": 1610
    },
    {
      "epoch": 1.503959012575687,
      "grad_norm": 0.3540552306957625,
      "learning_rate": 4.1620642043493266e-05,
      "loss": 0.4762,
      "step": 1615
    },
    {
      "epoch": 1.5086166744294365,
      "grad_norm": 0.3253538609557244,
      "learning_rate": 4.157749395926821e-05,
      "loss": 0.4733,
      "step": 1620
    },
    {
      "epoch": 1.5132743362831858,
      "grad_norm": 0.3080178298859202,
      "learning_rate": 4.153434587504315e-05,
      "loss": 0.4796,
      "step": 1625
    },
    {
      "epoch": 1.5179319981369352,
      "grad_norm": 0.34619581752316514,
      "learning_rate": 4.1491197790818086e-05,
      "loss": 0.476,
      "step": 1630
    },
    {
      "epoch": 1.5225896599906847,
      "grad_norm": 0.34807894430751296,
      "learning_rate": 4.144804970659303e-05,
      "loss": 0.4877,
      "step": 1635
    },
    {
      "epoch": 1.5272473218444342,
      "grad_norm": 0.35640275334870986,
      "learning_rate": 4.140490162236797e-05,
      "loss": 0.48,
      "step": 1640
    },
    {
      "epoch": 1.5319049836981835,
      "grad_norm": 0.3292028405667507,
      "learning_rate": 4.1361753538142914e-05,
      "loss": 0.486,
      "step": 1645
    },
    {
      "epoch": 1.5365626455519328,
      "grad_norm": 0.3603528514959395,
      "learning_rate": 4.131860545391785e-05,
      "loss": 0.4685,
      "step": 1650
    },
    {
      "epoch": 1.5412203074056823,
      "grad_norm": 0.39676261926256534,
      "learning_rate": 4.1275457369692785e-05,
      "loss": 0.4662,
      "step": 1655
    },
    {
      "epoch": 1.5458779692594318,
      "grad_norm": 0.31990377289987015,
      "learning_rate": 4.123230928546773e-05,
      "loss": 0.4646,
      "step": 1660
    },
    {
      "epoch": 1.5505356311131813,
      "grad_norm": 0.34707760473093296,
      "learning_rate": 4.1189161201242664e-05,
      "loss": 0.479,
      "step": 1665
    },
    {
      "epoch": 1.5551932929669308,
      "grad_norm": 0.37908804378386346,
      "learning_rate": 4.1146013117017606e-05,
      "loss": 0.4728,
      "step": 1670
    },
    {
      "epoch": 1.55985095482068,
      "grad_norm": 0.3045188338766208,
      "learning_rate": 4.110286503279255e-05,
      "loss": 0.47,
      "step": 1675
    },
    {
      "epoch": 1.5645086166744293,
      "grad_norm": 0.3361741111890774,
      "learning_rate": 4.1059716948567484e-05,
      "loss": 0.4787,
      "step": 1680
    },
    {
      "epoch": 1.5691662785281788,
      "grad_norm": 0.3574832168029641,
      "learning_rate": 4.101656886434243e-05,
      "loss": 0.4678,
      "step": 1685
    },
    {
      "epoch": 1.5738239403819283,
      "grad_norm": 0.367180841158958,
      "learning_rate": 4.097342078011737e-05,
      "loss": 0.4749,
      "step": 1690
    },
    {
      "epoch": 1.5784816022356778,
      "grad_norm": 0.35818608823714465,
      "learning_rate": 4.0930272695892305e-05,
      "loss": 0.4722,
      "step": 1695
    },
    {
      "epoch": 1.583139264089427,
      "grad_norm": 0.31578465880076306,
      "learning_rate": 4.088712461166724e-05,
      "loss": 0.4774,
      "step": 1700
    },
    {
      "epoch": 1.5877969259431766,
      "grad_norm": 0.31508866562481463,
      "learning_rate": 4.084397652744218e-05,
      "loss": 0.471,
      "step": 1705
    },
    {
      "epoch": 1.5924545877969258,
      "grad_norm": 0.36824611622656683,
      "learning_rate": 4.080082844321712e-05,
      "loss": 0.4762,
      "step": 1710
    },
    {
      "epoch": 1.5971122496506753,
      "grad_norm": 0.3426993674515455,
      "learning_rate": 4.075768035899206e-05,
      "loss": 0.484,
      "step": 1715
    },
    {
      "epoch": 1.6017699115044248,
      "grad_norm": 0.3008918327462232,
      "learning_rate": 4.0714532274767004e-05,
      "loss": 0.467,
      "step": 1720
    },
    {
      "epoch": 1.6064275733581743,
      "grad_norm": 0.32833150736856775,
      "learning_rate": 4.0671384190541946e-05,
      "loss": 0.4701,
      "step": 1725
    },
    {
      "epoch": 1.6110852352119236,
      "grad_norm": 0.32063097511559757,
      "learning_rate": 4.062823610631688e-05,
      "loss": 0.4835,
      "step": 1730
    },
    {
      "epoch": 1.6157428970656729,
      "grad_norm": 0.36352500964855944,
      "learning_rate": 4.0585088022091825e-05,
      "loss": 0.4802,
      "step": 1735
    },
    {
      "epoch": 1.6204005589194224,
      "grad_norm": 0.3225535587995063,
      "learning_rate": 4.054193993786676e-05,
      "loss": 0.4604,
      "step": 1740
    },
    {
      "epoch": 1.6250582207731719,
      "grad_norm": 0.31867713270373665,
      "learning_rate": 4.0498791853641696e-05,
      "loss": 0.4963,
      "step": 1745
    },
    {
      "epoch": 1.6297158826269214,
      "grad_norm": 0.33926758810703683,
      "learning_rate": 4.045564376941664e-05,
      "loss": 0.4668,
      "step": 1750
    },
    {
      "epoch": 1.6343735444806708,
      "grad_norm": 0.31591352952091095,
      "learning_rate": 4.041249568519158e-05,
      "loss": 0.4851,
      "step": 1755
    },
    {
      "epoch": 1.6390312063344201,
      "grad_norm": 0.332915269981055,
      "learning_rate": 4.036934760096652e-05,
      "loss": 0.4636,
      "step": 1760
    },
    {
      "epoch": 1.6436888681881694,
      "grad_norm": 0.32584428396025494,
      "learning_rate": 4.032619951674146e-05,
      "loss": 0.4702,
      "step": 1765
    },
    {
      "epoch": 1.648346530041919,
      "grad_norm": 0.31461845523635235,
      "learning_rate": 4.02830514325164e-05,
      "loss": 0.4642,
      "step": 1770
    },
    {
      "epoch": 1.6530041918956684,
      "grad_norm": 0.3139180698192341,
      "learning_rate": 4.023990334829134e-05,
      "loss": 0.4818,
      "step": 1775
    },
    {
      "epoch": 1.6576618537494179,
      "grad_norm": 0.32017852481151093,
      "learning_rate": 4.019675526406628e-05,
      "loss": 0.4658,
      "step": 1780
    },
    {
      "epoch": 1.6623195156031674,
      "grad_norm": 0.31297448792098553,
      "learning_rate": 4.0153607179841216e-05,
      "loss": 0.4762,
      "step": 1785
    },
    {
      "epoch": 1.6669771774569166,
      "grad_norm": 0.3007144033803569,
      "learning_rate": 4.011045909561615e-05,
      "loss": 0.4666,
      "step": 1790
    },
    {
      "epoch": 1.671634839310666,
      "grad_norm": 0.3381471136534159,
      "learning_rate": 4.0067311011391094e-05,
      "loss": 0.4643,
      "step": 1795
    },
    {
      "epoch": 1.6762925011644154,
      "grad_norm": 0.2884812365679434,
      "learning_rate": 4.0024162927166037e-05,
      "loss": 0.4644,
      "step": 1800
    },
    {
      "epoch": 1.680950163018165,
      "grad_norm": 0.28987193235455067,
      "learning_rate": 3.998101484294098e-05,
      "loss": 0.4708,
      "step": 1805
    },
    {
      "epoch": 1.6856078248719144,
      "grad_norm": 0.3077486024488941,
      "learning_rate": 3.9937866758715915e-05,
      "loss": 0.4693,
      "step": 1810
    },
    {
      "epoch": 1.6902654867256637,
      "grad_norm": 0.32704118182174213,
      "learning_rate": 3.989471867449086e-05,
      "loss": 0.4707,
      "step": 1815
    },
    {
      "epoch": 1.6949231485794132,
      "grad_norm": 0.3283819140906282,
      "learning_rate": 3.98515705902658e-05,
      "loss": 0.4543,
      "step": 1820
    },
    {
      "epoch": 1.6995808104331624,
      "grad_norm": 0.3333682262370285,
      "learning_rate": 3.9808422506040736e-05,
      "loss": 0.4603,
      "step": 1825
    },
    {
      "epoch": 1.704238472286912,
      "grad_norm": 0.35199040828628375,
      "learning_rate": 3.976527442181567e-05,
      "loss": 0.4624,
      "step": 1830
    },
    {
      "epoch": 1.7088961341406614,
      "grad_norm": 0.3528620238838726,
      "learning_rate": 3.9722126337590614e-05,
      "loss": 0.4721,
      "step": 1835
    },
    {
      "epoch": 1.713553795994411,
      "grad_norm": 0.39010186425634624,
      "learning_rate": 3.967897825336555e-05,
      "loss": 0.4681,
      "step": 1840
    },
    {
      "epoch": 1.7182114578481602,
      "grad_norm": 0.31996702699290974,
      "learning_rate": 3.963583016914049e-05,
      "loss": 0.4699,
      "step": 1845
    },
    {
      "epoch": 1.7228691197019095,
      "grad_norm": 0.3186085018344506,
      "learning_rate": 3.9592682084915434e-05,
      "loss": 0.4812,
      "step": 1850
    },
    {
      "epoch": 1.727526781555659,
      "grad_norm": 0.31853419466768745,
      "learning_rate": 3.954953400069037e-05,
      "loss": 0.4587,
      "step": 1855
    },
    {
      "epoch": 1.7321844434094085,
      "grad_norm": 0.3066899660411568,
      "learning_rate": 3.950638591646531e-05,
      "loss": 0.4895,
      "step": 1860
    },
    {
      "epoch": 1.736842105263158,
      "grad_norm": 0.3449165466632536,
      "learning_rate": 3.9463237832240255e-05,
      "loss": 0.4646,
      "step": 1865
    },
    {
      "epoch": 1.7414997671169075,
      "grad_norm": 0.2905831966162277,
      "learning_rate": 3.942008974801519e-05,
      "loss": 0.4723,
      "step": 1870
    },
    {
      "epoch": 1.7461574289706567,
      "grad_norm": 0.3084846131529603,
      "learning_rate": 3.937694166379013e-05,
      "loss": 0.4883,
      "step": 1875
    },
    {
      "epoch": 1.750815090824406,
      "grad_norm": 0.32738525287630377,
      "learning_rate": 3.933379357956507e-05,
      "loss": 0.4774,
      "step": 1880
    },
    {
      "epoch": 1.7554727526781555,
      "grad_norm": 0.3553674850756716,
      "learning_rate": 3.9290645495340005e-05,
      "loss": 0.4793,
      "step": 1885
    },
    {
      "epoch": 1.760130414531905,
      "grad_norm": 0.32504835139136623,
      "learning_rate": 3.924749741111495e-05,
      "loss": 0.4733,
      "step": 1890
    },
    {
      "epoch": 1.7647880763856545,
      "grad_norm": 0.3660373919088391,
      "learning_rate": 3.920434932688989e-05,
      "loss": 0.4846,
      "step": 1895
    },
    {
      "epoch": 1.7694457382394038,
      "grad_norm": 0.29835372419955336,
      "learning_rate": 3.916120124266483e-05,
      "loss": 0.473,
      "step": 1900
    },
    {
      "epoch": 1.7741034000931533,
      "grad_norm": 0.32973502761026136,
      "learning_rate": 3.911805315843977e-05,
      "loss": 0.4723,
      "step": 1905
    },
    {
      "epoch": 1.7787610619469025,
      "grad_norm": 0.34149442775874744,
      "learning_rate": 3.9074905074214704e-05,
      "loss": 0.4635,
      "step": 1910
    },
    {
      "epoch": 1.783418723800652,
      "grad_norm": 0.3069244884670442,
      "learning_rate": 3.9031756989989646e-05,
      "loss": 0.4555,
      "step": 1915
    },
    {
      "epoch": 1.7880763856544015,
      "grad_norm": 0.32470264049742625,
      "learning_rate": 3.898860890576458e-05,
      "loss": 0.4646,
      "step": 1920
    },
    {
      "epoch": 1.792734047508151,
      "grad_norm": 0.3565233837135482,
      "learning_rate": 3.8945460821539525e-05,
      "loss": 0.4714,
      "step": 1925
    },
    {
      "epoch": 1.7973917093619003,
      "grad_norm": 0.3550740476288845,
      "learning_rate": 3.890231273731447e-05,
      "loss": 0.465,
      "step": 1930
    },
    {
      "epoch": 1.8020493712156498,
      "grad_norm": 0.322637125091311,
      "learning_rate": 3.88591646530894e-05,
      "loss": 0.4585,
      "step": 1935
    },
    {
      "epoch": 1.806707033069399,
      "grad_norm": 0.31190845331128736,
      "learning_rate": 3.8816016568864345e-05,
      "loss": 0.4609,
      "step": 1940
    },
    {
      "epoch": 1.8113646949231486,
      "grad_norm": 0.3040605517970688,
      "learning_rate": 3.877286848463929e-05,
      "loss": 0.474,
      "step": 1945
    },
    {
      "epoch": 1.816022356776898,
      "grad_norm": 0.3105217638590834,
      "learning_rate": 3.8729720400414224e-05,
      "loss": 0.4641,
      "step": 1950
    },
    {
      "epoch": 1.8206800186306475,
      "grad_norm": 0.29888163992617,
      "learning_rate": 3.868657231618916e-05,
      "loss": 0.4665,
      "step": 1955
    },
    {
      "epoch": 1.8253376804843968,
      "grad_norm": 0.3087539504225479,
      "learning_rate": 3.86434242319641e-05,
      "loss": 0.469,
      "step": 1960
    },
    {
      "epoch": 1.829995342338146,
      "grad_norm": 0.318033747002997,
      "learning_rate": 3.860027614773904e-05,
      "loss": 0.4713,
      "step": 1965
    },
    {
      "epoch": 1.8346530041918956,
      "grad_norm": 0.3317260951694933,
      "learning_rate": 3.855712806351398e-05,
      "loss": 0.4671,
      "step": 1970
    },
    {
      "epoch": 1.839310666045645,
      "grad_norm": 0.33033402082725805,
      "learning_rate": 3.851397997928892e-05,
      "loss": 0.473,
      "step": 1975
    },
    {
      "epoch": 1.8439683278993946,
      "grad_norm": 0.340633068907541,
      "learning_rate": 3.8470831895063865e-05,
      "loss": 0.4646,
      "step": 1980
    },
    {
      "epoch": 1.848625989753144,
      "grad_norm": 0.3264703509731795,
      "learning_rate": 3.84276838108388e-05,
      "loss": 0.4818,
      "step": 1985
    },
    {
      "epoch": 1.8532836516068933,
      "grad_norm": 0.3364791661773499,
      "learning_rate": 3.838453572661374e-05,
      "loss": 0.4719,
      "step": 1990
    },
    {
      "epoch": 1.8579413134606426,
      "grad_norm": 0.28301246367631205,
      "learning_rate": 3.834138764238868e-05,
      "loss": 0.455,
      "step": 1995
    },
    {
      "epoch": 1.8625989753143921,
      "grad_norm": 0.3134422465956333,
      "learning_rate": 3.8298239558163615e-05,
      "loss": 0.4684,
      "step": 2000
    },
    {
      "epoch": 1.8672566371681416,
      "grad_norm": 0.3151746089266361,
      "learning_rate": 3.825509147393856e-05,
      "loss": 0.4787,
      "step": 2005
    },
    {
      "epoch": 1.871914299021891,
      "grad_norm": 0.2840550078558945,
      "learning_rate": 3.82119433897135e-05,
      "loss": 0.4607,
      "step": 2010
    },
    {
      "epoch": 1.8765719608756404,
      "grad_norm": 0.27799980621874687,
      "learning_rate": 3.8168795305488435e-05,
      "loss": 0.4802,
      "step": 2015
    },
    {
      "epoch": 1.8812296227293899,
      "grad_norm": 0.316828357752238,
      "learning_rate": 3.812564722126338e-05,
      "loss": 0.4615,
      "step": 2020
    },
    {
      "epoch": 1.8858872845831391,
      "grad_norm": 0.31358265496964005,
      "learning_rate": 3.808249913703832e-05,
      "loss": 0.4686,
      "step": 2025
    },
    {
      "epoch": 1.8905449464368886,
      "grad_norm": 0.3281373864511338,
      "learning_rate": 3.8039351052813256e-05,
      "loss": 0.4644,
      "step": 2030
    },
    {
      "epoch": 1.8952026082906381,
      "grad_norm": 0.30142229938481957,
      "learning_rate": 3.79962029685882e-05,
      "loss": 0.4609,
      "step": 2035
    },
    {
      "epoch": 1.8998602701443876,
      "grad_norm": 0.3092006503428952,
      "learning_rate": 3.7953054884363134e-05,
      "loss": 0.4741,
      "step": 2040
    },
    {
      "epoch": 1.904517931998137,
      "grad_norm": 0.2938226791442561,
      "learning_rate": 3.790990680013807e-05,
      "loss": 0.479,
      "step": 2045
    },
    {
      "epoch": 1.9091755938518864,
      "grad_norm": 0.3240865470753573,
      "learning_rate": 3.786675871591301e-05,
      "loss": 0.4663,
      "step": 2050
    },
    {
      "epoch": 1.9138332557056357,
      "grad_norm": 0.2865453714409431,
      "learning_rate": 3.7823610631687955e-05,
      "loss": 0.4659,
      "step": 2055
    },
    {
      "epoch": 1.9184909175593852,
      "grad_norm": 0.317618776283293,
      "learning_rate": 3.77804625474629e-05,
      "loss": 0.4611,
      "step": 2060
    },
    {
      "epoch": 1.9231485794131347,
      "grad_norm": 0.30311910097970307,
      "learning_rate": 3.773731446323783e-05,
      "loss": 0.4656,
      "step": 2065
    },
    {
      "epoch": 1.9278062412668842,
      "grad_norm": 0.3163405445690529,
      "learning_rate": 3.7694166379012776e-05,
      "loss": 0.4685,
      "step": 2070
    },
    {
      "epoch": 1.9324639031206334,
      "grad_norm": 0.2948346352579754,
      "learning_rate": 3.765101829478772e-05,
      "loss": 0.4659,
      "step": 2075
    },
    {
      "epoch": 1.9371215649743827,
      "grad_norm": 0.3323488242746519,
      "learning_rate": 3.7607870210562654e-05,
      "loss": 0.4576,
      "step": 2080
    },
    {
      "epoch": 1.9417792268281322,
      "grad_norm": 0.29471611432748746,
      "learning_rate": 3.756472212633759e-05,
      "loss": 0.4732,
      "step": 2085
    },
    {
      "epoch": 1.9464368886818817,
      "grad_norm": 0.2874816006032476,
      "learning_rate": 3.752157404211253e-05,
      "loss": 0.4501,
      "step": 2090
    },
    {
      "epoch": 1.9510945505356312,
      "grad_norm": 0.29429791205316697,
      "learning_rate": 3.747842595788747e-05,
      "loss": 0.4629,
      "step": 2095
    },
    {
      "epoch": 1.9557522123893807,
      "grad_norm": 0.33763560708172646,
      "learning_rate": 3.743527787366241e-05,
      "loss": 0.4613,
      "step": 2100
    },
    {
      "epoch": 1.96040987424313,
      "grad_norm": 0.3268984958188359,
      "learning_rate": 3.739212978943735e-05,
      "loss": 0.472,
      "step": 2105
    },
    {
      "epoch": 1.9650675360968792,
      "grad_norm": 0.29902163689150446,
      "learning_rate": 3.734898170521229e-05,
      "loss": 0.4537,
      "step": 2110
    },
    {
      "epoch": 1.9697251979506287,
      "grad_norm": 0.33038712163483136,
      "learning_rate": 3.730583362098723e-05,
      "loss": 0.4666,
      "step": 2115
    },
    {
      "epoch": 1.9743828598043782,
      "grad_norm": 0.30803719639938826,
      "learning_rate": 3.7262685536762174e-05,
      "loss": 0.47,
      "step": 2120
    },
    {
      "epoch": 1.9790405216581277,
      "grad_norm": 0.33156631458190233,
      "learning_rate": 3.721953745253711e-05,
      "loss": 0.4592,
      "step": 2125
    },
    {
      "epoch": 1.983698183511877,
      "grad_norm": 0.29369096568163894,
      "learning_rate": 3.7176389368312045e-05,
      "loss": 0.4629,
      "step": 2130
    },
    {
      "epoch": 1.9883558453656265,
      "grad_norm": 0.3092328629194217,
      "learning_rate": 3.713324128408699e-05,
      "loss": 0.4696,
      "step": 2135
    },
    {
      "epoch": 1.9930135072193758,
      "grad_norm": 0.30115880075274587,
      "learning_rate": 3.709009319986193e-05,
      "loss": 0.4683,
      "step": 2140
    },
    {
      "epoch": 1.9976711690731253,
      "grad_norm": 0.33330205032492555,
      "learning_rate": 3.7046945115636866e-05,
      "loss": 0.4684,
      "step": 2145
    },
    {
      "epoch": 2.0018630647415,
      "grad_norm": 0.34876999866482683,
      "learning_rate": 3.700379703141181e-05,
      "loss": 0.4616,
      "step": 2150
    },
    {
      "epoch": 2.0065207265952494,
      "grad_norm": 0.3638647035628797,
      "learning_rate": 3.696064894718675e-05,
      "loss": 0.4313,
      "step": 2155
    },
    {
      "epoch": 2.0111783884489984,
      "grad_norm": 0.34428941420795,
      "learning_rate": 3.6917500862961687e-05,
      "loss": 0.4237,
      "step": 2160
    },
    {
      "epoch": 2.015836050302748,
      "grad_norm": 0.3397188179653322,
      "learning_rate": 3.687435277873663e-05,
      "loss": 0.4303,
      "step": 2165
    },
    {
      "epoch": 2.0204937121564974,
      "grad_norm": 0.32110541283149646,
      "learning_rate": 3.6831204694511565e-05,
      "loss": 0.4162,
      "step": 2170
    },
    {
      "epoch": 2.025151374010247,
      "grad_norm": 0.3151397785202642,
      "learning_rate": 3.67880566102865e-05,
      "loss": 0.4379,
      "step": 2175
    },
    {
      "epoch": 2.0298090358639964,
      "grad_norm": 0.3152281636572169,
      "learning_rate": 3.674490852606144e-05,
      "loss": 0.4237,
      "step": 2180
    },
    {
      "epoch": 2.034466697717746,
      "grad_norm": 0.3695132941596053,
      "learning_rate": 3.6701760441836386e-05,
      "loss": 0.4341,
      "step": 2185
    },
    {
      "epoch": 2.039124359571495,
      "grad_norm": 0.3459325169856267,
      "learning_rate": 3.665861235761132e-05,
      "loss": 0.4297,
      "step": 2190
    },
    {
      "epoch": 2.0437820214252445,
      "grad_norm": 0.31236527300389905,
      "learning_rate": 3.6615464273386264e-05,
      "loss": 0.4284,
      "step": 2195
    },
    {
      "epoch": 2.048439683278994,
      "grad_norm": 0.3176395777062693,
      "learning_rate": 3.6572316189161206e-05,
      "loss": 0.4267,
      "step": 2200
    },
    {
      "epoch": 2.0530973451327434,
      "grad_norm": 0.2950691850967291,
      "learning_rate": 3.652916810493614e-05,
      "loss": 0.4334,
      "step": 2205
    },
    {
      "epoch": 2.057755006986493,
      "grad_norm": 0.31299477347836824,
      "learning_rate": 3.6486020020711085e-05,
      "loss": 0.4186,
      "step": 2210
    },
    {
      "epoch": 2.062412668840242,
      "grad_norm": 0.377436219411388,
      "learning_rate": 3.644287193648602e-05,
      "loss": 0.4267,
      "step": 2215
    },
    {
      "epoch": 2.0670703306939915,
      "grad_norm": 0.31036015331464156,
      "learning_rate": 3.639972385226096e-05,
      "loss": 0.4304,
      "step": 2220
    },
    {
      "epoch": 2.071727992547741,
      "grad_norm": 0.33969558138730943,
      "learning_rate": 3.63565757680359e-05,
      "loss": 0.4268,
      "step": 2225
    },
    {
      "epoch": 2.0763856544014905,
      "grad_norm": 0.3107184842395639,
      "learning_rate": 3.631342768381084e-05,
      "loss": 0.4287,
      "step": 2230
    },
    {
      "epoch": 2.08104331625524,
      "grad_norm": 0.3051677866590822,
      "learning_rate": 3.6270279599585783e-05,
      "loss": 0.422,
      "step": 2235
    },
    {
      "epoch": 2.0857009781089895,
      "grad_norm": 0.29914215009236755,
      "learning_rate": 3.622713151536072e-05,
      "loss": 0.433,
      "step": 2240
    },
    {
      "epoch": 2.0903586399627385,
      "grad_norm": 0.3296867411174138,
      "learning_rate": 3.618398343113566e-05,
      "loss": 0.4329,
      "step": 2245
    },
    {
      "epoch": 2.095016301816488,
      "grad_norm": 0.33857097297537636,
      "learning_rate": 3.6140835346910604e-05,
      "loss": 0.425,
      "step": 2250
    },
    {
      "epoch": 2.0996739636702375,
      "grad_norm": 0.3521956476979629,
      "learning_rate": 3.609768726268554e-05,
      "loss": 0.4377,
      "step": 2255
    },
    {
      "epoch": 2.104331625523987,
      "grad_norm": 0.3568796005147484,
      "learning_rate": 3.6054539178460476e-05,
      "loss": 0.4241,
      "step": 2260
    },
    {
      "epoch": 2.1089892873777365,
      "grad_norm": 0.3069614388311688,
      "learning_rate": 3.601139109423542e-05,
      "loss": 0.4397,
      "step": 2265
    },
    {
      "epoch": 2.113646949231486,
      "grad_norm": 0.3356486573686718,
      "learning_rate": 3.5968243010010354e-05,
      "loss": 0.4273,
      "step": 2270
    },
    {
      "epoch": 2.118304611085235,
      "grad_norm": 0.35591164093478217,
      "learning_rate": 3.5925094925785296e-05,
      "loss": 0.4286,
      "step": 2275
    },
    {
      "epoch": 2.1229622729389845,
      "grad_norm": 0.3058132160755914,
      "learning_rate": 3.588194684156024e-05,
      "loss": 0.4285,
      "step": 2280
    },
    {
      "epoch": 2.127619934792734,
      "grad_norm": 0.3166446437861686,
      "learning_rate": 3.5838798757335175e-05,
      "loss": 0.4338,
      "step": 2285
    },
    {
      "epoch": 2.1322775966464835,
      "grad_norm": 0.36843753132192086,
      "learning_rate": 3.579565067311012e-05,
      "loss": 0.4313,
      "step": 2290
    },
    {
      "epoch": 2.136935258500233,
      "grad_norm": 0.3071033619445971,
      "learning_rate": 3.575250258888505e-05,
      "loss": 0.4388,
      "step": 2295
    },
    {
      "epoch": 2.1415929203539825,
      "grad_norm": 0.3145913245764276,
      "learning_rate": 3.5709354504659995e-05,
      "loss": 0.4399,
      "step": 2300
    },
    {
      "epoch": 2.1462505822077316,
      "grad_norm": 0.29132330028161085,
      "learning_rate": 3.566620642043493e-05,
      "loss": 0.4307,
      "step": 2305
    },
    {
      "epoch": 2.150908244061481,
      "grad_norm": 0.3083456086449428,
      "learning_rate": 3.5623058336209874e-05,
      "loss": 0.4146,
      "step": 2310
    },
    {
      "epoch": 2.1555659059152306,
      "grad_norm": 0.32348252695160706,
      "learning_rate": 3.5579910251984816e-05,
      "loss": 0.422,
      "step": 2315
    },
    {
      "epoch": 2.16022356776898,
      "grad_norm": 0.3034515740572575,
      "learning_rate": 3.553676216775975e-05,
      "loss": 0.425,
      "step": 2320
    },
    {
      "epoch": 2.1648812296227296,
      "grad_norm": 0.32576416461243163,
      "learning_rate": 3.5493614083534694e-05,
      "loss": 0.435,
      "step": 2325
    },
    {
      "epoch": 2.1695388914764786,
      "grad_norm": 0.2966282038467855,
      "learning_rate": 3.545046599930964e-05,
      "loss": 0.4371,
      "step": 2330
    },
    {
      "epoch": 2.174196553330228,
      "grad_norm": 0.3043342104692987,
      "learning_rate": 3.540731791508457e-05,
      "loss": 0.421,
      "step": 2335
    },
    {
      "epoch": 2.1788542151839776,
      "grad_norm": 0.290553413040975,
      "learning_rate": 3.536416983085951e-05,
      "loss": 0.4337,
      "step": 2340
    },
    {
      "epoch": 2.183511877037727,
      "grad_norm": 0.29062006657824413,
      "learning_rate": 3.532102174663445e-05,
      "loss": 0.4329,
      "step": 2345
    },
    {
      "epoch": 2.1881695388914766,
      "grad_norm": 0.3153163849705803,
      "learning_rate": 3.5277873662409386e-05,
      "loss": 0.4316,
      "step": 2350
    },
    {
      "epoch": 2.192827200745226,
      "grad_norm": 0.3073600575111337,
      "learning_rate": 3.523472557818433e-05,
      "loss": 0.4246,
      "step": 2355
    },
    {
      "epoch": 2.197484862598975,
      "grad_norm": 0.32837802681216166,
      "learning_rate": 3.519157749395927e-05,
      "loss": 0.434,
      "step": 2360
    },
    {
      "epoch": 2.2021425244527246,
      "grad_norm": 0.3261194867585345,
      "learning_rate": 3.514842940973421e-05,
      "loss": 0.4312,
      "step": 2365
    },
    {
      "epoch": 2.206800186306474,
      "grad_norm": 0.3186686180806262,
      "learning_rate": 3.510528132550915e-05,
      "loss": 0.4358,
      "step": 2370
    },
    {
      "epoch": 2.2114578481602236,
      "grad_norm": 0.4045213035548189,
      "learning_rate": 3.506213324128409e-05,
      "loss": 0.4386,
      "step": 2375
    },
    {
      "epoch": 2.216115510013973,
      "grad_norm": 0.3318798827416219,
      "learning_rate": 3.501898515705903e-05,
      "loss": 0.4275,
      "step": 2380
    },
    {
      "epoch": 2.2207731718677226,
      "grad_norm": 0.3018574448505957,
      "learning_rate": 3.4975837072833964e-05,
      "loss": 0.4423,
      "step": 2385
    },
    {
      "epoch": 2.2254308337214717,
      "grad_norm": 0.280964214000006,
      "learning_rate": 3.4932688988608906e-05,
      "loss": 0.431,
      "step": 2390
    },
    {
      "epoch": 2.230088495575221,
      "grad_norm": 0.3169517129245141,
      "learning_rate": 3.488954090438385e-05,
      "loss": 0.4307,
      "step": 2395
    },
    {
      "epoch": 2.2347461574289706,
      "grad_norm": 0.30771204080019765,
      "learning_rate": 3.4846392820158784e-05,
      "loss": 0.4219,
      "step": 2400
    },
    {
      "epoch": 2.23940381928272,
      "grad_norm": 0.29594438427833025,
      "learning_rate": 3.480324473593373e-05,
      "loss": 0.4319,
      "step": 2405
    },
    {
      "epoch": 2.2440614811364696,
      "grad_norm": 0.3106532020326551,
      "learning_rate": 3.476009665170867e-05,
      "loss": 0.4305,
      "step": 2410
    },
    {
      "epoch": 2.248719142990219,
      "grad_norm": 0.32086828679169443,
      "learning_rate": 3.4716948567483605e-05,
      "loss": 0.4266,
      "step": 2415
    },
    {
      "epoch": 2.253376804843968,
      "grad_norm": 0.31496970793657364,
      "learning_rate": 3.467380048325855e-05,
      "loss": 0.4145,
      "step": 2420
    },
    {
      "epoch": 2.2580344666977177,
      "grad_norm": 0.3199466913936517,
      "learning_rate": 3.463065239903348e-05,
      "loss": 0.44,
      "step": 2425
    },
    {
      "epoch": 2.262692128551467,
      "grad_norm": 0.28452569938107114,
      "learning_rate": 3.458750431480842e-05,
      "loss": 0.4221,
      "step": 2430
    },
    {
      "epoch": 2.2673497904052167,
      "grad_norm": 0.29408635322478704,
      "learning_rate": 3.454435623058336e-05,
      "loss": 0.4313,
      "step": 2435
    },
    {
      "epoch": 2.272007452258966,
      "grad_norm": 0.3236978178737715,
      "learning_rate": 3.4501208146358304e-05,
      "loss": 0.4287,
      "step": 2440
    },
    {
      "epoch": 2.276665114112715,
      "grad_norm": 0.30866791184876435,
      "learning_rate": 3.445806006213324e-05,
      "loss": 0.4277,
      "step": 2445
    },
    {
      "epoch": 2.2813227759664647,
      "grad_norm": 0.31001961320269333,
      "learning_rate": 3.441491197790818e-05,
      "loss": 0.4278,
      "step": 2450
    },
    {
      "epoch": 2.285980437820214,
      "grad_norm": 0.3165660533568062,
      "learning_rate": 3.4371763893683125e-05,
      "loss": 0.4236,
      "step": 2455
    },
    {
      "epoch": 2.2906380996739637,
      "grad_norm": 0.29798149264042,
      "learning_rate": 3.432861580945806e-05,
      "loss": 0.4201,
      "step": 2460
    },
    {
      "epoch": 2.295295761527713,
      "grad_norm": 0.3230554223714303,
      "learning_rate": 3.4285467725233e-05,
      "loss": 0.4419,
      "step": 2465
    },
    {
      "epoch": 2.2999534233814627,
      "grad_norm": 0.3111760899255027,
      "learning_rate": 3.424231964100794e-05,
      "loss": 0.4327,
      "step": 2470
    },
    {
      "epoch": 2.3046110852352117,
      "grad_norm": 0.2815109178483268,
      "learning_rate": 3.419917155678288e-05,
      "loss": 0.4353,
      "step": 2475
    },
    {
      "epoch": 2.3092687470889612,
      "grad_norm": 0.33736483651132515,
      "learning_rate": 3.415602347255782e-05,
      "loss": 0.4285,
      "step": 2480
    },
    {
      "epoch": 2.3139264089427107,
      "grad_norm": 0.3104191025907746,
      "learning_rate": 3.411287538833276e-05,
      "loss": 0.4409,
      "step": 2485
    },
    {
      "epoch": 2.3185840707964602,
      "grad_norm": 0.29555579717032815,
      "learning_rate": 3.40697273041077e-05,
      "loss": 0.4391,
      "step": 2490
    },
    {
      "epoch": 2.3232417326502097,
      "grad_norm": 0.33934038820293116,
      "learning_rate": 3.402657921988264e-05,
      "loss": 0.4294,
      "step": 2495
    },
    {
      "epoch": 2.3278993945039588,
      "grad_norm": 0.3358901251646842,
      "learning_rate": 3.398343113565758e-05,
      "loss": 0.4171,
      "step": 2500
    },
    {
      "epoch": 2.3325570563577083,
      "grad_norm": 0.307899669535085,
      "learning_rate": 3.394028305143252e-05,
      "loss": 0.4283,
      "step": 2505
    },
    {
      "epoch": 2.3372147182114578,
      "grad_norm": 0.3081772703192549,
      "learning_rate": 3.389713496720746e-05,
      "loss": 0.4295,
      "step": 2510
    },
    {
      "epoch": 2.3418723800652073,
      "grad_norm": 0.29721255722715945,
      "learning_rate": 3.3853986882982394e-05,
      "loss": 0.4306,
      "step": 2515
    },
    {
      "epoch": 2.3465300419189568,
      "grad_norm": 0.30468366765484745,
      "learning_rate": 3.381083879875734e-05,
      "loss": 0.4349,
      "step": 2520
    },
    {
      "epoch": 2.3511877037727063,
      "grad_norm": 0.28941211631647595,
      "learning_rate": 3.376769071453227e-05,
      "loss": 0.432,
      "step": 2525
    },
    {
      "epoch": 2.3558453656264557,
      "grad_norm": 0.31755752891066114,
      "learning_rate": 3.3724542630307215e-05,
      "loss": 0.4315,
      "step": 2530
    },
    {
      "epoch": 2.360503027480205,
      "grad_norm": 0.2805711003592571,
      "learning_rate": 3.368139454608216e-05,
      "loss": 0.4191,
      "step": 2535
    },
    {
      "epoch": 2.3651606893339543,
      "grad_norm": 0.2853179672968521,
      "learning_rate": 3.363824646185709e-05,
      "loss": 0.4169,
      "step": 2540
    },
    {
      "epoch": 2.369818351187704,
      "grad_norm": 0.299899215279406,
      "learning_rate": 3.3595098377632036e-05,
      "loss": 0.4327,
      "step": 2545
    },
    {
      "epoch": 2.3744760130414533,
      "grad_norm": 0.31095024191931675,
      "learning_rate": 3.355195029340698e-05,
      "loss": 0.4334,
      "step": 2550
    },
    {
      "epoch": 2.3791336748952028,
      "grad_norm": 0.30076514494762585,
      "learning_rate": 3.3508802209181914e-05,
      "loss": 0.4246,
      "step": 2555
    },
    {
      "epoch": 2.383791336748952,
      "grad_norm": 0.31448486126840525,
      "learning_rate": 3.346565412495685e-05,
      "loss": 0.4279,
      "step": 2560
    },
    {
      "epoch": 2.3884489986027013,
      "grad_norm": 0.2966752118078948,
      "learning_rate": 3.342250604073179e-05,
      "loss": 0.4305,
      "step": 2565
    },
    {
      "epoch": 2.393106660456451,
      "grad_norm": 0.29506990970778374,
      "learning_rate": 3.3379357956506735e-05,
      "loss": 0.429,
      "step": 2570
    },
    {
      "epoch": 2.3977643223102003,
      "grad_norm": 0.28014977614462866,
      "learning_rate": 3.333620987228167e-05,
      "loss": 0.4314,
      "step": 2575
    },
    {
      "epoch": 2.40242198416395,
      "grad_norm": 0.3186450878821365,
      "learning_rate": 3.329306178805661e-05,
      "loss": 0.4212,
      "step": 2580
    },
    {
      "epoch": 2.4070796460176993,
      "grad_norm": 0.30194723988287164,
      "learning_rate": 3.3249913703831555e-05,
      "loss": 0.4229,
      "step": 2585
    },
    {
      "epoch": 2.4117373078714484,
      "grad_norm": 0.2848302734640622,
      "learning_rate": 3.320676561960649e-05,
      "loss": 0.4174,
      "step": 2590
    },
    {
      "epoch": 2.416394969725198,
      "grad_norm": 0.31164725236471075,
      "learning_rate": 3.3163617535381434e-05,
      "loss": 0.4373,
      "step": 2595
    },
    {
      "epoch": 2.4210526315789473,
      "grad_norm": 0.2935852260735618,
      "learning_rate": 3.312046945115637e-05,
      "loss": 0.4331,
      "step": 2600
    },
    {
      "epoch": 2.425710293432697,
      "grad_norm": 0.3334206758431403,
      "learning_rate": 3.3077321366931305e-05,
      "loss": 0.424,
      "step": 2605
    },
    {
      "epoch": 2.4303679552864463,
      "grad_norm": 0.33071091975960704,
      "learning_rate": 3.303417328270625e-05,
      "loss": 0.4193,
      "step": 2610
    },
    {
      "epoch": 2.4350256171401954,
      "grad_norm": 0.33874264470019955,
      "learning_rate": 3.299102519848119e-05,
      "loss": 0.4227,
      "step": 2615
    },
    {
      "epoch": 2.439683278993945,
      "grad_norm": 0.3127602756699459,
      "learning_rate": 3.2947877114256126e-05,
      "loss": 0.4281,
      "step": 2620
    },
    {
      "epoch": 2.4443409408476944,
      "grad_norm": 0.2893531393186613,
      "learning_rate": 3.290472903003107e-05,
      "loss": 0.4265,
      "step": 2625
    },
    {
      "epoch": 2.448998602701444,
      "grad_norm": 0.31123964614981114,
      "learning_rate": 3.286158094580601e-05,
      "loss": 0.4208,
      "step": 2630
    },
    {
      "epoch": 2.4536562645551934,
      "grad_norm": 0.2997379816868446,
      "learning_rate": 3.281843286158095e-05,
      "loss": 0.4333,
      "step": 2635
    },
    {
      "epoch": 2.458313926408943,
      "grad_norm": 0.310528060493364,
      "learning_rate": 3.277528477735589e-05,
      "loss": 0.4317,
      "step": 2640
    },
    {
      "epoch": 2.4629715882626924,
      "grad_norm": 0.30904159079478105,
      "learning_rate": 3.2732136693130825e-05,
      "loss": 0.4313,
      "step": 2645
    },
    {
      "epoch": 2.4676292501164414,
      "grad_norm": 0.2994441728887888,
      "learning_rate": 3.268898860890577e-05,
      "loss": 0.423,
      "step": 2650
    },
    {
      "epoch": 2.472286911970191,
      "grad_norm": 0.3098138708331934,
      "learning_rate": 3.26458405246807e-05,
      "loss": 0.4312,
      "step": 2655
    },
    {
      "epoch": 2.4769445738239404,
      "grad_norm": 0.300798197098642,
      "learning_rate": 3.2602692440455645e-05,
      "loss": 0.4241,
      "step": 2660
    },
    {
      "epoch": 2.48160223567769,
      "grad_norm": 0.3427234845607514,
      "learning_rate": 3.255954435623059e-05,
      "loss": 0.4244,
      "step": 2665
    },
    {
      "epoch": 2.4862598975314394,
      "grad_norm": 0.3176208519254816,
      "learning_rate": 3.2516396272005524e-05,
      "loss": 0.4309,
      "step": 2670
    },
    {
      "epoch": 2.4909175593851884,
      "grad_norm": 0.30328199370922204,
      "learning_rate": 3.2473248187780466e-05,
      "loss": 0.4218,
      "step": 2675
    },
    {
      "epoch": 2.495575221238938,
      "grad_norm": 0.32715405904683964,
      "learning_rate": 3.243010010355541e-05,
      "loss": 0.4241,
      "step": 2680
    },
    {
      "epoch": 2.5002328830926874,
      "grad_norm": 0.3203922422446222,
      "learning_rate": 3.2386952019330344e-05,
      "loss": 0.4281,
      "step": 2685
    },
    {
      "epoch": 2.504890544946437,
      "grad_norm": 0.3193382692860072,
      "learning_rate": 3.234380393510528e-05,
      "loss": 0.4146,
      "step": 2690
    },
    {
      "epoch": 2.5095482068001864,
      "grad_norm": 0.29768357238524246,
      "learning_rate": 3.230065585088022e-05,
      "loss": 0.4261,
      "step": 2695
    },
    {
      "epoch": 2.514205868653936,
      "grad_norm": 0.29123159620806016,
      "learning_rate": 3.225750776665516e-05,
      "loss": 0.4215,
      "step": 2700
    },
    {
      "epoch": 2.5188635305076854,
      "grad_norm": 0.3147697661985007,
      "learning_rate": 3.22143596824301e-05,
      "loss": 0.4329,
      "step": 2705
    },
    {
      "epoch": 2.5235211923614345,
      "grad_norm": 0.31127884309392934,
      "learning_rate": 3.217121159820504e-05,
      "loss": 0.4242,
      "step": 2710
    },
    {
      "epoch": 2.528178854215184,
      "grad_norm": 0.2943217455562263,
      "learning_rate": 3.2128063513979986e-05,
      "loss": 0.4335,
      "step": 2715
    },
    {
      "epoch": 2.5328365160689335,
      "grad_norm": 0.30444806224577065,
      "learning_rate": 3.208491542975492e-05,
      "loss": 0.4438,
      "step": 2720
    },
    {
      "epoch": 2.537494177922683,
      "grad_norm": 0.2892372970626022,
      "learning_rate": 3.204176734552986e-05,
      "loss": 0.4268,
      "step": 2725
    },
    {
      "epoch": 2.542151839776432,
      "grad_norm": 0.3058777747817771,
      "learning_rate": 3.19986192613048e-05,
      "loss": 0.437,
      "step": 2730
    },
    {
      "epoch": 2.5468095016301815,
      "grad_norm": 0.3166188890900455,
      "learning_rate": 3.1955471177079735e-05,
      "loss": 0.4247,
      "step": 2735
    },
    {
      "epoch": 2.551467163483931,
      "grad_norm": 0.3180650561216633,
      "learning_rate": 3.191232309285468e-05,
      "loss": 0.4233,
      "step": 2740
    },
    {
      "epoch": 2.5561248253376805,
      "grad_norm": 0.28158379940389555,
      "learning_rate": 3.186917500862962e-05,
      "loss": 0.4224,
      "step": 2745
    },
    {
      "epoch": 2.56078248719143,
      "grad_norm": 0.2726355767856877,
      "learning_rate": 3.1826026924404556e-05,
      "loss": 0.422,
      "step": 2750
    },
    {
      "epoch": 2.5654401490451795,
      "grad_norm": 0.30025431393798413,
      "learning_rate": 3.17828788401795e-05,
      "loss": 0.4175,
      "step": 2755
    },
    {
      "epoch": 2.570097810898929,
      "grad_norm": 0.30689705034184633,
      "learning_rate": 3.173973075595444e-05,
      "loss": 0.4303,
      "step": 2760
    },
    {
      "epoch": 2.574755472752678,
      "grad_norm": 0.3108439840023687,
      "learning_rate": 3.169658267172938e-05,
      "loss": 0.4223,
      "step": 2765
    },
    {
      "epoch": 2.5794131346064275,
      "grad_norm": 0.3004591002002451,
      "learning_rate": 3.165343458750431e-05,
      "loss": 0.4247,
      "step": 2770
    },
    {
      "epoch": 2.584070796460177,
      "grad_norm": 0.2949798929862715,
      "learning_rate": 3.1610286503279255e-05,
      "loss": 0.423,
      "step": 2775
    },
    {
      "epoch": 2.5887284583139265,
      "grad_norm": 0.32801179066930836,
      "learning_rate": 3.156713841905419e-05,
      "loss": 0.4299,
      "step": 2780
    },
    {
      "epoch": 2.5933861201676756,
      "grad_norm": 0.2859880937881386,
      "learning_rate": 3.152399033482913e-05,
      "loss": 0.4345,
      "step": 2785
    },
    {
      "epoch": 2.598043782021425,
      "grad_norm": 0.3064204254281662,
      "learning_rate": 3.1480842250604076e-05,
      "loss": 0.4352,
      "step": 2790
    },
    {
      "epoch": 2.6027014438751745,
      "grad_norm": 0.29024367264885154,
      "learning_rate": 3.143769416637902e-05,
      "loss": 0.4248,
      "step": 2795
    },
    {
      "epoch": 2.607359105728924,
      "grad_norm": 0.2993218419091459,
      "learning_rate": 3.1394546082153954e-05,
      "loss": 0.426,
      "step": 2800
    },
    {
      "epoch": 2.6120167675826735,
      "grad_norm": 0.2609045039444352,
      "learning_rate": 3.13513979979289e-05,
      "loss": 0.4219,
      "step": 2805
    },
    {
      "epoch": 2.616674429436423,
      "grad_norm": 0.29807486533339206,
      "learning_rate": 3.130824991370383e-05,
      "loss": 0.429,
      "step": 2810
    },
    {
      "epoch": 2.6213320912901725,
      "grad_norm": 0.2740221460537428,
      "learning_rate": 3.126510182947877e-05,
      "loss": 0.4221,
      "step": 2815
    },
    {
      "epoch": 2.625989753143922,
      "grad_norm": 0.28780393763390794,
      "learning_rate": 3.122195374525371e-05,
      "loss": 0.4332,
      "step": 2820
    },
    {
      "epoch": 2.630647414997671,
      "grad_norm": 0.2871889425921111,
      "learning_rate": 3.117880566102865e-05,
      "loss": 0.4333,
      "step": 2825
    },
    {
      "epoch": 2.6353050768514206,
      "grad_norm": 0.36044361421969257,
      "learning_rate": 3.113565757680359e-05,
      "loss": 0.423,
      "step": 2830
    },
    {
      "epoch": 2.63996273870517,
      "grad_norm": 0.32176541030886585,
      "learning_rate": 3.109250949257853e-05,
      "loss": 0.4339,
      "step": 2835
    },
    {
      "epoch": 2.6446204005589196,
      "grad_norm": 0.2864091195579079,
      "learning_rate": 3.1049361408353474e-05,
      "loss": 0.437,
      "step": 2840
    },
    {
      "epoch": 2.6492780624126686,
      "grad_norm": 0.30950724733902146,
      "learning_rate": 3.100621332412841e-05,
      "loss": 0.4319,
      "step": 2845
    },
    {
      "epoch": 2.653935724266418,
      "grad_norm": 0.29209082163980316,
      "learning_rate": 3.096306523990335e-05,
      "loss": 0.4327,
      "step": 2850
    },
    {
      "epoch": 2.6585933861201676,
      "grad_norm": 0.2735949787594406,
      "learning_rate": 3.091991715567829e-05,
      "loss": 0.4256,
      "step": 2855
    },
    {
      "epoch": 2.663251047973917,
      "grad_norm": 0.30195572623724537,
      "learning_rate": 3.0876769071453223e-05,
      "loss": 0.4325,
      "step": 2860
    },
    {
      "epoch": 2.6679087098276666,
      "grad_norm": 0.3002192223611597,
      "learning_rate": 3.0833620987228166e-05,
      "loss": 0.4313,
      "step": 2865
    },
    {
      "epoch": 2.672566371681416,
      "grad_norm": 0.3134114743281483,
      "learning_rate": 3.079047290300311e-05,
      "loss": 0.4231,
      "step": 2870
    },
    {
      "epoch": 2.6772240335351656,
      "grad_norm": 0.2810448906645236,
      "learning_rate": 3.074732481877805e-05,
      "loss": 0.4281,
      "step": 2875
    },
    {
      "epoch": 2.6818816953889146,
      "grad_norm": 0.3035847869166119,
      "learning_rate": 3.070417673455299e-05,
      "loss": 0.4199,
      "step": 2880
    },
    {
      "epoch": 2.686539357242664,
      "grad_norm": 0.3055652823210175,
      "learning_rate": 3.066102865032793e-05,
      "loss": 0.4376,
      "step": 2885
    },
    {
      "epoch": 2.6911970190964136,
      "grad_norm": 0.2773053799847741,
      "learning_rate": 3.061788056610287e-05,
      "loss": 0.4236,
      "step": 2890
    },
    {
      "epoch": 2.695854680950163,
      "grad_norm": 0.32676715369682086,
      "learning_rate": 3.057473248187781e-05,
      "loss": 0.4307,
      "step": 2895
    },
    {
      "epoch": 2.700512342803912,
      "grad_norm": 0.3002032786782077,
      "learning_rate": 3.053158439765274e-05,
      "loss": 0.4258,
      "step": 2900
    },
    {
      "epoch": 2.7051700046576617,
      "grad_norm": 0.30235761190391164,
      "learning_rate": 3.048843631342769e-05,
      "loss": 0.4199,
      "step": 2905
    },
    {
      "epoch": 2.709827666511411,
      "grad_norm": 0.2791084004190902,
      "learning_rate": 3.0445288229202625e-05,
      "loss": 0.4234,
      "step": 2910
    },
    {
      "epoch": 2.7144853283651607,
      "grad_norm": 0.2949238879795532,
      "learning_rate": 3.0402140144977564e-05,
      "loss": 0.4327,
      "step": 2915
    },
    {
      "epoch": 2.71914299021891,
      "grad_norm": 0.2740462964213622,
      "learning_rate": 3.0358992060752506e-05,
      "loss": 0.4239,
      "step": 2920
    },
    {
      "epoch": 2.7238006520726596,
      "grad_norm": 0.2939351979146713,
      "learning_rate": 3.0315843976527442e-05,
      "loss": 0.437,
      "step": 2925
    },
    {
      "epoch": 2.728458313926409,
      "grad_norm": 0.2933777350867547,
      "learning_rate": 3.027269589230238e-05,
      "loss": 0.4354,
      "step": 2930
    },
    {
      "epoch": 2.7331159757801586,
      "grad_norm": 0.2953146169829949,
      "learning_rate": 3.0229547808077324e-05,
      "loss": 0.4242,
      "step": 2935
    },
    {
      "epoch": 2.7377736376339077,
      "grad_norm": 0.3008040738146734,
      "learning_rate": 3.018639972385226e-05,
      "loss": 0.4245,
      "step": 2940
    },
    {
      "epoch": 2.742431299487657,
      "grad_norm": 0.29187597727648856,
      "learning_rate": 3.0143251639627202e-05,
      "loss": 0.4337,
      "step": 2945
    },
    {
      "epoch": 2.7470889613414067,
      "grad_norm": 0.30642995020454517,
      "learning_rate": 3.010010355540214e-05,
      "loss": 0.4248,
      "step": 2950
    },
    {
      "epoch": 2.751746623195156,
      "grad_norm": 0.2940107819636768,
      "learning_rate": 3.0056955471177077e-05,
      "loss": 0.4216,
      "step": 2955
    },
    {
      "epoch": 2.7564042850489052,
      "grad_norm": 0.28203309593390086,
      "learning_rate": 3.001380738695202e-05,
      "loss": 0.4274,
      "step": 2960
    },
    {
      "epoch": 2.7610619469026547,
      "grad_norm": 0.27842750179385345,
      "learning_rate": 2.9970659302726962e-05,
      "loss": 0.4261,
      "step": 2965
    },
    {
      "epoch": 2.765719608756404,
      "grad_norm": 0.2995175613502464,
      "learning_rate": 2.99275112185019e-05,
      "loss": 0.4318,
      "step": 2970
    },
    {
      "epoch": 2.7703772706101537,
      "grad_norm": 0.31708753330959044,
      "learning_rate": 2.9884363134276837e-05,
      "loss": 0.4217,
      "step": 2975
    },
    {
      "epoch": 2.775034932463903,
      "grad_norm": 0.3045070731784432,
      "learning_rate": 2.984121505005178e-05,
      "loss": 0.4228,
      "step": 2980
    },
    {
      "epoch": 2.7796925943176527,
      "grad_norm": 0.32767969810265124,
      "learning_rate": 2.979806696582672e-05,
      "loss": 0.4329,
      "step": 2985
    },
    {
      "epoch": 2.784350256171402,
      "grad_norm": 0.28750640450571197,
      "learning_rate": 2.9754918881601657e-05,
      "loss": 0.425,
      "step": 2990
    },
    {
      "epoch": 2.7890079180251512,
      "grad_norm": 0.30639240790224964,
      "learning_rate": 2.9711770797376596e-05,
      "loss": 0.4229,
      "step": 2995
    },
    {
      "epoch": 2.7936655798789007,
      "grad_norm": 0.2878394704604041,
      "learning_rate": 2.966862271315154e-05,
      "loss": 0.4344,
      "step": 3000
    },
    {
      "epoch": 2.7983232417326502,
      "grad_norm": 0.28464908757211527,
      "learning_rate": 2.9625474628926475e-05,
      "loss": 0.4203,
      "step": 3005
    },
    {
      "epoch": 2.8029809035863997,
      "grad_norm": 0.30199255276469394,
      "learning_rate": 2.9582326544701417e-05,
      "loss": 0.4357,
      "step": 3010
    },
    {
      "epoch": 2.807638565440149,
      "grad_norm": 0.3134296731427256,
      "learning_rate": 2.9539178460476356e-05,
      "loss": 0.4128,
      "step": 3015
    },
    {
      "epoch": 2.8122962272938983,
      "grad_norm": 0.2856339245413438,
      "learning_rate": 2.9496030376251292e-05,
      "loss": 0.4301,
      "step": 3020
    },
    {
      "epoch": 2.8169538891476478,
      "grad_norm": 0.3018219907450628,
      "learning_rate": 2.9452882292026235e-05,
      "loss": 0.4271,
      "step": 3025
    },
    {
      "epoch": 2.8216115510013973,
      "grad_norm": 0.29582335457199427,
      "learning_rate": 2.9409734207801177e-05,
      "loss": 0.4273,
      "step": 3030
    },
    {
      "epoch": 2.8262692128551468,
      "grad_norm": 0.3022648122719967,
      "learning_rate": 2.9366586123576113e-05,
      "loss": 0.4296,
      "step": 3035
    },
    {
      "epoch": 2.8309268747088963,
      "grad_norm": 0.29671327045823936,
      "learning_rate": 2.9323438039351052e-05,
      "loss": 0.4388,
      "step": 3040
    },
    {
      "epoch": 2.8355845365626458,
      "grad_norm": 0.2723914813076014,
      "learning_rate": 2.9280289955125994e-05,
      "loss": 0.4254,
      "step": 3045
    },
    {
      "epoch": 2.840242198416395,
      "grad_norm": 0.2853312116749402,
      "learning_rate": 2.9237141870900937e-05,
      "loss": 0.4305,
      "step": 3050
    },
    {
      "epoch": 2.8448998602701443,
      "grad_norm": 0.28261463916089896,
      "learning_rate": 2.9193993786675873e-05,
      "loss": 0.4318,
      "step": 3055
    },
    {
      "epoch": 2.849557522123894,
      "grad_norm": 0.29312576921280675,
      "learning_rate": 2.9150845702450812e-05,
      "loss": 0.4333,
      "step": 3060
    },
    {
      "epoch": 2.8542151839776433,
      "grad_norm": 0.29976972583884404,
      "learning_rate": 2.9107697618225754e-05,
      "loss": 0.4276,
      "step": 3065
    },
    {
      "epoch": 2.858872845831393,
      "grad_norm": 0.2848889694476678,
      "learning_rate": 2.906454953400069e-05,
      "loss": 0.4177,
      "step": 3070
    },
    {
      "epoch": 2.863530507685142,
      "grad_norm": 0.2881367781495036,
      "learning_rate": 2.9021401449775632e-05,
      "loss": 0.4289,
      "step": 3075
    },
    {
      "epoch": 2.8681881695388913,
      "grad_norm": 0.29160032793206797,
      "learning_rate": 2.897825336555057e-05,
      "loss": 0.4171,
      "step": 3080
    },
    {
      "epoch": 2.872845831392641,
      "grad_norm": 0.2835921667970533,
      "learning_rate": 2.8935105281325507e-05,
      "loss": 0.4286,
      "step": 3085
    },
    {
      "epoch": 2.8775034932463903,
      "grad_norm": 0.2889502074172928,
      "learning_rate": 2.889195719710045e-05,
      "loss": 0.4336,
      "step": 3090
    },
    {
      "epoch": 2.88216115510014,
      "grad_norm": 0.3176863911744914,
      "learning_rate": 2.8848809112875392e-05,
      "loss": 0.4198,
      "step": 3095
    },
    {
      "epoch": 2.8868188169538893,
      "grad_norm": 0.29564088126235827,
      "learning_rate": 2.8805661028650328e-05,
      "loss": 0.4274,
      "step": 3100
    },
    {
      "epoch": 2.891476478807639,
      "grad_norm": 0.2844171885226471,
      "learning_rate": 2.8762512944425267e-05,
      "loss": 0.4332,
      "step": 3105
    },
    {
      "epoch": 2.896134140661388,
      "grad_norm": 0.27726847461608944,
      "learning_rate": 2.871936486020021e-05,
      "loss": 0.421,
      "step": 3110
    },
    {
      "epoch": 2.9007918025151374,
      "grad_norm": 0.3027840136644332,
      "learning_rate": 2.8676216775975145e-05,
      "loss": 0.4165,
      "step": 3115
    },
    {
      "epoch": 2.905449464368887,
      "grad_norm": 0.2939210447842672,
      "learning_rate": 2.8633068691750088e-05,
      "loss": 0.4254,
      "step": 3120
    },
    {
      "epoch": 2.9101071262226363,
      "grad_norm": 0.28154390215548525,
      "learning_rate": 2.8589920607525027e-05,
      "loss": 0.4312,
      "step": 3125
    },
    {
      "epoch": 2.9147647880763854,
      "grad_norm": 0.27685515316136466,
      "learning_rate": 2.854677252329997e-05,
      "loss": 0.414,
      "step": 3130
    },
    {
      "epoch": 2.919422449930135,
      "grad_norm": 0.2837843809298599,
      "learning_rate": 2.8503624439074905e-05,
      "loss": 0.4194,
      "step": 3135
    },
    {
      "epoch": 2.9240801117838844,
      "grad_norm": 0.26401227535785055,
      "learning_rate": 2.8460476354849848e-05,
      "loss": 0.4327,
      "step": 3140
    },
    {
      "epoch": 2.928737773637634,
      "grad_norm": 0.2800035376843444,
      "learning_rate": 2.8417328270624787e-05,
      "loss": 0.4353,
      "step": 3145
    },
    {
      "epoch": 2.9333954354913834,
      "grad_norm": 0.2962748240980881,
      "learning_rate": 2.8374180186399723e-05,
      "loss": 0.432,
      "step": 3150
    },
    {
      "epoch": 2.938053097345133,
      "grad_norm": 0.3512453488726108,
      "learning_rate": 2.8331032102174665e-05,
      "loss": 0.4171,
      "step": 3155
    },
    {
      "epoch": 2.9427107591988824,
      "grad_norm": 0.2934425561421481,
      "learning_rate": 2.8287884017949608e-05,
      "loss": 0.4289,
      "step": 3160
    },
    {
      "epoch": 2.9473684210526314,
      "grad_norm": 0.30631787007946426,
      "learning_rate": 2.8244735933724543e-05,
      "loss": 0.4296,
      "step": 3165
    },
    {
      "epoch": 2.952026082906381,
      "grad_norm": 0.2833840042401889,
      "learning_rate": 2.8201587849499482e-05,
      "loss": 0.4259,
      "step": 3170
    },
    {
      "epoch": 2.9566837447601304,
      "grad_norm": 0.2755546094385225,
      "learning_rate": 2.8158439765274425e-05,
      "loss": 0.4311,
      "step": 3175
    },
    {
      "epoch": 2.96134140661388,
      "grad_norm": 0.30135070303810446,
      "learning_rate": 2.811529168104936e-05,
      "loss": 0.4218,
      "step": 3180
    },
    {
      "epoch": 2.9659990684676294,
      "grad_norm": 0.2943466823211789,
      "learning_rate": 2.8072143596824303e-05,
      "loss": 0.4272,
      "step": 3185
    },
    {
      "epoch": 2.9706567303213784,
      "grad_norm": 0.2874547322054413,
      "learning_rate": 2.8028995512599242e-05,
      "loss": 0.4164,
      "step": 3190
    },
    {
      "epoch": 2.975314392175128,
      "grad_norm": 0.2752579172878629,
      "learning_rate": 2.7985847428374178e-05,
      "loss": 0.4331,
      "step": 3195
    },
    {
      "epoch": 2.9799720540288774,
      "grad_norm": 0.27715909543271294,
      "learning_rate": 2.794269934414912e-05,
      "loss": 0.4304,
      "step": 3200
    },
    {
      "epoch": 2.984629715882627,
      "grad_norm": 0.30775015870428507,
      "learning_rate": 2.7899551259924063e-05,
      "loss": 0.437,
      "step": 3205
    },
    {
      "epoch": 2.9892873777363764,
      "grad_norm": 0.27013537155609096,
      "learning_rate": 2.7856403175699002e-05,
      "loss": 0.4097,
      "step": 3210
    },
    {
      "epoch": 2.993945039590126,
      "grad_norm": 0.2714676411884666,
      "learning_rate": 2.7813255091473938e-05,
      "loss": 0.4107,
      "step": 3215
    },
    {
      "epoch": 2.9986027014438754,
      "grad_norm": 0.29247382522755994,
      "learning_rate": 2.777010700724888e-05,
      "loss": 0.4305,
      "step": 3220
    },
    {
      "epoch": 3.0027945971122496,
      "grad_norm": 0.3301419187791048,
      "learning_rate": 2.7726958923023823e-05,
      "loss": 0.3946,
      "step": 3225
    },
    {
      "epoch": 3.007452258965999,
      "grad_norm": 0.33961808096776813,
      "learning_rate": 2.768381083879876e-05,
      "loss": 0.3811,
      "step": 3230
    },
    {
      "epoch": 3.0121099208197486,
      "grad_norm": 0.31545470947499427,
      "learning_rate": 2.7640662754573698e-05,
      "loss": 0.3858,
      "step": 3235
    },
    {
      "epoch": 3.016767582673498,
      "grad_norm": 0.2942517122869261,
      "learning_rate": 2.759751467034864e-05,
      "loss": 0.3749,
      "step": 3240
    },
    {
      "epoch": 3.021425244527247,
      "grad_norm": 0.297554738991091,
      "learning_rate": 2.7554366586123576e-05,
      "loss": 0.3847,
      "step": 3245
    },
    {
      "epoch": 3.0260829063809966,
      "grad_norm": 0.28526557812415143,
      "learning_rate": 2.751121850189852e-05,
      "loss": 0.3868,
      "step": 3250
    },
    {
      "epoch": 3.030740568234746,
      "grad_norm": 0.28556193065584956,
      "learning_rate": 2.7468070417673457e-05,
      "loss": 0.3877,
      "step": 3255
    },
    {
      "epoch": 3.0353982300884956,
      "grad_norm": 0.29390575250352075,
      "learning_rate": 2.7424922333448393e-05,
      "loss": 0.3889,
      "step": 3260
    },
    {
      "epoch": 3.040055891942245,
      "grad_norm": 0.2917254576809486,
      "learning_rate": 2.7381774249223336e-05,
      "loss": 0.3911,
      "step": 3265
    },
    {
      "epoch": 3.0447135537959946,
      "grad_norm": 0.29906384047010054,
      "learning_rate": 2.7338626164998278e-05,
      "loss": 0.3794,
      "step": 3270
    },
    {
      "epoch": 3.0493712156497437,
      "grad_norm": 0.28816478274203994,
      "learning_rate": 2.7295478080773214e-05,
      "loss": 0.3851,
      "step": 3275
    },
    {
      "epoch": 3.054028877503493,
      "grad_norm": 0.27460716390596634,
      "learning_rate": 2.7252329996548153e-05,
      "loss": 0.3787,
      "step": 3280
    },
    {
      "epoch": 3.0586865393572427,
      "grad_norm": 0.3159386578853275,
      "learning_rate": 2.7209181912323096e-05,
      "loss": 0.3971,
      "step": 3285
    },
    {
      "epoch": 3.063344201210992,
      "grad_norm": 0.29274954904367356,
      "learning_rate": 2.7166033828098038e-05,
      "loss": 0.3831,
      "step": 3290
    },
    {
      "epoch": 3.0680018630647417,
      "grad_norm": 0.276755627830823,
      "learning_rate": 2.7122885743872974e-05,
      "loss": 0.3876,
      "step": 3295
    },
    {
      "epoch": 3.0726595249184907,
      "grad_norm": 0.3181054021263001,
      "learning_rate": 2.7079737659647913e-05,
      "loss": 0.3807,
      "step": 3300
    },
    {
      "epoch": 3.07731718677224,
      "grad_norm": 0.295245844331509,
      "learning_rate": 2.7036589575422855e-05,
      "loss": 0.3779,
      "step": 3305
    },
    {
      "epoch": 3.0819748486259897,
      "grad_norm": 0.2964640847119177,
      "learning_rate": 2.699344149119779e-05,
      "loss": 0.3831,
      "step": 3310
    },
    {
      "epoch": 3.086632510479739,
      "grad_norm": 0.27999743559094736,
      "learning_rate": 2.6950293406972734e-05,
      "loss": 0.3799,
      "step": 3315
    },
    {
      "epoch": 3.0912901723334887,
      "grad_norm": 0.28078607988524035,
      "learning_rate": 2.6907145322747673e-05,
      "loss": 0.3734,
      "step": 3320
    },
    {
      "epoch": 3.095947834187238,
      "grad_norm": 0.2942685729909238,
      "learning_rate": 2.686399723852261e-05,
      "loss": 0.391,
      "step": 3325
    },
    {
      "epoch": 3.1006054960409872,
      "grad_norm": 0.33596895487127043,
      "learning_rate": 2.682084915429755e-05,
      "loss": 0.3856,
      "step": 3330
    },
    {
      "epoch": 3.1052631578947367,
      "grad_norm": 0.3579029797992384,
      "learning_rate": 2.677770107007249e-05,
      "loss": 0.3808,
      "step": 3335
    },
    {
      "epoch": 3.109920819748486,
      "grad_norm": 0.30270920110616906,
      "learning_rate": 2.673455298584743e-05,
      "loss": 0.3793,
      "step": 3340
    },
    {
      "epoch": 3.1145784816022357,
      "grad_norm": 0.3007273438509959,
      "learning_rate": 2.6691404901622368e-05,
      "loss": 0.3854,
      "step": 3345
    },
    {
      "epoch": 3.119236143455985,
      "grad_norm": 0.3170860088030704,
      "learning_rate": 2.664825681739731e-05,
      "loss": 0.3991,
      "step": 3350
    },
    {
      "epoch": 3.1238938053097347,
      "grad_norm": 0.27132890960491696,
      "learning_rate": 2.6605108733172247e-05,
      "loss": 0.385,
      "step": 3355
    },
    {
      "epoch": 3.1285514671634838,
      "grad_norm": 0.30194142159174714,
      "learning_rate": 2.6561960648947186e-05,
      "loss": 0.3853,
      "step": 3360
    },
    {
      "epoch": 3.1332091290172333,
      "grad_norm": 0.3075127390168238,
      "learning_rate": 2.6518812564722128e-05,
      "loss": 0.3915,
      "step": 3365
    },
    {
      "epoch": 3.1378667908709827,
      "grad_norm": 0.28421974060334493,
      "learning_rate": 2.647566448049707e-05,
      "loss": 0.3867,
      "step": 3370
    },
    {
      "epoch": 3.1425244527247322,
      "grad_norm": 0.2878405214258504,
      "learning_rate": 2.6432516396272006e-05,
      "loss": 0.3863,
      "step": 3375
    },
    {
      "epoch": 3.1471821145784817,
      "grad_norm": 0.2731365603764057,
      "learning_rate": 2.6389368312046945e-05,
      "loss": 0.3849,
      "step": 3380
    },
    {
      "epoch": 3.1518397764322312,
      "grad_norm": 0.2787958008164386,
      "learning_rate": 2.6346220227821888e-05,
      "loss": 0.3835,
      "step": 3385
    },
    {
      "epoch": 3.1564974382859803,
      "grad_norm": 0.29450719491083643,
      "learning_rate": 2.6303072143596824e-05,
      "loss": 0.3883,
      "step": 3390
    },
    {
      "epoch": 3.16115510013973,
      "grad_norm": 0.29332734893304135,
      "learning_rate": 2.6259924059371766e-05,
      "loss": 0.3803,
      "step": 3395
    },
    {
      "epoch": 3.1658127619934793,
      "grad_norm": 0.30993426363245885,
      "learning_rate": 2.6216775975146705e-05,
      "loss": 0.3885,
      "step": 3400
    },
    {
      "epoch": 3.1704704238472288,
      "grad_norm": 0.2786210154602276,
      "learning_rate": 2.617362789092164e-05,
      "loss": 0.3913,
      "step": 3405
    },
    {
      "epoch": 3.1751280857009783,
      "grad_norm": 0.30331739656240075,
      "learning_rate": 2.6130479806696584e-05,
      "loss": 0.3972,
      "step": 3410
    },
    {
      "epoch": 3.1797857475547273,
      "grad_norm": 0.30628780441120507,
      "learning_rate": 2.6087331722471526e-05,
      "loss": 0.3816,
      "step": 3415
    },
    {
      "epoch": 3.184443409408477,
      "grad_norm": 0.3170520375881354,
      "learning_rate": 2.6044183638246462e-05,
      "loss": 0.3958,
      "step": 3420
    },
    {
      "epoch": 3.1891010712622263,
      "grad_norm": 0.3145715445557658,
      "learning_rate": 2.60010355540214e-05,
      "loss": 0.3859,
      "step": 3425
    },
    {
      "epoch": 3.193758733115976,
      "grad_norm": 0.3134045753718819,
      "learning_rate": 2.5957887469796343e-05,
      "loss": 0.3982,
      "step": 3430
    },
    {
      "epoch": 3.1984163949697253,
      "grad_norm": 0.2977634360024046,
      "learning_rate": 2.591473938557128e-05,
      "loss": 0.3948,
      "step": 3435
    },
    {
      "epoch": 3.203074056823475,
      "grad_norm": 0.31714906857567454,
      "learning_rate": 2.587159130134622e-05,
      "loss": 0.3918,
      "step": 3440
    },
    {
      "epoch": 3.207731718677224,
      "grad_norm": 0.29150752902512933,
      "learning_rate": 2.582844321712116e-05,
      "loss": 0.3798,
      "step": 3445
    },
    {
      "epoch": 3.2123893805309733,
      "grad_norm": 0.2753100900275418,
      "learning_rate": 2.5785295132896096e-05,
      "loss": 0.388,
      "step": 3450
    },
    {
      "epoch": 3.217047042384723,
      "grad_norm": 0.31377822019186963,
      "learning_rate": 2.574214704867104e-05,
      "loss": 0.392,
      "step": 3455
    },
    {
      "epoch": 3.2217047042384723,
      "grad_norm": 0.31511011208404827,
      "learning_rate": 2.569899896444598e-05,
      "loss": 0.3903,
      "step": 3460
    },
    {
      "epoch": 3.226362366092222,
      "grad_norm": 0.3057820991355504,
      "learning_rate": 2.565585088022092e-05,
      "loss": 0.3849,
      "step": 3465
    },
    {
      "epoch": 3.2310200279459713,
      "grad_norm": 0.31091535400014025,
      "learning_rate": 2.5612702795995856e-05,
      "loss": 0.3925,
      "step": 3470
    },
    {
      "epoch": 3.2356776897997204,
      "grad_norm": 0.3065298802796263,
      "learning_rate": 2.55695547117708e-05,
      "loss": 0.3851,
      "step": 3475
    },
    {
      "epoch": 3.24033535165347,
      "grad_norm": 0.28343229513527635,
      "learning_rate": 2.552640662754574e-05,
      "loss": 0.3901,
      "step": 3480
    },
    {
      "epoch": 3.2449930135072194,
      "grad_norm": 0.30699661465165284,
      "learning_rate": 2.5483258543320677e-05,
      "loss": 0.3992,
      "step": 3485
    },
    {
      "epoch": 3.249650675360969,
      "grad_norm": 0.304868839219622,
      "learning_rate": 2.5440110459095616e-05,
      "loss": 0.4028,
      "step": 3490
    },
    {
      "epoch": 3.2543083372147183,
      "grad_norm": 0.31137654276760296,
      "learning_rate": 2.539696237487056e-05,
      "loss": 0.4006,
      "step": 3495
    },
    {
      "epoch": 3.258965999068468,
      "grad_norm": 0.3273392185330208,
      "learning_rate": 2.5353814290645494e-05,
      "loss": 0.3992,
      "step": 3500
    },
    {
      "epoch": 3.263623660922217,
      "grad_norm": 0.2951900413998763,
      "learning_rate": 2.5310666206420437e-05,
      "loss": 0.4016,
      "step": 3505
    },
    {
      "epoch": 3.2682813227759664,
      "grad_norm": 0.2995447393207585,
      "learning_rate": 2.5267518122195376e-05,
      "loss": 0.4008,
      "step": 3510
    },
    {
      "epoch": 3.272938984629716,
      "grad_norm": 0.29558091137764736,
      "learning_rate": 2.5224370037970312e-05,
      "loss": 0.3942,
      "step": 3515
    },
    {
      "epoch": 3.2775966464834654,
      "grad_norm": 0.27799484196873125,
      "learning_rate": 2.5181221953745254e-05,
      "loss": 0.3926,
      "step": 3520
    },
    {
      "epoch": 3.282254308337215,
      "grad_norm": 0.3069638306109413,
      "learning_rate": 2.5138073869520197e-05,
      "loss": 0.4075,
      "step": 3525
    },
    {
      "epoch": 3.286911970190964,
      "grad_norm": 0.2826587602936511,
      "learning_rate": 2.5094925785295132e-05,
      "loss": 0.3824,
      "step": 3530
    },
    {
      "epoch": 3.2915696320447134,
      "grad_norm": 0.28472450068445904,
      "learning_rate": 2.505177770107007e-05,
      "loss": 0.3809,
      "step": 3535
    },
    {
      "epoch": 3.296227293898463,
      "grad_norm": 0.2947222681455483,
      "learning_rate": 2.5008629616845014e-05,
      "loss": 0.3856,
      "step": 3540
    },
    {
      "epoch": 3.3008849557522124,
      "grad_norm": 0.29021115288242527,
      "learning_rate": 2.4965481532619953e-05,
      "loss": 0.3943,
      "step": 3545
    },
    {
      "epoch": 3.305542617605962,
      "grad_norm": 0.28214352815906424,
      "learning_rate": 2.4922333448394892e-05,
      "loss": 0.3882,
      "step": 3550
    },
    {
      "epoch": 3.3102002794597114,
      "grad_norm": 0.28437008502074534,
      "learning_rate": 2.487918536416983e-05,
      "loss": 0.3794,
      "step": 3555
    },
    {
      "epoch": 3.3148579413134605,
      "grad_norm": 0.2925903792911444,
      "learning_rate": 2.483603727994477e-05,
      "loss": 0.3892,
      "step": 3560
    },
    {
      "epoch": 3.31951560316721,
      "grad_norm": 0.3025830879877864,
      "learning_rate": 2.4792889195719713e-05,
      "loss": 0.3903,
      "step": 3565
    },
    {
      "epoch": 3.3241732650209594,
      "grad_norm": 0.2861005682428569,
      "learning_rate": 2.4749741111494652e-05,
      "loss": 0.3852,
      "step": 3570
    },
    {
      "epoch": 3.328830926874709,
      "grad_norm": 0.2995761768029744,
      "learning_rate": 2.4706593027269588e-05,
      "loss": 0.3971,
      "step": 3575
    },
    {
      "epoch": 3.3334885887284584,
      "grad_norm": 0.31975916407560884,
      "learning_rate": 2.466344494304453e-05,
      "loss": 0.3919,
      "step": 3580
    },
    {
      "epoch": 3.3381462505822075,
      "grad_norm": 0.3274179860271926,
      "learning_rate": 2.462029685881947e-05,
      "loss": 0.3944,
      "step": 3585
    },
    {
      "epoch": 3.342803912435957,
      "grad_norm": 0.2939272376943077,
      "learning_rate": 2.457714877459441e-05,
      "loss": 0.3864,
      "step": 3590
    },
    {
      "epoch": 3.3474615742897065,
      "grad_norm": 0.288644161598518,
      "learning_rate": 2.4534000690369348e-05,
      "loss": 0.3854,
      "step": 3595
    },
    {
      "epoch": 3.352119236143456,
      "grad_norm": 0.3211161107401283,
      "learning_rate": 2.4490852606144287e-05,
      "loss": 0.3867,
      "step": 3600
    },
    {
      "epoch": 3.3567768979972055,
      "grad_norm": 0.286046929145414,
      "learning_rate": 2.444770452191923e-05,
      "loss": 0.3785,
      "step": 3605
    },
    {
      "epoch": 3.361434559850955,
      "grad_norm": 0.2828533569093587,
      "learning_rate": 2.440455643769417e-05,
      "loss": 0.3898,
      "step": 3610
    },
    {
      "epoch": 3.3660922217047045,
      "grad_norm": 0.30513937699316357,
      "learning_rate": 2.4361408353469108e-05,
      "loss": 0.3864,
      "step": 3615
    },
    {
      "epoch": 3.3707498835584535,
      "grad_norm": 0.3223546760067281,
      "learning_rate": 2.4318260269244047e-05,
      "loss": 0.3866,
      "step": 3620
    },
    {
      "epoch": 3.375407545412203,
      "grad_norm": 0.2877244096395237,
      "learning_rate": 2.4275112185018986e-05,
      "loss": 0.3958,
      "step": 3625
    },
    {
      "epoch": 3.3800652072659525,
      "grad_norm": 0.31128462272497315,
      "learning_rate": 2.4231964100793925e-05,
      "loss": 0.3891,
      "step": 3630
    },
    {
      "epoch": 3.384722869119702,
      "grad_norm": 0.2736957275302278,
      "learning_rate": 2.4188816016568867e-05,
      "loss": 0.3989,
      "step": 3635
    },
    {
      "epoch": 3.3893805309734515,
      "grad_norm": 0.27220417515606105,
      "learning_rate": 2.4145667932343803e-05,
      "loss": 0.3762,
      "step": 3640
    },
    {
      "epoch": 3.3940381928272005,
      "grad_norm": 0.29820469866238686,
      "learning_rate": 2.4102519848118746e-05,
      "loss": 0.3914,
      "step": 3645
    },
    {
      "epoch": 3.39869585468095,
      "grad_norm": 0.27269825853896434,
      "learning_rate": 2.4059371763893685e-05,
      "loss": 0.3817,
      "step": 3650
    },
    {
      "epoch": 3.4033535165346995,
      "grad_norm": 0.27328661988823943,
      "learning_rate": 2.4016223679668624e-05,
      "loss": 0.3986,
      "step": 3655
    },
    {
      "epoch": 3.408011178388449,
      "grad_norm": 0.29588445891962345,
      "learning_rate": 2.3973075595443563e-05,
      "loss": 0.3834,
      "step": 3660
    },
    {
      "epoch": 3.4126688402421985,
      "grad_norm": 0.2795509483622472,
      "learning_rate": 2.3929927511218502e-05,
      "loss": 0.384,
      "step": 3665
    },
    {
      "epoch": 3.417326502095948,
      "grad_norm": 0.2800952050501343,
      "learning_rate": 2.388677942699344e-05,
      "loss": 0.3984,
      "step": 3670
    },
    {
      "epoch": 3.421984163949697,
      "grad_norm": 0.2974731567138673,
      "learning_rate": 2.3843631342768384e-05,
      "loss": 0.3859,
      "step": 3675
    },
    {
      "epoch": 3.4266418258034466,
      "grad_norm": 0.3058211738554222,
      "learning_rate": 2.3800483258543323e-05,
      "loss": 0.3784,
      "step": 3680
    },
    {
      "epoch": 3.431299487657196,
      "grad_norm": 0.2802537695116323,
      "learning_rate": 2.3757335174318262e-05,
      "loss": 0.3935,
      "step": 3685
    },
    {
      "epoch": 3.4359571495109456,
      "grad_norm": 0.27473146047721636,
      "learning_rate": 2.37141870900932e-05,
      "loss": 0.3844,
      "step": 3690
    },
    {
      "epoch": 3.440614811364695,
      "grad_norm": 0.3154402817633416,
      "learning_rate": 2.367103900586814e-05,
      "loss": 0.3877,
      "step": 3695
    },
    {
      "epoch": 3.445272473218444,
      "grad_norm": 0.27913532714039613,
      "learning_rate": 2.3627890921643083e-05,
      "loss": 0.3961,
      "step": 3700
    },
    {
      "epoch": 3.4499301350721936,
      "grad_norm": 0.2747248769064523,
      "learning_rate": 2.358474283741802e-05,
      "loss": 0.3986,
      "step": 3705
    },
    {
      "epoch": 3.454587796925943,
      "grad_norm": 0.2960100849619128,
      "learning_rate": 2.3541594753192957e-05,
      "loss": 0.3932,
      "step": 3710
    },
    {
      "epoch": 3.4592454587796926,
      "grad_norm": 0.2925881190426392,
      "learning_rate": 2.34984466689679e-05,
      "loss": 0.3834,
      "step": 3715
    },
    {
      "epoch": 3.463903120633442,
      "grad_norm": 0.2949742672291092,
      "learning_rate": 2.345529858474284e-05,
      "loss": 0.3869,
      "step": 3720
    },
    {
      "epoch": 3.4685607824871916,
      "grad_norm": 0.3001625124422708,
      "learning_rate": 2.3412150500517778e-05,
      "loss": 0.3965,
      "step": 3725
    },
    {
      "epoch": 3.473218444340941,
      "grad_norm": 0.2833687242521679,
      "learning_rate": 2.3369002416292717e-05,
      "loss": 0.3916,
      "step": 3730
    },
    {
      "epoch": 3.47787610619469,
      "grad_norm": 0.2874383117959432,
      "learning_rate": 2.3325854332067656e-05,
      "loss": 0.3951,
      "step": 3735
    },
    {
      "epoch": 3.4825337680484396,
      "grad_norm": 0.27403854885302326,
      "learning_rate": 2.32827062478426e-05,
      "loss": 0.386,
      "step": 3740
    },
    {
      "epoch": 3.487191429902189,
      "grad_norm": 0.27719204269177605,
      "learning_rate": 2.3239558163617538e-05,
      "loss": 0.3905,
      "step": 3745
    },
    {
      "epoch": 3.4918490917559386,
      "grad_norm": 0.287653731463181,
      "learning_rate": 2.3196410079392474e-05,
      "loss": 0.3953,
      "step": 3750
    },
    {
      "epoch": 3.496506753609688,
      "grad_norm": 0.2948223996371498,
      "learning_rate": 2.3153261995167416e-05,
      "loss": 0.3909,
      "step": 3755
    },
    {
      "epoch": 3.501164415463437,
      "grad_norm": 0.30219608506077617,
      "learning_rate": 2.3110113910942355e-05,
      "loss": 0.389,
      "step": 3760
    },
    {
      "epoch": 3.5058220773171866,
      "grad_norm": 0.29813744190115027,
      "learning_rate": 2.3066965826717294e-05,
      "loss": 0.382,
      "step": 3765
    },
    {
      "epoch": 3.510479739170936,
      "grad_norm": 0.30910919239802104,
      "learning_rate": 2.3023817742492234e-05,
      "loss": 0.3949,
      "step": 3770
    },
    {
      "epoch": 3.5151374010246856,
      "grad_norm": 0.2831029987177164,
      "learning_rate": 2.2980669658267173e-05,
      "loss": 0.3843,
      "step": 3775
    },
    {
      "epoch": 3.519795062878435,
      "grad_norm": 0.26841211018546457,
      "learning_rate": 2.2937521574042115e-05,
      "loss": 0.3737,
      "step": 3780
    },
    {
      "epoch": 3.5244527247321846,
      "grad_norm": 0.28819108961982487,
      "learning_rate": 2.2894373489817054e-05,
      "loss": 0.3938,
      "step": 3785
    },
    {
      "epoch": 3.529110386585934,
      "grad_norm": 0.2719792711316809,
      "learning_rate": 2.285122540559199e-05,
      "loss": 0.3875,
      "step": 3790
    },
    {
      "epoch": 3.533768048439683,
      "grad_norm": 0.3236030795318485,
      "learning_rate": 2.2808077321366933e-05,
      "loss": 0.3935,
      "step": 3795
    },
    {
      "epoch": 3.5384257102934327,
      "grad_norm": 0.2932090836891162,
      "learning_rate": 2.276492923714187e-05,
      "loss": 0.3978,
      "step": 3800
    },
    {
      "epoch": 3.543083372147182,
      "grad_norm": 0.28478227554082075,
      "learning_rate": 2.2721781152916814e-05,
      "loss": 0.4038,
      "step": 3805
    },
    {
      "epoch": 3.5477410340009317,
      "grad_norm": 0.2789052904368932,
      "learning_rate": 2.267863306869175e-05,
      "loss": 0.3966,
      "step": 3810
    },
    {
      "epoch": 3.5523986958546807,
      "grad_norm": 0.26407110073513873,
      "learning_rate": 2.263548498446669e-05,
      "loss": 0.3939,
      "step": 3815
    },
    {
      "epoch": 3.55705635770843,
      "grad_norm": 0.271177492728878,
      "learning_rate": 2.259233690024163e-05,
      "loss": 0.3849,
      "step": 3820
    },
    {
      "epoch": 3.5617140195621797,
      "grad_norm": 0.286705219052838,
      "learning_rate": 2.254918881601657e-05,
      "loss": 0.3993,
      "step": 3825
    },
    {
      "epoch": 3.566371681415929,
      "grad_norm": 0.2726656594542779,
      "learning_rate": 2.250604073179151e-05,
      "loss": 0.3924,
      "step": 3830
    },
    {
      "epoch": 3.5710293432696787,
      "grad_norm": 0.26176377479317314,
      "learning_rate": 2.246289264756645e-05,
      "loss": 0.3962,
      "step": 3835
    },
    {
      "epoch": 3.575687005123428,
      "grad_norm": 0.28425297129088894,
      "learning_rate": 2.2419744563341388e-05,
      "loss": 0.3895,
      "step": 3840
    },
    {
      "epoch": 3.5803446669771777,
      "grad_norm": 0.2811747199255795,
      "learning_rate": 2.237659647911633e-05,
      "loss": 0.3841,
      "step": 3845
    },
    {
      "epoch": 3.5850023288309267,
      "grad_norm": 0.2719328874331291,
      "learning_rate": 2.233344839489127e-05,
      "loss": 0.3957,
      "step": 3850
    },
    {
      "epoch": 3.5896599906846762,
      "grad_norm": 0.3097004183564538,
      "learning_rate": 2.2290300310666205e-05,
      "loss": 0.3999,
      "step": 3855
    },
    {
      "epoch": 3.5943176525384257,
      "grad_norm": 0.28401589081949086,
      "learning_rate": 2.2247152226441148e-05,
      "loss": 0.3922,
      "step": 3860
    },
    {
      "epoch": 3.598975314392175,
      "grad_norm": 0.2993021706948892,
      "learning_rate": 2.2204004142216087e-05,
      "loss": 0.3975,
      "step": 3865
    },
    {
      "epoch": 3.6036329762459243,
      "grad_norm": 0.2807841258267723,
      "learning_rate": 2.2160856057991026e-05,
      "loss": 0.393,
      "step": 3870
    },
    {
      "epoch": 3.6082906380996738,
      "grad_norm": 0.28300699550732944,
      "learning_rate": 2.2117707973765965e-05,
      "loss": 0.3926,
      "step": 3875
    },
    {
      "epoch": 3.6129482999534233,
      "grad_norm": 0.28016397945372123,
      "learning_rate": 2.2074559889540904e-05,
      "loss": 0.3955,
      "step": 3880
    },
    {
      "epoch": 3.6176059618071728,
      "grad_norm": 0.28841428552819237,
      "learning_rate": 2.2031411805315847e-05,
      "loss": 0.3893,
      "step": 3885
    },
    {
      "epoch": 3.6222636236609222,
      "grad_norm": 0.27192230736226725,
      "learning_rate": 2.1988263721090786e-05,
      "loss": 0.3894,
      "step": 3890
    },
    {
      "epoch": 3.6269212855146717,
      "grad_norm": 0.2880712838694273,
      "learning_rate": 2.1945115636865725e-05,
      "loss": 0.3928,
      "step": 3895
    },
    {
      "epoch": 3.6315789473684212,
      "grad_norm": 0.281508215572758,
      "learning_rate": 2.1901967552640664e-05,
      "loss": 0.3793,
      "step": 3900
    },
    {
      "epoch": 3.6362366092221707,
      "grad_norm": 0.2851632424546725,
      "learning_rate": 2.1858819468415603e-05,
      "loss": 0.3993,
      "step": 3905
    },
    {
      "epoch": 3.64089427107592,
      "grad_norm": 0.2946789274909601,
      "learning_rate": 2.1815671384190542e-05,
      "loss": 0.3895,
      "step": 3910
    },
    {
      "epoch": 3.6455519329296693,
      "grad_norm": 0.26237787548104385,
      "learning_rate": 2.1772523299965485e-05,
      "loss": 0.394,
      "step": 3915
    },
    {
      "epoch": 3.6502095947834188,
      "grad_norm": 0.2675560038891023,
      "learning_rate": 2.172937521574042e-05,
      "loss": 0.3837,
      "step": 3920
    },
    {
      "epoch": 3.6548672566371683,
      "grad_norm": 0.255428926648271,
      "learning_rate": 2.1686227131515363e-05,
      "loss": 0.3717,
      "step": 3925
    },
    {
      "epoch": 3.6595249184909173,
      "grad_norm": 0.27264818143520664,
      "learning_rate": 2.1643079047290302e-05,
      "loss": 0.3903,
      "step": 3930
    },
    {
      "epoch": 3.664182580344667,
      "grad_norm": 0.286960913485588,
      "learning_rate": 2.159993096306524e-05,
      "loss": 0.3888,
      "step": 3935
    },
    {
      "epoch": 3.6688402421984163,
      "grad_norm": 0.272844833992394,
      "learning_rate": 2.155678287884018e-05,
      "loss": 0.3908,
      "step": 3940
    },
    {
      "epoch": 3.673497904052166,
      "grad_norm": 0.30084856181012987,
      "learning_rate": 2.151363479461512e-05,
      "loss": 0.388,
      "step": 3945
    },
    {
      "epoch": 3.6781555659059153,
      "grad_norm": 0.27121359774790504,
      "learning_rate": 2.147048671039006e-05,
      "loss": 0.3807,
      "step": 3950
    },
    {
      "epoch": 3.682813227759665,
      "grad_norm": 0.2500929873589587,
      "learning_rate": 2.1427338626165e-05,
      "loss": 0.394,
      "step": 3955
    },
    {
      "epoch": 3.6874708896134143,
      "grad_norm": 0.2992074256066594,
      "learning_rate": 2.1384190541939937e-05,
      "loss": 0.3959,
      "step": 3960
    },
    {
      "epoch": 3.6921285514671633,
      "grad_norm": 0.28091863573123865,
      "learning_rate": 2.134104245771488e-05,
      "loss": 0.3888,
      "step": 3965
    },
    {
      "epoch": 3.696786213320913,
      "grad_norm": 0.27514331715957924,
      "learning_rate": 2.129789437348982e-05,
      "loss": 0.3981,
      "step": 3970
    },
    {
      "epoch": 3.7014438751746623,
      "grad_norm": 0.27132628601912073,
      "learning_rate": 2.1254746289264758e-05,
      "loss": 0.378,
      "step": 3975
    },
    {
      "epoch": 3.706101537028412,
      "grad_norm": 0.28637584475753014,
      "learning_rate": 2.1211598205039697e-05,
      "loss": 0.385,
      "step": 3980
    },
    {
      "epoch": 3.710759198882161,
      "grad_norm": 0.2855785631885528,
      "learning_rate": 2.1168450120814636e-05,
      "loss": 0.4038,
      "step": 3985
    },
    {
      "epoch": 3.7154168607359104,
      "grad_norm": 0.3026831500563836,
      "learning_rate": 2.1125302036589575e-05,
      "loss": 0.3907,
      "step": 3990
    },
    {
      "epoch": 3.72007452258966,
      "grad_norm": 0.2688244760465103,
      "learning_rate": 2.1082153952364517e-05,
      "loss": 0.3852,
      "step": 3995
    },
    {
      "epoch": 3.7247321844434094,
      "grad_norm": 0.28167351614853864,
      "learning_rate": 2.1039005868139457e-05,
      "loss": 0.3894,
      "step": 4000
    },
    {
      "epoch": 3.729389846297159,
      "grad_norm": 0.2659914062282293,
      "learning_rate": 2.0995857783914392e-05,
      "loss": 0.389,
      "step": 4005
    },
    {
      "epoch": 3.7340475081509084,
      "grad_norm": 0.2726300954917552,
      "learning_rate": 2.0952709699689335e-05,
      "loss": 0.3965,
      "step": 4010
    },
    {
      "epoch": 3.738705170004658,
      "grad_norm": 0.2714476638340281,
      "learning_rate": 2.0909561615464274e-05,
      "loss": 0.3816,
      "step": 4015
    },
    {
      "epoch": 3.7433628318584073,
      "grad_norm": 0.27008222933511167,
      "learning_rate": 2.0866413531239216e-05,
      "loss": 0.3972,
      "step": 4020
    },
    {
      "epoch": 3.7480204937121564,
      "grad_norm": 0.27928779011245575,
      "learning_rate": 2.0823265447014152e-05,
      "loss": 0.3937,
      "step": 4025
    },
    {
      "epoch": 3.752678155565906,
      "grad_norm": 0.2592524710247309,
      "learning_rate": 2.078011736278909e-05,
      "loss": 0.3902,
      "step": 4030
    },
    {
      "epoch": 3.7573358174196554,
      "grad_norm": 0.270658519731479,
      "learning_rate": 2.0736969278564034e-05,
      "loss": 0.389,
      "step": 4035
    },
    {
      "epoch": 3.761993479273405,
      "grad_norm": 0.2661485863884677,
      "learning_rate": 2.0693821194338973e-05,
      "loss": 0.3851,
      "step": 4040
    },
    {
      "epoch": 3.766651141127154,
      "grad_norm": 0.2741633716660954,
      "learning_rate": 2.0650673110113912e-05,
      "loss": 0.4024,
      "step": 4045
    },
    {
      "epoch": 3.7713088029809034,
      "grad_norm": 0.25793163860184815,
      "learning_rate": 2.060752502588885e-05,
      "loss": 0.3959,
      "step": 4050
    },
    {
      "epoch": 3.775966464834653,
      "grad_norm": 0.2694341954451844,
      "learning_rate": 2.056437694166379e-05,
      "loss": 0.3993,
      "step": 4055
    },
    {
      "epoch": 3.7806241266884024,
      "grad_norm": 0.2913200301279311,
      "learning_rate": 2.0521228857438733e-05,
      "loss": 0.3887,
      "step": 4060
    },
    {
      "epoch": 3.785281788542152,
      "grad_norm": 0.26597082383682413,
      "learning_rate": 2.0478080773213672e-05,
      "loss": 0.3836,
      "step": 4065
    },
    {
      "epoch": 3.7899394503959014,
      "grad_norm": 0.2851547677543417,
      "learning_rate": 2.0434932688988608e-05,
      "loss": 0.3888,
      "step": 4070
    },
    {
      "epoch": 3.794597112249651,
      "grad_norm": 0.280447551625326,
      "learning_rate": 2.039178460476355e-05,
      "loss": 0.3862,
      "step": 4075
    },
    {
      "epoch": 3.7992547741034,
      "grad_norm": 0.28174315140977474,
      "learning_rate": 2.034863652053849e-05,
      "loss": 0.3916,
      "step": 4080
    },
    {
      "epoch": 3.8039124359571495,
      "grad_norm": 0.278318357641387,
      "learning_rate": 2.0305488436313428e-05,
      "loss": 0.391,
      "step": 4085
    },
    {
      "epoch": 3.808570097810899,
      "grad_norm": 0.26447082950523965,
      "learning_rate": 2.0262340352088367e-05,
      "loss": 0.3893,
      "step": 4090
    },
    {
      "epoch": 3.8132277596646484,
      "grad_norm": 0.2888891311686468,
      "learning_rate": 2.0219192267863306e-05,
      "loss": 0.3845,
      "step": 4095
    },
    {
      "epoch": 3.8178854215183975,
      "grad_norm": 0.28688676740591784,
      "learning_rate": 2.017604418363825e-05,
      "loss": 0.3868,
      "step": 4100
    },
    {
      "epoch": 3.822543083372147,
      "grad_norm": 0.2776846043888768,
      "learning_rate": 2.0132896099413188e-05,
      "loss": 0.3853,
      "step": 4105
    },
    {
      "epoch": 3.8272007452258965,
      "grad_norm": 0.27365031589851296,
      "learning_rate": 2.0089748015188127e-05,
      "loss": 0.3876,
      "step": 4110
    },
    {
      "epoch": 3.831858407079646,
      "grad_norm": 0.29560592972952127,
      "learning_rate": 2.0046599930963066e-05,
      "loss": 0.3956,
      "step": 4115
    },
    {
      "epoch": 3.8365160689333955,
      "grad_norm": 0.27729616984299,
      "learning_rate": 2.0003451846738005e-05,
      "loss": 0.3824,
      "step": 4120
    },
    {
      "epoch": 3.841173730787145,
      "grad_norm": 0.28395815203010316,
      "learning_rate": 1.9960303762512945e-05,
      "loss": 0.3942,
      "step": 4125
    },
    {
      "epoch": 3.8458313926408945,
      "grad_norm": 0.25978895758806464,
      "learning_rate": 1.9917155678287887e-05,
      "loss": 0.3972,
      "step": 4130
    },
    {
      "epoch": 3.850489054494644,
      "grad_norm": 0.2881082189657543,
      "learning_rate": 1.9874007594062823e-05,
      "loss": 0.3956,
      "step": 4135
    },
    {
      "epoch": 3.855146716348393,
      "grad_norm": 0.29734628743977953,
      "learning_rate": 1.9830859509837765e-05,
      "loss": 0.3927,
      "step": 4140
    },
    {
      "epoch": 3.8598043782021425,
      "grad_norm": 0.27460021818975866,
      "learning_rate": 1.9787711425612704e-05,
      "loss": 0.3954,
      "step": 4145
    },
    {
      "epoch": 3.864462040055892,
      "grad_norm": 0.2806353369310551,
      "learning_rate": 1.9744563341387643e-05,
      "loss": 0.3836,
      "step": 4150
    },
    {
      "epoch": 3.8691197019096415,
      "grad_norm": 0.26108125227760187,
      "learning_rate": 1.9701415257162583e-05,
      "loss": 0.3871,
      "step": 4155
    },
    {
      "epoch": 3.8737773637633905,
      "grad_norm": 0.2808839786453705,
      "learning_rate": 1.9658267172937522e-05,
      "loss": 0.4058,
      "step": 4160
    },
    {
      "epoch": 3.87843502561714,
      "grad_norm": 0.29779482740771607,
      "learning_rate": 1.961511908871246e-05,
      "loss": 0.3971,
      "step": 4165
    },
    {
      "epoch": 3.8830926874708895,
      "grad_norm": 0.29323894582592086,
      "learning_rate": 1.9571971004487403e-05,
      "loss": 0.3877,
      "step": 4170
    },
    {
      "epoch": 3.887750349324639,
      "grad_norm": 0.2824962965677371,
      "learning_rate": 1.952882292026234e-05,
      "loss": 0.3968,
      "step": 4175
    },
    {
      "epoch": 3.8924080111783885,
      "grad_norm": 0.28627442962734617,
      "learning_rate": 1.948567483603728e-05,
      "loss": 0.3904,
      "step": 4180
    },
    {
      "epoch": 3.897065673032138,
      "grad_norm": 0.26174250233328855,
      "learning_rate": 1.944252675181222e-05,
      "loss": 0.3926,
      "step": 4185
    },
    {
      "epoch": 3.9017233348858875,
      "grad_norm": 0.2829457980041702,
      "learning_rate": 1.939937866758716e-05,
      "loss": 0.3908,
      "step": 4190
    },
    {
      "epoch": 3.9063809967396366,
      "grad_norm": 0.2869706845561002,
      "learning_rate": 1.93562305833621e-05,
      "loss": 0.3826,
      "step": 4195
    },
    {
      "epoch": 3.911038658593386,
      "grad_norm": 0.27466387746724047,
      "learning_rate": 1.9313082499137038e-05,
      "loss": 0.3864,
      "step": 4200
    },
    {
      "epoch": 3.9156963204471356,
      "grad_norm": 0.25959019833846475,
      "learning_rate": 1.9269934414911977e-05,
      "loss": 0.3918,
      "step": 4205
    },
    {
      "epoch": 3.920353982300885,
      "grad_norm": 0.2613589715021177,
      "learning_rate": 1.922678633068692e-05,
      "loss": 0.3863,
      "step": 4210
    },
    {
      "epoch": 3.925011644154634,
      "grad_norm": 0.27570572527999143,
      "learning_rate": 1.918363824646186e-05,
      "loss": 0.3841,
      "step": 4215
    },
    {
      "epoch": 3.9296693060083836,
      "grad_norm": 0.26909313140855945,
      "learning_rate": 1.9140490162236798e-05,
      "loss": 0.3878,
      "step": 4220
    },
    {
      "epoch": 3.934326967862133,
      "grad_norm": 0.29428858327279167,
      "learning_rate": 1.9097342078011737e-05,
      "loss": 0.3851,
      "step": 4225
    },
    {
      "epoch": 3.9389846297158826,
      "grad_norm": 0.28221273708742245,
      "learning_rate": 1.9054193993786676e-05,
      "loss": 0.3891,
      "step": 4230
    },
    {
      "epoch": 3.943642291569632,
      "grad_norm": 0.3038927822292466,
      "learning_rate": 1.901104590956162e-05,
      "loss": 0.3874,
      "step": 4235
    },
    {
      "epoch": 3.9482999534233816,
      "grad_norm": 0.2770714797205628,
      "learning_rate": 1.8967897825336554e-05,
      "loss": 0.3843,
      "step": 4240
    },
    {
      "epoch": 3.952957615277131,
      "grad_norm": 0.2700271647174003,
      "learning_rate": 1.8924749741111493e-05,
      "loss": 0.3845,
      "step": 4245
    },
    {
      "epoch": 3.9576152771308806,
      "grad_norm": 0.27141592755950594,
      "learning_rate": 1.8881601656886436e-05,
      "loss": 0.3897,
      "step": 4250
    },
    {
      "epoch": 3.9622729389846296,
      "grad_norm": 0.272109950207333,
      "learning_rate": 1.8838453572661375e-05,
      "loss": 0.3902,
      "step": 4255
    },
    {
      "epoch": 3.966930600838379,
      "grad_norm": 0.27033269350211153,
      "learning_rate": 1.8795305488436314e-05,
      "loss": 0.3957,
      "step": 4260
    },
    {
      "epoch": 3.9715882626921286,
      "grad_norm": 0.26861957112880175,
      "learning_rate": 1.8752157404211253e-05,
      "loss": 0.396,
      "step": 4265
    },
    {
      "epoch": 3.976245924545878,
      "grad_norm": 0.26019428717438436,
      "learning_rate": 1.8709009319986192e-05,
      "loss": 0.3902,
      "step": 4270
    },
    {
      "epoch": 3.980903586399627,
      "grad_norm": 0.2746106502523926,
      "learning_rate": 1.8665861235761135e-05,
      "loss": 0.3973,
      "step": 4275
    },
    {
      "epoch": 3.9855612482533767,
      "grad_norm": 0.28586410494929826,
      "learning_rate": 1.8622713151536074e-05,
      "loss": 0.3942,
      "step": 4280
    },
    {
      "epoch": 3.990218910107126,
      "grad_norm": 0.2753913879030619,
      "learning_rate": 1.857956506731101e-05,
      "loss": 0.3843,
      "step": 4285
    },
    {
      "epoch": 3.9948765719608756,
      "grad_norm": 0.265279504466799,
      "learning_rate": 1.8536416983085952e-05,
      "loss": 0.3904,
      "step": 4290
    },
    {
      "epoch": 3.999534233814625,
      "grad_norm": 0.2583494549292401,
      "learning_rate": 1.849326889886089e-05,
      "loss": 0.3924,
      "step": 4295
    },
    {
      "epoch": 4.003726129483,
      "grad_norm": 0.2997677102135513,
      "learning_rate": 1.8450120814635834e-05,
      "loss": 0.3539,
      "step": 4300
    },
    {
      "epoch": 4.008383791336749,
      "grad_norm": 0.2942226963725421,
      "learning_rate": 1.840697273041077e-05,
      "loss": 0.3456,
      "step": 4305
    },
    {
      "epoch": 4.013041453190499,
      "grad_norm": 0.2914321960764821,
      "learning_rate": 1.836382464618571e-05,
      "loss": 0.3506,
      "step": 4310
    },
    {
      "epoch": 4.017699115044247,
      "grad_norm": 0.27404662706687904,
      "learning_rate": 1.832067656196065e-05,
      "loss": 0.3444,
      "step": 4315
    },
    {
      "epoch": 4.022356776897997,
      "grad_norm": 0.286867920364329,
      "learning_rate": 1.827752847773559e-05,
      "loss": 0.3477,
      "step": 4320
    },
    {
      "epoch": 4.027014438751746,
      "grad_norm": 0.3058300510827335,
      "learning_rate": 1.823438039351053e-05,
      "loss": 0.3387,
      "step": 4325
    },
    {
      "epoch": 4.031672100605496,
      "grad_norm": 0.3030099536408052,
      "learning_rate": 1.819123230928547e-05,
      "loss": 0.3492,
      "step": 4330
    },
    {
      "epoch": 4.036329762459245,
      "grad_norm": 0.27744991108422595,
      "learning_rate": 1.8148084225060408e-05,
      "loss": 0.3512,
      "step": 4335
    },
    {
      "epoch": 4.040987424312995,
      "grad_norm": 0.3015458724649459,
      "learning_rate": 1.810493614083535e-05,
      "loss": 0.351,
      "step": 4340
    },
    {
      "epoch": 4.045645086166744,
      "grad_norm": 0.2858313767984123,
      "learning_rate": 1.806178805661029e-05,
      "loss": 0.3469,
      "step": 4345
    },
    {
      "epoch": 4.050302748020494,
      "grad_norm": 0.30124854251489336,
      "learning_rate": 1.8018639972385225e-05,
      "loss": 0.3515,
      "step": 4350
    },
    {
      "epoch": 4.054960409874243,
      "grad_norm": 0.2913346958612146,
      "learning_rate": 1.7975491888160167e-05,
      "loss": 0.3575,
      "step": 4355
    },
    {
      "epoch": 4.059618071727993,
      "grad_norm": 0.30150449477000985,
      "learning_rate": 1.7932343803935107e-05,
      "loss": 0.3589,
      "step": 4360
    },
    {
      "epoch": 4.064275733581742,
      "grad_norm": 0.2796194899277704,
      "learning_rate": 1.7889195719710046e-05,
      "loss": 0.3536,
      "step": 4365
    },
    {
      "epoch": 4.068933395435492,
      "grad_norm": 0.2781378377050305,
      "learning_rate": 1.7846047635484985e-05,
      "loss": 0.354,
      "step": 4370
    },
    {
      "epoch": 4.07359105728924,
      "grad_norm": 0.29698823514946365,
      "learning_rate": 1.7802899551259924e-05,
      "loss": 0.3495,
      "step": 4375
    },
    {
      "epoch": 4.07824871914299,
      "grad_norm": 0.29725164717382124,
      "learning_rate": 1.7759751467034866e-05,
      "loss": 0.3482,
      "step": 4380
    },
    {
      "epoch": 4.082906380996739,
      "grad_norm": 0.2769832676120323,
      "learning_rate": 1.7716603382809806e-05,
      "loss": 0.3477,
      "step": 4385
    },
    {
      "epoch": 4.087564042850489,
      "grad_norm": 0.2793955587340995,
      "learning_rate": 1.767345529858474e-05,
      "loss": 0.356,
      "step": 4390
    },
    {
      "epoch": 4.092221704704238,
      "grad_norm": 0.2822851420639792,
      "learning_rate": 1.7630307214359684e-05,
      "loss": 0.3516,
      "step": 4395
    },
    {
      "epoch": 4.096879366557988,
      "grad_norm": 0.29615452754055965,
      "learning_rate": 1.7587159130134623e-05,
      "loss": 0.3516,
      "step": 4400
    },
    {
      "epoch": 4.101537028411737,
      "grad_norm": 0.296053935166179,
      "learning_rate": 1.7544011045909562e-05,
      "loss": 0.3517,
      "step": 4405
    },
    {
      "epoch": 4.106194690265487,
      "grad_norm": 0.29587563151996993,
      "learning_rate": 1.75008629616845e-05,
      "loss": 0.3542,
      "step": 4410
    },
    {
      "epoch": 4.110852352119236,
      "grad_norm": 0.2747018150032309,
      "learning_rate": 1.745771487745944e-05,
      "loss": 0.3496,
      "step": 4415
    },
    {
      "epoch": 4.115510013972986,
      "grad_norm": 0.2769672928327675,
      "learning_rate": 1.7414566793234383e-05,
      "loss": 0.3597,
      "step": 4420
    },
    {
      "epoch": 4.120167675826735,
      "grad_norm": 0.28282668987346365,
      "learning_rate": 1.7371418709009322e-05,
      "loss": 0.3544,
      "step": 4425
    },
    {
      "epoch": 4.124825337680484,
      "grad_norm": 0.297794383825851,
      "learning_rate": 1.732827062478426e-05,
      "loss": 0.3587,
      "step": 4430
    },
    {
      "epoch": 4.1294829995342335,
      "grad_norm": 0.2886305490012947,
      "learning_rate": 1.72851225405592e-05,
      "loss": 0.3438,
      "step": 4435
    },
    {
      "epoch": 4.134140661387983,
      "grad_norm": 0.28807267158714406,
      "learning_rate": 1.724197445633414e-05,
      "loss": 0.3529,
      "step": 4440
    },
    {
      "epoch": 4.1387983232417325,
      "grad_norm": 0.2802696493292379,
      "learning_rate": 1.7198826372109078e-05,
      "loss": 0.3541,
      "step": 4445
    },
    {
      "epoch": 4.143455985095482,
      "grad_norm": 0.29132723067191274,
      "learning_rate": 1.715567828788402e-05,
      "loss": 0.3549,
      "step": 4450
    },
    {
      "epoch": 4.1481136469492315,
      "grad_norm": 0.2822016763590866,
      "learning_rate": 1.7112530203658957e-05,
      "loss": 0.3529,
      "step": 4455
    },
    {
      "epoch": 4.152771308802981,
      "grad_norm": 0.30047881710009045,
      "learning_rate": 1.70693821194339e-05,
      "loss": 0.3521,
      "step": 4460
    },
    {
      "epoch": 4.1574289706567304,
      "grad_norm": 0.29178159884709615,
      "learning_rate": 1.7026234035208838e-05,
      "loss": 0.3616,
      "step": 4465
    },
    {
      "epoch": 4.16208663251048,
      "grad_norm": 0.2822256219383287,
      "learning_rate": 1.6983085950983777e-05,
      "loss": 0.362,
      "step": 4470
    },
    {
      "epoch": 4.166744294364229,
      "grad_norm": 0.28196178956034174,
      "learning_rate": 1.6939937866758716e-05,
      "loss": 0.3519,
      "step": 4475
    },
    {
      "epoch": 4.171401956217979,
      "grad_norm": 0.287880213044001,
      "learning_rate": 1.6896789782533655e-05,
      "loss": 0.352,
      "step": 4480
    },
    {
      "epoch": 4.1760596180717275,
      "grad_norm": 0.285638041017663,
      "learning_rate": 1.6853641698308595e-05,
      "loss": 0.3411,
      "step": 4485
    },
    {
      "epoch": 4.180717279925477,
      "grad_norm": 0.26936267270032804,
      "learning_rate": 1.6810493614083537e-05,
      "loss": 0.3476,
      "step": 4490
    },
    {
      "epoch": 4.1853749417792265,
      "grad_norm": 0.2723002448186648,
      "learning_rate": 1.6767345529858476e-05,
      "loss": 0.3565,
      "step": 4495
    },
    {
      "epoch": 4.190032603632976,
      "grad_norm": 0.2894090849146836,
      "learning_rate": 1.6724197445633415e-05,
      "loss": 0.3602,
      "step": 4500
    },
    {
      "epoch": 4.1946902654867255,
      "grad_norm": 0.2897158157701427,
      "learning_rate": 1.6681049361408354e-05,
      "loss": 0.3478,
      "step": 4505
    },
    {
      "epoch": 4.199347927340475,
      "grad_norm": 0.2700956168808804,
      "learning_rate": 1.6637901277183294e-05,
      "loss": 0.3565,
      "step": 4510
    },
    {
      "epoch": 4.2040055891942245,
      "grad_norm": 0.28941071896655934,
      "learning_rate": 1.6594753192958236e-05,
      "loss": 0.3532,
      "step": 4515
    },
    {
      "epoch": 4.208663251047974,
      "grad_norm": 0.29078577157840524,
      "learning_rate": 1.6551605108733172e-05,
      "loss": 0.3534,
      "step": 4520
    },
    {
      "epoch": 4.2133209129017235,
      "grad_norm": 0.2740062769079577,
      "learning_rate": 1.650845702450811e-05,
      "loss": 0.3505,
      "step": 4525
    },
    {
      "epoch": 4.217978574755473,
      "grad_norm": 0.311944106972006,
      "learning_rate": 1.6465308940283053e-05,
      "loss": 0.352,
      "step": 4530
    },
    {
      "epoch": 4.2226362366092225,
      "grad_norm": 0.2743335578623788,
      "learning_rate": 1.6422160856057992e-05,
      "loss": 0.3519,
      "step": 4535
    },
    {
      "epoch": 4.227293898462972,
      "grad_norm": 0.298472760025109,
      "learning_rate": 1.637901277183293e-05,
      "loss": 0.3562,
      "step": 4540
    },
    {
      "epoch": 4.231951560316721,
      "grad_norm": 0.28035279359709786,
      "learning_rate": 1.633586468760787e-05,
      "loss": 0.3562,
      "step": 4545
    },
    {
      "epoch": 4.23660922217047,
      "grad_norm": 0.3041924164051859,
      "learning_rate": 1.629271660338281e-05,
      "loss": 0.3529,
      "step": 4550
    },
    {
      "epoch": 4.24126688402422,
      "grad_norm": 0.29639487524246005,
      "learning_rate": 1.6249568519157752e-05,
      "loss": 0.3581,
      "step": 4555
    },
    {
      "epoch": 4.245924545877969,
      "grad_norm": 0.2866238322275492,
      "learning_rate": 1.620642043493269e-05,
      "loss": 0.3618,
      "step": 4560
    },
    {
      "epoch": 4.250582207731719,
      "grad_norm": 0.29346873770121057,
      "learning_rate": 1.6163272350707627e-05,
      "loss": 0.3537,
      "step": 4565
    },
    {
      "epoch": 4.255239869585468,
      "grad_norm": 0.293380520146751,
      "learning_rate": 1.612012426648257e-05,
      "loss": 0.3623,
      "step": 4570
    },
    {
      "epoch": 4.259897531439218,
      "grad_norm": 0.304136689772452,
      "learning_rate": 1.607697618225751e-05,
      "loss": 0.3609,
      "step": 4575
    },
    {
      "epoch": 4.264555193292967,
      "grad_norm": 0.2811839783920599,
      "learning_rate": 1.6033828098032448e-05,
      "loss": 0.3586,
      "step": 4580
    },
    {
      "epoch": 4.269212855146717,
      "grad_norm": 0.27014921558137983,
      "learning_rate": 1.5990680013807387e-05,
      "loss": 0.3559,
      "step": 4585
    },
    {
      "epoch": 4.273870517000466,
      "grad_norm": 0.2928255014797238,
      "learning_rate": 1.5947531929582326e-05,
      "loss": 0.3532,
      "step": 4590
    },
    {
      "epoch": 4.2785281788542155,
      "grad_norm": 0.3059914833698476,
      "learning_rate": 1.590438384535727e-05,
      "loss": 0.3576,
      "step": 4595
    },
    {
      "epoch": 4.283185840707965,
      "grad_norm": 0.28471402080847047,
      "learning_rate": 1.5861235761132208e-05,
      "loss": 0.3588,
      "step": 4600
    },
    {
      "epoch": 4.287843502561714,
      "grad_norm": 0.2819640945193662,
      "learning_rate": 1.5818087676907143e-05,
      "loss": 0.3631,
      "step": 4605
    },
    {
      "epoch": 4.292501164415463,
      "grad_norm": 0.2968211133448053,
      "learning_rate": 1.5774939592682086e-05,
      "loss": 0.354,
      "step": 4610
    },
    {
      "epoch": 4.297158826269213,
      "grad_norm": 0.2869046489760582,
      "learning_rate": 1.5731791508457025e-05,
      "loss": 0.3543,
      "step": 4615
    },
    {
      "epoch": 4.301816488122962,
      "grad_norm": 0.28217525152545014,
      "learning_rate": 1.5688643424231964e-05,
      "loss": 0.3514,
      "step": 4620
    },
    {
      "epoch": 4.306474149976712,
      "grad_norm": 0.2704450220092547,
      "learning_rate": 1.5645495340006903e-05,
      "loss": 0.3553,
      "step": 4625
    },
    {
      "epoch": 4.311131811830461,
      "grad_norm": 0.28154821235145583,
      "learning_rate": 1.5602347255781842e-05,
      "loss": 0.3587,
      "step": 4630
    },
    {
      "epoch": 4.315789473684211,
      "grad_norm": 0.2903477248978338,
      "learning_rate": 1.5559199171556785e-05,
      "loss": 0.354,
      "step": 4635
    },
    {
      "epoch": 4.32044713553796,
      "grad_norm": 0.2783575731868453,
      "learning_rate": 1.5516051087331724e-05,
      "loss": 0.3536,
      "step": 4640
    },
    {
      "epoch": 4.32510479739171,
      "grad_norm": 0.2849040866102702,
      "learning_rate": 1.5472903003106663e-05,
      "loss": 0.3517,
      "step": 4645
    },
    {
      "epoch": 4.329762459245459,
      "grad_norm": 0.27188713337921927,
      "learning_rate": 1.5429754918881602e-05,
      "loss": 0.3536,
      "step": 4650
    },
    {
      "epoch": 4.334420121099209,
      "grad_norm": 0.2690744172312399,
      "learning_rate": 1.538660683465654e-05,
      "loss": 0.3528,
      "step": 4655
    },
    {
      "epoch": 4.339077782952957,
      "grad_norm": 0.2790116559082339,
      "learning_rate": 1.534345875043148e-05,
      "loss": 0.3569,
      "step": 4660
    },
    {
      "epoch": 4.343735444806707,
      "grad_norm": 0.2962273316542597,
      "learning_rate": 1.5300310666206423e-05,
      "loss": 0.3624,
      "step": 4665
    },
    {
      "epoch": 4.348393106660456,
      "grad_norm": 0.28560400124903856,
      "learning_rate": 1.525716258198136e-05,
      "loss": 0.3584,
      "step": 4670
    },
    {
      "epoch": 4.353050768514206,
      "grad_norm": 0.30410487162772376,
      "learning_rate": 1.5214014497756301e-05,
      "loss": 0.3559,
      "step": 4675
    },
    {
      "epoch": 4.357708430367955,
      "grad_norm": 0.28817700314791256,
      "learning_rate": 1.517086641353124e-05,
      "loss": 0.3615,
      "step": 4680
    },
    {
      "epoch": 4.362366092221705,
      "grad_norm": 0.2888746380515395,
      "learning_rate": 1.5127718329306178e-05,
      "loss": 0.3517,
      "step": 4685
    },
    {
      "epoch": 4.367023754075454,
      "grad_norm": 0.28080699044558916,
      "learning_rate": 1.508457024508112e-05,
      "loss": 0.356,
      "step": 4690
    },
    {
      "epoch": 4.371681415929204,
      "grad_norm": 0.27990396131183143,
      "learning_rate": 1.5041422160856058e-05,
      "loss": 0.3523,
      "step": 4695
    },
    {
      "epoch": 4.376339077782953,
      "grad_norm": 0.2765344733075826,
      "learning_rate": 1.4998274076630997e-05,
      "loss": 0.3631,
      "step": 4700
    },
    {
      "epoch": 4.380996739636703,
      "grad_norm": 0.27040304823826045,
      "learning_rate": 1.4955125992405938e-05,
      "loss": 0.3537,
      "step": 4705
    },
    {
      "epoch": 4.385654401490452,
      "grad_norm": 0.2884617079614611,
      "learning_rate": 1.4911977908180877e-05,
      "loss": 0.3636,
      "step": 4710
    },
    {
      "epoch": 4.390312063344201,
      "grad_norm": 0.2906767194043775,
      "learning_rate": 1.4868829823955818e-05,
      "loss": 0.3475,
      "step": 4715
    },
    {
      "epoch": 4.39496972519795,
      "grad_norm": 0.29522972295155203,
      "learning_rate": 1.4825681739730757e-05,
      "loss": 0.3513,
      "step": 4720
    },
    {
      "epoch": 4.3996273870517,
      "grad_norm": 0.29487683719029817,
      "learning_rate": 1.4782533655505696e-05,
      "loss": 0.3466,
      "step": 4725
    },
    {
      "epoch": 4.404285048905449,
      "grad_norm": 0.28608160512368613,
      "learning_rate": 1.4739385571280637e-05,
      "loss": 0.3621,
      "step": 4730
    },
    {
      "epoch": 4.408942710759199,
      "grad_norm": 0.2721317689888656,
      "learning_rate": 1.4696237487055576e-05,
      "loss": 0.3612,
      "step": 4735
    },
    {
      "epoch": 4.413600372612948,
      "grad_norm": 0.2668933390933761,
      "learning_rate": 1.4653089402830513e-05,
      "loss": 0.3442,
      "step": 4740
    },
    {
      "epoch": 4.418258034466698,
      "grad_norm": 0.28779509201299375,
      "learning_rate": 1.4609941318605456e-05,
      "loss": 0.3564,
      "step": 4745
    },
    {
      "epoch": 4.422915696320447,
      "grad_norm": 0.2906197219451771,
      "learning_rate": 1.4566793234380393e-05,
      "loss": 0.3505,
      "step": 4750
    },
    {
      "epoch": 4.427573358174197,
      "grad_norm": 0.2982303757089387,
      "learning_rate": 1.4523645150155336e-05,
      "loss": 0.3652,
      "step": 4755
    },
    {
      "epoch": 4.432231020027946,
      "grad_norm": 0.28571701965825896,
      "learning_rate": 1.4480497065930273e-05,
      "loss": 0.3553,
      "step": 4760
    },
    {
      "epoch": 4.436888681881696,
      "grad_norm": 0.2635318288947632,
      "learning_rate": 1.4437348981705212e-05,
      "loss": 0.3551,
      "step": 4765
    },
    {
      "epoch": 4.441546343735445,
      "grad_norm": 0.2773827887105278,
      "learning_rate": 1.4394200897480153e-05,
      "loss": 0.3469,
      "step": 4770
    },
    {
      "epoch": 4.446204005589194,
      "grad_norm": 0.27842606252002,
      "learning_rate": 1.4351052813255092e-05,
      "loss": 0.3548,
      "step": 4775
    },
    {
      "epoch": 4.450861667442943,
      "grad_norm": 0.2844065046621681,
      "learning_rate": 1.4307904729030031e-05,
      "loss": 0.3573,
      "step": 4780
    },
    {
      "epoch": 4.455519329296693,
      "grad_norm": 0.27066300646218566,
      "learning_rate": 1.4264756644804972e-05,
      "loss": 0.3623,
      "step": 4785
    },
    {
      "epoch": 4.460176991150442,
      "grad_norm": 0.27788231561769355,
      "learning_rate": 1.4221608560579911e-05,
      "loss": 0.3531,
      "step": 4790
    },
    {
      "epoch": 4.464834653004192,
      "grad_norm": 0.28535509486262484,
      "learning_rate": 1.4178460476354852e-05,
      "loss": 0.3561,
      "step": 4795
    },
    {
      "epoch": 4.469492314857941,
      "grad_norm": 0.28264741116920256,
      "learning_rate": 1.4135312392129791e-05,
      "loss": 0.3602,
      "step": 4800
    },
    {
      "epoch": 4.474149976711691,
      "grad_norm": 0.2789995718199173,
      "learning_rate": 1.4092164307904728e-05,
      "loss": 0.3583,
      "step": 4805
    },
    {
      "epoch": 4.47880763856544,
      "grad_norm": 0.2817727161151365,
      "learning_rate": 1.404901622367967e-05,
      "loss": 0.3538,
      "step": 4810
    },
    {
      "epoch": 4.48346530041919,
      "grad_norm": 0.28639908115166013,
      "learning_rate": 1.4005868139454608e-05,
      "loss": 0.3498,
      "step": 4815
    },
    {
      "epoch": 4.488122962272939,
      "grad_norm": 0.27914398798962947,
      "learning_rate": 1.3962720055229547e-05,
      "loss": 0.3606,
      "step": 4820
    },
    {
      "epoch": 4.492780624126689,
      "grad_norm": 0.2813654348693679,
      "learning_rate": 1.3919571971004488e-05,
      "loss": 0.3577,
      "step": 4825
    },
    {
      "epoch": 4.497438285980438,
      "grad_norm": 0.2752822656120324,
      "learning_rate": 1.3876423886779427e-05,
      "loss": 0.3632,
      "step": 4830
    },
    {
      "epoch": 4.502095947834187,
      "grad_norm": 0.2810386431415785,
      "learning_rate": 1.3833275802554368e-05,
      "loss": 0.3561,
      "step": 4835
    },
    {
      "epoch": 4.506753609687936,
      "grad_norm": 0.26428155355977434,
      "learning_rate": 1.3790127718329307e-05,
      "loss": 0.358,
      "step": 4840
    },
    {
      "epoch": 4.511411271541686,
      "grad_norm": 0.28016809170744666,
      "learning_rate": 1.3746979634104246e-05,
      "loss": 0.3553,
      "step": 4845
    },
    {
      "epoch": 4.516068933395435,
      "grad_norm": 0.2688572853487068,
      "learning_rate": 1.3703831549879187e-05,
      "loss": 0.3486,
      "step": 4850
    },
    {
      "epoch": 4.520726595249185,
      "grad_norm": 0.27799357232599314,
      "learning_rate": 1.3660683465654126e-05,
      "loss": 0.3569,
      "step": 4855
    },
    {
      "epoch": 4.525384257102934,
      "grad_norm": 0.2780736247657236,
      "learning_rate": 1.3617535381429064e-05,
      "loss": 0.3503,
      "step": 4860
    },
    {
      "epoch": 4.530041918956684,
      "grad_norm": 0.2734575874343857,
      "learning_rate": 1.3574387297204006e-05,
      "loss": 0.3561,
      "step": 4865
    },
    {
      "epoch": 4.534699580810433,
      "grad_norm": 0.26893128888114864,
      "learning_rate": 1.3531239212978944e-05,
      "loss": 0.3575,
      "step": 4870
    },
    {
      "epoch": 4.539357242664183,
      "grad_norm": 0.2677397824132135,
      "learning_rate": 1.3488091128753886e-05,
      "loss": 0.3595,
      "step": 4875
    },
    {
      "epoch": 4.544014904517932,
      "grad_norm": 0.27517283545332755,
      "learning_rate": 1.3444943044528824e-05,
      "loss": 0.3588,
      "step": 4880
    },
    {
      "epoch": 4.548672566371682,
      "grad_norm": 0.2677261089032289,
      "learning_rate": 1.3401794960303763e-05,
      "loss": 0.3572,
      "step": 4885
    },
    {
      "epoch": 4.55333022822543,
      "grad_norm": 0.2746298763750638,
      "learning_rate": 1.3358646876078703e-05,
      "loss": 0.3539,
      "step": 4890
    },
    {
      "epoch": 4.55798789007918,
      "grad_norm": 0.27391748336161115,
      "learning_rate": 1.3315498791853643e-05,
      "loss": 0.3516,
      "step": 4895
    },
    {
      "epoch": 4.562645551932929,
      "grad_norm": 0.2755362548785992,
      "learning_rate": 1.327235070762858e-05,
      "loss": 0.3465,
      "step": 4900
    },
    {
      "epoch": 4.567303213786679,
      "grad_norm": 0.27474560126274555,
      "learning_rate": 1.3229202623403522e-05,
      "loss": 0.3576,
      "step": 4905
    },
    {
      "epoch": 4.571960875640428,
      "grad_norm": 0.2697453209892254,
      "learning_rate": 1.318605453917846e-05,
      "loss": 0.3546,
      "step": 4910
    },
    {
      "epoch": 4.576618537494178,
      "grad_norm": 0.26520093770118885,
      "learning_rate": 1.3142906454953402e-05,
      "loss": 0.3511,
      "step": 4915
    },
    {
      "epoch": 4.581276199347927,
      "grad_norm": 0.27096137395019976,
      "learning_rate": 1.309975837072834e-05,
      "loss": 0.3521,
      "step": 4920
    },
    {
      "epoch": 4.585933861201677,
      "grad_norm": 0.27785617922875316,
      "learning_rate": 1.3056610286503279e-05,
      "loss": 0.3587,
      "step": 4925
    },
    {
      "epoch": 4.590591523055426,
      "grad_norm": 0.27480597603601387,
      "learning_rate": 1.301346220227822e-05,
      "loss": 0.3503,
      "step": 4930
    },
    {
      "epoch": 4.595249184909176,
      "grad_norm": 0.262151778448164,
      "learning_rate": 1.2970314118053159e-05,
      "loss": 0.3596,
      "step": 4935
    },
    {
      "epoch": 4.599906846762925,
      "grad_norm": 0.2621226880798443,
      "learning_rate": 1.2927166033828098e-05,
      "loss": 0.3507,
      "step": 4940
    },
    {
      "epoch": 4.604564508616674,
      "grad_norm": 0.26737866860351167,
      "learning_rate": 1.2884017949603039e-05,
      "loss": 0.3657,
      "step": 4945
    },
    {
      "epoch": 4.6092221704704235,
      "grad_norm": 0.27493673776727556,
      "learning_rate": 1.2840869865377978e-05,
      "loss": 0.3514,
      "step": 4950
    },
    {
      "epoch": 4.613879832324173,
      "grad_norm": 0.2640603882765641,
      "learning_rate": 1.2797721781152919e-05,
      "loss": 0.3588,
      "step": 4955
    },
    {
      "epoch": 4.6185374941779225,
      "grad_norm": 0.27230399378932046,
      "learning_rate": 1.2754573696927858e-05,
      "loss": 0.3505,
      "step": 4960
    },
    {
      "epoch": 4.623195156031672,
      "grad_norm": 0.278301999224887,
      "learning_rate": 1.2711425612702795e-05,
      "loss": 0.3626,
      "step": 4965
    },
    {
      "epoch": 4.6278528178854215,
      "grad_norm": 0.27005375722436464,
      "learning_rate": 1.2668277528477738e-05,
      "loss": 0.361,
      "step": 4970
    },
    {
      "epoch": 4.632510479739171,
      "grad_norm": 0.26623328404490904,
      "learning_rate": 1.2625129444252675e-05,
      "loss": 0.3475,
      "step": 4975
    },
    {
      "epoch": 4.6371681415929205,
      "grad_norm": 0.2770027170420596,
      "learning_rate": 1.2581981360027614e-05,
      "loss": 0.3491,
      "step": 4980
    },
    {
      "epoch": 4.64182580344667,
      "grad_norm": 0.28468413390833586,
      "learning_rate": 1.2538833275802555e-05,
      "loss": 0.3638,
      "step": 4985
    },
    {
      "epoch": 4.6464834653004194,
      "grad_norm": 0.26931676006962463,
      "learning_rate": 1.2495685191577494e-05,
      "loss": 0.3495,
      "step": 4990
    },
    {
      "epoch": 4.651141127154169,
      "grad_norm": 0.27253615404131626,
      "learning_rate": 1.2452537107352433e-05,
      "loss": 0.3572,
      "step": 4995
    },
    {
      "epoch": 4.6557987890079175,
      "grad_norm": 0.2761487701387579,
      "learning_rate": 1.2409389023127374e-05,
      "loss": 0.3578,
      "step": 5000
    },
    {
      "epoch": 4.660456450861668,
      "grad_norm": 0.2811691117624798,
      "learning_rate": 1.2366240938902313e-05,
      "loss": 0.3567,
      "step": 5005
    },
    {
      "epoch": 4.6651141127154165,
      "grad_norm": 0.2674447731198243,
      "learning_rate": 1.2323092854677252e-05,
      "loss": 0.3611,
      "step": 5010
    },
    {
      "epoch": 4.669771774569166,
      "grad_norm": 0.2686181413739826,
      "learning_rate": 1.2279944770452193e-05,
      "loss": 0.3554,
      "step": 5015
    },
    {
      "epoch": 4.6744294364229155,
      "grad_norm": 0.2647405091177316,
      "learning_rate": 1.2236796686227132e-05,
      "loss": 0.3561,
      "step": 5020
    },
    {
      "epoch": 4.679087098276665,
      "grad_norm": 0.26899405957256156,
      "learning_rate": 1.2193648602002073e-05,
      "loss": 0.358,
      "step": 5025
    },
    {
      "epoch": 4.6837447601304145,
      "grad_norm": 0.26614080097210785,
      "learning_rate": 1.215050051777701e-05,
      "loss": 0.3536,
      "step": 5030
    },
    {
      "epoch": 4.688402421984164,
      "grad_norm": 0.2716762078659036,
      "learning_rate": 1.2107352433551951e-05,
      "loss": 0.3542,
      "step": 5035
    },
    {
      "epoch": 4.6930600838379135,
      "grad_norm": 0.2638300361857127,
      "learning_rate": 1.206420434932689e-05,
      "loss": 0.3527,
      "step": 5040
    },
    {
      "epoch": 4.697717745691663,
      "grad_norm": 0.2710735641414926,
      "learning_rate": 1.202105626510183e-05,
      "loss": 0.3488,
      "step": 5045
    },
    {
      "epoch": 4.7023754075454125,
      "grad_norm": 0.2858156597460044,
      "learning_rate": 1.1977908180876769e-05,
      "loss": 0.3507,
      "step": 5050
    },
    {
      "epoch": 4.707033069399162,
      "grad_norm": 0.2716147467961824,
      "learning_rate": 1.193476009665171e-05,
      "loss": 0.3514,
      "step": 5055
    },
    {
      "epoch": 4.7116907312529115,
      "grad_norm": 0.27686308709443985,
      "learning_rate": 1.1891612012426649e-05,
      "loss": 0.3528,
      "step": 5060
    },
    {
      "epoch": 4.71634839310666,
      "grad_norm": 0.27565973582537256,
      "learning_rate": 1.1848463928201588e-05,
      "loss": 0.3484,
      "step": 5065
    },
    {
      "epoch": 4.72100605496041,
      "grad_norm": 0.2653207933251761,
      "learning_rate": 1.1805315843976528e-05,
      "loss": 0.3521,
      "step": 5070
    },
    {
      "epoch": 4.725663716814159,
      "grad_norm": 0.2666947304007132,
      "learning_rate": 1.1762167759751468e-05,
      "loss": 0.3586,
      "step": 5075
    },
    {
      "epoch": 4.730321378667909,
      "grad_norm": 0.27626418707490286,
      "learning_rate": 1.1719019675526408e-05,
      "loss": 0.3585,
      "step": 5080
    },
    {
      "epoch": 4.734979040521658,
      "grad_norm": 0.2704026690139338,
      "learning_rate": 1.1675871591301346e-05,
      "loss": 0.3493,
      "step": 5085
    },
    {
      "epoch": 4.739636702375408,
      "grad_norm": 0.2604165995834615,
      "learning_rate": 1.1632723507076287e-05,
      "loss": 0.3548,
      "step": 5090
    },
    {
      "epoch": 4.744294364229157,
      "grad_norm": 0.2674642824694853,
      "learning_rate": 1.1589575422851226e-05,
      "loss": 0.3481,
      "step": 5095
    },
    {
      "epoch": 4.748952026082907,
      "grad_norm": 0.27085639635412356,
      "learning_rate": 1.1546427338626167e-05,
      "loss": 0.3571,
      "step": 5100
    },
    {
      "epoch": 4.753609687936656,
      "grad_norm": 0.3011011701234883,
      "learning_rate": 1.1503279254401104e-05,
      "loss": 0.3584,
      "step": 5105
    },
    {
      "epoch": 4.7582673497904056,
      "grad_norm": 0.28015708907257936,
      "learning_rate": 1.1460131170176045e-05,
      "loss": 0.3567,
      "step": 5110
    },
    {
      "epoch": 4.762925011644155,
      "grad_norm": 0.27827937543449927,
      "learning_rate": 1.1416983085950984e-05,
      "loss": 0.3533,
      "step": 5115
    },
    {
      "epoch": 4.767582673497904,
      "grad_norm": 0.260357789119031,
      "learning_rate": 1.1373835001725925e-05,
      "loss": 0.3482,
      "step": 5120
    },
    {
      "epoch": 4.772240335351653,
      "grad_norm": 0.28601958244416825,
      "learning_rate": 1.1330686917500862e-05,
      "loss": 0.3549,
      "step": 5125
    },
    {
      "epoch": 4.776897997205403,
      "grad_norm": 0.27374981912499063,
      "learning_rate": 1.1287538833275803e-05,
      "loss": 0.3579,
      "step": 5130
    },
    {
      "epoch": 4.781555659059152,
      "grad_norm": 0.28333177180832764,
      "learning_rate": 1.1244390749050742e-05,
      "loss": 0.3677,
      "step": 5135
    },
    {
      "epoch": 4.786213320912902,
      "grad_norm": 0.2682299611439379,
      "learning_rate": 1.1201242664825683e-05,
      "loss": 0.3489,
      "step": 5140
    },
    {
      "epoch": 4.790870982766651,
      "grad_norm": 0.262041844728982,
      "learning_rate": 1.1158094580600622e-05,
      "loss": 0.3552,
      "step": 5145
    },
    {
      "epoch": 4.795528644620401,
      "grad_norm": 0.27162611374112183,
      "learning_rate": 1.1114946496375561e-05,
      "loss": 0.3411,
      "step": 5150
    },
    {
      "epoch": 4.80018630647415,
      "grad_norm": 0.2743560959865273,
      "learning_rate": 1.1071798412150502e-05,
      "loss": 0.3469,
      "step": 5155
    },
    {
      "epoch": 4.8048439683279,
      "grad_norm": 0.2667725899301606,
      "learning_rate": 1.1028650327925441e-05,
      "loss": 0.348,
      "step": 5160
    },
    {
      "epoch": 4.809501630181649,
      "grad_norm": 0.26431300428323234,
      "learning_rate": 1.098550224370038e-05,
      "loss": 0.3592,
      "step": 5165
    },
    {
      "epoch": 4.814159292035399,
      "grad_norm": 0.26977443208815316,
      "learning_rate": 1.094235415947532e-05,
      "loss": 0.3597,
      "step": 5170
    },
    {
      "epoch": 4.818816953889147,
      "grad_norm": 0.2749396280422831,
      "learning_rate": 1.089920607525026e-05,
      "loss": 0.3638,
      "step": 5175
    },
    {
      "epoch": 4.823474615742897,
      "grad_norm": 0.27575480969552224,
      "learning_rate": 1.0856057991025199e-05,
      "loss": 0.3508,
      "step": 5180
    },
    {
      "epoch": 4.828132277596646,
      "grad_norm": 0.2617122009763159,
      "learning_rate": 1.0812909906800138e-05,
      "loss": 0.3505,
      "step": 5185
    },
    {
      "epoch": 4.832789939450396,
      "grad_norm": 0.3010829387811166,
      "learning_rate": 1.0769761822575077e-05,
      "loss": 0.3668,
      "step": 5190
    },
    {
      "epoch": 4.837447601304145,
      "grad_norm": 0.2793154169980902,
      "learning_rate": 1.0726613738350018e-05,
      "loss": 0.3539,
      "step": 5195
    },
    {
      "epoch": 4.842105263157895,
      "grad_norm": 0.2609131857448218,
      "learning_rate": 1.0683465654124957e-05,
      "loss": 0.3484,
      "step": 5200
    },
    {
      "epoch": 4.846762925011644,
      "grad_norm": 0.271426479811975,
      "learning_rate": 1.0640317569899896e-05,
      "loss": 0.3594,
      "step": 5205
    },
    {
      "epoch": 4.851420586865394,
      "grad_norm": 0.26289024588017484,
      "learning_rate": 1.0597169485674835e-05,
      "loss": 0.363,
      "step": 5210
    },
    {
      "epoch": 4.856078248719143,
      "grad_norm": 0.2669006326992202,
      "learning_rate": 1.0554021401449776e-05,
      "loss": 0.356,
      "step": 5215
    },
    {
      "epoch": 4.860735910572893,
      "grad_norm": 0.26243471063174134,
      "learning_rate": 1.0510873317224715e-05,
      "loss": 0.3573,
      "step": 5220
    },
    {
      "epoch": 4.865393572426642,
      "grad_norm": 0.2754779799558577,
      "learning_rate": 1.0467725232999655e-05,
      "loss": 0.349,
      "step": 5225
    },
    {
      "epoch": 4.870051234280391,
      "grad_norm": 0.26534186700888124,
      "learning_rate": 1.0424577148774595e-05,
      "loss": 0.3551,
      "step": 5230
    },
    {
      "epoch": 4.874708896134141,
      "grad_norm": 0.2739235579072256,
      "learning_rate": 1.0381429064549534e-05,
      "loss": 0.3614,
      "step": 5235
    },
    {
      "epoch": 4.87936655798789,
      "grad_norm": 0.28221566104762646,
      "learning_rate": 1.0338280980324475e-05,
      "loss": 0.3535,
      "step": 5240
    },
    {
      "epoch": 4.884024219841639,
      "grad_norm": 0.28133578172638957,
      "learning_rate": 1.0295132896099413e-05,
      "loss": 0.3662,
      "step": 5245
    },
    {
      "epoch": 4.888681881695389,
      "grad_norm": 0.2937283592964916,
      "learning_rate": 1.0251984811874353e-05,
      "loss": 0.3635,
      "step": 5250
    },
    {
      "epoch": 4.893339543549138,
      "grad_norm": 0.26796973906150834,
      "learning_rate": 1.0208836727649293e-05,
      "loss": 0.355,
      "step": 5255
    },
    {
      "epoch": 4.897997205402888,
      "grad_norm": 0.2867795928690151,
      "learning_rate": 1.0165688643424233e-05,
      "loss": 0.3486,
      "step": 5260
    },
    {
      "epoch": 4.902654867256637,
      "grad_norm": 0.2689245367170045,
      "learning_rate": 1.012254055919917e-05,
      "loss": 0.3671,
      "step": 5265
    },
    {
      "epoch": 4.907312529110387,
      "grad_norm": 0.2670285343280654,
      "learning_rate": 1.0079392474974112e-05,
      "loss": 0.3578,
      "step": 5270
    },
    {
      "epoch": 4.911970190964136,
      "grad_norm": 0.2769178069960969,
      "learning_rate": 1.003624439074905e-05,
      "loss": 0.3555,
      "step": 5275
    },
    {
      "epoch": 4.916627852817886,
      "grad_norm": 0.26791408916053544,
      "learning_rate": 9.993096306523992e-06,
      "loss": 0.3578,
      "step": 5280
    },
    {
      "epoch": 4.921285514671635,
      "grad_norm": 0.27718325538785127,
      "learning_rate": 9.94994822229893e-06,
      "loss": 0.353,
      "step": 5285
    },
    {
      "epoch": 4.925943176525385,
      "grad_norm": 0.2827749754502368,
      "learning_rate": 9.90680013807387e-06,
      "loss": 0.3616,
      "step": 5290
    },
    {
      "epoch": 4.930600838379133,
      "grad_norm": 0.27149505642364447,
      "learning_rate": 9.86365205384881e-06,
      "loss": 0.3508,
      "step": 5295
    },
    {
      "epoch": 4.935258500232883,
      "grad_norm": 0.2658067031976932,
      "learning_rate": 9.82050396962375e-06,
      "loss": 0.3551,
      "step": 5300
    },
    {
      "epoch": 4.939916162086632,
      "grad_norm": 0.28137071719792245,
      "learning_rate": 9.777355885398689e-06,
      "loss": 0.3589,
      "step": 5305
    },
    {
      "epoch": 4.944573823940382,
      "grad_norm": 0.27042844960524937,
      "learning_rate": 9.734207801173628e-06,
      "loss": 0.3542,
      "step": 5310
    },
    {
      "epoch": 4.949231485794131,
      "grad_norm": 0.2704079436398085,
      "learning_rate": 9.691059716948569e-06,
      "loss": 0.3465,
      "step": 5315
    },
    {
      "epoch": 4.953889147647881,
      "grad_norm": 0.2753323913555296,
      "learning_rate": 9.647911632723508e-06,
      "loss": 0.3551,
      "step": 5320
    },
    {
      "epoch": 4.95854680950163,
      "grad_norm": 0.2758756694935783,
      "learning_rate": 9.604763548498447e-06,
      "loss": 0.3582,
      "step": 5325
    },
    {
      "epoch": 4.96320447135538,
      "grad_norm": 0.2726776244342691,
      "learning_rate": 9.561615464273386e-06,
      "loss": 0.358,
      "step": 5330
    },
    {
      "epoch": 4.967862133209129,
      "grad_norm": 0.2575722130314255,
      "learning_rate": 9.518467380048327e-06,
      "loss": 0.3451,
      "step": 5335
    },
    {
      "epoch": 4.972519795062879,
      "grad_norm": 0.2693836738963707,
      "learning_rate": 9.475319295823266e-06,
      "loss": 0.3565,
      "step": 5340
    },
    {
      "epoch": 4.977177456916628,
      "grad_norm": 0.27054936652858963,
      "learning_rate": 9.432171211598205e-06,
      "loss": 0.3506,
      "step": 5345
    },
    {
      "epoch": 4.981835118770377,
      "grad_norm": 0.2569816702620192,
      "learning_rate": 9.389023127373144e-06,
      "loss": 0.3441,
      "step": 5350
    },
    {
      "epoch": 4.986492780624126,
      "grad_norm": 0.2692646565195524,
      "learning_rate": 9.345875043148085e-06,
      "loss": 0.3563,
      "step": 5355
    },
    {
      "epoch": 4.991150442477876,
      "grad_norm": 0.2675495858431079,
      "learning_rate": 9.302726958923024e-06,
      "loss": 0.3412,
      "step": 5360
    },
    {
      "epoch": 4.995808104331625,
      "grad_norm": 0.2705058171756177,
      "learning_rate": 9.259578874697963e-06,
      "loss": 0.3575,
      "step": 5365
    },
    {
      "epoch": 5.0,
      "grad_norm": 0.276635222248572,
      "learning_rate": 9.216430790472904e-06,
      "loss": 0.345,
      "step": 5370
    },
    {
      "epoch": 5.0046576618537495,
      "grad_norm": 0.2944769129643865,
      "learning_rate": 9.173282706247843e-06,
      "loss": 0.3296,
      "step": 5375
    },
    {
      "epoch": 5.009315323707499,
      "grad_norm": 0.306409379128906,
      "learning_rate": 9.130134622022784e-06,
      "loss": 0.3287,
      "step": 5380
    },
    {
      "epoch": 5.0139729855612485,
      "grad_norm": 0.28838261022656486,
      "learning_rate": 9.086986537797721e-06,
      "loss": 0.3301,
      "step": 5385
    },
    {
      "epoch": 5.018630647414998,
      "grad_norm": 0.29028475234323564,
      "learning_rate": 9.043838453572662e-06,
      "loss": 0.3348,
      "step": 5390
    },
    {
      "epoch": 5.0232883092687475,
      "grad_norm": 0.27431866306456176,
      "learning_rate": 9.000690369347601e-06,
      "loss": 0.3205,
      "step": 5395
    },
    {
      "epoch": 5.027945971122496,
      "grad_norm": 0.2838418041987447,
      "learning_rate": 8.957542285122542e-06,
      "loss": 0.3181,
      "step": 5400
    },
    {
      "epoch": 5.032603632976246,
      "grad_norm": 0.2796519859169487,
      "learning_rate": 8.91439420089748e-06,
      "loss": 0.3206,
      "step": 5405
    },
    {
      "epoch": 5.037261294829995,
      "grad_norm": 0.26182868722008823,
      "learning_rate": 8.87124611667242e-06,
      "loss": 0.3221,
      "step": 5410
    },
    {
      "epoch": 5.041918956683745,
      "grad_norm": 0.31509027043148635,
      "learning_rate": 8.82809803244736e-06,
      "loss": 0.3218,
      "step": 5415
    },
    {
      "epoch": 5.046576618537494,
      "grad_norm": 0.2751551096141409,
      "learning_rate": 8.7849499482223e-06,
      "loss": 0.3149,
      "step": 5420
    },
    {
      "epoch": 5.051234280391244,
      "grad_norm": 0.27002023429603983,
      "learning_rate": 8.741801863997238e-06,
      "loss": 0.3218,
      "step": 5425
    },
    {
      "epoch": 5.055891942244993,
      "grad_norm": 0.2793452429527823,
      "learning_rate": 8.698653779772179e-06,
      "loss": 0.3285,
      "step": 5430
    },
    {
      "epoch": 5.0605496040987425,
      "grad_norm": 0.27426431890039554,
      "learning_rate": 8.655505695547118e-06,
      "loss": 0.3098,
      "step": 5435
    },
    {
      "epoch": 5.065207265952492,
      "grad_norm": 0.2729960701186788,
      "learning_rate": 8.612357611322058e-06,
      "loss": 0.3143,
      "step": 5440
    },
    {
      "epoch": 5.0698649278062415,
      "grad_norm": 0.2642618381392171,
      "learning_rate": 8.569209527096998e-06,
      "loss": 0.3176,
      "step": 5445
    },
    {
      "epoch": 5.074522589659991,
      "grad_norm": 0.2637012986245897,
      "learning_rate": 8.526061442871937e-06,
      "loss": 0.3313,
      "step": 5450
    },
    {
      "epoch": 5.0791802515137405,
      "grad_norm": 0.29665598894713824,
      "learning_rate": 8.482913358646877e-06,
      "loss": 0.33,
      "step": 5455
    },
    {
      "epoch": 5.083837913367489,
      "grad_norm": 0.25928349855673133,
      "learning_rate": 8.439765274421817e-06,
      "loss": 0.3105,
      "step": 5460
    },
    {
      "epoch": 5.088495575221239,
      "grad_norm": 0.28175905267135914,
      "learning_rate": 8.396617190196756e-06,
      "loss": 0.3244,
      "step": 5465
    },
    {
      "epoch": 5.093153237074988,
      "grad_norm": 0.275064629066818,
      "learning_rate": 8.353469105971695e-06,
      "loss": 0.3283,
      "step": 5470
    },
    {
      "epoch": 5.097810898928738,
      "grad_norm": 0.2875220803312239,
      "learning_rate": 8.310321021746636e-06,
      "loss": 0.328,
      "step": 5475
    },
    {
      "epoch": 5.102468560782487,
      "grad_norm": 0.2843151825021687,
      "learning_rate": 8.267172937521575e-06,
      "loss": 0.3339,
      "step": 5480
    },
    {
      "epoch": 5.107126222636237,
      "grad_norm": 0.2732542071182327,
      "learning_rate": 8.224024853296514e-06,
      "loss": 0.3235,
      "step": 5485
    },
    {
      "epoch": 5.111783884489986,
      "grad_norm": 0.2645927475304701,
      "learning_rate": 8.180876769071453e-06,
      "loss": 0.3195,
      "step": 5490
    },
    {
      "epoch": 5.116441546343736,
      "grad_norm": 0.2909235258371715,
      "learning_rate": 8.137728684846394e-06,
      "loss": 0.3331,
      "step": 5495
    },
    {
      "epoch": 5.121099208197485,
      "grad_norm": 0.2815879037647107,
      "learning_rate": 8.094580600621333e-06,
      "loss": 0.3263,
      "step": 5500
    },
    {
      "epoch": 5.125756870051235,
      "grad_norm": 0.27221760102988546,
      "learning_rate": 8.051432516396272e-06,
      "loss": 0.3288,
      "step": 5505
    },
    {
      "epoch": 5.130414531904984,
      "grad_norm": 0.2793564453620868,
      "learning_rate": 8.008284432171211e-06,
      "loss": 0.334,
      "step": 5510
    },
    {
      "epoch": 5.135072193758733,
      "grad_norm": 0.2706148143410356,
      "learning_rate": 7.965136347946152e-06,
      "loss": 0.3174,
      "step": 5515
    },
    {
      "epoch": 5.139729855612482,
      "grad_norm": 0.27576205018512123,
      "learning_rate": 7.921988263721091e-06,
      "loss": 0.33,
      "step": 5520
    },
    {
      "epoch": 5.144387517466232,
      "grad_norm": 0.2663903879078167,
      "learning_rate": 7.87884017949603e-06,
      "loss": 0.3243,
      "step": 5525
    },
    {
      "epoch": 5.149045179319981,
      "grad_norm": 0.28261024102522386,
      "learning_rate": 7.835692095270971e-06,
      "loss": 0.3209,
      "step": 5530
    },
    {
      "epoch": 5.153702841173731,
      "grad_norm": 0.27903807514242895,
      "learning_rate": 7.79254401104591e-06,
      "loss": 0.3262,
      "step": 5535
    },
    {
      "epoch": 5.15836050302748,
      "grad_norm": 0.2794902226839403,
      "learning_rate": 7.749395926820851e-06,
      "loss": 0.323,
      "step": 5540
    },
    {
      "epoch": 5.16301816488123,
      "grad_norm": 0.2727277142872333,
      "learning_rate": 7.706247842595788e-06,
      "loss": 0.3228,
      "step": 5545
    },
    {
      "epoch": 5.167675826734979,
      "grad_norm": 0.28221587286473715,
      "learning_rate": 7.663099758370729e-06,
      "loss": 0.3283,
      "step": 5550
    },
    {
      "epoch": 5.172333488588729,
      "grad_norm": 0.27709436316289415,
      "learning_rate": 7.619951674145669e-06,
      "loss": 0.3268,
      "step": 5555
    },
    {
      "epoch": 5.176991150442478,
      "grad_norm": 0.2728097834761023,
      "learning_rate": 7.576803589920608e-06,
      "loss": 0.3391,
      "step": 5560
    },
    {
      "epoch": 5.181648812296228,
      "grad_norm": 0.2771389556175177,
      "learning_rate": 7.533655505695547e-06,
      "loss": 0.3228,
      "step": 5565
    },
    {
      "epoch": 5.186306474149977,
      "grad_norm": 0.27386669159063143,
      "learning_rate": 7.490507421470487e-06,
      "loss": 0.3234,
      "step": 5570
    },
    {
      "epoch": 5.190964136003726,
      "grad_norm": 0.2643704536632611,
      "learning_rate": 7.447359337245427e-06,
      "loss": 0.3258,
      "step": 5575
    },
    {
      "epoch": 5.195621797857475,
      "grad_norm": 0.27821305255057466,
      "learning_rate": 7.4042112530203655e-06,
      "loss": 0.3204,
      "step": 5580
    },
    {
      "epoch": 5.200279459711225,
      "grad_norm": 0.2709304765323627,
      "learning_rate": 7.361063168795305e-06,
      "loss": 0.324,
      "step": 5585
    },
    {
      "epoch": 5.204937121564974,
      "grad_norm": 0.2815262882923379,
      "learning_rate": 7.317915084570245e-06,
      "loss": 0.3273,
      "step": 5590
    },
    {
      "epoch": 5.209594783418724,
      "grad_norm": 0.2831876418724499,
      "learning_rate": 7.274767000345185e-06,
      "loss": 0.3266,
      "step": 5595
    },
    {
      "epoch": 5.214252445272473,
      "grad_norm": 0.2733400984856177,
      "learning_rate": 7.231618916120124e-06,
      "loss": 0.3302,
      "step": 5600
    },
    {
      "epoch": 5.218910107126223,
      "grad_norm": 0.26940794170158017,
      "learning_rate": 7.1884708318950636e-06,
      "loss": 0.3302,
      "step": 5605
    },
    {
      "epoch": 5.223567768979972,
      "grad_norm": 0.2789756352353802,
      "learning_rate": 7.1453227476700035e-06,
      "loss": 0.3324,
      "step": 5610
    },
    {
      "epoch": 5.228225430833722,
      "grad_norm": 0.28487371347270457,
      "learning_rate": 7.1021746634449435e-06,
      "loss": 0.3206,
      "step": 5615
    },
    {
      "epoch": 5.232883092687471,
      "grad_norm": 0.27160123667880864,
      "learning_rate": 7.059026579219883e-06,
      "loss": 0.3238,
      "step": 5620
    },
    {
      "epoch": 5.237540754541221,
      "grad_norm": 0.27928347263685827,
      "learning_rate": 7.0158784949948226e-06,
      "loss": 0.3151,
      "step": 5625
    },
    {
      "epoch": 5.242198416394969,
      "grad_norm": 0.2781358355739859,
      "learning_rate": 6.9727304107697625e-06,
      "loss": 0.3212,
      "step": 5630
    },
    {
      "epoch": 5.246856078248719,
      "grad_norm": 0.2862079262979005,
      "learning_rate": 6.9295823265447025e-06,
      "loss": 0.3274,
      "step": 5635
    },
    {
      "epoch": 5.251513740102468,
      "grad_norm": 0.2710030473306093,
      "learning_rate": 6.886434242319641e-06,
      "loss": 0.3189,
      "step": 5640
    },
    {
      "epoch": 5.256171401956218,
      "grad_norm": 0.26074803518533335,
      "learning_rate": 6.843286158094581e-06,
      "loss": 0.3226,
      "step": 5645
    },
    {
      "epoch": 5.260829063809967,
      "grad_norm": 0.277688154721736,
      "learning_rate": 6.800138073869521e-06,
      "loss": 0.3243,
      "step": 5650
    },
    {
      "epoch": 5.265486725663717,
      "grad_norm": 0.2738583543835661,
      "learning_rate": 6.756989989644461e-06,
      "loss": 0.3221,
      "step": 5655
    },
    {
      "epoch": 5.270144387517466,
      "grad_norm": 0.27383033130609186,
      "learning_rate": 6.713841905419399e-06,
      "loss": 0.3203,
      "step": 5660
    },
    {
      "epoch": 5.274802049371216,
      "grad_norm": 0.26870097080985705,
      "learning_rate": 6.670693821194339e-06,
      "loss": 0.3272,
      "step": 5665
    },
    {
      "epoch": 5.279459711224965,
      "grad_norm": 0.26977997106752877,
      "learning_rate": 6.627545736969279e-06,
      "loss": 0.3213,
      "step": 5670
    },
    {
      "epoch": 5.284117373078715,
      "grad_norm": 0.26150827467634463,
      "learning_rate": 6.584397652744219e-06,
      "loss": 0.3173,
      "step": 5675
    },
    {
      "epoch": 5.288775034932464,
      "grad_norm": 0.2774145928019438,
      "learning_rate": 6.541249568519157e-06,
      "loss": 0.3321,
      "step": 5680
    },
    {
      "epoch": 5.293432696786214,
      "grad_norm": 0.27429842944141813,
      "learning_rate": 6.498101484294097e-06,
      "loss": 0.3261,
      "step": 5685
    },
    {
      "epoch": 5.298090358639962,
      "grad_norm": 0.26463731097232146,
      "learning_rate": 6.454953400069037e-06,
      "loss": 0.3263,
      "step": 5690
    },
    {
      "epoch": 5.302748020493712,
      "grad_norm": 0.2708950580055502,
      "learning_rate": 6.411805315843977e-06,
      "loss": 0.3251,
      "step": 5695
    },
    {
      "epoch": 5.307405682347461,
      "grad_norm": 0.2754521917538764,
      "learning_rate": 6.368657231618916e-06,
      "loss": 0.3217,
      "step": 5700
    },
    {
      "epoch": 5.312063344201211,
      "grad_norm": 0.2648722292193049,
      "learning_rate": 6.325509147393856e-06,
      "loss": 0.321,
      "step": 5705
    },
    {
      "epoch": 5.31672100605496,
      "grad_norm": 0.2783474157736064,
      "learning_rate": 6.282361063168796e-06,
      "loss": 0.3261,
      "step": 5710
    },
    {
      "epoch": 5.32137866790871,
      "grad_norm": 0.2675979662284909,
      "learning_rate": 6.239212978943735e-06,
      "loss": 0.3258,
      "step": 5715
    },
    {
      "epoch": 5.326036329762459,
      "grad_norm": 0.28180346121203653,
      "learning_rate": 6.196064894718675e-06,
      "loss": 0.3253,
      "step": 5720
    },
    {
      "epoch": 5.330693991616209,
      "grad_norm": 0.2865301145006838,
      "learning_rate": 6.152916810493614e-06,
      "loss": 0.3278,
      "step": 5725
    },
    {
      "epoch": 5.335351653469958,
      "grad_norm": 0.2724091610549319,
      "learning_rate": 6.109768726268554e-06,
      "loss": 0.3278,
      "step": 5730
    },
    {
      "epoch": 5.340009315323708,
      "grad_norm": 0.2688083756592549,
      "learning_rate": 6.066620642043493e-06,
      "loss": 0.3261,
      "step": 5735
    },
    {
      "epoch": 5.344666977177457,
      "grad_norm": 0.2732778229755867,
      "learning_rate": 6.023472557818433e-06,
      "loss": 0.3278,
      "step": 5740
    },
    {
      "epoch": 5.349324639031206,
      "grad_norm": 0.28546452599505634,
      "learning_rate": 5.980324473593372e-06,
      "loss": 0.323,
      "step": 5745
    },
    {
      "epoch": 5.353982300884955,
      "grad_norm": 0.2722017835509351,
      "learning_rate": 5.937176389368312e-06,
      "loss": 0.321,
      "step": 5750
    },
    {
      "epoch": 5.358639962738705,
      "grad_norm": 0.28351926790675114,
      "learning_rate": 5.894028305143251e-06,
      "loss": 0.322,
      "step": 5755
    },
    {
      "epoch": 5.363297624592454,
      "grad_norm": 0.2701525437368716,
      "learning_rate": 5.850880220918191e-06,
      "loss": 0.3236,
      "step": 5760
    },
    {
      "epoch": 5.367955286446204,
      "grad_norm": 0.2776258778975566,
      "learning_rate": 5.807732136693131e-06,
      "loss": 0.3213,
      "step": 5765
    },
    {
      "epoch": 5.372612948299953,
      "grad_norm": 0.28116025849521575,
      "learning_rate": 5.76458405246807e-06,
      "loss": 0.3261,
      "step": 5770
    },
    {
      "epoch": 5.377270610153703,
      "grad_norm": 0.2627699896176315,
      "learning_rate": 5.72143596824301e-06,
      "loss": 0.3233,
      "step": 5775
    },
    {
      "epoch": 5.381928272007452,
      "grad_norm": 0.28065600992932016,
      "learning_rate": 5.67828788401795e-06,
      "loss": 0.3227,
      "step": 5780
    },
    {
      "epoch": 5.386585933861202,
      "grad_norm": 0.28045633724442537,
      "learning_rate": 5.6351397997928894e-06,
      "loss": 0.3244,
      "step": 5785
    },
    {
      "epoch": 5.391243595714951,
      "grad_norm": 0.2682133683313105,
      "learning_rate": 5.591991715567829e-06,
      "loss": 0.3332,
      "step": 5790
    },
    {
      "epoch": 5.395901257568701,
      "grad_norm": 0.2574831802872796,
      "learning_rate": 5.5488436313427685e-06,
      "loss": 0.3305,
      "step": 5795
    },
    {
      "epoch": 5.4005589194224495,
      "grad_norm": 0.27669542046186485,
      "learning_rate": 5.5056955471177085e-06,
      "loss": 0.3231,
      "step": 5800
    },
    {
      "epoch": 5.405216581276199,
      "grad_norm": 0.2787620680019865,
      "learning_rate": 5.462547462892648e-06,
      "loss": 0.3399,
      "step": 5805
    },
    {
      "epoch": 5.4098742431299485,
      "grad_norm": 0.2714431532007126,
      "learning_rate": 5.4193993786675876e-06,
      "loss": 0.321,
      "step": 5810
    },
    {
      "epoch": 5.414531904983698,
      "grad_norm": 0.2685902076479993,
      "learning_rate": 5.376251294442527e-06,
      "loss": 0.3207,
      "step": 5815
    },
    {
      "epoch": 5.4191895668374475,
      "grad_norm": 0.2583670908882243,
      "learning_rate": 5.333103210217467e-06,
      "loss": 0.3216,
      "step": 5820
    },
    {
      "epoch": 5.423847228691197,
      "grad_norm": 0.2698609916315855,
      "learning_rate": 5.289955125992406e-06,
      "loss": 0.325,
      "step": 5825
    },
    {
      "epoch": 5.4285048905449464,
      "grad_norm": 0.27976276630270236,
      "learning_rate": 5.246807041767346e-06,
      "loss": 0.3215,
      "step": 5830
    },
    {
      "epoch": 5.433162552398696,
      "grad_norm": 0.269800064030816,
      "learning_rate": 5.203658957542285e-06,
      "loss": 0.3235,
      "step": 5835
    },
    {
      "epoch": 5.437820214252445,
      "grad_norm": 0.270110822319812,
      "learning_rate": 5.160510873317225e-06,
      "loss": 0.336,
      "step": 5840
    },
    {
      "epoch": 5.442477876106195,
      "grad_norm": 0.2661720373161581,
      "learning_rate": 5.117362789092165e-06,
      "loss": 0.3204,
      "step": 5845
    },
    {
      "epoch": 5.447135537959944,
      "grad_norm": 0.27549687132675077,
      "learning_rate": 5.074214704867105e-06,
      "loss": 0.329,
      "step": 5850
    },
    {
      "epoch": 5.451793199813694,
      "grad_norm": 0.27454815444463065,
      "learning_rate": 5.031066620642044e-06,
      "loss": 0.3268,
      "step": 5855
    },
    {
      "epoch": 5.4564508616674425,
      "grad_norm": 0.2671432333765382,
      "learning_rate": 4.987918536416984e-06,
      "loss": 0.3259,
      "step": 5860
    },
    {
      "epoch": 5.461108523521192,
      "grad_norm": 0.2776308377069313,
      "learning_rate": 4.944770452191923e-06,
      "loss": 0.3361,
      "step": 5865
    },
    {
      "epoch": 5.4657661853749415,
      "grad_norm": 0.2741324724752247,
      "learning_rate": 4.901622367966863e-06,
      "loss": 0.3244,
      "step": 5870
    },
    {
      "epoch": 5.470423847228691,
      "grad_norm": 0.284773103620731,
      "learning_rate": 4.858474283741802e-06,
      "loss": 0.3109,
      "step": 5875
    },
    {
      "epoch": 5.4750815090824405,
      "grad_norm": 0.26337805671358033,
      "learning_rate": 4.815326199516742e-06,
      "loss": 0.3227,
      "step": 5880
    },
    {
      "epoch": 5.47973917093619,
      "grad_norm": 0.27290215356504455,
      "learning_rate": 4.772178115291681e-06,
      "loss": 0.324,
      "step": 5885
    },
    {
      "epoch": 5.4843968327899395,
      "grad_norm": 0.27099741713079173,
      "learning_rate": 4.729030031066621e-06,
      "loss": 0.329,
      "step": 5890
    },
    {
      "epoch": 5.489054494643689,
      "grad_norm": 0.2621758723791822,
      "learning_rate": 4.68588194684156e-06,
      "loss": 0.3301,
      "step": 5895
    },
    {
      "epoch": 5.4937121564974385,
      "grad_norm": 0.2550523051836718,
      "learning_rate": 4.6427338626165e-06,
      "loss": 0.3222,
      "step": 5900
    },
    {
      "epoch": 5.498369818351188,
      "grad_norm": 0.265048431825132,
      "learning_rate": 4.599585778391439e-06,
      "loss": 0.3357,
      "step": 5905
    },
    {
      "epoch": 5.5030274802049375,
      "grad_norm": 0.26939368040057626,
      "learning_rate": 4.556437694166379e-06,
      "loss": 0.3192,
      "step": 5910
    },
    {
      "epoch": 5.507685142058687,
      "grad_norm": 0.2719154846071998,
      "learning_rate": 4.513289609941319e-06,
      "loss": 0.3246,
      "step": 5915
    },
    {
      "epoch": 5.512342803912436,
      "grad_norm": 0.26863015623342323,
      "learning_rate": 4.470141525716258e-06,
      "loss": 0.3204,
      "step": 5920
    },
    {
      "epoch": 5.517000465766185,
      "grad_norm": 0.25331108427121823,
      "learning_rate": 4.426993441491198e-06,
      "loss": 0.3268,
      "step": 5925
    },
    {
      "epoch": 5.521658127619935,
      "grad_norm": 0.2695958191665748,
      "learning_rate": 4.383845357266138e-06,
      "loss": 0.3225,
      "step": 5930
    },
    {
      "epoch": 5.526315789473684,
      "grad_norm": 0.2627809242570627,
      "learning_rate": 4.340697273041077e-06,
      "loss": 0.3232,
      "step": 5935
    },
    {
      "epoch": 5.530973451327434,
      "grad_norm": 0.2671969458438391,
      "learning_rate": 4.297549188816017e-06,
      "loss": 0.3241,
      "step": 5940
    },
    {
      "epoch": 5.535631113181183,
      "grad_norm": 0.26059040855486354,
      "learning_rate": 4.254401104590956e-06,
      "loss": 0.326,
      "step": 5945
    },
    {
      "epoch": 5.5402887750349326,
      "grad_norm": 0.27012358560337557,
      "learning_rate": 4.211253020365896e-06,
      "loss": 0.336,
      "step": 5950
    },
    {
      "epoch": 5.544946436888682,
      "grad_norm": 0.28165102409472464,
      "learning_rate": 4.168104936140835e-06,
      "loss": 0.3132,
      "step": 5955
    },
    {
      "epoch": 5.5496040987424315,
      "grad_norm": 0.26319772799719426,
      "learning_rate": 4.124956851915775e-06,
      "loss": 0.3252,
      "step": 5960
    },
    {
      "epoch": 5.554261760596181,
      "grad_norm": 0.267475065175135,
      "learning_rate": 4.0818087676907145e-06,
      "loss": 0.3236,
      "step": 5965
    },
    {
      "epoch": 5.5589194224499305,
      "grad_norm": 0.2722132151367968,
      "learning_rate": 4.0386606834656544e-06,
      "loss": 0.3203,
      "step": 5970
    },
    {
      "epoch": 5.563577084303679,
      "grad_norm": 0.26629036068425055,
      "learning_rate": 3.9955125992405935e-06,
      "loss": 0.3263,
      "step": 5975
    },
    {
      "epoch": 5.568234746157429,
      "grad_norm": 0.26823733476686584,
      "learning_rate": 3.9523645150155335e-06,
      "loss": 0.3263,
      "step": 5980
    },
    {
      "epoch": 5.572892408011178,
      "grad_norm": 0.2769818914469904,
      "learning_rate": 3.909216430790473e-06,
      "loss": 0.3242,
      "step": 5985
    },
    {
      "epoch": 5.577550069864928,
      "grad_norm": 0.26362164902638746,
      "learning_rate": 3.8660683465654126e-06,
      "loss": 0.3212,
      "step": 5990
    },
    {
      "epoch": 5.582207731718677,
      "grad_norm": 0.2796612118034883,
      "learning_rate": 3.8229202623403525e-06,
      "loss": 0.3212,
      "step": 5995
    },
    {
      "epoch": 5.586865393572427,
      "grad_norm": 0.2815453969027614,
      "learning_rate": 3.779772178115292e-06,
      "loss": 0.3315,
      "step": 6000
    },
    {
      "epoch": 5.591523055426176,
      "grad_norm": 0.2633359775765832,
      "learning_rate": 3.736624093890231e-06,
      "loss": 0.3299,
      "step": 6005
    },
    {
      "epoch": 5.596180717279926,
      "grad_norm": 0.2607616107583197,
      "learning_rate": 3.693476009665171e-06,
      "loss": 0.3255,
      "step": 6010
    },
    {
      "epoch": 5.600838379133675,
      "grad_norm": 0.2667612618087867,
      "learning_rate": 3.6503279254401107e-06,
      "loss": 0.3291,
      "step": 6015
    },
    {
      "epoch": 5.605496040987425,
      "grad_norm": 0.2652062533004386,
      "learning_rate": 3.6071798412150506e-06,
      "loss": 0.322,
      "step": 6020
    },
    {
      "epoch": 5.610153702841174,
      "grad_norm": 0.2771634189985546,
      "learning_rate": 3.5640317569899898e-06,
      "loss": 0.3183,
      "step": 6025
    },
    {
      "epoch": 5.614811364694923,
      "grad_norm": 0.28002632163310337,
      "learning_rate": 3.5208836727649297e-06,
      "loss": 0.323,
      "step": 6030
    },
    {
      "epoch": 5.619469026548672,
      "grad_norm": 0.26420063684223355,
      "learning_rate": 3.477735588539869e-06,
      "loss": 0.3188,
      "step": 6035
    },
    {
      "epoch": 5.624126688402422,
      "grad_norm": 0.2725342248739957,
      "learning_rate": 3.434587504314809e-06,
      "loss": 0.3206,
      "step": 6040
    },
    {
      "epoch": 5.628784350256171,
      "grad_norm": 0.26013509574306537,
      "learning_rate": 3.3914394200897483e-06,
      "loss": 0.3249,
      "step": 6045
    },
    {
      "epoch": 5.633442012109921,
      "grad_norm": 0.27047301989412564,
      "learning_rate": 3.348291335864688e-06,
      "loss": 0.3334,
      "step": 6050
    },
    {
      "epoch": 5.63809967396367,
      "grad_norm": 0.25803949731361225,
      "learning_rate": 3.3051432516396274e-06,
      "loss": 0.3212,
      "step": 6055
    },
    {
      "epoch": 5.64275733581742,
      "grad_norm": 0.2686675235173311,
      "learning_rate": 3.2619951674145674e-06,
      "loss": 0.3237,
      "step": 6060
    },
    {
      "epoch": 5.647414997671169,
      "grad_norm": 0.2546745467593447,
      "learning_rate": 3.2188470831895065e-06,
      "loss": 0.3209,
      "step": 6065
    },
    {
      "epoch": 5.652072659524919,
      "grad_norm": 0.25822354290943483,
      "learning_rate": 3.1756989989644464e-06,
      "loss": 0.323,
      "step": 6070
    },
    {
      "epoch": 5.656730321378668,
      "grad_norm": 0.2629896006695681,
      "learning_rate": 3.1325509147393856e-06,
      "loss": 0.3266,
      "step": 6075
    },
    {
      "epoch": 5.661387983232418,
      "grad_norm": 0.2583600294815806,
      "learning_rate": 3.089402830514325e-06,
      "loss": 0.3269,
      "step": 6080
    },
    {
      "epoch": 5.666045645086166,
      "grad_norm": 0.27590958900340956,
      "learning_rate": 3.046254746289265e-06,
      "loss": 0.3303,
      "step": 6085
    },
    {
      "epoch": 5.670703306939917,
      "grad_norm": 0.26067164711818963,
      "learning_rate": 3.0031066620642046e-06,
      "loss": 0.3179,
      "step": 6090
    },
    {
      "epoch": 5.675360968793665,
      "grad_norm": 0.2658805466979597,
      "learning_rate": 2.959958577839144e-06,
      "loss": 0.318,
      "step": 6095
    },
    {
      "epoch": 5.680018630647415,
      "grad_norm": 0.2537552525951581,
      "learning_rate": 2.9168104936140837e-06,
      "loss": 0.3245,
      "step": 6100
    },
    {
      "epoch": 5.684676292501164,
      "grad_norm": 0.2695369749083651,
      "learning_rate": 2.873662409389023e-06,
      "loss": 0.3285,
      "step": 6105
    },
    {
      "epoch": 5.689333954354914,
      "grad_norm": 0.26490406079533985,
      "learning_rate": 2.8305143251639627e-06,
      "loss": 0.3243,
      "step": 6110
    },
    {
      "epoch": 5.693991616208663,
      "grad_norm": 0.2665741459640961,
      "learning_rate": 2.7873662409389023e-06,
      "loss": 0.3202,
      "step": 6115
    },
    {
      "epoch": 5.698649278062413,
      "grad_norm": 0.26787899641585156,
      "learning_rate": 2.7442181567138422e-06,
      "loss": 0.3286,
      "step": 6120
    },
    {
      "epoch": 5.703306939916162,
      "grad_norm": 0.2799076699319239,
      "learning_rate": 2.7010700724887818e-06,
      "loss": 0.3279,
      "step": 6125
    },
    {
      "epoch": 5.707964601769912,
      "grad_norm": 0.2481493445003915,
      "learning_rate": 2.6579219882637213e-06,
      "loss": 0.3199,
      "step": 6130
    },
    {
      "epoch": 5.712622263623661,
      "grad_norm": 0.2599845438111195,
      "learning_rate": 2.614773904038661e-06,
      "loss": 0.3255,
      "step": 6135
    },
    {
      "epoch": 5.717279925477411,
      "grad_norm": 0.27060899774271624,
      "learning_rate": 2.5716258198136004e-06,
      "loss": 0.327,
      "step": 6140
    },
    {
      "epoch": 5.72193758733116,
      "grad_norm": 0.2590176204790071,
      "learning_rate": 2.52847773558854e-06,
      "loss": 0.3198,
      "step": 6145
    },
    {
      "epoch": 5.726595249184909,
      "grad_norm": 0.26178333295429757,
      "learning_rate": 2.4853296513634795e-06,
      "loss": 0.3267,
      "step": 6150
    },
    {
      "epoch": 5.731252911038658,
      "grad_norm": 0.2630243539880319,
      "learning_rate": 2.442181567138419e-06,
      "loss": 0.3316,
      "step": 6155
    },
    {
      "epoch": 5.735910572892408,
      "grad_norm": 0.2631374882852561,
      "learning_rate": 2.399033482913359e-06,
      "loss": 0.3291,
      "step": 6160
    },
    {
      "epoch": 5.740568234746157,
      "grad_norm": 0.26393967808378715,
      "learning_rate": 2.3558853986882985e-06,
      "loss": 0.331,
      "step": 6165
    },
    {
      "epoch": 5.745225896599907,
      "grad_norm": 0.26760843920955996,
      "learning_rate": 2.312737314463238e-06,
      "loss": 0.3183,
      "step": 6170
    },
    {
      "epoch": 5.749883558453656,
      "grad_norm": 0.26251309263189054,
      "learning_rate": 2.2695892302381776e-06,
      "loss": 0.3266,
      "step": 6175
    },
    {
      "epoch": 5.754541220307406,
      "grad_norm": 0.25870529829179206,
      "learning_rate": 2.226441146013117e-06,
      "loss": 0.3251,
      "step": 6180
    },
    {
      "epoch": 5.759198882161155,
      "grad_norm": 0.27157525299149055,
      "learning_rate": 2.1832930617880566e-06,
      "loss": 0.3244,
      "step": 6185
    },
    {
      "epoch": 5.763856544014905,
      "grad_norm": 0.26234589524974233,
      "learning_rate": 2.140144977562996e-06,
      "loss": 0.3197,
      "step": 6190
    },
    {
      "epoch": 5.768514205868654,
      "grad_norm": 0.2633612044495917,
      "learning_rate": 2.096996893337936e-06,
      "loss": 0.3304,
      "step": 6195
    },
    {
      "epoch": 5.773171867722404,
      "grad_norm": 0.2632435548790163,
      "learning_rate": 2.0538488091128757e-06,
      "loss": 0.328,
      "step": 6200
    },
    {
      "epoch": 5.777829529576152,
      "grad_norm": 0.2655603685497178,
      "learning_rate": 2.010700724887815e-06,
      "loss": 0.3315,
      "step": 6205
    },
    {
      "epoch": 5.782487191429902,
      "grad_norm": 0.25308082006023114,
      "learning_rate": 1.9675526406627547e-06,
      "loss": 0.3211,
      "step": 6210
    },
    {
      "epoch": 5.787144853283651,
      "grad_norm": 0.2559704509031632,
      "learning_rate": 1.9244045564376943e-06,
      "loss": 0.3209,
      "step": 6215
    },
    {
      "epoch": 5.791802515137401,
      "grad_norm": 0.2626242916992896,
      "learning_rate": 1.8812564722126338e-06,
      "loss": 0.321,
      "step": 6220
    },
    {
      "epoch": 5.79646017699115,
      "grad_norm": 0.25671241184053134,
      "learning_rate": 1.8381083879875736e-06,
      "loss": 0.3263,
      "step": 6225
    },
    {
      "epoch": 5.8011178388449,
      "grad_norm": 0.2602343944514577,
      "learning_rate": 1.794960303762513e-06,
      "loss": 0.3336,
      "step": 6230
    },
    {
      "epoch": 5.805775500698649,
      "grad_norm": 0.26736758906804503,
      "learning_rate": 1.7518122195374526e-06,
      "loss": 0.3293,
      "step": 6235
    },
    {
      "epoch": 5.810433162552399,
      "grad_norm": 0.2605198442383094,
      "learning_rate": 1.7086641353123924e-06,
      "loss": 0.3251,
      "step": 6240
    },
    {
      "epoch": 5.815090824406148,
      "grad_norm": 0.25528187338633446,
      "learning_rate": 1.665516051087332e-06,
      "loss": 0.3241,
      "step": 6245
    },
    {
      "epoch": 5.819748486259898,
      "grad_norm": 0.26906071934823633,
      "learning_rate": 1.6223679668622715e-06,
      "loss": 0.3262,
      "step": 6250
    },
    {
      "epoch": 5.824406148113647,
      "grad_norm": 0.26152734523541193,
      "learning_rate": 1.579219882637211e-06,
      "loss": 0.3348,
      "step": 6255
    },
    {
      "epoch": 5.829063809967396,
      "grad_norm": 0.26429652028969647,
      "learning_rate": 1.5360717984121505e-06,
      "loss": 0.3191,
      "step": 6260
    },
    {
      "epoch": 5.833721471821145,
      "grad_norm": 0.2648363258431519,
      "learning_rate": 1.49292371418709e-06,
      "loss": 0.3246,
      "step": 6265
    },
    {
      "epoch": 5.838379133674895,
      "grad_norm": 0.26682311499437245,
      "learning_rate": 1.4497756299620296e-06,
      "loss": 0.3329,
      "step": 6270
    },
    {
      "epoch": 5.843036795528644,
      "grad_norm": 0.262174779773615,
      "learning_rate": 1.4066275457369694e-06,
      "loss": 0.3259,
      "step": 6275
    },
    {
      "epoch": 5.847694457382394,
      "grad_norm": 0.25716452932251094,
      "learning_rate": 1.363479461511909e-06,
      "loss": 0.325,
      "step": 6280
    },
    {
      "epoch": 5.852352119236143,
      "grad_norm": 0.25630900933943085,
      "learning_rate": 1.3203313772868484e-06,
      "loss": 0.3267,
      "step": 6285
    },
    {
      "epoch": 5.857009781089893,
      "grad_norm": 0.25836082753914014,
      "learning_rate": 1.277183293061788e-06,
      "loss": 0.3199,
      "step": 6290
    },
    {
      "epoch": 5.861667442943642,
      "grad_norm": 0.24773755524187502,
      "learning_rate": 1.2340352088367277e-06,
      "loss": 0.319,
      "step": 6295
    },
    {
      "epoch": 5.866325104797392,
      "grad_norm": 0.25561794591552056,
      "learning_rate": 1.1908871246116673e-06,
      "loss": 0.3218,
      "step": 6300
    },
    {
      "epoch": 5.870982766651141,
      "grad_norm": 0.26291390968554823,
      "learning_rate": 1.1477390403866068e-06,
      "loss": 0.3151,
      "step": 6305
    },
    {
      "epoch": 5.875640428504891,
      "grad_norm": 0.26031774279787967,
      "learning_rate": 1.1045909561615463e-06,
      "loss": 0.3304,
      "step": 6310
    },
    {
      "epoch": 5.8802980903586395,
      "grad_norm": 0.25829889881103313,
      "learning_rate": 1.061442871936486e-06,
      "loss": 0.3126,
      "step": 6315
    },
    {
      "epoch": 5.88495575221239,
      "grad_norm": 0.25282510814984144,
      "learning_rate": 1.0182947877114256e-06,
      "loss": 0.3214,
      "step": 6320
    },
    {
      "epoch": 5.8896134140661385,
      "grad_norm": 0.26442373579895045,
      "learning_rate": 9.751467034863652e-07,
      "loss": 0.3226,
      "step": 6325
    },
    {
      "epoch": 5.894271075919888,
      "grad_norm": 0.2560661677547824,
      "learning_rate": 9.319986192613048e-07,
      "loss": 0.3323,
      "step": 6330
    },
    {
      "epoch": 5.8989287377736375,
      "grad_norm": 0.26174285445855033,
      "learning_rate": 8.888505350362444e-07,
      "loss": 0.3188,
      "step": 6335
    },
    {
      "epoch": 5.903586399627387,
      "grad_norm": 0.26055593595724325,
      "learning_rate": 8.45702450811184e-07,
      "loss": 0.3185,
      "step": 6340
    },
    {
      "epoch": 5.9082440614811365,
      "grad_norm": 0.25976635590670094,
      "learning_rate": 8.025543665861236e-07,
      "loss": 0.3133,
      "step": 6345
    },
    {
      "epoch": 5.912901723334886,
      "grad_norm": 0.2541580223080462,
      "learning_rate": 7.594062823610632e-07,
      "loss": 0.3238,
      "step": 6350
    },
    {
      "epoch": 5.9175593851886354,
      "grad_norm": 0.2600371258220012,
      "learning_rate": 7.162581981360028e-07,
      "loss": 0.3242,
      "step": 6355
    },
    {
      "epoch": 5.922217047042385,
      "grad_norm": 0.25277140648963686,
      "learning_rate": 6.731101139109423e-07,
      "loss": 0.3161,
      "step": 6360
    },
    {
      "epoch": 5.926874708896134,
      "grad_norm": 0.25628940591983723,
      "learning_rate": 6.29962029685882e-07,
      "loss": 0.3295,
      "step": 6365
    },
    {
      "epoch": 5.931532370749884,
      "grad_norm": 0.25686383318821293,
      "learning_rate": 5.868139454608215e-07,
      "loss": 0.3155,
      "step": 6370
    },
    {
      "epoch": 5.936190032603633,
      "grad_norm": 0.2612275438410046,
      "learning_rate": 5.436658612357612e-07,
      "loss": 0.3342,
      "step": 6375
    },
    {
      "epoch": 5.940847694457382,
      "grad_norm": 0.2592026166543193,
      "learning_rate": 5.005177770107007e-07,
      "loss": 0.3223,
      "step": 6380
    },
    {
      "epoch": 5.9455053563111315,
      "grad_norm": 0.2604255559264583,
      "learning_rate": 4.5736969278564034e-07,
      "loss": 0.326,
      "step": 6385
    },
    {
      "epoch": 5.950163018164881,
      "grad_norm": 0.2647208781008018,
      "learning_rate": 4.142216085605799e-07,
      "loss": 0.3344,
      "step": 6390
    },
    {
      "epoch": 5.9548206800186305,
      "grad_norm": 0.25886616409323504,
      "learning_rate": 3.710735243355195e-07,
      "loss": 0.3293,
      "step": 6395
    },
    {
      "epoch": 5.95947834187238,
      "grad_norm": 0.255081815321021,
      "learning_rate": 3.279254401104591e-07,
      "loss": 0.3262,
      "step": 6400
    },
    {
      "epoch": 5.9641360037261295,
      "grad_norm": 0.24994544152471945,
      "learning_rate": 2.847773558853987e-07,
      "loss": 0.3201,
      "step": 6405
    },
    {
      "epoch": 5.968793665579879,
      "grad_norm": 0.260426235428334,
      "learning_rate": 2.416292716603383e-07,
      "loss": 0.323,
      "step": 6410
    },
    {
      "epoch": 5.9734513274336285,
      "grad_norm": 0.2540278359263777,
      "learning_rate": 1.984811874352779e-07,
      "loss": 0.3123,
      "step": 6415
    },
    {
      "epoch": 5.978108989287378,
      "grad_norm": 0.260832146351295,
      "learning_rate": 1.5533310321021747e-07,
      "loss": 0.3164,
      "step": 6420
    },
    {
      "epoch": 5.9827666511411275,
      "grad_norm": 0.25201373803146826,
      "learning_rate": 1.1218501898515707e-07,
      "loss": 0.3231,
      "step": 6425
    },
    {
      "epoch": 5.987424312994877,
      "grad_norm": 0.2572838674705241,
      "learning_rate": 6.903693476009665e-08,
      "loss": 0.3305,
      "step": 6430
    },
    {
      "epoch": 5.992081974848626,
      "grad_norm": 0.2557005663617927,
      "learning_rate": 2.5888850535036245e-08,
      "loss": 0.3262,
      "step": 6435
    },
    {
      "epoch": 5.994876571960876,
      "step": 6438,
      "total_flos": 2925479637680128.0,
      "train_loss": 0.4274832340603698,
      "train_runtime": 44714.5203,
      "train_samples_per_second": 2.304,
      "train_steps_per_second": 0.144
    }
  ],
  "logging_steps": 5,
  "max_steps": 6438,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 6,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 2925479637680128.0,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}