{
  "best_metric": 0.8444921415218445,
  "best_model_checkpoint": "/gscratch/xlab/hallisky/style/models/multilabel/03-21-2023_21:54:15/checkpoint-4272",
  "epoch": 4.0,
  "global_step": 4272,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.25,
      "learning_rate": 5e-06,
      "loss": 0.3538,
      "step": 267
    },
    {
      "epoch": 0.25,
      "eval_acc": 0.6889150453506889,
      "eval_loss": 0.15054994821548462,
      "eval_runtime": 41.4893,
      "eval_samples_per_second": 348.114,
      "eval_steps_per_second": 1.374,
      "step": 267
    },
    {
      "epoch": 0.5,
      "learning_rate": 1e-05,
      "loss": 0.1314,
      "step": 534
    },
    {
      "epoch": 0.5,
      "eval_acc": 0.7778162431627779,
      "eval_loss": 0.10427295416593552,
      "eval_runtime": 41.0952,
      "eval_samples_per_second": 351.453,
      "eval_steps_per_second": 1.387,
      "step": 534
    },
    {
      "epoch": 0.75,
      "learning_rate": 9.444444444444445e-06,
      "loss": 0.1031,
      "step": 801
    },
    {
      "epoch": 0.75,
      "eval_acc": 0.8037803780378038,
      "eval_loss": 0.0903468206524849,
      "eval_runtime": 41.1937,
      "eval_samples_per_second": 350.612,
      "eval_steps_per_second": 1.384,
      "step": 801
    },
    {
      "epoch": 1.0,
      "learning_rate": 8.888888888888888e-06,
      "loss": 0.0916,
      "step": 1068
    },
    {
      "epoch": 1.0,
      "eval_acc": 0.8165893512428166,
      "eval_loss": 0.08489712327718735,
      "eval_runtime": 41.1826,
      "eval_samples_per_second": 350.706,
      "eval_steps_per_second": 1.384,
      "step": 1068
    },
    {
      "epoch": 1.25,
      "learning_rate": 8.333333333333334e-06,
      "loss": 0.0807,
      "step": 1335
    },
    {
      "epoch": 1.25,
      "eval_acc": 0.8217821782178217,
      "eval_loss": 0.0828377977013588,
      "eval_runtime": 41.1771,
      "eval_samples_per_second": 350.754,
      "eval_steps_per_second": 1.384,
      "step": 1335
    },
    {
      "epoch": 1.5,
      "learning_rate": 7.77777777777778e-06,
      "loss": 0.0777,
      "step": 1602
    },
    {
      "epoch": 1.5,
      "eval_acc": 0.8273904313508273,
      "eval_loss": 0.07905308157205582,
      "eval_runtime": 41.1692,
      "eval_samples_per_second": 350.821,
      "eval_steps_per_second": 1.385,
      "step": 1602
    },
    {
      "epoch": 1.75,
      "learning_rate": 7.222222222222223e-06,
      "loss": 0.0748,
      "step": 1869
    },
    {
      "epoch": 1.75,
      "eval_acc": 0.8333448729488333,
      "eval_loss": 0.07718278467655182,
      "eval_runtime": 41.1712,
      "eval_samples_per_second": 350.804,
      "eval_steps_per_second": 1.384,
      "step": 1869
    },
    {
      "epoch": 2.0,
      "learning_rate": 6.666666666666667e-06,
      "loss": 0.0732,
      "step": 2136
    },
    {
      "epoch": 2.0,
      "eval_acc": 0.8379145606868379,
      "eval_loss": 0.07422468066215515,
      "eval_runtime": 41.1823,
      "eval_samples_per_second": 350.709,
      "eval_steps_per_second": 1.384,
      "step": 2136
    },
    {
      "epoch": 2.25,
      "learning_rate": 6.111111111111112e-06,
      "loss": 0.0635,
      "step": 2403
    },
    {
      "epoch": 2.25,
      "eval_acc": 0.8334141106418335,
      "eval_loss": 0.07540296018123627,
      "eval_runtime": 41.1954,
      "eval_samples_per_second": 350.597,
      "eval_steps_per_second": 1.384,
      "step": 2403
    },
    {
      "epoch": 2.5,
      "learning_rate": 5.555555555555557e-06,
      "loss": 0.0627,
      "step": 2670
    },
    {
      "epoch": 2.5,
      "eval_acc": 0.8368759952918369,
      "eval_loss": 0.07453104108572006,
      "eval_runtime": 41.204,
      "eval_samples_per_second": 350.524,
      "eval_steps_per_second": 1.383,
      "step": 2670
    },
    {
      "epoch": 2.75,
      "learning_rate": 5e-06,
      "loss": 0.0621,
      "step": 2937
    },
    {
      "epoch": 2.75,
      "eval_acc": 0.8378453229938378,
      "eval_loss": 0.07528460025787354,
      "eval_runtime": 41.1464,
      "eval_samples_per_second": 351.015,
      "eval_steps_per_second": 1.385,
      "step": 2937
    },
    {
      "epoch": 3.0,
      "learning_rate": 4.444444444444444e-06,
      "loss": 0.0616,
      "step": 3204
    },
    {
      "epoch": 3.0,
      "eval_acc": 0.8421380599598421,
      "eval_loss": 0.07257214188575745,
      "eval_runtime": 41.1192,
      "eval_samples_per_second": 351.247,
      "eval_steps_per_second": 1.386,
      "step": 3204
    },
    {
      "epoch": 3.25,
      "learning_rate": 3.88888888888889e-06,
      "loss": 0.0544,
      "step": 3471
    },
    {
      "epoch": 3.25,
      "eval_acc": 0.844007477670844,
      "eval_loss": 0.07361570745706558,
      "eval_runtime": 41.4565,
      "eval_samples_per_second": 348.389,
      "eval_steps_per_second": 1.375,
      "step": 3471
    },
    {
      "epoch": 3.5,
      "learning_rate": 3.3333333333333333e-06,
      "loss": 0.0539,
      "step": 3738
    },
    {
      "epoch": 3.5,
      "eval_acc": 0.8441459530568441,
      "eval_loss": 0.07325488328933716,
      "eval_runtime": 41.1118,
      "eval_samples_per_second": 351.311,
      "eval_steps_per_second": 1.386,
      "step": 3738
    },
    {
      "epoch": 3.75,
      "learning_rate": 2.7777777777777783e-06,
      "loss": 0.0538,
      "step": 4005
    },
    {
      "epoch": 3.75,
      "eval_acc": 0.8426227238108426,
      "eval_loss": 0.07311566174030304,
      "eval_runtime": 41.1396,
      "eval_samples_per_second": 351.073,
      "eval_steps_per_second": 1.386,
      "step": 4005
    },
    {
      "epoch": 4.0,
      "learning_rate": 2.222222222222222e-06,
      "loss": 0.0528,
      "step": 4272
    },
    {
      "epoch": 4.0,
      "eval_acc": 0.8444921415218445,
      "eval_loss": 0.0724639743566513,
      "eval_runtime": 41.153,
      "eval_samples_per_second": 350.959,
      "eval_steps_per_second": 1.385,
      "step": 4272
    }
  ],
  "max_steps": 5340,
  "num_train_epochs": 5,
  "total_flos": 1.758532164135154e+17,
  "trial_name": null,
  "trial_params": null
}