|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 3.9881235154394297, |
|
"eval_steps": 500, |
|
"global_step": 312, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.06334125098970704, |
|
"grad_norm": 440.0, |
|
"learning_rate": 7.8125e-06, |
|
"log_odds_chosen": 3.1748764514923096, |
|
"log_odds_ratio": -9.884220123291016, |
|
"logps/chosen": -23.121124267578125, |
|
"logps/rejected": -26.29541015625, |
|
"loss": 178.8174, |
|
"nll_loss": 10.681877136230469, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -1.1560561656951904, |
|
"rewards/margins": 0.1587144136428833, |
|
"rewards/rejected": -1.3147705793380737, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.12668250197941408, |
|
"grad_norm": 340.0, |
|
"learning_rate": 1.5625e-05, |
|
"log_odds_chosen": 6.453858852386475, |
|
"log_odds_ratio": -9.807241439819336, |
|
"logps/chosen": -22.413124084472656, |
|
"logps/rejected": -28.864864349365234, |
|
"loss": 178.3701, |
|
"nll_loss": 10.657767295837402, |
|
"rewards/accuracies": 0.578125, |
|
"rewards/chosen": -1.1206562519073486, |
|
"rewards/margins": 0.32258713245391846, |
|
"rewards/rejected": -1.443243384361267, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.19002375296912113, |
|
"grad_norm": 276.0, |
|
"learning_rate": 2.34375e-05, |
|
"log_odds_chosen": 6.200798988342285, |
|
"log_odds_ratio": -6.756936073303223, |
|
"logps/chosen": -19.177919387817383, |
|
"logps/rejected": -25.377355575561523, |
|
"loss": 148.0439, |
|
"nll_loss": 8.914896011352539, |
|
"rewards/accuracies": 0.559374988079071, |
|
"rewards/chosen": -0.9588959813117981, |
|
"rewards/margins": 0.3099718987941742, |
|
"rewards/rejected": -1.26886785030365, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.25336500395882816, |
|
"grad_norm": 900.0, |
|
"learning_rate": 3.125e-05, |
|
"log_odds_chosen": 2.6450843811035156, |
|
"log_odds_ratio": -5.436456203460693, |
|
"logps/chosen": -13.422134399414062, |
|
"logps/rejected": -16.063411712646484, |
|
"loss": 101.9164, |
|
"nll_loss": 6.097954750061035, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.6711066961288452, |
|
"rewards/margins": 0.1320638358592987, |
|
"rewards/rejected": -0.8031705617904663, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.3167062549485352, |
|
"grad_norm": 93.5, |
|
"learning_rate": 3.90625e-05, |
|
"log_odds_chosen": 0.6309150457382202, |
|
"log_odds_ratio": -1.286942481994629, |
|
"logps/chosen": -3.097271680831909, |
|
"logps/rejected": -3.7033658027648926, |
|
"loss": 42.6767, |
|
"nll_loss": 2.6029443740844727, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.15486359596252441, |
|
"rewards/margins": 0.03030467964708805, |
|
"rewards/rejected": -0.18516826629638672, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.38004750593824227, |
|
"grad_norm": 152.0, |
|
"learning_rate": 4.6875e-05, |
|
"log_odds_chosen": 0.14747688174247742, |
|
"log_odds_ratio": -0.863192081451416, |
|
"logps/chosen": -1.753333330154419, |
|
"logps/rejected": -1.8908967971801758, |
|
"loss": 33.7883, |
|
"nll_loss": 2.068608283996582, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.08766667544841766, |
|
"rewards/margins": 0.006878170184791088, |
|
"rewards/rejected": -0.09454484283924103, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.4433887569279493, |
|
"grad_norm": 320.0, |
|
"learning_rate": 4.998583895833834e-05, |
|
"log_odds_chosen": 0.0781320109963417, |
|
"log_odds_ratio": -0.8847883939743042, |
|
"logps/chosen": -1.601119041442871, |
|
"logps/rejected": -1.677050232887268, |
|
"loss": 30.4438, |
|
"nll_loss": 1.8584957122802734, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.08005593717098236, |
|
"rewards/margins": 0.003796567674726248, |
|
"rewards/rejected": -0.08385250717401505, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.5067300079176563, |
|
"grad_norm": 79.5, |
|
"learning_rate": 4.989935734988098e-05, |
|
"log_odds_chosen": 0.17777779698371887, |
|
"log_odds_ratio": -0.7818454504013062, |
|
"logps/chosen": -1.434004545211792, |
|
"logps/rejected": -1.5908782482147217, |
|
"loss": 29.0365, |
|
"nll_loss": 1.7756904363632202, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.07170023769140244, |
|
"rewards/margins": 0.00784367136657238, |
|
"rewards/rejected": -0.07954390347003937, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.5700712589073634, |
|
"grad_norm": 46.0, |
|
"learning_rate": 4.973453315132592e-05, |
|
"log_odds_chosen": 0.12557587027549744, |
|
"log_odds_ratio": -0.7924664616584778, |
|
"logps/chosen": -1.366229772567749, |
|
"logps/rejected": -1.46891450881958, |
|
"loss": 27.2712, |
|
"nll_loss": 1.6648271083831787, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.06831149756908417, |
|
"rewards/margins": 0.00513423141092062, |
|
"rewards/rejected": -0.07344572246074677, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.6334125098970704, |
|
"grad_norm": 322.0, |
|
"learning_rate": 4.9491884960580894e-05, |
|
"log_odds_chosen": 0.2877430319786072, |
|
"log_odds_ratio": -0.712502121925354, |
|
"logps/chosen": -1.2252671718597412, |
|
"logps/rejected": -1.4585528373718262, |
|
"loss": 25.6522, |
|
"nll_loss": 1.5676389932632446, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.06126336008310318, |
|
"rewards/margins": 0.011664283461868763, |
|
"rewards/rejected": -0.07292764633893967, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.6967537608867775, |
|
"grad_norm": 74.0, |
|
"learning_rate": 4.9172176238625164e-05, |
|
"log_odds_chosen": 0.12641319632530212, |
|
"log_odds_ratio": -0.7623997926712036, |
|
"logps/chosen": -1.275046467781067, |
|
"logps/rejected": -1.3824188709259033, |
|
"loss": 25.7811, |
|
"nll_loss": 1.5732007026672363, |
|
"rewards/accuracies": 0.5218750238418579, |
|
"rewards/chosen": -0.06375232338905334, |
|
"rewards/margins": 0.005368619225919247, |
|
"rewards/rejected": -0.06912095099687576, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.7600950118764845, |
|
"grad_norm": 49.25, |
|
"learning_rate": 4.877641290737884e-05, |
|
"log_odds_chosen": 0.19897113740444183, |
|
"log_odds_ratio": -0.7030541300773621, |
|
"logps/chosen": -1.1849174499511719, |
|
"logps/rejected": -1.3446745872497559, |
|
"loss": 25.5335, |
|
"nll_loss": 1.5606908798217773, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.05924587324261665, |
|
"rewards/margins": 0.007987860590219498, |
|
"rewards/rejected": -0.06723372638225555, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.8234362628661916, |
|
"grad_norm": 73.0, |
|
"learning_rate": 4.8305840184700356e-05, |
|
"log_odds_chosen": 0.2491961419582367, |
|
"log_odds_ratio": -0.693637490272522, |
|
"logps/chosen": -1.1569511890411377, |
|
"logps/rejected": -1.3474953174591064, |
|
"loss": 25.4002, |
|
"nll_loss": 1.5528333187103271, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.057847559452056885, |
|
"rewards/margins": 0.009527203626930714, |
|
"rewards/rejected": -0.06737476587295532, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.8867775138558986, |
|
"grad_norm": 62.75, |
|
"learning_rate": 4.7761938666470403e-05, |
|
"log_odds_chosen": 0.1128716841340065, |
|
"log_odds_ratio": -0.76786869764328, |
|
"logps/chosen": -1.2037312984466553, |
|
"logps/rejected": -1.294019103050232, |
|
"loss": 25.6293, |
|
"nll_loss": 1.5634351968765259, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.060186564922332764, |
|
"rewards/margins": 0.004514397121965885, |
|
"rewards/rejected": -0.06470096111297607, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.9501187648456056, |
|
"grad_norm": 43.25, |
|
"learning_rate": 4.71464196680895e-05, |
|
"log_odds_chosen": 0.15465858578681946, |
|
"log_odds_ratio": -0.7334991693496704, |
|
"logps/chosen": -1.1338040828704834, |
|
"logps/rejected": -1.2355806827545166, |
|
"loss": 23.9048, |
|
"nll_loss": 1.457373023033142, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.05669020861387253, |
|
"rewards/margins": 0.005088826175779104, |
|
"rewards/rejected": -0.06177903339266777, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 1.0253365003958828, |
|
"grad_norm": 39.25, |
|
"learning_rate": 4.6461219840046654e-05, |
|
"log_odds_chosen": 0.339913547039032, |
|
"log_odds_ratio": -0.6567232608795166, |
|
"logps/chosen": -1.0351102352142334, |
|
"logps/rejected": -1.276901364326477, |
|
"loss": 26.0349, |
|
"nll_loss": 1.4135476350784302, |
|
"rewards/accuracies": 0.6111111044883728, |
|
"rewards/chosen": -0.05175551399588585, |
|
"rewards/margins": 0.012089554220438004, |
|
"rewards/rejected": -0.06384507566690445, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 1.0886777513855899, |
|
"grad_norm": 39.5, |
|
"learning_rate": 4.5708495074500414e-05, |
|
"log_odds_chosen": 0.5737000703811646, |
|
"log_odds_ratio": -0.5414241552352905, |
|
"logps/chosen": -0.8744705319404602, |
|
"logps/rejected": -1.2483097314834595, |
|
"loss": 19.751, |
|
"nll_loss": 1.2073694467544556, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.04372352734208107, |
|
"rewards/margins": 0.018691958859562874, |
|
"rewards/rejected": -0.062415480613708496, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 1.152019002375297, |
|
"grad_norm": 41.25, |
|
"learning_rate": 4.489061372204453e-05, |
|
"log_odds_chosen": 0.6578646302223206, |
|
"log_odds_ratio": -0.5210095047950745, |
|
"logps/chosen": -0.8167413473129272, |
|
"logps/rejected": -1.2592405080795288, |
|
"loss": 19.3149, |
|
"nll_loss": 1.181132197380066, |
|
"rewards/accuracies": 0.734375, |
|
"rewards/chosen": -0.04083707183599472, |
|
"rewards/margins": 0.02212495729327202, |
|
"rewards/rejected": -0.06296203285455704, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 1.2153602533650039, |
|
"grad_norm": 27.0, |
|
"learning_rate": 4.401014914000078e-05, |
|
"log_odds_chosen": 0.5433515310287476, |
|
"log_odds_ratio": -0.5697063207626343, |
|
"logps/chosen": -0.8254748582839966, |
|
"logps/rejected": -1.1539433002471924, |
|
"loss": 19.3225, |
|
"nll_loss": 1.179168701171875, |
|
"rewards/accuracies": 0.690625011920929, |
|
"rewards/chosen": -0.04127374291419983, |
|
"rewards/margins": 0.016423424705863, |
|
"rewards/rejected": -0.05769716575741768, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 1.278701504354711, |
|
"grad_norm": 79.5, |
|
"learning_rate": 4.306987159568479e-05, |
|
"log_odds_chosen": 0.592140793800354, |
|
"log_odds_ratio": -0.5446491241455078, |
|
"logps/chosen": -0.8890258073806763, |
|
"logps/rejected": -1.2678449153900146, |
|
"loss": 19.6595, |
|
"nll_loss": 1.2014886140823364, |
|
"rewards/accuracies": 0.703125, |
|
"rewards/chosen": -0.044451288878917694, |
|
"rewards/margins": 0.018940959125757217, |
|
"rewards/rejected": -0.06339225172996521, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.342042755344418, |
|
"grad_norm": 26.25, |
|
"learning_rate": 4.207273955012018e-05, |
|
"log_odds_chosen": 0.5168221592903137, |
|
"log_odds_ratio": -0.5695356130599976, |
|
"logps/chosen": -0.8439987897872925, |
|
"logps/rejected": -1.158204197883606, |
|
"loss": 18.8408, |
|
"nll_loss": 1.1490731239318848, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.04219993203878403, |
|
"rewards/margins": 0.01571027934551239, |
|
"rewards/rejected": -0.05791021138429642, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 1.405384006334125, |
|
"grad_norm": 32.0, |
|
"learning_rate": 4.10218903496256e-05, |
|
"log_odds_chosen": 0.5098962187767029, |
|
"log_odds_ratio": -0.5828264355659485, |
|
"logps/chosen": -0.8689939379692078, |
|
"logps/rejected": -1.1937323808670044, |
|
"loss": 20.1778, |
|
"nll_loss": 1.231968641281128, |
|
"rewards/accuracies": 0.6781250238418579, |
|
"rewards/chosen": -0.04344969242811203, |
|
"rewards/margins": 0.0162369217723608, |
|
"rewards/rejected": -0.05968661978840828, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 1.468725257323832, |
|
"grad_norm": 34.0, |
|
"learning_rate": 3.992063035456259e-05, |
|
"log_odds_chosen": 0.6211134195327759, |
|
"log_odds_ratio": -0.5309170484542847, |
|
"logps/chosen": -0.7933008074760437, |
|
"logps/rejected": -1.1834070682525635, |
|
"loss": 18.7885, |
|
"nll_loss": 1.14773428440094, |
|
"rewards/accuracies": 0.734375, |
|
"rewards/chosen": -0.039665043354034424, |
|
"rewards/margins": 0.01950531266629696, |
|
"rewards/rejected": -0.059170354157686234, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 1.5320665083135392, |
|
"grad_norm": 28.5, |
|
"learning_rate": 3.8772424536302564e-05, |
|
"log_odds_chosen": 0.6670337915420532, |
|
"log_odds_ratio": -0.5194506049156189, |
|
"logps/chosen": -0.7793987989425659, |
|
"logps/rejected": -1.1679680347442627, |
|
"loss": 18.5035, |
|
"nll_loss": 1.1304987668991089, |
|
"rewards/accuracies": 0.734375, |
|
"rewards/chosen": -0.038969945162534714, |
|
"rewards/margins": 0.01942845992743969, |
|
"rewards/rejected": -0.058398403227329254, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.5954077593032463, |
|
"grad_norm": 24.375, |
|
"learning_rate": 3.758088557514501e-05, |
|
"log_odds_chosen": 0.5610660314559937, |
|
"log_odds_ratio": -0.552033543586731, |
|
"logps/chosen": -0.795187771320343, |
|
"logps/rejected": -1.1401742696762085, |
|
"loss": 18.7339, |
|
"nll_loss": 1.1432664394378662, |
|
"rewards/accuracies": 0.684374988079071, |
|
"rewards/chosen": -0.03975939005613327, |
|
"rewards/margins": 0.017249327152967453, |
|
"rewards/rejected": -0.05700872093439102, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 1.6587490102929534, |
|
"grad_norm": 29.625, |
|
"learning_rate": 3.634976249348867e-05, |
|
"log_odds_chosen": 0.4703447222709656, |
|
"log_odds_ratio": -0.5870831608772278, |
|
"logps/chosen": -0.7940428256988525, |
|
"logps/rejected": -1.0778682231903076, |
|
"loss": 18.7796, |
|
"nll_loss": 1.1443703174591064, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.039702143520116806, |
|
"rewards/margins": 0.014191272668540478, |
|
"rewards/rejected": -0.05389341711997986, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 1.7220902612826603, |
|
"grad_norm": 28.875, |
|
"learning_rate": 3.508292886002013e-05, |
|
"log_odds_chosen": 0.5749887228012085, |
|
"log_odds_ratio": -0.5368548631668091, |
|
"logps/chosen": -0.8258606195449829, |
|
"logps/rejected": -1.1885448694229126, |
|
"loss": 18.9446, |
|
"nll_loss": 1.1571968793869019, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.04129302501678467, |
|
"rewards/margins": 0.018134210258722305, |
|
"rewards/rejected": -0.05942724272608757, |
|
"step": 135 |
|
}, |
|
{ |
|
"epoch": 1.7854315122723674, |
|
"grad_norm": 26.75, |
|
"learning_rate": 3.378437060203357e-05, |
|
"log_odds_chosen": 0.5498477220535278, |
|
"log_odds_ratio": -0.5687417984008789, |
|
"logps/chosen": -0.81671142578125, |
|
"logps/rejected": -1.1441940069198608, |
|
"loss": 18.5467, |
|
"nll_loss": 1.130732774734497, |
|
"rewards/accuracies": 0.6781250238418579, |
|
"rewards/chosen": -0.04083556681871414, |
|
"rewards/margins": 0.0163741298019886, |
|
"rewards/rejected": -0.05720970779657364, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.8487727632620743, |
|
"grad_norm": 36.5, |
|
"learning_rate": 3.24581734642289e-05, |
|
"log_odds_chosen": 0.7345950603485107, |
|
"log_odds_ratio": -0.5215600728988647, |
|
"logps/chosen": -0.7965916991233826, |
|
"logps/rejected": -1.2689704895019531, |
|
"loss": 18.424, |
|
"nll_loss": 1.125421404838562, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.03982958942651749, |
|
"rewards/margins": 0.02361893653869629, |
|
"rewards/rejected": -0.06344853341579437, |
|
"step": 145 |
|
}, |
|
{ |
|
"epoch": 1.9121140142517814, |
|
"grad_norm": 30.5, |
|
"learning_rate": 3.110851015344735e-05, |
|
"log_odds_chosen": 0.5858422517776489, |
|
"log_odds_ratio": -0.5331650376319885, |
|
"logps/chosen": -0.8209154009819031, |
|
"logps/rejected": -1.1717817783355713, |
|
"loss": 18.0192, |
|
"nll_loss": 1.0995434522628784, |
|
"rewards/accuracies": 0.7281249761581421, |
|
"rewards/chosen": -0.04104577749967575, |
|
"rewards/margins": 0.017543310299515724, |
|
"rewards/rejected": -0.058589089661836624, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.9754552652414885, |
|
"grad_norm": 31.5, |
|
"learning_rate": 2.9739627209791965e-05, |
|
"log_odds_chosen": 0.6027151346206665, |
|
"log_odds_ratio": -0.5478641390800476, |
|
"logps/chosen": -0.8092836141586304, |
|
"logps/rejected": -1.1755163669586182, |
|
"loss": 18.8905, |
|
"nll_loss": 1.1532632112503052, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.04046417772769928, |
|
"rewards/margins": 0.0183116365224123, |
|
"rewards/rejected": -0.05877581983804703, |
|
"step": 155 |
|
}, |
|
{ |
|
"epoch": 2.0506730007917655, |
|
"grad_norm": 26.625, |
|
"learning_rate": 2.8355831645441388e-05, |
|
"log_odds_chosen": 0.9976516962051392, |
|
"log_odds_ratio": -0.412594735622406, |
|
"logps/chosen": -0.6453574299812317, |
|
"logps/rejected": -1.2032675743103027, |
|
"loss": 17.6716, |
|
"nll_loss": 0.9611276984214783, |
|
"rewards/accuracies": 0.8333333134651184, |
|
"rewards/chosen": -0.032267872244119644, |
|
"rewards/margins": 0.027895506471395493, |
|
"rewards/rejected": -0.06016337499022484, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 2.1140142517814726, |
|
"grad_norm": 25.125, |
|
"learning_rate": 2.6961477393196126e-05, |
|
"log_odds_chosen": 1.592689871788025, |
|
"log_odds_ratio": -0.27941542863845825, |
|
"logps/chosen": -0.5471819639205933, |
|
"logps/rejected": -1.4586117267608643, |
|
"loss": 13.9962, |
|
"nll_loss": 0.8607938885688782, |
|
"rewards/accuracies": 0.9281250238418579, |
|
"rewards/chosen": -0.027359094470739365, |
|
"rewards/margins": 0.04557149484753609, |
|
"rewards/rejected": -0.07293058931827545, |
|
"step": 165 |
|
}, |
|
{ |
|
"epoch": 2.1773555027711797, |
|
"grad_norm": 23.125, |
|
"learning_rate": 2.556095160739513e-05, |
|
"log_odds_chosen": 1.3663856983184814, |
|
"log_odds_ratio": -0.3143647313117981, |
|
"logps/chosen": -0.5476871728897095, |
|
"logps/rejected": -1.304485559463501, |
|
"loss": 13.0544, |
|
"nll_loss": 0.800182044506073, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -0.027384359389543533, |
|
"rewards/margins": 0.037839919328689575, |
|
"rewards/rejected": -0.06522427499294281, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 2.240696753760887, |
|
"grad_norm": 26.75, |
|
"learning_rate": 2.415866086030516e-05, |
|
"log_odds_chosen": 1.4820417165756226, |
|
"log_odds_ratio": -0.29762935638427734, |
|
"logps/chosen": -0.5468381643295288, |
|
"logps/rejected": -1.3758976459503174, |
|
"loss": 13.3913, |
|
"nll_loss": 0.8220744132995605, |
|
"rewards/accuracies": 0.921875, |
|
"rewards/chosen": -0.02734190784394741, |
|
"rewards/margins": 0.04145296663045883, |
|
"rewards/rejected": -0.06879488378763199, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 2.304038004750594, |
|
"grad_norm": 26.25, |
|
"learning_rate": 2.2759017277414166e-05, |
|
"log_odds_chosen": 1.4541139602661133, |
|
"log_odds_ratio": -0.30560511350631714, |
|
"logps/chosen": -0.5400351285934448, |
|
"logps/rejected": -1.3224079608917236, |
|
"loss": 13.3401, |
|
"nll_loss": 0.818473219871521, |
|
"rewards/accuracies": 0.8968750238418579, |
|
"rewards/chosen": -0.0270017571747303, |
|
"rewards/margins": 0.03911864012479782, |
|
"rewards/rejected": -0.06612040102481842, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 2.367379255740301, |
|
"grad_norm": 45.0, |
|
"learning_rate": 2.136642465525219e-05, |
|
"log_odds_chosen": 1.3566925525665283, |
|
"log_odds_ratio": -0.312085896730423, |
|
"logps/chosen": -0.5589757561683655, |
|
"logps/rejected": -1.2790095806121826, |
|
"loss": 13.2166, |
|
"nll_loss": 0.8104313015937805, |
|
"rewards/accuracies": 0.8968750238418579, |
|
"rewards/chosen": -0.027948787435889244, |
|
"rewards/margins": 0.03600168973207474, |
|
"rewards/rejected": -0.06395047903060913, |
|
"step": 185 |
|
}, |
|
{ |
|
"epoch": 2.4307205067300077, |
|
"grad_norm": 31.0, |
|
"learning_rate": 1.9985264605418183e-05, |
|
"log_odds_chosen": 1.4590914249420166, |
|
"log_odds_ratio": -0.3126595616340637, |
|
"logps/chosen": -0.5508120059967041, |
|
"logps/rejected": -1.3448957204818726, |
|
"loss": 13.3402, |
|
"nll_loss": 0.8181284666061401, |
|
"rewards/accuracies": 0.890625, |
|
"rewards/chosen": -0.027540603652596474, |
|
"rewards/margins": 0.03970418870449066, |
|
"rewards/rejected": -0.06724478304386139, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 2.494061757719715, |
|
"grad_norm": 25.375, |
|
"learning_rate": 1.8619882768408935e-05, |
|
"log_odds_chosen": 1.5256664752960205, |
|
"log_odds_ratio": -0.29186081886291504, |
|
"logps/chosen": -0.5439263582229614, |
|
"logps/rejected": -1.4009654521942139, |
|
"loss": 13.2504, |
|
"nll_loss": 0.8135560750961304, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -0.02719631791114807, |
|
"rewards/margins": 0.04285195842385292, |
|
"rewards/rejected": -0.07004828751087189, |
|
"step": 195 |
|
}, |
|
{ |
|
"epoch": 2.557403008709422, |
|
"grad_norm": 24.25, |
|
"learning_rate": 1.7274575140626318e-05, |
|
"log_odds_chosen": 1.4801175594329834, |
|
"log_odds_ratio": -0.3182724118232727, |
|
"logps/chosen": -0.5556176900863647, |
|
"logps/rejected": -1.355858564376831, |
|
"loss": 13.7318, |
|
"nll_loss": 0.8423222303390503, |
|
"rewards/accuracies": 0.8968750238418579, |
|
"rewards/chosen": -0.027780886739492416, |
|
"rewards/margins": 0.04001203924417496, |
|
"rewards/rejected": -0.06779292970895767, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 2.620744259699129, |
|
"grad_norm": 21.25, |
|
"learning_rate": 1.59535745575832e-05, |
|
"log_odds_chosen": 1.538774847984314, |
|
"log_odds_ratio": -0.2739660441875458, |
|
"logps/chosen": -0.5107089281082153, |
|
"logps/rejected": -1.3470090627670288, |
|
"loss": 12.7697, |
|
"nll_loss": 0.7844074368476868, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -0.025535445660352707, |
|
"rewards/margins": 0.041815001517534256, |
|
"rewards/rejected": -0.06735045462846756, |
|
"step": 205 |
|
}, |
|
{ |
|
"epoch": 2.684085510688836, |
|
"grad_norm": 26.5, |
|
"learning_rate": 1.466103737583699e-05, |
|
"log_odds_chosen": 1.6077247858047485, |
|
"log_odds_ratio": -0.27898281812667847, |
|
"logps/chosen": -0.5174364447593689, |
|
"logps/rejected": -1.3971433639526367, |
|
"loss": 12.9226, |
|
"nll_loss": 0.7937139272689819, |
|
"rewards/accuracies": 0.909375011920929, |
|
"rewards/chosen": -0.025871824473142624, |
|
"rewards/margins": 0.04398534819483757, |
|
"rewards/rejected": -0.0698571726679802, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 2.7474267616785433, |
|
"grad_norm": 37.5, |
|
"learning_rate": 1.340103039555415e-05, |
|
"log_odds_chosen": 1.4663047790527344, |
|
"log_odds_ratio": -0.3185238242149353, |
|
"logps/chosen": -0.5488572716712952, |
|
"logps/rejected": -1.336905837059021, |
|
"loss": 13.052, |
|
"nll_loss": 0.7998241782188416, |
|
"rewards/accuracies": 0.903124988079071, |
|
"rewards/chosen": -0.02744286321103573, |
|
"rewards/margins": 0.03940243273973465, |
|
"rewards/rejected": -0.06684528291225433, |
|
"step": 215 |
|
}, |
|
{ |
|
"epoch": 2.81076801266825, |
|
"grad_norm": 33.25, |
|
"learning_rate": 1.217751806485235e-05, |
|
"log_odds_chosen": 1.548749327659607, |
|
"log_odds_ratio": -0.2925991415977478, |
|
"logps/chosen": -0.5320712327957153, |
|
"logps/rejected": -1.400617241859436, |
|
"loss": 13.2408, |
|
"nll_loss": 0.8129196166992188, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -0.026603559032082558, |
|
"rewards/margins": 0.043427299708127975, |
|
"rewards/rejected": -0.07003085315227509, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 2.8741092636579575, |
|
"grad_norm": 28.25, |
|
"learning_rate": 1.0994350006179931e-05, |
|
"log_odds_chosen": 1.4858758449554443, |
|
"log_odds_ratio": -0.29021698236465454, |
|
"logps/chosen": -0.5439127683639526, |
|
"logps/rejected": -1.3659403324127197, |
|
"loss": 13.1169, |
|
"nll_loss": 0.8052948117256165, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -0.027195638045668602, |
|
"rewards/margins": 0.04110138118267059, |
|
"rewards/rejected": -0.06829701364040375, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 2.937450514647664, |
|
"grad_norm": 23.125, |
|
"learning_rate": 9.855248903979506e-06, |
|
"log_odds_chosen": 1.4533650875091553, |
|
"log_odds_ratio": -0.2951867878437042, |
|
"logps/chosen": -0.5295075178146362, |
|
"logps/rejected": -1.2960364818572998, |
|
"loss": 13.1057, |
|
"nll_loss": 0.8043462038040161, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -0.026475373655557632, |
|
"rewards/margins": 0.0383264534175396, |
|
"rewards/rejected": -0.06480182707309723, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 3.0126682501979416, |
|
"grad_norm": 29.375, |
|
"learning_rate": 8.763798791745411e-06, |
|
"log_odds_chosen": 1.5862672328948975, |
|
"log_odds_ratio": -0.2758994698524475, |
|
"logps/chosen": -0.5054166316986084, |
|
"logps/rejected": -1.356021761894226, |
|
"loss": 14.3167, |
|
"nll_loss": 0.7815780639648438, |
|
"rewards/accuracies": 0.9222221970558167, |
|
"rewards/chosen": -0.02527083270251751, |
|
"rewards/margins": 0.042530253529548645, |
|
"rewards/rejected": -0.0678010806441307, |
|
"step": 235 |
|
}, |
|
{ |
|
"epoch": 3.0760095011876483, |
|
"grad_norm": 21.25, |
|
"learning_rate": 7.723433775328384e-06, |
|
"log_odds_chosen": 2.0705292224884033, |
|
"log_odds_ratio": -0.19949007034301758, |
|
"logps/chosen": -0.4086482524871826, |
|
"logps/rejected": -1.4631153345108032, |
|
"loss": 10.5377, |
|
"nll_loss": 0.6486285924911499, |
|
"rewards/accuracies": 0.965624988079071, |
|
"rewards/chosen": -0.02043241262435913, |
|
"rewards/margins": 0.05272335559129715, |
|
"rewards/rejected": -0.07315576076507568, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 3.1393507521773554, |
|
"grad_norm": 30.0, |
|
"learning_rate": 6.737427227968063e-06, |
|
"log_odds_chosen": 2.158939838409424, |
|
"log_odds_ratio": -0.17655470967292786, |
|
"logps/chosen": -0.36677083373069763, |
|
"logps/rejected": -1.4107964038848877, |
|
"loss": 10.2002, |
|
"nll_loss": 0.6286830902099609, |
|
"rewards/accuracies": 0.9781249761581421, |
|
"rewards/chosen": -0.01833854243159294, |
|
"rewards/margins": 0.0522012785077095, |
|
"rewards/rejected": -0.07053982466459274, |
|
"step": 245 |
|
}, |
|
{ |
|
"epoch": 3.2026920031670625, |
|
"grad_norm": 24.375, |
|
"learning_rate": 5.808881491049723e-06, |
|
"log_odds_chosen": 2.1467604637145996, |
|
"log_odds_ratio": -0.21470721065998077, |
|
"logps/chosen": -0.45862817764282227, |
|
"logps/rejected": -1.5745422840118408, |
|
"loss": 10.6424, |
|
"nll_loss": 0.6544127464294434, |
|
"rewards/accuracies": 0.953125, |
|
"rewards/chosen": -0.022931408137083054, |
|
"rewards/margins": 0.05579570680856705, |
|
"rewards/rejected": -0.0787271112203598, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 3.2660332541567696, |
|
"grad_norm": 28.375, |
|
"learning_rate": 4.940718112990553e-06, |
|
"log_odds_chosen": 2.224365711212158, |
|
"log_odds_ratio": -0.18038669228553772, |
|
"logps/chosen": -0.38552433252334595, |
|
"logps/rejected": -1.5169192552566528, |
|
"loss": 10.064, |
|
"nll_loss": 0.6199790835380554, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.019276214763522148, |
|
"rewards/margins": 0.056569743901491165, |
|
"rewards/rejected": -0.07584595680236816, |
|
"step": 255 |
|
}, |
|
{ |
|
"epoch": 3.3293745051464767, |
|
"grad_norm": 22.875, |
|
"learning_rate": 4.135668656967434e-06, |
|
"log_odds_chosen": 2.1139309406280518, |
|
"log_odds_ratio": -0.18317027390003204, |
|
"logps/chosen": -0.42459002137184143, |
|
"logps/rejected": -1.4972463846206665, |
|
"loss": 10.2835, |
|
"nll_loss": 0.6335573792457581, |
|
"rewards/accuracies": 0.965624988079071, |
|
"rewards/chosen": -0.02122950181365013, |
|
"rewards/margins": 0.053632818162441254, |
|
"rewards/rejected": -0.07486231625080109, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 3.392715756136184, |
|
"grad_norm": 26.5, |
|
"learning_rate": 3.3962661064086354e-06, |
|
"log_odds_chosen": 2.135136842727661, |
|
"log_odds_ratio": -0.18204157054424286, |
|
"logps/chosen": -0.40990591049194336, |
|
"logps/rejected": -1.4916514158248901, |
|
"loss": 10.287, |
|
"nll_loss": 0.6338369250297546, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.020495299249887466, |
|
"rewards/margins": 0.05408727377653122, |
|
"rewards/rejected": -0.07458256930112839, |
|
"step": 265 |
|
}, |
|
{ |
|
"epoch": 3.456057007125891, |
|
"grad_norm": 22.625, |
|
"learning_rate": 2.7248368952908053e-06, |
|
"log_odds_chosen": 2.2748658657073975, |
|
"log_odds_ratio": -0.17176181077957153, |
|
"logps/chosen": -0.41875600814819336, |
|
"logps/rejected": -1.62033212184906, |
|
"loss": 10.5217, |
|
"nll_loss": 0.6490176916122437, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.020937800407409668, |
|
"rewards/margins": 0.060078807175159454, |
|
"rewards/rejected": -0.08101661503314972, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 3.519398258115598, |
|
"grad_norm": 21.25, |
|
"learning_rate": 2.1234935883170047e-06, |
|
"log_odds_chosen": 2.311600685119629, |
|
"log_odds_ratio": -0.17423459887504578, |
|
"logps/chosen": -0.3958762288093567, |
|
"logps/rejected": -1.5846991539001465, |
|
"loss": 9.5479, |
|
"nll_loss": 0.5880299806594849, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.019793810322880745, |
|
"rewards/margins": 0.05944114923477173, |
|
"rewards/rejected": -0.07923495769500732, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 3.5827395091053047, |
|
"grad_norm": 25.875, |
|
"learning_rate": 1.59412823400657e-06, |
|
"log_odds_chosen": 2.3392279148101807, |
|
"log_odds_ratio": -0.17188401520252228, |
|
"logps/chosen": -0.40492624044418335, |
|
"logps/rejected": -1.6255334615707397, |
|
"loss": 10.1858, |
|
"nll_loss": 0.628017783164978, |
|
"rewards/accuracies": 0.9593750238418579, |
|
"rewards/chosen": -0.020246315747499466, |
|
"rewards/margins": 0.061030369251966476, |
|
"rewards/rejected": -0.08127669245004654, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 3.646080760095012, |
|
"grad_norm": 26.625, |
|
"learning_rate": 1.138406411610482e-06, |
|
"log_odds_chosen": 2.3245720863342285, |
|
"log_odds_ratio": -0.16916075348854065, |
|
"logps/chosen": -0.37623462080955505, |
|
"logps/rejected": -1.5502853393554688, |
|
"loss": 9.6503, |
|
"nll_loss": 0.5946874618530273, |
|
"rewards/accuracies": 0.9468749761581421, |
|
"rewards/chosen": -0.018811732530593872, |
|
"rewards/margins": 0.058702535927295685, |
|
"rewards/rejected": -0.07751427590847015, |
|
"step": 285 |
|
}, |
|
{ |
|
"epoch": 3.709422011084719, |
|
"grad_norm": 21.875, |
|
"learning_rate": 7.577619905828282e-07, |
|
"log_odds_chosen": 2.351839065551758, |
|
"log_odds_ratio": -0.16314415633678436, |
|
"logps/chosen": -0.3871734142303467, |
|
"logps/rejected": -1.6283880472183228, |
|
"loss": 9.8769, |
|
"nll_loss": 0.6091512441635132, |
|
"rewards/accuracies": 0.971875011920929, |
|
"rewards/chosen": -0.019358674064278603, |
|
"rewards/margins": 0.06206073611974716, |
|
"rewards/rejected": -0.08141940087080002, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 3.772763262074426, |
|
"grad_norm": 23.0, |
|
"learning_rate": 4.5339261909704e-07, |
|
"log_odds_chosen": 2.405986785888672, |
|
"log_odds_ratio": -0.16327905654907227, |
|
"logps/chosen": -0.4028860628604889, |
|
"logps/rejected": -1.711242914199829, |
|
"loss": 10.0405, |
|
"nll_loss": 0.6193649172782898, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.020144306123256683, |
|
"rewards/margins": 0.06541784107685089, |
|
"rewards/rejected": -0.08556214720010757, |
|
"step": 295 |
|
}, |
|
{ |
|
"epoch": 3.836104513064133, |
|
"grad_norm": 23.375, |
|
"learning_rate": 2.262559558016325e-07, |
|
"log_odds_chosen": 2.3016839027404785, |
|
"log_odds_ratio": -0.16708031296730042, |
|
"logps/chosen": -0.38780677318573, |
|
"logps/rejected": -1.5517771244049072, |
|
"loss": 9.6048, |
|
"nll_loss": 0.5919488072395325, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.01939033903181553, |
|
"rewards/margins": 0.058198511600494385, |
|
"rewards/rejected": -0.07758884876966476, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 3.89944576405384, |
|
"grad_norm": 24.75, |
|
"learning_rate": 7.706665667180091e-08, |
|
"log_odds_chosen": 2.3498542308807373, |
|
"log_odds_ratio": -0.1596686691045761, |
|
"logps/chosen": -0.39143842458724976, |
|
"logps/rejected": -1.6413482427597046, |
|
"loss": 10.6228, |
|
"nll_loss": 0.6559435725212097, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.019571922719478607, |
|
"rewards/margins": 0.06249549239873886, |
|
"rewards/rejected": -0.08206741511821747, |
|
"step": 305 |
|
}, |
|
{ |
|
"epoch": 3.9627870150435474, |
|
"grad_norm": 23.625, |
|
"learning_rate": 6.294126437336734e-09, |
|
"log_odds_chosen": 2.3062288761138916, |
|
"log_odds_ratio": -0.17084279656410217, |
|
"logps/chosen": -0.3852413594722748, |
|
"logps/rejected": -1.5706732273101807, |
|
"loss": 10.5082, |
|
"nll_loss": 0.6482213139533997, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.019262069836258888, |
|
"rewards/margins": 0.059271592646837234, |
|
"rewards/rejected": -0.07853366434574127, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 3.9881235154394297, |
|
"step": 312, |
|
"total_flos": 0.0, |
|
"train_loss": 25.76104594499637, |
|
"train_runtime": 3640.7426, |
|
"train_samples_per_second": 5.548, |
|
"train_steps_per_second": 0.086 |
|
} |
|
], |
|
"logging_steps": 5, |
|
"max_steps": 312, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 4, |
|
"save_steps": 100, |
|
"stateful_callbacks": { |
|
"TrainerControl": { |
|
"args": { |
|
"should_epoch_stop": false, |
|
"should_evaluate": false, |
|
"should_log": false, |
|
"should_save": true, |
|
"should_training_stop": true |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 0.0, |
|
"train_batch_size": 1, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|