|
{ |
|
"best_metric": 0.8505691885948181, |
|
"best_model_checkpoint": "saves/Mistral-7B-Instruct-v0.3/lora/orpo-salt-half/checkpoint-1500", |
|
"epoch": 2.9974597798475866, |
|
"eval_steps": 500, |
|
"global_step": 1770, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.01693480101608806, |
|
"grad_norm": 8.384780883789062, |
|
"learning_rate": 4.999614014035063e-06, |
|
"logits/chosen": -2.9335973262786865, |
|
"logits/rejected": -2.9718575477600098, |
|
"logps/chosen": -1.0935328006744385, |
|
"logps/rejected": -1.5320154428482056, |
|
"loss": 1.1602, |
|
"odds_ratio_loss": 0.6671324968338013, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.10935328900814056, |
|
"rewards/margins": 0.04384826496243477, |
|
"rewards/rejected": -0.15320155024528503, |
|
"sft_loss": 1.0935328006744385, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.03386960203217612, |
|
"grad_norm": 5.07673454284668, |
|
"learning_rate": 4.998440543386042e-06, |
|
"logits/chosen": -2.899705648422241, |
|
"logits/rejected": -2.8991312980651855, |
|
"logps/chosen": -1.0815098285675049, |
|
"logps/rejected": -1.5599451065063477, |
|
"loss": 1.1396, |
|
"odds_ratio_loss": 0.5810434818267822, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.10815098136663437, |
|
"rewards/margins": 0.04784352704882622, |
|
"rewards/rejected": -0.1559945046901703, |
|
"sft_loss": 1.0815098285675049, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05080440304826418, |
|
"grad_norm": 4.302060127258301, |
|
"learning_rate": 4.996479918381253e-06, |
|
"logits/chosen": -2.944632053375244, |
|
"logits/rejected": -2.975834846496582, |
|
"logps/chosen": -1.056970238685608, |
|
"logps/rejected": -1.281690001487732, |
|
"loss": 1.1267, |
|
"odds_ratio_loss": 0.6977876424789429, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.10569703578948975, |
|
"rewards/margins": 0.022471977397799492, |
|
"rewards/rejected": -0.1281690150499344, |
|
"sft_loss": 1.056970238685608, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.06773920406435224, |
|
"grad_norm": 3.221508026123047, |
|
"learning_rate": 4.993732756731818e-06, |
|
"logits/chosen": -2.925711154937744, |
|
"logits/rejected": -2.972975730895996, |
|
"logps/chosen": -0.8646748661994934, |
|
"logps/rejected": -1.3530806303024292, |
|
"loss": 0.9216, |
|
"odds_ratio_loss": 0.5692964792251587, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.08646748960018158, |
|
"rewards/margins": 0.04884058237075806, |
|
"rewards/rejected": -0.13530807197093964, |
|
"sft_loss": 0.8646748661994934, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.0846740050804403, |
|
"grad_norm": 11.413677215576172, |
|
"learning_rate": 4.9901999239537345e-06, |
|
"logits/chosen": -2.9809317588806152, |
|
"logits/rejected": -2.983664035797119, |
|
"logps/chosen": -0.9910508990287781, |
|
"logps/rejected": -1.344020962715149, |
|
"loss": 1.0558, |
|
"odds_ratio_loss": 0.6479853987693787, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.0991050973534584, |
|
"rewards/margins": 0.035297006368637085, |
|
"rewards/rejected": -0.1344021111726761, |
|
"sft_loss": 0.9910508990287781, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.10160880609652836, |
|
"grad_norm": 1.7445276975631714, |
|
"learning_rate": 4.985882533095186e-06, |
|
"logits/chosen": -2.9525933265686035, |
|
"logits/rejected": -2.955658435821533, |
|
"logps/chosen": -0.9359370470046997, |
|
"logps/rejected": -1.2753798961639404, |
|
"loss": 1.0023, |
|
"odds_ratio_loss": 0.6634014248847961, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.09359370172023773, |
|
"rewards/margins": 0.03394431248307228, |
|
"rewards/rejected": -0.12753799557685852, |
|
"sft_loss": 0.9359370470046997, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.11854360711261643, |
|
"grad_norm": 3.4384100437164307, |
|
"learning_rate": 4.9807819443858705e-06, |
|
"logits/chosen": -2.956145763397217, |
|
"logits/rejected": -2.9750401973724365, |
|
"logps/chosen": -0.8652639389038086, |
|
"logps/rejected": -1.2253761291503906, |
|
"loss": 0.9227, |
|
"odds_ratio_loss": 0.5748167634010315, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.08652639389038086, |
|
"rewards/margins": 0.0360112190246582, |
|
"rewards/rejected": -0.12253761291503906, |
|
"sft_loss": 0.8652639389038086, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.1354784081287045, |
|
"grad_norm": 2.6011013984680176, |
|
"learning_rate": 4.9748997648084404e-06, |
|
"logits/chosen": -2.927016496658325, |
|
"logits/rejected": -2.961350917816162, |
|
"logps/chosen": -0.9362251162528992, |
|
"logps/rejected": -1.1120071411132812, |
|
"loss": 1.0068, |
|
"odds_ratio_loss": 0.7062079906463623, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.09362251311540604, |
|
"rewards/margins": 0.017578203231096268, |
|
"rewards/rejected": -0.111200712621212, |
|
"sft_loss": 0.9362251162528992, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.15241320914479256, |
|
"grad_norm": 2.413475751876831, |
|
"learning_rate": 4.96823784759222e-06, |
|
"logits/chosen": -2.869145154953003, |
|
"logits/rejected": -2.916862964630127, |
|
"logps/chosen": -0.8781692385673523, |
|
"logps/rejected": -1.2194924354553223, |
|
"loss": 0.9334, |
|
"odds_ratio_loss": 0.552145779132843, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.08781692385673523, |
|
"rewards/margins": 0.034132327884435654, |
|
"rewards/rejected": -0.12194924056529999, |
|
"sft_loss": 0.8781692385673523, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.1693480101608806, |
|
"grad_norm": 1.9281786680221558, |
|
"learning_rate": 4.960798291629323e-06, |
|
"logits/chosen": -2.9603519439697266, |
|
"logits/rejected": -2.9720394611358643, |
|
"logps/chosen": -0.8728249669075012, |
|
"logps/rejected": -1.0510112047195435, |
|
"loss": 0.9387, |
|
"odds_ratio_loss": 0.658331036567688, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.08728249371051788, |
|
"rewards/margins": 0.017818626016378403, |
|
"rewards/rejected": -0.10510112345218658, |
|
"sft_loss": 0.8728249669075012, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.18628281117696868, |
|
"grad_norm": 2.8832967281341553, |
|
"learning_rate": 4.952583440813383e-06, |
|
"logits/chosen": -2.9769511222839355, |
|
"logits/rejected": -2.991858720779419, |
|
"logps/chosen": -0.9942096471786499, |
|
"logps/rejected": -1.25453519821167, |
|
"loss": 1.0667, |
|
"odds_ratio_loss": 0.7251978516578674, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.09942097961902618, |
|
"rewards/margins": 0.026032552123069763, |
|
"rewards/rejected": -0.12545353174209595, |
|
"sft_loss": 0.9942096471786499, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.20321761219305673, |
|
"grad_norm": 1.2281321287155151, |
|
"learning_rate": 4.943595883301086e-06, |
|
"logits/chosen": -2.9461495876312256, |
|
"logits/rejected": -2.988556385040283, |
|
"logps/chosen": -0.8882713317871094, |
|
"logps/rejected": -1.1232919692993164, |
|
"loss": 0.9522, |
|
"odds_ratio_loss": 0.6390038728713989, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.08882713317871094, |
|
"rewards/margins": 0.02350207045674324, |
|
"rewards/rejected": -0.11232920736074448, |
|
"sft_loss": 0.8882713317871094, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.2201524132091448, |
|
"grad_norm": 2.2648568153381348, |
|
"learning_rate": 4.933838450696757e-06, |
|
"logits/chosen": -2.9817254543304443, |
|
"logits/rejected": -3.0018677711486816, |
|
"logps/chosen": -0.8987566828727722, |
|
"logps/rejected": -1.0799884796142578, |
|
"loss": 0.9707, |
|
"odds_ratio_loss": 0.7198494672775269, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.08987566828727722, |
|
"rewards/margins": 0.018123187124729156, |
|
"rewards/rejected": -0.10799884796142578, |
|
"sft_loss": 0.8987566828727722, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.23708721422523285, |
|
"grad_norm": 3.667508840560913, |
|
"learning_rate": 4.923314217160234e-06, |
|
"logits/chosen": -2.988133192062378, |
|
"logits/rejected": -3.033163070678711, |
|
"logps/chosen": -0.9288301467895508, |
|
"logps/rejected": -1.1756031513214111, |
|
"loss": 0.9973, |
|
"odds_ratio_loss": 0.6846222877502441, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.09288302063941956, |
|
"rewards/margins": 0.024677302688360214, |
|
"rewards/rejected": -0.11756031215190887, |
|
"sft_loss": 0.9288301467895508, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.2540220152413209, |
|
"grad_norm": 1.4715845584869385, |
|
"learning_rate": 4.9120264984383285e-06, |
|
"logits/chosen": -2.997676134109497, |
|
"logits/rejected": -2.9935667514801025, |
|
"logps/chosen": -0.8239797353744507, |
|
"logps/rejected": -0.9983611106872559, |
|
"loss": 0.8869, |
|
"odds_ratio_loss": 0.6288361549377441, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.08239796757698059, |
|
"rewards/margins": 0.017438137903809547, |
|
"rewards/rejected": -0.09983611106872559, |
|
"sft_loss": 0.8239797353744507, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.270956816257409, |
|
"grad_norm": 1.3789633512496948, |
|
"learning_rate": 4.899978850820176e-06, |
|
"logits/chosen": -3.004080295562744, |
|
"logits/rejected": -3.0049595832824707, |
|
"logps/chosen": -0.9372480511665344, |
|
"logps/rejected": -1.147570252418518, |
|
"loss": 1.0008, |
|
"odds_ratio_loss": 0.635485053062439, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.0937248021364212, |
|
"rewards/margins": 0.021032210439443588, |
|
"rewards/rejected": -0.11475701630115509, |
|
"sft_loss": 0.9372480511665344, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.28789161727349705, |
|
"grad_norm": 2.1128056049346924, |
|
"learning_rate": 4.887175070016795e-06, |
|
"logits/chosen": -3.027259349822998, |
|
"logits/rejected": -3.0453531742095947, |
|
"logps/chosen": -0.7962668538093567, |
|
"logps/rejected": -1.0727083683013916, |
|
"loss": 0.8585, |
|
"odds_ratio_loss": 0.6222477555274963, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.07962668687105179, |
|
"rewards/margins": 0.027644142508506775, |
|
"rewards/rejected": -0.10727082192897797, |
|
"sft_loss": 0.7962668538093567, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.3048264182895851, |
|
"grad_norm": 4.011683464050293, |
|
"learning_rate": 4.873619189965217e-06, |
|
"logits/chosen": -2.973634719848633, |
|
"logits/rejected": -2.989631414413452, |
|
"logps/chosen": -0.8350585103034973, |
|
"logps/rejected": -1.051026463508606, |
|
"loss": 0.9, |
|
"odds_ratio_loss": 0.6493778824806213, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.08350586146116257, |
|
"rewards/margins": 0.021596785634756088, |
|
"rewards/rejected": -0.10510264337062836, |
|
"sft_loss": 0.8350585103034973, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.32176121930567314, |
|
"grad_norm": 1.2796882390975952, |
|
"learning_rate": 4.859315481557563e-06, |
|
"logits/chosen": -2.9721431732177734, |
|
"logits/rejected": -2.9849448204040527, |
|
"logps/chosen": -0.821880042552948, |
|
"logps/rejected": -1.0275905132293701, |
|
"loss": 0.8867, |
|
"odds_ratio_loss": 0.648421585559845, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.08218801021575928, |
|
"rewards/margins": 0.0205710306763649, |
|
"rewards/rejected": -0.10275904089212418, |
|
"sft_loss": 0.821880042552948, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.3386960203217612, |
|
"grad_norm": 1.8488657474517822, |
|
"learning_rate": 4.84426845129546e-06, |
|
"logits/chosen": -3.0222136974334717, |
|
"logits/rejected": -3.031564235687256, |
|
"logps/chosen": -0.8574220538139343, |
|
"logps/rejected": -0.9888635873794556, |
|
"loss": 0.9247, |
|
"odds_ratio_loss": 0.6725481748580933, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.08574221283197403, |
|
"rewards/margins": 0.013144141063094139, |
|
"rewards/rejected": -0.09888636320829391, |
|
"sft_loss": 0.8574220538139343, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.3556308213378493, |
|
"grad_norm": 1.5179060697555542, |
|
"learning_rate": 4.828482839870233e-06, |
|
"logits/chosen": -3.010849714279175, |
|
"logits/rejected": -3.0288453102111816, |
|
"logps/chosen": -0.8511277437210083, |
|
"logps/rejected": -0.9780422449111938, |
|
"loss": 0.9215, |
|
"odds_ratio_loss": 0.7037326693534851, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.08511276543140411, |
|
"rewards/margins": 0.012691453099250793, |
|
"rewards/rejected": -0.0978042259812355, |
|
"sft_loss": 0.8511277437210083, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.37256562235393736, |
|
"grad_norm": 1.2089393138885498, |
|
"learning_rate": 4.811963620669314e-06, |
|
"logits/chosen": -3.0581696033477783, |
|
"logits/rejected": -3.099684476852417, |
|
"logps/chosen": -0.8706866502761841, |
|
"logps/rejected": -1.0248219966888428, |
|
"loss": 0.9377, |
|
"odds_ratio_loss": 0.6702378988265991, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.08706866204738617, |
|
"rewards/margins": 0.015413539484143257, |
|
"rewards/rejected": -0.10248219966888428, |
|
"sft_loss": 0.8706866502761841, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.3895004233700254, |
|
"grad_norm": 2.3151497840881348, |
|
"learning_rate": 4.794715998209328e-06, |
|
"logits/chosen": -2.9359335899353027, |
|
"logits/rejected": -2.9507675170898438, |
|
"logps/chosen": -0.8457564115524292, |
|
"logps/rejected": -1.0168864727020264, |
|
"loss": 0.9116, |
|
"odds_ratio_loss": 0.6584563255310059, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.08457564562559128, |
|
"rewards/margins": 0.017113011330366135, |
|
"rewards/rejected": -0.10168864578008652, |
|
"sft_loss": 0.8457564115524292, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.40643522438611346, |
|
"grad_norm": 2.3710029125213623, |
|
"learning_rate": 4.7767454064963724e-06, |
|
"logits/chosen": -2.9894230365753174, |
|
"logits/rejected": -3.022892475128174, |
|
"logps/chosen": -0.8459660410881042, |
|
"logps/rejected": -1.015751838684082, |
|
"loss": 0.9112, |
|
"odds_ratio_loss": 0.6523627042770386, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.08459659665822983, |
|
"rewards/margins": 0.016978587955236435, |
|
"rewards/rejected": -0.10157518088817596, |
|
"sft_loss": 0.8459660410881042, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.42337002540220153, |
|
"grad_norm": 2.1464998722076416, |
|
"learning_rate": 4.758057507313987e-06, |
|
"logits/chosen": -3.024229049682617, |
|
"logits/rejected": -3.034865617752075, |
|
"logps/chosen": -0.8055688142776489, |
|
"logps/rejected": -1.001123070716858, |
|
"loss": 0.869, |
|
"odds_ratio_loss": 0.6340595483779907, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.08055686950683594, |
|
"rewards/margins": 0.019555436447262764, |
|
"rewards/rejected": -0.10011231899261475, |
|
"sft_loss": 0.8055688142776489, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.4403048264182896, |
|
"grad_norm": 3.3724582195281982, |
|
"learning_rate": 4.73865818843936e-06, |
|
"logits/chosen": -3.042086124420166, |
|
"logits/rejected": -3.0638880729675293, |
|
"logps/chosen": -0.912920355796814, |
|
"logps/rejected": -1.1694436073303223, |
|
"loss": 0.9795, |
|
"odds_ratio_loss": 0.6657846570014954, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.09129203855991364, |
|
"rewards/margins": 0.025652330368757248, |
|
"rewards/rejected": -0.11694437265396118, |
|
"sft_loss": 0.912920355796814, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.4572396274343776, |
|
"grad_norm": 1.6793408393859863, |
|
"learning_rate": 4.718553561788339e-06, |
|
"logits/chosen": -3.0312306880950928, |
|
"logits/rejected": -3.060295820236206, |
|
"logps/chosen": -0.836656928062439, |
|
"logps/rejected": -0.986899733543396, |
|
"loss": 0.9016, |
|
"odds_ratio_loss": 0.6491778492927551, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.08366570621728897, |
|
"rewards/margins": 0.015024276450276375, |
|
"rewards/rejected": -0.0986899808049202, |
|
"sft_loss": 0.836656928062439, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.4741744284504657, |
|
"grad_norm": 2.6274185180664062, |
|
"learning_rate": 4.697749961489822e-06, |
|
"logits/chosen": -3.061577558517456, |
|
"logits/rejected": -3.0702619552612305, |
|
"logps/chosen": -0.8848905563354492, |
|
"logps/rejected": -1.0645692348480225, |
|
"loss": 0.9531, |
|
"odds_ratio_loss": 0.6824666857719421, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.08848904818296432, |
|
"rewards/margins": 0.017967868596315384, |
|
"rewards/rejected": -0.10645692050457001, |
|
"sft_loss": 0.8848905563354492, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.4911092294665538, |
|
"grad_norm": 2.089768886566162, |
|
"learning_rate": 4.67625394189013e-06, |
|
"logits/chosen": -3.0883841514587402, |
|
"logits/rejected": -3.090940475463867, |
|
"logps/chosen": -0.7822630405426025, |
|
"logps/rejected": -1.0061594247817993, |
|
"loss": 0.8425, |
|
"odds_ratio_loss": 0.6027771830558777, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.07822629809379578, |
|
"rewards/margins": 0.022389648482203484, |
|
"rewards/rejected": -0.100615955889225, |
|
"sft_loss": 0.7822630405426025, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.5080440304826418, |
|
"grad_norm": 2.2402989864349365, |
|
"learning_rate": 4.654072275488016e-06, |
|
"logits/chosen": -3.1045384407043457, |
|
"logits/rejected": -3.1042561531066895, |
|
"logps/chosen": -0.7530822157859802, |
|
"logps/rejected": -0.9303094744682312, |
|
"loss": 0.8152, |
|
"odds_ratio_loss": 0.621616542339325, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07530822604894638, |
|
"rewards/margins": 0.017722725868225098, |
|
"rewards/rejected": -0.09303095191717148, |
|
"sft_loss": 0.7530822157859802, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.5249788314987299, |
|
"grad_norm": 7.859750747680664, |
|
"learning_rate": 4.631211950800925e-06, |
|
"logits/chosen": -3.0885701179504395, |
|
"logits/rejected": -3.1314892768859863, |
|
"logps/chosen": -0.8193107843399048, |
|
"logps/rejected": -1.0114855766296387, |
|
"loss": 0.8855, |
|
"odds_ratio_loss": 0.6622810959815979, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.08193108439445496, |
|
"rewards/margins": 0.01921747997403145, |
|
"rewards/rejected": -0.1011485680937767, |
|
"sft_loss": 0.8193107843399048, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.541913632514818, |
|
"grad_norm": 3.1328227519989014, |
|
"learning_rate": 4.6076801701632095e-06, |
|
"logits/chosen": -3.087184190750122, |
|
"logits/rejected": -3.1436970233917236, |
|
"logps/chosen": -0.801365852355957, |
|
"logps/rejected": -0.9046095609664917, |
|
"loss": 0.8686, |
|
"odds_ratio_loss": 0.6719350814819336, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.08013658970594406, |
|
"rewards/margins": 0.010324367322027683, |
|
"rewards/rejected": -0.09046096354722977, |
|
"sft_loss": 0.801365852355957, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.558848433530906, |
|
"grad_norm": 2.4766950607299805, |
|
"learning_rate": 4.583484347456972e-06, |
|
"logits/chosen": -3.0817341804504395, |
|
"logits/rejected": -3.114269256591797, |
|
"logps/chosen": -0.8467103242874146, |
|
"logps/rejected": -0.9794108271598816, |
|
"loss": 0.9136, |
|
"odds_ratio_loss": 0.6689561009407043, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.0846710279583931, |
|
"rewards/margins": 0.013270048424601555, |
|
"rewards/rejected": -0.0979410782456398, |
|
"sft_loss": 0.8467103242874146, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.5757832345469941, |
|
"grad_norm": 1.4625582695007324, |
|
"learning_rate": 4.55863210577626e-06, |
|
"logits/chosen": -3.123152256011963, |
|
"logits/rejected": -3.1361846923828125, |
|
"logps/chosen": -0.8682034611701965, |
|
"logps/rejected": -1.090419888496399, |
|
"loss": 0.9349, |
|
"odds_ratio_loss": 0.6673767566680908, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.08682034909725189, |
|
"rewards/margins": 0.022221634164452553, |
|
"rewards/rejected": -0.10904198884963989, |
|
"sft_loss": 0.8682034611701965, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.5927180355630821, |
|
"grad_norm": 1.6330158710479736, |
|
"learning_rate": 4.5331312750253465e-06, |
|
"logits/chosen": -3.079641103744507, |
|
"logits/rejected": -3.1221771240234375, |
|
"logps/chosen": -0.7776955962181091, |
|
"logps/rejected": -0.978277325630188, |
|
"loss": 0.8441, |
|
"odds_ratio_loss": 0.6638715267181396, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.07776956260204315, |
|
"rewards/margins": 0.02005816623568535, |
|
"rewards/rejected": -0.0978277251124382, |
|
"sft_loss": 0.7776955962181091, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.6096528365791702, |
|
"grad_norm": 1.9444369077682495, |
|
"learning_rate": 4.506989889451858e-06, |
|
"logits/chosen": -3.1103367805480957, |
|
"logits/rejected": -3.1434414386749268, |
|
"logps/chosen": -0.8445068597793579, |
|
"logps/rejected": -0.9869591593742371, |
|
"loss": 0.9108, |
|
"odds_ratio_loss": 0.662727952003479, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.08445067703723907, |
|
"rewards/margins": 0.014245236292481422, |
|
"rewards/rejected": -0.09869591891765594, |
|
"sft_loss": 0.8445068597793579, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.6265876375952583, |
|
"grad_norm": 2.251150369644165, |
|
"learning_rate": 4.480216185115512e-06, |
|
"logits/chosen": -3.0998404026031494, |
|
"logits/rejected": -3.087653398513794, |
|
"logps/chosen": -0.8066253662109375, |
|
"logps/rejected": -0.9696222543716431, |
|
"loss": 0.8704, |
|
"odds_ratio_loss": 0.6379208564758301, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08066253364086151, |
|
"rewards/margins": 0.016299689188599586, |
|
"rewards/rejected": -0.09696222841739655, |
|
"sft_loss": 0.8066253662109375, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.6435224386113463, |
|
"grad_norm": 1.8241935968399048, |
|
"learning_rate": 4.4528185972932856e-06, |
|
"logits/chosen": -3.019221782684326, |
|
"logits/rejected": -3.083693265914917, |
|
"logps/chosen": -0.8054075241088867, |
|
"logps/rejected": -1.0394740104675293, |
|
"loss": 0.8728, |
|
"odds_ratio_loss": 0.6744040846824646, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08054076135158539, |
|
"rewards/margins": 0.023406637832522392, |
|
"rewards/rejected": -0.10394741594791412, |
|
"sft_loss": 0.8054075241088867, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.6604572396274344, |
|
"grad_norm": 7.428930759429932, |
|
"learning_rate": 4.424805757821803e-06, |
|
"logits/chosen": -3.0259501934051514, |
|
"logits/rejected": -3.0692214965820312, |
|
"logps/chosen": -0.8603572845458984, |
|
"logps/rejected": -0.9688261151313782, |
|
"loss": 0.9297, |
|
"odds_ratio_loss": 0.6930958032608032, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.08603573590517044, |
|
"rewards/margins": 0.010846875607967377, |
|
"rewards/rejected": -0.09688261151313782, |
|
"sft_loss": 0.8603572845458984, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.6773920406435224, |
|
"grad_norm": 1.7724531888961792, |
|
"learning_rate": 4.396186492377812e-06, |
|
"logits/chosen": -3.049534797668457, |
|
"logits/rejected": -3.0946240425109863, |
|
"logps/chosen": -0.804740309715271, |
|
"logps/rejected": -1.0287699699401855, |
|
"loss": 0.8638, |
|
"odds_ratio_loss": 0.5903798341751099, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.08047403395175934, |
|
"rewards/margins": 0.02240295708179474, |
|
"rewards/rejected": -0.10287699848413467, |
|
"sft_loss": 0.804740309715271, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.6943268416596104, |
|
"grad_norm": 2.2618014812469482, |
|
"learning_rate": 4.366969817697578e-06, |
|
"logits/chosen": -3.0235114097595215, |
|
"logits/rejected": -3.083573341369629, |
|
"logps/chosen": -0.7958801984786987, |
|
"logps/rejected": -0.9750314950942993, |
|
"loss": 0.8637, |
|
"odds_ratio_loss": 0.6784581542015076, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.07958801835775375, |
|
"rewards/margins": 0.01791512593626976, |
|
"rewards/rejected": -0.09750314056873322, |
|
"sft_loss": 0.7958801984786987, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.7112616426756986, |
|
"grad_norm": 2.504657506942749, |
|
"learning_rate": 4.337164938736086e-06, |
|
"logits/chosen": -3.049118757247925, |
|
"logits/rejected": -3.05595326423645, |
|
"logps/chosen": -0.8237883448600769, |
|
"logps/rejected": -0.918091893196106, |
|
"loss": 0.8939, |
|
"odds_ratio_loss": 0.7009984254837036, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.08237884193658829, |
|
"rewards/margins": 0.00943034328520298, |
|
"rewards/rejected": -0.09180918335914612, |
|
"sft_loss": 0.8237883448600769, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.7281964436917866, |
|
"grad_norm": 1.2601932287216187, |
|
"learning_rate": 4.306781245766945e-06, |
|
"logits/chosen": -3.0380876064300537, |
|
"logits/rejected": -3.0915472507476807, |
|
"logps/chosen": -0.7913134098052979, |
|
"logps/rejected": -1.0263410806655884, |
|
"loss": 0.8569, |
|
"odds_ratio_loss": 0.6553653478622437, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.0791313424706459, |
|
"rewards/margins": 0.023502767086029053, |
|
"rewards/rejected": -0.10263410955667496, |
|
"sft_loss": 0.7913134098052979, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.7451312447078747, |
|
"grad_norm": 4.184472560882568, |
|
"learning_rate": 4.275828311423903e-06, |
|
"logits/chosen": -3.093045473098755, |
|
"logits/rejected": -3.094749927520752, |
|
"logps/chosen": -0.8900951147079468, |
|
"logps/rejected": -0.9482099413871765, |
|
"loss": 0.9611, |
|
"odds_ratio_loss": 0.7105128765106201, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.08900952339172363, |
|
"rewards/margins": 0.005811482667922974, |
|
"rewards/rejected": -0.09482099860906601, |
|
"sft_loss": 0.8900951147079468, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.7620660457239627, |
|
"grad_norm": 5.296163082122803, |
|
"learning_rate": 4.244315887684912e-06, |
|
"logits/chosen": -3.063938617706299, |
|
"logits/rejected": -3.0689697265625, |
|
"logps/chosen": -0.7619583010673523, |
|
"logps/rejected": -0.9321343302726746, |
|
"loss": 0.8277, |
|
"odds_ratio_loss": 0.6576007008552551, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07619582116603851, |
|
"rewards/margins": 0.017017606645822525, |
|
"rewards/rejected": -0.09321344643831253, |
|
"sft_loss": 0.7619583010673523, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.7790008467400508, |
|
"grad_norm": 1.0935202836990356, |
|
"learning_rate": 4.212253902799685e-06, |
|
"logits/chosen": -3.107138156890869, |
|
"logits/rejected": -3.1228718757629395, |
|
"logps/chosen": -0.828906238079071, |
|
"logps/rejected": -1.0854886770248413, |
|
"loss": 0.8936, |
|
"odds_ratio_loss": 0.6471681594848633, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.08289062976837158, |
|
"rewards/margins": 0.02565823495388031, |
|
"rewards/rejected": -0.10854886472225189, |
|
"sft_loss": 0.828906238079071, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.7959356477561389, |
|
"grad_norm": 2.112194538116455, |
|
"learning_rate": 4.179652458161718e-06, |
|
"logits/chosen": -3.0884511470794678, |
|
"logits/rejected": -3.086930751800537, |
|
"logps/chosen": -0.8770621418952942, |
|
"logps/rejected": -0.9670013189315796, |
|
"loss": 0.9476, |
|
"odds_ratio_loss": 0.7056951522827148, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.08770622313022614, |
|
"rewards/margins": 0.008993919007480145, |
|
"rewards/rejected": -0.09670013934373856, |
|
"sft_loss": 0.8770621418952942, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.8128704487722269, |
|
"grad_norm": 1.4803324937820435, |
|
"learning_rate": 4.146521825125765e-06, |
|
"logits/chosen": -3.1123909950256348, |
|
"logits/rejected": -3.127878189086914, |
|
"logps/chosen": -0.8552261590957642, |
|
"logps/rejected": -0.9841393232345581, |
|
"loss": 0.9226, |
|
"odds_ratio_loss": 0.6738197803497314, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.0855226144194603, |
|
"rewards/margins": 0.012891319580376148, |
|
"rewards/rejected": -0.09841393679380417, |
|
"sft_loss": 0.8552261590957642, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.8298052497883149, |
|
"grad_norm": 1.639560580253601, |
|
"learning_rate": 4.11287244177176e-06, |
|
"logits/chosen": -3.1404106616973877, |
|
"logits/rejected": -3.1268608570098877, |
|
"logps/chosen": -0.7970486283302307, |
|
"logps/rejected": -1.0359351634979248, |
|
"loss": 0.8576, |
|
"odds_ratio_loss": 0.6055651307106018, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07970486581325531, |
|
"rewards/margins": 0.02388865128159523, |
|
"rewards/rejected": -0.10359351336956024, |
|
"sft_loss": 0.7970486283302307, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.8467400508044031, |
|
"grad_norm": 7.098996162414551, |
|
"learning_rate": 4.078714909616215e-06, |
|
"logits/chosen": -3.1334495544433594, |
|
"logits/rejected": -3.127417802810669, |
|
"logps/chosen": -0.8159440755844116, |
|
"logps/rejected": -1.052442193031311, |
|
"loss": 0.8758, |
|
"odds_ratio_loss": 0.5985492467880249, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.08159441500902176, |
|
"rewards/margins": 0.02364981174468994, |
|
"rewards/rejected": -0.10524420440196991, |
|
"sft_loss": 0.8159440755844116, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.8467400508044031, |
|
"eval_logits/chosen": -3.1088695526123047, |
|
"eval_logits/rejected": -3.1276028156280518, |
|
"eval_logps/chosen": -0.8053962588310242, |
|
"eval_logps/rejected": -1.0085769891738892, |
|
"eval_loss": 0.8691067099571228, |
|
"eval_odds_ratio_loss": 0.6371051073074341, |
|
"eval_rewards/accuracies": 0.5704761743545532, |
|
"eval_rewards/chosen": -0.08053962886333466, |
|
"eval_rewards/margins": 0.02031807415187359, |
|
"eval_rewards/rejected": -0.1008576974272728, |
|
"eval_runtime": 194.1916, |
|
"eval_samples_per_second": 5.407, |
|
"eval_sft_loss": 0.8053962588310242, |
|
"eval_steps_per_second": 2.704, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.8636748518204911, |
|
"grad_norm": 5.1792216300964355, |
|
"learning_rate": 4.044059990272125e-06, |
|
"logits/chosen": -3.1320395469665527, |
|
"logits/rejected": -3.1584267616271973, |
|
"logps/chosen": -0.8421246409416199, |
|
"logps/rejected": -1.076226830482483, |
|
"loss": 0.9067, |
|
"odds_ratio_loss": 0.64554762840271, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08421246707439423, |
|
"rewards/margins": 0.02341020107269287, |
|
"rewards/rejected": -0.10762268304824829, |
|
"sft_loss": 0.8421246409416199, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.8806096528365792, |
|
"grad_norm": 2.52073073387146, |
|
"learning_rate": 4.0089186020584345e-06, |
|
"logits/chosen": -3.1114721298217773, |
|
"logits/rejected": -3.1454081535339355, |
|
"logps/chosen": -0.8953489065170288, |
|
"logps/rejected": -1.0218431949615479, |
|
"loss": 0.9597, |
|
"odds_ratio_loss": 0.6433413028717041, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.08953489363193512, |
|
"rewards/margins": 0.012649421580135822, |
|
"rewards/rejected": -0.10218431800603867, |
|
"sft_loss": 0.8953489065170288, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.8975444538526672, |
|
"grad_norm": 3.2529776096343994, |
|
"learning_rate": 3.973301816560124e-06, |
|
"logits/chosen": -3.1164424419403076, |
|
"logits/rejected": -3.0959460735321045, |
|
"logps/chosen": -0.8098430633544922, |
|
"logps/rejected": -1.0107625722885132, |
|
"loss": 0.871, |
|
"odds_ratio_loss": 0.6116595268249512, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.08098430931568146, |
|
"rewards/margins": 0.0200919471681118, |
|
"rewards/rejected": -0.10107626020908356, |
|
"sft_loss": 0.8098430633544922, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.9144792548687553, |
|
"grad_norm": 2.4143612384796143, |
|
"learning_rate": 3.937220855140021e-06, |
|
"logits/chosen": -3.1310832500457764, |
|
"logits/rejected": -3.1453144550323486, |
|
"logps/chosen": -0.8188160061836243, |
|
"logps/rejected": -0.9051122665405273, |
|
"loss": 0.8891, |
|
"odds_ratio_loss": 0.7029509544372559, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.08188159763813019, |
|
"rewards/margins": 0.008629636839032173, |
|
"rewards/rejected": -0.09051123261451721, |
|
"sft_loss": 0.8188160061836243, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.9314140558848434, |
|
"grad_norm": 1.3889896869659424, |
|
"learning_rate": 3.900687085403418e-06, |
|
"logits/chosen": -3.1416361331939697, |
|
"logits/rejected": -3.169236421585083, |
|
"logps/chosen": -0.8118529319763184, |
|
"logps/rejected": -0.8874411582946777, |
|
"loss": 0.8817, |
|
"odds_ratio_loss": 0.6982238292694092, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.08118529617786407, |
|
"rewards/margins": 0.007558824960142374, |
|
"rewards/rejected": -0.08874412626028061, |
|
"sft_loss": 0.8118529319763184, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.9483488569009314, |
|
"grad_norm": 2.055143356323242, |
|
"learning_rate": 3.863712017616614e-06, |
|
"logits/chosen": -3.131472110748291, |
|
"logits/rejected": -3.160679340362549, |
|
"logps/chosen": -0.8101698160171509, |
|
"logps/rejected": -0.9761570692062378, |
|
"loss": 0.8764, |
|
"odds_ratio_loss": 0.6618725657463074, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.08101697266101837, |
|
"rewards/margins": 0.016598742455244064, |
|
"rewards/rejected": -0.09761571884155273, |
|
"sft_loss": 0.8101698160171509, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.9652836579170194, |
|
"grad_norm": 14.660100936889648, |
|
"learning_rate": 3.826307301080504e-06, |
|
"logits/chosen": -3.0291128158569336, |
|
"logits/rejected": -3.0481247901916504, |
|
"logps/chosen": -0.8100768327713013, |
|
"logps/rejected": -1.162418007850647, |
|
"loss": 0.8764, |
|
"odds_ratio_loss": 0.6627860069274902, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.08100768178701401, |
|
"rewards/margins": 0.03523411601781845, |
|
"rewards/rejected": -0.11624179035425186, |
|
"sft_loss": 0.8100768327713013, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.9822184589331076, |
|
"grad_norm": 2.563514471054077, |
|
"learning_rate": 3.7884847204603775e-06, |
|
"logits/chosen": -3.1155200004577637, |
|
"logits/rejected": -3.15079665184021, |
|
"logps/chosen": -0.8116699457168579, |
|
"logps/rejected": -0.9339970350265503, |
|
"loss": 0.8812, |
|
"odds_ratio_loss": 0.6949950456619263, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.08116699010133743, |
|
"rewards/margins": 0.012232715263962746, |
|
"rewards/rejected": -0.09339970350265503, |
|
"sft_loss": 0.8116699457168579, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.9991532599491956, |
|
"grad_norm": 2.7657859325408936, |
|
"learning_rate": 3.750256192073058e-06, |
|
"logits/chosen": -3.172693967819214, |
|
"logits/rejected": -3.1799886226654053, |
|
"logps/chosen": -0.9269296526908875, |
|
"logps/rejected": -0.9760646820068359, |
|
"loss": 1.0009, |
|
"odds_ratio_loss": 0.7394477128982544, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.09269297122955322, |
|
"rewards/margins": 0.004913498647511005, |
|
"rewards/rejected": -0.09760646522045135, |
|
"sft_loss": 0.9269296526908875, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 1.0160880609652836, |
|
"grad_norm": 7.485799789428711, |
|
"learning_rate": 3.7116337601325715e-06, |
|
"logits/chosen": -3.1055843830108643, |
|
"logits/rejected": -3.129669189453125, |
|
"logps/chosen": -0.7523837089538574, |
|
"logps/rejected": -0.8988542556762695, |
|
"loss": 0.8166, |
|
"odds_ratio_loss": 0.6426426768302917, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.07523836940526962, |
|
"rewards/margins": 0.014647054485976696, |
|
"rewards/rejected": -0.08988542854785919, |
|
"sft_loss": 0.7523837089538574, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 1.0330228619813717, |
|
"grad_norm": 2.2944157123565674, |
|
"learning_rate": 3.6726295929555154e-06, |
|
"logits/chosen": -3.077573776245117, |
|
"logits/rejected": -3.11928391456604, |
|
"logps/chosen": -0.7445582747459412, |
|
"logps/rejected": -0.9014002680778503, |
|
"loss": 0.8117, |
|
"odds_ratio_loss": 0.6712278723716736, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.07445583492517471, |
|
"rewards/margins": 0.015684202313423157, |
|
"rewards/rejected": -0.09014002978801727, |
|
"sft_loss": 0.7445582747459412, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 1.0499576629974599, |
|
"grad_norm": 2.4759469032287598, |
|
"learning_rate": 3.6332559791273307e-06, |
|
"logits/chosen": -3.070753812789917, |
|
"logits/rejected": -3.133881092071533, |
|
"logps/chosen": -0.7787492871284485, |
|
"logps/rejected": -0.9435569643974304, |
|
"loss": 0.843, |
|
"odds_ratio_loss": 0.6423131823539734, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.07787492126226425, |
|
"rewards/margins": 0.016480756923556328, |
|
"rewards/rejected": -0.09435568749904633, |
|
"sft_loss": 0.7787492871284485, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 1.0668924640135478, |
|
"grad_norm": 2.397857666015625, |
|
"learning_rate": 3.593525323630681e-06, |
|
"logits/chosen": -3.0695629119873047, |
|
"logits/rejected": -3.0961527824401855, |
|
"logps/chosen": -0.8072765469551086, |
|
"logps/rejected": -0.9367998838424683, |
|
"loss": 0.8715, |
|
"odds_ratio_loss": 0.6417396664619446, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.08072765171527863, |
|
"rewards/margins": 0.012952342629432678, |
|
"rewards/rejected": -0.0936800017952919, |
|
"sft_loss": 0.8072765469551086, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 1.083827265029636, |
|
"grad_norm": 2.2752127647399902, |
|
"learning_rate": 3.5534501439371615e-06, |
|
"logits/chosen": -3.1115057468414307, |
|
"logits/rejected": -3.1277289390563965, |
|
"logps/chosen": -0.7920553088188171, |
|
"logps/rejected": -0.9753093719482422, |
|
"loss": 0.8548, |
|
"odds_ratio_loss": 0.6271573901176453, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07920553535223007, |
|
"rewards/margins": 0.018325407058000565, |
|
"rewards/rejected": -0.09753094613552094, |
|
"sft_loss": 0.7920553088188171, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 1.100762066045724, |
|
"grad_norm": 1.311848521232605, |
|
"learning_rate": 3.5130430660635633e-06, |
|
"logits/chosen": -3.1020348072052, |
|
"logits/rejected": -3.1430366039276123, |
|
"logps/chosen": -0.8032411336898804, |
|
"logps/rejected": -0.9853434562683105, |
|
"loss": 0.864, |
|
"odds_ratio_loss": 0.6071646809577942, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.08032412827014923, |
|
"rewards/margins": 0.018210221081972122, |
|
"rewards/rejected": -0.09853433817625046, |
|
"sft_loss": 0.8032411336898804, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 1.117696867061812, |
|
"grad_norm": 1.6412379741668701, |
|
"learning_rate": 3.4723168205939444e-06, |
|
"logits/chosen": -3.0930168628692627, |
|
"logits/rejected": -3.1322126388549805, |
|
"logps/chosen": -0.7815112471580505, |
|
"logps/rejected": -0.9042918086051941, |
|
"loss": 0.8483, |
|
"odds_ratio_loss": 0.6679799556732178, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.07815112918615341, |
|
"rewards/margins": 0.012278061360120773, |
|
"rewards/rejected": -0.09042918682098389, |
|
"sft_loss": 0.7815112471580505, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 1.1346316680779, |
|
"grad_norm": 2.1591804027557373, |
|
"learning_rate": 3.431284238668754e-06, |
|
"logits/chosen": -3.062398910522461, |
|
"logits/rejected": -3.1151247024536133, |
|
"logps/chosen": -0.8405235409736633, |
|
"logps/rejected": -1.0043154954910278, |
|
"loss": 0.9059, |
|
"odds_ratio_loss": 0.6538293957710266, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.08405234664678574, |
|
"rewards/margins": 0.01637919992208481, |
|
"rewards/rejected": -0.10043156147003174, |
|
"sft_loss": 0.8405235409736633, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 1.1515664690939882, |
|
"grad_norm": 4.811310291290283, |
|
"learning_rate": 3.389958247942274e-06, |
|
"logits/chosen": -3.057506561279297, |
|
"logits/rejected": -3.1269962787628174, |
|
"logps/chosen": -0.8411120176315308, |
|
"logps/rejected": -1.1102626323699951, |
|
"loss": 0.9091, |
|
"odds_ratio_loss": 0.6803519129753113, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.08411119878292084, |
|
"rewards/margins": 0.026915064081549644, |
|
"rewards/rejected": -0.11102626472711563, |
|
"sft_loss": 0.8411120176315308, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 1.168501270110076, |
|
"grad_norm": 2.754120111465454, |
|
"learning_rate": 3.3483518685096588e-06, |
|
"logits/chosen": -3.08880615234375, |
|
"logits/rejected": -3.109083890914917, |
|
"logps/chosen": -0.8396803140640259, |
|
"logps/rejected": -0.9775651693344116, |
|
"loss": 0.9066, |
|
"odds_ratio_loss": 0.6687448620796204, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.08396803587675095, |
|
"rewards/margins": 0.013788496144115925, |
|
"rewards/rejected": -0.097756527364254, |
|
"sft_loss": 0.8396803140640259, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 1.1854360711261642, |
|
"grad_norm": 1.9642765522003174, |
|
"learning_rate": 3.306478208804839e-06, |
|
"logits/chosen": -3.0659680366516113, |
|
"logits/rejected": -3.1026992797851562, |
|
"logps/chosen": -0.7790535092353821, |
|
"logps/rejected": -0.9482936859130859, |
|
"loss": 0.8451, |
|
"odds_ratio_loss": 0.659988284111023, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07790535688400269, |
|
"rewards/margins": 0.016924021765589714, |
|
"rewards/rejected": -0.09482936561107635, |
|
"sft_loss": 0.7790535092353821, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 1.2023708721422524, |
|
"grad_norm": 2.6194725036621094, |
|
"learning_rate": 3.264350461470608e-06, |
|
"logits/chosen": -3.0373668670654297, |
|
"logits/rejected": -3.0688533782958984, |
|
"logps/chosen": -0.7216525077819824, |
|
"logps/rejected": -1.0178143978118896, |
|
"loss": 0.7814, |
|
"odds_ratio_loss": 0.597113311290741, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.07216525077819824, |
|
"rewards/margins": 0.02961619198322296, |
|
"rewards/rejected": -0.1017814427614212, |
|
"sft_loss": 0.7216525077819824, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 1.2193056731583405, |
|
"grad_norm": 2.312389373779297, |
|
"learning_rate": 3.2219818992021685e-06, |
|
"logits/chosen": -3.03488826751709, |
|
"logits/rejected": -3.087043285369873, |
|
"logps/chosen": -0.7246071696281433, |
|
"logps/rejected": -1.047911286354065, |
|
"loss": 0.7828, |
|
"odds_ratio_loss": 0.5823417901992798, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.07246071845293045, |
|
"rewards/margins": 0.03233041241765022, |
|
"rewards/rejected": -0.10479112714529037, |
|
"sft_loss": 0.7246071696281433, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 1.2362404741744284, |
|
"grad_norm": 7.234289169311523, |
|
"learning_rate": 3.1793858705654595e-06, |
|
"logits/chosen": -3.0948994159698486, |
|
"logits/rejected": -3.1080453395843506, |
|
"logps/chosen": -0.7130419611930847, |
|
"logps/rejected": -0.9181682467460632, |
|
"loss": 0.7733, |
|
"odds_ratio_loss": 0.6023129820823669, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.07130420207977295, |
|
"rewards/margins": 0.020512625575065613, |
|
"rewards/rejected": -0.09181682765483856, |
|
"sft_loss": 0.7130419611930847, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 1.2531752751905165, |
|
"grad_norm": 2.101661443710327, |
|
"learning_rate": 3.1365757957915787e-06, |
|
"logits/chosen": -3.0874876976013184, |
|
"logits/rejected": -3.123832941055298, |
|
"logps/chosen": -0.8178297877311707, |
|
"logps/rejected": -0.9593319892883301, |
|
"loss": 0.8832, |
|
"odds_ratio_loss": 0.6534532308578491, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.0817829817533493, |
|
"rewards/margins": 0.01415021438151598, |
|
"rewards/rejected": -0.09593319892883301, |
|
"sft_loss": 0.8178297877311707, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 1.2701100762066047, |
|
"grad_norm": 3.2394182682037354, |
|
"learning_rate": 3.093565162548633e-06, |
|
"logits/chosen": -3.0456290245056152, |
|
"logits/rejected": -3.0783424377441406, |
|
"logps/chosen": -0.8610566854476929, |
|
"logps/rejected": -1.0578378438949585, |
|
"loss": 0.9282, |
|
"odds_ratio_loss": 0.6714938879013062, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.08610567450523376, |
|
"rewards/margins": 0.019678115844726562, |
|
"rewards/rejected": -0.10578378289937973, |
|
"sft_loss": 0.8610566854476929, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 1.2870448772226926, |
|
"grad_norm": 3.4462602138519287, |
|
"learning_rate": 3.0503675216923294e-06, |
|
"logits/chosen": -3.112204074859619, |
|
"logits/rejected": -3.1115825176239014, |
|
"logps/chosen": -0.7481369376182556, |
|
"logps/rejected": -0.9443623423576355, |
|
"loss": 0.8093, |
|
"odds_ratio_loss": 0.6117558479309082, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.07481369376182556, |
|
"rewards/margins": 0.019622545689344406, |
|
"rewards/rejected": -0.09443624317646027, |
|
"sft_loss": 0.7481369376182556, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 1.3039796782387807, |
|
"grad_norm": 2.5497653484344482, |
|
"learning_rate": 3.0069964829966748e-06, |
|
"logits/chosen": -3.083761692047119, |
|
"logits/rejected": -3.1253674030303955, |
|
"logps/chosen": -0.7777606248855591, |
|
"logps/rejected": -0.9011049270629883, |
|
"loss": 0.844, |
|
"odds_ratio_loss": 0.6621277928352356, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.07777605950832367, |
|
"rewards/margins": 0.01233443059027195, |
|
"rewards/rejected": -0.09011048078536987, |
|
"sft_loss": 0.7777606248855591, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 1.3209144792548688, |
|
"grad_norm": 2.1115994453430176, |
|
"learning_rate": 2.963465710866094e-06, |
|
"logits/chosen": -3.098053455352783, |
|
"logits/rejected": -3.1173479557037354, |
|
"logps/chosen": -0.7589127421379089, |
|
"logps/rejected": -1.0739343166351318, |
|
"loss": 0.8183, |
|
"odds_ratio_loss": 0.5940018892288208, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07589127868413925, |
|
"rewards/margins": 0.03150214999914169, |
|
"rewards/rejected": -0.10739342123270035, |
|
"sft_loss": 0.7589127421379089, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 1.337849280270957, |
|
"grad_norm": 7.320650577545166, |
|
"learning_rate": 2.919788920030357e-06, |
|
"logits/chosen": -3.143812894821167, |
|
"logits/rejected": -3.1576976776123047, |
|
"logps/chosen": -0.8158448934555054, |
|
"logps/rejected": -0.9654959440231323, |
|
"loss": 0.8826, |
|
"odds_ratio_loss": 0.6670835614204407, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.08158449828624725, |
|
"rewards/margins": 0.014965096488595009, |
|
"rewards/rejected": -0.09654959291219711, |
|
"sft_loss": 0.8158448934555054, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 1.3547840812870449, |
|
"grad_norm": 1.3826831579208374, |
|
"learning_rate": 2.8759798712236303e-06, |
|
"logits/chosen": -3.1377153396606445, |
|
"logits/rejected": -3.138549566268921, |
|
"logps/chosen": -0.7528073191642761, |
|
"logps/rejected": -1.0746774673461914, |
|
"loss": 0.8173, |
|
"odds_ratio_loss": 0.6448089480400085, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07528072595596313, |
|
"rewards/margins": 0.032187022268772125, |
|
"rewards/rejected": -0.10746775567531586, |
|
"sft_loss": 0.7528073191642761, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 1.371718882303133, |
|
"grad_norm": 2.1327664852142334, |
|
"learning_rate": 2.8320523668490507e-06, |
|
"logits/chosen": -3.095376968383789, |
|
"logits/rejected": -3.129220485687256, |
|
"logps/chosen": -0.8166864514350891, |
|
"logps/rejected": -0.9957429766654968, |
|
"loss": 0.8841, |
|
"odds_ratio_loss": 0.6738199591636658, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.08166865259408951, |
|
"rewards/margins": 0.01790565252304077, |
|
"rewards/rejected": -0.09957430511713028, |
|
"sft_loss": 0.8166864514350891, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 1.388653683319221, |
|
"grad_norm": 2.5101046562194824, |
|
"learning_rate": 2.7880202466301597e-06, |
|
"logits/chosen": -3.075023651123047, |
|
"logits/rejected": -3.1098039150238037, |
|
"logps/chosen": -0.8058909177780151, |
|
"logps/rejected": -0.9401804208755493, |
|
"loss": 0.8735, |
|
"odds_ratio_loss": 0.6758350133895874, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.08058909326791763, |
|
"rewards/margins": 0.013428950682282448, |
|
"rewards/rejected": -0.09401804953813553, |
|
"sft_loss": 0.8058909177780151, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 1.405588484335309, |
|
"grad_norm": 2.202185869216919, |
|
"learning_rate": 2.7438973832505854e-06, |
|
"logits/chosen": -3.060824155807495, |
|
"logits/rejected": -3.0545971393585205, |
|
"logps/chosen": -0.7594717741012573, |
|
"logps/rejected": -0.9759091138839722, |
|
"loss": 0.8235, |
|
"odds_ratio_loss": 0.6401799321174622, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.07594718039035797, |
|
"rewards/margins": 0.021643735468387604, |
|
"rewards/rejected": -0.09759090840816498, |
|
"sft_loss": 0.7594717741012573, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 1.4225232853513972, |
|
"grad_norm": 5.100254058837891, |
|
"learning_rate": 2.699697677983341e-06, |
|
"logits/chosen": -3.1682240962982178, |
|
"logits/rejected": -3.182861804962158, |
|
"logps/chosen": -0.8106738924980164, |
|
"logps/rejected": -0.8703418970108032, |
|
"loss": 0.8831, |
|
"odds_ratio_loss": 0.7241007685661316, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.08106739819049835, |
|
"rewards/margins": 0.0059667956084012985, |
|
"rewards/rejected": -0.0870341882109642, |
|
"sft_loss": 0.8106738924980164, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 1.4394580863674853, |
|
"grad_norm": 7.667366981506348, |
|
"learning_rate": 2.6554350563111115e-06, |
|
"logits/chosen": -3.0851314067840576, |
|
"logits/rejected": -3.116276502609253, |
|
"logps/chosen": -0.8307350873947144, |
|
"logps/rejected": -0.9147516489028931, |
|
"loss": 0.9054, |
|
"odds_ratio_loss": 0.7461589574813843, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.08307350426912308, |
|
"rewards/margins": 0.0084016602486372, |
|
"rewards/rejected": -0.09147517383098602, |
|
"sft_loss": 0.8307350873947144, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 1.4563928873835732, |
|
"grad_norm": 6.640262126922607, |
|
"learning_rate": 2.611123463538913e-06, |
|
"logits/chosen": -3.0749902725219727, |
|
"logits/rejected": -3.0911591053009033, |
|
"logps/chosen": -0.7142345309257507, |
|
"logps/rejected": -0.9272225499153137, |
|
"loss": 0.7775, |
|
"odds_ratio_loss": 0.6324297189712524, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.07142344862222672, |
|
"rewards/margins": 0.02129879966378212, |
|
"rewards/rejected": -0.09272225201129913, |
|
"sft_loss": 0.7142345309257507, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 1.4733276883996613, |
|
"grad_norm": 8.2167329788208, |
|
"learning_rate": 2.566776860400514e-06, |
|
"logits/chosen": -3.1082987785339355, |
|
"logits/rejected": -3.1257712841033936, |
|
"logps/chosen": -0.8350755572319031, |
|
"logps/rejected": -1.056910753250122, |
|
"loss": 0.898, |
|
"odds_ratio_loss": 0.6290403604507446, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.08350756019353867, |
|
"rewards/margins": 0.022183528169989586, |
|
"rewards/rejected": -0.10569107532501221, |
|
"sft_loss": 0.8350755572319031, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 1.4902624894157492, |
|
"grad_norm": 1.4808557033538818, |
|
"learning_rate": 2.522409218659989e-06, |
|
"logits/chosen": -3.143266201019287, |
|
"logits/rejected": -3.170633316040039, |
|
"logps/chosen": -0.7854605913162231, |
|
"logps/rejected": -0.9068530797958374, |
|
"loss": 0.8512, |
|
"odds_ratio_loss": 0.6573347449302673, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.07854606211185455, |
|
"rewards/margins": 0.012139257043600082, |
|
"rewards/rejected": -0.09068530797958374, |
|
"sft_loss": 0.7854605913162231, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 1.5071972904318374, |
|
"grad_norm": 3.6185755729675293, |
|
"learning_rate": 2.4780345167097976e-06, |
|
"logits/chosen": -3.0972650051116943, |
|
"logits/rejected": -3.0835044384002686, |
|
"logps/chosen": -0.7852433919906616, |
|
"logps/rejected": -1.0855656862258911, |
|
"loss": 0.8459, |
|
"odds_ratio_loss": 0.6065645813941956, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.07852433621883392, |
|
"rewards/margins": 0.030032237991690636, |
|
"rewards/rejected": -0.1085565835237503, |
|
"sft_loss": 0.7852433919906616, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 1.5241320914479255, |
|
"grad_norm": 4.639472961425781, |
|
"learning_rate": 2.4336667351667747e-06, |
|
"logits/chosen": -3.114197015762329, |
|
"logits/rejected": -3.124145269393921, |
|
"logps/chosen": -0.8203206062316895, |
|
"logps/rejected": -1.0489810705184937, |
|
"loss": 0.8792, |
|
"odds_ratio_loss": 0.5887311697006226, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.08203206956386566, |
|
"rewards/margins": 0.022866051644086838, |
|
"rewards/rejected": -0.1048981174826622, |
|
"sft_loss": 0.8203206062316895, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 1.5410668924640136, |
|
"grad_norm": 3.153228282928467, |
|
"learning_rate": 2.3893198524674264e-06, |
|
"logits/chosen": -3.086516857147217, |
|
"logits/rejected": -3.104675769805908, |
|
"logps/chosen": -0.7846948504447937, |
|
"logps/rejected": -0.9825633764266968, |
|
"loss": 0.8478, |
|
"odds_ratio_loss": 0.6309666037559509, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.07846949249505997, |
|
"rewards/margins": 0.019786860793828964, |
|
"rewards/rejected": -0.09825634956359863, |
|
"sft_loss": 0.7846948504447937, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 1.5580016934801018, |
|
"grad_norm": 3.532456874847412, |
|
"learning_rate": 2.345007840463904e-06, |
|
"logits/chosen": -3.0608856678009033, |
|
"logits/rejected": -3.0965914726257324, |
|
"logps/chosen": -0.8089407682418823, |
|
"logps/rejected": -0.9399329423904419, |
|
"loss": 0.8759, |
|
"odds_ratio_loss": 0.6692665815353394, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.08089407533407211, |
|
"rewards/margins": 0.013099217787384987, |
|
"rewards/rejected": -0.09399329125881195, |
|
"sft_loss": 0.8089407682418823, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 1.5749364944961897, |
|
"grad_norm": 3.778723955154419, |
|
"learning_rate": 2.3007446600220572e-06, |
|
"logits/chosen": -3.122668504714966, |
|
"logits/rejected": -3.1052489280700684, |
|
"logps/chosen": -0.799084484577179, |
|
"logps/rejected": -0.9807415008544922, |
|
"loss": 0.8639, |
|
"odds_ratio_loss": 0.6483135223388672, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07990844547748566, |
|
"rewards/margins": 0.018165703862905502, |
|
"rewards/rejected": -0.09807415306568146, |
|
"sft_loss": 0.799084484577179, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 1.5918712955122776, |
|
"grad_norm": 3.475341558456421, |
|
"learning_rate": 2.2565442566229507e-06, |
|
"logits/chosen": -3.0740559101104736, |
|
"logits/rejected": -3.106633424758911, |
|
"logps/chosen": -0.7921947240829468, |
|
"logps/rejected": -0.8962046504020691, |
|
"loss": 0.862, |
|
"odds_ratio_loss": 0.6985131502151489, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.07921947538852692, |
|
"rewards/margins": 0.01040099561214447, |
|
"rewards/rejected": -0.08962046355009079, |
|
"sft_loss": 0.7921947240829468, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 1.6088060965283657, |
|
"grad_norm": 2.1668930053710938, |
|
"learning_rate": 2.2124205559692195e-06, |
|
"logits/chosen": -3.064317226409912, |
|
"logits/rejected": -3.1167845726013184, |
|
"logps/chosen": -0.7817971706390381, |
|
"logps/rejected": -1.0080362558364868, |
|
"loss": 0.8399, |
|
"odds_ratio_loss": 0.5807359218597412, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.0781797245144844, |
|
"rewards/margins": 0.02262391336262226, |
|
"rewards/rejected": -0.10080362856388092, |
|
"sft_loss": 0.7817971706390381, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 1.6257408975444538, |
|
"grad_norm": 1.8939071893692017, |
|
"learning_rate": 2.168387459597666e-06, |
|
"logits/chosen": -3.092590808868408, |
|
"logits/rejected": -3.138596773147583, |
|
"logps/chosen": -0.7956336140632629, |
|
"logps/rejected": -1.0347424745559692, |
|
"loss": 0.8565, |
|
"odds_ratio_loss": 0.6086810827255249, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.07956336438655853, |
|
"rewards/margins": 0.02391088381409645, |
|
"rewards/rejected": -0.10347424447536469, |
|
"sft_loss": 0.7956336140632629, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 1.642675698560542, |
|
"grad_norm": 2.1293139457702637, |
|
"learning_rate": 2.1244588404994648e-06, |
|
"logits/chosen": -3.0621867179870605, |
|
"logits/rejected": -3.0614800453186035, |
|
"logps/chosen": -0.7734932899475098, |
|
"logps/rejected": -0.9287079572677612, |
|
"loss": 0.8408, |
|
"odds_ratio_loss": 0.6727171540260315, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.07734932750463486, |
|
"rewards/margins": 0.015521461144089699, |
|
"rewards/rejected": -0.0928707867860794, |
|
"sft_loss": 0.7734932899475098, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 1.65961049957663, |
|
"grad_norm": 5.466368675231934, |
|
"learning_rate": 2.08064853874936e-06, |
|
"logits/chosen": -3.0892724990844727, |
|
"logits/rejected": -3.1403141021728516, |
|
"logps/chosen": -0.8142075538635254, |
|
"logps/rejected": -0.9884878396987915, |
|
"loss": 0.8786, |
|
"odds_ratio_loss": 0.6434410810470581, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.08142076432704926, |
|
"rewards/margins": 0.017428018152713776, |
|
"rewards/rejected": -0.09884877502918243, |
|
"sft_loss": 0.8142075538635254, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 1.676545300592718, |
|
"grad_norm": 12.251886367797852, |
|
"learning_rate": 2.0369703571452387e-06, |
|
"logits/chosen": -3.055989980697632, |
|
"logits/rejected": -3.0450901985168457, |
|
"logps/chosen": -0.716150164604187, |
|
"logps/rejected": -0.9780189394950867, |
|
"loss": 0.7743, |
|
"odds_ratio_loss": 0.5815138816833496, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.07161502540111542, |
|
"rewards/margins": 0.026186879724264145, |
|
"rewards/rejected": -0.09780190140008926, |
|
"sft_loss": 0.716150164604187, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 1.6934801016088061, |
|
"grad_norm": 3.6831836700439453, |
|
"learning_rate": 1.993438056859441e-06, |
|
"logits/chosen": -3.1155529022216797, |
|
"logits/rejected": -3.1060287952423096, |
|
"logps/chosen": -0.7484423518180847, |
|
"logps/rejected": -0.929480254650116, |
|
"loss": 0.8098, |
|
"odds_ratio_loss": 0.6138982772827148, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07484424859285355, |
|
"rewards/margins": 0.018103793263435364, |
|
"rewards/rejected": -0.09294802695512772, |
|
"sft_loss": 0.7484423518180847, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.6934801016088061, |
|
"eval_logits/chosen": -3.0966453552246094, |
|
"eval_logits/rejected": -3.117032051086426, |
|
"eval_logps/chosen": -0.7911127805709839, |
|
"eval_logps/rejected": -0.9985377192497253, |
|
"eval_loss": 0.8548597693443298, |
|
"eval_odds_ratio_loss": 0.6374707221984863, |
|
"eval_rewards/accuracies": 0.5676190257072449, |
|
"eval_rewards/chosen": -0.07911127805709839, |
|
"eval_rewards/margins": 0.020742492750287056, |
|
"eval_rewards/rejected": -0.0998537689447403, |
|
"eval_runtime": 195.0995, |
|
"eval_samples_per_second": 5.382, |
|
"eval_sft_loss": 0.7911127805709839, |
|
"eval_steps_per_second": 2.691, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.710414902624894, |
|
"grad_norm": 2.5681636333465576, |
|
"learning_rate": 1.9500653531031917e-06, |
|
"logits/chosen": -3.116891384124756, |
|
"logits/rejected": -3.1181325912475586, |
|
"logps/chosen": -0.8085691332817078, |
|
"logps/rejected": -1.0325000286102295, |
|
"loss": 0.8768, |
|
"odds_ratio_loss": 0.6825646162033081, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.08085691183805466, |
|
"rewards/margins": 0.022393101826310158, |
|
"rewards/rejected": -0.10325001180171967, |
|
"sft_loss": 0.8085691332817078, |
|
"step": 1010 |
|
}, |
|
{ |
|
"epoch": 1.7273497036409822, |
|
"grad_norm": 2.5795137882232666, |
|
"learning_rate": 1.9068659108055117e-06, |
|
"logits/chosen": -3.1321749687194824, |
|
"logits/rejected": -3.155836582183838, |
|
"logps/chosen": -0.7755477428436279, |
|
"logps/rejected": -0.9410096406936646, |
|
"loss": 0.8406, |
|
"odds_ratio_loss": 0.6507992744445801, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.07755477726459503, |
|
"rewards/margins": 0.016546186059713364, |
|
"rewards/rejected": -0.0941009670495987, |
|
"sft_loss": 0.7755477428436279, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 1.7442845046570703, |
|
"grad_norm": 2.360201835632324, |
|
"learning_rate": 1.863853340307962e-06, |
|
"logits/chosen": -3.078691005706787, |
|
"logits/rejected": -3.1044058799743652, |
|
"logps/chosen": -0.683570146560669, |
|
"logps/rejected": -0.9945551753044128, |
|
"loss": 0.7414, |
|
"odds_ratio_loss": 0.5782071352005005, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.06835701316595078, |
|
"rewards/margins": 0.03109849989414215, |
|
"rewards/rejected": -0.09945552051067352, |
|
"sft_loss": 0.683570146560669, |
|
"step": 1030 |
|
}, |
|
{ |
|
"epoch": 1.7612193056731584, |
|
"grad_norm": 2.180799961090088, |
|
"learning_rate": 1.8210411930766019e-06, |
|
"logits/chosen": -3.0688931941986084, |
|
"logits/rejected": -3.096926212310791, |
|
"logps/chosen": -0.7589991688728333, |
|
"logps/rejected": -0.9845136404037476, |
|
"loss": 0.8196, |
|
"odds_ratio_loss": 0.6064754128456116, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.07589991390705109, |
|
"rewards/margins": 0.02255145087838173, |
|
"rewards/rejected": -0.09845136106014252, |
|
"sft_loss": 0.7589991688728333, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 1.7781541066892466, |
|
"grad_norm": 2.4373152256011963, |
|
"learning_rate": 1.7784429574324803e-06, |
|
"logits/chosen": -3.079857349395752, |
|
"logits/rejected": -3.1024343967437744, |
|
"logps/chosen": -0.738146960735321, |
|
"logps/rejected": -0.9651134610176086, |
|
"loss": 0.7991, |
|
"odds_ratio_loss": 0.6093234419822693, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07381470501422882, |
|
"rewards/margins": 0.022696642205119133, |
|
"rewards/rejected": -0.0965113490819931, |
|
"sft_loss": 0.738146960735321, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 1.7950889077053345, |
|
"grad_norm": 1.4168922901153564, |
|
"learning_rate": 1.7360720543020327e-06, |
|
"logits/chosen": -3.158327102661133, |
|
"logits/rejected": -3.1482691764831543, |
|
"logps/chosen": -0.7429002523422241, |
|
"logps/rejected": -0.9451411366462708, |
|
"loss": 0.8021, |
|
"odds_ratio_loss": 0.5917203426361084, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.07429002225399017, |
|
"rewards/margins": 0.020224085077643394, |
|
"rewards/rejected": -0.09451410919427872, |
|
"sft_loss": 0.7429002523422241, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 1.8120237087214224, |
|
"grad_norm": 7.384251594543457, |
|
"learning_rate": 1.6939418329887042e-06, |
|
"logits/chosen": -3.1223366260528564, |
|
"logits/rejected": -3.157349109649658, |
|
"logps/chosen": -0.7720141410827637, |
|
"logps/rejected": -1.004219651222229, |
|
"loss": 0.8344, |
|
"odds_ratio_loss": 0.623358428478241, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.07720141112804413, |
|
"rewards/margins": 0.023220548406243324, |
|
"rewards/rejected": -0.1004219651222229, |
|
"sft_loss": 0.7720141410827637, |
|
"step": 1070 |
|
}, |
|
{ |
|
"epoch": 1.8289585097375105, |
|
"grad_norm": 1.976076602935791, |
|
"learning_rate": 1.6520655669671467e-06, |
|
"logits/chosen": -3.151508331298828, |
|
"logits/rejected": -3.174203395843506, |
|
"logps/chosen": -0.7854975461959839, |
|
"logps/rejected": -0.9663504362106323, |
|
"loss": 0.8499, |
|
"odds_ratio_loss": 0.6441539525985718, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.07854975759983063, |
|
"rewards/margins": 0.01808529533445835, |
|
"rewards/rejected": -0.09663505107164383, |
|
"sft_loss": 0.7854975461959839, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 1.8458933107535986, |
|
"grad_norm": 2.3342175483703613, |
|
"learning_rate": 1.610456449701294e-06, |
|
"logits/chosen": -3.110039234161377, |
|
"logits/rejected": -3.1346383094787598, |
|
"logps/chosen": -0.7955976724624634, |
|
"logps/rejected": -1.0117195844650269, |
|
"loss": 0.8634, |
|
"odds_ratio_loss": 0.6780521273612976, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07955978065729141, |
|
"rewards/margins": 0.02161218598484993, |
|
"rewards/rejected": -0.10117195546627045, |
|
"sft_loss": 0.7955976724624634, |
|
"step": 1090 |
|
}, |
|
{ |
|
"epoch": 1.8628281117696868, |
|
"grad_norm": 2.040372133255005, |
|
"learning_rate": 1.5691275904876545e-06, |
|
"logits/chosen": -3.11810302734375, |
|
"logits/rejected": -3.0907464027404785, |
|
"logps/chosen": -0.8171418905258179, |
|
"logps/rejected": -1.049744963645935, |
|
"loss": 0.8794, |
|
"odds_ratio_loss": 0.6221813559532166, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.0817141979932785, |
|
"rewards/margins": 0.02326030097901821, |
|
"rewards/rejected": -0.10497449338436127, |
|
"sft_loss": 0.8171418905258179, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.879762912785775, |
|
"grad_norm": 1.1363812685012817, |
|
"learning_rate": 1.5280920103251235e-06, |
|
"logits/chosen": -3.121904134750366, |
|
"logits/rejected": -3.1400184631347656, |
|
"logps/chosen": -0.7571579217910767, |
|
"logps/rejected": -0.9578372240066528, |
|
"loss": 0.8213, |
|
"odds_ratio_loss": 0.6416117548942566, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.07571578770875931, |
|
"rewards/margins": 0.020067930221557617, |
|
"rewards/rejected": -0.09578372538089752, |
|
"sft_loss": 0.7571579217910767, |
|
"step": 1110 |
|
}, |
|
{ |
|
"epoch": 1.8966977138018628, |
|
"grad_norm": 3.5236151218414307, |
|
"learning_rate": 1.4873626378126015e-06, |
|
"logits/chosen": -3.109051465988159, |
|
"logits/rejected": -3.123108386993408, |
|
"logps/chosen": -0.7997997403144836, |
|
"logps/rejected": -1.0338528156280518, |
|
"loss": 0.8642, |
|
"odds_ratio_loss": 0.6441462635993958, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.07997997850179672, |
|
"rewards/margins": 0.02340531349182129, |
|
"rewards/rejected": -0.10338529199361801, |
|
"sft_loss": 0.7997997403144836, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 1.913632514817951, |
|
"grad_norm": 4.680517673492432, |
|
"learning_rate": 1.446952305075738e-06, |
|
"logits/chosen": -3.1131389141082764, |
|
"logits/rejected": -3.151978015899658, |
|
"logps/chosen": -0.7748031616210938, |
|
"logps/rejected": -0.8985496759414673, |
|
"loss": 0.8402, |
|
"odds_ratio_loss": 0.6542429327964783, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07748032361268997, |
|
"rewards/margins": 0.012374645099043846, |
|
"rewards/rejected": -0.08985497057437897, |
|
"sft_loss": 0.7748031616210938, |
|
"step": 1130 |
|
}, |
|
{ |
|
"epoch": 1.9305673158340388, |
|
"grad_norm": 1.990546703338623, |
|
"learning_rate": 1.406873743724065e-06, |
|
"logits/chosen": -3.134593963623047, |
|
"logits/rejected": -3.088059186935425, |
|
"logps/chosen": -0.8194819688796997, |
|
"logps/rejected": -1.0895836353302002, |
|
"loss": 0.8842, |
|
"odds_ratio_loss": 0.6472653746604919, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.08194819837808609, |
|
"rewards/margins": 0.02701016701757908, |
|
"rewards/rejected": -0.10895836353302002, |
|
"sft_loss": 0.8194819688796997, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 1.947502116850127, |
|
"grad_norm": 2.0966198444366455, |
|
"learning_rate": 1.3671395808397898e-06, |
|
"logits/chosen": -3.1003191471099854, |
|
"logits/rejected": -3.159738779067993, |
|
"logps/chosen": -0.7502952218055725, |
|
"logps/rejected": -0.8763904571533203, |
|
"loss": 0.8142, |
|
"odds_ratio_loss": 0.6395031213760376, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07502951472997665, |
|
"rewards/margins": 0.012609531171619892, |
|
"rewards/rejected": -0.08763904869556427, |
|
"sft_loss": 0.7502952218055725, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 1.964436917866215, |
|
"grad_norm": 6.19160795211792, |
|
"learning_rate": 1.3277623349995418e-06, |
|
"logits/chosen": -3.1115336418151855, |
|
"logits/rejected": -3.1329808235168457, |
|
"logps/chosen": -0.7726483941078186, |
|
"logps/rejected": -0.9365378618240356, |
|
"loss": 0.8381, |
|
"odds_ratio_loss": 0.6545372605323792, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07726484537124634, |
|
"rewards/margins": 0.016388945281505585, |
|
"rewards/rejected": -0.09365378320217133, |
|
"sft_loss": 0.7726483941078186, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 1.9813717188823032, |
|
"grad_norm": 2.1759722232818604, |
|
"learning_rate": 1.2887544123302781e-06, |
|
"logits/chosen": -3.1343424320220947, |
|
"logits/rejected": -3.145904064178467, |
|
"logps/chosen": -0.795665442943573, |
|
"logps/rejected": -0.9278801679611206, |
|
"loss": 0.863, |
|
"odds_ratio_loss": 0.6731692552566528, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.07956655323505402, |
|
"rewards/margins": 0.01322146225720644, |
|
"rewards/rejected": -0.09278801828622818, |
|
"sft_loss": 0.795665442943573, |
|
"step": 1170 |
|
}, |
|
{ |
|
"epoch": 1.9983065198983911, |
|
"grad_norm": 2.340391159057617, |
|
"learning_rate": 1.2501281026006393e-06, |
|
"logits/chosen": -3.113882064819336, |
|
"logits/rejected": -3.1493611335754395, |
|
"logps/chosen": -0.7624078989028931, |
|
"logps/rejected": -0.9082571268081665, |
|
"loss": 0.8289, |
|
"odds_ratio_loss": 0.6652835607528687, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.07624078541994095, |
|
"rewards/margins": 0.014584928750991821, |
|
"rewards/rejected": -0.09082571417093277, |
|
"sft_loss": 0.7624078989028931, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 2.015241320914479, |
|
"grad_norm": 2.1316022872924805, |
|
"learning_rate": 1.2118955753489523e-06, |
|
"logits/chosen": -3.1328041553497314, |
|
"logits/rejected": -3.119529962539673, |
|
"logps/chosen": -0.7868901491165161, |
|
"logps/rejected": -0.9673662185668945, |
|
"loss": 0.8514, |
|
"odds_ratio_loss": 0.6448970437049866, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07868902385234833, |
|
"rewards/margins": 0.018047606572508812, |
|
"rewards/rejected": -0.09673662483692169, |
|
"sft_loss": 0.7868901491165161, |
|
"step": 1190 |
|
}, |
|
{ |
|
"epoch": 2.032176121930567, |
|
"grad_norm": 2.219909906387329, |
|
"learning_rate": 1.1740688760491189e-06, |
|
"logits/chosen": -3.1113944053649902, |
|
"logits/rejected": -3.1537253856658936, |
|
"logps/chosen": -0.7722674608230591, |
|
"logps/rejected": -1.0175268650054932, |
|
"loss": 0.8299, |
|
"odds_ratio_loss": 0.5762220621109009, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.07722674310207367, |
|
"rewards/margins": 0.02452593669295311, |
|
"rewards/rejected": -0.10175268352031708, |
|
"sft_loss": 0.7722674608230591, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 2.0491109229466553, |
|
"grad_norm": 6.625383377075195, |
|
"learning_rate": 1.1366599223155847e-06, |
|
"logits/chosen": -3.092228651046753, |
|
"logits/rejected": -3.137305498123169, |
|
"logps/chosen": -0.7458280324935913, |
|
"logps/rejected": -1.1138603687286377, |
|
"loss": 0.8056, |
|
"odds_ratio_loss": 0.5978988409042358, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.07458280026912689, |
|
"rewards/margins": 0.0368032269179821, |
|
"rewards/rejected": -0.1113860234618187, |
|
"sft_loss": 0.7458280324935913, |
|
"step": 1210 |
|
}, |
|
{ |
|
"epoch": 2.0660457239627434, |
|
"grad_norm": 3.234821081161499, |
|
"learning_rate": 1.0996805001486067e-06, |
|
"logits/chosen": -3.109764575958252, |
|
"logits/rejected": -3.1344008445739746, |
|
"logps/chosen": -0.7534822225570679, |
|
"logps/rejected": -0.9295659065246582, |
|
"loss": 0.8161, |
|
"odds_ratio_loss": 0.625883162021637, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.07534822076559067, |
|
"rewards/margins": 0.017608370631933212, |
|
"rewards/rejected": -0.09295658767223358, |
|
"sft_loss": 0.7534822225570679, |
|
"step": 1220 |
|
}, |
|
{ |
|
"epoch": 2.0829805249788316, |
|
"grad_norm": 3.5898754596710205, |
|
"learning_rate": 1.0631422602209608e-06, |
|
"logits/chosen": -3.1376945972442627, |
|
"logits/rejected": -3.1564555168151855, |
|
"logps/chosen": -0.7927883863449097, |
|
"logps/rejected": -0.9650642275810242, |
|
"loss": 0.8569, |
|
"odds_ratio_loss": 0.6410055160522461, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.0792788416147232, |
|
"rewards/margins": 0.017227591946721077, |
|
"rewards/rejected": -0.09650643169879913, |
|
"sft_loss": 0.7927883863449097, |
|
"step": 1230 |
|
}, |
|
{ |
|
"epoch": 2.0999153259949197, |
|
"grad_norm": 1.6415690183639526, |
|
"learning_rate": 1.027056714207319e-06, |
|
"logits/chosen": -3.1497724056243896, |
|
"logits/rejected": -3.1502339839935303, |
|
"logps/chosen": -0.8299382925033569, |
|
"logps/rejected": -1.0621525049209595, |
|
"loss": 0.8939, |
|
"odds_ratio_loss": 0.6398864984512329, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.08299383521080017, |
|
"rewards/margins": 0.023221401497721672, |
|
"rewards/rejected": -0.10621523857116699, |
|
"sft_loss": 0.8299382925033569, |
|
"step": 1240 |
|
}, |
|
{ |
|
"epoch": 2.116850127011008, |
|
"grad_norm": 2.1568691730499268, |
|
"learning_rate": 9.914352311573838e-07, |
|
"logits/chosen": -3.1183249950408936, |
|
"logits/rejected": -3.130068302154541, |
|
"logps/chosen": -0.680639386177063, |
|
"logps/rejected": -0.9202540516853333, |
|
"loss": 0.7394, |
|
"odds_ratio_loss": 0.5878284573554993, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.06806393712759018, |
|
"rewards/margins": 0.02396145835518837, |
|
"rewards/rejected": -0.09202539175748825, |
|
"sft_loss": 0.680639386177063, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 2.1337849280270955, |
|
"grad_norm": 1.6225751638412476, |
|
"learning_rate": 9.562890339139877e-07, |
|
"logits/chosen": -3.0915961265563965, |
|
"logits/rejected": -3.1580090522766113, |
|
"logps/chosen": -0.7393635511398315, |
|
"logps/rejected": -0.9581485986709595, |
|
"loss": 0.8032, |
|
"odds_ratio_loss": 0.6382402181625366, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.07393636554479599, |
|
"rewards/margins": 0.021878493949770927, |
|
"rewards/rejected": -0.09581486135721207, |
|
"sft_loss": 0.7393635511398315, |
|
"step": 1260 |
|
}, |
|
{ |
|
"epoch": 2.1507197290431836, |
|
"grad_norm": 1.8676691055297852, |
|
"learning_rate": 9.216291955772374e-07, |
|
"logits/chosen": -3.0996224880218506, |
|
"logits/rejected": -3.153738021850586, |
|
"logps/chosen": -0.7607396841049194, |
|
"logps/rejected": -0.9290043115615845, |
|
"loss": 0.8238, |
|
"odds_ratio_loss": 0.6303194761276245, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07607396692037582, |
|
"rewards/margins": 0.016826456412672997, |
|
"rewards/rejected": -0.09290042519569397, |
|
"sft_loss": 0.7607396841049194, |
|
"step": 1270 |
|
}, |
|
{ |
|
"epoch": 2.167654530059272, |
|
"grad_norm": 2.096705198287964, |
|
"learning_rate": 8.874666360158457e-07, |
|
"logits/chosen": -3.082366466522217, |
|
"logits/rejected": -3.0917139053344727, |
|
"logps/chosen": -0.7106717824935913, |
|
"logps/rejected": -0.9525697827339172, |
|
"loss": 0.774, |
|
"odds_ratio_loss": 0.6332431435585022, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07106717675924301, |
|
"rewards/margins": 0.024189796298742294, |
|
"rewards/rejected": -0.0952569767832756, |
|
"sft_loss": 0.7106717824935913, |
|
"step": 1280 |
|
}, |
|
{ |
|
"epoch": 2.18458933107536, |
|
"grad_norm": 2.435957431793213, |
|
"learning_rate": 8.538121184267315e-07, |
|
"logits/chosen": -3.1009063720703125, |
|
"logits/rejected": -3.121425151824951, |
|
"logps/chosen": -0.6681427955627441, |
|
"logps/rejected": -0.9359544515609741, |
|
"loss": 0.7277, |
|
"odds_ratio_loss": 0.595305323600769, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.06681428104639053, |
|
"rewards/margins": 0.026781165972352028, |
|
"rewards/rejected": -0.09359544515609741, |
|
"sft_loss": 0.6681427955627441, |
|
"step": 1290 |
|
}, |
|
{ |
|
"epoch": 2.201524132091448, |
|
"grad_norm": 2.3414995670318604, |
|
"learning_rate": 8.206762459439907e-07, |
|
"logits/chosen": -3.1007769107818604, |
|
"logits/rejected": -3.142437696456909, |
|
"logps/chosen": -0.7944619059562683, |
|
"logps/rejected": -0.9959940910339355, |
|
"loss": 0.8604, |
|
"odds_ratio_loss": 0.659730076789856, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.07944619655609131, |
|
"rewards/margins": 0.020153220742940903, |
|
"rewards/rejected": -0.09959942102432251, |
|
"sft_loss": 0.7944619059562683, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 2.218458933107536, |
|
"grad_norm": 2.9054501056671143, |
|
"learning_rate": 7.880694582982898e-07, |
|
"logits/chosen": -3.165544271469116, |
|
"logits/rejected": -3.1803879737854004, |
|
"logps/chosen": -0.8287284970283508, |
|
"logps/rejected": -1.0228365659713745, |
|
"loss": 0.8926, |
|
"odds_ratio_loss": 0.6389774084091187, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.08287284523248672, |
|
"rewards/margins": 0.019410807639360428, |
|
"rewards/rejected": -0.10228364169597626, |
|
"sft_loss": 0.8287284970283508, |
|
"step": 1310 |
|
}, |
|
{ |
|
"epoch": 2.235393734123624, |
|
"grad_norm": 3.0612242221832275, |
|
"learning_rate": 7.560020285277401e-07, |
|
"logits/chosen": -3.0883891582489014, |
|
"logits/rejected": -3.142484426498413, |
|
"logps/chosen": -0.7654698491096497, |
|
"logps/rejected": -0.8691753149032593, |
|
"loss": 0.8363, |
|
"odds_ratio_loss": 0.708137035369873, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.07654698193073273, |
|
"rewards/margins": 0.010370554402470589, |
|
"rewards/rejected": -0.08691753447055817, |
|
"sft_loss": 0.7654698491096497, |
|
"step": 1320 |
|
}, |
|
{ |
|
"epoch": 2.252328535139712, |
|
"grad_norm": 3.074390172958374, |
|
"learning_rate": 7.244840597412956e-07, |
|
"logits/chosen": -3.0840373039245605, |
|
"logits/rejected": -3.113865613937378, |
|
"logps/chosen": -0.7826686501502991, |
|
"logps/rejected": -0.8837703466415405, |
|
"loss": 0.8534, |
|
"odds_ratio_loss": 0.7077327370643616, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.07826686650514603, |
|
"rewards/margins": 0.010110177099704742, |
|
"rewards/rejected": -0.08837703615427017, |
|
"sft_loss": 0.7826686501502991, |
|
"step": 1330 |
|
}, |
|
{ |
|
"epoch": 2.2692633361558, |
|
"grad_norm": 6.461457252502441, |
|
"learning_rate": 6.935254819356796e-07, |
|
"logits/chosen": -3.1297388076782227, |
|
"logits/rejected": -3.1470894813537598, |
|
"logps/chosen": -0.7817297577857971, |
|
"logps/rejected": -0.9597527384757996, |
|
"loss": 0.8464, |
|
"odds_ratio_loss": 0.6471723318099976, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.07817298173904419, |
|
"rewards/margins": 0.017802301794290543, |
|
"rewards/rejected": -0.09597527980804443, |
|
"sft_loss": 0.7817297577857971, |
|
"step": 1340 |
|
}, |
|
{ |
|
"epoch": 2.2861981371718882, |
|
"grad_norm": 1.7192656993865967, |
|
"learning_rate": 6.631360488668662e-07, |
|
"logits/chosen": -3.138521194458008, |
|
"logits/rejected": -3.150005578994751, |
|
"logps/chosen": -0.7190467715263367, |
|
"logps/rejected": -0.9147864580154419, |
|
"loss": 0.7824, |
|
"odds_ratio_loss": 0.6330953240394592, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07190467417240143, |
|
"rewards/margins": 0.019573967903852463, |
|
"rewards/rejected": -0.09147863835096359, |
|
"sft_loss": 0.7190467715263367, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 2.3031329381879764, |
|
"grad_norm": 2.849228620529175, |
|
"learning_rate": 6.333253349770672e-07, |
|
"logits/chosen": -3.1443207263946533, |
|
"logits/rejected": -3.1502537727355957, |
|
"logps/chosen": -0.8002703785896301, |
|
"logps/rejected": -0.9851358532905579, |
|
"loss": 0.8656, |
|
"odds_ratio_loss": 0.6537164449691772, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.0800270363688469, |
|
"rewards/margins": 0.018486548215150833, |
|
"rewards/rejected": -0.09851358830928802, |
|
"sft_loss": 0.8002703785896301, |
|
"step": 1360 |
|
}, |
|
{ |
|
"epoch": 2.3200677392040645, |
|
"grad_norm": 1.3985761404037476, |
|
"learning_rate": 6.041027323782364e-07, |
|
"logits/chosen": -3.136557102203369, |
|
"logits/rejected": -3.1552491188049316, |
|
"logps/chosen": -0.7381452918052673, |
|
"logps/rejected": -0.9497495889663696, |
|
"loss": 0.7993, |
|
"odds_ratio_loss": 0.611694872379303, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.07381454110145569, |
|
"rewards/margins": 0.021160420030355453, |
|
"rewards/rejected": -0.09497495740652084, |
|
"sft_loss": 0.7381452918052673, |
|
"step": 1370 |
|
}, |
|
{ |
|
"epoch": 2.337002540220152, |
|
"grad_norm": 2.8737893104553223, |
|
"learning_rate": 5.754774478929969e-07, |
|
"logits/chosen": -3.1531028747558594, |
|
"logits/rejected": -3.1830787658691406, |
|
"logps/chosen": -0.7512658834457397, |
|
"logps/rejected": -0.9563377499580383, |
|
"loss": 0.81, |
|
"odds_ratio_loss": 0.5872438549995422, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.07512658089399338, |
|
"rewards/margins": 0.020507195964455605, |
|
"rewards/rejected": -0.09563378244638443, |
|
"sft_loss": 0.7512658834457397, |
|
"step": 1380 |
|
}, |
|
{ |
|
"epoch": 2.3539373412362403, |
|
"grad_norm": 3.9828193187713623, |
|
"learning_rate": 5.474585001539634e-07, |
|
"logits/chosen": -3.159769296646118, |
|
"logits/rejected": -3.1827354431152344, |
|
"logps/chosen": -0.716742753982544, |
|
"logps/rejected": -0.9266722798347473, |
|
"loss": 0.7742, |
|
"odds_ratio_loss": 0.5742050409317017, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.07167427241802216, |
|
"rewards/margins": 0.020992957055568695, |
|
"rewards/rejected": -0.09266723692417145, |
|
"sft_loss": 0.716742753982544, |
|
"step": 1390 |
|
}, |
|
{ |
|
"epoch": 2.3708721422523285, |
|
"grad_norm": 1.6741605997085571, |
|
"learning_rate": 5.200547167623424e-07, |
|
"logits/chosen": -3.172938823699951, |
|
"logits/rejected": -3.1454200744628906, |
|
"logps/chosen": -0.7799001932144165, |
|
"logps/rejected": -1.040466070175171, |
|
"loss": 0.8399, |
|
"odds_ratio_loss": 0.5997284650802612, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07799001783132553, |
|
"rewards/margins": 0.026056593284010887, |
|
"rewards/rejected": -0.10404660552740097, |
|
"sft_loss": 0.7799001932144165, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 2.3878069432684166, |
|
"grad_norm": 1.3470762968063354, |
|
"learning_rate": 4.932747315067271e-07, |
|
"logits/chosen": -3.15238356590271, |
|
"logits/rejected": -3.1575927734375, |
|
"logps/chosen": -0.7618133425712585, |
|
"logps/rejected": -1.033060073852539, |
|
"loss": 0.8193, |
|
"odds_ratio_loss": 0.574641764163971, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.0761813372373581, |
|
"rewards/margins": 0.027124667540192604, |
|
"rewards/rejected": -0.10330601036548615, |
|
"sft_loss": 0.7618133425712585, |
|
"step": 1410 |
|
}, |
|
{ |
|
"epoch": 2.4047417442845047, |
|
"grad_norm": 4.0449371337890625, |
|
"learning_rate": 4.6712698164294553e-07, |
|
"logits/chosen": -3.1169888973236084, |
|
"logits/rejected": -3.131412982940674, |
|
"logps/chosen": -0.7451744079589844, |
|
"logps/rejected": -0.9645574688911438, |
|
"loss": 0.8029, |
|
"odds_ratio_loss": 0.5777136087417603, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.07451744377613068, |
|
"rewards/margins": 0.021938303485512733, |
|
"rewards/rejected": -0.09645574539899826, |
|
"sft_loss": 0.7451744079589844, |
|
"step": 1420 |
|
}, |
|
{ |
|
"epoch": 2.421676545300593, |
|
"grad_norm": 1.97295343875885, |
|
"learning_rate": 4.41619705235842e-07, |
|
"logits/chosen": -3.1363072395324707, |
|
"logits/rejected": -3.1507456302642822, |
|
"logps/chosen": -0.7545329332351685, |
|
"logps/rejected": -1.082715392112732, |
|
"loss": 0.8124, |
|
"odds_ratio_loss": 0.5782482624053955, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07545328885316849, |
|
"rewards/margins": 0.0328182615339756, |
|
"rewards/rejected": -0.10827155411243439, |
|
"sft_loss": 0.7545329332351685, |
|
"step": 1430 |
|
}, |
|
{ |
|
"epoch": 2.438611346316681, |
|
"grad_norm": 2.880293846130371, |
|
"learning_rate": 4.167609385637961e-07, |
|
"logits/chosen": -3.182002544403076, |
|
"logits/rejected": -3.165118455886841, |
|
"logps/chosen": -0.7962235808372498, |
|
"logps/rejected": -0.975197970867157, |
|
"loss": 0.8596, |
|
"odds_ratio_loss": 0.634021520614624, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.07962235808372498, |
|
"rewards/margins": 0.017897438257932663, |
|
"rewards/rejected": -0.09751980006694794, |
|
"sft_loss": 0.7962235808372498, |
|
"step": 1440 |
|
}, |
|
{ |
|
"epoch": 2.4555461473327687, |
|
"grad_norm": 2.545621871948242, |
|
"learning_rate": 3.9255851358683567e-07, |
|
"logits/chosen": -3.1265337467193604, |
|
"logits/rejected": -3.140825033187866, |
|
"logps/chosen": -0.7502083778381348, |
|
"logps/rejected": -0.9492311477661133, |
|
"loss": 0.817, |
|
"odds_ratio_loss": 0.6676734685897827, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.07502084225416183, |
|
"rewards/margins": 0.019902262836694717, |
|
"rewards/rejected": -0.09492311626672745, |
|
"sft_loss": 0.7502083778381348, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 2.472480948348857, |
|
"grad_norm": 2.91165828704834, |
|
"learning_rate": 3.690200554791082e-07, |
|
"logits/chosen": -3.093761444091797, |
|
"logits/rejected": -3.0926525592803955, |
|
"logps/chosen": -0.7423545122146606, |
|
"logps/rejected": -0.9597676992416382, |
|
"loss": 0.8001, |
|
"odds_ratio_loss": 0.5777243375778198, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.0742354542016983, |
|
"rewards/margins": 0.02174132689833641, |
|
"rewards/rejected": -0.09597676992416382, |
|
"sft_loss": 0.7423545122146606, |
|
"step": 1460 |
|
}, |
|
{ |
|
"epoch": 2.489415749364945, |
|
"grad_norm": 5.057043075561523, |
|
"learning_rate": 3.461529802265079e-07, |
|
"logits/chosen": -3.153536796569824, |
|
"logits/rejected": -3.16294264793396, |
|
"logps/chosen": -0.7584110498428345, |
|
"logps/rejected": -0.9484812617301941, |
|
"loss": 0.8201, |
|
"odds_ratio_loss": 0.6163991093635559, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.07584110647439957, |
|
"rewards/margins": 0.01900702901184559, |
|
"rewards/rejected": -0.0948481336236, |
|
"sft_loss": 0.7584110498428345, |
|
"step": 1470 |
|
}, |
|
{ |
|
"epoch": 2.506350550381033, |
|
"grad_norm": 2.0298070907592773, |
|
"learning_rate": 3.2396449229020883e-07, |
|
"logits/chosen": -3.18571400642395, |
|
"logits/rejected": -3.1676132678985596, |
|
"logps/chosen": -0.7937291264533997, |
|
"logps/rejected": -0.935733437538147, |
|
"loss": 0.8599, |
|
"odds_ratio_loss": 0.6614922881126404, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.07937291264533997, |
|
"rewards/margins": 0.014200428500771523, |
|
"rewards/rejected": -0.09357334673404694, |
|
"sft_loss": 0.7937291264533997, |
|
"step": 1480 |
|
}, |
|
{ |
|
"epoch": 2.523285351397121, |
|
"grad_norm": 4.000706195831299, |
|
"learning_rate": 3.024615823368371e-07, |
|
"logits/chosen": -3.1206421852111816, |
|
"logits/rejected": -3.152639150619507, |
|
"logps/chosen": -0.7629178166389465, |
|
"logps/rejected": -0.9828959703445435, |
|
"loss": 0.8252, |
|
"odds_ratio_loss": 0.622687041759491, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.07629178464412689, |
|
"rewards/margins": 0.021997807547450066, |
|
"rewards/rejected": -0.09828958660364151, |
|
"sft_loss": 0.7629178166389465, |
|
"step": 1490 |
|
}, |
|
{ |
|
"epoch": 2.5402201524132093, |
|
"grad_norm": 2.449856758117676, |
|
"learning_rate": 2.8165102503600716e-07, |
|
"logits/chosen": -3.095520496368408, |
|
"logits/rejected": -3.1078646183013916, |
|
"logps/chosen": -0.7517341375350952, |
|
"logps/rejected": -0.9984272718429565, |
|
"loss": 0.8135, |
|
"odds_ratio_loss": 0.6172657012939453, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.07517342269420624, |
|
"rewards/margins": 0.024669310078024864, |
|
"rewards/rejected": -0.09984272718429565, |
|
"sft_loss": 0.7517341375350952, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 2.5402201524132093, |
|
"eval_logits/chosen": -3.130527973175049, |
|
"eval_logits/rejected": -3.1507296562194824, |
|
"eval_logps/chosen": -0.7867480516433716, |
|
"eval_logps/rejected": -0.9955620169639587, |
|
"eval_loss": 0.8505691885948181, |
|
"eval_odds_ratio_loss": 0.638211727142334, |
|
"eval_rewards/accuracies": 0.5723809599876404, |
|
"eval_rewards/chosen": -0.0786748081445694, |
|
"eval_rewards/margins": 0.020881392061710358, |
|
"eval_rewards/rejected": -0.09955620020627975, |
|
"eval_runtime": 194.4899, |
|
"eval_samples_per_second": 5.399, |
|
"eval_sft_loss": 0.7867480516433716, |
|
"eval_steps_per_second": 2.699, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 2.557154953429297, |
|
"grad_norm": 1.5426675081253052, |
|
"learning_rate": 2.615393769259039e-07, |
|
"logits/chosen": -3.1200222969055176, |
|
"logits/rejected": -3.151517152786255, |
|
"logps/chosen": -0.845578670501709, |
|
"logps/rejected": -0.9381099939346313, |
|
"loss": 0.9194, |
|
"odds_ratio_loss": 0.7380752563476562, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.08455787599086761, |
|
"rewards/margins": 0.00925312377512455, |
|
"rewards/rejected": -0.09381099790334702, |
|
"sft_loss": 0.845578670501709, |
|
"step": 1510 |
|
}, |
|
{ |
|
"epoch": 2.574089754445385, |
|
"grad_norm": 4.487434387207031, |
|
"learning_rate": 2.421329743475917e-07, |
|
"logits/chosen": -3.1358139514923096, |
|
"logits/rejected": -3.157161235809326, |
|
"logps/chosen": -0.7332046627998352, |
|
"logps/rejected": -0.9182003140449524, |
|
"loss": 0.7981, |
|
"odds_ratio_loss": 0.6488706469535828, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.07332046329975128, |
|
"rewards/margins": 0.018499553203582764, |
|
"rewards/rejected": -0.09182002395391464, |
|
"sft_loss": 0.7332046627998352, |
|
"step": 1520 |
|
}, |
|
{ |
|
"epoch": 2.5910245554614733, |
|
"grad_norm": 2.3630266189575195, |
|
"learning_rate": 2.234379314486973e-07, |
|
"logits/chosen": -3.1165411472320557, |
|
"logits/rejected": -3.1539194583892822, |
|
"logps/chosen": -0.8104802370071411, |
|
"logps/rejected": -0.9215306043624878, |
|
"loss": 0.8756, |
|
"odds_ratio_loss": 0.6507223844528198, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08104802668094635, |
|
"rewards/margins": 0.011105035431683064, |
|
"rewards/rejected": -0.09215305000543594, |
|
"sft_loss": 0.8104802370071411, |
|
"step": 1530 |
|
}, |
|
{ |
|
"epoch": 2.6079593564775614, |
|
"grad_norm": 1.243202805519104, |
|
"learning_rate": 2.0546013825709783e-07, |
|
"logits/chosen": -3.1084282398223877, |
|
"logits/rejected": -3.129692792892456, |
|
"logps/chosen": -0.7565353512763977, |
|
"logps/rejected": -1.0974582433700562, |
|
"loss": 0.8149, |
|
"odds_ratio_loss": 0.5834510326385498, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.07565353065729141, |
|
"rewards/margins": 0.034092292189598083, |
|
"rewards/rejected": -0.1097458228468895, |
|
"sft_loss": 0.7565353512763977, |
|
"step": 1540 |
|
}, |
|
{ |
|
"epoch": 2.6248941574936495, |
|
"grad_norm": 1.8778706789016724, |
|
"learning_rate": 1.88205258825217e-07, |
|
"logits/chosen": -3.0997517108917236, |
|
"logits/rejected": -3.0937392711639404, |
|
"logps/chosen": -0.672328770160675, |
|
"logps/rejected": -0.9460701942443848, |
|
"loss": 0.7295, |
|
"odds_ratio_loss": 0.5718902349472046, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.06723286956548691, |
|
"rewards/margins": 0.02737414464354515, |
|
"rewards/rejected": -0.09460701793432236, |
|
"sft_loss": 0.672328770160675, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 2.6418289585097376, |
|
"grad_norm": 3.3697926998138428, |
|
"learning_rate": 1.7167872944552245e-07, |
|
"logits/chosen": -3.0951905250549316, |
|
"logits/rejected": -3.143221378326416, |
|
"logps/chosen": -0.6983746290206909, |
|
"logps/rejected": -0.9046236276626587, |
|
"loss": 0.7572, |
|
"odds_ratio_loss": 0.5877906084060669, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.06983745843172073, |
|
"rewards/margins": 0.020624909549951553, |
|
"rewards/rejected": -0.09046236425638199, |
|
"sft_loss": 0.6983746290206909, |
|
"step": 1560 |
|
}, |
|
{ |
|
"epoch": 2.6587637595258258, |
|
"grad_norm": 2.3217389583587646, |
|
"learning_rate": 1.5588575693777142e-07, |
|
"logits/chosen": -3.1411678791046143, |
|
"logits/rejected": -3.1680665016174316, |
|
"logps/chosen": -0.7495251297950745, |
|
"logps/rejected": -0.8962133526802063, |
|
"loss": 0.8135, |
|
"odds_ratio_loss": 0.6393758058547974, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.07495252043008804, |
|
"rewards/margins": 0.01466882973909378, |
|
"rewards/rejected": -0.08962134271860123, |
|
"sft_loss": 0.7495251297950745, |
|
"step": 1570 |
|
}, |
|
{ |
|
"epoch": 2.675698560541914, |
|
"grad_norm": 2.695345401763916, |
|
"learning_rate": 1.4083131700856428e-07, |
|
"logits/chosen": -3.093736410140991, |
|
"logits/rejected": -3.154810905456543, |
|
"logps/chosen": -0.7956855893135071, |
|
"logps/rejected": -0.9356697201728821, |
|
"loss": 0.8605, |
|
"odds_ratio_loss": 0.6480044722557068, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.07956856489181519, |
|
"rewards/margins": 0.013998406007885933, |
|
"rewards/rejected": -0.09356696903705597, |
|
"sft_loss": 0.7956855893135071, |
|
"step": 1580 |
|
}, |
|
{ |
|
"epoch": 2.6926333615580016, |
|
"grad_norm": 4.358800888061523, |
|
"learning_rate": 1.2652015268370315e-07, |
|
"logits/chosen": -3.150224208831787, |
|
"logits/rejected": -3.183022975921631, |
|
"logps/chosen": -0.7435088753700256, |
|
"logps/rejected": -0.9757458567619324, |
|
"loss": 0.8044, |
|
"odds_ratio_loss": 0.6088230013847351, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.07435088604688644, |
|
"rewards/margins": 0.02322370745241642, |
|
"rewards/rejected": -0.09757460653781891, |
|
"sft_loss": 0.7435088753700256, |
|
"step": 1590 |
|
}, |
|
{ |
|
"epoch": 2.7095681625740897, |
|
"grad_norm": 1.9499818086624146, |
|
"learning_rate": 1.1295677281386502e-07, |
|
"logits/chosen": -3.1664371490478516, |
|
"logits/rejected": -3.1615631580352783, |
|
"logps/chosen": -0.8278031349182129, |
|
"logps/rejected": -1.062387228012085, |
|
"loss": 0.8914, |
|
"odds_ratio_loss": 0.6362180113792419, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08278031647205353, |
|
"rewards/margins": 0.02345840446650982, |
|
"rewards/rejected": -0.1062387228012085, |
|
"sft_loss": 0.8278031349182129, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 2.726502963590178, |
|
"grad_norm": 1.277694821357727, |
|
"learning_rate": 1.0014545065404973e-07, |
|
"logits/chosen": -3.12839674949646, |
|
"logits/rejected": -3.1691813468933105, |
|
"logps/chosen": -0.8073375821113586, |
|
"logps/rejected": -1.0353208780288696, |
|
"loss": 0.8777, |
|
"odds_ratio_loss": 0.7032862305641174, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.0807337611913681, |
|
"rewards/margins": 0.022798333317041397, |
|
"rewards/rejected": -0.1035320907831192, |
|
"sft_loss": 0.8073375821113586, |
|
"step": 1610 |
|
}, |
|
{ |
|
"epoch": 2.743437764606266, |
|
"grad_norm": 1.525190830230713, |
|
"learning_rate": 8.809022251725502e-08, |
|
"logits/chosen": -3.1842474937438965, |
|
"logits/rejected": -3.156172752380371, |
|
"logps/chosen": -0.7726050615310669, |
|
"logps/rejected": -1.0311490297317505, |
|
"loss": 0.8348, |
|
"odds_ratio_loss": 0.6221681833267212, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07726050913333893, |
|
"rewards/margins": 0.02585439942777157, |
|
"rewards/rejected": -0.10311490297317505, |
|
"sft_loss": 0.7726050615310669, |
|
"step": 1620 |
|
}, |
|
{ |
|
"epoch": 2.7603725656223537, |
|
"grad_norm": 2.015005111694336, |
|
"learning_rate": 7.679488650280509e-08, |
|
"logits/chosen": -3.192863702774048, |
|
"logits/rejected": -3.213048219680786, |
|
"logps/chosen": -0.7830113172531128, |
|
"logps/rejected": -0.9780160188674927, |
|
"loss": 0.8429, |
|
"odds_ratio_loss": 0.5990433096885681, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.07830111682415009, |
|
"rewards/margins": 0.019500473514199257, |
|
"rewards/rejected": -0.09780160337686539, |
|
"sft_loss": 0.7830113172531128, |
|
"step": 1630 |
|
}, |
|
{ |
|
"epoch": 2.777307366638442, |
|
"grad_norm": 2.5389175415039062, |
|
"learning_rate": 6.626300129972563e-08, |
|
"logits/chosen": -3.122981071472168, |
|
"logits/rejected": -3.214785099029541, |
|
"logps/chosen": -0.7296528220176697, |
|
"logps/rejected": -0.955167293548584, |
|
"loss": 0.792, |
|
"odds_ratio_loss": 0.6230596303939819, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.07296527922153473, |
|
"rewards/margins": 0.02255145087838173, |
|
"rewards/rejected": -0.09551674872636795, |
|
"sft_loss": 0.7296528220176697, |
|
"step": 1640 |
|
}, |
|
{ |
|
"epoch": 2.79424216765453, |
|
"grad_norm": 1.7503160238265991, |
|
"learning_rate": 5.649788506555065e-08, |
|
"logits/chosen": -3.115635871887207, |
|
"logits/rejected": -3.166555881500244, |
|
"logps/chosen": -0.7383006811141968, |
|
"logps/rejected": -0.9469397664070129, |
|
"loss": 0.797, |
|
"odds_ratio_loss": 0.5870878100395203, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.07383007556200027, |
|
"rewards/margins": 0.020863894373178482, |
|
"rewards/rejected": -0.09469397366046906, |
|
"sft_loss": 0.7383006811141968, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 2.811176968670618, |
|
"grad_norm": 5.490494728088379, |
|
"learning_rate": 4.7502614380908474e-08, |
|
"logits/chosen": -3.1543877124786377, |
|
"logits/rejected": -3.134704113006592, |
|
"logps/chosen": -0.7694907188415527, |
|
"logps/rejected": -0.9553998112678528, |
|
"loss": 0.8333, |
|
"odds_ratio_loss": 0.6379188895225525, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.07694907486438751, |
|
"rewards/margins": 0.018590910360217094, |
|
"rewards/rejected": -0.09553998708724976, |
|
"sft_loss": 0.7694907188415527, |
|
"step": 1660 |
|
}, |
|
{ |
|
"epoch": 2.828111769686706, |
|
"grad_norm": 3.784432888031006, |
|
"learning_rate": 3.9280023280222066e-08, |
|
"logits/chosen": -3.1404221057891846, |
|
"logits/rejected": -3.1622607707977295, |
|
"logps/chosen": -0.757176399230957, |
|
"logps/rejected": -0.9332196116447449, |
|
"loss": 0.8222, |
|
"odds_ratio_loss": 0.6507169008255005, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.07571764290332794, |
|
"rewards/margins": 0.017604324966669083, |
|
"rewards/rejected": -0.09332196414470673, |
|
"sft_loss": 0.757176399230957, |
|
"step": 1670 |
|
}, |
|
{ |
|
"epoch": 2.8450465707027943, |
|
"grad_norm": 2.088921546936035, |
|
"learning_rate": 3.1832702358818855e-08, |
|
"logits/chosen": -3.116088390350342, |
|
"logits/rejected": -3.1639227867126465, |
|
"logps/chosen": -0.8132543563842773, |
|
"logps/rejected": -1.068440556526184, |
|
"loss": 0.8739, |
|
"odds_ratio_loss": 0.6062491536140442, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.08132544159889221, |
|
"rewards/margins": 0.025518611073493958, |
|
"rewards/rejected": -0.10684405267238617, |
|
"sft_loss": 0.8132543563842773, |
|
"step": 1680 |
|
}, |
|
{ |
|
"epoch": 2.8619813717188824, |
|
"grad_norm": 2.5282764434814453, |
|
"learning_rate": 2.5162997956746647e-08, |
|
"logits/chosen": -3.1500442028045654, |
|
"logits/rejected": -3.171433210372925, |
|
"logps/chosen": -0.7479134202003479, |
|
"logps/rejected": -0.9974180459976196, |
|
"loss": 0.8076, |
|
"odds_ratio_loss": 0.596734344959259, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.0747913345694542, |
|
"rewards/margins": 0.02495047077536583, |
|
"rewards/rejected": -0.09974180907011032, |
|
"sft_loss": 0.7479134202003479, |
|
"step": 1690 |
|
}, |
|
{ |
|
"epoch": 2.8789161727349706, |
|
"grad_norm": 3.469899892807007, |
|
"learning_rate": 1.9273011419536914e-08, |
|
"logits/chosen": -3.140916585922241, |
|
"logits/rejected": -3.16453218460083, |
|
"logps/chosen": -0.7706011533737183, |
|
"logps/rejected": -0.9274336695671082, |
|
"loss": 0.8378, |
|
"odds_ratio_loss": 0.6717150211334229, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.07706011831760406, |
|
"rewards/margins": 0.0156832467764616, |
|
"rewards/rejected": -0.09274337440729141, |
|
"sft_loss": 0.7706011533737183, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 2.8958509737510583, |
|
"grad_norm": 1.3444654941558838, |
|
"learning_rate": 1.4164598436159083e-08, |
|
"logits/chosen": -3.1658730506896973, |
|
"logits/rejected": -3.1828174591064453, |
|
"logps/chosen": -0.7568970322608948, |
|
"logps/rejected": -0.9151817560195923, |
|
"loss": 0.82, |
|
"odds_ratio_loss": 0.630837082862854, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.07568971067667007, |
|
"rewards/margins": 0.01582847535610199, |
|
"rewards/rejected": -0.09151817858219147, |
|
"sft_loss": 0.7568970322608948, |
|
"step": 1710 |
|
}, |
|
{ |
|
"epoch": 2.9127857747671464, |
|
"grad_norm": 2.7227256298065186, |
|
"learning_rate": 9.839368454371556e-09, |
|
"logits/chosen": -3.1114795207977295, |
|
"logits/rejected": -3.11403751373291, |
|
"logps/chosen": -0.7409245371818542, |
|
"logps/rejected": -1.0069682598114014, |
|
"loss": 0.8046, |
|
"odds_ratio_loss": 0.6370644569396973, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.07409246265888214, |
|
"rewards/margins": 0.026604369282722473, |
|
"rewards/rejected": -0.10069682449102402, |
|
"sft_loss": 0.7409245371818542, |
|
"step": 1720 |
|
}, |
|
{ |
|
"epoch": 2.9297205757832345, |
|
"grad_norm": 1.9127988815307617, |
|
"learning_rate": 6.298684173650649e-09, |
|
"logits/chosen": -3.0886847972869873, |
|
"logits/rejected": -3.123133897781372, |
|
"logps/chosen": -0.7336040735244751, |
|
"logps/rejected": -1.0557795763015747, |
|
"loss": 0.7977, |
|
"odds_ratio_loss": 0.640912652015686, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.07336040586233139, |
|
"rewards/margins": 0.03221754729747772, |
|
"rewards/rejected": -0.10557796061038971, |
|
"sft_loss": 0.7336040735244751, |
|
"step": 1730 |
|
}, |
|
{ |
|
"epoch": 2.9466553767993227, |
|
"grad_norm": 4.324622631072998, |
|
"learning_rate": 3.543661115860686e-09, |
|
"logits/chosen": -3.103651523590088, |
|
"logits/rejected": -3.136838674545288, |
|
"logps/chosen": -0.790323793888092, |
|
"logps/rejected": -1.0052763223648071, |
|
"loss": 0.8552, |
|
"odds_ratio_loss": 0.6486952900886536, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07903237640857697, |
|
"rewards/margins": 0.021495262160897255, |
|
"rewards/rejected": -0.10052764415740967, |
|
"sft_loss": 0.790323793888092, |
|
"step": 1740 |
|
}, |
|
{ |
|
"epoch": 2.963590177815411, |
|
"grad_norm": 2.1066298484802246, |
|
"learning_rate": 1.575167273800693e-09, |
|
"logits/chosen": -3.1322402954101562, |
|
"logits/rejected": -3.141758680343628, |
|
"logps/chosen": -0.7295509576797485, |
|
"logps/rejected": -0.8775620460510254, |
|
"loss": 0.7914, |
|
"odds_ratio_loss": 0.6189672350883484, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.07295509427785873, |
|
"rewards/margins": 0.01480111014097929, |
|
"rewards/rejected": -0.0877562090754509, |
|
"sft_loss": 0.7295509576797485, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 2.9805249788314985, |
|
"grad_norm": 2.692870855331421, |
|
"learning_rate": 3.9382283773564676e-10, |
|
"logits/chosen": -3.153014659881592, |
|
"logits/rejected": -3.1698267459869385, |
|
"logps/chosen": -0.8286467790603638, |
|
"logps/rejected": -1.0176646709442139, |
|
"loss": 0.8971, |
|
"odds_ratio_loss": 0.6844185590744019, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.0828646719455719, |
|
"rewards/margins": 0.018901783972978592, |
|
"rewards/rejected": -0.10176645219326019, |
|
"sft_loss": 0.8286467790603638, |
|
"step": 1760 |
|
}, |
|
{ |
|
"epoch": 2.9974597798475866, |
|
"grad_norm": 11.2490234375, |
|
"learning_rate": 0.0, |
|
"logits/chosen": -3.122014284133911, |
|
"logits/rejected": -3.142989158630371, |
|
"logps/chosen": -0.8531309962272644, |
|
"logps/rejected": -1.0983434915542603, |
|
"loss": 0.9206, |
|
"odds_ratio_loss": 0.6742203235626221, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.0853130966424942, |
|
"rewards/margins": 0.024521255865693092, |
|
"rewards/rejected": -0.10983435064554214, |
|
"sft_loss": 0.8531309962272644, |
|
"step": 1770 |
|
}, |
|
{ |
|
"epoch": 2.9974597798475866, |
|
"step": 1770, |
|
"total_flos": 2.0399855839629804e+18, |
|
"train_loss": 0.8663494454938813, |
|
"train_runtime": 17638.0542, |
|
"train_samples_per_second": 1.607, |
|
"train_steps_per_second": 0.1 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 1770, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"total_flos": 2.0399855839629804e+18, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|