{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9997706947947719,
  "eval_steps": 500,
  "global_step": 2180,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.00045861041045631735,
      "grad_norm": 99.26005382206641,
      "learning_rate": 0.0,
      "loss": 25.2517,
      "step": 1
    },
    {
      "epoch": 0.0009172208209126347,
      "grad_norm": 107.11130420998052,
      "learning_rate": 1.5151515151515153e-05,
      "loss": 25.5128,
      "step": 2
    },
    {
      "epoch": 0.001375831231368952,
      "grad_norm": 81.47925165442301,
      "learning_rate": 3.0303030303030306e-05,
      "loss": 24.8029,
      "step": 3
    },
    {
      "epoch": 0.0018344416418252694,
      "grad_norm": 62.33276634502426,
      "learning_rate": 4.545454545454546e-05,
      "loss": 23.0281,
      "step": 4
    },
    {
      "epoch": 0.002293052052281587,
      "grad_norm": 44.91719596046084,
      "learning_rate": 6.060606060606061e-05,
      "loss": 21.6801,
      "step": 5
    },
    {
      "epoch": 0.002751662462737904,
      "grad_norm": 32.221351318721,
      "learning_rate": 7.575757575757576e-05,
      "loss": 20.6189,
      "step": 6
    },
    {
      "epoch": 0.0032102728731942215,
      "grad_norm": 24.514265004467482,
      "learning_rate": 9.090909090909092e-05,
      "loss": 19.9338,
      "step": 7
    },
    {
      "epoch": 0.0036688832836505388,
      "grad_norm": 20.856872031241448,
      "learning_rate": 0.00010606060606060606,
      "loss": 19.2699,
      "step": 8
    },
    {
      "epoch": 0.0041274936941068565,
      "grad_norm": 16.918708512051364,
      "learning_rate": 0.00012121212121212122,
      "loss": 18.5825,
      "step": 9
    },
    {
      "epoch": 0.004586104104563174,
      "grad_norm": 13.991535650817168,
      "learning_rate": 0.00013636363636363637,
      "loss": 18.0148,
      "step": 10
    },
    {
      "epoch": 0.005044714515019491,
      "grad_norm": 10.27022861508554,
      "learning_rate": 0.00015151515151515152,
      "loss": 17.8324,
      "step": 11
    },
    {
      "epoch": 0.005503324925475808,
      "grad_norm": 9.482196959981854,
      "learning_rate": 0.00016666666666666666,
      "loss": 17.8241,
      "step": 12
    },
    {
      "epoch": 0.005961935335932126,
      "grad_norm": 5.854164281066282,
      "learning_rate": 0.00018181818181818183,
      "loss": 17.4011,
      "step": 13
    },
    {
      "epoch": 0.006420545746388443,
      "grad_norm": 5.066912103015744,
      "learning_rate": 0.00019696969696969695,
      "loss": 17.2108,
      "step": 14
    },
    {
      "epoch": 0.00687915615684476,
      "grad_norm": 5.65380880459776,
      "learning_rate": 0.00021212121212121213,
      "loss": 17.1605,
      "step": 15
    },
    {
      "epoch": 0.0073377665673010775,
      "grad_norm": 5.205589311558462,
      "learning_rate": 0.00022727272727272727,
      "loss": 17.0027,
      "step": 16
    },
    {
      "epoch": 0.007796376977757395,
      "grad_norm": 3.75123944363194,
      "learning_rate": 0.00024242424242424245,
      "loss": 16.7943,
      "step": 17
    },
    {
      "epoch": 0.008254987388213713,
      "grad_norm": 3.310166334263357,
      "learning_rate": 0.00025757575757575756,
      "loss": 16.7913,
      "step": 18
    },
    {
      "epoch": 0.00871359779867003,
      "grad_norm": 2.7561164495438533,
      "learning_rate": 0.00027272727272727274,
      "loss": 16.5953,
      "step": 19
    },
    {
      "epoch": 0.009172208209126348,
      "grad_norm": 2.7526690754535084,
      "learning_rate": 0.0002878787878787879,
      "loss": 16.7151,
      "step": 20
    },
    {
      "epoch": 0.009630818619582664,
      "grad_norm": 2.071800583783916,
      "learning_rate": 0.00030303030303030303,
      "loss": 16.5098,
      "step": 21
    },
    {
      "epoch": 0.010089429030038982,
      "grad_norm": 2.0540679488914315,
      "learning_rate": 0.0003181818181818182,
      "loss": 16.2118,
      "step": 22
    },
    {
      "epoch": 0.010548039440495299,
      "grad_norm": 2.2442923553473872,
      "learning_rate": 0.0003333333333333333,
      "loss": 16.3651,
      "step": 23
    },
    {
      "epoch": 0.011006649850951617,
      "grad_norm": 2.4747607855477485,
      "learning_rate": 0.0003484848484848485,
      "loss": 16.4972,
      "step": 24
    },
    {
      "epoch": 0.011465260261407933,
      "grad_norm": 2.804595491499318,
      "learning_rate": 0.00036363636363636367,
      "loss": 16.1968,
      "step": 25
    },
    {
      "epoch": 0.011923870671864251,
      "grad_norm": 1.6033283814310728,
      "learning_rate": 0.0003787878787878788,
      "loss": 16.128,
      "step": 26
    },
    {
      "epoch": 0.01238248108232057,
      "grad_norm": 1.91427928611653,
      "learning_rate": 0.0003939393939393939,
      "loss": 16.1679,
      "step": 27
    },
    {
      "epoch": 0.012841091492776886,
      "grad_norm": 1.6643709537121716,
      "learning_rate": 0.00040909090909090913,
      "loss": 16.1723,
      "step": 28
    },
    {
      "epoch": 0.013299701903233204,
      "grad_norm": 1.2508113020104807,
      "learning_rate": 0.00042424242424242425,
      "loss": 16.2068,
      "step": 29
    },
    {
      "epoch": 0.01375831231368952,
      "grad_norm": 1.2998735755675677,
      "learning_rate": 0.0004393939393939394,
      "loss": 16.0213,
      "step": 30
    },
    {
      "epoch": 0.014216922724145839,
      "grad_norm": 1.9947122161911097,
      "learning_rate": 0.00045454545454545455,
      "loss": 15.9767,
      "step": 31
    },
    {
      "epoch": 0.014675533134602155,
      "grad_norm": 1.3424810295772567,
      "learning_rate": 0.0004696969696969697,
      "loss": 16.0103,
      "step": 32
    },
    {
      "epoch": 0.015134143545058473,
      "grad_norm": 1.023050420666843,
      "learning_rate": 0.0004848484848484849,
      "loss": 15.934,
      "step": 33
    },
    {
      "epoch": 0.01559275395551479,
      "grad_norm": 1.0215023237040977,
      "learning_rate": 0.0005,
      "loss": 15.8301,
      "step": 34
    },
    {
      "epoch": 0.016051364365971106,
      "grad_norm": 1.2049622939872568,
      "learning_rate": 0.0005151515151515151,
      "loss": 15.9375,
      "step": 35
    },
    {
      "epoch": 0.016509974776427426,
      "grad_norm": 0.9070255921308118,
      "learning_rate": 0.0005303030303030302,
      "loss": 15.3409,
      "step": 36
    },
    {
      "epoch": 0.016968585186883742,
      "grad_norm": 0.8209869121529517,
      "learning_rate": 0.0005454545454545455,
      "loss": 15.7462,
      "step": 37
    },
    {
      "epoch": 0.01742719559734006,
      "grad_norm": 0.8236996283690791,
      "learning_rate": 0.0005606060606060606,
      "loss": 15.6093,
      "step": 38
    },
    {
      "epoch": 0.017885806007796375,
      "grad_norm": 1.0167698606345679,
      "learning_rate": 0.0005757575757575758,
      "loss": 15.7997,
      "step": 39
    },
    {
      "epoch": 0.018344416418252695,
      "grad_norm": 0.8492338540007056,
      "learning_rate": 0.0005909090909090909,
      "loss": 15.6869,
      "step": 40
    },
    {
      "epoch": 0.01880302682870901,
      "grad_norm": 0.8188135979381675,
      "learning_rate": 0.0006060606060606061,
      "loss": 15.4534,
      "step": 41
    },
    {
      "epoch": 0.019261637239165328,
      "grad_norm": 0.7732199750727218,
      "learning_rate": 0.0006212121212121212,
      "loss": 15.4955,
      "step": 42
    },
    {
      "epoch": 0.019720247649621648,
      "grad_norm": 0.7713223890869648,
      "learning_rate": 0.0006363636363636364,
      "loss": 15.5883,
      "step": 43
    },
    {
      "epoch": 0.020178858060077964,
      "grad_norm": 0.781251961586397,
      "learning_rate": 0.0006515151515151515,
      "loss": 15.5059,
      "step": 44
    },
    {
      "epoch": 0.02063746847053428,
      "grad_norm": 0.7594969131458282,
      "learning_rate": 0.0006666666666666666,
      "loss": 15.3821,
      "step": 45
    },
    {
      "epoch": 0.021096078880990597,
      "grad_norm": 0.7017687379712232,
      "learning_rate": 0.0006818181818181818,
      "loss": 15.3604,
      "step": 46
    },
    {
      "epoch": 0.021554689291446917,
      "grad_norm": 0.7209743613510067,
      "learning_rate": 0.000696969696969697,
      "loss": 15.2353,
      "step": 47
    },
    {
      "epoch": 0.022013299701903234,
      "grad_norm": 0.8549829155665253,
      "learning_rate": 0.0007121212121212122,
      "loss": 15.362,
      "step": 48
    },
    {
      "epoch": 0.02247191011235955,
      "grad_norm": 0.7285255461495379,
      "learning_rate": 0.0007272727272727273,
      "loss": 15.4395,
      "step": 49
    },
    {
      "epoch": 0.022930520522815866,
      "grad_norm": 0.8781335963052438,
      "learning_rate": 0.0007424242424242425,
      "loss": 15.352,
      "step": 50
    },
    {
      "epoch": 0.023389130933272186,
      "grad_norm": 0.844449453711172,
      "learning_rate": 0.0007575757575757576,
      "loss": 15.433,
      "step": 51
    },
    {
      "epoch": 0.023847741343728503,
      "grad_norm": 0.7335223783385711,
      "learning_rate": 0.0007727272727272727,
      "loss": 15.4348,
      "step": 52
    },
    {
      "epoch": 0.02430635175418482,
      "grad_norm": 1.0577590443058638,
      "learning_rate": 0.0007878787878787878,
      "loss": 15.38,
      "step": 53
    },
    {
      "epoch": 0.02476496216464114,
      "grad_norm": 0.9220371452277866,
      "learning_rate": 0.000803030303030303,
      "loss": 15.2993,
      "step": 54
    },
    {
      "epoch": 0.025223572575097455,
      "grad_norm": 1.094393727207949,
      "learning_rate": 0.0008181818181818183,
      "loss": 15.3775,
      "step": 55
    },
    {
      "epoch": 0.025682182985553772,
      "grad_norm": 1.0866329934181784,
      "learning_rate": 0.0008333333333333334,
      "loss": 15.3757,
      "step": 56
    },
    {
      "epoch": 0.02614079339601009,
      "grad_norm": 0.8177430767247976,
      "learning_rate": 0.0008484848484848485,
      "loss": 15.1457,
      "step": 57
    },
    {
      "epoch": 0.026599403806466408,
      "grad_norm": 1.0562645812410703,
      "learning_rate": 0.0008636363636363636,
      "loss": 15.4363,
      "step": 58
    },
    {
      "epoch": 0.027058014216922725,
      "grad_norm": 1.1971577670441162,
      "learning_rate": 0.0008787878787878789,
      "loss": 15.0848,
      "step": 59
    },
    {
      "epoch": 0.02751662462737904,
      "grad_norm": 0.9914303226757337,
      "learning_rate": 0.000893939393939394,
      "loss": 15.0097,
      "step": 60
    },
    {
      "epoch": 0.027975235037835357,
      "grad_norm": 1.2827843413822768,
      "learning_rate": 0.0009090909090909091,
      "loss": 14.9807,
      "step": 61
    },
    {
      "epoch": 0.028433845448291677,
      "grad_norm": 1.2447131874039858,
      "learning_rate": 0.0009242424242424242,
      "loss": 15.1049,
      "step": 62
    },
    {
      "epoch": 0.028892455858747994,
      "grad_norm": 1.2806197272069295,
      "learning_rate": 0.0009393939393939394,
      "loss": 15.0899,
      "step": 63
    },
    {
      "epoch": 0.02935106626920431,
      "grad_norm": 1.1447674528706258,
      "learning_rate": 0.0009545454545454546,
      "loss": 15.3125,
      "step": 64
    },
    {
      "epoch": 0.029809676679660627,
      "grad_norm": 1.1958111170255468,
      "learning_rate": 0.0009696969696969698,
      "loss": 14.8599,
      "step": 65
    },
    {
      "epoch": 0.030268287090116947,
      "grad_norm": 1.1853783564698552,
      "learning_rate": 0.000984848484848485,
      "loss": 15.1386,
      "step": 66
    },
    {
      "epoch": 0.030726897500573263,
      "grad_norm": 1.09860909839044,
      "learning_rate": 0.001,
      "loss": 14.9748,
      "step": 67
    },
    {
      "epoch": 0.03118550791102958,
      "grad_norm": 1.2493748383047238,
      "learning_rate": 0.0009999994478847943,
      "loss": 14.7389,
      "step": 68
    },
    {
      "epoch": 0.0316441183214859,
      "grad_norm": 1.2271707493600774,
      "learning_rate": 0.0009999977915403962,
      "loss": 14.9127,
      "step": 69
    },
    {
      "epoch": 0.03210272873194221,
      "grad_norm": 1.3273843312809783,
      "learning_rate": 0.0009999950309704639,
      "loss": 14.938,
      "step": 70
    },
    {
      "epoch": 0.03256133914239853,
      "grad_norm": 0.941986629735444,
      "learning_rate": 0.000999991166181094,
      "loss": 14.9414,
      "step": 71
    },
    {
      "epoch": 0.03301994955285485,
      "grad_norm": 1.1225828796188861,
      "learning_rate": 0.0009999861971808216,
      "loss": 14.7712,
      "step": 72
    },
    {
      "epoch": 0.033478559963311165,
      "grad_norm": 1.0121117472022703,
      "learning_rate": 0.0009999801239806208,
      "loss": 14.7634,
      "step": 73
    },
    {
      "epoch": 0.033937170373767485,
      "grad_norm": 1.09841651482702,
      "learning_rate": 0.0009999729465939035,
      "loss": 14.9266,
      "step": 74
    },
    {
      "epoch": 0.034395780784223805,
      "grad_norm": 1.6371838573405093,
      "learning_rate": 0.0009999646650365212,
      "loss": 14.9077,
      "step": 75
    },
    {
      "epoch": 0.03485439119468012,
      "grad_norm": 1.2208524654589825,
      "learning_rate": 0.0009999552793267634,
      "loss": 14.672,
      "step": 76
    },
    {
      "epoch": 0.03531300160513644,
      "grad_norm": 1.1776188743157354,
      "learning_rate": 0.0009999447894853577,
      "loss": 14.7105,
      "step": 77
    },
    {
      "epoch": 0.03577161201559275,
      "grad_norm": 1.337419194327386,
      "learning_rate": 0.0009999331955354708,
      "loss": 14.7454,
      "step": 78
    },
    {
      "epoch": 0.03623022242604907,
      "grad_norm": 1.3407373868077603,
      "learning_rate": 0.0009999204975027073,
      "loss": 14.6488,
      "step": 79
    },
    {
      "epoch": 0.03668883283650539,
      "grad_norm": 1.0966241938001464,
      "learning_rate": 0.0009999066954151103,
      "loss": 14.6245,
      "step": 80
    },
    {
      "epoch": 0.0371474432469617,
      "grad_norm": 1.0203298550454396,
      "learning_rate": 0.0009998917893031614,
      "loss": 14.6681,
      "step": 81
    },
    {
      "epoch": 0.03760605365741802,
      "grad_norm": 0.9755958627266572,
      "learning_rate": 0.0009998757791997801,
      "loss": 14.8171,
      "step": 82
    },
    {
      "epoch": 0.03806466406787434,
      "grad_norm": 0.9211743053853596,
      "learning_rate": 0.0009998586651403238,
      "loss": 14.6743,
      "step": 83
    },
    {
      "epoch": 0.038523274478330656,
      "grad_norm": 0.9819503348358914,
      "learning_rate": 0.0009998404471625885,
      "loss": 14.4026,
      "step": 84
    },
    {
      "epoch": 0.038981884888786976,
      "grad_norm": 1.254374588941118,
      "learning_rate": 0.0009998211253068078,
      "loss": 14.6649,
      "step": 85
    },
    {
      "epoch": 0.039440495299243296,
      "grad_norm": 1.103026799577137,
      "learning_rate": 0.0009998006996156535,
      "loss": 14.718,
      "step": 86
    },
    {
      "epoch": 0.03989910570969961,
      "grad_norm": 1.315397563080452,
      "learning_rate": 0.0009997791701342347,
      "loss": 14.4128,
      "step": 87
    },
    {
      "epoch": 0.04035771612015593,
      "grad_norm": 0.9658559425230993,
      "learning_rate": 0.0009997565369100983,
      "loss": 14.6637,
      "step": 88
    },
    {
      "epoch": 0.04081632653061224,
      "grad_norm": 1.038055115329216,
      "learning_rate": 0.0009997327999932291,
      "loss": 14.7041,
      "step": 89
    },
    {
      "epoch": 0.04127493694106856,
      "grad_norm": 1.0554041352813202,
      "learning_rate": 0.000999707959436049,
      "loss": 14.557,
      "step": 90
    },
    {
      "epoch": 0.04173354735152488,
      "grad_norm": 0.9212115520028313,
      "learning_rate": 0.0009996820152934176,
      "loss": 14.4034,
      "step": 91
    },
    {
      "epoch": 0.042192157761981194,
      "grad_norm": 0.9536782324138351,
      "learning_rate": 0.000999654967622631,
      "loss": 14.488,
      "step": 92
    },
    {
      "epoch": 0.042650768172437514,
      "grad_norm": 0.9751086957915003,
      "learning_rate": 0.0009996268164834238,
      "loss": 14.6462,
      "step": 93
    },
    {
      "epoch": 0.043109378582893834,
      "grad_norm": 1.0012216956341355,
      "learning_rate": 0.000999597561937966,
      "loss": 14.4129,
      "step": 94
    },
    {
      "epoch": 0.04356798899335015,
      "grad_norm": 0.940256998164777,
      "learning_rate": 0.0009995672040508656,
      "loss": 14.2974,
      "step": 95
    },
    {
      "epoch": 0.04402659940380647,
      "grad_norm": 0.8768513435651646,
      "learning_rate": 0.0009995357428891662,
      "loss": 14.5366,
      "step": 96
    },
    {
      "epoch": 0.04448520981426279,
      "grad_norm": 0.9616990725381331,
      "learning_rate": 0.0009995031785223491,
      "loss": 14.4533,
      "step": 97
    },
    {
      "epoch": 0.0449438202247191,
      "grad_norm": 0.9750770257210554,
      "learning_rate": 0.000999469511022331,
      "loss": 14.3159,
      "step": 98
    },
    {
      "epoch": 0.04540243063517542,
      "grad_norm": 1.0137729989402147,
      "learning_rate": 0.0009994347404634657,
      "loss": 14.4405,
      "step": 99
    },
    {
      "epoch": 0.04586104104563173,
      "grad_norm": 0.9960968508391641,
      "learning_rate": 0.0009993988669225423,
      "loss": 14.517,
      "step": 100
    },
    {
      "epoch": 0.04631965145608805,
      "grad_norm": 1.0877803283437801,
      "learning_rate": 0.000999361890478786,
      "loss": 14.5907,
      "step": 101
    },
    {
      "epoch": 0.04677826186654437,
      "grad_norm": 0.9344528746287649,
      "learning_rate": 0.0009993238112138583,
      "loss": 14.3073,
      "step": 102
    },
    {
      "epoch": 0.047236872277000685,
      "grad_norm": 0.9449729176494712,
      "learning_rate": 0.0009992846292118554,
      "loss": 14.0676,
      "step": 103
    },
    {
      "epoch": 0.047695482687457005,
      "grad_norm": 0.9503697953565843,
      "learning_rate": 0.000999244344559309,
      "loss": 14.3356,
      "step": 104
    },
    {
      "epoch": 0.048154093097913325,
      "grad_norm": 0.9277618785596007,
      "learning_rate": 0.0009992029573451869,
      "loss": 14.1718,
      "step": 105
    },
    {
      "epoch": 0.04861270350836964,
      "grad_norm": 0.9852363123945397,
      "learning_rate": 0.0009991604676608905,
      "loss": 14.4984,
      "step": 106
    },
    {
      "epoch": 0.04907131391882596,
      "grad_norm": 1.086495979698606,
      "learning_rate": 0.0009991168756002568,
      "loss": 14.4014,
      "step": 107
    },
    {
      "epoch": 0.04952992432928228,
      "grad_norm": 1.0370539553431135,
      "learning_rate": 0.0009990721812595574,
      "loss": 14.3689,
      "step": 108
    },
    {
      "epoch": 0.04998853473973859,
      "grad_norm": 1.0817911225048398,
      "learning_rate": 0.0009990263847374976,
      "loss": 14.4904,
      "step": 109
    },
    {
      "epoch": 0.05044714515019491,
      "grad_norm": 1.0101023067526118,
      "learning_rate": 0.0009989794861352173,
      "loss": 14.3605,
      "step": 110
    },
    {
      "epoch": 0.050905755560651224,
      "grad_norm": 1.1708336470990706,
      "learning_rate": 0.0009989314855562905,
      "loss": 14.2928,
      "step": 111
    },
    {
      "epoch": 0.051364365971107544,
      "grad_norm": 0.9920041410473656,
      "learning_rate": 0.0009988823831067245,
      "loss": 14.3405,
      "step": 112
    },
    {
      "epoch": 0.051822976381563864,
      "grad_norm": 1.0720592023885727,
      "learning_rate": 0.0009988321788949597,
      "loss": 14.1967,
      "step": 113
    },
    {
      "epoch": 0.05228158679202018,
      "grad_norm": 0.9153496002591678,
      "learning_rate": 0.0009987808730318709,
      "loss": 14.3601,
      "step": 114
    },
    {
      "epoch": 0.052740197202476496,
      "grad_norm": 0.9528787042591962,
      "learning_rate": 0.0009987284656307644,
      "loss": 14.194,
      "step": 115
    },
    {
      "epoch": 0.053198807612932816,
      "grad_norm": 0.9681838769445387,
      "learning_rate": 0.0009986749568073802,
      "loss": 14.4711,
      "step": 116
    },
    {
      "epoch": 0.05365741802338913,
      "grad_norm": 1.0776760809709376,
      "learning_rate": 0.0009986203466798905,
      "loss": 14.4439,
      "step": 117
    },
    {
      "epoch": 0.05411602843384545,
      "grad_norm": 0.9493049859092937,
      "learning_rate": 0.0009985646353688996,
      "loss": 14.3177,
      "step": 118
    },
    {
      "epoch": 0.05457463884430177,
      "grad_norm": 0.9665402235515973,
      "learning_rate": 0.0009985078229974437,
      "loss": 14.4309,
      "step": 119
    },
    {
      "epoch": 0.05503324925475808,
      "grad_norm": 0.9317627752747084,
      "learning_rate": 0.0009984499096909905,
      "loss": 14.3978,
      "step": 120
    },
    {
      "epoch": 0.0554918596652144,
      "grad_norm": 0.9011201561339094,
      "learning_rate": 0.0009983908955774397,
      "loss": 14.4728,
      "step": 121
    },
    {
      "epoch": 0.055950470075670715,
      "grad_norm": 0.954281378076419,
      "learning_rate": 0.0009983307807871211,
      "loss": 14.334,
      "step": 122
    },
    {
      "epoch": 0.056409080486127035,
      "grad_norm": 0.9287406378127513,
      "learning_rate": 0.0009982695654527965,
      "loss": 14.3169,
      "step": 123
    },
    {
      "epoch": 0.056867690896583355,
      "grad_norm": 1.0217723349377639,
      "learning_rate": 0.0009982072497096571,
      "loss": 14.2202,
      "step": 124
    },
    {
      "epoch": 0.05732630130703967,
      "grad_norm": 1.0335325678553218,
      "learning_rate": 0.000998143833695325,
      "loss": 14.2291,
      "step": 125
    },
    {
      "epoch": 0.05778491171749599,
      "grad_norm": 0.9980223031201237,
      "learning_rate": 0.0009980793175498517,
      "loss": 14.1907,
      "step": 126
    },
    {
      "epoch": 0.05824352212795231,
      "grad_norm": 1.009997460684765,
      "learning_rate": 0.000998013701415719,
      "loss": 14.2987,
      "step": 127
    },
    {
      "epoch": 0.05870213253840862,
      "grad_norm": 0.9871920508175357,
      "learning_rate": 0.0009979469854378372,
      "loss": 14.1411,
      "step": 128
    },
    {
      "epoch": 0.05916074294886494,
      "grad_norm": 0.8939579470639442,
      "learning_rate": 0.000997879169763546,
      "loss": 14.1859,
      "step": 129
    },
    {
      "epoch": 0.05961935335932125,
      "grad_norm": 1.030680522223415,
      "learning_rate": 0.000997810254542614,
      "loss": 14.1557,
      "step": 130
    },
    {
      "epoch": 0.06007796376977757,
      "grad_norm": 0.8618724567027946,
      "learning_rate": 0.0009977402399272374,
      "loss": 13.8624,
      "step": 131
    },
    {
      "epoch": 0.06053657418023389,
      "grad_norm": 1.1631107415758148,
      "learning_rate": 0.0009976691260720407,
      "loss": 14.1947,
      "step": 132
    },
    {
      "epoch": 0.060995184590690206,
      "grad_norm": 0.9793885429101116,
      "learning_rate": 0.0009975969131340763,
      "loss": 14.1716,
      "step": 133
    },
    {
      "epoch": 0.061453795001146526,
      "grad_norm": 0.9553272638109744,
      "learning_rate": 0.0009975236012728236,
      "loss": 14.2149,
      "step": 134
    },
    {
      "epoch": 0.061912405411602846,
      "grad_norm": 0.8969373701253779,
      "learning_rate": 0.0009974491906501886,
      "loss": 14.2273,
      "step": 135
    },
    {
      "epoch": 0.06237101582205916,
      "grad_norm": 1.0190150097610968,
      "learning_rate": 0.0009973736814305049,
      "loss": 14.2899,
      "step": 136
    },
    {
      "epoch": 0.06282962623251548,
      "grad_norm": 0.9729001709328742,
      "learning_rate": 0.0009972970737805312,
      "loss": 14.0648,
      "step": 137
    },
    {
      "epoch": 0.0632882366429718,
      "grad_norm": 1.0088917808269098,
      "learning_rate": 0.0009972193678694525,
      "loss": 14.1464,
      "step": 138
    },
    {
      "epoch": 0.06374684705342812,
      "grad_norm": 1.063952894086814,
      "learning_rate": 0.0009971405638688794,
      "loss": 14.0549,
      "step": 139
    },
    {
      "epoch": 0.06420545746388442,
      "grad_norm": 0.8570571142004274,
      "learning_rate": 0.0009970606619528475,
      "loss": 14.1771,
      "step": 140
    },
    {
      "epoch": 0.06466406787434074,
      "grad_norm": 1.0211308876973242,
      "learning_rate": 0.000996979662297817,
      "loss": 13.9167,
      "step": 141
    },
    {
      "epoch": 0.06512267828479706,
      "grad_norm": 0.8883011799340447,
      "learning_rate": 0.0009968975650826721,
      "loss": 14.1102,
      "step": 142
    },
    {
      "epoch": 0.06558128869525338,
      "grad_norm": 0.9498697156254627,
      "learning_rate": 0.000996814370488722,
      "loss": 14.1868,
      "step": 143
    },
    {
      "epoch": 0.0660398991057097,
      "grad_norm": 0.9975747710721354,
      "learning_rate": 0.000996730078699698,
      "loss": 13.8342,
      "step": 144
    },
    {
      "epoch": 0.06649850951616602,
      "grad_norm": 0.9856737964374593,
      "learning_rate": 0.0009966446899017558,
      "loss": 14.2594,
      "step": 145
    },
    {
      "epoch": 0.06695711992662233,
      "grad_norm": 0.9354999451822342,
      "learning_rate": 0.0009965582042834728,
      "loss": 14.1801,
      "step": 146
    },
    {
      "epoch": 0.06741573033707865,
      "grad_norm": 0.9988048636913349,
      "learning_rate": 0.0009964706220358492,
      "loss": 14.0866,
      "step": 147
    },
    {
      "epoch": 0.06787434074753497,
      "grad_norm": 1.0025384533596213,
      "learning_rate": 0.000996381943352307,
      "loss": 14.1463,
      "step": 148
    },
    {
      "epoch": 0.06833295115799129,
      "grad_norm": 0.9421052736529498,
      "learning_rate": 0.0009962921684286896,
      "loss": 14.2356,
      "step": 149
    },
    {
      "epoch": 0.06879156156844761,
      "grad_norm": 1.032057498829817,
      "learning_rate": 0.0009962012974632614,
      "loss": 14.2268,
      "step": 150
    },
    {
      "epoch": 0.06925017197890392,
      "grad_norm": 0.9363501510766711,
      "learning_rate": 0.0009961093306567075,
      "loss": 14.2247,
      "step": 151
    },
    {
      "epoch": 0.06970878238936024,
      "grad_norm": 1.0695259452344137,
      "learning_rate": 0.0009960162682121328,
      "loss": 14.2713,
      "step": 152
    },
    {
      "epoch": 0.07016739279981656,
      "grad_norm": 0.9410698950649806,
      "learning_rate": 0.0009959221103350623,
      "loss": 14.0763,
      "step": 153
    },
    {
      "epoch": 0.07062600321027288,
      "grad_norm": 0.9886333828663348,
      "learning_rate": 0.0009958268572334394,
      "loss": 14.0919,
      "step": 154
    },
    {
      "epoch": 0.0710846136207292,
      "grad_norm": 0.9360105633676409,
      "learning_rate": 0.0009957305091176274,
      "loss": 14.0247,
      "step": 155
    },
    {
      "epoch": 0.0715432240311855,
      "grad_norm": 1.1119651891732945,
      "learning_rate": 0.0009956330662004075,
      "loss": 14.1982,
      "step": 156
    },
    {
      "epoch": 0.07200183444164182,
      "grad_norm": 0.9426897142171574,
      "learning_rate": 0.0009955345286969779,
      "loss": 13.9816,
      "step": 157
    },
    {
      "epoch": 0.07246044485209814,
      "grad_norm": 0.9882606609747967,
      "learning_rate": 0.0009954348968249551,
      "loss": 14.166,
      "step": 158
    },
    {
      "epoch": 0.07291905526255446,
      "grad_norm": 0.9660021357167172,
      "learning_rate": 0.0009953341708043724,
      "loss": 13.9635,
      "step": 159
    },
    {
      "epoch": 0.07337766567301078,
      "grad_norm": 0.9808253636154712,
      "learning_rate": 0.0009952323508576793,
      "loss": 13.9935,
      "step": 160
    },
    {
      "epoch": 0.0738362760834671,
      "grad_norm": 0.9642236658153981,
      "learning_rate": 0.0009951294372097406,
      "loss": 14.0638,
      "step": 161
    },
    {
      "epoch": 0.0742948864939234,
      "grad_norm": 0.9576300368087867,
      "learning_rate": 0.0009950254300878378,
      "loss": 14.1028,
      "step": 162
    },
    {
      "epoch": 0.07475349690437973,
      "grad_norm": 1.043203606040842,
      "learning_rate": 0.000994920329721666,
      "loss": 13.8592,
      "step": 163
    },
    {
      "epoch": 0.07521210731483605,
      "grad_norm": 1.243919659100132,
      "learning_rate": 0.0009948141363433356,
      "loss": 13.7336,
      "step": 164
    },
    {
      "epoch": 0.07567071772529237,
      "grad_norm": 1.010057825333289,
      "learning_rate": 0.00099470685018737,
      "loss": 14.0528,
      "step": 165
    },
    {
      "epoch": 0.07612932813574869,
      "grad_norm": 1.3830456591016136,
      "learning_rate": 0.0009945984714907073,
      "loss": 13.9526,
      "step": 166
    },
    {
      "epoch": 0.07658793854620499,
      "grad_norm": 0.9773664622217801,
      "learning_rate": 0.000994489000492697,
      "loss": 14.0551,
      "step": 167
    },
    {
      "epoch": 0.07704654895666131,
      "grad_norm": 1.2850904853713843,
      "learning_rate": 0.0009943784374351016,
      "loss": 13.9929,
      "step": 168
    },
    {
      "epoch": 0.07750515936711763,
      "grad_norm": 0.85500712791591,
      "learning_rate": 0.0009942667825620951,
      "loss": 13.5275,
      "step": 169
    },
    {
      "epoch": 0.07796376977757395,
      "grad_norm": 1.0129487516942262,
      "learning_rate": 0.0009941540361202634,
      "loss": 13.9599,
      "step": 170
    },
    {
      "epoch": 0.07842238018803027,
      "grad_norm": 1.0968595189173094,
      "learning_rate": 0.0009940401983586022,
      "loss": 14.1968,
      "step": 171
    },
    {
      "epoch": 0.07888099059848659,
      "grad_norm": 0.8832264020653982,
      "learning_rate": 0.000993925269528518,
      "loss": 14.1757,
      "step": 172
    },
    {
      "epoch": 0.0793396010089429,
      "grad_norm": 1.0651465130547408,
      "learning_rate": 0.0009938092498838265,
      "loss": 14.085,
      "step": 173
    },
    {
      "epoch": 0.07979821141939922,
      "grad_norm": 1.0187552881798372,
      "learning_rate": 0.0009936921396807524,
      "loss": 13.913,
      "step": 174
    },
    {
      "epoch": 0.08025682182985554,
      "grad_norm": 0.8809841418135833,
      "learning_rate": 0.0009935739391779292,
      "loss": 14.0014,
      "step": 175
    },
    {
      "epoch": 0.08071543224031186,
      "grad_norm": 0.925813988718372,
      "learning_rate": 0.000993454648636398,
      "loss": 13.8332,
      "step": 176
    },
    {
      "epoch": 0.08117404265076818,
      "grad_norm": 1.0501823757226874,
      "learning_rate": 0.0009933342683196074,
      "loss": 14.2632,
      "step": 177
    },
    {
      "epoch": 0.08163265306122448,
      "grad_norm": 0.8628724357912126,
      "learning_rate": 0.0009932127984934125,
      "loss": 13.963,
      "step": 178
    },
    {
      "epoch": 0.0820912634716808,
      "grad_norm": 1.020877298516112,
      "learning_rate": 0.0009930902394260745,
      "loss": 14.1151,
      "step": 179
    },
    {
      "epoch": 0.08254987388213712,
      "grad_norm": 0.9286129415416989,
      "learning_rate": 0.0009929665913882607,
      "loss": 13.8277,
      "step": 180
    },
    {
      "epoch": 0.08300848429259344,
      "grad_norm": 0.9930097845536702,
      "learning_rate": 0.0009928418546530425,
      "loss": 13.9658,
      "step": 181
    },
    {
      "epoch": 0.08346709470304976,
      "grad_norm": 0.976015139724463,
      "learning_rate": 0.0009927160294958964,
      "loss": 14.0074,
      "step": 182
    },
    {
      "epoch": 0.08392570511350608,
      "grad_norm": 1.0167753330349252,
      "learning_rate": 0.000992589116194702,
      "loss": 13.9547,
      "step": 183
    },
    {
      "epoch": 0.08438431552396239,
      "grad_norm": 1.066386006061831,
      "learning_rate": 0.000992461115029743,
      "loss": 13.8401,
      "step": 184
    },
    {
      "epoch": 0.08484292593441871,
      "grad_norm": 0.9120956865055443,
      "learning_rate": 0.000992332026283704,
      "loss": 13.7789,
      "step": 185
    },
    {
      "epoch": 0.08530153634487503,
      "grad_norm": 0.9031013976134684,
      "learning_rate": 0.0009922018502416736,
      "loss": 13.7069,
      "step": 186
    },
    {
      "epoch": 0.08576014675533135,
      "grad_norm": 1.0431189522652191,
      "learning_rate": 0.0009920705871911395,
      "loss": 14.0701,
      "step": 187
    },
    {
      "epoch": 0.08621875716578767,
      "grad_norm": 1.0067617895978556,
      "learning_rate": 0.0009919382374219915,
      "loss": 13.8173,
      "step": 188
    },
    {
      "epoch": 0.08667736757624397,
      "grad_norm": 0.9596657265112611,
      "learning_rate": 0.0009918048012265187,
      "loss": 13.7727,
      "step": 189
    },
    {
      "epoch": 0.0871359779867003,
      "grad_norm": 1.0235585303122163,
      "learning_rate": 0.0009916702788994097,
      "loss": 13.8623,
      "step": 190
    },
    {
      "epoch": 0.08759458839715661,
      "grad_norm": 0.9560129811168615,
      "learning_rate": 0.0009915346707377519,
      "loss": 13.8903,
      "step": 191
    },
    {
      "epoch": 0.08805319880761293,
      "grad_norm": 0.8934671046374726,
      "learning_rate": 0.0009913979770410305,
      "loss": 13.9531,
      "step": 192
    },
    {
      "epoch": 0.08851180921806925,
      "grad_norm": 0.9231707384293193,
      "learning_rate": 0.0009912601981111285,
      "loss": 14.0158,
      "step": 193
    },
    {
      "epoch": 0.08897041962852557,
      "grad_norm": 0.9946128865460853,
      "learning_rate": 0.0009911213342523248,
      "loss": 14.0164,
      "step": 194
    },
    {
      "epoch": 0.08942903003898188,
      "grad_norm": 1.154988634276254,
      "learning_rate": 0.000990981385771295,
      "loss": 13.9505,
      "step": 195
    },
    {
      "epoch": 0.0898876404494382,
      "grad_norm": 0.8937850467606177,
      "learning_rate": 0.00099084035297711,
      "loss": 13.9134,
      "step": 196
    },
    {
      "epoch": 0.09034625085989452,
      "grad_norm": 1.1397657195137032,
      "learning_rate": 0.000990698236181235,
      "loss": 13.9093,
      "step": 197
    },
    {
      "epoch": 0.09080486127035084,
      "grad_norm": 1.0341165390943183,
      "learning_rate": 0.0009905550356975293,
      "loss": 13.9839,
      "step": 198
    },
    {
      "epoch": 0.09126347168080716,
      "grad_norm": 1.8774326959216947,
      "learning_rate": 0.0009904107518422457,
      "loss": 13.6113,
      "step": 199
    },
    {
      "epoch": 0.09172208209126347,
      "grad_norm": 1.2802023576926649,
      "learning_rate": 0.0009902653849340295,
      "loss": 14.0893,
      "step": 200
    },
    {
      "epoch": 0.09218069250171979,
      "grad_norm": 1.1610760506470743,
      "learning_rate": 0.0009901189352939177,
      "loss": 13.9004,
      "step": 201
    },
    {
      "epoch": 0.0926393029121761,
      "grad_norm": 0.967047708897973,
      "learning_rate": 0.0009899714032453387,
      "loss": 13.9026,
      "step": 202
    },
    {
      "epoch": 0.09309791332263243,
      "grad_norm": 1.4898684784897878,
      "learning_rate": 0.000989822789114111,
      "loss": 13.8456,
      "step": 203
    },
    {
      "epoch": 0.09355652373308875,
      "grad_norm": 0.9057607080088937,
      "learning_rate": 0.0009896730932284434,
      "loss": 13.6305,
      "step": 204
    },
    {
      "epoch": 0.09401513414354506,
      "grad_norm": 1.4062292256410258,
      "learning_rate": 0.0009895223159189332,
      "loss": 13.6996,
      "step": 205
    },
    {
      "epoch": 0.09447374455400137,
      "grad_norm": 0.9993287870895877,
      "learning_rate": 0.0009893704575185663,
      "loss": 14.0593,
      "step": 206
    },
    {
      "epoch": 0.09493235496445769,
      "grad_norm": 1.3577506758044278,
      "learning_rate": 0.000989217518362716,
      "loss": 14.004,
      "step": 207
    },
    {
      "epoch": 0.09539096537491401,
      "grad_norm": 1.1629140340091035,
      "learning_rate": 0.0009890634987891425,
      "loss": 13.9355,
      "step": 208
    },
    {
      "epoch": 0.09584957578537033,
      "grad_norm": 1.1328990796316065,
      "learning_rate": 0.0009889083991379917,
      "loss": 13.9463,
      "step": 209
    },
    {
      "epoch": 0.09630818619582665,
      "grad_norm": 1.177934126734097,
      "learning_rate": 0.0009887522197517954,
      "loss": 14.0784,
      "step": 210
    },
    {
      "epoch": 0.09676679660628296,
      "grad_norm": 0.9046782504030542,
      "learning_rate": 0.0009885949609754693,
      "loss": 14.0448,
      "step": 211
    },
    {
      "epoch": 0.09722540701673928,
      "grad_norm": 1.2851234746365416,
      "learning_rate": 0.000988436623156314,
      "loss": 13.7646,
      "step": 212
    },
    {
      "epoch": 0.0976840174271956,
      "grad_norm": 0.8444428383625682,
      "learning_rate": 0.0009882772066440114,
      "loss": 13.8915,
      "step": 213
    },
    {
      "epoch": 0.09814262783765192,
      "grad_norm": 1.1371982520072912,
      "learning_rate": 0.0009881167117906276,
      "loss": 13.7597,
      "step": 214
    },
    {
      "epoch": 0.09860123824810824,
      "grad_norm": 0.8887785465020905,
      "learning_rate": 0.0009879551389506084,
      "loss": 13.7944,
      "step": 215
    },
    {
      "epoch": 0.09905984865856456,
      "grad_norm": 0.8974237890929635,
      "learning_rate": 0.0009877924884807814,
      "loss": 14.021,
      "step": 216
    },
    {
      "epoch": 0.09951845906902086,
      "grad_norm": 0.942984064052092,
      "learning_rate": 0.000987628760740354,
      "loss": 13.6772,
      "step": 217
    },
    {
      "epoch": 0.09997706947947718,
      "grad_norm": 1.1309968670423036,
      "learning_rate": 0.0009874639560909118,
      "loss": 13.7696,
      "step": 218
    },
    {
      "epoch": 0.1004356798899335,
      "grad_norm": 1.0071010004817962,
      "learning_rate": 0.0009872980748964202,
      "loss": 13.8369,
      "step": 219
    },
    {
      "epoch": 0.10089429030038982,
      "grad_norm": 1.1301508431454945,
      "learning_rate": 0.000987131117523221,
      "loss": 14.0964,
      "step": 220
    },
    {
      "epoch": 0.10135290071084614,
      "grad_norm": 1.0683364721873478,
      "learning_rate": 0.000986963084340033,
      "loss": 14.0147,
      "step": 221
    },
    {
      "epoch": 0.10181151112130245,
      "grad_norm": 1.4563616768640353,
      "learning_rate": 0.0009867939757179508,
      "loss": 14.0218,
      "step": 222
    },
    {
      "epoch": 0.10227012153175877,
      "grad_norm": 1.0128263542308527,
      "learning_rate": 0.0009866237920304443,
      "loss": 13.9073,
      "step": 223
    },
    {
      "epoch": 0.10272873194221509,
      "grad_norm": 1.1505238078665918,
      "learning_rate": 0.0009864525336533577,
      "loss": 13.9473,
      "step": 224
    },
    {
      "epoch": 0.10318734235267141,
      "grad_norm": 0.8580804752379698,
      "learning_rate": 0.000986280200964908,
      "loss": 13.8477,
      "step": 225
    },
    {
      "epoch": 0.10364595276312773,
      "grad_norm": 1.0644479348992324,
      "learning_rate": 0.0009861067943456856,
      "loss": 13.848,
      "step": 226
    },
    {
      "epoch": 0.10410456317358405,
      "grad_norm": 0.9411636782978205,
      "learning_rate": 0.000985932314178652,
      "loss": 13.861,
      "step": 227
    },
    {
      "epoch": 0.10456317358404035,
      "grad_norm": 0.9876762256094398,
      "learning_rate": 0.00098575676084914,
      "loss": 13.9532,
      "step": 228
    },
    {
      "epoch": 0.10502178399449667,
      "grad_norm": 1.1251831191259247,
      "learning_rate": 0.0009855801347448518,
      "loss": 13.7917,
      "step": 229
    },
    {
      "epoch": 0.10548039440495299,
      "grad_norm": 0.8967014275418302,
      "learning_rate": 0.0009854024362558596,
      "loss": 13.9552,
      "step": 230
    },
    {
      "epoch": 0.10593900481540931,
      "grad_norm": 1.1060731018496455,
      "learning_rate": 0.0009852236657746035,
      "loss": 13.9075,
      "step": 231
    },
    {
      "epoch": 0.10639761522586563,
      "grad_norm": 0.897012698775401,
      "learning_rate": 0.0009850438236958911,
      "loss": 13.7777,
      "step": 232
    },
    {
      "epoch": 0.10685622563632194,
      "grad_norm": 0.9543683948878471,
      "learning_rate": 0.0009848629104168966,
      "loss": 14.0395,
      "step": 233
    },
    {
      "epoch": 0.10731483604677826,
      "grad_norm": 1.079279851641137,
      "learning_rate": 0.00098468092633716,
      "loss": 13.8962,
      "step": 234
    },
    {
      "epoch": 0.10777344645723458,
      "grad_norm": 0.8379677512500622,
      "learning_rate": 0.0009844978718585855,
      "loss": 13.6721,
      "step": 235
    },
    {
      "epoch": 0.1082320568676909,
      "grad_norm": 0.9380439898634857,
      "learning_rate": 0.0009843137473854423,
      "loss": 13.7439,
      "step": 236
    },
    {
      "epoch": 0.10869066727814722,
      "grad_norm": 0.9284078773422768,
      "learning_rate": 0.000984128553324362,
      "loss": 13.8623,
      "step": 237
    },
    {
      "epoch": 0.10914927768860354,
      "grad_norm": 0.8571993880224613,
      "learning_rate": 0.0009839422900843383,
      "loss": 13.9767,
      "step": 238
    },
    {
      "epoch": 0.10960788809905984,
      "grad_norm": 0.864238428178636,
      "learning_rate": 0.0009837549580767261,
      "loss": 13.7184,
      "step": 239
    },
    {
      "epoch": 0.11006649850951616,
      "grad_norm": 0.8214844497906465,
      "learning_rate": 0.0009835665577152411,
      "loss": 13.8156,
      "step": 240
    },
    {
      "epoch": 0.11052510891997248,
      "grad_norm": 0.8029256161285192,
      "learning_rate": 0.000983377089415958,
      "loss": 13.9412,
      "step": 241
    },
    {
      "epoch": 0.1109837193304288,
      "grad_norm": 0.9223145133272985,
      "learning_rate": 0.0009831865535973102,
      "loss": 13.8071,
      "step": 242
    },
    {
      "epoch": 0.11144232974088512,
      "grad_norm": 0.7648406198304079,
      "learning_rate": 0.0009829949506800885,
      "loss": 13.7499,
      "step": 243
    },
    {
      "epoch": 0.11190094015134143,
      "grad_norm": 0.8110573040431716,
      "learning_rate": 0.0009828022810874405,
      "loss": 13.6242,
      "step": 244
    },
    {
      "epoch": 0.11235955056179775,
      "grad_norm": 0.745629418893408,
      "learning_rate": 0.0009826085452448693,
      "loss": 13.6316,
      "step": 245
    },
    {
      "epoch": 0.11281816097225407,
      "grad_norm": 0.7670005492013716,
      "learning_rate": 0.000982413743580233,
      "loss": 13.7603,
      "step": 246
    },
    {
      "epoch": 0.11327677138271039,
      "grad_norm": 0.8940007779527512,
      "learning_rate": 0.0009822178765237436,
      "loss": 13.9127,
      "step": 247
    },
    {
      "epoch": 0.11373538179316671,
      "grad_norm": 0.7701368402807639,
      "learning_rate": 0.0009820209445079654,
      "loss": 13.8305,
      "step": 248
    },
    {
      "epoch": 0.11419399220362302,
      "grad_norm": 0.8709259205635433,
      "learning_rate": 0.0009818229479678158,
      "loss": 13.958,
      "step": 249
    },
    {
      "epoch": 0.11465260261407934,
      "grad_norm": 0.8068454637636883,
      "learning_rate": 0.0009816238873405615,
      "loss": 13.6482,
      "step": 250
    },
    {
      "epoch": 0.11511121302453566,
      "grad_norm": 0.7644148076278124,
      "learning_rate": 0.0009814237630658207,
      "loss": 13.7377,
      "step": 251
    },
    {
      "epoch": 0.11556982343499198,
      "grad_norm": 0.7703662519871242,
      "learning_rate": 0.00098122257558556,
      "loss": 13.8259,
      "step": 252
    },
    {
      "epoch": 0.1160284338454483,
      "grad_norm": 0.8389404798315125,
      "learning_rate": 0.0009810203253440937,
      "loss": 13.6886,
      "step": 253
    },
    {
      "epoch": 0.11648704425590461,
      "grad_norm": 0.7412007997721695,
      "learning_rate": 0.0009808170127880837,
      "loss": 13.8803,
      "step": 254
    },
    {
      "epoch": 0.11694565466636092,
      "grad_norm": 0.9070152322987837,
      "learning_rate": 0.000980612638366538,
      "loss": 13.7033,
      "step": 255
    },
    {
      "epoch": 0.11740426507681724,
      "grad_norm": 0.8831926409315498,
      "learning_rate": 0.0009804072025308096,
      "loss": 13.8789,
      "step": 256
    },
    {
      "epoch": 0.11786287548727356,
      "grad_norm": 0.7490717233216119,
      "learning_rate": 0.000980200705734595,
      "loss": 13.6346,
      "step": 257
    },
    {
      "epoch": 0.11832148589772988,
      "grad_norm": 0.7785244154182621,
      "learning_rate": 0.0009799931484339344,
      "loss": 13.8397,
      "step": 258
    },
    {
      "epoch": 0.1187800963081862,
      "grad_norm": 1.00677816870912,
      "learning_rate": 0.0009797845310872103,
      "loss": 13.7281,
      "step": 259
    },
    {
      "epoch": 0.1192387067186425,
      "grad_norm": 0.7708739026236421,
      "learning_rate": 0.0009795748541551457,
      "loss": 13.7987,
      "step": 260
    },
    {
      "epoch": 0.11969731712909883,
      "grad_norm": 1.0096608302156165,
      "learning_rate": 0.000979364118100804,
      "loss": 13.7848,
      "step": 261
    },
    {
      "epoch": 0.12015592753955515,
      "grad_norm": 0.7969344694839878,
      "learning_rate": 0.0009791523233895875,
      "loss": 13.7455,
      "step": 262
    },
    {
      "epoch": 0.12061453795001147,
      "grad_norm": 0.8959817235558271,
      "learning_rate": 0.0009789394704892364,
      "loss": 13.7969,
      "step": 263
    },
    {
      "epoch": 0.12107314836046779,
      "grad_norm": 1.0418687008097784,
      "learning_rate": 0.0009787255598698282,
      "loss": 13.9132,
      "step": 264
    },
    {
      "epoch": 0.1215317587709241,
      "grad_norm": 0.7978040664711227,
      "learning_rate": 0.0009785105920037758,
      "loss": 13.7153,
      "step": 265
    },
    {
      "epoch": 0.12199036918138041,
      "grad_norm": 0.890182248079558,
      "learning_rate": 0.0009782945673658275,
      "loss": 13.7011,
      "step": 266
    },
    {
      "epoch": 0.12244897959183673,
      "grad_norm": 0.7806893253104089,
      "learning_rate": 0.0009780774864330654,
      "loss": 13.7174,
      "step": 267
    },
    {
      "epoch": 0.12290759000229305,
      "grad_norm": 0.9616198191358778,
      "learning_rate": 0.000977859349684904,
      "loss": 13.4483,
      "step": 268
    },
    {
      "epoch": 0.12336620041274937,
      "grad_norm": 0.7756817139779628,
      "learning_rate": 0.00097764015760309,
      "loss": 13.9278,
      "step": 269
    },
    {
      "epoch": 0.12382481082320569,
      "grad_norm": 0.8578117973163208,
      "learning_rate": 0.0009774199106717004,
      "loss": 13.679,
      "step": 270
    },
    {
      "epoch": 0.124283421233662,
      "grad_norm": 0.754427739063696,
      "learning_rate": 0.0009771986093771417,
      "loss": 13.5531,
      "step": 271
    },
    {
      "epoch": 0.12474203164411832,
      "grad_norm": 0.8251413632425897,
      "learning_rate": 0.0009769762542081496,
      "loss": 13.6442,
      "step": 272
    },
    {
      "epoch": 0.12520064205457465,
      "grad_norm": 0.7754800090989906,
      "learning_rate": 0.000976752845655786,
      "loss": 13.895,
      "step": 273
    },
    {
      "epoch": 0.12565925246503096,
      "grad_norm": 0.7764035585269081,
      "learning_rate": 0.0009765283842134411,
      "loss": 13.6932,
      "step": 274
    },
    {
      "epoch": 0.12611786287548726,
      "grad_norm": 0.7854154865188464,
      "learning_rate": 0.0009763028703768282,
      "loss": 13.8871,
      "step": 275
    },
    {
      "epoch": 0.1265764732859436,
      "grad_norm": 0.7372044624645672,
      "learning_rate": 0.0009760763046439862,
      "loss": 13.8037,
      "step": 276
    },
    {
      "epoch": 0.1270350836963999,
      "grad_norm": 0.7392287292705926,
      "learning_rate": 0.0009758486875152766,
      "loss": 13.7349,
      "step": 277
    },
    {
      "epoch": 0.12749369410685624,
      "grad_norm": 0.7250284322453953,
      "learning_rate": 0.0009756200194933829,
      "loss": 13.5183,
      "step": 278
    },
    {
      "epoch": 0.12795230451731254,
      "grad_norm": 0.7752547550429824,
      "learning_rate": 0.0009753903010833094,
      "loss": 13.7745,
      "step": 279
    },
    {
      "epoch": 0.12841091492776885,
      "grad_norm": 0.7559064818397717,
      "learning_rate": 0.0009751595327923803,
      "loss": 13.9034,
      "step": 280
    },
    {
      "epoch": 0.12886952533822518,
      "grad_norm": 0.9811160849653299,
      "learning_rate": 0.0009749277151302382,
      "loss": 13.8314,
      "step": 281
    },
    {
      "epoch": 0.1293281357486815,
      "grad_norm": 0.7538810825223687,
      "learning_rate": 0.0009746948486088435,
      "loss": 13.6988,
      "step": 282
    },
    {
      "epoch": 0.12978674615913782,
      "grad_norm": 0.9871223707743749,
      "learning_rate": 0.0009744609337424727,
      "loss": 13.7707,
      "step": 283
    },
    {
      "epoch": 0.13024535656959413,
      "grad_norm": 0.765153231074411,
      "learning_rate": 0.0009742259710477177,
      "loss": 13.9285,
      "step": 284
    },
    {
      "epoch": 0.13070396698005043,
      "grad_norm": 0.7638115456363833,
      "learning_rate": 0.0009739899610434841,
      "loss": 14.0567,
      "step": 285
    },
    {
      "epoch": 0.13116257739050677,
      "grad_norm": 0.9524130370429609,
      "learning_rate": 0.0009737529042509913,
      "loss": 13.715,
      "step": 286
    },
    {
      "epoch": 0.13162118780096307,
      "grad_norm": 0.8054776536706572,
      "learning_rate": 0.0009735148011937693,
      "loss": 13.8157,
      "step": 287
    },
    {
      "epoch": 0.1320797982114194,
      "grad_norm": 1.0075429109465774,
      "learning_rate": 0.00097327565239766,
      "loss": 13.8936,
      "step": 288
    },
    {
      "epoch": 0.13253840862187571,
      "grad_norm": 0.7405173077406935,
      "learning_rate": 0.0009730354583908136,
      "loss": 13.7401,
      "step": 289
    },
    {
      "epoch": 0.13299701903233205,
      "grad_norm": 0.9325450669608536,
      "learning_rate": 0.0009727942197036895,
      "loss": 13.9856,
      "step": 290
    },
    {
      "epoch": 0.13345562944278835,
      "grad_norm": 0.7790219606605414,
      "learning_rate": 0.0009725519368690539,
      "loss": 13.8485,
      "step": 291
    },
    {
      "epoch": 0.13391423985324466,
      "grad_norm": 0.9775871799489847,
      "learning_rate": 0.0009723086104219787,
      "loss": 13.8109,
      "step": 292
    },
    {
      "epoch": 0.134372850263701,
      "grad_norm": 0.7751558777848235,
      "learning_rate": 0.0009720642408998409,
      "loss": 13.6773,
      "step": 293
    },
    {
      "epoch": 0.1348314606741573,
      "grad_norm": 0.8169918920045409,
      "learning_rate": 0.0009718188288423211,
      "loss": 13.9159,
      "step": 294
    },
    {
      "epoch": 0.13529007108461363,
      "grad_norm": 0.9890774494663955,
      "learning_rate": 0.0009715723747914022,
      "loss": 13.8365,
      "step": 295
    },
    {
      "epoch": 0.13574868149506994,
      "grad_norm": 0.9293875092564508,
      "learning_rate": 0.0009713248792913685,
      "loss": 13.5732,
      "step": 296
    },
    {
      "epoch": 0.13620729190552625,
      "grad_norm": 0.9807749560742014,
      "learning_rate": 0.0009710763428888037,
      "loss": 13.8498,
      "step": 297
    },
    {
      "epoch": 0.13666590231598258,
      "grad_norm": 0.7362752634268502,
      "learning_rate": 0.0009708267661325909,
      "loss": 13.8545,
      "step": 298
    },
    {
      "epoch": 0.13712451272643889,
      "grad_norm": 0.8897250940145203,
      "learning_rate": 0.0009705761495739107,
      "loss": 13.8071,
      "step": 299
    },
    {
      "epoch": 0.13758312313689522,
      "grad_norm": 0.7933792214021931,
      "learning_rate": 0.0009703244937662399,
      "loss": 13.7934,
      "step": 300
    },
    {
      "epoch": 0.13804173354735153,
      "grad_norm": 0.807511742234018,
      "learning_rate": 0.0009700717992653505,
      "loss": 13.6457,
      "step": 301
    },
    {
      "epoch": 0.13850034395780783,
      "grad_norm": 0.9208179344637193,
      "learning_rate": 0.0009698180666293083,
      "loss": 13.6384,
      "step": 302
    },
    {
      "epoch": 0.13895895436826416,
      "grad_norm": 0.7795443814041014,
      "learning_rate": 0.000969563296418472,
      "loss": 13.7626,
      "step": 303
    },
    {
      "epoch": 0.13941756477872047,
      "grad_norm": 0.9046777498320203,
      "learning_rate": 0.0009693074891954914,
      "loss": 13.7078,
      "step": 304
    },
    {
      "epoch": 0.1398761751891768,
      "grad_norm": 0.8233797932962604,
      "learning_rate": 0.0009690506455253072,
      "loss": 13.8,
      "step": 305
    },
    {
      "epoch": 0.1403347855996331,
      "grad_norm": 0.9552657858659704,
      "learning_rate": 0.0009687927659751481,
      "loss": 13.7292,
      "step": 306
    },
    {
      "epoch": 0.14079339601008942,
      "grad_norm": 0.7584392488120145,
      "learning_rate": 0.0009685338511145312,
      "loss": 13.7829,
      "step": 307
    },
    {
      "epoch": 0.14125200642054575,
      "grad_norm": 0.8508520164562091,
      "learning_rate": 0.0009682739015152598,
      "loss": 13.6716,
      "step": 308
    },
    {
      "epoch": 0.14171061683100206,
      "grad_norm": 0.7953347848925247,
      "learning_rate": 0.0009680129177514226,
      "loss": 13.8961,
      "step": 309
    },
    {
      "epoch": 0.1421692272414584,
      "grad_norm": 0.7587353949426541,
      "learning_rate": 0.0009677509003993915,
      "loss": 13.767,
      "step": 310
    },
    {
      "epoch": 0.1426278376519147,
      "grad_norm": 0.7398356166340168,
      "learning_rate": 0.0009674878500378221,
      "loss": 13.8518,
      "step": 311
    },
    {
      "epoch": 0.143086448062371,
      "grad_norm": 0.806344234726789,
      "learning_rate": 0.0009672237672476505,
      "loss": 13.826,
      "step": 312
    },
    {
      "epoch": 0.14354505847282734,
      "grad_norm": 0.737622065476737,
      "learning_rate": 0.0009669586526120935,
      "loss": 13.972,
      "step": 313
    },
    {
      "epoch": 0.14400366888328364,
      "grad_norm": 0.8031021946724085,
      "learning_rate": 0.0009666925067166459,
      "loss": 13.6488,
      "step": 314
    },
    {
      "epoch": 0.14446227929373998,
      "grad_norm": 0.7267279865887276,
      "learning_rate": 0.000966425330149081,
      "loss": 13.8072,
      "step": 315
    },
    {
      "epoch": 0.14492088970419628,
      "grad_norm": 0.7409125011227768,
      "learning_rate": 0.0009661571234994475,
      "loss": 13.49,
      "step": 316
    },
    {
      "epoch": 0.14537950011465262,
      "grad_norm": 0.8315731296232655,
      "learning_rate": 0.0009658878873600691,
      "loss": 13.6033,
      "step": 317
    },
    {
      "epoch": 0.14583811052510892,
      "grad_norm": 1.1681063293518155,
      "learning_rate": 0.0009656176223255438,
      "loss": 13.9536,
      "step": 318
    },
    {
      "epoch": 0.14629672093556523,
      "grad_norm": 0.818518002153804,
      "learning_rate": 0.000965346328992741,
      "loss": 13.7393,
      "step": 319
    },
    {
      "epoch": 0.14675533134602156,
      "grad_norm": 0.8290601777305892,
      "learning_rate": 0.0009650740079608014,
      "loss": 13.7177,
      "step": 320
    },
    {
      "epoch": 0.14721394175647787,
      "grad_norm": 0.744896265716409,
      "learning_rate": 0.0009648006598311353,
      "loss": 13.8446,
      "step": 321
    },
    {
      "epoch": 0.1476725521669342,
      "grad_norm": 1.2717899242506208,
      "learning_rate": 0.0009645262852074214,
      "loss": 13.5396,
      "step": 322
    },
    {
      "epoch": 0.1481311625773905,
      "grad_norm": 0.7610674846331765,
      "learning_rate": 0.0009642508846956053,
      "loss": 13.7313,
      "step": 323
    },
    {
      "epoch": 0.1485897729878468,
      "grad_norm": 0.7714885548587029,
      "learning_rate": 0.0009639744589038983,
      "loss": 14.0417,
      "step": 324
    },
    {
      "epoch": 0.14904838339830315,
      "grad_norm": 0.7960694258093151,
      "learning_rate": 0.0009636970084427759,
      "loss": 13.7905,
      "step": 325
    },
    {
      "epoch": 0.14950699380875945,
      "grad_norm": 0.7052552457759382,
      "learning_rate": 0.0009634185339249766,
      "loss": 13.7772,
      "step": 326
    },
    {
      "epoch": 0.1499656042192158,
      "grad_norm": 0.7818590948852636,
      "learning_rate": 0.0009631390359655003,
      "loss": 13.4159,
      "step": 327
    },
    {
      "epoch": 0.1504242146296721,
      "grad_norm": 0.8494545603702908,
      "learning_rate": 0.0009628585151816074,
      "loss": 13.7381,
      "step": 328
    },
    {
      "epoch": 0.1508828250401284,
      "grad_norm": 0.7276109016716474,
      "learning_rate": 0.0009625769721928172,
      "loss": 13.6327,
      "step": 329
    },
    {
      "epoch": 0.15134143545058473,
      "grad_norm": 1.3688157977821749,
      "learning_rate": 0.0009622944076209061,
      "loss": 13.5831,
      "step": 330
    },
    {
      "epoch": 0.15180004586104104,
      "grad_norm": 0.6818493527600906,
      "learning_rate": 0.0009620108220899071,
      "loss": 13.5634,
      "step": 331
    },
    {
      "epoch": 0.15225865627149737,
      "grad_norm": 1.2481981978218017,
      "learning_rate": 0.0009617262162261075,
      "loss": 13.818,
      "step": 332
    },
    {
      "epoch": 0.15271726668195368,
      "grad_norm": 0.7287243145957369,
      "learning_rate": 0.0009614405906580486,
      "loss": 13.8968,
      "step": 333
    },
    {
      "epoch": 0.15317587709240998,
      "grad_norm": 0.8381841077206319,
      "learning_rate": 0.000961153946016523,
      "loss": 13.7389,
      "step": 334
    },
    {
      "epoch": 0.15363448750286632,
      "grad_norm": 0.7462458564707762,
      "learning_rate": 0.000960866282934574,
      "loss": 13.7493,
      "step": 335
    },
    {
      "epoch": 0.15409309791332262,
      "grad_norm": 1.0044627446965098,
      "learning_rate": 0.0009605776020474945,
      "loss": 13.6505,
      "step": 336
    },
    {
      "epoch": 0.15455170832377896,
      "grad_norm": 0.8170146309466773,
      "learning_rate": 0.0009602879039928249,
      "loss": 13.7291,
      "step": 337
    },
    {
      "epoch": 0.15501031873423526,
      "grad_norm": 0.8146589176982727,
      "learning_rate": 0.0009599971894103521,
      "loss": 13.7984,
      "step": 338
    },
    {
      "epoch": 0.1554689291446916,
      "grad_norm": 0.8757404479102209,
      "learning_rate": 0.0009597054589421077,
      "loss": 13.7521,
      "step": 339
    },
    {
      "epoch": 0.1559275395551479,
      "grad_norm": 0.7496534996255376,
      "learning_rate": 0.0009594127132323669,
      "loss": 13.5926,
      "step": 340
    },
    {
      "epoch": 0.1563861499656042,
      "grad_norm": 0.9821193691001876,
      "learning_rate": 0.0009591189529276474,
      "loss": 13.9611,
      "step": 341
    },
    {
      "epoch": 0.15684476037606054,
      "grad_norm": 0.6526627103455207,
      "learning_rate": 0.0009588241786767072,
      "loss": 13.5084,
      "step": 342
    },
    {
      "epoch": 0.15730337078651685,
      "grad_norm": 0.7892093551295954,
      "learning_rate": 0.0009585283911305436,
      "loss": 13.7212,
      "step": 343
    },
    {
      "epoch": 0.15776198119697318,
      "grad_norm": 0.681102585769687,
      "learning_rate": 0.000958231590942392,
      "loss": 13.9351,
      "step": 344
    },
    {
      "epoch": 0.1582205916074295,
      "grad_norm": 0.6867388601635088,
      "learning_rate": 0.0009579337787677238,
      "loss": 13.8999,
      "step": 345
    },
    {
      "epoch": 0.1586792020178858,
      "grad_norm": 0.6746854845886422,
      "learning_rate": 0.0009576349552642456,
      "loss": 13.675,
      "step": 346
    },
    {
      "epoch": 0.15913781242834213,
      "grad_norm": 1.0911837521347254,
      "learning_rate": 0.0009573351210918975,
      "loss": 13.557,
      "step": 347
    },
    {
      "epoch": 0.15959642283879844,
      "grad_norm": 0.7443089266685537,
      "learning_rate": 0.0009570342769128514,
      "loss": 13.7356,
      "step": 348
    },
    {
      "epoch": 0.16005503324925477,
      "grad_norm": 0.7571279137908835,
      "learning_rate": 0.0009567324233915099,
      "loss": 13.7918,
      "step": 349
    },
    {
      "epoch": 0.16051364365971107,
      "grad_norm": 0.7391802209753321,
      "learning_rate": 0.0009564295611945047,
      "loss": 13.6763,
      "step": 350
    },
    {
      "epoch": 0.16097225407016738,
      "grad_norm": 0.7881433431759961,
      "learning_rate": 0.000956125690990695,
      "loss": 13.8075,
      "step": 351
    },
    {
      "epoch": 0.16143086448062371,
      "grad_norm": 0.8545855400317967,
      "learning_rate": 0.0009558208134511665,
      "loss": 13.7591,
      "step": 352
    },
    {
      "epoch": 0.16188947489108002,
      "grad_norm": 0.7765158684831451,
      "learning_rate": 0.0009555149292492289,
      "loss": 13.5504,
      "step": 353
    },
    {
      "epoch": 0.16234808530153635,
      "grad_norm": 0.7735701727746269,
      "learning_rate": 0.0009552080390604159,
      "loss": 13.7747,
      "step": 354
    },
    {
      "epoch": 0.16280669571199266,
      "grad_norm": 0.7357218128154851,
      "learning_rate": 0.0009549001435624823,
      "loss": 13.7289,
      "step": 355
    },
    {
      "epoch": 0.16326530612244897,
      "grad_norm": 0.8095284440795063,
      "learning_rate": 0.0009545912434354029,
      "loss": 13.9221,
      "step": 356
    },
    {
      "epoch": 0.1637239165329053,
      "grad_norm": 0.7251115669174861,
      "learning_rate": 0.0009542813393613721,
      "loss": 14.0739,
      "step": 357
    },
    {
      "epoch": 0.1641825269433616,
      "grad_norm": 0.7772140769173241,
      "learning_rate": 0.0009539704320248006,
      "loss": 13.8878,
      "step": 358
    },
    {
      "epoch": 0.16464113735381794,
      "grad_norm": 0.7801344258918148,
      "learning_rate": 0.0009536585221123151,
      "loss": 13.9684,
      "step": 359
    },
    {
      "epoch": 0.16509974776427425,
      "grad_norm": 0.7391092101925681,
      "learning_rate": 0.0009533456103127565,
      "loss": 13.6496,
      "step": 360
    },
    {
      "epoch": 0.16555835817473058,
      "grad_norm": 0.7902277389922805,
      "learning_rate": 0.000953031697317178,
      "loss": 13.6675,
      "step": 361
    },
    {
      "epoch": 0.16601696858518689,
      "grad_norm": 0.6780907267103548,
      "learning_rate": 0.0009527167838188445,
      "loss": 13.5425,
      "step": 362
    },
    {
      "epoch": 0.1664755789956432,
      "grad_norm": 0.7116178500066342,
      "learning_rate": 0.0009524008705132299,
      "loss": 13.7311,
      "step": 363
    },
    {
      "epoch": 0.16693418940609953,
      "grad_norm": 0.7091534179217759,
      "learning_rate": 0.0009520839580980166,
      "loss": 13.631,
      "step": 364
    },
    {
      "epoch": 0.16739279981655583,
      "grad_norm": 0.7596302910967372,
      "learning_rate": 0.0009517660472730929,
      "loss": 13.9366,
      "step": 365
    },
    {
      "epoch": 0.16785141022701217,
      "grad_norm": 0.7545524700349207,
      "learning_rate": 0.0009514471387405526,
      "loss": 13.8503,
      "step": 366
    },
    {
      "epoch": 0.16831002063746847,
      "grad_norm": 0.7970163404647762,
      "learning_rate": 0.0009511272332046926,
      "loss": 13.5192,
      "step": 367
    },
    {
      "epoch": 0.16876863104792478,
      "grad_norm": 0.7539281829549066,
      "learning_rate": 0.0009508063313720119,
      "loss": 13.8146,
      "step": 368
    },
    {
      "epoch": 0.1692272414583811,
      "grad_norm": 0.7146489918376423,
      "learning_rate": 0.0009504844339512095,
      "loss": 13.8658,
      "step": 369
    },
    {
      "epoch": 0.16968585186883742,
      "grad_norm": 0.7309552863488173,
      "learning_rate": 0.0009501615416531835,
      "loss": 13.6323,
      "step": 370
    },
    {
      "epoch": 0.17014446227929375,
      "grad_norm": 0.6962630974635178,
      "learning_rate": 0.0009498376551910285,
      "loss": 13.9475,
      "step": 371
    },
    {
      "epoch": 0.17060307268975006,
      "grad_norm": 0.8318817058568982,
      "learning_rate": 0.0009495127752800352,
      "loss": 13.7771,
      "step": 372
    },
    {
      "epoch": 0.17106168310020636,
      "grad_norm": 0.7699412301237205,
      "learning_rate": 0.0009491869026376882,
      "loss": 13.6788,
      "step": 373
    },
    {
      "epoch": 0.1715202935106627,
      "grad_norm": 0.7527666026621627,
      "learning_rate": 0.0009488600379836648,
      "loss": 13.5922,
      "step": 374
    },
    {
      "epoch": 0.171978903921119,
      "grad_norm": 0.7468581718106209,
      "learning_rate": 0.0009485321820398321,
      "loss": 13.8485,
      "step": 375
    },
    {
      "epoch": 0.17243751433157534,
      "grad_norm": 0.863462298059065,
      "learning_rate": 0.0009482033355302475,
      "loss": 13.6355,
      "step": 376
    },
    {
      "epoch": 0.17289612474203164,
      "grad_norm": 0.779321939619823,
      "learning_rate": 0.0009478734991811556,
      "loss": 13.8341,
      "step": 377
    },
    {
      "epoch": 0.17335473515248795,
      "grad_norm": 0.7100367620372846,
      "learning_rate": 0.0009475426737209871,
      "loss": 13.7353,
      "step": 378
    },
    {
      "epoch": 0.17381334556294428,
      "grad_norm": 0.8854566516566347,
      "learning_rate": 0.000947210859880357,
      "loss": 13.6826,
      "step": 379
    },
    {
      "epoch": 0.1742719559734006,
      "grad_norm": 0.6908034917262756,
      "learning_rate": 0.0009468780583920631,
      "loss": 13.5216,
      "step": 380
    },
    {
      "epoch": 0.17473056638385692,
      "grad_norm": 0.8087953958767737,
      "learning_rate": 0.0009465442699910846,
      "loss": 13.7772,
      "step": 381
    },
    {
      "epoch": 0.17518917679431323,
      "grad_norm": 0.7477558123133452,
      "learning_rate": 0.0009462094954145801,
      "loss": 13.8953,
      "step": 382
    },
    {
      "epoch": 0.17564778720476956,
      "grad_norm": 0.8037737903491861,
      "learning_rate": 0.0009458737354018859,
      "loss": 14.0056,
      "step": 383
    },
    {
      "epoch": 0.17610639761522587,
      "grad_norm": 0.6885915318667153,
      "learning_rate": 0.000945536990694515,
      "loss": 13.684,
      "step": 384
    },
    {
      "epoch": 0.17656500802568217,
      "grad_norm": 0.7876287106554236,
      "learning_rate": 0.0009451992620361551,
      "loss": 13.577,
      "step": 385
    },
    {
      "epoch": 0.1770236184361385,
      "grad_norm": 0.7130935713203451,
      "learning_rate": 0.0009448605501726664,
      "loss": 13.766,
      "step": 386
    },
    {
      "epoch": 0.17748222884659481,
      "grad_norm": 0.7672495587687314,
      "learning_rate": 0.000944520855852081,
      "loss": 13.82,
      "step": 387
    },
    {
      "epoch": 0.17794083925705115,
      "grad_norm": 0.7116002449487904,
      "learning_rate": 0.0009441801798246002,
      "loss": 13.5955,
      "step": 388
    },
    {
      "epoch": 0.17839944966750745,
      "grad_norm": 0.7538444470239595,
      "learning_rate": 0.0009438385228425939,
      "loss": 13.6764,
      "step": 389
    },
    {
      "epoch": 0.17885806007796376,
      "grad_norm": 0.6829370760783905,
      "learning_rate": 0.0009434958856605982,
      "loss": 13.9547,
      "step": 390
    },
    {
      "epoch": 0.1793166704884201,
      "grad_norm": 0.6481631974208928,
      "learning_rate": 0.0009431522690353137,
      "loss": 13.5158,
      "step": 391
    },
    {
      "epoch": 0.1797752808988764,
      "grad_norm": 0.7264641693959245,
      "learning_rate": 0.0009428076737256044,
      "loss": 13.682,
      "step": 392
    },
    {
      "epoch": 0.18023389130933273,
      "grad_norm": 0.7095581847854678,
      "learning_rate": 0.0009424621004924954,
      "loss": 13.746,
      "step": 393
    },
    {
      "epoch": 0.18069250171978904,
      "grad_norm": 0.6999180482891129,
      "learning_rate": 0.0009421155500991719,
      "loss": 13.8754,
      "step": 394
    },
    {
      "epoch": 0.18115111213024535,
      "grad_norm": 0.7355271341325631,
      "learning_rate": 0.0009417680233109767,
      "loss": 13.4746,
      "step": 395
    },
    {
      "epoch": 0.18160972254070168,
      "grad_norm": 0.6545476525385338,
      "learning_rate": 0.000941419520895409,
      "loss": 13.5966,
      "step": 396
    },
    {
      "epoch": 0.18206833295115799,
      "grad_norm": 0.8243806841495611,
      "learning_rate": 0.0009410700436221229,
      "loss": 13.8226,
      "step": 397
    },
    {
      "epoch": 0.18252694336161432,
      "grad_norm": 0.6224415753285422,
      "learning_rate": 0.0009407195922629252,
      "loss": 13.5823,
      "step": 398
    },
    {
      "epoch": 0.18298555377207062,
      "grad_norm": 0.7688721763846482,
      "learning_rate": 0.000940368167591774,
      "loss": 13.636,
      "step": 399
    },
    {
      "epoch": 0.18344416418252693,
      "grad_norm": 0.692862091583957,
      "learning_rate": 0.0009400157703847769,
      "loss": 13.9231,
      "step": 400
    },
    {
      "epoch": 0.18390277459298326,
      "grad_norm": 0.6568758015781996,
      "learning_rate": 0.0009396624014201895,
      "loss": 13.5427,
      "step": 401
    },
    {
      "epoch": 0.18436138500343957,
      "grad_norm": 0.7797893008705403,
      "learning_rate": 0.000939308061478413,
      "loss": 13.7467,
      "step": 402
    },
    {
      "epoch": 0.1848199954138959,
      "grad_norm": 0.6727323104613856,
      "learning_rate": 0.0009389527513419935,
      "loss": 13.6725,
      "step": 403
    },
    {
      "epoch": 0.1852786058243522,
      "grad_norm": 0.7521422399163014,
      "learning_rate": 0.0009385964717956195,
      "loss": 13.6564,
      "step": 404
    },
    {
      "epoch": 0.18573721623480852,
      "grad_norm": 0.6370092976728949,
      "learning_rate": 0.0009382392236261201,
      "loss": 13.6779,
      "step": 405
    },
    {
      "epoch": 0.18619582664526485,
      "grad_norm": 0.7777881208814299,
      "learning_rate": 0.0009378810076224644,
      "loss": 13.8997,
      "step": 406
    },
    {
      "epoch": 0.18665443705572116,
      "grad_norm": 0.6460089293002719,
      "learning_rate": 0.0009375218245757582,
      "loss": 13.6885,
      "step": 407
    },
    {
      "epoch": 0.1871130474661775,
      "grad_norm": 0.7325098483975537,
      "learning_rate": 0.0009371616752792432,
      "loss": 13.54,
      "step": 408
    },
    {
      "epoch": 0.1875716578766338,
      "grad_norm": 0.6400739924951503,
      "learning_rate": 0.0009368005605282949,
      "loss": 13.6555,
      "step": 409
    },
    {
      "epoch": 0.18803026828709013,
      "grad_norm": 0.709264640413351,
      "learning_rate": 0.0009364384811204212,
      "loss": 13.6919,
      "step": 410
    },
    {
      "epoch": 0.18848887869754644,
      "grad_norm": 0.773996310525167,
      "learning_rate": 0.00093607543785526,
      "loss": 13.789,
      "step": 411
    },
    {
      "epoch": 0.18894748910800274,
      "grad_norm": 0.7221080356667195,
      "learning_rate": 0.0009357114315345787,
      "loss": 13.825,
      "step": 412
    },
    {
      "epoch": 0.18940609951845908,
      "grad_norm": 0.9504590486887079,
      "learning_rate": 0.0009353464629622705,
      "loss": 13.803,
      "step": 413
    },
    {
      "epoch": 0.18986470992891538,
      "grad_norm": 0.727310147531057,
      "learning_rate": 0.0009349805329443544,
      "loss": 13.5932,
      "step": 414
    },
    {
      "epoch": 0.19032332033937172,
      "grad_norm": 0.8787215054578811,
      "learning_rate": 0.0009346136422889724,
      "loss": 13.7718,
      "step": 415
    },
    {
      "epoch": 0.19078193074982802,
      "grad_norm": 0.6824103353749684,
      "learning_rate": 0.0009342457918063882,
      "loss": 13.6281,
      "step": 416
    },
    {
      "epoch": 0.19124054116028433,
      "grad_norm": 0.7962077154227134,
      "learning_rate": 0.0009338769823089853,
      "loss": 13.532,
      "step": 417
    },
    {
      "epoch": 0.19169915157074066,
      "grad_norm": 0.6675853796794086,
      "learning_rate": 0.0009335072146112648,
      "loss": 13.9406,
      "step": 418
    },
    {
      "epoch": 0.19215776198119697,
      "grad_norm": 0.6948896442286366,
      "learning_rate": 0.0009331364895298444,
      "loss": 13.6012,
      "step": 419
    },
    {
      "epoch": 0.1926163723916533,
      "grad_norm": 0.6649506435909566,
      "learning_rate": 0.0009327648078834559,
      "loss": 13.9464,
      "step": 420
    },
    {
      "epoch": 0.1930749828021096,
      "grad_norm": 0.7029975679606778,
      "learning_rate": 0.0009323921704929434,
      "loss": 13.6918,
      "step": 421
    },
    {
      "epoch": 0.1935335932125659,
      "grad_norm": 0.7496591348467734,
      "learning_rate": 0.0009320185781812623,
      "loss": 13.7281,
      "step": 422
    },
    {
      "epoch": 0.19399220362302225,
      "grad_norm": 0.7488164939136274,
      "learning_rate": 0.0009316440317734762,
      "loss": 13.8125,
      "step": 423
    },
    {
      "epoch": 0.19445081403347855,
      "grad_norm": 0.7533895739991607,
      "learning_rate": 0.0009312685320967565,
      "loss": 13.9622,
      "step": 424
    },
    {
      "epoch": 0.1949094244439349,
      "grad_norm": 0.7518070085176143,
      "learning_rate": 0.0009308920799803793,
      "loss": 13.5534,
      "step": 425
    },
    {
      "epoch": 0.1953680348543912,
      "grad_norm": 0.6522963475947716,
      "learning_rate": 0.0009305146762557246,
      "loss": 13.6944,
      "step": 426
    },
    {
      "epoch": 0.1958266452648475,
      "grad_norm": 0.6555244619480457,
      "learning_rate": 0.0009301363217562736,
      "loss": 13.6738,
      "step": 427
    },
    {
      "epoch": 0.19628525567530383,
      "grad_norm": 0.7060593668200922,
      "learning_rate": 0.0009297570173176074,
      "loss": 13.8414,
      "step": 428
    },
    {
      "epoch": 0.19674386608576014,
      "grad_norm": 0.6826728871750495,
      "learning_rate": 0.000929376763777405,
      "loss": 13.5807,
      "step": 429
    },
    {
      "epoch": 0.19720247649621647,
      "grad_norm": 0.6466739760428631,
      "learning_rate": 0.0009289955619754413,
      "loss": 13.9544,
      "step": 430
    },
    {
      "epoch": 0.19766108690667278,
      "grad_norm": 0.6177075884850658,
      "learning_rate": 0.0009286134127535859,
      "loss": 13.7909,
      "step": 431
    },
    {
      "epoch": 0.1981196973171291,
      "grad_norm": 0.6093654650926448,
      "learning_rate": 0.0009282303169558,
      "loss": 13.5297,
      "step": 432
    },
    {
      "epoch": 0.19857830772758542,
      "grad_norm": 0.7938452797801463,
      "learning_rate": 0.0009278462754281359,
      "loss": 13.6303,
      "step": 433
    },
    {
      "epoch": 0.19903691813804172,
      "grad_norm": 0.6899900223543579,
      "learning_rate": 0.0009274612890187342,
      "loss": 13.6961,
      "step": 434
    },
    {
      "epoch": 0.19949552854849806,
      "grad_norm": 0.6182954665116861,
      "learning_rate": 0.0009270753585778222,
      "loss": 13.7474,
      "step": 435
    },
    {
      "epoch": 0.19995413895895436,
      "grad_norm": 0.7742545644899762,
      "learning_rate": 0.0009266884849577124,
      "loss": 13.4837,
      "step": 436
    },
    {
      "epoch": 0.2004127493694107,
      "grad_norm": 0.6447472929113525,
      "learning_rate": 0.0009263006690127998,
      "loss": 13.5741,
      "step": 437
    },
    {
      "epoch": 0.200871359779867,
      "grad_norm": 0.8062383402537613,
      "learning_rate": 0.0009259119115995609,
      "loss": 13.6539,
      "step": 438
    },
    {
      "epoch": 0.2013299701903233,
      "grad_norm": 0.7762155096769764,
      "learning_rate": 0.0009255222135765511,
      "loss": 13.9636,
      "step": 439
    },
    {
      "epoch": 0.20178858060077964,
      "grad_norm": 0.6718246190177715,
      "learning_rate": 0.0009251315758044032,
      "loss": 13.6218,
      "step": 440
    },
    {
      "epoch": 0.20224719101123595,
      "grad_norm": 0.7483793141229315,
      "learning_rate": 0.0009247399991458255,
      "loss": 13.5872,
      "step": 441
    },
    {
      "epoch": 0.20270580142169228,
      "grad_norm": 0.6411411997664211,
      "learning_rate": 0.0009243474844655996,
      "loss": 13.4543,
      "step": 442
    },
    {
      "epoch": 0.2031644118321486,
      "grad_norm": 0.7513148230552935,
      "learning_rate": 0.0009239540326305791,
      "loss": 13.7521,
      "step": 443
    },
    {
      "epoch": 0.2036230222426049,
      "grad_norm": 0.6581938580927522,
      "learning_rate": 0.0009235596445096864,
      "loss": 13.7927,
      "step": 444
    },
    {
      "epoch": 0.20408163265306123,
      "grad_norm": 0.7531843295980462,
      "learning_rate": 0.0009231643209739127,
      "loss": 13.7082,
      "step": 445
    },
    {
      "epoch": 0.20454024306351754,
      "grad_norm": 0.672989734890873,
      "learning_rate": 0.0009227680628963145,
      "loss": 13.6258,
      "step": 446
    },
    {
      "epoch": 0.20499885347397387,
      "grad_norm": 0.7206105037683388,
      "learning_rate": 0.000922370871152012,
      "loss": 13.3639,
      "step": 447
    },
    {
      "epoch": 0.20545746388443017,
      "grad_norm": 0.7588739807891867,
      "learning_rate": 0.0009219727466181877,
      "loss": 13.6263,
      "step": 448
    },
    {
      "epoch": 0.20591607429488648,
      "grad_norm": 0.6609851536328583,
      "learning_rate": 0.0009215736901740841,
      "loss": 13.689,
      "step": 449
    },
    {
      "epoch": 0.20637468470534281,
      "grad_norm": 0.7213218118441327,
      "learning_rate": 0.0009211737027010016,
      "loss": 13.7603,
      "step": 450
    },
    {
      "epoch": 0.20683329511579912,
      "grad_norm": 0.714250362913897,
      "learning_rate": 0.0009207727850822971,
      "loss": 13.5682,
      "step": 451
    },
    {
      "epoch": 0.20729190552625545,
      "grad_norm": 0.8137804954208604,
      "learning_rate": 0.0009203709382033814,
      "loss": 13.6261,
      "step": 452
    },
    {
      "epoch": 0.20775051593671176,
      "grad_norm": 0.7547471459978163,
      "learning_rate": 0.0009199681629517173,
      "loss": 13.7785,
      "step": 453
    },
    {
      "epoch": 0.2082091263471681,
      "grad_norm": 0.7336285988083571,
      "learning_rate": 0.0009195644602168184,
      "loss": 13.5282,
      "step": 454
    },
    {
      "epoch": 0.2086677367576244,
      "grad_norm": 0.7092962922032829,
      "learning_rate": 0.0009191598308902464,
      "loss": 13.8466,
      "step": 455
    },
    {
      "epoch": 0.2091263471680807,
      "grad_norm": 0.6469613801225429,
      "learning_rate": 0.0009187542758656091,
      "loss": 13.9123,
      "step": 456
    },
    {
      "epoch": 0.20958495757853704,
      "grad_norm": 0.7313336467305243,
      "learning_rate": 0.0009183477960385591,
      "loss": 13.8772,
      "step": 457
    },
    {
      "epoch": 0.21004356798899335,
      "grad_norm": 0.6641471212830199,
      "learning_rate": 0.0009179403923067912,
      "loss": 13.7466,
      "step": 458
    },
    {
      "epoch": 0.21050217839944968,
      "grad_norm": 0.6782948069029384,
      "learning_rate": 0.0009175320655700406,
      "loss": 13.7337,
      "step": 459
    },
    {
      "epoch": 0.21096078880990599,
      "grad_norm": 0.7477055691054985,
      "learning_rate": 0.0009171228167300805,
      "loss": 13.7618,
      "step": 460
    },
    {
      "epoch": 0.2114193992203623,
      "grad_norm": 0.6499448052312934,
      "learning_rate": 0.0009167126466907215,
      "loss": 13.709,
      "step": 461
    },
    {
      "epoch": 0.21187800963081863,
      "grad_norm": 0.6856039039143623,
      "learning_rate": 0.0009163015563578074,
      "loss": 13.8545,
      "step": 462
    },
    {
      "epoch": 0.21233662004127493,
      "grad_norm": 0.691046770174566,
      "learning_rate": 0.0009158895466392158,
      "loss": 13.5986,
      "step": 463
    },
    {
      "epoch": 0.21279523045173127,
      "grad_norm": 0.7736371082599797,
      "learning_rate": 0.0009154766184448535,
      "loss": 13.8044,
      "step": 464
    },
    {
      "epoch": 0.21325384086218757,
      "grad_norm": 0.6160318090472421,
      "learning_rate": 0.0009150627726866568,
      "loss": 13.6751,
      "step": 465
    },
    {
      "epoch": 0.21371245127264388,
      "grad_norm": 0.7039824955854516,
      "learning_rate": 0.000914648010278587,
      "loss": 14.0128,
      "step": 466
    },
    {
      "epoch": 0.2141710616831002,
      "grad_norm": 0.7095799942495351,
      "learning_rate": 0.0009142323321366315,
      "loss": 13.813,
      "step": 467
    },
    {
      "epoch": 0.21462967209355652,
      "grad_norm": 0.6898279542237786,
      "learning_rate": 0.0009138157391787986,
      "loss": 13.7297,
      "step": 468
    },
    {
      "epoch": 0.21508828250401285,
      "grad_norm": 0.6415727632843697,
      "learning_rate": 0.0009133982323251177,
      "loss": 13.7562,
      "step": 469
    },
    {
      "epoch": 0.21554689291446916,
      "grad_norm": 0.655000047602952,
      "learning_rate": 0.0009129798124976365,
      "loss": 13.6725,
      "step": 470
    },
    {
      "epoch": 0.21600550332492546,
      "grad_norm": 0.6865759713461063,
      "learning_rate": 0.0009125604806204187,
      "loss": 13.7492,
      "step": 471
    },
    {
      "epoch": 0.2164641137353818,
      "grad_norm": 0.6451326687211513,
      "learning_rate": 0.0009121402376195421,
      "loss": 13.8243,
      "step": 472
    },
    {
      "epoch": 0.2169227241458381,
      "grad_norm": 0.6899489306520814,
      "learning_rate": 0.0009117190844230972,
      "loss": 13.707,
      "step": 473
    },
    {
      "epoch": 0.21738133455629444,
      "grad_norm": 0.6271103880075571,
      "learning_rate": 0.0009112970219611841,
      "loss": 13.7165,
      "step": 474
    },
    {
      "epoch": 0.21783994496675074,
      "grad_norm": 0.7342099875611011,
      "learning_rate": 0.0009108740511659115,
      "loss": 13.656,
      "step": 475
    },
    {
      "epoch": 0.21829855537720708,
      "grad_norm": 0.6386743437576409,
      "learning_rate": 0.0009104501729713935,
      "loss": 13.6723,
      "step": 476
    },
    {
      "epoch": 0.21875716578766338,
      "grad_norm": 0.6611995768961296,
      "learning_rate": 0.0009100253883137488,
      "loss": 13.7218,
      "step": 477
    },
    {
      "epoch": 0.2192157761981197,
      "grad_norm": 0.6592332260147772,
      "learning_rate": 0.0009095996981310974,
      "loss": 13.6506,
      "step": 478
    },
    {
      "epoch": 0.21967438660857602,
      "grad_norm": 0.6016227595999705,
      "learning_rate": 0.0009091731033635596,
      "loss": 13.7555,
      "step": 479
    },
    {
      "epoch": 0.22013299701903233,
      "grad_norm": 0.6287922415256326,
      "learning_rate": 0.0009087456049532529,
      "loss": 13.6971,
      "step": 480
    },
    {
      "epoch": 0.22059160742948866,
      "grad_norm": 0.6377178198389865,
      "learning_rate": 0.0009083172038442914,
      "loss": 13.5947,
      "step": 481
    },
    {
      "epoch": 0.22105021783994497,
      "grad_norm": 0.6917570232569092,
      "learning_rate": 0.0009078879009827817,
      "loss": 13.825,
      "step": 482
    },
    {
      "epoch": 0.22150882825040127,
      "grad_norm": 0.6698287580797633,
      "learning_rate": 0.0009074576973168223,
      "loss": 13.5707,
      "step": 483
    },
    {
      "epoch": 0.2219674386608576,
      "grad_norm": 0.7390184166021554,
      "learning_rate": 0.0009070265937965015,
      "loss": 13.9857,
      "step": 484
    },
    {
      "epoch": 0.2224260490713139,
      "grad_norm": 0.6409290638111081,
      "learning_rate": 0.0009065945913738942,
      "loss": 13.5775,
      "step": 485
    },
    {
      "epoch": 0.22288465948177025,
      "grad_norm": 0.5843140001161706,
      "learning_rate": 0.0009061616910030609,
      "loss": 13.5918,
      "step": 486
    },
    {
      "epoch": 0.22334326989222655,
      "grad_norm": 0.5929875330893412,
      "learning_rate": 0.0009057278936400453,
      "loss": 13.9574,
      "step": 487
    },
    {
      "epoch": 0.22380188030268286,
      "grad_norm": 0.680916610042174,
      "learning_rate": 0.0009052932002428715,
      "loss": 13.6332,
      "step": 488
    },
    {
      "epoch": 0.2242604907131392,
      "grad_norm": 0.5937848549497331,
      "learning_rate": 0.0009048576117715435,
      "loss": 13.5422,
      "step": 489
    },
    {
      "epoch": 0.2247191011235955,
      "grad_norm": 0.7159720243153749,
      "learning_rate": 0.0009044211291880407,
      "loss": 13.7053,
      "step": 490
    },
    {
      "epoch": 0.22517771153405183,
      "grad_norm": 0.612111143539907,
      "learning_rate": 0.000903983753456318,
      "loss": 13.6437,
      "step": 491
    },
    {
      "epoch": 0.22563632194450814,
      "grad_norm": 0.7072300081635284,
      "learning_rate": 0.0009035454855423026,
      "loss": 13.6123,
      "step": 492
    },
    {
      "epoch": 0.22609493235496445,
      "grad_norm": 0.6127392165861419,
      "learning_rate": 0.0009031063264138922,
      "loss": 13.6218,
      "step": 493
    },
    {
      "epoch": 0.22655354276542078,
      "grad_norm": 0.7120893593547446,
      "learning_rate": 0.0009026662770409522,
      "loss": 13.7412,
      "step": 494
    },
    {
      "epoch": 0.22701215317587708,
      "grad_norm": 0.6316844188810694,
      "learning_rate": 0.0009022253383953147,
      "loss": 13.482,
      "step": 495
    },
    {
      "epoch": 0.22747076358633342,
      "grad_norm": 0.6626190183508875,
      "learning_rate": 0.0009017835114507753,
      "loss": 13.6177,
      "step": 496
    },
    {
      "epoch": 0.22792937399678972,
      "grad_norm": 0.6440624634667765,
      "learning_rate": 0.0009013407971830914,
      "loss": 13.6401,
      "step": 497
    },
    {
      "epoch": 0.22838798440724603,
      "grad_norm": 0.6763998026644108,
      "learning_rate": 0.0009008971965699801,
      "loss": 13.7798,
      "step": 498
    },
    {
      "epoch": 0.22884659481770236,
      "grad_norm": 0.6284068513433705,
      "learning_rate": 0.0009004527105911163,
      "loss": 13.8565,
      "step": 499
    },
    {
      "epoch": 0.22930520522815867,
      "grad_norm": 0.6367385444679768,
      "learning_rate": 0.0009000073402281295,
      "loss": 13.6682,
      "step": 500
    },
    {
      "epoch": 0.229763815638615,
      "grad_norm": 0.658351668835815,
      "learning_rate": 0.0008995610864646028,
      "loss": 13.7972,
      "step": 501
    },
    {
      "epoch": 0.2302224260490713,
      "grad_norm": 0.6291763994542268,
      "learning_rate": 0.0008991139502860703,
      "loss": 13.5774,
      "step": 502
    },
    {
      "epoch": 0.23068103645952764,
      "grad_norm": 0.6800730936509578,
      "learning_rate": 0.0008986659326800146,
      "loss": 13.8797,
      "step": 503
    },
    {
      "epoch": 0.23113964686998395,
      "grad_norm": 0.6619843333953348,
      "learning_rate": 0.0008982170346358651,
      "loss": 13.6751,
      "step": 504
    },
    {
      "epoch": 0.23159825728044026,
      "grad_norm": 0.7360673491336337,
      "learning_rate": 0.0008977672571449956,
      "loss": 13.8151,
      "step": 505
    },
    {
      "epoch": 0.2320568676908966,
      "grad_norm": 0.5998995918118901,
      "learning_rate": 0.0008973166012007217,
      "loss": 13.657,
      "step": 506
    },
    {
      "epoch": 0.2325154781013529,
      "grad_norm": 0.6658977074847637,
      "learning_rate": 0.0008968650677982998,
      "loss": 13.4804,
      "step": 507
    },
    {
      "epoch": 0.23297408851180923,
      "grad_norm": 0.6802537634455459,
      "learning_rate": 0.0008964126579349236,
      "loss": 13.5963,
      "step": 508
    },
    {
      "epoch": 0.23343269892226554,
      "grad_norm": 0.5963146693862964,
      "learning_rate": 0.0008959593726097226,
      "loss": 13.5768,
      "step": 509
    },
    {
      "epoch": 0.23389130933272184,
      "grad_norm": 0.7144167893974194,
      "learning_rate": 0.0008955052128237596,
      "loss": 13.9008,
      "step": 510
    },
    {
      "epoch": 0.23434991974317818,
      "grad_norm": 0.662122582132281,
      "learning_rate": 0.0008950501795800288,
      "loss": 13.5907,
      "step": 511
    },
    {
      "epoch": 0.23480853015363448,
      "grad_norm": 0.7262252738350718,
      "learning_rate": 0.0008945942738834532,
      "loss": 13.6087,
      "step": 512
    },
    {
      "epoch": 0.23526714056409082,
      "grad_norm": 0.655602943466857,
      "learning_rate": 0.0008941374967408826,
      "loss": 13.6452,
      "step": 513
    },
    {
      "epoch": 0.23572575097454712,
      "grad_norm": 0.6988704042774838,
      "learning_rate": 0.0008936798491610916,
      "loss": 13.7525,
      "step": 514
    },
    {
      "epoch": 0.23618436138500343,
      "grad_norm": 0.678299470400607,
      "learning_rate": 0.0008932213321547768,
      "loss": 13.581,
      "step": 515
    },
    {
      "epoch": 0.23664297179545976,
      "grad_norm": 0.7342121728127388,
      "learning_rate": 0.0008927619467345554,
      "loss": 13.7322,
      "step": 516
    },
    {
      "epoch": 0.23710158220591607,
      "grad_norm": 0.6851681137770528,
      "learning_rate": 0.0008923016939149615,
      "loss": 13.4691,
      "step": 517
    },
    {
      "epoch": 0.2375601926163724,
      "grad_norm": 0.6299800215607042,
      "learning_rate": 0.0008918405747124458,
      "loss": 13.7826,
      "step": 518
    },
    {
      "epoch": 0.2380188030268287,
      "grad_norm": 0.6713296661411425,
      "learning_rate": 0.0008913785901453721,
      "loss": 13.6056,
      "step": 519
    },
    {
      "epoch": 0.238477413437285,
      "grad_norm": 0.6266097438628441,
      "learning_rate": 0.000890915741234015,
      "loss": 13.6176,
      "step": 520
    },
    {
      "epoch": 0.23893602384774135,
      "grad_norm": 0.878864051017724,
      "learning_rate": 0.0008904520290005582,
      "loss": 13.7706,
      "step": 521
    },
    {
      "epoch": 0.23939463425819765,
      "grad_norm": 0.6322985371415266,
      "learning_rate": 0.000889987454469092,
      "loss": 13.523,
      "step": 522
    },
    {
      "epoch": 0.239853244668654,
      "grad_norm": 0.6551158828607494,
      "learning_rate": 0.0008895220186656111,
      "loss": 13.6119,
      "step": 523
    },
    {
      "epoch": 0.2403118550791103,
      "grad_norm": 0.6531317314686287,
      "learning_rate": 0.0008890557226180122,
      "loss": 13.6342,
      "step": 524
    },
    {
      "epoch": 0.24077046548956663,
      "grad_norm": 0.6826997824322452,
      "learning_rate": 0.0008885885673560921,
      "loss": 13.691,
      "step": 525
    },
    {
      "epoch": 0.24122907590002293,
      "grad_norm": 0.7285397515379941,
      "learning_rate": 0.0008881205539115444,
      "loss": 13.7707,
      "step": 526
    },
    {
      "epoch": 0.24168768631047924,
      "grad_norm": 0.6198438483156526,
      "learning_rate": 0.0008876516833179589,
      "loss": 13.6405,
      "step": 527
    },
    {
      "epoch": 0.24214629672093557,
      "grad_norm": 0.6288998245124259,
      "learning_rate": 0.0008871819566108177,
      "loss": 13.5089,
      "step": 528
    },
    {
      "epoch": 0.24260490713139188,
      "grad_norm": 0.6416550458683039,
      "learning_rate": 0.000886711374827494,
      "loss": 13.4619,
      "step": 529
    },
    {
      "epoch": 0.2430635175418482,
      "grad_norm": 0.7052161891920301,
      "learning_rate": 0.0008862399390072491,
      "loss": 13.728,
      "step": 530
    },
    {
      "epoch": 0.24352212795230452,
      "grad_norm": 0.6780196659208899,
      "learning_rate": 0.0008857676501912305,
      "loss": 13.6975,
      "step": 531
    },
    {
      "epoch": 0.24398073836276082,
      "grad_norm": 0.6510701549575431,
      "learning_rate": 0.0008852945094224697,
      "loss": 13.6855,
      "step": 532
    },
    {
      "epoch": 0.24443934877321716,
      "grad_norm": 0.629676887577189,
      "learning_rate": 0.0008848205177458795,
      "loss": 13.4675,
      "step": 533
    },
    {
      "epoch": 0.24489795918367346,
      "grad_norm": 0.7210371602146268,
      "learning_rate": 0.0008843456762082518,
      "loss": 13.7976,
      "step": 534
    },
    {
      "epoch": 0.2453565695941298,
      "grad_norm": 0.618703977941099,
      "learning_rate": 0.0008838699858582557,
      "loss": 13.7729,
      "step": 535
    },
    {
      "epoch": 0.2458151800045861,
      "grad_norm": 0.722472499348574,
      "learning_rate": 0.0008833934477464347,
      "loss": 13.6692,
      "step": 536
    },
    {
      "epoch": 0.2462737904150424,
      "grad_norm": 0.6541896120087248,
      "learning_rate": 0.0008829160629252045,
      "loss": 13.7731,
      "step": 537
    },
    {
      "epoch": 0.24673240082549874,
      "grad_norm": 0.6753360833825547,
      "learning_rate": 0.0008824378324488509,
      "loss": 13.8366,
      "step": 538
    },
    {
      "epoch": 0.24719101123595505,
      "grad_norm": 0.5964237054640114,
      "learning_rate": 0.0008819587573735268,
      "loss": 13.5281,
      "step": 539
    },
    {
      "epoch": 0.24764962164641138,
      "grad_norm": 0.8445208128038056,
      "learning_rate": 0.0008814788387572513,
      "loss": 13.6482,
      "step": 540
    },
    {
      "epoch": 0.2481082320568677,
      "grad_norm": 0.6036155407333608,
      "learning_rate": 0.0008809980776599053,
      "loss": 13.6504,
      "step": 541
    },
    {
      "epoch": 0.248566842467324,
      "grad_norm": 0.6235654970298661,
      "learning_rate": 0.0008805164751432312,
      "loss": 13.6061,
      "step": 542
    },
    {
      "epoch": 0.24902545287778033,
      "grad_norm": 0.5710343628616998,
      "learning_rate": 0.0008800340322708292,
      "loss": 13.5018,
      "step": 543
    },
    {
      "epoch": 0.24948406328823663,
      "grad_norm": 0.6349562083754354,
      "learning_rate": 0.0008795507501081555,
      "loss": 13.696,
      "step": 544
    },
    {
      "epoch": 0.24994267369869297,
      "grad_norm": 0.6653671659629751,
      "learning_rate": 0.0008790666297225196,
      "loss": 13.7157,
      "step": 545
    },
    {
      "epoch": 0.2504012841091493,
      "grad_norm": 0.8023521715407523,
      "learning_rate": 0.0008785816721830829,
      "loss": 13.8227,
      "step": 546
    },
    {
      "epoch": 0.2508598945196056,
      "grad_norm": 0.8534185836672791,
      "learning_rate": 0.0008780958785608546,
      "loss": 13.7622,
      "step": 547
    },
    {
      "epoch": 0.2513185049300619,
      "grad_norm": 0.6629753297936218,
      "learning_rate": 0.0008776092499286912,
      "loss": 13.5608,
      "step": 548
    },
    {
      "epoch": 0.2517771153405182,
      "grad_norm": 0.9649920122813237,
      "learning_rate": 0.0008771217873612929,
      "loss": 13.7941,
      "step": 549
    },
    {
      "epoch": 0.2522357257509745,
      "grad_norm": 0.6137292426883455,
      "learning_rate": 0.0008766334919352017,
      "loss": 13.4545,
      "step": 550
    },
    {
      "epoch": 0.2526943361614309,
      "grad_norm": 0.6240775248320325,
      "learning_rate": 0.0008761443647287987,
      "loss": 13.7897,
      "step": 551
    },
    {
      "epoch": 0.2531529465718872,
      "grad_norm": 0.6309118710368381,
      "learning_rate": 0.0008756544068223026,
      "loss": 13.7788,
      "step": 552
    },
    {
      "epoch": 0.2536115569823435,
      "grad_norm": 0.6298798407624988,
      "learning_rate": 0.0008751636192977659,
      "loss": 13.8167,
      "step": 553
    },
    {
      "epoch": 0.2540701673927998,
      "grad_norm": 0.6122974253867214,
      "learning_rate": 0.0008746720032390737,
      "loss": 13.9606,
      "step": 554
    },
    {
      "epoch": 0.2545287778032561,
      "grad_norm": 0.5954346243210656,
      "learning_rate": 0.0008741795597319408,
      "loss": 13.5906,
      "step": 555
    },
    {
      "epoch": 0.2549873882137125,
      "grad_norm": 0.6179865402157853,
      "learning_rate": 0.0008736862898639095,
      "loss": 13.8221,
      "step": 556
    },
    {
      "epoch": 0.2554459986241688,
      "grad_norm": 0.6081900854081858,
      "learning_rate": 0.0008731921947243468,
      "loss": 13.656,
      "step": 557
    },
    {
      "epoch": 0.2559046090346251,
      "grad_norm": 0.6078959515072513,
      "learning_rate": 0.0008726972754044427,
      "loss": 13.5488,
      "step": 558
    },
    {
      "epoch": 0.2563632194450814,
      "grad_norm": 0.5915721889324325,
      "learning_rate": 0.0008722015329972069,
      "loss": 13.5147,
      "step": 559
    },
    {
      "epoch": 0.2568218298555377,
      "grad_norm": 0.6257405211705518,
      "learning_rate": 0.0008717049685974672,
      "loss": 13.7034,
      "step": 560
    },
    {
      "epoch": 0.25728044026599406,
      "grad_norm": 0.6845514026865301,
      "learning_rate": 0.0008712075833018665,
      "loss": 13.8102,
      "step": 561
    },
    {
      "epoch": 0.25773905067645037,
      "grad_norm": 0.687759680567341,
      "learning_rate": 0.0008707093782088608,
      "loss": 13.5352,
      "step": 562
    },
    {
      "epoch": 0.25819766108690667,
      "grad_norm": 0.6224412280135145,
      "learning_rate": 0.0008702103544187167,
      "loss": 13.782,
      "step": 563
    },
    {
      "epoch": 0.258656271497363,
      "grad_norm": 0.718787029727441,
      "learning_rate": 0.0008697105130335085,
      "loss": 13.696,
      "step": 564
    },
    {
      "epoch": 0.2591148819078193,
      "grad_norm": 0.6808482923716414,
      "learning_rate": 0.0008692098551571164,
      "loss": 13.6541,
      "step": 565
    },
    {
      "epoch": 0.25957349231827564,
      "grad_norm": 0.5731827838749779,
      "learning_rate": 0.0008687083818952235,
      "loss": 13.5457,
      "step": 566
    },
    {
      "epoch": 0.26003210272873195,
      "grad_norm": 0.6121208817758225,
      "learning_rate": 0.0008682060943553143,
      "loss": 13.7909,
      "step": 567
    },
    {
      "epoch": 0.26049071313918826,
      "grad_norm": 0.596907667319619,
      "learning_rate": 0.0008677029936466707,
      "loss": 13.6284,
      "step": 568
    },
    {
      "epoch": 0.26094932354964456,
      "grad_norm": 0.5814332656377997,
      "learning_rate": 0.0008671990808803711,
      "loss": 13.9783,
      "step": 569
    },
    {
      "epoch": 0.26140793396010087,
      "grad_norm": 0.6452524431011539,
      "learning_rate": 0.0008666943571692871,
      "loss": 13.6327,
      "step": 570
    },
    {
      "epoch": 0.26186654437055723,
      "grad_norm": 0.5876059518186987,
      "learning_rate": 0.0008661888236280813,
      "loss": 13.8199,
      "step": 571
    },
    {
      "epoch": 0.26232515478101354,
      "grad_norm": 0.6049191537748061,
      "learning_rate": 0.0008656824813732045,
      "loss": 13.5792,
      "step": 572
    },
    {
      "epoch": 0.26278376519146984,
      "grad_norm": 0.6068281429656994,
      "learning_rate": 0.000865175331522894,
      "loss": 13.3707,
      "step": 573
    },
    {
      "epoch": 0.26324237560192615,
      "grad_norm": 0.5800543296152699,
      "learning_rate": 0.0008646673751971703,
      "loss": 13.4994,
      "step": 574
    },
    {
      "epoch": 0.26370098601238245,
      "grad_norm": 0.6862540028354155,
      "learning_rate": 0.000864158613517835,
      "loss": 13.6945,
      "step": 575
    },
    {
      "epoch": 0.2641595964228388,
      "grad_norm": 0.6845859938687919,
      "learning_rate": 0.0008636490476084681,
      "loss": 13.6838,
      "step": 576
    },
    {
      "epoch": 0.2646182068332951,
      "grad_norm": 0.767424215533394,
      "learning_rate": 0.0008631386785944264,
      "loss": 13.9052,
      "step": 577
    },
    {
      "epoch": 0.26507681724375143,
      "grad_norm": 0.6554578011533805,
      "learning_rate": 0.0008626275076028397,
      "loss": 13.6466,
      "step": 578
    },
    {
      "epoch": 0.26553542765420773,
      "grad_norm": 0.6110261069766717,
      "learning_rate": 0.0008621155357626091,
      "loss": 13.6741,
      "step": 579
    },
    {
      "epoch": 0.2659940380646641,
      "grad_norm": 0.7882923523000077,
      "learning_rate": 0.0008616027642044042,
      "loss": 13.7387,
      "step": 580
    },
    {
      "epoch": 0.2664526484751204,
      "grad_norm": 0.6836638894882718,
      "learning_rate": 0.000861089194060661,
      "loss": 13.5888,
      "step": 581
    },
    {
      "epoch": 0.2669112588855767,
      "grad_norm": 0.7242907131241766,
      "learning_rate": 0.000860574826465579,
      "loss": 13.5364,
      "step": 582
    },
    {
      "epoch": 0.267369869296033,
      "grad_norm": 0.5991929999371074,
      "learning_rate": 0.0008600596625551191,
      "loss": 13.6602,
      "step": 583
    },
    {
      "epoch": 0.2678284797064893,
      "grad_norm": 0.7010936530719747,
      "learning_rate": 0.0008595437034670006,
      "loss": 13.6379,
      "step": 584
    },
    {
      "epoch": 0.2682870901169457,
      "grad_norm": 0.6476798852477783,
      "learning_rate": 0.0008590269503406985,
      "loss": 13.6192,
      "step": 585
    },
    {
      "epoch": 0.268745700527402,
      "grad_norm": 0.6510951000777048,
      "learning_rate": 0.0008585094043174423,
      "loss": 13.7727,
      "step": 586
    },
    {
      "epoch": 0.2692043109378583,
      "grad_norm": 0.6745972062517068,
      "learning_rate": 0.0008579910665402118,
      "loss": 13.7817,
      "step": 587
    },
    {
      "epoch": 0.2696629213483146,
      "grad_norm": 0.5895725388166738,
      "learning_rate": 0.000857471938153736,
      "loss": 13.6156,
      "step": 588
    },
    {
      "epoch": 0.2701215317587709,
      "grad_norm": 0.6811975877700501,
      "learning_rate": 0.0008569520203044892,
      "loss": 13.493,
      "step": 589
    },
    {
      "epoch": 0.27058014216922727,
      "grad_norm": 0.7110031157909367,
      "learning_rate": 0.0008564313141406901,
      "loss": 13.3585,
      "step": 590
    },
    {
      "epoch": 0.2710387525796836,
      "grad_norm": 0.6555900199827576,
      "learning_rate": 0.0008559098208122973,
      "loss": 13.6024,
      "step": 591
    },
    {
      "epoch": 0.2714973629901399,
      "grad_norm": 0.60172440642139,
      "learning_rate": 0.0008553875414710089,
      "loss": 13.6723,
      "step": 592
    },
    {
      "epoch": 0.2719559734005962,
      "grad_norm": 0.7309731909775504,
      "learning_rate": 0.0008548644772702579,
      "loss": 13.5808,
      "step": 593
    },
    {
      "epoch": 0.2724145838110525,
      "grad_norm": 0.6871355185162522,
      "learning_rate": 0.0008543406293652116,
      "loss": 13.5705,
      "step": 594
    },
    {
      "epoch": 0.27287319422150885,
      "grad_norm": 0.5915383318676095,
      "learning_rate": 0.0008538159989127671,
      "loss": 13.6347,
      "step": 595
    },
    {
      "epoch": 0.27333180463196516,
      "grad_norm": 0.675740096735486,
      "learning_rate": 0.0008532905870715505,
      "loss": 13.7269,
      "step": 596
    },
    {
      "epoch": 0.27379041504242146,
      "grad_norm": 0.6353994429301371,
      "learning_rate": 0.0008527643950019131,
      "loss": 13.5654,
      "step": 597
    },
    {
      "epoch": 0.27424902545287777,
      "grad_norm": 0.7190003733210985,
      "learning_rate": 0.0008522374238659296,
      "loss": 13.575,
      "step": 598
    },
    {
      "epoch": 0.2747076358633341,
      "grad_norm": 0.639991501908558,
      "learning_rate": 0.0008517096748273951,
      "loss": 13.6787,
      "step": 599
    },
    {
      "epoch": 0.27516624627379044,
      "grad_norm": 0.6317978490760873,
      "learning_rate": 0.0008511811490518227,
      "loss": 13.8091,
      "step": 600
    },
    {
      "epoch": 0.27562485668424674,
      "grad_norm": 0.6354003007467041,
      "learning_rate": 0.0008506518477064405,
      "loss": 13.5068,
      "step": 601
    },
    {
      "epoch": 0.27608346709470305,
      "grad_norm": 0.7067078797992797,
      "learning_rate": 0.0008501217719601903,
      "loss": 13.7871,
      "step": 602
    },
    {
      "epoch": 0.27654207750515936,
      "grad_norm": 0.6270814679593057,
      "learning_rate": 0.0008495909229837233,
      "loss": 13.7159,
      "step": 603
    },
    {
      "epoch": 0.27700068791561566,
      "grad_norm": 0.6717309492405003,
      "learning_rate": 0.000849059301949399,
      "loss": 13.6266,
      "step": 604
    },
    {
      "epoch": 0.277459298326072,
      "grad_norm": 0.7058341465385499,
      "learning_rate": 0.0008485269100312812,
      "loss": 13.7955,
      "step": 605
    },
    {
      "epoch": 0.27791790873652833,
      "grad_norm": 0.6421377737047564,
      "learning_rate": 0.0008479937484051368,
      "loss": 13.6673,
      "step": 606
    },
    {
      "epoch": 0.27837651914698464,
      "grad_norm": 0.6662488029978719,
      "learning_rate": 0.0008474598182484323,
      "loss": 13.6918,
      "step": 607
    },
    {
      "epoch": 0.27883512955744094,
      "grad_norm": 0.5912044098279146,
      "learning_rate": 0.0008469251207403317,
      "loss": 13.8377,
      "step": 608
    },
    {
      "epoch": 0.27929373996789725,
      "grad_norm": 0.7120185485609593,
      "learning_rate": 0.0008463896570616934,
      "loss": 13.6664,
      "step": 609
    },
    {
      "epoch": 0.2797523503783536,
      "grad_norm": 0.5833079357912311,
      "learning_rate": 0.0008458534283950678,
      "loss": 13.6679,
      "step": 610
    },
    {
      "epoch": 0.2802109607888099,
      "grad_norm": 0.6897782182024569,
      "learning_rate": 0.0008453164359246952,
      "loss": 13.5079,
      "step": 611
    },
    {
      "epoch": 0.2806695711992662,
      "grad_norm": 0.6049081083285837,
      "learning_rate": 0.0008447786808365022,
      "loss": 13.7958,
      "step": 612
    },
    {
      "epoch": 0.2811281816097225,
      "grad_norm": 0.6157651218424084,
      "learning_rate": 0.0008442401643181,
      "loss": 13.6067,
      "step": 613
    },
    {
      "epoch": 0.28158679202017883,
      "grad_norm": 0.7266301814467849,
      "learning_rate": 0.0008437008875587811,
      "loss": 13.5615,
      "step": 614
    },
    {
      "epoch": 0.2820454024306352,
      "grad_norm": 0.6000204936826586,
      "learning_rate": 0.0008431608517495171,
      "loss": 13.6604,
      "step": 615
    },
    {
      "epoch": 0.2825040128410915,
      "grad_norm": 0.7906322017985308,
      "learning_rate": 0.0008426200580829561,
      "loss": 13.5502,
      "step": 616
    },
    {
      "epoch": 0.2829626232515478,
      "grad_norm": 0.6140895944452744,
      "learning_rate": 0.0008420785077534195,
      "loss": 13.6585,
      "step": 617
    },
    {
      "epoch": 0.2834212336620041,
      "grad_norm": 0.7350724138141813,
      "learning_rate": 0.0008415362019569001,
      "loss": 13.5623,
      "step": 618
    },
    {
      "epoch": 0.2838798440724604,
      "grad_norm": 0.6861292699830437,
      "learning_rate": 0.0008409931418910591,
      "loss": 13.5635,
      "step": 619
    },
    {
      "epoch": 0.2843384544829168,
      "grad_norm": 0.6536729421193141,
      "learning_rate": 0.0008404493287552232,
      "loss": 13.6684,
      "step": 620
    },
    {
      "epoch": 0.2847970648933731,
      "grad_norm": 0.6411356014163625,
      "learning_rate": 0.0008399047637503825,
      "loss": 13.4878,
      "step": 621
    },
    {
      "epoch": 0.2852556753038294,
      "grad_norm": 0.607357533073758,
      "learning_rate": 0.0008393594480791875,
      "loss": 13.6968,
      "step": 622
    },
    {
      "epoch": 0.2857142857142857,
      "grad_norm": 0.6334062152500295,
      "learning_rate": 0.0008388133829459463,
      "loss": 13.789,
      "step": 623
    },
    {
      "epoch": 0.286172896124742,
      "grad_norm": 0.560637155475017,
      "learning_rate": 0.0008382665695566227,
      "loss": 13.7208,
      "step": 624
    },
    {
      "epoch": 0.28663150653519837,
      "grad_norm": 0.5762457899188208,
      "learning_rate": 0.0008377190091188324,
      "loss": 13.7066,
      "step": 625
    },
    {
      "epoch": 0.28709011694565467,
      "grad_norm": 0.6132476024448148,
      "learning_rate": 0.0008371707028418413,
      "loss": 13.7738,
      "step": 626
    },
    {
      "epoch": 0.287548727356111,
      "grad_norm": 0.5896928283102464,
      "learning_rate": 0.0008366216519365621,
      "loss": 13.7329,
      "step": 627
    },
    {
      "epoch": 0.2880073377665673,
      "grad_norm": 0.6174151138199498,
      "learning_rate": 0.0008360718576155525,
      "loss": 13.5624,
      "step": 628
    },
    {
      "epoch": 0.28846594817702365,
      "grad_norm": 0.6316071668066349,
      "learning_rate": 0.0008355213210930118,
      "loss": 13.6499,
      "step": 629
    },
    {
      "epoch": 0.28892455858747995,
      "grad_norm": 0.5939692675489842,
      "learning_rate": 0.0008349700435847778,
      "loss": 13.6922,
      "step": 630
    },
    {
      "epoch": 0.28938316899793626,
      "grad_norm": 0.5980000992997726,
      "learning_rate": 0.0008344180263083256,
      "loss": 13.6139,
      "step": 631
    },
    {
      "epoch": 0.28984177940839256,
      "grad_norm": 0.5999059621204073,
      "learning_rate": 0.000833865270482764,
      "loss": 13.5684,
      "step": 632
    },
    {
      "epoch": 0.29030038981884887,
      "grad_norm": 0.6062028601027973,
      "learning_rate": 0.0008333117773288324,
      "loss": 13.5779,
      "step": 633
    },
    {
      "epoch": 0.29075900022930523,
      "grad_norm": 0.6387829811530091,
      "learning_rate": 0.0008327575480688985,
      "loss": 13.7196,
      "step": 634
    },
    {
      "epoch": 0.29121761063976154,
      "grad_norm": 0.5434969675722822,
      "learning_rate": 0.000832202583926956,
      "loss": 13.7046,
      "step": 635
    },
    {
      "epoch": 0.29167622105021784,
      "grad_norm": 0.8224282024702176,
      "learning_rate": 0.0008316468861286217,
      "loss": 13.8086,
      "step": 636
    },
    {
      "epoch": 0.29213483146067415,
      "grad_norm": 0.6743593368012756,
      "learning_rate": 0.0008310904559011323,
      "loss": 13.6616,
      "step": 637
    },
    {
      "epoch": 0.29259344187113046,
      "grad_norm": 1.0019523246940405,
      "learning_rate": 0.0008305332944733419,
      "loss": 13.6613,
      "step": 638
    },
    {
      "epoch": 0.2930520522815868,
      "grad_norm": 0.7376207170173221,
      "learning_rate": 0.0008299754030757202,
      "loss": 13.5773,
      "step": 639
    },
    {
      "epoch": 0.2935106626920431,
      "grad_norm": 0.6206024223871144,
      "learning_rate": 0.0008294167829403481,
      "loss": 13.7781,
      "step": 640
    },
    {
      "epoch": 0.29396927310249943,
      "grad_norm": 0.6860436995575044,
      "learning_rate": 0.0008288574353009164,
      "loss": 13.4718,
      "step": 641
    },
    {
      "epoch": 0.29442788351295573,
      "grad_norm": 0.6519124191923366,
      "learning_rate": 0.0008282973613927225,
      "loss": 13.7426,
      "step": 642
    },
    {
      "epoch": 0.29488649392341204,
      "grad_norm": 0.7188122459337289,
      "learning_rate": 0.0008277365624526675,
      "loss": 13.7439,
      "step": 643
    },
    {
      "epoch": 0.2953451043338684,
      "grad_norm": 0.6368013895853121,
      "learning_rate": 0.0008271750397192541,
      "loss": 13.7798,
      "step": 644
    },
    {
      "epoch": 0.2958037147443247,
      "grad_norm": 0.6386020393279266,
      "learning_rate": 0.0008266127944325832,
      "loss": 13.6705,
      "step": 645
    },
    {
      "epoch": 0.296262325154781,
      "grad_norm": 0.6507137415264087,
      "learning_rate": 0.0008260498278343513,
      "loss": 13.6092,
      "step": 646
    },
    {
      "epoch": 0.2967209355652373,
      "grad_norm": 0.6475762225774433,
      "learning_rate": 0.0008254861411678485,
      "loss": 13.6507,
      "step": 647
    },
    {
      "epoch": 0.2971795459756936,
      "grad_norm": 0.8543462481532136,
      "learning_rate": 0.0008249217356779544,
      "loss": 13.5577,
      "step": 648
    },
    {
      "epoch": 0.29763815638615,
      "grad_norm": 0.5966037137632111,
      "learning_rate": 0.0008243566126111363,
      "loss": 13.6001,
      "step": 649
    },
    {
      "epoch": 0.2980967667966063,
      "grad_norm": 0.731303896239801,
      "learning_rate": 0.0008237907732154466,
      "loss": 13.6462,
      "step": 650
    },
    {
      "epoch": 0.2985553772070626,
      "grad_norm": 0.6248190085041893,
      "learning_rate": 0.0008232242187405194,
      "loss": 13.6242,
      "step": 651
    },
    {
      "epoch": 0.2990139876175189,
      "grad_norm": 0.6062889961139167,
      "learning_rate": 0.000822656950437568,
      "loss": 13.7575,
      "step": 652
    },
    {
      "epoch": 0.2994725980279752,
      "grad_norm": 0.6979941881359351,
      "learning_rate": 0.0008220889695593823,
      "loss": 13.5416,
      "step": 653
    },
    {
      "epoch": 0.2999312084384316,
      "grad_norm": 0.6051824842798617,
      "learning_rate": 0.0008215202773603259,
      "loss": 13.7264,
      "step": 654
    },
    {
      "epoch": 0.3003898188488879,
      "grad_norm": 0.640898190521148,
      "learning_rate": 0.0008209508750963328,
      "loss": 13.7597,
      "step": 655
    },
    {
      "epoch": 0.3008484292593442,
      "grad_norm": 0.5800522275082293,
      "learning_rate": 0.0008203807640249062,
      "loss": 13.7127,
      "step": 656
    },
    {
      "epoch": 0.3013070396698005,
      "grad_norm": 0.6231031493331002,
      "learning_rate": 0.0008198099454051136,
      "loss": 13.5776,
      "step": 657
    },
    {
      "epoch": 0.3017656500802568,
      "grad_norm": 0.6352422238154609,
      "learning_rate": 0.0008192384204975857,
      "loss": 13.5184,
      "step": 658
    },
    {
      "epoch": 0.30222426049071316,
      "grad_norm": 0.5690529583803215,
      "learning_rate": 0.000818666190564513,
      "loss": 13.6772,
      "step": 659
    },
    {
      "epoch": 0.30268287090116947,
      "grad_norm": 0.6165701392407311,
      "learning_rate": 0.0008180932568696426,
      "loss": 13.6487,
      "step": 660
    },
    {
      "epoch": 0.30314148131162577,
      "grad_norm": 0.5988464806475086,
      "learning_rate": 0.0008175196206782764,
      "loss": 13.5968,
      "step": 661
    },
    {
      "epoch": 0.3036000917220821,
      "grad_norm": 0.59725578493666,
      "learning_rate": 0.0008169452832572675,
      "loss": 13.5951,
      "step": 662
    },
    {
      "epoch": 0.3040587021325384,
      "grad_norm": 0.6042803830828911,
      "learning_rate": 0.0008163702458750173,
      "loss": 13.7321,
      "step": 663
    },
    {
      "epoch": 0.30451731254299474,
      "grad_norm": 0.5871912651508477,
      "learning_rate": 0.0008157945098014734,
      "loss": 13.6616,
      "step": 664
    },
    {
      "epoch": 0.30497592295345105,
      "grad_norm": 0.5813481855846618,
      "learning_rate": 0.0008152180763081267,
      "loss": 13.5294,
      "step": 665
    },
    {
      "epoch": 0.30543453336390736,
      "grad_norm": 0.5525605171345832,
      "learning_rate": 0.0008146409466680076,
      "loss": 13.6655,
      "step": 666
    },
    {
      "epoch": 0.30589314377436366,
      "grad_norm": 0.5891764747462735,
      "learning_rate": 0.0008140631221556845,
      "loss": 13.6422,
      "step": 667
    },
    {
      "epoch": 0.30635175418481997,
      "grad_norm": 0.6744317968391599,
      "learning_rate": 0.0008134846040472599,
      "loss": 13.3055,
      "step": 668
    },
    {
      "epoch": 0.30681036459527633,
      "grad_norm": 0.5882571503986089,
      "learning_rate": 0.0008129053936203688,
      "loss": 13.6421,
      "step": 669
    },
    {
      "epoch": 0.30726897500573264,
      "grad_norm": 0.6259585722392762,
      "learning_rate": 0.0008123254921541745,
      "loss": 13.5569,
      "step": 670
    },
    {
      "epoch": 0.30772758541618894,
      "grad_norm": 0.599055421474114,
      "learning_rate": 0.0008117449009293668,
      "loss": 13.4975,
      "step": 671
    },
    {
      "epoch": 0.30818619582664525,
      "grad_norm": 0.635042562228129,
      "learning_rate": 0.0008111636212281586,
      "loss": 13.5662,
      "step": 672
    },
    {
      "epoch": 0.3086448062371016,
      "grad_norm": 0.5754830277557716,
      "learning_rate": 0.0008105816543342833,
      "loss": 13.5825,
      "step": 673
    },
    {
      "epoch": 0.3091034166475579,
      "grad_norm": 0.615027784675393,
      "learning_rate": 0.0008099990015329919,
      "loss": 13.5802,
      "step": 674
    },
    {
      "epoch": 0.3095620270580142,
      "grad_norm": 0.5709854380812334,
      "learning_rate": 0.0008094156641110504,
      "loss": 13.5946,
      "step": 675
    },
    {
      "epoch": 0.31002063746847053,
      "grad_norm": 0.5949022005551219,
      "learning_rate": 0.0008088316433567369,
      "loss": 13.5177,
      "step": 676
    },
    {
      "epoch": 0.31047924787892683,
      "grad_norm": 0.5274250038561585,
      "learning_rate": 0.0008082469405598378,
      "loss": 13.6159,
      "step": 677
    },
    {
      "epoch": 0.3109378582893832,
      "grad_norm": 0.527002202354421,
      "learning_rate": 0.0008076615570116468,
      "loss": 13.5994,
      "step": 678
    },
    {
      "epoch": 0.3113964686998395,
      "grad_norm": 0.5611555411890431,
      "learning_rate": 0.0008070754940049603,
      "loss": 13.4732,
      "step": 679
    },
    {
      "epoch": 0.3118550791102958,
      "grad_norm": 0.5766484808890594,
      "learning_rate": 0.0008064887528340756,
      "loss": 13.5428,
      "step": 680
    },
    {
      "epoch": 0.3123136895207521,
      "grad_norm": 0.5725411156405885,
      "learning_rate": 0.0008059013347947874,
      "loss": 13.6017,
      "step": 681
    },
    {
      "epoch": 0.3127722999312084,
      "grad_norm": 0.5665046039229136,
      "learning_rate": 0.0008053132411843857,
      "loss": 13.8041,
      "step": 682
    },
    {
      "epoch": 0.3132309103416648,
      "grad_norm": 0.6043913725210918,
      "learning_rate": 0.0008047244733016521,
      "loss": 13.6861,
      "step": 683
    },
    {
      "epoch": 0.3136895207521211,
      "grad_norm": 0.5659823315235768,
      "learning_rate": 0.0008041350324468573,
      "loss": 13.5889,
      "step": 684
    },
    {
      "epoch": 0.3141481311625774,
      "grad_norm": 0.595903371548551,
      "learning_rate": 0.0008035449199217583,
      "loss": 13.4057,
      "step": 685
    },
    {
      "epoch": 0.3146067415730337,
      "grad_norm": 0.7861096183245875,
      "learning_rate": 0.0008029541370295957,
      "loss": 13.6669,
      "step": 686
    },
    {
      "epoch": 0.31506535198349,
      "grad_norm": 0.5845041352194766,
      "learning_rate": 0.0008023626850750903,
      "loss": 13.4066,
      "step": 687
    },
    {
      "epoch": 0.31552396239394637,
      "grad_norm": 0.695752030388076,
      "learning_rate": 0.0008017705653644406,
      "loss": 13.6819,
      "step": 688
    },
    {
      "epoch": 0.3159825728044027,
      "grad_norm": 0.6264936107161541,
      "learning_rate": 0.0008011777792053195,
      "loss": 13.549,
      "step": 689
    },
    {
      "epoch": 0.316441183214859,
      "grad_norm": 0.5970335742487264,
      "learning_rate": 0.0008005843279068725,
      "loss": 13.4306,
      "step": 690
    },
    {
      "epoch": 0.3168997936253153,
      "grad_norm": 0.852831979727779,
      "learning_rate": 0.000799990212779713,
      "loss": 13.7319,
      "step": 691
    },
    {
      "epoch": 0.3173584040357716,
      "grad_norm": 0.5942070342729413,
      "learning_rate": 0.0007993954351359214,
      "loss": 13.6734,
      "step": 692
    },
    {
      "epoch": 0.31781701444622795,
      "grad_norm": 0.7741855353901514,
      "learning_rate": 0.0007987999962890406,
      "loss": 13.7868,
      "step": 693
    },
    {
      "epoch": 0.31827562485668426,
      "grad_norm": 0.5445618288126899,
      "learning_rate": 0.0007982038975540742,
      "loss": 13.5664,
      "step": 694
    },
    {
      "epoch": 0.31873423526714056,
      "grad_norm": 0.6326728537206822,
      "learning_rate": 0.0007976071402474826,
      "loss": 13.5426,
      "step": 695
    },
    {
      "epoch": 0.31919284567759687,
      "grad_norm": 0.6163699916623746,
      "learning_rate": 0.0007970097256871811,
      "loss": 13.5652,
      "step": 696
    },
    {
      "epoch": 0.3196514560880532,
      "grad_norm": 0.5421127361404238,
      "learning_rate": 0.0007964116551925364,
      "loss": 13.7544,
      "step": 697
    },
    {
      "epoch": 0.32011006649850954,
      "grad_norm": 0.7166105416032464,
      "learning_rate": 0.0007958129300843637,
      "loss": 13.5201,
      "step": 698
    },
    {
      "epoch": 0.32056867690896584,
      "grad_norm": 0.6701574721078409,
      "learning_rate": 0.0007952135516849239,
      "loss": 13.7483,
      "step": 699
    },
    {
      "epoch": 0.32102728731942215,
      "grad_norm": 0.6613629437508656,
      "learning_rate": 0.0007946135213179207,
      "loss": 13.623,
      "step": 700
    },
    {
      "epoch": 0.32148589772987846,
      "grad_norm": 0.612228680491199,
      "learning_rate": 0.0007940128403084977,
      "loss": 13.5603,
      "step": 701
    },
    {
      "epoch": 0.32194450814033476,
      "grad_norm": 0.5682998194658357,
      "learning_rate": 0.0007934115099832355,
      "loss": 13.6547,
      "step": 702
    },
    {
      "epoch": 0.3224031185507911,
      "grad_norm": 0.653547902373265,
      "learning_rate": 0.0007928095316701483,
      "loss": 13.7157,
      "step": 703
    },
    {
      "epoch": 0.32286172896124743,
      "grad_norm": 0.6330917420753438,
      "learning_rate": 0.0007922069066986819,
      "loss": 13.7094,
      "step": 704
    },
    {
      "epoch": 0.32332033937170374,
      "grad_norm": 0.6201539811535312,
      "learning_rate": 0.0007916036363997097,
      "loss": 13.6915,
      "step": 705
    },
    {
      "epoch": 0.32377894978216004,
      "grad_norm": 0.5645176772640019,
      "learning_rate": 0.0007909997221055308,
      "loss": 13.6368,
      "step": 706
    },
    {
      "epoch": 0.32423756019261635,
      "grad_norm": 0.6290197547671421,
      "learning_rate": 0.0007903951651498658,
      "loss": 13.2768,
      "step": 707
    },
    {
      "epoch": 0.3246961706030727,
      "grad_norm": 0.6263470126365256,
      "learning_rate": 0.0007897899668678557,
      "loss": 13.6746,
      "step": 708
    },
    {
      "epoch": 0.325154781013529,
      "grad_norm": 0.5757394752203031,
      "learning_rate": 0.0007891841285960566,
      "loss": 13.5241,
      "step": 709
    },
    {
      "epoch": 0.3256133914239853,
      "grad_norm": 0.6768499845913599,
      "learning_rate": 0.0007885776516724388,
      "loss": 13.4366,
      "step": 710
    },
    {
      "epoch": 0.3260720018344416,
      "grad_norm": 0.5824981457301253,
      "learning_rate": 0.0007879705374363831,
      "loss": 13.7639,
      "step": 711
    },
    {
      "epoch": 0.32653061224489793,
      "grad_norm": 0.5974274523440284,
      "learning_rate": 0.000787362787228677,
      "loss": 13.7198,
      "step": 712
    },
    {
      "epoch": 0.3269892226553543,
      "grad_norm": 0.6348554147891059,
      "learning_rate": 0.0007867544023915134,
      "loss": 13.5698,
      "step": 713
    },
    {
      "epoch": 0.3274478330658106,
      "grad_norm": 0.5602589831172908,
      "learning_rate": 0.0007861453842684861,
      "loss": 13.7539,
      "step": 714
    },
    {
      "epoch": 0.3279064434762669,
      "grad_norm": 0.6261857629067016,
      "learning_rate": 0.0007855357342045882,
      "loss": 13.6064,
      "step": 715
    },
    {
      "epoch": 0.3283650538867232,
      "grad_norm": 0.6219639286687317,
      "learning_rate": 0.0007849254535462074,
      "loss": 13.4581,
      "step": 716
    },
    {
      "epoch": 0.3288236642971795,
      "grad_norm": 0.6172994511359347,
      "learning_rate": 0.0007843145436411252,
      "loss": 13.7171,
      "step": 717
    },
    {
      "epoch": 0.3292822747076359,
      "grad_norm": 0.7466037777015748,
      "learning_rate": 0.0007837030058385117,
      "loss": 13.9966,
      "step": 718
    },
    {
      "epoch": 0.3297408851180922,
      "grad_norm": 0.5684012929297922,
      "learning_rate": 0.0007830908414889246,
      "loss": 13.7824,
      "step": 719
    },
    {
      "epoch": 0.3301994955285485,
      "grad_norm": 0.6753324226830371,
      "learning_rate": 0.0007824780519443046,
      "loss": 13.4168,
      "step": 720
    },
    {
      "epoch": 0.3306581059390048,
      "grad_norm": 0.5672718450889276,
      "learning_rate": 0.0007818646385579735,
      "loss": 13.7423,
      "step": 721
    },
    {
      "epoch": 0.33111671634946116,
      "grad_norm": 0.6273867434873771,
      "learning_rate": 0.0007812506026846307,
      "loss": 13.6878,
      "step": 722
    },
    {
      "epoch": 0.33157532675991747,
      "grad_norm": 0.5822824887576802,
      "learning_rate": 0.0007806359456803504,
      "loss": 13.5838,
      "step": 723
    },
    {
      "epoch": 0.33203393717037377,
      "grad_norm": 0.5751079651275747,
      "learning_rate": 0.0007800206689025785,
      "loss": 13.6329,
      "step": 724
    },
    {
      "epoch": 0.3324925475808301,
      "grad_norm": 0.5515455753475962,
      "learning_rate": 0.0007794047737101297,
      "loss": 13.4479,
      "step": 725
    },
    {
      "epoch": 0.3329511579912864,
      "grad_norm": 0.5754491293785099,
      "learning_rate": 0.0007787882614631843,
      "loss": 13.3855,
      "step": 726
    },
    {
      "epoch": 0.33340976840174275,
      "grad_norm": 0.54826736671818,
      "learning_rate": 0.0007781711335232856,
      "loss": 13.4326,
      "step": 727
    },
    {
      "epoch": 0.33386837881219905,
      "grad_norm": 0.6698047942957196,
      "learning_rate": 0.0007775533912533363,
      "loss": 13.7936,
      "step": 728
    },
    {
      "epoch": 0.33432698922265536,
      "grad_norm": 0.6249325246448334,
      "learning_rate": 0.0007769350360175962,
      "loss": 13.5757,
      "step": 729
    },
    {
      "epoch": 0.33478559963311166,
      "grad_norm": 0.5607919149096814,
      "learning_rate": 0.0007763160691816784,
      "loss": 13.5658,
      "step": 730
    },
    {
      "epoch": 0.33524421004356797,
      "grad_norm": 0.6169212931380329,
      "learning_rate": 0.000775696492112547,
      "loss": 13.5616,
      "step": 731
    },
    {
      "epoch": 0.33570282045402433,
      "grad_norm": 0.6063492440872426,
      "learning_rate": 0.0007750763061785137,
      "loss": 13.5934,
      "step": 732
    },
    {
      "epoch": 0.33616143086448064,
      "grad_norm": 0.5622640924683293,
      "learning_rate": 0.000774455512749235,
      "loss": 13.5473,
      "step": 733
    },
    {
      "epoch": 0.33662004127493694,
      "grad_norm": 0.5514358004948533,
      "learning_rate": 0.0007738341131957085,
      "loss": 13.5833,
      "step": 734
    },
    {
      "epoch": 0.33707865168539325,
      "grad_norm": 0.5769818613422089,
      "learning_rate": 0.000773212108890271,
      "loss": 13.3271,
      "step": 735
    },
    {
      "epoch": 0.33753726209584956,
      "grad_norm": 0.6084730558345768,
      "learning_rate": 0.0007725895012065947,
      "loss": 13.5402,
      "step": 736
    },
    {
      "epoch": 0.3379958725063059,
      "grad_norm": 0.5387636715662115,
      "learning_rate": 0.0007719662915196844,
      "loss": 13.6715,
      "step": 737
    },
    {
      "epoch": 0.3384544829167622,
      "grad_norm": 0.5661147366337727,
      "learning_rate": 0.0007713424812058736,
      "loss": 13.6999,
      "step": 738
    },
    {
      "epoch": 0.33891309332721853,
      "grad_norm": 0.5505645440618302,
      "learning_rate": 0.0007707180716428237,
      "loss": 13.7315,
      "step": 739
    },
    {
      "epoch": 0.33937170373767483,
      "grad_norm": 0.6146108004635424,
      "learning_rate": 0.0007700930642095184,
      "loss": 13.408,
      "step": 740
    },
    {
      "epoch": 0.33983031414813114,
      "grad_norm": 0.7211422280375244,
      "learning_rate": 0.0007694674602862621,
      "loss": 13.7341,
      "step": 741
    },
    {
      "epoch": 0.3402889245585875,
      "grad_norm": 0.5842275157207344,
      "learning_rate": 0.0007688412612546769,
      "loss": 13.7829,
      "step": 742
    },
    {
      "epoch": 0.3407475349690438,
      "grad_norm": 0.60592033653982,
      "learning_rate": 0.0007682144684976983,
      "loss": 13.6524,
      "step": 743
    },
    {
      "epoch": 0.3412061453795001,
      "grad_norm": 0.6156858301607667,
      "learning_rate": 0.0007675870833995739,
      "loss": 13.6142,
      "step": 744
    },
    {
      "epoch": 0.3416647557899564,
      "grad_norm": 0.6984714151795144,
      "learning_rate": 0.0007669591073458592,
      "loss": 13.5904,
      "step": 745
    },
    {
      "epoch": 0.3421233662004127,
      "grad_norm": 0.6327750918287087,
      "learning_rate": 0.0007663305417234146,
      "loss": 13.6638,
      "step": 746
    },
    {
      "epoch": 0.3425819766108691,
      "grad_norm": 0.6210749427629,
      "learning_rate": 0.0007657013879204022,
      "loss": 13.6126,
      "step": 747
    },
    {
      "epoch": 0.3430405870213254,
      "grad_norm": 0.6374492435214308,
      "learning_rate": 0.0007650716473262842,
      "loss": 13.826,
      "step": 748
    },
    {
      "epoch": 0.3434991974317817,
      "grad_norm": 0.5908286427076123,
      "learning_rate": 0.0007644413213318177,
      "loss": 13.79,
      "step": 749
    },
    {
      "epoch": 0.343957807842238,
      "grad_norm": 0.6162930496599677,
      "learning_rate": 0.0007638104113290531,
      "loss": 13.6148,
      "step": 750
    },
    {
      "epoch": 0.3444164182526943,
      "grad_norm": 0.6021727668859584,
      "learning_rate": 0.0007631789187113303,
      "loss": 13.8552,
      "step": 751
    },
    {
      "epoch": 0.3448750286631507,
      "grad_norm": 0.6591310875248856,
      "learning_rate": 0.000762546844873276,
      "loss": 13.6003,
      "step": 752
    },
    {
      "epoch": 0.345333639073607,
      "grad_norm": 0.5763630693234254,
      "learning_rate": 0.0007619141912108007,
      "loss": 13.759,
      "step": 753
    },
    {
      "epoch": 0.3457922494840633,
      "grad_norm": 0.5802655278555627,
      "learning_rate": 0.000761280959121095,
      "loss": 13.6221,
      "step": 754
    },
    {
      "epoch": 0.3462508598945196,
      "grad_norm": 0.5497032486112331,
      "learning_rate": 0.0007606471500026273,
      "loss": 13.5849,
      "step": 755
    },
    {
      "epoch": 0.3467094703049759,
      "grad_norm": 0.5800403273996597,
      "learning_rate": 0.0007600127652551401,
      "loss": 13.6906,
      "step": 756
    },
    {
      "epoch": 0.34716808071543226,
      "grad_norm": 0.6107545235811952,
      "learning_rate": 0.0007593778062796472,
      "loss": 13.7014,
      "step": 757
    },
    {
      "epoch": 0.34762669112588857,
      "grad_norm": 0.5940830175588679,
      "learning_rate": 0.000758742274478431,
      "loss": 13.7064,
      "step": 758
    },
    {
      "epoch": 0.34808530153634487,
      "grad_norm": 0.6230319108780368,
      "learning_rate": 0.0007581061712550381,
      "loss": 13.4463,
      "step": 759
    },
    {
      "epoch": 0.3485439119468012,
      "grad_norm": 0.5663296339405927,
      "learning_rate": 0.0007574694980142779,
      "loss": 13.53,
      "step": 760
    },
    {
      "epoch": 0.3490025223572575,
      "grad_norm": 0.6391213558636439,
      "learning_rate": 0.0007568322561622183,
      "loss": 13.7398,
      "step": 761
    },
    {
      "epoch": 0.34946113276771384,
      "grad_norm": 0.5659126570718102,
      "learning_rate": 0.0007561944471061826,
      "loss": 13.5837,
      "step": 762
    },
    {
      "epoch": 0.34991974317817015,
      "grad_norm": 0.6067280110439047,
      "learning_rate": 0.0007555560722547475,
      "loss": 13.8641,
      "step": 763
    },
    {
      "epoch": 0.35037835358862646,
      "grad_norm": 0.5998179813739034,
      "learning_rate": 0.0007549171330177387,
      "loss": 13.4654,
      "step": 764
    },
    {
      "epoch": 0.35083696399908276,
      "grad_norm": 0.5981603375117543,
      "learning_rate": 0.0007542776308062285,
      "loss": 13.5818,
      "step": 765
    },
    {
      "epoch": 0.3512955744095391,
      "grad_norm": 0.6229907353589016,
      "learning_rate": 0.0007536375670325325,
      "loss": 13.617,
      "step": 766
    },
    {
      "epoch": 0.35175418481999543,
      "grad_norm": 0.5623225163475115,
      "learning_rate": 0.0007529969431102063,
      "loss": 13.5044,
      "step": 767
    },
    {
      "epoch": 0.35221279523045174,
      "grad_norm": 0.7135878226989597,
      "learning_rate": 0.000752355760454043,
      "loss": 13.6595,
      "step": 768
    },
    {
      "epoch": 0.35267140564090804,
      "grad_norm": 0.5640450158457362,
      "learning_rate": 0.0007517140204800693,
      "loss": 13.6072,
      "step": 769
    },
    {
      "epoch": 0.35313001605136435,
      "grad_norm": 0.6268181878452884,
      "learning_rate": 0.0007510717246055425,
      "loss": 13.7391,
      "step": 770
    },
    {
      "epoch": 0.3535886264618207,
      "grad_norm": 0.6037830370665113,
      "learning_rate": 0.0007504288742489482,
      "loss": 13.5376,
      "step": 771
    },
    {
      "epoch": 0.354047236872277,
      "grad_norm": 0.6040480808796974,
      "learning_rate": 0.0007497854708299963,
      "loss": 13.5826,
      "step": 772
    },
    {
      "epoch": 0.3545058472827333,
      "grad_norm": 0.563109370184671,
      "learning_rate": 0.0007491415157696178,
      "loss": 13.7353,
      "step": 773
    },
    {
      "epoch": 0.35496445769318963,
      "grad_norm": 0.6052108661526936,
      "learning_rate": 0.0007484970104899623,
      "loss": 13.515,
      "step": 774
    },
    {
      "epoch": 0.35542306810364593,
      "grad_norm": 0.5749749962310987,
      "learning_rate": 0.0007478519564143945,
      "loss": 13.4048,
      "step": 775
    },
    {
      "epoch": 0.3558816785141023,
      "grad_norm": 0.8075842844892491,
      "learning_rate": 0.000747206354967491,
      "loss": 13.6736,
      "step": 776
    },
    {
      "epoch": 0.3563402889245586,
      "grad_norm": 0.6061447647727769,
      "learning_rate": 0.0007465602075750373,
      "loss": 13.7166,
      "step": 777
    },
    {
      "epoch": 0.3567988993350149,
      "grad_norm": 0.670362403458094,
      "learning_rate": 0.0007459135156640247,
      "loss": 13.6265,
      "step": 778
    },
    {
      "epoch": 0.3572575097454712,
      "grad_norm": 0.5597975398111034,
      "learning_rate": 0.0007452662806626468,
      "loss": 13.5565,
      "step": 779
    },
    {
      "epoch": 0.3577161201559275,
      "grad_norm": 0.6843542010922309,
      "learning_rate": 0.0007446185040002967,
      "loss": 13.7868,
      "step": 780
    },
    {
      "epoch": 0.3581747305663839,
      "grad_norm": 0.5522402133594846,
      "learning_rate": 0.0007439701871075642,
      "loss": 13.6095,
      "step": 781
    },
    {
      "epoch": 0.3586333409768402,
      "grad_norm": 0.5618556373060339,
      "learning_rate": 0.0007433213314162313,
      "loss": 13.7667,
      "step": 782
    },
    {
      "epoch": 0.3590919513872965,
      "grad_norm": 0.6288238432158225,
      "learning_rate": 0.0007426719383592705,
      "loss": 13.4389,
      "step": 783
    },
    {
      "epoch": 0.3595505617977528,
      "grad_norm": 0.5511345182738909,
      "learning_rate": 0.000742022009370841,
      "loss": 13.6031,
      "step": 784
    },
    {
      "epoch": 0.3600091722082091,
      "grad_norm": 0.6088400393062707,
      "learning_rate": 0.0007413715458862855,
      "loss": 13.5569,
      "step": 785
    },
    {
      "epoch": 0.36046778261866547,
      "grad_norm": 0.5765008147043353,
      "learning_rate": 0.0007407205493421272,
      "loss": 13.5339,
      "step": 786
    },
    {
      "epoch": 0.3609263930291218,
      "grad_norm": 0.5553279790129059,
      "learning_rate": 0.0007400690211760661,
      "loss": 13.7716,
      "step": 787
    },
    {
      "epoch": 0.3613850034395781,
      "grad_norm": 0.7015239019836854,
      "learning_rate": 0.0007394169628269771,
      "loss": 13.7586,
      "step": 788
    },
    {
      "epoch": 0.3618436138500344,
      "grad_norm": 0.6379801974039155,
      "learning_rate": 0.0007387643757349051,
      "loss": 13.3994,
      "step": 789
    },
    {
      "epoch": 0.3623022242604907,
      "grad_norm": 0.6840147292325841,
      "learning_rate": 0.0007381112613410635,
      "loss": 13.779,
      "step": 790
    },
    {
      "epoch": 0.36276083467094705,
      "grad_norm": 0.6284863216955484,
      "learning_rate": 0.0007374576210878298,
      "loss": 13.4585,
      "step": 791
    },
    {
      "epoch": 0.36321944508140336,
      "grad_norm": 0.5448512246022862,
      "learning_rate": 0.0007368034564187425,
      "loss": 13.5317,
      "step": 792
    },
    {
      "epoch": 0.36367805549185966,
      "grad_norm": 0.7263787992959148,
      "learning_rate": 0.0007361487687784989,
      "loss": 13.6492,
      "step": 793
    },
    {
      "epoch": 0.36413666590231597,
      "grad_norm": 0.5567903737617549,
      "learning_rate": 0.0007354935596129513,
      "loss": 13.7075,
      "step": 794
    },
    {
      "epoch": 0.3645952763127723,
      "grad_norm": 0.6373053065696512,
      "learning_rate": 0.000734837830369103,
      "loss": 13.4317,
      "step": 795
    },
    {
      "epoch": 0.36505388672322864,
      "grad_norm": 0.5825151611589479,
      "learning_rate": 0.0007341815824951066,
      "loss": 13.4738,
      "step": 796
    },
    {
      "epoch": 0.36551249713368494,
      "grad_norm": 0.5426160319381019,
      "learning_rate": 0.0007335248174402597,
      "loss": 13.3725,
      "step": 797
    },
    {
      "epoch": 0.36597110754414125,
      "grad_norm": 0.6128375781870472,
      "learning_rate": 0.0007328675366550023,
      "loss": 13.4346,
      "step": 798
    },
    {
      "epoch": 0.36642971795459756,
      "grad_norm": 0.6049220447177936,
      "learning_rate": 0.0007322097415909134,
      "loss": 13.4617,
      "step": 799
    },
    {
      "epoch": 0.36688832836505386,
      "grad_norm": 0.6602983455826635,
      "learning_rate": 0.0007315514337007071,
      "loss": 13.6256,
      "step": 800
    },
    {
      "epoch": 0.3673469387755102,
      "grad_norm": 0.5811025167061884,
      "learning_rate": 0.0007308926144382312,
      "loss": 13.8624,
      "step": 801
    },
    {
      "epoch": 0.36780554918596653,
      "grad_norm": 0.6863524348106546,
      "learning_rate": 0.0007302332852584619,
      "loss": 13.5824,
      "step": 802
    },
    {
      "epoch": 0.36826415959642284,
      "grad_norm": 0.6140838071078509,
      "learning_rate": 0.0007295734476175018,
      "loss": 13.56,
      "step": 803
    },
    {
      "epoch": 0.36872277000687914,
      "grad_norm": 0.6230794889111474,
      "learning_rate": 0.0007289131029725768,
      "loss": 13.6308,
      "step": 804
    },
    {
      "epoch": 0.36918138041733545,
      "grad_norm": 0.6434689078955128,
      "learning_rate": 0.0007282522527820319,
      "loss": 13.6069,
      "step": 805
    },
    {
      "epoch": 0.3696399908277918,
      "grad_norm": 0.5193694261966645,
      "learning_rate": 0.000727590898505329,
      "loss": 13.4914,
      "step": 806
    },
    {
      "epoch": 0.3700986012382481,
      "grad_norm": 0.6468562053746602,
      "learning_rate": 0.0007269290416030429,
      "loss": 13.6449,
      "step": 807
    },
    {
      "epoch": 0.3705572116487044,
      "grad_norm": 0.5987886016870558,
      "learning_rate": 0.000726266683536859,
      "loss": 13.8215,
      "step": 808
    },
    {
      "epoch": 0.3710158220591607,
      "grad_norm": 0.647012674425558,
      "learning_rate": 0.0007256038257695687,
      "loss": 13.5246,
      "step": 809
    },
    {
      "epoch": 0.37147443246961703,
      "grad_norm": 0.6383501476166696,
      "learning_rate": 0.0007249404697650678,
      "loss": 13.7277,
      "step": 810
    },
    {
      "epoch": 0.3719330428800734,
      "grad_norm": 0.6703769706186398,
      "learning_rate": 0.0007242766169883518,
      "loss": 13.6936,
      "step": 811
    },
    {
      "epoch": 0.3723916532905297,
      "grad_norm": 0.7028357023233633,
      "learning_rate": 0.0007236122689055138,
      "loss": 13.5373,
      "step": 812
    },
    {
      "epoch": 0.372850263700986,
      "grad_norm": 0.6772799686246744,
      "learning_rate": 0.0007229474269837401,
      "loss": 13.6186,
      "step": 813
    },
    {
      "epoch": 0.3733088741114423,
      "grad_norm": 0.6595711862961006,
      "learning_rate": 0.0007222820926913085,
      "loss": 13.7758,
      "step": 814
    },
    {
      "epoch": 0.3737674845218987,
      "grad_norm": 0.5618165834911644,
      "learning_rate": 0.0007216162674975833,
      "loss": 13.7693,
      "step": 815
    },
    {
      "epoch": 0.374226094932355,
      "grad_norm": 0.7947722679081993,
      "learning_rate": 0.0007209499528730138,
      "loss": 13.4541,
      "step": 816
    },
    {
      "epoch": 0.3746847053428113,
      "grad_norm": 0.5524108927096621,
      "learning_rate": 0.0007202831502891294,
      "loss": 13.4522,
      "step": 817
    },
    {
      "epoch": 0.3751433157532676,
      "grad_norm": 0.570268182020548,
      "learning_rate": 0.0007196158612185375,
      "loss": 13.439,
      "step": 818
    },
    {
      "epoch": 0.3756019261637239,
      "grad_norm": 0.6234469153415098,
      "learning_rate": 0.0007189480871349201,
      "loss": 13.4483,
      "step": 819
    },
    {
      "epoch": 0.37606053657418026,
      "grad_norm": 0.5242567161531967,
      "learning_rate": 0.0007182798295130299,
      "loss": 13.4993,
      "step": 820
    },
    {
      "epoch": 0.37651914698463657,
      "grad_norm": 0.5790371024697065,
      "learning_rate": 0.0007176110898286878,
      "loss": 13.7127,
      "step": 821
    },
    {
      "epoch": 0.37697775739509287,
      "grad_norm": 0.5699255047519811,
      "learning_rate": 0.0007169418695587791,
      "loss": 13.4793,
      "step": 822
    },
    {
      "epoch": 0.3774363678055492,
      "grad_norm": 0.5389203269812308,
      "learning_rate": 0.0007162721701812506,
      "loss": 13.456,
      "step": 823
    },
    {
      "epoch": 0.3778949782160055,
      "grad_norm": 0.5784486302974778,
      "learning_rate": 0.0007156019931751072,
      "loss": 13.6281,
      "step": 824
    },
    {
      "epoch": 0.37835358862646185,
      "grad_norm": 0.5579905358231279,
      "learning_rate": 0.0007149313400204082,
      "loss": 13.5216,
      "step": 825
    },
    {
      "epoch": 0.37881219903691815,
      "grad_norm": 0.5954340369991583,
      "learning_rate": 0.0007142602121982653,
      "loss": 13.756,
      "step": 826
    },
    {
      "epoch": 0.37927080944737446,
      "grad_norm": 0.5898301418263091,
      "learning_rate": 0.0007135886111908379,
      "loss": 13.7561,
      "step": 827
    },
    {
      "epoch": 0.37972941985783076,
      "grad_norm": 0.5990033115554539,
      "learning_rate": 0.0007129165384813303,
      "loss": 13.3801,
      "step": 828
    },
    {
      "epoch": 0.38018803026828707,
      "grad_norm": 0.5429651475519288,
      "learning_rate": 0.0007122439955539888,
      "loss": 13.6254,
      "step": 829
    },
    {
      "epoch": 0.38064664067874343,
      "grad_norm": 0.5741979184069959,
      "learning_rate": 0.0007115709838940983,
      "loss": 13.5516,
      "step": 830
    },
    {
      "epoch": 0.38110525108919974,
      "grad_norm": 0.5832429765814502,
      "learning_rate": 0.0007108975049879785,
      "loss": 13.5602,
      "step": 831
    },
    {
      "epoch": 0.38156386149965604,
      "grad_norm": 0.6189688950563406,
      "learning_rate": 0.0007102235603229814,
      "loss": 13.8463,
      "step": 832
    },
    {
      "epoch": 0.38202247191011235,
      "grad_norm": 0.5391250885861488,
      "learning_rate": 0.000709549151387487,
      "loss": 13.6255,
      "step": 833
    },
    {
      "epoch": 0.38248108232056865,
      "grad_norm": 0.6111209405186021,
      "learning_rate": 0.0007088742796709013,
      "loss": 13.4754,
      "step": 834
    },
    {
      "epoch": 0.382939692731025,
      "grad_norm": 0.5374004288507056,
      "learning_rate": 0.000708198946663652,
      "loss": 13.4852,
      "step": 835
    },
    {
      "epoch": 0.3833983031414813,
      "grad_norm": 0.6483721646318287,
      "learning_rate": 0.0007075231538571856,
      "loss": 13.5718,
      "step": 836
    },
    {
      "epoch": 0.38385691355193763,
      "grad_norm": 0.6674584152775463,
      "learning_rate": 0.0007068469027439641,
      "loss": 13.4637,
      "step": 837
    },
    {
      "epoch": 0.38431552396239393,
      "grad_norm": 0.5552600385244811,
      "learning_rate": 0.0007061701948174613,
      "loss": 13.5134,
      "step": 838
    },
    {
      "epoch": 0.38477413437285024,
      "grad_norm": 0.5705733417193543,
      "learning_rate": 0.0007054930315721606,
      "loss": 13.5824,
      "step": 839
    },
    {
      "epoch": 0.3852327447833066,
      "grad_norm": 0.5456747166441891,
      "learning_rate": 0.0007048154145035501,
      "loss": 13.7156,
      "step": 840
    },
    {
      "epoch": 0.3856913551937629,
      "grad_norm": 0.6147239471245861,
      "learning_rate": 0.0007041373451081207,
      "loss": 13.5761,
      "step": 841
    },
    {
      "epoch": 0.3861499656042192,
      "grad_norm": 0.5932836242053862,
      "learning_rate": 0.0007034588248833621,
      "loss": 13.7479,
      "step": 842
    },
    {
      "epoch": 0.3866085760146755,
      "grad_norm": 0.5954541712132768,
      "learning_rate": 0.0007027798553277595,
      "loss": 13.5286,
      "step": 843
    },
    {
      "epoch": 0.3870671864251318,
      "grad_norm": 0.7469582208676645,
      "learning_rate": 0.0007021004379407909,
      "loss": 13.5065,
      "step": 844
    },
    {
      "epoch": 0.3875257968355882,
      "grad_norm": 0.6700278728668074,
      "learning_rate": 0.0007014205742229227,
      "loss": 13.6506,
      "step": 845
    },
    {
      "epoch": 0.3879844072460445,
      "grad_norm": 0.6735468519262006,
      "learning_rate": 0.0007007402656756072,
      "loss": 13.5943,
      "step": 846
    },
    {
      "epoch": 0.3884430176565008,
      "grad_norm": 0.6528496716683133,
      "learning_rate": 0.0007000595138012797,
      "loss": 13.7134,
      "step": 847
    },
    {
      "epoch": 0.3889016280669571,
      "grad_norm": 0.5842050222048204,
      "learning_rate": 0.0006993783201033535,
      "loss": 13.5284,
      "step": 848
    },
    {
      "epoch": 0.3893602384774134,
      "grad_norm": 0.7094702533542087,
      "learning_rate": 0.0006986966860862182,
      "loss": 13.7846,
      "step": 849
    },
    {
      "epoch": 0.3898188488878698,
      "grad_norm": 0.6087642713277478,
      "learning_rate": 0.000698014613255236,
      "loss": 13.643,
      "step": 850
    },
    {
      "epoch": 0.3902774592983261,
      "grad_norm": 0.6098259001179887,
      "learning_rate": 0.0006973321031167382,
      "loss": 13.7148,
      "step": 851
    },
    {
      "epoch": 0.3907360697087824,
      "grad_norm": 0.6557523186351479,
      "learning_rate": 0.0006966491571780216,
      "loss": 13.5914,
      "step": 852
    },
    {
      "epoch": 0.3911946801192387,
      "grad_norm": 0.6253164324810742,
      "learning_rate": 0.0006959657769473453,
      "loss": 13.6339,
      "step": 853
    },
    {
      "epoch": 0.391653290529695,
      "grad_norm": 0.6745135891199311,
      "learning_rate": 0.000695281963933928,
      "loss": 13.7083,
      "step": 854
    },
    {
      "epoch": 0.39211190094015136,
      "grad_norm": 0.7507800135220241,
      "learning_rate": 0.0006945977196479438,
      "loss": 13.6527,
      "step": 855
    },
    {
      "epoch": 0.39257051135060766,
      "grad_norm": 0.8074019328562746,
      "learning_rate": 0.0006939130456005196,
      "loss": 13.675,
      "step": 856
    },
    {
      "epoch": 0.39302912176106397,
      "grad_norm": 0.7669427540675268,
      "learning_rate": 0.0006932279433037311,
      "loss": 13.4035,
      "step": 857
    },
    {
      "epoch": 0.3934877321715203,
      "grad_norm": 0.7917910545659315,
      "learning_rate": 0.0006925424142705997,
      "loss": 13.3121,
      "step": 858
    },
    {
      "epoch": 0.39394634258197664,
      "grad_norm": 0.6344210507415691,
      "learning_rate": 0.0006918564600150896,
      "loss": 13.6557,
      "step": 859
    },
    {
      "epoch": 0.39440495299243294,
      "grad_norm": 0.7786335066696232,
      "learning_rate": 0.0006911700820521042,
      "loss": 13.6959,
      "step": 860
    },
    {
      "epoch": 0.39486356340288925,
      "grad_norm": 0.5769855249975926,
      "learning_rate": 0.0006904832818974818,
      "loss": 13.7758,
      "step": 861
    },
    {
      "epoch": 0.39532217381334556,
      "grad_norm": 0.5810744821158009,
      "learning_rate": 0.0006897960610679939,
      "loss": 13.7138,
      "step": 862
    },
    {
      "epoch": 0.39578078422380186,
      "grad_norm": 0.7834265056224009,
      "learning_rate": 0.0006891084210813407,
      "loss": 13.5928,
      "step": 863
    },
    {
      "epoch": 0.3962393946342582,
      "grad_norm": 0.5692091433815627,
      "learning_rate": 0.0006884203634561483,
      "loss": 13.5474,
      "step": 864
    },
    {
      "epoch": 0.39669800504471453,
      "grad_norm": 0.6461040881591388,
      "learning_rate": 0.0006877318897119651,
      "loss": 13.5007,
      "step": 865
    },
    {
      "epoch": 0.39715661545517084,
      "grad_norm": 0.7014185825411137,
      "learning_rate": 0.0006870430013692579,
      "loss": 13.6671,
      "step": 866
    },
    {
      "epoch": 0.39761522586562714,
      "grad_norm": 0.5502336706217652,
      "learning_rate": 0.0006863536999494101,
      "loss": 13.6698,
      "step": 867
    },
    {
      "epoch": 0.39807383627608345,
      "grad_norm": 0.5722082773725581,
      "learning_rate": 0.0006856639869747167,
      "loss": 13.5968,
      "step": 868
    },
    {
      "epoch": 0.3985324466865398,
      "grad_norm": 0.614705260513655,
      "learning_rate": 0.0006849738639683818,
      "loss": 13.6979,
      "step": 869
    },
    {
      "epoch": 0.3989910570969961,
      "grad_norm": 0.5199829402102999,
      "learning_rate": 0.000684283332454515,
      "loss": 13.5741,
      "step": 870
    },
    {
      "epoch": 0.3994496675074524,
      "grad_norm": 0.5787389783699345,
      "learning_rate": 0.0006835923939581281,
      "loss": 13.6834,
      "step": 871
    },
    {
      "epoch": 0.3999082779179087,
      "grad_norm": 0.5690866154664359,
      "learning_rate": 0.0006829010500051318,
      "loss": 13.6424,
      "step": 872
    },
    {
      "epoch": 0.40036688832836503,
      "grad_norm": 0.5199668071775325,
      "learning_rate": 0.0006822093021223321,
      "loss": 13.5286,
      "step": 873
    },
    {
      "epoch": 0.4008254987388214,
      "grad_norm": 0.5574701458972539,
      "learning_rate": 0.0006815171518374268,
      "loss": 13.5737,
      "step": 874
    },
    {
      "epoch": 0.4012841091492777,
      "grad_norm": 0.5913911034626759,
      "learning_rate": 0.0006808246006790031,
      "loss": 13.6829,
      "step": 875
    },
    {
      "epoch": 0.401742719559734,
      "grad_norm": 0.5614027710507419,
      "learning_rate": 0.0006801316501765329,
      "loss": 13.6693,
      "step": 876
    },
    {
      "epoch": 0.4022013299701903,
      "grad_norm": 0.5640776612200458,
      "learning_rate": 0.0006794383018603704,
      "loss": 13.9246,
      "step": 877
    },
    {
      "epoch": 0.4026599403806466,
      "grad_norm": 0.5489785719567937,
      "learning_rate": 0.0006787445572617481,
      "loss": 13.5758,
      "step": 878
    },
    {
      "epoch": 0.403118550791103,
      "grad_norm": 0.5375751246807399,
      "learning_rate": 0.0006780504179127734,
      "loss": 13.6394,
      "step": 879
    },
    {
      "epoch": 0.4035771612015593,
      "grad_norm": 0.6131887537002043,
      "learning_rate": 0.0006773558853464265,
      "loss": 13.3936,
      "step": 880
    },
    {
      "epoch": 0.4040357716120156,
      "grad_norm": 0.5626704051848611,
      "learning_rate": 0.000676660961096555,
      "loss": 13.7854,
      "step": 881
    },
    {
      "epoch": 0.4044943820224719,
      "grad_norm": 0.5616389186947695,
      "learning_rate": 0.000675965646697872,
      "loss": 13.6478,
      "step": 882
    },
    {
      "epoch": 0.4049529924329282,
      "grad_norm": 0.6000554868828583,
      "learning_rate": 0.0006752699436859519,
      "loss": 13.6098,
      "step": 883
    },
    {
      "epoch": 0.40541160284338457,
      "grad_norm": 0.63714671123489,
      "learning_rate": 0.0006745738535972279,
      "loss": 13.5639,
      "step": 884
    },
    {
      "epoch": 0.4058702132538409,
      "grad_norm": 0.5798654094010567,
      "learning_rate": 0.0006738773779689874,
      "loss": 13.5807,
      "step": 885
    },
    {
      "epoch": 0.4063288236642972,
      "grad_norm": 0.58159295703214,
      "learning_rate": 0.0006731805183393696,
      "loss": 13.5936,
      "step": 886
    },
    {
      "epoch": 0.4067874340747535,
      "grad_norm": 0.6305286068846507,
      "learning_rate": 0.0006724832762473618,
      "loss": 13.5845,
      "step": 887
    },
    {
      "epoch": 0.4072460444852098,
      "grad_norm": 0.5446148284192537,
      "learning_rate": 0.0006717856532327956,
      "loss": 13.747,
      "step": 888
    },
    {
      "epoch": 0.40770465489566615,
      "grad_norm": 0.6196757115541541,
      "learning_rate": 0.0006710876508363444,
      "loss": 13.6656,
      "step": 889
    },
    {
      "epoch": 0.40816326530612246,
      "grad_norm": 0.5688798629255171,
      "learning_rate": 0.0006703892705995189,
      "loss": 13.516,
      "step": 890
    },
    {
      "epoch": 0.40862187571657876,
      "grad_norm": 0.623642947604135,
      "learning_rate": 0.0006696905140646647,
      "loss": 13.6385,
      "step": 891
    },
    {
      "epoch": 0.40908048612703507,
      "grad_norm": 0.5828104959222071,
      "learning_rate": 0.0006689913827749581,
      "loss": 13.583,
      "step": 892
    },
    {
      "epoch": 0.4095390965374914,
      "grad_norm": 0.5509320445383631,
      "learning_rate": 0.0006682918782744032,
      "loss": 13.6367,
      "step": 893
    },
    {
      "epoch": 0.40999770694794774,
      "grad_norm": 0.7158314442517139,
      "learning_rate": 0.0006675920021078282,
      "loss": 13.6332,
      "step": 894
    },
    {
      "epoch": 0.41045631735840404,
      "grad_norm": 0.9129818698920307,
      "learning_rate": 0.0006668917558208823,
      "loss": 13.298,
      "step": 895
    },
    {
      "epoch": 0.41091492776886035,
      "grad_norm": 0.8680777635581313,
      "learning_rate": 0.0006661911409600321,
      "loss": 13.5503,
      "step": 896
    },
    {
      "epoch": 0.41137353817931666,
      "grad_norm": 0.7998205661944121,
      "learning_rate": 0.0006654901590725577,
      "loss": 13.9304,
      "step": 897
    },
    {
      "epoch": 0.41183214858977296,
      "grad_norm": 0.583489729635096,
      "learning_rate": 0.0006647888117065507,
      "loss": 13.6074,
      "step": 898
    },
    {
      "epoch": 0.4122907590002293,
      "grad_norm": 0.63537814371727,
      "learning_rate": 0.0006640871004109086,
      "loss": 13.584,
      "step": 899
    },
    {
      "epoch": 0.41274936941068563,
      "grad_norm": 0.8434210037631955,
      "learning_rate": 0.000663385026735334,
      "loss": 13.7747,
      "step": 900
    },
    {
      "epoch": 0.41320797982114194,
      "grad_norm": 0.5431946168461869,
      "learning_rate": 0.0006626825922303285,
      "loss": 13.5891,
      "step": 901
    },
    {
      "epoch": 0.41366659023159824,
      "grad_norm": 0.6516284675959576,
      "learning_rate": 0.0006619797984471915,
      "loss": 13.6953,
      "step": 902
    },
    {
      "epoch": 0.41412520064205455,
      "grad_norm": 0.7842577524864894,
      "learning_rate": 0.0006612766469380158,
      "loss": 13.7858,
      "step": 903
    },
    {
      "epoch": 0.4145838110525109,
      "grad_norm": 0.5718529829965143,
      "learning_rate": 0.0006605731392556833,
      "loss": 13.6601,
      "step": 904
    },
    {
      "epoch": 0.4150424214629672,
      "grad_norm": 0.7022556904781722,
      "learning_rate": 0.0006598692769538637,
      "loss": 13.5688,
      "step": 905
    },
    {
      "epoch": 0.4155010318734235,
      "grad_norm": 0.7287559410135431,
      "learning_rate": 0.0006591650615870091,
      "loss": 13.5531,
      "step": 906
    },
    {
      "epoch": 0.4159596422838798,
      "grad_norm": 0.5436484904861051,
      "learning_rate": 0.0006584604947103514,
      "loss": 13.4622,
      "step": 907
    },
    {
      "epoch": 0.4164182526943362,
      "grad_norm": 0.7266660896538878,
      "learning_rate": 0.0006577555778798993,
      "loss": 13.7064,
      "step": 908
    },
    {
      "epoch": 0.4168768631047925,
      "grad_norm": 0.58647996289277,
      "learning_rate": 0.0006570503126524336,
      "loss": 13.4079,
      "step": 909
    },
    {
      "epoch": 0.4173354735152488,
      "grad_norm": 0.5507802088200082,
      "learning_rate": 0.0006563447005855054,
      "loss": 13.8059,
      "step": 910
    },
    {
      "epoch": 0.4177940839257051,
      "grad_norm": 0.6747146072243948,
      "learning_rate": 0.000655638743237431,
      "loss": 13.6231,
      "step": 911
    },
    {
      "epoch": 0.4182526943361614,
      "grad_norm": 0.7041702871869101,
      "learning_rate": 0.0006549324421672894,
      "loss": 13.5743,
      "step": 912
    },
    {
      "epoch": 0.4187113047466178,
      "grad_norm": 0.5824858674924581,
      "learning_rate": 0.0006542257989349194,
      "loss": 13.6585,
      "step": 913
    },
    {
      "epoch": 0.4191699151570741,
      "grad_norm": 0.6469421721420233,
      "learning_rate": 0.0006535188151009142,
      "loss": 13.8333,
      "step": 914
    },
    {
      "epoch": 0.4196285255675304,
      "grad_norm": 0.5638891817506644,
      "learning_rate": 0.0006528114922266204,
      "loss": 13.5467,
      "step": 915
    },
    {
      "epoch": 0.4200871359779867,
      "grad_norm": 0.536058396419222,
      "learning_rate": 0.0006521038318741327,
      "loss": 13.6483,
      "step": 916
    },
    {
      "epoch": 0.420545746388443,
      "grad_norm": 0.5384173815513661,
      "learning_rate": 0.0006513958356062912,
      "loss": 13.5105,
      "step": 917
    },
    {
      "epoch": 0.42100435679889936,
      "grad_norm": 0.6038073564554068,
      "learning_rate": 0.0006506875049866781,
      "loss": 13.8068,
      "step": 918
    },
    {
      "epoch": 0.42146296720935567,
      "grad_norm": 0.5586511888349721,
      "learning_rate": 0.0006499788415796137,
      "loss": 13.4753,
      "step": 919
    },
    {
      "epoch": 0.42192157761981197,
      "grad_norm": 0.5201993799157999,
      "learning_rate": 0.0006492698469501532,
      "loss": 13.3866,
      "step": 920
    },
    {
      "epoch": 0.4223801880302683,
      "grad_norm": 0.5873386922518576,
      "learning_rate": 0.0006485605226640837,
      "loss": 13.5007,
      "step": 921
    },
    {
      "epoch": 0.4228387984407246,
      "grad_norm": 0.626447656192024,
      "learning_rate": 0.00064785087028792,
      "loss": 13.5628,
      "step": 922
    },
    {
      "epoch": 0.42329740885118095,
      "grad_norm": 0.5339544118069889,
      "learning_rate": 0.0006471408913889019,
      "loss": 13.4904,
      "step": 923
    },
    {
      "epoch": 0.42375601926163725,
      "grad_norm": 0.5746351321060013,
      "learning_rate": 0.0006464305875349892,
      "loss": 13.6164,
      "step": 924
    },
    {
      "epoch": 0.42421462967209356,
      "grad_norm": 0.5386164186982964,
      "learning_rate": 0.000645719960294861,
      "loss": 13.4632,
      "step": 925
    },
    {
      "epoch": 0.42467324008254986,
      "grad_norm": 0.5473441965840881,
      "learning_rate": 0.0006450090112379092,
      "loss": 13.4932,
      "step": 926
    },
    {
      "epoch": 0.42513185049300617,
      "grad_norm": 0.5738911890591202,
      "learning_rate": 0.0006442977419342371,
      "loss": 13.5265,
      "step": 927
    },
    {
      "epoch": 0.42559046090346253,
      "grad_norm": 0.5571308940430735,
      "learning_rate": 0.000643586153954655,
      "loss": 13.7325,
      "step": 928
    },
    {
      "epoch": 0.42604907131391884,
      "grad_norm": 0.5649786216080434,
      "learning_rate": 0.0006428742488706772,
      "loss": 13.5873,
      "step": 929
    },
    {
      "epoch": 0.42650768172437514,
      "grad_norm": 0.649223606964218,
      "learning_rate": 0.0006421620282545182,
      "loss": 13.5805,
      "step": 930
    },
    {
      "epoch": 0.42696629213483145,
      "grad_norm": 0.5316641399444084,
      "learning_rate": 0.0006414494936790892,
      "loss": 13.5328,
      "step": 931
    },
    {
      "epoch": 0.42742490254528775,
      "grad_norm": 0.6312051296515806,
      "learning_rate": 0.0006407366467179951,
      "loss": 13.7293,
      "step": 932
    },
    {
      "epoch": 0.4278835129557441,
      "grad_norm": 0.5506050304797966,
      "learning_rate": 0.0006400234889455301,
      "loss": 13.611,
      "step": 933
    },
    {
      "epoch": 0.4283421233662004,
      "grad_norm": 0.5369166320980052,
      "learning_rate": 0.0006393100219366755,
      "loss": 13.5676,
      "step": 934
    },
    {
      "epoch": 0.42880073377665673,
      "grad_norm": 0.5364906210561445,
      "learning_rate": 0.0006385962472670953,
      "loss": 13.6845,
      "step": 935
    },
    {
      "epoch": 0.42925934418711303,
      "grad_norm": 0.5864192061409008,
      "learning_rate": 0.0006378821665131328,
      "loss": 13.5332,
      "step": 936
    },
    {
      "epoch": 0.42971795459756934,
      "grad_norm": 0.522368098415229,
      "learning_rate": 0.0006371677812518072,
      "loss": 13.6213,
      "step": 937
    },
    {
      "epoch": 0.4301765650080257,
      "grad_norm": 0.5700302985348981,
      "learning_rate": 0.0006364530930608107,
      "loss": 13.7127,
      "step": 938
    },
    {
      "epoch": 0.430635175418482,
      "grad_norm": 0.5995109415958063,
      "learning_rate": 0.0006357381035185038,
      "loss": 13.6234,
      "step": 939
    },
    {
      "epoch": 0.4310937858289383,
      "grad_norm": 0.5207639257804676,
      "learning_rate": 0.0006350228142039131,
      "loss": 13.6924,
      "step": 940
    },
    {
      "epoch": 0.4315523962393946,
      "grad_norm": 0.6743677628673935,
      "learning_rate": 0.000634307226696727,
      "loss": 13.7712,
      "step": 941
    },
    {
      "epoch": 0.4320110066498509,
      "grad_norm": 0.6422929504203501,
      "learning_rate": 0.0006335913425772926,
      "loss": 13.3079,
      "step": 942
    },
    {
      "epoch": 0.4324696170603073,
      "grad_norm": 0.6074723358061833,
      "learning_rate": 0.0006328751634266117,
      "loss": 13.7241,
      "step": 943
    },
    {
      "epoch": 0.4329282274707636,
      "grad_norm": 0.7152400927651041,
      "learning_rate": 0.0006321586908263382,
      "loss": 13.6952,
      "step": 944
    },
    {
      "epoch": 0.4333868378812199,
      "grad_norm": 0.5652985655493015,
      "learning_rate": 0.0006314419263587732,
      "loss": 13.6252,
      "step": 945
    },
    {
      "epoch": 0.4338454482916762,
      "grad_norm": 0.6928907173877982,
      "learning_rate": 0.0006307248716068637,
      "loss": 13.4072,
      "step": 946
    },
    {
      "epoch": 0.4343040587021325,
      "grad_norm": 0.6264192262781794,
      "learning_rate": 0.0006300075281541964,
      "loss": 13.8002,
      "step": 947
    },
    {
      "epoch": 0.4347626691125889,
      "grad_norm": 0.5613777952394443,
      "learning_rate": 0.0006292898975849966,
      "loss": 13.5809,
      "step": 948
    },
    {
      "epoch": 0.4352212795230452,
      "grad_norm": 0.6036977944382736,
      "learning_rate": 0.000628571981484123,
      "loss": 13.8232,
      "step": 949
    },
    {
      "epoch": 0.4356798899335015,
      "grad_norm": 0.5884920533790938,
      "learning_rate": 0.0006278537814370654,
      "loss": 13.6068,
      "step": 950
    },
    {
      "epoch": 0.4361385003439578,
      "grad_norm": 0.5606178230900211,
      "learning_rate": 0.0006271352990299406,
      "loss": 13.2918,
      "step": 951
    },
    {
      "epoch": 0.43659711075441415,
      "grad_norm": 0.5280602193037442,
      "learning_rate": 0.0006264165358494885,
      "loss": 13.8035,
      "step": 952
    },
    {
      "epoch": 0.43705572116487046,
      "grad_norm": 0.7124674527416099,
      "learning_rate": 0.0006256974934830694,
      "loss": 13.489,
      "step": 953
    },
    {
      "epoch": 0.43751433157532676,
      "grad_norm": 0.5517993428205346,
      "learning_rate": 0.0006249781735186606,
      "loss": 13.5779,
      "step": 954
    },
    {
      "epoch": 0.43797294198578307,
      "grad_norm": 0.5456022875703159,
      "learning_rate": 0.0006242585775448518,
      "loss": 13.5701,
      "step": 955
    },
    {
      "epoch": 0.4384315523962394,
      "grad_norm": 0.5284492616036103,
      "learning_rate": 0.0006235387071508427,
      "loss": 13.571,
      "step": 956
    },
    {
      "epoch": 0.43889016280669574,
      "grad_norm": 0.5531339877349373,
      "learning_rate": 0.0006228185639264384,
      "loss": 13.6655,
      "step": 957
    },
    {
      "epoch": 0.43934877321715204,
      "grad_norm": 0.5451087708064326,
      "learning_rate": 0.0006220981494620475,
      "loss": 13.7667,
      "step": 958
    },
    {
      "epoch": 0.43980738362760835,
      "grad_norm": 0.5876395215603394,
      "learning_rate": 0.000621377465348677,
      "loss": 13.4463,
      "step": 959
    },
    {
      "epoch": 0.44026599403806466,
      "grad_norm": 0.5550827373652599,
      "learning_rate": 0.0006206565131779293,
      "loss": 13.5879,
      "step": 960
    },
    {
      "epoch": 0.44072460444852096,
      "grad_norm": 0.5909209638566013,
      "learning_rate": 0.0006199352945419994,
      "loss": 13.6737,
      "step": 961
    },
    {
      "epoch": 0.4411832148589773,
      "grad_norm": 0.5878508804279182,
      "learning_rate": 0.00061921381103367,
      "loss": 13.7117,
      "step": 962
    },
    {
      "epoch": 0.44164182526943363,
      "grad_norm": 0.5390904616277219,
      "learning_rate": 0.0006184920642463094,
      "loss": 13.7174,
      "step": 963
    },
    {
      "epoch": 0.44210043567988994,
      "grad_norm": 0.5472956806230217,
      "learning_rate": 0.0006177700557738672,
      "loss": 13.7268,
      "step": 964
    },
    {
      "epoch": 0.44255904609034624,
      "grad_norm": 0.56964606423241,
      "learning_rate": 0.0006170477872108706,
      "loss": 13.5407,
      "step": 965
    },
    {
      "epoch": 0.44301765650080255,
      "grad_norm": 0.5520144580958275,
      "learning_rate": 0.0006163252601524216,
      "loss": 13.6705,
      "step": 966
    },
    {
      "epoch": 0.4434762669112589,
      "grad_norm": 0.5208243535678345,
      "learning_rate": 0.0006156024761941925,
      "loss": 13.5742,
      "step": 967
    },
    {
      "epoch": 0.4439348773217152,
      "grad_norm": 0.5617043371715799,
      "learning_rate": 0.000614879436932424,
      "loss": 13.6826,
      "step": 968
    },
    {
      "epoch": 0.4443934877321715,
      "grad_norm": 0.5347144289602631,
      "learning_rate": 0.0006141561439639196,
      "loss": 13.5344,
      "step": 969
    },
    {
      "epoch": 0.4448520981426278,
      "grad_norm": 0.5251484580027276,
      "learning_rate": 0.0006134325988860433,
      "loss": 13.6689,
      "step": 970
    },
    {
      "epoch": 0.44531070855308413,
      "grad_norm": 0.5709084400301729,
      "learning_rate": 0.0006127088032967165,
      "loss": 13.43,
      "step": 971
    },
    {
      "epoch": 0.4457693189635405,
      "grad_norm": 0.5420817984140488,
      "learning_rate": 0.0006119847587944131,
      "loss": 13.468,
      "step": 972
    },
    {
      "epoch": 0.4462279293739968,
      "grad_norm": 0.626803769165009,
      "learning_rate": 0.0006112604669781572,
      "loss": 13.6288,
      "step": 973
    },
    {
      "epoch": 0.4466865397844531,
      "grad_norm": 0.5507742521166112,
      "learning_rate": 0.0006105359294475188,
      "loss": 13.4703,
      "step": 974
    },
    {
      "epoch": 0.4471451501949094,
      "grad_norm": 0.5608514573420699,
      "learning_rate": 0.0006098111478026107,
      "loss": 13.6118,
      "step": 975
    },
    {
      "epoch": 0.4476037606053657,
      "grad_norm": 0.6193517831028252,
      "learning_rate": 0.0006090861236440848,
      "loss": 13.4056,
      "step": 976
    },
    {
      "epoch": 0.4480623710158221,
      "grad_norm": 0.5786944095055083,
      "learning_rate": 0.0006083608585731282,
      "loss": 13.7577,
      "step": 977
    },
    {
      "epoch": 0.4485209814262784,
      "grad_norm": 0.5391122570000473,
      "learning_rate": 0.0006076353541914609,
      "loss": 13.3104,
      "step": 978
    },
    {
      "epoch": 0.4489795918367347,
      "grad_norm": 0.5638206629971582,
      "learning_rate": 0.0006069096121013307,
      "loss": 13.7069,
      "step": 979
    },
    {
      "epoch": 0.449438202247191,
      "grad_norm": 0.5473393323135662,
      "learning_rate": 0.0006061836339055105,
      "loss": 13.614,
      "step": 980
    },
    {
      "epoch": 0.4498968126576473,
      "grad_norm": 0.5631987851453638,
      "learning_rate": 0.0006054574212072948,
      "loss": 13.4912,
      "step": 981
    },
    {
      "epoch": 0.45035542306810367,
      "grad_norm": 0.5874945262231633,
      "learning_rate": 0.0006047309756104958,
      "loss": 13.5175,
      "step": 982
    },
    {
      "epoch": 0.45081403347856,
      "grad_norm": 0.569728794704899,
      "learning_rate": 0.00060400429871944,
      "loss": 13.5233,
      "step": 983
    },
    {
      "epoch": 0.4512726438890163,
      "grad_norm": 0.5673939035746538,
      "learning_rate": 0.0006032773921389654,
      "loss": 13.3193,
      "step": 984
    },
    {
      "epoch": 0.4517312542994726,
      "grad_norm": 0.5545224478117882,
      "learning_rate": 0.0006025502574744162,
      "loss": 13.6475,
      "step": 985
    },
    {
      "epoch": 0.4521898647099289,
      "grad_norm": 0.5615595356077406,
      "learning_rate": 0.000601822896331641,
      "loss": 13.6248,
      "step": 986
    },
    {
      "epoch": 0.45264847512038525,
      "grad_norm": 0.6318716801139692,
      "learning_rate": 0.0006010953103169883,
      "loss": 13.6372,
      "step": 987
    },
    {
      "epoch": 0.45310708553084156,
      "grad_norm": 0.5418150275031935,
      "learning_rate": 0.0006003675010373034,
      "loss": 13.5027,
      "step": 988
    },
    {
      "epoch": 0.45356569594129786,
      "grad_norm": 0.5532263242645411,
      "learning_rate": 0.0005996394700999246,
      "loss": 13.2626,
      "step": 989
    },
    {
      "epoch": 0.45402430635175417,
      "grad_norm": 0.5755439719217146,
      "learning_rate": 0.0005989112191126794,
      "loss": 13.3475,
      "step": 990
    },
    {
      "epoch": 0.4544829167622105,
      "grad_norm": 0.5446536893679628,
      "learning_rate": 0.0005981827496838822,
      "loss": 13.7011,
      "step": 991
    },
    {
      "epoch": 0.45494152717266684,
      "grad_norm": 0.6065919648100918,
      "learning_rate": 0.0005974540634223286,
      "loss": 13.4726,
      "step": 992
    },
    {
      "epoch": 0.45540013758312314,
      "grad_norm": 0.5624960286195618,
      "learning_rate": 0.0005967251619372939,
      "loss": 13.6298,
      "step": 993
    },
    {
      "epoch": 0.45585874799357945,
      "grad_norm": 0.5588607895885466,
      "learning_rate": 0.0005959960468385284,
      "loss": 13.4934,
      "step": 994
    },
    {
      "epoch": 0.45631735840403576,
      "grad_norm": 0.6326813352160471,
      "learning_rate": 0.0005952667197362542,
      "loss": 13.536,
      "step": 995
    },
    {
      "epoch": 0.45677596881449206,
      "grad_norm": 0.5545399695884765,
      "learning_rate": 0.0005945371822411621,
      "loss": 13.5031,
      "step": 996
    },
    {
      "epoch": 0.4572345792249484,
      "grad_norm": 0.5564046623127789,
      "learning_rate": 0.0005938074359644063,
      "loss": 13.6966,
      "step": 997
    },
    {
      "epoch": 0.45769318963540473,
      "grad_norm": 0.5705837797060918,
      "learning_rate": 0.0005930774825176034,
      "loss": 13.5745,
      "step": 998
    },
    {
      "epoch": 0.45815180004586104,
      "grad_norm": 0.5522646894536747,
      "learning_rate": 0.0005923473235128268,
      "loss": 13.6171,
      "step": 999
    },
    {
      "epoch": 0.45861041045631734,
      "grad_norm": 0.5495097294871931,
      "learning_rate": 0.0005916169605626042,
      "loss": 13.3682,
      "step": 1000
    },
    {
      "epoch": 0.4590690208667737,
      "grad_norm": 0.5446530641725066,
      "learning_rate": 0.0005908863952799134,
      "loss": 13.5391,
      "step": 1001
    },
    {
      "epoch": 0.45952763127723,
      "grad_norm": 0.5354362933240567,
      "learning_rate": 0.0005901556292781793,
      "loss": 13.5497,
      "step": 1002
    },
    {
      "epoch": 0.4599862416876863,
      "grad_norm": 0.5421659197604202,
      "learning_rate": 0.0005894246641712698,
      "loss": 13.6245,
      "step": 1003
    },
    {
      "epoch": 0.4604448520981426,
      "grad_norm": 0.5326338965684002,
      "learning_rate": 0.0005886935015734931,
      "loss": 13.5405,
      "step": 1004
    },
    {
      "epoch": 0.4609034625085989,
      "grad_norm": 0.5095618852976368,
      "learning_rate": 0.0005879621430995928,
      "loss": 13.6536,
      "step": 1005
    },
    {
      "epoch": 0.4613620729190553,
      "grad_norm": 0.5431798069769697,
      "learning_rate": 0.0005872305903647455,
      "loss": 13.4719,
      "step": 1006
    },
    {
      "epoch": 0.4618206833295116,
      "grad_norm": 0.5528586565101086,
      "learning_rate": 0.0005864988449845569,
      "loss": 13.6054,
      "step": 1007
    },
    {
      "epoch": 0.4622792937399679,
      "grad_norm": 0.5741073961046554,
      "learning_rate": 0.0005857669085750578,
      "loss": 13.6855,
      "step": 1008
    },
    {
      "epoch": 0.4627379041504242,
      "grad_norm": 0.5321393890909472,
      "learning_rate": 0.0005850347827527013,
      "loss": 13.4473,
      "step": 1009
    },
    {
      "epoch": 0.4631965145608805,
      "grad_norm": 0.5306589626783951,
      "learning_rate": 0.0005843024691343584,
      "loss": 13.5849,
      "step": 1010
    },
    {
      "epoch": 0.4636551249713369,
      "grad_norm": 0.5606783933152849,
      "learning_rate": 0.000583569969337315,
      "loss": 13.7112,
      "step": 1011
    },
    {
      "epoch": 0.4641137353817932,
      "grad_norm": 0.6452494355217819,
      "learning_rate": 0.0005828372849792686,
      "loss": 13.5198,
      "step": 1012
    },
    {
      "epoch": 0.4645723457922495,
      "grad_norm": 0.5838895805714309,
      "learning_rate": 0.0005821044176783234,
      "loss": 13.7175,
      "step": 1013
    },
    {
      "epoch": 0.4650309562027058,
      "grad_norm": 0.5463365820721495,
      "learning_rate": 0.0005813713690529886,
      "loss": 13.4325,
      "step": 1014
    },
    {
      "epoch": 0.4654895666131621,
      "grad_norm": 0.5850512507698484,
      "learning_rate": 0.0005806381407221729,
      "loss": 13.6341,
      "step": 1015
    },
    {
      "epoch": 0.46594817702361846,
      "grad_norm": 0.5665778781805461,
      "learning_rate": 0.0005799047343051826,
      "loss": 13.4995,
      "step": 1016
    },
    {
      "epoch": 0.46640678743407477,
      "grad_norm": 0.5630051053271975,
      "learning_rate": 0.0005791711514217171,
      "loss": 13.4812,
      "step": 1017
    },
    {
      "epoch": 0.46686539784453107,
      "grad_norm": 0.5066016913217574,
      "learning_rate": 0.0005784373936918654,
      "loss": 13.6233,
      "step": 1018
    },
    {
      "epoch": 0.4673240082549874,
      "grad_norm": 0.5352734444808291,
      "learning_rate": 0.0005777034627361025,
      "loss": 13.6667,
      "step": 1019
    },
    {
      "epoch": 0.4677826186654437,
      "grad_norm": 0.5336298523156128,
      "learning_rate": 0.0005769693601752864,
      "loss": 13.7057,
      "step": 1020
    },
    {
      "epoch": 0.46824122907590005,
      "grad_norm": 0.5489733400358167,
      "learning_rate": 0.0005762350876306537,
      "loss": 13.5896,
      "step": 1021
    },
    {
      "epoch": 0.46869983948635635,
      "grad_norm": 0.5269054017078793,
      "learning_rate": 0.0005755006467238168,
      "loss": 13.5216,
      "step": 1022
    },
    {
      "epoch": 0.46915844989681266,
      "grad_norm": 0.5747892078741783,
      "learning_rate": 0.0005747660390767593,
      "loss": 13.391,
      "step": 1023
    },
    {
      "epoch": 0.46961706030726896,
      "grad_norm": 0.5649321228919727,
      "learning_rate": 0.0005740312663118338,
      "loss": 13.5465,
      "step": 1024
    },
    {
      "epoch": 0.47007567071772527,
      "grad_norm": 0.5659157592081904,
      "learning_rate": 0.0005732963300517568,
      "loss": 13.6411,
      "step": 1025
    },
    {
      "epoch": 0.47053428112818163,
      "grad_norm": 0.5188414645907043,
      "learning_rate": 0.0005725612319196064,
      "loss": 13.6592,
      "step": 1026
    },
    {
      "epoch": 0.47099289153863794,
      "grad_norm": 0.5183217134589021,
      "learning_rate": 0.000571825973538818,
      "loss": 13.625,
      "step": 1027
    },
    {
      "epoch": 0.47145150194909424,
      "grad_norm": 0.5246941140134582,
      "learning_rate": 0.0005710905565331811,
      "loss": 13.5301,
      "step": 1028
    },
    {
      "epoch": 0.47191011235955055,
      "grad_norm": 0.5604997421512843,
      "learning_rate": 0.0005703549825268353,
      "loss": 13.5355,
      "step": 1029
    },
    {
      "epoch": 0.47236872277000685,
      "grad_norm": 0.4982500699614927,
      "learning_rate": 0.0005696192531442667,
      "loss": 13.6564,
      "step": 1030
    },
    {
      "epoch": 0.4728273331804632,
      "grad_norm": 0.5671905809113867,
      "learning_rate": 0.000568883370010305,
      "loss": 13.6769,
      "step": 1031
    },
    {
      "epoch": 0.4732859435909195,
      "grad_norm": 0.5728989321590058,
      "learning_rate": 0.0005681473347501192,
      "loss": 13.5857,
      "step": 1032
    },
    {
      "epoch": 0.47374455400137583,
      "grad_norm": 0.5589176429796295,
      "learning_rate": 0.0005674111489892144,
      "loss": 13.6542,
      "step": 1033
    },
    {
      "epoch": 0.47420316441183213,
      "grad_norm": 0.5792032079441899,
      "learning_rate": 0.0005666748143534282,
      "loss": 13.4136,
      "step": 1034
    },
    {
      "epoch": 0.47466177482228844,
      "grad_norm": 0.5496629651729095,
      "learning_rate": 0.0005659383324689266,
      "loss": 13.5413,
      "step": 1035
    },
    {
      "epoch": 0.4751203852327448,
      "grad_norm": 0.609104331935357,
      "learning_rate": 0.0005652017049622007,
      "loss": 13.6902,
      "step": 1036
    },
    {
      "epoch": 0.4755789956432011,
      "grad_norm": 0.5412022680973836,
      "learning_rate": 0.0005644649334600641,
      "loss": 13.623,
      "step": 1037
    },
    {
      "epoch": 0.4760376060536574,
      "grad_norm": 0.5302355486679915,
      "learning_rate": 0.0005637280195896474,
      "loss": 13.6099,
      "step": 1038
    },
    {
      "epoch": 0.4764962164641137,
      "grad_norm": 0.5306127871376809,
      "learning_rate": 0.0005629909649783961,
      "loss": 13.6729,
      "step": 1039
    },
    {
      "epoch": 0.47695482687457,
      "grad_norm": 0.5335815913080861,
      "learning_rate": 0.0005622537712540664,
      "loss": 13.7135,
      "step": 1040
    },
    {
      "epoch": 0.4774134372850264,
      "grad_norm": 0.5325240795382683,
      "learning_rate": 0.0005615164400447218,
      "loss": 13.7371,
      "step": 1041
    },
    {
      "epoch": 0.4778720476954827,
      "grad_norm": 0.5125622928719061,
      "learning_rate": 0.0005607789729787294,
      "loss": 13.6654,
      "step": 1042
    },
    {
      "epoch": 0.478330658105939,
      "grad_norm": 0.6607567056858847,
      "learning_rate": 0.0005600413716847564,
      "loss": 13.3674,
      "step": 1043
    },
    {
      "epoch": 0.4787892685163953,
      "grad_norm": 0.527983704106386,
      "learning_rate": 0.000559303637791766,
      "loss": 13.6411,
      "step": 1044
    },
    {
      "epoch": 0.47924787892685167,
      "grad_norm": 0.6062075917629542,
      "learning_rate": 0.0005585657729290151,
      "loss": 13.4623,
      "step": 1045
    },
    {
      "epoch": 0.479706489337308,
      "grad_norm": 0.5531656470726274,
      "learning_rate": 0.000557827778726049,
      "loss": 13.3058,
      "step": 1046
    },
    {
      "epoch": 0.4801650997477643,
      "grad_norm": 0.5164073818467789,
      "learning_rate": 0.0005570896568126993,
      "loss": 13.4893,
      "step": 1047
    },
    {
      "epoch": 0.4806237101582206,
      "grad_norm": 0.5831637477484235,
      "learning_rate": 0.0005563514088190788,
      "loss": 13.3213,
      "step": 1048
    },
    {
      "epoch": 0.4810823205686769,
      "grad_norm": 0.5938303110430991,
      "learning_rate": 0.0005556130363755798,
      "loss": 13.4809,
      "step": 1049
    },
    {
      "epoch": 0.48154093097913325,
      "grad_norm": 0.5478721032992294,
      "learning_rate": 0.0005548745411128688,
      "loss": 13.5694,
      "step": 1050
    },
    {
      "epoch": 0.48199954138958956,
      "grad_norm": 0.5631899373269579,
      "learning_rate": 0.0005541359246618835,
      "loss": 13.5962,
      "step": 1051
    },
    {
      "epoch": 0.48245815180004586,
      "grad_norm": 0.5844787859854189,
      "learning_rate": 0.0005533971886538293,
      "loss": 13.6676,
      "step": 1052
    },
    {
      "epoch": 0.48291676221050217,
      "grad_norm": 0.6087934304779172,
      "learning_rate": 0.000552658334720176,
      "loss": 13.6718,
      "step": 1053
    },
    {
      "epoch": 0.4833753726209585,
      "grad_norm": 0.5990941963438141,
      "learning_rate": 0.0005519193644926535,
      "loss": 13.7445,
      "step": 1054
    },
    {
      "epoch": 0.48383398303141484,
      "grad_norm": 0.6448154675336014,
      "learning_rate": 0.0005511802796032485,
      "loss": 13.6487,
      "step": 1055
    },
    {
      "epoch": 0.48429259344187114,
      "grad_norm": 0.5683520368220308,
      "learning_rate": 0.0005504410816842009,
      "loss": 13.4263,
      "step": 1056
    },
    {
      "epoch": 0.48475120385232745,
      "grad_norm": 0.522500187968133,
      "learning_rate": 0.0005497017723680009,
      "loss": 13.3739,
      "step": 1057
    },
    {
      "epoch": 0.48520981426278376,
      "grad_norm": 0.6222830285945028,
      "learning_rate": 0.0005489623532873836,
      "loss": 13.6128,
      "step": 1058
    },
    {
      "epoch": 0.48566842467324006,
      "grad_norm": 0.5556757423735426,
      "learning_rate": 0.0005482228260753273,
      "loss": 13.4337,
      "step": 1059
    },
    {
      "epoch": 0.4861270350836964,
      "grad_norm": 0.5123794674969557,
      "learning_rate": 0.0005474831923650488,
      "loss": 13.5773,
      "step": 1060
    },
    {
      "epoch": 0.48658564549415273,
      "grad_norm": 0.6071483795162552,
      "learning_rate": 0.00054674345379,
      "loss": 13.567,
      "step": 1061
    },
    {
      "epoch": 0.48704425590460904,
      "grad_norm": 0.5327892168507891,
      "learning_rate": 0.000546003611983865,
      "loss": 13.5014,
      "step": 1062
    },
    {
      "epoch": 0.48750286631506534,
      "grad_norm": 0.5452538830338014,
      "learning_rate": 0.0005452636685805552,
      "loss": 13.6563,
      "step": 1063
    },
    {
      "epoch": 0.48796147672552165,
      "grad_norm": 0.6027490454366802,
      "learning_rate": 0.0005445236252142066,
      "loss": 13.5453,
      "step": 1064
    },
    {
      "epoch": 0.488420087135978,
      "grad_norm": 0.6230473025090296,
      "learning_rate": 0.000543783483519176,
      "loss": 13.5345,
      "step": 1065
    },
    {
      "epoch": 0.4888786975464343,
      "grad_norm": 0.5120829487830404,
      "learning_rate": 0.0005430432451300374,
      "loss": 13.4698,
      "step": 1066
    },
    {
      "epoch": 0.4893373079568906,
      "grad_norm": 0.646650680100527,
      "learning_rate": 0.0005423029116815781,
      "loss": 13.5414,
      "step": 1067
    },
    {
      "epoch": 0.4897959183673469,
      "grad_norm": 0.574855568908949,
      "learning_rate": 0.0005415624848087959,
      "loss": 13.5774,
      "step": 1068
    },
    {
      "epoch": 0.49025452877780323,
      "grad_norm": 0.6306276309116331,
      "learning_rate": 0.000540821966146894,
      "loss": 13.5124,
      "step": 1069
    },
    {
      "epoch": 0.4907131391882596,
      "grad_norm": 0.5921858710610072,
      "learning_rate": 0.0005400813573312793,
      "loss": 13.579,
      "step": 1070
    },
    {
      "epoch": 0.4911717495987159,
      "grad_norm": 0.5352437176715177,
      "learning_rate": 0.0005393406599975572,
      "loss": 13.5508,
      "step": 1071
    },
    {
      "epoch": 0.4916303600091722,
      "grad_norm": 0.6240772327105741,
      "learning_rate": 0.0005385998757815287,
      "loss": 13.4516,
      "step": 1072
    },
    {
      "epoch": 0.4920889704196285,
      "grad_norm": 0.5676249364591481,
      "learning_rate": 0.0005378590063191867,
      "loss": 13.5452,
      "step": 1073
    },
    {
      "epoch": 0.4925475808300848,
      "grad_norm": 0.5650395989703926,
      "learning_rate": 0.0005371180532467124,
      "loss": 13.5373,
      "step": 1074
    },
    {
      "epoch": 0.4930061912405412,
      "grad_norm": 0.655289127457637,
      "learning_rate": 0.000536377018200472,
      "loss": 13.739,
      "step": 1075
    },
    {
      "epoch": 0.4934648016509975,
      "grad_norm": 0.5709168941765268,
      "learning_rate": 0.0005356359028170118,
      "loss": 13.5262,
      "step": 1076
    },
    {
      "epoch": 0.4939234120614538,
      "grad_norm": 0.6088483277510108,
      "learning_rate": 0.0005348947087330564,
      "loss": 13.7953,
      "step": 1077
    },
    {
      "epoch": 0.4943820224719101,
      "grad_norm": 0.5689050273042651,
      "learning_rate": 0.0005341534375855037,
      "loss": 13.6665,
      "step": 1078
    },
    {
      "epoch": 0.4948406328823664,
      "grad_norm": 0.5259038554872832,
      "learning_rate": 0.0005334120910114222,
      "loss": 13.4695,
      "step": 1079
    },
    {
      "epoch": 0.49529924329282277,
      "grad_norm": 0.5337746047952354,
      "learning_rate": 0.0005326706706480467,
      "loss": 13.5459,
      "step": 1080
    },
    {
      "epoch": 0.49575785370327907,
      "grad_norm": 0.5521042584312661,
      "learning_rate": 0.0005319291781327749,
      "loss": 13.452,
      "step": 1081
    },
    {
      "epoch": 0.4962164641137354,
      "grad_norm": 0.5453238382653417,
      "learning_rate": 0.0005311876151031642,
      "loss": 13.3961,
      "step": 1082
    },
    {
      "epoch": 0.4966750745241917,
      "grad_norm": 0.6109971176355694,
      "learning_rate": 0.0005304459831969274,
      "loss": 13.4658,
      "step": 1083
    },
    {
      "epoch": 0.497133684934648,
      "grad_norm": 0.5161458186086,
      "learning_rate": 0.0005297042840519294,
      "loss": 13.5089,
      "step": 1084
    },
    {
      "epoch": 0.49759229534510435,
      "grad_norm": 0.549008929771026,
      "learning_rate": 0.0005289625193061838,
      "loss": 13.6973,
      "step": 1085
    },
    {
      "epoch": 0.49805090575556066,
      "grad_norm": 0.582386271997519,
      "learning_rate": 0.0005282206905978489,
      "loss": 13.3661,
      "step": 1086
    },
    {
      "epoch": 0.49850951616601696,
      "grad_norm": 0.5716607791848105,
      "learning_rate": 0.0005274787995652246,
      "loss": 13.383,
      "step": 1087
    },
    {
      "epoch": 0.49896812657647327,
      "grad_norm": 0.5254884105037807,
      "learning_rate": 0.000526736847846748,
      "loss": 13.4261,
      "step": 1088
    },
    {
      "epoch": 0.4994267369869296,
      "grad_norm": 0.6599174274197837,
      "learning_rate": 0.0005259948370809901,
      "loss": 13.6279,
      "step": 1089
    },
    {
      "epoch": 0.49988534739738594,
      "grad_norm": 0.5254273661012071,
      "learning_rate": 0.0005252527689066533,
      "loss": 13.3035,
      "step": 1090
    },
    {
      "epoch": 0.5003439578078422,
      "grad_norm": 0.5755989250660142,
      "learning_rate": 0.0005245106449625654,
      "loss": 13.6248,
      "step": 1091
    },
    {
      "epoch": 0.5008025682182986,
      "grad_norm": 0.5562509544771557,
      "learning_rate": 0.0005237684668876785,
      "loss": 13.6732,
      "step": 1092
    },
    {
      "epoch": 0.5012611786287549,
      "grad_norm": 0.588447420396966,
      "learning_rate": 0.0005230262363210637,
      "loss": 13.5461,
      "step": 1093
    },
    {
      "epoch": 0.5017197890392112,
      "grad_norm": 0.5411286872855782,
      "learning_rate": 0.0005222839549019079,
      "loss": 13.3598,
      "step": 1094
    },
    {
      "epoch": 0.5021783994496675,
      "grad_norm": 0.6229593135478078,
      "learning_rate": 0.0005215416242695108,
      "loss": 13.4135,
      "step": 1095
    },
    {
      "epoch": 0.5026370098601238,
      "grad_norm": 0.5956132768271483,
      "learning_rate": 0.0005207992460632804,
      "loss": 13.4672,
      "step": 1096
    },
    {
      "epoch": 0.5030956202705802,
      "grad_norm": 0.6606581627312751,
      "learning_rate": 0.0005200568219227299,
      "loss": 13.5542,
      "step": 1097
    },
    {
      "epoch": 0.5035542306810364,
      "grad_norm": 0.5819031551234072,
      "learning_rate": 0.000519314353487474,
      "loss": 13.5677,
      "step": 1098
    },
    {
      "epoch": 0.5040128410914928,
      "grad_norm": 0.5724218207445584,
      "learning_rate": 0.0005185718423972251,
      "loss": 13.4376,
      "step": 1099
    },
    {
      "epoch": 0.504471451501949,
      "grad_norm": 0.5193430256655747,
      "learning_rate": 0.0005178292902917898,
      "loss": 13.4346,
      "step": 1100
    },
    {
      "epoch": 0.5049300619124054,
      "grad_norm": 0.6307060679886018,
      "learning_rate": 0.0005170866988110656,
      "loss": 13.5749,
      "step": 1101
    },
    {
      "epoch": 0.5053886723228618,
      "grad_norm": 0.5268291222858491,
      "learning_rate": 0.0005163440695950362,
      "loss": 13.1056,
      "step": 1102
    },
    {
      "epoch": 0.505847282733318,
      "grad_norm": 0.5585407185223348,
      "learning_rate": 0.0005156014042837695,
      "loss": 13.7144,
      "step": 1103
    },
    {
      "epoch": 0.5063058931437744,
      "grad_norm": 0.5446501213625118,
      "learning_rate": 0.0005148587045174128,
      "loss": 13.7687,
      "step": 1104
    },
    {
      "epoch": 0.5067645035542306,
      "grad_norm": 0.5541981535325757,
      "learning_rate": 0.0005141159719361891,
      "loss": 13.4276,
      "step": 1105
    },
    {
      "epoch": 0.507223113964687,
      "grad_norm": 0.5655625533932969,
      "learning_rate": 0.0005133732081803945,
      "loss": 13.6324,
      "step": 1106
    },
    {
      "epoch": 0.5076817243751434,
      "grad_norm": 0.5552231020710297,
      "learning_rate": 0.0005126304148903936,
      "loss": 13.443,
      "step": 1107
    },
    {
      "epoch": 0.5081403347855996,
      "grad_norm": 0.5809036856074284,
      "learning_rate": 0.0005118875937066161,
      "loss": 13.553,
      "step": 1108
    },
    {
      "epoch": 0.508598945196056,
      "grad_norm": 0.5243886860687192,
      "learning_rate": 0.0005111447462695537,
      "loss": 13.6768,
      "step": 1109
    },
    {
      "epoch": 0.5090575556065122,
      "grad_norm": 0.5624873222803268,
      "learning_rate": 0.0005104018742197557,
      "loss": 13.4049,
      "step": 1110
    },
    {
      "epoch": 0.5095161660169686,
      "grad_norm": 0.5317228037400518,
      "learning_rate": 0.0005096589791978261,
      "loss": 13.7291,
      "step": 1111
    },
    {
      "epoch": 0.509974776427425,
      "grad_norm": 0.5219983829656455,
      "learning_rate": 0.0005089160628444192,
      "loss": 13.5938,
      "step": 1112
    },
    {
      "epoch": 0.5104333868378812,
      "grad_norm": 0.5203407578216053,
      "learning_rate": 0.0005081731268002371,
      "loss": 13.6156,
      "step": 1113
    },
    {
      "epoch": 0.5108919972483376,
      "grad_norm": 0.5414206145804362,
      "learning_rate": 0.0005074301727060243,
      "loss": 13.5575,
      "step": 1114
    },
    {
      "epoch": 0.5113506076587938,
      "grad_norm": 0.5539146018087978,
      "learning_rate": 0.0005066872022025663,
      "loss": 13.6549,
      "step": 1115
    },
    {
      "epoch": 0.5118092180692502,
      "grad_norm": 0.5369031076573004,
      "learning_rate": 0.0005059442169306844,
      "loss": 13.5273,
      "step": 1116
    },
    {
      "epoch": 0.5122678284797065,
      "grad_norm": 0.5466713500605188,
      "learning_rate": 0.0005052012185312321,
      "loss": 13.8026,
      "step": 1117
    },
    {
      "epoch": 0.5127264388901628,
      "grad_norm": 0.5451033637796265,
      "learning_rate": 0.0005044582086450925,
      "loss": 13.5454,
      "step": 1118
    },
    {
      "epoch": 0.5131850493006191,
      "grad_norm": 0.5551544202538341,
      "learning_rate": 0.0005037151889131737,
      "loss": 13.6834,
      "step": 1119
    },
    {
      "epoch": 0.5136436597110754,
      "grad_norm": 0.5481493401142311,
      "learning_rate": 0.0005029721609764059,
      "loss": 13.6082,
      "step": 1120
    },
    {
      "epoch": 0.5141022701215318,
      "grad_norm": 0.5531700408910959,
      "learning_rate": 0.000502229126475737,
      "loss": 13.6128,
      "step": 1121
    },
    {
      "epoch": 0.5145608805319881,
      "grad_norm": 0.5353190670794353,
      "learning_rate": 0.0005014860870521293,
      "loss": 13.7742,
      "step": 1122
    },
    {
      "epoch": 0.5150194909424444,
      "grad_norm": 0.6071539904112794,
      "learning_rate": 0.0005007430443465569,
      "loss": 13.7143,
      "step": 1123
    },
    {
      "epoch": 0.5154781013529007,
      "grad_norm": 0.888081069678747,
      "learning_rate": 0.0005,
      "loss": 13.4619,
      "step": 1124
    },
    {
      "epoch": 0.515936711763357,
      "grad_norm": 0.593952732351873,
      "learning_rate": 0.0004992569556534432,
      "loss": 13.4396,
      "step": 1125
    },
    {
      "epoch": 0.5163953221738133,
      "grad_norm": 0.6927230756160121,
      "learning_rate": 0.0004985139129478707,
      "loss": 13.6444,
      "step": 1126
    },
    {
      "epoch": 0.5168539325842697,
      "grad_norm": 0.547178386083209,
      "learning_rate": 0.0004977708735242633,
      "loss": 13.6732,
      "step": 1127
    },
    {
      "epoch": 0.517312542994726,
      "grad_norm": 0.5716593646714221,
      "learning_rate": 0.0004970278390235942,
      "loss": 13.6385,
      "step": 1128
    },
    {
      "epoch": 0.5177711534051823,
      "grad_norm": 0.6571510512106646,
      "learning_rate": 0.0004962848110868262,
      "loss": 13.4703,
      "step": 1129
    },
    {
      "epoch": 0.5182297638156386,
      "grad_norm": 0.5353501418504372,
      "learning_rate": 0.0004955417913549074,
      "loss": 13.7572,
      "step": 1130
    },
    {
      "epoch": 0.5186883742260949,
      "grad_norm": 0.5714465097544165,
      "learning_rate": 0.0004947987814687679,
      "loss": 13.4337,
      "step": 1131
    },
    {
      "epoch": 0.5191469846365513,
      "grad_norm": 0.5797804422252759,
      "learning_rate": 0.0004940557830693157,
      "loss": 13.5825,
      "step": 1132
    },
    {
      "epoch": 0.5196055950470075,
      "grad_norm": 0.56552341076434,
      "learning_rate": 0.0004933127977974338,
      "loss": 13.6782,
      "step": 1133
    },
    {
      "epoch": 0.5200642054574639,
      "grad_norm": 0.6211183501116199,
      "learning_rate": 0.0004925698272939757,
      "loss": 13.4904,
      "step": 1134
    },
    {
      "epoch": 0.5205228158679202,
      "grad_norm": 0.6245001589769252,
      "learning_rate": 0.0004918268731997632,
      "loss": 13.586,
      "step": 1135
    },
    {
      "epoch": 0.5209814262783765,
      "grad_norm": 0.5257449349127453,
      "learning_rate": 0.0004910839371555809,
      "loss": 13.612,
      "step": 1136
    },
    {
      "epoch": 0.5214400366888329,
      "grad_norm": 0.5922139921858391,
      "learning_rate": 0.0004903410208021739,
      "loss": 13.462,
      "step": 1137
    },
    {
      "epoch": 0.5218986470992891,
      "grad_norm": 0.6462460734871341,
      "learning_rate": 0.0004895981257802443,
      "loss": 13.7024,
      "step": 1138
    },
    {
      "epoch": 0.5223572575097455,
      "grad_norm": 0.5283194077425841,
      "learning_rate": 0.0004888552537304463,
      "loss": 13.4104,
      "step": 1139
    },
    {
      "epoch": 0.5228158679202017,
      "grad_norm": 0.547652214461585,
      "learning_rate": 0.00048811240629338394,
      "loss": 13.6076,
      "step": 1140
    },
    {
      "epoch": 0.5232744783306581,
      "grad_norm": 0.7612029624872978,
      "learning_rate": 0.00048736958510960663,
      "loss": 13.5745,
      "step": 1141
    },
    {
      "epoch": 0.5237330887411145,
      "grad_norm": 0.5703096591783996,
      "learning_rate": 0.00048662679181960564,
      "loss": 13.4051,
      "step": 1142
    },
    {
      "epoch": 0.5241916991515707,
      "grad_norm": 0.6717075268088895,
      "learning_rate": 0.00048588402806381094,
      "loss": 13.4261,
      "step": 1143
    },
    {
      "epoch": 0.5246503095620271,
      "grad_norm": 0.56804504522414,
      "learning_rate": 0.0004851412954825874,
      "loss": 13.4085,
      "step": 1144
    },
    {
      "epoch": 0.5251089199724833,
      "grad_norm": 0.5615226125711104,
      "learning_rate": 0.00048439859571623034,
      "loss": 13.6705,
      "step": 1145
    },
    {
      "epoch": 0.5255675303829397,
      "grad_norm": 0.5251947670717765,
      "learning_rate": 0.00048365593040496373,
      "loss": 13.5715,
      "step": 1146
    },
    {
      "epoch": 0.526026140793396,
      "grad_norm": 0.6232572117539389,
      "learning_rate": 0.00048291330118893443,
      "loss": 13.5484,
      "step": 1147
    },
    {
      "epoch": 0.5264847512038523,
      "grad_norm": 0.6659618164375236,
      "learning_rate": 0.0004821707097082102,
      "loss": 13.477,
      "step": 1148
    },
    {
      "epoch": 0.5269433616143087,
      "grad_norm": 0.5537240676063181,
      "learning_rate": 0.0004814281576027749,
      "loss": 13.8357,
      "step": 1149
    },
    {
      "epoch": 0.5274019720247649,
      "grad_norm": 0.6273112312308505,
      "learning_rate": 0.000480685646512526,
      "loss": 13.6684,
      "step": 1150
    },
    {
      "epoch": 0.5278605824352213,
      "grad_norm": 0.5860446867654938,
      "learning_rate": 0.00047994317807727025,
      "loss": 13.4247,
      "step": 1151
    },
    {
      "epoch": 0.5283191928456776,
      "grad_norm": 0.5222399507180763,
      "learning_rate": 0.00047920075393671974,
      "loss": 13.5397,
      "step": 1152
    },
    {
      "epoch": 0.5287778032561339,
      "grad_norm": 0.6576986594074476,
      "learning_rate": 0.0004784583757304893,
      "loss": 13.5562,
      "step": 1153
    },
    {
      "epoch": 0.5292364136665902,
      "grad_norm": 0.5173872339262062,
      "learning_rate": 0.00047771604509809214,
      "loss": 13.741,
      "step": 1154
    },
    {
      "epoch": 0.5296950240770465,
      "grad_norm": 0.5144029627702573,
      "learning_rate": 0.0004769737636789364,
      "loss": 13.2384,
      "step": 1155
    },
    {
      "epoch": 0.5301536344875029,
      "grad_norm": 0.578456345936532,
      "learning_rate": 0.00047623153311232157,
      "loss": 13.411,
      "step": 1156
    },
    {
      "epoch": 0.5306122448979592,
      "grad_norm": 0.5828861304476578,
      "learning_rate": 0.0004754893550374346,
      "loss": 13.5492,
      "step": 1157
    },
    {
      "epoch": 0.5310708553084155,
      "grad_norm": 0.5432249382911857,
      "learning_rate": 0.00047474723109334685,
      "loss": 13.3481,
      "step": 1158
    },
    {
      "epoch": 0.5315294657188718,
      "grad_norm": 0.571562308393684,
      "learning_rate": 0.00047400516291900993,
      "loss": 13.4764,
      "step": 1159
    },
    {
      "epoch": 0.5319880761293282,
      "grad_norm": 0.5966646765008524,
      "learning_rate": 0.0004732631521532522,
      "loss": 13.6952,
      "step": 1160
    },
    {
      "epoch": 0.5324466865397844,
      "grad_norm": 0.5234536477167421,
      "learning_rate": 0.0004725212004347755,
      "loss": 13.5703,
      "step": 1161
    },
    {
      "epoch": 0.5329052969502408,
      "grad_norm": 0.5249110997379028,
      "learning_rate": 0.00047177930940215095,
      "loss": 13.3846,
      "step": 1162
    },
    {
      "epoch": 0.533363907360697,
      "grad_norm": 0.5642987788099078,
      "learning_rate": 0.00047103748069381624,
      "loss": 13.4877,
      "step": 1163
    },
    {
      "epoch": 0.5338225177711534,
      "grad_norm": 0.4995529096293841,
      "learning_rate": 0.0004702957159480707,
      "loss": 13.4744,
      "step": 1164
    },
    {
      "epoch": 0.5342811281816098,
      "grad_norm": 0.5463113602853265,
      "learning_rate": 0.00046955401680307267,
      "loss": 13.4851,
      "step": 1165
    },
    {
      "epoch": 0.534739738592066,
      "grad_norm": 0.5393007395979699,
      "learning_rate": 0.0004688123848968359,
      "loss": 13.53,
      "step": 1166
    },
    {
      "epoch": 0.5351983490025224,
      "grad_norm": 0.5396963796899099,
      "learning_rate": 0.00046807082186722516,
      "loss": 13.5113,
      "step": 1167
    },
    {
      "epoch": 0.5356569594129786,
      "grad_norm": 0.5340707738073198,
      "learning_rate": 0.0004673293293519535,
      "loss": 13.6475,
      "step": 1168
    },
    {
      "epoch": 0.536115569823435,
      "grad_norm": 0.5302852195498664,
      "learning_rate": 0.00046658790898857806,
      "loss": 13.3448,
      "step": 1169
    },
    {
      "epoch": 0.5365741802338914,
      "grad_norm": 0.5569424611866779,
      "learning_rate": 0.0004658465624144963,
      "loss": 13.8516,
      "step": 1170
    },
    {
      "epoch": 0.5370327906443476,
      "grad_norm": 0.5508433279096292,
      "learning_rate": 0.0004651052912669438,
      "loss": 13.6789,
      "step": 1171
    },
    {
      "epoch": 0.537491401054804,
      "grad_norm": 0.5387740084531422,
      "learning_rate": 0.0004643640971829883,
      "loss": 13.6005,
      "step": 1172
    },
    {
      "epoch": 0.5379500114652602,
      "grad_norm": 0.5930724195184603,
      "learning_rate": 0.0004636229817995281,
      "loss": 13.4181,
      "step": 1173
    },
    {
      "epoch": 0.5384086218757166,
      "grad_norm": 0.596326178158726,
      "learning_rate": 0.0004628819467532876,
      "loss": 13.4719,
      "step": 1174
    },
    {
      "epoch": 0.538867232286173,
      "grad_norm": 0.533833703757643,
      "learning_rate": 0.00046214099368081335,
      "loss": 13.6088,
      "step": 1175
    },
    {
      "epoch": 0.5393258426966292,
      "grad_norm": 0.6526955242140658,
      "learning_rate": 0.0004614001242184714,
      "loss": 13.5306,
      "step": 1176
    },
    {
      "epoch": 0.5397844531070856,
      "grad_norm": 0.6290279934758455,
      "learning_rate": 0.000460659340002443,
      "loss": 13.712,
      "step": 1177
    },
    {
      "epoch": 0.5402430635175418,
      "grad_norm": 0.5310808915807036,
      "learning_rate": 0.00045991864266872073,
      "loss": 13.4775,
      "step": 1178
    },
    {
      "epoch": 0.5407016739279982,
      "grad_norm": 0.6438453742116592,
      "learning_rate": 0.00045917803385310595,
      "loss": 13.3737,
      "step": 1179
    },
    {
      "epoch": 0.5411602843384545,
      "grad_norm": 0.5505117192725582,
      "learning_rate": 0.00045843751519120417,
      "loss": 13.8135,
      "step": 1180
    },
    {
      "epoch": 0.5416188947489108,
      "grad_norm": 0.5273887462757542,
      "learning_rate": 0.00045769708831842193,
      "loss": 13.4839,
      "step": 1181
    },
    {
      "epoch": 0.5420775051593671,
      "grad_norm": 0.5300819356777146,
      "learning_rate": 0.00045695675486996266,
      "loss": 13.4746,
      "step": 1182
    },
    {
      "epoch": 0.5425361155698234,
      "grad_norm": 0.5572538158515721,
      "learning_rate": 0.00045621651648082405,
      "loss": 13.4864,
      "step": 1183
    },
    {
      "epoch": 0.5429947259802798,
      "grad_norm": 0.5454638336477624,
      "learning_rate": 0.00045547637478579356,
      "loss": 13.7383,
      "step": 1184
    },
    {
      "epoch": 0.5434533363907361,
      "grad_norm": 0.6319074593941993,
      "learning_rate": 0.0004547363314194449,
      "loss": 13.4085,
      "step": 1185
    },
    {
      "epoch": 0.5439119468011924,
      "grad_norm": 0.5461791496923941,
      "learning_rate": 0.000453996388016135,
      "loss": 13.4604,
      "step": 1186
    },
    {
      "epoch": 0.5443705572116487,
      "grad_norm": 0.5411380231715611,
      "learning_rate": 0.0004532565462099999,
      "loss": 13.5763,
      "step": 1187
    },
    {
      "epoch": 0.544829167622105,
      "grad_norm": 0.6290402071684896,
      "learning_rate": 0.0004525168076349513,
      "loss": 13.7601,
      "step": 1188
    },
    {
      "epoch": 0.5452877780325613,
      "grad_norm": 0.6381966637700975,
      "learning_rate": 0.0004517771739246729,
      "loss": 13.6816,
      "step": 1189
    },
    {
      "epoch": 0.5457463884430177,
      "grad_norm": 0.5243057664387419,
      "learning_rate": 0.0004510376467126165,
      "loss": 13.6297,
      "step": 1190
    },
    {
      "epoch": 0.546204998853474,
      "grad_norm": 0.5246035759108958,
      "learning_rate": 0.0004502982276319992,
      "loss": 13.5715,
      "step": 1191
    },
    {
      "epoch": 0.5466636092639303,
      "grad_norm": 0.5681457898603588,
      "learning_rate": 0.0004495589183157991,
      "loss": 13.5632,
      "step": 1192
    },
    {
      "epoch": 0.5471222196743866,
      "grad_norm": 0.539982539680145,
      "learning_rate": 0.0004488197203967517,
      "loss": 13.2656,
      "step": 1193
    },
    {
      "epoch": 0.5475808300848429,
      "grad_norm": 0.5803626371021418,
      "learning_rate": 0.0004480806355073467,
      "loss": 13.6716,
      "step": 1194
    },
    {
      "epoch": 0.5480394404952993,
      "grad_norm": 0.6825516175361033,
      "learning_rate": 0.000447341665279824,
      "loss": 13.5102,
      "step": 1195
    },
    {
      "epoch": 0.5484980509057555,
      "grad_norm": 0.5071704420790456,
      "learning_rate": 0.0004466028113461708,
      "loss": 13.3518,
      "step": 1196
    },
    {
      "epoch": 0.5489566613162119,
      "grad_norm": 0.5450323159150845,
      "learning_rate": 0.0004458640753381167,
      "loss": 13.5906,
      "step": 1197
    },
    {
      "epoch": 0.5494152717266682,
      "grad_norm": 0.5356714094993829,
      "learning_rate": 0.0004451254588871313,
      "loss": 13.637,
      "step": 1198
    },
    {
      "epoch": 0.5498738821371245,
      "grad_norm": 0.5179039599783133,
      "learning_rate": 0.0004443869636244203,
      "loss": 13.2315,
      "step": 1199
    },
    {
      "epoch": 0.5503324925475809,
      "grad_norm": 0.5667693723224301,
      "learning_rate": 0.0004436485911809212,
      "loss": 13.5797,
      "step": 1200
    },
    {
      "epoch": 0.5507911029580371,
      "grad_norm": 0.5631724860218962,
      "learning_rate": 0.00044291034318730087,
      "loss": 13.481,
      "step": 1201
    },
    {
      "epoch": 0.5512497133684935,
      "grad_norm": 0.5550839283983419,
      "learning_rate": 0.0004421722212739511,
      "loss": 13.8463,
      "step": 1202
    },
    {
      "epoch": 0.5517083237789497,
      "grad_norm": 0.5779430116656141,
      "learning_rate": 0.0004414342270709848,
      "loss": 13.3491,
      "step": 1203
    },
    {
      "epoch": 0.5521669341894061,
      "grad_norm": 0.5839245143163637,
      "learning_rate": 0.00044069636220823397,
      "loss": 13.5372,
      "step": 1204
    },
    {
      "epoch": 0.5526255445998625,
      "grad_norm": 0.551095027900809,
      "learning_rate": 0.0004399586283152437,
      "loss": 13.5015,
      "step": 1205
    },
    {
      "epoch": 0.5530841550103187,
      "grad_norm": 0.604303328172026,
      "learning_rate": 0.0004392210270212706,
      "loss": 13.8601,
      "step": 1206
    },
    {
      "epoch": 0.5535427654207751,
      "grad_norm": 0.6119876258177532,
      "learning_rate": 0.00043848355995527825,
      "loss": 13.7945,
      "step": 1207
    },
    {
      "epoch": 0.5540013758312313,
      "grad_norm": 0.5822769244991548,
      "learning_rate": 0.00043774622874593374,
      "loss": 13.3575,
      "step": 1208
    },
    {
      "epoch": 0.5544599862416877,
      "grad_norm": 0.5410129167475205,
      "learning_rate": 0.000437009035021604,
      "loss": 13.6865,
      "step": 1209
    },
    {
      "epoch": 0.554918596652144,
      "grad_norm": 0.5295442349805181,
      "learning_rate": 0.00043627198041035274,
      "loss": 13.5642,
      "step": 1210
    },
    {
      "epoch": 0.5553772070626003,
      "grad_norm": 0.5416058136164091,
      "learning_rate": 0.00043553506653993597,
      "loss": 13.5669,
      "step": 1211
    },
    {
      "epoch": 0.5558358174730567,
      "grad_norm": 0.5541394045980501,
      "learning_rate": 0.0004347982950377992,
      "loss": 13.5453,
      "step": 1212
    },
    {
      "epoch": 0.5562944278835129,
      "grad_norm": 0.5266321449604344,
      "learning_rate": 0.0004340616675310735,
      "loss": 13.6074,
      "step": 1213
    },
    {
      "epoch": 0.5567530382939693,
      "grad_norm": 0.6534522643928619,
      "learning_rate": 0.00043332518564657193,
      "loss": 13.6155,
      "step": 1214
    },
    {
      "epoch": 0.5572116487044256,
      "grad_norm": 0.531344941277238,
      "learning_rate": 0.0004325888510107856,
      "loss": 13.5178,
      "step": 1215
    },
    {
      "epoch": 0.5576702591148819,
      "grad_norm": 0.540517608560079,
      "learning_rate": 0.0004318526652498809,
      "loss": 13.6627,
      "step": 1216
    },
    {
      "epoch": 0.5581288695253382,
      "grad_norm": 0.5542465296897525,
      "learning_rate": 0.00043111662998969523,
      "loss": 13.6978,
      "step": 1217
    },
    {
      "epoch": 0.5585874799357945,
      "grad_norm": 0.5595555902711367,
      "learning_rate": 0.0004303807468557335,
      "loss": 13.6157,
      "step": 1218
    },
    {
      "epoch": 0.5590460903462509,
      "grad_norm": 0.5641121766823557,
      "learning_rate": 0.0004296450174731648,
      "loss": 13.5445,
      "step": 1219
    },
    {
      "epoch": 0.5595047007567072,
      "grad_norm": 0.5391366007571351,
      "learning_rate": 0.0004289094434668188,
      "loss": 13.7567,
      "step": 1220
    },
    {
      "epoch": 0.5599633111671635,
      "grad_norm": 0.5335424089586414,
      "learning_rate": 0.00042817402646118185,
      "loss": 13.5832,
      "step": 1221
    },
    {
      "epoch": 0.5604219215776198,
      "grad_norm": 0.5264857018092189,
      "learning_rate": 0.0004274387680803936,
      "loss": 13.5677,
      "step": 1222
    },
    {
      "epoch": 0.5608805319880761,
      "grad_norm": 0.535587489664779,
      "learning_rate": 0.00042670366994824327,
      "loss": 13.5583,
      "step": 1223
    },
    {
      "epoch": 0.5613391423985324,
      "grad_norm": 0.5629009814371878,
      "learning_rate": 0.0004259687336881663,
      "loss": 13.6415,
      "step": 1224
    },
    {
      "epoch": 0.5617977528089888,
      "grad_norm": 0.5497165898586301,
      "learning_rate": 0.0004252339609232408,
      "loss": 13.4163,
      "step": 1225
    },
    {
      "epoch": 0.562256363219445,
      "grad_norm": 0.5595159314894796,
      "learning_rate": 0.0004244993532761834,
      "loss": 13.6635,
      "step": 1226
    },
    {
      "epoch": 0.5627149736299014,
      "grad_norm": 0.5258493346521869,
      "learning_rate": 0.00042376491236934634,
      "loss": 13.5041,
      "step": 1227
    },
    {
      "epoch": 0.5631735840403577,
      "grad_norm": 0.522148190682579,
      "learning_rate": 0.0004230306398247136,
      "loss": 13.6209,
      "step": 1228
    },
    {
      "epoch": 0.563632194450814,
      "grad_norm": 0.540154019109576,
      "learning_rate": 0.0004222965372638976,
      "loss": 13.4443,
      "step": 1229
    },
    {
      "epoch": 0.5640908048612704,
      "grad_norm": 0.5888649021566148,
      "learning_rate": 0.0004215626063081348,
      "loss": 13.6427,
      "step": 1230
    },
    {
      "epoch": 0.5645494152717266,
      "grad_norm": 0.5215784213097057,
      "learning_rate": 0.000420828848578283,
      "loss": 13.6406,
      "step": 1231
    },
    {
      "epoch": 0.565008025682183,
      "grad_norm": 0.5121763813859848,
      "learning_rate": 0.0004200952656948175,
      "loss": 13.6665,
      "step": 1232
    },
    {
      "epoch": 0.5654666360926393,
      "grad_norm": 0.5174619449821071,
      "learning_rate": 0.0004193618592778272,
      "loss": 13.595,
      "step": 1233
    },
    {
      "epoch": 0.5659252465030956,
      "grad_norm": 0.5775028780695437,
      "learning_rate": 0.0004186286309470116,
      "loss": 13.5537,
      "step": 1234
    },
    {
      "epoch": 0.566383856913552,
      "grad_norm": 0.605162343434185,
      "learning_rate": 0.0004178955823216767,
      "loss": 13.6395,
      "step": 1235
    },
    {
      "epoch": 0.5668424673240082,
      "grad_norm": 0.5775093041275305,
      "learning_rate": 0.00041716271502073137,
      "loss": 13.6391,
      "step": 1236
    },
    {
      "epoch": 0.5673010777344646,
      "grad_norm": 0.5096021217361896,
      "learning_rate": 0.000416430030662685,
      "loss": 13.626,
      "step": 1237
    },
    {
      "epoch": 0.5677596881449208,
      "grad_norm": 0.5764151855520939,
      "learning_rate": 0.00041569753086564173,
      "loss": 13.6176,
      "step": 1238
    },
    {
      "epoch": 0.5682182985553772,
      "grad_norm": 0.5638341985621573,
      "learning_rate": 0.0004149652172472988,
      "loss": 13.6781,
      "step": 1239
    },
    {
      "epoch": 0.5686769089658336,
      "grad_norm": 0.5338980295704174,
      "learning_rate": 0.00041423309142494234,
      "loss": 13.5924,
      "step": 1240
    },
    {
      "epoch": 0.5691355193762898,
      "grad_norm": 0.5406011224807871,
      "learning_rate": 0.0004135011550154433,
      "loss": 13.5027,
      "step": 1241
    },
    {
      "epoch": 0.5695941297867462,
      "grad_norm": 0.5283464532826336,
      "learning_rate": 0.0004127694096352546,
      "loss": 13.4957,
      "step": 1242
    },
    {
      "epoch": 0.5700527401972024,
      "grad_norm": 0.5810667699871931,
      "learning_rate": 0.00041203785690040743,
      "loss": 13.6126,
      "step": 1243
    },
    {
      "epoch": 0.5705113506076588,
      "grad_norm": 0.5731300438456958,
      "learning_rate": 0.00041130649842650694,
      "loss": 13.5426,
      "step": 1244
    },
    {
      "epoch": 0.5709699610181151,
      "grad_norm": 0.5497954507139946,
      "learning_rate": 0.00041057533582873016,
      "loss": 13.6056,
      "step": 1245
    },
    {
      "epoch": 0.5714285714285714,
      "grad_norm": 0.5320870236239896,
      "learning_rate": 0.0004098443707218208,
      "loss": 13.6322,
      "step": 1246
    },
    {
      "epoch": 0.5718871818390278,
      "grad_norm": 0.5411876192476316,
      "learning_rate": 0.00040911360472008673,
      "loss": 13.535,
      "step": 1247
    },
    {
      "epoch": 0.572345792249484,
      "grad_norm": 0.5656281989162569,
      "learning_rate": 0.0004083830394373959,
      "loss": 13.6768,
      "step": 1248
    },
    {
      "epoch": 0.5728044026599404,
      "grad_norm": 0.5698189383771234,
      "learning_rate": 0.00040765267648717324,
      "loss": 13.4568,
      "step": 1249
    },
    {
      "epoch": 0.5732630130703967,
      "grad_norm": 0.5590706551323109,
      "learning_rate": 0.00040692251748239677,
      "loss": 13.4142,
      "step": 1250
    },
    {
      "epoch": 0.573721623480853,
      "grad_norm": 0.5425124107488488,
      "learning_rate": 0.00040619256403559383,
      "loss": 13.626,
      "step": 1251
    },
    {
      "epoch": 0.5741802338913093,
      "grad_norm": 0.5254112130158691,
      "learning_rate": 0.000405462817758838,
      "loss": 13.5359,
      "step": 1252
    },
    {
      "epoch": 0.5746388443017657,
      "grad_norm": 0.557032550901202,
      "learning_rate": 0.0004047332802637457,
      "loss": 13.6519,
      "step": 1253
    },
    {
      "epoch": 0.575097454712222,
      "grad_norm": 0.6621532439718544,
      "learning_rate": 0.00040400395316147157,
      "loss": 13.4587,
      "step": 1254
    },
    {
      "epoch": 0.5755560651226783,
      "grad_norm": 0.5558431019314388,
      "learning_rate": 0.00040327483806270627,
      "loss": 13.4451,
      "step": 1255
    },
    {
      "epoch": 0.5760146755331346,
      "grad_norm": 0.605934844275572,
      "learning_rate": 0.0004025459365776715,
      "loss": 13.4982,
      "step": 1256
    },
    {
      "epoch": 0.5764732859435909,
      "grad_norm": 0.5641320839326336,
      "learning_rate": 0.00040181725031611794,
      "loss": 13.645,
      "step": 1257
    },
    {
      "epoch": 0.5769318963540473,
      "grad_norm": 0.5582977359645971,
      "learning_rate": 0.0004010887808873206,
      "loss": 13.4717,
      "step": 1258
    },
    {
      "epoch": 0.5773905067645035,
      "grad_norm": 0.5352262199735797,
      "learning_rate": 0.00040036052990007553,
      "loss": 13.7326,
      "step": 1259
    },
    {
      "epoch": 0.5778491171749599,
      "grad_norm": 0.5295860634984869,
      "learning_rate": 0.0003996324989626967,
      "loss": 13.5295,
      "step": 1260
    },
    {
      "epoch": 0.5783077275854162,
      "grad_norm": 0.5277566891307411,
      "learning_rate": 0.00039890468968301166,
      "loss": 13.5032,
      "step": 1261
    },
    {
      "epoch": 0.5787663379958725,
      "grad_norm": 0.5374321162977763,
      "learning_rate": 0.0003981771036683591,
      "loss": 13.466,
      "step": 1262
    },
    {
      "epoch": 0.5792249484063289,
      "grad_norm": 0.5840743890175307,
      "learning_rate": 0.00039744974252558385,
      "loss": 13.6107,
      "step": 1263
    },
    {
      "epoch": 0.5796835588167851,
      "grad_norm": 0.5407655640271575,
      "learning_rate": 0.00039672260786103463,
      "loss": 13.5543,
      "step": 1264
    },
    {
      "epoch": 0.5801421692272415,
      "grad_norm": 0.5427406626552391,
      "learning_rate": 0.00039599570128055994,
      "loss": 13.7735,
      "step": 1265
    },
    {
      "epoch": 0.5806007796376977,
      "grad_norm": 0.674005459518343,
      "learning_rate": 0.0003952690243895044,
      "loss": 13.6046,
      "step": 1266
    },
    {
      "epoch": 0.5810593900481541,
      "grad_norm": 0.5294139024909046,
      "learning_rate": 0.0003945425787927054,
      "loss": 13.5999,
      "step": 1267
    },
    {
      "epoch": 0.5815180004586105,
      "grad_norm": 0.5170552876608634,
      "learning_rate": 0.00039381636609448975,
      "loss": 13.4929,
      "step": 1268
    },
    {
      "epoch": 0.5819766108690667,
      "grad_norm": 0.5101290746332502,
      "learning_rate": 0.0003930903878986693,
      "loss": 13.2936,
      "step": 1269
    },
    {
      "epoch": 0.5824352212795231,
      "grad_norm": 0.5556928620057082,
      "learning_rate": 0.00039236464580853916,
      "loss": 13.6393,
      "step": 1270
    },
    {
      "epoch": 0.5828938316899793,
      "grad_norm": 0.5506526889135192,
      "learning_rate": 0.0003916391414268718,
      "loss": 13.7136,
      "step": 1271
    },
    {
      "epoch": 0.5833524421004357,
      "grad_norm": 0.512975724375407,
      "learning_rate": 0.00039091387635591536,
      "loss": 13.4452,
      "step": 1272
    },
    {
      "epoch": 0.583811052510892,
      "grad_norm": 0.5425865477789773,
      "learning_rate": 0.0003901888521973894,
      "loss": 13.7229,
      "step": 1273
    },
    {
      "epoch": 0.5842696629213483,
      "grad_norm": 0.5051999207640931,
      "learning_rate": 0.0003894640705524813,
      "loss": 13.7031,
      "step": 1274
    },
    {
      "epoch": 0.5847282733318047,
      "grad_norm": 0.5713126552633427,
      "learning_rate": 0.00038873953302184284,
      "loss": 13.2536,
      "step": 1275
    },
    {
      "epoch": 0.5851868837422609,
      "grad_norm": 0.568923450094778,
      "learning_rate": 0.000388015241205587,
      "loss": 13.6497,
      "step": 1276
    },
    {
      "epoch": 0.5856454941527173,
      "grad_norm": 0.5663065820593444,
      "learning_rate": 0.00038729119670328355,
      "loss": 13.289,
      "step": 1277
    },
    {
      "epoch": 0.5861041045631736,
      "grad_norm": 0.6227252617431068,
      "learning_rate": 0.00038656740111395665,
      "loss": 13.7593,
      "step": 1278
    },
    {
      "epoch": 0.5865627149736299,
      "grad_norm": 0.5665255046956167,
      "learning_rate": 0.00038584385603608053,
      "loss": 13.6329,
      "step": 1279
    },
    {
      "epoch": 0.5870213253840862,
      "grad_norm": 0.5799119409056284,
      "learning_rate": 0.00038512056306757615,
      "loss": 13.4554,
      "step": 1280
    },
    {
      "epoch": 0.5874799357945425,
      "grad_norm": 0.5587483857682105,
      "learning_rate": 0.0003843975238058075,
      "loss": 13.3954,
      "step": 1281
    },
    {
      "epoch": 0.5879385462049989,
      "grad_norm": 0.5650178763580851,
      "learning_rate": 0.00038367473984757863,
      "loss": 13.7413,
      "step": 1282
    },
    {
      "epoch": 0.5883971566154552,
      "grad_norm": 0.5369815396855556,
      "learning_rate": 0.0003829522127891296,
      "loss": 13.4637,
      "step": 1283
    },
    {
      "epoch": 0.5888557670259115,
      "grad_norm": 0.6149579041930265,
      "learning_rate": 0.0003822299442261329,
      "loss": 13.6339,
      "step": 1284
    },
    {
      "epoch": 0.5893143774363678,
      "grad_norm": 0.592216491060348,
      "learning_rate": 0.00038150793575369063,
      "loss": 13.4025,
      "step": 1285
    },
    {
      "epoch": 0.5897729878468241,
      "grad_norm": 0.51873527676158,
      "learning_rate": 0.0003807861889663299,
      "loss": 13.6429,
      "step": 1286
    },
    {
      "epoch": 0.5902315982572804,
      "grad_norm": 0.6829498647548815,
      "learning_rate": 0.0003800647054580006,
      "loss": 13.4925,
      "step": 1287
    },
    {
      "epoch": 0.5906902086677368,
      "grad_norm": 0.5503064871500964,
      "learning_rate": 0.00037934348682207064,
      "loss": 13.4451,
      "step": 1288
    },
    {
      "epoch": 0.591148819078193,
      "grad_norm": 0.5526736073569668,
      "learning_rate": 0.00037862253465132306,
      "loss": 13.6655,
      "step": 1289
    },
    {
      "epoch": 0.5916074294886494,
      "grad_norm": 0.6102039072724988,
      "learning_rate": 0.00037790185053795245,
      "loss": 13.7055,
      "step": 1290
    },
    {
      "epoch": 0.5920660398991057,
      "grad_norm": 0.5416978141889558,
      "learning_rate": 0.0003771814360735616,
      "loss": 13.6228,
      "step": 1291
    },
    {
      "epoch": 0.592524650309562,
      "grad_norm": 0.5302757632835063,
      "learning_rate": 0.00037646129284915755,
      "loss": 13.7591,
      "step": 1292
    },
    {
      "epoch": 0.5929832607200184,
      "grad_norm": 0.5488503017221279,
      "learning_rate": 0.00037574142245514825,
      "loss": 13.5125,
      "step": 1293
    },
    {
      "epoch": 0.5934418711304746,
      "grad_norm": 0.5828807431151436,
      "learning_rate": 0.0003750218264813393,
      "loss": 13.4916,
      "step": 1294
    },
    {
      "epoch": 0.593900481540931,
      "grad_norm": 0.529133291057101,
      "learning_rate": 0.0003743025065169305,
      "loss": 13.6017,
      "step": 1295
    },
    {
      "epoch": 0.5943590919513873,
      "grad_norm": 0.5602658799876582,
      "learning_rate": 0.0003735834641505116,
      "loss": 13.5173,
      "step": 1296
    },
    {
      "epoch": 0.5948177023618436,
      "grad_norm": 0.5256617342492765,
      "learning_rate": 0.00037286470097005954,
      "loss": 13.4848,
      "step": 1297
    },
    {
      "epoch": 0.5952763127723,
      "grad_norm": 0.5523304128655313,
      "learning_rate": 0.0003721462185629347,
      "loss": 13.4918,
      "step": 1298
    },
    {
      "epoch": 0.5957349231827562,
      "grad_norm": 0.5122720008516294,
      "learning_rate": 0.00037142801851587707,
      "loss": 13.4595,
      "step": 1299
    },
    {
      "epoch": 0.5961935335932126,
      "grad_norm": 0.5192824728022998,
      "learning_rate": 0.00037071010241500357,
      "loss": 13.4535,
      "step": 1300
    },
    {
      "epoch": 0.5966521440036688,
      "grad_norm": 0.4821307953791148,
      "learning_rate": 0.00036999247184580383,
      "loss": 13.3774,
      "step": 1301
    },
    {
      "epoch": 0.5971107544141252,
      "grad_norm": 0.5312866842966091,
      "learning_rate": 0.00036927512839313636,
      "loss": 13.3337,
      "step": 1302
    },
    {
      "epoch": 0.5975693648245816,
      "grad_norm": 0.5196109810038507,
      "learning_rate": 0.0003685580736412268,
      "loss": 13.5911,
      "step": 1303
    },
    {
      "epoch": 0.5980279752350378,
      "grad_norm": 0.5673597285159359,
      "learning_rate": 0.000367841309173662,
      "loss": 13.4894,
      "step": 1304
    },
    {
      "epoch": 0.5984865856454942,
      "grad_norm": 0.5286802010960782,
      "learning_rate": 0.0003671248365733883,
      "loss": 13.4358,
      "step": 1305
    },
    {
      "epoch": 0.5989451960559504,
      "grad_norm": 0.5231666053005086,
      "learning_rate": 0.0003664086574227075,
      "loss": 13.6178,
      "step": 1306
    },
    {
      "epoch": 0.5994038064664068,
      "grad_norm": 0.530594387892871,
      "learning_rate": 0.000365692773303273,
      "loss": 13.3612,
      "step": 1307
    },
    {
      "epoch": 0.5998624168768631,
      "grad_norm": 0.5443838580041249,
      "learning_rate": 0.00036497718579608696,
      "loss": 13.5872,
      "step": 1308
    },
    {
      "epoch": 0.6003210272873194,
      "grad_norm": 0.5963247557776911,
      "learning_rate": 0.0003642618964814964,
      "loss": 13.719,
      "step": 1309
    },
    {
      "epoch": 0.6007796376977758,
      "grad_norm": 0.5510694368291167,
      "learning_rate": 0.00036354690693918946,
      "loss": 13.6538,
      "step": 1310
    },
    {
      "epoch": 0.601238248108232,
      "grad_norm": 0.575155824129918,
      "learning_rate": 0.00036283221874819284,
      "loss": 13.7663,
      "step": 1311
    },
    {
      "epoch": 0.6016968585186884,
      "grad_norm": 0.550860874359012,
      "learning_rate": 0.0003621178334868672,
      "loss": 13.5483,
      "step": 1312
    },
    {
      "epoch": 0.6021554689291447,
      "grad_norm": 0.5833668761245565,
      "learning_rate": 0.00036140375273290476,
      "loss": 13.5846,
      "step": 1313
    },
    {
      "epoch": 0.602614079339601,
      "grad_norm": 0.5401470798884267,
      "learning_rate": 0.0003606899780633245,
      "loss": 13.4776,
      "step": 1314
    },
    {
      "epoch": 0.6030726897500573,
      "grad_norm": 0.5613192821561401,
      "learning_rate": 0.0003599765110544699,
      "loss": 13.4695,
      "step": 1315
    },
    {
      "epoch": 0.6035313001605136,
      "grad_norm": 0.5428770111304969,
      "learning_rate": 0.0003592633532820052,
      "loss": 13.4459,
      "step": 1316
    },
    {
      "epoch": 0.60398991057097,
      "grad_norm": 0.5416818819474728,
      "learning_rate": 0.0003585505063209109,
      "loss": 13.7018,
      "step": 1317
    },
    {
      "epoch": 0.6044485209814263,
      "grad_norm": 0.5500503360966357,
      "learning_rate": 0.00035783797174548194,
      "loss": 13.5099,
      "step": 1318
    },
    {
      "epoch": 0.6049071313918826,
      "grad_norm": 0.5503555952296414,
      "learning_rate": 0.00035712575112932277,
      "loss": 13.7017,
      "step": 1319
    },
    {
      "epoch": 0.6053657418023389,
      "grad_norm": 0.5580529543265668,
      "learning_rate": 0.000356413846045345,
      "loss": 13.4153,
      "step": 1320
    },
    {
      "epoch": 0.6058243522127952,
      "grad_norm": 0.5404452642374905,
      "learning_rate": 0.000355702258065763,
      "loss": 13.5955,
      "step": 1321
    },
    {
      "epoch": 0.6062829626232515,
      "grad_norm": 0.5776959896312601,
      "learning_rate": 0.0003549909887620909,
      "loss": 13.4748,
      "step": 1322
    },
    {
      "epoch": 0.6067415730337079,
      "grad_norm": 0.5344818766833828,
      "learning_rate": 0.00035428003970513914,
      "loss": 13.7441,
      "step": 1323
    },
    {
      "epoch": 0.6072001834441642,
      "grad_norm": 0.520725108267128,
      "learning_rate": 0.00035356941246501085,
      "loss": 13.3331,
      "step": 1324
    },
    {
      "epoch": 0.6076587938546205,
      "grad_norm": 0.5663071212677051,
      "learning_rate": 0.0003528591086110984,
      "loss": 13.4885,
      "step": 1325
    },
    {
      "epoch": 0.6081174042650768,
      "grad_norm": 0.49165330054959866,
      "learning_rate": 0.00035214912971208,
      "loss": 13.183,
      "step": 1326
    },
    {
      "epoch": 0.6085760146755331,
      "grad_norm": 0.5524290776255246,
      "learning_rate": 0.0003514394773359163,
      "loss": 13.593,
      "step": 1327
    },
    {
      "epoch": 0.6090346250859895,
      "grad_norm": 0.5412818035703139,
      "learning_rate": 0.0003507301530498469,
      "loss": 13.3394,
      "step": 1328
    },
    {
      "epoch": 0.6094932354964457,
      "grad_norm": 0.5796245680645166,
      "learning_rate": 0.00035002115842038646,
      "loss": 13.4956,
      "step": 1329
    },
    {
      "epoch": 0.6099518459069021,
      "grad_norm": 0.5673650110524947,
      "learning_rate": 0.00034931249501332195,
      "loss": 13.5543,
      "step": 1330
    },
    {
      "epoch": 0.6104104563173584,
      "grad_norm": 0.5803808348022924,
      "learning_rate": 0.00034860416439370885,
      "loss": 13.5262,
      "step": 1331
    },
    {
      "epoch": 0.6108690667278147,
      "grad_norm": 0.5232796989381908,
      "learning_rate": 0.0003478961681258674,
      "loss": 13.2535,
      "step": 1332
    },
    {
      "epoch": 0.6113276771382711,
      "grad_norm": 0.5258075389854315,
      "learning_rate": 0.0003471885077733796,
      "loss": 13.4051,
      "step": 1333
    },
    {
      "epoch": 0.6117862875487273,
      "grad_norm": 0.6120004894552755,
      "learning_rate": 0.0003464811848990859,
      "loss": 13.5443,
      "step": 1334
    },
    {
      "epoch": 0.6122448979591837,
      "grad_norm": 0.5448298816725067,
      "learning_rate": 0.00034577420106508063,
      "loss": 13.2597,
      "step": 1335
    },
    {
      "epoch": 0.6127035083696399,
      "grad_norm": 0.5404951519757387,
      "learning_rate": 0.0003450675578327105,
      "loss": 13.3413,
      "step": 1336
    },
    {
      "epoch": 0.6131621187800963,
      "grad_norm": 0.5836123205090664,
      "learning_rate": 0.000344361256762569,
      "loss": 13.4853,
      "step": 1337
    },
    {
      "epoch": 0.6136207291905527,
      "grad_norm": 0.5522214585112442,
      "learning_rate": 0.00034365529941449456,
      "loss": 13.5603,
      "step": 1338
    },
    {
      "epoch": 0.6140793396010089,
      "grad_norm": 0.5307244296078154,
      "learning_rate": 0.0003429496873475664,
      "loss": 13.4894,
      "step": 1339
    },
    {
      "epoch": 0.6145379500114653,
      "grad_norm": 0.561738999289274,
      "learning_rate": 0.0003422444221201009,
      "loss": 13.5472,
      "step": 1340
    },
    {
      "epoch": 0.6149965604219215,
      "grad_norm": 0.5661250374135802,
      "learning_rate": 0.0003415395052896487,
      "loss": 13.5845,
      "step": 1341
    },
    {
      "epoch": 0.6154551708323779,
      "grad_norm": 0.5527586190261469,
      "learning_rate": 0.0003408349384129912,
      "loss": 13.4356,
      "step": 1342
    },
    {
      "epoch": 0.6159137812428342,
      "grad_norm": 0.5807100317681049,
      "learning_rate": 0.00034013072304613643,
      "loss": 13.3264,
      "step": 1343
    },
    {
      "epoch": 0.6163723916532905,
      "grad_norm": 0.5760522336316499,
      "learning_rate": 0.00033942686074431674,
      "loss": 13.3411,
      "step": 1344
    },
    {
      "epoch": 0.6168310020637469,
      "grad_norm": 0.5191405813258048,
      "learning_rate": 0.0003387233530619843,
      "loss": 13.5519,
      "step": 1345
    },
    {
      "epoch": 0.6172896124742032,
      "grad_norm": 0.514256143938495,
      "learning_rate": 0.0003380202015528084,
      "loss": 13.5243,
      "step": 1346
    },
    {
      "epoch": 0.6177482228846595,
      "grad_norm": 0.5593288401586259,
      "learning_rate": 0.0003373174077696715,
      "loss": 13.4908,
      "step": 1347
    },
    {
      "epoch": 0.6182068332951158,
      "grad_norm": 0.6181731321323357,
      "learning_rate": 0.0003366149732646661,
      "loss": 13.2634,
      "step": 1348
    },
    {
      "epoch": 0.6186654437055721,
      "grad_norm": 0.49725723007611966,
      "learning_rate": 0.00033591289958909143,
      "loss": 13.4227,
      "step": 1349
    },
    {
      "epoch": 0.6191240541160284,
      "grad_norm": 0.5909700641322707,
      "learning_rate": 0.00033521118829344954,
      "loss": 13.7762,
      "step": 1350
    },
    {
      "epoch": 0.6195826645264848,
      "grad_norm": 0.5430735265713996,
      "learning_rate": 0.0003345098409274423,
      "loss": 13.5135,
      "step": 1351
    },
    {
      "epoch": 0.6200412749369411,
      "grad_norm": 0.5370716064352082,
      "learning_rate": 0.00033380885903996796,
      "loss": 13.565,
      "step": 1352
    },
    {
      "epoch": 0.6204998853473974,
      "grad_norm": 0.5436605456298688,
      "learning_rate": 0.00033310824417911766,
      "loss": 13.6083,
      "step": 1353
    },
    {
      "epoch": 0.6209584957578537,
      "grad_norm": 0.5579336246830834,
      "learning_rate": 0.00033240799789217184,
      "loss": 13.5199,
      "step": 1354
    },
    {
      "epoch": 0.62141710616831,
      "grad_norm": 0.5818676682668728,
      "learning_rate": 0.00033170812172559694,
      "loss": 13.4529,
      "step": 1355
    },
    {
      "epoch": 0.6218757165787664,
      "grad_norm": 0.5490663202878974,
      "learning_rate": 0.000331008617225042,
      "loss": 13.5229,
      "step": 1356
    },
    {
      "epoch": 0.6223343269892226,
      "grad_norm": 0.6335703570895792,
      "learning_rate": 0.0003303094859353355,
      "loss": 13.5989,
      "step": 1357
    },
    {
      "epoch": 0.622792937399679,
      "grad_norm": 0.540645479199579,
      "learning_rate": 0.0003296107294004812,
      "loss": 13.4929,
      "step": 1358
    },
    {
      "epoch": 0.6232515478101353,
      "grad_norm": 0.5397285619479745,
      "learning_rate": 0.0003289123491636559,
      "loss": 13.6746,
      "step": 1359
    },
    {
      "epoch": 0.6237101582205916,
      "grad_norm": 0.6155499275403323,
      "learning_rate": 0.00032821434676720443,
      "loss": 13.4056,
      "step": 1360
    },
    {
      "epoch": 0.624168768631048,
      "grad_norm": 0.5407492701960069,
      "learning_rate": 0.00032751672375263836,
      "loss": 13.5944,
      "step": 1361
    },
    {
      "epoch": 0.6246273790415042,
      "grad_norm": 0.5460973558047634,
      "learning_rate": 0.0003268194816606305,
      "loss": 13.6177,
      "step": 1362
    },
    {
      "epoch": 0.6250859894519606,
      "grad_norm": 0.5124298407489012,
      "learning_rate": 0.00032612262203101267,
      "loss": 13.3317,
      "step": 1363
    },
    {
      "epoch": 0.6255445998624168,
      "grad_norm": 0.5102478969256616,
      "learning_rate": 0.00032542614640277225,
      "loss": 13.575,
      "step": 1364
    },
    {
      "epoch": 0.6260032102728732,
      "grad_norm": 0.4990517697811014,
      "learning_rate": 0.0003247300563140481,
      "loss": 13.6928,
      "step": 1365
    },
    {
      "epoch": 0.6264618206833296,
      "grad_norm": 0.5746873691796199,
      "learning_rate": 0.00032403435330212807,
      "loss": 13.6174,
      "step": 1366
    },
    {
      "epoch": 0.6269204310937858,
      "grad_norm": 0.5854488560513986,
      "learning_rate": 0.00032333903890344515,
      "loss": 13.5621,
      "step": 1367
    },
    {
      "epoch": 0.6273790415042422,
      "grad_norm": 0.5167466085756917,
      "learning_rate": 0.00032264411465357333,
      "loss": 13.6293,
      "step": 1368
    },
    {
      "epoch": 0.6278376519146984,
      "grad_norm": 0.5092338454742771,
      "learning_rate": 0.00032194958208722654,
      "loss": 13.2245,
      "step": 1369
    },
    {
      "epoch": 0.6282962623251548,
      "grad_norm": 0.5552611484586651,
      "learning_rate": 0.00032125544273825204,
      "loss": 13.6553,
      "step": 1370
    },
    {
      "epoch": 0.6287548727356111,
      "grad_norm": 0.5367882876399124,
      "learning_rate": 0.0003205616981396297,
      "loss": 13.3257,
      "step": 1371
    },
    {
      "epoch": 0.6292134831460674,
      "grad_norm": 0.5787005495503131,
      "learning_rate": 0.00031986834982346713,
      "loss": 13.6762,
      "step": 1372
    },
    {
      "epoch": 0.6296720935565238,
      "grad_norm": 0.5329682010355219,
      "learning_rate": 0.00031917539932099694,
      "loss": 13.6225,
      "step": 1373
    },
    {
      "epoch": 0.63013070396698,
      "grad_norm": 0.5161542481464411,
      "learning_rate": 0.00031848284816257336,
      "loss": 13.5768,
      "step": 1374
    },
    {
      "epoch": 0.6305893143774364,
      "grad_norm": 0.5888698344147439,
      "learning_rate": 0.0003177906978776682,
      "loss": 13.6492,
      "step": 1375
    },
    {
      "epoch": 0.6310479247878927,
      "grad_norm": 0.5559186695706811,
      "learning_rate": 0.0003170989499948683,
      "loss": 13.7343,
      "step": 1376
    },
    {
      "epoch": 0.631506535198349,
      "grad_norm": 0.5385139095228436,
      "learning_rate": 0.0003164076060418719,
      "loss": 13.4631,
      "step": 1377
    },
    {
      "epoch": 0.6319651456088053,
      "grad_norm": 0.5311511921681655,
      "learning_rate": 0.000315716667545485,
      "loss": 13.5231,
      "step": 1378
    },
    {
      "epoch": 0.6324237560192616,
      "grad_norm": 0.506971795142387,
      "learning_rate": 0.00031502613603161836,
      "loss": 13.5761,
      "step": 1379
    },
    {
      "epoch": 0.632882366429718,
      "grad_norm": 0.5380392404810582,
      "learning_rate": 0.00031433601302528335,
      "loss": 13.5146,
      "step": 1380
    },
    {
      "epoch": 0.6333409768401743,
      "grad_norm": 0.5484517357818066,
      "learning_rate": 0.00031364630005058995,
      "loss": 13.5431,
      "step": 1381
    },
    {
      "epoch": 0.6337995872506306,
      "grad_norm": 0.5330835223987049,
      "learning_rate": 0.0003129569986307422,
      "loss": 13.3503,
      "step": 1382
    },
    {
      "epoch": 0.6342581976610869,
      "grad_norm": 0.561714275107512,
      "learning_rate": 0.00031226811028803515,
      "loss": 13.4393,
      "step": 1383
    },
    {
      "epoch": 0.6347168080715432,
      "grad_norm": 0.5593698961958126,
      "learning_rate": 0.00031157963654385173,
      "loss": 13.636,
      "step": 1384
    },
    {
      "epoch": 0.6351754184819995,
      "grad_norm": 0.5349679664844316,
      "learning_rate": 0.0003108915789186592,
      "loss": 13.4459,
      "step": 1385
    },
    {
      "epoch": 0.6356340288924559,
      "grad_norm": 0.559459996833476,
      "learning_rate": 0.00031020393893200604,
      "loss": 13.4701,
      "step": 1386
    },
    {
      "epoch": 0.6360926393029122,
      "grad_norm": 0.5561158209690376,
      "learning_rate": 0.00030951671810251823,
      "loss": 13.5144,
      "step": 1387
    },
    {
      "epoch": 0.6365512497133685,
      "grad_norm": 0.5548883138828343,
      "learning_rate": 0.0003088299179478959,
      "loss": 13.5842,
      "step": 1388
    },
    {
      "epoch": 0.6370098601238248,
      "grad_norm": 0.5445143532833784,
      "learning_rate": 0.0003081435399849104,
      "loss": 13.5883,
      "step": 1389
    },
    {
      "epoch": 0.6374684705342811,
      "grad_norm": 0.5477117406350254,
      "learning_rate": 0.0003074575857294004,
      "loss": 13.7229,
      "step": 1390
    },
    {
      "epoch": 0.6379270809447375,
      "grad_norm": 0.5504587775369144,
      "learning_rate": 0.0003067720566962691,
      "loss": 13.6193,
      "step": 1391
    },
    {
      "epoch": 0.6383856913551937,
      "grad_norm": 0.5463002769742997,
      "learning_rate": 0.0003060869543994806,
      "loss": 13.4815,
      "step": 1392
    },
    {
      "epoch": 0.6388443017656501,
      "grad_norm": 0.5451207735821165,
      "learning_rate": 0.0003054022803520562,
      "loss": 13.6207,
      "step": 1393
    },
    {
      "epoch": 0.6393029121761064,
      "grad_norm": 0.5681585328995864,
      "learning_rate": 0.0003047180360660721,
      "loss": 13.5506,
      "step": 1394
    },
    {
      "epoch": 0.6397615225865627,
      "grad_norm": 0.5497037765747456,
      "learning_rate": 0.00030403422305265475,
      "loss": 13.4298,
      "step": 1395
    },
    {
      "epoch": 0.6402201329970191,
      "grad_norm": 0.543012452167815,
      "learning_rate": 0.0003033508428219785,
      "loss": 13.6597,
      "step": 1396
    },
    {
      "epoch": 0.6406787434074753,
      "grad_norm": 0.6200714575058316,
      "learning_rate": 0.00030266789688326184,
      "loss": 13.5372,
      "step": 1397
    },
    {
      "epoch": 0.6411373538179317,
      "grad_norm": 0.5491584508346852,
      "learning_rate": 0.00030198538674476393,
      "loss": 13.6003,
      "step": 1398
    },
    {
      "epoch": 0.6415959642283879,
      "grad_norm": 1.2492448291418197,
      "learning_rate": 0.00030130331391378185,
      "loss": 13.4634,
      "step": 1399
    },
    {
      "epoch": 0.6420545746388443,
      "grad_norm": 0.5570645109150317,
      "learning_rate": 0.0003006216798966468,
      "loss": 13.5514,
      "step": 1400
    },
    {
      "epoch": 0.6425131850493007,
      "grad_norm": 0.5498417311210865,
      "learning_rate": 0.00029994048619872034,
      "loss": 13.6332,
      "step": 1401
    },
    {
      "epoch": 0.6429717954597569,
      "grad_norm": 0.5218166273344833,
      "learning_rate": 0.0002992597343243927,
      "loss": 13.6423,
      "step": 1402
    },
    {
      "epoch": 0.6434304058702133,
      "grad_norm": 0.9196492796552431,
      "learning_rate": 0.0002985794257770773,
      "loss": 13.7836,
      "step": 1403
    },
    {
      "epoch": 0.6438890162806695,
      "grad_norm": 0.56398225670722,
      "learning_rate": 0.0002978995620592092,
      "loss": 13.7169,
      "step": 1404
    },
    {
      "epoch": 0.6443476266911259,
      "grad_norm": 0.5424169924738933,
      "learning_rate": 0.0002972201446722405,
      "loss": 13.309,
      "step": 1405
    },
    {
      "epoch": 0.6448062371015822,
      "grad_norm": 0.5508830381413284,
      "learning_rate": 0.00029654117511663803,
      "loss": 13.5321,
      "step": 1406
    },
    {
      "epoch": 0.6452648475120385,
      "grad_norm": 0.5776048983647247,
      "learning_rate": 0.0002958626548918795,
      "loss": 13.4332,
      "step": 1407
    },
    {
      "epoch": 0.6457234579224949,
      "grad_norm": 0.5759481337097193,
      "learning_rate": 0.00029518458549645014,
      "loss": 13.3877,
      "step": 1408
    },
    {
      "epoch": 0.6461820683329511,
      "grad_norm": 0.5223002627897635,
      "learning_rate": 0.00029450696842783954,
      "loss": 13.5292,
      "step": 1409
    },
    {
      "epoch": 0.6466406787434075,
      "grad_norm": 0.5880506450593801,
      "learning_rate": 0.00029382980518253865,
      "loss": 13.5888,
      "step": 1410
    },
    {
      "epoch": 0.6470992891538638,
      "grad_norm": 0.5897923640184561,
      "learning_rate": 0.00029315309725603595,
      "loss": 13.6684,
      "step": 1411
    },
    {
      "epoch": 0.6475578995643201,
      "grad_norm": 0.5270144980033254,
      "learning_rate": 0.00029247684614281446,
      "loss": 13.5778,
      "step": 1412
    },
    {
      "epoch": 0.6480165099747764,
      "grad_norm": 0.5778292619614019,
      "learning_rate": 0.0002918010533363481,
      "loss": 13.6338,
      "step": 1413
    },
    {
      "epoch": 0.6484751203852327,
      "grad_norm": 0.6259060476283632,
      "learning_rate": 0.0002911257203290987,
      "loss": 13.6136,
      "step": 1414
    },
    {
      "epoch": 0.6489337307956891,
      "grad_norm": 0.543011631704543,
      "learning_rate": 0.00029045084861251314,
      "loss": 13.7951,
      "step": 1415
    },
    {
      "epoch": 0.6493923412061454,
      "grad_norm": 0.5218334935830169,
      "learning_rate": 0.00028977643967701897,
      "loss": 13.355,
      "step": 1416
    },
    {
      "epoch": 0.6498509516166017,
      "grad_norm": 0.5868989526768155,
      "learning_rate": 0.00028910249501202156,
      "loss": 13.7561,
      "step": 1417
    },
    {
      "epoch": 0.650309562027058,
      "grad_norm": 0.5515366691313479,
      "learning_rate": 0.00028842901610590165,
      "loss": 13.7943,
      "step": 1418
    },
    {
      "epoch": 0.6507681724375143,
      "grad_norm": 0.526009140163767,
      "learning_rate": 0.00028775600444601123,
      "loss": 13.2867,
      "step": 1419
    },
    {
      "epoch": 0.6512267828479706,
      "grad_norm": 0.5526232137966265,
      "learning_rate": 0.00028708346151866973,
      "loss": 13.7035,
      "step": 1420
    },
    {
      "epoch": 0.651685393258427,
      "grad_norm": 0.5503509083957266,
      "learning_rate": 0.0002864113888091622,
      "loss": 13.5901,
      "step": 1421
    },
    {
      "epoch": 0.6521440036688833,
      "grad_norm": 0.5971505963020158,
      "learning_rate": 0.0002857397878017348,
      "loss": 13.4949,
      "step": 1422
    },
    {
      "epoch": 0.6526026140793396,
      "grad_norm": 0.6167733826672094,
      "learning_rate": 0.00028506865997959173,
      "loss": 13.5084,
      "step": 1423
    },
    {
      "epoch": 0.6530612244897959,
      "grad_norm": 0.5809123257537789,
      "learning_rate": 0.000284398006824893,
      "loss": 13.6324,
      "step": 1424
    },
    {
      "epoch": 0.6535198349002522,
      "grad_norm": 0.5843626462968052,
      "learning_rate": 0.00028372782981874963,
      "loss": 13.5316,
      "step": 1425
    },
    {
      "epoch": 0.6539784453107086,
      "grad_norm": 0.5642806487993266,
      "learning_rate": 0.00028305813044122096,
      "loss": 13.4508,
      "step": 1426
    },
    {
      "epoch": 0.6544370557211648,
      "grad_norm": 0.5482908789223221,
      "learning_rate": 0.0002823889101713122,
      "loss": 13.3054,
      "step": 1427
    },
    {
      "epoch": 0.6548956661316212,
      "grad_norm": 0.5545345678187219,
      "learning_rate": 0.0002817201704869701,
      "loss": 13.717,
      "step": 1428
    },
    {
      "epoch": 0.6553542765420775,
      "grad_norm": 0.5463088999038841,
      "learning_rate": 0.00028105191286508,
      "loss": 13.2427,
      "step": 1429
    },
    {
      "epoch": 0.6558128869525338,
      "grad_norm": 0.535676406235345,
      "learning_rate": 0.00028038413878146245,
      "loss": 13.6467,
      "step": 1430
    },
    {
      "epoch": 0.6562714973629902,
      "grad_norm": 0.5127611917222189,
      "learning_rate": 0.00027971684971087073,
      "loss": 13.4691,
      "step": 1431
    },
    {
      "epoch": 0.6567301077734464,
      "grad_norm": 0.5584004899808156,
      "learning_rate": 0.00027905004712698643,
      "loss": 13.5007,
      "step": 1432
    },
    {
      "epoch": 0.6571887181839028,
      "grad_norm": 0.5538030212268628,
      "learning_rate": 0.0002783837325024167,
      "loss": 13.4988,
      "step": 1433
    },
    {
      "epoch": 0.657647328594359,
      "grad_norm": 0.5967794345106837,
      "learning_rate": 0.00027771790730869153,
      "loss": 13.4489,
      "step": 1434
    },
    {
      "epoch": 0.6581059390048154,
      "grad_norm": 0.5579855089934264,
      "learning_rate": 0.0002770525730162599,
      "loss": 13.4551,
      "step": 1435
    },
    {
      "epoch": 0.6585645494152718,
      "grad_norm": 0.5321860726729923,
      "learning_rate": 0.00027638773109448645,
      "loss": 13.7518,
      "step": 1436
    },
    {
      "epoch": 0.659023159825728,
      "grad_norm": 0.5257940110933071,
      "learning_rate": 0.00027572338301164824,
      "loss": 13.5245,
      "step": 1437
    },
    {
      "epoch": 0.6594817702361844,
      "grad_norm": 0.5446572815027175,
      "learning_rate": 0.0002750595302349324,
      "loss": 13.4774,
      "step": 1438
    },
    {
      "epoch": 0.6599403806466407,
      "grad_norm": 0.5533790710263038,
      "learning_rate": 0.00027439617423043145,
      "loss": 13.6069,
      "step": 1439
    },
    {
      "epoch": 0.660398991057097,
      "grad_norm": 0.5739828005311295,
      "learning_rate": 0.00027373331646314114,
      "loss": 13.4129,
      "step": 1440
    },
    {
      "epoch": 0.6608576014675533,
      "grad_norm": 0.5575707862240626,
      "learning_rate": 0.0002730709583969572,
      "loss": 13.3657,
      "step": 1441
    },
    {
      "epoch": 0.6613162118780096,
      "grad_norm": 0.5483401031867183,
      "learning_rate": 0.0002724091014946711,
      "loss": 13.6299,
      "step": 1442
    },
    {
      "epoch": 0.661774822288466,
      "grad_norm": 0.5627509933848819,
      "learning_rate": 0.00027174774721796824,
      "loss": 13.446,
      "step": 1443
    },
    {
      "epoch": 0.6622334326989223,
      "grad_norm": 0.5241498114872815,
      "learning_rate": 0.0002710868970274232,
      "loss": 13.4219,
      "step": 1444
    },
    {
      "epoch": 0.6626920431093786,
      "grad_norm": 0.5318203206874241,
      "learning_rate": 0.0002704265523824982,
      "loss": 13.5936,
      "step": 1445
    },
    {
      "epoch": 0.6631506535198349,
      "grad_norm": 0.5378425185154989,
      "learning_rate": 0.00026976671474153826,
      "loss": 13.5326,
      "step": 1446
    },
    {
      "epoch": 0.6636092639302912,
      "grad_norm": 0.6777466177178139,
      "learning_rate": 0.00026910738556176886,
      "loss": 13.5713,
      "step": 1447
    },
    {
      "epoch": 0.6640678743407475,
      "grad_norm": 0.5989410124698219,
      "learning_rate": 0.0002684485662992929,
      "loss": 13.5329,
      "step": 1448
    },
    {
      "epoch": 0.6645264847512039,
      "grad_norm": 0.5498296476223969,
      "learning_rate": 0.0002677902584090869,
      "loss": 13.4207,
      "step": 1449
    },
    {
      "epoch": 0.6649850951616602,
      "grad_norm": 0.5719854072398715,
      "learning_rate": 0.00026713246334499774,
      "loss": 13.3709,
      "step": 1450
    },
    {
      "epoch": 0.6654437055721165,
      "grad_norm": 0.5746367821990462,
      "learning_rate": 0.00026647518255974023,
      "loss": 13.2712,
      "step": 1451
    },
    {
      "epoch": 0.6659023159825728,
      "grad_norm": 0.6121180891142158,
      "learning_rate": 0.0002658184175048934,
      "loss": 13.6939,
      "step": 1452
    },
    {
      "epoch": 0.6663609263930291,
      "grad_norm": 0.556945177678844,
      "learning_rate": 0.00026516216963089694,
      "loss": 13.5131,
      "step": 1453
    },
    {
      "epoch": 0.6668195368034855,
      "grad_norm": 0.6283372350701857,
      "learning_rate": 0.0002645064403870488,
      "loss": 13.3223,
      "step": 1454
    },
    {
      "epoch": 0.6672781472139417,
      "grad_norm": 0.6404893512715146,
      "learning_rate": 0.0002638512312215011,
      "loss": 13.4087,
      "step": 1455
    },
    {
      "epoch": 0.6677367576243981,
      "grad_norm": 0.5493771525809117,
      "learning_rate": 0.0002631965435812575,
      "loss": 13.439,
      "step": 1456
    },
    {
      "epoch": 0.6681953680348544,
      "grad_norm": 0.5520730890179801,
      "learning_rate": 0.00026254237891217046,
      "loss": 13.604,
      "step": 1457
    },
    {
      "epoch": 0.6686539784453107,
      "grad_norm": 0.5809672201663916,
      "learning_rate": 0.0002618887386589367,
      "loss": 13.2983,
      "step": 1458
    },
    {
      "epoch": 0.6691125888557671,
      "grad_norm": 0.5861329287220942,
      "learning_rate": 0.0002612356242650949,
      "loss": 13.5432,
      "step": 1459
    },
    {
      "epoch": 0.6695711992662233,
      "grad_norm": 0.5811165752631008,
      "learning_rate": 0.0002605830371730229,
      "loss": 13.6666,
      "step": 1460
    },
    {
      "epoch": 0.6700298096766797,
      "grad_norm": 0.5778459706429473,
      "learning_rate": 0.0002599309788239339,
      "loss": 13.5494,
      "step": 1461
    },
    {
      "epoch": 0.6704884200871359,
      "grad_norm": 0.6883469584743284,
      "learning_rate": 0.00025927945065787306,
      "loss": 13.4878,
      "step": 1462
    },
    {
      "epoch": 0.6709470304975923,
      "grad_norm": 0.5739117654497868,
      "learning_rate": 0.0002586284541137145,
      "loss": 13.4868,
      "step": 1463
    },
    {
      "epoch": 0.6714056409080487,
      "grad_norm": 0.5449389223026625,
      "learning_rate": 0.00025797799062915905,
      "loss": 13.6029,
      "step": 1464
    },
    {
      "epoch": 0.6718642513185049,
      "grad_norm": 0.5966990223091659,
      "learning_rate": 0.00025732806164072966,
      "loss": 13.4493,
      "step": 1465
    },
    {
      "epoch": 0.6723228617289613,
      "grad_norm": 0.6207681895760956,
      "learning_rate": 0.00025667866858376874,
      "loss": 13.521,
      "step": 1466
    },
    {
      "epoch": 0.6727814721394175,
      "grad_norm": 0.6432000584698955,
      "learning_rate": 0.0002560298128924358,
      "loss": 13.8566,
      "step": 1467
    },
    {
      "epoch": 0.6732400825498739,
      "grad_norm": 0.5232527953194311,
      "learning_rate": 0.0002553814959997032,
      "loss": 13.5897,
      "step": 1468
    },
    {
      "epoch": 0.6736986929603302,
      "grad_norm": 0.6331023542888623,
      "learning_rate": 0.00025473371933735334,
      "loss": 13.5468,
      "step": 1469
    },
    {
      "epoch": 0.6741573033707865,
      "grad_norm": 0.566911387665252,
      "learning_rate": 0.00025408648433597534,
      "loss": 13.4908,
      "step": 1470
    },
    {
      "epoch": 0.6746159137812429,
      "grad_norm": 0.5729519419575317,
      "learning_rate": 0.00025343979242496283,
      "loss": 13.5562,
      "step": 1471
    },
    {
      "epoch": 0.6750745241916991,
      "grad_norm": 0.5274612772630514,
      "learning_rate": 0.00025279364503250925,
      "loss": 13.4986,
      "step": 1472
    },
    {
      "epoch": 0.6755331346021555,
      "grad_norm": 0.5996324972733109,
      "learning_rate": 0.0002521480435856056,
      "loss": 13.562,
      "step": 1473
    },
    {
      "epoch": 0.6759917450126118,
      "grad_norm": 0.5707535975071363,
      "learning_rate": 0.0002515029895100378,
      "loss": 13.516,
      "step": 1474
    },
    {
      "epoch": 0.6764503554230681,
      "grad_norm": 0.5509540822711096,
      "learning_rate": 0.0002508584842303822,
      "loss": 13.3494,
      "step": 1475
    },
    {
      "epoch": 0.6769089658335244,
      "grad_norm": 0.5510002393733238,
      "learning_rate": 0.0002502145291700038,
      "loss": 13.5861,
      "step": 1476
    },
    {
      "epoch": 0.6773675762439807,
      "grad_norm": 0.5546847082289434,
      "learning_rate": 0.0002495711257510517,
      "loss": 13.5487,
      "step": 1477
    },
    {
      "epoch": 0.6778261866544371,
      "grad_norm": 0.5521528399696433,
      "learning_rate": 0.0002489282753944575,
      "loss": 13.385,
      "step": 1478
    },
    {
      "epoch": 0.6782847970648934,
      "grad_norm": 0.6469196492199153,
      "learning_rate": 0.00024828597951993093,
      "loss": 13.3337,
      "step": 1479
    },
    {
      "epoch": 0.6787434074753497,
      "grad_norm": 0.5156771771625902,
      "learning_rate": 0.00024764423954595706,
      "loss": 13.5615,
      "step": 1480
    },
    {
      "epoch": 0.679202017885806,
      "grad_norm": 0.5586472148917894,
      "learning_rate": 0.0002470030568897938,
      "loss": 13.4404,
      "step": 1481
    },
    {
      "epoch": 0.6796606282962623,
      "grad_norm": 0.6424783358152197,
      "learning_rate": 0.00024636243296746773,
      "loss": 13.5754,
      "step": 1482
    },
    {
      "epoch": 0.6801192387067186,
      "grad_norm": 0.553958671729653,
      "learning_rate": 0.0002457223691937716,
      "loss": 13.4605,
      "step": 1483
    },
    {
      "epoch": 0.680577849117175,
      "grad_norm": 0.5460289603333215,
      "learning_rate": 0.0002450828669822613,
      "loss": 13.4387,
      "step": 1484
    },
    {
      "epoch": 0.6810364595276313,
      "grad_norm": 0.53640174439462,
      "learning_rate": 0.00024444392774525253,
      "loss": 13.5069,
      "step": 1485
    },
    {
      "epoch": 0.6814950699380876,
      "grad_norm": 0.5335727068415598,
      "learning_rate": 0.00024380555289381733,
      "loss": 13.4952,
      "step": 1486
    },
    {
      "epoch": 0.6819536803485439,
      "grad_norm": 0.5448212310608584,
      "learning_rate": 0.00024316774383778184,
      "loss": 13.4925,
      "step": 1487
    },
    {
      "epoch": 0.6824122907590002,
      "grad_norm": 0.5378760458093091,
      "learning_rate": 0.0002425305019857222,
      "loss": 13.6785,
      "step": 1488
    },
    {
      "epoch": 0.6828709011694566,
      "grad_norm": 0.5364432291783341,
      "learning_rate": 0.00024189382874496184,
      "loss": 13.5427,
      "step": 1489
    },
    {
      "epoch": 0.6833295115799128,
      "grad_norm": 0.5235631802097139,
      "learning_rate": 0.00024125772552156916,
      "loss": 13.6768,
      "step": 1490
    },
    {
      "epoch": 0.6837881219903692,
      "grad_norm": 0.5467432550986876,
      "learning_rate": 0.00024062219372035292,
      "loss": 13.4713,
      "step": 1491
    },
    {
      "epoch": 0.6842467324008255,
      "grad_norm": 0.5540249830012741,
      "learning_rate": 0.00023998723474486007,
      "loss": 13.7431,
      "step": 1492
    },
    {
      "epoch": 0.6847053428112818,
      "grad_norm": 0.5653545722167688,
      "learning_rate": 0.00023935284999737272,
      "loss": 13.5541,
      "step": 1493
    },
    {
      "epoch": 0.6851639532217382,
      "grad_norm": 0.5444238101610456,
      "learning_rate": 0.00023871904087890505,
      "loss": 13.6007,
      "step": 1494
    },
    {
      "epoch": 0.6856225636321944,
      "grad_norm": 0.5658117380568989,
      "learning_rate": 0.00023808580878919945,
      "loss": 13.5859,
      "step": 1495
    },
    {
      "epoch": 0.6860811740426508,
      "grad_norm": 0.5156028810095354,
      "learning_rate": 0.00023745315512672398,
      "loss": 13.4311,
      "step": 1496
    },
    {
      "epoch": 0.686539784453107,
      "grad_norm": 0.5269626909593766,
      "learning_rate": 0.0002368210812886698,
      "loss": 13.6626,
      "step": 1497
    },
    {
      "epoch": 0.6869983948635634,
      "grad_norm": 0.6496326435412613,
      "learning_rate": 0.0002361895886709471,
      "loss": 13.435,
      "step": 1498
    },
    {
      "epoch": 0.6874570052740198,
      "grad_norm": 0.5842974929325577,
      "learning_rate": 0.0002355586786681823,
      "loss": 13.3706,
      "step": 1499
    },
    {
      "epoch": 0.687915615684476,
      "grad_norm": 0.552946740936479,
      "learning_rate": 0.00023492835267371575,
      "loss": 13.4821,
      "step": 1500
    },
    {
      "epoch": 0.6883742260949324,
      "grad_norm": 0.5231247105688088,
      "learning_rate": 0.0002342986120795978,
      "loss": 13.5623,
      "step": 1501
    },
    {
      "epoch": 0.6888328365053886,
      "grad_norm": 0.5360977953104581,
      "learning_rate": 0.0002336694582765857,
      "loss": 13.6501,
      "step": 1502
    },
    {
      "epoch": 0.689291446915845,
      "grad_norm": 0.5136968519117898,
      "learning_rate": 0.00023304089265414085,
      "loss": 13.4759,
      "step": 1503
    },
    {
      "epoch": 0.6897500573263013,
      "grad_norm": 0.5675303832323287,
      "learning_rate": 0.00023241291660042613,
      "loss": 13.4701,
      "step": 1504
    },
    {
      "epoch": 0.6902086677367576,
      "grad_norm": 0.5758474376066994,
      "learning_rate": 0.00023178553150230186,
      "loss": 13.5733,
      "step": 1505
    },
    {
      "epoch": 0.690667278147214,
      "grad_norm": 0.5352864707404559,
      "learning_rate": 0.00023115873874532324,
      "loss": 13.6872,
      "step": 1506
    },
    {
      "epoch": 0.6911258885576702,
      "grad_norm": 0.5623011028374948,
      "learning_rate": 0.00023053253971373796,
      "loss": 13.6485,
      "step": 1507
    },
    {
      "epoch": 0.6915844989681266,
      "grad_norm": 0.5521130300334472,
      "learning_rate": 0.00022990693579048166,
      "loss": 13.7561,
      "step": 1508
    },
    {
      "epoch": 0.6920431093785829,
      "grad_norm": 0.5464480768750508,
      "learning_rate": 0.00022928192835717644,
      "loss": 13.5441,
      "step": 1509
    },
    {
      "epoch": 0.6925017197890392,
      "grad_norm": 0.5541362716199804,
      "learning_rate": 0.00022865751879412634,
      "loss": 13.4817,
      "step": 1510
    },
    {
      "epoch": 0.6929603301994955,
      "grad_norm": 0.5278410652739096,
      "learning_rate": 0.00022803370848031585,
      "loss": 13.4437,
      "step": 1511
    },
    {
      "epoch": 0.6934189406099518,
      "grad_norm": 0.536162971384859,
      "learning_rate": 0.00022741049879340542,
      "loss": 13.5244,
      "step": 1512
    },
    {
      "epoch": 0.6938775510204082,
      "grad_norm": 0.5321751270025563,
      "learning_rate": 0.00022678789110972897,
      "loss": 13.5689,
      "step": 1513
    },
    {
      "epoch": 0.6943361614308645,
      "grad_norm": 0.5710537085360685,
      "learning_rate": 0.00022616588680429155,
      "loss": 13.5206,
      "step": 1514
    },
    {
      "epoch": 0.6947947718413208,
      "grad_norm": 0.5330799077438549,
      "learning_rate": 0.00022554448725076526,
      "loss": 13.4282,
      "step": 1515
    },
    {
      "epoch": 0.6952533822517771,
      "grad_norm": 0.5372596411314929,
      "learning_rate": 0.0002249236938214863,
      "loss": 13.5007,
      "step": 1516
    },
    {
      "epoch": 0.6957119926622334,
      "grad_norm": 0.5962682349575064,
      "learning_rate": 0.00022430350788745296,
      "loss": 13.5822,
      "step": 1517
    },
    {
      "epoch": 0.6961706030726897,
      "grad_norm": 0.5430586347262262,
      "learning_rate": 0.00022368393081832166,
      "loss": 13.3597,
      "step": 1518
    },
    {
      "epoch": 0.6966292134831461,
      "grad_norm": 0.5401217518475862,
      "learning_rate": 0.00022306496398240383,
      "loss": 13.4911,
      "step": 1519
    },
    {
      "epoch": 0.6970878238936024,
      "grad_norm": 0.5775406792691912,
      "learning_rate": 0.00022244660874666373,
      "loss": 13.3182,
      "step": 1520
    },
    {
      "epoch": 0.6975464343040587,
      "grad_norm": 0.5740289208090895,
      "learning_rate": 0.00022182886647671452,
      "loss": 13.6218,
      "step": 1521
    },
    {
      "epoch": 0.698005044714515,
      "grad_norm": 0.5685954607257492,
      "learning_rate": 0.0002212117385368157,
      "loss": 13.5227,
      "step": 1522
    },
    {
      "epoch": 0.6984636551249713,
      "grad_norm": 0.5596411601498252,
      "learning_rate": 0.00022059522628987038,
      "loss": 13.4942,
      "step": 1523
    },
    {
      "epoch": 0.6989222655354277,
      "grad_norm": 0.5360967529743529,
      "learning_rate": 0.00021997933109742162,
      "loss": 13.4307,
      "step": 1524
    },
    {
      "epoch": 0.6993808759458839,
      "grad_norm": 0.5287202918102737,
      "learning_rate": 0.00021936405431964969,
      "loss": 13.5545,
      "step": 1525
    },
    {
      "epoch": 0.6998394863563403,
      "grad_norm": 0.5814011618696034,
      "learning_rate": 0.00021874939731536926,
      "loss": 13.5243,
      "step": 1526
    },
    {
      "epoch": 0.7002980967667966,
      "grad_norm": 0.5545478858330721,
      "learning_rate": 0.00021813536144202656,
      "loss": 13.454,
      "step": 1527
    },
    {
      "epoch": 0.7007567071772529,
      "grad_norm": 0.6498602768247264,
      "learning_rate": 0.00021752194805569553,
      "loss": 13.339,
      "step": 1528
    },
    {
      "epoch": 0.7012153175877093,
      "grad_norm": 0.5584308161142751,
      "learning_rate": 0.0002169091585110754,
      "loss": 13.6976,
      "step": 1529
    },
    {
      "epoch": 0.7016739279981655,
      "grad_norm": 0.5726561875574226,
      "learning_rate": 0.00021629699416148828,
      "loss": 13.4387,
      "step": 1530
    },
    {
      "epoch": 0.7021325384086219,
      "grad_norm": 0.6777631378391582,
      "learning_rate": 0.000215685456358875,
      "loss": 13.7823,
      "step": 1531
    },
    {
      "epoch": 0.7025911488190782,
      "grad_norm": 0.5866631530501407,
      "learning_rate": 0.00021507454645379258,
      "loss": 13.5841,
      "step": 1532
    },
    {
      "epoch": 0.7030497592295345,
      "grad_norm": 0.5605390821058098,
      "learning_rate": 0.00021446426579541184,
      "loss": 13.5318,
      "step": 1533
    },
    {
      "epoch": 0.7035083696399909,
      "grad_norm": 0.5521483914608392,
      "learning_rate": 0.00021385461573151387,
      "loss": 13.678,
      "step": 1534
    },
    {
      "epoch": 0.7039669800504471,
      "grad_norm": 0.6150396436232447,
      "learning_rate": 0.00021324559760848677,
      "loss": 13.3026,
      "step": 1535
    },
    {
      "epoch": 0.7044255904609035,
      "grad_norm": 0.5458195082068804,
      "learning_rate": 0.00021263721277132303,
      "loss": 13.3969,
      "step": 1536
    },
    {
      "epoch": 0.7048842008713598,
      "grad_norm": 0.5384636581321947,
      "learning_rate": 0.0002120294625636171,
      "loss": 13.6527,
      "step": 1537
    },
    {
      "epoch": 0.7053428112818161,
      "grad_norm": 0.5686156579519182,
      "learning_rate": 0.0002114223483275613,
      "loss": 13.6602,
      "step": 1538
    },
    {
      "epoch": 0.7058014216922724,
      "grad_norm": 0.5377020123927249,
      "learning_rate": 0.0002108158714039435,
      "loss": 13.5538,
      "step": 1539
    },
    {
      "epoch": 0.7062600321027287,
      "grad_norm": 0.5195603611236671,
      "learning_rate": 0.00021021003313214455,
      "loss": 13.3063,
      "step": 1540
    },
    {
      "epoch": 0.7067186425131851,
      "grad_norm": 0.528832847385973,
      "learning_rate": 0.00020960483485013432,
      "loss": 13.4593,
      "step": 1541
    },
    {
      "epoch": 0.7071772529236414,
      "grad_norm": 0.5470243541669833,
      "learning_rate": 0.0002090002778944694,
      "loss": 13.4781,
      "step": 1542
    },
    {
      "epoch": 0.7076358633340977,
      "grad_norm": 0.5663583632483636,
      "learning_rate": 0.00020839636360029025,
      "loss": 13.6094,
      "step": 1543
    },
    {
      "epoch": 0.708094473744554,
      "grad_norm": 0.5413261733313783,
      "learning_rate": 0.00020779309330131818,
      "loss": 13.549,
      "step": 1544
    },
    {
      "epoch": 0.7085530841550103,
      "grad_norm": 0.5733595738860218,
      "learning_rate": 0.00020719046832985184,
      "loss": 13.2595,
      "step": 1545
    },
    {
      "epoch": 0.7090116945654666,
      "grad_norm": 0.5430547070302031,
      "learning_rate": 0.0002065884900167646,
      "loss": 13.6546,
      "step": 1546
    },
    {
      "epoch": 0.709470304975923,
      "grad_norm": 0.5391197634167381,
      "learning_rate": 0.0002059871596915024,
      "loss": 13.5458,
      "step": 1547
    },
    {
      "epoch": 0.7099289153863793,
      "grad_norm": 0.5390204727679598,
      "learning_rate": 0.0002053864786820795,
      "loss": 13.3235,
      "step": 1548
    },
    {
      "epoch": 0.7103875257968356,
      "grad_norm": 0.539956075679839,
      "learning_rate": 0.00020478644831507627,
      "loss": 13.6461,
      "step": 1549
    },
    {
      "epoch": 0.7108461362072919,
      "grad_norm": 0.5502961530144806,
      "learning_rate": 0.00020418706991563634,
      "loss": 13.5087,
      "step": 1550
    },
    {
      "epoch": 0.7113047466177482,
      "grad_norm": 0.5385655497979881,
      "learning_rate": 0.00020358834480746363,
      "loss": 13.5025,
      "step": 1551
    },
    {
      "epoch": 0.7117633570282046,
      "grad_norm": 0.5293459741932858,
      "learning_rate": 0.0002029902743128188,
      "loss": 13.5253,
      "step": 1552
    },
    {
      "epoch": 0.7122219674386608,
      "grad_norm": 0.5311751205514483,
      "learning_rate": 0.0002023928597525174,
      "loss": 13.1847,
      "step": 1553
    },
    {
      "epoch": 0.7126805778491172,
      "grad_norm": 0.5288047532816569,
      "learning_rate": 0.00020179610244592595,
      "loss": 13.5312,
      "step": 1554
    },
    {
      "epoch": 0.7131391882595735,
      "grad_norm": 0.5382962516831329,
      "learning_rate": 0.00020120000371095937,
      "loss": 13.4633,
      "step": 1555
    },
    {
      "epoch": 0.7135977986700298,
      "grad_norm": 0.5220548227868683,
      "learning_rate": 0.0002006045648640787,
      "loss": 13.4812,
      "step": 1556
    },
    {
      "epoch": 0.7140564090804862,
      "grad_norm": 0.5669383065788101,
      "learning_rate": 0.00020000978722028713,
      "loss": 13.8653,
      "step": 1557
    },
    {
      "epoch": 0.7145150194909424,
      "grad_norm": 0.5298159810077363,
      "learning_rate": 0.00019941567209312767,
      "loss": 13.6989,
      "step": 1558
    },
    {
      "epoch": 0.7149736299013988,
      "grad_norm": 0.536673725487106,
      "learning_rate": 0.00019882222079468036,
      "loss": 13.4997,
      "step": 1559
    },
    {
      "epoch": 0.715432240311855,
      "grad_norm": 0.5501219562846471,
      "learning_rate": 0.0001982294346355595,
      "loss": 13.344,
      "step": 1560
    },
    {
      "epoch": 0.7158908507223114,
      "grad_norm": 0.5804999488318231,
      "learning_rate": 0.00019763731492490976,
      "loss": 13.3961,
      "step": 1561
    },
    {
      "epoch": 0.7163494611327678,
      "grad_norm": 0.5436963212610942,
      "learning_rate": 0.00019704586297040422,
      "loss": 13.5221,
      "step": 1562
    },
    {
      "epoch": 0.716808071543224,
      "grad_norm": 0.568127748804818,
      "learning_rate": 0.0001964550800782417,
      "loss": 13.437,
      "step": 1563
    },
    {
      "epoch": 0.7172666819536804,
      "grad_norm": 0.5483312313154194,
      "learning_rate": 0.00019586496755314288,
      "loss": 13.6135,
      "step": 1564
    },
    {
      "epoch": 0.7177252923641366,
      "grad_norm": 0.5376198304529269,
      "learning_rate": 0.00019527552669834798,
      "loss": 13.4039,
      "step": 1565
    },
    {
      "epoch": 0.718183902774593,
      "grad_norm": 0.5305870829561916,
      "learning_rate": 0.0001946867588156142,
      "loss": 13.7464,
      "step": 1566
    },
    {
      "epoch": 0.7186425131850493,
      "grad_norm": 0.9475452447906493,
      "learning_rate": 0.00019409866520521258,
      "loss": 13.5232,
      "step": 1567
    },
    {
      "epoch": 0.7191011235955056,
      "grad_norm": 0.5667659776404311,
      "learning_rate": 0.00019351124716592455,
      "loss": 13.5107,
      "step": 1568
    },
    {
      "epoch": 0.719559734005962,
      "grad_norm": 0.5760691559280193,
      "learning_rate": 0.0001929245059950397,
      "loss": 13.4972,
      "step": 1569
    },
    {
      "epoch": 0.7200183444164182,
      "grad_norm": 0.5620827586468006,
      "learning_rate": 0.0001923384429883533,
      "loss": 13.4712,
      "step": 1570
    },
    {
      "epoch": 0.7204769548268746,
      "grad_norm": 0.5337853582405814,
      "learning_rate": 0.00019175305944016237,
      "loss": 13.4369,
      "step": 1571
    },
    {
      "epoch": 0.7209355652373309,
      "grad_norm": 0.6422084832911024,
      "learning_rate": 0.00019116835664326326,
      "loss": 13.7204,
      "step": 1572
    },
    {
      "epoch": 0.7213941756477872,
      "grad_norm": 0.5743488842259229,
      "learning_rate": 0.0001905843358889497,
      "loss": 13.545,
      "step": 1573
    },
    {
      "epoch": 0.7218527860582435,
      "grad_norm": 0.5950161462991493,
      "learning_rate": 0.00019000099846700836,
      "loss": 13.4386,
      "step": 1574
    },
    {
      "epoch": 0.7223113964686998,
      "grad_norm": 0.5170173564944466,
      "learning_rate": 0.00018941834566571692,
      "loss": 13.6576,
      "step": 1575
    },
    {
      "epoch": 0.7227700068791562,
      "grad_norm": 0.519997058526102,
      "learning_rate": 0.00018883637877184145,
      "loss": 13.5363,
      "step": 1576
    },
    {
      "epoch": 0.7232286172896125,
      "grad_norm": 0.5578983540841602,
      "learning_rate": 0.00018825509907063325,
      "loss": 13.4776,
      "step": 1577
    },
    {
      "epoch": 0.7236872277000688,
      "grad_norm": 0.5235599649898778,
      "learning_rate": 0.00018767450784582557,
      "loss": 13.6313,
      "step": 1578
    },
    {
      "epoch": 0.7241458381105251,
      "grad_norm": 0.5435184703656194,
      "learning_rate": 0.00018709460637963122,
      "loss": 13.551,
      "step": 1579
    },
    {
      "epoch": 0.7246044485209814,
      "grad_norm": 0.5659972817003611,
      "learning_rate": 0.00018651539595274013,
      "loss": 13.4018,
      "step": 1580
    },
    {
      "epoch": 0.7250630589314377,
      "grad_norm": 0.5379289425739562,
      "learning_rate": 0.00018593687784431578,
      "loss": 13.6342,
      "step": 1581
    },
    {
      "epoch": 0.7255216693418941,
      "grad_norm": 0.5401930216627505,
      "learning_rate": 0.00018535905333199248,
      "loss": 13.5511,
      "step": 1582
    },
    {
      "epoch": 0.7259802797523504,
      "grad_norm": 0.5927545927059403,
      "learning_rate": 0.0001847819236918733,
      "loss": 13.5183,
      "step": 1583
    },
    {
      "epoch": 0.7264388901628067,
      "grad_norm": 0.5753931923965674,
      "learning_rate": 0.00018420549019852655,
      "loss": 13.5378,
      "step": 1584
    },
    {
      "epoch": 0.726897500573263,
      "grad_norm": 0.5221449435622226,
      "learning_rate": 0.00018362975412498266,
      "loss": 13.6549,
      "step": 1585
    },
    {
      "epoch": 0.7273561109837193,
      "grad_norm": 0.5627611248738447,
      "learning_rate": 0.00018305471674273261,
      "loss": 13.5914,
      "step": 1586
    },
    {
      "epoch": 0.7278147213941757,
      "grad_norm": 0.590166191087603,
      "learning_rate": 0.0001824803793217237,
      "loss": 13.6252,
      "step": 1587
    },
    {
      "epoch": 0.7282733318046319,
      "grad_norm": 0.5638101431635401,
      "learning_rate": 0.00018190674313035737,
      "loss": 13.4437,
      "step": 1588
    },
    {
      "epoch": 0.7287319422150883,
      "grad_norm": 0.5953550535437816,
      "learning_rate": 0.00018133380943548716,
      "loss": 13.2286,
      "step": 1589
    },
    {
      "epoch": 0.7291905526255446,
      "grad_norm": 0.5708720245965843,
      "learning_rate": 0.00018076157950241452,
      "loss": 13.5544,
      "step": 1590
    },
    {
      "epoch": 0.7296491630360009,
      "grad_norm": 0.5851418974711632,
      "learning_rate": 0.00018019005459488652,
      "loss": 13.5933,
      "step": 1591
    },
    {
      "epoch": 0.7301077734464573,
      "grad_norm": 0.5618738841021477,
      "learning_rate": 0.00017961923597509388,
      "loss": 13.4304,
      "step": 1592
    },
    {
      "epoch": 0.7305663838569135,
      "grad_norm": 0.579298386809755,
      "learning_rate": 0.00017904912490366722,
      "loss": 13.3309,
      "step": 1593
    },
    {
      "epoch": 0.7310249942673699,
      "grad_norm": 0.5308806982189452,
      "learning_rate": 0.00017847972263967433,
      "loss": 13.5299,
      "step": 1594
    },
    {
      "epoch": 0.7314836046778261,
      "grad_norm": 0.560952778346823,
      "learning_rate": 0.0001779110304406177,
      "loss": 13.4932,
      "step": 1595
    },
    {
      "epoch": 0.7319422150882825,
      "grad_norm": 0.538674031447644,
      "learning_rate": 0.000177343049562432,
      "loss": 13.5887,
      "step": 1596
    },
    {
      "epoch": 0.7324008254987389,
      "grad_norm": 0.5280154363472388,
      "learning_rate": 0.0001767757812594807,
      "loss": 13.4316,
      "step": 1597
    },
    {
      "epoch": 0.7328594359091951,
      "grad_norm": 0.5486668584400226,
      "learning_rate": 0.0001762092267845534,
      "loss": 13.5858,
      "step": 1598
    },
    {
      "epoch": 0.7333180463196515,
      "grad_norm": 0.541421282856795,
      "learning_rate": 0.00017564338738886365,
      "loss": 13.5111,
      "step": 1599
    },
    {
      "epoch": 0.7337766567301077,
      "grad_norm": 0.5524104478363865,
      "learning_rate": 0.0001750782643220457,
      "loss": 13.5334,
      "step": 1600
    },
    {
      "epoch": 0.7342352671405641,
      "grad_norm": 0.5320000656514311,
      "learning_rate": 0.00017451385883215166,
      "loss": 13.4609,
      "step": 1601
    },
    {
      "epoch": 0.7346938775510204,
      "grad_norm": 0.5315447596152402,
      "learning_rate": 0.00017395017216564863,
      "loss": 13.4075,
      "step": 1602
    },
    {
      "epoch": 0.7351524879614767,
      "grad_norm": 0.530515835351447,
      "learning_rate": 0.00017338720556741687,
      "loss": 13.5529,
      "step": 1603
    },
    {
      "epoch": 0.7356110983719331,
      "grad_norm": 0.5409395357189516,
      "learning_rate": 0.00017282496028074606,
      "loss": 13.4571,
      "step": 1604
    },
    {
      "epoch": 0.7360697087823893,
      "grad_norm": 0.5262694599854476,
      "learning_rate": 0.00017226343754733254,
      "loss": 13.5585,
      "step": 1605
    },
    {
      "epoch": 0.7365283191928457,
      "grad_norm": 0.5103969045060346,
      "learning_rate": 0.00017170263860727769,
      "loss": 13.6517,
      "step": 1606
    },
    {
      "epoch": 0.736986929603302,
      "grad_norm": 0.5477179827244968,
      "learning_rate": 0.0001711425646990838,
      "loss": 13.6067,
      "step": 1607
    },
    {
      "epoch": 0.7374455400137583,
      "grad_norm": 0.5079361609131712,
      "learning_rate": 0.00017058321705965202,
      "loss": 13.6419,
      "step": 1608
    },
    {
      "epoch": 0.7379041504242146,
      "grad_norm": 0.5407806748756651,
      "learning_rate": 0.0001700245969242798,
      "loss": 13.5489,
      "step": 1609
    },
    {
      "epoch": 0.7383627608346709,
      "grad_norm": 0.5493257320460139,
      "learning_rate": 0.00016946670552665804,
      "loss": 13.6202,
      "step": 1610
    },
    {
      "epoch": 0.7388213712451273,
      "grad_norm": 0.5503265299543925,
      "learning_rate": 0.00016890954409886795,
      "loss": 13.6034,
      "step": 1611
    },
    {
      "epoch": 0.7392799816555836,
      "grad_norm": 0.5714997514388778,
      "learning_rate": 0.00016835311387137836,
      "loss": 13.5859,
      "step": 1612
    },
    {
      "epoch": 0.7397385920660399,
      "grad_norm": 0.5537941269304301,
      "learning_rate": 0.0001677974160730441,
      "loss": 13.6883,
      "step": 1613
    },
    {
      "epoch": 0.7401972024764962,
      "grad_norm": 0.5560908536853674,
      "learning_rate": 0.00016724245193110176,
      "loss": 13.6664,
      "step": 1614
    },
    {
      "epoch": 0.7406558128869525,
      "grad_norm": 0.5376880833454714,
      "learning_rate": 0.00016668822267116784,
      "loss": 13.278,
      "step": 1615
    },
    {
      "epoch": 0.7411144232974088,
      "grad_norm": 0.5815322930647687,
      "learning_rate": 0.00016613472951723597,
      "loss": 13.3605,
      "step": 1616
    },
    {
      "epoch": 0.7415730337078652,
      "grad_norm": 0.57099597088307,
      "learning_rate": 0.00016558197369167434,
      "loss": 13.6444,
      "step": 1617
    },
    {
      "epoch": 0.7420316441183215,
      "grad_norm": 0.5405207270016538,
      "learning_rate": 0.00016502995641522216,
      "loss": 13.4786,
      "step": 1618
    },
    {
      "epoch": 0.7424902545287778,
      "grad_norm": 0.5533689488971311,
      "learning_rate": 0.00016447867890698843,
      "loss": 13.4364,
      "step": 1619
    },
    {
      "epoch": 0.7429488649392341,
      "grad_norm": 0.5470512900012517,
      "learning_rate": 0.00016392814238444753,
      "loss": 13.5738,
      "step": 1620
    },
    {
      "epoch": 0.7434074753496904,
      "grad_norm": 0.5582366357025049,
      "learning_rate": 0.00016337834806343782,
      "loss": 13.4438,
      "step": 1621
    },
    {
      "epoch": 0.7438660857601468,
      "grad_norm": 0.5989201215634903,
      "learning_rate": 0.0001628292971581588,
      "loss": 13.5851,
      "step": 1622
    },
    {
      "epoch": 0.744324696170603,
      "grad_norm": 0.5471260924878542,
      "learning_rate": 0.00016228099088116772,
      "loss": 13.6736,
      "step": 1623
    },
    {
      "epoch": 0.7447833065810594,
      "grad_norm": 0.5386570934158313,
      "learning_rate": 0.00016173343044337734,
      "loss": 13.4792,
      "step": 1624
    },
    {
      "epoch": 0.7452419169915158,
      "grad_norm": 0.5580924160666677,
      "learning_rate": 0.00016118661705405356,
      "loss": 13.3619,
      "step": 1625
    },
    {
      "epoch": 0.745700527401972,
      "grad_norm": 0.5191235580663561,
      "learning_rate": 0.00016064055192081255,
      "loss": 13.2992,
      "step": 1626
    },
    {
      "epoch": 0.7461591378124284,
      "grad_norm": 0.5338562344834746,
      "learning_rate": 0.00016009523624961757,
      "loss": 13.3623,
      "step": 1627
    },
    {
      "epoch": 0.7466177482228846,
      "grad_norm": 0.5099509366419774,
      "learning_rate": 0.00015955067124477678,
      "loss": 13.4038,
      "step": 1628
    },
    {
      "epoch": 0.747076358633341,
      "grad_norm": 0.5318914881065777,
      "learning_rate": 0.000159006858108941,
      "loss": 13.5242,
      "step": 1629
    },
    {
      "epoch": 0.7475349690437973,
      "grad_norm": 0.5468060716122567,
      "learning_rate": 0.00015846379804310002,
      "loss": 13.5013,
      "step": 1630
    },
    {
      "epoch": 0.7479935794542536,
      "grad_norm": 0.5326806939470277,
      "learning_rate": 0.00015792149224658054,
      "loss": 13.5964,
      "step": 1631
    },
    {
      "epoch": 0.74845218986471,
      "grad_norm": 0.5550243681598793,
      "learning_rate": 0.00015737994191704385,
      "loss": 13.4026,
      "step": 1632
    },
    {
      "epoch": 0.7489108002751662,
      "grad_norm": 0.5579333768015667,
      "learning_rate": 0.0001568391482504829,
      "loss": 13.7428,
      "step": 1633
    },
    {
      "epoch": 0.7493694106856226,
      "grad_norm": 0.5770117263760374,
      "learning_rate": 0.00015629911244121903,
      "loss": 13.5708,
      "step": 1634
    },
    {
      "epoch": 0.7498280210960789,
      "grad_norm": 0.5454257411745042,
      "learning_rate": 0.0001557598356819,
      "loss": 13.5244,
      "step": 1635
    },
    {
      "epoch": 0.7502866315065352,
      "grad_norm": 0.5743686788858289,
      "learning_rate": 0.00015522131916349786,
      "loss": 13.6119,
      "step": 1636
    },
    {
      "epoch": 0.7507452419169915,
      "grad_norm": 0.566588176498822,
      "learning_rate": 0.00015468356407530493,
      "loss": 13.6647,
      "step": 1637
    },
    {
      "epoch": 0.7512038523274478,
      "grad_norm": 0.5372786495756049,
      "learning_rate": 0.00015414657160493217,
      "loss": 13.4767,
      "step": 1638
    },
    {
      "epoch": 0.7516624627379042,
      "grad_norm": 0.6152072310136396,
      "learning_rate": 0.00015361034293830673,
      "loss": 13.53,
      "step": 1639
    },
    {
      "epoch": 0.7521210731483605,
      "grad_norm": 0.5505043887038191,
      "learning_rate": 0.00015307487925966844,
      "loss": 13.6288,
      "step": 1640
    },
    {
      "epoch": 0.7525796835588168,
      "grad_norm": 0.559117354605604,
      "learning_rate": 0.00015254018175156776,
      "loss": 13.679,
      "step": 1641
    },
    {
      "epoch": 0.7530382939692731,
      "grad_norm": 0.5228742442004604,
      "learning_rate": 0.0001520062515948632,
      "loss": 13.3434,
      "step": 1642
    },
    {
      "epoch": 0.7534969043797294,
      "grad_norm": 0.5562361150921333,
      "learning_rate": 0.0001514730899687189,
      "loss": 13.4668,
      "step": 1643
    },
    {
      "epoch": 0.7539555147901857,
      "grad_norm": 0.5481232345225306,
      "learning_rate": 0.00015094069805060122,
      "loss": 13.4983,
      "step": 1644
    },
    {
      "epoch": 0.7544141252006421,
      "grad_norm": 0.5515527147807655,
      "learning_rate": 0.00015040907701627666,
      "loss": 13.5595,
      "step": 1645
    },
    {
      "epoch": 0.7548727356110984,
      "grad_norm": 0.5397281939767001,
      "learning_rate": 0.00014987822803980976,
      "loss": 13.3191,
      "step": 1646
    },
    {
      "epoch": 0.7553313460215547,
      "grad_norm": 0.576043180462775,
      "learning_rate": 0.00014934815229355965,
      "loss": 13.7945,
      "step": 1647
    },
    {
      "epoch": 0.755789956432011,
      "grad_norm": 0.5454505435449979,
      "learning_rate": 0.00014881885094817748,
      "loss": 13.8021,
      "step": 1648
    },
    {
      "epoch": 0.7562485668424673,
      "grad_norm": 0.5318229785612408,
      "learning_rate": 0.00014829032517260488,
      "loss": 13.469,
      "step": 1649
    },
    {
      "epoch": 0.7567071772529237,
      "grad_norm": 0.5257503878018898,
      "learning_rate": 0.0001477625761340704,
      "loss": 13.4278,
      "step": 1650
    },
    {
      "epoch": 0.7571657876633799,
      "grad_norm": 0.5423334226710979,
      "learning_rate": 0.0001472356049980868,
      "loss": 13.5171,
      "step": 1651
    },
    {
      "epoch": 0.7576243980738363,
      "grad_norm": 0.5222060036134608,
      "learning_rate": 0.00014670941292844954,
      "loss": 13.3866,
      "step": 1652
    },
    {
      "epoch": 0.7580830084842926,
      "grad_norm": 0.5285156996181376,
      "learning_rate": 0.00014618400108723295,
      "loss": 13.3745,
      "step": 1653
    },
    {
      "epoch": 0.7585416188947489,
      "grad_norm": 0.564986319681257,
      "learning_rate": 0.00014565937063478862,
      "loss": 13.4886,
      "step": 1654
    },
    {
      "epoch": 0.7590002293052053,
      "grad_norm": 0.5499157704431662,
      "learning_rate": 0.00014513552272974207,
      "loss": 13.2259,
      "step": 1655
    },
    {
      "epoch": 0.7594588397156615,
      "grad_norm": 0.542502972121505,
      "learning_rate": 0.0001446124585289913,
      "loss": 13.5399,
      "step": 1656
    },
    {
      "epoch": 0.7599174501261179,
      "grad_norm": 0.534815272633003,
      "learning_rate": 0.00014409017918770266,
      "loss": 13.4232,
      "step": 1657
    },
    {
      "epoch": 0.7603760605365741,
      "grad_norm": 0.5200171768332181,
      "learning_rate": 0.00014356868585930994,
      "loss": 13.0978,
      "step": 1658
    },
    {
      "epoch": 0.7608346709470305,
      "grad_norm": 0.5291539037376116,
      "learning_rate": 0.00014304797969551077,
      "loss": 13.3406,
      "step": 1659
    },
    {
      "epoch": 0.7612932813574869,
      "grad_norm": 0.6070583170087427,
      "learning_rate": 0.00014252806184626417,
      "loss": 13.5009,
      "step": 1660
    },
    {
      "epoch": 0.7617518917679431,
      "grad_norm": 0.5414738550008389,
      "learning_rate": 0.00014200893345978817,
      "loss": 13.3289,
      "step": 1661
    },
    {
      "epoch": 0.7622105021783995,
      "grad_norm": 0.5396419931359001,
      "learning_rate": 0.00014149059568255778,
      "loss": 13.5423,
      "step": 1662
    },
    {
      "epoch": 0.7626691125888557,
      "grad_norm": 0.5600187481605403,
      "learning_rate": 0.00014097304965930157,
      "loss": 13.4445,
      "step": 1663
    },
    {
      "epoch": 0.7631277229993121,
      "grad_norm": 0.5289499352838183,
      "learning_rate": 0.00014045629653299953,
      "loss": 13.6064,
      "step": 1664
    },
    {
      "epoch": 0.7635863334097684,
      "grad_norm": 0.5343507864500255,
      "learning_rate": 0.00013994033744488076,
      "loss": 13.4409,
      "step": 1665
    },
    {
      "epoch": 0.7640449438202247,
      "grad_norm": 0.572893389744657,
      "learning_rate": 0.00013942517353442092,
      "loss": 13.4334,
      "step": 1666
    },
    {
      "epoch": 0.7645035542306811,
      "grad_norm": 0.5722695761843646,
      "learning_rate": 0.0001389108059393391,
      "loss": 13.7322,
      "step": 1667
    },
    {
      "epoch": 0.7649621646411373,
      "grad_norm": 0.5414571630118011,
      "learning_rate": 0.00013839723579559581,
      "loss": 13.4604,
      "step": 1668
    },
    {
      "epoch": 0.7654207750515937,
      "grad_norm": 0.5447650103833626,
      "learning_rate": 0.00013788446423739103,
      "loss": 13.5364,
      "step": 1669
    },
    {
      "epoch": 0.76587938546205,
      "grad_norm": 0.5464773537578089,
      "learning_rate": 0.00013737249239716042,
      "loss": 13.2457,
      "step": 1670
    },
    {
      "epoch": 0.7663379958725063,
      "grad_norm": 0.5537409507263528,
      "learning_rate": 0.00013686132140557355,
      "loss": 13.6165,
      "step": 1671
    },
    {
      "epoch": 0.7667966062829626,
      "grad_norm": 0.5852219494029648,
      "learning_rate": 0.00013635095239153188,
      "loss": 13.557,
      "step": 1672
    },
    {
      "epoch": 0.7672552166934189,
      "grad_norm": 0.5589072578978043,
      "learning_rate": 0.00013584138648216527,
      "loss": 13.3283,
      "step": 1673
    },
    {
      "epoch": 0.7677138271038753,
      "grad_norm": 0.548465110655723,
      "learning_rate": 0.0001353326248028298,
      "loss": 13.7252,
      "step": 1674
    },
    {
      "epoch": 0.7681724375143316,
      "grad_norm": 0.5255006020070495,
      "learning_rate": 0.00013482466847710594,
      "loss": 13.5117,
      "step": 1675
    },
    {
      "epoch": 0.7686310479247879,
      "grad_norm": 0.5401790973671747,
      "learning_rate": 0.00013431751862679554,
      "loss": 13.5096,
      "step": 1676
    },
    {
      "epoch": 0.7690896583352442,
      "grad_norm": 0.533335454688884,
      "learning_rate": 0.00013381117637191887,
      "loss": 13.5561,
      "step": 1677
    },
    {
      "epoch": 0.7695482687457005,
      "grad_norm": 0.5565166885451792,
      "learning_rate": 0.00013330564283071293,
      "loss": 13.6072,
      "step": 1678
    },
    {
      "epoch": 0.7700068791561568,
      "grad_norm": 0.5451932194350434,
      "learning_rate": 0.000132800919119629,
      "loss": 13.636,
      "step": 1679
    },
    {
      "epoch": 0.7704654895666132,
      "grad_norm": 0.5260046372072681,
      "learning_rate": 0.00013229700635332948,
      "loss": 13.5309,
      "step": 1680
    },
    {
      "epoch": 0.7709240999770695,
      "grad_norm": 0.5673599798709942,
      "learning_rate": 0.00013179390564468585,
      "loss": 13.5541,
      "step": 1681
    },
    {
      "epoch": 0.7713827103875258,
      "grad_norm": 0.5353626339232774,
      "learning_rate": 0.00013129161810477641,
      "loss": 13.6285,
      "step": 1682
    },
    {
      "epoch": 0.7718413207979821,
      "grad_norm": 0.5254321926098489,
      "learning_rate": 0.0001307901448428837,
      "loss": 13.5357,
      "step": 1683
    },
    {
      "epoch": 0.7722999312084384,
      "grad_norm": 0.5717632815637701,
      "learning_rate": 0.0001302894869664916,
      "loss": 13.4181,
      "step": 1684
    },
    {
      "epoch": 0.7727585416188948,
      "grad_norm": 0.5247121910055932,
      "learning_rate": 0.00012978964558128336,
      "loss": 13.4881,
      "step": 1685
    },
    {
      "epoch": 0.773217152029351,
      "grad_norm": 0.5211923889964293,
      "learning_rate": 0.00012929062179113925,
      "loss": 13.3603,
      "step": 1686
    },
    {
      "epoch": 0.7736757624398074,
      "grad_norm": 0.5921622942555707,
      "learning_rate": 0.00012879241669813368,
      "loss": 13.5535,
      "step": 1687
    },
    {
      "epoch": 0.7741343728502637,
      "grad_norm": 0.548249693098221,
      "learning_rate": 0.00012829503140253295,
      "loss": 13.4876,
      "step": 1688
    },
    {
      "epoch": 0.77459298326072,
      "grad_norm": 0.5630323030129198,
      "learning_rate": 0.0001277984670027933,
      "loss": 13.6321,
      "step": 1689
    },
    {
      "epoch": 0.7750515936711764,
      "grad_norm": 0.5370873487897959,
      "learning_rate": 0.00012730272459555737,
      "loss": 13.5382,
      "step": 1690
    },
    {
      "epoch": 0.7755102040816326,
      "grad_norm": 0.5197750080800654,
      "learning_rate": 0.00012680780527565312,
      "loss": 13.3563,
      "step": 1691
    },
    {
      "epoch": 0.775968814492089,
      "grad_norm": 0.5275834296080708,
      "learning_rate": 0.0001263137101360905,
      "loss": 13.7071,
      "step": 1692
    },
    {
      "epoch": 0.7764274249025452,
      "grad_norm": 0.5457541513070224,
      "learning_rate": 0.00012582044026805922,
      "loss": 13.479,
      "step": 1693
    },
    {
      "epoch": 0.7768860353130016,
      "grad_norm": 0.5419452874623402,
      "learning_rate": 0.00012532799676092627,
      "loss": 13.686,
      "step": 1694
    },
    {
      "epoch": 0.777344645723458,
      "grad_norm": 0.5604855222637777,
      "learning_rate": 0.00012483638070223414,
      "loss": 13.5018,
      "step": 1695
    },
    {
      "epoch": 0.7778032561339142,
      "grad_norm": 0.5504701076682887,
      "learning_rate": 0.00012434559317769752,
      "loss": 13.6608,
      "step": 1696
    },
    {
      "epoch": 0.7782618665443706,
      "grad_norm": 0.5647509401361819,
      "learning_rate": 0.0001238556352712012,
      "loss": 13.7568,
      "step": 1697
    },
    {
      "epoch": 0.7787204769548268,
      "grad_norm": 0.5436540860743807,
      "learning_rate": 0.00012336650806479827,
      "loss": 13.506,
      "step": 1698
    },
    {
      "epoch": 0.7791790873652832,
      "grad_norm": 0.5817312082879185,
      "learning_rate": 0.00012287821263870708,
      "loss": 13.5032,
      "step": 1699
    },
    {
      "epoch": 0.7796376977757395,
      "grad_norm": 0.5521081820536219,
      "learning_rate": 0.00012239075007130885,
      "loss": 13.5647,
      "step": 1700
    },
    {
      "epoch": 0.7800963081861958,
      "grad_norm": 0.5312493629749769,
      "learning_rate": 0.00012190412143914536,
      "loss": 13.3112,
      "step": 1701
    },
    {
      "epoch": 0.7805549185966522,
      "grad_norm": 0.5568220717022704,
      "learning_rate": 0.0001214183278169172,
      "loss": 13.4981,
      "step": 1702
    },
    {
      "epoch": 0.7810135290071084,
      "grad_norm": 0.5318699315582688,
      "learning_rate": 0.00012093337027748042,
      "loss": 13.4332,
      "step": 1703
    },
    {
      "epoch": 0.7814721394175648,
      "grad_norm": 0.535655198690832,
      "learning_rate": 0.00012044924989184459,
      "loss": 13.4369,
      "step": 1704
    },
    {
      "epoch": 0.7819307498280211,
      "grad_norm": 0.5198158501054086,
      "learning_rate": 0.0001199659677291709,
      "loss": 13.6422,
      "step": 1705
    },
    {
      "epoch": 0.7823893602384774,
      "grad_norm": 0.5652620358543353,
      "learning_rate": 0.00011948352485676895,
      "loss": 13.393,
      "step": 1706
    },
    {
      "epoch": 0.7828479706489337,
      "grad_norm": 0.5270612549443519,
      "learning_rate": 0.00011900192234009477,
      "loss": 13.5423,
      "step": 1707
    },
    {
      "epoch": 0.78330658105939,
      "grad_norm": 0.5544695831188,
      "learning_rate": 0.00011852116124274875,
      "loss": 13.5501,
      "step": 1708
    },
    {
      "epoch": 0.7837651914698464,
      "grad_norm": 0.5604525112431605,
      "learning_rate": 0.00011804124262647314,
      "loss": 13.6705,
      "step": 1709
    },
    {
      "epoch": 0.7842238018803027,
      "grad_norm": 0.5506568471867203,
      "learning_rate": 0.00011756216755114929,
      "loss": 13.5718,
      "step": 1710
    },
    {
      "epoch": 0.784682412290759,
      "grad_norm": 0.545112351186286,
      "learning_rate": 0.00011708393707479548,
      "loss": 13.4963,
      "step": 1711
    },
    {
      "epoch": 0.7851410227012153,
      "grad_norm": 0.5344549815219278,
      "learning_rate": 0.00011660655225356531,
      "loss": 13.7018,
      "step": 1712
    },
    {
      "epoch": 0.7855996331116716,
      "grad_norm": 0.5551220908578232,
      "learning_rate": 0.0001161300141417444,
      "loss": 13.5234,
      "step": 1713
    },
    {
      "epoch": 0.7860582435221279,
      "grad_norm": 0.5489460947309001,
      "learning_rate": 0.00011565432379174823,
      "loss": 13.4846,
      "step": 1714
    },
    {
      "epoch": 0.7865168539325843,
      "grad_norm": 0.5437903568758541,
      "learning_rate": 0.00011517948225412056,
      "loss": 13.709,
      "step": 1715
    },
    {
      "epoch": 0.7869754643430406,
      "grad_norm": 0.5824064055383398,
      "learning_rate": 0.00011470549057753032,
      "loss": 13.4962,
      "step": 1716
    },
    {
      "epoch": 0.7874340747534969,
      "grad_norm": 0.5251233793218676,
      "learning_rate": 0.00011423234980876957,
      "loss": 13.3148,
      "step": 1717
    },
    {
      "epoch": 0.7878926851639533,
      "grad_norm": 0.5623469024339572,
      "learning_rate": 0.00011376006099275099,
      "loss": 13.43,
      "step": 1718
    },
    {
      "epoch": 0.7883512955744095,
      "grad_norm": 0.5419704656286205,
      "learning_rate": 0.00011328862517250609,
      "loss": 13.6692,
      "step": 1719
    },
    {
      "epoch": 0.7888099059848659,
      "grad_norm": 0.5839344821862489,
      "learning_rate": 0.00011281804338918239,
      "loss": 13.6583,
      "step": 1720
    },
    {
      "epoch": 0.7892685163953221,
      "grad_norm": 0.5392093536160134,
      "learning_rate": 0.00011234831668204115,
      "loss": 13.4815,
      "step": 1721
    },
    {
      "epoch": 0.7897271268057785,
      "grad_norm": 0.5517231774220016,
      "learning_rate": 0.00011187944608845569,
      "loss": 13.5899,
      "step": 1722
    },
    {
      "epoch": 0.7901857372162349,
      "grad_norm": 0.5533686356477816,
      "learning_rate": 0.00011141143264390801,
      "loss": 13.7717,
      "step": 1723
    },
    {
      "epoch": 0.7906443476266911,
      "grad_norm": 0.5522444195473251,
      "learning_rate": 0.0001109442773819877,
      "loss": 13.494,
      "step": 1724
    },
    {
      "epoch": 0.7911029580371475,
      "grad_norm": 0.5730727297008652,
      "learning_rate": 0.0001104779813343889,
      "loss": 13.2612,
      "step": 1725
    },
    {
      "epoch": 0.7915615684476037,
      "grad_norm": 0.5594265713241136,
      "learning_rate": 0.00011001254553090812,
      "loss": 13.3245,
      "step": 1726
    },
    {
      "epoch": 0.7920201788580601,
      "grad_norm": 0.5470754505111626,
      "learning_rate": 0.00010954797099944186,
      "loss": 13.4306,
      "step": 1727
    },
    {
      "epoch": 0.7924787892685164,
      "grad_norm": 0.5685049451066918,
      "learning_rate": 0.0001090842587659851,
      "loss": 13.5351,
      "step": 1728
    },
    {
      "epoch": 0.7929373996789727,
      "grad_norm": 0.5482974477634067,
      "learning_rate": 0.00010862140985462804,
      "loss": 13.5416,
      "step": 1729
    },
    {
      "epoch": 0.7933960100894291,
      "grad_norm": 0.6039250907574476,
      "learning_rate": 0.00010815942528755418,
      "loss": 13.5331,
      "step": 1730
    },
    {
      "epoch": 0.7938546204998853,
      "grad_norm": 0.563783787484826,
      "learning_rate": 0.00010769830608503844,
      "loss": 13.7211,
      "step": 1731
    },
    {
      "epoch": 0.7943132309103417,
      "grad_norm": 0.5594151970103536,
      "learning_rate": 0.00010723805326544473,
      "loss": 13.5728,
      "step": 1732
    },
    {
      "epoch": 0.794771841320798,
      "grad_norm": 0.5422715064449639,
      "learning_rate": 0.00010677866784522316,
      "loss": 13.5634,
      "step": 1733
    },
    {
      "epoch": 0.7952304517312543,
      "grad_norm": 0.5962067361137035,
      "learning_rate": 0.00010632015083890839,
      "loss": 13.7066,
      "step": 1734
    },
    {
      "epoch": 0.7956890621417106,
      "grad_norm": 0.5567080929508613,
      "learning_rate": 0.00010586250325911745,
      "loss": 13.3913,
      "step": 1735
    },
    {
      "epoch": 0.7961476725521669,
      "grad_norm": 0.5365077609326069,
      "learning_rate": 0.00010540572611654697,
      "loss": 13.5966,
      "step": 1736
    },
    {
      "epoch": 0.7966062829626233,
      "grad_norm": 0.5886049556602222,
      "learning_rate": 0.00010494982041997126,
      "loss": 13.6612,
      "step": 1737
    },
    {
      "epoch": 0.7970648933730796,
      "grad_norm": 0.5304730199577382,
      "learning_rate": 0.0001044947871762405,
      "loss": 13.5088,
      "step": 1738
    },
    {
      "epoch": 0.7975235037835359,
      "grad_norm": 0.5262878805866559,
      "learning_rate": 0.00010404062739027753,
      "loss": 13.6313,
      "step": 1739
    },
    {
      "epoch": 0.7979821141939922,
      "grad_norm": 0.5325478992359972,
      "learning_rate": 0.00010358734206507641,
      "loss": 13.3669,
      "step": 1740
    },
    {
      "epoch": 0.7984407246044485,
      "grad_norm": 0.5558865777099498,
      "learning_rate": 0.00010313493220170017,
      "loss": 13.4259,
      "step": 1741
    },
    {
      "epoch": 0.7988993350149048,
      "grad_norm": 0.5363351869014016,
      "learning_rate": 0.00010268339879927836,
      "loss": 13.5436,
      "step": 1742
    },
    {
      "epoch": 0.7993579454253612,
      "grad_norm": 0.5258544236118523,
      "learning_rate": 0.00010223274285500466,
      "loss": 13.5084,
      "step": 1743
    },
    {
      "epoch": 0.7998165558358175,
      "grad_norm": 0.570300669171802,
      "learning_rate": 0.00010178296536413495,
      "loss": 13.3452,
      "step": 1744
    },
    {
      "epoch": 0.8002751662462738,
      "grad_norm": 0.5312008806531676,
      "learning_rate": 0.00010133406731998546,
      "loss": 13.7218,
      "step": 1745
    },
    {
      "epoch": 0.8007337766567301,
      "grad_norm": 0.5323168358669599,
      "learning_rate": 0.00010088604971392979,
      "loss": 13.4739,
      "step": 1746
    },
    {
      "epoch": 0.8011923870671864,
      "grad_norm": 0.5345756085195843,
      "learning_rate": 0.0001004389135353972,
      "loss": 13.5028,
      "step": 1747
    },
    {
      "epoch": 0.8016509974776428,
      "grad_norm": 0.5249758203223255,
      "learning_rate": 9.999265977187049e-05,
      "loss": 13.5213,
      "step": 1748
    },
    {
      "epoch": 0.802109607888099,
      "grad_norm": 0.5298666498026577,
      "learning_rate": 9.95472894088838e-05,
      "loss": 13.4466,
      "step": 1749
    },
    {
      "epoch": 0.8025682182985554,
      "grad_norm": 0.5139917822958705,
      "learning_rate": 9.910280343001993e-05,
      "loss": 13.3142,
      "step": 1750
    },
    {
      "epoch": 0.8030268287090117,
      "grad_norm": 0.5391328595552553,
      "learning_rate": 9.865920281690866e-05,
      "loss": 13.5468,
      "step": 1751
    },
    {
      "epoch": 0.803485439119468,
      "grad_norm": 0.5607645558320712,
      "learning_rate": 9.821648854922482e-05,
      "loss": 13.4724,
      "step": 1752
    },
    {
      "epoch": 0.8039440495299244,
      "grad_norm": 0.5457680916377422,
      "learning_rate": 9.77746616046854e-05,
      "loss": 13.4681,
      "step": 1753
    },
    {
      "epoch": 0.8044026599403806,
      "grad_norm": 0.557237432760248,
      "learning_rate": 9.733372295904774e-05,
      "loss": 13.4962,
      "step": 1754
    },
    {
      "epoch": 0.804861270350837,
      "grad_norm": 0.5781360066178045,
      "learning_rate": 9.68936735861079e-05,
      "loss": 13.5304,
      "step": 1755
    },
    {
      "epoch": 0.8053198807612932,
      "grad_norm": 0.5388206977516945,
      "learning_rate": 9.645451445769737e-05,
      "loss": 13.4022,
      "step": 1756
    },
    {
      "epoch": 0.8057784911717496,
      "grad_norm": 0.5364090022659725,
      "learning_rate": 9.601624654368196e-05,
      "loss": 13.5625,
      "step": 1757
    },
    {
      "epoch": 0.806237101582206,
      "grad_norm": 0.5901960150364004,
      "learning_rate": 9.557887081195938e-05,
      "loss": 13.7652,
      "step": 1758
    },
    {
      "epoch": 0.8066957119926622,
      "grad_norm": 0.5341078626702204,
      "learning_rate": 9.514238822845667e-05,
      "loss": 13.5634,
      "step": 1759
    },
    {
      "epoch": 0.8071543224031186,
      "grad_norm": 0.5711518528402031,
      "learning_rate": 9.470679975712837e-05,
      "loss": 13.6046,
      "step": 1760
    },
    {
      "epoch": 0.8076129328135748,
      "grad_norm": 0.5807694819069271,
      "learning_rate": 9.427210635995481e-05,
      "loss": 13.4139,
      "step": 1761
    },
    {
      "epoch": 0.8080715432240312,
      "grad_norm": 0.547748302649367,
      "learning_rate": 9.383830899693923e-05,
      "loss": 13.6175,
      "step": 1762
    },
    {
      "epoch": 0.8085301536344875,
      "grad_norm": 0.558960988338915,
      "learning_rate": 9.340540862610591e-05,
      "loss": 13.4774,
      "step": 1763
    },
    {
      "epoch": 0.8089887640449438,
      "grad_norm": 0.5662687202492935,
      "learning_rate": 9.297340620349854e-05,
      "loss": 13.4329,
      "step": 1764
    },
    {
      "epoch": 0.8094473744554002,
      "grad_norm": 0.5499734354902137,
      "learning_rate": 9.25423026831777e-05,
      "loss": 13.4058,
      "step": 1765
    },
    {
      "epoch": 0.8099059848658564,
      "grad_norm": 0.6090859356446635,
      "learning_rate": 9.211209901721846e-05,
      "loss": 13.4496,
      "step": 1766
    },
    {
      "epoch": 0.8103645952763128,
      "grad_norm": 0.5598139251310551,
      "learning_rate": 9.168279615570863e-05,
      "loss": 13.3986,
      "step": 1767
    },
    {
      "epoch": 0.8108232056867691,
      "grad_norm": 0.5352087793447586,
      "learning_rate": 9.125439504674699e-05,
      "loss": 13.2244,
      "step": 1768
    },
    {
      "epoch": 0.8112818160972254,
      "grad_norm": 0.5819850569674263,
      "learning_rate": 9.082689663644057e-05,
      "loss": 13.5673,
      "step": 1769
    },
    {
      "epoch": 0.8117404265076817,
      "grad_norm": 0.534559004710724,
      "learning_rate": 9.040030186890264e-05,
      "loss": 13.5137,
      "step": 1770
    },
    {
      "epoch": 0.812199036918138,
      "grad_norm": 0.53862711632825,
      "learning_rate": 8.997461168625138e-05,
      "loss": 13.5881,
      "step": 1771
    },
    {
      "epoch": 0.8126576473285944,
      "grad_norm": 0.5345630751884013,
      "learning_rate": 8.954982702860664e-05,
      "loss": 13.4688,
      "step": 1772
    },
    {
      "epoch": 0.8131162577390507,
      "grad_norm": 0.6207062637899127,
      "learning_rate": 8.912594883408865e-05,
      "loss": 13.6609,
      "step": 1773
    },
    {
      "epoch": 0.813574868149507,
      "grad_norm": 0.6148065585759863,
      "learning_rate": 8.870297803881589e-05,
      "loss": 13.5272,
      "step": 1774
    },
    {
      "epoch": 0.8140334785599633,
      "grad_norm": 0.5387317007023281,
      "learning_rate": 8.828091557690287e-05,
      "loss": 13.5101,
      "step": 1775
    },
    {
      "epoch": 0.8144920889704196,
      "grad_norm": 0.537300201026007,
      "learning_rate": 8.785976238045801e-05,
      "loss": 13.2292,
      "step": 1776
    },
    {
      "epoch": 0.8149506993808759,
      "grad_norm": 0.5548597279570189,
      "learning_rate": 8.743951937958144e-05,
      "loss": 13.3254,
      "step": 1777
    },
    {
      "epoch": 0.8154093097913323,
      "grad_norm": 0.5812144358407436,
      "learning_rate": 8.702018750236357e-05,
      "loss": 13.4105,
      "step": 1778
    },
    {
      "epoch": 0.8158679202017886,
      "grad_norm": 0.61338688311213,
      "learning_rate": 8.660176767488237e-05,
      "loss": 13.6147,
      "step": 1779
    },
    {
      "epoch": 0.8163265306122449,
      "grad_norm": 0.5497443869488334,
      "learning_rate": 8.618426082120146e-05,
      "loss": 13.4219,
      "step": 1780
    },
    {
      "epoch": 0.8167851410227012,
      "grad_norm": 0.5880739846498726,
      "learning_rate": 8.576766786336854e-05,
      "loss": 13.4443,
      "step": 1781
    },
    {
      "epoch": 0.8172437514331575,
      "grad_norm": 0.5775762147003056,
      "learning_rate": 8.535198972141294e-05,
      "loss": 13.7785,
      "step": 1782
    },
    {
      "epoch": 0.8177023618436139,
      "grad_norm": 0.5524323456712231,
      "learning_rate": 8.493722731334347e-05,
      "loss": 13.4987,
      "step": 1783
    },
    {
      "epoch": 0.8181609722540701,
      "grad_norm": 0.5666834038054914,
      "learning_rate": 8.452338155514644e-05,
      "loss": 13.6622,
      "step": 1784
    },
    {
      "epoch": 0.8186195826645265,
      "grad_norm": 0.5926416366831214,
      "learning_rate": 8.411045336078426e-05,
      "loss": 13.4797,
      "step": 1785
    },
    {
      "epoch": 0.8190781930749828,
      "grad_norm": 0.5460360226557943,
      "learning_rate": 8.369844364219264e-05,
      "loss": 13.5222,
      "step": 1786
    },
    {
      "epoch": 0.8195368034854391,
      "grad_norm": 0.5698734585375914,
      "learning_rate": 8.328735330927873e-05,
      "loss": 13.3211,
      "step": 1787
    },
    {
      "epoch": 0.8199954138958955,
      "grad_norm": 0.6064709241038355,
      "learning_rate": 8.287718326991961e-05,
      "loss": 13.5655,
      "step": 1788
    },
    {
      "epoch": 0.8204540243063517,
      "grad_norm": 0.5414710660509572,
      "learning_rate": 8.246793442995954e-05,
      "loss": 13.5299,
      "step": 1789
    },
    {
      "epoch": 0.8209126347168081,
      "grad_norm": 0.5299818590092177,
      "learning_rate": 8.205960769320875e-05,
      "loss": 13.3363,
      "step": 1790
    },
    {
      "epoch": 0.8213712451272643,
      "grad_norm": 0.6329811950844169,
      "learning_rate": 8.165220396144085e-05,
      "loss": 13.6791,
      "step": 1791
    },
    {
      "epoch": 0.8218298555377207,
      "grad_norm": 0.5797548069350719,
      "learning_rate": 8.12457241343909e-05,
      "loss": 13.5161,
      "step": 1792
    },
    {
      "epoch": 0.8222884659481771,
      "grad_norm": 0.5571759769905743,
      "learning_rate": 8.084016910975367e-05,
      "loss": 13.5496,
      "step": 1793
    },
    {
      "epoch": 0.8227470763586333,
      "grad_norm": 0.5418177968491367,
      "learning_rate": 8.043553978318169e-05,
      "loss": 13.5075,
      "step": 1794
    },
    {
      "epoch": 0.8232056867690897,
      "grad_norm": 0.5491843037177586,
      "learning_rate": 8.003183704828281e-05,
      "loss": 13.3424,
      "step": 1795
    },
    {
      "epoch": 0.8236642971795459,
      "grad_norm": 0.5239105560138292,
      "learning_rate": 7.962906179661872e-05,
      "loss": 13.4363,
      "step": 1796
    },
    {
      "epoch": 0.8241229075900023,
      "grad_norm": 0.6578529130862799,
      "learning_rate": 7.922721491770296e-05,
      "loss": 13.3266,
      "step": 1797
    },
    {
      "epoch": 0.8245815180004586,
      "grad_norm": 0.5606911198263099,
      "learning_rate": 7.882629729899832e-05,
      "loss": 13.3436,
      "step": 1798
    },
    {
      "epoch": 0.8250401284109149,
      "grad_norm": 0.535883037780828,
      "learning_rate": 7.842630982591598e-05,
      "loss": 13.5292,
      "step": 1799
    },
    {
      "epoch": 0.8254987388213713,
      "grad_norm": 0.5388753944634778,
      "learning_rate": 7.802725338181232e-05,
      "loss": 13.5083,
      "step": 1800
    },
    {
      "epoch": 0.8259573492318275,
      "grad_norm": 0.57136221991916,
      "learning_rate": 7.762912884798812e-05,
      "loss": 13.4685,
      "step": 1801
    },
    {
      "epoch": 0.8264159596422839,
      "grad_norm": 0.5372863508310565,
      "learning_rate": 7.723193710368564e-05,
      "loss": 13.4387,
      "step": 1802
    },
    {
      "epoch": 0.8268745700527402,
      "grad_norm": 0.551072020241548,
      "learning_rate": 7.683567902608729e-05,
      "loss": 13.4088,
      "step": 1803
    },
    {
      "epoch": 0.8273331804631965,
      "grad_norm": 0.5807143807622616,
      "learning_rate": 7.644035549031364e-05,
      "loss": 13.4595,
      "step": 1804
    },
    {
      "epoch": 0.8277917908736528,
      "grad_norm": 0.5676446788850416,
      "learning_rate": 7.604596736942115e-05,
      "loss": 13.3177,
      "step": 1805
    },
    {
      "epoch": 0.8282504012841091,
      "grad_norm": 0.533133527122522,
      "learning_rate": 7.56525155344004e-05,
      "loss": 13.4769,
      "step": 1806
    },
    {
      "epoch": 0.8287090116945655,
      "grad_norm": 0.5681118194565966,
      "learning_rate": 7.52600008541745e-05,
      "loss": 13.5079,
      "step": 1807
    },
    {
      "epoch": 0.8291676221050218,
      "grad_norm": 0.5814793701691962,
      "learning_rate": 7.486842419559681e-05,
      "loss": 13.573,
      "step": 1808
    },
    {
      "epoch": 0.8296262325154781,
      "grad_norm": 0.5560540331339897,
      "learning_rate": 7.447778642344898e-05,
      "loss": 13.6791,
      "step": 1809
    },
    {
      "epoch": 0.8300848429259344,
      "grad_norm": 0.5403583912130793,
      "learning_rate": 7.408808840043912e-05,
      "loss": 13.6277,
      "step": 1810
    },
    {
      "epoch": 0.8305434533363908,
      "grad_norm": 0.581938623212003,
      "learning_rate": 7.369933098720021e-05,
      "loss": 13.621,
      "step": 1811
    },
    {
      "epoch": 0.831002063746847,
      "grad_norm": 0.5485847489691628,
      "learning_rate": 7.331151504228767e-05,
      "loss": 13.38,
      "step": 1812
    },
    {
      "epoch": 0.8314606741573034,
      "grad_norm": 0.5434718594837354,
      "learning_rate": 7.292464142217775e-05,
      "loss": 13.52,
      "step": 1813
    },
    {
      "epoch": 0.8319192845677597,
      "grad_norm": 0.5662499804159923,
      "learning_rate": 7.25387109812658e-05,
      "loss": 13.5112,
      "step": 1814
    },
    {
      "epoch": 0.832377894978216,
      "grad_norm": 0.5675480242172017,
      "learning_rate": 7.215372457186415e-05,
      "loss": 13.5253,
      "step": 1815
    },
    {
      "epoch": 0.8328365053886724,
      "grad_norm": 0.538916101695029,
      "learning_rate": 7.176968304420007e-05,
      "loss": 13.3898,
      "step": 1816
    },
    {
      "epoch": 0.8332951157991286,
      "grad_norm": 0.5157183062415637,
      "learning_rate": 7.138658724641417e-05,
      "loss": 13.5852,
      "step": 1817
    },
    {
      "epoch": 0.833753726209585,
      "grad_norm": 0.575249101911896,
      "learning_rate": 7.10044380245587e-05,
      "loss": 13.5973,
      "step": 1818
    },
    {
      "epoch": 0.8342123366200412,
      "grad_norm": 0.5433807349480185,
      "learning_rate": 7.062323622259515e-05,
      "loss": 13.3926,
      "step": 1819
    },
    {
      "epoch": 0.8346709470304976,
      "grad_norm": 0.5535095938267828,
      "learning_rate": 7.024298268239265e-05,
      "loss": 13.4155,
      "step": 1820
    },
    {
      "epoch": 0.835129557440954,
      "grad_norm": 0.5335738610167613,
      "learning_rate": 6.986367824372647e-05,
      "loss": 13.2624,
      "step": 1821
    },
    {
      "epoch": 0.8355881678514102,
      "grad_norm": 0.5411974406891142,
      "learning_rate": 6.948532374427541e-05,
      "loss": 13.6496,
      "step": 1822
    },
    {
      "epoch": 0.8360467782618666,
      "grad_norm": 0.5468772664619211,
      "learning_rate": 6.910792001962063e-05,
      "loss": 13.3079,
      "step": 1823
    },
    {
      "epoch": 0.8365053886723228,
      "grad_norm": 0.5401213550468917,
      "learning_rate": 6.873146790324358e-05,
      "loss": 13.6747,
      "step": 1824
    },
    {
      "epoch": 0.8369639990827792,
      "grad_norm": 0.5326790176953192,
      "learning_rate": 6.83559682265239e-05,
      "loss": 13.4479,
      "step": 1825
    },
    {
      "epoch": 0.8374226094932355,
      "grad_norm": 0.5398317617339123,
      "learning_rate": 6.798142181873784e-05,
      "loss": 13.6069,
      "step": 1826
    },
    {
      "epoch": 0.8378812199036918,
      "grad_norm": 0.5410079010866539,
      "learning_rate": 6.760782950705662e-05,
      "loss": 13.4436,
      "step": 1827
    },
    {
      "epoch": 0.8383398303141482,
      "grad_norm": 0.5455966601130966,
      "learning_rate": 6.723519211654422e-05,
      "loss": 13.5288,
      "step": 1828
    },
    {
      "epoch": 0.8387984407246044,
      "grad_norm": 0.5380455493481401,
      "learning_rate": 6.686351047015554e-05,
      "loss": 13.4647,
      "step": 1829
    },
    {
      "epoch": 0.8392570511350608,
      "grad_norm": 0.5381549835906776,
      "learning_rate": 6.649278538873515e-05,
      "loss": 13.6018,
      "step": 1830
    },
    {
      "epoch": 0.8397156615455171,
      "grad_norm": 0.5596426741013116,
      "learning_rate": 6.612301769101465e-05,
      "loss": 13.5858,
      "step": 1831
    },
    {
      "epoch": 0.8401742719559734,
      "grad_norm": 0.5384097336638081,
      "learning_rate": 6.575420819361177e-05,
      "loss": 13.5912,
      "step": 1832
    },
    {
      "epoch": 0.8406328823664297,
      "grad_norm": 0.5868401956965542,
      "learning_rate": 6.538635771102757e-05,
      "loss": 13.5521,
      "step": 1833
    },
    {
      "epoch": 0.841091492776886,
      "grad_norm": 0.5232477740562322,
      "learning_rate": 6.501946705564566e-05,
      "loss": 13.5248,
      "step": 1834
    },
    {
      "epoch": 0.8415501031873424,
      "grad_norm": 0.538349823578786,
      "learning_rate": 6.465353703772959e-05,
      "loss": 13.5131,
      "step": 1835
    },
    {
      "epoch": 0.8420087135977987,
      "grad_norm": 0.5451541332174803,
      "learning_rate": 6.428856846542136e-05,
      "loss": 13.5985,
      "step": 1836
    },
    {
      "epoch": 0.842467324008255,
      "grad_norm": 0.5273758044037524,
      "learning_rate": 6.392456214473996e-05,
      "loss": 13.5521,
      "step": 1837
    },
    {
      "epoch": 0.8429259344187113,
      "grad_norm": 0.5407587250093355,
      "learning_rate": 6.3561518879579e-05,
      "loss": 13.4839,
      "step": 1838
    },
    {
      "epoch": 0.8433845448291676,
      "grad_norm": 0.5492960171624327,
      "learning_rate": 6.31994394717052e-05,
      "loss": 13.4965,
      "step": 1839
    },
    {
      "epoch": 0.8438431552396239,
      "grad_norm": 0.5684791245023025,
      "learning_rate": 6.283832472075685e-05,
      "loss": 13.46,
      "step": 1840
    },
    {
      "epoch": 0.8443017656500803,
      "grad_norm": 0.567423000200974,
      "learning_rate": 6.247817542424178e-05,
      "loss": 13.5062,
      "step": 1841
    },
    {
      "epoch": 0.8447603760605366,
      "grad_norm": 0.549926224770748,
      "learning_rate": 6.211899237753559e-05,
      "loss": 13.7118,
      "step": 1842
    },
    {
      "epoch": 0.8452189864709929,
      "grad_norm": 0.5490739026433238,
      "learning_rate": 6.176077637387984e-05,
      "loss": 13.5355,
      "step": 1843
    },
    {
      "epoch": 0.8456775968814492,
      "grad_norm": 0.5518029283122398,
      "learning_rate": 6.140352820438066e-05,
      "loss": 13.5716,
      "step": 1844
    },
    {
      "epoch": 0.8461362072919055,
      "grad_norm": 0.5549831918165992,
      "learning_rate": 6.104724865800665e-05,
      "loss": 13.4876,
      "step": 1845
    },
    {
      "epoch": 0.8465948177023619,
      "grad_norm": 0.555474695292578,
      "learning_rate": 6.069193852158711e-05,
      "loss": 13.6222,
      "step": 1846
    },
    {
      "epoch": 0.8470534281128181,
      "grad_norm": 0.5622565546499402,
      "learning_rate": 6.0337598579810584e-05,
      "loss": 13.5643,
      "step": 1847
    },
    {
      "epoch": 0.8475120385232745,
      "grad_norm": 0.5617874403935718,
      "learning_rate": 5.9984229615223096e-05,
      "loss": 13.6014,
      "step": 1848
    },
    {
      "epoch": 0.8479706489337308,
      "grad_norm": 0.5289482709787681,
      "learning_rate": 5.963183240822606e-05,
      "loss": 13.5269,
      "step": 1849
    },
    {
      "epoch": 0.8484292593441871,
      "grad_norm": 0.5623389614159832,
      "learning_rate": 5.9280407737074825e-05,
      "loss": 13.314,
      "step": 1850
    },
    {
      "epoch": 0.8488878697546435,
      "grad_norm": 0.5613140277562351,
      "learning_rate": 5.8929956377877125e-05,
      "loss": 13.3871,
      "step": 1851
    },
    {
      "epoch": 0.8493464801650997,
      "grad_norm": 0.5260049492683099,
      "learning_rate": 5.8580479104591075e-05,
      "loss": 13.3586,
      "step": 1852
    },
    {
      "epoch": 0.8498050905755561,
      "grad_norm": 0.5544951039755222,
      "learning_rate": 5.823197668902341e-05,
      "loss": 13.3539,
      "step": 1853
    },
    {
      "epoch": 0.8502637009860123,
      "grad_norm": 0.5548775214789248,
      "learning_rate": 5.78844499008282e-05,
      "loss": 13.3855,
      "step": 1854
    },
    {
      "epoch": 0.8507223113964687,
      "grad_norm": 0.54648653990117,
      "learning_rate": 5.753789950750454e-05,
      "loss": 13.4363,
      "step": 1855
    },
    {
      "epoch": 0.8511809218069251,
      "grad_norm": 0.5605142912881383,
      "learning_rate": 5.719232627439558e-05,
      "loss": 13.4635,
      "step": 1856
    },
    {
      "epoch": 0.8516395322173813,
      "grad_norm": 0.5046479889148473,
      "learning_rate": 5.6847730964686315e-05,
      "loss": 13.3101,
      "step": 1857
    },
    {
      "epoch": 0.8520981426278377,
      "grad_norm": 0.5533650203293266,
      "learning_rate": 5.650411433940189e-05,
      "loss": 13.4577,
      "step": 1858
    },
    {
      "epoch": 0.8525567530382939,
      "grad_norm": 0.5184408045302561,
      "learning_rate": 5.61614771574061e-05,
      "loss": 13.2973,
      "step": 1859
    },
    {
      "epoch": 0.8530153634487503,
      "grad_norm": 0.5713067113155382,
      "learning_rate": 5.581982017539988e-05,
      "loss": 13.5145,
      "step": 1860
    },
    {
      "epoch": 0.8534739738592066,
      "grad_norm": 0.5508531929690254,
      "learning_rate": 5.5479144147919216e-05,
      "loss": 13.5254,
      "step": 1861
    },
    {
      "epoch": 0.8539325842696629,
      "grad_norm": 0.5800843541536558,
      "learning_rate": 5.51394498273336e-05,
      "loss": 13.4852,
      "step": 1862
    },
    {
      "epoch": 0.8543911946801193,
      "grad_norm": 0.5898816490471316,
      "learning_rate": 5.480073796384494e-05,
      "loss": 13.2591,
      "step": 1863
    },
    {
      "epoch": 0.8548498050905755,
      "grad_norm": 0.5295264792568616,
      "learning_rate": 5.446300930548492e-05,
      "loss": 13.3534,
      "step": 1864
    },
    {
      "epoch": 0.8553084155010319,
      "grad_norm": 0.5738102958823011,
      "learning_rate": 5.412626459811415e-05,
      "loss": 13.1257,
      "step": 1865
    },
    {
      "epoch": 0.8557670259114882,
      "grad_norm": 0.5529131579143296,
      "learning_rate": 5.3790504585419954e-05,
      "loss": 13.3979,
      "step": 1866
    },
    {
      "epoch": 0.8562256363219445,
      "grad_norm": 0.5388925695283162,
      "learning_rate": 5.345573000891541e-05,
      "loss": 13.5642,
      "step": 1867
    },
    {
      "epoch": 0.8566842467324008,
      "grad_norm": 0.5692019570757346,
      "learning_rate": 5.312194160793693e-05,
      "loss": 13.5173,
      "step": 1868
    },
    {
      "epoch": 0.8571428571428571,
      "grad_norm": 0.5472210168356675,
      "learning_rate": 5.278914011964303e-05,
      "loss": 13.4151,
      "step": 1869
    },
    {
      "epoch": 0.8576014675533135,
      "grad_norm": 0.5573180257564451,
      "learning_rate": 5.2457326279013006e-05,
      "loss": 13.564,
      "step": 1870
    },
    {
      "epoch": 0.8580600779637698,
      "grad_norm": 0.5370309690121505,
      "learning_rate": 5.2126500818844514e-05,
      "loss": 13.4407,
      "step": 1871
    },
    {
      "epoch": 0.8585186883742261,
      "grad_norm": 0.5376895560504822,
      "learning_rate": 5.1796664469752566e-05,
      "loss": 13.3021,
      "step": 1872
    },
    {
      "epoch": 0.8589772987846824,
      "grad_norm": 0.5411343308732943,
      "learning_rate": 5.1467817960167975e-05,
      "loss": 13.4594,
      "step": 1873
    },
    {
      "epoch": 0.8594359091951387,
      "grad_norm": 0.5792632014081616,
      "learning_rate": 5.113996201633536e-05,
      "loss": 13.4601,
      "step": 1874
    },
    {
      "epoch": 0.859894519605595,
      "grad_norm": 0.5558512084251119,
      "learning_rate": 5.0813097362311765e-05,
      "loss": 13.3475,
      "step": 1875
    },
    {
      "epoch": 0.8603531300160514,
      "grad_norm": 0.5373648332931754,
      "learning_rate": 5.048722471996475e-05,
      "loss": 13.3517,
      "step": 1876
    },
    {
      "epoch": 0.8608117404265077,
      "grad_norm": 0.5361107344415579,
      "learning_rate": 5.016234480897158e-05,
      "loss": 13.4935,
      "step": 1877
    },
    {
      "epoch": 0.861270350836964,
      "grad_norm": 0.5165656249448833,
      "learning_rate": 4.9838458346816664e-05,
      "loss": 13.5115,
      "step": 1878
    },
    {
      "epoch": 0.8617289612474203,
      "grad_norm": 0.5615070226592052,
      "learning_rate": 4.9515566048790485e-05,
      "loss": 13.5519,
      "step": 1879
    },
    {
      "epoch": 0.8621875716578766,
      "grad_norm": 0.5380807420268608,
      "learning_rate": 4.9193668627988074e-05,
      "loss": 13.3516,
      "step": 1880
    },
    {
      "epoch": 0.862646182068333,
      "grad_norm": 0.5498702116503652,
      "learning_rate": 4.887276679530744e-05,
      "loss": 13.5165,
      "step": 1881
    },
    {
      "epoch": 0.8631047924787892,
      "grad_norm": 0.5626392972265822,
      "learning_rate": 4.855286125944752e-05,
      "loss": 13.4807,
      "step": 1882
    },
    {
      "epoch": 0.8635634028892456,
      "grad_norm": 0.5710614207323736,
      "learning_rate": 4.8233952726907224e-05,
      "loss": 13.5449,
      "step": 1883
    },
    {
      "epoch": 0.8640220132997019,
      "grad_norm": 0.5453922331536566,
      "learning_rate": 4.7916041901983565e-05,
      "loss": 13.4256,
      "step": 1884
    },
    {
      "epoch": 0.8644806237101582,
      "grad_norm": 0.5699171118867322,
      "learning_rate": 4.7599129486770145e-05,
      "loss": 13.654,
      "step": 1885
    },
    {
      "epoch": 0.8649392341206146,
      "grad_norm": 0.5497161470403844,
      "learning_rate": 4.728321618115555e-05,
      "loss": 13.6247,
      "step": 1886
    },
    {
      "epoch": 0.8653978445310708,
      "grad_norm": 0.575390801135138,
      "learning_rate": 4.696830268282204e-05,
      "loss": 13.6471,
      "step": 1887
    },
    {
      "epoch": 0.8658564549415272,
      "grad_norm": 0.5514362710657545,
      "learning_rate": 4.665438968724361e-05,
      "loss": 13.4139,
      "step": 1888
    },
    {
      "epoch": 0.8663150653519834,
      "grad_norm": 0.5502732680838308,
      "learning_rate": 4.634147788768489e-05,
      "loss": 13.4729,
      "step": 1889
    },
    {
      "epoch": 0.8667736757624398,
      "grad_norm": 0.5438338320436362,
      "learning_rate": 4.6029567975199414e-05,
      "loss": 13.3651,
      "step": 1890
    },
    {
      "epoch": 0.8672322861728962,
      "grad_norm": 0.5553497634844335,
      "learning_rate": 4.571866063862795e-05,
      "loss": 13.6042,
      "step": 1891
    },
    {
      "epoch": 0.8676908965833524,
      "grad_norm": 0.5247852815902408,
      "learning_rate": 4.540875656459703e-05,
      "loss": 13.3835,
      "step": 1892
    },
    {
      "epoch": 0.8681495069938088,
      "grad_norm": 0.5521099426983994,
      "learning_rate": 4.509985643751785e-05,
      "loss": 13.5299,
      "step": 1893
    },
    {
      "epoch": 0.868608117404265,
      "grad_norm": 0.5594552591305388,
      "learning_rate": 4.479196093958421e-05,
      "loss": 13.5048,
      "step": 1894
    },
    {
      "epoch": 0.8690667278147214,
      "grad_norm": 0.5688087867587427,
      "learning_rate": 4.4485070750771187e-05,
      "loss": 13.5098,
      "step": 1895
    },
    {
      "epoch": 0.8695253382251777,
      "grad_norm": 0.5571989231511851,
      "learning_rate": 4.417918654883363e-05,
      "loss": 13.6267,
      "step": 1896
    },
    {
      "epoch": 0.869983948635634,
      "grad_norm": 0.5661286664546559,
      "learning_rate": 4.3874309009305e-05,
      "loss": 13.6415,
      "step": 1897
    },
    {
      "epoch": 0.8704425590460904,
      "grad_norm": 0.5341035884144844,
      "learning_rate": 4.357043880549538e-05,
      "loss": 13.3874,
      "step": 1898
    },
    {
      "epoch": 0.8709011694565466,
      "grad_norm": 0.5349279675120034,
      "learning_rate": 4.326757660849012e-05,
      "loss": 13.4402,
      "step": 1899
    },
    {
      "epoch": 0.871359779867003,
      "grad_norm": 0.524897831963954,
      "learning_rate": 4.2965723087148635e-05,
      "loss": 13.4901,
      "step": 1900
    },
    {
      "epoch": 0.8718183902774593,
      "grad_norm": 0.5890916355871391,
      "learning_rate": 4.266487890810256e-05,
      "loss": 13.2831,
      "step": 1901
    },
    {
      "epoch": 0.8722770006879156,
      "grad_norm": 0.5535273579805654,
      "learning_rate": 4.2365044735754365e-05,
      "loss": 13.3712,
      "step": 1902
    },
    {
      "epoch": 0.8727356110983719,
      "grad_norm": 0.5312526729986319,
      "learning_rate": 4.2066221232276266e-05,
      "loss": 13.6795,
      "step": 1903
    },
    {
      "epoch": 0.8731942215088283,
      "grad_norm": 0.5402792265944355,
      "learning_rate": 4.176840905760815e-05,
      "loss": 13.4039,
      "step": 1904
    },
    {
      "epoch": 0.8736528319192846,
      "grad_norm": 0.5599478796769317,
      "learning_rate": 4.1471608869456443e-05,
      "loss": 13.8301,
      "step": 1905
    },
    {
      "epoch": 0.8741114423297409,
      "grad_norm": 0.5357205928952036,
      "learning_rate": 4.117582132329284e-05,
      "loss": 13.4323,
      "step": 1906
    },
    {
      "epoch": 0.8745700527401972,
      "grad_norm": 0.5290958662455176,
      "learning_rate": 4.088104707235263e-05,
      "loss": 13.4646,
      "step": 1907
    },
    {
      "epoch": 0.8750286631506535,
      "grad_norm": 0.5774785701599092,
      "learning_rate": 4.058728676763313e-05,
      "loss": 13.3831,
      "step": 1908
    },
    {
      "epoch": 0.8754872735611099,
      "grad_norm": 0.5621546054924765,
      "learning_rate": 4.0294541057892375e-05,
      "loss": 13.4577,
      "step": 1909
    },
    {
      "epoch": 0.8759458839715661,
      "grad_norm": 0.5598262691207389,
      "learning_rate": 4.000281058964794e-05,
      "loss": 13.3789,
      "step": 1910
    },
    {
      "epoch": 0.8764044943820225,
      "grad_norm": 0.5470456455288183,
      "learning_rate": 3.971209600717507e-05,
      "loss": 13.3731,
      "step": 1911
    },
    {
      "epoch": 0.8768631047924788,
      "grad_norm": 0.5293420184246207,
      "learning_rate": 3.9422397952505465e-05,
      "loss": 13.4836,
      "step": 1912
    },
    {
      "epoch": 0.8773217152029351,
      "grad_norm": 0.5387992546538672,
      "learning_rate": 3.913371706542596e-05,
      "loss": 13.2896,
      "step": 1913
    },
    {
      "epoch": 0.8777803256133915,
      "grad_norm": 0.5673312858946148,
      "learning_rate": 3.884605398347707e-05,
      "loss": 13.2793,
      "step": 1914
    },
    {
      "epoch": 0.8782389360238477,
      "grad_norm": 0.555325480595792,
      "learning_rate": 3.8559409341951456e-05,
      "loss": 13.5612,
      "step": 1915
    },
    {
      "epoch": 0.8786975464343041,
      "grad_norm": 0.5519088365418223,
      "learning_rate": 3.8273783773892404e-05,
      "loss": 13.5074,
      "step": 1916
    },
    {
      "epoch": 0.8791561568447603,
      "grad_norm": 0.5541801209877283,
      "learning_rate": 3.798917791009293e-05,
      "loss": 13.3947,
      "step": 1917
    },
    {
      "epoch": 0.8796147672552167,
      "grad_norm": 0.5499619852652554,
      "learning_rate": 3.770559237909393e-05,
      "loss": 13.4532,
      "step": 1918
    },
    {
      "epoch": 0.8800733776656731,
      "grad_norm": 0.5417210349560008,
      "learning_rate": 3.742302780718288e-05,
      "loss": 13.3381,
      "step": 1919
    },
    {
      "epoch": 0.8805319880761293,
      "grad_norm": 0.5279159843047331,
      "learning_rate": 3.7141484818392635e-05,
      "loss": 13.2824,
      "step": 1920
    },
    {
      "epoch": 0.8809905984865857,
      "grad_norm": 0.5537393573585653,
      "learning_rate": 3.686096403449973e-05,
      "loss": 13.4809,
      "step": 1921
    },
    {
      "epoch": 0.8814492088970419,
      "grad_norm": 0.5234838810342162,
      "learning_rate": 3.658146607502344e-05,
      "loss": 13.4676,
      "step": 1922
    },
    {
      "epoch": 0.8819078193074983,
      "grad_norm": 0.5254168368735477,
      "learning_rate": 3.630299155722411e-05,
      "loss": 13.5108,
      "step": 1923
    },
    {
      "epoch": 0.8823664297179546,
      "grad_norm": 0.5467784302461038,
      "learning_rate": 3.6025541096101676e-05,
      "loss": 13.4632,
      "step": 1924
    },
    {
      "epoch": 0.8828250401284109,
      "grad_norm": 0.5471214003305611,
      "learning_rate": 3.574911530439473e-05,
      "loss": 13.4017,
      "step": 1925
    },
    {
      "epoch": 0.8832836505388673,
      "grad_norm": 0.5637726450053898,
      "learning_rate": 3.5473714792578606e-05,
      "loss": 13.6437,
      "step": 1926
    },
    {
      "epoch": 0.8837422609493235,
      "grad_norm": 0.5726066141422924,
      "learning_rate": 3.519934016886478e-05,
      "loss": 13.4172,
      "step": 1927
    },
    {
      "epoch": 0.8842008713597799,
      "grad_norm": 0.5945389135257796,
      "learning_rate": 3.4925992039198776e-05,
      "loss": 13.2252,
      "step": 1928
    },
    {
      "epoch": 0.8846594817702362,
      "grad_norm": 0.5224741027189378,
      "learning_rate": 3.465367100725908e-05,
      "loss": 13.4852,
      "step": 1929
    },
    {
      "epoch": 0.8851180921806925,
      "grad_norm": 0.5274015465605941,
      "learning_rate": 3.438237767445618e-05,
      "loss": 13.4466,
      "step": 1930
    },
    {
      "epoch": 0.8855767025911488,
      "grad_norm": 0.5627870603317782,
      "learning_rate": 3.4112112639930804e-05,
      "loss": 13.5164,
      "step": 1931
    },
    {
      "epoch": 0.8860353130016051,
      "grad_norm": 0.5492356887429409,
      "learning_rate": 3.3842876500552564e-05,
      "loss": 13.4102,
      "step": 1932
    },
    {
      "epoch": 0.8864939234120615,
      "grad_norm": 0.534541333674079,
      "learning_rate": 3.357466985091906e-05,
      "loss": 13.3916,
      "step": 1933
    },
    {
      "epoch": 0.8869525338225178,
      "grad_norm": 0.5545799585983883,
      "learning_rate": 3.330749328335414e-05,
      "loss": 13.4532,
      "step": 1934
    },
    {
      "epoch": 0.8874111442329741,
      "grad_norm": 0.5406489192454799,
      "learning_rate": 3.304134738790659e-05,
      "loss": 13.3823,
      "step": 1935
    },
    {
      "epoch": 0.8878697546434304,
      "grad_norm": 0.5381081934789953,
      "learning_rate": 3.277623275234953e-05,
      "loss": 13.4786,
      "step": 1936
    },
    {
      "epoch": 0.8883283650538867,
      "grad_norm": 0.5461677490900296,
      "learning_rate": 3.2512149962177994e-05,
      "loss": 13.5026,
      "step": 1937
    },
    {
      "epoch": 0.888786975464343,
      "grad_norm": 0.5746009008750664,
      "learning_rate": 3.224909960060851e-05,
      "loss": 13.5863,
      "step": 1938
    },
    {
      "epoch": 0.8892455858747994,
      "grad_norm": 0.5407886063252577,
      "learning_rate": 3.198708224857755e-05,
      "loss": 13.3725,
      "step": 1939
    },
    {
      "epoch": 0.8897041962852557,
      "grad_norm": 0.5559637044509137,
      "learning_rate": 3.172609848474023e-05,
      "loss": 13.6094,
      "step": 1940
    },
    {
      "epoch": 0.890162806695712,
      "grad_norm": 0.5616634741605981,
      "learning_rate": 3.1466148885468895e-05,
      "loss": 13.3531,
      "step": 1941
    },
    {
      "epoch": 0.8906214171061683,
      "grad_norm": 0.5735205457019257,
      "learning_rate": 3.120723402485198e-05,
      "loss": 13.2787,
      "step": 1942
    },
    {
      "epoch": 0.8910800275166246,
      "grad_norm": 0.6173232305040101,
      "learning_rate": 3.094935447469294e-05,
      "loss": 13.4261,
      "step": 1943
    },
    {
      "epoch": 0.891538637927081,
      "grad_norm": 0.543152097236227,
      "learning_rate": 3.069251080450863e-05,
      "loss": 13.5559,
      "step": 1944
    },
    {
      "epoch": 0.8919972483375372,
      "grad_norm": 0.5461858072340046,
      "learning_rate": 3.0436703581528113e-05,
      "loss": 13.5062,
      "step": 1945
    },
    {
      "epoch": 0.8924558587479936,
      "grad_norm": 0.5405764490701763,
      "learning_rate": 3.0181933370691694e-05,
      "loss": 13.4415,
      "step": 1946
    },
    {
      "epoch": 0.8929144691584499,
      "grad_norm": 0.5762640176252988,
      "learning_rate": 2.9928200734649523e-05,
      "loss": 13.2366,
      "step": 1947
    },
    {
      "epoch": 0.8933730795689062,
      "grad_norm": 0.5543078830662932,
      "learning_rate": 2.9675506233760142e-05,
      "loss": 13.521,
      "step": 1948
    },
    {
      "epoch": 0.8938316899793626,
      "grad_norm": 0.5810057681591982,
      "learning_rate": 2.942385042608925e-05,
      "loss": 13.4601,
      "step": 1949
    },
    {
      "epoch": 0.8942903003898188,
      "grad_norm": 0.5449640223993635,
      "learning_rate": 2.9173233867409054e-05,
      "loss": 13.3868,
      "step": 1950
    },
    {
      "epoch": 0.8947489108002752,
      "grad_norm": 0.5605857850984908,
      "learning_rate": 2.892365711119638e-05,
      "loss": 13.6032,
      "step": 1951
    },
    {
      "epoch": 0.8952075212107314,
      "grad_norm": 0.5532357505989361,
      "learning_rate": 2.8675120708631596e-05,
      "loss": 13.1761,
      "step": 1952
    },
    {
      "epoch": 0.8956661316211878,
      "grad_norm": 0.5419772393081201,
      "learning_rate": 2.8427625208597764e-05,
      "loss": 13.548,
      "step": 1953
    },
    {
      "epoch": 0.8961247420316442,
      "grad_norm": 0.5451349088538254,
      "learning_rate": 2.8181171157678874e-05,
      "loss": 13.4029,
      "step": 1954
    },
    {
      "epoch": 0.8965833524421004,
      "grad_norm": 0.5761830278303195,
      "learning_rate": 2.7935759100159053e-05,
      "loss": 13.5126,
      "step": 1955
    },
    {
      "epoch": 0.8970419628525568,
      "grad_norm": 0.5761949368888479,
      "learning_rate": 2.7691389578021365e-05,
      "loss": 13.4444,
      "step": 1956
    },
    {
      "epoch": 0.897500573263013,
      "grad_norm": 0.5327411089190593,
      "learning_rate": 2.7448063130946223e-05,
      "loss": 13.6095,
      "step": 1957
    },
    {
      "epoch": 0.8979591836734694,
      "grad_norm": 0.5442073668470376,
      "learning_rate": 2.7205780296310544e-05,
      "loss": 13.5045,
      "step": 1958
    },
    {
      "epoch": 0.8984177940839257,
      "grad_norm": 0.5439965786394341,
      "learning_rate": 2.6964541609186378e-05,
      "loss": 13.5988,
      "step": 1959
    },
    {
      "epoch": 0.898876404494382,
      "grad_norm": 0.5612836421119708,
      "learning_rate": 2.6724347602340104e-05,
      "loss": 13.2268,
      "step": 1960
    },
    {
      "epoch": 0.8993350149048384,
      "grad_norm": 0.5325904332908893,
      "learning_rate": 2.6485198806230682e-05,
      "loss": 13.6599,
      "step": 1961
    },
    {
      "epoch": 0.8997936253152946,
      "grad_norm": 0.5478318845390411,
      "learning_rate": 2.6247095749008797e-05,
      "loss": 13.3017,
      "step": 1962
    },
    {
      "epoch": 0.900252235725751,
      "grad_norm": 0.5498700361392806,
      "learning_rate": 2.6010038956515826e-05,
      "loss": 13.6411,
      "step": 1963
    },
    {
      "epoch": 0.9007108461362073,
      "grad_norm": 0.5648268527455583,
      "learning_rate": 2.5774028952282423e-05,
      "loss": 13.4091,
      "step": 1964
    },
    {
      "epoch": 0.9011694565466636,
      "grad_norm": 0.5268812014188228,
      "learning_rate": 2.5539066257527277e-05,
      "loss": 13.4104,
      "step": 1965
    },
    {
      "epoch": 0.90162806695712,
      "grad_norm": 0.5638962264242305,
      "learning_rate": 2.530515139115652e-05,
      "loss": 13.4145,
      "step": 1966
    },
    {
      "epoch": 0.9020866773675762,
      "grad_norm": 0.5589564711325797,
      "learning_rate": 2.5072284869761874e-05,
      "loss": 13.4274,
      "step": 1967
    },
    {
      "epoch": 0.9025452877780326,
      "grad_norm": 0.5494614079149541,
      "learning_rate": 2.4840467207619786e-05,
      "loss": 13.6298,
      "step": 1968
    },
    {
      "epoch": 0.9030038981884889,
      "grad_norm": 0.5459520346922234,
      "learning_rate": 2.460969891669068e-05,
      "loss": 13.4974,
      "step": 1969
    },
    {
      "epoch": 0.9034625085989452,
      "grad_norm": 0.5232099749836702,
      "learning_rate": 2.4379980506617272e-05,
      "loss": 13.557,
      "step": 1970
    },
    {
      "epoch": 0.9039211190094015,
      "grad_norm": 0.5727848750308846,
      "learning_rate": 2.4151312484723464e-05,
      "loss": 13.4871,
      "step": 1971
    },
    {
      "epoch": 0.9043797294198578,
      "grad_norm": 0.5236346423794064,
      "learning_rate": 2.3923695356013798e-05,
      "loss": 13.402,
      "step": 1972
    },
    {
      "epoch": 0.9048383398303141,
      "grad_norm": 0.5745566478437129,
      "learning_rate": 2.3697129623171833e-05,
      "loss": 13.4925,
      "step": 1973
    },
    {
      "epoch": 0.9052969502407705,
      "grad_norm": 0.5283288498874409,
      "learning_rate": 2.3471615786559042e-05,
      "loss": 13.5375,
      "step": 1974
    },
    {
      "epoch": 0.9057555606512268,
      "grad_norm": 0.5425316452124547,
      "learning_rate": 2.3247154344213818e-05,
      "loss": 13.5999,
      "step": 1975
    },
    {
      "epoch": 0.9062141710616831,
      "grad_norm": 0.5527546642171971,
      "learning_rate": 2.3023745791850625e-05,
      "loss": 13.6658,
      "step": 1976
    },
    {
      "epoch": 0.9066727814721394,
      "grad_norm": 0.5459767841560551,
      "learning_rate": 2.2801390622858354e-05,
      "loss": 13.532,
      "step": 1977
    },
    {
      "epoch": 0.9071313918825957,
      "grad_norm": 0.5395219570763186,
      "learning_rate": 2.2580089328299746e-05,
      "loss": 13.3745,
      "step": 1978
    },
    {
      "epoch": 0.9075900022930521,
      "grad_norm": 0.5347244134920799,
      "learning_rate": 2.235984239690997e-05,
      "loss": 13.3999,
      "step": 1979
    },
    {
      "epoch": 0.9080486127035083,
      "grad_norm": 0.546563624821302,
      "learning_rate": 2.2140650315095934e-05,
      "loss": 13.5623,
      "step": 1980
    },
    {
      "epoch": 0.9085072231139647,
      "grad_norm": 0.5328307703107512,
      "learning_rate": 2.192251356693459e-05,
      "loss": 13.5201,
      "step": 1981
    },
    {
      "epoch": 0.908965833524421,
      "grad_norm": 0.5466532067999261,
      "learning_rate": 2.170543263417246e-05,
      "loss": 13.5544,
      "step": 1982
    },
    {
      "epoch": 0.9094244439348773,
      "grad_norm": 0.5955512787415767,
      "learning_rate": 2.1489407996224286e-05,
      "loss": 13.2538,
      "step": 1983
    },
    {
      "epoch": 0.9098830543453337,
      "grad_norm": 0.5274086947539551,
      "learning_rate": 2.127444013017199e-05,
      "loss": 13.1889,
      "step": 1984
    },
    {
      "epoch": 0.9103416647557899,
      "grad_norm": 0.5802290957600929,
      "learning_rate": 2.1060529510763648e-05,
      "loss": 13.4528,
      "step": 1985
    },
    {
      "epoch": 0.9108002751662463,
      "grad_norm": 0.5672588686976763,
      "learning_rate": 2.084767661041259e-05,
      "loss": 13.4782,
      "step": 1986
    },
    {
      "epoch": 0.9112588855767025,
      "grad_norm": 0.5685829881211509,
      "learning_rate": 2.063588189919596e-05,
      "loss": 13.622,
      "step": 1987
    },
    {
      "epoch": 0.9117174959871589,
      "grad_norm": 0.5589162297500291,
      "learning_rate": 2.0425145844854275e-05,
      "loss": 13.6788,
      "step": 1988
    },
    {
      "epoch": 0.9121761063976153,
      "grad_norm": 0.567225949759816,
      "learning_rate": 2.0215468912789693e-05,
      "loss": 13.7451,
      "step": 1989
    },
    {
      "epoch": 0.9126347168080715,
      "grad_norm": 0.5595410948510176,
      "learning_rate": 2.0006851566065575e-05,
      "loss": 13.7109,
      "step": 1990
    },
    {
      "epoch": 0.9130933272185279,
      "grad_norm": 0.5521116577929553,
      "learning_rate": 1.9799294265405166e-05,
      "loss": 13.5131,
      "step": 1991
    },
    {
      "epoch": 0.9135519376289841,
      "grad_norm": 0.5417742505812609,
      "learning_rate": 1.9592797469190572e-05,
      "loss": 13.449,
      "step": 1992
    },
    {
      "epoch": 0.9140105480394405,
      "grad_norm": 0.5779265886061242,
      "learning_rate": 1.938736163346194e-05,
      "loss": 13.4907,
      "step": 1993
    },
    {
      "epoch": 0.9144691584498968,
      "grad_norm": 0.5677996186189833,
      "learning_rate": 1.9182987211916246e-05,
      "loss": 13.583,
      "step": 1994
    },
    {
      "epoch": 0.9149277688603531,
      "grad_norm": 0.5827354538985291,
      "learning_rate": 1.8979674655906332e-05,
      "loss": 13.3209,
      "step": 1995
    },
    {
      "epoch": 0.9153863792708095,
      "grad_norm": 0.5698103950036418,
      "learning_rate": 1.8777424414440024e-05,
      "loss": 13.5373,
      "step": 1996
    },
    {
      "epoch": 0.9158449896812658,
      "grad_norm": 0.5729564699205347,
      "learning_rate": 1.8576236934179202e-05,
      "loss": 13.4412,
      "step": 1997
    },
    {
      "epoch": 0.9163036000917221,
      "grad_norm": 0.5533257520524759,
      "learning_rate": 1.8376112659438393e-05,
      "loss": 13.4266,
      "step": 1998
    },
    {
      "epoch": 0.9167622105021784,
      "grad_norm": 0.5607480517193358,
      "learning_rate": 1.8177052032184282e-05,
      "loss": 13.5462,
      "step": 1999
    },
    {
      "epoch": 0.9172208209126347,
      "grad_norm": 0.5694770619678938,
      "learning_rate": 1.7979055492034435e-05,
      "loss": 13.4355,
      "step": 2000
    },
    {
      "epoch": 0.917679431323091,
      "grad_norm": 0.5675065889207742,
      "learning_rate": 1.7782123476256407e-05,
      "loss": 13.5208,
      "step": 2001
    },
    {
      "epoch": 0.9181380417335474,
      "grad_norm": 0.5283020821985154,
      "learning_rate": 1.7586256419766965e-05,
      "loss": 13.3985,
      "step": 2002
    },
    {
      "epoch": 0.9185966521440037,
      "grad_norm": 0.5403314998516543,
      "learning_rate": 1.7391454755130766e-05,
      "loss": 13.436,
      "step": 2003
    },
    {
      "epoch": 0.91905526255446,
      "grad_norm": 0.5456499067552346,
      "learning_rate": 1.7197718912559557e-05,
      "loss": 13.5548,
      "step": 2004
    },
    {
      "epoch": 0.9195138729649163,
      "grad_norm": 0.5386217715383396,
      "learning_rate": 1.700504931991148e-05,
      "loss": 13.3638,
      "step": 2005
    },
    {
      "epoch": 0.9199724833753726,
      "grad_norm": 0.5456781727732412,
      "learning_rate": 1.681344640268978e-05,
      "loss": 13.4812,
      "step": 2006
    },
    {
      "epoch": 0.920431093785829,
      "grad_norm": 0.5470531065028403,
      "learning_rate": 1.6622910584041974e-05,
      "loss": 13.3372,
      "step": 2007
    },
    {
      "epoch": 0.9208897041962852,
      "grad_norm": 0.5270821841011284,
      "learning_rate": 1.6433442284758903e-05,
      "loss": 13.3465,
      "step": 2008
    },
    {
      "epoch": 0.9213483146067416,
      "grad_norm": 0.5344020778957855,
      "learning_rate": 1.624504192327392e-05,
      "loss": 13.4435,
      "step": 2009
    },
    {
      "epoch": 0.9218069250171979,
      "grad_norm": 0.5661079739461179,
      "learning_rate": 1.6057709915661856e-05,
      "loss": 13.6406,
      "step": 2010
    },
    {
      "epoch": 0.9222655354276542,
      "grad_norm": 0.5609128561839235,
      "learning_rate": 1.5871446675638057e-05,
      "loss": 13.5609,
      "step": 2011
    },
    {
      "epoch": 0.9227241458381106,
      "grad_norm": 0.5634373166913894,
      "learning_rate": 1.5686252614557638e-05,
      "loss": 13.2935,
      "step": 2012
    },
    {
      "epoch": 0.9231827562485668,
      "grad_norm": 0.6261146113483335,
      "learning_rate": 1.5502128141414497e-05,
      "loss": 13.657,
      "step": 2013
    },
    {
      "epoch": 0.9236413666590232,
      "grad_norm": 0.5689209952168345,
      "learning_rate": 1.5319073662840188e-05,
      "loss": 13.6704,
      "step": 2014
    },
    {
      "epoch": 0.9240999770694794,
      "grad_norm": 0.592687870280418,
      "learning_rate": 1.5137089583103391e-05,
      "loss": 13.6837,
      "step": 2015
    },
    {
      "epoch": 0.9245585874799358,
      "grad_norm": 0.5599059559587164,
      "learning_rate": 1.4956176304108893e-05,
      "loss": 13.5992,
      "step": 2016
    },
    {
      "epoch": 0.9250171978903922,
      "grad_norm": 0.563966092607841,
      "learning_rate": 1.4776334225396481e-05,
      "loss": 13.2873,
      "step": 2017
    },
    {
      "epoch": 0.9254758083008484,
      "grad_norm": 0.5408375833125985,
      "learning_rate": 1.4597563744140397e-05,
      "loss": 13.7171,
      "step": 2018
    },
    {
      "epoch": 0.9259344187113048,
      "grad_norm": 0.5426600804407914,
      "learning_rate": 1.4419865255148269e-05,
      "loss": 13.3353,
      "step": 2019
    },
    {
      "epoch": 0.926393029121761,
      "grad_norm": 0.5562070358318214,
      "learning_rate": 1.4243239150860122e-05,
      "loss": 13.4374,
      "step": 2020
    },
    {
      "epoch": 0.9268516395322174,
      "grad_norm": 0.5397526369911251,
      "learning_rate": 1.4067685821347932e-05,
      "loss": 13.6708,
      "step": 2021
    },
    {
      "epoch": 0.9273102499426737,
      "grad_norm": 0.5572101966668784,
      "learning_rate": 1.389320565431429e-05,
      "loss": 13.4702,
      "step": 2022
    },
    {
      "epoch": 0.92776886035313,
      "grad_norm": 0.5253825568178809,
      "learning_rate": 1.3719799035091851e-05,
      "loss": 13.3627,
      "step": 2023
    },
    {
      "epoch": 0.9282274707635864,
      "grad_norm": 0.5409947253142823,
      "learning_rate": 1.3547466346642278e-05,
      "loss": 13.3803,
      "step": 2024
    },
    {
      "epoch": 0.9286860811740426,
      "grad_norm": 0.5361570874348569,
      "learning_rate": 1.3376207969555577e-05,
      "loss": 13.5784,
      "step": 2025
    },
    {
      "epoch": 0.929144691584499,
      "grad_norm": 0.5419613732780119,
      "learning_rate": 1.32060242820492e-05,
      "loss": 13.2716,
      "step": 2026
    },
    {
      "epoch": 0.9296033019949553,
      "grad_norm": 0.5494916019804705,
      "learning_rate": 1.3036915659967118e-05,
      "loss": 13.4359,
      "step": 2027
    },
    {
      "epoch": 0.9300619124054116,
      "grad_norm": 0.5630480508350736,
      "learning_rate": 1.2868882476779087e-05,
      "loss": 13.4818,
      "step": 2028
    },
    {
      "epoch": 0.930520522815868,
      "grad_norm": 0.5503408720868209,
      "learning_rate": 1.2701925103579815e-05,
      "loss": 13.4159,
      "step": 2029
    },
    {
      "epoch": 0.9309791332263242,
      "grad_norm": 0.5640155387568793,
      "learning_rate": 1.2536043909088191e-05,
      "loss": 13.8649,
      "step": 2030
    },
    {
      "epoch": 0.9314377436367806,
      "grad_norm": 0.5504308502860292,
      "learning_rate": 1.2371239259646228e-05,
      "loss": 13.5164,
      "step": 2031
    },
    {
      "epoch": 0.9318963540472369,
      "grad_norm": 0.5682943866548403,
      "learning_rate": 1.2207511519218672e-05,
      "loss": 13.7647,
      "step": 2032
    },
    {
      "epoch": 0.9323549644576932,
      "grad_norm": 0.5279580245120323,
      "learning_rate": 1.2044861049391676e-05,
      "loss": 13.2443,
      "step": 2033
    },
    {
      "epoch": 0.9328135748681495,
      "grad_norm": 0.5690341780782058,
      "learning_rate": 1.1883288209372512e-05,
      "loss": 13.4894,
      "step": 2034
    },
    {
      "epoch": 0.9332721852786058,
      "grad_norm": 0.5413327415251289,
      "learning_rate": 1.1722793355988471e-05,
      "loss": 13.4692,
      "step": 2035
    },
    {
      "epoch": 0.9337307956890621,
      "grad_norm": 0.5490903881062457,
      "learning_rate": 1.1563376843686135e-05,
      "loss": 13.7263,
      "step": 2036
    },
    {
      "epoch": 0.9341894060995185,
      "grad_norm": 0.5785993409388878,
      "learning_rate": 1.140503902453055e-05,
      "loss": 13.6379,
      "step": 2037
    },
    {
      "epoch": 0.9346480165099748,
      "grad_norm": 0.5666841193272576,
      "learning_rate": 1.1247780248204665e-05,
      "loss": 13.5003,
      "step": 2038
    },
    {
      "epoch": 0.9351066269204311,
      "grad_norm": 0.5392116885699068,
      "learning_rate": 1.1091600862008333e-05,
      "loss": 13.4056,
      "step": 2039
    },
    {
      "epoch": 0.9355652373308874,
      "grad_norm": 0.5557343812656218,
      "learning_rate": 1.0936501210857652e-05,
      "loss": 13.5147,
      "step": 2040
    },
    {
      "epoch": 0.9360238477413437,
      "grad_norm": 0.5523187208439919,
      "learning_rate": 1.0782481637284013e-05,
      "loss": 13.5468,
      "step": 2041
    },
    {
      "epoch": 0.9364824581518001,
      "grad_norm": 0.5564961342082605,
      "learning_rate": 1.0629542481433663e-05,
      "loss": 13.6047,
      "step": 2042
    },
    {
      "epoch": 0.9369410685622563,
      "grad_norm": 0.6089930269484368,
      "learning_rate": 1.0477684081066751e-05,
      "loss": 13.8041,
      "step": 2043
    },
    {
      "epoch": 0.9373996789727127,
      "grad_norm": 0.5337793774185032,
      "learning_rate": 1.0326906771556566e-05,
      "loss": 13.3901,
      "step": 2044
    },
    {
      "epoch": 0.937858289383169,
      "grad_norm": 0.5155607060604583,
      "learning_rate": 1.017721088588891e-05,
      "loss": 13.4529,
      "step": 2045
    },
    {
      "epoch": 0.9383168997936253,
      "grad_norm": 0.5299315817156387,
      "learning_rate": 1.0028596754661334e-05,
      "loss": 13.366,
      "step": 2046
    },
    {
      "epoch": 0.9387755102040817,
      "grad_norm": 0.5745982064640542,
      "learning_rate": 9.881064706082298e-06,
      "loss": 13.5669,
      "step": 2047
    },
    {
      "epoch": 0.9392341206145379,
      "grad_norm": 0.5483706844383441,
      "learning_rate": 9.734615065970454e-06,
      "loss": 13.5782,
      "step": 2048
    },
    {
      "epoch": 0.9396927310249943,
      "grad_norm": 0.5542190852784621,
      "learning_rate": 9.58924815775425e-06,
      "loss": 13.5763,
      "step": 2049
    },
    {
      "epoch": 0.9401513414354505,
      "grad_norm": 0.5786217839787257,
      "learning_rate": 9.444964302470715e-06,
      "loss": 13.2044,
      "step": 2050
    },
    {
      "epoch": 0.9406099518459069,
      "grad_norm": 0.5621030447025547,
      "learning_rate": 9.301763818765018e-06,
      "loss": 13.3683,
      "step": 2051
    },
    {
      "epoch": 0.9410685622563633,
      "grad_norm": 0.552598417354927,
      "learning_rate": 9.15964702288996e-06,
      "loss": 13.4351,
      "step": 2052
    },
    {
      "epoch": 0.9415271726668195,
      "grad_norm": 0.53045657281565,
      "learning_rate": 9.018614228704925e-06,
      "loss": 13.2765,
      "step": 2053
    },
    {
      "epoch": 0.9419857830772759,
      "grad_norm": 0.5332921618837491,
      "learning_rate": 8.878665747675152e-06,
      "loss": 13.5063,
      "step": 2054
    },
    {
      "epoch": 0.9424443934877321,
      "grad_norm": 0.5476587380333678,
      "learning_rate": 8.739801888871469e-06,
      "loss": 13.3063,
      "step": 2055
    },
    {
      "epoch": 0.9429030038981885,
      "grad_norm": 0.5376600297315113,
      "learning_rate": 8.602022958969336e-06,
      "loss": 13.4124,
      "step": 2056
    },
    {
      "epoch": 0.9433616143086448,
      "grad_norm": 0.5530996570987686,
      "learning_rate": 8.465329262248078e-06,
      "loss": 13.2459,
      "step": 2057
    },
    {
      "epoch": 0.9438202247191011,
      "grad_norm": 0.5528059726012191,
      "learning_rate": 8.32972110059027e-06,
      "loss": 13.415,
      "step": 2058
    },
    {
      "epoch": 0.9442788351295575,
      "grad_norm": 0.547481525369498,
      "learning_rate": 8.195198773481406e-06,
      "loss": 13.625,
      "step": 2059
    },
    {
      "epoch": 0.9447374455400137,
      "grad_norm": 0.5455268762291562,
      "learning_rate": 8.061762578008613e-06,
      "loss": 13.3985,
      "step": 2060
    },
    {
      "epoch": 0.9451960559504701,
      "grad_norm": 0.5687369150825248,
      "learning_rate": 7.929412808860559e-06,
      "loss": 13.6134,
      "step": 2061
    },
    {
      "epoch": 0.9456546663609264,
      "grad_norm": 0.5776166664522209,
      "learning_rate": 7.79814975832649e-06,
      "loss": 13.6526,
      "step": 2062
    },
    {
      "epoch": 0.9461132767713827,
      "grad_norm": 0.5437148704396156,
      "learning_rate": 7.667973716295851e-06,
      "loss": 13.5629,
      "step": 2063
    },
    {
      "epoch": 0.946571887181839,
      "grad_norm": 0.5558117963855099,
      "learning_rate": 7.5388849702571205e-06,
      "loss": 13.4762,
      "step": 2064
    },
    {
      "epoch": 0.9470304975922953,
      "grad_norm": 0.5634612435881566,
      "learning_rate": 7.4108838052979185e-06,
      "loss": 13.4165,
      "step": 2065
    },
    {
      "epoch": 0.9474891080027517,
      "grad_norm": 0.533864236098011,
      "learning_rate": 7.283970504103732e-06,
      "loss": 13.4552,
      "step": 2066
    },
    {
      "epoch": 0.947947718413208,
      "grad_norm": 0.547630624623949,
      "learning_rate": 7.1581453469575785e-06,
      "loss": 13.4704,
      "step": 2067
    },
    {
      "epoch": 0.9484063288236643,
      "grad_norm": 0.564803613798824,
      "learning_rate": 7.033408611739456e-06,
      "loss": 13.6325,
      "step": 2068
    },
    {
      "epoch": 0.9488649392341206,
      "grad_norm": 0.547945243854685,
      "learning_rate": 6.909760573925561e-06,
      "loss": 13.1609,
      "step": 2069
    },
    {
      "epoch": 0.9493235496445769,
      "grad_norm": 0.554914407508672,
      "learning_rate": 6.787201506587626e-06,
      "loss": 13.4336,
      "step": 2070
    },
    {
      "epoch": 0.9497821600550332,
      "grad_norm": 0.5252738017538887,
      "learning_rate": 6.66573168039264e-06,
      "loss": 13.6137,
      "step": 2071
    },
    {
      "epoch": 0.9502407704654896,
      "grad_norm": 0.5441692075989445,
      "learning_rate": 6.545351363601959e-06,
      "loss": 13.3102,
      "step": 2072
    },
    {
      "epoch": 0.9506993808759459,
      "grad_norm": 0.5281039782666413,
      "learning_rate": 6.426060822070812e-06,
      "loss": 13.5519,
      "step": 2073
    },
    {
      "epoch": 0.9511579912864022,
      "grad_norm": 0.552210084835681,
      "learning_rate": 6.3078603192475716e-06,
      "loss": 13.6333,
      "step": 2074
    },
    {
      "epoch": 0.9516166016968585,
      "grad_norm": 0.5395480529146559,
      "learning_rate": 6.1907501161735934e-06,
      "loss": 13.5387,
      "step": 2075
    },
    {
      "epoch": 0.9520752121073148,
      "grad_norm": 0.539984914193099,
      "learning_rate": 6.074730471482049e-06,
      "loss": 13.4252,
      "step": 2076
    },
    {
      "epoch": 0.9525338225177712,
      "grad_norm": 0.5647263616313631,
      "learning_rate": 5.959801641397755e-06,
      "loss": 13.3832,
      "step": 2077
    },
    {
      "epoch": 0.9529924329282274,
      "grad_norm": 0.5741486808704422,
      "learning_rate": 5.845963879736627e-06,
      "loss": 13.5418,
      "step": 2078
    },
    {
      "epoch": 0.9534510433386838,
      "grad_norm": 0.5486573226301247,
      "learning_rate": 5.733217437904892e-06,
      "loss": 13.2857,
      "step": 2079
    },
    {
      "epoch": 0.95390965374914,
      "grad_norm": 0.547440997424717,
      "learning_rate": 5.621562564898597e-06,
      "loss": 13.4311,
      "step": 2080
    },
    {
      "epoch": 0.9543682641595964,
      "grad_norm": 0.5486685484393291,
      "learning_rate": 5.51099950730316e-06,
      "loss": 13.3544,
      "step": 2081
    },
    {
      "epoch": 0.9548268745700528,
      "grad_norm": 0.5545219107901785,
      "learning_rate": 5.401528509292763e-06,
      "loss": 13.6734,
      "step": 2082
    },
    {
      "epoch": 0.955285484980509,
      "grad_norm": 0.5379474393718019,
      "learning_rate": 5.2931498126298495e-06,
      "loss": 13.4787,
      "step": 2083
    },
    {
      "epoch": 0.9557440953909654,
      "grad_norm": 0.5738296220200282,
      "learning_rate": 5.1858636566645135e-06,
      "loss": 13.4566,
      "step": 2084
    },
    {
      "epoch": 0.9562027058014216,
      "grad_norm": 0.5414247390451086,
      "learning_rate": 5.0796702783340035e-06,
      "loss": 13.4006,
      "step": 2085
    },
    {
      "epoch": 0.956661316211878,
      "grad_norm": 0.5761038123760133,
      "learning_rate": 4.97456991216233e-06,
      "loss": 13.7072,
      "step": 2086
    },
    {
      "epoch": 0.9571199266223344,
      "grad_norm": 0.5622152493131173,
      "learning_rate": 4.870562790259325e-06,
      "loss": 13.5582,
      "step": 2087
    },
    {
      "epoch": 0.9575785370327906,
      "grad_norm": 0.5806034058699016,
      "learning_rate": 4.7676491423208625e-06,
      "loss": 13.3992,
      "step": 2088
    },
    {
      "epoch": 0.958037147443247,
      "grad_norm": 0.568448432519177,
      "learning_rate": 4.66582919562758e-06,
      "loss": 13.521,
      "step": 2089
    },
    {
      "epoch": 0.9584957578537033,
      "grad_norm": 0.5518571498367986,
      "learning_rate": 4.5651031750448825e-06,
      "loss": 13.6692,
      "step": 2090
    },
    {
      "epoch": 0.9589543682641596,
      "grad_norm": 0.5631833340320282,
      "learning_rate": 4.465471303022217e-06,
      "loss": 13.6191,
      "step": 2091
    },
    {
      "epoch": 0.959412978674616,
      "grad_norm": 0.5378524079663521,
      "learning_rate": 4.366933799592743e-06,
      "loss": 13.5633,
      "step": 2092
    },
    {
      "epoch": 0.9598715890850722,
      "grad_norm": 0.523665595531304,
      "learning_rate": 4.269490882372551e-06,
      "loss": 13.5574,
      "step": 2093
    },
    {
      "epoch": 0.9603301994955286,
      "grad_norm": 0.5414284874170131,
      "learning_rate": 4.1731427665606115e-06,
      "loss": 13.5208,
      "step": 2094
    },
    {
      "epoch": 0.9607888099059849,
      "grad_norm": 0.5338114717272715,
      "learning_rate": 4.077889664937884e-06,
      "loss": 13.3793,
      "step": 2095
    },
    {
      "epoch": 0.9612474203164412,
      "grad_norm": 0.5611522015455934,
      "learning_rate": 3.983731787867207e-06,
      "loss": 13.3773,
      "step": 2096
    },
    {
      "epoch": 0.9617060307268975,
      "grad_norm": 0.5500827038962701,
      "learning_rate": 3.890669343292464e-06,
      "loss": 13.7894,
      "step": 2097
    },
    {
      "epoch": 0.9621646411373538,
      "grad_norm": 0.5528564705809732,
      "learning_rate": 3.7987025367384743e-06,
      "loss": 13.4555,
      "step": 2098
    },
    {
      "epoch": 0.9626232515478101,
      "grad_norm": 0.546615742214129,
      "learning_rate": 3.707831571310327e-06,
      "loss": 13.4583,
      "step": 2099
    },
    {
      "epoch": 0.9630818619582665,
      "grad_norm": 0.5459167414703254,
      "learning_rate": 3.6180566476929912e-06,
      "loss": 13.3009,
      "step": 2100
    },
    {
      "epoch": 0.9635404723687228,
      "grad_norm": 0.5423927450798798,
      "learning_rate": 3.529377964150815e-06,
      "loss": 13.4677,
      "step": 2101
    },
    {
      "epoch": 0.9639990827791791,
      "grad_norm": 0.5820830089321549,
      "learning_rate": 3.441795716527307e-06,
      "loss": 13.422,
      "step": 2102
    },
    {
      "epoch": 0.9644576931896354,
      "grad_norm": 0.5598512598756925,
      "learning_rate": 3.355310098244302e-06,
      "loss": 13.5952,
      "step": 2103
    },
    {
      "epoch": 0.9649163036000917,
      "grad_norm": 0.5524071750248192,
      "learning_rate": 3.269921300301959e-06,
      "loss": 13.5783,
      "step": 2104
    },
    {
      "epoch": 0.9653749140105481,
      "grad_norm": 0.5467371758424715,
      "learning_rate": 3.1856295112780988e-06,
      "loss": 13.5927,
      "step": 2105
    },
    {
      "epoch": 0.9658335244210043,
      "grad_norm": 0.537148871053374,
      "learning_rate": 3.102434917327812e-06,
      "loss": 13.6378,
      "step": 2106
    },
    {
      "epoch": 0.9662921348314607,
      "grad_norm": 0.5434553584615457,
      "learning_rate": 3.0203377021831292e-06,
      "loss": 13.3785,
      "step": 2107
    },
    {
      "epoch": 0.966750745241917,
      "grad_norm": 0.5423130848869916,
      "learning_rate": 2.939338047152573e-06,
      "loss": 13.5047,
      "step": 2108
    },
    {
      "epoch": 0.9672093556523733,
      "grad_norm": 0.5790243223939897,
      "learning_rate": 2.8594361311206073e-06,
      "loss": 13.2683,
      "step": 2109
    },
    {
      "epoch": 0.9676679660628297,
      "grad_norm": 0.53242397759233,
      "learning_rate": 2.7806321305475225e-06,
      "loss": 13.1926,
      "step": 2110
    },
    {
      "epoch": 0.9681265764732859,
      "grad_norm": 0.5292718153824285,
      "learning_rate": 2.7029262194688818e-06,
      "loss": 13.4184,
      "step": 2111
    },
    {
      "epoch": 0.9685851868837423,
      "grad_norm": 0.5586607413952258,
      "learning_rate": 2.626318569495134e-06,
      "loss": 13.3773,
      "step": 2112
    },
    {
      "epoch": 0.9690437972941985,
      "grad_norm": 0.5497904946356923,
      "learning_rate": 2.550809349811334e-06,
      "loss": 13.5917,
      "step": 2113
    },
    {
      "epoch": 0.9695024077046549,
      "grad_norm": 0.5470996799608197,
      "learning_rate": 2.476398727176532e-06,
      "loss": 13.7296,
      "step": 2114
    },
    {
      "epoch": 0.9699610181151113,
      "grad_norm": 0.5421229458211289,
      "learning_rate": 2.4030868659237204e-06,
      "loss": 13.6075,
      "step": 2115
    },
    {
      "epoch": 0.9704196285255675,
      "grad_norm": 0.5331511629377416,
      "learning_rate": 2.3308739279593317e-06,
      "loss": 13.2487,
      "step": 2116
    },
    {
      "epoch": 0.9708782389360239,
      "grad_norm": 0.5510903277096887,
      "learning_rate": 2.2597600727626845e-06,
      "loss": 13.3757,
      "step": 2117
    },
    {
      "epoch": 0.9713368493464801,
      "grad_norm": 0.5532911463223154,
      "learning_rate": 2.1897454573860387e-06,
      "loss": 13.5848,
      "step": 2118
    },
    {
      "epoch": 0.9717954597569365,
      "grad_norm": 0.5427227644432102,
      "learning_rate": 2.1208302364538746e-06,
      "loss": 13.5095,
      "step": 2119
    },
    {
      "epoch": 0.9722540701673928,
      "grad_norm": 0.5479034675057561,
      "learning_rate": 2.0530145621627804e-06,
      "loss": 13.3656,
      "step": 2120
    },
    {
      "epoch": 0.9727126805778491,
      "grad_norm": 0.5558914497005453,
      "learning_rate": 1.9862985842810653e-06,
      "loss": 13.5474,
      "step": 2121
    },
    {
      "epoch": 0.9731712909883055,
      "grad_norm": 0.5639676886647786,
      "learning_rate": 1.920682450148259e-06,
      "loss": 13.4432,
      "step": 2122
    },
    {
      "epoch": 0.9736299013987617,
      "grad_norm": 0.5731685664668064,
      "learning_rate": 1.856166304675111e-06,
      "loss": 13.4288,
      "step": 2123
    },
    {
      "epoch": 0.9740885118092181,
      "grad_norm": 0.5687051230223173,
      "learning_rate": 1.792750290342926e-06,
      "loss": 13.6002,
      "step": 2124
    },
    {
      "epoch": 0.9745471222196744,
      "grad_norm": 0.5489726580502259,
      "learning_rate": 1.7304345472035632e-06,
      "loss": 13.385,
      "step": 2125
    },
    {
      "epoch": 0.9750057326301307,
      "grad_norm": 0.5631785470340516,
      "learning_rate": 1.6692192128788253e-06,
      "loss": 13.6042,
      "step": 2126
    },
    {
      "epoch": 0.975464343040587,
      "grad_norm": 0.5502541413271315,
      "learning_rate": 1.6091044225604035e-06,
      "loss": 13.3559,
      "step": 2127
    },
    {
      "epoch": 0.9759229534510433,
      "grad_norm": 0.5430494864218129,
      "learning_rate": 1.5500903090094888e-06,
      "loss": 13.4612,
      "step": 2128
    },
    {
      "epoch": 0.9763815638614997,
      "grad_norm": 0.5537936041277711,
      "learning_rate": 1.492177002556383e-06,
      "loss": 13.4764,
      "step": 2129
    },
    {
      "epoch": 0.976840174271956,
      "grad_norm": 0.5521671296050199,
      "learning_rate": 1.4353646311004443e-06,
      "loss": 13.3298,
      "step": 2130
    },
    {
      "epoch": 0.9772987846824123,
      "grad_norm": 0.5553087022534245,
      "learning_rate": 1.3796533201094752e-06,
      "loss": 13.7095,
      "step": 2131
    },
    {
      "epoch": 0.9777573950928686,
      "grad_norm": 0.5485659194402972,
      "learning_rate": 1.3250431926197793e-06,
      "loss": 13.7034,
      "step": 2132
    },
    {
      "epoch": 0.9782160055033249,
      "grad_norm": 0.5444067334703886,
      "learning_rate": 1.2715343692356607e-06,
      "loss": 13.5164,
      "step": 2133
    },
    {
      "epoch": 0.9786746159137812,
      "grad_norm": 0.5499677291409165,
      "learning_rate": 1.2191269681292582e-06,
      "loss": 13.5221,
      "step": 2134
    },
    {
      "epoch": 0.9791332263242376,
      "grad_norm": 0.5625906910901343,
      "learning_rate": 1.1678211050402676e-06,
      "loss": 13.5477,
      "step": 2135
    },
    {
      "epoch": 0.9795918367346939,
      "grad_norm": 0.5267647799076003,
      "learning_rate": 1.117616893275719e-06,
      "loss": 13.4717,
      "step": 2136
    },
    {
      "epoch": 0.9800504471451502,
      "grad_norm": 0.5403473980116533,
      "learning_rate": 1.068514443709534e-06,
      "loss": 13.7245,
      "step": 2137
    },
    {
      "epoch": 0.9805090575556065,
      "grad_norm": 0.5649131281096698,
      "learning_rate": 1.0205138647826905e-06,
      "loss": 13.8612,
      "step": 2138
    },
    {
      "epoch": 0.9809676679660628,
      "grad_norm": 0.5745069097424316,
      "learning_rate": 9.73615262502503e-07,
      "loss": 13.4621,
      "step": 2139
    },
    {
      "epoch": 0.9814262783765192,
      "grad_norm": 0.5559786901404475,
      "learning_rate": 9.278187404426763e-07,
      "loss": 13.4564,
      "step": 2140
    },
    {
      "epoch": 0.9818848887869754,
      "grad_norm": 0.5575244038462969,
      "learning_rate": 8.831243997431404e-07,
      "loss": 13.452,
      "step": 2141
    },
    {
      "epoch": 0.9823434991974318,
      "grad_norm": 0.5323228930430751,
      "learning_rate": 8.395323391094944e-07,
      "loss": 13.1873,
      "step": 2142
    },
    {
      "epoch": 0.982802109607888,
      "grad_norm": 0.5506159474326113,
      "learning_rate": 7.970426548131183e-07,
      "loss": 13.2945,
      "step": 2143
    },
    {
      "epoch": 0.9832607200183444,
      "grad_norm": 0.5563401065616824,
      "learning_rate": 7.556554406908389e-07,
      "loss": 13.8218,
      "step": 2144
    },
    {
      "epoch": 0.9837193304288008,
      "grad_norm": 0.5664804681296799,
      "learning_rate": 7.153707881446536e-07,
      "loss": 13.425,
      "step": 2145
    },
    {
      "epoch": 0.984177940839257,
      "grad_norm": 0.5538544943906262,
      "learning_rate": 6.761887861417293e-07,
      "loss": 13.5052,
      "step": 2146
    },
    {
      "epoch": 0.9846365512497134,
      "grad_norm": 0.5700428920260526,
      "learning_rate": 6.381095212139032e-07,
      "loss": 13.5931,
      "step": 2147
    },
    {
      "epoch": 0.9850951616601696,
      "grad_norm": 0.580976862063177,
      "learning_rate": 6.011330774577384e-07,
      "loss": 13.5194,
      "step": 2148
    },
    {
      "epoch": 0.985553772070626,
      "grad_norm": 0.5603133605747243,
      "learning_rate": 5.652595365343016e-07,
      "loss": 13.5045,
      "step": 2149
    },
    {
      "epoch": 0.9860123824810824,
      "grad_norm": 0.570178457896819,
      "learning_rate": 5.304889776688859e-07,
      "loss": 13.5262,
      "step": 2150
    },
    {
      "epoch": 0.9864709928915386,
      "grad_norm": 0.537354257419322,
      "learning_rate": 4.968214776508994e-07,
      "loss": 13.4965,
      "step": 2151
    },
    {
      "epoch": 0.986929603301995,
      "grad_norm": 0.5435902287911796,
      "learning_rate": 4.6425711083375454e-07,
      "loss": 13.4589,
      "step": 2152
    },
    {
      "epoch": 0.9873882137124512,
      "grad_norm": 0.5343003995275457,
      "learning_rate": 4.3279594913447906e-07,
      "loss": 13.5414,
      "step": 2153
    },
    {
      "epoch": 0.9878468241229076,
      "grad_norm": 0.5626170056148846,
      "learning_rate": 4.02438062033883e-07,
      "loss": 13.3454,
      "step": 2154
    },
    {
      "epoch": 0.988305434533364,
      "grad_norm": 0.5440059911383701,
      "learning_rate": 3.7318351657616987e-07,
      "loss": 13.4986,
      "step": 2155
    },
    {
      "epoch": 0.9887640449438202,
      "grad_norm": 0.5569204421867654,
      "learning_rate": 3.4503237736882573e-07,
      "loss": 13.5802,
      "step": 2156
    },
    {
      "epoch": 0.9892226553542766,
      "grad_norm": 0.5394164388503304,
      "learning_rate": 3.179847065825081e-07,
      "loss": 13.4551,
      "step": 2157
    },
    {
      "epoch": 0.9896812657647328,
      "grad_norm": 0.5557118040837927,
      "learning_rate": 2.9204056395104594e-07,
      "loss": 13.1724,
      "step": 2158
    },
    {
      "epoch": 0.9901398761751892,
      "grad_norm": 0.5384706862844709,
      "learning_rate": 2.672000067709956e-07,
      "loss": 13.6073,
      "step": 2159
    },
    {
      "epoch": 0.9905984865856455,
      "grad_norm": 0.542932213910773,
      "learning_rate": 2.4346308990175204e-07,
      "loss": 13.2565,
      "step": 2160
    },
    {
      "epoch": 0.9910570969961018,
      "grad_norm": 0.539788321166138,
      "learning_rate": 2.208298657653818e-07,
      "loss": 13.4337,
      "step": 2161
    },
    {
      "epoch": 0.9915157074065581,
      "grad_norm": 0.5670760670698608,
      "learning_rate": 1.9930038434645692e-07,
      "loss": 13.4149,
      "step": 2162
    },
    {
      "epoch": 0.9919743178170144,
      "grad_norm": 0.5477180620041868,
      "learning_rate": 1.7887469319205484e-07,
      "loss": 13.5269,
      "step": 2163
    },
    {
      "epoch": 0.9924329282274708,
      "grad_norm": 0.5494744520649903,
      "learning_rate": 1.5955283741142523e-07,
      "loss": 13.5188,
      "step": 2164
    },
    {
      "epoch": 0.9928915386379271,
      "grad_norm": 0.55824995027188,
      "learning_rate": 1.4133485967615655e-07,
      "loss": 13.8001,
      "step": 2165
    },
    {
      "epoch": 0.9933501490483834,
      "grad_norm": 0.5426209836041237,
      "learning_rate": 1.2422080021995407e-07,
      "loss": 13.5917,
      "step": 2166
    },
    {
      "epoch": 0.9938087594588397,
      "grad_norm": 0.5455042940091859,
      "learning_rate": 1.082106968385288e-07,
      "loss": 13.4558,
      "step": 2167
    },
    {
      "epoch": 0.994267369869296,
      "grad_norm": 0.5329547767018725,
      "learning_rate": 9.330458488959748e-08,
      "loss": 13.4694,
      "step": 2168
    },
    {
      "epoch": 0.9947259802797523,
      "grad_norm": 0.5575055312889413,
      "learning_rate": 7.950249729271608e-08,
      "loss": 13.6983,
      "step": 2169
    },
    {
      "epoch": 0.9951845906902087,
      "grad_norm": 0.5371484779082286,
      "learning_rate": 6.680446452922429e-08,
      "loss": 13.4869,
      "step": 2170
    },
    {
      "epoch": 0.995643201100665,
      "grad_norm": 0.6084782447027622,
      "learning_rate": 5.521051464230098e-08,
      "loss": 13.4335,
      "step": 2171
    },
    {
      "epoch": 0.9961018115111213,
      "grad_norm": 0.5451095680382914,
      "learning_rate": 4.4720673236631206e-08,
      "loss": 13.561,
      "step": 2172
    },
    {
      "epoch": 0.9965604219215776,
      "grad_norm": 0.5304215695877842,
      "learning_rate": 3.53349634786837e-08,
      "loss": 13.4067,
      "step": 2173
    },
    {
      "epoch": 0.9970190323320339,
      "grad_norm": 0.5408815039495632,
      "learning_rate": 2.7053406096433365e-08,
      "loss": 13.3802,
      "step": 2174
    },
    {
      "epoch": 0.9974776427424903,
      "grad_norm": 0.5427678151506948,
      "learning_rate": 1.987601937930572e-08,
      "loss": 13.608,
      "step": 2175
    },
    {
      "epoch": 0.9979362531529465,
      "grad_norm": 0.5502683715264872,
      "learning_rate": 1.3802819178398984e-08,
      "loss": 13.3905,
      "step": 2176
    },
    {
      "epoch": 0.9983948635634029,
      "grad_norm": 0.5399904886737187,
      "learning_rate": 8.833818906039959e-09,
      "loss": 13.4314,
      "step": 2177
    },
    {
      "epoch": 0.9988534739738592,
      "grad_norm": 0.5666167694235265,
      "learning_rate": 4.969029536061598e-09,
      "loss": 13.4504,
      "step": 2178
    },
    {
      "epoch": 0.9993120843843155,
      "grad_norm": 0.5690374307508625,
      "learning_rate": 2.2084596038030037e-09,
      "loss": 13.5174,
      "step": 2179
    },
    {
      "epoch": 0.9997706947947719,
      "grad_norm": 0.5613442323662631,
      "learning_rate": 5.521152057763601e-10,
      "loss": 13.4249,
      "step": 2180
    },
    {
      "epoch": 0.9997706947947719,
      "step": 2180,
      "total_flos": 2.7748177392424714e+18,
      "train_loss": 13.717213252706266,
      "train_runtime": 24554.3416,
      "train_samples_per_second": 22.73,
      "train_steps_per_second": 0.089
    }
  ],
  "logging_steps": 1.0,
  "max_steps": 2180,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 24000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": false,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 2.7748177392424714e+18,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}