{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.6970573092003644,
  "eval_steps": 500,
  "global_step": 2200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0031684423145471106,
      "grad_norm": 0.19819921255111694,
      "learning_rate": 1.08e-05,
      "loss": 4.8784,
      "step": 10
    },
    {
      "epoch": 0.006336884629094221,
      "grad_norm": 0.34108418226242065,
      "learning_rate": 2.2800000000000002e-05,
      "loss": 5.0786,
      "step": 20
    },
    {
      "epoch": 0.009505326943641333,
      "grad_norm": 0.2952434718608856,
      "learning_rate": 3.48e-05,
      "loss": 5.0088,
      "step": 30
    },
    {
      "epoch": 0.012673769258188443,
      "grad_norm": 0.41625794768333435,
      "learning_rate": 4.6800000000000006e-05,
      "loss": 4.8262,
      "step": 40
    },
    {
      "epoch": 0.015842211572735555,
      "grad_norm": 1.1918175220489502,
      "learning_rate": 5.88e-05,
      "loss": 5.459,
      "step": 50
    },
    {
      "epoch": 0.019010653887282666,
      "grad_norm": 0.47633519768714905,
      "learning_rate": 5.999969438702549e-05,
      "loss": 5.6355,
      "step": 60
    },
    {
      "epoch": 0.022179096201829774,
      "grad_norm": 0.4251032769680023,
      "learning_rate": 5.999863795510765e-05,
      "loss": 5.5395,
      "step": 70
    },
    {
      "epoch": 0.025347538516376885,
      "grad_norm": 0.3730669617652893,
      "learning_rate": 5.999682695781323e-05,
      "loss": 5.539,
      "step": 80
    },
    {
      "epoch": 0.028515980830923996,
      "grad_norm": 0.3352891504764557,
      "learning_rate": 5.9994261440694826e-05,
      "loss": 5.5067,
      "step": 90
    },
    {
      "epoch": 0.03168442314547111,
      "grad_norm": 0.5066033005714417,
      "learning_rate": 5.9990941468283724e-05,
      "loss": 5.6052,
      "step": 100
    },
    {
      "epoch": 0.03485286546001822,
      "grad_norm": 0.4155944585800171,
      "learning_rate": 5.998686712408824e-05,
      "loss": 5.2992,
      "step": 110
    },
    {
      "epoch": 0.03802130777456533,
      "grad_norm": 0.4713999927043915,
      "learning_rate": 5.998203851059168e-05,
      "loss": 5.4492,
      "step": 120
    },
    {
      "epoch": 0.04118975008911244,
      "grad_norm": 0.32726186513900757,
      "learning_rate": 5.99764557492497e-05,
      "loss": 5.3036,
      "step": 130
    },
    {
      "epoch": 0.04435819240365955,
      "grad_norm": 0.4011114537715912,
      "learning_rate": 5.99701189804873e-05,
      "loss": 5.3122,
      "step": 140
    },
    {
      "epoch": 0.04752663471820666,
      "grad_norm": 0.46287617087364197,
      "learning_rate": 5.9963028363695244e-05,
      "loss": 5.3803,
      "step": 150
    },
    {
      "epoch": 0.05069507703275377,
      "grad_norm": 0.4292086660861969,
      "learning_rate": 5.995518407722611e-05,
      "loss": 5.1361,
      "step": 160
    },
    {
      "epoch": 0.05386351934730088,
      "grad_norm": 0.46384045481681824,
      "learning_rate": 5.994658631838974e-05,
      "loss": 5.226,
      "step": 170
    },
    {
      "epoch": 0.05703196166184799,
      "grad_norm": 0.41396230459213257,
      "learning_rate": 5.993723530344834e-05,
      "loss": 5.1451,
      "step": 180
    },
    {
      "epoch": 0.0602004039763951,
      "grad_norm": 0.5476190447807312,
      "learning_rate": 5.9927131267610975e-05,
      "loss": 5.1318,
      "step": 190
    },
    {
      "epoch": 0.06336884629094222,
      "grad_norm": 0.6672335267066956,
      "learning_rate": 5.991627446502772e-05,
      "loss": 5.188,
      "step": 200
    },
    {
      "epoch": 0.06653728860548933,
      "grad_norm": 0.5447375774383545,
      "learning_rate": 5.990466516878321e-05,
      "loss": 4.9756,
      "step": 210
    },
    {
      "epoch": 0.06970573092003644,
      "grad_norm": 0.38904011249542236,
      "learning_rate": 5.9892303670889794e-05,
      "loss": 5.0689,
      "step": 220
    },
    {
      "epoch": 0.07287417323458355,
      "grad_norm": 0.4386742115020752,
      "learning_rate": 5.987919028228022e-05,
      "loss": 5.0323,
      "step": 230
    },
    {
      "epoch": 0.07604261554913067,
      "grad_norm": 0.5731292366981506,
      "learning_rate": 5.986532533279975e-05,
      "loss": 5.0404,
      "step": 240
    },
    {
      "epoch": 0.07921105786367777,
      "grad_norm": 0.936024010181427,
      "learning_rate": 5.9850709171197906e-05,
      "loss": 5.0858,
      "step": 250
    },
    {
      "epoch": 0.08237950017822487,
      "grad_norm": 0.5377296209335327,
      "learning_rate": 5.9835342165119716e-05,
      "loss": 4.8807,
      "step": 260
    },
    {
      "epoch": 0.08554794249277199,
      "grad_norm": 0.5450035333633423,
      "learning_rate": 5.98192247010964e-05,
      "loss": 4.9438,
      "step": 270
    },
    {
      "epoch": 0.0887163848073191,
      "grad_norm": 0.5310191512107849,
      "learning_rate": 5.980235718453572e-05,
      "loss": 4.9821,
      "step": 280
    },
    {
      "epoch": 0.09188482712186621,
      "grad_norm": 1.1713744401931763,
      "learning_rate": 5.978474003971173e-05,
      "loss": 4.9609,
      "step": 290
    },
    {
      "epoch": 0.09505326943641332,
      "grad_norm": 0.6962549686431885,
      "learning_rate": 5.9766373709754145e-05,
      "loss": 4.9816,
      "step": 300
    },
    {
      "epoch": 0.09822171175096044,
      "grad_norm": 0.8181668519973755,
      "learning_rate": 5.974725865663715e-05,
      "loss": 4.8108,
      "step": 310
    },
    {
      "epoch": 0.10139015406550754,
      "grad_norm": 0.7138918042182922,
      "learning_rate": 5.9727395361167815e-05,
      "loss": 4.8991,
      "step": 320
    },
    {
      "epoch": 0.10455859638005466,
      "grad_norm": 0.754770040512085,
      "learning_rate": 5.970678432297399e-05,
      "loss": 4.9381,
      "step": 330
    },
    {
      "epoch": 0.10772703869460176,
      "grad_norm": 0.5500665307044983,
      "learning_rate": 5.9685426060491767e-05,
      "loss": 4.8761,
      "step": 340
    },
    {
      "epoch": 0.11089548100914888,
      "grad_norm": 0.9366258382797241,
      "learning_rate": 5.9663321110952344e-05,
      "loss": 4.9534,
      "step": 350
    },
    {
      "epoch": 0.11406392332369598,
      "grad_norm": 0.6130990386009216,
      "learning_rate": 5.964047003036864e-05,
      "loss": 4.736,
      "step": 360
    },
    {
      "epoch": 0.1172323656382431,
      "grad_norm": 0.5893264412879944,
      "learning_rate": 5.961687339352122e-05,
      "loss": 4.8846,
      "step": 370
    },
    {
      "epoch": 0.1204008079527902,
      "grad_norm": 0.5872120261192322,
      "learning_rate": 5.959253179394388e-05,
      "loss": 4.8049,
      "step": 380
    },
    {
      "epoch": 0.12356925026733732,
      "grad_norm": 0.9785071015357971,
      "learning_rate": 5.956744584390867e-05,
      "loss": 4.8355,
      "step": 390
    },
    {
      "epoch": 0.12673769258188444,
      "grad_norm": 1.110487461090088,
      "learning_rate": 5.9541616174410565e-05,
      "loss": 4.9169,
      "step": 400
    },
    {
      "epoch": 0.12990613489643155,
      "grad_norm": 0.8645092248916626,
      "learning_rate": 5.9515043435151535e-05,
      "loss": 4.7215,
      "step": 410
    },
    {
      "epoch": 0.13307457721097865,
      "grad_norm": 0.9703325033187866,
      "learning_rate": 5.948772829452423e-05,
      "loss": 4.832,
      "step": 420
    },
    {
      "epoch": 0.13624301952552575,
      "grad_norm": 0.6712486743927002,
      "learning_rate": 5.945967143959516e-05,
      "loss": 4.7938,
      "step": 430
    },
    {
      "epoch": 0.1394114618400729,
      "grad_norm": 0.8329195380210876,
      "learning_rate": 5.94308735760874e-05,
      "loss": 4.8012,
      "step": 440
    },
    {
      "epoch": 0.14257990415462,
      "grad_norm": 1.1987477540969849,
      "learning_rate": 5.940133542836286e-05,
      "loss": 4.8663,
      "step": 450
    },
    {
      "epoch": 0.1457483464691671,
      "grad_norm": 0.7384081482887268,
      "learning_rate": 5.937105773940407e-05,
      "loss": 4.6772,
      "step": 460
    },
    {
      "epoch": 0.1489167887837142,
      "grad_norm": 0.7325354814529419,
      "learning_rate": 5.9340041270795454e-05,
      "loss": 4.7705,
      "step": 470
    },
    {
      "epoch": 0.15208523109826133,
      "grad_norm": 0.8904541730880737,
      "learning_rate": 5.930828680270421e-05,
      "loss": 4.7959,
      "step": 480
    },
    {
      "epoch": 0.15525367341280844,
      "grad_norm": 0.8368484377861023,
      "learning_rate": 5.927579513386069e-05,
      "loss": 4.7283,
      "step": 490
    },
    {
      "epoch": 0.15842211572735554,
      "grad_norm": 0.8346347212791443,
      "learning_rate": 5.9242567081538255e-05,
      "loss": 4.805,
      "step": 500
    },
    {
      "epoch": 0.16159055804190264,
      "grad_norm": 0.6190924644470215,
      "learning_rate": 5.920860348153279e-05,
      "loss": 4.6284,
      "step": 510
    },
    {
      "epoch": 0.16475900035644975,
      "grad_norm": 1.251244068145752,
      "learning_rate": 5.917390518814164e-05,
      "loss": 4.6961,
      "step": 520
    },
    {
      "epoch": 0.16792744267099688,
      "grad_norm": 0.8339656591415405,
      "learning_rate": 5.913847307414212e-05,
      "loss": 4.6738,
      "step": 530
    },
    {
      "epoch": 0.17109588498554398,
      "grad_norm": 0.6742847561836243,
      "learning_rate": 5.910230803076957e-05,
      "loss": 4.6947,
      "step": 540
    },
    {
      "epoch": 0.1742643273000911,
      "grad_norm": 0.9937411546707153,
      "learning_rate": 5.906541096769495e-05,
      "loss": 4.7729,
      "step": 550
    },
    {
      "epoch": 0.1774327696146382,
      "grad_norm": 0.7207300662994385,
      "learning_rate": 5.9027782813001937e-05,
      "loss": 4.5756,
      "step": 560
    },
    {
      "epoch": 0.18060121192918532,
      "grad_norm": 0.7032576203346252,
      "learning_rate": 5.8989424513163595e-05,
      "loss": 4.6814,
      "step": 570
    },
    {
      "epoch": 0.18376965424373243,
      "grad_norm": 0.6017909049987793,
      "learning_rate": 5.895033703301855e-05,
      "loss": 4.6421,
      "step": 580
    },
    {
      "epoch": 0.18693809655827953,
      "grad_norm": 1.0589932203292847,
      "learning_rate": 5.8910521355746765e-05,
      "loss": 4.6204,
      "step": 590
    },
    {
      "epoch": 0.19010653887282664,
      "grad_norm": 1.3771517276763916,
      "learning_rate": 5.886997848284474e-05,
      "loss": 4.7115,
      "step": 600
    },
    {
      "epoch": 0.19327498118737377,
      "grad_norm": 0.6717742681503296,
      "learning_rate": 5.882870943410038e-05,
      "loss": 4.5742,
      "step": 610
    },
    {
      "epoch": 0.19644342350192087,
      "grad_norm": 1.0189399719238281,
      "learning_rate": 5.8786715247567346e-05,
      "loss": 4.6473,
      "step": 620
    },
    {
      "epoch": 0.19961186581646798,
      "grad_norm": 0.757310688495636,
      "learning_rate": 5.87439969795389e-05,
      "loss": 4.643,
      "step": 630
    },
    {
      "epoch": 0.20278030813101508,
      "grad_norm": 0.7095428705215454,
      "learning_rate": 5.870055570452133e-05,
      "loss": 4.5934,
      "step": 640
    },
    {
      "epoch": 0.2059487504455622,
      "grad_norm": 0.9886505603790283,
      "learning_rate": 5.8656392515207005e-05,
      "loss": 4.6542,
      "step": 650
    },
    {
      "epoch": 0.20911719276010932,
      "grad_norm": 0.6527100801467896,
      "learning_rate": 5.8611508522446806e-05,
      "loss": 4.5105,
      "step": 660
    },
    {
      "epoch": 0.21228563507465642,
      "grad_norm": 0.5825519561767578,
      "learning_rate": 5.856590485522222e-05,
      "loss": 4.5786,
      "step": 670
    },
    {
      "epoch": 0.21545407738920352,
      "grad_norm": 0.9389135241508484,
      "learning_rate": 5.851958266061694e-05,
      "loss": 4.5739,
      "step": 680
    },
    {
      "epoch": 0.21862251970375066,
      "grad_norm": 0.7726446986198425,
      "learning_rate": 5.847254310378801e-05,
      "loss": 4.5618,
      "step": 690
    },
    {
      "epoch": 0.22179096201829776,
      "grad_norm": 1.239856481552124,
      "learning_rate": 5.842478736793649e-05,
      "loss": 4.6292,
      "step": 700
    },
    {
      "epoch": 0.22495940433284486,
      "grad_norm": 0.6877331733703613,
      "learning_rate": 5.837631665427778e-05,
      "loss": 4.5048,
      "step": 710
    },
    {
      "epoch": 0.22812784664739197,
      "grad_norm": 0.801625669002533,
      "learning_rate": 5.832713218201127e-05,
      "loss": 4.533,
      "step": 720
    },
    {
      "epoch": 0.2312962889619391,
      "grad_norm": 0.967643141746521,
      "learning_rate": 5.827723518828983e-05,
      "loss": 4.5331,
      "step": 730
    },
    {
      "epoch": 0.2344647312764862,
      "grad_norm": 0.8205392956733704,
      "learning_rate": 5.822662692818855e-05,
      "loss": 4.5119,
      "step": 740
    },
    {
      "epoch": 0.2376331735910333,
      "grad_norm": 0.9368473291397095,
      "learning_rate": 5.817530867467324e-05,
      "loss": 4.5771,
      "step": 750
    },
    {
      "epoch": 0.2408016159055804,
      "grad_norm": 0.7767211198806763,
      "learning_rate": 5.812328171856844e-05,
      "loss": 4.4603,
      "step": 760
    },
    {
      "epoch": 0.24397005822012752,
      "grad_norm": 0.7029015421867371,
      "learning_rate": 5.807054736852488e-05,
      "loss": 4.5231,
      "step": 770
    },
    {
      "epoch": 0.24713850053467465,
      "grad_norm": 0.6616556644439697,
      "learning_rate": 5.80171069509866e-05,
      "loss": 4.5496,
      "step": 780
    },
    {
      "epoch": 0.2503069428492217,
      "grad_norm": 0.6856383085250854,
      "learning_rate": 5.79629618101576e-05,
      "loss": 4.5211,
      "step": 790
    },
    {
      "epoch": 0.2534753851637689,
      "grad_norm": 1.8431510925292969,
      "learning_rate": 5.790811330796798e-05,
      "loss": 4.5532,
      "step": 800
    },
    {
      "epoch": 0.256643827478316,
      "grad_norm": 0.6598497629165649,
      "learning_rate": 5.7852562824039755e-05,
      "loss": 4.4197,
      "step": 810
    },
    {
      "epoch": 0.2598122697928631,
      "grad_norm": 0.7226588726043701,
      "learning_rate": 5.779631175565207e-05,
      "loss": 4.5192,
      "step": 820
    },
    {
      "epoch": 0.2629807121074102,
      "grad_norm": 0.7087457776069641,
      "learning_rate": 5.773936151770615e-05,
      "loss": 4.4458,
      "step": 830
    },
    {
      "epoch": 0.2661491544219573,
      "grad_norm": 0.6751745343208313,
      "learning_rate": 5.768171354268959e-05,
      "loss": 4.4599,
      "step": 840
    },
    {
      "epoch": 0.2693175967365044,
      "grad_norm": 1.1364294290542603,
      "learning_rate": 5.762336928064043e-05,
      "loss": 4.5292,
      "step": 850
    },
    {
      "epoch": 0.2724860390510515,
      "grad_norm": 0.6589078307151794,
      "learning_rate": 5.756433019911064e-05,
      "loss": 4.4059,
      "step": 860
    },
    {
      "epoch": 0.2756544813655986,
      "grad_norm": 0.560736358165741,
      "learning_rate": 5.750459778312922e-05,
      "loss": 4.4956,
      "step": 870
    },
    {
      "epoch": 0.2788229236801458,
      "grad_norm": 0.5596962571144104,
      "learning_rate": 5.744417353516479e-05,
      "loss": 4.4729,
      "step": 880
    },
    {
      "epoch": 0.2819913659946929,
      "grad_norm": 0.7742477655410767,
      "learning_rate": 5.738305897508787e-05,
      "loss": 4.4419,
      "step": 890
    },
    {
      "epoch": 0.28515980830924,
      "grad_norm": 1.0753636360168457,
      "learning_rate": 5.7321255640132634e-05,
      "loss": 4.4975,
      "step": 900
    },
    {
      "epoch": 0.2883282506237871,
      "grad_norm": 0.6066401600837708,
      "learning_rate": 5.7258765084858206e-05,
      "loss": 4.3749,
      "step": 910
    },
    {
      "epoch": 0.2914966929383342,
      "grad_norm": 0.7095748782157898,
      "learning_rate": 5.71955888811096e-05,
      "loss": 4.4388,
      "step": 920
    },
    {
      "epoch": 0.2946651352528813,
      "grad_norm": 0.6218913197517395,
      "learning_rate": 5.713172861797817e-05,
      "loss": 4.4335,
      "step": 930
    },
    {
      "epoch": 0.2978335775674284,
      "grad_norm": 0.733177125453949,
      "learning_rate": 5.706718590176163e-05,
      "loss": 4.3898,
      "step": 940
    },
    {
      "epoch": 0.3010020198819755,
      "grad_norm": 0.7186071872711182,
      "learning_rate": 5.7001962355923634e-05,
      "loss": 4.4902,
      "step": 950
    },
    {
      "epoch": 0.30417046219652266,
      "grad_norm": 0.6571614742279053,
      "learning_rate": 5.693605962105299e-05,
      "loss": 4.3314,
      "step": 960
    },
    {
      "epoch": 0.30733890451106977,
      "grad_norm": 0.5829080939292908,
      "learning_rate": 5.686947935482238e-05,
      "loss": 4.4134,
      "step": 970
    },
    {
      "epoch": 0.31050734682561687,
      "grad_norm": 0.7523115277290344,
      "learning_rate": 5.680222323194659e-05,
      "loss": 4.4089,
      "step": 980
    },
    {
      "epoch": 0.313675789140164,
      "grad_norm": 0.656211256980896,
      "learning_rate": 5.67342929441405e-05,
      "loss": 4.3543,
      "step": 990
    },
    {
      "epoch": 0.3168442314547111,
      "grad_norm": 0.7600722908973694,
      "learning_rate": 5.666569020007646e-05,
      "loss": 4.4595,
      "step": 1000
    },
    {
      "epoch": 0.3200126737692582,
      "grad_norm": 0.656880795955658,
      "learning_rate": 5.659641672534133e-05,
      "loss": 4.3119,
      "step": 1010
    },
    {
      "epoch": 0.3231811160838053,
      "grad_norm": 0.6571980118751526,
      "learning_rate": 5.652647426239307e-05,
      "loss": 4.3918,
      "step": 1020
    },
    {
      "epoch": 0.3263495583983524,
      "grad_norm": 0.6933376789093018,
      "learning_rate": 5.645586457051691e-05,
      "loss": 4.375,
      "step": 1030
    },
    {
      "epoch": 0.3295180007128995,
      "grad_norm": 0.6467803120613098,
      "learning_rate": 5.638458942578113e-05,
      "loss": 4.3653,
      "step": 1040
    },
    {
      "epoch": 0.33268644302744665,
      "grad_norm": 0.953302800655365,
      "learning_rate": 5.631265062099231e-05,
      "loss": 4.4577,
      "step": 1050
    },
    {
      "epoch": 0.33585488534199376,
      "grad_norm": 0.6458297371864319,
      "learning_rate": 5.624004996565034e-05,
      "loss": 4.3215,
      "step": 1060
    },
    {
      "epoch": 0.33902332765654086,
      "grad_norm": 0.7524490356445312,
      "learning_rate": 5.616678928590282e-05,
      "loss": 4.3805,
      "step": 1070
    },
    {
      "epoch": 0.34219176997108797,
      "grad_norm": 0.6345677971839905,
      "learning_rate": 5.609287042449915e-05,
      "loss": 4.3734,
      "step": 1080
    },
    {
      "epoch": 0.34536021228563507,
      "grad_norm": 0.7871711254119873,
      "learning_rate": 5.601829524074421e-05,
      "loss": 4.3407,
      "step": 1090
    },
    {
      "epoch": 0.3485286546001822,
      "grad_norm": 0.9771536588668823,
      "learning_rate": 5.594306561045154e-05,
      "loss": 4.4314,
      "step": 1100
    },
    {
      "epoch": 0.3516970969147293,
      "grad_norm": 1.0256335735321045,
      "learning_rate": 5.58671834258962e-05,
      "loss": 4.2951,
      "step": 1110
    },
    {
      "epoch": 0.3548655392292764,
      "grad_norm": 0.6928941011428833,
      "learning_rate": 5.579065059576715e-05,
      "loss": 4.3752,
      "step": 1120
    },
    {
      "epoch": 0.35803398154382354,
      "grad_norm": 0.6043074727058411,
      "learning_rate": 5.571346904511926e-05,
      "loss": 4.3098,
      "step": 1130
    },
    {
      "epoch": 0.36120242385837065,
      "grad_norm": 0.7384766936302185,
      "learning_rate": 5.563564071532485e-05,
      "loss": 4.3167,
      "step": 1140
    },
    {
      "epoch": 0.36437086617291775,
      "grad_norm": 0.8783435821533203,
      "learning_rate": 5.5557167564024915e-05,
      "loss": 4.3998,
      "step": 1150
    },
    {
      "epoch": 0.36753930848746486,
      "grad_norm": 0.6876351237297058,
      "learning_rate": 5.547805156507982e-05,
      "loss": 4.2727,
      "step": 1160
    },
    {
      "epoch": 0.37070775080201196,
      "grad_norm": 0.7212591767311096,
      "learning_rate": 5.539829470851971e-05,
      "loss": 4.3282,
      "step": 1170
    },
    {
      "epoch": 0.37387619311655906,
      "grad_norm": 0.5978643298149109,
      "learning_rate": 5.531789900049443e-05,
      "loss": 4.3209,
      "step": 1180
    },
    {
      "epoch": 0.37704463543110617,
      "grad_norm": 0.664043664932251,
      "learning_rate": 5.523686646322302e-05,
      "loss": 4.2844,
      "step": 1190
    },
    {
      "epoch": 0.38021307774565327,
      "grad_norm": 0.7288404703140259,
      "learning_rate": 5.5155199134942934e-05,
      "loss": 4.3441,
      "step": 1200
    },
    {
      "epoch": 0.38338152006020043,
      "grad_norm": 0.7507999539375305,
      "learning_rate": 5.5072899069858713e-05,
      "loss": 4.2622,
      "step": 1210
    },
    {
      "epoch": 0.38654996237474754,
      "grad_norm": 0.6607810854911804,
      "learning_rate": 5.4989968338090334e-05,
      "loss": 4.303,
      "step": 1220
    },
    {
      "epoch": 0.38971840468929464,
      "grad_norm": 0.5649368762969971,
      "learning_rate": 5.4906409025621125e-05,
      "loss": 4.2851,
      "step": 1230
    },
    {
      "epoch": 0.39288684700384174,
      "grad_norm": 0.629998505115509,
      "learning_rate": 5.4822223234245316e-05,
      "loss": 4.3002,
      "step": 1240
    },
    {
      "epoch": 0.39605528931838885,
      "grad_norm": 0.8486919403076172,
      "learning_rate": 5.473741308151516e-05,
      "loss": 4.3504,
      "step": 1250
    },
    {
      "epoch": 0.39922373163293595,
      "grad_norm": 0.5649373531341553,
      "learning_rate": 5.465198070068767e-05,
      "loss": 4.2283,
      "step": 1260
    },
    {
      "epoch": 0.40239217394748306,
      "grad_norm": 0.5860993266105652,
      "learning_rate": 5.456592824067098e-05,
      "loss": 4.2915,
      "step": 1270
    },
    {
      "epoch": 0.40556061626203016,
      "grad_norm": 0.6270817518234253,
      "learning_rate": 5.447925786597022e-05,
      "loss": 4.2767,
      "step": 1280
    },
    {
      "epoch": 0.40872905857657726,
      "grad_norm": 0.804476261138916,
      "learning_rate": 5.43919717566332e-05,
      "loss": 4.2385,
      "step": 1290
    },
    {
      "epoch": 0.4118975008911244,
      "grad_norm": 0.7757463455200195,
      "learning_rate": 5.4304072108195446e-05,
      "loss": 4.3116,
      "step": 1300
    },
    {
      "epoch": 0.41506594320567153,
      "grad_norm": 0.6625142097473145,
      "learning_rate": 5.421556113162507e-05,
      "loss": 4.2071,
      "step": 1310
    },
    {
      "epoch": 0.41823438552021863,
      "grad_norm": 0.8527606725692749,
      "learning_rate": 5.412644105326707e-05,
      "loss": 4.2767,
      "step": 1320
    },
    {
      "epoch": 0.42140282783476574,
      "grad_norm": 0.6349330544471741,
      "learning_rate": 5.403671411478743e-05,
      "loss": 4.2691,
      "step": 1330
    },
    {
      "epoch": 0.42457127014931284,
      "grad_norm": 0.645797610282898,
      "learning_rate": 5.3946382573116634e-05,
      "loss": 4.241,
      "step": 1340
    },
    {
      "epoch": 0.42773971246385994,
      "grad_norm": 0.8075301051139832,
      "learning_rate": 5.385544870039298e-05,
      "loss": 4.3141,
      "step": 1350
    },
    {
      "epoch": 0.43090815477840705,
      "grad_norm": 0.6528099179267883,
      "learning_rate": 5.376391478390538e-05,
      "loss": 4.2255,
      "step": 1360
    },
    {
      "epoch": 0.43407659709295415,
      "grad_norm": 0.5646510124206543,
      "learning_rate": 5.3671783126035816e-05,
      "loss": 4.275,
      "step": 1370
    },
    {
      "epoch": 0.4372450394075013,
      "grad_norm": 0.5868734121322632,
      "learning_rate": 5.35790560442015e-05,
      "loss": 4.2824,
      "step": 1380
    },
    {
      "epoch": 0.4404134817220484,
      "grad_norm": 0.5941547155380249,
      "learning_rate": 5.3485735870796495e-05,
      "loss": 4.2643,
      "step": 1390
    },
    {
      "epoch": 0.4435819240365955,
      "grad_norm": 0.7245291471481323,
      "learning_rate": 5.33918249531331e-05,
      "loss": 4.3082,
      "step": 1400
    },
    {
      "epoch": 0.4467503663511426,
      "grad_norm": 0.7147409319877625,
      "learning_rate": 5.329732565338279e-05,
      "loss": 4.1898,
      "step": 1410
    },
    {
      "epoch": 0.44991880866568973,
      "grad_norm": 0.786919116973877,
      "learning_rate": 5.32022403485168e-05,
      "loss": 4.2158,
      "step": 1420
    },
    {
      "epoch": 0.45308725098023683,
      "grad_norm": 0.536165714263916,
      "learning_rate": 5.3106571430246365e-05,
      "loss": 4.2215,
      "step": 1430
    },
    {
      "epoch": 0.45625569329478394,
      "grad_norm": 0.6742566227912903,
      "learning_rate": 5.3010321304962504e-05,
      "loss": 4.2197,
      "step": 1440
    },
    {
      "epoch": 0.45942413560933104,
      "grad_norm": 0.7584452629089355,
      "learning_rate": 5.291349239367554e-05,
      "loss": 4.2972,
      "step": 1450
    },
    {
      "epoch": 0.4625925779238782,
      "grad_norm": 0.6290245652198792,
      "learning_rate": 5.2816087131954176e-05,
      "loss": 4.1603,
      "step": 1460
    },
    {
      "epoch": 0.4657610202384253,
      "grad_norm": 0.568684458732605,
      "learning_rate": 5.2718107969864256e-05,
      "loss": 4.2252,
      "step": 1470
    },
    {
      "epoch": 0.4689294625529724,
      "grad_norm": 0.5879467725753784,
      "learning_rate": 5.2619557371907104e-05,
      "loss": 4.2296,
      "step": 1480
    },
    {
      "epoch": 0.4720979048675195,
      "grad_norm": 0.5704094767570496,
      "learning_rate": 5.252043781695758e-05,
      "loss": 4.196,
      "step": 1490
    },
    {
      "epoch": 0.4752663471820666,
      "grad_norm": 0.7437434196472168,
      "learning_rate": 5.2420751798201685e-05,
      "loss": 4.2732,
      "step": 1500
    },
    {
      "epoch": 0.4784347894966137,
      "grad_norm": 0.6576899886131287,
      "learning_rate": 5.2320501823073866e-05,
      "loss": 4.148,
      "step": 1510
    },
    {
      "epoch": 0.4816032318111608,
      "grad_norm": 0.5717647671699524,
      "learning_rate": 5.221969041319396e-05,
      "loss": 4.2104,
      "step": 1520
    },
    {
      "epoch": 0.48477167412570793,
      "grad_norm": 0.5533893704414368,
      "learning_rate": 5.211832010430372e-05,
      "loss": 4.2493,
      "step": 1530
    },
    {
      "epoch": 0.48794011644025503,
      "grad_norm": 0.7069125771522522,
      "learning_rate": 5.201639344620311e-05,
      "loss": 4.1852,
      "step": 1540
    },
    {
      "epoch": 0.4911085587548022,
      "grad_norm": 0.7158598899841309,
      "learning_rate": 5.19139130026861e-05,
      "loss": 4.2715,
      "step": 1550
    },
    {
      "epoch": 0.4942770010693493,
      "grad_norm": 0.7333923578262329,
      "learning_rate": 5.181088135147618e-05,
      "loss": 4.1606,
      "step": 1560
    },
    {
      "epoch": 0.4974454433838964,
      "grad_norm": 0.591325581073761,
      "learning_rate": 5.17073010841616e-05,
      "loss": 4.2218,
      "step": 1570
    },
    {
      "epoch": 0.5006138856984435,
      "grad_norm": 0.5837592482566833,
      "learning_rate": 5.160317480613008e-05,
      "loss": 4.2246,
      "step": 1580
    },
    {
      "epoch": 0.5037823280129906,
      "grad_norm": 0.5487062931060791,
      "learning_rate": 5.1498505136503346e-05,
      "loss": 4.1844,
      "step": 1590
    },
    {
      "epoch": 0.5069507703275378,
      "grad_norm": 0.710067868232727,
      "learning_rate": 5.13932947080712e-05,
      "loss": 4.269,
      "step": 1600
    },
    {
      "epoch": 0.5101192126420848,
      "grad_norm": 0.6167586445808411,
      "learning_rate": 5.128754616722538e-05,
      "loss": 4.1127,
      "step": 1610
    },
    {
      "epoch": 0.513287654956632,
      "grad_norm": 0.5606287121772766,
      "learning_rate": 5.1181262173892856e-05,
      "loss": 4.2095,
      "step": 1620
    },
    {
      "epoch": 0.516456097271179,
      "grad_norm": 0.5047624707221985,
      "learning_rate": 5.1074445401469083e-05,
      "loss": 4.2135,
      "step": 1630
    },
    {
      "epoch": 0.5196245395857262,
      "grad_norm": 0.5399725437164307,
      "learning_rate": 5.0967098536750647e-05,
      "loss": 4.1672,
      "step": 1640
    },
    {
      "epoch": 0.5227929819002732,
      "grad_norm": 0.6514358520507812,
      "learning_rate": 5.085922427986772e-05,
      "loss": 4.225,
      "step": 1650
    },
    {
      "epoch": 0.5259614242148204,
      "grad_norm": 0.564847469329834,
      "learning_rate": 5.075082534421612e-05,
      "loss": 4.127,
      "step": 1660
    },
    {
      "epoch": 0.5291298665293676,
      "grad_norm": 0.5189889669418335,
      "learning_rate": 5.064190445638911e-05,
      "loss": 4.1728,
      "step": 1670
    },
    {
      "epoch": 0.5322983088439146,
      "grad_norm": 0.5365839600563049,
      "learning_rate": 5.053246435610877e-05,
      "loss": 4.2158,
      "step": 1680
    },
    {
      "epoch": 0.5354667511584618,
      "grad_norm": 0.6148782968521118,
      "learning_rate": 5.042250779615707e-05,
      "loss": 4.1489,
      "step": 1690
    },
    {
      "epoch": 0.5386351934730088,
      "grad_norm": 0.7295161485671997,
      "learning_rate": 5.031203754230673e-05,
      "loss": 4.2382,
      "step": 1700
    },
    {
      "epoch": 0.541803635787556,
      "grad_norm": 0.5799641609191895,
      "learning_rate": 5.020105637325149e-05,
      "loss": 4.1181,
      "step": 1710
    },
    {
      "epoch": 0.544972078102103,
      "grad_norm": 0.5710809826850891,
      "learning_rate": 5.008956708053636e-05,
      "loss": 4.1645,
      "step": 1720
    },
    {
      "epoch": 0.5481405204166502,
      "grad_norm": 0.5518040657043457,
      "learning_rate": 4.997757246848732e-05,
      "loss": 4.1606,
      "step": 1730
    },
    {
      "epoch": 0.5513089627311972,
      "grad_norm": 0.5413836240768433,
      "learning_rate": 4.986507535414084e-05,
      "loss": 4.1536,
      "step": 1740
    },
    {
      "epoch": 0.5544774050457444,
      "grad_norm": 0.7452698349952698,
      "learning_rate": 4.9752078567172964e-05,
      "loss": 4.2324,
      "step": 1750
    },
    {
      "epoch": 0.5576458473602915,
      "grad_norm": 0.6706964373588562,
      "learning_rate": 4.9638584949828173e-05,
      "loss": 4.1125,
      "step": 1760
    },
    {
      "epoch": 0.5608142896748386,
      "grad_norm": 0.5712190866470337,
      "learning_rate": 4.9524597356847874e-05,
      "loss": 4.1893,
      "step": 1770
    },
    {
      "epoch": 0.5639827319893858,
      "grad_norm": 0.5321968197822571,
      "learning_rate": 4.941011865539862e-05,
      "loss": 4.1646,
      "step": 1780
    },
    {
      "epoch": 0.5671511743039328,
      "grad_norm": 0.6286525130271912,
      "learning_rate": 4.929515172499995e-05,
      "loss": 4.1511,
      "step": 1790
    },
    {
      "epoch": 0.57031961661848,
      "grad_norm": 0.6667159795761108,
      "learning_rate": 4.9179699457452e-05,
      "loss": 4.2139,
      "step": 1800
    },
    {
      "epoch": 0.573488058933027,
      "grad_norm": 0.5475627779960632,
      "learning_rate": 4.906376475676274e-05,
      "loss": 4.0965,
      "step": 1810
    },
    {
      "epoch": 0.5766565012475742,
      "grad_norm": 0.48938098549842834,
      "learning_rate": 4.8947350539074924e-05,
      "loss": 4.1634,
      "step": 1820
    },
    {
      "epoch": 0.5798249435621212,
      "grad_norm": 0.5483301877975464,
      "learning_rate": 4.8830459732592786e-05,
      "loss": 4.1613,
      "step": 1830
    },
    {
      "epoch": 0.5829933858766684,
      "grad_norm": 0.577273964881897,
      "learning_rate": 4.871309527750832e-05,
      "loss": 4.1687,
      "step": 1840
    },
    {
      "epoch": 0.5861618281912155,
      "grad_norm": 0.7053927183151245,
      "learning_rate": 4.8595260125927354e-05,
      "loss": 4.2101,
      "step": 1850
    },
    {
      "epoch": 0.5893302705057626,
      "grad_norm": 0.5983800292015076,
      "learning_rate": 4.847695724179532e-05,
      "loss": 4.0919,
      "step": 1860
    },
    {
      "epoch": 0.5924987128203097,
      "grad_norm": 0.5336448550224304,
      "learning_rate": 4.835818960082266e-05,
      "loss": 4.1571,
      "step": 1870
    },
    {
      "epoch": 0.5956671551348568,
      "grad_norm": 0.5257425904273987,
      "learning_rate": 4.8238960190409985e-05,
      "loss": 4.1811,
      "step": 1880
    },
    {
      "epoch": 0.598835597449404,
      "grad_norm": 0.5968250632286072,
      "learning_rate": 4.8119272009572985e-05,
      "loss": 4.1438,
      "step": 1890
    },
    {
      "epoch": 0.602004039763951,
      "grad_norm": 0.7344405651092529,
      "learning_rate": 4.7999128068866885e-05,
      "loss": 4.2127,
      "step": 1900
    },
    {
      "epoch": 0.6051724820784982,
      "grad_norm": 0.5544557571411133,
      "learning_rate": 4.787853139031083e-05,
      "loss": 4.0893,
      "step": 1910
    },
    {
      "epoch": 0.6083409243930453,
      "grad_norm": 0.5338890552520752,
      "learning_rate": 4.775748500731181e-05,
      "loss": 4.1678,
      "step": 1920
    },
    {
      "epoch": 0.6115093667075924,
      "grad_norm": 0.47402748465538025,
      "learning_rate": 4.7635991964588386e-05,
      "loss": 4.1759,
      "step": 1930
    },
    {
      "epoch": 0.6146778090221395,
      "grad_norm": 0.4969950020313263,
      "learning_rate": 4.75140553180941e-05,
      "loss": 4.1487,
      "step": 1940
    },
    {
      "epoch": 0.6178462513366866,
      "grad_norm": 0.6715970635414124,
      "learning_rate": 4.739167813494056e-05,
      "loss": 4.1992,
      "step": 1950
    },
    {
      "epoch": 0.6210146936512337,
      "grad_norm": 0.5524842739105225,
      "learning_rate": 4.726886349332038e-05,
      "loss": 4.1092,
      "step": 1960
    },
    {
      "epoch": 0.6241831359657808,
      "grad_norm": 0.4865887463092804,
      "learning_rate": 4.7145614482429685e-05,
      "loss": 4.1495,
      "step": 1970
    },
    {
      "epoch": 0.627351578280328,
      "grad_norm": 0.4711080491542816,
      "learning_rate": 4.702193420239044e-05,
      "loss": 4.1299,
      "step": 1980
    },
    {
      "epoch": 0.630520020594875,
      "grad_norm": 0.5955727696418762,
      "learning_rate": 4.689782576417247e-05,
      "loss": 4.137,
      "step": 1990
    },
    {
      "epoch": 0.6336884629094222,
      "grad_norm": 0.7601689100265503,
      "learning_rate": 4.677329228951516e-05,
      "loss": 4.1742,
      "step": 2000
    },
    {
      "epoch": 0.6368569052239693,
      "grad_norm": 0.5171808004379272,
      "learning_rate": 4.6648336910849016e-05,
      "loss": 4.0932,
      "step": 2010
    },
    {
      "epoch": 0.6400253475385164,
      "grad_norm": 0.5088027119636536,
      "learning_rate": 4.6522962771216815e-05,
      "loss": 4.132,
      "step": 2020
    },
    {
      "epoch": 0.6431937898530635,
      "grad_norm": 0.5218876600265503,
      "learning_rate": 4.639717302419457e-05,
      "loss": 4.1397,
      "step": 2030
    },
    {
      "epoch": 0.6463622321676106,
      "grad_norm": 0.5443329811096191,
      "learning_rate": 4.6270970833812175e-05,
      "loss": 4.117,
      "step": 2040
    },
    {
      "epoch": 0.6495306744821577,
      "grad_norm": 0.6737596988677979,
      "learning_rate": 4.614435937447388e-05,
      "loss": 4.1727,
      "step": 2050
    },
    {
      "epoch": 0.6526991167967048,
      "grad_norm": 0.5711850523948669,
      "learning_rate": 4.601734183087838e-05,
      "loss": 4.0852,
      "step": 2060
    },
    {
      "epoch": 0.6558675591112519,
      "grad_norm": 0.5270329117774963,
      "learning_rate": 4.588992139793876e-05,
      "loss": 4.1357,
      "step": 2070
    },
    {
      "epoch": 0.659036001425799,
      "grad_norm": 0.4742993116378784,
      "learning_rate": 4.57621012807021e-05,
      "loss": 4.1446,
      "step": 2080
    },
    {
      "epoch": 0.6622044437403461,
      "grad_norm": 0.5585448741912842,
      "learning_rate": 4.563388469426885e-05,
      "loss": 4.0862,
      "step": 2090
    },
    {
      "epoch": 0.6653728860548933,
      "grad_norm": 0.7161533832550049,
      "learning_rate": 4.550527486371202e-05,
      "loss": 4.1865,
      "step": 2100
    },
    {
      "epoch": 0.6685413283694404,
      "grad_norm": 0.5912854671478271,
      "learning_rate": 4.5376275023995955e-05,
      "loss": 4.0892,
      "step": 2110
    },
    {
      "epoch": 0.6717097706839875,
      "grad_norm": 0.5607186555862427,
      "learning_rate": 4.524688841989509e-05,
      "loss": 4.1535,
      "step": 2120
    },
    {
      "epoch": 0.6748782129985346,
      "grad_norm": 0.4660521149635315,
      "learning_rate": 4.5117118305912235e-05,
      "loss": 4.1479,
      "step": 2130
    },
    {
      "epoch": 0.6780466553130817,
      "grad_norm": 0.5167705416679382,
      "learning_rate": 4.498696794619674e-05,
      "loss": 4.1185,
      "step": 2140
    },
    {
      "epoch": 0.6812150976276288,
      "grad_norm": 0.8260408639907837,
      "learning_rate": 4.485644061446242e-05,
      "loss": 4.1922,
      "step": 2150
    },
    {
      "epoch": 0.6843835399421759,
      "grad_norm": 0.5153993964195251,
      "learning_rate": 4.472553959390517e-05,
      "loss": 4.0546,
      "step": 2160
    },
    {
      "epoch": 0.6875519822567231,
      "grad_norm": 0.5324990749359131,
      "learning_rate": 4.459426817712041e-05,
      "loss": 4.1277,
      "step": 2170
    },
    {
      "epoch": 0.6907204245712701,
      "grad_norm": 0.49382662773132324,
      "learning_rate": 4.446262966602026e-05,
      "loss": 4.1258,
      "step": 2180
    },
    {
      "epoch": 0.6938888668858173,
      "grad_norm": 0.5588397979736328,
      "learning_rate": 4.433062737175047e-05,
      "loss": 4.1085,
      "step": 2190
    },
    {
      "epoch": 0.6970573092003644,
      "grad_norm": 0.6423320174217224,
      "learning_rate": 4.419826461460713e-05,
      "loss": 4.1834,
      "step": 2200
    }
  ],
  "logging_steps": 10,
  "max_steps": 6314,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 200,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 8.241908871552041e+17,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}