diff --git "a/trainer_state.json" "b/trainer_state.json" new file mode 100644--- /dev/null +++ "b/trainer_state.json" @@ -0,0 +1,3118 @@ +{ + "best_metric": null, + "best_model_checkpoint": null, + "epoch": 0.9997382884061764, + "eval_steps": 100, + "global_step": 1910, + "is_hyper_param_search": false, + "is_local_process_zero": true, + "is_world_process_zero": true, + "log_history": [ + { + "epoch": 0.0, + "learning_rate": 2.617801047120419e-09, + "logits/chosen": -1.1073582172393799, + "logits/rejected": -1.147336483001709, + "logps/chosen": -555.1083374023438, + "logps/rejected": -319.5801086425781, + "loss": 0.6931, + "pred_label": 0.0, + "rewards/accuracies": 0.0, + "rewards/chosen": 0.0, + "rewards/margins": 0.0, + "rewards/rejected": 0.0, + "step": 1, + "use_label": 3.0 + }, + { + "epoch": 0.01, + "learning_rate": 2.6178010471204188e-08, + "logits/chosen": -1.2926875352859497, + "logits/rejected": -1.3704100847244263, + "logps/chosen": -381.8235168457031, + "logps/rejected": -391.3815002441406, + "loss": 0.6938, + "pred_label": 0.0, + "rewards/accuracies": 0.5277777910232544, + "rewards/chosen": 0.011978763155639172, + "rewards/margins": 0.01244138553738594, + "rewards/rejected": -0.00046262191608548164, + "step": 10, + "use_label": 23.0 + }, + { + "epoch": 0.01, + "learning_rate": 5.2356020942408376e-08, + "logits/chosen": -1.3483327627182007, + "logits/rejected": -1.3216352462768555, + "logps/chosen": -363.7161560058594, + "logps/rejected": -315.9764709472656, + "loss": 0.6975, + "pred_label": 0.0, + "rewards/accuracies": 0.32499998807907104, + "rewards/chosen": -0.008383020758628845, + "rewards/margins": -0.02416672185063362, + "rewards/rejected": 0.015783701092004776, + "step": 20, + "use_label": 61.0 + }, + { + "epoch": 0.02, + "learning_rate": 7.853403141361257e-08, + "logits/chosen": -1.3223068714141846, + "logits/rejected": -1.3222100734710693, + "logps/chosen": -280.5625915527344, + "logps/rejected": -258.6246643066406, + "loss": 0.6945, + "pred_label": 0.0, + "rewards/accuracies": 0.5, + "rewards/chosen": 0.035853710025548935, + "rewards/margins": 0.010982252657413483, + "rewards/rejected": 0.024871457368135452, + "step": 30, + "use_label": 101.0 + }, + { + "epoch": 0.02, + "learning_rate": 1.0471204188481675e-07, + "logits/chosen": -1.235068678855896, + "logits/rejected": -1.1716922521591187, + "logps/chosen": -295.0016174316406, + "logps/rejected": -334.9220275878906, + "loss": 0.6876, + "pred_label": 0.0, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": 0.05318360775709152, + "rewards/margins": 0.013463905081152916, + "rewards/rejected": 0.03971971571445465, + "step": 40, + "use_label": 141.0 + }, + { + "epoch": 0.03, + "learning_rate": 1.3089005235602092e-07, + "logits/chosen": -1.2642734050750732, + "logits/rejected": -1.1366677284240723, + "logps/chosen": -365.61114501953125, + "logps/rejected": -289.4996032714844, + "loss": 0.6816, + "pred_label": 0.0, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.07207981497049332, + "rewards/margins": 0.05135401338338852, + "rewards/rejected": 0.020725790411233902, + "step": 50, + "use_label": 181.0 + }, + { + "epoch": 0.03, + "learning_rate": 1.5706806282722514e-07, + "logits/chosen": -1.2177165746688843, + "logits/rejected": -1.230103611946106, + "logps/chosen": -320.33563232421875, + "logps/rejected": -297.27508544921875, + "loss": 0.6684, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.10490015894174576, + "rewards/margins": 0.06382963806390762, + "rewards/rejected": 0.041070520877838135, + "step": 60, + "use_label": 221.0 + }, + { + "epoch": 0.04, + "learning_rate": 1.8324607329842932e-07, + "logits/chosen": -1.3265694379806519, + "logits/rejected": -1.228366732597351, + "logps/chosen": -353.4063720703125, + "logps/rejected": -329.4417724609375, + "loss": 0.6629, + "pred_label": 0.0, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.29588761925697327, + "rewards/margins": 0.1510847806930542, + "rewards/rejected": 0.14480280876159668, + "step": 70, + "use_label": 261.0 + }, + { + "epoch": 0.04, + "learning_rate": 2.094240837696335e-07, + "logits/chosen": -1.1528517007827759, + "logits/rejected": -1.1881415843963623, + "logps/chosen": -337.8947448730469, + "logps/rejected": -304.3246765136719, + "loss": 0.6595, + "pred_label": 0.0, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.24385130405426025, + "rewards/margins": 0.10451836884021759, + "rewards/rejected": 0.13933296501636505, + "step": 80, + "use_label": 301.0 + }, + { + "epoch": 0.05, + "learning_rate": 2.356020942408377e-07, + "logits/chosen": -1.2590320110321045, + "logits/rejected": -1.3250510692596436, + "logps/chosen": -300.1564025878906, + "logps/rejected": -253.16357421875, + "loss": 0.6333, + "pred_label": 0.4000000059604645, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.25413432717323303, + "rewards/margins": 0.1232389584183693, + "rewards/rejected": 0.13089534640312195, + "step": 90, + "use_label": 340.6000061035156 + }, + { + "epoch": 0.05, + "learning_rate": 2.6178010471204185e-07, + "logits/chosen": -1.2287836074829102, + "logits/rejected": -1.2516543865203857, + "logps/chosen": -284.37042236328125, + "logps/rejected": -261.1955261230469, + "loss": 0.6389, + "pred_label": 1.0, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.3382905125617981, + "rewards/margins": 0.1669512689113617, + "rewards/rejected": 0.17133919894695282, + "step": 100, + "use_label": 380.0 + }, + { + "epoch": 0.06, + "learning_rate": 2.879581151832461e-07, + "logits/chosen": -1.3341495990753174, + "logits/rejected": -1.3016605377197266, + "logps/chosen": -292.0628662109375, + "logps/rejected": -291.2762756347656, + "loss": 0.6436, + "pred_label": 1.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.34865301847457886, + "rewards/margins": 0.18116387724876404, + "rewards/rejected": 0.16748914122581482, + "step": 110, + "use_label": 420.0 + }, + { + "epoch": 0.06, + "learning_rate": 3.1413612565445027e-07, + "logits/chosen": -1.2554535865783691, + "logits/rejected": -1.2652862071990967, + "logps/chosen": -344.93731689453125, + "logps/rejected": -276.16986083984375, + "loss": 0.6143, + "pred_label": 1.2000000476837158, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.5013664364814758, + "rewards/margins": 0.28251418471336365, + "rewards/rejected": 0.21885231137275696, + "step": 120, + "use_label": 459.79998779296875 + }, + { + "epoch": 0.07, + "learning_rate": 3.4031413612565446e-07, + "logits/chosen": -1.3496097326278687, + "logits/rejected": -1.2879693508148193, + "logps/chosen": -386.6773376464844, + "logps/rejected": -273.1239013671875, + "loss": 0.5905, + "pred_label": 2.75, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.6433510184288025, + "rewards/margins": 0.37278619408607483, + "rewards/rejected": 0.27056482434272766, + "step": 130, + "use_label": 498.25 + }, + { + "epoch": 0.07, + "learning_rate": 3.6649214659685864e-07, + "logits/chosen": -1.3663859367370605, + "logits/rejected": -1.3222754001617432, + "logps/chosen": -272.4532775878906, + "logps/rejected": -275.90032958984375, + "loss": 0.5982, + "pred_label": 4.849999904632568, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.5143588185310364, + "rewards/margins": 0.23399925231933594, + "rewards/rejected": 0.28035953640937805, + "step": 140, + "use_label": 536.1500244140625 + }, + { + "epoch": 0.08, + "learning_rate": 3.926701570680628e-07, + "logits/chosen": -1.3226642608642578, + "logits/rejected": -1.2861449718475342, + "logps/chosen": -351.81878662109375, + "logps/rejected": -261.70208740234375, + "loss": 0.5397, + "pred_label": 10.800000190734863, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.6599913835525513, + "rewards/margins": 0.4658470153808594, + "rewards/rejected": 0.1941443383693695, + "step": 150, + "use_label": 570.2000122070312 + }, + { + "epoch": 0.08, + "learning_rate": 4.18848167539267e-07, + "logits/chosen": -1.1659371852874756, + "logits/rejected": -1.1296656131744385, + "logps/chosen": -412.0987243652344, + "logps/rejected": -323.6183776855469, + "loss": 0.5415, + "pred_label": 16.200000762939453, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.8539795875549316, + "rewards/margins": 0.553295373916626, + "rewards/rejected": 0.3006841838359833, + "step": 160, + "use_label": 604.7999877929688 + }, + { + "epoch": 0.09, + "learning_rate": 4.450261780104712e-07, + "logits/chosen": -1.2072548866271973, + "logits/rejected": -1.2403260469436646, + "logps/chosen": -385.46820068359375, + "logps/rejected": -320.39019775390625, + "loss": 0.5418, + "pred_label": 25.149999618530273, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.9536998867988586, + "rewards/margins": 0.5978282690048218, + "rewards/rejected": 0.35587143898010254, + "step": 170, + "use_label": 635.8499755859375 + }, + { + "epoch": 0.09, + "learning_rate": 4.712041884816754e-07, + "logits/chosen": -1.2297383546829224, + "logits/rejected": -1.258941888809204, + "logps/chosen": -341.0736999511719, + "logps/rejected": -295.0050964355469, + "loss": 0.5171, + "pred_label": 38.04999923706055, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 1.2062633037567139, + "rewards/margins": 0.8106303215026855, + "rewards/rejected": 0.3956330120563507, + "step": 180, + "use_label": 662.9500122070312 + }, + { + "epoch": 0.1, + "learning_rate": 4.973821989528796e-07, + "logits/chosen": -1.339400053024292, + "logits/rejected": -1.315095067024231, + "logps/chosen": -286.23583984375, + "logps/rejected": -274.06976318359375, + "loss": 0.5214, + "pred_label": 47.400001525878906, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.115768313407898, + "rewards/margins": 0.642764687538147, + "rewards/rejected": 0.4730035662651062, + "step": 190, + "use_label": 693.5999755859375 + }, + { + "epoch": 0.1, + "learning_rate": 4.973821989528796e-07, + "logits/chosen": -1.2894158363342285, + "logits/rejected": -1.3884027004241943, + "logps/chosen": -376.9474182128906, + "logps/rejected": -412.8877868652344, + "loss": 0.537, + "pred_label": 55.599998474121094, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 1.0884490013122559, + "rewards/margins": 0.6527363061904907, + "rewards/rejected": 0.4357127249240875, + "step": 200, + "use_label": 725.4000244140625 + }, + { + "epoch": 0.11, + "learning_rate": 4.944735311227458e-07, + "logits/chosen": -1.2643864154815674, + "logits/rejected": -1.2973284721374512, + "logps/chosen": -277.7148132324219, + "logps/rejected": -278.7284240722656, + "loss": 0.5029, + "pred_label": 67.3499984741211, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.057620644569397, + "rewards/margins": 0.7939804792404175, + "rewards/rejected": 0.26364022493362427, + "step": 210, + "use_label": 753.6500244140625 + }, + { + "epoch": 0.12, + "learning_rate": 4.91564863292612e-07, + "logits/chosen": -1.2208797931671143, + "logits/rejected": -1.216244101524353, + "logps/chosen": -309.74603271484375, + "logps/rejected": -312.56842041015625, + "loss": 0.4718, + "pred_label": 77.94999694824219, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.0285298824310303, + "rewards/margins": 0.5832255482673645, + "rewards/rejected": 0.4453042447566986, + "step": 220, + "use_label": 783.0499877929688 + }, + { + "epoch": 0.12, + "learning_rate": 4.886561954624782e-07, + "logits/chosen": -1.2997448444366455, + "logits/rejected": -1.270655870437622, + "logps/chosen": -440.33111572265625, + "logps/rejected": -398.54632568359375, + "loss": 0.4994, + "pred_label": 90.4000015258789, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.3546545505523682, + "rewards/margins": 1.087874174118042, + "rewards/rejected": 0.2667803168296814, + "step": 230, + "use_label": 810.5999755859375 + }, + { + "epoch": 0.13, + "learning_rate": 4.857475276323444e-07, + "logits/chosen": -1.338390588760376, + "logits/rejected": -1.2634680271148682, + "logps/chosen": -327.8422546386719, + "logps/rejected": -232.891845703125, + "loss": 0.4924, + "pred_label": 103.1500015258789, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 1.0152827501296997, + "rewards/margins": 1.013619303703308, + "rewards/rejected": 0.0016636044019833207, + "step": 240, + "use_label": 837.8499755859375 + }, + { + "epoch": 0.13, + "learning_rate": 4.828388598022106e-07, + "logits/chosen": -1.3745036125183105, + "logits/rejected": -1.362729787826538, + "logps/chosen": -346.9915771484375, + "logps/rejected": -278.84124755859375, + "loss": 0.4861, + "pred_label": 118.25, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.2634034156799316, + "rewards/margins": 0.8081910014152527, + "rewards/rejected": 0.4552123546600342, + "step": 250, + "use_label": 862.75 + }, + { + "epoch": 0.14, + "learning_rate": 4.799301919720768e-07, + "logits/chosen": -1.2410266399383545, + "logits/rejected": -1.1942863464355469, + "logps/chosen": -369.42156982421875, + "logps/rejected": -283.9178771972656, + "loss": 0.4563, + "pred_label": 129.89999389648438, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.2563164234161377, + "rewards/margins": 0.9190329313278198, + "rewards/rejected": 0.3372834324836731, + "step": 260, + "use_label": 891.0999755859375 + }, + { + "epoch": 0.14, + "learning_rate": 4.77021524141943e-07, + "logits/chosen": -1.3895528316497803, + "logits/rejected": -1.3751407861709595, + "logps/chosen": -268.8004455566406, + "logps/rejected": -284.19122314453125, + "loss": 0.4572, + "pred_label": 143.9499969482422, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.2117127180099487, + "rewards/margins": 0.6257116794586182, + "rewards/rejected": 0.5860010385513306, + "step": 270, + "use_label": 917.0499877929688 + }, + { + "epoch": 0.15, + "learning_rate": 4.7411285631180917e-07, + "logits/chosen": -1.3496041297912598, + "logits/rejected": -1.374954104423523, + "logps/chosen": -324.27777099609375, + "logps/rejected": -279.5069580078125, + "loss": 0.4693, + "pred_label": 153.35000610351562, + "rewards/accuracies": 0.5249999761581421, + "rewards/chosen": 1.1455862522125244, + "rewards/margins": 0.4547230303287506, + "rewards/rejected": 0.6908631920814514, + "step": 280, + "use_label": 947.6500244140625 + }, + { + "epoch": 0.15, + "learning_rate": 4.712041884816754e-07, + "logits/chosen": -1.3121898174285889, + "logits/rejected": -1.33925199508667, + "logps/chosen": -279.45233154296875, + "logps/rejected": -333.3996276855469, + "loss": 0.4481, + "pred_label": 166.14999389648438, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.7546733617782593, + "rewards/margins": 0.7626081705093384, + "rewards/rejected": -0.007934845983982086, + "step": 290, + "use_label": 974.8499755859375 + }, + { + "epoch": 0.16, + "learning_rate": 4.682955206515416e-07, + "logits/chosen": -1.2102586030960083, + "logits/rejected": -1.2808181047439575, + "logps/chosen": -326.1608581542969, + "logps/rejected": -304.8093566894531, + "loss": 0.4718, + "pred_label": 179.10000610351562, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.6026958227157593, + "rewards/margins": 0.38519367575645447, + "rewards/rejected": 0.217502161860466, + "step": 300, + "use_label": 1001.9000244140625 + }, + { + "epoch": 0.16, + "learning_rate": 4.653868528214078e-07, + "logits/chosen": -1.1727502346038818, + "logits/rejected": -1.2067267894744873, + "logps/chosen": -349.1585693359375, + "logps/rejected": -368.10772705078125, + "loss": 0.4232, + "pred_label": 194.14999389648438, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.496382474899292, + "rewards/margins": 1.2464112043380737, + "rewards/rejected": 0.24997124075889587, + "step": 310, + "use_label": 1026.8499755859375 + }, + { + "epoch": 0.17, + "learning_rate": 4.62478184991274e-07, + "logits/chosen": -1.2799582481384277, + "logits/rejected": -1.3597850799560547, + "logps/chosen": -314.4355773925781, + "logps/rejected": -300.247802734375, + "loss": 0.421, + "pred_label": 210.4499969482422, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.3173748254776, + "rewards/margins": 1.0775160789489746, + "rewards/rejected": 0.2398585081100464, + "step": 320, + "use_label": 1050.550048828125 + }, + { + "epoch": 0.17, + "learning_rate": 4.5956951716114016e-07, + "logits/chosen": -1.229280948638916, + "logits/rejected": -1.2974430322647095, + "logps/chosen": -345.85198974609375, + "logps/rejected": -309.1730651855469, + "loss": 0.4067, + "pred_label": 227.8000030517578, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.1288254261016846, + "rewards/margins": 0.8339816331863403, + "rewards/rejected": 0.2948438823223114, + "step": 330, + "use_label": 1073.199951171875 + }, + { + "epoch": 0.18, + "learning_rate": 4.566608493310064e-07, + "logits/chosen": -1.3431947231292725, + "logits/rejected": -1.3393170833587646, + "logps/chosen": -356.05810546875, + "logps/rejected": -269.2801818847656, + "loss": 0.3981, + "pred_label": 247.6999969482422, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.18732750415802, + "rewards/margins": 1.1738572120666504, + "rewards/rejected": 0.01347041130065918, + "step": 340, + "use_label": 1093.300048828125 + }, + { + "epoch": 0.18, + "learning_rate": 4.5375218150087257e-07, + "logits/chosen": -1.2831729650497437, + "logits/rejected": -1.2885137796401978, + "logps/chosen": -262.3739013671875, + "logps/rejected": -313.76385498046875, + "loss": 0.404, + "pred_label": 265.5, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.1653120517730713, + "rewards/margins": 0.48939722776412964, + "rewards/rejected": 0.6759146451950073, + "step": 350, + "use_label": 1115.5 + }, + { + "epoch": 0.19, + "learning_rate": 4.5084351367073883e-07, + "logits/chosen": -1.352308988571167, + "logits/rejected": -1.3754525184631348, + "logps/chosen": -239.6806182861328, + "logps/rejected": -248.5498046875, + "loss": 0.3967, + "pred_label": 285.3999938964844, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.0487511157989502, + "rewards/margins": 1.0042428970336914, + "rewards/rejected": 0.044508084654808044, + "step": 360, + "use_label": 1135.5999755859375 + }, + { + "epoch": 0.19, + "learning_rate": 4.47934845840605e-07, + "logits/chosen": -1.3563268184661865, + "logits/rejected": -1.3786267042160034, + "logps/chosen": -308.45672607421875, + "logps/rejected": -270.8168029785156, + "loss": 0.3685, + "pred_label": 303.54998779296875, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 1.2717820405960083, + "rewards/margins": 1.1417100429534912, + "rewards/rejected": 0.13007190823554993, + "step": 370, + "use_label": 1157.449951171875 + }, + { + "epoch": 0.2, + "learning_rate": 4.450261780104712e-07, + "logits/chosen": -1.325315237045288, + "logits/rejected": -1.279359221458435, + "logps/chosen": -325.87542724609375, + "logps/rejected": -285.46929931640625, + "loss": 0.3768, + "pred_label": 326.6000061035156, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 1.9515581130981445, + "rewards/margins": 1.806125283241272, + "rewards/rejected": 0.1454329490661621, + "step": 380, + "use_label": 1174.4000244140625 + }, + { + "epoch": 0.2, + "learning_rate": 4.421175101803374e-07, + "logits/chosen": -1.3155038356781006, + "logits/rejected": -1.3505948781967163, + "logps/chosen": -299.5450134277344, + "logps/rejected": -313.3736267089844, + "loss": 0.3309, + "pred_label": 345.95001220703125, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 2.0010504722595215, + "rewards/margins": 1.6885391473770142, + "rewards/rejected": 0.3125116229057312, + "step": 390, + "use_label": 1195.050048828125 + }, + { + "epoch": 0.21, + "learning_rate": 4.392088423502036e-07, + "logits/chosen": -1.305530309677124, + "logits/rejected": -1.3502506017684937, + "logps/chosen": -300.59857177734375, + "logps/rejected": -307.1338806152344, + "loss": 0.3581, + "pred_label": 362.79998779296875, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 1.3619585037231445, + "rewards/margins": 0.779948353767395, + "rewards/rejected": 0.5820102095603943, + "step": 400, + "use_label": 1218.199951171875 + }, + { + "epoch": 0.21, + "learning_rate": 4.3630017452006976e-07, + "logits/chosen": -1.2975621223449707, + "logits/rejected": -1.3306124210357666, + "logps/chosen": -328.60797119140625, + "logps/rejected": -269.7942810058594, + "loss": 0.3682, + "pred_label": 382.3999938964844, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.766815185546875, + "rewards/margins": 1.294459342956543, + "rewards/rejected": 0.47235584259033203, + "step": 410, + "use_label": 1238.5999755859375 + }, + { + "epoch": 0.22, + "learning_rate": 4.33391506689936e-07, + "logits/chosen": -1.3032572269439697, + "logits/rejected": -1.2937753200531006, + "logps/chosen": -330.87353515625, + "logps/rejected": -301.7434997558594, + "loss": 0.338, + "pred_label": 403.20001220703125, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.4692720174789429, + "rewards/margins": 1.3137176036834717, + "rewards/rejected": 0.15555439889431, + "step": 420, + "use_label": 1257.800048828125 + }, + { + "epoch": 0.23, + "learning_rate": 4.304828388598022e-07, + "logits/chosen": -1.1606519222259521, + "logits/rejected": -1.1426994800567627, + "logps/chosen": -382.469482421875, + "logps/rejected": -351.1206970214844, + "loss": 0.312, + "pred_label": 425.20001220703125, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 1.9217780828475952, + "rewards/margins": 2.1217658519744873, + "rewards/rejected": -0.19998787343502045, + "step": 430, + "use_label": 1275.800048828125 + }, + { + "epoch": 0.23, + "learning_rate": 4.2757417102966844e-07, + "logits/chosen": -1.2695534229278564, + "logits/rejected": -1.2679811716079712, + "logps/chosen": -311.1631164550781, + "logps/rejected": -303.9151611328125, + "loss": 0.3555, + "pred_label": 445.04998779296875, + "rewards/accuracies": 0.5, + "rewards/chosen": 1.351044774055481, + "rewards/margins": 0.4551563262939453, + "rewards/rejected": 0.89588862657547, + "step": 440, + "use_label": 1295.949951171875 + }, + { + "epoch": 0.24, + "learning_rate": 4.246655031995346e-07, + "logits/chosen": -1.198559284210205, + "logits/rejected": -1.194226861000061, + "logps/chosen": -286.85296630859375, + "logps/rejected": -267.08056640625, + "loss": 0.318, + "pred_label": 467.0, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 1.3747012615203857, + "rewards/margins": 2.1655492782592773, + "rewards/rejected": -0.7908478379249573, + "step": 450, + "use_label": 1314.0 + }, + { + "epoch": 0.24, + "learning_rate": 4.217568353694008e-07, + "logits/chosen": -1.3408215045928955, + "logits/rejected": -1.3759543895721436, + "logps/chosen": -340.38336181640625, + "logps/rejected": -321.1266174316406, + "loss": 0.3117, + "pred_label": 491.45001220703125, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 1.2093924283981323, + "rewards/margins": 1.9316478967666626, + "rewards/rejected": -0.7222554683685303, + "step": 460, + "use_label": 1329.550048828125 + }, + { + "epoch": 0.25, + "learning_rate": 4.18848167539267e-07, + "logits/chosen": -1.3295974731445312, + "logits/rejected": -1.3128505945205688, + "logps/chosen": -242.5327606201172, + "logps/rejected": -224.53195190429688, + "loss": 0.3026, + "pred_label": 512.9000244140625, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.9095451235771179, + "rewards/margins": 1.644636869430542, + "rewards/rejected": -0.7350918650627136, + "step": 470, + "use_label": 1348.0999755859375 + }, + { + "epoch": 0.25, + "learning_rate": 4.159394997091332e-07, + "logits/chosen": -1.3231532573699951, + "logits/rejected": -1.2729825973510742, + "logps/chosen": -264.4250183105469, + "logps/rejected": -219.93978881835938, + "loss": 0.3323, + "pred_label": 536.0, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.7578233480453491, + "rewards/margins": 1.116870403289795, + "rewards/rejected": -0.35904714465141296, + "step": 480, + "use_label": 1365.0 + }, + { + "epoch": 0.26, + "learning_rate": 4.1303083187899937e-07, + "logits/chosen": -1.126888394355774, + "logits/rejected": -1.1470086574554443, + "logps/chosen": -373.8638916015625, + "logps/rejected": -374.7835998535156, + "loss": 0.347, + "pred_label": 555.5999755859375, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.9710286855697632, + "rewards/margins": 1.8147733211517334, + "rewards/rejected": -0.8437445759773254, + "step": 490, + "use_label": 1385.4000244140625 + }, + { + "epoch": 0.26, + "learning_rate": 4.1012216404886563e-07, + "logits/chosen": -1.2007501125335693, + "logits/rejected": -1.1932474374771118, + "logps/chosen": -347.7434387207031, + "logps/rejected": -291.282958984375, + "loss": 0.3333, + "pred_label": 579.5999755859375, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.1730000972747803, + "rewards/margins": 1.3524248600006104, + "rewards/rejected": -0.17942459881305695, + "step": 500, + "use_label": 1401.4000244140625 + }, + { + "epoch": 0.27, + "learning_rate": 4.072134962187318e-07, + "logits/chosen": -1.3202276229858398, + "logits/rejected": -1.3420360088348389, + "logps/chosen": -267.11639404296875, + "logps/rejected": -283.467041015625, + "loss": 0.3203, + "pred_label": 603.7000122070312, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 1.058019995689392, + "rewards/margins": 2.189736843109131, + "rewards/rejected": -1.1317169666290283, + "step": 510, + "use_label": 1417.300048828125 + }, + { + "epoch": 0.27, + "learning_rate": 4.0430482838859805e-07, + "logits/chosen": -1.1408579349517822, + "logits/rejected": -1.2456986904144287, + "logps/chosen": -342.52593994140625, + "logps/rejected": -286.4843444824219, + "loss": 0.3375, + "pred_label": 627.4000244140625, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.5285916328430176, + "rewards/margins": 1.3080298900604248, + "rewards/rejected": -0.7794381380081177, + "step": 520, + "use_label": 1433.5999755859375 + }, + { + "epoch": 0.28, + "learning_rate": 4.013961605584642e-07, + "logits/chosen": -1.2992430925369263, + "logits/rejected": -1.3081134557724, + "logps/chosen": -313.76641845703125, + "logps/rejected": -381.96466064453125, + "loss": 0.2815, + "pred_label": 652.2000122070312, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.136601448059082, + "rewards/margins": 2.0457780361175537, + "rewards/rejected": -0.9091763496398926, + "step": 530, + "use_label": 1448.800048828125 + }, + { + "epoch": 0.28, + "learning_rate": 3.984874927283304e-07, + "logits/chosen": -1.278599739074707, + "logits/rejected": -1.319656491279602, + "logps/chosen": -306.2652893066406, + "logps/rejected": -265.16363525390625, + "loss": 0.3163, + "pred_label": 676.4000244140625, + "rewards/accuracies": 0.75, + "rewards/chosen": 2.4270758628845215, + "rewards/margins": 2.458122730255127, + "rewards/rejected": -0.031046796590089798, + "step": 540, + "use_label": 1464.5999755859375 + }, + { + "epoch": 0.29, + "learning_rate": 3.955788248981966e-07, + "logits/chosen": -1.2827562093734741, + "logits/rejected": -1.1873165369033813, + "logps/chosen": -303.70391845703125, + "logps/rejected": -265.5109558105469, + "loss": 0.3474, + "pred_label": 698.1500244140625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.532619595527649, + "rewards/margins": 2.0600717067718506, + "rewards/rejected": -0.5274521708488464, + "step": 550, + "use_label": 1482.8499755859375 + }, + { + "epoch": 0.29, + "learning_rate": 3.926701570680628e-07, + "logits/chosen": -1.3157541751861572, + "logits/rejected": -1.250253677368164, + "logps/chosen": -301.3360595703125, + "logps/rejected": -319.97308349609375, + "loss": 0.3382, + "pred_label": 726.5, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 2.103365182876587, + "rewards/margins": 1.9628784656524658, + "rewards/rejected": 0.14048661291599274, + "step": 560, + "use_label": 1494.5 + }, + { + "epoch": 0.3, + "learning_rate": 3.8976148923792903e-07, + "logits/chosen": -1.376778483390808, + "logits/rejected": -1.3120372295379639, + "logps/chosen": -333.80181884765625, + "logps/rejected": -280.1228332519531, + "loss": 0.3608, + "pred_label": 747.1500244140625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.266614317893982, + "rewards/margins": 1.6077734231948853, + "rewards/rejected": -0.34115907549858093, + "step": 570, + "use_label": 1513.8499755859375 + }, + { + "epoch": 0.3, + "learning_rate": 3.8685282140779524e-07, + "logits/chosen": -1.3779680728912354, + "logits/rejected": -1.3599121570587158, + "logps/chosen": -310.0410461425781, + "logps/rejected": -264.19580078125, + "loss": 0.2868, + "pred_label": 763.1500244140625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 2.130375385284424, + "rewards/margins": 2.0123019218444824, + "rewards/rejected": 0.11807334423065186, + "step": 580, + "use_label": 1537.8499755859375 + }, + { + "epoch": 0.31, + "learning_rate": 3.839441535776614e-07, + "logits/chosen": -1.3360798358917236, + "logits/rejected": -1.3371288776397705, + "logps/chosen": -339.41485595703125, + "logps/rejected": -361.7887268066406, + "loss": 0.3098, + "pred_label": 787.0, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 1.6782824993133545, + "rewards/margins": 1.6973320245742798, + "rewards/rejected": -0.01904923841357231, + "step": 590, + "use_label": 1554.0 + }, + { + "epoch": 0.31, + "learning_rate": 3.8103548574752766e-07, + "logits/chosen": -1.2587051391601562, + "logits/rejected": -1.1522762775421143, + "logps/chosen": -296.5978698730469, + "logps/rejected": -276.82574462890625, + "loss": 0.2843, + "pred_label": 815.4000244140625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 1.5922435522079468, + "rewards/margins": 1.7545421123504639, + "rewards/rejected": -0.16229870915412903, + "step": 600, + "use_label": 1565.5999755859375 + }, + { + "epoch": 0.32, + "learning_rate": 3.781268179173938e-07, + "logits/chosen": -1.2215590476989746, + "logits/rejected": -1.2082473039627075, + "logps/chosen": -304.05950927734375, + "logps/rejected": -402.14996337890625, + "loss": 0.2557, + "pred_label": 842.2999877929688, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.8012117147445679, + "rewards/margins": 2.9960241317749023, + "rewards/rejected": -2.194812774658203, + "step": 610, + "use_label": 1578.699951171875 + }, + { + "epoch": 0.32, + "learning_rate": 3.7521815008726e-07, + "logits/chosen": -1.247642993927002, + "logits/rejected": -1.2035638093948364, + "logps/chosen": -315.6612854003906, + "logps/rejected": -302.01751708984375, + "loss": 0.2351, + "pred_label": 869.2999877929688, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.4792921543121338, + "rewards/margins": 2.9789254665374756, + "rewards/rejected": -1.4996334314346313, + "step": 620, + "use_label": 1591.699951171875 + }, + { + "epoch": 0.33, + "learning_rate": 3.723094822571262e-07, + "logits/chosen": -1.1842483282089233, + "logits/rejected": -1.247299313545227, + "logps/chosen": -277.7129211425781, + "logps/rejected": -321.3715515136719, + "loss": 0.2968, + "pred_label": 896.5499877929688, + "rewards/accuracies": 0.574999988079071, + "rewards/chosen": 0.45529526472091675, + "rewards/margins": 1.2875747680664062, + "rewards/rejected": -0.8322793841362, + "step": 630, + "use_label": 1604.449951171875 + }, + { + "epoch": 0.33, + "learning_rate": 3.6940081442699243e-07, + "logits/chosen": -1.1754447221755981, + "logits/rejected": -1.2381727695465088, + "logps/chosen": -396.6316223144531, + "logps/rejected": -338.87164306640625, + "loss": 0.2743, + "pred_label": 923.0499877929688, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.854255199432373, + "rewards/margins": 2.8615994453430176, + "rewards/rejected": -2.0073440074920654, + "step": 640, + "use_label": 1617.949951171875 + }, + { + "epoch": 0.34, + "learning_rate": 3.6649214659685864e-07, + "logits/chosen": -1.3000792264938354, + "logits/rejected": -1.337820291519165, + "logps/chosen": -261.8930969238281, + "logps/rejected": -257.5715026855469, + "loss": 0.2734, + "pred_label": 950.7000122070312, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.9624491930007935, + "rewards/margins": 1.762574553489685, + "rewards/rejected": -0.8001253008842468, + "step": 650, + "use_label": 1630.300048828125 + }, + { + "epoch": 0.35, + "learning_rate": 3.6358347876672485e-07, + "logits/chosen": -1.2691199779510498, + "logits/rejected": -1.2646231651306152, + "logps/chosen": -337.8616943359375, + "logps/rejected": -244.37094116210938, + "loss": 0.2734, + "pred_label": 978.9500122070312, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 1.2994811534881592, + "rewards/margins": 2.3611044883728027, + "rewards/rejected": -1.061623215675354, + "step": 660, + "use_label": 1642.050048828125 + }, + { + "epoch": 0.35, + "learning_rate": 3.60674810936591e-07, + "logits/chosen": -1.261584997177124, + "logits/rejected": -1.1427786350250244, + "logps/chosen": -381.8892822265625, + "logps/rejected": -267.61029052734375, + "loss": 0.2662, + "pred_label": 1007.25, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.13106364011764526, + "rewards/margins": 2.517549991607666, + "rewards/rejected": -2.648613452911377, + "step": 670, + "use_label": 1653.75 + }, + { + "epoch": 0.36, + "learning_rate": 3.5776614310645726e-07, + "logits/chosen": -1.2966188192367554, + "logits/rejected": -1.304734706878662, + "logps/chosen": -329.88262939453125, + "logps/rejected": -311.21649169921875, + "loss": 0.278, + "pred_label": 1033.8499755859375, + "rewards/accuracies": 0.44999998807907104, + "rewards/chosen": -0.22023256123065948, + "rewards/margins": 0.27784207463264465, + "rewards/rejected": -0.49807462096214294, + "step": 680, + "use_label": 1667.1500244140625 + }, + { + "epoch": 0.36, + "learning_rate": 3.548574752763234e-07, + "logits/chosen": -1.0903209447860718, + "logits/rejected": -1.1760300397872925, + "logps/chosen": -370.1248474121094, + "logps/rejected": -320.8165283203125, + "loss": 0.2686, + "pred_label": 1061.199951171875, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 1.1078599691390991, + "rewards/margins": 2.3121399879455566, + "rewards/rejected": -1.204279899597168, + "step": 690, + "use_label": 1679.800048828125 + }, + { + "epoch": 0.37, + "learning_rate": 3.519488074461897e-07, + "logits/chosen": -1.2087898254394531, + "logits/rejected": -1.170114278793335, + "logps/chosen": -316.9244689941406, + "logps/rejected": -294.49346923828125, + "loss": 0.2622, + "pred_label": 1086.3499755859375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 1.7141532897949219, + "rewards/margins": 2.1581480503082275, + "rewards/rejected": -0.4439946711063385, + "step": 700, + "use_label": 1694.6500244140625 + }, + { + "epoch": 0.37, + "learning_rate": 3.4904013961605583e-07, + "logits/chosen": -1.233457326889038, + "logits/rejected": -1.2359874248504639, + "logps/chosen": -314.5318908691406, + "logps/rejected": -388.088623046875, + "loss": 0.2755, + "pred_label": 1111.300048828125, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 1.477555513381958, + "rewards/margins": -0.013068770989775658, + "rewards/rejected": 1.490624189376831, + "step": 710, + "use_label": 1709.699951171875 + }, + { + "epoch": 0.38, + "learning_rate": 3.4613147178592204e-07, + "logits/chosen": -1.2819852828979492, + "logits/rejected": -1.2896205186843872, + "logps/chosen": -241.99050903320312, + "logps/rejected": -281.1546325683594, + "loss": 0.2385, + "pred_label": 1134.5, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.46859627962112427, + "rewards/margins": 2.0235211849212646, + "rewards/rejected": -1.5549249649047852, + "step": 720, + "use_label": 1726.5 + }, + { + "epoch": 0.38, + "learning_rate": 3.4322280395578825e-07, + "logits/chosen": -1.0794142484664917, + "logits/rejected": -1.152029037475586, + "logps/chosen": -319.8411865234375, + "logps/rejected": -360.5047302246094, + "loss": 0.2396, + "pred_label": 1162.1500244140625, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.0120699405670166, + "rewards/margins": 2.247300624847412, + "rewards/rejected": -1.2352306842803955, + "step": 730, + "use_label": 1738.8499755859375 + }, + { + "epoch": 0.39, + "learning_rate": 3.4031413612565446e-07, + "logits/chosen": -1.333953619003296, + "logits/rejected": -1.2932109832763672, + "logps/chosen": -310.48443603515625, + "logps/rejected": -296.43182373046875, + "loss": 0.3141, + "pred_label": 1187.449951171875, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.4783060550689697, + "rewards/margins": 1.6771808862686157, + "rewards/rejected": -2.155486822128296, + "step": 740, + "use_label": 1753.550048828125 + }, + { + "epoch": 0.39, + "learning_rate": 3.374054682955206e-07, + "logits/chosen": -1.3608129024505615, + "logits/rejected": -1.3046305179595947, + "logps/chosen": -330.74310302734375, + "logps/rejected": -310.60198974609375, + "loss": 0.2691, + "pred_label": 1212.5999755859375, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.5366764068603516, + "rewards/margins": 1.5661550760269165, + "rewards/rejected": -1.029478669166565, + "step": 750, + "use_label": 1768.4000244140625 + }, + { + "epoch": 0.4, + "learning_rate": 3.3449680046538687e-07, + "logits/chosen": -1.2935998439788818, + "logits/rejected": -1.292982578277588, + "logps/chosen": -294.06005859375, + "logps/rejected": -312.6604309082031, + "loss": 0.2453, + "pred_label": 1240.4000244140625, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.0071651935577393, + "rewards/margins": 2.852874994277954, + "rewards/rejected": -1.8457095623016357, + "step": 760, + "use_label": 1780.5999755859375 + }, + { + "epoch": 0.4, + "learning_rate": 3.31588132635253e-07, + "logits/chosen": -1.062832236289978, + "logits/rejected": -0.9800139665603638, + "logps/chosen": -297.5071716308594, + "logps/rejected": -353.8055725097656, + "loss": 0.2174, + "pred_label": 1270.25, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.8052531480789185, + "rewards/margins": 4.82207727432251, + "rewards/rejected": -4.016823768615723, + "step": 770, + "use_label": 1790.75 + }, + { + "epoch": 0.41, + "learning_rate": 3.286794648051193e-07, + "logits/chosen": -1.1665582656860352, + "logits/rejected": -1.1822797060012817, + "logps/chosen": -369.43438720703125, + "logps/rejected": -359.2782897949219, + "loss": 0.2728, + "pred_label": 1301.5, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.6651790738105774, + "rewards/margins": 2.360260486602783, + "rewards/rejected": -1.695081353187561, + "step": 780, + "use_label": 1799.5 + }, + { + "epoch": 0.41, + "learning_rate": 3.2577079697498544e-07, + "logits/chosen": -1.3665610551834106, + "logits/rejected": -1.3496174812316895, + "logps/chosen": -404.8565979003906, + "logps/rejected": -373.46405029296875, + "loss": 0.2217, + "pred_label": 1332.300048828125, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.08961932361125946, + "rewards/margins": 3.382511854171753, + "rewards/rejected": -3.4721312522888184, + "step": 790, + "use_label": 1808.699951171875 + }, + { + "epoch": 0.42, + "learning_rate": 3.2286212914485165e-07, + "logits/chosen": -1.3454267978668213, + "logits/rejected": -1.3274281024932861, + "logps/chosen": -335.8668212890625, + "logps/rejected": -332.7888488769531, + "loss": 0.218, + "pred_label": 1360.800048828125, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.4208967685699463, + "rewards/margins": 1.5826103687286377, + "rewards/rejected": -1.1617136001586914, + "step": 800, + "use_label": 1820.199951171875 + }, + { + "epoch": 0.42, + "learning_rate": 3.1995346131471786e-07, + "logits/chosen": -1.2152721881866455, + "logits/rejected": -1.154066801071167, + "logps/chosen": -333.5192565917969, + "logps/rejected": -272.8238220214844, + "loss": 0.2319, + "pred_label": 1388.4000244140625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 1.0722098350524902, + "rewards/margins": 3.2645347118377686, + "rewards/rejected": -2.1923251152038574, + "step": 810, + "use_label": 1832.5999755859375 + }, + { + "epoch": 0.43, + "learning_rate": 3.1704479348458406e-07, + "logits/chosen": -1.312711238861084, + "logits/rejected": -1.3220988512039185, + "logps/chosen": -301.717529296875, + "logps/rejected": -309.1519470214844, + "loss": 0.2123, + "pred_label": 1418.0999755859375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 1.3842562437057495, + "rewards/margins": 1.5546679496765137, + "rewards/rejected": -0.1704115867614746, + "step": 820, + "use_label": 1842.9000244140625 + }, + { + "epoch": 0.43, + "learning_rate": 3.1413612565445027e-07, + "logits/chosen": -1.2439206838607788, + "logits/rejected": -1.2316957712173462, + "logps/chosen": -274.51019287109375, + "logps/rejected": -314.82940673828125, + "loss": 0.2971, + "pred_label": 1444.75, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.229410171508789, + "rewards/margins": 2.7308907508850098, + "rewards/rejected": -1.5014803409576416, + "step": 830, + "use_label": 1856.25 + }, + { + "epoch": 0.44, + "learning_rate": 3.112274578243165e-07, + "logits/chosen": -1.277681589126587, + "logits/rejected": -1.3482627868652344, + "logps/chosen": -245.6984405517578, + "logps/rejected": -274.3936462402344, + "loss": 0.271, + "pred_label": 1471.75, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.7067890167236328, + "rewards/margins": 2.375427007675171, + "rewards/rejected": -0.6686381101608276, + "step": 840, + "use_label": 1869.25 + }, + { + "epoch": 0.44, + "learning_rate": 3.0831878999418263e-07, + "logits/chosen": -1.223837971687317, + "logits/rejected": -1.2153947353363037, + "logps/chosen": -329.645751953125, + "logps/rejected": -368.6504211425781, + "loss": 0.2432, + "pred_label": 1502.050048828125, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.48693960905075073, + "rewards/margins": 1.9081785678863525, + "rewards/rejected": -1.421238660812378, + "step": 850, + "use_label": 1878.949951171875 + }, + { + "epoch": 0.45, + "learning_rate": 3.054101221640489e-07, + "logits/chosen": -1.208174705505371, + "logits/rejected": -1.2833263874053955, + "logps/chosen": -335.9885559082031, + "logps/rejected": -310.40545654296875, + "loss": 0.2323, + "pred_label": 1532.0999755859375, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.3224217891693115, + "rewards/margins": 2.202239513397217, + "rewards/rejected": -0.8798176050186157, + "step": 860, + "use_label": 1888.9000244140625 + }, + { + "epoch": 0.46, + "learning_rate": 3.0250145433391505e-07, + "logits/chosen": -1.2315248250961304, + "logits/rejected": -1.2171595096588135, + "logps/chosen": -362.20660400390625, + "logps/rejected": -303.99725341796875, + "loss": 0.2533, + "pred_label": 1559.4000244140625, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.4734199047088623, + "rewards/margins": 2.8212103843688965, + "rewards/rejected": -1.3477907180786133, + "step": 870, + "use_label": 1901.5999755859375 + }, + { + "epoch": 0.46, + "learning_rate": 2.9959278650378126e-07, + "logits/chosen": -1.2676366567611694, + "logits/rejected": -1.207371711730957, + "logps/chosen": -397.70416259765625, + "logps/rejected": -341.6063537597656, + "loss": 0.2686, + "pred_label": 1589.5999755859375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.5567753911018372, + "rewards/margins": 1.585971474647522, + "rewards/rejected": -1.0291959047317505, + "step": 880, + "use_label": 1911.4000244140625 + }, + { + "epoch": 0.47, + "learning_rate": 2.9668411867364746e-07, + "logits/chosen": -1.1451526880264282, + "logits/rejected": -1.0805937051773071, + "logps/chosen": -237.3237762451172, + "logps/rejected": -232.00241088867188, + "loss": 0.2663, + "pred_label": 1620.300048828125, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.1563435047864914, + "rewards/margins": 1.990779161453247, + "rewards/rejected": -1.8344357013702393, + "step": 890, + "use_label": 1920.699951171875 + }, + { + "epoch": 0.47, + "learning_rate": 2.9377545084351367e-07, + "logits/chosen": -1.1840555667877197, + "logits/rejected": -1.1891014575958252, + "logps/chosen": -219.21728515625, + "logps/rejected": -220.53848266601562, + "loss": 0.2378, + "pred_label": 1648.9000244140625, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.5114763975143433, + "rewards/margins": 1.1068259477615356, + "rewards/rejected": -1.6183021068572998, + "step": 900, + "use_label": 1932.0999755859375 + }, + { + "epoch": 0.48, + "learning_rate": 2.908667830133799e-07, + "logits/chosen": -1.2712427377700806, + "logits/rejected": -1.3315383195877075, + "logps/chosen": -369.0324401855469, + "logps/rejected": -281.7047424316406, + "loss": 0.2503, + "pred_label": 1677.4000244140625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.2587319612503052, + "rewards/margins": 2.2725837230682373, + "rewards/rejected": -1.0138520002365112, + "step": 910, + "use_label": 1943.5999755859375 + }, + { + "epoch": 0.48, + "learning_rate": 2.879581151832461e-07, + "logits/chosen": -1.237377405166626, + "logits/rejected": -1.2065725326538086, + "logps/chosen": -358.11248779296875, + "logps/rejected": -365.1144104003906, + "loss": 0.2472, + "pred_label": 1706.0, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.12370361387729645, + "rewards/margins": 1.9158817529678345, + "rewards/rejected": -1.7921781539916992, + "step": 920, + "use_label": 1955.0 + }, + { + "epoch": 0.49, + "learning_rate": 2.8504944735311224e-07, + "logits/chosen": -1.268371343612671, + "logits/rejected": -1.2534607648849487, + "logps/chosen": -273.5567626953125, + "logps/rejected": -315.6604919433594, + "loss": 0.2614, + "pred_label": 1733.5999755859375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.04811231419444084, + "rewards/margins": 2.209930181503296, + "rewards/rejected": -2.258042335510254, + "step": 930, + "use_label": 1967.4000244140625 + }, + { + "epoch": 0.49, + "learning_rate": 2.821407795229785e-07, + "logits/chosen": -1.164576768875122, + "logits/rejected": -1.0726913213729858, + "logps/chosen": -301.34490966796875, + "logps/rejected": -293.6893005371094, + "loss": 0.2803, + "pred_label": 1760.5, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.535773515701294, + "rewards/margins": 3.1747701168060303, + "rewards/rejected": -2.6389966011047363, + "step": 940, + "use_label": 1980.5 + }, + { + "epoch": 0.5, + "learning_rate": 2.7923211169284466e-07, + "logits/chosen": -1.2998483180999756, + "logits/rejected": -1.2662798166275024, + "logps/chosen": -343.672607421875, + "logps/rejected": -368.7283630371094, + "loss": 0.2189, + "pred_label": 1787.449951171875, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.820811927318573, + "rewards/margins": 1.670477271080017, + "rewards/rejected": -0.8496652841567993, + "step": 950, + "use_label": 1993.550048828125 + }, + { + "epoch": 0.5, + "learning_rate": 2.763234438627109e-07, + "logits/chosen": -1.313528060913086, + "logits/rejected": -1.2543927431106567, + "logps/chosen": -303.2686462402344, + "logps/rejected": -286.05535888671875, + "loss": 0.256, + "pred_label": 1810.1500244140625, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.3981182873249054, + "rewards/margins": 1.8850862979888916, + "rewards/rejected": -2.2832047939300537, + "step": 960, + "use_label": 2010.8499755859375 + }, + { + "epoch": 0.51, + "learning_rate": 2.7341477603257707e-07, + "logits/chosen": -1.2742154598236084, + "logits/rejected": -1.2301501035690308, + "logps/chosen": -339.7518310546875, + "logps/rejected": -329.7853088378906, + "loss": 0.2627, + "pred_label": 1836.300048828125, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.6415281295776367, + "rewards/margins": 1.2891826629638672, + "rewards/rejected": -1.930710792541504, + "step": 970, + "use_label": 2024.699951171875 + }, + { + "epoch": 0.51, + "learning_rate": 2.705061082024433e-07, + "logits/chosen": -1.2222896814346313, + "logits/rejected": -1.1963387727737427, + "logps/chosen": -375.39959716796875, + "logps/rejected": -348.4335632324219, + "loss": 0.2533, + "pred_label": 1864.6500244140625, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.4741919040679932, + "rewards/margins": 3.042367458343506, + "rewards/rejected": -1.5681754350662231, + "step": 980, + "use_label": 2036.3499755859375 + }, + { + "epoch": 0.52, + "learning_rate": 2.675974403723095e-07, + "logits/chosen": -1.1725823879241943, + "logits/rejected": -1.0960336923599243, + "logps/chosen": -304.5809326171875, + "logps/rejected": -284.103271484375, + "loss": 0.2378, + "pred_label": 1895.4000244140625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.33665040135383606, + "rewards/margins": 4.249828338623047, + "rewards/rejected": -3.913177967071533, + "step": 990, + "use_label": 2045.5999755859375 + }, + { + "epoch": 0.52, + "learning_rate": 2.646887725421757e-07, + "logits/chosen": -1.1304484605789185, + "logits/rejected": -1.125525712966919, + "logps/chosen": -365.712158203125, + "logps/rejected": -345.81195068359375, + "loss": 0.2517, + "pred_label": 1928.199951171875, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.9080309867858887, + "rewards/margins": 3.343575954437256, + "rewards/rejected": -2.435544967651367, + "step": 1000, + "use_label": 2052.800048828125 + }, + { + "epoch": 0.53, + "learning_rate": 2.6178010471204185e-07, + "logits/chosen": -1.2770113945007324, + "logits/rejected": -1.214017391204834, + "logps/chosen": -372.6104431152344, + "logps/rejected": -243.80184936523438, + "loss": 0.1902, + "pred_label": 1958.949951171875, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.21130351722240448, + "rewards/margins": 2.0323612689971924, + "rewards/rejected": -2.2436647415161133, + "step": 1010, + "use_label": 2062.050048828125 + }, + { + "epoch": 0.53, + "learning_rate": 2.588714368819081e-07, + "logits/chosen": -1.1657683849334717, + "logits/rejected": -1.1163619756698608, + "logps/chosen": -297.716064453125, + "logps/rejected": -282.97125244140625, + "loss": 0.2088, + "pred_label": 1987.5999755859375, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.1362704634666443, + "rewards/margins": 2.989813804626465, + "rewards/rejected": -3.126084089279175, + "step": 1020, + "use_label": 2073.39990234375 + }, + { + "epoch": 0.54, + "learning_rate": 2.5596276905177426e-07, + "logits/chosen": -1.226504921913147, + "logits/rejected": -1.194415807723999, + "logps/chosen": -339.902587890625, + "logps/rejected": -316.89312744140625, + "loss": 0.2374, + "pred_label": 2018.0, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.9146644473075867, + "rewards/margins": 2.7297873497009277, + "rewards/rejected": -1.8151226043701172, + "step": 1030, + "use_label": 2083.0 + }, + { + "epoch": 0.54, + "learning_rate": 2.530541012216405e-07, + "logits/chosen": -1.2352535724639893, + "logits/rejected": -1.1748406887054443, + "logps/chosen": -311.44940185546875, + "logps/rejected": -284.37603759765625, + "loss": 0.2823, + "pred_label": 2045.3499755859375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.10683774948120117, + "rewards/margins": 1.7979570627212524, + "rewards/rejected": -1.6911194324493408, + "step": 1040, + "use_label": 2095.64990234375 + }, + { + "epoch": 0.55, + "learning_rate": 2.501454333915067e-07, + "logits/chosen": -1.1737799644470215, + "logits/rejected": -1.2158236503601074, + "logps/chosen": -411.43634033203125, + "logps/rejected": -429.01605224609375, + "loss": 0.2445, + "pred_label": 2076.300048828125, + "rewards/accuracies": 0.625, + "rewards/chosen": 1.1737092733383179, + "rewards/margins": 2.0826451778411865, + "rewards/rejected": -0.9089359045028687, + "step": 1050, + "use_label": 2104.699951171875 + }, + { + "epoch": 0.55, + "learning_rate": 2.472367655613729e-07, + "logits/chosen": -1.195278525352478, + "logits/rejected": -1.1494415998458862, + "logps/chosen": -345.48663330078125, + "logps/rejected": -427.1460876464844, + "loss": 0.2202, + "pred_label": 2106.35009765625, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.7652619481086731, + "rewards/margins": 4.071230411529541, + "rewards/rejected": -3.3059685230255127, + "step": 1060, + "use_label": 2114.64990234375 + }, + { + "epoch": 0.56, + "learning_rate": 2.443280977312391e-07, + "logits/chosen": -1.2246100902557373, + "logits/rejected": -1.2212460041046143, + "logps/chosen": -376.6558532714844, + "logps/rejected": -344.31451416015625, + "loss": 0.2269, + "pred_label": 2134.449951171875, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 1.7661001682281494, + "rewards/margins": 4.430109977722168, + "rewards/rejected": -2.6640098094940186, + "step": 1070, + "use_label": 2126.550048828125 + }, + { + "epoch": 0.57, + "learning_rate": 2.414194299011053e-07, + "logits/chosen": -1.1105204820632935, + "logits/rejected": -1.0868804454803467, + "logps/chosen": -314.63348388671875, + "logps/rejected": -307.9922790527344, + "loss": 0.1703, + "pred_label": 2163.5, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.6920607089996338, + "rewards/margins": 2.328885555267334, + "rewards/rejected": -1.6368249654769897, + "step": 1080, + "use_label": 2137.5 + }, + { + "epoch": 0.57, + "learning_rate": 2.385107620709715e-07, + "logits/chosen": -1.1469954252243042, + "logits/rejected": -1.1428484916687012, + "logps/chosen": -335.100830078125, + "logps/rejected": -333.8580627441406, + "loss": 0.1841, + "pred_label": 2190.800048828125, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.8686046600341797, + "rewards/margins": 2.732456922531128, + "rewards/rejected": -1.8638522624969482, + "step": 1090, + "use_label": 2150.199951171875 + }, + { + "epoch": 0.58, + "learning_rate": 2.356020942408377e-07, + "logits/chosen": -1.1892510652542114, + "logits/rejected": -1.1579782962799072, + "logps/chosen": -343.0919189453125, + "logps/rejected": -357.2165832519531, + "loss": 0.2029, + "pred_label": 2219.800048828125, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.5540876388549805, + "rewards/margins": 2.639407157897949, + "rewards/rejected": -1.0853195190429688, + "step": 1100, + "use_label": 2161.199951171875 + }, + { + "epoch": 0.58, + "learning_rate": 2.326934264107039e-07, + "logits/chosen": -1.150119662284851, + "logits/rejected": -1.1163709163665771, + "logps/chosen": -424.14764404296875, + "logps/rejected": -254.954345703125, + "loss": 0.2579, + "pred_label": 2250.14990234375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.06615400314331055, + "rewards/margins": 2.4133870601654053, + "rewards/rejected": -2.347233295440674, + "step": 1110, + "use_label": 2170.85009765625 + }, + { + "epoch": 0.59, + "learning_rate": 2.2978475858057008e-07, + "logits/chosen": -1.1137053966522217, + "logits/rejected": -1.1033612489700317, + "logps/chosen": -301.53887939453125, + "logps/rejected": -317.7568054199219, + "loss": 0.2462, + "pred_label": 2276.85009765625, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 1.4118252992630005, + "rewards/margins": 1.6095244884490967, + "rewards/rejected": -0.19769929349422455, + "step": 1120, + "use_label": 2184.14990234375 + }, + { + "epoch": 0.59, + "learning_rate": 2.2687609075043629e-07, + "logits/chosen": -1.1289111375808716, + "logits/rejected": -1.0888346433639526, + "logps/chosen": -350.7413024902344, + "logps/rejected": -379.24273681640625, + "loss": 0.2236, + "pred_label": 2302.39990234375, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.709761619567871, + "rewards/margins": 3.1695258617401123, + "rewards/rejected": -1.4597642421722412, + "step": 1130, + "use_label": 2198.60009765625 + }, + { + "epoch": 0.6, + "learning_rate": 2.239674229203025e-07, + "logits/chosen": -1.162214994430542, + "logits/rejected": -1.2171916961669922, + "logps/chosen": -343.6764831542969, + "logps/rejected": -366.0986022949219, + "loss": 0.2462, + "pred_label": 2330.050048828125, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 1.5795726776123047, + "rewards/margins": 2.7293777465820312, + "rewards/rejected": -1.1498053073883057, + "step": 1140, + "use_label": 2210.949951171875 + }, + { + "epoch": 0.6, + "learning_rate": 2.210587550901687e-07, + "logits/chosen": -1.0336754322052002, + "logits/rejected": -1.0693806409835815, + "logps/chosen": -379.285888671875, + "logps/rejected": -329.3329162597656, + "loss": 0.1904, + "pred_label": 2355.050048828125, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.6715002059936523, + "rewards/margins": 4.020023822784424, + "rewards/rejected": -2.3485240936279297, + "step": 1150, + "use_label": 2225.949951171875 + }, + { + "epoch": 0.61, + "learning_rate": 2.1815008726003488e-07, + "logits/chosen": -1.1704041957855225, + "logits/rejected": -1.1589972972869873, + "logps/chosen": -324.221923828125, + "logps/rejected": -336.85528564453125, + "loss": 0.1874, + "pred_label": 2386.800048828125, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.6751067638397217, + "rewards/margins": 2.5076706409454346, + "rewards/rejected": -1.8325637578964233, + "step": 1160, + "use_label": 2234.199951171875 + }, + { + "epoch": 0.61, + "learning_rate": 2.152414194299011e-07, + "logits/chosen": -1.0068198442459106, + "logits/rejected": -1.000146508216858, + "logps/chosen": -282.8496398925781, + "logps/rejected": -321.56085205078125, + "loss": 0.2117, + "pred_label": 2416.5, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.7289396524429321, + "rewards/margins": 1.5910956859588623, + "rewards/rejected": -2.320035219192505, + "step": 1170, + "use_label": 2244.5 + }, + { + "epoch": 0.62, + "learning_rate": 2.123327515997673e-07, + "logits/chosen": -0.9953794479370117, + "logits/rejected": -0.970104992389679, + "logps/chosen": -317.9019470214844, + "logps/rejected": -309.2327575683594, + "loss": 0.1938, + "pred_label": 2443.75, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.3172712028026581, + "rewards/margins": 2.3301925659179688, + "rewards/rejected": -2.6474642753601074, + "step": 1180, + "use_label": 2257.25 + }, + { + "epoch": 0.62, + "learning_rate": 2.094240837696335e-07, + "logits/chosen": -1.0719118118286133, + "logits/rejected": -1.0339205265045166, + "logps/chosen": -371.0210876464844, + "logps/rejected": -436.35797119140625, + "loss": 0.1679, + "pred_label": 2471.5, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 1.381567120552063, + "rewards/margins": 4.559169292449951, + "rewards/rejected": -3.1776022911071777, + "step": 1190, + "use_label": 2269.5 + }, + { + "epoch": 0.63, + "learning_rate": 2.0651541593949969e-07, + "logits/chosen": -1.1298894882202148, + "logits/rejected": -1.1007440090179443, + "logps/chosen": -314.60009765625, + "logps/rejected": -374.28387451171875, + "loss": 0.2071, + "pred_label": 2505.75, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 1.431455373764038, + "rewards/margins": 4.434044361114502, + "rewards/rejected": -3.0025887489318848, + "step": 1200, + "use_label": 2275.25 + }, + { + "epoch": 0.63, + "learning_rate": 2.036067481093659e-07, + "logits/chosen": -0.998555064201355, + "logits/rejected": -1.096919298171997, + "logps/chosen": -364.8897399902344, + "logps/rejected": -305.2925720214844, + "loss": 0.2148, + "pred_label": 2539.550048828125, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": 0.8155422210693359, + "rewards/margins": 2.473628520965576, + "rewards/rejected": -1.6580864191055298, + "step": 1210, + "use_label": 2281.449951171875 + }, + { + "epoch": 0.64, + "learning_rate": 2.006980802792321e-07, + "logits/chosen": -1.093653678894043, + "logits/rejected": -1.1002882719039917, + "logps/chosen": -354.3478088378906, + "logps/rejected": -318.2798156738281, + "loss": 0.1867, + "pred_label": 2571.14990234375, + "rewards/accuracies": 0.75, + "rewards/chosen": 1.8111639022827148, + "rewards/margins": 3.449200391769409, + "rewards/rejected": -1.6380364894866943, + "step": 1220, + "use_label": 2289.85009765625 + }, + { + "epoch": 0.64, + "learning_rate": 1.977894124490983e-07, + "logits/chosen": -1.2149935960769653, + "logits/rejected": -1.1372343301773071, + "logps/chosen": -292.52294921875, + "logps/rejected": -265.5375061035156, + "loss": 0.1922, + "pred_label": 2601.39990234375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.0008518159156665206, + "rewards/margins": 2.4776790142059326, + "rewards/rejected": -2.4768271446228027, + "step": 1230, + "use_label": 2299.60009765625 + }, + { + "epoch": 0.65, + "learning_rate": 1.9488074461896452e-07, + "logits/chosen": -1.1095410585403442, + "logits/rejected": -1.108186960220337, + "logps/chosen": -381.8337707519531, + "logps/rejected": -426.4427185058594, + "loss": 0.2153, + "pred_label": 2632.60009765625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.9956940412521362, + "rewards/margins": 4.763395309448242, + "rewards/rejected": -3.7677009105682373, + "step": 1240, + "use_label": 2308.39990234375 + }, + { + "epoch": 0.65, + "learning_rate": 1.919720767888307e-07, + "logits/chosen": -1.2132353782653809, + "logits/rejected": -1.2195868492126465, + "logps/chosen": -322.24774169921875, + "logps/rejected": -393.90020751953125, + "loss": 0.1843, + "pred_label": 2666.699951171875, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.1081535592675209, + "rewards/margins": 2.927469253540039, + "rewards/rejected": -2.8193156719207764, + "step": 1250, + "use_label": 2314.300048828125 + }, + { + "epoch": 0.66, + "learning_rate": 1.890634089586969e-07, + "logits/chosen": -1.0909610986709595, + "logits/rejected": -1.0712100267410278, + "logps/chosen": -301.9937744140625, + "logps/rejected": -297.81365966796875, + "loss": 0.2063, + "pred_label": 2699.35009765625, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.6107617616653442, + "rewards/margins": 3.5464184284210205, + "rewards/rejected": -4.157180309295654, + "step": 1260, + "use_label": 2321.64990234375 + }, + { + "epoch": 0.66, + "learning_rate": 1.861547411285631e-07, + "logits/chosen": -1.1298341751098633, + "logits/rejected": -1.0467592477798462, + "logps/chosen": -324.2348937988281, + "logps/rejected": -344.5401611328125, + "loss": 0.2163, + "pred_label": 2732.89990234375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.3071551024913788, + "rewards/margins": 3.859396457672119, + "rewards/rejected": -4.16655158996582, + "step": 1270, + "use_label": 2328.10009765625 + }, + { + "epoch": 0.67, + "learning_rate": 1.8324607329842932e-07, + "logits/chosen": -1.0855954885482788, + "logits/rejected": -1.1321277618408203, + "logps/chosen": -276.91912841796875, + "logps/rejected": -345.2870178222656, + "loss": 0.2098, + "pred_label": 2764.39990234375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -1.4349310398101807, + "rewards/margins": 2.8562276363372803, + "rewards/rejected": -4.291158676147461, + "step": 1280, + "use_label": 2336.60009765625 + }, + { + "epoch": 0.68, + "learning_rate": 1.803374054682955e-07, + "logits/chosen": -0.9666376113891602, + "logits/rejected": -1.0153762102127075, + "logps/chosen": -300.74078369140625, + "logps/rejected": -354.02117919921875, + "loss": 0.2097, + "pred_label": 2795.449951171875, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.6840406656265259, + "rewards/margins": 2.9005465507507324, + "rewards/rejected": -3.5845870971679688, + "step": 1290, + "use_label": 2345.550048828125 + }, + { + "epoch": 0.68, + "learning_rate": 1.774287376381617e-07, + "logits/chosen": -1.1049658060073853, + "logits/rejected": -1.0567684173583984, + "logps/chosen": -389.1566162109375, + "logps/rejected": -305.6317443847656, + "loss": 0.1964, + "pred_label": 2828.35009765625, + "rewards/accuracies": 0.8999999761581421, + "rewards/chosen": 0.8954909443855286, + "rewards/margins": 4.705790042877197, + "rewards/rejected": -3.8102996349334717, + "step": 1300, + "use_label": 2352.64990234375 + }, + { + "epoch": 0.69, + "learning_rate": 1.7452006980802792e-07, + "logits/chosen": -1.1040902137756348, + "logits/rejected": -1.0495140552520752, + "logps/chosen": -460.7845764160156, + "logps/rejected": -396.0311279296875, + "loss": 0.2207, + "pred_label": 2858.25, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.5667216777801514, + "rewards/margins": 2.7725882530212402, + "rewards/rejected": -1.2058665752410889, + "step": 1310, + "use_label": 2362.75 + }, + { + "epoch": 0.69, + "learning_rate": 1.7161140197789412e-07, + "logits/chosen": -1.0317670106887817, + "logits/rejected": -1.0540131330490112, + "logps/chosen": -372.09661865234375, + "logps/rejected": -359.15264892578125, + "loss": 0.192, + "pred_label": 2887.85009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -2.242278575897217, + "rewards/margins": 1.178301453590393, + "rewards/rejected": -3.4205803871154785, + "step": 1320, + "use_label": 2373.14990234375 + }, + { + "epoch": 0.7, + "learning_rate": 1.687027341477603e-07, + "logits/chosen": -1.0154337882995605, + "logits/rejected": -1.026306390762329, + "logps/chosen": -246.6059112548828, + "logps/rejected": -313.6022033691406, + "loss": 0.1788, + "pred_label": 2915.949951171875, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.02297346666455269, + "rewards/margins": 3.024552583694458, + "rewards/rejected": -3.0475258827209473, + "step": 1330, + "use_label": 2385.050048828125 + }, + { + "epoch": 0.7, + "learning_rate": 1.657940663176265e-07, + "logits/chosen": -1.0699912309646606, + "logits/rejected": -1.0195733308792114, + "logps/chosen": -466.06085205078125, + "logps/rejected": -317.84381103515625, + "loss": 0.2023, + "pred_label": 2949.5, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.48599451780319214, + "rewards/margins": 4.262557029724121, + "rewards/rejected": -3.776562213897705, + "step": 1340, + "use_label": 2391.5 + }, + { + "epoch": 0.71, + "learning_rate": 1.6288539848749272e-07, + "logits/chosen": -1.0590379238128662, + "logits/rejected": -1.0681087970733643, + "logps/chosen": -320.8080139160156, + "logps/rejected": -345.09051513671875, + "loss": 0.2052, + "pred_label": 2979.60009765625, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.6778038144111633, + "rewards/margins": 2.3581137657165527, + "rewards/rejected": -3.0359177589416504, + "step": 1350, + "use_label": 2401.39990234375 + }, + { + "epoch": 0.71, + "learning_rate": 1.5997673065735893e-07, + "logits/chosen": -1.0564696788787842, + "logits/rejected": -0.9981715083122253, + "logps/chosen": -422.28436279296875, + "logps/rejected": -317.775146484375, + "loss": 0.2191, + "pred_label": 3008.10009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.9884247779846191, + "rewards/margins": 2.251682996749878, + "rewards/rejected": -3.240108013153076, + "step": 1360, + "use_label": 2412.89990234375 + }, + { + "epoch": 0.72, + "learning_rate": 1.5706806282722514e-07, + "logits/chosen": -1.1280796527862549, + "logits/rejected": -1.0177102088928223, + "logps/chosen": -372.64996337890625, + "logps/rejected": -325.7216796875, + "loss": 0.1683, + "pred_label": 3039.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.2417742758989334, + "rewards/margins": 3.3245158195495605, + "rewards/rejected": -3.5662899017333984, + "step": 1370, + "use_label": 2422.0 + }, + { + "epoch": 0.72, + "learning_rate": 1.5415939499709132e-07, + "logits/chosen": -1.0724250078201294, + "logits/rejected": -1.0401219129562378, + "logps/chosen": -272.91546630859375, + "logps/rejected": -230.291748046875, + "loss": 0.236, + "pred_label": 3070.25, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -1.9332826137542725, + "rewards/margins": 2.3847384452819824, + "rewards/rejected": -4.318020820617676, + "step": 1380, + "use_label": 2430.75 + }, + { + "epoch": 0.73, + "learning_rate": 1.5125072716695752e-07, + "logits/chosen": -1.0457665920257568, + "logits/rejected": -1.0457961559295654, + "logps/chosen": -352.20379638671875, + "logps/rejected": -325.810546875, + "loss": 0.183, + "pred_label": 3103.550048828125, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": -0.5959171652793884, + "rewards/margins": 3.479651927947998, + "rewards/rejected": -4.075569152832031, + "step": 1390, + "use_label": 2437.449951171875 + }, + { + "epoch": 0.73, + "learning_rate": 1.4834205933682373e-07, + "logits/chosen": -1.0471112728118896, + "logits/rejected": -1.0389719009399414, + "logps/chosen": -274.6317138671875, + "logps/rejected": -285.1258239746094, + "loss": 0.1871, + "pred_label": 3134.60009765625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.41235724091529846, + "rewards/margins": 3.290208339691162, + "rewards/rejected": -3.7025654315948486, + "step": 1400, + "use_label": 2446.39990234375 + }, + { + "epoch": 0.74, + "learning_rate": 1.4543339150668994e-07, + "logits/chosen": -1.1152770519256592, + "logits/rejected": -0.9952844381332397, + "logps/chosen": -310.62872314453125, + "logps/rejected": -284.43927001953125, + "loss": 0.2265, + "pred_label": 3163.60009765625, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.20648960769176483, + "rewards/margins": 2.571920871734619, + "rewards/rejected": -2.365431547164917, + "step": 1410, + "use_label": 2457.39990234375 + }, + { + "epoch": 0.74, + "learning_rate": 1.4252472367655612e-07, + "logits/chosen": -0.9487142562866211, + "logits/rejected": -0.9793168902397156, + "logps/chosen": -406.9850769042969, + "logps/rejected": -329.6900634765625, + "loss": 0.1839, + "pred_label": 3192.89990234375, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.578311562538147, + "rewards/margins": 3.119631290435791, + "rewards/rejected": -2.5413200855255127, + "step": 1420, + "use_label": 2468.10009765625 + }, + { + "epoch": 0.75, + "learning_rate": 1.3961605584642233e-07, + "logits/chosen": -1.0938589572906494, + "logits/rejected": -1.0605380535125732, + "logps/chosen": -306.23809814453125, + "logps/rejected": -274.6723937988281, + "loss": 0.1687, + "pred_label": 3223.800048828125, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.6292298436164856, + "rewards/margins": 3.2063794136047363, + "rewards/rejected": -3.8356094360351562, + "step": 1430, + "use_label": 2477.199951171875 + }, + { + "epoch": 0.75, + "learning_rate": 1.3670738801628854e-07, + "logits/chosen": -0.9974590539932251, + "logits/rejected": -1.0331952571868896, + "logps/chosen": -378.69158935546875, + "logps/rejected": -439.4248046875, + "loss": 0.1876, + "pred_label": 3252.699951171875, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.29973307251930237, + "rewards/margins": 3.267655849456787, + "rewards/rejected": -3.5673890113830566, + "step": 1440, + "use_label": 2488.300048828125 + }, + { + "epoch": 0.76, + "learning_rate": 1.3379872018615474e-07, + "logits/chosen": -1.0476627349853516, + "logits/rejected": -1.0306545495986938, + "logps/chosen": -381.58709716796875, + "logps/rejected": -317.57110595703125, + "loss": 0.203, + "pred_label": 3282.699951171875, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.4640182554721832, + "rewards/margins": 2.2011430263519287, + "rewards/rejected": -2.665161609649658, + "step": 1450, + "use_label": 2498.300048828125 + }, + { + "epoch": 0.76, + "learning_rate": 1.3089005235602092e-07, + "logits/chosen": -1.0808570384979248, + "logits/rejected": -1.0447580814361572, + "logps/chosen": -351.30523681640625, + "logps/rejected": -301.14276123046875, + "loss": 0.2173, + "pred_label": 3313.64990234375, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.037274181842803955, + "rewards/margins": 2.719515323638916, + "rewards/rejected": -2.756789445877075, + "step": 1460, + "use_label": 2507.35009765625 + }, + { + "epoch": 0.77, + "learning_rate": 1.2798138452588713e-07, + "logits/chosen": -1.0119661092758179, + "logits/rejected": -1.0127899646759033, + "logps/chosen": -452.9007263183594, + "logps/rejected": -350.81988525390625, + "loss": 0.2244, + "pred_label": 3344.35009765625, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.5907049179077148, + "rewards/margins": 4.506594181060791, + "rewards/rejected": -3.915889024734497, + "step": 1470, + "use_label": 2516.64990234375 + }, + { + "epoch": 0.77, + "learning_rate": 1.2507271669575334e-07, + "logits/chosen": -0.9774877429008484, + "logits/rejected": -1.0214698314666748, + "logps/chosen": -311.8316955566406, + "logps/rejected": -339.8797607421875, + "loss": 0.1938, + "pred_label": 3374.64990234375, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.14671669900417328, + "rewards/margins": 1.559191346168518, + "rewards/rejected": -1.7059080600738525, + "step": 1480, + "use_label": 2526.35009765625 + }, + { + "epoch": 0.78, + "learning_rate": 1.2216404886561955e-07, + "logits/chosen": -1.0379236936569214, + "logits/rejected": -1.0210058689117432, + "logps/chosen": -270.0731506347656, + "logps/rejected": -259.6552734375, + "loss": 0.2187, + "pred_label": 3405.449951171875, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.6488065719604492, + "rewards/margins": 2.2493479251861572, + "rewards/rejected": -2.8981544971466064, + "step": 1490, + "use_label": 2535.550048828125 + }, + { + "epoch": 0.79, + "learning_rate": 1.1925538103548575e-07, + "logits/chosen": -0.8977063894271851, + "logits/rejected": -0.984424889087677, + "logps/chosen": -356.542724609375, + "logps/rejected": -314.09893798828125, + "loss": 0.1712, + "pred_label": 3438.60009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.38773947954177856, + "rewards/margins": 2.917739152908325, + "rewards/rejected": -2.5299997329711914, + "step": 1500, + "use_label": 2542.39990234375 + }, + { + "epoch": 0.79, + "learning_rate": 1.1634671320535195e-07, + "logits/chosen": -1.0663585662841797, + "logits/rejected": -1.0083823204040527, + "logps/chosen": -318.5123291015625, + "logps/rejected": -304.78387451171875, + "loss": 0.1956, + "pred_label": 3471.699951171875, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.7554278373718262, + "rewards/margins": 3.285256862640381, + "rewards/rejected": -2.529829263687134, + "step": 1510, + "use_label": 2549.300048828125 + }, + { + "epoch": 0.8, + "learning_rate": 1.1343804537521814e-07, + "logits/chosen": -1.051598310470581, + "logits/rejected": -1.0226097106933594, + "logps/chosen": -377.0965881347656, + "logps/rejected": -338.76812744140625, + "loss": 0.2022, + "pred_label": 3499.949951171875, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.8909615278244019, + "rewards/margins": 1.9300997257232666, + "rewards/rejected": -1.039137840270996, + "step": 1520, + "use_label": 2561.050048828125 + }, + { + "epoch": 0.8, + "learning_rate": 1.1052937754508435e-07, + "logits/chosen": -1.105948567390442, + "logits/rejected": -1.1236369609832764, + "logps/chosen": -421.694580078125, + "logps/rejected": -365.2895202636719, + "loss": 0.2134, + "pred_label": 3530.699951171875, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 1.1226226091384888, + "rewards/margins": 4.040251731872559, + "rewards/rejected": -2.9176290035247803, + "step": 1530, + "use_label": 2570.300048828125 + }, + { + "epoch": 0.81, + "learning_rate": 1.0762070971495054e-07, + "logits/chosen": -1.0308454036712646, + "logits/rejected": -1.0807101726531982, + "logps/chosen": -355.47320556640625, + "logps/rejected": -407.9166259765625, + "loss": 0.2194, + "pred_label": 3559.35009765625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 1.4176074266433716, + "rewards/margins": 2.80814528465271, + "rewards/rejected": -1.390538215637207, + "step": 1540, + "use_label": 2581.64990234375 + }, + { + "epoch": 0.81, + "learning_rate": 1.0471204188481675e-07, + "logits/chosen": -1.0221843719482422, + "logits/rejected": -1.003930926322937, + "logps/chosen": -313.1138610839844, + "logps/rejected": -262.46295166015625, + "loss": 0.2012, + "pred_label": 3589.85009765625, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.20224742591381073, + "rewards/margins": 4.130466461181641, + "rewards/rejected": -3.9282188415527344, + "step": 1550, + "use_label": 2591.14990234375 + }, + { + "epoch": 0.82, + "learning_rate": 1.0180337405468295e-07, + "logits/chosen": -1.0754446983337402, + "logits/rejected": -1.0588288307189941, + "logps/chosen": -333.0397644042969, + "logps/rejected": -329.0122985839844, + "loss": 0.212, + "pred_label": 3616.550048828125, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -0.766626238822937, + "rewards/margins": 3.1271777153015137, + "rewards/rejected": -3.893803358078003, + "step": 1560, + "use_label": 2604.449951171875 + }, + { + "epoch": 0.82, + "learning_rate": 9.889470622454915e-08, + "logits/chosen": -1.0235944986343384, + "logits/rejected": -1.030893087387085, + "logps/chosen": -322.72857666015625, + "logps/rejected": -325.28912353515625, + "loss": 0.1968, + "pred_label": 3644.64990234375, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -1.4102725982666016, + "rewards/margins": 2.7497966289520264, + "rewards/rejected": -4.160068988800049, + "step": 1570, + "use_label": 2616.35009765625 + }, + { + "epoch": 0.83, + "learning_rate": 9.598603839441535e-08, + "logits/chosen": -1.0010260343551636, + "logits/rejected": -1.0144360065460205, + "logps/chosen": -380.1663818359375, + "logps/rejected": -335.47662353515625, + "loss": 0.2129, + "pred_label": 3672.25, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.20972386002540588, + "rewards/margins": 1.8820480108261108, + "rewards/rejected": -1.6723241806030273, + "step": 1580, + "use_label": 2628.75 + }, + { + "epoch": 0.83, + "learning_rate": 9.307737056428156e-08, + "logits/chosen": -0.800024151802063, + "logits/rejected": -0.8323174715042114, + "logps/chosen": -315.90447998046875, + "logps/rejected": -332.60150146484375, + "loss": 0.1793, + "pred_label": 3700.699951171875, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.8666568994522095, + "rewards/margins": 3.055481433868408, + "rewards/rejected": -3.9221386909484863, + "step": 1590, + "use_label": 2640.300048828125 + }, + { + "epoch": 0.84, + "learning_rate": 9.016870273414775e-08, + "logits/chosen": -0.8738692998886108, + "logits/rejected": -0.9172623753547668, + "logps/chosen": -350.1977233886719, + "logps/rejected": -301.5623474121094, + "loss": 0.2091, + "pred_label": 3731.60009765625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -1.1013541221618652, + "rewards/margins": 2.4148409366607666, + "rewards/rejected": -3.5161945819854736, + "step": 1600, + "use_label": 2649.39990234375 + }, + { + "epoch": 0.84, + "learning_rate": 8.726003490401396e-08, + "logits/chosen": -1.0966861248016357, + "logits/rejected": -1.0557001829147339, + "logps/chosen": -371.09173583984375, + "logps/rejected": -266.89263916015625, + "loss": 0.1972, + "pred_label": 3761.550048828125, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.1973705142736435, + "rewards/margins": 3.2193920612335205, + "rewards/rejected": -3.416762590408325, + "step": 1610, + "use_label": 2659.449951171875 + }, + { + "epoch": 0.85, + "learning_rate": 8.435136707388015e-08, + "logits/chosen": -0.9223749041557312, + "logits/rejected": -0.9156180620193481, + "logps/chosen": -407.92889404296875, + "logps/rejected": -357.1949157714844, + "loss": 0.1682, + "pred_label": 3793.60009765625, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.09885386377573013, + "rewards/margins": 2.7179036140441895, + "rewards/rejected": -2.8167572021484375, + "step": 1620, + "use_label": 2667.39990234375 + }, + { + "epoch": 0.85, + "learning_rate": 8.144269924374636e-08, + "logits/chosen": -0.8016144633293152, + "logits/rejected": -0.8304374814033508, + "logps/chosen": -392.49249267578125, + "logps/rejected": -387.77813720703125, + "loss": 0.2052, + "pred_label": 3822.14990234375, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.7084876298904419, + "rewards/margins": 3.3112130165100098, + "rewards/rejected": -4.019700527191162, + "step": 1630, + "use_label": 2678.85009765625 + }, + { + "epoch": 0.86, + "learning_rate": 7.853403141361257e-08, + "logits/chosen": -1.014793872833252, + "logits/rejected": -0.9608389139175415, + "logps/chosen": -301.42108154296875, + "logps/rejected": -262.4256286621094, + "loss": 0.23, + "pred_label": 3848.75, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.032284706830978394, + "rewards/margins": 2.6330177783966064, + "rewards/rejected": -2.6653025150299072, + "step": 1640, + "use_label": 2692.25 + }, + { + "epoch": 0.86, + "learning_rate": 7.562536358347876e-08, + "logits/chosen": -0.9088438749313354, + "logits/rejected": -0.9523930549621582, + "logps/chosen": -322.62933349609375, + "logps/rejected": -361.05670166015625, + "loss": 0.2088, + "pred_label": 3879.449951171875, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.8268980979919434, + "rewards/margins": 1.8542957305908203, + "rewards/rejected": -2.6811938285827637, + "step": 1650, + "use_label": 2701.550048828125 + }, + { + "epoch": 0.87, + "learning_rate": 7.271669575334497e-08, + "logits/chosen": -0.9642359614372253, + "logits/rejected": -0.8722718358039856, + "logps/chosen": -359.36968994140625, + "logps/rejected": -413.9576110839844, + "loss": 0.1724, + "pred_label": 3914.60009765625, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.34769096970558167, + "rewards/margins": 3.559408664703369, + "rewards/rejected": -3.907099485397339, + "step": 1660, + "use_label": 2706.39990234375 + }, + { + "epoch": 0.87, + "learning_rate": 6.980802792321116e-08, + "logits/chosen": -0.8372431993484497, + "logits/rejected": -0.8543712496757507, + "logps/chosen": -358.82550048828125, + "logps/rejected": -364.4158630371094, + "loss": 0.2092, + "pred_label": 3946.949951171875, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.024167824536561966, + "rewards/margins": 1.891296148300171, + "rewards/rejected": -1.9154638051986694, + "step": 1670, + "use_label": 2714.050048828125 + }, + { + "epoch": 0.88, + "learning_rate": 6.689936009307737e-08, + "logits/chosen": -0.8467708826065063, + "logits/rejected": -0.8923909068107605, + "logps/chosen": -311.6363830566406, + "logps/rejected": -330.9986267089844, + "loss": 0.1716, + "pred_label": 3979.300048828125, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -1.442889928817749, + "rewards/margins": 1.873750925064087, + "rewards/rejected": -3.316641330718994, + "step": 1680, + "use_label": 2721.699951171875 + }, + { + "epoch": 0.88, + "learning_rate": 6.399069226294357e-08, + "logits/chosen": -0.9512916803359985, + "logits/rejected": -0.9659273028373718, + "logps/chosen": -374.9173278808594, + "logps/rejected": -325.18695068359375, + "loss": 0.2179, + "pred_label": 4007.25, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -0.05676973983645439, + "rewards/margins": 2.8793885707855225, + "rewards/rejected": -2.9361586570739746, + "step": 1690, + "use_label": 2733.75 + }, + { + "epoch": 0.89, + "learning_rate": 6.108202443280977e-08, + "logits/chosen": -0.8465694189071655, + "logits/rejected": -0.9132062792778015, + "logps/chosen": -346.77032470703125, + "logps/rejected": -317.15716552734375, + "loss": 0.1967, + "pred_label": 4035.75, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.8769257664680481, + "rewards/margins": 3.0119924545288086, + "rewards/rejected": -3.88891863822937, + "step": 1700, + "use_label": 2745.25 + }, + { + "epoch": 0.9, + "learning_rate": 5.8173356602675974e-08, + "logits/chosen": -0.9974952936172485, + "logits/rejected": -0.9654536247253418, + "logps/chosen": -257.2732849121094, + "logps/rejected": -298.55157470703125, + "loss": 0.1809, + "pred_label": 4066.35009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -2.23062801361084, + "rewards/margins": 2.8383820056915283, + "rewards/rejected": -5.069010257720947, + "step": 1710, + "use_label": 2754.64990234375 + }, + { + "epoch": 0.9, + "learning_rate": 5.5264688772542175e-08, + "logits/chosen": -0.891338050365448, + "logits/rejected": -0.9084685444831848, + "logps/chosen": -470.7725524902344, + "logps/rejected": -485.2267150878906, + "loss": 0.1991, + "pred_label": 4097.75, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.26461061835289, + "rewards/margins": 3.3599886894226074, + "rewards/rejected": -3.6245994567871094, + "step": 1720, + "use_label": 2763.25 + }, + { + "epoch": 0.91, + "learning_rate": 5.2356020942408376e-08, + "logits/chosen": -0.8564783334732056, + "logits/rejected": -0.9062970876693726, + "logps/chosen": -296.5504455566406, + "logps/rejected": -302.5833740234375, + "loss": 0.1845, + "pred_label": 4129.0498046875, + "rewards/accuracies": 0.625, + "rewards/chosen": -1.253291130065918, + "rewards/margins": 1.962683916091919, + "rewards/rejected": -3.215975284576416, + "step": 1730, + "use_label": 2771.949951171875 + }, + { + "epoch": 0.91, + "learning_rate": 4.944735311227458e-08, + "logits/chosen": -0.8973791003227234, + "logits/rejected": -0.842978835105896, + "logps/chosen": -309.6785583496094, + "logps/rejected": -285.935791015625, + "loss": 0.2085, + "pred_label": 4161.39990234375, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.08151117712259293, + "rewards/margins": 3.9789955615997314, + "rewards/rejected": -3.897484540939331, + "step": 1740, + "use_label": 2779.60009765625 + }, + { + "epoch": 0.92, + "learning_rate": 4.653868528214078e-08, + "logits/chosen": -0.9628440141677856, + "logits/rejected": -1.0249922275543213, + "logps/chosen": -376.8697814941406, + "logps/rejected": -336.0437316894531, + "loss": 0.194, + "pred_label": 4192.5, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -1.799761414527893, + "rewards/margins": 3.394575595855713, + "rewards/rejected": -5.194336891174316, + "step": 1750, + "use_label": 2788.5 + }, + { + "epoch": 0.92, + "learning_rate": 4.363001745200698e-08, + "logits/chosen": -0.9686734080314636, + "logits/rejected": -1.011160135269165, + "logps/chosen": -344.68109130859375, + "logps/rejected": -338.3143615722656, + "loss": 0.1822, + "pred_label": 4224.9501953125, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.9003709554672241, + "rewards/margins": 4.197125434875488, + "rewards/rejected": -3.2967541217803955, + "step": 1760, + "use_label": 2796.050048828125 + }, + { + "epoch": 0.93, + "learning_rate": 4.072134962187318e-08, + "logits/chosen": -1.0619252920150757, + "logits/rejected": -1.0584673881530762, + "logps/chosen": -443.986083984375, + "logps/rejected": -363.45660400390625, + "loss": 0.2066, + "pred_label": 4258.4501953125, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.09187255799770355, + "rewards/margins": 2.6800460815429688, + "rewards/rejected": -2.5881736278533936, + "step": 1770, + "use_label": 2802.550048828125 + }, + { + "epoch": 0.93, + "learning_rate": 3.781268179173938e-08, + "logits/chosen": -0.9357656240463257, + "logits/rejected": -0.9777708053588867, + "logps/chosen": -364.95159912109375, + "logps/rejected": -321.37890625, + "loss": 0.1591, + "pred_label": 4284.2998046875, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.44581514596939087, + "rewards/margins": 3.383704423904419, + "rewards/rejected": -2.9378890991210938, + "step": 1780, + "use_label": 2816.699951171875 + }, + { + "epoch": 0.94, + "learning_rate": 3.490401396160558e-08, + "logits/chosen": -0.9476548433303833, + "logits/rejected": -0.9675912857055664, + "logps/chosen": -278.46197509765625, + "logps/rejected": -292.71759033203125, + "loss": 0.193, + "pred_label": 4318.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.4520017206668854, + "rewards/margins": 3.4121756553649902, + "rewards/rejected": -3.864177703857422, + "step": 1790, + "use_label": 2823.0 + }, + { + "epoch": 0.94, + "learning_rate": 3.199534613147178e-08, + "logits/chosen": -0.8004802465438843, + "logits/rejected": -0.8716949224472046, + "logps/chosen": -337.95367431640625, + "logps/rejected": -374.6369934082031, + "loss": 0.2068, + "pred_label": 4347.14990234375, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.5371319651603699, + "rewards/margins": 3.537078380584717, + "rewards/rejected": -4.074210166931152, + "step": 1800, + "use_label": 2833.85009765625 + }, + { + "epoch": 0.95, + "learning_rate": 2.9086678301337987e-08, + "logits/chosen": -1.0241055488586426, + "logits/rejected": -1.0045052766799927, + "logps/chosen": -336.6506042480469, + "logps/rejected": -323.0858459472656, + "loss": 0.1695, + "pred_label": 4374.85009765625, + "rewards/accuracies": 0.6000000238418579, + "rewards/chosen": -1.255857229232788, + "rewards/margins": 2.3176028728485107, + "rewards/rejected": -3.573460102081299, + "step": 1810, + "use_label": 2846.14990234375 + }, + { + "epoch": 0.95, + "learning_rate": 2.6178010471204188e-08, + "logits/chosen": -0.9693848490715027, + "logits/rejected": -0.9834591150283813, + "logps/chosen": -423.89404296875, + "logps/rejected": -360.3699035644531, + "loss": 0.1947, + "pred_label": 4406.10009765625, + "rewards/accuracies": 0.8500000238418579, + "rewards/chosen": 0.7960270047187805, + "rewards/margins": 5.266651153564453, + "rewards/rejected": -4.470623970031738, + "step": 1820, + "use_label": 2854.89990234375 + }, + { + "epoch": 0.96, + "learning_rate": 2.326934264107039e-08, + "logits/chosen": -0.8095889091491699, + "logits/rejected": -0.942054271697998, + "logps/chosen": -363.0487365722656, + "logps/rejected": -354.41607666015625, + "loss": 0.155, + "pred_label": 4439.2001953125, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": -1.144942045211792, + "rewards/margins": 2.5031371116638184, + "rewards/rejected": -3.6480789184570312, + "step": 1830, + "use_label": 2861.800048828125 + }, + { + "epoch": 0.96, + "learning_rate": 2.036067481093659e-08, + "logits/chosen": -0.8409671783447266, + "logits/rejected": -0.83684903383255, + "logps/chosen": -343.324951171875, + "logps/rejected": -362.7954406738281, + "loss": 0.1852, + "pred_label": 4470.85009765625, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.28511303663253784, + "rewards/margins": 3.413196563720703, + "rewards/rejected": -3.6983096599578857, + "step": 1840, + "use_label": 2870.14990234375 + }, + { + "epoch": 0.97, + "learning_rate": 1.745200698080279e-08, + "logits/chosen": -0.9692668914794922, + "logits/rejected": -0.9877262115478516, + "logps/chosen": -347.4239196777344, + "logps/rejected": -361.1467590332031, + "loss": 0.1698, + "pred_label": 4502.25, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.050318337976932526, + "rewards/margins": 4.280552864074707, + "rewards/rejected": -4.330872058868408, + "step": 1850, + "use_label": 2878.75 + }, + { + "epoch": 0.97, + "learning_rate": 1.4543339150668994e-08, + "logits/chosen": -0.9399221539497375, + "logits/rejected": -0.9900743365287781, + "logps/chosen": -236.5078582763672, + "logps/rejected": -264.27252197265625, + "loss": 0.186, + "pred_label": 4531.4501953125, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.7518811225891113, + "rewards/margins": 4.044899940490723, + "rewards/rejected": -3.293017864227295, + "step": 1860, + "use_label": 2889.550048828125 + }, + { + "epoch": 0.98, + "learning_rate": 1.1634671320535195e-08, + "logits/chosen": -1.0314300060272217, + "logits/rejected": -1.0113561153411865, + "logps/chosen": -360.2470703125, + "logps/rejected": -362.8114013671875, + "loss": 0.177, + "pred_label": 4561.4501953125, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.7401183843612671, + "rewards/margins": 2.7796757221221924, + "rewards/rejected": -3.51979398727417, + "step": 1870, + "use_label": 2899.550048828125 + }, + { + "epoch": 0.98, + "learning_rate": 8.726003490401395e-09, + "logits/chosen": -1.0057262182235718, + "logits/rejected": -1.0620293617248535, + "logps/chosen": -371.9859619140625, + "logps/rejected": -325.24932861328125, + "loss": 0.1686, + "pred_label": 4594.35009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.7309083938598633, + "rewards/margins": 2.8372626304626465, + "rewards/rejected": -3.5681710243225098, + "step": 1880, + "use_label": 2906.64990234375 + }, + { + "epoch": 0.99, + "learning_rate": 5.817335660267597e-09, + "logits/chosen": -0.9105908274650574, + "logits/rejected": -0.8740211725234985, + "logps/chosen": -344.8439025878906, + "logps/rejected": -438.42010498046875, + "loss": 0.1399, + "pred_label": 4625.9501953125, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -1.2992219924926758, + "rewards/margins": 3.6406044960021973, + "rewards/rejected": -4.939826011657715, + "step": 1890, + "use_label": 2915.050048828125 + }, + { + "epoch": 0.99, + "learning_rate": 2.9086678301337986e-09, + "logits/chosen": -1.0261294841766357, + "logits/rejected": -0.9415041208267212, + "logps/chosen": -310.5721130371094, + "logps/rejected": -294.1573486328125, + "loss": 0.1918, + "pred_label": 4658.85009765625, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -1.248200535774231, + "rewards/margins": 2.8387322425842285, + "rewards/rejected": -4.086933135986328, + "step": 1900, + "use_label": 2922.14990234375 + }, + { + "epoch": 1.0, + "learning_rate": 0.0, + "logits/chosen": -0.9566202163696289, + "logits/rejected": -0.9806071519851685, + "logps/chosen": -324.8318786621094, + "logps/rejected": -311.9029846191406, + "loss": 0.1714, + "pred_label": 4690.7998046875, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.9383823275566101, + "rewards/margins": 2.941167116165161, + "rewards/rejected": -3.879549741744995, + "step": 1910, + "use_label": 2930.199951171875 + }, + { + "epoch": 1.0, + "eval_logits/chosen": -0.9365912079811096, + "eval_logits/rejected": -0.926014244556427, + "eval_logps/chosen": -352.8179626464844, + "eval_logps/rejected": -349.7088317871094, + "eval_loss": 0.18906600773334503, + "eval_pred_label": 4808.27001953125, + "eval_rewards/accuracies": 0.7579365372657776, + "eval_rewards/chosen": 0.2332553267478943, + "eval_rewards/margins": 3.972490072250366, + "eval_rewards/rejected": -3.7392351627349854, + "eval_runtime": 284.6758, + "eval_samples_per_second": 7.026, + "eval_steps_per_second": 0.221, + "eval_use_label": 2959.730224609375, + "step": 1910 + }, + { + "epoch": 1.0, + "step": 1910, + "total_flos": 0.0, + "train_loss": 0.2926436502271922, + "train_runtime": 13370.5554, + "train_samples_per_second": 4.572, + "train_steps_per_second": 0.143 + } + ], + "logging_steps": 10, + "max_steps": 1910, + "num_train_epochs": 1, + "save_steps": 50, + "total_flos": 0.0, + "trial_name": null, + "trial_params": null +}