zephyr-7b-dpo-qlora / trainer_state.json
shenxq's picture
Model save
ac637be verified
raw
history blame
No virus
67.9 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9993998799759952,
"eval_steps": 100,
"global_step": 1249,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4e-08,
"logits/chosen": -2.5970396995544434,
"logits/rejected": -2.6002578735351562,
"logps/chosen": -391.7848205566406,
"logps/rejected": -379.88592529296875,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.01,
"learning_rate": 4.0000000000000003e-07,
"logits/chosen": -2.6321280002593994,
"logits/rejected": -2.654905319213867,
"logps/chosen": -415.5525207519531,
"logps/rejected": -451.3063049316406,
"loss": 0.6934,
"rewards/accuracies": 0.4166666567325592,
"rewards/chosen": 3.593924702727236e-05,
"rewards/margins": -0.0004077497578691691,
"rewards/rejected": 0.00044368900125846267,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 8.000000000000001e-07,
"logits/chosen": -2.5803229808807373,
"logits/rejected": -2.6041855812072754,
"logps/chosen": -378.57830810546875,
"logps/rejected": -406.0473937988281,
"loss": 0.6938,
"rewards/accuracies": 0.41874998807907104,
"rewards/chosen": -0.0005379965296015143,
"rewards/margins": -0.0012370327021926641,
"rewards/rejected": 0.0006990364054217935,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 1.2000000000000002e-06,
"logits/chosen": -2.5524168014526367,
"logits/rejected": -2.610682249069214,
"logps/chosen": -405.12225341796875,
"logps/rejected": -437.0355529785156,
"loss": 0.6931,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -2.3546338070445927e-06,
"rewards/margins": 4.741999146062881e-05,
"rewards/rejected": -4.97746623295825e-05,
"step": 30
},
{
"epoch": 0.03,
"learning_rate": 1.6000000000000001e-06,
"logits/chosen": -2.551060199737549,
"logits/rejected": -2.5709078311920166,
"logps/chosen": -374.8092346191406,
"logps/rejected": -387.48931884765625,
"loss": 0.6934,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.0006756076472811401,
"rewards/margins": -0.00045229861279949546,
"rewards/rejected": -0.00022330903448164463,
"step": 40
},
{
"epoch": 0.04,
"learning_rate": 2.0000000000000003e-06,
"logits/chosen": -2.5436787605285645,
"logits/rejected": -2.540389060974121,
"logps/chosen": -366.8270568847656,
"logps/rejected": -374.41497802734375,
"loss": 0.6934,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.0006743489648215473,
"rewards/margins": -0.00049586221575737,
"rewards/rejected": -0.0001784868072718382,
"step": 50
},
{
"epoch": 0.05,
"learning_rate": 2.4000000000000003e-06,
"logits/chosen": -2.5643861293792725,
"logits/rejected": -2.5933163166046143,
"logps/chosen": -389.86444091796875,
"logps/rejected": -415.7872619628906,
"loss": 0.6932,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.0006498451111838222,
"rewards/margins": -0.00017824865062721074,
"rewards/rejected": -0.0004715964605566114,
"step": 60
},
{
"epoch": 0.06,
"learning_rate": 2.8000000000000003e-06,
"logits/chosen": -2.608799457550049,
"logits/rejected": -2.6101136207580566,
"logps/chosen": -390.1094970703125,
"logps/rejected": -396.507568359375,
"loss": 0.693,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.0011631803354248405,
"rewards/margins": 0.00039177463622763753,
"rewards/rejected": -0.0015549547970294952,
"step": 70
},
{
"epoch": 0.06,
"learning_rate": 3.2000000000000003e-06,
"logits/chosen": -2.6286911964416504,
"logits/rejected": -2.6467819213867188,
"logps/chosen": -383.10284423828125,
"logps/rejected": -406.47393798828125,
"loss": 0.6933,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.0017519788816571236,
"rewards/margins": -0.0002883758279494941,
"rewards/rejected": -0.0014636029954999685,
"step": 80
},
{
"epoch": 0.07,
"learning_rate": 3.6000000000000003e-06,
"logits/chosen": -2.6428580284118652,
"logits/rejected": -2.6388907432556152,
"logps/chosen": -431.7904357910156,
"logps/rejected": -446.31732177734375,
"loss": 0.6927,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.0028546121902763844,
"rewards/margins": 0.0008213006658479571,
"rewards/rejected": -0.0036759127397090197,
"step": 90
},
{
"epoch": 0.08,
"learning_rate": 4.000000000000001e-06,
"logits/chosen": -2.6136410236358643,
"logits/rejected": -2.6054491996765137,
"logps/chosen": -382.89886474609375,
"logps/rejected": -399.26068115234375,
"loss": 0.6932,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.003379733068868518,
"rewards/margins": -0.00018416713282931596,
"rewards/rejected": -0.0031955656595528126,
"step": 100
},
{
"epoch": 0.08,
"eval_logits/chosen": -2.490821361541748,
"eval_logits/rejected": -2.497763156890869,
"eval_logps/chosen": -322.7948913574219,
"eval_logps/rejected": -332.520751953125,
"eval_loss": 0.6930087208747864,
"eval_rewards/accuracies": 0.5220000147819519,
"eval_rewards/chosen": -0.002995692892000079,
"eval_rewards/margins": 0.0002869492454919964,
"eval_rewards/rejected": -0.0032826424576342106,
"eval_runtime": 1592.8954,
"eval_samples_per_second": 1.88,
"eval_steps_per_second": 0.235,
"step": 100
},
{
"epoch": 0.09,
"learning_rate": 4.4e-06,
"logits/chosen": -2.6191463470458984,
"logits/rejected": -2.626410722732544,
"logps/chosen": -418.57061767578125,
"logps/rejected": -431.57568359375,
"loss": 0.6929,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.004728841129690409,
"rewards/margins": 0.00041932202293537557,
"rewards/rejected": -0.005148162599653006,
"step": 110
},
{
"epoch": 0.1,
"learning_rate": 4.800000000000001e-06,
"logits/chosen": -2.616450309753418,
"logits/rejected": -2.6316654682159424,
"logps/chosen": -387.65936279296875,
"logps/rejected": -406.0279235839844,
"loss": 0.6924,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.006956800818443298,
"rewards/margins": 0.00143801374360919,
"rewards/rejected": -0.008394813165068626,
"step": 120
},
{
"epoch": 0.1,
"learning_rate": 4.999755876225375e-06,
"logits/chosen": -2.5832996368408203,
"logits/rejected": -2.5880000591278076,
"logps/chosen": -411.5313415527344,
"logps/rejected": -434.4432678222656,
"loss": 0.6925,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.009584048762917519,
"rewards/margins": 0.0013888883404433727,
"rewards/rejected": -0.010972936637699604,
"step": 130
},
{
"epoch": 0.11,
"learning_rate": 4.997803172081864e-06,
"logits/chosen": -2.5645198822021484,
"logits/rejected": -2.5772173404693604,
"logps/chosen": -400.5196533203125,
"logps/rejected": -419.6172790527344,
"loss": 0.6928,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.013501152396202087,
"rewards/margins": 0.0006207471014931798,
"rewards/rejected": -0.01412190031260252,
"step": 140
},
{
"epoch": 0.12,
"learning_rate": 4.9938992891651825e-06,
"logits/chosen": -2.581132411956787,
"logits/rejected": -2.5872106552124023,
"logps/chosen": -382.4533386230469,
"logps/rejected": -414.8580627441406,
"loss": 0.6922,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.016398200765252113,
"rewards/margins": 0.0018568082014098763,
"rewards/rejected": -0.01825500652194023,
"step": 150
},
{
"epoch": 0.13,
"learning_rate": 4.988047277024456e-06,
"logits/chosen": -2.6221516132354736,
"logits/rejected": -2.6419806480407715,
"logps/chosen": -396.94757080078125,
"logps/rejected": -416.6105041503906,
"loss": 0.6922,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.019227243959903717,
"rewards/margins": 0.0020019258372485638,
"rewards/rejected": -0.021229172125458717,
"step": 160
},
{
"epoch": 0.14,
"learning_rate": 4.980251707005417e-06,
"logits/chosen": -2.618068218231201,
"logits/rejected": -2.6144068241119385,
"logps/chosen": -425.07244873046875,
"logps/rejected": -435.1886291503906,
"loss": 0.6935,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.02663121186196804,
"rewards/margins": -0.0007048315601423383,
"rewards/rejected": -0.0259263813495636,
"step": 170
},
{
"epoch": 0.14,
"learning_rate": 4.970518668679459e-06,
"logits/chosen": -2.5808920860290527,
"logits/rejected": -2.5735676288604736,
"logps/chosen": -417.1087341308594,
"logps/rejected": -421.8807678222656,
"loss": 0.6932,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.026894057169556618,
"rewards/margins": -6.247097917366773e-05,
"rewards/rejected": -0.02683158591389656,
"step": 180
},
{
"epoch": 0.15,
"learning_rate": 4.958855765086722e-06,
"logits/chosen": -2.606273651123047,
"logits/rejected": -2.608921527862549,
"logps/chosen": -391.02362060546875,
"logps/rejected": -399.1781311035156,
"loss": 0.6927,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.02517121471464634,
"rewards/margins": 0.0010087827686220407,
"rewards/rejected": -0.026179995387792587,
"step": 190
},
{
"epoch": 0.16,
"learning_rate": 4.945272106796919e-06,
"logits/chosen": -2.578331708908081,
"logits/rejected": -2.594592332839966,
"logps/chosen": -388.1728515625,
"logps/rejected": -412.2106018066406,
"loss": 0.6921,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.022485502064228058,
"rewards/margins": 0.0022102731745690107,
"rewards/rejected": -0.024695774540305138,
"step": 200
},
{
"epoch": 0.16,
"eval_logits/chosen": -2.4899628162384033,
"eval_logits/rejected": -2.497047185897827,
"eval_logps/chosen": -324.8166809082031,
"eval_logps/rejected": -334.6196594238281,
"eval_loss": 0.6926525831222534,
"eval_rewards/accuracies": 0.5183333158493042,
"eval_rewards/chosen": -0.023213516920804977,
"eval_rewards/margins": 0.0010577173670753837,
"eval_rewards/rejected": -0.02427123300731182,
"eval_runtime": 1586.0043,
"eval_samples_per_second": 1.888,
"eval_steps_per_second": 0.236,
"step": 200
},
{
"epoch": 0.17,
"learning_rate": 4.929778304792537e-06,
"logits/chosen": -2.5928235054016113,
"logits/rejected": -2.601378917694092,
"logps/chosen": -424.74066162109375,
"logps/rejected": -424.7421875,
"loss": 0.6914,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.029381360858678818,
"rewards/margins": 0.003531560767441988,
"rewards/rejected": -0.03291292116045952,
"step": 210
},
{
"epoch": 0.18,
"learning_rate": 4.912386462179987e-06,
"logits/chosen": -2.630770206451416,
"logits/rejected": -2.637812376022339,
"logps/chosen": -404.63275146484375,
"logps/rejected": -440.338134765625,
"loss": 0.6902,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.035039953887462616,
"rewards/margins": 0.005983405746519566,
"rewards/rejected": -0.041023362427949905,
"step": 220
},
{
"epoch": 0.18,
"learning_rate": 4.893110164735167e-06,
"logits/chosen": -2.6500868797302246,
"logits/rejected": -2.6668245792388916,
"logps/chosen": -419.52508544921875,
"logps/rejected": -435.5750427246094,
"loss": 0.6928,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.04090961441397667,
"rewards/margins": 0.000937508069910109,
"rewards/rejected": -0.04184712469577789,
"step": 230
},
{
"epoch": 0.19,
"learning_rate": 4.871964470290823e-06,
"logits/chosen": -2.5765552520751953,
"logits/rejected": -2.603816509246826,
"logps/chosen": -419.22332763671875,
"logps/rejected": -441.0245666503906,
"loss": 0.692,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.03845367208123207,
"rewards/margins": 0.0023850365541875362,
"rewards/rejected": -0.040838707238435745,
"step": 240
},
{
"epoch": 0.2,
"learning_rate": 4.848965896974006e-06,
"logits/chosen": -2.62872052192688,
"logits/rejected": -2.652649402618408,
"logps/chosen": -402.7193603515625,
"logps/rejected": -426.7840270996094,
"loss": 0.6924,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.038372453302145004,
"rewards/margins": 0.0015891485381871462,
"rewards/rejected": -0.03996160253882408,
"step": 250
},
{
"epoch": 0.21,
"learning_rate": 4.8241324103028055e-06,
"logits/chosen": -2.6180949211120605,
"logits/rejected": -2.5828635692596436,
"logps/chosen": -408.49200439453125,
"logps/rejected": -419.191650390625,
"loss": 0.6918,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.037816666066646576,
"rewards/margins": 0.002941995393484831,
"rewards/rejected": -0.040758661925792694,
"step": 260
},
{
"epoch": 0.22,
"learning_rate": 4.797483409152438e-06,
"logits/chosen": -2.617037296295166,
"logits/rejected": -2.636782169342041,
"logps/chosen": -391.20391845703125,
"logps/rejected": -412.9513244628906,
"loss": 0.691,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.0356525257229805,
"rewards/margins": 0.004444011487066746,
"rewards/rejected": -0.04009654000401497,
"step": 270
},
{
"epoch": 0.22,
"learning_rate": 4.769039710601669e-06,
"logits/chosen": -2.6231722831726074,
"logits/rejected": -2.6144392490386963,
"logps/chosen": -387.4334411621094,
"logps/rejected": -403.40057373046875,
"loss": 0.6898,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.03822268173098564,
"rewards/margins": 0.00696157943457365,
"rewards/rejected": -0.045184262096881866,
"step": 280
},
{
"epoch": 0.23,
"learning_rate": 4.738823533671383e-06,
"logits/chosen": -2.614532709121704,
"logits/rejected": -2.6345345973968506,
"logps/chosen": -421.5420837402344,
"logps/rejected": -442.87994384765625,
"loss": 0.6916,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.04416601359844208,
"rewards/margins": 0.003424243303015828,
"rewards/rejected": -0.047590263187885284,
"step": 290
},
{
"epoch": 0.24,
"learning_rate": 4.706858481968017e-06,
"logits/chosen": -2.5896575450897217,
"logits/rejected": -2.5868849754333496,
"logps/chosen": -402.4905700683594,
"logps/rejected": -409.7896423339844,
"loss": 0.6913,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.04842864349484444,
"rewards/margins": 0.003919907845556736,
"rewards/rejected": -0.052348554134368896,
"step": 300
},
{
"epoch": 0.24,
"eval_logits/chosen": -2.4894840717315674,
"eval_logits/rejected": -2.4966704845428467,
"eval_logps/chosen": -326.6393127441406,
"eval_logps/rejected": -336.60589599609375,
"eval_loss": 0.6919258236885071,
"eval_rewards/accuracies": 0.5339999794960022,
"eval_rewards/chosen": -0.0414397232234478,
"eval_rewards/margins": 0.0026943066623061895,
"eval_rewards/rejected": -0.04413403198122978,
"eval_runtime": 1585.7818,
"eval_samples_per_second": 1.888,
"eval_steps_per_second": 0.236,
"step": 300
},
{
"epoch": 0.25,
"learning_rate": 4.673169525245416e-06,
"logits/chosen": -2.5968692302703857,
"logits/rejected": -2.585181474685669,
"logps/chosen": -404.8775939941406,
"logps/rejected": -433.09954833984375,
"loss": 0.6902,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.04641267657279968,
"rewards/margins": 0.00627540796995163,
"rewards/rejected": -0.052688080817461014,
"step": 310
},
{
"epoch": 0.26,
"learning_rate": 4.63778297989952e-06,
"logits/chosen": -2.604461193084717,
"logits/rejected": -2.610708475112915,
"logps/chosen": -403.58184814453125,
"logps/rejected": -424.60772705078125,
"loss": 0.691,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.05133752152323723,
"rewards/margins": 0.004545481409877539,
"rewards/rejected": -0.0558830089867115,
"step": 320
},
{
"epoch": 0.26,
"learning_rate": 4.60072648841109e-06,
"logits/chosen": -2.589761734008789,
"logits/rejected": -2.5855839252471924,
"logps/chosen": -411.21026611328125,
"logps/rejected": -430.5553283691406,
"loss": 0.6872,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.0565360002219677,
"rewards/margins": 0.012383319437503815,
"rewards/rejected": -0.06891932338476181,
"step": 330
},
{
"epoch": 0.27,
"learning_rate": 4.562028997752574e-06,
"logits/chosen": -2.5468997955322266,
"logits/rejected": -2.5858261585235596,
"logps/chosen": -409.3692321777344,
"logps/rejected": -441.8675842285156,
"loss": 0.6876,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.04863084852695465,
"rewards/margins": 0.01160926092416048,
"rewards/rejected": -0.06024010106921196,
"step": 340
},
{
"epoch": 0.28,
"learning_rate": 4.521720736775947e-06,
"logits/chosen": -2.604952335357666,
"logits/rejected": -2.596958637237549,
"logps/chosen": -426.7057189941406,
"logps/rejected": -427.89971923828125,
"loss": 0.6907,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.06284764409065247,
"rewards/margins": 0.005581230856478214,
"rewards/rejected": -0.0684288740158081,
"step": 350
},
{
"epoch": 0.29,
"learning_rate": 4.479833192599198e-06,
"logits/chosen": -2.6308584213256836,
"logits/rejected": -2.627711057662964,
"logps/chosen": -435.3694763183594,
"logps/rejected": -454.00775146484375,
"loss": 0.6867,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.06149282306432724,
"rewards/margins": 0.013573038391768932,
"rewards/rejected": -0.07506586611270905,
"step": 360
},
{
"epoch": 0.3,
"learning_rate": 4.436399086009928e-06,
"logits/chosen": -2.6563782691955566,
"logits/rejected": -2.6729133129119873,
"logps/chosen": -409.5260925292969,
"logps/rejected": -424.39208984375,
"loss": 0.6874,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.07003921270370483,
"rewards/margins": 0.012296813540160656,
"rewards/rejected": -0.08233602344989777,
"step": 370
},
{
"epoch": 0.3,
"learning_rate": 4.391452345905239e-06,
"logits/chosen": -2.5911147594451904,
"logits/rejected": -2.6013450622558594,
"logps/chosen": -420.21124267578125,
"logps/rejected": -429.5535583496094,
"loss": 0.6907,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.10565753281116486,
"rewards/margins": 0.006660384126007557,
"rewards/rejected": -0.11231790482997894,
"step": 380
},
{
"epoch": 0.31,
"learning_rate": 4.3450280827879125e-06,
"logits/chosen": -2.6037206649780273,
"logits/rejected": -2.6106367111206055,
"logps/chosen": -413.44122314453125,
"logps/rejected": -427.99951171875,
"loss": 0.6882,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.10301616042852402,
"rewards/margins": 0.011568797752261162,
"rewards/rejected": -0.11458496004343033,
"step": 390
},
{
"epoch": 0.32,
"learning_rate": 4.297162561339554e-06,
"logits/chosen": -2.522566556930542,
"logits/rejected": -2.563248634338379,
"logps/chosen": -421.3636169433594,
"logps/rejected": -440.02276611328125,
"loss": 0.6893,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.09894236922264099,
"rewards/margins": 0.009308888576924801,
"rewards/rejected": -0.10825125873088837,
"step": 400
},
{
"epoch": 0.32,
"eval_logits/chosen": -2.4952824115753174,
"eval_logits/rejected": -2.5022895336151123,
"eval_logps/chosen": -330.40167236328125,
"eval_logps/rejected": -341.0243835449219,
"eval_loss": 0.6891305446624756,
"eval_rewards/accuracies": 0.5546666383743286,
"eval_rewards/chosen": -0.0790637731552124,
"eval_rewards/margins": 0.009254684671759605,
"eval_rewards/rejected": -0.08831847459077835,
"eval_runtime": 1591.081,
"eval_samples_per_second": 1.882,
"eval_steps_per_second": 0.236,
"step": 400
},
{
"epoch": 0.33,
"learning_rate": 4.247893172092157e-06,
"logits/chosen": -2.5598597526550293,
"logits/rejected": -2.5833513736724854,
"logps/chosen": -404.2002868652344,
"logps/rejected": -435.16357421875,
"loss": 0.6857,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.10219142585992813,
"rewards/margins": 0.016902286559343338,
"rewards/rejected": -0.11909371614456177,
"step": 410
},
{
"epoch": 0.34,
"learning_rate": 4.197258402220187e-06,
"logits/chosen": -2.5842721462249756,
"logits/rejected": -2.5888819694519043,
"logps/chosen": -414.70166015625,
"logps/rejected": -425.90667724609375,
"loss": 0.6896,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.12604860961437225,
"rewards/margins": 0.009105135686695576,
"rewards/rejected": -0.13515374064445496,
"step": 420
},
{
"epoch": 0.34,
"learning_rate": 4.145297805476023e-06,
"logits/chosen": -2.639338493347168,
"logits/rejected": -2.6249961853027344,
"logps/chosen": -412.44232177734375,
"logps/rejected": -427.6927185058594,
"loss": 0.6859,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.11504076421260834,
"rewards/margins": 0.016550499945878983,
"rewards/rejected": -0.13159126043319702,
"step": 430
},
{
"epoch": 0.35,
"learning_rate": 4.092051971292228e-06,
"logits/chosen": -2.598680019378662,
"logits/rejected": -2.617053508758545,
"logps/chosen": -404.762939453125,
"logps/rejected": -424.2640686035156,
"loss": 0.6839,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.1352802962064743,
"rewards/margins": 0.02169608511030674,
"rewards/rejected": -0.1569763869047165,
"step": 440
},
{
"epoch": 0.36,
"learning_rate": 4.037562493074792e-06,
"logits/chosen": -2.5920491218566895,
"logits/rejected": -2.59037446975708,
"logps/chosen": -434.03118896484375,
"logps/rejected": -439.4048767089844,
"loss": 0.6832,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.1662554293870926,
"rewards/margins": 0.024613002315163612,
"rewards/rejected": -0.19086843729019165,
"step": 450
},
{
"epoch": 0.37,
"learning_rate": 3.981871935712112e-06,
"logits/chosen": -2.6342484951019287,
"logits/rejected": -2.6563720703125,
"logps/chosen": -411.9359436035156,
"logps/rejected": -435.57586669921875,
"loss": 0.6833,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.16083207726478577,
"rewards/margins": 0.023298023268580437,
"rewards/rejected": -0.18413010239601135,
"step": 460
},
{
"epoch": 0.38,
"learning_rate": 3.925023802325094e-06,
"logits/chosen": -2.664499044418335,
"logits/rejected": -2.6798224449157715,
"logps/chosen": -434.82354736328125,
"logps/rejected": -453.67938232421875,
"loss": 0.6776,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.18204860389232635,
"rewards/margins": 0.03722093254327774,
"rewards/rejected": -0.2192695438861847,
"step": 470
},
{
"epoch": 0.38,
"learning_rate": 3.867062500284342e-06,
"logits/chosen": -2.6116766929626465,
"logits/rejected": -2.627462863922119,
"logps/chosen": -422.6748962402344,
"logps/rejected": -454.072509765625,
"loss": 0.6764,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.23436252772808075,
"rewards/margins": 0.041565194725990295,
"rewards/rejected": -0.27592775225639343,
"step": 480
},
{
"epoch": 0.39,
"learning_rate": 3.8080333065209885e-06,
"logits/chosen": -2.6146063804626465,
"logits/rejected": -2.6115143299102783,
"logps/chosen": -428.1953125,
"logps/rejected": -423.606201171875,
"loss": 0.7007,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.31299567222595215,
"rewards/margins": -0.0009805113077163696,
"rewards/rejected": -0.3120151460170746,
"step": 490
},
{
"epoch": 0.4,
"learning_rate": 3.7479823321582624e-06,
"logits/chosen": -2.6379127502441406,
"logits/rejected": -2.66550874710083,
"logps/chosen": -430.17572021484375,
"logps/rejected": -461.09942626953125,
"loss": 0.6724,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.24496836960315704,
"rewards/margins": 0.04783773794770241,
"rewards/rejected": -0.29280608892440796,
"step": 500
},
{
"epoch": 0.4,
"eval_logits/chosen": -2.5028529167175293,
"eval_logits/rejected": -2.5099613666534424,
"eval_logps/chosen": -342.6784973144531,
"eval_logps/rejected": -354.7255859375,
"eval_loss": 0.684412956237793,
"eval_rewards/accuracies": 0.5529999732971191,
"eval_rewards/chosen": -0.20183168351650238,
"eval_rewards/margins": 0.023498954251408577,
"eval_rewards/rejected": -0.2253306657075882,
"eval_runtime": 1585.7151,
"eval_samples_per_second": 1.888,
"eval_steps_per_second": 0.236,
"step": 500
},
{
"epoch": 0.41,
"learning_rate": 3.686956486491419e-06,
"logits/chosen": -2.59777569770813,
"logits/rejected": -2.600053071975708,
"logps/chosen": -440.66217041015625,
"logps/rejected": -454.6631774902344,
"loss": 0.6921,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.2590482234954834,
"rewards/margins": 0.0103686498478055,
"rewards/rejected": -0.269416868686676,
"step": 510
},
{
"epoch": 0.42,
"learning_rate": 3.625003440344166e-06,
"logits/chosen": -2.6117706298828125,
"logits/rejected": -2.625525712966919,
"logps/chosen": -416.2491760253906,
"logps/rejected": -427.0350646972656,
"loss": 0.6781,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.228049635887146,
"rewards/margins": 0.037227652966976166,
"rewards/rejected": -0.26527729630470276,
"step": 520
},
{
"epoch": 0.42,
"learning_rate": 3.562171588830231e-06,
"logits/chosen": -2.670912265777588,
"logits/rejected": -2.6928374767303467,
"logps/chosen": -428.692626953125,
"logps/rejected": -462.62274169921875,
"loss": 0.677,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.22094754874706268,
"rewards/margins": 0.041822899132966995,
"rewards/rejected": -0.26277047395706177,
"step": 530
},
{
"epoch": 0.43,
"learning_rate": 3.4985100135491245e-06,
"logits/chosen": -2.6906635761260986,
"logits/rejected": -2.708909511566162,
"logps/chosen": -444.14483642578125,
"logps/rejected": -480.6028747558594,
"loss": 0.6691,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.2567470967769623,
"rewards/margins": 0.058775972574949265,
"rewards/rejected": -0.31552308797836304,
"step": 540
},
{
"epoch": 0.44,
"learning_rate": 3.4340684442456673e-06,
"logits/chosen": -2.5944106578826904,
"logits/rejected": -2.6143410205841064,
"logps/chosen": -438.25738525390625,
"logps/rejected": -459.01763916015625,
"loss": 0.6759,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3285837173461914,
"rewards/margins": 0.04724721983075142,
"rewards/rejected": -0.37583091855049133,
"step": 550
},
{
"epoch": 0.45,
"learning_rate": 3.3688972199631974e-06,
"logits/chosen": -2.6361145973205566,
"logits/rejected": -2.613219976425171,
"logps/chosen": -441.1446838378906,
"logps/rejected": -451.8172912597656,
"loss": 0.6769,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.3550431728363037,
"rewards/margins": 0.05103546380996704,
"rewards/rejected": -0.40607863664627075,
"step": 560
},
{
"epoch": 0.46,
"learning_rate": 3.3030472497208354e-06,
"logits/chosen": -2.6290247440338135,
"logits/rejected": -2.652219295501709,
"logps/chosen": -426.34234619140625,
"logps/rejected": -484.0997009277344,
"loss": 0.6598,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.3327481150627136,
"rewards/margins": 0.08233290165662766,
"rewards/rejected": -0.4150810241699219,
"step": 570
},
{
"epoch": 0.46,
"learning_rate": 3.236569972745492e-06,
"logits/chosen": -2.640798807144165,
"logits/rejected": -2.6551499366760254,
"logps/chosen": -419.3091735839844,
"logps/rejected": -434.71527099609375,
"loss": 0.6828,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.31533199548721313,
"rewards/margins": 0.03718234971165657,
"rewards/rejected": -0.3525143265724182,
"step": 580
},
{
"epoch": 0.47,
"learning_rate": 3.1695173182897126e-06,
"logits/chosen": -2.5991015434265137,
"logits/rejected": -2.6085472106933594,
"logps/chosen": -435.3345642089844,
"logps/rejected": -467.74078369140625,
"loss": 0.6698,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.3784990906715393,
"rewards/margins": 0.060858141630887985,
"rewards/rejected": -0.4393572211265564,
"step": 590
},
{
"epoch": 0.48,
"learning_rate": 3.10194166506673e-06,
"logits/chosen": -2.624173164367676,
"logits/rejected": -2.649747848510742,
"logps/chosen": -443.6612243652344,
"logps/rejected": -470.52447509765625,
"loss": 0.6849,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.3967738747596741,
"rewards/margins": 0.03684181347489357,
"rewards/rejected": -0.43361568450927734,
"step": 600
},
{
"epoch": 0.48,
"eval_logits/chosen": -2.534677505493164,
"eval_logits/rejected": -2.5411627292633057,
"eval_logps/chosen": -356.15911865234375,
"eval_logps/rejected": -369.895751953125,
"eval_loss": 0.6804664731025696,
"eval_rewards/accuracies": 0.5596666932106018,
"eval_rewards/chosen": -0.33663830161094666,
"eval_rewards/margins": 0.04039384424686432,
"eval_rewards/rejected": -0.3770321309566498,
"eval_runtime": 1591.4153,
"eval_samples_per_second": 1.881,
"eval_steps_per_second": 0.236,
"step": 600
},
{
"epoch": 0.49,
"learning_rate": 3.0338958003344115e-06,
"logits/chosen": -2.628021717071533,
"logits/rejected": -2.654170036315918,
"logps/chosen": -423.56768798828125,
"logps/rejected": -451.36041259765625,
"loss": 0.6603,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.39979976415634155,
"rewards/margins": 0.08663686364889145,
"rewards/rejected": -0.486436665058136,
"step": 610
},
{
"epoch": 0.5,
"learning_rate": 2.9654328786600823e-06,
"logits/chosen": -2.6157093048095703,
"logits/rejected": -2.6425511837005615,
"logps/chosen": -437.9432678222656,
"logps/rejected": -482.203125,
"loss": 0.6558,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.4792413115501404,
"rewards/margins": 0.10463404655456543,
"rewards/rejected": -0.5838753581047058,
"step": 620
},
{
"epoch": 0.5,
"learning_rate": 2.896606380398402e-06,
"logits/chosen": -2.6413679122924805,
"logits/rejected": -2.6552627086639404,
"logps/chosen": -455.11260986328125,
"logps/rejected": -468.00885009765625,
"loss": 0.6878,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5618987083435059,
"rewards/margins": 0.03895387798547745,
"rewards/rejected": -0.6008526086807251,
"step": 630
},
{
"epoch": 0.51,
"learning_rate": 2.827470069914772e-06,
"logits/chosen": -2.681443214416504,
"logits/rejected": -2.6886324882507324,
"logps/chosen": -459.51953125,
"logps/rejected": -491.306396484375,
"loss": 0.6753,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5284784436225891,
"rewards/margins": 0.06967002153396606,
"rewards/rejected": -0.5981484651565552,
"step": 640
},
{
"epoch": 0.52,
"learning_rate": 2.7580779535868675e-06,
"logits/chosen": -2.630887508392334,
"logits/rejected": -2.645677328109741,
"logps/chosen": -457.000732421875,
"logps/rejected": -472.52545166015625,
"loss": 0.6748,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.47293543815612793,
"rewards/margins": 0.06035799905657768,
"rewards/rejected": -0.5332934260368347,
"step": 650
},
{
"epoch": 0.53,
"learning_rate": 2.688484237617129e-06,
"logits/chosen": -2.6493170261383057,
"logits/rejected": -2.68170166015625,
"logps/chosen": -453.5931091308594,
"logps/rejected": -479.94818115234375,
"loss": 0.6716,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.4414174556732178,
"rewards/margins": 0.06612064689397812,
"rewards/rejected": -0.5075381398200989,
"step": 660
},
{
"epoch": 0.54,
"learning_rate": 2.6187432856891585e-06,
"logits/chosen": -2.6594104766845703,
"logits/rejected": -2.6713552474975586,
"logps/chosen": -456.5660095214844,
"logps/rejected": -488.9071350097656,
"loss": 0.6665,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.44480568170547485,
"rewards/margins": 0.07257355749607086,
"rewards/rejected": -0.5173791646957397,
"step": 670
},
{
"epoch": 0.54,
"learning_rate": 2.548909576501096e-06,
"logits/chosen": -2.7302539348602295,
"logits/rejected": -2.7218852043151855,
"logps/chosen": -458.1416931152344,
"logps/rejected": -482.7391662597656,
"loss": 0.6629,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.4737418591976166,
"rewards/margins": 0.08715833723545074,
"rewards/rejected": -0.5609001517295837,
"step": 680
},
{
"epoch": 0.55,
"learning_rate": 2.4790376612091503e-06,
"logits/chosen": -2.7039058208465576,
"logits/rejected": -2.7273173332214355,
"logps/chosen": -473.97064208984375,
"logps/rejected": -498.3961486816406,
"loss": 0.6715,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.5430811643600464,
"rewards/margins": 0.08151289075613022,
"rewards/rejected": -0.6245940327644348,
"step": 690
},
{
"epoch": 0.56,
"learning_rate": 2.40918212081453e-06,
"logits/chosen": -2.6450188159942627,
"logits/rejected": -2.667860507965088,
"logps/chosen": -424.064208984375,
"logps/rejected": -463.056640625,
"loss": 0.6503,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.4541637897491455,
"rewards/margins": 0.10934968292713165,
"rewards/rejected": -0.5635135173797607,
"step": 700
},
{
"epoch": 0.56,
"eval_logits/chosen": -2.543064832687378,
"eval_logits/rejected": -2.5492172241210938,
"eval_logps/chosen": -366.25225830078125,
"eval_logps/rejected": -381.38427734375,
"eval_loss": 0.677395224571228,
"eval_rewards/accuracies": 0.5630000233650208,
"eval_rewards/chosen": -0.4375694692134857,
"eval_rewards/margins": 0.054348159581422806,
"eval_rewards/rejected": -0.4919176399707794,
"eval_runtime": 1584.8471,
"eval_samples_per_second": 1.889,
"eval_steps_per_second": 0.237,
"step": 700
},
{
"epoch": 0.57,
"learning_rate": 2.3393975235270654e-06,
"logits/chosen": -2.6429007053375244,
"logits/rejected": -2.635617256164551,
"logps/chosen": -460.6632385253906,
"logps/rejected": -495.57098388671875,
"loss": 0.6491,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5247341990470886,
"rewards/margins": 0.12096454948186874,
"rewards/rejected": -0.6456986665725708,
"step": 710
},
{
"epoch": 0.58,
"learning_rate": 2.2697383821388153e-06,
"logits/chosen": -2.665684700012207,
"logits/rejected": -2.63784122467041,
"logps/chosen": -454.69110107421875,
"logps/rejected": -466.15020751953125,
"loss": 0.6856,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.49990567564964294,
"rewards/margins": 0.04791254550218582,
"rewards/rejected": -0.547818124294281,
"step": 720
},
{
"epoch": 0.58,
"learning_rate": 2.2002591114409657e-06,
"logits/chosen": -2.672154426574707,
"logits/rejected": -2.6631722450256348,
"logps/chosen": -453.4434509277344,
"logps/rejected": -476.5977478027344,
"loss": 0.6716,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.4569157660007477,
"rewards/margins": 0.07301469147205353,
"rewards/rejected": -0.5299304723739624,
"step": 730
},
{
"epoch": 0.59,
"learning_rate": 2.131013985717285e-06,
"logits/chosen": -2.6686697006225586,
"logits/rejected": -2.662213087081909,
"logps/chosen": -456.8734436035156,
"logps/rejected": -498.776123046875,
"loss": 0.6499,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.38933560252189636,
"rewards/margins": 0.1104939803481102,
"rewards/rejected": -0.49982959032058716,
"step": 740
},
{
"epoch": 0.6,
"learning_rate": 2.062057096347338e-06,
"logits/chosen": -2.6757967472076416,
"logits/rejected": -2.683659553527832,
"logps/chosen": -449.1112365722656,
"logps/rejected": -460.0345764160156,
"loss": 0.6687,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.40121787786483765,
"rewards/margins": 0.07739736139774323,
"rewards/rejected": -0.47861528396606445,
"step": 750
},
{
"epoch": 0.61,
"learning_rate": 1.9934423095525733e-06,
"logits/chosen": -2.660757541656494,
"logits/rejected": -2.641571521759033,
"logps/chosen": -453.0924377441406,
"logps/rejected": -462.0196228027344,
"loss": 0.6656,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.3819692134857178,
"rewards/margins": 0.08219519257545471,
"rewards/rejected": -0.4641643464565277,
"step": 760
},
{
"epoch": 0.62,
"learning_rate": 1.9252232243182986e-06,
"logits/chosen": -2.6494193077087402,
"logits/rejected": -2.6644535064697266,
"logps/chosen": -420.4537658691406,
"logps/rejected": -452.49700927734375,
"loss": 0.6659,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.4124923348426819,
"rewards/margins": 0.08118386566638947,
"rewards/rejected": -0.49367618560791016,
"step": 770
},
{
"epoch": 0.62,
"learning_rate": 1.8574531305244043e-06,
"logits/chosen": -2.674877882003784,
"logits/rejected": -2.698631763458252,
"logps/chosen": -459.5841369628906,
"logps/rejected": -493.99969482421875,
"loss": 0.6453,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.4147106111049652,
"rewards/margins": 0.12960824370384216,
"rewards/rejected": -0.5443187952041626,
"step": 780
},
{
"epoch": 0.63,
"learning_rate": 1.7901849673175559e-06,
"logits/chosen": -2.69256329536438,
"logits/rejected": -2.7218129634857178,
"logps/chosen": -455.54736328125,
"logps/rejected": -486.2936096191406,
"loss": 0.66,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.38235577940940857,
"rewards/margins": 0.09029395878314972,
"rewards/rejected": -0.4726497530937195,
"step": 790
},
{
"epoch": 0.64,
"learning_rate": 1.7234712817573555e-06,
"logits/chosen": -2.670511484146118,
"logits/rejected": -2.6233620643615723,
"logps/chosen": -479.3063049316406,
"logps/rejected": -484.30718994140625,
"loss": 0.6841,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.45734840631484985,
"rewards/margins": 0.05288088321685791,
"rewards/rejected": -0.510229229927063,
"step": 800
},
{
"epoch": 0.64,
"eval_logits/chosen": -2.559168577194214,
"eval_logits/rejected": -2.56618070602417,
"eval_logps/chosen": -354.3205871582031,
"eval_logps/rejected": -370.067626953125,
"eval_loss": 0.6735012531280518,
"eval_rewards/accuracies": 0.5913333296775818,
"eval_rewards/chosen": -0.31825292110443115,
"eval_rewards/margins": 0.06049799174070358,
"eval_rewards/rejected": -0.37875089049339294,
"eval_runtime": 1585.6649,
"eval_samples_per_second": 1.888,
"eval_steps_per_second": 0.236,
"step": 800
},
{
"epoch": 0.65,
"learning_rate": 1.6573641877687936e-06,
"logits/chosen": -2.688758373260498,
"logits/rejected": -2.7048513889312744,
"logps/chosen": -433.21484375,
"logps/rejected": -472.264892578125,
"loss": 0.6559,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.3647409975528717,
"rewards/margins": 0.10189852863550186,
"rewards/rejected": -0.46663951873779297,
"step": 810
},
{
"epoch": 0.66,
"learning_rate": 1.591915325433034e-06,
"logits/chosen": -2.705111265182495,
"logits/rejected": -2.690777540206909,
"logps/chosen": -436.6507263183594,
"logps/rejected": -449.94891357421875,
"loss": 0.6732,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.39074164628982544,
"rewards/margins": 0.06702588498592377,
"rewards/rejected": -0.4577675461769104,
"step": 820
},
{
"epoch": 0.66,
"learning_rate": 1.5271758206483664e-06,
"logits/chosen": -2.7012417316436768,
"logits/rejected": -2.692389726638794,
"logps/chosen": -457.00238037109375,
"logps/rejected": -475.9202575683594,
"loss": 0.6669,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.4308633804321289,
"rewards/margins": 0.09242836385965347,
"rewards/rejected": -0.5232917666435242,
"step": 830
},
{
"epoch": 0.67,
"learning_rate": 1.4631962451927966e-06,
"logits/chosen": -2.675565719604492,
"logits/rejected": -2.708937168121338,
"logps/chosen": -453.7381286621094,
"logps/rejected": -485.38360595703125,
"loss": 0.6646,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.42438727617263794,
"rewards/margins": 0.09076879173517227,
"rewards/rejected": -0.515156090259552,
"step": 840
},
{
"epoch": 0.68,
"learning_rate": 1.4000265772195032e-06,
"logits/chosen": -2.706509590148926,
"logits/rejected": -2.7089927196502686,
"logps/chosen": -460.16326904296875,
"logps/rejected": -492.61163330078125,
"loss": 0.6616,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.4706672132015228,
"rewards/margins": 0.09530263394117355,
"rewards/rejected": -0.5659698247909546,
"step": 850
},
{
"epoch": 0.69,
"learning_rate": 1.3377161622160077e-06,
"logits/chosen": -2.688765048980713,
"logits/rejected": -2.686649799346924,
"logps/chosen": -452.43035888671875,
"logps/rejected": -478.95855712890625,
"loss": 0.6575,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.4564797878265381,
"rewards/margins": 0.09994712471961975,
"rewards/rejected": -0.5564268827438354,
"step": 860
},
{
"epoch": 0.7,
"learning_rate": 1.276313674457553e-06,
"logits/chosen": -2.687105417251587,
"logits/rejected": -2.6988847255706787,
"logps/chosen": -431.041259765625,
"logps/rejected": -468.3299865722656,
"loss": 0.644,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.4324752390384674,
"rewards/margins": 0.13809339702129364,
"rewards/rejected": -0.5705686211585999,
"step": 870
},
{
"epoch": 0.7,
"learning_rate": 1.2158670789848095e-06,
"logits/chosen": -2.6659626960754395,
"logits/rejected": -2.6399731636047363,
"logps/chosen": -455.6881408691406,
"logps/rejected": -471.60479736328125,
"loss": 0.6906,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.4899720549583435,
"rewards/margins": 0.03855130821466446,
"rewards/rejected": -0.528523325920105,
"step": 880
},
{
"epoch": 0.71,
"learning_rate": 1.1564235941356016e-06,
"logits/chosen": -2.636784791946411,
"logits/rejected": -2.665184497833252,
"logps/chosen": -443.228759765625,
"logps/rejected": -486.3818359375,
"loss": 0.6375,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.49858155846595764,
"rewards/margins": 0.15248027443885803,
"rewards/rejected": -0.6510618329048157,
"step": 890
},
{
"epoch": 0.72,
"learning_rate": 1.0980296546599254e-06,
"logits/chosen": -2.6810688972473145,
"logits/rejected": -2.66475248336792,
"logps/chosen": -478.31512451171875,
"logps/rejected": -483.4214782714844,
"loss": 0.6773,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.5605891942977905,
"rewards/margins": 0.07103095203638077,
"rewards/rejected": -0.6316200494766235,
"step": 900
},
{
"epoch": 0.72,
"eval_logits/chosen": -2.5706326961517334,
"eval_logits/rejected": -2.577380418777466,
"eval_logps/chosen": -362.3546142578125,
"eval_logps/rejected": -378.96929931640625,
"eval_loss": 0.6724240183830261,
"eval_rewards/accuracies": 0.5886666774749756,
"eval_rewards/chosen": -0.3985931873321533,
"eval_rewards/margins": 0.06917466968297958,
"eval_rewards/rejected": -0.4677678346633911,
"eval_runtime": 1591.0139,
"eval_samples_per_second": 1.882,
"eval_steps_per_second": 0.236,
"step": 900
},
{
"epoch": 0.73,
"learning_rate": 1.040730875447083e-06,
"logits/chosen": -2.6869008541107178,
"logits/rejected": -2.6931540966033936,
"logps/chosen": -458.8529357910156,
"logps/rejected": -475.8218688964844,
"loss": 0.6623,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.47533708810806274,
"rewards/margins": 0.09188753366470337,
"rewards/rejected": -0.5672245621681213,
"step": 910
},
{
"epoch": 0.74,
"learning_rate": 9.845720158932414e-07,
"logits/chosen": -2.655848741531372,
"logits/rejected": -2.650817632675171,
"logps/chosen": -427.18896484375,
"logps/rejected": -438.2098693847656,
"loss": 0.6659,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.4379213750362396,
"rewards/margins": 0.079404316842556,
"rewards/rejected": -0.5173256993293762,
"step": 920
},
{
"epoch": 0.74,
"learning_rate": 9.295969449372796e-07,
"logits/chosen": -2.682236433029175,
"logits/rejected": -2.706988573074341,
"logps/chosen": -434.9564514160156,
"logps/rejected": -458.70831298828125,
"loss": 0.6677,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.38662606477737427,
"rewards/margins": 0.08299224078655243,
"rewards/rejected": -0.4696182608604431,
"step": 930
},
{
"epoch": 0.75,
"learning_rate": 8.758486067922176e-07,
"logits/chosen": -2.68632435798645,
"logits/rejected": -2.700981616973877,
"logps/chosen": -425.8504943847656,
"logps/rejected": -465.1412048339844,
"loss": 0.651,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.372600257396698,
"rewards/margins": 0.1192459687590599,
"rewards/rejected": -0.4918461740016937,
"step": 940
},
{
"epoch": 0.76,
"learning_rate": 8.233689873990006e-07,
"logits/chosen": -2.661367654800415,
"logits/rejected": -2.663963794708252,
"logps/chosen": -430.2080993652344,
"logps/rejected": -466.76092529296875,
"loss": 0.6451,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.3554816246032715,
"rewards/margins": 0.12667140364646912,
"rewards/rejected": -0.4821530282497406,
"step": 950
},
{
"epoch": 0.77,
"learning_rate": 7.721990816288555e-07,
"logits/chosen": -2.7383949756622314,
"logits/rejected": -2.7589166164398193,
"logps/chosen": -413.57135009765625,
"logps/rejected": -441.5545349121094,
"loss": 0.6724,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.371616005897522,
"rewards/margins": 0.06721927225589752,
"rewards/rejected": -0.4388352930545807,
"step": 960
},
{
"epoch": 0.78,
"learning_rate": 7.223788612598148e-07,
"logits/chosen": -2.7295823097229004,
"logits/rejected": -2.740691661834717,
"logps/chosen": -443.89776611328125,
"logps/rejected": -474.42254638671875,
"loss": 0.6495,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.332157701253891,
"rewards/margins": 0.1200638860464096,
"rewards/rejected": -0.4522215723991394,
"step": 970
},
{
"epoch": 0.78,
"learning_rate": 6.73947243752448e-07,
"logits/chosen": -2.70650053024292,
"logits/rejected": -2.6953940391540527,
"logps/chosen": -460.09918212890625,
"logps/rejected": -455.58489990234375,
"loss": 0.6806,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.4144526422023773,
"rewards/margins": 0.05592719465494156,
"rewards/rejected": -0.4703798294067383,
"step": 980
},
{
"epoch": 0.79,
"learning_rate": 6.269420618491759e-07,
"logits/chosen": -2.7356605529785156,
"logits/rejected": -2.737133026123047,
"logps/chosen": -423.57537841796875,
"logps/rejected": -452.10260009765625,
"loss": 0.6665,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3517349660396576,
"rewards/margins": 0.08071886003017426,
"rewards/rejected": -0.43245387077331543,
"step": 990
},
{
"epoch": 0.8,
"learning_rate": 5.814000340209267e-07,
"logits/chosen": -2.7007267475128174,
"logits/rejected": -2.700061798095703,
"logps/chosen": -442.41705322265625,
"logps/rejected": -477.9020080566406,
"loss": 0.657,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.34119993448257446,
"rewards/margins": 0.10114644467830658,
"rewards/rejected": -0.44234639406204224,
"step": 1000
},
{
"epoch": 0.8,
"eval_logits/chosen": -2.570801258087158,
"eval_logits/rejected": -2.578376293182373,
"eval_logps/chosen": -350.23724365234375,
"eval_logps/rejected": -366.5909118652344,
"eval_loss": 0.6711118221282959,
"eval_rewards/accuracies": 0.5996666550636292,
"eval_rewards/chosen": -0.277418851852417,
"eval_rewards/margins": 0.06656476855278015,
"eval_rewards/rejected": -0.34398359060287476,
"eval_runtime": 1581.2525,
"eval_samples_per_second": 1.893,
"eval_steps_per_second": 0.237,
"step": 1000
},
{
"epoch": 0.81,
"learning_rate": 5.373567357842111e-07,
"logits/chosen": -2.6614608764648438,
"logits/rejected": -2.694235324859619,
"logps/chosen": -432.1556091308594,
"logps/rejected": -458.6222229003906,
"loss": 0.6625,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.2784090042114258,
"rewards/margins": 0.0895313173532486,
"rewards/rejected": -0.36794033646583557,
"step": 1010
},
{
"epoch": 0.82,
"learning_rate": 4.948465719110226e-07,
"logits/chosen": -2.6836206912994385,
"logits/rejected": -2.697563409805298,
"logps/chosen": -415.11041259765625,
"logps/rejected": -438.9033203125,
"loss": 0.6496,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.31501251459121704,
"rewards/margins": 0.11396019160747528,
"rewards/rejected": -0.4289727210998535,
"step": 1020
},
{
"epoch": 0.82,
"learning_rate": 4.539027495532766e-07,
"logits/chosen": -2.6660382747650146,
"logits/rejected": -2.6685073375701904,
"logps/chosen": -425.77490234375,
"logps/rejected": -446.0552673339844,
"loss": 0.6661,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.35333576798439026,
"rewards/margins": 0.08899492025375366,
"rewards/rejected": -0.4423306882381439,
"step": 1030
},
{
"epoch": 0.83,
"learning_rate": 4.14557252302783e-07,
"logits/chosen": -2.6789193153381348,
"logits/rejected": -2.6913952827453613,
"logps/chosen": -429.2127990722656,
"logps/rejected": -461.7294006347656,
"loss": 0.6592,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.3878576159477234,
"rewards/margins": 0.0975080356001854,
"rewards/rejected": -0.48536568880081177,
"step": 1040
},
{
"epoch": 0.84,
"learning_rate": 3.7684081520700884e-07,
"logits/chosen": -2.6964402198791504,
"logits/rejected": -2.70263934135437,
"logps/chosen": -462.135986328125,
"logps/rejected": -464.80035400390625,
"loss": 0.6855,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.437302827835083,
"rewards/margins": 0.041490085422992706,
"rewards/rejected": -0.4787929654121399,
"step": 1050
},
{
"epoch": 0.85,
"learning_rate": 3.407829007601507e-07,
"logits/chosen": -2.6650023460388184,
"logits/rejected": -2.6826632022857666,
"logps/chosen": -443.31982421875,
"logps/rejected": -482.0733947753906,
"loss": 0.6466,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.37219929695129395,
"rewards/margins": 0.13468633592128754,
"rewards/rejected": -0.5068856477737427,
"step": 1060
},
{
"epoch": 0.86,
"learning_rate": 3.064116758882724e-07,
"logits/chosen": -2.704041004180908,
"logits/rejected": -2.712378978729248,
"logps/chosen": -449.221435546875,
"logps/rejected": -488.82733154296875,
"loss": 0.6569,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.38919639587402344,
"rewards/margins": 0.10973678529262543,
"rewards/rejected": -0.49893322587013245,
"step": 1070
},
{
"epoch": 0.86,
"learning_rate": 2.737539899464908e-07,
"logits/chosen": -2.66728138923645,
"logits/rejected": -2.6824071407318115,
"logps/chosen": -408.9554138183594,
"logps/rejected": -435.3258361816406,
"loss": 0.6621,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.34276992082595825,
"rewards/margins": 0.08587387949228287,
"rewards/rejected": -0.4286437928676605,
"step": 1080
},
{
"epoch": 0.87,
"learning_rate": 2.4283535374538645e-07,
"logits/chosen": -2.731504440307617,
"logits/rejected": -2.733990430831909,
"logps/chosen": -450.702392578125,
"logps/rejected": -473.677490234375,
"loss": 0.6486,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.3203676640987396,
"rewards/margins": 0.12553098797798157,
"rewards/rejected": -0.4458986818790436,
"step": 1090
},
{
"epoch": 0.88,
"learning_rate": 2.1367991962303298e-07,
"logits/chosen": -2.6853950023651123,
"logits/rejected": -2.702267646789551,
"logps/chosen": -415.25048828125,
"logps/rejected": -443.6421813964844,
"loss": 0.6577,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.361422061920166,
"rewards/margins": 0.10248668491840363,
"rewards/rejected": -0.46390876173973083,
"step": 1100
},
{
"epoch": 0.88,
"eval_logits/chosen": -2.5729336738586426,
"eval_logits/rejected": -2.580476999282837,
"eval_logps/chosen": -351.83758544921875,
"eval_logps/rejected": -368.46795654296875,
"eval_loss": 0.6706418395042419,
"eval_rewards/accuracies": 0.5993333458900452,
"eval_rewards/chosen": -0.29342252016067505,
"eval_rewards/margins": 0.06933192163705826,
"eval_rewards/rejected": -0.3627544641494751,
"eval_runtime": 1589.982,
"eval_samples_per_second": 1.883,
"eval_steps_per_second": 0.236,
"step": 1100
},
{
"epoch": 0.89,
"learning_rate": 1.8631046257820278e-07,
"logits/chosen": -2.713442802429199,
"logits/rejected": -2.7153563499450684,
"logps/chosen": -448.9380798339844,
"logps/rejected": -459.6236267089844,
"loss": 0.6822,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.40070003271102905,
"rewards/margins": 0.04853721335530281,
"rewards/rejected": -0.44923728704452515,
"step": 1110
},
{
"epoch": 0.9,
"learning_rate": 1.6074836247950143e-07,
"logits/chosen": -2.7305519580841064,
"logits/rejected": -2.727928638458252,
"logps/chosen": -458.64208984375,
"logps/rejected": -479.4064025878906,
"loss": 0.6573,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.35037961602211,
"rewards/margins": 0.09948080033063889,
"rewards/rejected": -0.4498603940010071,
"step": 1120
},
{
"epoch": 0.9,
"learning_rate": 1.370135873643097e-07,
"logits/chosen": -2.6915557384490967,
"logits/rejected": -2.675395965576172,
"logps/chosen": -444.5713806152344,
"logps/rejected": -442.8504943847656,
"loss": 0.6738,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.3383481502532959,
"rewards/margins": 0.06244407966732979,
"rewards/rejected": -0.4007922112941742,
"step": 1130
},
{
"epoch": 0.91,
"learning_rate": 1.1512467784059372e-07,
"logits/chosen": -2.697908639907837,
"logits/rejected": -2.7043871879577637,
"logps/chosen": -409.3360595703125,
"logps/rejected": -430.56939697265625,
"loss": 0.6764,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.353982150554657,
"rewards/margins": 0.07203380763530731,
"rewards/rejected": -0.4260159432888031,
"step": 1140
},
{
"epoch": 0.92,
"learning_rate": 9.509873260376251e-08,
"logits/chosen": -2.6704964637756348,
"logits/rejected": -2.680117130279541,
"logps/chosen": -426.01568603515625,
"logps/rejected": -481.14056396484375,
"loss": 0.6301,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.292033851146698,
"rewards/margins": 0.16145673394203186,
"rewards/rejected": -0.45349058508872986,
"step": 1150
},
{
"epoch": 0.93,
"learning_rate": 7.695139507988559e-08,
"logits/chosen": -2.690162181854248,
"logits/rejected": -2.6840763092041016,
"logps/chosen": -455.21429443359375,
"logps/rejected": -471.9693908691406,
"loss": 0.6911,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.39844316244125366,
"rewards/margins": 0.04045991227030754,
"rewards/rejected": -0.4389030337333679,
"step": 1160
},
{
"epoch": 0.94,
"learning_rate": 6.069684120570684e-08,
"logits/chosen": -2.6355466842651367,
"logits/rejected": -2.6722214221954346,
"logps/chosen": -428.15887451171875,
"logps/rejected": -469.9241638183594,
"loss": 0.6556,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.31940728425979614,
"rewards/margins": 0.10891290009021759,
"rewards/rejected": -0.42832016944885254,
"step": 1170
},
{
"epoch": 0.94,
"learning_rate": 4.634776835499871e-08,
"logits/chosen": -2.648555278778076,
"logits/rejected": -2.648024082183838,
"logps/chosen": -414.77166748046875,
"logps/rejected": -438.7622985839844,
"loss": 0.6786,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.33630624413490295,
"rewards/margins": 0.059269875288009644,
"rewards/rejected": -0.3955760896205902,
"step": 1180
},
{
"epoch": 0.95,
"learning_rate": 3.3915385419908964e-08,
"logits/chosen": -2.724853992462158,
"logits/rejected": -2.7020461559295654,
"logps/chosen": -443.90399169921875,
"logps/rejected": -462.37384033203125,
"loss": 0.667,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.35228732228279114,
"rewards/margins": 0.08694271743297577,
"rewards/rejected": -0.4392300546169281,
"step": 1190
},
{
"epoch": 0.96,
"learning_rate": 2.3409404055043938e-08,
"logits/chosen": -2.6520884037017822,
"logits/rejected": -2.677919387817383,
"logps/chosen": -431.74053955078125,
"logps/rejected": -462.65179443359375,
"loss": 0.6444,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.3235710561275482,
"rewards/margins": 0.12747570872306824,
"rewards/rejected": -0.45104676485061646,
"step": 1200
},
{
"epoch": 0.96,
"eval_logits/chosen": -2.5725319385528564,
"eval_logits/rejected": -2.580101490020752,
"eval_logps/chosen": -351.09490966796875,
"eval_logps/rejected": -367.6591796875,
"eval_loss": 0.670768678188324,
"eval_rewards/accuracies": 0.5993333458900452,
"eval_rewards/chosen": -0.2859960198402405,
"eval_rewards/margins": 0.06867069751024246,
"eval_rewards/rejected": -0.35466670989990234,
"eval_runtime": 1585.5273,
"eval_samples_per_second": 1.888,
"eval_steps_per_second": 0.237,
"step": 1200
},
{
"epoch": 0.97,
"learning_rate": 1.4838031091134186e-08,
"logits/chosen": -2.6522107124328613,
"logits/rejected": -2.6851000785827637,
"logps/chosen": -404.271484375,
"logps/rejected": -455.322998046875,
"loss": 0.6397,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.3024134933948517,
"rewards/margins": 0.13375119864940643,
"rewards/rejected": -0.4361646771430969,
"step": 1210
},
{
"epoch": 0.98,
"learning_rate": 8.207962124201774e-09,
"logits/chosen": -2.687406063079834,
"logits/rejected": -2.6915080547332764,
"logps/chosen": -446.8517150878906,
"logps/rejected": -463.1326599121094,
"loss": 0.6692,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.33363211154937744,
"rewards/margins": 0.08175511658191681,
"rewards/rejected": -0.41538724303245544,
"step": 1220
},
{
"epoch": 0.98,
"learning_rate": 3.5243762852441023e-09,
"logits/chosen": -2.68792462348938,
"logits/rejected": -2.705341339111328,
"logps/chosen": -433.271484375,
"logps/rejected": -466.77362060546875,
"loss": 0.6609,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.34372276067733765,
"rewards/margins": 0.1005091518163681,
"rewards/rejected": -0.44423192739486694,
"step": 1230
},
{
"epoch": 0.99,
"learning_rate": 7.909321945129278e-10,
"logits/chosen": -2.704862117767334,
"logits/rejected": -2.7122840881347656,
"logps/chosen": -461.7477111816406,
"logps/rejected": -496.6314392089844,
"loss": 0.6514,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.3258623480796814,
"rewards/margins": 0.12274013459682465,
"rewards/rejected": -0.4486025273799896,
"step": 1240
},
{
"epoch": 1.0,
"step": 1249,
"total_flos": 0.0,
"train_loss": 0.67484232365751,
"train_runtime": 40604.4701,
"train_samples_per_second": 0.492,
"train_steps_per_second": 0.031
}
],
"logging_steps": 10,
"max_steps": 1249,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}