zephyr-7b-dpo-0k-15k-i1 / trainer_state.json
BraylonDash's picture
Model save
7026af4 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 1250,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4e-08,
"logits/chosen": -2.6577353477478027,
"logits/rejected": -2.043900489807129,
"logps/chosen": -505.98724365234375,
"logps/rejected": -319.40179443359375,
"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": 4.0000000000000003e-07,
"logits/chosen": -2.4769232273101807,
"logits/rejected": -2.134289503097534,
"logps/chosen": -285.32098388671875,
"logps/rejected": -191.55812072753906,
"loss": 0.6932,
"rewards/accuracies": 0.3333333432674408,
"rewards/chosen": 0.00036080353311263025,
"rewards/margins": -9.262199455406517e-05,
"rewards/rejected": 0.0004534254549071193,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 8.000000000000001e-07,
"logits/chosen": -2.3606486320495605,
"logits/rejected": -2.1270670890808105,
"logps/chosen": -271.58087158203125,
"logps/rejected": -208.8097381591797,
"loss": 0.6934,
"rewards/accuracies": 0.2750000059604645,
"rewards/chosen": -0.0005878033698536456,
"rewards/margins": -0.0015078920405358076,
"rewards/rejected": 0.0009200887870974839,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 1.2000000000000002e-06,
"logits/chosen": -2.277930498123169,
"logits/rejected": -2.2482991218566895,
"logps/chosen": -269.87554931640625,
"logps/rejected": -288.9063720703125,
"loss": 0.693,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.002121593803167343,
"rewards/margins": 0.0005655083805322647,
"rewards/rejected": 0.0015560855390504003,
"step": 30
},
{
"epoch": 0.03,
"learning_rate": 1.6000000000000001e-06,
"logits/chosen": -2.508906841278076,
"logits/rejected": -2.341235637664795,
"logps/chosen": -210.5034942626953,
"logps/rejected": -181.61361694335938,
"loss": 0.6926,
"rewards/accuracies": 0.5,
"rewards/chosen": 0.0023279793094843626,
"rewards/margins": 0.0018606262747198343,
"rewards/rejected": 0.0004673528019338846,
"step": 40
},
{
"epoch": 0.04,
"learning_rate": 2.0000000000000003e-06,
"logits/chosen": -2.36310076713562,
"logits/rejected": -2.3391873836517334,
"logps/chosen": -195.63548278808594,
"logps/rejected": -211.3179168701172,
"loss": 0.6924,
"rewards/accuracies": 0.5,
"rewards/chosen": 0.0008935689693316817,
"rewards/margins": 0.0021855938248336315,
"rewards/rejected": -0.0012920246226713061,
"step": 50
},
{
"epoch": 0.05,
"learning_rate": 2.4000000000000003e-06,
"logits/chosen": -2.4950430393218994,
"logits/rejected": -2.303813934326172,
"logps/chosen": -244.68856811523438,
"logps/rejected": -277.8478698730469,
"loss": 0.6922,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.010385877452790737,
"rewards/margins": -0.0004113853501621634,
"rewards/rejected": -0.009974492713809013,
"step": 60
},
{
"epoch": 0.06,
"learning_rate": 2.8000000000000003e-06,
"logits/chosen": -2.2493319511413574,
"logits/rejected": -2.266305446624756,
"logps/chosen": -229.941162109375,
"logps/rejected": -217.0909881591797,
"loss": 0.6901,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.011110167019069195,
"rewards/margins": 0.010483531281352043,
"rewards/rejected": -0.021593699231743813,
"step": 70
},
{
"epoch": 0.06,
"learning_rate": 3.2000000000000003e-06,
"logits/chosen": -2.399794816970825,
"logits/rejected": -2.393554449081421,
"logps/chosen": -266.1021423339844,
"logps/rejected": -263.08526611328125,
"loss": 0.6861,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.014540717005729675,
"rewards/margins": 0.021736864000558853,
"rewards/rejected": -0.03627758100628853,
"step": 80
},
{
"epoch": 0.07,
"learning_rate": 3.6000000000000003e-06,
"logits/chosen": -2.32847261428833,
"logits/rejected": -1.991908073425293,
"logps/chosen": -268.0962219238281,
"logps/rejected": -179.7996826171875,
"loss": 0.6863,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.027069438248872757,
"rewards/margins": 0.021108368411660194,
"rewards/rejected": -0.048177801072597504,
"step": 90
},
{
"epoch": 0.08,
"learning_rate": 4.000000000000001e-06,
"logits/chosen": -2.3088860511779785,
"logits/rejected": -2.312539577484131,
"logps/chosen": -274.9145202636719,
"logps/rejected": -250.87716674804688,
"loss": 0.6898,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.031584225594997406,
"rewards/margins": 0.010575409047305584,
"rewards/rejected": -0.042159631848335266,
"step": 100
},
{
"epoch": 0.09,
"learning_rate": 4.4e-06,
"logits/chosen": -2.2585253715515137,
"logits/rejected": -2.144835948944092,
"logps/chosen": -211.51986694335938,
"logps/rejected": -179.9736785888672,
"loss": 0.6819,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.005594385787844658,
"rewards/margins": 0.03688768297433853,
"rewards/rejected": -0.03129329904913902,
"step": 110
},
{
"epoch": 0.1,
"learning_rate": 4.800000000000001e-06,
"logits/chosen": -2.1611924171447754,
"logits/rejected": -2.171570301055908,
"logps/chosen": -237.06216430664062,
"logps/rejected": -269.0331115722656,
"loss": 0.6703,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.007297643460333347,
"rewards/margins": 0.047815777361392975,
"rewards/rejected": -0.055113423615694046,
"step": 120
},
{
"epoch": 0.1,
"learning_rate": 4.999756310023261e-06,
"logits/chosen": -2.448563814163208,
"logits/rejected": -2.3213229179382324,
"logps/chosen": -235.3451690673828,
"logps/rejected": -192.87477111816406,
"loss": 0.6764,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.011573897674679756,
"rewards/margins": 0.052028268575668335,
"rewards/rejected": -0.06360216438770294,
"step": 130
},
{
"epoch": 0.11,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": -2.3375329971313477,
"logits/rejected": -2.034670352935791,
"logps/chosen": -199.99537658691406,
"logps/rejected": -180.91787719726562,
"loss": 0.6737,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.08778422325849533,
"rewards/margins": 0.04929418861865997,
"rewards/rejected": -0.1370784044265747,
"step": 140
},
{
"epoch": 0.12,
"learning_rate": 4.993910125649561e-06,
"logits/chosen": -2.323729991912842,
"logits/rejected": -2.20963191986084,
"logps/chosen": -229.9593048095703,
"logps/rejected": -198.73062133789062,
"loss": 0.6566,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.018689800053834915,
"rewards/margins": 0.08985014259815216,
"rewards/rejected": -0.10853993892669678,
"step": 150
},
{
"epoch": 0.13,
"learning_rate": 4.988068499954578e-06,
"logits/chosen": -2.215893507003784,
"logits/rejected": -2.2889232635498047,
"logps/chosen": -277.0525207519531,
"logps/rejected": -287.1373596191406,
"loss": 0.6778,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.24399030208587646,
"rewards/margins": 0.03944157436490059,
"rewards/rejected": -0.28343185782432556,
"step": 160
},
{
"epoch": 0.14,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": -2.4302048683166504,
"logits/rejected": -2.032261848449707,
"logps/chosen": -285.89691162109375,
"logps/rejected": -248.88278198242188,
"loss": 0.6419,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.18989089131355286,
"rewards/margins": 0.04870374873280525,
"rewards/rejected": -0.23859462141990662,
"step": 170
},
{
"epoch": 0.14,
"learning_rate": 4.970570953616383e-06,
"logits/chosen": -2.2379422187805176,
"logits/rejected": -2.230316400527954,
"logps/chosen": -235.77371215820312,
"logps/rejected": -240.2162628173828,
"loss": 0.6573,
"rewards/accuracies": 0.42500001192092896,
"rewards/chosen": -0.0921173095703125,
"rewards/margins": 0.044399768114089966,
"rewards/rejected": -0.13651707768440247,
"step": 180
},
{
"epoch": 0.15,
"learning_rate": 4.958928677033465e-06,
"logits/chosen": -2.2377524375915527,
"logits/rejected": -2.1663427352905273,
"logps/chosen": -257.6595764160156,
"logps/rejected": -243.8786163330078,
"loss": 0.6519,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.2135959416627884,
"rewards/margins": 0.13946585357189178,
"rewards/rejected": -0.3530617952346802,
"step": 190
},
{
"epoch": 0.16,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": -2.29764723777771,
"logits/rejected": -2.1510560512542725,
"logps/chosen": -285.13702392578125,
"logps/rejected": -305.1742858886719,
"loss": 0.6381,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3077496588230133,
"rewards/margins": 0.12263032048940659,
"rewards/rejected": -0.4303800165653229,
"step": 200
},
{
"epoch": 0.17,
"learning_rate": 4.9299025014463665e-06,
"logits/chosen": -2.4579226970672607,
"logits/rejected": -2.097339153289795,
"logps/chosen": -314.6923828125,
"logps/rejected": -261.67041015625,
"loss": 0.613,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.4118514657020569,
"rewards/margins": 0.30009493231773376,
"rewards/rejected": -0.7119464874267578,
"step": 210
},
{
"epoch": 0.18,
"learning_rate": 4.912541236180779e-06,
"logits/chosen": -2.31429123878479,
"logits/rejected": -1.8840612173080444,
"logps/chosen": -303.4709777832031,
"logps/rejected": -296.21197509765625,
"loss": 0.6429,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6690406799316406,
"rewards/margins": 0.3159363567829132,
"rewards/rejected": -0.984977126121521,
"step": 220
},
{
"epoch": 0.18,
"learning_rate": 4.893298743830168e-06,
"logits/chosen": -2.283473491668701,
"logits/rejected": -2.319681167602539,
"logps/chosen": -240.64303588867188,
"logps/rejected": -283.9293212890625,
"loss": 0.6464,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.4180280268192291,
"rewards/margins": 0.16594772040843964,
"rewards/rejected": -0.5839756727218628,
"step": 230
},
{
"epoch": 0.19,
"learning_rate": 4.8721900291112415e-06,
"logits/chosen": -2.2171471118927,
"logits/rejected": -2.035769462585449,
"logps/chosen": -257.6659851074219,
"logps/rejected": -258.67364501953125,
"loss": 0.624,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5104702711105347,
"rewards/margins": 0.13771632313728333,
"rewards/rejected": -0.6481865644454956,
"step": 240
},
{
"epoch": 0.2,
"learning_rate": 4.849231551964771e-06,
"logits/chosen": -2.0292515754699707,
"logits/rejected": -1.9135668277740479,
"logps/chosen": -286.9338684082031,
"logps/rejected": -313.9806213378906,
"loss": 0.6291,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.169675588607788,
"rewards/margins": 0.24345961213111877,
"rewards/rejected": -1.4131351709365845,
"step": 250
},
{
"epoch": 0.21,
"learning_rate": 4.824441214720629e-06,
"logits/chosen": -2.203101634979248,
"logits/rejected": -1.946657419204712,
"logps/chosen": -382.1039123535156,
"logps/rejected": -404.14892578125,
"loss": 0.5794,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.0869503021240234,
"rewards/margins": 0.3835533559322357,
"rewards/rejected": -1.470503568649292,
"step": 260
},
{
"epoch": 0.22,
"learning_rate": 4.7978383481380865e-06,
"logits/chosen": -2.212496519088745,
"logits/rejected": -1.996314287185669,
"logps/chosen": -307.20587158203125,
"logps/rejected": -306.6591796875,
"loss": 0.5909,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5885969400405884,
"rewards/margins": 0.3659537732601166,
"rewards/rejected": -0.9545507431030273,
"step": 270
},
{
"epoch": 0.22,
"learning_rate": 4.769443696332272e-06,
"logits/chosen": -1.9687378406524658,
"logits/rejected": -2.04164719581604,
"logps/chosen": -249.0873565673828,
"logps/rejected": -323.9103088378906,
"loss": 0.5854,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8271968960762024,
"rewards/margins": 0.3957016170024872,
"rewards/rejected": -1.2228984832763672,
"step": 280
},
{
"epoch": 0.23,
"learning_rate": 4.7392794005985324e-06,
"logits/chosen": -2.2899293899536133,
"logits/rejected": -2.000715970993042,
"logps/chosen": -374.6996154785156,
"logps/rejected": -359.8489685058594,
"loss": 0.5851,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.9794802665710449,
"rewards/margins": 0.4062279760837555,
"rewards/rejected": -1.385708212852478,
"step": 290
},
{
"epoch": 0.24,
"learning_rate": 4.707368982147318e-06,
"logits/chosen": -2.028409242630005,
"logits/rejected": -1.9832637310028076,
"logps/chosen": -426.76385498046875,
"logps/rejected": -396.494384765625,
"loss": 0.5501,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.3581629991531372,
"rewards/margins": 0.4999775290489197,
"rewards/rejected": -1.8581405878067017,
"step": 300
},
{
"epoch": 0.25,
"learning_rate": 4.673737323763048e-06,
"logits/chosen": -1.773838996887207,
"logits/rejected": -1.8148994445800781,
"logps/chosen": -377.904541015625,
"logps/rejected": -456.285400390625,
"loss": 0.6464,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -1.8526346683502197,
"rewards/margins": 0.3698161244392395,
"rewards/rejected": -2.2224507331848145,
"step": 310
},
{
"epoch": 0.26,
"learning_rate": 4.638410650401267e-06,
"logits/chosen": -2.000642776489258,
"logits/rejected": -1.6485764980316162,
"logps/chosen": -337.97711181640625,
"logps/rejected": -298.59136962890625,
"loss": 0.6445,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.8390266299247742,
"rewards/margins": 0.39421704411506653,
"rewards/rejected": -1.233243703842163,
"step": 320
},
{
"epoch": 0.26,
"learning_rate": 4.601416508739211e-06,
"logits/chosen": -2.2264564037323,
"logits/rejected": -2.249499797821045,
"logps/chosen": -326.7769470214844,
"logps/rejected": -301.43341064453125,
"loss": 0.6091,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.5400243997573853,
"rewards/margins": 0.13523522019386292,
"rewards/rejected": -0.6752597093582153,
"step": 330
},
{
"epoch": 0.27,
"learning_rate": 4.562783745695738e-06,
"logits/chosen": -2.1335179805755615,
"logits/rejected": -1.8753182888031006,
"logps/chosen": -247.4087677001953,
"logps/rejected": -241.4481201171875,
"loss": 0.6052,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5356850624084473,
"rewards/margins": 0.29491013288497925,
"rewards/rejected": -0.8305951952934265,
"step": 340
},
{
"epoch": 0.28,
"learning_rate": 4.522542485937369e-06,
"logits/chosen": -2.284304141998291,
"logits/rejected": -1.9852168560028076,
"logps/chosen": -362.0438232421875,
"logps/rejected": -358.851806640625,
"loss": 0.5885,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.0619395971298218,
"rewards/margins": 0.26577672362327576,
"rewards/rejected": -1.32771635055542,
"step": 350
},
{
"epoch": 0.29,
"learning_rate": 4.4807241083879774e-06,
"logits/chosen": -2.22493052482605,
"logits/rejected": -2.0395853519439697,
"logps/chosen": -384.7464904785156,
"logps/rejected": -394.67999267578125,
"loss": 0.541,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.2721855640411377,
"rewards/margins": 0.4639417231082916,
"rewards/rejected": -1.736127495765686,
"step": 360
},
{
"epoch": 0.3,
"learning_rate": 4.437361221760449e-06,
"logits/chosen": -2.074812412261963,
"logits/rejected": -2.015082597732544,
"logps/chosen": -367.3848876953125,
"logps/rejected": -382.31573486328125,
"loss": 0.644,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.7959085702896118,
"rewards/margins": 0.18800289928913116,
"rewards/rejected": -1.983911156654358,
"step": 370
},
{
"epoch": 0.3,
"learning_rate": 4.3924876391293915e-06,
"logits/chosen": -1.972678542137146,
"logits/rejected": -1.9295990467071533,
"logps/chosen": -358.55126953125,
"logps/rejected": -405.638427734375,
"loss": 0.567,
"rewards/accuracies": 0.5,
"rewards/chosen": -1.6957597732543945,
"rewards/margins": 0.3802381753921509,
"rewards/rejected": -2.075997829437256,
"step": 380
},
{
"epoch": 0.31,
"learning_rate": 4.346138351564711e-06,
"logits/chosen": -2.031592607498169,
"logits/rejected": -1.8613895177841187,
"logps/chosen": -358.3297119140625,
"logps/rejected": -360.80023193359375,
"loss": 0.5674,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.6673065423965454,
"rewards/margins": 0.44297677278518677,
"rewards/rejected": -2.110283374786377,
"step": 390
},
{
"epoch": 0.32,
"learning_rate": 4.2983495008466285e-06,
"logits/chosen": -2.104931116104126,
"logits/rejected": -2.1332077980041504,
"logps/chosen": -411.54168701171875,
"logps/rejected": -463.42236328125,
"loss": 0.666,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.9785181283950806,
"rewards/margins": 0.3848969638347626,
"rewards/rejected": -2.363415241241455,
"step": 400
},
{
"epoch": 0.33,
"learning_rate": 4.249158351283414e-06,
"logits/chosen": -2.211599826812744,
"logits/rejected": -1.9848130941390991,
"logps/chosen": -430.388427734375,
"logps/rejected": -470.9698181152344,
"loss": 0.6333,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.7057955265045166,
"rewards/margins": 0.3454364538192749,
"rewards/rejected": -2.051231861114502,
"step": 410
},
{
"epoch": 0.34,
"learning_rate": 4.198603260653792e-06,
"logits/chosen": -2.3403067588806152,
"logits/rejected": -2.1165671348571777,
"logps/chosen": -418.66558837890625,
"logps/rejected": -407.5811462402344,
"loss": 0.5813,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.5130822658538818,
"rewards/margins": 0.13823339343070984,
"rewards/rejected": -1.651315689086914,
"step": 420
},
{
"epoch": 0.34,
"learning_rate": 4.146723650296701e-06,
"logits/chosen": -2.2038991451263428,
"logits/rejected": -2.2611026763916016,
"logps/chosen": -443.9969177246094,
"logps/rejected": -509.6358337402344,
"loss": 0.6094,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.4418891668319702,
"rewards/margins": 0.5296822190284729,
"rewards/rejected": -1.9715715646743774,
"step": 430
},
{
"epoch": 0.35,
"learning_rate": 4.093559974371725e-06,
"logits/chosen": -2.236635684967041,
"logits/rejected": -1.8922793865203857,
"logps/chosen": -270.41839599609375,
"logps/rejected": -274.03546142578125,
"loss": 0.5395,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.8648042678833008,
"rewards/margins": 0.4120641350746155,
"rewards/rejected": -1.2768683433532715,
"step": 440
},
{
"epoch": 0.36,
"learning_rate": 4.039153688314146e-06,
"logits/chosen": -2.2157764434814453,
"logits/rejected": -2.01257586479187,
"logps/chosen": -373.2973937988281,
"logps/rejected": -401.8234558105469,
"loss": 0.6029,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.1020699739456177,
"rewards/margins": 0.6321985125541687,
"rewards/rejected": -1.7342685461044312,
"step": 450
},
{
"epoch": 0.37,
"learning_rate": 3.983547216509254e-06,
"logits/chosen": -2.215888261795044,
"logits/rejected": -2.1177175045013428,
"logps/chosen": -399.17547607421875,
"logps/rejected": -416.84783935546875,
"loss": 0.5846,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.7063343524932861,
"rewards/margins": 0.4800395965576172,
"rewards/rejected": -2.186373472213745,
"step": 460
},
{
"epoch": 0.38,
"learning_rate": 3.92678391921108e-06,
"logits/chosen": -2.060939073562622,
"logits/rejected": -1.9635168313980103,
"logps/chosen": -346.0058898925781,
"logps/rejected": -369.8831481933594,
"loss": 0.6227,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.8079555034637451,
"rewards/margins": 0.4140779376029968,
"rewards/rejected": -2.2220332622528076,
"step": 470
},
{
"epoch": 0.38,
"learning_rate": 3.868908058731376e-06,
"logits/chosen": -1.974831223487854,
"logits/rejected": -1.8712928295135498,
"logps/chosen": -399.43365478515625,
"logps/rejected": -470.9222717285156,
"loss": 0.5688,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.336700439453125,
"rewards/margins": 0.5076223611831665,
"rewards/rejected": -2.844322681427002,
"step": 480
},
{
"epoch": 0.39,
"learning_rate": 3.8099647649251984e-06,
"logits/chosen": -2.060171127319336,
"logits/rejected": -1.7853256464004517,
"logps/chosen": -443.10223388671875,
"logps/rejected": -441.1519470214844,
"loss": 0.56,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.967047929763794,
"rewards/margins": 0.48686590790748596,
"rewards/rejected": -2.453913927078247,
"step": 490
},
{
"epoch": 0.4,
"learning_rate": 3.7500000000000005e-06,
"logits/chosen": -2.0951271057128906,
"logits/rejected": -2.0230164527893066,
"logps/chosen": -356.6705627441406,
"logps/rejected": -427.401611328125,
"loss": 0.5599,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.6239030361175537,
"rewards/margins": 0.6958804726600647,
"rewards/rejected": -2.3197836875915527,
"step": 500
},
{
"epoch": 0.41,
"learning_rate": 3.689060522675689e-06,
"logits/chosen": -2.1919496059417725,
"logits/rejected": -2.014500141143799,
"logps/chosen": -428.25701904296875,
"logps/rejected": -452.41729736328125,
"loss": 0.557,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.8657432794570923,
"rewards/margins": 0.5891915559768677,
"rewards/rejected": -2.454935073852539,
"step": 510
},
{
"epoch": 0.42,
"learning_rate": 3.627193851723577e-06,
"logits/chosen": -2.0486769676208496,
"logits/rejected": -1.9130027294158936,
"logps/chosen": -471.4190979003906,
"logps/rejected": -506.3304138183594,
"loss": 0.5804,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.426065444946289,
"rewards/margins": 0.6026751399040222,
"rewards/rejected": -3.028740406036377,
"step": 520
},
{
"epoch": 0.42,
"learning_rate": 3.564448228912682e-06,
"logits/chosen": -2.078453540802002,
"logits/rejected": -2.038710117340088,
"logps/chosen": -485.046875,
"logps/rejected": -521.7474365234375,
"loss": 0.5933,
"rewards/accuracies": 0.5,
"rewards/chosen": -2.38494873046875,
"rewards/margins": 0.25780850648880005,
"rewards/rejected": -2.6427571773529053,
"step": 530
},
{
"epoch": 0.43,
"learning_rate": 3.5008725813922383e-06,
"logits/chosen": -2.164614200592041,
"logits/rejected": -1.8823134899139404,
"logps/chosen": -410.06988525390625,
"logps/rejected": -389.2319030761719,
"loss": 0.5627,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.471362590789795,
"rewards/margins": 0.5164188146591187,
"rewards/rejected": -1.9877815246582031,
"step": 540
},
{
"epoch": 0.44,
"learning_rate": 3.436516483539781e-06,
"logits/chosen": -2.1140494346618652,
"logits/rejected": -1.9551842212677002,
"logps/chosen": -301.4757995605469,
"logps/rejected": -336.1775817871094,
"loss": 0.551,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.9967424273490906,
"rewards/margins": 0.5578289031982422,
"rewards/rejected": -1.554571270942688,
"step": 550
},
{
"epoch": 0.45,
"learning_rate": 3.3714301183045382e-06,
"logits/chosen": -2.2336459159851074,
"logits/rejected": -2.313737630844116,
"logps/chosen": -379.36846923828125,
"logps/rejected": -424.39483642578125,
"loss": 0.5247,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8149202466011047,
"rewards/margins": 0.6902198195457458,
"rewards/rejected": -1.5051400661468506,
"step": 560
},
{
"epoch": 0.46,
"learning_rate": 3.3056642380762783e-06,
"logits/chosen": -2.109157085418701,
"logits/rejected": -2.1128740310668945,
"logps/chosen": -370.62579345703125,
"logps/rejected": -388.05670166015625,
"loss": 0.5868,
"rewards/accuracies": 0.5,
"rewards/chosen": -1.1603527069091797,
"rewards/margins": 0.45087918639183044,
"rewards/rejected": -1.611232042312622,
"step": 570
},
{
"epoch": 0.46,
"learning_rate": 3.2392701251101172e-06,
"logits/chosen": -2.079946756362915,
"logits/rejected": -1.9534591436386108,
"logps/chosen": -384.3609313964844,
"logps/rejected": -391.6958312988281,
"loss": 0.5787,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.6679966449737549,
"rewards/margins": 0.5511924028396606,
"rewards/rejected": -2.219188928604126,
"step": 580
},
{
"epoch": 0.47,
"learning_rate": 3.1722995515381644e-06,
"logits/chosen": -1.974525809288025,
"logits/rejected": -1.8337838649749756,
"logps/chosen": -373.17828369140625,
"logps/rejected": -386.6225280761719,
"loss": 0.5649,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.2219445705413818,
"rewards/margins": 0.5704046487808228,
"rewards/rejected": -1.7923492193222046,
"step": 590
},
{
"epoch": 0.48,
"learning_rate": 3.1048047389991693e-06,
"logits/chosen": -2.0642426013946533,
"logits/rejected": -1.8965177536010742,
"logps/chosen": -323.34405517578125,
"logps/rejected": -380.21380615234375,
"loss": 0.5979,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.0704076290130615,
"rewards/margins": 0.48811253905296326,
"rewards/rejected": -1.5585200786590576,
"step": 600
},
{
"epoch": 0.49,
"learning_rate": 3.0368383179176584e-06,
"logits/chosen": -2.033987283706665,
"logits/rejected": -1.9596669673919678,
"logps/chosen": -305.8987731933594,
"logps/rejected": -335.6326904296875,
"loss": 0.5467,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6429412961006165,
"rewards/margins": 0.666652500629425,
"rewards/rejected": -1.309593677520752,
"step": 610
},
{
"epoch": 0.5,
"learning_rate": 2.9684532864643123e-06,
"logits/chosen": -2.0718424320220947,
"logits/rejected": -1.9492712020874023,
"logps/chosen": -354.63372802734375,
"logps/rejected": -367.26422119140625,
"loss": 0.5716,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.022562026977539,
"rewards/margins": 0.4337197244167328,
"rewards/rejected": -1.4562817811965942,
"step": 620
},
{
"epoch": 0.5,
"learning_rate": 2.8997029692295875e-06,
"logits/chosen": -2.2117860317230225,
"logits/rejected": -2.0400938987731934,
"logps/chosen": -381.23028564453125,
"logps/rejected": -429.6178283691406,
"loss": 0.531,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.1089333295822144,
"rewards/margins": 0.7748517990112305,
"rewards/rejected": -1.8837852478027344,
"step": 630
},
{
"epoch": 0.51,
"learning_rate": 2.8306409756428067e-06,
"logits/chosen": -2.042997121810913,
"logits/rejected": -1.6112009286880493,
"logps/chosen": -487.6568908691406,
"logps/rejected": -535.9304809570312,
"loss": 0.516,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.8848283290863037,
"rewards/margins": 1.166489601135254,
"rewards/rejected": -3.0513179302215576,
"step": 640
},
{
"epoch": 0.52,
"learning_rate": 2.761321158169134e-06,
"logits/chosen": -2.1288044452667236,
"logits/rejected": -1.8480589389801025,
"logps/chosen": -591.8642578125,
"logps/rejected": -573.7616577148438,
"loss": 0.5375,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -3.170348644256592,
"rewards/margins": 0.48750972747802734,
"rewards/rejected": -3.6578586101531982,
"step": 650
},
{
"epoch": 0.53,
"learning_rate": 2.6917975703170466e-06,
"logits/chosen": -1.834415078163147,
"logits/rejected": -1.542644739151001,
"logps/chosen": -506.2686462402344,
"logps/rejected": -612.8573608398438,
"loss": 0.5693,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.562810182571411,
"rewards/margins": 0.9861109852790833,
"rewards/rejected": -3.5489211082458496,
"step": 660
},
{
"epoch": 0.54,
"learning_rate": 2.6221244244890336e-06,
"logits/chosen": -1.8896293640136719,
"logits/rejected": -1.7028945684432983,
"logps/chosen": -479.379638671875,
"logps/rejected": -513.0144653320312,
"loss": 0.5561,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.36388897895813,
"rewards/margins": 0.6828749775886536,
"rewards/rejected": -3.0467634201049805,
"step": 670
},
{
"epoch": 0.54,
"learning_rate": 2.5523560497083927e-06,
"logits/chosen": -2.059697151184082,
"logits/rejected": -1.8784849643707275,
"logps/chosen": -337.88494873046875,
"logps/rejected": -363.4068298339844,
"loss": 0.6096,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.4112889766693115,
"rewards/margins": 0.4129563868045807,
"rewards/rejected": -1.8242454528808594,
"step": 680
},
{
"epoch": 0.55,
"learning_rate": 2.482546849255096e-06,
"logits/chosen": -1.9830563068389893,
"logits/rejected": -1.6811624765396118,
"logps/chosen": -361.8495788574219,
"logps/rejected": -382.3643493652344,
"loss": 0.5242,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.473449468612671,
"rewards/margins": 0.5172423720359802,
"rewards/rejected": -1.990692138671875,
"step": 690
},
{
"epoch": 0.56,
"learning_rate": 2.4127512582437486e-06,
"logits/chosen": -2.0440447330474854,
"logits/rejected": -1.9845294952392578,
"logps/chosen": -401.8277893066406,
"logps/rejected": -451.3299255371094,
"loss": 0.5288,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.1858041286468506,
"rewards/margins": 0.7083054184913635,
"rewards/rejected": -1.8941094875335693,
"step": 700
},
{
"epoch": 0.57,
"learning_rate": 2.3430237011767166e-06,
"logits/chosen": -1.833554983139038,
"logits/rejected": -1.6954238414764404,
"logps/chosen": -353.65924072265625,
"logps/rejected": -397.3764343261719,
"loss": 0.531,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.3097889423370361,
"rewards/margins": 0.7774924039840698,
"rewards/rejected": -2.0872814655303955,
"step": 710
},
{
"epoch": 0.58,
"learning_rate": 2.2734185495055503e-06,
"logits/chosen": -1.6967763900756836,
"logits/rejected": -1.4636828899383545,
"logps/chosen": -397.916748046875,
"logps/rejected": -437.44073486328125,
"loss": 0.4871,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.731325387954712,
"rewards/margins": 0.9450262188911438,
"rewards/rejected": -2.676351547241211,
"step": 720
},
{
"epoch": 0.58,
"learning_rate": 2.2039900792337477e-06,
"logits/chosen": -1.6748664379119873,
"logits/rejected": -1.4754512310028076,
"logps/chosen": -475.0547790527344,
"logps/rejected": -515.4742431640625,
"loss": 0.5481,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.1851890087127686,
"rewards/margins": 0.8229333758354187,
"rewards/rejected": -3.008122205734253,
"step": 730
},
{
"epoch": 0.59,
"learning_rate": 2.134792428593971e-06,
"logits/chosen": -1.8436079025268555,
"logits/rejected": -1.525818109512329,
"logps/chosen": -420.80810546875,
"logps/rejected": -455.9149475097656,
"loss": 0.5215,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.747849702835083,
"rewards/margins": 0.8033768534660339,
"rewards/rejected": -2.5512263774871826,
"step": 740
},
{
"epoch": 0.6,
"learning_rate": 2.0658795558326745e-06,
"logits/chosen": -1.76202392578125,
"logits/rejected": -1.825585126876831,
"logps/chosen": -335.0538635253906,
"logps/rejected": -410.6778259277344,
"loss": 0.5755,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.3501343727111816,
"rewards/margins": 0.7440234422683716,
"rewards/rejected": -2.0941579341888428,
"step": 750
},
{
"epoch": 0.61,
"learning_rate": 1.997305197135089e-06,
"logits/chosen": -1.8911097049713135,
"logits/rejected": -1.7050600051879883,
"logps/chosen": -344.654296875,
"logps/rejected": -365.8127136230469,
"loss": 0.5896,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.335740089416504,
"rewards/margins": 0.36764174699783325,
"rewards/rejected": -1.7033817768096924,
"step": 760
},
{
"epoch": 0.62,
"learning_rate": 1.9291228247233607e-06,
"logits/chosen": -2.199026107788086,
"logits/rejected": -1.8742659091949463,
"logps/chosen": -418.6163024902344,
"logps/rejected": -376.9792175292969,
"loss": 0.5747,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.2824668884277344,
"rewards/margins": 0.4385095536708832,
"rewards/rejected": -1.72097647190094,
"step": 770
},
{
"epoch": 0.62,
"learning_rate": 1.8613856051605242e-06,
"logits/chosen": -1.913412094116211,
"logits/rejected": -1.6813814640045166,
"logps/chosen": -404.4813232421875,
"logps/rejected": -463.2132263183594,
"loss": 0.5343,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.447469711303711,
"rewards/margins": 0.8074652552604675,
"rewards/rejected": -2.2549350261688232,
"step": 780
},
{
"epoch": 0.63,
"learning_rate": 1.7941463578928088e-06,
"logits/chosen": -2.0417885780334473,
"logits/rejected": -2.0130181312561035,
"logps/chosen": -379.37066650390625,
"logps/rejected": -394.95867919921875,
"loss": 0.6045,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.6197001934051514,
"rewards/margins": 0.29476988315582275,
"rewards/rejected": -1.9144700765609741,
"step": 790
},
{
"epoch": 0.64,
"learning_rate": 1.7274575140626318e-06,
"logits/chosen": -1.9285320043563843,
"logits/rejected": -2.0787336826324463,
"logps/chosen": -348.0567321777344,
"logps/rejected": -397.8705139160156,
"loss": 0.522,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.0203365087509155,
"rewards/margins": 0.4765065312385559,
"rewards/rejected": -1.4968432188034058,
"step": 800
},
{
"epoch": 0.65,
"learning_rate": 1.661371075624363e-06,
"logits/chosen": -1.9909842014312744,
"logits/rejected": -1.6936451196670532,
"logps/chosen": -265.6180419921875,
"logps/rejected": -281.07952880859375,
"loss": 0.5591,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.7331510782241821,
"rewards/margins": 0.5583804249763489,
"rewards/rejected": -1.2915314435958862,
"step": 810
},
{
"epoch": 0.66,
"learning_rate": 1.5959385747947697e-06,
"logits/chosen": -1.9582141637802124,
"logits/rejected": -1.9302053451538086,
"logps/chosen": -336.01959228515625,
"logps/rejected": -410.764404296875,
"loss": 0.4964,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.9309733510017395,
"rewards/margins": 0.613142728805542,
"rewards/rejected": -1.5441160202026367,
"step": 820
},
{
"epoch": 0.66,
"learning_rate": 1.5312110338697427e-06,
"logits/chosen": -2.052699327468872,
"logits/rejected": -1.9583276510238647,
"logps/chosen": -289.5865478515625,
"logps/rejected": -312.62725830078125,
"loss": 0.5951,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.9973894357681274,
"rewards/margins": 0.37033605575561523,
"rewards/rejected": -1.3677256107330322,
"step": 830
},
{
"epoch": 0.67,
"learning_rate": 1.467238925438646e-06,
"logits/chosen": -2.0705270767211914,
"logits/rejected": -1.8725826740264893,
"logps/chosen": -367.77471923828125,
"logps/rejected": -411.2198181152344,
"loss": 0.5386,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.9991909861564636,
"rewards/margins": 0.571709394454956,
"rewards/rejected": -1.5709004402160645,
"step": 840
},
{
"epoch": 0.68,
"learning_rate": 1.4040721330273063e-06,
"logits/chosen": -1.7584121227264404,
"logits/rejected": -1.4766103029251099,
"logps/chosen": -463.2286071777344,
"logps/rejected": -484.1163635253906,
"loss": 0.5445,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.6703122854232788,
"rewards/margins": 0.6411558389663696,
"rewards/rejected": -2.3114681243896484,
"step": 850
},
{
"epoch": 0.69,
"learning_rate": 1.3417599122003464e-06,
"logits/chosen": -2.1024575233459473,
"logits/rejected": -1.6672672033309937,
"logps/chosen": -391.9578857421875,
"logps/rejected": -422.2921447753906,
"loss": 0.4638,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.2933290004730225,
"rewards/margins": 0.9474877119064331,
"rewards/rejected": -2.240816831588745,
"step": 860
},
{
"epoch": 0.7,
"learning_rate": 1.280350852153168e-06,
"logits/chosen": -1.8131592273712158,
"logits/rejected": -1.9417848587036133,
"logps/chosen": -319.12799072265625,
"logps/rejected": -448.4112854003906,
"loss": 0.5363,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.285538911819458,
"rewards/margins": 0.8607152104377747,
"rewards/rejected": -2.146254062652588,
"step": 870
},
{
"epoch": 0.7,
"learning_rate": 1.2198928378235717e-06,
"logits/chosen": -1.8834339380264282,
"logits/rejected": -1.7587229013442993,
"logps/chosen": -424.78826904296875,
"logps/rejected": -522.1199340820312,
"loss": 0.5754,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.7636245489120483,
"rewards/margins": 0.8004873991012573,
"rewards/rejected": -2.5641119480133057,
"step": 880
},
{
"epoch": 0.71,
"learning_rate": 1.160433012552508e-06,
"logits/chosen": -2.0634958744049072,
"logits/rejected": -2.003753423690796,
"logps/chosen": -335.7600402832031,
"logps/rejected": -375.54852294921875,
"loss": 0.5125,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.9807103872299194,
"rewards/margins": 0.9115110635757446,
"rewards/rejected": -1.892221450805664,
"step": 890
},
{
"epoch": 0.72,
"learning_rate": 1.1020177413231334e-06,
"logits/chosen": -2.016789197921753,
"logits/rejected": -1.8195407390594482,
"logps/chosen": -353.8262939453125,
"logps/rejected": -411.1983947753906,
"loss": 0.5023,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.2073825597763062,
"rewards/margins": 0.8694759607315063,
"rewards/rejected": -2.0768585205078125,
"step": 900
},
{
"epoch": 0.73,
"learning_rate": 1.0446925746067768e-06,
"logits/chosen": -1.8060710430145264,
"logits/rejected": -1.5303637981414795,
"logps/chosen": -329.4360046386719,
"logps/rejected": -366.3887939453125,
"loss": 0.5347,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.0586979389190674,
"rewards/margins": 0.9634588956832886,
"rewards/rejected": -2.0221567153930664,
"step": 910
},
{
"epoch": 0.74,
"learning_rate": 9.88502212844063e-07,
"logits/chosen": -1.9487718343734741,
"logits/rejected": -1.8278251886367798,
"logps/chosen": -424.44256591796875,
"logps/rejected": -527.6069946289062,
"loss": 0.5597,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.5092089176177979,
"rewards/margins": 1.0597232580184937,
"rewards/rejected": -2.568932056427002,
"step": 920
},
{
"epoch": 0.74,
"learning_rate": 9.334904715888496e-07,
"logits/chosen": -1.9447358846664429,
"logits/rejected": -1.672228455543518,
"logps/chosen": -440.62774658203125,
"logps/rejected": -548.8284912109375,
"loss": 0.5634,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.6289888620376587,
"rewards/margins": 1.100258469581604,
"rewards/rejected": -2.7292473316192627,
"step": 930
},
{
"epoch": 0.75,
"learning_rate": 8.797002473421729e-07,
"logits/chosen": -1.8913949728012085,
"logits/rejected": -1.8936278820037842,
"logps/chosen": -384.93609619140625,
"logps/rejected": -455.0386657714844,
"loss": 0.5326,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.3333874940872192,
"rewards/margins": 0.7129821181297302,
"rewards/rejected": -2.046369791030884,
"step": 940
},
{
"epoch": 0.76,
"learning_rate": 8.271734841028553e-07,
"logits/chosen": -1.8880865573883057,
"logits/rejected": -1.6740175485610962,
"logps/chosen": -414.0435485839844,
"logps/rejected": -461.56158447265625,
"loss": 0.4911,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.596765398979187,
"rewards/margins": 0.815388560295105,
"rewards/rejected": -2.412154197692871,
"step": 950
},
{
"epoch": 0.77,
"learning_rate": 7.759511406608255e-07,
"logits/chosen": -2.010457754135132,
"logits/rejected": -1.8959916830062866,
"logps/chosen": -383.6888122558594,
"logps/rejected": -452.77105712890625,
"loss": 0.5708,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.487220048904419,
"rewards/margins": 0.7800397276878357,
"rewards/rejected": -2.2672595977783203,
"step": 960
},
{
"epoch": 0.78,
"learning_rate": 7.260731586586983e-07,
"logits/chosen": -1.7655563354492188,
"logits/rejected": -1.8372739553451538,
"logps/chosen": -259.60394287109375,
"logps/rejected": -331.3665466308594,
"loss": 0.541,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.3245407342910767,
"rewards/margins": 0.7111746072769165,
"rewards/rejected": -2.0357155799865723,
"step": 970
},
{
"epoch": 0.78,
"learning_rate": 6.775784314464717e-07,
"logits/chosen": -2.1715354919433594,
"logits/rejected": -1.837615728378296,
"logps/chosen": -360.0257873535156,
"logps/rejected": -413.18902587890625,
"loss": 0.4932,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.2116737365722656,
"rewards/margins": 1.0079423189163208,
"rewards/rejected": -2.219616413116455,
"step": 980
},
{
"epoch": 0.79,
"learning_rate": 6.305047737536707e-07,
"logits/chosen": -1.8406097888946533,
"logits/rejected": -1.8857877254486084,
"logps/chosen": -375.0958557128906,
"logps/rejected": -486.08197021484375,
"loss": 0.5066,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.7462670803070068,
"rewards/margins": 0.950217604637146,
"rewards/rejected": -2.6964848041534424,
"step": 990
},
{
"epoch": 0.8,
"learning_rate": 5.848888922025553e-07,
"logits/chosen": -1.8275880813598633,
"logits/rejected": -2.047853946685791,
"logps/chosen": -441.58966064453125,
"logps/rejected": -461.3450622558594,
"loss": 0.5374,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.767677664756775,
"rewards/margins": 0.5157124996185303,
"rewards/rejected": -2.283390522003174,
"step": 1000
},
{
"epoch": 0.81,
"learning_rate": 5.407663566854008e-07,
"logits/chosen": -2.1351914405822754,
"logits/rejected": -1.8889188766479492,
"logps/chosen": -394.17120361328125,
"logps/rejected": -436.7420959472656,
"loss": 0.5683,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.3438371419906616,
"rewards/margins": 0.9155893325805664,
"rewards/rejected": -2.2594265937805176,
"step": 1010
},
{
"epoch": 0.82,
"learning_rate": 4.981715726281666e-07,
"logits/chosen": -2.017271041870117,
"logits/rejected": -1.981236219406128,
"logps/chosen": -441.42254638671875,
"logps/rejected": -580.200927734375,
"loss": 0.4928,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.5878021717071533,
"rewards/margins": 1.3070558309555054,
"rewards/rejected": -2.8948581218719482,
"step": 1020
},
{
"epoch": 0.82,
"learning_rate": 4.5713775416217884e-07,
"logits/chosen": -1.6851351261138916,
"logits/rejected": -1.7757434844970703,
"logps/chosen": -348.5665588378906,
"logps/rejected": -457.87542724609375,
"loss": 0.5279,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.5111078023910522,
"rewards/margins": 1.0286327600479126,
"rewards/rejected": -2.5397403240203857,
"step": 1030
},
{
"epoch": 0.83,
"learning_rate": 4.1769689822475147e-07,
"logits/chosen": -1.674856424331665,
"logits/rejected": -1.817155122756958,
"logps/chosen": -318.9717712402344,
"logps/rejected": -434.81475830078125,
"loss": 0.5817,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.6043727397918701,
"rewards/margins": 0.6249055862426758,
"rewards/rejected": -2.229278087615967,
"step": 1040
},
{
"epoch": 0.84,
"learning_rate": 3.798797596089351e-07,
"logits/chosen": -1.953508973121643,
"logits/rejected": -1.7095199823379517,
"logps/chosen": -383.12371826171875,
"logps/rejected": -451.828125,
"loss": 0.5006,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.5027226209640503,
"rewards/margins": 0.9227194786071777,
"rewards/rejected": -2.4254419803619385,
"step": 1050
},
{
"epoch": 0.85,
"learning_rate": 3.4371582698185636e-07,
"logits/chosen": -2.011037826538086,
"logits/rejected": -1.858045220375061,
"logps/chosen": -455.5492248535156,
"logps/rejected": -558.2069091796875,
"loss": 0.4797,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.7510541677474976,
"rewards/margins": 0.8790451884269714,
"rewards/rejected": -2.6300997734069824,
"step": 1060
},
{
"epoch": 0.86,
"learning_rate": 3.092332998903416e-07,
"logits/chosen": -1.9804046154022217,
"logits/rejected": -2.04115891456604,
"logps/chosen": -395.303466796875,
"logps/rejected": -492.44329833984375,
"loss": 0.5034,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.4553591012954712,
"rewards/margins": 0.8879169225692749,
"rewards/rejected": -2.343275785446167,
"step": 1070
},
{
"epoch": 0.86,
"learning_rate": 2.764590667717562e-07,
"logits/chosen": -1.8032455444335938,
"logits/rejected": -1.563175916671753,
"logps/chosen": -345.4132385253906,
"logps/rejected": -335.91949462890625,
"loss": 0.5746,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -1.697852373123169,
"rewards/margins": 0.29076820611953735,
"rewards/rejected": -1.988620400428772,
"step": 1080
},
{
"epoch": 0.87,
"learning_rate": 2.454186839872158e-07,
"logits/chosen": -1.882430076599121,
"logits/rejected": -1.7509571313858032,
"logps/chosen": -354.9794616699219,
"logps/rejected": -450.55560302734375,
"loss": 0.4963,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.5439207553863525,
"rewards/margins": 1.0496200323104858,
"rewards/rejected": -2.5935404300689697,
"step": 1090
},
{
"epoch": 0.88,
"learning_rate": 2.1613635589349756e-07,
"logits/chosen": -1.9673278331756592,
"logits/rejected": -1.735865592956543,
"logps/chosen": -361.7403869628906,
"logps/rejected": -393.9466247558594,
"loss": 0.5859,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.3523650169372559,
"rewards/margins": 0.5490896105766296,
"rewards/rejected": -1.9014545679092407,
"step": 1100
},
{
"epoch": 0.89,
"learning_rate": 1.8863491596921745e-07,
"logits/chosen": -1.9650977849960327,
"logits/rejected": -1.674034833908081,
"logps/chosen": -398.31817626953125,
"logps/rejected": -418.9993591308594,
"loss": 0.4818,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.6353886127471924,
"rewards/margins": 0.8128952980041504,
"rewards/rejected": -2.4482839107513428,
"step": 1110
},
{
"epoch": 0.9,
"learning_rate": 1.629358090099639e-07,
"logits/chosen": -1.8651268482208252,
"logits/rejected": -1.7213342189788818,
"logps/chosen": -395.92608642578125,
"logps/rejected": -427.53082275390625,
"loss": 0.5434,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.8581531047821045,
"rewards/margins": 0.6955922842025757,
"rewards/rejected": -2.5537455081939697,
"step": 1120
},
{
"epoch": 0.9,
"learning_rate": 1.3905907440629752e-07,
"logits/chosen": -1.8218196630477905,
"logits/rejected": -1.603205680847168,
"logps/chosen": -371.84820556640625,
"logps/rejected": -423.8062438964844,
"loss": 0.5957,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.6403701305389404,
"rewards/margins": 0.8258684873580933,
"rewards/rejected": -2.466238498687744,
"step": 1130
},
{
"epoch": 0.91,
"learning_rate": 1.1702333051763271e-07,
"logits/chosen": -1.7774673700332642,
"logits/rejected": -1.5289005041122437,
"logps/chosen": -440.49407958984375,
"logps/rejected": -493.42486572265625,
"loss": 0.5159,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.8031593561172485,
"rewards/margins": 0.9965063333511353,
"rewards/rejected": -2.799665927886963,
"step": 1140
},
{
"epoch": 0.92,
"learning_rate": 9.684576015420277e-08,
"logits/chosen": -1.7342450618743896,
"logits/rejected": -1.5554310083389282,
"logps/chosen": -407.7359313964844,
"logps/rejected": -490.3565979003906,
"loss": 0.4897,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.6332868337631226,
"rewards/margins": 0.9035712480545044,
"rewards/rejected": -2.536858081817627,
"step": 1150
},
{
"epoch": 0.93,
"learning_rate": 7.854209717842231e-08,
"logits/chosen": -1.728171944618225,
"logits/rejected": -1.6524664163589478,
"logps/chosen": -420.43768310546875,
"logps/rejected": -411.3157653808594,
"loss": 0.6007,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.8913564682006836,
"rewards/margins": 0.21912887692451477,
"rewards/rejected": -2.110485553741455,
"step": 1160
},
{
"epoch": 0.94,
"learning_rate": 6.212661423609184e-08,
"logits/chosen": -1.7257133722305298,
"logits/rejected": -1.6627498865127563,
"logps/chosen": -435.08599853515625,
"logps/rejected": -524.0604248046875,
"loss": 0.6031,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.2094435691833496,
"rewards/margins": 0.7494264841079712,
"rewards/rejected": -2.9588701725006104,
"step": 1170
},
{
"epoch": 0.94,
"learning_rate": 4.761211162702117e-08,
"logits/chosen": -1.8422664403915405,
"logits/rejected": -1.7962735891342163,
"logps/chosen": -370.86566162109375,
"logps/rejected": -408.62042236328125,
"loss": 0.5591,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.4711788892745972,
"rewards/margins": 0.5541782379150391,
"rewards/rejected": -2.025357246398926,
"step": 1180
},
{
"epoch": 0.95,
"learning_rate": 3.5009907323737826e-08,
"logits/chosen": -1.8273169994354248,
"logits/rejected": -1.549336314201355,
"logps/chosen": -408.63140869140625,
"logps/rejected": -491.55303955078125,
"loss": 0.5183,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.7581342458724976,
"rewards/margins": 0.9740850329399109,
"rewards/rejected": -2.7322194576263428,
"step": 1190
},
{
"epoch": 0.96,
"learning_rate": 2.4329828146074096e-08,
"logits/chosen": -2.1215853691101074,
"logits/rejected": -1.810217261314392,
"logps/chosen": -454.5079040527344,
"logps/rejected": -459.13134765625,
"loss": 0.5396,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.8914794921875,
"rewards/margins": 0.6684591174125671,
"rewards/rejected": -2.559938430786133,
"step": 1200
},
{
"epoch": 0.97,
"learning_rate": 1.5580202098509078e-08,
"logits/chosen": -1.9334113597869873,
"logits/rejected": -1.882664680480957,
"logps/chosen": -512.0015869140625,
"logps/rejected": -529.5744018554688,
"loss": 0.5535,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.7347371578216553,
"rewards/margins": 0.7627021074295044,
"rewards/rejected": -2.49743914604187,
"step": 1210
},
{
"epoch": 0.98,
"learning_rate": 8.767851876239075e-09,
"logits/chosen": -1.6884361505508423,
"logits/rejected": -1.633782148361206,
"logps/chosen": -397.6700134277344,
"logps/rejected": -514.9879150390625,
"loss": 0.5168,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.720780611038208,
"rewards/margins": 1.0960266590118408,
"rewards/rejected": -2.816807270050049,
"step": 1220
},
{
"epoch": 0.98,
"learning_rate": 3.8980895450474455e-09,
"logits/chosen": -1.887984037399292,
"logits/rejected": -1.7204885482788086,
"logps/chosen": -388.229248046875,
"logps/rejected": -438.8857421875,
"loss": 0.5719,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.6957340240478516,
"rewards/margins": 0.7062219381332397,
"rewards/rejected": -2.4019558429718018,
"step": 1230
},
{
"epoch": 0.99,
"learning_rate": 9.747123991141193e-10,
"logits/chosen": -1.995326042175293,
"logits/rejected": -1.8855457305908203,
"logps/chosen": -367.4112243652344,
"logps/rejected": -460.2139587402344,
"loss": 0.5104,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.427626132965088,
"rewards/margins": 1.0081008672714233,
"rewards/rejected": -2.4357268810272217,
"step": 1240
},
{
"epoch": 1.0,
"learning_rate": 0.0,
"logits/chosen": -1.856300950050354,
"logits/rejected": -1.691680908203125,
"logps/chosen": -407.53253173828125,
"logps/rejected": -473.7496643066406,
"loss": 0.4906,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.637367606163025,
"rewards/margins": 0.8503096699714661,
"rewards/rejected": -2.4876773357391357,
"step": 1250
},
{
"epoch": 1.0,
"step": 1250,
"total_flos": 0.0,
"train_loss": 0.37276097068786623,
"train_runtime": 8917.3793,
"train_samples_per_second": 1.682,
"train_steps_per_second": 0.14
}
],
"logging_steps": 10,
"max_steps": 1250,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 20,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}