diff --git "a/trainer_state.json" "b/trainer_state.json"
--- "a/trainer_state.json"
+++ "b/trainer_state.json"
@@ -10,7 +10,7 @@
   "log_history": [
     {
       "epoch": 0.0010465724751439038,
-      "grad_norm": 13.052067067299667,
+      "grad_norm": 13.051470496546411,
       "learning_rate": 5.208333333333333e-09,
       "logits/chosen": -2.8506698608398438,
       "logits/rejected": -2.73948335647583,
@@ -25,1581 +25,1581 @@
     },
     {
       "epoch": 0.010465724751439037,
-      "grad_norm": 9.928706072255864,
+      "grad_norm": 9.930470236754596,
       "learning_rate": 5.208333333333333e-08,
-      "logits/chosen": -2.4740195274353027,
-      "logits/rejected": -2.444406270980835,
-      "logps/chosen": -266.9622802734375,
-      "logps/rejected": -241.96226501464844,
-      "loss": 0.693,
-      "rewards/accuracies": 0.4236111044883728,
-      "rewards/chosen": -0.00022456223086919636,
-      "rewards/margins": 0.00035724174813367426,
-      "rewards/rejected": -0.0005818039644509554,
+      "logits/chosen": -2.473865270614624,
+      "logits/rejected": -2.444345474243164,
+      "logps/chosen": -267.0189208984375,
+      "logps/rejected": -241.91099548339844,
+      "loss": 0.6931,
+      "rewards/accuracies": 0.3402777910232544,
+      "rewards/chosen": -0.0007909854757599533,
+      "rewards/margins": -0.0007218287792056799,
+      "rewards/rejected": -6.915684207342565e-05,
       "step": 10
     },
     {
       "epoch": 0.020931449502878074,
-      "grad_norm": 11.836447602814966,
+      "grad_norm": 11.896085597664555,
       "learning_rate": 1.0416666666666667e-07,
-      "logits/chosen": -2.4786019325256348,
-      "logits/rejected": -2.4329116344451904,
-      "logps/chosen": -291.01934814453125,
-      "logps/rejected": -289.3887939453125,
+      "logits/chosen": -2.4787793159484863,
+      "logits/rejected": -2.4331250190734863,
+      "logps/chosen": -290.9656677246094,
+      "logps/rejected": -289.4096984863281,
       "loss": 0.6931,
-      "rewards/accuracies": 0.48124998807907104,
-      "rewards/chosen": -0.00017613262753002346,
-      "rewards/margins": -0.0004834429419133812,
-      "rewards/rejected": 0.0003073102852795273,
+      "rewards/accuracies": 0.518750011920929,
+      "rewards/chosen": 0.00036076834658160806,
+      "rewards/margins": 0.0002623483305796981,
+      "rewards/rejected": 9.842005238169804e-05,
       "step": 20
     },
     {
       "epoch": 0.03139717425431711,
-      "grad_norm": 11.43939229787168,
+      "grad_norm": 11.466941852693491,
       "learning_rate": 1.5624999999999999e-07,
-      "logits/chosen": -2.5795392990112305,
-      "logits/rejected": -2.5510947704315186,
-      "logps/chosen": -280.17669677734375,
-      "logps/rejected": -281.76580810546875,
-      "loss": 0.6926,
-      "rewards/accuracies": 0.550000011920929,
-      "rewards/chosen": 0.0014820825308561325,
-      "rewards/margins": 0.0013314359821379185,
-      "rewards/rejected": 0.00015064656327012926,
+      "logits/chosen": -2.5793910026550293,
+      "logits/rejected": -2.5509467124938965,
+      "logps/chosen": -280.1932067871094,
+      "logps/rejected": -281.7767639160156,
+      "loss": 0.6928,
+      "rewards/accuracies": 0.6499999761581421,
+      "rewards/chosen": 0.0013170629972591996,
+      "rewards/margins": 0.0012757038930431008,
+      "rewards/rejected": 4.135915150982328e-05,
       "step": 30
     },
     {
       "epoch": 0.04186289900575615,
-      "grad_norm": 11.981014455337817,
+      "grad_norm": 11.971540688024561,
       "learning_rate": 2.0833333333333333e-07,
-      "logits/chosen": -2.5402791500091553,
-      "logits/rejected": -2.449491500854492,
-      "logps/chosen": -298.2373046875,
-      "logps/rejected": -286.4289245605469,
+      "logits/chosen": -2.5403754711151123,
+      "logits/rejected": -2.4496283531188965,
+      "logps/chosen": -298.2412414550781,
+      "logps/rejected": -286.4029235839844,
       "loss": 0.6918,
-      "rewards/accuracies": 0.5874999761581421,
-      "rewards/chosen": 0.0022775838151574135,
-      "rewards/margins": 0.0028102027717977762,
-      "rewards/rejected": -0.0005326189566403627,
+      "rewards/accuracies": 0.574999988079071,
+      "rewards/chosen": 0.002237908076494932,
+      "rewards/margins": 0.0025104572996497154,
+      "rewards/rejected": -0.0002725489321164787,
       "step": 40
     },
     {
       "epoch": 0.052328623757195186,
-      "grad_norm": 10.97354698714153,
+      "grad_norm": 10.953203550036346,
       "learning_rate": 2.604166666666667e-07,
-      "logits/chosen": -2.558612108230591,
-      "logits/rejected": -2.4638822078704834,
-      "logps/chosen": -274.1297912597656,
-      "logps/rejected": -240.5820770263672,
-      "loss": 0.6899,
-      "rewards/accuracies": 0.6625000238418579,
-      "rewards/chosen": 0.002552950754761696,
-      "rewards/margins": 0.005607770290225744,
-      "rewards/rejected": -0.0030548193026334047,
+      "logits/chosen": -2.5587046146392822,
+      "logits/rejected": -2.463972568511963,
+      "logps/chosen": -274.0966796875,
+      "logps/rejected": -240.55062866210938,
+      "loss": 0.6898,
+      "rewards/accuracies": 0.65625,
+      "rewards/chosen": 0.0028838147409260273,
+      "rewards/margins": 0.005623813718557358,
+      "rewards/rejected": -0.0027399989776313305,
       "step": 50
     },
     {
       "epoch": 0.06279434850863422,
-      "grad_norm": 11.035335415046434,
+      "grad_norm": 10.943227101823018,
       "learning_rate": 3.1249999999999997e-07,
-      "logits/chosen": -2.5303585529327393,
-      "logits/rejected": -2.4870927333831787,
-      "logps/chosen": -310.5282897949219,
-      "logps/rejected": -283.35943603515625,
-      "loss": 0.6857,
-      "rewards/accuracies": 0.637499988079071,
-      "rewards/chosen": 0.007333799265325069,
-      "rewards/margins": 0.013189129531383514,
-      "rewards/rejected": -0.005855330731719732,
+      "logits/chosen": -2.5307021141052246,
+      "logits/rejected": -2.4873709678649902,
+      "logps/chosen": -310.4714660644531,
+      "logps/rejected": -283.3081970214844,
+      "loss": 0.6858,
+      "rewards/accuracies": 0.643750011920929,
+      "rewards/chosen": 0.007902054116129875,
+      "rewards/margins": 0.013244956731796265,
+      "rewards/rejected": -0.005342902150005102,
       "step": 60
     },
     {
       "epoch": 0.07326007326007326,
-      "grad_norm": 14.219935605546187,
+      "grad_norm": 10.581002478128156,
       "learning_rate": 3.645833333333333e-07,
-      "logits/chosen": -2.410804033279419,
-      "logits/rejected": -2.4123597145080566,
-      "logps/chosen": -272.55145263671875,
-      "logps/rejected": -267.7709045410156,
-      "loss": 0.6809,
-      "rewards/accuracies": 0.699999988079071,
-      "rewards/chosen": -0.012238213792443275,
-      "rewards/margins": 0.02978568710386753,
-      "rewards/rejected": -0.04202389717102051,
+      "logits/chosen": -2.411393642425537,
+      "logits/rejected": -2.4127840995788574,
+      "logps/chosen": -272.46405029296875,
+      "logps/rejected": -267.72686767578125,
+      "loss": 0.681,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -0.011364299803972244,
+      "rewards/margins": 0.030219241976737976,
+      "rewards/rejected": -0.04158353805541992,
       "step": 70
     },
     {
       "epoch": 0.0837257980115123,
-      "grad_norm": 11.396556778995892,
+      "grad_norm": 11.42733355645212,
       "learning_rate": 4.1666666666666667e-07,
-      "logits/chosen": -2.4814910888671875,
-      "logits/rejected": -2.3977572917938232,
-      "logps/chosen": -288.310302734375,
-      "logps/rejected": -264.4598083496094,
-      "loss": 0.6714,
-      "rewards/accuracies": 0.6812499761581421,
-      "rewards/chosen": -0.032100919634103775,
-      "rewards/margins": 0.04889374226331711,
-      "rewards/rejected": -0.08099466562271118,
+      "logits/chosen": -2.481160879135132,
+      "logits/rejected": -2.397334575653076,
+      "logps/chosen": -288.3390197753906,
+      "logps/rejected": -264.55322265625,
+      "loss": 0.6712,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": -0.03238832578063011,
+      "rewards/margins": 0.04954018443822861,
+      "rewards/rejected": -0.08192850649356842,
       "step": 80
     },
     {
       "epoch": 0.09419152276295134,
-      "grad_norm": 11.246920998056021,
+      "grad_norm": 11.210988509809624,
       "learning_rate": 4.6874999999999996e-07,
-      "logits/chosen": -2.406747579574585,
-      "logits/rejected": -2.387026309967041,
-      "logps/chosen": -270.9520568847656,
-      "logps/rejected": -281.39276123046875,
-      "loss": 0.6686,
-      "rewards/accuracies": 0.637499988079071,
-      "rewards/chosen": -0.05444145202636719,
-      "rewards/margins": 0.03418709710240364,
-      "rewards/rejected": -0.08862854540348053,
+      "logits/chosen": -2.407602310180664,
+      "logits/rejected": -2.3880608081817627,
+      "logps/chosen": -270.7894592285156,
+      "logps/rejected": -281.30572509765625,
+      "loss": 0.6685,
+      "rewards/accuracies": 0.643750011920929,
+      "rewards/chosen": -0.05281522870063782,
+      "rewards/margins": 0.03494281694293022,
+      "rewards/rejected": -0.08775804191827774,
       "step": 90
     },
     {
       "epoch": 0.10465724751439037,
-      "grad_norm": 12.0711830754396,
+      "grad_norm": 11.737180826993443,
       "learning_rate": 4.999732492681437e-07,
-      "logits/chosen": -2.4021925926208496,
-      "logits/rejected": -2.336446523666382,
-      "logps/chosen": -313.94293212890625,
-      "logps/rejected": -299.0536804199219,
-      "loss": 0.6527,
-      "rewards/accuracies": 0.668749988079071,
-      "rewards/chosen": -0.04575753211975098,
-      "rewards/margins": 0.09000825881958008,
-      "rewards/rejected": -0.13576579093933105,
+      "logits/chosen": -2.4048094749450684,
+      "logits/rejected": -2.339949607849121,
+      "logps/chosen": -313.2823486328125,
+      "logps/rejected": -298.7939453125,
+      "loss": 0.6509,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -0.03915167227387428,
+      "rewards/margins": 0.094016894698143,
+      "rewards/rejected": -0.1331685483455658,
       "step": 100
     },
     {
       "epoch": 0.10465724751439037,
-      "eval_logits/chosen": -2.454146146774292,
-      "eval_logits/rejected": -2.390709400177002,
-      "eval_logps/chosen": -293.5221252441406,
-      "eval_logps/rejected": -278.3384094238281,
-      "eval_loss": 0.6527308225631714,
-      "eval_rewards/accuracies": 0.7003968358039856,
-      "eval_rewards/chosen": -0.017938869073987007,
-      "eval_rewards/margins": 0.10392169654369354,
-      "eval_rewards/rejected": -0.1218605563044548,
-      "eval_runtime": 207.8472,
-      "eval_samples_per_second": 9.622,
-      "eval_steps_per_second": 0.303,
+      "eval_logits/chosen": -2.4569716453552246,
+      "eval_logits/rejected": -2.39499831199646,
+      "eval_logps/chosen": -294.89752197265625,
+      "eval_logps/rejected": -280.17822265625,
+      "eval_loss": 0.6506768465042114,
+      "eval_rewards/accuracies": 0.6865079402923584,
+      "eval_rewards/chosen": -0.03169262409210205,
+      "eval_rewards/margins": 0.10856584459543228,
+      "eval_rewards/rejected": -0.14025846123695374,
+      "eval_runtime": 205.869,
+      "eval_samples_per_second": 9.715,
+      "eval_steps_per_second": 0.306,
       "step": 100
     },
     {
       "epoch": 0.1151229722658294,
-      "grad_norm": 15.521410395883006,
+      "grad_norm": 18.957823734967118,
       "learning_rate": 4.996723692767926e-07,
-      "logits/chosen": -2.443455457687378,
-      "logits/rejected": -2.398399829864502,
-      "logps/chosen": -253.86575317382812,
-      "logps/rejected": -260.5719299316406,
-      "loss": 0.6468,
-      "rewards/accuracies": 0.6875,
-      "rewards/chosen": -0.06694073975086212,
-      "rewards/margins": 0.11035104095935822,
-      "rewards/rejected": -0.17729176580905914,
+      "logits/chosen": -2.443211793899536,
+      "logits/rejected": -2.3995585441589355,
+      "logps/chosen": -254.23428344726562,
+      "logps/rejected": -261.7113952636719,
+      "loss": 0.645,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -0.07062581926584244,
+      "rewards/margins": 0.11806046962738037,
+      "rewards/rejected": -0.1886862814426422,
       "step": 110
     },
     {
       "epoch": 0.12558869701726844,
-      "grad_norm": 22.739816534288916,
+      "grad_norm": 211.9091102470587,
       "learning_rate": 4.990375746213598e-07,
-      "logits/chosen": -2.3525469303131104,
-      "logits/rejected": -2.293430805206299,
-      "logps/chosen": -293.82470703125,
-      "logps/rejected": -334.8168640136719,
-      "loss": 0.6272,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -0.22183768451213837,
-      "rewards/margins": 0.17258095741271973,
-      "rewards/rejected": -0.3944186568260193,
+      "logits/chosen": -2.338498115539551,
+      "logits/rejected": -2.2794461250305176,
+      "logps/chosen": -298.16986083984375,
+      "logps/rejected": -339.7401428222656,
+      "loss": 0.6298,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -0.2652895748615265,
+      "rewards/margins": 0.1783614456653595,
+      "rewards/rejected": -0.443651020526886,
       "step": 120
     },
     {
       "epoch": 0.1360544217687075,
-      "grad_norm": 20.276391537200777,
+      "grad_norm": 37.07462356301655,
       "learning_rate": 4.980697142834314e-07,
-      "logits/chosen": -2.2628631591796875,
-      "logits/rejected": -2.236845016479492,
-      "logps/chosen": -307.3517761230469,
-      "logps/rejected": -307.5767517089844,
-      "loss": 0.6066,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -0.16923505067825317,
-      "rewards/margins": 0.3002583086490631,
-      "rewards/rejected": -0.46949338912963867,
+      "logits/chosen": -2.244835615158081,
+      "logits/rejected": -2.2257189750671387,
+      "logps/chosen": -322.57122802734375,
+      "logps/rejected": -319.81500244140625,
+      "loss": 0.6229,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -0.3214291036128998,
+      "rewards/margins": 0.2704463005065918,
+      "rewards/rejected": -0.591875433921814,
       "step": 130
     },
     {
       "epoch": 0.14652014652014653,
-      "grad_norm": 17.987546221562845,
+      "grad_norm": 18.745612354229714,
       "learning_rate": 4.967700826904229e-07,
-      "logits/chosen": -2.134958505630493,
-      "logits/rejected": -2.08327054977417,
-      "logps/chosen": -298.15313720703125,
-      "logps/rejected": -315.8902893066406,
-      "loss": 0.6081,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.03450244665145874,
-      "rewards/margins": 0.24115605652332306,
-      "rewards/rejected": -0.2756584882736206,
+      "logits/chosen": -2.1823461055755615,
+      "logits/rejected": -2.132317304611206,
+      "logps/chosen": -301.0035705566406,
+      "logps/rejected": -319.9606628417969,
+      "loss": 0.6085,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -0.06300695240497589,
+      "rewards/margins": 0.2533551752567291,
+      "rewards/rejected": -0.3163621127605438,
       "step": 140
     },
     {
       "epoch": 0.15698587127158556,
-      "grad_norm": 23.008496326619163,
+      "grad_norm": 20.385737144331163,
       "learning_rate": 4.951404179843962e-07,
-      "logits/chosen": -2.194613456726074,
-      "logits/rejected": -2.1461760997772217,
-      "logps/chosen": -314.67279052734375,
-      "logps/rejected": -290.2945861816406,
-      "loss": 0.6108,
-      "rewards/accuracies": 0.643750011920929,
-      "rewards/chosen": -0.15121625363826752,
-      "rewards/margins": 0.22037005424499512,
-      "rewards/rejected": -0.37158629298210144,
+      "logits/chosen": -2.247217893600464,
+      "logits/rejected": -2.2084200382232666,
+      "logps/chosen": -331.0395812988281,
+      "logps/rejected": -310.9860534667969,
+      "loss": 0.609,
+      "rewards/accuracies": 0.6499999761581421,
+      "rewards/chosen": -0.31488415598869324,
+      "rewards/margins": 0.2636170983314514,
+      "rewards/rejected": -0.578501284122467,
       "step": 150
     },
     {
       "epoch": 0.1674515960230246,
-      "grad_norm": 29.398328418025,
+      "grad_norm": 17.50514107179683,
       "learning_rate": 4.931828996974498e-07,
-      "logits/chosen": -2.0455970764160156,
-      "logits/rejected": -1.9418065547943115,
-      "logps/chosen": -323.17498779296875,
-      "logps/rejected": -317.56756591796875,
-      "loss": 0.5728,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.17028222978115082,
-      "rewards/margins": 0.4041343629360199,
-      "rewards/rejected": -0.5744165778160095,
+      "logits/chosen": -2.186821937561035,
+      "logits/rejected": -2.093240261077881,
+      "logps/chosen": -330.5211181640625,
+      "logps/rejected": -320.204345703125,
+      "loss": 0.5859,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -0.2437433898448944,
+      "rewards/margins": 0.3570416569709778,
+      "rewards/rejected": -0.6007851362228394,
       "step": 160
     },
     {
       "epoch": 0.17791732077446362,
-      "grad_norm": 23.497103839633773,
+      "grad_norm": 18.503251999087315,
       "learning_rate": 4.909001458367866e-07,
-      "logits/chosen": -1.9791698455810547,
-      "logits/rejected": -1.8650147914886475,
-      "logps/chosen": -310.0411071777344,
-      "logps/rejected": -312.9387512207031,
-      "loss": 0.5794,
-      "rewards/accuracies": 0.71875,
-      "rewards/chosen": -0.24319405853748322,
-      "rewards/margins": 0.44290846586227417,
-      "rewards/rejected": -0.6861025094985962,
+      "logits/chosen": -2.151353597640991,
+      "logits/rejected": -2.0457262992858887,
+      "logps/chosen": -322.26434326171875,
+      "logps/rejected": -320.9549255371094,
+      "loss": 0.5782,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -0.3654264807701111,
+      "rewards/margins": 0.40083780884742737,
+      "rewards/rejected": -0.7662642598152161,
       "step": 170
     },
     {
       "epoch": 0.18838304552590268,
-      "grad_norm": 19.270246895233647,
+      "grad_norm": 22.1110027585031,
       "learning_rate": 4.882952093833627e-07,
-      "logits/chosen": -1.8387302160263062,
-      "logits/rejected": -1.7828607559204102,
-      "logps/chosen": -303.87457275390625,
-      "logps/rejected": -347.4334411621094,
-      "loss": 0.5813,
-      "rewards/accuracies": 0.6875,
-      "rewards/chosen": -0.5621883273124695,
-      "rewards/margins": 0.38899126648902893,
-      "rewards/rejected": -0.9511796832084656,
+      "logits/chosen": -2.0537924766540527,
+      "logits/rejected": -2.013256549835205,
+      "logps/chosen": -296.3823547363281,
+      "logps/rejected": -343.3452453613281,
+      "loss": 0.5758,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -0.487266480922699,
+      "rewards/margins": 0.42303138971328735,
+      "rewards/rejected": -0.9102978706359863,
       "step": 180
     },
     {
       "epoch": 0.1988487702773417,
-      "grad_norm": 21.20227835187441,
+      "grad_norm": 22.087134238338532,
       "learning_rate": 4.853715742087946e-07,
-      "logits/chosen": -1.604284644126892,
-      "logits/rejected": -1.5313122272491455,
-      "logps/chosen": -309.1009521484375,
-      "logps/rejected": -356.45343017578125,
-      "loss": 0.5712,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.6294011473655701,
-      "rewards/margins": 0.3671324849128723,
-      "rewards/rejected": -0.9965336918830872,
+      "logits/chosen": -1.8600575923919678,
+      "logits/rejected": -1.812792420387268,
+      "logps/chosen": -289.62152099609375,
+      "logps/rejected": -342.3699645996094,
+      "loss": 0.5745,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -0.43460702896118164,
+      "rewards/margins": 0.4210926592350006,
+      "rewards/rejected": -0.8556997179985046,
       "step": 190
     },
     {
       "epoch": 0.20931449502878074,
-      "grad_norm": 19.896447316671875,
+      "grad_norm": 22.27318395987777,
       "learning_rate": 4.821331504159906e-07,
-      "logits/chosen": -1.8829338550567627,
-      "logits/rejected": -1.7022336721420288,
-      "logps/chosen": -347.5841979980469,
-      "logps/rejected": -321.65185546875,
-      "loss": 0.5965,
-      "rewards/accuracies": 0.675000011920929,
-      "rewards/chosen": -0.23933956027030945,
-      "rewards/margins": 0.342945396900177,
-      "rewards/rejected": -0.5822849869728088,
+      "logits/chosen": -2.048029661178589,
+      "logits/rejected": -1.8949153423309326,
+      "logps/chosen": -378.06451416015625,
+      "logps/rejected": -363.1112976074219,
+      "loss": 0.5842,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": -0.5441427826881409,
+      "rewards/margins": 0.4527364671230316,
+      "rewards/rejected": -0.9968792796134949,
       "step": 200
     },
     {
       "epoch": 0.20931449502878074,
-      "eval_logits/chosen": -1.820605993270874,
-      "eval_logits/rejected": -1.7023367881774902,
-      "eval_logps/chosen": -319.80035400390625,
-      "eval_logps/rejected": -332.8750915527344,
-      "eval_loss": 0.5948727130889893,
-      "eval_rewards/accuracies": 0.7103174328804016,
-      "eval_rewards/chosen": -0.2807207703590393,
-      "eval_rewards/margins": 0.38650667667388916,
-      "eval_rewards/rejected": -0.6672274470329285,
-      "eval_runtime": 207.3956,
-      "eval_samples_per_second": 9.643,
-      "eval_steps_per_second": 0.304,
+      "eval_logits/chosen": -1.991257667541504,
+      "eval_logits/rejected": -1.8882709741592407,
+      "eval_logps/chosen": -348.01226806640625,
+      "eval_logps/rejected": -367.80316162109375,
+      "eval_loss": 0.586904764175415,
+      "eval_rewards/accuracies": 0.7182539701461792,
+      "eval_rewards/chosen": -0.5628401637077332,
+      "eval_rewards/margins": 0.45366790890693665,
+      "eval_rewards/rejected": -1.0165081024169922,
+      "eval_runtime": 206.2096,
+      "eval_samples_per_second": 9.699,
+      "eval_steps_per_second": 0.306,
       "step": 200
     },
     {
       "epoch": 0.21978021978021978,
-      "grad_norm": 23.83538654736134,
+      "grad_norm": 24.077327884786772,
       "learning_rate": 4.785842691097342e-07,
-      "logits/chosen": -1.751426100730896,
-      "logits/rejected": -1.600547432899475,
-      "logps/chosen": -337.1266174316406,
-      "logps/rejected": -363.1830749511719,
-      "loss": 0.5942,
-      "rewards/accuracies": 0.6499999761581421,
-      "rewards/chosen": -0.44608935713768005,
-      "rewards/margins": 0.3921416401863098,
-      "rewards/rejected": -0.8382309675216675,
+      "logits/chosen": -1.94464111328125,
+      "logits/rejected": -1.8161208629608154,
+      "logps/chosen": -353.6456604003906,
+      "logps/rejected": -386.1349792480469,
+      "loss": 0.5866,
+      "rewards/accuracies": 0.6812499761581421,
+      "rewards/chosen": -0.6112793684005737,
+      "rewards/margins": 0.45647066831588745,
+      "rewards/rejected": -1.0677502155303955,
       "step": 210
     },
     {
       "epoch": 0.2302459445316588,
-      "grad_norm": 21.783442575486323,
+      "grad_norm": 25.256951902015157,
       "learning_rate": 4.7472967660421603e-07,
-      "logits/chosen": -1.7183506488800049,
-      "logits/rejected": -1.5304996967315674,
-      "logps/chosen": -376.5364074707031,
-      "logps/rejected": -355.9579772949219,
-      "loss": 0.592,
-      "rewards/accuracies": 0.6812499761581421,
-      "rewards/chosen": -0.8024527430534363,
-      "rewards/margins": 0.3702601492404938,
-      "rewards/rejected": -1.1727129220962524,
+      "logits/chosen": -1.9311796426773071,
+      "logits/rejected": -1.7711641788482666,
+      "logps/chosen": -352.7847595214844,
+      "logps/rejected": -341.3170471191406,
+      "loss": 0.5697,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": -0.564936637878418,
+      "rewards/margins": 0.46136683225631714,
+      "rewards/rejected": -1.0263035297393799,
       "step": 220
     },
     {
       "epoch": 0.24071166928309787,
-      "grad_norm": 22.38037664465566,
+      "grad_norm": 25.586478019113667,
       "learning_rate": 4.705745280752585e-07,
-      "logits/chosen": -1.7021652460098267,
-      "logits/rejected": -1.6454862356185913,
-      "logps/chosen": -368.9347839355469,
-      "logps/rejected": -381.9288330078125,
-      "loss": 0.572,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -0.7253575921058655,
-      "rewards/margins": 0.3454781174659729,
-      "rewards/rejected": -1.0708357095718384,
+      "logits/chosen": -1.7587419748306274,
+      "logits/rejected": -1.7059828042984009,
+      "logps/chosen": -347.0582275390625,
+      "logps/rejected": -369.04644775390625,
+      "loss": 0.5672,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": -0.5065922737121582,
+      "rewards/margins": 0.435419499874115,
+      "rewards/rejected": -0.9420117139816284,
       "step": 230
     },
     {
       "epoch": 0.25117739403453687,
-      "grad_norm": 35.014618849070864,
+      "grad_norm": 30.28171698877856,
       "learning_rate": 4.6612438066572555e-07,
-      "logits/chosen": -1.6926515102386475,
-      "logits/rejected": -1.6362680196762085,
-      "logps/chosen": -324.94573974609375,
-      "logps/rejected": -367.96466064453125,
-      "loss": 0.5556,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -0.47211360931396484,
-      "rewards/margins": 0.5889753103256226,
-      "rewards/rejected": -1.0610889196395874,
+      "logits/chosen": -1.6479380130767822,
+      "logits/rejected": -1.585958480834961,
+      "logps/chosen": -339.24859619140625,
+      "logps/rejected": -380.8953552246094,
+      "loss": 0.5514,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -0.6151422262191772,
+      "rewards/margins": 0.5752537846565247,
+      "rewards/rejected": -1.1903960704803467,
       "step": 240
     },
     {
       "epoch": 0.2616431187859759,
-      "grad_norm": 26.49332081236356,
+      "grad_norm": 27.376050990007407,
       "learning_rate": 4.6138518605333664e-07,
-      "logits/chosen": -1.7838761806488037,
-      "logits/rejected": -1.7312650680541992,
-      "logps/chosen": -338.6064147949219,
-      "logps/rejected": -371.168701171875,
-      "loss": 0.5397,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -0.494678795337677,
-      "rewards/margins": 0.5390876531600952,
-      "rewards/rejected": -1.0337663888931274,
+      "logits/chosen": -1.6576675176620483,
+      "logits/rejected": -1.5993081331253052,
+      "logps/chosen": -355.48638916015625,
+      "logps/rejected": -390.65313720703125,
+      "loss": 0.5373,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -0.6634787321090698,
+      "rewards/margins": 0.5651322603225708,
+      "rewards/rejected": -1.228610873222351,
       "step": 250
     },
     {
       "epoch": 0.272108843537415,
-      "grad_norm": 24.98840517843348,
+      "grad_norm": 43.177740519027985,
       "learning_rate": 4.5636328249082514e-07,
-      "logits/chosen": -1.9279499053955078,
-      "logits/rejected": -1.8210124969482422,
-      "logps/chosen": -342.07049560546875,
-      "logps/rejected": -380.8410339355469,
-      "loss": 0.541,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.42370086908340454,
-      "rewards/margins": 0.49395614862442017,
-      "rewards/rejected": -0.9176570177078247,
+      "logits/chosen": -1.7310291528701782,
+      "logits/rejected": -1.6111475229263306,
+      "logps/chosen": -385.6434631347656,
+      "logps/rejected": -425.70599365234375,
+      "loss": 0.5521,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -0.8594309687614441,
+      "rewards/margins": 0.5068761706352234,
+      "rewards/rejected": -1.366307020187378,
       "step": 260
     },
     {
       "epoch": 0.282574568288854,
-      "grad_norm": 24.99592460312298,
+      "grad_norm": 26.026493875287635,
       "learning_rate": 4.510653863290871e-07,
-      "logits/chosen": -1.7682479619979858,
-      "logits/rejected": -1.682395339012146,
-      "logps/chosen": -367.8274230957031,
-      "logps/rejected": -373.8634033203125,
-      "loss": 0.5495,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -0.5512995719909668,
-      "rewards/margins": 0.5709925889968872,
-      "rewards/rejected": -1.122292160987854,
+      "logits/chosen": -1.52871572971344,
+      "logits/rejected": -1.4289811849594116,
+      "logps/chosen": -399.4200744628906,
+      "logps/rejected": -410.7467346191406,
+      "loss": 0.5441,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -0.8672260046005249,
+      "rewards/margins": 0.6238988637924194,
+      "rewards/rejected": -1.4911251068115234,
       "step": 270
     },
     {
       "epoch": 0.29304029304029305,
-      "grad_norm": 20.790540096080914,
+      "grad_norm": 37.6240820385195,
       "learning_rate": 4.4549858303465737e-07,
-      "logits/chosen": -1.8336889743804932,
-      "logits/rejected": -1.7349803447723389,
-      "logps/chosen": -365.58770751953125,
-      "logps/rejected": -396.11688232421875,
-      "loss": 0.5838,
+      "logits/chosen": -1.7061611413955688,
+      "logits/rejected": -1.602936029434204,
+      "logps/chosen": -387.9464111328125,
+      "logps/rejected": -430.4659729003906,
+      "loss": 0.5687,
       "rewards/accuracies": 0.75,
-      "rewards/chosen": -0.5898444652557373,
-      "rewards/margins": 0.508275032043457,
-      "rewards/rejected": -1.0981194972991943,
+      "rewards/chosen": -0.8134315609931946,
+      "rewards/margins": 0.6281790733337402,
+      "rewards/rejected": -1.4416106939315796,
       "step": 280
     },
     {
       "epoch": 0.3035060177917321,
-      "grad_norm": 28.914812216587325,
+      "grad_norm": 30.468189665035492,
       "learning_rate": 4.396703177135261e-07,
-      "logits/chosen": -1.725181221961975,
-      "logits/rejected": -1.5229023694992065,
-      "logps/chosen": -344.6284484863281,
-      "logps/rejected": -354.0973205566406,
-      "loss": 0.5535,
+      "logits/chosen": -1.7203712463378906,
+      "logits/rejected": -1.513145923614502,
+      "logps/chosen": -350.87872314453125,
+      "logps/rejected": -369.96343994140625,
+      "loss": 0.5572,
       "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.6310089826583862,
-      "rewards/margins": 0.48871439695358276,
-      "rewards/rejected": -1.1197235584259033,
+      "rewards/chosen": -0.6935116648674011,
+      "rewards/margins": 0.5848726034164429,
+      "rewards/rejected": -1.2783840894699097,
       "step": 290
     },
     {
       "epoch": 0.3139717425431711,
-      "grad_norm": 30.44777572316204,
+      "grad_norm": 25.896063593338788,
       "learning_rate": 4.335883851539693e-07,
-      "logits/chosen": -1.5577516555786133,
-      "logits/rejected": -1.4662883281707764,
-      "logps/chosen": -371.7145080566406,
-      "logps/rejected": -424.6953125,
-      "loss": 0.5658,
-      "rewards/accuracies": 0.6812499761581421,
-      "rewards/chosen": -0.9486764073371887,
-      "rewards/margins": 0.5460881590843201,
-      "rewards/rejected": -1.4947646856307983,
+      "logits/chosen": -1.7010329961776733,
+      "logits/rejected": -1.6187465190887451,
+      "logps/chosen": -348.5817565917969,
+      "logps/rejected": -401.21270751953125,
+      "loss": 0.5518,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": -0.7173490524291992,
+      "rewards/margins": 0.5425899624824524,
+      "rewards/rejected": -1.2599390745162964,
       "step": 300
     },
     {
       "epoch": 0.3139717425431711,
-      "eval_logits/chosen": -1.5125768184661865,
-      "eval_logits/rejected": -1.351996898651123,
-      "eval_logps/chosen": -368.3990478515625,
-      "eval_logps/rejected": -406.29046630859375,
-      "eval_loss": 0.5555263757705688,
-      "eval_rewards/accuracies": 0.7658730149269104,
-      "eval_rewards/chosen": -0.7667079567909241,
-      "eval_rewards/margins": 0.6346728801727295,
-      "eval_rewards/rejected": -1.4013807773590088,
-      "eval_runtime": 207.3223,
-      "eval_samples_per_second": 9.647,
-      "eval_steps_per_second": 0.304,
+      "eval_logits/chosen": -1.5902302265167236,
+      "eval_logits/rejected": -1.4401179552078247,
+      "eval_logps/chosen": -367.8922424316406,
+      "eval_logps/rejected": -402.97265625,
+      "eval_loss": 0.5573062300682068,
+      "eval_rewards/accuracies": 0.7420634627342224,
+      "eval_rewards/chosen": -0.7616395950317383,
+      "eval_rewards/margins": 0.606563150882721,
+      "eval_rewards/rejected": -1.368202805519104,
+      "eval_runtime": 205.6103,
+      "eval_samples_per_second": 9.727,
+      "eval_steps_per_second": 0.306,
       "step": 300
     },
     {
       "epoch": 0.32443746729461015,
-      "grad_norm": 24.53266227863393,
+      "grad_norm": 23.611287954852504,
       "learning_rate": 4.272609194017105e-07,
-      "logits/chosen": -1.7091705799102783,
-      "logits/rejected": -1.3815821409225464,
-      "logps/chosen": -355.83587646484375,
-      "logps/rejected": -361.08575439453125,
-      "loss": 0.5415,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.6351903676986694,
-      "rewards/margins": 0.6549168229103088,
-      "rewards/rejected": -1.2901071310043335,
+      "logits/chosen": -1.7664659023284912,
+      "logits/rejected": -1.4679615497589111,
+      "logps/chosen": -360.18707275390625,
+      "logps/rejected": -362.8342590332031,
+      "loss": 0.542,
+      "rewards/accuracies": 0.731249988079071,
+      "rewards/chosen": -0.678702712059021,
+      "rewards/margins": 0.6288893818855286,
+      "rewards/rejected": -1.3075921535491943,
       "step": 310
     },
     {
       "epoch": 0.3349031920460492,
-      "grad_norm": 23.055187734565123,
+      "grad_norm": 24.070079232095797,
       "learning_rate": 4.2069638288135547e-07,
-      "logits/chosen": -1.5925337076187134,
-      "logits/rejected": -1.4470938444137573,
-      "logps/chosen": -317.94647216796875,
-      "logps/rejected": -371.202880859375,
-      "loss": 0.5651,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -0.5869553089141846,
-      "rewards/margins": 0.6055953502655029,
-      "rewards/rejected": -1.1925506591796875,
+      "logits/chosen": -1.6632932424545288,
+      "logits/rejected": -1.5471082925796509,
+      "logps/chosen": -327.7480163574219,
+      "logps/rejected": -383.7098388671875,
+      "loss": 0.558,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": -0.6849709749221802,
+      "rewards/margins": 0.6326489448547363,
+      "rewards/rejected": -1.317620038986206,
       "step": 320
     },
     {
       "epoch": 0.3453689167974882,
-      "grad_norm": 36.10086224192863,
+      "grad_norm": 38.56102713027139,
       "learning_rate": 4.139035550786494e-07,
-      "logits/chosen": -1.376239538192749,
-      "logits/rejected": -1.2483571767807007,
-      "logps/chosen": -347.3015441894531,
-      "logps/rejected": -369.86407470703125,
-      "loss": 0.5485,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -0.8465766906738281,
-      "rewards/margins": 0.4942452311515808,
-      "rewards/rejected": -1.3408218622207642,
+      "logits/chosen": -1.4653488397598267,
+      "logits/rejected": -1.3664615154266357,
+      "logps/chosen": -347.2395935058594,
+      "logps/rejected": -377.60748291015625,
+      "loss": 0.5465,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -0.8459569811820984,
+      "rewards/margins": 0.5722988843917847,
+      "rewards/rejected": -1.4182560443878174,
       "step": 330
     },
     {
       "epoch": 0.35583464154892724,
-      "grad_norm": 29.84097305081729,
+      "grad_norm": 28.031822632702205,
       "learning_rate": 4.0689152079869306e-07,
-      "logits/chosen": -1.2099343538284302,
-      "logits/rejected": -1.0386359691619873,
-      "logps/chosen": -376.8388977050781,
-      "logps/rejected": -399.56024169921875,
-      "loss": 0.5574,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -0.9760238528251648,
-      "rewards/margins": 0.6029719114303589,
-      "rewards/rejected": -1.5789958238601685,
+      "logits/chosen": -1.3066160678863525,
+      "logits/rejected": -1.156473159790039,
+      "logps/chosen": -396.61676025390625,
+      "logps/rejected": -416.3009338378906,
+      "loss": 0.574,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -1.1738026142120361,
+      "rewards/margins": 0.5725993514060974,
+      "rewards/rejected": -1.7464020252227783,
       "step": 340
     },
     {
       "epoch": 0.3663003663003663,
-      "grad_norm": 23.516384637529317,
+      "grad_norm": 26.144173653631405,
       "learning_rate": 3.99669658015821e-07,
-      "logits/chosen": -1.3511981964111328,
-      "logits/rejected": -1.1930326223373413,
-      "logps/chosen": -384.8575744628906,
-      "logps/rejected": -408.7091369628906,
-      "loss": 0.527,
-      "rewards/accuracies": 0.6812499761581421,
-      "rewards/chosen": -0.8373312950134277,
-      "rewards/margins": 0.5571725964546204,
-      "rewards/rejected": -1.3945038318634033,
+      "logits/chosen": -1.540045976638794,
+      "logits/rejected": -1.4006694555282593,
+      "logps/chosen": -385.5912780761719,
+      "logps/rejected": -410.674072265625,
+      "loss": 0.536,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": -0.8446685075759888,
+      "rewards/margins": 0.569485068321228,
+      "rewards/rejected": -1.4141535758972168,
       "step": 350
     },
     {
       "epoch": 0.37676609105180536,
-      "grad_norm": 30.962318294654995,
+      "grad_norm": 33.22726534237683,
       "learning_rate": 3.92247625331392e-07,
-      "logits/chosen": -1.2773854732513428,
-      "logits/rejected": -1.1658525466918945,
-      "logps/chosen": -364.7725524902344,
-      "logps/rejected": -398.4949645996094,
-      "loss": 0.5148,
+      "logits/chosen": -1.4644501209259033,
+      "logits/rejected": -1.3674944639205933,
+      "logps/chosen": -364.0406188964844,
+      "logps/rejected": -396.0486145019531,
+      "loss": 0.5221,
       "rewards/accuracies": 0.7437499761581421,
-      "rewards/chosen": -0.7480669021606445,
-      "rewards/margins": 0.5878486037254333,
-      "rewards/rejected": -1.3359155654907227,
+      "rewards/chosen": -0.7407475709915161,
+      "rewards/margins": 0.5707039833068848,
+      "rewards/rejected": -1.3114515542984009,
       "step": 360
     },
     {
       "epoch": 0.3872318158032444,
-      "grad_norm": 27.389158880513975,
+      "grad_norm": 22.278385921862558,
       "learning_rate": 3.846353490562664e-07,
-      "logits/chosen": -1.1208012104034424,
-      "logits/rejected": -1.0132851600646973,
-      "logps/chosen": -367.5520935058594,
-      "logps/rejected": -390.63787841796875,
-      "loss": 0.5432,
+      "logits/chosen": -1.4632952213287354,
+      "logits/rejected": -1.364483118057251,
+      "logps/chosen": -353.5659484863281,
+      "logps/rejected": -371.3506774902344,
+      "loss": 0.5452,
       "rewards/accuracies": 0.6937500238418579,
-      "rewards/chosen": -0.9612551927566528,
-      "rewards/margins": 0.5348009467124939,
-      "rewards/rejected": -1.4960561990737915,
+      "rewards/chosen": -0.8213934898376465,
+      "rewards/margins": 0.4817906320095062,
+      "rewards/rejected": -1.3031842708587646,
       "step": 370
     },
     {
       "epoch": 0.3976975405546834,
-      "grad_norm": 33.89039091124739,
+      "grad_norm": 26.485819344845584,
       "learning_rate": 3.768430099352445e-07,
-      "logits/chosen": -1.071099042892456,
-      "logits/rejected": -0.8969755172729492,
-      "logps/chosen": -327.09857177734375,
-      "logps/rejected": -385.0154113769531,
-      "loss": 0.5284,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -0.6631902456283569,
-      "rewards/margins": 0.6594493389129639,
-      "rewards/rejected": -1.3226395845413208,
+      "logits/chosen": -1.4255541563034058,
+      "logits/rejected": -1.2725279331207275,
+      "logps/chosen": -331.7218933105469,
+      "logps/rejected": -386.83331298828125,
+      "loss": 0.5351,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -0.709423840045929,
+      "rewards/margins": 0.6313949823379517,
+      "rewards/rejected": -1.3408187627792358,
       "step": 380
     },
     {
       "epoch": 0.40816326530612246,
-      "grad_norm": 28.327278413320137,
+      "grad_norm": 26.88634695034012,
       "learning_rate": 3.6888102953122304e-07,
-      "logits/chosen": -1.0915015935897827,
-      "logits/rejected": -0.9525696039199829,
-      "logps/chosen": -349.69171142578125,
-      "logps/rejected": -393.8070373535156,
-      "loss": 0.553,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.6071487665176392,
-      "rewards/margins": 0.6005517244338989,
-      "rewards/rejected": -1.207700490951538,
+      "logits/chosen": -1.4283645153045654,
+      "logits/rejected": -1.313664197921753,
+      "logps/chosen": -346.01446533203125,
+      "logps/rejected": -394.5232849121094,
+      "loss": 0.5411,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -0.5703771710395813,
+      "rewards/margins": 0.6444854736328125,
+      "rewards/rejected": -1.2148627042770386,
       "step": 390
     },
     {
       "epoch": 0.4186289900575615,
-      "grad_norm": 39.79971412332236,
+      "grad_norm": 40.41797343152992,
       "learning_rate": 3.607600562872785e-07,
-      "logits/chosen": -1.1151034832000732,
-      "logits/rejected": -1.0535625219345093,
-      "logps/chosen": -351.47821044921875,
-      "logps/rejected": -393.27197265625,
-      "loss": 0.5589,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.6971405148506165,
-      "rewards/margins": 0.5149218440055847,
-      "rewards/rejected": -1.2120622396469116,
+      "logits/chosen": -1.3911030292510986,
+      "logits/rejected": -1.366856575012207,
+      "logps/chosen": -342.7760925292969,
+      "logps/rejected": -384.2821960449219,
+      "loss": 0.5596,
+      "rewards/accuracies": 0.706250011920929,
+      "rewards/chosen": -0.6101195216178894,
+      "rewards/margins": 0.5120452642440796,
+      "rewards/rejected": -1.1221649646759033,
       "step": 400
     },
     {
       "epoch": 0.4186289900575615,
-      "eval_logits/chosen": -1.3050860166549683,
-      "eval_logits/rejected": -1.1235393285751343,
-      "eval_logps/chosen": -345.56829833984375,
-      "eval_logps/rejected": -383.2520751953125,
-      "eval_loss": 0.5487203001976013,
-      "eval_rewards/accuracies": 0.7440476417541504,
-      "eval_rewards/chosen": -0.5384004712104797,
-      "eval_rewards/margins": 0.6325967311859131,
-      "eval_rewards/rejected": -1.1709972620010376,
-      "eval_runtime": 207.4367,
-      "eval_samples_per_second": 9.641,
-      "eval_steps_per_second": 0.304,
+      "eval_logits/chosen": -1.538719892501831,
+      "eval_logits/rejected": -1.3655085563659668,
+      "eval_logps/chosen": -346.5540771484375,
+      "eval_logps/rejected": -384.7071838378906,
+      "eval_loss": 0.5451058745384216,
+      "eval_rewards/accuracies": 0.7579365372657776,
+      "eval_rewards/chosen": -0.5482582449913025,
+      "eval_rewards/margins": 0.6372901797294617,
+      "eval_rewards/rejected": -1.1855483055114746,
+      "eval_runtime": 205.4146,
+      "eval_samples_per_second": 9.736,
+      "eval_steps_per_second": 0.307,
       "step": 400
     },
     {
       "epoch": 0.4290947148090005,
-      "grad_norm": 21.350151902719556,
+      "grad_norm": 21.999057622947905,
       "learning_rate": 3.5249095128531856e-07,
-      "logits/chosen": -1.0982998609542847,
-      "logits/rejected": -0.9201542735099792,
-      "logps/chosen": -343.6358337402344,
-      "logps/rejected": -367.96612548828125,
-      "loss": 0.556,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.6256347894668579,
-      "rewards/margins": 0.5450095534324646,
-      "rewards/rejected": -1.1706444025039673,
+      "logits/chosen": -1.391875982284546,
+      "logits/rejected": -1.2355632781982422,
+      "logps/chosen": -341.1160888671875,
+      "logps/rejected": -364.1993713378906,
+      "loss": 0.5513,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -0.6004370450973511,
+      "rewards/margins": 0.5325397253036499,
+      "rewards/rejected": -1.1329768896102905,
       "step": 410
     },
     {
       "epoch": 0.43956043956043955,
-      "grad_norm": 23.667050291577752,
+      "grad_norm": 24.300028128821324,
       "learning_rate": 3.4408477372034736e-07,
-      "logits/chosen": -1.060620665550232,
-      "logits/rejected": -0.7473148703575134,
-      "logps/chosen": -372.11016845703125,
-      "logps/rejected": -397.57098388671875,
-      "loss": 0.5625,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -0.7644386291503906,
-      "rewards/margins": 0.6210776567459106,
-      "rewards/rejected": -1.3855164051055908,
+      "logits/chosen": -1.4640319347381592,
+      "logits/rejected": -1.2146576642990112,
+      "logps/chosen": -346.3941345214844,
+      "logps/rejected": -369.1965637207031,
+      "loss": 0.5549,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -0.5072788596153259,
+      "rewards/margins": 0.5944929122924805,
+      "rewards/rejected": -1.1017718315124512,
       "step": 420
     },
     {
       "epoch": 0.4500261643118786,
-      "grad_norm": 22.367345567184383,
+      "grad_norm": 22.44936030672002,
       "learning_rate": 3.3555276610977276e-07,
-      "logits/chosen": -0.7368607521057129,
-      "logits/rejected": -0.5510228276252747,
-      "logps/chosen": -337.81658935546875,
-      "logps/rejected": -393.0298767089844,
-      "loss": 0.5383,
+      "logits/chosen": -1.2797183990478516,
+      "logits/rejected": -1.1341623067855835,
+      "logps/chosen": -323.0084533691406,
+      "logps/rejected": -373.71795654296875,
+      "loss": 0.5437,
       "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -0.7567154765129089,
-      "rewards/margins": 0.6072173118591309,
-      "rewards/rejected": -1.3639328479766846,
+      "rewards/chosen": -0.6086335778236389,
+      "rewards/margins": 0.5621797442436218,
+      "rewards/rejected": -1.1708133220672607,
       "step": 430
     },
     {
       "epoch": 0.4604918890633176,
-      "grad_norm": 32.61524223055801,
+      "grad_norm": 58.53891780109035,
       "learning_rate": 3.269063392575352e-07,
-      "logits/chosen": -0.4659114480018616,
-      "logits/rejected": -0.17648424208164215,
-      "logps/chosen": -384.61651611328125,
-      "logps/rejected": -391.88592529296875,
-      "loss": 0.5551,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -0.8541596531867981,
-      "rewards/margins": 0.5109161138534546,
-      "rewards/rejected": -1.3650758266448975,
+      "logits/chosen": -1.0216747522354126,
+      "logits/rejected": -0.7915207743644714,
+      "logps/chosen": -390.4550476074219,
+      "logps/rejected": -403.381103515625,
+      "loss": 0.5516,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -0.9125455617904663,
+      "rewards/margins": 0.5674823522567749,
+      "rewards/rejected": -1.4800279140472412,
       "step": 440
     },
     {
       "epoch": 0.47095761381475665,
-      "grad_norm": 27.18058847122058,
+      "grad_norm": 34.41084494366662,
       "learning_rate": 3.1815705699316964e-07,
-      "logits/chosen": -0.3145579397678375,
-      "logits/rejected": -0.058880001306533813,
-      "logps/chosen": -379.8831481933594,
-      "logps/rejected": -463.6725158691406,
-      "loss": 0.5317,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -0.8658321499824524,
-      "rewards/margins": 0.936010479927063,
-      "rewards/rejected": -1.801842451095581,
+      "logits/chosen": -0.9631792306900024,
+      "logits/rejected": -0.775741457939148,
+      "logps/chosen": -385.7535705566406,
+      "logps/rejected": -470.589111328125,
+      "loss": 0.538,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -0.9245363473892212,
+      "rewards/margins": 0.9464725255966187,
+      "rewards/rejected": -1.8710088729858398,
       "step": 450
     },
     {
       "epoch": 0.48142333856619574,
-      "grad_norm": 25.220099383777104,
+      "grad_norm": 26.318881177937072,
       "learning_rate": 3.0931662070620794e-07,
-      "logits/chosen": 0.08404725790023804,
-      "logits/rejected": 0.3625319302082062,
-      "logps/chosen": -365.56976318359375,
-      "logps/rejected": -411.714599609375,
-      "loss": 0.5566,
-      "rewards/accuracies": 0.7124999761581421,
-      "rewards/chosen": -1.0015758275985718,
-      "rewards/margins": 0.6278172135353088,
-      "rewards/rejected": -1.6293929815292358,
+      "logits/chosen": -0.9639442563056946,
+      "logits/rejected": -0.7544153928756714,
+      "logps/chosen": -342.7520751953125,
+      "logps/rejected": -388.74725341796875,
+      "loss": 0.5584,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -0.773398756980896,
+      "rewards/margins": 0.6263211369514465,
+      "rewards/rejected": -1.3997199535369873,
       "step": 460
     },
     {
       "epoch": 0.49188906331763477,
-      "grad_norm": 24.952621546278483,
+      "grad_norm": 42.18792096989277,
       "learning_rate": 3.003968536966078e-07,
-      "logits/chosen": 0.24087226390838623,
-      "logits/rejected": 0.563568651676178,
-      "logps/chosen": -365.004150390625,
-      "logps/rejected": -389.58001708984375,
-      "loss": 0.5096,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.7347763776779175,
-      "rewards/margins": 0.6258446574211121,
-      "rewards/rejected": -1.3606209754943848,
+      "logits/chosen": -1.0974524021148682,
+      "logits/rejected": -0.9117683172225952,
+      "logps/chosen": -347.5935363769531,
+      "logps/rejected": -371.4789123535156,
+      "loss": 0.5098,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -0.5606697797775269,
+      "rewards/margins": 0.618939995765686,
+      "rewards/rejected": -1.179609775543213,
       "step": 470
     },
     {
       "epoch": 0.5023547880690737,
-      "grad_norm": 32.74383372318901,
+      "grad_norm": 26.43002026478988,
       "learning_rate": 2.9140968536213693e-07,
-      "logits/chosen": 1.1872409582138062,
-      "logits/rejected": 1.7373387813568115,
-      "logps/chosen": -370.4136657714844,
-      "logps/rejected": -445.4076232910156,
-      "loss": 0.5137,
-      "rewards/accuracies": 0.8125,
-      "rewards/chosen": -0.9462141990661621,
-      "rewards/margins": 0.9510065317153931,
-      "rewards/rejected": -1.8972208499908447,
+      "logits/chosen": -0.8168869018554688,
+      "logits/rejected": -0.45926633477211,
+      "logps/chosen": -346.2986145019531,
+      "logps/rejected": -415.69403076171875,
+      "loss": 0.5151,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -0.70506352186203,
+      "rewards/margins": 0.8950220942497253,
+      "rewards/rejected": -1.6000858545303345,
       "step": 480
     },
     {
       "epoch": 0.5128205128205128,
-      "grad_norm": 32.930715234525394,
+      "grad_norm": 26.075603530748978,
       "learning_rate": 2.823671352438608e-07,
-      "logits/chosen": 1.1654322147369385,
-      "logits/rejected": 2.1016147136688232,
-      "logps/chosen": -418.4208984375,
-      "logps/rejected": -474.35040283203125,
-      "loss": 0.5111,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -0.9660051465034485,
-      "rewards/margins": 1.0909435749053955,
-      "rewards/rejected": -2.056948661804199,
+      "logits/chosen": -0.8554686307907104,
+      "logits/rejected": -0.26327481865882874,
+      "logps/chosen": -393.8193664550781,
+      "logps/rejected": -441.4097595214844,
+      "loss": 0.5253,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -0.7199895977973938,
+      "rewards/margins": 1.007552146911621,
+      "rewards/rejected": -1.7275416851043701,
       "step": 490
     },
     {
       "epoch": 0.5232862375719518,
-      "grad_norm": 30.523627120283177,
+      "grad_norm": 46.13803118076279,
       "learning_rate": 2.73281296951072e-07,
-      "logits/chosen": 1.876829743385315,
-      "logits/rejected": 2.2034175395965576,
-      "logps/chosen": -365.1771545410156,
-      "logps/rejected": -405.43560791015625,
-      "loss": 0.5239,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -0.9771045446395874,
-      "rewards/margins": 0.7711528539657593,
-      "rewards/rejected": -1.7482573986053467,
+      "logits/chosen": -0.16928303241729736,
+      "logits/rejected": 0.03523734211921692,
+      "logps/chosen": -358.99627685546875,
+      "logps/rejected": -392.8810119628906,
+      "loss": 0.5352,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -0.9152957201004028,
+      "rewards/margins": 0.707416296005249,
+      "rewards/rejected": -1.6227118968963623,
       "step": 500
     },
     {
       "epoch": 0.5232862375719518,
-      "eval_logits/chosen": 1.5218729972839355,
-      "eval_logits/rejected": 1.9714024066925049,
-      "eval_logps/chosen": -391.44976806640625,
-      "eval_logps/rejected": -448.5180358886719,
-      "eval_loss": 0.528627336025238,
-      "eval_rewards/accuracies": 0.7559523582458496,
-      "eval_rewards/chosen": -0.997215211391449,
-      "eval_rewards/margins": 0.8264416456222534,
-      "eval_rewards/rejected": -1.8236567974090576,
-      "eval_runtime": 207.241,
-      "eval_samples_per_second": 9.651,
-      "eval_steps_per_second": 0.304,
+      "eval_logits/chosen": -0.40079495310783386,
+      "eval_logits/rejected": -0.09374210983514786,
+      "eval_logps/chosen": -384.1307067871094,
+      "eval_logps/rejected": -433.48065185546875,
+      "eval_loss": 0.5308576822280884,
+      "eval_rewards/accuracies": 0.7440476417541504,
+      "eval_rewards/chosen": -0.9240241050720215,
+      "eval_rewards/margins": 0.7492585182189941,
+      "eval_rewards/rejected": -1.6732827425003052,
+      "eval_runtime": 205.6895,
+      "eval_samples_per_second": 9.723,
+      "eval_steps_per_second": 0.306,
       "step": 500
     },
     {
       "epoch": 0.533751962323391,
-      "grad_norm": 27.81451202293309,
+      "grad_norm": 35.03408862664356,
       "learning_rate": 2.641643219871597e-07,
-      "logits/chosen": 1.548216462135315,
-      "logits/rejected": 1.9589998722076416,
-      "logps/chosen": -383.6624755859375,
-      "logps/rejected": -464.35198974609375,
-      "loss": 0.4951,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -0.9046643972396851,
-      "rewards/margins": 1.0409786701202393,
-      "rewards/rejected": -1.9456428289413452,
+      "logits/chosen": -0.4317537248134613,
+      "logits/rejected": -0.18362993001937866,
+      "logps/chosen": -378.89105224609375,
+      "logps/rejected": -445.74951171875,
+      "loss": 0.5042,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -0.8569504618644714,
+      "rewards/margins": 0.902667224407196,
+      "rewards/rejected": -1.759617805480957,
       "step": 510
     },
     {
       "epoch": 0.54421768707483,
-      "grad_norm": 29.340931752899976,
+      "grad_norm": 34.20713430220626,
       "learning_rate": 2.550284034980507e-07,
-      "logits/chosen": 1.7957836389541626,
-      "logits/rejected": 2.101381301879883,
-      "logps/chosen": -404.4179382324219,
-      "logps/rejected": -452.2720642089844,
-      "loss": 0.5414,
-      "rewards/accuracies": 0.6875,
-      "rewards/chosen": -1.2435704469680786,
-      "rewards/margins": 0.6854804754257202,
-      "rewards/rejected": -1.9290510416030884,
+      "logits/chosen": 0.1857350617647171,
+      "logits/rejected": 0.4090241491794586,
+      "logps/chosen": -419.68463134765625,
+      "logps/rejected": -458.61297607421875,
+      "loss": 0.5497,
+      "rewards/accuracies": 0.643750011920929,
+      "rewards/chosen": -1.3962366580963135,
+      "rewards/margins": 0.5962225198745728,
+      "rewards/rejected": -1.9924590587615967,
       "step": 520
     },
     {
       "epoch": 0.554683411826269,
-      "grad_norm": 46.912061432724556,
+      "grad_norm": 57.42602366755521,
       "learning_rate": 2.4588575996495794e-07,
-      "logits/chosen": 1.731397271156311,
-      "logits/rejected": 2.056797981262207,
-      "logps/chosen": -382.4450988769531,
-      "logps/rejected": -446.0044860839844,
-      "loss": 0.5253,
-      "rewards/accuracies": 0.71875,
-      "rewards/chosen": -1.051786184310913,
-      "rewards/margins": 0.7878361344337463,
-      "rewards/rejected": -1.8396222591400146,
+      "logits/chosen": 0.09937302768230438,
+      "logits/rejected": 0.3888925313949585,
+      "logps/chosen": -389.2012023925781,
+      "logps/rejected": -448.75775146484375,
+      "loss": 0.5306,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.1193468570709229,
+      "rewards/margins": 0.7478076219558716,
+      "rewards/rejected": -1.8671547174453735,
       "step": 530
     },
     {
       "epoch": 0.565149136577708,
-      "grad_norm": 24.92567213025273,
+      "grad_norm": 24.347954290505303,
       "learning_rate": 2.367486188632446e-07,
-      "logits/chosen": 1.5656790733337402,
-      "logits/rejected": 2.234431028366089,
-      "logps/chosen": -393.55535888671875,
-      "logps/rejected": -436.7002868652344,
-      "loss": 0.5315,
-      "rewards/accuracies": 0.7124999761581421,
-      "rewards/chosen": -0.944146990776062,
-      "rewards/margins": 0.7639015316963196,
-      "rewards/rejected": -1.7080484628677368,
+      "logits/chosen": -0.15624569356441498,
+      "logits/rejected": 0.2739402949810028,
+      "logps/chosen": -381.8554382324219,
+      "logps/rejected": -420.7781677246094,
+      "loss": 0.5451,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": -0.8271482586860657,
+      "rewards/margins": 0.7216792702674866,
+      "rewards/rejected": -1.5488274097442627,
       "step": 540
     },
     {
       "epoch": 0.5756148613291471,
-      "grad_norm": 32.54831254952225,
+      "grad_norm": 37.95198659343882,
       "learning_rate": 2.276292003092593e-07,
-      "logits/chosen": 1.5801063776016235,
-      "logits/rejected": 2.281773805618286,
-      "logps/chosen": -372.85711669921875,
-      "logps/rejected": -411.4371032714844,
-      "loss": 0.5253,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -0.9322255849838257,
-      "rewards/margins": 0.8505610227584839,
-      "rewards/rejected": -1.7827867269515991,
+      "logits/chosen": -0.1706000417470932,
+      "logits/rejected": 0.3523309826850891,
+      "logps/chosen": -358.3487854003906,
+      "logps/rejected": -394.71575927734375,
+      "loss": 0.5281,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -0.7871420979499817,
+      "rewards/margins": 0.8284309506416321,
+      "rewards/rejected": -1.6155731678009033,
       "step": 550
     },
     {
       "epoch": 0.5860805860805861,
-      "grad_norm": 29.085900080414557,
+      "grad_norm": 31.617763354602737,
       "learning_rate": 2.185397007170141e-07,
-      "logits/chosen": 1.6783803701400757,
-      "logits/rejected": 2.3324828147888184,
-      "logps/chosen": -387.82562255859375,
-      "logps/rejected": -441.27496337890625,
-      "loss": 0.5448,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -0.9643680453300476,
-      "rewards/margins": 0.8284353017807007,
-      "rewards/rejected": -1.792803168296814,
+      "logits/chosen": 0.26206716895103455,
+      "logits/rejected": 0.7556803822517395,
+      "logps/chosen": -376.7598571777344,
+      "logps/rejected": -428.24951171875,
+      "loss": 0.5501,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -0.8537102937698364,
+      "rewards/margins": 0.808838963508606,
+      "rewards/rejected": -1.6625492572784424,
       "step": 560
     },
     {
       "epoch": 0.5965463108320251,
-      "grad_norm": 31.928924412077674,
+      "grad_norm": 34.132968228975315,
       "learning_rate": 2.094922764865619e-07,
-      "logits/chosen": 1.5882809162139893,
-      "logits/rejected": 2.0225436687469482,
-      "logps/chosen": -413.67156982421875,
-      "logps/rejected": -468.8702087402344,
-      "loss": 0.5212,
-      "rewards/accuracies": 0.7562500238418579,
-      "rewards/chosen": -1.2697303295135498,
-      "rewards/margins": 0.6714531183242798,
-      "rewards/rejected": -1.9411834478378296,
+      "logits/chosen": 0.1466689556837082,
+      "logits/rejected": 0.5085517168045044,
+      "logps/chosen": -406.31854248046875,
+      "logps/rejected": -458.3421936035156,
+      "loss": 0.5302,
+      "rewards/accuracies": 0.75,
+      "rewards/chosen": -1.196199655532837,
+      "rewards/margins": 0.6397031545639038,
+      "rewards/rejected": -1.8359029293060303,
       "step": 570
     },
     {
       "epoch": 0.6070120355834642,
-      "grad_norm": 31.841767976153722,
+      "grad_norm": 32.91671633286688,
       "learning_rate": 2.0049902774588797e-07,
-      "logits/chosen": 1.5401804447174072,
-      "logits/rejected": 2.043485641479492,
-      "logps/chosen": -369.1571350097656,
-      "logps/rejected": -399.03717041015625,
-      "loss": 0.5354,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.0988614559173584,
-      "rewards/margins": 0.7443856000900269,
-      "rewards/rejected": -1.8432470560073853,
+      "logits/chosen": 0.3107803463935852,
+      "logits/rejected": 0.6928397417068481,
+      "logps/chosen": -365.2535400390625,
+      "logps/rejected": -391.4024963378906,
+      "loss": 0.54,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.0598257780075073,
+      "rewards/margins": 0.707074761390686,
+      "rewards/rejected": -1.766900658607483,
       "step": 580
     },
     {
       "epoch": 0.6174777603349032,
-      "grad_norm": 26.610645499683343,
+      "grad_norm": 29.0857374634859,
       "learning_rate": 1.9157198216806238e-07,
-      "logits/chosen": 1.2823454141616821,
-      "logits/rejected": 1.9630553722381592,
-      "logps/chosen": -396.398193359375,
-      "logps/rejected": -458.5870056152344,
-      "loss": 0.5034,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -1.0586693286895752,
-      "rewards/margins": 0.7922993898391724,
-      "rewards/rejected": -1.8509687185287476,
+      "logits/chosen": 0.2890316843986511,
+      "logits/rejected": 0.8040902018547058,
+      "logps/chosen": -397.47650146484375,
+      "logps/rejected": -457.5138244628906,
+      "loss": 0.5115,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": -1.0694525241851807,
+      "rewards/margins": 0.7707843780517578,
+      "rewards/rejected": -1.8402369022369385,
       "step": 590
     },
     {
       "epoch": 0.6279434850863422,
-      "grad_norm": 33.482178387507815,
+      "grad_norm": 36.666534677506704,
       "learning_rate": 1.8272307888529274e-07,
-      "logits/chosen": 2.0136606693267822,
-      "logits/rejected": 2.522876262664795,
-      "logps/chosen": -394.36505126953125,
-      "logps/rejected": -452.51336669921875,
-      "loss": 0.4866,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.0494329929351807,
-      "rewards/margins": 0.8478507995605469,
-      "rewards/rejected": -1.8972837924957275,
+      "logits/chosen": 1.0131226778030396,
+      "logits/rejected": 1.3981071710586548,
+      "logps/chosen": -401.26397705078125,
+      "logps/rejected": -456.7733459472656,
+      "loss": 0.4998,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -1.118422269821167,
+      "rewards/margins": 0.8214613199234009,
+      "rewards/rejected": -1.9398835897445679,
       "step": 600
     },
     {
       "epoch": 0.6279434850863422,
-      "eval_logits/chosen": 1.6720942258834839,
-      "eval_logits/rejected": 2.1912178993225098,
-      "eval_logps/chosen": -394.2553405761719,
-      "eval_logps/rejected": -454.10687255859375,
-      "eval_loss": 0.520341157913208,
-      "eval_rewards/accuracies": 0.7519841194152832,
-      "eval_rewards/chosen": -1.025270938873291,
-      "eval_rewards/margins": 0.8542740941047668,
-      "eval_rewards/rejected": -1.8795452117919922,
-      "eval_runtime": 207.1957,
-      "eval_samples_per_second": 9.653,
+      "eval_logits/chosen": 0.6984339356422424,
+      "eval_logits/rejected": 1.1205775737762451,
+      "eval_logps/chosen": -395.91461181640625,
+      "eval_logps/rejected": -453.32965087890625,
+      "eval_loss": 0.522526741027832,
+      "eval_rewards/accuracies": 0.7400793433189392,
+      "eval_rewards/chosen": -1.0418637990951538,
+      "eval_rewards/margins": 0.8299089670181274,
+      "eval_rewards/rejected": -1.8717727661132812,
+      "eval_runtime": 206.9196,
+      "eval_samples_per_second": 9.666,
       "eval_steps_per_second": 0.304,
       "step": 600
     },
     {
       "epoch": 0.6384092098377813,
-      "grad_norm": 31.33778693478196,
+      "grad_norm": 30.673794835032183,
       "learning_rate": 1.7396415252139288e-07,
-      "logits/chosen": 1.6208579540252686,
-      "logits/rejected": 2.4816088676452637,
-      "logps/chosen": -423.79443359375,
-      "logps/rejected": -451.82684326171875,
-      "loss": 0.5341,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -1.0654724836349487,
-      "rewards/margins": 0.9426978230476379,
-      "rewards/rejected": -2.0081706047058105,
+      "logits/chosen": 0.5988572835922241,
+      "logits/rejected": 1.299567461013794,
+      "logps/chosen": -417.8856506347656,
+      "logps/rejected": -446.78515625,
+      "loss": 0.5258,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.0063849687576294,
+      "rewards/margins": 0.9513689875602722,
+      "rewards/rejected": -1.9577538967132568,
       "step": 610
     },
     {
       "epoch": 0.6488749345892203,
-      "grad_norm": 30.140974859445116,
+      "grad_norm": 32.211994842231405,
       "learning_rate": 1.6530691736402316e-07,
-      "logits/chosen": 1.8116317987442017,
-      "logits/rejected": 2.2730700969696045,
-      "logps/chosen": -387.0520935058594,
-      "logps/rejected": -434.83612060546875,
-      "loss": 0.5046,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.0636460781097412,
-      "rewards/margins": 0.8480000495910645,
-      "rewards/rejected": -1.9116461277008057,
+      "logits/chosen": 0.850233256816864,
+      "logits/rejected": 1.220767617225647,
+      "logps/chosen": -387.00848388671875,
+      "logps/rejected": -435.15289306640625,
+      "loss": 0.4986,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.0632097721099854,
+      "rewards/margins": 0.8516036868095398,
+      "rewards/rejected": -1.9148136377334595,
       "step": 620
     },
     {
       "epoch": 0.6593406593406593,
-      "grad_norm": 33.62285572259581,
+      "grad_norm": 38.864669583620696,
       "learning_rate": 1.5676295169786864e-07,
-      "logits/chosen": 1.6578181982040405,
-      "logits/rejected": 2.3820550441741943,
-      "logps/chosen": -386.489501953125,
-      "logps/rejected": -429.51043701171875,
-      "loss": 0.497,
+      "logits/chosen": 0.809965968132019,
+      "logits/rejected": 1.443335771560669,
+      "logps/chosen": -389.06817626953125,
+      "logps/rejected": -432.38336181640625,
+      "loss": 0.4981,
       "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -1.0371458530426025,
-      "rewards/margins": 0.8595516085624695,
-      "rewards/rejected": -1.8966976404190063,
+      "rewards/chosen": -1.0629326105117798,
+      "rewards/margins": 0.8624936938285828,
+      "rewards/rejected": -1.9254264831542969,
       "step": 630
     },
     {
       "epoch": 0.6698063840920984,
-      "grad_norm": 31.957391763882473,
+      "grad_norm": 34.096975466021526,
       "learning_rate": 1.483436823197092e-07,
-      "logits/chosen": 2.231616497039795,
-      "logits/rejected": 2.8293519020080566,
-      "logps/chosen": -395.12152099609375,
-      "logps/rejected": -440.7001953125,
-      "loss": 0.4957,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -1.2094372510910034,
-      "rewards/margins": 0.8334163427352905,
-      "rewards/rejected": -2.042853593826294,
+      "logits/chosen": 1.4999452829360962,
+      "logits/rejected": 1.9908607006072998,
+      "logps/chosen": -405.83526611328125,
+      "logps/rejected": -450.565185546875,
+      "loss": 0.502,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -1.3165745735168457,
+      "rewards/margins": 0.8249285817146301,
+      "rewards/rejected": -2.141503095626831,
       "step": 640
     },
     {
       "epoch": 0.6802721088435374,
-      "grad_norm": 32.022547433163076,
+      "grad_norm": 34.771815421003645,
       "learning_rate": 1.4006036925609243e-07,
-      "logits/chosen": 1.8036184310913086,
-      "logits/rejected": 2.0799317359924316,
-      "logps/chosen": -370.6817321777344,
-      "logps/rejected": -439.3890686035156,
-      "loss": 0.52,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -0.9587554931640625,
-      "rewards/margins": 0.6661632657051086,
-      "rewards/rejected": -1.6249186992645264,
+      "logits/chosen": 1.224877953529358,
+      "logits/rejected": 1.449184775352478,
+      "logps/chosen": -389.7561340332031,
+      "logps/rejected": -456.34295654296875,
+      "loss": 0.517,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -1.1494994163513184,
+      "rewards/margins": 0.6449583768844604,
+      "rewards/rejected": -1.7944577932357788,
       "step": 650
     },
     {
       "epoch": 0.6907378335949764,
-      "grad_norm": 33.23280647021671,
+      "grad_norm": 31.45885241937768,
       "learning_rate": 1.319240907040458e-07,
-      "logits/chosen": 1.6538280248641968,
-      "logits/rejected": 2.349470376968384,
-      "logps/chosen": -375.78582763671875,
-      "logps/rejected": -441.38958740234375,
-      "loss": 0.5006,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -0.8550441861152649,
-      "rewards/margins": 0.9203449487686157,
-      "rewards/rejected": -1.7753890752792358,
+      "logits/chosen": 1.1873584985733032,
+      "logits/rejected": 1.763824462890625,
+      "logps/chosen": -402.32806396484375,
+      "logps/rejected": -468.66339111328125,
+      "loss": 0.4886,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -1.120466709136963,
+      "rewards/margins": 0.9276604652404785,
+      "rewards/rejected": -2.0481271743774414,
       "step": 660
     },
     {
       "epoch": 0.7012035583464155,
-      "grad_norm": 31.81372246417421,
+      "grad_norm": 42.31478816659526,
       "learning_rate": 1.239457282149695e-07,
-      "logits/chosen": 1.4791467189788818,
-      "logits/rejected": 1.979636788368225,
-      "logps/chosen": -403.3299865722656,
-      "logps/rejected": -485.5584411621094,
-      "loss": 0.4843,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.0501943826675415,
-      "rewards/margins": 0.8925808668136597,
-      "rewards/rejected": -1.9427751302719116,
+      "logits/chosen": 1.06121826171875,
+      "logits/rejected": 1.486613154411316,
+      "logps/chosen": -420.27947998046875,
+      "logps/rejected": -495.8140563964844,
+      "loss": 0.4935,
+      "rewards/accuracies": 0.768750011920929,
+      "rewards/chosen": -1.2196897268295288,
+      "rewards/margins": 0.8256422877311707,
+      "rewards/rejected": -2.045332193374634,
       "step": 670
     },
     {
       "epoch": 0.7116692830978545,
-      "grad_norm": 34.19119539742106,
+      "grad_norm": 46.494592998931914,
       "learning_rate": 1.1613595214152711e-07,
-      "logits/chosen": 1.9757497310638428,
-      "logits/rejected": 2.6055572032928467,
-      "logps/chosen": -429.9534606933594,
-      "logps/rejected": -494.7481384277344,
-      "loss": 0.5396,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -1.1138195991516113,
-      "rewards/margins": 0.896401584148407,
-      "rewards/rejected": -2.010221481323242,
+      "logits/chosen": 1.2637029886245728,
+      "logits/rejected": 1.7961336374282837,
+      "logps/chosen": -429.7030334472656,
+      "logps/rejected": -490.2943420410156,
+      "loss": 0.5403,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -1.1113157272338867,
+      "rewards/margins": 0.8543672561645508,
+      "rewards/rejected": -1.9656829833984375,
       "step": 680
     },
     {
       "epoch": 0.7221350078492935,
-      "grad_norm": 25.591304043868,
+      "grad_norm": 28.22622857193106,
       "learning_rate": 1.0850520736699362e-07,
-      "logits/chosen": 1.7879974842071533,
-      "logits/rejected": 2.690200090408325,
-      "logps/chosen": -396.08428955078125,
-      "logps/rejected": -439.63409423828125,
-      "loss": 0.479,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -1.122511863708496,
-      "rewards/margins": 0.8859044313430786,
-      "rewards/rejected": -2.0084164142608643,
+      "logits/chosen": 1.1677778959274292,
+      "logits/rejected": 1.9492908716201782,
+      "logps/chosen": -397.92999267578125,
+      "logps/rejected": -434.52227783203125,
+      "loss": 0.4864,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.1409690380096436,
+      "rewards/margins": 0.8163288235664368,
+      "rewards/rejected": -1.9572980403900146,
       "step": 690
     },
     {
       "epoch": 0.7326007326007326,
-      "grad_norm": 34.46436848925927,
+      "grad_norm": 38.82903353111915,
       "learning_rate": 1.0106369933615042e-07,
-      "logits/chosen": 2.035653591156006,
-      "logits/rejected": 2.4581329822540283,
-      "logps/chosen": -372.4128723144531,
-      "logps/rejected": -428.73114013671875,
-      "loss": 0.4883,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -1.0989868640899658,
-      "rewards/margins": 0.7625324130058289,
-      "rewards/rejected": -1.86151921749115,
+      "logits/chosen": 1.49738347530365,
+      "logits/rejected": 1.8964507579803467,
+      "logps/chosen": -387.61126708984375,
+      "logps/rejected": -441.9359436035156,
+      "loss": 0.4936,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.250970482826233,
+      "rewards/margins": 0.7425969839096069,
+      "rewards/rejected": -1.9935674667358398,
       "step": 700
     },
     {
       "epoch": 0.7326007326007326,
-      "eval_logits/chosen": 1.878947377204895,
-      "eval_logits/rejected": 2.4788594245910645,
-      "eval_logps/chosen": -399.6694030761719,
-      "eval_logps/rejected": -462.8506164550781,
-      "eval_loss": 0.5169664025306702,
-      "eval_rewards/accuracies": 0.77182537317276,
-      "eval_rewards/chosen": -1.079411506652832,
-      "eval_rewards/margins": 0.8875707983970642,
-      "eval_rewards/rejected": -1.9669822454452515,
-      "eval_runtime": 206.5859,
-      "eval_samples_per_second": 9.681,
-      "eval_steps_per_second": 0.305,
+      "eval_logits/chosen": 1.4183756113052368,
+      "eval_logits/rejected": 1.9631398916244507,
+      "eval_logps/chosen": -412.60626220703125,
+      "eval_logps/rejected": -475.4923400878906,
+      "eval_loss": 0.5196009874343872,
+      "eval_rewards/accuracies": 0.7539682388305664,
+      "eval_rewards/chosen": -1.2087799310684204,
+      "eval_rewards/margins": 0.8846201300621033,
+      "eval_rewards/rejected": -2.0933997631073,
+      "eval_runtime": 206.1126,
+      "eval_samples_per_second": 9.703,
+      "eval_steps_per_second": 0.306,
       "step": 700
     },
     {
       "epoch": 0.7430664573521716,
-      "grad_norm": 34.11388019674118,
+      "grad_norm": 41.9438808858937,
       "learning_rate": 9.382138040640714e-08,
-      "logits/chosen": 2.2550535202026367,
-      "logits/rejected": 2.922189712524414,
-      "logps/chosen": -391.513671875,
-      "logps/rejected": -455.7442932128906,
-      "loss": 0.5537,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -1.2671797275543213,
-      "rewards/margins": 0.7816148996353149,
-      "rewards/rejected": -2.0487945079803467,
+      "logits/chosen": 1.7710202932357788,
+      "logits/rejected": 2.3763861656188965,
+      "logps/chosen": -406.9272155761719,
+      "logps/rejected": -463.2557067871094,
+      "loss": 0.5672,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -1.4213149547576904,
+      "rewards/margins": 0.7025940418243408,
+      "rewards/rejected": -2.1239089965820312,
       "step": 710
     },
     {
       "epoch": 0.7535321821036107,
-      "grad_norm": 34.82671206419468,
+      "grad_norm": 48.08337125318838,
       "learning_rate": 8.678793653740632e-08,
-      "logits/chosen": 2.182706117630005,
-      "logits/rejected": 2.837254047393799,
-      "logps/chosen": -403.3385314941406,
-      "logps/rejected": -465.86492919921875,
-      "loss": 0.4892,
-      "rewards/accuracies": 0.7437499761581421,
-      "rewards/chosen": -1.1041263341903687,
-      "rewards/margins": 1.0244553089141846,
-      "rewards/rejected": -2.1285815238952637,
+      "logits/chosen": 1.6012312173843384,
+      "logits/rejected": 2.216918468475342,
+      "logps/chosen": -419.3685607910156,
+      "logps/rejected": -477.35003662109375,
+      "loss": 0.5009,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.2644264698028564,
+      "rewards/margins": 0.9790066480636597,
+      "rewards/rejected": -2.2434329986572266,
       "step": 720
     },
     {
       "epoch": 0.7639979068550498,
-      "grad_norm": 35.53491885012393,
+      "grad_norm": 40.621592704466096,
       "learning_rate": 7.997277433690983e-08,
-      "logits/chosen": 2.3328869342803955,
-      "logits/rejected": 2.962507724761963,
-      "logps/chosen": -407.1231689453125,
-      "logps/rejected": -445.3768615722656,
-      "loss": 0.5213,
-      "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -1.3389291763305664,
-      "rewards/margins": 0.6952264904975891,
-      "rewards/rejected": -2.0341556072235107,
+      "logits/chosen": 1.654722809791565,
+      "logits/rejected": 2.163034677505493,
+      "logps/chosen": -413.20355224609375,
+      "logps/rejected": -446.3681640625,
+      "loss": 0.5238,
+      "rewards/accuracies": 0.6812499761581421,
+      "rewards/chosen": -1.399733543395996,
+      "rewards/margins": 0.6443352699279785,
+      "rewards/rejected": -2.0440685749053955,
       "step": 730
     },
     {
       "epoch": 0.7744636316064888,
-      "grad_norm": 29.972105990799896,
+      "grad_norm": 44.11198771694061,
       "learning_rate": 7.338500848029602e-08,
-      "logits/chosen": 2.8226966857910156,
-      "logits/rejected": 3.1715874671936035,
-      "logps/chosen": -380.815185546875,
-      "logps/rejected": -427.7220153808594,
-      "loss": 0.5094,
-      "rewards/accuracies": 0.699999988079071,
-      "rewards/chosen": -1.2078065872192383,
-      "rewards/margins": 0.7211803197860718,
-      "rewards/rejected": -1.9289871454238892,
+      "logits/chosen": 2.0758748054504395,
+      "logits/rejected": 2.3728959560394287,
+      "logps/chosen": -388.3576965332031,
+      "logps/rejected": -435.0791015625,
+      "loss": 0.5122,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -1.283231496810913,
+      "rewards/margins": 0.7193259000778198,
+      "rewards/rejected": -2.0025575160980225,
       "step": 740
     },
     {
       "epoch": 0.7849293563579278,
-      "grad_norm": 41.54151649066564,
+      "grad_norm": 39.45060030130625,
       "learning_rate": 6.70334495204884e-08,
-      "logits/chosen": 2.574078321456909,
-      "logits/rejected": 3.0375800132751465,
-      "logps/chosen": -385.7585754394531,
-      "logps/rejected": -465.62158203125,
-      "loss": 0.4929,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -1.298154592514038,
-      "rewards/margins": 0.868002712726593,
-      "rewards/rejected": -2.1661572456359863,
+      "logits/chosen": 1.8493086099624634,
+      "logits/rejected": 2.2490105628967285,
+      "logps/chosen": -393.843994140625,
+      "logps/rejected": -475.54132080078125,
+      "loss": 0.4984,
+      "rewards/accuracies": 0.75,
+      "rewards/chosen": -1.3790090084075928,
+      "rewards/margins": 0.8863463401794434,
+      "rewards/rejected": -2.265355348587036,
       "step": 750
     },
     {
       "epoch": 0.7953950811093669,
-      "grad_norm": 31.71564718135222,
+      "grad_norm": 41.13021738331374,
       "learning_rate": 6.092659210462231e-08,
-      "logits/chosen": 2.7071144580841064,
-      "logits/rejected": 3.148503065109253,
-      "logps/chosen": -371.3905334472656,
-      "logps/rejected": -459.02056884765625,
-      "loss": 0.5208,
-      "rewards/accuracies": 0.7562500238418579,
-      "rewards/chosen": -1.218873143196106,
-      "rewards/margins": 0.8540975451469421,
-      "rewards/rejected": -2.0729706287384033,
+      "logits/chosen": 1.9165847301483154,
+      "logits/rejected": 2.292691707611084,
+      "logps/chosen": -383.63818359375,
+      "logps/rejected": -474.6698303222656,
+      "loss": 0.5255,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -1.3413493633270264,
+      "rewards/margins": 0.8881141543388367,
+      "rewards/rejected": -2.229463577270508,
       "step": 760
     },
     {
       "epoch": 0.8058608058608059,
-      "grad_norm": 23.11012449409936,
+      "grad_norm": 23.227510052162174,
       "learning_rate": 5.507260361320737e-08,
-      "logits/chosen": 2.2510199546813965,
-      "logits/rejected": 2.30385160446167,
-      "logps/chosen": -430.4512634277344,
-      "logps/rejected": -524.1280517578125,
-      "loss": 0.4812,
+      "logits/chosen": 1.4680651426315308,
+      "logits/rejected": 1.5410151481628418,
+      "logps/chosen": -441.0107421875,
+      "logps/rejected": -535.9873657226562,
+      "loss": 0.4871,
       "rewards/accuracies": 0.706250011920929,
-      "rewards/chosen": -1.1405556201934814,
-      "rewards/margins": 0.765708327293396,
-      "rewards/rejected": -1.9062637090682983,
+      "rewards/chosen": -1.246150255203247,
+      "rewards/margins": 0.7787067294120789,
+      "rewards/rejected": -2.0248570442199707,
       "step": 770
     },
     {
       "epoch": 0.8163265306122449,
-      "grad_norm": 38.441651434975974,
+      "grad_norm": 40.34737389909196,
       "learning_rate": 4.947931323697982e-08,
-      "logits/chosen": 2.170044183731079,
-      "logits/rejected": 2.846287727355957,
-      "logps/chosen": -433.65606689453125,
-      "logps/rejected": -482.6353454589844,
-      "loss": 0.5037,
-      "rewards/accuracies": 0.7437499761581421,
-      "rewards/chosen": -1.0754050016403198,
-      "rewards/margins": 0.941206157207489,
-      "rewards/rejected": -2.0166115760803223,
+      "logits/chosen": 1.382004976272583,
+      "logits/rejected": 1.967145562171936,
+      "logps/chosen": -438.35150146484375,
+      "logps/rejected": -489.5238342285156,
+      "loss": 0.5137,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -1.1223596334457397,
+      "rewards/margins": 0.9631366729736328,
+      "rewards/rejected": -2.085496425628662,
       "step": 780
     },
     {
       "epoch": 0.826792255363684,
-      "grad_norm": 33.89230518812285,
+      "grad_norm": 40.27301508170132,
       "learning_rate": 4.415420150605398e-08,
-      "logits/chosen": 2.4628593921661377,
-      "logits/rejected": 3.4102680683135986,
-      "logps/chosen": -407.9674377441406,
-      "logps/rejected": -460.8292541503906,
-      "loss": 0.5319,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -1.2123013734817505,
-      "rewards/margins": 0.8870647549629211,
-      "rewards/rejected": -2.0993659496307373,
+      "logits/chosen": 1.519224762916565,
+      "logits/rejected": 2.341888904571533,
+      "logps/chosen": -414.23681640625,
+      "logps/rejected": -462.8794860839844,
+      "loss": 0.5392,
+      "rewards/accuracies": 0.737500011920929,
+      "rewards/chosen": -1.2749950885772705,
+      "rewards/margins": 0.8448735475540161,
+      "rewards/rejected": -2.119868516921997,
       "step": 790
     },
     {
       "epoch": 0.837257980115123,
-      "grad_norm": 30.213940104849513,
+      "grad_norm": 45.75902031403285,
       "learning_rate": 3.9104390285376374e-08,
-      "logits/chosen": 2.8522162437438965,
-      "logits/rejected": 3.206580400466919,
-      "logps/chosen": -409.0374755859375,
-      "logps/rejected": -480.83160400390625,
-      "loss": 0.5062,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -1.3300718069076538,
-      "rewards/margins": 0.7746312022209167,
-      "rewards/rejected": -2.104702949523926,
+      "logits/chosen": 1.8431495428085327,
+      "logits/rejected": 2.1607654094696045,
+      "logps/chosen": -412.042236328125,
+      "logps/rejected": -483.38787841796875,
+      "loss": 0.5125,
+      "rewards/accuracies": 0.7124999761581421,
+      "rewards/chosen": -1.3601198196411133,
+      "rewards/margins": 0.7701460719108582,
+      "rewards/rejected": -2.130265712738037,
       "step": 800
     },
     {
       "epoch": 0.837257980115123,
-      "eval_logits/chosen": 2.3343281745910645,
-      "eval_logits/rejected": 2.937108039855957,
-      "eval_logps/chosen": -408.60736083984375,
-      "eval_logps/rejected": -472.77459716796875,
-      "eval_loss": 0.513859212398529,
-      "eval_rewards/accuracies": 0.7678571343421936,
-      "eval_rewards/chosen": -1.1687912940979004,
-      "eval_rewards/margins": 0.8974311351776123,
-      "eval_rewards/rejected": -2.0662224292755127,
-      "eval_runtime": 207.3208,
-      "eval_samples_per_second": 9.647,
-      "eval_steps_per_second": 0.304,
+      "eval_logits/chosen": 1.3314255475997925,
+      "eval_logits/rejected": 1.873736023902893,
+      "eval_logps/chosen": -410.9924011230469,
+      "eval_logps/rejected": -473.3307189941406,
+      "eval_loss": 0.5170913338661194,
+      "eval_rewards/accuracies": 0.7658730149269104,
+      "eval_rewards/chosen": -1.1926417350769043,
+      "eval_rewards/margins": 0.8791419267654419,
+      "eval_rewards/rejected": -2.0717833042144775,
+      "eval_runtime": 206.6421,
+      "eval_samples_per_second": 9.679,
+      "eval_steps_per_second": 0.305,
       "step": 800
     },
     {
       "epoch": 0.847723704866562,
-      "grad_norm": 27.802052074609342,
+      "grad_norm": 32.21448583582019,
       "learning_rate": 3.433663324986208e-08,
-      "logits/chosen": 2.4477035999298096,
-      "logits/rejected": 3.375871181488037,
-      "logps/chosen": -420.7499084472656,
-      "logps/rejected": -460.06390380859375,
-      "loss": 0.5102,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -1.202946662902832,
-      "rewards/margins": 0.8900796175003052,
-      "rewards/rejected": -2.0930261611938477,
+      "logits/chosen": 1.3725779056549072,
+      "logits/rejected": 2.195798873901367,
+      "logps/chosen": -422.95086669921875,
+      "logps/rejected": -460.2098083496094,
+      "loss": 0.5186,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.224956750869751,
+      "rewards/margins": 0.8695290684700012,
+      "rewards/rejected": -2.0944857597351074,
       "step": 810
     },
     {
       "epoch": 0.858189429618001,
-      "grad_norm": 31.601124922739412,
+      "grad_norm": 32.56199170937255,
       "learning_rate": 2.9857306851953897e-08,
-      "logits/chosen": 2.4745118618011475,
-      "logits/rejected": 2.945732831954956,
-      "logps/chosen": -422.23492431640625,
-      "logps/rejected": -486.2650451660156,
-      "loss": 0.5107,
-      "rewards/accuracies": 0.7124999761581421,
-      "rewards/chosen": -1.0861989259719849,
-      "rewards/margins": 0.8801456689834595,
-      "rewards/rejected": -1.9663444757461548,
+      "logits/chosen": 1.3711284399032593,
+      "logits/rejected": 1.8220380544662476,
+      "logps/chosen": -428.7027893066406,
+      "logps/rejected": -492.301025390625,
+      "loss": 0.5116,
+      "rewards/accuracies": 0.75,
+      "rewards/chosen": -1.1508777141571045,
+      "rewards/margins": 0.8758268356323242,
+      "rewards/rejected": -2.0267043113708496,
       "step": 820
     },
     {
       "epoch": 0.8686551543694401,
-      "grad_norm": 31.69445271337221,
+      "grad_norm": 30.701473639362177,
       "learning_rate": 2.567240179368185e-08,
-      "logits/chosen": 2.5528483390808105,
-      "logits/rejected": 2.929966449737549,
-      "logps/chosen": -386.4670104980469,
-      "logps/rejected": -459.9654846191406,
-      "loss": 0.5211,
-      "rewards/accuracies": 0.7124999761581421,
-      "rewards/chosen": -1.2057501077651978,
-      "rewards/margins": 0.8482575416564941,
-      "rewards/rejected": -2.0540077686309814,
+      "logits/chosen": 1.444504976272583,
+      "logits/rejected": 1.7912876605987549,
+      "logps/chosen": -389.9057312011719,
+      "logps/rejected": -459.4881286621094,
+      "loss": 0.5196,
+      "rewards/accuracies": 0.6937500238418579,
+      "rewards/chosen": -1.2401378154754639,
+      "rewards/margins": 0.8090965151786804,
+      "rewards/rejected": -2.04923415184021,
       "step": 830
     },
     {
       "epoch": 0.8791208791208791,
-      "grad_norm": 28.37397907273284,
+      "grad_norm": 35.827147674342314,
       "learning_rate": 2.1787515014630357e-08,
-      "logits/chosen": 2.4352076053619385,
-      "logits/rejected": 3.2364249229431152,
-      "logps/chosen": -401.13031005859375,
-      "logps/rejected": -472.4002380371094,
-      "loss": 0.4836,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -1.1297938823699951,
-      "rewards/margins": 0.9000727534294128,
-      "rewards/rejected": -2.0298666954040527,
+      "logits/chosen": 1.3566821813583374,
+      "logits/rejected": 2.0197525024414062,
+      "logps/chosen": -408.1112365722656,
+      "logps/rejected": -478.15362548828125,
+      "loss": 0.4883,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -1.1996033191680908,
+      "rewards/margins": 0.8877967596054077,
+      "rewards/rejected": -2.087400197982788,
       "step": 840
     },
     {
       "epoch": 0.8895866038723181,
-      "grad_norm": 31.189751260957607,
+      "grad_norm": 31.663518232503517,
       "learning_rate": 1.820784220652766e-08,
-      "logits/chosen": 2.8544576168060303,
-      "logits/rejected": 3.519139051437378,
-      "logps/chosen": -401.4136962890625,
-      "logps/rejected": -444.1220703125,
-      "loss": 0.4733,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -1.1122504472732544,
-      "rewards/margins": 0.9509198069572449,
-      "rewards/rejected": -2.0631701946258545,
+      "logits/chosen": 1.6553951501846313,
+      "logits/rejected": 2.2624051570892334,
+      "logps/chosen": -405.8297424316406,
+      "logps/rejected": -444.5301818847656,
+      "loss": 0.4817,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.1564102172851562,
+      "rewards/margins": 0.9108415842056274,
+      "rewards/rejected": -2.067251682281494,
       "step": 850
     },
     {
       "epoch": 0.9000523286237572,
-      "grad_norm": 34.378886842289475,
+      "grad_norm": 37.26032497580661,
       "learning_rate": 1.4938170864468636e-08,
-      "logits/chosen": 2.683777332305908,
-      "logits/rejected": 3.4396414756774902,
-      "logps/chosen": -414.46282958984375,
-      "logps/rejected": -469.6669006347656,
-      "loss": 0.491,
-      "rewards/accuracies": 0.7875000238418579,
-      "rewards/chosen": -1.1743658781051636,
-      "rewards/margins": 0.8312996029853821,
-      "rewards/rejected": -2.0056653022766113,
+      "logits/chosen": 1.5842053890228271,
+      "logits/rejected": 2.242150068283081,
+      "logps/chosen": -416.7059020996094,
+      "logps/rejected": -471.46600341796875,
+      "loss": 0.4916,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.1967964172363281,
+      "rewards/margins": 0.8268603086471558,
+      "rewards/rejected": -2.0236568450927734,
       "step": 860
     },
     {
       "epoch": 0.9105180533751962,
-      "grad_norm": 33.77517436077429,
+      "grad_norm": 35.19582912121324,
       "learning_rate": 1.1982873884064465e-08,
-      "logits/chosen": 2.289219856262207,
-      "logits/rejected": 3.060136079788208,
-      "logps/chosen": -391.9703063964844,
-      "logps/rejected": -453.79864501953125,
-      "loss": 0.5145,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.1336462497711182,
-      "rewards/margins": 0.8525140881538391,
-      "rewards/rejected": -1.9861602783203125,
+      "logits/chosen": 1.2853723764419556,
+      "logits/rejected": 1.9532840251922607,
+      "logps/chosen": -392.630615234375,
+      "logps/rejected": -457.26348876953125,
+      "loss": 0.5073,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.140249490737915,
+      "rewards/margins": 0.8805596232414246,
+      "rewards/rejected": -2.0208089351654053,
       "step": 870
     },
     {
       "epoch": 0.9209837781266352,
-      "grad_norm": 30.19295068520873,
+      "grad_norm": 34.02189943600299,
       "learning_rate": 9.345903713082304e-09,
-      "logits/chosen": 2.3843894004821777,
-      "logits/rejected": 2.856482982635498,
-      "logps/chosen": -426.60205078125,
-      "logps/rejected": -496.14471435546875,
-      "loss": 0.5064,
-      "rewards/accuracies": 0.7437499761581421,
-      "rewards/chosen": -1.111513614654541,
-      "rewards/margins": 0.8207988739013672,
-      "rewards/rejected": -1.9323123693466187,
+      "logits/chosen": 1.409638524055481,
+      "logits/rejected": 1.783506989479065,
+      "logps/chosen": -433.36419677734375,
+      "logps/rejected": -498.5658264160156,
+      "loss": 0.51,
+      "rewards/accuracies": 0.699999988079071,
+      "rewards/chosen": -1.1791355609893799,
+      "rewards/margins": 0.7773876190185547,
+      "rewards/rejected": -1.9565229415893555,
       "step": 880
     },
     {
       "epoch": 0.9314495028780743,
-      "grad_norm": 105.03195364677694,
+      "grad_norm": 101.02453554023373,
       "learning_rate": 7.030787065396865e-09,
-      "logits/chosen": 2.2952702045440674,
-      "logits/rejected": 2.8261852264404297,
-      "logps/chosen": -415.36846923828125,
-      "logps/rejected": -494.0677795410156,
-      "loss": 0.5213,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -1.1505730152130127,
-      "rewards/margins": 0.8346196413040161,
-      "rewards/rejected": -1.9851925373077393,
+      "logits/chosen": 1.3969112634658813,
+      "logits/rejected": 1.868054747581482,
+      "logps/chosen": -416.21783447265625,
+      "logps/rejected": -492.66448974609375,
+      "loss": 0.5231,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -1.1590664386749268,
+      "rewards/margins": 0.812092661857605,
+      "rewards/rejected": -1.9711593389511108,
       "step": 890
     },
     {
       "epoch": 0.9419152276295133,
-      "grad_norm": 31.985249424086373,
+      "grad_norm": 33.05314756036431,
       "learning_rate": 5.04062020432286e-09,
-      "logits/chosen": 2.2670741081237793,
-      "logits/rejected": 2.6167309284210205,
-      "logps/chosen": -403.3427429199219,
-      "logps/rejected": -484.1851501464844,
-      "loss": 0.4881,
-      "rewards/accuracies": 0.737500011920929,
-      "rewards/chosen": -1.1589634418487549,
-      "rewards/margins": 0.7817413806915283,
-      "rewards/rejected": -1.9407049417495728,
+      "logits/chosen": 1.3796522617340088,
+      "logits/rejected": 1.653877854347229,
+      "logps/chosen": -404.7085876464844,
+      "logps/rejected": -482.3396911621094,
+      "loss": 0.4976,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -1.1726219654083252,
+      "rewards/margins": 0.7496272921562195,
+      "rewards/rejected": -1.9222491979599,
       "step": 900
     },
     {
       "epoch": 0.9419152276295133,
-      "eval_logits/chosen": 2.312872886657715,
-      "eval_logits/rejected": 2.927537202835083,
-      "eval_logps/chosen": -403.3655090332031,
-      "eval_logps/rejected": -467.9211730957031,
-      "eval_loss": 0.5130186676979065,
-      "eval_rewards/accuracies": 0.7678571343421936,
-      "eval_rewards/chosen": -1.1163722276687622,
-      "eval_rewards/margins": 0.9013155698776245,
-      "eval_rewards/rejected": -2.0176877975463867,
-      "eval_runtime": 206.8444,
-      "eval_samples_per_second": 9.669,
+      "eval_logits/chosen": 1.3479634523391724,
+      "eval_logits/rejected": 1.9026191234588623,
+      "eval_logps/chosen": -408.19427490234375,
+      "eval_logps/rejected": -470.8255310058594,
+      "eval_loss": 0.5151652693748474,
+      "eval_rewards/accuracies": 0.7599206566810608,
+      "eval_rewards/chosen": -1.1646604537963867,
+      "eval_rewards/margins": 0.8820713758468628,
+      "eval_rewards/rejected": -2.046732187271118,
+      "eval_runtime": 206.2937,
+      "eval_samples_per_second": 9.695,
       "eval_steps_per_second": 0.305,
       "step": 900
     },
     {
       "epoch": 0.9523809523809523,
-      "grad_norm": 35.487123394466245,
+      "grad_norm": 37.360762359784026,
       "learning_rate": 3.3780648016376866e-09,
-      "logits/chosen": 2.7584078311920166,
-      "logits/rejected": 3.168657064437866,
-      "logps/chosen": -337.892333984375,
-      "logps/rejected": -405.7276306152344,
-      "loss": 0.4997,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -1.0817266702651978,
-      "rewards/margins": 0.8273773193359375,
-      "rewards/rejected": -1.9091039896011353,
+      "logits/chosen": 1.7493425607681274,
+      "logits/rejected": 2.125293016433716,
+      "logps/chosen": -343.41607666015625,
+      "logps/rejected": -408.11895751953125,
+      "loss": 0.5068,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.1369637250900269,
+      "rewards/margins": 0.7960535883903503,
+      "rewards/rejected": -1.933017373085022,
       "step": 910
     },
     {
       "epoch": 0.9628466771323915,
-      "grad_norm": 36.09113184040287,
+      "grad_norm": 34.128880581694396,
       "learning_rate": 2.0453443778310766e-09,
-      "logits/chosen": 2.4235446453094482,
-      "logits/rejected": 3.1566996574401855,
-      "logps/chosen": -397.71240234375,
-      "logps/rejected": -446.8587341308594,
-      "loss": 0.5145,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -1.066408395767212,
-      "rewards/margins": 0.881396472454071,
-      "rewards/rejected": -1.9478048086166382,
+      "logits/chosen": 1.4454774856567383,
+      "logits/rejected": 2.106161594390869,
+      "logps/chosen": -403.2652893066406,
+      "logps/rejected": -456.05999755859375,
+      "loss": 0.5096,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -1.1219370365142822,
+      "rewards/margins": 0.9178807139396667,
+      "rewards/rejected": -2.0398173332214355,
       "step": 920
     },
     {
       "epoch": 0.9733124018838305,
-      "grad_norm": 37.77419919647761,
+      "grad_norm": 33.912988919352415,
       "learning_rate": 1.0442413283435758e-09,
-      "logits/chosen": 2.266357898712158,
-      "logits/rejected": 3.2177586555480957,
-      "logps/chosen": -427.22735595703125,
-      "logps/rejected": -462.27874755859375,
-      "loss": 0.5019,
-      "rewards/accuracies": 0.71875,
-      "rewards/chosen": -1.0245699882507324,
-      "rewards/margins": 0.867609977722168,
-      "rewards/rejected": -1.89218008518219,
+      "logits/chosen": 1.3229658603668213,
+      "logits/rejected": 2.128333330154419,
+      "logps/chosen": -431.3890075683594,
+      "logps/rejected": -464.32781982421875,
+      "loss": 0.5007,
+      "rewards/accuracies": 0.7250000238418579,
+      "rewards/chosen": -1.0661871433258057,
+      "rewards/margins": 0.8464835286140442,
+      "rewards/rejected": -1.9126704931259155,
       "step": 930
     },
     {
       "epoch": 0.9837781266352695,
-      "grad_norm": 31.024160932578784,
+      "grad_norm": 32.48448141098046,
       "learning_rate": 3.760945397705828e-10,
-      "logits/chosen": 2.280590534210205,
-      "logits/rejected": 2.8673043251037598,
-      "logps/chosen": -373.6363525390625,
-      "logps/rejected": -450.34417724609375,
-      "loss": 0.5014,
-      "rewards/accuracies": 0.7250000238418579,
-      "rewards/chosen": -1.0496900081634521,
-      "rewards/margins": 0.8248756527900696,
-      "rewards/rejected": -1.874565839767456,
+      "logits/chosen": 1.3461906909942627,
+      "logits/rejected": 1.898982048034668,
+      "logps/chosen": -381.6216125488281,
+      "logps/rejected": -456.62713623046875,
+      "loss": 0.5001,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -1.1295430660247803,
+      "rewards/margins": 0.8078519701957703,
+      "rewards/rejected": -1.9373950958251953,
       "step": 940
     },
     {
       "epoch": 0.9942438513867086,
-      "grad_norm": 34.67550870249938,
+      "grad_norm": 33.82160017136217,
       "learning_rate": 4.17975992204056e-11,
-      "logits/chosen": 2.359349250793457,
-      "logits/rejected": 2.7258801460266113,
-      "logps/chosen": -427.2079162597656,
-      "logps/rejected": -476.568115234375,
-      "loss": 0.5018,
+      "logits/chosen": 1.398242712020874,
+      "logits/rejected": 1.7302888631820679,
+      "logps/chosen": -433.99249267578125,
+      "logps/rejected": -482.1703186035156,
+      "loss": 0.5035,
       "rewards/accuracies": 0.71875,
-      "rewards/chosen": -1.1586647033691406,
-      "rewards/margins": 0.6969943046569824,
-      "rewards/rejected": -1.8556588888168335,
+      "rewards/chosen": -1.2265104055404663,
+      "rewards/margins": 0.6851701140403748,
+      "rewards/rejected": -1.9116805791854858,
       "step": 950
     },
     {
       "epoch": 0.9994767137624281,
       "step": 955,
       "total_flos": 0.0,
-      "train_loss": 0.5488577563101085,
-      "train_runtime": 17633.9629,
-      "train_samples_per_second": 3.467,
-      "train_steps_per_second": 0.054
+      "train_loss": 0.5508741353818883,
+      "train_runtime": 17482.7224,
+      "train_samples_per_second": 3.497,
+      "train_steps_per_second": 0.055
     }
   ],
   "logging_steps": 10,