gemma-7b-borpo-low-quality-v5 / trainer_state.json
c-alfano's picture
Model save
60f74d3 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.9865871833084947,
"eval_steps": 500,
"global_step": 501,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.029806259314456036,
"grad_norm": 1802.2630615234375,
"learning_rate": 3.75e-07,
"log_odds_chosen": -0.22044436633586884,
"log_odds_ratio": -1.0070970058441162,
"logits/chosen": 204.41079711914062,
"logits/rejected": 203.1068878173828,
"logps/chosen": -14.805323600769043,
"logps/rejected": -14.584878921508789,
"loss": 15.3769,
"nll_loss": 14.526931762695312,
"rewards/accuracies": 0.375,
"rewards/chosen": -7.4026618003845215,
"rewards/margins": -0.11022226512432098,
"rewards/rejected": -7.2924394607543945,
"step": 5
},
{
"epoch": 0.05961251862891207,
"grad_norm": 964.4558715820312,
"learning_rate": 7.5e-07,
"log_odds_chosen": 0.22153806686401367,
"log_odds_ratio": -0.7704828977584839,
"logits/chosen": 229.1735382080078,
"logits/rejected": 234.7494659423828,
"logps/chosen": -10.87135124206543,
"logps/rejected": -11.092044830322266,
"loss": 11.644,
"nll_loss": 10.984331130981445,
"rewards/accuracies": 0.5,
"rewards/chosen": -5.435675621032715,
"rewards/margins": 0.1103477030992508,
"rewards/rejected": -5.546022415161133,
"step": 10
},
{
"epoch": 0.08941877794336811,
"grad_norm": 325.32958984375,
"learning_rate": 1.125e-06,
"log_odds_chosen": 0.0398867093026638,
"log_odds_ratio": -0.7268143892288208,
"logits/chosen": 288.3202209472656,
"logits/rejected": 268.65325927734375,
"logps/chosen": -6.712647914886475,
"logps/rejected": -6.752583980560303,
"loss": 7.2706,
"nll_loss": 6.687516689300537,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -3.3563239574432373,
"rewards/margins": 0.019967829808592796,
"rewards/rejected": -3.3762919902801514,
"step": 15
},
{
"epoch": 0.11922503725782414,
"grad_norm": 153.8790740966797,
"learning_rate": 1.5e-06,
"log_odds_chosen": 0.005181239452213049,
"log_odds_ratio": -0.8698928952217102,
"logits/chosen": 279.0413513183594,
"logits/rejected": 273.7066955566406,
"logps/chosen": -4.006048202514648,
"logps/rejected": -4.012642860412598,
"loss": 4.5269,
"nll_loss": 4.109046459197998,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -2.003024101257324,
"rewards/margins": 0.003297415329143405,
"rewards/rejected": -2.006321430206299,
"step": 20
},
{
"epoch": 0.14903129657228018,
"grad_norm": 110.71721649169922,
"learning_rate": 1.875e-06,
"log_odds_chosen": -0.0372297465801239,
"log_odds_ratio": -0.8627766370773315,
"logits/chosen": 340.18084716796875,
"logits/rejected": 349.11151123046875,
"logps/chosen": -2.547694206237793,
"logps/rejected": -2.4806361198425293,
"loss": 3.1921,
"nll_loss": 2.749494791030884,
"rewards/accuracies": 0.5,
"rewards/chosen": -1.2738471031188965,
"rewards/margins": -0.03352922201156616,
"rewards/rejected": -1.2403180599212646,
"step": 25
},
{
"epoch": 0.17883755588673622,
"grad_norm": 75.32969665527344,
"learning_rate": 2.25e-06,
"log_odds_chosen": 0.018188167363405228,
"log_odds_ratio": -0.8359094858169556,
"logits/chosen": 373.6573791503906,
"logits/rejected": 400.1236572265625,
"logps/chosen": -2.330045223236084,
"logps/rejected": -2.3124964237213135,
"loss": 2.6112,
"nll_loss": 2.3758959770202637,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.165022611618042,
"rewards/margins": -0.008774325251579285,
"rewards/rejected": -1.1562482118606567,
"step": 30
},
{
"epoch": 0.20864381520119224,
"grad_norm": 45.19425964355469,
"learning_rate": 2.625e-06,
"log_odds_chosen": 0.25173941254615784,
"log_odds_ratio": -0.7035487294197083,
"logits/chosen": 391.4384765625,
"logits/rejected": 380.1146545410156,
"logps/chosen": -1.7061666250228882,
"logps/rejected": -1.8836677074432373,
"loss": 2.3484,
"nll_loss": 1.9240200519561768,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.8530833125114441,
"rewards/margins": 0.08875055611133575,
"rewards/rejected": -0.9418338537216187,
"step": 35
},
{
"epoch": 0.23845007451564829,
"grad_norm": 49.53385925292969,
"learning_rate": 3e-06,
"log_odds_chosen": 0.2747120261192322,
"log_odds_ratio": -0.6599891185760498,
"logits/chosen": 381.4653625488281,
"logits/rejected": 379.7186279296875,
"logps/chosen": -1.598753571510315,
"logps/rejected": -1.8254070281982422,
"loss": 2.2488,
"nll_loss": 2.010526418685913,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.7993767857551575,
"rewards/margins": 0.11332666873931885,
"rewards/rejected": -0.9127035140991211,
"step": 40
},
{
"epoch": 0.26825633383010433,
"grad_norm": 68.84819030761719,
"learning_rate": 3.3750000000000003e-06,
"log_odds_chosen": 0.29411372542381287,
"log_odds_ratio": -0.6252076625823975,
"logits/chosen": 395.29290771484375,
"logits/rejected": 404.8909912109375,
"logps/chosen": -1.651572823524475,
"logps/rejected": -1.8901698589324951,
"loss": 2.1795,
"nll_loss": 1.7518342733383179,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.8257864117622375,
"rewards/margins": 0.11929843574762344,
"rewards/rejected": -0.9450849294662476,
"step": 45
},
{
"epoch": 0.29806259314456035,
"grad_norm": 47.519283294677734,
"learning_rate": 3.75e-06,
"log_odds_chosen": 0.5501509308815002,
"log_odds_ratio": -0.5450042486190796,
"logits/chosen": 403.5310363769531,
"logits/rejected": 424.2149963378906,
"logps/chosen": -1.5113065242767334,
"logps/rejected": -1.964956521987915,
"loss": 2.1594,
"nll_loss": 1.8876510858535767,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7556532621383667,
"rewards/margins": 0.22682484984397888,
"rewards/rejected": -0.9824782609939575,
"step": 50
},
{
"epoch": 0.32786885245901637,
"grad_norm": 36.98734664916992,
"learning_rate": 4.125e-06,
"log_odds_chosen": 0.26053518056869507,
"log_odds_ratio": -0.6607739925384521,
"logits/chosen": 388.755859375,
"logits/rejected": 381.29827880859375,
"logps/chosen": -1.498511552810669,
"logps/rejected": -1.7075812816619873,
"loss": 2.1979,
"nll_loss": 1.9259532690048218,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7492557764053345,
"rewards/margins": 0.1045350432395935,
"rewards/rejected": -0.8537906408309937,
"step": 55
},
{
"epoch": 0.35767511177347244,
"grad_norm": 63.64994430541992,
"learning_rate": 4.5e-06,
"log_odds_chosen": 0.23358798027038574,
"log_odds_ratio": -0.6189488172531128,
"logits/chosen": 396.7557678222656,
"logits/rejected": 388.23681640625,
"logps/chosen": -1.4873348474502563,
"logps/rejected": -1.6665433645248413,
"loss": 2.0244,
"nll_loss": 1.826873540878296,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.7436674237251282,
"rewards/margins": 0.08960425853729248,
"rewards/rejected": -0.8332716822624207,
"step": 60
},
{
"epoch": 0.38748137108792846,
"grad_norm": 27.614044189453125,
"learning_rate": 4.875e-06,
"log_odds_chosen": 0.3461737036705017,
"log_odds_ratio": -0.6412385106086731,
"logits/chosen": 397.4202575683594,
"logits/rejected": 397.5688171386719,
"logps/chosen": -1.3429419994354248,
"logps/rejected": -1.6044105291366577,
"loss": 1.945,
"nll_loss": 1.5645616054534912,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6714709997177124,
"rewards/margins": 0.13073426485061646,
"rewards/rejected": -0.8022052645683289,
"step": 65
},
{
"epoch": 0.4172876304023845,
"grad_norm": 86.65094757080078,
"learning_rate": 5.25e-06,
"log_odds_chosen": 0.41045960783958435,
"log_odds_ratio": -0.5924609303474426,
"logits/chosen": 376.8070983886719,
"logits/rejected": 384.87994384765625,
"logps/chosen": -1.5336923599243164,
"logps/rejected": -1.8701730966567993,
"loss": 1.997,
"nll_loss": 1.7713584899902344,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7668461799621582,
"rewards/margins": 0.16824033856391907,
"rewards/rejected": -0.9350865483283997,
"step": 70
},
{
"epoch": 0.44709388971684055,
"grad_norm": 75.32307434082031,
"learning_rate": 5.625e-06,
"log_odds_chosen": 0.5093179941177368,
"log_odds_ratio": -0.5533502101898193,
"logits/chosen": 403.20697021484375,
"logits/rejected": 389.50067138671875,
"logps/chosen": -1.303440809249878,
"logps/rejected": -1.7079664468765259,
"loss": 1.901,
"nll_loss": 1.5844974517822266,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.651720404624939,
"rewards/margins": 0.20226280391216278,
"rewards/rejected": -0.8539832234382629,
"step": 75
},
{
"epoch": 0.47690014903129657,
"grad_norm": 52.309078216552734,
"learning_rate": 6e-06,
"log_odds_chosen": 0.5146449208259583,
"log_odds_ratio": -0.5773543119430542,
"logits/chosen": 392.9078063964844,
"logits/rejected": 413.81451416015625,
"logps/chosen": -1.4215633869171143,
"logps/rejected": -1.8292858600616455,
"loss": 1.9712,
"nll_loss": 1.699523687362671,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.7107816934585571,
"rewards/margins": 0.20386135578155518,
"rewards/rejected": -0.9146429300308228,
"step": 80
},
{
"epoch": 0.5067064083457526,
"grad_norm": 665.8851318359375,
"learning_rate": 6.375e-06,
"log_odds_chosen": 0.4371088147163391,
"log_odds_ratio": -0.5891958475112915,
"logits/chosen": 411.912109375,
"logits/rejected": 395.8421325683594,
"logps/chosen": -1.437474012374878,
"logps/rejected": -1.776432991027832,
"loss": 2.0743,
"nll_loss": 1.8243602514266968,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.718737006187439,
"rewards/margins": 0.16947945952415466,
"rewards/rejected": -0.888216495513916,
"step": 85
},
{
"epoch": 0.5365126676602087,
"grad_norm": 57.34864044189453,
"learning_rate": 6.750000000000001e-06,
"log_odds_chosen": 0.9314719438552856,
"log_odds_ratio": -0.49316421151161194,
"logits/chosen": 410.67156982421875,
"logits/rejected": 423.86431884765625,
"logps/chosen": -1.415283441543579,
"logps/rejected": -2.1992759704589844,
"loss": 1.8288,
"nll_loss": 1.5969065427780151,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7076417207717896,
"rewards/margins": 0.39199626445770264,
"rewards/rejected": -1.0996379852294922,
"step": 90
},
{
"epoch": 0.5663189269746647,
"grad_norm": 65.98050689697266,
"learning_rate": 7.1249999999999995e-06,
"log_odds_chosen": 0.38506197929382324,
"log_odds_ratio": -0.5869079828262329,
"logits/chosen": 378.6953430175781,
"logits/rejected": 390.01422119140625,
"logps/chosen": -1.212702989578247,
"logps/rejected": -1.4467040300369263,
"loss": 1.8818,
"nll_loss": 1.4865398406982422,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6063514947891235,
"rewards/margins": 0.1170005351305008,
"rewards/rejected": -0.7233520150184631,
"step": 95
},
{
"epoch": 0.5961251862891207,
"grad_norm": 42.52804183959961,
"learning_rate": 7.5e-06,
"log_odds_chosen": 0.6413298845291138,
"log_odds_ratio": -0.538486659526825,
"logits/chosen": 412.3399353027344,
"logits/rejected": 447.61865234375,
"logps/chosen": -1.3371002674102783,
"logps/rejected": -1.868131399154663,
"loss": 1.9035,
"nll_loss": 1.6383569240570068,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6685501337051392,
"rewards/margins": 0.2655157446861267,
"rewards/rejected": -0.9340656995773315,
"step": 100
},
{
"epoch": 0.6259314456035767,
"grad_norm": 48.187232971191406,
"learning_rate": 7.319250547113999e-06,
"log_odds_chosen": 0.7124954462051392,
"log_odds_ratio": -0.5506631731987,
"logits/chosen": 403.2848205566406,
"logits/rejected": 415.0179138183594,
"logps/chosen": -1.253138780593872,
"logps/rejected": -1.768273115158081,
"loss": 1.8545,
"nll_loss": 1.6297533512115479,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.626569390296936,
"rewards/margins": 0.25756722688674927,
"rewards/rejected": -0.8841365575790405,
"step": 105
},
{
"epoch": 0.6557377049180327,
"grad_norm": 195.05877685546875,
"learning_rate": 7.150969419341942e-06,
"log_odds_chosen": 0.4548686444759369,
"log_odds_ratio": -0.569240927696228,
"logits/chosen": 389.09381103515625,
"logits/rejected": 386.32513427734375,
"logps/chosen": -1.3020490407943726,
"logps/rejected": -1.6546863317489624,
"loss": 1.8713,
"nll_loss": 1.6556203365325928,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6510245203971863,
"rewards/margins": 0.1763187050819397,
"rewards/rejected": -0.8273431658744812,
"step": 110
},
{
"epoch": 0.6855439642324889,
"grad_norm": 53.028961181640625,
"learning_rate": 6.993786061802354e-06,
"log_odds_chosen": 0.8375183939933777,
"log_odds_ratio": -0.4603613018989563,
"logits/chosen": 408.5062561035156,
"logits/rejected": 414.22332763671875,
"logps/chosen": -1.220641851425171,
"logps/rejected": -1.8461415767669678,
"loss": 1.8867,
"nll_loss": 1.583599328994751,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.6103209257125854,
"rewards/margins": 0.3127499222755432,
"rewards/rejected": -0.9230707883834839,
"step": 115
},
{
"epoch": 0.7153502235469449,
"grad_norm": 49.91960144042969,
"learning_rate": 6.846531968814577e-06,
"log_odds_chosen": 0.6342799663543701,
"log_odds_ratio": -0.5294886827468872,
"logits/chosen": 393.26812744140625,
"logits/rejected": 391.69677734375,
"logps/chosen": -1.1840012073516846,
"logps/rejected": -1.6264234781265259,
"loss": 1.9171,
"nll_loss": 1.6122639179229736,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5920006036758423,
"rewards/margins": 0.22121115028858185,
"rewards/rejected": -0.8132117390632629,
"step": 120
},
{
"epoch": 0.7451564828614009,
"grad_norm": 26.490934371948242,
"learning_rate": 6.708203932499369e-06,
"log_odds_chosen": 0.3495193421840668,
"log_odds_ratio": -0.6338833570480347,
"logits/chosen": 394.9533386230469,
"logits/rejected": 406.28662109375,
"logps/chosen": -1.23225998878479,
"logps/rejected": -1.4777755737304688,
"loss": 1.8239,
"nll_loss": 1.5472067594528198,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.616129994392395,
"rewards/margins": 0.12275775521993637,
"rewards/rejected": -0.7388877868652344,
"step": 125
},
{
"epoch": 0.7749627421758569,
"grad_norm": 45.540184020996094,
"learning_rate": 6.577935144802719e-06,
"log_odds_chosen": 0.4649319052696228,
"log_odds_ratio": -0.5916207432746887,
"logits/chosen": 417.3863830566406,
"logits/rejected": 405.1355895996094,
"logps/chosen": -1.2279858589172363,
"logps/rejected": -1.5856168270111084,
"loss": 1.8861,
"nll_loss": 1.565514326095581,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.6139929294586182,
"rewards/margins": 0.17881537973880768,
"rewards/rejected": -0.7928084135055542,
"step": 130
},
{
"epoch": 0.8047690014903129,
"grad_norm": 42.16865921020508,
"learning_rate": 6.454972243679028e-06,
"log_odds_chosen": 0.46650758385658264,
"log_odds_ratio": -0.6158651113510132,
"logits/chosen": 421.04730224609375,
"logits/rejected": 423.00775146484375,
"logps/chosen": -1.509341835975647,
"logps/rejected": -1.8879953622817993,
"loss": 1.9281,
"nll_loss": 1.723570466041565,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.7546709179878235,
"rewards/margins": 0.18932664394378662,
"rewards/rejected": -0.9439976811408997,
"step": 135
},
{
"epoch": 0.834575260804769,
"grad_norm": 41.30878448486328,
"learning_rate": 6.338656910463875e-06,
"log_odds_chosen": 0.27211299538612366,
"log_odds_ratio": -0.680214524269104,
"logits/chosen": 404.1061096191406,
"logits/rejected": 409.54559326171875,
"logps/chosen": -1.4520971775054932,
"logps/rejected": -1.670898199081421,
"loss": 1.9311,
"nll_loss": 1.7876545190811157,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.7260485887527466,
"rewards/margins": 0.10940053313970566,
"rewards/rejected": -0.8354490995407104,
"step": 140
},
{
"epoch": 0.8643815201192251,
"grad_norm": 58.79216003417969,
"learning_rate": 6.228410989030498e-06,
"log_odds_chosen": 0.13751578330993652,
"log_odds_ratio": -0.6719902753829956,
"logits/chosen": 402.82342529296875,
"logits/rejected": 405.1291809082031,
"logps/chosen": -1.3680763244628906,
"logps/rejected": -1.4593302011489868,
"loss": 1.9064,
"nll_loss": 1.6308927536010742,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6840381622314453,
"rewards/margins": 0.045626964420080185,
"rewards/rejected": -0.7296651005744934,
"step": 145
},
{
"epoch": 0.8941877794336811,
"grad_norm": 28.582611083984375,
"learning_rate": 6.123724356957946e-06,
"log_odds_chosen": 0.8705786466598511,
"log_odds_ratio": -0.5025649666786194,
"logits/chosen": 407.1871643066406,
"logits/rejected": 423.8992614746094,
"logps/chosen": -1.2447984218597412,
"logps/rejected": -1.9607397317886353,
"loss": 1.8397,
"nll_loss": 1.4880900382995605,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6223992109298706,
"rewards/margins": 0.357970654964447,
"rewards/rejected": -0.9803698658943176,
"step": 150
},
{
"epoch": 0.9239940387481371,
"grad_norm": 56.87172317504883,
"learning_rate": 6.024144966768742e-06,
"log_odds_chosen": 0.32878491282463074,
"log_odds_ratio": -0.6313835382461548,
"logits/chosen": 388.4563293457031,
"logits/rejected": 402.23834228515625,
"logps/chosen": -1.365476131439209,
"logps/rejected": -1.6416184902191162,
"loss": 1.8579,
"nll_loss": 1.458609938621521,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6827380657196045,
"rewards/margins": 0.13807126879692078,
"rewards/rejected": -0.8208092451095581,
"step": 155
},
{
"epoch": 0.9538002980625931,
"grad_norm": 52.100013732910156,
"learning_rate": 5.929270612815711e-06,
"log_odds_chosen": 0.2496444433927536,
"log_odds_ratio": -0.6904889345169067,
"logits/chosen": 397.15887451171875,
"logits/rejected": 440.45587158203125,
"logps/chosen": -1.2235276699066162,
"logps/rejected": -1.4282524585723877,
"loss": 1.8144,
"nll_loss": 1.502598762512207,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.6117638349533081,
"rewards/margins": 0.10236239433288574,
"rewards/rejected": -0.7141262292861938,
"step": 160
},
{
"epoch": 0.9836065573770492,
"grad_norm": 62.34568786621094,
"learning_rate": 5.838742081211422e-06,
"log_odds_chosen": 0.25045713782310486,
"log_odds_ratio": -0.6064125895500183,
"logits/chosen": 407.72943115234375,
"logits/rejected": 420.569580078125,
"logps/chosen": -1.2549211978912354,
"logps/rejected": -1.4284086227416992,
"loss": 1.8398,
"nll_loss": 1.5022889375686646,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6274605989456177,
"rewards/margins": 0.0867437794804573,
"rewards/rejected": -0.7142043113708496,
"step": 165
},
{
"epoch": 0.9955290611028316,
"eval_log_odds_chosen": 0.25424009561538696,
"eval_log_odds_ratio": -0.6749479174613953,
"eval_logits/chosen": 331.1551513671875,
"eval_logits/rejected": 302.9094543457031,
"eval_logps/chosen": -1.1524564027786255,
"eval_logps/rejected": -1.343080997467041,
"eval_loss": 1.8982083797454834,
"eval_nll_loss": 1.5454045534133911,
"eval_rewards/accuracies": 0.5179855823516846,
"eval_rewards/chosen": -0.5762282013893127,
"eval_rewards/margins": 0.09531236439943314,
"eval_rewards/rejected": -0.6715404987335205,
"eval_runtime": 378.1633,
"eval_samples_per_second": 1.462,
"eval_steps_per_second": 0.368,
"step": 167
},
{
"epoch": 1.0134128166915053,
"grad_norm": 19.05902862548828,
"learning_rate": 5.752237416355279e-06,
"log_odds_chosen": 0.8472374081611633,
"log_odds_ratio": -0.5489901304244995,
"logits/chosen": 389.7673645019531,
"logits/rejected": 412.4244079589844,
"logps/chosen": -1.1339842081069946,
"logps/rejected": -1.749389410018921,
"loss": 1.6458,
"nll_loss": 1.2897634506225586,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5669921040534973,
"rewards/margins": 0.3077026307582855,
"rewards/rejected": -0.8746947050094604,
"step": 170
},
{
"epoch": 1.0432190760059612,
"grad_norm": 39.458702087402344,
"learning_rate": 5.669467095138408e-06,
"log_odds_chosen": 1.5121970176696777,
"log_odds_ratio": -0.30622434616088867,
"logits/chosen": 365.8326416015625,
"logits/rejected": 406.28411865234375,
"logps/chosen": -0.7569233775138855,
"logps/rejected": -1.806539535522461,
"loss": 1.2547,
"nll_loss": 1.183380365371704,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.37846168875694275,
"rewards/margins": 0.5248081088066101,
"rewards/rejected": -0.9032697677612305,
"step": 175
},
{
"epoch": 1.0730253353204173,
"grad_norm": 31.42799949645996,
"learning_rate": 5.5901699437494744e-06,
"log_odds_chosen": 1.5321556329727173,
"log_odds_ratio": -0.2848857045173645,
"logits/chosen": 364.0169677734375,
"logits/rejected": 339.5393371582031,
"logps/chosen": -0.8578670620918274,
"logps/rejected": -1.9486223459243774,
"loss": 1.3431,
"nll_loss": 1.3115805387496948,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.4289335310459137,
"rewards/margins": 0.5453776717185974,
"rewards/rejected": -0.9743111729621887,
"step": 180
},
{
"epoch": 1.1028315946348732,
"grad_norm": 35.6085090637207,
"learning_rate": 5.514109665703559e-06,
"log_odds_chosen": 2.223374843597412,
"log_odds_ratio": -0.2399306744337082,
"logits/chosen": 396.5127868652344,
"logits/rejected": 386.45831298828125,
"logps/chosen": -0.8754903078079224,
"logps/rejected": -2.4870636463165283,
"loss": 1.2172,
"nll_loss": 1.070143222808838,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.4377451539039612,
"rewards/margins": 0.8057867288589478,
"rewards/rejected": -1.2435318231582642,
"step": 185
},
{
"epoch": 1.1326378539493294,
"grad_norm": 22.422922134399414,
"learning_rate": 5.441071875825088e-06,
"log_odds_chosen": 1.2998098134994507,
"log_odds_ratio": -0.32330313324928284,
"logits/chosen": 364.8489074707031,
"logits/rejected": 408.52972412109375,
"logps/chosen": -0.870254397392273,
"logps/rejected": -1.7498779296875,
"loss": 1.2807,
"nll_loss": 1.1046603918075562,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4351271986961365,
"rewards/margins": 0.4398117661476135,
"rewards/rejected": -0.87493896484375,
"step": 190
},
{
"epoch": 1.1624441132637853,
"grad_norm": 25.98566246032715,
"learning_rate": 5.370861555295747e-06,
"log_odds_chosen": 2.002765417098999,
"log_odds_ratio": -0.24665918946266174,
"logits/chosen": 384.2296142578125,
"logits/rejected": 398.67205810546875,
"logps/chosen": -0.8954353332519531,
"logps/rejected": -2.385274648666382,
"loss": 1.2781,
"nll_loss": 1.1818829774856567,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.44771766662597656,
"rewards/margins": 0.7449195981025696,
"rewards/rejected": -1.192637324333191,
"step": 195
},
{
"epoch": 1.1922503725782414,
"grad_norm": 23.273193359375,
"learning_rate": 5.303300858899106e-06,
"log_odds_chosen": 2.1724612712860107,
"log_odds_ratio": -0.26438647508621216,
"logits/chosen": 413.26361083984375,
"logits/rejected": 396.8680419921875,
"logps/chosen": -0.730914831161499,
"logps/rejected": -2.3943963050842285,
"loss": 1.2803,
"nll_loss": 1.0805118083953857,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3654574155807495,
"rewards/margins": 0.83174067735672,
"rewards/rejected": -1.1971981525421143,
"step": 200
},
{
"epoch": 1.2220566318926975,
"grad_norm": 27.806596755981445,
"learning_rate": 5.238227218271837e-06,
"log_odds_chosen": 2.4484591484069824,
"log_odds_ratio": -0.22746221721172333,
"logits/chosen": 365.72119140625,
"logits/rejected": 359.6267395019531,
"logps/chosen": -0.758979320526123,
"logps/rejected": -2.573237180709839,
"loss": 1.2398,
"nll_loss": 1.1106595993041992,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.3794896602630615,
"rewards/margins": 0.9071289896965027,
"rewards/rejected": -1.2866185903549194,
"step": 205
},
{
"epoch": 1.2518628912071534,
"grad_norm": 22.43109130859375,
"learning_rate": 5.175491695067656e-06,
"log_odds_chosen": 2.793255567550659,
"log_odds_ratio": -0.23325839638710022,
"logits/chosen": 377.73443603515625,
"logits/rejected": 404.66644287109375,
"logps/chosen": -0.7964418530464172,
"logps/rejected": -3.0091164112091064,
"loss": 1.2371,
"nll_loss": 1.0737245082855225,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3982209265232086,
"rewards/margins": 1.106337308883667,
"rewards/rejected": -1.5045582056045532,
"step": 210
},
{
"epoch": 1.2816691505216096,
"grad_norm": 49.615081787109375,
"learning_rate": 5.114957546028551e-06,
"log_odds_chosen": 2.5913193225860596,
"log_odds_ratio": -0.23048730194568634,
"logits/chosen": 401.5857238769531,
"logits/rejected": 382.7472229003906,
"logps/chosen": -0.7830225229263306,
"logps/rejected": -2.782843828201294,
"loss": 1.2555,
"nll_loss": 1.1490776538848877,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3915112614631653,
"rewards/margins": 0.9999106526374817,
"rewards/rejected": -1.391421914100647,
"step": 215
},
{
"epoch": 1.3114754098360657,
"grad_norm": 37.056453704833984,
"learning_rate": 5.056498968474315e-06,
"log_odds_chosen": 1.4505621194839478,
"log_odds_ratio": -0.3243867754936218,
"logits/chosen": 393.34063720703125,
"logits/rejected": 387.152099609375,
"logps/chosen": -0.8490570783615112,
"logps/rejected": -1.8110103607177734,
"loss": 1.2897,
"nll_loss": 1.1027252674102783,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.4245285391807556,
"rewards/margins": 0.48097652196884155,
"rewards/rejected": -0.9055051803588867,
"step": 220
},
{
"epoch": 1.3412816691505216,
"grad_norm": 22.3747615814209,
"learning_rate": 4.9999999999999996e-06,
"log_odds_chosen": 1.0405768156051636,
"log_odds_ratio": -0.37774857878685,
"logits/chosen": 391.2264099121094,
"logits/rejected": 382.2864685058594,
"logps/chosen": -1.0447379350662231,
"logps/rejected": -1.8071935176849365,
"loss": 1.2781,
"nll_loss": 1.2478437423706055,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.5223689675331116,
"rewards/margins": 0.38122788071632385,
"rewards/rejected": -0.9035967588424683,
"step": 225
},
{
"epoch": 1.3710879284649775,
"grad_norm": 22.041149139404297,
"learning_rate": 4.945353550468404e-06,
"log_odds_chosen": 2.0077407360076904,
"log_odds_ratio": -0.25288665294647217,
"logits/chosen": 370.07696533203125,
"logits/rejected": 385.3541564941406,
"logps/chosen": -0.90089350938797,
"logps/rejected": -2.3661916255950928,
"loss": 1.2539,
"nll_loss": 0.989953339099884,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.450446754693985,
"rewards/margins": 0.7326492667198181,
"rewards/rejected": -1.1830958127975464,
"step": 230
},
{
"epoch": 1.4008941877794336,
"grad_norm": 33.446224212646484,
"learning_rate": 4.8924605479008165e-06,
"log_odds_chosen": 1.8689000606536865,
"log_odds_ratio": -0.24617977440357208,
"logits/chosen": 391.69732666015625,
"logits/rejected": 402.88836669921875,
"logps/chosen": -0.8436087369918823,
"logps/rejected": -2.213329553604126,
"loss": 1.2797,
"nll_loss": 1.1102927923202515,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.42180436849594116,
"rewards/margins": 0.684860348701477,
"rewards/rejected": -1.106664776802063,
"step": 235
},
{
"epoch": 1.4307004470938898,
"grad_norm": 19.861835479736328,
"learning_rate": 4.8412291827592715e-06,
"log_odds_chosen": 1.848902940750122,
"log_odds_ratio": -0.3047091066837311,
"logits/chosen": 415.73486328125,
"logits/rejected": 420.25909423828125,
"logps/chosen": -0.8824371099472046,
"logps/rejected": -2.3095035552978516,
"loss": 1.2891,
"nll_loss": 1.1334781646728516,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.4412185549736023,
"rewards/margins": 0.7135332226753235,
"rewards/rejected": -1.1547517776489258,
"step": 240
},
{
"epoch": 1.4605067064083457,
"grad_norm": 31.12167739868164,
"learning_rate": 4.79157423749955e-06,
"log_odds_chosen": 1.9949026107788086,
"log_odds_ratio": -0.24574021995067596,
"logits/chosen": 409.1098937988281,
"logits/rejected": 394.39593505859375,
"logps/chosen": -0.7639743089675903,
"logps/rejected": -2.1869406700134277,
"loss": 1.312,
"nll_loss": 1.2122286558151245,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.38198715448379517,
"rewards/margins": 0.7114832997322083,
"rewards/rejected": -1.0934703350067139,
"step": 245
},
{
"epoch": 1.4903129657228018,
"grad_norm": 20.75017547607422,
"learning_rate": 4.74341649025257e-06,
"log_odds_chosen": 2.6609864234924316,
"log_odds_ratio": -0.25564372539520264,
"logits/chosen": 380.4820251464844,
"logits/rejected": 385.7266845703125,
"logps/chosen": -0.804089367389679,
"logps/rejected": -2.9035263061523438,
"loss": 1.2332,
"nll_loss": 1.0138413906097412,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4020446836948395,
"rewards/margins": 1.0497183799743652,
"rewards/rejected": -1.4517631530761719,
"step": 250
},
{
"epoch": 1.520119225037258,
"grad_norm": 26.518312454223633,
"learning_rate": 4.696682183138622e-06,
"log_odds_chosen": 2.0447866916656494,
"log_odds_ratio": -0.22285214066505432,
"logits/chosen": 399.64971923828125,
"logits/rejected": 400.3153076171875,
"logps/chosen": -0.8079016804695129,
"logps/rejected": -2.2692599296569824,
"loss": 1.2551,
"nll_loss": 1.0453466176986694,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.40395084023475647,
"rewards/margins": 0.7306792140007019,
"rewards/rejected": -1.1346299648284912,
"step": 255
},
{
"epoch": 1.5499254843517138,
"grad_norm": 18.25040626525879,
"learning_rate": 4.6513025470953165e-06,
"log_odds_chosen": 1.7279441356658936,
"log_odds_ratio": -0.3020802140235901,
"logits/chosen": 380.86590576171875,
"logits/rejected": 420.64007568359375,
"logps/chosen": -0.8852326273918152,
"logps/rejected": -2.209996461868286,
"loss": 1.2132,
"nll_loss": 1.1014916896820068,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.4426163136959076,
"rewards/margins": 0.6623818278312683,
"rewards/rejected": -1.104998230934143,
"step": 260
},
{
"epoch": 1.5797317436661698,
"grad_norm": 17.494176864624023,
"learning_rate": 4.607213376254634e-06,
"log_odds_chosen": 1.6819546222686768,
"log_odds_ratio": -0.3340403437614441,
"logits/chosen": 388.51068115234375,
"logits/rejected": 409.89971923828125,
"logps/chosen": -0.8424839973449707,
"logps/rejected": -2.041083574295044,
"loss": 1.3572,
"nll_loss": 1.1232020854949951,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.42124199867248535,
"rewards/margins": 0.5992996692657471,
"rewards/rejected": -1.020541787147522,
"step": 265
},
{
"epoch": 1.6095380029806259,
"grad_norm": 34.47447967529297,
"learning_rate": 4.564354645876384e-06,
"log_odds_chosen": 1.859960913658142,
"log_odds_ratio": -0.2928032875061035,
"logits/chosen": 372.35052490234375,
"logits/rejected": 382.39117431640625,
"logps/chosen": -0.9743008613586426,
"logps/rejected": -2.432216167449951,
"loss": 1.3166,
"nll_loss": 1.270026683807373,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.4871504306793213,
"rewards/margins": 0.7289576530456543,
"rewards/rejected": -1.2161080837249756,
"step": 270
},
{
"epoch": 1.639344262295082,
"grad_norm": 21.639741897583008,
"learning_rate": 4.522670168666455e-06,
"log_odds_chosen": 1.463616967201233,
"log_odds_ratio": -0.3027709126472473,
"logits/chosen": 362.27044677734375,
"logits/rejected": 350.851318359375,
"logps/chosen": -0.7592640519142151,
"logps/rejected": -1.6749483346939087,
"loss": 1.2876,
"nll_loss": 1.0598416328430176,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.37963202595710754,
"rewards/margins": 0.45784205198287964,
"rewards/rejected": -0.8374741673469543,
"step": 275
},
{
"epoch": 1.669150521609538,
"grad_norm": 32.02434158325195,
"learning_rate": 4.482107285003976e-06,
"log_odds_chosen": 1.598225474357605,
"log_odds_ratio": -0.3603498339653015,
"logits/chosen": 410.98687744140625,
"logits/rejected": 397.2725830078125,
"logps/chosen": -0.8709872364997864,
"logps/rejected": -1.9590165615081787,
"loss": 1.285,
"nll_loss": 1.1194555759429932,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.4354936182498932,
"rewards/margins": 0.5440146923065186,
"rewards/rejected": -0.9795082807540894,
"step": 280
},
{
"epoch": 1.698956780923994,
"grad_norm": 18.135896682739258,
"learning_rate": 4.442616583193192e-06,
"log_odds_chosen": 2.131542921066284,
"log_odds_ratio": -0.2719998061656952,
"logits/chosen": 375.759033203125,
"logits/rejected": 377.7268981933594,
"logps/chosen": -0.7118241786956787,
"logps/rejected": -2.2590417861938477,
"loss": 1.2113,
"nll_loss": 1.054491400718689,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.35591208934783936,
"rewards/margins": 0.7736088037490845,
"rewards/rejected": -1.1295208930969238,
"step": 285
},
{
"epoch": 1.7287630402384502,
"grad_norm": 16.36552619934082,
"learning_rate": 4.404151646360276e-06,
"log_odds_chosen": 2.539419651031494,
"log_odds_ratio": -0.2885150611400604,
"logits/chosen": 395.09832763671875,
"logits/rejected": 429.5597229003906,
"logps/chosen": -0.8963809013366699,
"logps/rejected": -2.95475697517395,
"loss": 1.2765,
"nll_loss": 1.1113842725753784,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.44819045066833496,
"rewards/margins": 1.0291881561279297,
"rewards/rejected": -1.477378487586975,
"step": 290
},
{
"epoch": 1.758569299552906,
"grad_norm": 20.132436752319336,
"learning_rate": 4.366668823046865e-06,
"log_odds_chosen": 1.553693413734436,
"log_odds_ratio": -0.3133796155452728,
"logits/chosen": 371.56475830078125,
"logits/rejected": 413.505615234375,
"logps/chosen": -0.9781205058097839,
"logps/rejected": -2.0690226554870605,
"loss": 1.2963,
"nll_loss": 1.1234586238861084,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.48906025290489197,
"rewards/margins": 0.5454510450363159,
"rewards/rejected": -1.0345113277435303,
"step": 295
},
{
"epoch": 1.788375558867362,
"grad_norm": 26.103137969970703,
"learning_rate": 4.330127018922194e-06,
"log_odds_chosen": 2.1462044715881348,
"log_odds_ratio": -0.24198386073112488,
"logits/chosen": 409.97161865234375,
"logits/rejected": 383.2239990234375,
"logps/chosen": -0.7521153688430786,
"logps/rejected": -2.3262906074523926,
"loss": 1.1963,
"nll_loss": 1.022304654121399,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.3760576844215393,
"rewards/margins": 0.7870877385139465,
"rewards/rejected": -1.1631453037261963,
"step": 300
},
{
"epoch": 1.8181818181818183,
"grad_norm": 20.291587829589844,
"learning_rate": 4.294487507354011e-06,
"log_odds_chosen": 1.8153495788574219,
"log_odds_ratio": -0.3603833019733429,
"logits/chosen": 399.9996032714844,
"logits/rejected": 405.7534484863281,
"logps/chosen": -1.029065728187561,
"logps/rejected": -2.513669490814209,
"loss": 1.2618,
"nll_loss": 1.2645294666290283,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.5145328640937805,
"rewards/margins": 0.742301881313324,
"rewards/rejected": -1.2568347454071045,
"step": 305
},
{
"epoch": 1.8479880774962743,
"grad_norm": 19.601377487182617,
"learning_rate": 4.259713756852986e-06,
"log_odds_chosen": 3.238518238067627,
"log_odds_ratio": -0.22569267451763153,
"logits/chosen": 380.0532531738281,
"logits/rejected": 386.49920654296875,
"logps/chosen": -0.8288024067878723,
"logps/rejected": -3.505189895629883,
"loss": 1.3118,
"nll_loss": 1.1829016208648682,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.41440120339393616,
"rewards/margins": 1.338193655014038,
"rewards/rejected": -1.7525949478149414,
"step": 310
},
{
"epoch": 1.8777943368107302,
"grad_norm": 19.882904052734375,
"learning_rate": 4.225771273642583e-06,
"log_odds_chosen": 1.2986266613006592,
"log_odds_ratio": -0.3742119073867798,
"logits/chosen": 354.4183044433594,
"logits/rejected": 373.57757568359375,
"logps/chosen": -1.0900706052780151,
"logps/rejected": -2.065725326538086,
"loss": 1.3407,
"nll_loss": 1.3649649620056152,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.5450353026390076,
"rewards/margins": 0.4878273606300354,
"rewards/rejected": -1.032862663269043,
"step": 315
},
{
"epoch": 1.9076005961251863,
"grad_norm": 21.79350471496582,
"learning_rate": 4.192627457812106e-06,
"log_odds_chosen": 2.3978288173675537,
"log_odds_ratio": -0.24583116173744202,
"logits/chosen": 379.4220886230469,
"logits/rejected": 400.0596618652344,
"logps/chosen": -0.7697328329086304,
"logps/rejected": -2.6053688526153564,
"loss": 1.1966,
"nll_loss": 1.0758590698242188,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3848664164543152,
"rewards/margins": 0.9178179502487183,
"rewards/rejected": -1.3026844263076782,
"step": 320
},
{
"epoch": 1.9374068554396424,
"grad_norm": 25.364540100097656,
"learning_rate": 4.1602514716892185e-06,
"log_odds_chosen": 2.511842727661133,
"log_odds_ratio": -0.21811020374298096,
"logits/chosen": 377.62139892578125,
"logits/rejected": 409.9249572753906,
"logps/chosen": -0.8184257745742798,
"logps/rejected": -2.717911720275879,
"loss": 1.2025,
"nll_loss": 1.0009828805923462,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.4092128872871399,
"rewards/margins": 0.9497429728507996,
"rewards/rejected": -1.3589558601379395,
"step": 325
},
{
"epoch": 1.9672131147540983,
"grad_norm": 24.96152114868164,
"learning_rate": 4.128614119223853e-06,
"log_odds_chosen": 1.425945520401001,
"log_odds_ratio": -0.2635498642921448,
"logits/chosen": 370.4853515625,
"logits/rejected": 376.5760192871094,
"logps/chosen": -0.8058730959892273,
"logps/rejected": -1.770816445350647,
"loss": 1.257,
"nll_loss": 0.9558019638061523,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.40293654799461365,
"rewards/margins": 0.48247161507606506,
"rewards/rejected": -0.8854082226753235,
"step": 330
},
{
"epoch": 1.9970193740685542,
"grad_norm": 17.690673828125,
"learning_rate": 4.097687735311148e-06,
"log_odds_chosen": 1.0468049049377441,
"log_odds_ratio": -0.43120455741882324,
"logits/chosen": 390.3360900878906,
"logits/rejected": 377.1321105957031,
"logps/chosen": -1.1694064140319824,
"logps/rejected": -1.9213879108428955,
"loss": 1.299,
"nll_loss": 1.3618862628936768,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.5847032070159912,
"rewards/margins": 0.3759908080101013,
"rewards/rejected": -0.9606939554214478,
"step": 335
},
{
"epoch": 1.9970193740685542,
"eval_log_odds_chosen": 0.30739960074424744,
"eval_log_odds_ratio": -0.6665536761283875,
"eval_logits/chosen": 313.4974365234375,
"eval_logits/rejected": 282.3182678222656,
"eval_logps/chosen": -1.1013935804367065,
"eval_logps/rejected": -1.3020689487457275,
"eval_loss": 1.8186225891113281,
"eval_nll_loss": 1.4682071208953857,
"eval_rewards/accuracies": 0.5395683646202087,
"eval_rewards/chosen": -0.5506967902183533,
"eval_rewards/margins": 0.10033769905567169,
"eval_rewards/rejected": -0.6510344743728638,
"eval_runtime": 378.4505,
"eval_samples_per_second": 1.461,
"eval_steps_per_second": 0.367,
"step": 335
},
{
"epoch": 2.0268256333830106,
"grad_norm": 19.424360275268555,
"learning_rate": 4.0674460840998035e-06,
"log_odds_chosen": 3.000514507293701,
"log_odds_ratio": -0.10731367021799088,
"logits/chosen": 394.8528747558594,
"logits/rejected": 368.841796875,
"logps/chosen": -0.4828576147556305,
"logps/rejected": -2.412517547607422,
"loss": 0.7587,
"nll_loss": 0.8271492123603821,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24142880737781525,
"rewards/margins": 0.9648300409317017,
"rewards/rejected": -1.206258773803711,
"step": 340
},
{
"epoch": 2.0566318926974665,
"grad_norm": 13.576729774475098,
"learning_rate": 4.037864265436242e-06,
"log_odds_chosen": 3.940922498703003,
"log_odds_ratio": -0.07164354622364044,
"logits/chosen": 342.024658203125,
"logits/rejected": 342.9552307128906,
"logps/chosen": -0.48198527097702026,
"logps/rejected": -3.1786067485809326,
"loss": 0.6501,
"nll_loss": 0.6376822590827942,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.24099263548851013,
"rewards/margins": 1.3483107089996338,
"rewards/rejected": -1.5893033742904663,
"step": 345
},
{
"epoch": 2.0864381520119224,
"grad_norm": 13.920299530029297,
"learning_rate": 4.008918628686366e-06,
"log_odds_chosen": 3.9218780994415283,
"log_odds_ratio": -0.04757709056138992,
"logits/chosen": 334.36187744140625,
"logits/rejected": 374.3761291503906,
"logps/chosen": -0.4404733180999756,
"logps/rejected": -3.2984976768493652,
"loss": 0.6334,
"nll_loss": 0.6365537643432617,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2202366590499878,
"rewards/margins": 1.4290122985839844,
"rewards/rejected": -1.6492488384246826,
"step": 350
},
{
"epoch": 2.1162444113263787,
"grad_norm": 12.827689170837402,
"learning_rate": 3.980586693257065e-06,
"log_odds_chosen": 4.072226047515869,
"log_odds_ratio": -0.05848418548703194,
"logits/chosen": 343.1537170410156,
"logits/rejected": 358.5604248046875,
"logps/chosen": -0.41472020745277405,
"logps/rejected": -3.3263726234436035,
"loss": 0.6337,
"nll_loss": 0.6818060278892517,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.20736010372638702,
"rewards/margins": 1.4558260440826416,
"rewards/rejected": -1.6631863117218018,
"step": 355
},
{
"epoch": 2.1460506706408347,
"grad_norm": 14.782392501831055,
"learning_rate": 3.952847075210474e-06,
"log_odds_chosen": 4.195507049560547,
"log_odds_ratio": -0.03957627713680267,
"logits/chosen": 346.2444152832031,
"logits/rejected": 393.8316955566406,
"logps/chosen": -0.4012891352176666,
"logps/rejected": -3.2268481254577637,
"loss": 0.62,
"nll_loss": 0.5778239965438843,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2006445676088333,
"rewards/margins": 1.4127795696258545,
"rewards/rejected": -1.6134240627288818,
"step": 360
},
{
"epoch": 2.1758569299552906,
"grad_norm": 13.0780029296875,
"learning_rate": 3.925679419426603e-06,
"log_odds_chosen": 3.8077125549316406,
"log_odds_ratio": -0.06930537521839142,
"logits/chosen": 345.62933349609375,
"logits/rejected": 324.48724365234375,
"logps/chosen": -0.416290283203125,
"logps/rejected": -3.0446524620056152,
"loss": 0.6211,
"nll_loss": 0.5697866678237915,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.2081451416015625,
"rewards/margins": 1.3141810894012451,
"rewards/rejected": -1.5223262310028076,
"step": 365
},
{
"epoch": 2.2056631892697465,
"grad_norm": 13.796915054321289,
"learning_rate": 3.8990643368252725e-06,
"log_odds_chosen": 3.758181095123291,
"log_odds_ratio": -0.08863835036754608,
"logits/chosen": 309.7760009765625,
"logits/rejected": 327.28875732421875,
"logps/chosen": -0.4810813367366791,
"logps/rejected": -3.211531400680542,
"loss": 0.6805,
"nll_loss": 0.6794728636741638,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24054066836833954,
"rewards/margins": 1.365225076675415,
"rewards/rejected": -1.605765700340271,
"step": 370
},
{
"epoch": 2.235469448584203,
"grad_norm": 12.518247604370117,
"learning_rate": 3.872983346207417e-06,
"log_odds_chosen": 3.9930922985076904,
"log_odds_ratio": -0.07049717009067535,
"logits/chosen": 364.5960388183594,
"logits/rejected": 381.1838684082031,
"logps/chosen": -0.47075748443603516,
"logps/rejected": -3.3296546936035156,
"loss": 0.6191,
"nll_loss": 0.6227518320083618,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.23537874221801758,
"rewards/margins": 1.4294484853744507,
"rewards/rejected": -1.6648273468017578,
"step": 375
},
{
"epoch": 2.2652757078986587,
"grad_norm": 23.583436965942383,
"learning_rate": 3.847418820319329e-06,
"log_odds_chosen": 3.55949330329895,
"log_odds_ratio": -0.048723362386226654,
"logits/chosen": 364.0585632324219,
"logits/rejected": 337.66424560546875,
"logps/chosen": -0.41810473799705505,
"logps/rejected": -2.7338831424713135,
"loss": 0.6367,
"nll_loss": 0.6199051737785339,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.20905236899852753,
"rewards/margins": 1.1578892469406128,
"rewards/rejected": -1.3669415712356567,
"step": 380
},
{
"epoch": 2.2950819672131146,
"grad_norm": 12.274104118347168,
"learning_rate": 3.822353935782191e-06,
"log_odds_chosen": 4.226048946380615,
"log_odds_ratio": -0.04617181420326233,
"logits/chosen": 359.08154296875,
"logits/rejected": 344.7435607910156,
"logps/chosen": -0.36728304624557495,
"logps/rejected": -3.1797947883605957,
"loss": 0.5982,
"nll_loss": 0.536335289478302,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.18364152312278748,
"rewards/margins": 1.4062559604644775,
"rewards/rejected": -1.5898973941802979,
"step": 385
},
{
"epoch": 2.3248882265275705,
"grad_norm": 15.543499946594238,
"learning_rate": 3.79777262656375e-06,
"log_odds_chosen": 3.8094513416290283,
"log_odds_ratio": -0.03931097313761711,
"logits/chosen": 373.632568359375,
"logits/rejected": 393.21295166015625,
"logps/chosen": -0.46378618478775024,
"logps/rejected": -3.2327797412872314,
"loss": 0.667,
"nll_loss": 0.6460134387016296,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.23189309239387512,
"rewards/margins": 1.3844965696334839,
"rewards/rejected": -1.6163898706436157,
"step": 390
},
{
"epoch": 2.354694485842027,
"grad_norm": 13.162529945373535,
"learning_rate": 3.773659540699707e-06,
"log_odds_chosen": 3.9874942302703857,
"log_odds_ratio": -0.05443572998046875,
"logits/chosen": 368.2042236328125,
"logits/rejected": 346.73590087890625,
"logps/chosen": -0.3340614438056946,
"logps/rejected": -2.9790287017822266,
"loss": 0.5966,
"nll_loss": 0.5552054643630981,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1670307219028473,
"rewards/margins": 1.3224836587905884,
"rewards/rejected": -1.4895143508911133,
"step": 395
},
{
"epoch": 2.384500745156483,
"grad_norm": 11.184009552001953,
"learning_rate": 3.75e-06,
"log_odds_chosen": 4.703558444976807,
"log_odds_ratio": -0.06566660106182098,
"logits/chosen": 368.7803955078125,
"logits/rejected": 383.49188232421875,
"logps/chosen": -0.4126485288143158,
"logps/rejected": -3.8882851600646973,
"loss": 0.619,
"nll_loss": 0.5621041059494019,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.2063242644071579,
"rewards/margins": 1.7378184795379639,
"rewards/rejected": -1.9441425800323486,
"step": 400
},
{
"epoch": 2.4143070044709387,
"grad_norm": 12.980825424194336,
"learning_rate": 3.72677996249965e-06,
"log_odds_chosen": 4.317046165466309,
"log_odds_ratio": -0.07360969483852386,
"logits/chosen": 422.6468200683594,
"logits/rejected": 380.22998046875,
"logps/chosen": -0.43779826164245605,
"logps/rejected": -3.444517135620117,
"loss": 0.6456,
"nll_loss": 0.6048913598060608,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.21889913082122803,
"rewards/margins": 1.5033594369888306,
"rewards/rejected": -1.7222585678100586,
"step": 405
},
{
"epoch": 2.444113263785395,
"grad_norm": 11.386286735534668,
"learning_rate": 3.7039859874359616e-06,
"log_odds_chosen": 3.6852588653564453,
"log_odds_ratio": -0.06955203413963318,
"logits/chosen": 365.75726318359375,
"logits/rejected": 373.08935546875,
"logps/chosen": -0.4394207000732422,
"logps/rejected": -2.8892974853515625,
"loss": 0.6307,
"nll_loss": 0.6479864120483398,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.2197103500366211,
"rewards/margins": 1.2249382734298706,
"rewards/rejected": -1.4446487426757812,
"step": 410
},
{
"epoch": 2.473919523099851,
"grad_norm": 18.22783088684082,
"learning_rate": 3.68160520255362e-06,
"log_odds_chosen": 4.396167755126953,
"log_odds_ratio": -0.04461110755801201,
"logits/chosen": 371.11077880859375,
"logits/rejected": 393.236328125,
"logps/chosen": -0.4007198214530945,
"logps/rejected": -3.539832353591919,
"loss": 0.6134,
"nll_loss": 0.5493707656860352,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.20035991072654724,
"rewards/margins": 1.5695561170578003,
"rewards/rejected": -1.7699161767959595,
"step": 415
},
{
"epoch": 2.503725782414307,
"grad_norm": 12.961064338684082,
"learning_rate": 3.6596252735569993e-06,
"log_odds_chosen": 3.7051539421081543,
"log_odds_ratio": -0.06346116960048676,
"logits/chosen": 368.0414123535156,
"logits/rejected": 338.7422180175781,
"logps/chosen": -0.38283970952033997,
"logps/rejected": -2.9152991771698,
"loss": 0.6314,
"nll_loss": 0.6913596987724304,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.19141985476016998,
"rewards/margins": 1.2662298679351807,
"rewards/rejected": -1.4576495885849,
"step": 420
},
{
"epoch": 2.533532041728763,
"grad_norm": 14.195425033569336,
"learning_rate": 3.6380343755449946e-06,
"log_odds_chosen": 4.560431480407715,
"log_odds_ratio": -0.03129279986023903,
"logits/chosen": 362.96746826171875,
"logits/rejected": 349.57958984375,
"logps/chosen": -0.4239065647125244,
"logps/rejected": -3.7524616718292236,
"loss": 0.6005,
"nll_loss": 0.6241891980171204,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2119532823562622,
"rewards/margins": 1.66427743434906,
"rewards/rejected": -1.8762308359146118,
"step": 425
},
{
"epoch": 2.563338301043219,
"grad_norm": 13.435834884643555,
"learning_rate": 3.6168211662780912e-06,
"log_odds_chosen": 4.4234299659729,
"log_odds_ratio": -0.03766634315252304,
"logits/chosen": 361.0891418457031,
"logits/rejected": 362.5020446777344,
"logps/chosen": -0.3212960362434387,
"logps/rejected": -3.3532252311706543,
"loss": 0.6194,
"nll_loss": 0.5339005589485168,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.16064801812171936,
"rewards/margins": 1.5159647464752197,
"rewards/rejected": -1.6766126155853271,
"step": 430
},
{
"epoch": 2.593144560357675,
"grad_norm": 13.10000228881836,
"learning_rate": 3.5959747611403815e-06,
"log_odds_chosen": 3.667933702468872,
"log_odds_ratio": -0.049441076815128326,
"logits/chosen": 370.7209777832031,
"logits/rejected": 329.81048583984375,
"logps/chosen": -0.3787091374397278,
"logps/rejected": -2.830317735671997,
"loss": 0.6284,
"nll_loss": 0.5626553297042847,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1893545687198639,
"rewards/margins": 1.2258042097091675,
"rewards/rejected": -1.4151588678359985,
"step": 435
},
{
"epoch": 2.6229508196721314,
"grad_norm": 16.336698532104492,
"learning_rate": 3.575484709670971e-06,
"log_odds_chosen": 4.056857585906982,
"log_odds_ratio": -0.035340312868356705,
"logits/chosen": 350.6004333496094,
"logits/rejected": 356.38323974609375,
"logps/chosen": -0.3078530430793762,
"logps/rejected": -2.8841094970703125,
"loss": 0.6605,
"nll_loss": 0.6005590558052063,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1539265215396881,
"rewards/margins": 1.288128137588501,
"rewards/rejected": -1.4420547485351562,
"step": 440
},
{
"epoch": 2.6527570789865873,
"grad_norm": 16.911279678344727,
"learning_rate": 3.5553409735498294e-06,
"log_odds_chosen": 4.540930271148682,
"log_odds_ratio": -0.021845344454050064,
"logits/chosen": 333.2252197265625,
"logits/rejected": 360.4376525878906,
"logps/chosen": -0.3848707973957062,
"logps/rejected": -3.6115005016326904,
"loss": 0.6305,
"nll_loss": 0.616925060749054,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1924353986978531,
"rewards/margins": 1.6133148670196533,
"rewards/rejected": -1.8057502508163452,
"step": 445
},
{
"epoch": 2.682563338301043,
"grad_norm": 12.66332721710205,
"learning_rate": 3.535533905932738e-06,
"log_odds_chosen": 3.8172874450683594,
"log_odds_ratio": -0.04109153896570206,
"logits/chosen": 363.5258483886719,
"logits/rejected": 355.9951171875,
"logps/chosen": -0.442394495010376,
"logps/rejected": -3.0551743507385254,
"loss": 0.6594,
"nll_loss": 0.6155456304550171,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.221197247505188,
"rewards/margins": 1.3063898086547852,
"rewards/rejected": -1.5275871753692627,
"step": 450
},
{
"epoch": 2.712369597615499,
"grad_norm": 11.626214981079102,
"learning_rate": 3.516054232038716e-06,
"log_odds_chosen": 3.7958292961120605,
"log_odds_ratio": -0.05676658824086189,
"logits/chosen": 394.8970947265625,
"logits/rejected": 370.40692138671875,
"logps/chosen": -0.49928539991378784,
"logps/rejected": -3.0073413848876953,
"loss": 0.6971,
"nll_loss": 0.6259102821350098,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24964269995689392,
"rewards/margins": 1.2540279626846313,
"rewards/rejected": -1.5036706924438477,
"step": 455
},
{
"epoch": 2.742175856929955,
"grad_norm": 12.148904800415039,
"learning_rate": 3.496893030901177e-06,
"log_odds_chosen": 4.385091781616211,
"log_odds_ratio": -0.0362153984606266,
"logits/chosen": 390.84942626953125,
"logits/rejected": 403.558837890625,
"logps/chosen": -0.35050949454307556,
"logps/rejected": -3.5067138671875,
"loss": 0.64,
"nll_loss": 0.6797647476196289,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.17525474727153778,
"rewards/margins": 1.5781023502349854,
"rewards/rejected": -1.75335693359375,
"step": 460
},
{
"epoch": 2.7719821162444114,
"grad_norm": 16.0198974609375,
"learning_rate": 3.478041718201262e-06,
"log_odds_chosen": 4.281471252441406,
"log_odds_ratio": -0.04437633976340294,
"logits/chosen": 379.65142822265625,
"logits/rejected": 373.5622863769531,
"logps/chosen": -0.3376389443874359,
"logps/rejected": -3.2862961292266846,
"loss": 0.6475,
"nll_loss": 0.6045721769332886,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.16881947219371796,
"rewards/margins": 1.4743287563323975,
"rewards/rejected": -1.6431480646133423,
"step": 465
},
{
"epoch": 2.8017883755588673,
"grad_norm": 20.6240291595459,
"learning_rate": 3.459492030108319e-06,
"log_odds_chosen": 3.6953399181365967,
"log_odds_ratio": -0.07698026299476624,
"logits/chosen": 384.45758056640625,
"logits/rejected": 344.5814208984375,
"logps/chosen": -0.43422994017601013,
"logps/rejected": -2.9317374229431152,
"loss": 0.6539,
"nll_loss": 0.6051168441772461,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.21711497008800507,
"rewards/margins": 1.2487537860870361,
"rewards/rejected": -1.4658687114715576,
"step": 470
},
{
"epoch": 2.8315946348733236,
"grad_norm": 16.241731643676758,
"learning_rate": 3.441236008058426e-06,
"log_odds_chosen": 4.172859191894531,
"log_odds_ratio": -0.03734353557229042,
"logits/chosen": 359.6179504394531,
"logits/rejected": 372.67559814453125,
"logps/chosen": -0.3877125382423401,
"logps/rejected": -3.2986769676208496,
"loss": 0.6017,
"nll_loss": 0.5468435287475586,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.19385626912117004,
"rewards/margins": 1.4554822444915771,
"rewards/rejected": -1.6493384838104248,
"step": 475
},
{
"epoch": 2.8614008941877795,
"grad_norm": 13.836338996887207,
"learning_rate": 3.4232659844072883e-06,
"log_odds_chosen": 5.078559875488281,
"log_odds_ratio": -0.040774669498205185,
"logits/chosen": 369.23388671875,
"logits/rejected": 398.89654541015625,
"logps/chosen": -0.4116280674934387,
"logps/rejected": -4.2671427726745605,
"loss": 0.5823,
"nll_loss": 0.5545191764831543,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.20581403374671936,
"rewards/margins": 1.9277572631835938,
"rewards/rejected": -2.1335713863372803,
"step": 480
},
{
"epoch": 2.8912071535022354,
"grad_norm": 13.629563331604004,
"learning_rate": 3.4055745688987482e-06,
"log_odds_chosen": 3.756863832473755,
"log_odds_ratio": -0.05358726903796196,
"logits/chosen": 360.4635925292969,
"logits/rejected": 379.5230407714844,
"logps/chosen": -0.4683550000190735,
"logps/rejected": -3.0478286743164062,
"loss": 0.6356,
"nll_loss": 0.5991590023040771,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.23417750000953674,
"rewards/margins": 1.2897369861602783,
"rewards/rejected": -1.5239143371582031,
"step": 485
},
{
"epoch": 2.9210134128166914,
"grad_norm": 11.519225120544434,
"learning_rate": 3.3881546358946924e-06,
"log_odds_chosen": 4.434014320373535,
"log_odds_ratio": -0.048062875866889954,
"logits/chosen": 385.712158203125,
"logits/rejected": 336.68524169921875,
"logps/chosen": -0.45937252044677734,
"logps/rejected": -3.5976157188415527,
"loss": 0.6968,
"nll_loss": 0.5958720445632935,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.22968626022338867,
"rewards/margins": 1.5691215991973877,
"rewards/rejected": -1.7988078594207764,
"step": 490
},
{
"epoch": 2.9508196721311473,
"grad_norm": 16.139705657958984,
"learning_rate": 3.37099931231621e-06,
"log_odds_chosen": 4.680273532867432,
"log_odds_ratio": -0.062451474368572235,
"logits/chosen": 375.66119384765625,
"logits/rejected": 394.88690185546875,
"logps/chosen": -0.4953223764896393,
"logps/rejected": -3.950305938720703,
"loss": 0.6257,
"nll_loss": 0.628521203994751,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24766118824481964,
"rewards/margins": 1.7274919748306274,
"rewards/rejected": -1.9751529693603516,
"step": 495
},
{
"epoch": 2.9806259314456036,
"grad_norm": 15.226314544677734,
"learning_rate": 3.3541019662496845e-06,
"log_odds_chosen": 4.290003776550293,
"log_odds_ratio": -0.04303581267595291,
"logits/chosen": 379.34765625,
"logits/rejected": 365.36773681640625,
"logps/chosen": -0.3512369692325592,
"logps/rejected": -3.2438788414001465,
"loss": 0.6379,
"nll_loss": 0.5231742858886719,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1756184846162796,
"rewards/margins": 1.44632089138031,
"rewards/rejected": -1.6219394207000732,
"step": 500
},
{
"epoch": 2.9865871833084947,
"eval_log_odds_chosen": 0.33574143052101135,
"eval_log_odds_ratio": -0.6769095063209534,
"eval_logits/chosen": 292.74676513671875,
"eval_logits/rejected": 259.2041320800781,
"eval_logps/chosen": -1.2910676002502441,
"eval_logps/rejected": -1.5239651203155518,
"eval_loss": 2.003471851348877,
"eval_nll_loss": 1.6439613103866577,
"eval_rewards/accuracies": 0.6115108132362366,
"eval_rewards/chosen": -0.6455338001251221,
"eval_rewards/margins": 0.1164487823843956,
"eval_rewards/rejected": -0.7619825601577759,
"eval_runtime": 378.0521,
"eval_samples_per_second": 1.463,
"eval_steps_per_second": 0.368,
"step": 501
},
{
"epoch": 2.9865871833084947,
"step": 501,
"total_flos": 0.0,
"train_loss": 1.6206341998543805,
"train_runtime": 33576.7022,
"train_samples_per_second": 0.479,
"train_steps_per_second": 0.015
}
],
"logging_steps": 5,
"max_steps": 501,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}