|
{
|
|
"best_metric": 0.9340658097220965,
|
|
"best_model_checkpoint": "batch-size-16_FFPP-c23_1FPS_faces-expand-0-aligned_unaugmentation\\checkpoint-1381",
|
|
"epoch": 1.0,
|
|
"eval_steps": 500,
|
|
"global_step": 1381,
|
|
"is_hyper_param_search": false,
|
|
"is_local_process_zero": true,
|
|
"is_world_process_zero": true,
|
|
"log_history": [
|
|
{
|
|
"epoch": 0.01,
|
|
"grad_norm": 3.2925620079040527,
|
|
"learning_rate": 3.5971223021582732e-06,
|
|
"loss": 0.628,
|
|
"step": 10
|
|
},
|
|
{
|
|
"epoch": 0.01,
|
|
"grad_norm": 2.3982741832733154,
|
|
"learning_rate": 7.1942446043165465e-06,
|
|
"loss": 0.5814,
|
|
"step": 20
|
|
},
|
|
{
|
|
"epoch": 0.02,
|
|
"grad_norm": 2.776655673980713,
|
|
"learning_rate": 1.0791366906474821e-05,
|
|
"loss": 0.5151,
|
|
"step": 30
|
|
},
|
|
{
|
|
"epoch": 0.03,
|
|
"grad_norm": 2.050455331802368,
|
|
"learning_rate": 1.4388489208633093e-05,
|
|
"loss": 0.5308,
|
|
"step": 40
|
|
},
|
|
{
|
|
"epoch": 0.04,
|
|
"grad_norm": 2.005922794342041,
|
|
"learning_rate": 1.7985611510791367e-05,
|
|
"loss": 0.5363,
|
|
"step": 50
|
|
},
|
|
{
|
|
"epoch": 0.04,
|
|
"grad_norm": 1.7546412944793701,
|
|
"learning_rate": 2.1582733812949642e-05,
|
|
"loss": 0.4718,
|
|
"step": 60
|
|
},
|
|
{
|
|
"epoch": 0.05,
|
|
"grad_norm": 3.74489426612854,
|
|
"learning_rate": 2.5179856115107914e-05,
|
|
"loss": 0.5125,
|
|
"step": 70
|
|
},
|
|
{
|
|
"epoch": 0.06,
|
|
"grad_norm": 2.5463011264801025,
|
|
"learning_rate": 2.8776978417266186e-05,
|
|
"loss": 0.505,
|
|
"step": 80
|
|
},
|
|
{
|
|
"epoch": 0.07,
|
|
"grad_norm": 5.876346588134766,
|
|
"learning_rate": 3.237410071942446e-05,
|
|
"loss": 0.5294,
|
|
"step": 90
|
|
},
|
|
{
|
|
"epoch": 0.07,
|
|
"grad_norm": 3.9089746475219727,
|
|
"learning_rate": 3.597122302158273e-05,
|
|
"loss": 0.5465,
|
|
"step": 100
|
|
},
|
|
{
|
|
"epoch": 0.08,
|
|
"grad_norm": 5.623407363891602,
|
|
"learning_rate": 3.956834532374101e-05,
|
|
"loss": 0.5042,
|
|
"step": 110
|
|
},
|
|
{
|
|
"epoch": 0.09,
|
|
"grad_norm": 4.3956780433654785,
|
|
"learning_rate": 4.3165467625899284e-05,
|
|
"loss": 0.5046,
|
|
"step": 120
|
|
},
|
|
{
|
|
"epoch": 0.09,
|
|
"grad_norm": 4.22213077545166,
|
|
"learning_rate": 4.676258992805755e-05,
|
|
"loss": 0.4896,
|
|
"step": 130
|
|
},
|
|
{
|
|
"epoch": 0.1,
|
|
"grad_norm": 5.061640739440918,
|
|
"learning_rate": 4.99597423510467e-05,
|
|
"loss": 0.4615,
|
|
"step": 140
|
|
},
|
|
{
|
|
"epoch": 0.11,
|
|
"grad_norm": 3.777334213256836,
|
|
"learning_rate": 4.9557165861513685e-05,
|
|
"loss": 0.4725,
|
|
"step": 150
|
|
},
|
|
{
|
|
"epoch": 0.12,
|
|
"grad_norm": 8.781298637390137,
|
|
"learning_rate": 4.915458937198068e-05,
|
|
"loss": 0.4155,
|
|
"step": 160
|
|
},
|
|
{
|
|
"epoch": 0.12,
|
|
"grad_norm": 6.117956638336182,
|
|
"learning_rate": 4.875201288244767e-05,
|
|
"loss": 0.4209,
|
|
"step": 170
|
|
},
|
|
{
|
|
"epoch": 0.13,
|
|
"grad_norm": 3.1794047355651855,
|
|
"learning_rate": 4.834943639291466e-05,
|
|
"loss": 0.4149,
|
|
"step": 180
|
|
},
|
|
{
|
|
"epoch": 0.14,
|
|
"grad_norm": 4.002980709075928,
|
|
"learning_rate": 4.7946859903381646e-05,
|
|
"loss": 0.5147,
|
|
"step": 190
|
|
},
|
|
{
|
|
"epoch": 0.14,
|
|
"grad_norm": 2.6848464012145996,
|
|
"learning_rate": 4.7544283413848634e-05,
|
|
"loss": 0.4649,
|
|
"step": 200
|
|
},
|
|
{
|
|
"epoch": 0.15,
|
|
"grad_norm": 5.1571807861328125,
|
|
"learning_rate": 4.714170692431562e-05,
|
|
"loss": 0.4018,
|
|
"step": 210
|
|
},
|
|
{
|
|
"epoch": 0.16,
|
|
"grad_norm": 7.3869309425354,
|
|
"learning_rate": 4.673913043478261e-05,
|
|
"loss": 0.3477,
|
|
"step": 220
|
|
},
|
|
{
|
|
"epoch": 0.17,
|
|
"grad_norm": 5.493707656860352,
|
|
"learning_rate": 4.63365539452496e-05,
|
|
"loss": 0.3752,
|
|
"step": 230
|
|
},
|
|
{
|
|
"epoch": 0.17,
|
|
"grad_norm": 6.3577070236206055,
|
|
"learning_rate": 4.593397745571659e-05,
|
|
"loss": 0.3779,
|
|
"step": 240
|
|
},
|
|
{
|
|
"epoch": 0.18,
|
|
"grad_norm": 6.673830986022949,
|
|
"learning_rate": 4.553140096618358e-05,
|
|
"loss": 0.4141,
|
|
"step": 250
|
|
},
|
|
{
|
|
"epoch": 0.19,
|
|
"grad_norm": 3.704331636428833,
|
|
"learning_rate": 4.5128824476650565e-05,
|
|
"loss": 0.3926,
|
|
"step": 260
|
|
},
|
|
{
|
|
"epoch": 0.2,
|
|
"grad_norm": 5.406624794006348,
|
|
"learning_rate": 4.4726247987117554e-05,
|
|
"loss": 0.3742,
|
|
"step": 270
|
|
},
|
|
{
|
|
"epoch": 0.2,
|
|
"grad_norm": 5.6158528327941895,
|
|
"learning_rate": 4.432367149758454e-05,
|
|
"loss": 0.3715,
|
|
"step": 280
|
|
},
|
|
{
|
|
"epoch": 0.21,
|
|
"grad_norm": 5.550079345703125,
|
|
"learning_rate": 4.392109500805153e-05,
|
|
"loss": 0.3882,
|
|
"step": 290
|
|
},
|
|
{
|
|
"epoch": 0.22,
|
|
"grad_norm": 5.96751594543457,
|
|
"learning_rate": 4.351851851851852e-05,
|
|
"loss": 0.3615,
|
|
"step": 300
|
|
},
|
|
{
|
|
"epoch": 0.22,
|
|
"grad_norm": 5.48460054397583,
|
|
"learning_rate": 4.3115942028985515e-05,
|
|
"loss": 0.3233,
|
|
"step": 310
|
|
},
|
|
{
|
|
"epoch": 0.23,
|
|
"grad_norm": 5.41163969039917,
|
|
"learning_rate": 4.2713365539452496e-05,
|
|
"loss": 0.3723,
|
|
"step": 320
|
|
},
|
|
{
|
|
"epoch": 0.24,
|
|
"grad_norm": 6.076368808746338,
|
|
"learning_rate": 4.2310789049919485e-05,
|
|
"loss": 0.3684,
|
|
"step": 330
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"grad_norm": 9.160358428955078,
|
|
"learning_rate": 4.1908212560386474e-05,
|
|
"loss": 0.3381,
|
|
"step": 340
|
|
},
|
|
{
|
|
"epoch": 0.25,
|
|
"grad_norm": 5.847888469696045,
|
|
"learning_rate": 4.150563607085346e-05,
|
|
"loss": 0.3669,
|
|
"step": 350
|
|
},
|
|
{
|
|
"epoch": 0.26,
|
|
"grad_norm": 9.13357925415039,
|
|
"learning_rate": 4.110305958132045e-05,
|
|
"loss": 0.3717,
|
|
"step": 360
|
|
},
|
|
{
|
|
"epoch": 0.27,
|
|
"grad_norm": 6.672016143798828,
|
|
"learning_rate": 4.070048309178744e-05,
|
|
"loss": 0.3121,
|
|
"step": 370
|
|
},
|
|
{
|
|
"epoch": 0.28,
|
|
"grad_norm": 5.612246513366699,
|
|
"learning_rate": 4.0297906602254434e-05,
|
|
"loss": 0.3004,
|
|
"step": 380
|
|
},
|
|
{
|
|
"epoch": 0.28,
|
|
"grad_norm": 4.514873027801514,
|
|
"learning_rate": 3.9895330112721416e-05,
|
|
"loss": 0.3153,
|
|
"step": 390
|
|
},
|
|
{
|
|
"epoch": 0.29,
|
|
"grad_norm": 4.649542331695557,
|
|
"learning_rate": 3.9492753623188405e-05,
|
|
"loss": 0.3332,
|
|
"step": 400
|
|
},
|
|
{
|
|
"epoch": 0.3,
|
|
"grad_norm": 8.960000038146973,
|
|
"learning_rate": 3.90901771336554e-05,
|
|
"loss": 0.3117,
|
|
"step": 410
|
|
},
|
|
{
|
|
"epoch": 0.3,
|
|
"grad_norm": 4.450685977935791,
|
|
"learning_rate": 3.868760064412238e-05,
|
|
"loss": 0.2995,
|
|
"step": 420
|
|
},
|
|
{
|
|
"epoch": 0.31,
|
|
"grad_norm": 7.965936183929443,
|
|
"learning_rate": 3.828502415458937e-05,
|
|
"loss": 0.3106,
|
|
"step": 430
|
|
},
|
|
{
|
|
"epoch": 0.32,
|
|
"grad_norm": 5.865048885345459,
|
|
"learning_rate": 3.7882447665056365e-05,
|
|
"loss": 0.3165,
|
|
"step": 440
|
|
},
|
|
{
|
|
"epoch": 0.33,
|
|
"grad_norm": 6.963225841522217,
|
|
"learning_rate": 3.7479871175523354e-05,
|
|
"loss": 0.318,
|
|
"step": 450
|
|
},
|
|
{
|
|
"epoch": 0.33,
|
|
"grad_norm": 6.650681018829346,
|
|
"learning_rate": 3.7077294685990336e-05,
|
|
"loss": 0.3081,
|
|
"step": 460
|
|
},
|
|
{
|
|
"epoch": 0.34,
|
|
"grad_norm": 4.959587574005127,
|
|
"learning_rate": 3.667471819645733e-05,
|
|
"loss": 0.3109,
|
|
"step": 470
|
|
},
|
|
{
|
|
"epoch": 0.35,
|
|
"grad_norm": 5.5469183921813965,
|
|
"learning_rate": 3.627214170692432e-05,
|
|
"loss": 0.2793,
|
|
"step": 480
|
|
},
|
|
{
|
|
"epoch": 0.35,
|
|
"grad_norm": 10.897783279418945,
|
|
"learning_rate": 3.58695652173913e-05,
|
|
"loss": 0.2984,
|
|
"step": 490
|
|
},
|
|
{
|
|
"epoch": 0.36,
|
|
"grad_norm": 5.365045070648193,
|
|
"learning_rate": 3.5466988727858296e-05,
|
|
"loss": 0.2797,
|
|
"step": 500
|
|
},
|
|
{
|
|
"epoch": 0.37,
|
|
"grad_norm": 7.85481071472168,
|
|
"learning_rate": 3.5064412238325285e-05,
|
|
"loss": 0.3062,
|
|
"step": 510
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"grad_norm": 5.305897235870361,
|
|
"learning_rate": 3.4661835748792274e-05,
|
|
"loss": 0.3354,
|
|
"step": 520
|
|
},
|
|
{
|
|
"epoch": 0.38,
|
|
"grad_norm": 4.42397403717041,
|
|
"learning_rate": 3.425925925925926e-05,
|
|
"loss": 0.2454,
|
|
"step": 530
|
|
},
|
|
{
|
|
"epoch": 0.39,
|
|
"grad_norm": 3.9212090969085693,
|
|
"learning_rate": 3.385668276972625e-05,
|
|
"loss": 0.2564,
|
|
"step": 540
|
|
},
|
|
{
|
|
"epoch": 0.4,
|
|
"grad_norm": 7.084193229675293,
|
|
"learning_rate": 3.345410628019324e-05,
|
|
"loss": 0.2709,
|
|
"step": 550
|
|
},
|
|
{
|
|
"epoch": 0.41,
|
|
"grad_norm": 6.757343292236328,
|
|
"learning_rate": 3.305152979066023e-05,
|
|
"loss": 0.3131,
|
|
"step": 560
|
|
},
|
|
{
|
|
"epoch": 0.41,
|
|
"grad_norm": 8.7233304977417,
|
|
"learning_rate": 3.2648953301127216e-05,
|
|
"loss": 0.2729,
|
|
"step": 570
|
|
},
|
|
{
|
|
"epoch": 0.42,
|
|
"grad_norm": 6.0495100021362305,
|
|
"learning_rate": 3.2246376811594205e-05,
|
|
"loss": 0.2555,
|
|
"step": 580
|
|
},
|
|
{
|
|
"epoch": 0.43,
|
|
"grad_norm": 9.789109230041504,
|
|
"learning_rate": 3.184380032206119e-05,
|
|
"loss": 0.2636,
|
|
"step": 590
|
|
},
|
|
{
|
|
"epoch": 0.43,
|
|
"grad_norm": 8.58854866027832,
|
|
"learning_rate": 3.144122383252818e-05,
|
|
"loss": 0.2687,
|
|
"step": 600
|
|
},
|
|
{
|
|
"epoch": 0.44,
|
|
"grad_norm": 9.97248363494873,
|
|
"learning_rate": 3.103864734299517e-05,
|
|
"loss": 0.3013,
|
|
"step": 610
|
|
},
|
|
{
|
|
"epoch": 0.45,
|
|
"grad_norm": 25.611948013305664,
|
|
"learning_rate": 3.063607085346216e-05,
|
|
"loss": 0.3343,
|
|
"step": 620
|
|
},
|
|
{
|
|
"epoch": 0.46,
|
|
"grad_norm": 5.729062080383301,
|
|
"learning_rate": 3.023349436392915e-05,
|
|
"loss": 0.3457,
|
|
"step": 630
|
|
},
|
|
{
|
|
"epoch": 0.46,
|
|
"grad_norm": 7.5272216796875,
|
|
"learning_rate": 2.9830917874396136e-05,
|
|
"loss": 0.2456,
|
|
"step": 640
|
|
},
|
|
{
|
|
"epoch": 0.47,
|
|
"grad_norm": 7.43132209777832,
|
|
"learning_rate": 2.9428341384863124e-05,
|
|
"loss": 0.3469,
|
|
"step": 650
|
|
},
|
|
{
|
|
"epoch": 0.48,
|
|
"grad_norm": 7.458743572235107,
|
|
"learning_rate": 2.9025764895330116e-05,
|
|
"loss": 0.2884,
|
|
"step": 660
|
|
},
|
|
{
|
|
"epoch": 0.49,
|
|
"grad_norm": 6.541900634765625,
|
|
"learning_rate": 2.86231884057971e-05,
|
|
"loss": 0.2466,
|
|
"step": 670
|
|
},
|
|
{
|
|
"epoch": 0.49,
|
|
"grad_norm": 7.272885322570801,
|
|
"learning_rate": 2.822061191626409e-05,
|
|
"loss": 0.2345,
|
|
"step": 680
|
|
},
|
|
{
|
|
"epoch": 0.5,
|
|
"grad_norm": 10.58638858795166,
|
|
"learning_rate": 2.781803542673108e-05,
|
|
"loss": 0.2519,
|
|
"step": 690
|
|
},
|
|
{
|
|
"epoch": 0.51,
|
|
"grad_norm": 5.776723384857178,
|
|
"learning_rate": 2.741545893719807e-05,
|
|
"loss": 0.3001,
|
|
"step": 700
|
|
},
|
|
{
|
|
"epoch": 0.51,
|
|
"grad_norm": 5.556161880493164,
|
|
"learning_rate": 2.7012882447665055e-05,
|
|
"loss": 0.2799,
|
|
"step": 710
|
|
},
|
|
{
|
|
"epoch": 0.52,
|
|
"grad_norm": 5.732090950012207,
|
|
"learning_rate": 2.6610305958132047e-05,
|
|
"loss": 0.2197,
|
|
"step": 720
|
|
},
|
|
{
|
|
"epoch": 0.53,
|
|
"grad_norm": 17.775867462158203,
|
|
"learning_rate": 2.6207729468599036e-05,
|
|
"loss": 0.2938,
|
|
"step": 730
|
|
},
|
|
{
|
|
"epoch": 0.54,
|
|
"grad_norm": 19.362329483032227,
|
|
"learning_rate": 2.580515297906602e-05,
|
|
"loss": 0.2327,
|
|
"step": 740
|
|
},
|
|
{
|
|
"epoch": 0.54,
|
|
"grad_norm": 8.213146209716797,
|
|
"learning_rate": 2.5402576489533013e-05,
|
|
"loss": 0.2142,
|
|
"step": 750
|
|
},
|
|
{
|
|
"epoch": 0.55,
|
|
"grad_norm": 9.953818321228027,
|
|
"learning_rate": 2.5e-05,
|
|
"loss": 0.2403,
|
|
"step": 760
|
|
},
|
|
{
|
|
"epoch": 0.56,
|
|
"grad_norm": 7.865139007568359,
|
|
"learning_rate": 2.459742351046699e-05,
|
|
"loss": 0.2481,
|
|
"step": 770
|
|
},
|
|
{
|
|
"epoch": 0.56,
|
|
"grad_norm": 6.347203254699707,
|
|
"learning_rate": 2.4194847020933978e-05,
|
|
"loss": 0.2487,
|
|
"step": 780
|
|
},
|
|
{
|
|
"epoch": 0.57,
|
|
"grad_norm": 7.396281719207764,
|
|
"learning_rate": 2.3792270531400967e-05,
|
|
"loss": 0.2752,
|
|
"step": 790
|
|
},
|
|
{
|
|
"epoch": 0.58,
|
|
"grad_norm": 6.594508647918701,
|
|
"learning_rate": 2.338969404186796e-05,
|
|
"loss": 0.2389,
|
|
"step": 800
|
|
},
|
|
{
|
|
"epoch": 0.59,
|
|
"grad_norm": 7.60775089263916,
|
|
"learning_rate": 2.2987117552334944e-05,
|
|
"loss": 0.2621,
|
|
"step": 810
|
|
},
|
|
{
|
|
"epoch": 0.59,
|
|
"grad_norm": 13.000992774963379,
|
|
"learning_rate": 2.2584541062801932e-05,
|
|
"loss": 0.2691,
|
|
"step": 820
|
|
},
|
|
{
|
|
"epoch": 0.6,
|
|
"grad_norm": 11.92383098602295,
|
|
"learning_rate": 2.2181964573268924e-05,
|
|
"loss": 0.2484,
|
|
"step": 830
|
|
},
|
|
{
|
|
"epoch": 0.61,
|
|
"grad_norm": 10.424347877502441,
|
|
"learning_rate": 2.177938808373591e-05,
|
|
"loss": 0.2363,
|
|
"step": 840
|
|
},
|
|
{
|
|
"epoch": 0.62,
|
|
"grad_norm": 6.517064094543457,
|
|
"learning_rate": 2.13768115942029e-05,
|
|
"loss": 0.2489,
|
|
"step": 850
|
|
},
|
|
{
|
|
"epoch": 0.62,
|
|
"grad_norm": 6.191581726074219,
|
|
"learning_rate": 2.0974235104669886e-05,
|
|
"loss": 0.2632,
|
|
"step": 860
|
|
},
|
|
{
|
|
"epoch": 0.63,
|
|
"grad_norm": 7.870666027069092,
|
|
"learning_rate": 2.0571658615136878e-05,
|
|
"loss": 0.2527,
|
|
"step": 870
|
|
},
|
|
{
|
|
"epoch": 0.64,
|
|
"grad_norm": 7.662616729736328,
|
|
"learning_rate": 2.0169082125603867e-05,
|
|
"loss": 0.2355,
|
|
"step": 880
|
|
},
|
|
{
|
|
"epoch": 0.64,
|
|
"grad_norm": 11.486822128295898,
|
|
"learning_rate": 1.9766505636070852e-05,
|
|
"loss": 0.2169,
|
|
"step": 890
|
|
},
|
|
{
|
|
"epoch": 0.65,
|
|
"grad_norm": 12.382376670837402,
|
|
"learning_rate": 1.9363929146537844e-05,
|
|
"loss": 0.2483,
|
|
"step": 900
|
|
},
|
|
{
|
|
"epoch": 0.66,
|
|
"grad_norm": 10.059280395507812,
|
|
"learning_rate": 1.8961352657004832e-05,
|
|
"loss": 0.2159,
|
|
"step": 910
|
|
},
|
|
{
|
|
"epoch": 0.67,
|
|
"grad_norm": 18.096155166625977,
|
|
"learning_rate": 1.855877616747182e-05,
|
|
"loss": 0.2221,
|
|
"step": 920
|
|
},
|
|
{
|
|
"epoch": 0.67,
|
|
"grad_norm": 14.859966278076172,
|
|
"learning_rate": 1.815619967793881e-05,
|
|
"loss": 0.2633,
|
|
"step": 930
|
|
},
|
|
{
|
|
"epoch": 0.68,
|
|
"grad_norm": 10.2135648727417,
|
|
"learning_rate": 1.7753623188405798e-05,
|
|
"loss": 0.2817,
|
|
"step": 940
|
|
},
|
|
{
|
|
"epoch": 0.69,
|
|
"grad_norm": 15.141898155212402,
|
|
"learning_rate": 1.7351046698872786e-05,
|
|
"loss": 0.2481,
|
|
"step": 950
|
|
},
|
|
{
|
|
"epoch": 0.7,
|
|
"grad_norm": 11.335433959960938,
|
|
"learning_rate": 1.6948470209339775e-05,
|
|
"loss": 0.2236,
|
|
"step": 960
|
|
},
|
|
{
|
|
"epoch": 0.7,
|
|
"grad_norm": 10.088510513305664,
|
|
"learning_rate": 1.6545893719806767e-05,
|
|
"loss": 0.2564,
|
|
"step": 970
|
|
},
|
|
{
|
|
"epoch": 0.71,
|
|
"grad_norm": 11.138778686523438,
|
|
"learning_rate": 1.6143317230273752e-05,
|
|
"loss": 0.2301,
|
|
"step": 980
|
|
},
|
|
{
|
|
"epoch": 0.72,
|
|
"grad_norm": 9.613357543945312,
|
|
"learning_rate": 1.574074074074074e-05,
|
|
"loss": 0.2344,
|
|
"step": 990
|
|
},
|
|
{
|
|
"epoch": 0.72,
|
|
"grad_norm": 7.300370216369629,
|
|
"learning_rate": 1.533816425120773e-05,
|
|
"loss": 0.198,
|
|
"step": 1000
|
|
},
|
|
{
|
|
"epoch": 0.73,
|
|
"grad_norm": 8.088990211486816,
|
|
"learning_rate": 1.4935587761674719e-05,
|
|
"loss": 0.2218,
|
|
"step": 1010
|
|
},
|
|
{
|
|
"epoch": 0.74,
|
|
"grad_norm": 10.558765411376953,
|
|
"learning_rate": 1.4533011272141708e-05,
|
|
"loss": 0.2151,
|
|
"step": 1020
|
|
},
|
|
{
|
|
"epoch": 0.75,
|
|
"grad_norm": 13.40772819519043,
|
|
"learning_rate": 1.4130434782608694e-05,
|
|
"loss": 0.2308,
|
|
"step": 1030
|
|
},
|
|
{
|
|
"epoch": 0.75,
|
|
"grad_norm": 10.2852783203125,
|
|
"learning_rate": 1.3727858293075685e-05,
|
|
"loss": 0.195,
|
|
"step": 1040
|
|
},
|
|
{
|
|
"epoch": 0.76,
|
|
"grad_norm": 10.811461448669434,
|
|
"learning_rate": 1.3325281803542675e-05,
|
|
"loss": 0.1728,
|
|
"step": 1050
|
|
},
|
|
{
|
|
"epoch": 0.77,
|
|
"grad_norm": 6.249987602233887,
|
|
"learning_rate": 1.2922705314009662e-05,
|
|
"loss": 0.2358,
|
|
"step": 1060
|
|
},
|
|
{
|
|
"epoch": 0.77,
|
|
"grad_norm": 8.414009094238281,
|
|
"learning_rate": 1.2520128824476652e-05,
|
|
"loss": 0.2149,
|
|
"step": 1070
|
|
},
|
|
{
|
|
"epoch": 0.78,
|
|
"grad_norm": 7.3787689208984375,
|
|
"learning_rate": 1.211755233494364e-05,
|
|
"loss": 0.1711,
|
|
"step": 1080
|
|
},
|
|
{
|
|
"epoch": 0.79,
|
|
"grad_norm": 8.621789932250977,
|
|
"learning_rate": 1.1714975845410629e-05,
|
|
"loss": 0.1723,
|
|
"step": 1090
|
|
},
|
|
{
|
|
"epoch": 0.8,
|
|
"grad_norm": 7.638077735900879,
|
|
"learning_rate": 1.1312399355877617e-05,
|
|
"loss": 0.2104,
|
|
"step": 1100
|
|
},
|
|
{
|
|
"epoch": 0.8,
|
|
"grad_norm": 13.404261589050293,
|
|
"learning_rate": 1.0909822866344606e-05,
|
|
"loss": 0.2187,
|
|
"step": 1110
|
|
},
|
|
{
|
|
"epoch": 0.81,
|
|
"grad_norm": 15.54051399230957,
|
|
"learning_rate": 1.0507246376811594e-05,
|
|
"loss": 0.2342,
|
|
"step": 1120
|
|
},
|
|
{
|
|
"epoch": 0.82,
|
|
"grad_norm": 9.454499244689941,
|
|
"learning_rate": 1.0104669887278585e-05,
|
|
"loss": 0.2579,
|
|
"step": 1130
|
|
},
|
|
{
|
|
"epoch": 0.83,
|
|
"grad_norm": 8.479141235351562,
|
|
"learning_rate": 9.702093397745571e-06,
|
|
"loss": 0.1972,
|
|
"step": 1140
|
|
},
|
|
{
|
|
"epoch": 0.83,
|
|
"grad_norm": 9.336498260498047,
|
|
"learning_rate": 9.29951690821256e-06,
|
|
"loss": 0.1805,
|
|
"step": 1150
|
|
},
|
|
{
|
|
"epoch": 0.84,
|
|
"grad_norm": 7.811723709106445,
|
|
"learning_rate": 8.89694041867955e-06,
|
|
"loss": 0.2286,
|
|
"step": 1160
|
|
},
|
|
{
|
|
"epoch": 0.85,
|
|
"grad_norm": 6.146000862121582,
|
|
"learning_rate": 8.494363929146539e-06,
|
|
"loss": 0.2133,
|
|
"step": 1170
|
|
},
|
|
{
|
|
"epoch": 0.85,
|
|
"grad_norm": 10.112232208251953,
|
|
"learning_rate": 8.091787439613527e-06,
|
|
"loss": 0.2021,
|
|
"step": 1180
|
|
},
|
|
{
|
|
"epoch": 0.86,
|
|
"grad_norm": 12.056436538696289,
|
|
"learning_rate": 7.689210950080516e-06,
|
|
"loss": 0.1791,
|
|
"step": 1190
|
|
},
|
|
{
|
|
"epoch": 0.87,
|
|
"grad_norm": 9.647012710571289,
|
|
"learning_rate": 7.286634460547505e-06,
|
|
"loss": 0.1922,
|
|
"step": 1200
|
|
},
|
|
{
|
|
"epoch": 0.88,
|
|
"grad_norm": 5.874572277069092,
|
|
"learning_rate": 6.884057971014493e-06,
|
|
"loss": 0.205,
|
|
"step": 1210
|
|
},
|
|
{
|
|
"epoch": 0.88,
|
|
"grad_norm": 5.28275203704834,
|
|
"learning_rate": 6.481481481481481e-06,
|
|
"loss": 0.1951,
|
|
"step": 1220
|
|
},
|
|
{
|
|
"epoch": 0.89,
|
|
"grad_norm": 13.48544979095459,
|
|
"learning_rate": 6.078904991948471e-06,
|
|
"loss": 0.1871,
|
|
"step": 1230
|
|
},
|
|
{
|
|
"epoch": 0.9,
|
|
"grad_norm": 9.40962028503418,
|
|
"learning_rate": 5.676328502415459e-06,
|
|
"loss": 0.2078,
|
|
"step": 1240
|
|
},
|
|
{
|
|
"epoch": 0.91,
|
|
"grad_norm": 5.173947811126709,
|
|
"learning_rate": 5.273752012882448e-06,
|
|
"loss": 0.1712,
|
|
"step": 1250
|
|
},
|
|
{
|
|
"epoch": 0.91,
|
|
"grad_norm": 9.35919189453125,
|
|
"learning_rate": 4.871175523349437e-06,
|
|
"loss": 0.259,
|
|
"step": 1260
|
|
},
|
|
{
|
|
"epoch": 0.92,
|
|
"grad_norm": 9.794632911682129,
|
|
"learning_rate": 4.468599033816425e-06,
|
|
"loss": 0.2173,
|
|
"step": 1270
|
|
},
|
|
{
|
|
"epoch": 0.93,
|
|
"grad_norm": 8.138204574584961,
|
|
"learning_rate": 4.066022544283414e-06,
|
|
"loss": 0.1661,
|
|
"step": 1280
|
|
},
|
|
{
|
|
"epoch": 0.93,
|
|
"grad_norm": 7.02229118347168,
|
|
"learning_rate": 3.663446054750403e-06,
|
|
"loss": 0.1961,
|
|
"step": 1290
|
|
},
|
|
{
|
|
"epoch": 0.94,
|
|
"grad_norm": 8.913147926330566,
|
|
"learning_rate": 3.2608695652173914e-06,
|
|
"loss": 0.1956,
|
|
"step": 1300
|
|
},
|
|
{
|
|
"epoch": 0.95,
|
|
"grad_norm": 7.670225620269775,
|
|
"learning_rate": 2.85829307568438e-06,
|
|
"loss": 0.1706,
|
|
"step": 1310
|
|
},
|
|
{
|
|
"epoch": 0.96,
|
|
"grad_norm": 10.90648365020752,
|
|
"learning_rate": 2.455716586151369e-06,
|
|
"loss": 0.2055,
|
|
"step": 1320
|
|
},
|
|
{
|
|
"epoch": 0.96,
|
|
"grad_norm": 8.81640911102295,
|
|
"learning_rate": 2.053140096618358e-06,
|
|
"loss": 0.176,
|
|
"step": 1330
|
|
},
|
|
{
|
|
"epoch": 0.97,
|
|
"grad_norm": 7.678645610809326,
|
|
"learning_rate": 1.6505636070853463e-06,
|
|
"loss": 0.1897,
|
|
"step": 1340
|
|
},
|
|
{
|
|
"epoch": 0.98,
|
|
"grad_norm": 9.090548515319824,
|
|
"learning_rate": 1.247987117552335e-06,
|
|
"loss": 0.2035,
|
|
"step": 1350
|
|
},
|
|
{
|
|
"epoch": 0.98,
|
|
"grad_norm": 5.917918682098389,
|
|
"learning_rate": 8.454106280193237e-07,
|
|
"loss": 0.2102,
|
|
"step": 1360
|
|
},
|
|
{
|
|
"epoch": 0.99,
|
|
"grad_norm": 8.879599571228027,
|
|
"learning_rate": 4.428341384863124e-07,
|
|
"loss": 0.2003,
|
|
"step": 1370
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"grad_norm": 9.322664260864258,
|
|
"learning_rate": 4.025764895330113e-08,
|
|
"loss": 0.193,
|
|
"step": 1380
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"eval_accuracy": 0.9340658097220965,
|
|
"eval_f1": 0.9585606087544003,
|
|
"eval_loss": 0.16026902198791504,
|
|
"eval_precision": 0.9434178845400079,
|
|
"eval_recall": 0.9741973720348661,
|
|
"eval_roc_auc": 0.9777643781483449,
|
|
"eval_runtime": 159.6047,
|
|
"eval_samples_per_second": 553.718,
|
|
"eval_steps_per_second": 34.611,
|
|
"step": 1381
|
|
},
|
|
{
|
|
"epoch": 1.0,
|
|
"step": 1381,
|
|
"total_flos": 2.1966728826140099e+18,
|
|
"train_loss": 0.2970259432383502,
|
|
"train_runtime": 756.4932,
|
|
"train_samples_per_second": 116.823,
|
|
"train_steps_per_second": 1.826
|
|
}
|
|
],
|
|
"logging_steps": 10,
|
|
"max_steps": 1381,
|
|
"num_input_tokens_seen": 0,
|
|
"num_train_epochs": 1,
|
|
"save_steps": 500,
|
|
"total_flos": 2.1966728826140099e+18,
|
|
"train_batch_size": 16,
|
|
"trial_name": null,
|
|
"trial_params": null
|
|
}
|
|
|