{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.001239552893271397,
  "eval_steps": 500,
  "global_step": 50,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 2.479105786542794e-05,
      "grad_norm": 7.298285007476807,
      "learning_rate": 0.0001,
      "loss": 8.3458,
      "step": 1
    },
    {
      "epoch": 4.958211573085588e-05,
      "grad_norm": 8.879791259765625,
      "learning_rate": 0.0002,
      "loss": 8.3615,
      "step": 2
    },
    {
      "epoch": 7.437317359628382e-05,
      "grad_norm": 8.794269561767578,
      "learning_rate": 0.00019978589232386035,
      "loss": 7.5151,
      "step": 3
    },
    {
      "epoch": 9.916423146171176e-05,
      "grad_norm": 9.012420654296875,
      "learning_rate": 0.00019914448613738106,
      "loss": 5.6806,
      "step": 4
    },
    {
      "epoch": 0.0001239552893271397,
      "grad_norm": 7.061407566070557,
      "learning_rate": 0.00019807852804032305,
      "loss": 3.9168,
      "step": 5
    },
    {
      "epoch": 0.00014874634719256764,
      "grad_norm": 6.04531717300415,
      "learning_rate": 0.00019659258262890683,
      "loss": 2.518,
      "step": 6
    },
    {
      "epoch": 0.00017353740505799558,
      "grad_norm": 5.078192234039307,
      "learning_rate": 0.0001946930129495106,
      "loss": 1.6634,
      "step": 7
    },
    {
      "epoch": 0.00019832846292342353,
      "grad_norm": 5.033928871154785,
      "learning_rate": 0.0001923879532511287,
      "loss": 0.907,
      "step": 8
    },
    {
      "epoch": 0.00022311952078885147,
      "grad_norm": 3.0206494331359863,
      "learning_rate": 0.00018968727415326884,
      "loss": 0.6279,
      "step": 9
    },
    {
      "epoch": 0.0002479105786542794,
      "grad_norm": 2.810877799987793,
      "learning_rate": 0.00018660254037844388,
      "loss": 0.419,
      "step": 10
    },
    {
      "epoch": 0.00027270163651970736,
      "grad_norm": 3.125711441040039,
      "learning_rate": 0.00018314696123025454,
      "loss": 0.7372,
      "step": 11
    },
    {
      "epoch": 0.0002974926943851353,
      "grad_norm": 3.9015979766845703,
      "learning_rate": 0.00017933533402912354,
      "loss": 0.312,
      "step": 12
    },
    {
      "epoch": 0.0003222837522505632,
      "grad_norm": 1.802170753479004,
      "learning_rate": 0.00017518398074789775,
      "loss": 0.1449,
      "step": 13
    },
    {
      "epoch": 0.00034707481011599117,
      "grad_norm": 1.4817924499511719,
      "learning_rate": 0.00017071067811865476,
      "loss": 0.0737,
      "step": 14
    },
    {
      "epoch": 0.0003718658679814191,
      "grad_norm": 0.31127655506134033,
      "learning_rate": 0.00016593458151000688,
      "loss": 0.0081,
      "step": 15
    },
    {
      "epoch": 0.00039665692584684706,
      "grad_norm": 1.7345539331436157,
      "learning_rate": 0.00016087614290087208,
      "loss": 0.1763,
      "step": 16
    },
    {
      "epoch": 0.000421447983712275,
      "grad_norm": 0.861487865447998,
      "learning_rate": 0.00015555702330196023,
      "loss": 0.0059,
      "step": 17
    },
    {
      "epoch": 0.00044623904157770294,
      "grad_norm": 0.014484393410384655,
      "learning_rate": 0.00015000000000000001,
      "loss": 0.0003,
      "step": 18
    },
    {
      "epoch": 0.00047103009944313086,
      "grad_norm": 0.020801691338419914,
      "learning_rate": 0.00014422886902190014,
      "loss": 0.0003,
      "step": 19
    },
    {
      "epoch": 0.0004958211573085588,
      "grad_norm": 0.035404324531555176,
      "learning_rate": 0.000138268343236509,
      "loss": 0.0003,
      "step": 20
    },
    {
      "epoch": 0.0005206122151739868,
      "grad_norm": 15.371546745300293,
      "learning_rate": 0.00013214394653031616,
      "loss": 0.1066,
      "step": 21
    },
    {
      "epoch": 0.0005454032730394147,
      "grad_norm": 0.01011655479669571,
      "learning_rate": 0.00012588190451025207,
      "loss": 0.0002,
      "step": 22
    },
    {
      "epoch": 0.0005701943309048426,
      "grad_norm": 0.09178952872753143,
      "learning_rate": 0.00011950903220161285,
      "loss": 0.001,
      "step": 23
    },
    {
      "epoch": 0.0005949853887702706,
      "grad_norm": 0.00701731164008379,
      "learning_rate": 0.00011305261922200519,
      "loss": 0.0001,
      "step": 24
    },
    {
      "epoch": 0.0006197764466356985,
      "grad_norm": 0.014851056970655918,
      "learning_rate": 0.00010654031292301432,
      "loss": 0.0003,
      "step": 25
    },
    {
      "epoch": 0.0006445675045011264,
      "grad_norm": 2.1300110816955566,
      "learning_rate": 0.0001,
      "loss": 0.0058,
      "step": 26
    },
    {
      "epoch": 0.0006693585623665544,
      "grad_norm": 1.1658985614776611,
      "learning_rate": 9.345968707698569e-05,
      "loss": 0.0137,
      "step": 27
    },
    {
      "epoch": 0.0006941496202319823,
      "grad_norm": 0.008638538420200348,
      "learning_rate": 8.694738077799488e-05,
      "loss": 0.0001,
      "step": 28
    },
    {
      "epoch": 0.0007189406780974103,
      "grad_norm": 0.0034529389813542366,
      "learning_rate": 8.049096779838719e-05,
      "loss": 0.0001,
      "step": 29
    },
    {
      "epoch": 0.0007437317359628382,
      "grad_norm": 0.0020151117350906134,
      "learning_rate": 7.411809548974792e-05,
      "loss": 0.0,
      "step": 30
    },
    {
      "epoch": 0.0007685227938282661,
      "grad_norm": 2.9357662200927734,
      "learning_rate": 6.785605346968386e-05,
      "loss": 0.1847,
      "step": 31
    },
    {
      "epoch": 0.0007933138516936941,
      "grad_norm": 1.7568498849868774,
      "learning_rate": 6.173165676349103e-05,
      "loss": 0.0447,
      "step": 32
    },
    {
      "epoch": 0.000818104909559122,
      "grad_norm": 1.2945858240127563,
      "learning_rate": 5.577113097809989e-05,
      "loss": 0.029,
      "step": 33
    },
    {
      "epoch": 0.00084289596742455,
      "grad_norm": 0.003499679733067751,
      "learning_rate": 5.000000000000002e-05,
      "loss": 0.0001,
      "step": 34
    },
    {
      "epoch": 0.0008676870252899779,
      "grad_norm": 0.0012576651060953736,
      "learning_rate": 4.444297669803981e-05,
      "loss": 0.0,
      "step": 35
    },
    {
      "epoch": 0.0008924780831554059,
      "grad_norm": 0.0027500190772116184,
      "learning_rate": 3.9123857099127936e-05,
      "loss": 0.0001,
      "step": 36
    },
    {
      "epoch": 0.0009172691410208338,
      "grad_norm": 0.004673486575484276,
      "learning_rate": 3.406541848999312e-05,
      "loss": 0.0001,
      "step": 37
    },
    {
      "epoch": 0.0009420601988862617,
      "grad_norm": 0.0062264916487038136,
      "learning_rate": 2.9289321881345254e-05,
      "loss": 0.0001,
      "step": 38
    },
    {
      "epoch": 0.0009668512567516897,
      "grad_norm": 0.001728757400996983,
      "learning_rate": 2.4816019252102273e-05,
      "loss": 0.0,
      "step": 39
    },
    {
      "epoch": 0.0009916423146171176,
      "grad_norm": 0.004961659200489521,
      "learning_rate": 2.0664665970876496e-05,
      "loss": 0.0001,
      "step": 40
    },
    {
      "epoch": 0.0010164333724825456,
      "grad_norm": 0.0015779470559209585,
      "learning_rate": 1.6853038769745467e-05,
      "loss": 0.0,
      "step": 41
    },
    {
      "epoch": 0.0010412244303479735,
      "grad_norm": 0.0018025401514023542,
      "learning_rate": 1.339745962155613e-05,
      "loss": 0.0,
      "step": 42
    },
    {
      "epoch": 0.0010660154882134014,
      "grad_norm": 0.0029116978403180838,
      "learning_rate": 1.0312725846731175e-05,
      "loss": 0.0,
      "step": 43
    },
    {
      "epoch": 0.0010908065460788294,
      "grad_norm": 0.0026793968863785267,
      "learning_rate": 7.612046748871327e-06,
      "loss": 0.0,
      "step": 44
    },
    {
      "epoch": 0.0011155976039442573,
      "grad_norm": 0.002206047298386693,
      "learning_rate": 5.306987050489442e-06,
      "loss": 0.0,
      "step": 45
    },
    {
      "epoch": 0.0011403886618096852,
      "grad_norm": 0.002513278741389513,
      "learning_rate": 3.40741737109318e-06,
      "loss": 0.0,
      "step": 46
    },
    {
      "epoch": 0.0011651797196751133,
      "grad_norm": 0.00416777515783906,
      "learning_rate": 1.921471959676957e-06,
      "loss": 0.0001,
      "step": 47
    },
    {
      "epoch": 0.0011899707775405411,
      "grad_norm": 0.012963655404746532,
      "learning_rate": 8.555138626189618e-07,
      "loss": 0.0002,
      "step": 48
    },
    {
      "epoch": 0.001214761835405969,
      "grad_norm": 0.001530441571958363,
      "learning_rate": 2.141076761396521e-07,
      "loss": 0.0,
      "step": 49
    },
    {
      "epoch": 0.001239552893271397,
      "grad_norm": 0.002721717581152916,
      "learning_rate": 0.0,
      "loss": 0.0001,
      "step": 50
    }
  ],
  "logging_steps": 1,
  "max_steps": 50,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 25,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 6.6109411688448e+16,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}