trainer_state.json

{
  "best_metric": 2.6786022186279297,
  "best_model_checkpoint": "ClinicalT5-base-Amalq/shared_TaskA/checkpoint-453",
  "epoch": 3.0,
  "global_step": 453,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_gen_len": 19.0,
      "eval_loss": 2.707395315170288,
      "eval_rouge1": 26.1254,
      "eval_rouge2": 15.3623,
      "eval_rougeL": 26.0345,
      "eval_rougeLsum": 26.0439,
      "eval_runtime": 504.6047,
      "eval_samples_per_second": 0.198,
      "eval_steps_per_second": 0.026,
      "step": 151
    },
    {
      "epoch": 2.0,
      "eval_gen_len": 19.0,
      "eval_loss": 2.6932952404022217,
      "eval_rouge1": 25.1534,
      "eval_rouge2": 15.2987,
      "eval_rougeL": 25.0492,
      "eval_rougeLsum": 25.1223,
      "eval_runtime": 511.0491,
      "eval_samples_per_second": 0.196,
      "eval_steps_per_second": 0.025,
      "step": 302
    },
    {
      "epoch": 3.0,
      "eval_gen_len": 19.0,
      "eval_loss": 2.6786022186279297,
      "eval_rouge1": 26.6233,
      "eval_rouge2": 15.0278,
      "eval_rougeL": 26.42,
      "eval_rougeLsum": 26.6046,
      "eval_runtime": 497.4818,
      "eval_samples_per_second": 0.201,
      "eval_steps_per_second": 0.026,
      "step": 453
    }
  ],
  "max_steps": 755,
  "num_train_epochs": 5,
  "total_flos": 1.004475088982016e+16,
  "trial_name": null,
  "trial_params": null
}