diff --git "a/trainer_state.json" "b/trainer_state.json" --- "a/trainer_state.json" +++ "b/trainer_state.json" @@ -1,20 +1,20 @@ { "best_metric": null, "best_model_checkpoint": null, - "epoch": 0.9997382884061764, + "epoch": 1.0, "eval_steps": 100, - "global_step": 1910, + "global_step": 956, "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, + "learning_rate": 5.208333333333333e-09, + "logits/chosen": -3.0465383529663086, + "logits/rejected": -2.7803585529327393, + "logps/chosen": -262.69439697265625, + "logps/rejected": -200.3923797607422, "loss": 0.6931, "pred_label": 0.0, "rewards/accuracies": 0.0, @@ -22,3094 +22,1702 @@ "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1, - "use_label": 3.0 + "use_label": 0.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, + "learning_rate": 5.208333333333333e-08, + "logits/chosen": -2.765808582305908, + "logits/rejected": -2.763443946838379, + "logps/chosen": -299.00396728515625, + "logps/rejected": -250.18435668945312, + "loss": 0.6941, + "pred_label": 0.0, + "rewards/accuracies": 0.4444444477558136, + "rewards/chosen": 0.0011364308884367347, + "rewards/margins": 0.0017329632537439466, + "rewards/rejected": -0.0005965338204987347, "step": 10, - "use_label": 23.0 + "use_label": 0.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, + "epoch": 0.02, + "learning_rate": 1.0416666666666667e-07, + "logits/chosen": -2.7105605602264404, + "logits/rejected": -2.742635726928711, + "logps/chosen": -221.0091552734375, + "logps/rejected": -200.49070739746094, + "loss": 0.6855, + "pred_label": 0.0, + "rewards/accuracies": 0.5375000238418579, + "rewards/chosen": -0.006218413356691599, + "rewards/margins": 0.027181172743439674, + "rewards/rejected": -0.03339958190917969, "step": 20, - "use_label": 61.0 + "use_label": 0.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, + "epoch": 0.03, + "learning_rate": 1.5624999999999999e-07, + "logits/chosen": -2.7838003635406494, + "logits/rejected": -2.7471749782562256, + "logps/chosen": -286.62408447265625, + "logps/rejected": -256.59033203125, + "loss": 0.6656, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.032594505697488785, + "rewards/margins": 0.0683099776506424, + "rewards/rejected": -0.03571547567844391, "step": 30, - "use_label": 101.0 + "use_label": 0.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, + "epoch": 0.04, + "learning_rate": 2.0833333333333333e-07, + "logits/chosen": -2.6903152465820312, + "logits/rejected": -2.681668996810913, + "logps/chosen": -263.98040771484375, + "logps/rejected": -253.0127716064453, + "loss": 0.6449, + "pred_label": 0.0, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.07213224470615387, + "rewards/margins": 0.12824633717536926, + "rewards/rejected": -0.05611409991979599, "step": 40, - "use_label": 141.0 + "use_label": 0.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, + "epoch": 0.05, + "learning_rate": 2.604166666666667e-07, + "logits/chosen": -2.7848479747772217, + "logits/rejected": -2.7641561031341553, + "logps/chosen": -302.9726257324219, + "logps/rejected": -290.980712890625, + "loss": 0.63, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.0347287543118, + "rewards/margins": 0.08969736844301224, + "rewards/rejected": -0.12442612648010254, "step": 50, - "use_label": 181.0 + "use_label": 0.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, + "epoch": 0.06, + "learning_rate": 3.1249999999999997e-07, + "logits/chosen": -2.8073248863220215, + "logits/rejected": -2.7780768871307373, + "logps/chosen": -274.9972229003906, + "logps/rejected": -271.43719482421875, + "loss": 0.6385, "pred_label": 0.0, - "rewards/accuracies": 0.675000011920929, - "rewards/chosen": 0.10490015894174576, - "rewards/margins": 0.06382963806390762, - "rewards/rejected": 0.041070520877838135, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.0038132492918521166, + "rewards/margins": 0.19520750641822815, + "rewards/rejected": -0.19139425456523895, "step": 60, - "use_label": 221.0 + "use_label": 0.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, + "epoch": 0.07, + "learning_rate": 3.645833333333333e-07, + "logits/chosen": -2.8025059700012207, + "logits/rejected": -2.8569769859313965, + "logps/chosen": -308.70867919921875, + "logps/rejected": -294.216796875, + "loss": 0.5652, "pred_label": 0.0, "rewards/accuracies": 0.6499999761581421, - "rewards/chosen": 0.29588761925697327, - "rewards/margins": 0.1510847806930542, - "rewards/rejected": 0.14480280876159668, + "rewards/chosen": 0.10104294866323471, + "rewards/margins": 0.25741106271743774, + "rewards/rejected": -0.15636806190013885, "step": 70, - "use_label": 261.0 + "use_label": 0.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, + "epoch": 0.08, + "learning_rate": 4.1666666666666667e-07, + "logits/chosen": -2.659728527069092, + "logits/rejected": -2.5561347007751465, + "logps/chosen": -282.79150390625, + "logps/rejected": -264.9902648925781, + "loss": 0.5422, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.1746046096086502, + "rewards/margins": 0.6943714022636414, + "rewards/rejected": -0.5197668075561523, "step": 80, - "use_label": 301.0 + "use_label": 0.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, + "epoch": 0.09, + "learning_rate": 4.6874999999999996e-07, + "logits/chosen": -2.746777057647705, + "logits/rejected": -2.6942930221557617, + "logps/chosen": -290.62408447265625, + "logps/rejected": -270.763427734375, + "loss": 0.5694, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.2031947821378708, + "rewards/margins": 0.5417487025260925, + "rewards/rejected": -0.3385539650917053, "step": 90, - "use_label": 340.6000061035156 + "use_label": 0.0 }, { - "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, + "epoch": 0.1, + "learning_rate": 4.976744186046512e-07, + "logits/chosen": -2.6346707344055176, + "logits/rejected": -2.666553258895874, + "logps/chosen": -344.0323486328125, + "logps/rejected": -350.5570373535156, + "loss": 0.6368, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.40496858954429626, + "rewards/margins": 0.16033101081848145, + "rewards/rejected": -0.5652996301651001, "step": 100, - "use_label": 380.0 + "use_label": 0.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, + "epoch": 0.1, + "eval_logits/chosen": -2.824883460998535, + "eval_logits/rejected": -2.8082714080810547, + "eval_logps/chosen": -286.7236022949219, + "eval_logps/rejected": -265.6400451660156, + "eval_loss": 0.6040099859237671, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.6746031641960144, + "eval_rewards/chosen": 0.17999570071697235, + "eval_rewards/margins": 0.5800058245658875, + "eval_rewards/rejected": -0.4000100791454315, + "eval_runtime": 152.1323, + "eval_samples_per_second": 13.146, + "eval_steps_per_second": 0.414, + "eval_use_label": 0.0, + "step": 100 + }, + { + "epoch": 0.12, + "learning_rate": 4.91860465116279e-07, + "logits/chosen": -2.742032527923584, + "logits/rejected": -2.74245023727417, + "logps/chosen": -242.73056030273438, + "logps/rejected": -212.1634063720703, + "loss": 0.6054, + "pred_label": 0.0, + "rewards/accuracies": 0.625, + "rewards/chosen": 0.16584378480911255, + "rewards/margins": 0.5073415637016296, + "rewards/rejected": -0.3414977788925171, "step": 110, - "use_label": 420.0 + "use_label": 0.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, + "epoch": 0.13, + "learning_rate": 4.860465116279069e-07, + "logits/chosen": -2.648292064666748, + "logits/rejected": -2.6121315956115723, + "logps/chosen": -278.00640869140625, + "logps/rejected": -268.49859619140625, + "loss": 0.5983, + "pred_label": 0.0, "rewards/accuracies": 0.7250000238418579, - "rewards/chosen": 0.5013664364814758, - "rewards/margins": 0.28251418471336365, - "rewards/rejected": 0.21885231137275696, + "rewards/chosen": 0.15003331005573273, + "rewards/margins": 0.9967330098152161, + "rewards/rejected": -0.8466997146606445, "step": 120, - "use_label": 459.79998779296875 + "use_label": 0.0 }, { - "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, + "epoch": 0.14, + "learning_rate": 4.802325581395348e-07, + "logits/chosen": -2.7574169635772705, + "logits/rejected": -2.726195812225342, + "logps/chosen": -325.28851318359375, + "logps/rejected": -264.0757751464844, + "loss": 0.5869, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.13045839965343475, + "rewards/margins": 0.4203871190547943, + "rewards/rejected": -0.28992873430252075, "step": 130, - "use_label": 498.25 + "use_label": 0.0 }, { - "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, + "epoch": 0.15, + "learning_rate": 4.7441860465116277e-07, + "logits/chosen": -2.715982675552368, + "logits/rejected": -2.7288689613342285, + "logps/chosen": -239.6348419189453, + "logps/rejected": -251.3628692626953, + "loss": 0.5227, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.2861214280128479, + "rewards/margins": 0.9334532618522644, + "rewards/rejected": -0.6473318338394165, "step": 140, - "use_label": 536.1500244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.16, + "learning_rate": 4.6860465116279066e-07, + "logits/chosen": -2.7235076427459717, + "logits/rejected": -2.73270845413208, + "logps/chosen": -269.925048828125, + "logps/rejected": -257.32135009765625, + "loss": 0.5829, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.3011024594306946, + "rewards/margins": 0.7531024217605591, + "rewards/rejected": -0.4519999027252197, "step": 150, - "use_label": 570.2000122070312 + "use_label": 0.0 }, { - "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, + "epoch": 0.17, + "learning_rate": 4.627906976744186e-07, + "logits/chosen": -2.7294018268585205, + "logits/rejected": -2.7083096504211426, + "logps/chosen": -271.4284973144531, + "logps/rejected": -311.17718505859375, + "loss": 0.6106, + "pred_label": 0.0, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": 0.2270795851945877, + "rewards/margins": 0.7130987048149109, + "rewards/rejected": -0.4860190749168396, "step": 160, - "use_label": 604.7999877929688 + "use_label": 0.0 }, { - "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, + "epoch": 0.18, + "learning_rate": 4.569767441860465e-07, + "logits/chosen": -2.7488739490509033, + "logits/rejected": -2.731077194213867, + "logps/chosen": -296.81561279296875, + "logps/rejected": -279.2462158203125, + "loss": 0.5766, + "pred_label": 0.0, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.3102962076663971, + "rewards/margins": 0.6582878828048706, + "rewards/rejected": -0.3479916751384735, "step": 170, - "use_label": 635.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.19, + "learning_rate": 4.511627906976744e-07, + "logits/chosen": -2.678403377532959, + "logits/rejected": -2.6578478813171387, + "logps/chosen": -238.8046875, + "logps/rejected": -242.45297241210938, + "loss": 0.535, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": 0.11773022264242172, + "rewards/margins": 0.943315863609314, + "rewards/rejected": -0.8255857229232788, "step": 180, - "use_label": 662.9500122070312 + "use_label": 0.0 }, { - "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, + "epoch": 0.2, + "learning_rate": 4.4534883720930233e-07, + "logits/chosen": -2.7577908039093018, + "logits/rejected": -2.7466981410980225, + "logps/chosen": -255.8538818359375, + "logps/rejected": -249.35043334960938, + "loss": 0.515, + "pred_label": 0.0, + "rewards/accuracies": 0.8125, + "rewards/chosen": 0.29801803827285767, + "rewards/margins": 0.9082610011100769, + "rewards/rejected": -0.610243022441864, "step": 190, - "use_label": 693.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.21, + "learning_rate": 4.395348837209302e-07, + "logits/chosen": -2.726404905319214, + "logits/rejected": -2.684250593185425, + "logps/chosen": -314.0807189941406, + "logps/rejected": -294.7442626953125, + "loss": 0.558, + "pred_label": 0.0, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": 0.19561365246772766, + "rewards/margins": 0.6523147225379944, + "rewards/rejected": -0.45670104026794434, "step": 200, - "use_label": 725.4000244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.21, + "eval_logits/chosen": -2.808067798614502, + "eval_logits/rejected": -2.7980899810791016, + "eval_logps/chosen": -287.2000732421875, + "eval_logps/rejected": -269.501953125, + "eval_loss": 0.5651848316192627, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7420634627342224, + "eval_rewards/chosen": 0.1323493868112564, + "eval_rewards/margins": 0.9185545444488525, + "eval_rewards/rejected": -0.7862052321434021, + "eval_runtime": 151.5393, + "eval_samples_per_second": 13.198, + "eval_steps_per_second": 0.416, + "eval_use_label": 0.0, + "step": 200 + }, + { + "epoch": 0.22, + "learning_rate": 4.337209302325581e-07, + "logits/chosen": -2.7738518714904785, + "logits/rejected": -2.76870059967041, + "logps/chosen": -309.36224365234375, + "logps/rejected": -252.21908569335938, + "loss": 0.5574, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.1390099674463272, + "rewards/margins": 0.8792921900749207, + "rewards/rejected": -0.7402822375297546, "step": 210, - "use_label": 753.6500244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.23, + "learning_rate": 4.27906976744186e-07, + "logits/chosen": -2.7275898456573486, + "logits/rejected": -2.7462148666381836, + "logps/chosen": -330.2643127441406, + "logps/rejected": -309.7840576171875, + "loss": 0.545, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.24796506762504578, + "rewards/margins": 0.912385106086731, + "rewards/rejected": -0.6644200086593628, "step": 220, - "use_label": 783.0499877929688 + "use_label": 0.0 }, { - "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, + "epoch": 0.24, + "learning_rate": 4.220930232558139e-07, + "logits/chosen": -2.656540870666504, + "logits/rejected": -2.662473678588867, + "logps/chosen": -303.71685791015625, + "logps/rejected": -266.63275146484375, + "loss": 0.552, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.06451065838336945, + "rewards/margins": 0.7794097661972046, + "rewards/rejected": -0.8439203500747681, "step": 230, - "use_label": 810.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.25, + "learning_rate": 4.162790697674418e-07, + "logits/chosen": -2.670305013656616, + "logits/rejected": -2.6477558612823486, + "logps/chosen": -266.712158203125, + "logps/rejected": -217.30209350585938, + "loss": 0.5225, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.07932360470294952, + "rewards/margins": 0.7470332384109497, + "rewards/rejected": -0.8263567686080933, "step": 240, - "use_label": 837.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.26, + "learning_rate": 4.104651162790698e-07, + "logits/chosen": -2.7736430168151855, + "logits/rejected": -2.734602928161621, + "logps/chosen": -282.9905090332031, + "logps/rejected": -334.78863525390625, + "loss": 0.5718, + "pred_label": 0.0, "rewards/accuracies": 0.6499999761581421, - "rewards/chosen": 1.2634034156799316, - "rewards/margins": 0.8081910014152527, - "rewards/rejected": 0.4552123546600342, + "rewards/chosen": 0.052515070885419846, + "rewards/margins": 0.7567065954208374, + "rewards/rejected": -0.7041915655136108, "step": 250, - "use_label": 862.75 + "use_label": 0.0 }, { - "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, + "epoch": 0.27, + "learning_rate": 4.046511627906977e-07, + "logits/chosen": -2.671180009841919, + "logits/rejected": -2.6478466987609863, + "logps/chosen": -248.6728057861328, + "logps/rejected": -269.4361267089844, + "loss": 0.5466, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.02055490016937256, + "rewards/margins": 0.9694421887397766, + "rewards/rejected": -0.9488871693611145, "step": 260, - "use_label": 891.0999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.28, + "learning_rate": 3.9883720930232557e-07, + "logits/chosen": -2.7515838146209717, + "logits/rejected": -2.805983543395996, + "logps/chosen": -296.47076416015625, + "logps/rejected": -235.0985565185547, + "loss": 0.5395, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.11441866308450699, + "rewards/margins": 0.8921812772750854, + "rewards/rejected": -0.7777627110481262, "step": 270, - "use_label": 917.0499877929688 + "use_label": 0.0 }, { - "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, + "epoch": 0.29, + "learning_rate": 3.9302325581395346e-07, + "logits/chosen": -2.771331310272217, + "logits/rejected": -2.799328327178955, + "logps/chosen": -299.34796142578125, + "logps/rejected": -297.7183837890625, + "loss": 0.5239, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.07190613448619843, + "rewards/margins": 0.8066242933273315, + "rewards/rejected": -0.8785305023193359, "step": 280, - "use_label": 947.6500244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.3, + "learning_rate": 3.8720930232558135e-07, + "logits/chosen": -2.7138657569885254, + "logits/rejected": -2.7323174476623535, + "logps/chosen": -290.46563720703125, + "logps/rejected": -261.27117919921875, + "loss": 0.5757, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.09668038040399551, + "rewards/margins": 0.9626511335372925, + "rewards/rejected": -1.0593315362930298, "step": 290, - "use_label": 974.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.31, + "learning_rate": 3.813953488372093e-07, + "logits/chosen": -2.659846782684326, + "logits/rejected": -2.6316401958465576, + "logps/chosen": -293.4468078613281, + "logps/rejected": -243.309326171875, + "loss": 0.553, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.0554303340613842, + "rewards/margins": 1.0153982639312744, + "rewards/rejected": -1.0708284378051758, "step": 300, - "use_label": 1001.9000244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.31, + "eval_logits/chosen": -2.751713991165161, + "eval_logits/rejected": -2.742063045501709, + "eval_logps/chosen": -289.1977844238281, + "eval_logps/rejected": -272.0630187988281, + "eval_loss": 0.5431597828865051, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7341269850730896, + "eval_rewards/chosen": -0.06741873919963837, + "eval_rewards/margins": 0.9748923778533936, + "eval_rewards/rejected": -1.042311191558838, + "eval_runtime": 151.828, + "eval_samples_per_second": 13.173, + "eval_steps_per_second": 0.415, + "eval_use_label": 0.0, + "step": 300 + }, + { + "epoch": 0.32, + "learning_rate": 3.755813953488372e-07, + "logits/chosen": -2.612884283065796, + "logits/rejected": -2.65791916847229, + "logps/chosen": -306.4892578125, + "logps/rejected": -278.91619873046875, + "loss": 0.4974, + "pred_label": 0.0, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.24395442008972168, + "rewards/margins": 1.114572525024414, + "rewards/rejected": -0.8706180453300476, "step": 310, - "use_label": 1026.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.33, + "learning_rate": 3.697674418604651e-07, + "logits/chosen": -2.6776044368743896, + "logits/rejected": -2.622297525405884, + "logps/chosen": -270.759033203125, + "logps/rejected": -289.8017578125, + "loss": 0.5455, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.07885434478521347, + "rewards/margins": 1.0345350503921509, + "rewards/rejected": -1.1133893728256226, "step": 320, - "use_label": 1050.550048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.35, + "learning_rate": 3.63953488372093e-07, + "logits/chosen": -2.716804027557373, + "logits/rejected": -2.640516996383667, + "logps/chosen": -262.74810791015625, + "logps/rejected": -253.93417358398438, + "loss": 0.5639, + "pred_label": 0.0, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.14604653418064117, + "rewards/margins": 0.922707736492157, + "rewards/rejected": -0.7766610980033875, "step": 330, - "use_label": 1073.199951171875 + "use_label": 0.0 }, { - "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, + "epoch": 0.36, + "learning_rate": 3.581395348837209e-07, + "logits/chosen": -2.7240614891052246, + "logits/rejected": -2.6947460174560547, + "logps/chosen": -238.51412963867188, + "logps/rejected": -228.6656494140625, + "loss": 0.5925, + "pred_label": 0.0, + "rewards/accuracies": 0.612500011920929, + "rewards/chosen": -0.19918441772460938, + "rewards/margins": 0.44254446029663086, + "rewards/rejected": -0.6417288184165955, "step": 340, - "use_label": 1093.300048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.37, + "learning_rate": 3.5232558139534886e-07, + "logits/chosen": -2.7215187549591064, + "logits/rejected": -2.7599117755889893, + "logps/chosen": -307.45989990234375, + "logps/rejected": -334.6233825683594, + "loss": 0.577, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": 0.1541057825088501, + "rewards/margins": 1.1278841495513916, + "rewards/rejected": -0.9737783670425415, "step": 350, - "use_label": 1115.5 + "use_label": 0.0 }, { - "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, + "epoch": 0.38, + "learning_rate": 3.4651162790697675e-07, + "logits/chosen": -2.7533516883850098, + "logits/rejected": -2.8209774494171143, + "logps/chosen": -291.31658935546875, + "logps/rejected": -252.8400421142578, + "loss": 0.5715, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": 0.08639977127313614, + "rewards/margins": 0.8964889645576477, + "rewards/rejected": -0.8100892305374146, "step": 360, - "use_label": 1135.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.39, + "learning_rate": 3.4069767441860464e-07, + "logits/chosen": -2.7346293926239014, + "logits/rejected": -2.712937355041504, + "logps/chosen": -243.9089813232422, + "logps/rejected": -270.5567321777344, + "loss": 0.524, + "pred_label": 0.0, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.014877429232001305, + "rewards/margins": 0.8318921327590942, + "rewards/rejected": -0.8170146942138672, "step": 370, - "use_label": 1157.449951171875 + "use_label": 0.0 }, { - "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, + "epoch": 0.4, + "learning_rate": 3.3488372093023253e-07, + "logits/chosen": -2.820455312728882, + "logits/rejected": -2.7536873817443848, + "logps/chosen": -331.84075927734375, + "logps/rejected": -300.8788146972656, + "loss": 0.5174, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.10691157728433609, + "rewards/margins": 0.881051242351532, + "rewards/rejected": -0.9879627227783203, "step": 380, - "use_label": 1174.4000244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.41, + "learning_rate": 3.290697674418604e-07, + "logits/chosen": -2.560523509979248, + "logits/rejected": -2.5691637992858887, + "logps/chosen": -262.97808837890625, + "logps/rejected": -238.43594360351562, + "loss": 0.5326, + "pred_label": 0.0, + "rewards/accuracies": 0.6875, + "rewards/chosen": 0.1129593625664711, + "rewards/margins": 0.8537227511405945, + "rewards/rejected": -0.7407633662223816, "step": 390, - "use_label": 1195.050048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.42, + "learning_rate": 3.232558139534883e-07, + "logits/chosen": -2.748053789138794, + "logits/rejected": -2.720460891723633, + "logps/chosen": -307.76116943359375, + "logps/rejected": -274.69989013671875, + "loss": 0.5019, + "pred_label": 0.0, + "rewards/accuracies": 0.625, + "rewards/chosen": -0.10151912271976471, + "rewards/margins": 0.7108665704727173, + "rewards/rejected": -0.8123855590820312, "step": 400, - "use_label": 1218.199951171875 + "use_label": 0.0 }, { - "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, + "epoch": 0.42, + "eval_logits/chosen": -2.796114206314087, + "eval_logits/rejected": -2.7870821952819824, + "eval_logps/chosen": -287.2944030761719, + "eval_logps/rejected": -270.9002990722656, + "eval_loss": 0.5371336936950684, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7539682388305664, + "eval_rewards/chosen": 0.1229194775223732, + "eval_rewards/margins": 1.0489554405212402, + "eval_rewards/rejected": -0.9260359406471252, + "eval_runtime": 151.8781, + "eval_samples_per_second": 13.168, + "eval_steps_per_second": 0.415, + "eval_use_label": 0.0, + "step": 400 + }, + { + "epoch": 0.43, + "learning_rate": 3.1744186046511626e-07, + "logits/chosen": -2.8114898204803467, + "logits/rejected": -2.7443149089813232, + "logps/chosen": -308.30804443359375, + "logps/rejected": -297.49749755859375, + "loss": 0.5548, + "pred_label": 0.0, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.004237680230289698, + "rewards/margins": 0.7485671043395996, + "rewards/rejected": -0.7528048157691956, "step": 410, - "use_label": 1238.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.44, + "learning_rate": 3.116279069767442e-07, + "logits/chosen": -2.6755497455596924, + "logits/rejected": -2.623680591583252, + "logps/chosen": -251.78466796875, + "logps/rejected": -202.69125366210938, + "loss": 0.5576, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": 0.09348449110984802, + "rewards/margins": 0.8450796008110046, + "rewards/rejected": -0.751595139503479, "step": 420, - "use_label": 1257.800048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.45, + "learning_rate": 3.058139534883721e-07, + "logits/chosen": -2.7282052040100098, + "logits/rejected": -2.6777048110961914, + "logps/chosen": -286.0669860839844, + "logps/rejected": -259.3023376464844, + "loss": 0.5476, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": 0.2035273015499115, + "rewards/margins": 0.9668741226196289, + "rewards/rejected": -0.763346791267395, "step": 430, - "use_label": 1275.800048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.46, + "learning_rate": 3e-07, + "logits/chosen": -2.611921787261963, + "logits/rejected": -2.676251173019409, + "logps/chosen": -278.88824462890625, + "logps/rejected": -255.1051788330078, + "loss": 0.5149, + "pred_label": 0.0, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": 0.1889255791902542, + "rewards/margins": 0.9589231610298157, + "rewards/rejected": -0.7699976563453674, "step": 440, - "use_label": 1295.949951171875 + "use_label": 0.0 }, { - "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, + "epoch": 0.47, + "learning_rate": 2.941860465116279e-07, + "logits/chosen": -2.58278226852417, + "logits/rejected": -2.551997423171997, + "logps/chosen": -274.6985778808594, + "logps/rejected": -328.51422119140625, + "loss": 0.5667, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": 0.1181306466460228, + "rewards/margins": 1.005082368850708, + "rewards/rejected": -0.8869515657424927, "step": 450, - "use_label": 1314.0 + "use_label": 0.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, + "epoch": 0.48, + "learning_rate": 2.883720930232558e-07, + "logits/chosen": -2.6847758293151855, + "logits/rejected": -2.6493821144104004, + "logps/chosen": -277.9380798339844, + "logps/rejected": -285.26324462890625, + "loss": 0.533, + "pred_label": 0.0, + "rewards/accuracies": 0.6499999761581421, + "rewards/chosen": 0.1580689549446106, + "rewards/margins": 1.3047949075698853, + "rewards/rejected": -1.146726131439209, "step": 460, - "use_label": 1329.550048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.49, + "learning_rate": 2.825581395348837e-07, + "logits/chosen": -2.6480162143707275, + "logits/rejected": -2.5979321002960205, + "logps/chosen": -279.9370422363281, + "logps/rejected": -275.7652587890625, + "loss": 0.5222, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": 0.015222841873764992, + "rewards/margins": 0.9643913507461548, + "rewards/rejected": -0.9491683840751648, "step": 470, - "use_label": 1348.0999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.5, + "learning_rate": 2.767441860465116e-07, + "logits/chosen": -2.643878698348999, + "logits/rejected": -2.5506885051727295, + "logps/chosen": -231.51913452148438, + "logps/rejected": -236.3944091796875, + "loss": 0.5759, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": -0.12248492240905762, + "rewards/margins": 0.6468067169189453, + "rewards/rejected": -0.7692916393280029, "step": 480, - "use_label": 1365.0 + "use_label": 0.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, + "epoch": 0.51, + "learning_rate": 2.709302325581395e-07, + "logits/chosen": -2.70454740524292, + "logits/rejected": -2.6749463081359863, + "logps/chosen": -275.100830078125, + "logps/rejected": -261.96124267578125, + "loss": 0.5667, + "pred_label": 0.0, + "rewards/accuracies": 0.699999988079071, + "rewards/chosen": -0.030958181247115135, + "rewards/margins": 0.7133909463882446, + "rewards/rejected": -0.7443491220474243, "step": 490, - "use_label": 1385.4000244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.52, + "learning_rate": 2.651162790697674e-07, + "logits/chosen": -2.6521294116973877, + "logits/rejected": -2.639777421951294, + "logps/chosen": -267.96221923828125, + "logps/rejected": -275.6853942871094, + "loss": 0.5303, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": 0.0007066220277920365, + "rewards/margins": 0.7814868688583374, + "rewards/rejected": -0.7807803153991699, "step": 500, - "use_label": 1401.4000244140625 + "use_label": 0.0 }, { - "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, + "epoch": 0.52, + "eval_logits/chosen": -2.7494895458221436, + "eval_logits/rejected": -2.74151349067688, + "eval_logps/chosen": -287.7681579589844, + "eval_logps/rejected": -271.1743469238281, + "eval_loss": 0.5362493991851807, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.738095223903656, + "eval_rewards/chosen": 0.07554519176483154, + "eval_rewards/margins": 1.0289891958236694, + "eval_rewards/rejected": -0.9534440040588379, + "eval_runtime": 152.1284, + "eval_samples_per_second": 13.147, + "eval_steps_per_second": 0.414, + "eval_use_label": 0.0, + "step": 500 + }, + { + "epoch": 0.53, + "learning_rate": 2.593023255813954e-07, + "logits/chosen": -2.6452856063842773, + "logits/rejected": -2.635679244995117, + "logps/chosen": -292.32196044921875, + "logps/rejected": -265.02752685546875, + "loss": 0.5349, + "pred_label": 0.0, "rewards/accuracies": 0.7749999761581421, - "rewards/chosen": 1.058019995689392, - "rewards/margins": 2.189736843109131, - "rewards/rejected": -1.1317169666290283, + "rewards/chosen": 0.0869629979133606, + "rewards/margins": 0.8606294393539429, + "rewards/rejected": -0.7736665606498718, "step": 510, - "use_label": 1417.300048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.54, + "learning_rate": 2.534883720930233e-07, + "logits/chosen": -2.707009792327881, + "logits/rejected": -2.695380926132202, + "logps/chosen": -273.6498107910156, + "logps/rejected": -243.3380889892578, + "loss": 0.5236, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": 0.14563575387001038, + "rewards/margins": 1.0186597108840942, + "rewards/rejected": -0.8730238676071167, "step": 520, - "use_label": 1433.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.55, + "learning_rate": 2.4767441860465117e-07, + "logits/chosen": -2.677058458328247, + "logits/rejected": -2.6737570762634277, + "logps/chosen": -270.5714416503906, + "logps/rejected": -275.0865783691406, + "loss": 0.5015, + "pred_label": 0.0, "rewards/accuracies": 0.7250000238418579, - "rewards/chosen": 1.136601448059082, - "rewards/margins": 2.0457780361175537, - "rewards/rejected": -0.9091763496398926, + "rewards/chosen": -0.2971714735031128, + "rewards/margins": 1.0320571660995483, + "rewards/rejected": -1.3292286396026611, "step": 530, - "use_label": 1448.800048828125 + "use_label": 0.0 }, { - "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, + "epoch": 0.56, + "learning_rate": 2.4186046511627906e-07, + "logits/chosen": -2.7393388748168945, + "logits/rejected": -2.723223924636841, + "logps/chosen": -288.07904052734375, + "logps/rejected": -284.3858642578125, + "loss": 0.5325, + "pred_label": 0.0, "rewards/accuracies": 0.75, - "rewards/chosen": 2.4270758628845215, - "rewards/margins": 2.458122730255127, - "rewards/rejected": -0.031046796590089798, + "rewards/chosen": 0.023196930065751076, + "rewards/margins": 1.3002605438232422, + "rewards/rejected": -1.2770636081695557, "step": 540, - "use_label": 1464.5999755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.58, + "learning_rate": 2.3604651162790695e-07, + "logits/chosen": -2.7268896102905273, + "logits/rejected": -2.699005365371704, + "logps/chosen": -293.6747131347656, + "logps/rejected": -266.55059814453125, + "loss": 0.5283, + "pred_label": 0.0, "rewards/accuracies": 0.7250000238418579, - "rewards/chosen": 1.532619595527649, - "rewards/margins": 2.0600717067718506, - "rewards/rejected": -0.5274521708488464, + "rewards/chosen": 0.11607713997364044, + "rewards/margins": 1.205520510673523, + "rewards/rejected": -1.0894434452056885, "step": 550, - "use_label": 1482.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.59, + "learning_rate": 2.3023255813953487e-07, + "logits/chosen": -2.6480965614318848, + "logits/rejected": -2.66930890083313, + "logps/chosen": -264.1002197265625, + "logps/rejected": -240.21029663085938, + "loss": 0.6114, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.06454001367092133, + "rewards/margins": 0.681505560874939, + "rewards/rejected": -0.746045708656311, "step": 560, - "use_label": 1494.5 + "use_label": 0.0 }, { - "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, + "epoch": 0.6, + "learning_rate": 2.2441860465116278e-07, + "logits/chosen": -2.708428382873535, + "logits/rejected": -2.7417664527893066, + "logps/chosen": -296.1793212890625, + "logps/rejected": -299.2248840332031, + "loss": 0.5686, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": 0.24114087224006653, + "rewards/margins": 0.9173105955123901, + "rewards/rejected": -0.6761698126792908, "step": 570, - "use_label": 1513.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.61, + "learning_rate": 2.186046511627907e-07, + "logits/chosen": -2.6812610626220703, + "logits/rejected": -2.6830496788024902, + "logps/chosen": -283.55950927734375, + "logps/rejected": -271.852783203125, + "loss": 0.5146, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.09269438683986664, + "rewards/margins": 0.9761162996292114, + "rewards/rejected": -1.0688107013702393, "step": 580, - "use_label": 1537.8499755859375 + "use_label": 0.0 }, { - "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, + "epoch": 0.62, + "learning_rate": 2.127906976744186e-07, + "logits/chosen": -2.647878646850586, + "logits/rejected": -2.62599515914917, + "logps/chosen": -290.76544189453125, + "logps/rejected": -325.32684326171875, + "loss": 0.5352, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.14185315370559692, + "rewards/margins": 0.620773196220398, + "rewards/rejected": -0.7626262903213501, "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 + "use_label": 0.0 }, { "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 + "learning_rate": 2.0697674418604648e-07, + "logits/chosen": -2.6654324531555176, + "logits/rejected": -2.6334807872772217, + "logps/chosen": -318.82781982421875, + "logps/rejected": -323.73052978515625, + "loss": 0.5791, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": 0.07647746801376343, + "rewards/margins": 0.8629099726676941, + "rewards/rejected": -0.7864325642585754, + "step": 600, + "use_label": 0.0 }, { "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 + "eval_logits/chosen": -2.759521245956421, + "eval_logits/rejected": -2.751826763153076, + "eval_logps/chosen": -288.24688720703125, + "eval_logps/rejected": -271.9148864746094, + "eval_loss": 0.5281260013580322, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7460317611694336, + "eval_rewards/chosen": 0.02767007239162922, + "eval_rewards/margins": 1.0551676750183105, + "eval_rewards/rejected": -1.02749764919281, + "eval_runtime": 151.9139, + "eval_samples_per_second": 13.165, + "eval_steps_per_second": 0.415, + "eval_use_label": 0.0, + "step": 600 }, { "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 + "learning_rate": 2.0116279069767443e-07, + "logits/chosen": -2.727267026901245, + "logits/rejected": -2.6593177318573, + "logps/chosen": -285.02386474609375, + "logps/rejected": -229.25363159179688, + "loss": 0.5251, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.22663810849189758, + "rewards/margins": 0.9340082406997681, + "rewards/rejected": -1.1606463193893433, + "step": 610, + "use_label": 0.0 }, { "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 + "learning_rate": 1.9534883720930232e-07, + "logits/chosen": -2.7362422943115234, + "logits/rejected": -2.684910297393799, + "logps/chosen": -272.39776611328125, + "logps/rejected": -238.19973754882812, + "loss": 0.5186, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.16883151233196259, + "rewards/margins": 0.9152736663818359, + "rewards/rejected": -1.0841050148010254, + "step": 620, + "use_label": 0.0 }, { "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 + "learning_rate": 1.895348837209302e-07, + "logits/chosen": -2.6318321228027344, + "logits/rejected": -2.720790386199951, + "logps/chosen": -252.7714080810547, + "logps/rejected": -261.837158203125, + "loss": 0.508, + "pred_label": 0.0, + "rewards/accuracies": 0.8374999761581421, + "rewards/chosen": 0.06653375923633575, + "rewards/margins": 1.351811408996582, + "rewards/rejected": -1.2852776050567627, + "step": 630, + "use_label": 0.0 }, { "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 + "learning_rate": 1.8372093023255813e-07, + "logits/chosen": -2.708900213241577, + "logits/rejected": -2.6771461963653564, + "logps/chosen": -248.0977325439453, + "logps/rejected": -243.92178344726562, + "loss": 0.509, + "pred_label": 0.0, + "rewards/accuracies": 0.8125, + "rewards/chosen": 0.1819048821926117, + "rewards/margins": 1.3170161247253418, + "rewards/rejected": -1.1351112127304077, + "step": 640, + "use_label": 0.0 }, { "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 + "learning_rate": 1.7790697674418602e-07, + "logits/chosen": -2.767035722732544, + "logits/rejected": -2.7256782054901123, + "logps/chosen": -292.49127197265625, + "logps/rejected": -259.6093444824219, + "loss": 0.5359, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.06279899179935455, + "rewards/margins": 1.0030735731124878, + "rewards/rejected": -1.0658727884292603, + "step": 650, + "use_label": 0.0 }, { "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 + "learning_rate": 1.7209302325581396e-07, + "logits/chosen": -2.7260286808013916, + "logits/rejected": -2.7639498710632324, + "logps/chosen": -286.2920837402344, + "logps/rejected": -274.6990966796875, + "loss": 0.5137, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.031030695885419846, + "rewards/margins": 0.9901006817817688, + "rewards/rejected": -0.9590700268745422, + "step": 660, + "use_label": 0.0 }, { "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 + "learning_rate": 1.6627906976744186e-07, + "logits/chosen": -2.6322426795959473, + "logits/rejected": -2.632668972015381, + "logps/chosen": -303.038818359375, + "logps/rejected": -288.877197265625, + "loss": 0.514, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": 0.10557667911052704, + "rewards/margins": 1.2127177715301514, + "rewards/rejected": -1.1071412563323975, + "step": 670, + "use_label": 0.0 }, { "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 + "learning_rate": 1.6046511627906975e-07, + "logits/chosen": -2.5468828678131104, + "logits/rejected": -2.5757508277893066, + "logps/chosen": -228.77261352539062, + "logps/rejected": -234.0619659423828, + "loss": 0.5591, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": -0.0003312155604362488, + "rewards/margins": 0.6896525621414185, + "rewards/rejected": -0.6899837255477905, + "step": 680, + "use_label": 0.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, + "learning_rate": 1.5465116279069766e-07, + "logits/chosen": -2.636993885040283, + "logits/rejected": -2.585117816925049, + "logps/chosen": -250.43234252929688, + "logps/rejected": -217.508056640625, + "loss": 0.5132, + "pred_label": 0.0, "rewards/accuracies": 0.800000011920929, - "rewards/chosen": -0.5959171652793884, - "rewards/margins": 3.479651927947998, - "rewards/rejected": -4.075569152832031, - "step": 1390, - "use_label": 2437.449951171875 + "rewards/chosen": 0.18243907392024994, + "rewards/margins": 1.0671703815460205, + "rewards/rejected": -0.8847312927246094, + "step": 690, + "use_label": 0.0 }, { "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 + "learning_rate": 1.4883720930232558e-07, + "logits/chosen": -2.63033390045166, + "logits/rejected": -2.5760812759399414, + "logps/chosen": -255.9130096435547, + "logps/rejected": -274.47467041015625, + "loss": 0.5238, + "pred_label": 0.0, + "rewards/accuracies": 0.8125, + "rewards/chosen": 0.016175851225852966, + "rewards/margins": 1.1498843431472778, + "rewards/rejected": -1.1337083578109741, + "step": 700, + "use_label": 0.0 }, { - "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.73, + "eval_logits/chosen": -2.733815908432007, + "eval_logits/rejected": -2.726200580596924, + "eval_logps/chosen": -288.18280029296875, + "eval_logps/rejected": -272.3071594238281, + "eval_loss": 0.5295113325119019, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7539682388305664, + "eval_rewards/chosen": 0.034078579396009445, + "eval_rewards/margins": 1.1008045673370361, + "eval_rewards/rejected": -1.0667259693145752, + "eval_runtime": 151.6475, + "eval_samples_per_second": 13.188, + "eval_steps_per_second": 0.415, + "eval_use_label": 0.0, + "step": 700 }, { "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 + "learning_rate": 1.4302325581395347e-07, + "logits/chosen": -2.6521668434143066, + "logits/rejected": -2.6327109336853027, + "logps/chosen": -305.80029296875, + "logps/rejected": -239.8463897705078, + "loss": 0.5036, + "pred_label": 0.0, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": 0.14674702286720276, + "rewards/margins": 1.1667625904083252, + "rewards/rejected": -1.0200153589248657, + "step": 710, + "use_label": 0.0 }, { "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 + "learning_rate": 1.372093023255814e-07, + "logits/chosen": -2.6996383666992188, + "logits/rejected": -2.70395827293396, + "logps/chosen": -334.99652099609375, + "logps/rejected": -312.52581787109375, + "loss": 0.4854, + "pred_label": 0.0, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.011823808774352074, + "rewards/margins": 1.3162919282913208, + "rewards/rejected": -1.3044681549072266, + "step": 720, + "use_label": 0.0 }, { "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 + "learning_rate": 1.3139534883720928e-07, + "logits/chosen": -2.6853585243225098, + "logits/rejected": -2.634413480758667, + "logps/chosen": -327.4559020996094, + "logps/rejected": -264.45123291015625, + "loss": 0.5201, + "pred_label": 0.0, + "rewards/accuracies": 0.800000011920929, + "rewards/chosen": 0.2056887149810791, + "rewards/margins": 1.0861843824386597, + "rewards/rejected": -0.8804956674575806, + "step": 730, + "use_label": 0.0 }, { "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 + "learning_rate": 1.2558139534883723e-07, + "logits/chosen": -2.686652660369873, + "logits/rejected": -2.6892364025115967, + "logps/chosen": -297.3292541503906, + "logps/rejected": -277.6313781738281, + "loss": 0.5333, + "pred_label": 0.0, + "rewards/accuracies": 0.824999988079071, + "rewards/chosen": 0.06698469072580338, + "rewards/margins": 1.1885979175567627, + "rewards/rejected": -1.1216132640838623, + "step": 740, + "use_label": 0.0 }, { "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 + "learning_rate": 1.1976744186046512e-07, + "logits/chosen": -2.6189768314361572, + "logits/rejected": -2.598482370376587, + "logps/chosen": -260.7591247558594, + "logps/rejected": -294.9975280761719, + "loss": 0.523, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.0578201524913311, + "rewards/margins": 1.053215742111206, + "rewards/rejected": -1.1110358238220215, + "step": 750, + "use_label": 0.0 }, { "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 + "learning_rate": 1.1395348837209302e-07, + "logits/chosen": -2.705519199371338, + "logits/rejected": -2.696622371673584, + "logps/chosen": -306.7921447753906, + "logps/rejected": -287.7425842285156, + "loss": 0.4972, + "pred_label": 0.0, + "rewards/accuracies": 0.637499988079071, + "rewards/chosen": -0.11914374679327011, + "rewards/margins": 0.7654374837875366, + "rewards/rejected": -0.8845812082290649, + "step": 760, + "use_label": 0.0 }, { "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 + "learning_rate": 1.0813953488372093e-07, + "logits/chosen": -2.677530288696289, + "logits/rejected": -2.637589931488037, + "logps/chosen": -304.0654296875, + "logps/rejected": -326.1291809082031, + "loss": 0.5239, + "pred_label": 0.0, + "rewards/accuracies": 0.7749999761581421, + "rewards/chosen": -0.015724360942840576, + "rewards/margins": 1.0277526378631592, + "rewards/rejected": -1.0434770584106445, + "step": 770, + "use_label": 0.0 }, { "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, + "learning_rate": 1.0232558139534883e-07, + "logits/chosen": -2.674903154373169, + "logits/rejected": -2.6580498218536377, + "logps/chosen": -249.53701782226562, + "logps/rejected": -244.45858764648438, + "loss": 0.5269, + "pred_label": 0.0, "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 + "rewards/chosen": -0.016674162819981575, + "rewards/margins": 0.7822299599647522, + "rewards/rejected": -0.7989041805267334, + "step": 780, + "use_label": 0.0 }, { "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 + "learning_rate": 9.651162790697674e-08, + "logits/chosen": -2.661597728729248, + "logits/rejected": -2.700005054473877, + "logps/chosen": -275.3680114746094, + "logps/rejected": -290.7065734863281, + "loss": 0.5744, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.020705239847302437, + "rewards/margins": 1.2380138635635376, + "rewards/rejected": -1.2587188482284546, + "step": 790, + "use_label": 0.0 }, { "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 + "learning_rate": 9.069767441860465e-08, + "logits/chosen": -2.626075506210327, + "logits/rejected": -2.600184202194214, + "logps/chosen": -380.11334228515625, + "logps/rejected": -328.88665771484375, + "loss": 0.515, + "pred_label": 0.0, + "rewards/accuracies": 0.675000011920929, + "rewards/chosen": -0.10672910511493683, + "rewards/margins": 0.9256986379623413, + "rewards/rejected": -1.0324275493621826, + "step": 800, + "use_label": 0.0 }, { "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 + "eval_logits/chosen": -2.754448413848877, + "eval_logits/rejected": -2.747894763946533, + "eval_logps/chosen": -288.57720947265625, + "eval_logps/rejected": -272.82855224609375, + "eval_loss": 0.5257741212844849, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.7539682388305664, + "eval_rewards/chosen": -0.00535963149741292, + "eval_rewards/margins": 1.11350679397583, + "eval_rewards/rejected": -1.1188664436340332, + "eval_runtime": 152.0351, + "eval_samples_per_second": 13.155, + "eval_steps_per_second": 0.414, + "eval_use_label": 0.0, + "step": 800 }, { "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, + "learning_rate": 8.488372093023254e-08, + "logits/chosen": -2.720799684524536, + "logits/rejected": -2.6458606719970703, + "logps/chosen": -279.9139404296875, + "logps/rejected": -267.57965087890625, + "loss": 0.5294, + "pred_label": 0.0, "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 + "rewards/chosen": -0.003540521953254938, + "rewards/margins": 0.9458888173103333, + "rewards/rejected": -0.9494293332099915, + "step": 810, + "use_label": 0.0 }, { "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 + "learning_rate": 7.906976744186046e-08, + "logits/chosen": -2.6257495880126953, + "logits/rejected": -2.6502857208251953, + "logps/chosen": -245.8484344482422, + "logps/rejected": -254.8400115966797, + "loss": 0.5512, + "pred_label": 0.0, + "rewards/accuracies": 0.7124999761581421, + "rewards/chosen": 0.12137588113546371, + "rewards/margins": 1.019413709640503, + "rewards/rejected": -0.8980380296707153, + "step": 820, + "use_label": 0.0 }, { "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 + "learning_rate": 7.325581395348837e-08, + "logits/chosen": -2.715819835662842, + "logits/rejected": -2.6348624229431152, + "logps/chosen": -256.454833984375, + "logps/rejected": -287.5106506347656, + "loss": 0.4857, + "pred_label": 0.0, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -0.025987720116972923, + "rewards/margins": 1.199789047241211, + "rewards/rejected": -1.2257767915725708, + "step": 830, + "use_label": 0.0 }, { "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 + "learning_rate": 6.744186046511628e-08, + "logits/chosen": -2.6387436389923096, + "logits/rejected": -2.6407735347747803, + "logps/chosen": -301.9522705078125, + "logps/rejected": -316.30950927734375, + "loss": 0.5383, + "pred_label": 0.0, + "rewards/accuracies": 0.737500011920929, + "rewards/chosen": 0.08713732659816742, + "rewards/margins": 1.017157793045044, + "rewards/rejected": -0.9300203323364258, + "step": 840, + "use_label": 0.0 }, { "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 + "learning_rate": 6.162790697674419e-08, + "logits/chosen": -2.8061776161193848, + "logits/rejected": -2.7850382328033447, + "logps/chosen": -308.90155029296875, + "logps/rejected": -261.96075439453125, + "loss": 0.537, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": 0.09603701531887054, + "rewards/margins": 0.9941841959953308, + "rewards/rejected": -0.8981472253799438, + "step": 850, + "use_label": 0.0 }, { "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 + "learning_rate": 5.5813953488372087e-08, + "logits/chosen": -2.6717400550842285, + "logits/rejected": -2.691214084625244, + "logps/chosen": -292.5924072265625, + "logps/rejected": -275.4133605957031, + "loss": 0.5371, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.14807067811489105, + "rewards/margins": 1.216901183128357, + "rewards/rejected": -1.3649718761444092, + "step": 860, + "use_label": 0.0 }, { "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 + "learning_rate": 5e-08, + "logits/chosen": -2.7427048683166504, + "logits/rejected": -2.779627799987793, + "logps/chosen": -242.98812866210938, + "logps/rejected": -253.3491973876953, + "loss": 0.5322, + "pred_label": 0.0, + "rewards/accuracies": 0.75, + "rewards/chosen": -0.05029615759849548, + "rewards/margins": 1.148262619972229, + "rewards/rejected": -1.1985588073730469, + "step": 870, + "use_label": 0.0 }, { "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 + "learning_rate": 4.41860465116279e-08, + "logits/chosen": -2.7546846866607666, + "logits/rejected": -2.697383165359497, + "logps/chosen": -284.3351135253906, + "logps/rejected": -287.35260009765625, + "loss": 0.4992, + "pred_label": 0.0, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.2294842004776001, + "rewards/margins": 0.8660097122192383, + "rewards/rejected": -1.095494031906128, + "step": 880, + "use_label": 0.0 }, { "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 + "learning_rate": 3.837209302325581e-08, + "logits/chosen": -2.7058663368225098, + "logits/rejected": -2.6869189739227295, + "logps/chosen": -246.6070556640625, + "logps/rejected": -248.0365753173828, + "loss": 0.5038, + "pred_label": 0.0, + "rewards/accuracies": 0.7250000238418579, + "rewards/chosen": -0.039266109466552734, + "rewards/margins": 0.955298125743866, + "rewards/rejected": -0.9945642352104187, + "step": 890, + "use_label": 0.0 }, { "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 + "learning_rate": 3.255813953488372e-08, + "logits/chosen": -2.571969509124756, + "logits/rejected": -2.546536684036255, + "logps/chosen": -254.21237182617188, + "logps/rejected": -247.3881072998047, + "loss": 0.5166, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": 0.08884133398532867, + "rewards/margins": 1.227858304977417, + "rewards/rejected": -1.13901686668396, + "step": 900, + "use_label": 0.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 + "eval_logits/chosen": -2.782860040664673, + "eval_logits/rejected": -2.777499198913574, + "eval_logps/chosen": -289.315673828125, + "eval_logps/rejected": -273.0716552734375, + "eval_loss": 0.5272806286811829, + "eval_pred_label": 0.0, + "eval_rewards/accuracies": 0.761904776096344, + "eval_rewards/chosen": -0.07920397818088531, + "eval_rewards/margins": 1.0639697313308716, + "eval_rewards/rejected": -1.1431735754013062, + "eval_runtime": 151.9262, + "eval_samples_per_second": 13.164, + "eval_steps_per_second": 0.415, + "eval_use_label": 0.0, + "step": 900 }, { "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 + "learning_rate": 2.6744186046511626e-08, + "logits/chosen": -2.7148184776306152, + "logits/rejected": -2.7250776290893555, + "logps/chosen": -236.6240234375, + "logps/rejected": -242.1452178955078, + "loss": 0.4962, + "pred_label": 0.0, + "rewards/accuracies": 0.6875, + "rewards/chosen": -0.22720427811145782, + "rewards/margins": 0.8557177782058716, + "rewards/rejected": -1.0829219818115234, + "step": 910, + "use_label": 0.0 }, { "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 + "learning_rate": 2.0930232558139533e-08, + "logits/chosen": -2.641587734222412, + "logits/rejected": -2.6006321907043457, + "logps/chosen": -259.6849060058594, + "logps/rejected": -247.21450805664062, + "loss": 0.4621, + "pred_label": 0.0, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": 0.009412238374352455, + "rewards/margins": 1.2470829486846924, + "rewards/rejected": -1.237670660018921, + "step": 920, + "use_label": 0.0 }, { "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 + "learning_rate": 1.511627906976744e-08, + "logits/chosen": -2.7401022911071777, + "logits/rejected": -2.733879566192627, + "logps/chosen": -280.37432861328125, + "logps/rejected": -234.80264282226562, + "loss": 0.5163, + "pred_label": 0.0, + "rewards/accuracies": 0.8125, + "rewards/chosen": -0.03051159344613552, + "rewards/margins": 1.1736021041870117, + "rewards/rejected": -1.2041137218475342, + "step": 930, + "use_label": 0.0 }, { "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 + "learning_rate": 9.302325581395349e-09, + "logits/chosen": -2.7654836177825928, + "logits/rejected": -2.7249624729156494, + "logps/chosen": -323.6404724121094, + "logps/rejected": -323.39886474609375, + "loss": 0.4899, + "pred_label": 0.0, + "rewards/accuracies": 0.762499988079071, + "rewards/chosen": -0.04380204528570175, + "rewards/margins": 1.207780361175537, + "rewards/rejected": -1.2515825033187866, + "step": 940, + "use_label": 0.0 }, { "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 + "learning_rate": 3.4883720930232554e-09, + "logits/chosen": -2.725027084350586, + "logits/rejected": -2.711854934692383, + "logps/chosen": -288.2428894042969, + "logps/rejected": -280.0832214355469, + "loss": 0.5456, + "pred_label": 0.0, + "rewards/accuracies": 0.7875000238418579, + "rewards/chosen": -0.0814594253897667, + "rewards/margins": 1.5194867849349976, + "rewards/rejected": -1.6009461879730225, + "step": 950, + "use_label": 0.0 }, { "epoch": 1.0, - "step": 1910, + "step": 956, "total_flos": 0.0, - "train_loss": 0.2926436502271922, - "train_runtime": 13370.5554, - "train_samples_per_second": 4.572, - "train_steps_per_second": 0.143 + "train_loss": 0.5461576643349236, + "train_runtime": 10580.1696, + "train_samples_per_second": 5.778, + "train_steps_per_second": 0.09 } ], "logging_steps": 10, - "max_steps": 1910, + "max_steps": 956, "num_train_epochs": 1, "save_steps": 50, "total_flos": 0.0,