trainer_state.json

{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.8944543828264758,
  "eval_steps": 500,
  "global_step": 2000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.22361359570661896,
      "grad_norm": 55.74723434448242,
      "learning_rate": 3.899821109123435e-05,
      "logits/chosen": -0.6040188074111938,
      "logits/rejected": -0.7450945973396301,
      "logps/chosen": -420.3378601074219,
      "logps/rejected": -462.8780517578125,
      "loss": 1.797,
      "rewards/accuracies": 0.6399999856948853,
      "rewards/chosen": -8.708248138427734,
      "rewards/margins": 2.4897494316101074,
      "rewards/rejected": -11.197998046875,
      "step": 500
    },
    {
      "epoch": 0.4472271914132379,
      "grad_norm": 53.70317459106445,
      "learning_rate": 2.78175313059034e-05,
      "logits/chosen": -0.5758157968521118,
      "logits/rejected": -0.6371915936470032,
      "logps/chosen": -458.7496032714844,
      "logps/rejected": -508.36602783203125,
      "loss": 2.0075,
      "rewards/accuracies": 0.6168749928474426,
      "rewards/chosen": -12.504631042480469,
      "rewards/margins": 3.4706857204437256,
      "rewards/rejected": -15.975316047668457,
      "step": 1000
    },
    {
      "epoch": 0.6708407871198568,
      "grad_norm": 24.010499954223633,
      "learning_rate": 1.6636851520572452e-05,
      "logits/chosen": -0.9147199392318726,
      "logits/rejected": -0.9995958805084229,
      "logps/chosen": -451.2334899902344,
      "logps/rejected": -513.6981811523438,
      "loss": 1.7831,
      "rewards/accuracies": 0.6547499895095825,
      "rewards/chosen": -11.545605659484863,
      "rewards/margins": 4.830613613128662,
      "rewards/rejected": -16.376218795776367,
      "step": 1500
    },
    {
      "epoch": 0.8944543828264758,
      "grad_norm": 38.50263977050781,
      "learning_rate": 5.456171735241503e-06,
      "logits/chosen": -0.8216423392295837,
      "logits/rejected": -0.8953205347061157,
      "logps/chosen": -446.2303771972656,
      "logps/rejected": -521.4434814453125,
      "loss": 1.5334,
      "rewards/accuracies": 0.6894999742507935,
      "rewards/chosen": -11.270264625549316,
      "rewards/margins": 6.051136016845703,
      "rewards/rejected": -17.321399688720703,
      "step": 2000
    }
  ],
  "logging_steps": 500,
  "max_steps": 2236,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 1,
  "save_steps": 250,
  "total_flos": 0.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}