{
  "best_metric": 4.7972731590271,
  "best_model_checkpoint": "/mmfs1/gscratch/stf/abhinavp/corpus-filtering/outputs/pp-mod-subj2/lstm/0/checkpoints/checkpoint-76320",
  "epoch": 0.025000606015738065,
  "eval_steps": 10,
  "global_step": 76320,
  "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.8195,
      "step": 1
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.999161405248948e-05,
      "loss": 7.5629,
      "step": 512
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.998322810497896e-05,
      "loss": 7.0621,
      "step": 1024
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.997484215746844e-05,
      "loss": 6.9908,
      "step": 1536
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.996645620995792e-05,
      "loss": 6.9369,
      "step": 2048
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.99580702624474e-05,
      "loss": 6.8371,
      "step": 2560
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994968431493688e-05,
      "loss": 6.7009,
      "step": 3072
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994129836742636e-05,
      "loss": 6.5988,
      "step": 3584
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.993291241991584e-05,
      "loss": 6.5058,
      "step": 4096
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.992452647240532e-05,
      "loss": 6.4276,
      "step": 4608
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.99161405248948e-05,
      "loss": 6.3668,
      "step": 5120
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.990775457738428e-05,
      "loss": 6.2887,
      "step": 5632
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989936862987376e-05,
      "loss": 6.2278,
      "step": 6144
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989098268236324e-05,
      "loss": 6.1552,
      "step": 6656
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.988259673485272e-05,
      "loss": 6.1049,
      "step": 7168
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.98742107873422e-05,
      "loss": 6.04,
      "step": 7680
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.986582483983168e-05,
      "loss": 6.0126,
      "step": 8192
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.985743889232116e-05,
      "loss": 5.9574,
      "step": 8704
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.984905294481064e-05,
      "loss": 5.9143,
      "step": 9216
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.984066699730012e-05,
      "loss": 5.8808,
      "step": 9728
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.983229742859333e-05,
      "loss": 5.8575,
      "step": 10240
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.982391148108281e-05,
      "loss": 5.8109,
      "step": 10752
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.981552553357229e-05,
      "loss": 5.7867,
      "step": 11264
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.980713958606178e-05,
      "loss": 5.7509,
      "step": 11776
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9798770017354986e-05,
      "loss": 5.7184,
      "step": 12288
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9790384069844466e-05,
      "loss": 5.7003,
      "step": 12800
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9781998122333946e-05,
      "loss": 5.6629,
      "step": 13312
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9773628553627155e-05,
      "loss": 5.6507,
      "step": 13824
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9765242606116635e-05,
      "loss": 5.625,
      "step": 14336
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9756856658606115e-05,
      "loss": 5.5976,
      "step": 14848
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9748470711095595e-05,
      "loss": 5.5835,
      "step": 15360
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.974008476358507e-05,
      "loss": 5.561,
      "step": 15872
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.973169881607455e-05,
      "loss": 5.5482,
      "step": 16384
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.972331286856403e-05,
      "loss": 5.529,
      "step": 16896
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9714943299857244e-05,
      "loss": 5.5105,
      "step": 17408
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.970657373115046e-05,
      "loss": 5.5025,
      "step": 17920
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.969818778363994e-05,
      "loss": 5.495,
      "step": 18432
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.968980183612942e-05,
      "loss": 5.4556,
      "step": 18944
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.968143226742263e-05,
      "loss": 5.4478,
      "step": 19456
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.967304631991211e-05,
      "loss": 5.4339,
      "step": 19968
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.966466037240159e-05,
      "loss": 5.4193,
      "step": 20480
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.965627442489107e-05,
      "loss": 5.4053,
      "step": 20992
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.964788847738054e-05,
      "loss": 5.4073,
      "step": 21504
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.963950252987002e-05,
      "loss": 5.3857,
      "step": 22016
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.96311165823595e-05,
      "loss": 5.371,
      "step": 22528
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.962273063484898e-05,
      "loss": 5.3638,
      "step": 23040
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.961434468733847e-05,
      "loss": 5.3546,
      "step": 23552
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.960595873982795e-05,
      "loss": 5.3485,
      "step": 24064
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.959757279231743e-05,
      "loss": 5.3232,
      "step": 24576
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958918684480691e-05,
      "loss": 5.3127,
      "step": 25088
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958080089729639e-05,
      "loss": 5.3212,
      "step": 25600
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.95724313285896e-05,
      "loss": 5.31,
      "step": 26112
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.956404538107908e-05,
      "loss": 5.2974,
      "step": 26624
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.955567581237229e-05,
      "loss": 5.2747,
      "step": 27136
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.954728986486177e-05,
      "loss": 5.2834,
      "step": 27648
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.953890391735125e-05,
      "loss": 5.26,
      "step": 28160
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9530517969840727e-05,
      "loss": 5.2779,
      "step": 28672
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9522132022330207e-05,
      "loss": 5.243,
      "step": 29184
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9513746074819686e-05,
      "loss": 5.2466,
      "step": 29696
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9505360127309166e-05,
      "loss": 5.2271,
      "step": 30208
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9496974179798646e-05,
      "loss": 5.2167,
      "step": 30720
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948858823228813e-05,
      "loss": 5.2181,
      "step": 31232
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948020228477761e-05,
      "loss": 5.2027,
      "step": 31744
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9471816337267086e-05,
      "loss": 5.1976,
      "step": 32256
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.94634467685603e-05,
      "loss": 5.1888,
      "step": 32768
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.945506082104978e-05,
      "loss": 5.1986,
      "step": 33280
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9446674873539255e-05,
      "loss": 5.1712,
      "step": 33792
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9438288926028735e-05,
      "loss": 5.1614,
      "step": 34304
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9429902978518215e-05,
      "loss": 5.1454,
      "step": 34816
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.942153340981143e-05,
      "loss": 5.1456,
      "step": 35328
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9413147462300904e-05,
      "loss": 5.1594,
      "step": 35840
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9404761514790384e-05,
      "loss": 5.1494,
      "step": 36352
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.939637556727987e-05,
      "loss": 5.1299,
      "step": 36864
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.938798961976935e-05,
      "loss": 5.1276,
      "step": 37376
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.937960367225883e-05,
      "loss": 5.1282,
      "step": 37888
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.937121772474831e-05,
      "loss": 5.1223,
      "step": 38400
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.936284815604152e-05,
      "loss": 5.099,
      "step": 38912
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9354462208531e-05,
      "loss": 5.1024,
      "step": 39424
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.934607626102048e-05,
      "loss": 5.0889,
      "step": 39936
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.933769031350996e-05,
      "loss": 5.0915,
      "step": 40448
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.932932074480317e-05,
      "loss": 5.0867,
      "step": 40960
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.932093479729265e-05,
      "loss": 5.076,
      "step": 41472
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.931254884978213e-05,
      "loss": 5.078,
      "step": 41984
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.930416290227161e-05,
      "loss": 5.0683,
      "step": 42496
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.929577695476109e-05,
      "loss": 5.0516,
      "step": 43008
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9287407386054305e-05,
      "loss": 5.0508,
      "step": 43520
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9279021438543785e-05,
      "loss": 5.0557,
      "step": 44032
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9270651869836994e-05,
      "loss": 5.0489,
      "step": 44544
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9262265922326474e-05,
      "loss": 5.0365,
      "step": 45056
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9253879974815954e-05,
      "loss": 5.0226,
      "step": 45568
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9245494027305433e-05,
      "loss": 5.0284,
      "step": 46080
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9237108079794913e-05,
      "loss": 5.0152,
      "step": 46592
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.922872213228439e-05,
      "loss": 5.0247,
      "step": 47104
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.922033618477387e-05,
      "loss": 5.0043,
      "step": 47616
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.921195023726335e-05,
      "loss": 4.9997,
      "step": 48128
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.920358066855656e-05,
      "loss": 5.0168,
      "step": 48640
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.919519472104604e-05,
      "loss": 4.9837,
      "step": 49152
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.918680877353552e-05,
      "loss": 4.9776,
      "step": 49664
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.917842282602501e-05,
      "loss": 4.9822,
      "step": 50176
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.917003687851449e-05,
      "loss": 4.9802,
      "step": 50688
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.91616673098077e-05,
      "loss": 4.9811,
      "step": 51200
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.915328136229718e-05,
      "loss": 4.9687,
      "step": 51712
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.914489541478666e-05,
      "loss": 4.963,
      "step": 52224
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.913650946727614e-05,
      "loss": 4.9546,
      "step": 52736
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.912812351976562e-05,
      "loss": 4.9551,
      "step": 53248
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.911973757225509e-05,
      "loss": 4.9382,
      "step": 53760
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.911136800354831e-05,
      "loss": 4.9392,
      "step": 54272
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.910298205603779e-05,
      "loss": 4.9423,
      "step": 54784
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.909459610852727e-05,
      "loss": 4.9314,
      "step": 55296
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.908621016101675e-05,
      "loss": 4.9248,
      "step": 55808
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.907782421350623e-05,
      "loss": 4.92,
      "step": 56320
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906943826599571e-05,
      "loss": 4.9109,
      "step": 56832
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906105231848519e-05,
      "loss": 4.9207,
      "step": 57344
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9052682749778396e-05,
      "loss": 4.9065,
      "step": 57856
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9044296802267876e-05,
      "loss": 4.9033,
      "step": 58368
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9035910854757356e-05,
      "loss": 4.9045,
      "step": 58880
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9027524907246836e-05,
      "loss": 4.8985,
      "step": 59392
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9019155338540045e-05,
      "loss": 4.8923,
      "step": 59904
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9010769391029525e-05,
      "loss": 4.8907,
      "step": 60416
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9002383443519005e-05,
      "loss": 4.8915,
      "step": 60928
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8993997496008485e-05,
      "loss": 4.8907,
      "step": 61440
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.89856279273017e-05,
      "loss": 4.8697,
      "step": 61952
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.897724197979118e-05,
      "loss": 4.8797,
      "step": 62464
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.896885603228066e-05,
      "loss": 4.8704,
      "step": 62976
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.896047008477014e-05,
      "loss": 4.8727,
      "step": 63488
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.895208413725962e-05,
      "loss": 4.8536,
      "step": 64000
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.89436981897491e-05,
      "loss": 4.8626,
      "step": 64512
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.893531224223858e-05,
      "loss": 4.8477,
      "step": 65024
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.892694267353179e-05,
      "loss": 4.8542,
      "step": 65536
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.891855672602127e-05,
      "loss": 4.831,
      "step": 66048
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.891017077851075e-05,
      "loss": 4.8561,
      "step": 66560
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.890178483100023e-05,
      "loss": 4.8589,
      "step": 67072
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.889341526229344e-05,
      "loss": 4.8391,
      "step": 67584
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.888502931478292e-05,
      "loss": 4.8271,
      "step": 68096
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.88766433672724e-05,
      "loss": 4.8383,
      "step": 68608
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.886825741976188e-05,
      "loss": 4.831,
      "step": 69120
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8859871472251365e-05,
      "loss": 4.829,
      "step": 69632
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8851485524740845e-05,
      "loss": 4.8236,
      "step": 70144
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8843099577230325e-05,
      "loss": 4.8268,
      "step": 70656
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8834713629719805e-05,
      "loss": 4.8167,
      "step": 71168
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.882636043981674e-05,
      "loss": 4.8112,
      "step": 71680
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.881797449230622e-05,
      "loss": 4.7965,
      "step": 72192
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.88095885447957e-05,
      "loss": 4.808,
      "step": 72704
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.880120259728518e-05,
      "loss": 4.8048,
      "step": 73216
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.879281664977466e-05,
      "loss": 4.7955,
      "step": 73728
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.878443070226414e-05,
      "loss": 4.8082,
      "step": 74240
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.877604475475362e-05,
      "loss": 4.7873,
      "step": 74752
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.87676588072431e-05,
      "loss": 4.7961,
      "step": 75264
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.875928923853632e-05,
      "loss": 4.7969,
      "step": 75776
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.87509032910258e-05,
      "loss": 4.7865,
      "step": 76288
    },
    {
      "epoch": 0.03,
      "eval_loss": 4.7972731590271,
      "eval_runtime": 313.984,
      "eval_samples_per_second": 1215.32,
      "eval_steps_per_second": 37.98,
      "step": 76320
    }
  ],
  "logging_steps": 512,
  "max_steps": 3052726,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 10,
  "total_flos": 2.939484666167586e+16,
  "trial_name": null,
  "trial_params": null
}