diff --git "a/trainer_state.json" "b/trainer_state.json" new file mode 100644--- /dev/null +++ "b/trainer_state.json" @@ -0,0 +1,2981 @@ +{ + "best_metric": 0.6428677439689636, + "best_model_checkpoint": "/mnt/data/shesj/Trained/RL4CoT/DPO/llama2fullcontinue_largebeta_initialData_iter1Self_resetScoreExp_iter1_2_lowerLr16.json/checkpoint-1000", + "epoch": 0.5, + "eval_steps": 100, + "global_step": 1000, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0, + "learning_rate": 5e-09, + "logits/chosen": -1.5196828842163086, + "logits/rejected": -1.53714919090271, + "logps/chosen": -15.696101188659668, + "logps/rejected": -19.717897415161133, + "loss": 0.693, + "rewards/accuracies": 0.3062500059604645, + "rewards/chosen": -4.1580526158213615e-05, + "rewards/margins": 0.0009152223356068134, + "rewards/rejected": -0.000956802919972688, + "step": 5 + }, + { + "epoch": 0.01, + "learning_rate": 1e-08, + "logits/chosen": -1.4667831659317017, + "logits/rejected": -1.474163293838501, + "logps/chosen": -13.819122314453125, + "logps/rejected": -15.621783256530762, + "loss": 0.6933, + "rewards/accuracies": 0.46875, + "rewards/chosen": -0.0003026240156032145, + "rewards/margins": 0.0002748106198851019, + "rewards/rejected": -0.0005774348974227905, + "step": 10 + }, + { + "epoch": 0.01, + "learning_rate": 1.5e-08, + "logits/chosen": -1.4452192783355713, + "logits/rejected": -1.4530740976333618, + "logps/chosen": -16.253719329833984, + "logps/rejected": -19.002716064453125, + "loss": 0.693, + "rewards/accuracies": 0.518750011920929, + "rewards/chosen": 0.000418459705542773, + "rewards/margins": 0.0002198911242885515, + "rewards/rejected": 0.00019856853759847581, + "step": 15 + }, + { + "epoch": 0.01, + "learning_rate": 2e-08, + "logits/chosen": -1.4347673654556274, + "logits/rejected": -1.457056999206543, + "logps/chosen": -16.931005477905273, + "logps/rejected": -17.90321159362793, + "loss": 0.6938, + "rewards/accuracies": 0.45625001192092896, + "rewards/chosen": -0.0022916034795343876, + "rewards/margins": -0.003587479470297694, + "rewards/rejected": 0.0012958759907633066, + "step": 20 + }, + { + "epoch": 0.01, + "learning_rate": 2.5e-08, + "logits/chosen": -1.5136417150497437, + "logits/rejected": -1.5317498445510864, + "logps/chosen": -13.484731674194336, + "logps/rejected": -17.72646141052246, + "loss": 0.693, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.0013993926113471389, + "rewards/margins": 0.003486391855403781, + "rewards/rejected": -0.002086999360471964, + "step": 25 + }, + { + "epoch": 0.01, + "learning_rate": 3e-08, + "logits/chosen": -1.4325120449066162, + "logits/rejected": -1.423964262008667, + "logps/chosen": -16.99893569946289, + "logps/rejected": -21.196269989013672, + "loss": 0.6929, + "rewards/accuracies": 0.5, + "rewards/chosen": -0.0006942325853742659, + "rewards/margins": -0.00022374764375854284, + "rewards/rejected": -0.0004704846942331642, + "step": 30 + }, + { + "epoch": 0.02, + "learning_rate": 3.4999999999999996e-08, + "logits/chosen": -1.503702163696289, + "logits/rejected": -1.5359680652618408, + "logps/chosen": -15.859466552734375, + "logps/rejected": -18.37940788269043, + "loss": 0.6925, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.002690280321985483, + "rewards/margins": 0.0054735760204494, + "rewards/rejected": -0.002783295465633273, + "step": 35 + }, + { + "epoch": 0.02, + "learning_rate": 4e-08, + "logits/chosen": -1.474712610244751, + "logits/rejected": -1.4741761684417725, + "logps/chosen": -13.360185623168945, + "logps/rejected": -16.91265296936035, + "loss": 0.6935, + "rewards/accuracies": 0.5, + "rewards/chosen": -0.0006134170107543468, + "rewards/margins": 0.0004067299305461347, + "rewards/rejected": -0.0010201467666774988, + "step": 40 + }, + { + "epoch": 0.02, + "learning_rate": 4.5e-08, + "logits/chosen": -1.4315367937088013, + "logits/rejected": -1.4630348682403564, + "logps/chosen": -14.167867660522461, + "logps/rejected": -18.600988388061523, + "loss": 0.6931, + "rewards/accuracies": 0.518750011920929, + "rewards/chosen": 0.0011325415689498186, + "rewards/margins": -7.549161182396347e-06, + "rewards/rejected": 0.0011400904040783644, + "step": 45 + }, + { + "epoch": 0.03, + "learning_rate": 5e-08, + "logits/chosen": -1.466339349746704, + "logits/rejected": -1.4680755138397217, + "logps/chosen": -13.571925163269043, + "logps/rejected": -16.360881805419922, + "loss": 0.6933, + "rewards/accuracies": 0.4749999940395355, + "rewards/chosen": -0.0013324546162039042, + "rewards/margins": -5.83843320782762e-05, + "rewards/rejected": -0.0012740703532472253, + "step": 50 + }, + { + "epoch": 0.03, + "learning_rate": 5.5e-08, + "logits/chosen": -1.4321575164794922, + "logits/rejected": -1.4344770908355713, + "logps/chosen": -14.864067077636719, + "logps/rejected": -20.08001708984375, + "loss": 0.6929, + "rewards/accuracies": 0.5, + "rewards/chosen": 0.000479007518151775, + "rewards/margins": 0.001067839446477592, + "rewards/rejected": -0.0005888319574296474, + "step": 55 + }, + { + "epoch": 0.03, + "learning_rate": 6e-08, + "logits/chosen": -1.5018717050552368, + "logits/rejected": -1.5294861793518066, + "logps/chosen": -13.418993949890137, + "logps/rejected": -17.377300262451172, + "loss": 0.6929, + "rewards/accuracies": 0.46875, + "rewards/chosen": -0.0013912434224039316, + "rewards/margins": -0.0009791270131245255, + "rewards/rejected": -0.0004121163801755756, + "step": 60 + }, + { + "epoch": 0.03, + "learning_rate": 6.5e-08, + "logits/chosen": -1.40516197681427, + "logits/rejected": -1.4193370342254639, + "logps/chosen": -13.224818229675293, + "logps/rejected": -15.901044845581055, + "loss": 0.6926, + "rewards/accuracies": 0.48750001192092896, + "rewards/chosen": -0.0004983447724953294, + "rewards/margins": 0.0009676685440354049, + "rewards/rejected": -0.0014660133747383952, + "step": 65 + }, + { + "epoch": 0.04, + "learning_rate": 6.999999999999999e-08, + "logits/chosen": -1.5142648220062256, + "logits/rejected": -1.5050678253173828, + "logps/chosen": -15.645828247070312, + "logps/rejected": -20.22635269165039, + "loss": 0.6928, + "rewards/accuracies": 0.543749988079071, + "rewards/chosen": -0.0008540893904864788, + "rewards/margins": 0.0006531132385134697, + "rewards/rejected": -0.0015072030946612358, + "step": 70 + }, + { + "epoch": 0.04, + "learning_rate": 7.5e-08, + "logits/chosen": -1.4419431686401367, + "logits/rejected": -1.4324930906295776, + "logps/chosen": -14.863853454589844, + "logps/rejected": -16.58720588684082, + "loss": 0.6924, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": 0.0013656382216140628, + "rewards/margins": 0.0006723797414451838, + "rewards/rejected": 0.0006932583637535572, + "step": 75 + }, + { + "epoch": 0.04, + "learning_rate": 8e-08, + "logits/chosen": -1.532387375831604, + "logits/rejected": -1.5147764682769775, + "logps/chosen": -15.975011825561523, + "logps/rejected": -19.93300437927246, + "loss": 0.6925, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.0007912625442259014, + "rewards/margins": 0.00153960264287889, + "rewards/rejected": -0.0023308652453124523, + "step": 80 + }, + { + "epoch": 0.04, + "learning_rate": 8.5e-08, + "logits/chosen": -1.4505486488342285, + "logits/rejected": -1.4808809757232666, + "logps/chosen": -17.687313079833984, + "logps/rejected": -24.51900863647461, + "loss": 0.6921, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.00017726181249599904, + "rewards/margins": 0.0016221065307036042, + "rewards/rejected": -0.001799368066713214, + "step": 85 + }, + { + "epoch": 0.04, + "learning_rate": 9e-08, + "logits/chosen": -1.4988192319869995, + "logits/rejected": -1.513806939125061, + "logps/chosen": -14.488324165344238, + "logps/rejected": -19.464611053466797, + "loss": 0.6912, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.0011351148132234812, + "rewards/margins": 0.004005365073680878, + "rewards/rejected": -0.00287025049328804, + "step": 90 + }, + { + "epoch": 0.05, + "learning_rate": 9.499999999999999e-08, + "logits/chosen": -1.403247594833374, + "logits/rejected": -1.4147038459777832, + "logps/chosen": -16.571857452392578, + "logps/rejected": -16.457307815551758, + "loss": 0.6907, + "rewards/accuracies": 0.59375, + "rewards/chosen": 0.00036610427196137607, + "rewards/margins": 0.004560294561088085, + "rewards/rejected": -0.004194191191345453, + "step": 95 + }, + { + "epoch": 0.05, + "learning_rate": 1e-07, + "logits/chosen": -1.486196756362915, + "logits/rejected": -1.504416823387146, + "logps/chosen": -14.919549942016602, + "logps/rejected": -19.339908599853516, + "loss": 0.691, + "rewards/accuracies": 0.518750011920929, + "rewards/chosen": -0.003806027816608548, + "rewards/margins": 0.004251133184880018, + "rewards/rejected": -0.008057162165641785, + "step": 100 + }, + { + "epoch": 0.05, + "eval_logits/chosen": -1.7919771671295166, + "eval_logits/rejected": -1.8722617626190186, + "eval_logps/chosen": -15.013982772827148, + "eval_logps/rejected": -18.596895217895508, + "eval_loss": 0.6906684637069702, + "eval_rewards/accuracies": 0.5674920082092285, + "eval_rewards/chosen": -8.375057223020121e-05, + "eval_rewards/margins": 0.005779640283435583, + "eval_rewards/rejected": -0.005863390862941742, + "eval_runtime": 306.7, + "eval_samples_per_second": 65.21, + "eval_steps_per_second": 1.021, + "step": 100 + }, + { + "epoch": 0.05, + "learning_rate": 9.999238475781956e-08, + "logits/chosen": -1.415111780166626, + "logits/rejected": -1.409263253211975, + "logps/chosen": -15.464300155639648, + "logps/rejected": -18.61844253540039, + "loss": 0.6899, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.0006141990306787193, + "rewards/margins": 0.004516326356679201, + "rewards/rejected": -0.00513052474707365, + "step": 105 + }, + { + "epoch": 0.06, + "learning_rate": 9.996954135095479e-08, + "logits/chosen": -1.4188635349273682, + "logits/rejected": -1.432936668395996, + "logps/chosen": -15.272178649902344, + "logps/rejected": -19.53142738342285, + "loss": 0.689, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.00477779284119606, + "rewards/margins": 0.0066839540377259254, + "rewards/rejected": -0.011461746878921986, + "step": 110 + }, + { + "epoch": 0.06, + "learning_rate": 9.993147673772868e-08, + "logits/chosen": -1.4598616361618042, + "logits/rejected": -1.4905925989151, + "logps/chosen": -15.878756523132324, + "logps/rejected": -18.827163696289062, + "loss": 0.688, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": 0.002627016045153141, + "rewards/margins": 0.01106287818402052, + "rewards/rejected": -0.008435862138867378, + "step": 115 + }, + { + "epoch": 0.06, + "learning_rate": 9.98782025129912e-08, + "logits/chosen": -1.4647516012191772, + "logits/rejected": -1.4928150177001953, + "logps/chosen": -13.536163330078125, + "logps/rejected": -18.504684448242188, + "loss": 0.6892, + "rewards/accuracies": 0.518750011920929, + "rewards/chosen": -0.006495442241430283, + "rewards/margins": 0.0007650878978893161, + "rewards/rejected": -0.007260529790073633, + "step": 120 + }, + { + "epoch": 0.06, + "learning_rate": 9.980973490458727e-08, + "logits/chosen": -1.4349157810211182, + "logits/rejected": -1.4921131134033203, + "logps/chosen": -16.818748474121094, + "logps/rejected": -23.732484817504883, + "loss": 0.6884, + "rewards/accuracies": 0.48124998807907104, + "rewards/chosen": -0.014983393251895905, + "rewards/margins": 0.003936333581805229, + "rewards/rejected": -0.018919726833701134, + "step": 125 + }, + { + "epoch": 0.07, + "learning_rate": 9.972609476841366e-08, + "logits/chosen": -1.4056415557861328, + "logits/rejected": -1.4126735925674438, + "logps/chosen": -14.3095121383667, + "logps/rejected": -18.44950294494629, + "loss": 0.687, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.005277615040540695, + "rewards/margins": 0.011254631914198399, + "rewards/rejected": -0.01653224602341652, + "step": 130 + }, + { + "epoch": 0.07, + "learning_rate": 9.96273075820661e-08, + "logits/chosen": -1.5334028005599976, + "logits/rejected": -1.5822241306304932, + "logps/chosen": -15.960824966430664, + "logps/rejected": -21.098979949951172, + "loss": 0.6875, + "rewards/accuracies": 0.518750011920929, + "rewards/chosen": -0.016458828002214432, + "rewards/margins": 0.006202386226505041, + "rewards/rejected": -0.02266121283173561, + "step": 135 + }, + { + "epoch": 0.07, + "learning_rate": 9.951340343707851e-08, + "logits/chosen": -1.5316803455352783, + "logits/rejected": -1.5605201721191406, + "logps/chosen": -13.550819396972656, + "logps/rejected": -17.886049270629883, + "loss": 0.6851, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.012303833849728107, + "rewards/margins": 0.02252637967467308, + "rewards/rejected": -0.03483021631836891, + "step": 140 + }, + { + "epoch": 0.07, + "learning_rate": 9.938441702975688e-08, + "logits/chosen": -1.4327102899551392, + "logits/rejected": -1.446903944015503, + "logps/chosen": -13.809109687805176, + "logps/rejected": -17.256572723388672, + "loss": 0.685, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.016683299094438553, + "rewards/margins": 0.029044728726148605, + "rewards/rejected": -0.04572802782058716, + "step": 145 + }, + { + "epoch": 0.07, + "learning_rate": 9.92403876506104e-08, + "logits/chosen": -1.4938914775848389, + "logits/rejected": -1.5160696506500244, + "logps/chosen": -13.700773239135742, + "logps/rejected": -19.660675048828125, + "loss": 0.6815, + "rewards/accuracies": 0.4937500059604645, + "rewards/chosen": -0.03682475537061691, + "rewards/margins": 0.010936172679066658, + "rewards/rejected": -0.04776093363761902, + "step": 150 + }, + { + "epoch": 0.08, + "learning_rate": 9.90813591723832e-08, + "logits/chosen": -1.4303719997406006, + "logits/rejected": -1.4469249248504639, + "logps/chosen": -14.483743667602539, + "logps/rejected": -20.158580780029297, + "loss": 0.6835, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.03273214027285576, + "rewards/margins": 0.030618244782090187, + "rewards/rejected": -0.06335039436817169, + "step": 155 + }, + { + "epoch": 0.08, + "learning_rate": 9.890738003669028e-08, + "logits/chosen": -1.4257906675338745, + "logits/rejected": -1.440292477607727, + "logps/chosen": -13.087536811828613, + "logps/rejected": -18.774667739868164, + "loss": 0.6813, + "rewards/accuracies": 0.543749988079071, + "rewards/chosen": -0.042621154338121414, + "rewards/margins": 0.019855033606290817, + "rewards/rejected": -0.06247618794441223, + "step": 160 + }, + { + "epoch": 0.08, + "learning_rate": 9.871850323926176e-08, + "logits/chosen": -1.506829023361206, + "logits/rejected": -1.4980268478393555, + "logps/chosen": -15.794275283813477, + "logps/rejected": -22.559085845947266, + "loss": 0.6832, + "rewards/accuracies": 0.543749988079071, + "rewards/chosen": -0.04176632687449455, + "rewards/margins": 0.014226436614990234, + "rewards/rejected": -0.05599276348948479, + "step": 165 + }, + { + "epoch": 0.09, + "learning_rate": 9.851478631379982e-08, + "logits/chosen": -1.5072795152664185, + "logits/rejected": -1.5367839336395264, + "logps/chosen": -16.79397964477539, + "logps/rejected": -18.064043045043945, + "loss": 0.6843, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.037377405911684036, + "rewards/margins": 0.04563738405704498, + "rewards/rejected": -0.08301478624343872, + "step": 170 + }, + { + "epoch": 0.09, + "learning_rate": 9.82962913144534e-08, + "logits/chosen": -1.507598876953125, + "logits/rejected": -1.5444573163986206, + "logps/chosen": -17.070449829101562, + "logps/rejected": -20.720073699951172, + "loss": 0.6831, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.06209941580891609, + "rewards/margins": 0.03557855635881424, + "rewards/rejected": -0.09767796844244003, + "step": 175 + }, + { + "epoch": 0.09, + "learning_rate": 9.806308479691594e-08, + "logits/chosen": -1.4471648931503296, + "logits/rejected": -1.474890947341919, + "logps/chosen": -16.156829833984375, + "logps/rejected": -22.047147750854492, + "loss": 0.68, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.0555298812687397, + "rewards/margins": 0.0073929824866354465, + "rewards/rejected": -0.06292285770177841, + "step": 180 + }, + { + "epoch": 0.09, + "learning_rate": 9.781523779815177e-08, + "logits/chosen": -1.5135043859481812, + "logits/rejected": -1.5078579187393188, + "logps/chosen": -14.832931518554688, + "logps/rejected": -17.844093322753906, + "loss": 0.6827, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.06777982413768768, + "rewards/margins": 0.03908789902925491, + "rewards/rejected": -0.106867715716362, + "step": 185 + }, + { + "epoch": 0.1, + "learning_rate": 9.755282581475768e-08, + "logits/chosen": -1.4608676433563232, + "logits/rejected": -1.4990969896316528, + "logps/chosen": -14.89118766784668, + "logps/rejected": -20.47022247314453, + "loss": 0.6749, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.05767616629600525, + "rewards/margins": 0.0543893501162529, + "rewards/rejected": -0.11206551641225815, + "step": 190 + }, + { + "epoch": 0.1, + "learning_rate": 9.727592877996584e-08, + "logits/chosen": -1.4986900091171265, + "logits/rejected": -1.5200297832489014, + "logps/chosen": -14.921911239624023, + "logps/rejected": -19.319089889526367, + "loss": 0.6782, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.07736112922430038, + "rewards/margins": 0.037412129342556, + "rewards/rejected": -0.11477325856685638, + "step": 195 + }, + { + "epoch": 0.1, + "learning_rate": 9.698463103929542e-08, + "logits/chosen": -1.4672738313674927, + "logits/rejected": -1.4699958562850952, + "logps/chosen": -16.190366744995117, + "logps/rejected": -19.112018585205078, + "loss": 0.6787, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.06623920053243637, + "rewards/margins": 0.05223413184285164, + "rewards/rejected": -0.11847333610057831, + "step": 200 + }, + { + "epoch": 0.1, + "eval_logits/chosen": -1.7878340482711792, + "eval_logits/rejected": -1.868047833442688, + "eval_logps/chosen": -15.84277057647705, + "eval_logps/rejected": -19.780603408813477, + "eval_loss": 0.679453432559967, + "eval_rewards/accuracies": 0.5942491888999939, + "eval_rewards/chosen": -0.08296255767345428, + "eval_rewards/margins": 0.04127146303653717, + "eval_rewards/rejected": -0.12423399835824966, + "eval_runtime": 306.7549, + "eval_samples_per_second": 65.199, + "eval_steps_per_second": 1.02, + "step": 200 + }, + { + "epoch": 0.1, + "learning_rate": 9.667902132486008e-08, + "logits/chosen": -1.4577996730804443, + "logits/rejected": -1.458738088607788, + "logps/chosen": -14.40186882019043, + "logps/rejected": -19.614154815673828, + "loss": 0.6824, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": -0.1032874584197998, + "rewards/margins": 0.023597324267029762, + "rewards/rejected": -0.12688478827476501, + "step": 205 + }, + { + "epoch": 0.1, + "learning_rate": 9.635919272833936e-08, + "logits/chosen": -1.446189284324646, + "logits/rejected": -1.4532312154769897, + "logps/chosen": -13.800382614135742, + "logps/rejected": -16.633275985717773, + "loss": 0.6743, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.08655096590518951, + "rewards/margins": 0.04802204668521881, + "rewards/rejected": -0.13457301259040833, + "step": 210 + }, + { + "epoch": 0.11, + "learning_rate": 9.602524267262201e-08, + "logits/chosen": -1.5442548990249634, + "logits/rejected": -1.5252314805984497, + "logps/chosen": -16.61358070373535, + "logps/rejected": -19.388107299804688, + "loss": 0.6745, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.08506567776203156, + "rewards/margins": 0.061091698706150055, + "rewards/rejected": -0.14615735411643982, + "step": 215 + }, + { + "epoch": 0.11, + "learning_rate": 9.567727288213003e-08, + "logits/chosen": -1.5201714038848877, + "logits/rejected": -1.5108391046524048, + "logps/chosen": -16.2724609375, + "logps/rejected": -19.838550567626953, + "loss": 0.6755, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.1071196049451828, + "rewards/margins": 0.05034583806991577, + "rewards/rejected": -0.15746544301509857, + "step": 220 + }, + { + "epoch": 0.11, + "learning_rate": 9.53153893518325e-08, + "logits/chosen": -1.4277980327606201, + "logits/rejected": -1.4468252658843994, + "logps/chosen": -15.238229751586914, + "logps/rejected": -21.562265396118164, + "loss": 0.6815, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.11674360930919647, + "rewards/margins": 0.013696810230612755, + "rewards/rejected": -0.13044041395187378, + "step": 225 + }, + { + "epoch": 0.12, + "learning_rate": 9.493970231495834e-08, + "logits/chosen": -1.449592113494873, + "logits/rejected": -1.4903148412704468, + "logps/chosen": -16.092803955078125, + "logps/rejected": -21.503910064697266, + "loss": 0.6787, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.08694227039813995, + "rewards/margins": 0.06734081357717514, + "rewards/rejected": -0.1542830765247345, + "step": 230 + }, + { + "epoch": 0.12, + "learning_rate": 9.455032620941839e-08, + "logits/chosen": -1.4714128971099854, + "logits/rejected": -1.5002821683883667, + "logps/chosen": -18.0159969329834, + "logps/rejected": -20.794544219970703, + "loss": 0.6774, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.09392134845256805, + "rewards/margins": 0.06613980978727341, + "rewards/rejected": -0.16006115078926086, + "step": 235 + }, + { + "epoch": 0.12, + "learning_rate": 9.414737964294634e-08, + "logits/chosen": -1.4862656593322754, + "logits/rejected": -1.5194942951202393, + "logps/chosen": -18.190040588378906, + "logps/rejected": -20.4471492767334, + "loss": 0.6773, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.12713944911956787, + "rewards/margins": 0.034943290054798126, + "rewards/rejected": -0.1620827466249466, + "step": 240 + }, + { + "epoch": 0.12, + "learning_rate": 9.373098535696979e-08, + "logits/chosen": -1.4240232706069946, + "logits/rejected": -1.439587116241455, + "logps/chosen": -16.851634979248047, + "logps/rejected": -21.63901138305664, + "loss": 0.6721, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.11880864202976227, + "rewards/margins": 0.0625835433602333, + "rewards/rejected": -0.18139217793941498, + "step": 245 + }, + { + "epoch": 0.12, + "learning_rate": 9.330127018922194e-08, + "logits/chosen": -1.490593433380127, + "logits/rejected": -1.519042730331421, + "logps/chosen": -15.26930046081543, + "logps/rejected": -20.554950714111328, + "loss": 0.6788, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.12308211624622345, + "rewards/margins": 0.049174632877111435, + "rewards/rejected": -0.17225676774978638, + "step": 250 + }, + { + "epoch": 0.13, + "learning_rate": 9.285836503510561e-08, + "logits/chosen": -1.481011152267456, + "logits/rejected": -1.482369065284729, + "logps/chosen": -14.86046028137207, + "logps/rejected": -20.206274032592773, + "loss": 0.6683, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.09218485653400421, + "rewards/margins": 0.0633331835269928, + "rewards/rejected": -0.155518040060997, + "step": 255 + }, + { + "epoch": 0.13, + "learning_rate": 9.240240480782128e-08, + "logits/chosen": -1.4365761280059814, + "logits/rejected": -1.4659501314163208, + "logps/chosen": -16.5053768157959, + "logps/rejected": -20.457700729370117, + "loss": 0.6781, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.1196146234869957, + "rewards/margins": 0.0831097811460495, + "rewards/rejected": -0.2027243822813034, + "step": 260 + }, + { + "epoch": 0.13, + "learning_rate": 9.19335283972712e-08, + "logits/chosen": -1.4037281274795532, + "logits/rejected": -1.4219300746917725, + "logps/chosen": -17.773784637451172, + "logps/rejected": -21.13812828063965, + "loss": 0.6672, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.11854933202266693, + "rewards/margins": 0.08150772750377655, + "rewards/rejected": -0.20005705952644348, + "step": 265 + }, + { + "epoch": 0.14, + "learning_rate": 9.145187862775208e-08, + "logits/chosen": -1.5054607391357422, + "logits/rejected": -1.5367683172225952, + "logps/chosen": -16.548221588134766, + "logps/rejected": -20.993818283081055, + "loss": 0.6649, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.1312890648841858, + "rewards/margins": 0.06594224274158478, + "rewards/rejected": -0.19723130762577057, + "step": 270 + }, + { + "epoch": 0.14, + "learning_rate": 9.095760221444959e-08, + "logits/chosen": -1.5523430109024048, + "logits/rejected": -1.5738317966461182, + "logps/chosen": -16.80923080444336, + "logps/rejected": -20.842090606689453, + "loss": 0.6712, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.1634162962436676, + "rewards/margins": 0.06340184062719345, + "rewards/rejected": -0.22681812942028046, + "step": 275 + }, + { + "epoch": 0.14, + "learning_rate": 9.045084971874737e-08, + "logits/chosen": -1.4899578094482422, + "logits/rejected": -1.5115963220596313, + "logps/chosen": -15.990486145019531, + "logps/rejected": -21.041019439697266, + "loss": 0.6678, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.1481240689754486, + "rewards/margins": 0.04743753746151924, + "rewards/rejected": -0.19556160271167755, + "step": 280 + }, + { + "epoch": 0.14, + "learning_rate": 8.993177550236463e-08, + "logits/chosen": -1.484438180923462, + "logits/rejected": -1.499731183052063, + "logps/chosen": -16.02753448486328, + "logps/rejected": -20.631275177001953, + "loss": 0.6734, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.15427681803703308, + "rewards/margins": 0.016259009018540382, + "rewards/rejected": -0.1705358326435089, + "step": 285 + }, + { + "epoch": 0.14, + "learning_rate": 8.940053768033609e-08, + "logits/chosen": -1.4610778093338013, + "logits/rejected": -1.4643038511276245, + "logps/chosen": -18.099334716796875, + "logps/rejected": -21.200000762939453, + "loss": 0.6758, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.15338566899299622, + "rewards/margins": 0.03785661607980728, + "rewards/rejected": -0.1912422627210617, + "step": 290 + }, + { + "epoch": 0.15, + "learning_rate": 8.885729807284853e-08, + "logits/chosen": -1.5055420398712158, + "logits/rejected": -1.5390150547027588, + "logps/chosen": -14.061132431030273, + "logps/rejected": -19.544492721557617, + "loss": 0.6667, + "rewards/accuracies": 0.5687500238418579, + "rewards/chosen": -0.189667746424675, + "rewards/margins": 0.06658226996660233, + "rewards/rejected": -0.2562499940395355, + "step": 295 + }, + { + "epoch": 0.15, + "learning_rate": 8.83022221559489e-08, + "logits/chosen": -1.3812211751937866, + "logits/rejected": -1.394118070602417, + "logps/chosen": -17.90454864501953, + "logps/rejected": -21.622303009033203, + "loss": 0.6701, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.177689790725708, + "rewards/margins": 0.05468187481164932, + "rewards/rejected": -0.23237165808677673, + "step": 300 + }, + { + "epoch": 0.15, + "eval_logits/chosen": -1.7818100452423096, + "eval_logits/rejected": -1.8616912364959717, + "eval_logps/chosen": -16.538070678710938, + "eval_logps/rejected": -20.776979446411133, + "eval_loss": 0.6713529229164124, + "eval_rewards/accuracies": 0.6006389856338501, + "eval_rewards/chosen": -0.15249261260032654, + "eval_rewards/margins": 0.07137925922870636, + "eval_rewards/rejected": -0.2238718867301941, + "eval_runtime": 306.8553, + "eval_samples_per_second": 65.177, + "eval_steps_per_second": 1.02, + "step": 300 + }, + { + "epoch": 0.15, + "learning_rate": 8.77354790111386e-08, + "logits/chosen": -1.4351907968521118, + "logits/rejected": -1.4078080654144287, + "logps/chosen": -15.99042797088623, + "logps/rejected": -19.805822372436523, + "loss": 0.6643, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.14661024510860443, + "rewards/margins": 0.06514577567577362, + "rewards/rejected": -0.21175602078437805, + "step": 305 + }, + { + "epoch": 0.15, + "learning_rate": 8.715724127386971e-08, + "logits/chosen": -1.4537417888641357, + "logits/rejected": -1.4652783870697021, + "logps/chosen": -18.730056762695312, + "logps/rejected": -22.277685165405273, + "loss": 0.6727, + "rewards/accuracies": 0.543749988079071, + "rewards/chosen": -0.18776853382587433, + "rewards/margins": 0.04285965487360954, + "rewards/rejected": -0.23062816262245178, + "step": 310 + }, + { + "epoch": 0.16, + "learning_rate": 8.656768508095852e-08, + "logits/chosen": -1.400272250175476, + "logits/rejected": -1.3987435102462769, + "logps/chosen": -16.988048553466797, + "logps/rejected": -19.94150161743164, + "loss": 0.6662, + "rewards/accuracies": 0.53125, + "rewards/chosen": -0.1827572137117386, + "rewards/margins": 0.03802407532930374, + "rewards/rejected": -0.22078128159046173, + "step": 315 + }, + { + "epoch": 0.16, + "learning_rate": 8.596699001693255e-08, + "logits/chosen": -1.4178965091705322, + "logits/rejected": -1.4355350732803345, + "logps/chosen": -15.241655349731445, + "logps/rejected": -23.634593963623047, + "loss": 0.6629, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.16266849637031555, + "rewards/margins": 0.08267641067504883, + "rewards/rejected": -0.24534490704536438, + "step": 320 + }, + { + "epoch": 0.16, + "learning_rate": 8.535533905932736e-08, + "logits/chosen": -1.432398796081543, + "logits/rejected": -1.4598662853240967, + "logps/chosen": -16.167316436767578, + "logps/rejected": -20.810152053833008, + "loss": 0.6686, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.18219885230064392, + "rewards/margins": 0.06463146209716797, + "rewards/rejected": -0.2468303143978119, + "step": 325 + }, + { + "epoch": 0.17, + "learning_rate": 8.473291852294986e-08, + "logits/chosen": -1.4149553775787354, + "logits/rejected": -1.4176180362701416, + "logps/chosen": -15.214263916015625, + "logps/rejected": -20.01323890686035, + "loss": 0.6633, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.1562959998846054, + "rewards/margins": 0.10322503745555878, + "rewards/rejected": -0.2595210373401642, + "step": 330 + }, + { + "epoch": 0.17, + "learning_rate": 8.409991800312493e-08, + "logits/chosen": -1.5221331119537354, + "logits/rejected": -1.5471950769424438, + "logps/chosen": -15.114236831665039, + "logps/rejected": -18.62519645690918, + "loss": 0.6609, + "rewards/accuracies": 0.706250011920929, + "rewards/chosen": -0.12045025825500488, + "rewards/margins": 0.14489233493804932, + "rewards/rejected": -0.2653425931930542, + "step": 335 + }, + { + "epoch": 0.17, + "learning_rate": 8.34565303179429e-08, + "logits/chosen": -1.4450016021728516, + "logits/rejected": -1.4777119159698486, + "logps/chosen": -15.960481643676758, + "logps/rejected": -21.401214599609375, + "loss": 0.6669, + "rewards/accuracies": 0.53125, + "rewards/chosen": -0.20441024005413055, + "rewards/margins": 0.05542609095573425, + "rewards/rejected": -0.259836345911026, + "step": 340 + }, + { + "epoch": 0.17, + "learning_rate": 8.280295144952536e-08, + "logits/chosen": -1.3684567213058472, + "logits/rejected": -1.356687307357788, + "logps/chosen": -15.962387084960938, + "logps/rejected": -20.549938201904297, + "loss": 0.6613, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.15356414020061493, + "rewards/margins": 0.054951734840869904, + "rewards/rejected": -0.20851588249206543, + "step": 345 + }, + { + "epoch": 0.17, + "learning_rate": 8.213938048432696e-08, + "logits/chosen": -1.4533028602600098, + "logits/rejected": -1.47086763381958, + "logps/chosen": -15.168508529663086, + "logps/rejected": -20.278743743896484, + "loss": 0.6692, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.1997944861650467, + "rewards/margins": 0.06951111555099487, + "rewards/rejected": -0.26930561661720276, + "step": 350 + }, + { + "epoch": 0.18, + "learning_rate": 8.146601955249188e-08, + "logits/chosen": -1.4886163473129272, + "logits/rejected": -1.5108848810195923, + "logps/chosen": -18.865304946899414, + "logps/rejected": -24.56046485900879, + "loss": 0.6586, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.1855696439743042, + "rewards/margins": 0.1037302240729332, + "rewards/rejected": -0.289299875497818, + "step": 355 + }, + { + "epoch": 0.18, + "learning_rate": 8.07830737662829e-08, + "logits/chosen": -1.4619147777557373, + "logits/rejected": -1.4794903993606567, + "logps/chosen": -13.954874992370605, + "logps/rejected": -17.310880661010742, + "loss": 0.6687, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.18103952705860138, + "rewards/margins": 0.09420069307088852, + "rewards/rejected": -0.2752402424812317, + "step": 360 + }, + { + "epoch": 0.18, + "learning_rate": 8.009075115760243e-08, + "logits/chosen": -1.3863359689712524, + "logits/rejected": -1.3789931535720825, + "logps/chosen": -17.23479652404785, + "logps/rejected": -19.347734451293945, + "loss": 0.664, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.17440780997276306, + "rewards/margins": 0.10439540445804596, + "rewards/rejected": -0.2788031995296478, + "step": 365 + }, + { + "epoch": 0.18, + "learning_rate": 7.938926261462366e-08, + "logits/chosen": -1.355503797531128, + "logits/rejected": -1.3629181385040283, + "logps/chosen": -18.74729347229004, + "logps/rejected": -22.029096603393555, + "loss": 0.6606, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.21672053635120392, + "rewards/margins": 0.04532719776034355, + "rewards/rejected": -0.2620477080345154, + "step": 370 + }, + { + "epoch": 0.19, + "learning_rate": 7.86788218175523e-08, + "logits/chosen": -1.429302453994751, + "logits/rejected": -1.457215666770935, + "logps/chosen": -16.596994400024414, + "logps/rejected": -21.360082626342773, + "loss": 0.6667, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.18682262301445007, + "rewards/margins": 0.1307360827922821, + "rewards/rejected": -0.3175587058067322, + "step": 375 + }, + { + "epoch": 0.19, + "learning_rate": 7.795964517353733e-08, + "logits/chosen": -1.4372557401657104, + "logits/rejected": -1.4457218647003174, + "logps/chosen": -17.684959411621094, + "logps/rejected": -21.197628021240234, + "loss": 0.6617, + "rewards/accuracies": 0.5687500238418579, + "rewards/chosen": -0.1949060708284378, + "rewards/margins": 0.0896020457148552, + "rewards/rejected": -0.2845081090927124, + "step": 380 + }, + { + "epoch": 0.19, + "learning_rate": 7.723195175075135e-08, + "logits/chosen": -1.4864705801010132, + "logits/rejected": -1.4808794260025024, + "logps/chosen": -16.80634880065918, + "logps/rejected": -20.31828498840332, + "loss": 0.6581, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.1962084323167801, + "rewards/margins": 0.1290890872478485, + "rewards/rejected": -0.3252975046634674, + "step": 385 + }, + { + "epoch": 0.2, + "learning_rate": 7.649596321166024e-08, + "logits/chosen": -1.3273518085479736, + "logits/rejected": -1.3579399585723877, + "logps/chosen": -16.424484252929688, + "logps/rejected": -19.864755630493164, + "loss": 0.6686, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.17965315282344818, + "rewards/margins": 0.09440762549638748, + "rewards/rejected": -0.27406078577041626, + "step": 390 + }, + { + "epoch": 0.2, + "learning_rate": 7.575190374550272e-08, + "logits/chosen": -1.4504528045654297, + "logits/rejected": -1.4469316005706787, + "logps/chosen": -17.03780174255371, + "logps/rejected": -20.324657440185547, + "loss": 0.6587, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.22678837180137634, + "rewards/margins": 0.05626438185572624, + "rewards/rejected": -0.2830527722835541, + "step": 395 + }, + { + "epoch": 0.2, + "learning_rate": 7.5e-08, + "logits/chosen": -1.4595016241073608, + "logits/rejected": -1.4589979648590088, + "logps/chosen": -16.276988983154297, + "logps/rejected": -22.162479400634766, + "loss": 0.6676, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.2052340805530548, + "rewards/margins": 0.09203705936670303, + "rewards/rejected": -0.29727110266685486, + "step": 400 + }, + { + "epoch": 0.2, + "eval_logits/chosen": -1.7746305465698242, + "eval_logits/rejected": -1.8544888496398926, + "eval_logps/chosen": -16.942358016967773, + "eval_logps/rejected": -21.468765258789062, + "eval_loss": 0.6636533737182617, + "eval_rewards/accuracies": 0.6174121499061584, + "eval_rewards/chosen": -0.19292119145393372, + "eval_rewards/margins": 0.10012920200824738, + "eval_rewards/rejected": -0.2930504083633423, + "eval_runtime": 306.9122, + "eval_samples_per_second": 65.165, + "eval_steps_per_second": 1.02, + "step": 400 + }, + { + "epoch": 0.2, + "learning_rate": 7.424048101231686e-08, + "logits/chosen": -1.3691225051879883, + "logits/rejected": -1.3902453184127808, + "logps/chosen": -18.4908504486084, + "logps/rejected": -25.587356567382812, + "loss": 0.6637, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.2073945701122284, + "rewards/margins": 0.09110216796398163, + "rewards/rejected": -0.2984967529773712, + "step": 405 + }, + { + "epoch": 0.2, + "learning_rate": 7.347357813929453e-08, + "logits/chosen": -1.4559385776519775, + "logits/rejected": -1.4722373485565186, + "logps/chosen": -14.663101196289062, + "logps/rejected": -18.937580108642578, + "loss": 0.658, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.19805128872394562, + "rewards/margins": 0.08183437585830688, + "rewards/rejected": -0.2798856794834137, + "step": 410 + }, + { + "epoch": 0.21, + "learning_rate": 7.269952498697734e-08, + "logits/chosen": -1.4937471151351929, + "logits/rejected": -1.508141279220581, + "logps/chosen": -16.866802215576172, + "logps/rejected": -19.958166122436523, + "loss": 0.6675, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.16971763968467712, + "rewards/margins": 0.0776999443769455, + "rewards/rejected": -0.2474175989627838, + "step": 415 + }, + { + "epoch": 0.21, + "learning_rate": 7.191855733945387e-08, + "logits/chosen": -1.4956650733947754, + "logits/rejected": -1.4953656196594238, + "logps/chosen": -15.598005294799805, + "logps/rejected": -20.977344512939453, + "loss": 0.6545, + "rewards/accuracies": 0.6812499761581421, + "rewards/chosen": -0.20860418677330017, + "rewards/margins": 0.113286092877388, + "rewards/rejected": -0.321890264749527, + "step": 420 + }, + { + "epoch": 0.21, + "learning_rate": 7.113091308703497e-08, + "logits/chosen": -1.4557757377624512, + "logits/rejected": -1.4881502389907837, + "logps/chosen": -15.815752029418945, + "logps/rejected": -23.03244400024414, + "loss": 0.6648, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.18280960619449615, + "rewards/margins": 0.10344807803630829, + "rewards/rejected": -0.28625768423080444, + "step": 425 + }, + { + "epoch": 0.21, + "learning_rate": 7.033683215379002e-08, + "logits/chosen": -1.5411797761917114, + "logits/rejected": -1.52112877368927, + "logps/chosen": -19.202138900756836, + "logps/rejected": -21.153202056884766, + "loss": 0.6586, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.21284401416778564, + "rewards/margins": 0.06404918432235718, + "rewards/rejected": -0.2768932282924652, + "step": 430 + }, + { + "epoch": 0.22, + "learning_rate": 6.953655642446368e-08, + "logits/chosen": -1.3810409307479858, + "logits/rejected": -1.3840851783752441, + "logps/chosen": -19.21212387084961, + "logps/rejected": -23.80167579650879, + "loss": 0.6577, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.18992505967617035, + "rewards/margins": 0.10662909597158432, + "rewards/rejected": -0.2965541481971741, + "step": 435 + }, + { + "epoch": 0.22, + "learning_rate": 6.87303296707956e-08, + "logits/chosen": -1.3490447998046875, + "logits/rejected": -1.374977707862854, + "logps/chosen": -15.331153869628906, + "logps/rejected": -22.786605834960938, + "loss": 0.6525, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.2073613405227661, + "rewards/margins": 0.13075020909309387, + "rewards/rejected": -0.3381115794181824, + "step": 440 + }, + { + "epoch": 0.22, + "learning_rate": 6.7918397477265e-08, + "logits/chosen": -1.4382171630859375, + "logits/rejected": -1.4352459907531738, + "logps/chosen": -17.559337615966797, + "logps/rejected": -21.59587860107422, + "loss": 0.6659, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.20677068829536438, + "rewards/margins": 0.11824776977300644, + "rewards/rejected": -0.3250184655189514, + "step": 445 + }, + { + "epoch": 0.23, + "learning_rate": 6.710100716628345e-08, + "logits/chosen": -1.5328034162521362, + "logits/rejected": -1.5469181537628174, + "logps/chosen": -17.923799514770508, + "logps/rejected": -21.902999877929688, + "loss": 0.6573, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.21206100285053253, + "rewards/margins": 0.102953240275383, + "rewards/rejected": -0.3150142729282379, + "step": 450 + }, + { + "epoch": 0.23, + "learning_rate": 6.627840772285784e-08, + "logits/chosen": -1.4326767921447754, + "logits/rejected": -1.460436224937439, + "logps/chosen": -16.84982681274414, + "logps/rejected": -21.357173919677734, + "loss": 0.6489, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.18491019308567047, + "rewards/margins": 0.1146242767572403, + "rewards/rejected": -0.29953449964523315, + "step": 455 + }, + { + "epoch": 0.23, + "learning_rate": 6.545084971874738e-08, + "logits/chosen": -1.4505399465560913, + "logits/rejected": -1.4668846130371094, + "logps/chosen": -17.121883392333984, + "logps/rejected": -21.807418823242188, + "loss": 0.6563, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.22878995537757874, + "rewards/margins": 0.12952670454978943, + "rewards/rejected": -0.3583166301250458, + "step": 460 + }, + { + "epoch": 0.23, + "learning_rate": 6.461858523613683e-08, + "logits/chosen": -1.4589532613754272, + "logits/rejected": -1.4687752723693848, + "logps/chosen": -16.5905818939209, + "logps/rejected": -22.4600830078125, + "loss": 0.6516, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.23870249092578888, + "rewards/margins": 0.09481547772884369, + "rewards/rejected": -0.3335179388523102, + "step": 465 + }, + { + "epoch": 0.23, + "learning_rate": 6.378186779084996e-08, + "logits/chosen": -1.4866201877593994, + "logits/rejected": -1.4896290302276611, + "logps/chosen": -17.042253494262695, + "logps/rejected": -23.417699813842773, + "loss": 0.6498, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.22306545078754425, + "rewards/margins": 0.13384439051151276, + "rewards/rejected": -0.3569098114967346, + "step": 470 + }, + { + "epoch": 0.24, + "learning_rate": 6.294095225512604e-08, + "logits/chosen": -1.4022520780563354, + "logits/rejected": -1.3814033269882202, + "logps/chosen": -16.337562561035156, + "logps/rejected": -21.31450653076172, + "loss": 0.6569, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.2246430367231369, + "rewards/margins": 0.15105651319026947, + "rewards/rejected": -0.37569957971572876, + "step": 475 + }, + { + "epoch": 0.24, + "learning_rate": 6.209609477998338e-08, + "logits/chosen": -1.479479193687439, + "logits/rejected": -1.5034449100494385, + "logps/chosen": -17.957107543945312, + "logps/rejected": -23.098215103149414, + "loss": 0.6629, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.24643921852111816, + "rewards/margins": 0.1044594869017601, + "rewards/rejected": -0.3508986830711365, + "step": 480 + }, + { + "epoch": 0.24, + "learning_rate": 6.124755271719325e-08, + "logits/chosen": -1.4407473802566528, + "logits/rejected": -1.49507737159729, + "logps/chosen": -14.73084545135498, + "logps/rejected": -20.570337295532227, + "loss": 0.6481, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.19731993973255157, + "rewards/margins": 0.14940151572227478, + "rewards/rejected": -0.34672147035598755, + "step": 485 + }, + { + "epoch": 0.24, + "learning_rate": 6.039558454088796e-08, + "logits/chosen": -1.5169572830200195, + "logits/rejected": -1.5479459762573242, + "logps/chosen": -13.957262992858887, + "logps/rejected": -22.48187255859375, + "loss": 0.6496, + "rewards/accuracies": 0.706250011920929, + "rewards/chosen": -0.20421214401721954, + "rewards/margins": 0.16293281316757202, + "rewards/rejected": -0.36714497208595276, + "step": 490 + }, + { + "epoch": 0.25, + "learning_rate": 5.954044976882724e-08, + "logits/chosen": -1.4714252948760986, + "logits/rejected": -1.4994815587997437, + "logps/chosen": -17.487274169921875, + "logps/rejected": -23.63813591003418, + "loss": 0.6526, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.2372887134552002, + "rewards/margins": 0.12310683727264404, + "rewards/rejected": -0.36039552092552185, + "step": 495 + }, + { + "epoch": 0.25, + "learning_rate": 5.868240888334653e-08, + "logits/chosen": -1.4606778621673584, + "logits/rejected": -1.4551252126693726, + "logps/chosen": -16.36603355407715, + "logps/rejected": -21.119382858276367, + "loss": 0.6562, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.2265624701976776, + "rewards/margins": 0.16614754498004913, + "rewards/rejected": -0.39271003007888794, + "step": 500 + }, + { + "epoch": 0.25, + "eval_logits/chosen": -1.7672057151794434, + "eval_logits/rejected": -1.8467962741851807, + "eval_logps/chosen": -17.221099853515625, + "eval_logps/rejected": -22.007465362548828, + "eval_loss": 0.6560497283935547, + "eval_rewards/accuracies": 0.634984016418457, + "eval_rewards/chosen": -0.22079555690288544, + "eval_rewards/margins": 0.12612493336200714, + "eval_rewards/rejected": -0.3469204902648926, + "eval_runtime": 306.6923, + "eval_samples_per_second": 65.212, + "eval_steps_per_second": 1.021, + "step": 500 + }, + { + "epoch": 0.25, + "learning_rate": 5.7821723252011546e-08, + "logits/chosen": -1.4445126056671143, + "logits/rejected": -1.4752463102340698, + "logps/chosen": -18.014240264892578, + "logps/rejected": -25.679813385009766, + "loss": 0.6619, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.2500578463077545, + "rewards/margins": 0.07094712555408478, + "rewards/rejected": -0.3210049867630005, + "step": 505 + }, + { + "epoch": 0.26, + "learning_rate": 5.695865504800327e-08, + "logits/chosen": -1.4805431365966797, + "logits/rejected": -1.4579927921295166, + "logps/chosen": -16.328933715820312, + "logps/rejected": -23.8695068359375, + "loss": 0.6543, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.2136952131986618, + "rewards/margins": 0.15080443024635315, + "rewards/rejected": -0.36449965834617615, + "step": 510 + }, + { + "epoch": 0.26, + "learning_rate": 5.6093467170257366e-08, + "logits/chosen": -1.38763427734375, + "logits/rejected": -1.429253339767456, + "logps/chosen": -17.211883544921875, + "logps/rejected": -21.488162994384766, + "loss": 0.6454, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.23104102909564972, + "rewards/margins": 0.11019430309534073, + "rewards/rejected": -0.34123533964157104, + "step": 515 + }, + { + "epoch": 0.26, + "learning_rate": 5.5226423163382677e-08, + "logits/chosen": -1.3689727783203125, + "logits/rejected": -1.3758208751678467, + "logps/chosen": -18.692577362060547, + "logps/rejected": -21.925251007080078, + "loss": 0.6466, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.21853986382484436, + "rewards/margins": 0.09194198995828629, + "rewards/rejected": -0.31048184633255005, + "step": 520 + }, + { + "epoch": 0.26, + "learning_rate": 5.435778713738292e-08, + "logits/chosen": -1.3423216342926025, + "logits/rejected": -1.355452299118042, + "logps/chosen": -16.941513061523438, + "logps/rejected": -20.827672958374023, + "loss": 0.652, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2414543330669403, + "rewards/margins": 0.10109053552150726, + "rewards/rejected": -0.3425448536872864, + "step": 525 + }, + { + "epoch": 0.27, + "learning_rate": 5.3487823687206256e-08, + "logits/chosen": -1.4701192378997803, + "logits/rejected": -1.4796873331069946, + "logps/chosen": -19.578105926513672, + "logps/rejected": -25.23065757751465, + "loss": 0.6484, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.251248836517334, + "rewards/margins": 0.13180874288082123, + "rewards/rejected": -0.3830576241016388, + "step": 530 + }, + { + "epoch": 0.27, + "learning_rate": 5.261679781214719e-08, + "logits/chosen": -1.435497522354126, + "logits/rejected": -1.459380865097046, + "logps/chosen": -15.287922859191895, + "logps/rejected": -20.136037826538086, + "loss": 0.6582, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.23688891530036926, + "rewards/margins": 0.1384437382221222, + "rewards/rejected": -0.37533265352249146, + "step": 535 + }, + { + "epoch": 0.27, + "learning_rate": 5.1744974835125056e-08, + "logits/chosen": -1.4165819883346558, + "logits/rejected": -1.3940761089324951, + "logps/chosen": -17.587196350097656, + "logps/rejected": -21.539077758789062, + "loss": 0.6443, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.2061920464038849, + "rewards/margins": 0.133754700422287, + "rewards/rejected": -0.3399467468261719, + "step": 540 + }, + { + "epoch": 0.27, + "learning_rate": 5.087262032186418e-08, + "logits/chosen": -1.4642976522445679, + "logits/rejected": -1.469395637512207, + "logps/chosen": -19.382497787475586, + "logps/rejected": -25.097797393798828, + "loss": 0.6469, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.22812576591968536, + "rewards/margins": 0.1407867670059204, + "rewards/rejected": -0.3689125180244446, + "step": 545 + }, + { + "epoch": 0.28, + "learning_rate": 5e-08, + "logits/chosen": -1.4709419012069702, + "logits/rejected": -1.4961137771606445, + "logps/chosen": -18.24771499633789, + "logps/rejected": -22.102489471435547, + "loss": 0.6616, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.23151636123657227, + "rewards/margins": 0.10363288968801498, + "rewards/rejected": -0.33514922857284546, + "step": 550 + }, + { + "epoch": 0.28, + "learning_rate": 4.912737967813582e-08, + "logits/chosen": -1.425290584564209, + "logits/rejected": -1.4705363512039185, + "logps/chosen": -20.757022857666016, + "logps/rejected": -25.79976463317871, + "loss": 0.6412, + "rewards/accuracies": 0.606249988079071, + "rewards/chosen": -0.2047382891178131, + "rewards/margins": 0.147218257188797, + "rewards/rejected": -0.3519565165042877, + "step": 555 + }, + { + "epoch": 0.28, + "learning_rate": 4.8255025164874966e-08, + "logits/chosen": -1.4153515100479126, + "logits/rejected": -1.4406566619873047, + "logps/chosen": -15.806404113769531, + "logps/rejected": -23.537967681884766, + "loss": 0.6501, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.26336127519607544, + "rewards/margins": 0.0594928041100502, + "rewards/rejected": -0.32285410165786743, + "step": 560 + }, + { + "epoch": 0.28, + "learning_rate": 4.73832021878528e-08, + "logits/chosen": -1.4600375890731812, + "logits/rejected": -1.465736746788025, + "logps/chosen": -14.703231811523438, + "logps/rejected": -22.058746337890625, + "loss": 0.6494, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.2418874204158783, + "rewards/margins": 0.14205661416053772, + "rewards/rejected": -0.383944034576416, + "step": 565 + }, + { + "epoch": 0.28, + "learning_rate": 4.651217631279373e-08, + "logits/chosen": -1.488512396812439, + "logits/rejected": -1.5095858573913574, + "logps/chosen": -16.324678421020508, + "logps/rejected": -24.393007278442383, + "loss": 0.6459, + "rewards/accuracies": 0.6812499761581421, + "rewards/chosen": -0.22573427855968475, + "rewards/margins": 0.1321270912885666, + "rewards/rejected": -0.35786136984825134, + "step": 570 + }, + { + "epoch": 0.29, + "learning_rate": 4.5642212862617084e-08, + "logits/chosen": -1.4317299127578735, + "logits/rejected": -1.455801010131836, + "logps/chosen": -15.989962577819824, + "logps/rejected": -18.849998474121094, + "loss": 0.6477, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2763644754886627, + "rewards/margins": 0.11443523317575455, + "rewards/rejected": -0.39079970121383667, + "step": 575 + }, + { + "epoch": 0.29, + "learning_rate": 4.477357683661733e-08, + "logits/chosen": -1.4575328826904297, + "logits/rejected": -1.470690131187439, + "logps/chosen": -18.188373565673828, + "logps/rejected": -22.835786819458008, + "loss": 0.6512, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -0.22157052159309387, + "rewards/margins": 0.10469740629196167, + "rewards/rejected": -0.32626792788505554, + "step": 580 + }, + { + "epoch": 0.29, + "learning_rate": 4.390653282974263e-08, + "logits/chosen": -1.503989815711975, + "logits/rejected": -1.519313931465149, + "logps/chosen": -16.674240112304688, + "logps/rejected": -21.952165603637695, + "loss": 0.6365, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.2247585952281952, + "rewards/margins": 0.203842431306839, + "rewards/rejected": -0.4286009669303894, + "step": 585 + }, + { + "epoch": 0.29, + "learning_rate": 4.304134495199674e-08, + "logits/chosen": -1.4609280824661255, + "logits/rejected": -1.4766910076141357, + "logps/chosen": -19.406192779541016, + "logps/rejected": -22.686025619506836, + "loss": 0.6364, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2958422303199768, + "rewards/margins": 0.11995653063058853, + "rewards/rejected": -0.4157988131046295, + "step": 590 + }, + { + "epoch": 0.3, + "learning_rate": 4.217827674798844e-08, + "logits/chosen": -1.4871938228607178, + "logits/rejected": -1.5188095569610596, + "logps/chosen": -17.268047332763672, + "logps/rejected": -23.33865737915039, + "loss": 0.6532, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.31558969616889954, + "rewards/margins": 0.11393972486257553, + "rewards/rejected": -0.42952942848205566, + "step": 595 + }, + { + "epoch": 0.3, + "learning_rate": 4.131759111665348e-08, + "logits/chosen": -1.4645878076553345, + "logits/rejected": -1.4607641696929932, + "logps/chosen": -17.968856811523438, + "logps/rejected": -24.378482818603516, + "loss": 0.6483, + "rewards/accuracies": 0.6937500238418579, + "rewards/chosen": -0.20308199524879456, + "rewards/margins": 0.19312706589698792, + "rewards/rejected": -0.3962090015411377, + "step": 600 + }, + { + "epoch": 0.3, + "eval_logits/chosen": -1.762748122215271, + "eval_logits/rejected": -1.8424293994903564, + "eval_logps/chosen": -17.52169418334961, + "eval_logps/rejected": -22.536460876464844, + "eval_loss": 0.65033358335495, + "eval_rewards/accuracies": 0.6413738131523132, + "eval_rewards/chosen": -0.2508549690246582, + "eval_rewards/margins": 0.14896489679813385, + "eval_rewards/rejected": -0.39981985092163086, + "eval_runtime": 306.9145, + "eval_samples_per_second": 65.165, + "eval_steps_per_second": 1.02, + "step": 600 + }, + { + "epoch": 0.3, + "learning_rate": 4.0459550231172757e-08, + "logits/chosen": -1.3895206451416016, + "logits/rejected": -1.4285638332366943, + "logps/chosen": -17.535720825195312, + "logps/rejected": -24.525096893310547, + "loss": 0.6521, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.2898118793964386, + "rewards/margins": 0.08567583560943604, + "rewards/rejected": -0.37548771500587463, + "step": 605 + }, + { + "epoch": 0.3, + "learning_rate": 3.960441545911204e-08, + "logits/chosen": -1.4046005010604858, + "logits/rejected": -1.4093401432037354, + "logps/chosen": -19.01970863342285, + "logps/rejected": -22.846820831298828, + "loss": 0.6486, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.28651127219200134, + "rewards/margins": 0.07938197255134583, + "rewards/rejected": -0.36589327454566956, + "step": 610 + }, + { + "epoch": 0.31, + "learning_rate": 3.8752447282806754e-08, + "logits/chosen": -1.4743293523788452, + "logits/rejected": -1.481278419494629, + "logps/chosen": -15.964106559753418, + "logps/rejected": -22.171709060668945, + "loss": 0.6402, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.21257051825523376, + "rewards/margins": 0.2042216807603836, + "rewards/rejected": -0.41679221391677856, + "step": 615 + }, + { + "epoch": 0.31, + "learning_rate": 3.7903905220016615e-08, + "logits/chosen": -1.3876988887786865, + "logits/rejected": -1.410398244857788, + "logps/chosen": -17.697256088256836, + "logps/rejected": -23.332763671875, + "loss": 0.6429, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.23554810881614685, + "rewards/margins": 0.13481906056404114, + "rewards/rejected": -0.3703671991825104, + "step": 620 + }, + { + "epoch": 0.31, + "learning_rate": 3.705904774487396e-08, + "logits/chosen": -1.383541464805603, + "logits/rejected": -1.3796765804290771, + "logps/chosen": -18.24690818786621, + "logps/rejected": -24.86931610107422, + "loss": 0.6398, + "rewards/accuracies": 0.606249988079071, + "rewards/chosen": -0.23973222076892853, + "rewards/margins": 0.16422876715660095, + "rewards/rejected": -0.4039610028266907, + "step": 625 + }, + { + "epoch": 0.32, + "learning_rate": 3.621813220915004e-08, + "logits/chosen": -1.450282335281372, + "logits/rejected": -1.4380896091461182, + "logps/chosen": -16.290973663330078, + "logps/rejected": -21.838808059692383, + "loss": 0.6481, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.2356175184249878, + "rewards/margins": 0.14112675189971924, + "rewards/rejected": -0.37674424052238464, + "step": 630 + }, + { + "epoch": 0.32, + "learning_rate": 3.538141476386316e-08, + "logits/chosen": -1.4591354131698608, + "logits/rejected": -1.456343412399292, + "logps/chosen": -17.22078514099121, + "logps/rejected": -21.944005966186523, + "loss": 0.6491, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.2835259735584259, + "rewards/margins": 0.10515954345464706, + "rewards/rejected": -0.38868552446365356, + "step": 635 + }, + { + "epoch": 0.32, + "learning_rate": 3.4549150281252633e-08, + "logits/chosen": -1.428468108177185, + "logits/rejected": -1.4328949451446533, + "logps/chosen": -16.396862030029297, + "logps/rejected": -20.86203384399414, + "loss": 0.6369, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.2767188251018524, + "rewards/margins": 0.12265461683273315, + "rewards/rejected": -0.39937347173690796, + "step": 640 + }, + { + "epoch": 0.32, + "learning_rate": 3.372159227714218e-08, + "logits/chosen": -1.3802716732025146, + "logits/rejected": -1.4310414791107178, + "logps/chosen": -18.36510467529297, + "logps/rejected": -20.808429718017578, + "loss": 0.6429, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.2337242066860199, + "rewards/margins": 0.1597466617822647, + "rewards/rejected": -0.3934708833694458, + "step": 645 + }, + { + "epoch": 0.33, + "learning_rate": 3.2898992833716563e-08, + "logits/chosen": -1.482743263244629, + "logits/rejected": -1.4772045612335205, + "logps/chosen": -15.889050483703613, + "logps/rejected": -19.02865982055664, + "loss": 0.6478, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2292957603931427, + "rewards/margins": 0.16532939672470093, + "rewards/rejected": -0.39462512731552124, + "step": 650 + }, + { + "epoch": 0.33, + "learning_rate": 3.208160252273498e-08, + "logits/chosen": -1.4797611236572266, + "logits/rejected": -1.4938347339630127, + "logps/chosen": -18.688114166259766, + "logps/rejected": -22.962867736816406, + "loss": 0.644, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.2989271283149719, + "rewards/margins": 0.12799039483070374, + "rewards/rejected": -0.4269174635410309, + "step": 655 + }, + { + "epoch": 0.33, + "learning_rate": 3.126967032920439e-08, + "logits/chosen": -1.4451311826705933, + "logits/rejected": -1.4879848957061768, + "logps/chosen": -17.379915237426758, + "logps/rejected": -23.540796279907227, + "loss": 0.6363, + "rewards/accuracies": 0.5687500238418579, + "rewards/chosen": -0.23357510566711426, + "rewards/margins": 0.11232779920101166, + "rewards/rejected": -0.3459029197692871, + "step": 660 + }, + { + "epoch": 0.33, + "learning_rate": 3.046344357553632e-08, + "logits/chosen": -1.5276529788970947, + "logits/rejected": -1.5440245866775513, + "logps/chosen": -17.0556640625, + "logps/rejected": -22.744121551513672, + "loss": 0.6308, + "rewards/accuracies": 0.6937500238418579, + "rewards/chosen": -0.23212318122386932, + "rewards/margins": 0.16936565935611725, + "rewards/rejected": -0.4014888405799866, + "step": 665 + }, + { + "epoch": 0.34, + "learning_rate": 2.9663167846209998e-08, + "logits/chosen": -1.4147757291793823, + "logits/rejected": -1.418082356452942, + "logps/chosen": -19.272014617919922, + "logps/rejected": -27.263992309570312, + "loss": 0.6512, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.2706632614135742, + "rewards/margins": 0.18588735163211823, + "rewards/rejected": -0.45655059814453125, + "step": 670 + }, + { + "epoch": 0.34, + "learning_rate": 2.8869086912965035e-08, + "logits/chosen": -1.4479643106460571, + "logits/rejected": -1.4662295579910278, + "logps/chosen": -17.57897186279297, + "logps/rejected": -25.184152603149414, + "loss": 0.6423, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.2373163402080536, + "rewards/margins": 0.2066657841205597, + "rewards/rejected": -0.4439820647239685, + "step": 675 + }, + { + "epoch": 0.34, + "learning_rate": 2.8081442660546124e-08, + "logits/chosen": -1.36322820186615, + "logits/rejected": -1.3949863910675049, + "logps/chosen": -18.550283432006836, + "logps/rejected": -23.298870086669922, + "loss": 0.6416, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.25548475980758667, + "rewards/margins": 0.18017227947711945, + "rewards/rejected": -0.4356570839881897, + "step": 680 + }, + { + "epoch": 0.34, + "learning_rate": 2.730047501302266e-08, + "logits/chosen": -1.4136111736297607, + "logits/rejected": -1.4232200384140015, + "logps/chosen": -17.45450210571289, + "logps/rejected": -20.295621871948242, + "loss": 0.6415, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2979032099246979, + "rewards/margins": 0.12163282930850983, + "rewards/rejected": -0.4195360243320465, + "step": 685 + }, + { + "epoch": 0.34, + "learning_rate": 2.6526421860705472e-08, + "logits/chosen": -1.384798526763916, + "logits/rejected": -1.4583854675292969, + "logps/chosen": -16.522525787353516, + "logps/rejected": -23.442546844482422, + "loss": 0.6288, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.23055878281593323, + "rewards/margins": 0.23925617337226868, + "rewards/rejected": -0.4698149561882019, + "step": 690 + }, + { + "epoch": 0.35, + "learning_rate": 2.5759518987683148e-08, + "logits/chosen": -1.4321386814117432, + "logits/rejected": -1.4662306308746338, + "logps/chosen": -17.545879364013672, + "logps/rejected": -20.307212829589844, + "loss": 0.6431, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.31093353033065796, + "rewards/margins": 0.1719907969236374, + "rewards/rejected": -0.48292431235313416, + "step": 695 + }, + { + "epoch": 0.35, + "learning_rate": 2.500000000000001e-08, + "logits/chosen": -1.4065908193588257, + "logits/rejected": -1.4342257976531982, + "logps/chosen": -17.379558563232422, + "logps/rejected": -22.622777938842773, + "loss": 0.6413, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.27703213691711426, + "rewards/margins": 0.15564481914043427, + "rewards/rejected": -0.4326769709587097, + "step": 700 + }, + { + "epoch": 0.35, + "eval_logits/chosen": -1.7605791091918945, + "eval_logits/rejected": -1.8403115272521973, + "eval_logps/chosen": -17.779977798461914, + "eval_logps/rejected": -22.98041343688965, + "eval_loss": 0.6457803845405579, + "eval_rewards/accuracies": 0.6469648480415344, + "eval_rewards/chosen": -0.27668341994285583, + "eval_rewards/margins": 0.16753174364566803, + "eval_rewards/rejected": -0.44421514868736267, + "eval_runtime": 307.0164, + "eval_samples_per_second": 65.143, + "eval_steps_per_second": 1.019, + "step": 700 + }, + { + "epoch": 0.35, + "learning_rate": 2.4248096254497287e-08, + "logits/chosen": -1.4401991367340088, + "logits/rejected": -1.4596917629241943, + "logps/chosen": -17.921367645263672, + "logps/rejected": -23.322214126586914, + "loss": 0.6378, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.29132723808288574, + "rewards/margins": 0.19050189852714539, + "rewards/rejected": -0.4818291664123535, + "step": 705 + }, + { + "epoch": 0.35, + "learning_rate": 2.350403678833976e-08, + "logits/chosen": -1.4126722812652588, + "logits/rejected": -1.420345425605774, + "logps/chosen": -19.046810150146484, + "logps/rejected": -25.77749252319336, + "loss": 0.6416, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.30587273836135864, + "rewards/margins": 0.18231990933418274, + "rewards/rejected": -0.4881926476955414, + "step": 710 + }, + { + "epoch": 0.36, + "learning_rate": 2.2768048249248644e-08, + "logits/chosen": -1.3649728298187256, + "logits/rejected": -1.3823496103286743, + "logps/chosen": -17.098003387451172, + "logps/rejected": -21.837902069091797, + "loss": 0.6356, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.33593738079071045, + "rewards/margins": 0.1847352534532547, + "rewards/rejected": -0.520672619342804, + "step": 715 + }, + { + "epoch": 0.36, + "learning_rate": 2.2040354826462664e-08, + "logits/chosen": -1.3861441612243652, + "logits/rejected": -1.3953332901000977, + "logps/chosen": -17.742359161376953, + "logps/rejected": -24.968896865844727, + "loss": 0.6353, + "rewards/accuracies": 0.606249988079071, + "rewards/chosen": -0.32852208614349365, + "rewards/margins": 0.14815881848335266, + "rewards/rejected": -0.4766809046268463, + "step": 720 + }, + { + "epoch": 0.36, + "learning_rate": 2.1321178182447707e-08, + "logits/chosen": -1.3506596088409424, + "logits/rejected": -1.3595778942108154, + "logps/chosen": -19.100223541259766, + "logps/rejected": -24.44327163696289, + "loss": 0.6449, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.31565234065055847, + "rewards/margins": 0.11223573982715607, + "rewards/rejected": -0.42788806557655334, + "step": 725 + }, + { + "epoch": 0.36, + "learning_rate": 2.0610737385376347e-08, + "logits/chosen": -1.4714252948760986, + "logits/rejected": -1.4704372882843018, + "logps/chosen": -20.749143600463867, + "logps/rejected": -23.971107482910156, + "loss": 0.6357, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.31912142038345337, + "rewards/margins": 0.1428820788860321, + "rewards/rejected": -0.4620034694671631, + "step": 730 + }, + { + "epoch": 0.37, + "learning_rate": 1.990924884239758e-08, + "logits/chosen": -1.4702327251434326, + "logits/rejected": -1.4725419282913208, + "logps/chosen": -19.519115447998047, + "logps/rejected": -21.988479614257812, + "loss": 0.6404, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.30419957637786865, + "rewards/margins": 0.16355463862419128, + "rewards/rejected": -0.46775418519973755, + "step": 735 + }, + { + "epoch": 0.37, + "learning_rate": 1.9216926233717085e-08, + "logits/chosen": -1.4942646026611328, + "logits/rejected": -1.5009443759918213, + "logps/chosen": -16.522647857666016, + "logps/rejected": -24.716516494750977, + "loss": 0.6315, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.2690926194190979, + "rewards/margins": 0.16352799534797668, + "rewards/rejected": -0.4326205849647522, + "step": 740 + }, + { + "epoch": 0.37, + "learning_rate": 1.8533980447508135e-08, + "logits/chosen": -1.4108445644378662, + "logits/rejected": -1.4322090148925781, + "logps/chosen": -18.862674713134766, + "logps/rejected": -23.953310012817383, + "loss": 0.636, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.2986528277397156, + "rewards/margins": 0.14546221494674683, + "rewards/rejected": -0.4441150724887848, + "step": 745 + }, + { + "epoch": 0.38, + "learning_rate": 1.786061951567303e-08, + "logits/chosen": -1.4277067184448242, + "logits/rejected": -1.438820481300354, + "logps/chosen": -19.40964126586914, + "logps/rejected": -26.506017684936523, + "loss": 0.6373, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.29312849044799805, + "rewards/margins": 0.17660866677761078, + "rewards/rejected": -0.46973714232444763, + "step": 750 + }, + { + "epoch": 0.38, + "learning_rate": 1.719704855047464e-08, + "logits/chosen": -1.408469319343567, + "logits/rejected": -1.4118237495422363, + "logps/chosen": -17.609798431396484, + "logps/rejected": -22.488494873046875, + "loss": 0.636, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.29817792773246765, + "rewards/margins": 0.1390009969472885, + "rewards/rejected": -0.43717893958091736, + "step": 755 + }, + { + "epoch": 0.38, + "learning_rate": 1.6543469682057103e-08, + "logits/chosen": -1.3890222311019897, + "logits/rejected": -1.4223724603652954, + "logps/chosen": -16.866146087646484, + "logps/rejected": -21.382225036621094, + "loss": 0.6432, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.23102489113807678, + "rewards/margins": 0.185149148106575, + "rewards/rejected": -0.4161740243434906, + "step": 760 + }, + { + "epoch": 0.38, + "learning_rate": 1.590008199687508e-08, + "logits/chosen": -1.4136755466461182, + "logits/rejected": -1.4180434942245483, + "logps/chosen": -17.45306968688965, + "logps/rejected": -20.812774658203125, + "loss": 0.6425, + "rewards/accuracies": 0.5687500238418579, + "rewards/chosen": -0.29186105728149414, + "rewards/margins": 0.115916408598423, + "rewards/rejected": -0.40777745842933655, + "step": 765 + }, + { + "epoch": 0.39, + "learning_rate": 1.526708147705013e-08, + "logits/chosen": -1.4240939617156982, + "logits/rejected": -1.4586817026138306, + "logps/chosen": -17.540027618408203, + "logps/rejected": -22.493268966674805, + "loss": 0.6404, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.2557734549045563, + "rewards/margins": 0.17367486655712128, + "rewards/rejected": -0.42944836616516113, + "step": 770 + }, + { + "epoch": 0.39, + "learning_rate": 1.4644660940672625e-08, + "logits/chosen": -1.3837788105010986, + "logits/rejected": -1.4116965532302856, + "logps/chosen": -17.056177139282227, + "logps/rejected": -25.01016616821289, + "loss": 0.6338, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.28285446763038635, + "rewards/margins": 0.2001866102218628, + "rewards/rejected": -0.48304110765457153, + "step": 775 + }, + { + "epoch": 0.39, + "learning_rate": 1.4033009983067451e-08, + "logits/chosen": -1.4337577819824219, + "logits/rejected": -1.446617603302002, + "logps/chosen": -16.082469940185547, + "logps/rejected": -22.56252670288086, + "loss": 0.6439, + "rewards/accuracies": 0.71875, + "rewards/chosen": -0.2657018005847931, + "rewards/margins": 0.24111859500408173, + "rewards/rejected": -0.5068204402923584, + "step": 780 + }, + { + "epoch": 0.39, + "learning_rate": 1.3432314919041476e-08, + "logits/chosen": -1.47090744972229, + "logits/rejected": -1.508124589920044, + "logps/chosen": -14.742795944213867, + "logps/rejected": -19.045040130615234, + "loss": 0.6327, + "rewards/accuracies": 0.731249988079071, + "rewards/chosen": -0.2083434760570526, + "rewards/margins": 0.19431748986244202, + "rewards/rejected": -0.402660995721817, + "step": 785 + }, + { + "epoch": 0.4, + "learning_rate": 1.2842758726130282e-08, + "logits/chosen": -1.404345989227295, + "logits/rejected": -1.4321860074996948, + "logps/chosen": -18.46529769897461, + "logps/rejected": -22.419506072998047, + "loss": 0.6581, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.3104281723499298, + "rewards/margins": 0.12459827959537506, + "rewards/rejected": -0.43502646684646606, + "step": 790 + }, + { + "epoch": 0.4, + "learning_rate": 1.2264520988861398e-08, + "logits/chosen": -1.528306007385254, + "logits/rejected": -1.548837661743164, + "logps/chosen": -18.500003814697266, + "logps/rejected": -22.167816162109375, + "loss": 0.6485, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.29536038637161255, + "rewards/margins": 0.1245318204164505, + "rewards/rejected": -0.41989216208457947, + "step": 795 + }, + { + "epoch": 0.4, + "learning_rate": 1.1697777844051105e-08, + "logits/chosen": -1.4373068809509277, + "logits/rejected": -1.4536762237548828, + "logps/chosen": -19.13837242126465, + "logps/rejected": -21.966625213623047, + "loss": 0.6364, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.26146507263183594, + "rewards/margins": 0.19086112082004547, + "rewards/rejected": -0.4523262083530426, + "step": 800 + }, + { + "epoch": 0.4, + "eval_logits/chosen": -1.7574115991592407, + "eval_logits/rejected": -1.8370665311813354, + "eval_logps/chosen": -17.89972496032715, + "eval_logps/rejected": -23.20018768310547, + "eval_loss": 0.6436580419540405, + "eval_rewards/accuracies": 0.6449680328369141, + "eval_rewards/chosen": -0.28865811228752136, + "eval_rewards/margins": 0.17753452062606812, + "eval_rewards/rejected": -0.4661926031112671, + "eval_runtime": 306.9231, + "eval_samples_per_second": 65.163, + "eval_steps_per_second": 1.02, + "step": 800 + }, + { + "epoch": 0.4, + "learning_rate": 1.1142701927151454e-08, + "logits/chosen": -1.4606168270111084, + "logits/rejected": -1.4665342569351196, + "logps/chosen": -15.579089164733887, + "logps/rejected": -22.30219268798828, + "loss": 0.6483, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.2538829445838928, + "rewards/margins": 0.22965078055858612, + "rewards/rejected": -0.4835337698459625, + "step": 805 + }, + { + "epoch": 0.41, + "learning_rate": 1.0599462319663904e-08, + "logits/chosen": -1.3860244750976562, + "logits/rejected": -1.3826844692230225, + "logps/chosen": -17.648862838745117, + "logps/rejected": -25.444860458374023, + "loss": 0.6498, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.2895384132862091, + "rewards/margins": 0.12358560413122177, + "rewards/rejected": -0.4131239950656891, + "step": 810 + }, + { + "epoch": 0.41, + "learning_rate": 1.0068224497635369e-08, + "logits/chosen": -1.4392060041427612, + "logits/rejected": -1.460053563117981, + "logps/chosen": -17.04248809814453, + "logps/rejected": -25.65012550354004, + "loss": 0.6428, + "rewards/accuracies": 0.59375, + "rewards/chosen": -0.26920509338378906, + "rewards/margins": 0.1518903225660324, + "rewards/rejected": -0.42109543085098267, + "step": 815 + }, + { + "epoch": 0.41, + "learning_rate": 9.549150281252633e-09, + "logits/chosen": -1.380948781967163, + "logits/rejected": -1.38584566116333, + "logps/chosen": -19.414995193481445, + "logps/rejected": -24.585783004760742, + "loss": 0.6346, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.28162115812301636, + "rewards/margins": 0.15368224680423737, + "rewards/rejected": -0.43530339002609253, + "step": 820 + }, + { + "epoch": 0.41, + "learning_rate": 9.042397785550404e-09, + "logits/chosen": -1.3312206268310547, + "logits/rejected": -1.3463561534881592, + "logps/chosen": -15.558464050292969, + "logps/rejected": -20.260160446166992, + "loss": 0.6395, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.2858559489250183, + "rewards/margins": 0.15887019038200378, + "rewards/rejected": -0.4447261691093445, + "step": 825 + }, + { + "epoch": 0.41, + "learning_rate": 8.548121372247919e-09, + "logits/chosen": -1.403304100036621, + "logits/rejected": -1.3955562114715576, + "logps/chosen": -17.097698211669922, + "logps/rejected": -23.62860679626465, + "loss": 0.6436, + "rewards/accuracies": 0.5562499761581421, + "rewards/chosen": -0.3152267336845398, + "rewards/margins": 0.09969434142112732, + "rewards/rejected": -0.4149211049079895, + "step": 830 + }, + { + "epoch": 0.42, + "learning_rate": 8.066471602728803e-09, + "logits/chosen": -1.4208085536956787, + "logits/rejected": -1.4421093463897705, + "logps/chosen": -16.878705978393555, + "logps/rejected": -18.381912231445312, + "loss": 0.6342, + "rewards/accuracies": 0.6812499761581421, + "rewards/chosen": -0.2497948855161667, + "rewards/margins": 0.19130581617355347, + "rewards/rejected": -0.44110068678855896, + "step": 835 + }, + { + "epoch": 0.42, + "learning_rate": 7.597595192178703e-09, + "logits/chosen": -1.4911779165267944, + "logits/rejected": -1.4802215099334717, + "logps/chosen": -16.187580108642578, + "logps/rejected": -20.237377166748047, + "loss": 0.6369, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.30556631088256836, + "rewards/margins": 0.13740003108978271, + "rewards/rejected": -0.4429663121700287, + "step": 840 + }, + { + "epoch": 0.42, + "learning_rate": 7.1416349648943884e-09, + "logits/chosen": -1.4348905086517334, + "logits/rejected": -1.4343273639678955, + "logps/chosen": -18.559795379638672, + "logps/rejected": -23.7816219329834, + "loss": 0.6424, + "rewards/accuracies": 0.643750011920929, + "rewards/chosen": -0.3174014091491699, + "rewards/margins": 0.12262705713510513, + "rewards/rejected": -0.44002848863601685, + "step": 845 + }, + { + "epoch": 0.42, + "learning_rate": 6.698729810778064e-09, + "logits/chosen": -1.4852615594863892, + "logits/rejected": -1.4959180355072021, + "logps/chosen": -18.27218246459961, + "logps/rejected": -25.438560485839844, + "loss": 0.651, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.3046795427799225, + "rewards/margins": 0.14905641973018646, + "rewards/rejected": -0.45373591780662537, + "step": 850 + }, + { + "epoch": 0.43, + "learning_rate": 6.269014643030213e-09, + "logits/chosen": -1.4191768169403076, + "logits/rejected": -1.4377117156982422, + "logps/chosen": -17.66164207458496, + "logps/rejected": -24.158462524414062, + "loss": 0.6427, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.29421794414520264, + "rewards/margins": 0.1277884691953659, + "rewards/rejected": -0.4220064580440521, + "step": 855 + }, + { + "epoch": 0.43, + "learning_rate": 5.8526203570536504e-09, + "logits/chosen": -1.4127318859100342, + "logits/rejected": -1.4357506036758423, + "logps/chosen": -17.93946075439453, + "logps/rejected": -25.222787857055664, + "loss": 0.6391, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.3292141258716583, + "rewards/margins": 0.20317533612251282, + "rewards/rejected": -0.5323894619941711, + "step": 860 + }, + { + "epoch": 0.43, + "learning_rate": 5.44967379058161e-09, + "logits/chosen": -1.3914811611175537, + "logits/rejected": -1.4102437496185303, + "logps/chosen": -16.657636642456055, + "logps/rejected": -23.440505981445312, + "loss": 0.6564, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.3101702630519867, + "rewards/margins": 0.18297256529331207, + "rewards/rejected": -0.49314290285110474, + "step": 865 + }, + { + "epoch": 0.43, + "learning_rate": 5.060297685041659e-09, + "logits/chosen": -1.3386437892913818, + "logits/rejected": -1.3571398258209229, + "logps/chosen": -19.986854553222656, + "logps/rejected": -26.74784278869629, + "loss": 0.6279, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.284962922334671, + "rewards/margins": 0.19053895771503448, + "rewards/rejected": -0.4755018651485443, + "step": 870 + }, + { + "epoch": 0.44, + "learning_rate": 4.684610648167503e-09, + "logits/chosen": -1.3065433502197266, + "logits/rejected": -1.3413054943084717, + "logps/chosen": -17.01144027709961, + "logps/rejected": -24.851526260375977, + "loss": 0.6387, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.34229403734207153, + "rewards/margins": 0.10157414525747299, + "rewards/rejected": -0.44386816024780273, + "step": 875 + }, + { + "epoch": 0.44, + "learning_rate": 4.322727117869951e-09, + "logits/chosen": -1.3795998096466064, + "logits/rejected": -1.4208852052688599, + "logps/chosen": -18.107046127319336, + "logps/rejected": -24.59166145324707, + "loss": 0.6355, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.282823383808136, + "rewards/margins": 0.2188301533460617, + "rewards/rejected": -0.5016534924507141, + "step": 880 + }, + { + "epoch": 0.44, + "learning_rate": 3.974757327377981e-09, + "logits/chosen": -1.436051368713379, + "logits/rejected": -1.4475667476654053, + "logps/chosen": -18.486120223999023, + "logps/rejected": -22.97800064086914, + "loss": 0.6469, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.3353201746940613, + "rewards/margins": 0.08744947612285614, + "rewards/rejected": -0.4227696359157562, + "step": 885 + }, + { + "epoch": 0.45, + "learning_rate": 3.640807271660634e-09, + "logits/chosen": -1.464347004890442, + "logits/rejected": -1.4781545400619507, + "logps/chosen": -18.07439613342285, + "logps/rejected": -23.41990089416504, + "loss": 0.6422, + "rewards/accuracies": 0.71875, + "rewards/chosen": -0.259718656539917, + "rewards/margins": 0.25415921211242676, + "rewards/rejected": -0.5138779282569885, + "step": 890 + }, + { + "epoch": 0.45, + "learning_rate": 3.3209786751399183e-09, + "logits/chosen": -1.3919214010238647, + "logits/rejected": -1.4111039638519287, + "logps/chosen": -19.07809066772461, + "logps/rejected": -24.558177947998047, + "loss": 0.6417, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.3080528676509857, + "rewards/margins": 0.1815052181482315, + "rewards/rejected": -0.4895581305027008, + "step": 895 + }, + { + "epoch": 0.45, + "learning_rate": 3.015368960704584e-09, + "logits/chosen": -1.348677158355713, + "logits/rejected": -1.379531979560852, + "logps/chosen": -17.733901977539062, + "logps/rejected": -21.527729034423828, + "loss": 0.6456, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.34989431500434875, + "rewards/margins": 0.13681560754776, + "rewards/rejected": -0.48670992255210876, + "step": 900 + }, + { + "epoch": 0.45, + "eval_logits/chosen": -1.7564817667007446, + "eval_logits/rejected": -1.8361190557479858, + "eval_logps/chosen": -17.921520233154297, + "eval_logps/rejected": -23.25269317626953, + "eval_loss": 0.6429938077926636, + "eval_rewards/accuracies": 0.6417731642723083, + "eval_rewards/chosen": -0.2908374071121216, + "eval_rewards/margins": 0.1806056946516037, + "eval_rewards/rejected": -0.4714431166648865, + "eval_runtime": 307.0371, + "eval_samples_per_second": 65.139, + "eval_steps_per_second": 1.019, + "step": 900 + }, + { + "epoch": 0.45, + "learning_rate": 2.7240712200341577e-09, + "logits/chosen": -1.511375069618225, + "logits/rejected": -1.5339971780776978, + "logps/chosen": -17.652019500732422, + "logps/rejected": -21.41324234008789, + "loss": 0.6269, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.26256105303764343, + "rewards/margins": 0.18523547053337097, + "rewards/rejected": -0.4477965235710144, + "step": 905 + }, + { + "epoch": 0.46, + "learning_rate": 2.4471741852423233e-09, + "logits/chosen": -1.4349921941757202, + "logits/rejected": -1.4263644218444824, + "logps/chosen": -18.641862869262695, + "logps/rejected": -23.057044982910156, + "loss": 0.636, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.29802319407463074, + "rewards/margins": 0.17096181213855743, + "rewards/rejected": -0.468984991312027, + "step": 910 + }, + { + "epoch": 0.46, + "learning_rate": 2.184762201848228e-09, + "logits/chosen": -1.370429277420044, + "logits/rejected": -1.4054522514343262, + "logps/chosen": -17.3892765045166, + "logps/rejected": -24.669536590576172, + "loss": 0.6347, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.2652004361152649, + "rewards/margins": 0.2178630828857422, + "rewards/rejected": -0.4830635190010071, + "step": 915 + }, + { + "epoch": 0.46, + "learning_rate": 1.9369152030840553e-09, + "logits/chosen": -1.4306275844573975, + "logits/rejected": -1.4435356855392456, + "logps/chosen": -16.403339385986328, + "logps/rejected": -23.840957641601562, + "loss": 0.6289, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.23369142413139343, + "rewards/margins": 0.23300664126873016, + "rewards/rejected": -0.4666980803012848, + "step": 920 + }, + { + "epoch": 0.46, + "learning_rate": 1.70370868554659e-09, + "logits/chosen": -1.3685413599014282, + "logits/rejected": -1.3498773574829102, + "logps/chosen": -17.239362716674805, + "logps/rejected": -23.87692642211914, + "loss": 0.6242, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.25025489926338196, + "rewards/margins": 0.25468313694000244, + "rewards/rejected": -0.504938006401062, + "step": 925 + }, + { + "epoch": 0.47, + "learning_rate": 1.4852136862001763e-09, + "logits/chosen": -1.386918306350708, + "logits/rejected": -1.3831664323806763, + "logps/chosen": -19.643451690673828, + "logps/rejected": -24.284130096435547, + "loss": 0.6319, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.31439071893692017, + "rewards/margins": 0.20531371235847473, + "rewards/rejected": -0.5197044610977173, + "step": 930 + }, + { + "epoch": 0.47, + "learning_rate": 1.2814967607382432e-09, + "logits/chosen": -1.454646348953247, + "logits/rejected": -1.463849663734436, + "logps/chosen": -17.558238983154297, + "logps/rejected": -19.34381103515625, + "loss": 0.642, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.29885101318359375, + "rewards/margins": 0.1699037104845047, + "rewards/rejected": -0.46875467896461487, + "step": 935 + }, + { + "epoch": 0.47, + "learning_rate": 1.0926199633097156e-09, + "logits/chosen": -1.3675868511199951, + "logits/rejected": -1.4033797979354858, + "logps/chosen": -17.723363876342773, + "logps/rejected": -21.526758193969727, + "loss": 0.6345, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.29813557863235474, + "rewards/margins": 0.24015481770038605, + "rewards/rejected": -0.5382903814315796, + "step": 940 + }, + { + "epoch": 0.47, + "learning_rate": 9.186408276168012e-10, + "logits/chosen": -1.4319554567337036, + "logits/rejected": -1.4493201971054077, + "logps/chosen": -17.649295806884766, + "logps/rejected": -25.110567092895508, + "loss": 0.6347, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.25603246688842773, + "rewards/margins": 0.21903729438781738, + "rewards/rejected": -0.4750697612762451, + "step": 945 + }, + { + "epoch": 0.47, + "learning_rate": 7.59612349389599e-10, + "logits/chosen": -1.4553452730178833, + "logits/rejected": -1.4630682468414307, + "logps/chosen": -18.7999267578125, + "logps/rejected": -22.81121253967285, + "loss": 0.6351, + "rewards/accuracies": 0.581250011920929, + "rewards/chosen": -0.32581713795661926, + "rewards/margins": 0.13567964732646942, + "rewards/rejected": -0.4614967703819275, + "step": 950 + }, + { + "epoch": 0.48, + "learning_rate": 6.15582970243117e-10, + "logits/chosen": -1.4470014572143555, + "logits/rejected": -1.4603230953216553, + "logps/chosen": -18.624347686767578, + "logps/rejected": -21.459732055664062, + "loss": 0.643, + "rewards/accuracies": 0.6312500238418579, + "rewards/chosen": -0.3545226454734802, + "rewards/margins": 0.13958732783794403, + "rewards/rejected": -0.49411001801490784, + "step": 955 + }, + { + "epoch": 0.48, + "learning_rate": 4.865965629214819e-10, + "logits/chosen": -1.4853070974349976, + "logits/rejected": -1.5119067430496216, + "logps/chosen": -18.299137115478516, + "logps/rejected": -22.493701934814453, + "loss": 0.6401, + "rewards/accuracies": 0.543749988079071, + "rewards/chosen": -0.2782926559448242, + "rewards/margins": 0.11381447315216064, + "rewards/rejected": -0.39210715889930725, + "step": 960 + }, + { + "epoch": 0.48, + "learning_rate": 3.7269241793390084e-10, + "logits/chosen": -1.416803002357483, + "logits/rejected": -1.4471747875213623, + "logps/chosen": -18.96782875061035, + "logps/rejected": -25.271512985229492, + "loss": 0.6433, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.30018851161003113, + "rewards/margins": 0.2068859338760376, + "rewards/rejected": -0.5070745348930359, + "step": 965 + }, + { + "epoch": 0.48, + "learning_rate": 2.739052315863355e-10, + "logits/chosen": -1.475373387336731, + "logits/rejected": -1.4798928499221802, + "logps/chosen": -18.24342918395996, + "logps/rejected": -26.389270782470703, + "loss": 0.64, + "rewards/accuracies": 0.71875, + "rewards/chosen": -0.2784801125526428, + "rewards/margins": 0.23683655261993408, + "rewards/rejected": -0.5153166651725769, + "step": 970 + }, + { + "epoch": 0.49, + "learning_rate": 1.9026509541272272e-10, + "logits/chosen": -1.4730727672576904, + "logits/rejected": -1.482080101966858, + "logps/chosen": -17.36160659790039, + "logps/rejected": -26.266277313232422, + "loss": 0.635, + "rewards/accuracies": 0.668749988079071, + "rewards/chosen": -0.3075779676437378, + "rewards/margins": 0.23826391994953156, + "rewards/rejected": -0.5458418726921082, + "step": 975 + }, + { + "epoch": 0.49, + "learning_rate": 1.2179748700879012e-10, + "logits/chosen": -1.476564645767212, + "logits/rejected": -1.5101211071014404, + "logps/chosen": -17.65967559814453, + "logps/rejected": -22.493526458740234, + "loss": 0.6488, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.29659706354141235, + "rewards/margins": 0.1625634878873825, + "rewards/rejected": -0.45916056632995605, + "step": 980 + }, + { + "epoch": 0.49, + "learning_rate": 6.852326227130833e-11, + "logits/chosen": -1.3950709104537964, + "logits/rejected": -1.396350622177124, + "logps/chosen": -17.96800994873047, + "logps/rejected": -23.398609161376953, + "loss": 0.6321, + "rewards/accuracies": 0.6187499761581421, + "rewards/chosen": -0.26527708768844604, + "rewards/margins": 0.17985311150550842, + "rewards/rejected": -0.44513019919395447, + "step": 985 + }, + { + "epoch": 0.49, + "learning_rate": 3.0458649045211895e-11, + "logits/chosen": -1.3564696311950684, + "logits/rejected": -1.343924880027771, + "logps/chosen": -18.672687530517578, + "logps/rejected": -25.386516571044922, + "loss": 0.633, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.2513435184955597, + "rewards/margins": 0.25813814997673035, + "rewards/rejected": -0.50948166847229, + "step": 990 + }, + { + "epoch": 0.5, + "learning_rate": 7.615242180436521e-12, + "logits/chosen": -1.428175926208496, + "logits/rejected": -1.4705220460891724, + "logps/chosen": -18.455589294433594, + "logps/rejected": -23.739002227783203, + "loss": 0.6388, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.34945839643478394, + "rewards/margins": 0.16214387118816376, + "rewards/rejected": -0.5116022825241089, + "step": 995 + }, + { + "epoch": 0.5, + "learning_rate": 0.0, + "logits/chosen": -1.3395625352859497, + "logits/rejected": -1.3556944131851196, + "logps/chosen": -17.86135482788086, + "logps/rejected": -21.11099624633789, + "loss": 0.6448, + "rewards/accuracies": 0.65625, + "rewards/chosen": -0.3157894015312195, + "rewards/margins": 0.17568814754486084, + "rewards/rejected": -0.4914775788784027, + "step": 1000 + }, + { + "epoch": 0.5, + "eval_logits/chosen": -1.7563655376434326, + "eval_logits/rejected": -1.8358914852142334, + "eval_logps/chosen": -17.932043075561523, + "eval_logps/rejected": -23.265531539916992, + "eval_loss": 0.6428677439689636, + "eval_rewards/accuracies": 0.6453673839569092, + "eval_rewards/chosen": -0.2918897271156311, + "eval_rewards/margins": 0.18083742260932922, + "eval_rewards/rejected": -0.47272711992263794, + "eval_runtime": 307.0181, + "eval_samples_per_second": 65.143, + "eval_steps_per_second": 1.019, + "step": 1000 + } + ], + "logging_steps": 5, + "max_steps": 1000, + "num_input_tokens_seen": 0, + "num_train_epochs": 1, + "save_steps": 100, + "total_flos": 0.0, + "train_batch_size": 2, + "trial_name": null, + "trial_params": null +}