{ "best_metric": null, "best_model_checkpoint": null, "epoch": 3.0, "eval_steps": 50, "global_step": 201, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.07462686567164178, "grad_norm": 60.46895766728989, "learning_rate": 5e-07, "logits/chosen": -2.737339735031128, "logits/rejected": -2.717646837234497, "logps/chosen": -269.46966552734375, "logps/rejected": -218.519287109375, "loss": 0.6897, "rewards/accuracies": 0.33125001192092896, "rewards/chosen": 0.016166506335139275, "rewards/margins": 0.005314634181559086, "rewards/rejected": 0.010851874016225338, "step": 5 }, { "epoch": 0.14925373134328357, "grad_norm": 47.55077346828281, "learning_rate": 1e-06, "logits/chosen": -2.707444190979004, "logits/rejected": -2.6988959312438965, "logps/chosen": -259.7918395996094, "logps/rejected": -198.8335723876953, "loss": 0.6419, "rewards/accuracies": 0.668749988079071, "rewards/chosen": 0.5464529991149902, "rewards/margins": 0.22305770218372345, "rewards/rejected": 0.323395311832428, "step": 10 }, { "epoch": 0.22388059701492538, "grad_norm": 49.794624138997065, "learning_rate": 9.983100718730718e-07, "logits/chosen": -2.5312893390655518, "logits/rejected": -2.528350353240967, "logps/chosen": -250.9792022705078, "logps/rejected": -215.8843231201172, "loss": 0.6325, "rewards/accuracies": 0.668749988079071, "rewards/chosen": 1.2436325550079346, "rewards/margins": 0.6625381708145142, "rewards/rejected": 0.5810943841934204, "step": 15 }, { "epoch": 0.29850746268656714, "grad_norm": 45.89015113492354, "learning_rate": 9.932517109205849e-07, "logits/chosen": -2.345860719680786, "logits/rejected": -2.330219030380249, "logps/chosen": -261.6305236816406, "logps/rejected": -205.9855499267578, "loss": 0.5741, "rewards/accuracies": 0.731249988079071, "rewards/chosen": 1.5929079055786133, "rewards/margins": 0.922720730304718, "rewards/rejected": 0.6701873540878296, "step": 20 }, { "epoch": 0.373134328358209, "grad_norm": 63.53673896186562, "learning_rate": 9.848591102083375e-07, "logits/chosen": -2.2251970767974854, "logits/rejected": -2.210585832595825, "logps/chosen": -256.94158935546875, "logps/rejected": -198.90554809570312, "loss": 0.5846, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 1.4611073732376099, "rewards/margins": 0.925146758556366, "rewards/rejected": 0.5359605550765991, "step": 25 }, { "epoch": 0.44776119402985076, "grad_norm": 40.6998829965568, "learning_rate": 9.731890013043367e-07, "logits/chosen": -2.203108310699463, "logits/rejected": -2.188232898712158, "logps/chosen": -220.20474243164062, "logps/rejected": -174.1934356689453, "loss": 0.5755, "rewards/accuracies": 0.731249988079071, "rewards/chosen": 1.1598186492919922, "rewards/margins": 1.0439460277557373, "rewards/rejected": 0.11587252467870712, "step": 30 }, { "epoch": 0.5223880597014925, "grad_norm": 39.86530018922445, "learning_rate": 9.583202707897073e-07, "logits/chosen": -2.159541606903076, "logits/rejected": -2.146280288696289, "logps/chosen": -271.1134033203125, "logps/rejected": -222.50732421875, "loss": 0.5581, "rewards/accuracies": 0.78125, "rewards/chosen": 1.6402053833007812, "rewards/margins": 1.4621944427490234, "rewards/rejected": 0.1780107319355011, "step": 35 }, { "epoch": 0.5970149253731343, "grad_norm": 47.764041993597665, "learning_rate": 9.403534270080829e-07, "logits/chosen": -2.097519874572754, "logits/rejected": -2.0528969764709473, "logps/chosen": -246.8362274169922, "logps/rejected": -208.1300048828125, "loss": 0.5329, "rewards/accuracies": 0.668749988079071, "rewards/chosen": 1.0587269067764282, "rewards/margins": 1.0248701572418213, "rewards/rejected": 0.033856604248285294, "step": 40 }, { "epoch": 0.6716417910447762, "grad_norm": 33.33992129726437, "learning_rate": 9.19409920658098e-07, "logits/chosen": -2.1117985248565674, "logits/rejected": -2.0575757026672363, "logps/chosen": -238.47335815429688, "logps/rejected": -232.6268310546875, "loss": 0.5597, "rewards/accuracies": 0.706250011920929, "rewards/chosen": 1.020668387413025, "rewards/margins": 1.3137915134429932, "rewards/rejected": -0.2931232154369354, "step": 45 }, { "epoch": 0.746268656716418, "grad_norm": 36.75599018498795, "learning_rate": 8.956313238215823e-07, "logits/chosen": -2.192502021789551, "logits/rejected": -2.1830239295959473, "logps/chosen": -246.92733764648438, "logps/rejected": -219.84817504882812, "loss": 0.5242, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": 0.8840469121932983, "rewards/margins": 1.3445603847503662, "rewards/rejected": -0.4605133533477783, "step": 50 }, { "epoch": 0.746268656716418, "eval_logits/chosen": -2.2182116508483887, "eval_logits/rejected": -2.195751190185547, "eval_logps/chosen": -256.5732727050781, "eval_logps/rejected": -219.44544982910156, "eval_loss": 0.5641670823097229, "eval_rewards/accuracies": 0.7583333253860474, "eval_rewards/chosen": 0.8116810321807861, "eval_rewards/margins": 1.5396502017974854, "eval_rewards/rejected": -0.7279691696166992, "eval_runtime": 125.8473, "eval_samples_per_second": 15.098, "eval_steps_per_second": 0.238, "step": 50 }, { "epoch": 0.8208955223880597, "grad_norm": 33.66480221016764, "learning_rate": 8.691783729769873e-07, "logits/chosen": -2.252897024154663, "logits/rejected": -2.210509777069092, "logps/chosen": -273.36102294921875, "logps/rejected": -232.4573974609375, "loss": 0.5359, "rewards/accuracies": 0.793749988079071, "rewards/chosen": 1.0710394382476807, "rewards/margins": 1.8447446823120117, "rewards/rejected": -0.7737051248550415, "step": 55 }, { "epoch": 0.8955223880597015, "grad_norm": 42.22753576523594, "learning_rate": 8.402298824670029e-07, "logits/chosen": -2.2703890800476074, "logits/rejected": -2.252363443374634, "logps/chosen": -274.8329162597656, "logps/rejected": -254.45663452148438, "loss": 0.5811, "rewards/accuracies": 0.699999988079071, "rewards/chosen": 0.9601489901542664, "rewards/margins": 1.3747169971466064, "rewards/rejected": -0.4145679473876953, "step": 60 }, { "epoch": 0.9701492537313433, "grad_norm": 38.82616914526827, "learning_rate": 8.089815357650089e-07, "logits/chosen": -2.237731456756592, "logits/rejected": -2.245671510696411, "logps/chosen": -264.57989501953125, "logps/rejected": -241.2982940673828, "loss": 0.5166, "rewards/accuracies": 0.7562500238418579, "rewards/chosen": 1.0662319660186768, "rewards/margins": 1.8653961420059204, "rewards/rejected": -0.799164354801178, "step": 65 }, { "epoch": 1.044776119402985, "grad_norm": 21.744667598719886, "learning_rate": 7.756445627110522e-07, "logits/chosen": -2.2960398197174072, "logits/rejected": -2.2724010944366455, "logps/chosen": -264.32720947265625, "logps/rejected": -239.6345977783203, "loss": 0.346, "rewards/accuracies": 0.856249988079071, "rewards/chosen": 2.158931255340576, "rewards/margins": 2.908942222595215, "rewards/rejected": -0.7500110864639282, "step": 70 }, { "epoch": 1.1194029850746268, "grad_norm": 19.678947962162013, "learning_rate": 7.404443116588547e-07, "logits/chosen": -2.396239995956421, "logits/rejected": -2.362546443939209, "logps/chosen": -246.3358154296875, "logps/rejected": -195.57357788085938, "loss": 0.2283, "rewards/accuracies": 0.918749988079071, "rewards/chosen": 2.2872486114501953, "rewards/margins": 2.971306562423706, "rewards/rejected": -0.6840583086013794, "step": 75 }, { "epoch": 1.1940298507462686, "grad_norm": 20.2993371431914, "learning_rate": 7.036187261857288e-07, "logits/chosen": -2.4099860191345215, "logits/rejected": -2.384648084640503, "logps/chosen": -260.5028381347656, "logps/rejected": -223.2133331298828, "loss": 0.2312, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 2.834780693054199, "rewards/margins": 3.5776984691619873, "rewards/rejected": -0.7429176568984985, "step": 80 }, { "epoch": 1.2686567164179103, "grad_norm": 25.4573275034557, "learning_rate": 6.654167366624008e-07, "logits/chosen": -2.36556339263916, "logits/rejected": -2.349255084991455, "logps/chosen": -231.6350860595703, "logps/rejected": -221.1610870361328, "loss": 0.2307, "rewards/accuracies": 0.9437500238418579, "rewards/chosen": 1.9625335931777954, "rewards/margins": 3.2484028339385986, "rewards/rejected": -1.2858693599700928, "step": 85 }, { "epoch": 1.3432835820895521, "grad_norm": 29.79365948091661, "learning_rate": 6.260965775552713e-07, "logits/chosen": -2.374537944793701, "logits/rejected": -2.348567485809326, "logps/chosen": -247.57614135742188, "logps/rejected": -238.37899780273438, "loss": 0.2523, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 1.934003233909607, "rewards/margins": 3.8235995769500732, "rewards/rejected": -1.8895965814590454, "step": 90 }, { "epoch": 1.417910447761194, "grad_norm": 22.109779497676236, "learning_rate": 5.859240418356614e-07, "logits/chosen": -2.3896210193634033, "logits/rejected": -2.3736777305603027, "logps/chosen": -257.51898193359375, "logps/rejected": -228.7509765625, "loss": 0.2453, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 2.010754108428955, "rewards/margins": 3.4373581409454346, "rewards/rejected": -1.4266036748886108, "step": 95 }, { "epoch": 1.4925373134328357, "grad_norm": 18.59821752381129, "learning_rate": 5.451706842957421e-07, "logits/chosen": -2.3828063011169434, "logits/rejected": -2.3802871704101562, "logps/chosen": -238.89547729492188, "logps/rejected": -224.2349395751953, "loss": 0.2523, "rewards/accuracies": 0.918749988079071, "rewards/chosen": 2.049046039581299, "rewards/margins": 3.4148106575012207, "rewards/rejected": -1.3657643795013428, "step": 100 }, { "epoch": 1.4925373134328357, "eval_logits/chosen": -2.3702290058135986, "eval_logits/rejected": -2.352924108505249, "eval_logps/chosen": -250.75942993164062, "eval_logps/rejected": -224.2054901123047, "eval_loss": 0.5878031253814697, "eval_rewards/accuracies": 0.8041666746139526, "eval_rewards/chosen": 1.393066644668579, "eval_rewards/margins": 2.597040891647339, "eval_rewards/rejected": -1.2039742469787598, "eval_runtime": 125.6186, "eval_samples_per_second": 15.125, "eval_steps_per_second": 0.239, "step": 100 }, { "epoch": 1.5671641791044775, "grad_norm": 24.221807777872822, "learning_rate": 5.041119859162068e-07, "logits/chosen": -2.3542027473449707, "logits/rejected": -2.3176567554473877, "logps/chosen": -224.2234649658203, "logps/rejected": -210.6978302001953, "loss": 0.3, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 1.6206858158111572, "rewards/margins": 3.042532444000244, "rewards/rejected": -1.4218467473983765, "step": 105 }, { "epoch": 1.6417910447761193, "grad_norm": 21.221389216973705, "learning_rate": 4.630254916940423e-07, "logits/chosen": -2.3093810081481934, "logits/rejected": -2.300978660583496, "logps/chosen": -242.2421875, "logps/rejected": -230.2930145263672, "loss": 0.2471, "rewards/accuracies": 0.9312499761581421, "rewards/chosen": 1.8961149454116821, "rewards/margins": 3.936471462249756, "rewards/rejected": -2.0403571128845215, "step": 110 }, { "epoch": 1.716417910447761, "grad_norm": 28.90188297206431, "learning_rate": 4.2218893451814e-07, "logits/chosen": -2.299715042114258, "logits/rejected": -2.3139047622680664, "logps/chosen": -260.2466735839844, "logps/rejected": -238.0985870361328, "loss": 0.2589, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": 1.7004687786102295, "rewards/margins": 3.8741843700408936, "rewards/rejected": -2.173715353012085, "step": 115 }, { "epoch": 1.7910447761194028, "grad_norm": 25.74215758531903, "learning_rate": 3.8187835777481375e-07, "logits/chosen": -2.292194366455078, "logits/rejected": -2.290935754776001, "logps/chosen": -252.3697509765625, "logps/rejected": -246.54690551757812, "loss": 0.2766, "rewards/accuracies": 0.9312499761581421, "rewards/chosen": 1.6656386852264404, "rewards/margins": 3.7279200553894043, "rewards/rejected": -2.062281608581543, "step": 120 }, { "epoch": 1.8656716417910446, "grad_norm": 22.13418877428713, "learning_rate": 3.423662493738687e-07, "logits/chosen": -2.2597708702087402, "logits/rejected": -2.22979474067688, "logps/chosen": -242.34994506835938, "logps/rejected": -205.6407470703125, "loss": 0.2627, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": 1.9104535579681396, "rewards/margins": 3.547952175140381, "rewards/rejected": -1.6374982595443726, "step": 125 }, { "epoch": 1.9402985074626866, "grad_norm": 25.182255638442037, "learning_rate": 3.039196998086687e-07, "logits/chosen": -2.216444730758667, "logits/rejected": -2.2051291465759277, "logps/chosen": -240.48526000976562, "logps/rejected": -241.6737518310547, "loss": 0.278, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 2.2690188884735107, "rewards/margins": 3.9218125343322754, "rewards/rejected": -1.6527936458587646, "step": 130 }, { "epoch": 2.014925373134328, "grad_norm": 17.639179913168814, "learning_rate": 2.667985967011878e-07, "logits/chosen": -2.2279441356658936, "logits/rejected": -2.202115297317505, "logps/chosen": -256.98162841796875, "logps/rejected": -240.4128875732422, "loss": 0.2636, "rewards/accuracies": 0.90625, "rewards/chosen": 2.024724245071411, "rewards/margins": 3.793694257736206, "rewards/rejected": -1.7689695358276367, "step": 135 }, { "epoch": 2.08955223880597, "grad_norm": 16.436781206837193, "learning_rate": 2.3125386803640183e-07, "logits/chosen": -2.208228349685669, "logits/rejected": -2.1990818977355957, "logps/chosen": -241.15542602539062, "logps/rejected": -233.29031372070312, "loss": 0.1344, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 1.9513914585113525, "rewards/margins": 3.8950798511505127, "rewards/rejected": -1.9436886310577393, "step": 140 }, { "epoch": 2.1641791044776117, "grad_norm": 17.047112239762754, "learning_rate": 1.9752578596124952e-07, "logits/chosen": -2.233797550201416, "logits/rejected": -2.2078850269317627, "logps/chosen": -252.17111206054688, "logps/rejected": -218.3225555419922, "loss": 0.1554, "rewards/accuracies": 0.9312499761581421, "rewards/chosen": 2.0858609676361084, "rewards/margins": 3.9756836891174316, "rewards/rejected": -1.8898223638534546, "step": 145 }, { "epoch": 2.2388059701492535, "grad_norm": 14.963330285006984, "learning_rate": 1.6584234261399532e-07, "logits/chosen": -2.2354073524475098, "logits/rejected": -2.238706588745117, "logps/chosen": -226.289306640625, "logps/rejected": -246.685791015625, "loss": 0.1264, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 2.1262850761413574, "rewards/margins": 4.3760247230529785, "rewards/rejected": -2.249739408493042, "step": 150 }, { "epoch": 2.2388059701492535, "eval_logits/chosen": -2.265167713165283, "eval_logits/rejected": -2.2381813526153564, "eval_logps/chosen": -251.42041015625, "eval_logps/rejected": -226.3953399658203, "eval_loss": 0.5601586103439331, "eval_rewards/accuracies": 0.8166666626930237, "eval_rewards/chosen": 1.326967477798462, "eval_rewards/margins": 2.749927282333374, "eval_rewards/rejected": -1.422959566116333, "eval_runtime": 125.5687, "eval_samples_per_second": 15.131, "eval_steps_per_second": 0.239, "step": 150 }, { "epoch": 2.3134328358208958, "grad_norm": 17.27452729186339, "learning_rate": 1.3641770896292082e-07, "logits/chosen": -2.2842135429382324, "logits/rejected": -2.229306697845459, "logps/chosen": -242.30044555664062, "logps/rejected": -227.74948120117188, "loss": 0.1578, "rewards/accuracies": 0.981249988079071, "rewards/chosen": 2.336923360824585, "rewards/margins": 4.076951026916504, "rewards/rejected": -1.7400274276733398, "step": 155 }, { "epoch": 2.388059701492537, "grad_norm": 19.880826026501197, "learning_rate": 1.0945078707215221e-07, "logits/chosen": -2.263514757156372, "logits/rejected": -2.2274537086486816, "logps/chosen": -247.376220703125, "logps/rejected": -231.4269256591797, "loss": 0.1686, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 2.331441640853882, "rewards/margins": 3.9185550212860107, "rewards/rejected": -1.5871132612228394, "step": 160 }, { "epoch": 2.4626865671641793, "grad_norm": 18.037005682288928, "learning_rate": 8.512386558088919e-08, "logits/chosen": -2.2772862911224365, "logits/rejected": -2.248198986053467, "logps/chosen": -241.83132934570312, "logps/rejected": -266.8996887207031, "loss": 0.168, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 2.4042108058929443, "rewards/margins": 4.915772914886475, "rewards/rejected": -2.511561870574951, "step": 165 }, { "epoch": 2.5373134328358207, "grad_norm": 19.957029846976177, "learning_rate": 6.360138748461013e-08, "logits/chosen": -2.2610926628112793, "logits/rejected": -2.2388763427734375, "logps/chosen": -255.074951171875, "logps/rejected": -226.93997192382812, "loss": 0.1771, "rewards/accuracies": 0.9437500238418579, "rewards/chosen": 2.393348455429077, "rewards/margins": 4.332278251647949, "rewards/rejected": -1.9389299154281616, "step": 170 }, { "epoch": 2.611940298507463, "grad_norm": 15.104058742033901, "learning_rate": 4.5028838547699346e-08, "logits/chosen": -2.271852970123291, "logits/rejected": -2.2440991401672363, "logps/chosen": -242.74697875976562, "logps/rejected": -235.72128295898438, "loss": 0.1392, "rewards/accuracies": 0.9375, "rewards/chosen": 2.28426456451416, "rewards/margins": 4.252217769622803, "rewards/rejected": -1.9679529666900635, "step": 175 }, { "epoch": 2.6865671641791042, "grad_norm": 14.425654092236623, "learning_rate": 2.9531763861505964e-08, "logits/chosen": -2.2854719161987305, "logits/rejected": -2.243256092071533, "logps/chosen": -251.6754150390625, "logps/rejected": -246.1281280517578, "loss": 0.1574, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 2.5320801734924316, "rewards/margins": 4.495547771453857, "rewards/rejected": -1.9634678363800049, "step": 180 }, { "epoch": 2.7611940298507465, "grad_norm": 18.97471020277948, "learning_rate": 1.7214919195619125e-08, "logits/chosen": -2.2592127323150635, "logits/rejected": -2.2206928730010986, "logps/chosen": -236.07418823242188, "logps/rejected": -231.7993621826172, "loss": 0.1479, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 1.6844587326049805, "rewards/margins": 4.147632122039795, "rewards/rejected": -2.4631736278533936, "step": 185 }, { "epoch": 2.835820895522388, "grad_norm": 19.094777076672973, "learning_rate": 8.161562878982398e-09, "logits/chosen": -2.2626476287841797, "logits/rejected": -2.2289066314697266, "logps/chosen": -253.0904083251953, "logps/rejected": -223.6345672607422, "loss": 0.1273, "rewards/accuracies": 0.96875, "rewards/chosen": 2.360367774963379, "rewards/margins": 4.313665390014648, "rewards/rejected": -1.9532979726791382, "step": 190 }, { "epoch": 2.91044776119403, "grad_norm": 18.606332473099318, "learning_rate": 2.432892997526026e-09, "logits/chosen": -2.2822461128234863, "logits/rejected": -2.233109951019287, "logps/chosen": -226.60153198242188, "logps/rejected": -230.42422485351562, "loss": 0.1154, "rewards/accuracies": 0.9312499761581421, "rewards/chosen": 2.1634068489074707, "rewards/margins": 4.176701545715332, "rewards/rejected": -2.013294219970703, "step": 195 }, { "epoch": 2.9850746268656714, "grad_norm": 17.32503927255026, "learning_rate": 6.763371270035457e-11, "logits/chosen": -2.2751100063323975, "logits/rejected": -2.2504701614379883, "logps/chosen": -249.61959838867188, "logps/rejected": -224.1602020263672, "loss": 0.1301, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 2.2563796043395996, "rewards/margins": 4.354907512664795, "rewards/rejected": -2.0985279083251953, "step": 200 }, { "epoch": 2.9850746268656714, "eval_logits/chosen": -2.265265464782715, "eval_logits/rejected": -2.235792398452759, "eval_logps/chosen": -253.1015625, "eval_logps/rejected": -228.8603515625, "eval_loss": 0.5625271201133728, "eval_rewards/accuracies": 0.8208333253860474, "eval_rewards/chosen": 1.1588523387908936, "eval_rewards/margins": 2.8283116817474365, "eval_rewards/rejected": -1.6694589853286743, "eval_runtime": 125.4908, "eval_samples_per_second": 15.141, "eval_steps_per_second": 0.239, "step": 200 }, { "epoch": 3.0, "step": 201, "total_flos": 2369906314051584.0, "train_loss": 0.3258402518075497, "train_runtime": 7367.4174, "train_samples_per_second": 6.961, "train_steps_per_second": 0.027 } ], "logging_steps": 5, "max_steps": 201, "num_input_tokens_seen": 0, "num_train_epochs": 3, "save_steps": 100, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 2369906314051584.0, "train_batch_size": 8, "trial_name": null, "trial_params": null }