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.0010471204188481676,
-      "grad_norm": 1.806436074491372,
+      "grad_norm": 1.8064054769306461,
       "learning_rate": 1.7421602787456448e-07,
       "logits/chosen": -11.625,
       "logits/rejected": -11.75,
@@ -25,59 +25,59 @@
     },
     {
       "epoch": 0.010471204188481676,
-      "grad_norm": 1.753031830811777,
+      "grad_norm": 1.7542618091363311,
       "learning_rate": 1.7421602787456445e-06,
       "logits/chosen": -11.125,
       "logits/rejected": -10.5,
       "logps/chosen": -428.0,
       "logps/rejected": -756.0,
-      "loss": 0.6922,
-      "rewards/accuracies": 0.1527777761220932,
-      "rewards/chosen": 0.00015354156494140625,
-      "rewards/margins": 0.000335693359375,
-      "rewards/rejected": -0.00017833709716796875,
+      "loss": 0.6925,
+      "rewards/accuracies": 0.1319444477558136,
+      "rewards/chosen": -0.00046539306640625,
+      "rewards/margins": -0.00125885009765625,
+      "rewards/rejected": 0.00079345703125,
       "step": 10
     },
     {
       "epoch": 0.020942408376963352,
-      "grad_norm": 1.6525738053961263,
+      "grad_norm": 1.6408655968895125,
       "learning_rate": 3.484320557491289e-06,
       "logits/chosen": -11.0,
       "logits/rejected": -10.8125,
       "logps/chosen": -402.0,
       "logps/rejected": -576.0,
-      "loss": 0.6891,
+      "loss": 0.6883,
       "rewards/accuracies": 0.3499999940395355,
-      "rewards/chosen": -0.00055694580078125,
-      "rewards/margins": 0.00958251953125,
-      "rewards/rejected": -0.0101318359375,
+      "rewards/chosen": -0.00110626220703125,
+      "rewards/margins": 0.009765625,
+      "rewards/rejected": -0.0108642578125,
       "step": 20
     },
     {
       "epoch": 0.031413612565445025,
-      "grad_norm": 1.6752503762489155,
+      "grad_norm": 1.6772908345612645,
       "learning_rate": 5.226480836236934e-06,
       "logits/chosen": -11.625,
       "logits/rejected": -11.375,
-      "logps/chosen": -430.0,
+      "logps/chosen": -428.0,
       "logps/rejected": -532.0,
-      "loss": 0.6757,
-      "rewards/accuracies": 0.6000000238418579,
-      "rewards/chosen": -0.025634765625,
-      "rewards/margins": 0.02783203125,
-      "rewards/rejected": -0.053466796875,
+      "loss": 0.6755,
+      "rewards/accuracies": 0.574999988079071,
+      "rewards/chosen": -0.0252685546875,
+      "rewards/margins": 0.0264892578125,
+      "rewards/rejected": -0.0517578125,
       "step": 30
     },
     {
       "epoch": 0.041884816753926704,
-      "grad_norm": 1.965171751722186,
+      "grad_norm": 1.962821083552404,
       "learning_rate": 6.968641114982578e-06,
       "logits/chosen": -12.4375,
       "logits/rejected": -12.0,
       "logps/chosen": -448.0,
       "logps/rejected": -616.0,
-      "loss": 0.6364,
-      "rewards/accuracies": 0.6312500238418579,
+      "loss": 0.6368,
+      "rewards/accuracies": 0.625,
       "rewards/chosen": -0.189453125,
       "rewards/margins": 0.1220703125,
       "rewards/rejected": -0.3125,
@@ -85,4690 +85,4690 @@
     },
     {
       "epoch": 0.05235602094240838,
-      "grad_norm": 2.0214971099991823,
+      "grad_norm": 2.0215182953382964,
       "learning_rate": 8.710801393728225e-06,
       "logits/chosen": -12.1875,
       "logits/rejected": -12.0,
       "logps/chosen": -462.0,
       "logps/rejected": -700.0,
-      "loss": 0.5849,
-      "rewards/accuracies": 0.6625000238418579,
-      "rewards/chosen": -0.5078125,
+      "loss": 0.5846,
+      "rewards/accuracies": 0.675000011920929,
+      "rewards/chosen": -0.51171875,
       "rewards/margins": 0.419921875,
-      "rewards/rejected": -0.92578125,
+      "rewards/rejected": -0.9296875,
       "step": 50
     },
     {
       "epoch": 0.06282722513089005,
-      "grad_norm": 1.8574919158967238,
+      "grad_norm": 1.8626006846203826,
       "learning_rate": 1.0452961672473868e-05,
       "logits/chosen": -11.0,
       "logits/rejected": -10.9375,
-      "logps/chosen": -398.0,
+      "logps/chosen": -400.0,
       "logps/rejected": -692.0,
-      "loss": 0.5398,
-      "rewards/accuracies": 0.6937500238418579,
-      "rewards/chosen": -0.5,
+      "loss": 0.5394,
+      "rewards/accuracies": 0.6875,
+      "rewards/chosen": -0.50390625,
       "rewards/margins": 0.76171875,
       "rewards/rejected": -1.265625,
       "step": 60
     },
     {
       "epoch": 0.07329842931937172,
-      "grad_norm": 2.0180102172020864,
+      "grad_norm": 2.028796629399926,
       "learning_rate": 1.2195121951219513e-05,
       "logits/chosen": -12.0,
-      "logits/rejected": -11.6875,
+      "logits/rejected": -11.75,
       "logps/chosen": -516.0,
       "logps/rejected": -848.0,
-      "loss": 0.5346,
+      "loss": 0.5344,
       "rewards/accuracies": 0.706250011920929,
       "rewards/chosen": -0.81640625,
-      "rewards/margins": 0.99609375,
-      "rewards/rejected": -1.8125,
+      "rewards/margins": 1.0,
+      "rewards/rejected": -1.8203125,
       "step": 70
     },
     {
       "epoch": 0.08376963350785341,
-      "grad_norm": 2.118796993161168,
+      "grad_norm": 2.1328975360078974,
       "learning_rate": 1.3937282229965156e-05,
-      "logits/chosen": -12.3125,
+      "logits/chosen": -12.375,
       "logits/rejected": -12.1875,
       "logps/chosen": -506.0,
       "logps/rejected": -920.0,
-      "loss": 0.4982,
-      "rewards/accuracies": 0.7562500238418579,
-      "rewards/chosen": -1.0859375,
-      "rewards/margins": 1.5234375,
+      "loss": 0.4979,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -1.078125,
+      "rewards/margins": 1.53125,
       "rewards/rejected": -2.609375,
       "step": 80
     },
     {
       "epoch": 0.09424083769633508,
-      "grad_norm": 3.4881945422416933,
+      "grad_norm": 3.338986896006448,
       "learning_rate": 1.56794425087108e-05,
-      "logits/chosen": -12.0625,
-      "logits/rejected": -11.8125,
+      "logits/chosen": -12.125,
+      "logits/rejected": -11.875,
       "logps/chosen": -510.0,
       "logps/rejected": -724.0,
-      "loss": 0.5279,
-      "rewards/accuracies": 0.7124999761581421,
+      "loss": 0.5297,
+      "rewards/accuracies": 0.731249988079071,
       "rewards/chosen": -1.4921875,
-      "rewards/margins": 0.98828125,
+      "rewards/margins": 0.984375,
       "rewards/rejected": -2.484375,
       "step": 90
     },
     {
       "epoch": 0.10471204188481675,
-      "grad_norm": 2.152922060914835,
+      "grad_norm": 2.118774691989547,
       "learning_rate": 1.742160278745645e-05,
-      "logits/chosen": -11.8125,
-      "logits/rejected": -11.5625,
-      "logps/chosen": -660.0,
-      "logps/rejected": -960.0,
-      "loss": 0.5108,
-      "rewards/accuracies": 0.731249988079071,
-      "rewards/chosen": -1.8203125,
-      "rewards/margins": 1.1640625,
-      "rewards/rejected": -2.984375,
+      "logits/chosen": -11.5625,
+      "logits/rejected": -11.3125,
+      "logps/chosen": -668.0,
+      "logps/rejected": -964.0,
+      "loss": 0.5172,
+      "rewards/accuracies": 0.71875,
+      "rewards/chosen": -1.890625,
+      "rewards/margins": 1.1328125,
+      "rewards/rejected": -3.015625,
       "step": 100
     },
     {
       "epoch": 0.10471204188481675,
-      "eval_logits/chosen": -10.8125,
-      "eval_logits/rejected": -10.375,
-      "eval_logps/chosen": -442.0,
-      "eval_logps/rejected": -434.0,
-      "eval_loss": 0.6707968711853027,
-      "eval_rewards/accuracies": 0.609375,
-      "eval_rewards/chosen": -1.234375,
-      "eval_rewards/margins": 0.2177734375,
-      "eval_rewards/rejected": -1.453125,
-      "eval_runtime": 49.1287,
-      "eval_samples_per_second": 40.709,
-      "eval_steps_per_second": 0.651,
+      "eval_logits/chosen": -10.3125,
+      "eval_logits/rejected": -9.875,
+      "eval_logps/chosen": -458.0,
+      "eval_logps/rejected": -450.0,
+      "eval_loss": 0.664312481880188,
+      "eval_rewards/accuracies": 0.583984375,
+      "eval_rewards/chosen": -1.3984375,
+      "eval_rewards/margins": 0.2109375,
+      "eval_rewards/rejected": -1.6015625,
+      "eval_runtime": 48.0159,
+      "eval_samples_per_second": 41.653,
+      "eval_steps_per_second": 0.666,
       "step": 100
     },
     {
       "epoch": 0.11518324607329843,
-      "grad_norm": 2.7152160967946757,
+      "grad_norm": 2.498798874485681,
       "learning_rate": 1.9163763066202093e-05,
       "logits/chosen": -10.1875,
       "logits/rejected": -9.9375,
-      "logps/chosen": -588.0,
-      "logps/rejected": -912.0,
-      "loss": 0.4537,
-      "rewards/accuracies": 0.762499988079071,
-      "rewards/chosen": -1.703125,
-      "rewards/margins": 1.390625,
-      "rewards/rejected": -3.09375,
+      "logps/chosen": -576.0,
+      "logps/rejected": -900.0,
+      "loss": 0.4564,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -1.6171875,
+      "rewards/margins": 1.3984375,
+      "rewards/rejected": -3.0,
       "step": 110
     },
     {
       "epoch": 0.1256544502617801,
-      "grad_norm": 2.8078148154437925,
+      "grad_norm": 2.6828716333332543,
       "learning_rate": 2.0905923344947736e-05,
-      "logits/chosen": -9.375,
-      "logits/rejected": -9.3125,
-      "logps/chosen": -592.0,
-      "logps/rejected": -884.0,
-      "loss": 0.479,
-      "rewards/accuracies": 0.7875000238418579,
-      "rewards/chosen": -2.28125,
-      "rewards/margins": 1.296875,
-      "rewards/rejected": -3.578125,
+      "logits/chosen": -10.8125,
+      "logits/rejected": -10.8125,
+      "logps/chosen": -560.0,
+      "logps/rejected": -860.0,
+      "loss": 0.4882,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -1.96875,
+      "rewards/margins": 1.390625,
+      "rewards/rejected": -3.359375,
       "step": 120
     },
     {
       "epoch": 0.13612565445026178,
-      "grad_norm": 2.35281267807467,
+      "grad_norm": 2.515832633348086,
       "learning_rate": 2.264808362369338e-05,
-      "logits/chosen": -10.0625,
-      "logits/rejected": -10.0625,
-      "logps/chosen": -572.0,
-      "logps/rejected": -864.0,
-      "loss": 0.4769,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -1.6171875,
-      "rewards/margins": 1.109375,
-      "rewards/rejected": -2.71875,
+      "logits/chosen": -10.5625,
+      "logits/rejected": -10.5625,
+      "logps/chosen": -616.0,
+      "logps/rejected": -916.0,
+      "loss": 0.4672,
+      "rewards/accuracies": 0.768750011920929,
+      "rewards/chosen": -2.09375,
+      "rewards/margins": 1.171875,
+      "rewards/rejected": -3.265625,
       "step": 130
     },
     {
       "epoch": 0.14659685863874344,
-      "grad_norm": 2.972375209292521,
+      "grad_norm": 3.426140215005646,
       "learning_rate": 2.4390243902439026e-05,
-      "logits/chosen": -13.0625,
-      "logits/rejected": -12.8125,
-      "logps/chosen": -652.0,
-      "logps/rejected": -1004.0,
-      "loss": 0.4693,
-      "rewards/accuracies": 0.7875000238418579,
-      "rewards/chosen": -2.046875,
-      "rewards/margins": 1.4609375,
-      "rewards/rejected": -3.515625,
+      "logits/chosen": -13.375,
+      "logits/rejected": -13.1875,
+      "logps/chosen": -648.0,
+      "logps/rejected": -996.0,
+      "loss": 0.4736,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -2.03125,
+      "rewards/margins": 1.3984375,
+      "rewards/rejected": -3.4375,
       "step": 140
     },
     {
       "epoch": 0.15706806282722513,
-      "grad_norm": 2.7041561232033717,
+      "grad_norm": 2.3700312822607876,
       "learning_rate": 2.6132404181184672e-05,
-      "logits/chosen": -12.0,
-      "logits/rejected": -11.75,
-      "logps/chosen": -676.0,
-      "logps/rejected": -928.0,
-      "loss": 0.479,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -2.5,
-      "rewards/margins": 1.25,
-      "rewards/rejected": -3.75,
+      "logits/chosen": -11.6875,
+      "logits/rejected": -11.3125,
+      "logps/chosen": -628.0,
+      "logps/rejected": -880.0,
+      "loss": 0.4847,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -2.046875,
+      "rewards/margins": 1.2109375,
+      "rewards/rejected": -3.265625,
       "step": 150
     },
     {
       "epoch": 0.16753926701570682,
-      "grad_norm": 2.087713233064345,
+      "grad_norm": 2.340866477630366,
       "learning_rate": 2.7874564459930312e-05,
-      "logits/chosen": -11.1875,
-      "logits/rejected": -10.6875,
-      "logps/chosen": -660.0,
-      "logps/rejected": -932.0,
-      "loss": 0.4574,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -2.65625,
-      "rewards/margins": 1.265625,
-      "rewards/rejected": -3.921875,
+      "logits/chosen": -10.9375,
+      "logits/rejected": -10.375,
+      "logps/chosen": -608.0,
+      "logps/rejected": -884.0,
+      "loss": 0.4388,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -2.15625,
+      "rewards/margins": 1.28125,
+      "rewards/rejected": -3.4375,
       "step": 160
     },
     {
       "epoch": 0.17801047120418848,
-      "grad_norm": 2.381916331427287,
+      "grad_norm": 2.869031752062845,
       "learning_rate": 2.9616724738675962e-05,
-      "logits/chosen": -9.125,
-      "logits/rejected": -8.8125,
-      "logps/chosen": -724.0,
-      "logps/rejected": -1176.0,
-      "loss": 0.4194,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -2.984375,
-      "rewards/margins": 2.28125,
-      "rewards/rejected": -5.28125,
+      "logits/chosen": -8.5625,
+      "logits/rejected": -8.25,
+      "logps/chosen": -760.0,
+      "logps/rejected": -1200.0,
+      "loss": 0.4221,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -3.34375,
+      "rewards/margins": 2.1875,
+      "rewards/rejected": -5.53125,
       "step": 170
     },
     {
       "epoch": 0.18848167539267016,
-      "grad_norm": 2.527678455386163,
+      "grad_norm": 2.5731576053726934,
       "learning_rate": 3.13588850174216e-05,
-      "logits/chosen": -9.0,
-      "logits/rejected": -8.8125,
-      "logps/chosen": -724.0,
-      "logps/rejected": -1128.0,
+      "logits/chosen": -6.65625,
+      "logits/rejected": -6.46875,
+      "logps/chosen": -752.0,
+      "logps/rejected": -1152.0,
       "loss": 0.4248,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -2.875,
-      "rewards/margins": 2.109375,
-      "rewards/rejected": -4.96875,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -3.140625,
+      "rewards/margins": 2.0625,
+      "rewards/rejected": -5.21875,
       "step": 180
     },
     {
       "epoch": 0.19895287958115182,
-      "grad_norm": 3.03370138884052,
+      "grad_norm": 2.8656716776227964,
       "learning_rate": 3.310104529616725e-05,
-      "logits/chosen": -9.1875,
-      "logits/rejected": -9.125,
-      "logps/chosen": -760.0,
-      "logps/rejected": -1152.0,
-      "loss": 0.4235,
-      "rewards/accuracies": 0.78125,
-      "rewards/chosen": -3.640625,
-      "rewards/margins": 1.9140625,
-      "rewards/rejected": -5.5625,
+      "logits/chosen": -6.25,
+      "logits/rejected": -6.5,
+      "logps/chosen": -748.0,
+      "logps/rejected": -1128.0,
+      "loss": 0.4232,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -3.5,
+      "rewards/margins": 1.7890625,
+      "rewards/rejected": -5.28125,
       "step": 190
     },
     {
       "epoch": 0.2094240837696335,
-      "grad_norm": 2.275931638108211,
+      "grad_norm": 2.530505765588831,
       "learning_rate": 3.48432055749129e-05,
-      "logits/chosen": -10.5,
-      "logits/rejected": -10.125,
-      "logps/chosen": -756.0,
+      "logits/chosen": -8.4375,
+      "logits/rejected": -7.84375,
+      "logps/chosen": -748.0,
       "logps/rejected": -1080.0,
-      "loss": 0.4452,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -3.046875,
-      "rewards/margins": 1.4921875,
-      "rewards/rejected": -4.53125,
+      "loss": 0.4478,
+      "rewards/accuracies": 0.768750011920929,
+      "rewards/chosen": -2.953125,
+      "rewards/margins": 1.640625,
+      "rewards/rejected": -4.59375,
       "step": 200
     },
     {
       "epoch": 0.2094240837696335,
-      "eval_logits/chosen": -10.1875,
-      "eval_logits/rejected": -9.625,
-      "eval_logps/chosen": -600.0,
-      "eval_logps/rejected": -616.0,
-      "eval_loss": 0.7220468521118164,
-      "eval_rewards/accuracies": 0.6171875,
-      "eval_rewards/chosen": -2.828125,
-      "eval_rewards/margins": 0.443359375,
-      "eval_rewards/rejected": -3.28125,
-      "eval_runtime": 47.0795,
-      "eval_samples_per_second": 42.481,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -9.75,
+      "eval_logits/rejected": -9.25,
+      "eval_logps/chosen": -620.0,
+      "eval_logps/rejected": -636.0,
+      "eval_loss": 0.7484687566757202,
+      "eval_rewards/accuracies": 0.599609375,
+      "eval_rewards/chosen": -3.0,
+      "eval_rewards/margins": 0.47265625,
+      "eval_rewards/rejected": -3.46875,
+      "eval_runtime": 46.5502,
+      "eval_samples_per_second": 42.964,
+      "eval_steps_per_second": 0.687,
       "step": 200
     },
     {
       "epoch": 0.2198952879581152,
-      "grad_norm": 2.790099646654149,
+      "grad_norm": 2.5726793486723563,
       "learning_rate": 3.6585365853658535e-05,
-      "logits/chosen": -10.25,
-      "logits/rejected": -9.9375,
-      "logps/chosen": -624.0,
-      "logps/rejected": -880.0,
-      "loss": 0.4422,
-      "rewards/accuracies": 0.78125,
-      "rewards/chosen": -2.609375,
-      "rewards/margins": 1.15625,
-      "rewards/rejected": -3.765625,
+      "logits/chosen": -10.0,
+      "logits/rejected": -9.6875,
+      "logps/chosen": -616.0,
+      "logps/rejected": -868.0,
+      "loss": 0.451,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -2.53125,
+      "rewards/margins": 1.1328125,
+      "rewards/rejected": -3.65625,
       "step": 210
     },
     {
       "epoch": 0.23036649214659685,
-      "grad_norm": 2.4445867175858362,
+      "grad_norm": 2.4247863648839316,
       "learning_rate": 3.8327526132404185e-05,
-      "logits/chosen": -9.8125,
-      "logits/rejected": -9.3125,
-      "logps/chosen": -736.0,
-      "logps/rejected": -1136.0,
-      "loss": 0.4087,
-      "rewards/accuracies": 0.856249988079071,
-      "rewards/chosen": -3.15625,
-      "rewards/margins": 1.90625,
-      "rewards/rejected": -5.0625,
+      "logits/chosen": -9.25,
+      "logits/rejected": -8.75,
+      "logps/chosen": -712.0,
+      "logps/rejected": -1104.0,
+      "loss": 0.4163,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -2.921875,
+      "rewards/margins": 1.8203125,
+      "rewards/rejected": -4.75,
       "step": 220
     },
     {
       "epoch": 0.24083769633507854,
-      "grad_norm": 2.189536210082279,
+      "grad_norm": 4.510318426525849,
       "learning_rate": 4.006968641114983e-05,
-      "logits/chosen": -9.6875,
-      "logits/rejected": -9.625,
-      "logps/chosen": -716.0,
-      "logps/rejected": -952.0,
-      "loss": 0.4481,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -3.265625,
-      "rewards/margins": 1.3515625,
-      "rewards/rejected": -4.625,
+      "logits/chosen": -8.5,
+      "logits/rejected": -8.625,
+      "logps/chosen": -812.0,
+      "logps/rejected": -1056.0,
+      "loss": 0.4442,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -4.21875,
+      "rewards/margins": 1.40625,
+      "rewards/rejected": -5.625,
       "step": 230
     },
     {
       "epoch": 0.2513089005235602,
-      "grad_norm": 4.83195323954807,
+      "grad_norm": 3.551529834152487,
       "learning_rate": 4.181184668989547e-05,
-      "logits/chosen": -8.9375,
-      "logits/rejected": -8.375,
-      "logps/chosen": -760.0,
-      "logps/rejected": -1128.0,
-      "loss": 0.4543,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -3.875,
-      "rewards/margins": 2.09375,
-      "rewards/rejected": -6.0,
+      "logits/chosen": -8.5625,
+      "logits/rejected": -8.125,
+      "logps/chosen": -976.0,
+      "logps/rejected": -1336.0,
+      "loss": 0.4881,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -6.03125,
+      "rewards/margins": 2.03125,
+      "rewards/rejected": -8.0625,
       "step": 240
     },
     {
       "epoch": 0.2617801047120419,
-      "grad_norm": 2.9857595735325306,
+      "grad_norm": 3.1490996821133366,
       "learning_rate": 4.3554006968641115e-05,
-      "logits/chosen": -10.9375,
-      "logits/rejected": -10.5,
-      "logps/chosen": -728.0,
-      "logps/rejected": -1032.0,
-      "loss": 0.4357,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -2.75,
-      "rewards/margins": 1.609375,
-      "rewards/rejected": -4.34375,
+      "logits/chosen": -7.4375,
+      "logits/rejected": -7.125,
+      "logps/chosen": -716.0,
+      "logps/rejected": -1004.0,
+      "loss": 0.4411,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -2.625,
+      "rewards/margins": 1.4453125,
+      "rewards/rejected": -4.0625,
       "step": 250
     },
     {
       "epoch": 0.27225130890052357,
-      "grad_norm": 3.059164866934553,
+      "grad_norm": 1.8906480446452494,
       "learning_rate": 4.529616724738676e-05,
-      "logits/chosen": -11.3125,
-      "logits/rejected": -10.6875,
-      "logps/chosen": -668.0,
-      "logps/rejected": -1064.0,
-      "loss": 0.4133,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -2.90625,
-      "rewards/margins": 2.046875,
-      "rewards/rejected": -4.9375,
+      "logits/chosen": -6.9375,
+      "logits/rejected": -6.46875,
+      "logps/chosen": -620.0,
+      "logps/rejected": -1024.0,
+      "loss": 0.4106,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -2.453125,
+      "rewards/margins": 2.109375,
+      "rewards/rejected": -4.5625,
       "step": 260
     },
     {
       "epoch": 0.28272251308900526,
-      "grad_norm": 1.846300303313416,
+      "grad_norm": 2.7556469662874203,
       "learning_rate": 4.703832752613241e-05,
-      "logits/chosen": -10.0,
-      "logits/rejected": -9.625,
-      "logps/chosen": -736.0,
-      "logps/rejected": -1136.0,
-      "loss": 0.4582,
-      "rewards/accuracies": 0.768750011920929,
-      "rewards/chosen": -3.640625,
-      "rewards/margins": 1.8984375,
-      "rewards/rejected": -5.53125,
+      "logits/chosen": -9.25,
+      "logits/rejected": -9.0625,
+      "logps/chosen": -636.0,
+      "logps/rejected": -1056.0,
+      "loss": 0.4461,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -2.625,
+      "rewards/margins": 2.109375,
+      "rewards/rejected": -4.75,
       "step": 270
     },
     {
       "epoch": 0.2931937172774869,
-      "grad_norm": 2.5014181154607797,
+      "grad_norm": 2.03749442932726,
       "learning_rate": 4.878048780487805e-05,
-      "logits/chosen": -9.625,
-      "logits/rejected": -9.4375,
-      "logps/chosen": -696.0,
-      "logps/rejected": -1020.0,
-      "loss": 0.4562,
-      "rewards/accuracies": 0.8374999761581421,
-      "rewards/chosen": -3.375,
-      "rewards/margins": 1.484375,
-      "rewards/rejected": -4.84375,
+      "logits/chosen": -9.375,
+      "logits/rejected": -9.25,
+      "logps/chosen": -572.0,
+      "logps/rejected": -872.0,
+      "loss": 0.4817,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -2.109375,
+      "rewards/margins": 1.265625,
+      "rewards/rejected": -3.375,
       "step": 280
     },
     {
       "epoch": 0.3036649214659686,
-      "grad_norm": 2.3279926698388334,
+      "grad_norm": 2.6279902280720364,
       "learning_rate": 4.999983293481418e-05,
-      "logits/chosen": -8.9375,
-      "logits/rejected": -9.0625,
-      "logps/chosen": -820.0,
-      "logps/rejected": -1184.0,
-      "loss": 0.4449,
-      "rewards/accuracies": 0.78125,
-      "rewards/chosen": -4.34375,
-      "rewards/margins": 1.6015625,
-      "rewards/rejected": -5.96875,
+      "logits/chosen": -10.625,
+      "logits/rejected": -10.4375,
+      "logps/chosen": -616.0,
+      "logps/rejected": -980.0,
+      "loss": 0.4371,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -2.328125,
+      "rewards/margins": 1.5703125,
+      "rewards/rejected": -3.90625,
       "step": 290
     },
     {
       "epoch": 0.31413612565445026,
-      "grad_norm": 2.8143471281423187,
+      "grad_norm": 3.769418383316172,
       "learning_rate": 4.9996862949181595e-05,
-      "logits/chosen": -9.3125,
-      "logits/rejected": -8.75,
+      "logits/chosen": -8.6875,
+      "logits/rejected": -7.84375,
       "logps/chosen": -760.0,
-      "logps/rejected": -1120.0,
-      "loss": 0.4197,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -3.515625,
-      "rewards/margins": 1.8828125,
-      "rewards/rejected": -5.40625,
+      "logps/rejected": -1152.0,
+      "loss": 0.4548,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -3.53125,
+      "rewards/margins": 2.15625,
+      "rewards/rejected": -5.6875,
       "step": 300
     },
     {
       "epoch": 0.31413612565445026,
-      "eval_logits/chosen": -8.375,
-      "eval_logits/rejected": -7.5,
-      "eval_logps/chosen": -716.0,
-      "eval_logps/rejected": -748.0,
-      "eval_loss": 0.7965937256813049,
-      "eval_rewards/accuracies": 0.60546875,
-      "eval_rewards/chosen": -3.984375,
-      "eval_rewards/margins": 0.61328125,
-      "eval_rewards/rejected": -4.59375,
-      "eval_runtime": 47.0746,
-      "eval_samples_per_second": 42.486,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -9.0625,
+      "eval_logits/rejected": -8.25,
+      "eval_logps/chosen": -640.0,
+      "eval_logps/rejected": -652.0,
+      "eval_loss": 0.8017343878746033,
+      "eval_rewards/accuracies": 0.55859375,
+      "eval_rewards/chosen": -3.203125,
+      "eval_rewards/margins": 0.439453125,
+      "eval_rewards/rejected": -3.640625,
+      "eval_runtime": 46.5667,
+      "eval_samples_per_second": 42.949,
+      "eval_steps_per_second": 0.687,
       "step": 300
     },
     {
       "epoch": 0.32460732984293195,
-      "grad_norm": 3.3412549989746747,
+      "grad_norm": 2.266710520096018,
       "learning_rate": 4.9990180911526896e-05,
-      "logits/chosen": -8.75,
-      "logits/rejected": -8.375,
-      "logps/chosen": -868.0,
-      "logps/rejected": -1232.0,
-      "loss": 0.4075,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -4.96875,
-      "rewards/margins": 1.890625,
-      "rewards/rejected": -6.84375,
+      "logits/chosen": -10.0625,
+      "logits/rejected": -9.4375,
+      "logps/chosen": -680.0,
+      "logps/rejected": -1012.0,
+      "loss": 0.4244,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -3.109375,
+      "rewards/margins": 1.5859375,
+      "rewards/rejected": -4.6875,
       "step": 310
     },
     {
       "epoch": 0.33507853403141363,
-      "grad_norm": 2.190128732250711,
+      "grad_norm": 2.1542280893435914,
       "learning_rate": 4.997978781413745e-05,
       "logits/chosen": -9.5625,
       "logits/rejected": -9.1875,
-      "logps/chosen": -920.0,
-      "logps/rejected": -1232.0,
-      "loss": 0.4571,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.78125,
-      "rewards/margins": 1.9453125,
-      "rewards/rejected": -6.71875,
+      "logps/chosen": -792.0,
+      "logps/rejected": -1096.0,
+      "loss": 0.4623,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -3.46875,
+      "rewards/margins": 1.90625,
+      "rewards/rejected": -5.375,
       "step": 320
     },
     {
       "epoch": 0.34554973821989526,
-      "grad_norm": 3.536254358142373,
+      "grad_norm": 3.6034963009052707,
       "learning_rate": 4.996568520039565e-05,
-      "logits/chosen": -11.3125,
-      "logits/rejected": -10.6875,
-      "logps/chosen": -704.0,
-      "logps/rejected": -1104.0,
-      "loss": 0.4034,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -2.921875,
-      "rewards/margins": 2.078125,
-      "rewards/rejected": -5.0,
+      "logits/chosen": -10.5625,
+      "logits/rejected": -10.1875,
+      "logps/chosen": -648.0,
+      "logps/rejected": -1040.0,
+      "loss": 0.4152,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -2.359375,
+      "rewards/margins": 2.03125,
+      "rewards/rejected": -4.375,
       "step": 330
     },
     {
       "epoch": 0.35602094240837695,
-      "grad_norm": 2.893960585920598,
+      "grad_norm": 3.523383233256517,
       "learning_rate": 4.994787516454973e-05,
-      "logits/chosen": -11.5625,
-      "logits/rejected": -10.875,
-      "logps/chosen": -800.0,
-      "logps/rejected": -1096.0,
-      "loss": 0.447,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.03125,
-      "rewards/margins": 1.8203125,
-      "rewards/rejected": -5.84375,
+      "logits/chosen": -9.5,
+      "logits/rejected": -9.1875,
+      "logps/chosen": -720.0,
+      "logps/rejected": -1032.0,
+      "loss": 0.4566,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -3.25,
+      "rewards/margins": 1.9140625,
+      "rewards/rejected": -5.15625,
       "step": 340
     },
     {
       "epoch": 0.36649214659685864,
-      "grad_norm": 3.96725204340208,
+      "grad_norm": 3.317972825593402,
       "learning_rate": 4.992636035140273e-05,
-      "logits/chosen": -13.4375,
-      "logits/rejected": -12.8125,
-      "logps/chosen": -796.0,
-      "logps/rejected": -1248.0,
-      "loss": 0.4506,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -4.3125,
-      "rewards/margins": 2.21875,
-      "rewards/rejected": -6.53125,
+      "logits/chosen": -9.4375,
+      "logits/rejected": -9.125,
+      "logps/chosen": -664.0,
+      "logps/rejected": -1104.0,
+      "loss": 0.4604,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -2.953125,
+      "rewards/margins": 2.078125,
+      "rewards/rejected": -5.03125,
       "step": 350
     },
     {
       "epoch": 0.3769633507853403,
-      "grad_norm": 2.8513490162823105,
+      "grad_norm": 2.568077835049425,
       "learning_rate": 4.99011439559198e-05,
-      "logits/chosen": -12.5,
-      "logits/rejected": -11.875,
-      "logps/chosen": -792.0,
-      "logps/rejected": -1072.0,
-      "loss": 0.4222,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -3.90625,
-      "rewards/margins": 1.4296875,
-      "rewards/rejected": -5.34375,
+      "logits/chosen": -9.3125,
+      "logits/rejected": -8.6875,
+      "logps/chosen": -768.0,
+      "logps/rejected": -1104.0,
+      "loss": 0.3842,
+      "rewards/accuracies": 0.84375,
+      "rewards/chosen": -3.65625,
+      "rewards/margins": 1.953125,
+      "rewards/rejected": -5.59375,
       "step": 360
     },
     {
       "epoch": 0.387434554973822,
-      "grad_norm": 3.302024051268013,
+      "grad_norm": 3.370030371364723,
       "learning_rate": 4.987222972275369e-05,
-      "logits/chosen": -9.8125,
-      "logits/rejected": -8.8125,
-      "logps/chosen": -956.0,
-      "logps/rejected": -1384.0,
-      "loss": 0.3879,
-      "rewards/accuracies": 0.8500000238418579,
-      "rewards/chosen": -5.65625,
-      "rewards/margins": 2.671875,
-      "rewards/rejected": -8.3125,
+      "logits/chosen": -8.375,
+      "logits/rejected": -7.78125,
+      "logps/chosen": -880.0,
+      "logps/rejected": -1256.0,
+      "loss": 0.3781,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -4.875,
+      "rewards/margins": 2.203125,
+      "rewards/rejected": -7.09375,
       "step": 370
     },
     {
       "epoch": 0.39790575916230364,
-      "grad_norm": 2.1576846738942814,
+      "grad_norm": 1.9796366502802532,
       "learning_rate": 4.983962194568872e-05,
-      "logits/chosen": -9.25,
-      "logits/rejected": -8.125,
-      "logps/chosen": -824.0,
-      "logps/rejected": -1408.0,
-      "loss": 0.45,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -4.125,
-      "rewards/margins": 2.640625,
-      "rewards/rejected": -6.78125,
+      "logits/chosen": -10.125,
+      "logits/rejected": -9.375,
+      "logps/chosen": -832.0,
+      "logps/rejected": -1424.0,
+      "loss": 0.4361,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -4.1875,
+      "rewards/margins": 2.703125,
+      "rewards/rejected": -6.90625,
       "step": 380
     },
     {
       "epoch": 0.4083769633507853,
-      "grad_norm": 2.7628662083416833,
+      "grad_norm": 2.824891336811741,
       "learning_rate": 4.9803325467003084e-05,
-      "logits/chosen": -11.125,
-      "logits/rejected": -9.875,
-      "logps/chosen": -756.0,
-      "logps/rejected": -1224.0,
-      "loss": 0.4251,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -3.40625,
-      "rewards/margins": 2.125,
-      "rewards/rejected": -5.5625,
+      "logits/chosen": -11.9375,
+      "logits/rejected": -10.9375,
+      "logps/chosen": -736.0,
+      "logps/rejected": -1184.0,
+      "loss": 0.4234,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -3.1875,
+      "rewards/margins": 1.9765625,
+      "rewards/rejected": -5.15625,
       "step": 390
     },
     {
       "epoch": 0.418848167539267,
-      "grad_norm": 2.108113678780066,
+      "grad_norm": 2.657647692531284,
       "learning_rate": 4.9763345676749816e-05,
-      "logits/chosen": -10.875,
-      "logits/rejected": -10.25,
-      "logps/chosen": -772.0,
-      "logps/rejected": -1072.0,
-      "loss": 0.4624,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -3.8125,
-      "rewards/margins": 1.5546875,
-      "rewards/rejected": -5.375,
+      "logits/chosen": -9.625,
+      "logits/rejected": -9.25,
+      "logps/chosen": -796.0,
+      "logps/rejected": -1112.0,
+      "loss": 0.4474,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -4.0625,
+      "rewards/margins": 1.7109375,
+      "rewards/rejected": -5.78125,
       "step": 400
     },
     {
       "epoch": 0.418848167539267,
-      "eval_logits/chosen": -12.3125,
-      "eval_logits/rejected": -11.3125,
-      "eval_logps/chosen": -676.0,
-      "eval_logps/rejected": -692.0,
-      "eval_loss": 0.7233437299728394,
-      "eval_rewards/accuracies": 0.60546875,
-      "eval_rewards/chosen": -3.5625,
-      "eval_rewards/margins": 0.494140625,
-      "eval_rewards/rejected": -4.0625,
-      "eval_runtime": 47.2245,
-      "eval_samples_per_second": 42.351,
-      "eval_steps_per_second": 0.678,
+      "eval_logits/chosen": -10.0625,
+      "eval_logits/rejected": -9.125,
+      "eval_logps/chosen": -636.0,
+      "eval_logps/rejected": -664.0,
+      "eval_loss": 0.7340624928474426,
+      "eval_rewards/accuracies": 0.626953125,
+      "eval_rewards/chosen": -3.1875,
+      "eval_rewards/margins": 0.5625,
+      "eval_rewards/rejected": -3.75,
+      "eval_runtime": 46.5563,
+      "eval_samples_per_second": 42.959,
+      "eval_steps_per_second": 0.687,
       "step": 400
     },
     {
       "epoch": 0.4293193717277487,
-      "grad_norm": 2.565615926912782,
+      "grad_norm": 3.2080648942679466,
       "learning_rate": 4.9719688511956374e-05,
-      "logits/chosen": -11.75,
-      "logits/rejected": -10.6875,
-      "logps/chosen": -864.0,
-      "logps/rejected": -1280.0,
-      "loss": 0.4033,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -4.40625,
-      "rewards/margins": 2.234375,
-      "rewards/rejected": -6.65625,
+      "logits/chosen": -10.8125,
+      "logits/rejected": -10.0625,
+      "logps/chosen": -740.0,
+      "logps/rejected": -1136.0,
+      "loss": 0.41,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -3.1875,
+      "rewards/margins": 1.9921875,
+      "rewards/rejected": -5.1875,
       "step": 410
     },
     {
       "epoch": 0.4397905759162304,
-      "grad_norm": 2.5736428925524395,
+      "grad_norm": 3.1564908806998524,
       "learning_rate": 4.967236045574294e-05,
-      "logits/chosen": -9.9375,
-      "logits/rejected": -8.5625,
-      "logps/chosen": -912.0,
-      "logps/rejected": -1384.0,
-      "loss": 0.4054,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -5.09375,
-      "rewards/margins": 2.640625,
-      "rewards/rejected": -7.75,
+      "logits/chosen": -8.625,
+      "logits/rejected": -7.59375,
+      "logps/chosen": -844.0,
+      "logps/rejected": -1312.0,
+      "loss": 0.4502,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -4.375,
+      "rewards/margins": 2.59375,
+      "rewards/rejected": -7.0,
       "step": 420
     },
     {
       "epoch": 0.450261780104712,
-      "grad_norm": 2.859187192915181,
+      "grad_norm": 2.9110187005374866,
       "learning_rate": 4.96213685363597e-05,
-      "logits/chosen": -9.75,
-      "logits/rejected": -9.0,
-      "logps/chosen": -780.0,
-      "logps/rejected": -1168.0,
-      "loss": 0.4254,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -3.84375,
-      "rewards/margins": 1.8828125,
-      "rewards/rejected": -5.75,
+      "logits/chosen": -8.6875,
+      "logits/rejected": -8.1875,
+      "logps/chosen": -700.0,
+      "logps/rejected": -1064.0,
+      "loss": 0.4863,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -3.03125,
+      "rewards/margins": 1.6640625,
+      "rewards/rejected": -4.6875,
       "step": 430
     },
     {
       "epoch": 0.4607329842931937,
-      "grad_norm": 3.254685782656208,
+      "grad_norm": 5.14221421404816,
       "learning_rate": 4.956672032614315e-05,
-      "logits/chosen": -10.3125,
-      "logits/rejected": -9.625,
-      "logps/chosen": -812.0,
-      "logps/rejected": -1232.0,
-      "loss": 0.4422,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -3.984375,
-      "rewards/margins": 1.8984375,
-      "rewards/rejected": -5.875,
+      "logits/chosen": -9.8125,
+      "logits/rejected": -9.0,
+      "logps/chosen": -700.0,
+      "logps/rejected": -1096.0,
+      "loss": 0.4586,
+      "rewards/accuracies": 0.84375,
+      "rewards/chosen": -2.828125,
+      "rewards/margins": 1.734375,
+      "rewards/rejected": -4.5625,
       "step": 440
     },
     {
       "epoch": 0.4712041884816754,
-      "grad_norm": 2.575491895393729,
+      "grad_norm": 2.4107914996706583,
       "learning_rate": 4.950842394039157e-05,
-      "logits/chosen": -11.1875,
-      "logits/rejected": -10.3125,
-      "logps/chosen": -788.0,
-      "logps/rejected": -1256.0,
-      "loss": 0.3665,
-      "rewards/accuracies": 0.84375,
-      "rewards/chosen": -4.0,
-      "rewards/margins": 2.703125,
-      "rewards/rejected": -6.71875,
+      "logits/chosen": -10.875,
+      "logits/rejected": -9.625,
+      "logps/chosen": -716.0,
+      "logps/rejected": -1184.0,
+      "loss": 0.3751,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -3.296875,
+      "rewards/margins": 2.6875,
+      "rewards/rejected": -6.0,
       "step": 450
     },
     {
       "epoch": 0.4816753926701571,
-      "grad_norm": 3.297298961632602,
+      "grad_norm": 3.1640522302802743,
       "learning_rate": 4.944648803615994e-05,
-      "logits/chosen": -8.8125,
-      "logits/rejected": -7.5625,
-      "logps/chosen": -900.0,
-      "logps/rejected": -1504.0,
-      "loss": 0.3571,
-      "rewards/accuracies": 0.875,
-      "rewards/chosen": -4.90625,
-      "rewards/margins": 3.390625,
-      "rewards/rejected": -8.3125,
+      "logits/chosen": -8.875,
+      "logits/rejected": -7.75,
+      "logps/chosen": -804.0,
+      "logps/rejected": -1368.0,
+      "loss": 0.3746,
+      "rewards/accuracies": 0.862500011920929,
+      "rewards/chosen": -3.9375,
+      "rewards/margins": 3.0,
+      "rewards/rejected": -6.9375,
       "step": 460
     },
     {
       "epoch": 0.49214659685863876,
-      "grad_norm": 3.2052642555898516,
+      "grad_norm": 2.5483721623578712,
       "learning_rate": 4.938092181097433e-05,
-      "logits/chosen": -7.5625,
-      "logits/rejected": -6.84375,
-      "logps/chosen": -1008.0,
-      "logps/rejected": -1576.0,
-      "loss": 0.3558,
-      "rewards/accuracies": 0.862500011920929,
-      "rewards/chosen": -6.0625,
-      "rewards/margins": 3.78125,
-      "rewards/rejected": -9.875,
+      "logits/chosen": -8.1875,
+      "logits/rejected": -7.46875,
+      "logps/chosen": -868.0,
+      "logps/rejected": -1416.0,
+      "loss": 0.3944,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -4.65625,
+      "rewards/margins": 3.609375,
+      "rewards/rejected": -8.25,
       "step": 470
     },
     {
       "epoch": 0.5026178010471204,
-      "grad_norm": 5.923067620926881,
+      "grad_norm": 5.274023840028499,
       "learning_rate": 4.931173500146607e-05,
-      "logits/chosen": -8.6875,
-      "logits/rejected": -7.78125,
-      "logps/chosen": -940.0,
-      "logps/rejected": -1576.0,
-      "loss": 0.3837,
+      "logits/chosen": -7.875,
+      "logits/rejected": -7.0625,
+      "logps/chosen": -812.0,
+      "logps/rejected": -1416.0,
+      "loss": 0.3631,
       "rewards/accuracies": 0.8374999761581421,
-      "rewards/chosen": -5.625,
-      "rewards/margins": 3.875,
-      "rewards/rejected": -9.5,
+      "rewards/chosen": -4.34375,
+      "rewards/margins": 3.546875,
+      "rewards/rejected": -7.90625,
       "step": 480
     },
     {
       "epoch": 0.5130890052356021,
-      "grad_norm": 2.1013028895144013,
+      "grad_norm": 2.0850378369622335,
       "learning_rate": 4.923893788192582e-05,
-      "logits/chosen": -8.5625,
-      "logits/rejected": -7.625,
-      "logps/chosen": -920.0,
-      "logps/rejected": -1456.0,
-      "loss": 0.4372,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -5.1875,
-      "rewards/margins": 3.5,
-      "rewards/rejected": -8.6875,
+      "logits/chosen": -8.3125,
+      "logits/rejected": -7.5625,
+      "logps/chosen": -768.0,
+      "logps/rejected": -1232.0,
+      "loss": 0.4419,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -3.65625,
+      "rewards/margins": 2.765625,
+      "rewards/rejected": -6.4375,
       "step": 490
     },
     {
       "epoch": 0.5235602094240838,
-      "grad_norm": 2.698247192834292,
+      "grad_norm": 2.9930959885694786,
       "learning_rate": 4.9162541262777906e-05,
-      "logits/chosen": -10.125,
-      "logits/rejected": -8.9375,
-      "logps/chosen": -820.0,
-      "logps/rejected": -1216.0,
-      "loss": 0.373,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -4.0,
-      "rewards/margins": 2.34375,
-      "rewards/rejected": -6.34375,
+      "logits/chosen": -8.5625,
+      "logits/rejected": -7.78125,
+      "logps/chosen": -808.0,
+      "logps/rejected": -1200.0,
+      "loss": 0.398,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -3.890625,
+      "rewards/margins": 2.296875,
+      "rewards/rejected": -6.1875,
       "step": 500
     },
     {
       "epoch": 0.5235602094240838,
-      "eval_logits/chosen": -8.6875,
-      "eval_logits/rejected": -7.3125,
-      "eval_logps/chosen": -768.0,
-      "eval_logps/rejected": -808.0,
-      "eval_loss": 0.8017343878746033,
-      "eval_rewards/accuracies": 0.626953125,
-      "eval_rewards/chosen": -4.5,
-      "eval_rewards/margins": 0.67578125,
-      "eval_rewards/rejected": -5.1875,
-      "eval_runtime": 47.0783,
-      "eval_samples_per_second": 42.482,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -8.125,
+      "eval_logits/rejected": -6.6875,
+      "eval_logps/chosen": -692.0,
+      "eval_logps/rejected": -736.0,
+      "eval_loss": 0.8244531154632568,
+      "eval_rewards/accuracies": 0.63671875,
+      "eval_rewards/chosen": -3.734375,
+      "eval_rewards/margins": 0.7265625,
+      "eval_rewards/rejected": -4.46875,
+      "eval_runtime": 46.5485,
+      "eval_samples_per_second": 42.966,
+      "eval_steps_per_second": 0.687,
       "step": 500
     },
     {
       "epoch": 0.5340314136125655,
-      "grad_norm": 3.41827119232801,
+      "grad_norm": 3.0193870304515014,
       "learning_rate": 4.90825564889749e-05,
-      "logits/chosen": -9.125,
-      "logits/rejected": -8.4375,
-      "logps/chosen": -980.0,
-      "logps/rejected": -1288.0,
-      "loss": 0.4038,
-      "rewards/accuracies": 0.75,
-      "rewards/chosen": -5.40625,
-      "rewards/margins": 2.0625,
-      "rewards/rejected": -7.4375,
+      "logits/chosen": -9.25,
+      "logits/rejected": -8.5,
+      "logps/chosen": -912.0,
+      "logps/rejected": -1256.0,
+      "loss": 0.4008,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -4.75,
+      "rewards/margins": 2.4375,
+      "rewards/rejected": -7.1875,
       "step": 510
     },
     {
       "epoch": 0.5445026178010471,
-      "grad_norm": 2.440591801688025,
+      "grad_norm": 2.2565439791395074,
       "learning_rate": 4.899899543831288e-05,
-      "logits/chosen": -10.9375,
-      "logits/rejected": -9.75,
-      "logps/chosen": -952.0,
+      "logits/chosen": -9.375,
+      "logits/rejected": -8.25,
+      "logps/chosen": -948.0,
       "logps/rejected": -1336.0,
-      "loss": 0.3826,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -5.15625,
-      "rewards/margins": 2.78125,
-      "rewards/rejected": -7.96875,
+      "loss": 0.4143,
+      "rewards/accuracies": 0.762499988079071,
+      "rewards/chosen": -5.125,
+      "rewards/margins": 2.796875,
+      "rewards/rejected": -7.9375,
       "step": 520
     },
     {
       "epoch": 0.5549738219895288,
-      "grad_norm": 3.4988940244923685,
+      "grad_norm": 2.6859804756387176,
       "learning_rate": 4.891187051966765e-05,
-      "logits/chosen": -11.75,
-      "logits/rejected": -11.0625,
-      "logps/chosen": -960.0,
-      "logps/rejected": -1312.0,
-      "loss": 0.417,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -5.34375,
-      "rewards/margins": 2.25,
-      "rewards/rejected": -7.59375,
+      "logits/chosen": -10.6875,
+      "logits/rejected": -10.0625,
+      "logps/chosen": -848.0,
+      "logps/rejected": -1176.0,
+      "loss": 0.4261,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -4.21875,
+      "rewards/margins": 2.015625,
+      "rewards/rejected": -6.21875,
       "step": 530
     },
     {
       "epoch": 0.5654450261780105,
-      "grad_norm": 2.7529177257956894,
+      "grad_norm": 2.87608849473287,
       "learning_rate": 4.882119467115191e-05,
-      "logits/chosen": -12.1875,
-      "logits/rejected": -11.5,
-      "logps/chosen": -940.0,
-      "logps/rejected": -1344.0,
-      "loss": 0.3729,
-      "rewards/accuracies": 0.7875000238418579,
-      "rewards/chosen": -5.375,
-      "rewards/margins": 2.3125,
-      "rewards/rejected": -7.6875,
+      "logits/chosen": -10.875,
+      "logits/rejected": -10.125,
+      "logps/chosen": -908.0,
+      "logps/rejected": -1360.0,
+      "loss": 0.3868,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -5.0625,
+      "rewards/margins": 2.859375,
+      "rewards/rejected": -7.90625,
       "step": 540
     },
     {
       "epoch": 0.5759162303664922,
-      "grad_norm": 2.9444958471125102,
+      "grad_norm": 2.1911243334534465,
       "learning_rate": 4.872698135819402e-05,
-      "logits/chosen": -13.0625,
-      "logits/rejected": -12.125,
-      "logps/chosen": -848.0,
-      "logps/rejected": -1304.0,
-      "loss": 0.384,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -4.53125,
-      "rewards/margins": 2.171875,
-      "rewards/rejected": -6.71875,
+      "logits/chosen": -11.6875,
+      "logits/rejected": -10.6875,
+      "logps/chosen": -852.0,
+      "logps/rejected": -1344.0,
+      "loss": 0.3962,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -4.5625,
+      "rewards/margins": 2.53125,
+      "rewards/rejected": -7.09375,
       "step": 550
     },
     {
       "epoch": 0.5863874345549738,
-      "grad_norm": 2.164423145904123,
+      "grad_norm": 2.9440541491746846,
       "learning_rate": 4.862924457153831e-05,
-      "logits/chosen": -12.6875,
-      "logits/rejected": -11.5,
-      "logps/chosen": -772.0,
-      "logps/rejected": -1264.0,
-      "loss": 0.3553,
-      "rewards/accuracies": 0.8374999761581421,
-      "rewards/chosen": -4.34375,
-      "rewards/margins": 2.640625,
-      "rewards/rejected": -6.96875,
+      "logits/chosen": -11.375,
+      "logits/rejected": -10.3125,
+      "logps/chosen": -744.0,
+      "logps/rejected": -1224.0,
+      "loss": 0.3928,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -4.03125,
+      "rewards/margins": 2.484375,
+      "rewards/rejected": -6.53125,
       "step": 560
     },
     {
       "epoch": 0.5968586387434555,
-      "grad_norm": 2.5427492500745785,
+      "grad_norm": 2.2836689098851335,
       "learning_rate": 4.8527998825167484e-05,
-      "logits/chosen": -9.8125,
-      "logits/rejected": -8.6875,
-      "logps/chosen": -960.0,
-      "logps/rejected": -1448.0,
-      "loss": 0.3727,
-      "rewards/accuracies": 0.8125,
-      "rewards/chosen": -5.8125,
-      "rewards/margins": 2.78125,
-      "rewards/rejected": -8.625,
+      "logits/chosen": -10.9375,
+      "logits/rejected": -10.0625,
+      "logps/chosen": -840.0,
+      "logps/rejected": -1328.0,
+      "loss": 0.3612,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -4.65625,
+      "rewards/margins": 2.703125,
+      "rewards/rejected": -7.34375,
       "step": 570
     },
     {
       "epoch": 0.6073298429319371,
-      "grad_norm": 2.8173384255067138,
+      "grad_norm": 2.75251385602701,
       "learning_rate": 4.842325915414728e-05,
-      "logits/chosen": -9.0,
-      "logits/rejected": -7.90625,
-      "logps/chosen": -788.0,
-      "logps/rejected": -1128.0,
-      "loss": 0.3916,
+      "logits/chosen": -10.75,
+      "logits/rejected": -9.8125,
+      "logps/chosen": -848.0,
+      "logps/rejected": -1264.0,
+      "loss": 0.3846,
       "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -4.28125,
-      "rewards/margins": 1.9609375,
-      "rewards/rejected": -6.25,
+      "rewards/chosen": -4.875,
+      "rewards/margins": 2.703125,
+      "rewards/rejected": -7.59375,
       "step": 580
     },
     {
       "epoch": 0.6178010471204188,
-      "grad_norm": 4.092750582522732,
+      "grad_norm": 3.4086627548069908,
       "learning_rate": 4.831504111239373e-05,
-      "logits/chosen": -9.625,
-      "logits/rejected": -8.5625,
-      "logps/chosen": -976.0,
-      "logps/rejected": -1560.0,
-      "loss": 0.3583,
-      "rewards/accuracies": 0.8812500238418579,
-      "rewards/chosen": -5.59375,
-      "rewards/margins": 3.140625,
-      "rewards/rejected": -8.75,
+      "logits/chosen": -9.9375,
+      "logits/rejected": -9.0625,
+      "logps/chosen": -940.0,
+      "logps/rejected": -1528.0,
+      "loss": 0.3748,
+      "rewards/accuracies": 0.8500000238418579,
+      "rewards/chosen": -5.25,
+      "rewards/margins": 3.1875,
+      "rewards/rejected": -8.4375,
       "step": 590
     },
     {
       "epoch": 0.6282722513089005,
-      "grad_norm": 2.2019812207382214,
+      "grad_norm": 2.951743349805168,
       "learning_rate": 4.8203360770363425e-05,
-      "logits/chosen": -10.25,
-      "logits/rejected": -9.3125,
-      "logps/chosen": -984.0,
-      "logps/rejected": -1456.0,
-      "loss": 0.418,
-      "rewards/accuracies": 0.8125,
-      "rewards/chosen": -5.78125,
-      "rewards/margins": 2.953125,
-      "rewards/rejected": -8.75,
+      "logits/chosen": -10.0625,
+      "logits/rejected": -9.25,
+      "logps/chosen": -948.0,
+      "logps/rejected": -1424.0,
+      "loss": 0.4063,
+      "rewards/accuracies": 0.800000011920929,
+      "rewards/chosen": -5.40625,
+      "rewards/margins": 3.03125,
+      "rewards/rejected": -8.4375,
       "step": 600
     },
     {
       "epoch": 0.6282722513089005,
-      "eval_logits/chosen": -11.0,
-      "eval_logits/rejected": -9.8125,
-      "eval_logps/chosen": -768.0,
-      "eval_logps/rejected": -816.0,
-      "eval_loss": 0.8380625247955322,
-      "eval_rewards/accuracies": 0.61328125,
-      "eval_rewards/chosen": -4.46875,
-      "eval_rewards/margins": 0.78125,
-      "eval_rewards/rejected": -5.28125,
-      "eval_runtime": 47.079,
-      "eval_samples_per_second": 42.482,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -9.1875,
+      "eval_logits/rejected": -8.0625,
+      "eval_logps/chosen": -856.0,
+      "eval_logps/rejected": -912.0,
+      "eval_loss": 0.8893749713897705,
+      "eval_rewards/accuracies": 0.6171875,
+      "eval_rewards/chosen": -5.375,
+      "eval_rewards/margins": 0.84765625,
+      "eval_rewards/rejected": -6.21875,
+      "eval_runtime": 46.5463,
+      "eval_samples_per_second": 42.968,
+      "eval_steps_per_second": 0.687,
       "step": 600
     },
     {
       "epoch": 0.6387434554973822,
-      "grad_norm": 3.716517958121969,
+      "grad_norm": 3.5184732557371534,
       "learning_rate": 4.808823471266702e-05,
-      "logits/chosen": -11.5625,
-      "logits/rejected": -10.25,
-      "logps/chosen": -816.0,
-      "logps/rejected": -1296.0,
-      "loss": 0.3654,
-      "rewards/accuracies": 0.875,
-      "rewards/chosen": -4.21875,
-      "rewards/margins": 2.703125,
-      "rewards/rejected": -6.90625,
+      "logits/chosen": -10.5,
+      "logits/rejected": -9.625,
+      "logps/chosen": -836.0,
+      "logps/rejected": -1312.0,
+      "loss": 0.3881,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -4.4375,
+      "rewards/margins": 2.671875,
+      "rewards/rejected": -7.125,
       "step": 610
     },
     {
       "epoch": 0.6492146596858639,
-      "grad_norm": 3.753413187168972,
+      "grad_norm": 3.647126988918458,
       "learning_rate": 4.796968003560639e-05,
-      "logits/chosen": -12.5625,
-      "logits/rejected": -11.0,
-      "logps/chosen": -892.0,
-      "logps/rejected": -1352.0,
-      "loss": 0.452,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.875,
-      "rewards/margins": 2.53125,
-      "rewards/rejected": -7.375,
+      "logits/chosen": -11.9375,
+      "logits/rejected": -10.875,
+      "logps/chosen": -804.0,
+      "logps/rejected": -1248.0,
+      "loss": 0.4185,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -4.0,
+      "rewards/margins": 2.28125,
+      "rewards/rejected": -6.28125,
       "step": 620
     },
     {
       "epoch": 0.6596858638743456,
-      "grad_norm": 2.062359612493287,
+      "grad_norm": 2.360636929013847,
       "learning_rate": 4.7847714344635864e-05,
-      "logits/chosen": -14.6875,
-      "logits/rejected": -13.75,
-      "logps/chosen": -800.0,
-      "logps/rejected": -1160.0,
-      "loss": 0.4173,
-      "rewards/accuracies": 0.78125,
-      "rewards/chosen": -4.15625,
-      "rewards/margins": 2.1875,
-      "rewards/rejected": -6.34375,
+      "logits/chosen": -12.75,
+      "logits/rejected": -11.5625,
+      "logps/chosen": -728.0,
+      "logps/rejected": -1080.0,
+      "loss": 0.3991,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -3.453125,
+      "rewards/margins": 2.109375,
+      "rewards/rejected": -5.5625,
       "step": 630
     },
     {
       "epoch": 0.6701570680628273,
-      "grad_norm": 3.124870807293616,
+      "grad_norm": 4.859311628671399,
       "learning_rate": 4.772235575174776e-05,
-      "logits/chosen": -13.625,
-      "logits/rejected": -13.125,
-      "logps/chosen": -972.0,
-      "logps/rejected": -1336.0,
-      "loss": 0.3992,
-      "rewards/accuracies": 0.731249988079071,
+      "logits/chosen": -9.75,
+      "logits/rejected": -8.5625,
+      "logps/chosen": -976.0,
+      "logps/rejected": -1392.0,
+      "loss": 0.3911,
+      "rewards/accuracies": 0.800000011920929,
       "rewards/chosen": -5.40625,
-      "rewards/margins": 2.140625,
-      "rewards/rejected": -7.53125,
+      "rewards/margins": 2.6875,
+      "rewards/rejected": -8.0625,
       "step": 640
     },
     {
       "epoch": 0.680628272251309,
-      "grad_norm": 2.499316818622552,
+      "grad_norm": 3.8806965851278417,
       "learning_rate": 4.7593622872782754e-05,
-      "logits/chosen": -12.75,
-      "logits/rejected": -11.875,
-      "logps/chosen": -924.0,
-      "logps/rejected": -1320.0,
-      "loss": 0.4004,
-      "rewards/accuracies": 0.7749999761581421,
-      "rewards/chosen": -5.375,
-      "rewards/margins": 2.4375,
-      "rewards/rejected": -7.8125,
+      "logits/chosen": -8.875,
+      "logits/rejected": -8.3125,
+      "logps/chosen": -804.0,
+      "logps/rejected": -1144.0,
+      "loss": 0.423,
+      "rewards/accuracies": 0.7562500238418579,
+      "rewards/chosen": -4.15625,
+      "rewards/margins": 1.8671875,
+      "rewards/rejected": -6.03125,
       "step": 650
     },
     {
       "epoch": 0.6910994764397905,
-      "grad_norm": 3.1708092646053587,
+      "grad_norm": 2.6767693622168194,
       "learning_rate": 4.746153482466542e-05,
-      "logits/chosen": -11.3125,
-      "logits/rejected": -10.0625,
-      "logps/chosen": -796.0,
-      "logps/rejected": -1160.0,
-      "loss": 0.4005,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -4.125,
-      "rewards/margins": 2.15625,
-      "rewards/rejected": -6.28125,
+      "logits/chosen": -10.75,
+      "logits/rejected": -10.0,
+      "logps/chosen": -684.0,
+      "logps/rejected": -1012.0,
+      "loss": 0.415,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -3.015625,
+      "rewards/margins": 1.796875,
+      "rewards/rejected": -4.8125,
       "step": 660
     },
     {
       "epoch": 0.7015706806282722,
-      "grad_norm": 2.739188543095194,
+      "grad_norm": 5.086831999906795,
       "learning_rate": 4.732611122256537e-05,
-      "logits/chosen": -8.0625,
-      "logits/rejected": -7.1875,
-      "logps/chosen": -1088.0,
-      "logps/rejected": -1456.0,
-      "loss": 0.3637,
-      "rewards/accuracies": 0.8062499761581421,
-      "rewards/chosen": -6.5,
-      "rewards/margins": 2.546875,
-      "rewards/rejected": -9.0,
+      "logits/chosen": -7.9375,
+      "logits/rejected": -7.0,
+      "logps/chosen": -984.0,
+      "logps/rejected": -1384.0,
+      "loss": 0.3797,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -5.46875,
+      "rewards/margins": 2.84375,
+      "rewards/rejected": -8.3125,
       "step": 670
     },
     {
       "epoch": 0.7120418848167539,
-      "grad_norm": 2.0506422048897694,
+      "grad_norm": 2.144842657664273,
       "learning_rate": 4.718737217698435e-05,
-      "logits/chosen": -8.9375,
-      "logits/rejected": -8.125,
-      "logps/chosen": -868.0,
-      "logps/rejected": -1296.0,
-      "loss": 0.4323,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.875,
-      "rewards/margins": 2.125,
-      "rewards/rejected": -7.0,
+      "logits/chosen": -6.125,
+      "logits/rejected": -6.0625,
+      "logps/chosen": -820.0,
+      "logps/rejected": -1264.0,
+      "loss": 0.4382,
+      "rewards/accuracies": 0.7749999761581421,
+      "rewards/chosen": -4.40625,
+      "rewards/margins": 2.265625,
+      "rewards/rejected": -6.65625,
       "step": 680
     },
     {
       "epoch": 0.7225130890052356,
-      "grad_norm": 2.2563511617718204,
+      "grad_norm": 2.3506372684725543,
       "learning_rate": 4.704533829076982e-05,
-      "logits/chosen": -11.875,
-      "logits/rejected": -10.875,
-      "logps/chosen": -788.0,
-      "logps/rejected": -1216.0,
-      "loss": 0.3888,
-      "rewards/accuracies": 0.84375,
-      "rewards/chosen": -3.90625,
-      "rewards/margins": 2.25,
-      "rewards/rejected": -6.15625,
+      "logits/chosen": -7.6875,
+      "logits/rejected": -7.25,
+      "logps/chosen": -780.0,
+      "logps/rejected": -1184.0,
+      "loss": 0.3866,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -3.828125,
+      "rewards/margins": 1.9609375,
+      "rewards/rejected": -5.78125,
       "step": 690
     },
     {
       "epoch": 0.7329842931937173,
-      "grad_norm": 2.796314845940636,
+      "grad_norm": 2.526158749317099,
       "learning_rate": 4.6900030656055455e-05,
-      "logits/chosen": -12.0625,
-      "logits/rejected": -11.0,
-      "logps/chosen": -800.0,
-      "logps/rejected": -1160.0,
-      "loss": 0.4,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -3.90625,
-      "rewards/margins": 2.203125,
-      "rewards/rejected": -6.125,
-      "step": 700
-    },
+      "logits/chosen": -7.96875,
+      "logits/rejected": -6.96875,
+      "logps/chosen": -788.0,
+      "logps/rejected": -1144.0,
+      "loss": 0.3877,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -3.796875,
+      "rewards/margins": 2.15625,
+      "rewards/rejected": -5.9375,
+      "step": 700
+    },
     {
       "epoch": 0.7329842931937173,
-      "eval_logits/chosen": -11.6875,
-      "eval_logits/rejected": -10.75,
-      "eval_logps/chosen": -752.0,
-      "eval_logps/rejected": -800.0,
-      "eval_loss": 0.7899531126022339,
-      "eval_rewards/accuracies": 0.646484375,
-      "eval_rewards/chosen": -4.3125,
-      "eval_rewards/margins": 0.80078125,
-      "eval_rewards/rejected": -5.125,
-      "eval_runtime": 47.0119,
-      "eval_samples_per_second": 42.542,
-      "eval_steps_per_second": 0.681,
+      "eval_logits/chosen": -7.125,
+      "eval_logits/rejected": -5.59375,
+      "eval_logps/chosen": -768.0,
+      "eval_logps/rejected": -836.0,
+      "eval_loss": 0.8034218549728394,
+      "eval_rewards/accuracies": 0.640625,
+      "eval_rewards/chosen": -4.5,
+      "eval_rewards/margins": 0.9453125,
+      "eval_rewards/rejected": -5.46875,
+      "eval_runtime": 46.5561,
+      "eval_samples_per_second": 42.959,
+      "eval_steps_per_second": 0.687,
       "step": 700
     },
     {
       "epoch": 0.743455497382199,
-      "grad_norm": 2.5923435233959697,
+      "grad_norm": 2.5995482863598305,
       "learning_rate": 4.6751470851128906e-05,
-      "logits/chosen": -11.875,
-      "logits/rejected": -10.75,
-      "logps/chosen": -832.0,
-      "logps/rejected": -1272.0,
-      "loss": 0.3732,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -3.96875,
-      "rewards/margins": 1.9375,
-      "rewards/rejected": -5.90625,
+      "logits/chosen": -8.4375,
+      "logits/rejected": -7.28125,
+      "logps/chosen": -872.0,
+      "logps/rejected": -1336.0,
+      "loss": 0.3965,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -4.375,
+      "rewards/margins": 2.171875,
+      "rewards/rejected": -6.5625,
       "step": 710
     },
     {
       "epoch": 0.7539267015706806,
-      "grad_norm": 4.808763076398256,
+      "grad_norm": 2.524535994704713,
       "learning_rate": 4.6599680937227454e-05,
-      "logits/chosen": -11.375,
-      "logits/rejected": -9.875,
-      "logps/chosen": -932.0,
-      "logps/rejected": -1432.0,
-      "loss": 0.3776,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -5.09375,
-      "rewards/margins": 2.390625,
-      "rewards/rejected": -7.5,
+      "logits/chosen": -8.875,
+      "logits/rejected": -7.78125,
+      "logps/chosen": -972.0,
+      "logps/rejected": -1464.0,
+      "loss": 0.3944,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -5.5,
+      "rewards/margins": 2.265625,
+      "rewards/rejected": -7.78125,
       "step": 720
     },
     {
       "epoch": 0.7643979057591623,
-      "grad_norm": 3.3056631440336632,
+      "grad_norm": 2.562945854031945,
       "learning_rate": 4.644468345526183e-05,
-      "logits/chosen": -11.6875,
-      "logits/rejected": -10.625,
-      "logps/chosen": -884.0,
-      "logps/rejected": -1392.0,
-      "loss": 0.3878,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -4.71875,
-      "rewards/margins": 2.640625,
-      "rewards/rejected": -7.34375,
+      "logits/chosen": -12.125,
+      "logits/rejected": -11.25,
+      "logps/chosen": -828.0,
+      "logps/rejected": -1304.0,
+      "loss": 0.3816,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -4.15625,
+      "rewards/margins": 2.328125,
+      "rewards/rejected": -6.46875,
       "step": 730
     },
     {
       "epoch": 0.774869109947644,
-      "grad_norm": 2.1338574011001996,
+      "grad_norm": 2.8676753093725815,
       "learning_rate": 4.628650142246894e-05,
-      "logits/chosen": -7.3125,
-      "logits/rejected": -6.4375,
-      "logps/chosen": -1032.0,
-      "logps/rejected": -1600.0,
-      "loss": 0.3479,
-      "rewards/accuracies": 0.8687499761581421,
-      "rewards/chosen": -6.53125,
-      "rewards/margins": 2.734375,
-      "rewards/rejected": -9.25,
+      "logits/chosen": -11.625,
+      "logits/rejected": -10.0,
+      "logps/chosen": -840.0,
+      "logps/rejected": -1424.0,
+      "loss": 0.3356,
+      "rewards/accuracies": 0.8500000238418579,
+      "rewards/chosen": -4.625,
+      "rewards/margins": 2.9375,
+      "rewards/rejected": -7.5625,
       "step": 740
     },
     {
       "epoch": 0.7853403141361257,
-      "grad_norm": 2.3416942555594327,
+      "grad_norm": 2.574155280415414,
       "learning_rate": 4.6125158328993765e-05,
-      "logits/chosen": -7.96875,
-      "logits/rejected": -7.0,
-      "logps/chosen": -948.0,
-      "logps/rejected": -1400.0,
-      "loss": 0.378,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -5.625,
-      "rewards/margins": 2.375,
-      "rewards/rejected": -8.0,
+      "logits/chosen": -11.8125,
+      "logits/rejected": -10.375,
+      "logps/chosen": -788.0,
+      "logps/rejected": -1232.0,
+      "loss": 0.3716,
+      "rewards/accuracies": 0.8187500238418579,
+      "rewards/chosen": -4.03125,
+      "rewards/margins": 2.328125,
+      "rewards/rejected": -6.34375,
       "step": 750
     },
     {
       "epoch": 0.7958115183246073,
-      "grad_norm": 2.786491188516209,
+      "grad_norm": 3.0126485134282426,
       "learning_rate": 4.5960678134401064e-05,
-      "logits/chosen": -11.125,
-      "logits/rejected": -10.1875,
-      "logps/chosen": -844.0,
-      "logps/rejected": -1264.0,
-      "loss": 0.3708,
+      "logits/chosen": -11.625,
+      "logits/rejected": -10.6875,
+      "logps/chosen": -816.0,
+      "logps/rejected": -1232.0,
+      "loss": 0.3903,
       "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -4.40625,
-      "rewards/margins": 2.296875,
-      "rewards/rejected": -6.71875,
+      "rewards/chosen": -4.15625,
+      "rewards/margins": 2.171875,
+      "rewards/rejected": -6.3125,
       "step": 760
     },
     {
       "epoch": 0.806282722513089,
-      "grad_norm": 3.7399992316619675,
+      "grad_norm": 3.250752003058936,
       "learning_rate": 4.579308526411734e-05,
-      "logits/chosen": -11.0,
-      "logits/rejected": -9.8125,
-      "logps/chosen": -932.0,
-      "logps/rejected": -1624.0,
-      "loss": 0.3749,
-      "rewards/accuracies": 0.8500000238418579,
-      "rewards/chosen": -5.40625,
-      "rewards/margins": 3.625,
-      "rewards/rejected": -9.0625,
+      "logits/chosen": -8.75,
+      "logits/rejected": -7.96875,
+      "logps/chosen": -912.0,
+      "logps/rejected": -1552.0,
+      "loss": 0.3778,
+      "rewards/accuracies": 0.84375,
+      "rewards/chosen": -5.25,
+      "rewards/margins": 3.015625,
+      "rewards/rejected": -8.25,
       "step": 770
     },
     {
       "epoch": 0.8167539267015707,
-      "grad_norm": 3.419904484617841,
+      "grad_norm": 4.492756640519936,
       "learning_rate": 4.562240460580367e-05,
-      "logits/chosen": -12.125,
-      "logits/rejected": -11.1875,
-      "logps/chosen": -896.0,
-      "logps/rejected": -1408.0,
-      "loss": 0.3598,
-      "rewards/accuracies": 0.862500011920929,
-      "rewards/chosen": -5.03125,
-      "rewards/margins": 2.859375,
-      "rewards/rejected": -7.90625,
+      "logits/chosen": -8.625,
+      "logits/rejected": -7.84375,
+      "logps/chosen": -920.0,
+      "logps/rejected": -1456.0,
+      "loss": 0.35,
+      "rewards/accuracies": 0.875,
+      "rewards/chosen": -5.3125,
+      "rewards/margins": 3.046875,
+      "rewards/rejected": -8.375,
       "step": 780
     },
     {
       "epoch": 0.8272251308900523,
-      "grad_norm": 2.2317066770296665,
+      "grad_norm": 2.1405872613327643,
       "learning_rate": 4.544866150565985e-05,
-      "logits/chosen": -12.125,
-      "logits/rejected": -10.75,
-      "logps/chosen": -868.0,
-      "logps/rejected": -1344.0,
-      "loss": 0.3745,
-      "rewards/accuracies": 0.8374999761581421,
-      "rewards/chosen": -4.71875,
-      "rewards/margins": 2.53125,
-      "rewards/rejected": -7.25,
+      "logits/chosen": -8.4375,
+      "logits/rejected": -7.1875,
+      "logps/chosen": -808.0,
+      "logps/rejected": -1312.0,
+      "loss": 0.3831,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -4.15625,
+      "rewards/margins": 2.71875,
+      "rewards/rejected": -6.875,
       "step": 790
     },
     {
       "epoch": 0.837696335078534,
-      "grad_norm": 2.0984539072740658,
+      "grad_norm": 2.2176021491833477,
       "learning_rate": 4.527188176466051e-05,
-      "logits/chosen": -11.125,
-      "logits/rejected": -10.1875,
-      "logps/chosen": -904.0,
-      "logps/rejected": -1208.0,
-      "loss": 0.384,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.78125,
-      "rewards/margins": 1.8359375,
-      "rewards/rejected": -6.625,
+      "logits/chosen": -7.0,
+      "logits/rejected": -6.25,
+      "logps/chosen": -888.0,
+      "logps/rejected": -1184.0,
+      "loss": 0.4087,
+      "rewards/accuracies": 0.7437499761581421,
+      "rewards/chosen": -4.59375,
+      "rewards/margins": 1.8046875,
+      "rewards/rejected": -6.40625,
       "step": 800
     },
     {
       "epoch": 0.837696335078534,
-      "eval_logits/chosen": -8.6875,
-      "eval_logits/rejected": -7.21875,
-      "eval_logps/chosen": -844.0,
-      "eval_logps/rejected": -892.0,
-      "eval_loss": 0.8198124766349792,
-      "eval_rewards/accuracies": 0.63671875,
-      "eval_rewards/chosen": -5.25,
-      "eval_rewards/margins": 0.77734375,
-      "eval_rewards/rejected": -6.03125,
-      "eval_runtime": 47.0705,
-      "eval_samples_per_second": 42.489,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -5.84375,
+      "eval_logits/rejected": -4.375,
+      "eval_logps/chosen": -816.0,
+      "eval_logps/rejected": -864.0,
+      "eval_loss": 0.785937488079071,
+      "eval_rewards/accuracies": 0.611328125,
+      "eval_rewards/chosen": -4.96875,
+      "eval_rewards/margins": 0.79296875,
+      "eval_rewards/rejected": -5.75,
+      "eval_runtime": 46.5395,
+      "eval_samples_per_second": 42.974,
+      "eval_steps_per_second": 0.688,
       "step": 800
     },
     {
       "epoch": 0.8481675392670157,
-      "grad_norm": 3.617005189722071,
+      "grad_norm": 1.8694729486945647,
       "learning_rate": 4.509209163472357e-05,
-      "logits/chosen": -9.0625,
-      "logits/rejected": -7.9375,
-      "logps/chosen": -968.0,
-      "logps/rejected": -1400.0,
-      "loss": 0.3517,
-      "rewards/accuracies": 0.8125,
-      "rewards/chosen": -5.5625,
-      "rewards/margins": 2.5,
-      "rewards/rejected": -8.0625,
+      "logits/chosen": -7.875,
+      "logits/rejected": -7.0625,
+      "logps/chosen": -864.0,
+      "logps/rejected": -1280.0,
+      "loss": 0.344,
+      "rewards/accuracies": 0.8062499761581421,
+      "rewards/chosen": -4.53125,
+      "rewards/margins": 2.34375,
+      "rewards/rejected": -6.875,
       "step": 810
     },
     {
       "epoch": 0.8586387434554974,
-      "grad_norm": 2.4091883002327696,
+      "grad_norm": 1.7641224846920291,
       "learning_rate": 4.4909317814811896e-05,
-      "logits/chosen": -10.0625,
-      "logits/rejected": -9.0,
-      "logps/chosen": -896.0,
-      "logps/rejected": -1384.0,
-      "loss": 0.421,
-      "rewards/accuracies": 0.8125,
-      "rewards/chosen": -4.96875,
-      "rewards/margins": 2.59375,
-      "rewards/rejected": -7.5625,
+      "logits/chosen": -9.5,
+      "logits/rejected": -8.6875,
+      "logps/chosen": -760.0,
+      "logps/rejected": -1216.0,
+      "loss": 0.4136,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -3.640625,
+      "rewards/margins": 2.265625,
+      "rewards/rejected": -5.90625,
       "step": 820
     },
     {
       "epoch": 0.8691099476439791,
-      "grad_norm": 1.750962791430712,
+      "grad_norm": 1.7240672311282905,
       "learning_rate": 4.472358744696845e-05,
-      "logits/chosen": -11.9375,
-      "logits/rejected": -10.4375,
-      "logps/chosen": -772.0,
-      "logps/rejected": -1168.0,
-      "loss": 0.3731,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -3.734375,
-      "rewards/margins": 2.265625,
-      "rewards/rejected": -6.0,
+      "logits/chosen": -10.1875,
+      "logits/rejected": -8.875,
+      "logps/chosen": -720.0,
+      "logps/rejected": -1120.0,
+      "loss": 0.3659,
+      "rewards/accuracies": 0.824999988079071,
+      "rewards/chosen": -3.1875,
+      "rewards/margins": 2.328125,
+      "rewards/rejected": -5.53125,
       "step": 830
     },
     {
       "epoch": 0.8795811518324608,
-      "grad_norm": 4.3978815491579715,
+      "grad_norm": 4.064865108176253,
       "learning_rate": 4.4534928112285645e-05,
-      "logits/chosen": -7.6875,
-      "logits/rejected": -6.625,
-      "logps/chosen": -1040.0,
-      "logps/rejected": -1560.0,
-      "loss": 0.3799,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -6.34375,
-      "rewards/margins": 3.015625,
-      "rewards/rejected": -9.375,
+      "logits/chosen": -6.59375,
+      "logits/rejected": -5.90625,
+      "logps/chosen": -892.0,
+      "logps/rejected": -1376.0,
+      "loss": 0.3973,
+      "rewards/accuracies": 0.793749988079071,
+      "rewards/chosen": -4.84375,
+      "rewards/margins": 2.671875,
+      "rewards/rejected": -7.53125,
       "step": 840
     },
     {
       "epoch": 0.8900523560209425,
-      "grad_norm": 2.4350082449341772,
+      "grad_norm": 2.592229729817839,
       "learning_rate": 4.43433678268096e-05,
-      "logits/chosen": -7.0625,
-      "logits/rejected": -6.1875,
-      "logps/chosen": -1020.0,
-      "logps/rejected": -1496.0,
-      "loss": 0.3835,
-      "rewards/accuracies": 0.8500000238418579,
-      "rewards/chosen": -6.21875,
-      "rewards/margins": 2.59375,
-      "rewards/rejected": -8.8125,
+      "logits/chosen": -6.46875,
+      "logits/rejected": -5.8125,
+      "logps/chosen": -864.0,
+      "logps/rejected": -1328.0,
+      "loss": 0.3752,
+      "rewards/accuracies": 0.8687499761581421,
+      "rewards/chosen": -4.65625,
+      "rewards/margins": 2.546875,
+      "rewards/rejected": -7.1875,
       "step": 850
     },
     {
       "epoch": 0.900523560209424,
-      "grad_norm": 3.243850899843546,
+      "grad_norm": 2.909165564834623,
       "learning_rate": 4.414893503737968e-05,
-      "logits/chosen": -8.0625,
-      "logits/rejected": -6.53125,
-      "logps/chosen": -852.0,
-      "logps/rejected": -1328.0,
-      "loss": 0.325,
-      "rewards/accuracies": 0.8374999761581421,
-      "rewards/chosen": -4.8125,
+      "logits/chosen": -7.09375,
+      "logits/rejected": -6.0625,
+      "logps/chosen": -776.0,
+      "logps/rejected": -1256.0,
+      "loss": 0.3457,
+      "rewards/accuracies": 0.78125,
+      "rewards/chosen": -4.0625,
       "rewards/margins": 2.453125,
-      "rewards/rejected": -7.25,
+      "rewards/rejected": -6.5,
       "step": 860
     },
     {
       "epoch": 0.9109947643979057,
-      "grad_norm": 3.5841949842500322,
+      "grad_norm": 3.837090356580824,
       "learning_rate": 4.395165861740414e-05,
-      "logits/chosen": -7.34375,
-      "logits/rejected": -6.625,
-      "logps/chosen": -1024.0,
-      "logps/rejected": -1560.0,
-      "loss": 0.3574,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -6.5,
-      "rewards/margins": 3.359375,
-      "rewards/rejected": -9.875,
+      "logits/chosen": -7.5625,
+      "logits/rejected": -6.875,
+      "logps/chosen": -840.0,
+      "logps/rejected": -1320.0,
+      "loss": 0.352,
+      "rewards/accuracies": 0.7875000238418579,
+      "rewards/chosen": -4.65625,
+      "rewards/margins": 2.859375,
+      "rewards/rejected": -7.53125,
       "step": 870
     },
     {
       "epoch": 0.9214659685863874,
-      "grad_norm": 2.1992246953147747,
+      "grad_norm": 2.4185036112555367,
       "learning_rate": 4.375156786257241e-05,
-      "logits/chosen": -9.875,
-      "logits/rejected": -8.625,
-      "logps/chosen": -1008.0,
-      "logps/rejected": -1432.0,
-      "loss": 0.4129,
+      "logits/chosen": -8.0625,
+      "logits/rejected": -7.25,
+      "logps/chosen": -956.0,
+      "logps/rejected": -1368.0,
+      "loss": 0.42,
       "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -5.71875,
-      "rewards/margins": 2.640625,
-      "rewards/rejected": -8.375,
+      "rewards/chosen": -5.1875,
+      "rewards/margins": 2.5,
+      "rewards/rejected": -7.71875,
       "step": 880
     },
     {
       "epoch": 0.9319371727748691,
-      "grad_norm": 3.2427874400780143,
+      "grad_norm": 2.452906281212872,
       "learning_rate": 4.354869248650466e-05,
-      "logits/chosen": -11.75,
-      "logits/rejected": -10.25,
-      "logps/chosen": -776.0,
-      "logps/rejected": -1168.0,
-      "loss": 0.3742,
-      "rewards/accuracies": 0.831250011920929,
-      "rewards/chosen": -3.6875,
-      "rewards/margins": 1.828125,
-      "rewards/rejected": -5.53125,
+      "logits/chosen": -9.3125,
+      "logits/rejected": -8.0,
+      "logps/chosen": -764.0,
+      "logps/rejected": -1176.0,
+      "loss": 0.3538,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -3.578125,
+      "rewards/margins": 2.0625,
+      "rewards/rejected": -5.625,
       "step": 890
     },
     {
       "epoch": 0.9424083769633508,
-      "grad_norm": 4.560527139654481,
+      "grad_norm": 4.385511564544654,
       "learning_rate": 4.334306261633931e-05,
-      "logits/chosen": -9.5625,
-      "logits/rejected": -8.0625,
-      "logps/chosen": -980.0,
-      "logps/rejected": -1536.0,
-      "loss": 0.3594,
-      "rewards/accuracies": 0.793749988079071,
-      "rewards/chosen": -5.5,
-      "rewards/margins": 3.078125,
-      "rewards/rejected": -8.5625,
+      "logits/chosen": -6.6875,
+      "logits/rejected": -5.34375,
+      "logps/chosen": -996.0,
+      "logps/rejected": -1608.0,
+      "loss": 0.3409,
+      "rewards/accuracies": 0.831250011920929,
+      "rewards/chosen": -5.65625,
+      "rewards/margins": 3.625,
+      "rewards/rejected": -9.25,
       "step": 900
     },
     {
       "epoch": 0.9424083769633508,
-      "eval_logits/chosen": -7.09375,
-      "eval_logits/rejected": -5.71875,
-      "eval_logps/chosen": -1040.0,
-      "eval_logps/rejected": -1128.0,
-      "eval_loss": 1.0285311937332153,
-      "eval_rewards/accuracies": 0.62890625,
-      "eval_rewards/chosen": -7.25,
-      "eval_rewards/margins": 1.171875,
-      "eval_rewards/rejected": -8.4375,
-      "eval_runtime": 47.0901,
-      "eval_samples_per_second": 42.472,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -4.3125,
+      "eval_logits/rejected": -2.859375,
+      "eval_logps/chosen": -1064.0,
+      "eval_logps/rejected": -1160.0,
+      "eval_loss": 0.9967343807220459,
+      "eval_rewards/accuracies": 0.625,
+      "eval_rewards/chosen": -7.46875,
+      "eval_rewards/margins": 1.203125,
+      "eval_rewards/rejected": -8.6875,
+      "eval_runtime": 46.5542,
+      "eval_samples_per_second": 42.961,
+      "eval_steps_per_second": 0.687,
       "step": 900
     },
     {
       "epoch": 0.9528795811518325,
-      "grad_norm": 2.7133624345526663,
+      "grad_norm": 3.734087065103474,
       "learning_rate": 4.313470878825911e-05,
-      "logits/chosen": -8.75,
-      "logits/rejected": -7.625,
-      "logps/chosen": -1048.0,
-      "logps/rejected": -1560.0,
-      "loss": 0.3425,
+      "logits/chosen": -6.15625,
+      "logits/rejected": -5.3125,
+      "logps/chosen": -1064.0,
+      "logps/rejected": -1600.0,
+      "loss": 0.3588,
       "rewards/accuracies": 0.8687499761581421,
-      "rewards/chosen": -6.375,
-      "rewards/margins": 3.34375,
-      "rewards/rejected": -9.75,
+      "rewards/chosen": -6.5625,
+      "rewards/margins": 3.515625,
+      "rewards/rejected": -10.0625,
       "step": 910
     },
     {
       "epoch": 0.9633507853403142,
-      "grad_norm": 2.413433929191455,
+      "grad_norm": 2.4979823065196096,
       "learning_rate": 4.292366194295652e-05,
-      "logits/chosen": -8.4375,
-      "logits/rejected": -7.65625,
-      "logps/chosen": -944.0,
-      "logps/rejected": -1432.0,
-      "loss": 0.3655,
-      "rewards/accuracies": 0.8187500238418579,
-      "rewards/chosen": -5.40625,
-      "rewards/margins": 2.609375,
-      "rewards/rejected": -8.0,
+      "logits/chosen": -6.3125,
+      "logits/rejected": -5.75,
+      "logps/chosen": -916.0,
+      "logps/rejected": -1400.0,
+      "loss": 0.3693,
+      "rewards/accuracies": 0.862500011920929,
+      "rewards/chosen": -5.125,
+      "rewards/margins": 2.546875,
+      "rewards/rejected": -7.65625,
       "step": 920
     },
     {
       "epoch": 0.9738219895287958,
-      "grad_norm": 2.408759101192053,
+      "grad_norm": 2.8324619934180646,
       "learning_rate": 4.2709953421038964e-05,
-      "logits/chosen": -7.6875,
-      "logits/rejected": -6.96875,
-      "logps/chosen": -896.0,
-      "logps/rejected": -1296.0,
-      "loss": 0.3685,
+      "logits/chosen": -6.40625,
+      "logits/rejected": -5.75,
+      "logps/chosen": -820.0,
+      "logps/rejected": -1208.0,
+      "loss": 0.3661,
       "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -5.125,
-      "rewards/margins": 1.953125,
-      "rewards/rejected": -7.0625,
+      "rewards/chosen": -4.375,
+      "rewards/margins": 1.8515625,
+      "rewards/rejected": -6.21875,
       "step": 930
     },
     {
       "epoch": 0.9842931937172775,
-      "grad_norm": 3.3791897749286695,
+      "grad_norm": 2.8749777084985366,
       "learning_rate": 4.249361495837478e-05,
-      "logits/chosen": -7.96875,
-      "logits/rejected": -6.5625,
-      "logps/chosen": -868.0,
-      "logps/rejected": -1272.0,
-      "loss": 0.4109,
-      "rewards/accuracies": 0.800000011920929,
-      "rewards/chosen": -4.71875,
-      "rewards/margins": 2.078125,
-      "rewards/rejected": -6.8125,
+      "logits/chosen": -6.59375,
+      "logits/rejected": -5.375,
+      "logps/chosen": -832.0,
+      "logps/rejected": -1248.0,
+      "loss": 0.3879,
+      "rewards/accuracies": 0.8125,
+      "rewards/chosen": -4.375,
+      "rewards/margins": 2.21875,
+      "rewards/rejected": -6.59375,
       "step": 940
     },
     {
       "epoch": 0.9947643979057592,
-      "grad_norm": 2.4290280048530506,
+      "grad_norm": 1.9755194720353868,
       "learning_rate": 4.227467868138035e-05,
-      "logits/chosen": -7.53125,
-      "logits/rejected": -6.40625,
-      "logps/chosen": -900.0,
-      "logps/rejected": -1432.0,
-      "loss": 0.335,
-      "rewards/accuracies": 0.824999988079071,
-      "rewards/chosen": -4.96875,
-      "rewards/margins": 2.8125,
-      "rewards/rejected": -7.78125,
+      "logits/chosen": -6.0,
+      "logits/rejected": -5.125,
+      "logps/chosen": -836.0,
+      "logps/rejected": -1352.0,
+      "loss": 0.3442,
+      "rewards/accuracies": 0.8374999761581421,
+      "rewards/chosen": -4.34375,
+      "rewards/margins": 2.6875,
+      "rewards/rejected": -7.03125,
       "step": 950
     },
     {
       "epoch": 1.0052356020942408,
-      "grad_norm": 1.688081881995257,
+      "grad_norm": 1.5746176571675845,
       "learning_rate": 4.205317710224938e-05,
-      "logits/chosen": -8.5,
-      "logits/rejected": -7.0,
-      "logps/chosen": -912.0,
-      "logps/rejected": -1440.0,
-      "loss": 0.2601,
-      "rewards/accuracies": 0.90625,
-      "rewards/chosen": -4.9375,
-      "rewards/margins": 3.421875,
-      "rewards/rejected": -8.375,
+      "logits/chosen": -7.09375,
+      "logits/rejected": -6.03125,
+      "logps/chosen": -796.0,
+      "logps/rejected": -1304.0,
+      "loss": 0.2629,
+      "rewards/accuracies": 0.893750011920929,
+      "rewards/chosen": -3.796875,
+      "rewards/margins": 3.25,
+      "rewards/rejected": -7.03125,
       "step": 960
     },
     {
       "epoch": 1.0157068062827226,
-      "grad_norm": 1.0155288053060518,
+      "grad_norm": 0.7875953806963687,
       "learning_rate": 4.1829143114124733e-05,
-      "logits/chosen": -8.1875,
-      "logits/rejected": -6.5,
-      "logps/chosen": -984.0,
-      "logps/rejected": -1536.0,
-      "loss": 0.1329,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -5.6875,
-      "rewards/margins": 4.4375,
-      "rewards/rejected": -10.125,
+      "logits/chosen": -6.71875,
+      "logits/rejected": -5.375,
+      "logps/chosen": -860.0,
+      "logps/rejected": -1416.0,
+      "loss": 0.126,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -4.46875,
+      "rewards/margins": 4.5,
+      "rewards/rejected": -8.9375,
       "step": 970
     },
     {
       "epoch": 1.0261780104712042,
-      "grad_norm": 2.518665731566414,
+      "grad_norm": 2.2339789509319874,
       "learning_rate": 4.160260998621387e-05,
-      "logits/chosen": -7.0625,
-      "logits/rejected": -5.4375,
-      "logps/chosen": -1280.0,
-      "logps/rejected": -2000.0,
-      "loss": 0.1042,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.375,
-      "rewards/margins": 5.84375,
-      "rewards/rejected": -14.1875,
+      "logits/chosen": -6.34375,
+      "logits/rejected": -4.71875,
+      "logps/chosen": -1104.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.107,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -6.5625,
+      "rewards/margins": 6.09375,
+      "rewards/rejected": -12.6875,
       "step": 980
     },
     {
       "epoch": 1.036649214659686,
-      "grad_norm": 1.3435112945729693,
+      "grad_norm": 2.1057856095448555,
       "learning_rate": 4.137361135884824e-05,
-      "logits/chosen": -6.96875,
-      "logits/rejected": -5.78125,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -2128.0,
-      "loss": 0.0906,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.1875,
-      "rewards/margins": 6.6875,
-      "rewards/rejected": -14.875,
-      "step": 990
-    },
+      "logits/chosen": -5.375,
+      "logits/rejected": -4.40625,
+      "logps/chosen": -1088.0,
+      "logps/rejected": -1968.0,
+      "loss": 0.0986,
+      "rewards/accuracies": 0.9437500238418579,
+      "rewards/chosen": -6.71875,
+      "rewards/margins": 6.53125,
+      "rewards/rejected": -13.25,
+      "step": 990
+    },
     {
       "epoch": 1.0471204188481675,
-      "grad_norm": 0.8826496148665905,
+      "grad_norm": 1.5967919282724123,
       "learning_rate": 4.114218123848777e-05,
-      "logits/chosen": -5.625,
-      "logits/rejected": -4.46875,
-      "logps/chosen": -1272.0,
-      "logps/rejected": -2160.0,
-      "loss": 0.0833,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -8.875,
-      "rewards/margins": 6.9375,
-      "rewards/rejected": -15.75,
+      "logits/chosen": -6.09375,
+      "logits/rejected": -4.8125,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1912.0,
+      "loss": 0.095,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.6875,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -13.3125,
       "step": 1000
     },
     {
       "epoch": 1.0471204188481675,
-      "eval_logits/chosen": -4.09375,
-      "eval_logits/rejected": -2.578125,
-      "eval_logps/chosen": -1240.0,
-      "eval_logps/rejected": -1328.0,
-      "eval_loss": 1.1501250267028809,
-      "eval_rewards/accuracies": 0.60546875,
-      "eval_rewards/chosen": -9.25,
-      "eval_rewards/margins": 1.15625,
-      "eval_rewards/rejected": -10.375,
-      "eval_runtime": 47.0762,
-      "eval_samples_per_second": 42.484,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -6.21875,
+      "eval_logits/rejected": -4.5625,
+      "eval_logps/chosen": -1012.0,
+      "eval_logps/rejected": -1112.0,
+      "eval_loss": 1.2728281021118164,
+      "eval_rewards/accuracies": 0.599609375,
+      "eval_rewards/chosen": -6.9375,
+      "eval_rewards/margins": 1.3046875,
+      "eval_rewards/rejected": -8.25,
+      "eval_runtime": 46.7967,
+      "eval_samples_per_second": 42.738,
+      "eval_steps_per_second": 0.684,
       "step": 1000
     },
     {
       "epoch": 1.057591623036649,
-      "grad_norm": 2.4475079849098083,
+      "grad_norm": 2.1215536982575056,
       "learning_rate": 4.0908353992670824e-05,
-      "logits/chosen": -6.84375,
-      "logits/rejected": -5.53125,
-      "logps/chosen": -1072.0,
-      "logps/rejected": -1736.0,
-      "loss": 0.1142,
+      "logits/chosen": -7.34375,
+      "logits/rejected": -5.75,
+      "logps/chosen": -960.0,
+      "logps/rejected": -1672.0,
+      "loss": 0.1244,
       "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -6.8125,
-      "rewards/margins": 5.40625,
-      "rewards/rejected": -12.25,
+      "rewards/chosen": -5.65625,
+      "rewards/margins": 5.9375,
+      "rewards/rejected": -11.625,
       "step": 1010
     },
     {
       "epoch": 1.068062827225131,
-      "grad_norm": 2.4653747614145973,
+      "grad_norm": 2.067632907381316,
       "learning_rate": 4.067216434491059e-05,
-      "logits/chosen": -6.75,
-      "logits/rejected": -5.84375,
+      "logits/chosen": -5.8125,
+      "logits/rejected": -4.96875,
       "logps/chosen": -1088.0,
-      "logps/rejected": -1832.0,
-      "loss": 0.1033,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -6.875,
-      "rewards/margins": 5.65625,
-      "rewards/rejected": -12.5625,
+      "logps/rejected": -1864.0,
+      "loss": 0.1076,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.84375,
+      "rewards/margins": 6.0625,
+      "rewards/rejected": -12.9375,
       "step": 1020
     },
     {
       "epoch": 1.0785340314136125,
-      "grad_norm": 2.1310452666397706,
+      "grad_norm": 2.8539420054877915,
       "learning_rate": 4.043364736953866e-05,
-      "logits/chosen": -6.0,
-      "logits/rejected": -4.53125,
-      "logps/chosen": -1184.0,
-      "logps/rejected": -1944.0,
-      "loss": 0.1132,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -7.375,
-      "rewards/margins": 5.65625,
-      "rewards/rejected": -13.0625,
+      "logits/chosen": -5.78125,
+      "logits/rejected": -4.4375,
+      "logps/chosen": -1040.0,
+      "logps/rejected": -1768.0,
+      "loss": 0.1148,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -5.9375,
+      "rewards/margins": 5.34375,
+      "rewards/rejected": -11.3125,
       "step": 1030
     },
     {
       "epoch": 1.0890052356020943,
-      "grad_norm": 3.028771262320732,
+      "grad_norm": 2.0337473126810734,
       "learning_rate": 4.019283848649643e-05,
-      "logits/chosen": -5.65625,
+      "logits/chosen": -5.4375,
       "logits/rejected": -4.21875,
-      "logps/chosen": -1208.0,
-      "logps/rejected": -1976.0,
-      "loss": 0.139,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -8.0,
-      "rewards/margins": 5.84375,
-      "rewards/rejected": -13.8125,
+      "logps/chosen": -992.0,
+      "logps/rejected": -1720.0,
+      "loss": 0.1297,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -5.84375,
+      "rewards/margins": 5.46875,
+      "rewards/rejected": -11.3125,
       "step": 1040
     },
     {
       "epoch": 1.0994764397905759,
-      "grad_norm": 1.127373422789099,
+      "grad_norm": 1.5872238897114932,
       "learning_rate": 3.9949773456075215e-05,
-      "logits/chosen": -5.34375,
-      "logits/rejected": -3.96875,
-      "logps/chosen": -1040.0,
-      "logps/rejected": -1704.0,
-      "loss": 0.089,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -6.65625,
-      "rewards/margins": 4.59375,
-      "rewards/rejected": -11.25,
+      "logits/chosen": -4.09375,
+      "logits/rejected": -3.203125,
+      "logps/chosen": -988.0,
+      "logps/rejected": -1696.0,
+      "loss": 0.0962,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.15625,
+      "rewards/margins": 5.03125,
+      "rewards/rejected": -11.1875,
       "step": 1050
     },
     {
       "epoch": 1.1099476439790577,
-      "grad_norm": 2.952551522192677,
+      "grad_norm": 1.9359221881419222,
       "learning_rate": 3.970448837360585e-05,
-      "logits/chosen": -6.09375,
-      "logits/rejected": -4.71875,
-      "logps/chosen": -1360.0,
-      "logps/rejected": -2176.0,
-      "loss": 0.09,
-      "rewards/accuracies": 0.9375,
-      "rewards/chosen": -9.125,
-      "rewards/margins": 5.875,
-      "rewards/rejected": -15.0,
+      "logits/chosen": -5.4375,
+      "logits/rejected": -4.3125,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.1062,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.125,
+      "rewards/margins": 5.59375,
+      "rewards/rejected": -11.6875,
       "step": 1060
     },
     {
       "epoch": 1.1204188481675392,
-      "grad_norm": 1.0412579401484154,
+      "grad_norm": 1.2919228226744321,
       "learning_rate": 3.945701966409846e-05,
-      "logits/chosen": -6.0625,
+      "logits/chosen": -5.375,
       "logits/rejected": -4.625,
-      "logps/chosen": -1336.0,
-      "logps/rejected": -2400.0,
-      "loss": 0.0927,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -9.3125,
-      "rewards/margins": 8.1875,
-      "rewards/rejected": -17.5,
+      "logps/chosen": -976.0,
+      "logps/rejected": -1904.0,
+      "loss": 0.0959,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -5.75,
+      "rewards/margins": 6.75,
+      "rewards/rejected": -12.5,
       "step": 1070
     },
     {
       "epoch": 1.130890052356021,
-      "grad_norm": 1.2484808104674319,
+      "grad_norm": 1.1267122238659721,
       "learning_rate": 3.9207404076833374e-05,
-      "logits/chosen": -7.6875,
-      "logits/rejected": -5.3125,
-      "logps/chosen": -1152.0,
-      "logps/rejected": -2096.0,
-      "loss": 0.1113,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -7.125,
-      "rewards/margins": 7.25,
-      "rewards/rejected": -14.375,
+      "logits/chosen": -4.78125,
+      "logits/rejected": -3.1875,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1960.0,
+      "loss": 0.0989,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.3125,
+      "rewards/margins": 6.6875,
+      "rewards/rejected": -13.0,
       "step": 1080
     },
     {
       "epoch": 1.1413612565445026,
-      "grad_norm": 1.708984494773098,
+      "grad_norm": 0.772494809451675,
       "learning_rate": 3.895567867990379e-05,
-      "logits/chosen": -7.9375,
-      "logits/rejected": -5.5625,
-      "logps/chosen": -1128.0,
-      "logps/rejected": -2008.0,
-      "loss": 0.1217,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -6.65625,
-      "rewards/margins": 6.625,
-      "rewards/rejected": -13.3125,
+      "logits/chosen": -5.125,
+      "logits/rejected": -3.71875,
+      "logps/chosen": -1152.0,
+      "logps/rejected": -2032.0,
+      "loss": 0.1256,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -6.875,
+      "rewards/margins": 6.6875,
+      "rewards/rejected": -13.5625,
       "step": 1090
     },
     {
       "epoch": 1.1518324607329844,
-      "grad_norm": 1.206510038333789,
+      "grad_norm": 1.4742667836269225,
       "learning_rate": 3.870188085471114e-05,
-      "logits/chosen": -6.0625,
-      "logits/rejected": -4.90625,
-      "logps/chosen": -1184.0,
-      "logps/rejected": -1912.0,
-      "loss": 0.0953,
+      "logits/chosen": -5.28125,
+      "logits/rejected": -4.1875,
+      "logps/chosen": -1064.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.0768,
       "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -7.71875,
-      "rewards/margins": 5.71875,
-      "rewards/rejected": -13.4375,
+      "rewards/chosen": -6.53125,
+      "rewards/margins": 6.28125,
+      "rewards/rejected": -12.8125,
       "step": 1100
     },
     {
       "epoch": 1.1518324607329844,
-      "eval_logits/chosen": -3.59375,
-      "eval_logits/rejected": -2.0,
-      "eval_logps/chosen": -1264.0,
-      "eval_logps/rejected": -1360.0,
-      "eval_loss": 1.1999530792236328,
-      "eval_rewards/accuracies": 0.59375,
-      "eval_rewards/chosen": -9.5,
-      "eval_rewards/margins": 1.2421875,
-      "eval_rewards/rejected": -10.75,
-      "eval_runtime": 47.2586,
-      "eval_samples_per_second": 42.32,
-      "eval_steps_per_second": 0.677,
+      "eval_logits/chosen": -2.46875,
+      "eval_logits/rejected": -0.87109375,
+      "eval_logps/chosen": -1352.0,
+      "eval_logps/rejected": -1480.0,
+      "eval_loss": 1.4989843368530273,
+      "eval_rewards/accuracies": 0.59765625,
+      "eval_rewards/chosen": -10.375,
+      "eval_rewards/margins": 1.5625,
+      "eval_rewards/rejected": -11.9375,
+      "eval_runtime": 46.5426,
+      "eval_samples_per_second": 42.971,
+      "eval_steps_per_second": 0.688,
       "step": 1100
     },
     {
       "epoch": 1.162303664921466,
-      "grad_norm": 1.1820958698463258,
+      "grad_norm": 1.6787239199749102,
       "learning_rate": 3.844604829041395e-05,
-      "logits/chosen": -6.53125,
-      "logits/rejected": -5.03125,
-      "logps/chosen": -1056.0,
-      "logps/rejected": -1768.0,
-      "loss": 0.094,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -6.78125,
-      "rewards/margins": 5.8125,
-      "rewards/rejected": -12.5625,
+      "logits/chosen": -4.4375,
+      "logits/rejected": -3.46875,
+      "logps/chosen": -1088.0,
+      "logps/rejected": -1880.0,
+      "loss": 0.1048,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -7.09375,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -13.6875,
       "step": 1110
     },
     {
       "epoch": 1.1727748691099475,
-      "grad_norm": 1.9618932155541804,
+      "grad_norm": 2.410603933803685,
       "learning_rate": 3.818821897833095e-05,
-      "logits/chosen": -6.875,
-      "logits/rejected": -5.46875,
-      "logps/chosen": -1040.0,
-      "logps/rejected": -1736.0,
-      "loss": 0.1004,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -6.25,
-      "rewards/margins": 5.25,
-      "rewards/rejected": -11.5,
+      "logits/chosen": -4.375,
+      "logits/rejected": -3.734375,
+      "logps/chosen": -1032.0,
+      "logps/rejected": -1752.0,
+      "loss": 0.106,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -6.1875,
+      "rewards/margins": 5.4375,
+      "rewards/rejected": -11.625,
       "step": 1120
     },
     {
       "epoch": 1.1832460732984293,
-      "grad_norm": 2.00712723452005,
+      "grad_norm": 3.0656518656919967,
       "learning_rate": 3.792843120629935e-05,
-      "logits/chosen": -5.90625,
-      "logits/rejected": -4.09375,
-      "logps/chosen": -1096.0,
-      "logps/rejected": -1912.0,
-      "loss": 0.1232,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -7.03125,
-      "rewards/margins": 5.90625,
-      "rewards/rejected": -12.9375,
+      "logits/chosen": -5.1875,
+      "logits/rejected": -3.671875,
+      "logps/chosen": -948.0,
+      "logps/rejected": -1792.0,
+      "loss": 0.1154,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -5.59375,
+      "rewards/margins": 6.0625,
+      "rewards/rejected": -11.6875,
       "step": 1130
     },
     {
       "epoch": 1.193717277486911,
-      "grad_norm": 3.233388775541502,
+      "grad_norm": 2.7071402454075657,
       "learning_rate": 3.766672355298907e-05,
-      "logits/chosen": -7.3125,
-      "logits/rejected": -5.28125,
-      "logps/chosen": -1104.0,
-      "logps/rejected": -1880.0,
-      "loss": 0.1014,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -7.4375,
-      "rewards/margins": 5.84375,
-      "rewards/rejected": -13.25,
+      "logits/chosen": -6.375,
+      "logits/rejected": -5.03125,
+      "logps/chosen": -940.0,
+      "logps/rejected": -1728.0,
+      "loss": 0.1155,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -5.75,
+      "rewards/margins": 6.03125,
+      "rewards/rejected": -11.8125,
       "step": 1140
     },
     {
       "epoch": 1.2041884816753927,
-      "grad_norm": 4.471216632991261,
+      "grad_norm": 2.7929209995388273,
       "learning_rate": 3.740313488217373e-05,
-      "logits/chosen": -7.15625,
-      "logits/rejected": -4.6875,
-      "logps/chosen": -1280.0,
-      "logps/rejected": -2144.0,
-      "loss": 0.11,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.9375,
+      "logits/chosen": -5.75,
+      "logits/rejected": -4.40625,
+      "logps/chosen": -1104.0,
+      "logps/rejected": -1976.0,
+      "loss": 0.104,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -7.1875,
       "rewards/margins": 7.125,
-      "rewards/rejected": -16.125,
+      "rewards/rejected": -14.3125,
       "step": 1150
     },
     {
       "epoch": 1.2146596858638743,
-      "grad_norm": 2.0905299462725346,
+      "grad_norm": 1.5560329285795047,
       "learning_rate": 3.7137704336959464e-05,
-      "logits/chosen": -4.3125,
-      "logits/rejected": -2.28125,
-      "logps/chosen": -1280.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.1225,
-      "rewards/accuracies": 0.9312499761581421,
-      "rewards/chosen": -9.0,
-      "rewards/margins": 6.5625,
-      "rewards/rejected": -15.5,
+      "logits/chosen": -3.703125,
+      "logits/rejected": -2.328125,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1888.0,
+      "loss": 0.1264,
+      "rewards/accuracies": 0.918749988079071,
+      "rewards/chosen": -6.96875,
+      "rewards/margins": 6.375,
+      "rewards/rejected": -13.3125,
       "step": 1160
     },
     {
       "epoch": 1.225130890052356,
-      "grad_norm": 3.168573900259561,
+      "grad_norm": 1.6305002568776996,
       "learning_rate": 3.687047133397201e-05,
-      "logits/chosen": -5.8125,
-      "logits/rejected": -3.9375,
-      "logps/chosen": -1096.0,
-      "logps/rejected": -1888.0,
-      "loss": 0.1203,
+      "logits/chosen": -4.75,
+      "logits/rejected": -3.421875,
+      "logps/chosen": -932.0,
+      "logps/rejected": -1744.0,
+      "loss": 0.0834,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -7.09375,
-      "rewards/margins": 5.5,
-      "rewards/rejected": -12.5625,
+      "rewards/chosen": -5.46875,
+      "rewards/margins": 5.71875,
+      "rewards/rejected": -11.1875,
       "step": 1170
     },
     {
       "epoch": 1.2356020942408377,
-      "grad_norm": 2.331535911720474,
+      "grad_norm": 1.8016202115284654,
       "learning_rate": 3.6601475557503415e-05,
-      "logits/chosen": -6.34375,
-      "logits/rejected": -3.734375,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1864.0,
-      "loss": 0.0932,
+      "logits/chosen": -4.46875,
+      "logits/rejected": -2.640625,
+      "logps/chosen": -1024.0,
+      "logps/rejected": -1712.0,
+      "loss": 0.1026,
       "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -7.4375,
-      "rewards/margins": 5.8125,
-      "rewards/rejected": -13.25,
+      "rewards/chosen": -6.03125,
+      "rewards/margins": 5.65625,
+      "rewards/rejected": -11.6875,
       "step": 1180
     },
     {
       "epoch": 1.2460732984293195,
-      "grad_norm": 2.124068068600138,
+      "grad_norm": 0.818584910836671,
       "learning_rate": 3.633075695361882e-05,
-      "logits/chosen": -4.75,
-      "logits/rejected": -3.109375,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -2080.0,
-      "loss": 0.1008,
+      "logits/chosen": -2.796875,
+      "logits/rejected": -2.0625,
+      "logps/chosen": -1064.0,
+      "logps/rejected": -1840.0,
+      "loss": 0.1163,
       "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.5,
-      "rewards/margins": 6.375,
-      "rewards/rejected": -14.9375,
+      "rewards/chosen": -6.875,
+      "rewards/margins": 5.5,
+      "rewards/rejected": -12.375,
       "step": 1190
     },
     {
       "epoch": 1.256544502617801,
-      "grad_norm": 2.5186377852766966,
+      "grad_norm": 2.0922326793358503,
       "learning_rate": 3.6058355724224477e-05,
-      "logits/chosen": -7.0,
-      "logits/rejected": -5.1875,
-      "logps/chosen": -1128.0,
-      "logps/rejected": -1880.0,
-      "loss": 0.1198,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -7.34375,
-      "rewards/margins": 5.8125,
-      "rewards/rejected": -13.1875,
+      "logits/chosen": -4.0625,
+      "logits/rejected": -3.21875,
+      "logps/chosen": -964.0,
+      "logps/rejected": -1672.0,
+      "loss": 0.1176,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -5.6875,
+      "rewards/margins": 5.34375,
+      "rewards/rejected": -11.0,
       "step": 1200
     },
     {
       "epoch": 1.256544502617801,
-      "eval_logits/chosen": -4.96875,
-      "eval_logits/rejected": -2.9375,
-      "eval_logps/chosen": -1176.0,
-      "eval_logps/rejected": -1264.0,
-      "eval_loss": 1.2178750038146973,
-      "eval_rewards/accuracies": 0.607421875,
-      "eval_rewards/chosen": -8.5625,
-      "eval_rewards/margins": 1.1640625,
-      "eval_rewards/rejected": -9.75,
-      "eval_runtime": 47.1015,
-      "eval_samples_per_second": 42.461,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": -2.578125,
+      "eval_logits/rejected": -1.0390625,
+      "eval_logps/chosen": -964.0,
+      "eval_logps/rejected": -1040.0,
+      "eval_loss": 1.0658750534057617,
+      "eval_rewards/accuracies": 0.611328125,
+      "eval_rewards/chosen": -6.46875,
+      "eval_rewards/margins": 1.03125,
+      "eval_rewards/rejected": -7.5,
+      "eval_runtime": 46.5279,
+      "eval_samples_per_second": 42.985,
+      "eval_steps_per_second": 0.688,
       "step": 1200
     },
     {
       "epoch": 1.2670157068062826,
-      "grad_norm": 3.4582272711228526,
+      "grad_norm": 3.86986606969869,
       "learning_rate": 3.5784312321097744e-05,
-      "logits/chosen": -6.3125,
-      "logits/rejected": -4.6875,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1928.0,
-      "loss": 0.1032,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -7.5625,
-      "rewards/margins": 5.96875,
-      "rewards/rejected": -13.5,
+      "logits/chosen": -4.21875,
+      "logits/rejected": -3.34375,
+      "logps/chosen": -1000.0,
+      "logps/rejected": -1768.0,
+      "loss": 0.1025,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -5.875,
+      "rewards/margins": 6.0625,
+      "rewards/rejected": -11.9375,
       "step": 1210
     },
     {
       "epoch": 1.2774869109947644,
-      "grad_norm": 2.805737307455421,
+      "grad_norm": 2.8621526796330694,
       "learning_rate": 3.550866743987994e-05,
-      "logits/chosen": -7.15625,
-      "logits/rejected": -5.1875,
-      "logps/chosen": -1104.0,
-      "logps/rejected": -1832.0,
-      "loss": 0.1246,
-      "rewards/accuracies": 0.9375,
-      "rewards/chosen": -6.8125,
-      "rewards/margins": 5.6875,
-      "rewards/rejected": -12.5,
+      "logits/chosen": -3.53125,
+      "logits/rejected": -2.265625,
+      "logps/chosen": -1048.0,
+      "logps/rejected": -1840.0,
+      "loss": 0.1178,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.25,
+      "rewards/margins": 6.375,
+      "rewards/rejected": -12.625,
       "step": 1220
     },
     {
       "epoch": 1.2879581151832462,
-      "grad_norm": 2.284438570390368,
+      "grad_norm": 1.6083788899743747,
       "learning_rate": 3.5231462014033024e-05,
-      "logits/chosen": -7.96875,
-      "logits/rejected": -6.21875,
-      "logps/chosen": -1040.0,
-      "logps/rejected": -1856.0,
-      "loss": 0.0718,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -6.59375,
-      "rewards/margins": 5.90625,
-      "rewards/rejected": -12.5,
+      "logits/chosen": -3.671875,
+      "logits/rejected": -3.140625,
+      "logps/chosen": -960.0,
+      "logps/rejected": -1824.0,
+      "loss": 0.0618,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -5.78125,
+      "rewards/margins": 6.34375,
+      "rewards/rejected": -12.125,
       "step": 1230
     },
     {
       "epoch": 1.2984293193717278,
-      "grad_norm": 2.058840468124629,
+      "grad_norm": 3.1241619249952794,
       "learning_rate": 3.495273720876095e-05,
-      "logits/chosen": -5.125,
-      "logits/rejected": -3.421875,
-      "logps/chosen": -1312.0,
-      "logps/rejected": -2176.0,
-      "loss": 0.0799,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -9.0,
-      "rewards/margins": 7.21875,
-      "rewards/rejected": -16.25,
+      "logits/chosen": -2.796875,
+      "logits/rejected": -1.953125,
+      "logps/chosen": -1120.0,
+      "logps/rejected": -1888.0,
+      "loss": 0.0831,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -7.125,
+      "rewards/margins": 6.25,
+      "rewards/rejected": -13.375,
       "step": 1240
     },
     {
       "epoch": 1.3089005235602094,
-      "grad_norm": 0.9713263018281973,
+      "grad_norm": 2.96566035271667,
       "learning_rate": 3.4672534414896616e-05,
-      "logits/chosen": -5.46875,
-      "logits/rejected": -3.484375,
-      "logps/chosen": -1296.0,
-      "logps/rejected": -2208.0,
-      "loss": 0.0753,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.625,
-      "rewards/margins": 7.25,
-      "rewards/rejected": -15.875,
-      "step": 1250
-    },
-    {
-      "epoch": 1.3193717277486912,
-      "grad_norm": 2.7359457286619695,
+      "logits/chosen": -4.34375,
+      "logits/rejected": -3.109375,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1920.0,
+      "loss": 0.0945,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.25,
+      "rewards/margins": 6.71875,
+      "rewards/rejected": -12.9375,
+      "step": 1250
+    },
+    {
+      "epoch": 1.3193717277486912,
+      "grad_norm": 1.0713128980898727,
       "learning_rate": 3.439089524275526e-05,
-      "logits/chosen": -5.15625,
-      "logits/rejected": -2.984375,
-      "logps/chosen": -1136.0,
-      "logps/rejected": -1936.0,
-      "loss": 0.0976,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -7.46875,
-      "rewards/margins": 6.1875,
-      "rewards/rejected": -13.6875,
+      "logits/chosen": -4.75,
+      "logits/rejected": -3.265625,
+      "logps/chosen": -876.0,
+      "logps/rejected": -1624.0,
+      "loss": 0.1286,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -4.9375,
+      "rewards/margins": 5.59375,
+      "rewards/rejected": -10.5625,
       "step": 1260
     },
     {
       "epoch": 1.3298429319371727,
-      "grad_norm": 1.462005068992466,
+      "grad_norm": 2.589863981994399,
       "learning_rate": 3.410786151595533e-05,
-      "logits/chosen": -5.90625,
-      "logits/rejected": -3.703125,
-      "logps/chosen": -1144.0,
-      "logps/rejected": -1832.0,
-      "loss": 0.0939,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -7.59375,
-      "rewards/margins": 6.0,
-      "rewards/rejected": -13.625,
+      "logits/chosen": -4.875,
+      "logits/rejected": -3.671875,
+      "logps/chosen": -948.0,
+      "logps/rejected": -1616.0,
+      "loss": 0.1002,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -5.65625,
+      "rewards/margins": 5.84375,
+      "rewards/rejected": -11.5,
       "step": 1270
     },
     {
       "epoch": 1.3403141361256545,
-      "grad_norm": 2.901256884923763,
+      "grad_norm": 1.793448282096988,
       "learning_rate": 3.382347526520765e-05,
-      "logits/chosen": -6.25,
-      "logits/rejected": -4.125,
-      "logps/chosen": -1240.0,
-      "logps/rejected": -1992.0,
-      "loss": 0.0974,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.3125,
-      "rewards/margins": 6.1875,
-      "rewards/rejected": -14.5,
+      "logits/chosen": -5.03125,
+      "logits/rejected": -3.84375,
+      "logps/chosen": -992.0,
+      "logps/rejected": -1720.0,
+      "loss": 0.1128,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -5.84375,
+      "rewards/margins": 5.90625,
+      "rewards/rejected": -11.75,
       "step": 1280
     },
     {
       "epoch": 1.350785340314136,
-      "grad_norm": 3.6719217976392375,
+      "grad_norm": 3.5791499760215717,
       "learning_rate": 3.353777872207381e-05,
-      "logits/chosen": -6.875,
-      "logits/rejected": -4.71875,
-      "logps/chosen": -1192.0,
-      "logps/rejected": -2024.0,
-      "loss": 0.0909,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -7.8125,
-      "rewards/margins": 6.375,
-      "rewards/rejected": -14.1875,
+      "logits/chosen": -5.25,
+      "logits/rejected": -4.09375,
+      "logps/chosen": -952.0,
+      "logps/rejected": -1720.0,
+      "loss": 0.0889,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -5.46875,
+      "rewards/margins": 5.6875,
+      "rewards/rejected": -11.125,
       "step": 1290
     },
     {
       "epoch": 1.3612565445026177,
-      "grad_norm": 1.8708980452478416,
+      "grad_norm": 2.0456994847864176,
       "learning_rate": 3.325081431269475e-05,
-      "logits/chosen": -5.1875,
-      "logits/rejected": -3.5625,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -2144.0,
-      "loss": 0.1291,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -8.4375,
-      "rewards/margins": 7.15625,
-      "rewards/rejected": -15.5625,
+      "logits/chosen": -4.4375,
+      "logits/rejected": -3.5,
+      "logps/chosen": -968.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.0885,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -5.96875,
+      "rewards/margins": 6.6875,
+      "rewards/rejected": -12.625,
       "step": 1300
     },
     {
       "epoch": 1.3612565445026177,
-      "eval_logits/chosen": -4.59375,
+      "eval_logits/chosen": -4.28125,
       "eval_logits/rejected": -2.65625,
-      "eval_logps/chosen": -1192.0,
-      "eval_logps/rejected": -1296.0,
-      "eval_loss": 1.165765643119812,
-      "eval_rewards/accuracies": 0.61328125,
-      "eval_rewards/chosen": -8.75,
-      "eval_rewards/margins": 1.296875,
-      "eval_rewards/rejected": -10.0625,
-      "eval_runtime": 47.0911,
-      "eval_samples_per_second": 42.471,
-      "eval_steps_per_second": 0.68,
+      "eval_logps/chosen": -1056.0,
+      "eval_logps/rejected": -1160.0,
+      "eval_loss": 1.180437445640564,
+      "eval_rewards/accuracies": 0.6171875,
+      "eval_rewards/chosen": -7.375,
+      "eval_rewards/margins": 1.3671875,
+      "eval_rewards/rejected": -8.75,
+      "eval_runtime": 46.5476,
+      "eval_samples_per_second": 42.967,
+      "eval_steps_per_second": 0.687,
       "step": 1300
     },
     {
       "epoch": 1.3717277486910995,
-      "grad_norm": 1.2328616934690808,
+      "grad_norm": 1.6377321834646772,
       "learning_rate": 3.296262465149046e-05,
-      "logits/chosen": -6.46875,
-      "logits/rejected": -3.984375,
-      "logps/chosen": -1112.0,
-      "logps/rejected": -1896.0,
-      "loss": 0.0845,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -6.9375,
-      "rewards/margins": 6.1875,
-      "rewards/rejected": -13.0625,
+      "logits/chosen": -6.1875,
+      "logits/rejected": -4.0625,
+      "logps/chosen": -976.0,
+      "logps/rejected": -1840.0,
+      "loss": 0.0825,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -5.59375,
+      "rewards/margins": 6.90625,
+      "rewards/rejected": -12.5,
       "step": 1310
     },
     {
       "epoch": 1.3821989528795813,
-      "grad_norm": 1.1147731501344902,
+      "grad_norm": 0.5405761779118115,
       "learning_rate": 3.2673252534831687e-05,
-      "logits/chosen": -5.28125,
-      "logits/rejected": -2.984375,
-      "logps/chosen": -1120.0,
-      "logps/rejected": -1912.0,
-      "loss": 0.0872,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -7.25,
-      "rewards/margins": 6.0625,
-      "rewards/rejected": -13.3125,
+      "logits/chosen": -5.3125,
+      "logits/rejected": -3.578125,
+      "logps/chosen": -944.0,
+      "logps/rejected": -1712.0,
+      "loss": 0.0891,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -5.5,
+      "rewards/margins": 5.875,
+      "rewards/rejected": -11.375,
       "step": 1320
     },
     {
       "epoch": 1.3926701570680629,
-      "grad_norm": 2.9882590329216883,
+      "grad_norm": 2.202701460590514,
       "learning_rate": 3.2382740934684696e-05,
-      "logits/chosen": -3.21875,
-      "logits/rejected": -1.6875,
-      "logps/chosen": -1320.0,
-      "logps/rejected": -2272.0,
-      "loss": 0.1262,
+      "logits/chosen": -4.28125,
+      "logits/rejected": -2.90625,
+      "logps/chosen": -984.0,
+      "logps/rejected": -1888.0,
+      "loss": 0.1026,
       "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -9.1875,
-      "rewards/margins": 6.875,
-      "rewards/rejected": -16.125,
+      "rewards/chosen": -5.875,
+      "rewards/margins": 6.46875,
+      "rewards/rejected": -12.3125,
       "step": 1330
     },
     {
       "epoch": 1.4031413612565444,
-      "grad_norm": 1.2757300715297355,
+      "grad_norm": 1.1595637711487354,
       "learning_rate": 3.209113299222982e-05,
-      "logits/chosen": -3.125,
-      "logits/rejected": -2.734375,
-      "logps/chosen": -1200.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.0969,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.625,
-      "rewards/margins": 6.1875,
-      "rewards/rejected": -14.8125,
+      "logits/chosen": -3.875,
+      "logits/rejected": -3.546875,
+      "logps/chosen": -936.0,
+      "logps/rejected": -1792.0,
+      "loss": 0.1029,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -5.96875,
+      "rewards/margins": 6.3125,
+      "rewards/rejected": -12.25,
       "step": 1340
     },
     {
       "epoch": 1.4136125654450262,
-      "grad_norm": 2.6096513121300338,
+      "grad_norm": 2.548679541963918,
       "learning_rate": 3.179847201145505e-05,
-      "logits/chosen": -3.671875,
-      "logits/rejected": -2.515625,
-      "logps/chosen": -1280.0,
-      "logps/rejected": -2176.0,
-      "loss": 0.088,
+      "logits/chosen": -3.375,
+      "logits/rejected": -2.546875,
+      "logps/chosen": -1112.0,
+      "logps/rejected": -2064.0,
+      "loss": 0.0955,
       "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.6875,
-      "rewards/margins": 6.4375,
-      "rewards/rejected": -15.125,
+      "rewards/chosen": -6.96875,
+      "rewards/margins": 6.9375,
+      "rewards/rejected": -13.875,
       "step": 1350
     },
     {
       "epoch": 1.4240837696335078,
-      "grad_norm": 1.0809039561122553,
+      "grad_norm": 1.4161150568875678,
       "learning_rate": 3.150480145272528e-05,
-      "logits/chosen": -3.921875,
-      "logits/rejected": -2.984375,
-      "logps/chosen": -1368.0,
-      "logps/rejected": -2192.0,
-      "loss": 0.0968,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -9.6875,
-      "rewards/margins": 6.46875,
-      "rewards/rejected": -16.125,
+      "logits/chosen": -2.9375,
+      "logits/rejected": -2.703125,
+      "logps/chosen": -1160.0,
+      "logps/rejected": -1968.0,
+      "loss": 0.0828,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -7.53125,
+      "rewards/margins": 6.40625,
+      "rewards/rejected": -13.9375,
       "step": 1360
     },
     {
       "epoch": 1.4345549738219896,
-      "grad_norm": 1.2984827230940517,
+      "grad_norm": 1.140738857845775,
       "learning_rate": 3.121016492632848e-05,
-      "logits/chosen": -5.15625,
-      "logits/rejected": -3.53125,
-      "logps/chosen": -1288.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.0998,
+      "logits/chosen": -3.6875,
+      "logits/rejected": -2.6875,
+      "logps/chosen": -1120.0,
+      "logps/rejected": -1944.0,
+      "loss": 0.0989,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.8125,
-      "rewards/margins": 6.53125,
-      "rewards/rejected": -15.375,
+      "rewards/chosen": -7.125,
+      "rewards/margins": 6.5625,
+      "rewards/rejected": -13.6875,
       "step": 1370
     },
     {
       "epoch": 1.4450261780104712,
-      "grad_norm": 2.3153852483970154,
+      "grad_norm": 3.6384376420567937,
       "learning_rate": 3.091460618599951e-05,
-      "logits/chosen": -4.625,
-      "logits/rejected": -2.75,
-      "logps/chosen": -1256.0,
-      "logps/rejected": -2192.0,
-      "loss": 0.0897,
+      "logits/chosen": -3.5,
+      "logits/rejected": -2.3125,
+      "logps/chosen": -1032.0,
+      "logps/rejected": -1912.0,
+      "loss": 0.1019,
       "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -8.625,
-      "rewards/margins": 7.15625,
-      "rewards/rejected": -15.8125,
+      "rewards/chosen": -6.40625,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -13.0,
       "step": 1380
     },
     {
       "epoch": 1.4554973821989527,
-      "grad_norm": 2.568116251434959,
+      "grad_norm": 1.0206494167881448,
       "learning_rate": 3.061816912242265e-05,
-      "logits/chosen": -4.34375,
-      "logits/rejected": -3.03125,
-      "logps/chosen": -1248.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.0867,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.3125,
-      "rewards/margins": 6.3125,
-      "rewards/rejected": -14.625,
+      "logits/chosen": -3.953125,
+      "logits/rejected": -3.015625,
+      "logps/chosen": -1012.0,
+      "logps/rejected": -1856.0,
+      "loss": 0.0795,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -5.96875,
+      "rewards/margins": 6.1875,
+      "rewards/rejected": -12.1875,
       "step": 1390
     },
     {
       "epoch": 1.4659685863874345,
-      "grad_norm": 0.820343372273228,
+      "grad_norm": 0.5869459272359916,
       "learning_rate": 3.0320897756713785e-05,
-      "logits/chosen": -5.125,
-      "logits/rejected": -3.9375,
-      "logps/chosen": -1184.0,
-      "logps/rejected": -2208.0,
-      "loss": 0.0877,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -7.625,
-      "rewards/margins": 7.375,
-      "rewards/rejected": -15.0,
+      "logits/chosen": -3.34375,
+      "logits/rejected": -2.984375,
+      "logps/chosen": -1112.0,
+      "logps/rejected": -2160.0,
+      "loss": 0.0942,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.9375,
+      "rewards/margins": 7.5625,
+      "rewards/rejected": -14.5,
       "step": 1400
     },
     {
       "epoch": 1.4659685863874345,
-      "eval_logits/chosen": -3.296875,
-      "eval_logits/rejected": -1.3359375,
-      "eval_logps/chosen": -1216.0,
+      "eval_logits/chosen": -1.453125,
+      "eval_logits/rejected": 0.06982421875,
+      "eval_logps/chosen": -1200.0,
       "eval_logps/rejected": -1320.0,
-      "eval_loss": 1.1249375343322754,
-      "eval_rewards/accuracies": 0.625,
-      "eval_rewards/chosen": -9.0,
-      "eval_rewards/margins": 1.34375,
-      "eval_rewards/rejected": -10.375,
-      "eval_runtime": 47.0834,
-      "eval_samples_per_second": 42.478,
-      "eval_steps_per_second": 0.68,
+      "eval_loss": 1.1682031154632568,
+      "eval_rewards/accuracies": 0.607421875,
+      "eval_rewards/chosen": -8.8125,
+      "eval_rewards/margins": 1.5,
+      "eval_rewards/rejected": -10.3125,
+      "eval_runtime": 46.5526,
+      "eval_samples_per_second": 42.962,
+      "eval_steps_per_second": 0.687,
       "step": 1400
     },
     {
       "epoch": 1.4764397905759163,
-      "grad_norm": 2.1170140180674855,
+      "grad_norm": 2.095742513327814,
       "learning_rate": 3.002283623388332e-05,
-      "logits/chosen": -4.375,
-      "logits/rejected": -2.828125,
-      "logps/chosen": -1144.0,
-      "logps/rejected": -1928.0,
-      "loss": 0.1084,
-      "rewards/accuracies": 0.918749988079071,
-      "rewards/chosen": -7.8125,
-      "rewards/margins": 5.96875,
-      "rewards/rejected": -13.75,
+      "logits/chosen": -3.046875,
+      "logits/rejected": -2.171875,
+      "logps/chosen": -1040.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.0906,
+      "rewards/accuracies": 0.9437500238418579,
+      "rewards/chosen": -6.71875,
+      "rewards/margins": 6.21875,
+      "rewards/rejected": -12.9375,
       "step": 1410
     },
     {
       "epoch": 1.486910994764398,
-      "grad_norm": 2.7336680921377425,
+      "grad_norm": 2.6930921329937996,
       "learning_rate": 2.9724028816280507e-05,
-      "logits/chosen": -3.96875,
-      "logits/rejected": -2.890625,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1824.0,
-      "loss": 0.1188,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -7.90625,
-      "rewards/margins": 5.09375,
-      "rewards/rejected": -13.0,
+      "logits/chosen": -4.09375,
+      "logits/rejected": -3.296875,
+      "logps/chosen": -972.0,
+      "logps/rejected": -1640.0,
+      "loss": 0.1241,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -5.9375,
+      "rewards/margins": 5.21875,
+      "rewards/rejected": -11.125,
       "step": 1420
     },
     {
       "epoch": 1.4973821989528795,
-      "grad_norm": 0.89921737457481,
+      "grad_norm": 2.0217896282418404,
       "learning_rate": 2.9424519877020524e-05,
-      "logits/chosen": -5.0625,
-      "logits/rejected": -3.6875,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -2016.0,
-      "loss": 0.1061,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -7.875,
-      "rewards/margins": 5.40625,
-      "rewards/rejected": -13.3125,
+      "logits/chosen": -5.28125,
+      "logits/rejected": -4.0625,
+      "logps/chosen": -1048.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.0899,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.21875,
+      "rewards/margins": 5.84375,
+      "rewards/rejected": -12.0625,
       "step": 1430
     },
     {
       "epoch": 1.5078534031413613,
-      "grad_norm": 1.6877993547731167,
+      "grad_norm": 2.3195076673880894,
       "learning_rate": 2.912435389339504e-05,
-      "logits/chosen": -2.578125,
-      "logits/rejected": -0.2177734375,
-      "logps/chosen": -1256.0,
-      "logps/rejected": -1976.0,
-      "loss": 0.0907,
+      "logits/chosen": -3.078125,
+      "logits/rejected": -1.3984375,
+      "logps/chosen": -1120.0,
+      "logps/rejected": -1856.0,
+      "loss": 0.0987,
       "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -8.6875,
-      "rewards/margins": 5.46875,
-      "rewards/rejected": -14.125,
+      "rewards/chosen": -7.34375,
+      "rewards/margins": 5.625,
+      "rewards/rejected": -12.9375,
       "step": 1440
     },
     {
       "epoch": 1.518324607329843,
-      "grad_norm": 2.999790836072498,
+      "grad_norm": 3.5488880891277503,
       "learning_rate": 2.8823575440267226e-05,
-      "logits/chosen": -3.578125,
-      "logits/rejected": -2.46875,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -2040.0,
-      "loss": 0.1018,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -8.3125,
-      "rewards/margins": 5.78125,
-      "rewards/rejected": -14.0625,
+      "logits/chosen": -3.765625,
+      "logits/rejected": -3.046875,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.1021,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.875,
+      "rewards/margins": 5.71875,
+      "rewards/rejected": -12.625,
       "step": 1450
     },
     {
       "epoch": 1.5287958115183247,
-      "grad_norm": 2.332152261924408,
+      "grad_norm": 2.84535174525212,
       "learning_rate": 2.8522229183452465e-05,
-      "logits/chosen": -4.15625,
-      "logits/rejected": -1.703125,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1912.0,
-      "loss": 0.1236,
+      "logits/chosen": -3.640625,
+      "logits/rejected": -2.125,
+      "logps/chosen": -980.0,
+      "logps/rejected": -1680.0,
+      "loss": 0.1339,
       "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -7.53125,
-      "rewards/margins": 6.21875,
-      "rewards/rejected": -13.75,
+      "rewards/chosen": -5.65625,
+      "rewards/margins": 5.71875,
+      "rewards/rejected": -11.375,
       "step": 1460
     },
     {
       "epoch": 1.5392670157068062,
-      "grad_norm": 2.398247558300056,
+      "grad_norm": 1.9842212993322867,
       "learning_rate": 2.8220359873085373e-05,
-      "logits/chosen": -4.21875,
-      "logits/rejected": -2.3125,
-      "logps/chosen": -1128.0,
-      "logps/rejected": -1952.0,
-      "loss": 0.0829,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -7.21875,
-      "rewards/margins": 6.125,
-      "rewards/rejected": -13.3125,
+      "logits/chosen": -3.515625,
+      "logits/rejected": -2.234375,
+      "logps/chosen": -1024.0,
+      "logps/rejected": -1784.0,
+      "loss": 0.0935,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -6.15625,
+      "rewards/margins": 5.53125,
+      "rewards/rejected": -11.6875,
       "step": 1470
     },
     {
       "epoch": 1.5497382198952878,
-      "grad_norm": 1.8227549611865144,
+      "grad_norm": 2.4477955509300693,
       "learning_rate": 2.7918012336974388e-05,
-      "logits/chosen": -3.46875,
-      "logits/rejected": -2.359375,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1936.0,
-      "loss": 0.1023,
+      "logits/chosen": -3.0625,
+      "logits/rejected": -2.40625,
+      "logps/chosen": -1064.0,
+      "logps/rejected": -1848.0,
+      "loss": 0.0911,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -7.6875,
-      "rewards/margins": 6.125,
-      "rewards/rejected": -13.8125,
+      "rewards/chosen": -6.65625,
+      "rewards/margins": 6.28125,
+      "rewards/rejected": -12.9375,
       "step": 1480
     },
     {
       "epoch": 1.5602094240837696,
-      "grad_norm": 3.5892433035479483,
+      "grad_norm": 2.233689154630685,
       "learning_rate": 2.7615231473944815e-05,
-      "logits/chosen": -2.125,
-      "logits/rejected": -0.921875,
-      "logps/chosen": -1272.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.0891,
+      "logits/chosen": -2.78125,
+      "logits/rejected": -1.609375,
+      "logps/chosen": -1080.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.1029,
       "rewards/accuracies": 0.96875,
-      "rewards/chosen": -8.875,
-      "rewards/margins": 6.21875,
-      "rewards/rejected": -15.0625,
+      "rewards/chosen": -7.0,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -13.625,
       "step": 1490
     },
     {
       "epoch": 1.5706806282722514,
-      "grad_norm": 2.7830723360928036,
+      "grad_norm": 2.270347478489844,
       "learning_rate": 2.7312062247171327e-05,
-      "logits/chosen": -3.609375,
-      "logits/rejected": -2.390625,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -2080.0,
-      "loss": 0.1044,
-      "rewards/accuracies": 0.9375,
-      "rewards/chosen": -8.4375,
-      "rewards/margins": 6.15625,
-      "rewards/rejected": -14.625,
+      "logits/chosen": -3.703125,
+      "logits/rejected": -2.6875,
+      "logps/chosen": -972.0,
+      "logps/rejected": -1856.0,
+      "loss": 0.1054,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -6.0,
+      "rewards/margins": 6.40625,
+      "rewards/rejected": -12.375,
       "step": 1500
     },
     {
       "epoch": 1.5706806282722514,
-      "eval_logits/chosen": -1.53125,
-      "eval_logits/rejected": 0.318359375,
-      "eval_logps/chosen": -1296.0,
-      "eval_logps/rejected": -1424.0,
-      "eval_loss": 1.152273416519165,
-      "eval_rewards/accuracies": 0.6484375,
-      "eval_rewards/chosen": -9.75,
-      "eval_rewards/margins": 1.6328125,
-      "eval_rewards/rejected": -11.375,
-      "eval_runtime": 47.0923,
-      "eval_samples_per_second": 42.47,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -1.671875,
+      "eval_logits/rejected": -0.023193359375,
+      "eval_logps/chosen": -1040.0,
+      "eval_logps/rejected": -1160.0,
+      "eval_loss": 1.0601874589920044,
+      "eval_rewards/accuracies": 0.634765625,
+      "eval_rewards/chosen": -7.1875,
+      "eval_rewards/margins": 1.5078125,
+      "eval_rewards/rejected": -8.6875,
+      "eval_runtime": 46.5319,
+      "eval_samples_per_second": 42.981,
+      "eval_steps_per_second": 0.688,
       "step": 1500
     },
     {
       "epoch": 1.581151832460733,
-      "grad_norm": 0.542876855072402,
+      "grad_norm": 0.5891137509548409,
       "learning_rate": 2.7008549677500876e-05,
-      "logits/chosen": -4.9375,
-      "logits/rejected": -3.328125,
-      "logps/chosen": -1168.0,
-      "logps/rejected": -1888.0,
-      "loss": 0.0931,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -7.5,
-      "rewards/margins": 5.5625,
-      "rewards/rejected": -13.0625,
+      "logits/chosen": -4.625,
+      "logits/rejected": -3.359375,
+      "logps/chosen": -948.0,
+      "logps/rejected": -1680.0,
+      "loss": 0.0909,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -5.28125,
+      "rewards/margins": 5.6875,
+      "rewards/rejected": -10.9375,
       "step": 1510
     },
     {
       "epoch": 1.5916230366492146,
-      "grad_norm": 1.021372172518936,
+      "grad_norm": 1.7311869495427514,
       "learning_rate": 2.670473883676709e-05,
-      "logits/chosen": -3.421875,
-      "logits/rejected": -2.265625,
-      "logps/chosen": -1224.0,
-      "logps/rejected": -2160.0,
-      "loss": 0.0753,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -8.125,
-      "rewards/margins": 6.25,
-      "rewards/rejected": -14.375,
+      "logits/chosen": -3.4375,
+      "logits/rejected": -2.640625,
+      "logps/chosen": -984.0,
+      "logps/rejected": -1880.0,
+      "loss": 0.0872,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -5.6875,
+      "rewards/margins": 5.875,
+      "rewards/rejected": -11.5625,
       "step": 1520
     },
     {
       "epoch": 1.6020942408376964,
-      "grad_norm": 2.0816524921421324,
+      "grad_norm": 2.5972575684764747,
       "learning_rate": 2.6400674841097074e-05,
-      "logits/chosen": -4.3125,
-      "logits/rejected": -3.0,
-      "logps/chosen": -1360.0,
-      "logps/rejected": -2096.0,
-      "loss": 0.0919,
+      "logits/chosen": -3.78125,
+      "logits/rejected": -2.84375,
+      "logps/chosen": -1104.0,
+      "logps/rejected": -1808.0,
+      "loss": 0.08,
       "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -9.5,
-      "rewards/margins": 5.9375,
-      "rewards/rejected": -15.4375,
+      "rewards/chosen": -6.9375,
+      "rewards/margins": 5.65625,
+      "rewards/rejected": -12.5625,
       "step": 1530
     },
     {
       "epoch": 1.6125654450261782,
-      "grad_norm": 1.925179478757633,
+      "grad_norm": 1.6809176838986462,
       "learning_rate": 2.609640284421158e-05,
-      "logits/chosen": -5.25,
-      "logits/rejected": -3.5625,
-      "logps/chosen": -1256.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.1042,
+      "logits/chosen": -3.90625,
+      "logits/rejected": -2.8125,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1792.0,
+      "loss": 0.1161,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.5625,
-      "rewards/margins": 6.46875,
-      "rewards/rejected": -15.0625,
+      "rewards/chosen": -6.625,
+      "rewards/margins": 5.78125,
+      "rewards/rejected": -12.375,
       "step": 1540
     },
     {
       "epoch": 1.6230366492146597,
-      "grad_norm": 1.3879869247573788,
+      "grad_norm": 1.4294964111769837,
       "learning_rate": 2.5791968030719726e-05,
-      "logits/chosen": -6.875,
-      "logits/rejected": -4.65625,
-      "logps/chosen": -1184.0,
-      "logps/rejected": -2016.0,
-      "loss": 0.088,
+      "logits/chosen": -4.65625,
+      "logits/rejected": -3.296875,
+      "logps/chosen": -1032.0,
+      "logps/rejected": -1816.0,
+      "loss": 0.0969,
       "rewards/accuracies": 0.96875,
-      "rewards/chosen": -7.4375,
-      "rewards/margins": 6.375,
-      "rewards/rejected": -13.8125,
+      "rewards/chosen": -5.875,
+      "rewards/margins": 5.9375,
+      "rewards/rejected": -11.8125,
       "step": 1550
     },
     {
       "epoch": 1.6335078534031413,
-      "grad_norm": 1.9339185977439384,
+      "grad_norm": 1.7010489035697924,
       "learning_rate": 2.548741560940902e-05,
-      "logits/chosen": -6.71875,
-      "logits/rejected": -4.9375,
-      "logps/chosen": -1112.0,
-      "logps/rejected": -2016.0,
-      "loss": 0.0984,
+      "logits/chosen": -4.6875,
+      "logits/rejected": -3.578125,
+      "logps/chosen": -1040.0,
+      "logps/rejected": -1936.0,
+      "loss": 0.0954,
       "rewards/accuracies": 0.96875,
-      "rewards/chosen": -7.125,
-      "rewards/margins": 6.53125,
-      "rewards/rejected": -13.625,
+      "rewards/chosen": -6.4375,
+      "rewards/margins": 6.4375,
+      "rewards/rejected": -12.875,
       "step": 1560
     },
     {
       "epoch": 1.6439790575916229,
-      "grad_norm": 2.7758278606408666,
+      "grad_norm": 2.796625630980016,
       "learning_rate": 2.5182790806531787e-05,
-      "logits/chosen": -6.0,
-      "logits/rejected": -4.75,
-      "logps/chosen": -1176.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.113,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -7.5625,
-      "rewards/margins": 6.71875,
-      "rewards/rejected": -14.3125,
+      "logits/chosen": -3.203125,
+      "logits/rejected": -2.515625,
+      "logps/chosen": -1224.0,
+      "logps/rejected": -2192.0,
+      "loss": 0.1129,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -8.0,
+      "rewards/margins": 7.71875,
+      "rewards/rejected": -15.6875,
       "step": 1570
     },
     {
       "epoch": 1.6544502617801047,
-      "grad_norm": 2.015381760096074,
+      "grad_norm": 1.8903436601975896,
       "learning_rate": 2.4878138859089074e-05,
-      "logits/chosen": -6.1875,
-      "logits/rejected": -4.28125,
-      "logps/chosen": -1160.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.0796,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -7.59375,
-      "rewards/margins": 6.96875,
-      "rewards/rejected": -14.5625,
+      "logits/chosen": -3.8125,
+      "logits/rejected": -2.640625,
+      "logps/chosen": -1112.0,
+      "logps/rejected": -2040.0,
+      "loss": 0.1034,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -7.125,
+      "rewards/margins": 7.3125,
+      "rewards/rejected": -14.4375,
       "step": 1580
     },
     {
       "epoch": 1.6649214659685865,
-      "grad_norm": 1.1340180389635373,
+      "grad_norm": 1.1451627690250832,
       "learning_rate": 2.4573505008112924e-05,
-      "logits/chosen": -5.6875,
-      "logits/rejected": -3.859375,
-      "logps/chosen": -1240.0,
-      "logps/rejected": -1968.0,
-      "loss": 0.066,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -8.375,
-      "rewards/margins": 6.1875,
-      "rewards/rejected": -14.5,
+      "logits/chosen": -4.625,
+      "logits/rejected": -3.078125,
+      "logps/chosen": -1012.0,
+      "logps/rejected": -1696.0,
+      "loss": 0.0738,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -6.125,
+      "rewards/margins": 5.65625,
+      "rewards/rejected": -11.75,
       "step": 1590
     },
     {
       "epoch": 1.675392670157068,
-      "grad_norm": 1.4426622442082426,
+      "grad_norm": 1.5315042663974998,
       "learning_rate": 2.4268934491948032e-05,
-      "logits/chosen": -5.0625,
-      "logits/rejected": -3.875,
-      "logps/chosen": -1336.0,
-      "logps/rejected": -2064.0,
-      "loss": 0.0798,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -9.25,
-      "rewards/margins": 5.9375,
-      "rewards/rejected": -15.1875,
+      "logits/chosen": -3.5625,
+      "logits/rejected": -2.84375,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1768.0,
+      "loss": 0.0799,
+      "rewards/accuracies": 0.9437500238418579,
+      "rewards/chosen": -6.65625,
+      "rewards/margins": 5.5625,
+      "rewards/rejected": -12.1875,
       "step": 1600
     },
     {
       "epoch": 1.675392670157068,
-      "eval_logits/chosen": -2.234375,
-      "eval_logits/rejected": -0.361328125,
-      "eval_logps/chosen": -1488.0,
-      "eval_logps/rejected": -1624.0,
-      "eval_loss": 1.3625078201293945,
-      "eval_rewards/accuracies": 0.6171875,
-      "eval_rewards/chosen": -11.6875,
-      "eval_rewards/margins": 1.65625,
-      "eval_rewards/rejected": -13.375,
-      "eval_runtime": 47.0885,
-      "eval_samples_per_second": 42.473,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -1.0390625,
+      "eval_logits/rejected": 0.6484375,
+      "eval_logps/chosen": -1160.0,
+      "eval_logps/rejected": -1272.0,
+      "eval_loss": 1.1262344121932983,
+      "eval_rewards/accuracies": 0.638671875,
+      "eval_rewards/chosen": -8.4375,
+      "eval_rewards/margins": 1.4453125,
+      "eval_rewards/rejected": -9.875,
+      "eval_runtime": 46.5496,
+      "eval_samples_per_second": 42.965,
+      "eval_steps_per_second": 0.687,
       "step": 1600
     },
     {
       "epoch": 1.6858638743455496,
-      "grad_norm": 1.0310987175899597,
+      "grad_norm": 0.5995121490180322,
       "learning_rate": 2.396447253953385e-05,
-      "logits/chosen": -5.75,
-      "logits/rejected": -4.15625,
-      "logps/chosen": -1352.0,
-      "logps/rejected": -2240.0,
-      "loss": 0.0718,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -9.125,
-      "rewards/margins": 7.3125,
-      "rewards/rejected": -16.5,
+      "logits/chosen": -4.5,
+      "logits/rejected": -3.40625,
+      "logps/chosen": -1088.0,
+      "logps/rejected": -1912.0,
+      "loss": 0.0782,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.5625,
+      "rewards/margins": 6.59375,
+      "rewards/rejected": -13.1875,
       "step": 1610
     },
     {
       "epoch": 1.6963350785340314,
-      "grad_norm": 1.5346354875413641,
+      "grad_norm": 1.8823490209167353,
       "learning_rate": 2.3660164363687998e-05,
-      "logits/chosen": -6.09375,
-      "logits/rejected": -4.34375,
-      "logps/chosen": -1304.0,
-      "logps/rejected": -2096.0,
-      "loss": 0.0764,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.5625,
-      "rewards/margins": 6.3125,
-      "rewards/rejected": -14.8125,
+      "logits/chosen": -4.65625,
+      "logits/rejected": -3.109375,
+      "logps/chosen": -1104.0,
+      "logps/rejected": -1912.0,
+      "loss": 0.0829,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -6.53125,
+      "rewards/margins": 6.5,
+      "rewards/rejected": -13.0,
       "step": 1620
     },
     {
       "epoch": 1.7068062827225132,
-      "grad_norm": 2.616419576418859,
+      "grad_norm": 2.639598729920059,
       "learning_rate": 2.33560551543922e-05,
-      "logits/chosen": -6.09375,
-      "logits/rejected": -3.953125,
-      "logps/chosen": -1208.0,
-      "logps/rejected": -1992.0,
-      "loss": 0.073,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -7.78125,
-      "rewards/margins": 6.09375,
-      "rewards/rejected": -13.875,
+      "logits/chosen": -4.34375,
+      "logits/rejected": -3.09375,
+      "logps/chosen": -1040.0,
+      "logps/rejected": -1832.0,
+      "loss": 0.0873,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.09375,
+      "rewards/margins": 6.15625,
+      "rewards/rejected": -12.25,
       "step": 1630
     },
     {
       "epoch": 1.7172774869109948,
-      "grad_norm": 1.745924264164314,
+      "grad_norm": 3.071479027546342,
       "learning_rate": 2.3052190072081493e-05,
-      "logits/chosen": -5.78125,
-      "logits/rejected": -3.21875,
-      "logps/chosen": -1224.0,
-      "logps/rejected": -1992.0,
-      "loss": 0.0798,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.1875,
-      "rewards/margins": 6.40625,
-      "rewards/rejected": -14.5625,
+      "logits/chosen": -4.71875,
+      "logits/rejected": -3.03125,
+      "logps/chosen": -1040.0,
+      "logps/rejected": -1792.0,
+      "loss": 0.1025,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.28125,
+      "rewards/margins": 6.21875,
+      "rewards/rejected": -12.5,
       "step": 1640
     },
     {
       "epoch": 1.7277486910994764,
-      "grad_norm": 1.0855532344640153,
+      "grad_norm": 1.9828584967336726,
       "learning_rate": 2.2748614240937864e-05,
-      "logits/chosen": -4.59375,
-      "logits/rejected": -3.21875,
-      "logps/chosen": -1240.0,
-      "logps/rejected": -2208.0,
-      "loss": 0.0601,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.5,
-      "rewards/margins": 7.3125,
-      "rewards/rejected": -15.8125,
+      "logits/chosen": -4.34375,
+      "logits/rejected": -3.3125,
+      "logps/chosen": -976.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.0705,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -5.90625,
+      "rewards/margins": 6.875,
+      "rewards/rejected": -12.75,
       "step": 1650
     },
     {
       "epoch": 1.738219895287958,
-      "grad_norm": 2.2131216719549776,
+      "grad_norm": 1.26216299328458,
       "learning_rate": 2.2445372742189334e-05,
-      "logits/chosen": -4.21875,
-      "logits/rejected": -2.203125,
-      "logps/chosen": -1352.0,
-      "logps/rejected": -2336.0,
-      "loss": 0.1005,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -9.3125,
-      "rewards/margins": 7.5625,
-      "rewards/rejected": -16.875,
+      "logits/chosen": -3.484375,
+      "logits/rejected": -2.234375,
+      "logps/chosen": -1128.0,
+      "logps/rejected": -2032.0,
+      "loss": 0.0985,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -7.125,
+      "rewards/margins": 6.84375,
+      "rewards/rejected": -13.9375,
       "step": 1660
     },
     {
       "epoch": 1.7486910994764397,
-      "grad_norm": 1.532888727305927,
+      "grad_norm": 1.1026147510775932,
       "learning_rate": 2.214251060741528e-05,
-      "logits/chosen": -5.1875,
-      "logits/rejected": -3.28125,
-      "logps/chosen": -1296.0,
-      "logps/rejected": -2176.0,
-      "loss": 0.0749,
+      "logits/chosen": -3.78125,
+      "logits/rejected": -2.859375,
+      "logps/chosen": -1136.0,
+      "logps/rejected": -1992.0,
+      "loss": 0.0719,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -8.6875,
-      "rewards/margins": 6.78125,
-      "rewards/rejected": -15.4375,
+      "rewards/chosen": -7.09375,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -13.6875,
       "step": 1670
     },
     {
       "epoch": 1.7591623036649215,
-      "grad_norm": 2.7468523088707806,
+      "grad_norm": 2.6311725948300753,
       "learning_rate": 2.1840072811859302e-05,
-      "logits/chosen": -5.0625,
-      "logits/rejected": -3.0625,
-      "logps/chosen": -1296.0,
-      "logps/rejected": -2160.0,
-      "loss": 0.0872,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.5,
-      "rewards/margins": 6.8125,
-      "rewards/rejected": -15.3125,
+      "logits/chosen": -3.5625,
+      "logits/rejected": -2.5625,
+      "logps/chosen": -1160.0,
+      "logps/rejected": -2016.0,
+      "loss": 0.0828,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -7.21875,
+      "rewards/margins": 6.75,
+      "rewards/rejected": -13.9375,
       "step": 1680
     },
     {
       "epoch": 1.7696335078534031,
-      "grad_norm": 1.88322783183091,
+      "grad_norm": 1.711380337952449,
       "learning_rate": 2.1538104267750287e-05,
-      "logits/chosen": -4.0625,
-      "logits/rejected": -2.46875,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -1992.0,
-      "loss": 0.1035,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.4375,
-      "rewards/margins": 5.90625,
-      "rewards/rejected": -14.3125,
+      "logits/chosen": -3.234375,
+      "logits/rejected": -2.140625,
+      "logps/chosen": -1064.0,
+      "logps/rejected": -1816.0,
+      "loss": 0.0971,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.75,
+      "rewards/margins": 5.84375,
+      "rewards/rejected": -12.5625,
       "step": 1690
     },
     {
       "epoch": 1.7801047120418847,
-      "grad_norm": 1.6324407923143298,
+      "grad_norm": 1.6188001080199679,
       "learning_rate": 2.1236649817632952e-05,
-      "logits/chosen": -3.9375,
-      "logits/rejected": -3.0625,
-      "logps/chosen": -1272.0,
-      "logps/rejected": -2032.0,
-      "loss": 0.0847,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -9.125,
-      "rewards/margins": 5.6875,
-      "rewards/rejected": -14.8125,
+      "logits/chosen": -3.96875,
+      "logits/rejected": -3.34375,
+      "logps/chosen": -1020.0,
+      "logps/rejected": -1792.0,
+      "loss": 0.0894,
+      "rewards/accuracies": 0.9437500238418579,
+      "rewards/chosen": -6.5625,
+      "rewards/margins": 5.8125,
+      "rewards/rejected": -12.375,
       "step": 1700
     },
     {
       "epoch": 1.7801047120418847,
-      "eval_logits/chosen": -1.09375,
-      "eval_logits/rejected": 0.578125,
-      "eval_logps/chosen": -1544.0,
-      "eval_logps/rejected": -1656.0,
-      "eval_loss": 1.307437539100647,
-      "eval_rewards/accuracies": 0.62109375,
-      "eval_rewards/chosen": -12.25,
-      "eval_rewards/margins": 1.4375,
-      "eval_rewards/rejected": -13.6875,
-      "eval_runtime": 47.1105,
-      "eval_samples_per_second": 42.453,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": -2.125,
+      "eval_logits/rejected": -0.609375,
+      "eval_logps/chosen": -1184.0,
+      "eval_logps/rejected": -1304.0,
+      "eval_loss": 1.1620312929153442,
+      "eval_rewards/accuracies": 0.63671875,
+      "eval_rewards/chosen": -8.6875,
+      "eval_rewards/margins": 1.5234375,
+      "eval_rewards/rejected": -10.1875,
+      "eval_runtime": 46.5398,
+      "eval_samples_per_second": 42.974,
+      "eval_steps_per_second": 0.688,
       "step": 1700
     },
     {
       "epoch": 1.7905759162303665,
-      "grad_norm": 3.4698621331944133,
+      "grad_norm": 3.85850013777164,
       "learning_rate": 2.093575422770872e-05,
-      "logits/chosen": -2.5,
-      "logits/rejected": -1.546875,
-      "logps/chosen": -1408.0,
-      "logps/rejected": -2192.0,
-      "loss": 0.08,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -10.3125,
-      "rewards/margins": 6.34375,
-      "rewards/rejected": -16.625,
+      "logits/chosen": -3.359375,
+      "logits/rejected": -2.4375,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1840.0,
+      "loss": 0.0874,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.90625,
+      "rewards/margins": 6.15625,
+      "rewards/rejected": -13.0625,
       "step": 1710
     },
     {
       "epoch": 1.8010471204188483,
-      "grad_norm": 3.244021486396508,
+      "grad_norm": 3.906161983369402,
       "learning_rate": 2.0635462181187827e-05,
-      "logits/chosen": -3.4375,
-      "logits/rejected": -2.5625,
-      "logps/chosen": -1496.0,
-      "logps/rejected": -2432.0,
-      "loss": 0.0912,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -10.9375,
-      "rewards/margins": 6.96875,
-      "rewards/rejected": -17.875,
+      "logits/chosen": -4.0,
+      "logits/rejected": -3.125,
+      "logps/chosen": -1264.0,
+      "logps/rejected": -2208.0,
+      "loss": 0.0863,
+      "rewards/accuracies": 0.956250011920929,
+      "rewards/chosen": -8.5625,
+      "rewards/margins": 7.0625,
+      "rewards/rejected": -15.625,
       "step": 1720
     },
     {
       "epoch": 1.8115183246073299,
-      "grad_norm": 2.602033619168437,
+      "grad_norm": 2.0656072764590174,
       "learning_rate": 2.0335818271653917e-05,
-      "logits/chosen": -4.15625,
-      "logits/rejected": -3.015625,
-      "logps/chosen": -1312.0,
-      "logps/rejected": -2160.0,
-      "loss": 0.0834,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -9.125,
-      "rewards/margins": 6.34375,
-      "rewards/rejected": -15.4375,
+      "logits/chosen": -3.953125,
+      "logits/rejected": -3.1875,
+      "logps/chosen": -1184.0,
+      "logps/rejected": -2064.0,
+      "loss": 0.0777,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -7.78125,
+      "rewards/margins": 6.625,
+      "rewards/rejected": -14.4375,
       "step": 1730
     },
     {
       "epoch": 1.8219895287958114,
-      "grad_norm": 2.7601551424860826,
+      "grad_norm": 1.146435680405151,
       "learning_rate": 2.0036866996441815e-05,
-      "logits/chosen": -4.40625,
-      "logits/rejected": -2.375,
-      "logps/chosen": -1272.0,
-      "logps/rejected": -2096.0,
-      "loss": 0.0875,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.75,
-      "rewards/margins": 6.375,
-      "rewards/rejected": -15.125,
+      "logits/chosen": -3.421875,
+      "logits/rejected": -2.25,
+      "logps/chosen": -1224.0,
+      "logps/rejected": -2064.0,
+      "loss": 0.0836,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -8.25,
+      "rewards/margins": 6.6875,
+      "rewards/rejected": -14.9375,
       "step": 1740
     },
     {
       "epoch": 1.8324607329842932,
-      "grad_norm": 3.55242376181993,
+      "grad_norm": 2.3314660026790435,
       "learning_rate": 1.9738652750029677e-05,
-      "logits/chosen": -3.453125,
-      "logits/rejected": -2.296875,
-      "logps/chosen": -1352.0,
-      "logps/rejected": -2224.0,
-      "loss": 0.0833,
+      "logits/chosen": -3.53125,
+      "logits/rejected": -2.59375,
+      "logps/chosen": -1232.0,
+      "logps/rejected": -2112.0,
+      "loss": 0.079,
       "rewards/accuracies": 0.96875,
-      "rewards/chosen": -9.25,
-      "rewards/margins": 7.0,
-      "rewards/rejected": -16.25,
+      "rewards/chosen": -8.0625,
+      "rewards/margins": 7.125,
+      "rewards/rejected": -15.1875,
       "step": 1750
     },
     {
       "epoch": 1.8429319371727748,
-      "grad_norm": 2.4991335839252526,
+      "grad_norm": 2.8042308888456247,
       "learning_rate": 1.944121981744631e-05,
-      "logits/chosen": -5.21875,
-      "logits/rejected": -3.484375,
-      "logps/chosen": -1352.0,
-      "logps/rejected": -2240.0,
-      "loss": 0.0849,
-      "rewards/accuracies": 0.949999988079071,
-      "rewards/chosen": -9.0,
-      "rewards/margins": 6.5625,
-      "rewards/rejected": -15.625,
+      "logits/chosen": -5.0,
+      "logits/rejected": -3.453125,
+      "logps/chosen": -1184.0,
+      "logps/rejected": -2128.0,
+      "loss": 0.0746,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -7.3125,
+      "rewards/margins": 7.15625,
+      "rewards/rejected": -14.4375,
       "step": 1760
     },
