{
  "best_metric": 0.9036697247706422,
  "best_model_checkpoint": "distilbert-base-uncased-finetuned-sst2/run-0/checkpoint-4210",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 21050,
  "is_hyper_param_search": true,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.1187648456057007,
      "grad_norm": 9.763530731201172,
      "learning_rate": 2.4130303407723998e-05,
      "loss": 0.3459,
      "step": 500
    },
    {
      "epoch": 0.2375296912114014,
      "grad_norm": 7.90731954574585,
      "learning_rate": 2.3543191402669885e-05,
      "loss": 0.2557,
      "step": 1000
    },
    {
      "epoch": 0.35629453681710216,
      "grad_norm": 7.188111305236816,
      "learning_rate": 2.2956079397615776e-05,
      "loss": 0.2242,
      "step": 1500
    },
    {
      "epoch": 0.4750593824228028,
      "grad_norm": 9.416199684143066,
      "learning_rate": 2.2368967392561663e-05,
      "loss": 0.2281,
      "step": 2000
    },
    {
      "epoch": 0.5938242280285035,
      "grad_norm": 6.982799053192139,
      "learning_rate": 2.1781855387507554e-05,
      "loss": 0.2196,
      "step": 2500
    },
    {
      "epoch": 0.7125890736342043,
      "grad_norm": 3.475909948348999,
      "learning_rate": 2.1194743382453438e-05,
      "loss": 0.199,
      "step": 3000
    },
    {
      "epoch": 0.831353919239905,
      "grad_norm": 18.930234909057617,
      "learning_rate": 2.0607631377399328e-05,
      "loss": 0.1988,
      "step": 3500
    },
    {
      "epoch": 0.9501187648456056,
      "grad_norm": 21.44248390197754,
      "learning_rate": 2.0020519372345215e-05,
      "loss": 0.1964,
      "step": 4000
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.9036697247706422,
      "eval_loss": 0.279630184173584,
      "eval_runtime": 1.1981,
      "eval_samples_per_second": 727.831,
      "eval_steps_per_second": 45.907,
      "step": 4210
    },
    {
      "epoch": 1.0688836104513064,
      "grad_norm": 6.991998672485352,
      "learning_rate": 1.9433407367291103e-05,
      "loss": 0.1449,
      "step": 4500
    },
    {
      "epoch": 1.187648456057007,
      "grad_norm": 1.0682291984558105,
      "learning_rate": 1.8846295362236993e-05,
      "loss": 0.1307,
      "step": 5000
    },
    {
      "epoch": 1.3064133016627077,
      "grad_norm": 1.4475390911102295,
      "learning_rate": 1.825918335718288e-05,
      "loss": 0.1248,
      "step": 5500
    },
    {
      "epoch": 1.4251781472684084,
      "grad_norm": 0.4122050702571869,
      "learning_rate": 1.767207135212877e-05,
      "loss": 0.1288,
      "step": 6000
    },
    {
      "epoch": 1.5439429928741093,
      "grad_norm": 0.35570672154426575,
      "learning_rate": 1.7084959347074655e-05,
      "loss": 0.1353,
      "step": 6500
    },
    {
      "epoch": 1.66270783847981,
      "grad_norm": 0.9978006482124329,
      "learning_rate": 1.6497847342020545e-05,
      "loss": 0.1316,
      "step": 7000
    },
    {
      "epoch": 1.7814726840855108,
      "grad_norm": 7.568444728851318,
      "learning_rate": 1.5910735336966433e-05,
      "loss": 0.1363,
      "step": 7500
    },
    {
      "epoch": 1.9002375296912115,
      "grad_norm": 0.1252310872077942,
      "learning_rate": 1.532362333191232e-05,
      "loss": 0.1271,
      "step": 8000
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.8887614678899083,
      "eval_loss": 0.37789419293403625,
      "eval_runtime": 1.2,
      "eval_samples_per_second": 726.651,
      "eval_steps_per_second": 45.832,
      "step": 8420
    },
    {
      "epoch": 2.019002375296912,
      "grad_norm": 0.2641623616218567,
      "learning_rate": 1.4736511326858209e-05,
      "loss": 0.114,
      "step": 8500
    },
    {
      "epoch": 2.137767220902613,
      "grad_norm": 16.91781234741211,
      "learning_rate": 1.4149399321804098e-05,
      "loss": 0.0814,
      "step": 9000
    },
    {
      "epoch": 2.2565320665083135,
      "grad_norm": 0.13691911101341248,
      "learning_rate": 1.3562287316749987e-05,
      "loss": 0.0807,
      "step": 9500
    },
    {
      "epoch": 2.375296912114014,
      "grad_norm": 14.652079582214355,
      "learning_rate": 1.2975175311695872e-05,
      "loss": 0.0909,
      "step": 10000
    },
    {
      "epoch": 2.494061757719715,
      "grad_norm": 0.23481616377830505,
      "learning_rate": 1.2388063306641761e-05,
      "loss": 0.0823,
      "step": 10500
    },
    {
      "epoch": 2.6128266033254155,
      "grad_norm": 0.16778022050857544,
      "learning_rate": 1.1800951301587648e-05,
      "loss": 0.0866,
      "step": 11000
    },
    {
      "epoch": 2.731591448931116,
      "grad_norm": 22.94564437866211,
      "learning_rate": 1.1213839296533537e-05,
      "loss": 0.0894,
      "step": 11500
    },
    {
      "epoch": 2.850356294536817,
      "grad_norm": 4.668305397033691,
      "learning_rate": 1.0626727291479424e-05,
      "loss": 0.0969,
      "step": 12000
    },
    {
      "epoch": 2.969121140142518,
      "grad_norm": 0.056590911000967026,
      "learning_rate": 1.0039615286425313e-05,
      "loss": 0.0932,
      "step": 12500
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.8990825688073395,
      "eval_loss": 0.4282897412776947,
      "eval_runtime": 1.1944,
      "eval_samples_per_second": 730.051,
      "eval_steps_per_second": 46.047,
      "step": 12630
    },
    {
      "epoch": 3.0878859857482186,
      "grad_norm": 0.023897308856248856,
      "learning_rate": 9.452503281371202e-06,
      "loss": 0.0537,
      "step": 13000
    },
    {
      "epoch": 3.2066508313539193,
      "grad_norm": 0.7962842583656311,
      "learning_rate": 8.86539127631709e-06,
      "loss": 0.0473,
      "step": 13500
    },
    {
      "epoch": 3.32541567695962,
      "grad_norm": 0.47386619448661804,
      "learning_rate": 8.278279271262978e-06,
      "loss": 0.0608,
      "step": 14000
    },
    {
      "epoch": 3.4441805225653206,
      "grad_norm": 0.06599200516939163,
      "learning_rate": 7.691167266208865e-06,
      "loss": 0.0677,
      "step": 14500
    },
    {
      "epoch": 3.5629453681710213,
      "grad_norm": 0.18340125679969788,
      "learning_rate": 7.104055261154754e-06,
      "loss": 0.0579,
      "step": 15000
    },
    {
      "epoch": 3.6817102137767224,
      "grad_norm": 5.418150901794434,
      "learning_rate": 6.5169432561006415e-06,
      "loss": 0.0517,
      "step": 15500
    },
    {
      "epoch": 3.800475059382423,
      "grad_norm": 19.50031852722168,
      "learning_rate": 5.92983125104653e-06,
      "loss": 0.0617,
      "step": 16000
    },
    {
      "epoch": 3.9192399049881237,
      "grad_norm": 0.0843653604388237,
      "learning_rate": 5.3427192459924185e-06,
      "loss": 0.0599,
      "step": 16500
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.9036697247706422,
      "eval_loss": 0.466791033744812,
      "eval_runtime": 1.2178,
      "eval_samples_per_second": 716.054,
      "eval_steps_per_second": 45.164,
      "step": 16840
    },
    {
      "epoch": 4.038004750593824,
      "grad_norm": 0.04484863579273224,
      "learning_rate": 4.7556072409383065e-06,
      "loss": 0.0424,
      "step": 17000
    },
    {
      "epoch": 4.156769596199525,
      "grad_norm": 0.0025116826873272657,
      "learning_rate": 4.1684952358841946e-06,
      "loss": 0.0335,
      "step": 17500
    },
    {
      "epoch": 4.275534441805226,
      "grad_norm": 0.4570842981338501,
      "learning_rate": 3.581383230830083e-06,
      "loss": 0.0395,
      "step": 18000
    },
    {
      "epoch": 4.394299287410926,
      "grad_norm": 0.7941423654556274,
      "learning_rate": 2.9942712257759707e-06,
      "loss": 0.0334,
      "step": 18500
    },
    {
      "epoch": 4.513064133016627,
      "grad_norm": 3.7694590091705322,
      "learning_rate": 2.4071592207218587e-06,
      "loss": 0.035,
      "step": 19000
    },
    {
      "epoch": 4.631828978622328,
      "grad_norm": 0.018865738064050674,
      "learning_rate": 1.8200472156677468e-06,
      "loss": 0.0373,
      "step": 19500
    },
    {
      "epoch": 4.750593824228028,
      "grad_norm": 0.03485831245779991,
      "learning_rate": 1.232935210613635e-06,
      "loss": 0.0282,
      "step": 20000
    },
    {
      "epoch": 4.869358669833729,
      "grad_norm": 0.014678980223834515,
      "learning_rate": 6.458232055595231e-07,
      "loss": 0.0369,
      "step": 20500
    },
    {
      "epoch": 4.98812351543943,
      "grad_norm": 0.06822077184915543,
      "learning_rate": 5.871120050541119e-08,
      "loss": 0.0299,
      "step": 21000
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.9025229357798165,
      "eval_loss": 0.5252613425254822,
      "eval_runtime": 1.204,
      "eval_samples_per_second": 724.232,
      "eval_steps_per_second": 45.68,
      "step": 21050
    }
  ],
  "logging_steps": 500,
  "max_steps": 21050,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "total_flos": 3061523906193576.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": {
    "learning_rate": 2.471741541277811e-05,
    "num_train_epochs": 5,
    "per_device_train_batch_size": 16,
    "seed": 27
  }
}