{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0,
  "eval_steps": 500,
  "global_step": 384,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.02609262883235486,
      "grad_norm": 90.7743148803711,
      "learning_rate": 5.128205128205128e-07,
      "loss": 3.4203,
      "step": 5
    },
    {
      "epoch": 0.05218525766470972,
      "grad_norm": 52.55401611328125,
      "learning_rate": 1.153846153846154e-06,
      "loss": 2.8478,
      "step": 10
    },
    {
      "epoch": 0.07827788649706457,
      "grad_norm": 17.9848575592041,
      "learning_rate": 1.794871794871795e-06,
      "loss": 1.524,
      "step": 15
    },
    {
      "epoch": 0.10437051532941943,
      "grad_norm": 9.579212188720703,
      "learning_rate": 2.435897435897436e-06,
      "loss": 0.8379,
      "step": 20
    },
    {
      "epoch": 0.1304631441617743,
      "grad_norm": 6.430282115936279,
      "learning_rate": 3.0769230769230774e-06,
      "loss": 0.4496,
      "step": 25
    },
    {
      "epoch": 0.15655577299412915,
      "grad_norm": 5.243109703063965,
      "learning_rate": 3.7179487179487184e-06,
      "loss": 0.2512,
      "step": 30
    },
    {
      "epoch": 0.182648401826484,
      "grad_norm": 5.690948963165283,
      "learning_rate": 4.358974358974359e-06,
      "loss": 0.183,
      "step": 35
    },
    {
      "epoch": 0.20874103065883887,
      "grad_norm": 4.116969585418701,
      "learning_rate": 5e-06,
      "loss": 0.1788,
      "step": 40
    },
    {
      "epoch": 0.23483365949119372,
      "grad_norm": 2.8701882362365723,
      "learning_rate": 4.9974091841168195e-06,
      "loss": 0.1605,
      "step": 45
    },
    {
      "epoch": 0.2609262883235486,
      "grad_norm": 7.059335231781006,
      "learning_rate": 4.989642106328829e-06,
      "loss": 0.1394,
      "step": 50
    },
    {
      "epoch": 0.28701891715590344,
      "grad_norm": 3.2651431560516357,
      "learning_rate": 4.976714865090827e-06,
      "loss": 0.083,
      "step": 55
    },
    {
      "epoch": 0.3131115459882583,
      "grad_norm": 2.457581043243408,
      "learning_rate": 4.958654254084356e-06,
      "loss": 0.0896,
      "step": 60
    },
    {
      "epoch": 0.33920417482061316,
      "grad_norm": 2.72770619392395,
      "learning_rate": 4.935497706683698e-06,
      "loss": 0.0991,
      "step": 65
    },
    {
      "epoch": 0.365296803652968,
      "grad_norm": 2.0538723468780518,
      "learning_rate": 4.907293218369499e-06,
      "loss": 0.0979,
      "step": 70
    },
    {
      "epoch": 0.3913894324853229,
      "grad_norm": 2.1862363815307617,
      "learning_rate": 4.874099247250799e-06,
      "loss": 0.0933,
      "step": 75
    },
    {
      "epoch": 0.41748206131767773,
      "grad_norm": 2.605069637298584,
      "learning_rate": 4.835984592901678e-06,
      "loss": 0.0959,
      "step": 80
    },
    {
      "epoch": 0.4435746901500326,
      "grad_norm": 1.2768350839614868,
      "learning_rate": 4.793028253763633e-06,
      "loss": 0.0749,
      "step": 85
    },
    {
      "epoch": 0.46966731898238745,
      "grad_norm": 4.374709606170654,
      "learning_rate": 4.745319263409241e-06,
      "loss": 0.0991,
      "step": 90
    },
    {
      "epoch": 0.4957599478147423,
      "grad_norm": 1.9398008584976196,
      "learning_rate": 4.692956506006486e-06,
      "loss": 0.0833,
      "step": 95
    },
    {
      "epoch": 0.5218525766470972,
      "grad_norm": 1.7450673580169678,
      "learning_rate": 4.636048511366222e-06,
      "loss": 0.0804,
      "step": 100
    },
    {
      "epoch": 0.547945205479452,
      "grad_norm": 1.712754249572754,
      "learning_rate": 4.5747132299975634e-06,
      "loss": 0.0867,
      "step": 105
    },
    {
      "epoch": 0.5740378343118069,
      "grad_norm": 1.083371877670288,
      "learning_rate": 4.509077788637446e-06,
      "loss": 0.0688,
      "step": 110
    },
    {
      "epoch": 0.6001304631441617,
      "grad_norm": 1.936558485031128,
      "learning_rate": 4.43927822676105e-06,
      "loss": 0.0772,
      "step": 115
    },
    {
      "epoch": 0.6262230919765166,
      "grad_norm": 1.4485265016555786,
      "learning_rate": 4.3654592146192146e-06,
      "loss": 0.0767,
      "step": 120
    },
    {
      "epoch": 0.6523157208088715,
      "grad_norm": 1.4740047454833984,
      "learning_rate": 4.287773753387249e-06,
      "loss": 0.0715,
      "step": 125
    },
    {
      "epoch": 0.6784083496412263,
      "grad_norm": 1.7988250255584717,
      "learning_rate": 4.206382858046636e-06,
      "loss": 0.0823,
      "step": 130
    },
    {
      "epoch": 0.7045009784735812,
      "grad_norm": 0.9675837159156799,
      "learning_rate": 4.12145522365689e-06,
      "loss": 0.0955,
      "step": 135
    },
    {
      "epoch": 0.730593607305936,
      "grad_norm": 1.3645009994506836,
      "learning_rate": 4.033166875709291e-06,
      "loss": 0.062,
      "step": 140
    },
    {
      "epoch": 0.7566862361382909,
      "grad_norm": 1.3581523895263672,
      "learning_rate": 3.941700805287169e-06,
      "loss": 0.0868,
      "step": 145
    },
    {
      "epoch": 0.7827788649706457,
      "grad_norm": 1.8347728252410889,
      "learning_rate": 3.84724658978894e-06,
      "loss": 0.0681,
      "step": 150
    },
    {
      "epoch": 0.8088714938030006,
      "grad_norm": 1.278552770614624,
      "learning_rate": 3.7500000000000005e-06,
      "loss": 0.0709,
      "step": 155
    },
    {
      "epoch": 0.8349641226353555,
      "grad_norm": 1.676578402519226,
      "learning_rate": 3.650162594327881e-06,
      "loss": 0.0714,
      "step": 160
    },
    {
      "epoch": 0.8610567514677103,
      "grad_norm": 0.9458003640174866,
      "learning_rate": 3.5479413010416606e-06,
      "loss": 0.081,
      "step": 165
    },
    {
      "epoch": 0.8871493803000652,
      "grad_norm": 1.0945783853530884,
      "learning_rate": 3.443547989381536e-06,
      "loss": 0.0715,
      "step": 170
    },
    {
      "epoch": 0.91324200913242,
      "grad_norm": 1.2435601949691772,
      "learning_rate": 3.3371990304274654e-06,
      "loss": 0.0638,
      "step": 175
    },
    {
      "epoch": 0.9393346379647749,
      "grad_norm": 1.2059943675994873,
      "learning_rate": 3.2291148486370626e-06,
      "loss": 0.0683,
      "step": 180
    },
    {
      "epoch": 0.9654272667971298,
      "grad_norm": 0.6467380523681641,
      "learning_rate": 3.11951946498225e-06,
      "loss": 0.0709,
      "step": 185
    },
    {
      "epoch": 0.9915198956294846,
      "grad_norm": 0.8123487830162048,
      "learning_rate": 3.0086400326315853e-06,
      "loss": 0.0601,
      "step": 190
    },
    {
      "epoch": 1.015655577299413,
      "grad_norm": 1.2716907262802124,
      "learning_rate": 2.896706366140629e-06,
      "loss": 0.0579,
      "step": 195
    },
    {
      "epoch": 1.0417482061317678,
      "grad_norm": 1.656204104423523,
      "learning_rate": 2.7839504651261873e-06,
      "loss": 0.0623,
      "step": 200
    },
    {
      "epoch": 1.0678408349641226,
      "grad_norm": 1.195453405380249,
      "learning_rate": 2.670606033411678e-06,
      "loss": 0.0533,
      "step": 205
    },
    {
      "epoch": 1.0939334637964775,
      "grad_norm": 1.3654705286026,
      "learning_rate": 2.556907994640264e-06,
      "loss": 0.0556,
      "step": 210
    },
    {
      "epoch": 1.1200260926288323,
      "grad_norm": 0.6175478100776672,
      "learning_rate": 2.443092005359736e-06,
      "loss": 0.0459,
      "step": 215
    },
    {
      "epoch": 1.1461187214611872,
      "grad_norm": 0.884466826915741,
      "learning_rate": 2.3293939665883233e-06,
      "loss": 0.057,
      "step": 220
    },
    {
      "epoch": 1.172211350293542,
      "grad_norm": 1.4396722316741943,
      "learning_rate": 2.2160495348738127e-06,
      "loss": 0.063,
      "step": 225
    },
    {
      "epoch": 1.198303979125897,
      "grad_norm": 0.893132209777832,
      "learning_rate": 2.1032936338593716e-06,
      "loss": 0.0554,
      "step": 230
    },
    {
      "epoch": 1.2243966079582518,
      "grad_norm": 0.778917133808136,
      "learning_rate": 1.991359967368416e-06,
      "loss": 0.0538,
      "step": 235
    },
    {
      "epoch": 1.2504892367906066,
      "grad_norm": 1.2776492834091187,
      "learning_rate": 1.8804805350177507e-06,
      "loss": 0.0538,
      "step": 240
    },
    {
      "epoch": 1.2765818656229615,
      "grad_norm": 1.0802513360977173,
      "learning_rate": 1.7708851513629376e-06,
      "loss": 0.0541,
      "step": 245
    },
    {
      "epoch": 1.3026744944553164,
      "grad_norm": 0.9070869088172913,
      "learning_rate": 1.6628009695725348e-06,
      "loss": 0.0479,
      "step": 250
    },
    {
      "epoch": 1.3287671232876712,
      "grad_norm": 1.4413901567459106,
      "learning_rate": 1.5564520106184643e-06,
      "loss": 0.0514,
      "step": 255
    },
    {
      "epoch": 1.354859752120026,
      "grad_norm": 1.1004434823989868,
      "learning_rate": 1.4520586989583406e-06,
      "loss": 0.0486,
      "step": 260
    },
    {
      "epoch": 1.380952380952381,
      "grad_norm": 1.1776546239852905,
      "learning_rate": 1.3498374056721198e-06,
      "loss": 0.0467,
      "step": 265
    },
    {
      "epoch": 1.4070450097847358,
      "grad_norm": 1.00455641746521,
      "learning_rate": 1.2500000000000007e-06,
      "loss": 0.0555,
      "step": 270
    },
    {
      "epoch": 1.4331376386170906,
      "grad_norm": 1.1658108234405518,
      "learning_rate": 1.1527534102110613e-06,
      "loss": 0.05,
      "step": 275
    },
    {
      "epoch": 1.4592302674494455,
      "grad_norm": 0.5006061792373657,
      "learning_rate": 1.0582991947128324e-06,
      "loss": 0.0531,
      "step": 280
    },
    {
      "epoch": 1.4853228962818004,
      "grad_norm": 0.5298914909362793,
      "learning_rate": 9.66833124290709e-07,
      "loss": 0.0475,
      "step": 285
    },
    {
      "epoch": 1.5114155251141552,
      "grad_norm": 0.6051680445671082,
      "learning_rate": 8.785447763431101e-07,
      "loss": 0.0444,
      "step": 290
    },
    {
      "epoch": 1.53750815394651,
      "grad_norm": 1.11183500289917,
      "learning_rate": 7.936171419533653e-07,
      "loss": 0.0547,
      "step": 295
    },
    {
      "epoch": 1.563600782778865,
      "grad_norm": 2.068556308746338,
      "learning_rate": 7.122262466127513e-07,
      "loss": 0.0501,
      "step": 300
    },
    {
      "epoch": 1.5896934116112198,
      "grad_norm": 0.9381551742553711,
      "learning_rate": 6.345407853807864e-07,
      "loss": 0.0535,
      "step": 305
    },
    {
      "epoch": 1.6157860404435747,
      "grad_norm": 1.0716235637664795,
      "learning_rate": 5.607217732389503e-07,
      "loss": 0.0533,
      "step": 310
    },
    {
      "epoch": 1.6418786692759295,
      "grad_norm": 0.8771371245384216,
      "learning_rate": 4.909222113625545e-07,
      "loss": 0.0567,
      "step": 315
    },
    {
      "epoch": 1.6679712981082844,
      "grad_norm": 1.4611281156539917,
      "learning_rate": 4.252867700024374e-07,
      "loss": 0.0569,
      "step": 320
    },
    {
      "epoch": 1.6940639269406392,
      "grad_norm": 0.42611926794052124,
      "learning_rate": 3.639514886337786e-07,
      "loss": 0.0562,
      "step": 325
    },
    {
      "epoch": 1.720156555772994,
      "grad_norm": 1.3697057962417603,
      "learning_rate": 3.0704349399351437e-07,
      "loss": 0.0504,
      "step": 330
    },
    {
      "epoch": 1.746249184605349,
      "grad_norm": 1.6137443780899048,
      "learning_rate": 2.5468073659076e-07,
      "loss": 0.0481,
      "step": 335
    },
    {
      "epoch": 1.7723418134377038,
      "grad_norm": 0.8786627054214478,
      "learning_rate": 2.0697174623636795e-07,
      "loss": 0.0435,
      "step": 340
    },
    {
      "epoch": 1.7984344422700587,
      "grad_norm": 0.6073723435401917,
      "learning_rate": 1.6401540709832242e-07,
      "loss": 0.0448,
      "step": 345
    },
    {
      "epoch": 1.8245270711024135,
      "grad_norm": 0.4912233054637909,
      "learning_rate": 1.2590075274920206e-07,
      "loss": 0.0531,
      "step": 350
    },
    {
      "epoch": 1.8506196999347684,
      "grad_norm": 1.127508282661438,
      "learning_rate": 9.270678163050218e-08,
      "loss": 0.0521,
      "step": 355
    },
    {
      "epoch": 1.8767123287671232,
      "grad_norm": 1.1689162254333496,
      "learning_rate": 6.450229331630253e-08,
      "loss": 0.051,
      "step": 360
    },
    {
      "epoch": 1.902804957599478,
      "grad_norm": 0.8796715140342712,
      "learning_rate": 4.134574591564494e-08,
      "loss": 0.0514,
      "step": 365
    },
    {
      "epoch": 1.928897586431833,
      "grad_norm": 0.8864848017692566,
      "learning_rate": 2.3285134909173113e-08,
      "loss": 0.0484,
      "step": 370
    },
    {
      "epoch": 1.9549902152641878,
      "grad_norm": 0.7268601655960083,
      "learning_rate": 1.0357893671171793e-08,
      "loss": 0.0502,
      "step": 375
    },
    {
      "epoch": 1.9810828440965427,
      "grad_norm": 0.7900459170341492,
      "learning_rate": 2.590815883181108e-09,
      "loss": 0.0468,
      "step": 380
    },
    {
      "epoch": 2.0,
      "step": 384,
      "total_flos": 9.374131655263642e+17,
      "train_loss": 0.18539645452983677,
      "train_runtime": 8558.6373,
      "train_samples_per_second": 2.865,
      "train_steps_per_second": 0.045
    }
  ],
  "logging_steps": 5,
  "max_steps": 384,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 2,
  "save_steps": 200,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 9.374131655263642e+17,
  "train_batch_size": 4,
  "trial_name": null,
  "trial_params": null
}