{
  "best_metric": 4.811893463134766,
  "best_model_checkpoint": "/mmfs1/gscratch/stf/abhinavp/corpus-filtering/outputs/pp-mod-subj2/lstm/4/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.5627,
      "step": 512
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.998322810497896e-05,
      "loss": 7.0613,
      "step": 1024
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.997484215746844e-05,
      "loss": 6.9898,
      "step": 1536
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.996645620995792e-05,
      "loss": 6.9202,
      "step": 2048
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.99580702624474e-05,
      "loss": 6.7783,
      "step": 2560
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994968431493688e-05,
      "loss": 6.655,
      "step": 3072
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.994129836742636e-05,
      "loss": 6.5569,
      "step": 3584
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.993291241991584e-05,
      "loss": 6.4699,
      "step": 4096
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.992454285120906e-05,
      "loss": 6.398,
      "step": 4608
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.991615690369854e-05,
      "loss": 6.3437,
      "step": 5120
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.990777095618801e-05,
      "loss": 6.2739,
      "step": 5632
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989938500867749e-05,
      "loss": 6.2187,
      "step": 6144
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.989099906116697e-05,
      "loss": 6.1474,
      "step": 6656
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.988261311365645e-05,
      "loss": 6.0987,
      "step": 7168
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.987422716614593e-05,
      "loss": 6.0334,
      "step": 7680
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.986585759743914e-05,
      "loss": 6.0066,
      "step": 8192
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.985747164992862e-05,
      "loss": 5.9522,
      "step": 8704
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.98490857024181e-05,
      "loss": 5.9096,
      "step": 9216
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.984069975490758e-05,
      "loss": 5.8792,
      "step": 9728
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.983231380739706e-05,
      "loss": 5.8584,
      "step": 10240
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9823927859886547e-05,
      "loss": 5.8146,
      "step": 10752
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9815541912376026e-05,
      "loss": 5.7927,
      "step": 11264
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9807155964865506e-05,
      "loss": 5.7571,
      "step": 11776
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9798770017354986e-05,
      "loss": 5.7259,
      "step": 12288
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9790384069844466e-05,
      "loss": 5.7099,
      "step": 12800
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9781998122333946e-05,
      "loss": 5.6728,
      "step": 13312
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9773628553627155e-05,
      "loss": 5.6622,
      "step": 13824
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9765242606116635e-05,
      "loss": 5.6369,
      "step": 14336
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.9756856658606115e-05,
      "loss": 5.6098,
      "step": 14848
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9748470711095595e-05,
      "loss": 5.5975,
      "step": 15360
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.974008476358507e-05,
      "loss": 5.5742,
      "step": 15872
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.973169881607455e-05,
      "loss": 5.5614,
      "step": 16384
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9723329247367764e-05,
      "loss": 5.5445,
      "step": 16896
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9714943299857244e-05,
      "loss": 5.5246,
      "step": 17408
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9706557352346724e-05,
      "loss": 5.5164,
      "step": 17920
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9698171404836204e-05,
      "loss": 5.5095,
      "step": 18432
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.968980183612942e-05,
      "loss": 5.4711,
      "step": 18944
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.96814158886189e-05,
      "loss": 5.4617,
      "step": 19456
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.967302994110837e-05,
      "loss": 5.4483,
      "step": 19968
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.966464399359785e-05,
      "loss": 5.4321,
      "step": 20480
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.965625804608733e-05,
      "loss": 5.4199,
      "step": 20992
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.964787209857681e-05,
      "loss": 5.4204,
      "step": 21504
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.963948615106629e-05,
      "loss": 5.4004,
      "step": 22016
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.963110020355577e-05,
      "loss": 5.3849,
      "step": 22528
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.962273063484898e-05,
      "loss": 5.3784,
      "step": 23040
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.96143610661422e-05,
      "loss": 5.3678,
      "step": 23552
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.960597511863168e-05,
      "loss": 5.3617,
      "step": 24064
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.959758917112116e-05,
      "loss": 5.3377,
      "step": 24576
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958920322361064e-05,
      "loss": 5.3256,
      "step": 25088
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.958081727610012e-05,
      "loss": 5.3342,
      "step": 25600
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.95724313285896e-05,
      "loss": 5.3236,
      "step": 26112
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.956406175988281e-05,
      "loss": 5.31,
      "step": 26624
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.955567581237229e-05,
      "loss": 5.2879,
      "step": 27136
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.954728986486177e-05,
      "loss": 5.2967,
      "step": 27648
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.953890391735125e-05,
      "loss": 5.2745,
      "step": 28160
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9530517969840727e-05,
      "loss": 5.2905,
      "step": 28672
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9522132022330207e-05,
      "loss": 5.2562,
      "step": 29184
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9513746074819686e-05,
      "loss": 5.2603,
      "step": 29696
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9505360127309166e-05,
      "loss": 5.2399,
      "step": 30208
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.949700693740611e-05,
      "loss": 5.2296,
      "step": 30720
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948862098989559e-05,
      "loss": 5.232,
      "step": 31232
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.948023504238507e-05,
      "loss": 5.2168,
      "step": 31744
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.947184909487455e-05,
      "loss": 5.2111,
      "step": 32256
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.946346314736403e-05,
      "loss": 5.2033,
      "step": 32768
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.945507719985351e-05,
      "loss": 5.2122,
      "step": 33280
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.944670763114672e-05,
      "loss": 5.1851,
      "step": 33792
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.94383216836362e-05,
      "loss": 5.1755,
      "step": 34304
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.942993573612568e-05,
      "loss": 5.1604,
      "step": 34816
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.942154978861516e-05,
      "loss": 5.16,
      "step": 35328
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.941318021990837e-05,
      "loss": 5.174,
      "step": 35840
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9404794272397856e-05,
      "loss": 5.1638,
      "step": 36352
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9396408324887336e-05,
      "loss": 5.1459,
      "step": 36864
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9388022377376816e-05,
      "loss": 5.1416,
      "step": 37376
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9379636429866296e-05,
      "loss": 5.1417,
      "step": 37888
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9371250482355776e-05,
      "loss": 5.1357,
      "step": 38400
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9362880913648985e-05,
      "loss": 5.1136,
      "step": 38912
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9354494966138465e-05,
      "loss": 5.1178,
      "step": 39424
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9346109018627945e-05,
      "loss": 5.1047,
      "step": 39936
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9337723071117425e-05,
      "loss": 5.1061,
      "step": 40448
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9329337123606905e-05,
      "loss": 5.101,
      "step": 40960
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.932095117609638e-05,
      "loss": 5.0895,
      "step": 41472
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.931256522858586e-05,
      "loss": 5.0926,
      "step": 41984
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.930417928107534e-05,
      "loss": 5.0819,
      "step": 42496
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9295809712368554e-05,
      "loss": 5.0666,
      "step": 43008
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.9287423764858034e-05,
      "loss": 5.0658,
      "step": 43520
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.927907057495498e-05,
      "loss": 5.0708,
      "step": 44032
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.927068462744446e-05,
      "loss": 5.0638,
      "step": 44544
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.926229867993394e-05,
      "loss": 5.0512,
      "step": 45056
    },
    {
      "epoch": 0.01,
      "learning_rate": 4.925391273242342e-05,
      "loss": 5.0381,
      "step": 45568
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.92455267849129e-05,
      "loss": 5.0418,
      "step": 46080
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.923714083740238e-05,
      "loss": 5.0302,
      "step": 46592
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.922875488989185e-05,
      "loss": 5.0405,
      "step": 47104
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.922036894238133e-05,
      "loss": 5.019,
      "step": 47616
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.921198299487081e-05,
      "loss": 5.0136,
      "step": 48128
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.920361342616403e-05,
      "loss": 5.0313,
      "step": 48640
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.919522747865351e-05,
      "loss": 4.9991,
      "step": 49152
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.918684153114299e-05,
      "loss": 4.9929,
      "step": 49664
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.917845558363247e-05,
      "loss": 4.9976,
      "step": 50176
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9170086014925676e-05,
      "loss": 4.9954,
      "step": 50688
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9161700067415156e-05,
      "loss": 4.9961,
      "step": 51200
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9153314119904636e-05,
      "loss": 4.9851,
      "step": 51712
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9144928172394116e-05,
      "loss": 4.9779,
      "step": 52224
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9136542224883596e-05,
      "loss": 4.9699,
      "step": 52736
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9128156277373076e-05,
      "loss": 4.9699,
      "step": 53248
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9119770329862556e-05,
      "loss": 4.954,
      "step": 53760
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9111400761155765e-05,
      "loss": 4.9537,
      "step": 54272
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.9103014813645245e-05,
      "loss": 4.9583,
      "step": 54784
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.909462886613473e-05,
      "loss": 4.9465,
      "step": 55296
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.908624291862421e-05,
      "loss": 4.9388,
      "step": 55808
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.907785697111369e-05,
      "loss": 4.9357,
      "step": 56320
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906947102360317e-05,
      "loss": 4.9266,
      "step": 56832
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.906108507609265e-05,
      "loss": 4.9353,
      "step": 57344
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.905269912858213e-05,
      "loss": 4.9216,
      "step": 57856
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.904432955987534e-05,
      "loss": 4.9178,
      "step": 58368
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.903594361236482e-05,
      "loss": 4.9199,
      "step": 58880
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.90275576648543e-05,
      "loss": 4.9133,
      "step": 59392
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.901917171734378e-05,
      "loss": 4.9077,
      "step": 59904
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.901080214863699e-05,
      "loss": 4.9067,
      "step": 60416
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.900241620112647e-05,
      "loss": 4.9049,
      "step": 60928
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.899403025361595e-05,
      "loss": 4.9038,
      "step": 61440
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.898564430610543e-05,
      "loss": 4.8839,
      "step": 61952
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.897725835859491e-05,
      "loss": 4.8958,
      "step": 62464
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8968888789888125e-05,
      "loss": 4.8874,
      "step": 62976
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8960519221181335e-05,
      "loss": 4.8875,
      "step": 63488
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8952133273670815e-05,
      "loss": 4.8681,
      "step": 64000
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8943747326160294e-05,
      "loss": 4.8786,
      "step": 64512
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8935361378649774e-05,
      "loss": 4.8627,
      "step": 65024
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8926975431139254e-05,
      "loss": 4.8693,
      "step": 65536
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8918589483628734e-05,
      "loss": 4.8456,
      "step": 66048
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8910203536118214e-05,
      "loss": 4.8694,
      "step": 66560
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.890181758860769e-05,
      "loss": 4.8726,
      "step": 67072
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.889343164109717e-05,
      "loss": 4.8531,
      "step": 67584
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.888504569358665e-05,
      "loss": 4.8403,
      "step": 68096
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8876659746076134e-05,
      "loss": 4.8512,
      "step": 68608
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.8868273798565614e-05,
      "loss": 4.8454,
      "step": 69120
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.885990422985882e-05,
      "loss": 4.8435,
      "step": 69632
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.88515182823483e-05,
      "loss": 4.837,
      "step": 70144
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.884313233483778e-05,
      "loss": 4.8411,
      "step": 70656
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.883474638732726e-05,
      "loss": 4.8312,
      "step": 71168
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.882639319742421e-05,
      "loss": 4.8254,
      "step": 71680
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.881800724991369e-05,
      "loss": 4.8098,
      "step": 72192
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.880962130240316e-05,
      "loss": 4.8225,
      "step": 72704
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.880123535489264e-05,
      "loss": 4.8202,
      "step": 73216
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.879284940738212e-05,
      "loss": 4.8087,
      "step": 73728
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.87844634598716e-05,
      "loss": 4.8218,
      "step": 74240
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.877607751236109e-05,
      "loss": 4.8005,
      "step": 74752
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.87677079436543e-05,
      "loss": 4.8099,
      "step": 75264
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.875932199614378e-05,
      "loss": 4.8103,
      "step": 75776
    },
    {
      "epoch": 0.02,
      "learning_rate": 4.875093604863326e-05,
      "loss": 4.7988,
      "step": 76288
    },
    {
      "epoch": 0.03,
      "eval_loss": 4.811893463134766,
      "eval_runtime": 292.9712,
      "eval_samples_per_second": 1302.486,
      "eval_steps_per_second": 40.704,
      "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
}