{
  "best_metric": 0.6143561601638794,
  "best_model_checkpoint": "../../experiments_checkpoints/MAdAiLab/FacebookAI/roberta_base_ledgar/checkpoint-2800",
  "epoch": 2.9850746268656714,
  "eval_steps": 100,
  "global_step": 2800,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03,
      "grad_norm": 3.6866815090179443,
      "learning_rate": 1.9822316986496094e-05,
      "loss": 4.4897,
      "step": 25
    },
    {
      "epoch": 0.05,
      "grad_norm": 5.450005054473877,
      "learning_rate": 1.9644633972992183e-05,
      "loss": 4.0094,
      "step": 50
    },
    {
      "epoch": 0.08,
      "grad_norm": 5.619273662567139,
      "learning_rate": 1.9466950959488275e-05,
      "loss": 3.4988,
      "step": 75
    },
    {
      "epoch": 0.11,
      "grad_norm": 6.885162353515625,
      "learning_rate": 1.9289267945984364e-05,
      "loss": 3.1205,
      "step": 100
    },
    {
      "epoch": 0.11,
      "eval_accuracy": 0.5459,
      "eval_f1_macro": 0.27720224348887473,
      "eval_f1_micro": 0.5459,
      "eval_loss": 2.7419824600219727,
      "eval_runtime": 9.2035,
      "eval_samples_per_second": 1086.542,
      "eval_steps_per_second": 17.059,
      "step": 100
    },
    {
      "epoch": 0.13,
      "grad_norm": 5.539554595947266,
      "learning_rate": 1.9111584932480457e-05,
      "loss": 2.7314,
      "step": 125
    },
    {
      "epoch": 0.16,
      "grad_norm": 6.277410984039307,
      "learning_rate": 1.8933901918976546e-05,
      "loss": 2.5391,
      "step": 150
    },
    {
      "epoch": 0.19,
      "grad_norm": 7.819998264312744,
      "learning_rate": 1.8756218905472638e-05,
      "loss": 2.239,
      "step": 175
    },
    {
      "epoch": 0.21,
      "grad_norm": 6.953737735748291,
      "learning_rate": 1.857853589196873e-05,
      "loss": 2.0491,
      "step": 200
    },
    {
      "epoch": 0.21,
      "eval_accuracy": 0.6623,
      "eval_f1_macro": 0.3967046169089133,
      "eval_f1_micro": 0.6623,
      "eval_loss": 1.8505679368972778,
      "eval_runtime": 9.3275,
      "eval_samples_per_second": 1072.093,
      "eval_steps_per_second": 16.832,
      "step": 200
    },
    {
      "epoch": 0.24,
      "grad_norm": 6.609398365020752,
      "learning_rate": 1.8400852878464823e-05,
      "loss": 1.9233,
      "step": 225
    },
    {
      "epoch": 0.27,
      "grad_norm": 7.400420188903809,
      "learning_rate": 1.822316986496091e-05,
      "loss": 1.8128,
      "step": 250
    },
    {
      "epoch": 0.29,
      "grad_norm": 6.902711868286133,
      "learning_rate": 1.8045486851457e-05,
      "loss": 1.7228,
      "step": 275
    },
    {
      "epoch": 0.32,
      "grad_norm": 7.31109094619751,
      "learning_rate": 1.7867803837953093e-05,
      "loss": 1.6304,
      "step": 300
    },
    {
      "epoch": 0.32,
      "eval_accuracy": 0.722,
      "eval_f1_macro": 0.49532532352780545,
      "eval_f1_micro": 0.722,
      "eval_loss": 1.4552063941955566,
      "eval_runtime": 9.3229,
      "eval_samples_per_second": 1072.631,
      "eval_steps_per_second": 16.84,
      "step": 300
    },
    {
      "epoch": 0.35,
      "grad_norm": 7.23089599609375,
      "learning_rate": 1.7690120824449185e-05,
      "loss": 1.5689,
      "step": 325
    },
    {
      "epoch": 0.37,
      "grad_norm": 6.940632343292236,
      "learning_rate": 1.7512437810945274e-05,
      "loss": 1.4801,
      "step": 350
    },
    {
      "epoch": 0.4,
      "grad_norm": 7.126286029815674,
      "learning_rate": 1.7334754797441367e-05,
      "loss": 1.3823,
      "step": 375
    },
    {
      "epoch": 0.43,
      "grad_norm": 5.370332717895508,
      "learning_rate": 1.715707178393746e-05,
      "loss": 1.3418,
      "step": 400
    },
    {
      "epoch": 0.43,
      "eval_accuracy": 0.7574,
      "eval_f1_macro": 0.5559496659296457,
      "eval_f1_micro": 0.7574,
      "eval_loss": 1.210136890411377,
      "eval_runtime": 9.323,
      "eval_samples_per_second": 1072.616,
      "eval_steps_per_second": 16.84,
      "step": 400
    },
    {
      "epoch": 0.45,
      "grad_norm": 7.18474006652832,
      "learning_rate": 1.6979388770433548e-05,
      "loss": 1.3374,
      "step": 425
    },
    {
      "epoch": 0.48,
      "grad_norm": 7.533451080322266,
      "learning_rate": 1.6801705756929637e-05,
      "loss": 1.3839,
      "step": 450
    },
    {
      "epoch": 0.51,
      "grad_norm": 6.161320686340332,
      "learning_rate": 1.662402274342573e-05,
      "loss": 1.2614,
      "step": 475
    },
    {
      "epoch": 0.53,
      "grad_norm": 8.278363227844238,
      "learning_rate": 1.6446339729921822e-05,
      "loss": 1.2156,
      "step": 500
    },
    {
      "epoch": 0.53,
      "eval_accuracy": 0.7728,
      "eval_f1_macro": 0.5867794961553249,
      "eval_f1_micro": 0.7728,
      "eval_loss": 1.0700616836547852,
      "eval_runtime": 9.3443,
      "eval_samples_per_second": 1070.169,
      "eval_steps_per_second": 16.802,
      "step": 500
    },
    {
      "epoch": 0.56,
      "grad_norm": 7.487790584564209,
      "learning_rate": 1.626865671641791e-05,
      "loss": 1.1673,
      "step": 525
    },
    {
      "epoch": 0.59,
      "grad_norm": 4.386369705200195,
      "learning_rate": 1.6090973702914003e-05,
      "loss": 1.1326,
      "step": 550
    },
    {
      "epoch": 0.61,
      "grad_norm": 9.626976013183594,
      "learning_rate": 1.5913290689410095e-05,
      "loss": 1.0814,
      "step": 575
    },
    {
      "epoch": 0.64,
      "grad_norm": 8.249833106994629,
      "learning_rate": 1.5735607675906184e-05,
      "loss": 1.0994,
      "step": 600
    },
    {
      "epoch": 0.64,
      "eval_accuracy": 0.7922,
      "eval_f1_macro": 0.6223063439982247,
      "eval_f1_micro": 0.7922,
      "eval_loss": 0.9578111171722412,
      "eval_runtime": 9.3152,
      "eval_samples_per_second": 1073.512,
      "eval_steps_per_second": 16.854,
      "step": 600
    },
    {
      "epoch": 0.67,
      "grad_norm": 7.979560375213623,
      "learning_rate": 1.5557924662402277e-05,
      "loss": 1.0574,
      "step": 625
    },
    {
      "epoch": 0.69,
      "grad_norm": 8.0782470703125,
      "learning_rate": 1.5380241648898366e-05,
      "loss": 1.0363,
      "step": 650
    },
    {
      "epoch": 0.72,
      "grad_norm": 10.459425926208496,
      "learning_rate": 1.5202558635394458e-05,
      "loss": 1.0235,
      "step": 675
    },
    {
      "epoch": 0.75,
      "grad_norm": 4.731469631195068,
      "learning_rate": 1.5024875621890549e-05,
      "loss": 0.9857,
      "step": 700
    },
    {
      "epoch": 0.75,
      "eval_accuracy": 0.7968,
      "eval_f1_macro": 0.6274118616247372,
      "eval_f1_micro": 0.7968,
      "eval_loss": 0.8956872224807739,
      "eval_runtime": 9.3081,
      "eval_samples_per_second": 1074.331,
      "eval_steps_per_second": 16.867,
      "step": 700
    },
    {
      "epoch": 0.77,
      "grad_norm": 7.3887858390808105,
      "learning_rate": 1.484719260838664e-05,
      "loss": 0.9104,
      "step": 725
    },
    {
      "epoch": 0.8,
      "grad_norm": 7.011801719665527,
      "learning_rate": 1.4669509594882732e-05,
      "loss": 0.9964,
      "step": 750
    },
    {
      "epoch": 0.83,
      "grad_norm": 8.8153715133667,
      "learning_rate": 1.449182658137882e-05,
      "loss": 0.9792,
      "step": 775
    },
    {
      "epoch": 0.85,
      "grad_norm": 7.788893222808838,
      "learning_rate": 1.4314143567874911e-05,
      "loss": 0.9507,
      "step": 800
    },
    {
      "epoch": 0.85,
      "eval_accuracy": 0.7999,
      "eval_f1_macro": 0.636801164393828,
      "eval_f1_micro": 0.7999,
      "eval_loss": 0.847389280796051,
      "eval_runtime": 9.2025,
      "eval_samples_per_second": 1086.664,
      "eval_steps_per_second": 17.061,
      "step": 800
    },
    {
      "epoch": 0.88,
      "grad_norm": 9.77048110961914,
      "learning_rate": 1.4136460554371004e-05,
      "loss": 0.9459,
      "step": 825
    },
    {
      "epoch": 0.91,
      "grad_norm": 7.047522068023682,
      "learning_rate": 1.3958777540867094e-05,
      "loss": 0.8761,
      "step": 850
    },
    {
      "epoch": 0.93,
      "grad_norm": 11.120019912719727,
      "learning_rate": 1.3781094527363185e-05,
      "loss": 0.9177,
      "step": 875
    },
    {
      "epoch": 0.96,
      "grad_norm": 7.713557243347168,
      "learning_rate": 1.3603411513859277e-05,
      "loss": 0.8734,
      "step": 900
    },
    {
      "epoch": 0.96,
      "eval_accuracy": 0.814,
      "eval_f1_macro": 0.6675206332486842,
      "eval_f1_micro": 0.814,
      "eval_loss": 0.7989650368690491,
      "eval_runtime": 9.2989,
      "eval_samples_per_second": 1075.398,
      "eval_steps_per_second": 16.884,
      "step": 900
    },
    {
      "epoch": 0.99,
      "grad_norm": 8.386655807495117,
      "learning_rate": 1.3425728500355366e-05,
      "loss": 0.9284,
      "step": 925
    },
    {
      "epoch": 1.01,
      "grad_norm": 5.389517307281494,
      "learning_rate": 1.3248045486851457e-05,
      "loss": 0.8602,
      "step": 950
    },
    {
      "epoch": 1.04,
      "grad_norm": 6.257258415222168,
      "learning_rate": 1.307036247334755e-05,
      "loss": 0.8366,
      "step": 975
    },
    {
      "epoch": 1.07,
      "grad_norm": 6.529147148132324,
      "learning_rate": 1.289267945984364e-05,
      "loss": 0.7802,
      "step": 1000
    },
    {
      "epoch": 1.07,
      "eval_accuracy": 0.8128,
      "eval_f1_macro": 0.6605629121796295,
      "eval_f1_micro": 0.8128,
      "eval_loss": 0.7787585258483887,
      "eval_runtime": 9.303,
      "eval_samples_per_second": 1074.916,
      "eval_steps_per_second": 16.876,
      "step": 1000
    },
    {
      "epoch": 1.09,
      "grad_norm": 7.54701566696167,
      "learning_rate": 1.271499644633973e-05,
      "loss": 0.7882,
      "step": 1025
    },
    {
      "epoch": 1.12,
      "grad_norm": 9.281523704528809,
      "learning_rate": 1.2537313432835823e-05,
      "loss": 0.7479,
      "step": 1050
    },
    {
      "epoch": 1.15,
      "grad_norm": 7.9086689949035645,
      "learning_rate": 1.2359630419331914e-05,
      "loss": 0.7846,
      "step": 1075
    },
    {
      "epoch": 1.17,
      "grad_norm": 9.443074226379395,
      "learning_rate": 1.2181947405828003e-05,
      "loss": 0.7869,
      "step": 1100
    },
    {
      "epoch": 1.17,
      "eval_accuracy": 0.8178,
      "eval_f1_macro": 0.6741823550661081,
      "eval_f1_micro": 0.8178,
      "eval_loss": 0.7536987066268921,
      "eval_runtime": 9.2917,
      "eval_samples_per_second": 1076.224,
      "eval_steps_per_second": 16.897,
      "step": 1100
    },
    {
      "epoch": 1.2,
      "grad_norm": 4.2604146003723145,
      "learning_rate": 1.2004264392324093e-05,
      "loss": 0.747,
      "step": 1125
    },
    {
      "epoch": 1.23,
      "grad_norm": 8.130866050720215,
      "learning_rate": 1.1826581378820186e-05,
      "loss": 0.7166,
      "step": 1150
    },
    {
      "epoch": 1.25,
      "grad_norm": 5.583948135375977,
      "learning_rate": 1.1648898365316276e-05,
      "loss": 0.6869,
      "step": 1175
    },
    {
      "epoch": 1.28,
      "grad_norm": 6.114924430847168,
      "learning_rate": 1.1471215351812369e-05,
      "loss": 0.8341,
      "step": 1200
    },
    {
      "epoch": 1.28,
      "eval_accuracy": 0.8232,
      "eval_f1_macro": 0.6880953495522865,
      "eval_f1_micro": 0.8232,
      "eval_loss": 0.7308911681175232,
      "eval_runtime": 9.3042,
      "eval_samples_per_second": 1074.781,
      "eval_steps_per_second": 16.874,
      "step": 1200
    },
    {
      "epoch": 1.31,
      "grad_norm": 10.248847007751465,
      "learning_rate": 1.129353233830846e-05,
      "loss": 0.7763,
      "step": 1225
    },
    {
      "epoch": 1.33,
      "grad_norm": 6.819886207580566,
      "learning_rate": 1.111584932480455e-05,
      "loss": 0.7716,
      "step": 1250
    },
    {
      "epoch": 1.36,
      "grad_norm": 8.37248706817627,
      "learning_rate": 1.0938166311300639e-05,
      "loss": 0.7415,
      "step": 1275
    },
    {
      "epoch": 1.39,
      "grad_norm": 7.980863094329834,
      "learning_rate": 1.0760483297796731e-05,
      "loss": 0.7372,
      "step": 1300
    },
    {
      "epoch": 1.39,
      "eval_accuracy": 0.8219,
      "eval_f1_macro": 0.6875671569817423,
      "eval_f1_micro": 0.8219,
      "eval_loss": 0.7156826257705688,
      "eval_runtime": 9.2883,
      "eval_samples_per_second": 1076.628,
      "eval_steps_per_second": 16.903,
      "step": 1300
    },
    {
      "epoch": 1.41,
      "grad_norm": 8.064375877380371,
      "learning_rate": 1.0582800284292822e-05,
      "loss": 0.7554,
      "step": 1325
    },
    {
      "epoch": 1.44,
      "grad_norm": 6.63127326965332,
      "learning_rate": 1.0405117270788913e-05,
      "loss": 0.7419,
      "step": 1350
    },
    {
      "epoch": 1.47,
      "grad_norm": 7.705358028411865,
      "learning_rate": 1.0227434257285005e-05,
      "loss": 0.6624,
      "step": 1375
    },
    {
      "epoch": 1.49,
      "grad_norm": 6.472929000854492,
      "learning_rate": 1.0049751243781096e-05,
      "loss": 0.661,
      "step": 1400
    },
    {
      "epoch": 1.49,
      "eval_accuracy": 0.8224,
      "eval_f1_macro": 0.694073622954759,
      "eval_f1_micro": 0.8224,
      "eval_loss": 0.7058222889900208,
      "eval_runtime": 9.2938,
      "eval_samples_per_second": 1075.983,
      "eval_steps_per_second": 16.893,
      "step": 1400
    },
    {
      "epoch": 1.52,
      "grad_norm": 9.231730461120605,
      "learning_rate": 9.872068230277187e-06,
      "loss": 0.7197,
      "step": 1425
    },
    {
      "epoch": 1.55,
      "grad_norm": 7.591302394866943,
      "learning_rate": 9.694385216773277e-06,
      "loss": 0.7434,
      "step": 1450
    },
    {
      "epoch": 1.57,
      "grad_norm": 7.090827941894531,
      "learning_rate": 9.516702203269368e-06,
      "loss": 0.7368,
      "step": 1475
    },
    {
      "epoch": 1.6,
      "grad_norm": 9.221566200256348,
      "learning_rate": 9.339019189765458e-06,
      "loss": 0.6932,
      "step": 1500
    },
    {
      "epoch": 1.6,
      "eval_accuracy": 0.8258,
      "eval_f1_macro": 0.6980945742793786,
      "eval_f1_micro": 0.8258,
      "eval_loss": 0.6944009065628052,
      "eval_runtime": 9.2945,
      "eval_samples_per_second": 1075.909,
      "eval_steps_per_second": 16.892,
      "step": 1500
    },
    {
      "epoch": 1.63,
      "grad_norm": 9.705829620361328,
      "learning_rate": 9.161336176261551e-06,
      "loss": 0.6992,
      "step": 1525
    },
    {
      "epoch": 1.65,
      "grad_norm": 6.822689533233643,
      "learning_rate": 8.98365316275764e-06,
      "loss": 0.665,
      "step": 1550
    },
    {
      "epoch": 1.68,
      "grad_norm": 8.525530815124512,
      "learning_rate": 8.805970149253732e-06,
      "loss": 0.6499,
      "step": 1575
    },
    {
      "epoch": 1.71,
      "grad_norm": 9.714850425720215,
      "learning_rate": 8.628287135749823e-06,
      "loss": 0.7305,
      "step": 1600
    },
    {
      "epoch": 1.71,
      "eval_accuracy": 0.8292,
      "eval_f1_macro": 0.7058015695409668,
      "eval_f1_micro": 0.8292,
      "eval_loss": 0.6807318925857544,
      "eval_runtime": 9.2962,
      "eval_samples_per_second": 1075.707,
      "eval_steps_per_second": 16.889,
      "step": 1600
    },
    {
      "epoch": 1.73,
      "grad_norm": 8.261898040771484,
      "learning_rate": 8.450604122245915e-06,
      "loss": 0.6817,
      "step": 1625
    },
    {
      "epoch": 1.76,
      "grad_norm": 6.411789417266846,
      "learning_rate": 8.272921108742004e-06,
      "loss": 0.6775,
      "step": 1650
    },
    {
      "epoch": 1.79,
      "grad_norm": 6.365996360778809,
      "learning_rate": 8.095238095238097e-06,
      "loss": 0.6492,
      "step": 1675
    },
    {
      "epoch": 1.81,
      "grad_norm": 7.634685516357422,
      "learning_rate": 7.917555081734187e-06,
      "loss": 0.6952,
      "step": 1700
    },
    {
      "epoch": 1.81,
      "eval_accuracy": 0.8291,
      "eval_f1_macro": 0.7066258637507297,
      "eval_f1_micro": 0.8291,
      "eval_loss": 0.662746787071228,
      "eval_runtime": 9.2962,
      "eval_samples_per_second": 1075.707,
      "eval_steps_per_second": 16.889,
      "step": 1700
    },
    {
      "epoch": 1.84,
      "grad_norm": 11.271902084350586,
      "learning_rate": 7.739872068230278e-06,
      "loss": 0.7189,
      "step": 1725
    },
    {
      "epoch": 1.87,
      "grad_norm": 9.407271385192871,
      "learning_rate": 7.5621890547263685e-06,
      "loss": 0.6324,
      "step": 1750
    },
    {
      "epoch": 1.89,
      "grad_norm": 9.371041297912598,
      "learning_rate": 7.38450604122246e-06,
      "loss": 0.704,
      "step": 1775
    },
    {
      "epoch": 1.92,
      "grad_norm": 13.078332901000977,
      "learning_rate": 7.20682302771855e-06,
      "loss": 0.6583,
      "step": 1800
    },
    {
      "epoch": 1.92,
      "eval_accuracy": 0.8322,
      "eval_f1_macro": 0.7085973630583551,
      "eval_f1_micro": 0.8322,
      "eval_loss": 0.6509403586387634,
      "eval_runtime": 9.285,
      "eval_samples_per_second": 1077.009,
      "eval_steps_per_second": 16.909,
      "step": 1800
    },
    {
      "epoch": 1.95,
      "grad_norm": 9.226937294006348,
      "learning_rate": 7.029140014214641e-06,
      "loss": 0.7522,
      "step": 1825
    },
    {
      "epoch": 1.97,
      "grad_norm": 8.725870132446289,
      "learning_rate": 6.851457000710733e-06,
      "loss": 0.6921,
      "step": 1850
    },
    {
      "epoch": 2.0,
      "grad_norm": 8.459181785583496,
      "learning_rate": 6.673773987206824e-06,
      "loss": 0.606,
      "step": 1875
    },
    {
      "epoch": 2.03,
      "grad_norm": 7.4500932693481445,
      "learning_rate": 6.496090973702914e-06,
      "loss": 0.6157,
      "step": 1900
    },
    {
      "epoch": 2.03,
      "eval_accuracy": 0.8321,
      "eval_f1_macro": 0.7120242222649424,
      "eval_f1_micro": 0.8321,
      "eval_loss": 0.6486535668373108,
      "eval_runtime": 9.2948,
      "eval_samples_per_second": 1075.865,
      "eval_steps_per_second": 16.891,
      "step": 1900
    },
    {
      "epoch": 2.05,
      "grad_norm": 4.939641952514648,
      "learning_rate": 6.318407960199006e-06,
      "loss": 0.6133,
      "step": 1925
    },
    {
      "epoch": 2.08,
      "grad_norm": 7.50991153717041,
      "learning_rate": 6.140724946695097e-06,
      "loss": 0.6528,
      "step": 1950
    },
    {
      "epoch": 2.11,
      "grad_norm": 6.720021724700928,
      "learning_rate": 5.963041933191187e-06,
      "loss": 0.613,
      "step": 1975
    },
    {
      "epoch": 2.13,
      "grad_norm": 7.243627548217773,
      "learning_rate": 5.785358919687279e-06,
      "loss": 0.5817,
      "step": 2000
    },
    {
      "epoch": 2.13,
      "eval_accuracy": 0.8347,
      "eval_f1_macro": 0.7163666933931198,
      "eval_f1_micro": 0.8347,
      "eval_loss": 0.6429022550582886,
      "eval_runtime": 9.2894,
      "eval_samples_per_second": 1076.493,
      "eval_steps_per_second": 16.901,
      "step": 2000
    },
    {
      "epoch": 2.16,
      "grad_norm": 6.409624099731445,
      "learning_rate": 5.607675906183369e-06,
      "loss": 0.5955,
      "step": 2025
    },
    {
      "epoch": 2.19,
      "grad_norm": 9.208915710449219,
      "learning_rate": 5.42999289267946e-06,
      "loss": 0.6095,
      "step": 2050
    },
    {
      "epoch": 2.21,
      "grad_norm": 8.664913177490234,
      "learning_rate": 5.2523098791755514e-06,
      "loss": 0.5698,
      "step": 2075
    },
    {
      "epoch": 2.24,
      "grad_norm": 8.742962837219238,
      "learning_rate": 5.074626865671642e-06,
      "loss": 0.6002,
      "step": 2100
    },
    {
      "epoch": 2.24,
      "eval_accuracy": 0.836,
      "eval_f1_macro": 0.7201609696583678,
      "eval_f1_micro": 0.836,
      "eval_loss": 0.6375409960746765,
      "eval_runtime": 9.2901,
      "eval_samples_per_second": 1076.415,
      "eval_steps_per_second": 16.9,
      "step": 2100
    },
    {
      "epoch": 2.27,
      "grad_norm": 7.371341228485107,
      "learning_rate": 4.896943852167733e-06,
      "loss": 0.5847,
      "step": 2125
    },
    {
      "epoch": 2.29,
      "grad_norm": 6.305778503417969,
      "learning_rate": 4.719260838663824e-06,
      "loss": 0.5687,
      "step": 2150
    },
    {
      "epoch": 2.32,
      "grad_norm": 5.344340801239014,
      "learning_rate": 4.541577825159915e-06,
      "loss": 0.5202,
      "step": 2175
    },
    {
      "epoch": 2.35,
      "grad_norm": 10.841872215270996,
      "learning_rate": 4.363894811656006e-06,
      "loss": 0.5786,
      "step": 2200
    },
    {
      "epoch": 2.35,
      "eval_accuracy": 0.8401,
      "eval_f1_macro": 0.7318288671889359,
      "eval_f1_micro": 0.8401,
      "eval_loss": 0.634390652179718,
      "eval_runtime": 9.294,
      "eval_samples_per_second": 1075.966,
      "eval_steps_per_second": 16.893,
      "step": 2200
    },
    {
      "epoch": 2.37,
      "grad_norm": 7.411513328552246,
      "learning_rate": 4.186211798152097e-06,
      "loss": 0.5459,
      "step": 2225
    },
    {
      "epoch": 2.4,
      "grad_norm": 5.8190131187438965,
      "learning_rate": 4.008528784648188e-06,
      "loss": 0.5737,
      "step": 2250
    },
    {
      "epoch": 2.43,
      "grad_norm": 6.732067108154297,
      "learning_rate": 3.8308457711442784e-06,
      "loss": 0.5901,
      "step": 2275
    },
    {
      "epoch": 2.45,
      "grad_norm": 8.61303997039795,
      "learning_rate": 3.65316275764037e-06,
      "loss": 0.595,
      "step": 2300
    },
    {
      "epoch": 2.45,
      "eval_accuracy": 0.8382,
      "eval_f1_macro": 0.7221241023417111,
      "eval_f1_micro": 0.8382,
      "eval_loss": 0.627564013004303,
      "eval_runtime": 9.291,
      "eval_samples_per_second": 1076.316,
      "eval_steps_per_second": 16.898,
      "step": 2300
    },
    {
      "epoch": 2.48,
      "grad_norm": 10.592297554016113,
      "learning_rate": 3.4754797441364606e-06,
      "loss": 0.6044,
      "step": 2325
    },
    {
      "epoch": 2.51,
      "grad_norm": 13.547924041748047,
      "learning_rate": 3.297796730632552e-06,
      "loss": 0.603,
      "step": 2350
    },
    {
      "epoch": 2.53,
      "grad_norm": 8.350447654724121,
      "learning_rate": 3.120113717128643e-06,
      "loss": 0.5443,
      "step": 2375
    },
    {
      "epoch": 2.56,
      "grad_norm": 10.097490310668945,
      "learning_rate": 2.9424307036247335e-06,
      "loss": 0.564,
      "step": 2400
    },
    {
      "epoch": 2.56,
      "eval_accuracy": 0.8416,
      "eval_f1_macro": 0.7352520832907747,
      "eval_f1_micro": 0.8416,
      "eval_loss": 0.6197391748428345,
      "eval_runtime": 9.2987,
      "eval_samples_per_second": 1075.417,
      "eval_steps_per_second": 16.884,
      "step": 2400
    },
    {
      "epoch": 2.59,
      "grad_norm": 5.125119209289551,
      "learning_rate": 2.764747690120825e-06,
      "loss": 0.6586,
      "step": 2425
    },
    {
      "epoch": 2.61,
      "grad_norm": 8.446342468261719,
      "learning_rate": 2.5870646766169156e-06,
      "loss": 0.617,
      "step": 2450
    },
    {
      "epoch": 2.64,
      "grad_norm": 7.388770580291748,
      "learning_rate": 2.4093816631130067e-06,
      "loss": 0.5654,
      "step": 2475
    },
    {
      "epoch": 2.67,
      "grad_norm": 11.83965015411377,
      "learning_rate": 2.2316986496090974e-06,
      "loss": 0.5404,
      "step": 2500
    },
    {
      "epoch": 2.67,
      "eval_accuracy": 0.8438,
      "eval_f1_macro": 0.7412158553517405,
      "eval_f1_micro": 0.8438,
      "eval_loss": 0.6157177686691284,
      "eval_runtime": 9.2956,
      "eval_samples_per_second": 1075.778,
      "eval_steps_per_second": 16.89,
      "step": 2500
    },
    {
      "epoch": 2.69,
      "grad_norm": 6.929530143737793,
      "learning_rate": 2.0540156361051885e-06,
      "loss": 0.5531,
      "step": 2525
    },
    {
      "epoch": 2.72,
      "grad_norm": 13.461307525634766,
      "learning_rate": 1.8763326226012796e-06,
      "loss": 0.5699,
      "step": 2550
    },
    {
      "epoch": 2.75,
      "grad_norm": 9.756768226623535,
      "learning_rate": 1.6986496090973702e-06,
      "loss": 0.5622,
      "step": 2575
    },
    {
      "epoch": 2.77,
      "grad_norm": 7.3537983894348145,
      "learning_rate": 1.5209665955934613e-06,
      "loss": 0.5706,
      "step": 2600
    },
    {
      "epoch": 2.77,
      "eval_accuracy": 0.8418,
      "eval_f1_macro": 0.7367949154071943,
      "eval_f1_micro": 0.8418,
      "eval_loss": 0.6161669492721558,
      "eval_runtime": 9.3028,
      "eval_samples_per_second": 1074.943,
      "eval_steps_per_second": 16.877,
      "step": 2600
    },
    {
      "epoch": 2.8,
      "grad_norm": 5.742631435394287,
      "learning_rate": 1.3432835820895524e-06,
      "loss": 0.5536,
      "step": 2625
    },
    {
      "epoch": 2.83,
      "grad_norm": 11.36966609954834,
      "learning_rate": 1.1656005685856433e-06,
      "loss": 0.6294,
      "step": 2650
    },
    {
      "epoch": 2.85,
      "grad_norm": 11.894853591918945,
      "learning_rate": 9.879175550817342e-07,
      "loss": 0.5588,
      "step": 2675
    },
    {
      "epoch": 2.88,
      "grad_norm": 7.0773606300354,
      "learning_rate": 8.102345415778253e-07,
      "loss": 0.5419,
      "step": 2700
    },
    {
      "epoch": 2.88,
      "eval_accuracy": 0.844,
      "eval_f1_macro": 0.738274399314808,
      "eval_f1_micro": 0.844,
      "eval_loss": 0.6148083209991455,
      "eval_runtime": 9.2938,
      "eval_samples_per_second": 1075.984,
      "eval_steps_per_second": 16.893,
      "step": 2700
    },
    {
      "epoch": 2.91,
      "grad_norm": 8.430248260498047,
      "learning_rate": 6.325515280739161e-07,
      "loss": 0.5594,
      "step": 2725
    },
    {
      "epoch": 2.93,
      "grad_norm": 10.73851203918457,
      "learning_rate": 4.548685145700071e-07,
      "loss": 0.5574,
      "step": 2750
    },
    {
      "epoch": 2.96,
      "grad_norm": 8.823871612548828,
      "learning_rate": 2.771855010660981e-07,
      "loss": 0.502,
      "step": 2775
    },
    {
      "epoch": 2.99,
      "grad_norm": 12.509273529052734,
      "learning_rate": 9.950248756218906e-08,
      "loss": 0.5631,
      "step": 2800
    },
    {
      "epoch": 2.99,
      "eval_accuracy": 0.8449,
      "eval_f1_macro": 0.7395952726968685,
      "eval_f1_micro": 0.8449,
      "eval_loss": 0.6143561601638794,
      "eval_runtime": 9.3035,
      "eval_samples_per_second": 1074.867,
      "eval_steps_per_second": 16.875,
      "step": 2800
    }
  ],
  "logging_steps": 25,
  "max_steps": 2814,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 100,
  "total_flos": 1.1797746881134592e+16,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}