|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 100, |
|
"global_step": 385, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 1.282051282051282e-08, |
|
"logits/chosen": -1.7278180122375488, |
|
"logits/rejected": -1.7377450466156006, |
|
"logps/chosen": -29.553977966308594, |
|
"logps/rejected": -42.813133239746094, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.2820512820512818e-07, |
|
"logits/chosen": -1.8662651777267456, |
|
"logits/rejected": -1.8705823421478271, |
|
"logps/chosen": -37.01448440551758, |
|
"logps/rejected": -33.66516876220703, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.885291415732354e-05, |
|
"rewards/margins": 0.0002720804768614471, |
|
"rewards/rejected": -0.00035093340557068586, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.5641025641025636e-07, |
|
"logits/chosen": -1.9972978830337524, |
|
"logits/rejected": -1.9999454021453857, |
|
"logps/chosen": -29.626657485961914, |
|
"logps/rejected": -29.065475463867188, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.00015535543207079172, |
|
"rewards/margins": 0.00017565980670042336, |
|
"rewards/rejected": -2.0304325516917743e-05, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 3.8461538461538463e-07, |
|
"logits/chosen": -1.9197067022323608, |
|
"logits/rejected": -1.917022466659546, |
|
"logps/chosen": -31.42661476135254, |
|
"logps/rejected": -33.22706985473633, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.0001051858052960597, |
|
"rewards/margins": -1.9103928934782743e-05, |
|
"rewards/rejected": -8.608187636127695e-05, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.999896948438433e-07, |
|
"logits/chosen": -2.0163283348083496, |
|
"logits/rejected": -2.0075771808624268, |
|
"logps/chosen": -32.59123992919922, |
|
"logps/rejected": -32.48072052001953, |
|
"loss": 0.5001, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.00014431352610699832, |
|
"rewards/margins": -0.00046383924200199544, |
|
"rewards/rejected": 0.0003195256576873362, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.987541037542186e-07, |
|
"logits/chosen": -1.8642585277557373, |
|
"logits/rejected": -1.8534786701202393, |
|
"logps/chosen": -33.55145263671875, |
|
"logps/rejected": -35.415096282958984, |
|
"loss": 0.5001, |
|
"rewards/accuracies": 0.375, |
|
"rewards/chosen": 6.026435949024744e-05, |
|
"rewards/margins": -0.0003122163761872798, |
|
"rewards/rejected": 0.00037248071748763323, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.954691471941118e-07, |
|
"logits/chosen": -1.9458900690078735, |
|
"logits/rejected": -1.9478263854980469, |
|
"logps/chosen": -32.57926559448242, |
|
"logps/rejected": -33.194175720214844, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": 0.00020916396169923246, |
|
"rewards/margins": 0.00043710897443816066, |
|
"rewards/rejected": -0.00022794504184275866, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.901618883413548e-07, |
|
"logits/chosen": -2.0795435905456543, |
|
"logits/rejected": -2.084535837173462, |
|
"logps/chosen": -33.986812591552734, |
|
"logps/rejected": -36.573219299316406, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": 1.6729489288991317e-05, |
|
"rewards/margins": 6.016326369717717e-05, |
|
"rewards/rejected": -4.343378896010108e-05, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.828760511501322e-07, |
|
"logits/chosen": -1.9420961141586304, |
|
"logits/rejected": -1.9452673196792603, |
|
"logps/chosen": -34.41278839111328, |
|
"logps/rejected": -34.570655822753906, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": 0.00015023796004243195, |
|
"rewards/margins": -6.727596337441355e-05, |
|
"rewards/rejected": 0.0002175139234168455, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.736716601303429e-07, |
|
"logits/chosen": -1.9506244659423828, |
|
"logits/rejected": -1.9551340341567993, |
|
"logps/chosen": -32.45224380493164, |
|
"logps/rejected": -32.344200134277344, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": 0.0001517904020147398, |
|
"rewards/margins": 4.694242761615897e-06, |
|
"rewards/rejected": 0.00014709614333696663, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.62624545834521e-07, |
|
"logits/chosen": -2.048786163330078, |
|
"logits/rejected": -2.046797275543213, |
|
"logps/chosen": -32.23818588256836, |
|
"logps/rejected": -31.283069610595703, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": 2.6492711185710505e-05, |
|
"rewards/margins": 0.0001950610603671521, |
|
"rewards/rejected": -0.00016856836737133563, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"eval_logits/chosen": -2.243879556655884, |
|
"eval_logits/rejected": -2.2390010356903076, |
|
"eval_logps/chosen": -34.026790618896484, |
|
"eval_logps/rejected": -37.49333572387695, |
|
"eval_loss": 0.5000349879264832, |
|
"eval_rewards/accuracies": 0.5137043595314026, |
|
"eval_rewards/chosen": 7.762637687847018e-05, |
|
"eval_rewards/margins": -0.00015523387992288917, |
|
"eval_rewards/rejected": 0.00023286025680135936, |
|
"eval_runtime": 146.2199, |
|
"eval_samples_per_second": 2.346, |
|
"eval_steps_per_second": 0.294, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.4982572012636904e-07, |
|
"logits/chosen": -2.005380392074585, |
|
"logits/rejected": -2.0029690265655518, |
|
"logps/chosen": -33.23069381713867, |
|
"logps/rejected": -34.01119613647461, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": 0.0001312425301875919, |
|
"rewards/margins": -5.051253901910968e-05, |
|
"rewards/rejected": 0.00018175505101680756, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.353806263777677e-07, |
|
"logits/chosen": -2.016681432723999, |
|
"logits/rejected": -2.0083086490631104, |
|
"logps/chosen": -32.43777847290039, |
|
"logps/rejected": -32.1855354309082, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": 7.132566679501906e-05, |
|
"rewards/margins": 0.00031160234357230365, |
|
"rewards/rejected": -0.0002402766840532422, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.194082707715275e-07, |
|
"logits/chosen": -2.04573392868042, |
|
"logits/rejected": -2.0376956462860107, |
|
"logps/chosen": -30.474706649780273, |
|
"logps/rejected": -32.02512741088867, |
|
"loss": 0.5001, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": 1.0108102287631482e-05, |
|
"rewards/margins": -0.0002687241358216852, |
|
"rewards/rejected": 0.00027883221628144383, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.020402418666621e-07, |
|
"logits/chosen": -1.9768989086151123, |
|
"logits/rejected": -1.9871835708618164, |
|
"logps/chosen": -31.406789779663086, |
|
"logps/rejected": -32.54368209838867, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.00012870995851699263, |
|
"rewards/margins": 0.00021670905698556453, |
|
"rewards/rejected": -8.799910574452952e-05, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.8341962650351185e-07, |
|
"logits/chosen": -1.8907657861709595, |
|
"logits/rejected": -1.8918464183807373, |
|
"logps/chosen": -34.17363739013672, |
|
"logps/rejected": -34.758506774902344, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": 0.00019054378208238631, |
|
"rewards/margins": 0.00020306630176492035, |
|
"rewards/rejected": -1.2522534234449267e-05, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.636998309800572e-07, |
|
"logits/chosen": -1.9421974420547485, |
|
"logits/rejected": -1.9387152194976807, |
|
"logps/chosen": -36.14430618286133, |
|
"logps/rejected": -32.699371337890625, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": 0.0002737110189627856, |
|
"rewards/margins": 3.994655708083883e-05, |
|
"rewards/rejected": 0.00023376448370981961, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.430433172111807e-07, |
|
"logits/chosen": -2.042232036590576, |
|
"logits/rejected": -2.034842014312744, |
|
"logps/chosen": -33.76560974121094, |
|
"logps/rejected": -31.359146118164062, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.0003225207910872996, |
|
"rewards/margins": 0.00039547347114421427, |
|
"rewards/rejected": -7.295267278095707e-05, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.216202642830543e-07, |
|
"logits/chosen": -2.047433376312256, |
|
"logits/rejected": -2.052696704864502, |
|
"logps/chosen": -32.51101303100586, |
|
"logps/rejected": -32.49535369873047, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.00033775376505218446, |
|
"rewards/margins": 0.00043991307029500604, |
|
"rewards/rejected": -0.0001021592179313302, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 2.9960716642946403e-07, |
|
"logits/chosen": -2.048191785812378, |
|
"logits/rejected": -2.0454134941101074, |
|
"logps/chosen": -31.475040435791016, |
|
"logps/rejected": -31.315017700195312, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": 0.00013913081784266979, |
|
"rewards/margins": 0.00010714305244619027, |
|
"rewards/rejected": 3.1987779948394746e-05, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.771853789806683e-07, |
|
"logits/chosen": -1.9189927577972412, |
|
"logits/rejected": -1.9236657619476318, |
|
"logps/chosen": -31.597900390625, |
|
"logps/rejected": -32.774513244628906, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": 8.383009844692424e-05, |
|
"rewards/margins": -8.035473001655191e-05, |
|
"rewards/rejected": 0.0001641848066356033, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"eval_logits/chosen": -2.244427442550659, |
|
"eval_logits/rejected": -2.2395591735839844, |
|
"eval_logps/chosen": -34.002986907958984, |
|
"eval_logps/rejected": -37.49692916870117, |
|
"eval_loss": 0.4999743700027466, |
|
"eval_rewards/accuracies": 0.529900312423706, |
|
"eval_rewards/chosen": 0.00031563788070343435, |
|
"eval_rewards/margins": 0.0001187350062537007, |
|
"eval_rewards/rejected": 0.00019690288172569126, |
|
"eval_runtime": 145.9131, |
|
"eval_samples_per_second": 2.351, |
|
"eval_steps_per_second": 0.295, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.5453962426402e-07, |
|
"logits/chosen": -2.031794786453247, |
|
"logits/rejected": -2.0424611568450928, |
|
"logps/chosen": -31.942611694335938, |
|
"logps/rejected": -33.87322998046875, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": 0.0002869277377612889, |
|
"rewards/margins": 0.0003922684118151665, |
|
"rewards/rejected": -0.00010534075408941135, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.318564697655179e-07, |
|
"logits/chosen": -1.925797462463379, |
|
"logits/rejected": -1.9406683444976807, |
|
"logps/chosen": -30.079837799072266, |
|
"logps/rejected": -31.56089210510254, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": 0.00031700366525910795, |
|
"rewards/margins": 0.0003441544540692121, |
|
"rewards/rejected": -2.715078335313592e-05, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.093227910899832e-07, |
|
"logits/chosen": -1.9836803674697876, |
|
"logits/rejected": -1.9876391887664795, |
|
"logps/chosen": -33.396968841552734, |
|
"logps/rejected": -31.5497989654541, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.00020496857177931815, |
|
"rewards/margins": 0.0002099033008562401, |
|
"rewards/rejected": -4.934729986416642e-06, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.8712423238279356e-07, |
|
"logits/chosen": -1.9834476709365845, |
|
"logits/rejected": -1.9614884853363037, |
|
"logps/chosen": -34.1593017578125, |
|
"logps/rejected": -34.974971771240234, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -5.55333899683319e-05, |
|
"rewards/margins": 0.00018626711971592158, |
|
"rewards/rejected": -0.00024180053151212633, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.654436768970182e-07, |
|
"logits/chosen": -2.0252513885498047, |
|
"logits/rejected": -2.021942615509033, |
|
"logps/chosen": -32.891014099121094, |
|
"logps/rejected": -36.209808349609375, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": 0.00034049636451527476, |
|
"rewards/margins": 0.00018933510000351816, |
|
"rewards/rejected": 0.00015116124995984137, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.444597403062196e-07, |
|
"logits/chosen": -1.891709327697754, |
|
"logits/rejected": -1.889259934425354, |
|
"logps/chosen": -34.162227630615234, |
|
"logps/rejected": -35.52161407470703, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": 0.0002616822312120348, |
|
"rewards/margins": 0.0004316047707106918, |
|
"rewards/rejected": -0.0001699226413620636, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2434529917578887e-07, |
|
"logits/chosen": -1.8765720129013062, |
|
"logits/rejected": -1.8740289211273193, |
|
"logps/chosen": -34.38764190673828, |
|
"logps/rejected": -31.73568344116211, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": 9.958025475498289e-05, |
|
"rewards/margins": -0.00025685896980576217, |
|
"rewards/rejected": 0.00035643921000882983, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.0526606671603521e-07, |
|
"logits/chosen": -1.980292558670044, |
|
"logits/rejected": -1.9696547985076904, |
|
"logps/chosen": -35.3267936706543, |
|
"logps/rejected": -31.829299926757812, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": 0.00029847928090021014, |
|
"rewards/margins": 0.00018156312580686063, |
|
"rewards/rejected": 0.00011691616236930713, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 8.737922755071453e-08, |
|
"logits/chosen": -2.075955390930176, |
|
"logits/rejected": -2.0609400272369385, |
|
"logps/chosen": -30.922338485717773, |
|
"logps/rejected": -32.651031494140625, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": 1.7866250345832668e-05, |
|
"rewards/margins": 9.798279279493727e-06, |
|
"rewards/rejected": 8.067952876444906e-06, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.08321427484816e-08, |
|
"logits/chosen": -1.9476518630981445, |
|
"logits/rejected": -1.945123314857483, |
|
"logps/chosen": -32.88804626464844, |
|
"logps/rejected": -30.81081771850586, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": 0.00039066909812390804, |
|
"rewards/margins": 0.0003307634615339339, |
|
"rewards/rejected": 5.9905636589974165e-05, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"eval_logits/chosen": -2.2444517612457275, |
|
"eval_logits/rejected": -2.239577054977417, |
|
"eval_logps/chosen": -34.00169372558594, |
|
"eval_logps/rejected": -37.48704528808594, |
|
"eval_loss": 0.49999141693115234, |
|
"eval_rewards/accuracies": 0.5365448594093323, |
|
"eval_rewards/chosen": 0.0003285549464635551, |
|
"eval_rewards/margins": 3.2781601476017386e-05, |
|
"eval_rewards/rejected": 0.00029577332315966487, |
|
"eval_runtime": 145.927, |
|
"eval_samples_per_second": 2.35, |
|
"eval_steps_per_second": 0.295, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.576113578589034e-08, |
|
"logits/chosen": -1.929610013961792, |
|
"logits/rejected": -1.9263417720794678, |
|
"logps/chosen": -31.55474281311035, |
|
"logps/rejected": -33.74074935913086, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": 0.0004860997141804546, |
|
"rewards/margins": 0.0004720650613307953, |
|
"rewards/rejected": 1.4034636478754692e-05, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.229036944380912e-08, |
|
"logits/chosen": -1.980975866317749, |
|
"logits/rejected": -1.9686771631240845, |
|
"logps/chosen": -34.552978515625, |
|
"logps/rejected": -33.54259490966797, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 0.0004309539217501879, |
|
"rewards/margins": 0.00046852690866217017, |
|
"rewards/rejected": -3.757303420570679e-05, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 3.053082288996112e-08, |
|
"logits/chosen": -2.016475200653076, |
|
"logits/rejected": -2.015024185180664, |
|
"logps/chosen": -33.44532012939453, |
|
"logps/rejected": -32.47471237182617, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": 0.00040476821595802903, |
|
"rewards/margins": 0.0003560716868378222, |
|
"rewards/rejected": 4.869652912020683e-05, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.05793773749158e-08, |
|
"logits/chosen": -2.10372257232666, |
|
"logits/rejected": -2.0879228115081787, |
|
"logps/chosen": -34.1622200012207, |
|
"logps/rejected": -33.08952331542969, |
|
"loss": 0.5, |
|
"rewards/accuracies": 0.4124999940395355, |
|
"rewards/chosen": 0.00017170429055113345, |
|
"rewards/margins": -0.00011341056961100549, |
|
"rewards/rejected": 0.00028511486016213894, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.251801807404168e-08, |
|
"logits/chosen": -1.9753808975219727, |
|
"logits/rejected": -1.9744256734848022, |
|
"logps/chosen": -33.2431755065918, |
|
"logps/rejected": -32.457496643066406, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": 0.0003892013046424836, |
|
"rewards/margins": 0.00031266576843336225, |
|
"rewards/rejected": 7.653549255337566e-05, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 6.41315865106129e-09, |
|
"logits/chosen": -1.9311378002166748, |
|
"logits/rejected": -1.9414989948272705, |
|
"logps/chosen": -32.21795654296875, |
|
"logps/rejected": -35.26204299926758, |
|
"loss": 0.5001, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.00013266984024085104, |
|
"rewards/margins": -0.0003998771426267922, |
|
"rewards/rejected": 0.00026720724417828023, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.3150941078050324e-09, |
|
"logits/chosen": -2.0701680183410645, |
|
"logits/rejected": -2.063605546951294, |
|
"logps/chosen": -33.66097640991211, |
|
"logps/rejected": -29.19500732421875, |
|
"loss": 0.5001, |
|
"rewards/accuracies": 0.4000000059604645, |
|
"rewards/chosen": -9.009057976072654e-05, |
|
"rewards/margins": -0.0004042644868604839, |
|
"rewards/rejected": 0.0003141739289276302, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 2.575864278703266e-10, |
|
"logits/chosen": -1.9299176931381226, |
|
"logits/rejected": -1.9320815801620483, |
|
"logps/chosen": -34.23712921142578, |
|
"logps/rejected": -30.907180786132812, |
|
"loss": 0.4999, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": 1.9172584870830178e-05, |
|
"rewards/margins": 0.00036473083309829235, |
|
"rewards/rejected": -0.0003455582191236317, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 385, |
|
"total_flos": 0.0, |
|
"train_loss": 0.4999783658362054, |
|
"train_runtime": 3254.8055, |
|
"train_samples_per_second": 0.946, |
|
"train_steps_per_second": 0.118 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 385, |
|
"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 |
|
} |
|
|