matthieulel's picture
End of training
60a22e4 verified
{
"best_metric": 0.8669673055242391,
"best_model_checkpoint": "dinov2-base-imagenet1k-1-layer-finetuned-galaxy10-decals/checkpoint-1625",
"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.549,
"step": 10
},
{
"epoch": 0.32,
"learning_rate": 5.376344086021506e-06,
"loss": 1.8762,
"step": 20
},
{
"epoch": 0.48,
"learning_rate": 8.064516129032258e-06,
"loss": 1.3173,
"step": 30
},
{
"epoch": 0.64,
"learning_rate": 1.0752688172043012e-05,
"loss": 1.053,
"step": 40
},
{
"epoch": 0.8,
"learning_rate": 1.3440860215053763e-05,
"loss": 0.8899,
"step": 50
},
{
"epoch": 0.96,
"learning_rate": 1.6129032258064517e-05,
"loss": 0.8157,
"step": 60
},
{
"epoch": 0.99,
"eval_accuracy": 0.7812852311161218,
"eval_f1": 0.7853152390094879,
"eval_loss": 0.6739660501480103,
"eval_precision": 0.80460189132339,
"eval_recall": 0.7812852311161218,
"eval_runtime": 11.5622,
"eval_samples_per_second": 153.431,
"eval_steps_per_second": 2.422,
"step": 62
},
{
"epoch": 1.12,
"learning_rate": 1.881720430107527e-05,
"loss": 0.8066,
"step": 70
},
{
"epoch": 1.28,
"learning_rate": 2.1505376344086024e-05,
"loss": 0.7598,
"step": 80
},
{
"epoch": 1.44,
"learning_rate": 2.4193548387096777e-05,
"loss": 0.7406,
"step": 90
},
{
"epoch": 1.6,
"learning_rate": 2.6881720430107527e-05,
"loss": 0.7583,
"step": 100
},
{
"epoch": 1.76,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.7416,
"step": 110
},
{
"epoch": 1.92,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.8091,
"step": 120
},
{
"epoch": 2.0,
"eval_accuracy": 0.8021420518602029,
"eval_f1": 0.7949709860561608,
"eval_loss": 0.5948110818862915,
"eval_precision": 0.8016087087108787,
"eval_recall": 0.8021420518602029,
"eval_runtime": 11.5457,
"eval_samples_per_second": 153.65,
"eval_steps_per_second": 2.425,
"step": 125
},
{
"epoch": 2.08,
"learning_rate": 3.494623655913979e-05,
"loss": 0.7469,
"step": 130
},
{
"epoch": 2.24,
"learning_rate": 3.763440860215054e-05,
"loss": 0.7026,
"step": 140
},
{
"epoch": 2.4,
"learning_rate": 4.032258064516129e-05,
"loss": 0.7497,
"step": 150
},
{
"epoch": 2.56,
"learning_rate": 4.301075268817205e-05,
"loss": 0.7236,
"step": 160
},
{
"epoch": 2.72,
"learning_rate": 4.56989247311828e-05,
"loss": 0.6881,
"step": 170
},
{
"epoch": 2.88,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.6983,
"step": 180
},
{
"epoch": 2.99,
"eval_accuracy": 0.7965050732807215,
"eval_f1": 0.7909181013872687,
"eval_loss": 0.6016444563865662,
"eval_precision": 0.8077081664320911,
"eval_recall": 0.7965050732807215,
"eval_runtime": 11.5752,
"eval_samples_per_second": 153.259,
"eval_steps_per_second": 2.419,
"step": 187
},
{
"epoch": 3.04,
"learning_rate": 4.98805256869773e-05,
"loss": 0.8131,
"step": 190
},
{
"epoch": 3.2,
"learning_rate": 4.9581839904420555e-05,
"loss": 0.7253,
"step": 200
},
{
"epoch": 3.36,
"learning_rate": 4.92831541218638e-05,
"loss": 0.7012,
"step": 210
},
{
"epoch": 3.52,
"learning_rate": 4.898446833930705e-05,
"loss": 0.6939,
"step": 220
},
{
"epoch": 3.68,
"learning_rate": 4.86857825567503e-05,
"loss": 0.6652,
"step": 230
},
{
"epoch": 3.84,
"learning_rate": 4.8387096774193554e-05,
"loss": 0.7167,
"step": 240
},
{
"epoch": 4.0,
"learning_rate": 4.80884109916368e-05,
"loss": 0.6701,
"step": 250
},
{
"epoch": 4.0,
"eval_accuracy": 0.798196166854566,
"eval_f1": 0.7953625301219193,
"eval_loss": 0.5675845742225647,
"eval_precision": 0.8016177730685449,
"eval_recall": 0.798196166854566,
"eval_runtime": 11.5598,
"eval_samples_per_second": 153.463,
"eval_steps_per_second": 2.422,
"step": 250
},
{
"epoch": 4.16,
"learning_rate": 4.778972520908005e-05,
"loss": 0.6737,
"step": 260
},
{
"epoch": 4.32,
"learning_rate": 4.74910394265233e-05,
"loss": 0.6296,
"step": 270
},
{
"epoch": 4.48,
"learning_rate": 4.7192353643966546e-05,
"loss": 0.649,
"step": 280
},
{
"epoch": 4.64,
"learning_rate": 4.6893667861409805e-05,
"loss": 0.6319,
"step": 290
},
{
"epoch": 4.8,
"learning_rate": 4.659498207885305e-05,
"loss": 0.6448,
"step": 300
},
{
"epoch": 4.96,
"learning_rate": 4.62962962962963e-05,
"loss": 0.5998,
"step": 310
},
{
"epoch": 4.99,
"eval_accuracy": 0.8286358511837655,
"eval_f1": 0.830186011295156,
"eval_loss": 0.5116113424301147,
"eval_precision": 0.8400596294704076,
"eval_recall": 0.8286358511837655,
"eval_runtime": 11.5623,
"eval_samples_per_second": 153.43,
"eval_steps_per_second": 2.422,
"step": 312
},
{
"epoch": 5.12,
"learning_rate": 4.5997610513739546e-05,
"loss": 0.6066,
"step": 320
},
{
"epoch": 5.28,
"learning_rate": 4.56989247311828e-05,
"loss": 0.6128,
"step": 330
},
{
"epoch": 5.44,
"learning_rate": 4.540023894862604e-05,
"loss": 0.5735,
"step": 340
},
{
"epoch": 5.6,
"learning_rate": 4.51015531660693e-05,
"loss": 0.5786,
"step": 350
},
{
"epoch": 5.76,
"learning_rate": 4.4802867383512545e-05,
"loss": 0.615,
"step": 360
},
{
"epoch": 5.92,
"learning_rate": 4.4504181600955796e-05,
"loss": 0.5521,
"step": 370
},
{
"epoch": 6.0,
"eval_accuracy": 0.8354002254791432,
"eval_f1": 0.8325342621627388,
"eval_loss": 0.5154868364334106,
"eval_precision": 0.8374902079148381,
"eval_recall": 0.8354002254791432,
"eval_runtime": 11.5628,
"eval_samples_per_second": 153.423,
"eval_steps_per_second": 2.422,
"step": 375
},
{
"epoch": 6.08,
"learning_rate": 4.420549581839905e-05,
"loss": 0.5543,
"step": 380
},
{
"epoch": 6.24,
"learning_rate": 4.390681003584229e-05,
"loss": 0.5769,
"step": 390
},
{
"epoch": 6.4,
"learning_rate": 4.360812425328555e-05,
"loss": 0.5413,
"step": 400
},
{
"epoch": 6.56,
"learning_rate": 4.3309438470728796e-05,
"loss": 0.5743,
"step": 410
},
{
"epoch": 6.72,
"learning_rate": 4.301075268817205e-05,
"loss": 0.5464,
"step": 420
},
{
"epoch": 6.88,
"learning_rate": 4.271206690561529e-05,
"loss": 0.5441,
"step": 430
},
{
"epoch": 6.99,
"eval_accuracy": 0.8032694475760992,
"eval_f1": 0.7980463963295215,
"eval_loss": 0.5574168562889099,
"eval_precision": 0.810386236165255,
"eval_recall": 0.8032694475760992,
"eval_runtime": 11.572,
"eval_samples_per_second": 153.301,
"eval_steps_per_second": 2.42,
"step": 437
},
{
"epoch": 7.04,
"learning_rate": 4.241338112305854e-05,
"loss": 0.5855,
"step": 440
},
{
"epoch": 7.2,
"learning_rate": 4.2114695340501795e-05,
"loss": 0.5552,
"step": 450
},
{
"epoch": 7.36,
"learning_rate": 4.1816009557945046e-05,
"loss": 0.5519,
"step": 460
},
{
"epoch": 7.52,
"learning_rate": 4.15173237753883e-05,
"loss": 0.5466,
"step": 470
},
{
"epoch": 7.68,
"learning_rate": 4.121863799283154e-05,
"loss": 0.5219,
"step": 480
},
{
"epoch": 7.84,
"learning_rate": 4.0919952210274794e-05,
"loss": 0.5223,
"step": 490
},
{
"epoch": 8.0,
"learning_rate": 4.062126642771804e-05,
"loss": 0.5142,
"step": 500
},
{
"epoch": 8.0,
"eval_accuracy": 0.8410372040586246,
"eval_f1": 0.8375632661659843,
"eval_loss": 0.4818373918533325,
"eval_precision": 0.8417609296030333,
"eval_recall": 0.8410372040586246,
"eval_runtime": 11.6656,
"eval_samples_per_second": 152.071,
"eval_steps_per_second": 2.4,
"step": 500
},
{
"epoch": 8.16,
"learning_rate": 4.032258064516129e-05,
"loss": 0.4695,
"step": 510
},
{
"epoch": 8.32,
"learning_rate": 4.002389486260454e-05,
"loss": 0.4778,
"step": 520
},
{
"epoch": 8.48,
"learning_rate": 3.972520908004779e-05,
"loss": 0.5221,
"step": 530
},
{
"epoch": 8.64,
"learning_rate": 3.9426523297491045e-05,
"loss": 0.541,
"step": 540
},
{
"epoch": 8.8,
"learning_rate": 3.912783751493429e-05,
"loss": 0.5288,
"step": 550
},
{
"epoch": 8.96,
"learning_rate": 3.882915173237754e-05,
"loss": 0.5136,
"step": 560
},
{
"epoch": 8.99,
"eval_accuracy": 0.8337091319052987,
"eval_f1": 0.8316871457376268,
"eval_loss": 0.4913617670536041,
"eval_precision": 0.8353039996367163,
"eval_recall": 0.8337091319052987,
"eval_runtime": 11.6118,
"eval_samples_per_second": 152.775,
"eval_steps_per_second": 2.411,
"step": 562
},
{
"epoch": 9.12,
"learning_rate": 3.8530465949820786e-05,
"loss": 0.5007,
"step": 570
},
{
"epoch": 9.28,
"learning_rate": 3.8231780167264044e-05,
"loss": 0.487,
"step": 580
},
{
"epoch": 9.44,
"learning_rate": 3.793309438470729e-05,
"loss": 0.4923,
"step": 590
},
{
"epoch": 9.6,
"learning_rate": 3.763440860215054e-05,
"loss": 0.4958,
"step": 600
},
{
"epoch": 9.76,
"learning_rate": 3.7335722819593785e-05,
"loss": 0.456,
"step": 610
},
{
"epoch": 9.92,
"learning_rate": 3.7037037037037037e-05,
"loss": 0.4533,
"step": 620
},
{
"epoch": 10.0,
"eval_accuracy": 0.8320180383314544,
"eval_f1": 0.8294898477296314,
"eval_loss": 0.474043071269989,
"eval_precision": 0.8335131003551897,
"eval_recall": 0.8320180383314544,
"eval_runtime": 11.5585,
"eval_samples_per_second": 153.48,
"eval_steps_per_second": 2.422,
"step": 625
},
{
"epoch": 10.08,
"learning_rate": 3.673835125448029e-05,
"loss": 0.489,
"step": 630
},
{
"epoch": 10.24,
"learning_rate": 3.643966547192354e-05,
"loss": 0.4366,
"step": 640
},
{
"epoch": 10.4,
"learning_rate": 3.614097968936679e-05,
"loss": 0.4796,
"step": 650
},
{
"epoch": 10.56,
"learning_rate": 3.5842293906810036e-05,
"loss": 0.4461,
"step": 660
},
{
"epoch": 10.72,
"learning_rate": 3.554360812425329e-05,
"loss": 0.4448,
"step": 670
},
{
"epoch": 10.88,
"learning_rate": 3.524492234169653e-05,
"loss": 0.4904,
"step": 680
},
{
"epoch": 10.99,
"eval_accuracy": 0.8399098083427283,
"eval_f1": 0.8375098947235918,
"eval_loss": 0.5075283050537109,
"eval_precision": 0.8409445507007258,
"eval_recall": 0.8399098083427283,
"eval_runtime": 11.5984,
"eval_samples_per_second": 152.952,
"eval_steps_per_second": 2.414,
"step": 687
},
{
"epoch": 11.04,
"learning_rate": 3.494623655913979e-05,
"loss": 0.4216,
"step": 690
},
{
"epoch": 11.2,
"learning_rate": 3.4647550776583035e-05,
"loss": 0.4258,
"step": 700
},
{
"epoch": 11.36,
"learning_rate": 3.4348864994026287e-05,
"loss": 0.4058,
"step": 710
},
{
"epoch": 11.52,
"learning_rate": 3.405017921146954e-05,
"loss": 0.434,
"step": 720
},
{
"epoch": 11.68,
"learning_rate": 3.375149342891278e-05,
"loss": 0.4581,
"step": 730
},
{
"epoch": 11.84,
"learning_rate": 3.3452807646356034e-05,
"loss": 0.4476,
"step": 740
},
{
"epoch": 12.0,
"learning_rate": 3.3154121863799286e-05,
"loss": 0.4361,
"step": 750
},
{
"epoch": 12.0,
"eval_accuracy": 0.8562570462232244,
"eval_f1": 0.8540053975192837,
"eval_loss": 0.45521676540374756,
"eval_precision": 0.8554214225284571,
"eval_recall": 0.8562570462232244,
"eval_runtime": 11.576,
"eval_samples_per_second": 153.248,
"eval_steps_per_second": 2.419,
"step": 750
},
{
"epoch": 12.16,
"learning_rate": 3.285543608124254e-05,
"loss": 0.3985,
"step": 760
},
{
"epoch": 12.32,
"learning_rate": 3.255675029868578e-05,
"loss": 0.3962,
"step": 770
},
{
"epoch": 12.48,
"learning_rate": 3.2258064516129034e-05,
"loss": 0.4196,
"step": 780
},
{
"epoch": 12.64,
"learning_rate": 3.195937873357228e-05,
"loss": 0.4559,
"step": 790
},
{
"epoch": 12.8,
"learning_rate": 3.1660692951015537e-05,
"loss": 0.4057,
"step": 800
},
{
"epoch": 12.96,
"learning_rate": 3.136200716845878e-05,
"loss": 0.414,
"step": 810
},
{
"epoch": 12.99,
"eval_accuracy": 0.8365276211950394,
"eval_f1": 0.8374485204829644,
"eval_loss": 0.5025420784950256,
"eval_precision": 0.8454939581149772,
"eval_recall": 0.8365276211950394,
"eval_runtime": 11.579,
"eval_samples_per_second": 153.208,
"eval_steps_per_second": 2.418,
"step": 812
},
{
"epoch": 13.12,
"learning_rate": 3.106332138590203e-05,
"loss": 0.3862,
"step": 820
},
{
"epoch": 13.28,
"learning_rate": 3.0764635603345284e-05,
"loss": 0.3845,
"step": 830
},
{
"epoch": 13.44,
"learning_rate": 3.046594982078853e-05,
"loss": 0.4347,
"step": 840
},
{
"epoch": 13.6,
"learning_rate": 3.016726403823178e-05,
"loss": 0.413,
"step": 850
},
{
"epoch": 13.76,
"learning_rate": 2.9868578255675032e-05,
"loss": 0.3976,
"step": 860
},
{
"epoch": 13.92,
"learning_rate": 2.9569892473118284e-05,
"loss": 0.4114,
"step": 870
},
{
"epoch": 14.0,
"eval_accuracy": 0.846674182638106,
"eval_f1": 0.8420494248065613,
"eval_loss": 0.48215728998184204,
"eval_precision": 0.8437277030586722,
"eval_recall": 0.846674182638106,
"eval_runtime": 11.553,
"eval_samples_per_second": 153.553,
"eval_steps_per_second": 2.424,
"step": 875
},
{
"epoch": 14.08,
"learning_rate": 2.9271206690561532e-05,
"loss": 0.3816,
"step": 880
},
{
"epoch": 14.24,
"learning_rate": 2.897252090800478e-05,
"loss": 0.3634,
"step": 890
},
{
"epoch": 14.4,
"learning_rate": 2.8673835125448028e-05,
"loss": 0.3671,
"step": 900
},
{
"epoch": 14.56,
"learning_rate": 2.8375149342891276e-05,
"loss": 0.3937,
"step": 910
},
{
"epoch": 14.72,
"learning_rate": 2.807646356033453e-05,
"loss": 0.4028,
"step": 920
},
{
"epoch": 14.88,
"learning_rate": 2.777777777777778e-05,
"loss": 0.3878,
"step": 930
},
{
"epoch": 14.99,
"eval_accuracy": 0.8573844419391207,
"eval_f1": 0.8549254367261584,
"eval_loss": 0.46154502034187317,
"eval_precision": 0.8552249826170021,
"eval_recall": 0.8573844419391207,
"eval_runtime": 11.5764,
"eval_samples_per_second": 153.242,
"eval_steps_per_second": 2.419,
"step": 937
},
{
"epoch": 15.04,
"learning_rate": 2.747909199522103e-05,
"loss": 0.3596,
"step": 940
},
{
"epoch": 15.2,
"learning_rate": 2.718040621266428e-05,
"loss": 0.3673,
"step": 950
},
{
"epoch": 15.36,
"learning_rate": 2.6881720430107527e-05,
"loss": 0.3635,
"step": 960
},
{
"epoch": 15.52,
"learning_rate": 2.6583034647550775e-05,
"loss": 0.3778,
"step": 970
},
{
"epoch": 15.68,
"learning_rate": 2.628434886499403e-05,
"loss": 0.3291,
"step": 980
},
{
"epoch": 15.84,
"learning_rate": 2.5985663082437278e-05,
"loss": 0.3723,
"step": 990
},
{
"epoch": 16.0,
"learning_rate": 2.5686977299880526e-05,
"loss": 0.3756,
"step": 1000
},
{
"epoch": 16.0,
"eval_accuracy": 0.8444193912063134,
"eval_f1": 0.8449182009174938,
"eval_loss": 0.5017289519309998,
"eval_precision": 0.8523010405389694,
"eval_recall": 0.8444193912063134,
"eval_runtime": 11.5898,
"eval_samples_per_second": 153.066,
"eval_steps_per_second": 2.416,
"step": 1000
},
{
"epoch": 16.16,
"learning_rate": 2.5388291517323774e-05,
"loss": 0.3709,
"step": 1010
},
{
"epoch": 16.32,
"learning_rate": 2.5089605734767026e-05,
"loss": 0.3172,
"step": 1020
},
{
"epoch": 16.48,
"learning_rate": 2.4790919952210277e-05,
"loss": 0.3342,
"step": 1030
},
{
"epoch": 16.64,
"learning_rate": 2.4492234169653525e-05,
"loss": 0.3633,
"step": 1040
},
{
"epoch": 16.8,
"learning_rate": 2.4193548387096777e-05,
"loss": 0.3496,
"step": 1050
},
{
"epoch": 16.96,
"learning_rate": 2.3894862604540025e-05,
"loss": 0.3056,
"step": 1060
},
{
"epoch": 16.99,
"eval_accuracy": 0.8517474633596392,
"eval_f1": 0.8501410314829453,
"eval_loss": 0.49099722504615784,
"eval_precision": 0.8494646040840952,
"eval_recall": 0.8517474633596392,
"eval_runtime": 11.5714,
"eval_samples_per_second": 153.309,
"eval_steps_per_second": 2.42,
"step": 1062
},
{
"epoch": 17.12,
"learning_rate": 2.3596176821983273e-05,
"loss": 0.3386,
"step": 1070
},
{
"epoch": 17.28,
"learning_rate": 2.3297491039426525e-05,
"loss": 0.3355,
"step": 1080
},
{
"epoch": 17.44,
"learning_rate": 2.2998805256869773e-05,
"loss": 0.3319,
"step": 1090
},
{
"epoch": 17.6,
"learning_rate": 2.270011947431302e-05,
"loss": 0.3403,
"step": 1100
},
{
"epoch": 17.76,
"learning_rate": 2.2401433691756272e-05,
"loss": 0.3149,
"step": 1110
},
{
"epoch": 17.92,
"learning_rate": 2.2102747909199524e-05,
"loss": 0.3255,
"step": 1120
},
{
"epoch": 18.0,
"eval_accuracy": 0.8523111612175873,
"eval_f1": 0.8490918881883218,
"eval_loss": 0.5205659866333008,
"eval_precision": 0.8505355127557268,
"eval_recall": 0.8523111612175873,
"eval_runtime": 11.544,
"eval_samples_per_second": 153.673,
"eval_steps_per_second": 2.426,
"step": 1125
},
{
"epoch": 18.08,
"learning_rate": 2.1804062126642775e-05,
"loss": 0.3233,
"step": 1130
},
{
"epoch": 18.24,
"learning_rate": 2.1505376344086024e-05,
"loss": 0.291,
"step": 1140
},
{
"epoch": 18.4,
"learning_rate": 2.120669056152927e-05,
"loss": 0.2846,
"step": 1150
},
{
"epoch": 18.56,
"learning_rate": 2.0908004778972523e-05,
"loss": 0.3011,
"step": 1160
},
{
"epoch": 18.72,
"learning_rate": 2.060931899641577e-05,
"loss": 0.2922,
"step": 1170
},
{
"epoch": 18.88,
"learning_rate": 2.031063321385902e-05,
"loss": 0.3224,
"step": 1180
},
{
"epoch": 18.99,
"eval_accuracy": 0.8449830890642616,
"eval_f1": 0.8437753213575151,
"eval_loss": 0.5066474080085754,
"eval_precision": 0.8470210763602011,
"eval_recall": 0.8449830890642616,
"eval_runtime": 11.5926,
"eval_samples_per_second": 153.029,
"eval_steps_per_second": 2.415,
"step": 1187
},
{
"epoch": 19.04,
"learning_rate": 2.001194743130227e-05,
"loss": 0.3068,
"step": 1190
},
{
"epoch": 19.2,
"learning_rate": 1.9713261648745522e-05,
"loss": 0.287,
"step": 1200
},
{
"epoch": 19.36,
"learning_rate": 1.941457586618877e-05,
"loss": 0.311,
"step": 1210
},
{
"epoch": 19.52,
"learning_rate": 1.9115890083632022e-05,
"loss": 0.3114,
"step": 1220
},
{
"epoch": 19.68,
"learning_rate": 1.881720430107527e-05,
"loss": 0.29,
"step": 1230
},
{
"epoch": 19.84,
"learning_rate": 1.8518518518518518e-05,
"loss": 0.301,
"step": 1240
},
{
"epoch": 20.0,
"learning_rate": 1.821983273596177e-05,
"loss": 0.2763,
"step": 1250
},
{
"epoch": 20.0,
"eval_accuracy": 0.8573844419391207,
"eval_f1": 0.8534152371552309,
"eval_loss": 0.5043460130691528,
"eval_precision": 0.851867459670644,
"eval_recall": 0.8573844419391207,
"eval_runtime": 11.6061,
"eval_samples_per_second": 152.85,
"eval_steps_per_second": 2.413,
"step": 1250
},
{
"epoch": 20.16,
"learning_rate": 1.7921146953405018e-05,
"loss": 0.2712,
"step": 1260
},
{
"epoch": 20.32,
"learning_rate": 1.7622461170848266e-05,
"loss": 0.2689,
"step": 1270
},
{
"epoch": 20.48,
"learning_rate": 1.7323775388291518e-05,
"loss": 0.2703,
"step": 1280
},
{
"epoch": 20.64,
"learning_rate": 1.702508960573477e-05,
"loss": 0.2561,
"step": 1290
},
{
"epoch": 20.8,
"learning_rate": 1.6726403823178017e-05,
"loss": 0.2805,
"step": 1300
},
{
"epoch": 20.96,
"learning_rate": 1.642771804062127e-05,
"loss": 0.2926,
"step": 1310
},
{
"epoch": 20.99,
"eval_accuracy": 0.85456595264938,
"eval_f1": 0.8511974166549313,
"eval_loss": 0.5345116853713989,
"eval_precision": 0.8542415938887548,
"eval_recall": 0.85456595264938,
"eval_runtime": 11.5631,
"eval_samples_per_second": 153.419,
"eval_steps_per_second": 2.421,
"step": 1312
},
{
"epoch": 21.12,
"learning_rate": 1.6129032258064517e-05,
"loss": 0.2513,
"step": 1320
},
{
"epoch": 21.28,
"learning_rate": 1.5830346475507768e-05,
"loss": 0.2595,
"step": 1330
},
{
"epoch": 21.44,
"learning_rate": 1.5531660692951016e-05,
"loss": 0.2733,
"step": 1340
},
{
"epoch": 21.6,
"learning_rate": 1.5232974910394265e-05,
"loss": 0.257,
"step": 1350
},
{
"epoch": 21.76,
"learning_rate": 1.4934289127837516e-05,
"loss": 0.2711,
"step": 1360
},
{
"epoch": 21.92,
"learning_rate": 1.4635603345280766e-05,
"loss": 0.2824,
"step": 1370
},
{
"epoch": 22.0,
"eval_accuracy": 0.8528748590755355,
"eval_f1": 0.851657855208825,
"eval_loss": 0.5319970846176147,
"eval_precision": 0.8523432836985173,
"eval_recall": 0.8528748590755355,
"eval_runtime": 11.8233,
"eval_samples_per_second": 150.043,
"eval_steps_per_second": 2.368,
"step": 1375
},
{
"epoch": 22.08,
"learning_rate": 1.4336917562724014e-05,
"loss": 0.2557,
"step": 1380
},
{
"epoch": 22.24,
"learning_rate": 1.4038231780167265e-05,
"loss": 0.2365,
"step": 1390
},
{
"epoch": 22.4,
"learning_rate": 1.3739545997610515e-05,
"loss": 0.2544,
"step": 1400
},
{
"epoch": 22.56,
"learning_rate": 1.3440860215053763e-05,
"loss": 0.2326,
"step": 1410
},
{
"epoch": 22.72,
"learning_rate": 1.3142174432497015e-05,
"loss": 0.2598,
"step": 1420
},
{
"epoch": 22.88,
"learning_rate": 1.2843488649940263e-05,
"loss": 0.2613,
"step": 1430
},
{
"epoch": 22.99,
"eval_accuracy": 0.8562570462232244,
"eval_f1": 0.8541634844649634,
"eval_loss": 0.5253913402557373,
"eval_precision": 0.8542880575760138,
"eval_recall": 0.8562570462232244,
"eval_runtime": 11.5478,
"eval_samples_per_second": 153.623,
"eval_steps_per_second": 2.425,
"step": 1437
},
{
"epoch": 23.04,
"learning_rate": 1.2544802867383513e-05,
"loss": 0.2452,
"step": 1440
},
{
"epoch": 23.2,
"learning_rate": 1.2246117084826763e-05,
"loss": 0.2346,
"step": 1450
},
{
"epoch": 23.36,
"learning_rate": 1.1947431302270013e-05,
"loss": 0.225,
"step": 1460
},
{
"epoch": 23.52,
"learning_rate": 1.1648745519713262e-05,
"loss": 0.2314,
"step": 1470
},
{
"epoch": 23.68,
"learning_rate": 1.135005973715651e-05,
"loss": 0.2745,
"step": 1480
},
{
"epoch": 23.84,
"learning_rate": 1.1051373954599762e-05,
"loss": 0.2164,
"step": 1490
},
{
"epoch": 24.0,
"learning_rate": 1.0752688172043012e-05,
"loss": 0.2292,
"step": 1500
},
{
"epoch": 24.0,
"eval_accuracy": 0.85456595264938,
"eval_f1": 0.852830948081783,
"eval_loss": 0.5552685856819153,
"eval_precision": 0.852933725021029,
"eval_recall": 0.85456595264938,
"eval_runtime": 11.8572,
"eval_samples_per_second": 149.613,
"eval_steps_per_second": 2.361,
"step": 1500
},
{
"epoch": 24.16,
"learning_rate": 1.0454002389486262e-05,
"loss": 0.2497,
"step": 1510
},
{
"epoch": 24.32,
"learning_rate": 1.015531660692951e-05,
"loss": 0.2573,
"step": 1520
},
{
"epoch": 24.48,
"learning_rate": 9.856630824372761e-06,
"loss": 0.2351,
"step": 1530
},
{
"epoch": 24.64,
"learning_rate": 9.557945041816011e-06,
"loss": 0.2116,
"step": 1540
},
{
"epoch": 24.8,
"learning_rate": 9.259259259259259e-06,
"loss": 0.2208,
"step": 1550
},
{
"epoch": 24.96,
"learning_rate": 8.960573476702509e-06,
"loss": 0.2313,
"step": 1560
},
{
"epoch": 24.99,
"eval_accuracy": 0.8602029312288614,
"eval_f1": 0.8593340542503702,
"eval_loss": 0.560276985168457,
"eval_precision": 0.8612458347601163,
"eval_recall": 0.8602029312288614,
"eval_runtime": 11.6228,
"eval_samples_per_second": 152.63,
"eval_steps_per_second": 2.409,
"step": 1562
},
{
"epoch": 25.12,
"learning_rate": 8.661887694145759e-06,
"loss": 0.2091,
"step": 1570
},
{
"epoch": 25.28,
"learning_rate": 8.363201911589009e-06,
"loss": 0.1934,
"step": 1580
},
{
"epoch": 25.44,
"learning_rate": 8.064516129032258e-06,
"loss": 0.2039,
"step": 1590
},
{
"epoch": 25.6,
"learning_rate": 7.765830346475508e-06,
"loss": 0.2231,
"step": 1600
},
{
"epoch": 25.76,
"learning_rate": 7.467144563918758e-06,
"loss": 0.2064,
"step": 1610
},
{
"epoch": 25.92,
"learning_rate": 7.168458781362007e-06,
"loss": 0.2143,
"step": 1620
},
{
"epoch": 26.0,
"eval_accuracy": 0.8669673055242391,
"eval_f1": 0.8650490800677053,
"eval_loss": 0.5267350673675537,
"eval_precision": 0.8644566119604725,
"eval_recall": 0.8669673055242391,
"eval_runtime": 11.5256,
"eval_samples_per_second": 153.918,
"eval_steps_per_second": 2.429,
"step": 1625
},
{
"epoch": 26.08,
"learning_rate": 6.869772998805258e-06,
"loss": 0.2168,
"step": 1630
},
{
"epoch": 26.24,
"learning_rate": 6.5710872162485075e-06,
"loss": 0.2172,
"step": 1640
},
{
"epoch": 26.4,
"learning_rate": 6.2724014336917564e-06,
"loss": 0.2146,
"step": 1650
},
{
"epoch": 26.56,
"learning_rate": 5.973715651135006e-06,
"loss": 0.1982,
"step": 1660
},
{
"epoch": 26.72,
"learning_rate": 5.675029868578255e-06,
"loss": 0.2162,
"step": 1670
},
{
"epoch": 26.88,
"learning_rate": 5.376344086021506e-06,
"loss": 0.2075,
"step": 1680
},
{
"epoch": 26.99,
"eval_accuracy": 0.8573844419391207,
"eval_f1": 0.8572925425596268,
"eval_loss": 0.573656439781189,
"eval_precision": 0.8588592723050006,
"eval_recall": 0.8573844419391207,
"eval_runtime": 11.5433,
"eval_samples_per_second": 153.683,
"eval_steps_per_second": 2.426,
"step": 1687
},
{
"epoch": 27.04,
"learning_rate": 5.077658303464755e-06,
"loss": 0.2048,
"step": 1690
},
{
"epoch": 27.2,
"learning_rate": 4.7789725209080055e-06,
"loss": 0.2057,
"step": 1700
},
{
"epoch": 27.36,
"learning_rate": 4.4802867383512545e-06,
"loss": 0.2204,
"step": 1710
},
{
"epoch": 27.52,
"learning_rate": 4.181600955794504e-06,
"loss": 0.2,
"step": 1720
},
{
"epoch": 27.68,
"learning_rate": 3.882915173237754e-06,
"loss": 0.1801,
"step": 1730
},
{
"epoch": 27.84,
"learning_rate": 3.5842293906810035e-06,
"loss": 0.2057,
"step": 1740
},
{
"epoch": 28.0,
"learning_rate": 3.2855436081242537e-06,
"loss": 0.2121,
"step": 1750
},
{
"epoch": 28.0,
"eval_accuracy": 0.8618940248027057,
"eval_f1": 0.8603620821894493,
"eval_loss": 0.5748315453529358,
"eval_precision": 0.8601032519818212,
"eval_recall": 0.8618940248027057,
"eval_runtime": 11.5649,
"eval_samples_per_second": 153.395,
"eval_steps_per_second": 2.421,
"step": 1750
},
{
"epoch": 28.16,
"learning_rate": 2.986857825567503e-06,
"loss": 0.1974,
"step": 1760
},
{
"epoch": 28.32,
"learning_rate": 2.688172043010753e-06,
"loss": 0.1966,
"step": 1770
},
{
"epoch": 28.48,
"learning_rate": 2.3894862604540028e-06,
"loss": 0.1864,
"step": 1780
},
{
"epoch": 28.64,
"learning_rate": 2.090800477897252e-06,
"loss": 0.1915,
"step": 1790
},
{
"epoch": 28.8,
"learning_rate": 1.7921146953405017e-06,
"loss": 0.205,
"step": 1800
},
{
"epoch": 28.96,
"learning_rate": 1.4934289127837516e-06,
"loss": 0.1944,
"step": 1810
},
{
"epoch": 28.99,
"eval_accuracy": 0.8647125140924464,
"eval_f1": 0.8624452336163977,
"eval_loss": 0.5666339993476868,
"eval_precision": 0.861750515041048,
"eval_recall": 0.8647125140924464,
"eval_runtime": 11.5608,
"eval_samples_per_second": 153.45,
"eval_steps_per_second": 2.422,
"step": 1812
},
{
"epoch": 29.12,
"learning_rate": 1.1947431302270014e-06,
"loss": 0.1903,
"step": 1820
},
{
"epoch": 29.28,
"learning_rate": 8.960573476702509e-07,
"loss": 0.1832,
"step": 1830
},
{
"epoch": 29.44,
"learning_rate": 5.973715651135007e-07,
"loss": 0.1881,
"step": 1840
},
{
"epoch": 29.6,
"learning_rate": 2.9868578255675034e-07,
"loss": 0.1915,
"step": 1850
},
{
"epoch": 29.76,
"learning_rate": 0.0,
"loss": 0.1866,
"step": 1860
},
{
"epoch": 29.76,
"eval_accuracy": 0.8607666290868095,
"eval_f1": 0.8588674347943951,
"eval_loss": 0.5676339268684387,
"eval_precision": 0.8582657177956912,
"eval_recall": 0.8607666290868095,
"eval_runtime": 11.591,
"eval_samples_per_second": 153.05,
"eval_steps_per_second": 2.416,
"step": 1860
},
{
"epoch": 29.76,
"step": 1860,
"total_flos": 4.852841995091313e+19,
"train_loss": 0.431233105223666,
"train_runtime": 6824.8754,
"train_samples_per_second": 70.164,
"train_steps_per_second": 0.273
}
],
"logging_steps": 10,
"max_steps": 1860,
"num_input_tokens_seen": 0,
"num_train_epochs": 30,
"save_steps": 500,
"total_flos": 4.852841995091313e+19,
"train_batch_size": 64,
"trial_name": null,
"trial_params": null
}