{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.9976711690731253,
  "eval_steps": 500,
  "global_step": 3219,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.004657661853749418,
      "grad_norm": 10.188283956780763,
      "learning_rate": 7.763975155279503e-07,
      "loss": 0.9917,
      "step": 5
    },
    {
      "epoch": 0.009315323707498836,
      "grad_norm": 6.89342020049732,
      "learning_rate": 1.5527950310559006e-06,
      "loss": 0.9482,
      "step": 10
    },
    {
      "epoch": 0.013972985561248253,
      "grad_norm": 2.4710345904778768,
      "learning_rate": 2.329192546583851e-06,
      "loss": 0.8315,
      "step": 15
    },
    {
      "epoch": 0.018630647414997672,
      "grad_norm": 3.534878949967021,
      "learning_rate": 3.1055900621118013e-06,
      "loss": 0.7878,
      "step": 20
    },
    {
      "epoch": 0.02328830926874709,
      "grad_norm": 1.8092501281194573,
      "learning_rate": 3.881987577639752e-06,
      "loss": 0.723,
      "step": 25
    },
    {
      "epoch": 0.027945971122496506,
      "grad_norm": 1.7298801307473601,
      "learning_rate": 4.658385093167702e-06,
      "loss": 0.6778,
      "step": 30
    },
    {
      "epoch": 0.032603632976245925,
      "grad_norm": 2.927442859999318,
      "learning_rate": 5.4347826086956525e-06,
      "loss": 0.6462,
      "step": 35
    },
    {
      "epoch": 0.037261294829995344,
      "grad_norm": 1.0663579417274789,
      "learning_rate": 6.2111801242236025e-06,
      "loss": 0.6159,
      "step": 40
    },
    {
      "epoch": 0.04191895668374476,
      "grad_norm": 0.7084820498301786,
      "learning_rate": 6.9875776397515525e-06,
      "loss": 0.5687,
      "step": 45
    },
    {
      "epoch": 0.04657661853749418,
      "grad_norm": 0.6397123955298338,
      "learning_rate": 7.763975155279503e-06,
      "loss": 0.563,
      "step": 50
    },
    {
      "epoch": 0.05123428039124359,
      "grad_norm": 0.6087947198605437,
      "learning_rate": 8.540372670807453e-06,
      "loss": 0.5448,
      "step": 55
    },
    {
      "epoch": 0.05589194224499301,
      "grad_norm": 0.5590508269451022,
      "learning_rate": 9.316770186335403e-06,
      "loss": 0.5366,
      "step": 60
    },
    {
      "epoch": 0.06054960409874243,
      "grad_norm": 0.6347565773543817,
      "learning_rate": 1.0093167701863353e-05,
      "loss": 0.525,
      "step": 65
    },
    {
      "epoch": 0.06520726595249185,
      "grad_norm": 0.7688534068889166,
      "learning_rate": 1.0869565217391305e-05,
      "loss": 0.5019,
      "step": 70
    },
    {
      "epoch": 0.06986492780624126,
      "grad_norm": 0.6139375497512776,
      "learning_rate": 1.1645962732919255e-05,
      "loss": 0.5027,
      "step": 75
    },
    {
      "epoch": 0.07452258965999069,
      "grad_norm": 0.5971921611199407,
      "learning_rate": 1.2422360248447205e-05,
      "loss": 0.5121,
      "step": 80
    },
    {
      "epoch": 0.0791802515137401,
      "grad_norm": 0.6004837009895614,
      "learning_rate": 1.3198757763975155e-05,
      "loss": 0.4892,
      "step": 85
    },
    {
      "epoch": 0.08383791336748952,
      "grad_norm": 0.6465842903114484,
      "learning_rate": 1.3975155279503105e-05,
      "loss": 0.4955,
      "step": 90
    },
    {
      "epoch": 0.08849557522123894,
      "grad_norm": 0.6546136972520078,
      "learning_rate": 1.4751552795031057e-05,
      "loss": 0.4991,
      "step": 95
    },
    {
      "epoch": 0.09315323707498836,
      "grad_norm": 0.6522224551560621,
      "learning_rate": 1.5527950310559007e-05,
      "loss": 0.4899,
      "step": 100
    },
    {
      "epoch": 0.09781089892873777,
      "grad_norm": 1.016497595826677,
      "learning_rate": 1.630434782608696e-05,
      "loss": 0.4921,
      "step": 105
    },
    {
      "epoch": 0.10246856078248719,
      "grad_norm": 0.6283259471553262,
      "learning_rate": 1.7080745341614907e-05,
      "loss": 0.4813,
      "step": 110
    },
    {
      "epoch": 0.10712622263623661,
      "grad_norm": 0.7707380010729769,
      "learning_rate": 1.785714285714286e-05,
      "loss": 0.4798,
      "step": 115
    },
    {
      "epoch": 0.11178388448998602,
      "grad_norm": 0.7613331446751173,
      "learning_rate": 1.8633540372670807e-05,
      "loss": 0.4909,
      "step": 120
    },
    {
      "epoch": 0.11644154634373545,
      "grad_norm": 0.7680591677018838,
      "learning_rate": 1.940993788819876e-05,
      "loss": 0.4712,
      "step": 125
    },
    {
      "epoch": 0.12109920819748486,
      "grad_norm": 0.6327184822570511,
      "learning_rate": 2.0186335403726707e-05,
      "loss": 0.4728,
      "step": 130
    },
    {
      "epoch": 0.1257568700512343,
      "grad_norm": 0.6449375484537262,
      "learning_rate": 2.096273291925466e-05,
      "loss": 0.4794,
      "step": 135
    },
    {
      "epoch": 0.1304145319049837,
      "grad_norm": 0.8100284941174016,
      "learning_rate": 2.173913043478261e-05,
      "loss": 0.4733,
      "step": 140
    },
    {
      "epoch": 0.1350721937587331,
      "grad_norm": 1.022080006236384,
      "learning_rate": 2.2515527950310562e-05,
      "loss": 0.4829,
      "step": 145
    },
    {
      "epoch": 0.13972985561248252,
      "grad_norm": 1.1490144508364162,
      "learning_rate": 2.329192546583851e-05,
      "loss": 0.4766,
      "step": 150
    },
    {
      "epoch": 0.14438751746623196,
      "grad_norm": 0.7909850788366714,
      "learning_rate": 2.4068322981366462e-05,
      "loss": 0.4591,
      "step": 155
    },
    {
      "epoch": 0.14904517931998137,
      "grad_norm": 1.0198750042636018,
      "learning_rate": 2.484472049689441e-05,
      "loss": 0.4655,
      "step": 160
    },
    {
      "epoch": 0.1537028411737308,
      "grad_norm": 0.9484914888432379,
      "learning_rate": 2.5621118012422362e-05,
      "loss": 0.4671,
      "step": 165
    },
    {
      "epoch": 0.1583605030274802,
      "grad_norm": 1.1644724187022835,
      "learning_rate": 2.639751552795031e-05,
      "loss": 0.4653,
      "step": 170
    },
    {
      "epoch": 0.1630181648812296,
      "grad_norm": 0.6926062839385089,
      "learning_rate": 2.7173913043478262e-05,
      "loss": 0.4765,
      "step": 175
    },
    {
      "epoch": 0.16767582673497905,
      "grad_norm": 0.812050904096057,
      "learning_rate": 2.795031055900621e-05,
      "loss": 0.4707,
      "step": 180
    },
    {
      "epoch": 0.17233348858872846,
      "grad_norm": 0.9409363829602944,
      "learning_rate": 2.8726708074534165e-05,
      "loss": 0.4684,
      "step": 185
    },
    {
      "epoch": 0.17699115044247787,
      "grad_norm": 0.7266979812399479,
      "learning_rate": 2.9503105590062114e-05,
      "loss": 0.4618,
      "step": 190
    },
    {
      "epoch": 0.18164881229622729,
      "grad_norm": 0.6687811147283036,
      "learning_rate": 3.0279503105590062e-05,
      "loss": 0.4681,
      "step": 195
    },
    {
      "epoch": 0.18630647414997673,
      "grad_norm": 0.6913627334512608,
      "learning_rate": 3.1055900621118014e-05,
      "loss": 0.4428,
      "step": 200
    },
    {
      "epoch": 0.19096413600372614,
      "grad_norm": 1.0069119917873806,
      "learning_rate": 3.183229813664597e-05,
      "loss": 0.4634,
      "step": 205
    },
    {
      "epoch": 0.19562179785747555,
      "grad_norm": 1.0907389335734776,
      "learning_rate": 3.260869565217392e-05,
      "loss": 0.4537,
      "step": 210
    },
    {
      "epoch": 0.20027945971122496,
      "grad_norm": 0.812577537777315,
      "learning_rate": 3.3385093167701865e-05,
      "loss": 0.4651,
      "step": 215
    },
    {
      "epoch": 0.20493712156497437,
      "grad_norm": 0.6736908308827283,
      "learning_rate": 3.4161490683229814e-05,
      "loss": 0.4609,
      "step": 220
    },
    {
      "epoch": 0.2095947834187238,
      "grad_norm": 0.8707477474156307,
      "learning_rate": 3.493788819875777e-05,
      "loss": 0.4641,
      "step": 225
    },
    {
      "epoch": 0.21425244527247322,
      "grad_norm": 0.9454640044846007,
      "learning_rate": 3.571428571428572e-05,
      "loss": 0.4623,
      "step": 230
    },
    {
      "epoch": 0.21891010712622264,
      "grad_norm": 0.8306745122233324,
      "learning_rate": 3.6490683229813665e-05,
      "loss": 0.4575,
      "step": 235
    },
    {
      "epoch": 0.22356776897997205,
      "grad_norm": 0.7901900382606851,
      "learning_rate": 3.7267080745341614e-05,
      "loss": 0.4566,
      "step": 240
    },
    {
      "epoch": 0.22822543083372146,
      "grad_norm": 0.9688819685469978,
      "learning_rate": 3.804347826086957e-05,
      "loss": 0.4484,
      "step": 245
    },
    {
      "epoch": 0.2328830926874709,
      "grad_norm": 0.8062372592137216,
      "learning_rate": 3.881987577639752e-05,
      "loss": 0.4461,
      "step": 250
    },
    {
      "epoch": 0.2375407545412203,
      "grad_norm": 0.7641511825275663,
      "learning_rate": 3.9596273291925465e-05,
      "loss": 0.4566,
      "step": 255
    },
    {
      "epoch": 0.24219841639496972,
      "grad_norm": 0.8042506266307712,
      "learning_rate": 4.0372670807453414e-05,
      "loss": 0.461,
      "step": 260
    },
    {
      "epoch": 0.24685607824871914,
      "grad_norm": 1.0691188316274636,
      "learning_rate": 4.114906832298137e-05,
      "loss": 0.456,
      "step": 265
    },
    {
      "epoch": 0.2515137401024686,
      "grad_norm": 0.6816073576849953,
      "learning_rate": 4.192546583850932e-05,
      "loss": 0.4447,
      "step": 270
    },
    {
      "epoch": 0.25617140195621796,
      "grad_norm": 0.786684092309085,
      "learning_rate": 4.270186335403727e-05,
      "loss": 0.4597,
      "step": 275
    },
    {
      "epoch": 0.2608290638099674,
      "grad_norm": 0.9731728769390697,
      "learning_rate": 4.347826086956522e-05,
      "loss": 0.4492,
      "step": 280
    },
    {
      "epoch": 0.26548672566371684,
      "grad_norm": 0.9241645953731435,
      "learning_rate": 4.425465838509317e-05,
      "loss": 0.4487,
      "step": 285
    },
    {
      "epoch": 0.2701443875174662,
      "grad_norm": 0.9243759296412116,
      "learning_rate": 4.5031055900621124e-05,
      "loss": 0.4482,
      "step": 290
    },
    {
      "epoch": 0.27480204937121566,
      "grad_norm": 0.9724826474280194,
      "learning_rate": 4.580745341614907e-05,
      "loss": 0.4436,
      "step": 295
    },
    {
      "epoch": 0.27945971122496505,
      "grad_norm": 1.167227828042357,
      "learning_rate": 4.658385093167702e-05,
      "loss": 0.4559,
      "step": 300
    },
    {
      "epoch": 0.2841173730787145,
      "grad_norm": 0.9153182557676565,
      "learning_rate": 4.736024844720497e-05,
      "loss": 0.4602,
      "step": 305
    },
    {
      "epoch": 0.2887750349324639,
      "grad_norm": 0.9157744766589929,
      "learning_rate": 4.8136645962732924e-05,
      "loss": 0.4554,
      "step": 310
    },
    {
      "epoch": 0.2934326967862133,
      "grad_norm": 0.9103225109023383,
      "learning_rate": 4.891304347826087e-05,
      "loss": 0.4459,
      "step": 315
    },
    {
      "epoch": 0.29809035863996275,
      "grad_norm": 0.7854037877363089,
      "learning_rate": 4.968944099378882e-05,
      "loss": 0.4533,
      "step": 320
    },
    {
      "epoch": 0.30274802049371213,
      "grad_norm": 0.7248866382387523,
      "learning_rate": 4.994822229892993e-05,
      "loss": 0.4561,
      "step": 325
    },
    {
      "epoch": 0.3074056823474616,
      "grad_norm": 0.9234408398799464,
      "learning_rate": 4.986192613047981e-05,
      "loss": 0.4536,
      "step": 330
    },
    {
      "epoch": 0.312063344201211,
      "grad_norm": 1.117201435064076,
      "learning_rate": 4.977562996202969e-05,
      "loss": 0.4405,
      "step": 335
    },
    {
      "epoch": 0.3167210060549604,
      "grad_norm": 0.8383109216220387,
      "learning_rate": 4.968933379357957e-05,
      "loss": 0.4538,
      "step": 340
    },
    {
      "epoch": 0.32137866790870984,
      "grad_norm": 1.5031864585206804,
      "learning_rate": 4.9603037625129445e-05,
      "loss": 0.4461,
      "step": 345
    },
    {
      "epoch": 0.3260363297624592,
      "grad_norm": 1.2528732711373594,
      "learning_rate": 4.951674145667933e-05,
      "loss": 0.4479,
      "step": 350
    },
    {
      "epoch": 0.33069399161620866,
      "grad_norm": 1.0326050836718679,
      "learning_rate": 4.94304452882292e-05,
      "loss": 0.4411,
      "step": 355
    },
    {
      "epoch": 0.3353516534699581,
      "grad_norm": 0.8294258451312435,
      "learning_rate": 4.934414911977908e-05,
      "loss": 0.4491,
      "step": 360
    },
    {
      "epoch": 0.3400093153237075,
      "grad_norm": 0.6920233118225163,
      "learning_rate": 4.9257852951328965e-05,
      "loss": 0.4486,
      "step": 365
    },
    {
      "epoch": 0.3446669771774569,
      "grad_norm": 0.716055959377867,
      "learning_rate": 4.917155678287884e-05,
      "loss": 0.4583,
      "step": 370
    },
    {
      "epoch": 0.3493246390312063,
      "grad_norm": 0.8854847669687039,
      "learning_rate": 4.908526061442872e-05,
      "loss": 0.4536,
      "step": 375
    },
    {
      "epoch": 0.35398230088495575,
      "grad_norm": 0.6919174441646583,
      "learning_rate": 4.89989644459786e-05,
      "loss": 0.4406,
      "step": 380
    },
    {
      "epoch": 0.3586399627387052,
      "grad_norm": 0.7378018649069698,
      "learning_rate": 4.891266827752848e-05,
      "loss": 0.4357,
      "step": 385
    },
    {
      "epoch": 0.36329762459245457,
      "grad_norm": 0.6407230812695663,
      "learning_rate": 4.882637210907836e-05,
      "loss": 0.4404,
      "step": 390
    },
    {
      "epoch": 0.367955286446204,
      "grad_norm": 0.7383317931197969,
      "learning_rate": 4.874007594062824e-05,
      "loss": 0.4357,
      "step": 395
    },
    {
      "epoch": 0.37261294829995345,
      "grad_norm": 0.7425057031063464,
      "learning_rate": 4.865377977217811e-05,
      "loss": 0.4334,
      "step": 400
    },
    {
      "epoch": 0.37727061015370283,
      "grad_norm": 0.8114113228657345,
      "learning_rate": 4.8567483603728e-05,
      "loss": 0.436,
      "step": 405
    },
    {
      "epoch": 0.3819282720074523,
      "grad_norm": 0.5698710059249709,
      "learning_rate": 4.8481187435277875e-05,
      "loss": 0.4424,
      "step": 410
    },
    {
      "epoch": 0.38658593386120166,
      "grad_norm": 0.6994737184074745,
      "learning_rate": 4.839489126682776e-05,
      "loss": 0.4414,
      "step": 415
    },
    {
      "epoch": 0.3912435957149511,
      "grad_norm": 0.6385013806229771,
      "learning_rate": 4.830859509837763e-05,
      "loss": 0.4319,
      "step": 420
    },
    {
      "epoch": 0.39590125756870054,
      "grad_norm": 0.5959491211736601,
      "learning_rate": 4.822229892992751e-05,
      "loss": 0.4468,
      "step": 425
    },
    {
      "epoch": 0.4005589194224499,
      "grad_norm": 0.819699453502341,
      "learning_rate": 4.8136002761477395e-05,
      "loss": 0.4405,
      "step": 430
    },
    {
      "epoch": 0.40521658127619936,
      "grad_norm": 0.6348766348351661,
      "learning_rate": 4.804970659302727e-05,
      "loss": 0.4439,
      "step": 435
    },
    {
      "epoch": 0.40987424312994875,
      "grad_norm": 0.8059638244221499,
      "learning_rate": 4.796341042457715e-05,
      "loss": 0.451,
      "step": 440
    },
    {
      "epoch": 0.4145319049836982,
      "grad_norm": 0.5979193959619382,
      "learning_rate": 4.787711425612703e-05,
      "loss": 0.4307,
      "step": 445
    },
    {
      "epoch": 0.4191895668374476,
      "grad_norm": 0.5815262333926554,
      "learning_rate": 4.779081808767691e-05,
      "loss": 0.4351,
      "step": 450
    },
    {
      "epoch": 0.423847228691197,
      "grad_norm": 0.9929833143370682,
      "learning_rate": 4.770452191922679e-05,
      "loss": 0.452,
      "step": 455
    },
    {
      "epoch": 0.42850489054494645,
      "grad_norm": 0.5809843900351168,
      "learning_rate": 4.761822575077667e-05,
      "loss": 0.4343,
      "step": 460
    },
    {
      "epoch": 0.43316255239869583,
      "grad_norm": 0.6744414006957897,
      "learning_rate": 4.753192958232654e-05,
      "loss": 0.443,
      "step": 465
    },
    {
      "epoch": 0.43782021425244527,
      "grad_norm": 0.998193763185491,
      "learning_rate": 4.744563341387643e-05,
      "loss": 0.4358,
      "step": 470
    },
    {
      "epoch": 0.4424778761061947,
      "grad_norm": 0.7398951895532502,
      "learning_rate": 4.7359337245426306e-05,
      "loss": 0.4442,
      "step": 475
    },
    {
      "epoch": 0.4471355379599441,
      "grad_norm": 0.6601498362861044,
      "learning_rate": 4.7273041076976184e-05,
      "loss": 0.4303,
      "step": 480
    },
    {
      "epoch": 0.45179319981369354,
      "grad_norm": 0.5471251748953498,
      "learning_rate": 4.718674490852606e-05,
      "loss": 0.4317,
      "step": 485
    },
    {
      "epoch": 0.4564508616674429,
      "grad_norm": 0.5791221769322622,
      "learning_rate": 4.710044874007594e-05,
      "loss": 0.435,
      "step": 490
    },
    {
      "epoch": 0.46110852352119236,
      "grad_norm": 0.5530208900082192,
      "learning_rate": 4.7014152571625826e-05,
      "loss": 0.4279,
      "step": 495
    },
    {
      "epoch": 0.4657661853749418,
      "grad_norm": 0.709005380039522,
      "learning_rate": 4.6927856403175704e-05,
      "loss": 0.4346,
      "step": 500
    },
    {
      "epoch": 0.4704238472286912,
      "grad_norm": 0.4177160036682756,
      "learning_rate": 4.684156023472558e-05,
      "loss": 0.4288,
      "step": 505
    },
    {
      "epoch": 0.4750815090824406,
      "grad_norm": 0.5997886397097265,
      "learning_rate": 4.675526406627546e-05,
      "loss": 0.4247,
      "step": 510
    },
    {
      "epoch": 0.47973917093619,
      "grad_norm": 0.5902291201193781,
      "learning_rate": 4.666896789782534e-05,
      "loss": 0.4356,
      "step": 515
    },
    {
      "epoch": 0.48439683278993945,
      "grad_norm": 0.7871793279638812,
      "learning_rate": 4.658267172937522e-05,
      "loss": 0.4328,
      "step": 520
    },
    {
      "epoch": 0.4890544946436889,
      "grad_norm": 0.722745449359793,
      "learning_rate": 4.64963755609251e-05,
      "loss": 0.4407,
      "step": 525
    },
    {
      "epoch": 0.49371215649743827,
      "grad_norm": 0.823112622131515,
      "learning_rate": 4.641007939247497e-05,
      "loss": 0.4277,
      "step": 530
    },
    {
      "epoch": 0.4983698183511877,
      "grad_norm": 0.7724342365147061,
      "learning_rate": 4.632378322402486e-05,
      "loss": 0.4358,
      "step": 535
    },
    {
      "epoch": 0.5030274802049371,
      "grad_norm": 0.6117770443859092,
      "learning_rate": 4.6237487055574736e-05,
      "loss": 0.4378,
      "step": 540
    },
    {
      "epoch": 0.5076851420586865,
      "grad_norm": 0.8311108659999172,
      "learning_rate": 4.6151190887124615e-05,
      "loss": 0.4278,
      "step": 545
    },
    {
      "epoch": 0.5123428039124359,
      "grad_norm": 0.8129154894087233,
      "learning_rate": 4.606489471867449e-05,
      "loss": 0.4414,
      "step": 550
    },
    {
      "epoch": 0.5170004657661854,
      "grad_norm": 0.5951389969101163,
      "learning_rate": 4.597859855022437e-05,
      "loss": 0.4299,
      "step": 555
    },
    {
      "epoch": 0.5216581276199348,
      "grad_norm": 0.6867950993341094,
      "learning_rate": 4.589230238177425e-05,
      "loss": 0.4343,
      "step": 560
    },
    {
      "epoch": 0.5263157894736842,
      "grad_norm": 0.6969765247574474,
      "learning_rate": 4.5806006213324134e-05,
      "loss": 0.4326,
      "step": 565
    },
    {
      "epoch": 0.5309734513274337,
      "grad_norm": 0.700263667965796,
      "learning_rate": 4.5719710044874006e-05,
      "loss": 0.4317,
      "step": 570
    },
    {
      "epoch": 0.5356311131811831,
      "grad_norm": 0.5477903679849707,
      "learning_rate": 4.563341387642389e-05,
      "loss": 0.4288,
      "step": 575
    },
    {
      "epoch": 0.5402887750349324,
      "grad_norm": 0.4875449948646545,
      "learning_rate": 4.554711770797377e-05,
      "loss": 0.4291,
      "step": 580
    },
    {
      "epoch": 0.5449464368886818,
      "grad_norm": 0.6483883244276489,
      "learning_rate": 4.546082153952365e-05,
      "loss": 0.4255,
      "step": 585
    },
    {
      "epoch": 0.5496040987424313,
      "grad_norm": 0.5874237534879547,
      "learning_rate": 4.5374525371073526e-05,
      "loss": 0.4402,
      "step": 590
    },
    {
      "epoch": 0.5542617605961807,
      "grad_norm": 0.6738275443467228,
      "learning_rate": 4.5288229202623404e-05,
      "loss": 0.4316,
      "step": 595
    },
    {
      "epoch": 0.5589194224499301,
      "grad_norm": 0.7039615425689857,
      "learning_rate": 4.520193303417328e-05,
      "loss": 0.4366,
      "step": 600
    },
    {
      "epoch": 0.5635770843036796,
      "grad_norm": 0.6228516023637121,
      "learning_rate": 4.511563686572317e-05,
      "loss": 0.4133,
      "step": 605
    },
    {
      "epoch": 0.568234746157429,
      "grad_norm": 0.6178519124828229,
      "learning_rate": 4.5029340697273045e-05,
      "loss": 0.4311,
      "step": 610
    },
    {
      "epoch": 0.5728924080111784,
      "grad_norm": 0.6690682660152822,
      "learning_rate": 4.4943044528822923e-05,
      "loss": 0.4231,
      "step": 615
    },
    {
      "epoch": 0.5775500698649279,
      "grad_norm": 0.5773968331144422,
      "learning_rate": 4.48567483603728e-05,
      "loss": 0.4256,
      "step": 620
    },
    {
      "epoch": 0.5822077317186772,
      "grad_norm": 0.7744283063367543,
      "learning_rate": 4.477045219192268e-05,
      "loss": 0.4367,
      "step": 625
    },
    {
      "epoch": 0.5868653935724266,
      "grad_norm": 0.7491779356834112,
      "learning_rate": 4.4684156023472565e-05,
      "loss": 0.4231,
      "step": 630
    },
    {
      "epoch": 0.5915230554261761,
      "grad_norm": 0.7427532503654565,
      "learning_rate": 4.4597859855022436e-05,
      "loss": 0.4222,
      "step": 635
    },
    {
      "epoch": 0.5961807172799255,
      "grad_norm": 0.861064858619503,
      "learning_rate": 4.4511563686572315e-05,
      "loss": 0.4269,
      "step": 640
    },
    {
      "epoch": 0.6008383791336749,
      "grad_norm": 0.5637312203066887,
      "learning_rate": 4.44252675181222e-05,
      "loss": 0.4264,
      "step": 645
    },
    {
      "epoch": 0.6054960409874243,
      "grad_norm": 0.5648559548816346,
      "learning_rate": 4.433897134967208e-05,
      "loss": 0.4242,
      "step": 650
    },
    {
      "epoch": 0.6101537028411738,
      "grad_norm": 0.5669861772152599,
      "learning_rate": 4.4252675181221956e-05,
      "loss": 0.4297,
      "step": 655
    },
    {
      "epoch": 0.6148113646949231,
      "grad_norm": 0.5934050600649434,
      "learning_rate": 4.4166379012771834e-05,
      "loss": 0.4307,
      "step": 660
    },
    {
      "epoch": 0.6194690265486725,
      "grad_norm": 0.5588208620423528,
      "learning_rate": 4.408008284432171e-05,
      "loss": 0.4253,
      "step": 665
    },
    {
      "epoch": 0.624126688402422,
      "grad_norm": 0.5939836158631694,
      "learning_rate": 4.39937866758716e-05,
      "loss": 0.4283,
      "step": 670
    },
    {
      "epoch": 0.6287843502561714,
      "grad_norm": 0.5890048545951234,
      "learning_rate": 4.3907490507421476e-05,
      "loss": 0.4216,
      "step": 675
    },
    {
      "epoch": 0.6334420121099208,
      "grad_norm": 0.5315615827716382,
      "learning_rate": 4.382119433897135e-05,
      "loss": 0.425,
      "step": 680
    },
    {
      "epoch": 0.6380996739636703,
      "grad_norm": 0.5583181256253494,
      "learning_rate": 4.373489817052123e-05,
      "loss": 0.4354,
      "step": 685
    },
    {
      "epoch": 0.6427573358174197,
      "grad_norm": 0.6478305928379938,
      "learning_rate": 4.364860200207111e-05,
      "loss": 0.428,
      "step": 690
    },
    {
      "epoch": 0.6474149976711691,
      "grad_norm": 0.47607311475069264,
      "learning_rate": 4.356230583362099e-05,
      "loss": 0.4251,
      "step": 695
    },
    {
      "epoch": 0.6520726595249184,
      "grad_norm": 0.5509713279953086,
      "learning_rate": 4.347600966517087e-05,
      "loss": 0.4227,
      "step": 700
    },
    {
      "epoch": 0.6567303213786679,
      "grad_norm": 0.5584537283958807,
      "learning_rate": 4.3389713496720745e-05,
      "loss": 0.4197,
      "step": 705
    },
    {
      "epoch": 0.6613879832324173,
      "grad_norm": 0.6409629277291342,
      "learning_rate": 4.330341732827063e-05,
      "loss": 0.431,
      "step": 710
    },
    {
      "epoch": 0.6660456450861667,
      "grad_norm": 0.5683535669015193,
      "learning_rate": 4.321712115982051e-05,
      "loss": 0.4246,
      "step": 715
    },
    {
      "epoch": 0.6707033069399162,
      "grad_norm": 0.7989692157541006,
      "learning_rate": 4.3130824991370387e-05,
      "loss": 0.4162,
      "step": 720
    },
    {
      "epoch": 0.6753609687936656,
      "grad_norm": 0.6965110250543864,
      "learning_rate": 4.3044528822920265e-05,
      "loss": 0.4215,
      "step": 725
    },
    {
      "epoch": 0.680018630647415,
      "grad_norm": 0.7154488026364509,
      "learning_rate": 4.295823265447014e-05,
      "loss": 0.4279,
      "step": 730
    },
    {
      "epoch": 0.6846762925011645,
      "grad_norm": 0.5425458597574438,
      "learning_rate": 4.287193648602002e-05,
      "loss": 0.4169,
      "step": 735
    },
    {
      "epoch": 0.6893339543549138,
      "grad_norm": 0.5081622109419437,
      "learning_rate": 4.27856403175699e-05,
      "loss": 0.4214,
      "step": 740
    },
    {
      "epoch": 0.6939916162086632,
      "grad_norm": 0.6758841087317343,
      "learning_rate": 4.269934414911978e-05,
      "loss": 0.4182,
      "step": 745
    },
    {
      "epoch": 0.6986492780624126,
      "grad_norm": 0.8005270046487905,
      "learning_rate": 4.261304798066966e-05,
      "loss": 0.4251,
      "step": 750
    },
    {
      "epoch": 0.7033069399161621,
      "grad_norm": 0.570359705013777,
      "learning_rate": 4.252675181221954e-05,
      "loss": 0.4144,
      "step": 755
    },
    {
      "epoch": 0.7079646017699115,
      "grad_norm": 0.6205150187992302,
      "learning_rate": 4.244045564376942e-05,
      "loss": 0.4178,
      "step": 760
    },
    {
      "epoch": 0.7126222636236609,
      "grad_norm": 0.5081297336451379,
      "learning_rate": 4.23541594753193e-05,
      "loss": 0.4247,
      "step": 765
    },
    {
      "epoch": 0.7172799254774104,
      "grad_norm": 0.6217057259691577,
      "learning_rate": 4.2267863306869176e-05,
      "loss": 0.4246,
      "step": 770
    },
    {
      "epoch": 0.7219375873311598,
      "grad_norm": 0.5708472630777117,
      "learning_rate": 4.2181567138419054e-05,
      "loss": 0.4137,
      "step": 775
    },
    {
      "epoch": 0.7265952491849091,
      "grad_norm": 0.8064115063252781,
      "learning_rate": 4.209527096996894e-05,
      "loss": 0.429,
      "step": 780
    },
    {
      "epoch": 0.7312529110386586,
      "grad_norm": 0.5632862119043345,
      "learning_rate": 4.200897480151881e-05,
      "loss": 0.4224,
      "step": 785
    },
    {
      "epoch": 0.735910572892408,
      "grad_norm": 0.5355564887071951,
      "learning_rate": 4.1922678633068695e-05,
      "loss": 0.4157,
      "step": 790
    },
    {
      "epoch": 0.7405682347461574,
      "grad_norm": 0.5768074805294789,
      "learning_rate": 4.1836382464618573e-05,
      "loss": 0.422,
      "step": 795
    },
    {
      "epoch": 0.7452258965999069,
      "grad_norm": 0.4812749383008575,
      "learning_rate": 4.175008629616845e-05,
      "loss": 0.4179,
      "step": 800
    },
    {
      "epoch": 0.7498835584536563,
      "grad_norm": 0.5339846899941171,
      "learning_rate": 4.166379012771833e-05,
      "loss": 0.432,
      "step": 805
    },
    {
      "epoch": 0.7545412203074057,
      "grad_norm": 0.5623576201453402,
      "learning_rate": 4.157749395926821e-05,
      "loss": 0.4177,
      "step": 810
    },
    {
      "epoch": 0.759198882161155,
      "grad_norm": 0.620607656447653,
      "learning_rate": 4.1491197790818086e-05,
      "loss": 0.421,
      "step": 815
    },
    {
      "epoch": 0.7638565440149045,
      "grad_norm": 0.5231873298014822,
      "learning_rate": 4.140490162236797e-05,
      "loss": 0.4208,
      "step": 820
    },
    {
      "epoch": 0.7685142058686539,
      "grad_norm": 0.649270360189294,
      "learning_rate": 4.131860545391785e-05,
      "loss": 0.4102,
      "step": 825
    },
    {
      "epoch": 0.7731718677224033,
      "grad_norm": 0.640431306191139,
      "learning_rate": 4.123230928546773e-05,
      "loss": 0.4174,
      "step": 830
    },
    {
      "epoch": 0.7778295295761528,
      "grad_norm": 0.5051949815103178,
      "learning_rate": 4.1146013117017606e-05,
      "loss": 0.4246,
      "step": 835
    },
    {
      "epoch": 0.7824871914299022,
      "grad_norm": 0.5694239827330236,
      "learning_rate": 4.1059716948567484e-05,
      "loss": 0.4115,
      "step": 840
    },
    {
      "epoch": 0.7871448532836516,
      "grad_norm": 0.5858340369438,
      "learning_rate": 4.097342078011737e-05,
      "loss": 0.4159,
      "step": 845
    },
    {
      "epoch": 0.7918025151374011,
      "grad_norm": 0.6321382409367989,
      "learning_rate": 4.088712461166724e-05,
      "loss": 0.4181,
      "step": 850
    },
    {
      "epoch": 0.7964601769911505,
      "grad_norm": 0.5833020455213448,
      "learning_rate": 4.080082844321712e-05,
      "loss": 0.4101,
      "step": 855
    },
    {
      "epoch": 0.8011178388448998,
      "grad_norm": 0.7204608198037822,
      "learning_rate": 4.0714532274767004e-05,
      "loss": 0.4122,
      "step": 860
    },
    {
      "epoch": 0.8057755006986492,
      "grad_norm": 0.6437036434401001,
      "learning_rate": 4.062823610631688e-05,
      "loss": 0.4151,
      "step": 865
    },
    {
      "epoch": 0.8104331625523987,
      "grad_norm": 0.5904495022769027,
      "learning_rate": 4.054193993786676e-05,
      "loss": 0.4101,
      "step": 870
    },
    {
      "epoch": 0.8150908244061481,
      "grad_norm": 0.412201552171827,
      "learning_rate": 4.045564376941664e-05,
      "loss": 0.4192,
      "step": 875
    },
    {
      "epoch": 0.8197484862598975,
      "grad_norm": 0.42840467551296285,
      "learning_rate": 4.036934760096652e-05,
      "loss": 0.4043,
      "step": 880
    },
    {
      "epoch": 0.824406148113647,
      "grad_norm": 0.47848458612166145,
      "learning_rate": 4.02830514325164e-05,
      "loss": 0.4185,
      "step": 885
    },
    {
      "epoch": 0.8290638099673964,
      "grad_norm": 0.543847208383862,
      "learning_rate": 4.019675526406628e-05,
      "loss": 0.4169,
      "step": 890
    },
    {
      "epoch": 0.8337214718211458,
      "grad_norm": 0.4456297337432753,
      "learning_rate": 4.011045909561615e-05,
      "loss": 0.4182,
      "step": 895
    },
    {
      "epoch": 0.8383791336748952,
      "grad_norm": 0.535792214813578,
      "learning_rate": 4.0024162927166037e-05,
      "loss": 0.4125,
      "step": 900
    },
    {
      "epoch": 0.8430367955286446,
      "grad_norm": 0.5756699800837073,
      "learning_rate": 3.9937866758715915e-05,
      "loss": 0.4158,
      "step": 905
    },
    {
      "epoch": 0.847694457382394,
      "grad_norm": 0.535828537078377,
      "learning_rate": 3.98515705902658e-05,
      "loss": 0.4219,
      "step": 910
    },
    {
      "epoch": 0.8523521192361434,
      "grad_norm": 0.605677842689681,
      "learning_rate": 3.976527442181567e-05,
      "loss": 0.4143,
      "step": 915
    },
    {
      "epoch": 0.8570097810898929,
      "grad_norm": 0.5792731465851377,
      "learning_rate": 3.967897825336555e-05,
      "loss": 0.4212,
      "step": 920
    },
    {
      "epoch": 0.8616674429436423,
      "grad_norm": 0.49685236608393574,
      "learning_rate": 3.9592682084915434e-05,
      "loss": 0.4162,
      "step": 925
    },
    {
      "epoch": 0.8663251047973917,
      "grad_norm": 0.6603543063372369,
      "learning_rate": 3.950638591646531e-05,
      "loss": 0.4166,
      "step": 930
    },
    {
      "epoch": 0.8709827666511412,
      "grad_norm": 0.48412804279404464,
      "learning_rate": 3.942008974801519e-05,
      "loss": 0.4056,
      "step": 935
    },
    {
      "epoch": 0.8756404285048905,
      "grad_norm": 0.5075168163461252,
      "learning_rate": 3.933379357956507e-05,
      "loss": 0.4058,
      "step": 940
    },
    {
      "epoch": 0.8802980903586399,
      "grad_norm": 0.481651712881544,
      "learning_rate": 3.924749741111495e-05,
      "loss": 0.4096,
      "step": 945
    },
    {
      "epoch": 0.8849557522123894,
      "grad_norm": 0.5368050759753034,
      "learning_rate": 3.916120124266483e-05,
      "loss": 0.4005,
      "step": 950
    },
    {
      "epoch": 0.8896134140661388,
      "grad_norm": 0.5357951160484261,
      "learning_rate": 3.9074905074214704e-05,
      "loss": 0.4124,
      "step": 955
    },
    {
      "epoch": 0.8942710759198882,
      "grad_norm": 0.5043673748190443,
      "learning_rate": 3.898860890576458e-05,
      "loss": 0.4171,
      "step": 960
    },
    {
      "epoch": 0.8989287377736377,
      "grad_norm": 0.47072170917605627,
      "learning_rate": 3.890231273731447e-05,
      "loss": 0.4125,
      "step": 965
    },
    {
      "epoch": 0.9035863996273871,
      "grad_norm": 0.48784910056445,
      "learning_rate": 3.8816016568864345e-05,
      "loss": 0.4152,
      "step": 970
    },
    {
      "epoch": 0.9082440614811365,
      "grad_norm": 0.5851346168596476,
      "learning_rate": 3.8729720400414224e-05,
      "loss": 0.4099,
      "step": 975
    },
    {
      "epoch": 0.9129017233348858,
      "grad_norm": 0.6667368256711216,
      "learning_rate": 3.86434242319641e-05,
      "loss": 0.4128,
      "step": 980
    },
    {
      "epoch": 0.9175593851886353,
      "grad_norm": 0.6415929137545101,
      "learning_rate": 3.855712806351398e-05,
      "loss": 0.4177,
      "step": 985
    },
    {
      "epoch": 0.9222170470423847,
      "grad_norm": 0.5636304079043989,
      "learning_rate": 3.8470831895063865e-05,
      "loss": 0.3915,
      "step": 990
    },
    {
      "epoch": 0.9268747088961341,
      "grad_norm": 0.6112170530214519,
      "learning_rate": 3.838453572661374e-05,
      "loss": 0.4067,
      "step": 995
    },
    {
      "epoch": 0.9315323707498836,
      "grad_norm": 0.5502514153806973,
      "learning_rate": 3.8298239558163615e-05,
      "loss": 0.409,
      "step": 1000
    },
    {
      "epoch": 0.936190032603633,
      "grad_norm": 0.5967054768146283,
      "learning_rate": 3.82119433897135e-05,
      "loss": 0.4171,
      "step": 1005
    },
    {
      "epoch": 0.9408476944573824,
      "grad_norm": 0.45292831485300017,
      "learning_rate": 3.812564722126338e-05,
      "loss": 0.4065,
      "step": 1010
    },
    {
      "epoch": 0.9455053563111319,
      "grad_norm": 0.4552189648016451,
      "learning_rate": 3.8039351052813256e-05,
      "loss": 0.4086,
      "step": 1015
    },
    {
      "epoch": 0.9501630181648812,
      "grad_norm": 0.6316248598494597,
      "learning_rate": 3.7953054884363134e-05,
      "loss": 0.4171,
      "step": 1020
    },
    {
      "epoch": 0.9548206800186306,
      "grad_norm": 0.6115651135326055,
      "learning_rate": 3.786675871591301e-05,
      "loss": 0.4099,
      "step": 1025
    },
    {
      "epoch": 0.95947834187238,
      "grad_norm": 0.6273296160466062,
      "learning_rate": 3.77804625474629e-05,
      "loss": 0.4037,
      "step": 1030
    },
    {
      "epoch": 0.9641360037261295,
      "grad_norm": 0.4323296848465666,
      "learning_rate": 3.7694166379012776e-05,
      "loss": 0.4054,
      "step": 1035
    },
    {
      "epoch": 0.9687936655798789,
      "grad_norm": 0.6687111408493186,
      "learning_rate": 3.7607870210562654e-05,
      "loss": 0.4052,
      "step": 1040
    },
    {
      "epoch": 0.9734513274336283,
      "grad_norm": 0.4522927122320442,
      "learning_rate": 3.752157404211253e-05,
      "loss": 0.4057,
      "step": 1045
    },
    {
      "epoch": 0.9781089892873778,
      "grad_norm": 0.48769431997823864,
      "learning_rate": 3.743527787366241e-05,
      "loss": 0.4058,
      "step": 1050
    },
    {
      "epoch": 0.9827666511411272,
      "grad_norm": 0.6528810815132017,
      "learning_rate": 3.734898170521229e-05,
      "loss": 0.407,
      "step": 1055
    },
    {
      "epoch": 0.9874243129948765,
      "grad_norm": 0.5511171393125437,
      "learning_rate": 3.7262685536762174e-05,
      "loss": 0.4101,
      "step": 1060
    },
    {
      "epoch": 0.992081974848626,
      "grad_norm": 0.5204909377873204,
      "learning_rate": 3.7176389368312045e-05,
      "loss": 0.4069,
      "step": 1065
    },
    {
      "epoch": 0.9967396367023754,
      "grad_norm": 0.49724718921393213,
      "learning_rate": 3.709009319986193e-05,
      "loss": 0.3995,
      "step": 1070
    },
    {
      "epoch": 1.00093153237075,
      "grad_norm": 1.131703398299972,
      "learning_rate": 3.700379703141181e-05,
      "loss": 0.4096,
      "step": 1075
    },
    {
      "epoch": 1.0055891942244992,
      "grad_norm": 0.6419695512999521,
      "learning_rate": 3.6917500862961687e-05,
      "loss": 0.3525,
      "step": 1080
    },
    {
      "epoch": 1.0102468560782487,
      "grad_norm": 0.6016183638225807,
      "learning_rate": 3.6831204694511565e-05,
      "loss": 0.3416,
      "step": 1085
    },
    {
      "epoch": 1.0149045179319982,
      "grad_norm": 0.5918785782900216,
      "learning_rate": 3.674490852606144e-05,
      "loss": 0.3531,
      "step": 1090
    },
    {
      "epoch": 1.0195621797857475,
      "grad_norm": 0.5125644994191116,
      "learning_rate": 3.665861235761132e-05,
      "loss": 0.3494,
      "step": 1095
    },
    {
      "epoch": 1.024219841639497,
      "grad_norm": 0.4691917520614703,
      "learning_rate": 3.6572316189161206e-05,
      "loss": 0.3531,
      "step": 1100
    },
    {
      "epoch": 1.0288775034932465,
      "grad_norm": 0.5685444672610303,
      "learning_rate": 3.6486020020711085e-05,
      "loss": 0.3525,
      "step": 1105
    },
    {
      "epoch": 1.0335351653469957,
      "grad_norm": 0.5066529575440402,
      "learning_rate": 3.639972385226096e-05,
      "loss": 0.3525,
      "step": 1110
    },
    {
      "epoch": 1.0381928272007452,
      "grad_norm": 0.452084412510361,
      "learning_rate": 3.631342768381084e-05,
      "loss": 0.3432,
      "step": 1115
    },
    {
      "epoch": 1.0428504890544947,
      "grad_norm": 0.4653653890887597,
      "learning_rate": 3.622713151536072e-05,
      "loss": 0.3504,
      "step": 1120
    },
    {
      "epoch": 1.047508150908244,
      "grad_norm": 0.5786524786643038,
      "learning_rate": 3.6140835346910604e-05,
      "loss": 0.3566,
      "step": 1125
    },
    {
      "epoch": 1.0521658127619935,
      "grad_norm": 0.5536971348361633,
      "learning_rate": 3.6054539178460476e-05,
      "loss": 0.3415,
      "step": 1130
    },
    {
      "epoch": 1.056823474615743,
      "grad_norm": 0.47981725569345507,
      "learning_rate": 3.5968243010010354e-05,
      "loss": 0.3508,
      "step": 1135
    },
    {
      "epoch": 1.0614811364694923,
      "grad_norm": 0.5213789577783958,
      "learning_rate": 3.588194684156024e-05,
      "loss": 0.3542,
      "step": 1140
    },
    {
      "epoch": 1.0661387983232418,
      "grad_norm": 0.5440401632942263,
      "learning_rate": 3.579565067311012e-05,
      "loss": 0.3511,
      "step": 1145
    },
    {
      "epoch": 1.0707964601769913,
      "grad_norm": 0.4577451631796886,
      "learning_rate": 3.5709354504659995e-05,
      "loss": 0.3477,
      "step": 1150
    },
    {
      "epoch": 1.0754541220307405,
      "grad_norm": 0.542060009328887,
      "learning_rate": 3.5623058336209874e-05,
      "loss": 0.3513,
      "step": 1155
    },
    {
      "epoch": 1.08011178388449,
      "grad_norm": 0.6079326524813553,
      "learning_rate": 3.553676216775975e-05,
      "loss": 0.3582,
      "step": 1160
    },
    {
      "epoch": 1.0847694457382393,
      "grad_norm": 0.44794405329162973,
      "learning_rate": 3.545046599930964e-05,
      "loss": 0.3575,
      "step": 1165
    },
    {
      "epoch": 1.0894271075919888,
      "grad_norm": 0.45537754475901826,
      "learning_rate": 3.536416983085951e-05,
      "loss": 0.3549,
      "step": 1170
    },
    {
      "epoch": 1.0940847694457383,
      "grad_norm": 0.4291211671434693,
      "learning_rate": 3.5277873662409386e-05,
      "loss": 0.355,
      "step": 1175
    },
    {
      "epoch": 1.0987424312994876,
      "grad_norm": 0.487488634463869,
      "learning_rate": 3.519157749395927e-05,
      "loss": 0.3478,
      "step": 1180
    },
    {
      "epoch": 1.103400093153237,
      "grad_norm": 0.520950478977621,
      "learning_rate": 3.510528132550915e-05,
      "loss": 0.3492,
      "step": 1185
    },
    {
      "epoch": 1.1080577550069866,
      "grad_norm": 0.453710115001766,
      "learning_rate": 3.501898515705903e-05,
      "loss": 0.3319,
      "step": 1190
    },
    {
      "epoch": 1.1127154168607358,
      "grad_norm": 0.4987043498433735,
      "learning_rate": 3.4932688988608906e-05,
      "loss": 0.3413,
      "step": 1195
    },
    {
      "epoch": 1.1173730787144853,
      "grad_norm": 0.44029431944826875,
      "learning_rate": 3.4846392820158784e-05,
      "loss": 0.3485,
      "step": 1200
    },
    {
      "epoch": 1.1220307405682348,
      "grad_norm": 0.5042207045263483,
      "learning_rate": 3.476009665170867e-05,
      "loss": 0.357,
      "step": 1205
    },
    {
      "epoch": 1.126688402421984,
      "grad_norm": 0.443472956564033,
      "learning_rate": 3.467380048325855e-05,
      "loss": 0.354,
      "step": 1210
    },
    {
      "epoch": 1.1313460642757336,
      "grad_norm": 0.5492187609238193,
      "learning_rate": 3.458750431480842e-05,
      "loss": 0.3475,
      "step": 1215
    },
    {
      "epoch": 1.136003726129483,
      "grad_norm": 0.5718077527663505,
      "learning_rate": 3.4501208146358304e-05,
      "loss": 0.3568,
      "step": 1220
    },
    {
      "epoch": 1.1406613879832324,
      "grad_norm": 0.5159695298904204,
      "learning_rate": 3.441491197790818e-05,
      "loss": 0.3481,
      "step": 1225
    },
    {
      "epoch": 1.1453190498369819,
      "grad_norm": 0.6499550549648573,
      "learning_rate": 3.432861580945806e-05,
      "loss": 0.3504,
      "step": 1230
    },
    {
      "epoch": 1.1499767116907313,
      "grad_norm": 0.6360582483892888,
      "learning_rate": 3.424231964100794e-05,
      "loss": 0.3496,
      "step": 1235
    },
    {
      "epoch": 1.1546343735444806,
      "grad_norm": 0.4804246521892999,
      "learning_rate": 3.415602347255782e-05,
      "loss": 0.3543,
      "step": 1240
    },
    {
      "epoch": 1.1592920353982301,
      "grad_norm": 0.5416856509793349,
      "learning_rate": 3.40697273041077e-05,
      "loss": 0.3564,
      "step": 1245
    },
    {
      "epoch": 1.1639496972519794,
      "grad_norm": 0.5090896448452167,
      "learning_rate": 3.398343113565758e-05,
      "loss": 0.3588,
      "step": 1250
    },
    {
      "epoch": 1.1686073591057289,
      "grad_norm": 0.4433710508365059,
      "learning_rate": 3.389713496720746e-05,
      "loss": 0.3473,
      "step": 1255
    },
    {
      "epoch": 1.1732650209594784,
      "grad_norm": 0.43450996960215044,
      "learning_rate": 3.381083879875734e-05,
      "loss": 0.3579,
      "step": 1260
    },
    {
      "epoch": 1.1779226828132279,
      "grad_norm": 0.4639240544702146,
      "learning_rate": 3.3724542630307215e-05,
      "loss": 0.3455,
      "step": 1265
    },
    {
      "epoch": 1.1825803446669771,
      "grad_norm": 0.5168471960349775,
      "learning_rate": 3.363824646185709e-05,
      "loss": 0.3393,
      "step": 1270
    },
    {
      "epoch": 1.1872380065207266,
      "grad_norm": 0.5180715062017991,
      "learning_rate": 3.355195029340698e-05,
      "loss": 0.3509,
      "step": 1275
    },
    {
      "epoch": 1.191895668374476,
      "grad_norm": 0.6970312389377342,
      "learning_rate": 3.346565412495685e-05,
      "loss": 0.3538,
      "step": 1280
    },
    {
      "epoch": 1.1965533302282254,
      "grad_norm": 0.5101683277886183,
      "learning_rate": 3.3379357956506735e-05,
      "loss": 0.357,
      "step": 1285
    },
    {
      "epoch": 1.201210992081975,
      "grad_norm": 0.5101777824218828,
      "learning_rate": 3.329306178805661e-05,
      "loss": 0.3475,
      "step": 1290
    },
    {
      "epoch": 1.2058686539357242,
      "grad_norm": 0.4726689138411272,
      "learning_rate": 3.320676561960649e-05,
      "loss": 0.3511,
      "step": 1295
    },
    {
      "epoch": 1.2105263157894737,
      "grad_norm": 0.42018375632264665,
      "learning_rate": 3.312046945115637e-05,
      "loss": 0.3474,
      "step": 1300
    },
    {
      "epoch": 1.2151839776432232,
      "grad_norm": 0.43226827668944096,
      "learning_rate": 3.303417328270625e-05,
      "loss": 0.356,
      "step": 1305
    },
    {
      "epoch": 1.2198416394969724,
      "grad_norm": 0.43683104405759693,
      "learning_rate": 3.2947877114256126e-05,
      "loss": 0.3591,
      "step": 1310
    },
    {
      "epoch": 1.224499301350722,
      "grad_norm": 0.42555685283169953,
      "learning_rate": 3.286158094580601e-05,
      "loss": 0.3483,
      "step": 1315
    },
    {
      "epoch": 1.2291569632044714,
      "grad_norm": 0.416537069580912,
      "learning_rate": 3.277528477735589e-05,
      "loss": 0.3581,
      "step": 1320
    },
    {
      "epoch": 1.2338146250582207,
      "grad_norm": 0.41861673041456404,
      "learning_rate": 3.268898860890577e-05,
      "loss": 0.3506,
      "step": 1325
    },
    {
      "epoch": 1.2384722869119702,
      "grad_norm": 0.46889652427168527,
      "learning_rate": 3.2602692440455645e-05,
      "loss": 0.3561,
      "step": 1330
    },
    {
      "epoch": 1.2431299487657197,
      "grad_norm": 0.41067032917021723,
      "learning_rate": 3.2516396272005524e-05,
      "loss": 0.3486,
      "step": 1335
    },
    {
      "epoch": 1.247787610619469,
      "grad_norm": 0.5164136524967469,
      "learning_rate": 3.243010010355541e-05,
      "loss": 0.3506,
      "step": 1340
    },
    {
      "epoch": 1.2524452724732185,
      "grad_norm": 0.4091952474632887,
      "learning_rate": 3.234380393510528e-05,
      "loss": 0.3407,
      "step": 1345
    },
    {
      "epoch": 1.257102934326968,
      "grad_norm": 0.4909326039638265,
      "learning_rate": 3.225750776665516e-05,
      "loss": 0.3501,
      "step": 1350
    },
    {
      "epoch": 1.2617605961807172,
      "grad_norm": 0.5940284609530456,
      "learning_rate": 3.217121159820504e-05,
      "loss": 0.3467,
      "step": 1355
    },
    {
      "epoch": 1.2664182580344667,
      "grad_norm": 0.5669193907150314,
      "learning_rate": 3.208491542975492e-05,
      "loss": 0.3402,
      "step": 1360
    },
    {
      "epoch": 1.271075919888216,
      "grad_norm": 0.3941027761579512,
      "learning_rate": 3.19986192613048e-05,
      "loss": 0.3526,
      "step": 1365
    },
    {
      "epoch": 1.2757335817419655,
      "grad_norm": 0.5050588295544981,
      "learning_rate": 3.191232309285468e-05,
      "loss": 0.3502,
      "step": 1370
    },
    {
      "epoch": 1.280391243595715,
      "grad_norm": 0.4895886734894624,
      "learning_rate": 3.1826026924404556e-05,
      "loss": 0.3499,
      "step": 1375
    },
    {
      "epoch": 1.2850489054494645,
      "grad_norm": 0.48848047559261704,
      "learning_rate": 3.173973075595444e-05,
      "loss": 0.3497,
      "step": 1380
    },
    {
      "epoch": 1.2897065673032138,
      "grad_norm": 0.4165587299892872,
      "learning_rate": 3.165343458750431e-05,
      "loss": 0.3485,
      "step": 1385
    },
    {
      "epoch": 1.2943642291569633,
      "grad_norm": 0.4532938944634796,
      "learning_rate": 3.156713841905419e-05,
      "loss": 0.3563,
      "step": 1390
    },
    {
      "epoch": 1.2990218910107125,
      "grad_norm": 0.511849836322528,
      "learning_rate": 3.1480842250604076e-05,
      "loss": 0.3519,
      "step": 1395
    },
    {
      "epoch": 1.303679552864462,
      "grad_norm": 0.47351344444130267,
      "learning_rate": 3.1394546082153954e-05,
      "loss": 0.353,
      "step": 1400
    },
    {
      "epoch": 1.3083372147182115,
      "grad_norm": 0.43370189707506596,
      "learning_rate": 3.130824991370383e-05,
      "loss": 0.3462,
      "step": 1405
    },
    {
      "epoch": 1.312994876571961,
      "grad_norm": 0.4130583655164329,
      "learning_rate": 3.122195374525371e-05,
      "loss": 0.3453,
      "step": 1410
    },
    {
      "epoch": 1.3176525384257103,
      "grad_norm": 0.49082139605405356,
      "learning_rate": 3.113565757680359e-05,
      "loss": 0.3628,
      "step": 1415
    },
    {
      "epoch": 1.3223102002794598,
      "grad_norm": 0.4144787632420094,
      "learning_rate": 3.1049361408353474e-05,
      "loss": 0.3522,
      "step": 1420
    },
    {
      "epoch": 1.326967862133209,
      "grad_norm": 0.4349112186535909,
      "learning_rate": 3.096306523990335e-05,
      "loss": 0.3457,
      "step": 1425
    },
    {
      "epoch": 1.3316255239869585,
      "grad_norm": 0.4391572856497664,
      "learning_rate": 3.0876769071453223e-05,
      "loss": 0.3464,
      "step": 1430
    },
    {
      "epoch": 1.336283185840708,
      "grad_norm": 0.48123814696594097,
      "learning_rate": 3.079047290300311e-05,
      "loss": 0.3509,
      "step": 1435
    },
    {
      "epoch": 1.3409408476944573,
      "grad_norm": 0.48258038471952563,
      "learning_rate": 3.070417673455299e-05,
      "loss": 0.3499,
      "step": 1440
    },
    {
      "epoch": 1.3455985095482068,
      "grad_norm": 0.5327330788210805,
      "learning_rate": 3.061788056610287e-05,
      "loss": 0.3562,
      "step": 1445
    },
    {
      "epoch": 1.350256171401956,
      "grad_norm": 0.48425989112639334,
      "learning_rate": 3.053158439765274e-05,
      "loss": 0.3468,
      "step": 1450
    },
    {
      "epoch": 1.3549138332557056,
      "grad_norm": 0.42305323942857453,
      "learning_rate": 3.0445288229202625e-05,
      "loss": 0.3549,
      "step": 1455
    },
    {
      "epoch": 1.359571495109455,
      "grad_norm": 0.4058276237036279,
      "learning_rate": 3.0358992060752506e-05,
      "loss": 0.3387,
      "step": 1460
    },
    {
      "epoch": 1.3642291569632046,
      "grad_norm": 0.43968582707314,
      "learning_rate": 3.027269589230238e-05,
      "loss": 0.3459,
      "step": 1465
    },
    {
      "epoch": 1.3688868188169538,
      "grad_norm": 0.41508600875593327,
      "learning_rate": 3.018639972385226e-05,
      "loss": 0.3511,
      "step": 1470
    },
    {
      "epoch": 1.3735444806707033,
      "grad_norm": 0.4782627959963521,
      "learning_rate": 3.010010355540214e-05,
      "loss": 0.3444,
      "step": 1475
    },
    {
      "epoch": 1.3782021425244526,
      "grad_norm": 0.462177343042049,
      "learning_rate": 3.001380738695202e-05,
      "loss": 0.3469,
      "step": 1480
    },
    {
      "epoch": 1.382859804378202,
      "grad_norm": 0.5113846569287512,
      "learning_rate": 2.99275112185019e-05,
      "loss": 0.3523,
      "step": 1485
    },
    {
      "epoch": 1.3875174662319516,
      "grad_norm": 0.5490955228282127,
      "learning_rate": 2.984121505005178e-05,
      "loss": 0.3529,
      "step": 1490
    },
    {
      "epoch": 1.392175128085701,
      "grad_norm": 0.4760654179286793,
      "learning_rate": 2.9754918881601657e-05,
      "loss": 0.3529,
      "step": 1495
    },
    {
      "epoch": 1.3968327899394504,
      "grad_norm": 0.4872502642584581,
      "learning_rate": 2.966862271315154e-05,
      "loss": 0.3426,
      "step": 1500
    },
    {
      "epoch": 1.4014904517931999,
      "grad_norm": 0.40483751856604994,
      "learning_rate": 2.9582326544701417e-05,
      "loss": 0.3467,
      "step": 1505
    },
    {
      "epoch": 1.4061481136469491,
      "grad_norm": 0.4016809442001425,
      "learning_rate": 2.9496030376251292e-05,
      "loss": 0.3464,
      "step": 1510
    },
    {
      "epoch": 1.4108057755006986,
      "grad_norm": 0.464706786652168,
      "learning_rate": 2.9409734207801177e-05,
      "loss": 0.355,
      "step": 1515
    },
    {
      "epoch": 1.4154634373544481,
      "grad_norm": 0.4821581719631585,
      "learning_rate": 2.9323438039351052e-05,
      "loss": 0.3479,
      "step": 1520
    },
    {
      "epoch": 1.4201210992081974,
      "grad_norm": 0.5301152811790797,
      "learning_rate": 2.9237141870900937e-05,
      "loss": 0.3485,
      "step": 1525
    },
    {
      "epoch": 1.424778761061947,
      "grad_norm": 0.4932554713546422,
      "learning_rate": 2.9150845702450812e-05,
      "loss": 0.3401,
      "step": 1530
    },
    {
      "epoch": 1.4294364229156964,
      "grad_norm": 0.4857178590814053,
      "learning_rate": 2.906454953400069e-05,
      "loss": 0.3542,
      "step": 1535
    },
    {
      "epoch": 1.4340940847694457,
      "grad_norm": 0.4365051393877098,
      "learning_rate": 2.897825336555057e-05,
      "loss": 0.3462,
      "step": 1540
    },
    {
      "epoch": 1.4387517466231952,
      "grad_norm": 0.42282766872195476,
      "learning_rate": 2.889195719710045e-05,
      "loss": 0.3462,
      "step": 1545
    },
    {
      "epoch": 1.4434094084769447,
      "grad_norm": 0.40447709825837574,
      "learning_rate": 2.8805661028650328e-05,
      "loss": 0.3421,
      "step": 1550
    },
    {
      "epoch": 1.448067070330694,
      "grad_norm": 0.4512617659792673,
      "learning_rate": 2.871936486020021e-05,
      "loss": 0.3388,
      "step": 1555
    },
    {
      "epoch": 1.4527247321844434,
      "grad_norm": 0.39153759940705185,
      "learning_rate": 2.8633068691750088e-05,
      "loss": 0.346,
      "step": 1560
    },
    {
      "epoch": 1.4573823940381927,
      "grad_norm": 0.37936788170146635,
      "learning_rate": 2.854677252329997e-05,
      "loss": 0.353,
      "step": 1565
    },
    {
      "epoch": 1.4620400558919422,
      "grad_norm": 0.49562456539376953,
      "learning_rate": 2.8460476354849848e-05,
      "loss": 0.3641,
      "step": 1570
    },
    {
      "epoch": 1.4666977177456917,
      "grad_norm": 0.416899735961538,
      "learning_rate": 2.8374180186399723e-05,
      "loss": 0.3472,
      "step": 1575
    },
    {
      "epoch": 1.4713553795994412,
      "grad_norm": 0.594578480259912,
      "learning_rate": 2.8287884017949608e-05,
      "loss": 0.3397,
      "step": 1580
    },
    {
      "epoch": 1.4760130414531905,
      "grad_norm": 0.45739531092618846,
      "learning_rate": 2.8201587849499482e-05,
      "loss": 0.3477,
      "step": 1585
    },
    {
      "epoch": 1.48067070330694,
      "grad_norm": 0.555713798496605,
      "learning_rate": 2.811529168104936e-05,
      "loss": 0.3542,
      "step": 1590
    },
    {
      "epoch": 1.4853283651606892,
      "grad_norm": 0.4644143686984169,
      "learning_rate": 2.8028995512599242e-05,
      "loss": 0.3457,
      "step": 1595
    },
    {
      "epoch": 1.4899860270144387,
      "grad_norm": 0.4849873366929188,
      "learning_rate": 2.794269934414912e-05,
      "loss": 0.3506,
      "step": 1600
    },
    {
      "epoch": 1.4946436888681882,
      "grad_norm": 0.4058348111898449,
      "learning_rate": 2.7856403175699002e-05,
      "loss": 0.3429,
      "step": 1605
    },
    {
      "epoch": 1.4993013507219377,
      "grad_norm": 0.3903173871248913,
      "learning_rate": 2.777010700724888e-05,
      "loss": 0.3487,
      "step": 1610
    },
    {
      "epoch": 1.503959012575687,
      "grad_norm": 0.5111793997329473,
      "learning_rate": 2.768381083879876e-05,
      "loss": 0.3464,
      "step": 1615
    },
    {
      "epoch": 1.5086166744294365,
      "grad_norm": 0.43866836249196767,
      "learning_rate": 2.759751467034864e-05,
      "loss": 0.3485,
      "step": 1620
    },
    {
      "epoch": 1.5132743362831858,
      "grad_norm": 0.3747800542569933,
      "learning_rate": 2.751121850189852e-05,
      "loss": 0.352,
      "step": 1625
    },
    {
      "epoch": 1.5179319981369352,
      "grad_norm": 0.4297375639570572,
      "learning_rate": 2.7424922333448393e-05,
      "loss": 0.3487,
      "step": 1630
    },
    {
      "epoch": 1.5225896599906847,
      "grad_norm": 0.42889903781807254,
      "learning_rate": 2.7338626164998278e-05,
      "loss": 0.3582,
      "step": 1635
    },
    {
      "epoch": 1.5272473218444342,
      "grad_norm": 0.3914004420799744,
      "learning_rate": 2.7252329996548153e-05,
      "loss": 0.3523,
      "step": 1640
    },
    {
      "epoch": 1.5319049836981835,
      "grad_norm": 0.5155401559836418,
      "learning_rate": 2.7166033828098038e-05,
      "loss": 0.3554,
      "step": 1645
    },
    {
      "epoch": 1.5365626455519328,
      "grad_norm": 0.47481508502237424,
      "learning_rate": 2.7079737659647913e-05,
      "loss": 0.3406,
      "step": 1650
    },
    {
      "epoch": 1.5412203074056823,
      "grad_norm": 0.4724432537992889,
      "learning_rate": 2.699344149119779e-05,
      "loss": 0.3398,
      "step": 1655
    },
    {
      "epoch": 1.5458779692594318,
      "grad_norm": 0.43599422062090903,
      "learning_rate": 2.6907145322747673e-05,
      "loss": 0.3391,
      "step": 1660
    },
    {
      "epoch": 1.5505356311131813,
      "grad_norm": 0.4815350541934423,
      "learning_rate": 2.682084915429755e-05,
      "loss": 0.3515,
      "step": 1665
    },
    {
      "epoch": 1.5551932929669308,
      "grad_norm": 0.48670023690704234,
      "learning_rate": 2.673455298584743e-05,
      "loss": 0.3479,
      "step": 1670
    },
    {
      "epoch": 1.55985095482068,
      "grad_norm": 0.381486743975448,
      "learning_rate": 2.664825681739731e-05,
      "loss": 0.3455,
      "step": 1675
    },
    {
      "epoch": 1.5645086166744293,
      "grad_norm": 0.4811679115039236,
      "learning_rate": 2.6561960648947186e-05,
      "loss": 0.3525,
      "step": 1680
    },
    {
      "epoch": 1.5691662785281788,
      "grad_norm": 0.4390745529378594,
      "learning_rate": 2.647566448049707e-05,
      "loss": 0.3415,
      "step": 1685
    },
    {
      "epoch": 1.5738239403819283,
      "grad_norm": 0.4727376631513007,
      "learning_rate": 2.6389368312046945e-05,
      "loss": 0.3499,
      "step": 1690
    },
    {
      "epoch": 1.5784816022356778,
      "grad_norm": 0.42822721307610356,
      "learning_rate": 2.6303072143596824e-05,
      "loss": 0.346,
      "step": 1695
    },
    {
      "epoch": 1.583139264089427,
      "grad_norm": 0.44700207802428643,
      "learning_rate": 2.6216775975146705e-05,
      "loss": 0.3488,
      "step": 1700
    },
    {
      "epoch": 1.5877969259431766,
      "grad_norm": 0.40110082653386936,
      "learning_rate": 2.6130479806696584e-05,
      "loss": 0.3436,
      "step": 1705
    },
    {
      "epoch": 1.5924545877969258,
      "grad_norm": 0.4343846624345008,
      "learning_rate": 2.6044183638246462e-05,
      "loss": 0.3509,
      "step": 1710
    },
    {
      "epoch": 1.5971122496506753,
      "grad_norm": 0.43834253689890645,
      "learning_rate": 2.5957887469796343e-05,
      "loss": 0.3523,
      "step": 1715
    },
    {
      "epoch": 1.6017699115044248,
      "grad_norm": 0.3952245990105351,
      "learning_rate": 2.587159130134622e-05,
      "loss": 0.3402,
      "step": 1720
    },
    {
      "epoch": 1.6064275733581743,
      "grad_norm": 0.4137583141446628,
      "learning_rate": 2.5785295132896096e-05,
      "loss": 0.3462,
      "step": 1725
    },
    {
      "epoch": 1.6110852352119236,
      "grad_norm": 0.4270451542363023,
      "learning_rate": 2.569899896444598e-05,
      "loss": 0.357,
      "step": 1730
    },
    {
      "epoch": 1.6157428970656729,
      "grad_norm": 0.39983597405896965,
      "learning_rate": 2.5612702795995856e-05,
      "loss": 0.3557,
      "step": 1735
    },
    {
      "epoch": 1.6204005589194224,
      "grad_norm": 0.488038947004656,
      "learning_rate": 2.552640662754574e-05,
      "loss": 0.3377,
      "step": 1740
    },
    {
      "epoch": 1.6250582207731719,
      "grad_norm": 0.4365068477189154,
      "learning_rate": 2.5440110459095616e-05,
      "loss": 0.365,
      "step": 1745
    },
    {
      "epoch": 1.6297158826269214,
      "grad_norm": 0.3651077972832193,
      "learning_rate": 2.5353814290645494e-05,
      "loss": 0.3452,
      "step": 1750
    },
    {
      "epoch": 1.6343735444806708,
      "grad_norm": 0.3955926165335966,
      "learning_rate": 2.5267518122195376e-05,
      "loss": 0.3535,
      "step": 1755
    },
    {
      "epoch": 1.6390312063344201,
      "grad_norm": 0.45342658003940506,
      "learning_rate": 2.5181221953745254e-05,
      "loss": 0.3414,
      "step": 1760
    },
    {
      "epoch": 1.6436888681881694,
      "grad_norm": 0.4880007626564368,
      "learning_rate": 2.5094925785295132e-05,
      "loss": 0.3429,
      "step": 1765
    },
    {
      "epoch": 1.648346530041919,
      "grad_norm": 0.5072004917316565,
      "learning_rate": 2.5008629616845014e-05,
      "loss": 0.3397,
      "step": 1770
    },
    {
      "epoch": 1.6530041918956684,
      "grad_norm": 0.39026432176243964,
      "learning_rate": 2.4922333448394892e-05,
      "loss": 0.3535,
      "step": 1775
    },
    {
      "epoch": 1.6576618537494179,
      "grad_norm": 0.3685747616858067,
      "learning_rate": 2.483603727994477e-05,
      "loss": 0.3413,
      "step": 1780
    },
    {
      "epoch": 1.6623195156031674,
      "grad_norm": 0.4183115982720831,
      "learning_rate": 2.4749741111494652e-05,
      "loss": 0.3501,
      "step": 1785
    },
    {
      "epoch": 1.6669771774569166,
      "grad_norm": 0.41182943449330145,
      "learning_rate": 2.466344494304453e-05,
      "loss": 0.3426,
      "step": 1790
    },
    {
      "epoch": 1.671634839310666,
      "grad_norm": 0.5251058571201341,
      "learning_rate": 2.457714877459441e-05,
      "loss": 0.3407,
      "step": 1795
    },
    {
      "epoch": 1.6762925011644154,
      "grad_norm": 0.41048191414765883,
      "learning_rate": 2.4490852606144287e-05,
      "loss": 0.3401,
      "step": 1800
    },
    {
      "epoch": 1.680950163018165,
      "grad_norm": 0.43655864041454157,
      "learning_rate": 2.440455643769417e-05,
      "loss": 0.3475,
      "step": 1805
    },
    {
      "epoch": 1.6856078248719144,
      "grad_norm": 0.5519847591049524,
      "learning_rate": 2.4318260269244047e-05,
      "loss": 0.3452,
      "step": 1810
    },
    {
      "epoch": 1.6902654867256637,
      "grad_norm": 0.42249803070553016,
      "learning_rate": 2.4231964100793925e-05,
      "loss": 0.3455,
      "step": 1815
    },
    {
      "epoch": 1.6949231485794132,
      "grad_norm": 0.4115635191501472,
      "learning_rate": 2.4145667932343803e-05,
      "loss": 0.3343,
      "step": 1820
    },
    {
      "epoch": 1.6995808104331624,
      "grad_norm": 0.4457752093131425,
      "learning_rate": 2.4059371763893685e-05,
      "loss": 0.3376,
      "step": 1825
    },
    {
      "epoch": 1.704238472286912,
      "grad_norm": 0.4176018538153503,
      "learning_rate": 2.3973075595443563e-05,
      "loss": 0.3405,
      "step": 1830
    },
    {
      "epoch": 1.7088961341406614,
      "grad_norm": 0.344565413603055,
      "learning_rate": 2.388677942699344e-05,
      "loss": 0.3474,
      "step": 1835
    },
    {
      "epoch": 1.713553795994411,
      "grad_norm": 0.4091481454165795,
      "learning_rate": 2.3800483258543323e-05,
      "loss": 0.3418,
      "step": 1840
    },
    {
      "epoch": 1.7182114578481602,
      "grad_norm": 0.42255877165873534,
      "learning_rate": 2.37141870900932e-05,
      "loss": 0.3463,
      "step": 1845
    },
    {
      "epoch": 1.7228691197019095,
      "grad_norm": 0.3986400170303821,
      "learning_rate": 2.3627890921643083e-05,
      "loss": 0.3541,
      "step": 1850
    },
    {
      "epoch": 1.727526781555659,
      "grad_norm": 0.3925724522194303,
      "learning_rate": 2.3541594753192957e-05,
      "loss": 0.3381,
      "step": 1855
    },
    {
      "epoch": 1.7321844434094085,
      "grad_norm": 0.40120134440133326,
      "learning_rate": 2.345529858474284e-05,
      "loss": 0.3598,
      "step": 1860
    },
    {
      "epoch": 1.736842105263158,
      "grad_norm": 0.44257732981252107,
      "learning_rate": 2.3369002416292717e-05,
      "loss": 0.3397,
      "step": 1865
    },
    {
      "epoch": 1.7414997671169075,
      "grad_norm": 0.3500495410259064,
      "learning_rate": 2.32827062478426e-05,
      "loss": 0.3441,
      "step": 1870
    },
    {
      "epoch": 1.7461574289706567,
      "grad_norm": 0.4127245930747092,
      "learning_rate": 2.3196410079392474e-05,
      "loss": 0.3581,
      "step": 1875
    },
    {
      "epoch": 1.750815090824406,
      "grad_norm": 0.41661159230333694,
      "learning_rate": 2.3110113910942355e-05,
      "loss": 0.3497,
      "step": 1880
    },
    {
      "epoch": 1.7554727526781555,
      "grad_norm": 0.4518987523390677,
      "learning_rate": 2.3023817742492234e-05,
      "loss": 0.3503,
      "step": 1885
    },
    {
      "epoch": 1.760130414531905,
      "grad_norm": 0.3837933609690129,
      "learning_rate": 2.2937521574042115e-05,
      "loss": 0.3485,
      "step": 1890
    },
    {
      "epoch": 1.7647880763856545,
      "grad_norm": 0.43408463895449395,
      "learning_rate": 2.285122540559199e-05,
      "loss": 0.3568,
      "step": 1895
    },
    {
      "epoch": 1.7694457382394038,
      "grad_norm": 0.35839595909936256,
      "learning_rate": 2.276492923714187e-05,
      "loss": 0.3466,
      "step": 1900
    },
    {
      "epoch": 1.7741034000931533,
      "grad_norm": 0.5192679458207926,
      "learning_rate": 2.267863306869175e-05,
      "loss": 0.3465,
      "step": 1905
    },
    {
      "epoch": 1.7787610619469025,
      "grad_norm": 0.410882240906343,
      "learning_rate": 2.259233690024163e-05,
      "loss": 0.3386,
      "step": 1910
    },
    {
      "epoch": 1.783418723800652,
      "grad_norm": 0.4097036477686703,
      "learning_rate": 2.250604073179151e-05,
      "loss": 0.336,
      "step": 1915
    },
    {
      "epoch": 1.7880763856544015,
      "grad_norm": 0.42572798466762224,
      "learning_rate": 2.2419744563341388e-05,
      "loss": 0.3402,
      "step": 1920
    },
    {
      "epoch": 1.792734047508151,
      "grad_norm": 0.4526437001865557,
      "learning_rate": 2.233344839489127e-05,
      "loss": 0.349,
      "step": 1925
    },
    {
      "epoch": 1.7973917093619003,
      "grad_norm": 0.4543239654892199,
      "learning_rate": 2.2247152226441148e-05,
      "loss": 0.3383,
      "step": 1930
    },
    {
      "epoch": 1.8020493712156498,
      "grad_norm": 0.4835016026620863,
      "learning_rate": 2.2160856057991026e-05,
      "loss": 0.3358,
      "step": 1935
    },
    {
      "epoch": 1.806707033069399,
      "grad_norm": 0.3840153400762626,
      "learning_rate": 2.2074559889540904e-05,
      "loss": 0.3392,
      "step": 1940
    },
    {
      "epoch": 1.8113646949231486,
      "grad_norm": 0.45288867587660925,
      "learning_rate": 2.1988263721090786e-05,
      "loss": 0.3479,
      "step": 1945
    },
    {
      "epoch": 1.816022356776898,
      "grad_norm": 0.43954426928107787,
      "learning_rate": 2.1901967552640664e-05,
      "loss": 0.3387,
      "step": 1950
    },
    {
      "epoch": 1.8206800186306475,
      "grad_norm": 0.3888111305266047,
      "learning_rate": 2.1815671384190542e-05,
      "loss": 0.3423,
      "step": 1955
    },
    {
      "epoch": 1.8253376804843968,
      "grad_norm": 0.36280766120104074,
      "learning_rate": 2.172937521574042e-05,
      "loss": 0.3431,
      "step": 1960
    },
    {
      "epoch": 1.829995342338146,
      "grad_norm": 0.35872187797186245,
      "learning_rate": 2.1643079047290302e-05,
      "loss": 0.3459,
      "step": 1965
    },
    {
      "epoch": 1.8346530041918956,
      "grad_norm": 0.37450028176819045,
      "learning_rate": 2.155678287884018e-05,
      "loss": 0.3396,
      "step": 1970
    },
    {
      "epoch": 1.839310666045645,
      "grad_norm": 0.5010362480592447,
      "learning_rate": 2.147048671039006e-05,
      "loss": 0.3459,
      "step": 1975
    },
    {
      "epoch": 1.8439683278993946,
      "grad_norm": 0.41829410064951655,
      "learning_rate": 2.1384190541939937e-05,
      "loss": 0.3389,
      "step": 1980
    },
    {
      "epoch": 1.848625989753144,
      "grad_norm": 0.39360560324317606,
      "learning_rate": 2.129789437348982e-05,
      "loss": 0.3517,
      "step": 1985
    },
    {
      "epoch": 1.8532836516068933,
      "grad_norm": 0.4912959100610888,
      "learning_rate": 2.1211598205039697e-05,
      "loss": 0.347,
      "step": 1990
    },
    {
      "epoch": 1.8579413134606426,
      "grad_norm": 0.4469989431844116,
      "learning_rate": 2.1125302036589575e-05,
      "loss": 0.3351,
      "step": 1995
    },
    {
      "epoch": 1.8625989753143921,
      "grad_norm": 0.341803083992557,
      "learning_rate": 2.1039005868139457e-05,
      "loss": 0.3439,
      "step": 2000
    },
    {
      "epoch": 1.8672566371681416,
      "grad_norm": 0.39947807772865346,
      "learning_rate": 2.0952709699689335e-05,
      "loss": 0.3497,
      "step": 2005
    },
    {
      "epoch": 1.871914299021891,
      "grad_norm": 0.3668011199930861,
      "learning_rate": 2.0866413531239216e-05,
      "loss": 0.3407,
      "step": 2010
    },
    {
      "epoch": 1.8765719608756404,
      "grad_norm": 0.4140323339190993,
      "learning_rate": 2.078011736278909e-05,
      "loss": 0.3506,
      "step": 2015
    },
    {
      "epoch": 1.8812296227293899,
      "grad_norm": 0.4247115761097948,
      "learning_rate": 2.0693821194338973e-05,
      "loss": 0.3382,
      "step": 2020
    },
    {
      "epoch": 1.8858872845831391,
      "grad_norm": 0.3873750452328644,
      "learning_rate": 2.060752502588885e-05,
      "loss": 0.3432,
      "step": 2025
    },
    {
      "epoch": 1.8905449464368886,
      "grad_norm": 0.46101148181147794,
      "learning_rate": 2.0521228857438733e-05,
      "loss": 0.3405,
      "step": 2030
    },
    {
      "epoch": 1.8952026082906381,
      "grad_norm": 0.4043713696568898,
      "learning_rate": 2.0434932688988608e-05,
      "loss": 0.3385,
      "step": 2035
    },
    {
      "epoch": 1.8998602701443876,
      "grad_norm": 1.6308251131555385,
      "learning_rate": 2.034863652053849e-05,
      "loss": 0.3485,
      "step": 2040
    },
    {
      "epoch": 1.904517931998137,
      "grad_norm": 0.42977296093225265,
      "learning_rate": 2.0262340352088367e-05,
      "loss": 0.3493,
      "step": 2045
    },
    {
      "epoch": 1.9091755938518864,
      "grad_norm": 0.39158661566469094,
      "learning_rate": 2.017604418363825e-05,
      "loss": 0.345,
      "step": 2050
    },
    {
      "epoch": 1.9138332557056357,
      "grad_norm": 0.4649809261703152,
      "learning_rate": 2.0089748015188127e-05,
      "loss": 0.3436,
      "step": 2055
    },
    {
      "epoch": 1.9184909175593852,
      "grad_norm": 0.4576831055958142,
      "learning_rate": 2.0003451846738005e-05,
      "loss": 0.3408,
      "step": 2060
    },
    {
      "epoch": 1.9231485794131347,
      "grad_norm": 0.4977597912100062,
      "learning_rate": 1.9917155678287887e-05,
      "loss": 0.341,
      "step": 2065
    },
    {
      "epoch": 1.9278062412668842,
      "grad_norm": 0.41315229982340074,
      "learning_rate": 1.9830859509837765e-05,
      "loss": 0.3429,
      "step": 2070
    },
    {
      "epoch": 1.9324639031206334,
      "grad_norm": 0.3877041456470393,
      "learning_rate": 1.9744563341387643e-05,
      "loss": 0.3424,
      "step": 2075
    },
    {
      "epoch": 1.9371215649743827,
      "grad_norm": 0.4734848860006934,
      "learning_rate": 1.9658267172937522e-05,
      "loss": 0.3366,
      "step": 2080
    },
    {
      "epoch": 1.9417792268281322,
      "grad_norm": 0.3686041139963628,
      "learning_rate": 1.9571971004487403e-05,
      "loss": 0.3465,
      "step": 2085
    },
    {
      "epoch": 1.9464368886818817,
      "grad_norm": 0.3599711453940762,
      "learning_rate": 1.948567483603728e-05,
      "loss": 0.331,
      "step": 2090
    },
    {
      "epoch": 1.9510945505356312,
      "grad_norm": 0.3623534361004946,
      "learning_rate": 1.939937866758716e-05,
      "loss": 0.3387,
      "step": 2095
    },
    {
      "epoch": 1.9557522123893807,
      "grad_norm": 0.4611609057040844,
      "learning_rate": 1.9313082499137038e-05,
      "loss": 0.3399,
      "step": 2100
    },
    {
      "epoch": 1.96040987424313,
      "grad_norm": 0.43446509337589306,
      "learning_rate": 1.922678633068692e-05,
      "loss": 0.3486,
      "step": 2105
    },
    {
      "epoch": 1.9650675360968792,
      "grad_norm": 0.3676642923070972,
      "learning_rate": 1.9140490162236798e-05,
      "loss": 0.3359,
      "step": 2110
    },
    {
      "epoch": 1.9697251979506287,
      "grad_norm": 0.44376297339166915,
      "learning_rate": 1.9054193993786676e-05,
      "loss": 0.3415,
      "step": 2115
    },
    {
      "epoch": 1.9743828598043782,
      "grad_norm": 0.3883870723371515,
      "learning_rate": 1.8967897825336554e-05,
      "loss": 0.3443,
      "step": 2120
    },
    {
      "epoch": 1.9790405216581277,
      "grad_norm": 0.39250338911613497,
      "learning_rate": 1.8881601656886436e-05,
      "loss": 0.3399,
      "step": 2125
    },
    {
      "epoch": 1.983698183511877,
      "grad_norm": 0.3635940110663778,
      "learning_rate": 1.8795305488436314e-05,
      "loss": 0.3411,
      "step": 2130
    },
    {
      "epoch": 1.9883558453656265,
      "grad_norm": 0.3996995258897118,
      "learning_rate": 1.8709009319986192e-05,
      "loss": 0.3462,
      "step": 2135
    },
    {
      "epoch": 1.9930135072193758,
      "grad_norm": 0.4473119205714836,
      "learning_rate": 1.8622713151536074e-05,
      "loss": 0.3451,
      "step": 2140
    },
    {
      "epoch": 1.9976711690731253,
      "grad_norm": 0.4490735025991987,
      "learning_rate": 1.8536416983085952e-05,
      "loss": 0.3464,
      "step": 2145
    },
    {
      "epoch": 2.0018630647415,
      "grad_norm": 0.43028812379914894,
      "learning_rate": 1.8450120814635834e-05,
      "loss": 0.3225,
      "step": 2150
    },
    {
      "epoch": 2.0065207265952494,
      "grad_norm": 0.4054209905287257,
      "learning_rate": 1.836382464618571e-05,
      "loss": 0.2792,
      "step": 2155
    },
    {
      "epoch": 2.0111783884489984,
      "grad_norm": 0.35876300933824257,
      "learning_rate": 1.827752847773559e-05,
      "loss": 0.2735,
      "step": 2160
    },
    {
      "epoch": 2.015836050302748,
      "grad_norm": 0.3691273038336753,
      "learning_rate": 1.819123230928547e-05,
      "loss": 0.2722,
      "step": 2165
    },
    {
      "epoch": 2.0204937121564974,
      "grad_norm": 0.4177233189414795,
      "learning_rate": 1.810493614083535e-05,
      "loss": 0.2643,
      "step": 2170
    },
    {
      "epoch": 2.025151374010247,
      "grad_norm": 0.34458874465781053,
      "learning_rate": 1.8018639972385225e-05,
      "loss": 0.2772,
      "step": 2175
    },
    {
      "epoch": 2.0298090358639964,
      "grad_norm": 0.31887321159729404,
      "learning_rate": 1.7932343803935107e-05,
      "loss": 0.2671,
      "step": 2180
    },
    {
      "epoch": 2.034466697717746,
      "grad_norm": 0.3579147129970226,
      "learning_rate": 1.7846047635484985e-05,
      "loss": 0.2728,
      "step": 2185
    },
    {
      "epoch": 2.039124359571495,
      "grad_norm": 0.3784942361121672,
      "learning_rate": 1.7759751467034866e-05,
      "loss": 0.274,
      "step": 2190
    },
    {
      "epoch": 2.0437820214252445,
      "grad_norm": 0.3272900056996619,
      "learning_rate": 1.767345529858474e-05,
      "loss": 0.2705,
      "step": 2195
    },
    {
      "epoch": 2.048439683278994,
      "grad_norm": 0.35063142509597567,
      "learning_rate": 1.7587159130134623e-05,
      "loss": 0.2679,
      "step": 2200
    },
    {
      "epoch": 2.0530973451327434,
      "grad_norm": 0.35051174267675456,
      "learning_rate": 1.75008629616845e-05,
      "loss": 0.2736,
      "step": 2205
    },
    {
      "epoch": 2.057755006986493,
      "grad_norm": 0.34786100845413803,
      "learning_rate": 1.7414566793234383e-05,
      "loss": 0.2627,
      "step": 2210
    },
    {
      "epoch": 2.062412668840242,
      "grad_norm": 0.370350907908438,
      "learning_rate": 1.732827062478426e-05,
      "loss": 0.2712,
      "step": 2215
    },
    {
      "epoch": 2.0670703306939915,
      "grad_norm": 0.3725324605947192,
      "learning_rate": 1.724197445633414e-05,
      "loss": 0.2685,
      "step": 2220
    },
    {
      "epoch": 2.071727992547741,
      "grad_norm": 0.43416828814365316,
      "learning_rate": 1.715567828788402e-05,
      "loss": 0.2687,
      "step": 2225
    },
    {
      "epoch": 2.0763856544014905,
      "grad_norm": 0.36827023219145155,
      "learning_rate": 1.70693821194339e-05,
      "loss": 0.2705,
      "step": 2230
    },
    {
      "epoch": 2.08104331625524,
      "grad_norm": 0.34776031173858796,
      "learning_rate": 1.6983085950983777e-05,
      "loss": 0.265,
      "step": 2235
    },
    {
      "epoch": 2.0857009781089895,
      "grad_norm": 0.3449202439850243,
      "learning_rate": 1.6896789782533655e-05,
      "loss": 0.2748,
      "step": 2240
    },
    {
      "epoch": 2.0903586399627385,
      "grad_norm": 0.3482698035451559,
      "learning_rate": 1.6810493614083537e-05,
      "loss": 0.2716,
      "step": 2245
    },
    {
      "epoch": 2.095016301816488,
      "grad_norm": 0.3983159352725813,
      "learning_rate": 1.6724197445633415e-05,
      "loss": 0.2666,
      "step": 2250
    },
    {
      "epoch": 2.0996739636702375,
      "grad_norm": 0.35692856163758485,
      "learning_rate": 1.6637901277183294e-05,
      "loss": 0.2759,
      "step": 2255
    },
    {
      "epoch": 2.104331625523987,
      "grad_norm": 0.35950887809997184,
      "learning_rate": 1.6551605108733172e-05,
      "loss": 0.2672,
      "step": 2260
    },
    {
      "epoch": 2.1089892873777365,
      "grad_norm": 0.33930600276659506,
      "learning_rate": 1.6465308940283053e-05,
      "loss": 0.2748,
      "step": 2265
    },
    {
      "epoch": 2.113646949231486,
      "grad_norm": 0.3371202116546564,
      "learning_rate": 1.637901277183293e-05,
      "loss": 0.2689,
      "step": 2270
    },
    {
      "epoch": 2.118304611085235,
      "grad_norm": 0.6248581413728665,
      "learning_rate": 1.629271660338281e-05,
      "loss": 0.2703,
      "step": 2275
    },
    {
      "epoch": 2.1229622729389845,
      "grad_norm": 0.3358383673269155,
      "learning_rate": 1.620642043493269e-05,
      "loss": 0.2696,
      "step": 2280
    },
    {
      "epoch": 2.127619934792734,
      "grad_norm": 0.32596776335535316,
      "learning_rate": 1.612012426648257e-05,
      "loss": 0.2735,
      "step": 2285
    },
    {
      "epoch": 2.1322775966464835,
      "grad_norm": 0.3567801746323841,
      "learning_rate": 1.6033828098032448e-05,
      "loss": 0.2715,
      "step": 2290
    },
    {
      "epoch": 2.136935258500233,
      "grad_norm": 0.3415189696858835,
      "learning_rate": 1.5947531929582326e-05,
      "loss": 0.2751,
      "step": 2295
    },
    {
      "epoch": 2.1415929203539825,
      "grad_norm": 0.33587507970704916,
      "learning_rate": 1.5861235761132208e-05,
      "loss": 0.2761,
      "step": 2300
    },
    {
      "epoch": 2.1462505822077316,
      "grad_norm": 0.3301939199615534,
      "learning_rate": 1.5774939592682086e-05,
      "loss": 0.2732,
      "step": 2305
    },
    {
      "epoch": 2.150908244061481,
      "grad_norm": 0.3374609273477682,
      "learning_rate": 1.5688643424231964e-05,
      "loss": 0.2613,
      "step": 2310
    },
    {
      "epoch": 2.1555659059152306,
      "grad_norm": 0.3537406671099055,
      "learning_rate": 1.5602347255781842e-05,
      "loss": 0.2677,
      "step": 2315
    },
    {
      "epoch": 2.16022356776898,
      "grad_norm": 0.3254388017043705,
      "learning_rate": 1.5516051087331724e-05,
      "loss": 0.2657,
      "step": 2320
    },
    {
      "epoch": 2.1648812296227296,
      "grad_norm": 0.3347429433270765,
      "learning_rate": 1.5429754918881602e-05,
      "loss": 0.273,
      "step": 2325
    },
    {
      "epoch": 2.1695388914764786,
      "grad_norm": 0.32347332042126026,
      "learning_rate": 1.534345875043148e-05,
      "loss": 0.2729,
      "step": 2330
    },
    {
      "epoch": 2.174196553330228,
      "grad_norm": 0.3129217267720103,
      "learning_rate": 1.525716258198136e-05,
      "loss": 0.2663,
      "step": 2335
    },
    {
      "epoch": 2.1788542151839776,
      "grad_norm": 0.3096456883845045,
      "learning_rate": 1.517086641353124e-05,
      "loss": 0.2702,
      "step": 2340
    },
    {
      "epoch": 2.183511877037727,
      "grad_norm": 0.340914654671653,
      "learning_rate": 1.508457024508112e-05,
      "loss": 0.2754,
      "step": 2345
    },
    {
      "epoch": 2.1881695388914766,
      "grad_norm": 0.31268374201273935,
      "learning_rate": 1.4998274076630997e-05,
      "loss": 0.2697,
      "step": 2350
    },
    {
      "epoch": 2.192827200745226,
      "grad_norm": 0.31587628982001603,
      "learning_rate": 1.4911977908180877e-05,
      "loss": 0.2666,
      "step": 2355
    },
    {
      "epoch": 2.197484862598975,
      "grad_norm": 0.41265519030875064,
      "learning_rate": 1.4825681739730757e-05,
      "loss": 0.2748,
      "step": 2360
    },
    {
      "epoch": 2.2021425244527246,
      "grad_norm": 0.34480645493785267,
      "learning_rate": 1.4739385571280637e-05,
      "loss": 0.2733,
      "step": 2365
    },
    {
      "epoch": 2.206800186306474,
      "grad_norm": 0.3514431862813434,
      "learning_rate": 1.4653089402830513e-05,
      "loss": 0.2724,
      "step": 2370
    },
    {
      "epoch": 2.2114578481602236,
      "grad_norm": 0.36081808819969147,
      "learning_rate": 1.4566793234380393e-05,
      "loss": 0.2722,
      "step": 2375
    },
    {
      "epoch": 2.216115510013973,
      "grad_norm": 0.35307265393687487,
      "learning_rate": 1.4480497065930273e-05,
      "loss": 0.2697,
      "step": 2380
    },
    {
      "epoch": 2.2207731718677226,
      "grad_norm": 0.3447182585980463,
      "learning_rate": 1.4394200897480153e-05,
      "loss": 0.2854,
      "step": 2385
    },
    {
      "epoch": 2.2254308337214717,
      "grad_norm": 0.3004086881529523,
      "learning_rate": 1.4307904729030031e-05,
      "loss": 0.2751,
      "step": 2390
    },
    {
      "epoch": 2.230088495575221,
      "grad_norm": 0.35749476417900994,
      "learning_rate": 1.4221608560579911e-05,
      "loss": 0.2705,
      "step": 2395
    },
    {
      "epoch": 2.2347461574289706,
      "grad_norm": 0.3483805502173258,
      "learning_rate": 1.4135312392129791e-05,
      "loss": 0.2615,
      "step": 2400
    },
    {
      "epoch": 2.23940381928272,
      "grad_norm": 0.3236292375516467,
      "learning_rate": 1.404901622367967e-05,
      "loss": 0.2715,
      "step": 2405
    },
    {
      "epoch": 2.2440614811364696,
      "grad_norm": 0.33126778315005734,
      "learning_rate": 1.3962720055229547e-05,
      "loss": 0.2719,
      "step": 2410
    },
    {
      "epoch": 2.248719142990219,
      "grad_norm": 0.32225329762995847,
      "learning_rate": 1.3876423886779427e-05,
      "loss": 0.2702,
      "step": 2415
    },
    {
      "epoch": 2.253376804843968,
      "grad_norm": 0.37075911100946524,
      "learning_rate": 1.3790127718329307e-05,
      "loss": 0.2625,
      "step": 2420
    },
    {
      "epoch": 2.2580344666977177,
      "grad_norm": 0.32324953799129136,
      "learning_rate": 1.3703831549879187e-05,
      "loss": 0.2726,
      "step": 2425
    },
    {
      "epoch": 2.262692128551467,
      "grad_norm": 0.31199251649779275,
      "learning_rate": 1.3617535381429064e-05,
      "loss": 0.2653,
      "step": 2430
    },
    {
      "epoch": 2.2673497904052167,
      "grad_norm": 0.3403620311026147,
      "learning_rate": 1.3531239212978944e-05,
      "loss": 0.2724,
      "step": 2435
    },
    {
      "epoch": 2.272007452258966,
      "grad_norm": 0.3367570441629384,
      "learning_rate": 1.3444943044528824e-05,
      "loss": 0.269,
      "step": 2440
    },
    {
      "epoch": 2.276665114112715,
      "grad_norm": 0.32955307271701106,
      "learning_rate": 1.3358646876078703e-05,
      "loss": 0.2695,
      "step": 2445
    },
    {
      "epoch": 2.2813227759664647,
      "grad_norm": 0.31766711492777233,
      "learning_rate": 1.327235070762858e-05,
      "loss": 0.2676,
      "step": 2450
    },
    {
      "epoch": 2.285980437820214,
      "grad_norm": 0.332414527819616,
      "learning_rate": 1.318605453917846e-05,
      "loss": 0.2677,
      "step": 2455
    },
    {
      "epoch": 2.2906380996739637,
      "grad_norm": 0.3386457393609545,
      "learning_rate": 1.309975837072834e-05,
      "loss": 0.2649,
      "step": 2460
    },
    {
      "epoch": 2.295295761527713,
      "grad_norm": 0.32874628368668374,
      "learning_rate": 1.301346220227822e-05,
      "loss": 0.2804,
      "step": 2465
    },
    {
      "epoch": 2.2999534233814627,
      "grad_norm": 0.3717189613944638,
      "learning_rate": 1.2927166033828098e-05,
      "loss": 0.2686,
      "step": 2470
    },
    {
      "epoch": 2.3046110852352117,
      "grad_norm": 0.3078367814408983,
      "learning_rate": 1.2840869865377978e-05,
      "loss": 0.2757,
      "step": 2475
    },
    {
      "epoch": 2.3092687470889612,
      "grad_norm": 0.3163762519682407,
      "learning_rate": 1.2754573696927858e-05,
      "loss": 0.2666,
      "step": 2480
    },
    {
      "epoch": 2.3139264089427107,
      "grad_norm": 0.3461934224536042,
      "learning_rate": 1.2668277528477738e-05,
      "loss": 0.2752,
      "step": 2485
    },
    {
      "epoch": 2.3185840707964602,
      "grad_norm": 0.36487920734922646,
      "learning_rate": 1.2581981360027614e-05,
      "loss": 0.2756,
      "step": 2490
    },
    {
      "epoch": 2.3232417326502097,
      "grad_norm": 0.34135333676364193,
      "learning_rate": 1.2495685191577494e-05,
      "loss": 0.2677,
      "step": 2495
    },
    {
      "epoch": 2.3278993945039588,
      "grad_norm": 0.29412439897655684,
      "learning_rate": 1.2409389023127374e-05,
      "loss": 0.2618,
      "step": 2500
    },
    {
      "epoch": 2.3325570563577083,
      "grad_norm": 0.3469443902085298,
      "learning_rate": 1.2323092854677252e-05,
      "loss": 0.2687,
      "step": 2505
    },
    {
      "epoch": 2.3372147182114578,
      "grad_norm": 0.30520607437789915,
      "learning_rate": 1.2236796686227132e-05,
      "loss": 0.2691,
      "step": 2510
    },
    {
      "epoch": 2.3418723800652073,
      "grad_norm": 0.30375812241056543,
      "learning_rate": 1.215050051777701e-05,
      "loss": 0.2691,
      "step": 2515
    },
    {
      "epoch": 2.3465300419189568,
      "grad_norm": 0.30378403188035596,
      "learning_rate": 1.206420434932689e-05,
      "loss": 0.2715,
      "step": 2520
    },
    {
      "epoch": 2.3511877037727063,
      "grad_norm": 0.31807996860394366,
      "learning_rate": 1.1977908180876769e-05,
      "loss": 0.2729,
      "step": 2525
    },
    {
      "epoch": 2.3558453656264557,
      "grad_norm": 0.3198474127222539,
      "learning_rate": 1.1891612012426649e-05,
      "loss": 0.2689,
      "step": 2530
    },
    {
      "epoch": 2.360503027480205,
      "grad_norm": 0.322047324670104,
      "learning_rate": 1.1805315843976528e-05,
      "loss": 0.2632,
      "step": 2535
    },
    {
      "epoch": 2.3651606893339543,
      "grad_norm": 0.2920434471374932,
      "learning_rate": 1.1719019675526408e-05,
      "loss": 0.2624,
      "step": 2540
    },
    {
      "epoch": 2.369818351187704,
      "grad_norm": 0.2992397909930311,
      "learning_rate": 1.1632723507076287e-05,
      "loss": 0.2718,
      "step": 2545
    },
    {
      "epoch": 2.3744760130414533,
      "grad_norm": 0.3016980334570488,
      "learning_rate": 1.1546427338626167e-05,
      "loss": 0.2716,
      "step": 2550
    },
    {
      "epoch": 2.3791336748952028,
      "grad_norm": 0.29479015683701515,
      "learning_rate": 1.1460131170176045e-05,
      "loss": 0.2687,
      "step": 2555
    },
    {
      "epoch": 2.383791336748952,
      "grad_norm": 0.37092289097713493,
      "learning_rate": 1.1373835001725925e-05,
      "loss": 0.2681,
      "step": 2560
    },
    {
      "epoch": 2.3884489986027013,
      "grad_norm": 0.3169925528733488,
      "learning_rate": 1.1287538833275803e-05,
      "loss": 0.2704,
      "step": 2565
    },
    {
      "epoch": 2.393106660456451,
      "grad_norm": 0.3288103937639693,
      "learning_rate": 1.1201242664825683e-05,
      "loss": 0.2726,
      "step": 2570
    },
    {
      "epoch": 2.3977643223102003,
      "grad_norm": 0.33330135344712125,
      "learning_rate": 1.1114946496375561e-05,
      "loss": 0.2715,
      "step": 2575
    },
    {
      "epoch": 2.40242198416395,
      "grad_norm": 0.3143266064808191,
      "learning_rate": 1.1028650327925441e-05,
      "loss": 0.2678,
      "step": 2580
    },
    {
      "epoch": 2.4070796460176993,
      "grad_norm": 0.3089414591917859,
      "learning_rate": 1.094235415947532e-05,
      "loss": 0.2673,
      "step": 2585
    },
    {
      "epoch": 2.4117373078714484,
      "grad_norm": 0.33090254088999627,
      "learning_rate": 1.0856057991025199e-05,
      "loss": 0.2615,
      "step": 2590
    },
    {
      "epoch": 2.416394969725198,
      "grad_norm": 0.32046020454868757,
      "learning_rate": 1.0769761822575077e-05,
      "loss": 0.2759,
      "step": 2595
    },
    {
      "epoch": 2.4210526315789473,
      "grad_norm": 0.31086652926138264,
      "learning_rate": 1.0683465654124957e-05,
      "loss": 0.2713,
      "step": 2600
    },
    {
      "epoch": 2.425710293432697,
      "grad_norm": 0.35180943944343135,
      "learning_rate": 1.0597169485674835e-05,
      "loss": 0.2655,
      "step": 2605
    },
    {
      "epoch": 2.4303679552864463,
      "grad_norm": 0.2972576886709708,
      "learning_rate": 1.0510873317224715e-05,
      "loss": 0.2636,
      "step": 2610
    },
    {
      "epoch": 2.4350256171401954,
      "grad_norm": 0.3331594501764293,
      "learning_rate": 1.0424577148774595e-05,
      "loss": 0.2662,
      "step": 2615
    },
    {
      "epoch": 2.439683278993945,
      "grad_norm": 0.31641535541271265,
      "learning_rate": 1.0338280980324475e-05,
      "loss": 0.2689,
      "step": 2620
    },
    {
      "epoch": 2.4443409408476944,
      "grad_norm": 0.3141394480995625,
      "learning_rate": 1.0251984811874353e-05,
      "loss": 0.2636,
      "step": 2625
    },
    {
      "epoch": 2.448998602701444,
      "grad_norm": 0.3147939016682018,
      "learning_rate": 1.0165688643424233e-05,
      "loss": 0.2689,
      "step": 2630
    },
    {
      "epoch": 2.4536562645551934,
      "grad_norm": 0.31899583486198935,
      "learning_rate": 1.0079392474974112e-05,
      "loss": 0.2699,
      "step": 2635
    },
    {
      "epoch": 2.458313926408943,
      "grad_norm": 0.33177490161313883,
      "learning_rate": 9.993096306523992e-06,
      "loss": 0.2713,
      "step": 2640
    },
    {
      "epoch": 2.4629715882626924,
      "grad_norm": 0.29935857382899445,
      "learning_rate": 9.90680013807387e-06,
      "loss": 0.2709,
      "step": 2645
    },
    {
      "epoch": 2.4676292501164414,
      "grad_norm": 0.28105837326630384,
      "learning_rate": 9.82050396962375e-06,
      "loss": 0.2666,
      "step": 2650
    },
    {
      "epoch": 2.472286911970191,
      "grad_norm": 0.30304136669810483,
      "learning_rate": 9.734207801173628e-06,
      "loss": 0.2692,
      "step": 2655
    },
    {
      "epoch": 2.4769445738239404,
      "grad_norm": 0.28955490226085534,
      "learning_rate": 9.647911632723508e-06,
      "loss": 0.2626,
      "step": 2660
    },
    {
      "epoch": 2.48160223567769,
      "grad_norm": 0.2951377061299924,
      "learning_rate": 9.561615464273386e-06,
      "loss": 0.2664,
      "step": 2665
    },
    {
      "epoch": 2.4862598975314394,
      "grad_norm": 0.29826802346861564,
      "learning_rate": 9.475319295823266e-06,
      "loss": 0.2663,
      "step": 2670
    },
    {
      "epoch": 2.4909175593851884,
      "grad_norm": 0.3034793091098368,
      "learning_rate": 9.389023127373144e-06,
      "loss": 0.2644,
      "step": 2675
    },
    {
      "epoch": 2.495575221238938,
      "grad_norm": 0.32077904676215363,
      "learning_rate": 9.302726958923024e-06,
      "loss": 0.2637,
      "step": 2680
    },
    {
      "epoch": 2.5002328830926874,
      "grad_norm": 0.31558231865241043,
      "learning_rate": 9.216430790472904e-06,
      "loss": 0.2691,
      "step": 2685
    },
    {
      "epoch": 2.504890544946437,
      "grad_norm": 0.30622326682488266,
      "learning_rate": 9.130134622022784e-06,
      "loss": 0.2597,
      "step": 2690
    },
    {
      "epoch": 2.5095482068001864,
      "grad_norm": 0.31443863469595024,
      "learning_rate": 9.043838453572662e-06,
      "loss": 0.265,
      "step": 2695
    },
    {
      "epoch": 2.514205868653936,
      "grad_norm": 0.30508493796066816,
      "learning_rate": 8.957542285122542e-06,
      "loss": 0.2622,
      "step": 2700
    },
    {
      "epoch": 2.5188635305076854,
      "grad_norm": 0.290277140517799,
      "learning_rate": 8.87124611667242e-06,
      "loss": 0.2687,
      "step": 2705
    },
    {
      "epoch": 2.5235211923614345,
      "grad_norm": 0.2804105966669482,
      "learning_rate": 8.7849499482223e-06,
      "loss": 0.265,
      "step": 2710
    },
    {
      "epoch": 2.528178854215184,
      "grad_norm": 6.1479481751996605,
      "learning_rate": 8.698653779772179e-06,
      "loss": 0.2777,
      "step": 2715
    },
    {
      "epoch": 2.5328365160689335,
      "grad_norm": 0.3086526470990953,
      "learning_rate": 8.612357611322058e-06,
      "loss": 0.2747,
      "step": 2720
    },
    {
      "epoch": 2.537494177922683,
      "grad_norm": 0.30526808019405227,
      "learning_rate": 8.526061442871937e-06,
      "loss": 0.2682,
      "step": 2725
    },
    {
      "epoch": 2.542151839776432,
      "grad_norm": 0.29559206780785946,
      "learning_rate": 8.439765274421817e-06,
      "loss": 0.2754,
      "step": 2730
    },
    {
      "epoch": 2.5468095016301815,
      "grad_norm": 0.32013855137579844,
      "learning_rate": 8.353469105971695e-06,
      "loss": 0.2624,
      "step": 2735
    },
    {
      "epoch": 2.551467163483931,
      "grad_norm": 0.30046991659359606,
      "learning_rate": 8.267172937521575e-06,
      "loss": 0.2627,
      "step": 2740
    },
    {
      "epoch": 2.5561248253376805,
      "grad_norm": 0.3010828351674564,
      "learning_rate": 8.180876769071453e-06,
      "loss": 0.2661,
      "step": 2745
    },
    {
      "epoch": 2.56078248719143,
      "grad_norm": 0.29716737280159244,
      "learning_rate": 8.094580600621333e-06,
      "loss": 0.2643,
      "step": 2750
    },
    {
      "epoch": 2.5654401490451795,
      "grad_norm": 0.27621804882719136,
      "learning_rate": 8.008284432171211e-06,
      "loss": 0.2618,
      "step": 2755
    },
    {
      "epoch": 2.570097810898929,
      "grad_norm": 0.3030220173947267,
      "learning_rate": 7.921988263721091e-06,
      "loss": 0.2662,
      "step": 2760
    },
    {
      "epoch": 2.574755472752678,
      "grad_norm": 0.29400504049869375,
      "learning_rate": 7.835692095270971e-06,
      "loss": 0.2633,
      "step": 2765
    },
    {
      "epoch": 2.5794131346064275,
      "grad_norm": 0.306732227308194,
      "learning_rate": 7.749395926820851e-06,
      "loss": 0.2638,
      "step": 2770
    },
    {
      "epoch": 2.584070796460177,
      "grad_norm": 0.30788745139029816,
      "learning_rate": 7.663099758370729e-06,
      "loss": 0.2614,
      "step": 2775
    },
    {
      "epoch": 2.5887284583139265,
      "grad_norm": 0.3031265945147943,
      "learning_rate": 7.576803589920608e-06,
      "loss": 0.2689,
      "step": 2780
    },
    {
      "epoch": 2.5933861201676756,
      "grad_norm": 0.2877285616823859,
      "learning_rate": 7.490507421470487e-06,
      "loss": 0.2695,
      "step": 2785
    },
    {
      "epoch": 2.598043782021425,
      "grad_norm": 0.2934999900129627,
      "learning_rate": 7.4042112530203655e-06,
      "loss": 0.2715,
      "step": 2790
    },
    {
      "epoch": 2.6027014438751745,
      "grad_norm": 0.30329594318196373,
      "learning_rate": 7.317915084570245e-06,
      "loss": 0.2656,
      "step": 2795
    },
    {
      "epoch": 2.607359105728924,
      "grad_norm": 0.32253682128995537,
      "learning_rate": 7.231618916120124e-06,
      "loss": 0.2665,
      "step": 2800
    },
    {
      "epoch": 2.6120167675826735,
      "grad_norm": 0.30777748756481915,
      "learning_rate": 7.1453227476700035e-06,
      "loss": 0.2657,
      "step": 2805
    },
    {
      "epoch": 2.616674429436423,
      "grad_norm": 0.3057632029967134,
      "learning_rate": 7.059026579219883e-06,
      "loss": 0.2665,
      "step": 2810
    },
    {
      "epoch": 2.6213320912901725,
      "grad_norm": 0.27537170879877854,
      "learning_rate": 6.9727304107697625e-06,
      "loss": 0.26,
      "step": 2815
    },
    {
      "epoch": 2.625989753143922,
      "grad_norm": 0.28317280112866633,
      "learning_rate": 6.886434242319641e-06,
      "loss": 0.2688,
      "step": 2820
    },
    {
      "epoch": 2.630647414997671,
      "grad_norm": 0.3113422639817683,
      "learning_rate": 6.800138073869521e-06,
      "loss": 0.2694,
      "step": 2825
    },
    {
      "epoch": 2.6353050768514206,
      "grad_norm": 0.30303194362479696,
      "learning_rate": 6.713841905419399e-06,
      "loss": 0.2638,
      "step": 2830
    },
    {
      "epoch": 2.63996273870517,
      "grad_norm": 0.2839562988656851,
      "learning_rate": 6.627545736969279e-06,
      "loss": 0.2705,
      "step": 2835
    },
    {
      "epoch": 2.6446204005589196,
      "grad_norm": 0.27563661276149093,
      "learning_rate": 6.541249568519157e-06,
      "loss": 0.2739,
      "step": 2840
    },
    {
      "epoch": 2.6492780624126686,
      "grad_norm": 0.30198093934398546,
      "learning_rate": 6.454953400069037e-06,
      "loss": 0.2656,
      "step": 2845
    },
    {
      "epoch": 2.653935724266418,
      "grad_norm": 0.30991037777554664,
      "learning_rate": 6.368657231618916e-06,
      "loss": 0.2722,
      "step": 2850
    },
    {
      "epoch": 2.6585933861201676,
      "grad_norm": 0.30903335048255237,
      "learning_rate": 6.282361063168796e-06,
      "loss": 0.2612,
      "step": 2855
    },
    {
      "epoch": 2.663251047973917,
      "grad_norm": 0.2832792912640235,
      "learning_rate": 6.196064894718675e-06,
      "loss": 0.2673,
      "step": 2860
    },
    {
      "epoch": 2.6679087098276666,
      "grad_norm": 0.31138567160302016,
      "learning_rate": 6.109768726268554e-06,
      "loss": 0.2672,
      "step": 2865
    },
    {
      "epoch": 2.672566371681416,
      "grad_norm": 0.29655099700996046,
      "learning_rate": 6.023472557818433e-06,
      "loss": 0.2641,
      "step": 2870
    },
    {
      "epoch": 2.6772240335351656,
      "grad_norm": 0.28697175564566213,
      "learning_rate": 5.937176389368312e-06,
      "loss": 0.265,
      "step": 2875
    },
    {
      "epoch": 2.6818816953889146,
      "grad_norm": 0.30669541057087085,
      "learning_rate": 5.850880220918191e-06,
      "loss": 0.2659,
      "step": 2880
    },
    {
      "epoch": 2.686539357242664,
      "grad_norm": 0.3030117896925779,
      "learning_rate": 5.76458405246807e-06,
      "loss": 0.2731,
      "step": 2885
    },
    {
      "epoch": 2.6911970190964136,
      "grad_norm": 0.27836804342361304,
      "learning_rate": 5.67828788401795e-06,
      "loss": 0.2646,
      "step": 2890
    },
    {
      "epoch": 2.695854680950163,
      "grad_norm": 0.29924461798231633,
      "learning_rate": 5.591991715567829e-06,
      "loss": 0.2621,
      "step": 2895
    },
    {
      "epoch": 2.700512342803912,
      "grad_norm": 0.2799963727470457,
      "learning_rate": 5.5056955471177085e-06,
      "loss": 0.2648,
      "step": 2900
    },
    {
      "epoch": 2.7051700046576617,
      "grad_norm": 0.27244087384732596,
      "learning_rate": 5.4193993786675876e-06,
      "loss": 0.2583,
      "step": 2905
    },
    {
      "epoch": 2.709827666511411,
      "grad_norm": 0.27838335012792,
      "learning_rate": 5.333103210217467e-06,
      "loss": 0.2627,
      "step": 2910
    },
    {
      "epoch": 2.7144853283651607,
      "grad_norm": 0.28969294682295205,
      "learning_rate": 5.246807041767346e-06,
      "loss": 0.2691,
      "step": 2915
    },
    {
      "epoch": 2.71914299021891,
      "grad_norm": 0.2954244937564105,
      "learning_rate": 5.160510873317225e-06,
      "loss": 0.2636,
      "step": 2920
    },
    {
      "epoch": 2.7238006520726596,
      "grad_norm": 0.3033558084280074,
      "learning_rate": 5.074214704867105e-06,
      "loss": 0.2694,
      "step": 2925
    },
    {
      "epoch": 2.728458313926409,
      "grad_norm": 0.2880673580675071,
      "learning_rate": 4.987918536416984e-06,
      "loss": 0.2692,
      "step": 2930
    },
    {
      "epoch": 2.7331159757801586,
      "grad_norm": 0.2736154523291695,
      "learning_rate": 4.901622367966863e-06,
      "loss": 0.2633,
      "step": 2935
    },
    {
      "epoch": 2.7377736376339077,
      "grad_norm": 0.3090665768624068,
      "learning_rate": 4.815326199516742e-06,
      "loss": 0.2649,
      "step": 2940
    },
    {
      "epoch": 2.742431299487657,
      "grad_norm": 0.29106096286946964,
      "learning_rate": 4.729030031066621e-06,
      "loss": 0.272,
      "step": 2945
    },
    {
      "epoch": 2.7470889613414067,
      "grad_norm": 0.28629359418089845,
      "learning_rate": 4.6427338626165e-06,
      "loss": 0.2607,
      "step": 2950
    },
    {
      "epoch": 2.751746623195156,
      "grad_norm": 0.2899216210745862,
      "learning_rate": 4.556437694166379e-06,
      "loss": 0.2649,
      "step": 2955
    },
    {
      "epoch": 2.7564042850489052,
      "grad_norm": 0.2901843418240645,
      "learning_rate": 4.470141525716258e-06,
      "loss": 0.2667,
      "step": 2960
    },
    {
      "epoch": 2.7610619469026547,
      "grad_norm": 0.27642611679570656,
      "learning_rate": 4.383845357266138e-06,
      "loss": 0.2676,
      "step": 2965
    },
    {
      "epoch": 2.765719608756404,
      "grad_norm": 0.2700903260201578,
      "learning_rate": 4.297549188816017e-06,
      "loss": 0.2679,
      "step": 2970
    },
    {
      "epoch": 2.7703772706101537,
      "grad_norm": 0.3114018254754919,
      "learning_rate": 4.211253020365896e-06,
      "loss": 0.2662,
      "step": 2975
    },
    {
      "epoch": 2.775034932463903,
      "grad_norm": 0.3092109607728558,
      "learning_rate": 4.124956851915775e-06,
      "loss": 0.2626,
      "step": 2980
    },
    {
      "epoch": 2.7796925943176527,
      "grad_norm": 0.3055282038126083,
      "learning_rate": 4.0386606834656544e-06,
      "loss": 0.2695,
      "step": 2985
    },
    {
      "epoch": 2.784350256171402,
      "grad_norm": 0.28306882068846634,
      "learning_rate": 3.9523645150155335e-06,
      "loss": 0.2662,
      "step": 2990
    },
    {
      "epoch": 2.7890079180251512,
      "grad_norm": 0.2827108746288498,
      "learning_rate": 3.8660683465654126e-06,
      "loss": 0.2618,
      "step": 2995
    },
    {
      "epoch": 2.7936655798789007,
      "grad_norm": 0.2898456225548245,
      "learning_rate": 3.779772178115292e-06,
      "loss": 0.2686,
      "step": 3000
    },
    {
      "epoch": 2.7983232417326502,
      "grad_norm": 0.27424060863614574,
      "learning_rate": 3.693476009665171e-06,
      "loss": 0.2627,
      "step": 3005
    },
    {
      "epoch": 2.8029809035863997,
      "grad_norm": 0.27290134092267404,
      "learning_rate": 3.6071798412150506e-06,
      "loss": 0.269,
      "step": 3010
    },
    {
      "epoch": 2.807638565440149,
      "grad_norm": 0.28796373650746787,
      "learning_rate": 3.5208836727649297e-06,
      "loss": 0.2551,
      "step": 3015
    },
    {
      "epoch": 2.8122962272938983,
      "grad_norm": 0.2781225855318277,
      "learning_rate": 3.434587504314809e-06,
      "loss": 0.2667,
      "step": 3020
    },
    {
      "epoch": 2.8169538891476478,
      "grad_norm": 0.26326000490501106,
      "learning_rate": 3.348291335864688e-06,
      "loss": 0.2646,
      "step": 3025
    },
    {
      "epoch": 2.8216115510013973,
      "grad_norm": 0.2606023563840398,
      "learning_rate": 3.2619951674145674e-06,
      "loss": 0.2619,
      "step": 3030
    },
    {
      "epoch": 2.8262692128551468,
      "grad_norm": 0.27810164754919947,
      "learning_rate": 3.1756989989644464e-06,
      "loss": 0.2648,
      "step": 3035
    },
    {
      "epoch": 2.8309268747088963,
      "grad_norm": 0.29523783528795877,
      "learning_rate": 3.089402830514325e-06,
      "loss": 0.2701,
      "step": 3040
    },
    {
      "epoch": 2.8355845365626458,
      "grad_norm": 0.2580016014937091,
      "learning_rate": 3.0031066620642046e-06,
      "loss": 0.2622,
      "step": 3045
    },
    {
      "epoch": 2.840242198416395,
      "grad_norm": 0.30172007099883025,
      "learning_rate": 2.9168104936140837e-06,
      "loss": 0.2614,
      "step": 3050
    },
    {
      "epoch": 2.8448998602701443,
      "grad_norm": 0.29799681738240597,
      "learning_rate": 2.8305143251639627e-06,
      "loss": 0.2691,
      "step": 3055
    },
    {
      "epoch": 2.849557522123894,
      "grad_norm": 0.2649774150276832,
      "learning_rate": 2.7442181567138422e-06,
      "loss": 0.271,
      "step": 3060
    },
    {
      "epoch": 2.8542151839776433,
      "grad_norm": 0.2852446285396719,
      "learning_rate": 2.6579219882637213e-06,
      "loss": 0.2674,
      "step": 3065
    },
    {
      "epoch": 2.858872845831393,
      "grad_norm": 0.2765953270347661,
      "learning_rate": 2.5716258198136004e-06,
      "loss": 0.2582,
      "step": 3070
    },
    {
      "epoch": 2.863530507685142,
      "grad_norm": 0.2845253380416326,
      "learning_rate": 2.4853296513634795e-06,
      "loss": 0.2642,
      "step": 3075
    },
    {
      "epoch": 2.8681881695388913,
      "grad_norm": 0.28907774572406547,
      "learning_rate": 2.399033482913359e-06,
      "loss": 0.2583,
      "step": 3080
    },
    {
      "epoch": 2.872845831392641,
      "grad_norm": 0.2727970735427606,
      "learning_rate": 2.312737314463238e-06,
      "loss": 0.2642,
      "step": 3085
    },
    {
      "epoch": 2.8775034932463903,
      "grad_norm": 0.2733420123820053,
      "learning_rate": 2.226441146013117e-06,
      "loss": 0.2692,
      "step": 3090
    },
    {
      "epoch": 2.88216115510014,
      "grad_norm": 0.2797813055046872,
      "learning_rate": 2.140144977562996e-06,
      "loss": 0.2593,
      "step": 3095
    },
    {
      "epoch": 2.8868188169538893,
      "grad_norm": 0.2633383007165478,
      "learning_rate": 2.0538488091128757e-06,
      "loss": 0.2658,
      "step": 3100
    },
    {
      "epoch": 2.891476478807639,
      "grad_norm": 0.2769336388827699,
      "learning_rate": 1.9675526406627547e-06,
      "loss": 0.2663,
      "step": 3105
    },
    {
      "epoch": 2.896134140661388,
      "grad_norm": 0.2546568035332576,
      "learning_rate": 1.8812564722126338e-06,
      "loss": 0.2582,
      "step": 3110
    },
    {
      "epoch": 2.9007918025151374,
      "grad_norm": 0.25880729559847354,
      "learning_rate": 1.794960303762513e-06,
      "loss": 0.2574,
      "step": 3115
    },
    {
      "epoch": 2.905449464368887,
      "grad_norm": 0.27569006812368774,
      "learning_rate": 1.7086641353123924e-06,
      "loss": 0.2645,
      "step": 3120
    },
    {
      "epoch": 2.9101071262226363,
      "grad_norm": 0.26915391351888773,
      "learning_rate": 1.6223679668622715e-06,
      "loss": 0.2658,
      "step": 3125
    },
    {
      "epoch": 2.9147647880763854,
      "grad_norm": 0.2626594854816184,
      "learning_rate": 1.5360717984121505e-06,
      "loss": 0.2562,
      "step": 3130
    },
    {
      "epoch": 2.919422449930135,
      "grad_norm": 0.2620279806453697,
      "learning_rate": 1.4497756299620296e-06,
      "loss": 0.2627,
      "step": 3135
    },
    {
      "epoch": 2.9240801117838844,
      "grad_norm": 0.25616555933566926,
      "learning_rate": 1.363479461511909e-06,
      "loss": 0.2688,
      "step": 3140
    },
    {
      "epoch": 2.928737773637634,
      "grad_norm": 0.27394240411115406,
      "learning_rate": 1.277183293061788e-06,
      "loss": 0.2723,
      "step": 3145
    },
    {
      "epoch": 2.9333954354913834,
      "grad_norm": 0.2690479285493206,
      "learning_rate": 1.1908871246116673e-06,
      "loss": 0.2651,
      "step": 3150
    },
    {
      "epoch": 2.938053097345133,
      "grad_norm": 0.3712818208897552,
      "learning_rate": 1.1045909561615463e-06,
      "loss": 0.2611,
      "step": 3155
    },
    {
      "epoch": 2.9427107591988824,
      "grad_norm": 0.2774650328132279,
      "learning_rate": 1.0182947877114256e-06,
      "loss": 0.2646,
      "step": 3160
    },
    {
      "epoch": 2.9473684210526314,
      "grad_norm": 0.27275138892373524,
      "learning_rate": 9.319986192613048e-07,
      "loss": 0.2659,
      "step": 3165
    },
    {
      "epoch": 2.952026082906381,
      "grad_norm": 0.24604170225680166,
      "learning_rate": 8.45702450811184e-07,
      "loss": 0.2617,
      "step": 3170
    },
    {
      "epoch": 2.9566837447601304,
      "grad_norm": 0.2642587866106321,
      "learning_rate": 7.594062823610632e-07,
      "loss": 0.2656,
      "step": 3175
    },
    {
      "epoch": 2.96134140661388,
      "grad_norm": 0.26008096815246584,
      "learning_rate": 6.731101139109423e-07,
      "loss": 0.262,
      "step": 3180
    },
    {
      "epoch": 2.9659990684676294,
      "grad_norm": 0.2672889395247758,
      "learning_rate": 5.868139454608215e-07,
      "loss": 0.2622,
      "step": 3185
    },
    {
      "epoch": 2.9706567303213784,
      "grad_norm": 0.25575145904728286,
      "learning_rate": 5.005177770107007e-07,
      "loss": 0.2564,
      "step": 3190
    },
    {
      "epoch": 2.975314392175128,
      "grad_norm": 0.2709375247567089,
      "learning_rate": 4.142216085605799e-07,
      "loss": 0.268,
      "step": 3195
    },
    {
      "epoch": 2.9799720540288774,
      "grad_norm": 0.25664197436812547,
      "learning_rate": 3.279254401104591e-07,
      "loss": 0.2667,
      "step": 3200
    },
    {
      "epoch": 2.984629715882627,
      "grad_norm": 0.2559659898287829,
      "learning_rate": 2.416292716603383e-07,
      "loss": 0.2661,
      "step": 3205
    },
    {
      "epoch": 2.9892873777363764,
      "grad_norm": 0.2684555308311467,
      "learning_rate": 1.5533310321021747e-07,
      "loss": 0.2524,
      "step": 3210
    },
    {
      "epoch": 2.993945039590126,
      "grad_norm": 0.2742465979505438,
      "learning_rate": 6.903693476009665e-08,
      "loss": 0.2548,
      "step": 3215
    },
    {
      "epoch": 2.9976711690731253,
      "step": 3219,
      "total_flos": 2.754977641806168e+18,
      "train_loss": 0.35548364405085414,
      "train_runtime": 119587.5261,
      "train_samples_per_second": 0.431,
      "train_steps_per_second": 0.027
    }
  ],
  "logging_steps": 5,
  "max_steps": 3219,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "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": 2.754977641806168e+18,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}