dctanner's picture
Model save
50149aa verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.999508116084604,
"eval_steps": 200,
"global_step": 1016,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4.901960784313726e-08,
"logits/chosen": -2.0748214721679688,
"logits/rejected": -2.1528916358947754,
"logps/chosen": -78.83467102050781,
"logps/rejected": -90.04127502441406,
"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.901960784313725e-07,
"logits/chosen": -2.167158603668213,
"logits/rejected": -2.065119743347168,
"logps/chosen": -101.462646484375,
"logps/rejected": -67.91659545898438,
"loss": 0.6937,
"rewards/accuracies": 0.3611111044883728,
"rewards/chosen": -0.0003858842537738383,
"rewards/margins": -0.001147754956036806,
"rewards/rejected": 0.0007618705276399851,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 9.80392156862745e-07,
"logits/chosen": -2.139626979827881,
"logits/rejected": -2.0195488929748535,
"logps/chosen": -109.9924545288086,
"logps/rejected": -84.65155029296875,
"loss": 0.695,
"rewards/accuracies": 0.375,
"rewards/chosen": -0.002081229817122221,
"rewards/margins": -0.0036208624951541424,
"rewards/rejected": 0.0015396325616165996,
"step": 20
},
{
"epoch": 0.03,
"learning_rate": 1.4705882352941177e-06,
"logits/chosen": -2.2819817066192627,
"logits/rejected": -2.181386947631836,
"logps/chosen": -100.8692398071289,
"logps/rejected": -84.57305908203125,
"loss": 0.6931,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.0019000970060005784,
"rewards/margins": 0.00016770660295151174,
"rewards/rejected": 0.0017323906067758799,
"step": 30
},
{
"epoch": 0.04,
"learning_rate": 1.96078431372549e-06,
"logits/chosen": -2.274721622467041,
"logits/rejected": -2.156294584274292,
"logps/chosen": -105.41255950927734,
"logps/rejected": -83.99583435058594,
"loss": 0.6929,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.001408256939612329,
"rewards/margins": 0.0004250857455190271,
"rewards/rejected": 0.0009831712814047933,
"step": 40
},
{
"epoch": 0.05,
"learning_rate": 2.450980392156863e-06,
"logits/chosen": -2.29875111579895,
"logits/rejected": -2.20742130279541,
"logps/chosen": -103.51786041259766,
"logps/rejected": -77.90748596191406,
"loss": 0.692,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.005386353936046362,
"rewards/margins": 0.002320895902812481,
"rewards/rejected": 0.0030654582660645247,
"step": 50
},
{
"epoch": 0.06,
"learning_rate": 2.9411764705882355e-06,
"logits/chosen": -2.30989408493042,
"logits/rejected": -2.233097553253174,
"logps/chosen": -107.9452896118164,
"logps/rejected": -90.55638122558594,
"loss": 0.6912,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.008994068950414658,
"rewards/margins": 0.003984561190009117,
"rewards/rejected": 0.005009507294744253,
"step": 60
},
{
"epoch": 0.07,
"learning_rate": 3.431372549019608e-06,
"logits/chosen": -2.2964837551116943,
"logits/rejected": -2.192676544189453,
"logps/chosen": -98.52997589111328,
"logps/rejected": -73.2212142944336,
"loss": 0.6907,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.011149758473038673,
"rewards/margins": 0.004859681706875563,
"rewards/rejected": 0.006290078163146973,
"step": 70
},
{
"epoch": 0.08,
"learning_rate": 3.92156862745098e-06,
"logits/chosen": -2.147590398788452,
"logits/rejected": -2.0242252349853516,
"logps/chosen": -110.5780258178711,
"logps/rejected": -85.19944763183594,
"loss": 0.6903,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.018292848020792007,
"rewards/margins": 0.005854339338839054,
"rewards/rejected": 0.012438507750630379,
"step": 80
},
{
"epoch": 0.09,
"learning_rate": 4.411764705882353e-06,
"logits/chosen": -2.103201389312744,
"logits/rejected": -1.9731849431991577,
"logps/chosen": -86.84721374511719,
"logps/rejected": -63.73047637939453,
"loss": 0.6889,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.02545500174164772,
"rewards/margins": 0.008647125214338303,
"rewards/rejected": 0.016807876527309418,
"step": 90
},
{
"epoch": 0.1,
"learning_rate": 4.901960784313726e-06,
"logits/chosen": -2.071173906326294,
"logits/rejected": -1.8952083587646484,
"logps/chosen": -120.53657531738281,
"logps/rejected": -84.6740951538086,
"loss": 0.6804,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.051459141075611115,
"rewards/margins": 0.02619524672627449,
"rewards/rejected": 0.025263899937272072,
"step": 100
},
{
"epoch": 0.11,
"learning_rate": 4.9990549169459415e-06,
"logits/chosen": -2.2821316719055176,
"logits/rejected": -2.1455578804016113,
"logps/chosen": -103.51020812988281,
"logps/rejected": -78.41320037841797,
"loss": 0.6767,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": 0.06272837519645691,
"rewards/margins": 0.03374689072370529,
"rewards/rejected": 0.028981482610106468,
"step": 110
},
{
"epoch": 0.12,
"learning_rate": 4.995216741642263e-06,
"logits/chosen": -2.284080743789673,
"logits/rejected": -2.210580825805664,
"logps/chosen": -96.24214172363281,
"logps/rejected": -77.19042205810547,
"loss": 0.6812,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.08176051080226898,
"rewards/margins": 0.02550395205616951,
"rewards/rejected": 0.056256555020809174,
"step": 120
},
{
"epoch": 0.13,
"learning_rate": 4.988430936991089e-06,
"logits/chosen": -2.2925381660461426,
"logits/rejected": -2.149710178375244,
"logps/chosen": -106.36643981933594,
"logps/rejected": -80.01815795898438,
"loss": 0.6647,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.11486251652240753,
"rewards/margins": 0.06016199663281441,
"rewards/rejected": 0.054700516164302826,
"step": 130
},
{
"epoch": 0.14,
"learning_rate": 4.978705519144525e-06,
"logits/chosen": -2.1970953941345215,
"logits/rejected": -2.0385475158691406,
"logps/chosen": -118.03541564941406,
"logps/rejected": -81.73417663574219,
"loss": 0.6639,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.10544681549072266,
"rewards/margins": 0.06229867413640022,
"rewards/rejected": 0.04314813390374184,
"step": 140
},
{
"epoch": 0.15,
"learning_rate": 4.966051976854862e-06,
"logits/chosen": -2.339622735977173,
"logits/rejected": -2.2239785194396973,
"logps/chosen": -90.04592895507812,
"logps/rejected": -68.85932922363281,
"loss": 0.6506,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.1409004032611847,
"rewards/margins": 0.09264969825744629,
"rewards/rejected": 0.048250701278448105,
"step": 150
},
{
"epoch": 0.16,
"learning_rate": 4.950485257902782e-06,
"logits/chosen": -2.2208049297332764,
"logits/rejected": -2.1466565132141113,
"logps/chosen": -101.00809478759766,
"logps/rejected": -76.30353546142578,
"loss": 0.6618,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.13603894412517548,
"rewards/margins": 0.0690341368317604,
"rewards/rejected": 0.06700481474399567,
"step": 160
},
{
"epoch": 0.17,
"learning_rate": 4.932023751439358e-06,
"logits/chosen": -2.286597728729248,
"logits/rejected": -2.13307523727417,
"logps/chosen": -110.24186706542969,
"logps/rejected": -81.82392120361328,
"loss": 0.6666,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.13935917615890503,
"rewards/margins": 0.06269019842147827,
"rewards/rejected": 0.07666899263858795,
"step": 170
},
{
"epoch": 0.18,
"learning_rate": 4.9106892662627395e-06,
"logits/chosen": -2.357682466506958,
"logits/rejected": -2.237700939178467,
"logps/chosen": -104.79756927490234,
"logps/rejected": -81.48443603515625,
"loss": 0.6615,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.16287028789520264,
"rewards/margins": 0.07204125821590424,
"rewards/rejected": 0.0908290296792984,
"step": 180
},
{
"epoch": 0.19,
"learning_rate": 4.886507005055149e-06,
"logits/chosen": -2.3158655166625977,
"logits/rejected": -2.1448147296905518,
"logps/chosen": -113.81204986572266,
"logps/rejected": -80.2816390991211,
"loss": 0.646,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.15469492971897125,
"rewards/margins": 0.10541819036006927,
"rewards/rejected": 0.049276743084192276,
"step": 190
},
{
"epoch": 0.2,
"learning_rate": 4.859505534610658e-06,
"logits/chosen": -2.2734456062316895,
"logits/rejected": -2.1840872764587402,
"logps/chosen": -94.27523040771484,
"logps/rejected": -76.13407897949219,
"loss": 0.65,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.12800325453281403,
"rewards/margins": 0.10445593297481537,
"rewards/rejected": 0.023547304794192314,
"step": 200
},
{
"epoch": 0.2,
"eval_logits/chosen": -2.1420769691467285,
"eval_logits/rejected": -2.046384811401367,
"eval_logps/chosen": -98.08348083496094,
"eval_logps/rejected": -76.25611877441406,
"eval_loss": 0.656255841255188,
"eval_rewards/accuracies": 0.650943398475647,
"eval_rewards/chosen": 0.106992706656456,
"eval_rewards/margins": 0.08928683400154114,
"eval_rewards/rejected": 0.017705870792269707,
"eval_runtime": 425.8967,
"eval_samples_per_second": 0.981,
"eval_steps_per_second": 0.124,
"step": 200
},
{
"epoch": 0.21,
"learning_rate": 4.829716752088893e-06,
"logits/chosen": -2.2180724143981934,
"logits/rejected": -2.1603446006774902,
"logps/chosen": -85.03736877441406,
"logps/rejected": -78.87088012695312,
"loss": 0.6573,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.10240446031093597,
"rewards/margins": 0.08565853536128998,
"rewards/rejected": 0.016745951026678085,
"step": 210
},
{
"epoch": 0.22,
"learning_rate": 4.797175847334535e-06,
"logits/chosen": -2.2404112815856934,
"logits/rejected": -2.12317156791687,
"logps/chosen": -107.25093841552734,
"logps/rejected": -84.33776092529297,
"loss": 0.6679,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.06942565739154816,
"rewards/margins": 0.06278938055038452,
"rewards/rejected": 0.006636275444179773,
"step": 220
},
{
"epoch": 0.23,
"learning_rate": 4.761921261307143e-06,
"logits/chosen": -2.271301746368408,
"logits/rejected": -2.1368188858032227,
"logps/chosen": -102.20993041992188,
"logps/rejected": -80.41031646728516,
"loss": 0.656,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.06752457469701767,
"rewards/margins": 0.09335462749004364,
"rewards/rejected": -0.02583005093038082,
"step": 230
},
{
"epoch": 0.24,
"learning_rate": 4.723994640670377e-06,
"logits/chosen": -2.2616968154907227,
"logits/rejected": -2.090630292892456,
"logps/chosen": -112.83064270019531,
"logps/rejected": -78.7665023803711,
"loss": 0.613,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": 0.1038433164358139,
"rewards/margins": 0.18221423029899597,
"rewards/rejected": -0.07837091386318207,
"step": 240
},
{
"epoch": 0.25,
"learning_rate": 4.68344078859431e-06,
"logits/chosen": -2.214836597442627,
"logits/rejected": -2.1651015281677246,
"logps/chosen": -94.0736312866211,
"logps/rejected": -87.97615814208984,
"loss": 0.6446,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.05433257669210434,
"rewards/margins": 0.1173890084028244,
"rewards/rejected": -0.06305641680955887,
"step": 250
},
{
"epoch": 0.26,
"learning_rate": 4.6403076118289006e-06,
"logits/chosen": -2.191092014312744,
"logits/rejected": -2.0465681552886963,
"logps/chosen": -106.81085205078125,
"logps/rejected": -80.91065216064453,
"loss": 0.6325,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.04650095850229263,
"rewards/margins": 0.14686846733093262,
"rewards/rejected": -0.10036750882863998,
"step": 260
},
{
"epoch": 0.27,
"learning_rate": 4.5946460641111776e-06,
"logits/chosen": -2.3747944831848145,
"logits/rejected": -2.1630542278289795,
"logps/chosen": -103.29341125488281,
"logps/rejected": -80.09381103515625,
"loss": 0.6041,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.06812257319688797,
"rewards/margins": 0.2125866860151291,
"rewards/rejected": -0.14446412026882172,
"step": 270
},
{
"epoch": 0.28,
"learning_rate": 4.546510085972983e-06,
"logits/chosen": -2.385852813720703,
"logits/rejected": -2.2712764739990234,
"logps/chosen": -115.0221176147461,
"logps/rejected": -90.1436538696289,
"loss": 0.6212,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.017763856798410416,
"rewards/margins": 0.1770613044500351,
"rewards/rejected": -0.15929746627807617,
"step": 280
},
{
"epoch": 0.29,
"learning_rate": 4.495956541020376e-06,
"logits/chosen": -2.3901445865631104,
"logits/rejected": -2.247413396835327,
"logps/chosen": -119.44215393066406,
"logps/rejected": -101.89396667480469,
"loss": 0.613,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.03466514125466347,
"rewards/margins": 0.2126956284046173,
"rewards/rejected": -0.24736078083515167,
"step": 290
},
{
"epoch": 0.3,
"learning_rate": 4.443045148759978e-06,
"logits/chosen": -2.3851912021636963,
"logits/rejected": -2.246699094772339,
"logps/chosen": -126.57054138183594,
"logps/rejected": -93.43769836425781,
"loss": 0.5898,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.06855957210063934,
"rewards/margins": 0.2794524133205414,
"rewards/rejected": -0.3480120301246643,
"step": 300
},
{
"epoch": 0.3,
"learning_rate": 4.3878384140516025e-06,
"logits/chosen": -2.4334146976470947,
"logits/rejected": -2.3179290294647217,
"logps/chosen": -118.78206634521484,
"logps/rejected": -107.4043197631836,
"loss": 0.5792,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.15513427555561066,
"rewards/margins": 0.30940642952919006,
"rewards/rejected": -0.46454063057899475,
"step": 310
},
{
"epoch": 0.31,
"learning_rate": 4.330401553270522e-06,
"logits/chosen": -2.34911847114563,
"logits/rejected": -2.2519352436065674,
"logps/chosen": -116.80616760253906,
"logps/rejected": -108.0991439819336,
"loss": 0.5869,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.32395830750465393,
"rewards/margins": 0.30500033497810364,
"rewards/rejected": -0.6289585828781128,
"step": 320
},
{
"epoch": 0.32,
"learning_rate": 4.2708024172665795e-06,
"logits/chosen": -2.448741912841797,
"logits/rejected": -2.2790164947509766,
"logps/chosen": -119.88993072509766,
"logps/rejected": -94.18651580810547,
"loss": 0.5564,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.37532466650009155,
"rewards/margins": 0.37958258390426636,
"rewards/rejected": -0.7549072504043579,
"step": 330
},
{
"epoch": 0.33,
"learning_rate": 4.209111411211174e-06,
"logits/chosen": -2.4606077671051025,
"logits/rejected": -2.3158764839172363,
"logps/chosen": -108.91439056396484,
"logps/rejected": -94.19551086425781,
"loss": 0.5416,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.3131277859210968,
"rewards/margins": 0.4208511710166931,
"rewards/rejected": -0.7339790463447571,
"step": 340
},
{
"epoch": 0.34,
"learning_rate": 4.145401411426788e-06,
"logits/chosen": -2.4888482093811035,
"logits/rejected": -2.3739964962005615,
"logps/chosen": -106.79335021972656,
"logps/rejected": -94.10002136230469,
"loss": 0.5863,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.24795857071876526,
"rewards/margins": 0.3726752996444702,
"rewards/rejected": -0.6206337809562683,
"step": 350
},
{
"epoch": 0.35,
"learning_rate": 4.079747679297314e-06,
"logits/chosen": -2.320113182067871,
"logits/rejected": -2.214874744415283,
"logps/chosen": -118.53324890136719,
"logps/rejected": -100.85164642333984,
"loss": 0.574,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.29153329133987427,
"rewards/margins": 0.3650625944137573,
"rewards/rejected": -0.6565959453582764,
"step": 360
},
{
"epoch": 0.36,
"learning_rate": 4.012227772360889e-06,
"logits/chosen": -2.3389956951141357,
"logits/rejected": -2.1818504333496094,
"logps/chosen": -123.92159271240234,
"logps/rejected": -110.42405700683594,
"loss": 0.5134,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -0.2689405083656311,
"rewards/margins": 0.5353353023529053,
"rewards/rejected": -0.8042758703231812,
"step": 370
},
{
"epoch": 0.37,
"learning_rate": 3.942921452690245e-06,
"logits/chosen": -2.369523525238037,
"logits/rejected": -2.285311222076416,
"logps/chosen": -132.2254180908203,
"logps/rejected": -125.16448974609375,
"loss": 0.6273,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6799182891845703,
"rewards/margins": 0.3109356760978699,
"rewards/rejected": -0.9908539652824402,
"step": 380
},
{
"epoch": 0.38,
"learning_rate": 3.871910592668817e-06,
"logits/chosen": -2.4459290504455566,
"logits/rejected": -2.3258919715881348,
"logps/chosen": -127.33780670166016,
"logps/rejected": -129.97348022460938,
"loss": 0.4728,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.7220331430435181,
"rewards/margins": 0.7715710997581482,
"rewards/rejected": -1.4936041831970215,
"step": 390
},
{
"epoch": 0.39,
"learning_rate": 3.799279078273921e-06,
"logits/chosen": -2.3586697578430176,
"logits/rejected": -2.190298080444336,
"logps/chosen": -137.87445068359375,
"logps/rejected": -125.60152435302734,
"loss": 0.456,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.9262881278991699,
"rewards/margins": 0.8406116366386414,
"rewards/rejected": -1.766899824142456,
"step": 400
},
{
"epoch": 0.39,
"eval_logits/chosen": -2.21016526222229,
"eval_logits/rejected": -2.1202762126922607,
"eval_logps/chosen": -142.6661376953125,
"eval_logps/rejected": -139.3409881591797,
"eval_loss": 0.5446351766586304,
"eval_rewards/accuracies": 0.7216981053352356,
"eval_rewards/chosen": -1.23048734664917,
"eval_rewards/margins": 0.6443533301353455,
"eval_rewards/rejected": -1.8748407363891602,
"eval_runtime": 427.1638,
"eval_samples_per_second": 0.979,
"eval_steps_per_second": 0.124,
"step": 400
},
{
"epoch": 0.4,
"learning_rate": 3.725112709981249e-06,
"logits/chosen": -2.2144548892974854,
"logits/rejected": -2.0825304985046387,
"logps/chosen": -153.91371154785156,
"logps/rejected": -152.90274047851562,
"loss": 0.524,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.2480976581573486,
"rewards/margins": 0.7566108703613281,
"rewards/rejected": -2.0047080516815186,
"step": 410
},
{
"epoch": 0.41,
"learning_rate": 3.649499101407737e-06,
"logits/chosen": -2.3662781715393066,
"logits/rejected": -2.2161378860473633,
"logps/chosen": -157.3763427734375,
"logps/rejected": -154.7397918701172,
"loss": 0.5717,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.3450363874435425,
"rewards/margins": 0.736282467842102,
"rewards/rejected": -2.0813188552856445,
"step": 420
},
{
"epoch": 0.42,
"learning_rate": 3.5725275758125564e-06,
"logits/chosen": -2.2823150157928467,
"logits/rejected": -2.1387839317321777,
"logps/chosen": -135.34524536132812,
"logps/rejected": -134.94952392578125,
"loss": 0.4512,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -1.2761428356170654,
"rewards/margins": 1.0124304294586182,
"rewards/rejected": -2.2885732650756836,
"step": 430
},
{
"epoch": 0.43,
"learning_rate": 3.494289060578478e-06,
"logits/chosen": -2.3979570865631104,
"logits/rejected": -2.3127408027648926,
"logps/chosen": -148.89285278320312,
"logps/rejected": -157.427978515625,
"loss": 0.5081,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.634016990661621,
"rewards/margins": 0.8799861669540405,
"rewards/rejected": -2.514003276824951,
"step": 440
},
{
"epoch": 0.44,
"learning_rate": 3.414875979798272e-06,
"logits/chosen": -2.349696636199951,
"logits/rejected": -2.236372709274292,
"logps/chosen": -163.39578247070312,
"logps/rejected": -173.43084716796875,
"loss": 0.4372,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.1473162174224854,
"rewards/margins": 1.2105666399002075,
"rewards/rejected": -2.3578829765319824,
"step": 450
},
{
"epoch": 0.45,
"learning_rate": 3.3343821450930196e-06,
"logits/chosen": -2.468815326690674,
"logits/rejected": -2.319728374481201,
"logps/chosen": -147.9761199951172,
"logps/rejected": -140.23440551757812,
"loss": 0.4893,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.0942740440368652,
"rewards/margins": 1.0382341146469116,
"rewards/rejected": -2.132507801055908,
"step": 460
},
{
"epoch": 0.46,
"learning_rate": 3.252902644791325e-06,
"logits/chosen": -2.2242140769958496,
"logits/rejected": -2.078382968902588,
"logps/chosen": -138.49478149414062,
"logps/rejected": -156.64273071289062,
"loss": 0.4146,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -1.0762813091278076,
"rewards/margins": 1.254016399383545,
"rewards/rejected": -2.3302977085113525,
"step": 470
},
{
"epoch": 0.47,
"learning_rate": 3.170533731600339e-06,
"logits/chosen": -2.334146022796631,
"logits/rejected": -2.1843693256378174,
"logps/chosen": -145.91741943359375,
"logps/rejected": -150.4651336669922,
"loss": 0.5102,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.1809916496276855,
"rewards/margins": 1.0365092754364014,
"rewards/rejected": -2.217501163482666,
"step": 480
},
{
"epoch": 0.48,
"learning_rate": 3.0873727089012816e-06,
"logits/chosen": -2.4750094413757324,
"logits/rejected": -2.3860108852386475,
"logps/chosen": -168.5699005126953,
"logps/rejected": -172.41029357910156,
"loss": 0.522,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.693511962890625,
"rewards/margins": 1.0143020153045654,
"rewards/rejected": -2.7078135013580322,
"step": 490
},
{
"epoch": 0.49,
"learning_rate": 3.0035178158038026e-06,
"logits/chosen": -2.261049747467041,
"logits/rejected": -2.0733211040496826,
"logps/chosen": -178.0288543701172,
"logps/rejected": -181.95970153808594,
"loss": 0.4784,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.8506263494491577,
"rewards/margins": 1.296245813369751,
"rewards/rejected": -3.146872043609619,
"step": 500
},
{
"epoch": 0.5,
"learning_rate": 2.919068111094937e-06,
"logits/chosen": -2.3573222160339355,
"logits/rejected": -2.2329063415527344,
"logps/chosen": -169.11073303222656,
"logps/rejected": -186.6231231689453,
"loss": 0.6133,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.2830090522766113,
"rewards/margins": 1.0279150009155273,
"rewards/rejected": -3.3109240531921387,
"step": 510
},
{
"epoch": 0.51,
"learning_rate": 2.8341233562197895e-06,
"logits/chosen": -2.3493080139160156,
"logits/rejected": -2.2869277000427246,
"logps/chosen": -156.89324951171875,
"logps/rejected": -172.35809326171875,
"loss": 0.4775,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.8899619579315186,
"rewards/margins": 1.054958701133728,
"rewards/rejected": -2.944920778274536,
"step": 520
},
{
"epoch": 0.52,
"learning_rate": 2.7487838974321352e-06,
"logits/chosen": -2.294079065322876,
"logits/rejected": -2.1721999645233154,
"logps/chosen": -166.351806640625,
"logps/rejected": -181.2619171142578,
"loss": 0.501,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -2.059480905532837,
"rewards/margins": 1.2059848308563232,
"rewards/rejected": -3.265465259552002,
"step": 530
},
{
"epoch": 0.53,
"learning_rate": 2.6631505472541997e-06,
"logits/chosen": -2.284674644470215,
"logits/rejected": -2.151689291000366,
"logps/chosen": -170.77108764648438,
"logps/rejected": -185.42703247070312,
"loss": 0.4457,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.8532861471176147,
"rewards/margins": 1.2717969417572021,
"rewards/rejected": -3.1250832080841064,
"step": 540
},
{
"epoch": 0.54,
"learning_rate": 2.5773244653856173e-06,
"logits/chosen": -2.262991428375244,
"logits/rejected": -2.140122652053833,
"logps/chosen": -171.88180541992188,
"logps/rejected": -183.91680908203125,
"loss": 0.4782,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.8343322277069092,
"rewards/margins": 1.209377408027649,
"rewards/rejected": -3.0437099933624268,
"step": 550
},
{
"epoch": 0.55,
"learning_rate": 2.4914070392022717e-06,
"logits/chosen": -2.287393093109131,
"logits/rejected": -2.1789438724517822,
"logps/chosen": -167.02993774414062,
"logps/rejected": -178.21737670898438,
"loss": 0.4847,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.4670352935791016,
"rewards/margins": 1.1378867626190186,
"rewards/rejected": -2.60492205619812,
"step": 560
},
{
"epoch": 0.56,
"learning_rate": 2.4054997639861778e-06,
"logits/chosen": -2.2121901512145996,
"logits/rejected": -2.039933443069458,
"logps/chosen": -164.4762725830078,
"logps/rejected": -171.06558227539062,
"loss": 0.4581,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.45037043094635,
"rewards/margins": 1.332580327987671,
"rewards/rejected": -2.7829508781433105,
"step": 570
},
{
"epoch": 0.57,
"learning_rate": 2.3197041230278905e-06,
"logits/chosen": -2.336905002593994,
"logits/rejected": -2.2310211658477783,
"logps/chosen": -159.6435089111328,
"logps/rejected": -185.4768829345703,
"loss": 0.4182,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -1.4801210165023804,
"rewards/margins": 1.5004690885543823,
"rewards/rejected": -2.980590343475342,
"step": 580
},
{
"epoch": 0.58,
"learning_rate": 2.234121467743082e-06,
"logits/chosen": -2.356192111968994,
"logits/rejected": -2.269792079925537,
"logps/chosen": -152.16677856445312,
"logps/rejected": -166.55563354492188,
"loss": 0.545,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.5818209648132324,
"rewards/margins": 1.1744325160980225,
"rewards/rejected": -2.756253719329834,
"step": 590
},
{
"epoch": 0.59,
"learning_rate": 2.148852897944905e-06,
"logits/chosen": -2.4117207527160645,
"logits/rejected": -2.2672557830810547,
"logps/chosen": -149.95504760742188,
"logps/rejected": -165.35263061523438,
"loss": 0.4388,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.488303303718567,
"rewards/margins": 1.227616786956787,
"rewards/rejected": -2.7159199714660645,
"step": 600
},
{
"epoch": 0.59,
"eval_logits/chosen": -2.1954238414764404,
"eval_logits/rejected": -2.1017305850982666,
"eval_logps/chosen": -161.69036865234375,
"eval_logps/rejected": -173.2707977294922,
"eval_loss": 0.5324557423591614,
"eval_rewards/accuracies": 0.7122641801834106,
"eval_rewards/chosen": -1.8012140989303589,
"eval_rewards/margins": 1.091520071029663,
"eval_rewards/rejected": -2.8927345275878906,
"eval_runtime": 427.5006,
"eval_samples_per_second": 0.978,
"eval_steps_per_second": 0.124,
"step": 600
},
{
"epoch": 0.6,
"learning_rate": 2.063999142413574e-06,
"logits/chosen": -2.3014285564422607,
"logits/rejected": -2.1665964126586914,
"logps/chosen": -164.73809814453125,
"logps/rejected": -163.736083984375,
"loss": 0.4451,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.6745765209197998,
"rewards/margins": 1.155146598815918,
"rewards/rejected": -2.8297228813171387,
"step": 610
},
{
"epoch": 0.61,
"learning_rate": 1.9796604399042547e-06,
"logits/chosen": -2.375365972518921,
"logits/rejected": -2.2335057258605957,
"logps/chosen": -170.6911163330078,
"logps/rejected": -186.2073974609375,
"loss": 0.4407,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.9759544134140015,
"rewards/margins": 1.4642221927642822,
"rewards/rejected": -3.440176486968994,
"step": 620
},
{
"epoch": 0.62,
"learning_rate": 1.8959364207338216e-06,
"logits/chosen": -2.3572235107421875,
"logits/rejected": -2.1937854290008545,
"logps/chosen": -148.90460205078125,
"logps/rejected": -174.04818725585938,
"loss": 0.5111,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.8767086267471313,
"rewards/margins": 1.4870884418487549,
"rewards/rejected": -3.363797426223755,
"step": 630
},
{
"epoch": 0.63,
"learning_rate": 1.8129259890863825e-06,
"logits/chosen": -2.3194022178649902,
"logits/rejected": -2.2175240516662598,
"logps/chosen": -165.7982177734375,
"logps/rejected": -185.64183044433594,
"loss": 0.5652,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.662651777267456,
"rewards/margins": 1.1808096170425415,
"rewards/rejected": -2.843461513519287,
"step": 640
},
{
"epoch": 0.64,
"learning_rate": 1.7307272061765738e-06,
"logits/chosen": -2.355379819869995,
"logits/rejected": -2.256904125213623,
"logps/chosen": -160.78778076171875,
"logps/rejected": -178.40591430664062,
"loss": 0.489,
"rewards/accuracies": 0.8125,
"rewards/chosen": -1.5761786699295044,
"rewards/margins": 1.2927827835083008,
"rewards/rejected": -2.8689613342285156,
"step": 650
},
{
"epoch": 0.65,
"learning_rate": 1.649437174408685e-06,
"logits/chosen": -2.2938830852508545,
"logits/rejected": -2.162132740020752,
"logps/chosen": -150.61842346191406,
"logps/rejected": -170.839111328125,
"loss": 0.5404,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.1881531476974487,
"rewards/margins": 1.283790111541748,
"rewards/rejected": -2.4719433784484863,
"step": 660
},
{
"epoch": 0.66,
"learning_rate": 1.569151922668422e-06,
"logits/chosen": -2.3966734409332275,
"logits/rejected": -2.287233829498291,
"logps/chosen": -138.367431640625,
"logps/rejected": -159.68118286132812,
"loss": 0.5196,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.2400791645050049,
"rewards/margins": 1.232775330543518,
"rewards/rejected": -2.4728543758392334,
"step": 670
},
{
"epoch": 0.67,
"learning_rate": 1.4899662928828428e-06,
"logits/chosen": -2.333634853363037,
"logits/rejected": -2.157811403274536,
"logps/chosen": -127.2032699584961,
"logps/rejected": -149.32907104492188,
"loss": 0.4387,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.0127990245819092,
"rewards/margins": 1.4315626621246338,
"rewards/rejected": -2.444361686706543,
"step": 680
},
{
"epoch": 0.68,
"learning_rate": 1.4119738279824507e-06,
"logits/chosen": -2.272979259490967,
"logits/rejected": -2.127248764038086,
"logps/chosen": -131.46400451660156,
"logps/rejected": -144.866455078125,
"loss": 0.5303,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.0165386199951172,
"rewards/margins": 1.0676348209381104,
"rewards/rejected": -2.0841732025146484,
"step": 690
},
{
"epoch": 0.69,
"learning_rate": 1.3352666613978152e-06,
"logits/chosen": -2.192830801010132,
"logits/rejected": -2.111144542694092,
"logps/chosen": -134.23226928710938,
"logps/rejected": -143.62905883789062,
"loss": 0.5064,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.0449942350387573,
"rewards/margins": 1.0384838581085205,
"rewards/rejected": -2.0834782123565674,
"step": 700
},
{
"epoch": 0.7,
"learning_rate": 1.2599354082212523e-06,
"logits/chosen": -2.2571260929107666,
"logits/rejected": -2.1350224018096924,
"logps/chosen": -129.79287719726562,
"logps/rejected": -138.2635498046875,
"loss": 0.5149,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.9037138223648071,
"rewards/margins": 0.9939008951187134,
"rewards/rejected": -1.8976147174835205,
"step": 710
},
{
"epoch": 0.71,
"learning_rate": 1.186069058162127e-06,
"logits/chosen": -2.3779895305633545,
"logits/rejected": -2.2798473834991455,
"logps/chosen": -136.55984497070312,
"logps/rejected": -145.81918334960938,
"loss": 0.5318,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.2205983400344849,
"rewards/margins": 0.9511237144470215,
"rewards/rejected": -2.171722173690796,
"step": 720
},
{
"epoch": 0.72,
"learning_rate": 1.113754870422254e-06,
"logits/chosen": -2.4235808849334717,
"logits/rejected": -2.3193821907043457,
"logps/chosen": -142.02517700195312,
"logps/rejected": -161.961669921875,
"loss": 0.5042,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.2923016548156738,
"rewards/margins": 0.9127277135848999,
"rewards/rejected": -2.205029249191284,
"step": 730
},
{
"epoch": 0.73,
"learning_rate": 1.0430782706155545e-06,
"logits/chosen": -2.435793399810791,
"logits/rejected": -2.296560764312744,
"logps/chosen": -145.57290649414062,
"logps/rejected": -156.23214721679688,
"loss": 0.4346,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.0421411991119385,
"rewards/margins": 1.3148539066314697,
"rewards/rejected": -2.356995105743408,
"step": 740
},
{
"epoch": 0.74,
"learning_rate": 9.741227498537615e-07,
"logits/chosen": -2.4774973392486572,
"logits/rejected": -2.3496251106262207,
"logps/chosen": -133.50381469726562,
"logps/rejected": -148.293701171875,
"loss": 0.4295,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.2879221439361572,
"rewards/margins": 1.1688247919082642,
"rewards/rejected": -2.45674729347229,
"step": 750
},
{
"epoch": 0.75,
"learning_rate": 9.069697661173668e-07,
"logits/chosen": -2.3304691314697266,
"logits/rejected": -2.2422666549682617,
"logps/chosen": -143.60433959960938,
"logps/rejected": -161.01876831054688,
"loss": 0.6187,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.4187496900558472,
"rewards/margins": 0.990622878074646,
"rewards/rejected": -2.4093728065490723,
"step": 760
},
{
"epoch": 0.76,
"learning_rate": 8.416986480283434e-07,
"logits/chosen": -2.2641918659210205,
"logits/rejected": -2.1294054985046387,
"logps/chosen": -150.9503173828125,
"logps/rejected": -169.94259643554688,
"loss": 0.5414,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.6702228784561157,
"rewards/margins": 1.2004982233047485,
"rewards/rejected": -2.8707211017608643,
"step": 770
},
{
"epoch": 0.77,
"learning_rate": 7.783865011382876e-07,
"logits/chosen": -2.318986415863037,
"logits/rejected": -2.189507246017456,
"logps/chosen": -157.97384643554688,
"logps/rejected": -164.84359741210938,
"loss": 0.4953,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.3389387130737305,
"rewards/margins": 1.2802064418792725,
"rewards/rejected": -2.619144916534424,
"step": 780
},
{
"epoch": 0.78,
"learning_rate": 7.171081168427205e-07,
"logits/chosen": -2.3565828800201416,
"logits/rejected": -2.193580150604248,
"logps/chosen": -169.49929809570312,
"logps/rejected": -177.23731994628906,
"loss": 0.5052,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.3413608074188232,
"rewards/margins": 1.271929144859314,
"rewards/rejected": -2.6132900714874268,
"step": 790
},
{
"epoch": 0.79,
"learning_rate": 6.579358840291064e-07,
"logits/chosen": -2.3053181171417236,
"logits/rejected": -2.205723762512207,
"logps/chosen": -155.66004943847656,
"logps/rejected": -173.5889434814453,
"loss": 0.6137,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.5923388004302979,
"rewards/margins": 0.8797184228897095,
"rewards/rejected": -2.4720568656921387,
"step": 800
},
{
"epoch": 0.79,
"eval_logits/chosen": -2.19348406791687,
"eval_logits/rejected": -2.0962369441986084,
"eval_logps/chosen": -149.9387664794922,
"eval_logps/rejected": -160.84133911132812,
"eval_loss": 0.5198491215705872,
"eval_rewards/accuracies": 0.7382075190544128,
"eval_rewards/chosen": -1.448665976524353,
"eval_rewards/margins": 1.0711843967437744,
"eval_rewards/rejected": -2.519850254058838,
"eval_runtime": 426.9839,
"eval_samples_per_second": 0.979,
"eval_steps_per_second": 0.124,
"step": 800
},
{
"epoch": 0.8,
"learning_rate": 6.00939703563006e-07,
"logits/chosen": -2.3648667335510254,
"logits/rejected": -2.2830593585968018,
"logps/chosen": -136.9246826171875,
"logps/rejected": -156.57559204101562,
"loss": 0.5648,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.4630320072174072,
"rewards/margins": 0.9899848699569702,
"rewards/rejected": -2.453016757965088,
"step": 810
},
{
"epoch": 0.81,
"learning_rate": 5.461869057133412e-07,
"logits/chosen": -2.408620595932007,
"logits/rejected": -2.319190502166748,
"logps/chosen": -145.22872924804688,
"logps/rejected": -167.64926147460938,
"loss": 0.487,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.306095838546753,
"rewards/margins": 1.1592603921890259,
"rewards/rejected": -2.4653563499450684,
"step": 820
},
{
"epoch": 0.82,
"learning_rate": 4.937421706143497e-07,
"logits/chosen": -2.290102481842041,
"logits/rejected": -2.158676862716675,
"logps/chosen": -152.98867797851562,
"logps/rejected": -159.16566467285156,
"loss": 0.5218,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.3972251415252686,
"rewards/margins": 1.101043462753296,
"rewards/rejected": -2.4982683658599854,
"step": 830
},
{
"epoch": 0.83,
"learning_rate": 4.43667451858166e-07,
"logits/chosen": -2.2902016639709473,
"logits/rejected": -2.131404399871826,
"logps/chosen": -134.6024627685547,
"logps/rejected": -151.9069061279297,
"loss": 0.4313,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -1.2463924884796143,
"rewards/margins": 1.268364667892456,
"rewards/rejected": -2.514756917953491,
"step": 840
},
{
"epoch": 0.84,
"learning_rate": 3.9602190330830484e-07,
"logits/chosen": -2.2570953369140625,
"logits/rejected": -2.157632827758789,
"logps/chosen": -141.2190704345703,
"logps/rejected": -162.86306762695312,
"loss": 0.4876,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.3870246410369873,
"rewards/margins": 1.1176416873931885,
"rewards/rejected": -2.504666328430176,
"step": 850
},
{
"epoch": 0.85,
"learning_rate": 3.5086180922049295e-07,
"logits/chosen": -2.394253969192505,
"logits/rejected": -2.2592263221740723,
"logps/chosen": -165.12005615234375,
"logps/rejected": -166.54446411132812,
"loss": 0.4366,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.3376173973083496,
"rewards/margins": 1.2365379333496094,
"rewards/rejected": -2.574155330657959,
"step": 860
},
{
"epoch": 0.86,
"learning_rate": 3.0824051775340895e-07,
"logits/chosen": -2.424966335296631,
"logits/rejected": -2.3199541568756104,
"logps/chosen": -124.06752014160156,
"logps/rejected": -140.36489868164062,
"loss": 0.5696,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.2786959409713745,
"rewards/margins": 0.8720223307609558,
"rewards/rejected": -2.1507184505462646,
"step": 870
},
{
"epoch": 0.87,
"learning_rate": 2.6820837794786336e-07,
"logits/chosen": -2.2567453384399414,
"logits/rejected": -2.1820857524871826,
"logps/chosen": -145.2144317626953,
"logps/rejected": -155.57261657714844,
"loss": 0.6788,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.4533874988555908,
"rewards/margins": 0.4883599281311035,
"rewards/rejected": -1.9417474269866943,
"step": 880
},
{
"epoch": 0.88,
"learning_rate": 2.3081268024887694e-07,
"logits/chosen": -2.2515053749084473,
"logits/rejected": -2.0878632068634033,
"logps/chosen": -141.62442016601562,
"logps/rejected": -151.8519287109375,
"loss": 0.4185,
"rewards/accuracies": 0.8125,
"rewards/chosen": -1.220234990119934,
"rewards/margins": 1.2837200164794922,
"rewards/rejected": -2.5039544105529785,
"step": 890
},
{
"epoch": 0.89,
"learning_rate": 1.9609760064091044e-07,
"logits/chosen": -2.343606472015381,
"logits/rejected": -2.277740955352783,
"logps/chosen": -144.82054138183594,
"logps/rejected": -142.51901245117188,
"loss": 0.5214,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.2351983785629272,
"rewards/margins": 0.8959378004074097,
"rewards/rejected": -2.131136417388916,
"step": 900
},
{
"epoch": 0.9,
"learning_rate": 1.6410414846224992e-07,
"logits/chosen": -2.256499767303467,
"logits/rejected": -2.1499085426330566,
"logps/chosen": -139.65145874023438,
"logps/rejected": -156.27642822265625,
"loss": 0.429,
"rewards/accuracies": 0.8125,
"rewards/chosen": -1.2983112335205078,
"rewards/margins": 1.2496817111968994,
"rewards/rejected": -2.547992706298828,
"step": 910
},
{
"epoch": 0.91,
"learning_rate": 1.348701179601819e-07,
"logits/chosen": -2.4174904823303223,
"logits/rejected": -2.2869112491607666,
"logps/chosen": -156.2935028076172,
"logps/rejected": -166.49362182617188,
"loss": 0.4412,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -1.221897840499878,
"rewards/margins": 1.1692349910736084,
"rewards/rejected": -2.3911328315734863,
"step": 920
},
{
"epoch": 0.91,
"learning_rate": 1.0843004364420151e-07,
"logits/chosen": -2.2345454692840576,
"logits/rejected": -2.1238150596618652,
"logps/chosen": -157.7463836669922,
"logps/rejected": -174.59848022460938,
"loss": 0.5407,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.309340476989746,
"rewards/margins": 0.9167596101760864,
"rewards/rejected": -2.226100206375122,
"step": 930
},
{
"epoch": 0.92,
"learning_rate": 8.481515948997931e-08,
"logits/chosen": -2.384162425994873,
"logits/rejected": -2.2841312885284424,
"logps/chosen": -161.34854125976562,
"logps/rejected": -156.95523071289062,
"loss": 0.6309,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.4349567890167236,
"rewards/margins": 0.7904176115989685,
"rewards/rejected": -2.225374460220337,
"step": 940
},
{
"epoch": 0.93,
"learning_rate": 6.4053362042297e-08,
"logits/chosen": -2.284637928009033,
"logits/rejected": -2.121169090270996,
"logps/chosen": -147.6913299560547,
"logps/rejected": -160.8138885498047,
"loss": 0.4361,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.3165013790130615,
"rewards/margins": 1.2154157161712646,
"rewards/rejected": -2.5319173336029053,
"step": 950
},
{
"epoch": 0.94,
"learning_rate": 4.616917746052163e-08,
"logits/chosen": -2.3731329441070557,
"logits/rejected": -2.2411139011383057,
"logps/chosen": -147.39578247070312,
"logps/rejected": -155.5648651123047,
"loss": 0.5548,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.4831774234771729,
"rewards/margins": 1.0112624168395996,
"rewards/rejected": -2.4944396018981934,
"step": 960
},
{
"epoch": 0.95,
"learning_rate": 3.118373254556412e-08,
"logits/chosen": -2.4090075492858887,
"logits/rejected": -2.2784786224365234,
"logps/chosen": -138.59571838378906,
"logps/rejected": -139.35055541992188,
"loss": 0.4862,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.2963029146194458,
"rewards/margins": 1.0239927768707275,
"rewards/rejected": -2.320295810699463,
"step": 970
},
{
"epoch": 0.96,
"learning_rate": 1.9114729782535037e-08,
"logits/chosen": -2.432577133178711,
"logits/rejected": -2.3220748901367188,
"logps/chosen": -136.9718017578125,
"logps/rejected": -146.90240478515625,
"loss": 0.5037,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.215503454208374,
"rewards/margins": 0.9537515640258789,
"rewards/rejected": -2.169254779815674,
"step": 980
},
{
"epoch": 0.97,
"learning_rate": 9.97642642858815e-09,
"logits/chosen": -2.30542254447937,
"logits/rejected": -2.1695542335510254,
"logps/chosen": -152.14138793945312,
"logps/rejected": -153.89279174804688,
"loss": 0.5001,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.3888603448867798,
"rewards/margins": 0.9819372296333313,
"rewards/rejected": -2.3707973957061768,
"step": 990
},
{
"epoch": 0.98,
"learning_rate": 3.779617670651436e-09,
"logits/chosen": -2.278373956680298,
"logits/rejected": -2.17798113822937,
"logps/chosen": -157.8939208984375,
"logps/rejected": -168.04055786132812,
"loss": 0.5866,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.575810194015503,
"rewards/margins": 0.995488166809082,
"rewards/rejected": -2.571298599243164,
"step": 1000
},
{
"epoch": 0.98,
"eval_logits/chosen": -2.192172050476074,
"eval_logits/rejected": -2.095205545425415,
"eval_logps/chosen": -147.71951293945312,
"eval_logps/rejected": -158.54696655273438,
"eval_loss": 0.5194684863090515,
"eval_rewards/accuracies": 0.7358490824699402,
"eval_rewards/chosen": -1.382088541984558,
"eval_rewards/margins": 1.068930983543396,
"eval_rewards/rejected": -2.451019525527954,
"eval_runtime": 427.3583,
"eval_samples_per_second": 0.978,
"eval_steps_per_second": 0.124,
"step": 1000
},
{
"epoch": 0.99,
"learning_rate": 5.316238729444201e-10,
"logits/chosen": -2.2608113288879395,
"logits/rejected": -2.101694345474243,
"logps/chosen": -152.0623779296875,
"logps/rejected": -162.7583770751953,
"loss": 0.4821,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.5301237106323242,
"rewards/margins": 1.2161496877670288,
"rewards/rejected": -2.7462732791900635,
"step": 1010
},
{
"epoch": 1.0,
"step": 1016,
"total_flos": 0.0,
"train_loss": 0.5536013367607837,
"train_runtime": 17040.3461,
"train_samples_per_second": 0.477,
"train_steps_per_second": 0.06
}
],
"logging_steps": 10,
"max_steps": 1016,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}