{
  "best_metric": 0.9897161661867544,
  "best_model_checkpoint": "google/vit-base-patch16-224-in21k_dog_vs_cat_image_classification/checkpoint-3750",
  "epoch": 3.0,
  "global_step": 3750,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "learning_rate": 0.0001999466666666667,
      "loss": 0.7014,
      "step": 1
    },
    {
      "epoch": 0.4,
      "learning_rate": 0.00017333333333333334,
      "loss": 0.1057,
      "step": 500
    },
    {
      "epoch": 0.8,
      "learning_rate": 0.00014666666666666666,
      "loss": 0.0896,
      "step": 1000
    },
    {
      "epoch": 1.0,
      "eval_F1": 0.978256367778008,
      "eval_Precision": 0.9837567680133278,
      "eval_Recall": 0.9728171334431631,
      "eval_accuracy": 0.979,
      "eval_loss": 0.05895629897713661,
      "eval_runtime": 4548.903,
      "eval_samples_per_second": 1.099,
      "eval_steps_per_second": 0.137,
      "step": 1250
    },
    {
      "epoch": 1.2,
      "learning_rate": 0.00012,
      "loss": 0.0471,
      "step": 1500
    },
    {
      "epoch": 1.6,
      "learning_rate": 9.333333333333334e-05,
      "loss": 0.0322,
      "step": 2000
    },
    {
      "epoch": 2.0,
      "learning_rate": 6.666666666666667e-05,
      "loss": 0.0253,
      "step": 2500
    },
    {
      "epoch": 2.0,
      "eval_F1": 0.9836743128745609,
      "eval_Precision": 0.9871422646204894,
      "eval_Recall": 0.9802306425041186,
      "eval_accuracy": 0.9842,
      "eval_loss": 0.05425402894616127,
      "eval_runtime": 4485.3309,
      "eval_samples_per_second": 1.115,
      "eval_steps_per_second": 0.139,
      "step": 2500
    },
    {
      "epoch": 2.4,
      "learning_rate": 4e-05,
      "loss": 0.0064,
      "step": 3000
    },
    {
      "epoch": 2.8,
      "learning_rate": 1.3333333333333333e-05,
      "loss": 0.0066,
      "step": 3500
    },
    {
      "epoch": 3.0,
      "eval_F1": 0.9897161661867544,
      "eval_Precision": 0.9884963023829088,
      "eval_Recall": 0.9909390444810544,
      "eval_accuracy": 0.99,
      "eval_loss": 0.04044065997004509,
      "eval_runtime": 4387.6276,
      "eval_samples_per_second": 1.14,
      "eval_steps_per_second": 0.142,
      "step": 3750
    },
    {
      "epoch": 3.0,
      "step": 3750,
      "total_flos": 4.649054424937832e+18,
      "train_loss": 0.04228772549629212,
      "train_runtime": 172467.8398,
      "train_samples_per_second": 0.348,
      "train_steps_per_second": 0.022
    }
  ],
  "max_steps": 3750,
  "num_train_epochs": 3,
  "total_flos": 4.649054424937832e+18,
  "trial_name": null,
  "trial_params": null
}