{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.5421686746987951,
  "eval_steps": 500,
  "global_step": 90,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.006024096385542169,
      "grad_norm": 11.142176628112793,
      "learning_rate": 0.0,
      "loss": 0.428902804851532,
      "step": 1
    },
    {
      "epoch": 0.012048192771084338,
      "grad_norm": 11.034109115600586,
      "learning_rate": 5.555555555555555e-07,
      "loss": 0.42792809009552,
      "step": 2
    },
    {
      "epoch": 0.018072289156626505,
      "grad_norm": 8.74612045288086,
      "learning_rate": 1.111111111111111e-06,
      "loss": 0.3810642957687378,
      "step": 3
    },
    {
      "epoch": 0.024096385542168676,
      "grad_norm": 9.825540542602539,
      "learning_rate": 1.6666666666666667e-06,
      "loss": 0.4792688190937042,
      "step": 4
    },
    {
      "epoch": 0.030120481927710843,
      "grad_norm": 9.189689636230469,
      "learning_rate": 2.222222222222222e-06,
      "loss": 0.40543031692504883,
      "step": 5
    },
    {
      "epoch": 0.03614457831325301,
      "grad_norm": 7.393019199371338,
      "learning_rate": 2.7777777777777783e-06,
      "loss": 0.4046850800514221,
      "step": 6
    },
    {
      "epoch": 0.04216867469879518,
      "grad_norm": 5.906346797943115,
      "learning_rate": 3.3333333333333333e-06,
      "loss": 0.3256700038909912,
      "step": 7
    },
    {
      "epoch": 0.04819277108433735,
      "grad_norm": 4.1642632484436035,
      "learning_rate": 3.88888888888889e-06,
      "loss": 0.2622697651386261,
      "step": 8
    },
    {
      "epoch": 0.05421686746987952,
      "grad_norm": 4.239490032196045,
      "learning_rate": 4.444444444444444e-06,
      "loss": 0.31185054779052734,
      "step": 9
    },
    {
      "epoch": 0.060240963855421686,
      "grad_norm": 2.696002721786499,
      "learning_rate": 5e-06,
      "loss": 0.26603323221206665,
      "step": 10
    },
    {
      "epoch": 0.06626506024096386,
      "grad_norm": 2.6713006496429443,
      "learning_rate": 4.999881749961716e-06,
      "loss": 0.29827162623405457,
      "step": 11
    },
    {
      "epoch": 0.07228915662650602,
      "grad_norm": 2.532520294189453,
      "learning_rate": 4.999527011033321e-06,
      "loss": 0.2619355022907257,
      "step": 12
    },
    {
      "epoch": 0.0783132530120482,
      "grad_norm": 2.390526533126831,
      "learning_rate": 4.9989358167731285e-06,
      "loss": 0.2530556321144104,
      "step": 13
    },
    {
      "epoch": 0.08433734939759036,
      "grad_norm": 2.513714551925659,
      "learning_rate": 4.998108223108133e-06,
      "loss": 0.2906743884086609,
      "step": 14
    },
    {
      "epoch": 0.09036144578313253,
      "grad_norm": 2.0978147983551025,
      "learning_rate": 4.9970443083287224e-06,
      "loss": 0.23413600027561188,
      "step": 15
    },
    {
      "epoch": 0.0963855421686747,
      "grad_norm": 2.065729856491089,
      "learning_rate": 4.995744173081265e-06,
      "loss": 0.24827918410301208,
      "step": 16
    },
    {
      "epoch": 0.10240963855421686,
      "grad_norm": 1.884876012802124,
      "learning_rate": 4.994207940358597e-06,
      "loss": 0.26246076822280884,
      "step": 17
    },
    {
      "epoch": 0.10843373493975904,
      "grad_norm": 2.5683226585388184,
      "learning_rate": 4.99243575548838e-06,
      "loss": 0.3206707835197449,
      "step": 18
    },
    {
      "epoch": 0.1144578313253012,
      "grad_norm": 1.6468095779418945,
      "learning_rate": 4.9904277861193574e-06,
      "loss": 0.20793572068214417,
      "step": 19
    },
    {
      "epoch": 0.12048192771084337,
      "grad_norm": 2.026458263397217,
      "learning_rate": 4.988184222205492e-06,
      "loss": 0.23015552759170532,
      "step": 20
    },
    {
      "epoch": 0.12650602409638553,
      "grad_norm": 1.8717762231826782,
      "learning_rate": 4.9857052759880006e-06,
      "loss": 0.25441068410873413,
      "step": 21
    },
    {
      "epoch": 0.13253012048192772,
      "grad_norm": 2.0290212631225586,
      "learning_rate": 4.98299118197527e-06,
      "loss": 0.22494634985923767,
      "step": 22
    },
    {
      "epoch": 0.13855421686746988,
      "grad_norm": 2.214284658432007,
      "learning_rate": 4.980042196920677e-06,
      "loss": 0.26744288206100464,
      "step": 23
    },
    {
      "epoch": 0.14457831325301204,
      "grad_norm": 2.1135001182556152,
      "learning_rate": 4.976858599798299e-06,
      "loss": 0.27789992094039917,
      "step": 24
    },
    {
      "epoch": 0.15060240963855423,
      "grad_norm": 1.9528898000717163,
      "learning_rate": 4.97344069177652e-06,
      "loss": 0.2466088980436325,
      "step": 25
    },
    {
      "epoch": 0.1566265060240964,
      "grad_norm": 2.101134777069092,
      "learning_rate": 4.969788796189544e-06,
      "loss": 0.2756834030151367,
      "step": 26
    },
    {
      "epoch": 0.16265060240963855,
      "grad_norm": 2.343674421310425,
      "learning_rate": 4.965903258506806e-06,
      "loss": 0.28239595890045166,
      "step": 27
    },
    {
      "epoch": 0.1686746987951807,
      "grad_norm": 2.253816843032837,
      "learning_rate": 4.9617844463002895e-06,
      "loss": 0.2833530306816101,
      "step": 28
    },
    {
      "epoch": 0.1746987951807229,
      "grad_norm": 2.0298430919647217,
      "learning_rate": 4.957432749209755e-06,
      "loss": 0.25803640484809875,
      "step": 29
    },
    {
      "epoch": 0.18072289156626506,
      "grad_norm": 1.9530489444732666,
      "learning_rate": 4.952848578905881e-06,
      "loss": 0.2215431183576584,
      "step": 30
    },
    {
      "epoch": 0.18674698795180722,
      "grad_norm": 1.940727949142456,
      "learning_rate": 4.948032369051317e-06,
      "loss": 0.21956567466259003,
      "step": 31
    },
    {
      "epoch": 0.1927710843373494,
      "grad_norm": 1.7422584295272827,
      "learning_rate": 4.942984575259666e-06,
      "loss": 0.20336072146892548,
      "step": 32
    },
    {
      "epoch": 0.19879518072289157,
      "grad_norm": 1.7944018840789795,
      "learning_rate": 4.937705675052373e-06,
      "loss": 0.2340494990348816,
      "step": 33
    },
    {
      "epoch": 0.20481927710843373,
      "grad_norm": 1.9949196577072144,
      "learning_rate": 4.932196167813559e-06,
      "loss": 0.30935412645339966,
      "step": 34
    },
    {
      "epoch": 0.21084337349397592,
      "grad_norm": 1.9301334619522095,
      "learning_rate": 4.926456574742778e-06,
      "loss": 0.24893832206726074,
      "step": 35
    },
    {
      "epoch": 0.21686746987951808,
      "grad_norm": 1.830209493637085,
      "learning_rate": 4.920487438805711e-06,
      "loss": 0.24357952177524567,
      "step": 36
    },
    {
      "epoch": 0.22289156626506024,
      "grad_norm": 1.988722324371338,
      "learning_rate": 4.9142893246828e-06,
      "loss": 0.22917479276657104,
      "step": 37
    },
    {
      "epoch": 0.2289156626506024,
      "grad_norm": 1.8187060356140137,
      "learning_rate": 4.907862818715831e-06,
      "loss": 0.27028727531433105,
      "step": 38
    },
    {
      "epoch": 0.23493975903614459,
      "grad_norm": 2.018294095993042,
      "learning_rate": 4.901208528852465e-06,
      "loss": 0.2310202419757843,
      "step": 39
    },
    {
      "epoch": 0.24096385542168675,
      "grad_norm": 2.0524044036865234,
      "learning_rate": 4.894327084588727e-06,
      "loss": 0.2731711268424988,
      "step": 40
    },
    {
      "epoch": 0.2469879518072289,
      "grad_norm": 1.8044854402542114,
      "learning_rate": 4.887219136909455e-06,
      "loss": 0.21416833996772766,
      "step": 41
    },
    {
      "epoch": 0.25301204819277107,
      "grad_norm": 1.7802573442459106,
      "learning_rate": 4.879885358226718e-06,
      "loss": 0.19698399305343628,
      "step": 42
    },
    {
      "epoch": 0.25903614457831325,
      "grad_norm": 2.2974190711975098,
      "learning_rate": 4.872326442316203e-06,
      "loss": 0.27448707818984985,
      "step": 43
    },
    {
      "epoch": 0.26506024096385544,
      "grad_norm": 2.2874867916107178,
      "learning_rate": 4.864543104251587e-06,
      "loss": 0.25334182381629944,
      "step": 44
    },
    {
      "epoch": 0.2710843373493976,
      "grad_norm": 1.9537646770477295,
      "learning_rate": 4.856536080336889e-06,
      "loss": 0.23862236738204956,
      "step": 45
    },
    {
      "epoch": 0.27710843373493976,
      "grad_norm": 2.0207152366638184,
      "learning_rate": 4.848306128036817e-06,
      "loss": 0.21768009662628174,
      "step": 46
    },
    {
      "epoch": 0.28313253012048195,
      "grad_norm": 1.8584778308868408,
      "learning_rate": 4.839854025905111e-06,
      "loss": 0.22923576831817627,
      "step": 47
    },
    {
      "epoch": 0.2891566265060241,
      "grad_norm": 1.950571060180664,
      "learning_rate": 4.83118057351089e-06,
      "loss": 0.24856369197368622,
      "step": 48
    },
    {
      "epoch": 0.29518072289156627,
      "grad_norm": 1.9584742784500122,
      "learning_rate": 4.822286591363018e-06,
      "loss": 0.22143439948558807,
      "step": 49
    },
    {
      "epoch": 0.30120481927710846,
      "grad_norm": 1.7273216247558594,
      "learning_rate": 4.813172920832477e-06,
      "loss": 0.21043998003005981,
      "step": 50
    },
    {
      "epoch": 0.3072289156626506,
      "grad_norm": 1.895005702972412,
      "learning_rate": 4.8038404240727785e-06,
      "loss": 0.2245764136314392,
      "step": 51
    },
    {
      "epoch": 0.3132530120481928,
      "grad_norm": 2.0731658935546875,
      "learning_rate": 4.794289983938403e-06,
      "loss": 0.22999663650989532,
      "step": 52
    },
    {
      "epoch": 0.3192771084337349,
      "grad_norm": 1.8308407068252563,
      "learning_rate": 4.784522503901279e-06,
      "loss": 0.195350781083107,
      "step": 53
    },
    {
      "epoch": 0.3253012048192771,
      "grad_norm": 1.843331217765808,
      "learning_rate": 4.774538907965318e-06,
      "loss": 0.2074728161096573,
      "step": 54
    },
    {
      "epoch": 0.3313253012048193,
      "grad_norm": 1.7621232271194458,
      "learning_rate": 4.764340140578999e-06,
      "loss": 0.2011730819940567,
      "step": 55
    },
    {
      "epoch": 0.3373493975903614,
      "grad_norm": 2.133352518081665,
      "learning_rate": 4.753927166546032e-06,
      "loss": 0.2272721230983734,
      "step": 56
    },
    {
      "epoch": 0.3433734939759036,
      "grad_norm": 1.7030085325241089,
      "learning_rate": 4.743300970934077e-06,
      "loss": 0.21829667687416077,
      "step": 57
    },
    {
      "epoch": 0.3493975903614458,
      "grad_norm": 1.9551599025726318,
      "learning_rate": 4.732462558981566e-06,
      "loss": 0.27412763237953186,
      "step": 58
    },
    {
      "epoch": 0.35542168674698793,
      "grad_norm": 1.7485458850860596,
      "learning_rate": 4.7214129560026015e-06,
      "loss": 0.2115289866924286,
      "step": 59
    },
    {
      "epoch": 0.3614457831325301,
      "grad_norm": 2.274949789047241,
      "learning_rate": 4.710153207289963e-06,
      "loss": 0.2607116997241974,
      "step": 60
    },
    {
      "epoch": 0.3674698795180723,
      "grad_norm": 1.6567832231521606,
      "learning_rate": 4.698684378016223e-06,
      "loss": 0.2039959728717804,
      "step": 61
    },
    {
      "epoch": 0.37349397590361444,
      "grad_norm": 2.048426866531372,
      "learning_rate": 4.687007553132984e-06,
      "loss": 0.2799915075302124,
      "step": 62
    },
    {
      "epoch": 0.3795180722891566,
      "grad_norm": 1.7995165586471558,
      "learning_rate": 4.675123837268235e-06,
      "loss": 0.21062196791172028,
      "step": 63
    },
    {
      "epoch": 0.3855421686746988,
      "grad_norm": 1.8946254253387451,
      "learning_rate": 4.663034354621864e-06,
      "loss": 0.20754952728748322,
      "step": 64
    },
    {
      "epoch": 0.39156626506024095,
      "grad_norm": 1.7134093046188354,
      "learning_rate": 4.650740248859297e-06,
      "loss": 0.21463951468467712,
      "step": 65
    },
    {
      "epoch": 0.39759036144578314,
      "grad_norm": 1.9219496250152588,
      "learning_rate": 4.638242683003317e-06,
      "loss": 0.26067647337913513,
      "step": 66
    },
    {
      "epoch": 0.4036144578313253,
      "grad_norm": 1.7691570520401,
      "learning_rate": 4.625542839324036e-06,
      "loss": 0.23639249801635742,
      "step": 67
    },
    {
      "epoch": 0.40963855421686746,
      "grad_norm": 1.9222103357315063,
      "learning_rate": 4.6126419192270555e-06,
      "loss": 0.2021777331829071,
      "step": 68
    },
    {
      "epoch": 0.41566265060240964,
      "grad_norm": 1.7899214029312134,
      "learning_rate": 4.599541143139811e-06,
      "loss": 0.24930432438850403,
      "step": 69
    },
    {
      "epoch": 0.42168674698795183,
      "grad_norm": 2.099332094192505,
      "learning_rate": 4.586241750396123e-06,
      "loss": 0.22903309762477875,
      "step": 70
    },
    {
      "epoch": 0.42771084337349397,
      "grad_norm": 1.7237921953201294,
      "learning_rate": 4.572744999118951e-06,
      "loss": 0.21340343356132507,
      "step": 71
    },
    {
      "epoch": 0.43373493975903615,
      "grad_norm": 2.0501701831817627,
      "learning_rate": 4.55905216610138e-06,
      "loss": 0.2957020699977875,
      "step": 72
    },
    {
      "epoch": 0.4397590361445783,
      "grad_norm": 1.9058964252471924,
      "learning_rate": 4.545164546685833e-06,
      "loss": 0.1939191371202469,
      "step": 73
    },
    {
      "epoch": 0.4457831325301205,
      "grad_norm": 1.6917678117752075,
      "learning_rate": 4.531083454641531e-06,
      "loss": 0.17142519354820251,
      "step": 74
    },
    {
      "epoch": 0.45180722891566266,
      "grad_norm": 1.7717102766036987,
      "learning_rate": 4.5168102220402134e-06,
      "loss": 0.2252553552389145,
      "step": 75
    },
    {
      "epoch": 0.4578313253012048,
      "grad_norm": 1.658211588859558,
      "learning_rate": 4.502346199130122e-06,
      "loss": 0.23183956742286682,
      "step": 76
    },
    {
      "epoch": 0.463855421686747,
      "grad_norm": 1.8013274669647217,
      "learning_rate": 4.487692754208266e-06,
      "loss": 0.22619429230690002,
      "step": 77
    },
    {
      "epoch": 0.46987951807228917,
      "grad_norm": 1.7795627117156982,
      "learning_rate": 4.472851273490985e-06,
      "loss": 0.2013862431049347,
      "step": 78
    },
    {
      "epoch": 0.4759036144578313,
      "grad_norm": 1.939047932624817,
      "learning_rate": 4.457823160982807e-06,
      "loss": 0.24375547468662262,
      "step": 79
    },
    {
      "epoch": 0.4819277108433735,
      "grad_norm": 2.0435917377471924,
      "learning_rate": 4.442609838343639e-06,
      "loss": 0.2810860276222229,
      "step": 80
    },
    {
      "epoch": 0.4879518072289157,
      "grad_norm": 1.6849004030227661,
      "learning_rate": 4.427212744754265e-06,
      "loss": 0.20575395226478577,
      "step": 81
    },
    {
      "epoch": 0.4939759036144578,
      "grad_norm": 1.7432836294174194,
      "learning_rate": 4.411633336780213e-06,
      "loss": 0.24015243351459503,
      "step": 82
    },
    {
      "epoch": 0.5,
      "grad_norm": 2.005021810531616,
      "learning_rate": 4.395873088233954e-06,
      "loss": 0.2686704397201538,
      "step": 83
    },
    {
      "epoch": 0.5060240963855421,
      "grad_norm": 1.9157148599624634,
      "learning_rate": 4.379933490035482e-06,
      "loss": 0.2602766752243042,
      "step": 84
    },
    {
      "epoch": 0.5120481927710844,
      "grad_norm": 1.8026031255722046,
      "learning_rate": 4.363816050071275e-06,
      "loss": 0.2138879895210266,
      "step": 85
    },
    {
      "epoch": 0.5180722891566265,
      "grad_norm": 1.8697198629379272,
      "learning_rate": 4.3475222930516484e-06,
      "loss": 0.24326106905937195,
      "step": 86
    },
    {
      "epoch": 0.5240963855421686,
      "grad_norm": 1.840242624282837,
      "learning_rate": 4.331053760366514e-06,
      "loss": 0.23169377446174622,
      "step": 87
    },
    {
      "epoch": 0.5301204819277109,
      "grad_norm": 1.9570574760437012,
      "learning_rate": 4.314412009939568e-06,
      "loss": 0.21268793940544128,
      "step": 88
    },
    {
      "epoch": 0.536144578313253,
      "grad_norm": 6.490395545959473,
      "learning_rate": 4.297598616080912e-06,
      "loss": 0.23041915893554688,
      "step": 89
    },
    {
      "epoch": 0.5421686746987951,
      "grad_norm": 1.701446294784546,
      "learning_rate": 4.28061516933812e-06,
      "loss": 0.19419308006763458,
      "step": 90
    }
  ],
  "logging_steps": 1,
  "max_steps": 332,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 30,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 5.1649109246450074e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}