dpo-selective-longerrun / trainer_state.json
wxzhang's picture
Model save
59c7c4d verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.5702695629416383,
"eval_steps": 100,
"global_step": 1500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 3.3333333333333334e-09,
"logits/chosen": -2.606384754180908,
"logits/rejected": -2.4982504844665527,
"logps/chosen": -296.16644287109375,
"logps/rejected": -225.2230224609375,
"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": 3.3333333333333334e-08,
"logits/chosen": -2.517352819442749,
"logits/rejected": -2.4637084007263184,
"logps/chosen": -342.76483154296875,
"logps/rejected": -293.0290222167969,
"loss": 0.6947,
"rewards/accuracies": 0.4305555522441864,
"rewards/chosen": 0.0013976963236927986,
"rewards/margins": -0.0040576523169875145,
"rewards/rejected": 0.005455348175019026,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 6.666666666666667e-08,
"logits/chosen": -2.396883726119995,
"logits/rejected": -2.3743655681610107,
"logps/chosen": -238.6938018798828,
"logps/rejected": -248.8633575439453,
"loss": 0.6943,
"rewards/accuracies": 0.48124998807907104,
"rewards/chosen": -0.0019892356358468533,
"rewards/margins": -0.0029293105471879244,
"rewards/rejected": 0.0009400752605870366,
"step": 20
},
{
"epoch": 0.03,
"learning_rate": 1e-07,
"logits/chosen": -2.490886688232422,
"logits/rejected": -2.408034563064575,
"logps/chosen": -289.0502014160156,
"logps/rejected": -257.93157958984375,
"loss": 0.6915,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": 8.620424341643229e-05,
"rewards/margins": 0.0004866129602305591,
"rewards/rejected": -0.000400408694986254,
"step": 30
},
{
"epoch": 0.04,
"learning_rate": 1.3333333333333334e-07,
"logits/chosen": -2.479222536087036,
"logits/rejected": -2.4444398880004883,
"logps/chosen": -298.08795166015625,
"logps/rejected": -280.84173583984375,
"loss": 0.6898,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.010273179039359093,
"rewards/margins": 0.005869547836482525,
"rewards/rejected": 0.004403631202876568,
"step": 40
},
{
"epoch": 0.05,
"learning_rate": 1.6666666666666665e-07,
"logits/chosen": -2.505727767944336,
"logits/rejected": -2.4711194038391113,
"logps/chosen": -268.0919494628906,
"logps/rejected": -254.45358276367188,
"loss": 0.6838,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.024473872035741806,
"rewards/margins": 0.018633713945746422,
"rewards/rejected": 0.0058401599526405334,
"step": 50
},
{
"epoch": 0.06,
"learning_rate": 2e-07,
"logits/chosen": -2.456902265548706,
"logits/rejected": -2.411407947540283,
"logps/chosen": -293.38934326171875,
"logps/rejected": -269.47882080078125,
"loss": 0.677,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": 0.04791393131017685,
"rewards/margins": 0.03436511009931564,
"rewards/rejected": 0.01354882400482893,
"step": 60
},
{
"epoch": 0.07,
"learning_rate": 2.3333333333333333e-07,
"logits/chosen": -2.463378429412842,
"logits/rejected": -2.385895013809204,
"logps/chosen": -290.9529724121094,
"logps/rejected": -276.04107666015625,
"loss": 0.6614,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": 0.09689084440469742,
"rewards/margins": 0.058380376547575,
"rewards/rejected": 0.03851046413183212,
"step": 70
},
{
"epoch": 0.08,
"learning_rate": 2.6666666666666667e-07,
"logits/chosen": -2.4076247215270996,
"logits/rejected": -2.3244595527648926,
"logps/chosen": -321.92596435546875,
"logps/rejected": -284.5071105957031,
"loss": 0.6387,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": 0.15204539895057678,
"rewards/margins": 0.11996610462665558,
"rewards/rejected": 0.032079294323921204,
"step": 80
},
{
"epoch": 0.09,
"learning_rate": 3e-07,
"logits/chosen": -2.3717925548553467,
"logits/rejected": -2.362457752227783,
"logps/chosen": -314.3221435546875,
"logps/rejected": -272.3472595214844,
"loss": 0.6254,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": 0.159285306930542,
"rewards/margins": 0.18115119636058807,
"rewards/rejected": -0.02186589315533638,
"step": 90
},
{
"epoch": 0.1,
"learning_rate": 3.333333333333333e-07,
"logits/chosen": -2.3045754432678223,
"logits/rejected": -2.326310634613037,
"logps/chosen": -290.2933349609375,
"logps/rejected": -319.2317199707031,
"loss": 0.6163,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": 0.04698774218559265,
"rewards/margins": 0.2375941276550293,
"rewards/rejected": -0.19060640037059784,
"step": 100
},
{
"epoch": 0.1,
"eval_logits/chosen": -2.3541910648345947,
"eval_logits/rejected": -2.306934118270874,
"eval_logps/chosen": -296.0254211425781,
"eval_logps/rejected": -276.9588623046875,
"eval_loss": 0.6145145297050476,
"eval_rewards/accuracies": 0.7023809552192688,
"eval_rewards/chosen": 0.01473198737949133,
"eval_rewards/margins": 0.2757962942123413,
"eval_rewards/rejected": -0.2610643208026886,
"eval_runtime": 413.5508,
"eval_samples_per_second": 4.836,
"eval_steps_per_second": 0.152,
"step": 100
},
{
"epoch": 0.12,
"learning_rate": 3.666666666666666e-07,
"logits/chosen": -2.4004225730895996,
"logits/rejected": -2.336871385574341,
"logps/chosen": -265.05352783203125,
"logps/rejected": -254.93212890625,
"loss": 0.6078,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.014697420410811901,
"rewards/margins": 0.27927759289741516,
"rewards/rejected": -0.29397499561309814,
"step": 110
},
{
"epoch": 0.13,
"learning_rate": 4e-07,
"logits/chosen": -2.400813579559326,
"logits/rejected": -2.317667245864868,
"logps/chosen": -289.80938720703125,
"logps/rejected": -268.9970703125,
"loss": 0.6067,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": 0.11111988872289658,
"rewards/margins": 0.3709365725517273,
"rewards/rejected": -0.2598166763782501,
"step": 120
},
{
"epoch": 0.14,
"learning_rate": 4.3333333333333335e-07,
"logits/chosen": -2.4633655548095703,
"logits/rejected": -2.370208263397217,
"logps/chosen": -321.96954345703125,
"logps/rejected": -289.915771484375,
"loss": 0.5958,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.06524975597858429,
"rewards/margins": 0.3465055525302887,
"rewards/rejected": -0.2812557816505432,
"step": 130
},
{
"epoch": 0.15,
"learning_rate": 4.6666666666666666e-07,
"logits/chosen": -2.4572839736938477,
"logits/rejected": -2.402984857559204,
"logps/chosen": -267.6603088378906,
"logps/rejected": -240.7797088623047,
"loss": 0.5789,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.05029069632291794,
"rewards/margins": 0.3604304790496826,
"rewards/rejected": -0.3101397752761841,
"step": 140
},
{
"epoch": 0.16,
"learning_rate": 5e-07,
"logits/chosen": -2.3665835857391357,
"logits/rejected": -2.3545050621032715,
"logps/chosen": -304.1943054199219,
"logps/rejected": -298.508056640625,
"loss": 0.6084,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.032247237861156464,
"rewards/margins": 0.392182320356369,
"rewards/rejected": -0.35993510484695435,
"step": 150
},
{
"epoch": 0.17,
"learning_rate": 4.999323102948654e-07,
"logits/chosen": -2.453312397003174,
"logits/rejected": -2.3983371257781982,
"logps/chosen": -269.2843322753906,
"logps/rejected": -277.93682861328125,
"loss": 0.5419,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.10002978146076202,
"rewards/margins": 0.553473174571991,
"rewards/rejected": -0.4534434676170349,
"step": 160
},
{
"epoch": 0.18,
"learning_rate": 4.997292778346312e-07,
"logits/chosen": -2.493710517883301,
"logits/rejected": -2.4315335750579834,
"logps/chosen": -300.1785888671875,
"logps/rejected": -257.55157470703125,
"loss": 0.562,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.0674915537238121,
"rewards/margins": 0.48717761039733887,
"rewards/rejected": -0.5546691417694092,
"step": 170
},
{
"epoch": 0.19,
"learning_rate": 4.99391012564956e-07,
"logits/chosen": -2.3834595680236816,
"logits/rejected": -2.3311541080474854,
"logps/chosen": -271.11114501953125,
"logps/rejected": -289.841552734375,
"loss": 0.5335,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.27489790320396423,
"rewards/margins": 0.6237329244613647,
"rewards/rejected": -0.8986309170722961,
"step": 180
},
{
"epoch": 0.2,
"learning_rate": 4.989176976624511e-07,
"logits/chosen": -2.546043634414673,
"logits/rejected": -2.4504528045654297,
"logps/chosen": -307.6429138183594,
"logps/rejected": -274.8311462402344,
"loss": 0.5298,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": 0.14119036495685577,
"rewards/margins": 0.8371874094009399,
"rewards/rejected": -0.6959971189498901,
"step": 190
},
{
"epoch": 0.21,
"learning_rate": 4.983095894354857e-07,
"logits/chosen": -2.489591121673584,
"logits/rejected": -2.4581377506256104,
"logps/chosen": -268.5490417480469,
"logps/rejected": -291.10986328125,
"loss": 0.5608,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.1413675844669342,
"rewards/margins": 0.6023878455162048,
"rewards/rejected": -0.7437554597854614,
"step": 200
},
{
"epoch": 0.21,
"eval_logits/chosen": -2.4285523891448975,
"eval_logits/rejected": -2.3805854320526123,
"eval_logps/chosen": -298.1169128417969,
"eval_logps/rejected": -287.88702392578125,
"eval_loss": 0.5506500005722046,
"eval_rewards/accuracies": 0.7400793433189392,
"eval_rewards/chosen": -0.08984197676181793,
"eval_rewards/margins": 0.7176319360733032,
"eval_rewards/rejected": -0.80747389793396,
"eval_runtime": 409.3127,
"eval_samples_per_second": 4.886,
"eval_steps_per_second": 0.154,
"step": 200
},
{
"epoch": 0.22,
"learning_rate": 4.975670171853925e-07,
"logits/chosen": -2.48987078666687,
"logits/rejected": -2.4578893184661865,
"logps/chosen": -309.9327087402344,
"logps/rejected": -268.804443359375,
"loss": 0.5555,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.00010260194540023804,
"rewards/margins": 0.6477264165878296,
"rewards/rejected": -0.6476237773895264,
"step": 210
},
{
"epoch": 0.23,
"learning_rate": 4.966903830281448e-07,
"logits/chosen": -2.5101141929626465,
"logits/rejected": -2.5068182945251465,
"logps/chosen": -302.55804443359375,
"logps/rejected": -306.7073059082031,
"loss": 0.5371,
"rewards/accuracies": 0.71875,
"rewards/chosen": 0.022780220955610275,
"rewards/margins": 0.6337054371833801,
"rewards/rejected": -0.6109251379966736,
"step": 220
},
{
"epoch": 0.24,
"learning_rate": 4.956801616766034e-07,
"logits/chosen": -2.477936267852783,
"logits/rejected": -2.4627578258514404,
"logps/chosen": -334.6808776855469,
"logps/rejected": -300.4921569824219,
"loss": 0.5438,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.660649299621582,
"rewards/margins": 0.6745242476463318,
"rewards/rejected": -1.3351736068725586,
"step": 230
},
{
"epoch": 0.25,
"learning_rate": 4.945369001834514e-07,
"logits/chosen": -2.4449329376220703,
"logits/rejected": -2.3937153816223145,
"logps/chosen": -280.98468017578125,
"logps/rejected": -243.7540283203125,
"loss": 0.5362,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.5287132263183594,
"rewards/margins": 0.6548066139221191,
"rewards/rejected": -1.1835198402404785,
"step": 240
},
{
"epoch": 0.26,
"learning_rate": 4.932612176449559e-07,
"logits/chosen": -2.3831703662872314,
"logits/rejected": -2.3532094955444336,
"logps/chosen": -308.6506042480469,
"logps/rejected": -295.2610778808594,
"loss": 0.5517,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.20035383105278015,
"rewards/margins": 0.6892730593681335,
"rewards/rejected": -0.8896268606185913,
"step": 250
},
{
"epoch": 0.27,
"learning_rate": 4.918538048657159e-07,
"logits/chosen": -2.381070852279663,
"logits/rejected": -2.356685161590576,
"logps/chosen": -260.6903076171875,
"logps/rejected": -273.7911682128906,
"loss": 0.5333,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.3518093228340149,
"rewards/margins": 0.6164388060569763,
"rewards/rejected": -0.9682480692863464,
"step": 260
},
{
"epoch": 0.28,
"learning_rate": 4.903154239845797e-07,
"logits/chosen": -2.385469436645508,
"logits/rejected": -2.3523507118225098,
"logps/chosen": -312.2256774902344,
"logps/rejected": -312.83526611328125,
"loss": 0.5092,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.35005810856819153,
"rewards/margins": 0.9718685150146484,
"rewards/rejected": -1.3219265937805176,
"step": 270
},
{
"epoch": 0.29,
"learning_rate": 4.88646908061933e-07,
"logits/chosen": -2.4627230167388916,
"logits/rejected": -2.418569564819336,
"logps/chosen": -317.95703125,
"logps/rejected": -289.7148742675781,
"loss": 0.5104,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.29435399174690247,
"rewards/margins": 0.9088577032089233,
"rewards/rejected": -1.2032116651535034,
"step": 280
},
{
"epoch": 0.3,
"learning_rate": 4.868491606285823e-07,
"logits/chosen": -2.4506328105926514,
"logits/rejected": -2.3806400299072266,
"logps/chosen": -302.1065673828125,
"logps/rejected": -251.01498413085938,
"loss": 0.5353,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.43174147605895996,
"rewards/margins": 0.6914370656013489,
"rewards/rejected": -1.123178482055664,
"step": 290
},
{
"epoch": 0.31,
"learning_rate": 4.849231551964771e-07,
"logits/chosen": -2.431826591491699,
"logits/rejected": -2.354094982147217,
"logps/chosen": -268.3138122558594,
"logps/rejected": -259.5442199707031,
"loss": 0.4934,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.18001610040664673,
"rewards/margins": 0.8297437429428101,
"rewards/rejected": -1.009759783744812,
"step": 300
},
{
"epoch": 0.31,
"eval_logits/chosen": -2.3849596977233887,
"eval_logits/rejected": -2.3416402339935303,
"eval_logps/chosen": -299.6116638183594,
"eval_logps/rejected": -292.5220947265625,
"eval_loss": 0.5225438475608826,
"eval_rewards/accuracies": 0.7579365372657776,
"eval_rewards/chosen": -0.16457918286323547,
"eval_rewards/margins": 0.8746477365493774,
"eval_rewards/rejected": -1.03922700881958,
"eval_runtime": 407.614,
"eval_samples_per_second": 4.907,
"eval_steps_per_second": 0.155,
"step": 300
},
{
"epoch": 0.32,
"learning_rate": 4.828699347315356e-07,
"logits/chosen": -2.3300023078918457,
"logits/rejected": -2.291306972503662,
"logps/chosen": -281.8946228027344,
"logps/rejected": -328.680908203125,
"loss": 0.4757,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -0.23267221450805664,
"rewards/margins": 1.1307260990142822,
"rewards/rejected": -1.3633983135223389,
"step": 310
},
{
"epoch": 0.33,
"learning_rate": 4.806906110888606e-07,
"logits/chosen": -2.2734484672546387,
"logits/rejected": -2.2220654487609863,
"logps/chosen": -344.95184326171875,
"logps/rejected": -317.2240295410156,
"loss": 0.5251,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.40918412804603577,
"rewards/margins": 1.0021181106567383,
"rewards/rejected": -1.4113022089004517,
"step": 320
},
{
"epoch": 0.35,
"learning_rate": 4.783863644106502e-07,
"logits/chosen": -2.2840628623962402,
"logits/rejected": -2.2298922538757324,
"logps/chosen": -281.0782775878906,
"logps/rejected": -242.35263061523438,
"loss": 0.553,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.017911773175001144,
"rewards/margins": 0.6968544721603394,
"rewards/rejected": -0.714766263961792,
"step": 330
},
{
"epoch": 0.36,
"learning_rate": 4.759584424871301e-07,
"logits/chosen": -2.270648717880249,
"logits/rejected": -2.2538352012634277,
"logps/chosen": -297.785888671875,
"logps/rejected": -256.94305419921875,
"loss": 0.5741,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.040852271020412445,
"rewards/margins": 0.6195092797279358,
"rewards/rejected": -0.5786570310592651,
"step": 340
},
{
"epoch": 0.37,
"learning_rate": 4.7340816008085305e-07,
"logits/chosen": -2.3321385383605957,
"logits/rejected": -2.301793336868286,
"logps/chosen": -321.775390625,
"logps/rejected": -314.6920471191406,
"loss": 0.5205,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -0.3098350465297699,
"rewards/margins": 0.9677619934082031,
"rewards/rejected": -1.2775970697402954,
"step": 350
},
{
"epoch": 0.38,
"learning_rate": 4.707368982147317e-07,
"logits/chosen": -2.460111379623413,
"logits/rejected": -2.429689884185791,
"logps/chosen": -289.2377014160156,
"logps/rejected": -296.45123291015625,
"loss": 0.5145,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.8857641220092773,
"rewards/margins": 0.7435467839241028,
"rewards/rejected": -1.6293108463287354,
"step": 360
},
{
"epoch": 0.39,
"learning_rate": 4.6794610342419056e-07,
"logits/chosen": -2.403916120529175,
"logits/rejected": -2.338923931121826,
"logps/chosen": -287.01434326171875,
"logps/rejected": -307.77728271484375,
"loss": 0.4818,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.28949111700057983,
"rewards/margins": 1.0141589641571045,
"rewards/rejected": -1.303650140762329,
"step": 370
},
{
"epoch": 0.4,
"learning_rate": 4.650372869738414e-07,
"logits/chosen": -2.4299778938293457,
"logits/rejected": -2.404153347015381,
"logps/chosen": -310.12152099609375,
"logps/rejected": -294.5881652832031,
"loss": 0.5046,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.4136735796928406,
"rewards/margins": 0.9840124249458313,
"rewards/rejected": -1.3976860046386719,
"step": 380
},
{
"epoch": 0.41,
"learning_rate": 4.6201202403910643e-07,
"logits/chosen": -2.3376293182373047,
"logits/rejected": -2.289973020553589,
"logps/chosen": -299.35546875,
"logps/rejected": -284.88323974609375,
"loss": 0.5184,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.6497005224227905,
"rewards/margins": 1.0391151905059814,
"rewards/rejected": -1.688815712928772,
"step": 390
},
{
"epoch": 0.42,
"learning_rate": 4.588719528532341e-07,
"logits/chosen": -2.458130359649658,
"logits/rejected": -2.4228129386901855,
"logps/chosen": -315.5902404785156,
"logps/rejected": -292.9128112792969,
"loss": 0.4812,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5732877254486084,
"rewards/margins": 0.7860868573188782,
"rewards/rejected": -1.359374761581421,
"step": 400
},
{
"epoch": 0.42,
"eval_logits/chosen": -2.369720458984375,
"eval_logits/rejected": -2.3284802436828613,
"eval_logps/chosen": -300.5794982910156,
"eval_logps/rejected": -295.33331298828125,
"eval_loss": 0.5147603154182434,
"eval_rewards/accuracies": 0.7599206566810608,
"eval_rewards/chosen": -0.21297024190425873,
"eval_rewards/margins": 0.9668172597885132,
"eval_rewards/rejected": -1.179787516593933,
"eval_runtime": 407.8456,
"eval_samples_per_second": 4.904,
"eval_steps_per_second": 0.154,
"step": 400
},
{
"epoch": 0.43,
"learning_rate": 4.5561877382016553e-07,
"logits/chosen": -2.4733567237854004,
"logits/rejected": -2.429727792739868,
"logps/chosen": -300.5076599121094,
"logps/rejected": -275.7524108886719,
"loss": 0.5144,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.15985186398029327,
"rewards/margins": 0.7756703495979309,
"rewards/rejected": -0.935522198677063,
"step": 410
},
{
"epoch": 0.44,
"learning_rate": 4.5225424859373684e-07,
"logits/chosen": -2.3798115253448486,
"logits/rejected": -2.3424534797668457,
"logps/chosen": -243.74472045898438,
"logps/rejected": -266.5513916015625,
"loss": 0.5302,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.005200001411139965,
"rewards/margins": 0.7427350878715515,
"rewards/rejected": -0.7479350566864014,
"step": 420
},
{
"epoch": 0.45,
"learning_rate": 4.487801991237119e-07,
"logits/chosen": -2.441993236541748,
"logits/rejected": -2.414069890975952,
"logps/chosen": -303.46893310546875,
"logps/rejected": -286.26470947265625,
"loss": 0.5387,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.12445823103189468,
"rewards/margins": 0.7546092867851257,
"rewards/rejected": -0.8790675401687622,
"step": 430
},
{
"epoch": 0.46,
"learning_rate": 4.451985066691648e-07,
"logits/chosen": -2.444979429244995,
"logits/rejected": -2.402029514312744,
"logps/chosen": -300.88037109375,
"logps/rejected": -275.1875915527344,
"loss": 0.4965,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.5577610731124878,
"rewards/margins": 0.7658098340034485,
"rewards/rejected": -1.323570966720581,
"step": 440
},
{
"epoch": 0.47,
"learning_rate": 4.415111107797445e-07,
"logits/chosen": -2.290574550628662,
"logits/rejected": -2.2670164108276367,
"logps/chosen": -236.2515106201172,
"logps/rejected": -255.0359344482422,
"loss": 0.5183,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.5639761090278625,
"rewards/margins": 0.8503490686416626,
"rewards/rejected": -1.4143251180648804,
"step": 450
},
{
"epoch": 0.48,
"learning_rate": 4.377200082453748e-07,
"logits/chosen": -2.467921495437622,
"logits/rejected": -2.422136068344116,
"logps/chosen": -329.4405822753906,
"logps/rejected": -305.3108825683594,
"loss": 0.4962,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.41245952248573303,
"rewards/margins": 0.9212630987167358,
"rewards/rejected": -1.333722710609436,
"step": 460
},
{
"epoch": 0.49,
"learning_rate": 4.3382725201495717e-07,
"logits/chosen": -2.3804707527160645,
"logits/rejected": -2.3205363750457764,
"logps/chosen": -282.99945068359375,
"logps/rejected": -292.9880065917969,
"loss": 0.4927,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.5624107718467712,
"rewards/margins": 0.8573344349861145,
"rewards/rejected": -1.4197450876235962,
"step": 470
},
{
"epoch": 0.5,
"learning_rate": 4.2983495008466273e-07,
"logits/chosen": -2.302927255630493,
"logits/rejected": -2.272263765335083,
"logps/chosen": -271.0894470214844,
"logps/rejected": -286.90118408203125,
"loss": 0.5345,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5512509346008301,
"rewards/margins": 0.8141476511955261,
"rewards/rejected": -1.365398645401001,
"step": 480
},
{
"epoch": 0.51,
"learning_rate": 4.2574526435641546e-07,
"logits/chosen": -2.311328172683716,
"logits/rejected": -2.276010274887085,
"logps/chosen": -285.27960205078125,
"logps/rejected": -270.05401611328125,
"loss": 0.5116,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.49702826142311096,
"rewards/margins": 0.8841336369514465,
"rewards/rejected": -1.3811619281768799,
"step": 490
},
{
"epoch": 0.52,
"learning_rate": 4.2156040946718343e-07,
"logits/chosen": -2.3711423873901367,
"logits/rejected": -2.286292552947998,
"logps/chosen": -304.9607238769531,
"logps/rejected": -297.2706604003906,
"loss": 0.5217,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.5594313740730286,
"rewards/margins": 1.1461693048477173,
"rewards/rejected": -1.7056007385253906,
"step": 500
},
{
"epoch": 0.52,
"eval_logits/chosen": -2.3431873321533203,
"eval_logits/rejected": -2.3073697090148926,
"eval_logps/chosen": -299.8136291503906,
"eval_logps/rejected": -294.8787841796875,
"eval_loss": 0.5093801617622375,
"eval_rewards/accuracies": 0.7599206566810608,
"eval_rewards/chosen": -0.17467793822288513,
"eval_rewards/margins": 0.982383131980896,
"eval_rewards/rejected": -1.157060980796814,
"eval_runtime": 409.4153,
"eval_samples_per_second": 4.885,
"eval_steps_per_second": 0.154,
"step": 500
},
{
"epoch": 0.53,
"learning_rate": 4.172826515897145e-07,
"logits/chosen": -2.380160331726074,
"logits/rejected": -2.3342180252075195,
"logps/chosen": -285.99005126953125,
"logps/rejected": -262.095703125,
"loss": 0.4819,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -0.1991649866104126,
"rewards/margins": 0.9838080406188965,
"rewards/rejected": -1.1829731464385986,
"step": 510
},
{
"epoch": 0.54,
"learning_rate": 4.129143072053638e-07,
"logits/chosen": -2.430504322052002,
"logits/rejected": -2.3967652320861816,
"logps/chosen": -295.1175842285156,
"logps/rejected": -298.79034423828125,
"loss": 0.4879,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.2341586798429489,
"rewards/margins": 0.9701293110847473,
"rewards/rejected": -1.2042880058288574,
"step": 520
},
{
"epoch": 0.55,
"learning_rate": 4.084577418496775e-07,
"logits/chosen": -2.459068775177002,
"logits/rejected": -2.4153661727905273,
"logps/chosen": -303.06683349609375,
"logps/rejected": -333.80609130859375,
"loss": 0.4868,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.4716320037841797,
"rewards/margins": 1.0053110122680664,
"rewards/rejected": -1.4769428968429565,
"step": 530
},
{
"epoch": 0.57,
"learning_rate": 4.039153688314145e-07,
"logits/chosen": -2.4637537002563477,
"logits/rejected": -2.4269044399261475,
"logps/chosen": -304.27166748046875,
"logps/rejected": -302.20477294921875,
"loss": 0.4986,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.45887041091918945,
"rewards/margins": 1.0204750299453735,
"rewards/rejected": -1.4793453216552734,
"step": 540
},
{
"epoch": 0.58,
"learning_rate": 3.9928964792569654e-07,
"logits/chosen": -2.4167208671569824,
"logits/rejected": -2.3573527336120605,
"logps/chosen": -303.3789978027344,
"logps/rejected": -296.81060791015625,
"loss": 0.5343,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.59644615650177,
"rewards/margins": 0.9994708895683289,
"rewards/rejected": -1.595916986465454,
"step": 550
},
{
"epoch": 0.59,
"learning_rate": 3.945830840419966e-07,
"logits/chosen": -2.3017094135284424,
"logits/rejected": -2.2486953735351562,
"logps/chosen": -298.501708984375,
"logps/rejected": -250.04556274414062,
"loss": 0.5037,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.4263156056404114,
"rewards/margins": 0.9166458249092102,
"rewards/rejected": -1.3429615497589111,
"step": 560
},
{
"epoch": 0.6,
"learning_rate": 3.8979822586768666e-07,
"logits/chosen": -2.308485507965088,
"logits/rejected": -2.259333610534668,
"logps/chosen": -274.44329833984375,
"logps/rejected": -297.6800537109375,
"loss": 0.5011,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.19708169996738434,
"rewards/margins": 0.9468144178390503,
"rewards/rejected": -1.1438961029052734,
"step": 570
},
{
"epoch": 0.61,
"learning_rate": 3.849376644878782e-07,
"logits/chosen": -2.3406989574432373,
"logits/rejected": -2.2872297763824463,
"logps/chosen": -293.8192443847656,
"logps/rejected": -292.8212585449219,
"loss": 0.5012,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -0.24491167068481445,
"rewards/margins": 1.1491771936416626,
"rewards/rejected": -1.394088864326477,
"step": 580
},
{
"epoch": 0.62,
"learning_rate": 3.800040319823038e-07,
"logits/chosen": -2.2375845909118652,
"logits/rejected": -2.21081280708313,
"logps/chosen": -281.64678955078125,
"logps/rejected": -281.1907958984375,
"loss": 0.5189,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.33811599016189575,
"rewards/margins": 0.640630841255188,
"rewards/rejected": -0.9787468910217285,
"step": 590
},
{
"epoch": 0.63,
"learning_rate": 3.75e-07,
"logits/chosen": -2.336167812347412,
"logits/rejected": -2.3122599124908447,
"logps/chosen": -316.1159362792969,
"logps/rejected": -348.52728271484375,
"loss": 0.5069,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": 0.08971457183361053,
"rewards/margins": 1.1014090776443481,
"rewards/rejected": -1.0116945505142212,
"step": 600
},
{
"epoch": 0.63,
"eval_logits/chosen": -2.2853615283966064,
"eval_logits/rejected": -2.2443532943725586,
"eval_logps/chosen": -297.127197265625,
"eval_logps/rejected": -292.7251281738281,
"eval_loss": 0.5037240982055664,
"eval_rewards/accuracies": 0.7658730149269104,
"eval_rewards/chosen": -0.04035760462284088,
"eval_rewards/margins": 1.009021520614624,
"eval_rewards/rejected": -1.0493791103363037,
"eval_runtime": 411.3039,
"eval_samples_per_second": 4.863,
"eval_steps_per_second": 0.153,
"step": 600
},
{
"epoch": 0.64,
"learning_rate": 3.699282783125616e-07,
"logits/chosen": -2.29797625541687,
"logits/rejected": -2.2567286491394043,
"logps/chosen": -333.64312744140625,
"logps/rejected": -276.2323913574219,
"loss": 0.4939,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.5196839570999146,
"rewards/margins": 0.9884433746337891,
"rewards/rejected": -1.508127212524414,
"step": 610
},
{
"epoch": 0.65,
"learning_rate": 3.647916133467529e-07,
"logits/chosen": -2.3571763038635254,
"logits/rejected": -2.2932493686676025,
"logps/chosen": -308.9863586425781,
"logps/rejected": -301.33367919921875,
"loss": 0.4919,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.0287619829177856,
"rewards/margins": 0.9334309697151184,
"rewards/rejected": -1.9621927738189697,
"step": 620
},
{
"epoch": 0.66,
"learning_rate": 3.595927866972693e-07,
"logits/chosen": -2.352038860321045,
"logits/rejected": -2.2862563133239746,
"logps/chosen": -293.9407958984375,
"logps/rejected": -288.50885009765625,
"loss": 0.4733,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.5753384828567505,
"rewards/margins": 1.169237494468689,
"rewards/rejected": -1.744576096534729,
"step": 630
},
{
"epoch": 0.67,
"learning_rate": 3.5433461362045447e-07,
"logits/chosen": -2.3236405849456787,
"logits/rejected": -2.302849054336548,
"logps/chosen": -262.9497375488281,
"logps/rejected": -284.64642333984375,
"loss": 0.4962,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.15619780123233795,
"rewards/margins": 1.1346657276153564,
"rewards/rejected": -1.2908635139465332,
"step": 640
},
{
"epoch": 0.68,
"learning_rate": 3.490199415097892e-07,
"logits/chosen": -2.3758909702301025,
"logits/rejected": -2.300546169281006,
"logps/chosen": -324.71881103515625,
"logps/rejected": -303.84881591796875,
"loss": 0.494,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": 0.004534644540399313,
"rewards/margins": 1.049611210823059,
"rewards/rejected": -1.045076608657837,
"step": 650
},
{
"epoch": 0.69,
"learning_rate": 3.43651648353978e-07,
"logits/chosen": -2.4268946647644043,
"logits/rejected": -2.3779337406158447,
"logps/chosen": -320.46429443359375,
"logps/rejected": -281.7370910644531,
"loss": 0.5142,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.28596049547195435,
"rewards/margins": 0.7663468718528748,
"rewards/rejected": -1.052307367324829,
"step": 660
},
{
"epoch": 0.7,
"learning_rate": 3.3823264117846717e-07,
"logits/chosen": -2.353480100631714,
"logits/rejected": -2.3156867027282715,
"logps/chosen": -294.9992980957031,
"logps/rejected": -279.8269958496094,
"loss": 0.5042,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.3519328236579895,
"rewards/margins": 1.0733650922775269,
"rewards/rejected": -1.4252979755401611,
"step": 670
},
{
"epoch": 0.71,
"learning_rate": 3.327658544712395e-07,
"logits/chosen": -2.349534273147583,
"logits/rejected": -2.2964000701904297,
"logps/chosen": -302.86798095703125,
"logps/rejected": -278.29827880859375,
"loss": 0.5219,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.47534093260765076,
"rewards/margins": 0.663826584815979,
"rewards/rejected": -1.1391674280166626,
"step": 680
},
{
"epoch": 0.72,
"learning_rate": 3.272542485937368e-07,
"logits/chosen": -2.3173270225524902,
"logits/rejected": -2.2520382404327393,
"logps/chosen": -266.728515625,
"logps/rejected": -234.1517333984375,
"loss": 0.5044,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.2846493721008301,
"rewards/margins": 0.9897111058235168,
"rewards/rejected": -1.2743605375289917,
"step": 690
},
{
"epoch": 0.73,
"learning_rate": 3.2170080817777257e-07,
"logits/chosen": -2.3343617916107178,
"logits/rejected": -2.2195441722869873,
"logps/chosen": -297.25250244140625,
"logps/rejected": -286.7340393066406,
"loss": 0.4582,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.5840951800346375,
"rewards/margins": 1.0429940223693848,
"rewards/rejected": -1.6270891427993774,
"step": 700
},
{
"epoch": 0.73,
"eval_logits/chosen": -2.289731979370117,
"eval_logits/rejected": -2.246857166290283,
"eval_logps/chosen": -308.9957580566406,
"eval_logps/rejected": -306.2008056640625,
"eval_loss": 0.500346839427948,
"eval_rewards/accuracies": 0.7599206566810608,
"eval_rewards/chosen": -0.6337829232215881,
"eval_rewards/margins": 1.0893793106079102,
"eval_rewards/rejected": -1.7231621742248535,
"eval_runtime": 438.5736,
"eval_samples_per_second": 4.56,
"eval_steps_per_second": 0.144,
"step": 700
},
{
"epoch": 0.74,
"learning_rate": 3.1610854050930057e-07,
"logits/chosen": -2.3422415256500244,
"logits/rejected": -2.2789344787597656,
"logps/chosen": -298.95867919921875,
"logps/rejected": -271.91290283203125,
"loss": 0.474,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.5943951606750488,
"rewards/margins": 0.9698866009712219,
"rewards/rejected": -1.5642818212509155,
"step": 710
},
{
"epoch": 0.75,
"learning_rate": 3.104804738999169e-07,
"logits/chosen": -2.296921968460083,
"logits/rejected": -2.277675151824951,
"logps/chosen": -288.68511962890625,
"logps/rejected": -294.5284729003906,
"loss": 0.4741,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.4995554983615875,
"rewards/margins": 1.2392946481704712,
"rewards/rejected": -1.7388503551483154,
"step": 720
},
{
"epoch": 0.76,
"learning_rate": 3.048196560469758e-07,
"logits/chosen": -2.320403575897217,
"logits/rejected": -2.276542901992798,
"logps/chosen": -317.94012451171875,
"logps/rejected": -283.49920654296875,
"loss": 0.4819,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.4403966963291168,
"rewards/margins": 1.0142490863800049,
"rewards/rejected": -1.4546458721160889,
"step": 730
},
{
"epoch": 0.77,
"learning_rate": 2.991291523832075e-07,
"logits/chosen": -2.3161144256591797,
"logits/rejected": -2.2607274055480957,
"logps/chosen": -326.93829345703125,
"logps/rejected": -328.2031555175781,
"loss": 0.4746,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.6947178840637207,
"rewards/margins": 1.2727077007293701,
"rewards/rejected": -1.9674255847930908,
"step": 740
},
{
"epoch": 0.79,
"learning_rate": 2.934120444167326e-07,
"logits/chosen": -2.292912006378174,
"logits/rejected": -2.2443642616271973,
"logps/chosen": -282.31536865234375,
"logps/rejected": -312.34735107421875,
"loss": 0.4777,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.9387601017951965,
"rewards/margins": 1.1481364965438843,
"rewards/rejected": -2.0868964195251465,
"step": 750
},
{
"epoch": 0.8,
"learning_rate": 2.8767142806237077e-07,
"logits/chosen": -2.369844913482666,
"logits/rejected": -2.368522882461548,
"logps/chosen": -308.49700927734375,
"logps/rejected": -300.10455322265625,
"loss": 0.4665,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.539741039276123,
"rewards/margins": 1.0256460905075073,
"rewards/rejected": -1.5653870105743408,
"step": 760
},
{
"epoch": 0.81,
"learning_rate": 2.819104119651487e-07,
"logits/chosen": -2.3911471366882324,
"logits/rejected": -2.3795692920684814,
"logps/chosen": -315.2966613769531,
"logps/rejected": -321.0699157714844,
"loss": 0.5096,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.39289337396621704,
"rewards/margins": 1.0190023183822632,
"rewards/rejected": -1.411895751953125,
"step": 770
},
{
"epoch": 0.82,
"learning_rate": 2.761321158169134e-07,
"logits/chosen": -2.3824098110198975,
"logits/rejected": -2.333876132965088,
"logps/chosen": -309.57403564453125,
"logps/rejected": -271.635009765625,
"loss": 0.4965,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.31998205184936523,
"rewards/margins": 1.0204508304595947,
"rewards/rejected": -1.3404327630996704,
"step": 780
},
{
"epoch": 0.83,
"learning_rate": 2.703396686669646e-07,
"logits/chosen": -2.3626012802124023,
"logits/rejected": -2.336550235748291,
"logps/chosen": -282.4324035644531,
"logps/rejected": -289.73907470703125,
"loss": 0.4969,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.40831103920936584,
"rewards/margins": 0.9678645133972168,
"rewards/rejected": -1.3761756420135498,
"step": 790
},
{
"epoch": 0.84,
"learning_rate": 2.6453620722761895e-07,
"logits/chosen": -2.2754549980163574,
"logits/rejected": -2.227632999420166,
"logps/chosen": -324.36065673828125,
"logps/rejected": -315.53265380859375,
"loss": 0.457,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.40555134415626526,
"rewards/margins": 1.2698910236358643,
"rewards/rejected": -1.6754423379898071,
"step": 800
},
{
"epoch": 0.84,
"eval_logits/chosen": -2.3341588973999023,
"eval_logits/rejected": -2.292834520339966,
"eval_logps/chosen": -306.1227722167969,
"eval_logps/rejected": -303.84637451171875,
"eval_loss": 0.4906502664089203,
"eval_rewards/accuracies": 0.7638888955116272,
"eval_rewards/chosen": -0.4901339113712311,
"eval_rewards/margins": 1.1153074502944946,
"eval_rewards/rejected": -1.6054412126541138,
"eval_runtime": 442.4535,
"eval_samples_per_second": 4.52,
"eval_steps_per_second": 0.142,
"step": 800
},
{
"epoch": 0.85,
"learning_rate": 2.5872487417562527e-07,
"logits/chosen": -2.3688154220581055,
"logits/rejected": -2.2756457328796387,
"logps/chosen": -302.0091552734375,
"logps/rejected": -267.61712646484375,
"loss": 0.5048,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.739054799079895,
"rewards/margins": 0.9176353216171265,
"rewards/rejected": -1.656690001487732,
"step": 810
},
{
"epoch": 0.86,
"learning_rate": 2.5290881645034926e-07,
"logits/chosen": -2.371279239654541,
"logits/rejected": -2.3216042518615723,
"logps/chosen": -307.4602966308594,
"logps/rejected": -276.8441162109375,
"loss": 0.5156,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.4735354781150818,
"rewards/margins": 1.1825616359710693,
"rewards/rejected": -1.656097173690796,
"step": 820
},
{
"epoch": 0.87,
"learning_rate": 2.4709118354965077e-07,
"logits/chosen": -2.3698923587799072,
"logits/rejected": -2.3812479972839355,
"logps/chosen": -283.11395263671875,
"logps/rejected": -314.9549865722656,
"loss": 0.4728,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.5311114192008972,
"rewards/margins": 0.8778692483901978,
"rewards/rejected": -1.4089806079864502,
"step": 830
},
{
"epoch": 0.88,
"learning_rate": 2.412751258243748e-07,
"logits/chosen": -2.377016305923462,
"logits/rejected": -2.3782734870910645,
"logps/chosen": -294.67608642578125,
"logps/rejected": -310.6152038574219,
"loss": 0.4955,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.6087425351142883,
"rewards/margins": 0.8135608434677124,
"rewards/rejected": -1.4223034381866455,
"step": 840
},
{
"epoch": 0.89,
"learning_rate": 2.3546379277238103e-07,
"logits/chosen": -2.422614574432373,
"logits/rejected": -2.39127779006958,
"logps/chosen": -313.9159240722656,
"logps/rejected": -270.3652648925781,
"loss": 0.4874,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.7056568264961243,
"rewards/margins": 0.9642340540885925,
"rewards/rejected": -1.6698909997940063,
"step": 850
},
{
"epoch": 0.9,
"learning_rate": 2.2966033133303545e-07,
"logits/chosen": -2.378150463104248,
"logits/rejected": -2.3483357429504395,
"logps/chosen": -326.0942687988281,
"logps/rejected": -330.057373046875,
"loss": 0.4994,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.8599470257759094,
"rewards/margins": 0.9078429341316223,
"rewards/rejected": -1.7677898406982422,
"step": 860
},
{
"epoch": 0.91,
"learning_rate": 2.2386788418308665e-07,
"logits/chosen": -2.4225101470947266,
"logits/rejected": -2.4114060401916504,
"logps/chosen": -266.55108642578125,
"logps/rejected": -283.1296081542969,
"loss": 0.501,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.5755284428596497,
"rewards/margins": 1.0859739780426025,
"rewards/rejected": -1.6615022420883179,
"step": 870
},
{
"epoch": 0.92,
"learning_rate": 2.1808958803485133e-07,
"logits/chosen": -2.483870506286621,
"logits/rejected": -2.426514148712158,
"logps/chosen": -305.41119384765625,
"logps/rejected": -296.4634704589844,
"loss": 0.4761,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.4832102358341217,
"rewards/margins": 0.9984723925590515,
"rewards/rejected": -1.481682538986206,
"step": 880
},
{
"epoch": 0.93,
"learning_rate": 2.123285719376292e-07,
"logits/chosen": -2.4902679920196533,
"logits/rejected": -2.4276280403137207,
"logps/chosen": -306.55987548828125,
"logps/rejected": -308.28216552734375,
"loss": 0.4792,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -0.5795844793319702,
"rewards/margins": 1.21144700050354,
"rewards/rejected": -1.7910315990447998,
"step": 890
},
{
"epoch": 0.94,
"learning_rate": 2.065879555832674e-07,
"logits/chosen": -2.3825860023498535,
"logits/rejected": -2.32081937789917,
"logps/chosen": -300.0443115234375,
"logps/rejected": -298.1470642089844,
"loss": 0.4723,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.6022388339042664,
"rewards/margins": 1.1292412281036377,
"rewards/rejected": -1.7314800024032593,
"step": 900
},
{
"epoch": 0.94,
"eval_logits/chosen": -2.3762104511260986,
"eval_logits/rejected": -2.3354876041412354,
"eval_logps/chosen": -305.15618896484375,
"eval_logps/rejected": -302.8718566894531,
"eval_loss": 0.4933333098888397,
"eval_rewards/accuracies": 0.7658730149269104,
"eval_rewards/chosen": -0.4418059289455414,
"eval_rewards/margins": 1.114911437034607,
"eval_rewards/rejected": -1.5567171573638916,
"eval_runtime": 427.8237,
"eval_samples_per_second": 4.675,
"eval_steps_per_second": 0.147,
"step": 900
},
{
"epoch": 0.95,
"learning_rate": 2.0087084761679243e-07,
"logits/chosen": -2.433973789215088,
"logits/rejected": -2.3652501106262207,
"logps/chosen": -312.04620361328125,
"logps/rejected": -285.4141540527344,
"loss": 0.4747,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.3190104365348816,
"rewards/margins": 1.0960520505905151,
"rewards/rejected": -1.4150625467300415,
"step": 910
},
{
"epoch": 0.96,
"learning_rate": 1.9518034395302412e-07,
"logits/chosen": -2.3947741985321045,
"logits/rejected": -2.349290370941162,
"logps/chosen": -300.0207214355469,
"logps/rejected": -293.3101501464844,
"loss": 0.4644,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.2700619399547577,
"rewards/margins": 1.2211670875549316,
"rewards/rejected": -1.4912290573120117,
"step": 920
},
{
"epoch": 0.97,
"learning_rate": 1.895195261000831e-07,
"logits/chosen": -2.426356792449951,
"logits/rejected": -2.363300323486328,
"logps/chosen": -273.56988525390625,
"logps/rejected": -281.79144287109375,
"loss": 0.4792,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.19313231110572815,
"rewards/margins": 1.6159725189208984,
"rewards/rejected": -1.8091049194335938,
"step": 930
},
{
"epoch": 0.98,
"learning_rate": 1.8389145949069951e-07,
"logits/chosen": -2.504002094268799,
"logits/rejected": -2.434199810028076,
"logps/chosen": -312.13922119140625,
"logps/rejected": -303.17279052734375,
"loss": 0.4749,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.8358801603317261,
"rewards/margins": 1.0024197101593018,
"rewards/rejected": -1.8382999897003174,
"step": 940
},
{
"epoch": 0.99,
"learning_rate": 1.782991918222275e-07,
"logits/chosen": -2.4335973262786865,
"logits/rejected": -2.371967315673828,
"logps/chosen": -297.8493957519531,
"logps/rejected": -308.2081604003906,
"loss": 0.5115,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -1.042378544807434,
"rewards/margins": 0.9967982172966003,
"rewards/rejected": -2.0391769409179688,
"step": 950
},
{
"epoch": 1.0,
"learning_rate": 1.7274575140626315e-07,
"logits/chosen": -2.4119739532470703,
"logits/rejected": -2.3814332485198975,
"logps/chosen": -291.4102478027344,
"logps/rejected": -332.5351257324219,
"loss": 0.3924,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.6064465045928955,
"rewards/margins": 1.3707728385925293,
"rewards/rejected": -1.9772193431854248,
"step": 960
},
{
"epoch": 1.02,
"learning_rate": 1.672341455287605e-07,
"logits/chosen": -2.4356229305267334,
"logits/rejected": -2.380049228668213,
"logps/chosen": -317.14300537109375,
"logps/rejected": -315.25616455078125,
"loss": 0.2962,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.3146964907646179,
"rewards/margins": 1.8185898065567017,
"rewards/rejected": -2.133286237716675,
"step": 970
},
{
"epoch": 1.03,
"learning_rate": 1.617673588215328e-07,
"logits/chosen": -2.4142088890075684,
"logits/rejected": -2.3693430423736572,
"logps/chosen": -248.46224975585938,
"logps/rejected": -285.1511535644531,
"loss": 0.2886,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.18761667609214783,
"rewards/margins": 1.7726829051971436,
"rewards/rejected": -1.9602997303009033,
"step": 980
},
{
"epoch": 1.04,
"learning_rate": 1.5634835164602196e-07,
"logits/chosen": -2.4898295402526855,
"logits/rejected": -2.4521660804748535,
"logps/chosen": -336.2506103515625,
"logps/rejected": -302.1678161621094,
"loss": 0.2961,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.21727129817008972,
"rewards/margins": 1.7004649639129639,
"rewards/rejected": -1.917736291885376,
"step": 990
},
{
"epoch": 1.05,
"learning_rate": 1.5098005849021078e-07,
"logits/chosen": -2.4500224590301514,
"logits/rejected": -2.4026894569396973,
"logps/chosen": -296.7429504394531,
"logps/rejected": -283.27789306640625,
"loss": 0.3094,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -0.8134616017341614,
"rewards/margins": 1.605277419090271,
"rewards/rejected": -2.418738842010498,
"step": 1000
},
{
"epoch": 1.05,
"eval_logits/chosen": -2.40937876701355,
"eval_logits/rejected": -2.369751214981079,
"eval_logps/chosen": -312.38043212890625,
"eval_logps/rejected": -312.68560791015625,
"eval_loss": 0.49218976497650146,
"eval_rewards/accuracies": 0.7638888955116272,
"eval_rewards/chosen": -0.803017795085907,
"eval_rewards/margins": 1.2443830966949463,
"eval_rewards/rejected": -2.04740047454834,
"eval_runtime": 407.9979,
"eval_samples_per_second": 4.902,
"eval_steps_per_second": 0.154,
"step": 1000
},
{
"epoch": 1.06,
"learning_rate": 1.4566538637954554e-07,
"logits/chosen": -2.456105947494507,
"logits/rejected": -2.391247272491455,
"logps/chosen": -317.70330810546875,
"logps/rejected": -292.7303771972656,
"loss": 0.2867,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -0.4770506024360657,
"rewards/margins": 1.7201404571533203,
"rewards/rejected": -2.1971912384033203,
"step": 1010
},
{
"epoch": 1.07,
"learning_rate": 1.404072133027306e-07,
"logits/chosen": -2.468541383743286,
"logits/rejected": -2.4047131538391113,
"logps/chosen": -324.1610107421875,
"logps/rejected": -310.6138610839844,
"loss": 0.3011,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -0.3531934320926666,
"rewards/margins": 1.7022035121917725,
"rewards/rejected": -2.055396795272827,
"step": 1020
},
{
"epoch": 1.08,
"learning_rate": 1.3520838665324703e-07,
"logits/chosen": -2.451047658920288,
"logits/rejected": -2.426084280014038,
"logps/chosen": -289.8973083496094,
"logps/rejected": -315.23968505859375,
"loss": 0.2848,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.3617556095123291,
"rewards/margins": 1.7169090509414673,
"rewards/rejected": -2.078664779663086,
"step": 1030
},
{
"epoch": 1.09,
"learning_rate": 1.3007172168743852e-07,
"logits/chosen": -2.4811718463897705,
"logits/rejected": -2.442286968231201,
"logps/chosen": -315.90509033203125,
"logps/rejected": -345.96820068359375,
"loss": 0.2943,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.2960956394672394,
"rewards/margins": 1.6934324502944946,
"rewards/rejected": -1.9895280599594116,
"step": 1040
},
{
"epoch": 1.1,
"learning_rate": 1.2500000000000005e-07,
"logits/chosen": -2.4405837059020996,
"logits/rejected": -2.38596773147583,
"logps/chosen": -287.0311279296875,
"logps/rejected": -303.80535888671875,
"loss": 0.2866,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.2767558693885803,
"rewards/margins": 1.9063446521759033,
"rewards/rejected": -2.183100461959839,
"step": 1050
},
{
"epoch": 1.11,
"learning_rate": 1.1999596801769616e-07,
"logits/chosen": -2.440513849258423,
"logits/rejected": -2.3295609951019287,
"logps/chosen": -326.0518493652344,
"logps/rejected": -333.1617431640625,
"loss": 0.2744,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.25197187066078186,
"rewards/margins": 2.045137882232666,
"rewards/rejected": -2.297109603881836,
"step": 1060
},
{
"epoch": 1.12,
"learning_rate": 1.1506233551212185e-07,
"logits/chosen": -2.414334535598755,
"logits/rejected": -2.3740944862365723,
"logps/chosen": -306.10992431640625,
"logps/rejected": -327.01959228515625,
"loss": 0.2711,
"rewards/accuracies": 0.90625,
"rewards/chosen": -0.261447936296463,
"rewards/margins": 2.056182384490967,
"rewards/rejected": -2.3176300525665283,
"step": 1070
},
{
"epoch": 1.13,
"learning_rate": 1.1020177413231332e-07,
"logits/chosen": -2.424375057220459,
"logits/rejected": -2.3422465324401855,
"logps/chosen": -267.51129150390625,
"logps/rejected": -272.04876708984375,
"loss": 0.2992,
"rewards/accuracies": 0.84375,
"rewards/chosen": -0.5566546320915222,
"rewards/margins": 1.5810120105743408,
"rewards/rejected": -2.1376664638519287,
"step": 1080
},
{
"epoch": 1.14,
"learning_rate": 1.0541691595800336e-07,
"logits/chosen": -2.4174036979675293,
"logits/rejected": -2.417484760284424,
"logps/chosen": -288.2060852050781,
"logps/rejected": -316.9212341308594,
"loss": 0.2776,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.14382892847061157,
"rewards/margins": 1.983428716659546,
"rewards/rejected": -2.127257823944092,
"step": 1090
},
{
"epoch": 1.15,
"learning_rate": 1.007103520743035e-07,
"logits/chosen": -2.4386072158813477,
"logits/rejected": -2.383044719696045,
"logps/chosen": -316.8988342285156,
"logps/rejected": -311.27642822265625,
"loss": 0.2725,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.27822011709213257,
"rewards/margins": 1.802807092666626,
"rewards/rejected": -2.081026792526245,
"step": 1100
},
{
"epoch": 1.15,
"eval_logits/chosen": -2.3785088062286377,
"eval_logits/rejected": -2.3381552696228027,
"eval_logps/chosen": -307.5903015136719,
"eval_logps/rejected": -309.0183410644531,
"eval_loss": 0.49208447337150574,
"eval_rewards/accuracies": 0.7460317611694336,
"eval_rewards/chosen": -0.5635129809379578,
"eval_rewards/margins": 1.3005269765853882,
"eval_rewards/rejected": -1.8640400171279907,
"eval_runtime": 431.8636,
"eval_samples_per_second": 4.631,
"eval_steps_per_second": 0.146,
"step": 1100
},
{
"epoch": 1.16,
"learning_rate": 9.608463116858542e-08,
"logits/chosen": -2.4096856117248535,
"logits/rejected": -2.3269238471984863,
"logps/chosen": -319.33123779296875,
"logps/rejected": -314.37982177734375,
"loss": 0.2832,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.4104345738887787,
"rewards/margins": 1.9642629623413086,
"rewards/rejected": -2.37469744682312,
"step": 1110
},
{
"epoch": 1.17,
"learning_rate": 9.15422581503224e-08,
"logits/chosen": -2.4082019329071045,
"logits/rejected": -2.3823094367980957,
"logps/chosen": -281.8961181640625,
"logps/rejected": -310.1343078613281,
"loss": 0.2659,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.4686800539493561,
"rewards/margins": 1.9623409509658813,
"rewards/rejected": -2.431021213531494,
"step": 1120
},
{
"epoch": 1.18,
"learning_rate": 8.70856927946362e-08,
"logits/chosen": -2.4076130390167236,
"logits/rejected": -2.377281427383423,
"logps/chosen": -313.17877197265625,
"logps/rejected": -320.7528381347656,
"loss": 0.2712,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.3578185439109802,
"rewards/margins": 1.8895679712295532,
"rewards/rejected": -2.2473864555358887,
"step": 1130
},
{
"epoch": 1.19,
"learning_rate": 8.271734841028552e-08,
"logits/chosen": -2.4748058319091797,
"logits/rejected": -2.3889963626861572,
"logps/chosen": -317.75933837890625,
"logps/rejected": -335.86553955078125,
"loss": 0.2587,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.460686057806015,
"rewards/margins": 1.9483623504638672,
"rewards/rejected": -2.409048557281494,
"step": 1140
},
{
"epoch": 1.2,
"learning_rate": 7.843959053281663e-08,
"logits/chosen": -2.419320583343506,
"logits/rejected": -2.3716225624084473,
"logps/chosen": -308.89141845703125,
"logps/rejected": -291.35797119140625,
"loss": 0.3091,
"rewards/accuracies": 0.90625,
"rewards/chosen": -0.3241550624370575,
"rewards/margins": 1.9242489337921143,
"rewards/rejected": -2.248404026031494,
"step": 1150
},
{
"epoch": 1.21,
"learning_rate": 7.425473564358456e-08,
"logits/chosen": -2.4155445098876953,
"logits/rejected": -2.374995708465576,
"logps/chosen": -285.75823974609375,
"logps/rejected": -320.6084899902344,
"loss": 0.2787,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.321144163608551,
"rewards/margins": 2.070187568664551,
"rewards/rejected": -2.391331911087036,
"step": 1160
},
{
"epoch": 1.22,
"learning_rate": 7.016504991533726e-08,
"logits/chosen": -2.4452130794525146,
"logits/rejected": -2.416900157928467,
"logps/chosen": -333.56890869140625,
"logps/rejected": -288.9085998535156,
"loss": 0.2819,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.34982940554618835,
"rewards/margins": 1.8246349096298218,
"rewards/rejected": -2.174464702606201,
"step": 1170
},
{
"epoch": 1.24,
"learning_rate": 6.617274798504286e-08,
"logits/chosen": -2.4035491943359375,
"logits/rejected": -2.3629653453826904,
"logps/chosen": -306.9756164550781,
"logps/rejected": -289.35479736328125,
"loss": 0.2805,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.36044904589653015,
"rewards/margins": 1.8948723077774048,
"rewards/rejected": -2.2553212642669678,
"step": 1180
},
{
"epoch": 1.25,
"learning_rate": 6.22799917546252e-08,
"logits/chosen": -2.459360361099243,
"logits/rejected": -2.391692638397217,
"logps/chosen": -314.09185791015625,
"logps/rejected": -300.3113098144531,
"loss": 0.2546,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.302379310131073,
"rewards/margins": 2.0588433742523193,
"rewards/rejected": -2.361222743988037,
"step": 1190
},
{
"epoch": 1.26,
"learning_rate": 5.848888922025552e-08,
"logits/chosen": -2.473045825958252,
"logits/rejected": -2.456024408340454,
"logps/chosen": -312.7765808105469,
"logps/rejected": -322.79498291015625,
"loss": 0.2932,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -0.290397971868515,
"rewards/margins": 1.813227653503418,
"rewards/rejected": -2.103625774383545,
"step": 1200
},
{
"epoch": 1.26,
"eval_logits/chosen": -2.391538381576538,
"eval_logits/rejected": -2.351095199584961,
"eval_logps/chosen": -309.3631591796875,
"eval_logps/rejected": -311.79766845703125,
"eval_loss": 0.49240604043006897,
"eval_rewards/accuracies": 0.7579365372657776,
"eval_rewards/chosen": -0.6521540880203247,
"eval_rewards/margins": 1.3508530855178833,
"eval_rewards/rejected": -2.003007173538208,
"eval_runtime": 414.0649,
"eval_samples_per_second": 4.83,
"eval_steps_per_second": 0.152,
"step": 1200
},
{
"epoch": 1.27,
"learning_rate": 5.48014933308352e-08,
"logits/chosen": -2.407109022140503,
"logits/rejected": -2.3259758949279785,
"logps/chosen": -230.8815155029297,
"logps/rejected": -267.571044921875,
"loss": 0.3043,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -0.5537179112434387,
"rewards/margins": 1.7985881567001343,
"rewards/rejected": -2.3523058891296387,
"step": 1210
},
{
"epoch": 1.28,
"learning_rate": 5.121980087628802e-08,
"logits/chosen": -2.468632221221924,
"logits/rejected": -2.407593250274658,
"logps/chosen": -321.1299133300781,
"logps/rejected": -319.76788330078125,
"loss": 0.2968,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.3099203407764435,
"rewards/margins": 2.04097580909729,
"rewards/rejected": -2.350896120071411,
"step": 1220
},
{
"epoch": 1.29,
"learning_rate": 4.774575140626316e-08,
"logits/chosen": -2.399498462677002,
"logits/rejected": -2.317718982696533,
"logps/chosen": -311.56378173828125,
"logps/rejected": -322.6271057128906,
"loss": 0.2885,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.34509938955307007,
"rewards/margins": 1.8977705240249634,
"rewards/rejected": -2.2428698539733887,
"step": 1230
},
{
"epoch": 1.3,
"learning_rate": 4.438122617983442e-08,
"logits/chosen": -2.3652095794677734,
"logits/rejected": -2.3344194889068604,
"logps/chosen": -290.28790283203125,
"logps/rejected": -294.1825866699219,
"loss": 0.2989,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4549785256385803,
"rewards/margins": 1.7224676609039307,
"rewards/rejected": -2.177445888519287,
"step": 1240
},
{
"epoch": 1.31,
"learning_rate": 4.112804714676593e-08,
"logits/chosen": -2.459439754486084,
"logits/rejected": -2.4127402305603027,
"logps/chosen": -292.1894226074219,
"logps/rejected": -307.1498107910156,
"loss": 0.2864,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.346891850233078,
"rewards/margins": 1.9702911376953125,
"rewards/rejected": -2.317183017730713,
"step": 1250
},
{
"epoch": 1.32,
"learning_rate": 3.798797596089351e-08,
"logits/chosen": -2.427961587905884,
"logits/rejected": -2.4093945026397705,
"logps/chosen": -316.4217224121094,
"logps/rejected": -341.4559631347656,
"loss": 0.2918,
"rewards/accuracies": 0.90625,
"rewards/chosen": -0.42415857315063477,
"rewards/margins": 1.8887336254119873,
"rewards/rejected": -2.312892198562622,
"step": 1260
},
{
"epoch": 1.33,
"learning_rate": 3.496271302615869e-08,
"logits/chosen": -2.4277498722076416,
"logits/rejected": -2.35754656791687,
"logps/chosen": -290.10235595703125,
"logps/rejected": -304.0705261230469,
"loss": 0.2807,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.38870421051979065,
"rewards/margins": 2.0075557231903076,
"rewards/rejected": -2.3962597846984863,
"step": 1270
},
{
"epoch": 1.34,
"learning_rate": 3.205389657580943e-08,
"logits/chosen": -2.438321590423584,
"logits/rejected": -2.3755178451538086,
"logps/chosen": -283.269775390625,
"logps/rejected": -337.3073425292969,
"loss": 0.2519,
"rewards/accuracies": 0.90625,
"rewards/chosen": -0.2695261836051941,
"rewards/margins": 2.0451509952545166,
"rewards/rejected": -2.3146772384643555,
"step": 1280
},
{
"epoch": 1.35,
"learning_rate": 2.9263101785268252e-08,
"logits/chosen": -2.4288270473480225,
"logits/rejected": -2.39312481880188,
"logps/chosen": -325.607421875,
"logps/rejected": -317.9021911621094,
"loss": 0.2867,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.3596641421318054,
"rewards/margins": 1.9093034267425537,
"rewards/rejected": -2.268967390060425,
"step": 1290
},
{
"epoch": 1.36,
"learning_rate": 2.659183991914696e-08,
"logits/chosen": -2.419738531112671,
"logits/rejected": -2.362929105758667,
"logps/chosen": -296.9184875488281,
"logps/rejected": -324.33587646484375,
"loss": 0.275,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.3492244482040405,
"rewards/margins": 2.0286529064178467,
"rewards/rejected": -2.3778772354125977,
"step": 1300
},
{
"epoch": 1.36,
"eval_logits/chosen": -2.39336895942688,
"eval_logits/rejected": -2.353144407272339,
"eval_logps/chosen": -309.0525817871094,
"eval_logps/rejected": -311.2369079589844,
"eval_loss": 0.4915713965892792,
"eval_rewards/accuracies": 0.7599206566810608,
"eval_rewards/chosen": -0.6366247534751892,
"eval_rewards/margins": 1.3383426666259766,
"eval_rewards/rejected": -1.9749674797058105,
"eval_runtime": 412.9249,
"eval_samples_per_second": 4.843,
"eval_steps_per_second": 0.153,
"step": 1300
},
{
"epoch": 1.37,
"learning_rate": 2.4041557512869876e-08,
"logits/chosen": -2.3756937980651855,
"logits/rejected": -2.3104584217071533,
"logps/chosen": -307.74395751953125,
"logps/rejected": -316.4643859863281,
"loss": 0.301,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -0.6529245972633362,
"rewards/margins": 1.7977030277252197,
"rewards/rejected": -2.4506278038024902,
"step": 1310
},
{
"epoch": 1.38,
"learning_rate": 2.1613635589349756e-08,
"logits/chosen": -2.4178683757781982,
"logits/rejected": -2.3683836460113525,
"logps/chosen": -293.5295715332031,
"logps/rejected": -329.43756103515625,
"loss": 0.2802,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.4704504907131195,
"rewards/margins": 1.998552680015564,
"rewards/rejected": -2.469003200531006,
"step": 1320
},
{
"epoch": 1.39,
"learning_rate": 1.9309388911139424e-08,
"logits/chosen": -2.4530060291290283,
"logits/rejected": -2.417515993118286,
"logps/chosen": -303.9961853027344,
"logps/rejected": -316.8196716308594,
"loss": 0.2883,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -0.4491092562675476,
"rewards/margins": 1.8476078510284424,
"rewards/rejected": -2.2967171669006348,
"step": 1330
},
{
"epoch": 1.4,
"learning_rate": 1.713006526846439e-08,
"logits/chosen": -2.3352913856506348,
"logits/rejected": -2.3332459926605225,
"logps/chosen": -309.6740417480469,
"logps/rejected": -316.5276184082031,
"loss": 0.3034,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -0.5785962343215942,
"rewards/margins": 1.668442964553833,
"rewards/rejected": -2.247039318084717,
"step": 1340
},
{
"epoch": 1.41,
"learning_rate": 1.507684480352292e-08,
"logits/chosen": -2.444365978240967,
"logits/rejected": -2.3748979568481445,
"logps/chosen": -273.01177978515625,
"logps/rejected": -300.568115234375,
"loss": 0.2704,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.5734509229660034,
"rewards/margins": 1.8402652740478516,
"rewards/rejected": -2.4137163162231445,
"step": 1350
},
{
"epoch": 1.42,
"learning_rate": 1.3150839371417699e-08,
"logits/chosen": -2.3913302421569824,
"logits/rejected": -2.3794515132904053,
"logps/chosen": -270.5587463378906,
"logps/rejected": -277.6205749511719,
"loss": 0.3003,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.3957940936088562,
"rewards/margins": 1.739976167678833,
"rewards/rejected": -2.135770320892334,
"step": 1360
},
{
"epoch": 1.43,
"learning_rate": 1.1353091938067023e-08,
"logits/chosen": -2.40122652053833,
"logits/rejected": -2.317656993865967,
"logps/chosen": -306.97454833984375,
"logps/rejected": -288.46038818359375,
"loss": 0.2884,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.4694311022758484,
"rewards/margins": 1.7464059591293335,
"rewards/rejected": -2.215837240219116,
"step": 1370
},
{
"epoch": 1.44,
"learning_rate": 9.684576015420275e-09,
"logits/chosen": -2.429955005645752,
"logits/rejected": -2.340874433517456,
"logps/chosen": -284.2897033691406,
"logps/rejected": -346.4473571777344,
"loss": 0.259,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.4956478178501129,
"rewards/margins": 2.128859043121338,
"rewards/rejected": -2.624507188796997,
"step": 1380
},
{
"epoch": 1.46,
"learning_rate": 8.14619513428405e-09,
"logits/chosen": -2.41424822807312,
"logits/rejected": -2.4188144207000732,
"logps/chosen": -284.09356689453125,
"logps/rejected": -301.265380859375,
"loss": 0.3023,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -0.6043499708175659,
"rewards/margins": 1.725403070449829,
"rewards/rejected": -2.3297529220581055,
"step": 1390
},
{
"epoch": 1.47,
"learning_rate": 6.738782355044048e-09,
"logits/chosen": -2.420943260192871,
"logits/rejected": -2.3737356662750244,
"logps/chosen": -295.2271423339844,
"logps/rejected": -300.8685607910156,
"loss": 0.2768,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.4684749245643616,
"rewards/margins": 2.138996124267578,
"rewards/rejected": -2.607471227645874,
"step": 1400
},
{
"epoch": 1.47,
"eval_logits/chosen": -2.3908376693725586,
"eval_logits/rejected": -2.3505396842956543,
"eval_logps/chosen": -310.34185791015625,
"eval_logps/rejected": -312.6646423339844,
"eval_loss": 0.49215617775917053,
"eval_rewards/accuracies": 0.7579365372657776,
"eval_rewards/chosen": -0.7010902166366577,
"eval_rewards/margins": 1.345264196395874,
"eval_rewards/rejected": -2.0463547706604004,
"eval_runtime": 434.5194,
"eval_samples_per_second": 4.603,
"eval_steps_per_second": 0.145,
"step": 1400
},
{
"epoch": 1.48,
"learning_rate": 5.463099816548577e-09,
"logits/chosen": -2.435065269470215,
"logits/rejected": -2.368187427520752,
"logps/chosen": -305.482666015625,
"logps/rejected": -319.5362548828125,
"loss": 0.2907,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.512501060962677,
"rewards/margins": 1.8626611232757568,
"rewards/rejected": -2.375162124633789,
"step": 1410
},
{
"epoch": 1.49,
"learning_rate": 4.319838323396691e-09,
"logits/chosen": -2.425668239593506,
"logits/rejected": -2.394260883331299,
"logps/chosen": -285.38983154296875,
"logps/rejected": -326.1392822265625,
"loss": 0.28,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.43002137541770935,
"rewards/margins": 1.9287302494049072,
"rewards/rejected": -2.3587517738342285,
"step": 1420
},
{
"epoch": 1.5,
"learning_rate": 3.309616971855195e-09,
"logits/chosen": -2.3961329460144043,
"logits/rejected": -2.3518338203430176,
"logps/chosen": -313.55267333984375,
"logps/rejected": -286.34320068359375,
"loss": 0.2816,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.41422533988952637,
"rewards/margins": 1.944323182106018,
"rewards/rejected": -2.358548402786255,
"step": 1430
},
{
"epoch": 1.51,
"learning_rate": 2.4329828146074096e-09,
"logits/chosen": -2.386317491531372,
"logits/rejected": -2.3524794578552246,
"logps/chosen": -313.0305480957031,
"logps/rejected": -332.1868896484375,
"loss": 0.2798,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.3209957182407379,
"rewards/margins": 2.147188663482666,
"rewards/rejected": -2.46818470954895,
"step": 1440
},
{
"epoch": 1.52,
"learning_rate": 1.690410564514244e-09,
"logits/chosen": -2.411320209503174,
"logits/rejected": -2.3845181465148926,
"logps/chosen": -296.78228759765625,
"logps/rejected": -306.8433837890625,
"loss": 0.2898,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -0.4139639735221863,
"rewards/margins": 1.8331149816513062,
"rewards/rejected": -2.2470791339874268,
"step": 1450
},
{
"epoch": 1.53,
"learning_rate": 1.0823023375489126e-09,
"logits/chosen": -2.468296766281128,
"logits/rejected": -2.4058051109313965,
"logps/chosen": -282.4747619628906,
"logps/rejected": -294.89923095703125,
"loss": 0.2817,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.3540436029434204,
"rewards/margins": 1.8670495748519897,
"rewards/rejected": -2.22109317779541,
"step": 1460
},
{
"epoch": 1.54,
"learning_rate": 6.089874350439505e-10,
"logits/chosen": -2.4278903007507324,
"logits/rejected": -2.412611961364746,
"logps/chosen": -309.91180419921875,
"logps/rejected": -351.41607666015625,
"loss": 0.2826,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -0.38388413190841675,
"rewards/margins": 1.9360164403915405,
"rewards/rejected": -2.3199005126953125,
"step": 1470
},
{
"epoch": 1.55,
"learning_rate": 2.707221653688585e-10,
"logits/chosen": -2.3936755657196045,
"logits/rejected": -2.3537259101867676,
"logps/chosen": -316.6966247558594,
"logps/rejected": -323.77154541015625,
"loss": 0.2674,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.3255850076675415,
"rewards/margins": 2.003681182861328,
"rewards/rejected": -2.329266309738159,
"step": 1480
},
{
"epoch": 1.56,
"learning_rate": 6.768970513457151e-11,
"logits/chosen": -2.3720383644104004,
"logits/rejected": -2.346740245819092,
"logps/chosen": -301.3364562988281,
"logps/rejected": -312.578125,
"loss": 0.2749,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -0.46938952803611755,
"rewards/margins": 1.996596097946167,
"rewards/rejected": -2.4659857749938965,
"step": 1490
},
{
"epoch": 1.57,
"learning_rate": 0.0,
"logits/chosen": -2.345372200012207,
"logits/rejected": -2.3051817417144775,
"logps/chosen": -275.92779541015625,
"logps/rejected": -335.74761962890625,
"loss": 0.2863,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -0.5277979969978333,
"rewards/margins": 2.0437979698181152,
"rewards/rejected": -2.5715959072113037,
"step": 1500
},
{
"epoch": 1.57,
"eval_logits/chosen": -2.3900513648986816,
"eval_logits/rejected": -2.3497886657714844,
"eval_logps/chosen": -310.23736572265625,
"eval_logps/rejected": -312.59942626953125,
"eval_loss": 0.4916023015975952,
"eval_rewards/accuracies": 0.7579365372657776,
"eval_rewards/chosen": -0.6958636045455933,
"eval_rewards/margins": 1.3472286462783813,
"eval_rewards/rejected": -2.0430922508239746,
"eval_runtime": 454.0528,
"eval_samples_per_second": 4.405,
"eval_steps_per_second": 0.139,
"step": 1500
},
{
"epoch": 1.57,
"step": 1500,
"total_flos": 0.0,
"train_loss": 0.4390208276112874,
"train_runtime": 42485.337,
"train_samples_per_second": 2.26,
"train_steps_per_second": 0.035
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}