training_checkpoints/checkpoint-321/trainer_state.json

{
  "best_metric": 0.3947617401663212,
  "best_model_checkpoint": "bart-base-lora/checkpoint-321",
  "epoch": 1.9968895800933126,
  "eval_steps": 500,
  "global_step": 321,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.611154144074361,
      "eval_f1_macro": 0.28705545756004525,
      "eval_f1_micro": 0.611154144074361,
      "eval_f1_weighted": 0.5574695078229231,
      "eval_loss": 1.3204560279846191,
      "eval_macro_fpr": 0.046390243375546196,
      "eval_macro_sensitivity": 0.302378554842108,
      "eval_macro_specificity": 0.9691815961629444,
      "eval_precision": 0.5321554731456403,
      "eval_precision_macro": 0.2887439395071009,
      "eval_recall": 0.611154144074361,
      "eval_recall_macro": 0.302378554842108,
      "eval_runtime": 45.1674,
      "eval_samples_per_second": 28.583,
      "eval_steps_per_second": 3.587,
      "eval_weighted_fpr": 0.04347073086248701,
      "eval_weighted_sensitivity": 0.611154144074361,
      "eval_weighted_specificity": 0.9265697983698051,
      "step": 160
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.6994577846630519,
      "eval_f1_macro": 0.3947617401663212,
      "eval_f1_micro": 0.6994577846630519,
      "eval_f1_weighted": 0.6807748226338182,
      "eval_loss": 0.887535810470581,
      "eval_macro_fpr": 0.03062685085400977,
      "eval_macro_sensitivity": 0.4253981018601158,
      "eval_macro_specificity": 0.977359054380045,
      "eval_precision": 0.6728416067370143,
      "eval_precision_macro": 0.3822291983692299,
      "eval_recall": 0.6994577846630519,
      "eval_recall_macro": 0.4253981018601158,
      "eval_runtime": 44.0045,
      "eval_samples_per_second": 29.338,
      "eval_steps_per_second": 3.681,
      "eval_weighted_fpr": 0.02977743668457406,
      "eval_weighted_sensitivity": 0.6994577846630519,
      "eval_weighted_specificity": 0.9609280310376243,
      "step": 321
    }
  ],
  "logging_steps": 500,
  "max_steps": 2400,
  "num_train_epochs": 15,
  "save_steps": 500,
  "total_flos": 3255412271763456.0,
  "trial_name": null,
  "trial_params": null
}