training_checkpoints/checkpoint-964/trainer_state.json

{
  "best_metric": 0.5371165387852578,
  "best_model_checkpoint": "legal-bert-lora/checkpoint-964",
  "epoch": 5.996889580093312,
  "eval_steps": 500,
  "global_step": 964,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.6374903175832688,
      "eval_f1_macro": 0.3394539856434376,
      "eval_f1_micro": 0.6374903175832688,
      "eval_f1_weighted": 0.5765523433851403,
      "eval_loss": 1.3127473592758179,
      "eval_macro_fpr": 0.040137717078130394,
      "eval_macro_sensitivity": 0.4017115530536667,
      "eval_macro_specificity": 0.9725650393599338,
      "eval_precision": 0.5466227876047401,
      "eval_precision_macro": 0.31141306924471374,
      "eval_recall": 0.6374903175832688,
      "eval_recall_macro": 0.4017115530536667,
      "eval_runtime": 44.4128,
      "eval_samples_per_second": 29.068,
      "eval_steps_per_second": 3.648,
      "eval_weighted_fpr": 0.0390325271059216,
      "eval_weighted_sensitivity": 0.6374903175832688,
      "eval_weighted_specificity": 0.950985272815738,
      "step": 160
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7273431448489543,
      "eval_f1_macro": 0.3969800602789428,
      "eval_f1_micro": 0.7273431448489543,
      "eval_f1_weighted": 0.6960627383550935,
      "eval_loss": 0.8160030841827393,
      "eval_macro_fpr": 0.026417996388603746,
      "eval_macro_sensitivity": 0.450135625445356,
      "eval_macro_specificity": 0.9796396430402219,
      "eval_precision": 0.6997929823446861,
      "eval_precision_macro": 0.39453164470926966,
      "eval_recall": 0.7273431448489543,
      "eval_recall_macro": 0.450135625445356,
      "eval_runtime": 34.3148,
      "eval_samples_per_second": 37.622,
      "eval_steps_per_second": 4.721,
      "eval_weighted_fpr": 0.026077937472218107,
      "eval_weighted_sensitivity": 0.7273431448489543,
      "eval_weighted_specificity": 0.9672515007543734,
      "step": 321
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7443841982958946,
      "eval_f1_macro": 0.4210575074216338,
      "eval_f1_micro": 0.7443841982958946,
      "eval_f1_weighted": 0.7223603623418946,
      "eval_loss": 0.7322217226028442,
      "eval_macro_fpr": 0.024451015698283406,
      "eval_macro_sensitivity": 0.46251578703813245,
      "eval_macro_specificity": 0.9809603839363172,
      "eval_precision": 0.711002444877192,
      "eval_precision_macro": 0.4030436676815596,
      "eval_recall": 0.7443841982958946,
      "eval_recall_macro": 0.46251578703813245,
      "eval_runtime": 32.8708,
      "eval_samples_per_second": 39.275,
      "eval_steps_per_second": 4.928,
      "eval_weighted_fpr": 0.023940800928612886,
      "eval_weighted_sensitivity": 0.7443841982958946,
      "eval_weighted_specificity": 0.9700215607488639,
      "step": 482
    },
    {
      "epoch": 3.11,
      "learning_rate": 3.958333333333333e-05,
      "loss": 1.2229,
      "step": 500
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.7536793183578622,
      "eval_f1_macro": 0.46141164864389705,
      "eval_f1_micro": 0.7536793183578622,
      "eval_f1_weighted": 0.7337325056665881,
      "eval_loss": 0.7034952640533447,
      "eval_macro_fpr": 0.023326126599014015,
      "eval_macro_sensitivity": 0.4865735295815666,
      "eval_macro_specificity": 0.9815978320880049,
      "eval_precision": 0.7340611054176746,
      "eval_precision_macro": 0.4982390488816621,
      "eval_recall": 0.7536793183578622,
      "eval_recall_macro": 0.4865735295815666,
      "eval_runtime": 33.282,
      "eval_samples_per_second": 38.79,
      "eval_steps_per_second": 4.868,
      "eval_weighted_fpr": 0.022812051649928264,
      "eval_weighted_sensitivity": 0.7536793183578622,
      "eval_weighted_specificity": 0.9702881629622103,
      "step": 643
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.7652982184353214,
      "eval_f1_macro": 0.4824234674423381,
      "eval_f1_micro": 0.7652982184353214,
      "eval_f1_weighted": 0.7415202198150465,
      "eval_loss": 0.6963701844215393,
      "eval_macro_fpr": 0.022125198961045537,
      "eval_macro_sensitivity": 0.5099998711032655,
      "eval_macro_specificity": 0.982427621700107,
      "eval_precision": 0.7403884847240659,
      "eval_precision_macro": 0.5015009503316971,
      "eval_recall": 0.7652982184353214,
      "eval_recall_macro": 0.5099998711032655,
      "eval_runtime": 29.8835,
      "eval_samples_per_second": 43.201,
      "eval_steps_per_second": 5.421,
      "eval_weighted_fpr": 0.02143615139724089,
      "eval_weighted_sensitivity": 0.7652982184353214,
      "eval_weighted_specificity": 0.9711161070662828,
      "step": 803
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.7807900852052673,
      "eval_f1_macro": 0.5371165387852578,
      "eval_f1_micro": 0.7807900852052673,
      "eval_f1_weighted": 0.7659242764458758,
      "eval_loss": 0.684776246547699,
      "eval_macro_fpr": 0.020129954279912522,
      "eval_macro_sensitivity": 0.5604626117585171,
      "eval_macro_specificity": 0.9836211797331506,
      "eval_precision": 0.7919069236021963,
      "eval_precision_macro": 0.5968972302554038,
      "eval_recall": 0.7807900852052673,
      "eval_recall_macro": 0.5604626117585171,
      "eval_runtime": 29.4753,
      "eval_samples_per_second": 43.799,
      "eval_steps_per_second": 5.496,
      "eval_weighted_fpr": 0.019659604029176797,
      "eval_weighted_sensitivity": 0.7807900852052673,
      "eval_weighted_specificity": 0.9735276107919913,
      "step": 964
    }
  ],
  "logging_steps": 500,
  "max_steps": 2400,
  "num_train_epochs": 15,
  "save_steps": 500,
  "total_flos": 8366917304180736.0,
  "trial_name": null,
  "trial_params": null
}