alvarobartt's picture
alvarobartt HF staff
Model save
3c4912e verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 500,
"global_step": 711,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.04,
"grad_norm": 49.48526136809559,
"learning_rate": 5.000000000000001e-07,
"log_odds_chosen": 0.14267238974571228,
"log_odds_ratio": -0.6910250782966614,
"logits/chosen": -2.8734333515167236,
"logits/rejected": -2.8792202472686768,
"logps/chosen": -1.2169833183288574,
"logps/rejected": -1.3137216567993164,
"loss": 1.1778,
"nll_loss": 1.1760419607162476,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.06084916740655899,
"rewards/margins": 0.004836911801248789,
"rewards/rejected": -0.06568608433008194,
"step": 10
},
{
"epoch": 0.08,
"grad_norm": 13.42628442764115,
"learning_rate": 1.0000000000000002e-06,
"log_odds_chosen": 0.2992035746574402,
"log_odds_ratio": -0.5965026617050171,
"logits/chosen": -2.9965202808380127,
"logits/rejected": -2.9929251670837402,
"logps/chosen": -0.9654132723808289,
"logps/rejected": -1.1591482162475586,
"loss": 1.0,
"nll_loss": 0.9463264346122742,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.04827066510915756,
"rewards/margins": 0.009686745703220367,
"rewards/rejected": -0.05795741081237793,
"step": 20
},
{
"epoch": 0.13,
"grad_norm": 9.276480900326913,
"learning_rate": 1.5e-06,
"log_odds_chosen": 0.35957902669906616,
"log_odds_ratio": -0.5845057368278503,
"logits/chosen": -3.0192956924438477,
"logits/rejected": -3.018825054168701,
"logps/chosen": -0.8531354665756226,
"logps/rejected": -1.0771838426589966,
"loss": 0.9246,
"nll_loss": 0.8811793327331543,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.04265677183866501,
"rewards/margins": 0.011202414520084858,
"rewards/rejected": -0.05385918542742729,
"step": 30
},
{
"epoch": 0.17,
"grad_norm": 8.627451557441312,
"learning_rate": 2.0000000000000003e-06,
"log_odds_chosen": 0.25122153759002686,
"log_odds_ratio": -0.6361113786697388,
"logits/chosen": -2.915778160095215,
"logits/rejected": -2.919457197189331,
"logps/chosen": -0.8648379445075989,
"logps/rejected": -0.9924612045288086,
"loss": 0.974,
"nll_loss": 0.9005156755447388,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.04324189946055412,
"rewards/margins": 0.0063811675645411015,
"rewards/rejected": -0.04962306469678879,
"step": 40
},
{
"epoch": 0.21,
"grad_norm": 10.358595971591946,
"learning_rate": 2.5e-06,
"log_odds_chosen": 0.20706939697265625,
"log_odds_ratio": -0.6479698419570923,
"logits/chosen": -2.931480884552002,
"logits/rejected": -2.941484212875366,
"logps/chosen": -0.9367877244949341,
"logps/rejected": -1.0564123392105103,
"loss": 0.9112,
"nll_loss": 0.9363681674003601,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.046839382499456406,
"rewards/margins": 0.005981230176985264,
"rewards/rejected": -0.052820611745119095,
"step": 50
},
{
"epoch": 0.25,
"grad_norm": 10.184305920593003,
"learning_rate": 3e-06,
"log_odds_chosen": 0.274976909160614,
"log_odds_ratio": -0.6372017860412598,
"logits/chosen": -2.887624979019165,
"logits/rejected": -2.902449369430542,
"logps/chosen": -0.8582770228385925,
"logps/rejected": -1.012649416923523,
"loss": 0.9268,
"nll_loss": 0.9307074546813965,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.04291384667158127,
"rewards/margins": 0.007718615233898163,
"rewards/rejected": -0.05063246563076973,
"step": 60
},
{
"epoch": 0.3,
"grad_norm": 10.247661529064816,
"learning_rate": 3.5e-06,
"log_odds_chosen": 0.29217422008514404,
"log_odds_ratio": -0.6399118304252625,
"logits/chosen": -2.9243764877319336,
"logits/rejected": -2.9292612075805664,
"logps/chosen": -0.8583773374557495,
"logps/rejected": -1.0340784788131714,
"loss": 0.9441,
"nll_loss": 0.9250191450119019,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.042918868362903595,
"rewards/margins": 0.008785055950284004,
"rewards/rejected": -0.05170391872525215,
"step": 70
},
{
"epoch": 0.34,
"grad_norm": 8.823827784258985,
"learning_rate": 4.000000000000001e-06,
"log_odds_chosen": 0.33998143672943115,
"log_odds_ratio": -0.6087978482246399,
"logits/chosen": -2.901061773300171,
"logits/rejected": -2.909398078918457,
"logps/chosen": -0.8597785234451294,
"logps/rejected": -1.0702154636383057,
"loss": 0.9265,
"nll_loss": 0.883587658405304,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.04298892617225647,
"rewards/margins": 0.010521845892071724,
"rewards/rejected": -0.053510766476392746,
"step": 80
},
{
"epoch": 0.38,
"grad_norm": 9.3837801640132,
"learning_rate": 4.5e-06,
"log_odds_chosen": 0.32301634550094604,
"log_odds_ratio": -0.645144522190094,
"logits/chosen": -2.9571635723114014,
"logits/rejected": -2.9651668071746826,
"logps/chosen": -0.8539797067642212,
"logps/rejected": -1.0550405979156494,
"loss": 0.9336,
"nll_loss": 0.9078629612922668,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.04269898682832718,
"rewards/margins": 0.010053041391074657,
"rewards/rejected": -0.05275202915072441,
"step": 90
},
{
"epoch": 0.42,
"grad_norm": 8.451957287235773,
"learning_rate": 5e-06,
"log_odds_chosen": 0.27936068177223206,
"log_odds_ratio": -0.619978129863739,
"logits/chosen": -2.9700396060943604,
"logits/rejected": -2.9722213745117188,
"logps/chosen": -0.892767608165741,
"logps/rejected": -1.0630358457565308,
"loss": 0.9536,
"nll_loss": 0.9281327128410339,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.04463838413357735,
"rewards/margins": 0.008513410575687885,
"rewards/rejected": -0.05315179377794266,
"step": 100
},
{
"epoch": 0.46,
"grad_norm": 10.228379998303796,
"learning_rate": 4.767312946227961e-06,
"log_odds_chosen": 0.18160709738731384,
"log_odds_ratio": -0.7056410908699036,
"logits/chosen": -2.980069637298584,
"logits/rejected": -2.9743785858154297,
"logps/chosen": -0.9338927268981934,
"logps/rejected": -1.0694698095321655,
"loss": 0.949,
"nll_loss": 0.9810144305229187,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.046694640070199966,
"rewards/margins": 0.006778855808079243,
"rewards/rejected": -0.053473494946956635,
"step": 110
},
{
"epoch": 0.51,
"grad_norm": 15.501078885228303,
"learning_rate": 4.564354645876385e-06,
"log_odds_chosen": 0.3752610683441162,
"log_odds_ratio": -0.5817548036575317,
"logits/chosen": -2.8941051959991455,
"logits/rejected": -2.898369073867798,
"logps/chosen": -0.8418710827827454,
"logps/rejected": -1.0447475910186768,
"loss": 0.9253,
"nll_loss": 0.908086895942688,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.04209355264902115,
"rewards/margins": 0.010143828578293324,
"rewards/rejected": -0.0522373802959919,
"step": 120
},
{
"epoch": 0.55,
"grad_norm": 8.598131028345922,
"learning_rate": 4.385290096535147e-06,
"log_odds_chosen": 0.2913978397846222,
"log_odds_ratio": -0.6210304498672485,
"logits/chosen": -2.8913044929504395,
"logits/rejected": -2.900383234024048,
"logps/chosen": -0.9610312581062317,
"logps/rejected": -1.1445128917694092,
"loss": 0.979,
"nll_loss": 1.0131185054779053,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.048051562160253525,
"rewards/margins": 0.00917408149689436,
"rewards/rejected": -0.05722564458847046,
"step": 130
},
{
"epoch": 0.59,
"grad_norm": 8.665950821616725,
"learning_rate": 4.2257712736425835e-06,
"log_odds_chosen": 0.31699851155281067,
"log_odds_ratio": -0.6054994463920593,
"logits/chosen": -2.866349697113037,
"logits/rejected": -2.872480869293213,
"logps/chosen": -0.8127169609069824,
"logps/rejected": -1.005972146987915,
"loss": 0.9264,
"nll_loss": 0.8691350817680359,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.040635846555233,
"rewards/margins": 0.009662764146924019,
"rewards/rejected": -0.05029861256480217,
"step": 140
},
{
"epoch": 0.63,
"grad_norm": 7.610661708462692,
"learning_rate": 4.082482904638631e-06,
"log_odds_chosen": 0.26200634241104126,
"log_odds_ratio": -0.635285496711731,
"logits/chosen": -2.813951253890991,
"logits/rejected": -2.831160306930542,
"logps/chosen": -0.8999479413032532,
"logps/rejected": -1.0443265438079834,
"loss": 0.9319,
"nll_loss": 0.9246827363967896,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.04499739781022072,
"rewards/margins": 0.0072189331986010075,
"rewards/rejected": -0.05221632868051529,
"step": 150
},
{
"epoch": 0.68,
"grad_norm": 7.926717564356759,
"learning_rate": 3.952847075210474e-06,
"log_odds_chosen": 0.4248151183128357,
"log_odds_ratio": -0.5570122599601746,
"logits/chosen": -2.8344883918762207,
"logits/rejected": -2.8441624641418457,
"logps/chosen": -0.8424416780471802,
"logps/rejected": -1.1050217151641846,
"loss": 0.9371,
"nll_loss": 0.8941017985343933,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.04212208464741707,
"rewards/margins": 0.01312900148332119,
"rewards/rejected": -0.05525108426809311,
"step": 160
},
{
"epoch": 0.72,
"grad_norm": 8.768416551022222,
"learning_rate": 3.834824944236852e-06,
"log_odds_chosen": 0.280124694108963,
"log_odds_ratio": -0.6224346160888672,
"logits/chosen": -2.8287386894226074,
"logits/rejected": -2.83266019821167,
"logps/chosen": -0.8510664105415344,
"logps/rejected": -1.0343472957611084,
"loss": 0.9355,
"nll_loss": 0.8820816278457642,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.04255332052707672,
"rewards/margins": 0.009164043702185154,
"rewards/rejected": -0.0517173632979393,
"step": 170
},
{
"epoch": 0.76,
"grad_norm": 7.46076847939469,
"learning_rate": 3.72677996249965e-06,
"log_odds_chosen": 0.17997188866138458,
"log_odds_ratio": -0.6717725992202759,
"logits/chosen": -2.8428828716278076,
"logits/rejected": -2.8347556591033936,
"logps/chosen": -0.9121964573860168,
"logps/rejected": -0.9958919286727905,
"loss": 0.946,
"nll_loss": 0.9758418202400208,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.04560982435941696,
"rewards/margins": 0.004184772726148367,
"rewards/rejected": -0.04979459568858147,
"step": 180
},
{
"epoch": 0.8,
"grad_norm": 7.680949233012938,
"learning_rate": 3.6273812505500587e-06,
"log_odds_chosen": 0.35985735058784485,
"log_odds_ratio": -0.5938600301742554,
"logits/chosen": -2.859172821044922,
"logits/rejected": -2.8654944896698,
"logps/chosen": -0.847649097442627,
"logps/rejected": -1.0266352891921997,
"loss": 0.9285,
"nll_loss": 0.8830558657646179,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.04238245636224747,
"rewards/margins": 0.008949300274252892,
"rewards/rejected": -0.05133175849914551,
"step": 190
},
{
"epoch": 0.84,
"grad_norm": 9.078983143397938,
"learning_rate": 3.5355339059327378e-06,
"log_odds_chosen": 0.4591992497444153,
"log_odds_ratio": -0.5660054087638855,
"logits/chosen": -2.9353525638580322,
"logits/rejected": -2.9381651878356934,
"logps/chosen": -0.9107562303543091,
"logps/rejected": -1.1601167917251587,
"loss": 0.943,
"nll_loss": 1.0148041248321533,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.045537810772657394,
"rewards/margins": 0.012468031607568264,
"rewards/rejected": -0.05800584703683853,
"step": 200
},
{
"epoch": 0.89,
"grad_norm": 8.44546881476589,
"learning_rate": 3.450327796711771e-06,
"log_odds_chosen": 0.29461508989334106,
"log_odds_ratio": -0.6495056748390198,
"logits/chosen": -2.90964412689209,
"logits/rejected": -2.9097769260406494,
"logps/chosen": -0.8788071870803833,
"logps/rejected": -1.0280598402023315,
"loss": 0.9482,
"nll_loss": 0.9459528923034668,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.043940357863903046,
"rewards/margins": 0.007462637964636087,
"rewards/rejected": -0.051402997225522995,
"step": 210
},
{
"epoch": 0.93,
"grad_norm": 8.193251853529034,
"learning_rate": 3.3709993123162106e-06,
"log_odds_chosen": 0.4032968580722809,
"log_odds_ratio": -0.568686842918396,
"logits/chosen": -2.9146342277526855,
"logits/rejected": -2.9228734970092773,
"logps/chosen": -0.8045447468757629,
"logps/rejected": -1.03438401222229,
"loss": 0.9203,
"nll_loss": 0.8637560606002808,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.04022723808884621,
"rewards/margins": 0.011491960845887661,
"rewards/rejected": -0.05171919986605644,
"step": 220
},
{
"epoch": 0.97,
"grad_norm": 8.324361077294697,
"learning_rate": 3.296902366978936e-06,
"log_odds_chosen": 0.36563342809677124,
"log_odds_ratio": -0.5865170955657959,
"logits/chosen": -2.9637176990509033,
"logits/rejected": -2.9688830375671387,
"logps/chosen": -0.8463845252990723,
"logps/rejected": -1.061700701713562,
"loss": 0.9112,
"nll_loss": 0.9032902717590332,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.04231923073530197,
"rewards/margins": 0.010765810497105122,
"rewards/rejected": -0.05308504030108452,
"step": 230
},
{
"epoch": 1.01,
"grad_norm": 9.173894745062437,
"learning_rate": 3.2274861218395142e-06,
"log_odds_chosen": 0.5646923184394836,
"log_odds_ratio": -0.550338625907898,
"logits/chosen": -2.9580464363098145,
"logits/rejected": -2.9572625160217285,
"logps/chosen": -0.7748560309410095,
"logps/rejected": -1.0799105167388916,
"loss": 0.8136,
"nll_loss": 0.8241481781005859,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.038742803037166595,
"rewards/margins": 0.015252722427248955,
"rewards/rejected": -0.0539955273270607,
"step": 240
},
{
"epoch": 1.05,
"grad_norm": 9.099732749463024,
"learning_rate": 3.1622776601683796e-06,
"log_odds_chosen": 1.4823018312454224,
"log_odds_ratio": -0.2659846544265747,
"logits/chosen": -2.9596664905548096,
"logits/rejected": -2.9568517208099365,
"logps/chosen": -0.47842296957969666,
"logps/rejected": -1.2101627588272095,
"loss": 0.5233,
"nll_loss": 0.5237408876419067,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.023921148851513863,
"rewards/margins": 0.03658699244260788,
"rewards/rejected": -0.06050813943147659,
"step": 250
},
{
"epoch": 1.1,
"grad_norm": 9.707924912075288,
"learning_rate": 3.1008683647302113e-06,
"log_odds_chosen": 1.2276315689086914,
"log_odds_ratio": -0.337185263633728,
"logits/chosen": -2.9387476444244385,
"logits/rejected": -2.947472333908081,
"logps/chosen": -0.507585346698761,
"logps/rejected": -1.0944607257843018,
"loss": 0.5243,
"nll_loss": 0.5015721321105957,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.02537926472723484,
"rewards/margins": 0.02934376522898674,
"rewards/rejected": -0.05472303181886673,
"step": 260
},
{
"epoch": 1.14,
"grad_norm": 9.72287079567634,
"learning_rate": 3.0429030972509227e-06,
"log_odds_chosen": 1.6538162231445312,
"log_odds_ratio": -0.23319308459758759,
"logits/chosen": -3.0101027488708496,
"logits/rejected": -3.0216686725616455,
"logps/chosen": -0.4229302406311035,
"logps/rejected": -1.2452623844146729,
"loss": 0.4944,
"nll_loss": 0.4760821759700775,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.021146511659026146,
"rewards/margins": 0.04111660644412041,
"rewards/rejected": -0.0622631199657917,
"step": 270
},
{
"epoch": 1.18,
"grad_norm": 8.91121542892108,
"learning_rate": 2.988071523335984e-06,
"log_odds_chosen": 1.5969759225845337,
"log_odds_ratio": -0.24297292530536652,
"logits/chosen": -2.9810919761657715,
"logits/rejected": -2.984095335006714,
"logps/chosen": -0.4323214888572693,
"logps/rejected": -1.1894398927688599,
"loss": 0.4899,
"nll_loss": 0.4894355237483978,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.021616075187921524,
"rewards/margins": 0.03785592317581177,
"rewards/rejected": -0.05947200208902359,
"step": 280
},
{
"epoch": 1.22,
"grad_norm": 9.172297878492826,
"learning_rate": 2.9361010975735177e-06,
"log_odds_chosen": 1.5589730739593506,
"log_odds_ratio": -0.2480790615081787,
"logits/chosen": -2.936096668243408,
"logits/rejected": -2.9368419647216797,
"logps/chosen": -0.4527331292629242,
"logps/rejected": -1.1788005828857422,
"loss": 0.489,
"nll_loss": 0.48634010553359985,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.0226366575807333,
"rewards/margins": 0.03630336746573448,
"rewards/rejected": -0.05894003063440323,
"step": 290
},
{
"epoch": 1.27,
"grad_norm": 9.341330783727024,
"learning_rate": 2.8867513459481293e-06,
"log_odds_chosen": 1.5156631469726562,
"log_odds_ratio": -0.2381705492734909,
"logits/chosen": -2.949042320251465,
"logits/rejected": -2.934840202331543,
"logps/chosen": -0.43807506561279297,
"logps/rejected": -1.1634423732757568,
"loss": 0.5031,
"nll_loss": 0.4857659935951233,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.0219037514179945,
"rewards/margins": 0.03626837208867073,
"rewards/rejected": -0.05817211791872978,
"step": 300
},
{
"epoch": 1.31,
"grad_norm": 9.51833987789547,
"learning_rate": 2.839809171235324e-06,
"log_odds_chosen": 1.4176826477050781,
"log_odds_ratio": -0.281759113073349,
"logits/chosen": -2.953523874282837,
"logits/rejected": -2.954210042953491,
"logps/chosen": -0.47912734746932983,
"logps/rejected": -1.2172271013259888,
"loss": 0.5149,
"nll_loss": 0.4759243428707123,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.02395637147128582,
"rewards/margins": 0.03690498322248459,
"rewards/rejected": -0.060861360281705856,
"step": 310
},
{
"epoch": 1.35,
"grad_norm": 9.733927381742513,
"learning_rate": 2.7950849718747376e-06,
"log_odds_chosen": 1.33830726146698,
"log_odds_ratio": -0.2881857454776764,
"logits/chosen": -3.0379624366760254,
"logits/rejected": -3.0340828895568848,
"logps/chosen": -0.5138579607009888,
"logps/rejected": -1.1911945343017578,
"loss": 0.5378,
"nll_loss": 0.5391749739646912,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.025692898780107498,
"rewards/margins": 0.03386683017015457,
"rewards/rejected": -0.05955972522497177,
"step": 320
},
{
"epoch": 1.39,
"grad_norm": 8.811470336283001,
"learning_rate": 2.752409412815902e-06,
"log_odds_chosen": 1.485058307647705,
"log_odds_ratio": -0.26495322585105896,
"logits/chosen": -2.9619758129119873,
"logits/rejected": -2.954977035522461,
"logps/chosen": -0.440424382686615,
"logps/rejected": -1.131378412246704,
"loss": 0.4785,
"nll_loss": 0.4669272303581238,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.0220212172716856,
"rewards/margins": 0.034547705203294754,
"rewards/rejected": -0.056568920612335205,
"step": 330
},
{
"epoch": 1.43,
"grad_norm": 8.015826159950372,
"learning_rate": 2.711630722733202e-06,
"log_odds_chosen": 1.5126841068267822,
"log_odds_ratio": -0.25113973021507263,
"logits/chosen": -2.979628801345825,
"logits/rejected": -2.9764034748077393,
"logps/chosen": -0.42568206787109375,
"logps/rejected": -1.1240874528884888,
"loss": 0.5006,
"nll_loss": 0.46929699182510376,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.021284103393554688,
"rewards/margins": 0.03492026776075363,
"rewards/rejected": -0.05620437115430832,
"step": 340
},
{
"epoch": 1.48,
"grad_norm": 8.45163306773323,
"learning_rate": 2.6726124191242444e-06,
"log_odds_chosen": 1.3055660724639893,
"log_odds_ratio": -0.3173283338546753,
"logits/chosen": -2.961458683013916,
"logits/rejected": -2.9598679542541504,
"logps/chosen": -0.5269508957862854,
"logps/rejected": -1.1968458890914917,
"loss": 0.4854,
"nll_loss": 0.5489671230316162,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.02634754404425621,
"rewards/margins": 0.033494751900434494,
"rewards/rejected": -0.059842299669981,
"step": 350
},
{
"epoch": 1.52,
"grad_norm": 8.682519620016398,
"learning_rate": 2.6352313834736496e-06,
"log_odds_chosen": 1.644923210144043,
"log_odds_ratio": -0.23544283211231232,
"logits/chosen": -2.930689573287964,
"logits/rejected": -2.928385019302368,
"logps/chosen": -0.4198189675807953,
"logps/rejected": -1.2138855457305908,
"loss": 0.5002,
"nll_loss": 0.45420289039611816,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.020990949124097824,
"rewards/margins": 0.03970332816243172,
"rewards/rejected": -0.06069428473711014,
"step": 360
},
{
"epoch": 1.56,
"grad_norm": 9.273291108083583,
"learning_rate": 2.599376224550182e-06,
"log_odds_chosen": 1.5730557441711426,
"log_odds_ratio": -0.2532302439212799,
"logits/chosen": -2.9642691612243652,
"logits/rejected": -2.9628634452819824,
"logps/chosen": -0.4353133738040924,
"logps/rejected": -1.1966689825057983,
"loss": 0.5056,
"nll_loss": 0.48967617750167847,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.02176566980779171,
"rewards/margins": 0.0380677804350853,
"rewards/rejected": -0.05983344465494156,
"step": 370
},
{
"epoch": 1.6,
"grad_norm": 8.84508067219799,
"learning_rate": 2.564945880212886e-06,
"log_odds_chosen": 1.6267578601837158,
"log_odds_ratio": -0.24944674968719482,
"logits/chosen": -2.934312582015991,
"logits/rejected": -2.9307126998901367,
"logps/chosen": -0.44339004158973694,
"logps/rejected": -1.2226251363754272,
"loss": 0.4946,
"nll_loss": 0.44669169187545776,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.022169504314661026,
"rewards/margins": 0.038961753249168396,
"rewards/rejected": -0.06113125756382942,
"step": 380
},
{
"epoch": 1.65,
"grad_norm": 7.973489385408531,
"learning_rate": 2.5318484177091667e-06,
"log_odds_chosen": 1.4293986558914185,
"log_odds_ratio": -0.26934128999710083,
"logits/chosen": -2.927151918411255,
"logits/rejected": -2.9325613975524902,
"logps/chosen": -0.48316580057144165,
"logps/rejected": -1.1627215147018433,
"loss": 0.5043,
"nll_loss": 0.5102673172950745,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.024158289656043053,
"rewards/margins": 0.03397778421640396,
"rewards/rejected": -0.05813607573509216,
"step": 390
},
{
"epoch": 1.69,
"grad_norm": 8.824910828626702,
"learning_rate": 2.5e-06,
"log_odds_chosen": 1.6508582830429077,
"log_odds_ratio": -0.24078328907489777,
"logits/chosen": -2.9304981231689453,
"logits/rejected": -2.9384522438049316,
"logps/chosen": -0.4645829200744629,
"logps/rejected": -1.2513707876205444,
"loss": 0.528,
"nll_loss": 0.5313082933425903,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.023229148238897324,
"rewards/margins": 0.03933940455317497,
"rewards/rejected": -0.0625685527920723,
"step": 400
},
{
"epoch": 1.73,
"grad_norm": 9.94593557333145,
"learning_rate": 2.4693239916239746e-06,
"log_odds_chosen": 1.5409417152404785,
"log_odds_ratio": -0.26054129004478455,
"logits/chosen": -2.8831980228424072,
"logits/rejected": -2.881943702697754,
"logps/chosen": -0.42643943428993225,
"logps/rejected": -1.1698906421661377,
"loss": 0.5037,
"nll_loss": 0.464557409286499,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.021321970969438553,
"rewards/margins": 0.03717255964875221,
"rewards/rejected": -0.05849452689290047,
"step": 410
},
{
"epoch": 1.77,
"grad_norm": 8.290147408251284,
"learning_rate": 2.4397501823713327e-06,
"log_odds_chosen": 1.3443034887313843,
"log_odds_ratio": -0.3111021816730499,
"logits/chosen": -2.867410182952881,
"logits/rejected": -2.867697238922119,
"logps/chosen": -0.5130647420883179,
"logps/rejected": -1.2247389554977417,
"loss": 0.4997,
"nll_loss": 0.5347863435745239,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.025653237476944923,
"rewards/margins": 0.03558371216058731,
"rewards/rejected": -0.06123695522546768,
"step": 420
},
{
"epoch": 1.81,
"grad_norm": 10.095422116087253,
"learning_rate": 2.411214110852061e-06,
"log_odds_chosen": 1.538722276687622,
"log_odds_ratio": -0.25320082902908325,
"logits/chosen": -2.9051260948181152,
"logits/rejected": -2.907191038131714,
"logps/chosen": -0.49453073740005493,
"logps/rejected": -1.2816953659057617,
"loss": 0.5557,
"nll_loss": 0.5553343892097473,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.024726541712880135,
"rewards/margins": 0.0393582321703434,
"rewards/rejected": -0.06408476829528809,
"step": 430
},
{
"epoch": 1.86,
"grad_norm": 9.482272989808482,
"learning_rate": 2.3836564731139807e-06,
"log_odds_chosen": 1.701960802078247,
"log_odds_ratio": -0.23092034459114075,
"logits/chosen": -2.88254976272583,
"logits/rejected": -2.880296230316162,
"logps/chosen": -0.41986674070358276,
"logps/rejected": -1.1859886646270752,
"loss": 0.5171,
"nll_loss": 0.4521302282810211,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.020993337035179138,
"rewards/margins": 0.0383060947060585,
"rewards/rejected": -0.05929943174123764,
"step": 440
},
{
"epoch": 1.9,
"grad_norm": 10.287479798932107,
"learning_rate": 2.357022603955159e-06,
"log_odds_chosen": 1.5226378440856934,
"log_odds_ratio": -0.26804596185684204,
"logits/chosen": -2.9180657863616943,
"logits/rejected": -2.9176905155181885,
"logps/chosen": -0.4231870174407959,
"logps/rejected": -1.1624265909194946,
"loss": 0.4775,
"nll_loss": 0.42117613554000854,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.021159350872039795,
"rewards/margins": 0.036961980164051056,
"rewards/rejected": -0.05812133103609085,
"step": 450
},
{
"epoch": 1.94,
"grad_norm": 8.629730433025097,
"learning_rate": 2.3312620206007847e-06,
"log_odds_chosen": 1.3161282539367676,
"log_odds_ratio": -0.3122040629386902,
"logits/chosen": -2.9203555583953857,
"logits/rejected": -2.9158670902252197,
"logps/chosen": -0.46862277388572693,
"logps/rejected": -1.085384726524353,
"loss": 0.5357,
"nll_loss": 0.4903482496738434,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.023431137204170227,
"rewards/margins": 0.030838098376989365,
"rewards/rejected": -0.05426923185586929,
"step": 460
},
{
"epoch": 1.98,
"grad_norm": 9.462108983797318,
"learning_rate": 2.3063280200722128e-06,
"log_odds_chosen": 1.5001277923583984,
"log_odds_ratio": -0.2745407521724701,
"logits/chosen": -2.918792724609375,
"logits/rejected": -2.920560598373413,
"logps/chosen": -0.45306238532066345,
"logps/rejected": -1.1488702297210693,
"loss": 0.5047,
"nll_loss": 0.46945127844810486,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.02265312150120735,
"rewards/margins": 0.034790389239788055,
"rewards/rejected": -0.05744351074099541,
"step": 470
},
{
"epoch": 2.03,
"grad_norm": 12.033704149037684,
"learning_rate": 2.2821773229381924e-06,
"log_odds_chosen": 2.368060350418091,
"log_odds_ratio": -0.13723622262477875,
"logits/chosen": -2.896536111831665,
"logits/rejected": -2.9012863636016846,
"logps/chosen": -0.26962071657180786,
"logps/rejected": -1.277259349822998,
"loss": 0.2775,
"nll_loss": 0.2636214792728424,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.013481035828590393,
"rewards/margins": 0.05038193613290787,
"rewards/rejected": -0.06386296451091766,
"step": 480
},
{
"epoch": 2.07,
"grad_norm": 8.31587372487098,
"learning_rate": 2.2587697572631284e-06,
"log_odds_chosen": 3.0361766815185547,
"log_odds_ratio": -0.0762939602136612,
"logits/chosen": -2.872748374938965,
"logits/rejected": -2.8818652629852295,
"logps/chosen": -0.17086732387542725,
"logps/rejected": -1.341355562210083,
"loss": 0.1837,
"nll_loss": 0.18102502822875977,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008543366566300392,
"rewards/margins": 0.05852441117167473,
"rewards/rejected": -0.06706777215003967,
"step": 490
},
{
"epoch": 2.11,
"grad_norm": 9.109087690728854,
"learning_rate": 2.23606797749979e-06,
"log_odds_chosen": 3.1204092502593994,
"log_odds_ratio": -0.0752432718873024,
"logits/chosen": -2.842435121536255,
"logits/rejected": -2.851132869720459,
"logps/chosen": -0.1684246063232422,
"logps/rejected": -1.4159977436065674,
"loss": 0.1865,
"nll_loss": 0.1788669377565384,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008421231061220169,
"rewards/margins": 0.0623786523938179,
"rewards/rejected": -0.07079987227916718,
"step": 500
},
{
"epoch": 2.15,
"grad_norm": 11.342602042204765,
"learning_rate": 2.2140372138502386e-06,
"log_odds_chosen": 3.1781134605407715,
"log_odds_ratio": -0.06865429133176804,
"logits/chosen": -2.842808485031128,
"logits/rejected": -2.8529012203216553,
"logps/chosen": -0.15066179633140564,
"logps/rejected": -1.4150397777557373,
"loss": 0.1774,
"nll_loss": 0.15856973826885223,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007533090654760599,
"rewards/margins": 0.06321889907121658,
"rewards/rejected": -0.07075198739767075,
"step": 510
},
{
"epoch": 2.19,
"grad_norm": 9.015217844997702,
"learning_rate": 2.1926450482675734e-06,
"log_odds_chosen": 3.1590893268585205,
"log_odds_ratio": -0.05701795965433121,
"logits/chosen": -2.8566677570343018,
"logits/rejected": -2.8613815307617188,
"logps/chosen": -0.17371872067451477,
"logps/rejected": -1.5091571807861328,
"loss": 0.1875,
"nll_loss": 0.1826210618019104,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008685936219990253,
"rewards/margins": 0.06677192449569702,
"rewards/rejected": -0.0754578709602356,
"step": 520
},
{
"epoch": 2.24,
"grad_norm": 7.705666021658087,
"learning_rate": 2.1718612138153473e-06,
"log_odds_chosen": 3.129737615585327,
"log_odds_ratio": -0.07283337414264679,
"logits/chosen": -2.857102632522583,
"logits/rejected": -2.8588953018188477,
"logps/chosen": -0.16752207279205322,
"logps/rejected": -1.4430062770843506,
"loss": 0.1864,
"nll_loss": 0.18135443329811096,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008376102894544601,
"rewards/margins": 0.0637742131948471,
"rewards/rejected": -0.072150319814682,
"step": 530
},
{
"epoch": 2.28,
"grad_norm": 10.453820153776633,
"learning_rate": 2.151657414559676e-06,
"log_odds_chosen": 3.152796983718872,
"log_odds_ratio": -0.07412304729223251,
"logits/chosen": -2.8558595180511475,
"logits/rejected": -2.8634510040283203,
"logps/chosen": -0.1711394190788269,
"logps/rejected": -1.419811725616455,
"loss": 0.1867,
"nll_loss": 0.18491069972515106,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.0085569703951478,
"rewards/margins": 0.06243361160159111,
"rewards/rejected": -0.07099058479070663,
"step": 540
},
{
"epoch": 2.32,
"grad_norm": 9.218290542920707,
"learning_rate": 2.132007163556104e-06,
"log_odds_chosen": 2.85337233543396,
"log_odds_ratio": -0.10947513580322266,
"logits/chosen": -2.851966142654419,
"logits/rejected": -2.8540444374084473,
"logps/chosen": -0.19084951281547546,
"logps/rejected": -1.3400590419769287,
"loss": 0.1955,
"nll_loss": 0.19958652555942535,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.009542474523186684,
"rewards/margins": 0.057460468262434006,
"rewards/rejected": -0.06700295209884644,
"step": 550
},
{
"epoch": 2.36,
"grad_norm": 10.317365217804,
"learning_rate": 2.1128856368212917e-06,
"log_odds_chosen": 2.9810779094696045,
"log_odds_ratio": -0.09573061764240265,
"logits/chosen": -2.842881441116333,
"logits/rejected": -2.8447251319885254,
"logps/chosen": -0.19754521548748016,
"logps/rejected": -1.4727751016616821,
"loss": 0.1741,
"nll_loss": 0.20112493634223938,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.009877260774374008,
"rewards/margins": 0.06376149505376816,
"rewards/rejected": -0.07363875210285187,
"step": 560
},
{
"epoch": 2.41,
"grad_norm": 8.85810509927659,
"learning_rate": 2.0942695414584777e-06,
"log_odds_chosen": 3.2707130908966064,
"log_odds_ratio": -0.05988941714167595,
"logits/chosen": -2.8249504566192627,
"logits/rejected": -2.835737705230713,
"logps/chosen": -0.1614939272403717,
"logps/rejected": -1.4856724739074707,
"loss": 0.188,
"nll_loss": 0.17816588282585144,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008074694313108921,
"rewards/margins": 0.06620893627405167,
"rewards/rejected": -0.07428362965583801,
"step": 570
},
{
"epoch": 2.45,
"grad_norm": 8.66523647376059,
"learning_rate": 2.0761369963434992e-06,
"log_odds_chosen": 2.9631845951080322,
"log_odds_ratio": -0.08499319851398468,
"logits/chosen": -2.785752058029175,
"logits/rejected": -2.796713352203369,
"logps/chosen": -0.18658441305160522,
"logps/rejected": -1.3940823078155518,
"loss": 0.1968,
"nll_loss": 0.1999494731426239,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.009329221211373806,
"rewards/margins": 0.060374896973371506,
"rewards/rejected": -0.06970412284135818,
"step": 580
},
{
"epoch": 2.49,
"grad_norm": 7.663377813440287,
"learning_rate": 2.058467423981546e-06,
"log_odds_chosen": 3.1809768676757812,
"log_odds_ratio": -0.0660170167684555,
"logits/chosen": -2.7790160179138184,
"logits/rejected": -2.7909960746765137,
"logps/chosen": -0.16522589325904846,
"logps/rejected": -1.5188618898391724,
"loss": 0.1828,
"nll_loss": 0.18212279677391052,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008261295035481453,
"rewards/margins": 0.06768180429935455,
"rewards/rejected": -0.07594309747219086,
"step": 590
},
{
"epoch": 2.53,
"grad_norm": 9.43229140775389,
"learning_rate": 2.0412414523193154e-06,
"log_odds_chosen": 3.2196316719055176,
"log_odds_ratio": -0.05672148987650871,
"logits/chosen": -2.8662352561950684,
"logits/rejected": -2.872044086456299,
"logps/chosen": -0.16352586448192596,
"logps/rejected": -1.5207058191299438,
"loss": 0.1882,
"nll_loss": 0.1710083782672882,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008176293224096298,
"rewards/margins": 0.06785898655653,
"rewards/rejected": -0.0760352835059166,
"step": 600
},
{
"epoch": 2.57,
"grad_norm": 9.251985707632262,
"learning_rate": 2.0244408254472904e-06,
"log_odds_chosen": 3.181028366088867,
"log_odds_ratio": -0.06368444114923477,
"logits/chosen": -2.8114702701568604,
"logits/rejected": -2.827327013015747,
"logps/chosen": -0.16818083822727203,
"logps/rejected": -1.5173419713974,
"loss": 0.1928,
"nll_loss": 0.19379279017448425,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008409040980041027,
"rewards/margins": 0.06745805591344833,
"rewards/rejected": -0.07586710155010223,
"step": 610
},
{
"epoch": 2.62,
"grad_norm": 8.85560770094724,
"learning_rate": 2.0080483222562476e-06,
"log_odds_chosen": 3.0775837898254395,
"log_odds_ratio": -0.08850983530282974,
"logits/chosen": -2.8276491165161133,
"logits/rejected": -2.8322935104370117,
"logps/chosen": -0.17984047532081604,
"logps/rejected": -1.4078468084335327,
"loss": 0.1874,
"nll_loss": 0.18593069911003113,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008992023766040802,
"rewards/margins": 0.061400316655635834,
"rewards/rejected": -0.07039234042167664,
"step": 620
},
{
"epoch": 2.66,
"grad_norm": 9.092588797512901,
"learning_rate": 1.9920476822239895e-06,
"log_odds_chosen": 3.452129364013672,
"log_odds_ratio": -0.048345282673835754,
"logits/chosen": -2.828294277191162,
"logits/rejected": -2.8251712322235107,
"logps/chosen": -0.15338510274887085,
"logps/rejected": -1.581225872039795,
"loss": 0.1829,
"nll_loss": 0.18267366290092468,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0076692551374435425,
"rewards/margins": 0.07139204442501068,
"rewards/rejected": -0.07906129956245422,
"step": 630
},
{
"epoch": 2.7,
"grad_norm": 8.85624863375054,
"learning_rate": 1.976423537605237e-06,
"log_odds_chosen": 3.381225109100342,
"log_odds_ratio": -0.0466490238904953,
"logits/chosen": -2.8369762897491455,
"logits/rejected": -2.8300728797912598,
"logps/chosen": -0.13137118518352509,
"logps/rejected": -1.4386910200119019,
"loss": 0.1705,
"nll_loss": 0.1446365863084793,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006568559445440769,
"rewards/margins": 0.0653659924864769,
"rewards/rejected": -0.07193455845117569,
"step": 640
},
{
"epoch": 2.74,
"grad_norm": 9.126901201726705,
"learning_rate": 1.961161351381841e-06,
"log_odds_chosen": 3.2348713874816895,
"log_odds_ratio": -0.059200920164585114,
"logits/chosen": -2.8560492992401123,
"logits/rejected": -2.8655595779418945,
"logps/chosen": -0.15543992817401886,
"logps/rejected": -1.4683492183685303,
"loss": 0.1718,
"nll_loss": 0.17798814177513123,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007771995849907398,
"rewards/margins": 0.06564547121524811,
"rewards/rejected": -0.07341746240854263,
"step": 650
},
{
"epoch": 2.78,
"grad_norm": 9.459348581011165,
"learning_rate": 1.9462473604038077e-06,
"log_odds_chosen": 3.150841236114502,
"log_odds_ratio": -0.08523111045360565,
"logits/chosen": -2.883835792541504,
"logits/rejected": -2.880549669265747,
"logps/chosen": -0.16044911742210388,
"logps/rejected": -1.4550909996032715,
"loss": 0.1827,
"nll_loss": 0.1803114414215088,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008022455498576164,
"rewards/margins": 0.06473208963871002,
"rewards/rejected": -0.07275454699993134,
"step": 660
},
{
"epoch": 2.83,
"grad_norm": 8.641861937738083,
"learning_rate": 1.9316685232156397e-06,
"log_odds_chosen": 3.220848798751831,
"log_odds_ratio": -0.06002454087138176,
"logits/chosen": -2.8484086990356445,
"logits/rejected": -2.8489136695861816,
"logps/chosen": -0.15638573467731476,
"logps/rejected": -1.4806524515151978,
"loss": 0.1855,
"nll_loss": 0.177657350897789,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007819287478923798,
"rewards/margins": 0.06621333211660385,
"rewards/rejected": -0.07403261959552765,
"step": 670
},
{
"epoch": 2.87,
"grad_norm": 9.590977394670045,
"learning_rate": 1.917412472118426e-06,
"log_odds_chosen": 3.190727949142456,
"log_odds_ratio": -0.06804339587688446,
"logits/chosen": -2.8769946098327637,
"logits/rejected": -2.8714823722839355,
"logps/chosen": -0.16566260159015656,
"logps/rejected": -1.4919040203094482,
"loss": 0.1888,
"nll_loss": 0.1725473403930664,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008283129893243313,
"rewards/margins": 0.06631207466125488,
"rewards/rejected": -0.07459519803524017,
"step": 680
},
{
"epoch": 2.91,
"grad_norm": 8.352762820112602,
"learning_rate": 1.9034674690672024e-06,
"log_odds_chosen": 3.2079219818115234,
"log_odds_ratio": -0.08390762656927109,
"logits/chosen": -2.8269476890563965,
"logits/rejected": -2.837005376815796,
"logps/chosen": -0.18829426169395447,
"logps/rejected": -1.5218613147735596,
"loss": 0.1972,
"nll_loss": 0.18950611352920532,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.009414712898433208,
"rewards/margins": 0.06667836010456085,
"rewards/rejected": -0.07609307020902634,
"step": 690
},
{
"epoch": 2.95,
"grad_norm": 7.962144706780938,
"learning_rate": 1.8898223650461362e-06,
"log_odds_chosen": 3.216644763946533,
"log_odds_ratio": -0.060684747993946075,
"logits/chosen": -2.856433868408203,
"logits/rejected": -2.8612186908721924,
"logps/chosen": -0.16995587944984436,
"logps/rejected": -1.478044867515564,
"loss": 0.1852,
"nll_loss": 0.18319742381572723,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008497794158756733,
"rewards/margins": 0.06540445238351822,
"rewards/rejected": -0.07390224188566208,
"step": 700
},
{
"epoch": 3.0,
"grad_norm": 8.826772322191102,
"learning_rate": 1.876466562602004e-06,
"log_odds_chosen": 3.2584176063537598,
"log_odds_ratio": -0.05594608932733536,
"logits/chosen": -2.808353900909424,
"logits/rejected": -2.8018672466278076,
"logps/chosen": -0.1637522280216217,
"logps/rejected": -1.5413471460342407,
"loss": 0.1864,
"nll_loss": 0.16807685792446136,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008187612518668175,
"rewards/margins": 0.06887974590063095,
"rewards/rejected": -0.07706736028194427,
"step": 710
},
{
"epoch": 3.0,
"step": 711,
"total_flos": 0.0,
"train_loss": 0.5470885709871219,
"train_runtime": 7181.4173,
"train_samples_per_second": 3.159,
"train_steps_per_second": 0.099
}
],
"logging_steps": 10,
"max_steps": 711,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"total_flos": 0.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}