dpo-selective-buffer-safeipo / trainer_state.json
wxzhang's picture
Model save
581934d verified
raw
history blame
114 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9995965030262273,
"eval_steps": 300,
"global_step": 1858,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 2.6881720430107528e-09,
"logits/chosen": -2.670260429382324,
"logits/rejected": -2.1533777713775635,
"logps/chosen": -299.33551025390625,
"logps/rejected": -186.81130981445312,
"loss": 13125.0,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"rewards/safe_rewards": 0.0,
"rewards/unsafe_rewards": 0.0,
"step": 1
},
{
"epoch": 0.01,
"learning_rate": 2.6881720430107527e-08,
"logits/chosen": -2.4763858318328857,
"logits/rejected": -2.354341745376587,
"logps/chosen": -201.82504272460938,
"logps/rejected": -189.45822143554688,
"loss": 14971.9097,
"rewards/accuracies": 0.4930555522441864,
"rewards/chosen": 9.199242413160391e-06,
"rewards/margins": -8.218608854804188e-05,
"rewards/rejected": 9.138535824604332e-05,
"rewards/safe_rewards": 1.8223654478788376e-05,
"rewards/unsafe_rewards": 1.7482994962847442e-07,
"step": 10
},
{
"epoch": 0.01,
"learning_rate": 5.3763440860215054e-08,
"logits/chosen": -2.44279408454895,
"logits/rejected": -2.218726873397827,
"logps/chosen": -226.34970092773438,
"logps/rejected": -181.1803436279297,
"loss": 15239.6516,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.0002221722388640046,
"rewards/margins": -0.0001655549422139302,
"rewards/rejected": -5.6617216614540666e-05,
"rewards/safe_rewards": -0.0002462912234477699,
"rewards/unsafe_rewards": -0.0001980532251764089,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 8.064516129032257e-08,
"logits/chosen": -2.4326512813568115,
"logits/rejected": -2.2918035984039307,
"logps/chosen": -215.0783233642578,
"logps/rejected": -189.28309631347656,
"loss": 14993.0922,
"rewards/accuracies": 0.375,
"rewards/chosen": -0.0004244670271873474,
"rewards/margins": -0.000839566346257925,
"rewards/rejected": 0.0004150994645897299,
"rewards/safe_rewards": -0.0002224749478045851,
"rewards/unsafe_rewards": -0.0006264590774662793,
"step": 30
},
{
"epoch": 0.02,
"learning_rate": 1.0752688172043011e-07,
"logits/chosen": -2.375096082687378,
"logits/rejected": -2.157045841217041,
"logps/chosen": -180.7657470703125,
"logps/rejected": -173.87054443359375,
"loss": 14723.7563,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.00011304272629786283,
"rewards/margins": 0.0008593280799686909,
"rewards/rejected": -0.0009723707917146385,
"rewards/safe_rewards": 0.0004990470479242504,
"rewards/unsafe_rewards": -0.0007251326460391283,
"step": 40
},
{
"epoch": 0.03,
"learning_rate": 1.3440860215053762e-07,
"logits/chosen": -2.489865303039551,
"logits/rejected": -2.1804325580596924,
"logps/chosen": -209.59664916992188,
"logps/rejected": -167.6984100341797,
"loss": 14816.0938,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": 0.0011160348076373339,
"rewards/margins": 0.0046949503012001514,
"rewards/rejected": -0.0035789154935628176,
"rewards/safe_rewards": 0.00042248546378687024,
"rewards/unsafe_rewards": 0.0018095843261107802,
"step": 50
},
{
"epoch": 0.03,
"learning_rate": 1.6129032258064515e-07,
"logits/chosen": -2.4289777278900146,
"logits/rejected": -2.2897238731384277,
"logps/chosen": -185.92630004882812,
"logps/rejected": -185.1912841796875,
"loss": 14593.3875,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.0020237788558006287,
"rewards/margins": 0.002217882312834263,
"rewards/rejected": -0.004241660702973604,
"rewards/safe_rewards": -0.001639070687815547,
"rewards/unsafe_rewards": -0.002408486558124423,
"step": 60
},
{
"epoch": 0.04,
"learning_rate": 1.8817204301075268e-07,
"logits/chosen": -2.423598289489746,
"logits/rejected": -2.2320618629455566,
"logps/chosen": -202.7837677001953,
"logps/rejected": -184.62033081054688,
"loss": 14276.2172,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.005977807100862265,
"rewards/margins": 0.005661749746650457,
"rewards/rejected": -0.011639557778835297,
"rewards/safe_rewards": -0.00685838982462883,
"rewards/unsafe_rewards": -0.005097225774079561,
"step": 70
},
{
"epoch": 0.04,
"learning_rate": 2.1505376344086022e-07,
"logits/chosen": -2.4342398643493652,
"logits/rejected": -2.271530866622925,
"logps/chosen": -221.3511505126953,
"logps/rejected": -196.20684814453125,
"loss": 14430.5766,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.017918918281793594,
"rewards/margins": 0.013557764701545238,
"rewards/rejected": -0.03147668391466141,
"rewards/safe_rewards": -0.020387938246130943,
"rewards/unsafe_rewards": -0.015449894592165947,
"step": 80
},
{
"epoch": 0.05,
"learning_rate": 2.4193548387096775e-07,
"logits/chosen": -2.4415786266326904,
"logits/rejected": -2.301713466644287,
"logps/chosen": -211.0481414794922,
"logps/rejected": -171.99159240722656,
"loss": 14501.5938,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.05115891247987747,
"rewards/margins": 0.025721188634634018,
"rewards/rejected": -0.07688009738922119,
"rewards/safe_rewards": -0.04981667920947075,
"rewards/unsafe_rewards": -0.05250114947557449,
"step": 90
},
{
"epoch": 0.05,
"learning_rate": 2.6881720430107523e-07,
"logits/chosen": -2.4574618339538574,
"logits/rejected": -2.3291263580322266,
"logps/chosen": -204.63087463378906,
"logps/rejected": -179.63002014160156,
"loss": 14364.2516,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.08542316406965256,
"rewards/margins": 0.03905550017952919,
"rewards/rejected": -0.12447866052389145,
"rewards/safe_rewards": -0.08581903576850891,
"rewards/unsafe_rewards": -0.0850272923707962,
"step": 100
},
{
"epoch": 0.06,
"learning_rate": 2.956989247311828e-07,
"logits/chosen": -2.429097890853882,
"logits/rejected": -2.2684311866760254,
"logps/chosen": -222.95913696289062,
"logps/rejected": -187.91293334960938,
"loss": 14761.6688,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.11019313335418701,
"rewards/margins": 0.048916272819042206,
"rewards/rejected": -0.159109428524971,
"rewards/safe_rewards": -0.10632093250751495,
"rewards/unsafe_rewards": -0.11406532675027847,
"step": 110
},
{
"epoch": 0.06,
"learning_rate": 3.225806451612903e-07,
"logits/chosen": -2.4783873558044434,
"logits/rejected": -2.262389659881592,
"logps/chosen": -220.35800170898438,
"logps/rejected": -169.4020233154297,
"loss": 13980.0906,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.09087369590997696,
"rewards/margins": 0.07458756119012833,
"rewards/rejected": -0.1654612421989441,
"rewards/safe_rewards": -0.08417809009552002,
"rewards/unsafe_rewards": -0.0975693017244339,
"step": 120
},
{
"epoch": 0.07,
"learning_rate": 3.4946236559139783e-07,
"logits/chosen": -2.47499942779541,
"logits/rejected": -2.3391237258911133,
"logps/chosen": -224.6880645751953,
"logps/rejected": -181.26388549804688,
"loss": 13439.1469,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.10466276109218597,
"rewards/margins": 0.0889045000076294,
"rewards/rejected": -0.19356727600097656,
"rewards/safe_rewards": -0.07197071611881256,
"rewards/unsafe_rewards": -0.13735483586788177,
"step": 130
},
{
"epoch": 0.08,
"learning_rate": 3.7634408602150537e-07,
"logits/chosen": -2.5111544132232666,
"logits/rejected": -2.323676347732544,
"logps/chosen": -232.02120971679688,
"logps/rejected": -207.3649444580078,
"loss": 13156.9813,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.20595864951610565,
"rewards/margins": 0.0673115998506546,
"rewards/rejected": -0.27327024936676025,
"rewards/safe_rewards": -0.19269119203090668,
"rewards/unsafe_rewards": -0.21922609210014343,
"step": 140
},
{
"epoch": 0.08,
"learning_rate": 4.0322580645161285e-07,
"logits/chosen": -2.440796375274658,
"logits/rejected": -2.294173002243042,
"logps/chosen": -229.3905792236328,
"logps/rejected": -201.4041748046875,
"loss": 12286.168,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.2244747132062912,
"rewards/margins": 0.09208513051271439,
"rewards/rejected": -0.3165598511695862,
"rewards/safe_rewards": -0.2293473184108734,
"rewards/unsafe_rewards": -0.21960210800170898,
"step": 150
},
{
"epoch": 0.09,
"learning_rate": 4.3010752688172043e-07,
"logits/chosen": -2.4001574516296387,
"logits/rejected": -2.2244515419006348,
"logps/chosen": -242.0061492919922,
"logps/rejected": -220.7064666748047,
"loss": 13450.5672,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.2182481735944748,
"rewards/margins": 0.09150904417037964,
"rewards/rejected": -0.30975720286369324,
"rewards/safe_rewards": -0.20791450142860413,
"rewards/unsafe_rewards": -0.22858186066150665,
"step": 160
},
{
"epoch": 0.09,
"learning_rate": 4.569892473118279e-07,
"logits/chosen": -2.4316458702087402,
"logits/rejected": -2.3187336921691895,
"logps/chosen": -229.7932891845703,
"logps/rejected": -191.04086303710938,
"loss": 12580.4609,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.15405751764774323,
"rewards/margins": 0.08396536856889725,
"rewards/rejected": -0.23802292346954346,
"rewards/safe_rewards": -0.13653624057769775,
"rewards/unsafe_rewards": -0.1715788096189499,
"step": 170
},
{
"epoch": 0.1,
"learning_rate": 4.838709677419355e-07,
"logits/chosen": -2.473118305206299,
"logits/rejected": -2.3191967010498047,
"logps/chosen": -217.89785766601562,
"logps/rejected": -209.47775268554688,
"loss": 12886.1875,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.2591598629951477,
"rewards/margins": 0.1223045364022255,
"rewards/rejected": -0.3814643919467926,
"rewards/safe_rewards": -0.2669682800769806,
"rewards/unsafe_rewards": -0.25135138630867004,
"step": 180
},
{
"epoch": 0.1,
"learning_rate": 4.999929391798331e-07,
"logits/chosen": -2.4944000244140625,
"logits/rejected": -2.323819637298584,
"logps/chosen": -224.4022674560547,
"logps/rejected": -209.09341430664062,
"loss": 12964.0625,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.30063313245773315,
"rewards/margins": 0.11855147778987885,
"rewards/rejected": -0.4191845953464508,
"rewards/safe_rewards": -0.30268028378486633,
"rewards/unsafe_rewards": -0.2985859215259552,
"step": 190
},
{
"epoch": 0.11,
"learning_rate": 4.9991350953333e-07,
"logits/chosen": -2.4175021648406982,
"logits/rejected": -2.261751413345337,
"logps/chosen": -257.97100830078125,
"logps/rejected": -248.92697143554688,
"loss": 12408.5,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.3828560709953308,
"rewards/margins": 0.09374293684959412,
"rewards/rejected": -0.4765990376472473,
"rewards/safe_rewards": -0.37296319007873535,
"rewards/unsafe_rewards": -0.39274901151657104,
"step": 200
},
{
"epoch": 0.11,
"learning_rate": 4.997458523498236e-07,
"logits/chosen": -2.4530012607574463,
"logits/rejected": -2.3131752014160156,
"logps/chosen": -222.4829559326172,
"logps/rejected": -198.89447021484375,
"loss": 12109.7687,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3033100664615631,
"rewards/margins": 0.09060867875814438,
"rewards/rejected": -0.3939187526702881,
"rewards/safe_rewards": -0.29972246289253235,
"rewards/unsafe_rewards": -0.30689769983291626,
"step": 210
},
{
"epoch": 0.12,
"learning_rate": 4.99490026817712e-07,
"logits/chosen": -2.415255308151245,
"logits/rejected": -2.290353775024414,
"logps/chosen": -225.1759490966797,
"logps/rejected": -215.8201446533203,
"loss": 12286.7969,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.308417946100235,
"rewards/margins": 0.15336203575134277,
"rewards/rejected": -0.46177998185157776,
"rewards/safe_rewards": -0.2879489064216614,
"rewards/unsafe_rewards": -0.3288869559764862,
"step": 220
},
{
"epoch": 0.12,
"learning_rate": 4.991461232516674e-07,
"logits/chosen": -2.403742551803589,
"logits/rejected": -2.2323994636535645,
"logps/chosen": -265.6297607421875,
"logps/rejected": -247.0828399658203,
"loss": 13271.4391,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.35555627942085266,
"rewards/margins": 0.13768498599529266,
"rewards/rejected": -0.49324122071266174,
"rewards/safe_rewards": -0.34540650248527527,
"rewards/unsafe_rewards": -0.36570602655410767,
"step": 230
},
{
"epoch": 0.13,
"learning_rate": 4.98714263060751e-07,
"logits/chosen": -2.5071444511413574,
"logits/rejected": -2.31742525100708,
"logps/chosen": -209.05905151367188,
"logps/rejected": -179.74855041503906,
"loss": 12854.6859,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.2065323293209076,
"rewards/margins": 0.12446670234203339,
"rewards/rejected": -0.33099907636642456,
"rewards/safe_rewards": -0.21961939334869385,
"rewards/unsafe_rewards": -0.19344526529312134,
"step": 240
},
{
"epoch": 0.13,
"learning_rate": 4.98194598705552e-07,
"logits/chosen": -2.460813045501709,
"logits/rejected": -2.3778514862060547,
"logps/chosen": -238.38693237304688,
"logps/rejected": -223.18795776367188,
"loss": 13078.5102,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.48650604486465454,
"rewards/margins": 0.07239247858524323,
"rewards/rejected": -0.5588985681533813,
"rewards/safe_rewards": -0.4960268437862396,
"rewards/unsafe_rewards": -0.47698527574539185,
"step": 250
},
{
"epoch": 0.14,
"learning_rate": 4.975873136443648e-07,
"logits/chosen": -2.5023112297058105,
"logits/rejected": -2.350739002227783,
"logps/chosen": -268.8946228027344,
"logps/rejected": -247.2296600341797,
"loss": 11995.3344,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.3785567581653595,
"rewards/margins": 0.12816599011421204,
"rewards/rejected": -0.5067228078842163,
"rewards/safe_rewards": -0.39122840762138367,
"rewards/unsafe_rewards": -0.36588507890701294,
"step": 260
},
{
"epoch": 0.15,
"learning_rate": 4.968926222684212e-07,
"logits/chosen": -2.4438796043395996,
"logits/rejected": -2.3461410999298096,
"logps/chosen": -229.61386108398438,
"logps/rejected": -223.92630004882812,
"loss": 12123.7047,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.294075608253479,
"rewards/margins": 0.13579340279102325,
"rewards/rejected": -0.42986902594566345,
"rewards/safe_rewards": -0.300296813249588,
"rewards/unsafe_rewards": -0.2878544330596924,
"step": 270
},
{
"epoch": 0.15,
"learning_rate": 4.961107698262044e-07,
"logits/chosen": -2.4015512466430664,
"logits/rejected": -2.263901948928833,
"logps/chosen": -255.80538940429688,
"logps/rejected": -228.9020233154297,
"loss": 12352.2758,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.375516414642334,
"rewards/margins": 0.10626471042633057,
"rewards/rejected": -0.48178109526634216,
"rewards/safe_rewards": -0.37126559019088745,
"rewards/unsafe_rewards": -0.37976714968681335,
"step": 280
},
{
"epoch": 0.16,
"learning_rate": 4.952420323368673e-07,
"logits/chosen": -2.4498801231384277,
"logits/rejected": -2.382075548171997,
"logps/chosen": -239.36386108398438,
"logps/rejected": -241.67861938476562,
"loss": 12562.6453,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.47904032468795776,
"rewards/margins": 0.12548959255218506,
"rewards/rejected": -0.6045299172401428,
"rewards/safe_rewards": -0.4896617829799652,
"rewards/unsafe_rewards": -0.46841883659362793,
"step": 290
},
{
"epoch": 0.16,
"learning_rate": 4.942867164927899e-07,
"logits/chosen": -2.448484182357788,
"logits/rejected": -2.3499300479888916,
"logps/chosen": -233.24447631835938,
"logps/rejected": -213.6923828125,
"loss": 13096.7359,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.3095199167728424,
"rewards/margins": 0.13528046011924744,
"rewards/rejected": -0.44480031728744507,
"rewards/safe_rewards": -0.2981899082660675,
"rewards/unsafe_rewards": -0.3208498954772949,
"step": 300
},
{
"epoch": 0.16,
"eval_logits/chosen": -2.319474458694458,
"eval_logits/rejected": -2.1815290451049805,
"eval_logps/chosen": -170.00274658203125,
"eval_logps/rejected": -140.18304443359375,
"eval_loss": 4529.67333984375,
"eval_rewards/accuracies": 0.6583978533744812,
"eval_rewards/chosen": -0.39572206139564514,
"eval_rewards/margins": 0.08145187050104141,
"eval_rewards/rejected": -0.4771738648414612,
"eval_rewards/safe_rewards": -0.39296460151672363,
"eval_rewards/unsafe_rewards": -0.3956325650215149,
"eval_runtime": 996.6135,
"eval_samples_per_second": 33.156,
"eval_steps_per_second": 1.037,
"step": 300
},
{
"epoch": 0.17,
"learning_rate": 4.932451595513062e-07,
"logits/chosen": -2.464543342590332,
"logits/rejected": -2.3167717456817627,
"logps/chosen": -244.10421752929688,
"logps/rejected": -229.1746368408203,
"loss": 11442.1586,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.25108686089515686,
"rewards/margins": 0.15766267478466034,
"rewards/rejected": -0.408749520778656,
"rewards/safe_rewards": -0.2550010681152344,
"rewards/unsafe_rewards": -0.24717266857624054,
"step": 310
},
{
"epoch": 0.17,
"learning_rate": 4.921177292156419e-07,
"logits/chosen": -2.486952543258667,
"logits/rejected": -2.317573070526123,
"logps/chosen": -234.8188934326172,
"logps/rejected": -228.7853240966797,
"loss": 11535.2578,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.2879149317741394,
"rewards/margins": 0.18317629396915436,
"rewards/rejected": -0.47109121084213257,
"rewards/safe_rewards": -0.2716086506843567,
"rewards/unsafe_rewards": -0.30422115325927734,
"step": 320
},
{
"epoch": 0.18,
"learning_rate": 4.909048235051033e-07,
"logits/chosen": -2.408730983734131,
"logits/rejected": -2.313384532928467,
"logps/chosen": -240.1051483154297,
"logps/rejected": -238.6286163330078,
"loss": 11998.6938,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.4070356488227844,
"rewards/margins": 0.15228822827339172,
"rewards/rejected": -0.559323787689209,
"rewards/safe_rewards": -0.38683491945266724,
"rewards/unsafe_rewards": -0.42723625898361206,
"step": 330
},
{
"epoch": 0.18,
"learning_rate": 4.896068706145631e-07,
"logits/chosen": -2.430915355682373,
"logits/rejected": -2.3069653511047363,
"logps/chosen": -263.82440185546875,
"logps/rejected": -228.1202392578125,
"loss": 12415.2109,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.5314391255378723,
"rewards/margins": 0.14294834434986115,
"rewards/rejected": -0.674387514591217,
"rewards/safe_rewards": -0.5249046087265015,
"rewards/unsafe_rewards": -0.5379736423492432,
"step": 340
},
{
"epoch": 0.19,
"learning_rate": 4.882243287632946e-07,
"logits/chosen": -2.5057172775268555,
"logits/rejected": -2.3827576637268066,
"logps/chosen": -237.1790771484375,
"logps/rejected": -237.27392578125,
"loss": 11917.3391,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.44174814224243164,
"rewards/margins": 0.1188526302576065,
"rewards/rejected": -0.5606008172035217,
"rewards/safe_rewards": -0.4454471170902252,
"rewards/unsafe_rewards": -0.4380492568016052,
"step": 350
},
{
"epoch": 0.19,
"learning_rate": 4.867576860332048e-07,
"logits/chosen": -2.5187079906463623,
"logits/rejected": -2.4128689765930176,
"logps/chosen": -207.39404296875,
"logps/rejected": -220.6639404296875,
"loss": 11619.4062,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.353740930557251,
"rewards/margins": 0.16230328381061554,
"rewards/rejected": -0.5160442590713501,
"rewards/safe_rewards": -0.37073829770088196,
"rewards/unsafe_rewards": -0.33674362301826477,
"step": 360
},
{
"epoch": 0.2,
"learning_rate": 4.85207460196526e-07,
"logits/chosen": -2.5497612953186035,
"logits/rejected": -2.4256086349487305,
"logps/chosen": -251.84036254882812,
"logps/rejected": -246.384033203125,
"loss": 11846.7016,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.47489339113235474,
"rewards/margins": 0.13903862237930298,
"rewards/rejected": -0.6139320135116577,
"rewards/safe_rewards": -0.4880181849002838,
"rewards/unsafe_rewards": -0.46176856756210327,
"step": 370
},
{
"epoch": 0.2,
"learning_rate": 4.835741985330259e-07,
"logits/chosen": -2.5404162406921387,
"logits/rejected": -2.4239704608917236,
"logps/chosen": -246.53939819335938,
"logps/rejected": -233.34353637695312,
"loss": 11351.725,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.45782560110092163,
"rewards/margins": 0.15662749111652374,
"rewards/rejected": -0.614453136920929,
"rewards/safe_rewards": -0.42890676856040955,
"rewards/unsafe_rewards": -0.48674440383911133,
"step": 380
},
{
"epoch": 0.21,
"learning_rate": 4.818584776367992e-07,
"logits/chosen": -2.5068519115448,
"logits/rejected": -2.4303596019744873,
"logps/chosen": -244.3930206298828,
"logps/rejected": -249.14187622070312,
"loss": 11726.7609,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.40478819608688354,
"rewards/margins": 0.16650350391864777,
"rewards/rejected": -0.5712917447090149,
"rewards/safe_rewards": -0.43173927068710327,
"rewards/unsafe_rewards": -0.377837210893631,
"step": 390
},
{
"epoch": 0.22,
"learning_rate": 4.800609032127122e-07,
"logits/chosen": -2.4628491401672363,
"logits/rejected": -2.374331474304199,
"logps/chosen": -254.27822875976562,
"logps/rejected": -228.73330688476562,
"loss": 12046.5695,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.48730865120887756,
"rewards/margins": 0.11220350116491318,
"rewards/rejected": -0.5995121598243713,
"rewards/safe_rewards": -0.4591635763645172,
"rewards/unsafe_rewards": -0.5154536962509155,
"step": 400
},
{
"epoch": 0.22,
"learning_rate": 4.78182109862569e-07,
"logits/chosen": -2.4446866512298584,
"logits/rejected": -2.408660650253296,
"logps/chosen": -230.39208984375,
"logps/rejected": -231.1884765625,
"loss": 12854.2617,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.3896816670894623,
"rewards/margins": 0.12279312312602997,
"rewards/rejected": -0.5124748945236206,
"rewards/safe_rewards": -0.35095858573913574,
"rewards/unsafe_rewards": -0.4284047484397888,
"step": 410
},
{
"epoch": 0.23,
"learning_rate": 4.7622276086107677e-07,
"logits/chosen": -2.4900124073028564,
"logits/rejected": -2.3827693462371826,
"logps/chosen": -251.73007202148438,
"logps/rejected": -243.5518341064453,
"loss": 12053.4797,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.4146080017089844,
"rewards/margins": 0.16079458594322205,
"rewards/rejected": -0.5754026174545288,
"rewards/safe_rewards": -0.4179013669490814,
"rewards/unsafe_rewards": -0.41131457686424255,
"step": 420
},
{
"epoch": 0.23,
"learning_rate": 4.741835479216879e-07,
"logits/chosen": -2.445971727371216,
"logits/rejected": -2.286686420440674,
"logps/chosen": -294.9362487792969,
"logps/rejected": -262.58404541015625,
"loss": 11785.9859,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.48062044382095337,
"rewards/margins": 0.16858412325382233,
"rewards/rejected": -0.6492044925689697,
"rewards/safe_rewards": -0.5107543468475342,
"rewards/unsafe_rewards": -0.450486421585083,
"step": 430
},
{
"epoch": 0.24,
"learning_rate": 4.720651909524036e-07,
"logits/chosen": -2.3577795028686523,
"logits/rejected": -2.2147059440612793,
"logps/chosen": -236.0032196044922,
"logps/rejected": -218.00820922851562,
"loss": 11980.575,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.3994632661342621,
"rewards/margins": 0.1237812489271164,
"rewards/rejected": -0.5232445001602173,
"rewards/safe_rewards": -0.4401687681674957,
"rewards/unsafe_rewards": -0.3587578237056732,
"step": 440
},
{
"epoch": 0.24,
"learning_rate": 4.698684378016222e-07,
"logits/chosen": -2.2142910957336426,
"logits/rejected": -2.0808310508728027,
"logps/chosen": -249.1070556640625,
"logps/rejected": -246.5348663330078,
"loss": 12188.9062,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6178480982780457,
"rewards/margins": 0.13151735067367554,
"rewards/rejected": -0.7493655681610107,
"rewards/safe_rewards": -0.6169500350952148,
"rewards/unsafe_rewards": -0.6187463402748108,
"step": 450
},
{
"epoch": 0.25,
"learning_rate": 4.675940639941256e-07,
"logits/chosen": -2.2552475929260254,
"logits/rejected": -2.087047815322876,
"logps/chosen": -268.5553283691406,
"logps/rejected": -256.51849365234375,
"loss": 11744.7563,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6386964917182922,
"rewards/margins": 0.15222086012363434,
"rewards/rejected": -0.7909173965454102,
"rewards/safe_rewards": -0.6151617765426636,
"rewards/unsafe_rewards": -0.6622311472892761,
"step": 460
},
{
"epoch": 0.25,
"learning_rate": 4.6524287245729286e-07,
"logits/chosen": -2.2173266410827637,
"logits/rejected": -2.0759875774383545,
"logps/chosen": -247.02444458007812,
"logps/rejected": -235.70431518554688,
"loss": 11539.7117,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5241425633430481,
"rewards/margins": 0.1680615395307541,
"rewards/rejected": -0.6922041177749634,
"rewards/safe_rewards": -0.5475858449935913,
"rewards/unsafe_rewards": -0.5006993412971497,
"step": 470
},
{
"epoch": 0.26,
"learning_rate": 4.628156932376418e-07,
"logits/chosen": -2.254505157470703,
"logits/rejected": -2.057737112045288,
"logps/chosen": -252.7125244140625,
"logps/rejected": -227.06204223632812,
"loss": 11626.5016,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5887488126754761,
"rewards/margins": 0.16838543117046356,
"rewards/rejected": -0.7571342587471008,
"rewards/safe_rewards": -0.5993281602859497,
"rewards/unsafe_rewards": -0.5781695246696472,
"step": 480
},
{
"epoch": 0.26,
"learning_rate": 4.603133832077953e-07,
"logits/chosen": -2.2982699871063232,
"logits/rejected": -2.1872057914733887,
"logps/chosen": -293.5793762207031,
"logps/rejected": -291.3560791015625,
"loss": 11510.6367,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.609082043170929,
"rewards/margins": 0.1590423583984375,
"rewards/rejected": -0.7681244015693665,
"rewards/safe_rewards": -0.6237068176269531,
"rewards/unsafe_rewards": -0.5944572687149048,
"step": 490
},
{
"epoch": 0.27,
"learning_rate": 4.5773682576397776e-07,
"logits/chosen": -2.2795727252960205,
"logits/rejected": -2.150864839553833,
"logps/chosen": -247.63088989257812,
"logps/rejected": -234.78408813476562,
"loss": 11965.7203,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.47748714685440063,
"rewards/margins": 0.13509830832481384,
"rewards/rejected": -0.6125854849815369,
"rewards/safe_rewards": -0.48745498061180115,
"rewards/unsafe_rewards": -0.4675193428993225,
"step": 500
},
{
"epoch": 0.27,
"learning_rate": 4.5508693051414774e-07,
"logits/chosen": -2.3894762992858887,
"logits/rejected": -2.2791035175323486,
"logps/chosen": -240.63394165039062,
"logps/rejected": -236.6797332763672,
"loss": 11414.6078,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.4428652822971344,
"rewards/margins": 0.16295991837978363,
"rewards/rejected": -0.6058252453804016,
"rewards/safe_rewards": -0.4437985420227051,
"rewards/unsafe_rewards": -0.44193196296691895,
"step": 510
},
{
"epoch": 0.28,
"learning_rate": 4.52364632956877e-07,
"logits/chosen": -2.376743793487549,
"logits/rejected": -2.2216954231262207,
"logps/chosen": -262.9234313964844,
"logps/rejected": -226.41116333007812,
"loss": 12766.6898,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.4680143892765045,
"rewards/margins": 0.11080136150121689,
"rewards/rejected": -0.578815758228302,
"rewards/safe_rewards": -0.4691530764102936,
"rewards/unsafe_rewards": -0.4668757915496826,
"step": 520
},
{
"epoch": 0.29,
"learning_rate": 4.4957089415108895e-07,
"logits/chosen": -2.3553097248077393,
"logits/rejected": -2.2464723587036133,
"logps/chosen": -242.263671875,
"logps/rejected": -255.6951141357422,
"loss": 11431.9516,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.4732840061187744,
"rewards/margins": 0.19163894653320312,
"rewards/rejected": -0.6649229526519775,
"rewards/safe_rewards": -0.4660876393318176,
"rewards/unsafe_rewards": -0.4804803729057312,
"step": 530
},
{
"epoch": 0.29,
"learning_rate": 4.467067003767745e-07,
"logits/chosen": -2.3562865257263184,
"logits/rejected": -2.181304931640625,
"logps/chosen": -247.70449829101562,
"logps/rejected": -240.8135528564453,
"loss": 12058.2938,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5476623177528381,
"rewards/margins": 0.19659535586833954,
"rewards/rejected": -0.7442576885223389,
"rewards/safe_rewards": -0.5530461072921753,
"rewards/unsafe_rewards": -0.5422784686088562,
"step": 540
},
{
"epoch": 0.3,
"learning_rate": 4.437730627868027e-07,
"logits/chosen": -2.3266587257385254,
"logits/rejected": -2.120724678039551,
"logps/chosen": -231.5970916748047,
"logps/rejected": -221.3756103515625,
"loss": 11200.6187,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5469081997871399,
"rewards/margins": 0.20715579390525818,
"rewards/rejected": -0.7540639638900757,
"rewards/safe_rewards": -0.5482960939407349,
"rewards/unsafe_rewards": -0.5455203652381897,
"step": 550
},
{
"epoch": 0.3,
"learning_rate": 4.4077101704995163e-07,
"logits/chosen": -2.3897128105163574,
"logits/rejected": -2.2368927001953125,
"logps/chosen": -238.13796997070312,
"logps/rejected": -232.872314453125,
"loss": 11633.2375,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.4596262574195862,
"rewards/margins": 0.13524024188518524,
"rewards/rejected": -0.5948664546012878,
"rewards/safe_rewards": -0.4546341300010681,
"rewards/unsafe_rewards": -0.4646182954311371,
"step": 560
},
{
"epoch": 0.31,
"learning_rate": 4.3770162298528356e-07,
"logits/chosen": -2.396641969680786,
"logits/rejected": -2.2687771320343018,
"logps/chosen": -244.2262420654297,
"logps/rejected": -219.45248413085938,
"loss": 12088.3172,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.493887335062027,
"rewards/margins": 0.15973922610282898,
"rewards/rejected": -0.653626561164856,
"rewards/safe_rewards": -0.48037633299827576,
"rewards/unsafe_rewards": -0.5073983073234558,
"step": 570
},
{
"epoch": 0.31,
"learning_rate": 4.3456596418799476e-07,
"logits/chosen": -2.3043763637542725,
"logits/rejected": -2.173457145690918,
"logps/chosen": -257.493896484375,
"logps/rejected": -241.50222778320312,
"loss": 10751.843,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.5369631052017212,
"rewards/margins": 0.15752388536930084,
"rewards/rejected": -0.6944869756698608,
"rewards/safe_rewards": -0.5310293436050415,
"rewards/unsafe_rewards": -0.5428968667984009,
"step": 580
},
{
"epoch": 0.32,
"learning_rate": 4.313651476468715e-07,
"logits/chosen": -2.270984172821045,
"logits/rejected": -2.1114678382873535,
"logps/chosen": -245.8052520751953,
"logps/rejected": -238.4367218017578,
"loss": 11156.9102,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.4919458329677582,
"rewards/margins": 0.16596952080726624,
"rewards/rejected": -0.6579153537750244,
"rewards/safe_rewards": -0.533513605594635,
"rewards/unsafe_rewards": -0.45037803053855896,
"step": 590
},
{
"epoch": 0.32,
"learning_rate": 4.2810030335348693e-07,
"logits/chosen": -2.199920177459717,
"logits/rejected": -2.001626491546631,
"logps/chosen": -263.06634521484375,
"logps/rejected": -236.2901153564453,
"loss": 11584.7875,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.5653510689735413,
"rewards/margins": 0.14779801666736603,
"rewards/rejected": -0.7131490707397461,
"rewards/safe_rewards": -0.5651000142097473,
"rewards/unsafe_rewards": -0.56560218334198,
"step": 600
},
{
"epoch": 0.32,
"eval_logits/chosen": -1.993375301361084,
"eval_logits/rejected": -1.7938207387924194,
"eval_logps/chosen": -211.25747680664062,
"eval_logps/rejected": -180.6571044921875,
"eval_loss": 4406.71337890625,
"eval_rewards/accuracies": 0.6338334679603577,
"eval_rewards/chosen": -0.8082689642906189,
"eval_rewards/margins": 0.07364560663700104,
"eval_rewards/rejected": -0.8819145560264587,
"eval_rewards/safe_rewards": -0.8027816414833069,
"eval_rewards/unsafe_rewards": -0.8050090670585632,
"eval_runtime": 995.4426,
"eval_samples_per_second": 33.195,
"eval_steps_per_second": 1.038,
"step": 600
},
{
"epoch": 0.33,
"learning_rate": 4.2477258390327806e-07,
"logits/chosen": -2.2489686012268066,
"logits/rejected": -2.072410821914673,
"logps/chosen": -241.9006805419922,
"logps/rejected": -251.07376098632812,
"loss": 11304.5898,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.5690335035324097,
"rewards/margins": 0.20418229699134827,
"rewards/rejected": -0.7732157707214355,
"rewards/safe_rewards": -0.5881134271621704,
"rewards/unsafe_rewards": -0.5499535799026489,
"step": 610
},
{
"epoch": 0.33,
"learning_rate": 4.2138316408864197e-07,
"logits/chosen": -2.334721803665161,
"logits/rejected": -2.150754928588867,
"logps/chosen": -246.8868865966797,
"logps/rejected": -237.44033813476562,
"loss": 10365.4461,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.48651427030563354,
"rewards/margins": 0.2631617486476898,
"rewards/rejected": -0.7496760487556458,
"rewards/safe_rewards": -0.4882384240627289,
"rewards/unsafe_rewards": -0.4847901463508606,
"step": 620
},
{
"epoch": 0.34,
"learning_rate": 4.179332404841962e-07,
"logits/chosen": -2.2644031047821045,
"logits/rejected": -2.0903661251068115,
"logps/chosen": -267.8985290527344,
"logps/rejected": -251.5312042236328,
"loss": 10894.2219,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5778486728668213,
"rewards/margins": 0.22166796028614044,
"rewards/rejected": -0.7995165586471558,
"rewards/safe_rewards": -0.5826883316040039,
"rewards/unsafe_rewards": -0.5730089545249939,
"step": 630
},
{
"epoch": 0.34,
"learning_rate": 4.1442403102434954e-07,
"logits/chosen": -2.282378911972046,
"logits/rejected": -2.113847255706787,
"logps/chosen": -270.49737548828125,
"logps/rejected": -250.124755859375,
"loss": 11471.2906,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.5048245787620544,
"rewards/margins": 0.20923128724098206,
"rewards/rejected": -0.7140558958053589,
"rewards/safe_rewards": -0.5091570615768433,
"rewards/unsafe_rewards": -0.5004920959472656,
"step": 640
},
{
"epoch": 0.35,
"learning_rate": 4.108567745733318e-07,
"logits/chosen": -2.283024311065674,
"logits/rejected": -2.1171913146972656,
"logps/chosen": -216.49264526367188,
"logps/rejected": -221.82421875,
"loss": 11842.7375,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.4298115372657776,
"rewards/margins": 0.178109809756279,
"rewards/rejected": -0.6079213619232178,
"rewards/safe_rewards": -0.41675862669944763,
"rewards/unsafe_rewards": -0.4428643584251404,
"step": 650
},
{
"epoch": 0.36,
"learning_rate": 4.0723273048783426e-07,
"logits/chosen": -2.3322463035583496,
"logits/rejected": -2.1912810802459717,
"logps/chosen": -271.88299560546875,
"logps/rejected": -232.814208984375,
"loss": 11859.6484,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.47567135095596313,
"rewards/margins": 0.16224288940429688,
"rewards/rejected": -0.6379141807556152,
"rewards/safe_rewards": -0.4415673315525055,
"rewards/unsafe_rewards": -0.5097752809524536,
"step": 660
},
{
"epoch": 0.36,
"learning_rate": 4.0355317817241697e-07,
"logits/chosen": -2.3272366523742676,
"logits/rejected": -2.1486003398895264,
"logps/chosen": -275.4678039550781,
"logps/rejected": -224.7365264892578,
"loss": 11542.4758,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.465405136346817,
"rewards/margins": 0.16741091012954712,
"rewards/rejected": -0.6328160166740417,
"rewards/safe_rewards": -0.43667951226234436,
"rewards/unsafe_rewards": -0.4941307008266449,
"step": 670
},
{
"epoch": 0.37,
"learning_rate": 3.998194166278367e-07,
"logits/chosen": -2.378793239593506,
"logits/rejected": -2.2765145301818848,
"logps/chosen": -242.05722045898438,
"logps/rejected": -228.02822875976562,
"loss": 12028.9203,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5077196359634399,
"rewards/margins": 0.10326583683490753,
"rewards/rejected": -0.6109854578971863,
"rewards/safe_rewards": -0.519899845123291,
"rewards/unsafe_rewards": -0.4955393671989441,
"step": 680
},
{
"epoch": 0.37,
"learning_rate": 3.9603276399245855e-07,
"logits/chosen": -2.3500876426696777,
"logits/rejected": -2.2054431438446045,
"logps/chosen": -266.26800537109375,
"logps/rejected": -240.6906280517578,
"loss": 11937.7102,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.5368236303329468,
"rewards/margins": 0.19003283977508545,
"rewards/rejected": -0.7268564105033875,
"rewards/safe_rewards": -0.5230244994163513,
"rewards/unsafe_rewards": -0.5506226420402527,
"step": 690
},
{
"epoch": 0.38,
"learning_rate": 3.9219455707691e-07,
"logits/chosen": -2.3385093212127686,
"logits/rejected": -2.194088935852051,
"logps/chosen": -262.49957275390625,
"logps/rejected": -248.57937622070312,
"loss": 11237.9461,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.5866261720657349,
"rewards/margins": 0.18088310956954956,
"rewards/rejected": -0.7675093412399292,
"rewards/safe_rewards": -0.5994827747344971,
"rewards/unsafe_rewards": -0.5737696886062622,
"step": 700
},
{
"epoch": 0.38,
"learning_rate": 3.883061508921439e-07,
"logits/chosen": -2.4109256267547607,
"logits/rejected": -2.309866428375244,
"logps/chosen": -253.4579620361328,
"logps/rejected": -269.43060302734375,
"loss": 11465.2805,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.5475128889083862,
"rewards/margins": 0.13771310448646545,
"rewards/rejected": -0.6852259635925293,
"rewards/safe_rewards": -0.5719602704048157,
"rewards/unsafe_rewards": -0.5230655670166016,
"step": 710
},
{
"epoch": 0.39,
"learning_rate": 3.8436891817107555e-07,
"logits/chosen": -2.3683512210845947,
"logits/rejected": -2.29433012008667,
"logps/chosen": -238.59597778320312,
"logps/rejected": -241.7389678955078,
"loss": 11748.5664,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.481682687997818,
"rewards/margins": 0.18611426651477814,
"rewards/rejected": -0.6677969694137573,
"rewards/safe_rewards": -0.4765700697898865,
"rewards/unsafe_rewards": -0.4867952764034271,
"step": 720
},
{
"epoch": 0.39,
"learning_rate": 3.8038424888396414e-07,
"logits/chosen": -2.3680453300476074,
"logits/rejected": -2.233586072921753,
"logps/chosen": -255.9365692138672,
"logps/rejected": -252.37393188476562,
"loss": 10985.4922,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.5011948347091675,
"rewards/margins": 0.1678631603717804,
"rewards/rejected": -0.6690580248832703,
"rewards/safe_rewards": -0.5107179880142212,
"rewards/unsafe_rewards": -0.49167174100875854,
"step": 730
},
{
"epoch": 0.4,
"learning_rate": 3.763535497477079e-07,
"logits/chosen": -2.376601457595825,
"logits/rejected": -2.2197787761688232,
"logps/chosen": -260.691650390625,
"logps/rejected": -245.3418731689453,
"loss": 11114.0781,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.5256686210632324,
"rewards/margins": 0.21716609597206116,
"rewards/rejected": -0.742834746837616,
"rewards/safe_rewards": -0.5268703103065491,
"rewards/unsafe_rewards": -0.524466872215271,
"step": 740
},
{
"epoch": 0.4,
"learning_rate": 3.7227824372922795e-07,
"logits/chosen": -2.3355045318603516,
"logits/rejected": -2.202324867248535,
"logps/chosen": -244.21817016601562,
"logps/rejected": -234.26766967773438,
"loss": 11000.375,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.5600683093070984,
"rewards/margins": 0.15717127919197083,
"rewards/rejected": -0.7172395586967468,
"rewards/safe_rewards": -0.5515316724777222,
"rewards/unsafe_rewards": -0.5686048269271851,
"step": 750
},
{
"epoch": 0.41,
"learning_rate": 3.681597695431148e-07,
"logits/chosen": -2.351304292678833,
"logits/rejected": -2.2070152759552,
"logps/chosen": -249.6879425048828,
"logps/rejected": -252.49722290039062,
"loss": 11395.6094,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.5104349851608276,
"rewards/margins": 0.22875532507896423,
"rewards/rejected": -0.7391902804374695,
"rewards/safe_rewards": -0.5327340364456177,
"rewards/unsafe_rewards": -0.4881359040737152,
"step": 760
},
{
"epoch": 0.41,
"learning_rate": 3.639995811437159e-07,
"logits/chosen": -2.3204877376556396,
"logits/rejected": -2.2027907371520996,
"logps/chosen": -252.81082153320312,
"logps/rejected": -258.1892395019531,
"loss": 11175.6797,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5397385358810425,
"rewards/margins": 0.19889900088310242,
"rewards/rejected": -0.7386375069618225,
"rewards/safe_rewards": -0.5588012337684631,
"rewards/unsafe_rewards": -0.520675778388977,
"step": 770
},
{
"epoch": 0.42,
"learning_rate": 3.597991472118426e-07,
"logits/chosen": -2.3824949264526367,
"logits/rejected": -2.2542223930358887,
"logps/chosen": -267.04437255859375,
"logps/rejected": -254.86337280273438,
"loss": 11511.8547,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5231243371963501,
"rewards/margins": 0.17550811171531677,
"rewards/rejected": -0.6986324191093445,
"rewards/safe_rewards": -0.5508791208267212,
"rewards/unsafe_rewards": -0.49536967277526855,
"step": 780
},
{
"epoch": 0.43,
"learning_rate": 3.5555995063627836e-07,
"logits/chosen": -2.3838717937469482,
"logits/rejected": -2.264705181121826,
"logps/chosen": -288.45098876953125,
"logps/rejected": -261.7911071777344,
"loss": 11126.725,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5807030200958252,
"rewards/margins": 0.1779719889163971,
"rewards/rejected": -0.7586749792098999,
"rewards/safe_rewards": -0.5821424722671509,
"rewards/unsafe_rewards": -0.5792635679244995,
"step": 790
},
{
"epoch": 0.43,
"learning_rate": 3.512834879902715e-07,
"logits/chosen": -2.325700521469116,
"logits/rejected": -2.20318603515625,
"logps/chosen": -262.4079284667969,
"logps/rejected": -256.8218078613281,
"loss": 11113.3211,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6240984797477722,
"rewards/margins": 0.1907762587070465,
"rewards/rejected": -0.8148747682571411,
"rewards/safe_rewards": -0.6181318759918213,
"rewards/unsafe_rewards": -0.6300650835037231,
"step": 800
},
{
"epoch": 0.44,
"learning_rate": 3.4697126900319616e-07,
"logits/chosen": -2.2967851161956787,
"logits/rejected": -2.1394267082214355,
"logps/chosen": -259.38018798828125,
"logps/rejected": -245.7625732421875,
"loss": 11771.4461,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.5871919393539429,
"rewards/margins": 0.2232244461774826,
"rewards/rejected": -0.8104164004325867,
"rewards/safe_rewards": -0.5566633939743042,
"rewards/unsafe_rewards": -0.6177204251289368,
"step": 810
},
{
"epoch": 0.44,
"learning_rate": 3.426248160275693e-07,
"logits/chosen": -2.33948016166687,
"logits/rejected": -2.2020907402038574,
"logps/chosen": -251.8537139892578,
"logps/rejected": -247.29443359375,
"loss": 11619.0344,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.5438165068626404,
"rewards/margins": 0.17501583695411682,
"rewards/rejected": -0.7188323736190796,
"rewards/safe_rewards": -0.5689524412155151,
"rewards/unsafe_rewards": -0.5186805725097656,
"step": 820
},
{
"epoch": 0.45,
"learning_rate": 3.3824566350161094e-07,
"logits/chosen": -2.3640289306640625,
"logits/rejected": -2.195350408554077,
"logps/chosen": -256.7790222167969,
"logps/rejected": -235.6261444091797,
"loss": 11238.1625,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.4845482409000397,
"rewards/margins": 0.1882091611623764,
"rewards/rejected": -0.6727573871612549,
"rewards/safe_rewards": -0.48699751496315,
"rewards/unsafe_rewards": -0.48209887742996216,
"step": 830
},
{
"epoch": 0.45,
"learning_rate": 3.338353574075381e-07,
"logits/chosen": -2.319687843322754,
"logits/rejected": -2.2427151203155518,
"logps/chosen": -229.2741241455078,
"logps/rejected": -230.6369171142578,
"loss": 13030.1109,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5387877225875854,
"rewards/margins": 0.1425672322511673,
"rewards/rejected": -0.6813548803329468,
"rewards/safe_rewards": -0.5679432153701782,
"rewards/unsafe_rewards": -0.5096321702003479,
"step": 840
},
{
"epoch": 0.46,
"learning_rate": 3.2939545472578314e-07,
"logits/chosen": -2.3902430534362793,
"logits/rejected": -2.1937272548675537,
"logps/chosen": -279.47705078125,
"logps/rejected": -250.4775848388672,
"loss": 11581.1313,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.5328287482261658,
"rewards/margins": 0.13031499087810516,
"rewards/rejected": -0.6631438136100769,
"rewards/safe_rewards": -0.5133041143417358,
"rewards/unsafe_rewards": -0.5523533821105957,
"step": 850
},
{
"epoch": 0.46,
"learning_rate": 3.2492752288532916e-07,
"logits/chosen": -2.325867176055908,
"logits/rejected": -2.1522111892700195,
"logps/chosen": -253.1811981201172,
"logps/rejected": -233.44174194335938,
"loss": 11311.8109,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.4949401319026947,
"rewards/margins": 0.15677447617053986,
"rewards/rejected": -0.6517146825790405,
"rewards/safe_rewards": -0.47377505898475647,
"rewards/unsafe_rewards": -0.5161052942276001,
"step": 860
},
{
"epoch": 0.47,
"learning_rate": 3.204331392103574e-07,
"logits/chosen": -2.337698459625244,
"logits/rejected": -2.1352732181549072,
"logps/chosen": -268.2446594238281,
"logps/rejected": -234.0172882080078,
"loss": 11545.3641,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.5469434261322021,
"rewards/margins": 0.15391257405281067,
"rewards/rejected": -0.7008560299873352,
"rewards/safe_rewards": -0.5454758405685425,
"rewards/unsafe_rewards": -0.5484111905097961,
"step": 870
},
{
"epoch": 0.47,
"learning_rate": 3.159138903634006e-07,
"logits/chosen": -2.1978166103363037,
"logits/rejected": -2.0318470001220703,
"logps/chosen": -280.46319580078125,
"logps/rejected": -256.2702941894531,
"loss": 11248.9852,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.7179843187332153,
"rewards/margins": 0.11917723715305328,
"rewards/rejected": -0.837161660194397,
"rewards/safe_rewards": -0.7153393030166626,
"rewards/unsafe_rewards": -0.7206293940544128,
"step": 880
},
{
"epoch": 0.48,
"learning_rate": 3.1137137178519977e-07,
"logits/chosen": -2.1384453773498535,
"logits/rejected": -2.003512382507324,
"logps/chosen": -253.1010284423828,
"logps/rejected": -262.6402893066406,
"loss": 11450.3414,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7104918956756592,
"rewards/margins": 0.18712057173252106,
"rewards/rejected": -0.897612452507019,
"rewards/safe_rewards": -0.6939632296562195,
"rewards/unsafe_rewards": -0.7270206212997437,
"step": 890
},
{
"epoch": 0.48,
"learning_rate": 3.068071871314626e-07,
"logits/chosen": -2.135709524154663,
"logits/rejected": -2.028604507446289,
"logps/chosen": -242.0713653564453,
"logps/rejected": -236.3900909423828,
"loss": 10862.3484,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6334865689277649,
"rewards/margins": 0.13563254475593567,
"rewards/rejected": -0.7691190838813782,
"rewards/safe_rewards": -0.63458251953125,
"rewards/unsafe_rewards": -0.6323906183242798,
"step": 900
},
{
"epoch": 0.48,
"eval_logits/chosen": -1.9376695156097412,
"eval_logits/rejected": -1.7468173503875732,
"eval_logps/chosen": -218.71400451660156,
"eval_logps/rejected": -187.7608642578125,
"eval_loss": 4377.5634765625,
"eval_rewards/accuracies": 0.6195546984672546,
"eval_rewards/chosen": -0.8828346133232117,
"eval_rewards/margins": 0.07011755555868149,
"eval_rewards/rejected": -0.9529521465301514,
"eval_rewards/safe_rewards": -0.8774688839912415,
"eval_rewards/unsafe_rewards": -0.8777852058410645,
"eval_runtime": 994.9448,
"eval_samples_per_second": 33.212,
"eval_steps_per_second": 1.038,
"step": 900
},
{
"epoch": 0.49,
"learning_rate": 3.022229477067205e-07,
"logits/chosen": -2.1737711429595947,
"logits/rejected": -1.9998136758804321,
"logps/chosen": -283.02313232421875,
"logps/rejected": -255.57656860351562,
"loss": 10507.2695,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6587027311325073,
"rewards/margins": 0.193914994597435,
"rewards/rejected": -0.8526177406311035,
"rewards/safe_rewards": -0.6458637714385986,
"rewards/unsafe_rewards": -0.6715416312217712,
"step": 910
},
{
"epoch": 0.49,
"learning_rate": 2.976202718954869e-07,
"logits/chosen": -2.183156728744507,
"logits/rejected": -1.9853588342666626,
"logps/chosen": -282.3530578613281,
"logps/rejected": -271.27069091796875,
"loss": 11861.3852,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.7082624435424805,
"rewards/margins": 0.18091826140880585,
"rewards/rejected": -0.8891807794570923,
"rewards/safe_rewards": -0.7145110368728638,
"rewards/unsafe_rewards": -0.7020138502120972,
"step": 920
},
{
"epoch": 0.5,
"learning_rate": 2.930007845909146e-07,
"logits/chosen": -2.177462339401245,
"logits/rejected": -2.0282797813415527,
"logps/chosen": -268.3103942871094,
"logps/rejected": -266.30548095703125,
"loss": 11683.3781,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.7064036726951599,
"rewards/margins": 0.13596151769161224,
"rewards/rejected": -0.8423651456832886,
"rewards/safe_rewards": -0.7049628496170044,
"rewards/unsafe_rewards": -0.7078445553779602,
"step": 930
},
{
"epoch": 0.51,
"learning_rate": 2.8836611662115634e-07,
"logits/chosen": -2.1535885334014893,
"logits/rejected": -1.9535331726074219,
"logps/chosen": -287.9471130371094,
"logps/rejected": -254.46157836914062,
"loss": 12032.6703,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.6779440641403198,
"rewards/margins": 0.1851780265569687,
"rewards/rejected": -0.8631221652030945,
"rewards/safe_rewards": -0.6615537405014038,
"rewards/unsafe_rewards": -0.6943344473838806,
"step": 940
},
{
"epoch": 0.51,
"learning_rate": 2.8371790417362986e-07,
"logits/chosen": -2.1404051780700684,
"logits/rejected": -1.9921302795410156,
"logps/chosen": -250.58895874023438,
"logps/rejected": -252.75021362304688,
"loss": 12575.8664,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6325485110282898,
"rewards/margins": 0.13314202427864075,
"rewards/rejected": -0.7656905651092529,
"rewards/safe_rewards": -0.615986168384552,
"rewards/unsafe_rewards": -0.6491108536720276,
"step": 950
},
{
"epoch": 0.52,
"learning_rate": 2.7905778821739056e-07,
"logits/chosen": -2.1147265434265137,
"logits/rejected": -1.9771724939346313,
"logps/chosen": -253.51351928710938,
"logps/rejected": -233.7639923095703,
"loss": 11670.4188,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5872529745101929,
"rewards/margins": 0.1395425647497177,
"rewards/rejected": -0.7267955541610718,
"rewards/safe_rewards": -0.5655093193054199,
"rewards/unsafe_rewards": -0.6089966297149658,
"step": 960
},
{
"epoch": 0.52,
"learning_rate": 2.74387413923817e-07,
"logits/chosen": -2.056212902069092,
"logits/rejected": -1.9303762912750244,
"logps/chosen": -294.0152893066406,
"logps/rejected": -272.24615478515625,
"loss": 11716.1664,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6957898736000061,
"rewards/margins": 0.17264129221439362,
"rewards/rejected": -0.8684310913085938,
"rewards/safe_rewards": -0.688141405582428,
"rewards/unsafe_rewards": -0.7034383416175842,
"step": 970
},
{
"epoch": 0.53,
"learning_rate": 2.69708430085812e-07,
"logits/chosen": -2.0949666500091553,
"logits/rejected": -1.8358476161956787,
"logps/chosen": -303.7754821777344,
"logps/rejected": -279.3113708496094,
"loss": 12070.975,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.8039963841438293,
"rewards/margins": 0.2226048707962036,
"rewards/rejected": -1.0266011953353882,
"rewards/safe_rewards": -0.7967602014541626,
"rewards/unsafe_rewards": -0.8112322688102722,
"step": 980
},
{
"epoch": 0.53,
"learning_rate": 2.6502248853572504e-07,
"logits/chosen": -2.018751859664917,
"logits/rejected": -1.81708562374115,
"logps/chosen": -268.05535888671875,
"logps/rejected": -263.68487548828125,
"loss": 12117.2781,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.8013098835945129,
"rewards/margins": 0.21267572045326233,
"rewards/rejected": -1.013985514640808,
"rewards/safe_rewards": -0.7930777668952942,
"rewards/unsafe_rewards": -0.8095420002937317,
"step": 990
},
{
"epoch": 0.54,
"learning_rate": 2.6033124356220325e-07,
"logits/chosen": -2.040283441543579,
"logits/rejected": -1.8361635208129883,
"logps/chosen": -266.68841552734375,
"logps/rejected": -244.3878631591797,
"loss": 11242.6422,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.639050304889679,
"rewards/margins": 0.21101757884025574,
"rewards/rejected": -0.8500677943229675,
"rewards/safe_rewards": -0.6206936836242676,
"rewards/unsafe_rewards": -0.6574069261550903,
"step": 1000
},
{
"epoch": 0.54,
"learning_rate": 2.55636351326173e-07,
"logits/chosen": -2.074812412261963,
"logits/rejected": -1.9070053100585938,
"logps/chosen": -267.3968505859375,
"logps/rejected": -242.2158660888672,
"loss": 10413.1977,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.6132654547691345,
"rewards/margins": 0.21324153244495392,
"rewards/rejected": -0.8265069723129272,
"rewards/safe_rewards": -0.638529360294342,
"rewards/unsafe_rewards": -0.588001549243927,
"step": 1010
},
{
"epoch": 0.55,
"learning_rate": 2.509394692761622e-07,
"logits/chosen": -2.1109747886657715,
"logits/rejected": -1.8711456060409546,
"logps/chosen": -266.0138244628906,
"logps/rejected": -246.04159545898438,
"loss": 11730.4734,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6028767824172974,
"rewards/margins": 0.20766966044902802,
"rewards/rejected": -0.8105465173721313,
"rewards/safe_rewards": -0.5901221632957458,
"rewards/unsafe_rewards": -0.6156314611434937,
"step": 1020
},
{
"epoch": 0.55,
"learning_rate": 2.462422555631674e-07,
"logits/chosen": -2.073171854019165,
"logits/rejected": -1.8314094543457031,
"logps/chosen": -267.2673034667969,
"logps/rejected": -234.9337921142578,
"loss": 11384.8359,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6479398012161255,
"rewards/margins": 0.20152795314788818,
"rewards/rejected": -0.8494676351547241,
"rewards/safe_rewards": -0.6472254991531372,
"rewards/unsafe_rewards": -0.6486541032791138,
"step": 1030
},
{
"epoch": 0.56,
"learning_rate": 2.415463684552728e-07,
"logits/chosen": -2.0377871990203857,
"logits/rejected": -1.8571240901947021,
"logps/chosen": -264.3353576660156,
"logps/rejected": -250.5574493408203,
"loss": 11869.4391,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6739367246627808,
"rewards/margins": 0.1584298014640808,
"rewards/rejected": -0.8323665857315063,
"rewards/safe_rewards": -0.6686100363731384,
"rewards/unsafe_rewards": -0.6792632341384888,
"step": 1040
},
{
"epoch": 0.56,
"learning_rate": 2.3685346575222807e-07,
"logits/chosen": -2.111082077026367,
"logits/rejected": -1.8854366540908813,
"logps/chosen": -280.7798767089844,
"logps/rejected": -254.7810821533203,
"loss": 11278.0977,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6307443976402283,
"rewards/margins": 0.16730864346027374,
"rewards/rejected": -0.7980531454086304,
"rewards/safe_rewards": -0.6317691802978516,
"rewards/unsafe_rewards": -0.6297196745872498,
"step": 1050
},
{
"epoch": 0.57,
"learning_rate": 2.321652042001919e-07,
"logits/chosen": -2.124929189682007,
"logits/rejected": -1.8838306665420532,
"logps/chosen": -272.2227783203125,
"logps/rejected": -273.8869323730469,
"loss": 11306.2313,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6101773381233215,
"rewards/margins": 0.20483005046844482,
"rewards/rejected": -0.8150073885917664,
"rewards/safe_rewards": -0.6046205759048462,
"rewards/unsafe_rewards": -0.6157340407371521,
"step": 1060
},
{
"epoch": 0.58,
"learning_rate": 2.2748323890684662e-07,
"logits/chosen": -2.1761631965637207,
"logits/rejected": -1.8986377716064453,
"logps/chosen": -270.1111145019531,
"logps/rejected": -247.43576049804688,
"loss": 11255.1672,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.5767061710357666,
"rewards/margins": 0.2480432540178299,
"rewards/rejected": -0.8247492909431458,
"rewards/safe_rewards": -0.5869646072387695,
"rewards/unsafe_rewards": -0.5664476156234741,
"step": 1070
},
{
"epoch": 0.58,
"learning_rate": 2.2280922275709213e-07,
"logits/chosen": -2.1585605144500732,
"logits/rejected": -2.006892681121826,
"logps/chosen": -272.51593017578125,
"logps/rejected": -259.193603515625,
"loss": 11610.1609,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6135168671607971,
"rewards/margins": 0.16986827552318573,
"rewards/rejected": -0.783385157585144,
"rewards/safe_rewards": -0.5932785272598267,
"rewards/unsafe_rewards": -0.6337552666664124,
"step": 1080
},
{
"epoch": 0.59,
"learning_rate": 2.1814480582952375e-07,
"logits/chosen": -2.176574230194092,
"logits/rejected": -2.0007762908935547,
"logps/chosen": -264.99627685546875,
"logps/rejected": -260.7152099609375,
"loss": 11244.9641,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6294008493423462,
"rewards/margins": 0.1786467581987381,
"rewards/rejected": -0.8080476522445679,
"rewards/safe_rewards": -0.610293984413147,
"rewards/unsafe_rewards": -0.6485077738761902,
"step": 1090
},
{
"epoch": 0.59,
"learning_rate": 2.1349163481390187e-07,
"logits/chosen": -2.1403450965881348,
"logits/rejected": -1.9742145538330078,
"logps/chosen": -258.41851806640625,
"logps/rejected": -253.95596313476562,
"loss": 11432.9805,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.579275369644165,
"rewards/margins": 0.1978916972875595,
"rewards/rejected": -0.7771671414375305,
"rewards/safe_rewards": -0.5507979393005371,
"rewards/unsafe_rewards": -0.6077528595924377,
"step": 1100
},
{
"epoch": 0.6,
"learning_rate": 2.0885135242981647e-07,
"logits/chosen": -2.110644817352295,
"logits/rejected": -1.8921115398406982,
"logps/chosen": -295.12310791015625,
"logps/rejected": -242.7633056640625,
"loss": 10866.6008,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6292051672935486,
"rewards/margins": 0.1930016279220581,
"rewards/rejected": -0.8222068548202515,
"rewards/safe_rewards": -0.6636418104171753,
"rewards/unsafe_rewards": -0.5947686433792114,
"step": 1110
},
{
"epoch": 0.6,
"learning_rate": 2.0422559684675494e-07,
"logits/chosen": -2.07668137550354,
"logits/rejected": -1.870774269104004,
"logps/chosen": -279.4585876464844,
"logps/rejected": -252.98422241210938,
"loss": 11123.7453,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.622269868850708,
"rewards/margins": 0.18196968734264374,
"rewards/rejected": -0.8042396306991577,
"rewards/safe_rewards": -0.6149351596832275,
"rewards/unsafe_rewards": -0.629604697227478,
"step": 1120
},
{
"epoch": 0.61,
"learning_rate": 1.9961600110577457e-07,
"logits/chosen": -2.0477161407470703,
"logits/rejected": -1.8454376459121704,
"logps/chosen": -281.2010803222656,
"logps/rejected": -276.59503173828125,
"loss": 11451.4273,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.7229688763618469,
"rewards/margins": 0.15784484148025513,
"rewards/rejected": -0.880813717842102,
"rewards/safe_rewards": -0.6920644640922546,
"rewards/unsafe_rewards": -0.7538732290267944,
"step": 1130
},
{
"epoch": 0.61,
"learning_rate": 1.950241925429867e-07,
"logits/chosen": -2.1420865058898926,
"logits/rejected": -1.9303150177001953,
"logps/chosen": -264.2613830566406,
"logps/rejected": -247.3074493408203,
"loss": 11185.8914,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6415218114852905,
"rewards/margins": 0.21913623809814453,
"rewards/rejected": -0.8606580495834351,
"rewards/safe_rewards": -0.6352616548538208,
"rewards/unsafe_rewards": -0.647782027721405,
"step": 1140
},
{
"epoch": 0.62,
"learning_rate": 1.9045179221505495e-07,
"logits/chosen": -2.1051979064941406,
"logits/rejected": -1.9602015018463135,
"logps/chosen": -292.8426513671875,
"logps/rejected": -272.5882568359375,
"loss": 10919.0359,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.6359490156173706,
"rewards/margins": 0.21196213364601135,
"rewards/rejected": -0.8479111790657043,
"rewards/safe_rewards": -0.605754554271698,
"rewards/unsafe_rewards": -0.666143536567688,
"step": 1150
},
{
"epoch": 0.62,
"learning_rate": 1.8590041432690893e-07,
"logits/chosen": -2.0412395000457764,
"logits/rejected": -1.9340324401855469,
"logps/chosen": -256.9598693847656,
"logps/rejected": -250.4047393798828,
"loss": 11277.7938,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.6930029988288879,
"rewards/margins": 0.13424697518348694,
"rewards/rejected": -0.8272498846054077,
"rewards/safe_rewards": -0.6941550970077515,
"rewards/unsafe_rewards": -0.6918508410453796,
"step": 1160
},
{
"epoch": 0.63,
"learning_rate": 1.813716656618788e-07,
"logits/chosen": -2.0102741718292236,
"logits/rejected": -1.9039026498794556,
"logps/chosen": -259.2917175292969,
"logps/rejected": -252.873046875,
"loss": 11988.2266,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7098476886749268,
"rewards/margins": 0.1709776222705841,
"rewards/rejected": -0.8808252215385437,
"rewards/safe_rewards": -0.7011247873306274,
"rewards/unsafe_rewards": -0.7185705304145813,
"step": 1170
},
{
"epoch": 0.63,
"learning_rate": 1.7686714501444788e-07,
"logits/chosen": -2.084367275238037,
"logits/rejected": -1.8202540874481201,
"logps/chosen": -288.35626220703125,
"logps/rejected": -254.33651733398438,
"loss": 11612.3062,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.7599023580551147,
"rewards/margins": 0.16640828549861908,
"rewards/rejected": -0.926310658454895,
"rewards/safe_rewards": -0.7765257954597473,
"rewards/unsafe_rewards": -0.7432790994644165,
"step": 1180
},
{
"epoch": 0.64,
"learning_rate": 1.7238844262582768e-07,
"logits/chosen": -2.052704095840454,
"logits/rejected": -1.9864280223846436,
"logps/chosen": -275.9404602050781,
"logps/rejected": -276.5259094238281,
"loss": 10682.7672,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.6891142129898071,
"rewards/margins": 0.15954537689685822,
"rewards/rejected": -0.8486596345901489,
"rewards/safe_rewards": -0.6521228551864624,
"rewards/unsafe_rewards": -0.7261057496070862,
"step": 1190
},
{
"epoch": 0.65,
"learning_rate": 1.679371396225504e-07,
"logits/chosen": -2.0941474437713623,
"logits/rejected": -1.8897063732147217,
"logps/chosen": -271.2891540527344,
"logps/rejected": -270.78924560546875,
"loss": 11671.4219,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7015270590782166,
"rewards/margins": 0.19374321401119232,
"rewards/rejected": -0.8952702283859253,
"rewards/safe_rewards": -0.6884527802467346,
"rewards/unsafe_rewards": -0.7146013379096985,
"step": 1200
},
{
"epoch": 0.65,
"eval_logits/chosen": -1.8665262460708618,
"eval_logits/rejected": -1.6740189790725708,
"eval_logps/chosen": -228.5369415283203,
"eval_logps/rejected": -197.5587921142578,
"eval_loss": 4346.4052734375,
"eval_rewards/accuracies": 0.6158034801483154,
"eval_rewards/chosen": -0.9810636043548584,
"eval_rewards/margins": 0.06986771523952484,
"eval_rewards/rejected": -1.0509313344955444,
"eval_rewards/safe_rewards": -0.9763943552970886,
"eval_rewards/unsafe_rewards": -0.9767957329750061,
"eval_runtime": 992.8374,
"eval_samples_per_second": 33.282,
"eval_steps_per_second": 1.04,
"step": 1200
},
{
"epoch": 0.65,
"learning_rate": 1.6351480745828096e-07,
"logits/chosen": -2.1039538383483887,
"logits/rejected": -1.9394419193267822,
"logps/chosen": -265.77984619140625,
"logps/rejected": -253.25137329101562,
"loss": 10276.6594,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6483036279678345,
"rewards/margins": 0.1812000572681427,
"rewards/rejected": -0.8295037150382996,
"rewards/safe_rewards": -0.6423134803771973,
"rewards/unsafe_rewards": -0.6542937755584717,
"step": 1210
},
{
"epoch": 0.66,
"learning_rate": 1.5912300735904248e-07,
"logits/chosen": -2.1374449729919434,
"logits/rejected": -1.983070969581604,
"logps/chosen": -290.1751403808594,
"logps/rejected": -259.19097900390625,
"loss": 11657.7609,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6737682223320007,
"rewards/margins": 0.14703810214996338,
"rewards/rejected": -0.8208063244819641,
"rewards/safe_rewards": -0.6786805987358093,
"rewards/unsafe_rewards": -0.6688558459281921,
"step": 1220
},
{
"epoch": 0.66,
"learning_rate": 1.5476328977205395e-07,
"logits/chosen": -2.1445086002349854,
"logits/rejected": -1.9767332077026367,
"logps/chosen": -271.8307189941406,
"logps/rejected": -251.6882781982422,
"loss": 11008.7156,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.6544319987297058,
"rewards/margins": 0.21836614608764648,
"rewards/rejected": -0.8727981448173523,
"rewards/safe_rewards": -0.6966055631637573,
"rewards/unsafe_rewards": -0.6122584342956543,
"step": 1230
},
{
"epoch": 0.67,
"learning_rate": 1.5043719381837112e-07,
"logits/chosen": -2.08951473236084,
"logits/rejected": -1.9547208547592163,
"logps/chosen": -287.0835266113281,
"logps/rejected": -271.43670654296875,
"loss": 11362.9422,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6643306612968445,
"rewards/margins": 0.17835856974124908,
"rewards/rejected": -0.8426891565322876,
"rewards/safe_rewards": -0.6574736833572388,
"rewards/unsafe_rewards": -0.6711875200271606,
"step": 1240
},
{
"epoch": 0.67,
"learning_rate": 1.461462467495284e-07,
"logits/chosen": -2.0471625328063965,
"logits/rejected": -1.8923746347427368,
"logps/chosen": -252.0188446044922,
"logps/rejected": -260.54522705078125,
"loss": 10503.6422,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7411776185035706,
"rewards/margins": 0.21499836444854736,
"rewards/rejected": -0.9561759829521179,
"rewards/safe_rewards": -0.7554097175598145,
"rewards/unsafe_rewards": -0.7269455194473267,
"step": 1250
},
{
"epoch": 0.68,
"learning_rate": 1.4189196340836865e-07,
"logits/chosen": -2.1763851642608643,
"logits/rejected": -1.9493157863616943,
"logps/chosen": -269.4758605957031,
"logps/rejected": -258.123291015625,
"loss": 10853.5781,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.7069443464279175,
"rewards/margins": 0.17843110859394073,
"rewards/rejected": -0.8853754997253418,
"rewards/safe_rewards": -0.7389132380485535,
"rewards/unsafe_rewards": -0.6749754548072815,
"step": 1260
},
{
"epoch": 0.68,
"learning_rate": 1.3767584569425561e-07,
"logits/chosen": -2.19474458694458,
"logits/rejected": -1.9729722738265991,
"logps/chosen": -277.60992431640625,
"logps/rejected": -258.0458984375,
"loss": 11364.5312,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6931368708610535,
"rewards/margins": 0.18750238418579102,
"rewards/rejected": -0.8806392550468445,
"rewards/safe_rewards": -0.7000595927238464,
"rewards/unsafe_rewards": -0.6862143278121948,
"step": 1270
},
{
"epoch": 0.69,
"learning_rate": 1.334993820328541e-07,
"logits/chosen": -2.0907111167907715,
"logits/rejected": -1.9239038228988647,
"logps/chosen": -246.83642578125,
"logps/rejected": -245.0505828857422,
"loss": 10652.2328,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.7036882638931274,
"rewards/margins": 0.24492530524730682,
"rewards/rejected": -0.9486135244369507,
"rewards/safe_rewards": -0.6736655235290527,
"rewards/unsafe_rewards": -0.7337108850479126,
"step": 1280
},
{
"epoch": 0.69,
"learning_rate": 1.2936404685066852e-07,
"logits/chosen": -2.0505154132843018,
"logits/rejected": -1.9092705249786377,
"logps/chosen": -286.58990478515625,
"logps/rejected": -280.3271789550781,
"loss": 11526.9,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.68525230884552,
"rewards/margins": 0.18679992854595184,
"rewards/rejected": -0.8720521926879883,
"rewards/safe_rewards": -0.7125786542892456,
"rewards/unsafe_rewards": -0.6579257249832153,
"step": 1290
},
{
"epoch": 0.7,
"learning_rate": 1.252713000545221e-07,
"logits/chosen": -2.2164149284362793,
"logits/rejected": -2.019528388977051,
"logps/chosen": -282.0648498535156,
"logps/rejected": -258.7922058105469,
"loss": 10187.7117,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.6330300569534302,
"rewards/margins": 0.23740728199481964,
"rewards/rejected": -0.8704373240470886,
"rewards/safe_rewards": -0.6281386613845825,
"rewards/unsafe_rewards": -0.6379214525222778,
"step": 1300
},
{
"epoch": 0.7,
"learning_rate": 1.2122258651616304e-07,
"logits/chosen": -2.183960437774658,
"logits/rejected": -1.976144790649414,
"logps/chosen": -275.6617126464844,
"logps/rejected": -244.86434936523438,
"loss": 10813.9688,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6538507342338562,
"rewards/margins": 0.17538292706012726,
"rewards/rejected": -0.8292337656021118,
"rewards/safe_rewards": -0.6337584853172302,
"rewards/unsafe_rewards": -0.673943042755127,
"step": 1310
},
{
"epoch": 0.71,
"learning_rate": 1.1721933556217792e-07,
"logits/chosen": -2.167842388153076,
"logits/rejected": -2.0031533241271973,
"logps/chosen": -265.863525390625,
"logps/rejected": -261.1900329589844,
"loss": 11171.5523,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6531133651733398,
"rewards/margins": 0.19861795008182526,
"rewards/rejected": -0.8517313003540039,
"rewards/safe_rewards": -0.6670490503311157,
"rewards/unsafe_rewards": -0.639177680015564,
"step": 1320
},
{
"epoch": 0.72,
"learning_rate": 1.1326296046939333e-07,
"logits/chosen": -2.134734869003296,
"logits/rejected": -1.9698076248168945,
"logps/chosen": -257.0758972167969,
"logps/rejected": -242.34640502929688,
"loss": 10608.8219,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6214333772659302,
"rewards/margins": 0.21942774951457977,
"rewards/rejected": -0.8408611416816711,
"rewards/safe_rewards": -0.6008444428443909,
"rewards/unsafe_rewards": -0.6420222520828247,
"step": 1330
},
{
"epoch": 0.72,
"learning_rate": 1.0935485796594351e-07,
"logits/chosen": -2.174132823944092,
"logits/rejected": -1.9792499542236328,
"logps/chosen": -291.65814208984375,
"logps/rejected": -260.0698547363281,
"loss": 11895.8789,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.6717311143875122,
"rewards/margins": 0.22725185751914978,
"rewards/rejected": -0.8989830017089844,
"rewards/safe_rewards": -0.6783491373062134,
"rewards/unsafe_rewards": -0.665113091468811,
"step": 1340
},
{
"epoch": 0.73,
"learning_rate": 1.0549640773818028e-07,
"logits/chosen": -2.139070749282837,
"logits/rejected": -2.040194034576416,
"logps/chosen": -272.8217468261719,
"logps/rejected": -252.80606079101562,
"loss": 11776.8328,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.724553108215332,
"rewards/margins": 0.14907190203666687,
"rewards/rejected": -0.8736250996589661,
"rewards/safe_rewards": -0.764702320098877,
"rewards/unsafe_rewards": -0.6844038963317871,
"step": 1350
},
{
"epoch": 0.73,
"learning_rate": 1.0168897194359921e-07,
"logits/chosen": -2.1789050102233887,
"logits/rejected": -1.9954169988632202,
"logps/chosen": -300.97711181640625,
"logps/rejected": -274.7542724609375,
"loss": 10877.8953,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.7322704195976257,
"rewards/margins": 0.15806125104427338,
"rewards/rejected": -0.8903317451477051,
"rewards/safe_rewards": -0.6902467608451843,
"rewards/unsafe_rewards": -0.7742940783500671,
"step": 1360
},
{
"epoch": 0.74,
"learning_rate": 9.793389472995392e-08,
"logits/chosen": -2.155348539352417,
"logits/rejected": -1.9421207904815674,
"logps/chosen": -279.4602966308594,
"logps/rejected": -240.51791381835938,
"loss": 10089.3242,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.639030933380127,
"rewards/margins": 0.252288281917572,
"rewards/rejected": -0.8913192749023438,
"rewards/safe_rewards": -0.6204690933227539,
"rewards/unsafe_rewards": -0.6575928330421448,
"step": 1370
},
{
"epoch": 0.74,
"learning_rate": 9.423250176072874e-08,
"logits/chosen": -2.1420297622680664,
"logits/rejected": -1.946412444114685,
"logps/chosen": -260.1219177246094,
"logps/rejected": -234.1204833984375,
"loss": 12708.7586,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.71236252784729,
"rewards/margins": 0.1536431610584259,
"rewards/rejected": -0.8660055994987488,
"rewards/safe_rewards": -0.6938896179199219,
"rewards/unsafe_rewards": -0.7308354377746582,
"step": 1380
},
{
"epoch": 0.75,
"learning_rate": 9.058609974713654e-08,
"logits/chosen": -2.1647183895111084,
"logits/rejected": -1.9520107507705688,
"logps/chosen": -270.0279235839844,
"logps/rejected": -260.9911193847656,
"loss": 10654.6578,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6530768275260925,
"rewards/margins": 0.2193538248538971,
"rewards/rejected": -0.872430682182312,
"rewards/safe_rewards": -0.6400060653686523,
"rewards/unsafe_rewards": -0.6661475896835327,
"step": 1390
},
{
"epoch": 0.75,
"learning_rate": 8.699597598680753e-08,
"logits/chosen": -2.1252872943878174,
"logits/rejected": -1.9515492916107178,
"logps/chosen": -257.53680419921875,
"logps/rejected": -247.7872314453125,
"loss": 10438.4352,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6546493768692017,
"rewards/margins": 0.19867801666259766,
"rewards/rejected": -0.8533273935317993,
"rewards/safe_rewards": -0.6472882032394409,
"rewards/unsafe_rewards": -0.6620105504989624,
"step": 1400
},
{
"epoch": 0.76,
"learning_rate": 8.346339790933166e-08,
"logits/chosen": -2.163567543029785,
"logits/rejected": -1.9771531820297241,
"logps/chosen": -261.906982421875,
"logps/rejected": -245.43344116210938,
"loss": 11502.7297,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6902798414230347,
"rewards/margins": 0.19094708561897278,
"rewards/rejected": -0.8812268376350403,
"rewards/safe_rewards": -0.6901260018348694,
"rewards/unsafe_rewards": -0.6904336214065552,
"step": 1410
},
{
"epoch": 0.76,
"learning_rate": 7.998961262881506e-08,
"logits/chosen": -2.136033535003662,
"logits/rejected": -1.915771245956421,
"logps/chosen": -282.3092346191406,
"logps/rejected": -245.1779022216797,
"loss": 10907.2641,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.6517400741577148,
"rewards/margins": 0.2158946692943573,
"rewards/rejected": -0.8676347732543945,
"rewards/safe_rewards": -0.6872564554214478,
"rewards/unsafe_rewards": -0.6162236928939819,
"step": 1420
},
{
"epoch": 0.77,
"learning_rate": 7.657584650360846e-08,
"logits/chosen": -2.068126678466797,
"logits/rejected": -1.9587552547454834,
"logps/chosen": -251.7869873046875,
"logps/rejected": -236.9489288330078,
"loss": 11409.3359,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6544846296310425,
"rewards/margins": 0.2086174488067627,
"rewards/rejected": -0.8631020784378052,
"rewards/safe_rewards": -0.6824710965156555,
"rewards/unsafe_rewards": -0.6264981627464294,
"step": 1430
},
{
"epoch": 0.77,
"learning_rate": 7.322330470336313e-08,
"logits/chosen": -2.122067928314209,
"logits/rejected": -1.9209505319595337,
"logps/chosen": -269.77520751953125,
"logps/rejected": -268.40972900390625,
"loss": 10821.843,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6080222129821777,
"rewards/margins": 0.25041937828063965,
"rewards/rejected": -0.8584416508674622,
"rewards/safe_rewards": -0.5671194195747375,
"rewards/unsafe_rewards": -0.6489250659942627,
"step": 1440
},
{
"epoch": 0.78,
"learning_rate": 6.993317078356709e-08,
"logits/chosen": -2.1524910926818848,
"logits/rejected": -2.056199789047241,
"logps/chosen": -279.29583740234375,
"logps/rejected": -255.84521484375,
"loss": 11391.4539,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.691224217414856,
"rewards/margins": 0.13461162149906158,
"rewards/rejected": -0.8258358240127563,
"rewards/safe_rewards": -0.7272582054138184,
"rewards/unsafe_rewards": -0.6551901698112488,
"step": 1450
},
{
"epoch": 0.79,
"learning_rate": 6.67066062677118e-08,
"logits/chosen": -2.191707134246826,
"logits/rejected": -2.0195794105529785,
"logps/chosen": -263.27642822265625,
"logps/rejected": -235.50973510742188,
"loss": 11926.3453,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6236655116081238,
"rewards/margins": 0.16336306929588318,
"rewards/rejected": -0.7870286107063293,
"rewards/safe_rewards": -0.6308915615081787,
"rewards/unsafe_rewards": -0.6164394617080688,
"step": 1460
},
{
"epoch": 0.79,
"learning_rate": 6.354475023723685e-08,
"logits/chosen": -2.1155574321746826,
"logits/rejected": -1.9720067977905273,
"logps/chosen": -301.7005920410156,
"logps/rejected": -267.5169372558594,
"loss": 11094.9625,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6763675808906555,
"rewards/margins": 0.23334193229675293,
"rewards/rejected": -0.9097094535827637,
"rewards/safe_rewards": -0.6679859757423401,
"rewards/unsafe_rewards": -0.6847492456436157,
"step": 1470
},
{
"epoch": 0.8,
"learning_rate": 6.044871892939746e-08,
"logits/chosen": -2.184211254119873,
"logits/rejected": -1.9999923706054688,
"logps/chosen": -281.41900634765625,
"logps/rejected": -268.82464599609375,
"loss": 10924.2703,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.6950246691703796,
"rewards/margins": 0.19827482104301453,
"rewards/rejected": -0.8932995796203613,
"rewards/safe_rewards": -0.6960557699203491,
"rewards/unsafe_rewards": -0.6939936280250549,
"step": 1480
},
{
"epoch": 0.8,
"learning_rate": 5.741960534319676e-08,
"logits/chosen": -2.134298801422119,
"logits/rejected": -2.02238130569458,
"logps/chosen": -241.8282012939453,
"logps/rejected": -240.76290893554688,
"loss": 11064.8711,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.687341570854187,
"rewards/margins": 0.16167931258678436,
"rewards/rejected": -0.8490209579467773,
"rewards/safe_rewards": -0.7233562469482422,
"rewards/unsafe_rewards": -0.6513269543647766,
"step": 1490
},
{
"epoch": 0.81,
"learning_rate": 5.44584788535217e-08,
"logits/chosen": -2.1548972129821777,
"logits/rejected": -2.0020337104797363,
"logps/chosen": -288.1227111816406,
"logps/rejected": -265.52337646484375,
"loss": 10202.4125,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.7084766626358032,
"rewards/margins": 0.21978096663951874,
"rewards/rejected": -0.9282576441764832,
"rewards/safe_rewards": -0.6754325032234192,
"rewards/unsafe_rewards": -0.741520881652832,
"step": 1500
},
{
"epoch": 0.81,
"eval_logits/chosen": -1.9419655799865723,
"eval_logits/rejected": -1.764477252960205,
"eval_logps/chosen": -226.97747802734375,
"eval_logps/rejected": -195.17942810058594,
"eval_loss": 4320.98779296875,
"eval_rewards/accuracies": 0.6022507548332214,
"eval_rewards/chosen": -0.9654689431190491,
"eval_rewards/margins": 0.061668772250413895,
"eval_rewards/rejected": -1.0271376371383667,
"eval_rewards/safe_rewards": -0.9610660672187805,
"eval_rewards/unsafe_rewards": -0.9618369936943054,
"eval_runtime": 993.2958,
"eval_samples_per_second": 33.267,
"eval_steps_per_second": 1.04,
"step": 1500
},
{
"epoch": 0.81,
"learning_rate": 5.156638483361933e-08,
"logits/chosen": -2.2049641609191895,
"logits/rejected": -2.0106749534606934,
"logps/chosen": -285.2719421386719,
"logps/rejected": -273.9066467285156,
"loss": 11107.9672,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6739949584007263,
"rewards/margins": 0.23203198611736298,
"rewards/rejected": -0.9060269594192505,
"rewards/safe_rewards": -0.6972517967224121,
"rewards/unsafe_rewards": -0.6507382988929749,
"step": 1510
},
{
"epoch": 0.82,
"learning_rate": 4.8744344286046236e-08,
"logits/chosen": -2.148716926574707,
"logits/rejected": -2.0059280395507812,
"logps/chosen": -285.7190856933594,
"logps/rejected": -266.8186950683594,
"loss": 11427.2711,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.6893398761749268,
"rewards/margins": 0.15746144950389862,
"rewards/rejected": -0.8468014001846313,
"rewards/safe_rewards": -0.7316367030143738,
"rewards/unsafe_rewards": -0.6470431089401245,
"step": 1520
},
{
"epoch": 0.82,
"learning_rate": 4.599335348222169e-08,
"logits/chosen": -2.174553871154785,
"logits/rejected": -2.05711030960083,
"logps/chosen": -288.70849609375,
"logps/rejected": -289.8013610839844,
"loss": 10682.943,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6674157977104187,
"rewards/margins": 0.21851542592048645,
"rewards/rejected": -0.8859313130378723,
"rewards/safe_rewards": -0.6756128668785095,
"rewards/unsafe_rewards": -0.6592189073562622,
"step": 1530
},
{
"epoch": 0.83,
"learning_rate": 4.331438361071163e-08,
"logits/chosen": -2.1525204181671143,
"logits/rejected": -2.059199571609497,
"logps/chosen": -302.69195556640625,
"logps/rejected": -287.97161865234375,
"loss": 11478.0016,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6294053792953491,
"rewards/margins": 0.17254997789859772,
"rewards/rejected": -0.8019553422927856,
"rewards/safe_rewards": -0.6268101334571838,
"rewards/unsafe_rewards": -0.6320004463195801,
"step": 1540
},
{
"epoch": 0.83,
"learning_rate": 4.0708380434367864e-08,
"logits/chosen": -2.1821987628936768,
"logits/rejected": -1.9993159770965576,
"logps/chosen": -265.56854248046875,
"logps/rejected": -261.4121398925781,
"loss": 10532.0812,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6764332056045532,
"rewards/margins": 0.19545750319957733,
"rewards/rejected": -0.8718908429145813,
"rewards/safe_rewards": -0.6992810964584351,
"rewards/unsafe_rewards": -0.6535855531692505,
"step": 1550
},
{
"epoch": 0.84,
"learning_rate": 3.817626395644305e-08,
"logits/chosen": -2.1872050762176514,
"logits/rejected": -2.0328307151794434,
"logps/chosen": -265.10205078125,
"logps/rejected": -248.8824462890625,
"loss": 11308.0203,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.6716850399971008,
"rewards/margins": 0.13508987426757812,
"rewards/rejected": -0.806774914264679,
"rewards/safe_rewards": -0.6651741862297058,
"rewards/unsafe_rewards": -0.6781958341598511,
"step": 1560
},
{
"epoch": 0.84,
"learning_rate": 3.571892809580013e-08,
"logits/chosen": -2.157172203063965,
"logits/rejected": -2.0089964866638184,
"logps/chosen": -260.57525634765625,
"logps/rejected": -251.6233367919922,
"loss": 11400.9055,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6940633058547974,
"rewards/margins": 0.136986643075943,
"rewards/rejected": -0.831049919128418,
"rewards/safe_rewards": -0.6672950983047485,
"rewards/unsafe_rewards": -0.7208314538002014,
"step": 1570
},
{
"epoch": 0.85,
"learning_rate": 3.333724037132976e-08,
"logits/chosen": -2.180004596710205,
"logits/rejected": -2.0329430103302,
"logps/chosen": -274.8149108886719,
"logps/rejected": -270.170654296875,
"loss": 11022.3203,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6869722008705139,
"rewards/margins": 0.17142841219902039,
"rewards/rejected": -0.8584005236625671,
"rewards/safe_rewards": -0.664495587348938,
"rewards/unsafe_rewards": -0.7094486951828003,
"step": 1580
},
{
"epoch": 0.86,
"learning_rate": 3.1032041595688506e-08,
"logits/chosen": -2.107234477996826,
"logits/rejected": -1.9175293445587158,
"logps/chosen": -265.2936706542969,
"logps/rejected": -258.67987060546875,
"loss": 10853.2336,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.6839567422866821,
"rewards/margins": 0.1879226267337799,
"rewards/rejected": -0.8718793988227844,
"rewards/safe_rewards": -0.6944083571434021,
"rewards/unsafe_rewards": -0.6735051870346069,
"step": 1590
},
{
"epoch": 0.86,
"learning_rate": 2.880414557846453e-08,
"logits/chosen": -2.131779432296753,
"logits/rejected": -2.045048236846924,
"logps/chosen": -249.33114624023438,
"logps/rejected": -240.8230438232422,
"loss": 10239.6547,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6429909467697144,
"rewards/margins": 0.1874755173921585,
"rewards/rejected": -0.8304663896560669,
"rewards/safe_rewards": -0.6290788650512695,
"rewards/unsafe_rewards": -0.6569029092788696,
"step": 1600
},
{
"epoch": 0.87,
"learning_rate": 2.6654338838876662e-08,
"logits/chosen": -2.2182207107543945,
"logits/rejected": -1.985608458518982,
"logps/chosen": -280.660400390625,
"logps/rejected": -240.7994384765625,
"loss": 11038.0031,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.6624557971954346,
"rewards/margins": 0.23739679157733917,
"rewards/rejected": -0.8998525738716125,
"rewards/safe_rewards": -0.68045574426651,
"rewards/unsafe_rewards": -0.6444558501243591,
"step": 1610
},
{
"epoch": 0.87,
"learning_rate": 2.4583380328107805e-08,
"logits/chosen": -2.1470413208007812,
"logits/rejected": -1.985733985900879,
"logps/chosen": -291.9235534667969,
"logps/rejected": -258.4531555175781,
"loss": 11528.2828,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6623538136482239,
"rewards/margins": 0.20262321829795837,
"rewards/rejected": -0.8649770021438599,
"rewards/safe_rewards": -0.6641503572463989,
"rewards/unsafe_rewards": -0.660557210445404,
"step": 1620
},
{
"epoch": 0.88,
"learning_rate": 2.259200116137039e-08,
"logits/chosen": -2.1394782066345215,
"logits/rejected": -1.9972803592681885,
"logps/chosen": -284.652099609375,
"logps/rejected": -278.02423095703125,
"loss": 11545.0234,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6758708357810974,
"rewards/margins": 0.17819495499134064,
"rewards/rejected": -0.8540657758712769,
"rewards/safe_rewards": -0.6568797826766968,
"rewards/unsafe_rewards": -0.6948619484901428,
"step": 1630
},
{
"epoch": 0.88,
"learning_rate": 2.068090435979958e-08,
"logits/chosen": -2.1189610958099365,
"logits/rejected": -2.0093135833740234,
"logps/chosen": -268.87371826171875,
"logps/rejected": -250.8845672607422,
"loss": 11243.7469,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6307465434074402,
"rewards/margins": 0.16744786500930786,
"rewards/rejected": -0.798194408416748,
"rewards/safe_rewards": -0.6202843189239502,
"rewards/unsafe_rewards": -0.6412087678909302,
"step": 1640
},
{
"epoch": 0.89,
"learning_rate": 1.8850764602263423e-08,
"logits/chosen": -2.1497740745544434,
"logits/rejected": -1.9683231115341187,
"logps/chosen": -263.89495849609375,
"logps/rejected": -266.2047424316406,
"loss": 11196.7602,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6697946786880493,
"rewards/margins": 0.1570041924715042,
"rewards/rejected": -0.8267987966537476,
"rewards/safe_rewards": -0.6728789806365967,
"rewards/unsafe_rewards": -0.666710376739502,
"step": 1650
},
{
"epoch": 0.89,
"learning_rate": 1.710222798718028e-08,
"logits/chosen": -2.16522479057312,
"logits/rejected": -2.045431613922119,
"logps/chosen": -276.67889404296875,
"logps/rejected": -280.6698303222656,
"loss": 10997.7469,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.6575156450271606,
"rewards/margins": 0.1838151067495346,
"rewards/rejected": -0.8413307070732117,
"rewards/safe_rewards": -0.6694937944412231,
"rewards/unsafe_rewards": -0.6455374360084534,
"step": 1660
},
{
"epoch": 0.9,
"learning_rate": 1.5435911804424356e-08,
"logits/chosen": -2.159721851348877,
"logits/rejected": -2.048668384552002,
"logps/chosen": -290.70916748046875,
"logps/rejected": -270.44683837890625,
"loss": 11924.0805,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6324605345726013,
"rewards/margins": 0.19088464975357056,
"rewards/rejected": -0.8233451843261719,
"rewards/safe_rewards": -0.6401357054710388,
"rewards/unsafe_rewards": -0.624785304069519,
"step": 1670
},
{
"epoch": 0.9,
"learning_rate": 1.3852404317403199e-08,
"logits/chosen": -2.108271360397339,
"logits/rejected": -1.9915637969970703,
"logps/chosen": -250.95361328125,
"logps/rejected": -268.6639099121094,
"loss": 11402.2625,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.6399115324020386,
"rewards/margins": 0.16807249188423157,
"rewards/rejected": -0.8079840540885925,
"rewards/safe_rewards": -0.6549087762832642,
"rewards/unsafe_rewards": -0.6249145269393921,
"step": 1680
},
{
"epoch": 0.91,
"learning_rate": 1.235226455538113e-08,
"logits/chosen": -2.1952316761016846,
"logits/rejected": -2.0874345302581787,
"logps/chosen": -270.4241943359375,
"logps/rejected": -263.0255432128906,
"loss": 11680.2766,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.6519337892532349,
"rewards/margins": 0.18009743094444275,
"rewards/rejected": -0.8320311307907104,
"rewards/safe_rewards": -0.662136435508728,
"rewards/unsafe_rewards": -0.6417311429977417,
"step": 1690
},
{
"epoch": 0.91,
"learning_rate": 1.0936022116124321e-08,
"logits/chosen": -2.155649185180664,
"logits/rejected": -2.0004172325134277,
"logps/chosen": -262.668212890625,
"logps/rejected": -252.78994750976562,
"loss": 10840.3375,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6261528134346008,
"rewards/margins": 0.2126920521259308,
"rewards/rejected": -0.838844895362854,
"rewards/safe_rewards": -0.6410089731216431,
"rewards/unsafe_rewards": -0.6112965941429138,
"step": 1700
},
{
"epoch": 0.92,
"learning_rate": 9.60417697893534e-09,
"logits/chosen": -2.155482292175293,
"logits/rejected": -2.0352649688720703,
"logps/chosen": -259.88397216796875,
"logps/rejected": -262.71746826171875,
"loss": 11428.7422,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.632883608341217,
"rewards/margins": 0.17666089534759521,
"rewards/rejected": -0.8095444440841675,
"rewards/safe_rewards": -0.5953446626663208,
"rewards/unsafe_rewards": -0.6704224348068237,
"step": 1710
},
{
"epoch": 0.93,
"learning_rate": 8.357199328144576e-09,
"logits/chosen": -2.1424155235290527,
"logits/rejected": -2.0332658290863037,
"logps/chosen": -308.6863708496094,
"logps/rejected": -289.75830078125,
"loss": 10265.0281,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6669701337814331,
"rewards/margins": 0.17594949901103973,
"rewards/rejected": -0.8429197072982788,
"rewards/safe_rewards": -0.7015780210494995,
"rewards/unsafe_rewards": -0.6323622465133667,
"step": 1720
},
{
"epoch": 0.93,
"learning_rate": 7.1955293871198144e-09,
"logits/chosen": -2.112140417098999,
"logits/rejected": -2.030024528503418,
"logps/chosen": -241.85476684570312,
"logps/rejected": -241.1482391357422,
"loss": 11823.4781,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6754211187362671,
"rewards/margins": 0.13721489906311035,
"rewards/rejected": -0.8126360177993774,
"rewards/safe_rewards": -0.6722853779792786,
"rewards/unsafe_rewards": -0.6785567998886108,
"step": 1730
},
{
"epoch": 0.94,
"learning_rate": 6.119577262853254e-09,
"logits/chosen": -2.1218502521514893,
"logits/rejected": -1.944239854812622,
"logps/chosen": -251.3634490966797,
"logps/rejected": -230.8335723876953,
"loss": 11556.3273,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.638350248336792,
"rewards/margins": 0.21106202900409698,
"rewards/rejected": -0.8494122624397278,
"rewards/safe_rewards": -0.613598108291626,
"rewards/unsafe_rewards": -0.6631024479866028,
"step": 1740
},
{
"epoch": 0.94,
"learning_rate": 5.129722801180542e-09,
"logits/chosen": -2.1147098541259766,
"logits/rejected": -2.0030481815338135,
"logps/chosen": -270.8954162597656,
"logps/rejected": -255.99600219726562,
"loss": 10173.3906,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6771684885025024,
"rewards/margins": 0.1781720221042633,
"rewards/rejected": -0.8553404808044434,
"rewards/safe_rewards": -0.7092324495315552,
"rewards/unsafe_rewards": -0.6451044082641602,
"step": 1750
},
{
"epoch": 0.95,
"learning_rate": 4.226315452682816e-09,
"logits/chosen": -2.1494641304016113,
"logits/rejected": -2.025930404663086,
"logps/chosen": -264.47100830078125,
"logps/rejected": -256.65093994140625,
"loss": 10950.5703,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6349852681159973,
"rewards/margins": 0.19823208451271057,
"rewards/rejected": -0.833217442035675,
"rewards/safe_rewards": -0.6437097787857056,
"rewards/unsafe_rewards": -0.6262607574462891,
"step": 1760
},
{
"epoch": 0.95,
"learning_rate": 3.4096741493194193e-09,
"logits/chosen": -2.2127685546875,
"logits/rejected": -2.0972578525543213,
"logps/chosen": -264.87530517578125,
"logps/rejected": -257.4256286621094,
"loss": 11913.2609,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6370722651481628,
"rewards/margins": 0.15080325305461884,
"rewards/rejected": -0.7878755927085876,
"rewards/safe_rewards": -0.617976725101471,
"rewards/unsafe_rewards": -0.6561679840087891,
"step": 1770
},
{
"epoch": 0.96,
"learning_rate": 2.6800871918346846e-09,
"logits/chosen": -2.220578670501709,
"logits/rejected": -2.04236102104187,
"logps/chosen": -274.6324768066406,
"logps/rejected": -260.865234375,
"loss": 11005.4078,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.6086142659187317,
"rewards/margins": 0.19890496134757996,
"rewards/rejected": -0.8075191378593445,
"rewards/safe_rewards": -0.6277071237564087,
"rewards/unsafe_rewards": -0.5895212888717651,
"step": 1780
},
{
"epoch": 0.96,
"learning_rate": 2.0378121479783796e-09,
"logits/chosen": -2.132037401199341,
"logits/rejected": -1.972651481628418,
"logps/chosen": -265.41961669921875,
"logps/rejected": -256.274658203125,
"loss": 11459.243,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6822707056999207,
"rewards/margins": 0.2049897462129593,
"rewards/rejected": -0.8872605562210083,
"rewards/safe_rewards": -0.6853871941566467,
"rewards/unsafe_rewards": -0.6791542768478394,
"step": 1790
},
{
"epoch": 0.97,
"learning_rate": 1.4830757615760247e-09,
"logits/chosen": -2.1159682273864746,
"logits/rejected": -1.9551427364349365,
"logps/chosen": -281.0558166503906,
"logps/rejected": -260.3757629394531,
"loss": 11785.8336,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6576239466667175,
"rewards/margins": 0.15307986736297607,
"rewards/rejected": -0.8107039332389832,
"rewards/safe_rewards": -0.689802348613739,
"rewards/unsafe_rewards": -0.6254457235336304,
"step": 1800
},
{
"epoch": 0.97,
"eval_logits/chosen": -1.954970121383667,
"eval_logits/rejected": -1.7745355367660522,
"eval_logps/chosen": -224.6013946533203,
"eval_logps/rejected": -193.1151123046875,
"eval_loss": 4320.82080078125,
"eval_rewards/accuracies": 0.6027347445487976,
"eval_rewards/chosen": -0.9417080879211426,
"eval_rewards/margins": 0.06478659808635712,
"eval_rewards/rejected": -1.0064946413040161,
"eval_rewards/safe_rewards": -0.9369282722473145,
"eval_rewards/unsafe_rewards": -0.9372634291648865,
"eval_runtime": 993.1393,
"eval_samples_per_second": 33.272,
"eval_steps_per_second": 1.04,
"step": 1800
},
{
"epoch": 0.97,
"learning_rate": 1.0160738724809548e-09,
"logits/chosen": -2.1704697608947754,
"logits/rejected": -1.9731838703155518,
"logps/chosen": -259.1123046875,
"logps/rejected": -257.42254638671875,
"loss": 10364.2625,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6492170691490173,
"rewards/margins": 0.19766607880592346,
"rewards/rejected": -0.8468831777572632,
"rewards/safe_rewards": -0.6583928465843201,
"rewards/unsafe_rewards": -0.6400412321090698,
"step": 1810
},
{
"epoch": 0.98,
"learning_rate": 6.369713474366212e-10,
"logits/chosen": -2.1475830078125,
"logits/rejected": -1.9969937801361084,
"logps/chosen": -294.9414367675781,
"logps/rejected": -288.337890625,
"loss": 10068.2195,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.6980792880058289,
"rewards/margins": 0.2098824679851532,
"rewards/rejected": -0.9079617261886597,
"rewards/safe_rewards": -0.7020989656448364,
"rewards/unsafe_rewards": -0.6940596699714661,
"step": 1820
},
{
"epoch": 0.98,
"learning_rate": 3.459020218731512e-10,
"logits/chosen": -2.1376261711120605,
"logits/rejected": -2.0157809257507324,
"logps/chosen": -254.5150604248047,
"logps/rejected": -243.1461181640625,
"loss": 10131.9508,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.6527233719825745,
"rewards/margins": 0.2155275046825409,
"rewards/rejected": -0.8682507276535034,
"rewards/safe_rewards": -0.6437439322471619,
"rewards/unsafe_rewards": -0.6617026329040527,
"step": 1830
},
{
"epoch": 0.99,
"learning_rate": 1.429686526593088e-10,
"logits/chosen": -2.116459608078003,
"logits/rejected": -1.9903713464736938,
"logps/chosen": -270.08428955078125,
"logps/rejected": -265.38336181640625,
"loss": 11322.0812,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.6661794185638428,
"rewards/margins": 0.18438491225242615,
"rewards/rejected": -0.8505643606185913,
"rewards/safe_rewards": -0.6831308603286743,
"rewards/unsafe_rewards": -0.6492279767990112,
"step": 1840
},
{
"epoch": 1.0,
"learning_rate": 2.824288182584622e-11,
"logits/chosen": -2.1975011825561523,
"logits/rejected": -2.020158290863037,
"logps/chosen": -284.1047668457031,
"logps/rejected": -259.551513671875,
"loss": 10472.5969,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6722114682197571,
"rewards/margins": 0.17123940587043762,
"rewards/rejected": -0.8434508442878723,
"rewards/safe_rewards": -0.6627709865570068,
"rewards/unsafe_rewards": -0.6816519498825073,
"step": 1850
},
{
"epoch": 1.0,
"step": 1858,
"total_flos": 0.0,
"train_loss": 11692.422362755651,
"train_runtime": 23508.9776,
"train_samples_per_second": 2.53,
"train_steps_per_second": 0.079
}
],
"logging_steps": 10,
"max_steps": 1858,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 300,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}