{
  "best_metric": 0.7374663072776281,
  "best_model_checkpoint": "models/ner-roberta-es-clinical-trials-ner/checkpoint-1506",
  "epoch": 3.0,
  "global_step": 1506,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "learning_rate": 1.3333333333333333e-05,
      "loss": 0.3951,
      "step": 502
    },
    {
      "epoch": 1.0,
      "eval_Body_Part_f1": 0.7081339712918661,
      "eval_Body_Part_number": 197,
      "eval_Body_Part_precision": 0.669683257918552,
      "eval_Body_Part_recall": 0.751269035532995,
      "eval_Disease_f1": 0.7342256214149141,
      "eval_Disease_number": 521,
      "eval_Disease_precision": 0.7314285714285714,
      "eval_Disease_recall": 0.7370441458733206,
      "eval_Family_Member_f1": 0.9166666666666666,
      "eval_Family_Member_number": 13,
      "eval_Family_Member_precision": 1.0,
      "eval_Family_Member_recall": 0.8461538461538461,
      "eval_Medication_f1": 0.6493506493506493,
      "eval_Medication_number": 37,
      "eval_Medication_precision": 0.625,
      "eval_Medication_recall": 0.6756756756756757,
      "eval_Procedure_f1": 0.608108108108108,
      "eval_Procedure_number": 134,
      "eval_Procedure_precision": 0.5555555555555556,
      "eval_Procedure_recall": 0.6716417910447762,
      "eval_loss": 0.2830584943294525,
      "eval_overall_accuracy": 0.9144236572118286,
      "eval_overall_f1": 0.7071466953250941,
      "eval_overall_precision": 0.6861313868613139,
      "eval_overall_recall": 0.729490022172949,
      "eval_runtime": 5.8394,
      "eval_samples_per_second": 76.378,
      "eval_steps_per_second": 4.795,
      "step": 502
    },
    {
      "epoch": 2.0,
      "learning_rate": 6.666666666666667e-06,
      "loss": 0.2123,
      "step": 1004
    },
    {
      "epoch": 2.0,
      "eval_Body_Part_f1": 0.7105882352941175,
      "eval_Body_Part_number": 197,
      "eval_Body_Part_precision": 0.6622807017543859,
      "eval_Body_Part_recall": 0.766497461928934,
      "eval_Disease_f1": 0.7360308285163777,
      "eval_Disease_number": 521,
      "eval_Disease_precision": 0.7388781431334622,
      "eval_Disease_recall": 0.7332053742802304,
      "eval_Family_Member_f1": 0.8148148148148148,
      "eval_Family_Member_number": 13,
      "eval_Family_Member_precision": 0.7857142857142857,
      "eval_Family_Member_recall": 0.8461538461538461,
      "eval_Medication_f1": 0.7012987012987014,
      "eval_Medication_number": 37,
      "eval_Medication_precision": 0.675,
      "eval_Medication_recall": 0.7297297297297297,
      "eval_Procedure_f1": 0.6825938566552902,
      "eval_Procedure_number": 134,
      "eval_Procedure_precision": 0.6289308176100629,
      "eval_Procedure_recall": 0.746268656716418,
      "eval_loss": 0.29119569063186646,
      "eval_overall_accuracy": 0.9132166566083283,
      "eval_overall_f1": 0.7215053763440861,
      "eval_overall_precision": 0.7004175365344467,
      "eval_overall_recall": 0.7439024390243902,
      "eval_runtime": 5.6078,
      "eval_samples_per_second": 79.532,
      "eval_steps_per_second": 4.993,
      "step": 1004
    },
    {
      "epoch": 3.0,
      "learning_rate": 0.0,
      "loss": 0.1686,
      "step": 1506
    },
    {
      "epoch": 3.0,
      "eval_Body_Part_f1": 0.7285714285714285,
      "eval_Body_Part_number": 197,
      "eval_Body_Part_precision": 0.6860986547085202,
      "eval_Body_Part_recall": 0.7766497461928934,
      "eval_Disease_f1": 0.7490421455938698,
      "eval_Disease_number": 521,
      "eval_Disease_precision": 0.7476099426386233,
      "eval_Disease_recall": 0.7504798464491362,
      "eval_Family_Member_f1": 0.8461538461538461,
      "eval_Family_Member_number": 13,
      "eval_Family_Member_precision": 0.8461538461538461,
      "eval_Family_Member_recall": 0.8461538461538461,
      "eval_Medication_f1": 0.8266666666666665,
      "eval_Medication_number": 37,
      "eval_Medication_precision": 0.8157894736842105,
      "eval_Medication_recall": 0.8378378378378378,
      "eval_Procedure_f1": 0.6758620689655171,
      "eval_Procedure_number": 134,
      "eval_Procedure_precision": 0.6282051282051282,
      "eval_Procedure_recall": 0.7313432835820896,
      "eval_loss": 0.28113484382629395,
      "eval_overall_accuracy": 0.9174411587205794,
      "eval_overall_f1": 0.7374663072776281,
      "eval_overall_precision": 0.7177334732423925,
      "eval_overall_recall": 0.7583148558758315,
      "eval_runtime": 5.4011,
      "eval_samples_per_second": 82.576,
      "eval_steps_per_second": 5.184,
      "step": 1506
    },
    {
      "epoch": 3.0,
      "step": 1506,
      "total_flos": 6291231284505600.0,
      "train_loss": 0.25864365819599205,
      "train_runtime": 931.1664,
      "train_samples_per_second": 25.855,
      "train_steps_per_second": 1.617
    }
  ],
  "max_steps": 1506,
  "num_train_epochs": 3,
  "total_flos": 6291231284505600.0,
  "trial_name": null,
  "trial_params": null
}