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.6881732940673828, + "best_model_checkpoint": "/mnt/data/shesj/Trained/RL4CoT/DPO/Parallel_13B_numglueCorrect_extend_10lang_v3_iter2.json/checkpoint-1000", + "epoch": 0.2508938091952581, + "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": 1e-08, + "logits/chosen": -1.5961366891860962, + "logits/rejected": -1.4505422115325928, + "logps/chosen": -5.596881866455078, + "logps/rejected": -9.411199569702148, + "loss": 0.6934, + "rewards/accuracies": 0.26249998807907104, + "rewards/chosen": -0.0024300559889525175, + "rewards/margins": -0.003987783100455999, + "rewards/rejected": 0.00155772699508816, + "step": 5 + }, + { + "epoch": 0.0, + "learning_rate": 2e-08, + "logits/chosen": -1.7575080394744873, + "logits/rejected": -1.5709590911865234, + "logps/chosen": -5.7118635177612305, + "logps/rejected": -8.229207992553711, + "loss": 0.6932, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": 0.005083759315311909, + "rewards/margins": 0.007960619404911995, + "rewards/rejected": -0.0028768605552613735, + "step": 10 + }, + { + "epoch": 0.0, + "learning_rate": 3e-08, + "logits/chosen": -1.3776355981826782, + "logits/rejected": -1.1297904253005981, + "logps/chosen": -5.6487884521484375, + "logps/rejected": -6.3460259437561035, + "loss": 0.6928, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.0028433147817850113, + "rewards/margins": 0.004021753557026386, + "rewards/rejected": -0.0011784390080720186, + "step": 15 + }, + { + "epoch": 0.01, + "learning_rate": 4e-08, + "logits/chosen": -1.3416035175323486, + "logits/rejected": -1.4808504581451416, + "logps/chosen": -6.05244255065918, + "logps/rejected": -6.185896873474121, + "loss": 0.6939, + "rewards/accuracies": 0.44999998807907104, + "rewards/chosen": -0.002370675327256322, + "rewards/margins": -0.003694503800943494, + "rewards/rejected": 0.0013238281244412065, + "step": 20 + }, + { + "epoch": 0.01, + "learning_rate": 5e-08, + "logits/chosen": -1.5403258800506592, + "logits/rejected": -1.6286535263061523, + "logps/chosen": -5.387463569641113, + "logps/rejected": -6.760479927062988, + "loss": 0.6931, + "rewards/accuracies": 0.42500001192092896, + "rewards/chosen": -0.002958612982183695, + "rewards/margins": -0.0027186197694391012, + "rewards/rejected": -0.0002399933582637459, + "step": 25 + }, + { + "epoch": 0.01, + "learning_rate": 6e-08, + "logits/chosen": -1.3823367357254028, + "logits/rejected": -1.3627607822418213, + "logps/chosen": -5.9833083152771, + "logps/rejected": -5.850650787353516, + "loss": 0.6936, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": 0.002619728446006775, + "rewards/margins": 0.004939082078635693, + "rewards/rejected": -0.002319354098290205, + "step": 30 + }, + { + "epoch": 0.01, + "learning_rate": 6.999999999999999e-08, + "logits/chosen": -1.4706413745880127, + "logits/rejected": -1.4250898361206055, + "logps/chosen": -5.871586799621582, + "logps/rejected": -7.329322814941406, + "loss": 0.6927, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.0053069936111569405, + "rewards/margins": 0.006646360270678997, + "rewards/rejected": -0.001339366426691413, + "step": 35 + }, + { + "epoch": 0.01, + "learning_rate": 8e-08, + "logits/chosen": -1.3677151203155518, + "logits/rejected": -1.4711055755615234, + "logps/chosen": -5.220858573913574, + "logps/rejected": -6.019894599914551, + "loss": 0.6922, + "rewards/accuracies": 0.48750001192092896, + "rewards/chosen": 0.0007630128529854119, + "rewards/margins": 0.0009836136596277356, + "rewards/rejected": -0.0002206008939538151, + "step": 40 + }, + { + "epoch": 0.01, + "learning_rate": 9e-08, + "logits/chosen": -1.5524793863296509, + "logits/rejected": -1.5295554399490356, + "logps/chosen": -5.439484596252441, + "logps/rejected": -6.122335433959961, + "loss": 0.6937, + "rewards/accuracies": 0.4625000059604645, + "rewards/chosen": -0.0029087450820952654, + "rewards/margins": -0.0008771896245889366, + "rewards/rejected": -0.002031555399298668, + "step": 45 + }, + { + "epoch": 0.01, + "learning_rate": 1e-07, + "logits/chosen": -1.5465644598007202, + "logits/rejected": -1.5324052572250366, + "logps/chosen": -5.974602699279785, + "logps/rejected": -6.071439743041992, + "loss": 0.6932, + "rewards/accuracies": 0.4749999940395355, + "rewards/chosen": 0.00048004291602410376, + "rewards/margins": -0.0023091284092515707, + "rewards/rejected": 0.0027891716454178095, + "step": 50 + }, + { + "epoch": 0.01, + "learning_rate": 1.1e-07, + "logits/chosen": -1.4663435220718384, + "logits/rejected": -1.4614862203598022, + "logps/chosen": -5.4424333572387695, + "logps/rejected": -6.563809394836426, + "loss": 0.6937, + "rewards/accuracies": 0.512499988079071, + "rewards/chosen": 0.0017671022797003388, + "rewards/margins": 0.00013901680358685553, + "rewards/rejected": 0.0016280856216326356, + "step": 55 + }, + { + "epoch": 0.02, + "learning_rate": 1.2e-07, + "logits/chosen": -1.3572139739990234, + "logits/rejected": -1.5018689632415771, + "logps/chosen": -5.387210845947266, + "logps/rejected": -5.628620147705078, + "loss": 0.6934, + "rewards/accuracies": 0.5, + "rewards/chosen": -0.0009455516701564193, + "rewards/margins": -0.0012193130096420646, + "rewards/rejected": 0.0002737622708082199, + "step": 60 + }, + { + "epoch": 0.02, + "learning_rate": 1.3e-07, + "logits/chosen": -1.4500153064727783, + "logits/rejected": -1.5059070587158203, + "logps/chosen": -5.2743024826049805, + "logps/rejected": -5.707627773284912, + "loss": 0.6932, + "rewards/accuracies": 0.44999998807907104, + "rewards/chosen": -8.441717363893986e-05, + "rewards/margins": -0.0039388397708535194, + "rewards/rejected": 0.0038544225972145796, + "step": 65 + }, + { + "epoch": 0.02, + "learning_rate": 1.3999999999999998e-07, + "logits/chosen": -1.400269627571106, + "logits/rejected": -1.4793713092803955, + "logps/chosen": -5.212708473205566, + "logps/rejected": -7.475738525390625, + "loss": 0.6938, + "rewards/accuracies": 0.42500001192092896, + "rewards/chosen": -0.0009079872397705913, + "rewards/margins": -0.004652161151170731, + "rewards/rejected": 0.0037441744934767485, + "step": 70 + }, + { + "epoch": 0.02, + "learning_rate": 1.5e-07, + "logits/chosen": -1.6910864114761353, + "logits/rejected": -1.6056768894195557, + "logps/chosen": -5.710541725158691, + "logps/rejected": -6.734696388244629, + "loss": 0.6925, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -0.0002835780323948711, + "rewards/margins": 0.001549507724121213, + "rewards/rejected": -0.0018330859020352364, + "step": 75 + }, + { + "epoch": 0.02, + "learning_rate": 1.6e-07, + "logits/chosen": -1.348565936088562, + "logits/rejected": -1.3284913301467896, + "logps/chosen": -5.4736833572387695, + "logps/rejected": -6.32749080657959, + "loss": 0.6916, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.004232033155858517, + "rewards/margins": 0.008012665435671806, + "rewards/rejected": -0.003780632745474577, + "step": 80 + }, + { + "epoch": 0.02, + "learning_rate": 1.7e-07, + "logits/chosen": -1.4623692035675049, + "logits/rejected": -1.3986611366271973, + "logps/chosen": -5.681046485900879, + "logps/rejected": -6.193971157073975, + "loss": 0.6923, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": -0.003290227148681879, + "rewards/margins": 0.00016479431360494345, + "rewards/rejected": -0.003455021884292364, + "step": 85 + }, + { + "epoch": 0.02, + "learning_rate": 1.8e-07, + "logits/chosen": -1.6442101001739502, + "logits/rejected": -1.7291923761367798, + "logps/chosen": -6.625936985015869, + "logps/rejected": -8.430340766906738, + "loss": 0.6935, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.004968828056007624, + "rewards/margins": 0.004905154462903738, + "rewards/rejected": 6.367354944813997e-05, + "step": 90 + }, + { + "epoch": 0.02, + "learning_rate": 1.8999999999999998e-07, + "logits/chosen": -1.3575623035430908, + "logits/rejected": -1.3733503818511963, + "logps/chosen": -5.06931209564209, + "logps/rejected": -5.564260005950928, + "loss": 0.6925, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.0030343024991452694, + "rewards/margins": 0.004197937436401844, + "rewards/rejected": -0.0011636342387646437, + "step": 95 + }, + { + "epoch": 0.03, + "learning_rate": 2e-07, + "logits/chosen": -1.4199776649475098, + "logits/rejected": -1.3951759338378906, + "logps/chosen": -4.820796012878418, + "logps/rejected": -5.737803936004639, + "loss": 0.6933, + "rewards/accuracies": 0.4625000059604645, + "rewards/chosen": -0.0034723032731562853, + "rewards/margins": -0.003412929829210043, + "rewards/rejected": -5.9372840041760355e-05, + "step": 100 + }, + { + "epoch": 0.03, + "eval_logits/chosen": -3.167680501937866, + "eval_logits/rejected": -3.1470484733581543, + "eval_logps/chosen": -6.138812065124512, + "eval_logps/rejected": -6.94625997543335, + "eval_loss": 0.6930756568908691, + "eval_rewards/accuracies": 0.5112179517745972, + "eval_rewards/chosen": 0.00018124215421266854, + "eval_rewards/margins": 0.00026574215735308826, + "eval_rewards/rejected": -8.450019231531769e-05, + "eval_runtime": 615.2356, + "eval_samples_per_second": 32.401, + "eval_steps_per_second": 0.507, + "step": 100 + }, + { + "epoch": 0.03, + "learning_rate": 1.9999658256641745e-07, + "logits/chosen": -1.5726451873779297, + "logits/rejected": -1.4517858028411865, + "logps/chosen": -6.400355339050293, + "logps/rejected": -8.746145248413086, + "loss": 0.6923, + "rewards/accuracies": 0.48750001192092896, + "rewards/chosen": 0.0017850773874670267, + "rewards/margins": 0.003671336220577359, + "rewards/rejected": -0.001886258483864367, + "step": 105 + }, + { + "epoch": 0.03, + "learning_rate": 1.999863304992469e-07, + "logits/chosen": -1.4806731939315796, + "logits/rejected": -1.4488584995269775, + "logps/chosen": -4.718806266784668, + "logps/rejected": -6.621995449066162, + "loss": 0.6931, + "rewards/accuracies": 0.512499988079071, + "rewards/chosen": -0.003079561050981283, + "rewards/margins": -0.0010764991166070104, + "rewards/rejected": -0.002003061817958951, + "step": 110 + }, + { + "epoch": 0.03, + "learning_rate": 1.9996924449920347e-07, + "logits/chosen": -1.3951839208602905, + "logits/rejected": -1.5014991760253906, + "logps/chosen": -5.849172115325928, + "logps/rejected": -6.110888957977295, + "loss": 0.6916, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": -0.002346306573599577, + "rewards/margins": -0.0002563331217970699, + "rewards/rejected": -0.0020899735391139984, + "step": 115 + }, + { + "epoch": 0.03, + "learning_rate": 1.999453257340926e-07, + "logits/chosen": -1.709111213684082, + "logits/rejected": -1.6502996683120728, + "logps/chosen": -5.375964164733887, + "logps/rejected": -6.7974419593811035, + "loss": 0.6908, + "rewards/accuracies": 0.5, + "rewards/chosen": 0.0006786620942875743, + "rewards/margins": 0.002161815296858549, + "rewards/rejected": -0.0014831533189862967, + "step": 120 + }, + { + "epoch": 0.03, + "learning_rate": 1.9991457583873009e-07, + "logits/chosen": -1.6699622869491577, + "logits/rejected": -1.6052825450897217, + "logps/chosen": -5.923104286193848, + "logps/rejected": -6.845878601074219, + "loss": 0.6904, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": 0.002157662995159626, + "rewards/margins": 0.0037304076831787825, + "rewards/rejected": -0.0015727445716038346, + "step": 125 + }, + { + "epoch": 0.03, + "learning_rate": 1.9987699691483047e-07, + "logits/chosen": -1.5257129669189453, + "logits/rejected": -1.5561866760253906, + "logps/chosen": -4.64754581451416, + "logps/rejected": -6.103314399719238, + "loss": 0.69, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.00055070681264624, + "rewards/margins": 0.008159702643752098, + "rewards/rejected": -0.0076089962385594845, + "step": 130 + }, + { + "epoch": 0.03, + "learning_rate": 1.9983259153086325e-07, + "logits/chosen": -1.5833688974380493, + "logits/rejected": -1.5711325407028198, + "logps/chosen": -5.653990745544434, + "logps/rejected": -7.226287841796875, + "loss": 0.6918, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.0021617007441818714, + "rewards/margins": 0.005132616497576237, + "rewards/rejected": -0.007294316776096821, + "step": 135 + }, + { + "epoch": 0.04, + "learning_rate": 1.9978136272187745e-07, + "logits/chosen": -1.3831666707992554, + "logits/rejected": -1.3798400163650513, + "logps/chosen": -4.3581929206848145, + "logps/rejected": -5.750641345977783, + "loss": 0.6905, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.0044810837134718895, + "rewards/margins": 0.002624013228341937, + "rewards/rejected": 0.0018570702522993088, + "step": 140 + }, + { + "epoch": 0.04, + "learning_rate": 1.997233139892941e-07, + "logits/chosen": -1.4784475564956665, + "logits/rejected": -1.453616976737976, + "logps/chosen": -5.93430757522583, + "logps/rejected": -6.510165214538574, + "loss": 0.6904, + "rewards/accuracies": 0.5625, + "rewards/chosen": 0.0023500092793256044, + "rewards/margins": 0.007515914738178253, + "rewards/rejected": -0.005165906623005867, + "step": 145 + }, + { + "epoch": 0.04, + "learning_rate": 1.9965844930066698e-07, + "logits/chosen": -1.4583405256271362, + "logits/rejected": -1.611533761024475, + "logps/chosen": -5.597588539123535, + "logps/rejected": -7.059754848480225, + "loss": 0.6884, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.0004460577911231667, + "rewards/margins": 0.008472367189824581, + "rewards/rejected": -0.00891842506825924, + "step": 150 + }, + { + "epoch": 0.04, + "learning_rate": 1.9958677308941136e-07, + "logits/chosen": -1.590573787689209, + "logits/rejected": -1.5703933238983154, + "logps/chosen": -8.305377006530762, + "logps/rejected": -6.981376647949219, + "loss": 0.6898, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": 0.0003787805908359587, + "rewards/margins": 0.006806156598031521, + "rewards/rejected": -0.006427376065403223, + "step": 155 + }, + { + "epoch": 0.04, + "learning_rate": 1.9950829025450114e-07, + "logits/chosen": -1.6168692111968994, + "logits/rejected": -1.3941491842269897, + "logps/chosen": -4.724443435668945, + "logps/rejected": -7.058165073394775, + "loss": 0.6897, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.002109301509335637, + "rewards/margins": 0.008472011424601078, + "rewards/rejected": -0.00636270921677351, + "step": 160 + }, + { + "epoch": 0.04, + "learning_rate": 1.9942300616013377e-07, + "logits/chosen": -1.7095611095428467, + "logits/rejected": -1.7674989700317383, + "logps/chosen": -5.134617328643799, + "logps/rejected": -5.987712383270264, + "loss": 0.6906, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -0.0022057683672755957, + "rewards/margins": 0.0033194683492183685, + "rewards/rejected": -0.0055252364836633205, + "step": 165 + }, + { + "epoch": 0.04, + "learning_rate": 1.993309266353638e-07, + "logits/chosen": -1.3613307476043701, + "logits/rejected": -1.4475274085998535, + "logps/chosen": -6.0252180099487305, + "logps/rejected": -6.769097328186035, + "loss": 0.6906, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.00019303560839034617, + "rewards/margins": 0.0034273392520844936, + "rewards/rejected": -0.0036203742492944, + "step": 170 + }, + { + "epoch": 0.04, + "learning_rate": 1.992320579737045e-07, + "logits/chosen": -1.2607237100601196, + "logits/rejected": -1.2645083665847778, + "logps/chosen": -5.982339859008789, + "logps/rejected": -5.792677879333496, + "loss": 0.687, + "rewards/accuracies": 0.6875, + "rewards/chosen": 0.006508027669042349, + "rewards/margins": 0.01678382232785225, + "rewards/rejected": -0.010275794193148613, + "step": 175 + }, + { + "epoch": 0.05, + "learning_rate": 1.9912640693269751e-07, + "logits/chosen": -1.4913660287857056, + "logits/rejected": -1.467761516571045, + "logps/chosen": -4.889002799987793, + "logps/rejected": -6.567469120025635, + "loss": 0.6895, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": 0.007032574620097876, + "rewards/margins": 0.015939272940158844, + "rewards/rejected": -0.008906697854399681, + "step": 180 + }, + { + "epoch": 0.05, + "learning_rate": 1.9901398073345117e-07, + "logits/chosen": -1.6264896392822266, + "logits/rejected": -1.787503957748413, + "logps/chosen": -5.5176873207092285, + "logps/rejected": -7.238592624664307, + "loss": 0.6888, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": 0.0024650241248309612, + "rewards/margins": 0.014559340663254261, + "rewards/rejected": -0.012094316072762012, + "step": 185 + }, + { + "epoch": 0.05, + "learning_rate": 1.9889478706014683e-07, + "logits/chosen": -1.5220094919204712, + "logits/rejected": -1.3731660842895508, + "logps/chosen": -4.48035192489624, + "logps/rejected": -5.973559856414795, + "loss": 0.6892, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.002238192595541477, + "rewards/margins": 0.0097486088052392, + "rewards/rejected": -0.007510416209697723, + "step": 190 + }, + { + "epoch": 0.05, + "learning_rate": 1.9876883405951376e-07, + "logits/chosen": -1.3608970642089844, + "logits/rejected": -1.2253252267837524, + "logps/chosen": -6.359659194946289, + "logps/rejected": -6.143754005432129, + "loss": 0.6891, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.0014768632827326655, + "rewards/margins": 0.004477277398109436, + "rewards/rejected": -0.0030004139989614487, + "step": 195 + }, + { + "epoch": 0.05, + "learning_rate": 1.9863613034027222e-07, + "logits/chosen": -1.3653035163879395, + "logits/rejected": -1.3042250871658325, + "logps/chosen": -5.9716901779174805, + "logps/rejected": -6.985511779785156, + "loss": 0.6853, + "rewards/accuracies": 0.5625, + "rewards/chosen": 0.003856885712593794, + "rewards/margins": 0.011315730400383472, + "rewards/rejected": -0.0074588460847735405, + "step": 200 + }, + { + "epoch": 0.05, + "eval_logits/chosen": -3.174004077911377, + "eval_logits/rejected": -3.153486490249634, + "eval_logps/chosen": -6.161444664001465, + "eval_logps/rejected": -6.977956771850586, + "eval_loss": 0.6930607557296753, + "eval_rewards/accuracies": 0.5140224099159241, + "eval_rewards/chosen": -0.0020819876808673143, + "eval_rewards/margins": 0.0011722741182893515, + "eval_rewards/rejected": -0.0032542620319873095, + "eval_runtime": 618.1553, + "eval_samples_per_second": 32.248, + "eval_steps_per_second": 0.505, + "step": 200 + }, + { + "epoch": 0.05, + "learning_rate": 1.9849668497254518e-07, + "logits/chosen": -1.366420030593872, + "logits/rejected": -1.418501377105713, + "logps/chosen": -4.841452121734619, + "logps/rejected": -7.0436296463012695, + "loss": 0.6882, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.0067129782401025295, + "rewards/margins": 0.013699628412723541, + "rewards/rejected": -0.006986652500927448, + "step": 205 + }, + { + "epoch": 0.05, + "learning_rate": 1.9835050748723822e-07, + "logits/chosen": -1.5236493349075317, + "logits/rejected": -1.4237031936645508, + "logps/chosen": -5.028156280517578, + "logps/rejected": -6.966952323913574, + "loss": 0.6862, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.004889342468231916, + "rewards/margins": 0.01445357222110033, + "rewards/rejected": -0.009564228355884552, + "step": 210 + }, + { + "epoch": 0.05, + "learning_rate": 1.9819760787538837e-07, + "logits/chosen": -1.7881724834442139, + "logits/rejected": -1.702300786972046, + "logps/chosen": -5.008362293243408, + "logps/rejected": -7.346070766448975, + "loss": 0.6862, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.003184701083227992, + "rewards/margins": 0.009327715262770653, + "rewards/rejected": -0.012512415647506714, + "step": 215 + }, + { + "epoch": 0.06, + "learning_rate": 1.9803799658748093e-07, + "logits/chosen": -1.7177143096923828, + "logits/rejected": -1.579524278640747, + "logps/chosen": -5.551783561706543, + "logps/rejected": -8.588147163391113, + "loss": 0.6859, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.00936056487262249, + "rewards/margins": -0.0009745029965415597, + "rewards/rejected": -0.00838606245815754, + "step": 220 + }, + { + "epoch": 0.06, + "learning_rate": 1.9787168453273545e-07, + "logits/chosen": -1.4372203350067139, + "logits/rejected": -1.4148085117340088, + "logps/chosen": -5.2672553062438965, + "logps/rejected": -5.7078680992126465, + "loss": 0.685, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.004485909827053547, + "rewards/margins": 0.01839793473482132, + "rewards/rejected": -0.013912022113800049, + "step": 225 + }, + { + "epoch": 0.06, + "learning_rate": 1.9769868307835993e-07, + "logits/chosen": -1.6145379543304443, + "logits/rejected": -1.536407709121704, + "logps/chosen": -5.295413970947266, + "logps/rejected": -7.089319705963135, + "loss": 0.6852, + "rewards/accuracies": 0.550000011920929, + "rewards/chosen": 0.00589085603132844, + "rewards/margins": 0.01651725545525551, + "rewards/rejected": -0.010626398026943207, + "step": 230 + }, + { + "epoch": 0.06, + "learning_rate": 1.9751900404877398e-07, + "logits/chosen": -1.5907440185546875, + "logits/rejected": -1.5271718502044678, + "logps/chosen": -4.938467979431152, + "logps/rejected": -5.892498970031738, + "loss": 0.6846, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.0029959643725305796, + "rewards/margins": 0.021143438294529915, + "rewards/rejected": -0.01814747229218483, + "step": 235 + }, + { + "epoch": 0.06, + "learning_rate": 1.9733265972480058e-07, + "logits/chosen": -1.6651685237884521, + "logits/rejected": -1.5356318950653076, + "logps/chosen": -7.418261528015137, + "logps/rejected": -8.457880973815918, + "loss": 0.6852, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.0033951636869460344, + "rewards/margins": 0.026372741907835007, + "rewards/rejected": -0.02297757938504219, + "step": 240 + }, + { + "epoch": 0.06, + "learning_rate": 1.9713966284282674e-07, + "logits/chosen": -1.4138095378875732, + "logits/rejected": -1.3163411617279053, + "logps/chosen": -6.462838172912598, + "logps/rejected": -7.8138837814331055, + "loss": 0.6839, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.007619897834956646, + "rewards/margins": 0.0050119319930672646, + "rewards/rejected": -0.012631828896701336, + "step": 245 + }, + { + "epoch": 0.06, + "learning_rate": 1.9694002659393302e-07, + "logits/chosen": -1.5196382999420166, + "logits/rejected": -1.477994680404663, + "logps/chosen": -5.463156700134277, + "logps/rejected": -6.823214530944824, + "loss": 0.6831, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.0001128343865275383, + "rewards/margins": 0.022026551887392998, + "rewards/rejected": -0.02213938534259796, + "step": 250 + }, + { + "epoch": 0.06, + "learning_rate": 1.9673376462299182e-07, + "logits/chosen": -1.522629976272583, + "logits/rejected": -1.4759032726287842, + "logps/chosen": -6.2718963623046875, + "logps/rejected": -7.001539707183838, + "loss": 0.6825, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.00030401311232708395, + "rewards/margins": 0.021813327446579933, + "rewards/rejected": -0.021509312093257904, + "step": 255 + }, + { + "epoch": 0.07, + "learning_rate": 1.9652089102773487e-07, + "logits/chosen": -1.815498948097229, + "logits/rejected": -1.9041208028793335, + "logps/chosen": -4.9399566650390625, + "logps/rejected": -6.410666465759277, + "loss": 0.6844, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.0037350181955844164, + "rewards/margins": 0.023170799016952515, + "rewards/rejected": -0.01943577639758587, + "step": 260 + }, + { + "epoch": 0.07, + "learning_rate": 1.963014203577896e-07, + "logits/chosen": -1.5250468254089355, + "logits/rejected": -1.5195215940475464, + "logps/chosen": -4.857203483581543, + "logps/rejected": -6.858325958251953, + "loss": 0.6829, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.007676561363041401, + "rewards/margins": 0.008902650326490402, + "rewards/rejected": -0.016579212620854378, + "step": 265 + }, + { + "epoch": 0.07, + "learning_rate": 1.9607536761368482e-07, + "logits/chosen": -1.398611068725586, + "logits/rejected": -1.4291143417358398, + "logps/chosen": -5.944151878356934, + "logps/rejected": -6.5233049392700195, + "loss": 0.682, + "rewards/accuracies": 0.6875, + "rewards/chosen": 0.004370951093733311, + "rewards/margins": 0.02115224488079548, + "rewards/rejected": -0.016781292855739594, + "step": 270 + }, + { + "epoch": 0.07, + "learning_rate": 1.9584274824582527e-07, + "logits/chosen": -1.3120192289352417, + "logits/rejected": -1.3574376106262207, + "logps/chosen": -4.558770179748535, + "logps/rejected": -5.6866583824157715, + "loss": 0.6821, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.004323200322687626, + "rewards/margins": 0.01929726079106331, + "rewards/rejected": -0.014974060468375683, + "step": 275 + }, + { + "epoch": 0.07, + "learning_rate": 1.9560357815343574e-07, + "logits/chosen": -1.5410289764404297, + "logits/rejected": -1.5422722101211548, + "logps/chosen": -7.481714725494385, + "logps/rejected": -6.8584442138671875, + "loss": 0.6795, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": 0.004259726498275995, + "rewards/margins": 0.026613563299179077, + "rewards/rejected": -0.02235383726656437, + "step": 280 + }, + { + "epoch": 0.07, + "learning_rate": 1.9535787368347442e-07, + "logits/chosen": -1.6866031885147095, + "logits/rejected": -1.573209524154663, + "logps/chosen": -6.985440731048584, + "logps/rejected": -6.92074728012085, + "loss": 0.6813, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.003224983811378479, + "rewards/margins": 0.03093186393380165, + "rewards/rejected": -0.03415685147047043, + "step": 285 + }, + { + "epoch": 0.07, + "learning_rate": 1.9510565162951537e-07, + "logits/chosen": -1.5131175518035889, + "logits/rejected": -1.5239301919937134, + "logps/chosen": -5.186364650726318, + "logps/rejected": -6.2596564292907715, + "loss": 0.6808, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.012961247935891151, + "rewards/margins": 0.05755941942334175, + "rewards/rejected": -0.04459817335009575, + "step": 290 + }, + { + "epoch": 0.07, + "learning_rate": 1.9484692923060094e-07, + "logits/chosen": -1.540378212928772, + "logits/rejected": -1.5740854740142822, + "logps/chosen": -6.078642845153809, + "logps/rejected": -6.573060512542725, + "loss": 0.6809, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.011097406968474388, + "rewards/margins": 0.030860627070069313, + "rewards/rejected": -0.041958026587963104, + "step": 295 + }, + { + "epoch": 0.08, + "learning_rate": 1.9458172417006346e-07, + "logits/chosen": -1.5138806104660034, + "logits/rejected": -1.3892674446105957, + "logps/chosen": -5.267967224121094, + "logps/rejected": -6.3208441734313965, + "loss": 0.6801, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.0035496647469699383, + "rewards/margins": 0.038710370659828186, + "rewards/rejected": -0.03516070172190666, + "step": 300 + }, + { + "epoch": 0.08, + "eval_logits/chosen": -3.184242010116577, + "eval_logits/rejected": -3.1636037826538086, + "eval_logps/chosen": -6.222862720489502, + "eval_logps/rejected": -7.056950569152832, + "eval_loss": 0.6925042867660522, + "eval_rewards/accuracies": 0.5252403616905212, + "eval_rewards/chosen": -0.008223854936659336, + "eval_rewards/margins": 0.0029297315049916506, + "eval_rewards/rejected": -0.011153585277497768, + "eval_runtime": 620.1964, + "eval_samples_per_second": 32.141, + "eval_steps_per_second": 0.503, + "step": 300 + }, + { + "epoch": 0.08, + "learning_rate": 1.943100545743165e-07, + "logits/chosen": -1.6512733697891235, + "logits/rejected": -1.721056342124939, + "logps/chosen": -5.9783034324646, + "logps/rejected": -5.989422798156738, + "loss": 0.6787, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.009316826239228249, + "rewards/margins": 0.029956454411149025, + "rewards/rejected": -0.039273280650377274, + "step": 305 + }, + { + "epoch": 0.08, + "learning_rate": 1.9403193901161612e-07, + "logits/chosen": -1.4372011423110962, + "logits/rejected": -1.2608239650726318, + "logps/chosen": -5.216915130615234, + "logps/rejected": -7.368325233459473, + "loss": 0.6783, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.00843791477382183, + "rewards/margins": 0.019914906471967697, + "rewards/rejected": -0.028352823108434677, + "step": 310 + }, + { + "epoch": 0.08, + "learning_rate": 1.9374739649079154e-07, + "logits/chosen": -1.4851138591766357, + "logits/rejected": -1.5565675497055054, + "logps/chosen": -6.530230522155762, + "logps/rejected": -5.977784156799316, + "loss": 0.6795, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.005143959075212479, + "rewards/margins": 0.03382311016321182, + "rewards/rejected": -0.028679147362709045, + "step": 315 + }, + { + "epoch": 0.08, + "learning_rate": 1.9345644645994608e-07, + "logits/chosen": -1.7324178218841553, + "logits/rejected": -1.6093097925186157, + "logps/chosen": -5.668793201446533, + "logps/rejected": -8.58001708984375, + "loss": 0.6813, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.007929663173854351, + "rewards/margins": 0.016542982310056686, + "rewards/rejected": -0.02447264827787876, + "step": 320 + }, + { + "epoch": 0.08, + "learning_rate": 1.9315910880512788e-07, + "logits/chosen": -1.664690613746643, + "logits/rejected": -1.6959164142608643, + "logps/chosen": -4.798338413238525, + "logps/rejected": -6.650571346282959, + "loss": 0.6793, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.002946319757029414, + "rewards/margins": 0.03725341707468033, + "rewards/rejected": -0.04019974544644356, + "step": 325 + }, + { + "epoch": 0.08, + "learning_rate": 1.928554038489707e-07, + "logits/chosen": -1.6366933584213257, + "logits/rejected": -1.6312482357025146, + "logps/chosen": -5.757204532623291, + "logps/rejected": -6.9877214431762695, + "loss": 0.6771, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.004410495515912771, + "rewards/margins": 0.03300872817635536, + "rewards/rejected": -0.03741922602057457, + "step": 330 + }, + { + "epoch": 0.08, + "learning_rate": 1.9254535234930483e-07, + "logits/chosen": -1.383345365524292, + "logits/rejected": -1.4103121757507324, + "logps/chosen": -7.0129594802856445, + "logps/rejected": -6.204277038574219, + "loss": 0.6795, + "rewards/accuracies": 0.5625, + "rewards/chosen": -0.0018229834968224168, + "rewards/margins": 0.018833067268133163, + "rewards/rejected": -0.020656052976846695, + "step": 335 + }, + { + "epoch": 0.09, + "learning_rate": 1.9222897549773846e-07, + "logits/chosen": -1.3323512077331543, + "logits/rejected": -1.3201320171356201, + "logps/chosen": -5.118114471435547, + "logps/rejected": -7.212060451507568, + "loss": 0.6776, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.009920697659254074, + "rewards/margins": 0.029569601640105247, + "rewards/rejected": -0.03949030116200447, + "step": 340 + }, + { + "epoch": 0.09, + "learning_rate": 1.9190629491820908e-07, + "logits/chosen": -1.4638566970825195, + "logits/rejected": -1.4451267719268799, + "logps/chosen": -5.2605671882629395, + "logps/rejected": -6.085940837860107, + "loss": 0.6739, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.005447469651699066, + "rewards/margins": 0.034440845251083374, + "rewards/rejected": -0.03988831490278244, + "step": 345 + }, + { + "epoch": 0.09, + "learning_rate": 1.9157733266550572e-07, + "logits/chosen": -1.5962337255477905, + "logits/rejected": -1.6220163106918335, + "logps/chosen": -6.018959045410156, + "logps/rejected": -6.889263153076172, + "loss": 0.674, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.0057447971776127815, + "rewards/margins": 0.07764319330453873, + "rewards/rejected": -0.08338797837495804, + "step": 350 + }, + { + "epoch": 0.09, + "learning_rate": 1.9124211122376135e-07, + "logits/chosen": -1.4028089046478271, + "logits/rejected": -1.3758150339126587, + "logps/chosen": -5.279055595397949, + "logps/rejected": -8.15546989440918, + "loss": 0.6746, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.004057350568473339, + "rewards/margins": 0.041755061596632004, + "rewards/rejected": -0.03769771382212639, + "step": 355 + }, + { + "epoch": 0.09, + "learning_rate": 1.9090065350491624e-07, + "logits/chosen": -1.466732144355774, + "logits/rejected": -1.505338430404663, + "logps/chosen": -6.279904365539551, + "logps/rejected": -6.962622165679932, + "loss": 0.6722, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.0070061227306723595, + "rewards/margins": 0.04479537159204483, + "rewards/rejected": -0.037789251655340195, + "step": 360 + }, + { + "epoch": 0.09, + "learning_rate": 1.905529828471519e-07, + "logits/chosen": -1.5488044023513794, + "logits/rejected": -1.4669849872589111, + "logps/chosen": -6.46566104888916, + "logps/rejected": -6.615389823913574, + "loss": 0.6752, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.009375613182783127, + "rewards/margins": 0.0315098911523819, + "rewards/rejected": -0.04088550806045532, + "step": 365 + }, + { + "epoch": 0.09, + "learning_rate": 1.901991230132959e-07, + "logits/chosen": -1.5748860836029053, + "logits/rejected": -1.5498300790786743, + "logps/chosen": -5.943437099456787, + "logps/rejected": -7.429045677185059, + "loss": 0.6752, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.012559416703879833, + "rewards/margins": 0.03320794180035591, + "rewards/rejected": -0.04576735943555832, + "step": 370 + }, + { + "epoch": 0.09, + "learning_rate": 1.8983909818919788e-07, + "logits/chosen": -1.4835641384124756, + "logits/rejected": -1.4734997749328613, + "logps/chosen": -5.776877403259277, + "logps/rejected": -6.821134090423584, + "loss": 0.6746, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 8.074291690718383e-05, + "rewards/margins": 0.041906945407390594, + "rewards/rejected": -0.04182619974017143, + "step": 375 + }, + { + "epoch": 0.1, + "learning_rate": 1.8947293298207635e-07, + "logits/chosen": -1.590511441230774, + "logits/rejected": -1.5636823177337646, + "logps/chosen": -6.252840042114258, + "logps/rejected": -7.568482398986816, + "loss": 0.6773, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.009611086919903755, + "rewards/margins": 0.04731304943561554, + "rewards/rejected": -0.056924134492874146, + "step": 380 + }, + { + "epoch": 0.1, + "learning_rate": 1.8910065241883678e-07, + "logits/chosen": -1.7463115453720093, + "logits/rejected": -1.552099347114563, + "logps/chosen": -5.755384922027588, + "logps/rejected": -9.108583450317383, + "loss": 0.6762, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.014299096539616585, + "rewards/margins": 0.04487228766083717, + "rewards/rejected": -0.05917138606309891, + "step": 385 + }, + { + "epoch": 0.1, + "learning_rate": 1.8872228194436116e-07, + "logits/chosen": -1.4452903270721436, + "logits/rejected": -1.4187732934951782, + "logps/chosen": -5.313692092895508, + "logps/rejected": -6.416788578033447, + "loss": 0.6713, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.017885476350784302, + "rewards/margins": 0.03439543396234512, + "rewards/rejected": -0.052280914038419724, + "step": 390 + }, + { + "epoch": 0.1, + "learning_rate": 1.8833784741976886e-07, + "logits/chosen": -1.5387766361236572, + "logits/rejected": -1.5287269353866577, + "logps/chosen": -5.880291938781738, + "logps/rejected": -6.970144748687744, + "loss": 0.6682, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.018293332308530807, + "rewards/margins": 0.044221844524145126, + "rewards/rejected": -0.06251517683267593, + "step": 395 + }, + { + "epoch": 0.1, + "learning_rate": 1.8794737512064888e-07, + "logits/chosen": -1.640610933303833, + "logits/rejected": -1.539394736289978, + "logps/chosen": -5.029040336608887, + "logps/rejected": -6.50949764251709, + "loss": 0.6676, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.008673004806041718, + "rewards/margins": 0.04148182272911072, + "rewards/rejected": -0.050154827535152435, + "step": 400 + }, + { + "epoch": 0.1, + "eval_logits/chosen": -3.1871755123138428, + "eval_logits/rejected": -3.1664505004882812, + "eval_logps/chosen": -6.327807903289795, + "eval_logps/rejected": -7.203298091888428, + "eval_loss": 0.6918210983276367, + "eval_rewards/accuracies": 0.5324519276618958, + "eval_rewards/chosen": -0.018718333914875984, + "eval_rewards/margins": 0.007069970481097698, + "eval_rewards/rejected": -0.025788303464651108, + "eval_runtime": 624.3628, + "eval_samples_per_second": 31.927, + "eval_steps_per_second": 0.5, + "step": 400 + }, + { + "epoch": 0.1, + "learning_rate": 1.875508917352643e-07, + "logits/chosen": -1.3193857669830322, + "logits/rejected": -1.3139396905899048, + "logps/chosen": -5.116377830505371, + "logps/rejected": -7.48989200592041, + "loss": 0.6702, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.00014799665950704366, + "rewards/margins": 0.06528617441654205, + "rewards/rejected": -0.06513817608356476, + "step": 405 + }, + { + "epoch": 0.1, + "learning_rate": 1.871484243627277e-07, + "logits/chosen": -1.6524467468261719, + "logits/rejected": -1.6226232051849365, + "logps/chosen": -5.752681732177734, + "logps/rejected": -7.157814979553223, + "loss": 0.6721, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.02660415694117546, + "rewards/margins": 0.04491695761680603, + "rewards/rejected": -0.07152111828327179, + "step": 410 + }, + { + "epoch": 0.1, + "learning_rate": 1.867400005111495e-07, + "logits/chosen": -1.4393689632415771, + "logits/rejected": -1.369827151298523, + "logps/chosen": -6.169909477233887, + "logps/rejected": -7.438270568847656, + "loss": 0.6732, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.005500434432178736, + "rewards/margins": 0.06915347278118134, + "rewards/rejected": -0.07465390115976334, + "step": 415 + }, + { + "epoch": 0.11, + "learning_rate": 1.8632564809575738e-07, + "logits/chosen": -1.622018814086914, + "logits/rejected": -1.367488145828247, + "logps/chosen": -5.794252872467041, + "logps/rejected": -7.3042449951171875, + "loss": 0.6702, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.016323495656251907, + "rewards/margins": 0.07359044253826141, + "rewards/rejected": -0.057266950607299805, + "step": 420 + }, + { + "epoch": 0.11, + "learning_rate": 1.859053954369885e-07, + "logits/chosen": -1.5450842380523682, + "logits/rejected": -1.5149590969085693, + "logps/chosen": -4.754348278045654, + "logps/rejected": -6.53006649017334, + "loss": 0.6752, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.005179319530725479, + "rewards/margins": 0.04943736642599106, + "rewards/rejected": -0.05461668223142624, + "step": 425 + }, + { + "epoch": 0.11, + "learning_rate": 1.854792712585539e-07, + "logits/chosen": -1.6106449365615845, + "logits/rejected": -1.472044587135315, + "logps/chosen": -5.688014030456543, + "logps/rejected": -7.954685211181641, + "loss": 0.6644, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.01195032149553299, + "rewards/margins": 0.08017265051603317, + "rewards/rejected": -0.09212296456098557, + "step": 430 + }, + { + "epoch": 0.11, + "learning_rate": 1.8504730468547506e-07, + "logits/chosen": -1.7078673839569092, + "logits/rejected": -1.6030772924423218, + "logps/chosen": -5.776875972747803, + "logps/rejected": -8.993513107299805, + "loss": 0.6652, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.03464338928461075, + "rewards/margins": 0.04098203405737877, + "rewards/rejected": -0.07562542706727982, + "step": 435 + }, + { + "epoch": 0.11, + "learning_rate": 1.846095252420935e-07, + "logits/chosen": -1.5690264701843262, + "logits/rejected": -1.466038465499878, + "logps/chosen": -5.061371326446533, + "logps/rejected": -7.400214195251465, + "loss": 0.6721, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.0034414425026625395, + "rewards/margins": 0.06261752545833588, + "rewards/rejected": -0.06605897843837738, + "step": 440 + }, + { + "epoch": 0.11, + "learning_rate": 1.841659628500527e-07, + "logits/chosen": -1.5850152969360352, + "logits/rejected": -1.626733422279358, + "logps/chosen": -5.730135440826416, + "logps/rejected": -7.101342678070068, + "loss": 0.6687, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.0294067170470953, + "rewards/margins": 0.06425820291042328, + "rewards/rejected": -0.09366491436958313, + "step": 445 + }, + { + "epoch": 0.11, + "learning_rate": 1.8371664782625284e-07, + "logits/chosen": -1.4130651950836182, + "logits/rejected": -1.4390531778335571, + "logps/chosen": -6.015176296234131, + "logps/rejected": -6.557525634765625, + "loss": 0.6697, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.019521493464708328, + "rewards/margins": 0.048861414194107056, + "rewards/rejected": -0.06838290393352509, + "step": 450 + }, + { + "epoch": 0.11, + "learning_rate": 1.8326161088077904e-07, + "logits/chosen": -1.5105092525482178, + "logits/rejected": -1.5329506397247314, + "logps/chosen": -5.465790748596191, + "logps/rejected": -6.999060153961182, + "loss": 0.6671, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.01001508068293333, + "rewards/margins": 0.05736090615391731, + "rewards/rejected": -0.06737598031759262, + "step": 455 + }, + { + "epoch": 0.12, + "learning_rate": 1.82800883114802e-07, + "logits/chosen": -1.3845876455307007, + "logits/rejected": -1.4282658100128174, + "logps/chosen": -6.4336256980896, + "logps/rejected": -6.6461639404296875, + "loss": 0.6661, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.023983094841241837, + "rewards/margins": 0.0474400632083416, + "rewards/rejected": -0.07142315804958344, + "step": 460 + }, + { + "epoch": 0.12, + "learning_rate": 1.8233449601845256e-07, + "logits/chosen": -1.5965187549591064, + "logits/rejected": -1.6954765319824219, + "logps/chosen": -7.093475341796875, + "logps/rejected": -8.442846298217773, + "loss": 0.662, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.0007729934295639396, + "rewards/margins": 0.09539145231246948, + "rewards/rejected": -0.09616444259881973, + "step": 465 + }, + { + "epoch": 0.12, + "learning_rate": 1.8186248146866925e-07, + "logits/chosen": -1.4333058595657349, + "logits/rejected": -1.3761308193206787, + "logps/chosen": -6.583089351654053, + "logps/rejected": -8.258574485778809, + "loss": 0.6664, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.0373983308672905, + "rewards/margins": 0.028671253472566605, + "rewards/rejected": -0.0660695806145668, + "step": 470 + }, + { + "epoch": 0.12, + "learning_rate": 1.8138487172701948e-07, + "logits/chosen": -1.4833250045776367, + "logits/rejected": -1.5289559364318848, + "logps/chosen": -5.485930442810059, + "logps/rejected": -6.312931060791016, + "loss": 0.6665, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.00883159227669239, + "rewards/margins": 0.05059976503252983, + "rewards/rejected": -0.059431351721286774, + "step": 475 + }, + { + "epoch": 0.12, + "learning_rate": 1.8090169943749475e-07, + "logits/chosen": -1.3064377307891846, + "logits/rejected": -1.246734619140625, + "logps/chosen": -4.533870220184326, + "logps/rejected": -6.509539604187012, + "loss": 0.6642, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.005517001263797283, + "rewards/margins": 0.06515659391880035, + "rewards/rejected": -0.07067359238862991, + "step": 480 + }, + { + "epoch": 0.12, + "learning_rate": 1.8041299762427914e-07, + "logits/chosen": -1.605373740196228, + "logits/rejected": -1.5201643705368042, + "logps/chosen": -6.204909801483154, + "logps/rejected": -8.210671424865723, + "loss": 0.6623, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.024663742631673813, + "rewards/margins": 0.06276446580886841, + "rewards/rejected": -0.08742821216583252, + "step": 485 + }, + { + "epoch": 0.12, + "learning_rate": 1.7991879968949247e-07, + "logits/chosen": -1.2218480110168457, + "logits/rejected": -1.1649186611175537, + "logps/chosen": -4.534018516540527, + "logps/rejected": -5.6343183517456055, + "loss": 0.6628, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.007990372367203236, + "rewards/margins": 0.07446859031915665, + "rewards/rejected": -0.0824589654803276, + "step": 490 + }, + { + "epoch": 0.12, + "learning_rate": 1.794191394109071e-07, + "logits/chosen": -1.4611269235610962, + "logits/rejected": -1.4972448348999023, + "logps/chosen": -6.087195873260498, + "logps/rejected": -7.726794242858887, + "loss": 0.6659, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.02641758695244789, + "rewards/margins": 0.03801130875945091, + "rewards/rejected": -0.0644288882613182, + "step": 495 + }, + { + "epoch": 0.13, + "learning_rate": 1.7891405093963936e-07, + "logits/chosen": -1.6527025699615479, + "logits/rejected": -1.5353628396987915, + "logps/chosen": -4.67386531829834, + "logps/rejected": -6.836350917816162, + "loss": 0.6582, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.008362488821148872, + "rewards/margins": 0.07853694260120392, + "rewards/rejected": -0.08689942955970764, + "step": 500 + }, + { + "epoch": 0.13, + "eval_logits/chosen": -3.196692705154419, + "eval_logits/rejected": -3.1757330894470215, + "eval_logps/chosen": -6.479369640350342, + "eval_logps/rejected": -7.3778076171875, + "eval_loss": 0.691510021686554, + "eval_rewards/accuracies": 0.5356570482254028, + "eval_rewards/chosen": -0.03387455642223358, + "eval_rewards/margins": 0.0093647176399827, + "eval_rewards/rejected": -0.043239280581474304, + "eval_runtime": 630.1142, + "eval_samples_per_second": 31.636, + "eval_steps_per_second": 0.495, + "step": 500 + }, + { + "epoch": 0.13, + "learning_rate": 1.7840356879781529e-07, + "logits/chosen": -1.5877869129180908, + "logits/rejected": -1.5104506015777588, + "logps/chosen": -6.052631378173828, + "logps/rejected": -9.69563102722168, + "loss": 0.6607, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.04113469645380974, + "rewards/margins": 0.08254175633192062, + "rewards/rejected": -0.12367645651102066, + "step": 505 + }, + { + "epoch": 0.13, + "learning_rate": 1.7788772787621125e-07, + "logits/chosen": -1.5799081325531006, + "logits/rejected": -1.6663821935653687, + "logps/chosen": -6.070633888244629, + "logps/rejected": -6.397655487060547, + "loss": 0.6615, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.03644927218556404, + "rewards/margins": 0.06373059004545212, + "rewards/rejected": -0.10017986595630646, + "step": 510 + }, + { + "epoch": 0.13, + "learning_rate": 1.7736656343186894e-07, + "logits/chosen": -1.5198477506637573, + "logits/rejected": -1.5320686101913452, + "logps/chosen": -4.8720879554748535, + "logps/rejected": -6.920307159423828, + "loss": 0.6546, + "rewards/accuracies": 0.8125, + "rewards/chosen": 0.011885623447597027, + "rewards/margins": 0.12990930676460266, + "rewards/rejected": -0.11802370846271515, + "step": 515 + }, + { + "epoch": 0.13, + "learning_rate": 1.768401110856859e-07, + "logits/chosen": -1.4357372522354126, + "logits/rejected": -1.3116753101348877, + "logps/chosen": -5.608946323394775, + "logps/rejected": -7.910216331481934, + "loss": 0.6588, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.03163328021764755, + "rewards/margins": 0.06858251243829727, + "rewards/rejected": -0.10021580755710602, + "step": 520 + }, + { + "epoch": 0.13, + "learning_rate": 1.7630840681998066e-07, + "logits/chosen": -1.5340244770050049, + "logits/rejected": -1.417262315750122, + "logps/chosen": -5.505238056182861, + "logps/rejected": -9.424348831176758, + "loss": 0.657, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.043297283351421356, + "rewards/margins": 0.06860554218292236, + "rewards/rejected": -0.11190283298492432, + "step": 525 + }, + { + "epoch": 0.13, + "learning_rate": 1.7577148697603348e-07, + "logits/chosen": -1.7233011722564697, + "logits/rejected": -1.6292282342910767, + "logps/chosen": -5.7183942794799805, + "logps/rejected": -6.829588890075684, + "loss": 0.6596, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.0456002876162529, + "rewards/margins": 0.06835009157657623, + "rewards/rejected": -0.11395038664340973, + "step": 530 + }, + { + "epoch": 0.13, + "learning_rate": 1.7522938825160247e-07, + "logits/chosen": -1.3610880374908447, + "logits/rejected": -1.3855262994766235, + "logps/chosen": -5.5726470947265625, + "logps/rejected": -7.635517120361328, + "loss": 0.6577, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.03400539606809616, + "rewards/margins": 0.03809204697608948, + "rewards/rejected": -0.07209744304418564, + "step": 535 + }, + { + "epoch": 0.14, + "learning_rate": 1.7468214769841538e-07, + "logits/chosen": -1.5978724956512451, + "logits/rejected": -1.4684964418411255, + "logps/chosen": -5.9864959716796875, + "logps/rejected": -7.737608432769775, + "loss": 0.6572, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.044120945036411285, + "rewards/margins": 0.06804494559764862, + "rewards/rejected": -0.11216588318347931, + "step": 540 + }, + { + "epoch": 0.14, + "learning_rate": 1.7412980271963708e-07, + "logits/chosen": -1.567631721496582, + "logits/rejected": -1.54317045211792, + "logps/chosen": -6.259519100189209, + "logps/rejected": -7.840844631195068, + "loss": 0.6614, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.006099678575992584, + "rewards/margins": 0.1069895401597023, + "rewards/rejected": -0.11308921873569489, + "step": 545 + }, + { + "epoch": 0.14, + "learning_rate": 1.7357239106731316e-07, + "logits/chosen": -1.6185524463653564, + "logits/rejected": -1.6444612741470337, + "logps/chosen": -7.109327793121338, + "logps/rejected": -8.976202964782715, + "loss": 0.6555, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.04382595047354698, + "rewards/margins": 0.09603992104530334, + "rewards/rejected": -0.13986587524414062, + "step": 550 + }, + { + "epoch": 0.14, + "learning_rate": 1.7300995083978961e-07, + "logits/chosen": -1.7173988819122314, + "logits/rejected": -1.5636038780212402, + "logps/chosen": -5.074532508850098, + "logps/rejected": -7.6101531982421875, + "loss": 0.6536, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.04033565893769264, + "rewards/margins": 0.08647169172763824, + "rewards/rejected": -0.1268073469400406, + "step": 555 + }, + { + "epoch": 0.14, + "learning_rate": 1.724425204791089e-07, + "logits/chosen": -1.3909645080566406, + "logits/rejected": -1.4084515571594238, + "logps/chosen": -5.7829999923706055, + "logps/rejected": -6.748002529144287, + "loss": 0.6638, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.03097168169915676, + "rewards/margins": 0.04933810234069824, + "rewards/rejected": -0.08030977845191956, + "step": 560 + }, + { + "epoch": 0.14, + "learning_rate": 1.7187013876838238e-07, + "logits/chosen": -1.8116487264633179, + "logits/rejected": -1.7086451053619385, + "logps/chosen": -6.233320236206055, + "logps/rejected": -8.50374698638916, + "loss": 0.6505, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.014792789705097675, + "rewards/margins": 0.09041319042444229, + "rewards/rejected": -0.1052059754729271, + "step": 565 + }, + { + "epoch": 0.14, + "learning_rate": 1.712928448291397e-07, + "logits/chosen": -1.4688892364501953, + "logits/rejected": -1.531427025794983, + "logps/chosen": -6.673917293548584, + "logps/rejected": -6.955280303955078, + "loss": 0.6468, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.013536572456359863, + "rewards/margins": 0.09923507273197174, + "rewards/rejected": -0.11277163028717041, + "step": 570 + }, + { + "epoch": 0.14, + "learning_rate": 1.7071067811865473e-07, + "logits/chosen": -1.7241573333740234, + "logits/rejected": -1.6227924823760986, + "logps/chosen": -6.4433488845825195, + "logps/rejected": -9.645414352416992, + "loss": 0.6523, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.054280836135149, + "rewards/margins": 0.09889966994524002, + "rewards/rejected": -0.15318050980567932, + "step": 575 + }, + { + "epoch": 0.15, + "learning_rate": 1.7012367842724884e-07, + "logits/chosen": -1.4540927410125732, + "logits/rejected": -1.448845624923706, + "logps/chosen": -7.165667533874512, + "logps/rejected": -9.222039222717285, + "loss": 0.6556, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.03299577161669731, + "rewards/margins": 0.09829618781805038, + "rewards/rejected": -0.1312919557094574, + "step": 580 + }, + { + "epoch": 0.15, + "learning_rate": 1.695318858755712e-07, + "logits/chosen": -1.6242682933807373, + "logits/rejected": -1.507509708404541, + "logps/chosen": -4.892175197601318, + "logps/rejected": -6.663909912109375, + "loss": 0.6587, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": -0.05031831935048103, + "rewards/margins": 0.06618543714284897, + "rewards/rejected": -0.1165037602186203, + "step": 585 + }, + { + "epoch": 0.15, + "learning_rate": 1.6893534091185658e-07, + "logits/chosen": -1.655474305152893, + "logits/rejected": -1.6360971927642822, + "logps/chosen": -6.940272331237793, + "logps/rejected": -10.048660278320312, + "loss": 0.6463, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.006456449627876282, + "rewards/margins": 0.1727210134267807, + "rewards/rejected": -0.17917747795581818, + "step": 590 + }, + { + "epoch": 0.15, + "learning_rate": 1.6833408430916082e-07, + "logits/chosen": -1.3439010381698608, + "logits/rejected": -1.429510235786438, + "logps/chosen": -5.466313362121582, + "logps/rejected": -7.18454647064209, + "loss": 0.6487, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.05292614549398422, + "rewards/margins": 0.12489708513021469, + "rewards/rejected": -0.17782321572303772, + "step": 595 + }, + { + "epoch": 0.15, + "learning_rate": 1.6772815716257412e-07, + "logits/chosen": -1.51864492893219, + "logits/rejected": -1.455951452255249, + "logps/chosen": -6.4935503005981445, + "logps/rejected": -8.119839668273926, + "loss": 0.6537, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.0410698801279068, + "rewards/margins": 0.07796063274145126, + "rewards/rejected": -0.11903052031993866, + "step": 600 + }, + { + "epoch": 0.15, + "eval_logits/chosen": -3.1874260902404785, + "eval_logits/rejected": -3.1666018962860107, + "eval_logps/chosen": -6.68336296081543, + "eval_logps/rejected": -7.635651588439941, + "eval_loss": 0.6910278797149658, + "eval_rewards/accuracies": 0.5352563858032227, + "eval_rewards/chosen": -0.05427387356758118, + "eval_rewards/margins": 0.014749797061085701, + "eval_rewards/rejected": -0.06902367621660233, + "eval_runtime": 633.9901, + "eval_samples_per_second": 31.442, + "eval_steps_per_second": 0.492, + "step": 600 + }, + { + "epoch": 0.15, + "learning_rate": 1.6711760088641197e-07, + "logits/chosen": -1.707196593284607, + "logits/rejected": -1.7934414148330688, + "logps/chosen": -6.166356086730957, + "logps/rejected": -8.222631454467773, + "loss": 0.6509, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.04317507520318031, + "rewards/margins": 0.09725725650787354, + "rewards/rejected": -0.14043232798576355, + "step": 605 + }, + { + "epoch": 0.15, + "learning_rate": 1.665024572113848e-07, + "logits/chosen": -1.545772910118103, + "logits/rejected": -1.6466634273529053, + "logps/chosen": -6.318427085876465, + "logps/rejected": -8.524378776550293, + "loss": 0.6514, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.0410161130130291, + "rewards/margins": 0.10211832821369171, + "rewards/rejected": -0.14313443005084991, + "step": 610 + }, + { + "epoch": 0.15, + "learning_rate": 1.6588276818174578e-07, + "logits/chosen": -1.8286006450653076, + "logits/rejected": -1.8894586563110352, + "logps/chosen": -6.069952011108398, + "logps/rejected": -9.206938743591309, + "loss": 0.651, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.020980229601264, + "rewards/margins": 0.11290951818227768, + "rewards/rejected": -0.13388976454734802, + "step": 615 + }, + { + "epoch": 0.16, + "learning_rate": 1.6525857615241686e-07, + "logits/chosen": -1.47043776512146, + "logits/rejected": -1.5266094207763672, + "logps/chosen": -6.51092529296875, + "logps/rejected": -7.585784912109375, + "loss": 0.6509, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.04623446241021156, + "rewards/margins": 0.08200391381978989, + "rewards/rejected": -0.12823837995529175, + "step": 620 + }, + { + "epoch": 0.16, + "learning_rate": 1.6462992378609406e-07, + "logits/chosen": -1.3878867626190186, + "logits/rejected": -1.427337884902954, + "logps/chosen": -5.593390464782715, + "logps/rejected": -8.398979187011719, + "loss": 0.6468, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.024669837206602097, + "rewards/margins": 0.16932611167430878, + "rewards/rejected": -0.19399593770503998, + "step": 625 + }, + { + "epoch": 0.16, + "learning_rate": 1.6399685405033166e-07, + "logits/chosen": -1.4157261848449707, + "logits/rejected": -1.3857667446136475, + "logps/chosen": -7.09283447265625, + "logps/rejected": -7.571598052978516, + "loss": 0.6464, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.06203197315335274, + "rewards/margins": 0.08331136405467987, + "rewards/rejected": -0.1453433334827423, + "step": 630 + }, + { + "epoch": 0.16, + "learning_rate": 1.6335941021460504e-07, + "logits/chosen": -1.4545994997024536, + "logits/rejected": -1.5172913074493408, + "logps/chosen": -6.2775068283081055, + "logps/rejected": -9.521524429321289, + "loss": 0.6483, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.045348554849624634, + "rewards/margins": 0.10597596317529678, + "rewards/rejected": -0.15132452547550201, + "step": 635 + }, + { + "epoch": 0.16, + "learning_rate": 1.627176358473537e-07, + "logits/chosen": -1.5027107000350952, + "logits/rejected": -1.3006885051727295, + "logps/chosen": -6.991432189941406, + "logps/rejected": -8.119719505310059, + "loss": 0.6497, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.07398322969675064, + "rewards/margins": 0.03964931517839432, + "rewards/rejected": -0.11363253742456436, + "step": 640 + }, + { + "epoch": 0.16, + "learning_rate": 1.6207157481300312e-07, + "logits/chosen": -1.5265876054763794, + "logits/rejected": -1.4646865129470825, + "logps/chosen": -6.282595634460449, + "logps/rejected": -7.717820167541504, + "loss": 0.6474, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.07993953675031662, + "rewards/margins": 0.09873492270708084, + "rewards/rejected": -0.17867444455623627, + "step": 645 + }, + { + "epoch": 0.16, + "learning_rate": 1.614212712689668e-07, + "logits/chosen": -1.3973186016082764, + "logits/rejected": -1.3966352939605713, + "logps/chosen": -6.049260139465332, + "logps/rejected": -6.803788185119629, + "loss": 0.6457, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.053327836096286774, + "rewards/margins": 0.13262109458446503, + "rewards/rejected": -0.1859489530324936, + "step": 650 + }, + { + "epoch": 0.16, + "learning_rate": 1.607667696626281e-07, + "logits/chosen": -1.604760766029358, + "logits/rejected": -1.6347767114639282, + "logps/chosen": -5.876626014709473, + "logps/rejected": -9.119338989257812, + "loss": 0.6534, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.058147139847278595, + "rewards/margins": 0.09601394087076187, + "rewards/rejected": -0.15416109561920166, + "step": 655 + }, + { + "epoch": 0.17, + "learning_rate": 1.601081147283025e-07, + "logits/chosen": -1.517514944076538, + "logits/rejected": -1.4594268798828125, + "logps/chosen": -5.623353004455566, + "logps/rejected": -8.838793754577637, + "loss": 0.6423, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.034189894795417786, + "rewards/margins": 0.120149627327919, + "rewards/rejected": -0.1543395221233368, + "step": 660 + }, + { + "epoch": 0.17, + "learning_rate": 1.594453514841798e-07, + "logits/chosen": -1.4519226551055908, + "logits/rejected": -1.4916832447052002, + "logps/chosen": -6.024622917175293, + "logps/rejected": -8.156673431396484, + "loss": 0.6421, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.03883025795221329, + "rewards/margins": 0.14476829767227173, + "rewards/rejected": -0.18359854817390442, + "step": 665 + }, + { + "epoch": 0.17, + "learning_rate": 1.5877852522924732e-07, + "logits/chosen": -1.4973927736282349, + "logits/rejected": -1.5078349113464355, + "logps/chosen": -5.639887809753418, + "logps/rejected": -6.538842678070068, + "loss": 0.6519, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.046415191143751144, + "rewards/margins": 0.08505970239639282, + "rewards/rejected": -0.13147488236427307, + "step": 670 + }, + { + "epoch": 0.17, + "learning_rate": 1.5810768154019382e-07, + "logits/chosen": -1.6372696161270142, + "logits/rejected": -1.604660987854004, + "logps/chosen": -6.092626571655273, + "logps/rejected": -7.028026580810547, + "loss": 0.6493, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.07696966081857681, + "rewards/margins": 0.10025089979171753, + "rewards/rejected": -0.17722055315971375, + "step": 675 + }, + { + "epoch": 0.17, + "learning_rate": 1.5743286626829435e-07, + "logits/chosen": -1.3090213537216187, + "logits/rejected": -1.3185111284255981, + "logps/chosen": -4.761218070983887, + "logps/rejected": -6.58809757232666, + "loss": 0.6395, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.03409787639975548, + "rewards/margins": 0.1063770055770874, + "rewards/rejected": -0.14047487080097198, + "step": 680 + }, + { + "epoch": 0.17, + "learning_rate": 1.5675412553627636e-07, + "logits/chosen": -1.4627618789672852, + "logits/rejected": -1.4510043859481812, + "logps/chosen": -5.731337547302246, + "logps/rejected": -10.051878929138184, + "loss": 0.6498, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.03276892006397247, + "rewards/margins": 0.13961870968341827, + "rewards/rejected": -0.17238759994506836, + "step": 685 + }, + { + "epoch": 0.17, + "learning_rate": 1.5607150573516727e-07, + "logits/chosen": -1.450136423110962, + "logits/rejected": -1.4004848003387451, + "logps/chosen": -6.355518341064453, + "logps/rejected": -7.548236846923828, + "loss": 0.6462, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.07692820578813553, + "rewards/margins": 0.07390854507684708, + "rewards/rejected": -0.1508367359638214, + "step": 690 + }, + { + "epoch": 0.17, + "learning_rate": 1.5538505352112372e-07, + "logits/chosen": -1.7112722396850586, + "logits/rejected": -1.4571855068206787, + "logps/chosen": -6.127640724182129, + "logps/rejected": -7.857392311096191, + "loss": 0.6451, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.07270301878452301, + "rewards/margins": 0.13500170409679413, + "rewards/rejected": -0.20770475268363953, + "step": 695 + }, + { + "epoch": 0.18, + "learning_rate": 1.546948158122427e-07, + "logits/chosen": -1.7370418310165405, + "logits/rejected": -1.7001125812530518, + "logps/chosen": -5.915326118469238, + "logps/rejected": -8.323230743408203, + "loss": 0.6435, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.07308584451675415, + "rewards/margins": 0.15741673111915588, + "rewards/rejected": -0.23050260543823242, + "step": 700 + }, + { + "epoch": 0.18, + "eval_logits/chosen": -3.186901092529297, + "eval_logits/rejected": -3.1663575172424316, + "eval_logps/chosen": -6.852160930633545, + "eval_logps/rejected": -7.856788158416748, + "eval_loss": 0.6899130344390869, + "eval_rewards/accuracies": 0.5376602411270142, + "eval_rewards/chosen": -0.0711536630988121, + "eval_rewards/margins": 0.019983632490038872, + "eval_rewards/rejected": -0.09113729745149612, + "eval_runtime": 640.2435, + "eval_samples_per_second": 31.135, + "eval_steps_per_second": 0.487, + "step": 700 + }, + { + "epoch": 0.18, + "learning_rate": 1.540008397853547e-07, + "logits/chosen": -1.5285913944244385, + "logits/rejected": -1.506296992301941, + "logps/chosen": -4.79876184463501, + "logps/rejected": -7.281890869140625, + "loss": 0.6362, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.01859206147491932, + "rewards/margins": 0.11602876335382462, + "rewards/rejected": -0.1346208155155182, + "step": 705 + }, + { + "epoch": 0.18, + "learning_rate": 1.5330317287279937e-07, + "logits/chosen": -1.489984393119812, + "logits/rejected": -1.384037733078003, + "logps/chosen": -6.871964931488037, + "logps/rejected": -9.540796279907227, + "loss": 0.6343, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.09499697387218475, + "rewards/margins": 0.11305873095989227, + "rewards/rejected": -0.20805568993091583, + "step": 710 + }, + { + "epoch": 0.18, + "learning_rate": 1.526018627591834e-07, + "logits/chosen": -1.4063003063201904, + "logits/rejected": -1.392664909362793, + "logps/chosen": -7.688409328460693, + "logps/rejected": -8.560149192810059, + "loss": 0.6462, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.08843918889760971, + "rewards/margins": 0.12001106888055801, + "rewards/rejected": -0.20845024287700653, + "step": 715 + }, + { + "epoch": 0.18, + "learning_rate": 1.5189695737812152e-07, + "logits/chosen": -1.4897139072418213, + "logits/rejected": -1.437138319015503, + "logps/chosen": -5.6488165855407715, + "logps/rejected": -8.617653846740723, + "loss": 0.637, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.08085988461971283, + "rewards/margins": 0.1466083824634552, + "rewards/rejected": -0.22746825218200684, + "step": 720 + }, + { + "epoch": 0.18, + "learning_rate": 1.511885049089601e-07, + "logits/chosen": -1.4502151012420654, + "logits/rejected": -1.558260440826416, + "logps/chosen": -6.4499664306640625, + "logps/rejected": -8.321115493774414, + "loss": 0.6455, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.08257200568914413, + "rewards/margins": 0.10597743839025497, + "rewards/rejected": -0.1885494440793991, + "step": 725 + }, + { + "epoch": 0.18, + "learning_rate": 1.5047655377348439e-07, + "logits/chosen": -1.3676706552505493, + "logits/rejected": -1.3527071475982666, + "logps/chosen": -6.454686641693115, + "logps/rejected": -7.9275712966918945, + "loss": 0.6392, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.07613205164670944, + "rewards/margins": 0.11505208909511566, + "rewards/rejected": -0.1911841332912445, + "step": 730 + }, + { + "epoch": 0.18, + "learning_rate": 1.4976115263260874e-07, + "logits/chosen": -1.4929918050765991, + "logits/rejected": -1.4645825624465942, + "logps/chosen": -5.613523006439209, + "logps/rejected": -8.017206192016602, + "loss": 0.6462, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.043209511786699295, + "rewards/margins": 0.09339593350887299, + "rewards/rejected": -0.13660545647144318, + "step": 735 + }, + { + "epoch": 0.19, + "learning_rate": 1.4904235038305082e-07, + "logits/chosen": -1.430387258529663, + "logits/rejected": -1.5527921915054321, + "logps/chosen": -7.354534149169922, + "logps/rejected": -8.235260963439941, + "loss": 0.6458, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.07374344021081924, + "rewards/margins": 0.14879578351974487, + "rewards/rejected": -0.22253921627998352, + "step": 740 + }, + { + "epoch": 0.19, + "learning_rate": 1.483201961539896e-07, + "logits/chosen": -1.4932714700698853, + "logits/rejected": -1.4173743724822998, + "logps/chosen": -5.170498847961426, + "logps/rejected": -7.974796295166016, + "loss": 0.6307, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.04303674027323723, + "rewards/margins": 0.15367302298545837, + "rewards/rejected": -0.1967097669839859, + "step": 745 + }, + { + "epoch": 0.19, + "learning_rate": 1.4759473930370737e-07, + "logits/chosen": -1.7062429189682007, + "logits/rejected": -1.6499900817871094, + "logps/chosen": -6.257295608520508, + "logps/rejected": -7.387567043304443, + "loss": 0.6434, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.10747422277927399, + "rewards/margins": 0.08016838133335114, + "rewards/rejected": -0.18764260411262512, + "step": 750 + }, + { + "epoch": 0.19, + "learning_rate": 1.4686602941621615e-07, + "logits/chosen": -1.244971752166748, + "logits/rejected": -1.2606076002120972, + "logps/chosen": -6.677346706390381, + "logps/rejected": -9.851409912109375, + "loss": 0.6353, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.08402319997549057, + "rewards/margins": 0.10642798244953156, + "rewards/rejected": -0.19045117497444153, + "step": 755 + }, + { + "epoch": 0.19, + "learning_rate": 1.4613411629786877e-07, + "logits/chosen": -1.4713022708892822, + "logits/rejected": -1.5048084259033203, + "logps/chosen": -6.147872447967529, + "logps/rejected": -7.910445213317871, + "loss": 0.6417, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.07068430632352829, + "rewards/margins": 0.16299991309642792, + "rewards/rejected": -0.2336842119693756, + "step": 760 + }, + { + "epoch": 0.19, + "learning_rate": 1.4539904997395468e-07, + "logits/chosen": -1.3301279544830322, + "logits/rejected": -1.3574144840240479, + "logps/chosen": -6.699339389801025, + "logps/rejected": -9.695663452148438, + "loss": 0.6279, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.06842740625143051, + "rewards/margins": 0.13677598536014557, + "rewards/rejected": -0.20520341396331787, + "step": 765 + }, + { + "epoch": 0.19, + "learning_rate": 1.4466088068528067e-07, + "logits/chosen": -1.51852548122406, + "logits/rejected": -1.4426764249801636, + "logps/chosen": -5.722136497497559, + "logps/rejected": -8.637394905090332, + "loss": 0.6389, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.0995694026350975, + "rewards/margins": 0.11200227588415146, + "rewards/rejected": -0.21157169342041016, + "step": 770 + }, + { + "epoch": 0.19, + "learning_rate": 1.4391965888473702e-07, + "logits/chosen": -1.5839837789535522, + "logits/rejected": -1.5870234966278076, + "logps/chosen": -6.29650354385376, + "logps/rejected": -9.936391830444336, + "loss": 0.63, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.08398771286010742, + "rewards/margins": 0.11872847378253937, + "rewards/rejected": -0.20271620154380798, + "step": 775 + }, + { + "epoch": 0.2, + "learning_rate": 1.4317543523384928e-07, + "logits/chosen": -1.694641351699829, + "logits/rejected": -1.6382360458374023, + "logps/chosen": -6.944577217102051, + "logps/rejected": -8.599451065063477, + "loss": 0.6283, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.08465109765529633, + "rewards/margins": 0.16851410269737244, + "rewards/rejected": -0.25316524505615234, + "step": 780 + }, + { + "epoch": 0.2, + "learning_rate": 1.4242826059931536e-07, + "logits/chosen": -1.4301807880401611, + "logits/rejected": -1.3396793603897095, + "logps/chosen": -5.984529972076416, + "logps/rejected": -8.90467643737793, + "loss": 0.6422, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.11020038276910782, + "rewards/margins": 0.13162291049957275, + "rewards/rejected": -0.24182331562042236, + "step": 785 + }, + { + "epoch": 0.2, + "learning_rate": 1.4167818604952903e-07, + "logits/chosen": -1.8726218938827515, + "logits/rejected": -1.8315349817276, + "logps/chosen": -6.4224066734313965, + "logps/rejected": -9.21501350402832, + "loss": 0.6316, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.08100078999996185, + "rewards/margins": 0.16276691854000092, + "rewards/rejected": -0.24376770853996277, + "step": 790 + }, + { + "epoch": 0.2, + "learning_rate": 1.4092526285108939e-07, + "logits/chosen": -1.6488440036773682, + "logits/rejected": -1.6959072351455688, + "logps/chosen": -5.05312442779541, + "logps/rejected": -8.503290176391602, + "loss": 0.6354, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.025250781327486038, + "rewards/margins": 0.17517754435539246, + "rewards/rejected": -0.2004283368587494, + "step": 795 + }, + { + "epoch": 0.2, + "learning_rate": 1.4016954246529695e-07, + "logits/chosen": -1.5666801929473877, + "logits/rejected": -1.6438080072402954, + "logps/chosen": -6.046337127685547, + "logps/rejected": -8.245607376098633, + "loss": 0.6332, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.08281711488962173, + "rewards/margins": 0.1388954371213913, + "rewards/rejected": -0.22171254456043243, + "step": 800 + }, + { + "epoch": 0.2, + "eval_logits/chosen": -3.1937944889068604, + "eval_logits/rejected": -3.173306941986084, + "eval_logps/chosen": -7.060844421386719, + "eval_logps/rejected": -8.098358154296875, + "eval_loss": 0.6894627213478088, + "eval_rewards/accuracies": 0.5336538553237915, + "eval_rewards/chosen": -0.09202194213867188, + "eval_rewards/margins": 0.023272372782230377, + "eval_rewards/rejected": -0.11529432237148285, + "eval_runtime": 642.2741, + "eval_samples_per_second": 31.037, + "eval_steps_per_second": 0.486, + "step": 800 + }, + { + "epoch": 0.2, + "learning_rate": 1.3941107654463616e-07, + "logits/chosen": -1.6347227096557617, + "logits/rejected": -1.5565736293792725, + "logps/chosen": -6.210860252380371, + "logps/rejected": -8.191205978393555, + "loss": 0.633, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.0968286544084549, + "rewards/margins": 0.13983137905597687, + "rewards/rejected": -0.23666004836559296, + "step": 805 + }, + { + "epoch": 0.2, + "learning_rate": 1.3864991692924522e-07, + "logits/chosen": -1.6105167865753174, + "logits/rejected": -1.5960274934768677, + "logps/chosen": -6.203638553619385, + "logps/rejected": -9.020109176635742, + "loss": 0.6296, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.09015347063541412, + "rewards/margins": 0.13406077027320862, + "rewards/rejected": -0.22421424090862274, + "step": 810 + }, + { + "epoch": 0.2, + "learning_rate": 1.3788611564337276e-07, + "logits/chosen": -1.5526584386825562, + "logits/rejected": -1.5093594789505005, + "logps/chosen": -6.796191215515137, + "logps/rejected": -8.405149459838867, + "loss": 0.6223, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.08056505024433136, + "rewards/margins": 0.13686737418174744, + "rewards/rejected": -0.21743245422840118, + "step": 815 + }, + { + "epoch": 0.21, + "learning_rate": 1.3711972489182207e-07, + "logits/chosen": -1.311798334121704, + "logits/rejected": -1.2505333423614502, + "logps/chosen": -6.062074184417725, + "logps/rejected": -8.897860527038574, + "loss": 0.6306, + "rewards/accuracies": 0.8125, + "rewards/chosen": -0.030773481354117393, + "rewards/margins": 0.19692835211753845, + "rewards/rejected": -0.22770185768604279, + "step": 820 + }, + { + "epoch": 0.21, + "learning_rate": 1.3635079705638297e-07, + "logits/chosen": -1.5671780109405518, + "logits/rejected": -1.5879342555999756, + "logps/chosen": -6.3402533531188965, + "logps/rejected": -9.183328628540039, + "loss": 0.6338, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.06746585667133331, + "rewards/margins": 0.13011819124221802, + "rewards/rejected": -0.19758403301239014, + "step": 825 + }, + { + "epoch": 0.21, + "learning_rate": 1.3557938469225164e-07, + "logits/chosen": -1.423302412033081, + "logits/rejected": -1.4359217882156372, + "logps/chosen": -6.0072126388549805, + "logps/rejected": -9.093564987182617, + "loss": 0.6206, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.09784204512834549, + "rewards/margins": 0.13212862610816956, + "rewards/rejected": -0.22997066378593445, + "step": 830 + }, + { + "epoch": 0.21, + "learning_rate": 1.3480554052443843e-07, + "logits/chosen": -1.7213542461395264, + "logits/rejected": -1.6981548070907593, + "logps/chosen": -7.577986240386963, + "logps/rejected": -10.498396873474121, + "loss": 0.6315, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.10542915016412735, + "rewards/margins": 0.1352270096540451, + "rewards/rejected": -0.24065613746643066, + "step": 835 + }, + { + "epoch": 0.21, + "learning_rate": 1.340293174441643e-07, + "logits/chosen": -1.6847597360610962, + "logits/rejected": -1.6527271270751953, + "logps/chosen": -8.06293773651123, + "logps/rejected": -9.775976181030273, + "loss": 0.6238, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.11751914024353027, + "rewards/margins": 0.1262793093919754, + "rewards/rejected": -0.24379844963550568, + "step": 840 + }, + { + "epoch": 0.21, + "learning_rate": 1.332507685052457e-07, + "logits/chosen": -1.3307305574417114, + "logits/rejected": -1.296414852142334, + "logps/chosen": -5.506723880767822, + "logps/rejected": -9.845108032226562, + "loss": 0.6284, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.05931394174695015, + "rewards/margins": 0.17723853886127472, + "rewards/rejected": -0.23655244708061218, + "step": 845 + }, + { + "epoch": 0.21, + "learning_rate": 1.3246994692046836e-07, + "logits/chosen": -1.4784770011901855, + "logits/rejected": -1.4177029132843018, + "logps/chosen": -6.148890495300293, + "logps/rejected": -9.090556144714355, + "loss": 0.6283, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.07422871887683868, + "rewards/margins": 0.16514147818088531, + "rewards/rejected": -0.239370197057724, + "step": 850 + }, + { + "epoch": 0.21, + "learning_rate": 1.3168690605795043e-07, + "logits/chosen": -1.4130356311798096, + "logits/rejected": -1.453981637954712, + "logps/chosen": -5.568487167358398, + "logps/rejected": -9.608223915100098, + "loss": 0.6299, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.087576724588871, + "rewards/margins": 0.2175295650959015, + "rewards/rejected": -0.3051062524318695, + "step": 855 + }, + { + "epoch": 0.22, + "learning_rate": 1.3090169943749475e-07, + "logits/chosen": -1.4787486791610718, + "logits/rejected": -1.437145471572876, + "logps/chosen": -5.853695869445801, + "logps/rejected": -8.345501899719238, + "loss": 0.6359, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.05679728835821152, + "rewards/margins": 0.1354844868183136, + "rewards/rejected": -0.19228176772594452, + "step": 860 + }, + { + "epoch": 0.22, + "learning_rate": 1.3011438072693074e-07, + "logits/chosen": -1.5212247371673584, + "logits/rejected": -1.4433953762054443, + "logps/chosen": -6.458197593688965, + "logps/rejected": -9.791971206665039, + "loss": 0.6192, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.07254395633935928, + "rewards/margins": 0.23906107246875763, + "rewards/rejected": -0.3116050362586975, + "step": 865 + }, + { + "epoch": 0.22, + "learning_rate": 1.2932500373844649e-07, + "logits/chosen": -1.6338106393814087, + "logits/rejected": -1.4234752655029297, + "logps/chosen": -5.266790866851807, + "logps/rejected": -8.406009674072266, + "loss": 0.6266, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.09523673355579376, + "rewards/margins": 0.15846498310565948, + "rewards/rejected": -0.25370171666145325, + "step": 870 + }, + { + "epoch": 0.22, + "learning_rate": 1.2853362242491051e-07, + "logits/chosen": -1.4831264019012451, + "logits/rejected": -1.5002310276031494, + "logps/chosen": -5.224946022033691, + "logps/rejected": -7.2889556884765625, + "loss": 0.6128, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.06067908555269241, + "rewards/margins": 0.17062053084373474, + "rewards/rejected": -0.23129959404468536, + "step": 875 + }, + { + "epoch": 0.22, + "learning_rate": 1.2774029087618446e-07, + "logits/chosen": -1.758953332901001, + "logits/rejected": -1.6146351099014282, + "logps/chosen": -7.610304355621338, + "logps/rejected": -10.307147026062012, + "loss": 0.619, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.07071717083454132, + "rewards/margins": 0.20475205779075623, + "rewards/rejected": -0.27546921372413635, + "step": 880 + }, + { + "epoch": 0.22, + "learning_rate": 1.2694506331542577e-07, + "logits/chosen": -1.591770052909851, + "logits/rejected": -1.4535247087478638, + "logps/chosen": -5.625763893127441, + "logps/rejected": -9.85864543914795, + "loss": 0.6345, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.0788426622748375, + "rewards/margins": 0.15841318666934967, + "rewards/rejected": -0.23725584149360657, + "step": 885 + }, + { + "epoch": 0.22, + "learning_rate": 1.2614799409538198e-07, + "logits/chosen": -1.4659042358398438, + "logits/rejected": -1.4563062191009521, + "logps/chosen": -5.280413627624512, + "logps/rejected": -9.323397636413574, + "loss": 0.6226, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.09044213593006134, + "rewards/margins": 0.1876935213804245, + "rewards/rejected": -0.2781356871128082, + "step": 890 + }, + { + "epoch": 0.22, + "learning_rate": 1.253491376946754e-07, + "logits/chosen": -1.3718680143356323, + "logits/rejected": -1.3589755296707153, + "logps/chosen": -6.7582688331604, + "logps/rejected": -10.708890914916992, + "loss": 0.6275, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.06859103590250015, + "rewards/margins": 0.16634361445903778, + "rewards/rejected": -0.23493464291095734, + "step": 895 + }, + { + "epoch": 0.23, + "learning_rate": 1.2454854871407992e-07, + "logits/chosen": -1.3176133632659912, + "logits/rejected": -1.3136779069900513, + "logps/chosen": -6.8164777755737305, + "logps/rejected": -9.460234642028809, + "loss": 0.6217, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.1115039810538292, + "rewards/margins": 0.17541712522506714, + "rewards/rejected": -0.28692108392715454, + "step": 900 + }, + { + "epoch": 0.23, + "eval_logits/chosen": -3.18924617767334, + "eval_logits/rejected": -3.1685447692871094, + "eval_logps/chosen": -7.315045356750488, + "eval_logps/rejected": -8.408605575561523, + "eval_loss": 0.6886266469955444, + "eval_rewards/accuracies": 0.5352563858032227, + "eval_rewards/chosen": -0.11744209378957748, + "eval_rewards/margins": 0.0288768969476223, + "eval_rewards/rejected": -0.14631898701190948, + "eval_runtime": 646.5131, + "eval_samples_per_second": 30.833, + "eval_steps_per_second": 0.483, + "step": 900 + }, + { + "epoch": 0.23, + "learning_rate": 1.2374628187278885e-07, + "logits/chosen": -1.6750404834747314, + "logits/rejected": -1.6324043273925781, + "logps/chosen": -6.7752885818481445, + "logps/rejected": -9.884492874145508, + "loss": 0.6205, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.08316051214933395, + "rewards/margins": 0.19135987758636475, + "rewards/rejected": -0.2745203971862793, + "step": 905 + }, + { + "epoch": 0.23, + "learning_rate": 1.2294239200467515e-07, + "logits/chosen": -1.4121886491775513, + "logits/rejected": -1.4634597301483154, + "logps/chosen": -8.272059440612793, + "logps/rejected": -10.233227729797363, + "loss": 0.6219, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.1476745754480362, + "rewards/margins": 0.1988331377506256, + "rewards/rejected": -0.346507728099823, + "step": 910 + }, + { + "epoch": 0.23, + "learning_rate": 1.2213693405454345e-07, + "logits/chosen": -1.6356998682022095, + "logits/rejected": -1.6384683847427368, + "logps/chosen": -5.971181392669678, + "logps/rejected": -8.54847240447998, + "loss": 0.6212, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.13469815254211426, + "rewards/margins": 0.1848105490207672, + "rewards/rejected": -0.31950870156288147, + "step": 915 + }, + { + "epoch": 0.23, + "learning_rate": 1.213299630743747e-07, + "logits/chosen": -1.4081295728683472, + "logits/rejected": -1.333418846130371, + "logps/chosen": -7.040464878082275, + "logps/rejected": -8.731954574584961, + "loss": 0.6179, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.12728752195835114, + "rewards/margins": 0.17130446434020996, + "rewards/rejected": -0.2985920011997223, + "step": 920 + }, + { + "epoch": 0.23, + "learning_rate": 1.205215342195634e-07, + "logits/chosen": -1.5196300745010376, + "logits/rejected": -1.5172007083892822, + "logps/chosen": -5.7593536376953125, + "logps/rejected": -9.683046340942383, + "loss": 0.6175, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.10798660665750504, + "rewards/margins": 0.25386783480644226, + "rewards/rejected": -0.3618544340133667, + "step": 925 + }, + { + "epoch": 0.23, + "learning_rate": 1.1971170274514802e-07, + "logits/chosen": -1.6615116596221924, + "logits/rejected": -1.6914132833480835, + "logps/chosen": -7.840911865234375, + "logps/rejected": -9.82246208190918, + "loss": 0.6232, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.14061352610588074, + "rewards/margins": 0.14244748651981354, + "rewards/rejected": -0.2830609679222107, + "step": 930 + }, + { + "epoch": 0.23, + "learning_rate": 1.1890052400203402e-07, + "logits/chosen": -1.834905982017517, + "logits/rejected": -1.7889604568481445, + "logps/chosen": -7.288022041320801, + "logps/rejected": -10.376224517822266, + "loss": 0.6095, + "rewards/accuracies": 0.5874999761581421, + "rewards/chosen": -0.1694309115409851, + "rewards/margins": 0.1324268877506256, + "rewards/rejected": -0.3018577992916107, + "step": 935 + }, + { + "epoch": 0.24, + "learning_rate": 1.18088053433211e-07, + "logits/chosen": -1.5051194429397583, + "logits/rejected": -1.5041710138320923, + "logps/chosen": -6.950632572174072, + "logps/rejected": -9.823295593261719, + "loss": 0.6208, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.11773916333913803, + "rewards/margins": 0.17614324390888214, + "rewards/rejected": -0.29388242959976196, + "step": 940 + }, + { + "epoch": 0.24, + "learning_rate": 1.1727434656996305e-07, + "logits/chosen": -1.6864219903945923, + "logits/rejected": -1.7134393453598022, + "logps/chosen": -8.343110084533691, + "logps/rejected": -10.6605863571167, + "loss": 0.6269, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.1767118275165558, + "rewards/margins": 0.15801793336868286, + "rewards/rejected": -0.33472976088523865, + "step": 945 + }, + { + "epoch": 0.24, + "learning_rate": 1.1645945902807339e-07, + "logits/chosen": -1.521843671798706, + "logits/rejected": -1.4162412881851196, + "logps/chosen": -5.303705215454102, + "logps/rejected": -8.564371109008789, + "loss": 0.6195, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.08533243834972382, + "rewards/margins": 0.2330024540424347, + "rewards/rejected": -0.3183349072933197, + "step": 950 + }, + { + "epoch": 0.24, + "learning_rate": 1.1564344650402309e-07, + "logits/chosen": -1.4854631423950195, + "logits/rejected": -1.5085594654083252, + "logps/chosen": -6.938723087310791, + "logps/rejected": -9.417280197143555, + "loss": 0.6228, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.11080266535282135, + "rewards/margins": 0.18629567325115204, + "rewards/rejected": -0.2970983386039734, + "step": 955 + }, + { + "epoch": 0.24, + "learning_rate": 1.1482636477118419e-07, + "logits/chosen": -1.5579372644424438, + "logits/rejected": -1.5621235370635986, + "logps/chosen": -6.055428981781006, + "logps/rejected": -8.274986267089844, + "loss": 0.617, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.08702854812145233, + "rewards/margins": 0.19237728416919708, + "rewards/rejected": -0.2794058322906494, + "step": 960 + }, + { + "epoch": 0.24, + "learning_rate": 1.1400826967600779e-07, + "logits/chosen": -1.6652123928070068, + "logits/rejected": -1.5670098066329956, + "logps/chosen": -6.44259786605835, + "logps/rejected": -9.023821830749512, + "loss": 0.6279, + "rewards/accuracies": 0.6625000238418579, + "rewards/chosen": -0.12496472895145416, + "rewards/margins": 0.1399177610874176, + "rewards/rejected": -0.26488250494003296, + "step": 965 + }, + { + "epoch": 0.24, + "learning_rate": 1.131892171342069e-07, + "logits/chosen": -1.5863568782806396, + "logits/rejected": -1.4539623260498047, + "logps/chosen": -8.477279663085938, + "logps/rejected": -11.437823295593262, + "loss": 0.6105, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.12180640548467636, + "rewards/margins": 0.20829221606254578, + "rewards/rejected": -0.3300986588001251, + "step": 970 + }, + { + "epoch": 0.24, + "learning_rate": 1.1236926312693478e-07, + "logits/chosen": -1.4525238275527954, + "logits/rejected": -1.3738415241241455, + "logps/chosen": -6.627744197845459, + "logps/rejected": -10.264178276062012, + "loss": 0.6171, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -0.12857326865196228, + "rewards/margins": 0.23741576075553894, + "rewards/rejected": -0.365989089012146, + "step": 975 + }, + { + "epoch": 0.25, + "learning_rate": 1.1154846369695863e-07, + "logits/chosen": -1.2888273000717163, + "logits/rejected": -1.2684919834136963, + "logps/chosen": -7.151463985443115, + "logps/rejected": -10.000380516052246, + "loss": 0.6203, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.09203344583511353, + "rewards/margins": 0.21837130188941956, + "rewards/rejected": -0.3104047179222107, + "step": 980 + }, + { + "epoch": 0.25, + "learning_rate": 1.1072687494482918e-07, + "logits/chosen": -1.3414087295532227, + "logits/rejected": -1.3495652675628662, + "logps/chosen": -7.086977481842041, + "logps/rejected": -10.637145042419434, + "loss": 0.6076, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.11822090297937393, + "rewards/margins": 0.2522438168525696, + "rewards/rejected": -0.3704647123813629, + "step": 985 + }, + { + "epoch": 0.25, + "learning_rate": 1.0990455302504628e-07, + "logits/chosen": -1.2805410623550415, + "logits/rejected": -1.3479764461517334, + "logps/chosen": -6.615203857421875, + "logps/rejected": -9.906420707702637, + "loss": 0.6078, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.1480402648448944, + "rewards/margins": 0.21094676852226257, + "rewards/rejected": -0.358987033367157, + "step": 990 + }, + { + "epoch": 0.25, + "learning_rate": 1.0908155414222082e-07, + "logits/chosen": -1.3715662956237793, + "logits/rejected": -1.4605109691619873, + "logps/chosen": -6.958237648010254, + "logps/rejected": -8.883742332458496, + "loss": 0.6167, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.13871873915195465, + "rewards/margins": 0.16220495104789734, + "rewards/rejected": -0.3009237051010132, + "step": 995 + }, + { + "epoch": 0.25, + "learning_rate": 1.0825793454723325e-07, + "logits/chosen": -1.5984854698181152, + "logits/rejected": -1.618065595626831, + "logps/chosen": -7.258332252502441, + "logps/rejected": -9.4107666015625, + "loss": 0.6015, + "rewards/accuracies": 0.8374999761581421, + "rewards/chosen": -0.0792965292930603, + "rewards/margins": 0.28216832876205444, + "rewards/rejected": -0.36146482825279236, + "step": 1000 + }, + { + "epoch": 0.25, + "eval_logits/chosen": -3.169694662094116, + "eval_logits/rejected": -3.1489059925079346, + "eval_logps/chosen": -7.470149040222168, + "eval_logps/rejected": -8.596240043640137, + "eval_loss": 0.6881732940673828, + "eval_rewards/accuracies": 0.5396634340286255, + "eval_rewards/chosen": -0.13295257091522217, + "eval_rewards/margins": 0.032129984349012375, + "eval_rewards/rejected": -0.16508255898952484, + "eval_runtime": 650.2113, + "eval_samples_per_second": 30.658, + "eval_steps_per_second": 0.48, + "step": 1000 + } + ], + "logging_steps": 5, + "max_steps": 2000, + "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 +}