Mistral-7B-Base-SimPO2-5e-7 / trainer_state.json
tengxiao1
TX
7a271e3
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9987943737441393,
"eval_steps": 400,
"global_step": 466,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.010716677829872739,
"grad_norm": 8.776682238147446,
"learning_rate": 5.3191489361702123e-08,
"logits/chosen": -2.8905415534973145,
"logits/rejected": -2.8877615928649902,
"logps/chosen": -0.9742914438247681,
"logps/rejected": -0.9831219911575317,
"loss": 0.0001,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.9742914438247681,
"rewards/margins": 0.008830604143440723,
"rewards/rejected": -0.9831219911575317,
"step": 5
},
{
"epoch": 0.021433355659745478,
"grad_norm": 9.810248541108757,
"learning_rate": 1.0638297872340425e-07,
"logits/chosen": -2.9070327281951904,
"logits/rejected": -2.8929152488708496,
"logps/chosen": -0.978168785572052,
"logps/rejected": -1.001371145248413,
"loss": -0.0,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.978168785572052,
"rewards/margins": 0.023202406242489815,
"rewards/rejected": -1.001371145248413,
"step": 10
},
{
"epoch": 0.032150033489618215,
"grad_norm": 7.0995253496741135,
"learning_rate": 1.5957446808510638e-07,
"logits/chosen": -2.895216703414917,
"logits/rejected": -2.9180421829223633,
"logps/chosen": -0.9596306085586548,
"logps/rejected": -0.9847423434257507,
"loss": -0.0001,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.9596306085586548,
"rewards/margins": 0.02511175535619259,
"rewards/rejected": -0.9847423434257507,
"step": 15
},
{
"epoch": 0.042866711319490956,
"grad_norm": 7.785343608601427,
"learning_rate": 2.127659574468085e-07,
"logits/chosen": -2.902580976486206,
"logits/rejected": -2.9009933471679688,
"logps/chosen": -0.9715825319290161,
"logps/rejected": -0.9759886860847473,
"loss": -0.0,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.9715825319290161,
"rewards/margins": 0.00440616998821497,
"rewards/rejected": -0.9759886860847473,
"step": 20
},
{
"epoch": 0.0535833891493637,
"grad_norm": 6.677386701450202,
"learning_rate": 2.659574468085106e-07,
"logits/chosen": -2.8314852714538574,
"logits/rejected": -2.8363277912139893,
"logps/chosen": -0.9519669413566589,
"logps/rejected": -1.0002533197402954,
"loss": -0.0003,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.9519669413566589,
"rewards/margins": 0.048286281526088715,
"rewards/rejected": -1.0002533197402954,
"step": 25
},
{
"epoch": 0.06430006697923643,
"grad_norm": 8.4552870442767,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -2.895077705383301,
"logits/rejected": -2.9221832752227783,
"logps/chosen": -0.983117938041687,
"logps/rejected": -0.9829673767089844,
"loss": -0.0002,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.983117938041687,
"rewards/margins": -0.00015044640167616308,
"rewards/rejected": -0.9829673767089844,
"step": 30
},
{
"epoch": 0.07501674480910918,
"grad_norm": 7.60248558460589,
"learning_rate": 3.7234042553191484e-07,
"logits/chosen": -2.9559242725372314,
"logits/rejected": -2.9249794483184814,
"logps/chosen": -0.95380699634552,
"logps/rejected": -0.9984177350997925,
"loss": -0.0004,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.95380699634552,
"rewards/margins": 0.04461073502898216,
"rewards/rejected": -0.9984177350997925,
"step": 35
},
{
"epoch": 0.08573342263898191,
"grad_norm": 8.131750300550957,
"learning_rate": 4.25531914893617e-07,
"logits/chosen": -2.8956754207611084,
"logits/rejected": -2.8939049243927,
"logps/chosen": -0.9987373352050781,
"logps/rejected": -0.9930660128593445,
"loss": -0.0015,
"rewards/accuracies": 0.48124998807907104,
"rewards/chosen": -0.9987373352050781,
"rewards/margins": -0.005671312101185322,
"rewards/rejected": -0.9930660128593445,
"step": 40
},
{
"epoch": 0.09645010046885466,
"grad_norm": 7.488747662199227,
"learning_rate": 4.787234042553192e-07,
"logits/chosen": -2.8674330711364746,
"logits/rejected": -2.856853485107422,
"logps/chosen": -0.9764472246170044,
"logps/rejected": -0.9902997016906738,
"loss": -0.001,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.9764472246170044,
"rewards/margins": 0.013852563686668873,
"rewards/rejected": -0.9902997016906738,
"step": 45
},
{
"epoch": 0.1071667782987274,
"grad_norm": 7.447612263721868,
"learning_rate": 4.999367579541646e-07,
"logits/chosen": -2.875795364379883,
"logits/rejected": -2.8871512413024902,
"logps/chosen": -0.9369581937789917,
"logps/rejected": -0.9167054891586304,
"loss": -0.0027,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.9369581937789917,
"rewards/margins": -0.020252685993909836,
"rewards/rejected": -0.9167054891586304,
"step": 50
},
{
"epoch": 0.11788345612860013,
"grad_norm": 8.432358893020524,
"learning_rate": 4.99550394651012e-07,
"logits/chosen": -2.915107011795044,
"logits/rejected": -2.925875186920166,
"logps/chosen": -0.9737237095832825,
"logps/rejected": -1.0020536184310913,
"loss": -0.0022,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.9737237095832825,
"rewards/margins": 0.028329964727163315,
"rewards/rejected": -1.0020536184310913,
"step": 55
},
{
"epoch": 0.12860013395847286,
"grad_norm": 9.97321343058875,
"learning_rate": 4.988133448023071e-07,
"logits/chosen": -2.8224921226501465,
"logits/rejected": -2.8447964191436768,
"logps/chosen": -0.9149552583694458,
"logps/rejected": -0.9350795745849609,
"loss": -0.0041,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.9149552583694458,
"rewards/margins": 0.02012430503964424,
"rewards/rejected": -0.9350795745849609,
"step": 60
},
{
"epoch": 0.13931681178834562,
"grad_norm": 8.412802955896868,
"learning_rate": 4.977266441646379e-07,
"logits/chosen": -2.900336503982544,
"logits/rejected": -2.9115452766418457,
"logps/chosen": -0.8882263898849487,
"logps/rejected": -0.9012020230293274,
"loss": -0.0057,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.8882263898849487,
"rewards/margins": 0.012975621037185192,
"rewards/rejected": -0.9012020230293274,
"step": 65
},
{
"epoch": 0.15003348961821836,
"grad_norm": 8.21023142941744,
"learning_rate": 4.962918198495654e-07,
"logits/chosen": -2.816331386566162,
"logits/rejected": -2.839690685272217,
"logps/chosen": -0.9164741635322571,
"logps/rejected": -0.9583846926689148,
"loss": -0.007,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.9164741635322571,
"rewards/margins": 0.04191048443317413,
"rewards/rejected": -0.9583846926689148,
"step": 70
},
{
"epoch": 0.1607501674480911,
"grad_norm": 8.354728288045035,
"learning_rate": 4.945108881776135e-07,
"logits/chosen": -2.8184382915496826,
"logits/rejected": -2.81858229637146,
"logps/chosen": -0.8914631009101868,
"logps/rejected": -0.9294856190681458,
"loss": -0.0066,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.8914631009101868,
"rewards/margins": 0.038022562861442566,
"rewards/rejected": -0.9294856190681458,
"step": 75
},
{
"epoch": 0.17146684527796383,
"grad_norm": 9.546162491226012,
"learning_rate": 4.923863518447879e-07,
"logits/chosen": -2.840766191482544,
"logits/rejected": -2.8635830879211426,
"logps/chosen": -0.887871265411377,
"logps/rejected": -0.958912193775177,
"loss": -0.006,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.887871265411377,
"rewards/margins": 0.07104082405567169,
"rewards/rejected": -0.958912193775177,
"step": 80
},
{
"epoch": 0.18218352310783656,
"grad_norm": 9.354318155678731,
"learning_rate": 4.899211964056035e-07,
"logits/chosen": -2.854689359664917,
"logits/rejected": -2.866008758544922,
"logps/chosen": -0.9082525968551636,
"logps/rejected": -0.9574602842330933,
"loss": -0.0132,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.9082525968551636,
"rewards/margins": 0.04920773208141327,
"rewards/rejected": -0.9574602842330933,
"step": 85
},
{
"epoch": 0.19290020093770932,
"grad_norm": 9.38237658589797,
"learning_rate": 4.871188860775643e-07,
"logits/chosen": -2.830294132232666,
"logits/rejected": -2.8332040309906006,
"logps/chosen": -0.9233630895614624,
"logps/rejected": -0.9633795022964478,
"loss": -0.0142,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.9233630895614624,
"rewards/margins": 0.040016382932662964,
"rewards/rejected": -0.9633795022964478,
"step": 90
},
{
"epoch": 0.20361687876758205,
"grad_norm": 8.979685614931006,
"learning_rate": 4.839833588729897e-07,
"logits/chosen": -2.8387644290924072,
"logits/rejected": -2.8495821952819824,
"logps/chosen": -0.9509484171867371,
"logps/rejected": -0.974162220954895,
"loss": -0.0085,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.9509484171867371,
"rewards/margins": 0.02321377769112587,
"rewards/rejected": -0.974162220954895,
"step": 95
},
{
"epoch": 0.2143335565974548,
"grad_norm": 9.427728619757332,
"learning_rate": 4.80519021065031e-07,
"logits/chosen": -2.860654354095459,
"logits/rejected": -2.8680408000946045,
"logps/chosen": -0.9869228601455688,
"logps/rejected": -1.0385963916778564,
"loss": -0.015,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.9869228601455688,
"rewards/margins": 0.05167357996106148,
"rewards/rejected": -1.0385963916778564,
"step": 100
},
{
"epoch": 0.22505023442732752,
"grad_norm": 8.191485445947501,
"learning_rate": 4.767307409956528e-07,
"logits/chosen": -2.859048843383789,
"logits/rejected": -2.8625106811523438,
"logps/chosen": -0.8909996151924133,
"logps/rejected": -0.9282458424568176,
"loss": -0.0154,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.8909996151924133,
"rewards/margins": 0.03724620118737221,
"rewards/rejected": -0.9282458424568176,
"step": 105
},
{
"epoch": 0.23576691225720026,
"grad_norm": 9.657244741544229,
"learning_rate": 4.726238422342805e-07,
"logits/chosen": -2.8523192405700684,
"logits/rejected": -2.84688138961792,
"logps/chosen": -0.9256642460823059,
"logps/rejected": -0.9944097399711609,
"loss": -0.0136,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9256642460823059,
"rewards/margins": 0.06874561309814453,
"rewards/rejected": -0.9944097399711609,
"step": 110
},
{
"epoch": 0.24648359008707302,
"grad_norm": 11.45037121550009,
"learning_rate": 4.6820409609673027e-07,
"logits/chosen": -2.895726203918457,
"logits/rejected": -2.900205612182617,
"logps/chosen": -0.9234002828598022,
"logps/rejected": -0.938769519329071,
"loss": -0.0216,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.9234002828598022,
"rewards/margins": 0.015369323082268238,
"rewards/rejected": -0.938769519329071,
"step": 115
},
{
"epoch": 0.2572002679169457,
"grad_norm": 9.77719736840442,
"learning_rate": 4.6347771353493173e-07,
"logits/chosen": -2.9136910438537598,
"logits/rejected": -2.930593252182007,
"logps/chosen": -0.9225813150405884,
"logps/rejected": -0.917222797870636,
"loss": -0.0168,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.9225813150405884,
"rewards/margins": -0.005358507391065359,
"rewards/rejected": -0.917222797870636,
"step": 120
},
{
"epoch": 0.2679169457468185,
"grad_norm": 10.417139695198527,
"learning_rate": 4.5845133640884217e-07,
"logits/chosen": -2.898656129837036,
"logits/rejected": -2.8996481895446777,
"logps/chosen": -0.9208585619926453,
"logps/rejected": -0.9479470252990723,
"loss": -0.0193,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.9208585619926453,
"rewards/margins": 0.0270883496850729,
"rewards/rejected": -0.9479470252990723,
"step": 125
},
{
"epoch": 0.27863362357669125,
"grad_norm": 11.697763554027677,
"learning_rate": 4.531320281528169e-07,
"logits/chosen": -2.9303739070892334,
"logits/rejected": -2.9241251945495605,
"logps/chosen": -0.9110285639762878,
"logps/rejected": -0.9726667404174805,
"loss": -0.0227,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.9110285639762878,
"rewards/margins": 0.06163806468248367,
"rewards/rejected": -0.9726667404174805,
"step": 130
},
{
"epoch": 0.289350301406564,
"grad_norm": 11.144902490144164,
"learning_rate": 4.4752726384955293e-07,
"logits/chosen": -2.9404830932617188,
"logits/rejected": -2.947343111038208,
"logps/chosen": -0.9363792538642883,
"logps/rejected": -0.9744508862495422,
"loss": -0.0195,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.9363792538642883,
"rewards/margins": 0.03807152435183525,
"rewards/rejected": -0.9744508862495422,
"step": 135
},
{
"epoch": 0.3000669792364367,
"grad_norm": 11.921950508886521,
"learning_rate": 4.4164491972555294e-07,
"logits/chosen": -2.8976681232452393,
"logits/rejected": -2.898315906524658,
"logps/chosen": -0.9219269752502441,
"logps/rejected": -0.9640507698059082,
"loss": -0.0259,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.9219269752502441,
"rewards/margins": 0.04212382063269615,
"rewards/rejected": -0.9640507698059082,
"step": 140
},
{
"epoch": 0.31078365706630945,
"grad_norm": 13.471098814591679,
"learning_rate": 4.3549326208287393e-07,
"logits/chosen": -2.9619617462158203,
"logits/rejected": -2.9274888038635254,
"logps/chosen": -0.9526675343513489,
"logps/rejected": -1.007359266281128,
"loss": -0.0355,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.9526675343513489,
"rewards/margins": 0.0546916201710701,
"rewards/rejected": -1.007359266281128,
"step": 145
},
{
"epoch": 0.3215003348961822,
"grad_norm": 12.766870563789471,
"learning_rate": 4.2908093568271187e-07,
"logits/chosen": -2.9606540203094482,
"logits/rejected": -2.9726486206054688,
"logps/chosen": -0.9194633364677429,
"logps/rejected": -0.9709534645080566,
"loss": -0.0371,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.9194633364677429,
"rewards/margins": 0.05149019509553909,
"rewards/rejected": -0.9709534645080566,
"step": 150
},
{
"epoch": 0.3322170127260549,
"grad_norm": 12.28868938106036,
"learning_rate": 4.2241695159714843e-07,
"logits/chosen": -3.043233633041382,
"logits/rejected": -3.0254199504852295,
"logps/chosen": -0.9320791959762573,
"logps/rejected": -0.9708281755447388,
"loss": -0.0274,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.9320791959762573,
"rewards/margins": 0.038748957216739655,
"rewards/rejected": -0.9708281755447388,
"step": 155
},
{
"epoch": 0.34293369055592765,
"grad_norm": 16.212616603609877,
"learning_rate": 4.1551067454613074e-07,
"logits/chosen": -2.943720579147339,
"logits/rejected": -2.928537607192993,
"logps/chosen": -0.921017050743103,
"logps/rejected": -0.9882138967514038,
"loss": -0.0359,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.921017050743103,
"rewards/margins": 0.06719694286584854,
"rewards/rejected": -0.9882138967514038,
"step": 160
},
{
"epoch": 0.3536503683858004,
"grad_norm": 15.792664568932544,
"learning_rate": 4.0837180973747876e-07,
"logits/chosen": -3.0298409461975098,
"logits/rejected": -3.0069289207458496,
"logps/chosen": -0.9174503087997437,
"logps/rejected": -0.9726656079292297,
"loss": -0.0342,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.9174503087997437,
"rewards/margins": 0.05521535873413086,
"rewards/rejected": -0.9726656079292297,
"step": 165
},
{
"epoch": 0.3643670462156731,
"grad_norm": 10.787104513853434,
"learning_rate": 4.0101038922841455e-07,
"logits/chosen": -2.968522548675537,
"logits/rejected": -2.9739623069763184,
"logps/chosen": -0.9771224856376648,
"logps/rejected": -1.0373847484588623,
"loss": -0.0466,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.9771224856376648,
"rewards/margins": 0.06026231497526169,
"rewards/rejected": -1.0373847484588623,
"step": 170
},
{
"epoch": 0.3750837240455459,
"grad_norm": 13.17830711446757,
"learning_rate": 3.934367578277783e-07,
"logits/chosen": -2.9837470054626465,
"logits/rejected": -2.9880218505859375,
"logps/chosen": -1.000985026359558,
"logps/rejected": -1.0419981479644775,
"loss": -0.0381,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.000985026359558,
"rewards/margins": 0.041013117879629135,
"rewards/rejected": -1.0419981479644775,
"step": 175
},
{
"epoch": 0.38580040187541864,
"grad_norm": 14.72612824066128,
"learning_rate": 3.856615585587434e-07,
"logits/chosen": -2.9933667182922363,
"logits/rejected": -3.0047719478607178,
"logps/chosen": -1.0653940439224243,
"logps/rejected": -1.1087322235107422,
"loss": -0.0319,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -1.0653940439224243,
"rewards/margins": 0.043338216841220856,
"rewards/rejected": -1.1087322235107422,
"step": 180
},
{
"epoch": 0.3965170797052914,
"grad_norm": 13.397319728202142,
"learning_rate": 3.776957177024581e-07,
"logits/chosen": -2.9812378883361816,
"logits/rejected": -2.9986259937286377,
"logps/chosen": -1.0316323041915894,
"logps/rejected": -1.0616919994354248,
"loss": -0.033,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.0316323041915894,
"rewards/margins": 0.030059661716222763,
"rewards/rejected": -1.0616919994354248,
"step": 185
},
{
"epoch": 0.4072337575351641,
"grad_norm": 14.059564629221503,
"learning_rate": 3.69550429443633e-07,
"logits/chosen": -2.8960227966308594,
"logits/rejected": -2.913637161254883,
"logps/chosen": -1.012019157409668,
"logps/rejected": -1.215237021446228,
"loss": -0.0447,
"rewards/accuracies": 0.65625,
"rewards/chosen": -1.012019157409668,
"rewards/margins": 0.20321793854236603,
"rewards/rejected": -1.215237021446228,
"step": 190
},
{
"epoch": 0.41795043536503684,
"grad_norm": 14.952805148520074,
"learning_rate": 3.612371401396502e-07,
"logits/chosen": -3.007540225982666,
"logits/rejected": -2.9880564212799072,
"logps/chosen": -1.0366661548614502,
"logps/rejected": -1.0943790674209595,
"loss": -0.0368,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.0366661548614502,
"rewards/margins": 0.05771297961473465,
"rewards/rejected": -1.0943790674209595,
"step": 195
},
{
"epoch": 0.4286671131949096,
"grad_norm": 14.170450035952996,
"learning_rate": 3.527675322353011e-07,
"logits/chosen": -2.922602415084839,
"logits/rejected": -2.9457390308380127,
"logps/chosen": -0.970561146736145,
"logps/rejected": -1.0033313035964966,
"loss": -0.0473,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.970561146736145,
"rewards/margins": 0.032770149409770966,
"rewards/rejected": -1.0033313035964966,
"step": 200
},
{
"epoch": 0.4393837910247823,
"grad_norm": 15.488305590600266,
"learning_rate": 3.4415350784575693e-07,
"logits/chosen": -3.004404067993164,
"logits/rejected": -2.9882898330688477,
"logps/chosen": -0.9613070487976074,
"logps/rejected": -1.0151447057724,
"loss": -0.044,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.9613070487976074,
"rewards/margins": 0.053837649524211884,
"rewards/rejected": -1.0151447057724,
"step": 205
},
{
"epoch": 0.45010046885465504,
"grad_norm": 13.543647630514888,
"learning_rate": 3.3540717203084224e-07,
"logits/chosen": -2.956808090209961,
"logits/rejected": -2.9622817039489746,
"logps/chosen": -0.9170505404472351,
"logps/rejected": -0.9884069561958313,
"loss": -0.0514,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9170505404472351,
"rewards/margins": 0.07135634124279022,
"rewards/rejected": -0.9884069561958313,
"step": 210
},
{
"epoch": 0.4608171466845278,
"grad_norm": 14.183913951293327,
"learning_rate": 3.265408157841153e-07,
"logits/chosen": -2.9880080223083496,
"logits/rejected": -2.986445903778076,
"logps/chosen": -0.9918737411499023,
"logps/rejected": -1.0286133289337158,
"loss": -0.0446,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.9918737411499023,
"rewards/margins": 0.036739740520715714,
"rewards/rejected": -1.0286133289337158,
"step": 215
},
{
"epoch": 0.4715338245144005,
"grad_norm": 16.687524624276737,
"learning_rate": 3.1756689876066187e-07,
"logits/chosen": -2.940664291381836,
"logits/rejected": -2.940556049346924,
"logps/chosen": -0.9756967425346375,
"logps/rejected": -1.0494619607925415,
"loss": -0.0496,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.9756967425346375,
"rewards/margins": 0.07376515865325928,
"rewards/rejected": -1.0494619607925415,
"step": 220
},
{
"epoch": 0.4822505023442733,
"grad_norm": 15.106449736541158,
"learning_rate": 3.0849803176787184e-07,
"logits/chosen": -2.978764057159424,
"logits/rejected": -3.02057147026062,
"logps/chosen": -0.9675830602645874,
"logps/rejected": -1.0782067775726318,
"loss": -0.0528,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9675830602645874,
"rewards/margins": 0.11062371730804443,
"rewards/rejected": -1.0782067775726318,
"step": 225
},
{
"epoch": 0.49296718017414604,
"grad_norm": 19.732642999044504,
"learning_rate": 2.993469590438065e-07,
"logits/chosen": -2.91894793510437,
"logits/rejected": -2.92021107673645,
"logps/chosen": -0.9777423143386841,
"logps/rejected": -1.0478038787841797,
"loss": -0.0487,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9777423143386841,
"rewards/margins": 0.07006163895130157,
"rewards/rejected": -1.0478038787841797,
"step": 230
},
{
"epoch": 0.5036838580040187,
"grad_norm": 15.470655040763342,
"learning_rate": 2.901265403480597e-07,
"logits/chosen": -2.868373394012451,
"logits/rejected": -2.8822197914123535,
"logps/chosen": -0.9746626615524292,
"logps/rejected": -1.109392523765564,
"loss": -0.0532,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.9746626615524292,
"rewards/margins": 0.13472989201545715,
"rewards/rejected": -1.109392523765564,
"step": 235
},
{
"epoch": 0.5144005358338914,
"grad_norm": 18.117444556863273,
"learning_rate": 2.808497328902782e-07,
"logits/chosen": -2.829807758331299,
"logits/rejected": -2.837395191192627,
"logps/chosen": -1.0404951572418213,
"logps/rejected": -1.1007030010223389,
"loss": -0.0453,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.0404951572418213,
"rewards/margins": 0.06020789220929146,
"rewards/rejected": -1.1007030010223389,
"step": 240
},
{
"epoch": 0.5251172136637642,
"grad_norm": 18.628678925066545,
"learning_rate": 2.715295731217396e-07,
"logits/chosen": -2.81119966506958,
"logits/rejected": -2.815497636795044,
"logps/chosen": -1.0369569063186646,
"logps/rejected": -1.112683892250061,
"loss": -0.0596,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -1.0369569063186646,
"rewards/margins": 0.07572712004184723,
"rewards/rejected": -1.112683892250061,
"step": 245
},
{
"epoch": 0.535833891493637,
"grad_norm": 17.866763466807367,
"learning_rate": 2.621791584155725e-07,
"logits/chosen": -2.83900785446167,
"logits/rejected": -2.859978199005127,
"logps/chosen": -1.1023390293121338,
"logps/rejected": -1.1515963077545166,
"loss": -0.0617,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -1.1023390293121338,
"rewards/margins": 0.04925721138715744,
"rewards/rejected": -1.1515963077545166,
"step": 250
},
{
"epoch": 0.5465505693235098,
"grad_norm": 17.478329638335467,
"learning_rate": 2.52811628661367e-07,
"logits/chosen": -2.8983101844787598,
"logits/rejected": -2.908754348754883,
"logps/chosen": -1.0288032293319702,
"logps/rejected": -1.134108066558838,
"loss": -0.0631,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -1.0288032293319702,
"rewards/margins": 0.10530481487512589,
"rewards/rejected": -1.134108066558838,
"step": 255
},
{
"epoch": 0.5572672471533825,
"grad_norm": 15.049601923886751,
"learning_rate": 2.434401478000353e-07,
"logits/chosen": -2.8707144260406494,
"logits/rejected": -2.876131057739258,
"logps/chosen": -1.0516908168792725,
"logps/rejected": -1.1409704685211182,
"loss": -0.0405,
"rewards/accuracies": 0.5,
"rewards/chosen": -1.0516908168792725,
"rewards/margins": 0.08927960693836212,
"rewards/rejected": -1.1409704685211182,
"step": 260
},
{
"epoch": 0.5679839249832552,
"grad_norm": 14.714810625194913,
"learning_rate": 2.340778853248759e-07,
"logits/chosen": -2.86578369140625,
"logits/rejected": -2.8567006587982178,
"logps/chosen": -1.0564695596694946,
"logps/rejected": -1.199512004852295,
"loss": -0.0695,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -1.0564695596694946,
"rewards/margins": 0.143042653799057,
"rewards/rejected": -1.199512004852295,
"step": 265
},
{
"epoch": 0.578700602813128,
"grad_norm": 18.053080062016626,
"learning_rate": 2.247379977748339e-07,
"logits/chosen": -2.9022586345672607,
"logits/rejected": -2.9201228618621826,
"logps/chosen": -1.0403499603271484,
"logps/rejected": -1.1265124082565308,
"loss": -0.0673,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -1.0403499603271484,
"rewards/margins": 0.08616244792938232,
"rewards/rejected": -1.1265124082565308,
"step": 270
},
{
"epoch": 0.5894172806430007,
"grad_norm": 14.15566071309189,
"learning_rate": 2.1543361024596573e-07,
"logits/chosen": -2.9802463054656982,
"logits/rejected": -2.988959550857544,
"logps/chosen": -0.9800962209701538,
"logps/rejected": -1.0803120136260986,
"loss": -0.0525,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.9800962209701538,
"rewards/margins": 0.1002158373594284,
"rewards/rejected": -1.0803120136260986,
"step": 275
},
{
"epoch": 0.6001339584728734,
"grad_norm": 17.56629547418189,
"learning_rate": 2.0617779794709082e-07,
"logits/chosen": -2.8620638847351074,
"logits/rejected": -2.8629186153411865,
"logps/chosen": -1.103073000907898,
"logps/rejected": -1.196608304977417,
"loss": -0.0632,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -1.103073000907898,
"rewards/margins": 0.09353531897068024,
"rewards/rejected": -1.196608304977417,
"step": 280
},
{
"epoch": 0.6108506363027462,
"grad_norm": 15.46761530330997,
"learning_rate": 1.9698356782554687e-07,
"logits/chosen": -2.9556548595428467,
"logits/rejected": -2.9529995918273926,
"logps/chosen": -1.109263300895691,
"logps/rejected": -1.1731650829315186,
"loss": -0.0564,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.109263300895691,
"rewards/margins": 0.06390184909105301,
"rewards/rejected": -1.1731650829315186,
"step": 285
},
{
"epoch": 0.6215673141326189,
"grad_norm": 16.8037567850014,
"learning_rate": 1.878638402888727e-07,
"logits/chosen": -2.9940178394317627,
"logits/rejected": -2.9970099925994873,
"logps/chosen": -0.9795197248458862,
"logps/rejected": -1.0418453216552734,
"loss": -0.0668,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.9795197248458862,
"rewards/margins": 0.06232555955648422,
"rewards/rejected": -1.0418453216552734,
"step": 290
},
{
"epoch": 0.6322839919624916,
"grad_norm": 16.60747046570722,
"learning_rate": 1.7883143104810223e-07,
"logits/chosen": -2.9415085315704346,
"logits/rejected": -2.9544739723205566,
"logps/chosen": -1.069349765777588,
"logps/rejected": -1.2494323253631592,
"loss": -0.0739,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.069349765777588,
"rewards/margins": 0.1800827533006668,
"rewards/rejected": -1.2494323253631592,
"step": 295
},
{
"epoch": 0.6430006697923644,
"grad_norm": 15.422934564351497,
"learning_rate": 1.6989903310818548e-07,
"logits/chosen": -3.0317275524139404,
"logits/rejected": -3.049278736114502,
"logps/chosen": -1.0510863065719604,
"logps/rejected": -1.1395753622055054,
"loss": -0.0517,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.0510863065719604,
"rewards/margins": 0.08848897367715836,
"rewards/rejected": -1.1395753622055054,
"step": 300
},
{
"epoch": 0.6537173476222371,
"grad_norm": 17.56071364411128,
"learning_rate": 1.610791989308466e-07,
"logits/chosen": -2.9625232219696045,
"logits/rejected": -2.9480719566345215,
"logps/chosen": -1.0810691118240356,
"logps/rejected": -1.1474703550338745,
"loss": -0.0711,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -1.0810691118240356,
"rewards/margins": 0.06640125066041946,
"rewards/rejected": -1.1474703550338745,
"step": 305
},
{
"epoch": 0.6644340254521098,
"grad_norm": 19.400175535223255,
"learning_rate": 1.523843227949418e-07,
"logits/chosen": -3.0158638954162598,
"logits/rejected": -3.014029026031494,
"logps/chosen": -1.054434061050415,
"logps/rejected": -1.1792142391204834,
"loss": -0.0711,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.054434061050415,
"rewards/margins": 0.12478007376194,
"rewards/rejected": -1.1792142391204834,
"step": 310
},
{
"epoch": 0.6751507032819826,
"grad_norm": 18.063066961906372,
"learning_rate": 1.4382662337910867e-07,
"logits/chosen": -2.9173102378845215,
"logits/rejected": -2.9165103435516357,
"logps/chosen": -1.1207153797149658,
"logps/rejected": -1.2241191864013672,
"loss": -0.0582,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -1.1207153797149658,
"rewards/margins": 0.10340368747711182,
"rewards/rejected": -1.2241191864013672,
"step": 315
},
{
"epoch": 0.6858673811118553,
"grad_norm": 17.656998709658712,
"learning_rate": 1.3541812659118104e-07,
"logits/chosen": -2.924173593521118,
"logits/rejected": -2.901674747467041,
"logps/chosen": -1.1284511089324951,
"logps/rejected": -1.2347378730773926,
"loss": -0.0605,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.1284511089324951,
"rewards/margins": 0.10628668963909149,
"rewards/rejected": -1.2347378730773926,
"step": 320
},
{
"epoch": 0.696584058941728,
"grad_norm": 17.674353062779463,
"learning_rate": 1.2717064866850047e-07,
"logits/chosen": -2.9069221019744873,
"logits/rejected": -2.9266703128814697,
"logps/chosen": -1.0635308027267456,
"logps/rejected": -1.2443645000457764,
"loss": -0.076,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.0635308027267456,
"rewards/margins": 0.1808338761329651,
"rewards/rejected": -1.2443645000457764,
"step": 325
},
{
"epoch": 0.7073007367716008,
"grad_norm": 16.692003983984833,
"learning_rate": 1.1909577957287117e-07,
"logits/chosen": -2.9482550621032715,
"logits/rejected": -2.9383444786071777,
"logps/chosen": -1.1115998029708862,
"logps/rejected": -1.269691824913025,
"loss": -0.0875,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -1.1115998029708862,
"rewards/margins": 0.15809203684329987,
"rewards/rejected": -1.269691824913025,
"step": 330
},
{
"epoch": 0.7180174146014735,
"grad_norm": 16.832132437998734,
"learning_rate": 1.1120486670349441e-07,
"logits/chosen": -3.0022974014282227,
"logits/rejected": -3.0084285736083984,
"logps/chosen": -1.1076433658599854,
"logps/rejected": -1.2158708572387695,
"loss": -0.0631,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -1.1076433658599854,
"rewards/margins": 0.10822737216949463,
"rewards/rejected": -1.2158708572387695,
"step": 335
},
{
"epoch": 0.7287340924313462,
"grad_norm": 16.746834458412387,
"learning_rate": 1.0350899895076948e-07,
"logits/chosen": -2.960242509841919,
"logits/rejected": -2.9590137004852295,
"logps/chosen": -1.1618969440460205,
"logps/rejected": -1.3054715394973755,
"loss": -0.063,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.1618969440460205,
"rewards/margins": 0.14357443153858185,
"rewards/rejected": -1.3054715394973755,
"step": 340
},
{
"epoch": 0.739450770261219,
"grad_norm": 16.472011449839883,
"learning_rate": 9.601899111337069e-08,
"logits/chosen": -2.9834091663360596,
"logits/rejected": -2.9975457191467285,
"logps/chosen": -1.126527190208435,
"logps/rejected": -1.3144327402114868,
"loss": -0.0777,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -1.126527190208435,
"rewards/margins": 0.18790560960769653,
"rewards/rejected": -1.3144327402114868,
"step": 345
},
{
"epoch": 0.7501674480910918,
"grad_norm": 17.215572295903478,
"learning_rate": 8.874536870049765e-08,
"logits/chosen": -2.969423294067383,
"logits/rejected": -2.969230890274048,
"logps/chosen": -1.127826452255249,
"logps/rejected": -1.271470069885254,
"loss": -0.0752,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.127826452255249,
"rewards/margins": 0.1436435580253601,
"rewards/rejected": -1.271470069885254,
"step": 350
},
{
"epoch": 0.7608841259209645,
"grad_norm": 19.033752258043222,
"learning_rate": 8.169835314065663e-08,
"logits/chosen": -2.9632205963134766,
"logits/rejected": -2.9548847675323486,
"logps/chosen": -1.1820392608642578,
"logps/rejected": -1.2698676586151123,
"loss": -0.0668,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -1.1820392608642578,
"rewards/margins": 0.08782847970724106,
"rewards/rejected": -1.2698676586151123,
"step": 355
},
{
"epoch": 0.7716008037508373,
"grad_norm": 17.064795493163807,
"learning_rate": 7.488784741775816e-08,
"logits/chosen": -2.976466178894043,
"logits/rejected": -2.9713473320007324,
"logps/chosen": -1.054131269454956,
"logps/rejected": -1.1566834449768066,
"loss": -0.0663,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.054131269454956,
"rewards/margins": 0.10255217552185059,
"rewards/rejected": -1.1566834449768066,
"step": 360
},
{
"epoch": 0.78231748158071,
"grad_norm": 17.868156711335107,
"learning_rate": 6.832342215471715e-08,
"logits/chosen": -2.98659086227417,
"logits/rejected": -2.986912488937378,
"logps/chosen": -1.1025737524032593,
"logps/rejected": -1.242219090461731,
"loss": -0.0823,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.1025737524032593,
"rewards/margins": 0.13964524865150452,
"rewards/rejected": -1.242219090461731,
"step": 365
},
{
"epoch": 0.7930341594105828,
"grad_norm": 24.714287844140323,
"learning_rate": 6.201430216410985e-08,
"logits/chosen": -3.0034446716308594,
"logits/rejected": -3.001354217529297,
"logps/chosen": -1.1147871017456055,
"logps/rejected": -1.200984239578247,
"loss": -0.081,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.1147871017456055,
"rewards/margins": 0.08619704097509384,
"rewards/rejected": -1.200984239578247,
"step": 370
},
{
"epoch": 0.8037508372404555,
"grad_norm": 20.89166421346418,
"learning_rate": 5.596935348479029e-08,
"logits/chosen": -2.9388105869293213,
"logits/rejected": -2.948521137237549,
"logps/chosen": -1.2442572116851807,
"logps/rejected": -1.393157958984375,
"loss": -0.0861,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.2442572116851807,
"rewards/margins": 0.1489005833864212,
"rewards/rejected": -1.393157958984375,
"step": 375
},
{
"epoch": 0.8144675150703282,
"grad_norm": 22.406198534031674,
"learning_rate": 5.01970709226808e-08,
"logits/chosen": -3.028179883956909,
"logits/rejected": -3.009589672088623,
"logps/chosen": -1.0915172100067139,
"logps/rejected": -1.243607997894287,
"loss": -0.0837,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.0915172100067139,
"rewards/margins": 0.1520906686782837,
"rewards/rejected": -1.243607997894287,
"step": 380
},
{
"epoch": 0.825184192900201,
"grad_norm": 24.898962036749563,
"learning_rate": 4.4705566113246635e-08,
"logits/chosen": -2.9715943336486816,
"logits/rejected": -2.949638605117798,
"logps/chosen": -1.228461503982544,
"logps/rejected": -1.3865272998809814,
"loss": -0.0897,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.228461503982544,
"rewards/margins": 0.15806564688682556,
"rewards/rejected": -1.3865272998809814,
"step": 385
},
{
"epoch": 0.8359008707300737,
"grad_norm": 19.449984589836117,
"learning_rate": 3.950255612243042e-08,
"logits/chosen": -2.951904058456421,
"logits/rejected": -2.928208827972412,
"logps/chosen": -1.2001017332077026,
"logps/rejected": -1.3440535068511963,
"loss": -0.0724,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.2001017332077026,
"rewards/margins": 0.1439516693353653,
"rewards/rejected": -1.3440535068511963,
"step": 390
},
{
"epoch": 0.8466175485599464,
"grad_norm": 19.857222600639766,
"learning_rate": 3.459535260206384e-08,
"logits/chosen": -2.9977169036865234,
"logits/rejected": -3.020946979522705,
"logps/chosen": -1.096103549003601,
"logps/rejected": -1.21526300907135,
"loss": -0.0898,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -1.096103549003601,
"rewards/margins": 0.11915931850671768,
"rewards/rejected": -1.21526300907135,
"step": 395
},
{
"epoch": 0.8573342263898192,
"grad_norm": 21.73532568075866,
"learning_rate": 2.999085151499764e-08,
"logits/chosen": -2.9534425735473633,
"logits/rejected": -2.9359939098358154,
"logps/chosen": -1.0908204317092896,
"logps/rejected": -1.356081485748291,
"loss": -0.0905,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.0908204317092896,
"rewards/margins": 0.2652612626552582,
"rewards/rejected": -1.356081485748291,
"step": 400
},
{
"epoch": 0.8573342263898192,
"eval_logits/chosen": -2.8999884128570557,
"eval_logits/rejected": -2.892754316329956,
"eval_logps/chosen": -1.1346372365951538,
"eval_logps/rejected": -1.2466005086898804,
"eval_loss": -0.07080772519111633,
"eval_rewards/accuracies": 0.5757978558540344,
"eval_rewards/chosen": -1.1346372365951538,
"eval_rewards/margins": 0.11196334660053253,
"eval_rewards/rejected": -1.2466005086898804,
"eval_runtime": 432.4273,
"eval_samples_per_second": 6.924,
"eval_steps_per_second": 0.435,
"step": 400
},
{
"epoch": 0.8680509042196919,
"grad_norm": 19.440021908588918,
"learning_rate": 2.5695523444387747e-08,
"logits/chosen": -2.9388387203216553,
"logits/rejected": -2.9628615379333496,
"logps/chosen": -1.111882209777832,
"logps/rejected": -1.2351758480072021,
"loss": -0.09,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.111882209777832,
"rewards/margins": 0.12329357862472534,
"rewards/rejected": -1.2351758480072021,
"step": 405
},
{
"epoch": 0.8787675820495646,
"grad_norm": 19.454852119565878,
"learning_rate": 2.1715404500756875e-08,
"logits/chosen": -2.98136305809021,
"logits/rejected": -3.0059518814086914,
"logps/chosen": -1.090047836303711,
"logps/rejected": -1.203460693359375,
"loss": -0.0823,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.090047836303711,
"rewards/margins": 0.11341279745101929,
"rewards/rejected": -1.203460693359375,
"step": 410
},
{
"epoch": 0.8894842598794374,
"grad_norm": 27.119138813457198,
"learning_rate": 1.8056087839608546e-08,
"logits/chosen": -2.9424166679382324,
"logits/rejected": -2.9254748821258545,
"logps/chosen": -1.1260377168655396,
"logps/rejected": -1.3184921741485596,
"loss": -0.1086,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.1260377168655396,
"rewards/margins": 0.19245454668998718,
"rewards/rejected": -1.3184921741485596,
"step": 415
},
{
"epoch": 0.9002009377093101,
"grad_norm": 19.3060602023721,
"learning_rate": 1.4722715801513935e-08,
"logits/chosen": -2.9605276584625244,
"logits/rejected": -2.967475414276123,
"logps/chosen": -1.0344493389129639,
"logps/rejected": -1.0916310548782349,
"loss": -0.0802,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.0344493389129639,
"rewards/margins": 0.057181768119335175,
"rewards/rejected": -1.0916310548782349,
"step": 420
},
{
"epoch": 0.9109176155391828,
"grad_norm": 20.048601077954377,
"learning_rate": 1.1719972685717395e-08,
"logits/chosen": -2.970942258834839,
"logits/rejected": -2.9757351875305176,
"logps/chosen": -1.0653189420700073,
"logps/rejected": -1.3184258937835693,
"loss": -0.1043,
"rewards/accuracies": 0.65625,
"rewards/chosen": -1.0653189420700073,
"rewards/margins": 0.2531070411205292,
"rewards/rejected": -1.3184258937835693,
"step": 425
},
{
"epoch": 0.9216342933690556,
"grad_norm": 21.66418470338373,
"learning_rate": 9.05207816741485e-09,
"logits/chosen": -2.939943313598633,
"logits/rejected": -2.9273343086242676,
"logps/chosen": -1.2503459453582764,
"logps/rejected": -1.4009029865264893,
"loss": -0.1013,
"rewards/accuracies": 0.59375,
"rewards/chosen": -1.2503459453582764,
"rewards/margins": 0.15055687725543976,
"rewards/rejected": -1.4009029865264893,
"step": 430
},
{
"epoch": 0.9323509711989283,
"grad_norm": 21.774320932145653,
"learning_rate": 6.722781367956481e-09,
"logits/chosen": -3.0610954761505127,
"logits/rejected": -3.0484843254089355,
"logps/chosen": -1.0789310932159424,
"logps/rejected": -1.2742266654968262,
"loss": -0.1009,
"rewards/accuracies": 0.59375,
"rewards/chosen": -1.0789310932159424,
"rewards/margins": 0.19529545307159424,
"rewards/rejected": -1.2742266654968262,
"step": 435
},
{
"epoch": 0.943067649028801,
"grad_norm": 19.319233646112725,
"learning_rate": 4.735355586305861e-09,
"logits/chosen": -2.94116473197937,
"logits/rejected": -2.949489116668701,
"logps/chosen": -1.1311179399490356,
"logps/rejected": -1.3173315525054932,
"loss": -0.0993,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.1311179399490356,
"rewards/margins": 0.1862136423587799,
"rewards/rejected": -1.3173315525054932,
"step": 440
},
{
"epoch": 0.9537843268586738,
"grad_norm": 19.556600822562356,
"learning_rate": 3.0925936991598454e-09,
"logits/chosen": -2.9415977001190186,
"logits/rejected": -2.944352388381958,
"logps/chosen": -1.1981102228164673,
"logps/rejected": -1.329925775527954,
"loss": -0.0746,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -1.1981102228164673,
"rewards/margins": 0.1318153738975525,
"rewards/rejected": -1.329925775527954,
"step": 445
},
{
"epoch": 0.9645010046885466,
"grad_norm": 24.43813858171124,
"learning_rate": 1.7968042361930225e-09,
"logits/chosen": -2.958519697189331,
"logits/rejected": -2.9468445777893066,
"logps/chosen": -1.1451342105865479,
"logps/rejected": -1.265005111694336,
"loss": -0.0894,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -1.1451342105865479,
"rewards/margins": 0.11987074464559555,
"rewards/rejected": -1.265005111694336,
"step": 450
},
{
"epoch": 0.9752176825184193,
"grad_norm": 23.495311684580507,
"learning_rate": 8.498081359423759e-10,
"logits/chosen": -2.9030869007110596,
"logits/rejected": -2.932225465774536,
"logps/chosen": -1.1186686754226685,
"logps/rejected": -1.335756540298462,
"loss": -0.1008,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -1.1186686754226685,
"rewards/margins": 0.21708789467811584,
"rewards/rejected": -1.335756540298462,
"step": 455
},
{
"epoch": 0.9859343603482921,
"grad_norm": 20.358552883416746,
"learning_rate": 2.5293618689056774e-10,
"logits/chosen": -2.9370243549346924,
"logits/rejected": -2.953622341156006,
"logps/chosen": -1.1718332767486572,
"logps/rejected": -1.4025189876556396,
"loss": -0.0805,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -1.1718332767486572,
"rewards/margins": 0.230685755610466,
"rewards/rejected": -1.4025189876556396,
"step": 460
},
{
"epoch": 0.9966510381781648,
"grad_norm": 27.657325299971138,
"learning_rate": 7.02715734424153e-12,
"logits/chosen": -2.9429261684417725,
"logits/rejected": -2.951002359390259,
"logps/chosen": -1.2534146308898926,
"logps/rejected": -1.3702634572982788,
"loss": -0.0688,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -1.2534146308898926,
"rewards/margins": 0.11684894561767578,
"rewards/rejected": -1.3702634572982788,
"step": 465
},
{
"epoch": 0.9987943737441393,
"step": 466,
"total_flos": 0.0,
"train_loss": 0.0,
"train_runtime": 4.3106,
"train_samples_per_second": 13854.197,
"train_steps_per_second": 108.105
}
],
"logging_steps": 5,
"max_steps": 466,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}