Time_SFT/checkpoint-100/trainer_state.json

{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 3.587443946188341,
  "eval_steps": 500,
  "global_step": 100,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.35874439461883406,
      "grad_norm": 5.573123455047607,
      "learning_rate": 9.259259259259259e-07,
      "logits/chosen": -2.3576977252960205,
      "logits/rejected": -2.3553850650787354,
      "logps/chosen": -61.0846061706543,
      "logps/rejected": -126.1152572631836,
      "loss": 0.6942,
      "rewards/accuracies": 0.36250001192092896,
      "rewards/chosen": -0.0012381849810481071,
      "rewards/margins": -0.010831715539097786,
      "rewards/rejected": 0.009593529626727104,
      "step": 10
    },
    {
      "epoch": 0.7174887892376681,
      "grad_norm": 3.8175244331359863,
      "learning_rate": 1.8518518518518519e-06,
      "logits/chosen": -2.3618054389953613,
      "logits/rejected": -2.384873867034912,
      "logps/chosen": -63.2865104675293,
      "logps/rejected": -81.8724365234375,
      "loss": 0.6925,
      "rewards/accuracies": 0.4749999940395355,
      "rewards/chosen": -0.002874236088246107,
      "rewards/margins": -0.003056168556213379,
      "rewards/rejected": 0.00018193255527876318,
      "step": 20
    },
    {
      "epoch": 1.0762331838565022,
      "grad_norm": 4.383662700653076,
      "learning_rate": 2.7777777777777783e-06,
      "logits/chosen": -2.3333091735839844,
      "logits/rejected": -2.3545360565185547,
      "logps/chosen": -76.02193450927734,
      "logps/rejected": -88.19218444824219,
      "loss": 0.6932,
      "rewards/accuracies": 0.5249999761581421,
      "rewards/chosen": -0.008687756024301052,
      "rewards/margins": -0.0016246589366346598,
      "rewards/rejected": -0.007063096854835749,
      "step": 30
    },
    {
      "epoch": 1.4349775784753362,
      "grad_norm": 4.095139503479004,
      "learning_rate": 3.7037037037037037e-06,
      "logits/chosen": -2.3483197689056396,
      "logits/rejected": -2.363131523132324,
      "logps/chosen": -66.08761596679688,
      "logps/rejected": -78.22222900390625,
      "loss": 0.6889,
      "rewards/accuracies": 0.5249999761581421,
      "rewards/chosen": -0.014177674427628517,
      "rewards/margins": 0.012472175993025303,
      "rewards/rejected": -0.026649847626686096,
      "step": 40
    },
    {
      "epoch": 1.7937219730941703,
      "grad_norm": 4.034746170043945,
      "learning_rate": 4.62962962962963e-06,
      "logits/chosen": -2.31459903717041,
      "logits/rejected": -2.335706949234009,
      "logps/chosen": -69.31814575195312,
      "logps/rejected": -77.91527557373047,
      "loss": 0.6823,
      "rewards/accuracies": 0.4749999940395355,
      "rewards/chosen": -0.04294499754905701,
      "rewards/margins": 0.007987136952579021,
      "rewards/rejected": -0.050932131707668304,
      "step": 50
    },
    {
      "epoch": 2.1524663677130045,
      "grad_norm": 4.620233535766602,
      "learning_rate": 4.998119881260576e-06,
      "logits/chosen": -2.364583969116211,
      "logits/rejected": -2.3598217964172363,
      "logps/chosen": -69.84207153320312,
      "logps/rejected": -79.67817687988281,
      "loss": 0.6717,
      "rewards/accuracies": 0.699999988079071,
      "rewards/chosen": -0.06739393621683121,
      "rewards/margins": 0.050740111619234085,
      "rewards/rejected": -0.118134044110775,
      "step": 60
    },
    {
      "epoch": 2.5112107623318387,
      "grad_norm": 4.450082302093506,
      "learning_rate": 4.9866405060165044e-06,
      "logits/chosen": -2.298379421234131,
      "logits/rejected": -2.3169469833374023,
      "logps/chosen": -79.13768005371094,
      "logps/rejected": -79.24192810058594,
      "loss": 0.6499,
      "rewards/accuracies": 0.7124999761581421,
      "rewards/chosen": -0.1282506138086319,
      "rewards/margins": 0.08815944194793701,
      "rewards/rejected": -0.21641004085540771,
      "step": 70
    },
    {
      "epoch": 2.8699551569506725,
      "grad_norm": 4.7328877449035645,
      "learning_rate": 4.964774158361991e-06,
      "logits/chosen": -2.3363795280456543,
      "logits/rejected": -2.3331849575042725,
      "logps/chosen": -77.52214050292969,
      "logps/rejected": -86.90077209472656,
      "loss": 0.6346,
      "rewards/accuracies": 0.75,
      "rewards/chosen": -0.22365322709083557,
      "rewards/margins": 0.1937221735715866,
      "rewards/rejected": -0.4173754155635834,
      "step": 80
    },
    {
      "epoch": 3.2286995515695067,
      "grad_norm": 4.016219615936279,
      "learning_rate": 4.93261217644956e-06,
      "logits/chosen": -2.30871844291687,
      "logits/rejected": -2.3049044609069824,
      "logps/chosen": -65.6957778930664,
      "logps/rejected": -77.02860260009766,
      "loss": 0.6004,
      "rewards/accuracies": 0.762499988079071,
      "rewards/chosen": -0.23905189335346222,
      "rewards/margins": 0.2932591438293457,
      "rewards/rejected": -0.5323110222816467,
      "step": 90
    },
    {
      "epoch": 3.587443946188341,
      "grad_norm": 4.553714752197266,
      "learning_rate": 4.8902889044347e-06,
      "logits/chosen": -2.3068125247955322,
      "logits/rejected": -2.294462203979492,
      "logps/chosen": -85.46375274658203,
      "logps/rejected": -79.72854614257812,
      "loss": 0.5816,
      "rewards/accuracies": 0.6875,
      "rewards/chosen": -0.3506511151790619,
      "rewards/margins": 0.2997886538505554,
      "rewards/rejected": -0.6504397392272949,
      "step": 100
    }
  ],
  "logging_steps": 10,
  "max_steps": 540,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 100,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 3.648437622595912e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}