diff --git "a/final/trainer_state.json" "b/final/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/final/trainer_state.json"
@@ -0,0 +1,4185 @@
+{
+  "best_global_step": null,
+  "best_metric": null,
+  "best_model_checkpoint": null,
+  "epoch": 0.31061914481603603,
+  "eval_steps": 500,
+  "global_step": 5935,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.0005233683990160674,
+      "grad_norm": 1.1370724439620972,
+      "learning_rate": 2.910112359550562e-06,
+      "loss": 0.8003,
+      "step": 10
+    },
+    {
+      "epoch": 0.0010467367980321349,
+      "grad_norm": 0.9877656102180481,
+      "learning_rate": 3.921348314606742e-06,
+      "loss": 0.7587,
+      "step": 20
+    },
+    {
+      "epoch": 0.0015701051970482022,
+      "grad_norm": 0.8628589510917664,
+      "learning_rate": 4.932584269662922e-06,
+      "loss": 0.7476,
+      "step": 30
+    },
+    {
+      "epoch": 0.0020934735960642698,
+      "grad_norm": 0.7837583422660828,
+      "learning_rate": 5.943820224719102e-06,
+      "loss": 0.7334,
+      "step": 40
+    },
+    {
+      "epoch": 0.002616841995080337,
+      "grad_norm": 0.7342932224273682,
+      "learning_rate": 6.955056179775282e-06,
+      "loss": 0.7397,
+      "step": 50
+    },
+    {
+      "epoch": 0.0031402103940964044,
+      "grad_norm": 0.9271304607391357,
+      "learning_rate": 7.966292134831462e-06,
+      "loss": 0.7296,
+      "step": 60
+    },
+    {
+      "epoch": 0.0036635787931124718,
+      "grad_norm": 0.8066033720970154,
+      "learning_rate": 8.977528089887641e-06,
+      "loss": 0.6864,
+      "step": 70
+    },
+    {
+      "epoch": 0.0041869471921285395,
+      "grad_norm": 0.8114457726478577,
+      "learning_rate": 9.988764044943822e-06,
+      "loss": 0.6792,
+      "step": 80
+    },
+    {
+      "epoch": 0.004710315591144607,
+      "grad_norm": 0.8880093097686768,
+      "learning_rate": 1.1000000000000001e-05,
+      "loss": 0.7508,
+      "step": 90
+    },
+    {
+      "epoch": 0.005233683990160674,
+      "grad_norm": 1.1273611783981323,
+      "learning_rate": 1.2011235955056182e-05,
+      "loss": 0.6849,
+      "step": 100
+    },
+    {
+      "epoch": 0.0057570523891767415,
+      "grad_norm": 0.7791317701339722,
+      "learning_rate": 1.3022471910112362e-05,
+      "loss": 0.698,
+      "step": 110
+    },
+    {
+      "epoch": 0.006280420788192809,
+      "grad_norm": 0.8140974044799805,
+      "learning_rate": 1.403370786516854e-05,
+      "loss": 0.6947,
+      "step": 120
+    },
+    {
+      "epoch": 0.006803789187208876,
+      "grad_norm": 0.8656443357467651,
+      "learning_rate": 1.504494382022472e-05,
+      "loss": 0.7158,
+      "step": 130
+    },
+    {
+      "epoch": 0.0073271575862249435,
+      "grad_norm": 0.744526743888855,
+      "learning_rate": 1.60561797752809e-05,
+      "loss": 0.6701,
+      "step": 140
+    },
+    {
+      "epoch": 0.00785052598524101,
+      "grad_norm": 0.7715036273002625,
+      "learning_rate": 1.706741573033708e-05,
+      "loss": 0.6794,
+      "step": 150
+    },
+    {
+      "epoch": 0.008373894384257079,
+      "grad_norm": 0.7816882729530334,
+      "learning_rate": 1.8078651685393256e-05,
+      "loss": 0.7059,
+      "step": 160
+    },
+    {
+      "epoch": 0.008897262783273146,
+      "grad_norm": 0.7964625954627991,
+      "learning_rate": 1.908988764044944e-05,
+      "loss": 0.6954,
+      "step": 170
+    },
+    {
+      "epoch": 0.009420631182289214,
+      "grad_norm": 1.0829215049743652,
+      "learning_rate": 2e-05,
+      "loss": 0.6888,
+      "step": 180
+    },
+    {
+      "epoch": 0.00994399958130528,
+      "grad_norm": 0.6891264319419861,
+      "learning_rate": 2e-05,
+      "loss": 0.6539,
+      "step": 190
+    },
+    {
+      "epoch": 0.010467367980321348,
+      "grad_norm": 0.7293897867202759,
+      "learning_rate": 2e-05,
+      "loss": 0.7164,
+      "step": 200
+    },
+    {
+      "epoch": 0.010990736379337415,
+      "grad_norm": 0.6318921446800232,
+      "learning_rate": 2e-05,
+      "loss": 0.7084,
+      "step": 210
+    },
+    {
+      "epoch": 0.011514104778353483,
+      "grad_norm": 0.6405470371246338,
+      "learning_rate": 2e-05,
+      "loss": 0.6812,
+      "step": 220
+    },
+    {
+      "epoch": 0.012037473177369551,
+      "grad_norm": 0.6632219552993774,
+      "learning_rate": 2e-05,
+      "loss": 0.6733,
+      "step": 230
+    },
+    {
+      "epoch": 0.012560841576385618,
+      "grad_norm": 0.7196585536003113,
+      "learning_rate": 2e-05,
+      "loss": 0.6835,
+      "step": 240
+    },
+    {
+      "epoch": 0.013084209975401686,
+      "grad_norm": 0.5942214131355286,
+      "learning_rate": 2e-05,
+      "loss": 0.6613,
+      "step": 250
+    },
+    {
+      "epoch": 0.013607578374417752,
+      "grad_norm": 0.5716996788978577,
+      "learning_rate": 2e-05,
+      "loss": 0.6759,
+      "step": 260
+    },
+    {
+      "epoch": 0.01413094677343382,
+      "grad_norm": 0.5713549256324768,
+      "learning_rate": 2e-05,
+      "loss": 0.7041,
+      "step": 270
+    },
+    {
+      "epoch": 0.014654315172449887,
+      "grad_norm": 0.5658433437347412,
+      "learning_rate": 2e-05,
+      "loss": 0.6866,
+      "step": 280
+    },
+    {
+      "epoch": 0.015177683571465955,
+      "grad_norm": 0.5381125211715698,
+      "learning_rate": 2e-05,
+      "loss": 0.7024,
+      "step": 290
+    },
+    {
+      "epoch": 0.01570105197048202,
+      "grad_norm": 0.5079479217529297,
+      "learning_rate": 2e-05,
+      "loss": 0.686,
+      "step": 300
+    },
+    {
+      "epoch": 0.016224420369498088,
+      "grad_norm": 0.5530030727386475,
+      "learning_rate": 2e-05,
+      "loss": 0.6912,
+      "step": 310
+    },
+    {
+      "epoch": 0.016747788768514158,
+      "grad_norm": 0.5396811962127686,
+      "learning_rate": 2e-05,
+      "loss": 0.6751,
+      "step": 320
+    },
+    {
+      "epoch": 0.017271157167530225,
+      "grad_norm": 0.4894005060195923,
+      "learning_rate": 2e-05,
+      "loss": 0.6649,
+      "step": 330
+    },
+    {
+      "epoch": 0.01779452556654629,
+      "grad_norm": 0.5230134129524231,
+      "learning_rate": 2e-05,
+      "loss": 0.684,
+      "step": 340
+    },
+    {
+      "epoch": 0.01831789396556236,
+      "grad_norm": 0.5332402586936951,
+      "learning_rate": 2e-05,
+      "loss": 0.6767,
+      "step": 350
+    },
+    {
+      "epoch": 0.018841262364578427,
+      "grad_norm": 0.5048417448997498,
+      "learning_rate": 2e-05,
+      "loss": 0.6922,
+      "step": 360
+    },
+    {
+      "epoch": 0.019364630763594494,
+      "grad_norm": 0.5345087051391602,
+      "learning_rate": 2e-05,
+      "loss": 0.6976,
+      "step": 370
+    },
+    {
+      "epoch": 0.01988799916261056,
+      "grad_norm": 0.511880099773407,
+      "learning_rate": 2e-05,
+      "loss": 0.6814,
+      "step": 380
+    },
+    {
+      "epoch": 0.02041136756162663,
+      "grad_norm": 0.46280574798583984,
+      "learning_rate": 2e-05,
+      "loss": 0.6566,
+      "step": 390
+    },
+    {
+      "epoch": 0.020934735960642697,
+      "grad_norm": 0.464054137468338,
+      "learning_rate": 2e-05,
+      "loss": 0.6483,
+      "step": 400
+    },
+    {
+      "epoch": 0.021458104359658763,
+      "grad_norm": 0.5449008941650391,
+      "learning_rate": 2e-05,
+      "loss": 0.6841,
+      "step": 410
+    },
+    {
+      "epoch": 0.02198147275867483,
+      "grad_norm": 0.5352079272270203,
+      "learning_rate": 2e-05,
+      "loss": 0.6604,
+      "step": 420
+    },
+    {
+      "epoch": 0.0225048411576909,
+      "grad_norm": 0.5088271498680115,
+      "learning_rate": 2e-05,
+      "loss": 0.6683,
+      "step": 430
+    },
+    {
+      "epoch": 0.023028209556706966,
+      "grad_norm": 0.5593430399894714,
+      "learning_rate": 2e-05,
+      "loss": 0.6795,
+      "step": 440
+    },
+    {
+      "epoch": 0.023551577955723033,
+      "grad_norm": 0.507481575012207,
+      "learning_rate": 2e-05,
+      "loss": 0.6654,
+      "step": 450
+    },
+    {
+      "epoch": 0.024074946354739103,
+      "grad_norm": 0.4815616011619568,
+      "learning_rate": 2e-05,
+      "loss": 0.6843,
+      "step": 460
+    },
+    {
+      "epoch": 0.02459831475375517,
+      "grad_norm": 0.5703353881835938,
+      "learning_rate": 2e-05,
+      "loss": 0.6598,
+      "step": 470
+    },
+    {
+      "epoch": 0.025121683152771235,
+      "grad_norm": 0.48705339431762695,
+      "learning_rate": 2e-05,
+      "loss": 0.6614,
+      "step": 480
+    },
+    {
+      "epoch": 0.025645051551787302,
+      "grad_norm": 0.547533392906189,
+      "learning_rate": 2e-05,
+      "loss": 0.6585,
+      "step": 490
+    },
+    {
+      "epoch": 0.026168419950803372,
+      "grad_norm": 0.6569129228591919,
+      "learning_rate": 2e-05,
+      "loss": 0.6716,
+      "step": 500
+    },
+    {
+      "epoch": 0.02669178834981944,
+      "grad_norm": 0.5733374357223511,
+      "learning_rate": 2e-05,
+      "loss": 0.6465,
+      "step": 510
+    },
+    {
+      "epoch": 0.027215156748835505,
+      "grad_norm": 0.4885842502117157,
+      "learning_rate": 2e-05,
+      "loss": 0.682,
+      "step": 520
+    },
+    {
+      "epoch": 0.02773852514785157,
+      "grad_norm": 0.42785435914993286,
+      "learning_rate": 2e-05,
+      "loss": 0.6594,
+      "step": 530
+    },
+    {
+      "epoch": 0.02826189354686764,
+      "grad_norm": 0.4955022931098938,
+      "learning_rate": 2e-05,
+      "loss": 0.6674,
+      "step": 540
+    },
+    {
+      "epoch": 0.028785261945883708,
+      "grad_norm": 0.48021453619003296,
+      "learning_rate": 2e-05,
+      "loss": 0.6796,
+      "step": 550
+    },
+    {
+      "epoch": 0.029308630344899774,
+      "grad_norm": 0.4767313301563263,
+      "learning_rate": 2e-05,
+      "loss": 0.6865,
+      "step": 560
+    },
+    {
+      "epoch": 0.029831998743915844,
+      "grad_norm": 0.5284311771392822,
+      "learning_rate": 2e-05,
+      "loss": 0.6489,
+      "step": 570
+    },
+    {
+      "epoch": 0.03035536714293191,
+      "grad_norm": 0.4461924731731415,
+      "learning_rate": 2e-05,
+      "loss": 0.686,
+      "step": 580
+    },
+    {
+      "epoch": 0.030878735541947977,
+      "grad_norm": 0.45115983486175537,
+      "learning_rate": 2e-05,
+      "loss": 0.6539,
+      "step": 590
+    },
+    {
+      "epoch": 0.03140210394096404,
+      "grad_norm": 0.5064119100570679,
+      "learning_rate": 2e-05,
+      "loss": 0.6946,
+      "step": 600
+    },
+    {
+      "epoch": 0.03192547233998011,
+      "grad_norm": 0.5319550037384033,
+      "learning_rate": 2e-05,
+      "loss": 0.645,
+      "step": 610
+    },
+    {
+      "epoch": 0.032448840738996176,
+      "grad_norm": 0.4419703781604767,
+      "learning_rate": 2e-05,
+      "loss": 0.7003,
+      "step": 620
+    },
+    {
+      "epoch": 0.03297220913801225,
+      "grad_norm": 0.42887452244758606,
+      "learning_rate": 2e-05,
+      "loss": 0.6783,
+      "step": 630
+    },
+    {
+      "epoch": 0.033495577537028316,
+      "grad_norm": 0.46661144495010376,
+      "learning_rate": 2e-05,
+      "loss": 0.6686,
+      "step": 640
+    },
+    {
+      "epoch": 0.03401894593604438,
+      "grad_norm": 0.5283986330032349,
+      "learning_rate": 2e-05,
+      "loss": 0.6679,
+      "step": 650
+    },
+    {
+      "epoch": 0.03454231433506045,
+      "grad_norm": 0.5113454461097717,
+      "learning_rate": 2e-05,
+      "loss": 0.6506,
+      "step": 660
+    },
+    {
+      "epoch": 0.035065682734076516,
+      "grad_norm": 0.4744948148727417,
+      "learning_rate": 2e-05,
+      "loss": 0.6724,
+      "step": 670
+    },
+    {
+      "epoch": 0.03558905113309258,
+      "grad_norm": 0.49696066975593567,
+      "learning_rate": 2e-05,
+      "loss": 0.6652,
+      "step": 680
+    },
+    {
+      "epoch": 0.03611241953210865,
+      "grad_norm": 0.7813011407852173,
+      "learning_rate": 2e-05,
+      "loss": 0.6621,
+      "step": 690
+    },
+    {
+      "epoch": 0.03663578793112472,
+      "grad_norm": 0.49424248933792114,
+      "learning_rate": 2e-05,
+      "loss": 0.6727,
+      "step": 700
+    },
+    {
+      "epoch": 0.03715915633014079,
+      "grad_norm": 0.4447154104709625,
+      "learning_rate": 2e-05,
+      "loss": 0.6659,
+      "step": 710
+    },
+    {
+      "epoch": 0.037682524729156855,
+      "grad_norm": 0.44892066717147827,
+      "learning_rate": 2e-05,
+      "loss": 0.6554,
+      "step": 720
+    },
+    {
+      "epoch": 0.03820589312817292,
+      "grad_norm": 0.49734535813331604,
+      "learning_rate": 2e-05,
+      "loss": 0.6779,
+      "step": 730
+    },
+    {
+      "epoch": 0.03872926152718899,
+      "grad_norm": 0.5591990947723389,
+      "learning_rate": 2e-05,
+      "loss": 0.6927,
+      "step": 740
+    },
+    {
+      "epoch": 0.039252629926205054,
+      "grad_norm": 0.45224255323410034,
+      "learning_rate": 2e-05,
+      "loss": 0.6398,
+      "step": 750
+    },
+    {
+      "epoch": 0.03977599832522112,
+      "grad_norm": 0.41793182492256165,
+      "learning_rate": 2e-05,
+      "loss": 0.6668,
+      "step": 760
+    },
+    {
+      "epoch": 0.04029936672423719,
+      "grad_norm": 0.43993815779685974,
+      "learning_rate": 2e-05,
+      "loss": 0.6456,
+      "step": 770
+    },
+    {
+      "epoch": 0.04082273512325326,
+      "grad_norm": 0.44372230768203735,
+      "learning_rate": 2e-05,
+      "loss": 0.6669,
+      "step": 780
+    },
+    {
+      "epoch": 0.04134610352226933,
+      "grad_norm": 0.4827944040298462,
+      "learning_rate": 2e-05,
+      "loss": 0.657,
+      "step": 790
+    },
+    {
+      "epoch": 0.041869471921285394,
+      "grad_norm": 0.4187488257884979,
+      "learning_rate": 2e-05,
+      "loss": 0.6456,
+      "step": 800
+    },
+    {
+      "epoch": 0.04239284032030146,
+      "grad_norm": 0.47213008999824524,
+      "learning_rate": 2e-05,
+      "loss": 0.6555,
+      "step": 810
+    },
+    {
+      "epoch": 0.042916208719317527,
+      "grad_norm": 0.43557852506637573,
+      "learning_rate": 2e-05,
+      "loss": 0.6545,
+      "step": 820
+    },
+    {
+      "epoch": 0.04343957711833359,
+      "grad_norm": 0.45871806144714355,
+      "learning_rate": 2e-05,
+      "loss": 0.6538,
+      "step": 830
+    },
+    {
+      "epoch": 0.04396294551734966,
+      "grad_norm": 0.46028947830200195,
+      "learning_rate": 2e-05,
+      "loss": 0.6356,
+      "step": 840
+    },
+    {
+      "epoch": 0.04448631391636573,
+      "grad_norm": 0.548116147518158,
+      "learning_rate": 2e-05,
+      "loss": 0.6872,
+      "step": 850
+    },
+    {
+      "epoch": 0.0450096823153818,
+      "grad_norm": 0.49427151679992676,
+      "learning_rate": 2e-05,
+      "loss": 0.6372,
+      "step": 860
+    },
+    {
+      "epoch": 0.045533050714397866,
+      "grad_norm": 0.46103739738464355,
+      "learning_rate": 2e-05,
+      "loss": 0.6991,
+      "step": 870
+    },
+    {
+      "epoch": 0.04605641911341393,
+      "grad_norm": 0.4400480091571808,
+      "learning_rate": 2e-05,
+      "loss": 0.667,
+      "step": 880
+    },
+    {
+      "epoch": 0.04657978751243,
+      "grad_norm": 0.45002153515815735,
+      "learning_rate": 2e-05,
+      "loss": 0.6582,
+      "step": 890
+    },
+    {
+      "epoch": 0.047103155911446065,
+      "grad_norm": 0.44472604990005493,
+      "learning_rate": 2e-05,
+      "loss": 0.6706,
+      "step": 900
+    },
+    {
+      "epoch": 0.04762652431046213,
+      "grad_norm": 0.4121211767196655,
+      "learning_rate": 2e-05,
+      "loss": 0.6602,
+      "step": 910
+    },
+    {
+      "epoch": 0.048149892709478205,
+      "grad_norm": 0.5010108351707458,
+      "learning_rate": 2e-05,
+      "loss": 0.6761,
+      "step": 920
+    },
+    {
+      "epoch": 0.04867326110849427,
+      "grad_norm": 0.45920342206954956,
+      "learning_rate": 2e-05,
+      "loss": 0.6671,
+      "step": 930
+    },
+    {
+      "epoch": 0.04919662950751034,
+      "grad_norm": 0.5177173018455505,
+      "learning_rate": 2e-05,
+      "loss": 0.652,
+      "step": 940
+    },
+    {
+      "epoch": 0.049719997906526404,
+      "grad_norm": 0.49086499214172363,
+      "learning_rate": 2e-05,
+      "loss": 0.6795,
+      "step": 950
+    },
+    {
+      "epoch": 0.05024336630554247,
+      "grad_norm": 0.582841157913208,
+      "learning_rate": 2e-05,
+      "loss": 0.6376,
+      "step": 960
+    },
+    {
+      "epoch": 0.05076673470455854,
+      "grad_norm": 0.4690440595149994,
+      "learning_rate": 2e-05,
+      "loss": 0.6465,
+      "step": 970
+    },
+    {
+      "epoch": 0.051290103103574604,
+      "grad_norm": 0.45363402366638184,
+      "learning_rate": 2e-05,
+      "loss": 0.6511,
+      "step": 980
+    },
+    {
+      "epoch": 0.05181347150259067,
+      "grad_norm": 0.43920376896858215,
+      "learning_rate": 2e-05,
+      "loss": 0.6943,
+      "step": 990
+    },
+    {
+      "epoch": 0.052336839901606744,
+      "grad_norm": 0.421144038438797,
+      "learning_rate": 2e-05,
+      "loss": 0.6434,
+      "step": 1000
+    },
+    {
+      "epoch": 0.05286020830062281,
+      "grad_norm": 0.45188188552856445,
+      "learning_rate": 2e-05,
+      "loss": 0.6467,
+      "step": 1010
+    },
+    {
+      "epoch": 0.05338357669963888,
+      "grad_norm": 0.4339853823184967,
+      "learning_rate": 2e-05,
+      "loss": 0.6596,
+      "step": 1020
+    },
+    {
+      "epoch": 0.05390694509865494,
+      "grad_norm": 0.4234607517719269,
+      "learning_rate": 2e-05,
+      "loss": 0.674,
+      "step": 1030
+    },
+    {
+      "epoch": 0.05443031349767101,
+      "grad_norm": 0.46552717685699463,
+      "learning_rate": 2e-05,
+      "loss": 0.6649,
+      "step": 1040
+    },
+    {
+      "epoch": 0.054953681896687076,
+      "grad_norm": 0.4405843913555145,
+      "learning_rate": 2e-05,
+      "loss": 0.6577,
+      "step": 1050
+    },
+    {
+      "epoch": 0.05547705029570314,
+      "grad_norm": 0.5044958591461182,
+      "learning_rate": 2e-05,
+      "loss": 0.6152,
+      "step": 1060
+    },
+    {
+      "epoch": 0.056000418694719216,
+      "grad_norm": 0.43265852332115173,
+      "learning_rate": 2e-05,
+      "loss": 0.6317,
+      "step": 1070
+    },
+    {
+      "epoch": 0.05652378709373528,
+      "grad_norm": 0.41526952385902405,
+      "learning_rate": 2e-05,
+      "loss": 0.6501,
+      "step": 1080
+    },
+    {
+      "epoch": 0.05704715549275135,
+      "grad_norm": 0.43947693705558777,
+      "learning_rate": 2e-05,
+      "loss": 0.6542,
+      "step": 1090
+    },
+    {
+      "epoch": 0.057570523891767415,
+      "grad_norm": 0.4375130534172058,
+      "learning_rate": 2e-05,
+      "loss": 0.671,
+      "step": 1100
+    },
+    {
+      "epoch": 0.05809389229078348,
+      "grad_norm": 0.4554004371166229,
+      "learning_rate": 2e-05,
+      "loss": 0.6507,
+      "step": 1110
+    },
+    {
+      "epoch": 0.05861726068979955,
+      "grad_norm": 0.5064834356307983,
+      "learning_rate": 2e-05,
+      "loss": 0.6467,
+      "step": 1120
+    },
+    {
+      "epoch": 0.059140629088815615,
+      "grad_norm": 0.446809321641922,
+      "learning_rate": 2e-05,
+      "loss": 0.6751,
+      "step": 1130
+    },
+    {
+      "epoch": 0.05966399748783169,
+      "grad_norm": 0.4745899736881256,
+      "learning_rate": 2e-05,
+      "loss": 0.6564,
+      "step": 1140
+    },
+    {
+      "epoch": 0.060187365886847755,
+      "grad_norm": 0.5119066834449768,
+      "learning_rate": 2e-05,
+      "loss": 0.6618,
+      "step": 1150
+    },
+    {
+      "epoch": 0.06071073428586382,
+      "grad_norm": 0.41898113489151,
+      "learning_rate": 2e-05,
+      "loss": 0.6622,
+      "step": 1160
+    },
+    {
+      "epoch": 0.06123410268487989,
+      "grad_norm": 0.6409226059913635,
+      "learning_rate": 2e-05,
+      "loss": 0.6708,
+      "step": 1170
+    },
+    {
+      "epoch": 0.061757471083895954,
+      "grad_norm": 0.439377099275589,
+      "learning_rate": 2e-05,
+      "loss": 0.6473,
+      "step": 1180
+    },
+    {
+      "epoch": 0.06228083948291202,
+      "grad_norm": 0.4160749316215515,
+      "learning_rate": 2e-05,
+      "loss": 0.6798,
+      "step": 1190
+    },
+    {
+      "epoch": 0.06280420788192809,
+      "grad_norm": 0.5204553604125977,
+      "learning_rate": 2e-05,
+      "loss": 0.6866,
+      "step": 1200
+    },
+    {
+      "epoch": 0.06332757628094415,
+      "grad_norm": 0.48897305130958557,
+      "learning_rate": 2e-05,
+      "loss": 0.6618,
+      "step": 1210
+    },
+    {
+      "epoch": 0.06385094467996022,
+      "grad_norm": 0.45014750957489014,
+      "learning_rate": 2e-05,
+      "loss": 0.6647,
+      "step": 1220
+    },
+    {
+      "epoch": 0.06437431307897629,
+      "grad_norm": 0.4595189690589905,
+      "learning_rate": 2e-05,
+      "loss": 0.6625,
+      "step": 1230
+    },
+    {
+      "epoch": 0.06489768147799235,
+      "grad_norm": 0.49671754240989685,
+      "learning_rate": 2e-05,
+      "loss": 0.663,
+      "step": 1240
+    },
+    {
+      "epoch": 0.06542104987700843,
+      "grad_norm": 0.44070637226104736,
+      "learning_rate": 2e-05,
+      "loss": 0.6613,
+      "step": 1250
+    },
+    {
+      "epoch": 0.0659444182760245,
+      "grad_norm": 0.45897576212882996,
+      "learning_rate": 2e-05,
+      "loss": 0.6516,
+      "step": 1260
+    },
+    {
+      "epoch": 0.06646778667504057,
+      "grad_norm": 0.4450547695159912,
+      "learning_rate": 2e-05,
+      "loss": 0.641,
+      "step": 1270
+    },
+    {
+      "epoch": 0.06699115507405663,
+      "grad_norm": 0.4288750886917114,
+      "learning_rate": 2e-05,
+      "loss": 0.6656,
+      "step": 1280
+    },
+    {
+      "epoch": 0.0675145234730727,
+      "grad_norm": 0.47525206208229065,
+      "learning_rate": 2e-05,
+      "loss": 0.6915,
+      "step": 1290
+    },
+    {
+      "epoch": 0.06803789187208877,
+      "grad_norm": 0.41285258531570435,
+      "learning_rate": 2e-05,
+      "loss": 0.6401,
+      "step": 1300
+    },
+    {
+      "epoch": 0.06856126027110483,
+      "grad_norm": 0.46445295214653015,
+      "learning_rate": 2e-05,
+      "loss": 0.6692,
+      "step": 1310
+    },
+    {
+      "epoch": 0.0690846286701209,
+      "grad_norm": 0.44153061509132385,
+      "learning_rate": 2e-05,
+      "loss": 0.6291,
+      "step": 1320
+    },
+    {
+      "epoch": 0.06960799706913696,
+      "grad_norm": 0.7016943693161011,
+      "learning_rate": 2e-05,
+      "loss": 0.6405,
+      "step": 1330
+    },
+    {
+      "epoch": 0.07013136546815303,
+      "grad_norm": 0.498710572719574,
+      "learning_rate": 2e-05,
+      "loss": 0.6433,
+      "step": 1340
+    },
+    {
+      "epoch": 0.0706547338671691,
+      "grad_norm": 0.4415338337421417,
+      "learning_rate": 2e-05,
+      "loss": 0.673,
+      "step": 1350
+    },
+    {
+      "epoch": 0.07117810226618516,
+      "grad_norm": 0.5377183556556702,
+      "learning_rate": 2e-05,
+      "loss": 0.6556,
+      "step": 1360
+    },
+    {
+      "epoch": 0.07170147066520123,
+      "grad_norm": 0.40461525321006775,
+      "learning_rate": 2e-05,
+      "loss": 0.6384,
+      "step": 1370
+    },
+    {
+      "epoch": 0.0722248390642173,
+      "grad_norm": 0.4240797758102417,
+      "learning_rate": 2e-05,
+      "loss": 0.6467,
+      "step": 1380
+    },
+    {
+      "epoch": 0.07274820746323336,
+      "grad_norm": 0.4052056074142456,
+      "learning_rate": 2e-05,
+      "loss": 0.6448,
+      "step": 1390
+    },
+    {
+      "epoch": 0.07327157586224944,
+      "grad_norm": 0.41825127601623535,
+      "learning_rate": 2e-05,
+      "loss": 0.6465,
+      "step": 1400
+    },
+    {
+      "epoch": 0.07379494426126551,
+      "grad_norm": 0.41024652123451233,
+      "learning_rate": 2e-05,
+      "loss": 0.6428,
+      "step": 1410
+    },
+    {
+      "epoch": 0.07431831266028158,
+      "grad_norm": 0.38469892740249634,
+      "learning_rate": 2e-05,
+      "loss": 0.6652,
+      "step": 1420
+    },
+    {
+      "epoch": 0.07484168105929764,
+      "grad_norm": 0.4639175236225128,
+      "learning_rate": 2e-05,
+      "loss": 0.6503,
+      "step": 1430
+    },
+    {
+      "epoch": 0.07536504945831371,
+      "grad_norm": 0.4508153200149536,
+      "learning_rate": 2e-05,
+      "loss": 0.6464,
+      "step": 1440
+    },
+    {
+      "epoch": 0.07588841785732978,
+      "grad_norm": 0.5173973441123962,
+      "learning_rate": 2e-05,
+      "loss": 0.6475,
+      "step": 1450
+    },
+    {
+      "epoch": 0.07641178625634584,
+      "grad_norm": 0.47181883454322815,
+      "learning_rate": 2e-05,
+      "loss": 0.6662,
+      "step": 1460
+    },
+    {
+      "epoch": 0.07693515465536191,
+      "grad_norm": 0.43690988421440125,
+      "learning_rate": 2e-05,
+      "loss": 0.6459,
+      "step": 1470
+    },
+    {
+      "epoch": 0.07745852305437798,
+      "grad_norm": 0.40047961473464966,
+      "learning_rate": 2e-05,
+      "loss": 0.6808,
+      "step": 1480
+    },
+    {
+      "epoch": 0.07798189145339404,
+      "grad_norm": 0.3962308466434479,
+      "learning_rate": 2e-05,
+      "loss": 0.6249,
+      "step": 1490
+    },
+    {
+      "epoch": 0.07850525985241011,
+      "grad_norm": 0.5075916051864624,
+      "learning_rate": 2e-05,
+      "loss": 0.6562,
+      "step": 1500
+    },
+    {
+      "epoch": 0.07902862825142618,
+      "grad_norm": 0.44439756870269775,
+      "learning_rate": 2e-05,
+      "loss": 0.6744,
+      "step": 1510
+    },
+    {
+      "epoch": 0.07955199665044224,
+      "grad_norm": 0.4172062575817108,
+      "learning_rate": 2e-05,
+      "loss": 0.6328,
+      "step": 1520
+    },
+    {
+      "epoch": 0.08007536504945831,
+      "grad_norm": 0.9272112250328064,
+      "learning_rate": 2e-05,
+      "loss": 0.6554,
+      "step": 1530
+    },
+    {
+      "epoch": 0.08059873344847437,
+      "grad_norm": 0.44365498423576355,
+      "learning_rate": 2e-05,
+      "loss": 0.6567,
+      "step": 1540
+    },
+    {
+      "epoch": 0.08112210184749045,
+      "grad_norm": 0.5808505415916443,
+      "learning_rate": 2e-05,
+      "loss": 0.6671,
+      "step": 1550
+    },
+    {
+      "epoch": 0.08164547024650652,
+      "grad_norm": 0.4238007068634033,
+      "learning_rate": 2e-05,
+      "loss": 0.6655,
+      "step": 1560
+    },
+    {
+      "epoch": 0.08216883864552259,
+      "grad_norm": 0.4158059358596802,
+      "learning_rate": 2e-05,
+      "loss": 0.6502,
+      "step": 1570
+    },
+    {
+      "epoch": 0.08269220704453865,
+      "grad_norm": 0.4266833961009979,
+      "learning_rate": 2e-05,
+      "loss": 0.6389,
+      "step": 1580
+    },
+    {
+      "epoch": 0.08321557544355472,
+      "grad_norm": 0.4183817505836487,
+      "learning_rate": 2e-05,
+      "loss": 0.6297,
+      "step": 1590
+    },
+    {
+      "epoch": 0.08373894384257079,
+      "grad_norm": 0.4242572784423828,
+      "learning_rate": 2e-05,
+      "loss": 0.6575,
+      "step": 1600
+    },
+    {
+      "epoch": 0.08426231224158685,
+      "grad_norm": 0.42805418372154236,
+      "learning_rate": 2e-05,
+      "loss": 0.6316,
+      "step": 1610
+    },
+    {
+      "epoch": 0.08478568064060292,
+      "grad_norm": 0.441866010427475,
+      "learning_rate": 2e-05,
+      "loss": 0.6702,
+      "step": 1620
+    },
+    {
+      "epoch": 0.08530904903961899,
+      "grad_norm": 0.4371151626110077,
+      "learning_rate": 2e-05,
+      "loss": 0.6623,
+      "step": 1630
+    },
+    {
+      "epoch": 0.08583241743863505,
+      "grad_norm": 0.43131211400032043,
+      "learning_rate": 2e-05,
+      "loss": 0.6448,
+      "step": 1640
+    },
+    {
+      "epoch": 0.08635578583765112,
+      "grad_norm": 0.4316512644290924,
+      "learning_rate": 2e-05,
+      "loss": 0.6374,
+      "step": 1650
+    },
+    {
+      "epoch": 0.08687915423666719,
+      "grad_norm": 0.48419904708862305,
+      "learning_rate": 2e-05,
+      "loss": 0.6224,
+      "step": 1660
+    },
+    {
+      "epoch": 0.08740252263568325,
+      "grad_norm": 0.3933804929256439,
+      "learning_rate": 2e-05,
+      "loss": 0.6522,
+      "step": 1670
+    },
+    {
+      "epoch": 0.08792589103469932,
+      "grad_norm": 0.4437639117240906,
+      "learning_rate": 2e-05,
+      "loss": 0.6237,
+      "step": 1680
+    },
+    {
+      "epoch": 0.08844925943371539,
+      "grad_norm": 0.4899040460586548,
+      "learning_rate": 2e-05,
+      "loss": 0.6321,
+      "step": 1690
+    },
+    {
+      "epoch": 0.08897262783273147,
+      "grad_norm": 0.4382249712944031,
+      "learning_rate": 2e-05,
+      "loss": 0.6469,
+      "step": 1700
+    },
+    {
+      "epoch": 0.08949599623174753,
+      "grad_norm": 0.4095931053161621,
+      "learning_rate": 2e-05,
+      "loss": 0.6588,
+      "step": 1710
+    },
+    {
+      "epoch": 0.0900193646307636,
+      "grad_norm": 0.43513184785842896,
+      "learning_rate": 2e-05,
+      "loss": 0.6569,
+      "step": 1720
+    },
+    {
+      "epoch": 0.09054273302977967,
+      "grad_norm": 0.46793821454048157,
+      "learning_rate": 2e-05,
+      "loss": 0.6322,
+      "step": 1730
+    },
+    {
+      "epoch": 0.09106610142879573,
+      "grad_norm": 0.430164098739624,
+      "learning_rate": 2e-05,
+      "loss": 0.6224,
+      "step": 1740
+    },
+    {
+      "epoch": 0.0915894698278118,
+      "grad_norm": 0.3761891722679138,
+      "learning_rate": 2e-05,
+      "loss": 0.606,
+      "step": 1750
+    },
+    {
+      "epoch": 0.09211283822682786,
+      "grad_norm": 0.373146116733551,
+      "learning_rate": 2e-05,
+      "loss": 0.6329,
+      "step": 1760
+    },
+    {
+      "epoch": 0.09263620662584393,
+      "grad_norm": 0.42644017934799194,
+      "learning_rate": 2e-05,
+      "loss": 0.6466,
+      "step": 1770
+    },
+    {
+      "epoch": 0.09315957502486,
+      "grad_norm": 0.40171223878860474,
+      "learning_rate": 2e-05,
+      "loss": 0.6567,
+      "step": 1780
+    },
+    {
+      "epoch": 0.09368294342387606,
+      "grad_norm": 0.5002732872962952,
+      "learning_rate": 2e-05,
+      "loss": 0.6645,
+      "step": 1790
+    },
+    {
+      "epoch": 0.09420631182289213,
+      "grad_norm": 0.45148345828056335,
+      "learning_rate": 2e-05,
+      "loss": 0.6735,
+      "step": 1800
+    },
+    {
+      "epoch": 0.0947296802219082,
+      "grad_norm": 0.4408950209617615,
+      "learning_rate": 2e-05,
+      "loss": 0.64,
+      "step": 1810
+    },
+    {
+      "epoch": 0.09525304862092426,
+      "grad_norm": 0.41224032640457153,
+      "learning_rate": 2e-05,
+      "loss": 0.6483,
+      "step": 1820
+    },
+    {
+      "epoch": 0.09577641701994033,
+      "grad_norm": 0.3857000768184662,
+      "learning_rate": 2e-05,
+      "loss": 0.6591,
+      "step": 1830
+    },
+    {
+      "epoch": 0.09629978541895641,
+      "grad_norm": 0.5142344832420349,
+      "learning_rate": 2e-05,
+      "loss": 0.6613,
+      "step": 1840
+    },
+    {
+      "epoch": 0.09682315381797248,
+      "grad_norm": 0.43707653880119324,
+      "learning_rate": 2e-05,
+      "loss": 0.6463,
+      "step": 1850
+    },
+    {
+      "epoch": 0.09734652221698854,
+      "grad_norm": 0.5951546430587769,
+      "learning_rate": 2e-05,
+      "loss": 0.6634,
+      "step": 1860
+    },
+    {
+      "epoch": 0.09786989061600461,
+      "grad_norm": 0.451498419046402,
+      "learning_rate": 2e-05,
+      "loss": 0.6228,
+      "step": 1870
+    },
+    {
+      "epoch": 0.09839325901502068,
+      "grad_norm": 0.43872302770614624,
+      "learning_rate": 2e-05,
+      "loss": 0.6417,
+      "step": 1880
+    },
+    {
+      "epoch": 0.09891662741403674,
+      "grad_norm": 0.3905869722366333,
+      "learning_rate": 2e-05,
+      "loss": 0.6195,
+      "step": 1890
+    },
+    {
+      "epoch": 0.09943999581305281,
+      "grad_norm": 0.46226704120635986,
+      "learning_rate": 2e-05,
+      "loss": 0.652,
+      "step": 1900
+    },
+    {
+      "epoch": 0.09996336421206888,
+      "grad_norm": 0.4350249469280243,
+      "learning_rate": 2e-05,
+      "loss": 0.6814,
+      "step": 1910
+    },
+    {
+      "epoch": 0.10048673261108494,
+      "grad_norm": 0.547003448009491,
+      "learning_rate": 2e-05,
+      "loss": 0.6376,
+      "step": 1920
+    },
+    {
+      "epoch": 0.10101010101010101,
+      "grad_norm": 0.40520697832107544,
+      "learning_rate": 2e-05,
+      "loss": 0.6281,
+      "step": 1930
+    },
+    {
+      "epoch": 0.10153346940911707,
+      "grad_norm": 0.3886966109275818,
+      "learning_rate": 2e-05,
+      "loss": 0.6349,
+      "step": 1940
+    },
+    {
+      "epoch": 0.10205683780813314,
+      "grad_norm": 0.4416722059249878,
+      "learning_rate": 2e-05,
+      "loss": 0.6626,
+      "step": 1950
+    },
+    {
+      "epoch": 0.10258020620714921,
+      "grad_norm": 0.4272310137748718,
+      "learning_rate": 2e-05,
+      "loss": 0.627,
+      "step": 1960
+    },
+    {
+      "epoch": 0.10310357460616527,
+      "grad_norm": 0.4342178404331207,
+      "learning_rate": 2e-05,
+      "loss": 0.6616,
+      "step": 1970
+    },
+    {
+      "epoch": 0.10362694300518134,
+      "grad_norm": 0.4413588047027588,
+      "learning_rate": 2e-05,
+      "loss": 0.6602,
+      "step": 1980
+    },
+    {
+      "epoch": 0.10415031140419742,
+      "grad_norm": 0.39807382225990295,
+      "learning_rate": 2e-05,
+      "loss": 0.6184,
+      "step": 1990
+    },
+    {
+      "epoch": 0.10467367980321349,
+      "grad_norm": 0.39922189712524414,
+      "learning_rate": 2e-05,
+      "loss": 0.6348,
+      "step": 2000
+    },
+    {
+      "epoch": 0.10519704820222955,
+      "grad_norm": 0.4405508041381836,
+      "learning_rate": 2e-05,
+      "loss": 0.6527,
+      "step": 2010
+    },
+    {
+      "epoch": 0.10572041660124562,
+      "grad_norm": 0.4154644310474396,
+      "learning_rate": 2e-05,
+      "loss": 0.6383,
+      "step": 2020
+    },
+    {
+      "epoch": 0.10624378500026169,
+      "grad_norm": 0.4121659994125366,
+      "learning_rate": 2e-05,
+      "loss": 0.6337,
+      "step": 2030
+    },
+    {
+      "epoch": 0.10676715339927775,
+      "grad_norm": 0.4231645166873932,
+      "learning_rate": 2e-05,
+      "loss": 0.6648,
+      "step": 2040
+    },
+    {
+      "epoch": 0.10729052179829382,
+      "grad_norm": 0.5495308041572571,
+      "learning_rate": 2e-05,
+      "loss": 0.6655,
+      "step": 2050
+    },
+    {
+      "epoch": 0.10781389019730989,
+      "grad_norm": 0.4083961546421051,
+      "learning_rate": 2e-05,
+      "loss": 0.6091,
+      "step": 2060
+    },
+    {
+      "epoch": 0.10833725859632595,
+      "grad_norm": 0.394467830657959,
+      "learning_rate": 2e-05,
+      "loss": 0.6472,
+      "step": 2070
+    },
+    {
+      "epoch": 0.10886062699534202,
+      "grad_norm": 0.4051300585269928,
+      "learning_rate": 2e-05,
+      "loss": 0.6295,
+      "step": 2080
+    },
+    {
+      "epoch": 0.10938399539435809,
+      "grad_norm": 0.44103720784187317,
+      "learning_rate": 2e-05,
+      "loss": 0.6316,
+      "step": 2090
+    },
+    {
+      "epoch": 0.10990736379337415,
+      "grad_norm": 0.4452396333217621,
+      "learning_rate": 2e-05,
+      "loss": 0.6367,
+      "step": 2100
+    },
+    {
+      "epoch": 0.11043073219239022,
+      "grad_norm": 0.5510708689689636,
+      "learning_rate": 2e-05,
+      "loss": 0.6258,
+      "step": 2110
+    },
+    {
+      "epoch": 0.11095410059140629,
+      "grad_norm": 0.5372933149337769,
+      "learning_rate": 2e-05,
+      "loss": 0.6476,
+      "step": 2120
+    },
+    {
+      "epoch": 0.11147746899042235,
+      "grad_norm": 0.4383063316345215,
+      "learning_rate": 2e-05,
+      "loss": 0.6605,
+      "step": 2130
+    },
+    {
+      "epoch": 0.11200083738943843,
+      "grad_norm": 0.4241836667060852,
+      "learning_rate": 2e-05,
+      "loss": 0.6362,
+      "step": 2140
+    },
+    {
+      "epoch": 0.1125242057884545,
+      "grad_norm": 0.442451149225235,
+      "learning_rate": 2e-05,
+      "loss": 0.6433,
+      "step": 2150
+    },
+    {
+      "epoch": 0.11304757418747056,
+      "grad_norm": 0.41505861282348633,
+      "learning_rate": 2e-05,
+      "loss": 0.6239,
+      "step": 2160
+    },
+    {
+      "epoch": 0.11357094258648663,
+      "grad_norm": 0.43439918756484985,
+      "learning_rate": 2e-05,
+      "loss": 0.6274,
+      "step": 2170
+    },
+    {
+      "epoch": 0.1140943109855027,
+      "grad_norm": 0.43937069177627563,
+      "learning_rate": 2e-05,
+      "loss": 0.6414,
+      "step": 2180
+    },
+    {
+      "epoch": 0.11461767938451876,
+      "grad_norm": 0.43210020661354065,
+      "learning_rate": 2e-05,
+      "loss": 0.6257,
+      "step": 2190
+    },
+    {
+      "epoch": 0.11514104778353483,
+      "grad_norm": 0.49416911602020264,
+      "learning_rate": 2e-05,
+      "loss": 0.6442,
+      "step": 2200
+    },
+    {
+      "epoch": 0.1156644161825509,
+      "grad_norm": 0.40978503227233887,
+      "learning_rate": 2e-05,
+      "loss": 0.6329,
+      "step": 2210
+    },
+    {
+      "epoch": 0.11618778458156696,
+      "grad_norm": 0.4340236485004425,
+      "learning_rate": 2e-05,
+      "loss": 0.6347,
+      "step": 2220
+    },
+    {
+      "epoch": 0.11671115298058303,
+      "grad_norm": 0.41441839933395386,
+      "learning_rate": 2e-05,
+      "loss": 0.6205,
+      "step": 2230
+    },
+    {
+      "epoch": 0.1172345213795991,
+      "grad_norm": 0.4365447461605072,
+      "learning_rate": 2e-05,
+      "loss": 0.6329,
+      "step": 2240
+    },
+    {
+      "epoch": 0.11775788977861516,
+      "grad_norm": 0.44118615984916687,
+      "learning_rate": 2e-05,
+      "loss": 0.6549,
+      "step": 2250
+    },
+    {
+      "epoch": 0.11828125817763123,
+      "grad_norm": 0.40321072936058044,
+      "learning_rate": 2e-05,
+      "loss": 0.6284,
+      "step": 2260
+    },
+    {
+      "epoch": 0.1188046265766473,
+      "grad_norm": 0.4358213543891907,
+      "learning_rate": 2e-05,
+      "loss": 0.6375,
+      "step": 2270
+    },
+    {
+      "epoch": 0.11932799497566338,
+      "grad_norm": 0.42884060740470886,
+      "learning_rate": 2e-05,
+      "loss": 0.6452,
+      "step": 2280
+    },
+    {
+      "epoch": 0.11985136337467944,
+      "grad_norm": 0.43688151240348816,
+      "learning_rate": 2e-05,
+      "loss": 0.6433,
+      "step": 2290
+    },
+    {
+      "epoch": 0.12037473177369551,
+      "grad_norm": 0.42834022641181946,
+      "learning_rate": 2e-05,
+      "loss": 0.6461,
+      "step": 2300
+    },
+    {
+      "epoch": 0.12089810017271158,
+      "grad_norm": 0.44969525933265686,
+      "learning_rate": 2e-05,
+      "loss": 0.6704,
+      "step": 2310
+    },
+    {
+      "epoch": 0.12142146857172764,
+      "grad_norm": 0.4053316116333008,
+      "learning_rate": 2e-05,
+      "loss": 0.6188,
+      "step": 2320
+    },
+    {
+      "epoch": 0.12194483697074371,
+      "grad_norm": 0.3716852366924286,
+      "learning_rate": 2e-05,
+      "loss": 0.6249,
+      "step": 2330
+    },
+    {
+      "epoch": 0.12246820536975978,
+      "grad_norm": 0.4359619617462158,
+      "learning_rate": 2e-05,
+      "loss": 0.6571,
+      "step": 2340
+    },
+    {
+      "epoch": 0.12299157376877584,
+      "grad_norm": 0.38862133026123047,
+      "learning_rate": 2e-05,
+      "loss": 0.641,
+      "step": 2350
+    },
+    {
+      "epoch": 0.12351494216779191,
+      "grad_norm": 0.4286133646965027,
+      "learning_rate": 2e-05,
+      "loss": 0.6514,
+      "step": 2360
+    },
+    {
+      "epoch": 0.12403831056680797,
+      "grad_norm": 0.4758356809616089,
+      "learning_rate": 2e-05,
+      "loss": 0.628,
+      "step": 2370
+    },
+    {
+      "epoch": 0.12456167896582404,
+      "grad_norm": 0.4074471592903137,
+      "learning_rate": 2e-05,
+      "loss": 0.6161,
+      "step": 2380
+    },
+    {
+      "epoch": 0.1250850473648401,
+      "grad_norm": 0.4395975172519684,
+      "learning_rate": 2e-05,
+      "loss": 0.6514,
+      "step": 2390
+    },
+    {
+      "epoch": 0.12560841576385617,
+      "grad_norm": 0.4212956130504608,
+      "learning_rate": 2e-05,
+      "loss": 0.6459,
+      "step": 2400
+    },
+    {
+      "epoch": 0.12613178416287224,
+      "grad_norm": 0.43518325686454773,
+      "learning_rate": 2e-05,
+      "loss": 0.6631,
+      "step": 2410
+    },
+    {
+      "epoch": 0.1266551525618883,
+      "grad_norm": 0.41669315099716187,
+      "learning_rate": 2e-05,
+      "loss": 0.6433,
+      "step": 2420
+    },
+    {
+      "epoch": 0.12717852096090437,
+      "grad_norm": 0.47521886229515076,
+      "learning_rate": 2e-05,
+      "loss": 0.6008,
+      "step": 2430
+    },
+    {
+      "epoch": 0.12770188935992044,
+      "grad_norm": 0.5306794047355652,
+      "learning_rate": 2e-05,
+      "loss": 0.6499,
+      "step": 2440
+    },
+    {
+      "epoch": 0.1282252577589365,
+      "grad_norm": 0.4078957438468933,
+      "learning_rate": 2e-05,
+      "loss": 0.632,
+      "step": 2450
+    },
+    {
+      "epoch": 0.12874862615795257,
+      "grad_norm": 0.4263654351234436,
+      "learning_rate": 2e-05,
+      "loss": 0.6367,
+      "step": 2460
+    },
+    {
+      "epoch": 0.12927199455696864,
+      "grad_norm": 0.4139572083950043,
+      "learning_rate": 2e-05,
+      "loss": 0.642,
+      "step": 2470
+    },
+    {
+      "epoch": 0.1297953629559847,
+      "grad_norm": 0.38871797919273376,
+      "learning_rate": 2e-05,
+      "loss": 0.6342,
+      "step": 2480
+    },
+    {
+      "epoch": 0.13031873135500077,
+      "grad_norm": 0.429225891828537,
+      "learning_rate": 2e-05,
+      "loss": 0.6249,
+      "step": 2490
+    },
+    {
+      "epoch": 0.13084209975401687,
+      "grad_norm": 0.45410633087158203,
+      "learning_rate": 2e-05,
+      "loss": 0.6395,
+      "step": 2500
+    },
+    {
+      "epoch": 0.13136546815303293,
+      "grad_norm": 0.45400431752204895,
+      "learning_rate": 2e-05,
+      "loss": 0.6244,
+      "step": 2510
+    },
+    {
+      "epoch": 0.131888836552049,
+      "grad_norm": 0.4065890312194824,
+      "learning_rate": 2e-05,
+      "loss": 0.6394,
+      "step": 2520
+    },
+    {
+      "epoch": 0.13241220495106507,
+      "grad_norm": 0.4191737771034241,
+      "learning_rate": 2e-05,
+      "loss": 0.6409,
+      "step": 2530
+    },
+    {
+      "epoch": 0.13293557335008113,
+      "grad_norm": 0.45096731185913086,
+      "learning_rate": 2e-05,
+      "loss": 0.6208,
+      "step": 2540
+    },
+    {
+      "epoch": 0.1334589417490972,
+      "grad_norm": 0.46258044242858887,
+      "learning_rate": 2e-05,
+      "loss": 0.6395,
+      "step": 2550
+    },
+    {
+      "epoch": 0.13398231014811327,
+      "grad_norm": 0.4282276928424835,
+      "learning_rate": 2e-05,
+      "loss": 0.601,
+      "step": 2560
+    },
+    {
+      "epoch": 0.13450567854712933,
+      "grad_norm": 0.41605663299560547,
+      "learning_rate": 2e-05,
+      "loss": 0.6501,
+      "step": 2570
+    },
+    {
+      "epoch": 0.1350290469461454,
+      "grad_norm": 0.41660454869270325,
+      "learning_rate": 2e-05,
+      "loss": 0.6546,
+      "step": 2580
+    },
+    {
+      "epoch": 0.13555241534516146,
+      "grad_norm": 0.4275190234184265,
+      "learning_rate": 2e-05,
+      "loss": 0.6458,
+      "step": 2590
+    },
+    {
+      "epoch": 0.13607578374417753,
+      "grad_norm": 0.4305119514465332,
+      "learning_rate": 2e-05,
+      "loss": 0.6403,
+      "step": 2600
+    },
+    {
+      "epoch": 0.1365991521431936,
+      "grad_norm": 0.42454832792282104,
+      "learning_rate": 2e-05,
+      "loss": 0.6368,
+      "step": 2610
+    },
+    {
+      "epoch": 0.13712252054220966,
+      "grad_norm": 0.44257181882858276,
+      "learning_rate": 2e-05,
+      "loss": 0.6343,
+      "step": 2620
+    },
+    {
+      "epoch": 0.13764588894122573,
+      "grad_norm": 0.3995290696620941,
+      "learning_rate": 2e-05,
+      "loss": 0.6616,
+      "step": 2630
+    },
+    {
+      "epoch": 0.1381692573402418,
+      "grad_norm": 0.4199754297733307,
+      "learning_rate": 2e-05,
+      "loss": 0.6286,
+      "step": 2640
+    },
+    {
+      "epoch": 0.13869262573925786,
+      "grad_norm": 0.42020726203918457,
+      "learning_rate": 2e-05,
+      "loss": 0.6371,
+      "step": 2650
+    },
+    {
+      "epoch": 0.13921599413827393,
+      "grad_norm": 0.39368361234664917,
+      "learning_rate": 2e-05,
+      "loss": 0.6517,
+      "step": 2660
+    },
+    {
+      "epoch": 0.13973936253729,
+      "grad_norm": 0.4302023947238922,
+      "learning_rate": 2e-05,
+      "loss": 0.6139,
+      "step": 2670
+    },
+    {
+      "epoch": 0.14026273093630606,
+      "grad_norm": 0.49658092856407166,
+      "learning_rate": 2e-05,
+      "loss": 0.6565,
+      "step": 2680
+    },
+    {
+      "epoch": 0.14078609933532213,
+      "grad_norm": 0.3941155672073364,
+      "learning_rate": 2e-05,
+      "loss": 0.6527,
+      "step": 2690
+    },
+    {
+      "epoch": 0.1413094677343382,
+      "grad_norm": 0.43608132004737854,
+      "learning_rate": 2e-05,
+      "loss": 0.6309,
+      "step": 2700
+    },
+    {
+      "epoch": 0.14183283613335426,
+      "grad_norm": 0.4120804965496063,
+      "learning_rate": 2e-05,
+      "loss": 0.6326,
+      "step": 2710
+    },
+    {
+      "epoch": 0.14235620453237033,
+      "grad_norm": 0.4378657639026642,
+      "learning_rate": 2e-05,
+      "loss": 0.6479,
+      "step": 2720
+    },
+    {
+      "epoch": 0.1428795729313864,
+      "grad_norm": 0.39626598358154297,
+      "learning_rate": 2e-05,
+      "loss": 0.6537,
+      "step": 2730
+    },
+    {
+      "epoch": 0.14340294133040246,
+      "grad_norm": 0.40195828676223755,
+      "learning_rate": 2e-05,
+      "loss": 0.6075,
+      "step": 2740
+    },
+    {
+      "epoch": 0.14392630972941853,
+      "grad_norm": 0.3953752815723419,
+      "learning_rate": 2e-05,
+      "loss": 0.6179,
+      "step": 2750
+    },
+    {
+      "epoch": 0.1444496781284346,
+      "grad_norm": 0.4433000385761261,
+      "learning_rate": 2e-05,
+      "loss": 0.6743,
+      "step": 2760
+    },
+    {
+      "epoch": 0.14497304652745066,
+      "grad_norm": 0.42893922328948975,
+      "learning_rate": 2e-05,
+      "loss": 0.6272,
+      "step": 2770
+    },
+    {
+      "epoch": 0.14549641492646673,
+      "grad_norm": 0.4065708518028259,
+      "learning_rate": 2e-05,
+      "loss": 0.6417,
+      "step": 2780
+    },
+    {
+      "epoch": 0.1460197833254828,
+      "grad_norm": 0.4181337058544159,
+      "learning_rate": 2e-05,
+      "loss": 0.6392,
+      "step": 2790
+    },
+    {
+      "epoch": 0.1465431517244989,
+      "grad_norm": 0.4179243743419647,
+      "learning_rate": 2e-05,
+      "loss": 0.6026,
+      "step": 2800
+    },
+    {
+      "epoch": 0.14706652012351495,
+      "grad_norm": 0.46388933062553406,
+      "learning_rate": 2e-05,
+      "loss": 0.6335,
+      "step": 2810
+    },
+    {
+      "epoch": 0.14758988852253102,
+      "grad_norm": 0.4350208640098572,
+      "learning_rate": 2e-05,
+      "loss": 0.6349,
+      "step": 2820
+    },
+    {
+      "epoch": 0.1481132569215471,
+      "grad_norm": 0.4281371831893921,
+      "learning_rate": 2e-05,
+      "loss": 0.5927,
+      "step": 2830
+    },
+    {
+      "epoch": 0.14863662532056315,
+      "grad_norm": 0.3796765208244324,
+      "learning_rate": 2e-05,
+      "loss": 0.6244,
+      "step": 2840
+    },
+    {
+      "epoch": 0.14915999371957922,
+      "grad_norm": 0.400057315826416,
+      "learning_rate": 2e-05,
+      "loss": 0.6083,
+      "step": 2850
+    },
+    {
+      "epoch": 0.1496833621185953,
+      "grad_norm": 0.5580277442932129,
+      "learning_rate": 2e-05,
+      "loss": 0.6088,
+      "step": 2860
+    },
+    {
+      "epoch": 0.15020673051761135,
+      "grad_norm": 0.4220695197582245,
+      "learning_rate": 2e-05,
+      "loss": 0.6339,
+      "step": 2870
+    },
+    {
+      "epoch": 0.15073009891662742,
+      "grad_norm": 0.4196612536907196,
+      "learning_rate": 2e-05,
+      "loss": 0.6415,
+      "step": 2880
+    },
+    {
+      "epoch": 0.1512534673156435,
+      "grad_norm": 0.4167044162750244,
+      "learning_rate": 2e-05,
+      "loss": 0.6276,
+      "step": 2890
+    },
+    {
+      "epoch": 0.15177683571465955,
+      "grad_norm": 0.4121061861515045,
+      "learning_rate": 2e-05,
+      "loss": 0.6311,
+      "step": 2900
+    },
+    {
+      "epoch": 0.15230020411367562,
+      "grad_norm": 0.4583412706851959,
+      "learning_rate": 2e-05,
+      "loss": 0.6415,
+      "step": 2910
+    },
+    {
+      "epoch": 0.15282357251269169,
+      "grad_norm": 0.4203738272190094,
+      "learning_rate": 2e-05,
+      "loss": 0.657,
+      "step": 2920
+    },
+    {
+      "epoch": 0.15334694091170775,
+      "grad_norm": 0.4278572201728821,
+      "learning_rate": 2e-05,
+      "loss": 0.6457,
+      "step": 2930
+    },
+    {
+      "epoch": 0.15387030931072382,
+      "grad_norm": 0.41517725586891174,
+      "learning_rate": 2e-05,
+      "loss": 0.6506,
+      "step": 2940
+    },
+    {
+      "epoch": 0.15439367770973988,
+      "grad_norm": 0.39440619945526123,
+      "learning_rate": 2e-05,
+      "loss": 0.6247,
+      "step": 2950
+    },
+    {
+      "epoch": 0.15491704610875595,
+      "grad_norm": 0.3762326240539551,
+      "learning_rate": 2e-05,
+      "loss": 0.6464,
+      "step": 2960
+    },
+    {
+      "epoch": 0.15544041450777202,
+      "grad_norm": 0.43617796897888184,
+      "learning_rate": 2e-05,
+      "loss": 0.6162,
+      "step": 2970
+    },
+    {
+      "epoch": 0.15596378290678808,
+      "grad_norm": 0.41636911034584045,
+      "learning_rate": 2e-05,
+      "loss": 0.6327,
+      "step": 2980
+    },
+    {
+      "epoch": 0.15648715130580415,
+      "grad_norm": 0.4136853814125061,
+      "learning_rate": 2e-05,
+      "loss": 0.6292,
+      "step": 2990
+    },
+    {
+      "epoch": 0.15701051970482022,
+      "grad_norm": 0.44505634903907776,
+      "learning_rate": 2e-05,
+      "loss": 0.646,
+      "step": 3000
+    },
+    {
+      "epoch": 0.15753388810383628,
+      "grad_norm": 0.42681798338890076,
+      "learning_rate": 2e-05,
+      "loss": 0.6274,
+      "step": 3010
+    },
+    {
+      "epoch": 0.15805725650285235,
+      "grad_norm": 0.3740960657596588,
+      "learning_rate": 2e-05,
+      "loss": 0.6233,
+      "step": 3020
+    },
+    {
+      "epoch": 0.15858062490186842,
+      "grad_norm": 0.4044531285762787,
+      "learning_rate": 2e-05,
+      "loss": 0.655,
+      "step": 3030
+    },
+    {
+      "epoch": 0.15910399330088448,
+      "grad_norm": 0.38664835691452026,
+      "learning_rate": 2e-05,
+      "loss": 0.6201,
+      "step": 3040
+    },
+    {
+      "epoch": 0.15962736169990055,
+      "grad_norm": 0.5274628400802612,
+      "learning_rate": 2e-05,
+      "loss": 0.6374,
+      "step": 3050
+    },
+    {
+      "epoch": 0.16015073009891662,
+      "grad_norm": 0.4279114305973053,
+      "learning_rate": 2e-05,
+      "loss": 0.6433,
+      "step": 3060
+    },
+    {
+      "epoch": 0.16067409849793268,
+      "grad_norm": 0.396634578704834,
+      "learning_rate": 2e-05,
+      "loss": 0.6155,
+      "step": 3070
+    },
+    {
+      "epoch": 0.16119746689694875,
+      "grad_norm": 0.41256171464920044,
+      "learning_rate": 2e-05,
+      "loss": 0.6274,
+      "step": 3080
+    },
+    {
+      "epoch": 0.16172083529596484,
+      "grad_norm": 0.47806742787361145,
+      "learning_rate": 2e-05,
+      "loss": 0.6288,
+      "step": 3090
+    },
+    {
+      "epoch": 0.1622442036949809,
+      "grad_norm": 0.41482001543045044,
+      "learning_rate": 2e-05,
+      "loss": 0.6097,
+      "step": 3100
+    },
+    {
+      "epoch": 0.16276757209399698,
+      "grad_norm": 0.4124285876750946,
+      "learning_rate": 2e-05,
+      "loss": 0.6381,
+      "step": 3110
+    },
+    {
+      "epoch": 0.16329094049301304,
+      "grad_norm": 0.39308762550354004,
+      "learning_rate": 2e-05,
+      "loss": 0.6418,
+      "step": 3120
+    },
+    {
+      "epoch": 0.1638143088920291,
+      "grad_norm": 0.5976954698562622,
+      "learning_rate": 2e-05,
+      "loss": 0.6403,
+      "step": 3130
+    },
+    {
+      "epoch": 0.16433767729104518,
+      "grad_norm": 0.4556901752948761,
+      "learning_rate": 2e-05,
+      "loss": 0.6415,
+      "step": 3140
+    },
+    {
+      "epoch": 0.16486104569006124,
+      "grad_norm": 0.4634074866771698,
+      "learning_rate": 2e-05,
+      "loss": 0.6142,
+      "step": 3150
+    },
+    {
+      "epoch": 0.1653844140890773,
+      "grad_norm": 0.4238816797733307,
+      "learning_rate": 2e-05,
+      "loss": 0.6528,
+      "step": 3160
+    },
+    {
+      "epoch": 0.16590778248809337,
+      "grad_norm": 0.5388637185096741,
+      "learning_rate": 2e-05,
+      "loss": 0.6723,
+      "step": 3170
+    },
+    {
+      "epoch": 0.16643115088710944,
+      "grad_norm": 0.3934621512889862,
+      "learning_rate": 2e-05,
+      "loss": 0.6347,
+      "step": 3180
+    },
+    {
+      "epoch": 0.1669545192861255,
+      "grad_norm": 0.3961109519004822,
+      "learning_rate": 2e-05,
+      "loss": 0.6345,
+      "step": 3190
+    },
+    {
+      "epoch": 0.16747788768514157,
+      "grad_norm": 0.41134336590766907,
+      "learning_rate": 2e-05,
+      "loss": 0.6227,
+      "step": 3200
+    },
+    {
+      "epoch": 0.16800125608415764,
+      "grad_norm": 0.4064910411834717,
+      "learning_rate": 2e-05,
+      "loss": 0.6087,
+      "step": 3210
+    },
+    {
+      "epoch": 0.1685246244831737,
+      "grad_norm": 0.411873459815979,
+      "learning_rate": 2e-05,
+      "loss": 0.6291,
+      "step": 3220
+    },
+    {
+      "epoch": 0.16904799288218977,
+      "grad_norm": 0.41122299432754517,
+      "learning_rate": 2e-05,
+      "loss": 0.6217,
+      "step": 3230
+    },
+    {
+      "epoch": 0.16957136128120584,
+      "grad_norm": 0.4552735686302185,
+      "learning_rate": 2e-05,
+      "loss": 0.6334,
+      "step": 3240
+    },
+    {
+      "epoch": 0.1700947296802219,
+      "grad_norm": 0.457132488489151,
+      "learning_rate": 2e-05,
+      "loss": 0.6213,
+      "step": 3250
+    },
+    {
+      "epoch": 0.17061809807923797,
+      "grad_norm": 0.38603243231773376,
+      "learning_rate": 2e-05,
+      "loss": 0.6363,
+      "step": 3260
+    },
+    {
+      "epoch": 0.17114146647825404,
+      "grad_norm": 0.4321597218513489,
+      "learning_rate": 2e-05,
+      "loss": 0.6194,
+      "step": 3270
+    },
+    {
+      "epoch": 0.1716648348772701,
+      "grad_norm": 0.43034639954566956,
+      "learning_rate": 2e-05,
+      "loss": 0.6409,
+      "step": 3280
+    },
+    {
+      "epoch": 0.17218820327628617,
+      "grad_norm": 0.42103973031044006,
+      "learning_rate": 2e-05,
+      "loss": 0.6154,
+      "step": 3290
+    },
+    {
+      "epoch": 0.17271157167530224,
+      "grad_norm": 0.3927100896835327,
+      "learning_rate": 2e-05,
+      "loss": 0.6313,
+      "step": 3300
+    },
+    {
+      "epoch": 0.1732349400743183,
+      "grad_norm": 0.4339587688446045,
+      "learning_rate": 2e-05,
+      "loss": 0.614,
+      "step": 3310
+    },
+    {
+      "epoch": 0.17375830847333437,
+      "grad_norm": 0.40991151332855225,
+      "learning_rate": 2e-05,
+      "loss": 0.6456,
+      "step": 3320
+    },
+    {
+      "epoch": 0.17428167687235044,
+      "grad_norm": 0.4325578808784485,
+      "learning_rate": 2e-05,
+      "loss": 0.6041,
+      "step": 3330
+    },
+    {
+      "epoch": 0.1748050452713665,
+      "grad_norm": 0.41213691234588623,
+      "learning_rate": 2e-05,
+      "loss": 0.6349,
+      "step": 3340
+    },
+    {
+      "epoch": 0.17532841367038257,
+      "grad_norm": 0.47782137989997864,
+      "learning_rate": 2e-05,
+      "loss": 0.636,
+      "step": 3350
+    },
+    {
+      "epoch": 0.17585178206939864,
+      "grad_norm": 0.42506250739097595,
+      "learning_rate": 2e-05,
+      "loss": 0.626,
+      "step": 3360
+    },
+    {
+      "epoch": 0.1763751504684147,
+      "grad_norm": 0.3940761983394623,
+      "learning_rate": 2e-05,
+      "loss": 0.6193,
+      "step": 3370
+    },
+    {
+      "epoch": 0.17689851886743077,
+      "grad_norm": 0.4069708287715912,
+      "learning_rate": 2e-05,
+      "loss": 0.6365,
+      "step": 3380
+    },
+    {
+      "epoch": 0.17742188726644686,
+      "grad_norm": 0.504355788230896,
+      "learning_rate": 2e-05,
+      "loss": 0.5985,
+      "step": 3390
+    },
+    {
+      "epoch": 0.17794525566546293,
+      "grad_norm": 0.3881029486656189,
+      "learning_rate": 2e-05,
+      "loss": 0.6328,
+      "step": 3400
+    },
+    {
+      "epoch": 0.178468624064479,
+      "grad_norm": 2.036409854888916,
+      "learning_rate": 2e-05,
+      "loss": 0.6237,
+      "step": 3410
+    },
+    {
+      "epoch": 0.17899199246349506,
+      "grad_norm": 0.41256919503211975,
+      "learning_rate": 2e-05,
+      "loss": 0.6066,
+      "step": 3420
+    },
+    {
+      "epoch": 0.17951536086251113,
+      "grad_norm": 0.4488495886325836,
+      "learning_rate": 2e-05,
+      "loss": 0.6247,
+      "step": 3430
+    },
+    {
+      "epoch": 0.1800387292615272,
+      "grad_norm": 0.4205232560634613,
+      "learning_rate": 2e-05,
+      "loss": 0.6094,
+      "step": 3440
+    },
+    {
+      "epoch": 0.18056209766054326,
+      "grad_norm": 0.4089616537094116,
+      "learning_rate": 2e-05,
+      "loss": 0.6264,
+      "step": 3450
+    },
+    {
+      "epoch": 0.18108546605955933,
+      "grad_norm": 0.37557995319366455,
+      "learning_rate": 2e-05,
+      "loss": 0.6322,
+      "step": 3460
+    },
+    {
+      "epoch": 0.1816088344585754,
+      "grad_norm": 0.39397501945495605,
+      "learning_rate": 2e-05,
+      "loss": 0.6647,
+      "step": 3470
+    },
+    {
+      "epoch": 0.18213220285759146,
+      "grad_norm": 0.4009655714035034,
+      "learning_rate": 2e-05,
+      "loss": 0.6203,
+      "step": 3480
+    },
+    {
+      "epoch": 0.18265557125660753,
+      "grad_norm": 0.40983593463897705,
+      "learning_rate": 2e-05,
+      "loss": 0.6684,
+      "step": 3490
+    },
+    {
+      "epoch": 0.1831789396556236,
+      "grad_norm": 0.41632992029190063,
+      "learning_rate": 2e-05,
+      "loss": 0.6055,
+      "step": 3500
+    },
+    {
+      "epoch": 0.18370230805463966,
+      "grad_norm": 0.42826464772224426,
+      "learning_rate": 2e-05,
+      "loss": 0.6114,
+      "step": 3510
+    },
+    {
+      "epoch": 0.18422567645365573,
+      "grad_norm": 0.4451156556606293,
+      "learning_rate": 2e-05,
+      "loss": 0.6164,
+      "step": 3520
+    },
+    {
+      "epoch": 0.1847490448526718,
+      "grad_norm": 0.7244383692741394,
+      "learning_rate": 2e-05,
+      "loss": 0.643,
+      "step": 3530
+    },
+    {
+      "epoch": 0.18527241325168786,
+      "grad_norm": 0.41192492842674255,
+      "learning_rate": 2e-05,
+      "loss": 0.6389,
+      "step": 3540
+    },
+    {
+      "epoch": 0.18579578165070393,
+      "grad_norm": 0.42937004566192627,
+      "learning_rate": 2e-05,
+      "loss": 0.6449,
+      "step": 3550
+    },
+    {
+      "epoch": 0.18631915004972,
+      "grad_norm": 0.4111078977584839,
+      "learning_rate": 2e-05,
+      "loss": 0.6202,
+      "step": 3560
+    },
+    {
+      "epoch": 0.18684251844873606,
+      "grad_norm": 0.4172969460487366,
+      "learning_rate": 2e-05,
+      "loss": 0.6188,
+      "step": 3570
+    },
+    {
+      "epoch": 0.18736588684775213,
+      "grad_norm": 0.38246381282806396,
+      "learning_rate": 2e-05,
+      "loss": 0.609,
+      "step": 3580
+    },
+    {
+      "epoch": 0.1878892552467682,
+      "grad_norm": 0.467737078666687,
+      "learning_rate": 2e-05,
+      "loss": 0.6177,
+      "step": 3590
+    },
+    {
+      "epoch": 0.18841262364578426,
+      "grad_norm": 0.3787391185760498,
+      "learning_rate": 2e-05,
+      "loss": 0.6156,
+      "step": 3600
+    },
+    {
+      "epoch": 0.18893599204480033,
+      "grad_norm": 0.39074644446372986,
+      "learning_rate": 2e-05,
+      "loss": 0.6275,
+      "step": 3610
+    },
+    {
+      "epoch": 0.1894593604438164,
+      "grad_norm": 0.4353400468826294,
+      "learning_rate": 2e-05,
+      "loss": 0.613,
+      "step": 3620
+    },
+    {
+      "epoch": 0.18998272884283246,
+      "grad_norm": 0.4572504758834839,
+      "learning_rate": 2e-05,
+      "loss": 0.6243,
+      "step": 3630
+    },
+    {
+      "epoch": 0.19050609724184853,
+      "grad_norm": 0.3868942856788635,
+      "learning_rate": 2e-05,
+      "loss": 0.6303,
+      "step": 3640
+    },
+    {
+      "epoch": 0.1910294656408646,
+      "grad_norm": 0.41320866346359253,
+      "learning_rate": 2e-05,
+      "loss": 0.6255,
+      "step": 3650
+    },
+    {
+      "epoch": 0.19155283403988066,
+      "grad_norm": 0.39754143357276917,
+      "learning_rate": 2e-05,
+      "loss": 0.6064,
+      "step": 3660
+    },
+    {
+      "epoch": 0.19207620243889673,
+      "grad_norm": 0.45954880118370056,
+      "learning_rate": 2e-05,
+      "loss": 0.6355,
+      "step": 3670
+    },
+    {
+      "epoch": 0.19259957083791282,
+      "grad_norm": 0.42518895864486694,
+      "learning_rate": 2e-05,
+      "loss": 0.6339,
+      "step": 3680
+    },
+    {
+      "epoch": 0.1931229392369289,
+      "grad_norm": 0.40126708149909973,
+      "learning_rate": 2e-05,
+      "loss": 0.6226,
+      "step": 3690
+    },
+    {
+      "epoch": 0.19364630763594495,
+      "grad_norm": 0.5110933184623718,
+      "learning_rate": 2e-05,
+      "loss": 0.6236,
+      "step": 3700
+    },
+    {
+      "epoch": 0.19416967603496102,
+      "grad_norm": 0.4530631899833679,
+      "learning_rate": 2e-05,
+      "loss": 0.6132,
+      "step": 3710
+    },
+    {
+      "epoch": 0.19469304443397709,
+      "grad_norm": 0.40570759773254395,
+      "learning_rate": 2e-05,
+      "loss": 0.6205,
+      "step": 3720
+    },
+    {
+      "epoch": 0.19521641283299315,
+      "grad_norm": 0.4424304962158203,
+      "learning_rate": 2e-05,
+      "loss": 0.6318,
+      "step": 3730
+    },
+    {
+      "epoch": 0.19573978123200922,
+      "grad_norm": 0.4104612171649933,
+      "learning_rate": 2e-05,
+      "loss": 0.6112,
+      "step": 3740
+    },
+    {
+      "epoch": 0.19626314963102529,
+      "grad_norm": 0.4605982303619385,
+      "learning_rate": 2e-05,
+      "loss": 0.63,
+      "step": 3750
+    },
+    {
+      "epoch": 0.19678651803004135,
+      "grad_norm": 0.41476568579673767,
+      "learning_rate": 2e-05,
+      "loss": 0.6301,
+      "step": 3760
+    },
+    {
+      "epoch": 0.19730988642905742,
+      "grad_norm": 0.4510325491428375,
+      "learning_rate": 2e-05,
+      "loss": 0.6311,
+      "step": 3770
+    },
+    {
+      "epoch": 0.19783325482807348,
+      "grad_norm": 0.3622564673423767,
+      "learning_rate": 2e-05,
+      "loss": 0.6028,
+      "step": 3780
+    },
+    {
+      "epoch": 0.19835662322708955,
+      "grad_norm": 0.40758174657821655,
+      "learning_rate": 2e-05,
+      "loss": 0.6328,
+      "step": 3790
+    },
+    {
+      "epoch": 0.19887999162610562,
+      "grad_norm": 0.42536821961402893,
+      "learning_rate": 2e-05,
+      "loss": 0.6343,
+      "step": 3800
+    },
+    {
+      "epoch": 0.19940336002512168,
+      "grad_norm": 0.4246145486831665,
+      "learning_rate": 2e-05,
+      "loss": 0.617,
+      "step": 3810
+    },
+    {
+      "epoch": 0.19992672842413775,
+      "grad_norm": 0.4054490327835083,
+      "learning_rate": 2e-05,
+      "loss": 0.6374,
+      "step": 3820
+    },
+    {
+      "epoch": 0.20045009682315382,
+      "grad_norm": 0.4975704848766327,
+      "learning_rate": 2e-05,
+      "loss": 0.6006,
+      "step": 3830
+    },
+    {
+      "epoch": 0.20097346522216988,
+      "grad_norm": 0.5053114891052246,
+      "learning_rate": 2e-05,
+      "loss": 0.6319,
+      "step": 3840
+    },
+    {
+      "epoch": 0.20149683362118595,
+      "grad_norm": 0.41113340854644775,
+      "learning_rate": 2e-05,
+      "loss": 0.6342,
+      "step": 3850
+    },
+    {
+      "epoch": 0.20202020202020202,
+      "grad_norm": 0.3836223781108856,
+      "learning_rate": 2e-05,
+      "loss": 0.6196,
+      "step": 3860
+    },
+    {
+      "epoch": 0.20254357041921808,
+      "grad_norm": 0.3937544822692871,
+      "learning_rate": 2e-05,
+      "loss": 0.641,
+      "step": 3870
+    },
+    {
+      "epoch": 0.20306693881823415,
+      "grad_norm": 0.4237481951713562,
+      "learning_rate": 2e-05,
+      "loss": 0.6125,
+      "step": 3880
+    },
+    {
+      "epoch": 0.20359030721725022,
+      "grad_norm": 0.3751864731311798,
+      "learning_rate": 2e-05,
+      "loss": 0.6197,
+      "step": 3890
+    },
+    {
+      "epoch": 0.20411367561626628,
+      "grad_norm": 0.5081929564476013,
+      "learning_rate": 2e-05,
+      "loss": 0.6203,
+      "step": 3900
+    },
+    {
+      "epoch": 0.20463704401528235,
+      "grad_norm": 0.4540235698223114,
+      "learning_rate": 2e-05,
+      "loss": 0.6168,
+      "step": 3910
+    },
+    {
+      "epoch": 0.20516041241429842,
+      "grad_norm": 0.41905754804611206,
+      "learning_rate": 2e-05,
+      "loss": 0.616,
+      "step": 3920
+    },
+    {
+      "epoch": 0.20568378081331448,
+      "grad_norm": 0.5326279997825623,
+      "learning_rate": 2e-05,
+      "loss": 0.6211,
+      "step": 3930
+    },
+    {
+      "epoch": 0.20620714921233055,
+      "grad_norm": 0.41054078936576843,
+      "learning_rate": 2e-05,
+      "loss": 0.6113,
+      "step": 3940
+    },
+    {
+      "epoch": 0.20673051761134661,
+      "grad_norm": 0.4346046447753906,
+      "learning_rate": 2e-05,
+      "loss": 0.6495,
+      "step": 3950
+    },
+    {
+      "epoch": 0.20725388601036268,
+      "grad_norm": 0.410095751285553,
+      "learning_rate": 2e-05,
+      "loss": 0.62,
+      "step": 3960
+    },
+    {
+      "epoch": 0.20777725440937878,
+      "grad_norm": 0.40288132429122925,
+      "learning_rate": 2e-05,
+      "loss": 0.6214,
+      "step": 3970
+    },
+    {
+      "epoch": 0.20830062280839484,
+      "grad_norm": 0.40218082070350647,
+      "learning_rate": 2e-05,
+      "loss": 0.6318,
+      "step": 3980
+    },
+    {
+      "epoch": 0.2088239912074109,
+      "grad_norm": 0.43279343843460083,
+      "learning_rate": 2e-05,
+      "loss": 0.6341,
+      "step": 3990
+    },
+    {
+      "epoch": 0.20934735960642697,
+      "grad_norm": 0.39592084288597107,
+      "learning_rate": 2e-05,
+      "loss": 0.6042,
+      "step": 4000
+    },
+    {
+      "epoch": 0.20987072800544304,
+      "grad_norm": 0.4071950912475586,
+      "learning_rate": 2e-05,
+      "loss": 0.5843,
+      "step": 4010
+    },
+    {
+      "epoch": 0.2103940964044591,
+      "grad_norm": 0.3897649347782135,
+      "learning_rate": 2e-05,
+      "loss": 0.6269,
+      "step": 4020
+    },
+    {
+      "epoch": 0.21091746480347517,
+      "grad_norm": 0.7160203456878662,
+      "learning_rate": 2e-05,
+      "loss": 0.6499,
+      "step": 4030
+    },
+    {
+      "epoch": 0.21144083320249124,
+      "grad_norm": 0.38851386308670044,
+      "learning_rate": 2e-05,
+      "loss": 0.6142,
+      "step": 4040
+    },
+    {
+      "epoch": 0.2119642016015073,
+      "grad_norm": 0.3943970799446106,
+      "learning_rate": 2e-05,
+      "loss": 0.5983,
+      "step": 4050
+    },
+    {
+      "epoch": 0.21248757000052337,
+      "grad_norm": 0.41590937972068787,
+      "learning_rate": 2e-05,
+      "loss": 0.6363,
+      "step": 4060
+    },
+    {
+      "epoch": 0.21301093839953944,
+      "grad_norm": 0.38856077194213867,
+      "learning_rate": 2e-05,
+      "loss": 0.611,
+      "step": 4070
+    },
+    {
+      "epoch": 0.2135343067985555,
+      "grad_norm": 0.4386836886405945,
+      "learning_rate": 2e-05,
+      "loss": 0.6145,
+      "step": 4080
+    },
+    {
+      "epoch": 0.21405767519757157,
+      "grad_norm": 0.46340975165367126,
+      "learning_rate": 2e-05,
+      "loss": 0.5949,
+      "step": 4090
+    },
+    {
+      "epoch": 0.21458104359658764,
+      "grad_norm": 0.4160849153995514,
+      "learning_rate": 2e-05,
+      "loss": 0.6446,
+      "step": 4100
+    },
+    {
+      "epoch": 0.2151044119956037,
+      "grad_norm": 0.4251480996608734,
+      "learning_rate": 2e-05,
+      "loss": 0.6031,
+      "step": 4110
+    },
+    {
+      "epoch": 0.21562778039461977,
+      "grad_norm": 0.41039806604385376,
+      "learning_rate": 2e-05,
+      "loss": 0.6165,
+      "step": 4120
+    },
+    {
+      "epoch": 0.21615114879363584,
+      "grad_norm": 0.4157627522945404,
+      "learning_rate": 2e-05,
+      "loss": 0.6219,
+      "step": 4130
+    },
+    {
+      "epoch": 0.2166745171926519,
+      "grad_norm": 0.3525838553905487,
+      "learning_rate": 2e-05,
+      "loss": 0.6062,
+      "step": 4140
+    },
+    {
+      "epoch": 0.21719788559166797,
+      "grad_norm": 0.4129687249660492,
+      "learning_rate": 2e-05,
+      "loss": 0.6104,
+      "step": 4150
+    },
+    {
+      "epoch": 0.21772125399068404,
+      "grad_norm": 0.4055493175983429,
+      "learning_rate": 2e-05,
+      "loss": 0.5947,
+      "step": 4160
+    },
+    {
+      "epoch": 0.2182446223897001,
+      "grad_norm": 0.42468446493148804,
+      "learning_rate": 2e-05,
+      "loss": 0.6507,
+      "step": 4170
+    },
+    {
+      "epoch": 0.21876799078871617,
+      "grad_norm": 0.4344474673271179,
+      "learning_rate": 2e-05,
+      "loss": 0.6062,
+      "step": 4180
+    },
+    {
+      "epoch": 0.21929135918773224,
+      "grad_norm": 0.4484303295612335,
+      "learning_rate": 2e-05,
+      "loss": 0.6128,
+      "step": 4190
+    },
+    {
+      "epoch": 0.2198147275867483,
+      "grad_norm": 0.38424041867256165,
+      "learning_rate": 2e-05,
+      "loss": 0.611,
+      "step": 4200
+    },
+    {
+      "epoch": 0.22033809598576437,
+      "grad_norm": 0.40774214267730713,
+      "learning_rate": 2e-05,
+      "loss": 0.6128,
+      "step": 4210
+    },
+    {
+      "epoch": 0.22086146438478044,
+      "grad_norm": 0.39545294642448425,
+      "learning_rate": 2e-05,
+      "loss": 0.6196,
+      "step": 4220
+    },
+    {
+      "epoch": 0.2213848327837965,
+      "grad_norm": 0.42367228865623474,
+      "learning_rate": 2e-05,
+      "loss": 0.6324,
+      "step": 4230
+    },
+    {
+      "epoch": 0.22190820118281257,
+      "grad_norm": 0.4583156406879425,
+      "learning_rate": 2e-05,
+      "loss": 0.6011,
+      "step": 4240
+    },
+    {
+      "epoch": 0.22243156958182864,
+      "grad_norm": 0.36509275436401367,
+      "learning_rate": 2e-05,
+      "loss": 0.6227,
+      "step": 4250
+    },
+    {
+      "epoch": 0.2229549379808447,
+      "grad_norm": 0.41607242822647095,
+      "learning_rate": 2e-05,
+      "loss": 0.625,
+      "step": 4260
+    },
+    {
+      "epoch": 0.2234783063798608,
+      "grad_norm": 0.41028332710266113,
+      "learning_rate": 2e-05,
+      "loss": 0.5968,
+      "step": 4270
+    },
+    {
+      "epoch": 0.22400167477887686,
+      "grad_norm": 0.4013286530971527,
+      "learning_rate": 2e-05,
+      "loss": 0.6143,
+      "step": 4280
+    },
+    {
+      "epoch": 0.22452504317789293,
+      "grad_norm": 0.42334598302841187,
+      "learning_rate": 2e-05,
+      "loss": 0.6036,
+      "step": 4290
+    },
+    {
+      "epoch": 0.225048411576909,
+      "grad_norm": 0.47744619846343994,
+      "learning_rate": 2e-05,
+      "loss": 0.6329,
+      "step": 4300
+    },
+    {
+      "epoch": 0.22557177997592506,
+      "grad_norm": 0.378061443567276,
+      "learning_rate": 2e-05,
+      "loss": 0.6191,
+      "step": 4310
+    },
+    {
+      "epoch": 0.22609514837494113,
+      "grad_norm": 0.4232480823993683,
+      "learning_rate": 2e-05,
+      "loss": 0.6167,
+      "step": 4320
+    },
+    {
+      "epoch": 0.2266185167739572,
+      "grad_norm": 0.3921084702014923,
+      "learning_rate": 2e-05,
+      "loss": 0.6363,
+      "step": 4330
+    },
+    {
+      "epoch": 0.22714188517297326,
+      "grad_norm": 0.3987734913825989,
+      "learning_rate": 2e-05,
+      "loss": 0.611,
+      "step": 4340
+    },
+    {
+      "epoch": 0.22766525357198933,
+      "grad_norm": 0.4001140296459198,
+      "learning_rate": 2e-05,
+      "loss": 0.6025,
+      "step": 4350
+    },
+    {
+      "epoch": 0.2281886219710054,
+      "grad_norm": 0.4162245988845825,
+      "learning_rate": 2e-05,
+      "loss": 0.6151,
+      "step": 4360
+    },
+    {
+      "epoch": 0.22871199037002146,
+      "grad_norm": 0.4258664548397064,
+      "learning_rate": 2e-05,
+      "loss": 0.6107,
+      "step": 4370
+    },
+    {
+      "epoch": 0.22923535876903753,
+      "grad_norm": 0.39507806301116943,
+      "learning_rate": 2e-05,
+      "loss": 0.6155,
+      "step": 4380
+    },
+    {
+      "epoch": 0.2297587271680536,
+      "grad_norm": 0.49936944246292114,
+      "learning_rate": 2e-05,
+      "loss": 0.6367,
+      "step": 4390
+    },
+    {
+      "epoch": 0.23028209556706966,
+      "grad_norm": 0.4109865128993988,
+      "learning_rate": 2e-05,
+      "loss": 0.6163,
+      "step": 4400
+    },
+    {
+      "epoch": 0.23080546396608573,
+      "grad_norm": 0.4059872031211853,
+      "learning_rate": 2e-05,
+      "loss": 0.6087,
+      "step": 4410
+    },
+    {
+      "epoch": 0.2313288323651018,
+      "grad_norm": 0.41509491205215454,
+      "learning_rate": 2e-05,
+      "loss": 0.6034,
+      "step": 4420
+    },
+    {
+      "epoch": 0.23185220076411786,
+      "grad_norm": 0.4001082181930542,
+      "learning_rate": 2e-05,
+      "loss": 0.5977,
+      "step": 4430
+    },
+    {
+      "epoch": 0.23237556916313393,
+      "grad_norm": 0.4086252748966217,
+      "learning_rate": 2e-05,
+      "loss": 0.606,
+      "step": 4440
+    },
+    {
+      "epoch": 0.23289893756215,
+      "grad_norm": 0.3729661703109741,
+      "learning_rate": 2e-05,
+      "loss": 0.6201,
+      "step": 4450
+    },
+    {
+      "epoch": 0.23342230596116606,
+      "grad_norm": 0.39218300580978394,
+      "learning_rate": 2e-05,
+      "loss": 0.6174,
+      "step": 4460
+    },
+    {
+      "epoch": 0.23394567436018213,
+      "grad_norm": 0.3822868764400482,
+      "learning_rate": 2e-05,
+      "loss": 0.6395,
+      "step": 4470
+    },
+    {
+      "epoch": 0.2344690427591982,
+      "grad_norm": 0.4116714596748352,
+      "learning_rate": 2e-05,
+      "loss": 0.6101,
+      "step": 4480
+    },
+    {
+      "epoch": 0.23499241115821426,
+      "grad_norm": 0.4040875732898712,
+      "learning_rate": 2e-05,
+      "loss": 0.626,
+      "step": 4490
+    },
+    {
+      "epoch": 0.23551577955723033,
+      "grad_norm": 0.3950350880622864,
+      "learning_rate": 2e-05,
+      "loss": 0.5975,
+      "step": 4500
+    },
+    {
+      "epoch": 0.2360391479562464,
+      "grad_norm": 0.39547741413116455,
+      "learning_rate": 2e-05,
+      "loss": 0.6092,
+      "step": 4510
+    },
+    {
+      "epoch": 0.23656251635526246,
+      "grad_norm": 0.4103192985057831,
+      "learning_rate": 2e-05,
+      "loss": 0.5902,
+      "step": 4520
+    },
+    {
+      "epoch": 0.23708588475427853,
+      "grad_norm": 0.4051879048347473,
+      "learning_rate": 2e-05,
+      "loss": 0.6064,
+      "step": 4530
+    },
+    {
+      "epoch": 0.2376092531532946,
+      "grad_norm": 0.4002724885940552,
+      "learning_rate": 2e-05,
+      "loss": 0.618,
+      "step": 4540
+    },
+    {
+      "epoch": 0.23813262155231066,
+      "grad_norm": 0.4022727310657501,
+      "learning_rate": 2e-05,
+      "loss": 0.6145,
+      "step": 4550
+    },
+    {
+      "epoch": 0.23865598995132675,
+      "grad_norm": 0.4159553050994873,
+      "learning_rate": 2e-05,
+      "loss": 0.5922,
+      "step": 4560
+    },
+    {
+      "epoch": 0.23917935835034282,
+      "grad_norm": 0.3999551236629486,
+      "learning_rate": 2e-05,
+      "loss": 0.5905,
+      "step": 4570
+    },
+    {
+      "epoch": 0.23970272674935889,
+      "grad_norm": 0.4089633822441101,
+      "learning_rate": 2e-05,
+      "loss": 0.6132,
+      "step": 4580
+    },
+    {
+      "epoch": 0.24022609514837495,
+      "grad_norm": 0.37393105030059814,
+      "learning_rate": 2e-05,
+      "loss": 0.6094,
+      "step": 4590
+    },
+    {
+      "epoch": 0.24074946354739102,
+      "grad_norm": 0.6190505623817444,
+      "learning_rate": 2e-05,
+      "loss": 0.6066,
+      "step": 4600
+    },
+    {
+      "epoch": 0.24127283194640708,
+      "grad_norm": 0.42357298731803894,
+      "learning_rate": 2e-05,
+      "loss": 0.63,
+      "step": 4610
+    },
+    {
+      "epoch": 0.24179620034542315,
+      "grad_norm": 0.7327708601951599,
+      "learning_rate": 2e-05,
+      "loss": 0.5942,
+      "step": 4620
+    },
+    {
+      "epoch": 0.24231956874443922,
+      "grad_norm": 0.4425099790096283,
+      "learning_rate": 2e-05,
+      "loss": 0.6078,
+      "step": 4630
+    },
+    {
+      "epoch": 0.24284293714345528,
+      "grad_norm": 0.3585580289363861,
+      "learning_rate": 2e-05,
+      "loss": 0.5886,
+      "step": 4640
+    },
+    {
+      "epoch": 0.24336630554247135,
+      "grad_norm": 1.4051371812820435,
+      "learning_rate": 2e-05,
+      "loss": 0.6031,
+      "step": 4650
+    },
+    {
+      "epoch": 0.24388967394148742,
+      "grad_norm": 0.4339183270931244,
+      "learning_rate": 2e-05,
+      "loss": 0.6104,
+      "step": 4660
+    },
+    {
+      "epoch": 0.24441304234050348,
+      "grad_norm": 0.40716227889060974,
+      "learning_rate": 2e-05,
+      "loss": 0.6532,
+      "step": 4670
+    },
+    {
+      "epoch": 0.24493641073951955,
+      "grad_norm": 0.3990173041820526,
+      "learning_rate": 2e-05,
+      "loss": 0.6159,
+      "step": 4680
+    },
+    {
+      "epoch": 0.24545977913853562,
+      "grad_norm": 0.5576123595237732,
+      "learning_rate": 2e-05,
+      "loss": 0.6279,
+      "step": 4690
+    },
+    {
+      "epoch": 0.24598314753755168,
+      "grad_norm": 0.3955235183238983,
+      "learning_rate": 2e-05,
+      "loss": 0.6232,
+      "step": 4700
+    },
+    {
+      "epoch": 0.24650651593656775,
+      "grad_norm": 0.4591265320777893,
+      "learning_rate": 2e-05,
+      "loss": 0.6311,
+      "step": 4710
+    },
+    {
+      "epoch": 0.24702988433558382,
+      "grad_norm": 0.3967943787574768,
+      "learning_rate": 2e-05,
+      "loss": 0.6285,
+      "step": 4720
+    },
+    {
+      "epoch": 0.24755325273459988,
+      "grad_norm": 0.5398264527320862,
+      "learning_rate": 2e-05,
+      "loss": 0.6331,
+      "step": 4730
+    },
+    {
+      "epoch": 0.24807662113361595,
+      "grad_norm": 0.41231897473335266,
+      "learning_rate": 2e-05,
+      "loss": 0.6024,
+      "step": 4740
+    },
+    {
+      "epoch": 0.24859998953263202,
+      "grad_norm": 0.40629324316978455,
+      "learning_rate": 2e-05,
+      "loss": 0.6271,
+      "step": 4750
+    },
+    {
+      "epoch": 0.24912335793164808,
+      "grad_norm": 0.42986491322517395,
+      "learning_rate": 2e-05,
+      "loss": 0.5869,
+      "step": 4760
+    },
+    {
+      "epoch": 0.24964672633066415,
+      "grad_norm": 0.45604678988456726,
+      "learning_rate": 2e-05,
+      "loss": 0.632,
+      "step": 4770
+    },
+    {
+      "epoch": 0.2501700947296802,
+      "grad_norm": 0.4645533859729767,
+      "learning_rate": 2e-05,
+      "loss": 0.6167,
+      "step": 4780
+    },
+    {
+      "epoch": 0.2506934631286963,
+      "grad_norm": 0.3783343732357025,
+      "learning_rate": 2e-05,
+      "loss": 0.6157,
+      "step": 4790
+    },
+    {
+      "epoch": 0.25121683152771235,
+      "grad_norm": 0.38526099920272827,
+      "learning_rate": 2e-05,
+      "loss": 0.5896,
+      "step": 4800
+    },
+    {
+      "epoch": 0.2517401999267284,
+      "grad_norm": 0.3877682685852051,
+      "learning_rate": 2e-05,
+      "loss": 0.6102,
+      "step": 4810
+    },
+    {
+      "epoch": 0.2522635683257445,
+      "grad_norm": 0.39466989040374756,
+      "learning_rate": 2e-05,
+      "loss": 0.5981,
+      "step": 4820
+    },
+    {
+      "epoch": 0.25278693672476055,
+      "grad_norm": 0.4054771065711975,
+      "learning_rate": 2e-05,
+      "loss": 0.5967,
+      "step": 4830
+    },
+    {
+      "epoch": 0.2533103051237766,
+      "grad_norm": 0.37173131108283997,
+      "learning_rate": 2e-05,
+      "loss": 0.5909,
+      "step": 4840
+    },
+    {
+      "epoch": 0.2538336735227927,
+      "grad_norm": 0.41072943806648254,
+      "learning_rate": 2e-05,
+      "loss": 0.6111,
+      "step": 4850
+    },
+    {
+      "epoch": 0.25435704192180875,
+      "grad_norm": 0.41547060012817383,
+      "learning_rate": 2e-05,
+      "loss": 0.6222,
+      "step": 4860
+    },
+    {
+      "epoch": 0.2548804103208248,
+      "grad_norm": 0.409974604845047,
+      "learning_rate": 2e-05,
+      "loss": 0.6307,
+      "step": 4870
+    },
+    {
+      "epoch": 0.2554037787198409,
+      "grad_norm": 0.38040807843208313,
+      "learning_rate": 2e-05,
+      "loss": 0.6125,
+      "step": 4880
+    },
+    {
+      "epoch": 0.25592714711885695,
+      "grad_norm": 0.4161960482597351,
+      "learning_rate": 2e-05,
+      "loss": 0.6311,
+      "step": 4890
+    },
+    {
+      "epoch": 0.256450515517873,
+      "grad_norm": 0.41063639521598816,
+      "learning_rate": 2e-05,
+      "loss": 0.6335,
+      "step": 4900
+    },
+    {
+      "epoch": 0.2569738839168891,
+      "grad_norm": 0.4094699025154114,
+      "learning_rate": 2e-05,
+      "loss": 0.6166,
+      "step": 4910
+    },
+    {
+      "epoch": 0.25749725231590515,
+      "grad_norm": 0.4404178559780121,
+      "learning_rate": 2e-05,
+      "loss": 0.6109,
+      "step": 4920
+    },
+    {
+      "epoch": 0.2580206207149212,
+      "grad_norm": 0.42314210534095764,
+      "learning_rate": 2e-05,
+      "loss": 0.6089,
+      "step": 4930
+    },
+    {
+      "epoch": 0.2585439891139373,
+      "grad_norm": 0.4365096688270569,
+      "learning_rate": 2e-05,
+      "loss": 0.6193,
+      "step": 4940
+    },
+    {
+      "epoch": 0.25906735751295334,
+      "grad_norm": 0.4479396939277649,
+      "learning_rate": 2e-05,
+      "loss": 0.6254,
+      "step": 4950
+    },
+    {
+      "epoch": 0.2595907259119694,
+      "grad_norm": 0.3912433385848999,
+      "learning_rate": 2e-05,
+      "loss": 0.6169,
+      "step": 4960
+    },
+    {
+      "epoch": 0.2601140943109855,
+      "grad_norm": 0.3662976622581482,
+      "learning_rate": 2e-05,
+      "loss": 0.6037,
+      "step": 4970
+    },
+    {
+      "epoch": 0.26063746271000154,
+      "grad_norm": 0.4002629518508911,
+      "learning_rate": 2e-05,
+      "loss": 0.5935,
+      "step": 4980
+    },
+    {
+      "epoch": 0.2611608311090176,
+      "grad_norm": 0.39754584431648254,
+      "learning_rate": 2e-05,
+      "loss": 0.6045,
+      "step": 4990
+    },
+    {
+      "epoch": 0.26168419950803373,
+      "grad_norm": 0.44900912046432495,
+      "learning_rate": 2e-05,
+      "loss": 0.6234,
+      "step": 5000
+    },
+    {
+      "epoch": 0.2622075679070498,
+      "grad_norm": 0.46902185678482056,
+      "learning_rate": 2e-05,
+      "loss": 0.5961,
+      "step": 5010
+    },
+    {
+      "epoch": 0.26273093630606587,
+      "grad_norm": 0.41309309005737305,
+      "learning_rate": 2e-05,
+      "loss": 0.6275,
+      "step": 5020
+    },
+    {
+      "epoch": 0.26325430470508193,
+      "grad_norm": 0.396864652633667,
+      "learning_rate": 2e-05,
+      "loss": 0.638,
+      "step": 5030
+    },
+    {
+      "epoch": 0.263777673104098,
+      "grad_norm": 0.39725908637046814,
+      "learning_rate": 2e-05,
+      "loss": 0.638,
+      "step": 5040
+    },
+    {
+      "epoch": 0.26430104150311406,
+      "grad_norm": 0.40473729372024536,
+      "learning_rate": 2e-05,
+      "loss": 0.6166,
+      "step": 5050
+    },
+    {
+      "epoch": 0.26482440990213013,
+      "grad_norm": 0.45297154784202576,
+      "learning_rate": 2e-05,
+      "loss": 0.6167,
+      "step": 5060
+    },
+    {
+      "epoch": 0.2653477783011462,
+      "grad_norm": 0.3637174367904663,
+      "learning_rate": 2e-05,
+      "loss": 0.59,
+      "step": 5070
+    },
+    {
+      "epoch": 0.26587114670016226,
+      "grad_norm": 0.3888295590877533,
+      "learning_rate": 2e-05,
+      "loss": 0.5946,
+      "step": 5080
+    },
+    {
+      "epoch": 0.26639451509917833,
+      "grad_norm": 0.3925114870071411,
+      "learning_rate": 2e-05,
+      "loss": 0.6454,
+      "step": 5090
+    },
+    {
+      "epoch": 0.2669178834981944,
+      "grad_norm": 0.42430034279823303,
+      "learning_rate": 2e-05,
+      "loss": 0.598,
+      "step": 5100
+    },
+    {
+      "epoch": 0.26744125189721046,
+      "grad_norm": 0.4356481432914734,
+      "learning_rate": 2e-05,
+      "loss": 0.5841,
+      "step": 5110
+    },
+    {
+      "epoch": 0.26796462029622653,
+      "grad_norm": 0.40845903754234314,
+      "learning_rate": 2e-05,
+      "loss": 0.6214,
+      "step": 5120
+    },
+    {
+      "epoch": 0.2684879886952426,
+      "grad_norm": 0.37573105096817017,
+      "learning_rate": 2e-05,
+      "loss": 0.6283,
+      "step": 5130
+    },
+    {
+      "epoch": 0.26901135709425866,
+      "grad_norm": 0.40981823205947876,
+      "learning_rate": 2e-05,
+      "loss": 0.5933,
+      "step": 5140
+    },
+    {
+      "epoch": 0.26953472549327473,
+      "grad_norm": 0.39033955335617065,
+      "learning_rate": 2e-05,
+      "loss": 0.5921,
+      "step": 5150
+    },
+    {
+      "epoch": 0.2700580938922908,
+      "grad_norm": 0.4205704927444458,
+      "learning_rate": 2e-05,
+      "loss": 0.6128,
+      "step": 5160
+    },
+    {
+      "epoch": 0.27058146229130686,
+      "grad_norm": 0.37669286131858826,
+      "learning_rate": 2e-05,
+      "loss": 0.6054,
+      "step": 5170
+    },
+    {
+      "epoch": 0.27110483069032293,
+      "grad_norm": 0.40791818499565125,
+      "learning_rate": 2e-05,
+      "loss": 0.6309,
+      "step": 5180
+    },
+    {
+      "epoch": 0.271628199089339,
+      "grad_norm": 0.42482030391693115,
+      "learning_rate": 2e-05,
+      "loss": 0.6297,
+      "step": 5190
+    },
+    {
+      "epoch": 0.27215156748835506,
+      "grad_norm": 0.4918680787086487,
+      "learning_rate": 2e-05,
+      "loss": 0.5974,
+      "step": 5200
+    },
+    {
+      "epoch": 0.27267493588737113,
+      "grad_norm": 0.4212997257709503,
+      "learning_rate": 2e-05,
+      "loss": 0.6251,
+      "step": 5210
+    },
+    {
+      "epoch": 0.2731983042863872,
+      "grad_norm": 0.40279462933540344,
+      "learning_rate": 2e-05,
+      "loss": 0.5907,
+      "step": 5220
+    },
+    {
+      "epoch": 0.27372167268540326,
+      "grad_norm": 0.4512042999267578,
+      "learning_rate": 2e-05,
+      "loss": 0.6082,
+      "step": 5230
+    },
+    {
+      "epoch": 0.2742450410844193,
+      "grad_norm": 0.4068576693534851,
+      "learning_rate": 2e-05,
+      "loss": 0.6062,
+      "step": 5240
+    },
+    {
+      "epoch": 0.2747684094834354,
+      "grad_norm": 0.4271692931652069,
+      "learning_rate": 2e-05,
+      "loss": 0.6326,
+      "step": 5250
+    },
+    {
+      "epoch": 0.27529177788245146,
+      "grad_norm": 0.41401100158691406,
+      "learning_rate": 2e-05,
+      "loss": 0.6105,
+      "step": 5260
+    },
+    {
+      "epoch": 0.2758151462814675,
+      "grad_norm": 0.4413628578186035,
+      "learning_rate": 2e-05,
+      "loss": 0.5898,
+      "step": 5270
+    },
+    {
+      "epoch": 0.2763385146804836,
+      "grad_norm": 0.41365042328834534,
+      "learning_rate": 2e-05,
+      "loss": 0.5958,
+      "step": 5280
+    },
+    {
+      "epoch": 0.27686188307949966,
+      "grad_norm": 0.45695561170578003,
+      "learning_rate": 2e-05,
+      "loss": 0.6283,
+      "step": 5290
+    },
+    {
+      "epoch": 0.2773852514785157,
+      "grad_norm": 0.4205687940120697,
+      "learning_rate": 2e-05,
+      "loss": 0.6257,
+      "step": 5300
+    },
+    {
+      "epoch": 0.2779086198775318,
+      "grad_norm": 0.3920809030532837,
+      "learning_rate": 2e-05,
+      "loss": 0.6518,
+      "step": 5310
+    },
+    {
+      "epoch": 0.27843198827654786,
+      "grad_norm": 0.34857314825057983,
+      "learning_rate": 2e-05,
+      "loss": 0.5915,
+      "step": 5320
+    },
+    {
+      "epoch": 0.2789553566755639,
+      "grad_norm": 0.47157859802246094,
+      "learning_rate": 2e-05,
+      "loss": 0.6315,
+      "step": 5330
+    },
+    {
+      "epoch": 0.27947872507458,
+      "grad_norm": 0.4518488943576813,
+      "learning_rate": 2e-05,
+      "loss": 0.6172,
+      "step": 5340
+    },
+    {
+      "epoch": 0.28000209347359606,
+      "grad_norm": 0.44076552987098694,
+      "learning_rate": 1.999194520388485e-05,
+      "loss": 0.6251,
+      "step": 5350
+    },
+    {
+      "epoch": 0.2805254618726121,
+      "grad_norm": 0.38890475034713745,
+      "learning_rate": 1.9959247303157763e-05,
+      "loss": 0.6194,
+      "step": 5360
+    },
+    {
+      "epoch": 0.2810488302716282,
+      "grad_norm": 0.3902517855167389,
+      "learning_rate": 1.990149424459817e-05,
+      "loss": 0.6009,
+      "step": 5370
+    },
+    {
+      "epoch": 0.28157219867064426,
+      "grad_norm": 0.39967110753059387,
+      "learning_rate": 1.9818847538648688e-05,
+      "loss": 0.6081,
+      "step": 5380
+    },
+    {
+      "epoch": 0.2820955670696603,
+      "grad_norm": 0.3647516965866089,
+      "learning_rate": 1.971153831256721e-05,
+      "loss": 0.5785,
+      "step": 5390
+    },
+    {
+      "epoch": 0.2826189354686764,
+      "grad_norm": 0.4686444401741028,
+      "learning_rate": 1.9579866664063435e-05,
+      "loss": 0.635,
+      "step": 5400
+    },
+    {
+      "epoch": 0.28314230386769246,
+      "grad_norm": 0.4206537902355194,
+      "learning_rate": 1.9424200822054782e-05,
+      "loss": 0.628,
+      "step": 5410
+    },
+    {
+      "epoch": 0.2836656722667085,
+      "grad_norm": 0.43939080834388733,
+      "learning_rate": 1.9244976116888626e-05,
+      "loss": 0.6335,
+      "step": 5420
+    },
+    {
+      "epoch": 0.2841890406657246,
+      "grad_norm": 0.3906099796295166,
+      "learning_rate": 1.904269376291071e-05,
+      "loss": 0.5958,
+      "step": 5430
+    },
+    {
+      "epoch": 0.28471240906474066,
+      "grad_norm": 0.3985750377178192,
+      "learning_rate": 1.88179194567844e-05,
+      "loss": 0.6213,
+      "step": 5440
+    },
+    {
+      "epoch": 0.2852357774637567,
+      "grad_norm": 0.4336455762386322,
+      "learning_rate": 1.8571281795480632e-05,
+      "loss": 0.604,
+      "step": 5450
+    },
+    {
+      "epoch": 0.2857591458627728,
+      "grad_norm": 0.4126925468444824,
+      "learning_rate": 1.8303470518362808e-05,
+      "loss": 0.5991,
+      "step": 5460
+    },
+    {
+      "epoch": 0.28628251426178886,
+      "grad_norm": 0.4074404835700989,
+      "learning_rate": 1.801523457828271e-05,
+      "loss": 0.6045,
+      "step": 5470
+    },
+    {
+      "epoch": 0.2868058826608049,
+      "grad_norm": 0.40034711360931396,
+      "learning_rate": 1.7707380047081796e-05,
+      "loss": 0.6052,
+      "step": 5480
+    },
+    {
+      "epoch": 0.287329251059821,
+      "grad_norm": 0.41930365562438965,
+      "learning_rate": 1.7380767861355252e-05,
+      "loss": 0.6316,
+      "step": 5490
+    },
+    {
+      "epoch": 0.28785261945883706,
+      "grad_norm": 0.42970430850982666,
+      "learning_rate": 1.703631141478299e-05,
+      "loss": 0.643,
+      "step": 5500
+    },
+    {
+      "epoch": 0.2883759878578531,
+      "grad_norm": 0.6285122632980347,
+      "learning_rate": 1.6674974003760705e-05,
+      "loss": 0.5952,
+      "step": 5510
+    },
+    {
+      "epoch": 0.2888993562568692,
+      "grad_norm": 0.4414973556995392,
+      "learning_rate": 1.6297766133474556e-05,
+      "loss": 0.5934,
+      "step": 5520
+    },
+    {
+      "epoch": 0.28942272465588526,
+      "grad_norm": 0.3714768886566162,
+      "learning_rate": 1.5905742691953186e-05,
+      "loss": 0.5846,
+      "step": 5530
+    },
+    {
+      "epoch": 0.2899460930549013,
+      "grad_norm": 0.3532920181751251,
+      "learning_rate": 1.55e-05,
+      "loss": 0.5995,
+      "step": 5540
+    },
+    {
+      "epoch": 0.2904694614539174,
+      "grad_norm": 0.4924233853816986,
+      "learning_rate": 1.5081672745255862e-05,
+      "loss": 0.5818,
+      "step": 5550
+    },
+    {
+      "epoch": 0.29099282985293345,
+      "grad_norm": 0.4407850503921509,
+      "learning_rate": 1.4651930808966206e-05,
+      "loss": 0.6016,
+      "step": 5560
+    },
+    {
+      "epoch": 0.2915161982519495,
+      "grad_norm": 0.42552539706230164,
+      "learning_rate": 1.4211975994326848e-05,
+      "loss": 0.603,
+      "step": 5570
+    },
+    {
+      "epoch": 0.2920395666509656,
+      "grad_norm": 0.36129358410835266,
+      "learning_rate": 1.376303866555779e-05,
+      "loss": 0.6101,
+      "step": 5580
+    },
+    {
+      "epoch": 0.2925629350499817,
+      "grad_norm": 0.406896710395813,
+      "learning_rate": 1.3306374307104164e-05,
+      "loss": 0.6115,
+      "step": 5590
+    },
+    {
+      "epoch": 0.2930863034489978,
+      "grad_norm": 0.43325328826904297,
+      "learning_rate": 1.2843260012586718e-05,
+      "loss": 0.6105,
+      "step": 5600
+    },
+    {
+      "epoch": 0.29360967184801384,
+      "grad_norm": 0.40178170800209045,
+      "learning_rate": 1.2374990913320726e-05,
+      "loss": 0.6258,
+      "step": 5610
+    },
+    {
+      "epoch": 0.2941330402470299,
+      "grad_norm": 0.37656369805336,
+      "learning_rate": 1.1902876556391247e-05,
+      "loss": 0.6063,
+      "step": 5620
+    },
+    {
+      "epoch": 0.294656408646046,
+      "grad_norm": 0.4082496166229248,
+      "learning_rate": 1.1428237242413685e-05,
+      "loss": 0.579,
+      "step": 5630
+    },
+    {
+      "epoch": 0.29517977704506204,
+      "grad_norm": 0.40694573521614075,
+      "learning_rate": 1.095240033322132e-05,
+      "loss": 0.5992,
+      "step": 5640
+    },
+    {
+      "epoch": 0.2957031454440781,
+      "grad_norm": 0.4091090261936188,
+      "learning_rate": 1.047669653980572e-05,
+      "loss": 0.5843,
+      "step": 5650
+    },
+    {
+      "epoch": 0.2962265138430942,
+      "grad_norm": 0.44866329431533813,
+      "learning_rate": 1.0002456200890902e-05,
+      "loss": 0.6118,
+      "step": 5660
+    },
+    {
+      "epoch": 0.29674988224211024,
+      "grad_norm": 0.4145185649394989,
+      "learning_rate": 9.531005562548668e-06,
+      "loss": 0.5998,
+      "step": 5670
+    },
+    {
+      "epoch": 0.2972732506411263,
+      "grad_norm": 0.4247947037220001,
+      "learning_rate": 9.06366306925925e-06,
+      "loss": 0.5885,
+      "step": 5680
+    },
+    {
+      "epoch": 0.2977966190401424,
+      "grad_norm": 0.3585066795349121,
+      "learning_rate": 8.601735676789686e-06,
+      "loss": 0.6015,
+      "step": 5690
+    },
+    {
+      "epoch": 0.29831998743915844,
+      "grad_norm": 0.3997299373149872,
+      "learning_rate": 8.14651519720118e-06,
+      "loss": 0.6087,
+      "step": 5700
+    },
+    {
+      "epoch": 0.2988433558381745,
+      "grad_norm": 0.4697336256504059,
+      "learning_rate": 7.699274686206866e-06,
+      "loss": 0.6262,
+      "step": 5710
+    },
+    {
+      "epoch": 0.2993667242371906,
+      "grad_norm": 0.3803713023662567,
+      "learning_rate": 7.261264882983024e-06,
+      "loss": 0.6104,
+      "step": 5720
+    },
+    {
+      "epoch": 0.29989009263620664,
+      "grad_norm": 0.382207989692688,
+      "learning_rate": 6.8337107123900386e-06,
+      "loss": 0.5738,
+      "step": 5730
+    },
+    {
+      "epoch": 0.3004134610352227,
+      "grad_norm": 0.39442846179008484,
+      "learning_rate": 6.417807859384881e-06,
+      "loss": 0.6163,
+      "step": 5740
+    },
+    {
+      "epoch": 0.3009368294342388,
+      "grad_norm": 0.3684907555580139,
+      "learning_rate": 6.014719425205009e-06,
+      "loss": 0.6046,
+      "step": 5750
+    },
+    {
+      "epoch": 0.30146019783325484,
+      "grad_norm": 0.37866348028182983,
+      "learning_rate": 5.6255726746749275e-06,
+      "loss": 0.5828,
+      "step": 5760
+    },
+    {
+      "epoch": 0.3019835662322709,
+      "grad_norm": 0.38500234484672546,
+      "learning_rate": 5.251455883731728e-06,
+      "loss": 0.6056,
+      "step": 5770
+    },
+    {
+      "epoch": 0.302506934631287,
+      "grad_norm": 0.4282638430595398,
+      "learning_rate": 4.893415295985812e-06,
+      "loss": 0.6071,
+      "step": 5780
+    },
+    {
+      "epoch": 0.30303030303030304,
+      "grad_norm": 0.4080576002597809,
+      "learning_rate": 4.552452196827885e-06,
+      "loss": 0.5835,
+      "step": 5790
+    },
+    {
+      "epoch": 0.3035536714293191,
+      "grad_norm": 0.3698746860027313,
+      "learning_rate": 4.229520113264785e-06,
+      "loss": 0.6342,
+      "step": 5800
+    },
+    {
+      "epoch": 0.30407703982833517,
+      "grad_norm": 0.388528972864151,
+      "learning_rate": 3.925522147314915e-06,
+      "loss": 0.6107,
+      "step": 5810
+    },
+    {
+      "epoch": 0.30460040822735124,
+      "grad_norm": 0.4334981441497803,
+      "learning_rate": 3.641308450420675e-06,
+      "loss": 0.5793,
+      "step": 5820
+    },
+    {
+      "epoch": 0.3051237766263673,
+      "grad_norm": 0.4103776514530182,
+      "learning_rate": 3.37767384594087e-06,
+      "loss": 0.6098,
+      "step": 5830
+    },
+    {
+      "epoch": 0.30564714502538337,
+      "grad_norm": 0.386342316865921,
+      "learning_rate": 3.1353556063719357e-06,
+      "loss": 0.5895,
+      "step": 5840
+    },
+    {
+      "epoch": 0.30617051342439944,
+      "grad_norm": 0.39483997225761414,
+      "learning_rate": 2.9150313915141474e-06,
+      "loss": 0.5973,
+      "step": 5850
+    },
+    {
+      "epoch": 0.3066938818234155,
+      "grad_norm": 0.3758963942527771,
+      "learning_rate": 2.717317353348891e-06,
+      "loss": 0.5851,
+      "step": 5860
+    },
+    {
+      "epoch": 0.30721725022243157,
+      "grad_norm": 0.37651148438453674,
+      "learning_rate": 2.5427664129268253e-06,
+      "loss": 0.5944,
+      "step": 5870
+    },
+    {
+      "epoch": 0.30774061862144764,
+      "grad_norm": 0.3909638226032257,
+      "learning_rate": 2.391866714085717e-06,
+      "loss": 0.5946,
+      "step": 5880
+    },
+    {
+      "epoch": 0.3082639870204637,
+      "grad_norm": 0.3788841664791107,
+      "learning_rate": 2.2650402583222753e-06,
+      "loss": 0.6036,
+      "step": 5890
+    },
+    {
+      "epoch": 0.30878735541947977,
+      "grad_norm": 0.3904835283756256,
+      "learning_rate": 2.1626417246356398e-06,
+      "loss": 0.5993,
+      "step": 5900
+    },
+    {
+      "epoch": 0.30931072381849584,
+      "grad_norm": 0.4416792690753937,
+      "learning_rate": 2.084957477642894e-06,
+      "loss": 0.5919,
+      "step": 5910
+    },
+    {
+      "epoch": 0.3098340922175119,
+      "grad_norm": 0.4406580626964569,
+      "learning_rate": 2.0322047667405147e-06,
+      "loss": 0.5997,
+      "step": 5920
+    },
+    {
+      "epoch": 0.31035746061652797,
+      "grad_norm": 0.39945074915885925,
+      "learning_rate": 2.0045311185513342e-06,
+      "loss": 0.6132,
+      "step": 5930
+    }
+  ],
+  "logging_steps": 10,
+  "max_steps": 5935,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 1,
+  "save_steps": 2035,
+  "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.6494000762504348e+19,
+  "train_batch_size": 1,
+  "trial_name": null,
+  "trial_params": null
+}