trainer_state.json

{
  "best_metric": 0.02071884274482727,
  "best_model_checkpoint": "hateBERT-hate-offensive-normal-speech-lr-2e-05/checkpoint-765",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 765,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 1.997385620915033e-05,
      "loss": 1.2298,
      "step": 1
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.6000000000000003e-05,
      "loss": 0.6155,
      "step": 153
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.9801385062748699,
      "eval_Macro Precision": 0.9792887667887668,
      "eval_Macro Recall": 0.9811135158920202,
      "eval_Micro F1": 0.9804560260586319,
      "eval_Micro Precision": 0.9804560260586319,
      "eval_Micro Recall": 0.9804560260586319,
      "eval_Weighted F1": 0.9804745709987049,
      "eval_Weighted Precision": 0.9806165061865388,
      "eval_Weighted Recall": 0.9804560260586319,
      "eval_accuracy": 0.9804560260586319,
      "eval_loss": 0.08891552686691284,
      "eval_runtime": 0.6023,
      "eval_samples_per_second": 509.742,
      "eval_steps_per_second": 33.208,
      "step": 153
    },
    {
      "epoch": 2.0,
      "learning_rate": 1.2e-05,
      "loss": 0.0665,
      "step": 306
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.9864455652898299,
      "eval_Macro Precision": 0.9863515802256096,
      "eval_Macro Recall": 0.9865779967663372,
      "eval_Micro F1": 0.9869706840390879,
      "eval_Micro Precision": 0.9869706840390879,
      "eval_Micro Recall": 0.9869706840390879,
      "eval_Weighted F1": 0.9869723989743543,
      "eval_Weighted Precision": 0.9870087814541784,
      "eval_Weighted Recall": 0.9869706840390879,
      "eval_accuracy": 0.9869706840390879,
      "eval_loss": 0.03681240230798721,
      "eval_runtime": 0.5956,
      "eval_samples_per_second": 515.439,
      "eval_steps_per_second": 33.579,
      "step": 306
    },
    {
      "epoch": 3.0,
      "learning_rate": 8.000000000000001e-06,
      "loss": 0.0235,
      "step": 459
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.9900552839293133,
      "eval_Macro Precision": 0.9899417522368342,
      "eval_Macro Recall": 0.9903233151183971,
      "eval_Micro F1": 0.990228013029316,
      "eval_Micro Precision": 0.990228013029316,
      "eval_Micro Recall": 0.990228013029316,
      "eval_Weighted F1": 0.9902299179000704,
      "eval_Weighted Precision": 0.9903711923255897,
      "eval_Weighted Recall": 0.990228013029316,
      "eval_accuracy": 0.990228013029316,
      "eval_loss": 0.026355858892202377,
      "eval_runtime": 0.5997,
      "eval_samples_per_second": 511.883,
      "eval_steps_per_second": 33.347,
      "step": 459
    },
    {
      "epoch": 4.0,
      "learning_rate": 4.000000000000001e-06,
      "loss": 0.0182,
      "step": 612
    },
    {
      "epoch": 4.0,
      "eval_Macro F1": 0.9864518923933833,
      "eval_Macro Precision": 0.9863981943454503,
      "eval_Macro Recall": 0.9868510928961749,
      "eval_Micro F1": 0.9869706840390879,
      "eval_Micro Precision": 0.9869706840390879,
      "eval_Micro Recall": 0.9869706840390879,
      "eval_Weighted F1": 0.9869741119043762,
      "eval_Weighted Precision": 0.9872893357881319,
      "eval_Weighted Recall": 0.9869706840390879,
      "eval_accuracy": 0.9869706840390879,
      "eval_loss": 0.04140869900584221,
      "eval_runtime": 0.5943,
      "eval_samples_per_second": 516.568,
      "eval_steps_per_second": 33.653,
      "step": 612
    },
    {
      "epoch": 5.0,
      "learning_rate": 0.0,
      "loss": 0.012,
      "step": 765
    },
    {
      "epoch": 5.0,
      "eval_Macro F1": 0.9900552839293133,
      "eval_Macro Precision": 0.9899417522368342,
      "eval_Macro Recall": 0.9903233151183971,
      "eval_Micro F1": 0.990228013029316,
      "eval_Micro Precision": 0.990228013029316,
      "eval_Micro Recall": 0.990228013029316,
      "eval_Weighted F1": 0.9902299179000704,
      "eval_Weighted Precision": 0.9903711923255897,
      "eval_Weighted Recall": 0.990228013029316,
      "eval_accuracy": 0.990228013029316,
      "eval_loss": 0.02071884274482727,
      "eval_runtime": 0.6008,
      "eval_samples_per_second": 510.999,
      "eval_steps_per_second": 33.29,
      "step": 765
    },
    {
      "epoch": 5.0,
      "step": 765,
      "total_flos": 717066286251840.0,
      "train_loss": 0.1479466404011047,
      "train_runtime": 81.9262,
      "train_samples_per_second": 149.403,
      "train_steps_per_second": 9.338
    }
  ],
  "logging_steps": 500,
  "max_steps": 765,
  "num_train_epochs": 5,
  "save_steps": 500,
  "total_flos": 717066286251840.0,
  "trial_name": null,
  "trial_params": null
}