{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.0, "eval_steps": 100, "global_step": 385, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.0, "learning_rate": 1.282051282051282e-08, "logits/chosen": -1.7278180122375488, "logits/rejected": -1.7377450466156006, "logps/chosen": -29.553977966308594, "logps/rejected": -42.813133239746094, "loss": 1.0, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.03, "learning_rate": 1.2820512820512818e-07, "logits/chosen": -1.866068720817566, "logits/rejected": -1.8703863620758057, "logps/chosen": -36.99105453491211, "logps/rejected": -33.65992736816406, "loss": 0.9728, "rewards/accuracies": 0.5138888955116272, "rewards/chosen": 0.009325359016656876, "rewards/margins": 0.027240369468927383, "rewards/rejected": -0.017915012314915657, "step": 10 }, { "epoch": 0.05, "learning_rate": 2.5641025641025636e-07, "logits/chosen": -1.9970887899398804, "logits/rejected": -1.9997413158416748, "logps/chosen": -29.63595199584961, "logps/rejected": -29.070571899414062, "loss": 0.992, "rewards/accuracies": 0.5, "rewards/chosen": 0.003744622226804495, "rewards/margins": 0.008021654561161995, "rewards/rejected": -0.004277031868696213, "step": 20 }, { "epoch": 0.08, "learning_rate": 3.8461538461538463e-07, "logits/chosen": -1.9195963144302368, "logits/rejected": -1.9168994426727295, "logps/chosen": -31.414175033569336, "logps/rejected": -33.25041961669922, "loss": 0.9797, "rewards/accuracies": 0.5625, "rewards/chosen": 0.0011531396303325891, "rewards/margins": 0.02032935619354248, "rewards/rejected": -0.019176214933395386, "step": 30 }, { "epoch": 0.1, "learning_rate": 4.999896948438433e-07, "logits/chosen": -2.01688814163208, "logits/rejected": -2.0081212520599365, "logps/chosen": -32.583614349365234, "logps/rejected": -32.48966979980469, "loss": 1.0179, "rewards/accuracies": 0.4375, "rewards/chosen": -0.004085154738277197, "rewards/margins": -0.017886672168970108, "rewards/rejected": 0.013801517896354198, "step": 40 }, { "epoch": 0.13, "learning_rate": 4.987541037542186e-07, "logits/chosen": -1.8644500970840454, "logits/rejected": -1.853655219078064, "logps/chosen": -33.54804229736328, "logps/rejected": -35.43767166137695, "loss": 1.0031, "rewards/accuracies": 0.4375, "rewards/chosen": 0.005664288066327572, "rewards/margins": -0.0031387731432914734, "rewards/rejected": 0.00880306214094162, "step": 50 }, { "epoch": 0.16, "learning_rate": 4.954691471941118e-07, "logits/chosen": -1.9449504613876343, "logits/rejected": -1.946890115737915, "logps/chosen": -32.58049774169922, "logps/rejected": -33.199378967285156, "loss": 0.9714, "rewards/accuracies": 0.612500011920929, "rewards/chosen": 0.011811850592494011, "rewards/margins": 0.02860717847943306, "rewards/rejected": -0.016795331612229347, "step": 60 }, { "epoch": 0.18, "learning_rate": 4.901618883413548e-07, "logits/chosen": -2.0795464515686035, "logits/rejected": -2.0845184326171875, "logps/chosen": -33.98798370361328, "logps/rejected": -36.569732666015625, "loss": 0.9992, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 0.0003018675488419831, "rewards/margins": 0.0008206713828258216, "rewards/rejected": -0.0005188033101148903, "step": 70 }, { "epoch": 0.21, "learning_rate": 4.828760511501322e-07, "logits/chosen": -1.9425585269927979, "logits/rejected": -1.945704698562622, "logps/chosen": -34.408721923828125, "logps/rejected": -34.581504821777344, "loss": 0.9951, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.011452903971076012, "rewards/margins": 0.0049109989777207375, "rewards/rejected": 0.006541903130710125, "step": 80 }, { "epoch": 0.23, "learning_rate": 4.736716601303429e-07, "logits/chosen": -1.9505088329315186, "logits/rejected": -1.9550243616104126, "logps/chosen": -32.451255798339844, "logps/rejected": -32.366790771484375, "loss": 0.9856, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 0.009697502478957176, "rewards/margins": 0.014425436034798622, "rewards/rejected": -0.0047279344871640205, "step": 90 }, { "epoch": 0.26, "learning_rate": 4.62624545834521e-07, "logits/chosen": -2.048947334289551, "logits/rejected": -2.046935796737671, "logps/chosen": -32.25363540649414, "logps/rejected": -31.277917861938477, "loss": 1.0007, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": -0.007680465932935476, "rewards/margins": -0.0006578800384886563, "rewards/rejected": -0.007022587116807699, "step": 100 }, { "epoch": 0.26, "eval_logits/chosen": -2.243957042694092, "eval_logits/rejected": -2.2390689849853516, "eval_logps/chosen": -34.019203186035156, "eval_logps/rejected": -37.50347900390625, "eval_loss": 0.998858630657196, "eval_rewards/accuracies": 0.5402824282646179, "eval_rewards/chosen": 0.009209612384438515, "eval_rewards/margins": 0.0013267018366605043, "eval_rewards/rejected": 0.00788290984928608, "eval_runtime": 145.9923, "eval_samples_per_second": 2.349, "eval_steps_per_second": 0.295, "step": 100 }, { "epoch": 0.29, "learning_rate": 4.4982572012636904e-07, "logits/chosen": -2.005302667617798, "logits/rejected": -2.002894878387451, "logps/chosen": -33.23944854736328, "logps/rejected": -34.005104064941406, "loss": 1.0119, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 0.002622986678034067, "rewards/margins": -0.01194014959037304, "rewards/rejected": 0.014563137665390968, "step": 110 }, { "epoch": 0.31, "learning_rate": 4.353806263777677e-07, "logits/chosen": -2.0167059898376465, "logits/rejected": -2.0083398818969727, "logps/chosen": -32.45713424682617, "logps/rejected": -32.17597198486328, "loss": 0.9987, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": -0.0073361145332455635, "rewards/margins": 0.0013455990701913834, "rewards/rejected": -0.008681714534759521, "step": 120 }, { "epoch": 0.34, "learning_rate": 4.194082707715275e-07, "logits/chosen": -2.0464558601379395, "logits/rejected": -2.038412094116211, "logps/chosen": -30.482311248779297, "logps/rejected": -32.04933166503906, "loss": 1.0062, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": -0.003956255037337542, "rewards/margins": -0.006162940990179777, "rewards/rejected": 0.002206685720011592, "step": 130 }, { "epoch": 0.36, "learning_rate": 4.020402418666621e-07, "logits/chosen": -1.9769861698150635, "logits/rejected": -1.9872652292251587, "logps/chosen": -31.415592193603516, "logps/rejected": -32.5562858581543, "loss": 0.9847, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.0024404728319495916, "rewards/margins": 0.015285758301615715, "rewards/rejected": -0.01284528523683548, "step": 140 }, { "epoch": 0.39, "learning_rate": 3.8341962650351185e-07, "logits/chosen": -1.890750527381897, "logits/rejected": -1.8918195962905884, "logps/chosen": -34.196632385253906, "logps/rejected": -34.74388885498047, "loss": 1.0104, "rewards/accuracies": 0.4000000059604645, "rewards/chosen": -0.002364098560065031, "rewards/margins": -0.0103833619505167, "rewards/rejected": 0.008019264787435532, "step": 150 }, { "epoch": 0.42, "learning_rate": 3.636998309800572e-07, "logits/chosen": -1.9424158334732056, "logits/rejected": -1.9389365911483765, "logps/chosen": -36.13062286376953, "logps/rejected": -32.720314025878906, "loss": 0.9768, "rewards/accuracies": 0.637499988079071, "rewards/chosen": 0.02463117241859436, "rewards/margins": 0.023165332153439522, "rewards/rejected": 0.0014658428262919188, "step": 160 }, { "epoch": 0.44, "learning_rate": 3.430433172111807e-07, "logits/chosen": -2.04191517829895, "logits/rejected": -2.034518003463745, "logps/chosen": -33.780067443847656, "logps/rejected": -31.35748863220215, "loss": 0.9859, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.010676576755940914, "rewards/margins": 0.014060018584132195, "rewards/rejected": -0.0033834422938525677, "step": 170 }, { "epoch": 0.47, "learning_rate": 3.216202642830543e-07, "logits/chosen": -2.0474770069122314, "logits/rejected": -2.0527572631835938, "logps/chosen": -32.51685333251953, "logps/rejected": -32.48542022705078, "loss": 0.9831, "rewards/accuracies": 0.5, "rewards/chosen": 0.01676076650619507, "rewards/margins": 0.016928378492593765, "rewards/rejected": -0.00016761067672632635, "step": 180 }, { "epoch": 0.49, "learning_rate": 2.9960716642946403e-07, "logits/chosen": -2.0486884117126465, "logits/rejected": -2.045897960662842, "logps/chosen": -31.490026473999023, "logps/rejected": -31.336299896240234, "loss": 0.9898, "rewards/accuracies": 0.5, "rewards/chosen": -0.0006444025784730911, "rewards/margins": 0.010206506587564945, "rewards/rejected": -0.010850909166038036, "step": 190 }, { "epoch": 0.52, "learning_rate": 2.771853789806683e-07, "logits/chosen": -1.9185329675674438, "logits/rejected": -1.9231984615325928, "logps/chosen": -31.579875946044922, "logps/rejected": -32.781455993652344, "loss": 0.9898, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.015846019610762596, "rewards/margins": 0.010161412879824638, "rewards/rejected": 0.005684606730937958, "step": 200 }, { "epoch": 0.52, "eval_logits/chosen": -2.2439582347869873, "eval_logits/rejected": -2.239088296890259, "eval_logps/chosen": -34.01380920410156, "eval_logps/rejected": -37.500911712646484, "eval_loss": 0.9969704151153564, "eval_rewards/accuracies": 0.5307309031486511, "eval_rewards/chosen": 0.01244510430842638, "eval_rewards/margins": 0.0030194728169590235, "eval_rewards/rejected": 0.009425631724298, "eval_runtime": 145.8589, "eval_samples_per_second": 2.352, "eval_steps_per_second": 0.295, "step": 200 }, { "epoch": 0.55, "learning_rate": 2.5453962426402e-07, "logits/chosen": -2.0318081378936768, "logits/rejected": -2.0424768924713135, "logps/chosen": -31.95905113220215, "logps/rejected": -33.87731170654297, "loss": 0.9839, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.007351105101406574, "rewards/margins": 0.01611880213022232, "rewards/rejected": -0.008767696097493172, "step": 210 }, { "epoch": 0.57, "learning_rate": 2.318564697655179e-07, "logits/chosen": -1.9253685474395752, "logits/rejected": -1.940233826637268, "logps/chosen": -30.08770179748535, "logps/rejected": -31.577301025390625, "loss": 0.9742, "rewards/accuracies": 0.612500011920929, "rewards/chosen": 0.014303619973361492, "rewards/margins": 0.025777745991945267, "rewards/rejected": -0.011474122293293476, "step": 220 }, { "epoch": 0.6, "learning_rate": 2.093227910899832e-07, "logits/chosen": -1.9832963943481445, "logits/rejected": -1.9872655868530273, "logps/chosen": -33.39131546020508, "logps/rejected": -31.548086166381836, "loss": 0.985, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.015694385394454002, "rewards/margins": 0.014964587986469269, "rewards/rejected": 0.0007297966512851417, "step": 230 }, { "epoch": 0.62, "learning_rate": 1.8712423238279356e-07, "logits/chosen": -1.9828859567642212, "logits/rejected": -1.9609191417694092, "logps/chosen": -34.16459655761719, "logps/rejected": -34.95134735107422, "loss": 1.0062, "rewards/accuracies": 0.4124999940395355, "rewards/chosen": -0.0065087564289569855, "rewards/margins": -0.006178082898259163, "rewards/rejected": -0.0003306727739982307, "step": 240 }, { "epoch": 0.65, "learning_rate": 1.654436768970182e-07, "logits/chosen": -2.0246593952178955, "logits/rejected": -2.0213539600372314, "logps/chosen": -32.9222412109375, "logps/rejected": -36.228416442871094, "loss": 0.9962, "rewards/accuracies": 0.4749999940395355, "rewards/chosen": 0.0016961356159299612, "rewards/margins": 0.0037924889475107193, "rewards/rejected": -0.0020963551942259073, "step": 250 }, { "epoch": 0.68, "learning_rate": 1.444597403062196e-07, "logits/chosen": -1.8918907642364502, "logits/rejected": -1.8894577026367188, "logps/chosen": -34.189327239990234, "logps/rejected": -35.505496978759766, "loss": 1.0, "rewards/accuracies": 0.5, "rewards/chosen": -0.0005569729837588966, "rewards/margins": -2.999706157424953e-05, "rewards/rejected": -0.0005269756657071412, "step": 260 }, { "epoch": 0.7, "learning_rate": 1.2434529917578887e-07, "logits/chosen": -1.8758213520050049, "logits/rejected": -1.8732786178588867, "logps/chosen": -34.38149642944336, "logps/rejected": -31.752059936523438, "loss": 1.0019, "rewards/accuracies": 0.550000011920929, "rewards/chosen": 0.009660609066486359, "rewards/margins": -0.0019017171580344439, "rewards/rejected": 0.011562327854335308, "step": 270 }, { "epoch": 0.73, "learning_rate": 1.0526606671603521e-07, "logits/chosen": -1.9798027276992798, "logits/rejected": -1.9691746234893799, "logps/chosen": -35.328853607177734, "logps/rejected": -31.868383407592773, "loss": 0.9669, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.016671547666192055, "rewards/margins": 0.03310702368617058, "rewards/rejected": -0.016435474157333374, "step": 280 }, { "epoch": 0.75, "learning_rate": 8.737922755071453e-08, "logits/chosen": -2.075648546218872, "logits/rejected": -2.0605978965759277, "logps/chosen": -30.926809310913086, "logps/rejected": -32.621070861816406, "loss": 1.0201, "rewards/accuracies": 0.4124999940395355, "rewards/chosen": -0.0016108205309137702, "rewards/margins": -0.020070534199476242, "rewards/rejected": 0.018459713086485863, "step": 290 }, { "epoch": 0.78, "learning_rate": 7.08321427484816e-08, "logits/chosen": -1.9466993808746338, "logits/rejected": -1.9441711902618408, "logps/chosen": -32.90770721435547, "logps/rejected": -30.82305335998535, "loss": 0.9846, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.011643724516034126, "rewards/margins": 0.015391023829579353, "rewards/rejected": -0.0037472969852387905, "step": 300 }, { "epoch": 0.78, "eval_logits/chosen": -2.243670701980591, "eval_logits/rejected": -2.2387943267822266, "eval_logps/chosen": -34.02626419067383, "eval_logps/rejected": -37.50523376464844, "eval_loss": 1.001994013786316, "eval_rewards/accuracies": 0.4634551703929901, "eval_rewards/chosen": 0.0049731116741895676, "eval_rewards/margins": -0.0018590801628306508, "eval_rewards/rejected": 0.006832191254943609, "eval_runtime": 145.8336, "eval_samples_per_second": 2.352, "eval_steps_per_second": 0.295, "step": 300 }, { "epoch": 0.81, "learning_rate": 5.576113578589034e-08, "logits/chosen": -1.9287983179092407, "logits/rejected": -1.9255239963531494, "logps/chosen": -31.573007583618164, "logps/rejected": -33.742637634277344, "loss": 0.9815, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": 0.018206708133220673, "rewards/margins": 0.018496429547667503, "rewards/rejected": -0.0002897246740758419, "step": 310 }, { "epoch": 0.83, "learning_rate": 4.229036944380912e-08, "logits/chosen": -1.9805524349212646, "logits/rejected": -1.9682422876358032, "logps/chosen": -34.57670211791992, "logps/rejected": -33.561363220214844, "loss": 0.9749, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.011624744161963463, "rewards/margins": 0.0251374039798975, "rewards/rejected": -0.013512656092643738, "step": 320 }, { "epoch": 0.86, "learning_rate": 3.053082288996112e-08, "logits/chosen": -2.016091823577881, "logits/rejected": -2.0146238803863525, "logps/chosen": -33.47222137451172, "logps/rejected": -32.46577835083008, "loss": 1.0001, "rewards/accuracies": 0.4749999940395355, "rewards/chosen": 0.008143280632793903, "rewards/margins": -0.0001393534184899181, "rewards/rejected": 0.008282631635665894, "step": 330 }, { "epoch": 0.88, "learning_rate": 2.05793773749158e-08, "logits/chosen": -2.1032156944274902, "logits/rejected": -2.0874197483062744, "logps/chosen": -34.174861907958984, "logps/rejected": -33.088890075683594, "loss": 1.0148, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.002716648392379284, "rewards/margins": -0.014771336689591408, "rewards/rejected": 0.017487986013293266, "step": 340 }, { "epoch": 0.91, "learning_rate": 1.251801807404168e-08, "logits/chosen": -1.9747329950332642, "logits/rejected": -1.9737837314605713, "logps/chosen": -33.25817108154297, "logps/rejected": -32.459861755371094, "loss": 0.9888, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 0.014356844127178192, "rewards/margins": 0.01118617132306099, "rewards/rejected": 0.003170671407133341, "step": 350 }, { "epoch": 0.94, "learning_rate": 6.41315865106129e-09, "logits/chosen": -1.9303538799285889, "logits/rejected": -1.94071364402771, "logps/chosen": -32.20769500732422, "logps/rejected": -35.273780822753906, "loss": 1.0108, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": -0.0018019669223576784, "rewards/margins": -0.010786842554807663, "rewards/rejected": 0.008984875865280628, "step": 360 }, { "epoch": 0.96, "learning_rate": 2.3150941078050324e-09, "logits/chosen": -2.069509744644165, "logits/rejected": -2.0629658699035645, "logps/chosen": -33.647117614746094, "logps/rejected": -29.217060089111328, "loss": 1.0027, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 0.002911838237196207, "rewards/margins": -0.0027081891894340515, "rewards/rejected": 0.005620025563985109, "step": 370 }, { "epoch": 0.99, "learning_rate": 2.575864278703266e-10, "logits/chosen": -1.929449439048767, "logits/rejected": -1.9315989017486572, "logps/chosen": -34.25099563598633, "logps/rejected": -30.906116485595703, "loss": 0.9871, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": -0.007165629416704178, "rewards/margins": 0.012927901931107044, "rewards/rejected": -0.020093530416488647, "step": 380 }, { "epoch": 1.0, "step": 385, "total_flos": 0.0, "train_loss": 0.9932638019710393, "train_runtime": 3254.1177, "train_samples_per_second": 0.946, "train_steps_per_second": 0.118 } ], "logging_steps": 10, "max_steps": 385, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 100, "total_flos": 0.0, "train_batch_size": 4, "trial_name": null, "trial_params": null }