|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.9996020692399522, |
|
"eval_steps": 500, |
|
"global_step": 1884, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 2.645502645502645e-09, |
|
"logits/chosen": -1.7827545404434204, |
|
"logits/rejected": -1.8475866317749023, |
|
"logps/chosen": -188.38653564453125, |
|
"logps/rejected": -220.2354736328125, |
|
"loss": 0.6931, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 2.6455026455026453e-08, |
|
"logits/chosen": -1.9714936017990112, |
|
"logits/rejected": -1.876778483390808, |
|
"logps/chosen": -288.3788146972656, |
|
"logps/rejected": -199.33457946777344, |
|
"loss": 0.6931, |
|
"rewards/accuracies": 0.3819444477558136, |
|
"rewards/chosen": -0.00010286699398420751, |
|
"rewards/margins": 0.00021866995666641742, |
|
"rewards/rejected": -0.00032153705251403153, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 5.2910052910052905e-08, |
|
"logits/chosen": -1.9301083087921143, |
|
"logits/rejected": -1.9124500751495361, |
|
"logps/chosen": -180.6164093017578, |
|
"logps/rejected": -186.2301483154297, |
|
"loss": 0.6931, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -4.3160351197002456e-05, |
|
"rewards/margins": -7.290135545190424e-05, |
|
"rewards/rejected": 2.974100425490178e-05, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 7.936507936507936e-08, |
|
"logits/chosen": -1.9063631296157837, |
|
"logits/rejected": -1.937107801437378, |
|
"logps/chosen": -190.45803833007812, |
|
"logps/rejected": -201.83729553222656, |
|
"loss": 0.6933, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.0003868662752211094, |
|
"rewards/margins": -0.00044071293086744845, |
|
"rewards/rejected": 5.3846673836233094e-05, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.0582010582010581e-07, |
|
"logits/chosen": -1.8966213464736938, |
|
"logits/rejected": -1.8653703927993774, |
|
"logps/chosen": -212.13247680664062, |
|
"logps/rejected": -186.06394958496094, |
|
"loss": 0.693, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": 0.00024294569448102266, |
|
"rewards/margins": 0.00039846505387686193, |
|
"rewards/rejected": -0.00015551943215541542, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.3227513227513225e-07, |
|
"logits/chosen": -1.9822158813476562, |
|
"logits/rejected": -1.8613579273223877, |
|
"logps/chosen": -217.9025421142578, |
|
"logps/rejected": -174.90078735351562, |
|
"loss": 0.6923, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": 0.003075198968872428, |
|
"rewards/margins": 0.0013211520854383707, |
|
"rewards/rejected": 0.001754046999849379, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.5873015873015872e-07, |
|
"logits/chosen": -1.9449065923690796, |
|
"logits/rejected": -1.9192440509796143, |
|
"logps/chosen": -176.40371704101562, |
|
"logps/rejected": -172.91717529296875, |
|
"loss": 0.6921, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": 0.003195743542164564, |
|
"rewards/margins": 0.0035919342190027237, |
|
"rewards/rejected": -0.0003961907350458205, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 1.8518518518518516e-07, |
|
"logits/chosen": -1.9199409484863281, |
|
"logits/rejected": -1.8474018573760986, |
|
"logps/chosen": -174.99351501464844, |
|
"logps/rejected": -197.31735229492188, |
|
"loss": 0.6911, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": 0.0023753070272505283, |
|
"rewards/margins": 0.006421695463359356, |
|
"rewards/rejected": -0.004046388436108828, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.1164021164021162e-07, |
|
"logits/chosen": -1.8320897817611694, |
|
"logits/rejected": -1.844348669052124, |
|
"logps/chosen": -195.1573028564453, |
|
"logps/rejected": -195.3834228515625, |
|
"loss": 0.6894, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.001768267946317792, |
|
"rewards/margins": 0.007629372179508209, |
|
"rewards/rejected": -0.009397639892995358, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.3809523809523806e-07, |
|
"logits/chosen": -1.909330129623413, |
|
"logits/rejected": -1.9080867767333984, |
|
"logps/chosen": -196.3611297607422, |
|
"logps/rejected": -187.08120727539062, |
|
"loss": 0.6879, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.0239607784897089, |
|
"rewards/margins": 0.006291805300861597, |
|
"rewards/rejected": -0.030252579599618912, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.645502645502645e-07, |
|
"logits/chosen": -1.8753902912139893, |
|
"logits/rejected": -1.8588756322860718, |
|
"logps/chosen": -197.9770050048828, |
|
"logps/rejected": -210.04736328125, |
|
"loss": 0.6842, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.05865932255983353, |
|
"rewards/margins": 0.00952382292598486, |
|
"rewards/rejected": -0.06818314641714096, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 2.9100529100529097e-07, |
|
"logits/chosen": -1.942865014076233, |
|
"logits/rejected": -1.8896948099136353, |
|
"logps/chosen": -222.5757598876953, |
|
"logps/rejected": -227.91476440429688, |
|
"loss": 0.6796, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07471577823162079, |
|
"rewards/margins": 0.0264597050845623, |
|
"rewards/rejected": -0.1011754721403122, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.1746031746031743e-07, |
|
"logits/chosen": -1.8088843822479248, |
|
"logits/rejected": -1.7518575191497803, |
|
"logps/chosen": -222.64022827148438, |
|
"logps/rejected": -209.8594512939453, |
|
"loss": 0.6776, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.11961501836776733, |
|
"rewards/margins": 0.06059970706701279, |
|
"rewards/rejected": -0.18021473288536072, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.439153439153439e-07, |
|
"logits/chosen": -1.7773793935775757, |
|
"logits/rejected": -1.7670618295669556, |
|
"logps/chosen": -222.9905548095703, |
|
"logps/rejected": -228.2213134765625, |
|
"loss": 0.669, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.18275949358940125, |
|
"rewards/margins": 0.04632816091179848, |
|
"rewards/rejected": -0.22908766567707062, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.703703703703703e-07, |
|
"logits/chosen": -1.777341604232788, |
|
"logits/rejected": -1.6979541778564453, |
|
"logps/chosen": -227.1598358154297, |
|
"logps/rejected": -202.8821258544922, |
|
"loss": 0.6663, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.19891491532325745, |
|
"rewards/margins": 0.05222553759813309, |
|
"rewards/rejected": -0.25114041566848755, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 3.968253968253968e-07, |
|
"logits/chosen": -1.639807939529419, |
|
"logits/rejected": -1.5844175815582275, |
|
"logps/chosen": -248.316650390625, |
|
"logps/rejected": -248.49240112304688, |
|
"loss": 0.6549, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.45894140005111694, |
|
"rewards/margins": 0.053535498678684235, |
|
"rewards/rejected": -0.512476921081543, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 4.2328042328042324e-07, |
|
"logits/chosen": -1.4632054567337036, |
|
"logits/rejected": -1.4093120098114014, |
|
"logps/chosen": -265.42877197265625, |
|
"logps/rejected": -252.86727905273438, |
|
"loss": 0.6609, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.6529312133789062, |
|
"rewards/margins": 0.09157294034957886, |
|
"rewards/rejected": -0.7445041537284851, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.497354497354497e-07, |
|
"logits/chosen": -1.4973636865615845, |
|
"logits/rejected": -1.3732246160507202, |
|
"logps/chosen": -299.1934509277344, |
|
"logps/rejected": -286.1786804199219, |
|
"loss": 0.6473, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.6386870741844177, |
|
"rewards/margins": 0.15349723398685455, |
|
"rewards/rejected": -0.7921843528747559, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.761904761904761e-07, |
|
"logits/chosen": -1.419722318649292, |
|
"logits/rejected": -1.3405072689056396, |
|
"logps/chosen": -262.6790466308594, |
|
"logps/rejected": -247.8793487548828, |
|
"loss": 0.6447, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.6575330495834351, |
|
"rewards/margins": 0.12454842031002045, |
|
"rewards/rejected": -0.7820814847946167, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.999995705919032e-07, |
|
"logits/chosen": -1.4580373764038086, |
|
"logits/rejected": -1.3449883460998535, |
|
"logps/chosen": -265.0930480957031, |
|
"logps/rejected": -266.20880126953125, |
|
"loss": 0.649, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.6115543246269226, |
|
"rewards/margins": 0.11900641769170761, |
|
"rewards/rejected": -0.7305607795715332, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.999480434051858e-07, |
|
"logits/chosen": -1.7100050449371338, |
|
"logits/rejected": -1.6133902072906494, |
|
"logps/chosen": -312.9882507324219, |
|
"logps/rejected": -298.0437927246094, |
|
"loss": 0.6499, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.9183656573295593, |
|
"rewards/margins": 0.07652232050895691, |
|
"rewards/rejected": -0.9948879480361938, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.998106548810311e-07, |
|
"logits/chosen": -1.6054027080535889, |
|
"logits/rejected": -1.5358479022979736, |
|
"logps/chosen": -283.46551513671875, |
|
"logps/rejected": -306.21820068359375, |
|
"loss": 0.6371, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.8442398905754089, |
|
"rewards/margins": 0.2072862684726715, |
|
"rewards/rejected": -1.0515261888504028, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.995874522146975e-07, |
|
"logits/chosen": -1.5969936847686768, |
|
"logits/rejected": -1.5442743301391602, |
|
"logps/chosen": -294.7565002441406, |
|
"logps/rejected": -295.92779541015625, |
|
"loss": 0.6223, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.9133985638618469, |
|
"rewards/margins": 0.20787055790424347, |
|
"rewards/rejected": -1.1212691068649292, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.992785120800375e-07, |
|
"logits/chosen": -1.0925428867340088, |
|
"logits/rejected": -1.0447893142700195, |
|
"logps/chosen": -327.7795104980469, |
|
"logps/rejected": -339.69500732421875, |
|
"loss": 0.6349, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.108730435371399, |
|
"rewards/margins": 0.24315257370471954, |
|
"rewards/rejected": -1.3518829345703125, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.988839406031596e-07, |
|
"logits/chosen": -0.13352422416210175, |
|
"logits/rejected": 0.016487866640090942, |
|
"logps/chosen": -324.3548278808594, |
|
"logps/rejected": -332.9447326660156, |
|
"loss": 0.63, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.3696115016937256, |
|
"rewards/margins": 0.30091235041618347, |
|
"rewards/rejected": -1.6705236434936523, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.98403873325972e-07, |
|
"logits/chosen": -0.25656959414482117, |
|
"logits/rejected": -0.22962741553783417, |
|
"logps/chosen": -276.41180419921875, |
|
"logps/rejected": -297.1820983886719, |
|
"loss": 0.6098, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.8686075210571289, |
|
"rewards/margins": 0.2754421532154083, |
|
"rewards/rejected": -1.1440496444702148, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.978384751596212e-07, |
|
"logits/chosen": 0.7076885104179382, |
|
"logits/rejected": 0.7922934293746948, |
|
"logps/chosen": -315.2431640625, |
|
"logps/rejected": -349.3119812011719, |
|
"loss": 0.6038, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.1041444540023804, |
|
"rewards/margins": 0.35109204053878784, |
|
"rewards/rejected": -1.4552366733551025, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.971879403278432e-07, |
|
"logits/chosen": 1.446623682975769, |
|
"logits/rejected": 1.5377308130264282, |
|
"logps/chosen": -370.0694580078125, |
|
"logps/rejected": -390.1954650878906, |
|
"loss": 0.604, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.6548852920532227, |
|
"rewards/margins": 0.3384679853916168, |
|
"rewards/rejected": -1.9933536052703857, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.964524923002436e-07, |
|
"logits/chosen": 0.9782055020332336, |
|
"logits/rejected": 1.0214025974273682, |
|
"logps/chosen": -374.83990478515625, |
|
"logps/rejected": -407.66644287109375, |
|
"loss": 0.5943, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.666853666305542, |
|
"rewards/margins": 0.4476628303527832, |
|
"rewards/rejected": -2.1145167350769043, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.956323837155325e-07, |
|
"logits/chosen": 1.1159193515777588, |
|
"logits/rejected": 1.1185142993927002, |
|
"logps/chosen": -357.0001220703125, |
|
"logps/rejected": -394.30279541015625, |
|
"loss": 0.6107, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.6487280130386353, |
|
"rewards/margins": 0.3608878254890442, |
|
"rewards/rejected": -2.009615898132324, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.947278962947386e-07, |
|
"logits/chosen": 1.6991510391235352, |
|
"logits/rejected": 1.9209808111190796, |
|
"logps/chosen": -396.7066955566406, |
|
"logps/rejected": -431.6412658691406, |
|
"loss": 0.6001, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -2.0254971981048584, |
|
"rewards/margins": 0.42958149313926697, |
|
"rewards/rejected": -2.455078601837158, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.937393407444337e-07, |
|
"logits/chosen": 1.100517988204956, |
|
"logits/rejected": 1.2365609407424927, |
|
"logps/chosen": -356.17681884765625, |
|
"logps/rejected": -412.0787048339844, |
|
"loss": 0.5925, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.7612950801849365, |
|
"rewards/margins": 0.4073117673397064, |
|
"rewards/rejected": -2.168606758117676, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.926670566499992e-07, |
|
"logits/chosen": 1.4866538047790527, |
|
"logits/rejected": 1.5181796550750732, |
|
"logps/chosen": -370.19793701171875, |
|
"logps/rejected": -399.6379089355469, |
|
"loss": 0.5865, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.8391116857528687, |
|
"rewards/margins": 0.3383946418762207, |
|
"rewards/rejected": -2.1775062084198, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.915114123589732e-07, |
|
"logits/chosen": 3.2927498817443848, |
|
"logits/rejected": 3.4409637451171875, |
|
"logps/chosen": -474.81134033203125, |
|
"logps/rejected": -486.77264404296875, |
|
"loss": 0.5826, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.5577211380004883, |
|
"rewards/margins": 0.5131586790084839, |
|
"rewards/rejected": -3.0708799362182617, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.90272804854517e-07, |
|
"logits/chosen": 4.935120582580566, |
|
"logits/rejected": 5.241981029510498, |
|
"logps/chosen": -558.8336181640625, |
|
"logps/rejected": -587.92578125, |
|
"loss": 0.6196, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -3.244816303253174, |
|
"rewards/margins": 0.4380508065223694, |
|
"rewards/rejected": -3.6828665733337402, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.889516596190448e-07, |
|
"logits/chosen": 3.1185314655303955, |
|
"logits/rejected": 3.6742959022521973, |
|
"logps/chosen": -524.4172973632812, |
|
"logps/rejected": -528.1964111328125, |
|
"loss": 0.602, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.8538341522216797, |
|
"rewards/margins": 0.4846366345882416, |
|
"rewards/rejected": -3.338470935821533, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.875484304880629e-07, |
|
"logits/chosen": 1.2977662086486816, |
|
"logits/rejected": 1.6007190942764282, |
|
"logps/chosen": -468.362060546875, |
|
"logps/rejected": -483.639892578125, |
|
"loss": 0.6246, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -2.143881320953369, |
|
"rewards/margins": 0.36468884348869324, |
|
"rewards/rejected": -2.5085701942443848, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.860635994942702e-07, |
|
"logits/chosen": 1.5786962509155273, |
|
"logits/rejected": 1.9745759963989258, |
|
"logps/chosen": -395.7223205566406, |
|
"logps/rejected": -411.2752990722656, |
|
"loss": 0.5782, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.866349458694458, |
|
"rewards/margins": 0.4314461350440979, |
|
"rewards/rejected": -2.297795534133911, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.844976767019714e-07, |
|
"logits/chosen": 2.636617660522461, |
|
"logits/rejected": 2.620858669281006, |
|
"logps/chosen": -369.3752136230469, |
|
"logps/rejected": -416.45574951171875, |
|
"loss": 0.576, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -2.1540982723236084, |
|
"rewards/margins": 0.4857994019985199, |
|
"rewards/rejected": -2.6398978233337402, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.828512000318616e-07, |
|
"logits/chosen": 2.6557507514953613, |
|
"logits/rejected": 3.367990016937256, |
|
"logps/chosen": -490.51708984375, |
|
"logps/rejected": -534.2076416015625, |
|
"loss": 0.5691, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -2.498900890350342, |
|
"rewards/margins": 0.7616626620292664, |
|
"rewards/rejected": -3.260563373565674, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.811247350762418e-07, |
|
"logits/chosen": 2.237056016921997, |
|
"logits/rejected": 2.7285072803497314, |
|
"logps/chosen": -381.19000244140625, |
|
"logps/rejected": -417.8363342285156, |
|
"loss": 0.5659, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.9070937633514404, |
|
"rewards/margins": 0.5606156587600708, |
|
"rewards/rejected": -2.46770977973938, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.79318874904728e-07, |
|
"logits/chosen": 4.107834339141846, |
|
"logits/rejected": 4.020180702209473, |
|
"logps/chosen": -422.3435974121094, |
|
"logps/rejected": -466.9422912597656, |
|
"loss": 0.5826, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -2.08524751663208, |
|
"rewards/margins": 0.5145748853683472, |
|
"rewards/rejected": -2.599822521209717, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.774342398605221e-07, |
|
"logits/chosen": 4.6263508796691895, |
|
"logits/rejected": 5.094324588775635, |
|
"logps/chosen": -447.66790771484375, |
|
"logps/rejected": -472.98541259765625, |
|
"loss": 0.5864, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -2.5026144981384277, |
|
"rewards/margins": 0.5022978782653809, |
|
"rewards/rejected": -3.0049126148223877, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.754714773473134e-07, |
|
"logits/chosen": 3.137293577194214, |
|
"logits/rejected": 3.4239907264709473, |
|
"logps/chosen": -383.85272216796875, |
|
"logps/rejected": -439.3240661621094, |
|
"loss": 0.5614, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.8029075860977173, |
|
"rewards/margins": 0.6045055389404297, |
|
"rewards/rejected": -2.4074130058288574, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.734312616068851e-07, |
|
"logits/chosen": 3.090550184249878, |
|
"logits/rejected": 3.005370855331421, |
|
"logps/chosen": -393.83636474609375, |
|
"logps/rejected": -456.08917236328125, |
|
"loss": 0.5636, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.2565479278564453, |
|
"rewards/margins": 0.5205822587013245, |
|
"rewards/rejected": -2.777130365371704, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.713142934875005e-07, |
|
"logits/chosen": 3.673936128616333, |
|
"logits/rejected": 4.1635026931762695, |
|
"logps/chosen": -492.8273010253906, |
|
"logps/rejected": -526.05810546875, |
|
"loss": 0.5837, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.6836562156677246, |
|
"rewards/margins": 0.5476377010345459, |
|
"rewards/rejected": -3.2312939167022705, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.6912130020314996e-07, |
|
"logits/chosen": 3.0116238594055176, |
|
"logits/rejected": 3.2378668785095215, |
|
"logps/chosen": -389.37811279296875, |
|
"logps/rejected": -445.2620544433594, |
|
"loss": 0.5657, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -2.161726951599121, |
|
"rewards/margins": 0.5274554491043091, |
|
"rewards/rejected": -2.6891825199127197, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.668530350837408e-07, |
|
"logits/chosen": 2.4322941303253174, |
|
"logits/rejected": 2.488206148147583, |
|
"logps/chosen": -389.4209899902344, |
|
"logps/rejected": -435.225341796875, |
|
"loss": 0.5843, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.8818559646606445, |
|
"rewards/margins": 0.40773385763168335, |
|
"rewards/rejected": -2.2895896434783936, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.64510277316316e-07, |
|
"logits/chosen": 2.517045736312866, |
|
"logits/rejected": 2.917738676071167, |
|
"logps/chosen": -429.2723693847656, |
|
"logps/rejected": -474.98681640625, |
|
"loss": 0.568, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.0465638637542725, |
|
"rewards/margins": 0.6840267777442932, |
|
"rewards/rejected": -2.730590581893921, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.6209383167739015e-07, |
|
"logits/chosen": 2.8990161418914795, |
|
"logits/rejected": 3.3626670837402344, |
|
"logps/chosen": -449.22235107421875, |
|
"logps/rejected": -482.134033203125, |
|
"loss": 0.5814, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -2.392901659011841, |
|
"rewards/margins": 0.5466801524162292, |
|
"rewards/rejected": -2.939582109451294, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.5960452825649526e-07, |
|
"logits/chosen": 2.796527147293091, |
|
"logits/rejected": 2.8022494316101074, |
|
"logps/chosen": -416.53692626953125, |
|
"logps/rejected": -469.519287109375, |
|
"loss": 0.5481, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -2.1346237659454346, |
|
"rewards/margins": 0.5421117544174194, |
|
"rewards/rejected": -2.6767354011535645, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"eval_logits/chosen": 4.160411834716797, |
|
"eval_logits/rejected": 3.967071294784546, |
|
"eval_logps/chosen": -386.0565185546875, |
|
"eval_logps/rejected": -439.2520751953125, |
|
"eval_loss": 0.6089237928390503, |
|
"eval_rewards/accuracies": 0.6705063581466675, |
|
"eval_rewards/chosen": -2.6821529865264893, |
|
"eval_rewards/margins": 0.44139912724494934, |
|
"eval_rewards/rejected": -3.123551845550537, |
|
"eval_runtime": 2833.1592, |
|
"eval_samples_per_second": 12.369, |
|
"eval_steps_per_second": 0.387, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.570432221710314e-07, |
|
"logits/chosen": 2.99139666557312, |
|
"logits/rejected": 2.9831995964050293, |
|
"logps/chosen": -414.53643798828125, |
|
"logps/rejected": -447.3922424316406, |
|
"loss": 0.5896, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -1.9070537090301514, |
|
"rewards/margins": 0.6989824175834656, |
|
"rewards/rejected": -2.6060359477996826, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.5441079327251927e-07, |
|
"logits/chosen": 2.8209099769592285, |
|
"logits/rejected": 3.297544002532959, |
|
"logps/chosen": -415.64990234375, |
|
"logps/rejected": -483.9609375, |
|
"loss": 0.5711, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -2.1026933193206787, |
|
"rewards/margins": 0.5386781096458435, |
|
"rewards/rejected": -2.641371726989746, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.5170814584435644e-07, |
|
"logits/chosen": 3.1211986541748047, |
|
"logits/rejected": 3.634293794631958, |
|
"logps/chosen": -445.901611328125, |
|
"logps/rejected": -477.486083984375, |
|
"loss": 0.5723, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -2.098557949066162, |
|
"rewards/margins": 0.7294325828552246, |
|
"rewards/rejected": -2.8279905319213867, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.4893620829118124e-07, |
|
"logits/chosen": 4.876758575439453, |
|
"logits/rejected": 4.832733154296875, |
|
"logps/chosen": -390.5097351074219, |
|
"logps/rejected": -458.13433837890625, |
|
"loss": 0.5553, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -2.3329362869262695, |
|
"rewards/margins": 0.6324361562728882, |
|
"rewards/rejected": -2.965372323989868, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.460959328199497e-07, |
|
"logits/chosen": 5.745593070983887, |
|
"logits/rejected": 6.544689178466797, |
|
"logps/chosen": -479.26312255859375, |
|
"logps/rejected": -547.4561767578125, |
|
"loss": 0.5592, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -2.804473876953125, |
|
"rewards/margins": 0.6489624977111816, |
|
"rewards/rejected": -3.4534363746643066, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.4318829511283707e-07, |
|
"logits/chosen": 5.760529518127441, |
|
"logits/rejected": 5.908038139343262, |
|
"logps/chosen": -500.1220703125, |
|
"logps/rejected": -579.2546997070312, |
|
"loss": 0.5787, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -3.121411085128784, |
|
"rewards/margins": 0.5803627967834473, |
|
"rewards/rejected": -3.7017738819122314, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.40214293992074e-07, |
|
"logits/chosen": 4.968967914581299, |
|
"logits/rejected": 5.733763694763184, |
|
"logps/chosen": -449.43182373046875, |
|
"logps/rejected": -481.9095153808594, |
|
"loss": 0.5705, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -2.2817494869232178, |
|
"rewards/margins": 0.5884373188018799, |
|
"rewards/rejected": -2.8701870441436768, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.3717495107683516e-07, |
|
"logits/chosen": 6.087460041046143, |
|
"logits/rejected": 6.749306678771973, |
|
"logps/chosen": -554.03173828125, |
|
"logps/rejected": -600.6712646484375, |
|
"loss": 0.5706, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -3.579016923904419, |
|
"rewards/margins": 0.5537681579589844, |
|
"rewards/rejected": -4.132784843444824, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.340713104322953e-07, |
|
"logits/chosen": 5.846838474273682, |
|
"logits/rejected": 6.166060447692871, |
|
"logps/chosen": -547.8532104492188, |
|
"logps/rejected": -635.3438110351562, |
|
"loss": 0.5318, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.4989750385284424, |
|
"rewards/margins": 0.7716847658157349, |
|
"rewards/rejected": -4.270659446716309, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.3090443821097566e-07, |
|
"logits/chosen": 4.9042229652404785, |
|
"logits/rejected": 5.197253704071045, |
|
"logps/chosen": -483.651123046875, |
|
"logps/rejected": -557.0797119140625, |
|
"loss": 0.5436, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -2.6425604820251465, |
|
"rewards/margins": 0.6980206370353699, |
|
"rewards/rejected": -3.340581178665161, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.276754222865029e-07, |
|
"logits/chosen": 4.094830513000488, |
|
"logits/rejected": 4.941472053527832, |
|
"logps/chosen": -461.6971130371094, |
|
"logps/rejected": -490.43377685546875, |
|
"loss": 0.5627, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -2.5023880004882812, |
|
"rewards/margins": 0.6450821757316589, |
|
"rewards/rejected": -3.147469997406006, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.2438537187990565e-07, |
|
"logits/chosen": 4.136041641235352, |
|
"logits/rejected": 4.636130332946777, |
|
"logps/chosen": -462.34136962890625, |
|
"logps/rejected": -513.6154174804688, |
|
"loss": 0.5457, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -2.5274147987365723, |
|
"rewards/margins": 0.8207530975341797, |
|
"rewards/rejected": -3.348167896270752, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.210354171785795e-07, |
|
"logits/chosen": 4.823207855224609, |
|
"logits/rejected": 5.325142860412598, |
|
"logps/chosen": -488.6162109375, |
|
"logps/rejected": -548.1345825195312, |
|
"loss": 0.5523, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -2.699159622192383, |
|
"rewards/margins": 0.712811291217804, |
|
"rewards/rejected": -3.4119713306427, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.1762670894804775e-07, |
|
"logits/chosen": 4.793757915496826, |
|
"logits/rejected": 5.079339027404785, |
|
"logps/chosen": -517.7249755859375, |
|
"logps/rejected": -555.1099243164062, |
|
"loss": 0.5609, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.0854828357696533, |
|
"rewards/margins": 0.5394307374954224, |
|
"rewards/rejected": -3.624913454055786, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.1416041813665493e-07, |
|
"logits/chosen": 4.005798816680908, |
|
"logits/rejected": 4.380357265472412, |
|
"logps/chosen": -465.93701171875, |
|
"logps/rejected": -518.0279541015625, |
|
"loss": 0.5483, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -2.6373207569122314, |
|
"rewards/margins": 0.5303686857223511, |
|
"rewards/rejected": -3.167689323425293, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.1063773547332584e-07, |
|
"logits/chosen": 4.460536956787109, |
|
"logits/rejected": 4.873723030090332, |
|
"logps/chosen": -462.9532775878906, |
|
"logps/rejected": -519.1263427734375, |
|
"loss": 0.5651, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -2.6806063652038574, |
|
"rewards/margins": 0.6171140670776367, |
|
"rewards/rejected": -3.2977206707000732, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.0705987105853077e-07, |
|
"logits/chosen": 4.531912803649902, |
|
"logits/rejected": 4.875602722167969, |
|
"logps/chosen": -430.66015625, |
|
"logps/rejected": -490.43505859375, |
|
"loss": 0.5664, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.494072437286377, |
|
"rewards/margins": 0.5989924669265747, |
|
"rewards/rejected": -3.0930652618408203, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.034280539485952e-07, |
|
"logits/chosen": 3.7882587909698486, |
|
"logits/rejected": 4.701764106750488, |
|
"logps/chosen": -472.9560546875, |
|
"logps/rejected": -545.2316284179688, |
|
"loss": 0.5388, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -2.2864925861358643, |
|
"rewards/margins": 0.8441749811172485, |
|
"rewards/rejected": -3.130667209625244, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.997435317334988e-07, |
|
"logits/chosen": 5.6066789627075195, |
|
"logits/rejected": 6.343667507171631, |
|
"logps/chosen": -546.0533447265625, |
|
"logps/rejected": -598.7782592773438, |
|
"loss": 0.5491, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.1853551864624023, |
|
"rewards/margins": 0.7388944625854492, |
|
"rewards/rejected": -3.9242501258850098, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.960075701083074e-07, |
|
"logits/chosen": 6.01334285736084, |
|
"logits/rejected": 6.114634037017822, |
|
"logps/chosen": -553.4805908203125, |
|
"logps/rejected": -612.23095703125, |
|
"loss": 0.5694, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -3.704514980316162, |
|
"rewards/margins": 0.5731467008590698, |
|
"rewards/rejected": -4.27766227722168, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.92221452438385e-07, |
|
"logits/chosen": 5.397694110870361, |
|
"logits/rejected": 5.5348615646362305, |
|
"logps/chosen": -530.5993041992188, |
|
"logps/rejected": -563.1759033203125, |
|
"loss": 0.5445, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.2607131004333496, |
|
"rewards/margins": 0.5572507977485657, |
|
"rewards/rejected": -3.817964553833008, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.8838647931853684e-07, |
|
"logits/chosen": 5.427793025970459, |
|
"logits/rejected": 5.946757793426514, |
|
"logps/chosen": -589.3401489257812, |
|
"logps/rejected": -668.6383666992188, |
|
"loss": 0.5744, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.8846659660339355, |
|
"rewards/margins": 0.7481729388237, |
|
"rewards/rejected": -4.632839679718018, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.845039681262332e-07, |
|
"logits/chosen": 5.047539710998535, |
|
"logits/rejected": 5.657217025756836, |
|
"logps/chosen": -554.4708862304688, |
|
"logps/rejected": -616.8558349609375, |
|
"loss": 0.5279, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -3.465214967727661, |
|
"rewards/margins": 0.6963861584663391, |
|
"rewards/rejected": -4.1616010665893555, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.805752525690681e-07, |
|
"logits/chosen": 5.46529483795166, |
|
"logits/rejected": 5.889483451843262, |
|
"logps/chosen": -564.59228515625, |
|
"logps/rejected": -660.7513427734375, |
|
"loss": 0.5431, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.814227342605591, |
|
"rewards/margins": 0.7818762063980103, |
|
"rewards/rejected": -4.596103191375732, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 3.7660168222660824e-07, |
|
"logits/chosen": 5.152674198150635, |
|
"logits/rejected": 5.510188102722168, |
|
"logps/chosen": -587.24560546875, |
|
"logps/rejected": -626.5244140625, |
|
"loss": 0.5788, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.4678547382354736, |
|
"rewards/margins": 0.5621030926704407, |
|
"rewards/rejected": -4.0299577713012695, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 3.725846220867901e-07, |
|
"logits/chosen": 4.9715166091918945, |
|
"logits/rejected": 5.927099227905273, |
|
"logps/chosen": -540.8013916015625, |
|
"logps/rejected": -573.8363037109375, |
|
"loss": 0.5665, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -3.4430599212646484, |
|
"rewards/margins": 0.5011590123176575, |
|
"rewards/rejected": -3.944218873977661, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.6852545207702393e-07, |
|
"logits/chosen": 5.292831897735596, |
|
"logits/rejected": 6.226684093475342, |
|
"logps/chosen": -585.9905395507812, |
|
"logps/rejected": -621.8675537109375, |
|
"loss": 0.5763, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.4966049194335938, |
|
"rewards/margins": 0.7289837598800659, |
|
"rewards/rejected": -4.225588798522949, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.6442556659016475e-07, |
|
"logits/chosen": 5.347090244293213, |
|
"logits/rejected": 6.540925025939941, |
|
"logps/chosen": -568.36572265625, |
|
"logps/rejected": -611.6026611328125, |
|
"loss": 0.5235, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.558441162109375, |
|
"rewards/margins": 0.7847374081611633, |
|
"rewards/rejected": -4.343178749084473, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.602863740055161e-07, |
|
"logits/chosen": 5.637946128845215, |
|
"logits/rejected": 6.208116054534912, |
|
"logps/chosen": -585.2955322265625, |
|
"logps/rejected": -676.1448974609375, |
|
"loss": 0.5476, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.8783912658691406, |
|
"rewards/margins": 0.8531737327575684, |
|
"rewards/rejected": -4.731564998626709, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.5610929620502747e-07, |
|
"logits/chosen": 4.267381191253662, |
|
"logits/rejected": 5.2271528244018555, |
|
"logps/chosen": -578.3743286132812, |
|
"logps/rejected": -665.9542846679688, |
|
"loss": 0.5381, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -3.779019594192505, |
|
"rewards/margins": 0.8331762552261353, |
|
"rewards/rejected": -4.61219596862793, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.5189576808485404e-07, |
|
"logits/chosen": 4.498036861419678, |
|
"logits/rejected": 5.546736717224121, |
|
"logps/chosen": -622.6543579101562, |
|
"logps/rejected": -662.39990234375, |
|
"loss": 0.5631, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.9034676551818848, |
|
"rewards/margins": 0.7405894994735718, |
|
"rewards/rejected": -4.644057273864746, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.476472370624464e-07, |
|
"logits/chosen": 5.130145072937012, |
|
"logits/rejected": 5.417891502380371, |
|
"logps/chosen": -601.1119384765625, |
|
"logps/rejected": -665.1697387695312, |
|
"loss": 0.5375, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -4.171017646789551, |
|
"rewards/margins": 0.6281455159187317, |
|
"rewards/rejected": -4.799162864685059, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.43365162579338e-07, |
|
"logits/chosen": 5.479987144470215, |
|
"logits/rejected": 5.379258155822754, |
|
"logps/chosen": -553.6651611328125, |
|
"logps/rejected": -656.02880859375, |
|
"loss": 0.5448, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.905271053314209, |
|
"rewards/margins": 0.8390300869941711, |
|
"rewards/rejected": -4.744300842285156, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.390510155998023e-07, |
|
"logits/chosen": 5.235553741455078, |
|
"logits/rejected": 5.778375148773193, |
|
"logps/chosen": -578.1744384765625, |
|
"logps/rejected": -612.1620483398438, |
|
"loss": 0.5477, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.762741804122925, |
|
"rewards/margins": 0.507239043712616, |
|
"rewards/rejected": -4.269980430603027, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.347062781055526e-07, |
|
"logits/chosen": 5.177485942840576, |
|
"logits/rejected": 5.3884077072143555, |
|
"logps/chosen": -517.0951538085938, |
|
"logps/rejected": -611.8797607421875, |
|
"loss": 0.5674, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -3.408946990966797, |
|
"rewards/margins": 0.6858997344970703, |
|
"rewards/rejected": -4.094846725463867, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.303324425866559e-07, |
|
"logits/chosen": 4.779083728790283, |
|
"logits/rejected": 4.99451208114624, |
|
"logps/chosen": -534.8001708984375, |
|
"logps/rejected": -588.9842529296875, |
|
"loss": 0.5692, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -3.043729543685913, |
|
"rewards/margins": 0.7375171184539795, |
|
"rewards/rejected": -3.7812466621398926, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.2593101152883795e-07, |
|
"logits/chosen": 4.893560409545898, |
|
"logits/rejected": 4.912612438201904, |
|
"logps/chosen": -477.8765563964844, |
|
"logps/rejected": -577.2595825195312, |
|
"loss": 0.5559, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -2.9390878677368164, |
|
"rewards/margins": 0.735144853591919, |
|
"rewards/rejected": -3.6742329597473145, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.21503496897354e-07, |
|
"logits/chosen": 4.999721527099609, |
|
"logits/rejected": 5.3239426612854, |
|
"logps/chosen": -539.9871215820312, |
|
"logps/rejected": -568.0904541015625, |
|
"loss": 0.5742, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -3.2623775005340576, |
|
"rewards/margins": 0.4343135356903076, |
|
"rewards/rejected": -3.6966910362243652, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.170514196176037e-07, |
|
"logits/chosen": 4.89313268661499, |
|
"logits/rejected": 4.976138114929199, |
|
"logps/chosen": -531.0595703125, |
|
"logps/rejected": -615.2716064453125, |
|
"loss": 0.5382, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.470829486846924, |
|
"rewards/margins": 0.6645902395248413, |
|
"rewards/rejected": -4.1354193687438965, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.125763090526674e-07, |
|
"logits/chosen": 4.8970746994018555, |
|
"logits/rejected": 5.335020542144775, |
|
"logps/chosen": -570.1527709960938, |
|
"logps/rejected": -649.9376831054688, |
|
"loss": 0.5099, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.5448403358459473, |
|
"rewards/margins": 0.861894965171814, |
|
"rewards/rejected": -4.406734943389893, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.080797024779447e-07, |
|
"logits/chosen": 4.570401191711426, |
|
"logits/rejected": 5.245053291320801, |
|
"logps/chosen": -538.8987426757812, |
|
"logps/rejected": -598.7467651367188, |
|
"loss": 0.529, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.5106894969940186, |
|
"rewards/margins": 0.7811245918273926, |
|
"rewards/rejected": -4.29181432723999, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.035631445530743e-07, |
|
"logits/chosen": 5.412999629974365, |
|
"logits/rejected": 5.755091190338135, |
|
"logps/chosen": -584.6710205078125, |
|
"logps/rejected": -669.8858642578125, |
|
"loss": 0.5345, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.7473666667938232, |
|
"rewards/margins": 0.8245238065719604, |
|
"rewards/rejected": -4.571890830993652, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 2.9902818679131775e-07, |
|
"logits/chosen": 5.077193260192871, |
|
"logits/rejected": 5.424740791320801, |
|
"logps/chosen": -592.6712646484375, |
|
"logps/rejected": -663.4064331054688, |
|
"loss": 0.5308, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -4.113033294677734, |
|
"rewards/margins": 0.7320185899734497, |
|
"rewards/rejected": -4.845052242279053, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.944763870265886e-07, |
|
"logits/chosen": 4.157199859619141, |
|
"logits/rejected": 5.055262565612793, |
|
"logps/chosen": -593.670166015625, |
|
"logps/rejected": -670.92578125, |
|
"loss": 0.5311, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.861306667327881, |
|
"rewards/margins": 0.8220443725585938, |
|
"rewards/rejected": -4.683350563049316, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.899093088783105e-07, |
|
"logits/chosen": 4.214892864227295, |
|
"logits/rejected": 4.872607707977295, |
|
"logps/chosen": -627.0443115234375, |
|
"logps/rejected": -696.4767456054688, |
|
"loss": 0.5336, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.9733481407165527, |
|
"rewards/margins": 0.7376152276992798, |
|
"rewards/rejected": -4.710963726043701, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.8532852121428733e-07, |
|
"logits/chosen": 4.1724419593811035, |
|
"logits/rejected": 4.470523834228516, |
|
"logps/chosen": -547.5701904296875, |
|
"logps/rejected": -630.979736328125, |
|
"loss": 0.5351, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.5739071369171143, |
|
"rewards/margins": 0.9093371629714966, |
|
"rewards/rejected": -4.4832444190979, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.807355976117716e-07, |
|
"logits/chosen": 3.997878313064575, |
|
"logits/rejected": 4.5441575050354, |
|
"logps/chosen": -573.764892578125, |
|
"logps/rejected": -626.8602294921875, |
|
"loss": 0.5637, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -3.4168217182159424, |
|
"rewards/margins": 0.7197359800338745, |
|
"rewards/rejected": -4.1365580558776855, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.761321158169134e-07, |
|
"logits/chosen": 3.821613311767578, |
|
"logits/rejected": 4.272428512573242, |
|
"logps/chosen": -539.5350341796875, |
|
"logps/rejected": -624.6610717773438, |
|
"loss": 0.5437, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.425632953643799, |
|
"rewards/margins": 0.7200672030448914, |
|
"rewards/rejected": -4.145700454711914, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.715196572027789e-07, |
|
"logits/chosen": 3.830516815185547, |
|
"logits/rejected": 3.83209228515625, |
|
"logps/chosen": -553.0765380859375, |
|
"logps/rejected": -614.8712768554688, |
|
"loss": 0.5591, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.4271316528320312, |
|
"rewards/margins": 0.6863579750061035, |
|
"rewards/rejected": -4.113489627838135, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.6689980622612204e-07, |
|
"logits/chosen": 4.027700424194336, |
|
"logits/rejected": 4.4086809158325195, |
|
"logps/chosen": -504.0587463378906, |
|
"logps/rejected": -589.3375854492188, |
|
"loss": 0.5519, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -3.1788105964660645, |
|
"rewards/margins": 0.6995212435722351, |
|
"rewards/rejected": -3.8783316612243652, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"eval_logits/chosen": 5.348609447479248, |
|
"eval_logits/rejected": 5.197441101074219, |
|
"eval_logps/chosen": -543.0738525390625, |
|
"eval_logps/rejected": -602.8671264648438, |
|
"eval_loss": 0.5866538882255554, |
|
"eval_rewards/accuracies": 0.6894388794898987, |
|
"eval_rewards/chosen": -4.252326488494873, |
|
"eval_rewards/margins": 0.5073760151863098, |
|
"eval_rewards/rejected": -4.759702205657959, |
|
"eval_runtime": 2829.1466, |
|
"eval_samples_per_second": 12.387, |
|
"eval_steps_per_second": 0.387, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.622741498830969e-07, |
|
"logits/chosen": 4.356446266174316, |
|
"logits/rejected": 4.390655517578125, |
|
"logps/chosen": -531.9536743164062, |
|
"logps/rejected": -595.4317626953125, |
|
"loss": 0.5685, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.2892327308654785, |
|
"rewards/margins": 0.6568211913108826, |
|
"rewards/rejected": -3.9460537433624268, |
|
"step": 1010 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.5764427716409815e-07, |
|
"logits/chosen": 3.622380495071411, |
|
"logits/rejected": 4.40954065322876, |
|
"logps/chosen": -515.4381103515625, |
|
"logps/rejected": -565.3131103515625, |
|
"loss": 0.5428, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.0056488513946533, |
|
"rewards/margins": 0.683438777923584, |
|
"rewards/rejected": -3.6890876293182373, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.5301177850791616e-07, |
|
"logits/chosen": 3.9004855155944824, |
|
"logits/rejected": 4.826648712158203, |
|
"logps/chosen": -568.6829833984375, |
|
"logps/rejected": -630.1458740234375, |
|
"loss": 0.5394, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.4316725730895996, |
|
"rewards/margins": 0.825964093208313, |
|
"rewards/rejected": -4.257637023925781, |
|
"step": 1030 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4837824525539477e-07, |
|
"logits/chosen": 3.6745972633361816, |
|
"logits/rejected": 4.580860614776611, |
|
"logps/chosen": -539.4935302734375, |
|
"logps/rejected": -591.1002197265625, |
|
"loss": 0.5549, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -3.340437650680542, |
|
"rewards/margins": 0.6588428020477295, |
|
"rewards/rejected": -3.9992804527282715, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 2.4374526910277886e-07, |
|
"logits/chosen": 4.059652805328369, |
|
"logits/rejected": 4.569832801818848, |
|
"logps/chosen": -524.096435546875, |
|
"logps/rejected": -599.4136962890625, |
|
"loss": 0.566, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.2275619506835938, |
|
"rewards/margins": 0.767623782157898, |
|
"rewards/rejected": -3.9951858520507812, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 2.391144415549403e-07, |
|
"logits/chosen": 4.6053290367126465, |
|
"logits/rejected": 4.822667121887207, |
|
"logps/chosen": -529.7521362304688, |
|
"logps/rejected": -581.6422729492188, |
|
"loss": 0.5563, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -3.478646755218506, |
|
"rewards/margins": 0.4808022081851959, |
|
"rewards/rejected": -3.959449052810669, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.3448735337866919e-07, |
|
"logits/chosen": 4.680068016052246, |
|
"logits/rejected": 5.09098482131958, |
|
"logps/chosen": -486.0177307128906, |
|
"logps/rejected": -539.49658203125, |
|
"loss": 0.5976, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -3.101942539215088, |
|
"rewards/margins": 0.5219632983207703, |
|
"rewards/rejected": -3.623905897140503, |
|
"step": 1070 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.2986559405621886e-07, |
|
"logits/chosen": 4.115077495574951, |
|
"logits/rejected": 4.646478652954102, |
|
"logps/chosen": -508.98480224609375, |
|
"logps/rejected": -547.0084838867188, |
|
"loss": 0.5705, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.912672281265259, |
|
"rewards/margins": 0.5327907204627991, |
|
"rewards/rejected": -3.445463180541992, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2525075123929213e-07, |
|
"logits/chosen": 4.412012100219727, |
|
"logits/rejected": 4.801966190338135, |
|
"logps/chosen": -494.9298400878906, |
|
"logps/rejected": -553.3126220703125, |
|
"loss": 0.53, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -3.0377612113952637, |
|
"rewards/margins": 0.6897190809249878, |
|
"rewards/rejected": -3.72748064994812, |
|
"step": 1090 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.206444102036565e-07, |
|
"logits/chosen": 4.899998664855957, |
|
"logits/rejected": 5.011576175689697, |
|
"logps/chosen": -533.6759643554688, |
|
"logps/rejected": -604.164306640625, |
|
"loss": 0.5622, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.346374034881592, |
|
"rewards/margins": 0.5992105603218079, |
|
"rewards/rejected": -3.945584774017334, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.160481533045751e-07, |
|
"logits/chosen": 4.868978977203369, |
|
"logits/rejected": 5.403256416320801, |
|
"logps/chosen": -569.0045776367188, |
|
"logps/rejected": -625.4342041015625, |
|
"loss": 0.547, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.556847333908081, |
|
"rewards/margins": 0.7111159563064575, |
|
"rewards/rejected": -4.267963409423828, |
|
"step": 1110 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.1146355943324148e-07, |
|
"logits/chosen": 5.191876411437988, |
|
"logits/rejected": 5.507510185241699, |
|
"logps/chosen": -557.5202026367188, |
|
"logps/rejected": -637.4208374023438, |
|
"loss": 0.5456, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.72900652885437, |
|
"rewards/margins": 0.7132843732833862, |
|
"rewards/rejected": -4.442290782928467, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.0689220347440374e-07, |
|
"logits/chosen": 4.4830732345581055, |
|
"logits/rejected": 5.013645172119141, |
|
"logps/chosen": -587.7266235351562, |
|
"logps/rejected": -625.2584228515625, |
|
"loss": 0.5546, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.556133985519409, |
|
"rewards/margins": 0.6282913684844971, |
|
"rewards/rejected": -4.184426307678223, |
|
"step": 1130 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.0233565576536564e-07, |
|
"logits/chosen": 4.331673622131348, |
|
"logits/rejected": 4.588160037994385, |
|
"logps/chosen": -579.3844604492188, |
|
"logps/rejected": -643.15380859375, |
|
"loss": 0.5457, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.580660581588745, |
|
"rewards/margins": 0.58534175157547, |
|
"rewards/rejected": -4.16600227355957, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 1.97795481556549e-07, |
|
"logits/chosen": 3.8659698963165283, |
|
"logits/rejected": 4.494038105010986, |
|
"logps/chosen": -526.2404174804688, |
|
"logps/rejected": -560.0923461914062, |
|
"loss": 0.5521, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.2603347301483154, |
|
"rewards/margins": 0.6364775896072388, |
|
"rewards/rejected": -3.8968119621276855, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.9327324047380422e-07, |
|
"logits/chosen": 3.7696309089660645, |
|
"logits/rejected": 4.594429016113281, |
|
"logps/chosen": -504.9520568847656, |
|
"logps/rejected": -586.5645751953125, |
|
"loss": 0.5119, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.0566916465759277, |
|
"rewards/margins": 0.8342288732528687, |
|
"rewards/rejected": -3.8909201622009277, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.887704859826528e-07, |
|
"logits/chosen": 4.6678924560546875, |
|
"logits/rejected": 5.033539772033691, |
|
"logps/chosen": -579.7088012695312, |
|
"logps/rejected": -635.5621948242188, |
|
"loss": 0.5359, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -3.5427486896514893, |
|
"rewards/margins": 0.6539695262908936, |
|
"rewards/rejected": -4.196718692779541, |
|
"step": 1170 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.8428876485464572e-07, |
|
"logits/chosen": 4.120070934295654, |
|
"logits/rejected": 4.802687644958496, |
|
"logps/chosen": -546.7890014648438, |
|
"logps/rejected": -595.0975341796875, |
|
"loss": 0.5792, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.6790192127227783, |
|
"rewards/margins": 0.6043803095817566, |
|
"rewards/rejected": -4.28339958190918, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.798296166360216e-07, |
|
"logits/chosen": 4.212487697601318, |
|
"logits/rejected": 5.026057243347168, |
|
"logps/chosen": -570.1341552734375, |
|
"logps/rejected": -631.1887817382812, |
|
"loss": 0.554, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.4938530921936035, |
|
"rewards/margins": 0.8111255764961243, |
|
"rewards/rejected": -4.304978370666504, |
|
"step": 1190 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.7539457311884675e-07, |
|
"logits/chosen": 4.184502601623535, |
|
"logits/rejected": 4.767962455749512, |
|
"logps/chosen": -532.1156005859375, |
|
"logps/rejected": -596.5296630859375, |
|
"loss": 0.5289, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.360400438308716, |
|
"rewards/margins": 0.7973008155822754, |
|
"rewards/rejected": -4.157701015472412, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.7098515781481883e-07, |
|
"logits/chosen": 5.043318748474121, |
|
"logits/rejected": 5.477365970611572, |
|
"logps/chosen": -554.041015625, |
|
"logps/rejected": -602.6400146484375, |
|
"loss": 0.5381, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -3.55816388130188, |
|
"rewards/margins": 0.6002733111381531, |
|
"rewards/rejected": -4.1584367752075195, |
|
"step": 1210 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6660288543191568e-07, |
|
"logits/chosen": 4.342132568359375, |
|
"logits/rejected": 5.391766548156738, |
|
"logps/chosen": -545.3358764648438, |
|
"logps/rejected": -593.21044921875, |
|
"loss": 0.5273, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.2604167461395264, |
|
"rewards/margins": 0.7034675478935242, |
|
"rewards/rejected": -3.9638848304748535, |
|
"step": 1220 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6224926135406693e-07, |
|
"logits/chosen": 4.523472785949707, |
|
"logits/rejected": 4.9561238288879395, |
|
"logps/chosen": -541.8357543945312, |
|
"logps/rejected": -617.3576049804688, |
|
"loss": 0.5279, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.246079206466675, |
|
"rewards/margins": 0.7863640785217285, |
|
"rewards/rejected": -4.032443046569824, |
|
"step": 1230 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.579257811240298e-07, |
|
"logits/chosen": 4.174516201019287, |
|
"logits/rejected": 4.948155403137207, |
|
"logps/chosen": -566.966796875, |
|
"logps/rejected": -629.2576293945312, |
|
"loss": 0.5383, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.5496108531951904, |
|
"rewards/margins": 0.6297346949577332, |
|
"rewards/rejected": -4.179345607757568, |
|
"step": 1240 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.5363392992964523e-07, |
|
"logits/chosen": 4.812657833099365, |
|
"logits/rejected": 5.1759843826293945, |
|
"logps/chosen": -545.2940673828125, |
|
"logps/rejected": -610.6871948242188, |
|
"loss": 0.5599, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.605778217315674, |
|
"rewards/margins": 0.5686632394790649, |
|
"rewards/rejected": -4.174441337585449, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4937518209365108e-07, |
|
"logits/chosen": 4.399840354919434, |
|
"logits/rejected": 5.024414539337158, |
|
"logps/chosen": -565.787353515625, |
|
"logps/rejected": -621.1328125, |
|
"loss": 0.5422, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.319702625274658, |
|
"rewards/margins": 0.7489235997200012, |
|
"rewards/rejected": -4.0686259269714355, |
|
"step": 1260 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4515100056722708e-07, |
|
"logits/chosen": 4.322859287261963, |
|
"logits/rejected": 4.647487163543701, |
|
"logps/chosen": -473.44915771484375, |
|
"logps/rejected": -556.26806640625, |
|
"loss": 0.5335, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -2.9221551418304443, |
|
"rewards/margins": 0.8288917541503906, |
|
"rewards/rejected": -3.7510476112365723, |
|
"step": 1270 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.4096283642744716e-07, |
|
"logits/chosen": 4.445618152618408, |
|
"logits/rejected": 5.310964107513428, |
|
"logps/chosen": -529.9594116210938, |
|
"logps/rejected": -583.6397094726562, |
|
"loss": 0.5489, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -3.098684310913086, |
|
"rewards/margins": 0.7321470975875854, |
|
"rewards/rejected": -3.8308310508728027, |
|
"step": 1280 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.3681212837880977e-07, |
|
"logits/chosen": 4.398858070373535, |
|
"logits/rejected": 4.8625688552856445, |
|
"logps/chosen": -509.6946716308594, |
|
"logps/rejected": -594.6348876953125, |
|
"loss": 0.5397, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -2.9974119663238525, |
|
"rewards/margins": 0.8787434697151184, |
|
"rewards/rejected": -3.876155376434326, |
|
"step": 1290 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.3270030225901908e-07, |
|
"logits/chosen": 4.207759857177734, |
|
"logits/rejected": 5.357243537902832, |
|
"logps/chosen": -540.0505981445312, |
|
"logps/rejected": -591.9456787109375, |
|
"loss": 0.5405, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -2.9670040607452393, |
|
"rewards/margins": 0.9191484451293945, |
|
"rewards/rejected": -3.8861522674560547, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2862877054918572e-07, |
|
"logits/chosen": 3.810641050338745, |
|
"logits/rejected": 4.167244911193848, |
|
"logps/chosen": -455.54962158203125, |
|
"logps/rejected": -525.954833984375, |
|
"loss": 0.558, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -2.536816358566284, |
|
"rewards/margins": 0.7640112638473511, |
|
"rewards/rejected": -3.300827741622925, |
|
"step": 1310 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2459893188861613e-07, |
|
"logits/chosen": 3.5092411041259766, |
|
"logits/rejected": 4.095297813415527, |
|
"logps/chosen": -398.01702880859375, |
|
"logps/rejected": -486.40106201171875, |
|
"loss": 0.5194, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -2.352292060852051, |
|
"rewards/margins": 0.8707995414733887, |
|
"rewards/rejected": -3.2230911254882812, |
|
"step": 1320 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.206121705943558e-07, |
|
"logits/chosen": 3.8648457527160645, |
|
"logits/rejected": 4.79974365234375, |
|
"logps/chosen": -450.3089904785156, |
|
"logps/rejected": -489.1319274902344, |
|
"loss": 0.5501, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -2.3793253898620605, |
|
"rewards/margins": 0.6881033778190613, |
|
"rewards/rejected": -3.0674290657043457, |
|
"step": 1330 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.1666985618565422e-07, |
|
"logits/chosen": 4.9529547691345215, |
|
"logits/rejected": 5.19863224029541, |
|
"logps/chosen": -469.70855712890625, |
|
"logps/rejected": -529.8267822265625, |
|
"loss": 0.5422, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -2.916710615158081, |
|
"rewards/margins": 0.6288083791732788, |
|
"rewards/rejected": -3.545518398284912, |
|
"step": 1340 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 1.1277334291351145e-07, |
|
"logits/chosen": 5.23069953918457, |
|
"logits/rejected": 5.482142925262451, |
|
"logps/chosen": -469.21307373046875, |
|
"logps/rejected": -589.8045654296875, |
|
"loss": 0.5183, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -2.9651408195495605, |
|
"rewards/margins": 0.8228071331977844, |
|
"rewards/rejected": -3.7879481315612793, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 1.089239692954701e-07, |
|
"logits/chosen": 4.438393592834473, |
|
"logits/rejected": 5.2009711265563965, |
|
"logps/chosen": -496.31463623046875, |
|
"logps/rejected": -561.5128173828125, |
|
"loss": 0.5497, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -2.9592137336730957, |
|
"rewards/margins": 0.6417844891548157, |
|
"rewards/rejected": -3.6009984016418457, |
|
"step": 1360 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.051230576558127e-07, |
|
"logits/chosen": 5.329151630401611, |
|
"logits/rejected": 5.1115946769714355, |
|
"logps/chosen": -510.32379150390625, |
|
"logps/rejected": -633.5717163085938, |
|
"loss": 0.5281, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.305920124053955, |
|
"rewards/margins": 0.6949362754821777, |
|
"rewards/rejected": -4.000856399536133, |
|
"step": 1370 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.0137191367132078e-07, |
|
"logits/chosen": 4.635811805725098, |
|
"logits/rejected": 5.594052314758301, |
|
"logps/chosen": -538.316650390625, |
|
"logps/rejected": -593.9783325195312, |
|
"loss": 0.5179, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.257719039916992, |
|
"rewards/margins": 0.8566193580627441, |
|
"rewards/rejected": -4.1143388748168945, |
|
"step": 1380 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.76718259227532e-08, |
|
"logits/chosen": 5.040810585021973, |
|
"logits/rejected": 5.872241973876953, |
|
"logps/chosen": -559.94140625, |
|
"logps/rejected": -600.4876098632812, |
|
"loss": 0.5346, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.3652939796447754, |
|
"rewards/margins": 0.7677735090255737, |
|
"rewards/rejected": -4.1330671310424805, |
|
"step": 1390 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.402406545219676e-08, |
|
"logits/chosen": 5.209423542022705, |
|
"logits/rejected": 5.7783403396606445, |
|
"logps/chosen": -570.754150390625, |
|
"logps/rejected": -631.2447509765625, |
|
"loss": 0.5343, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -3.6044132709503174, |
|
"rewards/margins": 0.8434345126152039, |
|
"rewards/rejected": -4.447848320007324, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.042988532644249e-08, |
|
"logits/chosen": 5.212072372436523, |
|
"logits/rejected": 5.8972063064575195, |
|
"logps/chosen": -583.494873046875, |
|
"logps/rejected": -651.96630859375, |
|
"loss": 0.5139, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.3433117866516113, |
|
"rewards/margins": 0.9961609840393066, |
|
"rewards/rejected": -4.339472770690918, |
|
"step": 1410 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 8.689052020653592e-08, |
|
"logits/chosen": 4.78833532333374, |
|
"logits/rejected": 5.89553165435791, |
|
"logps/chosen": -561.2659912109375, |
|
"logps/rejected": -631.704833984375, |
|
"loss": 0.5514, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.425673007965088, |
|
"rewards/margins": 0.8838043212890625, |
|
"rewards/rejected": -4.309477806091309, |
|
"step": 1420 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 8.340718592365037e-08, |
|
"logits/chosen": 5.993227958679199, |
|
"logits/rejected": 6.045858860015869, |
|
"logps/chosen": -575.0015869140625, |
|
"logps/rejected": -656.5886840820312, |
|
"loss": 0.5134, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.8804218769073486, |
|
"rewards/margins": 0.6545805931091309, |
|
"rewards/rejected": -4.535001754760742, |
|
"step": 1430 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 7.998107906142839e-08, |
|
"logits/chosen": 5.676799774169922, |
|
"logits/rejected": 6.391809940338135, |
|
"logps/chosen": -568.3199462890625, |
|
"logps/rejected": -637.5294189453125, |
|
"loss": 0.5254, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.7702269554138184, |
|
"rewards/margins": 0.8325515985488892, |
|
"rewards/rejected": -4.602778434753418, |
|
"step": 1440 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.661337654493575e-08, |
|
"logits/chosen": 5.199474334716797, |
|
"logits/rejected": 5.9724626541137695, |
|
"logps/chosen": -604.0126953125, |
|
"logps/rejected": -679.3408203125, |
|
"loss": 0.5511, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.842377185821533, |
|
"rewards/margins": 0.8208445310592651, |
|
"rewards/rejected": -4.663221836090088, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.330523523636751e-08, |
|
"logits/chosen": 4.888028144836426, |
|
"logits/rejected": 5.282634258270264, |
|
"logps/chosen": -574.2249755859375, |
|
"logps/rejected": -649.12158203125, |
|
"loss": 0.5353, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.574962615966797, |
|
"rewards/margins": 0.7398016452789307, |
|
"rewards/rejected": -4.314764499664307, |
|
"step": 1460 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.005779153764682e-08, |
|
"logits/chosen": 5.4382123947143555, |
|
"logits/rejected": 5.483049392700195, |
|
"logps/chosen": -550.1482543945312, |
|
"logps/rejected": -629.5477294921875, |
|
"loss": 0.5274, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.7003333568573, |
|
"rewards/margins": 0.6894980669021606, |
|
"rewards/rejected": -4.389832019805908, |
|
"step": 1470 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.687216100005138e-08, |
|
"logits/chosen": 5.554679870605469, |
|
"logits/rejected": 6.150310516357422, |
|
"logps/chosen": -604.5882568359375, |
|
"logps/rejected": -694.1824340820312, |
|
"loss": 0.5241, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.837186336517334, |
|
"rewards/margins": 0.8338931202888489, |
|
"rewards/rejected": -4.671080589294434, |
|
"step": 1480 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.374943794100349e-08, |
|
"logits/chosen": 4.995267868041992, |
|
"logits/rejected": 6.089870452880859, |
|
"logps/chosen": -540.8165283203125, |
|
"logps/rejected": -621.503662109375, |
|
"loss": 0.5181, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.447101593017578, |
|
"rewards/margins": 0.998428463935852, |
|
"rewards/rejected": -4.445529937744141, |
|
"step": 1490 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 6.069069506815325e-08, |
|
"logits/chosen": 5.665683269500732, |
|
"logits/rejected": 5.919187545776367, |
|
"logps/chosen": -557.50341796875, |
|
"logps/rejected": -635.5902099609375, |
|
"loss": 0.5597, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -3.7898292541503906, |
|
"rewards/margins": 0.7044454216957092, |
|
"rewards/rejected": -4.494275093078613, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"eval_logits/chosen": 6.629395008087158, |
|
"eval_logits/rejected": 6.464417934417725, |
|
"eval_logps/chosen": -596.9036865234375, |
|
"eval_logps/rejected": -659.0733032226562, |
|
"eval_loss": 0.5820685625076294, |
|
"eval_rewards/accuracies": 0.6959397792816162, |
|
"eval_rewards/chosen": -4.79062557220459, |
|
"eval_rewards/margins": 0.5311391949653625, |
|
"eval_rewards/rejected": -5.321763515472412, |
|
"eval_runtime": 2832.2152, |
|
"eval_samples_per_second": 12.373, |
|
"eval_steps_per_second": 0.387, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.7696983110885746e-08, |
|
"logits/chosen": 6.144343376159668, |
|
"logits/rejected": 6.1946892738342285, |
|
"logps/chosen": -568.4108276367188, |
|
"logps/rejected": -656.3994750976562, |
|
"loss": 0.5713, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.8781726360321045, |
|
"rewards/margins": 0.7777887582778931, |
|
"rewards/rejected": -4.655961036682129, |
|
"step": 1510 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.47693304593777e-08, |
|
"logits/chosen": 5.610565185546875, |
|
"logits/rejected": 6.231629371643066, |
|
"logps/chosen": -578.0348510742188, |
|
"logps/rejected": -634.0302124023438, |
|
"loss": 0.5254, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.640714168548584, |
|
"rewards/margins": 0.8447046279907227, |
|
"rewards/rejected": -4.485418796539307, |
|
"step": 1520 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.190874281132851e-08, |
|
"logits/chosen": 5.386014938354492, |
|
"logits/rejected": 5.870030403137207, |
|
"logps/chosen": -544.43212890625, |
|
"logps/rejected": -605.0162963867188, |
|
"loss": 0.5313, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.533620834350586, |
|
"rewards/margins": 0.7047785520553589, |
|
"rewards/rejected": -4.238399505615234, |
|
"step": 1530 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.9116202826486045e-08, |
|
"logits/chosen": 5.818629741668701, |
|
"logits/rejected": 6.154656410217285, |
|
"logps/chosen": -583.7796630859375, |
|
"logps/rejected": -635.4794921875, |
|
"loss": 0.5337, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.777665615081787, |
|
"rewards/margins": 0.6795842051506042, |
|
"rewards/rejected": -4.457250118255615, |
|
"step": 1540 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.639266978908676e-08, |
|
"logits/chosen": 5.711030006408691, |
|
"logits/rejected": 6.358799934387207, |
|
"logps/chosen": -611.2662353515625, |
|
"logps/rejected": -676.8931884765625, |
|
"loss": 0.5187, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -3.886317491531372, |
|
"rewards/margins": 0.7436650991439819, |
|
"rewards/rejected": -4.629981994628906, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.373907927832513e-08, |
|
"logits/chosen": 5.366969108581543, |
|
"logits/rejected": 5.813288688659668, |
|
"logps/chosen": -553.2315673828125, |
|
"logps/rejected": -654.0559692382812, |
|
"loss": 0.5274, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.5649421215057373, |
|
"rewards/margins": 0.8516329526901245, |
|
"rewards/rejected": -4.416574954986572, |
|
"step": 1560 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.115634284696698e-08, |
|
"logits/chosen": 5.473841667175293, |
|
"logits/rejected": 6.142959117889404, |
|
"logps/chosen": -595.4747314453125, |
|
"logps/rejected": -675.5442504882812, |
|
"loss": 0.4995, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.8835277557373047, |
|
"rewards/margins": 0.8822458386421204, |
|
"rewards/rejected": -4.765774250030518, |
|
"step": 1570 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.864534770821559e-08, |
|
"logits/chosen": 5.575949668884277, |
|
"logits/rejected": 6.3084797859191895, |
|
"logps/chosen": -557.6798706054688, |
|
"logps/rejected": -628.6642456054688, |
|
"loss": 0.5549, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.6573452949523926, |
|
"rewards/margins": 0.8320104479789734, |
|
"rewards/rejected": -4.489355564117432, |
|
"step": 1580 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.620695643093924e-08, |
|
"logits/chosen": 5.1766839027404785, |
|
"logits/rejected": 6.165072441101074, |
|
"logps/chosen": -578.5045166015625, |
|
"logps/rejected": -645.0850830078125, |
|
"loss": 0.5174, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.7886593341827393, |
|
"rewards/margins": 0.9527637362480164, |
|
"rewards/rejected": -4.7414231300354, |
|
"step": 1590 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.384200664336412e-08, |
|
"logits/chosen": 5.577385425567627, |
|
"logits/rejected": 6.328660011291504, |
|
"logps/chosen": -591.4779052734375, |
|
"logps/rejected": -644.4967041015625, |
|
"loss": 0.5339, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.7442920207977295, |
|
"rewards/margins": 0.8128402829170227, |
|
"rewards/rejected": -4.557132244110107, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.155131074533529e-08, |
|
"logits/chosen": 4.875297546386719, |
|
"logits/rejected": 5.868630886077881, |
|
"logps/chosen": -590.7601318359375, |
|
"logps/rejected": -644.1740112304688, |
|
"loss": 0.5532, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.6774280071258545, |
|
"rewards/margins": 0.69183349609375, |
|
"rewards/rejected": -4.369261741638184, |
|
"step": 1610 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.9335655629243645e-08, |
|
"logits/chosen": 5.1115217208862305, |
|
"logits/rejected": 5.944631099700928, |
|
"logps/chosen": -572.9088134765625, |
|
"logps/rejected": -635.9785766601562, |
|
"loss": 0.5635, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.67173433303833, |
|
"rewards/margins": 0.6962157487869263, |
|
"rewards/rejected": -4.367949962615967, |
|
"step": 1620 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.7195802409715197e-08, |
|
"logits/chosen": 5.198755741119385, |
|
"logits/rejected": 5.86753511428833, |
|
"logps/chosen": -608.7496948242188, |
|
"logps/rejected": -661.3677978515625, |
|
"loss": 0.5482, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -3.853783130645752, |
|
"rewards/margins": 0.7783033847808838, |
|
"rewards/rejected": -4.632086753845215, |
|
"step": 1630 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.513248616215527e-08, |
|
"logits/chosen": 4.92915153503418, |
|
"logits/rejected": 5.753048896789551, |
|
"logps/chosen": -574.3338623046875, |
|
"logps/rejected": -647.4317626953125, |
|
"loss": 0.5148, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.5955843925476074, |
|
"rewards/margins": 0.8324259519577026, |
|
"rewards/rejected": -4.428009986877441, |
|
"step": 1640 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.31464156702382e-08, |
|
"logits/chosen": 4.705880165100098, |
|
"logits/rejected": 5.527700901031494, |
|
"logps/chosen": -576.8226928710938, |
|
"logps/rejected": -628.70849609375, |
|
"loss": 0.5499, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.532841205596924, |
|
"rewards/margins": 0.7848026752471924, |
|
"rewards/rejected": -4.317643642425537, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.1238273182427933e-08, |
|
"logits/chosen": 5.681646347045898, |
|
"logits/rejected": 6.597903251647949, |
|
"logps/chosen": -569.5862426757812, |
|
"logps/rejected": -620.6526489257812, |
|
"loss": 0.5361, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.6919121742248535, |
|
"rewards/margins": 0.7138382196426392, |
|
"rewards/rejected": -4.405750274658203, |
|
"step": 1660 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.9408714177614306e-08, |
|
"logits/chosen": 5.2946391105651855, |
|
"logits/rejected": 6.044753074645996, |
|
"logps/chosen": -564.3060302734375, |
|
"logps/rejected": -626.62353515625, |
|
"loss": 0.5135, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -3.511505126953125, |
|
"rewards/margins": 0.8757253885269165, |
|
"rewards/rejected": -4.38723087310791, |
|
"step": 1670 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.7658367139945228e-08, |
|
"logits/chosen": 5.4068779945373535, |
|
"logits/rejected": 5.994559288024902, |
|
"logps/chosen": -599.1517333984375, |
|
"logps/rejected": -650.2825317382812, |
|
"loss": 0.5239, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -3.7978172302246094, |
|
"rewards/margins": 0.7726759910583496, |
|
"rewards/rejected": -4.570493221282959, |
|
"step": 1680 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.5987833342931745e-08, |
|
"logits/chosen": 5.3490800857543945, |
|
"logits/rejected": 6.137597560882568, |
|
"logps/chosen": -583.5009155273438, |
|
"logps/rejected": -632.0625, |
|
"loss": 0.5467, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.8200180530548096, |
|
"rewards/margins": 0.7188345193862915, |
|
"rewards/rejected": -4.538852691650391, |
|
"step": 1690 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.439768664290053e-08, |
|
"logits/chosen": 5.457294940948486, |
|
"logits/rejected": 5.772470474243164, |
|
"logps/chosen": -603.9612426757812, |
|
"logps/rejected": -673.6254272460938, |
|
"loss": 0.5239, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.798058032989502, |
|
"rewards/margins": 0.787087619304657, |
|
"rewards/rejected": -4.585144996643066, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.2888473281864597e-08, |
|
"logits/chosen": 5.0094194412231445, |
|
"logits/rejected": 5.467037200927734, |
|
"logps/chosen": -566.6424560546875, |
|
"logps/rejected": -654.4783935546875, |
|
"loss": 0.5305, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -3.774543046951294, |
|
"rewards/margins": 0.7515050172805786, |
|
"rewards/rejected": -4.526047706604004, |
|
"step": 1710 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.1460711699880082e-08, |
|
"logits/chosen": 4.981472015380859, |
|
"logits/rejected": 6.010613918304443, |
|
"logps/chosen": -587.7794189453125, |
|
"logps/rejected": -658.3820190429688, |
|
"loss": 0.5277, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.5816383361816406, |
|
"rewards/margins": 0.8929538726806641, |
|
"rewards/rejected": -4.474592208862305, |
|
"step": 1720 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 1.0114892356953397e-08, |
|
"logits/chosen": 4.724175453186035, |
|
"logits/rejected": 5.377623558044434, |
|
"logps/chosen": -572.1964111328125, |
|
"logps/rejected": -616.9322509765625, |
|
"loss": 0.5707, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.6713428497314453, |
|
"rewards/margins": 0.6362916231155396, |
|
"rewards/rejected": -4.3076348304748535, |
|
"step": 1730 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 8.851477564560061e-09, |
|
"logits/chosen": 5.281374454498291, |
|
"logits/rejected": 5.823667526245117, |
|
"logps/chosen": -574.9295654296875, |
|
"logps/rejected": -682.01611328125, |
|
"loss": 0.5272, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -3.782670497894287, |
|
"rewards/margins": 1.0396969318389893, |
|
"rewards/rejected": -4.8223676681518555, |
|
"step": 1740 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.670901326832763e-09, |
|
"logits/chosen": 5.813240051269531, |
|
"logits/rejected": 6.113523006439209, |
|
"logps/chosen": -583.0259399414062, |
|
"logps/rejected": -707.0408935546875, |
|
"loss": 0.5068, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -3.896685838699341, |
|
"rewards/margins": 0.9632736444473267, |
|
"rewards/rejected": -4.859958648681641, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 6.5735691914738936e-09, |
|
"logits/chosen": 4.764095783233643, |
|
"logits/rejected": 5.158614158630371, |
|
"logps/chosen": -563.710693359375, |
|
"logps/rejected": -646.1812744140625, |
|
"loss": 0.5635, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.6481738090515137, |
|
"rewards/margins": 0.7430108785629272, |
|
"rewards/rejected": -4.3911848068237305, |
|
"step": 1760 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 5.559858110443016e-09, |
|
"logits/chosen": 4.941697120666504, |
|
"logits/rejected": 5.857718467712402, |
|
"logps/chosen": -589.8416137695312, |
|
"logps/rejected": -651.1143798828125, |
|
"loss": 0.5218, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.781978130340576, |
|
"rewards/margins": 0.8180473446846008, |
|
"rewards/rejected": -4.600025177001953, |
|
"step": 1770 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 4.6301163104676685e-09, |
|
"logits/chosen": 5.284440517425537, |
|
"logits/rejected": 5.553745269775391, |
|
"logps/chosen": -551.4322509765625, |
|
"logps/rejected": -651.212646484375, |
|
"loss": 0.5407, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -3.6018593311309814, |
|
"rewards/margins": 0.7678863406181335, |
|
"rewards/rejected": -4.36974573135376, |
|
"step": 1780 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 3.784663173421438e-09, |
|
"logits/chosen": 5.374969959259033, |
|
"logits/rejected": 5.7753376960754395, |
|
"logps/chosen": -622.2989501953125, |
|
"logps/rejected": -689.6795654296875, |
|
"loss": 0.5308, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.73246693611145, |
|
"rewards/margins": 0.8092583417892456, |
|
"rewards/rejected": -4.5417256355285645, |
|
"step": 1790 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 3.023789126611137e-09, |
|
"logits/chosen": 5.826882839202881, |
|
"logits/rejected": 6.438210964202881, |
|
"logps/chosen": -580.4129638671875, |
|
"logps/rejected": -642.33740234375, |
|
"loss": 0.5188, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.765054702758789, |
|
"rewards/margins": 0.8300830721855164, |
|
"rewards/rejected": -4.595137596130371, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.3477555430100604e-09, |
|
"logits/chosen": 5.330014228820801, |
|
"logits/rejected": 6.266722679138184, |
|
"logps/chosen": -545.0280151367188, |
|
"logps/rejected": -623.4840087890625, |
|
"loss": 0.5255, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.382579803466797, |
|
"rewards/margins": 1.0013315677642822, |
|
"rewards/rejected": -4.383911609649658, |
|
"step": 1810 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.7567946514721322e-09, |
|
"logits/chosen": 5.459120273590088, |
|
"logits/rejected": 5.671516418457031, |
|
"logps/chosen": -565.1712036132812, |
|
"logps/rejected": -651.7200927734375, |
|
"loss": 0.5631, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.7416863441467285, |
|
"rewards/margins": 0.7198008894920349, |
|
"rewards/rejected": -4.46148681640625, |
|
"step": 1820 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.2511094569571668e-09, |
|
"logits/chosen": 4.77047061920166, |
|
"logits/rejected": 5.680732727050781, |
|
"logps/chosen": -540.1213989257812, |
|
"logps/rejected": -612.4068603515625, |
|
"loss": 0.5294, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.508056640625, |
|
"rewards/margins": 0.8371505737304688, |
|
"rewards/rejected": -4.345207214355469, |
|
"step": 1830 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 8.308736707954289e-10, |
|
"logits/chosen": 5.112320899963379, |
|
"logits/rejected": 6.371096611022949, |
|
"logps/chosen": -591.5335693359375, |
|
"logps/rejected": -622.4219360351562, |
|
"loss": 0.5351, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.7684085369110107, |
|
"rewards/margins": 0.8042453527450562, |
|
"rewards/rejected": -4.572653770446777, |
|
"step": 1840 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 4.962316510149222e-10, |
|
"logits/chosen": 4.924439430236816, |
|
"logits/rejected": 5.723077774047852, |
|
"logps/chosen": -528.1791381835938, |
|
"logps/rejected": -613.794189453125, |
|
"loss": 0.5155, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -3.5361733436584473, |
|
"rewards/margins": 0.8941107988357544, |
|
"rewards/rejected": -4.43028450012207, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 2.4729835275189016e-10, |
|
"logits/chosen": 5.2165093421936035, |
|
"logits/rejected": 5.809523105621338, |
|
"logps/chosen": -512.5826416015625, |
|
"logps/rejected": -562.5357055664062, |
|
"loss": 0.5279, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.3880043029785156, |
|
"rewards/margins": 0.6139224767684937, |
|
"rewards/rejected": -4.001926898956299, |
|
"step": 1860 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 8.415928876176482e-11, |
|
"logits/chosen": 5.226788520812988, |
|
"logits/rejected": 5.463950157165527, |
|
"logps/chosen": -568.6761474609375, |
|
"logps/rejected": -642.3060302734375, |
|
"loss": 0.5796, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -3.871469020843506, |
|
"rewards/margins": 0.6007556319236755, |
|
"rewards/rejected": -4.472224235534668, |
|
"step": 1870 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 6.870500044303673e-12, |
|
"logits/chosen": 5.113773345947266, |
|
"logits/rejected": 5.360236167907715, |
|
"logps/chosen": -532.0494995117188, |
|
"logps/rejected": -629.88818359375, |
|
"loss": 0.537, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.353315830230713, |
|
"rewards/margins": 0.727187991142273, |
|
"rewards/rejected": -4.080503940582275, |
|
"step": 1880 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 1884, |
|
"total_flos": 0.0, |
|
"train_loss": 0.5658232939218006, |
|
"train_runtime": 45309.3847, |
|
"train_samples_per_second": 2.662, |
|
"train_steps_per_second": 0.042 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 1884, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 500, |
|
"total_flos": 0.0, |
|
"train_batch_size": 4, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|