trainer_state.json

{
  "best_metric": 0.8454911866853387,
  "best_model_checkpoint": "result/my-unsup-simcse-roberta-base/SupCon-48-0.2-1e-7-0.07-0.00",
  "epoch": 0.11199283245872264,
  "global_step": 875,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.02,
      "eval_avg_sts": 0.7685225645328353,
      "eval_sickr_spearman": 0.7173618872369332,
      "eval_stsb_spearman": 0.8196832418287373,
      "step": 125
    },
    {
      "epoch": 0.03,
      "eval_avg_sts": 0.773152170020105,
      "eval_sickr_spearman": 0.7157021725267392,
      "eval_stsb_spearman": 0.8306021675134706,
      "step": 250
    },
    {
      "epoch": 0.05,
      "eval_avg_sts": 0.7719290374646383,
      "eval_sickr_spearman": 0.7119126627165764,
      "eval_stsb_spearman": 0.8319454122127001,
      "step": 375
    },
    {
      "epoch": 0.06,
      "learning_rate": 1.872008191475746e-05,
      "loss": 0.1453,
      "step": 500
    },
    {
      "epoch": 0.06,
      "eval_avg_sts": 0.7734885653744008,
      "eval_sickr_spearman": 0.7109199559120535,
      "eval_stsb_spearman": 0.8360571748367481,
      "step": 500
    },
    {
      "epoch": 0.08,
      "eval_avg_sts": 0.7796495889494159,
      "eval_sickr_spearman": 0.7263016759921019,
      "eval_stsb_spearman": 0.8329975019067298,
      "step": 625
    },
    {
      "epoch": 0.1,
      "eval_avg_sts": 0.7808295955536356,
      "eval_sickr_spearman": 0.7253872118518216,
      "eval_stsb_spearman": 0.8362719792554496,
      "step": 750
    },
    {
      "epoch": 0.11,
      "eval_avg_sts": 0.7831225631840171,
      "eval_sickr_spearman": 0.7207539396826955,
      "eval_stsb_spearman": 0.8454911866853387,
      "step": 875
    }
  ],
  "max_steps": 7813,
  "num_train_epochs": 1,
  "total_flos": 0,
  "trial_name": null,
  "trial_params": null
}