{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 4.993927617196988,
  "eval_steps": 500,
  "global_step": 2570,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.9715812484819043,
      "grad_norm": 0.0,
      "learning_rate": 1.9460259489596192e-05,
      "loss": 0.0282,
      "step": 500
    },
    {
      "epoch": 0.9987855234393976,
      "eval_reranking_1_loss": 0.182732954621315,
      "eval_reranking_1_runtime": 1.2858,
      "eval_reranking_1_samples_per_second": 29.553,
      "eval_reranking_1_steps_per_second": 14.777,
      "step": 514
    },
    {
      "epoch": 0.9987855234393976,
      "eval_retrival_1_loss": 0.1305331587791443,
      "eval_retrival_1_runtime": 26.5236,
      "eval_retrival_1_samples_per_second": 32.726,
      "eval_retrival_1_steps_per_second": 16.363,
      "step": 514
    },
    {
      "epoch": 0.9987855234393976,
      "eval_sts_1_loss": 0.2950426936149597,
      "eval_sts_1_runtime": 0.2829,
      "eval_sts_1_samples_per_second": 35.351,
      "eval_sts_1_steps_per_second": 17.676,
      "step": 514
    },
    {
      "epoch": 1.9431624969638086,
      "grad_norm": 6.779000282287598,
      "learning_rate": 1.5325659218524745e-05,
      "loss": 0.0555,
      "step": 1000
    },
    {
      "epoch": 1.999514209375759,
      "eval_reranking_1_loss": 0.3932221829891205,
      "eval_reranking_1_runtime": 1.258,
      "eval_reranking_1_samples_per_second": 30.208,
      "eval_reranking_1_steps_per_second": 15.104,
      "step": 1029
    },
    {
      "epoch": 1.999514209375759,
      "eval_retrival_1_loss": 0.06928794831037521,
      "eval_retrival_1_runtime": 28.2993,
      "eval_retrival_1_samples_per_second": 30.672,
      "eval_retrival_1_steps_per_second": 15.336,
      "step": 1029
    },
    {
      "epoch": 1.999514209375759,
      "eval_sts_1_loss": 0.8648566007614136,
      "eval_sts_1_runtime": 0.2978,
      "eval_sts_1_samples_per_second": 33.582,
      "eval_sts_1_steps_per_second": 16.791,
      "step": 1029
    },
    {
      "epoch": 2.914743745445713,
      "grad_norm": 0.030390629544854164,
      "learning_rate": 8.827829529818852e-06,
      "loss": 0.0151,
      "step": 1500
    },
    {
      "epoch": 2.998299732815157,
      "eval_reranking_1_loss": 0.21115952730178833,
      "eval_reranking_1_runtime": 1.2048,
      "eval_reranking_1_samples_per_second": 31.54,
      "eval_reranking_1_steps_per_second": 15.77,
      "step": 1543
    },
    {
      "epoch": 2.998299732815157,
      "eval_retrival_1_loss": 0.05546686798334122,
      "eval_retrival_1_runtime": 26.8806,
      "eval_retrival_1_samples_per_second": 32.291,
      "eval_retrival_1_steps_per_second": 16.145,
      "step": 1543
    },
    {
      "epoch": 2.998299732815157,
      "eval_sts_1_loss": 0.5478371977806091,
      "eval_sts_1_runtime": 0.2864,
      "eval_sts_1_samples_per_second": 34.913,
      "eval_sts_1_steps_per_second": 17.457,
      "step": 1543
    },
    {
      "epoch": 3.886324993927617,
      "grad_norm": 0.0,
      "learning_rate": 2.8501434747835012e-06,
      "loss": 0.0036,
      "step": 2000
    },
    {
      "epoch": 3.999028418751518,
      "eval_reranking_1_loss": 0.19206289947032928,
      "eval_reranking_1_runtime": 1.196,
      "eval_reranking_1_samples_per_second": 31.774,
      "eval_reranking_1_steps_per_second": 15.887,
      "step": 2058
    },
    {
      "epoch": 3.999028418751518,
      "eval_retrival_1_loss": 0.04320598766207695,
      "eval_retrival_1_runtime": 27.2793,
      "eval_retrival_1_samples_per_second": 31.819,
      "eval_retrival_1_steps_per_second": 15.909,
      "step": 2058
    },
    {
      "epoch": 3.999028418751518,
      "eval_sts_1_loss": 0.591204047203064,
      "eval_sts_1_runtime": 0.2948,
      "eval_sts_1_samples_per_second": 33.923,
      "eval_sts_1_steps_per_second": 16.961,
      "step": 2058
    },
    {
      "epoch": 4.857906242409522,
      "grad_norm": 0.0018939984729513526,
      "learning_rate": 4.5163479025036596e-08,
      "loss": 0.0013,
      "step": 2500
    },
    {
      "epoch": 4.993927617196988,
      "eval_reranking_1_loss": 0.19038808345794678,
      "eval_reranking_1_runtime": 1.207,
      "eval_reranking_1_samples_per_second": 31.484,
      "eval_reranking_1_steps_per_second": 15.742,
      "step": 2570
    },
    {
      "epoch": 4.993927617196988,
      "eval_retrival_1_loss": 0.04124879837036133,
      "eval_retrival_1_runtime": 27.5211,
      "eval_retrival_1_samples_per_second": 31.539,
      "eval_retrival_1_steps_per_second": 15.77,
      "step": 2570
    },
    {
      "epoch": 4.993927617196988,
      "eval_sts_1_loss": 0.6356238722801208,
      "eval_sts_1_runtime": 0.3051,
      "eval_sts_1_samples_per_second": 32.781,
      "eval_sts_1_steps_per_second": 16.39,
      "step": 2570
    }
  ],
  "logging_steps": 500,
  "max_steps": 2570,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "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": 0.0,
  "train_batch_size": 2,
  "trial_name": null,
  "trial_params": null
}