|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.9994666666666666, |
|
"eval_steps": 500, |
|
"global_step": 937, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 5.319148936170213e-08, |
|
"logits/chosen": 0.0328633114695549, |
|
"logits/rejected": 0.22779616713523865, |
|
"logps/chosen": -407.2765197753906, |
|
"logps/rejected": -295.7368469238281, |
|
"loss": 0.4178, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 5.319148936170213e-07, |
|
"logits/chosen": 0.12644250690937042, |
|
"logits/rejected": 0.25490710139274597, |
|
"logps/chosen": -342.5318603515625, |
|
"logps/rejected": -339.8170471191406, |
|
"loss": 0.3561, |
|
"rewards/accuracies": 0.4305555522441864, |
|
"rewards/chosen": 2.7654659788822755e-05, |
|
"rewards/margins": 8.612655801698565e-05, |
|
"rewards/rejected": -5.84718982281629e-05, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.0638297872340427e-06, |
|
"logits/chosen": 0.11099036037921906, |
|
"logits/rejected": 0.24831095337867737, |
|
"logps/chosen": -378.44439697265625, |
|
"logps/rejected": -353.71014404296875, |
|
"loss": 0.3639, |
|
"rewards/accuracies": 0.4000000059604645, |
|
"rewards/chosen": -4.667802568292245e-05, |
|
"rewards/margins": -9.378789400216192e-05, |
|
"rewards/rejected": 4.710986831923947e-05, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.595744680851064e-06, |
|
"logits/chosen": 0.13326099514961243, |
|
"logits/rejected": 0.1938765048980713, |
|
"logps/chosen": -396.07440185546875, |
|
"logps/rejected": -353.0478210449219, |
|
"loss": 0.3768, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": 4.021604763693176e-05, |
|
"rewards/margins": -1.0102905434905551e-05, |
|
"rewards/rejected": 5.0318958528805524e-05, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.1276595744680853e-06, |
|
"logits/chosen": 0.15499012172222137, |
|
"logits/rejected": 0.24013516306877136, |
|
"logps/chosen": -357.16259765625, |
|
"logps/rejected": -330.24609375, |
|
"loss": 0.3838, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": 5.2471255912678316e-05, |
|
"rewards/margins": 0.00012843578588217497, |
|
"rewards/rejected": -7.596451177960262e-05, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.6595744680851065e-06, |
|
"logits/chosen": 0.08975875377655029, |
|
"logits/rejected": 0.1924462616443634, |
|
"logps/chosen": -360.7787780761719, |
|
"logps/rejected": -345.1267395019531, |
|
"loss": 0.3709, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": 0.00022406328935176134, |
|
"rewards/margins": 0.0003413105441723019, |
|
"rewards/rejected": -0.00011724724026862532, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.191489361702128e-06, |
|
"logits/chosen": 0.11686824262142181, |
|
"logits/rejected": 0.2060682326555252, |
|
"logps/chosen": -301.4020690917969, |
|
"logps/rejected": -306.5481872558594, |
|
"loss": 0.379, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": 0.0006540452013723552, |
|
"rewards/margins": 0.001110218232497573, |
|
"rewards/rejected": -0.00045617311843670905, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.723404255319149e-06, |
|
"logits/chosen": 0.061485785990953445, |
|
"logits/rejected": 0.14367519319057465, |
|
"logps/chosen": -309.8575134277344, |
|
"logps/rejected": -319.0173645019531, |
|
"loss": 0.3678, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": 0.000993106747046113, |
|
"rewards/margins": 0.0025861565954983234, |
|
"rewards/rejected": -0.001593050197698176, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.255319148936171e-06, |
|
"logits/chosen": 0.08192186057567596, |
|
"logits/rejected": 0.18210968375205994, |
|
"logps/chosen": -313.7688293457031, |
|
"logps/rejected": -354.7149658203125, |
|
"loss": 0.3692, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": 0.0009643015218898654, |
|
"rewards/margins": 0.005697316490113735, |
|
"rewards/rejected": -0.004733014386147261, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.787234042553192e-06, |
|
"logits/chosen": 0.11128780990839005, |
|
"logits/rejected": 0.16921164095401764, |
|
"logps/chosen": -364.577392578125, |
|
"logps/rejected": -352.7664489746094, |
|
"loss": 0.372, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.002881403313949704, |
|
"rewards/margins": 0.005687831435352564, |
|
"rewards/rejected": -0.008569234982132912, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.999375059004058e-06, |
|
"logits/chosen": 0.08404798805713654, |
|
"logits/rejected": 0.17940422892570496, |
|
"logps/chosen": -375.03900146484375, |
|
"logps/rejected": -333.00775146484375, |
|
"loss": 0.3432, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.0073157488368451595, |
|
"rewards/margins": 0.015402473509311676, |
|
"rewards/rejected": -0.022718224674463272, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.9955571065548795e-06, |
|
"logits/chosen": 0.0811820849776268, |
|
"logits/rejected": 0.12991295754909515, |
|
"logps/chosen": -374.31103515625, |
|
"logps/rejected": -350.5423278808594, |
|
"loss": 0.3508, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.02395443804562092, |
|
"rewards/margins": 0.01998896151781082, |
|
"rewards/rejected": -0.04394339770078659, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.9882736864879e-06, |
|
"logits/chosen": 0.07552331686019897, |
|
"logits/rejected": 0.10529439151287079, |
|
"logps/chosen": -385.13165283203125, |
|
"logps/rejected": -430.96142578125, |
|
"loss": 0.3218, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.06972260773181915, |
|
"rewards/margins": 0.05090062692761421, |
|
"rewards/rejected": -0.12062324583530426, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.977534912960124e-06, |
|
"logits/chosen": 0.1114310622215271, |
|
"logits/rejected": 0.16347981989383698, |
|
"logps/chosen": -484.64544677734375, |
|
"logps/rejected": -562.412109375, |
|
"loss": 0.3292, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.13593843579292297, |
|
"rewards/margins": 0.09484948217868805, |
|
"rewards/rejected": -0.23078791797161102, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.963355698422092e-06, |
|
"logits/chosen": 0.12350474298000336, |
|
"logits/rejected": 0.17375971376895905, |
|
"logps/chosen": -554.6981201171875, |
|
"logps/rejected": -652.0894775390625, |
|
"loss": 0.3099, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.20147164165973663, |
|
"rewards/margins": 0.13413165509700775, |
|
"rewards/rejected": -0.3356032967567444, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.945755732909625e-06, |
|
"logits/chosen": 0.0980851799249649, |
|
"logits/rejected": 0.21910460293293, |
|
"logps/chosen": -511.48046875, |
|
"logps/rejected": -560.3663940429688, |
|
"loss": 0.3019, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.1676788479089737, |
|
"rewards/margins": 0.08112357556819916, |
|
"rewards/rejected": -0.24880242347717285, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.924759456701167e-06, |
|
"logits/chosen": 0.06849007308483124, |
|
"logits/rejected": 0.16746065020561218, |
|
"logps/chosen": -522.7907104492188, |
|
"logps/rejected": -604.654541015625, |
|
"loss": 0.2982, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.16378344595432281, |
|
"rewards/margins": 0.10557951033115387, |
|
"rewards/rejected": -0.2693629264831543, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.900396026378671e-06, |
|
"logits/chosen": 0.09706642478704453, |
|
"logits/rejected": 0.22275814414024353, |
|
"logps/chosen": -541.6332397460938, |
|
"logps/rejected": -655.224609375, |
|
"loss": 0.2733, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -0.1621033251285553, |
|
"rewards/margins": 0.12445729970932007, |
|
"rewards/rejected": -0.286560595035553, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.872699274339169e-06, |
|
"logits/chosen": 0.14361754059791565, |
|
"logits/rejected": 0.1780318021774292, |
|
"logps/chosen": -505.50262451171875, |
|
"logps/rejected": -618.1754760742188, |
|
"loss": 0.2927, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.15032632648944855, |
|
"rewards/margins": 0.1392049789428711, |
|
"rewards/rejected": -0.28953129053115845, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.8417076618132434e-06, |
|
"logits/chosen": 0.07448507845401764, |
|
"logits/rejected": 0.11850146949291229, |
|
"logps/chosen": -497.01702880859375, |
|
"logps/rejected": -640.9612426757812, |
|
"loss": 0.2788, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.1675926297903061, |
|
"rewards/margins": 0.15780577063560486, |
|
"rewards/rejected": -0.32539841532707214, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.807464225455655e-06, |
|
"logits/chosen": 0.1496686488389969, |
|
"logits/rejected": 0.15641534328460693, |
|
"logps/chosen": -477.30438232421875, |
|
"logps/rejected": -642.1568603515625, |
|
"loss": 0.2886, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.18345065414905548, |
|
"rewards/margins": 0.14235511422157288, |
|
"rewards/rejected": -0.32580578327178955, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.770016517582283e-06, |
|
"logits/chosen": 0.029586156830191612, |
|
"logits/rejected": 0.10910207033157349, |
|
"logps/chosen": -545.3736572265625, |
|
"logps/rejected": -619.1397705078125, |
|
"loss": 0.3316, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.2202235460281372, |
|
"rewards/margins": 0.1027408093214035, |
|
"rewards/rejected": -0.3229643702507019, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.7294165401363616e-06, |
|
"logits/chosen": 0.07184920459985733, |
|
"logits/rejected": 0.10738018900156021, |
|
"logps/chosen": -551.0501708984375, |
|
"logps/rejected": -633.4860229492188, |
|
"loss": 0.2955, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.21274058520793915, |
|
"rewards/margins": 0.10448522865772247, |
|
"rewards/rejected": -0.3172258138656616, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.68572067247573e-06, |
|
"logits/chosen": 0.07210972160100937, |
|
"logits/rejected": 0.15474052727222443, |
|
"logps/chosen": -530.9338989257812, |
|
"logps/rejected": -641.7281494140625, |
|
"loss": 0.2748, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -0.20435886085033417, |
|
"rewards/margins": 0.1295766979455948, |
|
"rewards/rejected": -0.33393555879592896, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.638989593081364e-06, |
|
"logits/chosen": 0.1024290919303894, |
|
"logits/rejected": 0.1865706890821457, |
|
"logps/chosen": -584.1736450195312, |
|
"logps/rejected": -625.4171142578125, |
|
"loss": 0.309, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.20201639831066132, |
|
"rewards/margins": 0.09904564172029495, |
|
"rewards/rejected": -0.30106204748153687, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.5892881952959015e-06, |
|
"logits/chosen": 0.02927372418344021, |
|
"logits/rejected": 0.09811622649431229, |
|
"logps/chosen": -533.7764892578125, |
|
"logps/rejected": -629.5748901367188, |
|
"loss": 0.2861, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.18896381556987762, |
|
"rewards/margins": 0.11862628161907196, |
|
"rewards/rejected": -0.3075900971889496, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.536685497209182e-06, |
|
"logits/chosen": -0.005825365893542767, |
|
"logits/rejected": 0.09718175232410431, |
|
"logps/chosen": -493.98675537109375, |
|
"logps/rejected": -583.2804565429688, |
|
"loss": 0.2896, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.20344126224517822, |
|
"rewards/margins": 0.09909246116876602, |
|
"rewards/rejected": -0.30253371596336365, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.481254545815943e-06, |
|
"logits/chosen": 0.044513873755931854, |
|
"logits/rejected": 0.025778833776712418, |
|
"logps/chosen": -558.865966796875, |
|
"logps/rejected": -615.7289428710938, |
|
"loss": 0.2943, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.2074216902256012, |
|
"rewards/margins": 0.11120656877756119, |
|
"rewards/rejected": -0.3186282515525818, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.42307231557875e-06, |
|
"logits/chosen": -0.028864026069641113, |
|
"logits/rejected": 0.07541537284851074, |
|
"logps/chosen": -592.0741577148438, |
|
"logps/rejected": -668.5736083984375, |
|
"loss": 0.3057, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.21832111477851868, |
|
"rewards/margins": 0.11492238938808441, |
|
"rewards/rejected": -0.3332434892654419, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.3622196015370305e-06, |
|
"logits/chosen": 0.0018409714102745056, |
|
"logits/rejected": 0.11235576868057251, |
|
"logps/chosen": -553.7425537109375, |
|
"logps/rejected": -655.2655639648438, |
|
"loss": 0.3027, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.20998337864875793, |
|
"rewards/margins": 0.11225197464227676, |
|
"rewards/rejected": -0.3222353756427765, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.298780907110648e-06, |
|
"logits/chosen": 0.02845548465847969, |
|
"logits/rejected": 0.060444425791502, |
|
"logps/chosen": -538.5903930664062, |
|
"logps/rejected": -638.1737060546875, |
|
"loss": 0.2801, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.21026530861854553, |
|
"rewards/margins": 0.11142082512378693, |
|
"rewards/rejected": -0.32168611884117126, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.23284432675381e-06, |
|
"logits/chosen": 0.03200574219226837, |
|
"logits/rejected": 0.09657691419124603, |
|
"logps/chosen": -566.3619995117188, |
|
"logps/rejected": -676.0072631835938, |
|
"loss": 0.2852, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.209668830037117, |
|
"rewards/margins": 0.10983745008707047, |
|
"rewards/rejected": -0.3195062577724457, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.164501423622277e-06, |
|
"logits/chosen": 0.009218290448188782, |
|
"logits/rejected": 0.07068648189306259, |
|
"logps/chosen": -493.89532470703125, |
|
"logps/rejected": -595.8672485351562, |
|
"loss": 0.2807, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.19013245403766632, |
|
"rewards/margins": 0.11394868791103363, |
|
"rewards/rejected": -0.30408114194869995, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.0938471024237355e-06, |
|
"logits/chosen": 0.040950141847133636, |
|
"logits/rejected": 0.15327873826026917, |
|
"logps/chosen": -557.1869506835938, |
|
"logps/rejected": -644.422119140625, |
|
"loss": 0.286, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.21056047081947327, |
|
"rewards/margins": 0.11874810606241226, |
|
"rewards/rejected": -0.32930856943130493, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.020979477627907e-06, |
|
"logits/chosen": 0.05679699778556824, |
|
"logits/rejected": 0.10968853533267975, |
|
"logps/chosen": -576.4144287109375, |
|
"logps/rejected": -666.2213745117188, |
|
"loss": 0.2967, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.23255427181720734, |
|
"rewards/margins": 0.12515048682689667, |
|
"rewards/rejected": -0.3577048182487488, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.9459997372194105e-06, |
|
"logits/chosen": 0.06001776456832886, |
|
"logits/rejected": 0.1731962114572525, |
|
"logps/chosen": -578.7145385742188, |
|
"logps/rejected": -695.6481323242188, |
|
"loss": 0.2869, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.23581092059612274, |
|
"rewards/margins": 0.13799193501472473, |
|
"rewards/rejected": -0.37380287051200867, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.869012002182573e-06, |
|
"logits/chosen": 0.12720148265361786, |
|
"logits/rejected": 0.11129485070705414, |
|
"logps/chosen": -585.1661376953125, |
|
"logps/rejected": -674.7999877929688, |
|
"loss": 0.2751, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.24305827915668488, |
|
"rewards/margins": 0.1303601711988449, |
|
"rewards/rejected": -0.3734184205532074, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.7901231819133104e-06, |
|
"logits/chosen": 0.046785108745098114, |
|
"logits/rejected": 0.12497253715991974, |
|
"logps/chosen": -613.0939331054688, |
|
"logps/rejected": -778.9556274414062, |
|
"loss": 0.2816, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.26795250177383423, |
|
"rewards/margins": 0.17298182845115662, |
|
"rewards/rejected": -0.44093433022499084, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.709442825758875e-06, |
|
"logits/chosen": -0.016536608338356018, |
|
"logits/rejected": 0.08203241974115372, |
|
"logps/chosen": -619.8163452148438, |
|
"logps/rejected": -756.1831665039062, |
|
"loss": 0.2809, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.28122490644454956, |
|
"rewards/margins": 0.1667798012495041, |
|
"rewards/rejected": -0.44800466299057007, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.6270829708916113e-06, |
|
"logits/chosen": -0.013438811525702477, |
|
"logits/rejected": 0.056597866117954254, |
|
"logps/chosen": -660.5250244140625, |
|
"logps/rejected": -809.8643798828125, |
|
"loss": 0.2683, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.30733537673950195, |
|
"rewards/margins": 0.1618378460407257, |
|
"rewards/rejected": -0.46917325258255005, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.543157986727991e-06, |
|
"logits/chosen": 0.033776335418224335, |
|
"logits/rejected": 0.08476515114307404, |
|
"logps/chosen": -612.4915161132812, |
|
"logps/rejected": -791.068603515625, |
|
"loss": 0.2687, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.298635333776474, |
|
"rewards/margins": 0.14923743903636932, |
|
"rewards/rejected": -0.4478727877140045, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.4577844161089614e-06, |
|
"logits/chosen": 0.04248402267694473, |
|
"logits/rejected": 0.13372144103050232, |
|
"logps/chosen": -653.1471557617188, |
|
"logps/rejected": -754.6251831054688, |
|
"loss": 0.2659, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -0.30667322874069214, |
|
"rewards/margins": 0.1425372064113617, |
|
"rewards/rejected": -0.4492104649543762, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3710808134621577e-06, |
|
"logits/chosen": 0.016178572550415993, |
|
"logits/rejected": 0.09873557835817337, |
|
"logps/chosen": -632.523193359375, |
|
"logps/rejected": -767.3197021484375, |
|
"loss": 0.2715, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.28947144746780396, |
|
"rewards/margins": 0.1752663552761078, |
|
"rewards/rejected": -0.46473783254623413, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.2831675801707126e-06, |
|
"logits/chosen": 0.010262581519782543, |
|
"logits/rejected": 0.09065081179141998, |
|
"logps/chosen": -607.0426025390625, |
|
"logps/rejected": -816.6497192382812, |
|
"loss": 0.2543, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.3082737326622009, |
|
"rewards/margins": 0.19562499225139618, |
|
"rewards/rejected": -0.5038987398147583, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.194166797377289e-06, |
|
"logits/chosen": 0.03939592093229294, |
|
"logits/rejected": 0.051005780696868896, |
|
"logps/chosen": -630.1431884765625, |
|
"logps/rejected": -726.6351318359375, |
|
"loss": 0.2793, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.28141292929649353, |
|
"rewards/margins": 0.13616052269935608, |
|
"rewards/rejected": -0.4175735116004944, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.104202056455501e-06, |
|
"logits/chosen": 0.03237801790237427, |
|
"logits/rejected": 0.128599613904953, |
|
"logps/chosen": -699.2102661132812, |
|
"logps/rejected": -776.4459228515625, |
|
"loss": 0.2784, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.2957838177680969, |
|
"rewards/margins": 0.1545422077178955, |
|
"rewards/rejected": -0.4503260552883148, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.013398287384144e-06, |
|
"logits/chosen": -0.011108310893177986, |
|
"logits/rejected": 0.0545300655066967, |
|
"logps/chosen": -605.3892822265625, |
|
"logps/rejected": -729.7080078125, |
|
"loss": 0.3022, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.27824196219444275, |
|
"rewards/margins": 0.1415918618440628, |
|
"rewards/rejected": -0.41983380913734436, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.9218815852625717e-06, |
|
"logits/chosen": -0.010739525780081749, |
|
"logits/rejected": 0.1061888337135315, |
|
"logps/chosen": -675.0782470703125, |
|
"logps/rejected": -811.0116577148438, |
|
"loss": 0.2543, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.32911890745162964, |
|
"rewards/margins": 0.14946062862873077, |
|
"rewards/rejected": -0.4785795211791992, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.829779035208113e-06, |
|
"logits/chosen": -0.007434132508933544, |
|
"logits/rejected": 0.005777590908110142, |
|
"logps/chosen": -675.082275390625, |
|
"logps/rejected": -852.5006103515625, |
|
"loss": 0.2821, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.3303678631782532, |
|
"rewards/margins": 0.17629772424697876, |
|
"rewards/rejected": -0.5066655874252319, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.737218535878705e-06, |
|
"logits/chosen": -0.010230760090053082, |
|
"logits/rejected": 0.06991882622241974, |
|
"logps/chosen": -657.92919921875, |
|
"logps/rejected": -769.2183227539062, |
|
"loss": 0.2732, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.29096969962120056, |
|
"rewards/margins": 0.1650645136833191, |
|
"rewards/rejected": -0.45603418350219727, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.64432862186579e-06, |
|
"logits/chosen": 0.010941145941615105, |
|
"logits/rejected": 0.06464840471744537, |
|
"logps/chosen": -660.7554931640625, |
|
"logps/rejected": -808.1925048828125, |
|
"loss": 0.2525, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.3170284628868103, |
|
"rewards/margins": 0.16747084259986877, |
|
"rewards/rejected": -0.4844993054866791, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.551238285204126e-06, |
|
"logits/chosen": 0.01938098482787609, |
|
"logits/rejected": 0.10472504049539566, |
|
"logps/chosen": -665.8427734375, |
|
"logps/rejected": -784.5371704101562, |
|
"loss": 0.2657, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.29728442430496216, |
|
"rewards/margins": 0.14796006679534912, |
|
"rewards/rejected": -0.4452444911003113, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4580767962463688e-06, |
|
"logits/chosen": 0.014335828833281994, |
|
"logits/rejected": 0.10104580968618393, |
|
"logps/chosen": -667.510986328125, |
|
"logps/rejected": -799.0164794921875, |
|
"loss": 0.2633, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -0.30110082030296326, |
|
"rewards/margins": 0.16439731419086456, |
|
"rewards/rejected": -0.4654981195926666, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.3649735241511546e-06, |
|
"logits/chosen": 0.028812021017074585, |
|
"logits/rejected": 0.09060634672641754, |
|
"logps/chosen": -620.7320556640625, |
|
"logps/rejected": -778.6414184570312, |
|
"loss": 0.2401, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.2825697362422943, |
|
"rewards/margins": 0.1802894026041031, |
|
"rewards/rejected": -0.4628591537475586, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2720577572339914e-06, |
|
"logits/chosen": -0.015088727697730064, |
|
"logits/rejected": 0.08517689257860184, |
|
"logps/chosen": -646.2489013671875, |
|
"logps/rejected": -777.7031860351562, |
|
"loss": 0.261, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.2979923486709595, |
|
"rewards/margins": 0.16907855868339539, |
|
"rewards/rejected": -0.46707090735435486, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.1794585234303995e-06, |
|
"logits/chosen": 0.0029529170133173466, |
|
"logits/rejected": 0.08159308135509491, |
|
"logps/chosen": -626.0859985351562, |
|
"logps/rejected": -760.864990234375, |
|
"loss": 0.2519, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.2840486168861389, |
|
"rewards/margins": 0.16967108845710754, |
|
"rewards/rejected": -0.4537196755409241, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.0873044111206407e-06, |
|
"logits/chosen": 0.04737887904047966, |
|
"logits/rejected": 0.09217114746570587, |
|
"logps/chosen": -670.5003662109375, |
|
"logps/rejected": -781.8992309570312, |
|
"loss": 0.2568, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.3173016309738159, |
|
"rewards/margins": 0.15325884521007538, |
|
"rewards/rejected": -0.4705604612827301, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 1.9957233905648293e-06, |
|
"logits/chosen": 0.05551145225763321, |
|
"logits/rejected": 0.07084596157073975, |
|
"logps/chosen": -708.875732421875, |
|
"logps/rejected": -875.9669189453125, |
|
"loss": 0.2604, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.3540639579296112, |
|
"rewards/margins": 0.1768617331981659, |
|
"rewards/rejected": -0.5309256315231323, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.904842636196402e-06, |
|
"logits/chosen": 0.05049177259206772, |
|
"logits/rejected": 0.09144983440637589, |
|
"logps/chosen": -662.368896484375, |
|
"logps/rejected": -762.068359375, |
|
"loss": 0.2818, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.2937859892845154, |
|
"rewards/margins": 0.15120849013328552, |
|
"rewards/rejected": -0.4449945092201233, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.814788350020726e-06, |
|
"logits/chosen": 0.013850146904587746, |
|
"logits/rejected": 0.08181043714284897, |
|
"logps/chosen": -655.6898193359375, |
|
"logps/rejected": -799.6488037109375, |
|
"loss": 0.2732, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.2904633581638336, |
|
"rewards/margins": 0.16650694608688354, |
|
"rewards/rejected": -0.45697030425071716, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.725685586364051e-06, |
|
"logits/chosen": 0.026665305718779564, |
|
"logits/rejected": 0.10492020845413208, |
|
"logps/chosen": -667.7979736328125, |
|
"logps/rejected": -860.6505737304688, |
|
"loss": 0.2529, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.2826516032218933, |
|
"rewards/margins": 0.18477080762386322, |
|
"rewards/rejected": -0.4674224257469177, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6376580782162172e-06, |
|
"logits/chosen": 0.04557771235704422, |
|
"logits/rejected": 0.06558578461408615, |
|
"logps/chosen": -630.7008056640625, |
|
"logps/rejected": -797.5521240234375, |
|
"loss": 0.2553, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.29132452607154846, |
|
"rewards/margins": 0.18406057357788086, |
|
"rewards/rejected": -0.4753850996494293, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.550828065408227e-06, |
|
"logits/chosen": 0.0122049106284976, |
|
"logits/rejected": 0.05510292202234268, |
|
"logps/chosen": -702.2847900390625, |
|
"logps/rejected": -777.2984619140625, |
|
"loss": 0.2793, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.32928770780563354, |
|
"rewards/margins": 0.15212389826774597, |
|
"rewards/rejected": -0.4814116060733795, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4653161248633053e-06, |
|
"logits/chosen": 0.027809197083115578, |
|
"logits/rejected": 0.12329945713281631, |
|
"logps/chosen": -693.8055419921875, |
|
"logps/rejected": -849.8357543945312, |
|
"loss": 0.2712, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.3381686210632324, |
|
"rewards/margins": 0.15602417290210724, |
|
"rewards/rejected": -0.4941927492618561, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.381241003157162e-06, |
|
"logits/chosen": 0.018910501152276993, |
|
"logits/rejected": 0.06395964324474335, |
|
"logps/chosen": -668.9143676757812, |
|
"logps/rejected": -784.6621704101562, |
|
"loss": 0.304, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.3322431445121765, |
|
"rewards/margins": 0.13410837948322296, |
|
"rewards/rejected": -0.4663515090942383, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.298719451619979e-06, |
|
"logits/chosen": 0.032313134521245956, |
|
"logits/rejected": 0.09517451375722885, |
|
"logps/chosen": -650.8678588867188, |
|
"logps/rejected": -775.7073974609375, |
|
"loss": 0.2671, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.302242636680603, |
|
"rewards/margins": 0.15537433326244354, |
|
"rewards/rejected": -0.45761698484420776, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2178660642091036e-06, |
|
"logits/chosen": -0.0034284330904483795, |
|
"logits/rejected": 0.08462049067020416, |
|
"logps/chosen": -646.4454345703125, |
|
"logps/rejected": -803.33056640625, |
|
"loss": 0.2708, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.3094111979007721, |
|
"rewards/margins": 0.16766026616096497, |
|
"rewards/rejected": -0.47707146406173706, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.1387931183775821e-06, |
|
"logits/chosen": 0.01865808106958866, |
|
"logits/rejected": 0.07240469753742218, |
|
"logps/chosen": -656.0631103515625, |
|
"logps/rejected": -819.2371826171875, |
|
"loss": 0.2438, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -0.31997841596603394, |
|
"rewards/margins": 0.1749737411737442, |
|
"rewards/rejected": -0.49495214223861694, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.061610419159532e-06, |
|
"logits/chosen": 0.007644282188266516, |
|
"logits/rejected": 0.12147297710180283, |
|
"logps/chosen": -666.0787353515625, |
|
"logps/rejected": -814.1128540039062, |
|
"loss": 0.27, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -0.31804323196411133, |
|
"rewards/margins": 0.1811751127243042, |
|
"rewards/rejected": -0.49921831488609314, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.864251466888364e-07, |
|
"logits/chosen": 0.002938181161880493, |
|
"logits/rejected": 0.05325765162706375, |
|
"logps/chosen": -659.8150634765625, |
|
"logps/rejected": -826.27294921875, |
|
"loss": 0.2679, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -0.33443090319633484, |
|
"rewards/margins": 0.17140033841133118, |
|
"rewards/rejected": -0.505831241607666, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.133417073629288e-07, |
|
"logits/chosen": 0.009498851373791695, |
|
"logits/rejected": 0.10675854980945587, |
|
"logps/chosen": -641.385986328125, |
|
"logps/rejected": -772.0265502929688, |
|
"loss": 0.2745, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.33288660645484924, |
|
"rewards/margins": 0.14625254273414612, |
|
"rewards/rejected": -0.47913914918899536, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 8.424615888583332e-07, |
|
"logits/chosen": 0.01779480278491974, |
|
"logits/rejected": 0.06852717697620392, |
|
"logps/chosen": -658.0755615234375, |
|
"logps/rejected": -790.7129516601562, |
|
"loss": 0.2472, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.3279511630535126, |
|
"rewards/margins": 0.15394431352615356, |
|
"rewards/rejected": -0.48189544677734375, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.738832191993092e-07, |
|
"logits/chosen": -0.008415929041802883, |
|
"logits/rejected": 0.025492772459983826, |
|
"logps/chosen": -639.2633056640625, |
|
"logps/rejected": -759.4827880859375, |
|
"loss": 0.2888, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.31683486700057983, |
|
"rewards/margins": 0.16635778546333313, |
|
"rewards/rejected": -0.48319268226623535, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.077018300752917e-07, |
|
"logits/chosen": 0.036400362849235535, |
|
"logits/rejected": 0.06037065386772156, |
|
"logps/chosen": -635.8643798828125, |
|
"logps/rejected": -806.2174072265625, |
|
"loss": 0.2624, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.3112700879573822, |
|
"rewards/margins": 0.17294862866401672, |
|
"rewards/rejected": -0.48421868681907654, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.440093245969342e-07, |
|
"logits/chosen": 0.006747276522219181, |
|
"logits/rejected": 0.036448314785957336, |
|
"logps/chosen": -682.7285766601562, |
|
"logps/rejected": -770.6795654296875, |
|
"loss": 0.2916, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.3282105326652527, |
|
"rewards/margins": 0.13396592438220978, |
|
"rewards/rejected": -0.46217647194862366, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.828941496744075e-07, |
|
"logits/chosen": -0.008215249516069889, |
|
"logits/rejected": 0.039237380027770996, |
|
"logps/chosen": -615.4769897460938, |
|
"logps/rejected": -744.2301025390625, |
|
"loss": 0.2858, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.31000182032585144, |
|
"rewards/margins": 0.12073155492544174, |
|
"rewards/rejected": -0.43073344230651855, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.244411731951671e-07, |
|
"logits/chosen": -0.018334541469812393, |
|
"logits/rejected": 0.0751979798078537, |
|
"logps/chosen": -653.4116821289062, |
|
"logps/rejected": -775.2245483398438, |
|
"loss": 0.2744, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.3128724694252014, |
|
"rewards/margins": 0.16823306679725647, |
|
"rewards/rejected": -0.4811055064201355, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.6873156617173594e-07, |
|
"logits/chosen": -0.01749856024980545, |
|
"logits/rejected": 0.0061824568547308445, |
|
"logps/chosen": -671.5619506835938, |
|
"logps/rejected": -798.6920776367188, |
|
"loss": 0.2781, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.32307571172714233, |
|
"rewards/margins": 0.15197516977787018, |
|
"rewards/rejected": -0.4750508666038513, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.1584269002318653e-07, |
|
"logits/chosen": -0.031299591064453125, |
|
"logits/rejected": 0.11445782333612442, |
|
"logps/chosen": -633.4989624023438, |
|
"logps/rejected": -759.5948486328125, |
|
"loss": 0.276, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.2988068461418152, |
|
"rewards/margins": 0.14791473746299744, |
|
"rewards/rejected": -0.446721613407135, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.658479891468258e-07, |
|
"logits/chosen": 0.01681383140385151, |
|
"logits/rejected": 0.06748757511377335, |
|
"logps/chosen": -703.5291748046875, |
|
"logps/rejected": -790.0008544921875, |
|
"loss": 0.2645, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.3278062641620636, |
|
"rewards/margins": 0.1466275155544281, |
|
"rewards/rejected": -0.4744337499141693, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.18816888929272e-07, |
|
"logits/chosen": -0.006874733604490757, |
|
"logits/rejected": 0.09821389615535736, |
|
"logps/chosen": -696.3887939453125, |
|
"logps/rejected": -831.1228637695312, |
|
"loss": 0.2456, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.31410273909568787, |
|
"rewards/margins": 0.19106796383857727, |
|
"rewards/rejected": -0.5051707029342651, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.748146993385484e-07, |
|
"logits/chosen": 0.02232988551259041, |
|
"logits/rejected": 0.11882144212722778, |
|
"logps/chosen": -737.1534423828125, |
|
"logps/rejected": -799.8753662109375, |
|
"loss": 0.2901, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.3397114872932434, |
|
"rewards/margins": 0.11829358339309692, |
|
"rewards/rejected": -0.45800504088401794, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.3390252423108077e-07, |
|
"logits/chosen": -0.005960514303296804, |
|
"logits/rejected": 0.08128391206264496, |
|
"logps/chosen": -600.1956787109375, |
|
"logps/rejected": -725.2703857421875, |
|
"loss": 0.2618, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.28923314809799194, |
|
"rewards/margins": 0.14518693089485168, |
|
"rewards/rejected": -0.43442004919052124, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.961371764995243e-07, |
|
"logits/chosen": -0.027128243818879128, |
|
"logits/rejected": 0.08068221807479858, |
|
"logps/chosen": -648.6610107421875, |
|
"logps/rejected": -814.8963012695312, |
|
"loss": 0.2565, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.31799566745758057, |
|
"rewards/margins": 0.18419261276721954, |
|
"rewards/rejected": -0.5021882653236389, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.61571099179261e-07, |
|
"logits/chosen": 0.04195733740925789, |
|
"logits/rejected": 0.05591679736971855, |
|
"logps/chosen": -618.9046630859375, |
|
"logps/rejected": -764.7545776367188, |
|
"loss": 0.2554, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.3056134879589081, |
|
"rewards/margins": 0.1550132781267166, |
|
"rewards/rejected": -0.4606267511844635, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.3025229262312367e-07, |
|
"logits/chosen": 0.007567937020212412, |
|
"logits/rejected": 0.08190431445837021, |
|
"logps/chosen": -702.777587890625, |
|
"logps/rejected": -844.9808349609375, |
|
"loss": 0.2738, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.33942800760269165, |
|
"rewards/margins": 0.16622349619865417, |
|
"rewards/rejected": -0.5056515336036682, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 1.0222424784546853e-07, |
|
"logits/chosen": -0.011982535012066364, |
|
"logits/rejected": 0.029620468616485596, |
|
"logps/chosen": -671.661865234375, |
|
"logps/rejected": -762.5552978515625, |
|
"loss": 0.297, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.355450838804245, |
|
"rewards/margins": 0.11624264717102051, |
|
"rewards/rejected": -0.4716935157775879, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.752588612816553e-08, |
|
"logits/chosen": -0.010390078648924828, |
|
"logits/rejected": 0.059226177632808685, |
|
"logps/chosen": -709.0919189453125, |
|
"logps/rejected": -846.7449340820312, |
|
"loss": 0.296, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.3520728349685669, |
|
"rewards/margins": 0.14244066178798676, |
|
"rewards/rejected": -0.49451351165771484, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 5.619150497236991e-08, |
|
"logits/chosen": 0.030358577147126198, |
|
"logits/rejected": 0.1358805149793625, |
|
"logps/chosen": -658.2210693359375, |
|
"logps/rejected": -764.5792236328125, |
|
"loss": 0.2756, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.31784769892692566, |
|
"rewards/margins": 0.1385163813829422, |
|
"rewards/rejected": -0.45636409521102905, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 3.825073047112743e-08, |
|
"logits/chosen": 0.07536911964416504, |
|
"logits/rejected": 0.11495663970708847, |
|
"logps/chosen": -674.4959716796875, |
|
"logps/rejected": -830.2818603515625, |
|
"loss": 0.2589, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.32807761430740356, |
|
"rewards/margins": 0.15293636918067932, |
|
"rewards/rejected": -0.48101386427879333, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.372847616895685e-08, |
|
"logits/chosen": 0.015253638848662376, |
|
"logits/rejected": 0.014745418913662434, |
|
"logps/chosen": -697.5753173828125, |
|
"logps/rejected": -841.8138427734375, |
|
"loss": 0.2527, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.33767208456993103, |
|
"rewards/margins": 0.15997497737407684, |
|
"rewards/rejected": -0.4976470470428467, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.264490846553279e-08, |
|
"logits/chosen": 0.04343446344137192, |
|
"logits/rejected": 0.05051248520612717, |
|
"logps/chosen": -659.9283447265625, |
|
"logps/rejected": -793.960205078125, |
|
"loss": 0.2637, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.3193858563899994, |
|
"rewards/margins": 0.15680603682994843, |
|
"rewards/rejected": -0.4761919379234314, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 5.015418611516165e-09, |
|
"logits/chosen": -0.0020548023749142885, |
|
"logits/rejected": 0.06288515031337738, |
|
"logps/chosen": -699.4730224609375, |
|
"logps/rejected": -794.502197265625, |
|
"loss": 0.268, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.34633034467697144, |
|
"rewards/margins": 0.136570543050766, |
|
"rewards/rejected": -0.48290085792541504, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 8.506013354186993e-10, |
|
"logits/chosen": 0.013004904612898827, |
|
"logits/rejected": 0.07795099914073944, |
|
"logps/chosen": -636.2744750976562, |
|
"logps/rejected": -769.828125, |
|
"loss": 0.264, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.2970486283302307, |
|
"rewards/margins": 0.1587102711200714, |
|
"rewards/rejected": -0.4557588994503021, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 937, |
|
"total_flos": 0.0, |
|
"train_loss": 0.06840609829860919, |
|
"train_runtime": 2864.3388, |
|
"train_samples_per_second": 10.474, |
|
"train_steps_per_second": 0.327 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 937, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 100, |
|
"total_flos": 0.0, |
|
"train_batch_size": 4, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|