trainer_state.json

{
  "best_metric": 0.9660714285714286,
  "best_model_checkpoint": "vit-large-patch16-224-dungeon-geo-morphs-0-4-30Nov24-010/checkpoint-30",
  "epoch": 23.90909090909091,
  "eval_steps": 10,
  "global_step": 60,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 3.909090909090909,
      "grad_norm": 24.15110969543457,
      "learning_rate": 1.888888888888889e-05,
      "loss": 1.4393,
      "step": 10
    },
    {
      "epoch": 3.909090909090909,
      "eval_accuracy": 0.7089285714285715,
      "eval_loss": 0.9143770337104797,
      "eval_runtime": 7.5272,
      "eval_samples_per_second": 74.397,
      "eval_steps_per_second": 9.3,
      "step": 10
    },
    {
      "epoch": 7.909090909090909,
      "grad_norm": 3.7548530101776123,
      "learning_rate": 1.5185185185185187e-05,
      "loss": 0.2882,
      "step": 20
    },
    {
      "epoch": 7.909090909090909,
      "eval_accuracy": 0.9267857142857143,
      "eval_loss": 0.33463338017463684,
      "eval_runtime": 8.6699,
      "eval_samples_per_second": 64.591,
      "eval_steps_per_second": 8.074,
      "step": 20
    },
    {
      "epoch": 11.909090909090908,
      "grad_norm": 0.41169193387031555,
      "learning_rate": 1.1481481481481482e-05,
      "loss": 0.0231,
      "step": 30
    },
    {
      "epoch": 11.909090909090908,
      "eval_accuracy": 0.9660714285714286,
      "eval_loss": 0.1486058533191681,
      "eval_runtime": 8.3231,
      "eval_samples_per_second": 67.283,
      "eval_steps_per_second": 8.41,
      "step": 30
    },
    {
      "epoch": 15.909090909090908,
      "grad_norm": 0.092295341193676,
      "learning_rate": 7.77777777777778e-06,
      "loss": 0.0024,
      "step": 40
    },
    {
      "epoch": 15.909090909090908,
      "eval_accuracy": 0.9607142857142857,
      "eval_loss": 0.11993860453367233,
      "eval_runtime": 7.9285,
      "eval_samples_per_second": 70.631,
      "eval_steps_per_second": 8.829,
      "step": 40
    },
    {
      "epoch": 19.90909090909091,
      "grad_norm": 0.05062829330563545,
      "learning_rate": 4.074074074074074e-06,
      "loss": 0.001,
      "step": 50
    },
    {
      "epoch": 19.90909090909091,
      "eval_accuracy": 0.9642857142857143,
      "eval_loss": 0.10705247521400452,
      "eval_runtime": 9.033,
      "eval_samples_per_second": 61.995,
      "eval_steps_per_second": 7.749,
      "step": 50
    },
    {
      "epoch": 23.90909090909091,
      "grad_norm": 0.03906424716114998,
      "learning_rate": 3.7037037037037036e-07,
      "loss": 0.0007,
      "step": 60
    },
    {
      "epoch": 23.90909090909091,
      "eval_accuracy": 0.9625,
      "eval_loss": 0.10578501969575882,
      "eval_runtime": 8.3191,
      "eval_samples_per_second": 67.315,
      "eval_steps_per_second": 8.414,
      "step": 60
    },
    {
      "epoch": 23.90909090909091,
      "step": 60,
      "total_flos": 5.259589576438579e+17,
      "train_loss": 0.2924559221913417,
      "train_runtime": 346.8638,
      "train_samples_per_second": 7.092,
      "train_steps_per_second": 0.173
    }
  ],
  "logging_steps": 10,
  "max_steps": 60,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 30,
  "save_steps": 10,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 5.259589576438579e+17,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}