trainer_state.json

{
  "best_metric": 0.8268827257542107,
  "best_model_checkpoint": "/content/result/unsup-simcse-roberta-large-facebook-election-ads",
  "epoch": 5.0,
  "global_step": 1425,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.44,
      "eval_avg_sts": 0.7713986248458491,
      "eval_sickr_spearman": 0.7196138734589604,
      "eval_stsb_spearman": 0.8231833762327377,
      "step": 125
    },
    {
      "epoch": 0.88,
      "eval_avg_sts": 0.7816935841568983,
      "eval_sickr_spearman": 0.7365044425595858,
      "eval_stsb_spearman": 0.8268827257542107,
      "step": 250
    },
    {
      "epoch": 1.32,
      "eval_avg_sts": 0.7645407603583253,
      "eval_sickr_spearman": 0.7141448601248113,
      "eval_stsb_spearman": 0.8149366605918393,
      "step": 375
    },
    {
      "epoch": 1.75,
      "learning_rate": 3.24561403508772e-05,
      "loss": 0.0019,
      "step": 500
    },
    {
      "epoch": 1.75,
      "eval_avg_sts": 0.7604863072298488,
      "eval_sickr_spearman": 0.7226733585583653,
      "eval_stsb_spearman": 0.7982992559013322,
      "step": 500
    },
    {
      "epoch": 2.19,
      "eval_avg_sts": 0.7556532453851812,
      "eval_sickr_spearman": 0.7095467467219377,
      "eval_stsb_spearman": 0.8017597440484248,
      "step": 625
    },
    {
      "epoch": 2.63,
      "eval_avg_sts": 0.7581763481027948,
      "eval_sickr_spearman": 0.7125405183600657,
      "eval_stsb_spearman": 0.803812177845524,
      "step": 750
    },
    {
      "epoch": 3.07,
      "eval_avg_sts": 0.7366916009615327,
      "eval_sickr_spearman": 0.6934594017306621,
      "eval_stsb_spearman": 0.7799238001924034,
      "step": 875
    },
    {
      "epoch": 3.51,
      "learning_rate": 1.4912280701754386e-05,
      "loss": 0.0011,
      "step": 1000
    },
    {
      "epoch": 3.51,
      "eval_avg_sts": 0.7261565018963125,
      "eval_sickr_spearman": 0.6887153218105183,
      "eval_stsb_spearman": 0.7635976819821068,
      "step": 1000
    },
    {
      "epoch": 3.95,
      "eval_avg_sts": 0.73942694473879,
      "eval_sickr_spearman": 0.699325680304806,
      "eval_stsb_spearman": 0.7795282091727741,
      "step": 1125
    },
    {
      "epoch": 4.39,
      "eval_avg_sts": 0.7486781326241809,
      "eval_sickr_spearman": 0.707213684000683,
      "eval_stsb_spearman": 0.7901425812476789,
      "step": 1250
    },
    {
      "epoch": 4.82,
      "eval_avg_sts": 0.753180147484883,
      "eval_sickr_spearman": 0.7108770641384692,
      "eval_stsb_spearman": 0.7954832308312969,
      "step": 1375
    },
    {
      "epoch": 5.0,
      "step": 1425,
      "train_runtime": 3104.0027,
      "train_samples_per_second": 0.459
    }
  ],
  "max_steps": 1425,
  "num_train_epochs": 5,
  "total_flos": 12419737766461440,
  "trial_name": null,
  "trial_params": null
}