training_checkpoints/checkpoint-803/trainer_state.json

{
  "best_metric": 0.6353946839596228,
  "best_model_checkpoint": "bart-base-lora/checkpoint-803",
  "epoch": 4.995334370139969,
  "eval_steps": 500,
  "global_step": 803,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.715724244771495,
      "eval_f1_macro": 0.3957532686340496,
      "eval_f1_micro": 0.715724244771495,
      "eval_f1_weighted": 0.6834645556416842,
      "eval_loss": 0.9525313973426819,
      "eval_macro_fpr": 0.028501626692807605,
      "eval_macro_sensitivity": 0.4415906678049911,
      "eval_macro_specificity": 0.9786637894661117,
      "eval_precision": 0.6787596592263883,
      "eval_precision_macro": 0.38751339695744513,
      "eval_recall": 0.715724244771495,
      "eval_recall_macro": 0.4415906678049911,
      "eval_runtime": 120.9594,
      "eval_samples_per_second": 10.673,
      "eval_steps_per_second": 1.339,
      "eval_weighted_fpr": 0.027587762158911525,
      "eval_weighted_sensitivity": 0.715724244771495,
      "eval_weighted_specificity": 0.9642325972201805,
      "step": 160
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7412858249419055,
      "eval_f1_macro": 0.43370046411892155,
      "eval_f1_micro": 0.7412858249419055,
      "eval_f1_weighted": 0.7231329193382661,
      "eval_loss": 0.7733433842658997,
      "eval_macro_fpr": 0.025167370687133015,
      "eval_macro_sensitivity": 0.4686708989603177,
      "eval_macro_specificity": 0.9805374797463638,
      "eval_precision": 0.729625861926521,
      "eval_precision_macro": 0.4491323101141271,
      "eval_recall": 0.7412858249419055,
      "eval_recall_macro": 0.4686708989603177,
      "eval_runtime": 119.8649,
      "eval_samples_per_second": 10.77,
      "eval_steps_per_second": 1.352,
      "eval_weighted_fpr": 0.024322749781532187,
      "eval_weighted_sensitivity": 0.7412858249419055,
      "eval_weighted_specificity": 0.9667763712535515,
      "step": 321
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7738187451587917,
      "eval_f1_macro": 0.5270593942960143,
      "eval_f1_micro": 0.7738187451587917,
      "eval_f1_weighted": 0.7610767338978833,
      "eval_loss": 0.7104586958885193,
      "eval_macro_fpr": 0.021211805484311928,
      "eval_macro_sensitivity": 0.5407847836684627,
      "eval_macro_specificity": 0.9830897092527341,
      "eval_precision": 0.7630858215778352,
      "eval_precision_macro": 0.5565386466479504,
      "eval_recall": 0.7738187451587917,
      "eval_recall_macro": 0.5407847836684627,
      "eval_runtime": 119.5419,
      "eval_samples_per_second": 10.8,
      "eval_steps_per_second": 1.355,
      "eval_weighted_fpr": 0.020451043563524302,
      "eval_weighted_sensitivity": 0.7738187451587917,
      "eval_weighted_specificity": 0.9725268936322198,
      "step": 482
    },
    {
      "epoch": 3.11,
      "learning_rate": 3.958333333333333e-05,
      "loss": 1.08,
      "step": 500
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.7575522850503486,
      "eval_f1_macro": 0.5497163954183316,
      "eval_f1_micro": 0.7575522850503486,
      "eval_f1_weighted": 0.7437674055775699,
      "eval_loss": 0.7538753151893616,
      "eval_macro_fpr": 0.023393901824967867,
      "eval_macro_sensitivity": 0.5613408412198008,
      "eval_macro_specificity": 0.9817091819688993,
      "eval_precision": 0.7583906478192325,
      "eval_precision_macro": 0.5791225123984725,
      "eval_recall": 0.7575522850503486,
      "eval_recall_macro": 0.5613408412198008,
      "eval_runtime": 120.3095,
      "eval_samples_per_second": 10.731,
      "eval_steps_per_second": 1.347,
      "eval_weighted_fpr": 0.022349161013923598,
      "eval_weighted_sensitivity": 0.7575522850503486,
      "eval_weighted_specificity": 0.9680854444831427,
      "step": 643
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.7831138652207591,
      "eval_f1_macro": 0.6353946839596228,
      "eval_f1_micro": 0.783113865220759,
      "eval_f1_weighted": 0.770250750407997,
      "eval_loss": 0.6978110074996948,
      "eval_macro_fpr": 0.020328728105215688,
      "eval_macro_sensitivity": 0.6491872351233146,
      "eval_macro_specificity": 0.983610303127039,
      "eval_precision": 0.790031827496624,
      "eval_precision_macro": 0.7409525878258343,
      "eval_recall": 0.7831138652207591,
      "eval_recall_macro": 0.6491872351233146,
      "eval_runtime": 119.9411,
      "eval_samples_per_second": 10.764,
      "eval_steps_per_second": 1.351,
      "eval_weighted_fpr": 0.019398642095053348,
      "eval_weighted_sensitivity": 0.7831138652207591,
      "eval_weighted_specificity": 0.9710406816848265,
      "step": 803
    }
  ],
  "logging_steps": 500,
  "max_steps": 2400,
  "num_train_epochs": 15,
  "save_steps": 500,
  "total_flos": 2.860711868077056e+16,
  "trial_name": null,
  "trial_params": null
}