trainer_state.json

{
  "best_metric": 0.8373493975903614,
  "best_model_checkpoint": "vit-base-aihub_model-v2/checkpoint-9",
  "epoch": 5.0,
  "global_step": 15,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.6746987951807228,
      "eval_f1": 0.5943616223134295,
      "eval_loss": 1.6293878555297852,
      "eval_precision": 0.6434404116256432,
      "eval_recall": 0.6238199585809991,
      "eval_runtime": 6.3604,
      "eval_samples_per_second": 26.099,
      "eval_steps_per_second": 0.314,
      "step": 3
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7530120481927711,
      "eval_f1": 0.6875213018670031,
      "eval_loss": 1.4494768381118774,
      "eval_precision": 0.7775814126820318,
      "eval_recall": 0.701835255443846,
      "eval_runtime": 5.6618,
      "eval_samples_per_second": 29.319,
      "eval_steps_per_second": 0.353,
      "step": 6
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.8373493975903614,
      "eval_f1": 0.8021719960125756,
      "eval_loss": 1.3162782192230225,
      "eval_precision": 0.8562555276264954,
      "eval_recall": 0.7993336310123969,
      "eval_runtime": 5.0768,
      "eval_samples_per_second": 32.698,
      "eval_steps_per_second": 0.394,
      "step": 9
    },
    {
      "epoch": 3.33,
      "learning_rate": 1.923076923076923e-05,
      "loss": 1.5378,
      "step": 10
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.8373493975903614,
      "eval_f1": 0.8035411001594577,
      "eval_loss": 1.232733964920044,
      "eval_precision": 0.8735820318716042,
      "eval_recall": 0.7993336310123969,
      "eval_runtime": 5.2569,
      "eval_samples_per_second": 31.578,
      "eval_steps_per_second": 0.38,
      "step": 12
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.8373493975903614,
      "eval_f1": 0.8036849674785987,
      "eval_loss": 1.1992783546447754,
      "eval_precision": 0.8745971666076694,
      "eval_recall": 0.7993336310123969,
      "eval_runtime": 5.464,
      "eval_samples_per_second": 30.381,
      "eval_steps_per_second": 0.366,
      "step": 15
    },
    {
      "epoch": 5.0,
      "step": 15,
      "total_flos": 5.757861244520448e+17,
      "train_loss": 1.4401942888895671,
      "train_runtime": 84.6609,
      "train_samples_per_second": 87.762,
      "train_steps_per_second": 0.177
    }
  ],
  "max_steps": 15,
  "num_train_epochs": 5,
  "total_flos": 5.757861244520448e+17,
  "trial_name": null,
  "trial_params": null
}