-    {
-      "epoch": 1.8534031413612566,
-      "grad_norm": 1.0867713650815005,
-      "learning_rate": 1.9144612367694887e-05,
-      "logits/chosen": -5.125,
-      "logits/rejected": -3.3125,
-      "logps/chosen": -1240.0,
-      "logps/rejected": -2008.0,
-      "loss": 0.0667,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.375,
-      "rewards/margins": 6.3125,
-      "rewards/rejected": -14.6875,
+    {
+      "epoch": 1.8534031413612566,
+      "grad_norm": 1.1244485421795494,
+      "learning_rate": 1.9144612367694887e-05,
+      "logits/chosen": -4.0,
+      "logits/rejected": -2.640625,
+      "logps/chosen": -1096.0,
+      "logps/rejected": -1880.0,
+      "loss": 0.0729,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -6.96875,
+      "rewards/margins": 6.5,
+      "rewards/rejected": -13.5,
       "step": 1770
     },
     {
       "epoch": 1.8638743455497382,
-      "grad_norm": 1.5280300607044794,
+      "grad_norm": 2.151245483052861,
       "learning_rate": 1.8848874447193803e-05,
-      "logits/chosen": -4.1875,
-      "logits/rejected": -2.4375,
-      "logps/chosen": -1304.0,
-      "logps/rejected": -2128.0,
-      "loss": 0.0814,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -9.125,
+      "logits/chosen": -3.15625,
+      "logits/rejected": -1.84375,
+      "logps/chosen": -1168.0,
+      "logps/rejected": -1992.0,
+      "loss": 0.1011,
+      "rewards/accuracies": 0.9437500238418579,
+      "rewards/chosen": -7.78125,
       "rewards/margins": 6.90625,
-      "rewards/rejected": -16.0,
+      "rewards/rejected": -14.6875,
       "step": 1780
     },
     {
       "epoch": 1.8743455497382198,
-      "grad_norm": 2.4360778138595505,
+      "grad_norm": 3.14257094250862,
       "learning_rate": 1.855404997323571e-05,
-      "logits/chosen": -5.125,
-      "logits/rejected": -3.671875,
-      "logps/chosen": -1336.0,
-      "logps/rejected": -2176.0,
-      "loss": 0.0884,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -9.25,
-      "rewards/margins": 6.96875,
-      "rewards/rejected": -16.25,
+      "logits/chosen": -4.125,
+      "logits/rejected": -3.21875,
+      "logps/chosen": -1120.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.0929,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -7.0625,
+      "rewards/margins": 6.3125,
+      "rewards/rejected": -13.375,
       "step": 1790
     },
     {
       "epoch": 1.8848167539267016,
-      "grad_norm": 1.3496661620645678,
+      "grad_norm": 1.3127759553165996,
       "learning_rate": 1.82601827274658e-05,
-      "logits/chosen": -5.5625,
-      "logits/rejected": -3.703125,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -1952.0,
-      "loss": 0.1018,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -8.375,
-      "rewards/margins": 6.0,
-      "rewards/rejected": -14.375,
+      "logits/chosen": -4.34375,
+      "logits/rejected": -3.125,
+      "logps/chosen": -988.0,
+      "logps/rejected": -1688.0,
+      "loss": 0.1011,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.09375,
+      "rewards/margins": 5.6875,
+      "rewards/rejected": -11.75,
       "step": 1800
     },
     {
       "epoch": 1.8848167539267016,
-      "eval_logits/chosen": -3.3125,
-      "eval_logits/rejected": -1.421875,
-      "eval_logps/chosen": -1264.0,
-      "eval_logps/rejected": -1376.0,
-      "eval_loss": 1.1159921884536743,
-      "eval_rewards/accuracies": 0.638671875,
-      "eval_rewards/chosen": -9.4375,
-      "eval_rewards/margins": 1.421875,
-      "eval_rewards/rejected": -10.8125,
-      "eval_runtime": 47.1075,
-      "eval_samples_per_second": 42.456,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": -1.7890625,
+      "eval_logits/rejected": -0.25390625,
+      "eval_logps/chosen": -1096.0,
+      "eval_logps/rejected": -1208.0,
+      "eval_loss": 1.0581562519073486,
+      "eval_rewards/accuracies": 0.64453125,
+      "eval_rewards/chosen": -7.78125,
+      "eval_rewards/margins": 1.3984375,
+      "eval_rewards/rejected": -9.1875,
+      "eval_runtime": 46.5487,
+      "eval_samples_per_second": 42.966,
+      "eval_steps_per_second": 0.687,
       "step": 1800
     },
     {
       "epoch": 1.8952879581151834,
-      "grad_norm": 2.506337636253115,
+      "grad_norm": 2.3449118736046883,
       "learning_rate": 1.7967316349380213e-05,
-      "logits/chosen": -5.03125,
-      "logits/rejected": -3.71875,
-      "logps/chosen": -1176.0,
-      "logps/rejected": -1936.0,
-      "loss": 0.0892,
+      "logits/chosen": -3.0625,
+      "logits/rejected": -2.375,
+      "logps/chosen": -1032.0,
+      "logps/rejected": -1752.0,
+      "loss": 0.0975,
       "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.0625,
-      "rewards/margins": 6.09375,
-      "rewards/rejected": -14.125,
+      "rewards/chosen": -6.625,
+      "rewards/margins": 5.71875,
+      "rewards/rejected": -12.3125,
       "step": 1810
     },
     {
       "epoch": 1.905759162303665,
-      "grad_norm": 1.9237111956589796,
+      "grad_norm": 2.3483923360148626,
       "learning_rate": 1.7675494329845516e-05,
-      "logits/chosen": -4.75,
-      "logits/rejected": -3.171875,
-      "logps/chosen": -1312.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.0655,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -9.1875,
-      "rewards/margins": 6.21875,
-      "rewards/rejected": -15.4375,
+      "logits/chosen": -2.65625,
+      "logits/rejected": -1.6953125,
+      "logps/chosen": -1152.0,
+      "logps/rejected": -1928.0,
+      "loss": 0.0795,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -7.65625,
+      "rewards/margins": 5.90625,
+      "rewards/rejected": -13.5625,
       "step": 1820
     },
     {
       "epoch": 1.9162303664921465,
-      "grad_norm": 0.582904913258408,
+      "grad_norm": 1.3045307413619847,
       "learning_rate": 1.738476000464026e-05,
-      "logits/chosen": -5.25,
-      "logits/rejected": -3.5,
-      "logps/chosen": -1248.0,
-      "logps/rejected": -2064.0,
-      "loss": 0.0875,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -8.5625,
-      "rewards/margins": 6.5625,
-      "rewards/rejected": -15.125,
+      "logits/chosen": -3.015625,
+      "logits/rejected": -1.859375,
+      "logps/chosen": -1072.0,
+      "logps/rejected": -1904.0,
+      "loss": 0.082,
+      "rewards/accuracies": 0.9624999761581421,
+      "rewards/chosen": -6.84375,
+      "rewards/margins": 6.71875,
+      "rewards/rejected": -13.5625,
       "step": 1830
     },
     {
       "epoch": 1.9267015706806283,
-      "grad_norm": 3.7238681440342476,
+      "grad_norm": 1.4428388312263158,
       "learning_rate": 1.7095156548019646e-05,
-      "logits/chosen": -6.75,
-      "logits/rejected": -4.8125,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.0851,
-      "rewards/accuracies": 0.9624999761581421,
-      "rewards/chosen": -8.5,
-      "rewards/margins": 6.8125,
-      "rewards/rejected": -15.3125,
+      "logits/chosen": -4.40625,
+      "logits/rejected": -3.1875,
+      "logps/chosen": -1096.0,
+      "logps/rejected": -1984.0,
+      "loss": 0.0808,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -7.125,
+      "rewards/margins": 6.84375,
+      "rewards/rejected": -14.0,
       "step": 1840
     },
     {
       "epoch": 1.93717277486911,
-      "grad_norm": 3.2782252173379,
+      "grad_norm": 3.3471269640085137,
       "learning_rate": 1.6806726966304064e-05,
-      "logits/chosen": -6.125,
-      "logits/rejected": -4.25,
-      "logps/chosen": -1344.0,
-      "logps/rejected": -2112.0,
-      "loss": 0.1105,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -9.1875,
-      "rewards/margins": 6.03125,
-      "rewards/rejected": -15.1875,
+      "logits/chosen": -4.4375,
+      "logits/rejected": -2.9375,
+      "logps/chosen": -1184.0,
+      "logps/rejected": -1984.0,
+      "loss": 0.1092,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -7.625,
+      "rewards/margins": 6.34375,
+      "rewards/rejected": -13.9375,
       "step": 1850
     },
     {
       "epoch": 1.9476439790575917,
-      "grad_norm": 1.6699418962967145,
+      "grad_norm": 1.4797943824903212,
       "learning_rate": 1.6519514091492625e-05,
-      "logits/chosen": -5.40625,
-      "logits/rejected": -4.03125,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -2040.0,
-      "loss": 0.0809,
-      "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.6875,
-      "rewards/margins": 6.53125,
-      "rewards/rejected": -15.25,
+      "logits/chosen": -4.09375,
+      "logits/rejected": -3.09375,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.0901,
+      "rewards/accuracies": 0.949999988079071,
+      "rewards/chosen": -6.96875,
+      "rewards/margins": 6.875,
+      "rewards/rejected": -13.8125,
       "step": 1860
     },
     {
       "epoch": 1.9581151832460733,
-      "grad_norm": 1.3679248396395456,
+      "grad_norm": 2.185346931882519,
       "learning_rate": 1.6233560574902627e-05,
-      "logits/chosen": -5.21875,
-      "logits/rejected": -3.453125,
-      "logps/chosen": -1200.0,
-      "logps/rejected": -2080.0,
-      "loss": 0.0936,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -8.4375,
-      "rewards/margins": 6.4375,
-      "rewards/rejected": -14.875,
+      "logits/chosen": -3.203125,
+      "logits/rejected": -2.21875,
+      "logps/chosen": -1056.0,
+      "logps/rejected": -1936.0,
+      "loss": 0.1104,
+      "rewards/accuracies": 0.96875,
+      "rewards/chosen": -7.03125,
+      "rewards/margins": 6.5,
+      "rewards/rejected": -13.5625,
       "step": 1870
     },
     {
       "epoch": 1.9685863874345548,
-      "grad_norm": 2.4854442522097564,
+      "grad_norm": 2.1549552693416674,
       "learning_rate": 1.5948908880835753e-05,
-      "logits/chosen": -5.625,
-      "logits/rejected": -3.3125,
-      "logps/chosen": -1272.0,
-      "logps/rejected": -2048.0,
-      "loss": 0.081,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -8.8125,
-      "rewards/margins": 6.4375,
-      "rewards/rejected": -15.25,
+      "logits/chosen": -3.125,
+      "logits/rejected": -1.8046875,
+      "logps/chosen": -1096.0,
+      "logps/rejected": -1888.0,
+      "loss": 0.0977,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -7.09375,
+      "rewards/margins": 6.5,
+      "rewards/rejected": -13.5625,
       "step": 1880
     },
     {
       "epoch": 1.9790575916230366,
-      "grad_norm": 1.01399499181541,
+      "grad_norm": 1.2083693717335389,
       "learning_rate": 1.5665601280272124e-05,
-      "logits/chosen": -6.15625,
-      "logits/rejected": -4.5,
-      "logps/chosen": -1216.0,
-      "logps/rejected": -2064.0,
-      "loss": 0.0831,
-      "rewards/accuracies": 0.9437500238418579,
-      "rewards/chosen": -8.3125,
-      "rewards/margins": 6.28125,
-      "rewards/rejected": -14.5625,
+      "logits/chosen": -3.609375,
+      "logits/rejected": -2.828125,
+      "logps/chosen": -1080.0,
+      "logps/rejected": -1896.0,
+      "loss": 0.0842,
+      "rewards/accuracies": 0.9375,
+      "rewards/chosen": -6.90625,
+      "rewards/margins": 6.03125,
+      "rewards/rejected": -12.9375,
       "step": 1890
     },
     {
       "epoch": 1.9895287958115184,
-      "grad_norm": 2.2420767620448676,
+      "grad_norm": 1.2503137014249368,
       "learning_rate": 1.538367984459298e-05,
-      "logits/chosen": -5.71875,
-      "logits/rejected": -3.46875,
-      "logps/chosen": -1152.0,
-      "logps/rejected": -2024.0,
-      "loss": 0.0649,
+      "logits/chosen": -2.953125,
+      "logits/rejected": -1.8515625,
+      "logps/chosen": -1024.0,
+      "logps/rejected": -1872.0,
+      "loss": 0.068,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -7.90625,
-      "rewards/margins": 6.59375,
-      "rewards/rejected": -14.5,
+      "rewards/chosen": -6.59375,
+      "rewards/margins": 6.40625,
+      "rewards/rejected": -13.0,
       "step": 1900
     },
     {
       "epoch": 1.9895287958115184,
-      "eval_logits/chosen": -3.046875,
-      "eval_logits/rejected": -1.03125,
-      "eval_logps/chosen": -1392.0,
-      "eval_logps/rejected": -1536.0,
-      "eval_loss": 1.314226508140564,
-      "eval_rewards/accuracies": 0.62109375,
-      "eval_rewards/chosen": -10.75,
-      "eval_rewards/margins": 1.6796875,
-      "eval_rewards/rejected": -12.4375,
-      "eval_runtime": 47.0816,
-      "eval_samples_per_second": 42.479,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -1.0625,
+      "eval_logits/rejected": 0.5078125,
+      "eval_logps/chosen": -1248.0,
+      "eval_logps/rejected": -1368.0,
+      "eval_loss": 1.246999979019165,
+      "eval_rewards/accuracies": 0.63671875,
+      "eval_rewards/chosen": -9.25,
+      "eval_rewards/margins": 1.5625,
+      "eval_rewards/rejected": -10.8125,
+      "eval_runtime": 46.522,
+      "eval_samples_per_second": 42.99,
+      "eval_steps_per_second": 0.688,
       "step": 1900
     },
     {
       "epoch": 2.0,
-      "grad_norm": 2.039933962862527,
+      "grad_norm": 3.3680459141885217,
       "learning_rate": 1.5103186439333133e-05,
-      "logits/chosen": -4.4375,
-      "logits/rejected": -3.3125,
-      "logps/chosen": -1256.0,
-      "logps/rejected": -2224.0,
-      "loss": 0.1064,
+      "logits/chosen": -2.28125,
+      "logits/rejected": -2.109375,
+      "logps/chosen": -1120.0,
+      "logps/rejected": -2080.0,
+      "loss": 0.1041,
       "rewards/accuracies": 0.956250011920929,
-      "rewards/chosen": -8.75,
-      "rewards/margins": 7.78125,
-      "rewards/rejected": -16.5,
+      "rewards/chosen": -7.375,
+      "rewards/margins": 7.71875,
+      "rewards/rejected": -15.0625,
       "step": 1910
     },
     {
       "epoch": 2.0104712041884816,
-      "grad_norm": 0.5977599378677944,
+      "grad_norm": 0.46903067654475533,
       "learning_rate": 1.4824162717963829e-05,
-      "logits/chosen": -5.1875,
-      "logits/rejected": -2.125,
-      "logps/chosen": -1232.0,
-      "logps/rejected": -2192.0,
-      "loss": 0.0153,
+      "logits/chosen": -2.71875,
+      "logits/rejected": -0.9296875,
+      "logps/chosen": -1128.0,
+      "logps/rejected": -2064.0,
+      "loss": 0.0156,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -8.375,
-      "rewards/margins": 8.0625,
-      "rewards/rejected": -16.375,
+      "rewards/chosen": -7.34375,
+      "rewards/margins": 7.90625,
+      "rewards/rejected": -15.25,
       "step": 1920
     },
     {
       "epoch": 2.020942408376963,
-      "grad_norm": 0.36787439417115725,
+      "grad_norm": 0.14327604030943603,
       "learning_rate": 1.4546650115707192e-05,
-      "logits/chosen": -5.53125,
-      "logits/rejected": -3.484375,
-      "logps/chosen": -1240.0,
-      "logps/rejected": -2160.0,
-      "loss": 0.0127,
+      "logits/chosen": -3.609375,
+      "logits/rejected": -2.453125,
+      "logps/chosen": -1104.0,
+      "logps/rejected": -1992.0,
+      "loss": 0.0171,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -8.6875,
-      "rewards/margins": 8.0625,
-      "rewards/rejected": -16.75,
+      "rewards/chosen": -7.34375,
+      "rewards/margins": 7.8125,
+      "rewards/rejected": -15.125,
       "step": 1930
     },
     {
       "epoch": 2.031413612565445,
-      "grad_norm": 0.061516779366227535,
+      "grad_norm": 0.2957848801139986,
       "learning_rate": 1.4270689843383111e-05,
-      "logits/chosen": -3.765625,
-      "logits/rejected": -1.625,
-      "logps/chosen": -1384.0,
-      "logps/rejected": -2448.0,
-      "loss": 0.0133,
+      "logits/chosen": -2.234375,
+      "logits/rejected": -0.7578125,
+      "logps/chosen": -1176.0,
+      "logps/rejected": -2224.0,
+      "loss": 0.0161,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -9.8125,
-      "rewards/margins": 9.1875,
-      "rewards/rejected": -19.0,
+      "rewards/chosen": -7.75,
+      "rewards/margins": 8.9375,
+      "rewards/rejected": -16.625,
       "step": 1940
     },
     {
       "epoch": 2.0418848167539267,
-      "grad_norm": 0.3655557372419591,
+      "grad_norm": 0.44101228635606704,
       "learning_rate": 1.3996322881289347e-05,
-      "logits/chosen": -4.5,
-      "logits/rejected": -2.375,
-      "logps/chosen": -1376.0,
-      "logps/rejected": -2496.0,
-      "loss": 0.0212,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.0625,
+      "logits/chosen": -3.078125,
+      "logits/rejected": -1.546875,
+      "logps/chosen": -1168.0,
+      "logps/rejected": -2288.0,
+      "loss": 0.0163,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -7.96875,
       "rewards/margins": 9.375,
-      "rewards/rejected": -19.375,
+      "rewards/rejected": -17.375,
       "step": 1950
     },
     {
       "epoch": 2.0523560209424083,
-      "grad_norm": 0.3483109003084252,
+      "grad_norm": 0.07048245610445522,
       "learning_rate": 1.3723589973115963e-05,
-      "logits/chosen": -4.125,
-      "logits/rejected": -2.09375,
-      "logps/chosen": -1296.0,
-      "logps/rejected": -2576.0,
+      "logits/chosen": -2.8125,
+      "logits/rejected": -1.2890625,
+      "logps/chosen": -1136.0,
+      "logps/rejected": -2384.0,
       "loss": 0.0116,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -9.3125,
-      "rewards/margins": 10.3125,
-      "rewards/rejected": -19.625,
+      "rewards/chosen": -7.625,
+      "rewards/margins": 10.125,
+      "rewards/rejected": -17.75,
       "step": 1960
     },
     {
       "epoch": 2.06282722513089,
-      "grad_norm": 0.2220265302563937,
+      "grad_norm": 0.11498906748003677,
       "learning_rate": 1.3452531619894837e-05,
-      "logits/chosen": -3.859375,
-      "logits/rejected": -1.9765625,
-      "logps/chosen": -1416.0,
-      "logps/rejected": -2640.0,
-      "loss": 0.0088,
+      "logits/chosen": -2.546875,
+      "logits/rejected": -1.328125,
+      "logps/chosen": -1264.0,
+      "logps/rejected": -2464.0,
+      "loss": 0.01,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -10.3125,
-      "rewards/margins": 10.1875,
-      "rewards/rejected": -20.5,
+      "rewards/chosen": -8.75,
+      "rewards/margins": 9.9375,
+      "rewards/rejected": -18.75,
       "step": 1970
     },
     {
       "epoch": 2.073298429319372,
-      "grad_norm": 0.2494338602476271,
+      "grad_norm": 0.6213429156994286,
       "learning_rate": 1.3183188073985245e-05,
-      "logits/chosen": -4.125,
-      "logits/rejected": -2.3125,
-      "logps/chosen": -1424.0,
-      "logps/rejected": -2544.0,
-      "loss": 0.0184,
+      "logits/chosen": -2.84375,
+      "logits/rejected": -1.6796875,
+      "logps/chosen": -1272.0,
+      "logps/rejected": -2432.0,
+      "loss": 0.0267,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -10.4375,
-      "rewards/margins": 9.4375,
-      "rewards/rejected": -19.875,
+      "rewards/chosen": -9.0,
+      "rewards/margins": 9.75,
+      "rewards/rejected": -18.75,
       "step": 1980
     },
     {
       "epoch": 2.0837696335078535,
-      "grad_norm": 0.1354327516775415,
+      "grad_norm": 0.12557149275546892,
       "learning_rate": 1.2915599333096351e-05,
-      "logits/chosen": -4.5,
-      "logits/rejected": -2.65625,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2752.0,
-      "loss": 0.0084,
+      "logits/chosen": -3.390625,
+      "logits/rejected": -2.203125,
+      "logps/chosen": -1344.0,
+      "logps/rejected": -2512.0,
+      "loss": 0.0106,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 10.3125,
-      "rewards/rejected": -21.375,
+      "rewards/chosen": -9.3125,
+      "rewards/margins": 9.625,
+      "rewards/rejected": -19.0,
       "step": 1990
     },
     {
       "epoch": 2.094240837696335,
-      "grad_norm": 0.059306827743589294,
+      "grad_norm": 0.10901452946527279,
       "learning_rate": 1.264980513434752e-05,
-      "logits/chosen": -4.375,
-      "logits/rejected": -1.71875,
-      "logps/chosen": -1560.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.0155,
+      "logits/chosen": -2.96875,
+      "logits/rejected": -1.0546875,
+      "logps/chosen": -1344.0,
+      "logps/rejected": -2464.0,
+      "loss": 0.0203,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.25,
-      "rewards/margins": 9.8125,
-      "rewards/rejected": -21.0,
+      "rewards/chosen": -9.1875,
+      "rewards/margins": 9.25,
+      "rewards/rejected": -18.375,
       "step": 2000
     },
     {
       "epoch": 2.094240837696335,
-      "eval_logits/chosen": -1.1640625,
-      "eval_logits/rejected": 0.66015625,
-      "eval_logps/chosen": -1768.0,
-      "eval_logps/rejected": -1944.0,
-      "eval_loss": 1.7396718263626099,
-      "eval_rewards/accuracies": 0.630859375,
-      "eval_rewards/chosen": -14.4375,
-      "eval_rewards/margins": 2.078125,
-      "eval_rewards/rejected": -16.5,
-      "eval_runtime": 47.1041,
-      "eval_samples_per_second": 42.459,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": -0.134765625,
+      "eval_logits/rejected": 1.3828125,
+      "eval_logps/chosen": -1504.0,
+      "eval_logps/rejected": -1648.0,
+      "eval_loss": 1.5381640195846558,
+      "eval_rewards/accuracies": 0.62109375,
+      "eval_rewards/chosen": -11.875,
+      "eval_rewards/margins": 1.7265625,
+      "eval_rewards/rejected": -13.625,
+      "eval_runtime": 46.548,
+      "eval_samples_per_second": 42.966,
+      "eval_steps_per_second": 0.687,
       "step": 2000
     },
     {
       "epoch": 2.1047120418848166,
-      "grad_norm": 0.13848395497179677,
+      "grad_norm": 0.2808906980570263,
       "learning_rate": 1.2385844948367324e-05,
-      "logits/chosen": -4.28125,
-      "logits/rejected": -2.359375,
-      "logps/chosen": -1440.0,
-      "logps/rejected": -2544.0,
-      "loss": 0.0217,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -10.6875,
-      "rewards/margins": 9.75,
-      "rewards/rejected": -20.5,
+      "logits/chosen": -2.703125,
+      "logits/rejected": -1.671875,
+      "logps/chosen": -1232.0,
+      "logps/rejected": -2224.0,
+      "loss": 0.0202,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -8.5625,
+      "rewards/margins": 8.625,
+      "rewards/rejected": -17.25,
       "step": 2010
     },
     {
       "epoch": 2.115183246073298,
-      "grad_norm": 0.22370134033723152,
-      "learning_rate": 1.2123757973432114e-05,
-      "logits/chosen": -4.8125,
-      "logits/rejected": -2.734375,
-      "logps/chosen": -1472.0,
-      "logps/rejected": -2464.0,
-      "loss": 0.0173,
+      "grad_norm": 0.27712575136132994,
+      "learning_rate": 1.2123757973432114e-05,
+      "logits/chosen": -3.84375,
+      "logits/rejected": -2.125,
+      "logps/chosen": -1224.0,
+      "logps/rejected": -2144.0,
+      "loss": 0.0169,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -10.625,
-      "rewards/margins": 9.0,
-      "rewards/rejected": -19.625,
+      "rewards/chosen": -8.0625,
+      "rewards/margins": 8.4375,
+      "rewards/rejected": -16.5,
       "step": 2020
     },
     {
       "epoch": 2.1256544502617802,
-      "grad_norm": 0.1993368306799259,
+      "grad_norm": 0.5344916679933674,
       "learning_rate": 1.1863583129645106e-05,
-      "logits/chosen": -4.15625,
-      "logits/rejected": -2.9375,
-      "logps/chosen": -1424.0,
-      "logps/rejected": -2672.0,
-      "loss": 0.0201,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -10.375,
-      "rewards/margins": 9.75,
-      "rewards/rejected": -20.125,
+      "logits/chosen": -3.078125,
+      "logits/rejected": -2.46875,
+      "logps/chosen": -1192.0,
+      "logps/rejected": -2416.0,
+      "loss": 0.0191,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -8.0,
+      "rewards/margins": 9.625,
+      "rewards/rejected": -17.625,
       "step": 2030
     },
     {
       "epoch": 2.136125654450262,
-      "grad_norm": 0.22485964142272405,
+      "grad_norm": 0.39874129841143435,
       "learning_rate": 1.1605359053156607e-05,
-      "logits/chosen": -4.4375,
-      "logits/rejected": -1.96875,
-      "logps/chosen": -1472.0,
-      "logps/rejected": -2608.0,
-      "loss": 0.0118,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.4375,
-      "rewards/margins": 9.6875,
-      "rewards/rejected": -20.125,
+      "logits/chosen": -3.671875,
+      "logits/rejected": -1.8515625,
+      "logps/chosen": -1240.0,
+      "logps/rejected": -2320.0,
+      "loss": 0.0132,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -8.1875,
+      "rewards/margins": 9.0625,
+      "rewards/rejected": -17.25,
       "step": 2040
     },
     {
       "epoch": 2.1465968586387434,
-      "grad_norm": 0.08852618802289816,
+      "grad_norm": 0.11624150506743383,
       "learning_rate": 1.1349124090426644e-05,
-      "logits/chosen": -3.515625,
-      "logits/rejected": -1.5234375,
-      "logps/chosen": -1424.0,
-      "logps/rejected": -2720.0,
-      "loss": 0.0134,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.25,
-      "rewards/margins": 10.5625,
-      "rewards/rejected": -20.875,
+      "logits/chosen": -2.484375,
+      "logits/rejected": -1.078125,
+      "logps/chosen": -1240.0,
+      "logps/rejected": -2480.0,
+      "loss": 0.0139,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -8.375,
+      "rewards/margins": 10.0,
+      "rewards/rejected": -18.375,
       "step": 2050
     },
     {
       "epoch": 2.157068062827225,
-      "grad_norm": 0.47479179910766867,
+      "grad_norm": 0.053072936305807145,
       "learning_rate": 1.1094916292530406e-05,
-      "logits/chosen": -3.34375,
-      "logits/rejected": -1.9609375,
-      "logps/chosen": -1440.0,
-      "logps/rejected": -2528.0,
-      "loss": 0.0154,
+      "logits/chosen": -2.1875,
+      "logits/rejected": -1.421875,
+      "logps/chosen": -1240.0,
+      "logps/rejected": -2304.0,
+      "loss": 0.0152,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -10.25,
-      "rewards/margins": 9.25,
-      "rewards/rejected": -19.5,
+      "rewards/chosen": -8.25,
+      "rewards/margins": 9.0625,
+      "rewards/rejected": -17.375,
       "step": 2060
     },
     {
       "epoch": 2.167539267015707,
-      "grad_norm": 0.09578626557974271,
+      "grad_norm": 0.36118743665834424,
       "learning_rate": 1.0842773409507623e-05,
-      "logits/chosen": -3.3125,
-      "logits/rejected": -2.09375,
-      "logps/chosen": -1416.0,
-      "logps/rejected": -2656.0,
-      "loss": 0.018,
+      "logits/chosen": -2.203125,
+      "logits/rejected": -1.421875,
+      "logps/chosen": -1208.0,
+      "logps/rejected": -2432.0,
+      "loss": 0.0138,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.4375,
+      "rewards/chosen": -8.375,
       "rewards/margins": 10.0,
-      "rewards/rejected": -20.5,
+      "rewards/rejected": -18.25,
       "step": 2070
     },
     {
       "epoch": 2.1780104712041886,
-      "grad_norm": 0.23092446662860747,
+      "grad_norm": 0.5555422637893556,
       "learning_rate": 1.0592732884756754e-05,
-      "logits/chosen": -4.3125,
-      "logits/rejected": -2.734375,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2880.0,
-      "loss": 0.0128,
+      "logits/chosen": -2.9375,
+      "logits/rejected": -1.9453125,
+      "logps/chosen": -1312.0,
+      "logps/rejected": -2704.0,
+      "loss": 0.0109,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -10.8125,
-      "rewards/margins": 10.8125,
-      "rewards/rejected": -21.625,
+      "rewards/chosen": -9.0,
+      "rewards/margins": 10.875,
+      "rewards/rejected": -19.875,
       "step": 2080
     },
     {
       "epoch": 2.18848167539267,
-      "grad_norm": 0.08950959724147078,
+      "grad_norm": 0.13776730262877457,
       "learning_rate": 1.0344831849474507e-05,
-      "logits/chosen": -3.9375,
-      "logits/rejected": -2.0,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2768.0,
-      "loss": 0.01,
+      "logits/chosen": -2.609375,
+      "logits/rejected": -1.1796875,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0088,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.1875,
-      "rewards/margins": 10.3125,
-      "rewards/rejected": -21.5,
+      "rewards/chosen": -9.6875,
+      "rewards/margins": 10.6875,
+      "rewards/rejected": -20.375,
       "step": 2090
     },
     {
       "epoch": 2.1989528795811517,
-      "grad_norm": 0.14829506514767885,
+      "grad_norm": 0.6880302116803799,
       "learning_rate": 1.0099107117141879e-05,
-      "logits/chosen": -3.40625,
-      "logits/rejected": -1.390625,
-      "logps/chosen": -1464.0,
-      "logps/rejected": -2656.0,
-      "loss": 0.0091,
+      "logits/chosen": -1.765625,
+      "logits/rejected": -0.50390625,
+      "logps/chosen": -1336.0,
+      "logps/rejected": -2512.0,
+      "loss": 0.033,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -10.875,
-      "rewards/margins": 10.125,
-      "rewards/rejected": -21.0,
+      "rewards/chosen": -9.5625,
+      "rewards/margins": 10.0625,
+      "rewards/rejected": -19.625,
       "step": 2100
     },
     {
       "epoch": 2.1989528795811517,
-      "eval_logits/chosen": -1.3671875,
-      "eval_logits/rejected": 0.50390625,
-      "eval_logps/chosen": -1720.0,
-      "eval_logps/rejected": -1904.0,
-      "eval_loss": 1.7226641178131104,
-      "eval_rewards/accuracies": 0.625,
-      "eval_rewards/chosen": -14.0,
-      "eval_rewards/margins": 2.15625,
-      "eval_rewards/rejected": -16.125,
-      "eval_runtime": 47.0855,
-      "eval_samples_per_second": 42.476,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -0.2431640625,
+      "eval_logits/rejected": 1.3359375,
+      "eval_logps/chosen": -1544.0,
+      "eval_logps/rejected": -1720.0,
+      "eval_loss": 1.6682578325271606,
+      "eval_rewards/accuracies": 0.63671875,
+      "eval_rewards/chosen": -12.25,
+      "eval_rewards/margins": 2.0625,
+      "eval_rewards/rejected": -14.3125,
+      "eval_runtime": 46.5424,
+      "eval_samples_per_second": 42.972,
+      "eval_steps_per_second": 0.688,
       "step": 2100
     },
     {
       "epoch": 2.2094240837696333,
-      "grad_norm": 0.14880506586335354,
+      "grad_norm": 0.2777428850827323,
       "learning_rate": 9.855595178057334e-06,
-      "logits/chosen": -3.84375,
-      "logits/rejected": -2.171875,
-      "logps/chosen": -1560.0,
-      "logps/rejected": -2800.0,
-      "loss": 0.0099,
+      "logits/chosen": -2.671875,
+      "logits/rejected": -1.4921875,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2592.0,
+      "loss": 0.0113,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.5,
-      "rewards/margins": 10.625,
-      "rewards/rejected": -22.125,
+      "rewards/chosen": -9.4375,
+      "rewards/margins": 10.5,
+      "rewards/rejected": -20.0,
       "step": 2110
     },
     {
       "epoch": 2.2198952879581153,
-      "grad_norm": 0.11619690142811516,
+      "grad_norm": 0.3904280154605227,
       "learning_rate": 9.614332193917949e-06,
-      "logits/chosen": -3.671875,
-      "logits/rejected": -1.453125,
-      "logps/chosen": -1568.0,
-      "logps/rejected": -2832.0,
-      "loss": 0.0204,
+      "logits/chosen": -2.421875,
+      "logits/rejected": -0.875,
+      "logps/chosen": -1320.0,
+      "logps/rejected": -2528.0,
+      "loss": 0.0142,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 10.625,
-      "rewards/rejected": -22.0,
+      "rewards/chosen": -8.9375,
+      "rewards/margins": 10.125,
+      "rewards/rejected": -19.0,
       "step": 2120
     },
     {
       "epoch": 2.230366492146597,
-      "grad_norm": 0.6474828677254899,
+      "grad_norm": 0.3445355559272349,
       "learning_rate": 9.375353992449385e-06,
-      "logits/chosen": -4.5,
-      "logits/rejected": -2.953125,
-      "logps/chosen": -1568.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0077,
+      "logits/chosen": -3.1875,
+      "logits/rejected": -2.234375,
+      "logps/chosen": -1312.0,
+      "logps/rejected": -2528.0,
+      "loss": 0.0125,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.75,
-      "rewards/margins": 11.0,
-      "rewards/rejected": -22.75,
+      "rewards/chosen": -9.125,
+      "rewards/margins": 10.375,
+      "rewards/rejected": -19.5,
       "step": 2130
     },
     {
       "epoch": 2.2408376963350785,
-      "grad_norm": 1.2532545067430394,
+      "grad_norm": 0.22325334490451682,
       "learning_rate": 9.138696062085442e-06,
-      "logits/chosen": -4.21875,
-      "logits/rejected": -2.171875,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2960.0,
-      "loss": 0.0113,
+      "logits/chosen": -2.921875,
+      "logits/rejected": -1.84375,
+      "logps/chosen": -1272.0,
+      "logps/rejected": -2640.0,
+      "loss": 0.0137,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.25,
-      "rewards/margins": 12.25,
-      "rewards/rejected": -23.5,
+      "rewards/chosen": -8.8125,
+      "rewards/margins": 11.4375,
+      "rewards/rejected": -20.25,
       "step": 2140
     },
     {
       "epoch": 2.25130890052356,
-      "grad_norm": 0.17305712795534955,
+      "grad_norm": 0.157948914293651,
       "learning_rate": 8.904393546698007e-06,
-      "logits/chosen": -4.40625,
-      "logits/rejected": -1.9375,
-      "logps/chosen": -1576.0,
-      "logps/rejected": -2944.0,
-      "loss": 0.0151,
+      "logits/chosen": -3.25,
+      "logits/rejected": -1.765625,
+      "logps/chosen": -1288.0,
+      "logps/rejected": -2560.0,
+      "loss": 0.0198,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.75,
-      "rewards/margins": 11.5,
-      "rewards/rejected": -23.25,
+      "rewards/chosen": -8.8125,
+      "rewards/margins": 10.5,
+      "rewards/rejected": -19.375,
       "step": 2150
     },
     {
       "epoch": 2.261780104712042,
-      "grad_norm": 0.19183191663432336,
+      "grad_norm": 0.3095113913495355,
       "learning_rate": 8.672481240378142e-06,
-      "logits/chosen": -3.4375,
-      "logits/rejected": -1.4296875,
-      "logps/chosen": -1472.0,
-      "logps/rejected": -2752.0,
-      "loss": 0.0199,
+      "logits/chosen": -2.4375,
+      "logits/rejected": -1.328125,
+      "logps/chosen": -1200.0,
+      "logps/rejected": -2384.0,
+      "loss": 0.0185,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.1875,
-      "rewards/margins": 10.75,
-      "rewards/rejected": -21.875,
+      "rewards/chosen": -8.5,
+      "rewards/margins": 9.75,
+      "rewards/rejected": -18.25,
       "step": 2160
     },
     {
       "epoch": 2.2722513089005236,
-      "grad_norm": 1.0618485036326681,
+      "grad_norm": 0.1629079365572,
       "learning_rate": 8.44299358226919e-06,
-      "logits/chosen": -3.78125,
-      "logits/rejected": -1.6953125,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2848.0,
-      "loss": 0.0174,
+      "logits/chosen": -2.390625,
+      "logits/rejected": -1.2109375,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2544.0,
+      "loss": 0.015,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.1875,
-      "rewards/margins": 10.875,
-      "rewards/rejected": -22.125,
+      "rewards/chosen": -9.0,
+      "rewards/margins": 10.0625,
+      "rewards/rejected": -19.125,
       "step": 2170
     },
     {
       "epoch": 2.282722513089005,
-      "grad_norm": 0.276997394354002,
+      "grad_norm": 0.11813293644061723,
       "learning_rate": 8.215964651452457e-06,
-      "logits/chosen": -2.984375,
-      "logits/rejected": -0.828125,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0074,
+      "logits/chosen": -2.046875,
+      "logits/rejected": -0.51953125,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2640.0,
+      "loss": 0.0125,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.3125,
-      "rewards/margins": 10.9375,
-      "rewards/rejected": -22.25,
+      "rewards/chosen": -9.125,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -20.0,
       "step": 2180
     },
     {
       "epoch": 2.2931937172774868,
-      "grad_norm": 0.23799460846539638,
+      "grad_norm": 0.47894716942467785,
       "learning_rate": 7.991428161886502e-06,
-      "logits/chosen": -3.9375,
-      "logits/rejected": -1.90625,
-      "logps/chosen": -1504.0,
-      "logps/rejected": -2672.0,
-      "loss": 0.0072,
+      "logits/chosen": -2.71875,
+      "logits/rejected": -1.28125,
+      "logps/chosen": -1320.0,
+      "logps/rejected": -2528.0,
+      "loss": 0.0076,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 10.25,
-      "rewards/rejected": -21.375,
+      "rewards/chosen": -9.1875,
+      "rewards/margins": 10.625,
+      "rewards/rejected": -19.875,
       "step": 2190
     },
     {
       "epoch": 2.303664921465969,
-      "grad_norm": 0.18076747142854013,
+      "grad_norm": 0.1876519612378706,
       "learning_rate": 7.76941745740061e-06,
-      "logits/chosen": -3.75,
-      "logits/rejected": -1.8671875,
-      "logps/chosen": -1624.0,
-      "logps/rejected": -2928.0,
-      "loss": 0.0101,
+      "logits/chosen": -2.578125,
+      "logits/rejected": -1.53125,
+      "logps/chosen": -1456.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0123,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -11.9375,
-      "rewards/margins": 11.3125,
-      "rewards/rejected": -23.25,
+      "rewards/chosen": -10.3125,
+      "rewards/margins": 10.3125,
+      "rewards/rejected": -20.625,
       "step": 2200
     },
     {
       "epoch": 2.303664921465969,
-      "eval_logits/chosen": -0.8359375,
-      "eval_logits/rejected": 0.984375,
-      "eval_logps/chosen": -1824.0,
-      "eval_logps/rejected": -2016.0,
-      "eval_loss": 1.8446015119552612,
-      "eval_rewards/accuracies": 0.640625,
-      "eval_rewards/chosen": -15.0625,
-      "eval_rewards/margins": 2.234375,
-      "eval_rewards/rejected": -17.25,
-      "eval_runtime": 47.0945,
-      "eval_samples_per_second": 42.468,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": 0.142578125,
+      "eval_logits/rejected": 1.6640625,
+      "eval_logps/chosen": -1616.0,
+      "eval_logps/rejected": -1792.0,
+      "eval_loss": 1.68359375,
+      "eval_rewards/accuracies": 0.62890625,
+      "eval_rewards/chosen": -13.0,
+      "eval_rewards/margins": 2.03125,
+      "eval_rewards/rejected": -15.0,
+      "eval_runtime": 46.5365,
+      "eval_samples_per_second": 42.977,
+      "eval_steps_per_second": 0.688,
       "step": 2200
     },
     {
       "epoch": 2.3141361256544504,
-      "grad_norm": 0.22755235241110316,
+      "grad_norm": 0.27939244593341905,
       "learning_rate": 7.549965506743123e-06,
-      "logits/chosen": -3.546875,
-      "logits/rejected": -2.015625,
-      "logps/chosen": -1576.0,
-      "logps/rejected": -2832.0,
-      "loss": 0.0102,
+      "logits/chosen": -2.15625,
+      "logits/rejected": -1.2734375,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2592.0,
+      "loss": 0.0077,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.9375,
-      "rewards/margins": 10.8125,
-      "rewards/rejected": -22.75,
+      "rewards/chosen": -10.0625,
+      "rewards/margins": 10.4375,
+      "rewards/rejected": -20.5,
       "step": 2210
     },
     {
       "epoch": 2.324607329842932,
-      "grad_norm": 0.04908119155051022,
+      "grad_norm": 0.21038205797566167,
       "learning_rate": 7.333104898685631e-06,
-      "logits/chosen": -3.453125,
-      "logits/rejected": -1.921875,
-      "logps/chosen": -1544.0,
-      "logps/rejected": -2800.0,
-      "loss": 0.0108,
+      "logits/chosen": -1.953125,
+      "logits/rejected": -1.140625,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2592.0,
+      "loss": 0.0102,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.5,
-      "rewards/margins": 11.0,
-      "rewards/rejected": -22.5,
+      "rewards/chosen": -9.9375,
+      "rewards/margins": 10.5625,
+      "rewards/rejected": -20.5,
       "step": 2220
     },
     {
       "epoch": 2.3350785340314135,
-      "grad_norm": 0.08915612839685524,
+      "grad_norm": 0.23309906486257237,
       "learning_rate": 7.118867837183518e-06,
-      "logits/chosen": -4.625,
-      "logits/rejected": -1.875,
-      "logps/chosen": -1616.0,
-      "logps/rejected": -2976.0,
-      "loss": 0.0202,
+      "logits/chosen": -3.03125,
+      "logits/rejected": -1.1171875,
+      "logps/chosen": -1496.0,
+      "logps/rejected": -2848.0,
+      "loss": 0.0093,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.5,
-      "rewards/margins": 12.125,
-      "rewards/rejected": -23.625,
+      "rewards/chosen": -10.375,
+      "rewards/margins": 12.0625,
+      "rewards/rejected": -22.375,
       "step": 2230
     },
     {
       "epoch": 2.345549738219895,
-      "grad_norm": 0.12457909704050821,
+      "grad_norm": 0.06877555528767784,
       "learning_rate": 6.9072861365936045e-06,
-      "logits/chosen": -4.4375,
-      "logits/rejected": -2.484375,
-      "logps/chosen": -1640.0,
-      "logps/rejected": -3024.0,
-      "loss": 0.0078,
+      "logits/chosen": -2.953125,
+      "logits/rejected": -1.7421875,
+      "logps/chosen": -1568.0,
+      "logps/rejected": -2976.0,
+      "loss": 0.0089,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.8125,
-      "rewards/margins": 11.6875,
-      "rewards/rejected": -23.5,
+      "rewards/chosen": -11.1875,
+      "rewards/margins": 11.8125,
+      "rewards/rejected": -23.0,
       "step": 2240
     },
     {
       "epoch": 2.356020942408377,
-      "grad_norm": 0.2286841613244401,
+      "grad_norm": 0.8432324682845347,
       "learning_rate": 6.6983912169497005e-06,
-      "logits/chosen": -3.71875,
-      "logits/rejected": -1.6328125,
-      "logps/chosen": -1560.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0135,
+      "logits/chosen": -2.3125,
+      "logits/rejected": -0.84765625,
+      "logps/chosen": -1504.0,
+      "logps/rejected": -2832.0,
+      "loss": 0.0134,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.25,
-      "rewards/margins": 10.9375,
-      "rewards/rejected": -22.125,
+      "rewards/chosen": -10.625,
+      "rewards/margins": 11.125,
+      "rewards/rejected": -21.75,
       "step": 2250
     },
     {
       "epoch": 2.3664921465968587,
-      "grad_norm": 0.042937808849063154,
+      "grad_norm": 0.2322416734759718,
       "learning_rate": 6.49221409929677e-06,
-      "logits/chosen": -2.71875,
-      "logits/rejected": -0.54296875,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2720.0,
-      "loss": 0.0097,
+      "logits/chosen": -1.4765625,
+      "logits/rejected": -0.056396484375,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2624.0,
+      "loss": 0.0086,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 10.9375,
-      "rewards/rejected": -22.0,
+      "rewards/chosen": -10.0,
+      "rewards/margins": 11.125,
+      "rewards/rejected": -21.125,
       "step": 2260
     },
     {
       "epoch": 2.3769633507853403,
-      "grad_norm": 0.35863060407012703,
+      "grad_norm": 0.1276694450821154,
       "learning_rate": 6.2887854010842075e-06,
-      "logits/chosen": -2.5,
-      "logits/rejected": -1.4609375,
-      "logps/chosen": -1616.0,
-      "logps/rejected": -2896.0,
-      "loss": 0.0124,
+      "logits/chosen": -1.6953125,
+      "logits/rejected": -1.015625,
+      "logps/chosen": -1456.0,
+      "logps/rejected": -2816.0,
+      "loss": 0.0104,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -12.4375,
-      "rewards/margins": 10.9375,
-      "rewards/rejected": -23.375,
+      "rewards/chosen": -10.8125,
+      "rewards/margins": 11.75,
+      "rewards/rejected": -22.5,
       "step": 2270
     },
     {
       "epoch": 2.387434554973822,
-      "grad_norm": 1.6091554416518914,
+      "grad_norm": 0.2818628533877098,
       "learning_rate": 6.0881353316191385e-06,
-      "logits/chosen": -3.828125,
-      "logits/rejected": -2.265625,
-      "logps/chosen": -1576.0,
-      "logps/rejected": -2928.0,
-      "loss": 0.0125,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.625,
-      "rewards/margins": 11.625,
-      "rewards/rejected": -23.25,
+      "logits/chosen": -2.40625,
+      "logits/rejected": -1.5,
+      "logps/chosen": -1456.0,
+      "logps/rejected": -2784.0,
+      "loss": 0.0142,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -10.375,
+      "rewards/margins": 11.3125,
+      "rewards/rejected": -21.625,
       "step": 2280
     },
     {
       "epoch": 2.3979057591623034,
-      "grad_norm": 0.1263364201350342,
+      "grad_norm": 0.1189279025623444,
       "learning_rate": 5.890293687580381e-06,
-      "logits/chosen": -2.671875,
-      "logits/rejected": -0.83984375,
-      "logps/chosen": -1600.0,
-      "logps/rejected": -2880.0,
-      "loss": 0.0122,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.875,
-      "rewards/margins": 11.375,
-      "rewards/rejected": -23.25,
+      "logits/chosen": -1.71875,
+      "logits/rejected": -0.39453125,
+      "logps/chosen": -1472.0,
+      "logps/rejected": -2704.0,
+      "loss": 0.0097,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -10.625,
+      "rewards/margins": 10.875,
+      "rewards/rejected": -21.5,
       "step": 2290
     },
     {
       "epoch": 2.4083769633507854,
-      "grad_norm": 0.6813179391799218,
-      "learning_rate": 5.695289848593532e-06,
-      "logits/chosen": -4.0,
-      "logits/rejected": -2.71875,
-      "logps/chosen": -1480.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.0145,
-      "rewards/accuracies": 0.96875,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 10.625,
-      "rewards/rejected": -21.625,
+      "grad_norm": 0.2499631722690142,
+      "learning_rate": 5.695289848593532e-06,
+      "logits/chosen": -2.828125,
+      "logits/rejected": -2.203125,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0123,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 10.875,
+      "rewards/rejected": -20.75,
       "step": 2300
     },
     {
       "epoch": 2.4083769633507854,
-      "eval_logits/chosen": -0.66796875,
-      "eval_logits/rejected": 1.171875,
-      "eval_logps/chosen": -1768.0,
-      "eval_logps/rejected": -1968.0,
-      "eval_loss": 1.791085958480835,
-      "eval_rewards/accuracies": 0.62890625,
-      "eval_rewards/chosen": -14.5,
-      "eval_rewards/margins": 2.265625,
-      "eval_rewards/rejected": -16.75,
-      "eval_runtime": 47.0854,
-      "eval_samples_per_second": 42.476,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": 0.193359375,
+      "eval_logits/rejected": 1.75,
+      "eval_logps/chosen": -1656.0,
+      "eval_logps/rejected": -1848.0,
+      "eval_loss": 1.731601595878601,
+      "eval_rewards/accuracies": 0.626953125,
+      "eval_rewards/chosen": -13.375,
+      "eval_rewards/margins": 2.203125,
+      "eval_rewards/rejected": -15.5625,
+      "eval_runtime": 46.5202,
+      "eval_samples_per_second": 42.992,
+      "eval_steps_per_second": 0.688,
       "step": 2300
     },
     {
       "epoch": 2.418848167539267,
-      "grad_norm": 0.3636623403280782,
+      "grad_norm": 0.12776226950000688,
       "learning_rate": 5.50315277286812e-06,
-      "logits/chosen": -3.8125,
-      "logits/rejected": -2.078125,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.0128,
+      "logits/chosen": -2.28125,
+      "logits/rejected": -1.234375,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2704.0,
+      "loss": 0.0132,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.125,
-      "rewards/margins": 11.125,
-      "rewards/rejected": -22.25,
+      "rewards/chosen": -10.125,
+      "rewards/margins": 11.75,
+      "rewards/rejected": -21.875,
       "step": 2310
     },
     {
       "epoch": 2.4293193717277486,
-      "grad_norm": 0.19419140933440066,
+      "grad_norm": 0.23021586294411645,
       "learning_rate": 5.313910992897281e-06,
-      "logits/chosen": -3.296875,
-      "logits/rejected": -1.25,
-      "logps/chosen": -1472.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0079,
+      "logits/chosen": -1.8125,
+      "logits/rejected": -0.62890625,
+      "logps/chosen": -1352.0,
+      "logps/rejected": -2752.0,
+      "loss": 0.0083,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -10.8125,
-      "rewards/margins": 11.625,
-      "rewards/rejected": -22.375,
+      "rewards/chosen": -9.625,
+      "rewards/margins": 11.6875,
+      "rewards/rejected": -21.375,
       "step": 2320
     },
     {
       "epoch": 2.4397905759162306,
-      "grad_norm": 0.3216749125997076,
+      "grad_norm": 0.20300961825486086,
       "learning_rate": 5.127592611220658e-06,
-      "logits/chosen": -3.515625,
-      "logits/rejected": -1.7734375,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2896.0,
-      "loss": 0.0205,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.25,
-      "rewards/margins": 11.3125,
-      "rewards/rejected": -22.625,
+      "logits/chosen": -2.328125,
+      "logits/rejected": -1.328125,
+      "logps/chosen": -1432.0,
+      "logps/rejected": -2736.0,
+      "loss": 0.018,
+      "rewards/accuracies": 0.9750000238418579,
+      "rewards/chosen": -10.1875,
+      "rewards/margins": 10.875,
+      "rewards/rejected": -21.0,
       "step": 2330
     },
     {
       "epoch": 2.450261780104712,
-      "grad_norm": 0.20579957150346426,
+      "grad_norm": 1.0057435281197067,
       "learning_rate": 4.944225296251159e-06,
-      "logits/chosen": -3.609375,
-      "logits/rejected": -2.421875,
-      "logps/chosen": -1448.0,
-      "logps/rejected": -2752.0,
-      "loss": 0.0106,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.875,
+      "logits/chosen": -2.0,
+      "logits/rejected": -1.609375,
+      "logps/chosen": -1376.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.012,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -10.125,
       "rewards/margins": 11.0,
-      "rewards/rejected": -21.875,
+      "rewards/rejected": -21.125,
       "step": 2340
     },
     {
       "epoch": 2.4607329842931938,
-      "grad_norm": 0.11492631505791115,
+      "grad_norm": 0.041410474843955034,
       "learning_rate": 4.763836278166172e-06,
-      "logits/chosen": -4.125,
-      "logits/rejected": -1.8359375,
-      "logps/chosen": -1424.0,
-      "logps/rejected": -2624.0,
-      "loss": 0.0092,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -10.375,
-      "rewards/margins": 10.4375,
-      "rewards/rejected": -20.875,
+      "logits/chosen": -2.15625,
+      "logits/rejected": -0.68359375,
+      "logps/chosen": -1336.0,
+      "logps/rejected": -2576.0,
+      "loss": 0.0099,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -9.5,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -20.375,
       "step": 2350
     },
     {
       "epoch": 2.4712041884816753,
-      "grad_norm": 0.05241783911536356,
+      "grad_norm": 0.12230997071302514,
       "learning_rate": 4.586452344863876e-06,
-      "logits/chosen": -3.90625,
-      "logits/rejected": -1.9921875,
-      "logps/chosen": -1512.0,
-      "logps/rejected": -2784.0,
-      "loss": 0.0124,
-      "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -11.0,
-      "rewards/margins": 11.0,
-      "rewards/rejected": -22.0,
+      "logits/chosen": -2.484375,
+      "logits/rejected": -1.2890625,
+      "logps/chosen": -1392.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.0119,
+      "rewards/accuracies": 0.981249988079071,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 11.125,
+      "rewards/rejected": -21.0,
       "step": 2360
     },
     {
       "epoch": 2.481675392670157,
-      "grad_norm": 0.03555405707770405,
+      "grad_norm": 0.6743478551412059,
       "learning_rate": 4.412099837985192e-06,
-      "logits/chosen": -3.46875,
-      "logits/rejected": -1.25,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2928.0,
-      "loss": 0.0117,
+      "logits/chosen": -2.15625,
+      "logits/rejected": -0.6796875,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2784.0,
+      "loss": 0.0123,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.0625,
+      "rewards/chosen": -9.625,
       "rewards/margins": 11.9375,
-      "rewards/rejected": -23.0,
+      "rewards/rejected": -21.625,
       "step": 2370
     },
     {
       "epoch": 2.492146596858639,
-      "grad_norm": 0.33592489701696837,
+      "grad_norm": 0.1524624556571847,
       "learning_rate": 4.240804649002089e-06,
-      "logits/chosen": -3.828125,
-      "logits/rejected": -1.5078125,
-      "logps/chosen": -1432.0,
-      "logps/rejected": -2624.0,
-      "loss": 0.0184,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -10.75,
-      "rewards/margins": 10.6875,
-      "rewards/rejected": -21.375,
+      "logits/chosen": -2.640625,
+      "logits/rejected": -0.95703125,
+      "logps/chosen": -1280.0,
+      "logps/rejected": -2432.0,
+      "loss": 0.0202,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -9.1875,
+      "rewards/margins": 10.3125,
+      "rewards/rejected": -19.5,
       "step": 2380
     },
     {
       "epoch": 2.5026178010471205,
-      "grad_norm": 1.128534879252504,
+      "grad_norm": 0.6119292876305849,
       "learning_rate": 4.072592215372598e-06,
-      "logits/chosen": -4.40625,
-      "logits/rejected": -2.34375,
-      "logps/chosen": -1560.0,
-      "logps/rejected": -2848.0,
-      "loss": 0.0122,
+      "logits/chosen": -3.53125,
+      "logits/rejected": -2.0,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2592.0,
+      "loss": 0.0125,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.25,
-      "rewards/margins": 11.6875,
-      "rewards/rejected": -23.0,
+      "rewards/chosen": -9.375,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -20.375,
       "step": 2390
     },
     {
       "epoch": 2.513089005235602,
-      "grad_norm": 0.415737848268598,
+      "grad_norm": 2.5651159703779505,
       "learning_rate": 3.907487516763389e-06,
-      "logits/chosen": -3.109375,
-      "logits/rejected": -1.6796875,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2848.0,
-      "loss": 0.0114,
+      "logits/chosen": -2.09375,
+      "logits/rejected": -1.2578125,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2560.0,
+      "loss": 0.0167,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.3125,
-      "rewards/margins": 11.3125,
-      "rewards/rejected": -22.625,
+      "rewards/chosen": -9.125,
+      "rewards/margins": 10.6875,
+      "rewards/rejected": -19.875,
       "step": 2400
     },
     {
       "epoch": 2.513089005235602,
-      "eval_logits/chosen": -0.54296875,
-      "eval_logits/rejected": 1.234375,
-      "eval_logps/chosen": -1792.0,
-      "eval_logps/rejected": -1992.0,
-      "eval_loss": 1.7978359460830688,
-      "eval_rewards/accuracies": 0.626953125,
-      "eval_rewards/chosen": -14.75,
-      "eval_rewards/margins": 2.28125,
-      "eval_rewards/rejected": -17.0,
-      "eval_runtime": 47.0926,
-      "eval_samples_per_second": 42.469,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": -0.1259765625,
+      "eval_logits/rejected": 1.46875,
+      "eval_logps/chosen": -1496.0,
+      "eval_logps/rejected": -1672.0,
+      "eval_loss": 1.6171094179153442,
+      "eval_rewards/accuracies": 0.630859375,
+      "eval_rewards/chosen": -11.8125,
+      "eval_rewards/margins": 2.03125,
+      "eval_rewards/rejected": -13.8125,
+      "eval_runtime": 46.5414,
+      "eval_samples_per_second": 42.973,
+      "eval_steps_per_second": 0.688,
       "step": 2400
     },
     {
       "epoch": 2.5235602094240837,
-      "grad_norm": 0.28650815039070604,
+      "grad_norm": 0.2103146095010995,
       "learning_rate": 3.745515071340286e-06,
-      "logits/chosen": -3.0,
-      "logits/rejected": -1.2734375,
-      "logps/chosen": -1584.0,
-      "logps/rejected": -2928.0,
-      "loss": 0.0125,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.6875,
-      "rewards/margins": 11.625,
-      "rewards/rejected": -23.25,
+      "logits/chosen": -2.15625,
+      "logits/rejected": -0.96484375,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0123,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -9.4375,
+      "rewards/margins": 11.0625,
+      "rewards/rejected": -20.5,
       "step": 2410
     },
     {
       "epoch": 2.5340314136125652,
-      "grad_norm": 0.050769491764024226,
+      "grad_norm": 0.10079461122383392,
       "learning_rate": 3.586698932127236e-06,
-      "logits/chosen": -4.0,
-      "logits/rejected": -1.8828125,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2768.0,
-      "loss": 0.0127,
+      "logits/chosen": -3.015625,
+      "logits/rejected": -1.5,
+      "logps/chosen": -1288.0,
+      "logps/rejected": -2528.0,
+      "loss": 0.0094,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.25,
+      "rewards/chosen": -8.9375,
       "rewards/margins": 10.9375,
-      "rewards/rejected": -22.25,
+      "rewards/rejected": -19.75,
       "step": 2420
     },
     {
       "epoch": 2.5445026178010473,
-      "grad_norm": 0.6324762557871049,
+      "grad_norm": 0.6597356039693393,
       "learning_rate": 3.4310626834344744e-06,
-      "logits/chosen": -4.0625,
-      "logits/rejected": -2.640625,
-      "logps/chosen": -1608.0,
-      "logps/rejected": -2912.0,
-      "loss": 0.0119,
-      "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.875,
-      "rewards/margins": 11.1875,
-      "rewards/rejected": -23.0,
+      "logits/chosen": -3.265625,
+      "logits/rejected": -2.125,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2624.0,
+      "loss": 0.0114,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.4375,
+      "rewards/margins": 10.75,
+      "rewards/rejected": -20.125,
       "step": 2430
     },
     {
       "epoch": 2.554973821989529,
-      "grad_norm": 0.0572509741374605,
+      "grad_norm": 0.03490918064950846,
       "learning_rate": 3.2786294373562343e-06,
-      "logits/chosen": -3.359375,
-      "logits/rejected": -1.296875,
-      "logps/chosen": -1552.0,
-      "logps/rejected": -3008.0,
-      "loss": 0.0103,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 12.3125,
-      "rewards/rejected": -23.75,
+      "logits/chosen": -2.359375,
+      "logits/rejected": -0.88671875,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2736.0,
+      "loss": 0.0074,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.5,
+      "rewards/margins": 11.4375,
+      "rewards/rejected": -21.0,
       "step": 2440
     },
     {
       "epoch": 2.5654450261780104,
-      "grad_norm": 0.41395884185200127,
+      "grad_norm": 0.9904612967863327,
       "learning_rate": 3.1294218303385526e-06,
-      "logits/chosen": -3.875,
-      "logits/rejected": -2.234375,
-      "logps/chosen": -1624.0,
-      "logps/rejected": -3088.0,
-      "loss": 0.008,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.875,
-      "rewards/margins": 11.875,
-      "rewards/rejected": -23.75,
+      "logits/chosen": -2.6875,
+      "logits/rejected": -1.8203125,
+      "logps/chosen": -1440.0,
+      "logps/rejected": -2832.0,
+      "loss": 0.0075,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -10.0,
+      "rewards/margins": 11.1875,
+      "rewards/rejected": -21.25,
       "step": 2450
     },
     {
       "epoch": 2.5759162303664924,
-      "grad_norm": 0.45416520058418547,
+      "grad_norm": 1.1764375262626894,
       "learning_rate": 2.9834620198177483e-06,
-      "logits/chosen": -3.515625,
-      "logits/rejected": -1.6640625,
-      "logps/chosen": -1584.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0141,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.75,
-      "rewards/margins": 11.3125,
-      "rewards/rejected": -23.125,
+      "logits/chosen": -2.546875,
+      "logits/rejected": -1.1875,
+      "logps/chosen": -1424.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.018,
+      "rewards/accuracies": 0.987500011920929,
+      "rewards/chosen": -10.1875,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -21.125,
       "step": 2460
     },
     {
       "epoch": 2.5863874345549736,
-      "grad_norm": 0.06555993289453634,
+      "grad_norm": 0.038405686392041144,
       "learning_rate": 2.840771680930068e-06,
-      "logits/chosen": -3.6875,
-      "logits/rejected": -2.015625,
-      "logps/chosen": -1656.0,
-      "logps/rejected": -3200.0,
-      "loss": 0.0108,
+      "logits/chosen": -2.640625,
+      "logits/rejected": -1.453125,
+      "logps/chosen": -1504.0,
+      "logps/rejected": -3024.0,
+      "loss": 0.0078,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -12.3125,
-      "rewards/margins": 12.625,
-      "rewards/rejected": -25.0,
+      "rewards/chosen": -10.75,
+      "rewards/margins": 12.4375,
+      "rewards/rejected": -23.125,
       "step": 2470
     },
     {
       "epoch": 2.5968586387434556,
-      "grad_norm": 0.14730237647348368,
+      "grad_norm": 0.3103971774674795,
       "learning_rate": 2.701372003292868e-06,
-      "logits/chosen": -2.546875,
-      "logits/rejected": -0.80859375,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2816.0,
-      "loss": 0.013,
+      "logits/chosen": -1.484375,
+      "logits/rejected": -0.35546875,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0139,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.375,
+      "rewards/chosen": -9.75,
       "rewards/margins": 10.9375,
-      "rewards/rejected": -22.25,
+      "rewards/rejected": -20.75,
       "step": 2480
     },
     {
       "epoch": 2.607329842931937,
-      "grad_norm": 0.19443148530885196,
+      "grad_norm": 0.4659432843933814,
       "learning_rate": 2.56528368785795e-06,
-      "logits/chosen": -2.78125,
-      "logits/rejected": -1.03125,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2704.0,
-      "loss": 0.0154,
+      "logits/chosen": -1.5859375,
+      "logits/rejected": -0.431640625,
+      "logps/chosen": -1400.0,
+      "logps/rejected": -2576.0,
+      "loss": 0.0075,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.125,
+      "rewards/chosen": -9.875,
       "rewards/margins": 10.5625,
-      "rewards/rejected": -21.625,
+      "rewards/rejected": -20.5,
       "step": 2490
     },
     {
       "epoch": 2.6178010471204187,
-      "grad_norm": 0.10815148657918729,
+      "grad_norm": 0.20957715724701875,
       "learning_rate": 2.432526943837471e-06,
-      "logits/chosen": -3.046875,
-      "logits/rejected": -0.859375,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2880.0,
-      "loss": 0.012,
+      "logits/chosen": -1.828125,
+      "logits/rejected": -0.4140625,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2752.0,
+      "loss": 0.0138,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.125,
+      "rewards/chosen": -9.875,
       "rewards/margins": 11.5,
-      "rewards/rejected": -22.625,
+      "rewards/rejected": -21.375,
       "step": 2500
     },
     {
       "epoch": 2.6178010471204187,
-      "eval_logits/chosen": -0.5234375,
-      "eval_logits/rejected": 1.265625,
-      "eval_logps/chosen": -1784.0,
-      "eval_logps/rejected": -1984.0,
-      "eval_loss": 1.7939844131469727,
-      "eval_rewards/accuracies": 0.630859375,
-      "eval_rewards/chosen": -14.6875,
-      "eval_rewards/margins": 2.28125,
-      "eval_rewards/rejected": -17.0,
-      "eval_runtime": 47.0982,
-      "eval_samples_per_second": 42.464,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": 0.2041015625,
+      "eval_logits/rejected": 1.75,
+      "eval_logps/chosen": -1664.0,
+      "eval_logps/rejected": -1848.0,
+      "eval_loss": 1.7572109699249268,
+      "eval_rewards/accuracies": 0.626953125,
+      "eval_rewards/chosen": -13.4375,
+      "eval_rewards/margins": 2.140625,
+      "eval_rewards/rejected": -15.625,
+      "eval_runtime": 46.5465,
+      "eval_samples_per_second": 42.968,
+      "eval_steps_per_second": 0.687,
       "step": 2500
     },
     {
       "epoch": 2.6282722513089007,
-      "grad_norm": 2.0052364520965056,
+      "grad_norm": 0.36075500749373224,
       "learning_rate": 2.3031214857028446e-06,
-      "logits/chosen": -3.078125,
-      "logits/rejected": -1.609375,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2912.0,
-      "loss": 0.0118,
+      "logits/chosen": -1.7734375,
+      "logits/rejected": -1.0703125,
+      "logps/chosen": -1416.0,
+      "logps/rejected": -2768.0,
+      "loss": 0.0134,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.1875,
-      "rewards/margins": 12.0,
-      "rewards/rejected": -23.125,
+      "rewards/chosen": -10.0,
+      "rewards/margins": 11.6875,
+      "rewards/rejected": -21.625,
       "step": 2510
     },
     {
       "epoch": 2.6387434554973823,
-      "grad_norm": 0.09847657007482878,
+      "grad_norm": 0.2180974537074586,
       "learning_rate": 2.177086530257139e-06,
-      "logits/chosen": -3.578125,
-      "logits/rejected": -1.671875,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2784.0,
-      "loss": 0.0146,
+      "logits/chosen": -2.546875,
+      "logits/rejected": -1.1953125,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2592.0,
+      "loss": 0.0138,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 10.9375,
-      "rewards/rejected": -22.375,
+      "rewards/chosen": -9.875,
+      "rewards/margins": 10.5625,
+      "rewards/rejected": -20.5,
       "step": 2520
     },
     {
       "epoch": 2.649214659685864,
-      "grad_norm": 0.16869525008885042,
+      "grad_norm": 0.16949992244173948,
       "learning_rate": 2.054440793781354e-06,
-      "logits/chosen": -3.609375,
-      "logits/rejected": -1.265625,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2816.0,
-      "loss": 0.0124,
+      "logits/chosen": -2.59375,
+      "logits/rejected": -0.921875,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2624.0,
+      "loss": 0.015,
       "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 11.125,
-      "rewards/rejected": -22.5,
+      "rewards/chosen": -9.625,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -20.5,
       "step": 2530
     },
     {
       "epoch": 2.6596858638743455,
-      "grad_norm": 0.10456975971332139,
+      "grad_norm": 0.1033446788995966,
       "learning_rate": 1.9352024892550587e-06,
-      "logits/chosen": -3.046875,
-      "logits/rejected": -1.1953125,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.012,
+      "logits/chosen": -2.25,
+      "logits/rejected": -0.85546875,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2544.0,
+      "loss": 0.0114,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.3125,
-      "rewards/margins": 10.8125,
-      "rewards/rejected": -22.125,
+      "rewards/chosen": -9.6875,
+      "rewards/margins": 10.5,
+      "rewards/rejected": -20.125,
       "step": 2540
     },
     {
       "epoch": 2.670157068062827,
-      "grad_norm": 0.23987183672477652,
+      "grad_norm": 0.236714433891974,
       "learning_rate": 1.8193893236517379e-06,
-      "logits/chosen": -3.703125,
-      "logits/rejected": -1.734375,
-      "logps/chosen": -1592.0,
-      "logps/rejected": -2912.0,
-      "loss": 0.0177,
-      "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.6875,
-      "rewards/margins": 11.25,
-      "rewards/rejected": -23.0,
+      "logits/chosen": -2.875,
+      "logits/rejected": -1.4140625,
+      "logps/chosen": -1408.0,
+      "logps/rejected": -2704.0,
+      "loss": 0.0122,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.875,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -20.875,
       "step": 2550
     },
     {
       "epoch": 2.680628272251309,
-      "grad_norm": 0.28280240621718694,
+      "grad_norm": 0.18278914845003005,
       "learning_rate": 1.7070184953092956e-06,
-      "logits/chosen": -3.6875,
-      "logits/rejected": -0.94921875,
-      "logps/chosen": -1448.0,
-      "logps/rejected": -2720.0,
-      "loss": 0.0117,
+      "logits/chosen": -2.625,
+      "logits/rejected": -0.69921875,
+      "logps/chosen": -1264.0,
+      "logps/rejected": -2496.0,
+      "loss": 0.0132,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -10.4375,
-      "rewards/margins": 11.25,
-      "rewards/rejected": -21.75,
+      "rewards/chosen": -8.625,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -19.5,
       "step": 2560
     },
     {
-      "epoch": 2.6910994764397906,
-      "grad_norm": 0.17868788890856102,
-      "learning_rate": 1.5981066913760918e-06,
-      "logits/chosen": -2.59375,
-      "logits/rejected": -0.53125,
-      "logps/chosen": -1432.0,
-      "logps/rejected": -2608.0,
-      "loss": 0.0076,
+      "epoch": 2.6910994764397906,
+      "grad_norm": 0.14335468662911013,
+      "learning_rate": 1.5981066913760918e-06,
+      "logits/chosen": -1.4453125,
+      "logits/rejected": -0.041015625,
+      "logps/chosen": -1264.0,
+      "logps/rejected": -2432.0,
+      "loss": 0.0058,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -10.5625,
-      "rewards/margins": 10.25,
-      "rewards/rejected": -20.875,
+      "rewards/chosen": -8.9375,
+      "rewards/margins": 10.0625,
+      "rewards/rejected": -19.0,
       "step": 2570
     },
     {
       "epoch": 2.701570680628272,
-      "grad_norm": 0.15309507992376212,
+      "grad_norm": 0.8298323257931932,
       "learning_rate": 1.492670085332884e-06,
-      "logits/chosen": -3.921875,
-      "logits/rejected": -1.8671875,
-      "logps/chosen": -1544.0,
-      "logps/rejected": -2960.0,
-      "loss": 0.0093,
+      "logits/chosen": -3.0,
+      "logits/rejected": -1.546875,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2816.0,
+      "loss": 0.0109,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 11.5,
-      "rewards/rejected": -23.0,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 11.6875,
+      "rewards/rejected": -21.5,
       "step": 2580
     },
     {
       "epoch": 2.712041884816754,
-      "grad_norm": 0.11211247878950366,
+      "grad_norm": 0.0761633488626917,
       "learning_rate": 1.3907243345910787e-06,
-      "logits/chosen": -3.640625,
-      "logits/rejected": -1.375,
-      "logps/chosen": -1592.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0134,
+      "logits/chosen": -2.4375,
+      "logits/rejected": -1.046875,
+      "logps/chosen": -1416.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.0103,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.5625,
-      "rewards/margins": 11.1875,
-      "rewards/rejected": -22.75,
+      "rewards/chosen": -9.75,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -20.75,
       "step": 2590
     },
     {
       "epoch": 2.7225130890052354,
-      "grad_norm": 0.0878471738085003,
+      "grad_norm": 0.5202855684948315,
       "learning_rate": 1.292284578167538e-06,
-      "logits/chosen": -3.140625,
-      "logits/rejected": -1.3671875,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0075,
-      "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 11.1875,
-      "rewards/rejected": -22.625,
+      "logits/chosen": -2.21875,
+      "logits/rejected": -1.0390625,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2688.0,
+      "loss": 0.0097,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.875,
+      "rewards/margins": 11.0625,
+      "rewards/rejected": -20.875,
       "step": 2600
     },
     {
       "epoch": 2.7225130890052354,
-      "eval_logits/chosen": -0.40625,
-      "eval_logits/rejected": 1.359375,
-      "eval_logps/chosen": -1792.0,
-      "eval_logps/rejected": -1992.0,
-      "eval_loss": 1.8008984327316284,
-      "eval_rewards/accuracies": 0.630859375,
-      "eval_rewards/chosen": -14.75,
-      "eval_rewards/margins": 2.265625,
-      "eval_rewards/rejected": -17.0,
-      "eval_runtime": 47.0873,
-      "eval_samples_per_second": 42.474,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": 0.14453125,
+      "eval_logits/rejected": 1.703125,
+      "eval_logps/chosen": -1616.0,
+      "eval_logps/rejected": -1792.0,
+      "eval_loss": 1.690609335899353,
+      "eval_rewards/accuracies": 0.623046875,
+      "eval_rewards/chosen": -12.9375,
+      "eval_rewards/margins": 2.109375,
+      "eval_rewards/rejected": -15.0625,
+      "eval_runtime": 46.5379,
+      "eval_samples_per_second": 42.976,
+      "eval_steps_per_second": 0.688,
       "step": 2600
     },
     {
       "epoch": 2.7329842931937174,
-      "grad_norm": 0.07370534954517018,
+      "grad_norm": 0.14241808037264714,
       "learning_rate": 1.1973654344364925e-06,
-      "logits/chosen": -3.171875,
-      "logits/rejected": -1.953125,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2896.0,
-      "loss": 0.0086,
+      "logits/chosen": -2.328125,
+      "logits/rejected": -1.5,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2688.0,
+      "loss": 0.009,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.1875,
-      "rewards/margins": 11.375,
-      "rewards/rejected": -22.5,
+      "rewards/chosen": -9.5,
+      "rewards/margins": 11.0,
+      "rewards/rejected": -20.5,
       "step": 2610
     },
     {
       "epoch": 2.743455497382199,
-      "grad_norm": 0.3365025689074477,
+      "grad_norm": 0.15094842631741057,
       "learning_rate": 1.1059809989586583e-06,
-      "logits/chosen": -3.046875,
-      "logits/rejected": -0.94921875,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -3008.0,
-      "loss": 0.0093,
+      "logits/chosen": -2.21875,
+      "logits/rejected": -0.609375,
+      "logps/chosen": -1384.0,
+      "logps/rejected": -2832.0,
+      "loss": 0.0089,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 12.125,
-      "rewards/rejected": -23.5,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 11.875,
+      "rewards/rejected": -21.75,
       "step": 2620
     },
     {
       "epoch": 2.7539267015706805,
-      "grad_norm": 0.08829671354669028,
+      "grad_norm": 0.11733894310713607,
       "learning_rate": 1.018144842388044e-06,
-      "logits/chosen": -2.46875,
-      "logits/rejected": -0.255859375,
-      "logps/chosen": -1528.0,
-      "logps/rejected": -2848.0,
+      "logits/chosen": -1.3671875,
+      "logits/rejected": 0.0206298828125,
+      "logps/chosen": -1344.0,
+      "logps/rejected": -2608.0,
       "loss": 0.0105,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 11.375,
-      "rewards/rejected": -22.375,
+      "rewards/chosen": -9.1875,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -20.0,
       "step": 2630
     },
     {
       "epoch": 2.7643979057591626,
-      "grad_norm": 0.1732226483399812,
+      "grad_norm": 0.1584391069723244,
       "learning_rate": 9.338700084567109e-07,
-      "logits/chosen": -3.109375,
-      "logits/rejected": -1.4765625,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0078,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.0625,
-      "rewards/margins": 11.6875,
-      "rewards/rejected": -22.75,
+      "logits/chosen": -2.296875,
+      "logits/rejected": -1.2109375,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0075,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -9.25,
+      "rewards/margins": 11.5,
+      "rewards/rejected": -20.75,
       "step": 2640
     },
     {
       "epoch": 2.774869109947644,
-      "grad_norm": 0.21408267013956153,
+      "grad_norm": 1.2400041033000138,
       "learning_rate": 8.531690120377606e-07,
-      "logits/chosen": -2.96875,
-      "logits/rejected": -1.453125,
-      "logps/chosen": -1728.0,
-      "logps/rejected": -3024.0,
-      "loss": 0.0122,
+      "logits/chosen": -2.40625,
+      "logits/rejected": -1.3125,
+      "logps/chosen": -1488.0,
+      "logps/rejected": -2784.0,
+      "loss": 0.0094,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -12.875,
-      "rewards/margins": 11.375,
-      "rewards/rejected": -24.25,
+      "rewards/chosen": -10.4375,
+      "rewards/margins": 11.25,
+      "rewards/rejected": -21.75,
       "step": 2650
     },
     {
       "epoch": 2.7853403141361257,
-      "grad_norm": 0.12461493186298428,
+      "grad_norm": 0.20826158151499208,
       "learning_rate": 7.760538372868636e-07,
-      "logits/chosen": -3.3125,
-      "logits/rejected": -1.4453125,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2832.0,
-      "loss": 0.0137,
+      "logits/chosen": -2.4375,
+      "logits/rejected": -1.1171875,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2640.0,
+      "loss": 0.0127,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 11.5,
-      "rewards/rejected": -22.875,
+      "rewards/chosen": -9.6875,
+      "rewards/margins": 11.1875,
+      "rewards/rejected": -20.875,
       "step": 2660
     },
     {
       "epoch": 2.7958115183246073,
-      "grad_norm": 0.08539447462187891,
+      "grad_norm": 1.5685191352602552,
       "learning_rate": 7.025359358626166e-07,
-      "logits/chosen": -3.34375,
-      "logits/rejected": -0.84765625,
-      "logps/chosen": -1648.0,
-      "logps/rejected": -2944.0,
-      "loss": 0.0076,
+      "logits/chosen": -2.515625,
+      "logits/rejected": -0.478515625,
+      "logps/chosen": -1496.0,
+      "logps/rejected": -2704.0,
+      "loss": 0.0088,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -12.0,
-      "rewards/margins": 11.75,
-      "rewards/rejected": -23.75,
+      "rewards/chosen": -10.5,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -21.375,
       "step": 2670
     },
     {
       "epoch": 2.806282722513089,
-      "grad_norm": 0.03546327245127398,
+      "grad_norm": 0.1402645958227768,
       "learning_rate": 6.326262252259463e-07,
-      "logits/chosen": -3.15625,
-      "logits/rejected": -0.90234375,
-      "logps/chosen": -1592.0,
-      "logps/rejected": -3008.0,
+      "logits/chosen": -2.453125,
+      "logits/rejected": -0.6171875,
+      "logps/chosen": -1408.0,
+      "logps/rejected": -2784.0,
       "loss": 0.0098,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.6875,
-      "rewards/margins": 12.125,
-      "rewards/rejected": -23.875,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 11.875,
+      "rewards/rejected": -21.625,
       "step": 2680
     },
     {
       "epoch": 2.816753926701571,
-      "grad_norm": 0.22716375235408476,
+      "grad_norm": 0.4578046756623097,
       "learning_rate": 5.66335087018871e-07,
-      "logits/chosen": -3.25,
-      "logits/rejected": -1.6484375,
-      "logps/chosen": -1584.0,
-      "logps/rejected": -3024.0,
-      "loss": 0.0076,
+      "logits/chosen": -2.515625,
+      "logits/rejected": -1.4375,
+      "logps/chosen": -1392.0,
+      "logps/rejected": -2816.0,
+      "loss": 0.0099,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.625,
-      "rewards/margins": 11.6875,
-      "rewards/rejected": -23.25,
+      "rewards/chosen": -9.6875,
+      "rewards/margins": 11.5625,
+      "rewards/rejected": -21.25,
       "step": 2690
     },
     {
       "epoch": 2.8272251308900525,
-      "grad_norm": 0.5299507901285828,
+      "grad_norm": 1.3556804362869708,
       "learning_rate": 5.036723655228226e-07,
-      "logits/chosen": -2.765625,
-      "logits/rejected": -1.3046875,
-      "logps/chosen": -1512.0,
-      "logps/rejected": -2880.0,
-      "loss": 0.0142,
+      "logits/chosen": -1.953125,
+      "logits/rejected": -0.984375,
+      "logps/chosen": -1352.0,
+      "logps/rejected": -2656.0,
+      "loss": 0.0149,
       "rewards/accuracies": 0.987500011920929,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 11.625,
-      "rewards/rejected": -23.0,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 10.9375,
+      "rewards/rejected": -20.75,
       "step": 2700
     },
     {
       "epoch": 2.8272251308900525,
-      "eval_logits/chosen": -0.302734375,
-      "eval_logits/rejected": 1.4453125,
-      "eval_logps/chosen": -1816.0,
-      "eval_logps/rejected": -2016.0,
-      "eval_loss": 1.8148125410079956,
-      "eval_rewards/accuracies": 0.630859375,
-      "eval_rewards/chosen": -14.9375,
-      "eval_rewards/margins": 2.296875,
-      "eval_rewards/rejected": -17.25,
-      "eval_runtime": 47.0841,
-      "eval_samples_per_second": 42.477,
-      "eval_steps_per_second": 0.68,
+      "eval_logits/chosen": 0.18359375,
+      "eval_logits/rejected": 1.734375,
+      "eval_logps/chosen": -1632.0,
+      "eval_logps/rejected": -1808.0,
+      "eval_loss": 1.7064374685287476,
+      "eval_rewards/accuracies": 0.62890625,
+      "eval_rewards/chosen": -13.125,
+      "eval_rewards/margins": 2.109375,
+      "eval_rewards/rejected": -15.25,
+      "eval_runtime": 46.5425,
+      "eval_samples_per_second": 42.971,
+      "eval_steps_per_second": 0.688,
       "step": 2700
     },
     {
       "epoch": 2.837696335078534,
-      "grad_norm": 1.7591547555964455,
+      "grad_norm": 0.4125675835147402,
       "learning_rate": 4.446473661967432e-07,
-      "logits/chosen": -3.625,
-      "logits/rejected": -1.8671875,
-      "logps/chosen": -1592.0,
-      "logps/rejected": -2800.0,
-      "loss": 0.0123,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.9375,
-      "rewards/margins": 10.6875,
-      "rewards/rejected": -22.625,
+      "logits/chosen": -2.703125,
+      "logits/rejected": -1.453125,
+      "logps/chosen": -1376.0,
+      "logps/rejected": -2608.0,
+      "loss": 0.0121,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -20.625,
       "step": 2710
     },
     {
       "epoch": 2.8481675392670156,
-      "grad_norm": 1.0528192304485011,
+      "grad_norm": 0.4787121699779127,
       "learning_rate": 3.892688542952411e-07,
-      "logits/chosen": -2.5,
-      "logits/rejected": -0.67578125,
-      "logps/chosen": -1464.0,
-      "logps/rejected": -2816.0,
-      "loss": 0.0128,
+      "logits/chosen": -1.6171875,
+      "logits/rejected": -0.40625,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2640.0,
+      "loss": 0.0116,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -10.8125,
-      "rewards/margins": 11.5,
-      "rewards/rejected": -22.25,
+      "rewards/chosen": -9.25,
+      "rewards/margins": 11.375,
+      "rewards/rejected": -20.625,
       "step": 2720
     },
     {
       "epoch": 2.858638743455497,
-      "grad_norm": 0.21803424026895357,
+      "grad_norm": 1.9039112982993742,
       "learning_rate": 3.3754505356694e-07,
-      "logits/chosen": -3.984375,
-      "logits/rejected": -1.90625,
-      "logps/chosen": -1608.0,
-      "logps/rejected": -2800.0,
-      "loss": 0.0136,
+      "logits/chosen": -3.0625,
+      "logits/rejected": -1.5,
+      "logps/chosen": -1472.0,
+      "logps/rejected": -2608.0,
+      "loss": 0.013,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.875,
-      "rewards/margins": 10.4375,
-      "rewards/rejected": -22.25,
+      "rewards/chosen": -10.5,
+      "rewards/margins": 9.8125,
+      "rewards/rejected": -20.375,
       "step": 2730
     },
     {
       "epoch": 2.869109947643979,
-      "grad_norm": 0.1258121861832878,
+      "grad_norm": 0.6053495752014403,
       "learning_rate": 2.8948364503322276e-07,
-      "logits/chosen": -2.75,
-      "logits/rejected": -0.8359375,
-      "logps/chosen": -1504.0,
-      "logps/rejected": -2752.0,
-      "loss": 0.0123,
+      "logits/chosen": -1.9921875,
+      "logits/rejected": -0.546875,
+      "logps/chosen": -1312.0,
+      "logps/rejected": -2560.0,
+      "loss": 0.0115,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.3125,
-      "rewards/margins": 11.125,
-      "rewards/rejected": -22.375,
+      "rewards/chosen": -9.375,
+      "rewards/margins": 11.0,
+      "rewards/rejected": -20.375,
       "step": 2740
     },
     {
       "epoch": 2.8795811518324608,
-      "grad_norm": 0.5767318390725228,
+      "grad_norm": 0.08360094814884637,
       "learning_rate": 2.4509176584762706e-07,
-      "logits/chosen": -2.984375,
-      "logits/rejected": -0.98046875,
-      "logps/chosen": -1488.0,
-      "logps/rejected": -2752.0,
-      "loss": 0.0097,
-      "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.0,
-      "rewards/margins": 10.625,
-      "rewards/rejected": -21.5,
+      "logits/chosen": -1.9375,
+      "logits/rejected": -0.703125,
+      "logps/chosen": -1320.0,
+      "logps/rejected": -2560.0,
+      "loss": 0.0115,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -9.3125,
+      "rewards/margins": 10.375,
+      "rewards/rejected": -19.75,
       "step": 2750
     },
     {
       "epoch": 2.8900523560209423,
-      "grad_norm": 0.09425982935238154,
+      "grad_norm": 0.6088919310408218,
       "learning_rate": 2.0437600823595694e-07,
-      "logits/chosen": -3.484375,
-      "logits/rejected": -1.40625,
-      "logps/chosen": -1592.0,
-      "logps/rejected": -2880.0,
-      "loss": 0.0105,
-      "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.5,
-      "rewards/margins": 11.0,
-      "rewards/rejected": -22.5,
+      "logits/chosen": -2.59375,
+      "logits/rejected": -1.078125,
+      "logps/chosen": -1424.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.0098,
+      "rewards/accuracies": 1.0,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 10.5625,
+      "rewards/rejected": -20.375,
       "step": 2760
     },
     {
       "epoch": 2.900523560209424,
-      "grad_norm": 0.314688452433757,
+      "grad_norm": 0.40022001701622,
       "learning_rate": 1.6734241851733e-07,
-      "logits/chosen": -3.390625,
-      "logits/rejected": -1.53125,
-      "logps/chosen": -1672.0,
-      "logps/rejected": -2992.0,
-      "loss": 0.0092,
+      "logits/chosen": -2.640625,
+      "logits/rejected": -1.296875,
+      "logps/chosen": -1504.0,
+      "logps/rejected": -2800.0,
+      "loss": 0.0074,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -12.375,
-      "rewards/margins": 11.4375,
-      "rewards/rejected": -23.75,
+      "rewards/chosen": -10.625,
+      "rewards/margins": 11.25,
+      "rewards/rejected": -21.875,
       "step": 2770
     },
     {
       "epoch": 2.9109947643979055,
-      "grad_norm": 1.533065818931154,
+      "grad_norm": 0.17414845517895167,
       "learning_rate": 1.339964962062984e-07,
-      "logits/chosen": -3.28125,
-      "logits/rejected": -2.015625,
-      "logps/chosen": -1512.0,
-      "logps/rejected": -2800.0,
-      "loss": 0.0067,
+      "logits/chosen": -2.5625,
+      "logits/rejected": -1.625,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2672.0,
+      "loss": 0.0054,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 11.1875,
-      "rewards/rejected": -22.625,
+      "rewards/chosen": -9.875,
+      "rewards/margins": 11.375,
+      "rewards/rejected": -21.25,
       "step": 2780
     },
     {
       "epoch": 2.9214659685863875,
-      "grad_norm": 0.5216332775391516,
+      "grad_norm": 0.16825072185723725,
       "learning_rate": 1.0434319319617136e-07,
-      "logits/chosen": -3.234375,
-      "logits/rejected": -0.93359375,
-      "logps/chosen": -1536.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.01,
+      "logits/chosen": -2.5625,
+      "logits/rejected": -0.7578125,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2560.0,
+      "loss": 0.0102,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 10.75,
-      "rewards/rejected": -22.125,
+      "rewards/chosen": -9.75,
+      "rewards/margins": 10.625,
+      "rewards/rejected": -20.375,
       "step": 2790
     },
     {
       "epoch": 2.931937172774869,
-      "grad_norm": 0.341009303562548,
+      "grad_norm": 0.0891543916770425,
       "learning_rate": 7.838691302365363e-08,
-      "logits/chosen": -2.8125,
-      "logits/rejected": -1.0078125,
-      "logps/chosen": -1552.0,
-      "logps/rejected": -2912.0,
-      "loss": 0.0102,
+      "logits/chosen": -1.9375,
+      "logits/rejected": -0.6875,
+      "logps/chosen": -1392.0,
+      "logps/rejected": -2768.0,
+      "loss": 0.0092,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.4375,
+      "rewards/chosen": -9.9375,
       "rewards/margins": 11.5,
-      "rewards/rejected": -22.875,
+      "rewards/rejected": -21.375,
       "step": 2800
     },
     {
       "epoch": 2.931937172774869,
-      "eval_logits/chosen": -0.3125,
-      "eval_logits/rejected": 1.4375,
-      "eval_logps/chosen": -1816.0,
-      "eval_logps/rejected": -2016.0,
-      "eval_loss": 1.8180078268051147,
-      "eval_rewards/accuracies": 0.626953125,
-      "eval_rewards/chosen": -14.9375,
-      "eval_rewards/margins": 2.3125,
-      "eval_rewards/rejected": -17.25,
-      "eval_runtime": 47.0939,
-      "eval_samples_per_second": 42.468,
-      "eval_steps_per_second": 0.679,
+      "eval_logits/chosen": 0.1669921875,
+      "eval_logits/rejected": 1.71875,
+      "eval_logps/chosen": -1624.0,
+      "eval_logps/rejected": -1808.0,
+      "eval_loss": 1.7073436975479126,
+      "eval_rewards/accuracies": 0.62890625,
+      "eval_rewards/chosen": -13.0625,
+      "eval_rewards/margins": 2.109375,
+      "eval_rewards/rejected": -15.125,
+      "eval_runtime": 46.5475,
+      "eval_samples_per_second": 42.967,
+      "eval_steps_per_second": 0.687,
       "step": 2800
     },
     {
       "epoch": 2.9424083769633507,
-      "grad_norm": 0.1707304319835883,
+      "grad_norm": 0.1737058779642528,
       "learning_rate": 5.613151021490459e-08,
-      "logits/chosen": -2.640625,
-      "logits/rejected": -1.5234375,
-      "logps/chosen": -1472.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.0058,
+      "logits/chosen": -1.7265625,
+      "logits/rejected": -1.09375,
+      "logps/chosen": -1304.0,
+      "logps/rejected": -2640.0,
+      "loss": 0.0077,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.3125,
-      "rewards/margins": 11.75,
-      "rewards/rejected": -23.0,
+      "rewards/chosen": -9.625,
+      "rewards/margins": 11.0625,
+      "rewards/rejected": -20.625,
       "step": 2810
     },
     {
       "epoch": 2.9528795811518327,
-      "grad_norm": 0.5595562848919824,
+      "grad_norm": 0.07528946685652511,
       "learning_rate": 3.7580289713159965e-08,
-      "logits/chosen": -2.921875,
-      "logits/rejected": -0.72265625,
-      "logps/chosen": -1600.0,
-      "logps/rejected": -2976.0,
-      "loss": 0.0094,
+      "logits/chosen": -2.109375,
+      "logits/rejected": -0.38671875,
+      "logps/chosen": -1424.0,
+      "logps/rejected": -2720.0,
+      "loss": 0.0077,
       "rewards/accuracies": 1.0,
-      "rewards/chosen": -11.8125,
-      "rewards/margins": 11.875,
-      "rewards/rejected": -23.75,
+      "rewards/chosen": -10.0,
+      "rewards/margins": 11.125,
+      "rewards/rejected": -21.25,
       "step": 2820
     },
     {
       "epoch": 2.9633507853403143,
-      "grad_norm": 0.34613763016441956,
+      "grad_norm": 0.24425733321453835,
       "learning_rate": 2.2736006387927277e-08,
-      "logits/chosen": -3.234375,
-      "logits/rejected": -0.93359375,
-      "logps/chosen": -1576.0,
-      "logps/rejected": -2768.0,
-      "loss": 0.012,
-      "rewards/accuracies": 0.981249988079071,
-      "rewards/chosen": -11.625,
-      "rewards/margins": 10.75,
-      "rewards/rejected": -22.375,
+      "logits/chosen": -2.265625,
+      "logits/rejected": -0.609375,
+      "logps/chosen": -1432.0,
+      "logps/rejected": -2576.0,
+      "loss": 0.0074,
+      "rewards/accuracies": 0.9937499761581421,
+      "rewards/chosen": -10.125,
+      "rewards/margins": 10.25,
+      "rewards/rejected": -20.375,
       "step": 2830
     },
     {
       "epoch": 2.973821989528796,
-      "grad_norm": 1.0870109941132398,
+      "grad_norm": 0.1576221321644675,
       "learning_rate": 1.1600864625893604e-08,
-      "logits/chosen": -3.15625,
-      "logits/rejected": -1.6796875,
-      "logps/chosen": -1544.0,
-      "logps/rejected": -2976.0,
-      "loss": 0.0113,
+      "logits/chosen": -2.5,
+      "logits/rejected": -1.4296875,
+      "logps/chosen": -1368.0,
+      "logps/rejected": -2752.0,
+      "loss": 0.0137,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.375,
-      "rewards/margins": 12.0,
-      "rewards/rejected": -23.375,
+      "rewards/chosen": -9.625,
+      "rewards/margins": 11.4375,
+      "rewards/rejected": -21.0,
       "step": 2840
     },
     {
       "epoch": 2.9842931937172774,
-      "grad_norm": 0.017941993915583238,
+      "grad_norm": 0.1470090005782711,
       "learning_rate": 4.176518003573549e-09,
-      "logits/chosen": -3.703125,
-      "logits/rejected": -1.53125,
-      "logps/chosen": -1560.0,
-      "logps/rejected": -2864.0,
-      "loss": 0.016,
+      "logits/chosen": -2.671875,
+      "logits/rejected": -1.1953125,
+      "logps/chosen": -1392.0,
+      "logps/rejected": -2624.0,
+      "loss": 0.0199,
       "rewards/accuracies": 0.9937499761581421,
-      "rewards/chosen": -11.625,
-      "rewards/margins": 11.375,
-      "rewards/rejected": -23.0,
+      "rewards/chosen": -9.9375,
+      "rewards/margins": 10.8125,
+      "rewards/rejected": -20.75,
       "step": 2850
     },
     {
       "epoch": 2.994764397905759,
-      "grad_norm": 0.11366211393856238,
+      "grad_norm": 0.41792938060583534,
       "learning_rate": 4.640690417528903e-10,
-      "logits/chosen": -2.984375,
-      "logits/rejected": -1.484375,
-      "logps/chosen": -1520.0,
-      "logps/rejected": -2736.0,
-      "loss": 0.0163,
+      "logits/chosen": -2.140625,
+      "logits/rejected": -1.15625,
+      "logps/chosen": -1360.0,
+      "logps/rejected": -2528.0,
+      "loss": 0.0184,
       "rewards/accuracies": 0.9750000238418579,
-      "rewards/chosen": -11.4375,
-      "rewards/margins": 10.8125,
-      "rewards/rejected": -22.25,
+      "rewards/chosen": -9.8125,
+      "rewards/margins": 10.4375,
+      "rewards/rejected": -20.25,
       "step": 2860
     },
     {
       "epoch": 3.0,
       "step": 2865,
       "total_flos": 0.0,
-      "train_loss": 0.17683555641411486,
-      "train_runtime": 12375.7398,
-      "train_samples_per_second": 14.816,
-      "train_steps_per_second": 0.232
+      "train_loss": 0.17899337792687808,
+      "train_runtime": 12124.1231,
+      "train_samples_per_second": 15.123,
+      "train_steps_per_second": 0.236
     }
   ],
   "logging_steps": 10,