1128 lines
44 KiB
JSON
1128 lines
44 KiB
JSON
|
{
|
||
|
|
"best_global_step": null,
|
||
|
|
"best_metric": null,
|
||
|
|
"best_model_checkpoint": null,
|
||
|
|
"epoch": 0.26721891909947226,
|
||
|
|
"eval_steps": 500,
|
||
|
|
"global_step": 500,
|
||
|
|
"is_hyper_param_search": false,
|
||
|
|
"is_local_process_zero": true,
|
||
|
|
"is_world_process_zero": true,
|
||
|
|
"log_history": [
|
||
|
|
{
|
||
|
|
"epoch": 0.0005344378381989445,
|
||
|
|
"grad_norm": 63.52920150756836,
|
||
|
|
"learning_rate": 0.0,
|
||
|
|
"logits/chosen": -0.144775390625,
|
||
|
|
"logits/rejected": -0.1036376953125,
|
||
|
|
"logps/chosen": -110.828125,
|
||
|
|
"logps/rejected": -116.828125,
|
||
|
|
"logps/weighted_chosen": -0.2086181640625,
|
||
|
|
"logps/weighted_rejected": -1.271392822265625,
|
||
|
|
"loss": 0.6914,
|
||
|
|
"rewards/accuracies": 0.0,
|
||
|
|
"rewards/chosen": 0.0,
|
||
|
|
"rewards/margins": 0.0,
|
||
|
|
"rewards/rejected": 0.0,
|
||
|
|
"rewards/weighted_accuracies": 0.0,
|
||
|
|
"rewards/weighted_chosen": 0.0,
|
||
|
|
"rewards/weighted_margins": 0.0,
|
||
|
|
"rewards/weighted_rejected": 0.0,
|
||
|
|
"step": 1
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.005344378381989445,
|
||
|
|
"grad_norm": 102.77810668945312,
|
||
|
|
"learning_rate": 4.787234042553191e-08,
|
||
|
|
"logits/chosen": -0.2549913227558136,
|
||
|
|
"logits/rejected": -0.24070316553115845,
|
||
|
|
"logps/chosen": -135.32899475097656,
|
||
|
|
"logps/rejected": -138.99305725097656,
|
||
|
|
"logps/weighted_chosen": -0.8547499179840088,
|
||
|
|
"logps/weighted_rejected": -0.9027608036994934,
|
||
|
|
"loss": 0.6928,
|
||
|
|
"rewards/accuracies": 0.2951388955116272,
|
||
|
|
"rewards/chosen": -0.0164930559694767,
|
||
|
|
"rewards/margins": -0.0349392369389534,
|
||
|
|
"rewards/rejected": 0.0184461809694767,
|
||
|
|
"rewards/weighted_accuracies": 0.4027777910232544,
|
||
|
|
"rewards/weighted_chosen": -0.002239651046693325,
|
||
|
|
"rewards/weighted_margins": 0.0028160943184047937,
|
||
|
|
"rewards/weighted_rejected": -0.0050794812850654125,
|
||
|
|
"step": 10
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.01068875676397889,
|
||
|
|
"grad_norm": 9.201544761657715,
|
||
|
|
"learning_rate": 1.0106382978723404e-07,
|
||
|
|
"logits/chosen": -0.19488906860351562,
|
||
|
|
"logits/rejected": -0.183399960398674,
|
||
|
|
"logps/chosen": -116.0374984741211,
|
||
|
|
"logps/rejected": -115.5054702758789,
|
||
|
|
"logps/weighted_chosen": -0.6929916143417358,
|
||
|
|
"logps/weighted_rejected": -0.6838119626045227,
|
||
|
|
"loss": 0.6917,
|
||
|
|
"rewards/accuracies": 0.38749998807907104,
|
||
|
|
"rewards/chosen": 0.01796874962747097,
|
||
|
|
"rewards/margins": 0.03007812425494194,
|
||
|
|
"rewards/rejected": -0.01210937462747097,
|
||
|
|
"rewards/weighted_accuracies": 0.4781250059604645,
|
||
|
|
"rewards/weighted_chosen": 0.0022293091751635075,
|
||
|
|
"rewards/weighted_margins": 0.00339679722674191,
|
||
|
|
"rewards/weighted_rejected": -0.0011629104847088456,
|
||
|
|
"step": 20
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.016033135145968335,
|
||
|
|
"grad_norm": 7.109442234039307,
|
||
|
|
"learning_rate": 1.5425531914893615e-07,
|
||
|
|
"logits/chosen": -0.1776607483625412,
|
||
|
|
"logits/rejected": -0.17884215712547302,
|
||
|
|
"logps/chosen": -123.2867202758789,
|
||
|
|
"logps/rejected": -129.75,
|
||
|
|
"logps/weighted_chosen": -0.7567901611328125,
|
||
|
|
"logps/weighted_rejected": -1.2207733392715454,
|
||
|
|
"loss": 0.6913,
|
||
|
|
"rewards/accuracies": 0.3968749940395355,
|
||
|
|
"rewards/chosen": -0.0074218749068677425,
|
||
|
|
"rewards/margins": 0.04843749850988388,
|
||
|
|
"rewards/rejected": -0.05585937574505806,
|
||
|
|
"rewards/weighted_accuracies": 0.5,
|
||
|
|
"rewards/weighted_chosen": 0.0007101058727130294,
|
||
|
|
"rewards/weighted_margins": 0.004384422209113836,
|
||
|
|
"rewards/weighted_rejected": -0.0036745071411132812,
|
||
|
|
"step": 30
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.02137751352795778,
|
||
|
|
"grad_norm": 6.0048956871032715,
|
||
|
|
"learning_rate": 2.074468085106383e-07,
|
||
|
|
"logits/chosen": -0.23842163383960724,
|
||
|
|
"logits/rejected": -0.2298881560564041,
|
||
|
|
"logps/chosen": -127.34062194824219,
|
||
|
|
"logps/rejected": -128.1710968017578,
|
||
|
|
"logps/weighted_chosen": -0.8051223754882812,
|
||
|
|
"logps/weighted_rejected": -0.5736129879951477,
|
||
|
|
"loss": 0.6934,
|
||
|
|
"rewards/accuracies": 0.34062498807907104,
|
||
|
|
"rewards/chosen": -0.01992187462747097,
|
||
|
|
"rewards/margins": -0.05351562425494194,
|
||
|
|
"rewards/rejected": 0.03359375149011612,
|
||
|
|
"rewards/weighted_accuracies": 0.4593749940395355,
|
||
|
|
"rewards/weighted_chosen": -7.83920258982107e-05,
|
||
|
|
"rewards/weighted_margins": -0.0009832382202148438,
|
||
|
|
"rewards/weighted_rejected": 0.000904083251953125,
|
||
|
|
"step": 40
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.026721891909947223,
|
||
|
|
"grad_norm": 32.7233772277832,
|
||
|
|
"learning_rate": 2.6063829787234044e-07,
|
||
|
|
"logits/chosen": -0.18835143744945526,
|
||
|
|
"logits/rejected": -0.170085147023201,
|
||
|
|
"logps/chosen": -120.88594055175781,
|
||
|
|
"logps/rejected": -121.3515625,
|
||
|
|
"logps/weighted_chosen": -0.6867736577987671,
|
||
|
|
"logps/weighted_rejected": -1.138249158859253,
|
||
|
|
"loss": 0.6903,
|
||
|
|
"rewards/accuracies": 0.3656249940395355,
|
||
|
|
"rewards/chosen": 0.03554687649011612,
|
||
|
|
"rewards/margins": -0.03164062649011612,
|
||
|
|
"rewards/rejected": 0.06718750298023224,
|
||
|
|
"rewards/weighted_accuracies": 0.515625,
|
||
|
|
"rewards/weighted_chosen": -0.006287765689194202,
|
||
|
|
"rewards/weighted_margins": 0.0064636231400072575,
|
||
|
|
"rewards/weighted_rejected": -0.01274185162037611,
|
||
|
|
"step": 50
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.03206627029193667,
|
||
|
|
"grad_norm": 55.560707092285156,
|
||
|
|
"learning_rate": 3.1382978723404253e-07,
|
||
|
|
"logits/chosen": -0.1308601349592209,
|
||
|
|
"logits/rejected": -0.1308906525373459,
|
||
|
|
"logps/chosen": -116.5875015258789,
|
||
|
|
"logps/rejected": -119.5406265258789,
|
||
|
|
"logps/weighted_chosen": -0.8019469976425171,
|
||
|
|
"logps/weighted_rejected": -0.8834930658340454,
|
||
|
|
"loss": 0.7002,
|
||
|
|
"rewards/accuracies": 0.3343749940395355,
|
||
|
|
"rewards/chosen": 0.03476562350988388,
|
||
|
|
"rewards/margins": -0.06171875074505806,
|
||
|
|
"rewards/rejected": 0.09648437798023224,
|
||
|
|
"rewards/weighted_accuracies": 0.4781250059604645,
|
||
|
|
"rewards/weighted_chosen": -0.01903533935546875,
|
||
|
|
"rewards/weighted_margins": -2.880096508306451e-05,
|
||
|
|
"rewards/weighted_rejected": -0.019007300958037376,
|
||
|
|
"step": 60
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.037410648673926114,
|
||
|
|
"grad_norm": 20.22484588623047,
|
||
|
|
"learning_rate": 3.6702127659574467e-07,
|
||
|
|
"logits/chosen": -0.20572929084300995,
|
||
|
|
"logits/rejected": -0.19457244873046875,
|
||
|
|
"logps/chosen": -114.4507827758789,
|
||
|
|
"logps/rejected": -118.63594055175781,
|
||
|
|
"logps/weighted_chosen": -0.6070801019668579,
|
||
|
|
"logps/weighted_rejected": -0.6768280267715454,
|
||
|
|
"loss": 0.6911,
|
||
|
|
"rewards/accuracies": 0.42500001192092896,
|
||
|
|
"rewards/chosen": 0.16484375298023224,
|
||
|
|
"rewards/margins": 0.07148437201976776,
|
||
|
|
"rewards/rejected": 0.09335937350988388,
|
||
|
|
"rewards/weighted_accuracies": 0.4749999940395355,
|
||
|
|
"rewards/weighted_chosen": -0.010281562805175781,
|
||
|
|
"rewards/weighted_margins": 0.007259177975356579,
|
||
|
|
"rewards/weighted_rejected": -0.017521286383271217,
|
||
|
|
"step": 70
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.04275502705591556,
|
||
|
|
"grad_norm": 15.305061340332031,
|
||
|
|
"learning_rate": 4.202127659574468e-07,
|
||
|
|
"logits/chosen": -0.19508513808250427,
|
||
|
|
"logits/rejected": -0.12093047797679901,
|
||
|
|
"logps/chosen": -112.57734680175781,
|
||
|
|
"logps/rejected": -112.77734375,
|
||
|
|
"logps/weighted_chosen": -0.4597320556640625,
|
||
|
|
"logps/weighted_rejected": -0.9711883664131165,
|
||
|
|
"loss": 0.6952,
|
||
|
|
"rewards/accuracies": 0.37812501192092896,
|
||
|
|
"rewards/chosen": 0.04570312425494194,
|
||
|
|
"rewards/margins": 0.01914062537252903,
|
||
|
|
"rewards/rejected": 0.02656250074505806,
|
||
|
|
"rewards/weighted_accuracies": 0.4625000059604645,
|
||
|
|
"rewards/weighted_chosen": -0.02759857103228569,
|
||
|
|
"rewards/weighted_margins": 0.021048355847597122,
|
||
|
|
"rewards/weighted_rejected": -0.04872245714068413,
|
||
|
|
"step": 80
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.048099405437905,
|
||
|
|
"grad_norm": 11.991273880004883,
|
||
|
|
"learning_rate": 4.734042553191489e-07,
|
||
|
|
"logits/chosen": -0.17391815781593323,
|
||
|
|
"logits/rejected": -0.11656417697668076,
|
||
|
|
"logps/chosen": -119.3070297241211,
|
||
|
|
"logps/rejected": -118.51249694824219,
|
||
|
|
"logps/weighted_chosen": -1.0416854619979858,
|
||
|
|
"logps/weighted_rejected": -1.5859466791152954,
|
||
|
|
"loss": 0.6736,
|
||
|
|
"rewards/accuracies": 0.4375,
|
||
|
|
"rewards/chosen": -0.08164062350988388,
|
||
|
|
"rewards/margins": 0.341796875,
|
||
|
|
"rewards/rejected": -0.4234375059604645,
|
||
|
|
"rewards/weighted_accuracies": 0.515625,
|
||
|
|
"rewards/weighted_chosen": -0.04084587097167969,
|
||
|
|
"rewards/weighted_margins": 0.10567589104175568,
|
||
|
|
"rewards/weighted_rejected": -0.14661255478858948,
|
||
|
|
"step": 90
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.053443783819894446,
|
||
|
|
"grad_norm": 17.195886611938477,
|
||
|
|
"learning_rate": 5.26595744680851e-07,
|
||
|
|
"logits/chosen": -0.204203799366951,
|
||
|
|
"logits/rejected": -0.14691162109375,
|
||
|
|
"logps/chosen": -125.05155944824219,
|
||
|
|
"logps/rejected": -125.3671875,
|
||
|
|
"logps/weighted_chosen": -0.7374237179756165,
|
||
|
|
"logps/weighted_rejected": -0.7348953485488892,
|
||
|
|
"loss": 0.7064,
|
||
|
|
"rewards/accuracies": 0.4281249940395355,
|
||
|
|
"rewards/chosen": -0.3832031190395355,
|
||
|
|
"rewards/margins": 0.15585938096046448,
|
||
|
|
"rewards/rejected": -0.5390625,
|
||
|
|
"rewards/weighted_accuracies": 0.46875,
|
||
|
|
"rewards/weighted_chosen": -0.060018159449100494,
|
||
|
|
"rewards/weighted_margins": 0.0024120330344885588,
|
||
|
|
"rewards/weighted_rejected": -0.06242714077234268,
|
||
|
|
"step": 100
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.058788162201883896,
|
||
|
|
"grad_norm": 9.103920936584473,
|
||
|
|
"learning_rate": 5.797872340425531e-07,
|
||
|
|
"logits/chosen": -0.2138931304216385,
|
||
|
|
"logits/rejected": -0.19011077284812927,
|
||
|
|
"logps/chosen": -118.6820297241211,
|
||
|
|
"logps/rejected": -121.28593444824219,
|
||
|
|
"logps/weighted_chosen": -0.4821228086948395,
|
||
|
|
"logps/weighted_rejected": -0.7759445309638977,
|
||
|
|
"loss": 0.6939,
|
||
|
|
"rewards/accuracies": 0.4281249940395355,
|
||
|
|
"rewards/chosen": -0.896484375,
|
||
|
|
"rewards/margins": 0.09531249850988388,
|
||
|
|
"rewards/rejected": -0.9917968511581421,
|
||
|
|
"rewards/weighted_accuracies": 0.528124988079071,
|
||
|
|
"rewards/weighted_chosen": -0.05413341522216797,
|
||
|
|
"rewards/weighted_margins": 0.04503212124109268,
|
||
|
|
"rewards/weighted_rejected": -0.09918193519115448,
|
||
|
|
"step": 110
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06413254058387334,
|
||
|
|
"grad_norm": 8.539896011352539,
|
||
|
|
"learning_rate": 6.329787234042553e-07,
|
||
|
|
"logits/chosen": -0.17510680854320526,
|
||
|
|
"logits/rejected": -0.14726562798023224,
|
||
|
|
"logps/chosen": -122.58906555175781,
|
||
|
|
"logps/rejected": -128.4656219482422,
|
||
|
|
"logps/weighted_chosen": -0.9249496459960938,
|
||
|
|
"logps/weighted_rejected": -0.762683093547821,
|
||
|
|
"loss": 0.7325,
|
||
|
|
"rewards/accuracies": 0.4375,
|
||
|
|
"rewards/chosen": -1.068359375,
|
||
|
|
"rewards/margins": 0.28398436307907104,
|
||
|
|
"rewards/rejected": -1.3523437976837158,
|
||
|
|
"rewards/weighted_accuracies": 0.4781250059604645,
|
||
|
|
"rewards/weighted_chosen": -0.098027803003788,
|
||
|
|
"rewards/weighted_margins": -0.029563140124082565,
|
||
|
|
"rewards/weighted_rejected": -0.06836700439453125,
|
||
|
|
"step": 120
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.06947691896586278,
|
||
|
|
"grad_norm": 15.564044952392578,
|
||
|
|
"learning_rate": 6.861702127659574e-07,
|
||
|
|
"logits/chosen": -0.20233002305030823,
|
||
|
|
"logits/rejected": -0.18396148085594177,
|
||
|
|
"logps/chosen": -122.2671890258789,
|
||
|
|
"logps/rejected": -123.8382797241211,
|
||
|
|
"logps/weighted_chosen": -0.7501281499862671,
|
||
|
|
"logps/weighted_rejected": -0.821154773235321,
|
||
|
|
"loss": 0.7078,
|
||
|
|
"rewards/accuracies": 0.4000000059604645,
|
||
|
|
"rewards/chosen": -1.45703125,
|
||
|
|
"rewards/margins": 0.29179686307907104,
|
||
|
|
"rewards/rejected": -1.7488281726837158,
|
||
|
|
"rewards/weighted_accuracies": 0.5062500238418579,
|
||
|
|
"rewards/weighted_chosen": -0.007884788326919079,
|
||
|
|
"rewards/weighted_margins": 0.006314086727797985,
|
||
|
|
"rewards/weighted_rejected": -0.014200759120285511,
|
||
|
|
"step": 130
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.07482129734785223,
|
||
|
|
"grad_norm": 9.176558494567871,
|
||
|
|
"learning_rate": 7.393617021276596e-07,
|
||
|
|
"logits/chosen": -0.16455383598804474,
|
||
|
|
"logits/rejected": -0.14660033583641052,
|
||
|
|
"logps/chosen": -116.328125,
|
||
|
|
"logps/rejected": -119.9664077758789,
|
||
|
|
"logps/weighted_chosen": -0.9873703122138977,
|
||
|
|
"logps/weighted_rejected": -0.6730011105537415,
|
||
|
|
"loss": 0.7174,
|
||
|
|
"rewards/accuracies": 0.4749999940395355,
|
||
|
|
"rewards/chosen": -1.6007812023162842,
|
||
|
|
"rewards/margins": 0.533203125,
|
||
|
|
"rewards/rejected": -2.133984327316284,
|
||
|
|
"rewards/weighted_accuracies": 0.515625,
|
||
|
|
"rewards/weighted_chosen": 0.02084655687212944,
|
||
|
|
"rewards/weighted_margins": 0.05102996900677681,
|
||
|
|
"rewards/weighted_rejected": -0.030106354504823685,
|
||
|
|
"step": 140
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08016567572984168,
|
||
|
|
"grad_norm": 10.172534942626953,
|
||
|
|
"learning_rate": 7.925531914893616e-07,
|
||
|
|
"logits/chosen": -0.1509498655796051,
|
||
|
|
"logits/rejected": -0.13156357407569885,
|
||
|
|
"logps/chosen": -120.0562515258789,
|
||
|
|
"logps/rejected": -117.85546875,
|
||
|
|
"logps/weighted_chosen": -0.8198806643486023,
|
||
|
|
"logps/weighted_rejected": -0.7782562375068665,
|
||
|
|
"loss": 0.7005,
|
||
|
|
"rewards/accuracies": 0.46562498807907104,
|
||
|
|
"rewards/chosen": -2.0234375,
|
||
|
|
"rewards/margins": 0.607421875,
|
||
|
|
"rewards/rejected": -2.630859375,
|
||
|
|
"rewards/weighted_accuracies": 0.543749988079071,
|
||
|
|
"rewards/weighted_chosen": -0.0781441181898117,
|
||
|
|
"rewards/weighted_margins": 0.034273337572813034,
|
||
|
|
"rewards/weighted_rejected": -0.11248989403247833,
|
||
|
|
"step": 150
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.08551005411183112,
|
||
|
|
"grad_norm": 25.879701614379883,
|
||
|
|
"learning_rate": 8.457446808510637e-07,
|
||
|
|
"logits/chosen": -0.17474937438964844,
|
||
|
|
"logits/rejected": -0.17009583115577698,
|
||
|
|
"logps/chosen": -121.2367172241211,
|
||
|
|
"logps/rejected": -121.62109375,
|
||
|
|
"logps/weighted_chosen": -0.8304061889648438,
|
||
|
|
"logps/weighted_rejected": -0.866925060749054,
|
||
|
|
"loss": 0.7072,
|
||
|
|
"rewards/accuracies": 0.4781250059604645,
|
||
|
|
"rewards/chosen": -2.219531297683716,
|
||
|
|
"rewards/margins": 0.716796875,
|
||
|
|
"rewards/rejected": -2.936328172683716,
|
||
|
|
"rewards/weighted_accuracies": 0.47187501192092896,
|
||
|
|
"rewards/weighted_chosen": -0.07932891696691513,
|
||
|
|
"rewards/weighted_margins": 0.017035294324159622,
|
||
|
|
"rewards/weighted_rejected": -0.09644661098718643,
|
||
|
|
"step": 160
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09085443249382057,
|
||
|
|
"grad_norm": 9.220558166503906,
|
||
|
|
"learning_rate": 8.989361702127659e-07,
|
||
|
|
"logits/chosen": -0.22574539482593536,
|
||
|
|
"logits/rejected": -0.19882050156593323,
|
||
|
|
"logps/chosen": -120.62968444824219,
|
||
|
|
"logps/rejected": -124.78437805175781,
|
||
|
|
"logps/weighted_chosen": -0.9401000738143921,
|
||
|
|
"logps/weighted_rejected": -0.8424423336982727,
|
||
|
|
"loss": 0.7174,
|
||
|
|
"rewards/accuracies": 0.4937500059604645,
|
||
|
|
"rewards/chosen": -2.4033203125,
|
||
|
|
"rewards/margins": 0.613085925579071,
|
||
|
|
"rewards/rejected": -3.016406297683716,
|
||
|
|
"rewards/weighted_accuracies": 0.48750001192092896,
|
||
|
|
"rewards/weighted_chosen": -0.04837837070226669,
|
||
|
|
"rewards/weighted_margins": 0.04595222324132919,
|
||
|
|
"rewards/weighted_rejected": -0.09437103569507599,
|
||
|
|
"step": 170
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.09619881087581,
|
||
|
|
"grad_norm": 55.62498092651367,
|
||
|
|
"learning_rate": 9.52127659574468e-07,
|
||
|
|
"logits/chosen": -0.2216789275407791,
|
||
|
|
"logits/rejected": -0.20373383164405823,
|
||
|
|
"logps/chosen": -120.3851547241211,
|
||
|
|
"logps/rejected": -121.2328109741211,
|
||
|
|
"logps/weighted_chosen": -0.435232549905777,
|
||
|
|
"logps/weighted_rejected": -0.5373809933662415,
|
||
|
|
"loss": 0.7302,
|
||
|
|
"rewards/accuracies": 0.47187501192092896,
|
||
|
|
"rewards/chosen": -1.338281273841858,
|
||
|
|
"rewards/margins": 0.3658203184604645,
|
||
|
|
"rewards/rejected": -1.7041015625,
|
||
|
|
"rewards/weighted_accuracies": 0.53125,
|
||
|
|
"rewards/weighted_chosen": -0.08163909614086151,
|
||
|
|
"rewards/weighted_margins": -0.008893203921616077,
|
||
|
|
"rewards/weighted_rejected": -0.07281456142663956,
|
||
|
|
"step": 180
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.10154318925779945,
|
||
|
|
"grad_norm": 39.38788986206055,
|
||
|
|
"learning_rate": 9.99999129927477e-07,
|
||
|
|
"logits/chosen": -0.19243164360523224,
|
||
|
|
"logits/rejected": -0.15758666396141052,
|
||
|
|
"logps/chosen": -121.8960952758789,
|
||
|
|
"logps/rejected": -123.01094055175781,
|
||
|
|
"logps/weighted_chosen": -0.953173816204071,
|
||
|
|
"logps/weighted_rejected": -1.0853087902069092,
|
||
|
|
"loss": 0.6702,
|
||
|
|
"rewards/accuracies": 0.48750001192092896,
|
||
|
|
"rewards/chosen": -2.100390672683716,
|
||
|
|
"rewards/margins": 0.662890613079071,
|
||
|
|
"rewards/rejected": -2.7632813453674316,
|
||
|
|
"rewards/weighted_accuracies": 0.578125,
|
||
|
|
"rewards/weighted_chosen": -0.09999503940343857,
|
||
|
|
"rewards/weighted_margins": 0.12822990119457245,
|
||
|
|
"rewards/weighted_rejected": -0.22818031907081604,
|
||
|
|
"step": 190
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.10688756763978889,
|
||
|
|
"grad_norm": 12.883384704589844,
|
||
|
|
"learning_rate": 9.99894724888679e-07,
|
||
|
|
"logits/chosen": -0.17392578721046448,
|
||
|
|
"logits/rejected": -0.147491455078125,
|
||
|
|
"logps/chosen": -129.53750610351562,
|
||
|
|
"logps/rejected": -128.95703125,
|
||
|
|
"logps/weighted_chosen": -0.7322753667831421,
|
||
|
|
"logps/weighted_rejected": -1.0728164911270142,
|
||
|
|
"loss": 0.6777,
|
||
|
|
"rewards/accuracies": 0.4749999940395355,
|
||
|
|
"rewards/chosen": -3.215625047683716,
|
||
|
|
"rewards/margins": 0.9507812261581421,
|
||
|
|
"rewards/rejected": -4.166406154632568,
|
||
|
|
"rewards/weighted_accuracies": 0.5375000238418579,
|
||
|
|
"rewards/weighted_chosen": -0.1104557067155838,
|
||
|
|
"rewards/weighted_margins": 0.13128623366355896,
|
||
|
|
"rewards/weighted_rejected": -0.2415626496076584,
|
||
|
|
"step": 200
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11223194602177834,
|
||
|
|
"grad_norm": 17.96660041809082,
|
||
|
|
"learning_rate": 9.996163469793475e-07,
|
||
|
|
"logits/chosen": -0.18083801865577698,
|
||
|
|
"logits/rejected": -0.12766647338867188,
|
||
|
|
"logps/chosen": -128.1953125,
|
||
|
|
"logps/rejected": -123.03047180175781,
|
||
|
|
"logps/weighted_chosen": -0.8699920773506165,
|
||
|
|
"logps/weighted_rejected": -0.8845611810684204,
|
||
|
|
"loss": 0.7572,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -3.5113282203674316,
|
||
|
|
"rewards/margins": 1.3683593273162842,
|
||
|
|
"rewards/rejected": -4.879296779632568,
|
||
|
|
"rewards/weighted_accuracies": 0.5531250238418579,
|
||
|
|
"rewards/weighted_chosen": -0.07919616997241974,
|
||
|
|
"rewards/weighted_margins": 0.04895629733800888,
|
||
|
|
"rewards/weighted_rejected": -0.12815475463867188,
|
||
|
|
"step": 210
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.11757632440376779,
|
||
|
|
"grad_norm": 63.790077209472656,
|
||
|
|
"learning_rate": 9.991640930802883e-07,
|
||
|
|
"logits/chosen": -0.14619751274585724,
|
||
|
|
"logits/rejected": -0.13860473036766052,
|
||
|
|
"logps/chosen": -126.61015319824219,
|
||
|
|
"logps/rejected": -129.9773406982422,
|
||
|
|
"logps/weighted_chosen": -1.0776779651641846,
|
||
|
|
"logps/weighted_rejected": -0.9309722781181335,
|
||
|
|
"loss": 0.7107,
|
||
|
|
"rewards/accuracies": 0.5531250238418579,
|
||
|
|
"rewards/chosen": -4.296484470367432,
|
||
|
|
"rewards/margins": 1.10546875,
|
||
|
|
"rewards/rejected": -5.40234375,
|
||
|
|
"rewards/weighted_accuracies": 0.5406249761581421,
|
||
|
|
"rewards/weighted_chosen": -0.013729858212172985,
|
||
|
|
"rewards/weighted_margins": 0.14467468857765198,
|
||
|
|
"rewards/weighted_rejected": -0.1582542359828949,
|
||
|
|
"step": 220
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12292070278575723,
|
||
|
|
"grad_norm": 9.634794235229492,
|
||
|
|
"learning_rate": 9.98538120584459e-07,
|
||
|
|
"logits/chosen": -0.1792152374982834,
|
||
|
|
"logits/rejected": -0.1362098753452301,
|
||
|
|
"logps/chosen": -133.7140655517578,
|
||
|
|
"logps/rejected": -137.7546844482422,
|
||
|
|
"logps/weighted_chosen": -0.6676223874092102,
|
||
|
|
"logps/weighted_rejected": -0.9518234133720398,
|
||
|
|
"loss": 0.7614,
|
||
|
|
"rewards/accuracies": 0.515625,
|
||
|
|
"rewards/chosen": -4.493554592132568,
|
||
|
|
"rewards/margins": 1.0623047351837158,
|
||
|
|
"rewards/rejected": -5.555859565734863,
|
||
|
|
"rewards/weighted_accuracies": 0.565625011920929,
|
||
|
|
"rewards/weighted_chosen": -0.1120578795671463,
|
||
|
|
"rewards/weighted_margins": 0.031700897961854935,
|
||
|
|
"rewards/weighted_rejected": -0.14376182854175568,
|
||
|
|
"step": 230
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.12826508116774668,
|
||
|
|
"grad_norm": 8.717682838439941,
|
||
|
|
"learning_rate": 9.977386473421917e-07,
|
||
|
|
"logits/chosen": -0.09295501559972763,
|
||
|
|
"logits/rejected": -0.07836151123046875,
|
||
|
|
"logps/chosen": -121.78202819824219,
|
||
|
|
"logps/rejected": -126.39453125,
|
||
|
|
"logps/weighted_chosen": -0.8116515874862671,
|
||
|
|
"logps/weighted_rejected": -1.2165496349334717,
|
||
|
|
"loss": 0.6731,
|
||
|
|
"rewards/accuracies": 0.5874999761581421,
|
||
|
|
"rewards/chosen": -7.553515434265137,
|
||
|
|
"rewards/margins": 1.669921875,
|
||
|
|
"rewards/rejected": -9.224218368530273,
|
||
|
|
"rewards/weighted_accuracies": 0.534375011920929,
|
||
|
|
"rewards/weighted_chosen": -0.1755084991455078,
|
||
|
|
"rewards/weighted_margins": 0.15414047241210938,
|
||
|
|
"rewards/weighted_rejected": -0.32989272475242615,
|
||
|
|
"step": 240
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.13360945954973613,
|
||
|
|
"grad_norm": 35.197269439697266,
|
||
|
|
"learning_rate": 9.96765951585378e-07,
|
||
|
|
"logits/chosen": -0.021549224853515625,
|
||
|
|
"logits/rejected": 0.00345611572265625,
|
||
|
|
"logps/chosen": -123.0296859741211,
|
||
|
|
"logps/rejected": -127.9671859741211,
|
||
|
|
"logps/weighted_chosen": -1.186151146888733,
|
||
|
|
"logps/weighted_rejected": -1.386474609375,
|
||
|
|
"loss": 0.7026,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -9.133593559265137,
|
||
|
|
"rewards/margins": 1.150781273841858,
|
||
|
|
"rewards/rejected": -10.283984184265137,
|
||
|
|
"rewards/weighted_accuracies": 0.59375,
|
||
|
|
"rewards/weighted_chosen": -0.3282794952392578,
|
||
|
|
"rewards/weighted_margins": 0.1091407760977745,
|
||
|
|
"rewards/weighted_rejected": -0.4371795654296875,
|
||
|
|
"step": 250
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.13895383793172555,
|
||
|
|
"grad_norm": 84.2197494506836,
|
||
|
|
"learning_rate": 9.956203718306388e-07,
|
||
|
|
"logits/chosen": 0.025267411023378372,
|
||
|
|
"logits/rejected": 0.06335143744945526,
|
||
|
|
"logps/chosen": -128.8429718017578,
|
||
|
|
"logps/rejected": -133.50625610351562,
|
||
|
|
"logps/weighted_chosen": -0.9628280401229858,
|
||
|
|
"logps/weighted_rejected": -1.2008453607559204,
|
||
|
|
"loss": 0.7443,
|
||
|
|
"rewards/accuracies": 0.5,
|
||
|
|
"rewards/chosen": -10.861719131469727,
|
||
|
|
"rewards/margins": 0.871874988079071,
|
||
|
|
"rewards/rejected": -11.733593940734863,
|
||
|
|
"rewards/weighted_accuracies": 0.49687498807907104,
|
||
|
|
"rewards/weighted_chosen": -0.26259079575538635,
|
||
|
|
"rewards/weighted_margins": 0.06770706176757812,
|
||
|
|
"rewards/weighted_rejected": -0.3305191099643707,
|
||
|
|
"step": 260
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.144298216313715,
|
||
|
|
"grad_norm": 57.09628677368164,
|
||
|
|
"learning_rate": 9.943023067615136e-07,
|
||
|
|
"logits/chosen": -0.017973709851503372,
|
||
|
|
"logits/rejected": -0.0057998658157885075,
|
||
|
|
"logps/chosen": -137.60311889648438,
|
||
|
|
"logps/rejected": -140.91561889648438,
|
||
|
|
"logps/weighted_chosen": -0.8694183230400085,
|
||
|
|
"logps/weighted_rejected": -0.8910156488418579,
|
||
|
|
"loss": 0.7462,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -10.946093559265137,
|
||
|
|
"rewards/margins": 1.662109375,
|
||
|
|
"rewards/rejected": -12.608202934265137,
|
||
|
|
"rewards/weighted_accuracies": 0.565625011920929,
|
||
|
|
"rewards/weighted_chosen": -0.06702041625976562,
|
||
|
|
"rewards/weighted_margins": 0.13949890434741974,
|
||
|
|
"rewards/weighted_rejected": -0.20627327263355255,
|
||
|
|
"step": 270
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.14964259469570446,
|
||
|
|
"grad_norm": 45.0294303894043,
|
||
|
|
"learning_rate": 9.928122150897112e-07,
|
||
|
|
"logits/chosen": -0.045014191418886185,
|
||
|
|
"logits/rejected": 0.02437591552734375,
|
||
|
|
"logps/chosen": -130.1125030517578,
|
||
|
|
"logps/rejected": -131.6171875,
|
||
|
|
"logps/weighted_chosen": -0.7659690976142883,
|
||
|
|
"logps/weighted_rejected": -0.8499618768692017,
|
||
|
|
"loss": 0.7079,
|
||
|
|
"rewards/accuracies": 0.5406249761581421,
|
||
|
|
"rewards/chosen": -12.664453506469727,
|
||
|
|
"rewards/margins": 1.584375023841858,
|
||
|
|
"rewards/rejected": -14.249608993530273,
|
||
|
|
"rewards/weighted_accuracies": 0.543749988079071,
|
||
|
|
"rewards/weighted_chosen": -0.22265930473804474,
|
||
|
|
"rewards/weighted_margins": 0.07556991279125214,
|
||
|
|
"rewards/weighted_rejected": -0.298013299703598,
|
||
|
|
"step": 280
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.1549869730776939,
|
||
|
|
"grad_norm": 9.370684623718262,
|
||
|
|
"learning_rate": 9.9115061539547e-07,
|
||
|
|
"logits/chosen": -0.02069244347512722,
|
||
|
|
"logits/rejected": 0.01569976843893528,
|
||
|
|
"logps/chosen": -139.4453125,
|
||
|
|
"logps/rejected": -141.2578125,
|
||
|
|
"logps/weighted_chosen": -0.8775177001953125,
|
||
|
|
"logps/weighted_rejected": -1.2375701665878296,
|
||
|
|
"loss": 0.6949,
|
||
|
|
"rewards/accuracies": 0.5375000238418579,
|
||
|
|
"rewards/chosen": -13.07421875,
|
||
|
|
"rewards/margins": 1.177734375,
|
||
|
|
"rewards/rejected": -14.249608993530273,
|
||
|
|
"rewards/weighted_accuracies": 0.546875,
|
||
|
|
"rewards/weighted_chosen": -0.18484345078468323,
|
||
|
|
"rewards/weighted_margins": 0.12048187106847763,
|
||
|
|
"rewards/weighted_rejected": -0.30528029799461365,
|
||
|
|
"step": 290
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.16033135145968336,
|
||
|
|
"grad_norm": 30.68979263305664,
|
||
|
|
"learning_rate": 9.893180859470818e-07,
|
||
|
|
"logits/chosen": -0.0137786865234375,
|
||
|
|
"logits/rejected": 0.041573334485292435,
|
||
|
|
"logps/chosen": -130.22265625,
|
||
|
|
"logps/rejected": -132.06094360351562,
|
||
|
|
"logps/weighted_chosen": -1.1769134998321533,
|
||
|
|
"logps/weighted_rejected": -1.2438628673553467,
|
||
|
|
"loss": 0.7261,
|
||
|
|
"rewards/accuracies": 0.590624988079071,
|
||
|
|
"rewards/chosen": -13.093358993530273,
|
||
|
|
"rewards/margins": 2.108203172683716,
|
||
|
|
"rewards/rejected": -15.200780868530273,
|
||
|
|
"rewards/weighted_accuracies": 0.5843750238418579,
|
||
|
|
"rewards/weighted_chosen": -0.2848831117153168,
|
||
|
|
"rewards/weighted_margins": 0.07841148227453232,
|
||
|
|
"rewards/weighted_rejected": -0.36317139863967896,
|
||
|
|
"step": 300
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.16567572984167278,
|
||
|
|
"grad_norm": 71.56196594238281,
|
||
|
|
"learning_rate": 9.873152644996424e-07,
|
||
|
|
"logits/chosen": -0.11114044487476349,
|
||
|
|
"logits/rejected": -0.11599807441234589,
|
||
|
|
"logps/chosen": -131.01718139648438,
|
||
|
|
"logps/rejected": -132.28125,
|
||
|
|
"logps/weighted_chosen": -1.1032531261444092,
|
||
|
|
"logps/weighted_rejected": -0.8420807123184204,
|
||
|
|
"loss": 0.7872,
|
||
|
|
"rewards/accuracies": 0.5718749761581421,
|
||
|
|
"rewards/chosen": -12.979687690734863,
|
||
|
|
"rewards/margins": 2.407421827316284,
|
||
|
|
"rewards/rejected": -15.388280868530273,
|
||
|
|
"rewards/weighted_accuracies": 0.5218750238418579,
|
||
|
|
"rewards/weighted_chosen": 0.05532989650964737,
|
||
|
|
"rewards/weighted_margins": 0.2024696320295334,
|
||
|
|
"rewards/weighted_rejected": -0.1469375640153885,
|
||
|
|
"step": 310
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.17102010822366223,
|
||
|
|
"grad_norm": 12.055798530578613,
|
||
|
|
"learning_rate": 9.85142848073103e-07,
|
||
|
|
"logits/chosen": -0.10557708889245987,
|
||
|
|
"logits/rejected": -0.06234588474035263,
|
||
|
|
"logps/chosen": -133.39218139648438,
|
||
|
|
"logps/rejected": -134.2109375,
|
||
|
|
"logps/weighted_chosen": -0.700244128704071,
|
||
|
|
"logps/weighted_rejected": -1.4902985095977783,
|
||
|
|
"loss": 0.7558,
|
||
|
|
"rewards/accuracies": 0.534375011920929,
|
||
|
|
"rewards/chosen": -14.2265625,
|
||
|
|
"rewards/margins": 0.6207031011581421,
|
||
|
|
"rewards/rejected": -14.846094131469727,
|
||
|
|
"rewards/weighted_accuracies": 0.515625,
|
||
|
|
"rewards/weighted_chosen": -0.2170669585466385,
|
||
|
|
"rewards/weighted_margins": 0.03446922451257706,
|
||
|
|
"rewards/weighted_rejected": -0.25187987089157104,
|
||
|
|
"step": 320
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.17636448660565168,
|
||
|
|
"grad_norm": 41.381507873535156,
|
||
|
|
"learning_rate": 9.828015927096914e-07,
|
||
|
|
"logits/chosen": -0.085169218480587,
|
||
|
|
"logits/rejected": -0.05568351596593857,
|
||
|
|
"logps/chosen": -143.8234405517578,
|
||
|
|
"logps/rejected": -142.39608764648438,
|
||
|
|
"logps/weighted_chosen": -0.977459728717804,
|
||
|
|
"logps/weighted_rejected": -0.7958831787109375,
|
||
|
|
"loss": 0.6935,
|
||
|
|
"rewards/accuracies": 0.550000011920929,
|
||
|
|
"rewards/chosen": -14.643750190734863,
|
||
|
|
"rewards/margins": 1.578125,
|
||
|
|
"rewards/rejected": -16.222265243530273,
|
||
|
|
"rewards/weighted_accuracies": 0.581250011920929,
|
||
|
|
"rewards/weighted_chosen": -0.16206398606300354,
|
||
|
|
"rewards/weighted_margins": 0.07153511047363281,
|
||
|
|
"rewards/weighted_rejected": -0.23361511528491974,
|
||
|
|
"step": 330
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.18170886498764113,
|
||
|
|
"grad_norm": 7.313599586486816,
|
||
|
|
"learning_rate": 9.802923132107968e-07,
|
||
|
|
"logits/chosen": -0.10784301906824112,
|
||
|
|
"logits/rejected": -0.052645109593868256,
|
||
|
|
"logps/chosen": -143.91561889648438,
|
||
|
|
"logps/rejected": -147.7234344482422,
|
||
|
|
"logps/weighted_chosen": -0.7267853021621704,
|
||
|
|
"logps/weighted_rejected": -1.24786376953125,
|
||
|
|
"loss": 0.7102,
|
||
|
|
"rewards/accuracies": 0.5562499761581421,
|
||
|
|
"rewards/chosen": -16.782812118530273,
|
||
|
|
"rewards/margins": 1.23046875,
|
||
|
|
"rewards/rejected": -18.012500762939453,
|
||
|
|
"rewards/weighted_accuracies": 0.5687500238418579,
|
||
|
|
"rewards/weighted_chosen": -0.1999809294939041,
|
||
|
|
"rewards/weighted_margins": 0.056002043187618256,
|
||
|
|
"rewards/weighted_rejected": -0.2561977505683899,
|
||
|
|
"step": 340
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.18705324336963056,
|
||
|
|
"grad_norm": 66.15936279296875,
|
||
|
|
"learning_rate": 9.776158828534024e-07,
|
||
|
|
"logits/chosen": -0.12608642876148224,
|
||
|
|
"logits/rejected": -0.09711913764476776,
|
||
|
|
"logps/chosen": -143.48281860351562,
|
||
|
|
"logps/rejected": -188.25936889648438,
|
||
|
|
"logps/weighted_chosen": -1.260833740234375,
|
||
|
|
"logps/weighted_rejected": -0.9638397097587585,
|
||
|
|
"loss": 0.7338,
|
||
|
|
"rewards/accuracies": 0.612500011920929,
|
||
|
|
"rewards/chosen": -16.727733612060547,
|
||
|
|
"rewards/margins": 3.159374952316284,
|
||
|
|
"rewards/rejected": -19.883594512939453,
|
||
|
|
"rewards/weighted_accuracies": 0.559374988079071,
|
||
|
|
"rewards/weighted_chosen": -0.26622849702835083,
|
||
|
|
"rewards/weighted_margins": 0.05975799635052681,
|
||
|
|
"rewards/weighted_rejected": -0.3257743716239929,
|
||
|
|
"step": 350
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.19239762175162,
|
||
|
|
"grad_norm": 10.4098539352417,
|
||
|
|
"learning_rate": 9.747732330861695e-07,
|
||
|
|
"logits/chosen": -0.05247039720416069,
|
||
|
|
"logits/rejected": -0.024548912420868874,
|
||
|
|
"logps/chosen": -135.78750610351562,
|
||
|
|
"logps/rejected": -137.03671264648438,
|
||
|
|
"logps/weighted_chosen": -0.8440704345703125,
|
||
|
|
"logps/weighted_rejected": -0.8307411074638367,
|
||
|
|
"loss": 0.7415,
|
||
|
|
"rewards/accuracies": 0.578125,
|
||
|
|
"rewards/chosen": -16.864452362060547,
|
||
|
|
"rewards/margins": 2.5316405296325684,
|
||
|
|
"rewards/rejected": -19.396093368530273,
|
||
|
|
"rewards/weighted_accuracies": 0.5249999761581421,
|
||
|
|
"rewards/weighted_chosen": 0.005719375796616077,
|
||
|
|
"rewards/weighted_margins": 0.17883948981761932,
|
||
|
|
"rewards/weighted_rejected": -0.17323684692382812,
|
||
|
|
"step": 360
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.19774200013360946,
|
||
|
|
"grad_norm": 8.8024263381958,
|
||
|
|
"learning_rate": 9.717653532052742e-07,
|
||
|
|
"logits/chosen": -0.0410335548222065,
|
||
|
|
"logits/rejected": -0.049485016614198685,
|
||
|
|
"logps/chosen": -130.2781219482422,
|
||
|
|
"logps/rejected": -137.9343719482422,
|
||
|
|
"logps/weighted_chosen": -1.1705901622772217,
|
||
|
|
"logps/weighted_rejected": -0.7656036615371704,
|
||
|
|
"loss": 0.7402,
|
||
|
|
"rewards/accuracies": 0.6031249761581421,
|
||
|
|
"rewards/chosen": -17.155858993530273,
|
||
|
|
"rewards/margins": 2.863085985183716,
|
||
|
|
"rewards/rejected": -20.016016006469727,
|
||
|
|
"rewards/weighted_accuracies": 0.53125,
|
||
|
|
"rewards/weighted_chosen": -0.07308082282543182,
|
||
|
|
"rewards/weighted_margins": 0.04105224460363388,
|
||
|
|
"rewards/weighted_rejected": -0.114354707300663,
|
||
|
|
"step": 370
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2030863785155989,
|
||
|
|
"grad_norm": 26.81167221069336,
|
||
|
|
"learning_rate": 9.685932900101146e-07,
|
||
|
|
"logits/chosen": -0.07604064792394638,
|
||
|
|
"logits/rejected": -0.06302948296070099,
|
||
|
|
"logps/chosen": -134.7859344482422,
|
||
|
|
"logps/rejected": -137.11093139648438,
|
||
|
|
"logps/weighted_chosen": -0.98638916015625,
|
||
|
|
"logps/weighted_rejected": -0.6243560910224915,
|
||
|
|
"loss": 0.7116,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -17.729297637939453,
|
||
|
|
"rewards/margins": 1.8025391101837158,
|
||
|
|
"rewards/rejected": -19.528514862060547,
|
||
|
|
"rewards/weighted_accuracies": 0.546875,
|
||
|
|
"rewards/weighted_chosen": -0.11945419013500214,
|
||
|
|
"rewards/weighted_margins": 0.07185058295726776,
|
||
|
|
"rewards/weighted_rejected": -0.1911766082048416,
|
||
|
|
"step": 380
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.20843075689758836,
|
||
|
|
"grad_norm": 40.89613342285156,
|
||
|
|
"learning_rate": 9.652581474390043e-07,
|
||
|
|
"logits/chosen": -0.09274902194738388,
|
||
|
|
"logits/rejected": -0.05452308803796768,
|
||
|
|
"logps/chosen": -139.3249969482422,
|
||
|
|
"logps/rejected": -141.53750610351562,
|
||
|
|
"logps/weighted_chosen": -0.9199386835098267,
|
||
|
|
"logps/weighted_rejected": -0.9217513799667358,
|
||
|
|
"loss": 0.7323,
|
||
|
|
"rewards/accuracies": 0.550000011920929,
|
||
|
|
"rewards/chosen": -19.379688262939453,
|
||
|
|
"rewards/margins": 2.321093797683716,
|
||
|
|
"rewards/rejected": -21.702733993530273,
|
||
|
|
"rewards/weighted_accuracies": 0.543749988079071,
|
||
|
|
"rewards/weighted_chosen": -0.05511016771197319,
|
||
|
|
"rewards/weighted_margins": 0.04057302325963974,
|
||
|
|
"rewards/weighted_rejected": -0.09564266353845596,
|
||
|
|
"step": 390
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.21377513527957778,
|
||
|
|
"grad_norm": 51.860347747802734,
|
||
|
|
"learning_rate": 9.61761086184981e-07,
|
||
|
|
"logits/chosen": -0.10268402099609375,
|
||
|
|
"logits/rejected": -0.05923309177160263,
|
||
|
|
"logps/chosen": -139.50936889648438,
|
||
|
|
"logps/rejected": -141.4015655517578,
|
||
|
|
"logps/weighted_chosen": -0.9449096918106079,
|
||
|
|
"logps/weighted_rejected": -1.177301049232483,
|
||
|
|
"loss": 0.7387,
|
||
|
|
"rewards/accuracies": 0.5625,
|
||
|
|
"rewards/chosen": -20.024608612060547,
|
||
|
|
"rewards/margins": 2.9205079078674316,
|
||
|
|
"rewards/rejected": -22.942577362060547,
|
||
|
|
"rewards/weighted_accuracies": 0.5625,
|
||
|
|
"rewards/weighted_chosen": -0.26060789823532104,
|
||
|
|
"rewards/weighted_margins": 0.017241667956113815,
|
||
|
|
"rewards/weighted_rejected": -0.2779647707939148,
|
||
|
|
"step": 400
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.21911951366156723,
|
||
|
|
"grad_norm": 7.539150238037109,
|
||
|
|
"learning_rate": 9.581033232918629e-07,
|
||
|
|
"logits/chosen": -0.015906523913145065,
|
||
|
|
"logits/rejected": 0.02000865899026394,
|
||
|
|
"logps/chosen": -138.39999389648438,
|
||
|
|
"logps/rejected": -141.68905639648438,
|
||
|
|
"logps/weighted_chosen": -1.0313812494277954,
|
||
|
|
"logps/weighted_rejected": -1.1196472644805908,
|
||
|
|
"loss": 0.7283,
|
||
|
|
"rewards/accuracies": 0.5874999761581421,
|
||
|
|
"rewards/chosen": -21.702342987060547,
|
||
|
|
"rewards/margins": 2.826953172683716,
|
||
|
|
"rewards/rejected": -24.53125,
|
||
|
|
"rewards/weighted_accuracies": 0.5249999761581421,
|
||
|
|
"rewards/weighted_chosen": -0.2783859372138977,
|
||
|
|
"rewards/weighted_margins": 0.06903991848230362,
|
||
|
|
"rewards/weighted_rejected": -0.3477470278739929,
|
||
|
|
"step": 410
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.22446389204355668,
|
||
|
|
"grad_norm": 6.638446807861328,
|
||
|
|
"learning_rate": 9.542861317306952e-07,
|
||
|
|
"logits/chosen": -0.05767669528722763,
|
||
|
|
"logits/rejected": -0.03759918361902237,
|
||
|
|
"logps/chosen": -140.64688110351562,
|
||
|
|
"logps/rejected": -141.9718780517578,
|
||
|
|
"logps/weighted_chosen": -1.5940430164337158,
|
||
|
|
"logps/weighted_rejected": -1.458154320716858,
|
||
|
|
"loss": 0.7371,
|
||
|
|
"rewards/accuracies": 0.5406249761581421,
|
||
|
|
"rewards/chosen": -22.33203125,
|
||
|
|
"rewards/margins": 1.1359374523162842,
|
||
|
|
"rewards/rejected": -23.46484375,
|
||
|
|
"rewards/weighted_accuracies": 0.515625,
|
||
|
|
"rewards/weighted_chosen": -0.299835205078125,
|
||
|
|
"rewards/weighted_margins": 0.032170869410037994,
|
||
|
|
"rewards/weighted_rejected": -0.3319748044013977,
|
||
|
|
"step": 420
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.22980827042554614,
|
||
|
|
"grad_norm": 24.621816635131836,
|
||
|
|
"learning_rate": 9.503108399567308e-07,
|
||
|
|
"logits/chosen": -0.15150222182273865,
|
||
|
|
"logits/rejected": -0.09339599311351776,
|
||
|
|
"logps/chosen": -149.8671875,
|
||
|
|
"logps/rejected": -153.6296844482422,
|
||
|
|
"logps/weighted_chosen": -0.7455719113349915,
|
||
|
|
"logps/weighted_rejected": -0.826556384563446,
|
||
|
|
"loss": 0.7125,
|
||
|
|
"rewards/accuracies": 0.550000011920929,
|
||
|
|
"rewards/chosen": -22.685546875,
|
||
|
|
"rewards/margins": 1.7355468273162842,
|
||
|
|
"rewards/rejected": -24.420312881469727,
|
||
|
|
"rewards/weighted_accuracies": 0.5406249761581421,
|
||
|
|
"rewards/weighted_chosen": -0.19403991103172302,
|
||
|
|
"rewards/weighted_margins": 0.02692871168255806,
|
||
|
|
"rewards/weighted_rejected": -0.2209148406982422,
|
||
|
|
"step": 430
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.23515264880753559,
|
||
|
|
"grad_norm": 10.726137161254883,
|
||
|
|
"learning_rate": 9.461788314471034e-07,
|
||
|
|
"logits/chosen": -0.08421440422534943,
|
||
|
|
"logits/rejected": -0.031351469457149506,
|
||
|
|
"logps/chosen": -143.1171875,
|
||
|
|
"logps/rejected": -146.890625,
|
||
|
|
"logps/weighted_chosen": -0.7524811029434204,
|
||
|
|
"logps/weighted_rejected": -0.8818389773368835,
|
||
|
|
"loss": 0.7062,
|
||
|
|
"rewards/accuracies": 0.550000011920929,
|
||
|
|
"rewards/chosen": -24.516796112060547,
|
||
|
|
"rewards/margins": 2.0640625953674316,
|
||
|
|
"rewards/rejected": -26.577733993530273,
|
||
|
|
"rewards/weighted_accuracies": 0.528124988079071,
|
||
|
|
"rewards/weighted_chosen": -0.22655639052391052,
|
||
|
|
"rewards/weighted_margins": 0.07337188720703125,
|
||
|
|
"rewards/weighted_rejected": -0.30005186796188354,
|
||
|
|
"step": 440
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.240497027189525,
|
||
|
|
"grad_norm": 8.586684226989746,
|
||
|
|
"learning_rate": 9.418915442193509e-07,
|
||
|
|
"logits/chosen": -0.03446502611041069,
|
||
|
|
"logits/rejected": 0.007968139834702015,
|
||
|
|
"logps/chosen": -144.7843780517578,
|
||
|
|
"logps/rejected": -155.96249389648438,
|
||
|
|
"logps/weighted_chosen": -1.1978485584259033,
|
||
|
|
"logps/weighted_rejected": -1.2447845935821533,
|
||
|
|
"loss": 0.6841,
|
||
|
|
"rewards/accuracies": 0.6312500238418579,
|
||
|
|
"rewards/chosen": -26.516016006469727,
|
||
|
|
"rewards/margins": 4.714453220367432,
|
||
|
|
"rewards/rejected": -31.236328125,
|
||
|
|
"rewards/weighted_accuracies": 0.5218750238418579,
|
||
|
|
"rewards/weighted_chosen": -0.25520628690719604,
|
||
|
|
"rewards/weighted_margins": 0.09325332939624786,
|
||
|
|
"rewards/weighted_rejected": -0.3483436703681946,
|
||
|
|
"step": 450
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.24584140557151446,
|
||
|
|
"grad_norm": 10.014894485473633,
|
||
|
|
"learning_rate": 9.374504703309579e-07,
|
||
|
|
"logits/chosen": -0.1379852294921875,
|
||
|
|
"logits/rejected": -0.0861412063241005,
|
||
|
|
"logps/chosen": -141.83828735351562,
|
||
|
|
"logps/rejected": -143.34530639648438,
|
||
|
|
"logps/weighted_chosen": -1.1420501470565796,
|
||
|
|
"logps/weighted_rejected": -1.055877685546875,
|
||
|
|
"loss": 0.6804,
|
||
|
|
"rewards/accuracies": 0.534375011920929,
|
||
|
|
"rewards/chosen": -25.53515625,
|
||
|
|
"rewards/margins": 2.038281202316284,
|
||
|
|
"rewards/rejected": -27.571874618530273,
|
||
|
|
"rewards/weighted_accuracies": 0.5218750238418579,
|
||
|
|
"rewards/weighted_chosen": -0.21983489394187927,
|
||
|
|
"rewards/weighted_margins": 0.17051848769187927,
|
||
|
|
"rewards/weighted_rejected": -0.3904357850551605,
|
||
|
|
"step": 460
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2511857839535039,
|
||
|
|
"grad_norm": 10.024358749389648,
|
||
|
|
"learning_rate": 9.328571553600915e-07,
|
||
|
|
"logits/chosen": -0.1318313628435135,
|
||
|
|
"logits/rejected": -0.0828731507062912,
|
||
|
|
"logps/chosen": -144.39688110351562,
|
||
|
|
"logps/rejected": -147.41796875,
|
||
|
|
"logps/weighted_chosen": -0.9446808099746704,
|
||
|
|
"logps/weighted_rejected": -1.138859510421753,
|
||
|
|
"loss": 0.7137,
|
||
|
|
"rewards/accuracies": 0.5718749761581421,
|
||
|
|
"rewards/chosen": -25.697265625,
|
||
|
|
"rewards/margins": 1.9656250476837158,
|
||
|
|
"rewards/rejected": -27.665233612060547,
|
||
|
|
"rewards/weighted_accuracies": 0.4906249940395355,
|
||
|
|
"rewards/weighted_chosen": -0.2078697234392166,
|
||
|
|
"rewards/weighted_margins": 0.052256010472774506,
|
||
|
|
"rewards/weighted_rejected": -0.260000616312027,
|
||
|
|
"step": 470
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.25653016233549336,
|
||
|
|
"grad_norm": 13.618422508239746,
|
||
|
|
"learning_rate": 9.281131978677106e-07,
|
||
|
|
"logits/chosen": -0.16686096787452698,
|
||
|
|
"logits/rejected": -0.1246364563703537,
|
||
|
|
"logps/chosen": -149.2546844482422,
|
||
|
|
"logps/rejected": -152.47811889648438,
|
||
|
|
"logps/weighted_chosen": -0.686419665813446,
|
||
|
|
"logps/weighted_rejected": -1.0338714122772217,
|
||
|
|
"loss": 0.6805,
|
||
|
|
"rewards/accuracies": 0.59375,
|
||
|
|
"rewards/chosen": -25.559375762939453,
|
||
|
|
"rewards/margins": 3.1332030296325684,
|
||
|
|
"rewards/rejected": -28.689844131469727,
|
||
|
|
"rewards/weighted_accuracies": 0.534375011920929,
|
||
|
|
"rewards/weighted_chosen": -0.22538375854492188,
|
||
|
|
"rewards/weighted_margins": 0.13199158012866974,
|
||
|
|
"rewards/weighted_rejected": -0.35721856355667114,
|
||
|
|
"step": 480
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.2618745407174828,
|
||
|
|
"grad_norm": 30.852863311767578,
|
||
|
|
"learning_rate": 9.232202488412361e-07,
|
||
|
|
"logits/chosen": -0.10437698662281036,
|
||
|
|
"logits/rejected": -0.05600614473223686,
|
||
|
|
"logps/chosen": -143.5671844482422,
|
||
|
|
"logps/rejected": -149.11874389648438,
|
||
|
|
"logps/weighted_chosen": -1.236029028892517,
|
||
|
|
"logps/weighted_rejected": -1.6542327404022217,
|
||
|
|
"loss": 0.6937,
|
||
|
|
"rewards/accuracies": 0.5375000238418579,
|
||
|
|
"rewards/chosen": -28.171483993530273,
|
||
|
|
"rewards/margins": 2.311328172683716,
|
||
|
|
"rewards/rejected": -30.477733612060547,
|
||
|
|
"rewards/weighted_accuracies": 0.534375011920929,
|
||
|
|
"rewards/weighted_chosen": -0.27268296480178833,
|
||
|
|
"rewards/weighted_margins": 0.08454742282629013,
|
||
|
|
"rewards/weighted_rejected": -0.35735321044921875,
|
||
|
|
"step": 490
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.26721891909947226,
|
||
|
|
"grad_norm": 9.151970863342285,
|
||
|
|
"learning_rate": 9.181800111199766e-07,
|
||
|
|
"logits/chosen": -0.1230342835187912,
|
||
|
|
"logits/rejected": -0.0801507979631424,
|
||
|
|
"logps/chosen": -147.54843139648438,
|
||
|
|
"logps/rejected": -147.5437469482422,
|
||
|
|
"logps/weighted_chosen": -1.5909912586212158,
|
||
|
|
"logps/weighted_rejected": -1.286199927330017,
|
||
|
|
"loss": 0.7005,
|
||
|
|
"rewards/accuracies": 0.578125,
|
||
|
|
"rewards/chosen": -29.640233993530273,
|
||
|
|
"rewards/margins": 3.1507811546325684,
|
||
|
|
"rewards/rejected": -32.789451599121094,
|
||
|
|
"rewards/weighted_accuracies": 0.5874999761581421,
|
||
|
|
"rewards/weighted_chosen": -0.34203338623046875,
|
||
|
|
"rewards/weighted_margins": 0.12214355170726776,
|
||
|
|
"rewards/weighted_rejected": -0.4641265869140625,
|
||
|
|
"step": 500
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"epoch": 0.26721891909947226,
|
||
|
|
"eval_logits/chosen": -0.29476672410964966,
|
||
|
|
"eval_logits/rejected": -0.26891082525253296,
|
||
|
|
"eval_logps/chosen": -149.5992889404297,
|
||
|
|
"eval_logps/rejected": -153.5249481201172,
|
||
|
|
"eval_logps/weighted_chosen": -0.8761399984359741,
|
||
|
|
"eval_logps/weighted_rejected": -1.120854377746582,
|
||
|
|
"eval_loss": 0.6999610066413879,
|
||
|
|
"eval_rewards/accuracies": 0.5784114003181458,
|
||
|
|
"eval_rewards/chosen": -28.267057418823242,
|
||
|
|
"eval_rewards/margins": 2.7842414379119873,
|
||
|
|
"eval_rewards/rejected": -31.05422592163086,
|
||
|
|
"eval_rewards/weighted_accuracies": 0.5264765620231628,
|
||
|
|
"eval_rewards/weighted_chosen": -0.23699289560317993,
|
||
|
|
"eval_rewards/weighted_margins": 0.08454981446266174,
|
||
|
|
"eval_rewards/weighted_rejected": -0.3215427100658417,
|
||
|
|
"eval_runtime": 1248.3443,
|
||
|
|
"eval_samples_per_second": 1.571,
|
||
|
|
"eval_steps_per_second": 0.393,
|
||
|
|
"step": 500
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"logging_steps": 10,
|
||
|
|
"max_steps": 1872,
|
||
|
|
"num_input_tokens_seen": 0,
|
||
|
|
"num_train_epochs": 1,
|
||
|
|
"save_steps": 500,
|
||
|
|
"stateful_callbacks": {
|
||
|
|
"TrainerControl": {
|
||
|
|
"args": {
|
||
|
|
"should_epoch_stop": false,
|
||
|
|
"should_evaluate": false,
|
||
|
|
"should_log": false,
|
||
|
|
"should_save": true,
|
||
|
|
"should_training_stop": false
|
||
|
|
},
|
||
|
|
"attributes": {}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"total_flos": 0.0,
|
||
|
|
"train_batch_size": 1,
|
||
|
|
"trial_name": null,
|
||
|
|
"trial_params": null
|
||
|
|
}
|