{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.0354131534569984,
  "eval_steps": 500,
  "global_step": 5400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.005621135469364812,
      "grad_norm": 19.072066040018505,
      "learning_rate": 7.223113964686999e-07,
      "loss": 8.2549,
      "step": 10
    },
    {
      "epoch": 0.011242270938729624,
      "grad_norm": 19.552394701284502,
      "learning_rate": 1.5248796147672553e-06,
      "loss": 8.1791,
      "step": 20
    },
    {
      "epoch": 0.016863406408094434,
      "grad_norm": 9.93392906727828,
      "learning_rate": 2.3274478330658108e-06,
      "loss": 7.6759,
      "step": 30
    },
    {
      "epoch": 0.022484541877459248,
      "grad_norm": 7.426576059473498,
      "learning_rate": 3.1300160513643665e-06,
      "loss": 6.913,
      "step": 40
    },
    {
      "epoch": 0.028105677346824058,
      "grad_norm": 3.5646171014288366,
      "learning_rate": 3.932584269662922e-06,
      "loss": 6.3931,
      "step": 50
    },
    {
      "epoch": 0.03372681281618887,
      "grad_norm": 2.8779163477648666,
      "learning_rate": 4.735152487961477e-06,
      "loss": 6.0533,
      "step": 60
    },
    {
      "epoch": 0.03934794828555368,
      "grad_norm": 3.3997232487978883,
      "learning_rate": 5.537720706260032e-06,
      "loss": 5.7246,
      "step": 70
    },
    {
      "epoch": 0.044969083754918496,
      "grad_norm": 4.059106771548473,
      "learning_rate": 6.340288924558588e-06,
      "loss": 5.2465,
      "step": 80
    },
    {
      "epoch": 0.050590219224283306,
      "grad_norm": 4.7307302237777895,
      "learning_rate": 7.142857142857143e-06,
      "loss": 4.6393,
      "step": 90
    },
    {
      "epoch": 0.056211354693648116,
      "grad_norm": 5.242698651813807,
      "learning_rate": 7.9454253611557e-06,
      "loss": 4.0362,
      "step": 100
    },
    {
      "epoch": 0.061832490163012926,
      "grad_norm": 4.723996169899083,
      "learning_rate": 8.747993579454254e-06,
      "loss": 3.549,
      "step": 110
    },
    {
      "epoch": 0.06745362563237774,
      "grad_norm": 2.885143750378497,
      "learning_rate": 9.550561797752809e-06,
      "loss": 3.2171,
      "step": 120
    },
    {
      "epoch": 0.07307476110174255,
      "grad_norm": 6.825878781249646,
      "learning_rate": 1.0353130016051364e-05,
      "loss": 3.027,
      "step": 130
    },
    {
      "epoch": 0.07869589657110736,
      "grad_norm": 6.737230278162123,
      "learning_rate": 1.115569823434992e-05,
      "loss": 2.9055,
      "step": 140
    },
    {
      "epoch": 0.08431703204047218,
      "grad_norm": 6.716587811023919,
      "learning_rate": 1.1958266452648475e-05,
      "loss": 2.8089,
      "step": 150
    },
    {
      "epoch": 0.08993816750983699,
      "grad_norm": 7.201415197148535,
      "learning_rate": 1.276083467094703e-05,
      "loss": 2.7296,
      "step": 160
    },
    {
      "epoch": 0.0955593029792018,
      "grad_norm": 7.301380585480233,
      "learning_rate": 1.3563402889245585e-05,
      "loss": 2.6609,
      "step": 170
    },
    {
      "epoch": 0.10118043844856661,
      "grad_norm": 11.575326761105941,
      "learning_rate": 1.4365971107544143e-05,
      "loss": 2.5979,
      "step": 180
    },
    {
      "epoch": 0.10680157391793142,
      "grad_norm": 8.619354697957048,
      "learning_rate": 1.5168539325842698e-05,
      "loss": 2.5341,
      "step": 190
    },
    {
      "epoch": 0.11242270938729623,
      "grad_norm": 7.390257991673745,
      "learning_rate": 1.5971107544141253e-05,
      "loss": 2.4767,
      "step": 200
    },
    {
      "epoch": 0.11804384485666104,
      "grad_norm": 13.39500588081804,
      "learning_rate": 1.6773675762439807e-05,
      "loss": 2.4379,
      "step": 210
    },
    {
      "epoch": 0.12366498032602585,
      "grad_norm": 27.117334976761317,
      "learning_rate": 1.7576243980738362e-05,
      "loss": 2.3867,
      "step": 220
    },
    {
      "epoch": 0.12928611579539068,
      "grad_norm": 11.695868441564258,
      "learning_rate": 1.8378812199036917e-05,
      "loss": 2.3207,
      "step": 230
    },
    {
      "epoch": 0.13490725126475547,
      "grad_norm": 10.785553590242758,
      "learning_rate": 1.9181380417335472e-05,
      "loss": 2.2455,
      "step": 240
    },
    {
      "epoch": 0.1405283867341203,
      "grad_norm": 25.384076838053424,
      "learning_rate": 1.998394863563403e-05,
      "loss": 2.1723,
      "step": 250
    },
    {
      "epoch": 0.1461495222034851,
      "grad_norm": 16.42978149349179,
      "learning_rate": 2.0786516853932585e-05,
      "loss": 2.1322,
      "step": 260
    },
    {
      "epoch": 0.15177065767284992,
      "grad_norm": 15.299053481717623,
      "learning_rate": 2.158908507223114e-05,
      "loss": 2.0356,
      "step": 270
    },
    {
      "epoch": 0.15739179314221471,
      "grad_norm": 16.598966653865332,
      "learning_rate": 2.2391653290529695e-05,
      "loss": 1.9785,
      "step": 280
    },
    {
      "epoch": 0.16301292861157954,
      "grad_norm": 8.000521357822016,
      "learning_rate": 2.319422150882825e-05,
      "loss": 1.8885,
      "step": 290
    },
    {
      "epoch": 0.16863406408094436,
      "grad_norm": 12.337240959757104,
      "learning_rate": 2.3996789727126808e-05,
      "loss": 1.7929,
      "step": 300
    },
    {
      "epoch": 0.17425519955030916,
      "grad_norm": 9.313580975453032,
      "learning_rate": 2.4799357945425362e-05,
      "loss": 1.7252,
      "step": 310
    },
    {
      "epoch": 0.17987633501967398,
      "grad_norm": 9.713971076745432,
      "learning_rate": 2.5601926163723917e-05,
      "loss": 1.6366,
      "step": 320
    },
    {
      "epoch": 0.18549747048903878,
      "grad_norm": 6.604247664050388,
      "learning_rate": 2.6404494382022472e-05,
      "loss": 1.5532,
      "step": 330
    },
    {
      "epoch": 0.1911186059584036,
      "grad_norm": 8.06631111933664,
      "learning_rate": 2.7207062600321027e-05,
      "loss": 1.5014,
      "step": 340
    },
    {
      "epoch": 0.1967397414277684,
      "grad_norm": 5.369237324186266,
      "learning_rate": 2.8009630818619585e-05,
      "loss": 1.4407,
      "step": 350
    },
    {
      "epoch": 0.20236087689713322,
      "grad_norm": 6.309483928633083,
      "learning_rate": 2.881219903691814e-05,
      "loss": 1.398,
      "step": 360
    },
    {
      "epoch": 0.20798201236649802,
      "grad_norm": 6.173807170921828,
      "learning_rate": 2.9614767255216698e-05,
      "loss": 1.3462,
      "step": 370
    },
    {
      "epoch": 0.21360314783586284,
      "grad_norm": 7.913665828353373,
      "learning_rate": 3.0417335473515253e-05,
      "loss": 1.2942,
      "step": 380
    },
    {
      "epoch": 0.21922428330522767,
      "grad_norm": 6.320887435288008,
      "learning_rate": 3.1219903691813805e-05,
      "loss": 1.2581,
      "step": 390
    },
    {
      "epoch": 0.22484541877459246,
      "grad_norm": 4.001174978157139,
      "learning_rate": 3.202247191011236e-05,
      "loss": 1.2068,
      "step": 400
    },
    {
      "epoch": 0.2304665542439573,
      "grad_norm": 5.833384870789841,
      "learning_rate": 3.2825040128410914e-05,
      "loss": 1.1774,
      "step": 410
    },
    {
      "epoch": 0.23608768971332209,
      "grad_norm": 5.418298506902902,
      "learning_rate": 3.362760834670947e-05,
      "loss": 1.1285,
      "step": 420
    },
    {
      "epoch": 0.2417088251826869,
      "grad_norm": 6.822807280672466,
      "learning_rate": 3.4430176565008024e-05,
      "loss": 1.0971,
      "step": 430
    },
    {
      "epoch": 0.2473299606520517,
      "grad_norm": 5.1633853492959725,
      "learning_rate": 3.523274478330658e-05,
      "loss": 1.0631,
      "step": 440
    },
    {
      "epoch": 0.25295109612141653,
      "grad_norm": 4.794453623656116,
      "learning_rate": 3.6035313001605133e-05,
      "loss": 1.0233,
      "step": 450
    },
    {
      "epoch": 0.25857223159078135,
      "grad_norm": 6.493147309773226,
      "learning_rate": 3.683788121990369e-05,
      "loss": 1.0093,
      "step": 460
    },
    {
      "epoch": 0.2641933670601461,
      "grad_norm": 7.38273852542595,
      "learning_rate": 3.764044943820225e-05,
      "loss": 0.9745,
      "step": 470
    },
    {
      "epoch": 0.26981450252951095,
      "grad_norm": 5.953745454074908,
      "learning_rate": 3.8443017656500805e-05,
      "loss": 0.9609,
      "step": 480
    },
    {
      "epoch": 0.27543563799887577,
      "grad_norm": 5.92674703602574,
      "learning_rate": 3.924558587479936e-05,
      "loss": 0.9295,
      "step": 490
    },
    {
      "epoch": 0.2810567734682406,
      "grad_norm": 4.500282226524396,
      "learning_rate": 4.0048154093097914e-05,
      "loss": 0.8992,
      "step": 500
    },
    {
      "epoch": 0.2866779089376054,
      "grad_norm": 4.320018263048497,
      "learning_rate": 4.085072231139647e-05,
      "loss": 0.8855,
      "step": 510
    },
    {
      "epoch": 0.2922990444069702,
      "grad_norm": 6.008360095853525,
      "learning_rate": 4.1653290529695024e-05,
      "loss": 0.8589,
      "step": 520
    },
    {
      "epoch": 0.297920179876335,
      "grad_norm": 5.252075871228091,
      "learning_rate": 4.245585874799358e-05,
      "loss": 0.8272,
      "step": 530
    },
    {
      "epoch": 0.30354131534569984,
      "grad_norm": 6.427285353419691,
      "learning_rate": 4.3258426966292134e-05,
      "loss": 0.808,
      "step": 540
    },
    {
      "epoch": 0.30916245081506466,
      "grad_norm": 5.854800753365741,
      "learning_rate": 4.4060995184590695e-05,
      "loss": 0.7901,
      "step": 550
    },
    {
      "epoch": 0.31478358628442943,
      "grad_norm": 7.046341886366144,
      "learning_rate": 4.486356340288925e-05,
      "loss": 0.7701,
      "step": 560
    },
    {
      "epoch": 0.32040472175379425,
      "grad_norm": 5.594440118602691,
      "learning_rate": 4.5666131621187805e-05,
      "loss": 0.7462,
      "step": 570
    },
    {
      "epoch": 0.3260258572231591,
      "grad_norm": 5.653552049437021,
      "learning_rate": 4.646869983948636e-05,
      "loss": 0.7284,
      "step": 580
    },
    {
      "epoch": 0.3316469926925239,
      "grad_norm": 5.196665356454571,
      "learning_rate": 4.7271268057784915e-05,
      "loss": 0.712,
      "step": 590
    },
    {
      "epoch": 0.3372681281618887,
      "grad_norm": 6.1381022060931025,
      "learning_rate": 4.807383627608347e-05,
      "loss": 0.6868,
      "step": 600
    },
    {
      "epoch": 0.3428892636312535,
      "grad_norm": 5.052546164435208,
      "learning_rate": 4.8876404494382024e-05,
      "loss": 0.6672,
      "step": 610
    },
    {
      "epoch": 0.3485103991006183,
      "grad_norm": 4.916726449591455,
      "learning_rate": 4.9678972712680586e-05,
      "loss": 0.6501,
      "step": 620
    },
    {
      "epoch": 0.35413153456998314,
      "grad_norm": 4.832632041097204,
      "learning_rate": 5.0481540930979134e-05,
      "loss": 0.6313,
      "step": 630
    },
    {
      "epoch": 0.35975267003934797,
      "grad_norm": 4.751573877646501,
      "learning_rate": 5.1284109149277696e-05,
      "loss": 0.6165,
      "step": 640
    },
    {
      "epoch": 0.36537380550871273,
      "grad_norm": 5.912237963361695,
      "learning_rate": 5.2086677367576244e-05,
      "loss": 0.5907,
      "step": 650
    },
    {
      "epoch": 0.37099494097807756,
      "grad_norm": 4.086167788806448,
      "learning_rate": 5.2889245585874805e-05,
      "loss": 0.5768,
      "step": 660
    },
    {
      "epoch": 0.3766160764474424,
      "grad_norm": 4.582982284092926,
      "learning_rate": 5.369181380417335e-05,
      "loss": 0.5579,
      "step": 670
    },
    {
      "epoch": 0.3822372119168072,
      "grad_norm": 4.620402866105723,
      "learning_rate": 5.4494382022471915e-05,
      "loss": 0.5379,
      "step": 680
    },
    {
      "epoch": 0.38785834738617203,
      "grad_norm": 3.8404903506330834,
      "learning_rate": 5.5296950240770476e-05,
      "loss": 0.5112,
      "step": 690
    },
    {
      "epoch": 0.3934794828555368,
      "grad_norm": 3.2956728880451784,
      "learning_rate": 5.6099518459069025e-05,
      "loss": 0.502,
      "step": 700
    },
    {
      "epoch": 0.3991006183249016,
      "grad_norm": 4.455851011958456,
      "learning_rate": 5.6902086677367586e-05,
      "loss": 0.4889,
      "step": 710
    },
    {
      "epoch": 0.40472175379426645,
      "grad_norm": 3.3081515499225693,
      "learning_rate": 5.7704654895666134e-05,
      "loss": 0.4703,
      "step": 720
    },
    {
      "epoch": 0.41034288926363127,
      "grad_norm": 3.5766535302434037,
      "learning_rate": 5.8507223113964696e-05,
      "loss": 0.4591,
      "step": 730
    },
    {
      "epoch": 0.41596402473299604,
      "grad_norm": 3.1816986592066154,
      "learning_rate": 5.9309791332263244e-05,
      "loss": 0.437,
      "step": 740
    },
    {
      "epoch": 0.42158516020236086,
      "grad_norm": 2.746088460027587,
      "learning_rate": 6.0112359550561805e-05,
      "loss": 0.429,
      "step": 750
    },
    {
      "epoch": 0.4272062956717257,
      "grad_norm": 2.7638950111486777,
      "learning_rate": 6.0914927768860353e-05,
      "loss": 0.4056,
      "step": 760
    },
    {
      "epoch": 0.4328274311410905,
      "grad_norm": 2.621320210943319,
      "learning_rate": 6.171749598715892e-05,
      "loss": 0.3951,
      "step": 770
    },
    {
      "epoch": 0.43844856661045534,
      "grad_norm": 2.9039501807988635,
      "learning_rate": 6.252006420545746e-05,
      "loss": 0.3752,
      "step": 780
    },
    {
      "epoch": 0.4440697020798201,
      "grad_norm": 2.672708782303941,
      "learning_rate": 6.332263242375602e-05,
      "loss": 0.3732,
      "step": 790
    },
    {
      "epoch": 0.44969083754918493,
      "grad_norm": 2.534243622485374,
      "learning_rate": 6.412520064205457e-05,
      "loss": 0.3544,
      "step": 800
    },
    {
      "epoch": 0.45531197301854975,
      "grad_norm": 2.2060854376474297,
      "learning_rate": 6.492776886035313e-05,
      "loss": 0.337,
      "step": 810
    },
    {
      "epoch": 0.4609331084879146,
      "grad_norm": 2.8950080552663353,
      "learning_rate": 6.573033707865169e-05,
      "loss": 0.3278,
      "step": 820
    },
    {
      "epoch": 0.46655424395727935,
      "grad_norm": 2.8077239251031183,
      "learning_rate": 6.653290529695024e-05,
      "loss": 0.3188,
      "step": 830
    },
    {
      "epoch": 0.47217537942664417,
      "grad_norm": 2.4109713129957173,
      "learning_rate": 6.73354735152488e-05,
      "loss": 0.3023,
      "step": 840
    },
    {
      "epoch": 0.477796514896009,
      "grad_norm": 2.4108771548493864,
      "learning_rate": 6.813804173354735e-05,
      "loss": 0.2907,
      "step": 850
    },
    {
      "epoch": 0.4834176503653738,
      "grad_norm": 2.381688936764209,
      "learning_rate": 6.894060995184591e-05,
      "loss": 0.28,
      "step": 860
    },
    {
      "epoch": 0.48903878583473864,
      "grad_norm": 2.226240645538486,
      "learning_rate": 6.974317817014446e-05,
      "loss": 0.265,
      "step": 870
    },
    {
      "epoch": 0.4946599213041034,
      "grad_norm": 2.36477519460133,
      "learning_rate": 7.054574638844302e-05,
      "loss": 0.2563,
      "step": 880
    },
    {
      "epoch": 0.5002810567734682,
      "grad_norm": 2.1877534753089325,
      "learning_rate": 7.134831460674157e-05,
      "loss": 0.2458,
      "step": 890
    },
    {
      "epoch": 0.5059021922428331,
      "grad_norm": 1.9910110516875703,
      "learning_rate": 7.215088282504013e-05,
      "loss": 0.237,
      "step": 900
    },
    {
      "epoch": 0.5115233277121979,
      "grad_norm": 2.6541637478760203,
      "learning_rate": 7.295345104333868e-05,
      "loss": 0.2274,
      "step": 910
    },
    {
      "epoch": 0.5171444631815627,
      "grad_norm": 2.1481697620944273,
      "learning_rate": 7.375601926163724e-05,
      "loss": 0.2125,
      "step": 920
    },
    {
      "epoch": 0.5227655986509275,
      "grad_norm": 2.2194085230343283,
      "learning_rate": 7.455858747993579e-05,
      "loss": 0.2086,
      "step": 930
    },
    {
      "epoch": 0.5283867341202922,
      "grad_norm": 1.8687719324053211,
      "learning_rate": 7.536115569823435e-05,
      "loss": 0.2002,
      "step": 940
    },
    {
      "epoch": 0.5340078695896571,
      "grad_norm": 1.809218029026303,
      "learning_rate": 7.616372391653292e-05,
      "loss": 0.19,
      "step": 950
    },
    {
      "epoch": 0.5396290050590219,
      "grad_norm": 1.544075304452749,
      "learning_rate": 7.696629213483147e-05,
      "loss": 0.1824,
      "step": 960
    },
    {
      "epoch": 0.5452501405283867,
      "grad_norm": 1.7029729768183128,
      "learning_rate": 7.776886035313003e-05,
      "loss": 0.1766,
      "step": 970
    },
    {
      "epoch": 0.5508712759977515,
      "grad_norm": 1.9321640691371915,
      "learning_rate": 7.857142857142858e-05,
      "loss": 0.1674,
      "step": 980
    },
    {
      "epoch": 0.5564924114671164,
      "grad_norm": 1.6567308102186273,
      "learning_rate": 7.937399678972713e-05,
      "loss": 0.1616,
      "step": 990
    },
    {
      "epoch": 0.5621135469364812,
      "grad_norm": 1.75914588466869,
      "learning_rate": 8.017656500802569e-05,
      "loss": 0.1496,
      "step": 1000
    },
    {
      "epoch": 0.567734682405846,
      "grad_norm": 1.6033430089086067,
      "learning_rate": 8.097913322632424e-05,
      "loss": 0.1411,
      "step": 1010
    },
    {
      "epoch": 0.5733558178752108,
      "grad_norm": 1.5592637787510577,
      "learning_rate": 8.17817014446228e-05,
      "loss": 0.1349,
      "step": 1020
    },
    {
      "epoch": 0.5789769533445756,
      "grad_norm": 1.490862273444445,
      "learning_rate": 8.258426966292135e-05,
      "loss": 0.1321,
      "step": 1030
    },
    {
      "epoch": 0.5845980888139404,
      "grad_norm": 1.103403083662274,
      "learning_rate": 8.338683788121991e-05,
      "loss": 0.1237,
      "step": 1040
    },
    {
      "epoch": 0.5902192242833052,
      "grad_norm": 1.3322879027309737,
      "learning_rate": 8.418940609951846e-05,
      "loss": 0.1175,
      "step": 1050
    },
    {
      "epoch": 0.59584035975267,
      "grad_norm": 1.1302428773890956,
      "learning_rate": 8.499197431781702e-05,
      "loss": 0.1134,
      "step": 1060
    },
    {
      "epoch": 0.6014614952220348,
      "grad_norm": 1.043524404705153,
      "learning_rate": 8.579454253611557e-05,
      "loss": 0.1079,
      "step": 1070
    },
    {
      "epoch": 0.6070826306913997,
      "grad_norm": 1.0807715777082445,
      "learning_rate": 8.659711075441413e-05,
      "loss": 0.1027,
      "step": 1080
    },
    {
      "epoch": 0.6127037661607645,
      "grad_norm": 1.0613560487233085,
      "learning_rate": 8.739967897271268e-05,
      "loss": 0.1014,
      "step": 1090
    },
    {
      "epoch": 0.6183249016301293,
      "grad_norm": 1.7970050911102724,
      "learning_rate": 8.820224719101124e-05,
      "loss": 0.0956,
      "step": 1100
    },
    {
      "epoch": 0.6239460370994941,
      "grad_norm": 1.2518666492303272,
      "learning_rate": 8.900481540930979e-05,
      "loss": 0.0934,
      "step": 1110
    },
    {
      "epoch": 0.6295671725688589,
      "grad_norm": 1.1953063290224126,
      "learning_rate": 8.980738362760836e-05,
      "loss": 0.1004,
      "step": 1120
    },
    {
      "epoch": 0.6351883080382237,
      "grad_norm": 0.9616154833979669,
      "learning_rate": 9.06099518459069e-05,
      "loss": 0.0843,
      "step": 1130
    },
    {
      "epoch": 0.6408094435075885,
      "grad_norm": 0.8282021805405166,
      "learning_rate": 9.141252006420547e-05,
      "loss": 0.0788,
      "step": 1140
    },
    {
      "epoch": 0.6464305789769533,
      "grad_norm": 0.7645874993850531,
      "learning_rate": 9.221508828250401e-05,
      "loss": 0.0741,
      "step": 1150
    },
    {
      "epoch": 0.6520517144463182,
      "grad_norm": 1.0333771317137108,
      "learning_rate": 9.301765650080258e-05,
      "loss": 0.0708,
      "step": 1160
    },
    {
      "epoch": 0.657672849915683,
      "grad_norm": 0.6189773458607581,
      "learning_rate": 9.382022471910112e-05,
      "loss": 0.0722,
      "step": 1170
    },
    {
      "epoch": 0.6632939853850478,
      "grad_norm": 0.5553259399184006,
      "learning_rate": 9.462279293739969e-05,
      "loss": 0.0654,
      "step": 1180
    },
    {
      "epoch": 0.6689151208544126,
      "grad_norm": 0.6418699826913336,
      "learning_rate": 9.542536115569823e-05,
      "loss": 0.0625,
      "step": 1190
    },
    {
      "epoch": 0.6745362563237775,
      "grad_norm": 0.692560675949567,
      "learning_rate": 9.62279293739968e-05,
      "loss": 0.0591,
      "step": 1200
    },
    {
      "epoch": 0.6801573917931422,
      "grad_norm": 0.4228488541088058,
      "learning_rate": 9.703049759229535e-05,
      "loss": 0.0565,
      "step": 1210
    },
    {
      "epoch": 0.685778527262507,
      "grad_norm": 0.4504043674769784,
      "learning_rate": 9.783306581059391e-05,
      "loss": 0.0546,
      "step": 1220
    },
    {
      "epoch": 0.6913996627318718,
      "grad_norm": 0.44916766960088017,
      "learning_rate": 9.863563402889246e-05,
      "loss": 0.0518,
      "step": 1230
    },
    {
      "epoch": 0.6970207982012366,
      "grad_norm": 0.46844229890271083,
      "learning_rate": 9.943820224719102e-05,
      "loss": 0.0504,
      "step": 1240
    },
    {
      "epoch": 0.7026419336706015,
      "grad_norm": 0.4911174405862922,
      "learning_rate": 9.999998231913558e-05,
      "loss": 0.0482,
      "step": 1250
    },
    {
      "epoch": 0.7082630691399663,
      "grad_norm": 0.3927574276223742,
      "learning_rate": 9.999966799300477e-05,
      "loss": 0.0469,
      "step": 1260
    },
    {
      "epoch": 0.7138842046093311,
      "grad_norm": 0.3446568196906315,
      "learning_rate": 9.999896076161874e-05,
      "loss": 0.0447,
      "step": 1270
    },
    {
      "epoch": 0.7195053400786959,
      "grad_norm": 0.6708037379046565,
      "learning_rate": 9.999786063053502e-05,
      "loss": 0.0437,
      "step": 1280
    },
    {
      "epoch": 0.7251264755480608,
      "grad_norm": 0.5265622160752478,
      "learning_rate": 9.999636760839859e-05,
      "loss": 0.0428,
      "step": 1290
    },
    {
      "epoch": 0.7307476110174255,
      "grad_norm": 0.3741304206840176,
      "learning_rate": 9.999448170694188e-05,
      "loss": 0.0418,
      "step": 1300
    },
    {
      "epoch": 0.7363687464867903,
      "grad_norm": 0.3488892086457768,
      "learning_rate": 9.999220294098459e-05,
      "loss": 0.0401,
      "step": 1310
    },
    {
      "epoch": 0.7419898819561551,
      "grad_norm": 0.364156745530358,
      "learning_rate": 9.998953132843362e-05,
      "loss": 0.0392,
      "step": 1320
    },
    {
      "epoch": 0.7476110174255199,
      "grad_norm": 0.316984179052284,
      "learning_rate": 9.998646689028292e-05,
      "loss": 0.038,
      "step": 1330
    },
    {
      "epoch": 0.7532321528948848,
      "grad_norm": 0.4122286636140672,
      "learning_rate": 9.998300965061335e-05,
      "loss": 0.0369,
      "step": 1340
    },
    {
      "epoch": 0.7588532883642496,
      "grad_norm": 0.2828777162977292,
      "learning_rate": 9.997915963659241e-05,
      "loss": 0.0358,
      "step": 1350
    },
    {
      "epoch": 0.7644744238336144,
      "grad_norm": 0.35038412611293795,
      "learning_rate": 9.997491687847413e-05,
      "loss": 0.0344,
      "step": 1360
    },
    {
      "epoch": 0.7700955593029792,
      "grad_norm": 0.3359568867412438,
      "learning_rate": 9.997028140959878e-05,
      "loss": 0.0348,
      "step": 1370
    },
    {
      "epoch": 0.7757166947723441,
      "grad_norm": 0.3008434778248411,
      "learning_rate": 9.99652532663926e-05,
      "loss": 0.0331,
      "step": 1380
    },
    {
      "epoch": 0.7813378302417088,
      "grad_norm": 0.24375407955159972,
      "learning_rate": 9.995983248836749e-05,
      "loss": 0.0321,
      "step": 1390
    },
    {
      "epoch": 0.7869589657110736,
      "grad_norm": 0.2580356766574075,
      "learning_rate": 9.995401911812084e-05,
      "loss": 0.0305,
      "step": 1400
    },
    {
      "epoch": 0.7925801011804384,
      "grad_norm": 0.22377682987678044,
      "learning_rate": 9.994781320133499e-05,
      "loss": 0.0297,
      "step": 1410
    },
    {
      "epoch": 0.7982012366498032,
      "grad_norm": 0.3192038760895579,
      "learning_rate": 9.9941214786777e-05,
      "loss": 0.0295,
      "step": 1420
    },
    {
      "epoch": 0.8038223721191681,
      "grad_norm": 0.42209606301924135,
      "learning_rate": 9.99342239262983e-05,
      "loss": 0.0283,
      "step": 1430
    },
    {
      "epoch": 0.8094435075885329,
      "grad_norm": 0.2740277370912408,
      "learning_rate": 9.992684067483411e-05,
      "loss": 0.0276,
      "step": 1440
    },
    {
      "epoch": 0.8150646430578977,
      "grad_norm": 0.2403536361378785,
      "learning_rate": 9.991906509040323e-05,
      "loss": 0.027,
      "step": 1450
    },
    {
      "epoch": 0.8206857785272625,
      "grad_norm": 0.25177256736722925,
      "learning_rate": 9.99108972341074e-05,
      "loss": 0.026,
      "step": 1460
    },
    {
      "epoch": 0.8263069139966274,
      "grad_norm": 0.2611127538329995,
      "learning_rate": 9.990233717013092e-05,
      "loss": 0.0254,
      "step": 1470
    },
    {
      "epoch": 0.8319280494659921,
      "grad_norm": 0.26600699519251914,
      "learning_rate": 9.989338496574013e-05,
      "loss": 0.0246,
      "step": 1480
    },
    {
      "epoch": 0.8375491849353569,
      "grad_norm": 0.2511518661704276,
      "learning_rate": 9.988404069128286e-05,
      "loss": 0.0241,
      "step": 1490
    },
    {
      "epoch": 0.8431703204047217,
      "grad_norm": 0.24275092866635561,
      "learning_rate": 9.987430442018787e-05,
      "loss": 0.0231,
      "step": 1500
    },
    {
      "epoch": 0.8487914558740866,
      "grad_norm": 0.2117797837985066,
      "learning_rate": 9.986417622896433e-05,
      "loss": 0.0222,
      "step": 1510
    },
    {
      "epoch": 0.8544125913434514,
      "grad_norm": 0.23706464693081078,
      "learning_rate": 9.985365619720114e-05,
      "loss": 0.0224,
      "step": 1520
    },
    {
      "epoch": 0.8600337268128162,
      "grad_norm": 0.21681891631644465,
      "learning_rate": 9.984274440756639e-05,
      "loss": 0.0216,
      "step": 1530
    },
    {
      "epoch": 0.865654862282181,
      "grad_norm": 0.21895203687881423,
      "learning_rate": 9.983144094580662e-05,
      "loss": 0.0205,
      "step": 1540
    },
    {
      "epoch": 0.8712759977515458,
      "grad_norm": 0.20958525636292016,
      "learning_rate": 9.981974590074623e-05,
      "loss": 0.0197,
      "step": 1550
    },
    {
      "epoch": 0.8768971332209107,
      "grad_norm": 0.2092568671749629,
      "learning_rate": 9.980765936428668e-05,
      "loss": 0.0192,
      "step": 1560
    },
    {
      "epoch": 0.8825182686902754,
      "grad_norm": 0.20242771809585056,
      "learning_rate": 9.979518143140591e-05,
      "loss": 0.0186,
      "step": 1570
    },
    {
      "epoch": 0.8881394041596402,
      "grad_norm": 0.2067518737526698,
      "learning_rate": 9.97823122001575e-05,
      "loss": 0.0184,
      "step": 1580
    },
    {
      "epoch": 0.893760539629005,
      "grad_norm": 0.2131831404574167,
      "learning_rate": 9.976905177166984e-05,
      "loss": 0.0179,
      "step": 1590
    },
    {
      "epoch": 0.8993816750983699,
      "grad_norm": 0.19975090191123349,
      "learning_rate": 9.975540025014551e-05,
      "loss": 0.0171,
      "step": 1600
    },
    {
      "epoch": 0.9050028105677347,
      "grad_norm": 0.1840388312874417,
      "learning_rate": 9.974135774286027e-05,
      "loss": 0.0167,
      "step": 1610
    },
    {
      "epoch": 0.9106239460370995,
      "grad_norm": 0.16080844246323525,
      "learning_rate": 9.972692436016237e-05,
      "loss": 0.0162,
      "step": 1620
    },
    {
      "epoch": 0.9162450815064643,
      "grad_norm": 0.7406842420136835,
      "learning_rate": 9.971210021547163e-05,
      "loss": 0.0164,
      "step": 1630
    },
    {
      "epoch": 0.9218662169758292,
      "grad_norm": 0.6842280048869254,
      "learning_rate": 9.969688542527847e-05,
      "loss": 0.0185,
      "step": 1640
    },
    {
      "epoch": 0.927487352445194,
      "grad_norm": 1.580555984265757,
      "learning_rate": 9.968128010914313e-05,
      "loss": 0.0316,
      "step": 1650
    },
    {
      "epoch": 0.9331084879145587,
      "grad_norm": 0.37860426611015985,
      "learning_rate": 9.966528438969462e-05,
      "loss": 0.0249,
      "step": 1660
    },
    {
      "epoch": 0.9387296233839235,
      "grad_norm": 0.23262955055081527,
      "learning_rate": 9.964889839262983e-05,
      "loss": 0.0186,
      "step": 1670
    },
    {
      "epoch": 0.9443507588532883,
      "grad_norm": 0.20273477480045216,
      "learning_rate": 9.963212224671249e-05,
      "loss": 0.0161,
      "step": 1680
    },
    {
      "epoch": 0.9499718943226532,
      "grad_norm": 0.13881800111147724,
      "learning_rate": 9.961495608377221e-05,
      "loss": 0.0148,
      "step": 1690
    },
    {
      "epoch": 0.955593029792018,
      "grad_norm": 0.17108269648499383,
      "learning_rate": 9.959740003870339e-05,
      "loss": 0.0144,
      "step": 1700
    },
    {
      "epoch": 0.9612141652613828,
      "grad_norm": 0.16536831484992337,
      "learning_rate": 9.95794542494642e-05,
      "loss": 0.0137,
      "step": 1710
    },
    {
      "epoch": 0.9668353007307476,
      "grad_norm": 0.17400543803234356,
      "learning_rate": 9.956111885707546e-05,
      "loss": 0.0131,
      "step": 1720
    },
    {
      "epoch": 0.9724564362001125,
      "grad_norm": 0.14754960398432485,
      "learning_rate": 9.954239400561958e-05,
      "loss": 0.0126,
      "step": 1730
    },
    {
      "epoch": 0.9780775716694773,
      "grad_norm": 0.13872283525841225,
      "learning_rate": 9.952327984223938e-05,
      "loss": 0.012,
      "step": 1740
    },
    {
      "epoch": 0.983698707138842,
      "grad_norm": 0.15345071148869308,
      "learning_rate": 9.950377651713698e-05,
      "loss": 0.0119,
      "step": 1750
    },
    {
      "epoch": 0.9893198426082068,
      "grad_norm": 0.15868096773505136,
      "learning_rate": 9.948388418357255e-05,
      "loss": 0.0117,
      "step": 1760
    },
    {
      "epoch": 0.9949409780775716,
      "grad_norm": 0.13964728528048598,
      "learning_rate": 9.94636029978632e-05,
      "loss": 0.0111,
      "step": 1770
    },
    {
      "epoch": 1.0005621135469365,
      "grad_norm": 0.12990131986586648,
      "learning_rate": 9.944293311938164e-05,
      "loss": 0.0111,
      "step": 1780
    },
    {
      "epoch": 1.0061832490163012,
      "grad_norm": 0.15109117991919174,
      "learning_rate": 9.942187471055507e-05,
      "loss": 0.0106,
      "step": 1790
    },
    {
      "epoch": 1.0118043844856661,
      "grad_norm": 0.15023868490530576,
      "learning_rate": 9.940042793686378e-05,
      "loss": 0.0101,
      "step": 1800
    },
    {
      "epoch": 1.0174255199550308,
      "grad_norm": 0.13748040160742883,
      "learning_rate": 9.937859296683984e-05,
      "loss": 0.0098,
      "step": 1810
    },
    {
      "epoch": 1.0230466554243958,
      "grad_norm": 0.13694721938981147,
      "learning_rate": 9.935636997206592e-05,
      "loss": 0.0099,
      "step": 1820
    },
    {
      "epoch": 1.0286677908937605,
      "grad_norm": 0.13775398006820014,
      "learning_rate": 9.933375912717379e-05,
      "loss": 0.0096,
      "step": 1830
    },
    {
      "epoch": 1.0342889263631254,
      "grad_norm": 0.12873175645779594,
      "learning_rate": 9.931076060984306e-05,
      "loss": 0.0094,
      "step": 1840
    },
    {
      "epoch": 1.0399100618324901,
      "grad_norm": 0.13491704932600562,
      "learning_rate": 9.928737460079964e-05,
      "loss": 0.0093,
      "step": 1850
    },
    {
      "epoch": 1.045531197301855,
      "grad_norm": 0.13052217132802058,
      "learning_rate": 9.926360128381452e-05,
      "loss": 0.0089,
      "step": 1860
    },
    {
      "epoch": 1.0511523327712198,
      "grad_norm": 0.15139988915587,
      "learning_rate": 9.923944084570214e-05,
      "loss": 0.0089,
      "step": 1870
    },
    {
      "epoch": 1.0567734682405847,
      "grad_norm": 0.11412503222119044,
      "learning_rate": 9.9214893476319e-05,
      "loss": 0.0085,
      "step": 1880
    },
    {
      "epoch": 1.0623946037099494,
      "grad_norm": 0.13727989640704788,
      "learning_rate": 9.918995936856224e-05,
      "loss": 0.0085,
      "step": 1890
    },
    {
      "epoch": 1.0680157391793141,
      "grad_norm": 0.1249207256562971,
      "learning_rate": 9.916463871836796e-05,
      "loss": 0.0084,
      "step": 1900
    },
    {
      "epoch": 1.073636874648679,
      "grad_norm": 0.12406222038106689,
      "learning_rate": 9.913893172470982e-05,
      "loss": 0.0081,
      "step": 1910
    },
    {
      "epoch": 1.0792580101180438,
      "grad_norm": 0.13127597326529594,
      "learning_rate": 9.911283858959746e-05,
      "loss": 0.0083,
      "step": 1920
    },
    {
      "epoch": 1.0848791455874087,
      "grad_norm": 0.1368675589547782,
      "learning_rate": 9.908635951807477e-05,
      "loss": 0.0082,
      "step": 1930
    },
    {
      "epoch": 1.0905002810567734,
      "grad_norm": 0.1172965848549566,
      "learning_rate": 9.905949471821851e-05,
      "loss": 0.008,
      "step": 1940
    },
    {
      "epoch": 1.0961214165261384,
      "grad_norm": 0.11798553949395943,
      "learning_rate": 9.903224440113649e-05,
      "loss": 0.008,
      "step": 1950
    },
    {
      "epoch": 1.101742551995503,
      "grad_norm": 0.13016996611164902,
      "learning_rate": 9.900460878096598e-05,
      "loss": 0.0078,
      "step": 1960
    },
    {
      "epoch": 1.107363687464868,
      "grad_norm": 0.12122199750464839,
      "learning_rate": 9.897658807487202e-05,
      "loss": 0.0077,
      "step": 1970
    },
    {
      "epoch": 1.1129848229342327,
      "grad_norm": 0.11846619937347036,
      "learning_rate": 9.894818250304576e-05,
      "loss": 0.0075,
      "step": 1980
    },
    {
      "epoch": 1.1186059584035974,
      "grad_norm": 0.13622498242403422,
      "learning_rate": 9.891939228870261e-05,
      "loss": 0.0076,
      "step": 1990
    },
    {
      "epoch": 1.1242270938729624,
      "grad_norm": 0.1132057504919735,
      "learning_rate": 9.889021765808064e-05,
      "loss": 0.0076,
      "step": 2000
    },
    {
      "epoch": 1.129848229342327,
      "grad_norm": 0.12782091049281527,
      "learning_rate": 9.886065884043868e-05,
      "loss": 0.0076,
      "step": 2010
    },
    {
      "epoch": 1.135469364811692,
      "grad_norm": 0.1327283957213651,
      "learning_rate": 9.883071606805454e-05,
      "loss": 0.0072,
      "step": 2020
    },
    {
      "epoch": 1.1410905002810567,
      "grad_norm": 0.11466155852542638,
      "learning_rate": 9.880038957622327e-05,
      "loss": 0.0074,
      "step": 2030
    },
    {
      "epoch": 1.1467116357504217,
      "grad_norm": 0.1248526409625004,
      "learning_rate": 9.876967960325518e-05,
      "loss": 0.0071,
      "step": 2040
    },
    {
      "epoch": 1.1523327712197864,
      "grad_norm": 0.12458128006542332,
      "learning_rate": 9.873858639047409e-05,
      "loss": 0.0069,
      "step": 2050
    },
    {
      "epoch": 1.1579539066891513,
      "grad_norm": 0.11664744118592543,
      "learning_rate": 9.870711018221535e-05,
      "loss": 0.0066,
      "step": 2060
    },
    {
      "epoch": 1.163575042158516,
      "grad_norm": 0.12050522669971526,
      "learning_rate": 9.867525122582394e-05,
      "loss": 0.0068,
      "step": 2070
    },
    {
      "epoch": 1.1691961776278808,
      "grad_norm": 0.13065303841446785,
      "learning_rate": 9.864300977165259e-05,
      "loss": 0.0067,
      "step": 2080
    },
    {
      "epoch": 1.1748173130972457,
      "grad_norm": 0.10805422192260604,
      "learning_rate": 9.861038607305968e-05,
      "loss": 0.0066,
      "step": 2090
    },
    {
      "epoch": 1.1804384485666104,
      "grad_norm": 0.11507644557756966,
      "learning_rate": 9.857738038640737e-05,
      "loss": 0.0063,
      "step": 2100
    },
    {
      "epoch": 1.1860595840359753,
      "grad_norm": 0.10761515703888919,
      "learning_rate": 9.854399297105957e-05,
      "loss": 0.0061,
      "step": 2110
    },
    {
      "epoch": 1.19168071950534,
      "grad_norm": 0.12594601810241263,
      "learning_rate": 9.851022408937979e-05,
      "loss": 0.0067,
      "step": 2120
    },
    {
      "epoch": 1.197301854974705,
      "grad_norm": 0.12986420025775458,
      "learning_rate": 9.847607400672925e-05,
      "loss": 0.0065,
      "step": 2130
    },
    {
      "epoch": 1.2029229904440697,
      "grad_norm": 0.10577243861744662,
      "learning_rate": 9.844154299146466e-05,
      "loss": 0.0062,
      "step": 2140
    },
    {
      "epoch": 1.2085441259134346,
      "grad_norm": 0.11560521707742587,
      "learning_rate": 9.840663131493617e-05,
      "loss": 0.0063,
      "step": 2150
    },
    {
      "epoch": 1.2141652613827993,
      "grad_norm": 0.12172276197577853,
      "learning_rate": 9.837133925148523e-05,
      "loss": 0.006,
      "step": 2160
    },
    {
      "epoch": 1.219786396852164,
      "grad_norm": 0.11989705951251413,
      "learning_rate": 9.833566707844241e-05,
      "loss": 0.0061,
      "step": 2170
    },
    {
      "epoch": 1.225407532321529,
      "grad_norm": 0.12073736751173343,
      "learning_rate": 9.829961507612526e-05,
      "loss": 0.0062,
      "step": 2180
    },
    {
      "epoch": 1.2310286677908937,
      "grad_norm": 0.10581806093891255,
      "learning_rate": 9.82631835278361e-05,
      "loss": 0.006,
      "step": 2190
    },
    {
      "epoch": 1.2366498032602586,
      "grad_norm": 0.12350567107631101,
      "learning_rate": 9.822637271985977e-05,
      "loss": 0.0058,
      "step": 2200
    },
    {
      "epoch": 1.2422709387296234,
      "grad_norm": 0.10706106863986312,
      "learning_rate": 9.818918294146136e-05,
      "loss": 0.0055,
      "step": 2210
    },
    {
      "epoch": 1.2478920741989883,
      "grad_norm": 0.11497496683839335,
      "learning_rate": 9.815161448488403e-05,
      "loss": 0.0059,
      "step": 2220
    },
    {
      "epoch": 1.253513209668353,
      "grad_norm": 0.1230577251222716,
      "learning_rate": 9.81136676453466e-05,
      "loss": 0.0056,
      "step": 2230
    },
    {
      "epoch": 1.259134345137718,
      "grad_norm": 0.11408509041104613,
      "learning_rate": 9.80753427210413e-05,
      "loss": 0.0058,
      "step": 2240
    },
    {
      "epoch": 1.2647554806070826,
      "grad_norm": 0.13501233611835498,
      "learning_rate": 9.803664001313142e-05,
      "loss": 0.0057,
      "step": 2250
    },
    {
      "epoch": 1.2703766160764474,
      "grad_norm": 0.09995034213120912,
      "learning_rate": 9.799755982574894e-05,
      "loss": 0.0057,
      "step": 2260
    },
    {
      "epoch": 1.2759977515458123,
      "grad_norm": 0.10016312243592881,
      "learning_rate": 9.79581024659921e-05,
      "loss": 0.0055,
      "step": 2270
    },
    {
      "epoch": 1.281618887015177,
      "grad_norm": 0.12356597886645763,
      "learning_rate": 9.791826824392305e-05,
      "loss": 0.0056,
      "step": 2280
    },
    {
      "epoch": 1.287240022484542,
      "grad_norm": 0.10251488515673644,
      "learning_rate": 9.787805747256536e-05,
      "loss": 0.0055,
      "step": 2290
    },
    {
      "epoch": 1.2928611579539067,
      "grad_norm": 0.10694624192420116,
      "learning_rate": 9.783747046790162e-05,
      "loss": 0.0054,
      "step": 2300
    },
    {
      "epoch": 1.2984822934232714,
      "grad_norm": 0.0837306764941164,
      "learning_rate": 9.779650754887086e-05,
      "loss": 0.0052,
      "step": 2310
    },
    {
      "epoch": 1.3041034288926363,
      "grad_norm": 0.12536680825139915,
      "learning_rate": 9.775516903736616e-05,
      "loss": 0.0053,
      "step": 2320
    },
    {
      "epoch": 1.3097245643620012,
      "grad_norm": 0.09801113540564874,
      "learning_rate": 9.771345525823208e-05,
      "loss": 0.0053,
      "step": 2330
    },
    {
      "epoch": 1.315345699831366,
      "grad_norm": 0.3194597362808477,
      "learning_rate": 9.767136653926199e-05,
      "loss": 0.0055,
      "step": 2340
    },
    {
      "epoch": 1.3209668353007307,
      "grad_norm": 1.8141673380574084,
      "learning_rate": 9.762890321119572e-05,
      "loss": 0.0263,
      "step": 2350
    },
    {
      "epoch": 1.3265879707700956,
      "grad_norm": 2.0373228278291986,
      "learning_rate": 9.758606560771675e-05,
      "loss": 0.0595,
      "step": 2360
    },
    {
      "epoch": 1.3322091062394603,
      "grad_norm": 0.24089249295422568,
      "learning_rate": 9.754285406544969e-05,
      "loss": 0.0223,
      "step": 2370
    },
    {
      "epoch": 1.3378302417088253,
      "grad_norm": 0.14596942738787602,
      "learning_rate": 9.749926892395768e-05,
      "loss": 0.0095,
      "step": 2380
    },
    {
      "epoch": 1.34345137717819,
      "grad_norm": 0.11444722263474487,
      "learning_rate": 9.745531052573961e-05,
      "loss": 0.0072,
      "step": 2390
    },
    {
      "epoch": 1.3490725126475547,
      "grad_norm": 0.0997521525083771,
      "learning_rate": 9.741097921622746e-05,
      "loss": 0.0063,
      "step": 2400
    },
    {
      "epoch": 1.3546936481169196,
      "grad_norm": 0.10164810149706918,
      "learning_rate": 9.736627534378364e-05,
      "loss": 0.0058,
      "step": 2410
    },
    {
      "epoch": 1.3603147835862845,
      "grad_norm": 0.10637858866117042,
      "learning_rate": 9.732119925969823e-05,
      "loss": 0.0055,
      "step": 2420
    },
    {
      "epoch": 1.3659359190556493,
      "grad_norm": 0.1154184870469501,
      "learning_rate": 9.727575131818617e-05,
      "loss": 0.0052,
      "step": 2430
    },
    {
      "epoch": 1.371557054525014,
      "grad_norm": 0.09812174226964852,
      "learning_rate": 9.722993187638454e-05,
      "loss": 0.0051,
      "step": 2440
    },
    {
      "epoch": 1.377178189994379,
      "grad_norm": 0.08717591400124859,
      "learning_rate": 9.718374129434971e-05,
      "loss": 0.0051,
      "step": 2450
    },
    {
      "epoch": 1.3827993254637436,
      "grad_norm": 0.08608962077541132,
      "learning_rate": 9.713717993505456e-05,
      "loss": 0.0049,
      "step": 2460
    },
    {
      "epoch": 1.3884204609331086,
      "grad_norm": 0.10182986875302574,
      "learning_rate": 9.709024816438555e-05,
      "loss": 0.0047,
      "step": 2470
    },
    {
      "epoch": 1.3940415964024733,
      "grad_norm": 0.10115287527791918,
      "learning_rate": 9.704294635113993e-05,
      "loss": 0.0049,
      "step": 2480
    },
    {
      "epoch": 1.399662731871838,
      "grad_norm": 0.0784538159472712,
      "learning_rate": 9.69952748670228e-05,
      "loss": 0.0048,
      "step": 2490
    },
    {
      "epoch": 1.405283867341203,
      "grad_norm": 0.0776387928340622,
      "learning_rate": 9.694723408664414e-05,
      "loss": 0.0044,
      "step": 2500
    },
    {
      "epoch": 1.4109050028105679,
      "grad_norm": 0.0837732573729813,
      "learning_rate": 9.689882438751601e-05,
      "loss": 0.0046,
      "step": 2510
    },
    {
      "epoch": 1.4165261382799326,
      "grad_norm": 0.09489114181755515,
      "learning_rate": 9.68500461500494e-05,
      "loss": 0.0046,
      "step": 2520
    },
    {
      "epoch": 1.4221472737492973,
      "grad_norm": 0.101494695984547,
      "learning_rate": 9.680089975755139e-05,
      "loss": 0.0045,
      "step": 2530
    },
    {
      "epoch": 1.4277684092186622,
      "grad_norm": 0.08821477058203668,
      "learning_rate": 9.675138559622205e-05,
      "loss": 0.0043,
      "step": 2540
    },
    {
      "epoch": 1.433389544688027,
      "grad_norm": 0.07881075966273893,
      "learning_rate": 9.670150405515147e-05,
      "loss": 0.0044,
      "step": 2550
    },
    {
      "epoch": 1.4390106801573919,
      "grad_norm": 0.09899683557846178,
      "learning_rate": 9.665125552631662e-05,
      "loss": 0.0043,
      "step": 2560
    },
    {
      "epoch": 1.4446318156267566,
      "grad_norm": 0.08629556264118085,
      "learning_rate": 9.660064040457837e-05,
      "loss": 0.0044,
      "step": 2570
    },
    {
      "epoch": 1.4502529510961213,
      "grad_norm": 0.09415390850669993,
      "learning_rate": 9.654965908767827e-05,
      "loss": 0.0044,
      "step": 2580
    },
    {
      "epoch": 1.4558740865654862,
      "grad_norm": 0.09596874126004533,
      "learning_rate": 9.649831197623556e-05,
      "loss": 0.0043,
      "step": 2590
    },
    {
      "epoch": 1.4614952220348512,
      "grad_norm": 0.09072868361315219,
      "learning_rate": 9.64465994737439e-05,
      "loss": 0.0041,
      "step": 2600
    },
    {
      "epoch": 1.4671163575042159,
      "grad_norm": 0.08557262009476094,
      "learning_rate": 9.639452198656828e-05,
      "loss": 0.0041,
      "step": 2610
    },
    {
      "epoch": 1.4727374929735806,
      "grad_norm": 0.08310295695036411,
      "learning_rate": 9.634207992394178e-05,
      "loss": 0.0041,
      "step": 2620
    },
    {
      "epoch": 1.4783586284429455,
      "grad_norm": 0.09055939067871042,
      "learning_rate": 9.628927369796236e-05,
      "loss": 0.0041,
      "step": 2630
    },
    {
      "epoch": 1.4839797639123102,
      "grad_norm": 0.09604647791927998,
      "learning_rate": 9.623610372358969e-05,
      "loss": 0.0042,
      "step": 2640
    },
    {
      "epoch": 1.4896008993816752,
      "grad_norm": 0.08577660993704818,
      "learning_rate": 9.618257041864176e-05,
      "loss": 0.0043,
      "step": 2650
    },
    {
      "epoch": 1.4952220348510399,
      "grad_norm": 0.09123279036915018,
      "learning_rate": 9.61286742037917e-05,
      "loss": 0.0041,
      "step": 2660
    },
    {
      "epoch": 1.5008431703204046,
      "grad_norm": 0.08260166738909931,
      "learning_rate": 9.607441550256445e-05,
      "loss": 0.0039,
      "step": 2670
    },
    {
      "epoch": 1.5064643057897695,
      "grad_norm": 0.07923250866368316,
      "learning_rate": 9.601979474133343e-05,
      "loss": 0.004,
      "step": 2680
    },
    {
      "epoch": 1.5120854412591345,
      "grad_norm": 0.08239031912656816,
      "learning_rate": 9.596481234931713e-05,
      "loss": 0.0038,
      "step": 2690
    },
    {
      "epoch": 1.5177065767284992,
      "grad_norm": 0.07610828082112323,
      "learning_rate": 9.59094687585759e-05,
      "loss": 0.0039,
      "step": 2700
    },
    {
      "epoch": 1.523327712197864,
      "grad_norm": 0.11252793520644211,
      "learning_rate": 9.585376440400833e-05,
      "loss": 0.0041,
      "step": 2710
    },
    {
      "epoch": 1.5289488476672288,
      "grad_norm": 0.09330378622902911,
      "learning_rate": 9.579769972334802e-05,
      "loss": 0.004,
      "step": 2720
    },
    {
      "epoch": 1.5345699831365935,
      "grad_norm": 0.07857369281450616,
      "learning_rate": 9.574127515716004e-05,
      "loss": 0.0039,
      "step": 2730
    },
    {
      "epoch": 1.5401911186059585,
      "grad_norm": 0.07210626541279375,
      "learning_rate": 9.56844911488375e-05,
      "loss": 0.0038,
      "step": 2740
    },
    {
      "epoch": 1.5458122540753232,
      "grad_norm": 0.07707927197752867,
      "learning_rate": 9.562734814459811e-05,
      "loss": 0.0038,
      "step": 2750
    },
    {
      "epoch": 1.551433389544688,
      "grad_norm": 0.08248750325983276,
      "learning_rate": 9.556984659348052e-05,
      "loss": 0.0039,
      "step": 2760
    },
    {
      "epoch": 1.5570545250140528,
      "grad_norm": 0.08508167021142606,
      "learning_rate": 9.551198694734102e-05,
      "loss": 0.0039,
      "step": 2770
    },
    {
      "epoch": 1.5626756604834178,
      "grad_norm": 0.08080512980956885,
      "learning_rate": 9.545376966084977e-05,
      "loss": 0.0038,
      "step": 2780
    },
    {
      "epoch": 1.5682967959527825,
      "grad_norm": 0.09346422370365523,
      "learning_rate": 9.539519519148738e-05,
      "loss": 0.0037,
      "step": 2790
    },
    {
      "epoch": 1.5739179314221472,
      "grad_norm": 0.08917779155675297,
      "learning_rate": 9.533626399954126e-05,
      "loss": 0.0036,
      "step": 2800
    },
    {
      "epoch": 1.5795390668915121,
      "grad_norm": 0.06721304489449485,
      "learning_rate": 9.527697654810195e-05,
      "loss": 0.0035,
      "step": 2810
    },
    {
      "epoch": 1.5851602023608768,
      "grad_norm": 0.07427558428573001,
      "learning_rate": 9.52173333030596e-05,
      "loss": 0.0036,
      "step": 2820
    },
    {
      "epoch": 1.5907813378302418,
      "grad_norm": 0.08659228174058138,
      "learning_rate": 9.515733473310019e-05,
      "loss": 0.0036,
      "step": 2830
    },
    {
      "epoch": 1.5964024732996065,
      "grad_norm": 0.0719969223939997,
      "learning_rate": 9.509698130970193e-05,
      "loss": 0.0036,
      "step": 2840
    },
    {
      "epoch": 1.6020236087689712,
      "grad_norm": 0.08181084829875987,
      "learning_rate": 9.50362735071315e-05,
      "loss": 0.0034,
      "step": 2850
    },
    {
      "epoch": 1.6076447442383361,
      "grad_norm": 0.0894284800559242,
      "learning_rate": 9.497521180244038e-05,
      "loss": 0.0035,
      "step": 2860
    },
    {
      "epoch": 1.613265879707701,
      "grad_norm": 0.08282207277574022,
      "learning_rate": 9.491379667546104e-05,
      "loss": 0.0034,
      "step": 2870
    },
    {
      "epoch": 1.6188870151770658,
      "grad_norm": 0.145986818674443,
      "learning_rate": 9.485202860880321e-05,
      "loss": 0.0037,
      "step": 2880
    },
    {
      "epoch": 1.6245081506464305,
      "grad_norm": 0.08546744479627431,
      "learning_rate": 9.478990808785006e-05,
      "loss": 0.0036,
      "step": 2890
    },
    {
      "epoch": 1.6301292861157954,
      "grad_norm": 0.08759751416458819,
      "learning_rate": 9.472743560075446e-05,
      "loss": 0.0034,
      "step": 2900
    },
    {
      "epoch": 1.6357504215851602,
      "grad_norm": 0.07818496950657797,
      "learning_rate": 9.466461163843497e-05,
      "loss": 0.0033,
      "step": 2910
    },
    {
      "epoch": 1.641371557054525,
      "grad_norm": 0.07598460655098531,
      "learning_rate": 9.460143669457221e-05,
      "loss": 0.0034,
      "step": 2920
    },
    {
      "epoch": 1.6469926925238898,
      "grad_norm": 0.07204405112584633,
      "learning_rate": 9.453791126560482e-05,
      "loss": 0.0033,
      "step": 2930
    },
    {
      "epoch": 1.6526138279932545,
      "grad_norm": 0.07548100465060122,
      "learning_rate": 9.44740358507256e-05,
      "loss": 0.0033,
      "step": 2940
    },
    {
      "epoch": 1.6582349634626194,
      "grad_norm": 0.07627624404632878,
      "learning_rate": 9.440981095187764e-05,
      "loss": 0.0034,
      "step": 2950
    },
    {
      "epoch": 1.6638560989319844,
      "grad_norm": 0.07801223336358269,
      "learning_rate": 9.434523707375024e-05,
      "loss": 0.0033,
      "step": 2960
    },
    {
      "epoch": 1.669477234401349,
      "grad_norm": 0.07014240339623043,
      "learning_rate": 9.428031472377516e-05,
      "loss": 0.0031,
      "step": 2970
    },
    {
      "epoch": 1.6750983698707138,
      "grad_norm": 0.0725952595345933,
      "learning_rate": 9.421504441212238e-05,
      "loss": 0.0032,
      "step": 2980
    },
    {
      "epoch": 1.6807195053400787,
      "grad_norm": 0.08432049334212283,
      "learning_rate": 9.414942665169631e-05,
      "loss": 0.0032,
      "step": 2990
    },
    {
      "epoch": 1.6863406408094435,
      "grad_norm": 0.07105741598976323,
      "learning_rate": 9.408346195813159e-05,
      "loss": 0.0034,
      "step": 3000
    },
    {
      "epoch": 1.6919617762788084,
      "grad_norm": 0.13906184784395983,
      "learning_rate": 9.401715084978922e-05,
      "loss": 0.0052,
      "step": 3010
    },
    {
      "epoch": 1.697582911748173,
      "grad_norm": 0.11601157873012202,
      "learning_rate": 9.395049384775225e-05,
      "loss": 0.0042,
      "step": 3020
    },
    {
      "epoch": 1.7032040472175378,
      "grad_norm": 0.09015124532353191,
      "learning_rate": 9.388349147582194e-05,
      "loss": 0.0037,
      "step": 3030
    },
    {
      "epoch": 1.7088251826869028,
      "grad_norm": 0.07000264697695573,
      "learning_rate": 9.381614426051344e-05,
      "loss": 0.0034,
      "step": 3040
    },
    {
      "epoch": 1.7144463181562677,
      "grad_norm": 0.08113336516882994,
      "learning_rate": 9.374845273105177e-05,
      "loss": 0.0033,
      "step": 3050
    },
    {
      "epoch": 1.7200674536256324,
      "grad_norm": 0.08261811708445045,
      "learning_rate": 9.368041741936762e-05,
      "loss": 0.0033,
      "step": 3060
    },
    {
      "epoch": 1.7256885890949971,
      "grad_norm": 0.07838609207688987,
      "learning_rate": 9.361203886009316e-05,
      "loss": 0.0033,
      "step": 3070
    },
    {
      "epoch": 1.731309724564362,
      "grad_norm": 0.08665499355732291,
      "learning_rate": 9.354331759055787e-05,
      "loss": 0.0033,
      "step": 3080
    },
    {
      "epoch": 1.7369308600337268,
      "grad_norm": 0.07918424347748348,
      "learning_rate": 9.347425415078427e-05,
      "loss": 0.0034,
      "step": 3090
    },
    {
      "epoch": 1.7425519955030917,
      "grad_norm": 0.08191128168223484,
      "learning_rate": 9.34048490834838e-05,
      "loss": 0.0031,
      "step": 3100
    },
    {
      "epoch": 1.7481731309724564,
      "grad_norm": 0.0759010615813787,
      "learning_rate": 9.333510293405231e-05,
      "loss": 0.003,
      "step": 3110
    },
    {
      "epoch": 1.7537942664418211,
      "grad_norm": 0.08477221828549904,
      "learning_rate": 9.326501625056606e-05,
      "loss": 0.0031,
      "step": 3120
    },
    {
      "epoch": 1.759415401911186,
      "grad_norm": 0.08622296059752232,
      "learning_rate": 9.319458958377723e-05,
      "loss": 0.0032,
      "step": 3130
    },
    {
      "epoch": 1.765036537380551,
      "grad_norm": 0.07276169675225246,
      "learning_rate": 9.312382348710962e-05,
      "loss": 0.0031,
      "step": 3140
    },
    {
      "epoch": 1.7706576728499157,
      "grad_norm": 0.07298764401064435,
      "learning_rate": 9.305271851665438e-05,
      "loss": 0.0031,
      "step": 3150
    },
    {
      "epoch": 1.7762788083192804,
      "grad_norm": 0.0765870165752984,
      "learning_rate": 9.298127523116555e-05,
      "loss": 0.003,
      "step": 3160
    },
    {
      "epoch": 1.7818999437886454,
      "grad_norm": 0.0963853743897375,
      "learning_rate": 9.290949419205567e-05,
      "loss": 0.0032,
      "step": 3170
    },
    {
      "epoch": 1.78752107925801,
      "grad_norm": 0.08167306430005795,
      "learning_rate": 9.283737596339147e-05,
      "loss": 0.003,
      "step": 3180
    },
    {
      "epoch": 1.793142214727375,
      "grad_norm": 0.07386683946543225,
      "learning_rate": 9.276492111188933e-05,
      "loss": 0.0029,
      "step": 3190
    },
    {
      "epoch": 1.7987633501967397,
      "grad_norm": 0.079496491310148,
      "learning_rate": 9.269213020691083e-05,
      "loss": 0.003,
      "step": 3200
    },
    {
      "epoch": 1.8043844856661044,
      "grad_norm": 0.08006049495470177,
      "learning_rate": 9.261900382045841e-05,
      "loss": 0.003,
      "step": 3210
    },
    {
      "epoch": 1.8100056211354694,
      "grad_norm": 0.06946915908105838,
      "learning_rate": 9.254554252717067e-05,
      "loss": 0.003,
      "step": 3220
    },
    {
      "epoch": 1.8156267566048343,
      "grad_norm": 0.06692045669453084,
      "learning_rate": 9.247174690431803e-05,
      "loss": 0.0028,
      "step": 3230
    },
    {
      "epoch": 1.821247892074199,
      "grad_norm": 0.06938244252654852,
      "learning_rate": 9.23976175317981e-05,
      "loss": 0.0029,
      "step": 3240
    },
    {
      "epoch": 1.8268690275435637,
      "grad_norm": 0.0677124057287239,
      "learning_rate": 9.232315499213114e-05,
      "loss": 0.0028,
      "step": 3250
    },
    {
      "epoch": 1.8324901630129287,
      "grad_norm": 0.0659706764858958,
      "learning_rate": 9.224835987045549e-05,
      "loss": 0.0027,
      "step": 3260
    },
    {
      "epoch": 1.8381112984822934,
      "grad_norm": 0.07797163812538724,
      "learning_rate": 9.217323275452301e-05,
      "loss": 0.0029,
      "step": 3270
    },
    {
      "epoch": 1.8437324339516583,
      "grad_norm": 0.06371412754471345,
      "learning_rate": 9.209777423469436e-05,
      "loss": 0.0027,
      "step": 3280
    },
    {
      "epoch": 1.849353569421023,
      "grad_norm": 0.07195198600776227,
      "learning_rate": 9.202198490393448e-05,
      "loss": 0.0027,
      "step": 3290
    },
    {
      "epoch": 1.8549747048903877,
      "grad_norm": 0.09032730876019279,
      "learning_rate": 9.194586535780785e-05,
      "loss": 0.0027,
      "step": 3300
    },
    {
      "epoch": 1.8605958403597527,
      "grad_norm": 0.07184863572635682,
      "learning_rate": 9.186941619447382e-05,
      "loss": 0.0028,
      "step": 3310
    },
    {
      "epoch": 1.8662169758291176,
      "grad_norm": 0.07972046752123726,
      "learning_rate": 9.179263801468196e-05,
      "loss": 0.0028,
      "step": 3320
    },
    {
      "epoch": 1.8718381112984823,
      "grad_norm": 0.06462786356243252,
      "learning_rate": 9.171553142176725e-05,
      "loss": 0.0026,
      "step": 3330
    },
    {
      "epoch": 1.877459246767847,
      "grad_norm": 0.0748752457223882,
      "learning_rate": 9.163809702164547e-05,
      "loss": 0.0028,
      "step": 3340
    },
    {
      "epoch": 1.883080382237212,
      "grad_norm": 0.06644669684356859,
      "learning_rate": 9.156033542280829e-05,
      "loss": 0.0027,
      "step": 3350
    },
    {
      "epoch": 1.8887015177065767,
      "grad_norm": 0.06481410104589969,
      "learning_rate": 9.148224723631858e-05,
      "loss": 0.0028,
      "step": 3360
    },
    {
      "epoch": 1.8943226531759416,
      "grad_norm": 0.07493396927909493,
      "learning_rate": 9.14038330758056e-05,
      "loss": 0.0028,
      "step": 3370
    },
    {
      "epoch": 1.8999437886453063,
      "grad_norm": 0.07930938338708406,
      "learning_rate": 9.132509355746016e-05,
      "loss": 0.0028,
      "step": 3380
    },
    {
      "epoch": 1.905564924114671,
      "grad_norm": 0.0703227155299206,
      "learning_rate": 9.124602930002977e-05,
      "loss": 0.0027,
      "step": 3390
    },
    {
      "epoch": 1.911186059584036,
      "grad_norm": 0.06415414989508922,
      "learning_rate": 9.11666409248138e-05,
      "loss": 0.0027,
      "step": 3400
    },
    {
      "epoch": 1.916807195053401,
      "grad_norm": 0.05989446878848808,
      "learning_rate": 9.10869290556586e-05,
      "loss": 0.0026,
      "step": 3410
    },
    {
      "epoch": 1.9224283305227656,
      "grad_norm": 0.06596944372327754,
      "learning_rate": 9.100689431895257e-05,
      "loss": 0.0026,
      "step": 3420
    },
    {
      "epoch": 1.9280494659921303,
      "grad_norm": 0.06839374667585654,
      "learning_rate": 9.092653734362126e-05,
      "loss": 0.0025,
      "step": 3430
    },
    {
      "epoch": 1.9336706014614953,
      "grad_norm": 0.05902981335451462,
      "learning_rate": 9.084585876112241e-05,
      "loss": 0.0025,
      "step": 3440
    },
    {
      "epoch": 1.93929173693086,
      "grad_norm": 0.06610787957287352,
      "learning_rate": 9.076485920544105e-05,
      "loss": 0.0027,
      "step": 3450
    },
    {
      "epoch": 1.944912872400225,
      "grad_norm": 0.08257300358198327,
      "learning_rate": 9.068353931308437e-05,
      "loss": 0.0027,
      "step": 3460
    },
    {
      "epoch": 1.9505340078695896,
      "grad_norm": 0.0643606698153111,
      "learning_rate": 9.060189972307691e-05,
      "loss": 0.0025,
      "step": 3470
    },
    {
      "epoch": 1.9561551433389543,
      "grad_norm": 0.06558027003590208,
      "learning_rate": 9.051994107695541e-05,
      "loss": 0.0025,
      "step": 3480
    },
    {
      "epoch": 1.9617762788083193,
      "grad_norm": 0.0752177480017892,
      "learning_rate": 9.043766401876376e-05,
      "loss": 0.0024,
      "step": 3490
    },
    {
      "epoch": 1.9673974142776842,
      "grad_norm": 0.07854251356309212,
      "learning_rate": 9.03550691950481e-05,
      "loss": 0.0026,
      "step": 3500
    },
    {
      "epoch": 1.973018549747049,
      "grad_norm": 0.07298249855242128,
      "learning_rate": 9.027215725485148e-05,
      "loss": 0.0026,
      "step": 3510
    },
    {
      "epoch": 1.9786396852164136,
      "grad_norm": 0.08010026435433024,
      "learning_rate": 9.018892884970899e-05,
      "loss": 0.0026,
      "step": 3520
    },
    {
      "epoch": 1.9842608206857786,
      "grad_norm": 0.06202515462439953,
      "learning_rate": 9.010538463364256e-05,
      "loss": 0.0024,
      "step": 3530
    },
    {
      "epoch": 1.9898819561551433,
      "grad_norm": 0.0638298747620675,
      "learning_rate": 9.002152526315571e-05,
      "loss": 0.0025,
      "step": 3540
    },
    {
      "epoch": 1.9955030916245082,
      "grad_norm": 0.05989999228664472,
      "learning_rate": 8.993735139722862e-05,
      "loss": 0.0024,
      "step": 3550
    },
    {
      "epoch": 2.001124227093873,
      "grad_norm": 0.058109118290538206,
      "learning_rate": 8.985286369731275e-05,
      "loss": 0.0024,
      "step": 3560
    },
    {
      "epoch": 2.0067453625632377,
      "grad_norm": 0.057457616331219645,
      "learning_rate": 8.976806282732572e-05,
      "loss": 0.0019,
      "step": 3570
    },
    {
      "epoch": 2.0123664980326024,
      "grad_norm": 0.05651975998498543,
      "learning_rate": 8.968294945364613e-05,
      "loss": 0.002,
      "step": 3580
    },
    {
      "epoch": 2.0179876335019675,
      "grad_norm": 0.05599584314470216,
      "learning_rate": 8.959752424510824e-05,
      "loss": 0.0021,
      "step": 3590
    },
    {
      "epoch": 2.0236087689713322,
      "grad_norm": 0.06816841901562025,
      "learning_rate": 8.951178787299679e-05,
      "loss": 0.0018,
      "step": 3600
    },
    {
      "epoch": 2.029229904440697,
      "grad_norm": 0.07476474513368925,
      "learning_rate": 8.942574101104164e-05,
      "loss": 0.0019,
      "step": 3610
    },
    {
      "epoch": 2.0348510399100617,
      "grad_norm": 0.058566412874928996,
      "learning_rate": 8.933938433541262e-05,
      "loss": 0.0019,
      "step": 3620
    },
    {
      "epoch": 2.040472175379427,
      "grad_norm": 0.054106525795158976,
      "learning_rate": 8.925271852471403e-05,
      "loss": 0.0019,
      "step": 3630
    },
    {
      "epoch": 2.0460933108487915,
      "grad_norm": 0.0594202177574867,
      "learning_rate": 8.916574425997946e-05,
      "loss": 0.0018,
      "step": 3640
    },
    {
      "epoch": 2.0517144463181562,
      "grad_norm": 0.053841873095984816,
      "learning_rate": 8.907846222466634e-05,
      "loss": 0.0017,
      "step": 3650
    },
    {
      "epoch": 2.057335581787521,
      "grad_norm": 0.052808507853994574,
      "learning_rate": 8.899087310465066e-05,
      "loss": 0.0016,
      "step": 3660
    },
    {
      "epoch": 2.0629567172568857,
      "grad_norm": 0.057001105318726125,
      "learning_rate": 8.890297758822148e-05,
      "loss": 0.0018,
      "step": 3670
    },
    {
      "epoch": 2.068577852726251,
      "grad_norm": 0.057273072509835384,
      "learning_rate": 8.881477636607562e-05,
      "loss": 0.0018,
      "step": 3680
    },
    {
      "epoch": 2.0741989881956155,
      "grad_norm": 0.0546263439951423,
      "learning_rate": 8.872627013131211e-05,
      "loss": 0.0017,
      "step": 3690
    },
    {
      "epoch": 2.0798201236649803,
      "grad_norm": 0.06711361958170248,
      "learning_rate": 8.863745957942691e-05,
      "loss": 0.0018,
      "step": 3700
    },
    {
      "epoch": 2.085441259134345,
      "grad_norm": 0.050819332117840434,
      "learning_rate": 8.854834540830728e-05,
      "loss": 0.0017,
      "step": 3710
    },
    {
      "epoch": 2.09106239460371,
      "grad_norm": 0.06830828572411692,
      "learning_rate": 8.845892831822639e-05,
      "loss": 0.0017,
      "step": 3720
    },
    {
      "epoch": 2.096683530073075,
      "grad_norm": 0.061987972777272506,
      "learning_rate": 8.83692090118378e-05,
      "loss": 0.0017,
      "step": 3730
    },
    {
      "epoch": 2.1023046655424396,
      "grad_norm": 0.0669128652733702,
      "learning_rate": 8.827918819416996e-05,
      "loss": 0.0018,
      "step": 3740
    },
    {
      "epoch": 2.1079258010118043,
      "grad_norm": 0.07873101299778994,
      "learning_rate": 8.818886657262058e-05,
      "loss": 0.0019,
      "step": 3750
    },
    {
      "epoch": 2.1135469364811694,
      "grad_norm": 0.07345024781427896,
      "learning_rate": 8.809824485695118e-05,
      "loss": 0.0017,
      "step": 3760
    },
    {
      "epoch": 2.119168071950534,
      "grad_norm": 0.06739076371941057,
      "learning_rate": 8.800732375928146e-05,
      "loss": 0.0018,
      "step": 3770
    },
    {
      "epoch": 2.124789207419899,
      "grad_norm": 0.07680093288435995,
      "learning_rate": 8.791610399408369e-05,
      "loss": 0.002,
      "step": 3780
    },
    {
      "epoch": 2.1304103428892636,
      "grad_norm": 0.0884538226879485,
      "learning_rate": 8.782458627817716e-05,
      "loss": 0.0019,
      "step": 3790
    },
    {
      "epoch": 2.1360314783586283,
      "grad_norm": 0.06512081179717873,
      "learning_rate": 8.773277133072244e-05,
      "loss": 0.0019,
      "step": 3800
    },
    {
      "epoch": 2.1416526138279934,
      "grad_norm": 0.06562103971194272,
      "learning_rate": 8.764065987321586e-05,
      "loss": 0.002,
      "step": 3810
    },
    {
      "epoch": 2.147273749297358,
      "grad_norm": 0.06091852571490337,
      "learning_rate": 8.754825262948371e-05,
      "loss": 0.002,
      "step": 3820
    },
    {
      "epoch": 2.152894884766723,
      "grad_norm": 0.05792467251440998,
      "learning_rate": 8.745555032567667e-05,
      "loss": 0.0019,
      "step": 3830
    },
    {
      "epoch": 2.1585160202360876,
      "grad_norm": 0.053473661297281636,
      "learning_rate": 8.736255369026403e-05,
      "loss": 0.0017,
      "step": 3840
    },
    {
      "epoch": 2.1641371557054523,
      "grad_norm": 0.08632132178323625,
      "learning_rate": 8.726926345402796e-05,
      "loss": 0.0018,
      "step": 3850
    },
    {
      "epoch": 2.1697582911748174,
      "grad_norm": 0.05824894493699371,
      "learning_rate": 8.71756803500578e-05,
      "loss": 0.0018,
      "step": 3860
    },
    {
      "epoch": 2.175379426644182,
      "grad_norm": 0.05568698678649655,
      "learning_rate": 8.708180511374432e-05,
      "loss": 0.0018,
      "step": 3870
    },
    {
      "epoch": 2.181000562113547,
      "grad_norm": 0.056802968519803224,
      "learning_rate": 8.69876384827739e-05,
      "loss": 0.0017,
      "step": 3880
    },
    {
      "epoch": 2.1866216975829116,
      "grad_norm": 0.06697937497255127,
      "learning_rate": 8.689318119712269e-05,
      "loss": 0.0018,
      "step": 3890
    },
    {
      "epoch": 2.1922428330522767,
      "grad_norm": 0.1352898553987361,
      "learning_rate": 8.679843399905096e-05,
      "loss": 0.002,
      "step": 3900
    },
    {
      "epoch": 2.1978639685216415,
      "grad_norm": 0.06346871111752156,
      "learning_rate": 8.670339763309706e-05,
      "loss": 0.0019,
      "step": 3910
    },
    {
      "epoch": 2.203485103991006,
      "grad_norm": 0.07365846761282421,
      "learning_rate": 8.660807284607171e-05,
      "loss": 0.002,
      "step": 3920
    },
    {
      "epoch": 2.209106239460371,
      "grad_norm": 0.07340874139355225,
      "learning_rate": 8.651246038705206e-05,
      "loss": 0.002,
      "step": 3930
    },
    {
      "epoch": 2.214727374929736,
      "grad_norm": 0.07574517541710105,
      "learning_rate": 8.641656100737588e-05,
      "loss": 0.0021,
      "step": 3940
    },
    {
      "epoch": 2.2203485103991007,
      "grad_norm": 0.06886932484491658,
      "learning_rate": 8.632037546063556e-05,
      "loss": 0.0018,
      "step": 3950
    },
    {
      "epoch": 2.2259696458684655,
      "grad_norm": 0.04966512197542119,
      "learning_rate": 8.622390450267227e-05,
      "loss": 0.0018,
      "step": 3960
    },
    {
      "epoch": 2.23159078133783,
      "grad_norm": 0.06298016565071278,
      "learning_rate": 8.612714889156996e-05,
      "loss": 0.0018,
      "step": 3970
    },
    {
      "epoch": 2.237211916807195,
      "grad_norm": 0.09602521126309173,
      "learning_rate": 8.603010938764944e-05,
      "loss": 0.0017,
      "step": 3980
    },
    {
      "epoch": 2.24283305227656,
      "grad_norm": 0.06179034901494223,
      "learning_rate": 8.593278675346239e-05,
      "loss": 0.0018,
      "step": 3990
    },
    {
      "epoch": 2.2484541877459248,
      "grad_norm": 0.05768935033293568,
      "learning_rate": 8.58351817537854e-05,
      "loss": 0.0017,
      "step": 4000
    },
    {
      "epoch": 2.2540753232152895,
      "grad_norm": 0.05638492675248429,
      "learning_rate": 8.573729515561387e-05,
      "loss": 0.0019,
      "step": 4010
    },
    {
      "epoch": 2.259696458684654,
      "grad_norm": 0.06657528366005455,
      "learning_rate": 8.563912772815612e-05,
      "loss": 0.0018,
      "step": 4020
    },
    {
      "epoch": 2.265317594154019,
      "grad_norm": 0.06296651541934814,
      "learning_rate": 8.554068024282722e-05,
      "loss": 0.0018,
      "step": 4030
    },
    {
      "epoch": 2.270938729623384,
      "grad_norm": 0.06738339509707009,
      "learning_rate": 8.5441953473243e-05,
      "loss": 0.0017,
      "step": 4040
    },
    {
      "epoch": 2.2765598650927488,
      "grad_norm": 0.07704835455024767,
      "learning_rate": 8.534294819521394e-05,
      "loss": 0.0016,
      "step": 4050
    },
    {
      "epoch": 2.2821810005621135,
      "grad_norm": 0.0567808801938328,
      "learning_rate": 8.524366518673907e-05,
      "loss": 0.0017,
      "step": 4060
    },
    {
      "epoch": 2.287802136031478,
      "grad_norm": 0.06033970611341973,
      "learning_rate": 8.514410522799992e-05,
      "loss": 0.0017,
      "step": 4070
    },
    {
      "epoch": 2.2934232715008434,
      "grad_norm": 0.06295828590005369,
      "learning_rate": 8.504426910135429e-05,
      "loss": 0.0016,
      "step": 4080
    },
    {
      "epoch": 2.299044406970208,
      "grad_norm": 0.06724936488740965,
      "learning_rate": 8.494415759133014e-05,
      "loss": 0.0017,
      "step": 4090
    },
    {
      "epoch": 2.304665542439573,
      "grad_norm": 0.06118776673887835,
      "learning_rate": 8.48437714846195e-05,
      "loss": 0.0016,
      "step": 4100
    },
    {
      "epoch": 2.3102866779089375,
      "grad_norm": 0.09003973689457073,
      "learning_rate": 8.474311157007218e-05,
      "loss": 0.0018,
      "step": 4110
    },
    {
      "epoch": 2.3159078133783026,
      "grad_norm": 0.06239353412217714,
      "learning_rate": 8.46421786386896e-05,
      "loss": 0.0018,
      "step": 4120
    },
    {
      "epoch": 2.3215289488476674,
      "grad_norm": 0.05655954006890423,
      "learning_rate": 8.45409734836186e-05,
      "loss": 0.0016,
      "step": 4130
    },
    {
      "epoch": 2.327150084317032,
      "grad_norm": 0.0521687512636754,
      "learning_rate": 8.443949690014527e-05,
      "loss": 0.0016,
      "step": 4140
    },
    {
      "epoch": 2.332771219786397,
      "grad_norm": 0.05228400769684489,
      "learning_rate": 8.433774968568849e-05,
      "loss": 0.0016,
      "step": 4150
    },
    {
      "epoch": 2.3383923552557615,
      "grad_norm": 0.054685428841644984,
      "learning_rate": 8.423573263979392e-05,
      "loss": 0.0016,
      "step": 4160
    },
    {
      "epoch": 2.3440134907251267,
      "grad_norm": 0.05279358062852916,
      "learning_rate": 8.413344656412754e-05,
      "loss": 0.0017,
      "step": 4170
    },
    {
      "epoch": 2.3496346261944914,
      "grad_norm": 0.0619594130610791,
      "learning_rate": 8.40308922624694e-05,
      "loss": 0.0015,
      "step": 4180
    },
    {
      "epoch": 2.355255761663856,
      "grad_norm": 0.05549436621885542,
      "learning_rate": 8.392807054070738e-05,
      "loss": 0.0015,
      "step": 4190
    },
    {
      "epoch": 2.360876897133221,
      "grad_norm": 0.05844615472881089,
      "learning_rate": 8.382498220683068e-05,
      "loss": 0.0017,
      "step": 4200
    },
    {
      "epoch": 2.3664980326025855,
      "grad_norm": 0.05785480991010937,
      "learning_rate": 8.37216280709237e-05,
      "loss": 0.0016,
      "step": 4210
    },
    {
      "epoch": 2.3721191680719507,
      "grad_norm": 0.07481921771572901,
      "learning_rate": 8.361800894515946e-05,
      "loss": 0.0016,
      "step": 4220
    },
    {
      "epoch": 2.3777403035413154,
      "grad_norm": 0.05822829929196074,
      "learning_rate": 8.351412564379336e-05,
      "loss": 0.0016,
      "step": 4230
    },
    {
      "epoch": 2.38336143901068,
      "grad_norm": 0.05597231845386826,
      "learning_rate": 8.340997898315671e-05,
      "loss": 0.0017,
      "step": 4240
    },
    {
      "epoch": 2.388982574480045,
      "grad_norm": 0.05518800111012908,
      "learning_rate": 8.330556978165036e-05,
      "loss": 0.0017,
      "step": 4250
    },
    {
      "epoch": 2.39460370994941,
      "grad_norm": 0.06684935678037553,
      "learning_rate": 8.320089885973826e-05,
      "loss": 0.0018,
      "step": 4260
    },
    {
      "epoch": 2.4002248454187747,
      "grad_norm": 0.072573592930173,
      "learning_rate": 8.309596703994094e-05,
      "loss": 0.0017,
      "step": 4270
    },
    {
      "epoch": 2.4058459808881394,
      "grad_norm": 0.06709976923029225,
      "learning_rate": 8.299077514682915e-05,
      "loss": 0.0017,
      "step": 4280
    },
    {
      "epoch": 2.411467116357504,
      "grad_norm": 0.06006598574157663,
      "learning_rate": 8.288532400701735e-05,
      "loss": 0.0018,
      "step": 4290
    },
    {
      "epoch": 2.4170882518268693,
      "grad_norm": 0.06053465610650408,
      "learning_rate": 8.277961444915716e-05,
      "loss": 0.0017,
      "step": 4300
    },
    {
      "epoch": 2.422709387296234,
      "grad_norm": 0.0633679769120732,
      "learning_rate": 8.267364730393093e-05,
      "loss": 0.0016,
      "step": 4310
    },
    {
      "epoch": 2.4283305227655987,
      "grad_norm": 0.04905130321561388,
      "learning_rate": 8.256742340404515e-05,
      "loss": 0.0016,
      "step": 4320
    },
    {
      "epoch": 2.4339516582349634,
      "grad_norm": 0.05084581543250133,
      "learning_rate": 8.246094358422393e-05,
      "loss": 0.0015,
      "step": 4330
    },
    {
      "epoch": 2.439572793704328,
      "grad_norm": 0.06929941783453905,
      "learning_rate": 8.235420868120245e-05,
      "loss": 0.0015,
      "step": 4340
    },
    {
      "epoch": 2.4451939291736933,
      "grad_norm": 0.0495981785149641,
      "learning_rate": 8.224721953372035e-05,
      "loss": 0.0015,
      "step": 4350
    },
    {
      "epoch": 2.450815064643058,
      "grad_norm": 0.06154487277663696,
      "learning_rate": 8.213997698251519e-05,
      "loss": 0.0015,
      "step": 4360
    },
    {
      "epoch": 2.4564362001124227,
      "grad_norm": 0.06702227730246357,
      "learning_rate": 8.203248187031577e-05,
      "loss": 0.0016,
      "step": 4370
    },
    {
      "epoch": 2.4620573355817874,
      "grad_norm": 0.056017350947715615,
      "learning_rate": 8.192473504183566e-05,
      "loss": 0.0015,
      "step": 4380
    },
    {
      "epoch": 2.467678471051152,
      "grad_norm": 0.11198849203086247,
      "learning_rate": 8.181673734376634e-05,
      "loss": 0.0016,
      "step": 4390
    },
    {
      "epoch": 2.4732996065205173,
      "grad_norm": 0.06329875901984851,
      "learning_rate": 8.170848962477068e-05,
      "loss": 0.0018,
      "step": 4400
    },
    {
      "epoch": 2.478920741989882,
      "grad_norm": 0.06331119073409751,
      "learning_rate": 8.159999273547636e-05,
      "loss": 0.0016,
      "step": 4410
    },
    {
      "epoch": 2.4845418774592467,
      "grad_norm": 0.052348939039535504,
      "learning_rate": 8.149124752846894e-05,
      "loss": 0.0016,
      "step": 4420
    },
    {
      "epoch": 2.4901630129286114,
      "grad_norm": 0.054010497916272666,
      "learning_rate": 8.138225485828537e-05,
      "loss": 0.0015,
      "step": 4430
    },
    {
      "epoch": 2.4957841483979766,
      "grad_norm": 0.0654068401724483,
      "learning_rate": 8.12730155814072e-05,
      "loss": 0.0016,
      "step": 4440
    },
    {
      "epoch": 2.5014052838673413,
      "grad_norm": 0.06774434911902472,
      "learning_rate": 8.116353055625385e-05,
      "loss": 0.0016,
      "step": 4450
    },
    {
      "epoch": 2.507026419336706,
      "grad_norm": 0.058645050551856606,
      "learning_rate": 8.105380064317587e-05,
      "loss": 0.0015,
      "step": 4460
    },
    {
      "epoch": 2.5126475548060707,
      "grad_norm": 0.06378242762268096,
      "learning_rate": 8.094382670444817e-05,
      "loss": 0.0014,
      "step": 4470
    },
    {
      "epoch": 2.518268690275436,
      "grad_norm": 0.07035140206345245,
      "learning_rate": 8.083360960426327e-05,
      "loss": 0.0015,
      "step": 4480
    },
    {
      "epoch": 2.5238898257448006,
      "grad_norm": 0.054138269993327354,
      "learning_rate": 8.072315020872448e-05,
      "loss": 0.0014,
      "step": 4490
    },
    {
      "epoch": 2.5295109612141653,
      "grad_norm": 0.0520380194907875,
      "learning_rate": 8.061244938583907e-05,
      "loss": 0.0015,
      "step": 4500
    },
    {
      "epoch": 2.53513209668353,
      "grad_norm": 0.0652285964745877,
      "learning_rate": 8.050150800551155e-05,
      "loss": 0.0014,
      "step": 4510
    },
    {
      "epoch": 2.5407532321528947,
      "grad_norm": 0.06146797888465651,
      "learning_rate": 8.039032693953672e-05,
      "loss": 0.0015,
      "step": 4520
    },
    {
      "epoch": 2.54637436762226,
      "grad_norm": 0.04413018291772843,
      "learning_rate": 8.02789070615929e-05,
      "loss": 0.0015,
      "step": 4530
    },
    {
      "epoch": 2.5519955030916246,
      "grad_norm": 0.04939992283587218,
      "learning_rate": 8.016724924723501e-05,
      "loss": 0.0015,
      "step": 4540
    },
    {
      "epoch": 2.5576166385609893,
      "grad_norm": 0.043663817829164675,
      "learning_rate": 8.00553543738877e-05,
      "loss": 0.0016,
      "step": 4550
    },
    {
      "epoch": 2.563237774030354,
      "grad_norm": 0.054145213065889916,
      "learning_rate": 7.994322332083852e-05,
      "loss": 0.0016,
      "step": 4560
    },
    {
      "epoch": 2.5688589094997187,
      "grad_norm": 0.05699514555987272,
      "learning_rate": 7.983085696923087e-05,
      "loss": 0.0019,
      "step": 4570
    },
    {
      "epoch": 2.574480044969084,
      "grad_norm": 0.0640274235075573,
      "learning_rate": 7.971825620205726e-05,
      "loss": 0.0018,
      "step": 4580
    },
    {
      "epoch": 2.5801011804384486,
      "grad_norm": 0.05404722214829411,
      "learning_rate": 7.96054219041522e-05,
      "loss": 0.0016,
      "step": 4590
    },
    {
      "epoch": 2.5857223159078133,
      "grad_norm": 0.060989090707734825,
      "learning_rate": 7.94923549621854e-05,
      "loss": 0.0015,
      "step": 4600
    },
    {
      "epoch": 2.591343451377178,
      "grad_norm": 0.05089975294543041,
      "learning_rate": 7.937905626465458e-05,
      "loss": 0.0015,
      "step": 4610
    },
    {
      "epoch": 2.5969645868465427,
      "grad_norm": 0.047753743345937585,
      "learning_rate": 7.926552670187881e-05,
      "loss": 0.0014,
      "step": 4620
    },
    {
      "epoch": 2.602585722315908,
      "grad_norm": 0.04387585711392661,
      "learning_rate": 7.915176716599122e-05,
      "loss": 0.0015,
      "step": 4630
    },
    {
      "epoch": 2.6082068577852726,
      "grad_norm": 0.05324929700338459,
      "learning_rate": 7.903777855093213e-05,
      "loss": 0.0014,
      "step": 4640
    },
    {
      "epoch": 2.6138279932546373,
      "grad_norm": 0.04210024748842374,
      "learning_rate": 7.892356175244202e-05,
      "loss": 0.0014,
      "step": 4650
    },
    {
      "epoch": 2.6194491287240025,
      "grad_norm": 0.09552386039463091,
      "learning_rate": 7.880911766805444e-05,
      "loss": 0.0014,
      "step": 4660
    },
    {
      "epoch": 2.625070264193367,
      "grad_norm": 0.059038210930734324,
      "learning_rate": 7.869444719708903e-05,
      "loss": 0.0013,
      "step": 4670
    },
    {
      "epoch": 2.630691399662732,
      "grad_norm": 0.04816659963124649,
      "learning_rate": 7.857955124064435e-05,
      "loss": 0.0012,
      "step": 4680
    },
    {
      "epoch": 2.6363125351320966,
      "grad_norm": 0.05442452499531946,
      "learning_rate": 7.846443070159096e-05,
      "loss": 0.0014,
      "step": 4690
    },
    {
      "epoch": 2.6419336706014613,
      "grad_norm": 0.05419618414147171,
      "learning_rate": 7.834908648456408e-05,
      "loss": 0.0013,
      "step": 4700
    },
    {
      "epoch": 2.6475548060708265,
      "grad_norm": 0.05046575142656625,
      "learning_rate": 7.823351949595676e-05,
      "loss": 0.0013,
      "step": 4710
    },
    {
      "epoch": 2.653175941540191,
      "grad_norm": 0.046485397857171505,
      "learning_rate": 7.811773064391257e-05,
      "loss": 0.0013,
      "step": 4720
    },
    {
      "epoch": 2.658797077009556,
      "grad_norm": 0.043494442406102805,
      "learning_rate": 7.800172083831849e-05,
      "loss": 0.0012,
      "step": 4730
    },
    {
      "epoch": 2.6644182124789206,
      "grad_norm": 0.05197640361791803,
      "learning_rate": 7.788549099079787e-05,
      "loss": 0.0013,
      "step": 4740
    },
    {
      "epoch": 2.6700393479482853,
      "grad_norm": 0.050588815857194186,
      "learning_rate": 7.77690420147031e-05,
      "loss": 0.0013,
      "step": 4750
    },
    {
      "epoch": 2.6756604834176505,
      "grad_norm": 0.055277287421162696,
      "learning_rate": 7.765237482510857e-05,
      "loss": 0.0012,
      "step": 4760
    },
    {
      "epoch": 2.681281618887015,
      "grad_norm": 0.047387945885664126,
      "learning_rate": 7.753549033880336e-05,
      "loss": 0.0012,
      "step": 4770
    },
    {
      "epoch": 2.68690275435638,
      "grad_norm": 0.055314706331344304,
      "learning_rate": 7.74183894742842e-05,
      "loss": 0.0013,
      "step": 4780
    },
    {
      "epoch": 2.6925238898257446,
      "grad_norm": 0.054991171970991085,
      "learning_rate": 7.730107315174808e-05,
      "loss": 0.0015,
      "step": 4790
    },
    {
      "epoch": 2.6981450252951094,
      "grad_norm": 0.07142443615869051,
      "learning_rate": 7.718354229308511e-05,
      "loss": 0.0014,
      "step": 4800
    },
    {
      "epoch": 2.7037661607644745,
      "grad_norm": 0.06880613464299018,
      "learning_rate": 7.706579782187127e-05,
      "loss": 0.0014,
      "step": 4810
    },
    {
      "epoch": 2.7093872962338392,
      "grad_norm": 0.10332828544462125,
      "learning_rate": 7.694784066336114e-05,
      "loss": 0.0015,
      "step": 4820
    },
    {
      "epoch": 2.715008431703204,
      "grad_norm": 0.06452511979597532,
      "learning_rate": 7.68296717444806e-05,
      "loss": 0.0015,
      "step": 4830
    },
    {
      "epoch": 2.720629567172569,
      "grad_norm": 0.05595821471841563,
      "learning_rate": 7.671129199381963e-05,
      "loss": 0.0015,
      "step": 4840
    },
    {
      "epoch": 2.726250702641934,
      "grad_norm": 0.04384518983020577,
      "learning_rate": 7.659270234162491e-05,
      "loss": 0.0015,
      "step": 4850
    },
    {
      "epoch": 2.7318718381112985,
      "grad_norm": 0.04831975704039876,
      "learning_rate": 7.647390371979257e-05,
      "loss": 0.0014,
      "step": 4860
    },
    {
      "epoch": 2.7374929735806632,
      "grad_norm": 0.059347623006389266,
      "learning_rate": 7.635489706186092e-05,
      "loss": 0.0012,
      "step": 4870
    },
    {
      "epoch": 2.743114109050028,
      "grad_norm": 0.05694220115635119,
      "learning_rate": 7.623568330300294e-05,
      "loss": 0.0013,
      "step": 4880
    },
    {
      "epoch": 2.748735244519393,
      "grad_norm": 0.0468828291912844,
      "learning_rate": 7.611626338001912e-05,
      "loss": 0.0013,
      "step": 4890
    },
    {
      "epoch": 2.754356379988758,
      "grad_norm": 0.04787035904833046,
      "learning_rate": 7.599663823132998e-05,
      "loss": 0.0014,
      "step": 4900
    },
    {
      "epoch": 2.7599775154581225,
      "grad_norm": 0.05657663969108575,
      "learning_rate": 7.587680879696876e-05,
      "loss": 0.0013,
      "step": 4910
    },
    {
      "epoch": 2.7655986509274872,
      "grad_norm": 0.04466607117951967,
      "learning_rate": 7.575677601857402e-05,
      "loss": 0.0013,
      "step": 4920
    },
    {
      "epoch": 2.771219786396852,
      "grad_norm": 0.06027260612305898,
      "learning_rate": 7.563654083938221e-05,
      "loss": 0.0013,
      "step": 4930
    },
    {
      "epoch": 2.776840921866217,
      "grad_norm": 0.05081088548969293,
      "learning_rate": 7.551610420422026e-05,
      "loss": 0.0013,
      "step": 4940
    },
    {
      "epoch": 2.782462057335582,
      "grad_norm": 0.05428661275304198,
      "learning_rate": 7.53954670594982e-05,
      "loss": 0.0012,
      "step": 4950
    },
    {
      "epoch": 2.7880831928049465,
      "grad_norm": 0.0482700567695289,
      "learning_rate": 7.52746303532017e-05,
      "loss": 0.0012,
      "step": 4960
    },
    {
      "epoch": 2.7937043282743113,
      "grad_norm": 0.04707913457424893,
      "learning_rate": 7.515359503488457e-05,
      "loss": 0.0012,
      "step": 4970
    },
    {
      "epoch": 2.799325463743676,
      "grad_norm": 0.06312636376317708,
      "learning_rate": 7.503236205566142e-05,
      "loss": 0.0011,
      "step": 4980
    },
    {
      "epoch": 2.804946599213041,
      "grad_norm": 0.039101096639687,
      "learning_rate": 7.491093236820003e-05,
      "loss": 0.0011,
      "step": 4990
    },
    {
      "epoch": 2.810567734682406,
      "grad_norm": 0.04203545515495065,
      "learning_rate": 7.478930692671403e-05,
      "loss": 0.0013,
      "step": 5000
    },
    {
      "epoch": 2.8161888701517706,
      "grad_norm": 0.05994425738017975,
      "learning_rate": 7.466748668695522e-05,
      "loss": 0.0012,
      "step": 5010
    },
    {
      "epoch": 2.8218100056211357,
      "grad_norm": 0.053168584599147396,
      "learning_rate": 7.454547260620623e-05,
      "loss": 0.0013,
      "step": 5020
    },
    {
      "epoch": 2.8274311410905004,
      "grad_norm": 0.050877693867899004,
      "learning_rate": 7.442326564327291e-05,
      "loss": 0.0013,
      "step": 5030
    },
    {
      "epoch": 2.833052276559865,
      "grad_norm": 0.06380365444654545,
      "learning_rate": 7.430086675847676e-05,
      "loss": 0.0012,
      "step": 5040
    },
    {
      "epoch": 2.83867341202923,
      "grad_norm": 0.04679933950214477,
      "learning_rate": 7.417827691364751e-05,
      "loss": 0.0011,
      "step": 5050
    },
    {
      "epoch": 2.8442945474985946,
      "grad_norm": 0.04377109248658636,
      "learning_rate": 7.40554970721154e-05,
      "loss": 0.0012,
      "step": 5060
    },
    {
      "epoch": 2.8499156829679597,
      "grad_norm": 0.0608311362918617,
      "learning_rate": 7.393252819870375e-05,
      "loss": 0.0014,
      "step": 5070
    },
    {
      "epoch": 2.8555368184373244,
      "grad_norm": 0.0943118819372207,
      "learning_rate": 7.380937125972133e-05,
      "loss": 0.0014,
      "step": 5080
    },
    {
      "epoch": 2.861157953906689,
      "grad_norm": 0.0403091749249693,
      "learning_rate": 7.368602722295471e-05,
      "loss": 0.0013,
      "step": 5090
    },
    {
      "epoch": 2.866779089376054,
      "grad_norm": 0.055971258310376075,
      "learning_rate": 7.356249705766075e-05,
      "loss": 0.0013,
      "step": 5100
    },
    {
      "epoch": 2.8724002248454186,
      "grad_norm": 0.08519796434852828,
      "learning_rate": 7.34387817345589e-05,
      "loss": 0.0013,
      "step": 5110
    },
    {
      "epoch": 2.8780213603147837,
      "grad_norm": 0.06853625802370912,
      "learning_rate": 7.331488222582364e-05,
      "loss": 0.0012,
      "step": 5120
    },
    {
      "epoch": 2.8836424957841484,
      "grad_norm": 0.04218234679052211,
      "learning_rate": 7.319079950507682e-05,
      "loss": 0.0013,
      "step": 5130
    },
    {
      "epoch": 2.889263631253513,
      "grad_norm": 0.03601990309692009,
      "learning_rate": 7.306653454737992e-05,
      "loss": 0.0012,
      "step": 5140
    },
    {
      "epoch": 2.894884766722878,
      "grad_norm": 0.05970290400064692,
      "learning_rate": 7.294208832922655e-05,
      "loss": 0.0011,
      "step": 5150
    },
    {
      "epoch": 2.9005059021922426,
      "grad_norm": 0.06211859449356491,
      "learning_rate": 7.281746182853469e-05,
      "loss": 0.0017,
      "step": 5160
    },
    {
      "epoch": 2.9061270376616077,
      "grad_norm": 0.04244606481235483,
      "learning_rate": 7.269265602463895e-05,
      "loss": 0.0013,
      "step": 5170
    },
    {
      "epoch": 2.9117481731309725,
      "grad_norm": 0.04796952285255335,
      "learning_rate": 7.256767189828297e-05,
      "loss": 0.0012,
      "step": 5180
    },
    {
      "epoch": 2.917369308600337,
      "grad_norm": 0.04891082555346954,
      "learning_rate": 7.24425104316117e-05,
      "loss": 0.0012,
      "step": 5190
    },
    {
      "epoch": 2.9229904440697023,
      "grad_norm": 0.04272228076487222,
      "learning_rate": 7.231717260816361e-05,
      "loss": 0.0011,
      "step": 5200
    },
    {
      "epoch": 2.928611579539067,
      "grad_norm": 0.04454348084467995,
      "learning_rate": 7.219165941286303e-05,
      "loss": 0.0012,
      "step": 5210
    },
    {
      "epoch": 2.9342327150084317,
      "grad_norm": 0.05868816541252896,
      "learning_rate": 7.20659718320124e-05,
      "loss": 0.0012,
      "step": 5220
    },
    {
      "epoch": 2.9398538504777965,
      "grad_norm": 0.052032777228837916,
      "learning_rate": 7.194011085328448e-05,
      "loss": 0.0013,
      "step": 5230
    },
    {
      "epoch": 2.945474985947161,
      "grad_norm": 0.049334668563237004,
      "learning_rate": 7.181407746571466e-05,
      "loss": 0.0011,
      "step": 5240
    },
    {
      "epoch": 2.9510961214165263,
      "grad_norm": 0.05111824349302509,
      "learning_rate": 7.16878726596931e-05,
      "loss": 0.0012,
      "step": 5250
    },
    {
      "epoch": 2.956717256885891,
      "grad_norm": 0.047943356040951235,
      "learning_rate": 7.156149742695703e-05,
      "loss": 0.0012,
      "step": 5260
    },
    {
      "epoch": 2.9623383923552558,
      "grad_norm": 0.04666570165091871,
      "learning_rate": 7.143495276058292e-05,
      "loss": 0.0013,
      "step": 5270
    },
    {
      "epoch": 2.9679595278246205,
      "grad_norm": 0.04633872778424716,
      "learning_rate": 7.130823965497863e-05,
      "loss": 0.0012,
      "step": 5280
    },
    {
      "epoch": 2.973580663293985,
      "grad_norm": 0.04728600175386692,
      "learning_rate": 7.11813591058757e-05,
      "loss": 0.001,
      "step": 5290
    },
    {
      "epoch": 2.9792017987633503,
      "grad_norm": 0.04331703035403907,
      "learning_rate": 7.105431211032142e-05,
      "loss": 0.001,
      "step": 5300
    },
    {
      "epoch": 2.984822934232715,
      "grad_norm": 0.0411566396160935,
      "learning_rate": 7.092709966667107e-05,
      "loss": 0.001,
      "step": 5310
    },
    {
      "epoch": 2.9904440697020798,
      "grad_norm": 0.045270747800133036,
      "learning_rate": 7.079972277458007e-05,
      "loss": 0.0011,
      "step": 5320
    },
    {
      "epoch": 2.9960652051714445,
      "grad_norm": 0.03583276636219469,
      "learning_rate": 7.0672182434996e-05,
      "loss": 0.0009,
      "step": 5330
    },
    {
      "epoch": 3.0016863406408096,
      "grad_norm": 0.049772188634408326,
      "learning_rate": 7.054447965015099e-05,
      "loss": 0.001,
      "step": 5340
    },
    {
      "epoch": 3.0073074761101743,
      "grad_norm": 0.04417115124033103,
      "learning_rate": 7.041661542355356e-05,
      "loss": 0.0008,
      "step": 5350
    },
    {
      "epoch": 3.012928611579539,
      "grad_norm": 0.04016838015205063,
      "learning_rate": 7.028859075998093e-05,
      "loss": 0.0007,
      "step": 5360
    },
    {
      "epoch": 3.0185497470489038,
      "grad_norm": 0.03429961640956788,
      "learning_rate": 7.016040666547105e-05,
      "loss": 0.0008,
      "step": 5370
    },
    {
      "epoch": 3.0241708825182685,
      "grad_norm": 0.03891109220250057,
      "learning_rate": 7.003206414731467e-05,
      "loss": 0.0007,
      "step": 5380
    },
    {
      "epoch": 3.0297920179876336,
      "grad_norm": 0.04617952784664274,
      "learning_rate": 6.990356421404751e-05,
      "loss": 0.0008,
      "step": 5390
    },
    {
      "epoch": 3.0354131534569984,
      "grad_norm": 0.04730962147871257,
      "learning_rate": 6.977490787544224e-05,
      "loss": 0.0008,
      "step": 5400
    }
  ],
  "logging_steps": 10,
  "max_steps": 12453,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 7,
  "save_steps": 1800,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 440429050331136.0,
  "train_batch_size": 128,
  "trial_name": null,
  "trial_params": null
}