phi-2-irepo-chatml-v6-i1 / trainer_state.json
lole25's picture
Model save
8c151a7 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9996190476190476,
"eval_steps": 500,
"global_step": 1312,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 3.787878787878788e-08,
"logits/chosen": 0.3414086103439331,
"logits/rejected": 0.3618736267089844,
"logps/chosen": -457.8209228515625,
"logps/rejected": -477.39923095703125,
"loss": 0.2223,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.01,
"learning_rate": 3.787878787878788e-07,
"logits/chosen": 0.3330618441104889,
"logits/rejected": 0.3205517530441284,
"logps/chosen": -306.0744323730469,
"logps/rejected": -326.5745849609375,
"loss": 0.2099,
"rewards/accuracies": 0.2777777910232544,
"rewards/chosen": -0.00023350439732894301,
"rewards/margins": 6.652800948359072e-05,
"rewards/rejected": -0.0003000324359163642,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 7.575757575757576e-07,
"logits/chosen": 0.3300573229789734,
"logits/rejected": 0.3766365945339203,
"logps/chosen": -294.01226806640625,
"logps/rejected": -328.8618469238281,
"loss": 0.2087,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 9.059391595656052e-05,
"rewards/margins": 1.657838947721757e-05,
"rewards/rejected": 7.401553739327937e-05,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 1.1363636363636364e-06,
"logits/chosen": 0.33157598972320557,
"logits/rejected": 0.3647070527076721,
"logps/chosen": -308.50250244140625,
"logps/rejected": -316.5055236816406,
"loss": 0.21,
"rewards/accuracies": 0.26249998807907104,
"rewards/chosen": -9.306786523666233e-05,
"rewards/margins": -5.975523890811019e-06,
"rewards/rejected": -8.709232497494668e-05,
"step": 30
},
{
"epoch": 0.03,
"learning_rate": 1.5151515151515152e-06,
"logits/chosen": 0.30125534534454346,
"logits/rejected": 0.3123665452003479,
"logps/chosen": -278.3026123046875,
"logps/rejected": -290.43365478515625,
"loss": 0.221,
"rewards/accuracies": 0.20624999701976776,
"rewards/chosen": -1.4783197912038304e-05,
"rewards/margins": 5.556107498705387e-05,
"rewards/rejected": -7.034426380414516e-05,
"step": 40
},
{
"epoch": 0.04,
"learning_rate": 1.8939393939393941e-06,
"logits/chosen": 0.33402004837989807,
"logits/rejected": 0.3797036111354828,
"logps/chosen": -250.8700408935547,
"logps/rejected": -274.0110778808594,
"loss": 0.2,
"rewards/accuracies": 0.26249998807907104,
"rewards/chosen": 0.0001303141616517678,
"rewards/margins": 2.952014074253384e-05,
"rewards/rejected": 0.00010079403000418097,
"step": 50
},
{
"epoch": 0.05,
"learning_rate": 2.2727272727272728e-06,
"logits/chosen": 0.3142412304878235,
"logits/rejected": 0.3297632336616516,
"logps/chosen": -261.3359069824219,
"logps/rejected": -272.4247131347656,
"loss": 0.216,
"rewards/accuracies": 0.3125,
"rewards/chosen": 7.56668159738183e-05,
"rewards/margins": 0.0001469437702326104,
"rewards/rejected": -7.12769542587921e-05,
"step": 60
},
{
"epoch": 0.05,
"learning_rate": 2.6515151515151514e-06,
"logits/chosen": 0.31074008345603943,
"logits/rejected": 0.3008107542991638,
"logps/chosen": -305.2731628417969,
"logps/rejected": -301.9687805175781,
"loss": 0.21,
"rewards/accuracies": 0.25,
"rewards/chosen": -0.0002583070017863065,
"rewards/margins": -3.105907671852037e-05,
"rewards/rejected": -0.0002272479177918285,
"step": 70
},
{
"epoch": 0.06,
"learning_rate": 3.0303030303030305e-06,
"logits/chosen": 0.310350626707077,
"logits/rejected": 0.3315640389919281,
"logps/chosen": -232.4719696044922,
"logps/rejected": -277.22833251953125,
"loss": 0.1962,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 9.660933574195951e-05,
"rewards/margins": 0.00016913158469833434,
"rewards/rejected": -7.252227078424767e-05,
"step": 80
},
{
"epoch": 0.07,
"learning_rate": 3.409090909090909e-06,
"logits/chosen": 0.28212469816207886,
"logits/rejected": 0.35147595405578613,
"logps/chosen": -291.93121337890625,
"logps/rejected": -309.50341796875,
"loss": 0.2222,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": -0.0006029252544976771,
"rewards/margins": 0.00015305924171116203,
"rewards/rejected": -0.0007559844525530934,
"step": 90
},
{
"epoch": 0.08,
"learning_rate": 3.7878787878787882e-06,
"logits/chosen": 0.2799969017505646,
"logits/rejected": 0.33390945196151733,
"logps/chosen": -313.59515380859375,
"logps/rejected": -333.5437316894531,
"loss": 0.2158,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": -0.0013320285361260176,
"rewards/margins": 0.00041345407953485847,
"rewards/rejected": -0.001745482673868537,
"step": 100
},
{
"epoch": 0.08,
"learning_rate": 4.166666666666667e-06,
"logits/chosen": 0.3202076852321625,
"logits/rejected": 0.31327325105667114,
"logps/chosen": -339.7900085449219,
"logps/rejected": -391.5009765625,
"loss": 0.2034,
"rewards/accuracies": 0.39375001192092896,
"rewards/chosen": -0.0016250547487288713,
"rewards/margins": 0.000587599235586822,
"rewards/rejected": -0.002212654100731015,
"step": 110
},
{
"epoch": 0.09,
"learning_rate": 4.5454545454545455e-06,
"logits/chosen": 0.2636483311653137,
"logits/rejected": 0.3315524756908417,
"logps/chosen": -257.42376708984375,
"logps/rejected": -279.6871643066406,
"loss": 0.2195,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": -0.0009949387749657035,
"rewards/margins": 0.0003876467817462981,
"rewards/rejected": -0.0013825856149196625,
"step": 120
},
{
"epoch": 0.1,
"learning_rate": 4.924242424242425e-06,
"logits/chosen": 0.25745201110839844,
"logits/rejected": 0.3214574456214905,
"logps/chosen": -302.13787841796875,
"logps/rejected": -334.5384826660156,
"loss": 0.1975,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": -0.0025944842491298914,
"rewards/margins": 0.0014145069289952517,
"rewards/rejected": -0.004008991178125143,
"step": 130
},
{
"epoch": 0.11,
"learning_rate": 4.999432965739786e-06,
"logits/chosen": 0.26108819246292114,
"logits/rejected": 0.21493813395500183,
"logps/chosen": -321.00067138671875,
"logps/rejected": -349.4842834472656,
"loss": 0.2053,
"rewards/accuracies": 0.3687500059604645,
"rewards/chosen": -0.003485953900963068,
"rewards/margins": 0.0010465523228049278,
"rewards/rejected": -0.0045325057581067085,
"step": 140
},
{
"epoch": 0.11,
"learning_rate": 4.997129829895409e-06,
"logits/chosen": 0.16764743626117706,
"logits/rejected": 0.20418021082878113,
"logps/chosen": -216.89199829101562,
"logps/rejected": -234.42935180664062,
"loss": 0.208,
"rewards/accuracies": 0.2562499940395355,
"rewards/chosen": -0.0010050686541944742,
"rewards/margins": 0.0011736680753529072,
"rewards/rejected": -0.0021787367295473814,
"step": 150
},
{
"epoch": 0.12,
"learning_rate": 4.9930567839810125e-06,
"logits/chosen": 0.13134139776229858,
"logits/rejected": 0.14908884465694427,
"logps/chosen": -268.61419677734375,
"logps/rejected": -301.322265625,
"loss": 0.2203,
"rewards/accuracies": 0.3375000059604645,
"rewards/chosen": -0.0038432697765529156,
"rewards/margins": 0.005182007793337107,
"rewards/rejected": -0.009025278501212597,
"step": 160
},
{
"epoch": 0.13,
"learning_rate": 4.987216714880929e-06,
"logits/chosen": 0.1437113881111145,
"logits/rejected": 0.13215014338493347,
"logps/chosen": -259.3898010253906,
"logps/rejected": -267.3370056152344,
"loss": 0.2012,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": -0.01510892529040575,
"rewards/margins": 0.0029346179217100143,
"rewards/rejected": -0.01804354228079319,
"step": 170
},
{
"epoch": 0.14,
"learning_rate": 4.979613761906212e-06,
"logits/chosen": 0.07960179448127747,
"logits/rejected": 0.08153261244297028,
"logps/chosen": -357.0888366699219,
"logps/rejected": -368.5904846191406,
"loss": 0.2207,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": -0.021911371499300003,
"rewards/margins": 0.0033388130832463503,
"rewards/rejected": -0.025250181555747986,
"step": 180
},
{
"epoch": 0.14,
"learning_rate": 4.970253313860788e-06,
"logits/chosen": 0.12686966359615326,
"logits/rejected": 0.092551589012146,
"logps/chosen": -323.23162841796875,
"logps/rejected": -354.8710632324219,
"loss": 0.2065,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": -0.027633970603346825,
"rewards/margins": 0.007219144143164158,
"rewards/rejected": -0.03485311195254326,
"step": 190
},
{
"epoch": 0.15,
"learning_rate": 4.959142005221991e-06,
"logits/chosen": 0.030479246750473976,
"logits/rejected": 0.03255733102560043,
"logps/chosen": -324.7137145996094,
"logps/rejected": -360.15802001953125,
"loss": 0.2118,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": -0.036979734897613525,
"rewards/margins": 0.0006652610609307885,
"rewards/rejected": -0.037644993513822556,
"step": 200
},
{
"epoch": 0.16,
"learning_rate": 4.94628771143819e-06,
"logits/chosen": 0.011406493373215199,
"logits/rejected": 0.0717320591211319,
"logps/chosen": -335.43402099609375,
"logps/rejected": -334.8843078613281,
"loss": 0.2127,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": -0.02695571817457676,
"rewards/margins": 0.0005523961153812706,
"rewards/rejected": -0.027508113533258438,
"step": 210
},
{
"epoch": 0.17,
"learning_rate": 4.931699543346854e-06,
"logits/chosen": 0.11805645376443863,
"logits/rejected": 0.13258619606494904,
"logps/chosen": -321.3063049316406,
"logps/rejected": -337.80084228515625,
"loss": 0.221,
"rewards/accuracies": 0.2874999940395355,
"rewards/chosen": -0.009172406047582626,
"rewards/margins": 0.0013926494866609573,
"rewards/rejected": -0.010565054602921009,
"step": 220
},
{
"epoch": 0.18,
"learning_rate": 4.9153878407169815e-06,
"logits/chosen": 0.04168776422739029,
"logits/rejected": 0.052837539464235306,
"logps/chosen": -321.91546630859375,
"logps/rejected": -337.6319274902344,
"loss": 0.2142,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": 0.00017253603436984122,
"rewards/margins": 0.0029792343266308308,
"rewards/rejected": -0.002806698437780142,
"step": 230
},
{
"epoch": 0.18,
"learning_rate": 4.897364164920515e-06,
"logits/chosen": 0.0712689757347107,
"logits/rejected": 0.07041509449481964,
"logps/chosen": -275.91632080078125,
"logps/rejected": -272.06927490234375,
"loss": 0.2087,
"rewards/accuracies": 0.2562499940395355,
"rewards/chosen": 0.0034352331422269344,
"rewards/margins": 0.0012143031926825643,
"rewards/rejected": 0.002220930065959692,
"step": 240
},
{
"epoch": 0.19,
"learning_rate": 4.8776412907378845e-06,
"logits/chosen": 0.0784018337726593,
"logits/rejected": 0.07138744741678238,
"logps/chosen": -270.2529296875,
"logps/rejected": -276.2494812011719,
"loss": 0.2158,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.004147468134760857,
"rewards/margins": 0.0022368025965988636,
"rewards/rejected": 0.0019106656545773149,
"step": 250
},
{
"epoch": 0.2,
"learning_rate": 4.8562331973035396e-06,
"logits/chosen": 0.04922521486878395,
"logits/rejected": 0.03360098600387573,
"logps/chosen": -256.4879455566406,
"logps/rejected": -295.20220947265625,
"loss": 0.2052,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.0018207021057605743,
"rewards/margins": -4.667430039262399e-05,
"rewards/rejected": 0.0018673762679100037,
"step": 260
},
{
"epoch": 0.21,
"learning_rate": 4.833155058197842e-06,
"logits/chosen": 0.003297035349532962,
"logits/rejected": 0.05275546759366989,
"logps/chosen": -241.99462890625,
"logps/rejected": -263.92120361328125,
"loss": 0.2039,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.005943115334957838,
"rewards/margins": 0.004178833216428757,
"rewards/rejected": 0.0017642822349444032,
"step": 270
},
{
"epoch": 0.21,
"learning_rate": 4.808423230692374e-06,
"logits/chosen": 0.05449346452951431,
"logits/rejected": 0.06165250390768051,
"logps/chosen": -253.69369506835938,
"logps/rejected": -293.31585693359375,
"loss": 0.2029,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": 0.012908290140330791,
"rewards/margins": 0.007234405726194382,
"rewards/rejected": 0.005673886742442846,
"step": 280
},
{
"epoch": 0.22,
"learning_rate": 4.7820552441562625e-06,
"logits/chosen": -0.005410021636635065,
"logits/rejected": 0.010305285453796387,
"logps/chosen": -270.04083251953125,
"logps/rejected": -316.2156066894531,
"loss": 0.1928,
"rewards/accuracies": 0.35624998807907104,
"rewards/chosen": 0.010301874950528145,
"rewards/margins": 0.007835518568754196,
"rewards/rejected": 0.002466357545927167,
"step": 290
},
{
"epoch": 0.23,
"learning_rate": 4.754069787631761e-06,
"logits/chosen": -0.05866356939077377,
"logits/rejected": -0.09499181061983109,
"logps/chosen": -276.0648193359375,
"logps/rejected": -317.32037353515625,
"loss": 0.203,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.008139651268720627,
"rewards/margins": 0.008907916024327278,
"rewards/rejected": -0.0007682637078687549,
"step": 300
},
{
"epoch": 0.24,
"learning_rate": 4.724486696587862e-06,
"logits/chosen": -0.08903555572032928,
"logits/rejected": -0.07178256660699844,
"logps/chosen": -291.4513244628906,
"logps/rejected": -304.63726806640625,
"loss": 0.2106,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.001623433781787753,
"rewards/margins": 0.0021732584573328495,
"rewards/rejected": -0.0005498243262991309,
"step": 310
},
{
"epoch": 0.24,
"learning_rate": 4.693326938861367e-06,
"logits/chosen": -0.12407398223876953,
"logits/rejected": -0.06067138910293579,
"logps/chosen": -249.7843475341797,
"logps/rejected": -290.00408935546875,
"loss": 0.2082,
"rewards/accuracies": 0.34375,
"rewards/chosen": 0.010884495452046394,
"rewards/margins": 0.008355050347745419,
"rewards/rejected": 0.002529443707317114,
"step": 320
},
{
"epoch": 0.25,
"learning_rate": 4.660612599795343e-06,
"logits/chosen": -0.11239596456289291,
"logits/rejected": -0.12089481204748154,
"logps/chosen": -290.61737060546875,
"logps/rejected": -312.4447937011719,
"loss": 0.2046,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": 0.007386790122836828,
"rewards/margins": 0.009295441210269928,
"rewards/rejected": -0.0019086506217718124,
"step": 330
},
{
"epoch": 0.26,
"learning_rate": 4.626366866585528e-06,
"logits/chosen": -0.12021325528621674,
"logits/rejected": -0.11920841783285141,
"logps/chosen": -334.55303955078125,
"logps/rejected": -341.6528625488281,
"loss": 0.2095,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.005449886433780193,
"rewards/margins": -0.0001580852986080572,
"rewards/rejected": 0.0056079719215631485,
"step": 340
},
{
"epoch": 0.27,
"learning_rate": 4.590614011845758e-06,
"logits/chosen": -0.0888114720582962,
"logits/rejected": -0.11601515114307404,
"logps/chosen": -275.80316162109375,
"logps/rejected": -300.4934997558594,
"loss": 0.1974,
"rewards/accuracies": 0.3687500059604645,
"rewards/chosen": 0.01670825481414795,
"rewards/margins": 0.00685430783778429,
"rewards/rejected": 0.009853946976363659,
"step": 350
},
{
"epoch": 0.27,
"learning_rate": 4.553379376404085e-06,
"logits/chosen": -0.1058160662651062,
"logits/rejected": -0.11791692674160004,
"logps/chosen": -266.61322021484375,
"logps/rejected": -279.0782470703125,
"loss": 0.1983,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.021091196686029434,
"rewards/margins": 0.004850545898079872,
"rewards/rejected": 0.01624065265059471,
"step": 360
},
{
"epoch": 0.28,
"learning_rate": 4.514689351341751e-06,
"logits/chosen": -0.16150256991386414,
"logits/rejected": -0.1245572566986084,
"logps/chosen": -277.8587341308594,
"logps/rejected": -272.55792236328125,
"loss": 0.208,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.017171379178762436,
"rewards/margins": 8.804257959127426e-05,
"rewards/rejected": 0.017083335667848587,
"step": 370
},
{
"epoch": 0.29,
"learning_rate": 4.474571359287791e-06,
"logits/chosen": -0.1678142547607422,
"logits/rejected": -0.1356990933418274,
"logps/chosen": -280.3147277832031,
"logps/rejected": -318.61224365234375,
"loss": 0.2096,
"rewards/accuracies": 0.375,
"rewards/chosen": 0.024694612249732018,
"rewards/margins": 0.010597179643809795,
"rewards/rejected": 0.014097435399889946,
"step": 380
},
{
"epoch": 0.3,
"learning_rate": 4.4330538349824684e-06,
"logits/chosen": -0.12732991576194763,
"logits/rejected": -0.12050570547580719,
"logps/chosen": -287.99615478515625,
"logps/rejected": -286.3104553222656,
"loss": 0.2079,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.017507802695035934,
"rewards/margins": 1.922808587551117e-05,
"rewards/rejected": 0.017488572746515274,
"step": 390
},
{
"epoch": 0.3,
"learning_rate": 4.3901662051233755e-06,
"logits/chosen": -0.18693214654922485,
"logits/rejected": -0.11668721586465836,
"logps/chosen": -293.45611572265625,
"logps/rejected": -291.56011962890625,
"loss": 0.2156,
"rewards/accuracies": 0.2562499940395355,
"rewards/chosen": 0.011068233288824558,
"rewards/margins": -0.005082954186946154,
"rewards/rejected": 0.01615118607878685,
"step": 400
},
{
"epoch": 0.31,
"learning_rate": 4.345938867508439e-06,
"logits/chosen": -0.06379786878824234,
"logits/rejected": -0.07422082126140594,
"logps/chosen": -299.4942626953125,
"logps/rejected": -322.50201416015625,
"loss": 0.209,
"rewards/accuracies": 0.2874999940395355,
"rewards/chosen": 0.015014107339084148,
"rewards/margins": 0.003927754703909159,
"rewards/rejected": 0.011086350306868553,
"step": 410
},
{
"epoch": 0.32,
"learning_rate": 4.30040316949064e-06,
"logits/chosen": -0.0699755996465683,
"logits/rejected": -0.08232170343399048,
"logps/chosen": -288.906494140625,
"logps/rejected": -310.1832580566406,
"loss": 0.2186,
"rewards/accuracies": 0.3375000059604645,
"rewards/chosen": 0.015213333070278168,
"rewards/margins": 0.0060978480614721775,
"rewards/rejected": 0.009115484543144703,
"step": 420
},
{
"epoch": 0.33,
"learning_rate": 4.253591385759705e-06,
"logits/chosen": -0.10615179687738419,
"logits/rejected": -0.06481219828128815,
"logps/chosen": -278.0174255371094,
"logps/rejected": -334.2236328125,
"loss": 0.2108,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": 0.012311488389968872,
"rewards/margins": 0.00595985259860754,
"rewards/rejected": 0.006351633928716183,
"step": 430
},
{
"epoch": 0.34,
"learning_rate": 4.205536695466524e-06,
"logits/chosen": -0.13743802905082703,
"logits/rejected": -0.133576899766922,
"logps/chosen": -274.42333984375,
"logps/rejected": -305.0398864746094,
"loss": 0.1894,
"rewards/accuracies": 0.28125,
"rewards/chosen": 0.00840366818010807,
"rewards/margins": 0.004010985605418682,
"rewards/rejected": 0.004392682109028101,
"step": 440
},
{
"epoch": 0.34,
"learning_rate": 4.15627315870651e-06,
"logits/chosen": -0.2103540152311325,
"logits/rejected": -0.19342456758022308,
"logps/chosen": -282.6641540527344,
"logps/rejected": -310.85113525390625,
"loss": 0.1974,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": 0.00448573287576437,
"rewards/margins": 0.0029663294553756714,
"rewards/rejected": 0.0015194038860499859,
"step": 450
},
{
"epoch": 0.35,
"learning_rate": 4.105835692378557e-06,
"logits/chosen": -0.1941739320755005,
"logits/rejected": -0.22001402080059052,
"logps/chosen": -267.8673095703125,
"logps/rejected": -265.15618896484375,
"loss": 0.2101,
"rewards/accuracies": 0.2562499940395355,
"rewards/chosen": 0.007565724663436413,
"rewards/margins": 0.0029779626056551933,
"rewards/rejected": 0.004587762989103794,
"step": 460
},
{
"epoch": 0.36,
"learning_rate": 4.05426004543672e-06,
"logits/chosen": -0.2290777713060379,
"logits/rejected": -0.22092656791210175,
"logps/chosen": -267.7095947265625,
"logps/rejected": -307.23468017578125,
"loss": 0.2034,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": 0.013477683067321777,
"rewards/margins": 0.006231679115444422,
"rewards/rejected": 0.007246003951877356,
"step": 470
},
{
"epoch": 0.37,
"learning_rate": 4.001582773552153e-06,
"logits/chosen": -0.2154918909072876,
"logits/rejected": -0.2263382226228714,
"logps/chosen": -236.22079467773438,
"logps/rejected": -281.47308349609375,
"loss": 0.208,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.008721614256501198,
"rewards/margins": 0.004514150787144899,
"rewards/rejected": 0.004207463003695011,
"step": 480
},
{
"epoch": 0.37,
"learning_rate": 3.947841213203262e-06,
"logits/chosen": -0.25622615218162537,
"logits/rejected": -0.28361567854881287,
"logps/chosen": -307.4781494140625,
"logps/rejected": -311.0342712402344,
"loss": 0.2118,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.00395470205694437,
"rewards/margins": 0.005559145472943783,
"rewards/rejected": -0.001604443765245378,
"step": 490
},
{
"epoch": 0.38,
"learning_rate": 3.893073455212438e-06,
"logits/chosen": -0.2454405277967453,
"logits/rejected": -0.23036327958106995,
"logps/chosen": -305.7126159667969,
"logps/rejected": -320.9441833496094,
"loss": 0.1978,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.006379422731697559,
"rewards/margins": 0.0032655359245836735,
"rewards/rejected": 0.003113886807113886,
"step": 500
},
{
"epoch": 0.39,
"learning_rate": 3.837318317748134e-06,
"logits/chosen": -0.23187056183815002,
"logits/rejected": -0.25806108117103577,
"logps/chosen": -295.49847412109375,
"logps/rejected": -312.9036560058594,
"loss": 0.2095,
"rewards/accuracies": 0.35624998807907104,
"rewards/chosen": 0.010056470520794392,
"rewards/margins": 0.005588938947767019,
"rewards/rejected": 0.004467530641704798,
"step": 510
},
{
"epoch": 0.4,
"learning_rate": 3.7806153188114027e-06,
"logits/chosen": -0.28555235266685486,
"logits/rejected": -0.27182719111442566,
"logps/chosen": -278.7008361816406,
"logps/rejected": -286.72845458984375,
"loss": 0.2112,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.0004449138359632343,
"rewards/margins": 0.0027928180061280727,
"rewards/rejected": -0.002347904024645686,
"step": 520
},
{
"epoch": 0.4,
"learning_rate": 3.7230046482264256e-06,
"logits/chosen": -0.29695600271224976,
"logits/rejected": -0.30071455240249634,
"logps/chosen": -303.6511535644531,
"logps/rejected": -311.7902526855469,
"loss": 0.2241,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.0028660064563155174,
"rewards/margins": -0.001624327152967453,
"rewards/rejected": 0.00449033360928297,
"step": 530
},
{
"epoch": 0.41,
"learning_rate": 3.6645271391548542e-06,
"logits/chosen": -0.2697906196117401,
"logits/rejected": -0.2420235425233841,
"logps/chosen": -287.1878662109375,
"logps/rejected": -289.3710632324219,
"loss": 0.1866,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.010242622345685959,
"rewards/margins": 0.0021990591194480658,
"rewards/rejected": 0.008043562062084675,
"step": 540
},
{
"epoch": 0.42,
"learning_rate": 3.6052242391541746e-06,
"logits/chosen": -0.2508285641670227,
"logits/rejected": -0.22416014969348907,
"logps/chosen": -282.50885009765625,
"logps/rejected": -304.0210876464844,
"loss": 0.1951,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.014059506356716156,
"rewards/margins": 0.006549051962792873,
"rewards/rejected": 0.007510454393923283,
"step": 550
},
{
"epoch": 0.43,
"learning_rate": 3.5451379808006014e-06,
"logits/chosen": -0.2925412654876709,
"logits/rejected": -0.27184224128723145,
"logps/chosen": -226.45083618164062,
"logps/rejected": -252.47207641601562,
"loss": 0.2026,
"rewards/accuracies": 0.26249998807907104,
"rewards/chosen": 0.006832784973084927,
"rewards/margins": 0.006407728884369135,
"rewards/rejected": 0.00042505591409280896,
"step": 560
},
{
"epoch": 0.43,
"learning_rate": 3.484310951897323e-06,
"logits/chosen": -0.280683308839798,
"logits/rejected": -0.2360977679491043,
"logps/chosen": -272.83953857421875,
"logps/rejected": -299.12847900390625,
"loss": 0.1933,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.0044427914544939995,
"rewards/margins": 0.007019062992185354,
"rewards/rejected": -0.0025762729346752167,
"step": 570
},
{
"epoch": 0.44,
"learning_rate": 3.4227862652892106e-06,
"logits/chosen": -0.23029498755931854,
"logits/rejected": -0.30575060844421387,
"logps/chosen": -284.2434997558594,
"logps/rejected": -325.15606689453125,
"loss": 0.2113,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.0008473257767036557,
"rewards/margins": 0.0063803717494010925,
"rewards/rejected": -0.007227697875350714,
"step": 580
},
{
"epoch": 0.45,
"learning_rate": 3.3606075283054005e-06,
"logits/chosen": -0.21216976642608643,
"logits/rejected": -0.24321305751800537,
"logps/chosen": -289.8013610839844,
"logps/rejected": -292.233642578125,
"loss": 0.2076,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": -0.006494040135294199,
"rewards/margins": 0.0029291484970599413,
"rewards/rejected": -0.009423188865184784,
"step": 590
},
{
"epoch": 0.46,
"learning_rate": 3.2978188118513814e-06,
"logits/chosen": -0.3345397114753723,
"logits/rejected": -0.2698872983455658,
"logps/chosen": -288.9801025390625,
"logps/rejected": -325.5001525878906,
"loss": 0.2204,
"rewards/accuracies": 0.3375000059604645,
"rewards/chosen": -0.005164691712707281,
"rewards/margins": 0.006596796214580536,
"rewards/rejected": -0.011761486530303955,
"step": 600
},
{
"epoch": 0.46,
"learning_rate": 3.234464619172522e-06,
"logits/chosen": -0.28216275572776794,
"logits/rejected": -0.36517441272735596,
"logps/chosen": -275.3839416503906,
"logps/rejected": -297.3681640625,
"loss": 0.2018,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.007699407637119293,
"rewards/margins": 0.0056534623727202415,
"rewards/rejected": -0.01335287094116211,
"step": 610
},
{
"epoch": 0.47,
"learning_rate": 3.1705898543111576e-06,
"logits/chosen": -0.32491374015808105,
"logits/rejected": -0.31058216094970703,
"logps/chosen": -318.451904296875,
"logps/rejected": -336.5116271972656,
"loss": 0.2052,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": -0.007857877761125565,
"rewards/margins": 0.005408051423728466,
"rewards/rejected": -0.013265928253531456,
"step": 620
},
{
"epoch": 0.48,
"learning_rate": 3.106239790279606e-06,
"logits/chosen": -0.3316212594509125,
"logits/rejected": -0.27467650175094604,
"logps/chosen": -306.818359375,
"logps/rejected": -310.139892578125,
"loss": 0.2198,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.008624967187643051,
"rewards/margins": 0.0013632235350087285,
"rewards/rejected": -0.009988190606236458,
"step": 630
},
{
"epoch": 0.49,
"learning_rate": 3.041460036971664e-06,
"logits/chosen": -0.30510929226875305,
"logits/rejected": -0.2651751637458801,
"logps/chosen": -289.76177978515625,
"logps/rejected": -306.5693664550781,
"loss": 0.2091,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": -0.0034012228716164827,
"rewards/margins": 0.005689640529453754,
"rewards/rejected": -0.009090864099562168,
"step": 640
},
{
"epoch": 0.5,
"learning_rate": 2.976296508835326e-06,
"logits/chosen": -0.27622857689857483,
"logits/rejected": -0.3029255270957947,
"logps/chosen": -254.8801727294922,
"logps/rejected": -280.85552978515625,
"loss": 0.2118,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.003135355655103922,
"rewards/margins": 0.00885056797415018,
"rewards/rejected": -0.005715211853384972,
"step": 650
},
{
"epoch": 0.5,
"learning_rate": 2.910795392329649e-06,
"logits/chosen": -0.29605865478515625,
"logits/rejected": -0.27770885825157166,
"logps/chosen": -277.80804443359375,
"logps/rejected": -309.5155029296875,
"loss": 0.2133,
"rewards/accuracies": 0.3687500059604645,
"rewards/chosen": 0.0010512445587664843,
"rewards/margins": 0.008048586547374725,
"rewards/rejected": -0.006997342221438885,
"step": 660
},
{
"epoch": 0.51,
"learning_rate": 2.8450031131888147e-06,
"logits/chosen": -0.35795870423316956,
"logits/rejected": -0.37449702620506287,
"logps/chosen": -289.4441833496094,
"logps/rejected": -332.01055908203125,
"loss": 0.1977,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": 0.004084504209458828,
"rewards/margins": 0.011728955432772636,
"rewards/rejected": -0.0076444498263299465,
"step": 670
},
{
"epoch": 0.52,
"learning_rate": 2.7789663035166035e-06,
"logits/chosen": -0.35911503434181213,
"logits/rejected": -0.3697941303253174,
"logps/chosen": -251.0248565673828,
"logps/rejected": -283.61279296875,
"loss": 0.2036,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.0005092100473120809,
"rewards/margins": 0.011741384863853455,
"rewards/rejected": -0.011232174932956696,
"step": 680
},
{
"epoch": 0.53,
"learning_rate": 2.7127317687345973e-06,
"logits/chosen": -0.4329107403755188,
"logits/rejected": -0.38218554854393005,
"logps/chosen": -275.99176025390625,
"logps/rejected": -282.9251708984375,
"loss": 0.2255,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": -0.0015629607951268554,
"rewards/margins": 0.005656304769217968,
"rewards/rejected": -0.007219265215098858,
"step": 690
},
{
"epoch": 0.53,
"learning_rate": 2.6463464544075344e-06,
"logits/chosen": -0.38624441623687744,
"logits/rejected": -0.4270317554473877,
"logps/chosen": -313.55047607421875,
"logps/rejected": -360.2120361328125,
"loss": 0.1998,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": -0.0040723588317632675,
"rewards/margins": 0.013303135521709919,
"rewards/rejected": -0.01737549528479576,
"step": 700
},
{
"epoch": 0.54,
"learning_rate": 2.579857412969345e-06,
"logits/chosen": -0.4271603226661682,
"logits/rejected": -0.4234057366847992,
"logps/chosen": -304.10211181640625,
"logps/rejected": -354.1361389160156,
"loss": 0.1979,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": -0.007728138472884893,
"rewards/margins": 0.016483409330248833,
"rewards/rejected": -0.024211544543504715,
"step": 710
},
{
"epoch": 0.55,
"learning_rate": 2.513311770373421e-06,
"logits/chosen": -0.5024863481521606,
"logits/rejected": -0.47420111298561096,
"logps/chosen": -317.60174560546875,
"logps/rejected": -362.4761657714844,
"loss": 0.192,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": -0.011080259457230568,
"rewards/margins": 0.010900650173425674,
"rewards/rejected": -0.021980909630656242,
"step": 720
},
{
"epoch": 0.56,
"learning_rate": 2.446756692690804e-06,
"logits/chosen": -0.4976615905761719,
"logits/rejected": -0.4921053349971771,
"logps/chosen": -306.67547607421875,
"logps/rejected": -361.6275939941406,
"loss": 0.2069,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": -0.006612093187868595,
"rewards/margins": 0.014301796443760395,
"rewards/rejected": -0.02091388963162899,
"step": 730
},
{
"epoch": 0.56,
"learning_rate": 2.380239352679908e-06,
"logits/chosen": -0.45196834206581116,
"logits/rejected": -0.48185840249061584,
"logps/chosen": -313.2336730957031,
"logps/rejected": -313.48260498046875,
"loss": 0.2149,
"rewards/accuracies": 0.2874999940395355,
"rewards/chosen": -0.010375128127634525,
"rewards/margins": 0.004926495254039764,
"rewards/rejected": -0.015301624312996864,
"step": 740
},
{
"epoch": 0.57,
"learning_rate": 2.313806896351529e-06,
"logits/chosen": -0.5004686117172241,
"logits/rejected": -0.47765883803367615,
"logps/chosen": -274.7811279296875,
"logps/rejected": -311.44207763671875,
"loss": 0.1904,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": 0.002466167788952589,
"rewards/margins": 0.012883700430393219,
"rewards/rejected": -0.010417533107101917,
"step": 750
},
{
"epoch": 0.58,
"learning_rate": 2.247506409552795e-06,
"logits/chosen": -0.563784122467041,
"logits/rejected": -0.5083848237991333,
"logps/chosen": -286.28277587890625,
"logps/rejected": -303.66925048828125,
"loss": 0.2037,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.0052384668961167336,
"rewards/margins": 0.004287950228899717,
"rewards/rejected": -0.009526416659355164,
"step": 760
},
{
"epoch": 0.59,
"learning_rate": 2.1813848845937695e-06,
"logits/chosen": -0.5057908296585083,
"logits/rejected": -0.48972076177597046,
"logps/chosen": -297.547607421875,
"logps/rejected": -294.28863525390625,
"loss": 0.2101,
"rewards/accuracies": 0.28125,
"rewards/chosen": -0.0030983686447143555,
"rewards/margins": 0.0027166479267179966,
"rewards/rejected": -0.00581501517444849,
"step": 770
},
{
"epoch": 0.59,
"learning_rate": 2.1154891869403436e-06,
"logits/chosen": -0.472622811794281,
"logits/rejected": -0.4686339497566223,
"logps/chosen": -241.92630004882812,
"logps/rejected": -279.33062744140625,
"loss": 0.2091,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.009422892704606056,
"rewards/margins": 0.009348099119961262,
"rewards/rejected": 7.479395571863279e-05,
"step": 780
},
{
"epoch": 0.6,
"learning_rate": 2.0498660219970395e-06,
"logits/chosen": -0.44067639112472534,
"logits/rejected": -0.46692126989364624,
"logps/chosen": -285.769287109375,
"logps/rejected": -305.04296875,
"loss": 0.2271,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": 0.008178983815014362,
"rewards/margins": 0.007535718381404877,
"rewards/rejected": 0.0006432650843635201,
"step": 790
},
{
"epoch": 0.61,
"learning_rate": 1.9845619020032552e-06,
"logits/chosen": -0.4349413514137268,
"logits/rejected": -0.4484075605869293,
"logps/chosen": -305.90972900390625,
"logps/rejected": -287.2285461425781,
"loss": 0.2105,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.006356480531394482,
"rewards/margins": 0.0014576372923329473,
"rewards/rejected": 0.004898843355476856,
"step": 800
},
{
"epoch": 0.62,
"learning_rate": 1.9196231130664282e-06,
"logits/chosen": -0.45880216360092163,
"logits/rejected": -0.44850587844848633,
"logps/chosen": -263.9127502441406,
"logps/rejected": -292.25177001953125,
"loss": 0.2033,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.0159307774156332,
"rewards/margins": 0.009099806658923626,
"rewards/rejected": 0.006830970756709576,
"step": 810
},
{
"epoch": 0.62,
"learning_rate": 1.8550956823554708e-06,
"logits/chosen": -0.4190438687801361,
"logits/rejected": -0.39324015378952026,
"logps/chosen": -297.383056640625,
"logps/rejected": -307.9434814453125,
"loss": 0.2115,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": 0.008881737478077412,
"rewards/margins": 0.0028643650002777576,
"rewards/rejected": 0.006017371080815792,
"step": 820
},
{
"epoch": 0.63,
"learning_rate": 1.7910253454777346e-06,
"logits/chosen": -0.4339013993740082,
"logits/rejected": -0.4780551791191101,
"logps/chosen": -264.0601501464844,
"logps/rejected": -309.7989807128906,
"loss": 0.2071,
"rewards/accuracies": 0.3687500059604645,
"rewards/chosen": 0.01502480823546648,
"rewards/margins": 0.013155427761375904,
"rewards/rejected": 0.0018693817546591163,
"step": 830
},
{
"epoch": 0.64,
"learning_rate": 1.7274575140626318e-06,
"logits/chosen": -0.413571834564209,
"logits/rejected": -0.4285905361175537,
"logps/chosen": -263.08453369140625,
"logps/rejected": -286.2958068847656,
"loss": 0.2184,
"rewards/accuracies": 0.25,
"rewards/chosen": 0.008341384120285511,
"rewards/margins": 0.0012967393267899752,
"rewards/rejected": 0.007044644560664892,
"step": 840
},
{
"epoch": 0.65,
"learning_rate": 1.6644372435748823e-06,
"logits/chosen": -0.408627986907959,
"logits/rejected": -0.4011750817298889,
"logps/chosen": -229.0798797607422,
"logps/rejected": -262.96075439453125,
"loss": 0.2043,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": 0.0140065373852849,
"rewards/margins": 0.007881510071456432,
"rewards/rejected": 0.006125027313828468,
"step": 850
},
{
"epoch": 0.66,
"learning_rate": 1.6020092013802002e-06,
"logits/chosen": -0.43513408303260803,
"logits/rejected": -0.4182719588279724,
"logps/chosen": -283.6298828125,
"logps/rejected": -299.7771301269531,
"loss": 0.2094,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.015776053071022034,
"rewards/margins": 0.005827675573527813,
"rewards/rejected": 0.00994837749749422,
"step": 860
},
{
"epoch": 0.66,
"learning_rate": 1.5402176350860653e-06,
"logits/chosen": -0.4543350338935852,
"logits/rejected": -0.43857789039611816,
"logps/chosen": -258.48822021484375,
"logps/rejected": -309.0599365234375,
"loss": 0.2144,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": 0.01519864983856678,
"rewards/margins": 0.0170084610581398,
"rewards/rejected": -0.0018098097061738372,
"step": 870
},
{
"epoch": 0.67,
"learning_rate": 1.4791063411799938e-06,
"logits/chosen": -0.4367285370826721,
"logits/rejected": -0.4124454855918884,
"logps/chosen": -258.4836120605469,
"logps/rejected": -256.78948974609375,
"loss": 0.2125,
"rewards/accuracies": 0.26249998807907104,
"rewards/chosen": 0.010697437450289726,
"rewards/margins": 0.0008281940827146173,
"rewards/rejected": 0.009869244880974293,
"step": 880
},
{
"epoch": 0.68,
"learning_rate": 1.4187186339875697e-06,
"logits/chosen": -0.3717727065086365,
"logits/rejected": -0.3890214264392853,
"logps/chosen": -238.35391235351562,
"logps/rejected": -270.5140686035156,
"loss": 0.202,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.015574706718325615,
"rewards/margins": 0.009048102423548698,
"rewards/rejected": 0.006526602897793055,
"step": 890
},
{
"epoch": 0.69,
"learning_rate": 1.3590973149722103e-06,
"logits/chosen": -0.4208298325538635,
"logits/rejected": -0.4319419860839844,
"logps/chosen": -272.0270690917969,
"logps/rejected": -303.1417541503906,
"loss": 0.2083,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.015811121091246605,
"rewards/margins": 0.010845163837075233,
"rewards/rejected": 0.004965959116816521,
"step": 900
},
{
"epoch": 0.69,
"learning_rate": 1.300284642398445e-06,
"logits/chosen": -0.47379741072654724,
"logits/rejected": -0.4520508348941803,
"logps/chosen": -262.06292724609375,
"logps/rejected": -311.0146789550781,
"loss": 0.1942,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": 0.014805925078690052,
"rewards/margins": 0.01618744060397148,
"rewards/rejected": -0.0013815152924507856,
"step": 910
},
{
"epoch": 0.7,
"learning_rate": 1.2423223013801946e-06,
"logits/chosen": -0.45992302894592285,
"logits/rejected": -0.4493522644042969,
"logps/chosen": -277.81231689453125,
"logps/rejected": -303.0189514160156,
"loss": 0.2059,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": 0.007296368479728699,
"rewards/margins": 0.007703060749918222,
"rewards/rejected": -0.00040669291047379375,
"step": 920
},
{
"epoch": 0.71,
"learning_rate": 1.1852513743352886e-06,
"logits/chosen": -0.4434199929237366,
"logits/rejected": -0.4254834055900574,
"logps/chosen": -236.1838836669922,
"logps/rejected": -280.4830627441406,
"loss": 0.2005,
"rewards/accuracies": 0.2874999940395355,
"rewards/chosen": 0.00997086614370346,
"rewards/margins": 0.012648127973079681,
"rewards/rejected": -0.0026772604323923588,
"step": 930
},
{
"epoch": 0.72,
"learning_rate": 1.1291123118671665e-06,
"logits/chosen": -0.4678889811038971,
"logits/rejected": -0.4497453570365906,
"logps/chosen": -260.39630126953125,
"logps/rejected": -310.59783935546875,
"loss": 0.2033,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.013568739406764507,
"rewards/margins": 0.013490801677107811,
"rewards/rejected": 7.793791883159429e-05,
"step": 940
},
{
"epoch": 0.72,
"learning_rate": 1.073944904094385e-06,
"logits/chosen": -0.4590323865413666,
"logits/rejected": -0.4857376217842102,
"logps/chosen": -250.05941772460938,
"logps/rejected": -269.81182861328125,
"loss": 0.2081,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.009169178083539009,
"rewards/margins": 0.0009465640177950263,
"rewards/rejected": 0.008222613483667374,
"step": 950
},
{
"epoch": 0.73,
"learning_rate": 1.019788252448267e-06,
"logits/chosen": -0.42771902680397034,
"logits/rejected": -0.4784491956233978,
"logps/chosen": -286.91033935546875,
"logps/rejected": -356.3207092285156,
"loss": 0.2176,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": 0.009989687241613865,
"rewards/margins": 0.01874762959778309,
"rewards/rejected": -0.008757943287491798,
"step": 960
},
{
"epoch": 0.74,
"learning_rate": 9.66680741958685e-07,
"logits/chosen": -0.5021045804023743,
"logits/rejected": -0.4944073557853699,
"logps/chosen": -256.72845458984375,
"logps/rejected": -282.7173767089844,
"loss": 0.2023,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": 0.008730728179216385,
"rewards/margins": 0.010655781254172325,
"rewards/rejected": -0.001925053307786584,
"step": 970
},
{
"epoch": 0.75,
"learning_rate": 9.146600140475945e-07,
"logits/chosen": -0.49039751291275024,
"logits/rejected": -0.4602015018463135,
"logps/chosen": -319.5450134277344,
"logps/rejected": -356.46478271484375,
"loss": 0.2141,
"rewards/accuracies": 0.3375000059604645,
"rewards/chosen": 0.0025001950562000275,
"rewards/margins": 0.0142419608309865,
"rewards/rejected": -0.011741766706109047,
"step": 980
},
{
"epoch": 0.75,
"learning_rate": 8.637629398496378e-07,
"logits/chosen": -0.4612942636013031,
"logits/rejected": -0.4650956988334656,
"logps/chosen": -314.468017578125,
"logps/rejected": -336.6165466308594,
"loss": 0.2201,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.004640791565179825,
"rewards/margins": 0.011841602623462677,
"rewards/rejected": -0.0072008101269602776,
"step": 990
},
{
"epoch": 0.76,
"learning_rate": 8.140255940787059e-07,
"logits/chosen": -0.4051927924156189,
"logits/rejected": -0.41572269797325134,
"logps/chosen": -266.1190185546875,
"logps/rejected": -307.9905090332031,
"loss": 0.1983,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.005232260562479496,
"rewards/margins": 0.011333976872265339,
"rewards/rejected": -0.006101716309785843,
"step": 1000
},
{
"epoch": 0.77,
"learning_rate": 7.654832294589776e-07,
"logits/chosen": -0.433448851108551,
"logits/rejected": -0.4530878961086273,
"logps/chosen": -264.8626403808594,
"logps/rejected": -305.2262878417969,
"loss": 0.2114,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": 0.004321249667555094,
"rewards/margins": 0.009284446947276592,
"rewards/rejected": -0.004963197745382786,
"step": 1010
},
{
"epoch": 0.78,
"learning_rate": 7.181702517385789e-07,
"logits/chosen": -0.470588743686676,
"logits/rejected": -0.4782032370567322,
"logps/chosen": -274.71527099609375,
"logps/rejected": -307.9129943847656,
"loss": 0.1997,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.007287648506462574,
"rewards/margins": 0.00972357951104641,
"rewards/rejected": -0.0024359312374144793,
"step": 1020
},
{
"epoch": 0.78,
"learning_rate": 6.721201953035511e-07,
"logits/chosen": -0.4322214126586914,
"logits/rejected": -0.458743155002594,
"logps/chosen": -262.1369323730469,
"logps/rejected": -275.8923034667969,
"loss": 0.2108,
"rewards/accuracies": 0.23749999701976776,
"rewards/chosen": 0.0005865416023880243,
"rewards/margins": 0.000941948383115232,
"rewards/rejected": -0.00035540637327358127,
"step": 1030
},
{
"epoch": 0.79,
"learning_rate": 6.273656994094232e-07,
"logits/chosen": -0.4811081886291504,
"logits/rejected": -0.40350890159606934,
"logps/chosen": -272.11419677734375,
"logps/rejected": -307.12371826171875,
"loss": 0.1991,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.005647097248584032,
"rewards/margins": 0.007634321693331003,
"rewards/rejected": -0.0019872249104082584,
"step": 1040
},
{
"epoch": 0.8,
"learning_rate": 5.839384850472359e-07,
"logits/chosen": -0.46519404649734497,
"logits/rejected": -0.40799570083618164,
"logps/chosen": -289.98883056640625,
"logps/rejected": -292.3999938964844,
"loss": 0.2163,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.008130472153425217,
"rewards/margins": 0.002877952065318823,
"rewards/rejected": 0.005252520553767681,
"step": 1050
},
{
"epoch": 0.81,
"learning_rate": 5.418693324604082e-07,
"logits/chosen": -0.5294805765151978,
"logits/rejected": -0.5169610977172852,
"logps/chosen": -272.34686279296875,
"logps/rejected": -307.16412353515625,
"loss": 0.1938,
"rewards/accuracies": 0.35624998807907104,
"rewards/chosen": 0.012028077617287636,
"rewards/margins": 0.013656511902809143,
"rewards/rejected": -0.0016284309094771743,
"step": 1060
},
{
"epoch": 0.82,
"learning_rate": 5.01188059328386e-07,
"logits/chosen": -0.5478223562240601,
"logits/rejected": -0.4814823269844055,
"logps/chosen": -289.8566589355469,
"logps/rejected": -289.57525634765625,
"loss": 0.2065,
"rewards/accuracies": 0.28125,
"rewards/chosen": 0.008872238919138908,
"rewards/margins": 0.006515635643154383,
"rewards/rejected": 0.0023566028103232384,
"step": 1070
},
{
"epoch": 0.82,
"learning_rate": 4.619234996325314e-07,
"logits/chosen": -0.4742591977119446,
"logits/rejected": -0.45757928490638733,
"logps/chosen": -328.6953125,
"logps/rejected": -340.1205139160156,
"loss": 0.2115,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.006806342396885157,
"rewards/margins": 0.0063201361335814,
"rewards/rejected": 0.000486206088680774,
"step": 1080
},
{
"epoch": 0.83,
"learning_rate": 4.241034832192434e-07,
"logits/chosen": -0.5209555625915527,
"logits/rejected": -0.5151633024215698,
"logps/chosen": -276.89581298828125,
"logps/rejected": -273.1256103515625,
"loss": 0.2135,
"rewards/accuracies": 0.24375000596046448,
"rewards/chosen": 0.0027219385374337435,
"rewards/margins": 0.0031283677089959383,
"rewards/rejected": -0.00040642902604304254,
"step": 1090
},
{
"epoch": 0.84,
"learning_rate": 3.877548160747768e-07,
"logits/chosen": -0.4666425585746765,
"logits/rejected": -0.48001235723495483,
"logps/chosen": -280.6470031738281,
"logps/rejected": -315.48443603515625,
"loss": 0.2048,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": 0.002273663878440857,
"rewards/margins": 0.008108812384307384,
"rewards/rejected": -0.0058351485058665276,
"step": 1100
},
{
"epoch": 0.85,
"learning_rate": 3.529032613257574e-07,
"logits/chosen": -0.4815247654914856,
"logits/rejected": -0.45564061403274536,
"logps/chosen": -300.01959228515625,
"logps/rejected": -330.40899658203125,
"loss": 0.2076,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": 0.0031171857845038176,
"rewards/margins": 0.009788629598915577,
"rewards/rejected": -0.006671445909887552,
"step": 1110
},
{
"epoch": 0.85,
"learning_rate": 3.195735209788528e-07,
"logits/chosen": -0.4933408796787262,
"logits/rejected": -0.5491820573806763,
"logps/chosen": -258.11822509765625,
"logps/rejected": -296.2974853515625,
"loss": 0.213,
"rewards/accuracies": 0.3062500059604645,
"rewards/chosen": 0.008377974852919579,
"rewards/margins": 0.012130925431847572,
"rewards/rejected": -0.003752949880436063,
"step": 1120
},
{
"epoch": 0.86,
"learning_rate": 2.8778921841253774e-07,
"logits/chosen": -0.43251729011535645,
"logits/rejected": -0.5054982900619507,
"logps/chosen": -281.94403076171875,
"logps/rejected": -280.0823669433594,
"loss": 0.2204,
"rewards/accuracies": 0.23125000298023224,
"rewards/chosen": 0.0010522855445742607,
"rewards/margins": 0.00011768531840061769,
"rewards/rejected": 0.0009346003644168377,
"step": 1130
},
{
"epoch": 0.87,
"learning_rate": 2.5757288163336806e-07,
"logits/chosen": -0.4674537181854248,
"logits/rejected": -0.43040981888771057,
"logps/chosen": -267.8955078125,
"logps/rejected": -268.64312744140625,
"loss": 0.2134,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": -0.00031215575290843844,
"rewards/margins": -0.00015194570005405694,
"rewards/rejected": -0.00016021067858673632,
"step": 1140
},
{
"epoch": 0.88,
"learning_rate": 2.2894592730863336e-07,
"logits/chosen": -0.4933416247367859,
"logits/rejected": -0.4184231758117676,
"logps/chosen": -274.43487548828125,
"logps/rejected": -287.72259521484375,
"loss": 0.2058,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.012380121275782585,
"rewards/margins": 0.007005924824625254,
"rewards/rejected": 0.005374195985496044,
"step": 1150
},
{
"epoch": 0.88,
"learning_rate": 2.019286455866981e-07,
"logits/chosen": -0.5218170881271362,
"logits/rejected": -0.46864786744117737,
"logps/chosen": -298.1045227050781,
"logps/rejected": -322.55181884765625,
"loss": 0.1925,
"rewards/accuracies": 0.32499998807907104,
"rewards/chosen": -0.0008321186760440469,
"rewards/margins": 0.004535870626568794,
"rewards/rejected": -0.005367989186197519,
"step": 1160
},
{
"epoch": 0.89,
"learning_rate": 1.7654018571579557e-07,
"logits/chosen": -0.48008307814598083,
"logits/rejected": -0.4698426127433777,
"logps/chosen": -307.2906494140625,
"logps/rejected": -356.3848571777344,
"loss": 0.2072,
"rewards/accuracies": 0.3812499940395355,
"rewards/chosen": 0.013308033347129822,
"rewards/margins": 0.019972598180174828,
"rewards/rejected": -0.00666456576436758,
"step": 1170
},
{
"epoch": 0.9,
"learning_rate": 1.5279854247146703e-07,
"logits/chosen": -0.46893006563186646,
"logits/rejected": -0.4585728645324707,
"logps/chosen": -340.1523132324219,
"logps/rejected": -351.4723205566406,
"loss": 0.2135,
"rewards/accuracies": 0.3187499940395355,
"rewards/chosen": 0.0029886148404330015,
"rewards/margins": 0.005356112495064735,
"rewards/rejected": -0.002367498353123665,
"step": 1180
},
{
"epoch": 0.91,
"learning_rate": 1.307205434022671e-07,
"logits/chosen": -0.4998309016227722,
"logits/rejected": -0.49961596727371216,
"logps/chosen": -272.09832763671875,
"logps/rejected": -272.3662109375,
"loss": 0.1976,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.0032546469010412693,
"rewards/margins": 0.003977665212005377,
"rewards/rejected": -0.0007230177288874984,
"step": 1190
},
{
"epoch": 0.91,
"learning_rate": 1.1032183690276754e-07,
"logits/chosen": -0.518333911895752,
"logits/rejected": -0.5002647638320923,
"logps/chosen": -280.9651794433594,
"logps/rejected": -300.6822204589844,
"loss": 0.2098,
"rewards/accuracies": 0.26875001192092896,
"rewards/chosen": 0.002680544275790453,
"rewards/margins": 0.003743491368368268,
"rewards/rejected": -0.001062947092577815,
"step": 1200
},
{
"epoch": 0.92,
"learning_rate": 9.161688112232836e-08,
"logits/chosen": -0.46787554025650024,
"logits/rejected": -0.45076021552085876,
"logps/chosen": -282.44903564453125,
"logps/rejected": -328.4521789550781,
"loss": 0.2076,
"rewards/accuracies": 0.35624998807907104,
"rewards/chosen": 0.012471871450543404,
"rewards/margins": 0.01295357383787632,
"rewards/rejected": -0.00048170099034905434,
"step": 1210
},
{
"epoch": 0.93,
"learning_rate": 7.46189337174788e-08,
"logits/chosen": -0.4574583172798157,
"logits/rejected": -0.48262113332748413,
"logps/chosen": -287.9381408691406,
"logps/rejected": -307.37017822265625,
"loss": 0.2135,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": 0.01407331507652998,
"rewards/margins": 0.009314117953181267,
"rewards/rejected": 0.004759198985993862,
"step": 1220
},
{
"epoch": 0.94,
"learning_rate": 5.934004245518793e-08,
"logits/chosen": -0.5343510508537292,
"logits/rejected": -0.42381030321121216,
"logps/chosen": -301.0657958984375,
"logps/rejected": -314.10260009765625,
"loss": 0.2144,
"rewards/accuracies": 0.33125001192092896,
"rewards/chosen": 0.011122044175863266,
"rewards/margins": 0.00779320765286684,
"rewards/rejected": 0.0033288367558270693,
"step": 1230
},
{
"epoch": 0.94,
"learning_rate": 4.579103667367385e-08,
"logits/chosen": -0.5401466488838196,
"logits/rejected": -0.4905480444431305,
"logps/chosen": -302.44818115234375,
"logps/rejected": -316.00640869140625,
"loss": 0.2009,
"rewards/accuracies": 0.3125,
"rewards/chosen": 0.009371964260935783,
"rewards/margins": 0.005004683043807745,
"rewards/rejected": 0.004367280751466751,
"step": 1240
},
{
"epoch": 0.95,
"learning_rate": 3.398151960681162e-08,
"logits/chosen": -0.4864376485347748,
"logits/rejected": -0.474257230758667,
"logps/chosen": -272.18450927734375,
"logps/rejected": -293.8534851074219,
"loss": 0.212,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.010555078275501728,
"rewards/margins": 0.007184301503002644,
"rewards/rejected": 0.003370775608345866,
"step": 1250
},
{
"epoch": 0.96,
"learning_rate": 2.3919861577572924e-08,
"logits/chosen": -0.48694509267807007,
"logits/rejected": -0.4881502091884613,
"logps/chosen": -285.6047668457031,
"logps/rejected": -344.7032470703125,
"loss": 0.207,
"rewards/accuracies": 0.3375000059604645,
"rewards/chosen": 0.010286719538271427,
"rewards/margins": 0.011380873620510101,
"rewards/rejected": -0.0010941538494080305,
"step": 1260
},
{
"epoch": 0.97,
"learning_rate": 1.5613194065327854e-08,
"logits/chosen": -0.4984816610813141,
"logits/rejected": -0.4886694550514221,
"logps/chosen": -341.03369140625,
"logps/rejected": -376.7138977050781,
"loss": 0.1955,
"rewards/accuracies": 0.38749998807907104,
"rewards/chosen": 0.008053203113377094,
"rewards/margins": 0.012355836108326912,
"rewards/rejected": -0.004302632994949818,
"step": 1270
},
{
"epoch": 0.98,
"learning_rate": 9.067404651211808e-09,
"logits/chosen": -0.47365856170654297,
"logits/rejected": -0.463469922542572,
"logps/chosen": -296.1658630371094,
"logps/rejected": -324.7825012207031,
"loss": 0.2016,
"rewards/accuracies": 0.35624998807907104,
"rewards/chosen": 0.00670830812305212,
"rewards/margins": 0.009644337929785252,
"rewards/rejected": -0.0029360298067331314,
"step": 1280
},
{
"epoch": 0.98,
"learning_rate": 4.287132845137709e-09,
"logits/chosen": -0.4601454734802246,
"logits/rejected": -0.44960451126098633,
"logps/chosen": -305.7701110839844,
"logps/rejected": -349.0350341796875,
"loss": 0.1976,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": 0.007751205004751682,
"rewards/margins": 0.011196482926607132,
"rewards/rejected": -0.0034452793188393116,
"step": 1290
},
{
"epoch": 0.99,
"learning_rate": 1.2757667974155896e-09,
"logits/chosen": -0.4950791001319885,
"logits/rejected": -0.47172340750694275,
"logps/chosen": -312.46453857421875,
"logps/rejected": -315.9293212890625,
"loss": 0.2221,
"rewards/accuracies": 0.2874999940395355,
"rewards/chosen": 0.00022761887521483004,
"rewards/margins": -0.004159080795943737,
"rewards/rejected": 0.0043866997584700584,
"step": 1300
},
{
"epoch": 1.0,
"learning_rate": 3.544089730633804e-11,
"logits/chosen": -0.48454493284225464,
"logits/rejected": -0.46828117966651917,
"logps/chosen": -290.0190734863281,
"logps/rejected": -295.50665283203125,
"loss": 0.1916,
"rewards/accuracies": 0.29374998807907104,
"rewards/chosen": 0.0072511411271989346,
"rewards/margins": 0.00542284082621336,
"rewards/rejected": 0.001828300068154931,
"step": 1310
},
{
"epoch": 1.0,
"step": 1312,
"total_flos": 0.0,
"train_loss": 0.20755663267677513,
"train_runtime": 11765.5627,
"train_samples_per_second": 1.785,
"train_steps_per_second": 0.112
}
],
"logging_steps": 10,
"max_steps": 1312,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}