trainer_state.json

{
  "best_metric": 0.8242545236101876,
  "best_model_checkpoint": "/content/result/unsup-simcse-roberta-large-semeval2016-restaurants",
  "epoch": 30.0,
  "global_step": 960,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 3.91,
      "eval_avg_sts": 0.7698010352632874,
      "eval_sickr_spearman": 0.7266049923976835,
      "eval_stsb_spearman": 0.8129970781288911,
      "step": 125
    },
    {
      "epoch": 7.81,
      "eval_avg_sts": 0.7792034906317966,
      "eval_sickr_spearman": 0.7348533254165491,
      "eval_stsb_spearman": 0.8235536558470441,
      "step": 250
    },
    {
      "epoch": 11.72,
      "eval_avg_sts": 0.7651583327248668,
      "eval_sickr_spearman": 0.7195221820863151,
      "eval_stsb_spearman": 0.8107944833634184,
      "step": 375
    },
    {
      "epoch": 15.62,
      "learning_rate": 2.3958333333333334e-05,
      "loss": 0.0019,
      "step": 500
    },
    {
      "epoch": 15.62,
      "eval_avg_sts": 0.7728941176212885,
      "eval_sickr_spearman": 0.7267549454963733,
      "eval_stsb_spearman": 0.8190332897462036,
      "step": 500
    },
    {
      "epoch": 19.53,
      "eval_avg_sts": 0.7803452705388955,
      "eval_sickr_spearman": 0.7396146485021092,
      "eval_stsb_spearman": 0.8210758925756817,
      "step": 625
    },
    {
      "epoch": 23.44,
      "eval_avg_sts": 0.7784229361678052,
      "eval_sickr_spearman": 0.7325913487254228,
      "eval_stsb_spearman": 0.8242545236101876,
      "step": 750
    },
    {
      "epoch": 27.34,
      "eval_avg_sts": 0.7792486791116137,
      "eval_sickr_spearman": 0.7348576962467801,
      "eval_stsb_spearman": 0.8236396619764472,
      "step": 875
    },
    {
      "epoch": 30.0,
      "step": 960,
      "train_runtime": 1919.7644,
      "train_samples_per_second": 0.5
    }
  ],
  "max_steps": 960,
  "num_train_epochs": 30,
  "total_flos": 8183872361005056,
  "trial_name": null,
  "trial_params": null
}