{
  "best_metric": 0.7590687601515972,
  "best_model_checkpoint": "models/ner-roberta-es-clinical-trials-ner/checkpoint-1004",
  "epoch": 2.0,
  "global_step": 1004,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "learning_rate": 2.5e-05,
      "loss": 0.3329,
      "step": 502
    },
    {
      "epoch": 1.0,
      "eval_Body_Part_f1": 0.7268408551068883,
      "eval_Body_Part_number": 197,
      "eval_Body_Part_precision": 0.6830357142857143,
      "eval_Body_Part_recall": 0.7766497461928934,
      "eval_Disease_f1": 0.7687861271676301,
      "eval_Disease_number": 521,
      "eval_Disease_precision": 0.7717601547388782,
      "eval_Disease_recall": 0.7658349328214972,
      "eval_Family_Member_f1": 0.9230769230769231,
      "eval_Family_Member_number": 13,
      "eval_Family_Member_precision": 0.9230769230769231,
      "eval_Family_Member_recall": 0.9230769230769231,
      "eval_Medication_f1": 0.7792207792207791,
      "eval_Medication_number": 37,
      "eval_Medication_precision": 0.75,
      "eval_Medication_recall": 0.8108108108108109,
      "eval_Procedure_f1": 0.6689655172413793,
      "eval_Procedure_number": 134,
      "eval_Procedure_precision": 0.6217948717948718,
      "eval_Procedure_recall": 0.7238805970149254,
      "eval_loss": 0.2561482787132263,
      "eval_overall_accuracy": 0.9219070609535305,
      "eval_overall_f1": 0.7462203023758099,
      "eval_overall_precision": 0.7273684210526316,
      "eval_overall_recall": 0.7660753880266076,
      "eval_runtime": 7.4415,
      "eval_samples_per_second": 59.934,
      "eval_steps_per_second": 3.763,
      "step": 502
    },
    {
      "epoch": 2.0,
      "learning_rate": 0.0,
      "loss": 0.1699,
      "step": 1004
    },
    {
      "epoch": 2.0,
      "eval_Body_Part_f1": 0.7612293144208038,
      "eval_Body_Part_number": 197,
      "eval_Body_Part_precision": 0.7123893805309734,
      "eval_Body_Part_recall": 0.817258883248731,
      "eval_Disease_f1": 0.7704130643611912,
      "eval_Disease_number": 521,
      "eval_Disease_precision": 0.7711538461538462,
      "eval_Disease_recall": 0.7696737044145874,
      "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.8378378378378378,
      "eval_Medication_number": 37,
      "eval_Medication_precision": 0.8378378378378378,
      "eval_Medication_recall": 0.8378378378378378,
      "eval_Procedure_f1": 0.685512367491166,
      "eval_Procedure_number": 134,
      "eval_Procedure_precision": 0.6510067114093959,
      "eval_Procedure_recall": 0.7238805970149254,
      "eval_loss": 0.2660825252532959,
      "eval_overall_accuracy": 0.923838261919131,
      "eval_overall_f1": 0.7590687601515972,
      "eval_overall_precision": 0.7417989417989418,
      "eval_overall_recall": 0.7771618625277162,
      "eval_runtime": 6.1412,
      "eval_samples_per_second": 72.624,
      "eval_steps_per_second": 4.559,
      "step": 1004
    },
    {
      "epoch": 2.0,
      "step": 1004,
      "total_flos": 4194154189670400.0,
      "train_loss": 0.2513850238693663,
      "train_runtime": 633.278,
      "train_samples_per_second": 25.344,
      "train_steps_per_second": 1.585
    }
  ],
  "max_steps": 1004,
  "num_train_epochs": 2,
  "total_flos": 4194154189670400.0,
  "trial_name": null,
  "trial_params": null
}