{
  "best_metric": 0.996,
  "best_model_checkpoint": "AI_ImageClassification_SDXL/checkpoint-586",
  "epoch": 0.9995735607675906,
  "eval_steps": 500,
  "global_step": 586,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.02,
      "learning_rate": 2.840909090909091e-06,
      "loss": 1.2947,
      "step": 10
    },
    {
      "epoch": 0.03,
      "learning_rate": 5.681818181818182e-06,
      "loss": 0.689,
      "step": 20
    },
    {
      "epoch": 0.05,
      "learning_rate": 8.522727272727273e-06,
      "loss": 0.2576,
      "step": 30
    },
    {
      "epoch": 0.07,
      "learning_rate": 1.1363636363636365e-05,
      "loss": 0.23,
      "step": 40
    },
    {
      "epoch": 0.09,
      "learning_rate": 1.4204545454545456e-05,
      "loss": 0.1137,
      "step": 50
    },
    {
      "epoch": 0.1,
      "learning_rate": 1.7045454545454546e-05,
      "loss": 0.0554,
      "step": 60
    },
    {
      "epoch": 0.12,
      "learning_rate": 1.9886363636363638e-05,
      "loss": 0.059,
      "step": 70
    },
    {
      "epoch": 0.14,
      "learning_rate": 2.272727272727273e-05,
      "loss": 0.0583,
      "step": 80
    },
    {
      "epoch": 0.15,
      "learning_rate": 2.5568181818181817e-05,
      "loss": 0.0349,
      "step": 90
    },
    {
      "epoch": 0.17,
      "learning_rate": 2.8409090909090912e-05,
      "loss": 0.0864,
      "step": 100
    },
    {
      "epoch": 0.19,
      "learning_rate": 3.125e-05,
      "loss": 0.0145,
      "step": 110
    },
    {
      "epoch": 0.2,
      "learning_rate": 3.409090909090909e-05,
      "loss": 0.0358,
      "step": 120
    },
    {
      "epoch": 0.22,
      "learning_rate": 3.6931818181818184e-05,
      "loss": 0.0473,
      "step": 130
    },
    {
      "epoch": 0.24,
      "learning_rate": 3.9772727272727275e-05,
      "loss": 0.0253,
      "step": 140
    },
    {
      "epoch": 0.26,
      "learning_rate": 4.261363636363637e-05,
      "loss": 0.0574,
      "step": 150
    },
    {
      "epoch": 0.27,
      "learning_rate": 4.545454545454546e-05,
      "loss": 0.0403,
      "step": 160
    },
    {
      "epoch": 0.29,
      "learning_rate": 4.829545454545455e-05,
      "loss": 0.0481,
      "step": 170
    },
    {
      "epoch": 0.31,
      "learning_rate": 4.9873577749683945e-05,
      "loss": 0.0246,
      "step": 180
    },
    {
      "epoch": 0.32,
      "learning_rate": 4.955752212389381e-05,
      "loss": 0.0094,
      "step": 190
    },
    {
      "epoch": 0.34,
      "learning_rate": 4.924146649810367e-05,
      "loss": 0.0413,
      "step": 200
    },
    {
      "epoch": 0.36,
      "learning_rate": 4.892541087231353e-05,
      "loss": 0.0339,
      "step": 210
    },
    {
      "epoch": 0.38,
      "learning_rate": 4.860935524652339e-05,
      "loss": 0.0523,
      "step": 220
    },
    {
      "epoch": 0.39,
      "learning_rate": 4.829329962073325e-05,
      "loss": 0.0104,
      "step": 230
    },
    {
      "epoch": 0.41,
      "learning_rate": 4.797724399494311e-05,
      "loss": 0.0309,
      "step": 240
    },
    {
      "epoch": 0.43,
      "learning_rate": 4.7661188369152975e-05,
      "loss": 0.0248,
      "step": 250
    },
    {
      "epoch": 0.44,
      "learning_rate": 4.734513274336283e-05,
      "loss": 0.0306,
      "step": 260
    },
    {
      "epoch": 0.46,
      "learning_rate": 4.70290771175727e-05,
      "loss": 0.0162,
      "step": 270
    },
    {
      "epoch": 0.48,
      "learning_rate": 4.6713021491782554e-05,
      "loss": 0.016,
      "step": 280
    },
    {
      "epoch": 0.49,
      "learning_rate": 4.639696586599241e-05,
      "loss": 0.0103,
      "step": 290
    },
    {
      "epoch": 0.51,
      "learning_rate": 4.608091024020228e-05,
      "loss": 0.0196,
      "step": 300
    },
    {
      "epoch": 0.53,
      "learning_rate": 4.576485461441214e-05,
      "loss": 0.0264,
      "step": 310
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.5448798988622e-05,
      "loss": 0.0344,
      "step": 320
    },
    {
      "epoch": 0.56,
      "learning_rate": 4.5132743362831855e-05,
      "loss": 0.036,
      "step": 330
    },
    {
      "epoch": 0.58,
      "learning_rate": 4.4816687737041726e-05,
      "loss": 0.0082,
      "step": 340
    },
    {
      "epoch": 0.6,
      "learning_rate": 4.4500632111251584e-05,
      "loss": 0.0149,
      "step": 350
    },
    {
      "epoch": 0.61,
      "learning_rate": 4.418457648546144e-05,
      "loss": 0.0082,
      "step": 360
    },
    {
      "epoch": 0.63,
      "learning_rate": 4.3868520859671306e-05,
      "loss": 0.0451,
      "step": 370
    },
    {
      "epoch": 0.65,
      "learning_rate": 4.355246523388117e-05,
      "loss": 0.0055,
      "step": 380
    },
    {
      "epoch": 0.67,
      "learning_rate": 4.323640960809103e-05,
      "loss": 0.0087,
      "step": 390
    },
    {
      "epoch": 0.68,
      "learning_rate": 4.2920353982300885e-05,
      "loss": 0.0052,
      "step": 400
    },
    {
      "epoch": 0.7,
      "learning_rate": 4.260429835651075e-05,
      "loss": 0.0033,
      "step": 410
    },
    {
      "epoch": 0.72,
      "learning_rate": 4.2288242730720607e-05,
      "loss": 0.0148,
      "step": 420
    },
    {
      "epoch": 0.73,
      "learning_rate": 4.197218710493047e-05,
      "loss": 0.0134,
      "step": 430
    },
    {
      "epoch": 0.75,
      "learning_rate": 4.165613147914033e-05,
      "loss": 0.0125,
      "step": 440
    },
    {
      "epoch": 0.77,
      "learning_rate": 4.134007585335019e-05,
      "loss": 0.0531,
      "step": 450
    },
    {
      "epoch": 0.78,
      "learning_rate": 4.102402022756005e-05,
      "loss": 0.0059,
      "step": 460
    },
    {
      "epoch": 0.8,
      "learning_rate": 4.0707964601769914e-05,
      "loss": 0.0059,
      "step": 470
    },
    {
      "epoch": 0.82,
      "learning_rate": 4.039190897597978e-05,
      "loss": 0.0194,
      "step": 480
    },
    {
      "epoch": 0.84,
      "learning_rate": 4.0075853350189636e-05,
      "loss": 0.0111,
      "step": 490
    },
    {
      "epoch": 0.85,
      "learning_rate": 3.9759797724399494e-05,
      "loss": 0.0321,
      "step": 500
    },
    {
      "epoch": 0.87,
      "learning_rate": 3.944374209860936e-05,
      "loss": 0.0261,
      "step": 510
    },
    {
      "epoch": 0.89,
      "learning_rate": 3.912768647281922e-05,
      "loss": 0.0196,
      "step": 520
    },
    {
      "epoch": 0.9,
      "learning_rate": 3.881163084702908e-05,
      "loss": 0.0038,
      "step": 530
    },
    {
      "epoch": 0.92,
      "learning_rate": 3.849557522123894e-05,
      "loss": 0.0216,
      "step": 540
    },
    {
      "epoch": 0.94,
      "learning_rate": 3.81795195954488e-05,
      "loss": 0.0047,
      "step": 550
    },
    {
      "epoch": 0.96,
      "learning_rate": 3.7863463969658666e-05,
      "loss": 0.0095,
      "step": 560
    },
    {
      "epoch": 0.97,
      "learning_rate": 3.754740834386852e-05,
      "loss": 0.0036,
      "step": 570
    },
    {
      "epoch": 0.99,
      "learning_rate": 3.723135271807838e-05,
      "loss": 0.0025,
      "step": 580
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.996,
      "eval_loss": 0.015431854873895645,
      "eval_runtime": 44.7221,
      "eval_samples_per_second": 44.721,
      "eval_steps_per_second": 2.795,
      "step": 586
    }
  ],
  "logging_steps": 10,
  "max_steps": 1758,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "total_flos": 2.9390551738299924e+18,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}