zephyr-7b-ipo-10k-40k-0.001-i1 / trainer_state.json
BraylonDash's picture
Model save
fd70a54 verified
raw
history blame contribute delete
No virus
60.7 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 500,
"global_step": 1250,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4e-08,
"logits/chosen": -3.1403026580810547,
"logits/rejected": -2.4295239448547363,
"logps/chosen": -392.2955017089844,
"logps/rejected": -613.5740966796875,
"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.442929267883301,
"logits/rejected": -2.321220636367798,
"logps/chosen": -261.98870849609375,
"logps/rejected": -221.4100799560547,
"loss": 0.6931,
"rewards/accuracies": 0.4722222089767456,
"rewards/chosen": 7.1744441811461e-05,
"rewards/margins": 5.5226653785211965e-05,
"rewards/rejected": 1.6517798940185457e-05,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 8.000000000000001e-07,
"logits/chosen": -2.42722225189209,
"logits/rejected": -2.285809278488159,
"logps/chosen": -244.7586212158203,
"logps/rejected": -254.1322784423828,
"loss": 0.6931,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -4.93793522764463e-05,
"rewards/margins": 0.00013055391900707036,
"rewards/rejected": -0.00017993326764553785,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 1.2000000000000002e-06,
"logits/chosen": -2.468625068664551,
"logits/rejected": -2.3896145820617676,
"logps/chosen": -251.3707733154297,
"logps/rejected": -300.4630126953125,
"loss": 0.6931,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.00025913942954503,
"rewards/margins": 9.763217531144619e-05,
"rewards/rejected": 0.00016150726878549904,
"step": 30
},
{
"epoch": 0.03,
"learning_rate": 1.6000000000000001e-06,
"logits/chosen": -2.483097553253174,
"logits/rejected": -2.135282516479492,
"logps/chosen": -252.2518310546875,
"logps/rejected": -206.49740600585938,
"loss": 0.693,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.0007439021137543023,
"rewards/margins": 0.0003426038019824773,
"rewards/rejected": 0.00040129836997948587,
"step": 40
},
{
"epoch": 0.04,
"learning_rate": 2.0000000000000003e-06,
"logits/chosen": -2.6477389335632324,
"logits/rejected": -2.486539363861084,
"logps/chosen": -245.1098175048828,
"logps/rejected": -234.5915985107422,
"loss": 0.6928,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.0010015963343903422,
"rewards/margins": 0.00047645685845054686,
"rewards/rejected": 0.0005251394468359649,
"step": 50
},
{
"epoch": 0.05,
"learning_rate": 2.4000000000000003e-06,
"logits/chosen": -2.5867533683776855,
"logits/rejected": -1.8461745977401733,
"logps/chosen": -288.46343994140625,
"logps/rejected": -191.20741271972656,
"loss": 0.6924,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.0025676097720861435,
"rewards/margins": 0.001473221811465919,
"rewards/rejected": 0.0010943879606202245,
"step": 60
},
{
"epoch": 0.06,
"learning_rate": 2.8000000000000003e-06,
"logits/chosen": -2.4274649620056152,
"logits/rejected": -2.21500825881958,
"logps/chosen": -206.21188354492188,
"logps/rejected": -188.66172790527344,
"loss": 0.6918,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.004373534582555294,
"rewards/margins": 0.0019022986525669694,
"rewards/rejected": 0.0024712360464036465,
"step": 70
},
{
"epoch": 0.06,
"learning_rate": 3.2000000000000003e-06,
"logits/chosen": -2.272698163986206,
"logits/rejected": -2.2475826740264893,
"logps/chosen": -218.1708984375,
"logps/rejected": -247.9695281982422,
"loss": 0.6919,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.00605297926813364,
"rewards/margins": 0.002590904710814357,
"rewards/rejected": 0.0034620738588273525,
"step": 80
},
{
"epoch": 0.07,
"learning_rate": 3.6000000000000003e-06,
"logits/chosen": -2.5878186225891113,
"logits/rejected": -2.174198627471924,
"logps/chosen": -284.7269287109375,
"logps/rejected": -222.1537322998047,
"loss": 0.6908,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.00458324421197176,
"rewards/margins": 0.005124131683260202,
"rewards/rejected": -0.0005408872966654599,
"step": 90
},
{
"epoch": 0.08,
"learning_rate": 4.000000000000001e-06,
"logits/chosen": -2.4876809120178223,
"logits/rejected": -2.304762363433838,
"logps/chosen": -258.6788330078125,
"logps/rejected": -230.6868133544922,
"loss": 0.6878,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.0009092213585972786,
"rewards/margins": 0.013424595817923546,
"rewards/rejected": -0.012515375390648842,
"step": 100
},
{
"epoch": 0.09,
"learning_rate": 4.4e-06,
"logits/chosen": -2.24761700630188,
"logits/rejected": -2.0283989906311035,
"logps/chosen": -242.88485717773438,
"logps/rejected": -253.8578643798828,
"loss": 0.6832,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.02561577782034874,
"rewards/margins": 0.02318538725376129,
"rewards/rejected": -0.04880116507411003,
"step": 110
},
{
"epoch": 0.1,
"learning_rate": 4.800000000000001e-06,
"logits/chosen": -2.0988376140594482,
"logits/rejected": -2.193089008331299,
"logps/chosen": -242.8556365966797,
"logps/rejected": -317.71917724609375,
"loss": 0.6753,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.047444917261600494,
"rewards/margins": 0.03924765810370445,
"rewards/rejected": -0.08669257164001465,
"step": 120
},
{
"epoch": 0.1,
"learning_rate": 4.999756310023261e-06,
"logits/chosen": -2.241037607192993,
"logits/rejected": -2.157954692840576,
"logps/chosen": -331.423828125,
"logps/rejected": -348.91046142578125,
"loss": 0.6688,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.08836604654788971,
"rewards/margins": 0.0334697850048542,
"rewards/rejected": -0.12183582782745361,
"step": 130
},
{
"epoch": 0.11,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": -1.7352321147918701,
"logits/rejected": -1.5876859426498413,
"logps/chosen": -398.1210021972656,
"logps/rejected": -428.30328369140625,
"loss": 0.6519,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.22779341042041779,
"rewards/margins": 0.0445188470184803,
"rewards/rejected": -0.2723122239112854,
"step": 140
},
{
"epoch": 0.12,
"learning_rate": 4.993910125649561e-06,
"logits/chosen": -1.735691785812378,
"logits/rejected": -1.7642030715942383,
"logps/chosen": -525.7499389648438,
"logps/rejected": -695.8289184570312,
"loss": 0.6354,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.30290618538856506,
"rewards/margins": 0.11975699663162231,
"rewards/rejected": -0.422663152217865,
"step": 150
},
{
"epoch": 0.13,
"learning_rate": 4.988068499954578e-06,
"logits/chosen": -2.004002332687378,
"logits/rejected": -1.7766345739364624,
"logps/chosen": -464.7952575683594,
"logps/rejected": -630.8657836914062,
"loss": 0.6461,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.23786354064941406,
"rewards/margins": 0.18317463994026184,
"rewards/rejected": -0.4210381507873535,
"step": 160
},
{
"epoch": 0.14,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": -2.0650768280029297,
"logits/rejected": -1.572537899017334,
"logps/chosen": -591.57421875,
"logps/rejected": -780.8646240234375,
"loss": 0.607,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.2891489565372467,
"rewards/margins": 0.27657198905944824,
"rewards/rejected": -0.5657209157943726,
"step": 170
},
{
"epoch": 0.14,
"learning_rate": 4.970570953616383e-06,
"logits/chosen": -1.4938254356384277,
"logits/rejected": -1.661007285118103,
"logps/chosen": -764.5491943359375,
"logps/rejected": -1000.9505615234375,
"loss": 0.6384,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5677521228790283,
"rewards/margins": 0.19688071310520172,
"rewards/rejected": -0.7646328806877136,
"step": 180
},
{
"epoch": 0.15,
"learning_rate": 4.958928677033465e-06,
"logits/chosen": -1.6295101642608643,
"logits/rejected": -1.3617414236068726,
"logps/chosen": -720.4210815429688,
"logps/rejected": -925.54638671875,
"loss": 0.6042,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.4372948706150055,
"rewards/margins": 0.2413189709186554,
"rewards/rejected": -0.6786138415336609,
"step": 190
},
{
"epoch": 0.16,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": -1.502726674079895,
"logits/rejected": -1.3334269523620605,
"logps/chosen": -752.0069580078125,
"logps/rejected": -1068.231201171875,
"loss": 0.64,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5011454820632935,
"rewards/margins": 0.2965936064720154,
"rewards/rejected": -0.7977390289306641,
"step": 200
},
{
"epoch": 0.17,
"learning_rate": 4.9299025014463665e-06,
"logits/chosen": -1.1884076595306396,
"logits/rejected": -1.0619341135025024,
"logps/chosen": -890.1341552734375,
"logps/rejected": -1238.742431640625,
"loss": 0.6007,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.6316002607345581,
"rewards/margins": 0.3692142069339752,
"rewards/rejected": -1.000814437866211,
"step": 210
},
{
"epoch": 0.18,
"learning_rate": 4.912541236180779e-06,
"logits/chosen": -1.0146244764328003,
"logits/rejected": -0.9316056966781616,
"logps/chosen": -979.2394409179688,
"logps/rejected": -1465.3270263671875,
"loss": 0.6018,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7414819002151489,
"rewards/margins": 0.49276018142700195,
"rewards/rejected": -1.2342422008514404,
"step": 220
},
{
"epoch": 0.18,
"learning_rate": 4.893298743830168e-06,
"logits/chosen": -1.2160202264785767,
"logits/rejected": -1.0113346576690674,
"logps/chosen": -1194.5838623046875,
"logps/rejected": -1710.5823974609375,
"loss": 0.5866,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9506030082702637,
"rewards/margins": 0.5406419634819031,
"rewards/rejected": -1.491244912147522,
"step": 230
},
{
"epoch": 0.19,
"learning_rate": 4.8721900291112415e-06,
"logits/chosen": -1.7798312902450562,
"logits/rejected": -1.6643425226211548,
"logps/chosen": -1057.5977783203125,
"logps/rejected": -1005.8741455078125,
"loss": 0.694,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7734619379043579,
"rewards/margins": -0.0247958954423666,
"rewards/rejected": -0.7486661076545715,
"step": 240
},
{
"epoch": 0.2,
"learning_rate": 4.849231551964771e-06,
"logits/chosen": -2.0089643001556396,
"logits/rejected": -1.7859022617340088,
"logps/chosen": -399.1302795410156,
"logps/rejected": -577.6984252929688,
"loss": 0.6424,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.1754738688468933,
"rewards/margins": 0.13275322318077087,
"rewards/rejected": -0.3082270920276642,
"step": 250
},
{
"epoch": 0.21,
"learning_rate": 4.824441214720629e-06,
"logits/chosen": -1.7344192266464233,
"logits/rejected": -1.632328748703003,
"logps/chosen": -632.4493408203125,
"logps/rejected": -760.2273559570312,
"loss": 0.6252,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.41359105706214905,
"rewards/margins": 0.12629616260528564,
"rewards/rejected": -0.5398872494697571,
"step": 260
},
{
"epoch": 0.22,
"learning_rate": 4.7978383481380865e-06,
"logits/chosen": -1.5548350811004639,
"logits/rejected": -1.5831079483032227,
"logps/chosen": -598.45556640625,
"logps/rejected": -712.7095336914062,
"loss": 0.6096,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3854503631591797,
"rewards/margins": 0.13576054573059082,
"rewards/rejected": -0.5212109088897705,
"step": 270
},
{
"epoch": 0.22,
"learning_rate": 4.769443696332272e-06,
"logits/chosen": -2.0036838054656982,
"logits/rejected": -1.7330118417739868,
"logps/chosen": -426.128662109375,
"logps/rejected": -617.08984375,
"loss": 0.6208,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.2103346288204193,
"rewards/margins": 0.19718368351459503,
"rewards/rejected": -0.40751829743385315,
"step": 280
},
{
"epoch": 0.23,
"learning_rate": 4.7392794005985324e-06,
"logits/chosen": -1.8511890172958374,
"logits/rejected": -1.5459794998168945,
"logps/chosen": -737.9730224609375,
"logps/rejected": -961.6851806640625,
"loss": 0.6214,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.48743873834609985,
"rewards/margins": 0.20717349648475647,
"rewards/rejected": -0.6946121454238892,
"step": 290
},
{
"epoch": 0.24,
"learning_rate": 4.707368982147318e-06,
"logits/chosen": -1.7424571514129639,
"logits/rejected": -1.4730459451675415,
"logps/chosen": -833.6031494140625,
"logps/rejected": -950.7183837890625,
"loss": 0.5805,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5092605352401733,
"rewards/margins": 0.1874401867389679,
"rewards/rejected": -0.6967006921768188,
"step": 300
},
{
"epoch": 0.25,
"learning_rate": 4.673737323763048e-06,
"logits/chosen": -1.373366355895996,
"logits/rejected": -1.2923896312713623,
"logps/chosen": -975.0471801757812,
"logps/rejected": -1144.96533203125,
"loss": 0.6091,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7069741487503052,
"rewards/margins": 0.21132870018482208,
"rewards/rejected": -0.9183028340339661,
"step": 310
},
{
"epoch": 0.26,
"learning_rate": 4.638410650401267e-06,
"logits/chosen": -1.477074384689331,
"logits/rejected": -1.2639570236206055,
"logps/chosen": -1050.070556640625,
"logps/rejected": -1336.6949462890625,
"loss": 0.5927,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.8159879446029663,
"rewards/margins": 0.2922418415546417,
"rewards/rejected": -1.1082298755645752,
"step": 320
},
{
"epoch": 0.26,
"learning_rate": 4.601416508739211e-06,
"logits/chosen": -1.5806219577789307,
"logits/rejected": -1.5797499418258667,
"logps/chosen": -1069.904296875,
"logps/rejected": -1329.1131591796875,
"loss": 0.6329,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.8489059209823608,
"rewards/margins": 0.2538915276527405,
"rewards/rejected": -1.102797508239746,
"step": 330
},
{
"epoch": 0.27,
"learning_rate": 4.562783745695738e-06,
"logits/chosen": -1.578185796737671,
"logits/rejected": -1.3778866529464722,
"logps/chosen": -898.2579956054688,
"logps/rejected": -1295.5048828125,
"loss": 0.5816,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6616091728210449,
"rewards/margins": 0.3867344856262207,
"rewards/rejected": -1.0483437776565552,
"step": 340
},
{
"epoch": 0.28,
"learning_rate": 4.522542485937369e-06,
"logits/chosen": -1.3196460008621216,
"logits/rejected": -1.0029346942901611,
"logps/chosen": -1030.9923095703125,
"logps/rejected": -1271.06494140625,
"loss": 0.6062,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7833272218704224,
"rewards/margins": 0.2811380922794342,
"rewards/rejected": -1.0644652843475342,
"step": 350
},
{
"epoch": 0.29,
"learning_rate": 4.4807241083879774e-06,
"logits/chosen": -1.4315626621246338,
"logits/rejected": -1.1289265155792236,
"logps/chosen": -1395.56640625,
"logps/rejected": -2238.73681640625,
"loss": 0.5686,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.1033650636672974,
"rewards/margins": 0.8617695569992065,
"rewards/rejected": -1.965134620666504,
"step": 360
},
{
"epoch": 0.3,
"learning_rate": 4.437361221760449e-06,
"logits/chosen": -1.0882983207702637,
"logits/rejected": -0.9719398617744446,
"logps/chosen": -1259.433349609375,
"logps/rejected": -1656.2109375,
"loss": 0.5973,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.069411277770996,
"rewards/margins": 0.3754344880580902,
"rewards/rejected": -1.4448457956314087,
"step": 370
},
{
"epoch": 0.3,
"learning_rate": 4.3924876391293915e-06,
"logits/chosen": -0.9243530035018921,
"logits/rejected": -0.9347097277641296,
"logps/chosen": -1276.580810546875,
"logps/rejected": -1660.425048828125,
"loss": 0.5932,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.1309508085250854,
"rewards/margins": 0.35707855224609375,
"rewards/rejected": -1.4880293607711792,
"step": 380
},
{
"epoch": 0.31,
"learning_rate": 4.346138351564711e-06,
"logits/chosen": -1.2673766613006592,
"logits/rejected": -1.1589628458023071,
"logps/chosen": -1206.547607421875,
"logps/rejected": -1615.82568359375,
"loss": 0.5588,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.9714127779006958,
"rewards/margins": 0.4280439019203186,
"rewards/rejected": -1.39945650100708,
"step": 390
},
{
"epoch": 0.32,
"learning_rate": 4.2983495008466285e-06,
"logits/chosen": -1.5774317979812622,
"logits/rejected": -1.2660281658172607,
"logps/chosen": -1082.2581787109375,
"logps/rejected": -1570.060791015625,
"loss": 0.5484,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.8231463432312012,
"rewards/margins": 0.5096359848976135,
"rewards/rejected": -1.3327823877334595,
"step": 400
},
{
"epoch": 0.33,
"learning_rate": 4.249158351283414e-06,
"logits/chosen": -1.4447729587554932,
"logits/rejected": -1.1427133083343506,
"logps/chosen": -1174.7425537109375,
"logps/rejected": -1466.5479736328125,
"loss": 0.5432,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9159573316574097,
"rewards/margins": 0.33625325560569763,
"rewards/rejected": -1.2522106170654297,
"step": 410
},
{
"epoch": 0.34,
"learning_rate": 4.198603260653792e-06,
"logits/chosen": -0.7784031629562378,
"logits/rejected": -0.5429641008377075,
"logps/chosen": -2375.2294921875,
"logps/rejected": -2860.4833984375,
"loss": 0.5023,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -2.1455845832824707,
"rewards/margins": 0.4987269341945648,
"rewards/rejected": -2.6443114280700684,
"step": 420
},
{
"epoch": 0.34,
"learning_rate": 4.146723650296701e-06,
"logits/chosen": -0.8310529589653015,
"logits/rejected": -0.47391629219055176,
"logps/chosen": -3060.8125,
"logps/rejected": -3744.44140625,
"loss": 0.6076,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.840507984161377,
"rewards/margins": 0.725513219833374,
"rewards/rejected": -3.56602144241333,
"step": 430
},
{
"epoch": 0.35,
"learning_rate": 4.093559974371725e-06,
"logits/chosen": -1.8024406433105469,
"logits/rejected": -1.4066709280014038,
"logps/chosen": -1206.559326171875,
"logps/rejected": -1958.577392578125,
"loss": 0.5517,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.948940098285675,
"rewards/margins": 0.76356440782547,
"rewards/rejected": -1.7125043869018555,
"step": 440
},
{
"epoch": 0.36,
"learning_rate": 4.039153688314146e-06,
"logits/chosen": -1.7638174295425415,
"logits/rejected": -1.6592317819595337,
"logps/chosen": -888.7286376953125,
"logps/rejected": -1274.7813720703125,
"loss": 0.5681,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6799342036247253,
"rewards/margins": 0.4119555950164795,
"rewards/rejected": -1.0918899774551392,
"step": 450
},
{
"epoch": 0.37,
"learning_rate": 3.983547216509254e-06,
"logits/chosen": -1.8817472457885742,
"logits/rejected": -1.766122817993164,
"logps/chosen": -1327.4205322265625,
"logps/rejected": -1799.326904296875,
"loss": 0.5546,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.9942048192024231,
"rewards/margins": 0.49855145812034607,
"rewards/rejected": -1.4927562475204468,
"step": 460
},
{
"epoch": 0.38,
"learning_rate": 3.92678391921108e-06,
"logits/chosen": -1.842795968055725,
"logits/rejected": -1.6158469915390015,
"logps/chosen": -1413.307373046875,
"logps/rejected": -1754.1558837890625,
"loss": 0.6049,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.0906217098236084,
"rewards/margins": 0.3714645802974701,
"rewards/rejected": -1.4620864391326904,
"step": 470
},
{
"epoch": 0.38,
"learning_rate": 3.868908058731376e-06,
"logits/chosen": -1.6520286798477173,
"logits/rejected": -1.5457327365875244,
"logps/chosen": -1587.302490234375,
"logps/rejected": -1965.464599609375,
"loss": 0.5766,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.3615858554840088,
"rewards/margins": 0.39060306549072266,
"rewards/rejected": -1.7521886825561523,
"step": 480
},
{
"epoch": 0.39,
"learning_rate": 3.8099647649251984e-06,
"logits/chosen": -1.8340473175048828,
"logits/rejected": -1.6425994634628296,
"logps/chosen": -1667.3792724609375,
"logps/rejected": -2022.701904296875,
"loss": 0.5243,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.4147528409957886,
"rewards/margins": 0.4026387333869934,
"rewards/rejected": -1.8173916339874268,
"step": 490
},
{
"epoch": 0.4,
"learning_rate": 3.7500000000000005e-06,
"logits/chosen": -1.565735936164856,
"logits/rejected": -1.4348723888397217,
"logps/chosen": -1869.2935791015625,
"logps/rejected": -2379.262939453125,
"loss": 0.5181,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.658975601196289,
"rewards/margins": 0.5309830904006958,
"rewards/rejected": -2.1899585723876953,
"step": 500
},
{
"epoch": 0.41,
"learning_rate": 3.689060522675689e-06,
"logits/chosen": -1.8067477941513062,
"logits/rejected": -1.4334272146224976,
"logps/chosen": -1798.0201416015625,
"logps/rejected": -2005.1083984375,
"loss": 0.5859,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.5768072605133057,
"rewards/margins": 0.24011941254138947,
"rewards/rejected": -1.8169265985488892,
"step": 510
},
{
"epoch": 0.42,
"learning_rate": 3.627193851723577e-06,
"logits/chosen": -1.9058891534805298,
"logits/rejected": -1.5708621740341187,
"logps/chosen": -1344.331787109375,
"logps/rejected": -2066.19677734375,
"loss": 0.5302,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.0564631223678589,
"rewards/margins": 0.7303067445755005,
"rewards/rejected": -1.7867698669433594,
"step": 520
},
{
"epoch": 0.42,
"learning_rate": 3.564448228912682e-06,
"logits/chosen": -1.8703407049179077,
"logits/rejected": -1.8959243297576904,
"logps/chosen": -1418.962890625,
"logps/rejected": -1918.7298583984375,
"loss": 0.5437,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.2395734786987305,
"rewards/margins": 0.4607962965965271,
"rewards/rejected": -1.7003698348999023,
"step": 530
},
{
"epoch": 0.43,
"learning_rate": 3.5008725813922383e-06,
"logits/chosen": -1.4778034687042236,
"logits/rejected": -1.42378830909729,
"logps/chosen": -2291.814453125,
"logps/rejected": -2982.93212890625,
"loss": 0.4967,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.091310977935791,
"rewards/margins": 0.6848040819168091,
"rewards/rejected": -2.7761149406433105,
"step": 540
},
{
"epoch": 0.44,
"learning_rate": 3.436516483539781e-06,
"logits/chosen": -1.495721459388733,
"logits/rejected": -1.364970326423645,
"logps/chosen": -2159.25634765625,
"logps/rejected": -2941.31982421875,
"loss": 0.5244,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.9415247440338135,
"rewards/margins": 0.7436372637748718,
"rewards/rejected": -2.685161828994751,
"step": 550
},
{
"epoch": 0.45,
"learning_rate": 3.3714301183045382e-06,
"logits/chosen": -1.7252031564712524,
"logits/rejected": -1.3431731462478638,
"logps/chosen": -2346.15673828125,
"logps/rejected": -2880.50341796875,
"loss": 0.5818,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.072269916534424,
"rewards/margins": 0.5818148851394653,
"rewards/rejected": -2.6540849208831787,
"step": 560
},
{
"epoch": 0.46,
"learning_rate": 3.3056642380762783e-06,
"logits/chosen": -1.7475759983062744,
"logits/rejected": -1.6887413263320923,
"logps/chosen": -2365.16650390625,
"logps/rejected": -3232.527587890625,
"loss": 0.4838,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.112750291824341,
"rewards/margins": 0.815180778503418,
"rewards/rejected": -2.927931070327759,
"step": 570
},
{
"epoch": 0.46,
"learning_rate": 3.2392701251101172e-06,
"logits/chosen": -1.6227779388427734,
"logits/rejected": -1.371058702468872,
"logps/chosen": -2093.24365234375,
"logps/rejected": -2805.071533203125,
"loss": 0.5206,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.8952558040618896,
"rewards/margins": 0.6814407706260681,
"rewards/rejected": -2.5766968727111816,
"step": 580
},
{
"epoch": 0.47,
"learning_rate": 3.1722995515381644e-06,
"logits/chosen": -1.6335232257843018,
"logits/rejected": -1.2477694749832153,
"logps/chosen": -2436.821044921875,
"logps/rejected": -3250.9990234375,
"loss": 0.5081,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.2077012062072754,
"rewards/margins": 0.8523558378219604,
"rewards/rejected": -3.0600569248199463,
"step": 590
},
{
"epoch": 0.48,
"learning_rate": 3.1048047389991693e-06,
"logits/chosen": -1.898289442062378,
"logits/rejected": -1.5713775157928467,
"logps/chosen": -1884.5660400390625,
"logps/rejected": -2746.454833984375,
"loss": 0.5623,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.6871427297592163,
"rewards/margins": 0.8711158633232117,
"rewards/rejected": -2.5582587718963623,
"step": 600
},
{
"epoch": 0.49,
"learning_rate": 3.0368383179176584e-06,
"logits/chosen": -1.4297007322311401,
"logits/rejected": -1.3948471546173096,
"logps/chosen": -2035.6175537109375,
"logps/rejected": -3012.0087890625,
"loss": 0.5395,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.860030174255371,
"rewards/margins": 0.9378849267959595,
"rewards/rejected": -2.797914981842041,
"step": 610
},
{
"epoch": 0.5,
"learning_rate": 2.9684532864643123e-06,
"logits/chosen": -1.7777831554412842,
"logits/rejected": -1.3341039419174194,
"logps/chosen": -2097.790283203125,
"logps/rejected": -2818.270263671875,
"loss": 0.4943,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.8172571659088135,
"rewards/margins": 0.7645599246025085,
"rewards/rejected": -2.581817150115967,
"step": 620
},
{
"epoch": 0.5,
"learning_rate": 2.8997029692295875e-06,
"logits/chosen": -1.7123359441757202,
"logits/rejected": -1.4930963516235352,
"logps/chosen": -1850.494873046875,
"logps/rejected": -2223.23779296875,
"loss": 0.5415,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.5890159606933594,
"rewards/margins": 0.4220260679721832,
"rewards/rejected": -2.0110421180725098,
"step": 630
},
{
"epoch": 0.51,
"learning_rate": 2.8306409756428067e-06,
"logits/chosen": -1.7620747089385986,
"logits/rejected": -1.273874282836914,
"logps/chosen": -1751.422119140625,
"logps/rejected": -2349.684814453125,
"loss": 0.5267,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.5236599445343018,
"rewards/margins": 0.6361467838287354,
"rewards/rejected": -2.159806728363037,
"step": 640
},
{
"epoch": 0.52,
"learning_rate": 2.761321158169134e-06,
"logits/chosen": -1.8131462335586548,
"logits/rejected": -1.548789381980896,
"logps/chosen": -1939.831298828125,
"logps/rejected": -2459.304443359375,
"loss": 0.5526,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.7154333591461182,
"rewards/margins": 0.5310146808624268,
"rewards/rejected": -2.246448040008545,
"step": 650
},
{
"epoch": 0.53,
"learning_rate": 2.6917975703170466e-06,
"logits/chosen": -1.4764631986618042,
"logits/rejected": -1.2985259294509888,
"logps/chosen": -2074.831787109375,
"logps/rejected": -2988.404296875,
"loss": 0.5177,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.8811687231063843,
"rewards/margins": 0.8929063677787781,
"rewards/rejected": -2.7740752696990967,
"step": 660
},
{
"epoch": 0.54,
"learning_rate": 2.6221244244890336e-06,
"logits/chosen": -1.606785774230957,
"logits/rejected": -1.4506292343139648,
"logps/chosen": -2944.038330078125,
"logps/rejected": -3770.83203125,
"loss": 0.5202,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.648715019226074,
"rewards/margins": 0.8418272733688354,
"rewards/rejected": -3.4905428886413574,
"step": 670
},
{
"epoch": 0.54,
"learning_rate": 2.5523560497083927e-06,
"logits/chosen": -1.7594387531280518,
"logits/rejected": -1.4435147047042847,
"logps/chosen": -2691.794921875,
"logps/rejected": -2907.418212890625,
"loss": 0.6347,
"rewards/accuracies": 0.5,
"rewards/chosen": -2.4359123706817627,
"rewards/margins": 0.3157985210418701,
"rewards/rejected": -2.751710891723633,
"step": 680
},
{
"epoch": 0.55,
"learning_rate": 2.482546849255096e-06,
"logits/chosen": -1.6512796878814697,
"logits/rejected": -1.2861438989639282,
"logps/chosen": -2340.228515625,
"logps/rejected": -3065.31591796875,
"loss": 0.4919,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -2.0771026611328125,
"rewards/margins": 0.6874440908432007,
"rewards/rejected": -2.7645466327667236,
"step": 690
},
{
"epoch": 0.56,
"learning_rate": 2.4127512582437486e-06,
"logits/chosen": -1.7903190851211548,
"logits/rejected": -1.5637561082839966,
"logps/chosen": -1961.098876953125,
"logps/rejected": -2661.05126953125,
"loss": 0.5352,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -1.727064847946167,
"rewards/margins": 0.7038751840591431,
"rewards/rejected": -2.4309401512145996,
"step": 700
},
{
"epoch": 0.57,
"learning_rate": 2.3430237011767166e-06,
"logits/chosen": -1.5963449478149414,
"logits/rejected": -1.5752345323562622,
"logps/chosen": -2412.08447265625,
"logps/rejected": -2860.287109375,
"loss": 0.5364,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.1989083290100098,
"rewards/margins": 0.4225144386291504,
"rewards/rejected": -2.62142276763916,
"step": 710
},
{
"epoch": 0.58,
"learning_rate": 2.2734185495055503e-06,
"logits/chosen": -1.581274390220642,
"logits/rejected": -1.338945746421814,
"logps/chosen": -2689.8212890625,
"logps/rejected": -3143.3369140625,
"loss": 0.5331,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.4156579971313477,
"rewards/margins": 0.4918283522129059,
"rewards/rejected": -2.9074864387512207,
"step": 720
},
{
"epoch": 0.58,
"learning_rate": 2.2039900792337477e-06,
"logits/chosen": -1.6093494892120361,
"logits/rejected": -1.2763807773590088,
"logps/chosen": -1861.92578125,
"logps/rejected": -2657.68212890625,
"loss": 0.4712,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.6057929992675781,
"rewards/margins": 0.847273051738739,
"rewards/rejected": -2.453066110610962,
"step": 730
},
{
"epoch": 0.59,
"learning_rate": 2.134792428593971e-06,
"logits/chosen": -1.3526313304901123,
"logits/rejected": -1.3224549293518066,
"logps/chosen": -2195.27197265625,
"logps/rejected": -2851.3291015625,
"loss": 0.5536,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.023761749267578,
"rewards/margins": 0.6121652722358704,
"rewards/rejected": -2.635927200317383,
"step": 740
},
{
"epoch": 0.6,
"learning_rate": 2.0658795558326745e-06,
"logits/chosen": -1.6000654697418213,
"logits/rejected": -1.4151827096939087,
"logps/chosen": -2148.247802734375,
"logps/rejected": -2729.99853515625,
"loss": 0.5222,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.903270959854126,
"rewards/margins": 0.5996490716934204,
"rewards/rejected": -2.5029196739196777,
"step": 750
},
{
"epoch": 0.61,
"learning_rate": 1.997305197135089e-06,
"logits/chosen": -1.8795032501220703,
"logits/rejected": -1.8359801769256592,
"logps/chosen": -2000.689208984375,
"logps/rejected": -2926.07275390625,
"loss": 0.5271,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.7533314228057861,
"rewards/margins": 0.9173868298530579,
"rewards/rejected": -2.6707184314727783,
"step": 760
},
{
"epoch": 0.62,
"learning_rate": 1.9291228247233607e-06,
"logits/chosen": -1.6307891607284546,
"logits/rejected": -1.496146321296692,
"logps/chosen": -1922.004150390625,
"logps/rejected": -2854.74365234375,
"loss": 0.4909,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.7167818546295166,
"rewards/margins": 0.9143074750900269,
"rewards/rejected": -2.631089448928833,
"step": 770
},
{
"epoch": 0.62,
"learning_rate": 1.8613856051605242e-06,
"logits/chosen": -1.5307934284210205,
"logits/rejected": -1.4985663890838623,
"logps/chosen": -1941.616455078125,
"logps/rejected": -2720.62109375,
"loss": 0.495,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.7477505207061768,
"rewards/margins": 0.7529212236404419,
"rewards/rejected": -2.500671863555908,
"step": 780
},
{
"epoch": 0.63,
"learning_rate": 1.7941463578928088e-06,
"logits/chosen": -1.6022170782089233,
"logits/rejected": -1.5732767581939697,
"logps/chosen": -2065.362548828125,
"logps/rejected": -2565.924560546875,
"loss": 0.5311,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.8574340343475342,
"rewards/margins": 0.4975149631500244,
"rewards/rejected": -2.3549489974975586,
"step": 790
},
{
"epoch": 0.64,
"learning_rate": 1.7274575140626318e-06,
"logits/chosen": -1.5523189306259155,
"logits/rejected": -1.2900121212005615,
"logps/chosen": -2249.78466796875,
"logps/rejected": -3071.79541015625,
"loss": 0.5322,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.0411088466644287,
"rewards/margins": 0.8322113752365112,
"rewards/rejected": -2.8733203411102295,
"step": 800
},
{
"epoch": 0.65,
"learning_rate": 1.661371075624363e-06,
"logits/chosen": -1.750195860862732,
"logits/rejected": -1.4256199598312378,
"logps/chosen": -2680.12548828125,
"logps/rejected": -3339.82177734375,
"loss": 0.5582,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -2.413327693939209,
"rewards/margins": 0.6952977180480957,
"rewards/rejected": -3.108625650405884,
"step": 810
},
{
"epoch": 0.66,
"learning_rate": 1.5959385747947697e-06,
"logits/chosen": -1.6673352718353271,
"logits/rejected": -1.4309087991714478,
"logps/chosen": -2934.502685546875,
"logps/rejected": -3557.366455078125,
"loss": 0.5826,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.6931636333465576,
"rewards/margins": 0.6061044931411743,
"rewards/rejected": -3.2992682456970215,
"step": 820
},
{
"epoch": 0.66,
"learning_rate": 1.5312110338697427e-06,
"logits/chosen": -1.4265161752700806,
"logits/rejected": -1.3840219974517822,
"logps/chosen": -2902.10205078125,
"logps/rejected": -3211.46484375,
"loss": 0.5793,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.7132956981658936,
"rewards/margins": 0.2958894670009613,
"rewards/rejected": -3.009185314178467,
"step": 830
},
{
"epoch": 0.67,
"learning_rate": 1.467238925438646e-06,
"logits/chosen": -1.93195378780365,
"logits/rejected": -1.7828657627105713,
"logps/chosen": -2659.49609375,
"logps/rejected": -3121.871826171875,
"loss": 0.5751,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -2.427194118499756,
"rewards/margins": 0.47958964109420776,
"rewards/rejected": -2.9067838191986084,
"step": 840
},
{
"epoch": 0.68,
"learning_rate": 1.4040721330273063e-06,
"logits/chosen": -1.6783872842788696,
"logits/rejected": -1.3621320724487305,
"logps/chosen": -2476.935791015625,
"logps/rejected": -3240.056640625,
"loss": 0.5022,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -2.256714344024658,
"rewards/margins": 0.7921364307403564,
"rewards/rejected": -3.0488505363464355,
"step": 850
},
{
"epoch": 0.69,
"learning_rate": 1.3417599122003464e-06,
"logits/chosen": -1.631744146347046,
"logits/rejected": -1.4881222248077393,
"logps/chosen": -2611.767333984375,
"logps/rejected": -3342.803466796875,
"loss": 0.5239,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.361072301864624,
"rewards/margins": 0.6957711577415466,
"rewards/rejected": -3.0568435192108154,
"step": 860
},
{
"epoch": 0.7,
"learning_rate": 1.280350852153168e-06,
"logits/chosen": -1.6476237773895264,
"logits/rejected": -1.2135014533996582,
"logps/chosen": -2210.89697265625,
"logps/rejected": -3153.556884765625,
"loss": 0.49,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -1.9529176950454712,
"rewards/margins": 0.9891131520271301,
"rewards/rejected": -2.942031145095825,
"step": 870
},
{
"epoch": 0.7,
"learning_rate": 1.2198928378235717e-06,
"logits/chosen": -1.8185428380966187,
"logits/rejected": -1.4744179248809814,
"logps/chosen": -1982.8447265625,
"logps/rejected": -2986.81591796875,
"loss": 0.4908,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.720696210861206,
"rewards/margins": 1.0247818231582642,
"rewards/rejected": -2.7454781532287598,
"step": 880
},
{
"epoch": 0.71,
"learning_rate": 1.160433012552508e-06,
"logits/chosen": -1.9028065204620361,
"logits/rejected": -1.5971095561981201,
"logps/chosen": -1988.5980224609375,
"logps/rejected": -2843.60302734375,
"loss": 0.5298,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.695246934890747,
"rewards/margins": 0.9005265235900879,
"rewards/rejected": -2.595773458480835,
"step": 890
},
{
"epoch": 0.72,
"learning_rate": 1.1020177413231334e-06,
"logits/chosen": -1.7520854473114014,
"logits/rejected": -1.4778826236724854,
"logps/chosen": -1867.814453125,
"logps/rejected": -2722.72900390625,
"loss": 0.5195,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.6399002075195312,
"rewards/margins": 0.8625130653381348,
"rewards/rejected": -2.502413272857666,
"step": 900
},
{
"epoch": 0.73,
"learning_rate": 1.0446925746067768e-06,
"logits/chosen": -1.6180340051651,
"logits/rejected": -1.4504057168960571,
"logps/chosen": -2156.244140625,
"logps/rejected": -3364.479736328125,
"loss": 0.4724,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.9639689922332764,
"rewards/margins": 1.1746752262115479,
"rewards/rejected": -3.138643980026245,
"step": 910
},
{
"epoch": 0.74,
"learning_rate": 9.88502212844063e-07,
"logits/chosen": -1.8366491794586182,
"logits/rejected": -1.353752851486206,
"logps/chosen": -1940.987060546875,
"logps/rejected": -2697.123779296875,
"loss": 0.506,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.714422583580017,
"rewards/margins": 0.7562412023544312,
"rewards/rejected": -2.4706637859344482,
"step": 920
},
{
"epoch": 0.74,
"learning_rate": 9.334904715888496e-07,
"logits/chosen": -1.8136298656463623,
"logits/rejected": -1.5429127216339111,
"logps/chosen": -2025.2021484375,
"logps/rejected": -2842.939453125,
"loss": 0.5274,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.792414903640747,
"rewards/margins": 0.8433243036270142,
"rewards/rejected": -2.63573956489563,
"step": 930
},
{
"epoch": 0.75,
"learning_rate": 8.797002473421729e-07,
"logits/chosen": -1.6221929788589478,
"logits/rejected": -1.536675214767456,
"logps/chosen": -2179.556884765625,
"logps/rejected": -2724.71142578125,
"loss": 0.5293,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.8998562097549438,
"rewards/margins": 0.5943135023117065,
"rewards/rejected": -2.4941699504852295,
"step": 940
},
{
"epoch": 0.76,
"learning_rate": 8.271734841028553e-07,
"logits/chosen": -1.599360704421997,
"logits/rejected": -1.4945180416107178,
"logps/chosen": -2247.677001953125,
"logps/rejected": -2832.541015625,
"loss": 0.5645,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.041861057281494,
"rewards/margins": 0.5804768800735474,
"rewards/rejected": -2.622338056564331,
"step": 950
},
{
"epoch": 0.77,
"learning_rate": 7.759511406608255e-07,
"logits/chosen": -1.9042466878890991,
"logits/rejected": -1.5334478616714478,
"logps/chosen": -1946.2919921875,
"logps/rejected": -2741.82861328125,
"loss": 0.466,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.713287115097046,
"rewards/margins": 0.8301644325256348,
"rewards/rejected": -2.5434517860412598,
"step": 960
},
{
"epoch": 0.78,
"learning_rate": 7.260731586586983e-07,
"logits/chosen": -1.5018055438995361,
"logits/rejected": -1.4937880039215088,
"logps/chosen": -2611.099609375,
"logps/rejected": -3151.30810546875,
"loss": 0.5775,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.3841938972473145,
"rewards/margins": 0.5148266553878784,
"rewards/rejected": -2.8990206718444824,
"step": 970
},
{
"epoch": 0.78,
"learning_rate": 6.775784314464717e-07,
"logits/chosen": -1.6764729022979736,
"logits/rejected": -1.5057140588760376,
"logps/chosen": -2298.930908203125,
"logps/rejected": -2712.99169921875,
"loss": 0.556,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -2.0648088455200195,
"rewards/margins": 0.4065318703651428,
"rewards/rejected": -2.4713406562805176,
"step": 980
},
{
"epoch": 0.79,
"learning_rate": 6.305047737536707e-07,
"logits/chosen": -1.7027795314788818,
"logits/rejected": -1.3025496006011963,
"logps/chosen": -2075.1337890625,
"logps/rejected": -2931.57080078125,
"loss": 0.4584,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.846689224243164,
"rewards/margins": 0.8770244717597961,
"rewards/rejected": -2.7237138748168945,
"step": 990
},
{
"epoch": 0.8,
"learning_rate": 5.848888922025553e-07,
"logits/chosen": -1.8110439777374268,
"logits/rejected": -1.4326590299606323,
"logps/chosen": -2325.86474609375,
"logps/rejected": -3089.902099609375,
"loss": 0.5249,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.067896604537964,
"rewards/margins": 0.7900792360305786,
"rewards/rejected": -2.857975482940674,
"step": 1000
},
{
"epoch": 0.81,
"learning_rate": 5.407663566854008e-07,
"logits/chosen": -1.6088895797729492,
"logits/rejected": -1.4155261516571045,
"logps/chosen": -2324.50927734375,
"logps/rejected": -3341.44970703125,
"loss": 0.525,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.1127965450286865,
"rewards/margins": 0.9637571573257446,
"rewards/rejected": -3.0765535831451416,
"step": 1010
},
{
"epoch": 0.82,
"learning_rate": 4.981715726281666e-07,
"logits/chosen": -1.6132185459136963,
"logits/rejected": -1.2942755222320557,
"logps/chosen": -2351.848388671875,
"logps/rejected": -3291.189453125,
"loss": 0.5183,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -2.1332767009735107,
"rewards/margins": 0.9448688626289368,
"rewards/rejected": -3.0781455039978027,
"step": 1020
},
{
"epoch": 0.82,
"learning_rate": 4.5713775416217884e-07,
"logits/chosen": -1.770582914352417,
"logits/rejected": -1.515246868133545,
"logps/chosen": -2241.62109375,
"logps/rejected": -3231.28125,
"loss": 0.4636,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.9664819240570068,
"rewards/margins": 1.0160348415374756,
"rewards/rejected": -2.9825170040130615,
"step": 1030
},
{
"epoch": 0.83,
"learning_rate": 4.1769689822475147e-07,
"logits/chosen": -1.5879560708999634,
"logits/rejected": -1.5036201477050781,
"logps/chosen": -1936.926025390625,
"logps/rejected": -2708.7578125,
"loss": 0.4947,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.7738593816757202,
"rewards/margins": 0.7560992240905762,
"rewards/rejected": -2.529958486557007,
"step": 1040
},
{
"epoch": 0.84,
"learning_rate": 3.798797596089351e-07,
"logits/chosen": -1.4568753242492676,
"logits/rejected": -1.477905511856079,
"logps/chosen": -2428.654296875,
"logps/rejected": -3266.05712890625,
"loss": 0.5201,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.2049851417541504,
"rewards/margins": 0.8197803497314453,
"rewards/rejected": -3.0247654914855957,
"step": 1050
},
{
"epoch": 0.85,
"learning_rate": 3.4371582698185636e-07,
"logits/chosen": -1.6677932739257812,
"logits/rejected": -1.5246306657791138,
"logps/chosen": -2556.70751953125,
"logps/rejected": -3194.56884765625,
"loss": 0.5391,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.3062329292297363,
"rewards/margins": 0.6159250140190125,
"rewards/rejected": -2.9221577644348145,
"step": 1060
},
{
"epoch": 0.86,
"learning_rate": 3.092332998903416e-07,
"logits/chosen": -1.7090551853179932,
"logits/rejected": -1.4194791316986084,
"logps/chosen": -2150.77392578125,
"logps/rejected": -2783.882568359375,
"loss": 0.5195,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.9338016510009766,
"rewards/margins": 0.6355168223381042,
"rewards/rejected": -2.5693182945251465,
"step": 1070
},
{
"epoch": 0.86,
"learning_rate": 2.764590667717562e-07,
"logits/chosen": -1.606188178062439,
"logits/rejected": -1.4986306428909302,
"logps/chosen": -2124.823974609375,
"logps/rejected": -2895.18310546875,
"loss": 0.4628,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.912660837173462,
"rewards/margins": 0.7604308724403381,
"rewards/rejected": -2.673091411590576,
"step": 1080
},
{
"epoch": 0.87,
"learning_rate": 2.454186839872158e-07,
"logits/chosen": -1.5546796321868896,
"logits/rejected": -1.550550103187561,
"logps/chosen": -2096.05078125,
"logps/rejected": -2596.4248046875,
"loss": 0.5839,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.9206135272979736,
"rewards/margins": 0.4739322066307068,
"rewards/rejected": -2.394545793533325,
"step": 1090
},
{
"epoch": 0.88,
"learning_rate": 2.1613635589349756e-07,
"logits/chosen": -1.657184362411499,
"logits/rejected": -1.3406727313995361,
"logps/chosen": -2227.37841796875,
"logps/rejected": -3146.83349609375,
"loss": 0.459,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.9884710311889648,
"rewards/margins": 0.9205359220504761,
"rewards/rejected": -2.9090065956115723,
"step": 1100
},
{
"epoch": 0.89,
"learning_rate": 1.8863491596921745e-07,
"logits/chosen": -1.4567536115646362,
"logits/rejected": -1.1964176893234253,
"logps/chosen": -2488.02099609375,
"logps/rejected": -3093.317626953125,
"loss": 0.5213,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.2580440044403076,
"rewards/margins": 0.6173666715621948,
"rewards/rejected": -2.875410556793213,
"step": 1110
},
{
"epoch": 0.9,
"learning_rate": 1.629358090099639e-07,
"logits/chosen": -1.8676036596298218,
"logits/rejected": -1.5324196815490723,
"logps/chosen": -2476.497314453125,
"logps/rejected": -3140.97607421875,
"loss": 0.5292,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.2105040550231934,
"rewards/margins": 0.6873103380203247,
"rewards/rejected": -2.8978142738342285,
"step": 1120
},
{
"epoch": 0.9,
"learning_rate": 1.3905907440629752e-07,
"logits/chosen": -1.5062824487686157,
"logits/rejected": -1.7191336154937744,
"logps/chosen": -2101.674560546875,
"logps/rejected": -2914.34228515625,
"loss": 0.4889,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.8925731182098389,
"rewards/margins": 0.7410818338394165,
"rewards/rejected": -2.633654832839966,
"step": 1130
},
{
"epoch": 0.91,
"learning_rate": 1.1702333051763271e-07,
"logits/chosen": -1.885053277015686,
"logits/rejected": -1.7132413387298584,
"logps/chosen": -2271.51806640625,
"logps/rejected": -3019.444580078125,
"loss": 0.4746,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.0059704780578613,
"rewards/margins": 0.8035289645195007,
"rewards/rejected": -2.809499740600586,
"step": 1140
},
{
"epoch": 0.92,
"learning_rate": 9.684576015420277e-08,
"logits/chosen": -1.7674312591552734,
"logits/rejected": -1.7866573333740234,
"logps/chosen": -2304.083740234375,
"logps/rejected": -3042.02783203125,
"loss": 0.5235,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.0314338207244873,
"rewards/margins": 0.7121332883834839,
"rewards/rejected": -2.7435669898986816,
"step": 1150
},
{
"epoch": 0.93,
"learning_rate": 7.854209717842231e-08,
"logits/chosen": -1.6722157001495361,
"logits/rejected": -1.738372564315796,
"logps/chosen": -2533.139404296875,
"logps/rejected": -3169.639892578125,
"loss": 0.508,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -2.2765398025512695,
"rewards/margins": 0.6401499509811401,
"rewards/rejected": -2.91668963432312,
"step": 1160
},
{
"epoch": 0.94,
"learning_rate": 6.212661423609184e-08,
"logits/chosen": -1.6138194799423218,
"logits/rejected": -1.4168260097503662,
"logps/chosen": -2651.40087890625,
"logps/rejected": -2953.78564453125,
"loss": 0.5389,
"rewards/accuracies": 0.5,
"rewards/chosen": -2.369499683380127,
"rewards/margins": 0.3613986372947693,
"rewards/rejected": -2.730898380279541,
"step": 1170
},
{
"epoch": 0.94,
"learning_rate": 4.761211162702117e-08,
"logits/chosen": -1.8268349170684814,
"logits/rejected": -1.389704942703247,
"logps/chosen": -2308.206298828125,
"logps/rejected": -3084.591796875,
"loss": 0.4561,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.0564706325531006,
"rewards/margins": 0.8305153846740723,
"rewards/rejected": -2.8869857788085938,
"step": 1180
},
{
"epoch": 0.95,
"learning_rate": 3.5009907323737826e-08,
"logits/chosen": -1.5044516324996948,
"logits/rejected": -1.3144376277923584,
"logps/chosen": -2185.82275390625,
"logps/rejected": -3079.09130859375,
"loss": 0.4552,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.9693256616592407,
"rewards/margins": 0.8803497552871704,
"rewards/rejected": -2.8496756553649902,
"step": 1190
},
{
"epoch": 0.96,
"learning_rate": 2.4329828146074096e-08,
"logits/chosen": -1.5807933807373047,
"logits/rejected": -1.4424628019332886,
"logps/chosen": -2456.582763671875,
"logps/rejected": -2916.781982421875,
"loss": 0.5377,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -2.223078489303589,
"rewards/margins": 0.4909304976463318,
"rewards/rejected": -2.7140090465545654,
"step": 1200
},
{
"epoch": 0.97,
"learning_rate": 1.5580202098509078e-08,
"logits/chosen": -1.4222395420074463,
"logits/rejected": -1.4579776525497437,
"logps/chosen": -2549.04052734375,
"logps/rejected": -3383.83203125,
"loss": 0.4934,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.3147170543670654,
"rewards/margins": 0.7796028852462769,
"rewards/rejected": -3.094320058822632,
"step": 1210
},
{
"epoch": 0.98,
"learning_rate": 8.767851876239075e-09,
"logits/chosen": -1.9085347652435303,
"logits/rejected": -1.5333627462387085,
"logps/chosen": -2555.34228515625,
"logps/rejected": -3125.24853515625,
"loss": 0.5644,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.2714717388153076,
"rewards/margins": 0.6149500012397766,
"rewards/rejected": -2.8864216804504395,
"step": 1220
},
{
"epoch": 0.98,
"learning_rate": 3.8980895450474455e-09,
"logits/chosen": -1.6516891717910767,
"logits/rejected": -1.3506808280944824,
"logps/chosen": -2481.84228515625,
"logps/rejected": -3454.458984375,
"loss": 0.4893,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.2125697135925293,
"rewards/margins": 1.0183699131011963,
"rewards/rejected": -3.2309398651123047,
"step": 1230
},
{
"epoch": 0.99,
"learning_rate": 9.747123991141193e-10,
"logits/chosen": -1.2762700319290161,
"logits/rejected": -1.1082122325897217,
"logps/chosen": -2387.552490234375,
"logps/rejected": -3240.255615234375,
"loss": 0.4634,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.163398265838623,
"rewards/margins": 0.8387178182601929,
"rewards/rejected": -3.0021164417266846,
"step": 1240
},
{
"epoch": 1.0,
"learning_rate": 0.0,
"logits/chosen": -1.7946316003799438,
"logits/rejected": -1.654803991317749,
"logps/chosen": -2349.287109375,
"logps/rejected": -2943.7216796875,
"loss": 0.547,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -2.1279029846191406,
"rewards/margins": 0.613613486289978,
"rewards/rejected": -2.74151611328125,
"step": 1250
},
{
"epoch": 1.0,
"step": 1250,
"total_flos": 0.0,
"train_loss": 0.5607085178375244,
"train_runtime": 13233.1545,
"train_samples_per_second": 1.134,
"train_steps_per_second": 0.094
}
],
"logging_steps": 10,
"max_steps": 1250,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 20,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}