{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9989071038251366, "eval_steps": 400, "global_step": 457, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01092896174863388, "grad_norm": 56.07717009412548, "learning_rate": 9.782608695652174e-08, "logits/chosen": -1.0135059356689453, "logits/rejected": -1.0081793069839478, "logps/chosen": -0.28040584921836853, "logps/rejected": -0.28580397367477417, "loss": 4.1045, "rewards/accuracies": 0.53125, "rewards/chosen": -2.804058313369751, "rewards/margins": 0.05398142337799072, "rewards/rejected": -2.8580400943756104, "semantic_entropy": 0.7516118884086609, "step": 5 }, { "epoch": 0.02185792349726776, "grad_norm": 72.03538939992451, "learning_rate": 1.9565217391304347e-07, "logits/chosen": -1.060633659362793, "logits/rejected": -1.009230375289917, "logps/chosen": -0.25705546140670776, "logps/rejected": -0.270809531211853, "loss": 4.0645, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -2.570554494857788, "rewards/margins": 0.13754098117351532, "rewards/rejected": -2.7080955505371094, "semantic_entropy": 0.709734320640564, "step": 10 }, { "epoch": 0.03278688524590164, "grad_norm": 60.80949908739426, "learning_rate": 2.9347826086956523e-07, "logits/chosen": -1.0096722841262817, "logits/rejected": -0.9631622433662415, "logps/chosen": -0.2670869827270508, "logps/rejected": -0.2734836935997009, "loss": 4.0535, "rewards/accuracies": 0.512499988079071, "rewards/chosen": -2.670869827270508, "rewards/margins": 0.06396698206663132, "rewards/rejected": -2.734837055206299, "semantic_entropy": 0.7273667454719543, "step": 15 }, { "epoch": 0.04371584699453552, "grad_norm": 72.87577064643818, "learning_rate": 3.9130434782608694e-07, "logits/chosen": -0.9465762376785278, "logits/rejected": -0.8959148526191711, "logps/chosen": -0.27229827642440796, "logps/rejected": -0.28428345918655396, "loss": 4.0997, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -2.7229831218719482, "rewards/margins": 0.1198514848947525, "rewards/rejected": -2.842834711074829, "semantic_entropy": 0.743862509727478, "step": 20 }, { "epoch": 0.0546448087431694, "grad_norm": 35.73788589652073, "learning_rate": 4.891304347826087e-07, "logits/chosen": -0.9495905637741089, "logits/rejected": -0.8757770657539368, "logps/chosen": -0.2737862467765808, "logps/rejected": -0.2946169972419739, "loss": 4.0619, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -2.7378625869750977, "rewards/margins": 0.2083074301481247, "rewards/rejected": -2.9461700916290283, "semantic_entropy": 0.7538469433784485, "step": 25 }, { "epoch": 0.06557377049180328, "grad_norm": 56.199447993934285, "learning_rate": 5.869565217391305e-07, "logits/chosen": -1.0543675422668457, "logits/rejected": -0.9891208410263062, "logps/chosen": -0.2664986550807953, "logps/rejected": -0.28541743755340576, "loss": 4.0445, "rewards/accuracies": 0.5375000238418579, "rewards/chosen": -2.6649868488311768, "rewards/margins": 0.18918730318546295, "rewards/rejected": -2.8541738986968994, "semantic_entropy": 0.7244771122932434, "step": 30 }, { "epoch": 0.07650273224043716, "grad_norm": 64.82660227272935, "learning_rate": 6.847826086956522e-07, "logits/chosen": -0.9974210858345032, "logits/rejected": -0.9332624673843384, "logps/chosen": -0.2572976052761078, "logps/rejected": -0.2904777228832245, "loss": 3.9891, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -2.5729756355285645, "rewards/margins": 0.33180171251296997, "rewards/rejected": -2.9047772884368896, "semantic_entropy": 0.7203728556632996, "step": 35 }, { "epoch": 0.08743169398907104, "grad_norm": 74.43759886551268, "learning_rate": 7.826086956521739e-07, "logits/chosen": -0.9570473432540894, "logits/rejected": -0.8978286981582642, "logps/chosen": -0.28474289178848267, "logps/rejected": -0.31194889545440674, "loss": 4.1025, "rewards/accuracies": 0.512499988079071, "rewards/chosen": -2.847428798675537, "rewards/margins": 0.27205973863601685, "rewards/rejected": -3.1194887161254883, "semantic_entropy": 0.7648034691810608, "step": 40 }, { "epoch": 0.09836065573770492, "grad_norm": 34.176663547437016, "learning_rate": 8.804347826086956e-07, "logits/chosen": -1.021121859550476, "logits/rejected": -0.9401710629463196, "logps/chosen": -0.2921110689640045, "logps/rejected": -0.33298516273498535, "loss": 4.0055, "rewards/accuracies": 0.518750011920929, "rewards/chosen": -2.9211106300354004, "rewards/margins": 0.4087409973144531, "rewards/rejected": -3.3298516273498535, "semantic_entropy": 0.7668289542198181, "step": 45 }, { "epoch": 0.1092896174863388, "grad_norm": 80.21226502338544, "learning_rate": 8.997896779678812e-07, "logits/chosen": -0.955331027507782, "logits/rejected": -0.8773272633552551, "logps/chosen": -0.3032563030719757, "logps/rejected": -0.3285720944404602, "loss": 4.0066, "rewards/accuracies": 0.4937500059604645, "rewards/chosen": -3.0325629711151123, "rewards/margins": 0.2531580626964569, "rewards/rejected": -3.2857213020324707, "semantic_entropy": 0.765364944934845, "step": 50 }, { "epoch": 0.12021857923497267, "grad_norm": 43.70683163065837, "learning_rate": 8.989355816502526e-07, "logits/chosen": -0.9907596707344055, "logits/rejected": -0.8809078931808472, "logps/chosen": -0.2901116907596588, "logps/rejected": -0.3407103419303894, "loss": 3.896, "rewards/accuracies": 0.581250011920929, "rewards/chosen": -2.9011168479919434, "rewards/margins": 0.5059866905212402, "rewards/rejected": -3.4071037769317627, "semantic_entropy": 0.7688600420951843, "step": 55 }, { "epoch": 0.13114754098360656, "grad_norm": 40.17800097293932, "learning_rate": 8.974258123725301e-07, "logits/chosen": -1.0390875339508057, "logits/rejected": -0.995086669921875, "logps/chosen": -0.2762259840965271, "logps/rejected": -0.34549298882484436, "loss": 3.8357, "rewards/accuracies": 0.612500011920929, "rewards/chosen": -2.7622599601745605, "rewards/margins": 0.6926698684692383, "rewards/rejected": -3.454929828643799, "semantic_entropy": 0.7664283514022827, "step": 60 }, { "epoch": 0.14207650273224043, "grad_norm": 59.513778490950706, "learning_rate": 8.952625751612967e-07, "logits/chosen": -1.0262492895126343, "logits/rejected": -0.9614121317863464, "logps/chosen": -0.324962317943573, "logps/rejected": -0.35770028829574585, "loss": 3.9978, "rewards/accuracies": 0.53125, "rewards/chosen": -3.2496230602264404, "rewards/margins": 0.3273797631263733, "rewards/rejected": -3.577003002166748, "semantic_entropy": 0.8130875825881958, "step": 65 }, { "epoch": 0.15300546448087432, "grad_norm": 207.7015350833011, "learning_rate": 8.924490294367533e-07, "logits/chosen": -0.9874998331069946, "logits/rejected": -0.9699292182922363, "logps/chosen": -0.31067711114883423, "logps/rejected": -0.364091157913208, "loss": 3.8828, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -3.1067709922790527, "rewards/margins": 0.534140408039093, "rewards/rejected": -3.64091157913208, "semantic_entropy": 0.7709770798683167, "step": 70 }, { "epoch": 0.16393442622950818, "grad_norm": 44.352990988039096, "learning_rate": 8.889892843983686e-07, "logits/chosen": -0.9768747091293335, "logits/rejected": -0.9624359011650085, "logps/chosen": -0.3188754916191101, "logps/rejected": -0.39044779539108276, "loss": 3.9066, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -3.1887545585632324, "rewards/margins": 0.715722918510437, "rewards/rejected": -3.904478073120117, "semantic_entropy": 0.8010842204093933, "step": 75 }, { "epoch": 0.17486338797814208, "grad_norm": 96.76821204915929, "learning_rate": 8.848883930233695e-07, "logits/chosen": -0.9784610867500305, "logits/rejected": -0.9192510843276978, "logps/chosen": -0.332768976688385, "logps/rejected": -0.36657267808914185, "loss": 3.8709, "rewards/accuracies": 0.606249988079071, "rewards/chosen": -3.3276901245117188, "rewards/margins": 0.3380367159843445, "rewards/rejected": -3.665726900100708, "semantic_entropy": 0.7843155860900879, "step": 80 }, { "epoch": 0.18579234972677597, "grad_norm": 75.68920123621399, "learning_rate": 8.801523446868397e-07, "logits/chosen": -0.9815126657485962, "logits/rejected": -0.9339221119880676, "logps/chosen": -0.3239663243293762, "logps/rejected": -0.4012156128883362, "loss": 3.832, "rewards/accuracies": 0.625, "rewards/chosen": -3.2396633625030518, "rewards/margins": 0.7724922895431519, "rewards/rejected": -4.012156009674072, "semantic_entropy": 0.8030881881713867, "step": 85 }, { "epoch": 0.19672131147540983, "grad_norm": 52.172085418459005, "learning_rate": 8.74788056414203e-07, "logits/chosen": -1.024124264717102, "logits/rejected": -0.9470185041427612, "logps/chosen": -0.3651575446128845, "logps/rejected": -0.40893810987472534, "loss": 3.8895, "rewards/accuracies": 0.5687500238418579, "rewards/chosen": -3.651575803756714, "rewards/margins": 0.43780526518821716, "rewards/rejected": -4.089381217956543, "semantic_entropy": 0.8523303270339966, "step": 90 }, { "epoch": 0.20765027322404372, "grad_norm": 55.856723424209804, "learning_rate": 8.688033627788689e-07, "logits/chosen": -0.9588603973388672, "logits/rejected": -0.9630708694458008, "logps/chosen": -0.35158681869506836, "logps/rejected": -0.3956569731235504, "loss": 3.7738, "rewards/accuracies": 0.574999988079071, "rewards/chosen": -3.5158684253692627, "rewards/margins": 0.4407011568546295, "rewards/rejected": -3.956569194793701, "semantic_entropy": 0.8362972140312195, "step": 95 }, { "epoch": 0.2185792349726776, "grad_norm": 133.22842224119972, "learning_rate": 8.622070044597933e-07, "logits/chosen": -0.9748640060424805, "logits/rejected": -0.929074764251709, "logps/chosen": -0.4111453890800476, "logps/rejected": -0.49381446838378906, "loss": 3.8025, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -4.111453533172607, "rewards/margins": 0.8266904950141907, "rewards/rejected": -4.938143730163574, "semantic_entropy": 0.8799806833267212, "step": 100 }, { "epoch": 0.22950819672131148, "grad_norm": 63.064394179489504, "learning_rate": 8.550086154756677e-07, "logits/chosen": -1.0247594118118286, "logits/rejected": -0.9960424304008484, "logps/chosen": -0.3660830855369568, "logps/rejected": -0.44689756631851196, "loss": 3.7174, "rewards/accuracies": 0.637499988079071, "rewards/chosen": -3.6608314514160156, "rewards/margins": 0.8081439733505249, "rewards/rejected": -4.468975067138672, "semantic_entropy": 0.8604310154914856, "step": 105 }, { "epoch": 0.24043715846994534, "grad_norm": 63.6155571920592, "learning_rate": 8.472187091143803e-07, "logits/chosen": -1.0404878854751587, "logits/rejected": -1.019997000694275, "logps/chosen": -0.40719980001449585, "logps/rejected": -0.5280216336250305, "loss": 3.7079, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": -4.07199764251709, "rewards/margins": 1.2082184553146362, "rewards/rejected": -5.280216217041016, "semantic_entropy": 0.903515636920929, "step": 110 }, { "epoch": 0.25136612021857924, "grad_norm": 68.78614353875135, "learning_rate": 8.388486625782994e-07, "logits/chosen": -1.0444709062576294, "logits/rejected": -1.0056098699569702, "logps/chosen": -0.40871691703796387, "logps/rejected": -0.5650209188461304, "loss": 3.6489, "rewards/accuracies": 0.7124999761581421, "rewards/chosen": -4.087169170379639, "rewards/margins": 1.563040018081665, "rewards/rejected": -5.650209426879883, "semantic_entropy": 0.8942676782608032, "step": 115 }, { "epoch": 0.26229508196721313, "grad_norm": 53.32792414437455, "learning_rate": 8.299107003678049e-07, "logits/chosen": -1.0611729621887207, "logits/rejected": -0.9919866323471069, "logps/chosen": -0.42750421166419983, "logps/rejected": -0.5232139825820923, "loss": 3.5791, "rewards/accuracies": 0.699999988079071, "rewards/chosen": -4.2750420570373535, "rewards/margins": 0.9570980072021484, "rewards/rejected": -5.232140064239502, "semantic_entropy": 0.9266929626464844, "step": 120 }, { "epoch": 0.273224043715847, "grad_norm": 121.11035694148153, "learning_rate": 8.204178764273363e-07, "logits/chosen": -0.9674042463302612, "logits/rejected": -0.9337356686592102, "logps/chosen": -0.4846254885196686, "logps/rejected": -0.6142443418502808, "loss": 3.5767, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -4.846255302429199, "rewards/margins": 1.2961883544921875, "rewards/rejected": -6.1424431800842285, "semantic_entropy": 0.92425936460495, "step": 125 }, { "epoch": 0.28415300546448086, "grad_norm": 64.53667052645014, "learning_rate": 8.103840550800328e-07, "logits/chosen": -0.9902731776237488, "logits/rejected": -0.987317681312561, "logps/chosen": -0.5054492950439453, "logps/rejected": -0.7366899251937866, "loss": 3.5275, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": -5.054492473602295, "rewards/margins": 2.3124070167541504, "rewards/rejected": -7.366899013519287, "semantic_entropy": 0.9140235185623169, "step": 130 }, { "epoch": 0.29508196721311475, "grad_norm": 59.7508648355179, "learning_rate": 7.998238907788109e-07, "logits/chosen": -1.0142452716827393, "logits/rejected": -0.9352253079414368, "logps/chosen": -0.5328418612480164, "logps/rejected": -0.72784823179245, "loss": 3.3724, "rewards/accuracies": 0.668749988079071, "rewards/chosen": -5.328418254852295, "rewards/margins": 1.9500631093978882, "rewards/rejected": -7.278481960296631, "semantic_entropy": 0.9628833532333374, "step": 135 }, { "epoch": 0.30601092896174864, "grad_norm": 78.9540175820449, "learning_rate": 7.887528067034528e-07, "logits/chosen": -1.017315149307251, "logits/rejected": -0.9636947512626648, "logps/chosen": -0.6016183495521545, "logps/rejected": -0.8618828058242798, "loss": 3.0483, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -6.016183853149414, "rewards/margins": 2.602644681930542, "rewards/rejected": -8.618828773498535, "semantic_entropy": 0.9850225448608398, "step": 140 }, { "epoch": 0.31693989071038253, "grad_norm": 99.62302257315461, "learning_rate": 7.77186972234965e-07, "logits/chosen": -0.9608656764030457, "logits/rejected": -0.9462074041366577, "logps/chosen": -0.6700846552848816, "logps/rejected": -0.8206523060798645, "loss": 3.2001, "rewards/accuracies": 0.6937500238418579, "rewards/chosen": -6.7008466720581055, "rewards/margins": 1.505676031112671, "rewards/rejected": -8.206521987915039, "semantic_entropy": 0.9759008288383484, "step": 145 }, { "epoch": 0.32786885245901637, "grad_norm": 89.43625982319897, "learning_rate": 7.651432793401065e-07, "logits/chosen": -1.010655164718628, "logits/rejected": -0.9779534339904785, "logps/chosen": -0.8374897241592407, "logps/rejected": -1.1644561290740967, "loss": 3.0851, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": -8.374897956848145, "rewards/margins": 3.269664764404297, "rewards/rejected": -11.644561767578125, "semantic_entropy": 0.9569616317749023, "step": 150 }, { "epoch": 0.33879781420765026, "grad_norm": 89.48638184015236, "learning_rate": 7.526393179005737e-07, "logits/chosen": -0.9409087896347046, "logits/rejected": -0.8860008120536804, "logps/chosen": -0.9283156394958496, "logps/rejected": -1.1858878135681152, "loss": 2.8999, "rewards/accuracies": 0.731249988079071, "rewards/chosen": -9.283157348632812, "rewards/margins": 2.575721263885498, "rewards/rejected": -11.858877182006836, "semantic_entropy": 0.8997370004653931, "step": 155 }, { "epoch": 0.34972677595628415, "grad_norm": 106.63607424638896, "learning_rate": 7.396933500228808e-07, "logits/chosen": -0.9566726684570312, "logits/rejected": -0.9040724039077759, "logps/chosen": -1.1487252712249756, "logps/rejected": -1.4483044147491455, "loss": 2.801, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -11.487253189086914, "rewards/margins": 2.9957919120788574, "rewards/rejected": -14.48304557800293, "semantic_entropy": 0.8422845005989075, "step": 160 }, { "epoch": 0.36065573770491804, "grad_norm": 173.29746360828278, "learning_rate": 7.263242833664484e-07, "logits/chosen": -0.911471962928772, "logits/rejected": -0.8458994030952454, "logps/chosen": -1.2985560894012451, "logps/rejected": -1.5788644552230835, "loss": 2.8804, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": -12.985560417175293, "rewards/margins": 2.8030834197998047, "rewards/rejected": -15.788644790649414, "semantic_entropy": 0.802132785320282, "step": 165 }, { "epoch": 0.37158469945355194, "grad_norm": 100.11131084617674, "learning_rate": 7.1255164352886e-07, "logits/chosen": -0.9127036929130554, "logits/rejected": -0.8471270799636841, "logps/chosen": -1.3799912929534912, "logps/rejected": -1.7123008966445923, "loss": 2.8315, "rewards/accuracies": 0.768750011920929, "rewards/chosen": -13.79991340637207, "rewards/margins": 3.3230953216552734, "rewards/rejected": -17.12300682067871, "semantic_entropy": 0.7457509636878967, "step": 170 }, { "epoch": 0.3825136612021858, "grad_norm": 111.52780471006011, "learning_rate": 6.98395545528617e-07, "logits/chosen": -0.9586197137832642, "logits/rejected": -0.9303746223449707, "logps/chosen": -1.5328519344329834, "logps/rejected": -1.8097922801971436, "loss": 2.5002, "rewards/accuracies": 0.6875, "rewards/chosen": -15.328518867492676, "rewards/margins": 2.769404411315918, "rewards/rejected": -18.097923278808594, "semantic_entropy": 0.6919700503349304, "step": 175 }, { "epoch": 0.39344262295081966, "grad_norm": 112.60716502155312, "learning_rate": 6.838766644270385e-07, "logits/chosen": -0.9389745593070984, "logits/rejected": -0.9307812452316284, "logps/chosen": -1.5702974796295166, "logps/rejected": -2.0943307876586914, "loss": 2.4165, "rewards/accuracies": 0.78125, "rewards/chosen": -15.702975273132324, "rewards/margins": 5.24033260345459, "rewards/rejected": -20.943307876586914, "semantic_entropy": 0.6626415252685547, "step": 180 }, { "epoch": 0.40437158469945356, "grad_norm": 178.966427933753, "learning_rate": 6.690162051322191e-07, "logits/chosen": -0.9021973609924316, "logits/rejected": -0.9323358535766602, "logps/chosen": -1.6979179382324219, "logps/rejected": -2.232943058013916, "loss": 2.2909, "rewards/accuracies": 0.84375, "rewards/chosen": -16.979177474975586, "rewards/margins": 5.350252151489258, "rewards/rejected": -22.329431533813477, "semantic_entropy": 0.6147797107696533, "step": 185 }, { "epoch": 0.41530054644808745, "grad_norm": 146.482478913679, "learning_rate": 6.538358714291389e-07, "logits/chosen": -0.9412842988967896, "logits/rejected": -0.8962854146957397, "logps/chosen": -1.7343536615371704, "logps/rejected": -2.2405014038085938, "loss": 2.3785, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -17.343536376953125, "rewards/margins": 5.061476707458496, "rewards/rejected": -22.405012130737305, "semantic_entropy": 0.6030288934707642, "step": 190 }, { "epoch": 0.4262295081967213, "grad_norm": 150.31719448352644, "learning_rate": 6.383578342811638e-07, "logits/chosen": -0.9521581530570984, "logits/rejected": -0.9250186085700989, "logps/chosen": -1.9811210632324219, "logps/rejected": -2.550245523452759, "loss": 2.3547, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -19.81121253967285, "rewards/margins": 5.691242694854736, "rewards/rejected": -25.502452850341797, "semantic_entropy": 0.5285295844078064, "step": 195 }, { "epoch": 0.4371584699453552, "grad_norm": 142.7731322709338, "learning_rate": 6.226046994492272e-07, "logits/chosen": -0.9036038517951965, "logits/rejected": -0.7988227605819702, "logps/chosen": -1.8621279001235962, "logps/rejected": -2.401458263397217, "loss": 2.1514, "rewards/accuracies": 0.78125, "rewards/chosen": -18.621280670166016, "rewards/margins": 5.393301486968994, "rewards/rejected": -24.01457977294922, "semantic_entropy": 0.5528938174247742, "step": 200 }, { "epoch": 0.44808743169398907, "grad_norm": 109.73924196519458, "learning_rate": 6.065994744759878e-07, "logits/chosen": -0.8110615015029907, "logits/rejected": -0.8245576024055481, "logps/chosen": -1.8828060626983643, "logps/rejected": -2.4408397674560547, "loss": 2.3483, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -18.828060150146484, "rewards/margins": 5.58033561706543, "rewards/rejected": -24.408397674560547, "semantic_entropy": 0.5373791456222534, "step": 205 }, { "epoch": 0.45901639344262296, "grad_norm": 168.41087843878026, "learning_rate": 5.903655350831845e-07, "logits/chosen": -0.8903123736381531, "logits/rejected": -0.8840487599372864, "logps/chosen": -1.9938600063323975, "logps/rejected": -2.660202980041504, "loss": 2.0737, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -19.938602447509766, "rewards/margins": 6.663429260253906, "rewards/rejected": -26.602031707763672, "semantic_entropy": 0.5236510634422302, "step": 210 }, { "epoch": 0.46994535519125685, "grad_norm": 180.98552250261446, "learning_rate": 5.73926591031261e-07, "logits/chosen": -0.8701471090316772, "logits/rejected": -0.7955431938171387, "logps/chosen": -2.0766396522521973, "logps/rejected": -2.737276077270508, "loss": 2.2802, "rewards/accuracies": 0.8125, "rewards/chosen": -20.76639747619629, "rewards/margins": 6.606365203857422, "rewards/rejected": -27.37276268005371, "semantic_entropy": 0.4769493639469147, "step": 215 }, { "epoch": 0.4808743169398907, "grad_norm": 113.13891029235478, "learning_rate": 5.573066514911273e-07, "logits/chosen": -0.9388381838798523, "logits/rejected": -0.916387677192688, "logps/chosen": -2.1589951515197754, "logps/rejected": -2.7171058654785156, "loss": 2.1685, "rewards/accuracies": 0.731249988079071, "rewards/chosen": -21.589946746826172, "rewards/margins": 5.581110954284668, "rewards/rejected": -27.171056747436523, "semantic_entropy": 0.46796584129333496, "step": 220 }, { "epoch": 0.4918032786885246, "grad_norm": 104.5369543115771, "learning_rate": 5.405299899786287e-07, "logits/chosen": -0.9614135026931763, "logits/rejected": -0.9081883430480957, "logps/chosen": -2.1882309913635254, "logps/rejected": -2.7919082641601562, "loss": 2.2561, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -21.882308959960938, "rewards/margins": 6.036771297454834, "rewards/rejected": -27.919086456298828, "semantic_entropy": 0.45887380838394165, "step": 225 }, { "epoch": 0.5027322404371585, "grad_norm": 122.89716590246043, "learning_rate": 5.236211089029395e-07, "logits/chosen": -0.8663997650146484, "logits/rejected": -0.874904453754425, "logps/chosen": -2.1786980628967285, "logps/rejected": -2.7758872509002686, "loss": 2.3111, "rewards/accuracies": 0.8125, "rewards/chosen": -21.78697967529297, "rewards/margins": 5.971897125244141, "rewards/rejected": -27.758874893188477, "semantic_entropy": 0.45373231172561646, "step": 230 }, { "epoch": 0.5136612021857924, "grad_norm": 163.16841178090678, "learning_rate": 5.066047037806549e-07, "logits/chosen": -0.8985625505447388, "logits/rejected": -0.8524802923202515, "logps/chosen": -2.2341809272766113, "logps/rejected": -2.762202262878418, "loss": 1.952, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -22.34181022644043, "rewards/margins": 5.280209541320801, "rewards/rejected": -27.622018814086914, "semantic_entropy": 0.4423500895500183, "step": 235 }, { "epoch": 0.5245901639344263, "grad_norm": 126.28807692275711, "learning_rate": 4.895056271678488e-07, "logits/chosen": -0.9690831303596497, "logits/rejected": -0.9802172780036926, "logps/chosen": -2.1863203048706055, "logps/rejected": -2.8334527015686035, "loss": 1.9608, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -21.863203048706055, "rewards/margins": 6.4713263511657715, "rewards/rejected": -28.33452796936035, "semantic_entropy": 0.4405677318572998, "step": 240 }, { "epoch": 0.5355191256830601, "grad_norm": 203.4242664926994, "learning_rate": 4.723488523627758e-07, "logits/chosen": -0.9145323038101196, "logits/rejected": -0.8553255796432495, "logps/chosen": -2.0377883911132812, "logps/rejected": -2.7260096073150635, "loss": 2.0939, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -20.377883911132812, "rewards/margins": 6.882212162017822, "rewards/rejected": -27.260095596313477, "semantic_entropy": 0.49542418122291565, "step": 245 }, { "epoch": 0.546448087431694, "grad_norm": 99.66072304681889, "learning_rate": 4.55159436932227e-07, "logits/chosen": -0.9824727773666382, "logits/rejected": -0.9378935694694519, "logps/chosen": -2.12608003616333, "logps/rejected": -2.6889073848724365, "loss": 2.0268, "rewards/accuracies": 0.78125, "rewards/chosen": -21.26080322265625, "rewards/margins": 5.628273010253906, "rewards/rejected": -26.889074325561523, "semantic_entropy": 0.46515989303588867, "step": 250 }, { "epoch": 0.5573770491803278, "grad_norm": 126.02875800766768, "learning_rate": 4.3796248611481193e-07, "logits/chosen": -0.982226550579071, "logits/rejected": -0.947576642036438, "logps/chosen": -2.2678322792053223, "logps/rejected": -2.873493194580078, "loss": 2.1411, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -22.67832374572754, "rewards/margins": 6.056609153747559, "rewards/rejected": -28.734933853149414, "semantic_entropy": 0.4328394830226898, "step": 255 }, { "epoch": 0.5683060109289617, "grad_norm": 97.8896168726136, "learning_rate": 4.2078311615461545e-07, "logits/chosen": -0.961346447467804, "logits/rejected": -0.9446791410446167, "logps/chosen": -2.171950578689575, "logps/rejected": -2.801100492477417, "loss": 1.9527, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -21.719501495361328, "rewards/margins": 6.291501998901367, "rewards/rejected": -28.011005401611328, "semantic_entropy": 0.45725923776626587, "step": 260 }, { "epoch": 0.5792349726775956, "grad_norm": 92.04238840519959, "learning_rate": 4.0364641761878056e-07, "logits/chosen": -0.940658688545227, "logits/rejected": -0.8965815305709839, "logps/chosen": -2.2573418617248535, "logps/rejected": -2.931023597717285, "loss": 1.9009, "rewards/accuracies": 0.8125, "rewards/chosen": -22.573415756225586, "rewards/margins": 6.736820220947266, "rewards/rejected": -29.31023597717285, "semantic_entropy": 0.4323546886444092, "step": 265 }, { "epoch": 0.5901639344262295, "grad_norm": 132.08002962791124, "learning_rate": 3.8657741875259324e-07, "logits/chosen": -0.9581838846206665, "logits/rejected": -0.9301847219467163, "logps/chosen": -2.3147079944610596, "logps/rejected": -3.0581116676330566, "loss": 1.7921, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -23.147079467773438, "rewards/margins": 7.4340338706970215, "rewards/rejected": -30.581111907958984, "semantic_entropy": 0.41289639472961426, "step": 270 }, { "epoch": 0.6010928961748634, "grad_norm": 114.39315473375268, "learning_rate": 3.6960104892558897e-07, "logits/chosen": -0.9345510601997375, "logits/rejected": -0.862471878528595, "logps/chosen": -2.36122989654541, "logps/rejected": -2.888824701309204, "loss": 2.0653, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -23.6122989654541, "rewards/margins": 5.275951385498047, "rewards/rejected": -28.88825035095215, "semantic_entropy": 0.40508565306663513, "step": 275 }, { "epoch": 0.6120218579234973, "grad_norm": 124.01203860773359, "learning_rate": 3.5274210222206623e-07, "logits/chosen": -0.9259498715400696, "logits/rejected": -0.9285569190979004, "logps/chosen": -2.254788637161255, "logps/rejected": -2.976097583770752, "loss": 2.0752, "rewards/accuracies": 0.78125, "rewards/chosen": -22.54788589477539, "rewards/margins": 7.213089942932129, "rewards/rejected": -29.760974884033203, "semantic_entropy": 0.44933634996414185, "step": 280 }, { "epoch": 0.6229508196721312, "grad_norm": 117.08336782700206, "learning_rate": 3.360252012291866e-07, "logits/chosen": -1.0664575099945068, "logits/rejected": -1.0288488864898682, "logps/chosen": -2.2899527549743652, "logps/rejected": -3.0293755531311035, "loss": 1.654, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -22.899526596069336, "rewards/margins": 7.394228458404541, "rewards/rejected": -30.29375648498535, "semantic_entropy": 0.4348418712615967, "step": 285 }, { "epoch": 0.6338797814207651, "grad_norm": 121.62024337688516, "learning_rate": 3.1947476107554687e-07, "logits/chosen": -0.9870282411575317, "logits/rejected": -0.9533792734146118, "logps/chosen": -2.296736240386963, "logps/rejected": -2.971813917160034, "loss": 1.9371, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -22.967363357543945, "rewards/margins": 6.750775337219238, "rewards/rejected": -29.7181396484375, "semantic_entropy": 0.4140281081199646, "step": 290 }, { "epoch": 0.644808743169399, "grad_norm": 104.51223192918087, "learning_rate": 3.0311495377274497e-07, "logits/chosen": -0.9511557817459106, "logits/rejected": -0.9528511166572571, "logps/chosen": -2.3220698833465576, "logps/rejected": -3.1148197650909424, "loss": 1.5373, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -23.2206974029541, "rewards/margins": 7.927499294281006, "rewards/rejected": -31.148197174072266, "semantic_entropy": 0.41551512479782104, "step": 295 }, { "epoch": 0.6557377049180327, "grad_norm": 128.04282767271587, "learning_rate": 2.8696967291202045e-07, "logits/chosen": -1.0152642726898193, "logits/rejected": -0.9992272257804871, "logps/chosen": -2.2516703605651855, "logps/rejected": -2.9054460525512695, "loss": 1.8182, "rewards/accuracies": 0.8062499761581421, "rewards/chosen": -22.51670265197754, "rewards/margins": 6.537758827209473, "rewards/rejected": -29.054462432861328, "semantic_entropy": 0.4529096186161041, "step": 300 }, { "epoch": 0.6666666666666666, "grad_norm": 130.0472804365796, "learning_rate": 2.710624987675287e-07, "logits/chosen": -0.9901336431503296, "logits/rejected": -0.9319698214530945, "logps/chosen": -2.382580518722534, "logps/rejected": -3.083923816680908, "loss": 1.8928, "rewards/accuracies": 0.8125, "rewards/chosen": -23.8258056640625, "rewards/margins": 7.013431549072266, "rewards/rejected": -30.8392333984375, "semantic_entropy": 0.4096023142337799, "step": 305 }, { "epoch": 0.6775956284153005, "grad_norm": 135.85405045383933, "learning_rate": 2.554166638572174e-07, "logits/chosen": -0.9127476811408997, "logits/rejected": -0.9295948147773743, "logps/chosen": -2.382692575454712, "logps/rejected": -3.132550001144409, "loss": 1.9167, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -23.826923370361328, "rewards/margins": 7.498571872711182, "rewards/rejected": -31.32549476623535, "semantic_entropy": 0.3914796710014343, "step": 310 }, { "epoch": 0.6885245901639344, "grad_norm": 128.2044181966222, "learning_rate": 2.4005501901160237e-07, "logits/chosen": -0.9594842791557312, "logits/rejected": -0.9629005193710327, "logps/chosen": -2.5815389156341553, "logps/rejected": -3.1138715744018555, "loss": 1.7964, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -25.815387725830078, "rewards/margins": 5.323328495025635, "rewards/rejected": -31.138717651367188, "semantic_entropy": 0.3702000677585602, "step": 315 }, { "epoch": 0.6994535519125683, "grad_norm": 119.46232342615077, "learning_rate": 2.250000000000001e-07, "logits/chosen": -1.0557501316070557, "logits/rejected": -0.9983032941818237, "logps/chosen": -2.5179853439331055, "logps/rejected": -3.2003746032714844, "loss": 1.7829, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": -25.179855346679688, "rewards/margins": 6.823893070220947, "rewards/rejected": -32.003746032714844, "semantic_entropy": 0.3816224932670593, "step": 320 }, { "epoch": 0.7103825136612022, "grad_norm": 168.54223657148586, "learning_rate": 2.1027359476295932e-07, "logits/chosen": -1.0253689289093018, "logits/rejected": -1.0082193613052368, "logps/chosen": -2.3298420906066895, "logps/rejected": -2.9450843334198, "loss": 1.7987, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": -23.29842185974121, "rewards/margins": 6.152422904968262, "rewards/rejected": -29.450841903686523, "semantic_entropy": 0.43097060918807983, "step": 325 }, { "epoch": 0.7213114754098361, "grad_norm": 116.75076161131324, "learning_rate": 1.9589731129874957e-07, "logits/chosen": -0.9325264096260071, "logits/rejected": -0.9063482284545898, "logps/chosen": -2.3737525939941406, "logps/rejected": -3.1893439292907715, "loss": 1.7401, "rewards/accuracies": 0.8500000238418579, "rewards/chosen": -23.73752212524414, "rewards/margins": 8.15591812133789, "rewards/rejected": -31.893442153930664, "semantic_entropy": 0.42511922121047974, "step": 330 }, { "epoch": 0.73224043715847, "grad_norm": 112.9807304911019, "learning_rate": 1.8189214625080817e-07, "logits/chosen": -0.9031323194503784, "logits/rejected": -0.8634198904037476, "logps/chosen": -2.4427285194396973, "logps/rejected": -3.1281769275665283, "loss": 1.7125, "rewards/accuracies": 0.824999988079071, "rewards/chosen": -24.427288055419922, "rewards/margins": 6.854477882385254, "rewards/rejected": -31.28176498413086, "semantic_entropy": 0.4148736000061035, "step": 335 }, { "epoch": 0.7431693989071039, "grad_norm": 144.29558828539845, "learning_rate": 1.6827855424202287e-07, "logits/chosen": -0.9443111419677734, "logits/rejected": -0.8933967351913452, "logps/chosen": -2.487252712249756, "logps/rejected": -3.187016248703003, "loss": 1.935, "rewards/accuracies": 0.800000011920929, "rewards/chosen": -24.872528076171875, "rewards/margins": 6.997633457183838, "rewards/rejected": -31.870159149169922, "semantic_entropy": 0.3925535976886749, "step": 340 }, { "epoch": 0.7540983606557377, "grad_norm": 124.60190547977228, "learning_rate": 1.550764180006395e-07, "logits/chosen": -1.0004165172576904, "logits/rejected": -0.9962259531021118, "logps/chosen": -2.543126106262207, "logps/rejected": -3.2367870807647705, "loss": 1.9263, "rewards/accuracies": 0.8125, "rewards/chosen": -25.43126106262207, "rewards/margins": 6.936606407165527, "rewards/rejected": -32.36787033081055, "semantic_entropy": 0.38035836815834045, "step": 345 }, { "epoch": 0.7650273224043715, "grad_norm": 113.81092904036393, "learning_rate": 1.4230501932142528e-07, "logits/chosen": -0.9861739277839661, "logits/rejected": -0.9617244601249695, "logps/chosen": -2.442134141921997, "logps/rejected": -3.0121092796325684, "loss": 1.8226, "rewards/accuracies": 0.84375, "rewards/chosen": -24.421342849731445, "rewards/margins": 5.6997480392456055, "rewards/rejected": -30.12109375, "semantic_entropy": 0.3996952772140503, "step": 350 }, { "epoch": 0.7759562841530054, "grad_norm": 95.58347606179626, "learning_rate": 1.2998301090449732e-07, "logits/chosen": -0.9443106651306152, "logits/rejected": -0.9029051065444946, "logps/chosen": -2.4110779762268066, "logps/rejected": -3.094087600708008, "loss": 1.757, "rewards/accuracies": 0.862500011920929, "rewards/chosen": -24.110782623291016, "rewards/margins": 6.8300933837890625, "rewards/rejected": -30.940876007080078, "semantic_entropy": 0.4104097783565521, "step": 355 }, { "epoch": 0.7868852459016393, "grad_norm": 113.47773960736876, "learning_rate": 1.181283891129494e-07, "logits/chosen": -0.8974916338920593, "logits/rejected": -0.9033212661743164, "logps/chosen": -2.495509386062622, "logps/rejected": -3.1737542152404785, "loss": 1.8246, "rewards/accuracies": 0.7875000238418579, "rewards/chosen": -24.955089569091797, "rewards/margins": 6.782454490661621, "rewards/rejected": -31.7375431060791, "semantic_entropy": 0.397562712430954, "step": 360 }, { "epoch": 0.7978142076502732, "grad_norm": 115.18767750795475, "learning_rate": 1.0675846768906093e-07, "logits/chosen": -0.997270405292511, "logits/rejected": -0.9995688199996948, "logps/chosen": -2.479579210281372, "logps/rejected": -3.164058208465576, "loss": 1.913, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -24.795791625976562, "rewards/margins": 6.84478759765625, "rewards/rejected": -31.640579223632812, "semantic_entropy": 0.38974660634994507, "step": 365 }, { "epoch": 0.8087431693989071, "grad_norm": 132.48592611233022, "learning_rate": 9.588985246747924e-08, "logits/chosen": -1.0140767097473145, "logits/rejected": -0.9754125475883484, "logps/chosen": -2.5327906608581543, "logps/rejected": -3.2368435859680176, "loss": 1.7973, "rewards/accuracies": 0.8125, "rewards/chosen": -25.32790184020996, "rewards/margins": 7.040532112121582, "rewards/rejected": -32.36843490600586, "semantic_entropy": 0.37877336144447327, "step": 370 }, { "epoch": 0.819672131147541, "grad_norm": 94.56211622988587, "learning_rate": 8.553841712230287e-08, "logits/chosen": -1.0174239873886108, "logits/rejected": -0.9506980180740356, "logps/chosen": -2.4886462688446045, "logps/rejected": -3.2155399322509766, "loss": 1.7098, "rewards/accuracies": 0.8374999761581421, "rewards/chosen": -24.88646125793457, "rewards/margins": 7.268939018249512, "rewards/rejected": -32.155399322509766, "semantic_entropy": 0.3927253186702728, "step": 375 }, { "epoch": 0.8306010928961749, "grad_norm": 112.32547052410033, "learning_rate": 7.571927998349143e-08, "logits/chosen": -0.878455638885498, "logits/rejected": -0.895898163318634, "logps/chosen": -2.6072468757629395, "logps/rejected": -3.277003765106201, "loss": 1.7202, "rewards/accuracies": 0.8687499761581421, "rewards/chosen": -26.072467803955078, "rewards/margins": 6.69757080078125, "rewards/rejected": -32.77003860473633, "semantic_entropy": 0.3782675564289093, "step": 380 }, { "epoch": 0.8415300546448088, "grad_norm": 124.71355945194576, "learning_rate": 6.644678195645878e-08, "logits/chosen": -0.9814226031303406, "logits/rejected": -0.9645193815231323, "logps/chosen": -2.661468982696533, "logps/rejected": -3.464935779571533, "loss": 2.1272, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -26.61469078063965, "rewards/margins": 8.034662246704102, "rewards/rejected": -34.649356842041016, "semantic_entropy": 0.34283363819122314, "step": 385 }, { "epoch": 0.8524590163934426, "grad_norm": 139.6935787868564, "learning_rate": 5.77344655770996e-08, "logits/chosen": -0.9859868884086609, "logits/rejected": -0.961321234703064, "logps/chosen": -2.5931482315063477, "logps/rejected": -3.4174599647521973, "loss": 1.7084, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -25.93148422241211, "rewards/margins": 8.243112564086914, "rewards/rejected": -34.174598693847656, "semantic_entropy": 0.35839056968688965, "step": 390 }, { "epoch": 0.8633879781420765, "grad_norm": 151.19395191677748, "learning_rate": 4.9595055232838666e-08, "logits/chosen": -0.9353212118148804, "logits/rejected": -0.9160464406013489, "logps/chosen": -2.742602825164795, "logps/rejected": -3.4214999675750732, "loss": 1.7141, "rewards/accuracies": 0.8062499761581421, "rewards/chosen": -27.426029205322266, "rewards/margins": 6.7889723777771, "rewards/rejected": -34.21500015258789, "semantic_entropy": 0.32845446467399597, "step": 395 }, { "epoch": 0.8743169398907104, "grad_norm": 132.6235448510247, "learning_rate": 4.204043857859128e-08, "logits/chosen": -0.9146641492843628, "logits/rejected": -0.8824892044067383, "logps/chosen": -2.5529472827911377, "logps/rejected": -3.324016571044922, "loss": 1.7533, "rewards/accuracies": 0.8812500238418579, "rewards/chosen": -25.52947235107422, "rewards/margins": 7.710694789886475, "rewards/rejected": -33.24016571044922, "semantic_entropy": 0.3849184811115265, "step": 400 }, { "epoch": 0.8743169398907104, "eval_logits/chosen": -1.217973232269287, "eval_logits/rejected": -1.1664073467254639, "eval_logps/chosen": -2.541196346282959, "eval_logps/rejected": -3.292675256729126, "eval_loss": 1.7650105953216553, "eval_rewards/accuracies": 0.8433734774589539, "eval_rewards/chosen": -25.411962509155273, "eval_rewards/margins": 7.514789581298828, "eval_rewards/rejected": -32.926753997802734, "eval_runtime": 37.4693, "eval_samples_per_second": 35.175, "eval_semantic_entropy": 0.371148943901062, "eval_steps_per_second": 2.215, "step": 400 }, { "epoch": 0.8852459016393442, "grad_norm": 144.66318541442615, "learning_rate": 3.508164917477564e-08, "logits/chosen": -0.9309503436088562, "logits/rejected": -0.8550406694412231, "logps/chosen": -2.522739887237549, "logps/rejected": -3.200605869293213, "loss": 1.8143, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -25.227399826049805, "rewards/margins": 6.778659820556641, "rewards/rejected": -32.00605773925781, "semantic_entropy": 0.38521987199783325, "step": 405 }, { "epoch": 0.8961748633879781, "grad_norm": 109.19303106749234, "learning_rate": 2.872885037273573e-08, "logits/chosen": -0.9444577097892761, "logits/rejected": -0.9762741923332214, "logps/chosen": -2.642904758453369, "logps/rejected": -3.4438624382019043, "loss": 1.7524, "rewards/accuracies": 0.8125, "rewards/chosen": -26.42904281616211, "rewards/margins": 8.0095796585083, "rewards/rejected": -34.438629150390625, "semantic_entropy": 0.35392093658447266, "step": 410 }, { "epoch": 0.907103825136612, "grad_norm": 140.70600085298142, "learning_rate": 2.2991320471108737e-08, "logits/chosen": -0.9723037481307983, "logits/rejected": -0.9694955945014954, "logps/chosen": -2.5781261920928955, "logps/rejected": -3.383686065673828, "loss": 2.0281, "rewards/accuracies": 0.893750011920929, "rewards/chosen": -25.781265258789062, "rewards/margins": 8.055593490600586, "rewards/rejected": -33.836856842041016, "semantic_entropy": 0.37105756998062134, "step": 415 }, { "epoch": 0.9180327868852459, "grad_norm": 114.03524937369127, "learning_rate": 1.787743916481737e-08, "logits/chosen": -0.9833318591117859, "logits/rejected": -0.9596291780471802, "logps/chosen": -2.6041696071624756, "logps/rejected": -3.4229087829589844, "loss": 1.6196, "rewards/accuracies": 0.84375, "rewards/chosen": -26.041698455810547, "rewards/margins": 8.187392234802246, "rewards/rejected": -34.229087829589844, "semantic_entropy": 0.34770655632019043, "step": 420 }, { "epoch": 0.9289617486338798, "grad_norm": 105.90948195293248, "learning_rate": 1.3394675306478253e-08, "logits/chosen": -0.9838679432868958, "logits/rejected": -0.9677348136901855, "logps/chosen": -2.711045265197754, "logps/rejected": -3.550402879714966, "loss": 1.8917, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -27.110454559326172, "rewards/margins": 8.393570899963379, "rewards/rejected": -35.5040283203125, "semantic_entropy": 0.33215636014938354, "step": 425 }, { "epoch": 0.9398907103825137, "grad_norm": 116.33746873521659, "learning_rate": 9.549575998100024e-09, "logits/chosen": -0.9154612421989441, "logits/rejected": -0.9118291139602661, "logps/chosen": -2.4197230339050293, "logps/rejected": -3.2603530883789062, "loss": 1.5031, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": -24.197229385375977, "rewards/margins": 8.406302452087402, "rewards/rejected": -32.60353088378906, "semantic_entropy": 0.39185765385627747, "step": 430 }, { "epoch": 0.9508196721311475, "grad_norm": 129.942185849395, "learning_rate": 6.3477570290039974e-09, "logits/chosen": -0.967807412147522, "logits/rejected": -0.9948493242263794, "logps/chosen": -2.624267101287842, "logps/rejected": -3.4724323749542236, "loss": 1.7083, "rewards/accuracies": 0.893750011920929, "rewards/chosen": -26.2426700592041, "rewards/margins": 8.481654167175293, "rewards/rejected": -34.72432327270508, "semantic_entropy": 0.3495063781738281, "step": 435 }, { "epoch": 0.9617486338797814, "grad_norm": 130.00917349864633, "learning_rate": 3.7938946739319575e-09, "logits/chosen": -0.9625166058540344, "logits/rejected": -0.9211773872375488, "logps/chosen": -2.5590291023254395, "logps/rejected": -3.1604175567626953, "loss": 1.7398, "rewards/accuracies": 0.7749999761581421, "rewards/chosen": -25.590290069580078, "rewards/margins": 6.013881683349609, "rewards/rejected": -31.604175567626953, "semantic_entropy": 0.3838638365268707, "step": 440 }, { "epoch": 0.9726775956284153, "grad_norm": 114.30411553161933, "learning_rate": 1.8917188633203684e-09, "logits/chosen": -0.9689086675643921, "logits/rejected": -0.9481277465820312, "logps/chosen": -2.5780527591705322, "logps/rejected": -3.3827826976776123, "loss": 1.7293, "rewards/accuracies": 0.831250011920929, "rewards/chosen": -25.780529022216797, "rewards/margins": 8.0472993850708, "rewards/rejected": -33.82782745361328, "semantic_entropy": 0.3513355553150177, "step": 445 }, { "epoch": 0.9836065573770492, "grad_norm": 141.94643359249721, "learning_rate": 6.440077357159901e-10, "logits/chosen": -0.8894301652908325, "logits/rejected": -0.8833738565444946, "logps/chosen": -2.633645534515381, "logps/rejected": -3.3513991832733154, "loss": 1.7421, "rewards/accuracies": 0.856249988079071, "rewards/chosen": -26.336456298828125, "rewards/margins": 7.177537441253662, "rewards/rejected": -33.51399230957031, "semantic_entropy": 0.34072810411453247, "step": 450 }, { "epoch": 0.994535519125683, "grad_norm": 119.0733411035781, "learning_rate": 5.258358028848442e-11, "logits/chosen": -0.9054166674613953, "logits/rejected": -0.8974553346633911, "logps/chosen": -2.7232465744018555, "logps/rejected": -3.295553684234619, "loss": 2.1021, "rewards/accuracies": 0.75, "rewards/chosen": -27.23246192932129, "rewards/margins": 5.723072052001953, "rewards/rejected": -32.955535888671875, "semantic_entropy": 0.34370288252830505, "step": 455 }, { "epoch": 0.9989071038251366, "step": 457, "total_flos": 0.0, "train_loss": 2.585477126766347, "train_runtime": 5989.402, "train_samples_per_second": 9.777, "train_steps_per_second": 0.076 } ], "logging_steps": 5, "max_steps": 457, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 1000000, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 0.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }