{
  "best_metric": 0.9076670631807177,
  "best_model_checkpoint": "./output//roberta-large_ipc1_H_5_32_5e-6_0.01_0.06_07-08-22_06-40/checkpoint-28000",
  "epoch": 0.2084682793177131,
  "global_step": 28000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 2.481759070829404e-07,
      "loss": 0.4329,
      "step": 2000
    },
    {
      "epoch": 0.01,
      "eval_accuracy": 0.5884747561984958,
      "eval_f1": 0.6872181807983729,
      "eval_loss": 0.2462294101715088,
      "eval_roc_auc": 0.7804816717606253,
      "eval_runtime": 6569.6147,
      "eval_samples_per_second": 33.839,
      "eval_steps_per_second": 2.115,
      "step": 2000
    },
    {
      "epoch": 0.03,
      "learning_rate": 4.963518141658809e-07,
      "loss": 0.1951,
      "step": 4000
    },
    {
      "epoch": 0.03,
      "eval_accuracy": 0.7922514304221094,
      "eval_f1": 0.8496544620977544,
      "eval_loss": 0.14517635107040405,
      "eval_roc_auc": 0.895243709044785,
      "eval_runtime": 6569.3584,
      "eval_samples_per_second": 33.841,
      "eval_steps_per_second": 2.115,
      "step": 4000
    },
    {
      "epoch": 0.04,
      "learning_rate": 7.445277212488212e-07,
      "loss": 0.1382,
      "step": 6000
    },
    {
      "epoch": 0.04,
      "eval_accuracy": 0.8181744575191623,
      "eval_f1": 0.8705377939577977,
      "eval_loss": 0.12493952363729477,
      "eval_roc_auc": 0.9109245202637991,
      "eval_runtime": 6567.6398,
      "eval_samples_per_second": 33.85,
      "eval_steps_per_second": 2.116,
      "step": 6000
    },
    {
      "epoch": 0.06,
      "learning_rate": 9.927036283317617e-07,
      "loss": 0.1226,
      "step": 8000
    },
    {
      "epoch": 0.06,
      "eval_accuracy": 0.8257044154161719,
      "eval_f1": 0.8795041743265541,
      "eval_loss": 0.11273977905511856,
      "eval_roc_auc": 0.9185852786225337,
      "eval_runtime": 6573.4579,
      "eval_samples_per_second": 33.82,
      "eval_steps_per_second": 2.114,
      "step": 8000
    },
    {
      "epoch": 0.07,
      "learning_rate": 1.240879535414702e-06,
      "loss": 0.1121,
      "step": 10000
    },
    {
      "epoch": 0.07,
      "eval_accuracy": 0.8326721004714096,
      "eval_f1": 0.8861235840391877,
      "eval_loss": 0.10739490389823914,
      "eval_roc_auc": 0.9245890326912541,
      "eval_runtime": 6572.6062,
      "eval_samples_per_second": 33.824,
      "eval_steps_per_second": 2.114,
      "step": 10000
    },
    {
      "epoch": 0.09,
      "learning_rate": 1.4890554424976424e-06,
      "loss": 0.1059,
      "step": 12000
    },
    {
      "epoch": 0.09,
      "eval_accuracy": 0.8378045269711036,
      "eval_f1": 0.8920632767595246,
      "eval_loss": 0.1017272025346756,
      "eval_roc_auc": 0.929144259699995,
      "eval_runtime": 6575.5269,
      "eval_samples_per_second": 33.809,
      "eval_steps_per_second": 2.113,
      "step": 12000
    },
    {
      "epoch": 0.1,
      "learning_rate": 1.7372313495805828e-06,
      "loss": 0.1024,
      "step": 14000
    },
    {
      "epoch": 0.1,
      "eval_accuracy": 0.8408677894130772,
      "eval_f1": 0.893626255792282,
      "eval_loss": 0.09984249621629715,
      "eval_roc_auc": 0.928852296252917,
      "eval_runtime": 6574.5786,
      "eval_samples_per_second": 33.814,
      "eval_steps_per_second": 2.113,
      "step": 14000
    },
    {
      "epoch": 0.12,
      "learning_rate": 1.9854072566635234e-06,
      "loss": 0.0974,
      "step": 16000
    },
    {
      "epoch": 0.12,
      "eval_accuracy": 0.8442549210119112,
      "eval_f1": 0.898560228806424,
      "eval_loss": 0.09467251598834991,
      "eval_roc_auc": 0.9336472029116307,
      "eval_runtime": 6575.3988,
      "eval_samples_per_second": 33.81,
      "eval_steps_per_second": 2.113,
      "step": 16000
    },
    {
      "epoch": 0.13,
      "learning_rate": 2.2335831637464636e-06,
      "loss": 0.0964,
      "step": 18000
    },
    {
      "epoch": 0.13,
      "eval_accuracy": 0.8498776494296304,
      "eval_f1": 0.9009191664807081,
      "eval_loss": 0.09292563796043396,
      "eval_roc_auc": 0.9335645456901214,
      "eval_runtime": 6575.3606,
      "eval_samples_per_second": 33.81,
      "eval_steps_per_second": 2.113,
      "step": 18000
    },
    {
      "epoch": 0.15,
      "learning_rate": 2.481759070829404e-06,
      "loss": 0.0954,
      "step": 20000
    },
    {
      "epoch": 0.15,
      "eval_accuracy": 0.8467199251502393,
      "eval_f1": 0.8968800725980649,
      "eval_loss": 0.09573328495025635,
      "eval_roc_auc": 0.9303951610003275,
      "eval_runtime": 6576.7864,
      "eval_samples_per_second": 33.803,
      "eval_steps_per_second": 2.113,
      "step": 20000
    },
    {
      "epoch": 0.16,
      "learning_rate": 2.7299349779123447e-06,
      "loss": 0.0926,
      "step": 22000
    },
    {
      "epoch": 0.16,
      "eval_accuracy": 0.8511686278743388,
      "eval_f1": 0.9032656201127709,
      "eval_loss": 0.09094775468111038,
      "eval_roc_auc": 0.9367924962877248,
      "eval_runtime": 6579.9338,
      "eval_samples_per_second": 33.786,
      "eval_steps_per_second": 2.112,
      "step": 22000
    },
    {
      "epoch": 0.18,
      "learning_rate": 2.978110884995285e-06,
      "loss": 0.0916,
      "step": 24000
    },
    {
      "epoch": 0.18,
      "eval_accuracy": 0.8545737522041096,
      "eval_f1": 0.9044808139596631,
      "eval_loss": 0.08910883218050003,
      "eval_roc_auc": 0.9358689602259472,
      "eval_runtime": 6577.1239,
      "eval_samples_per_second": 33.801,
      "eval_steps_per_second": 2.113,
      "step": 24000
    },
    {
      "epoch": 0.19,
      "learning_rate": 3.2262867920782255e-06,
      "loss": 0.0907,
      "step": 26000
    },
    {
      "epoch": 0.19,
      "eval_accuracy": 0.8548076577062866,
      "eval_f1": 0.9047564303992575,
      "eval_loss": 0.08734705299139023,
      "eval_roc_auc": 0.9352966950206927,
      "eval_runtime": 6579.4149,
      "eval_samples_per_second": 33.789,
      "eval_steps_per_second": 2.112,
      "step": 26000
    },
    {
      "epoch": 0.21,
      "learning_rate": 3.4744626991611656e-06,
      "loss": 0.09,
      "step": 28000
    },
    {
      "epoch": 0.21,
      "eval_accuracy": 0.8557117924358559,
      "eval_f1": 0.9076670631807177,
      "eval_loss": 0.08580958843231201,
      "eval_roc_auc": 0.9401102386424043,
      "eval_runtime": 6641.2181,
      "eval_samples_per_second": 33.475,
      "eval_steps_per_second": 2.092,
      "step": 28000
    }
  ],
  "max_steps": 671565,
  "num_train_epochs": 5,
  "total_flos": 8.35021786841088e+17,
  "trial_name": null,
  "trial_params": null
}