matthieulel's picture
End of training
9511d21 verified
{
"best_metric": 0.8573844419391207,
"best_model_checkpoint": "swinv2-base-patch4-window16-256-finetuned-galaxy10-decals/checkpoint-562",
"epoch": 29.76,
"eval_steps": 500,
"global_step": 1860,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.16,
"learning_rate": 2.688172043010753e-06,
"loss": 2.3072,
"step": 10
},
{
"epoch": 0.32,
"learning_rate": 5.376344086021506e-06,
"loss": 2.2442,
"step": 20
},
{
"epoch": 0.48,
"learning_rate": 8.064516129032258e-06,
"loss": 2.1693,
"step": 30
},
{
"epoch": 0.64,
"learning_rate": 1.0752688172043012e-05,
"loss": 2.0304,
"step": 40
},
{
"epoch": 0.8,
"learning_rate": 1.3440860215053763e-05,
"loss": 1.8132,
"step": 50
},
{
"epoch": 0.96,
"learning_rate": 1.6129032258064517e-05,
"loss": 1.5098,
"step": 60
},
{
"epoch": 0.99,
"eval_accuracy": 0.5569334836527621,
"eval_f1": 0.5320699263015376,
"eval_loss": 1.2358192205429077,
"eval_precision": 0.5493242914747882,
"eval_recall": 0.5569334836527621,
"eval_runtime": 14.1239,
"eval_samples_per_second": 125.602,
"eval_steps_per_second": 1.982,
"step": 62
},
{
"epoch": 1.12,
"learning_rate": 1.881720430107527e-05,
"loss": 1.3817,
"step": 70
},
{
"epoch": 1.28,
"learning_rate": 2.1505376344086024e-05,
"loss": 1.2146,
"step": 80
},
{
"epoch": 1.44,
"learning_rate": 2.4193548387096777e-05,
"loss": 1.1001,
"step": 90
},
{
"epoch": 1.6,
"learning_rate": 2.6881720430107527e-05,
"loss": 1.0165,
"step": 100
},
{
"epoch": 1.76,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.8885,
"step": 110
},
{
"epoch": 1.92,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.8845,
"step": 120
},
{
"epoch": 2.0,
"eval_accuracy": 0.7598647125140925,
"eval_f1": 0.7496936828527331,
"eval_loss": 0.739095151424408,
"eval_precision": 0.7800422073276804,
"eval_recall": 0.7598647125140925,
"eval_runtime": 13.5805,
"eval_samples_per_second": 130.628,
"eval_steps_per_second": 2.062,
"step": 125
},
{
"epoch": 2.08,
"learning_rate": 3.494623655913979e-05,
"loss": 0.8624,
"step": 130
},
{
"epoch": 2.24,
"learning_rate": 3.763440860215054e-05,
"loss": 0.8028,
"step": 140
},
{
"epoch": 2.4,
"learning_rate": 4.032258064516129e-05,
"loss": 0.8034,
"step": 150
},
{
"epoch": 2.56,
"learning_rate": 4.301075268817205e-05,
"loss": 0.8272,
"step": 160
},
{
"epoch": 2.72,
"learning_rate": 4.56989247311828e-05,
"loss": 0.7559,
"step": 170
},
{
"epoch": 2.88,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.753,
"step": 180
},
{
"epoch": 2.99,
"eval_accuracy": 0.7970687711386697,
"eval_f1": 0.7903031006048434,
"eval_loss": 0.5997146964073181,
"eval_precision": 0.8062170618116372,
"eval_recall": 0.7970687711386697,
"eval_runtime": 14.0607,
"eval_samples_per_second": 126.167,
"eval_steps_per_second": 1.991,
"step": 187
},
{
"epoch": 3.04,
"learning_rate": 4.98805256869773e-05,
"loss": 0.7863,
"step": 190
},
{
"epoch": 3.2,
"learning_rate": 4.9581839904420555e-05,
"loss": 0.6978,
"step": 200
},
{
"epoch": 3.36,
"learning_rate": 4.92831541218638e-05,
"loss": 0.7132,
"step": 210
},
{
"epoch": 3.52,
"learning_rate": 4.898446833930705e-05,
"loss": 0.7213,
"step": 220
},
{
"epoch": 3.68,
"learning_rate": 4.86857825567503e-05,
"loss": 0.6416,
"step": 230
},
{
"epoch": 3.84,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.6757,
"step": 240
},
{
"epoch": 4.0,
"learning_rate": 4.80884109916368e-05,
"loss": 0.6149,
"step": 250
},
{
"epoch": 4.0,
"eval_accuracy": 0.8331454340473506,
"eval_f1": 0.8276437966288983,
"eval_loss": 0.4920002520084381,
"eval_precision": 0.8285049115438993,
"eval_recall": 0.8331454340473506,
"eval_runtime": 14.1664,
"eval_samples_per_second": 125.226,
"eval_steps_per_second": 1.977,
"step": 250
},
{
"epoch": 4.16,
"learning_rate": 4.778972520908005e-05,
"loss": 0.6591,
"step": 260
},
{
"epoch": 4.32,
"learning_rate": 4.74910394265233e-05,
"loss": 0.6553,
"step": 270
},
{
"epoch": 4.48,
"learning_rate": 4.7192353643966546e-05,
"loss": 0.6243,
"step": 280
},
{
"epoch": 4.64,
"learning_rate": 4.6893667861409805e-05,
"loss": 0.6246,
"step": 290
},
{
"epoch": 4.8,
"learning_rate": 4.659498207885305e-05,
"loss": 0.6415,
"step": 300
},
{
"epoch": 4.96,
"learning_rate": 4.62962962962963e-05,
"loss": 0.5807,
"step": 310
},
{
"epoch": 4.99,
"eval_accuracy": 0.8325817361894025,
"eval_f1": 0.8314819415051848,
"eval_loss": 0.462306410074234,
"eval_precision": 0.832266174147116,
"eval_recall": 0.8325817361894025,
"eval_runtime": 14.3076,
"eval_samples_per_second": 123.99,
"eval_steps_per_second": 1.957,
"step": 312
},
{
"epoch": 5.12,
"learning_rate": 4.5997610513739546e-05,
"loss": 0.5912,
"step": 320
},
{
"epoch": 5.28,
"learning_rate": 4.56989247311828e-05,
"loss": 0.6187,
"step": 330
},
{
"epoch": 5.44,
"learning_rate": 4.540023894862604e-05,
"loss": 0.5915,
"step": 340
},
{
"epoch": 5.6,
"learning_rate": 4.51015531660693e-05,
"loss": 0.5729,
"step": 350
},
{
"epoch": 5.76,
"learning_rate": 4.4802867383512545e-05,
"loss": 0.5681,
"step": 360
},
{
"epoch": 5.92,
"learning_rate": 4.4504181600955796e-05,
"loss": 0.5938,
"step": 370
},
{
"epoch": 6.0,
"eval_accuracy": 0.8365276211950394,
"eval_f1": 0.8294012027924638,
"eval_loss": 0.48571065068244934,
"eval_precision": 0.8403290323376461,
"eval_recall": 0.8365276211950394,
"eval_runtime": 14.0743,
"eval_samples_per_second": 126.045,
"eval_steps_per_second": 1.989,
"step": 375
},
{
"epoch": 6.08,
"learning_rate": 4.420549581839905e-05,
"loss": 0.5502,
"step": 380
},
{
"epoch": 6.24,
"learning_rate": 4.390681003584229e-05,
"loss": 0.5647,
"step": 390
},
{
"epoch": 6.4,
"learning_rate": 4.360812425328555e-05,
"loss": 0.5253,
"step": 400
},
{
"epoch": 6.56,
"learning_rate": 4.3309438470728796e-05,
"loss": 0.5629,
"step": 410
},
{
"epoch": 6.72,
"learning_rate": 4.301075268817205e-05,
"loss": 0.5436,
"step": 420
},
{
"epoch": 6.88,
"learning_rate": 4.271206690561529e-05,
"loss": 0.5583,
"step": 430
},
{
"epoch": 6.99,
"eval_accuracy": 0.826381059751973,
"eval_f1": 0.8242512176217561,
"eval_loss": 0.46798598766326904,
"eval_precision": 0.8314254568647421,
"eval_recall": 0.826381059751973,
"eval_runtime": 13.97,
"eval_samples_per_second": 126.986,
"eval_steps_per_second": 2.004,
"step": 437
},
{
"epoch": 7.04,
"learning_rate": 4.241338112305854e-05,
"loss": 0.5768,
"step": 440
},
{
"epoch": 7.2,
"learning_rate": 4.2114695340501795e-05,
"loss": 0.5508,
"step": 450
},
{
"epoch": 7.36,
"learning_rate": 4.1816009557945046e-05,
"loss": 0.5188,
"step": 460
},
{
"epoch": 7.52,
"learning_rate": 4.15173237753883e-05,
"loss": 0.5504,
"step": 470
},
{
"epoch": 7.68,
"learning_rate": 4.121863799283154e-05,
"loss": 0.5099,
"step": 480
},
{
"epoch": 7.84,
"learning_rate": 4.0919952210274794e-05,
"loss": 0.5188,
"step": 490
},
{
"epoch": 8.0,
"learning_rate": 4.062126642771804e-05,
"loss": 0.5103,
"step": 500
},
{
"epoch": 8.0,
"eval_accuracy": 0.8190529875986471,
"eval_f1": 0.8180354734220666,
"eval_loss": 0.4882398545742035,
"eval_precision": 0.8312265650848392,
"eval_recall": 0.8190529875986471,
"eval_runtime": 14.0396,
"eval_samples_per_second": 126.357,
"eval_steps_per_second": 1.994,
"step": 500
},
{
"epoch": 8.16,
"learning_rate": 4.032258064516129e-05,
"loss": 0.4906,
"step": 510
},
{
"epoch": 8.32,
"learning_rate": 4.002389486260454e-05,
"loss": 0.5145,
"step": 520
},
{
"epoch": 8.48,
"learning_rate": 3.972520908004779e-05,
"loss": 0.5426,
"step": 530
},
{
"epoch": 8.64,
"learning_rate": 3.9426523297491045e-05,
"loss": 0.4938,
"step": 540
},
{
"epoch": 8.8,
"learning_rate": 3.912783751493429e-05,
"loss": 0.5294,
"step": 550
},
{
"epoch": 8.96,
"learning_rate": 3.882915173237754e-05,
"loss": 0.5186,
"step": 560
},
{
"epoch": 8.99,
"eval_accuracy": 0.8573844419391207,
"eval_f1": 0.854574130005493,
"eval_loss": 0.4341418147087097,
"eval_precision": 0.8588925760449824,
"eval_recall": 0.8573844419391207,
"eval_runtime": 14.1319,
"eval_samples_per_second": 125.532,
"eval_steps_per_second": 1.981,
"step": 562
},
{
"epoch": 9.12,
"learning_rate": 3.8530465949820786e-05,
"loss": 0.4958,
"step": 570
},
{
"epoch": 9.28,
"learning_rate": 3.8231780167264044e-05,
"loss": 0.4714,
"step": 580
},
{
"epoch": 9.44,
"learning_rate": 3.793309438470729e-05,
"loss": 0.4876,
"step": 590
},
{
"epoch": 9.6,
"learning_rate": 3.763440860215054e-05,
"loss": 0.4944,
"step": 600
},
{
"epoch": 9.76,
"learning_rate": 3.7335722819593785e-05,
"loss": 0.4562,
"step": 610
},
{
"epoch": 9.92,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.4696,
"step": 620
},
{
"epoch": 10.0,
"eval_accuracy": 0.8494926719278467,
"eval_f1": 0.8481056611842751,
"eval_loss": 0.42929476499557495,
"eval_precision": 0.8483832338340108,
"eval_recall": 0.8494926719278467,
"eval_runtime": 14.4925,
"eval_samples_per_second": 122.408,
"eval_steps_per_second": 1.932,
"step": 625
},
{
"epoch": 10.08,
"learning_rate": 3.673835125448029e-05,
"loss": 0.4727,
"step": 630
},
{
"epoch": 10.24,
"learning_rate": 3.643966547192354e-05,
"loss": 0.4773,
"step": 640
},
{
"epoch": 10.4,
"learning_rate": 3.614097968936679e-05,
"loss": 0.5013,
"step": 650
},
{
"epoch": 10.56,
"learning_rate": 3.5842293906810036e-05,
"loss": 0.4657,
"step": 660
},
{
"epoch": 10.72,
"learning_rate": 3.554360812425329e-05,
"loss": 0.4554,
"step": 670
},
{
"epoch": 10.88,
"learning_rate": 3.524492234169653e-05,
"loss": 0.4711,
"step": 680
},
{
"epoch": 10.99,
"eval_accuracy": 0.8421645997745209,
"eval_f1": 0.841376954933891,
"eval_loss": 0.4396071135997772,
"eval_precision": 0.8430720391153448,
"eval_recall": 0.8421645997745209,
"eval_runtime": 13.8918,
"eval_samples_per_second": 127.701,
"eval_steps_per_second": 2.016,
"step": 687
},
{
"epoch": 11.04,
"learning_rate": 3.494623655913979e-05,
"loss": 0.4237,
"step": 690
},
{
"epoch": 11.2,
"learning_rate": 3.4647550776583035e-05,
"loss": 0.4357,
"step": 700
},
{
"epoch": 11.36,
"learning_rate": 3.4348864994026287e-05,
"loss": 0.4275,
"step": 710
},
{
"epoch": 11.52,
"learning_rate": 3.405017921146954e-05,
"loss": 0.4192,
"step": 720
},
{
"epoch": 11.68,
"learning_rate": 3.375149342891278e-05,
"loss": 0.4679,
"step": 730
},
{
"epoch": 11.84,
"learning_rate": 3.3452807646356034e-05,
"loss": 0.4545,
"step": 740
},
{
"epoch": 12.0,
"learning_rate": 3.3154121863799286e-05,
"loss": 0.4271,
"step": 750
},
{
"epoch": 12.0,
"eval_accuracy": 0.8489289740698985,
"eval_f1": 0.8479793252918201,
"eval_loss": 0.4547128677368164,
"eval_precision": 0.8500074699440431,
"eval_recall": 0.8489289740698985,
"eval_runtime": 14.1403,
"eval_samples_per_second": 125.457,
"eval_steps_per_second": 1.98,
"step": 750
},
{
"epoch": 12.16,
"learning_rate": 3.285543608124254e-05,
"loss": 0.4396,
"step": 760
},
{
"epoch": 12.32,
"learning_rate": 3.255675029868578e-05,
"loss": 0.4256,
"step": 770
},
{
"epoch": 12.48,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.4252,
"step": 780
},
{
"epoch": 12.64,
"learning_rate": 3.195937873357228e-05,
"loss": 0.4578,
"step": 790
},
{
"epoch": 12.8,
"learning_rate": 3.1660692951015537e-05,
"loss": 0.4381,
"step": 800
},
{
"epoch": 12.96,
"learning_rate": 3.136200716845878e-05,
"loss": 0.4576,
"step": 810
},
{
"epoch": 12.99,
"eval_accuracy": 0.8489289740698985,
"eval_f1": 0.847344503637895,
"eval_loss": 0.4424493610858917,
"eval_precision": 0.8522067023126736,
"eval_recall": 0.8489289740698985,
"eval_runtime": 14.3056,
"eval_samples_per_second": 124.007,
"eval_steps_per_second": 1.957,
"step": 812
},
{
"epoch": 13.12,
"learning_rate": 3.106332138590203e-05,
"loss": 0.4097,
"step": 820
},
{
"epoch": 13.28,
"learning_rate": 3.0764635603345284e-05,
"loss": 0.4107,
"step": 830
},
{
"epoch": 13.44,
"learning_rate": 3.046594982078853e-05,
"loss": 0.4588,
"step": 840
},
{
"epoch": 13.6,
"learning_rate": 3.016726403823178e-05,
"loss": 0.4117,
"step": 850
},
{
"epoch": 13.76,
"learning_rate": 2.9868578255675032e-05,
"loss": 0.44,
"step": 860
},
{
"epoch": 13.92,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.4483,
"step": 870
},
{
"epoch": 14.0,
"eval_accuracy": 0.8494926719278467,
"eval_f1": 0.8492425838074925,
"eval_loss": 0.4355337917804718,
"eval_precision": 0.8530872306633641,
"eval_recall": 0.8494926719278467,
"eval_runtime": 14.2373,
"eval_samples_per_second": 124.602,
"eval_steps_per_second": 1.967,
"step": 875
},
{
"epoch": 14.08,
"learning_rate": 2.9271206690561532e-05,
"loss": 0.4228,
"step": 880
},
{
"epoch": 14.24,
"learning_rate": 2.897252090800478e-05,
"loss": 0.3787,
"step": 890
},
{
"epoch": 14.4,
"learning_rate": 2.8673835125448028e-05,
"loss": 0.4008,
"step": 900
},
{
"epoch": 14.56,
"learning_rate": 2.8375149342891276e-05,
"loss": 0.426,
"step": 910
},
{
"epoch": 14.72,
"learning_rate": 2.807646356033453e-05,
"loss": 0.4017,
"step": 920
},
{
"epoch": 14.88,
"learning_rate": 2.777777777777778e-05,
"loss": 0.3914,
"step": 930
},
{
"epoch": 14.99,
"eval_accuracy": 0.8540022547914318,
"eval_f1": 0.8531865003871237,
"eval_loss": 0.4359700679779053,
"eval_precision": 0.8532564316361854,
"eval_recall": 0.8540022547914318,
"eval_runtime": 14.4662,
"eval_samples_per_second": 122.63,
"eval_steps_per_second": 1.936,
"step": 937
},
{
"epoch": 15.04,
"learning_rate": 2.747909199522103e-05,
"loss": 0.403,
"step": 940
},
{
"epoch": 15.2,
"learning_rate": 2.718040621266428e-05,
"loss": 0.379,
"step": 950
},
{
"epoch": 15.36,
"learning_rate": 2.6881720430107527e-05,
"loss": 0.3996,
"step": 960
},
{
"epoch": 15.52,
"learning_rate": 2.6583034647550775e-05,
"loss": 0.4133,
"step": 970
},
{
"epoch": 15.68,
"learning_rate": 2.628434886499403e-05,
"loss": 0.3847,
"step": 980
},
{
"epoch": 15.84,
"learning_rate": 2.5985663082437278e-05,
"loss": 0.41,
"step": 990
},
{
"epoch": 16.0,
"learning_rate": 2.5686977299880526e-05,
"loss": 0.3883,
"step": 1000
},
{
"epoch": 16.0,
"eval_accuracy": 0.85456595264938,
"eval_f1": 0.8526324439725788,
"eval_loss": 0.44642314314842224,
"eval_precision": 0.8550297386149915,
"eval_recall": 0.85456595264938,
"eval_runtime": 14.2582,
"eval_samples_per_second": 124.419,
"eval_steps_per_second": 1.964,
"step": 1000
},
{
"epoch": 16.16,
"learning_rate": 2.5388291517323774e-05,
"loss": 0.3831,
"step": 1010
},
{
"epoch": 16.32,
"learning_rate": 2.5089605734767026e-05,
"loss": 0.3698,
"step": 1020
},
{
"epoch": 16.48,
"learning_rate": 2.4790919952210277e-05,
"loss": 0.3822,
"step": 1030
},
{
"epoch": 16.64,
"learning_rate": 2.4492234169653525e-05,
"loss": 0.4215,
"step": 1040
},
{
"epoch": 16.8,
"learning_rate": 2.4193548387096777e-05,
"loss": 0.3799,
"step": 1050
},
{
"epoch": 16.96,
"learning_rate": 2.3894862604540025e-05,
"loss": 0.3421,
"step": 1060
},
{
"epoch": 16.99,
"eval_accuracy": 0.8489289740698985,
"eval_f1": 0.8479111609298762,
"eval_loss": 0.44728776812553406,
"eval_precision": 0.8486260152172298,
"eval_recall": 0.8489289740698985,
"eval_runtime": 14.6558,
"eval_samples_per_second": 121.044,
"eval_steps_per_second": 1.911,
"step": 1062
},
{
"epoch": 17.12,
"learning_rate": 2.3596176821983273e-05,
"loss": 0.3786,
"step": 1070
},
{
"epoch": 17.28,
"learning_rate": 2.3297491039426525e-05,
"loss": 0.3708,
"step": 1080
},
{
"epoch": 17.44,
"learning_rate": 2.2998805256869773e-05,
"loss": 0.3813,
"step": 1090
},
{
"epoch": 17.6,
"learning_rate": 2.270011947431302e-05,
"loss": 0.3823,
"step": 1100
},
{
"epoch": 17.76,
"learning_rate": 2.2401433691756272e-05,
"loss": 0.3715,
"step": 1110
},
{
"epoch": 17.92,
"learning_rate": 2.2102747909199524e-05,
"loss": 0.3666,
"step": 1120
},
{
"epoch": 18.0,
"eval_accuracy": 0.8540022547914318,
"eval_f1": 0.8528008682046607,
"eval_loss": 0.4455113708972931,
"eval_precision": 0.8540569189661774,
"eval_recall": 0.8540022547914318,
"eval_runtime": 14.5314,
"eval_samples_per_second": 122.08,
"eval_steps_per_second": 1.927,
"step": 1125
},
{
"epoch": 18.08,
"learning_rate": 2.1804062126642775e-05,
"loss": 0.3528,
"step": 1130
},
{
"epoch": 18.24,
"learning_rate": 2.1505376344086024e-05,
"loss": 0.3515,
"step": 1140
},
{
"epoch": 18.4,
"learning_rate": 2.120669056152927e-05,
"loss": 0.3586,
"step": 1150
},
{
"epoch": 18.56,
"learning_rate": 2.0908004778972523e-05,
"loss": 0.3522,
"step": 1160
},
{
"epoch": 18.72,
"learning_rate": 2.060931899641577e-05,
"loss": 0.3596,
"step": 1170
},
{
"epoch": 18.88,
"learning_rate": 2.031063321385902e-05,
"loss": 0.3737,
"step": 1180
},
{
"epoch": 18.99,
"eval_accuracy": 0.8573844419391207,
"eval_f1": 0.8560565093621151,
"eval_loss": 0.45872262120246887,
"eval_precision": 0.8560472962449797,
"eval_recall": 0.8573844419391207,
"eval_runtime": 13.9382,
"eval_samples_per_second": 127.276,
"eval_steps_per_second": 2.009,
"step": 1187
},
{
"epoch": 19.04,
"learning_rate": 2.001194743130227e-05,
"loss": 0.3618,
"step": 1190
},
{
"epoch": 19.2,
"learning_rate": 1.9713261648745522e-05,
"loss": 0.3624,
"step": 1200
},
{
"epoch": 19.36,
"learning_rate": 1.941457586618877e-05,
"loss": 0.3525,
"step": 1210
},
{
"epoch": 19.52,
"learning_rate": 1.9115890083632022e-05,
"loss": 0.3446,
"step": 1220
},
{
"epoch": 19.68,
"learning_rate": 1.881720430107527e-05,
"loss": 0.3392,
"step": 1230
},
{
"epoch": 19.84,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.3629,
"step": 1240
},
{
"epoch": 20.0,
"learning_rate": 1.821983273596177e-05,
"loss": 0.3694,
"step": 1250
},
{
"epoch": 20.0,
"eval_accuracy": 0.855129650507328,
"eval_f1": 0.8522853423001466,
"eval_loss": 0.45834314823150635,
"eval_precision": 0.8528348433176285,
"eval_recall": 0.855129650507328,
"eval_runtime": 14.4317,
"eval_samples_per_second": 122.924,
"eval_steps_per_second": 1.94,
"step": 1250
},
{
"epoch": 20.16,
"learning_rate": 1.7921146953405018e-05,
"loss": 0.3379,
"step": 1260
},
{
"epoch": 20.32,
"learning_rate": 1.7622461170848266e-05,
"loss": 0.3519,
"step": 1270
},
{
"epoch": 20.48,
"learning_rate": 1.7323775388291518e-05,
"loss": 0.3307,
"step": 1280
},
{
"epoch": 20.64,
"learning_rate": 1.702508960573477e-05,
"loss": 0.3337,
"step": 1290
},
{
"epoch": 20.8,
"learning_rate": 1.6726403823178017e-05,
"loss": 0.3585,
"step": 1300
},
{
"epoch": 20.96,
"learning_rate": 1.642771804062127e-05,
"loss": 0.3269,
"step": 1310
},
{
"epoch": 20.99,
"eval_accuracy": 0.850620067643743,
"eval_f1": 0.8487164054190969,
"eval_loss": 0.4882984757423401,
"eval_precision": 0.8494157131229852,
"eval_recall": 0.850620067643743,
"eval_runtime": 14.6519,
"eval_samples_per_second": 121.076,
"eval_steps_per_second": 1.911,
"step": 1312
},
{
"epoch": 21.12,
"learning_rate": 1.6129032258064517e-05,
"loss": 0.3222,
"step": 1320
},
{
"epoch": 21.28,
"learning_rate": 1.5830346475507768e-05,
"loss": 0.338,
"step": 1330
},
{
"epoch": 21.44,
"learning_rate": 1.5531660692951016e-05,
"loss": 0.346,
"step": 1340
},
{
"epoch": 21.6,
"learning_rate": 1.5232974910394265e-05,
"loss": 0.33,
"step": 1350
},
{
"epoch": 21.76,
"learning_rate": 1.4934289127837516e-05,
"loss": 0.3371,
"step": 1360
},
{
"epoch": 21.92,
"learning_rate": 1.4635603345280766e-05,
"loss": 0.3699,
"step": 1370
},
{
"epoch": 22.0,
"eval_accuracy": 0.8500563697857948,
"eval_f1": 0.8485861921693642,
"eval_loss": 0.4807828664779663,
"eval_precision": 0.8514470196649129,
"eval_recall": 0.8500563697857948,
"eval_runtime": 13.9392,
"eval_samples_per_second": 127.267,
"eval_steps_per_second": 2.009,
"step": 1375
},
{
"epoch": 22.08,
"learning_rate": 1.4336917562724014e-05,
"loss": 0.3314,
"step": 1380
},
{
"epoch": 22.24,
"learning_rate": 1.4038231780167265e-05,
"loss": 0.3094,
"step": 1390
},
{
"epoch": 22.4,
"learning_rate": 1.3739545997610515e-05,
"loss": 0.3281,
"step": 1400
},
{
"epoch": 22.56,
"learning_rate": 1.3440860215053763e-05,
"loss": 0.3277,
"step": 1410
},
{
"epoch": 22.72,
"learning_rate": 1.3142174432497015e-05,
"loss": 0.3321,
"step": 1420
},
{
"epoch": 22.88,
"learning_rate": 1.2843488649940263e-05,
"loss": 0.3395,
"step": 1430
},
{
"epoch": 22.99,
"eval_accuracy": 0.8483652762119503,
"eval_f1": 0.8476840953762947,
"eval_loss": 0.4706174433231354,
"eval_precision": 0.8492797248601933,
"eval_recall": 0.8483652762119503,
"eval_runtime": 14.2474,
"eval_samples_per_second": 124.514,
"eval_steps_per_second": 1.965,
"step": 1437
},
{
"epoch": 23.04,
"learning_rate": 1.2544802867383513e-05,
"loss": 0.319,
"step": 1440
},
{
"epoch": 23.2,
"learning_rate": 1.2246117084826763e-05,
"loss": 0.313,
"step": 1450
},
{
"epoch": 23.36,
"learning_rate": 1.1947431302270013e-05,
"loss": 0.3065,
"step": 1460
},
{
"epoch": 23.52,
"learning_rate": 1.1648745519713262e-05,
"loss": 0.3158,
"step": 1470
},
{
"epoch": 23.68,
"learning_rate": 1.135005973715651e-05,
"loss": 0.3424,
"step": 1480
},
{
"epoch": 23.84,
"learning_rate": 1.1051373954599762e-05,
"loss": 0.2878,
"step": 1490
},
{
"epoch": 24.0,
"learning_rate": 1.0752688172043012e-05,
"loss": 0.3147,
"step": 1500
},
{
"epoch": 24.0,
"eval_accuracy": 0.8568207440811725,
"eval_f1": 0.8556665798513748,
"eval_loss": 0.4675993323326111,
"eval_precision": 0.8556454616708037,
"eval_recall": 0.8568207440811725,
"eval_runtime": 14.4786,
"eval_samples_per_second": 122.525,
"eval_steps_per_second": 1.934,
"step": 1500
},
{
"epoch": 24.16,
"learning_rate": 1.0454002389486262e-05,
"loss": 0.3206,
"step": 1510
},
{
"epoch": 24.32,
"learning_rate": 1.015531660692951e-05,
"loss": 0.3371,
"step": 1520
},
{
"epoch": 24.48,
"learning_rate": 9.856630824372761e-06,
"loss": 0.3092,
"step": 1530
},
{
"epoch": 24.64,
"learning_rate": 9.557945041816011e-06,
"loss": 0.3041,
"step": 1540
},
{
"epoch": 24.8,
"learning_rate": 9.259259259259259e-06,
"loss": 0.2997,
"step": 1550
},
{
"epoch": 24.96,
"learning_rate": 8.960573476702509e-06,
"loss": 0.3352,
"step": 1560
},
{
"epoch": 24.99,
"eval_accuracy": 0.8556933483652762,
"eval_f1": 0.8537584363348232,
"eval_loss": 0.48680880665779114,
"eval_precision": 0.8543439587001157,
"eval_recall": 0.8556933483652762,
"eval_runtime": 14.3831,
"eval_samples_per_second": 123.339,
"eval_steps_per_second": 1.947,
"step": 1562
},
{
"epoch": 25.12,
"learning_rate": 8.661887694145759e-06,
"loss": 0.292,
"step": 1570
},
{
"epoch": 25.28,
"learning_rate": 8.363201911589009e-06,
"loss": 0.3025,
"step": 1580
},
{
"epoch": 25.44,
"learning_rate": 8.064516129032258e-06,
"loss": 0.2996,
"step": 1590
},
{
"epoch": 25.6,
"learning_rate": 7.765830346475508e-06,
"loss": 0.3184,
"step": 1600
},
{
"epoch": 25.76,
"learning_rate": 7.467144563918758e-06,
"loss": 0.3006,
"step": 1610
},
{
"epoch": 25.92,
"learning_rate": 7.168458781362007e-06,
"loss": 0.3007,
"step": 1620
},
{
"epoch": 26.0,
"eval_accuracy": 0.8489289740698985,
"eval_f1": 0.8474670469443679,
"eval_loss": 0.48870715498924255,
"eval_precision": 0.8491891204056768,
"eval_recall": 0.8489289740698985,
"eval_runtime": 14.3108,
"eval_samples_per_second": 123.963,
"eval_steps_per_second": 1.957,
"step": 1625
},
{
"epoch": 26.08,
"learning_rate": 6.869772998805258e-06,
"loss": 0.3049,
"step": 1630
},
{
"epoch": 26.24,
"learning_rate": 6.5710872162485075e-06,
"loss": 0.3276,
"step": 1640
},
{
"epoch": 26.4,
"learning_rate": 6.2724014336917564e-06,
"loss": 0.3091,
"step": 1650
},
{
"epoch": 26.56,
"learning_rate": 5.973715651135006e-06,
"loss": 0.288,
"step": 1660
},
{
"epoch": 26.72,
"learning_rate": 5.675029868578255e-06,
"loss": 0.3112,
"step": 1670
},
{
"epoch": 26.88,
"learning_rate": 5.376344086021506e-06,
"loss": 0.3049,
"step": 1680
},
{
"epoch": 26.99,
"eval_accuracy": 0.8534385569334837,
"eval_f1": 0.85263850114306,
"eval_loss": 0.4837624430656433,
"eval_precision": 0.8531960514889521,
"eval_recall": 0.8534385569334837,
"eval_runtime": 13.9142,
"eval_samples_per_second": 127.496,
"eval_steps_per_second": 2.012,
"step": 1687
},
{
"epoch": 27.04,
"learning_rate": 5.077658303464755e-06,
"loss": 0.3046,
"step": 1690
},
{
"epoch": 27.2,
"learning_rate": 4.7789725209080055e-06,
"loss": 0.2958,
"step": 1700
},
{
"epoch": 27.36,
"learning_rate": 4.4802867383512545e-06,
"loss": 0.3213,
"step": 1710
},
{
"epoch": 27.52,
"learning_rate": 4.181600955794504e-06,
"loss": 0.2831,
"step": 1720
},
{
"epoch": 27.68,
"learning_rate": 3.882915173237754e-06,
"loss": 0.2672,
"step": 1730
},
{
"epoch": 27.84,
"learning_rate": 3.5842293906810035e-06,
"loss": 0.3145,
"step": 1740
},
{
"epoch": 28.0,
"learning_rate": 3.2855436081242537e-06,
"loss": 0.3228,
"step": 1750
},
{
"epoch": 28.0,
"eval_accuracy": 0.855129650507328,
"eval_f1": 0.8535889981721714,
"eval_loss": 0.49099063873291016,
"eval_precision": 0.8538511857164837,
"eval_recall": 0.855129650507328,
"eval_runtime": 14.4445,
"eval_samples_per_second": 122.815,
"eval_steps_per_second": 1.938,
"step": 1750
},
{
"epoch": 28.16,
"learning_rate": 2.986857825567503e-06,
"loss": 0.2961,
"step": 1760
},
{
"epoch": 28.32,
"learning_rate": 2.688172043010753e-06,
"loss": 0.3,
"step": 1770
},
{
"epoch": 28.48,
"learning_rate": 2.3894862604540028e-06,
"loss": 0.2902,
"step": 1780
},
{
"epoch": 28.64,
"learning_rate": 2.090800477897252e-06,
"loss": 0.3021,
"step": 1790
},
{
"epoch": 28.8,
"learning_rate": 1.7921146953405017e-06,
"loss": 0.3001,
"step": 1800
},
{
"epoch": 28.96,
"learning_rate": 1.4934289127837516e-06,
"loss": 0.3005,
"step": 1810
},
{
"epoch": 28.99,
"eval_accuracy": 0.8534385569334837,
"eval_f1": 0.8517521631952123,
"eval_loss": 0.48459768295288086,
"eval_precision": 0.8517293054811527,
"eval_recall": 0.8534385569334837,
"eval_runtime": 14.7082,
"eval_samples_per_second": 120.613,
"eval_steps_per_second": 1.904,
"step": 1812
},
{
"epoch": 29.12,
"learning_rate": 1.1947431302270014e-06,
"loss": 0.3103,
"step": 1820
},
{
"epoch": 29.28,
"learning_rate": 8.960573476702509e-07,
"loss": 0.2916,
"step": 1830
},
{
"epoch": 29.44,
"learning_rate": 5.973715651135007e-07,
"loss": 0.2845,
"step": 1840
},
{
"epoch": 29.6,
"learning_rate": 2.9868578255675034e-07,
"loss": 0.2974,
"step": 1850
},
{
"epoch": 29.76,
"learning_rate": 0.0,
"loss": 0.2972,
"step": 1860
},
{
"epoch": 29.76,
"eval_accuracy": 0.8556933483652762,
"eval_f1": 0.8542865471295588,
"eval_loss": 0.48264560103416443,
"eval_precision": 0.8543714689014588,
"eval_recall": 0.8556933483652762,
"eval_runtime": 14.1506,
"eval_samples_per_second": 125.366,
"eval_steps_per_second": 1.979,
"step": 1860
},
{
"epoch": 29.76,
"step": 1860,
"total_flos": 4.870114457320843e+19,
"train_loss": 0.5012358284765674,
"train_runtime": 8248.1266,
"train_samples_per_second": 58.057,
"train_steps_per_second": 0.226
}
],
"logging_steps": 10,
"max_steps": 1860,
"num_input_tokens_seen": 0,
"num_train_epochs": 30,
"save_steps": 500,
"total_flos": 4.870114457320843e+19,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}