{ "best_metric": 0.8375910584575753, "best_model_checkpoint": "result/roberta-large-cls_before_pooler-sym_mlp-mlp_bert-bs64-gpu8-gs1-lr5e-5-m=stsb-norm0.05-l32-wiki-id-1epoch", "epoch": 0.8316274309109519, "global_step": 1625, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 4.9744114636642786e-05, "loss": 6.5887, "step": 10 }, { "epoch": 0.01, "learning_rate": 4.948822927328557e-05, "loss": 2.7792, "step": 20 }, { "epoch": 0.02, "learning_rate": 4.923234390992835e-05, "loss": 0.1035, "step": 30 }, { "epoch": 0.02, "learning_rate": 4.897645854657114e-05, "loss": 0.0265, "step": 40 }, { "epoch": 0.03, "learning_rate": 4.8720573183213926e-05, "loss": 0.0104, "step": 50 }, { "epoch": 0.03, "learning_rate": 4.846468781985671e-05, "loss": 0.0094, "step": 60 }, { "epoch": 0.04, "learning_rate": 4.820880245649949e-05, "loss": 0.0064, "step": 70 }, { "epoch": 0.04, "learning_rate": 4.795291709314228e-05, "loss": 0.0038, "step": 80 }, { "epoch": 0.05, "learning_rate": 4.769703172978506e-05, "loss": 0.0062, "step": 90 }, { "epoch": 0.05, "learning_rate": 4.7441146366427843e-05, "loss": 0.0032, "step": 100 }, { "epoch": 0.06, "learning_rate": 4.718526100307063e-05, "loss": 0.0032, "step": 110 }, { "epoch": 0.06, "learning_rate": 4.692937563971341e-05, "loss": 0.0034, "step": 120 }, { "epoch": 0.06, "eval_avg_sts": 0.7766676482869428, "eval_sickr_spearman": 0.7413213856606112, "eval_stsb_spearman": 0.8120139109132745, "step": 125 }, { "epoch": 0.07, "learning_rate": 4.6673490276356194e-05, "loss": 0.0034, "step": 130 }, { "epoch": 0.07, "learning_rate": 4.641760491299898e-05, "loss": 0.0027, "step": 140 }, { "epoch": 0.08, "learning_rate": 4.616171954964176e-05, "loss": 0.0028, "step": 150 }, { "epoch": 0.08, "learning_rate": 4.5905834186284544e-05, "loss": 0.0024, "step": 160 }, { "epoch": 0.09, "learning_rate": 4.564994882292733e-05, "loss": 0.0014, "step": 170 }, { "epoch": 0.09, "learning_rate": 4.539406345957011e-05, "loss": 0.0023, "step": 180 }, { "epoch": 0.1, "learning_rate": 4.5138178096212894e-05, "loss": 0.0022, "step": 190 }, { "epoch": 0.1, "learning_rate": 4.4882292732855685e-05, "loss": 0.0017, "step": 200 }, { "epoch": 0.11, "learning_rate": 4.462640736949847e-05, "loss": 0.0022, "step": 210 }, { "epoch": 0.11, "learning_rate": 4.437052200614125e-05, "loss": 0.0016, "step": 220 }, { "epoch": 0.12, "learning_rate": 4.4114636642784035e-05, "loss": 0.0013, "step": 230 }, { "epoch": 0.12, "learning_rate": 4.385875127942682e-05, "loss": 0.002, "step": 240 }, { "epoch": 0.13, "learning_rate": 4.36028659160696e-05, "loss": 0.0018, "step": 250 }, { "epoch": 0.13, "eval_avg_sts": 0.7848935997485145, "eval_sickr_spearman": 0.7482028495947509, "eval_stsb_spearman": 0.8215843499022781, "step": 250 }, { "epoch": 0.13, "learning_rate": 4.3346980552712385e-05, "loss": 0.0019, "step": 260 }, { "epoch": 0.14, "learning_rate": 4.309109518935517e-05, "loss": 0.0019, "step": 270 }, { "epoch": 0.14, "learning_rate": 4.283520982599796e-05, "loss": 0.0014, "step": 280 }, { "epoch": 0.15, "learning_rate": 4.257932446264074e-05, "loss": 0.0018, "step": 290 }, { "epoch": 0.15, "learning_rate": 4.2323439099283526e-05, "loss": 0.0026, "step": 300 }, { "epoch": 0.16, "learning_rate": 4.206755373592631e-05, "loss": 0.0022, "step": 310 }, { "epoch": 0.16, "learning_rate": 4.181166837256909e-05, "loss": 0.0022, "step": 320 }, { "epoch": 0.17, "learning_rate": 4.1555783009211876e-05, "loss": 0.0015, "step": 330 }, { "epoch": 0.17, "learning_rate": 4.129989764585466e-05, "loss": 0.001, "step": 340 }, { "epoch": 0.18, "learning_rate": 4.104401228249744e-05, "loss": 0.0011, "step": 350 }, { "epoch": 0.18, "learning_rate": 4.0788126919140226e-05, "loss": 0.0025, "step": 360 }, { "epoch": 0.19, "learning_rate": 4.053224155578301e-05, "loss": 0.0011, "step": 370 }, { "epoch": 0.19, "eval_avg_sts": 0.7915789886236229, "eval_sickr_spearman": 0.7516040759808881, "eval_stsb_spearman": 0.8315539012663576, "step": 375 }, { "epoch": 0.19, "learning_rate": 4.027635619242579e-05, "loss": 0.0017, "step": 380 }, { "epoch": 0.2, "learning_rate": 4.0020470829068577e-05, "loss": 0.0018, "step": 390 }, { "epoch": 0.2, "learning_rate": 3.976458546571136e-05, "loss": 0.002, "step": 400 }, { "epoch": 0.21, "learning_rate": 3.9508700102354143e-05, "loss": 0.0012, "step": 410 }, { "epoch": 0.21, "learning_rate": 3.925281473899693e-05, "loss": 0.0008, "step": 420 }, { "epoch": 0.22, "learning_rate": 3.899692937563972e-05, "loss": 0.0015, "step": 430 }, { "epoch": 0.23, "learning_rate": 3.87410440122825e-05, "loss": 0.0009, "step": 440 }, { "epoch": 0.23, "learning_rate": 3.8485158648925284e-05, "loss": 0.0009, "step": 450 }, { "epoch": 0.24, "learning_rate": 3.822927328556807e-05, "loss": 0.0017, "step": 460 }, { "epoch": 0.24, "learning_rate": 3.797338792221085e-05, "loss": 0.0019, "step": 470 }, { "epoch": 0.25, "learning_rate": 3.7717502558853634e-05, "loss": 0.0009, "step": 480 }, { "epoch": 0.25, "learning_rate": 3.746161719549642e-05, "loss": 0.0017, "step": 490 }, { "epoch": 0.26, "learning_rate": 3.72057318321392e-05, "loss": 0.0016, "step": 500 }, { "epoch": 0.26, "eval_avg_sts": 0.7912523903898834, "eval_sickr_spearman": 0.7523470690890343, "eval_stsb_spearman": 0.8301577116907326, "step": 500 }, { "epoch": 0.26, "learning_rate": 3.694984646878199e-05, "loss": 0.0005, "step": 510 }, { "epoch": 0.27, "learning_rate": 3.6693961105424775e-05, "loss": 0.0015, "step": 520 }, { "epoch": 0.27, "learning_rate": 3.643807574206756e-05, "loss": 0.0014, "step": 530 }, { "epoch": 0.28, "learning_rate": 3.618219037871034e-05, "loss": 0.0032, "step": 540 }, { "epoch": 0.28, "learning_rate": 3.5926305015353125e-05, "loss": 0.0029, "step": 550 }, { "epoch": 0.29, "learning_rate": 3.567041965199591e-05, "loss": 0.0021, "step": 560 }, { "epoch": 0.29, "learning_rate": 3.541453428863869e-05, "loss": 0.0007, "step": 570 }, { "epoch": 0.3, "learning_rate": 3.5158648925281475e-05, "loss": 0.0019, "step": 580 }, { "epoch": 0.3, "learning_rate": 3.4902763561924265e-05, "loss": 0.0015, "step": 590 }, { "epoch": 0.31, "learning_rate": 3.464687819856704e-05, "loss": 0.0013, "step": 600 }, { "epoch": 0.31, "learning_rate": 3.4390992835209826e-05, "loss": 0.0009, "step": 610 }, { "epoch": 0.32, "learning_rate": 3.413510747185261e-05, "loss": 0.0013, "step": 620 }, { "epoch": 0.32, "eval_avg_sts": 0.7924654796499744, "eval_sickr_spearman": 0.7521532635947314, "eval_stsb_spearman": 0.8327776957052173, "step": 625 }, { "epoch": 0.32, "learning_rate": 3.387922210849539e-05, "loss": 0.0016, "step": 630 }, { "epoch": 0.33, "learning_rate": 3.3623336745138176e-05, "loss": 0.0005, "step": 640 }, { "epoch": 0.33, "learning_rate": 3.336745138178096e-05, "loss": 0.001, "step": 650 }, { "epoch": 0.34, "learning_rate": 3.311156601842374e-05, "loss": 0.0012, "step": 660 }, { "epoch": 0.34, "learning_rate": 3.285568065506653e-05, "loss": 0.002, "step": 670 }, { "epoch": 0.35, "learning_rate": 3.2599795291709316e-05, "loss": 0.0009, "step": 680 }, { "epoch": 0.35, "learning_rate": 3.23439099283521e-05, "loss": 0.0009, "step": 690 }, { "epoch": 0.36, "learning_rate": 3.208802456499488e-05, "loss": 0.0018, "step": 700 }, { "epoch": 0.36, "learning_rate": 3.183213920163767e-05, "loss": 0.0009, "step": 710 }, { "epoch": 0.37, "learning_rate": 3.157625383828045e-05, "loss": 0.0023, "step": 720 }, { "epoch": 0.37, "learning_rate": 3.1320368474923234e-05, "loss": 0.0009, "step": 730 }, { "epoch": 0.38, "learning_rate": 3.106448311156602e-05, "loss": 0.0014, "step": 740 }, { "epoch": 0.38, "learning_rate": 3.080859774820881e-05, "loss": 0.001, "step": 750 }, { "epoch": 0.38, "eval_avg_sts": 0.791243313094975, "eval_sickr_spearman": 0.7516190136534353, "eval_stsb_spearman": 0.8308676125365148, "step": 750 }, { "epoch": 0.39, "learning_rate": 3.055271238485159e-05, "loss": 0.0014, "step": 760 }, { "epoch": 0.39, "learning_rate": 3.0296827021494374e-05, "loss": 0.0008, "step": 770 }, { "epoch": 0.4, "learning_rate": 3.0040941658137157e-05, "loss": 0.0021, "step": 780 }, { "epoch": 0.4, "learning_rate": 2.978505629477994e-05, "loss": 0.0014, "step": 790 }, { "epoch": 0.41, "learning_rate": 2.9529170931422724e-05, "loss": 0.001, "step": 800 }, { "epoch": 0.41, "learning_rate": 2.9273285568065504e-05, "loss": 0.0004, "step": 810 }, { "epoch": 0.42, "learning_rate": 2.9017400204708288e-05, "loss": 0.0009, "step": 820 }, { "epoch": 0.42, "learning_rate": 2.8761514841351078e-05, "loss": 0.0004, "step": 830 }, { "epoch": 0.43, "learning_rate": 2.850562947799386e-05, "loss": 0.0007, "step": 840 }, { "epoch": 0.44, "learning_rate": 2.8249744114636645e-05, "loss": 0.0007, "step": 850 }, { "epoch": 0.44, "learning_rate": 2.7993858751279428e-05, "loss": 0.0013, "step": 860 }, { "epoch": 0.45, "learning_rate": 2.7737973387922212e-05, "loss": 0.0004, "step": 870 }, { "epoch": 0.45, "eval_avg_sts": 0.7907165330286824, "eval_sickr_spearman": 0.7509387971948688, "eval_stsb_spearman": 0.830494268862496, "step": 875 }, { "epoch": 0.45, "learning_rate": 2.7482088024564995e-05, "loss": 0.0011, "step": 880 }, { "epoch": 0.46, "learning_rate": 2.722620266120778e-05, "loss": 0.0007, "step": 890 }, { "epoch": 0.46, "learning_rate": 2.6970317297850565e-05, "loss": 0.0015, "step": 900 }, { "epoch": 0.47, "learning_rate": 2.671443193449335e-05, "loss": 0.0009, "step": 910 }, { "epoch": 0.47, "learning_rate": 2.6458546571136132e-05, "loss": 0.0011, "step": 920 }, { "epoch": 0.48, "learning_rate": 2.6202661207778916e-05, "loss": 0.0005, "step": 930 }, { "epoch": 0.48, "learning_rate": 2.59467758444217e-05, "loss": 0.0006, "step": 940 }, { "epoch": 0.49, "learning_rate": 2.5690890481064483e-05, "loss": 0.0009, "step": 950 }, { "epoch": 0.49, "learning_rate": 2.5435005117707266e-05, "loss": 0.0004, "step": 960 }, { "epoch": 0.5, "learning_rate": 2.517911975435005e-05, "loss": 0.0011, "step": 970 }, { "epoch": 0.5, "learning_rate": 2.4923234390992836e-05, "loss": 0.0006, "step": 980 }, { "epoch": 0.51, "learning_rate": 2.466734902763562e-05, "loss": 0.0008, "step": 990 }, { "epoch": 0.51, "learning_rate": 2.4411463664278406e-05, "loss": 0.0006, "step": 1000 }, { "epoch": 0.51, "eval_avg_sts": 0.7917366161729372, "eval_sickr_spearman": 0.7491932509064049, "eval_stsb_spearman": 0.8342799814394696, "step": 1000 }, { "epoch": 0.52, "learning_rate": 2.415557830092119e-05, "loss": 0.0006, "step": 1010 }, { "epoch": 0.52, "learning_rate": 2.3899692937563973e-05, "loss": 0.0007, "step": 1020 }, { "epoch": 0.53, "learning_rate": 2.3643807574206757e-05, "loss": 0.0007, "step": 1030 }, { "epoch": 0.53, "learning_rate": 2.338792221084954e-05, "loss": 0.0017, "step": 1040 }, { "epoch": 0.54, "learning_rate": 2.3132036847492324e-05, "loss": 0.0008, "step": 1050 }, { "epoch": 0.54, "learning_rate": 2.2876151484135107e-05, "loss": 0.0014, "step": 1060 }, { "epoch": 0.55, "learning_rate": 2.262026612077789e-05, "loss": 0.0003, "step": 1070 }, { "epoch": 0.55, "learning_rate": 2.2364380757420677e-05, "loss": 0.0009, "step": 1080 }, { "epoch": 0.56, "learning_rate": 2.210849539406346e-05, "loss": 0.0013, "step": 1090 }, { "epoch": 0.56, "learning_rate": 2.1852610030706244e-05, "loss": 0.001, "step": 1100 }, { "epoch": 0.57, "learning_rate": 2.1596724667349028e-05, "loss": 0.0004, "step": 1110 }, { "epoch": 0.57, "learning_rate": 2.1340839303991814e-05, "loss": 0.0005, "step": 1120 }, { "epoch": 0.58, "eval_avg_sts": 0.7931919187577379, "eval_sickr_spearman": 0.7501849490577976, "eval_stsb_spearman": 0.836198888457678, "step": 1125 }, { "epoch": 0.58, "learning_rate": 2.1084953940634598e-05, "loss": 0.0008, "step": 1130 }, { "epoch": 0.58, "learning_rate": 2.082906857727738e-05, "loss": 0.0018, "step": 1140 }, { "epoch": 0.59, "learning_rate": 2.0573183213920165e-05, "loss": 0.001, "step": 1150 }, { "epoch": 0.59, "learning_rate": 2.0317297850562948e-05, "loss": 0.0009, "step": 1160 }, { "epoch": 0.6, "learning_rate": 2.006141248720573e-05, "loss": 0.0005, "step": 1170 }, { "epoch": 0.6, "learning_rate": 1.9805527123848515e-05, "loss": 0.0009, "step": 1180 }, { "epoch": 0.61, "learning_rate": 1.95496417604913e-05, "loss": 0.0004, "step": 1190 }, { "epoch": 0.61, "learning_rate": 1.9293756397134085e-05, "loss": 0.0014, "step": 1200 }, { "epoch": 0.62, "learning_rate": 1.903787103377687e-05, "loss": 0.0012, "step": 1210 }, { "epoch": 0.62, "learning_rate": 1.8781985670419652e-05, "loss": 0.0014, "step": 1220 }, { "epoch": 0.63, "learning_rate": 1.8526100307062436e-05, "loss": 0.0008, "step": 1230 }, { "epoch": 0.63, "learning_rate": 1.8270214943705222e-05, "loss": 0.0007, "step": 1240 }, { "epoch": 0.64, "learning_rate": 1.8014329580348006e-05, "loss": 0.0007, "step": 1250 }, { "epoch": 0.64, "eval_avg_sts": 0.7889672204342415, "eval_sickr_spearman": 0.7468257979165204, "eval_stsb_spearman": 0.8311086429519627, "step": 1250 }, { "epoch": 0.64, "learning_rate": 1.775844421699079e-05, "loss": 0.0012, "step": 1260 }, { "epoch": 0.65, "learning_rate": 1.7502558853633573e-05, "loss": 0.0003, "step": 1270 }, { "epoch": 0.66, "learning_rate": 1.7246673490276356e-05, "loss": 0.0009, "step": 1280 }, { "epoch": 0.66, "learning_rate": 1.699078812691914e-05, "loss": 0.001, "step": 1290 }, { "epoch": 0.67, "learning_rate": 1.6734902763561923e-05, "loss": 0.0011, "step": 1300 }, { "epoch": 0.67, "learning_rate": 1.647901740020471e-05, "loss": 0.0013, "step": 1310 }, { "epoch": 0.68, "learning_rate": 1.6223132036847493e-05, "loss": 0.0012, "step": 1320 }, { "epoch": 0.68, "learning_rate": 1.5967246673490277e-05, "loss": 0.001, "step": 1330 }, { "epoch": 0.69, "learning_rate": 1.571136131013306e-05, "loss": 0.0015, "step": 1340 }, { "epoch": 0.69, "learning_rate": 1.5455475946775847e-05, "loss": 0.0017, "step": 1350 }, { "epoch": 0.7, "learning_rate": 1.519959058341863e-05, "loss": 0.0004, "step": 1360 }, { "epoch": 0.7, "learning_rate": 1.4943705220061414e-05, "loss": 0.0008, "step": 1370 }, { "epoch": 0.7, "eval_avg_sts": 0.7906249920381556, "eval_sickr_spearman": 0.747464803690053, "eval_stsb_spearman": 0.8337851803862584, "step": 1375 }, { "epoch": 0.71, "learning_rate": 1.4687819856704197e-05, "loss": 0.0008, "step": 1380 }, { "epoch": 0.71, "learning_rate": 1.4431934493346982e-05, "loss": 0.0016, "step": 1390 }, { "epoch": 0.72, "learning_rate": 1.4176049129989766e-05, "loss": 0.0006, "step": 1400 }, { "epoch": 0.72, "learning_rate": 1.392016376663255e-05, "loss": 0.0008, "step": 1410 }, { "epoch": 0.73, "learning_rate": 1.3664278403275333e-05, "loss": 0.001, "step": 1420 }, { "epoch": 0.73, "learning_rate": 1.3408393039918118e-05, "loss": 0.0002, "step": 1430 }, { "epoch": 0.74, "learning_rate": 1.3152507676560901e-05, "loss": 0.003, "step": 1440 }, { "epoch": 0.74, "learning_rate": 1.2896622313203685e-05, "loss": 0.0008, "step": 1450 }, { "epoch": 0.75, "learning_rate": 1.2640736949846468e-05, "loss": 0.0017, "step": 1460 }, { "epoch": 0.75, "learning_rate": 1.2384851586489253e-05, "loss": 0.0008, "step": 1470 }, { "epoch": 0.76, "learning_rate": 1.2128966223132038e-05, "loss": 0.0003, "step": 1480 }, { "epoch": 0.76, "learning_rate": 1.1873080859774822e-05, "loss": 0.0005, "step": 1490 }, { "epoch": 0.77, "learning_rate": 1.1617195496417605e-05, "loss": 0.0004, "step": 1500 }, { "epoch": 0.77, "eval_avg_sts": 0.7942094737119515, "eval_sickr_spearman": 0.7523370786199352, "eval_stsb_spearman": 0.8360818688039678, "step": 1500 }, { "epoch": 0.77, "learning_rate": 1.1361310133060389e-05, "loss": 0.0005, "step": 1510 }, { "epoch": 0.78, "learning_rate": 1.1105424769703174e-05, "loss": 0.0004, "step": 1520 }, { "epoch": 0.78, "learning_rate": 1.0849539406345957e-05, "loss": 0.0011, "step": 1530 }, { "epoch": 0.79, "learning_rate": 1.0593654042988742e-05, "loss": 0.0005, "step": 1540 }, { "epoch": 0.79, "learning_rate": 1.0337768679631526e-05, "loss": 0.0012, "step": 1550 }, { "epoch": 0.8, "learning_rate": 1.0081883316274309e-05, "loss": 0.0011, "step": 1560 }, { "epoch": 0.8, "learning_rate": 9.825997952917093e-06, "loss": 0.001, "step": 1570 }, { "epoch": 0.81, "learning_rate": 9.570112589559878e-06, "loss": 0.001, "step": 1580 }, { "epoch": 0.81, "learning_rate": 9.314227226202661e-06, "loss": 0.0008, "step": 1590 }, { "epoch": 0.82, "learning_rate": 9.058341862845446e-06, "loss": 0.0009, "step": 1600 }, { "epoch": 0.82, "learning_rate": 8.80245649948823e-06, "loss": 0.0005, "step": 1610 }, { "epoch": 0.83, "learning_rate": 8.546571136131013e-06, "loss": 0.0012, "step": 1620 }, { "epoch": 0.83, "eval_avg_sts": 0.7953541291135341, "eval_sickr_spearman": 0.7531171997694929, "eval_stsb_spearman": 0.8375910584575753, "step": 1625 } ], "max_steps": 1954, "num_train_epochs": 1, "total_flos": 0, "trial_name": null, "trial_params": null }