matthieulel's picture
End of training
b07723c verified
raw
history blame
34.4 kB
{
"best_metric": 0.855129650507328,
"best_model_checkpoint": "swinv2-tiny-patch4-window8-256-finetuned-galaxy10-decals/checkpoint-1750",
"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.338,
"step": 10
},
{
"epoch": 0.32,
"learning_rate": 5.376344086021506e-06,
"loss": 2.2865,
"step": 20
},
{
"epoch": 0.48,
"learning_rate": 8.064516129032258e-06,
"loss": 2.2056,
"step": 30
},
{
"epoch": 0.64,
"learning_rate": 1.0752688172043012e-05,
"loss": 2.1143,
"step": 40
},
{
"epoch": 0.8,
"learning_rate": 1.3440860215053763e-05,
"loss": 1.9656,
"step": 50
},
{
"epoch": 0.96,
"learning_rate": 1.6129032258064517e-05,
"loss": 1.7462,
"step": 60
},
{
"epoch": 0.99,
"eval_accuracy": 0.44306651634723787,
"eval_f1": 0.3967373381053456,
"eval_loss": 1.459241509437561,
"eval_precision": 0.4309454337536254,
"eval_recall": 0.44306651634723787,
"eval_runtime": 13.0297,
"eval_samples_per_second": 136.15,
"eval_steps_per_second": 2.149,
"step": 62
},
{
"epoch": 1.12,
"learning_rate": 1.881720430107527e-05,
"loss": 1.5959,
"step": 70
},
{
"epoch": 1.28,
"learning_rate": 2.1505376344086024e-05,
"loss": 1.4636,
"step": 80
},
{
"epoch": 1.44,
"learning_rate": 2.4193548387096777e-05,
"loss": 1.3638,
"step": 90
},
{
"epoch": 1.6,
"learning_rate": 2.6881720430107527e-05,
"loss": 1.2923,
"step": 100
},
{
"epoch": 1.76,
"learning_rate": 2.9569892473118284e-05,
"loss": 1.2027,
"step": 110
},
{
"epoch": 1.92,
"learning_rate": 3.2258064516129034e-05,
"loss": 1.1805,
"step": 120
},
{
"epoch": 2.0,
"eval_accuracy": 0.6459977452085682,
"eval_f1": 0.6240514141720535,
"eval_loss": 1.033475399017334,
"eval_precision": 0.674110352354924,
"eval_recall": 0.6459977452085682,
"eval_runtime": 11.9962,
"eval_samples_per_second": 147.88,
"eval_steps_per_second": 2.334,
"step": 125
},
{
"epoch": 2.08,
"learning_rate": 3.494623655913979e-05,
"loss": 1.1111,
"step": 130
},
{
"epoch": 2.24,
"learning_rate": 3.763440860215054e-05,
"loss": 1.0623,
"step": 140
},
{
"epoch": 2.4,
"learning_rate": 4.032258064516129e-05,
"loss": 1.0135,
"step": 150
},
{
"epoch": 2.56,
"learning_rate": 4.301075268817205e-05,
"loss": 0.9981,
"step": 160
},
{
"epoch": 2.72,
"learning_rate": 4.56989247311828e-05,
"loss": 1.0009,
"step": 170
},
{
"epoch": 2.88,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.9342,
"step": 180
},
{
"epoch": 2.99,
"eval_accuracy": 0.7536640360766629,
"eval_f1": 0.7394089362506209,
"eval_loss": 0.7051301598548889,
"eval_precision": 0.7477910225482289,
"eval_recall": 0.7536640360766629,
"eval_runtime": 12.1983,
"eval_samples_per_second": 145.43,
"eval_steps_per_second": 2.295,
"step": 187
},
{
"epoch": 3.04,
"learning_rate": 4.98805256869773e-05,
"loss": 0.9806,
"step": 190
},
{
"epoch": 3.2,
"learning_rate": 4.9581839904420555e-05,
"loss": 0.8707,
"step": 200
},
{
"epoch": 3.36,
"learning_rate": 4.92831541218638e-05,
"loss": 0.8585,
"step": 210
},
{
"epoch": 3.52,
"learning_rate": 4.898446833930705e-05,
"loss": 0.8877,
"step": 220
},
{
"epoch": 3.68,
"learning_rate": 4.86857825567503e-05,
"loss": 0.8385,
"step": 230
},
{
"epoch": 3.84,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.8254,
"step": 240
},
{
"epoch": 4.0,
"learning_rate": 4.80884109916368e-05,
"loss": 0.786,
"step": 250
},
{
"epoch": 4.0,
"eval_accuracy": 0.7745208568207441,
"eval_f1": 0.7637327701817007,
"eval_loss": 0.6467944383621216,
"eval_precision": 0.7730637730047877,
"eval_recall": 0.7745208568207441,
"eval_runtime": 12.5729,
"eval_samples_per_second": 141.098,
"eval_steps_per_second": 2.227,
"step": 250
},
{
"epoch": 4.16,
"learning_rate": 4.778972520908005e-05,
"loss": 0.8039,
"step": 260
},
{
"epoch": 4.32,
"learning_rate": 4.74910394265233e-05,
"loss": 0.7894,
"step": 270
},
{
"epoch": 4.48,
"learning_rate": 4.7192353643966546e-05,
"loss": 0.7725,
"step": 280
},
{
"epoch": 4.64,
"learning_rate": 4.6893667861409805e-05,
"loss": 0.7457,
"step": 290
},
{
"epoch": 4.8,
"learning_rate": 4.659498207885305e-05,
"loss": 0.7854,
"step": 300
},
{
"epoch": 4.96,
"learning_rate": 4.62962962962963e-05,
"loss": 0.7062,
"step": 310
},
{
"epoch": 4.99,
"eval_accuracy": 0.8038331454340474,
"eval_f1": 0.8007998868962807,
"eval_loss": 0.6012569665908813,
"eval_precision": 0.8051953963851154,
"eval_recall": 0.8038331454340474,
"eval_runtime": 12.1624,
"eval_samples_per_second": 145.86,
"eval_steps_per_second": 2.302,
"step": 312
},
{
"epoch": 5.12,
"learning_rate": 4.5997610513739546e-05,
"loss": 0.7495,
"step": 320
},
{
"epoch": 5.28,
"learning_rate": 4.56989247311828e-05,
"loss": 0.747,
"step": 330
},
{
"epoch": 5.44,
"learning_rate": 4.540023894862604e-05,
"loss": 0.7313,
"step": 340
},
{
"epoch": 5.6,
"learning_rate": 4.51015531660693e-05,
"loss": 0.7226,
"step": 350
},
{
"epoch": 5.76,
"learning_rate": 4.4802867383512545e-05,
"loss": 0.6906,
"step": 360
},
{
"epoch": 5.92,
"learning_rate": 4.4504181600955796e-05,
"loss": 0.7011,
"step": 370
},
{
"epoch": 6.0,
"eval_accuracy": 0.8122886133032694,
"eval_f1": 0.8040545665580083,
"eval_loss": 0.5372710227966309,
"eval_precision": 0.8171054089841352,
"eval_recall": 0.8122886133032694,
"eval_runtime": 12.0921,
"eval_samples_per_second": 146.708,
"eval_steps_per_second": 2.316,
"step": 375
},
{
"epoch": 6.08,
"learning_rate": 4.420549581839905e-05,
"loss": 0.6861,
"step": 380
},
{
"epoch": 6.24,
"learning_rate": 4.390681003584229e-05,
"loss": 0.7043,
"step": 390
},
{
"epoch": 6.4,
"learning_rate": 4.360812425328555e-05,
"loss": 0.6644,
"step": 400
},
{
"epoch": 6.56,
"learning_rate": 4.3309438470728796e-05,
"loss": 0.689,
"step": 410
},
{
"epoch": 6.72,
"learning_rate": 4.301075268817205e-05,
"loss": 0.6727,
"step": 420
},
{
"epoch": 6.88,
"learning_rate": 4.271206690561529e-05,
"loss": 0.7014,
"step": 430
},
{
"epoch": 6.99,
"eval_accuracy": 0.8043968432919955,
"eval_f1": 0.799539873702415,
"eval_loss": 0.5469949841499329,
"eval_precision": 0.8048356150099172,
"eval_recall": 0.8043968432919955,
"eval_runtime": 12.0324,
"eval_samples_per_second": 147.436,
"eval_steps_per_second": 2.327,
"step": 437
},
{
"epoch": 7.04,
"learning_rate": 4.241338112305854e-05,
"loss": 0.7202,
"step": 440
},
{
"epoch": 7.2,
"learning_rate": 4.2114695340501795e-05,
"loss": 0.6881,
"step": 450
},
{
"epoch": 7.36,
"learning_rate": 4.1816009557945046e-05,
"loss": 0.6631,
"step": 460
},
{
"epoch": 7.52,
"learning_rate": 4.15173237753883e-05,
"loss": 0.6689,
"step": 470
},
{
"epoch": 7.68,
"learning_rate": 4.121863799283154e-05,
"loss": 0.6496,
"step": 480
},
{
"epoch": 7.84,
"learning_rate": 4.0919952210274794e-05,
"loss": 0.6414,
"step": 490
},
{
"epoch": 8.0,
"learning_rate": 4.062126642771804e-05,
"loss": 0.6447,
"step": 500
},
{
"epoch": 8.0,
"eval_accuracy": 0.8083427282976324,
"eval_f1": 0.8024677836866688,
"eval_loss": 0.530910849571228,
"eval_precision": 0.8086669637121374,
"eval_recall": 0.8083427282976324,
"eval_runtime": 12.4768,
"eval_samples_per_second": 142.184,
"eval_steps_per_second": 2.244,
"step": 500
},
{
"epoch": 8.16,
"learning_rate": 4.032258064516129e-05,
"loss": 0.6111,
"step": 510
},
{
"epoch": 8.32,
"learning_rate": 4.002389486260454e-05,
"loss": 0.648,
"step": 520
},
{
"epoch": 8.48,
"learning_rate": 3.972520908004779e-05,
"loss": 0.6786,
"step": 530
},
{
"epoch": 8.64,
"learning_rate": 3.9426523297491045e-05,
"loss": 0.6227,
"step": 540
},
{
"epoch": 8.8,
"learning_rate": 3.912783751493429e-05,
"loss": 0.656,
"step": 550
},
{
"epoch": 8.96,
"learning_rate": 3.882915173237754e-05,
"loss": 0.608,
"step": 560
},
{
"epoch": 8.99,
"eval_accuracy": 0.8337091319052987,
"eval_f1": 0.8299833390063015,
"eval_loss": 0.4836348593235016,
"eval_precision": 0.8323481302951424,
"eval_recall": 0.8337091319052987,
"eval_runtime": 12.2681,
"eval_samples_per_second": 144.602,
"eval_steps_per_second": 2.282,
"step": 562
},
{
"epoch": 9.12,
"learning_rate": 3.8530465949820786e-05,
"loss": 0.645,
"step": 570
},
{
"epoch": 9.28,
"learning_rate": 3.8231780167264044e-05,
"loss": 0.606,
"step": 580
},
{
"epoch": 9.44,
"learning_rate": 3.793309438470729e-05,
"loss": 0.6187,
"step": 590
},
{
"epoch": 9.6,
"learning_rate": 3.763440860215054e-05,
"loss": 0.6577,
"step": 600
},
{
"epoch": 9.76,
"learning_rate": 3.7335722819593785e-05,
"loss": 0.5886,
"step": 610
},
{
"epoch": 9.92,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.6196,
"step": 620
},
{
"epoch": 10.0,
"eval_accuracy": 0.8331454340473506,
"eval_f1": 0.8268299353906539,
"eval_loss": 0.47972893714904785,
"eval_precision": 0.829346527739505,
"eval_recall": 0.8331454340473506,
"eval_runtime": 12.3006,
"eval_samples_per_second": 144.22,
"eval_steps_per_second": 2.276,
"step": 625
},
{
"epoch": 10.08,
"learning_rate": 3.673835125448029e-05,
"loss": 0.604,
"step": 630
},
{
"epoch": 10.24,
"learning_rate": 3.643966547192354e-05,
"loss": 0.6252,
"step": 640
},
{
"epoch": 10.4,
"learning_rate": 3.614097968936679e-05,
"loss": 0.6096,
"step": 650
},
{
"epoch": 10.56,
"learning_rate": 3.5842293906810036e-05,
"loss": 0.616,
"step": 660
},
{
"epoch": 10.72,
"learning_rate": 3.554360812425329e-05,
"loss": 0.6008,
"step": 670
},
{
"epoch": 10.88,
"learning_rate": 3.524492234169653e-05,
"loss": 0.6031,
"step": 680
},
{
"epoch": 10.99,
"eval_accuracy": 0.826381059751973,
"eval_f1": 0.8238666386963341,
"eval_loss": 0.48629868030548096,
"eval_precision": 0.8274450429447017,
"eval_recall": 0.826381059751973,
"eval_runtime": 12.2042,
"eval_samples_per_second": 145.36,
"eval_steps_per_second": 2.294,
"step": 687
},
{
"epoch": 11.04,
"learning_rate": 3.494623655913979e-05,
"loss": 0.5625,
"step": 690
},
{
"epoch": 11.2,
"learning_rate": 3.4647550776583035e-05,
"loss": 0.5738,
"step": 700
},
{
"epoch": 11.36,
"learning_rate": 3.4348864994026287e-05,
"loss": 0.5596,
"step": 710
},
{
"epoch": 11.52,
"learning_rate": 3.405017921146954e-05,
"loss": 0.5708,
"step": 720
},
{
"epoch": 11.68,
"learning_rate": 3.375149342891278e-05,
"loss": 0.5985,
"step": 730
},
{
"epoch": 11.84,
"learning_rate": 3.3452807646356034e-05,
"loss": 0.5801,
"step": 740
},
{
"epoch": 12.0,
"learning_rate": 3.3154121863799286e-05,
"loss": 0.5462,
"step": 750
},
{
"epoch": 12.0,
"eval_accuracy": 0.8354002254791432,
"eval_f1": 0.8313120284958877,
"eval_loss": 0.47487151622772217,
"eval_precision": 0.8341308199677026,
"eval_recall": 0.8354002254791432,
"eval_runtime": 12.178,
"eval_samples_per_second": 145.672,
"eval_steps_per_second": 2.299,
"step": 750
},
{
"epoch": 12.16,
"learning_rate": 3.285543608124254e-05,
"loss": 0.5717,
"step": 760
},
{
"epoch": 12.32,
"learning_rate": 3.255675029868578e-05,
"loss": 0.5475,
"step": 770
},
{
"epoch": 12.48,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.5737,
"step": 780
},
{
"epoch": 12.64,
"learning_rate": 3.195937873357228e-05,
"loss": 0.5946,
"step": 790
},
{
"epoch": 12.8,
"learning_rate": 3.1660692951015537e-05,
"loss": 0.5901,
"step": 800
},
{
"epoch": 12.96,
"learning_rate": 3.136200716845878e-05,
"loss": 0.5868,
"step": 810
},
{
"epoch": 12.99,
"eval_accuracy": 0.8235625704622322,
"eval_f1": 0.8170801995102805,
"eval_loss": 0.5268704295158386,
"eval_precision": 0.8268442109822676,
"eval_recall": 0.8235625704622322,
"eval_runtime": 12.5064,
"eval_samples_per_second": 141.847,
"eval_steps_per_second": 2.239,
"step": 812
},
{
"epoch": 13.12,
"learning_rate": 3.106332138590203e-05,
"loss": 0.5488,
"step": 820
},
{
"epoch": 13.28,
"learning_rate": 3.0764635603345284e-05,
"loss": 0.5416,
"step": 830
},
{
"epoch": 13.44,
"learning_rate": 3.046594982078853e-05,
"loss": 0.5903,
"step": 840
},
{
"epoch": 13.6,
"learning_rate": 3.016726403823178e-05,
"loss": 0.555,
"step": 850
},
{
"epoch": 13.76,
"learning_rate": 2.9868578255675032e-05,
"loss": 0.5909,
"step": 860
},
{
"epoch": 13.92,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.5844,
"step": 870
},
{
"epoch": 14.0,
"eval_accuracy": 0.8472378804960541,
"eval_f1": 0.8430459045092068,
"eval_loss": 0.44020360708236694,
"eval_precision": 0.8447437765748704,
"eval_recall": 0.8472378804960541,
"eval_runtime": 11.365,
"eval_samples_per_second": 156.093,
"eval_steps_per_second": 2.464,
"step": 875
},
{
"epoch": 14.08,
"learning_rate": 2.9271206690561532e-05,
"loss": 0.5555,
"step": 880
},
{
"epoch": 14.24,
"learning_rate": 2.897252090800478e-05,
"loss": 0.524,
"step": 890
},
{
"epoch": 14.4,
"learning_rate": 2.8673835125448028e-05,
"loss": 0.566,
"step": 900
},
{
"epoch": 14.56,
"learning_rate": 2.8375149342891276e-05,
"loss": 0.5792,
"step": 910
},
{
"epoch": 14.72,
"learning_rate": 2.807646356033453e-05,
"loss": 0.5546,
"step": 920
},
{
"epoch": 14.88,
"learning_rate": 2.777777777777778e-05,
"loss": 0.5326,
"step": 930
},
{
"epoch": 14.99,
"eval_accuracy": 0.8393461104847801,
"eval_f1": 0.8353216124081483,
"eval_loss": 0.46350377798080444,
"eval_precision": 0.835916331519358,
"eval_recall": 0.8393461104847801,
"eval_runtime": 12.1081,
"eval_samples_per_second": 146.514,
"eval_steps_per_second": 2.313,
"step": 937
},
{
"epoch": 15.04,
"learning_rate": 2.747909199522103e-05,
"loss": 0.5364,
"step": 940
},
{
"epoch": 15.2,
"learning_rate": 2.718040621266428e-05,
"loss": 0.5306,
"step": 950
},
{
"epoch": 15.36,
"learning_rate": 2.6881720430107527e-05,
"loss": 0.522,
"step": 960
},
{
"epoch": 15.52,
"learning_rate": 2.6583034647550775e-05,
"loss": 0.5433,
"step": 970
},
{
"epoch": 15.68,
"learning_rate": 2.628434886499403e-05,
"loss": 0.5323,
"step": 980
},
{
"epoch": 15.84,
"learning_rate": 2.5985663082437278e-05,
"loss": 0.5513,
"step": 990
},
{
"epoch": 16.0,
"learning_rate": 2.5686977299880526e-05,
"loss": 0.5313,
"step": 1000
},
{
"epoch": 16.0,
"eval_accuracy": 0.8365276211950394,
"eval_f1": 0.8299910051062035,
"eval_loss": 0.47337380051612854,
"eval_precision": 0.8345002341789073,
"eval_recall": 0.8365276211950394,
"eval_runtime": 12.4556,
"eval_samples_per_second": 142.426,
"eval_steps_per_second": 2.248,
"step": 1000
},
{
"epoch": 16.16,
"learning_rate": 2.5388291517323774e-05,
"loss": 0.5216,
"step": 1010
},
{
"epoch": 16.32,
"learning_rate": 2.5089605734767026e-05,
"loss": 0.5043,
"step": 1020
},
{
"epoch": 16.48,
"learning_rate": 2.4790919952210277e-05,
"loss": 0.5244,
"step": 1030
},
{
"epoch": 16.64,
"learning_rate": 2.4492234169653525e-05,
"loss": 0.5608,
"step": 1040
},
{
"epoch": 16.8,
"learning_rate": 2.4193548387096777e-05,
"loss": 0.5296,
"step": 1050
},
{
"epoch": 16.96,
"learning_rate": 2.3894862604540025e-05,
"loss": 0.4893,
"step": 1060
},
{
"epoch": 16.99,
"eval_accuracy": 0.8365276211950394,
"eval_f1": 0.8316068499333706,
"eval_loss": 0.46750184893608093,
"eval_precision": 0.8334895055423712,
"eval_recall": 0.8365276211950394,
"eval_runtime": 12.3548,
"eval_samples_per_second": 143.588,
"eval_steps_per_second": 2.266,
"step": 1062
},
{
"epoch": 17.12,
"learning_rate": 2.3596176821983273e-05,
"loss": 0.4959,
"step": 1070
},
{
"epoch": 17.28,
"learning_rate": 2.3297491039426525e-05,
"loss": 0.5219,
"step": 1080
},
{
"epoch": 17.44,
"learning_rate": 2.2998805256869773e-05,
"loss": 0.5567,
"step": 1090
},
{
"epoch": 17.6,
"learning_rate": 2.270011947431302e-05,
"loss": 0.547,
"step": 1100
},
{
"epoch": 17.76,
"learning_rate": 2.2401433691756272e-05,
"loss": 0.517,
"step": 1110
},
{
"epoch": 17.92,
"learning_rate": 2.2102747909199524e-05,
"loss": 0.4983,
"step": 1120
},
{
"epoch": 18.0,
"eval_accuracy": 0.8444193912063134,
"eval_f1": 0.8401485583042129,
"eval_loss": 0.4440980553627014,
"eval_precision": 0.843126201021357,
"eval_recall": 0.8444193912063134,
"eval_runtime": 12.7246,
"eval_samples_per_second": 139.415,
"eval_steps_per_second": 2.2,
"step": 1125
},
{
"epoch": 18.08,
"learning_rate": 2.1804062126642775e-05,
"loss": 0.4937,
"step": 1130
},
{
"epoch": 18.24,
"learning_rate": 2.1505376344086024e-05,
"loss": 0.5066,
"step": 1140
},
{
"epoch": 18.4,
"learning_rate": 2.120669056152927e-05,
"loss": 0.5068,
"step": 1150
},
{
"epoch": 18.56,
"learning_rate": 2.0908004778972523e-05,
"loss": 0.5157,
"step": 1160
},
{
"epoch": 18.72,
"learning_rate": 2.060931899641577e-05,
"loss": 0.4944,
"step": 1170
},
{
"epoch": 18.88,
"learning_rate": 2.031063321385902e-05,
"loss": 0.518,
"step": 1180
},
{
"epoch": 18.99,
"eval_accuracy": 0.8416009019165727,
"eval_f1": 0.8375802807575766,
"eval_loss": 0.4692534804344177,
"eval_precision": 0.844069444456146,
"eval_recall": 0.8416009019165727,
"eval_runtime": 12.3566,
"eval_samples_per_second": 143.567,
"eval_steps_per_second": 2.266,
"step": 1187
},
{
"epoch": 19.04,
"learning_rate": 2.001194743130227e-05,
"loss": 0.5197,
"step": 1190
},
{
"epoch": 19.2,
"learning_rate": 1.9713261648745522e-05,
"loss": 0.4825,
"step": 1200
},
{
"epoch": 19.36,
"learning_rate": 1.941457586618877e-05,
"loss": 0.5073,
"step": 1210
},
{
"epoch": 19.52,
"learning_rate": 1.9115890083632022e-05,
"loss": 0.5127,
"step": 1220
},
{
"epoch": 19.68,
"learning_rate": 1.881720430107527e-05,
"loss": 0.4979,
"step": 1230
},
{
"epoch": 19.84,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.5064,
"step": 1240
},
{
"epoch": 20.0,
"learning_rate": 1.821983273596177e-05,
"loss": 0.5228,
"step": 1250
},
{
"epoch": 20.0,
"eval_accuracy": 0.8410372040586246,
"eval_f1": 0.8358383632663606,
"eval_loss": 0.47317034006118774,
"eval_precision": 0.8379460715174784,
"eval_recall": 0.8410372040586246,
"eval_runtime": 12.1895,
"eval_samples_per_second": 145.535,
"eval_steps_per_second": 2.297,
"step": 1250
},
{
"epoch": 20.16,
"learning_rate": 1.7921146953405018e-05,
"loss": 0.4944,
"step": 1260
},
{
"epoch": 20.32,
"learning_rate": 1.7622461170848266e-05,
"loss": 0.5183,
"step": 1270
},
{
"epoch": 20.48,
"learning_rate": 1.7323775388291518e-05,
"loss": 0.4846,
"step": 1280
},
{
"epoch": 20.64,
"learning_rate": 1.702508960573477e-05,
"loss": 0.4575,
"step": 1290
},
{
"epoch": 20.8,
"learning_rate": 1.6726403823178017e-05,
"loss": 0.5073,
"step": 1300
},
{
"epoch": 20.96,
"learning_rate": 1.642771804062127e-05,
"loss": 0.4761,
"step": 1310
},
{
"epoch": 20.99,
"eval_accuracy": 0.8489289740698985,
"eval_f1": 0.8459979793062321,
"eval_loss": 0.4566962420940399,
"eval_precision": 0.8493409108571578,
"eval_recall": 0.8489289740698985,
"eval_runtime": 12.4372,
"eval_samples_per_second": 142.637,
"eval_steps_per_second": 2.251,
"step": 1312
},
{
"epoch": 21.12,
"learning_rate": 1.6129032258064517e-05,
"loss": 0.4721,
"step": 1320
},
{
"epoch": 21.28,
"learning_rate": 1.5830346475507768e-05,
"loss": 0.4947,
"step": 1330
},
{
"epoch": 21.44,
"learning_rate": 1.5531660692951016e-05,
"loss": 0.4604,
"step": 1340
},
{
"epoch": 21.6,
"learning_rate": 1.5232974910394265e-05,
"loss": 0.4866,
"step": 1350
},
{
"epoch": 21.76,
"learning_rate": 1.4934289127837516e-05,
"loss": 0.4898,
"step": 1360
},
{
"epoch": 21.92,
"learning_rate": 1.4635603345280766e-05,
"loss": 0.5311,
"step": 1370
},
{
"epoch": 22.0,
"eval_accuracy": 0.8483652762119503,
"eval_f1": 0.8433313368394385,
"eval_loss": 0.4581642746925354,
"eval_precision": 0.8469029471949041,
"eval_recall": 0.8483652762119503,
"eval_runtime": 12.319,
"eval_samples_per_second": 144.005,
"eval_steps_per_second": 2.273,
"step": 1375
},
{
"epoch": 22.08,
"learning_rate": 1.4336917562724014e-05,
"loss": 0.472,
"step": 1380
},
{
"epoch": 22.24,
"learning_rate": 1.4038231780167265e-05,
"loss": 0.4625,
"step": 1390
},
{
"epoch": 22.4,
"learning_rate": 1.3739545997610515e-05,
"loss": 0.4761,
"step": 1400
},
{
"epoch": 22.56,
"learning_rate": 1.3440860215053763e-05,
"loss": 0.4925,
"step": 1410
},
{
"epoch": 22.72,
"learning_rate": 1.3142174432497015e-05,
"loss": 0.5046,
"step": 1420
},
{
"epoch": 22.88,
"learning_rate": 1.2843488649940263e-05,
"loss": 0.4894,
"step": 1430
},
{
"epoch": 22.99,
"eval_accuracy": 0.846674182638106,
"eval_f1": 0.8432993819735607,
"eval_loss": 0.46270185708999634,
"eval_precision": 0.8449705056102179,
"eval_recall": 0.846674182638106,
"eval_runtime": 12.2439,
"eval_samples_per_second": 144.889,
"eval_steps_per_second": 2.287,
"step": 1437
},
{
"epoch": 23.04,
"learning_rate": 1.2544802867383513e-05,
"loss": 0.4916,
"step": 1440
},
{
"epoch": 23.2,
"learning_rate": 1.2246117084826763e-05,
"loss": 0.4597,
"step": 1450
},
{
"epoch": 23.36,
"learning_rate": 1.1947431302270013e-05,
"loss": 0.4827,
"step": 1460
},
{
"epoch": 23.52,
"learning_rate": 1.1648745519713262e-05,
"loss": 0.4734,
"step": 1470
},
{
"epoch": 23.68,
"learning_rate": 1.135005973715651e-05,
"loss": 0.4993,
"step": 1480
},
{
"epoch": 23.84,
"learning_rate": 1.1051373954599762e-05,
"loss": 0.4327,
"step": 1490
},
{
"epoch": 24.0,
"learning_rate": 1.0752688172043012e-05,
"loss": 0.4791,
"step": 1500
},
{
"epoch": 24.0,
"eval_accuracy": 0.850620067643743,
"eval_f1": 0.8481016264122521,
"eval_loss": 0.4579758942127228,
"eval_precision": 0.8492590570394511,
"eval_recall": 0.850620067643743,
"eval_runtime": 12.3087,
"eval_samples_per_second": 144.125,
"eval_steps_per_second": 2.275,
"step": 1500
},
{
"epoch": 24.16,
"learning_rate": 1.0454002389486262e-05,
"loss": 0.4944,
"step": 1510
},
{
"epoch": 24.32,
"learning_rate": 1.015531660692951e-05,
"loss": 0.4851,
"step": 1520
},
{
"epoch": 24.48,
"learning_rate": 9.856630824372761e-06,
"loss": 0.4575,
"step": 1530
},
{
"epoch": 24.64,
"learning_rate": 9.557945041816011e-06,
"loss": 0.4647,
"step": 1540
},
{
"epoch": 24.8,
"learning_rate": 9.259259259259259e-06,
"loss": 0.4724,
"step": 1550
},
{
"epoch": 24.96,
"learning_rate": 8.960573476702509e-06,
"loss": 0.479,
"step": 1560
},
{
"epoch": 24.99,
"eval_accuracy": 0.8472378804960541,
"eval_f1": 0.8433338370724657,
"eval_loss": 0.46248483657836914,
"eval_precision": 0.8442736543740463,
"eval_recall": 0.8472378804960541,
"eval_runtime": 12.5223,
"eval_samples_per_second": 141.667,
"eval_steps_per_second": 2.236,
"step": 1562
},
{
"epoch": 25.12,
"learning_rate": 8.661887694145759e-06,
"loss": 0.4403,
"step": 1570
},
{
"epoch": 25.28,
"learning_rate": 8.363201911589009e-06,
"loss": 0.4646,
"step": 1580
},
{
"epoch": 25.44,
"learning_rate": 8.064516129032258e-06,
"loss": 0.4619,
"step": 1590
},
{
"epoch": 25.6,
"learning_rate": 7.765830346475508e-06,
"loss": 0.4866,
"step": 1600
},
{
"epoch": 25.76,
"learning_rate": 7.467144563918758e-06,
"loss": 0.4613,
"step": 1610
},
{
"epoch": 25.92,
"learning_rate": 7.168458781362007e-06,
"loss": 0.4487,
"step": 1620
},
{
"epoch": 26.0,
"eval_accuracy": 0.8494926719278467,
"eval_f1": 0.8447270351294509,
"eval_loss": 0.4557044208049774,
"eval_precision": 0.8468737036527202,
"eval_recall": 0.8494926719278467,
"eval_runtime": 12.0207,
"eval_samples_per_second": 147.579,
"eval_steps_per_second": 2.329,
"step": 1625
},
{
"epoch": 26.08,
"learning_rate": 6.869772998805258e-06,
"loss": 0.4644,
"step": 1630
},
{
"epoch": 26.24,
"learning_rate": 6.5710872162485075e-06,
"loss": 0.4634,
"step": 1640
},
{
"epoch": 26.4,
"learning_rate": 6.2724014336917564e-06,
"loss": 0.4648,
"step": 1650
},
{
"epoch": 26.56,
"learning_rate": 5.973715651135006e-06,
"loss": 0.4495,
"step": 1660
},
{
"epoch": 26.72,
"learning_rate": 5.675029868578255e-06,
"loss": 0.459,
"step": 1670
},
{
"epoch": 26.88,
"learning_rate": 5.376344086021506e-06,
"loss": 0.4515,
"step": 1680
},
{
"epoch": 26.99,
"eval_accuracy": 0.8534385569334837,
"eval_f1": 0.8499588607029948,
"eval_loss": 0.4501287341117859,
"eval_precision": 0.8509867433588109,
"eval_recall": 0.8534385569334837,
"eval_runtime": 12.2216,
"eval_samples_per_second": 145.153,
"eval_steps_per_second": 2.291,
"step": 1687
},
{
"epoch": 27.04,
"learning_rate": 5.077658303464755e-06,
"loss": 0.4715,
"step": 1690
},
{
"epoch": 27.2,
"learning_rate": 4.7789725209080055e-06,
"loss": 0.4635,
"step": 1700
},
{
"epoch": 27.36,
"learning_rate": 4.4802867383512545e-06,
"loss": 0.4725,
"step": 1710
},
{
"epoch": 27.52,
"learning_rate": 4.181600955794504e-06,
"loss": 0.4528,
"step": 1720
},
{
"epoch": 27.68,
"learning_rate": 3.882915173237754e-06,
"loss": 0.4152,
"step": 1730
},
{
"epoch": 27.84,
"learning_rate": 3.5842293906810035e-06,
"loss": 0.4803,
"step": 1740
},
{
"epoch": 28.0,
"learning_rate": 3.2855436081242537e-06,
"loss": 0.4862,
"step": 1750
},
{
"epoch": 28.0,
"eval_accuracy": 0.855129650507328,
"eval_f1": 0.8513189116456382,
"eval_loss": 0.4551778733730316,
"eval_precision": 0.8528824827232304,
"eval_recall": 0.855129650507328,
"eval_runtime": 12.1816,
"eval_samples_per_second": 145.629,
"eval_steps_per_second": 2.299,
"step": 1750
},
{
"epoch": 28.16,
"learning_rate": 2.986857825567503e-06,
"loss": 0.4571,
"step": 1760
},
{
"epoch": 28.32,
"learning_rate": 2.688172043010753e-06,
"loss": 0.4539,
"step": 1770
},
{
"epoch": 28.48,
"learning_rate": 2.3894862604540028e-06,
"loss": 0.4544,
"step": 1780
},
{
"epoch": 28.64,
"learning_rate": 2.090800477897252e-06,
"loss": 0.4407,
"step": 1790
},
{
"epoch": 28.8,
"learning_rate": 1.7921146953405017e-06,
"loss": 0.4537,
"step": 1800
},
{
"epoch": 28.96,
"learning_rate": 1.4934289127837516e-06,
"loss": 0.4348,
"step": 1810
},
{
"epoch": 28.99,
"eval_accuracy": 0.850620067643743,
"eval_f1": 0.8468804656079502,
"eval_loss": 0.45116734504699707,
"eval_precision": 0.848642512013622,
"eval_recall": 0.850620067643743,
"eval_runtime": 12.1421,
"eval_samples_per_second": 146.103,
"eval_steps_per_second": 2.306,
"step": 1812
},
{
"epoch": 29.12,
"learning_rate": 1.1947431302270014e-06,
"loss": 0.4538,
"step": 1820
},
{
"epoch": 29.28,
"learning_rate": 8.960573476702509e-07,
"loss": 0.4298,
"step": 1830
},
{
"epoch": 29.44,
"learning_rate": 5.973715651135007e-07,
"loss": 0.4407,
"step": 1840
},
{
"epoch": 29.6,
"learning_rate": 2.9868578255675034e-07,
"loss": 0.4455,
"step": 1850
},
{
"epoch": 29.76,
"learning_rate": 0.0,
"loss": 0.4623,
"step": 1860
},
{
"epoch": 29.76,
"eval_accuracy": 0.855129650507328,
"eval_f1": 0.8515522043491148,
"eval_loss": 0.4539188742637634,
"eval_precision": 0.8533430464385425,
"eval_recall": 0.855129650507328,
"eval_runtime": 12.2921,
"eval_samples_per_second": 144.321,
"eval_steps_per_second": 2.278,
"step": 1860
},
{
"epoch": 29.76,
"step": 1860,
"total_flos": 1.5459140620853969e+19,
"train_loss": 0.6520651725030715,
"train_runtime": 5096.4521,
"train_samples_per_second": 93.959,
"train_steps_per_second": 0.365
}
],
"logging_steps": 10,
"max_steps": 1860,
"num_input_tokens_seen": 0,
"num_train_epochs": 30,
"save_steps": 500,
"total_flos": 1.5459140620853969e+19,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}