trainer_state.json

{
  "best_metric": 1.0,
  "best_model_checkpoint": "swin-base-patch4-window7-224-finetuned-microsoft/swin-base-patch4-window7-224/checkpoint-5",
  "epoch": 5.714285714285714,
  "eval_steps": 500,
  "global_step": 10,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.5714285714285714,
      "eval_accuracy": 0.391304347826087,
      "eval_loss": 0.8119199275970459,
      "eval_runtime": 0.2068,
      "eval_samples_per_second": 111.241,
      "eval_steps_per_second": 4.837,
      "step": 1
    },
    {
      "epoch": 1.7142857142857144,
      "eval_accuracy": 0.5217391304347826,
      "eval_loss": 0.6543549299240112,
      "eval_runtime": 0.2025,
      "eval_samples_per_second": 113.593,
      "eval_steps_per_second": 4.939,
      "step": 3
    },
    {
      "epoch": 2.857142857142857,
      "eval_accuracy": 1.0,
      "eval_loss": 0.3809329867362976,
      "eval_runtime": 0.2093,
      "eval_samples_per_second": 109.896,
      "eval_steps_per_second": 4.778,
      "step": 5
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 1.0,
      "eval_loss": 0.23518137633800507,
      "eval_runtime": 0.2087,
      "eval_samples_per_second": 110.206,
      "eval_steps_per_second": 4.792,
      "step": 7
    },
    {
      "epoch": 4.571428571428571,
      "eval_accuracy": 1.0,
      "eval_loss": 0.1875050812959671,
      "eval_runtime": 0.2057,
      "eval_samples_per_second": 111.824,
      "eval_steps_per_second": 4.862,
      "step": 8
    },
    {
      "epoch": 5.714285714285714,
      "grad_norm": 3.835265874862671,
      "learning_rate": 0.0,
      "loss": 0.4988,
      "step": 10
    },
    {
      "epoch": 5.714285714285714,
      "eval_accuracy": 1.0,
      "eval_loss": 0.14810918271541595,
      "eval_runtime": 0.2143,
      "eval_samples_per_second": 107.346,
      "eval_steps_per_second": 4.667,
      "step": 10
    },
    {
      "epoch": 5.714285714285714,
      "step": 10,
      "total_flos": 9.323121119735808e+16,
      "train_loss": 0.498783016204834,
      "train_runtime": 51.3045,
      "train_samples_per_second": 40.152,
      "train_steps_per_second": 0.195
    }
  ],
  "logging_steps": 10,
  "max_steps": 10,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 9.323121119735808e+16,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}