{ "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.8663256168365479, "logits/rejected": -1.8706417083740234, "logps/chosen": -36.98673629760742, "logps/rejected": -33.66697311401367, "loss": 0.9994, "rewards/accuracies": 0.5, "rewards/chosen": 0.00019861094187945127, "rewards/margins": 0.0005675733555108309, "rewards/rejected": -0.0003689624136313796, "step": 10 }, { "epoch": 0.05, "learning_rate": 2.5641025641025636e-07, "logits/chosen": -1.997527837753296, "logits/rejected": -2.0001606941223145, "logps/chosen": -29.65478515625, "logps/rejected": -29.070556640625, "loss": 1.0001, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": -0.00012590530968736857, "rewards/margins": -5.477964805322699e-05, "rewards/rejected": -7.112567982403561e-05, "step": 20 }, { "epoch": 0.08, "learning_rate": 3.8461538461538463e-07, "logits/chosen": -1.9195702075958252, "logits/rejected": -1.9168812036514282, "logps/chosen": -31.421009063720703, "logps/rejected": -33.225616455078125, "loss": 1.0, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": -4.912544318358414e-05, "rewards/margins": 2.244002462248318e-05, "rewards/rejected": -7.15654605301097e-05, "step": 30 }, { "epoch": 0.1, "learning_rate": 4.999896948438433e-07, "logits/chosen": -2.0165441036224365, "logits/rejected": -2.0078024864196777, "logps/chosen": -32.57805252075195, "logps/rejected": -32.504085540771484, "loss": 1.0001, "rewards/accuracies": 0.4375, "rewards/chosen": -1.2443718333088327e-05, "rewards/margins": -9.829048212850466e-05, "rewards/rejected": 8.584677561884746e-05, "step": 40 }, { "epoch": 0.13, "learning_rate": 4.987541037542186e-07, "logits/chosen": -1.8641563653945923, "logits/rejected": -1.8533931970596313, "logps/chosen": -33.555824279785156, "logps/rejected": -35.42582321166992, "loss": 1.0002, "rewards/accuracies": 0.36250001192092896, "rewards/chosen": 1.6570575098739937e-05, "rewards/margins": -0.0002486651937942952, "rewards/rejected": 0.000265235808910802, "step": 50 }, { "epoch": 0.16, "learning_rate": 4.954691471941118e-07, "logits/chosen": -1.9452263116836548, "logits/rejected": -1.9471622705459595, "logps/chosen": -32.5916633605957, "logps/rejected": -33.1885871887207, "loss": 0.9997, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": 8.520439587300643e-05, "rewards/margins": 0.00025727329193614423, "rewards/rejected": -0.0001720689469948411, "step": 60 }, { "epoch": 0.18, "learning_rate": 4.901618883413548e-07, "logits/chosen": -2.0798425674438477, "logits/rejected": -2.08483624458313, "logps/chosen": -33.98626708984375, "logps/rejected": -36.57551956176758, "loss": 0.9999, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 2.217638393631205e-05, "rewards/margins": 8.864814299158752e-05, "rewards/rejected": -6.647173722740263e-05, "step": 70 }, { "epoch": 0.21, "learning_rate": 4.828760511501322e-07, "logits/chosen": -1.942161202430725, "logits/rejected": -1.9453051090240479, "logps/chosen": -34.400901794433594, "logps/rejected": -34.57304382324219, "loss": 0.9999, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.00026908880681730807, "rewards/margins": 7.544312393292785e-05, "rewards/rejected": 0.000193645668332465, "step": 80 }, { "epoch": 0.23, "learning_rate": 4.736716601303429e-07, "logits/chosen": -1.9505256414413452, "logits/rejected": -1.9550361633300781, "logps/chosen": -32.452125549316406, "logps/rejected": -32.35675811767578, "loss": 0.9999, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.00015296679339371622, "rewards/margins": 0.0001315164554398507, "rewards/rejected": 2.145032885891851e-05, "step": 90 }, { "epoch": 0.26, "learning_rate": 4.62624545834521e-07, "logits/chosen": -2.0490074157714844, "logits/rejected": -2.0470099449157715, "logps/chosen": -32.25472640991211, "logps/rejected": -31.2731990814209, "loss": 1.0001, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": -0.00013893461436964571, "rewards/margins": -6.904623296577483e-05, "rewards/rejected": -6.98883886798285e-05, "step": 100 }, { "epoch": 0.26, "eval_logits/chosen": -2.2441036701202393, "eval_logits/rejected": -2.239219903945923, "eval_logps/chosen": -34.02959442138672, "eval_logps/rejected": -37.50800704956055, "eval_loss": 1.0000391006469727, "eval_rewards/accuracies": 0.5099667906761169, "eval_rewards/chosen": 4.9558610044186935e-05, "eval_rewards/margins": -3.6566227208822966e-05, "eval_rewards/rejected": 8.61248336150311e-05, "eval_runtime": 145.9645, "eval_samples_per_second": 2.35, "eval_steps_per_second": 0.295, "step": 100 }, { "epoch": 0.29, "learning_rate": 4.4982572012636904e-07, "logits/chosen": -2.005286931991577, "logits/rejected": -2.0028929710388184, "logps/chosen": -33.22523498535156, "logps/rejected": -34.039817810058594, "loss": 0.9997, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.00018582009943202138, "rewards/margins": 0.0002903031127061695, "rewards/rejected": -0.00010448300599819049, "step": 110 }, { "epoch": 0.31, "learning_rate": 4.353806263777677e-07, "logits/chosen": -2.0164072513580322, "logits/rejected": -2.008033514022827, "logps/chosen": -32.473182678222656, "logps/rejected": -32.18474197387695, "loss": 1.0001, "rewards/accuracies": 0.4625000059604645, "rewards/chosen": -0.00028277214732952416, "rewards/margins": -5.0430990086169913e-05, "rewards/rejected": -0.00023234116088133305, "step": 120 }, { "epoch": 0.34, "learning_rate": 4.194082707715275e-07, "logits/chosen": -2.0463318824768066, "logits/rejected": -2.03826642036438, "logps/chosen": -30.479427337646484, "logps/rejected": -32.030479431152344, "loss": 1.0003, "rewards/accuracies": 0.4000000059604645, "rewards/chosen": -3.712148100021295e-05, "rewards/margins": -0.00026237856945954263, "rewards/rejected": 0.0002252570993732661, "step": 130 }, { "epoch": 0.36, "learning_rate": 4.020402418666621e-07, "logits/chosen": -1.9770317077636719, "logits/rejected": -1.987318754196167, "logps/chosen": -31.41189193725586, "logps/rejected": -32.552860260009766, "loss": 0.9997, "rewards/accuracies": 0.4625000059604645, "rewards/chosen": 7.772201206535101e-05, "rewards/margins": 0.000257515290286392, "rewards/rejected": -0.00017979330732487142, "step": 140 }, { "epoch": 0.39, "learning_rate": 3.8341962650351185e-07, "logits/chosen": -1.8904685974121094, "logits/rejected": -1.8915725946426392, "logps/chosen": -34.178749084472656, "logps/rejected": -34.7522087097168, "loss": 0.9999, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 0.00013941082579549402, "rewards/margins": 8.894065831555054e-05, "rewards/rejected": 5.047018566983752e-05, "step": 150 }, { "epoch": 0.42, "learning_rate": 3.636998309800572e-07, "logits/chosen": -1.9424184560775757, "logits/rejected": -1.9389402866363525, "logps/chosen": -36.14480209350586, "logps/rejected": -32.753517150878906, "loss": 0.9994, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.0002687961678020656, "rewards/margins": 0.0005764620145782828, "rewards/rejected": -0.0003076658467762172, "step": 160 }, { "epoch": 0.44, "learning_rate": 3.430433172111807e-07, "logits/chosen": -2.0420711040496826, "logits/rejected": -2.0346856117248535, "logps/chosen": -33.783363342285156, "logps/rejected": -31.349674224853516, "loss": 0.9999, "rewards/accuracies": 0.5249999761581421, "rewards/chosen": 0.00014495299546979368, "rewards/margins": 0.00012318804510869086, "rewards/rejected": 2.1765008568763733e-05, "step": 170 }, { "epoch": 0.47, "learning_rate": 3.216202642830543e-07, "logits/chosen": -2.0475425720214844, "logits/rejected": -2.052799940109253, "logps/chosen": -32.50774383544922, "logps/rejected": -32.48485565185547, "loss": 0.9996, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.00037043425254523754, "rewards/margins": 0.0003675749758258462, "rewards/rejected": 2.859340838767821e-06, "step": 180 }, { "epoch": 0.49, "learning_rate": 2.9960716642946403e-07, "logits/chosen": -2.0484375953674316, "logits/rejected": -2.045671224594116, "logps/chosen": -31.50165367126465, "logps/rejected": -31.33974266052246, "loss": 0.9999, "rewards/accuracies": 0.5625, "rewards/chosen": -0.00012700485240202397, "rewards/margins": 8.82898602867499e-05, "rewards/rejected": -0.00021529469813685864, "step": 190 }, { "epoch": 0.52, "learning_rate": 2.771853789806683e-07, "logits/chosen": -1.9182907342910767, "logits/rejected": -1.9229600429534912, "logps/chosen": -31.5924072265625, "logps/rejected": -32.794044494628906, "loss": 0.9998, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 0.00013877781748306006, "rewards/margins": 0.0001699000713415444, "rewards/rejected": -3.112227204837836e-05, "step": 200 }, { "epoch": 0.52, "eval_logits/chosen": -2.2440481185913086, "eval_logits/rejected": -2.2391700744628906, "eval_logps/chosen": -34.02214813232422, "eval_logps/rejected": -37.503997802734375, "eval_loss": 1.0000054836273193, "eval_rewards/accuracies": 0.5045680999755859, "eval_rewards/chosen": 0.00012402697757352144, "eval_rewards/margins": -2.209126932939398e-06, "eval_rewards/rejected": 0.00012623611837625504, "eval_runtime": 145.6161, "eval_samples_per_second": 2.356, "eval_steps_per_second": 0.295, "step": 200 }, { "epoch": 0.55, "learning_rate": 2.5453962426402e-07, "logits/chosen": -2.0312139987945557, "logits/rejected": -2.041886568069458, "logps/chosen": -31.938335418701172, "logps/rejected": -33.87850570678711, "loss": 0.9995, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.00032970920437946916, "rewards/margins": 0.0004878062754869461, "rewards/rejected": -0.00015809701289981604, "step": 210 }, { "epoch": 0.57, "learning_rate": 2.318564697655179e-07, "logits/chosen": -1.925606369972229, "logits/rejected": -1.940452218055725, "logps/chosen": -30.110549926757812, "logps/rejected": -31.573543548583984, "loss": 0.9998, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 9.872082955553196e-06, "rewards/margins": 0.00016354399849660695, "rewards/rejected": -0.00015367190644610673, "step": 220 }, { "epoch": 0.6, "learning_rate": 2.093227910899832e-07, "logits/chosen": -1.983096718788147, "logits/rejected": -1.9870779514312744, "logps/chosen": -33.37377166748047, "logps/rejected": -31.558422088623047, "loss": 0.9995, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.0004369509988464415, "rewards/margins": 0.0005281810299493372, "rewards/rejected": -9.123003837885335e-05, "step": 230 }, { "epoch": 0.62, "learning_rate": 1.8712423238279356e-07, "logits/chosen": -1.9832874536514282, "logits/rejected": -1.9613001346588135, "logps/chosen": -34.15814971923828, "logps/rejected": -34.962947845458984, "loss": 0.9999, "rewards/accuracies": 0.5, "rewards/chosen": -4.397642987896688e-05, "rewards/margins": 7.76300803408958e-05, "rewards/rejected": -0.00012160651385784149, "step": 240 }, { "epoch": 0.65, "learning_rate": 1.654436768970182e-07, "logits/chosen": -2.024568557739258, "logits/rejected": -2.021247386932373, "logps/chosen": -32.91596221923828, "logps/rejected": -36.22909164428711, "loss": 0.9999, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 9.10547751118429e-05, "rewards/margins": 0.00013272995420265943, "rewards/rejected": -4.1675208194646984e-05, "step": 250 }, { "epoch": 0.68, "learning_rate": 1.444597403062196e-07, "logits/chosen": -1.891326904296875, "logits/rejected": -1.8888810873031616, "logps/chosen": -34.165008544921875, "logps/rejected": -35.511131286621094, "loss": 0.9997, "rewards/accuracies": 0.550000011920929, "rewards/chosen": 0.00023390687420032918, "rewards/margins": 0.00029896467458456755, "rewards/rejected": -6.505780766019598e-05, "step": 260 }, { "epoch": 0.7, "learning_rate": 1.2434529917578887e-07, "logits/chosen": -1.8756563663482666, "logits/rejected": -1.87309992313385, "logps/chosen": -34.39522171020508, "logps/rejected": -31.7664852142334, "loss": 1.0, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 2.3789130864315666e-05, "rewards/margins": -2.4663497242727317e-05, "rewards/rejected": 4.8452602641191334e-05, "step": 270 }, { "epoch": 0.73, "learning_rate": 1.0526606671603521e-07, "logits/chosen": -1.9794845581054688, "logits/rejected": -1.9688661098480225, "logps/chosen": -35.318504333496094, "logps/rejected": -31.839763641357422, "loss": 0.9996, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 0.0003813380899373442, "rewards/margins": 0.0003690699231810868, "rewards/rejected": 1.2268178579688538e-05, "step": 280 }, { "epoch": 0.75, "learning_rate": 8.737922755071453e-08, "logits/chosen": -2.075482130050659, "logits/rejected": -2.060464859008789, "logps/chosen": -30.921621322631836, "logps/rejected": -32.64299774169922, "loss": 1.0001, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 2.502280767657794e-05, "rewards/margins": -6.332028715405613e-05, "rewards/rejected": 8.834306936478242e-05, "step": 290 }, { "epoch": 0.78, "learning_rate": 7.08321427484816e-08, "logits/chosen": -1.9463533163070679, "logits/rejected": -1.9438135623931885, "logps/chosen": -32.896881103515625, "logps/rejected": -30.816125869750977, "loss": 0.9997, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.0003022877499461174, "rewards/margins": 0.0002954433439299464, "rewards/rejected": 6.84439146425575e-06, "step": 300 }, { "epoch": 0.78, "eval_logits/chosen": -2.243636131286621, "eval_logits/rejected": -2.2387540340423584, "eval_logps/chosen": -34.02006530761719, "eval_logps/rejected": -37.50526428222656, "eval_loss": 0.9999700784683228, "eval_rewards/accuracies": 0.4987541437149048, "eval_rewards/chosen": 0.0001448561524739489, "eval_rewards/margins": 3.130395998596214e-05, "eval_rewards/rejected": 0.00011355219612596557, "eval_runtime": 145.6582, "eval_samples_per_second": 2.355, "eval_steps_per_second": 0.295, "step": 300 }, { "epoch": 0.81, "learning_rate": 5.576113578589034e-08, "logits/chosen": -1.9288015365600586, "logits/rejected": -1.9255383014678955, "logps/chosen": -31.57794189453125, "logps/rejected": -33.73435592651367, "loss": 0.9998, "rewards/accuracies": 0.574999988079071, "rewards/chosen": 0.000254074577242136, "rewards/margins": 0.00017612635565456003, "rewards/rejected": 7.794820703566074e-05, "step": 310 }, { "epoch": 0.83, "learning_rate": 4.229036944380912e-08, "logits/chosen": -1.9808727502822876, "logits/rejected": -1.9685646295547485, "logps/chosen": -34.54682159423828, "logps/rejected": -33.555274963378906, "loss": 0.9993, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": 0.0004924900131300092, "rewards/margins": 0.0006568533135578036, "rewards/rejected": -0.00016436330042779446, "step": 320 }, { "epoch": 0.86, "learning_rate": 3.053082288996112e-08, "logits/chosen": -2.0161168575286865, "logits/rejected": -2.0146450996398926, "logps/chosen": -33.46837615966797, "logps/rejected": -32.443458557128906, "loss": 1.0002, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 0.00017420158837921917, "rewards/margins": -0.00018708205607254058, "rewards/rejected": 0.00036128362989984453, "step": 330 }, { "epoch": 0.88, "learning_rate": 2.05793773749158e-08, "logits/chosen": -2.102954626083374, "logits/rejected": -2.087148666381836, "logps/chosen": -34.159751892089844, "logps/rejected": -33.093017578125, "loss": 1.0001, "rewards/accuracies": 0.512499988079071, "rewards/chosen": 0.00019638515368569642, "rewards/margins": -5.3783714974997565e-05, "rewards/rejected": 0.0002501688722986728, "step": 340 }, { "epoch": 0.91, "learning_rate": 1.251801807404168e-08, "logits/chosen": -1.974347710609436, "logits/rejected": -1.9733997583389282, "logps/chosen": -33.24498748779297, "logps/rejected": -32.44942855834961, "loss": 0.9998, "rewards/accuracies": 0.48750001192092896, "rewards/chosen": 0.00037107508978806436, "rewards/margins": 0.0002138734853360802, "rewards/rejected": 0.00015720158990006894, "step": 350 }, { "epoch": 0.94, "learning_rate": 6.41315865106129e-09, "logits/chosen": -1.9305756092071533, "logits/rejected": -1.9409431219100952, "logps/chosen": -32.20996856689453, "logps/rejected": -35.26984405517578, "loss": 1.0002, "rewards/accuracies": 0.4749999940395355, "rewards/chosen": -5.276153751765378e-05, "rewards/margins": -0.00024183676578104496, "rewards/rejected": 0.00018907521734945476, "step": 360 }, { "epoch": 0.96, "learning_rate": 2.3150941078050324e-09, "logits/chosen": -2.0696487426757812, "logits/rejected": -2.0630972385406494, "logps/chosen": -33.643394470214844, "logps/rejected": -29.242889404296875, "loss": 0.9997, "rewards/accuracies": 0.550000011920929, "rewards/chosen": 8.579110726714134e-05, "rewards/margins": 0.0002504353760741651, "rewards/rejected": -0.00016464429791085422, "step": 370 }, { "epoch": 0.99, "learning_rate": 2.575864278703266e-10, "logits/chosen": -1.9290872812271118, "logits/rejected": -1.9312543869018555, "logps/chosen": -34.22844696044922, "logps/rejected": -30.902618408203125, "loss": 0.9996, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.00010604139242786914, "rewards/margins": 0.0004059396160300821, "rewards/rejected": -0.00029989826725795865, "step": 380 }, { "epoch": 1.0, "step": 385, "total_flos": 0.0, "train_loss": 0.9998476387618425, "train_runtime": 3251.6099, "train_samples_per_second": 0.947, "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 }