llama-7b-dpo-qlora / trainer_state.json
chanchan7's picture
Model save
aef4b53 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9997382884061764,
"eval_steps": 100,
"global_step": 1910,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 2.617801047120419e-08,
"logits/chosen": -0.58705735206604,
"logits/rejected": -0.5808682441711426,
"logps/chosen": -386.9568786621094,
"logps/rejected": -358.9479064941406,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.01,
"learning_rate": 2.617801047120419e-07,
"logits/chosen": -0.4515784680843353,
"logits/rejected": -0.4993375241756439,
"logps/chosen": -340.8123474121094,
"logps/rejected": -277.9239501953125,
"loss": 0.6934,
"rewards/accuracies": 0.4166666567325592,
"rewards/chosen": -0.0005649582017213106,
"rewards/margins": -6.073586700949818e-05,
"rewards/rejected": -0.0005042223492637277,
"step": 10
},
{
"epoch": 0.01,
"learning_rate": 5.235602094240838e-07,
"logits/chosen": -0.4952339231967926,
"logits/rejected": -0.5711107850074768,
"logps/chosen": -382.88726806640625,
"logps/rejected": -338.0599670410156,
"loss": 0.6931,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": 0.0028820387087762356,
"rewards/margins": 0.00026537588564679027,
"rewards/rejected": 0.002616662997752428,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 7.853403141361258e-07,
"logits/chosen": -0.48099619150161743,
"logits/rejected": -0.4897836148738861,
"logps/chosen": -309.4810791015625,
"logps/rejected": -301.8321838378906,
"loss": 0.6929,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.004554978106170893,
"rewards/margins": 0.0005212987889535725,
"rewards/rejected": 0.004033679608255625,
"step": 30
},
{
"epoch": 0.02,
"learning_rate": 1.0471204188481676e-06,
"logits/chosen": -0.5069094896316528,
"logits/rejected": -0.5498972535133362,
"logps/chosen": -294.9288330078125,
"logps/rejected": -289.6306457519531,
"loss": 0.6928,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.00764175970107317,
"rewards/margins": 0.0012837459798902273,
"rewards/rejected": 0.006358013954013586,
"step": 40
},
{
"epoch": 0.03,
"learning_rate": 1.3089005235602096e-06,
"logits/chosen": -0.4766748547554016,
"logits/rejected": -0.6654168367385864,
"logps/chosen": -353.96112060546875,
"logps/rejected": -258.60479736328125,
"loss": 0.6918,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.013894833624362946,
"rewards/margins": 0.004527085926383734,
"rewards/rejected": 0.009367748163640499,
"step": 50
},
{
"epoch": 0.03,
"learning_rate": 1.5706806282722515e-06,
"logits/chosen": -0.5836187601089478,
"logits/rejected": -0.5082155466079712,
"logps/chosen": -311.2541809082031,
"logps/rejected": -351.26812744140625,
"loss": 0.691,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": 0.019704651087522507,
"rewards/margins": -0.0016706284368410707,
"rewards/rejected": 0.021375281736254692,
"step": 60
},
{
"epoch": 0.04,
"learning_rate": 1.8324607329842933e-06,
"logits/chosen": -0.4787808954715729,
"logits/rejected": -0.5381947755813599,
"logps/chosen": -394.5220031738281,
"logps/rejected": -304.90826416015625,
"loss": 0.6899,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.037270233035087585,
"rewards/margins": 0.01263844221830368,
"rewards/rejected": 0.024631790816783905,
"step": 70
},
{
"epoch": 0.04,
"learning_rate": 2.094240837696335e-06,
"logits/chosen": -0.4997861981391907,
"logits/rejected": -0.5136772990226746,
"logps/chosen": -334.28045654296875,
"logps/rejected": -317.20477294921875,
"loss": 0.6896,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.042300838977098465,
"rewards/margins": 0.006197690032422543,
"rewards/rejected": 0.036103151738643646,
"step": 80
},
{
"epoch": 0.05,
"learning_rate": 2.356020942408377e-06,
"logits/chosen": -0.502038836479187,
"logits/rejected": -0.5897258520126343,
"logps/chosen": -346.77593994140625,
"logps/rejected": -291.6183776855469,
"loss": 0.6867,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.053455304354429245,
"rewards/margins": 0.013971390202641487,
"rewards/rejected": 0.03948391601443291,
"step": 90
},
{
"epoch": 0.05,
"learning_rate": 2.617801047120419e-06,
"logits/chosen": -0.41690191626548767,
"logits/rejected": -0.5484428405761719,
"logps/chosen": -355.68505859375,
"logps/rejected": -283.31048583984375,
"loss": 0.6856,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.07831202447414398,
"rewards/margins": 0.021578781306743622,
"rewards/rejected": 0.05673323944211006,
"step": 100
},
{
"epoch": 0.05,
"eval_logits/chosen": -0.31417590379714966,
"eval_logits/rejected": -0.33973267674446106,
"eval_logps/chosen": -338.1841735839844,
"eval_logps/rejected": -307.254638671875,
"eval_loss": 0.6868060231208801,
"eval_rewards/accuracies": 0.5376983880996704,
"eval_rewards/chosen": 0.08433966338634491,
"eval_rewards/margins": 0.015132981352508068,
"eval_rewards/rejected": 0.06920668482780457,
"eval_runtime": 303.4683,
"eval_samples_per_second": 6.59,
"eval_steps_per_second": 0.208,
"step": 100
},
{
"epoch": 0.06,
"learning_rate": 2.8795811518324613e-06,
"logits/chosen": -0.5858451724052429,
"logits/rejected": -0.5726209282875061,
"logps/chosen": -321.0362243652344,
"logps/rejected": -316.78009033203125,
"loss": 0.686,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.09119822084903717,
"rewards/margins": 0.01558433473110199,
"rewards/rejected": 0.07561388611793518,
"step": 110
},
{
"epoch": 0.06,
"learning_rate": 3.141361256544503e-06,
"logits/chosen": -0.4955894947052002,
"logits/rejected": -0.5918242335319519,
"logps/chosen": -307.3293762207031,
"logps/rejected": -277.9299011230469,
"loss": 0.683,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.09477759897708893,
"rewards/margins": 0.022622792050242424,
"rewards/rejected": 0.07215481251478195,
"step": 120
},
{
"epoch": 0.07,
"learning_rate": 3.403141361256545e-06,
"logits/chosen": -0.433607816696167,
"logits/rejected": -0.5049300193786621,
"logps/chosen": -317.8403625488281,
"logps/rejected": -276.2845153808594,
"loss": 0.6795,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.12141671031713486,
"rewards/margins": 0.027771497145295143,
"rewards/rejected": 0.09364522993564606,
"step": 130
},
{
"epoch": 0.07,
"learning_rate": 3.6649214659685865e-06,
"logits/chosen": -0.35517334938049316,
"logits/rejected": -0.41019129753112793,
"logps/chosen": -331.9830627441406,
"logps/rejected": -290.9942321777344,
"loss": 0.6796,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.150104820728302,
"rewards/margins": 0.034656353294849396,
"rewards/rejected": 0.11544845998287201,
"step": 140
},
{
"epoch": 0.08,
"learning_rate": 3.926701570680629e-06,
"logits/chosen": -0.41458845138549805,
"logits/rejected": -0.45303764939308167,
"logps/chosen": -374.31268310546875,
"logps/rejected": -304.92242431640625,
"loss": 0.6705,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.17660722136497498,
"rewards/margins": 0.06261179596185684,
"rewards/rejected": 0.11399543285369873,
"step": 150
},
{
"epoch": 0.08,
"learning_rate": 4.18848167539267e-06,
"logits/chosen": -0.4083201289176941,
"logits/rejected": -0.4408670961856842,
"logps/chosen": -372.0410461425781,
"logps/rejected": -322.4227600097656,
"loss": 0.6734,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.19496707618236542,
"rewards/margins": 0.05301988124847412,
"rewards/rejected": 0.1419472098350525,
"step": 160
},
{
"epoch": 0.09,
"learning_rate": 4.450261780104713e-06,
"logits/chosen": -0.46804505586624146,
"logits/rejected": -0.4163655638694763,
"logps/chosen": -262.0623474121094,
"logps/rejected": -281.7099609375,
"loss": 0.6724,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.15472838282585144,
"rewards/margins": 0.0040833535604178905,
"rewards/rejected": 0.15064503252506256,
"step": 170
},
{
"epoch": 0.09,
"learning_rate": 4.712041884816754e-06,
"logits/chosen": -0.41915637254714966,
"logits/rejected": -0.41267627477645874,
"logps/chosen": -316.70330810546875,
"logps/rejected": -292.5999450683594,
"loss": 0.6642,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.2095501720905304,
"rewards/margins": 0.06560155004262924,
"rewards/rejected": 0.14394858479499817,
"step": 180
},
{
"epoch": 0.1,
"learning_rate": 4.9738219895287965e-06,
"logits/chosen": -0.30535200238227844,
"logits/rejected": -0.4938386082649231,
"logps/chosen": -324.11016845703125,
"logps/rejected": -270.20721435546875,
"loss": 0.6688,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.23200471699237823,
"rewards/margins": 0.09016872942447662,
"rewards/rejected": 0.1418360024690628,
"step": 190
},
{
"epoch": 0.1,
"learning_rate": 4.999661831436499e-06,
"logits/chosen": -0.3207201361656189,
"logits/rejected": -0.45532283186912537,
"logps/chosen": -347.30316162109375,
"logps/rejected": -325.71929931640625,
"loss": 0.6704,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.23967811465263367,
"rewards/margins": 0.06717831641435623,
"rewards/rejected": 0.17249980568885803,
"step": 200
},
{
"epoch": 0.1,
"eval_logits/chosen": -0.3406147360801697,
"eval_logits/rejected": -0.3758181631565094,
"eval_logps/chosen": -322.39111328125,
"eval_logps/rejected": -296.1336669921875,
"eval_loss": 0.6715283989906311,
"eval_rewards/accuracies": 0.5714285969734192,
"eval_rewards/chosen": 0.24227052927017212,
"eval_rewards/margins": 0.061853885650634766,
"eval_rewards/rejected": 0.18041667342185974,
"eval_runtime": 307.592,
"eval_samples_per_second": 6.502,
"eval_steps_per_second": 0.205,
"step": 200
},
{
"epoch": 0.11,
"learning_rate": 4.9984929711403395e-06,
"logits/chosen": -0.4271935820579529,
"logits/rejected": -0.5097531676292419,
"logps/chosen": -321.6978454589844,
"logps/rejected": -279.8887939453125,
"loss": 0.6615,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.21578499674797058,
"rewards/margins": 0.05311957001686096,
"rewards/rejected": 0.1626654416322708,
"step": 210
},
{
"epoch": 0.12,
"learning_rate": 4.996489634487865e-06,
"logits/chosen": -0.2373839169740677,
"logits/rejected": -0.3769948184490204,
"logps/chosen": -373.025634765625,
"logps/rejected": -327.1902160644531,
"loss": 0.6723,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.287474125623703,
"rewards/margins": 0.09721784293651581,
"rewards/rejected": 0.19025632739067078,
"step": 220
},
{
"epoch": 0.12,
"learning_rate": 4.9936524905772466e-06,
"logits/chosen": -0.3467353880405426,
"logits/rejected": -0.49967822432518005,
"logps/chosen": -360.65362548828125,
"logps/rejected": -339.7496643066406,
"loss": 0.6611,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.27538615465164185,
"rewards/margins": 0.08647196739912033,
"rewards/rejected": 0.18891416490077972,
"step": 230
},
{
"epoch": 0.13,
"learning_rate": 4.9899824869915e-06,
"logits/chosen": -0.3842504024505615,
"logits/rejected": -0.4908681809902191,
"logps/chosen": -323.8313293457031,
"logps/rejected": -300.1197204589844,
"loss": 0.6608,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.2536216378211975,
"rewards/margins": 0.08490858227014542,
"rewards/rejected": 0.16871307790279388,
"step": 240
},
{
"epoch": 0.13,
"learning_rate": 4.985480849482012e-06,
"logits/chosen": -0.37045997381210327,
"logits/rejected": -0.41814011335372925,
"logps/chosen": -311.5799560546875,
"logps/rejected": -304.9263610839844,
"loss": 0.6622,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.22143617272377014,
"rewards/margins": 0.06613994389772415,
"rewards/rejected": 0.1552962213754654,
"step": 250
},
{
"epoch": 0.14,
"learning_rate": 4.980149081559142e-06,
"logits/chosen": -0.33250561356544495,
"logits/rejected": -0.38181784749031067,
"logps/chosen": -346.22845458984375,
"logps/rejected": -355.2063293457031,
"loss": 0.6566,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.2670690417289734,
"rewards/margins": 0.0644298866391182,
"rewards/rejected": 0.20263917744159698,
"step": 260
},
{
"epoch": 0.14,
"learning_rate": 4.9739889639900655e-06,
"logits/chosen": -0.46652165055274963,
"logits/rejected": -0.4939687252044678,
"logps/chosen": -316.5768127441406,
"logps/rejected": -301.8284606933594,
"loss": 0.6537,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.22106440365314484,
"rewards/margins": 0.05704628303647041,
"rewards/rejected": 0.16401812434196472,
"step": 270
},
{
"epoch": 0.15,
"learning_rate": 4.967002554204009e-06,
"logits/chosen": -0.325555682182312,
"logits/rejected": -0.42002320289611816,
"logps/chosen": -332.63031005859375,
"logps/rejected": -311.92181396484375,
"loss": 0.6594,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.19830381870269775,
"rewards/margins": 0.07296115905046463,
"rewards/rejected": 0.12534265220165253,
"step": 280
},
{
"epoch": 0.15,
"learning_rate": 4.959192185605089e-06,
"logits/chosen": -0.36657431721687317,
"logits/rejected": -0.4914252758026123,
"logps/chosen": -323.62890625,
"logps/rejected": -286.2757873535156,
"loss": 0.6571,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.17159457504749298,
"rewards/margins": 0.07951690256595612,
"rewards/rejected": 0.09207765758037567,
"step": 290
},
{
"epoch": 0.16,
"learning_rate": 4.950560466792969e-06,
"logits/chosen": -0.4312424659729004,
"logits/rejected": -0.4952569007873535,
"logps/chosen": -284.1667785644531,
"logps/rejected": -273.22900390625,
"loss": 0.6506,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.14056554436683655,
"rewards/margins": 0.13772882521152496,
"rewards/rejected": 0.002836701227352023,
"step": 300
},
{
"epoch": 0.16,
"eval_logits/chosen": -0.442807137966156,
"eval_logits/rejected": -0.475908488035202,
"eval_logps/chosen": -331.02752685546875,
"eval_logps/rejected": -309.7589111328125,
"eval_loss": 0.6529492735862732,
"eval_rewards/accuracies": 0.6646825671195984,
"eval_rewards/chosen": 0.155906543135643,
"eval_rewards/margins": 0.11174201965332031,
"eval_rewards/rejected": 0.04416452720761299,
"eval_runtime": 303.7935,
"eval_samples_per_second": 6.583,
"eval_steps_per_second": 0.207,
"step": 300
},
{
"epoch": 0.16,
"learning_rate": 4.9411102806916185e-06,
"logits/chosen": -0.41296476125717163,
"logits/rejected": -0.5174468755722046,
"logps/chosen": -362.47833251953125,
"logps/rejected": -323.5210266113281,
"loss": 0.6355,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": 0.1841851770877838,
"rewards/margins": 0.19086548686027527,
"rewards/rejected": -0.00668031070381403,
"step": 310
},
{
"epoch": 0.17,
"learning_rate": 4.930844783586424e-06,
"logits/chosen": -0.5255850553512573,
"logits/rejected": -0.5664322376251221,
"logps/chosen": -277.1454162597656,
"logps/rejected": -295.079833984375,
"loss": 0.6426,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.09250400960445404,
"rewards/margins": 0.09653620421886444,
"rewards/rejected": -0.004032188560813665,
"step": 320
},
{
"epoch": 0.17,
"learning_rate": 4.919767404070033e-06,
"logits/chosen": -0.5266455411911011,
"logits/rejected": -0.4706489145755768,
"logps/chosen": -285.48236083984375,
"logps/rejected": -343.0083923339844,
"loss": 0.6482,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.06514567881822586,
"rewards/margins": 0.07579077780246735,
"rewards/rejected": -0.010645096190273762,
"step": 330
},
{
"epoch": 0.18,
"learning_rate": 4.907881841897216e-06,
"logits/chosen": -0.43907618522644043,
"logits/rejected": -0.5358369946479797,
"logps/chosen": -377.4954528808594,
"logps/rejected": -327.7589111328125,
"loss": 0.641,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.13037431240081787,
"rewards/margins": 0.13443666696548462,
"rewards/rejected": -0.004062363877892494,
"step": 340
},
{
"epoch": 0.18,
"learning_rate": 4.89519206674919e-06,
"logits/chosen": -0.4344902038574219,
"logits/rejected": -0.5312429666519165,
"logps/chosen": -323.53448486328125,
"logps/rejected": -311.77496337890625,
"loss": 0.6242,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.09232774376869202,
"rewards/margins": 0.17667637765407562,
"rewards/rejected": -0.0843486562371254,
"step": 350
},
{
"epoch": 0.19,
"learning_rate": 4.881702316907769e-06,
"logits/chosen": -0.5391712784767151,
"logits/rejected": -0.533171534538269,
"logps/chosen": -264.92230224609375,
"logps/rejected": -285.822998046875,
"loss": 0.6303,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.05296499282121658,
"rewards/margins": 0.13872875273227692,
"rewards/rejected": -0.08576375991106033,
"step": 360
},
{
"epoch": 0.19,
"learning_rate": 4.86741709783982e-06,
"logits/chosen": -0.5374969244003296,
"logits/rejected": -0.5305294394493103,
"logps/chosen": -330.28271484375,
"logps/rejected": -368.3117980957031,
"loss": 0.6336,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.04659426584839821,
"rewards/margins": 0.18263372778892517,
"rewards/rejected": -0.13603946566581726,
"step": 370
},
{
"epoch": 0.2,
"learning_rate": 4.852341180692471e-06,
"logits/chosen": -0.506037712097168,
"logits/rejected": -0.5441786646842957,
"logps/chosen": -345.60064697265625,
"logps/rejected": -334.61407470703125,
"loss": 0.6298,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.023661959916353226,
"rewards/margins": 0.12598641216754913,
"rewards/rejected": -0.14964835345745087,
"step": 380
},
{
"epoch": 0.2,
"learning_rate": 4.836479600699579e-06,
"logits/chosen": -0.6301513314247131,
"logits/rejected": -0.615659236907959,
"logps/chosen": -330.850830078125,
"logps/rejected": -338.16070556640625,
"loss": 0.6262,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.07495202869176865,
"rewards/margins": 0.1975657194852829,
"rewards/rejected": -0.27251774072647095,
"step": 390
},
{
"epoch": 0.21,
"learning_rate": 4.819837655500014e-06,
"logits/chosen": -0.5807877779006958,
"logits/rejected": -0.6708138585090637,
"logps/chosen": -367.5466003417969,
"logps/rejected": -348.1293640136719,
"loss": 0.6372,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.1158147007226944,
"rewards/margins": 0.1703101098537445,
"rewards/rejected": -0.2861248254776001,
"step": 400
},
{
"epoch": 0.21,
"eval_logits/chosen": -0.5491507649421692,
"eval_logits/rejected": -0.5776416659355164,
"eval_logps/chosen": -357.9351806640625,
"eval_logps/rejected": -345.4769287109375,
"eval_loss": 0.6271597743034363,
"eval_rewards/accuracies": 0.6865079402923584,
"eval_rewards/chosen": -0.1131698414683342,
"eval_rewards/margins": 0.19984641671180725,
"eval_rewards/rejected": -0.31301626563072205,
"eval_runtime": 303.4179,
"eval_samples_per_second": 6.592,
"eval_steps_per_second": 0.208,
"step": 400
},
{
"epoch": 0.21,
"learning_rate": 4.802420903368286e-06,
"logits/chosen": -0.6273578405380249,
"logits/rejected": -0.6960797309875488,
"logps/chosen": -361.93359375,
"logps/rejected": -319.00341796875,
"loss": 0.6421,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.197979137301445,
"rewards/margins": 0.10214652866125107,
"rewards/rejected": -0.3001256585121155,
"step": 410
},
{
"epoch": 0.22,
"learning_rate": 4.784235161358124e-06,
"logits/chosen": -0.5516955852508545,
"logits/rejected": -0.6049106121063232,
"logps/chosen": -367.198974609375,
"logps/rejected": -363.4229736328125,
"loss": 0.6268,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.12621311843395233,
"rewards/margins": 0.13846664130687714,
"rewards/rejected": -0.26467975974082947,
"step": 420
},
{
"epoch": 0.23,
"learning_rate": 4.765286503359632e-06,
"logits/chosen": -0.5340523719787598,
"logits/rejected": -0.6084105372428894,
"logps/chosen": -407.4958190917969,
"logps/rejected": -401.229736328125,
"loss": 0.6107,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.05744675546884537,
"rewards/margins": 0.27851784229278564,
"rewards/rejected": -0.3359646201133728,
"step": 430
},
{
"epoch": 0.23,
"learning_rate": 4.745581258070654e-06,
"logits/chosen": -0.5335565805435181,
"logits/rejected": -0.5681129693984985,
"logps/chosen": -342.0080261230469,
"logps/rejected": -333.02569580078125,
"loss": 0.6243,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.16289742290973663,
"rewards/margins": 0.25955528020858765,
"rewards/rejected": -0.4224526882171631,
"step": 440
},
{
"epoch": 0.24,
"learning_rate": 4.725126006883047e-06,
"logits/chosen": -0.554370105266571,
"logits/rejected": -0.6364303827285767,
"logps/chosen": -396.9653625488281,
"logps/rejected": -362.92730712890625,
"loss": 0.6176,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.19718773663043976,
"rewards/margins": 0.19910624623298645,
"rewards/rejected": -0.396293967962265,
"step": 450
},
{
"epoch": 0.24,
"learning_rate": 4.70392758168454e-06,
"logits/chosen": -0.624381422996521,
"logits/rejected": -0.6213779449462891,
"logps/chosen": -394.22039794921875,
"logps/rejected": -391.92388916015625,
"loss": 0.6054,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.17801611125469208,
"rewards/margins": 0.1999357044696808,
"rewards/rejected": -0.37795180082321167,
"step": 460
},
{
"epoch": 0.25,
"learning_rate": 4.68199306257695e-06,
"logits/chosen": -0.5325026512145996,
"logits/rejected": -0.6415473222732544,
"logps/chosen": -373.08441162109375,
"logps/rejected": -333.419189453125,
"loss": 0.6012,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.15178225934505463,
"rewards/margins": 0.23722997307777405,
"rewards/rejected": -0.38901224732398987,
"step": 470
},
{
"epoch": 0.25,
"learning_rate": 4.659329775511478e-06,
"logits/chosen": -0.5378426313400269,
"logits/rejected": -0.5987221002578735,
"logps/chosen": -354.837158203125,
"logps/rejected": -358.8296813964844,
"loss": 0.6251,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.1486106812953949,
"rewards/margins": 0.21964044868946075,
"rewards/rejected": -0.36825114488601685,
"step": 480
},
{
"epoch": 0.26,
"learning_rate": 4.635945289841902e-06,
"logits/chosen": -0.594926118850708,
"logits/rejected": -0.6234263181686401,
"logps/chosen": -376.20172119140625,
"logps/rejected": -401.26068115234375,
"loss": 0.602,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.07999664545059204,
"rewards/margins": 0.32993897795677185,
"rewards/rejected": -0.4099356234073639,
"step": 490
},
{
"epoch": 0.26,
"learning_rate": 4.611847415796476e-06,
"logits/chosen": -0.5384534001350403,
"logits/rejected": -0.5856305956840515,
"logps/chosen": -344.09375,
"logps/rejected": -328.9897766113281,
"loss": 0.6233,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.11651408672332764,
"rewards/margins": 0.17748796939849854,
"rewards/rejected": -0.2940020263195038,
"step": 500
},
{
"epoch": 0.26,
"eval_logits/chosen": -0.5494967103004456,
"eval_logits/rejected": -0.5819825530052185,
"eval_logps/chosen": -362.3848571777344,
"eval_logps/rejected": -356.7882080078125,
"eval_loss": 0.6162292957305908,
"eval_rewards/accuracies": 0.682539701461792,
"eval_rewards/chosen": -0.15766657888889313,
"eval_rewards/margins": 0.2684621512889862,
"eval_rewards/rejected": -0.4261287748813629,
"eval_runtime": 304.7575,
"eval_samples_per_second": 6.563,
"eval_steps_per_second": 0.207,
"step": 500
},
{
"epoch": 0.27,
"learning_rate": 4.587044201869378e-06,
"logits/chosen": -0.5900930166244507,
"logits/rejected": -0.604232907295227,
"logps/chosen": -365.4901123046875,
"logps/rejected": -390.84246826171875,
"loss": 0.5966,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.15175417065620422,
"rewards/margins": 0.3504869043827057,
"rewards/rejected": -0.5022410750389099,
"step": 510
},
{
"epoch": 0.27,
"learning_rate": 4.561543932132574e-06,
"logits/chosen": -0.5539441108703613,
"logits/rejected": -0.6341567039489746,
"logps/chosen": -371.22308349609375,
"logps/rejected": -340.50457763671875,
"loss": 0.6261,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.36344796419143677,
"rewards/margins": 0.220851331949234,
"rewards/rejected": -0.5842992663383484,
"step": 520
},
{
"epoch": 0.28,
"learning_rate": 4.535355123469009e-06,
"logits/chosen": -0.7428617477416992,
"logits/rejected": -0.7538031935691833,
"logps/chosen": -303.5591735839844,
"logps/rejected": -322.505126953125,
"loss": 0.5975,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.45155367255210876,
"rewards/margins": 0.2669391930103302,
"rewards/rejected": -0.718492865562439,
"step": 530
},
{
"epoch": 0.28,
"learning_rate": 4.508486522728037e-06,
"logits/chosen": -0.5991286039352417,
"logits/rejected": -0.6453845500946045,
"logps/chosen": -350.3877258300781,
"logps/rejected": -354.8768615722656,
"loss": 0.6001,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.24457433819770813,
"rewards/margins": 0.3572844862937927,
"rewards/rejected": -0.6018588542938232,
"step": 540
},
{
"epoch": 0.29,
"learning_rate": 4.480947103804044e-06,
"logits/chosen": -0.6068693399429321,
"logits/rejected": -0.6633307337760925,
"logps/chosen": -366.67510986328125,
"logps/rejected": -344.18536376953125,
"loss": 0.6145,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.4160049557685852,
"rewards/margins": 0.27529022097587585,
"rewards/rejected": -0.6912952065467834,
"step": 550
},
{
"epoch": 0.29,
"learning_rate": 4.452746064639239e-06,
"logits/chosen": -0.6586125493049622,
"logits/rejected": -0.6525910496711731,
"logps/chosen": -389.03436279296875,
"logps/rejected": -459.2569274902344,
"loss": 0.602,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.4362773895263672,
"rewards/margins": 0.41235917806625366,
"rewards/rejected": -0.8486365079879761,
"step": 560
},
{
"epoch": 0.3,
"learning_rate": 4.423892824151617e-06,
"logits/chosen": -0.6040334701538086,
"logits/rejected": -0.6174503564834595,
"logps/chosen": -361.7618408203125,
"logps/rejected": -354.9851379394531,
"loss": 0.6068,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.33209994435310364,
"rewards/margins": 0.2956308126449585,
"rewards/rejected": -0.6277307271957397,
"step": 570
},
{
"epoch": 0.3,
"learning_rate": 4.3943970190891164e-06,
"logits/chosen": -0.5234232544898987,
"logits/rejected": -0.6443308591842651,
"logps/chosen": -349.57867431640625,
"logps/rejected": -357.2449645996094,
"loss": 0.584,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.3364105820655823,
"rewards/margins": 0.3499498665332794,
"rewards/rejected": -0.6863604784011841,
"step": 580
},
{
"epoch": 0.31,
"learning_rate": 4.364268500811025e-06,
"logits/chosen": -0.5796949863433838,
"logits/rejected": -0.6078803539276123,
"logps/chosen": -393.03057861328125,
"logps/rejected": -394.01068115234375,
"loss": 0.5883,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.3195868730545044,
"rewards/margins": 0.2843206822872162,
"rewards/rejected": -0.603907585144043,
"step": 590
},
{
"epoch": 0.31,
"learning_rate": 4.333517331997704e-06,
"logits/chosen": -0.4854974150657654,
"logits/rejected": -0.5861517786979675,
"logps/chosen": -409.95849609375,
"logps/rejected": -371.15240478515625,
"loss": 0.5951,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.351412832736969,
"rewards/margins": 0.25703001022338867,
"rewards/rejected": -0.6084428429603577,
"step": 600
},
{
"epoch": 0.31,
"eval_logits/chosen": -0.573330819606781,
"eval_logits/rejected": -0.6100495457649231,
"eval_logps/chosen": -380.79119873046875,
"eval_logps/rejected": -382.4302978515625,
"eval_loss": 0.6062807440757751,
"eval_rewards/accuracies": 0.6805555820465088,
"eval_rewards/chosen": -0.3417300879955292,
"eval_rewards/margins": 0.34081971645355225,
"eval_rewards/rejected": -0.6825497150421143,
"eval_runtime": 305.3418,
"eval_samples_per_second": 6.55,
"eval_steps_per_second": 0.206,
"step": 600
},
{
"epoch": 0.32,
"learning_rate": 4.302153783289737e-06,
"logits/chosen": -0.5746784210205078,
"logits/rejected": -0.6247537136077881,
"logps/chosen": -327.1573791503906,
"logps/rejected": -373.3064270019531,
"loss": 0.5644,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.30045533180236816,
"rewards/margins": 0.5708844661712646,
"rewards/rejected": -0.8713397979736328,
"step": 610
},
{
"epoch": 0.32,
"learning_rate": 4.270188329857613e-06,
"logits/chosen": -0.5362870097160339,
"logits/rejected": -0.603136420249939,
"logps/chosen": -377.772705078125,
"logps/rejected": -375.9126281738281,
"loss": 0.5612,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.35553207993507385,
"rewards/margins": 0.38167324662208557,
"rewards/rejected": -0.7372053265571594,
"step": 620
},
{
"epoch": 0.33,
"learning_rate": 4.237631647903115e-06,
"logits/chosen": -0.5939895510673523,
"logits/rejected": -0.6751977801322937,
"logps/chosen": -431.8272399902344,
"logps/rejected": -434.0873107910156,
"loss": 0.5784,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.404622882604599,
"rewards/margins": 0.4936921000480652,
"rewards/rejected": -0.8983149528503418,
"step": 630
},
{
"epoch": 0.33,
"learning_rate": 4.204494611093548e-06,
"logits/chosen": -0.5942648649215698,
"logits/rejected": -0.6419891119003296,
"logps/chosen": -398.41192626953125,
"logps/rejected": -402.21881103515625,
"loss": 0.5819,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5822068452835083,
"rewards/margins": 0.2493685930967331,
"rewards/rejected": -0.8315755128860474,
"step": 640
},
{
"epoch": 0.34,
"learning_rate": 4.170788286930024e-06,
"logits/chosen": -0.6248338222503662,
"logits/rejected": -0.6994636654853821,
"logps/chosen": -358.5780029296875,
"logps/rejected": -354.32513427734375,
"loss": 0.6212,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5734933614730835,
"rewards/margins": 0.3680447041988373,
"rewards/rejected": -0.9415380358695984,
"step": 650
},
{
"epoch": 0.35,
"learning_rate": 4.136523933051005e-06,
"logits/chosen": -0.5910335183143616,
"logits/rejected": -0.6014568209648132,
"logps/chosen": -367.816162109375,
"logps/rejected": -425.21478271484375,
"loss": 0.5867,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6200834512710571,
"rewards/margins": 0.457810640335083,
"rewards/rejected": -1.0778939723968506,
"step": 660
},
{
"epoch": 0.35,
"learning_rate": 4.101712993472348e-06,
"logits/chosen": -0.45557427406311035,
"logits/rejected": -0.586103081703186,
"logps/chosen": -435.7613220214844,
"logps/rejected": -412.81610107421875,
"loss": 0.5938,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5693609714508057,
"rewards/margins": 0.3782386779785156,
"rewards/rejected": -0.9475995898246765,
"step": 670
},
{
"epoch": 0.36,
"learning_rate": 4.066367094765091e-06,
"logits/chosen": -0.6208279132843018,
"logits/rejected": -0.6261738538742065,
"logps/chosen": -346.9092712402344,
"logps/rejected": -385.6962890625,
"loss": 0.6158,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6417191624641418,
"rewards/margins": 0.22618098556995392,
"rewards/rejected": -0.8679000735282898,
"step": 680
},
{
"epoch": 0.36,
"learning_rate": 4.030498042172277e-06,
"logits/chosen": -0.4598918855190277,
"logits/rejected": -0.6179080009460449,
"logps/chosen": -456.4967346191406,
"logps/rejected": -428.18096923828125,
"loss": 0.58,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.3372291326522827,
"rewards/margins": 0.46832937002182007,
"rewards/rejected": -0.8055585026741028,
"step": 690
},
{
"epoch": 0.37,
"learning_rate": 3.994117815666095e-06,
"logits/chosen": -0.514435350894928,
"logits/rejected": -0.5911833047866821,
"logps/chosen": -425.69085693359375,
"logps/rejected": -387.5975646972656,
"loss": 0.6051,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.4158889651298523,
"rewards/margins": 0.2637188136577606,
"rewards/rejected": -0.6796077489852905,
"step": 700
},
{
"epoch": 0.37,
"eval_logits/chosen": -0.5761136412620544,
"eval_logits/rejected": -0.6107844710350037,
"eval_logps/chosen": -395.6782531738281,
"eval_logps/rejected": -402.2430725097656,
"eval_loss": 0.5972979068756104,
"eval_rewards/accuracies": 0.6944444179534912,
"eval_rewards/chosen": -0.4906010627746582,
"eval_rewards/margins": 0.39007675647735596,
"eval_rewards/rejected": -0.8806778192520142,
"eval_runtime": 305.6972,
"eval_samples_per_second": 6.542,
"eval_steps_per_second": 0.206,
"step": 700
},
{
"epoch": 0.37,
"learning_rate": 3.957238565946672e-06,
"logits/chosen": -0.5273677706718445,
"logits/rejected": -0.5377794504165649,
"logps/chosen": -402.70660400390625,
"logps/rejected": -437.87628173828125,
"loss": 0.5903,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.46286138892173767,
"rewards/margins": 0.4086821675300598,
"rewards/rejected": -0.8715435266494751,
"step": 710
},
{
"epoch": 0.38,
"learning_rate": 3.919872610383831e-06,
"logits/chosen": -0.6572099924087524,
"logits/rejected": -0.6273663640022278,
"logps/chosen": -354.00579833984375,
"logps/rejected": -382.08441162109375,
"loss": 0.6022,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.5593703389167786,
"rewards/margins": 0.3321402668952942,
"rewards/rejected": -0.8915106654167175,
"step": 720
},
{
"epoch": 0.38,
"learning_rate": 3.882032428903195e-06,
"logits/chosen": -0.6124129891395569,
"logits/rejected": -0.6364959478378296,
"logps/chosen": -382.92462158203125,
"logps/rejected": -385.5881042480469,
"loss": 0.5791,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6445373892784119,
"rewards/margins": 0.42275238037109375,
"rewards/rejected": -1.0672898292541504,
"step": 730
},
{
"epoch": 0.39,
"learning_rate": 3.84373065981799e-06,
"logits/chosen": -0.5338658690452576,
"logits/rejected": -0.6029759049415588,
"logps/chosen": -454.47900390625,
"logps/rejected": -433.13140869140625,
"loss": 0.5668,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.659946858882904,
"rewards/margins": 0.5364845395088196,
"rewards/rejected": -1.1964313983917236,
"step": 740
},
{
"epoch": 0.39,
"learning_rate": 3.8049800956079552e-06,
"logits/chosen": -0.6849664449691772,
"logits/rejected": -0.7146345376968384,
"logps/chosen": -375.855224609375,
"logps/rejected": -412.647216796875,
"loss": 0.5968,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.7736431360244751,
"rewards/margins": 0.37692931294441223,
"rewards/rejected": -1.150572419166565,
"step": 750
},
{
"epoch": 0.4,
"learning_rate": 3.765793678646753e-06,
"logits/chosen": -0.5949341058731079,
"logits/rejected": -0.630172610282898,
"logps/chosen": -380.9452209472656,
"logps/rejected": -441.0630798339844,
"loss": 0.567,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.4381617605686188,
"rewards/margins": 0.5467316508293152,
"rewards/rejected": -0.9848934412002563,
"step": 760
},
{
"epoch": 0.4,
"learning_rate": 3.726184496879323e-06,
"logits/chosen": -0.5766685009002686,
"logits/rejected": -0.6258134841918945,
"logps/chosen": -385.54351806640625,
"logps/rejected": -396.5734558105469,
"loss": 0.5823,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5654970407485962,
"rewards/margins": 0.3876551687717438,
"rewards/rejected": -0.9531521797180176,
"step": 770
},
{
"epoch": 0.41,
"learning_rate": 3.686165779450619e-06,
"logits/chosen": -0.5654915571212769,
"logits/rejected": -0.6695746183395386,
"logps/chosen": -422.5735778808594,
"logps/rejected": -401.92034912109375,
"loss": 0.6027,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6763705015182495,
"rewards/margins": 0.4181916117668152,
"rewards/rejected": -1.0945621728897095,
"step": 780
},
{
"epoch": 0.41,
"learning_rate": 3.645750892287178e-06,
"logits/chosen": -0.5739408731460571,
"logits/rejected": -0.649164617061615,
"logps/chosen": -439.37615966796875,
"logps/rejected": -465.0233459472656,
"loss": 0.5805,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.711835503578186,
"rewards/margins": 0.45104461908340454,
"rewards/rejected": -1.1628801822662354,
"step": 790
},
{
"epoch": 0.42,
"learning_rate": 3.604953333633009e-06,
"logits/chosen": -0.5284892916679382,
"logits/rejected": -0.6252182722091675,
"logps/chosen": -437.35821533203125,
"logps/rejected": -434.4740295410156,
"loss": 0.5632,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6502100229263306,
"rewards/margins": 0.4485486149787903,
"rewards/rejected": -1.098758578300476,
"step": 800
},
{
"epoch": 0.42,
"eval_logits/chosen": -0.5841491222381592,
"eval_logits/rejected": -0.6245321035385132,
"eval_logps/chosen": -409.95855712890625,
"eval_logps/rejected": -422.52947998046875,
"eval_loss": 0.5927994251251221,
"eval_rewards/accuracies": 0.7023809552192688,
"eval_rewards/chosen": -0.6334035396575928,
"eval_rewards/margins": 0.45013830065727234,
"eval_rewards/rejected": -1.083541750907898,
"eval_runtime": 304.2196,
"eval_samples_per_second": 6.574,
"eval_steps_per_second": 0.207,
"step": 800
},
{
"epoch": 0.42,
"learning_rate": 3.56378672954129e-06,
"logits/chosen": -0.5947645902633667,
"logits/rejected": -0.6664719581604004,
"logps/chosen": -407.41339111328125,
"logps/rejected": -436.47381591796875,
"loss": 0.5897,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6832540035247803,
"rewards/margins": 0.5449843406677246,
"rewards/rejected": -1.2282384634017944,
"step": 810
},
{
"epoch": 0.43,
"learning_rate": 3.5222648293233806e-06,
"logits/chosen": -0.5373682379722595,
"logits/rejected": -0.5452297329902649,
"logps/chosen": -385.5528869628906,
"logps/rejected": -416.3194885253906,
"loss": 0.601,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.4152299463748932,
"rewards/margins": 0.42649930715560913,
"rewards/rejected": -0.8417292833328247,
"step": 820
},
{
"epoch": 0.43,
"learning_rate": 3.4804015009566573e-06,
"logits/chosen": -0.5158123970031738,
"logits/rejected": -0.6795430183410645,
"logps/chosen": -412.291015625,
"logps/rejected": -380.50860595703125,
"loss": 0.5845,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.5058903694152832,
"rewards/margins": 0.47143155336380005,
"rewards/rejected": -0.977321982383728,
"step": 830
},
{
"epoch": 0.44,
"learning_rate": 3.4382107264527244e-06,
"logits/chosen": -0.587105929851532,
"logits/rejected": -0.6171265840530396,
"logps/chosen": -391.59991455078125,
"logps/rejected": -412.62750244140625,
"loss": 0.5836,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5484491586685181,
"rewards/margins": 0.42554783821105957,
"rewards/rejected": -0.9739969968795776,
"step": 840
},
{
"epoch": 0.44,
"learning_rate": 3.3957065971875387e-06,
"logits/chosen": -0.5018014311790466,
"logits/rejected": -0.5997334718704224,
"logps/chosen": -488.0768127441406,
"logps/rejected": -467.55072021484375,
"loss": 0.61,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7004522085189819,
"rewards/margins": 0.5092984437942505,
"rewards/rejected": -1.2097506523132324,
"step": 850
},
{
"epoch": 0.45,
"learning_rate": 3.352903309194999e-06,
"logits/chosen": -0.5440029501914978,
"logits/rejected": -0.5686444044113159,
"logps/chosen": -364.2088623046875,
"logps/rejected": -364.9720153808594,
"loss": 0.6065,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5453570485115051,
"rewards/margins": 0.4008842408657074,
"rewards/rejected": -0.9462413787841797,
"step": 860
},
{
"epoch": 0.46,
"learning_rate": 3.309815158425591e-06,
"logits/chosen": -0.5663027167320251,
"logits/rejected": -0.6566844582557678,
"logps/chosen": -410.7333984375,
"logps/rejected": -389.6551513671875,
"loss": 0.5911,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.4486373960971832,
"rewards/margins": 0.37078288197517395,
"rewards/rejected": -0.8194202184677124,
"step": 870
},
{
"epoch": 0.46,
"learning_rate": 3.266456535971654e-06,
"logits/chosen": -0.5438618659973145,
"logits/rejected": -0.5869525671005249,
"logps/chosen": -430.4832458496094,
"logps/rejected": -415.47711181640625,
"loss": 0.6088,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6006536483764648,
"rewards/margins": 0.3156343400478363,
"rewards/rejected": -0.9162880778312683,
"step": 880
},
{
"epoch": 0.47,
"learning_rate": 3.2228419232608692e-06,
"logits/chosen": -0.5569460391998291,
"logits/rejected": -0.5293912887573242,
"logps/chosen": -342.6893615722656,
"logps/rejected": -377.5931701660156,
"loss": 0.5956,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5601915121078491,
"rewards/margins": 0.33136099576950073,
"rewards/rejected": -0.8915525674819946,
"step": 890
},
{
"epoch": 0.47,
"learning_rate": 3.1789858872195888e-06,
"logits/chosen": -0.49963292479515076,
"logits/rejected": -0.6308623552322388,
"logps/chosen": -407.0679931640625,
"logps/rejected": -378.2735900878906,
"loss": 0.6015,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.5700783133506775,
"rewards/margins": 0.47629982233047485,
"rewards/rejected": -1.0463781356811523,
"step": 900
},
{
"epoch": 0.47,
"eval_logits/chosen": -0.5358740091323853,
"eval_logits/rejected": -0.5755640268325806,
"eval_logps/chosen": -407.6412353515625,
"eval_logps/rejected": -420.5953063964844,
"eval_loss": 0.5896004438400269,
"eval_rewards/accuracies": 0.7123016119003296,
"eval_rewards/chosen": -0.6102306246757507,
"eval_rewards/margins": 0.4539690315723419,
"eval_rewards/rejected": -1.064199686050415,
"eval_runtime": 305.5138,
"eval_samples_per_second": 6.546,
"eval_steps_per_second": 0.206,
"step": 900
},
{
"epoch": 0.48,
"learning_rate": 3.1349030754075945e-06,
"logits/chosen": -0.5590487718582153,
"logits/rejected": -0.6419841051101685,
"logps/chosen": -426.57550048828125,
"logps/rejected": -397.7510681152344,
"loss": 0.5658,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6472785472869873,
"rewards/margins": 0.4323779046535492,
"rewards/rejected": -1.0796566009521484,
"step": 910
},
{
"epoch": 0.48,
"learning_rate": 3.0906082111259313e-06,
"logits/chosen": -0.571009635925293,
"logits/rejected": -0.5964629054069519,
"logps/chosen": -411.31201171875,
"logps/rejected": -410.85260009765625,
"loss": 0.593,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6876448392868042,
"rewards/margins": 0.34134042263031006,
"rewards/rejected": -1.0289852619171143,
"step": 920
},
{
"epoch": 0.49,
"learning_rate": 3.046116088499449e-06,
"logits/chosen": -0.5807533264160156,
"logits/rejected": -0.6887992024421692,
"logps/chosen": -456.3690490722656,
"logps/rejected": -431.4046936035156,
"loss": 0.5647,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6661675572395325,
"rewards/margins": 0.4776817262172699,
"rewards/rejected": -1.14384925365448,
"step": 930
},
{
"epoch": 0.49,
"learning_rate": 3.0014415675356813e-06,
"logits/chosen": -0.5229719877243042,
"logits/rejected": -0.6582568287849426,
"logps/chosen": -381.76239013671875,
"logps/rejected": -381.06927490234375,
"loss": 0.5761,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.5734530091285706,
"rewards/margins": 0.5467861294746399,
"rewards/rejected": -1.120239019393921,
"step": 940
},
{
"epoch": 0.5,
"learning_rate": 2.9565995691617242e-06,
"logits/chosen": -0.5676053166389465,
"logits/rejected": -0.5334717035293579,
"logps/chosen": -374.41717529296875,
"logps/rejected": -421.0732421875,
"loss": 0.6225,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.6005482077598572,
"rewards/margins": 0.21005001664161682,
"rewards/rejected": -0.8105981945991516,
"step": 950
},
{
"epoch": 0.5,
"learning_rate": 2.9116050702407706e-06,
"logits/chosen": -0.44915372133255005,
"logits/rejected": -0.5996781587600708,
"logps/chosen": -453.83343505859375,
"logps/rejected": -393.98333740234375,
"loss": 0.5957,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6119068264961243,
"rewards/margins": 0.29583027958869934,
"rewards/rejected": -0.9077370762825012,
"step": 960
},
{
"epoch": 0.51,
"learning_rate": 2.8664730985699537e-06,
"logits/chosen": -0.5181512832641602,
"logits/rejected": -0.5320832133293152,
"logps/chosen": -359.99542236328125,
"logps/rejected": -414.4845275878906,
"loss": 0.5795,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5852144956588745,
"rewards/margins": 0.4881064295768738,
"rewards/rejected": -1.073320984840393,
"step": 970
},
{
"epoch": 0.51,
"learning_rate": 2.8212187278611907e-06,
"logits/chosen": -0.5162011981010437,
"logits/rejected": -0.6026550531387329,
"logps/chosen": -435.4556579589844,
"logps/rejected": -434.9972229003906,
"loss": 0.5648,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.5601040124893188,
"rewards/margins": 0.5399636030197144,
"rewards/rejected": -1.1000676155090332,
"step": 980
},
{
"epoch": 0.52,
"learning_rate": 2.7758570727066843e-06,
"logits/chosen": -0.5119590759277344,
"logits/rejected": -0.6090120673179626,
"logps/chosen": -407.456298828125,
"logps/rejected": -399.5018310546875,
"loss": 0.5684,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6310015916824341,
"rewards/margins": 0.42230549454689026,
"rewards/rejected": -1.053307056427002,
"step": 990
},
{
"epoch": 0.52,
"learning_rate": 2.730403283530767e-06,
"logits/chosen": -0.5462077856063843,
"logits/rejected": -0.603754997253418,
"logps/chosen": -436.95245361328125,
"logps/rejected": -415.5531311035156,
"loss": 0.5756,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.71515291929245,
"rewards/margins": 0.3384590148925781,
"rewards/rejected": -1.0536119937896729,
"step": 1000
},
{
"epoch": 0.52,
"eval_logits/chosen": -0.50582355260849,
"eval_logits/rejected": -0.5430945754051208,
"eval_logps/chosen": -411.3542785644531,
"eval_logps/rejected": -426.3283996582031,
"eval_loss": 0.5865140557289124,
"eval_rewards/accuracies": 0.6984127163887024,
"eval_rewards/chosen": -0.6473610401153564,
"eval_rewards/margins": 0.4741695821285248,
"eval_rewards/rejected": -1.1215306520462036,
"eval_runtime": 302.4755,
"eval_samples_per_second": 6.612,
"eval_steps_per_second": 0.208,
"step": 1000
},
{
"epoch": 0.53,
"learning_rate": 2.6848725415297888e-06,
"logits/chosen": -0.5315025448799133,
"logits/rejected": -0.6642025709152222,
"logps/chosen": -466.88250732421875,
"logps/rejected": -440.3934020996094,
"loss": 0.5408,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.6935629844665527,
"rewards/margins": 0.5693783760070801,
"rewards/rejected": -1.2629413604736328,
"step": 1010
},
{
"epoch": 0.53,
"learning_rate": 2.639280053601719e-06,
"logits/chosen": -0.4503622055053711,
"logits/rejected": -0.515163779258728,
"logps/chosen": -388.44940185546875,
"logps/rejected": -394.9497985839844,
"loss": 0.5909,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6574369668960571,
"rewards/margins": 0.3425886929035187,
"rewards/rejected": -1.000025749206543,
"step": 1020
},
{
"epoch": 0.54,
"learning_rate": 2.59364104726716e-06,
"logits/chosen": -0.5630380511283875,
"logits/rejected": -0.6658009886741638,
"logps/chosen": -403.2070617675781,
"logps/rejected": -395.23150634765625,
"loss": 0.5597,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.5559983253479004,
"rewards/margins": 0.5452179908752441,
"rewards/rejected": -1.1012163162231445,
"step": 1030
},
{
"epoch": 0.54,
"learning_rate": 2.547970765583491e-06,
"logits/chosen": -0.5626110434532166,
"logits/rejected": -0.633199155330658,
"logps/chosen": -394.472412109375,
"logps/rejected": -418.28741455078125,
"loss": 0.5756,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.8999873399734497,
"rewards/margins": 0.44079795479774475,
"rewards/rejected": -1.340785264968872,
"step": 1040
},
{
"epoch": 0.55,
"learning_rate": 2.502284462053799e-06,
"logits/chosen": -0.5563883185386658,
"logits/rejected": -0.6150022745132446,
"logps/chosen": -443.32977294921875,
"logps/rejected": -443.04248046875,
"loss": 0.5724,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.826148509979248,
"rewards/margins": 0.5606396198272705,
"rewards/rejected": -1.3867881298065186,
"step": 1050
},
{
"epoch": 0.55,
"learning_rate": 2.456597395532338e-06,
"logits/chosen": -0.4988153874874115,
"logits/rejected": -0.5230491161346436,
"logps/chosen": -409.4883728027344,
"logps/rejected": -461.45184326171875,
"loss": 0.5785,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.9212621450424194,
"rewards/margins": 0.43989840149879456,
"rewards/rejected": -1.3611605167388916,
"step": 1060
},
{
"epoch": 0.56,
"learning_rate": 2.4109248251281953e-06,
"logits/chosen": -0.5152363777160645,
"logits/rejected": -0.5550749897956848,
"logps/chosen": -412.5245056152344,
"logps/rejected": -469.71392822265625,
"loss": 0.5708,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.7094501256942749,
"rewards/margins": 0.5959488749504089,
"rewards/rejected": -1.3053990602493286,
"step": 1070
},
{
"epoch": 0.57,
"learning_rate": 2.365282005108875e-06,
"logits/chosen": -0.40664130449295044,
"logits/rejected": -0.4538194537162781,
"logps/chosen": -474.70269775390625,
"logps/rejected": -478.2893981933594,
"loss": 0.5514,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6573441028594971,
"rewards/margins": 0.4294394850730896,
"rewards/rejected": -1.086783528327942,
"step": 1080
},
{
"epoch": 0.57,
"learning_rate": 2.319684179805491e-06,
"logits/chosen": -0.47670966386795044,
"logits/rejected": -0.5509120225906372,
"logps/chosen": -417.5166015625,
"logps/rejected": -451.0810546875,
"loss": 0.595,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.8204286694526672,
"rewards/margins": 0.3873991370201111,
"rewards/rejected": -1.2078278064727783,
"step": 1090
},
{
"epoch": 0.58,
"learning_rate": 2.2741465785212905e-06,
"logits/chosen": -0.5169596076011658,
"logits/rejected": -0.5811697244644165,
"logps/chosen": -409.420166015625,
"logps/rejected": -383.60638427734375,
"loss": 0.6024,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.791795015335083,
"rewards/margins": 0.3366108238697052,
"rewards/rejected": -1.1284058094024658,
"step": 1100
},
{
"epoch": 0.58,
"eval_logits/chosen": -0.5104440450668335,
"eval_logits/rejected": -0.5501194596290588,
"eval_logps/chosen": -419.2626037597656,
"eval_logps/rejected": -437.0024719238281,
"eval_loss": 0.5854523777961731,
"eval_rewards/accuracies": 0.7063491940498352,
"eval_rewards/chosen": -0.7264440655708313,
"eval_rewards/margins": 0.5018272399902344,
"eval_rewards/rejected": -1.228271245956421,
"eval_runtime": 308.3584,
"eval_samples_per_second": 6.486,
"eval_steps_per_second": 0.204,
"step": 1100
},
{
"epoch": 0.58,
"learning_rate": 2.2286844104451848e-06,
"logits/chosen": -0.5197004079818726,
"logits/rejected": -0.6333814859390259,
"logps/chosen": -416.119873046875,
"logps/rejected": -442.78363037109375,
"loss": 0.5806,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7725197076797485,
"rewards/margins": 0.5796698331832886,
"rewards/rejected": -1.3521894216537476,
"step": 1110
},
{
"epoch": 0.59,
"learning_rate": 2.183312859572008e-06,
"logits/chosen": -0.46561068296432495,
"logits/rejected": -0.5527454614639282,
"logps/chosen": -446.88372802734375,
"logps/rejected": -421.5301208496094,
"loss": 0.5718,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.8624758720397949,
"rewards/margins": 0.5891597867012024,
"rewards/rejected": -1.451635718345642,
"step": 1120
},
{
"epoch": 0.59,
"learning_rate": 2.1380470796311843e-06,
"logits/chosen": -0.38593560457229614,
"logits/rejected": -0.4362197518348694,
"logps/chosen": -429.46893310546875,
"logps/rejected": -482.09014892578125,
"loss": 0.5891,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9297833442687988,
"rewards/margins": 0.5348304510116577,
"rewards/rejected": -1.464613676071167,
"step": 1130
},
{
"epoch": 0.6,
"learning_rate": 2.092902189025507e-06,
"logits/chosen": -0.5493001937866211,
"logits/rejected": -0.6307727098464966,
"logps/chosen": -408.87017822265625,
"logps/rejected": -407.6255187988281,
"loss": 0.5762,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7089771032333374,
"rewards/margins": 0.5302673578262329,
"rewards/rejected": -1.2392443418502808,
"step": 1140
},
{
"epoch": 0.6,
"learning_rate": 2.0478932657817105e-06,
"logits/chosen": -0.44064339995384216,
"logits/rejected": -0.5361225605010986,
"logps/chosen": -439.63775634765625,
"logps/rejected": -438.837158203125,
"loss": 0.5686,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6102380156517029,
"rewards/margins": 0.5436003804206848,
"rewards/rejected": -1.1538383960723877,
"step": 1150
},
{
"epoch": 0.61,
"learning_rate": 2.0030353425145376e-06,
"logits/chosen": -0.526265561580658,
"logits/rejected": -0.5658246874809265,
"logps/chosen": -346.7317199707031,
"logps/rejected": -412.7989807128906,
"loss": 0.5961,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.733574390411377,
"rewards/margins": 0.4724843502044678,
"rewards/rejected": -1.2060587406158447,
"step": 1160
},
{
"epoch": 0.61,
"learning_rate": 1.958343401405964e-06,
"logits/chosen": -0.5053200721740723,
"logits/rejected": -0.5343925952911377,
"logps/chosen": -354.54681396484375,
"logps/rejected": -389.521240234375,
"loss": 0.5935,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.6786689758300781,
"rewards/margins": 0.24279697239398956,
"rewards/rejected": -0.921466052532196,
"step": 1170
},
{
"epoch": 0.62,
"learning_rate": 1.9138323692012734e-06,
"logits/chosen": -0.48046913743019104,
"logits/rejected": -0.5356425046920776,
"logps/chosen": -404.2420654296875,
"logps/rejected": -372.3519287109375,
"loss": 0.5909,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7317395806312561,
"rewards/margins": 0.2877105176448822,
"rewards/rejected": -1.019450068473816,
"step": 1180
},
{
"epoch": 0.62,
"learning_rate": 1.8695171122236443e-06,
"logits/chosen": -0.44227689504623413,
"logits/rejected": -0.5266855359077454,
"logps/chosen": -438.1207580566406,
"logps/rejected": -435.01824951171875,
"loss": 0.5636,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.6313630938529968,
"rewards/margins": 0.5005151629447937,
"rewards/rejected": -1.1318782567977905,
"step": 1190
},
{
"epoch": 0.63,
"learning_rate": 1.8254124314089225e-06,
"logits/chosen": -0.44814664125442505,
"logits/rejected": -0.3982391357421875,
"logps/chosen": -371.9776916503906,
"logps/rejected": -453.76397705078125,
"loss": 0.5578,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6842934489250183,
"rewards/margins": 0.3474181890487671,
"rewards/rejected": -1.0317118167877197,
"step": 1200
},
{
"epoch": 0.63,
"eval_logits/chosen": -0.4896259009838104,
"eval_logits/rejected": -0.5296720266342163,
"eval_logps/chosen": -415.6815490722656,
"eval_logps/rejected": -434.11138916015625,
"eval_loss": 0.582290768623352,
"eval_rewards/accuracies": 0.7142857313156128,
"eval_rewards/chosen": -0.6906338334083557,
"eval_rewards/margins": 0.5087268352508545,
"eval_rewards/rejected": -1.199360728263855,
"eval_runtime": 305.1471,
"eval_samples_per_second": 6.554,
"eval_steps_per_second": 0.206,
"step": 1200
},
{
"epoch": 0.63,
"learning_rate": 1.781533057362221e-06,
"logits/chosen": -0.45607271790504456,
"logits/rejected": -0.5114004015922546,
"logps/chosen": -411.0953063964844,
"logps/rejected": -410.81622314453125,
"loss": 0.564,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6678163409233093,
"rewards/margins": 0.5182808637619019,
"rewards/rejected": -1.1860973834991455,
"step": 1210
},
{
"epoch": 0.64,
"learning_rate": 1.7378936454380277e-06,
"logits/chosen": -0.4843919277191162,
"logits/rejected": -0.5591267347335815,
"logps/chosen": -461.26971435546875,
"logps/rejected": -451.92987060546875,
"loss": 0.5887,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.848225474357605,
"rewards/margins": 0.39363884925842285,
"rewards/rejected": -1.2418644428253174,
"step": 1220
},
{
"epoch": 0.64,
"learning_rate": 1.6945087708454273e-06,
"logits/chosen": -0.48674625158309937,
"logits/rejected": -0.5596768260002136,
"logps/chosen": -418.35748291015625,
"logps/rejected": -432.4639587402344,
"loss": 0.5957,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.8621279001235962,
"rewards/margins": 0.6262621879577637,
"rewards/rejected": -1.4883902072906494,
"step": 1230
},
{
"epoch": 0.65,
"learning_rate": 1.651392923780105e-06,
"logits/chosen": -0.4551656246185303,
"logits/rejected": -0.4784488081932068,
"logps/chosen": -442.0116271972656,
"logps/rejected": -449.7799377441406,
"loss": 0.5509,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.7251280546188354,
"rewards/margins": 0.5167814493179321,
"rewards/rejected": -1.2419095039367676,
"step": 1240
},
{
"epoch": 0.65,
"learning_rate": 1.608560504584737e-06,
"logits/chosen": -0.4653591215610504,
"logits/rejected": -0.5175148844718933,
"logps/chosen": -433.5137634277344,
"logps/rejected": -450.39263916015625,
"loss": 0.5434,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7668915390968323,
"rewards/margins": 0.5189617276191711,
"rewards/rejected": -1.2858531475067139,
"step": 1250
},
{
"epoch": 0.66,
"learning_rate": 1.5660258189393945e-06,
"logits/chosen": -0.45602017641067505,
"logits/rejected": -0.5941805243492126,
"logps/chosen": -421.2008361816406,
"logps/rejected": -438.20172119140625,
"loss": 0.6055,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7053720951080322,
"rewards/margins": 0.5101194977760315,
"rewards/rejected": -1.215491533279419,
"step": 1260
},
{
"epoch": 0.66,
"learning_rate": 1.5238030730835578e-06,
"logits/chosen": -0.4900667071342468,
"logits/rejected": -0.5974335670471191,
"logps/chosen": -439.05084228515625,
"logps/rejected": -448.49224853515625,
"loss": 0.5813,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6744861006736755,
"rewards/margins": 0.5117014050483704,
"rewards/rejected": -1.186187505722046,
"step": 1270
},
{
"epoch": 0.67,
"learning_rate": 1.4819063690713565e-06,
"logits/chosen": -0.5272679328918457,
"logits/rejected": -0.555342435836792,
"logps/chosen": -392.576416015625,
"logps/rejected": -438.9559631347656,
"loss": 0.5508,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.8570607304573059,
"rewards/margins": 0.5523840188980103,
"rewards/rejected": -1.4094446897506714,
"step": 1280
},
{
"epoch": 0.68,
"learning_rate": 1.4403497000615885e-06,
"logits/chosen": -0.364164263010025,
"logits/rejected": -0.3934030830860138,
"logps/chosen": -453.59210205078125,
"logps/rejected": -465.28912353515625,
"loss": 0.5526,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6626067161560059,
"rewards/margins": 0.4651568531990051,
"rewards/rejected": -1.1277635097503662,
"step": 1290
},
{
"epoch": 0.68,
"learning_rate": 1.3991469456441273e-06,
"logits/chosen": -0.4578070044517517,
"logits/rejected": -0.5681411027908325,
"logps/chosen": -439.03289794921875,
"logps/rejected": -432.52105712890625,
"loss": 0.5243,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6801882982254028,
"rewards/margins": 0.5355648398399353,
"rewards/rejected": -1.2157530784606934,
"step": 1300
},
{
"epoch": 0.68,
"eval_logits/chosen": -0.4929659962654114,
"eval_logits/rejected": -0.5339555740356445,
"eval_logps/chosen": -421.1521911621094,
"eval_logps/rejected": -441.3783264160156,
"eval_loss": 0.5803412795066833,
"eval_rewards/accuracies": 0.7142857313156128,
"eval_rewards/chosen": -0.7453399896621704,
"eval_rewards/margins": 0.5266899466514587,
"eval_rewards/rejected": -1.2720301151275635,
"eval_runtime": 305.6529,
"eval_samples_per_second": 6.543,
"eval_steps_per_second": 0.206,
"step": 1300
},
{
"epoch": 0.69,
"learning_rate": 1.3583118672042441e-06,
"logits/chosen": -0.5170468688011169,
"logits/rejected": -0.572604775428772,
"logps/chosen": -372.7608642578125,
"logps/rejected": -370.40911865234375,
"loss": 0.5896,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7620792388916016,
"rewards/margins": 0.42411160469055176,
"rewards/rejected": -1.1861908435821533,
"step": 1310
},
{
"epoch": 0.69,
"learning_rate": 1.3178581033264218e-06,
"logits/chosen": -0.5389008522033691,
"logits/rejected": -0.5711629986763,
"logps/chosen": -405.4247131347656,
"logps/rejected": -452.078125,
"loss": 0.5835,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8531595468521118,
"rewards/margins": 0.46848931908607483,
"rewards/rejected": -1.3216488361358643,
"step": 1320
},
{
"epoch": 0.7,
"learning_rate": 1.2777991652391757e-06,
"logits/chosen": -0.4288851320743561,
"logits/rejected": -0.4834524989128113,
"logps/chosen": -371.2089538574219,
"logps/rejected": -410.0118103027344,
"loss": 0.5504,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.8295769691467285,
"rewards/margins": 0.5004912614822388,
"rewards/rejected": -1.3300683498382568,
"step": 1330
},
{
"epoch": 0.7,
"learning_rate": 1.2381484323024178e-06,
"logits/chosen": -0.47288522124290466,
"logits/rejected": -0.5918864011764526,
"logps/chosen": -463.8346252441406,
"logps/rejected": -432.7979431152344,
"loss": 0.5696,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7991669774055481,
"rewards/margins": 0.45910316705703735,
"rewards/rejected": -1.2582701444625854,
"step": 1340
},
{
"epoch": 0.71,
"learning_rate": 1.1989191475388518e-06,
"logits/chosen": -0.41379016637802124,
"logits/rejected": -0.5847631692886353,
"logps/chosen": -464.99267578125,
"logps/rejected": -418.6041564941406,
"loss": 0.6093,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.67491215467453,
"rewards/margins": 0.5438855886459351,
"rewards/rejected": -1.2187979221343994,
"step": 1350
},
{
"epoch": 0.71,
"learning_rate": 1.160124413210918e-06,
"logits/chosen": -0.4751303791999817,
"logits/rejected": -0.5804455876350403,
"logps/chosen": -409.5224609375,
"logps/rejected": -407.48443603515625,
"loss": 0.5771,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7855092883110046,
"rewards/margins": 0.5121535062789917,
"rewards/rejected": -1.2976628541946411,
"step": 1360
},
{
"epoch": 0.72,
"learning_rate": 1.1217771864447396e-06,
"logits/chosen": -0.45879894495010376,
"logits/rejected": -0.5505935549736023,
"logps/chosen": -442.29931640625,
"logps/rejected": -427.472412109375,
"loss": 0.522,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.7745510339736938,
"rewards/margins": 0.5624163746833801,
"rewards/rejected": -1.3369674682617188,
"step": 1370
},
{
"epoch": 0.72,
"learning_rate": 1.08389027490255e-06,
"logits/chosen": -0.5081161856651306,
"logits/rejected": -0.5460919737815857,
"logps/chosen": -387.21722412109375,
"logps/rejected": -389.8815002441406,
"loss": 0.6193,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.8121119737625122,
"rewards/margins": 0.3596836030483246,
"rewards/rejected": -1.1717956066131592,
"step": 1380
},
{
"epoch": 0.73,
"learning_rate": 1.046476332505036e-06,
"logits/chosen": -0.47852253913879395,
"logits/rejected": -0.5535346269607544,
"logps/chosen": -409.1778869628906,
"logps/rejected": -408.6339111328125,
"loss": 0.5707,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.8597895503044128,
"rewards/margins": 0.43825215101242065,
"rewards/rejected": -1.298041582107544,
"step": 1390
},
{
"epoch": 0.73,
"learning_rate": 1.0095478552050348e-06,
"logits/chosen": -0.47451940178871155,
"logits/rejected": -0.5755375027656555,
"logps/chosen": -351.31610107421875,
"logps/rejected": -404.18328857421875,
"loss": 0.5343,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.6808342933654785,
"rewards/margins": 0.7804089784622192,
"rewards/rejected": -1.4612432718276978,
"step": 1400
},
{
"epoch": 0.73,
"eval_logits/chosen": -0.4871543347835541,
"eval_logits/rejected": -0.5270918011665344,
"eval_logps/chosen": -420.1601867675781,
"eval_logps/rejected": -440.79998779296875,
"eval_loss": 0.5805416107177734,
"eval_rewards/accuracies": 0.7103174328804016,
"eval_rewards/chosen": -0.7354201674461365,
"eval_rewards/margins": 0.5308260321617126,
"eval_rewards/rejected": -1.2662461996078491,
"eval_runtime": 304.0241,
"eval_samples_per_second": 6.578,
"eval_steps_per_second": 0.207,
"step": 1400
},
{
"epoch": 0.74,
"learning_rate": 9.731171768139808e-07,
"logits/chosen": -0.48386329412460327,
"logits/rejected": -0.5785359144210815,
"logps/chosen": -431.7328186035156,
"logps/rejected": -404.26019287109375,
"loss": 0.5709,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.6404847502708435,
"rewards/margins": 0.5482622385025024,
"rewards/rejected": -1.1887470483779907,
"step": 1410
},
{
"epoch": 0.74,
"learning_rate": 9.371964648825221e-07,
"logits/chosen": -0.4259588122367859,
"logits/rejected": -0.512924075126648,
"logps/chosen": -449.75921630859375,
"logps/rejected": -424.4246520996094,
"loss": 0.5421,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6056379079818726,
"rewards/margins": 0.6098628044128418,
"rewards/rejected": -1.2155007123947144,
"step": 1420
},
{
"epoch": 0.75,
"learning_rate": 9.017977166366445e-07,
"logits/chosen": -0.3833313286304474,
"logits/rejected": -0.4866362512111664,
"logps/chosen": -418.40606689453125,
"logps/rejected": -465.56805419921875,
"loss": 0.586,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6852587461471558,
"rewards/margins": 0.4969066083431244,
"rewards/rejected": -1.1821653842926025,
"step": 1430
},
{
"epoch": 0.75,
"learning_rate": 8.669327549707096e-07,
"logits/chosen": -0.5720198154449463,
"logits/rejected": -0.6064139604568481,
"logps/chosen": -372.54229736328125,
"logps/rejected": -417.3109436035156,
"loss": 0.5577,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.7984358072280884,
"rewards/margins": 0.6686285138130188,
"rewards/rejected": -1.467064380645752,
"step": 1440
},
{
"epoch": 0.76,
"learning_rate": 8.326132244986932e-07,
"logits/chosen": -0.48107537627220154,
"logits/rejected": -0.4693182110786438,
"logps/chosen": -417.77197265625,
"logps/rejected": -460.15692138671875,
"loss": 0.5416,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.7300211787223816,
"rewards/margins": 0.5799316763877869,
"rewards/rejected": -1.309952974319458,
"step": 1450
},
{
"epoch": 0.76,
"learning_rate": 7.988505876649863e-07,
"logits/chosen": -0.5137463808059692,
"logits/rejected": -0.5962769389152527,
"logps/chosen": -412.7781677246094,
"logps/rejected": -412.116943359375,
"loss": 0.5912,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7613190412521362,
"rewards/margins": 0.5068355798721313,
"rewards/rejected": -1.268154501914978,
"step": 1460
},
{
"epoch": 0.77,
"learning_rate": 7.656561209160248e-07,
"logits/chosen": -0.3398676812648773,
"logits/rejected": -0.5123169422149658,
"logps/chosen": -464.4519958496094,
"logps/rejected": -445.8551330566406,
"loss": 0.551,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.6963375210762024,
"rewards/margins": 0.6760309934616089,
"rewards/rejected": -1.3723684549331665,
"step": 1470
},
{
"epoch": 0.77,
"learning_rate": 7.330409109340563e-07,
"logits/chosen": -0.44637346267700195,
"logits/rejected": -0.4979945123195648,
"logps/chosen": -436.58123779296875,
"logps/rejected": -432.36126708984375,
"loss": 0.5833,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.779857337474823,
"rewards/margins": 0.5138452649116516,
"rewards/rejected": -1.293702483177185,
"step": 1480
},
{
"epoch": 0.78,
"learning_rate": 7.010158509342682e-07,
"logits/chosen": -0.5019239783287048,
"logits/rejected": -0.5278982520103455,
"logps/chosen": -399.7169494628906,
"logps/rejected": -425.452392578125,
"loss": 0.5786,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7415592074394226,
"rewards/margins": 0.5297465920448303,
"rewards/rejected": -1.271305799484253,
"step": 1490
},
{
"epoch": 0.79,
"learning_rate": 6.695916370265529e-07,
"logits/chosen": -0.4993990957736969,
"logits/rejected": -0.5913614630699158,
"logps/chosen": -424.80047607421875,
"logps/rejected": -439.79193115234375,
"loss": 0.5707,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6962804794311523,
"rewards/margins": 0.5849637985229492,
"rewards/rejected": -1.2812442779541016,
"step": 1500
},
{
"epoch": 0.79,
"eval_logits/chosen": -0.48642095923423767,
"eval_logits/rejected": -0.5267860889434814,
"eval_logps/chosen": -418.4040222167969,
"eval_logps/rejected": -439.218994140625,
"eval_loss": 0.5798574090003967,
"eval_rewards/accuracies": 0.7123016119003296,
"eval_rewards/chosen": -0.7178582549095154,
"eval_rewards/margins": 0.5325784683227539,
"eval_rewards/rejected": -1.250436782836914,
"eval_runtime": 305.4058,
"eval_samples_per_second": 6.549,
"eval_steps_per_second": 0.206,
"step": 1500
},
{
"epoch": 0.79,
"learning_rate": 6.387787646430854e-07,
"logits/chosen": -0.39493924379348755,
"logits/rejected": -0.4824163317680359,
"logps/chosen": -435.23944091796875,
"logps/rejected": -434.67938232421875,
"loss": 0.5349,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6798096895217896,
"rewards/margins": 0.579690158367157,
"rewards/rejected": -1.2594999074935913,
"step": 1510
},
{
"epoch": 0.8,
"learning_rate": 6.085875250329401e-07,
"logits/chosen": -0.5511216521263123,
"logits/rejected": -0.5999202728271484,
"logps/chosen": -452.65020751953125,
"logps/rejected": -435.93902587890625,
"loss": 0.5403,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.8256019353866577,
"rewards/margins": 0.45076680183410645,
"rewards/rejected": -1.2763688564300537,
"step": 1520
},
{
"epoch": 0.8,
"learning_rate": 5.79028001824894e-07,
"logits/chosen": -0.5369911193847656,
"logits/rejected": -0.519089937210083,
"logps/chosen": -452.34210205078125,
"logps/rejected": -478.9400939941406,
"loss": 0.5801,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7827368974685669,
"rewards/margins": 0.420412540435791,
"rewards/rejected": -1.2031495571136475,
"step": 1530
},
{
"epoch": 0.81,
"learning_rate": 5.501100676595761e-07,
"logits/chosen": -0.4579439163208008,
"logits/rejected": -0.44433537125587463,
"logps/chosen": -364.8140869140625,
"logps/rejected": -434.513916015625,
"loss": 0.5814,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7198919653892517,
"rewards/margins": 0.45986801385879517,
"rewards/rejected": -1.1797600984573364,
"step": 1540
},
{
"epoch": 0.81,
"learning_rate": 5.218433808920884e-07,
"logits/chosen": -0.5248664021492004,
"logits/rejected": -0.5349477529525757,
"logps/chosen": -437.393798828125,
"logps/rejected": -468.2730407714844,
"loss": 0.5523,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.881768524646759,
"rewards/margins": 0.5896928310394287,
"rewards/rejected": -1.471461296081543,
"step": 1550
},
{
"epoch": 0.82,
"learning_rate": 4.942373823661928e-07,
"logits/chosen": -0.4769573211669922,
"logits/rejected": -0.5523253679275513,
"logps/chosen": -423.78399658203125,
"logps/rejected": -398.61163330078125,
"loss": 0.6046,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.8796138763427734,
"rewards/margins": 0.2528652250766754,
"rewards/rejected": -1.132478952407837,
"step": 1560
},
{
"epoch": 0.82,
"learning_rate": 4.6730129226114363e-07,
"logits/chosen": -0.4421960711479187,
"logits/rejected": -0.4300961494445801,
"logps/chosen": -378.70233154296875,
"logps/rejected": -440.5601501464844,
"loss": 0.5843,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.8324155807495117,
"rewards/margins": 0.5455555319786072,
"rewards/rejected": -1.3779711723327637,
"step": 1570
},
{
"epoch": 0.83,
"learning_rate": 4.4104410701222703e-07,
"logits/chosen": -0.4712623655796051,
"logits/rejected": -0.5238856673240662,
"logps/chosen": -422.5208435058594,
"logps/rejected": -434.720947265625,
"loss": 0.586,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.8706847429275513,
"rewards/margins": 0.3749724328517914,
"rewards/rejected": -1.245656967163086,
"step": 1580
},
{
"epoch": 0.83,
"learning_rate": 4.154745963060197e-07,
"logits/chosen": -0.40005379915237427,
"logits/rejected": -0.4531996250152588,
"logps/chosen": -439.6205139160156,
"logps/rejected": -432.45318603515625,
"loss": 0.5706,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6769813895225525,
"rewards/margins": 0.5456241369247437,
"rewards/rejected": -1.222605586051941,
"step": 1590
},
{
"epoch": 0.84,
"learning_rate": 3.9060130015138863e-07,
"logits/chosen": -0.41965001821517944,
"logits/rejected": -0.5496610403060913,
"logps/chosen": -458.8155212402344,
"logps/rejected": -414.1279296875,
"loss": 0.5582,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6864231824874878,
"rewards/margins": 0.4698265492916107,
"rewards/rejected": -1.156249761581421,
"step": 1600
},
{
"epoch": 0.84,
"eval_logits/chosen": -0.4870251715183258,
"eval_logits/rejected": -0.5271284580230713,
"eval_logps/chosen": -419.6229553222656,
"eval_logps/rejected": -440.7270812988281,
"eval_loss": 0.5795175433158875,
"eval_rewards/accuracies": 0.7123016119003296,
"eval_rewards/chosen": -0.7300478219985962,
"eval_rewards/margins": 0.5354698896408081,
"eval_rewards/rejected": -1.2655178308486938,
"eval_runtime": 305.2401,
"eval_samples_per_second": 6.552,
"eval_steps_per_second": 0.206,
"step": 1600
},
{
"epoch": 0.84,
"learning_rate": 3.664325260271953e-07,
"logits/chosen": -0.528279721736908,
"logits/rejected": -0.6684530973434448,
"logps/chosen": -430.3279724121094,
"logps/rejected": -415.4619140625,
"loss": 0.5372,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.6265103220939636,
"rewards/margins": 0.6475422978401184,
"rewards/rejected": -1.274052619934082,
"step": 1610
},
{
"epoch": 0.85,
"learning_rate": 3.429763461076677e-07,
"logits/chosen": -0.3826778829097748,
"logits/rejected": -0.5138348937034607,
"logps/chosen": -440.0087890625,
"logps/rejected": -389.94049072265625,
"loss": 0.5901,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7695432901382446,
"rewards/margins": 0.3364180028438568,
"rewards/rejected": -1.1059613227844238,
"step": 1620
},
{
"epoch": 0.85,
"learning_rate": 3.202405945663556e-07,
"logits/chosen": -0.46464791893959045,
"logits/rejected": -0.5091901421546936,
"logps/chosen": -438.41461181640625,
"logps/rejected": -474.1409606933594,
"loss": 0.5839,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7322907447814941,
"rewards/margins": 0.397641658782959,
"rewards/rejected": -1.1299324035644531,
"step": 1630
},
{
"epoch": 0.86,
"learning_rate": 2.982328649595856e-07,
"logits/chosen": -0.5395783185958862,
"logits/rejected": -0.5762434601783752,
"logps/chosen": -355.19256591796875,
"logps/rejected": -356.80377197265625,
"loss": 0.5679,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.5796637535095215,
"rewards/margins": 0.501792311668396,
"rewards/rejected": -1.0814560651779175,
"step": 1640
},
{
"epoch": 0.86,
"learning_rate": 2.7696050769026954e-07,
"logits/chosen": -0.4488208293914795,
"logits/rejected": -0.5003235340118408,
"logps/chosen": -406.1018371582031,
"logps/rejected": -456.99688720703125,
"loss": 0.5767,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.835650622844696,
"rewards/margins": 0.4407881796360016,
"rewards/rejected": -1.2764387130737305,
"step": 1650
},
{
"epoch": 0.87,
"learning_rate": 2.564306275529341e-07,
"logits/chosen": -0.4920230507850647,
"logits/rejected": -0.5208636522293091,
"logps/chosen": -429.73516845703125,
"logps/rejected": -437.10992431640625,
"loss": 0.5907,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7671813368797302,
"rewards/margins": 0.3512424826622009,
"rewards/rejected": -1.1184238195419312,
"step": 1660
},
{
"epoch": 0.87,
"learning_rate": 2.3665008136077332e-07,
"logits/chosen": -0.46324095129966736,
"logits/rejected": -0.5051876306533813,
"logps/chosen": -455.948486328125,
"logps/rejected": -432.7015686035156,
"loss": 0.5709,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.749980092048645,
"rewards/margins": 0.4463537633419037,
"rewards/rejected": -1.1963337659835815,
"step": 1670
},
{
"epoch": 0.88,
"learning_rate": 2.1762547565553293e-07,
"logits/chosen": -0.4768814444541931,
"logits/rejected": -0.5184912085533142,
"logps/chosen": -447.73699951171875,
"logps/rejected": -459.3815002441406,
"loss": 0.5918,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.7029529809951782,
"rewards/margins": 0.4211914539337158,
"rewards/rejected": -1.1241443157196045,
"step": 1680
},
{
"epoch": 0.88,
"learning_rate": 1.993631645009747e-07,
"logits/chosen": -0.5452268123626709,
"logits/rejected": -0.5993489027023315,
"logps/chosen": -437.57574462890625,
"logps/rejected": -418.8999938964844,
"loss": 0.5761,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7408413887023926,
"rewards/margins": 0.390348881483078,
"rewards/rejected": -1.1311901807785034,
"step": 1690
},
{
"epoch": 0.89,
"learning_rate": 1.818692473606748e-07,
"logits/chosen": -0.413705438375473,
"logits/rejected": -0.4628356099128723,
"logps/chosen": -434.87677001953125,
"logps/rejected": -442.89697265625,
"loss": 0.5722,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.7070480585098267,
"rewards/margins": 0.4665676951408386,
"rewards/rejected": -1.17361581325531,
"step": 1700
},
{
"epoch": 0.89,
"eval_logits/chosen": -0.4876059889793396,
"eval_logits/rejected": -0.527910053730011,
"eval_logps/chosen": -418.4286193847656,
"eval_logps/rejected": -439.34423828125,
"eval_loss": 0.5797783136367798,
"eval_rewards/accuracies": 0.7142857313156128,
"eval_rewards/chosen": -0.7181041836738586,
"eval_rewards/margins": 0.5335846543312073,
"eval_rewards/rejected": -1.251688838005066,
"eval_runtime": 304.6366,
"eval_samples_per_second": 6.565,
"eval_steps_per_second": 0.207,
"step": 1700
},
{
"epoch": 0.9,
"learning_rate": 1.6514956706084885e-07,
"logits/chosen": -0.4525040090084076,
"logits/rejected": -0.49936771392822266,
"logps/chosen": -421.5797424316406,
"logps/rejected": -447.3121643066406,
"loss": 0.5836,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.7661898732185364,
"rewards/margins": 0.37991976737976074,
"rewards/rejected": -1.1461094617843628,
"step": 1710
},
{
"epoch": 0.9,
"learning_rate": 1.4920970783889737e-07,
"logits/chosen": -0.4586445689201355,
"logits/rejected": -0.4811577796936035,
"logps/chosen": -412.69036865234375,
"logps/rejected": -422.6893615722656,
"loss": 0.5803,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.8322868347167969,
"rewards/margins": 0.4826812148094177,
"rewards/rejected": -1.3149678707122803,
"step": 1720
},
{
"epoch": 0.91,
"learning_rate": 1.340549934783164e-07,
"logits/chosen": -0.4814591407775879,
"logits/rejected": -0.4792747497558594,
"logps/chosen": -403.96356201171875,
"logps/rejected": -428.724365234375,
"loss": 0.5786,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.8141158223152161,
"rewards/margins": 0.33095940947532654,
"rewards/rejected": -1.1450750827789307,
"step": 1730
},
{
"epoch": 0.91,
"learning_rate": 1.196904855305961e-07,
"logits/chosen": -0.48179054260253906,
"logits/rejected": -0.5652682185173035,
"logps/chosen": -372.9039001464844,
"logps/rejected": -408.64044189453125,
"loss": 0.5559,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7646127939224243,
"rewards/margins": 0.5597863793373108,
"rewards/rejected": -1.3243992328643799,
"step": 1740
},
{
"epoch": 0.92,
"learning_rate": 1.0612098162470302e-07,
"logits/chosen": -0.4800513684749603,
"logits/rejected": -0.5272001624107361,
"logps/chosen": -375.82318115234375,
"logps/rejected": -419.78912353515625,
"loss": 0.5415,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.700168251991272,
"rewards/margins": 0.6963863968849182,
"rewards/rejected": -1.396554708480835,
"step": 1750
},
{
"epoch": 0.92,
"learning_rate": 9.335101386471285e-08,
"logits/chosen": -0.4976566731929779,
"logits/rejected": -0.5353574752807617,
"logps/chosen": -427.3526306152344,
"logps/rejected": -440.6385803222656,
"loss": 0.5983,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.7464424967765808,
"rewards/margins": 0.34794315695762634,
"rewards/rejected": -1.0943857431411743,
"step": 1760
},
{
"epoch": 0.93,
"learning_rate": 8.138484731612273e-08,
"logits/chosen": -0.47734421491622925,
"logits/rejected": -0.49538105726242065,
"logps/chosen": -419.697021484375,
"logps/rejected": -415.8753356933594,
"loss": 0.619,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7787846922874451,
"rewards/margins": 0.4026049077510834,
"rewards/rejected": -1.1813896894454956,
"step": 1770
},
{
"epoch": 0.93,
"learning_rate": 7.022647858135501e-08,
"logits/chosen": -0.46987882256507874,
"logits/rejected": -0.548054039478302,
"logps/chosen": -452.89031982421875,
"logps/rejected": -434.79412841796875,
"loss": 0.5841,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.5671200156211853,
"rewards/margins": 0.5638147592544556,
"rewards/rejected": -1.1309348344802856,
"step": 1780
},
{
"epoch": 0.94,
"learning_rate": 5.987963446492384e-08,
"logits/chosen": -0.451104074716568,
"logits/rejected": -0.5252590179443359,
"logps/chosen": -411.8269958496094,
"logps/rejected": -424.35711669921875,
"loss": 0.5847,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.7318955659866333,
"rewards/margins": 0.4990832805633545,
"rewards/rejected": -1.2309788465499878,
"step": 1790
},
{
"epoch": 0.94,
"learning_rate": 5.034777072871394e-08,
"logits/chosen": -0.5124109387397766,
"logits/rejected": -0.5305658578872681,
"logps/chosen": -402.7662048339844,
"logps/rejected": -395.8765563964844,
"loss": 0.5964,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6769779324531555,
"rewards/margins": 0.48363232612609863,
"rewards/rejected": -1.1606101989746094,
"step": 1800
},
{
"epoch": 0.94,
"eval_logits/chosen": -0.4875066578388214,
"eval_logits/rejected": -0.5277794599533081,
"eval_logps/chosen": -418.2664489746094,
"eval_logps/rejected": -439.24761962890625,
"eval_loss": 0.5795618891716003,
"eval_rewards/accuracies": 0.716269850730896,
"eval_rewards/chosen": -0.7164830565452576,
"eval_rewards/margins": 0.5342397689819336,
"eval_rewards/rejected": -1.2507227659225464,
"eval_runtime": 305.9708,
"eval_samples_per_second": 6.537,
"eval_steps_per_second": 0.206,
"step": 1800
},
{
"epoch": 0.95,
"learning_rate": 4.163407093778243e-08,
"logits/chosen": -0.4408170282840729,
"logits/rejected": -0.5464833974838257,
"logps/chosen": -402.9147644042969,
"logps/rejected": -394.70172119140625,
"loss": 0.5482,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6640672087669373,
"rewards/margins": 0.5019302368164062,
"rewards/rejected": -1.1659975051879883,
"step": 1810
},
{
"epoch": 0.95,
"learning_rate": 3.37414453970758e-08,
"logits/chosen": -0.4044371247291565,
"logits/rejected": -0.44484180212020874,
"logps/chosen": -477.49853515625,
"logps/rejected": -480.2476501464844,
"loss": 0.5723,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6610472202301025,
"rewards/margins": 0.5942956209182739,
"rewards/rejected": -1.2553428411483765,
"step": 1820
},
{
"epoch": 0.96,
"learning_rate": 2.6672530179410183e-08,
"logits/chosen": -0.45786118507385254,
"logits/rejected": -0.5099942088127136,
"logps/chosen": -395.73785400390625,
"logps/rejected": -440.14495849609375,
"loss": 0.554,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.6342514753341675,
"rewards/margins": 0.6357657313346863,
"rewards/rejected": -1.2700172662734985,
"step": 1830
},
{
"epoch": 0.96,
"learning_rate": 2.04296862450451e-08,
"logits/chosen": -0.4265124797821045,
"logits/rejected": -0.4525698721408844,
"logps/chosen": -421.3204650878906,
"logps/rejected": -476.8326110839844,
"loss": 0.5625,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.7191395163536072,
"rewards/margins": 0.5786597728729248,
"rewards/rejected": -1.2977991104125977,
"step": 1840
},
{
"epoch": 0.97,
"learning_rate": 1.501499865314171e-08,
"logits/chosen": -0.4681014120578766,
"logits/rejected": -0.530280590057373,
"logps/chosen": -422.77197265625,
"logps/rejected": -424.14056396484375,
"loss": 0.5514,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7412760257720947,
"rewards/margins": 0.6121469736099243,
"rewards/rejected": -1.3534228801727295,
"step": 1850
},
{
"epoch": 0.97,
"learning_rate": 1.0430275865371265e-08,
"logits/chosen": -0.45095524191856384,
"logits/rejected": -0.5133798122406006,
"logps/chosen": -359.3103332519531,
"logps/rejected": -419.0379333496094,
"loss": 0.5218,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.862167477607727,
"rewards/margins": 0.4554516673088074,
"rewards/rejected": -1.3176189661026,
"step": 1860
},
{
"epoch": 0.98,
"learning_rate": 6.677049141901315e-09,
"logits/chosen": -0.4633178114891052,
"logits/rejected": -0.542106032371521,
"logps/chosen": -444.0020446777344,
"logps/rejected": -472.94390869140625,
"loss": 0.5761,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6808910369873047,
"rewards/margins": 0.5797610878944397,
"rewards/rejected": -1.2606520652770996,
"step": 1870
},
{
"epoch": 0.98,
"learning_rate": 3.756572029968708e-09,
"logits/chosen": -0.49038058519363403,
"logits/rejected": -0.5752415060997009,
"logps/chosen": -397.6487121582031,
"logps/rejected": -417.7991638183594,
"loss": 0.5427,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.7362946271896362,
"rewards/margins": 0.7132160067558289,
"rewards/rejected": -1.4495106935501099,
"step": 1880
},
{
"epoch": 0.99,
"learning_rate": 1.6698199452053199e-09,
"logits/chosen": -0.4550807476043701,
"logits/rejected": -0.5087591409683228,
"logps/chosen": -394.6986999511719,
"logps/rejected": -468.04742431640625,
"loss": 0.5651,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7036923170089722,
"rewards/margins": 0.5819169282913208,
"rewards/rejected": -1.285609245300293,
"step": 1890
},
{
"epoch": 0.99,
"learning_rate": 4.1748984585560094e-10,
"logits/chosen": -0.47594064474105835,
"logits/rejected": -0.4794263243675232,
"logps/chosen": -370.91998291015625,
"logps/rejected": -406.54864501953125,
"loss": 0.5896,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6993014216423035,
"rewards/margins": 0.5366551280021667,
"rewards/rejected": -1.2359565496444702,
"step": 1900
},
{
"epoch": 0.99,
"eval_logits/chosen": -0.48749276995658875,
"eval_logits/rejected": -0.5277734398841858,
"eval_logps/chosen": -418.4147033691406,
"eval_logps/rejected": -439.3841857910156,
"eval_loss": 0.5797061920166016,
"eval_rewards/accuracies": 0.716269850730896,
"eval_rewards/chosen": -0.7179651260375977,
"eval_rewards/margins": 0.5341233611106873,
"eval_rewards/rejected": -1.2520886659622192,
"eval_runtime": 302.9913,
"eval_samples_per_second": 6.601,
"eval_steps_per_second": 0.208,
"step": 1900
},
{
"epoch": 1.0,
"learning_rate": 0.0,
"logits/chosen": -0.5706673860549927,
"logits/rejected": -0.613754153251648,
"logps/chosen": -418.0396423339844,
"logps/rejected": -457.19842529296875,
"loss": 0.5615,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.8323566317558289,
"rewards/margins": 0.60826575756073,
"rewards/rejected": -1.440622329711914,
"step": 1910
},
{
"epoch": 1.0,
"step": 1910,
"total_flos": 0.0,
"train_loss": 0.598351346759896,
"train_runtime": 27817.4474,
"train_samples_per_second": 2.198,
"train_steps_per_second": 0.069
}
],
"logging_steps": 10,
"max_steps": 1910,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}