trainer_state.json

{
  "best_metric": 0.8439716312056738,
  "best_model_checkpoint": "swinv2-large-patch4-window12-192-22k-augmented\\checkpoint-20",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 25,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_accuracy": 0.7446808510638298,
      "eval_loss": 0.834225058555603,
      "eval_runtime": 25.3317,
      "eval_samples_per_second": 5.566,
      "eval_steps_per_second": 0.197,
      "step": 5
    },
    {
      "epoch": 2.0,
      "learning_rate": 6.818181818181818e-05,
      "loss": 1.3135,
      "step": 10
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7872340425531915,
      "eval_loss": 0.6567438244819641,
      "eval_runtime": 23.9812,
      "eval_samples_per_second": 5.88,
      "eval_steps_per_second": 0.208,
      "step": 10
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.8226950354609929,
      "eval_loss": 0.48486486077308655,
      "eval_runtime": 24.8122,
      "eval_samples_per_second": 5.683,
      "eval_steps_per_second": 0.202,
      "step": 15
    },
    {
      "epoch": 4.0,
      "learning_rate": 2.272727272727273e-05,
      "loss": 0.4762,
      "step": 20
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.8439716312056738,
      "eval_loss": 0.48768967390060425,
      "eval_runtime": 24.0078,
      "eval_samples_per_second": 5.873,
      "eval_steps_per_second": 0.208,
      "step": 20
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.8439716312056738,
      "eval_loss": 0.4229961931705475,
      "eval_runtime": 24.9551,
      "eval_samples_per_second": 5.65,
      "eval_steps_per_second": 0.2,
      "step": 25
    },
    {
      "epoch": 5.0,
      "step": 25,
      "total_flos": 8.206071072340378e+17,
      "train_loss": 0.7788310098648071,
      "train_runtime": 2728.3378,
      "train_samples_per_second": 2.322,
      "train_steps_per_second": 0.009
    }
  ],
  "logging_steps": 10,
  "max_steps": 25,
  "num_train_epochs": 5,
  "save_steps": 500,
  "total_flos": 8.206071072340378e+17,
  "trial_name": null,
  "trial_params": null
}