Mistral-Crab-DPO / trainer_state.json
Kikkk's picture
commit
c1412d7
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 931,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0010741138560687433,
"grad_norm": 38.01819610595703,
"learning_rate": 5.3191489361702125e-09,
"logits/chosen": -1.7826814651489258,
"logits/rejected": -1.9013216495513916,
"logps/chosen": -277.2808837890625,
"logps/rejected": -667.4992065429688,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.010741138560687433,
"grad_norm": 34.09270477294922,
"learning_rate": 5.3191489361702123e-08,
"logits/chosen": -2.232851266860962,
"logits/rejected": -2.225285530090332,
"logps/chosen": -305.5078125,
"logps/rejected": -258.5682067871094,
"loss": 0.6934,
"rewards/accuracies": 0.3333333432674408,
"rewards/chosen": 0.00024179136380553246,
"rewards/margins": -0.0007453116122633219,
"rewards/rejected": 0.000987103208899498,
"step": 10
},
{
"epoch": 0.021482277121374866,
"grad_norm": 33.920143127441406,
"learning_rate": 1.0638297872340425e-07,
"logits/chosen": -2.0934910774230957,
"logits/rejected": -2.0479633808135986,
"logps/chosen": -278.1986389160156,
"logps/rejected": -270.8542175292969,
"loss": 0.6927,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": 0.002228913828730583,
"rewards/margins": 0.0013251047348603606,
"rewards/rejected": 0.0009038090938702226,
"step": 20
},
{
"epoch": 0.0322234156820623,
"grad_norm": 36.32479476928711,
"learning_rate": 1.5957446808510638e-07,
"logits/chosen": -2.2630741596221924,
"logits/rejected": -2.158916473388672,
"logps/chosen": -329.6264343261719,
"logps/rejected": -238.6651153564453,
"loss": 0.6906,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.015065526589751244,
"rewards/margins": 0.008328847587108612,
"rewards/rejected": 0.006736679468303919,
"step": 30
},
{
"epoch": 0.04296455424274973,
"grad_norm": 29.183805465698242,
"learning_rate": 2.127659574468085e-07,
"logits/chosen": -2.290919542312622,
"logits/rejected": -2.1671340465545654,
"logps/chosen": -298.9222412109375,
"logps/rejected": -290.89227294921875,
"loss": 0.685,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.04827628284692764,
"rewards/margins": 0.017906121909618378,
"rewards/rejected": 0.030370160937309265,
"step": 40
},
{
"epoch": 0.05370569280343716,
"grad_norm": 29.736236572265625,
"learning_rate": 2.659574468085106e-07,
"logits/chosen": -2.400148868560791,
"logits/rejected": -2.2399775981903076,
"logps/chosen": -286.6379699707031,
"logps/rejected": -248.4271697998047,
"loss": 0.6748,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.08774404227733612,
"rewards/margins": 0.037349071353673935,
"rewards/rejected": 0.05039495974779129,
"step": 50
},
{
"epoch": 0.0644468313641246,
"grad_norm": 29.81789207458496,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -2.3034303188323975,
"logits/rejected": -2.119563102722168,
"logps/chosen": -292.1720275878906,
"logps/rejected": -246.50967407226562,
"loss": 0.6684,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.1263999193906784,
"rewards/margins": 0.08450669795274734,
"rewards/rejected": 0.04189322888851166,
"step": 60
},
{
"epoch": 0.07518796992481203,
"grad_norm": 28.368366241455078,
"learning_rate": 3.7234042553191484e-07,
"logits/chosen": -2.194551944732666,
"logits/rejected": -2.0943870544433594,
"logps/chosen": -270.94085693359375,
"logps/rejected": -222.0772705078125,
"loss": 0.6529,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.15133240818977356,
"rewards/margins": 0.09613539278507233,
"rewards/rejected": 0.05519700050354004,
"step": 70
},
{
"epoch": 0.08592910848549946,
"grad_norm": 32.63807678222656,
"learning_rate": 4.25531914893617e-07,
"logits/chosen": -2.110665798187256,
"logits/rejected": -2.0879478454589844,
"logps/chosen": -302.55810546875,
"logps/rejected": -276.37298583984375,
"loss": 0.6364,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.12729570269584656,
"rewards/margins": 0.13354265689849854,
"rewards/rejected": -0.006246985401958227,
"step": 80
},
{
"epoch": 0.0966702470461869,
"grad_norm": 33.7596549987793,
"learning_rate": 4.787234042553192e-07,
"logits/chosen": -2.011441707611084,
"logits/rejected": -1.953314185142517,
"logps/chosen": -275.5764465332031,
"logps/rejected": -278.00384521484375,
"loss": 0.6135,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.013616062700748444,
"rewards/margins": 0.2872251570224762,
"rewards/rejected": -0.30084121227264404,
"step": 90
},
{
"epoch": 0.10741138560687433,
"grad_norm": 49.362213134765625,
"learning_rate": 4.999366067542832e-07,
"logits/chosen": -2.1004366874694824,
"logits/rejected": -1.8379312753677368,
"logps/chosen": -325.45672607421875,
"logps/rejected": -329.73748779296875,
"loss": 0.6027,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.26145023107528687,
"rewards/margins": 0.2924392819404602,
"rewards/rejected": -0.5538895726203918,
"step": 100
},
{
"epoch": 0.11815252416756176,
"grad_norm": 53.218753814697266,
"learning_rate": 4.995493200065303e-07,
"logits/chosen": -2.0087809562683105,
"logits/rejected": -1.840013861656189,
"logps/chosen": -386.00592041015625,
"logps/rejected": -363.0046081542969,
"loss": 0.5916,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.4549340605735779,
"rewards/margins": 0.5379987359046936,
"rewards/rejected": -0.9929327964782715,
"step": 110
},
{
"epoch": 0.1288936627282492,
"grad_norm": 47.64997482299805,
"learning_rate": 4.988105098641859e-07,
"logits/chosen": -1.9923064708709717,
"logits/rejected": -1.8889319896697998,
"logps/chosen": -368.4234313964844,
"logps/rejected": -384.9602355957031,
"loss": 0.5813,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.44473686814308167,
"rewards/margins": 0.4787873327732086,
"rewards/rejected": -0.9235240817070007,
"step": 120
},
{
"epoch": 0.13963480128893663,
"grad_norm": 38.56282424926758,
"learning_rate": 4.977212170395597e-07,
"logits/chosen": -1.8189750909805298,
"logits/rejected": -1.6740334033966064,
"logps/chosen": -328.65789794921875,
"logps/rejected": -318.7656555175781,
"loss": 0.5653,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.4469381272792816,
"rewards/margins": 0.42922893166542053,
"rewards/rejected": -0.8761669993400574,
"step": 130
},
{
"epoch": 0.15037593984962405,
"grad_norm": 60.210453033447266,
"learning_rate": 4.962829759464157e-07,
"logits/chosen": -1.896112084388733,
"logits/rejected": -1.7687921524047852,
"logps/chosen": -346.6222229003906,
"logps/rejected": -360.77880859375,
"loss": 0.6072,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5491372346878052,
"rewards/margins": 0.2989022731781006,
"rewards/rejected": -0.8480396270751953,
"step": 140
},
{
"epoch": 0.1611170784103115,
"grad_norm": 48.47090148925781,
"learning_rate": 4.944978125385465e-07,
"logits/chosen": -2.036083698272705,
"logits/rejected": -1.8509035110473633,
"logps/chosen": -321.4471130371094,
"logps/rejected": -355.3517150878906,
"loss": 0.5702,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.34621065855026245,
"rewards/margins": 0.6038306951522827,
"rewards/rejected": -0.9500414133071899,
"step": 150
},
{
"epoch": 0.17185821697099893,
"grad_norm": 43.58736801147461,
"learning_rate": 4.923682414559481e-07,
"logits/chosen": -1.8674886226654053,
"logits/rejected": -1.689318299293518,
"logps/chosen": -357.22833251953125,
"logps/rejected": -359.64666748046875,
"loss": 0.5438,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.3814140856266022,
"rewards/margins": 0.5234302282333374,
"rewards/rejected": -0.9048442840576172,
"step": 160
},
{
"epoch": 0.18259935553168635,
"grad_norm": 68.03955841064453,
"learning_rate": 4.898972624826147e-07,
"logits/chosen": -1.8476066589355469,
"logits/rejected": -1.6870009899139404,
"logps/chosen": -403.6802673339844,
"logps/rejected": -395.5543212890625,
"loss": 0.5301,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.5655828714370728,
"rewards/margins": 0.6453245282173157,
"rewards/rejected": -1.210907220840454,
"step": 170
},
{
"epoch": 0.1933404940923738,
"grad_norm": 54.94865036010742,
"learning_rate": 4.870883563209449e-07,
"logits/chosen": -1.838039755821228,
"logits/rejected": -1.630350112915039,
"logps/chosen": -387.94305419921875,
"logps/rejected": -392.7327880859375,
"loss": 0.5404,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.5131097435951233,
"rewards/margins": 0.5987601280212402,
"rewards/rejected": -1.1118699312210083,
"step": 180
},
{
"epoch": 0.20408163265306123,
"grad_norm": 57.38792037963867,
"learning_rate": 4.839454796887089e-07,
"logits/chosen": -2.1099352836608887,
"logits/rejected": -1.886162519454956,
"logps/chosen": -425.6485290527344,
"logps/rejected": -410.21478271484375,
"loss": 0.5418,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.5665225982666016,
"rewards/margins": 0.683864414691925,
"rewards/rejected": -1.2503869533538818,
"step": 190
},
{
"epoch": 0.21482277121374865,
"grad_norm": 44.05226135253906,
"learning_rate": 4.804730597454859e-07,
"logits/chosen": -2.0036187171936035,
"logits/rejected": -1.8792438507080078,
"logps/chosen": -399.13043212890625,
"logps/rejected": -396.62396240234375,
"loss": 0.5527,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.4406675696372986,
"rewards/margins": 0.7380279302597046,
"rewards/rejected": -1.1786954402923584,
"step": 200
},
{
"epoch": 0.22556390977443608,
"grad_norm": 51.885616302490234,
"learning_rate": 4.7667598785642125e-07,
"logits/chosen": -2.023136615753174,
"logits/rejected": -1.8735549449920654,
"logps/chosen": -310.3551025390625,
"logps/rejected": -374.1414489746094,
"loss": 0.5322,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.42302098870277405,
"rewards/margins": 0.5779639482498169,
"rewards/rejected": -1.0009849071502686,
"step": 210
},
{
"epoch": 0.23630504833512353,
"grad_norm": 81.05691528320312,
"learning_rate": 4.725596127020879e-07,
"logits/chosen": -1.7004687786102295,
"logits/rejected": -1.7193877696990967,
"logps/chosen": -350.72308349609375,
"logps/rejected": -415.49591064453125,
"loss": 0.5508,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7310193777084351,
"rewards/margins": 0.5848621129989624,
"rewards/rejected": -1.315881609916687,
"step": 220
},
{
"epoch": 0.24704618689581095,
"grad_norm": 53.66835021972656,
"learning_rate": 4.6812973274415924e-07,
"logits/chosen": -1.7762218713760376,
"logits/rejected": -1.6072883605957031,
"logps/chosen": -355.7483825683594,
"logps/rejected": -385.47930908203125,
"loss": 0.5362,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.48422494530677795,
"rewards/margins": 0.607512354850769,
"rewards/rejected": -1.091737151145935,
"step": 230
},
{
"epoch": 0.2577873254564984,
"grad_norm": 61.94645309448242,
"learning_rate": 4.633925880575046e-07,
"logits/chosen": -1.9013745784759521,
"logits/rejected": -1.7701694965362549,
"logps/chosen": -420.7286682128906,
"logps/rejected": -435.3941955566406,
"loss": 0.518,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.7990320920944214,
"rewards/margins": 0.7329047918319702,
"rewards/rejected": -1.5319368839263916,
"step": 240
},
{
"epoch": 0.26852846401718583,
"grad_norm": 72.49921417236328,
"learning_rate": 4.5835485154021437e-07,
"logits/chosen": -1.7423961162567139,
"logits/rejected": -1.6719386577606201,
"logps/chosen": -402.57159423828125,
"logps/rejected": -477.9698791503906,
"loss": 0.5094,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.6995307803153992,
"rewards/margins": 0.7330148220062256,
"rewards/rejected": -1.4325456619262695,
"step": 250
},
{
"epoch": 0.27926960257787325,
"grad_norm": 65.41045379638672,
"learning_rate": 4.530236195139358e-07,
"logits/chosen": -1.8140243291854858,
"logits/rejected": -1.6735155582427979,
"logps/chosen": -393.9617919921875,
"logps/rejected": -421.1630859375,
"loss": 0.5377,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5548108816146851,
"rewards/margins": 0.7069646716117859,
"rewards/rejected": -1.2617756128311157,
"step": 260
},
{
"epoch": 0.2900107411385607,
"grad_norm": 49.067054748535156,
"learning_rate": 4.474064017277605e-07,
"logits/chosen": -1.8181359767913818,
"logits/rejected": -1.7128044366836548,
"logps/chosen": -370.6304016113281,
"logps/rejected": -398.3502502441406,
"loss": 0.544,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.7615430951118469,
"rewards/margins": 0.4647931158542633,
"rewards/rejected": -1.2263362407684326,
"step": 270
},
{
"epoch": 0.3007518796992481,
"grad_norm": 67.5071792602539,
"learning_rate": 4.415111107797445e-07,
"logits/chosen": -1.7189642190933228,
"logits/rejected": -1.7779271602630615,
"logps/chosen": -348.5400695800781,
"logps/rejected": -425.5625915527344,
"loss": 0.5203,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6641637682914734,
"rewards/margins": 0.7238661050796509,
"rewards/rejected": -1.3880298137664795,
"step": 280
},
{
"epoch": 0.31149301825993553,
"grad_norm": 50.27863311767578,
"learning_rate": 4.3534605097096176e-07,
"logits/chosen": -1.651114821434021,
"logits/rejected": -1.5353507995605469,
"logps/chosen": -335.2391357421875,
"logps/rejected": -443.3240661621094,
"loss": 0.5236,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.7241489291191101,
"rewards/margins": 0.6447311639785767,
"rewards/rejected": -1.368880033493042,
"step": 290
},
{
"epoch": 0.322234156820623,
"grad_norm": 52.84365463256836,
"learning_rate": 4.289199066077922e-07,
"logits/chosen": -1.712467908859253,
"logits/rejected": -1.4087601900100708,
"logps/chosen": -359.71502685546875,
"logps/rejected": -427.78045654296875,
"loss": 0.5071,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -0.7437978386878967,
"rewards/margins": 0.8904324769973755,
"rewards/rejected": -1.6342302560806274,
"step": 300
},
{
"epoch": 0.33297529538131043,
"grad_norm": 73.85507202148438,
"learning_rate": 4.2224172976892166e-07,
"logits/chosen": -1.4589178562164307,
"logits/rejected": -1.3919631242752075,
"logps/chosen": -395.29888916015625,
"logps/rejected": -469.9727478027344,
"loss": 0.5085,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.9644006490707397,
"rewards/margins": 0.9210098385810852,
"rewards/rejected": -1.8854105472564697,
"step": 310
},
{
"epoch": 0.34371643394199786,
"grad_norm": 163.49395751953125,
"learning_rate": 4.1532092755428525e-07,
"logits/chosen": -1.453866958618164,
"logits/rejected": -1.392270803451538,
"logps/chosen": -352.23748779296875,
"logps/rejected": -432.1480407714844,
"loss": 0.5065,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.8710242509841919,
"rewards/margins": 0.7999363541603088,
"rewards/rejected": -1.6709606647491455,
"step": 320
},
{
"epoch": 0.3544575725026853,
"grad_norm": 87.29029846191406,
"learning_rate": 4.0816724883391707e-07,
"logits/chosen": -1.7530879974365234,
"logits/rejected": -1.5067665576934814,
"logps/chosen": -385.63702392578125,
"logps/rejected": -428.7237854003906,
"loss": 0.5205,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.8045473098754883,
"rewards/margins": 0.8592826128005981,
"rewards/rejected": -1.663830041885376,
"step": 330
},
{
"epoch": 0.3651987110633727,
"grad_norm": 59.684043884277344,
"learning_rate": 4.007907705153699e-07,
"logits/chosen": -1.6779248714447021,
"logits/rejected": -1.6681525707244873,
"logps/chosen": -394.14007568359375,
"logps/rejected": -467.6216735839844,
"loss": 0.543,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.5750120878219604,
"rewards/margins": 0.646149754524231,
"rewards/rejected": -1.2211616039276123,
"step": 340
},
{
"epoch": 0.37593984962406013,
"grad_norm": 53.85585403442383,
"learning_rate": 3.932018833490518e-07,
"logits/chosen": -1.727351427078247,
"logits/rejected": -1.599186658859253,
"logps/chosen": -391.8357849121094,
"logps/rejected": -442.3279724121094,
"loss": 0.5156,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6497470140457153,
"rewards/margins": 0.7530845403671265,
"rewards/rejected": -1.4028315544128418,
"step": 350
},
{
"epoch": 0.3866809881847476,
"grad_norm": 55.50624084472656,
"learning_rate": 3.854112772914723e-07,
"logits/chosen": -1.479723572731018,
"logits/rejected": -1.5374826192855835,
"logps/chosen": -342.92791748046875,
"logps/rejected": -446.47503662109375,
"loss": 0.5091,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.8904348611831665,
"rewards/margins": 0.907945990562439,
"rewards/rejected": -1.7983808517456055,
"step": 360
},
{
"epoch": 0.39742212674543503,
"grad_norm": 53.516597747802734,
"learning_rate": 3.774299264470177e-07,
"logits/chosen": -1.506805181503296,
"logits/rejected": -1.5588204860687256,
"logps/chosen": -392.9142150878906,
"logps/rejected": -493.38812255859375,
"loss": 0.5008,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9804698824882507,
"rewards/margins": 0.8168279528617859,
"rewards/rejected": -1.7972980737686157,
"step": 370
},
{
"epoch": 0.40816326530612246,
"grad_norm": 57.87943649291992,
"learning_rate": 3.6926907360946604e-07,
"logits/chosen": -1.7936798334121704,
"logits/rejected": -1.573922872543335,
"logps/chosen": -382.94732666015625,
"logps/rejected": -445.59423828125,
"loss": 0.5065,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.781761109828949,
"rewards/margins": 0.822665810585022,
"rewards/rejected": -1.6044269800186157,
"step": 380
},
{
"epoch": 0.4189044038668099,
"grad_norm": 54.115848541259766,
"learning_rate": 3.609402144250171e-07,
"logits/chosen": -1.8267205953598022,
"logits/rejected": -1.5302643775939941,
"logps/chosen": -418.0675354003906,
"logps/rejected": -446.8143005371094,
"loss": 0.4838,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.7378665208816528,
"rewards/margins": 1.0708403587341309,
"rewards/rejected": -1.8087068796157837,
"step": 390
},
{
"epoch": 0.4296455424274973,
"grad_norm": 48.267066955566406,
"learning_rate": 3.5245508119914683e-07,
"logits/chosen": -1.695697546005249,
"logits/rejected": -1.5788294076919556,
"logps/chosen": -400.60382080078125,
"logps/rejected": -472.8611755371094,
"loss": 0.4735,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.0140430927276611,
"rewards/margins": 0.6619123220443726,
"rewards/rejected": -1.6759554147720337,
"step": 400
},
{
"epoch": 0.44038668098818473,
"grad_norm": 54.982364654541016,
"learning_rate": 3.4382562637009484e-07,
"logits/chosen": -1.76909601688385,
"logits/rejected": -1.6182836294174194,
"logps/chosen": -387.72088623046875,
"logps/rejected": -486.9130859375,
"loss": 0.5232,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.9676576852798462,
"rewards/margins": 0.9475955963134766,
"rewards/rejected": -1.9152532815933228,
"step": 410
},
{
"epoch": 0.45112781954887216,
"grad_norm": 63.29111099243164,
"learning_rate": 3.350640056722662e-07,
"logits/chosen": -1.5833942890167236,
"logits/rejected": -1.4724894762039185,
"logps/chosen": -380.0611877441406,
"logps/rejected": -419.60699462890625,
"loss": 0.506,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9589203596115112,
"rewards/margins": 0.64564049243927,
"rewards/rejected": -1.6045608520507812,
"step": 420
},
{
"epoch": 0.46186895810955964,
"grad_norm": 54.812477111816406,
"learning_rate": 3.261825610132634e-07,
"logits/chosen": -1.825090765953064,
"logits/rejected": -1.742057204246521,
"logps/chosen": -434.72491455078125,
"logps/rejected": -475.3501892089844,
"loss": 0.5288,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6500536799430847,
"rewards/margins": 0.9928058385848999,
"rewards/rejected": -1.642859697341919,
"step": 430
},
{
"epoch": 0.47261009667024706,
"grad_norm": 55.682682037353516,
"learning_rate": 3.1719380308866924e-07,
"logits/chosen": -1.8630441427230835,
"logits/rejected": -1.7151546478271484,
"logps/chosen": -378.4299011230469,
"logps/rejected": -382.890869140625,
"loss": 0.4969,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.6695320010185242,
"rewards/margins": 0.7351147532463074,
"rewards/rejected": -1.404646873474121,
"step": 440
},
{
"epoch": 0.4833512352309345,
"grad_norm": 55.0916862487793,
"learning_rate": 3.0811039375906764e-07,
"logits/chosen": -1.9064140319824219,
"logits/rejected": -1.714078664779663,
"logps/chosen": -397.0462951660156,
"logps/rejected": -441.3729553222656,
"loss": 0.5073,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.7625714540481567,
"rewards/margins": 0.8189951181411743,
"rewards/rejected": -1.581566572189331,
"step": 450
},
{
"epoch": 0.4940923737916219,
"grad_norm": 58.015785217285156,
"learning_rate": 2.9894512821413075e-07,
"logits/chosen": -1.7865148782730103,
"logits/rejected": -1.5167757272720337,
"logps/chosen": -416.95550537109375,
"logps/rejected": -436.30450439453125,
"loss": 0.4966,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.9333486557006836,
"rewards/margins": 0.8736823797225952,
"rewards/rejected": -1.807031273841858,
"step": 460
},
{
"epoch": 0.5048335123523093,
"grad_norm": 68.10478210449219,
"learning_rate": 2.8971091694889264e-07,
"logits/chosen": -1.7999794483184814,
"logits/rejected": -1.7058521509170532,
"logps/chosen": -399.33038330078125,
"logps/rejected": -435.877197265625,
"loss": 0.5401,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9552398920059204,
"rewards/margins": 0.6859758496284485,
"rewards/rejected": -1.6412159204483032,
"step": 470
},
{
"epoch": 0.5155746509129968,
"grad_norm": 72.0062255859375,
"learning_rate": 2.8042076757760066e-07,
"logits/chosen": -1.8483903408050537,
"logits/rejected": -1.52974534034729,
"logps/chosen": -399.46746826171875,
"logps/rejected": -427.24981689453125,
"loss": 0.5003,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.8091894388198853,
"rewards/margins": 0.6731428503990173,
"rewards/rejected": -1.4823323488235474,
"step": 480
},
{
"epoch": 0.5263157894736842,
"grad_norm": 61.39365768432617,
"learning_rate": 2.7108776651076116e-07,
"logits/chosen": -1.7545459270477295,
"logits/rejected": -1.6814851760864258,
"logps/chosen": -398.98699951171875,
"logps/rejected": -491.4130859375,
"loss": 0.4783,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.6736662983894348,
"rewards/margins": 0.9081080555915833,
"rewards/rejected": -1.5817744731903076,
"step": 490
},
{
"epoch": 0.5370569280343717,
"grad_norm": 54.34407424926758,
"learning_rate": 2.6172506052119147e-07,
"logits/chosen": -1.7606862783432007,
"logits/rejected": -1.576406478881836,
"logps/chosen": -404.3669128417969,
"logps/rejected": -468.853515625,
"loss": 0.5235,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.9447774887084961,
"rewards/margins": 0.9778174161911011,
"rewards/rejected": -1.9225950241088867,
"step": 500
},
{
"epoch": 0.547798066595059,
"grad_norm": 76.07147216796875,
"learning_rate": 2.523458382250412e-07,
"logits/chosen": -1.7519705295562744,
"logits/rejected": -1.6435455083847046,
"logps/chosen": -410.5835876464844,
"logps/rejected": -490.4002380371094,
"loss": 0.4922,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7955667972564697,
"rewards/margins": 0.8166013956069946,
"rewards/rejected": -1.612168312072754,
"step": 510
},
{
"epoch": 0.5585392051557465,
"grad_norm": 56.88628387451172,
"learning_rate": 2.429633115038737e-07,
"logits/chosen": -1.7606322765350342,
"logits/rejected": -1.561812162399292,
"logps/chosen": -399.27294921875,
"logps/rejected": -455.17462158203125,
"loss": 0.4751,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.7019720673561096,
"rewards/margins": 1.1464049816131592,
"rewards/rejected": -1.8483772277832031,
"step": 520
},
{
"epoch": 0.569280343716434,
"grad_norm": 108.08949279785156,
"learning_rate": 2.3359069689397467e-07,
"logits/chosen": -1.5748560428619385,
"logits/rejected": -1.5276682376861572,
"logps/chosen": -402.18023681640625,
"logps/rejected": -493.3003845214844,
"loss": 0.509,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -1.0235780477523804,
"rewards/margins": 0.8902117609977722,
"rewards/rejected": -1.9137897491455078,
"step": 530
},
{
"epoch": 0.5800214822771214,
"grad_norm": 81.13088989257812,
"learning_rate": 2.2424119696910278e-07,
"logits/chosen": -1.6776129007339478,
"logits/rejected": -1.4775701761245728,
"logps/chosen": -423.53778076171875,
"logps/rejected": -458.8199768066406,
"loss": 0.4854,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.8803874254226685,
"rewards/margins": 0.8511655926704407,
"rewards/rejected": -1.7315528392791748,
"step": 540
},
{
"epoch": 0.5907626208378088,
"grad_norm": 55.828590393066406,
"learning_rate": 2.1492798174291006e-07,
"logits/chosen": -1.6470720767974854,
"logits/rejected": -1.4030835628509521,
"logps/chosen": -370.0120544433594,
"logps/rejected": -398.987060546875,
"loss": 0.4997,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6602271795272827,
"rewards/margins": 0.8755744099617004,
"rewards/rejected": -1.535801649093628,
"step": 550
},
{
"epoch": 0.6015037593984962,
"grad_norm": 50.89604187011719,
"learning_rate": 2.0566417011722652e-07,
"logits/chosen": -1.7970349788665771,
"logits/rejected": -1.5006736516952515,
"logps/chosen": -410.7815856933594,
"logps/rejected": -538.999267578125,
"loss": 0.4874,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -0.8133628964424133,
"rewards/margins": 1.2357568740844727,
"rewards/rejected": -2.049119710922241,
"step": 560
},
{
"epoch": 0.6122448979591837,
"grad_norm": 61.7906494140625,
"learning_rate": 1.9646281140234222e-07,
"logits/chosen": -1.5956964492797852,
"logits/rejected": -1.560612440109253,
"logps/chosen": -411.87615966796875,
"logps/rejected": -522.2528076171875,
"loss": 0.4574,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -1.0511554479599,
"rewards/margins": 1.075621485710144,
"rewards/rejected": -2.126776933670044,
"step": 570
},
{
"epoch": 0.6229860365198711,
"grad_norm": 69.56195831298828,
"learning_rate": 1.8733686693531982e-07,
"logits/chosen": -1.601663589477539,
"logits/rejected": -1.5559356212615967,
"logps/chosen": -434.58880615234375,
"logps/rejected": -539.5415649414062,
"loss": 0.4856,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.2848224639892578,
"rewards/margins": 0.8390380144119263,
"rewards/rejected": -2.1238605976104736,
"step": 580
},
{
"epoch": 0.6337271750805585,
"grad_norm": 67.03250122070312,
"learning_rate": 1.782991918222275e-07,
"logits/chosen": -1.6579906940460205,
"logits/rejected": -1.3883017301559448,
"logps/chosen": -424.90643310546875,
"logps/rejected": -479.0643005371094,
"loss": 0.4948,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.1206614971160889,
"rewards/margins": 0.9874579310417175,
"rewards/rejected": -2.108119487762451,
"step": 590
},
{
"epoch": 0.644468313641246,
"grad_norm": 49.616844177246094,
"learning_rate": 1.693625168300127e-07,
"logits/chosen": -1.5930492877960205,
"logits/rejected": -1.3797900676727295,
"logps/chosen": -399.57135009765625,
"logps/rejected": -451.79022216796875,
"loss": 0.5333,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.0476266145706177,
"rewards/margins": 0.8083570599555969,
"rewards/rejected": -1.8559834957122803,
"step": 600
},
{
"epoch": 0.6552094522019334,
"grad_norm": 66.87409973144531,
"learning_rate": 1.6053943045352516e-07,
"logits/chosen": -1.5892466306686401,
"logits/rejected": -1.5528477430343628,
"logps/chosen": -399.47808837890625,
"logps/rejected": -506.40374755859375,
"loss": 0.4995,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.6764686107635498,
"rewards/margins": 0.8436732292175293,
"rewards/rejected": -1.5201416015625,
"step": 610
},
{
"epoch": 0.6659505907626209,
"grad_norm": 55.837158203125,
"learning_rate": 1.5184236118294787e-07,
"logits/chosen": -1.5849072933197021,
"logits/rejected": -1.4045408964157104,
"logps/chosen": -453.03070068359375,
"logps/rejected": -508.41705322265625,
"loss": 0.4857,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.2203357219696045,
"rewards/margins": 0.8617902994155884,
"rewards/rejected": -2.0821259021759033,
"step": 620
},
{
"epoch": 0.6766917293233082,
"grad_norm": 45.06595230102539,
"learning_rate": 1.4328355999661586e-07,
"logits/chosen": -1.540052890777588,
"logits/rejected": -1.3541651964187622,
"logps/chosen": -389.3322448730469,
"logps/rejected": -468.869384765625,
"loss": 0.5054,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.0844593048095703,
"rewards/margins": 1.0829648971557617,
"rewards/rejected": -2.167423963546753,
"step": 630
},
{
"epoch": 0.6874328678839957,
"grad_norm": 60.282310485839844,
"learning_rate": 1.3487508310388534e-07,
"logits/chosen": -1.434007167816162,
"logits/rejected": -1.4812556505203247,
"logps/chosen": -397.49090576171875,
"logps/rejected": -508.686767578125,
"loss": 0.5134,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.238869309425354,
"rewards/margins": 0.7315988540649414,
"rewards/rejected": -1.9704681634902954,
"step": 640
},
{
"epoch": 0.6981740064446831,
"grad_norm": 45.42427444458008,
"learning_rate": 1.266287749623594e-07,
"logits/chosen": -1.7065637111663818,
"logits/rejected": -1.4134385585784912,
"logps/chosen": -430.59197998046875,
"logps/rejected": -482.938720703125,
"loss": 0.4871,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.0648348331451416,
"rewards/margins": 1.0008587837219238,
"rewards/rejected": -2.0656933784484863,
"step": 650
},
{
"epoch": 0.7089151450053706,
"grad_norm": 55.1621208190918,
"learning_rate": 1.1855625159339452e-07,
"logits/chosen": -1.6233896017074585,
"logits/rejected": -1.4236741065979004,
"logps/chosen": -445.6341247558594,
"logps/rejected": -489.8377380371094,
"loss": 0.4783,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.1494815349578857,
"rewards/margins": 0.9501803517341614,
"rewards/rejected": -2.0996615886688232,
"step": 660
},
{
"epoch": 0.719656283566058,
"grad_norm": 64.62177276611328,
"learning_rate": 1.1066888421939092e-07,
"logits/chosen": -1.6147754192352295,
"logits/rejected": -1.3309791088104248,
"logps/chosen": -438.37042236328125,
"logps/rejected": -420.2880859375,
"loss": 0.4984,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.012224555015564,
"rewards/margins": 0.7375470399856567,
"rewards/rejected": -1.7497714757919312,
"step": 670
},
{
"epoch": 0.7303974221267454,
"grad_norm": 59.346073150634766,
"learning_rate": 1.0297778324591339e-07,
"logits/chosen": -1.5413581132888794,
"logits/rejected": -1.2392480373382568,
"logps/chosen": -423.99853515625,
"logps/rejected": -454.7129821777344,
"loss": 0.5099,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.9406285285949707,
"rewards/margins": 1.0149680376052856,
"rewards/rejected": -1.9555965662002563,
"step": 680
},
{
"epoch": 0.7411385606874329,
"grad_norm": 49.851619720458984,
"learning_rate": 9.549378261120816e-08,
"logits/chosen": -1.583778977394104,
"logits/rejected": -1.4962458610534668,
"logps/chosen": -394.4914245605469,
"logps/rejected": -450.92218017578125,
"loss": 0.4907,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.9432915449142456,
"rewards/margins": 0.7003445029258728,
"rewards/rejected": -1.6436359882354736,
"step": 690
},
{
"epoch": 0.7518796992481203,
"grad_norm": 87.44384002685547,
"learning_rate": 8.822742452516063e-08,
"logits/chosen": -1.6286656856536865,
"logits/rejected": -1.4821763038635254,
"logps/chosen": -421.04083251953125,
"logps/rejected": -503.68365478515625,
"loss": 0.5017,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.912327766418457,
"rewards/margins": 0.9774397015571594,
"rewards/rejected": -1.8897672891616821,
"step": 700
},
{
"epoch": 0.7626208378088077,
"grad_norm": 50.42548751831055,
"learning_rate": 8.118894461919074e-08,
"logits/chosen": -1.4116138219833374,
"logits/rejected": -1.2908035516738892,
"logps/chosen": -368.16656494140625,
"logps/rejected": -451.15087890625,
"loss": 0.4812,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.0490543842315674,
"rewards/margins": 0.8819605708122253,
"rewards/rejected": -1.9310150146484375,
"step": 710
},
{
"epoch": 0.7733619763694952,
"grad_norm": 53.07373809814453,
"learning_rate": 7.438825752800579e-08,
"logits/chosen": -1.6596364974975586,
"logits/rejected": -1.556316614151001,
"logps/chosen": -442.6805114746094,
"logps/rejected": -490.9193420410156,
"loss": 0.4715,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.8844410181045532,
"rewards/margins": 0.9804731607437134,
"rewards/rejected": -1.8649142980575562,
"step": 720
},
{
"epoch": 0.7841031149301826,
"grad_norm": 53.835243225097656,
"learning_rate": 6.783494292351907e-08,
"logits/chosen": -1.4840538501739502,
"logits/rejected": -1.2394304275512695,
"logps/chosen": -376.4036560058594,
"logps/rejected": -482.4645080566406,
"loss": 0.4999,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.9420751333236694,
"rewards/margins": 1.132046103477478,
"rewards/rejected": -2.0741212368011475,
"step": 730
},
{
"epoch": 0.7948442534908701,
"grad_norm": 62.356510162353516,
"learning_rate": 6.153823202060846e-08,
"logits/chosen": -1.4445642232894897,
"logits/rejected": -1.271066665649414,
"logps/chosen": -390.6706848144531,
"logps/rejected": -478.68463134765625,
"loss": 0.4794,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.236981749534607,
"rewards/margins": 0.9016791582107544,
"rewards/rejected": -2.1386606693267822,
"step": 740
},
{
"epoch": 0.8055853920515574,
"grad_norm": 79.2643051147461,
"learning_rate": 5.550699457372282e-08,
"logits/chosen": -1.568433403968811,
"logits/rejected": -1.4340345859527588,
"logps/chosen": -396.7769470214844,
"logps/rejected": -476.18585205078125,
"loss": 0.4811,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.8553333282470703,
"rewards/margins": 1.078909158706665,
"rewards/rejected": -1.9342422485351562,
"step": 750
},
{
"epoch": 0.8163265306122449,
"grad_norm": 75.26664733886719,
"learning_rate": 4.97497263826539e-08,
"logits/chosen": -1.6515172719955444,
"logits/rejected": -1.4298070669174194,
"logps/chosen": -439.5928649902344,
"logps/rejected": -471.7366638183594,
"loss": 0.4988,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -1.0839134454727173,
"rewards/margins": 0.8585917353630066,
"rewards/rejected": -1.942505121231079,
"step": 760
},
{
"epoch": 0.8270676691729323,
"grad_norm": 55.086814880371094,
"learning_rate": 4.4274537325072384e-08,
"logits/chosen": -1.7422411441802979,
"logits/rejected": -1.465787649154663,
"logps/chosen": -413.664306640625,
"logps/rejected": -463.29180908203125,
"loss": 0.4741,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.8051220178604126,
"rewards/margins": 1.0510156154632568,
"rewards/rejected": -1.8561376333236694,
"step": 770
},
{
"epoch": 0.8378088077336198,
"grad_norm": 52.02236557006836,
"learning_rate": 3.9089139932687534e-08,
"logits/chosen": -1.5885370969772339,
"logits/rejected": -1.232320785522461,
"logps/chosen": -414.0470275878906,
"logps/rejected": -457.7579650878906,
"loss": 0.4909,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.0070761442184448,
"rewards/margins": 1.148498773574829,
"rewards/rejected": -2.1555750370025635,
"step": 780
},
{
"epoch": 0.8485499462943072,
"grad_norm": 57.56011199951172,
"learning_rate": 3.4200838527120525e-08,
"logits/chosen": -1.5626875162124634,
"logits/rejected": -1.446763277053833,
"logps/chosen": -366.77081298828125,
"logps/rejected": -446.97760009765625,
"loss": 0.4699,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.8740879893302917,
"rewards/margins": 0.9469636082649231,
"rewards/rejected": -1.8210515975952148,
"step": 790
},
{
"epoch": 0.8592910848549946,
"grad_norm": 66.50774383544922,
"learning_rate": 2.9616518930796447e-08,
"logits/chosen": -1.6846020221710205,
"logits/rejected": -1.5415384769439697,
"logps/chosen": -471.2176818847656,
"logps/rejected": -504.05035400390625,
"loss": 0.4866,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.8976691961288452,
"rewards/margins": 1.1193997859954834,
"rewards/rejected": -2.017069101333618,
"step": 800
},
{
"epoch": 0.8700322234156821,
"grad_norm": 49.84431457519531,
"learning_rate": 2.534263876734802e-08,
"logits/chosen": -1.6844720840454102,
"logits/rejected": -1.4568878412246704,
"logps/chosen": -412.49371337890625,
"logps/rejected": -470.7568359375,
"loss": 0.4711,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9692155718803406,
"rewards/margins": 1.001009464263916,
"rewards/rejected": -1.9702249765396118,
"step": 810
},
{
"epoch": 0.8807733619763695,
"grad_norm": 54.95064926147461,
"learning_rate": 2.138521836519408e-08,
"logits/chosen": -1.6106141805648804,
"logits/rejected": -1.4224177598953247,
"logps/chosen": -422.16796875,
"logps/rejected": -472.8387145996094,
"loss": 0.4862,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.0354992151260376,
"rewards/margins": 0.9904156923294067,
"rewards/rejected": -2.0259149074554443,
"step": 820
},
{
"epoch": 0.8915145005370569,
"grad_norm": 59.206207275390625,
"learning_rate": 1.7749832277106524e-08,
"logits/chosen": -1.5170490741729736,
"logits/rejected": -1.3669121265411377,
"logps/chosen": -387.67791748046875,
"logps/rejected": -476.3672790527344,
"loss": 0.4937,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.0796597003936768,
"rewards/margins": 1.1205719709396362,
"rewards/rejected": -2.2002315521240234,
"step": 830
},
{
"epoch": 0.9022556390977443,
"grad_norm": 76.18675231933594,
"learning_rate": 1.4441601427711836e-08,
"logits/chosen": -1.6329158544540405,
"logits/rejected": -1.3677482604980469,
"logps/chosen": -407.67523193359375,
"logps/rejected": -457.119384765625,
"loss": 0.4827,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.0345627069473267,
"rewards/margins": 0.9469528198242188,
"rewards/rejected": -1.9815152883529663,
"step": 840
},
{
"epoch": 0.9129967776584318,
"grad_norm": 52.468204498291016,
"learning_rate": 1.1465185899987794e-08,
"logits/chosen": -1.5776712894439697,
"logits/rejected": -1.3938223123550415,
"logps/chosen": -396.31195068359375,
"logps/rejected": -443.401123046875,
"loss": 0.5082,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.0158368349075317,
"rewards/margins": 0.9373232126235962,
"rewards/rejected": -1.953160047531128,
"step": 850
},
{
"epoch": 0.9237379162191193,
"grad_norm": 69.90252685546875,
"learning_rate": 8.824778370916996e-09,
"logits/chosen": -1.5731096267700195,
"logits/rejected": -1.47088623046875,
"logps/chosen": -432.2276916503906,
"logps/rejected": -516.0203857421875,
"loss": 0.4968,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.9636839628219604,
"rewards/margins": 0.9173566102981567,
"rewards/rejected": -1.881040334701538,
"step": 860
},
{
"epoch": 0.9344790547798066,
"grad_norm": 53.79716873168945,
"learning_rate": 6.5240982055440044e-09,
"logits/chosen": -1.592310905456543,
"logits/rejected": -1.4055845737457275,
"logps/chosen": -392.08526611328125,
"logps/rejected": -461.09051513671875,
"loss": 0.5285,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.1570608615875244,
"rewards/margins": 0.8815321922302246,
"rewards/rejected": -2.038593292236328,
"step": 870
},
{
"epoch": 0.9452201933404941,
"grad_norm": 55.78624725341797,
"learning_rate": 4.56638621775518e-09,
"logits/chosen": -1.7353003025054932,
"logits/rejected": -1.592315435409546,
"logps/chosen": -423.08868408203125,
"logps/rejected": -525.6699829101562,
"loss": 0.4699,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.9968613386154175,
"rewards/margins": 0.8165915608406067,
"rewards/rejected": -1.8134530782699585,
"step": 880
},
{
"epoch": 0.9559613319011815,
"grad_norm": 64.26811981201172,
"learning_rate": 2.9544001051613478e-09,
"logits/chosen": -1.5674078464508057,
"logits/rejected": -1.477450966835022,
"logps/chosen": -431.13037109375,
"logps/rejected": -499.76849365234375,
"loss": 0.4941,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -1.012650489807129,
"rewards/margins": 0.9235459566116333,
"rewards/rejected": -1.9361963272094727,
"step": 890
},
{
"epoch": 0.966702470461869,
"grad_norm": 63.50950622558594,
"learning_rate": 1.690410564514244e-09,
"logits/chosen": -1.5680744647979736,
"logits/rejected": -1.3493303060531616,
"logps/chosen": -389.19879150390625,
"logps/rejected": -441.45098876953125,
"loss": 0.4976,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.0372302532196045,
"rewards/margins": 0.7519098520278931,
"rewards/rejected": -1.7891401052474976,
"step": 900
},
{
"epoch": 0.9774436090225563,
"grad_norm": 62.333988189697266,
"learning_rate": 7.761980931282053e-10,
"logits/chosen": -1.7527889013290405,
"logits/rejected": -1.5912810564041138,
"logps/chosen": -469.765625,
"logps/rejected": -504.1661071777344,
"loss": 0.4791,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.1952266693115234,
"rewards/margins": 0.7948960065841675,
"rewards/rejected": -1.9901227951049805,
"step": 910
},
{
"epoch": 0.9881847475832438,
"grad_norm": 56.70859909057617,
"learning_rate": 2.1305048081288191e-10,
"logits/chosen": -1.6409803628921509,
"logits/rejected": -1.3718929290771484,
"logps/chosen": -389.1742858886719,
"logps/rejected": -463.9933166503906,
"loss": 0.4863,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.9753316044807434,
"rewards/margins": 1.0109294652938843,
"rewards/rejected": -1.986261010169983,
"step": 920
},
{
"epoch": 0.9989258861439313,
"grad_norm": 53.95326614379883,
"learning_rate": 1.7609958501973998e-12,
"logits/chosen": -1.5267393589019775,
"logits/rejected": -1.3435981273651123,
"logps/chosen": -425.5747985839844,
"logps/rejected": -472.15435791015625,
"loss": 0.4831,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.9775764346122742,
"rewards/margins": 0.9130144119262695,
"rewards/rejected": -1.8905909061431885,
"step": 930
},
{
"epoch": 1.0,
"step": 931,
"total_flos": 0.0,
"train_loss": 0.524798985838762,
"train_runtime": 12257.7363,
"train_samples_per_second": 4.859,
"train_steps_per_second": 0.076
}
],
"logging_steps": 10,
"max_steps": 931,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}