trainer_state.json

{
  "best_metric": 0.03334970399737358,
  "best_model_checkpoint": "/home/jcanete/ft-data/all_results/pos/bertin_roberta_base/epochs_2_bs_16_lr_5e-5/checkpoint-1600",
  "epoch": 2.0,
  "global_step": 1790,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.22,
      "eval_accuracy": 0.983631166259442,
      "eval_f1": 0.9806329240291506,
      "eval_loss": 0.06692877411842346,
      "eval_precision": 0.9799676769289092,
      "eval_recall": 0.9812990749435576,
      "eval_runtime": 2.9547,
      "eval_samples_per_second": 559.789,
      "eval_steps_per_second": 35.198,
      "step": 200
    },
    {
      "epoch": 0.45,
      "eval_accuracy": 0.9877353518900014,
      "eval_f1": 0.9855665575341371,
      "eval_loss": 0.0487518236041069,
      "eval_precision": 0.9847605369716554,
      "eval_recall": 0.9863738986234041,
      "eval_runtime": 2.9562,
      "eval_samples_per_second": 559.497,
      "eval_steps_per_second": 35.18,
      "step": 400
    },
    {
      "epoch": 0.56,
      "learning_rate": 3.606145251396648e-05,
      "loss": 0.17,
      "step": 500
    },
    {
      "epoch": 0.67,
      "eval_accuracy": 0.9896197639694022,
      "eval_f1": 0.9876868689894844,
      "eval_loss": 0.0419081449508667,
      "eval_precision": 0.9873417721518988,
      "eval_recall": 0.9880322071487083,
      "eval_runtime": 2.9607,
      "eval_samples_per_second": 558.66,
      "eval_steps_per_second": 35.127,
      "step": 600
    },
    {
      "epoch": 0.89,
      "eval_accuracy": 0.9898912470655872,
      "eval_f1": 0.9883067213884146,
      "eval_loss": 0.039409052580595016,
      "eval_precision": 0.987902259841891,
      "eval_recall": 0.9887115142554594,
      "eval_runtime": 2.9548,
      "eval_samples_per_second": 559.772,
      "eval_steps_per_second": 35.197,
      "step": 800
    },
    {
      "epoch": 1.12,
      "learning_rate": 2.2094972067039108e-05,
      "loss": 0.0356,
      "step": 1000
    },
    {
      "epoch": 1.12,
      "eval_accuracy": 0.9899072166594803,
      "eval_f1": 0.9888547117804499,
      "eval_loss": 0.03804260864853859,
      "eval_precision": 0.9885585351730197,
      "eval_recall": 0.989151065912769,
      "eval_runtime": 2.9587,
      "eval_samples_per_second": 559.024,
      "eval_steps_per_second": 35.15,
      "step": 1000
    },
    {
      "epoch": 1.34,
      "eval_accuracy": 0.9901786997556652,
      "eval_f1": 0.9885585351730197,
      "eval_loss": 0.04127265512943268,
      "eval_precision": 0.9879667138951528,
      "eval_recall": 0.989151065912769,
      "eval_runtime": 3.0133,
      "eval_samples_per_second": 548.892,
      "eval_steps_per_second": 34.513,
      "step": 1200
    },
    {
      "epoch": 1.56,
      "eval_accuracy": 0.9912646321404047,
      "eval_f1": 0.9899833221814985,
      "eval_loss": 0.035211607813835144,
      "eval_precision": 0.9896769298350705,
      "eval_recall": 0.9902899042976164,
      "eval_runtime": 2.9833,
      "eval_samples_per_second": 554.427,
      "eval_steps_per_second": 34.861,
      "step": 1400
    },
    {
      "epoch": 1.68,
      "learning_rate": 8.128491620111732e-06,
      "loss": 0.0176,
      "step": 1500
    },
    {
      "epoch": 1.79,
      "eval_accuracy": 0.9915840240182692,
      "eval_f1": 0.9903327607558025,
      "eval_loss": 0.03334970399737358,
      "eval_precision": 0.9900361414508496,
      "eval_recall": 0.9906295578509919,
      "eval_runtime": 2.9696,
      "eval_samples_per_second": 556.981,
      "eval_steps_per_second": 35.022,
      "step": 1600
    },
    {
      "epoch": 2.0,
      "step": 1790,
      "total_flos": 1190583063933120.0,
      "train_loss": 0.06446368148206999,
      "train_runtime": 136.8646,
      "train_samples_per_second": 209.039,
      "train_steps_per_second": 13.079
    }
  ],
  "max_steps": 1790,
  "num_train_epochs": 2,
  "total_flos": 1190583063933120.0,
  "trial_name": null,
  "trial_params": null
}