checkpoint-540/trainer_state.json

{
  "best_metric": 0.15804019570350647,
  "best_model_checkpoint": "headgear_image_detection/checkpoint-540",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 540,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.9943181818181818,
      "eval_loss": 0.16189882159233093,
      "eval_runtime": 19.8459,
      "eval_samples_per_second": 115.288,
      "eval_steps_per_second": 14.411,
      "step": 108
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.9934440559440559,
      "eval_loss": 0.16086503863334656,
      "eval_runtime": 19.4034,
      "eval_samples_per_second": 117.918,
      "eval_steps_per_second": 14.74,
      "step": 216
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.9943181818181818,
      "eval_loss": 0.1594339907169342,
      "eval_runtime": 19.5979,
      "eval_samples_per_second": 116.747,
      "eval_steps_per_second": 14.593,
      "step": 324
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.9934440559440559,
      "eval_loss": 0.1583821177482605,
      "eval_runtime": 19.5953,
      "eval_samples_per_second": 116.763,
      "eval_steps_per_second": 14.595,
      "step": 432
    },
    {
      "epoch": 4.63,
      "grad_norm": 0.35583406686782837,
      "learning_rate": 8.163265306122448e-08,
      "loss": 0.1381,
      "step": 500
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.9938811188811189,
      "eval_loss": 0.15804019570350647,
      "eval_runtime": 19.5607,
      "eval_samples_per_second": 116.969,
      "eval_steps_per_second": 14.621,
      "step": 540
    }
  ],
  "logging_steps": 500,
  "max_steps": 540,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "total_flos": 1.3299770701253837e+18,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}