{
  "best_metric": 0.9836132732486685,
  "best_model_checkpoint": "./output/neuralmind/bert-large-portuguese-cased-finetune-wiki_section_disease/seed42-seq512-lr5e-05-epoch5-bs4-ts1.0-tssp1.0-cl0.5-2024-05-30_21:01:41/checkpoint-8040",
  "epoch": 5.0,
  "global_step": 13400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "learning_rate": 4e-05,
      "loss": 0.5631,
      "step": 2680
    },
    {
      "epoch": 1.0,
      "eval_EOP_f1": 0.9767820773930753,
      "eval_EOP_number": 2446,
      "eval_EOP_precision": 0.9732142857142857,
      "eval_EOP_recall": 0.9803761242845462,
      "eval_da_EOP_f1": 0.9276059988890947,
      "eval_da_EOP_number": 2736,
      "eval_da_EOP_precision": 0.9399624765478424,
      "eval_da_EOP_recall": 0.9155701754385965,
      "eval_da_overall_accuracy": 0.9829540500479553,
      "eval_da_overall_f1": 0.9276059988890947,
      "eval_da_overall_precision": 0.9399624765478424,
      "eval_da_overall_recall": 0.9155701754385965,
      "eval_loss": NaN,
      "eval_overall_accuracy": 0.9944503943140882,
      "eval_overall_f1": 0.9767820773930753,
      "eval_overall_precision": 0.9732142857142857,
      "eval_overall_recall": 0.9803761242845462,
      "eval_runtime": 350.1554,
      "eval_samples_per_second": 4.361,
      "eval_steps_per_second": 1.091,
      "step": 2680
    },
    {
      "epoch": 2.0,
      "learning_rate": 3e-05,
      "loss": 0.4083,
      "step": 5360
    },
    {
      "epoch": 2.0,
      "eval_EOP_f1": 0.9791751735402207,
      "eval_EOP_number": 2446,
      "eval_EOP_precision": 0.9779771615008157,
      "eval_EOP_recall": 0.9803761242845462,
      "eval_da_EOP_f1": 0.9395747855277882,
      "eval_da_EOP_number": 2736,
      "eval_da_EOP_precision": 0.9592536176694593,
      "eval_da_EOP_recall": 0.9206871345029239,
      "eval_da_overall_accuracy": 0.985874967303165,
      "eval_da_overall_f1": 0.9395747855277882,
      "eval_da_overall_precision": 0.9592536176694593,
      "eval_da_overall_recall": 0.9206871345029239,
      "eval_loss": NaN,
      "eval_overall_accuracy": 0.9950345633336579,
      "eval_overall_f1": 0.9791751735402207,
      "eval_overall_precision": 0.9779771615008157,
      "eval_overall_recall": 0.9803761242845462,
      "eval_runtime": 351.0792,
      "eval_samples_per_second": 4.349,
      "eval_steps_per_second": 1.088,
      "step": 5360
    },
    {
      "epoch": 3.0,
      "learning_rate": 2e-05,
      "loss": 0.3595,
      "step": 8040
    },
    {
      "epoch": 3.0,
      "eval_EOP_f1": 0.9836132732486685,
      "eval_EOP_number": 2446,
      "eval_EOP_precision": 0.985632183908046,
      "eval_EOP_recall": 0.9816026165167621,
      "eval_da_EOP_f1": 0.9373584905660378,
      "eval_da_EOP_number": 2736,
      "eval_da_EOP_precision": 0.968798751950078,
      "eval_da_EOP_recall": 0.9078947368421053,
      "eval_da_overall_accuracy": 0.985526201063737,
      "eval_da_overall_f1": 0.9373584905660378,
      "eval_da_overall_precision": 0.968798751950078,
      "eval_da_overall_recall": 0.9078947368421053,
      "eval_loss": NaN,
      "eval_overall_accuracy": 0.9961055398695355,
      "eval_overall_f1": 0.9836132732486685,
      "eval_overall_precision": 0.985632183908046,
      "eval_overall_recall": 0.9816026165167621,
      "eval_runtime": 350.0594,
      "eval_samples_per_second": 4.362,
      "eval_steps_per_second": 1.091,
      "step": 8040
    },
    {
      "epoch": 4.0,
      "learning_rate": 1e-05,
      "loss": 0.315,
      "step": 10720
    },
    {
      "epoch": 4.0,
      "eval_EOP_f1": 0.981170691772411,
      "eval_EOP_number": 2446,
      "eval_EOP_precision": 0.9823770491803279,
      "eval_EOP_recall": 0.9799672935404743,
      "eval_da_EOP_f1": 0.9371944339977436,
      "eval_da_EOP_number": 2736,
      "eval_da_EOP_precision": 0.965143299767622,
      "eval_da_EOP_recall": 0.9108187134502924,
      "eval_da_overall_accuracy": 0.98543900950388,
      "eval_da_overall_f1": 0.9371944339977436,
      "eval_da_overall_precision": 0.965143299767622,
      "eval_da_overall_recall": 0.9108187134502924,
      "eval_loss": NaN,
      "eval_overall_accuracy": 0.995521370849966,
      "eval_overall_f1": 0.981170691772411,
      "eval_overall_precision": 0.9823770491803279,
      "eval_overall_recall": 0.9799672935404743,
      "eval_runtime": 351.0283,
      "eval_samples_per_second": 4.35,
      "eval_steps_per_second": 1.088,
      "step": 10720
    },
    {
      "epoch": 5.0,
      "learning_rate": 0.0,
      "loss": 0.2695,
      "step": 13400
    },
    {
      "epoch": 5.0,
      "eval_EOP_f1": 0.9823553549446039,
      "eval_EOP_number": 2446,
      "eval_EOP_precision": 0.985996705107084,
      "eval_EOP_recall": 0.9787408013082584,
      "eval_da_EOP_f1": 0.9391206313416008,
      "eval_da_EOP_number": 2736,
      "eval_da_EOP_precision": 0.9663573085846868,
      "eval_da_EOP_recall": 0.9133771929824561,
      "eval_da_overall_accuracy": 0.985874967303165,
      "eval_da_overall_f1": 0.9391206313416008,
      "eval_da_overall_precision": 0.9663573085846868,
      "eval_da_overall_recall": 0.9133771929824561,
      "eval_loss": NaN,
      "eval_overall_accuracy": 0.9958134553597507,
      "eval_overall_f1": 0.9823553549446039,
      "eval_overall_precision": 0.985996705107084,
      "eval_overall_recall": 0.9787408013082584,
      "eval_runtime": 351.0391,
      "eval_samples_per_second": 4.35,
      "eval_steps_per_second": 1.088,
      "step": 13400
    },
    {
      "epoch": 5.0,
      "step": 13400,
      "total_flos": 9.99040548028416e+16,
      "train_loss": 0.3830805900915345,
      "train_runtime": 38367.8373,
      "train_samples_per_second": 1.397,
      "train_steps_per_second": 0.349
    }
  ],
  "max_steps": 13400,
  "num_train_epochs": 5,
  "total_flos": 9.99040548028416e+16,
  "trial_name": null,
  "trial_params": null
}