aftonposten-6b-align-scan / trainer_state.json
hugodk-sch's picture
Model save
3be5042 verified
raw
history blame
83.6 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 4.0,
"eval_steps": 100,
"global_step": 1540,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 1.282051282051282e-07,
"logits/chosen": -1.7278180122375488,
"logits/rejected": -1.7377450466156006,
"logps/chosen": -29.553977966308594,
"logps/rejected": -42.813133239746094,
"loss": 0.5,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.03,
"learning_rate": 1.282051282051282e-06,
"logits/chosen": -1.866452693939209,
"logits/rejected": -1.8707809448242188,
"logps/chosen": -36.98366165161133,
"logps/rejected": -33.666290283203125,
"loss": 0.4914,
"rewards/accuracies": 0.5555555820465088,
"rewards/chosen": 0.01834939420223236,
"rewards/margins": 0.04732292518019676,
"rewards/rejected": -0.02897353284060955,
"step": 10
},
{
"epoch": 0.05,
"learning_rate": 2.564102564102564e-06,
"logits/chosen": -1.9971332550048828,
"logits/rejected": -1.999770164489746,
"logps/chosen": -29.6362247467041,
"logps/rejected": -29.044448852539062,
"loss": 0.5009,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": 0.004776014480739832,
"rewards/margins": -0.01042011845856905,
"rewards/rejected": 0.015196132473647594,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 3.846153846153847e-06,
"logits/chosen": -1.9208379983901978,
"logits/rejected": -1.9181442260742188,
"logps/chosen": -31.425155639648438,
"logps/rejected": -33.237945556640625,
"loss": 0.4997,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.007249836809933186,
"rewards/margins": 0.008341209962964058,
"rewards/rejected": -0.015591045841574669,
"step": 30
},
{
"epoch": 0.1,
"learning_rate": 4.999896948438434e-06,
"logits/chosen": -2.0180625915527344,
"logits/rejected": -2.009308338165283,
"logps/chosen": -32.559749603271484,
"logps/rejected": -32.524681091308594,
"loss": 0.4962,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.013646525330841541,
"rewards/margins": 0.023250887170433998,
"rewards/rejected": -0.009604359045624733,
"step": 40
},
{
"epoch": 0.13,
"learning_rate": 4.987541037542187e-06,
"logits/chosen": -1.8639278411865234,
"logits/rejected": -1.8531490564346313,
"logps/chosen": -33.53788375854492,
"logps/rejected": -35.42852020263672,
"loss": 0.5015,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": 0.015677569434046745,
"rewards/margins": -0.0033827773295342922,
"rewards/rejected": 0.019060343503952026,
"step": 50
},
{
"epoch": 0.16,
"learning_rate": 4.954691471941119e-06,
"logits/chosen": -1.9435577392578125,
"logits/rejected": -1.9454940557479858,
"logps/chosen": -32.49355697631836,
"logps/rejected": -33.1856575012207,
"loss": 0.4817,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.08530057966709137,
"rewards/margins": 0.09672373533248901,
"rewards/rejected": -0.011423162184655666,
"step": 60
},
{
"epoch": 0.18,
"learning_rate": 4.901618883413549e-06,
"logits/chosen": -2.075368881225586,
"logits/rejected": -2.0803427696228027,
"logps/chosen": -33.89010238647461,
"logps/rejected": -36.549957275390625,
"loss": 0.4868,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.07870620489120483,
"rewards/margins": 0.06357409805059433,
"rewards/rejected": 0.015132094733417034,
"step": 70
},
{
"epoch": 0.21,
"learning_rate": 4.828760511501322e-06,
"logits/chosen": -1.9372116327285767,
"logits/rejected": -1.9403302669525146,
"logps/chosen": -34.205604553222656,
"logps/rejected": -34.49314880371094,
"loss": 0.4788,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": 0.17776472866535187,
"rewards/margins": 0.09835849702358246,
"rewards/rejected": 0.0794062465429306,
"step": 80
},
{
"epoch": 0.23,
"learning_rate": 4.7367166013034295e-06,
"logits/chosen": -1.945416808128357,
"logits/rejected": -1.9499365091323853,
"logps/chosen": -32.26136016845703,
"logps/rejected": -32.26020050048828,
"loss": 0.4788,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.16484788060188293,
"rewards/margins": 0.08588258177042007,
"rewards/rejected": 0.07896529138088226,
"step": 90
},
{
"epoch": 0.26,
"learning_rate": 4.626245458345211e-06,
"logits/chosen": -2.043586254119873,
"logits/rejected": -2.0416104793548584,
"logps/chosen": -31.968154907226562,
"logps/rejected": -31.14252281188965,
"loss": 0.4749,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.21814338862895966,
"rewards/margins": 0.11919467151165009,
"rewards/rejected": 0.09894871711730957,
"step": 100
},
{
"epoch": 0.26,
"eval_logits/chosen": -2.2375268936157227,
"eval_logits/rejected": -2.2326817512512207,
"eval_logps/chosen": -33.851219177246094,
"eval_logps/rejected": -37.3537483215332,
"eval_loss": 0.49629145860671997,
"eval_rewards/accuracies": 0.5336378812789917,
"eval_rewards/chosen": 0.14666710793972015,
"eval_rewards/margins": 0.016369162127375603,
"eval_rewards/rejected": 0.1302979290485382,
"eval_runtime": 145.7938,
"eval_samples_per_second": 2.353,
"eval_steps_per_second": 0.295,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 4.498257201263691e-06,
"logits/chosen": -1.998853087425232,
"logits/rejected": -1.9964749813079834,
"logps/chosen": -32.937965393066406,
"logps/rejected": -33.846004486083984,
"loss": 0.474,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.2446821928024292,
"rewards/margins": 0.09798828512430191,
"rewards/rejected": 0.1466939002275467,
"step": 110
},
{
"epoch": 0.31,
"learning_rate": 4.353806263777678e-06,
"logits/chosen": -2.0093610286712646,
"logits/rejected": -2.0010437965393066,
"logps/chosen": -32.1511116027832,
"logps/rejected": -31.939533233642578,
"loss": 0.4873,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.2350343018770218,
"rewards/margins": 0.057456426322460175,
"rewards/rejected": 0.17757786810398102,
"step": 120
},
{
"epoch": 0.34,
"learning_rate": 4.1940827077152755e-06,
"logits/chosen": -2.0374419689178467,
"logits/rejected": -2.029500961303711,
"logps/chosen": -30.157424926757812,
"logps/rejected": -31.90500831604004,
"loss": 0.4716,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.2546332776546478,
"rewards/margins": 0.1362351030111313,
"rewards/rejected": 0.11839816719293594,
"step": 130
},
{
"epoch": 0.36,
"learning_rate": 4.0204024186666215e-06,
"logits/chosen": -1.9675471782684326,
"logits/rejected": -1.9777450561523438,
"logps/chosen": -31.059162139892578,
"logps/rejected": -32.384300231933594,
"loss": 0.4616,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.28839823603630066,
"rewards/margins": 0.16793587803840637,
"rewards/rejected": 0.12046238034963608,
"step": 140
},
{
"epoch": 0.39,
"learning_rate": 3.834196265035119e-06,
"logits/chosen": -1.8793624639511108,
"logits/rejected": -1.8805418014526367,
"logps/chosen": -33.646915435791016,
"logps/rejected": -34.56272888183594,
"loss": 0.4385,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.4366222321987152,
"rewards/margins": 0.2809991240501404,
"rewards/rejected": 0.15562310814857483,
"step": 150
},
{
"epoch": 0.42,
"learning_rate": 3.636998309800573e-06,
"logits/chosen": -1.9307949542999268,
"logits/rejected": -1.9274256229400635,
"logps/chosen": -35.765926361083984,
"logps/rejected": -32.48193359375,
"loss": 0.4685,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.32460084557533264,
"rewards/margins": 0.13194666802883148,
"rewards/rejected": 0.19265416264533997,
"step": 160
},
{
"epoch": 0.44,
"learning_rate": 3.4304331721118078e-06,
"logits/chosen": -2.0318870544433594,
"logits/rejected": -2.0245723724365234,
"logps/chosen": -33.25593566894531,
"logps/rejected": -31.189266204833984,
"loss": 0.4315,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.4335424304008484,
"rewards/margins": 0.3034728467464447,
"rewards/rejected": 0.1300695389509201,
"step": 170
},
{
"epoch": 0.47,
"learning_rate": 3.2162026428305436e-06,
"logits/chosen": -2.0387840270996094,
"logits/rejected": -2.0439517498016357,
"logps/chosen": -31.977941513061523,
"logps/rejected": -32.183441162109375,
"loss": 0.4499,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.4534761905670166,
"rewards/margins": 0.2121172845363617,
"rewards/rejected": 0.2413589060306549,
"step": 180
},
{
"epoch": 0.49,
"learning_rate": 2.996071664294641e-06,
"logits/chosen": -2.039411783218384,
"logits/rejected": -2.036684513092041,
"logps/chosen": -31.094594955444336,
"logps/rejected": -31.10666275024414,
"loss": 0.4653,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.3154846429824829,
"rewards/margins": 0.1462438404560089,
"rewards/rejected": 0.169240802526474,
"step": 190
},
{
"epoch": 0.52,
"learning_rate": 2.7718537898066833e-06,
"logits/chosen": -1.909691572189331,
"logits/rejected": -1.914358139038086,
"logps/chosen": -31.08693504333496,
"logps/rejected": -32.61079406738281,
"loss": 0.4376,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.41547876596450806,
"rewards/margins": 0.27136775851249695,
"rewards/rejected": 0.14411096274852753,
"step": 200
},
{
"epoch": 0.52,
"eval_logits/chosen": -2.233895778656006,
"eval_logits/rejected": -2.2290823459625244,
"eval_logps/chosen": -33.78962326049805,
"eval_logps/rejected": -37.29551696777344,
"eval_loss": 0.4955826997756958,
"eval_rewards/accuracies": 0.5485880374908447,
"eval_rewards/chosen": 0.19594170153141022,
"eval_rewards/margins": 0.019058095291256905,
"eval_rewards/rejected": 0.17688362300395966,
"eval_runtime": 145.7262,
"eval_samples_per_second": 2.354,
"eval_steps_per_second": 0.295,
"step": 200
},
{
"epoch": 0.55,
"learning_rate": 2.5453962426402006e-06,
"logits/chosen": -2.0220179557800293,
"logits/rejected": -2.0326101779937744,
"logps/chosen": -31.51509666442871,
"logps/rejected": -33.655662536621094,
"loss": 0.4554,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.36496439576148987,
"rewards/margins": 0.19933317601680756,
"rewards/rejected": 0.16563120484352112,
"step": 210
},
{
"epoch": 0.57,
"learning_rate": 2.3185646976551794e-06,
"logits/chosen": -1.9142780303955078,
"logits/rejected": -1.9289791584014893,
"logps/chosen": -29.601001739501953,
"logps/rejected": -31.418704986572266,
"loss": 0.4339,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": 0.4084321856498718,
"rewards/margins": 0.2968555688858032,
"rewards/rejected": 0.11157669872045517,
"step": 220
},
{
"epoch": 0.6,
"learning_rate": 2.0932279108998323e-06,
"logits/chosen": -1.9722967147827148,
"logits/rejected": -1.9762967824935913,
"logps/chosen": -32.825157165527344,
"logps/rejected": -31.445331573486328,
"loss": 0.4142,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.47385087609291077,
"rewards/margins": 0.39067354798316956,
"rewards/rejected": 0.0831773579120636,
"step": 230
},
{
"epoch": 0.62,
"learning_rate": 1.8712423238279358e-06,
"logits/chosen": -1.9708006381988525,
"logits/rejected": -1.9490963220596313,
"logps/chosen": -33.599205017089844,
"logps/rejected": -34.856910705566406,
"loss": 0.4183,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.4436335563659668,
"rewards/margins": 0.36852845549583435,
"rewards/rejected": 0.07510510087013245,
"step": 240
},
{
"epoch": 0.65,
"learning_rate": 1.6544367689701824e-06,
"logits/chosen": -2.011307954788208,
"logits/rejected": -2.008012533187866,
"logps/chosen": -32.494956970214844,
"logps/rejected": -35.98812484741211,
"loss": 0.4641,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.3440879285335541,
"rewards/margins": 0.15464913845062256,
"rewards/rejected": 0.18943879008293152,
"step": 250
},
{
"epoch": 0.68,
"learning_rate": 1.4445974030621963e-06,
"logits/chosen": -1.8780018091201782,
"logits/rejected": -1.8755817413330078,
"logps/chosen": -33.72126388549805,
"logps/rejected": -35.290069580078125,
"loss": 0.4544,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.37370795011520386,
"rewards/margins": 0.20206721127033234,
"rewards/rejected": 0.17164072394371033,
"step": 260
},
{
"epoch": 0.7,
"learning_rate": 1.243452991757889e-06,
"logits/chosen": -1.8628206253051758,
"logits/rejected": -1.8603187799453735,
"logps/chosen": -33.929344177246094,
"logps/rejected": -31.61127281188965,
"loss": 0.4451,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.3746057152748108,
"rewards/margins": 0.246560737490654,
"rewards/rejected": 0.128044992685318,
"step": 270
},
{
"epoch": 0.73,
"learning_rate": 1.0526606671603523e-06,
"logits/chosen": -1.9666433334350586,
"logits/rejected": -1.9561984539031982,
"logps/chosen": -34.74022674560547,
"logps/rejected": -31.632495880126953,
"loss": 0.4239,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.4931296706199646,
"rewards/margins": 0.3263343572616577,
"rewards/rejected": 0.16679534316062927,
"step": 280
},
{
"epoch": 0.75,
"learning_rate": 8.737922755071455e-07,
"logits/chosen": -2.0619094371795654,
"logits/rejected": -2.0470869541168213,
"logps/chosen": -30.43317222595215,
"logps/rejected": -32.3420524597168,
"loss": 0.4679,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.39276057481765747,
"rewards/margins": 0.1449340283870697,
"rewards/rejected": 0.24782654643058777,
"step": 290
},
{
"epoch": 0.78,
"learning_rate": 7.08321427484816e-07,
"logits/chosen": -1.9332126379013062,
"logits/rejected": -1.9307750463485718,
"logps/chosen": -32.100502014160156,
"logps/rejected": -30.6780948638916,
"loss": 0.3835,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.6612862348556519,
"rewards/margins": 0.5503143072128296,
"rewards/rejected": 0.1109718531370163,
"step": 300
},
{
"epoch": 0.78,
"eval_logits/chosen": -2.231203556060791,
"eval_logits/rejected": -2.226395606994629,
"eval_logps/chosen": -33.778892517089844,
"eval_logps/rejected": -37.28717803955078,
"eval_loss": 0.4949820339679718,
"eval_rewards/accuracies": 0.5245016813278198,
"eval_rewards/chosen": 0.204526886343956,
"eval_rewards/margins": 0.020971858873963356,
"eval_rewards/rejected": 0.1835550218820572,
"eval_runtime": 145.6189,
"eval_samples_per_second": 2.355,
"eval_steps_per_second": 0.295,
"step": 300
},
{
"epoch": 0.81,
"learning_rate": 4.84533120650964e-06,
"logits/chosen": -1.9174331426620483,
"logits/rejected": -1.9142847061157227,
"logps/chosen": -31.049779891967773,
"logps/rejected": -33.54362487792969,
"loss": 0.4366,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.4428572654724121,
"rewards/margins": 0.28403452038764954,
"rewards/rejected": 0.15882274508476257,
"step": 310
},
{
"epoch": 0.83,
"learning_rate": 4.825108134172131e-06,
"logits/chosen": -1.9670381546020508,
"logits/rejected": -1.9549226760864258,
"logps/chosen": -34.00459671020508,
"logps/rejected": -33.429378509521484,
"loss": 0.4145,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.4731821119785309,
"rewards/margins": 0.38561564683914185,
"rewards/rejected": 0.08756652474403381,
"step": 320
},
{
"epoch": 0.86,
"learning_rate": 4.80369052967602e-06,
"logits/chosen": -2.001861095428467,
"logits/rejected": -2.0005335807800293,
"logps/chosen": -32.78104019165039,
"logps/rejected": -32.248443603515625,
"loss": 0.4183,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.5638058185577393,
"rewards/margins": 0.37889450788497925,
"rewards/rejected": 0.18491129577159882,
"step": 330
},
{
"epoch": 0.88,
"learning_rate": 4.781089396387968e-06,
"logits/chosen": -2.0892767906188965,
"logits/rejected": -2.073693037033081,
"logps/chosen": -33.377323150634766,
"logps/rejected": -32.779090881347656,
"loss": 0.4167,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.641654908657074,
"rewards/margins": 0.37049978971481323,
"rewards/rejected": 0.2711551785469055,
"step": 340
},
{
"epoch": 0.91,
"learning_rate": 4.757316345716554e-06,
"logits/chosen": -1.9618675708770752,
"logits/rejected": -1.9610551595687866,
"logps/chosen": -32.51572799682617,
"logps/rejected": -32.19974899291992,
"loss": 0.4137,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.6130937933921814,
"rewards/margins": 0.4007722735404968,
"rewards/rejected": 0.21232159435749054,
"step": 350
},
{
"epoch": 0.94,
"learning_rate": 4.73238359114687e-06,
"logits/chosen": -1.916259765625,
"logits/rejected": -1.9264857769012451,
"logps/chosen": -31.4000244140625,
"logps/rejected": -35.018882751464844,
"loss": 0.4065,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.6437316536903381,
"rewards/margins": 0.427828311920166,
"rewards/rejected": 0.21590332686901093,
"step": 360
},
{
"epoch": 0.96,
"learning_rate": 4.706303941965804e-06,
"logits/chosen": -2.050283908843994,
"logits/rejected": -2.043849468231201,
"logps/chosen": -32.88298797607422,
"logps/rejected": -28.950063705444336,
"loss": 0.4093,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.6151864528656006,
"rewards/margins": 0.3940979838371277,
"rewards/rejected": 0.2210884839296341,
"step": 370
},
{
"epoch": 0.99,
"learning_rate": 4.679090796681225e-06,
"logits/chosen": -1.9082043170928955,
"logits/rejected": -1.9103626012802124,
"logps/chosen": -33.2851676940918,
"logps/rejected": -30.665771484375,
"loss": 0.3759,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.7631045579910278,
"rewards/margins": 0.5976192355155945,
"rewards/rejected": 0.1654852330684662,
"step": 380
},
{
"epoch": 1.01,
"learning_rate": 4.650758136138454e-06,
"logits/chosen": -1.9357115030288696,
"logits/rejected": -1.9344127178192139,
"logps/chosen": -33.36685562133789,
"logps/rejected": -35.62567138671875,
"loss": 0.3409,
"rewards/accuracies": 0.7583333253860474,
"rewards/chosen": 0.8143496513366699,
"rewards/margins": 0.790077805519104,
"rewards/rejected": 0.02427184209227562,
"step": 390
},
{
"epoch": 1.04,
"learning_rate": 4.621320516337559e-06,
"logits/chosen": -1.8698232173919678,
"logits/rejected": -1.8616269826889038,
"logps/chosen": -30.623443603515625,
"logps/rejected": -36.03126907348633,
"loss": 0.3117,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": 0.920426070690155,
"rewards/margins": 0.9929585456848145,
"rewards/rejected": -0.07253243774175644,
"step": 400
},
{
"epoch": 1.04,
"eval_logits/chosen": -2.2160162925720215,
"eval_logits/rejected": -2.2111997604370117,
"eval_logps/chosen": -33.652793884277344,
"eval_logps/rejected": -37.193359375,
"eval_loss": 0.4891091287136078,
"eval_rewards/accuracies": 0.5651993155479431,
"eval_rewards/chosen": 0.30540817975997925,
"eval_rewards/margins": 0.04680241644382477,
"eval_rewards/rejected": 0.2586057484149933,
"eval_runtime": 146.2599,
"eval_samples_per_second": 2.345,
"eval_steps_per_second": 0.294,
"step": 400
},
{
"epoch": 1.06,
"learning_rate": 4.590793060955158e-06,
"logits/chosen": -2.044127941131592,
"logits/rejected": -2.046945571899414,
"logps/chosen": -31.79901695251465,
"logps/rejected": -34.87114334106445,
"loss": 0.299,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.003937005996704,
"rewards/margins": 1.0665733814239502,
"rewards/rejected": -0.0626363754272461,
"step": 410
},
{
"epoch": 1.09,
"learning_rate": 4.559191453574582e-06,
"logits/chosen": -1.8841445446014404,
"logits/rejected": -1.8827756643295288,
"logps/chosen": -27.981430053710938,
"logps/rejected": -32.39961624145508,
"loss": 0.3151,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 0.868902325630188,
"rewards/margins": 0.9344655871391296,
"rewards/rejected": -0.06556323915719986,
"step": 420
},
{
"epoch": 1.12,
"learning_rate": 4.52653192962838e-06,
"logits/chosen": -1.8422701358795166,
"logits/rejected": -1.835268259048462,
"logps/chosen": -32.6194953918457,
"logps/rejected": -34.0600700378418,
"loss": 0.2904,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.1636759042739868,
"rewards/margins": 1.0155143737792969,
"rewards/rejected": 0.14816154539585114,
"step": 430
},
{
"epoch": 1.14,
"learning_rate": 4.492831268057307e-06,
"logits/chosen": -2.0123181343078613,
"logits/rejected": -2.0072364807128906,
"logps/chosen": -30.41460609436035,
"logps/rejected": -32.033443450927734,
"loss": 0.284,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.0949517488479614,
"rewards/margins": 1.1476168632507324,
"rewards/rejected": -0.05266512185335159,
"step": 440
},
{
"epoch": 1.17,
"learning_rate": 4.458106782690094e-06,
"logits/chosen": -1.8972448110580444,
"logits/rejected": -1.9013574123382568,
"logps/chosen": -33.069793701171875,
"logps/rejected": -32.7395133972168,
"loss": 0.263,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.156842589378357,
"rewards/margins": 1.2775455713272095,
"rewards/rejected": -0.12070278078317642,
"step": 450
},
{
"epoch": 1.19,
"learning_rate": 4.422376313348405e-06,
"logits/chosen": -1.9023644924163818,
"logits/rejected": -1.8966976404190063,
"logps/chosen": -33.858001708984375,
"logps/rejected": -35.19942092895508,
"loss": 0.2537,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.249218225479126,
"rewards/margins": 1.4494789838790894,
"rewards/rejected": -0.20026080310344696,
"step": 460
},
{
"epoch": 1.22,
"learning_rate": 4.3856582166815696e-06,
"logits/chosen": -1.9303083419799805,
"logits/rejected": -1.9299156665802002,
"logps/chosen": -32.68412399291992,
"logps/rejected": -34.166542053222656,
"loss": 0.2809,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.1873619556427002,
"rewards/margins": 1.1722075939178467,
"rewards/rejected": 0.01515440084040165,
"step": 470
},
{
"epoch": 1.25,
"learning_rate": 4.347971356735789e-06,
"logits/chosen": -1.97918701171875,
"logits/rejected": -1.960694670677185,
"logps/chosen": -32.545894622802734,
"logps/rejected": -33.19091033935547,
"loss": 0.2501,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.291896104812622,
"rewards/margins": 1.3886420726776123,
"rewards/rejected": -0.09674612432718277,
"step": 480
},
{
"epoch": 1.27,
"learning_rate": 4.309335095262675e-06,
"logits/chosen": -1.9456336498260498,
"logits/rejected": -1.9451408386230469,
"logps/chosen": -30.120983123779297,
"logps/rejected": -31.22079849243164,
"loss": 0.2773,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.1922577619552612,
"rewards/margins": 1.1720596551895142,
"rewards/rejected": 0.020198041573166847,
"step": 490
},
{
"epoch": 1.3,
"learning_rate": 4.269769281772082e-06,
"logits/chosen": -1.9087337255477905,
"logits/rejected": -1.9019416570663452,
"logps/chosen": -30.988162994384766,
"logps/rejected": -34.79794692993164,
"loss": 0.2459,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.345572829246521,
"rewards/margins": 1.4825364351272583,
"rewards/rejected": -0.13696344196796417,
"step": 500
},
{
"epoch": 1.3,
"eval_logits/chosen": -2.1905555725097656,
"eval_logits/rejected": -2.1857681274414062,
"eval_logps/chosen": -33.6363525390625,
"eval_logps/rejected": -37.182701110839844,
"eval_loss": 0.4884692430496216,
"eval_rewards/accuracies": 0.5622923374176025,
"eval_rewards/chosen": 0.3185593783855438,
"eval_rewards/margins": 0.05142458155751228,
"eval_rewards/rejected": 0.26713478565216064,
"eval_runtime": 145.9464,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 500
},
{
"epoch": 1.32,
"learning_rate": 4.22929424333435e-06,
"logits/chosen": -1.903969168663025,
"logits/rejected": -1.9078088998794556,
"logps/chosen": -27.756145477294922,
"logps/rejected": -33.003501892089844,
"loss": 0.2721,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.1349422931671143,
"rewards/margins": 1.2503130435943604,
"rewards/rejected": -0.11537061631679535,
"step": 510
},
{
"epoch": 1.35,
"learning_rate": 4.1879307741372085e-06,
"logits/chosen": -1.9105024337768555,
"logits/rejected": -1.9210312366485596,
"logps/chosen": -31.711151123046875,
"logps/rejected": -30.852153778076172,
"loss": 0.2636,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.240944743156433,
"rewards/margins": 1.4013227224349976,
"rewards/rejected": -0.1603778600692749,
"step": 520
},
{
"epoch": 1.38,
"learning_rate": 4.145700124802693e-06,
"logits/chosen": -1.8562660217285156,
"logits/rejected": -1.8539501428604126,
"logps/chosen": -30.094646453857422,
"logps/rejected": -30.31314468383789,
"loss": 0.2542,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.2579472064971924,
"rewards/margins": 1.3670498132705688,
"rewards/rejected": -0.10910268872976303,
"step": 530
},
{
"epoch": 1.4,
"learning_rate": 4.102623991469562e-06,
"logits/chosen": -1.9388301372528076,
"logits/rejected": -1.931828260421753,
"logps/chosen": -32.52373504638672,
"logps/rejected": -33.19480514526367,
"loss": 0.234,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.3672724962234497,
"rewards/margins": 1.5091286897659302,
"rewards/rejected": -0.14185625314712524,
"step": 540
},
{
"epoch": 1.43,
"learning_rate": 4.058724504646834e-06,
"logits/chosen": -1.9054105281829834,
"logits/rejected": -1.9118045568466187,
"logps/chosen": -30.23921775817871,
"logps/rejected": -32.79788589477539,
"loss": 0.2692,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.211173176765442,
"rewards/margins": 1.2424962520599365,
"rewards/rejected": -0.03132311627268791,
"step": 550
},
{
"epoch": 1.45,
"learning_rate": 4.014024217844167e-06,
"logits/chosen": -1.9779608249664307,
"logits/rejected": -1.9552392959594727,
"logps/chosen": -29.897476196289062,
"logps/rejected": -33.047569274902344,
"loss": 0.2672,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.2320712804794312,
"rewards/margins": 1.273280143737793,
"rewards/rejected": -0.041208960115909576,
"step": 560
},
{
"epoch": 1.48,
"learning_rate": 3.968546095984911e-06,
"logits/chosen": -1.9123926162719727,
"logits/rejected": -1.9074643850326538,
"logps/chosen": -30.69790267944336,
"logps/rejected": -32.02743911743164,
"loss": 0.2612,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": 1.3750633001327515,
"rewards/margins": 1.3002793788909912,
"rewards/rejected": 0.07478378713130951,
"step": 570
},
{
"epoch": 1.51,
"learning_rate": 3.922313503607806e-06,
"logits/chosen": -1.9546325206756592,
"logits/rejected": -1.956498384475708,
"logps/chosen": -32.70453643798828,
"logps/rejected": -35.04417037963867,
"loss": 0.2395,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.337005853652954,
"rewards/margins": 1.5316107273101807,
"rewards/rejected": -0.19460472464561462,
"step": 580
},
{
"epoch": 1.53,
"learning_rate": 3.875350192863368e-06,
"logits/chosen": -1.9385898113250732,
"logits/rejected": -1.9379469156265259,
"logps/chosen": -28.891321182250977,
"logps/rejected": -31.504505157470703,
"loss": 0.2493,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.3605870008468628,
"rewards/margins": 1.3948794603347778,
"rewards/rejected": -0.034292541444301605,
"step": 590
},
{
"epoch": 1.56,
"learning_rate": 3.8276802913111436e-06,
"logits/chosen": -1.9496517181396484,
"logits/rejected": -1.9475892782211304,
"logps/chosen": -31.307537078857422,
"logps/rejected": -32.269371032714844,
"loss": 0.2639,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3525952100753784,
"rewards/margins": 1.3281266689300537,
"rewards/rejected": 0.024468522518873215,
"step": 600
},
{
"epoch": 1.56,
"eval_logits/chosen": -2.183182716369629,
"eval_logits/rejected": -2.178420066833496,
"eval_logps/chosen": -33.58167266845703,
"eval_logps/rejected": -37.20375442504883,
"eval_loss": 0.47496160864830017,
"eval_rewards/accuracies": 0.5855481624603271,
"eval_rewards/chosen": 0.3623029887676239,
"eval_rewards/margins": 0.11201038211584091,
"eval_rewards/rejected": 0.2502925992012024,
"eval_runtime": 145.9651,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 600
},
{
"epoch": 1.58,
"learning_rate": 3.7793282895240927e-06,
"logits/chosen": -1.9928886890411377,
"logits/rejected": -1.9994275569915771,
"logps/chosen": -30.57297706604004,
"logps/rejected": -32.0395622253418,
"loss": 0.2611,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.3315242528915405,
"rewards/margins": 1.3126256465911865,
"rewards/rejected": 0.01889852061867714,
"step": 610
},
{
"epoch": 1.61,
"learning_rate": 3.730319028506478e-06,
"logits/chosen": -1.9484421014785767,
"logits/rejected": -1.946126937866211,
"logps/chosen": -32.853248596191406,
"logps/rejected": -30.977664947509766,
"loss": 0.2377,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.5221078395843506,
"rewards/margins": 1.5158333778381348,
"rewards/rejected": 0.0062743364833295345,
"step": 620
},
{
"epoch": 1.64,
"learning_rate": 3.6806776869317074e-06,
"logits/chosen": -1.8969027996063232,
"logits/rejected": -1.8902422189712524,
"logps/chosen": -33.44438171386719,
"logps/rejected": -32.37852478027344,
"loss": 0.2252,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.6158740520477295,
"rewards/margins": 1.6329265832901,
"rewards/rejected": -0.017052406445145607,
"step": 630
},
{
"epoch": 1.66,
"learning_rate": 3.6304297682067146e-06,
"logits/chosen": -1.9018688201904297,
"logits/rejected": -1.9080841541290283,
"logps/chosen": -32.16619110107422,
"logps/rejected": -33.13113021850586,
"loss": 0.2565,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.450488567352295,
"rewards/margins": 1.3631832599639893,
"rewards/rejected": 0.08730525523424149,
"step": 640
},
{
"epoch": 1.69,
"learning_rate": 3.579601087369492e-06,
"logits/chosen": -1.98688542842865,
"logits/rejected": -2.0009474754333496,
"logps/chosen": -30.202587127685547,
"logps/rejected": -31.93996238708496,
"loss": 0.2613,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 1.3199015855789185,
"rewards/margins": 1.2737209796905518,
"rewards/rejected": 0.046180594712495804,
"step": 650
},
{
"epoch": 1.71,
"learning_rate": 3.5282177578265295e-06,
"logits/chosen": -1.861903429031372,
"logits/rejected": -1.8589446544647217,
"logps/chosen": -31.801767349243164,
"logps/rejected": -34.892234802246094,
"loss": 0.1937,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.6969530582427979,
"rewards/margins": 1.8505403995513916,
"rewards/rejected": -0.15358731150627136,
"step": 660
},
{
"epoch": 1.74,
"learning_rate": 3.476306177936961e-06,
"logits/chosen": -1.9529634714126587,
"logits/rejected": -1.9532558917999268,
"logps/chosen": -29.594532012939453,
"logps/rejected": -33.946571350097656,
"loss": 0.2422,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.3528481721878052,
"rewards/margins": 1.449000597000122,
"rewards/rejected": -0.09615238755941391,
"step": 670
},
{
"epoch": 1.77,
"learning_rate": 3.423893017450324e-06,
"logits/chosen": -1.9031528234481812,
"logits/rejected": -1.9001739025115967,
"logps/chosen": -29.16634178161621,
"logps/rejected": -32.86220169067383,
"loss": 0.2577,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": 1.3648045063018799,
"rewards/margins": 1.40032160282135,
"rewards/rejected": -0.03551710769534111,
"step": 680
},
{
"epoch": 1.79,
"learning_rate": 3.3710052038048794e-06,
"logits/chosen": -1.9385061264038086,
"logits/rejected": -1.9385082721710205,
"logps/chosen": -27.996692657470703,
"logps/rejected": -30.74009132385254,
"loss": 0.2085,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.7331215143203735,
"rewards/margins": 1.7887967824935913,
"rewards/rejected": -0.05567514896392822,
"step": 690
},
{
"epoch": 1.82,
"learning_rate": 3.3176699082935546e-06,
"logits/chosen": -1.8473536968231201,
"logits/rejected": -1.850669503211975,
"logps/chosen": -32.4169921875,
"logps/rejected": -31.4881649017334,
"loss": 0.2437,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": 1.6491864919662476,
"rewards/margins": 1.6679694652557373,
"rewards/rejected": -0.01878293789923191,
"step": 700
},
{
"epoch": 1.82,
"eval_logits/chosen": -2.178642988204956,
"eval_logits/rejected": -2.173909902572632,
"eval_logps/chosen": -33.59923553466797,
"eval_logps/rejected": -37.22935485839844,
"eval_loss": 0.4741733968257904,
"eval_rewards/accuracies": 0.5747508406639099,
"eval_rewards/chosen": 0.34825241565704346,
"eval_rewards/margins": 0.1184391975402832,
"eval_rewards/rejected": 0.22981324791908264,
"eval_runtime": 145.9455,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 700
},
{
"epoch": 1.84,
"learning_rate": 3.2639145321045933e-06,
"logits/chosen": -1.9303743839263916,
"logits/rejected": -1.9213898181915283,
"logps/chosen": -34.50909423828125,
"logps/rejected": -31.8797550201416,
"loss": 0.2249,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.602341890335083,
"rewards/margins": 1.5814943313598633,
"rewards/rejected": 0.02084733545780182,
"step": 710
},
{
"epoch": 1.87,
"learning_rate": 3.2097666922441107e-06,
"logits/chosen": -1.9459857940673828,
"logits/rejected": -1.9470113515853882,
"logps/chosen": -34.359092712402344,
"logps/rejected": -33.046634674072266,
"loss": 0.2173,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.6610389947891235,
"rewards/margins": 1.6375271081924438,
"rewards/rejected": 0.02351185865700245,
"step": 720
},
{
"epoch": 1.9,
"learning_rate": 3.1552542073477554e-06,
"logits/chosen": -1.962262511253357,
"logits/rejected": -1.9598932266235352,
"logps/chosen": -30.564239501953125,
"logps/rejected": -32.997047424316406,
"loss": 0.23,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.5992707014083862,
"rewards/margins": 1.5993889570236206,
"rewards/rejected": -0.00011831596202682704,
"step": 730
},
{
"epoch": 1.92,
"learning_rate": 3.100405083388799e-06,
"logits/chosen": -1.9474010467529297,
"logits/rejected": -1.952619194984436,
"logps/chosen": -29.694049835205078,
"logps/rejected": -33.282554626464844,
"loss": 0.2069,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.6733427047729492,
"rewards/margins": 1.7755340337753296,
"rewards/rejected": -0.10219136625528336,
"step": 740
},
{
"epoch": 1.95,
"learning_rate": 3.0452474992899645e-06,
"logits/chosen": -1.892179250717163,
"logits/rejected": -1.8910681009292603,
"logps/chosen": -31.141448974609375,
"logps/rejected": -34.69000244140625,
"loss": 0.24,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.5868273973464966,
"rewards/margins": 1.5900630950927734,
"rewards/rejected": -0.0032357454765588045,
"step": 750
},
{
"epoch": 1.97,
"learning_rate": 2.989809792446417e-06,
"logits/chosen": -1.7668031454086304,
"logits/rejected": -1.7617915868759155,
"logps/chosen": -33.82783889770508,
"logps/rejected": -35.33191680908203,
"loss": 0.2193,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.7364822626113892,
"rewards/margins": 1.8352861404418945,
"rewards/rejected": -0.0988037958741188,
"step": 760
},
{
"epoch": 2.0,
"learning_rate": 2.9341204441673267e-06,
"logits/chosen": -1.9073164463043213,
"logits/rejected": -1.9110107421875,
"logps/chosen": -33.157711029052734,
"logps/rejected": -33.576969146728516,
"loss": 0.2221,
"rewards/accuracies": 0.8708333969116211,
"rewards/chosen": 1.6447594165802002,
"rewards/margins": 1.6304632425308228,
"rewards/rejected": 0.014296052046120167,
"step": 770
},
{
"epoch": 2.03,
"learning_rate": 2.878208065043501e-06,
"logits/chosen": -1.8515024185180664,
"logits/rejected": -1.8497060537338257,
"logps/chosen": -31.297021865844727,
"logps/rejected": -35.01959991455078,
"loss": 0.1312,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.0836594104766846,
"rewards/margins": 2.575618028640747,
"rewards/rejected": -0.4919588565826416,
"step": 780
},
{
"epoch": 2.05,
"learning_rate": 2.8221013802485974e-06,
"logits/chosen": -1.899592638015747,
"logits/rejected": -1.8983027935028076,
"logps/chosen": -30.751636505126953,
"logps/rejected": -33.40520477294922,
"loss": 0.1431,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.0992822647094727,
"rewards/margins": 2.354191541671753,
"rewards/rejected": -0.25490933656692505,
"step": 790
},
{
"epoch": 2.08,
"learning_rate": 2.76582921478147e-06,
"logits/chosen": -1.83371901512146,
"logits/rejected": -1.8279550075531006,
"logps/chosen": -31.871252059936523,
"logps/rejected": -31.73343849182129,
"loss": 0.1567,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.0094635486602783,
"rewards/margins": 2.224806547164917,
"rewards/rejected": -0.21534299850463867,
"step": 800
},
{
"epoch": 2.08,
"eval_logits/chosen": -2.1802878379821777,
"eval_logits/rejected": -2.1755456924438477,
"eval_logps/chosen": -33.54963302612305,
"eval_logps/rejected": -37.20660400390625,
"eval_loss": 0.4695200026035309,
"eval_rewards/accuracies": 0.5826411843299866,
"eval_rewards/chosen": 0.3879339396953583,
"eval_rewards/margins": 0.1399237960577011,
"eval_rewards/rejected": 0.24801018834114075,
"eval_runtime": 145.8879,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 800
},
{
"epoch": 2.1,
"learning_rate": 2.7094204786572254e-06,
"logits/chosen": -1.9300823211669922,
"logits/rejected": -1.9375168085098267,
"logps/chosen": -29.62691307067871,
"logps/rejected": -34.63849639892578,
"loss": 0.1613,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.0010743141174316,
"rewards/margins": 2.3072001934051514,
"rewards/rejected": -0.3061259686946869,
"step": 810
},
{
"epoch": 2.13,
"learning_rate": 2.6529041520546072e-06,
"logits/chosen": -1.9036821126937866,
"logits/rejected": -1.9064573049545288,
"logps/chosen": -30.34005355834961,
"logps/rejected": -33.30119705200195,
"loss": 0.1999,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.784820556640625,
"rewards/margins": 1.8419349193572998,
"rewards/rejected": -0.05711423233151436,
"step": 820
},
{
"epoch": 2.16,
"learning_rate": 2.5963092704273302e-06,
"logits/chosen": -1.805069923400879,
"logits/rejected": -1.8091821670532227,
"logps/chosen": -30.127094268798828,
"logps/rejected": -35.30133056640625,
"loss": 0.1557,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.9714915752410889,
"rewards/margins": 2.2844126224517822,
"rewards/rejected": -0.31292054057121277,
"step": 830
},
{
"epoch": 2.18,
"learning_rate": 2.53966490958702e-06,
"logits/chosen": -1.8699356317520142,
"logits/rejected": -1.865962028503418,
"logps/chosen": -30.581180572509766,
"logps/rejected": -33.02351379394531,
"loss": 0.1821,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 1.9099775552749634,
"rewards/margins": 1.9330689907073975,
"rewards/rejected": -0.02309143915772438,
"step": 840
},
{
"epoch": 2.21,
"learning_rate": 2.4830001707654135e-06,
"logits/chosen": -1.9513721466064453,
"logits/rejected": -1.9533140659332275,
"logps/chosen": -30.263479232788086,
"logps/rejected": -36.080474853515625,
"loss": 0.1402,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 2.0094857215881348,
"rewards/margins": 2.344318389892578,
"rewards/rejected": -0.3348326086997986,
"step": 850
},
{
"epoch": 2.23,
"learning_rate": 2.4263441656635054e-06,
"logits/chosen": -1.7577623128890991,
"logits/rejected": -1.7520900964736938,
"logps/chosen": -33.64806365966797,
"logps/rejected": -33.127098083496094,
"loss": 0.157,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.106886625289917,
"rewards/margins": 2.2950429916381836,
"rewards/rejected": -0.18815645575523376,
"step": 860
},
{
"epoch": 2.26,
"learning_rate": 2.3697260014953107e-06,
"logits/chosen": -1.8190982341766357,
"logits/rejected": -1.8191255331039429,
"logps/chosen": -33.16907501220703,
"logps/rejected": -35.244049072265625,
"loss": 0.1428,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.138986110687256,
"rewards/margins": 2.457213878631592,
"rewards/rejected": -0.31822749972343445,
"step": 870
},
{
"epoch": 2.29,
"learning_rate": 2.3131747660339396e-06,
"logits/chosen": -1.867693543434143,
"logits/rejected": -1.8564653396606445,
"logps/chosen": -31.62196922302246,
"logps/rejected": -33.34123992919922,
"loss": 0.1484,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.9876899719238281,
"rewards/margins": 2.3230578899383545,
"rewards/rejected": -0.3353681266307831,
"step": 880
},
{
"epoch": 2.31,
"learning_rate": 2.256719512667651e-06,
"logits/chosen": -1.9699180126190186,
"logits/rejected": -1.9743999242782593,
"logps/chosen": -30.770462036132812,
"logps/rejected": -32.943992614746094,
"loss": 0.1449,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.0475802421569824,
"rewards/margins": 2.453592300415039,
"rewards/rejected": -0.40601205825805664,
"step": 890
},
{
"epoch": 2.34,
"learning_rate": 2.2003892454735786e-06,
"logits/chosen": -1.8890403509140015,
"logits/rejected": -1.881980538368225,
"logps/chosen": -32.13774490356445,
"logps/rejected": -32.89883041381836,
"loss": 0.131,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.173067808151245,
"rewards/margins": 2.5704257488250732,
"rewards/rejected": -0.397358238697052,
"step": 900
},
{
"epoch": 2.34,
"eval_logits/chosen": -2.1704888343811035,
"eval_logits/rejected": -2.165771961212158,
"eval_logps/chosen": -33.59296798706055,
"eval_logps/rejected": -37.2408447265625,
"eval_loss": 0.4716451168060303,
"eval_rewards/accuracies": 0.5859634280204773,
"eval_rewards/chosen": 0.3532681465148926,
"eval_rewards/margins": 0.1326470971107483,
"eval_rewards/rejected": 0.22062106430530548,
"eval_runtime": 145.952,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 900
},
{
"epoch": 2.36,
"learning_rate": 2.1442129043167877e-06,
"logits/chosen": -1.8803346157073975,
"logits/rejected": -1.8807852268218994,
"logps/chosen": -28.69057846069336,
"logps/rejected": -35.19638442993164,
"loss": 0.1531,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.9812660217285156,
"rewards/margins": 2.3866193294525146,
"rewards/rejected": -0.4053533673286438,
"step": 910
},
{
"epoch": 2.39,
"learning_rate": 2.088219349982323e-06,
"logits/chosen": -1.8385648727416992,
"logits/rejected": -1.8306169509887695,
"logps/chosen": -29.56528091430664,
"logps/rejected": -34.04883575439453,
"loss": 0.1693,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.8189325332641602,
"rewards/margins": 2.2009072303771973,
"rewards/rejected": -0.3819747567176819,
"step": 920
},
{
"epoch": 2.42,
"learning_rate": 2.0324373493478803e-06,
"logits/chosen": -2.004300117492676,
"logits/rejected": -2.004765033721924,
"logps/chosen": -27.645904541015625,
"logps/rejected": -33.17176818847656,
"loss": 0.1901,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.8809070587158203,
"rewards/margins": 2.126433849334717,
"rewards/rejected": -0.24552695453166962,
"step": 930
},
{
"epoch": 2.44,
"learning_rate": 1.976895560604729e-06,
"logits/chosen": -1.883927583694458,
"logits/rejected": -1.893393874168396,
"logps/chosen": -31.66170883178711,
"logps/rejected": -33.47935104370117,
"loss": 0.1308,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.2106852531433105,
"rewards/margins": 2.602447509765625,
"rewards/rejected": -0.3917620778083801,
"step": 940
},
{
"epoch": 2.47,
"learning_rate": 1.921622518534466e-06,
"logits/chosen": -1.9267995357513428,
"logits/rejected": -1.9306179285049438,
"logps/chosen": -28.294620513916016,
"logps/rejected": -32.00605010986328,
"loss": 0.1628,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.882483720779419,
"rewards/margins": 2.0984606742858887,
"rewards/rejected": -0.21597695350646973,
"step": 950
},
{
"epoch": 2.49,
"learning_rate": 1.8666466198491794e-06,
"logits/chosen": -1.9250777959823608,
"logits/rejected": -1.921263337135315,
"logps/chosen": -31.393346786499023,
"logps/rejected": -34.266326904296875,
"loss": 0.1531,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.145495891571045,
"rewards/margins": 2.4334449768066406,
"rewards/rejected": -0.28794899582862854,
"step": 960
},
{
"epoch": 2.52,
"learning_rate": 1.8119961086025376e-06,
"logits/chosen": -1.8407471179962158,
"logits/rejected": -1.84335458278656,
"logps/chosen": -30.177082061767578,
"logps/rejected": -35.5606803894043,
"loss": 0.15,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.1085331439971924,
"rewards/margins": 2.411801815032959,
"rewards/rejected": -0.3032683730125427,
"step": 970
},
{
"epoch": 2.55,
"learning_rate": 1.7576990616793139e-06,
"logits/chosen": -1.8759149312973022,
"logits/rejected": -1.8695703744888306,
"logps/chosen": -32.879905700683594,
"logps/rejected": -36.688812255859375,
"loss": 0.1591,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.020131826400757,
"rewards/margins": 2.3616156578063965,
"rewards/rejected": -0.34148353338241577,
"step": 980
},
{
"epoch": 2.57,
"learning_rate": 1.7037833743707892e-06,
"logits/chosen": -1.8502181768417358,
"logits/rejected": -1.845171332359314,
"logps/chosen": -28.556514739990234,
"logps/rejected": -36.38850784301758,
"loss": 0.167,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.9571492671966553,
"rewards/margins": 2.2760462760925293,
"rewards/rejected": -0.31889697909355164,
"step": 990
},
{
"epoch": 2.6,
"learning_rate": 1.6502767460434588e-06,
"logits/chosen": -1.8279926776885986,
"logits/rejected": -1.8174269199371338,
"logps/chosen": -29.500579833984375,
"logps/rejected": -29.74460220336914,
"loss": 0.1784,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.798485517501831,
"rewards/margins": 1.9143791198730469,
"rewards/rejected": -0.11589355766773224,
"step": 1000
},
{
"epoch": 2.6,
"eval_logits/chosen": -2.169346570968628,
"eval_logits/rejected": -2.1646294593811035,
"eval_logps/chosen": -33.57487106323242,
"eval_logps/rejected": -37.250404357910156,
"eval_loss": 0.4672938883304596,
"eval_rewards/accuracies": 0.5859634280204773,
"eval_rewards/chosen": 0.3677443265914917,
"eval_rewards/margins": 0.15477292239665985,
"eval_rewards/rejected": 0.21297141909599304,
"eval_runtime": 145.9529,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1000
},
{
"epoch": 2.62,
"learning_rate": 1.5972066659083796e-06,
"logits/chosen": -1.938071846961975,
"logits/rejected": -1.9374967813491821,
"logps/chosen": -29.38279151916504,
"logps/rejected": -30.51906394958496,
"loss": 0.1659,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.953122854232788,
"rewards/margins": 2.1615920066833496,
"rewards/rejected": -0.20846888422966003,
"step": 1010
},
{
"epoch": 2.65,
"learning_rate": 1.5446003988985041e-06,
"logits/chosen": -1.9777361154556274,
"logits/rejected": -1.9785385131835938,
"logps/chosen": -29.455398559570312,
"logps/rejected": -31.60211753845215,
"loss": 0.1457,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.9905421733856201,
"rewards/margins": 2.3395895957946777,
"rewards/rejected": -0.3490474820137024,
"step": 1020
},
{
"epoch": 2.68,
"learning_rate": 1.4924849716612211e-06,
"logits/chosen": -1.948656439781189,
"logits/rejected": -1.951843500137329,
"logps/chosen": -29.8292236328125,
"logps/rejected": -28.163440704345703,
"loss": 0.1712,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.960677146911621,
"rewards/margins": 2.1044540405273438,
"rewards/rejected": -0.14377683401107788,
"step": 1030
},
{
"epoch": 2.7,
"learning_rate": 1.440887158673332e-06,
"logits/chosen": -1.9582103490829468,
"logits/rejected": -1.9505630731582642,
"logps/chosen": -28.690776824951172,
"logps/rejected": -33.91931915283203,
"loss": 0.1486,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.9896939992904663,
"rewards/margins": 2.3249640464782715,
"rewards/rejected": -0.33527034521102905,
"step": 1040
},
{
"epoch": 2.73,
"learning_rate": 1.3898334684855647e-06,
"logits/chosen": -1.8943002223968506,
"logits/rejected": -1.905225157737732,
"logps/chosen": -30.873615264892578,
"logps/rejected": -32.47500228881836,
"loss": 0.1427,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.143594264984131,
"rewards/margins": 2.3275885581970215,
"rewards/rejected": -0.18399406969547272,
"step": 1050
},
{
"epoch": 2.75,
"learning_rate": 1.3393501301037245e-06,
"logits/chosen": -1.9732818603515625,
"logits/rejected": -1.9640705585479736,
"logps/chosen": -30.988611221313477,
"logps/rejected": -36.728416442871094,
"loss": 0.171,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 2.009733200073242,
"rewards/margins": 2.327359199523926,
"rewards/rejected": -0.31762614846229553,
"step": 1060
},
{
"epoch": 2.78,
"learning_rate": 1.2894630795134454e-06,
"logits/chosen": -1.8804638385772705,
"logits/rejected": -1.8824098110198975,
"logps/chosen": -33.24834442138672,
"logps/rejected": -33.089820861816406,
"loss": 0.1391,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.1927542686462402,
"rewards/margins": 2.506317615509033,
"rewards/rejected": -0.31356340646743774,
"step": 1070
},
{
"epoch": 2.81,
"learning_rate": 1.2401979463554984e-06,
"logits/chosen": -2.0114994049072266,
"logits/rejected": -2.012619972229004,
"logps/chosen": -30.447467803955078,
"logps/rejected": -34.64774703979492,
"loss": 0.1376,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 2.088136911392212,
"rewards/margins": 2.513786792755127,
"rewards/rejected": -0.42564988136291504,
"step": 1080
},
{
"epoch": 2.83,
"learning_rate": 1.1915800407584705e-06,
"logits/chosen": -1.9806941747665405,
"logits/rejected": -1.9849655628204346,
"logps/chosen": -28.488088607788086,
"logps/rejected": -33.84292984008789,
"loss": 0.1691,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.9241825342178345,
"rewards/margins": 2.160259485244751,
"rewards/rejected": -0.23607663810253143,
"step": 1090
},
{
"epoch": 2.86,
"learning_rate": 1.1436343403356019e-06,
"logits/chosen": -1.9700733423233032,
"logits/rejected": -1.9752953052520752,
"logps/chosen": -31.235626220703125,
"logps/rejected": -30.409204483032227,
"loss": 0.1956,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.8667011260986328,
"rewards/margins": 1.8727245330810547,
"rewards/rejected": -0.0060233683325350285,
"step": 1100
},
{
"epoch": 2.86,
"eval_logits/chosen": -2.1669461727142334,
"eval_logits/rejected": -2.162247896194458,
"eval_logps/chosen": -33.58707809448242,
"eval_logps/rejected": -37.24416732788086,
"eval_loss": 0.47056397795677185,
"eval_rewards/accuracies": 0.5859634280204773,
"eval_rewards/chosen": 0.3579804301261902,
"eval_rewards/margins": 0.14001740515232086,
"eval_rewards/rejected": 0.21796302497386932,
"eval_runtime": 145.9563,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1100
},
{
"epoch": 2.88,
"learning_rate": 1.0963854773524548e-06,
"logits/chosen": -1.9610811471939087,
"logits/rejected": -1.9610563516616821,
"logps/chosen": -30.302623748779297,
"logps/rejected": -31.24424171447754,
"loss": 0.1495,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.1100215911865234,
"rewards/margins": 2.2926793098449707,
"rewards/rejected": -0.18265783786773682,
"step": 1110
},
{
"epoch": 2.91,
"learning_rate": 1.049857726072005e-06,
"logits/chosen": -1.8055435419082642,
"logits/rejected": -1.8071558475494385,
"logps/chosen": -31.964313507080078,
"logps/rejected": -33.263916015625,
"loss": 0.1722,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 2.08052921295166,
"rewards/margins": 2.258647918701172,
"rewards/rejected": -0.17811879515647888,
"step": 1120
},
{
"epoch": 2.94,
"learning_rate": 1.0040749902836508e-06,
"logits/chosen": -1.8441641330718994,
"logits/rejected": -1.8414533138275146,
"logps/chosen": -28.680904388427734,
"logps/rejected": -31.291534423828125,
"loss": 0.1972,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.8646504878997803,
"rewards/margins": 1.9652849435806274,
"rewards/rejected": -0.10063423216342926,
"step": 1130
},
{
"epoch": 2.96,
"learning_rate": 9.59060791022566e-07,
"logits/chosen": -1.979723334312439,
"logits/rejected": -1.9741865396499634,
"logps/chosen": -30.334341049194336,
"logps/rejected": -33.13202667236328,
"loss": 0.1522,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.1643624305725098,
"rewards/margins": 2.3067753314971924,
"rewards/rejected": -0.14241299033164978,
"step": 1140
},
{
"epoch": 2.99,
"learning_rate": 9.148382544856885e-07,
"logits/chosen": -1.8336684703826904,
"logits/rejected": -1.823862075805664,
"logps/chosen": -30.90520668029785,
"logps/rejected": -31.456531524658203,
"loss": 0.1763,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.037224292755127,
"rewards/margins": 2.1685280799865723,
"rewards/rejected": -0.13130377233028412,
"step": 1150
},
{
"epoch": 3.01,
"learning_rate": 8.714301001505568e-07,
"logits/chosen": -1.9085521697998047,
"logits/rejected": -1.908399224281311,
"logps/chosen": -31.05959701538086,
"logps/rejected": -31.56801414489746,
"loss": 0.1469,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.09771990776062,
"rewards/margins": 2.3909716606140137,
"rewards/rejected": -0.2932516634464264,
"step": 1160
},
{
"epoch": 3.04,
"learning_rate": 8.288586291031025e-07,
"logits/chosen": -1.9817787408828735,
"logits/rejected": -1.9764562845230103,
"logps/chosen": -30.92386817932129,
"logps/rejected": -33.19076919555664,
"loss": 0.1496,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.0816802978515625,
"rewards/margins": 2.2813220024108887,
"rewards/rejected": -0.19964155554771423,
"step": 1170
},
{
"epoch": 3.06,
"learning_rate": 7.871457125803897e-07,
"logits/chosen": -1.8389742374420166,
"logits/rejected": -1.8466594219207764,
"logps/chosen": -30.842212677001953,
"logps/rejected": -32.71555709838867,
"loss": 0.1498,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.9926265478134155,
"rewards/margins": 2.2759203910827637,
"rewards/rejected": -0.28329357504844666,
"step": 1180
},
{
"epoch": 3.09,
"learning_rate": 7.463127807341966e-07,
"logits/chosen": -1.9125378131866455,
"logits/rejected": -1.9067026376724243,
"logps/chosen": -29.305688858032227,
"logps/rejected": -33.628910064697266,
"loss": 0.1274,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.294342279434204,
"rewards/margins": 2.570711612701416,
"rewards/rejected": -0.27636927366256714,
"step": 1190
},
{
"epoch": 3.12,
"learning_rate": 7.063808116212021e-07,
"logits/chosen": -1.8545643091201782,
"logits/rejected": -1.8566617965698242,
"logps/chosen": -30.835662841796875,
"logps/rejected": -33.26459503173828,
"loss": 0.137,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.1238210201263428,
"rewards/margins": 2.5274574756622314,
"rewards/rejected": -0.403636634349823,
"step": 1200
},
{
"epoch": 3.12,
"eval_logits/chosen": -2.167179584503174,
"eval_logits/rejected": -2.1624755859375,
"eval_logps/chosen": -33.572837829589844,
"eval_logps/rejected": -37.24382400512695,
"eval_loss": 0.4679512083530426,
"eval_rewards/accuracies": 0.6063122749328613,
"eval_rewards/chosen": 0.36937177181243896,
"eval_rewards/margins": 0.15113388001918793,
"eval_rewards/rejected": 0.21823787689208984,
"eval_runtime": 145.8851,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 1200
},
{
"epoch": 3.14,
"learning_rate": 6.673703204254348e-07,
"logits/chosen": -1.7856515645980835,
"logits/rejected": -1.784690260887146,
"logps/chosen": -33.143531799316406,
"logps/rejected": -33.10202407836914,
"loss": 0.1273,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.3271608352661133,
"rewards/margins": 2.641291618347168,
"rewards/rejected": -0.3141304850578308,
"step": 1210
},
{
"epoch": 3.17,
"learning_rate": 6.293013489185315e-07,
"logits/chosen": -1.9604772329330444,
"logits/rejected": -1.9546642303466797,
"logps/chosen": -28.979385375976562,
"logps/rejected": -33.37938690185547,
"loss": 0.1352,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.1794400215148926,
"rewards/margins": 2.5801169872283936,
"rewards/rejected": -0.40067729353904724,
"step": 1220
},
{
"epoch": 3.19,
"learning_rate": 5.921934551632086e-07,
"logits/chosen": -1.8098831176757812,
"logits/rejected": -1.798370361328125,
"logps/chosen": -31.50105857849121,
"logps/rejected": -33.25225067138672,
"loss": 0.1311,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.3076980113983154,
"rewards/margins": 2.5944409370422363,
"rewards/rejected": -0.2867427468299866,
"step": 1230
},
{
"epoch": 3.22,
"learning_rate": 5.560657034652405e-07,
"logits/chosen": -1.9010969400405884,
"logits/rejected": -1.8942989110946655,
"logps/chosen": -28.258697509765625,
"logps/rejected": -29.34195327758789,
"loss": 0.1603,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.9532296657562256,
"rewards/margins": 2.33274507522583,
"rewards/rejected": -0.3795151114463806,
"step": 1240
},
{
"epoch": 3.25,
"learning_rate": 5.2093665457911e-07,
"logits/chosen": -1.9308841228485107,
"logits/rejected": -1.9373613595962524,
"logps/chosen": -32.524845123291016,
"logps/rejected": -31.774621963500977,
"loss": 0.1315,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.2964370250701904,
"rewards/margins": 2.5014843940734863,
"rewards/rejected": -0.20504704117774963,
"step": 1250
},
{
"epoch": 3.27,
"learning_rate": 4.868243561723535e-07,
"logits/chosen": -1.9048112630844116,
"logits/rejected": -1.9054410457611084,
"logps/chosen": -30.393224716186523,
"logps/rejected": -33.286048889160156,
"loss": 0.1317,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.1082680225372314,
"rewards/margins": 2.485210418701172,
"rewards/rejected": -0.37694239616394043,
"step": 1260
},
{
"epoch": 3.3,
"learning_rate": 4.537463335535161e-07,
"logits/chosen": -1.846727728843689,
"logits/rejected": -1.8463557958602905,
"logps/chosen": -30.23922348022461,
"logps/rejected": -33.92576217651367,
"loss": 0.1224,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.2812910079956055,
"rewards/margins": 2.6054136753082275,
"rewards/rejected": -0.3241223692893982,
"step": 1270
},
{
"epoch": 3.32,
"learning_rate": 4.217195806684629e-07,
"logits/chosen": -1.730064034461975,
"logits/rejected": -1.7253376245498657,
"logps/chosen": -32.30054473876953,
"logps/rejected": -31.246471405029297,
"loss": 0.1154,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.349520206451416,
"rewards/margins": 2.605508327484131,
"rewards/rejected": -0.2559877932071686,
"step": 1280
},
{
"epoch": 3.35,
"learning_rate": 3.907605513696808e-07,
"logits/chosen": -1.9358218908309937,
"logits/rejected": -1.9203758239746094,
"logps/chosen": -31.7402400970459,
"logps/rejected": -35.388179779052734,
"loss": 0.1234,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 2.085869550704956,
"rewards/margins": 2.5841598510742188,
"rewards/rejected": -0.4982902407646179,
"step": 1290
},
{
"epoch": 3.38,
"learning_rate": 3.6088515096305675e-07,
"logits/chosen": -1.879227638244629,
"logits/rejected": -1.8840656280517578,
"logps/chosen": -30.69500732421875,
"logps/rejected": -36.70278549194336,
"loss": 0.1211,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 2.2580931186676025,
"rewards/margins": 2.7377448081970215,
"rewards/rejected": -0.4796522259712219,
"step": 1300
},
{
"epoch": 3.38,
"eval_logits/chosen": -2.1668777465820312,
"eval_logits/rejected": -2.1621785163879395,
"eval_logps/chosen": -33.58045959472656,
"eval_logps/rejected": -37.23927688598633,
"eval_loss": 0.47049835324287415,
"eval_rewards/accuracies": 0.5917773842811584,
"eval_rewards/chosen": 0.3632733225822449,
"eval_rewards/margins": 0.14139726758003235,
"eval_rewards/rejected": 0.22187604010105133,
"eval_runtime": 145.9507,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1300
},
{
"epoch": 3.4,
"learning_rate": 3.321087280364757e-07,
"logits/chosen": -1.8520374298095703,
"logits/rejected": -1.8522069454193115,
"logps/chosen": -33.18568801879883,
"logps/rejected": -37.46868896484375,
"loss": 0.134,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.446664333343506,
"rewards/margins": 2.7543227672576904,
"rewards/rejected": -0.30765873193740845,
"step": 1310
},
{
"epoch": 3.43,
"learning_rate": 3.044460665744284e-07,
"logits/chosen": -1.9394071102142334,
"logits/rejected": -1.9380232095718384,
"logps/chosen": -29.4769344329834,
"logps/rejected": -31.30813980102539,
"loss": 0.159,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 2.093536853790283,
"rewards/margins": 2.3773601055145264,
"rewards/rejected": -0.2838229537010193,
"step": 1320
},
{
"epoch": 3.45,
"learning_rate": 2.779113783626916e-07,
"logits/chosen": -1.844186782836914,
"logits/rejected": -1.8454091548919678,
"logps/chosen": -31.52813720703125,
"logps/rejected": -33.989559173583984,
"loss": 0.1366,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.276893138885498,
"rewards/margins": 2.549605131149292,
"rewards/rejected": -0.27271172404289246,
"step": 1330
},
{
"epoch": 3.48,
"learning_rate": 2.5251829568697204e-07,
"logits/chosen": -1.9051072597503662,
"logits/rejected": -1.9038896560668945,
"logps/chosen": -28.674198150634766,
"logps/rejected": -32.49821090698242,
"loss": 0.1373,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.01957631111145,
"rewards/margins": 2.424564838409424,
"rewards/rejected": -0.4049888551235199,
"step": 1340
},
{
"epoch": 3.51,
"learning_rate": 2.2827986432927774e-07,
"logits/chosen": -1.9206342697143555,
"logits/rejected": -1.9062645435333252,
"logps/chosen": -31.68826675415039,
"logps/rejected": -36.69408416748047,
"loss": 0.1396,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.102721691131592,
"rewards/margins": 2.4464478492736816,
"rewards/rejected": -0.34372615814208984,
"step": 1350
},
{
"epoch": 3.53,
"learning_rate": 2.0520853686560177e-07,
"logits/chosen": -1.921002984046936,
"logits/rejected": -1.9334627389907837,
"logps/chosen": -29.323871612548828,
"logps/rejected": -32.485260009765625,
"loss": 0.1368,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.195680856704712,
"rewards/margins": 2.3920950889587402,
"rewards/rejected": -0.1964142769575119,
"step": 1360
},
{
"epoch": 3.56,
"learning_rate": 1.833161662683672e-07,
"logits/chosen": -2.0142505168914795,
"logits/rejected": -2.0135538578033447,
"logps/chosen": -29.253097534179688,
"logps/rejected": -36.559532165527344,
"loss": 0.1185,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 2.2218658924102783,
"rewards/margins": 2.8047871589660645,
"rewards/rejected": -0.5829211473464966,
"step": 1370
},
{
"epoch": 3.58,
"learning_rate": 1.626139998169246e-07,
"logits/chosen": -1.8797496557235718,
"logits/rejected": -1.8874984979629517,
"logps/chosen": -31.32064437866211,
"logps/rejected": -37.89208221435547,
"loss": 0.144,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.287868022918701,
"rewards/margins": 2.641819477081299,
"rewards/rejected": -0.3539513051509857,
"step": 1380
},
{
"epoch": 3.61,
"learning_rate": 1.4311267331922535e-07,
"logits/chosen": -1.8338594436645508,
"logits/rejected": -1.8298187255859375,
"logps/chosen": -31.842514038085938,
"logps/rejected": -31.77047348022461,
"loss": 0.1419,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 2.3397603034973145,
"rewards/margins": 2.4782257080078125,
"rewards/rejected": -0.13846543431282043,
"step": 1390
},
{
"epoch": 3.64,
"learning_rate": 1.2482220564763669e-07,
"logits/chosen": -1.9958465099334717,
"logits/rejected": -1.9933058023452759,
"logps/chosen": -28.800048828125,
"logps/rejected": -32.510841369628906,
"loss": 0.1553,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.9972022771835327,
"rewards/margins": 2.2903003692626953,
"rewards/rejected": -0.29309794306755066,
"step": 1400
},
{
"epoch": 3.64,
"eval_logits/chosen": -2.1668412685394287,
"eval_logits/rejected": -2.162137985229492,
"eval_logps/chosen": -33.57229232788086,
"eval_logps/rejected": -37.258155822753906,
"eval_loss": 0.46543803811073303,
"eval_rewards/accuracies": 0.6034052968025208,
"eval_rewards/chosen": 0.36980658769607544,
"eval_rewards/margins": 0.16303294897079468,
"eval_rewards/rejected": 0.20677369832992554,
"eval_runtime": 145.9296,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1400
},
{
"epoch": 3.66,
"learning_rate": 1.0775199359171346e-07,
"logits/chosen": -1.9372856616973877,
"logits/rejected": -1.9331865310668945,
"logps/chosen": -30.8883113861084,
"logps/rejected": -29.84084129333496,
"loss": 0.1363,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.2033774852752686,
"rewards/margins": 2.4324822425842285,
"rewards/rejected": -0.22910502552986145,
"step": 1410
},
{
"epoch": 3.69,
"learning_rate": 9.191080703056604e-08,
"logits/chosen": -1.8948686122894287,
"logits/rejected": -1.8960151672363281,
"logps/chosen": -30.679889678955078,
"logps/rejected": -35.14348602294922,
"loss": 0.152,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.1479902267456055,
"rewards/margins": 2.323575735092163,
"rewards/rejected": -0.17558541893959045,
"step": 1420
},
{
"epoch": 3.71,
"learning_rate": 7.730678442730539e-08,
"logits/chosen": -1.8465667963027954,
"logits/rejected": -1.8398901224136353,
"logps/chosen": -31.448232650756836,
"logps/rejected": -37.158573150634766,
"loss": 0.1367,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.1548283100128174,
"rewards/margins": 2.530158519744873,
"rewards/rejected": -0.3753301799297333,
"step": 1430
},
{
"epoch": 3.74,
"learning_rate": 6.394742864787806e-08,
"logits/chosen": -1.859572172164917,
"logits/rejected": -1.8541902303695679,
"logps/chosen": -26.713485717773438,
"logps/rejected": -32.027252197265625,
"loss": 0.1488,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 2.0798051357269287,
"rewards/margins": 2.317080020904541,
"rewards/rejected": -0.2372746467590332,
"step": 1440
},
{
"epoch": 3.77,
"learning_rate": 5.183960310644748e-08,
"logits/chosen": -1.8826467990875244,
"logits/rejected": -1.8725192546844482,
"logps/chosen": -30.08587074279785,
"logps/rejected": -36.018455505371094,
"loss": 0.1431,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.9165277481079102,
"rewards/margins": 2.503002405166626,
"rewards/rejected": -0.5864745378494263,
"step": 1450
},
{
"epoch": 3.79,
"learning_rate": 4.098952823928693e-08,
"logits/chosen": -1.8568382263183594,
"logits/rejected": -1.8540499210357666,
"logps/chosen": -30.78998374938965,
"logps/rejected": -31.02651596069336,
"loss": 0.1648,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.0158348083496094,
"rewards/margins": 2.124389886856079,
"rewards/rejected": -0.10855510085821152,
"step": 1460
},
{
"epoch": 3.82,
"learning_rate": 3.1402778309014284e-08,
"logits/chosen": -1.9178192615509033,
"logits/rejected": -1.924325704574585,
"logps/chosen": -29.135822296142578,
"logps/rejected": -33.322288513183594,
"loss": 0.1271,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.3466856479644775,
"rewards/margins": 2.634577989578247,
"rewards/rejected": -0.28789228200912476,
"step": 1470
},
{
"epoch": 3.84,
"learning_rate": 2.3084278540791427e-08,
"logits/chosen": -1.9184119701385498,
"logits/rejected": -1.9287493228912354,
"logps/chosen": -29.0635986328125,
"logps/rejected": -29.924457550048828,
"loss": 0.1386,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.1053242683410645,
"rewards/margins": 2.342036008834839,
"rewards/rejected": -0.2367115318775177,
"step": 1480
},
{
"epoch": 3.87,
"learning_rate": 1.6038302591975807e-08,
"logits/chosen": -1.8501653671264648,
"logits/rejected": -1.8432512283325195,
"logps/chosen": -31.104766845703125,
"logps/rejected": -32.60163879394531,
"loss": 0.144,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 2.0413756370544434,
"rewards/margins": 2.3543033599853516,
"rewards/rejected": -0.31292763352394104,
"step": 1490
},
{
"epoch": 3.9,
"learning_rate": 1.0268470356514237e-08,
"logits/chosen": -1.9025371074676514,
"logits/rejected": -1.8992702960968018,
"logps/chosen": -31.016992568969727,
"logps/rejected": -33.745033264160156,
"loss": 0.1447,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 2.086864709854126,
"rewards/margins": 2.474714517593384,
"rewards/rejected": -0.3878494203090668,
"step": 1500
},
{
"epoch": 3.9,
"eval_logits/chosen": -2.1669082641601562,
"eval_logits/rejected": -2.1621992588043213,
"eval_logps/chosen": -33.575931549072266,
"eval_logps/rejected": -37.24650192260742,
"eval_loss": 0.46842658519744873,
"eval_rewards/accuracies": 0.574335515499115,
"eval_rewards/chosen": 0.3668961524963379,
"eval_rewards/margins": 0.15079911053180695,
"eval_rewards/rejected": 0.21609705686569214,
"eval_runtime": 145.9178,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 1500
},
{
"epoch": 3.92,
"learning_rate": 5.777746105209147e-09,
"logits/chosen": -1.9817174673080444,
"logits/rejected": -1.9823877811431885,
"logps/chosen": -27.261791229248047,
"logps/rejected": -33.12643814086914,
"loss": 0.1654,
"rewards/accuracies": 0.875,
"rewards/chosen": 2.0435855388641357,
"rewards/margins": 2.35237717628479,
"rewards/rejected": -0.30879148840904236,
"step": 1510
},
{
"epoch": 3.95,
"learning_rate": 2.5684369628148352e-09,
"logits/chosen": -1.834775686264038,
"logits/rejected": -1.8347667455673218,
"logps/chosen": -30.5068416595459,
"logps/rejected": -34.2733154296875,
"loss": 0.1565,
"rewards/accuracies": 0.9375,
"rewards/chosen": 2.0280237197875977,
"rewards/margins": 2.347689151763916,
"rewards/rejected": -0.31966525316238403,
"step": 1520
},
{
"epoch": 3.97,
"learning_rate": 6.421917227455999e-10,
"logits/chosen": -1.986581563949585,
"logits/rejected": -1.9839556217193604,
"logps/chosen": -29.06634521484375,
"logps/rejected": -31.9202938079834,
"loss": 0.1594,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.926171898841858,
"rewards/margins": 2.2754859924316406,
"rewards/rejected": -0.34931421279907227,
"step": 1530
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"logits/chosen": -1.9706960916519165,
"logits/rejected": -1.9718106985092163,
"logps/chosen": -28.249774932861328,
"logps/rejected": -30.143047332763672,
"loss": 0.1594,
"rewards/accuracies": 0.9583333730697632,
"rewards/chosen": 1.8961349725723267,
"rewards/margins": 2.178429126739502,
"rewards/rejected": -0.2822941541671753,
"step": 1540
},
{
"epoch": 4.0,
"step": 1540,
"total_flos": 0.0,
"train_loss": 0.16076272354497537,
"train_runtime": 10802.3033,
"train_samples_per_second": 1.14,
"train_steps_per_second": 0.143
}
],
"logging_steps": 10,
"max_steps": 1540,
"num_input_tokens_seen": 0,
"num_train_epochs": 4,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}