trainer_state.json

{
  "best_metric": 0.22135336697101593,
  "best_model_checkpoint": "./outputs/google/flan-t5-xl_06072024_22_39_08/checkpoint-130",
  "epoch": 5.0,
  "global_step": 325,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_f1": 0.5706292333786693,
      "eval_loss": 0.23904749751091003,
      "eval_precision": 0.6808422340944231,
      "eval_recall": 0.5369166453901002,
      "eval_runtime": 219.4425,
      "eval_samples_per_second": 4.161,
      "eval_steps_per_second": 0.036,
      "step": 65
    },
    {
      "epoch": 2.0,
      "eval_f1": 0.6301624612855913,
      "eval_loss": 0.22135336697101593,
      "eval_precision": 0.6489075109749957,
      "eval_recall": 0.6250635711770931,
      "eval_runtime": 225.8096,
      "eval_samples_per_second": 4.043,
      "eval_steps_per_second": 0.035,
      "step": 130
    },
    {
      "epoch": 3.0,
      "eval_f1": 0.621258077547598,
      "eval_loss": 0.2238980382680893,
      "eval_precision": 0.638580268076411,
      "eval_recall": 0.6117899756082343,
      "eval_runtime": 222.9642,
      "eval_samples_per_second": 4.095,
      "eval_steps_per_second": 0.036,
      "step": 195
    },
    {
      "epoch": 4.0,
      "eval_f1": 0.6213726430535729,
      "eval_loss": 0.2307296097278595,
      "eval_precision": 0.6537968022434235,
      "eval_recall": 0.5984045776817679,
      "eval_runtime": 222.6245,
      "eval_samples_per_second": 4.101,
      "eval_steps_per_second": 0.036,
      "step": 260
    },
    {
      "epoch": 5.0,
      "eval_f1": 0.60185801623338,
      "eval_loss": 0.2448112666606903,
      "eval_precision": 0.6105407586491245,
      "eval_recall": 0.6070169192821365,
      "eval_runtime": 230.757,
      "eval_samples_per_second": 3.957,
      "eval_steps_per_second": 0.035,
      "step": 325
    },
    {
      "epoch": 5.0,
      "step": 325,
      "total_flos": 17362999574528.0,
      "train_loss": 0.22278970571664664,
      "train_runtime": 6258.6879,
      "train_samples_per_second": 6.564,
      "train_steps_per_second": 0.052
    }
  ],
  "max_steps": 325,
  "num_train_epochs": 5,
  "total_flos": 17362999574528.0,
  "trial_name": null,
  "trial_params": null
}