{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.9996261682242991,
  "eval_steps": 250,
  "global_step": 1337,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.05009345794392523,
      "grad_norm": 5.180806636810303,
      "learning_rate": 4.925373134328357e-07,
      "loss": 0.4797,
      "step": 67
    },
    {
      "epoch": 0.10018691588785046,
      "grad_norm": 4.254225730895996,
      "learning_rate": 9.925373134328357e-07,
      "loss": 0.4389,
      "step": 134
    },
    {
      "epoch": 0.1502803738317757,
      "grad_norm": 1.3476804494857788,
      "learning_rate": 9.451371571072319e-07,
      "loss": 0.3254,
      "step": 201
    },
    {
      "epoch": 0.18691588785046728,
      "eval_loss": 0.27628111839294434,
      "eval_runtime": 28.3849,
      "eval_samples_per_second": 17.615,
      "eval_steps_per_second": 8.807,
      "step": 250
    },
    {
      "epoch": 0.20037383177570092,
      "grad_norm": 3.236503839492798,
      "learning_rate": 8.894430590191188e-07,
      "loss": 0.3114,
      "step": 268
    },
    {
      "epoch": 0.2504672897196262,
      "grad_norm": 2.4669530391693115,
      "learning_rate": 8.337489609310058e-07,
      "loss": 0.2587,
      "step": 335
    },
    {
      "epoch": 0.3005607476635514,
      "grad_norm": 1.2674647569656372,
      "learning_rate": 7.780548628428927e-07,
      "loss": 0.2137,
      "step": 402
    },
    {
      "epoch": 0.3506542056074766,
      "grad_norm": 2.225503444671631,
      "learning_rate": 7.223607647547797e-07,
      "loss": 0.2079,
      "step": 469
    },
    {
      "epoch": 0.37383177570093457,
      "eval_loss": 0.19800835847854614,
      "eval_runtime": 28.3888,
      "eval_samples_per_second": 17.613,
      "eval_steps_per_second": 8.806,
      "step": 500
    },
    {
      "epoch": 0.40074766355140184,
      "grad_norm": 1.3379801511764526,
      "learning_rate": 6.666666666666666e-07,
      "loss": 0.1884,
      "step": 536
    },
    {
      "epoch": 0.4508411214953271,
      "grad_norm": 1.3498804569244385,
      "learning_rate": 6.109725685785536e-07,
      "loss": 0.183,
      "step": 603
    },
    {
      "epoch": 0.5009345794392523,
      "grad_norm": 2.507737874984741,
      "learning_rate": 5.552784704904405e-07,
      "loss": 0.1772,
      "step": 670
    },
    {
      "epoch": 0.5510280373831775,
      "grad_norm": 2.597299575805664,
      "learning_rate": 4.995843724023275e-07,
      "loss": 0.1729,
      "step": 737
    },
    {
      "epoch": 0.5607476635514018,
      "eval_loss": 0.17047198116779327,
      "eval_runtime": 28.3578,
      "eval_samples_per_second": 17.632,
      "eval_steps_per_second": 8.816,
      "step": 750
    },
    {
      "epoch": 0.6011214953271028,
      "grad_norm": 3.0745019912719727,
      "learning_rate": 4.438902743142144e-07,
      "loss": 0.163,
      "step": 804
    },
    {
      "epoch": 0.6512149532710281,
      "grad_norm": 2.3236138820648193,
      "learning_rate": 3.881961762261014e-07,
      "loss": 0.1656,
      "step": 871
    },
    {
      "epoch": 0.7013084112149532,
      "grad_norm": 1.7440522909164429,
      "learning_rate": 3.3250207813798835e-07,
      "loss": 0.1551,
      "step": 938
    },
    {
      "epoch": 0.7476635514018691,
      "eval_loss": 0.15852276980876923,
      "eval_runtime": 28.3587,
      "eval_samples_per_second": 17.631,
      "eval_steps_per_second": 8.816,
      "step": 1000
    },
    {
      "epoch": 0.7514018691588785,
      "grad_norm": 1.2390040159225464,
      "learning_rate": 2.7680798004987534e-07,
      "loss": 0.1359,
      "step": 1005
    },
    {
      "epoch": 0.8014953271028037,
      "grad_norm": 1.791257619857788,
      "learning_rate": 2.2111388196176226e-07,
      "loss": 0.1392,
      "step": 1072
    },
    {
      "epoch": 0.851588785046729,
      "grad_norm": 2.2460813522338867,
      "learning_rate": 1.6541978387364923e-07,
      "loss": 0.1537,
      "step": 1139
    },
    {
      "epoch": 0.9016822429906542,
      "grad_norm": 0.983316957950592,
      "learning_rate": 1.0972568578553615e-07,
      "loss": 0.1371,
      "step": 1206
    },
    {
      "epoch": 0.9345794392523364,
      "eval_loss": 0.1559886783361435,
      "eval_runtime": 28.3644,
      "eval_samples_per_second": 17.628,
      "eval_steps_per_second": 8.814,
      "step": 1250
    },
    {
      "epoch": 0.9517757009345794,
      "grad_norm": 2.740877866744995,
      "learning_rate": 5.403158769742311e-08,
      "loss": 0.1372,
      "step": 1273
    },
    {
      "epoch": 0.9996261682242991,
      "step": 1337,
      "total_flos": 9.176656366338048e+16,
      "train_loss": 0.214627830859165,
      "train_runtime": 2095.2086,
      "train_samples_per_second": 5.107,
      "train_steps_per_second": 0.638
    }
  ],
  "logging_steps": 67,
  "max_steps": 1337,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 500,
  "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.176656366338048e+16,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}