{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.05044136191677175,
  "eval_steps": 10,
  "global_step": 20,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "computed_total": 0.45715999603271484,
      "epoch": 0.0025220680958385876,
      "grad_norm": 102.097412109375,
      "kl_loss": 6.5083294309431494e-09,
      "learning_rate": 0.0,
      "loss": 4.3237,
      "sft_loss": 0.45650917291641235,
      "step": 1
    },
    {
      "computed_total": 0.555546760559082,
      "epoch": 0.005044136191677175,
      "grad_norm": 87.49225616455078,
      "kl_loss": 5.4945092919922445e-09,
      "learning_rate": 4.000000000000001e-06,
      "loss": 4.3036,
      "sft_loss": 0.5549973249435425,
      "step": 2
    },
    {
      "computed_total": 0.5587829947471619,
      "epoch": 0.007566204287515763,
      "grad_norm": 64.52155303955078,
      "kl_loss": 3.4872137177899276e-08,
      "learning_rate": 8.000000000000001e-06,
      "loss": 3.5946,
      "sft_loss": 0.5552957653999329,
      "step": 3
    },
    {
      "computed_total": 0.4053139090538025,
      "epoch": 0.01008827238335435,
      "grad_norm": 79.32300567626953,
      "kl_loss": 7.649157964806363e-07,
      "learning_rate": 1.2e-05,
      "loss": 3.3472,
      "sft_loss": 0.3288223147392273,
      "step": 4
    },
    {
      "computed_total": 0.3922015130519867,
      "epoch": 0.012610340479192938,
      "grad_norm": 49.438594818115234,
      "kl_loss": 9.65095068750088e-07,
      "learning_rate": 1.6000000000000003e-05,
      "loss": 3.0682,
      "sft_loss": 0.29569199681282043,
      "step": 5
    },
    {
      "computed_total": 0.3126932382583618,
      "epoch": 0.015132408575031526,
      "grad_norm": 36.26791000366211,
      "kl_loss": 9.634337629904621e-07,
      "learning_rate": 2e-05,
      "loss": 2.7815,
      "sft_loss": 0.21634985506534576,
      "step": 6
    },
    {
      "computed_total": 0.2588472068309784,
      "epoch": 0.017654476670870115,
      "grad_norm": 28.21232795715332,
      "kl_loss": 7.276251494658936e-07,
      "learning_rate": 1.9948979591836737e-05,
      "loss": 2.499,
      "sft_loss": 0.18608468770980835,
      "step": 7
    },
    {
      "computed_total": 0.32910704612731934,
      "epoch": 0.0201765447667087,
      "grad_norm": 24.3712158203125,
      "kl_loss": 8.822735821922834e-07,
      "learning_rate": 1.9897959183673473e-05,
      "loss": 2.4776,
      "sft_loss": 0.24087969958782196,
      "step": 8
    },
    {
      "computed_total": 0.3658309280872345,
      "epoch": 0.02269861286254729,
      "grad_norm": 31.98260498046875,
      "kl_loss": 7.195029070317105e-07,
      "learning_rate": 1.9846938775510205e-05,
      "loss": 2.3718,
      "sft_loss": 0.2938806414604187,
      "step": 9
    },
    {
      "computed_total": 0.22178040444850922,
      "epoch": 0.025220680958385876,
      "grad_norm": 34.0799674987793,
      "kl_loss": 6.166483217384666e-07,
      "learning_rate": 1.979591836734694e-05,
      "loss": 2.4129,
      "sft_loss": 0.1601155698299408,
      "step": 10
    },
    {
      "computed_total": 2.1096696853637695,
      "epoch": 0.025220680958385876,
      "eval_loss": 1.9816868305206299,
      "eval_runtime": 67.6013,
      "eval_samples_per_second": 10.429,
      "eval_steps_per_second": 5.222,
      "kl_loss": 3.9526437944914505e-07,
      "sft_loss": 2.070143222808838,
      "step": 10
    },
    {
      "computed_total": 0.4089103043079376,
      "epoch": 0.027742749054224466,
      "grad_norm": 26.954879760742188,
      "kl_loss": 4.996533107259893e-07,
      "learning_rate": 1.9744897959183677e-05,
      "loss": 2.277,
      "sft_loss": 0.35894498229026794,
      "step": 11
    },
    {
      "computed_total": 0.23354634642601013,
      "epoch": 0.03026481715006305,
      "grad_norm": 18.464841842651367,
      "kl_loss": 5.376580816118803e-07,
      "learning_rate": 1.969387755102041e-05,
      "loss": 2.2583,
      "sft_loss": 0.17978054285049438,
      "step": 12
    },
    {
      "computed_total": 0.30147990584373474,
      "epoch": 0.03278688524590164,
      "grad_norm": 26.595727920532227,
      "kl_loss": 5.573554062721087e-07,
      "learning_rate": 1.9642857142857145e-05,
      "loss": 2.2624,
      "sft_loss": 0.24574437737464905,
      "step": 13
    },
    {
      "computed_total": 0.24872392416000366,
      "epoch": 0.03530895334174023,
      "grad_norm": 16.613079071044922,
      "kl_loss": 4.973371687810868e-07,
      "learning_rate": 1.9591836734693877e-05,
      "loss": 2.1151,
      "sft_loss": 0.19899021089076996,
      "step": 14
    },
    {
      "computed_total": 0.19776779413223267,
      "epoch": 0.03783102143757881,
      "grad_norm": 15.080662727355957,
      "kl_loss": 4.196848806259368e-07,
      "learning_rate": 1.9540816326530613e-05,
      "loss": 2.0897,
      "sft_loss": 0.15579931437969208,
      "step": 15
    },
    {
      "computed_total": 0.284699946641922,
      "epoch": 0.0403530895334174,
      "grad_norm": 11.973522186279297,
      "kl_loss": 4.3317794506947394e-07,
      "learning_rate": 1.948979591836735e-05,
      "loss": 2.216,
      "sft_loss": 0.2413821518421173,
      "step": 16
    },
    {
      "computed_total": 0.21680015325546265,
      "epoch": 0.04287515762925599,
      "grad_norm": 11.934453964233398,
      "kl_loss": 3.477235850368743e-07,
      "learning_rate": 1.9438775510204085e-05,
      "loss": 1.9235,
      "sft_loss": 0.18202780187129974,
      "step": 17
    },
    {
      "computed_total": 0.18643273413181305,
      "epoch": 0.04539722572509458,
      "grad_norm": 10.074520111083984,
      "kl_loss": 2.590618919384724e-07,
      "learning_rate": 1.9387755102040817e-05,
      "loss": 1.8522,
      "sft_loss": 0.16052654385566711,
      "step": 18
    },
    {
      "computed_total": 0.22932234406471252,
      "epoch": 0.04791929382093316,
      "grad_norm": 15.449592590332031,
      "kl_loss": 2.511018522000086e-07,
      "learning_rate": 1.9336734693877553e-05,
      "loss": 1.7982,
      "sft_loss": 0.20421215891838074,
      "step": 19
    },
    {
      "computed_total": 0.29722726345062256,
      "epoch": 0.05044136191677175,
      "grad_norm": 12.395018577575684,
      "kl_loss": 2.6658420892999857e-07,
      "learning_rate": 1.928571428571429e-05,
      "loss": 1.817,
      "sft_loss": 0.27056884765625,
      "step": 20
    },
    {
      "computed_total": 1.6116889715194702,
      "epoch": 0.05044136191677175,
      "eval_loss": 1.5718247890472412,
      "eval_runtime": 69.8078,
      "eval_samples_per_second": 10.099,
      "eval_steps_per_second": 5.057,
      "kl_loss": 3.4212027344437956e-07,
      "sft_loss": 1.577476978302002,
      "step": 20
    }
  ],
  "logging_steps": 1,
  "max_steps": 397,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 10,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 3691795705651200.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}