{
  "best_metric": 4.525084018707275,
  "best_model_checkpoint": "/mmfs1/gscratch/stf/abhinavp/corpus-filtering/outputs/pp-mod-subj2/lstm/3/checkpoints/checkpoint-152640",
  "epoch": 1.0250006060157382,
  "eval_steps": 10,
  "global_step": 152640,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "learning_rate": 4.999998362119627e-05,
      "loss": 10.8194,
      "step": 1
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.999161405248948e-05,
      "loss": 7.5719,
      "step": 512
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.998322810497896e-05,
      "loss": 7.067,
      "step": 1024
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.997484215746844e-05,
      "loss": 6.9956,
      "step": 1536
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.996645620995792e-05,
      "loss": 6.9424,
      "step": 2048
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.99580702624474e-05,
      "loss": 6.8558,
      "step": 2560
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994968431493688e-05,
      "loss": 6.6935,
      "step": 3072
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994129836742636e-05,
      "loss": 6.5846,
      "step": 3584
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.993291241991584e-05,
      "loss": 6.4975,
      "step": 4096
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.992452647240532e-05,
      "loss": 6.4297,
      "step": 4608
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.99161405248948e-05,
      "loss": 6.3811,
      "step": 5120
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.990775457738428e-05,
      "loss": 6.3144,
      "step": 5632
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989936862987376e-05,
      "loss": 6.263,
      "step": 6144
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989098268236324e-05,
      "loss": 6.1904,
      "step": 6656
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.988261311365645e-05,
      "loss": 6.136,
      "step": 7168
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.987422716614593e-05,
      "loss": 6.067,
      "step": 7680
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.986584121863541e-05,
      "loss": 6.035,
      "step": 8192
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.985745527112489e-05,
      "loss": 5.9783,
      "step": 8704
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.984906932361437e-05,
      "loss": 5.9322,
      "step": 9216
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.984068337610385e-05,
      "loss": 5.8985,
      "step": 9728
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.983229742859333e-05,
      "loss": 5.8769,
      "step": 10240
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.982391148108281e-05,
      "loss": 5.8304,
      "step": 10752
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.981552553357229e-05,
      "loss": 5.8075,
      "step": 11264
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.980713958606178e-05,
      "loss": 5.7724,
      "step": 11776
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.979875363855125e-05,
      "loss": 5.7408,
      "step": 12288
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9790384069844466e-05,
      "loss": 5.7232,
      "step": 12800
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9781998122333946e-05,
      "loss": 5.6853,
      "step": 13312
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9773612174823426e-05,
      "loss": 5.673,
      "step": 13824
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9765242606116635e-05,
      "loss": 5.648,
      "step": 14336
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9756856658606115e-05,
      "loss": 5.6195,
      "step": 14848
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9748470711095595e-05,
      "loss": 5.6054,
      "step": 15360
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.974008476358507e-05,
      "loss": 5.5823,
      "step": 15872
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.973169881607455e-05,
      "loss": 5.5682,
      "step": 16384
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.972331286856403e-05,
      "loss": 5.5513,
      "step": 16896
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9714926921053515e-05,
      "loss": 5.5308,
      "step": 17408
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9706540973542995e-05,
      "loss": 5.5215,
      "step": 17920
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9698171404836204e-05,
      "loss": 5.5139,
      "step": 18432
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9689785457325684e-05,
      "loss": 5.4756,
      "step": 18944
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9681399509815164e-05,
      "loss": 5.4657,
      "step": 19456
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9673013562304644e-05,
      "loss": 5.4519,
      "step": 19968
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9664627614794124e-05,
      "loss": 5.4357,
      "step": 20480
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9656241667283604e-05,
      "loss": 5.4216,
      "step": 20992
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.964787209857681e-05,
      "loss": 5.4229,
      "step": 21504
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.963948615106629e-05,
      "loss": 5.4026,
      "step": 22016
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.963110020355577e-05,
      "loss": 5.387,
      "step": 22528
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.962271425604525e-05,
      "loss": 5.3803,
      "step": 23040
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.961434468733847e-05,
      "loss": 5.3695,
      "step": 23552
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.960595873982795e-05,
      "loss": 5.3622,
      "step": 24064
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.959758917112116e-05,
      "loss": 5.3381,
      "step": 24576
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958920322361064e-05,
      "loss": 5.327,
      "step": 25088
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958081727610012e-05,
      "loss": 5.3347,
      "step": 25600
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.95724313285896e-05,
      "loss": 5.3237,
      "step": 26112
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.956404538107908e-05,
      "loss": 5.3118,
      "step": 26624
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.955565943356856e-05,
      "loss": 5.2875,
      "step": 27136
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.954727348605804e-05,
      "loss": 5.2966,
      "step": 27648
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.953888753854752e-05,
      "loss": 5.275,
      "step": 28160
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9530501591037e-05,
      "loss": 5.2905,
      "step": 28672
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9522132022330207e-05,
      "loss": 5.2563,
      "step": 29184
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9513746074819686e-05,
      "loss": 5.2599,
      "step": 29696
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9505360127309166e-05,
      "loss": 5.2398,
      "step": 30208
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9496974179798646e-05,
      "loss": 5.2303,
      "step": 30720
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948858823228813e-05,
      "loss": 5.2318,
      "step": 31232
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948020228477761e-05,
      "loss": 5.2149,
      "step": 31744
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.947183271607082e-05,
      "loss": 5.21,
      "step": 32256
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.94634467685603e-05,
      "loss": 5.2032,
      "step": 32768
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.945506082104978e-05,
      "loss": 5.2125,
      "step": 33280
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9446674873539255e-05,
      "loss": 5.1849,
      "step": 33792
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9438288926028735e-05,
      "loss": 5.1752,
      "step": 34304
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9429902978518215e-05,
      "loss": 5.1602,
      "step": 34816
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.942153340981143e-05,
      "loss": 5.1583,
      "step": 35328
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9413147462300904e-05,
      "loss": 5.1737,
      "step": 35840
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9404761514790384e-05,
      "loss": 5.1619,
      "step": 36352
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.939637556727987e-05,
      "loss": 5.1444,
      "step": 36864
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.938798961976935e-05,
      "loss": 5.1404,
      "step": 37376
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.937960367225883e-05,
      "loss": 5.143,
      "step": 37888
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9371250482355776e-05,
      "loss": 5.1359,
      "step": 38400
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9362864534845256e-05,
      "loss": 5.1131,
      "step": 38912
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.935447858733473e-05,
      "loss": 5.118,
      "step": 39424
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.934609263982421e-05,
      "loss": 5.1037,
      "step": 39936
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.933770669231369e-05,
      "loss": 5.1055,
      "step": 40448
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.932932074480317e-05,
      "loss": 5.1022,
      "step": 40960
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.932093479729265e-05,
      "loss": 5.0905,
      "step": 41472
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.931254884978213e-05,
      "loss": 5.0925,
      "step": 41984
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.930416290227161e-05,
      "loss": 5.0832,
      "step": 42496
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9295793333564825e-05,
      "loss": 5.0675,
      "step": 43008
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9287407386054305e-05,
      "loss": 5.0665,
      "step": 43520
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9279021438543785e-05,
      "loss": 5.0716,
      "step": 44032
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9270635491033264e-05,
      "loss": 5.0642,
      "step": 44544
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9262249543522744e-05,
      "loss": 5.0513,
      "step": 45056
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9253863596012224e-05,
      "loss": 5.0386,
      "step": 45568
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9245494027305433e-05,
      "loss": 5.0431,
      "step": 46080
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9237108079794913e-05,
      "loss": 5.0302,
      "step": 46592
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.922873851108812e-05,
      "loss": 5.0416,
      "step": 47104
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.92203525635776e-05,
      "loss": 5.0192,
      "step": 47616
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.921196661606708e-05,
      "loss": 5.0148,
      "step": 48128
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.920358066855656e-05,
      "loss": 5.032,
      "step": 48640
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.919519472104604e-05,
      "loss": 5.0004,
      "step": 49152
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.918680877353552e-05,
      "loss": 4.9947,
      "step": 49664
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.917842282602501e-05,
      "loss": 4.9991,
      "step": 50176
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.917003687851449e-05,
      "loss": 4.9971,
      "step": 50688
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.916165093100397e-05,
      "loss": 4.9973,
      "step": 51200
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.915328136229718e-05,
      "loss": 4.9856,
      "step": 51712
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.914489541478666e-05,
      "loss": 4.9793,
      "step": 52224
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.913650946727614e-05,
      "loss": 4.9719,
      "step": 52736
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.912812351976562e-05,
      "loss": 4.9706,
      "step": 53248
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.911973757225509e-05,
      "loss": 4.9546,
      "step": 53760
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.911136800354831e-05,
      "loss": 4.9543,
      "step": 54272
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.910298205603779e-05,
      "loss": 4.9592,
      "step": 54784
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.909459610852727e-05,
      "loss": 4.9467,
      "step": 55296
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.908621016101675e-05,
      "loss": 4.9415,
      "step": 55808
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.907782421350623e-05,
      "loss": 4.9358,
      "step": 56320
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906943826599571e-05,
      "loss": 4.9273,
      "step": 56832
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906105231848519e-05,
      "loss": 4.9364,
      "step": 57344
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.905266637097467e-05,
      "loss": 4.922,
      "step": 57856
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9044296802267876e-05,
      "loss": 4.9191,
      "step": 58368
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.903592723356109e-05,
      "loss": 4.9196,
      "step": 58880
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9027541286050565e-05,
      "loss": 4.9136,
      "step": 59392
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9019155338540045e-05,
      "loss": 4.9097,
      "step": 59904
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9010769391029525e-05,
      "loss": 4.9072,
      "step": 60416
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9002383443519005e-05,
      "loss": 4.9064,
      "step": 60928
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.899403025361595e-05,
      "loss": 4.9048,
      "step": 61440
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.898564430610543e-05,
      "loss": 4.8852,
      "step": 61952
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.897725835859491e-05,
      "loss": 4.8965,
      "step": 62464
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8968872411084396e-05,
      "loss": 4.8864,
      "step": 62976
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.896048646357387e-05,
      "loss": 4.8876,
      "step": 63488
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.895210051606335e-05,
      "loss": 4.8696,
      "step": 64000
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.894371456855283e-05,
      "loss": 4.8786,
      "step": 64512
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.893532862104231e-05,
      "loss": 4.8627,
      "step": 65024
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.892694267353179e-05,
      "loss": 4.8685,
      "step": 65536
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.891855672602127e-05,
      "loss": 4.8463,
      "step": 66048
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.891017077851075e-05,
      "loss": 4.869,
      "step": 66560
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.890178483100023e-05,
      "loss": 4.874,
      "step": 67072
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.889343164109717e-05,
      "loss": 4.8545,
      "step": 67584
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.888504569358665e-05,
      "loss": 4.8404,
      "step": 68096
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8876659746076134e-05,
      "loss": 4.8522,
      "step": 68608
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8868273798565614e-05,
      "loss": 4.8459,
      "step": 69120
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8859887851055094e-05,
      "loss": 4.8439,
      "step": 69632
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8851501903544574e-05,
      "loss": 4.8368,
      "step": 70144
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8843115956034054e-05,
      "loss": 4.8399,
      "step": 70656
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8834730008523534e-05,
      "loss": 4.8305,
      "step": 71168
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.882637681862047e-05,
      "loss": 4.8263,
      "step": 71680
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.881799087110995e-05,
      "loss": 4.8124,
      "step": 72192
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.880960492359943e-05,
      "loss": 4.823,
      "step": 72704
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.880121897608891e-05,
      "loss": 4.8183,
      "step": 73216
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.879283302857839e-05,
      "loss": 4.8102,
      "step": 73728
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.878444708106787e-05,
      "loss": 4.8228,
      "step": 74240
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.877606113355735e-05,
      "loss": 4.8007,
      "step": 74752
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.876769156485057e-05,
      "loss": 4.8102,
      "step": 75264
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.875930561734005e-05,
      "loss": 4.8111,
      "step": 75776
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.875091966982953e-05,
      "loss": 4.7998,
      "step": 76288
    },
    {
      "epoch": 0.03,
      "eval_loss": 4.812014102935791,
      "eval_runtime": 329.0718,
      "eval_samples_per_second": 1159.598,
      "eval_steps_per_second": 36.238,
      "step": 76320
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.874253372231901e-05,
      "loss": 4.7881,
      "step": 76800
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.873414777480849e-05,
      "loss": 4.794,
      "step": 77312
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.872576182729797e-05,
      "loss": 4.7965,
      "step": 77824
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8717408637394906e-05,
      "loss": 4.7873,
      "step": 78336
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8709022689884386e-05,
      "loss": 4.788,
      "step": 78848
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8700636742373866e-05,
      "loss": 4.7708,
      "step": 79360
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8692250794863346e-05,
      "loss": 4.777,
      "step": 79872
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8683864847352826e-05,
      "loss": 4.7661,
      "step": 80384
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8675478899842306e-05,
      "loss": 4.7728,
      "step": 80896
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.866710933113552e-05,
      "loss": 4.7638,
      "step": 81408
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.8658723383625e-05,
      "loss": 4.7689,
      "step": 81920
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.865035381491821e-05,
      "loss": 4.7706,
      "step": 82432
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.864196786740769e-05,
      "loss": 4.7564,
      "step": 82944
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.863358191989717e-05,
      "loss": 4.7541,
      "step": 83456
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.862519597238665e-05,
      "loss": 4.7389,
      "step": 83968
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.861681002487613e-05,
      "loss": 4.7494,
      "step": 84480
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.860842407736561e-05,
      "loss": 4.7428,
      "step": 84992
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.860003812985509e-05,
      "loss": 4.7369,
      "step": 85504
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.859165218234457e-05,
      "loss": 4.7337,
      "step": 86016
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.858326623483405e-05,
      "loss": 4.7617,
      "step": 86528
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.857488028732352e-05,
      "loss": 4.7338,
      "step": 87040
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.856649433981301e-05,
      "loss": 4.7344,
      "step": 87552
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.855810839230249e-05,
      "loss": 4.7383,
      "step": 88064
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.854972244479197e-05,
      "loss": 4.7276,
      "step": 88576
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.854135287608518e-05,
      "loss": 4.7265,
      "step": 89088
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.853296692857466e-05,
      "loss": 4.7141,
      "step": 89600
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.852458098106414e-05,
      "loss": 4.7234,
      "step": 90112
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.851619503355362e-05,
      "loss": 4.718,
      "step": 90624
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.85078090860431e-05,
      "loss": 4.7051,
      "step": 91136
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.849942313853258e-05,
      "loss": 4.7132,
      "step": 91648
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.849103719102206e-05,
      "loss": 4.7046,
      "step": 92160
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.848266762231527e-05,
      "loss": 4.7136,
      "step": 92672
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.847428167480475e-05,
      "loss": 4.7108,
      "step": 93184
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.846589572729423e-05,
      "loss": 4.6974,
      "step": 93696
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.845750977978371e-05,
      "loss": 4.706,
      "step": 94208
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8449123832273194e-05,
      "loss": 4.7134,
      "step": 94720
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8440754263566404e-05,
      "loss": 4.6825,
      "step": 95232
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8432368316055884e-05,
      "loss": 4.6939,
      "step": 95744
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8423982368545363e-05,
      "loss": 4.6896,
      "step": 96256
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8415596421034843e-05,
      "loss": 4.6815,
      "step": 96768
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.840721047352432e-05,
      "loss": 4.6821,
      "step": 97280
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.83988245260138e-05,
      "loss": 4.6906,
      "step": 97792
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.839045495730701e-05,
      "loss": 4.6842,
      "step": 98304
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.838206900979649e-05,
      "loss": 4.678,
      "step": 98816
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.83736994410897e-05,
      "loss": 4.6824,
      "step": 99328
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.836531349357918e-05,
      "loss": 4.6748,
      "step": 99840
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.835692754606866e-05,
      "loss": 4.679,
      "step": 100352
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.834854159855814e-05,
      "loss": 4.6607,
      "step": 100864
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.834015565104763e-05,
      "loss": 4.6578,
      "step": 101376
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.833178608234084e-05,
      "loss": 4.674,
      "step": 101888
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.832340013483032e-05,
      "loss": 4.6727,
      "step": 102400
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.83150141873198e-05,
      "loss": 4.6614,
      "step": 102912
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.830662823980928e-05,
      "loss": 4.652,
      "step": 103424
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.829824229229876e-05,
      "loss": 4.6634,
      "step": 103936
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.828985634478824e-05,
      "loss": 4.6453,
      "step": 104448
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.828147039727771e-05,
      "loss": 4.6713,
      "step": 104960
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.827308444976719e-05,
      "loss": 4.6471,
      "step": 105472
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.826469850225667e-05,
      "loss": 4.6591,
      "step": 105984
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.825631255474615e-05,
      "loss": 4.645,
      "step": 106496
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.824792660723563e-05,
      "loss": 4.6375,
      "step": 107008
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.823954065972511e-05,
      "loss": 4.6416,
      "step": 107520
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8231171091018326e-05,
      "loss": 4.6462,
      "step": 108032
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8222785143507806e-05,
      "loss": 4.6295,
      "step": 108544
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8214399195997286e-05,
      "loss": 4.6322,
      "step": 109056
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8206013248486766e-05,
      "loss": 4.6472,
      "step": 109568
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8197627300976246e-05,
      "loss": 4.6271,
      "step": 110080
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8189241353465726e-05,
      "loss": 4.6174,
      "step": 110592
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8180871784758935e-05,
      "loss": 4.6184,
      "step": 111104
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8172502216052144e-05,
      "loss": 4.6168,
      "step": 111616
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8164116268541624e-05,
      "loss": 4.6382,
      "step": 112128
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8155730321031104e-05,
      "loss": 4.6323,
      "step": 112640
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8147344373520584e-05,
      "loss": 4.6167,
      "step": 113152
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.8138958426010064e-05,
      "loss": 4.6206,
      "step": 113664
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.813057247849955e-05,
      "loss": 4.6233,
      "step": 114176
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.812218653098903e-05,
      "loss": 4.622,
      "step": 114688
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.811380058347851e-05,
      "loss": 4.6052,
      "step": 115200
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.810543101477172e-05,
      "loss": 4.6188,
      "step": 115712
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.80970450672612e-05,
      "loss": 4.6063,
      "step": 116224
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.808865911975068e-05,
      "loss": 4.6141,
      "step": 116736
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.808027317224016e-05,
      "loss": 4.6061,
      "step": 117248
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.807188722472964e-05,
      "loss": 4.6047,
      "step": 117760
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.806350127721912e-05,
      "loss": 4.6082,
      "step": 118272
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.80551153297086e-05,
      "loss": 4.6081,
      "step": 118784
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.804672938219807e-05,
      "loss": 4.5911,
      "step": 119296
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.803835981349129e-05,
      "loss": 4.5996,
      "step": 119808
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.802997386598077e-05,
      "loss": 4.5972,
      "step": 120320
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.802158791847025e-05,
      "loss": 4.6029,
      "step": 120832
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.801320197095973e-05,
      "loss": 4.5948,
      "step": 121344
    },
    {
      "epoch": 1.01,
      "learning_rate": 4.800481602344921e-05,
      "loss": 4.5827,
      "step": 121856
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7996446454742424e-05,
      "loss": 4.5923,
      "step": 122368
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.79880605072319e-05,
      "loss": 4.5834,
      "step": 122880
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.797967455972138e-05,
      "loss": 4.5957,
      "step": 123392
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.797128861221086e-05,
      "loss": 4.577,
      "step": 123904
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.796290266470034e-05,
      "loss": 4.5814,
      "step": 124416
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7954533095993546e-05,
      "loss": 4.5918,
      "step": 124928
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7946147148483026e-05,
      "loss": 4.5816,
      "step": 125440
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7937761200972506e-05,
      "loss": 4.5627,
      "step": 125952
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7929375253461986e-05,
      "loss": 4.5757,
      "step": 126464
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.792098930595147e-05,
      "loss": 4.579,
      "step": 126976
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.791260335844095e-05,
      "loss": 4.5809,
      "step": 127488
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.790421741093043e-05,
      "loss": 4.5698,
      "step": 128000
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.789584784222364e-05,
      "loss": 4.5701,
      "step": 128512
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.788746189471312e-05,
      "loss": 4.5639,
      "step": 129024
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.78790759472026e-05,
      "loss": 4.5699,
      "step": 129536
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.787068999969208e-05,
      "loss": 4.5512,
      "step": 130048
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.786232043098529e-05,
      "loss": 4.5595,
      "step": 130560
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.785393448347477e-05,
      "loss": 4.566,
      "step": 131072
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.784554853596425e-05,
      "loss": 4.5617,
      "step": 131584
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.783716258845373e-05,
      "loss": 4.5464,
      "step": 132096
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.782877664094321e-05,
      "loss": 4.555,
      "step": 132608
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.782040707223642e-05,
      "loss": 4.5441,
      "step": 133120
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7812021124725906e-05,
      "loss": 4.5602,
      "step": 133632
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7803635177215386e-05,
      "loss": 4.5462,
      "step": 134144
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7795249229704866e-05,
      "loss": 4.5378,
      "step": 134656
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7786863282194346e-05,
      "loss": 4.5567,
      "step": 135168
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7778493713487555e-05,
      "loss": 4.5497,
      "step": 135680
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7770107765977035e-05,
      "loss": 4.5439,
      "step": 136192
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7761721818466515e-05,
      "loss": 4.5462,
      "step": 136704
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7753335870955995e-05,
      "loss": 4.544,
      "step": 137216
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7744966302249204e-05,
      "loss": 4.549,
      "step": 137728
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7736580354738684e-05,
      "loss": 4.5282,
      "step": 138240
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7728194407228164e-05,
      "loss": 4.5482,
      "step": 138752
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7719808459717644e-05,
      "loss": 4.5363,
      "step": 139264
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.771143889101086e-05,
      "loss": 4.5396,
      "step": 139776
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.770305294350034e-05,
      "loss": 4.5301,
      "step": 140288
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.769466699598982e-05,
      "loss": 4.5355,
      "step": 140800
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.76862810484793e-05,
      "loss": 4.5243,
      "step": 141312
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.767791147977251e-05,
      "loss": 4.5326,
      "step": 141824
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.766952553226199e-05,
      "loss": 4.5091,
      "step": 142336
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.766113958475147e-05,
      "loss": 4.5433,
      "step": 142848
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.765275363724095e-05,
      "loss": 4.5358,
      "step": 143360
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.764436768973043e-05,
      "loss": 4.5304,
      "step": 143872
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.763599812102364e-05,
      "loss": 4.5148,
      "step": 144384
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.762761217351312e-05,
      "loss": 4.5308,
      "step": 144896
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.76192262260026e-05,
      "loss": 4.5304,
      "step": 145408
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.761084027849208e-05,
      "loss": 4.5232,
      "step": 145920
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.760245433098156e-05,
      "loss": 4.5159,
      "step": 146432
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.7594084762274773e-05,
      "loss": 4.5234,
      "step": 146944
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.758571519356798e-05,
      "loss": 4.5201,
      "step": 147456
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.757732924605746e-05,
      "loss": 4.5264,
      "step": 147968
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.756894329854694e-05,
      "loss": 4.5028,
      "step": 148480
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.756055735103642e-05,
      "loss": 4.5184,
      "step": 148992
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.75521714035259e-05,
      "loss": 4.5155,
      "step": 149504
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.754380183481911e-05,
      "loss": 4.5093,
      "step": 150016
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.753541588730859e-05,
      "loss": 4.5219,
      "step": 150528
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.752702993979807e-05,
      "loss": 4.5025,
      "step": 151040
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.751864399228755e-05,
      "loss": 4.5115,
      "step": 151552
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.751027442358077e-05,
      "loss": 4.5097,
      "step": 152064
    },
    {
      "epoch": 1.02,
      "learning_rate": 4.750188847607025e-05,
      "loss": 4.5096,
      "step": 152576
    },
    {
      "epoch": 1.03,
      "eval_loss": 4.525084018707275,
      "eval_runtime": 315.6819,
      "eval_samples_per_second": 1208.783,
      "eval_steps_per_second": 37.775,
      "step": 152640
    }
  ],
  "logging_steps": 512,
  "max_steps": 3052726,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 10,
  "total_flos": 5.878969332335171e+16,
  "trial_name": null,
  "trial_params": null
}