gemma-7b-softplus-basic-5e-5 / trainer_state.json
silviasapora's picture
Model save
c57d7a0 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.9765925925925925,
"eval_steps": 500,
"global_step": 315,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.047407407407407405,
"grad_norm": 899.4839477539062,
"learning_rate": 7.8125e-06,
"log_odds_chosen": -3.253441572189331,
"log_odds_ratio": -11.290441513061523,
"logits/chosen": 151.96023559570312,
"logits/rejected": 125.16461181640625,
"logps/chosen": -21.246450424194336,
"logps/rejected": -17.994016647338867,
"loss": 456.0568,
"nll_loss": 9.107990264892578,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -10.623225212097168,
"rewards/margins": -1.6262180805206299,
"rewards/rejected": -8.997008323669434,
"step": 5
},
{
"epoch": 0.09481481481481481,
"grad_norm": 665.6010131835938,
"learning_rate": 1.5625e-05,
"log_odds_chosen": -0.5072727203369141,
"log_odds_ratio": -9.659940719604492,
"logits/chosen": 151.84805297851562,
"logits/rejected": 114.96217346191406,
"logps/chosen": -19.77961540222168,
"logps/rejected": -19.273290634155273,
"loss": 394.0516,
"nll_loss": 8.324145317077637,
"rewards/accuracies": 0.40625,
"rewards/chosen": -9.88980770111084,
"rewards/margins": -0.25316303968429565,
"rewards/rejected": -9.636645317077637,
"step": 10
},
{
"epoch": 0.14222222222222222,
"grad_norm": 584.8187255859375,
"learning_rate": 2.34375e-05,
"log_odds_chosen": -3.0495519638061523,
"log_odds_ratio": -12.564888000488281,
"logits/chosen": 165.22348022460938,
"logits/rejected": 135.7709197998047,
"logps/chosen": -23.73749351501465,
"logps/rejected": -20.685997009277344,
"loss": 398.6285,
"nll_loss": 8.013145446777344,
"rewards/accuracies": 0.45625001192092896,
"rewards/chosen": -11.868746757507324,
"rewards/margins": -1.5257480144500732,
"rewards/rejected": -10.342998504638672,
"step": 15
},
{
"epoch": 0.18962962962962962,
"grad_norm": 2222.182861328125,
"learning_rate": 3.125e-05,
"log_odds_chosen": -1.6242220401763916,
"log_odds_ratio": -6.402337551116943,
"logits/chosen": 166.9097137451172,
"logits/rejected": 148.36062622070312,
"logps/chosen": -13.310510635375977,
"logps/rejected": -11.688575744628906,
"loss": 263.4254,
"nll_loss": 5.480126857757568,
"rewards/accuracies": 0.4312500059604645,
"rewards/chosen": -6.655255317687988,
"rewards/margins": -0.8109671473503113,
"rewards/rejected": -5.844287872314453,
"step": 20
},
{
"epoch": 0.23703703703703705,
"grad_norm": 239.38790893554688,
"learning_rate": 3.90625e-05,
"log_odds_chosen": -0.3463691771030426,
"log_odds_ratio": -1.4706324338912964,
"logits/chosen": 178.37814331054688,
"logits/rejected": 188.9853515625,
"logps/chosen": -2.9337403774261475,
"logps/rejected": -2.58659291267395,
"loss": 101.069,
"nll_loss": 2.465604305267334,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -1.4668701887130737,
"rewards/margins": -0.17357376217842102,
"rewards/rejected": -1.293296456336975,
"step": 25
},
{
"epoch": 0.28444444444444444,
"grad_norm": 168.3093719482422,
"learning_rate": 4.6875e-05,
"log_odds_chosen": -0.08818509429693222,
"log_odds_ratio": -0.9202126264572144,
"logits/chosen": 208.46914672851562,
"logits/rejected": 211.0690155029297,
"logps/chosen": -1.8011804819107056,
"logps/rejected": -1.7284847497940063,
"loss": 76.8201,
"nll_loss": 1.9414145946502686,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.9005902409553528,
"rewards/margins": -0.036347877234220505,
"rewards/rejected": -0.8642423748970032,
"step": 30
},
{
"epoch": 0.33185185185185184,
"grad_norm": 344.3992919921875,
"learning_rate": 4.998613757348784e-05,
"log_odds_chosen": 0.2355690896511078,
"log_odds_ratio": -0.7713075876235962,
"logits/chosen": 221.82272338867188,
"logits/rejected": 217.92764282226562,
"logps/chosen": -1.5275888442993164,
"logps/rejected": -1.7404054403305054,
"loss": 73.3205,
"nll_loss": 1.7808992862701416,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.7637944221496582,
"rewards/margins": 0.1064082607626915,
"rewards/rejected": -0.8702027201652527,
"step": 35
},
{
"epoch": 0.37925925925925924,
"grad_norm": 205.59912109375,
"learning_rate": 4.990147841143462e-05,
"log_odds_chosen": 0.3138998746871948,
"log_odds_ratio": -0.7420972585678101,
"logits/chosen": 225.82821655273438,
"logits/rejected": 226.96774291992188,
"logps/chosen": -1.5016671419143677,
"logps/rejected": -1.7676680088043213,
"loss": 71.1531,
"nll_loss": 1.7955719232559204,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.7508335709571838,
"rewards/margins": 0.1330004334449768,
"rewards/rejected": -0.8838340044021606,
"step": 40
},
{
"epoch": 0.4266666666666667,
"grad_norm": 139.50970458984375,
"learning_rate": 4.97401218720448e-05,
"log_odds_chosen": 0.1954536736011505,
"log_odds_ratio": -0.72107994556427,
"logits/chosen": 208.2395782470703,
"logits/rejected": 205.417236328125,
"logps/chosen": -1.3215817213058472,
"logps/rejected": -1.5023378133773804,
"loss": 66.4276,
"nll_loss": 1.647528886795044,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.6607908606529236,
"rewards/margins": 0.0903780609369278,
"rewards/rejected": -0.7511689066886902,
"step": 45
},
{
"epoch": 0.4740740740740741,
"grad_norm": 77.09749603271484,
"learning_rate": 4.9502564938797946e-05,
"log_odds_chosen": 0.19976872205734253,
"log_odds_ratio": -0.7074006199836731,
"logits/chosen": 202.8539276123047,
"logits/rejected": 195.70155334472656,
"logps/chosen": -1.2350599765777588,
"logps/rejected": -1.4153530597686768,
"loss": 63.0281,
"nll_loss": 1.5506360530853271,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.6175299882888794,
"rewards/margins": 0.0901465192437172,
"rewards/rejected": -0.7076765298843384,
"step": 50
},
{
"epoch": 0.5214814814814814,
"grad_norm": 59.40414810180664,
"learning_rate": 4.918953929490768e-05,
"log_odds_chosen": 0.14897426962852478,
"log_odds_ratio": -0.7380977869033813,
"logits/chosen": 190.48110961914062,
"logits/rejected": 195.70455932617188,
"logps/chosen": -1.221083641052246,
"logps/rejected": -1.3429195880889893,
"loss": 62.9068,
"nll_loss": 1.543068528175354,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.610541820526123,
"rewards/margins": 0.0609179362654686,
"rewards/rejected": -0.6714597940444946,
"step": 55
},
{
"epoch": 0.5688888888888889,
"grad_norm": 126.07240295410156,
"learning_rate": 4.88020090697132e-05,
"log_odds_chosen": 0.15872475504875183,
"log_odds_ratio": -0.7120335102081299,
"logits/chosen": 186.23619079589844,
"logits/rejected": 179.56393432617188,
"logps/chosen": -1.2440166473388672,
"logps/rejected": -1.3791967630386353,
"loss": 63.2646,
"nll_loss": 1.5736169815063477,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.6220083236694336,
"rewards/margins": 0.06759000569581985,
"rewards/rejected": -0.6895983815193176,
"step": 60
},
{
"epoch": 0.6162962962962963,
"grad_norm": 139.16448974609375,
"learning_rate": 4.834116786912897e-05,
"log_odds_chosen": 0.18847045302391052,
"log_odds_ratio": -0.7004148364067078,
"logits/chosen": 180.8818359375,
"logits/rejected": 185.00570678710938,
"logps/chosen": -1.2013404369354248,
"logps/rejected": -1.3533903360366821,
"loss": 60.8734,
"nll_loss": 1.485313057899475,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.6006702184677124,
"rewards/margins": 0.07602496445178986,
"rewards/rejected": -0.6766951680183411,
"step": 65
},
{
"epoch": 0.6637037037037037,
"grad_norm": 251.40016174316406,
"learning_rate": 4.7808435099299045e-05,
"log_odds_chosen": 0.18260684609413147,
"log_odds_ratio": -0.7129999995231628,
"logits/chosen": 182.959716796875,
"logits/rejected": 183.2157745361328,
"logps/chosen": -1.1487746238708496,
"logps/rejected": -1.2979029417037964,
"loss": 61.4203,
"nll_loss": 1.4665228128433228,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5743873119354248,
"rewards/margins": 0.07456417381763458,
"rewards/rejected": -0.6489514708518982,
"step": 70
},
{
"epoch": 0.7111111111111111,
"grad_norm": 124.29930877685547,
"learning_rate": 4.720545159477922e-05,
"log_odds_chosen": 0.17588524520397186,
"log_odds_ratio": -0.7309717535972595,
"logits/chosen": 175.79225158691406,
"logits/rejected": 185.4251708984375,
"logps/chosen": -1.171048641204834,
"logps/rejected": -1.3039839267730713,
"loss": 60.8133,
"nll_loss": 1.5102540254592896,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.585524320602417,
"rewards/margins": 0.06646756827831268,
"rewards/rejected": -0.6519919633865356,
"step": 75
},
{
"epoch": 0.7585185185185185,
"grad_norm": 358.4803771972656,
"learning_rate": 4.653407456471222e-05,
"log_odds_chosen": 0.1115400418639183,
"log_odds_ratio": -0.7338749170303345,
"logits/chosen": 191.45970153808594,
"logits/rejected": 181.09698486328125,
"logps/chosen": -1.1183122396469116,
"logps/rejected": -1.2054791450500488,
"loss": 57.9734,
"nll_loss": 1.4206385612487793,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.5591561198234558,
"rewards/margins": 0.04358343034982681,
"rewards/rejected": -0.6027395725250244,
"step": 80
},
{
"epoch": 0.8059259259259259,
"grad_norm": 136.4271697998047,
"learning_rate": 4.579637187256222e-05,
"log_odds_chosen": 0.04175865277647972,
"log_odds_ratio": -0.7415879964828491,
"logits/chosen": 192.60121154785156,
"logits/rejected": 192.03916931152344,
"logps/chosen": -1.0868916511535645,
"logps/rejected": -1.1372497081756592,
"loss": 59.0801,
"nll_loss": 1.4550647735595703,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5434458255767822,
"rewards/margins": 0.025179097428917885,
"rewards/rejected": -0.5686248540878296,
"step": 85
},
{
"epoch": 0.8533333333333334,
"grad_norm": 56.09324645996094,
"learning_rate": 4.499461566702685e-05,
"log_odds_chosen": 0.12758071720600128,
"log_odds_ratio": -0.7168221473693848,
"logits/chosen": 182.20901489257812,
"logits/rejected": 187.005126953125,
"logps/chosen": -1.0412800312042236,
"logps/rejected": -1.140512466430664,
"loss": 56.929,
"nll_loss": 1.3427871465682983,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5206400156021118,
"rewards/margins": 0.04961626976728439,
"rewards/rejected": -0.570256233215332,
"step": 90
},
{
"epoch": 0.9007407407407407,
"grad_norm": 101.89226531982422,
"learning_rate": 4.413127538374411e-05,
"log_odds_chosen": 0.03433248773217201,
"log_odds_ratio": -0.7621655464172363,
"logits/chosen": 183.79061889648438,
"logits/rejected": 181.02096557617188,
"logps/chosen": -1.1271755695343018,
"logps/rejected": -1.155602216720581,
"loss": 57.6873,
"nll_loss": 1.4238755702972412,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.5635877847671509,
"rewards/margins": 0.014213320799171925,
"rewards/rejected": -0.5778011083602905,
"step": 95
},
{
"epoch": 0.9481481481481482,
"grad_norm": 56.19839859008789,
"learning_rate": 4.320901013934887e-05,
"log_odds_chosen": 0.19043493270874023,
"log_odds_ratio": -0.6898019313812256,
"logits/chosen": 187.52651977539062,
"logits/rejected": 188.4263458251953,
"logps/chosen": -1.064992904663086,
"logps/rejected": -1.2005788087844849,
"loss": 56.784,
"nll_loss": 1.3723514080047607,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.532496452331543,
"rewards/margins": 0.06779297441244125,
"rewards/rejected": -0.6002894043922424,
"step": 100
},
{
"epoch": 0.9955555555555555,
"grad_norm": 134.25718688964844,
"learning_rate": 4.223066054130568e-05,
"log_odds_chosen": 0.19098126888275146,
"log_odds_ratio": -0.708988606929779,
"logits/chosen": 186.099365234375,
"logits/rejected": 191.89564514160156,
"logps/chosen": -1.0854551792144775,
"logps/rejected": -1.2490203380584717,
"loss": 59.6264,
"nll_loss": 1.4756276607513428,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5427275896072388,
"rewards/margins": 0.0817825049161911,
"rewards/rejected": -0.6245101690292358,
"step": 105
},
{
"epoch": 1.037925925925926,
"grad_norm": 48.896156311035156,
"learning_rate": 4.1199239938743797e-05,
"log_odds_chosen": 0.09143175929784775,
"log_odds_ratio": -0.7199470400810242,
"logits/chosen": 192.02879333496094,
"logits/rejected": 189.01097106933594,
"logps/chosen": -1.0210695266723633,
"logps/rejected": -1.0847587585449219,
"loss": 46.1645,
"nll_loss": 1.2595568895339966,
"rewards/accuracies": 0.5384615659713745,
"rewards/chosen": -0.5105347633361816,
"rewards/margins": 0.031844645738601685,
"rewards/rejected": -0.5423793792724609,
"step": 110
},
{
"epoch": 1.0853333333333333,
"grad_norm": 44.063316345214844,
"learning_rate": 4.0117925141242174e-05,
"log_odds_chosen": 0.09873426705598831,
"log_odds_ratio": -0.741472601890564,
"logits/chosen": 185.77395629882812,
"logits/rejected": 194.8834686279297,
"logps/chosen": -0.9893190264701843,
"logps/rejected": -1.0386199951171875,
"loss": 50.7825,
"nll_loss": 1.2114965915679932,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.49465951323509216,
"rewards/margins": 0.024650435894727707,
"rewards/rejected": -0.5193099975585938,
"step": 115
},
{
"epoch": 1.1327407407407408,
"grad_norm": 42.30266571044922,
"learning_rate": 3.899004663415084e-05,
"log_odds_chosen": 0.03399781137704849,
"log_odds_ratio": -0.7742182612419128,
"logits/chosen": 182.58143615722656,
"logits/rejected": 186.83212280273438,
"logps/chosen": -0.9977661967277527,
"logps/rejected": -1.0048539638519287,
"loss": 51.3458,
"nll_loss": 1.2454302310943604,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.49888309836387634,
"rewards/margins": 0.0035438979975879192,
"rewards/rejected": -0.5024269819259644,
"step": 120
},
{
"epoch": 1.1801481481481482,
"grad_norm": 125.85904693603516,
"learning_rate": 3.781907832058587e-05,
"log_odds_chosen": 0.09150873124599457,
"log_odds_ratio": -0.7233962416648865,
"logits/chosen": 181.49447631835938,
"logits/rejected": 180.0559844970703,
"logps/chosen": -0.9917734861373901,
"logps/rejected": -1.051242470741272,
"loss": 49.8693,
"nll_loss": 1.1629549264907837,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.49588674306869507,
"rewards/margins": 0.029734421521425247,
"rewards/rejected": -0.525621235370636,
"step": 125
},
{
"epoch": 1.2275555555555555,
"grad_norm": 39.95571517944336,
"learning_rate": 3.660862682169282e-05,
"log_odds_chosen": 0.09172184765338898,
"log_odds_ratio": -0.7451839447021484,
"logits/chosen": 179.13302612304688,
"logits/rejected": 183.4129638671875,
"logps/chosen": -1.0290215015411377,
"logps/rejected": -1.0855556726455688,
"loss": 51.0956,
"nll_loss": 1.1969176530838013,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.5145107507705688,
"rewards/margins": 0.028267022222280502,
"rewards/rejected": -0.5427778363227844,
"step": 130
},
{
"epoch": 1.274962962962963,
"grad_norm": 115.57888793945312,
"learning_rate": 3.5362420368134356e-05,
"log_odds_chosen": 0.07470552623271942,
"log_odds_ratio": -0.7265952825546265,
"logits/chosen": 183.11488342285156,
"logits/rejected": 177.27346801757812,
"logps/chosen": -0.9192112684249878,
"logps/rejected": -0.9482501745223999,
"loss": 49.3732,
"nll_loss": 1.1410008668899536,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.4596056342124939,
"rewards/margins": 0.0145194623619318,
"rewards/rejected": -0.47412508726119995,
"step": 135
},
{
"epoch": 1.3223703703703704,
"grad_norm": 40.40840148925781,
"learning_rate": 3.408429731701635e-05,
"log_odds_chosen": 0.15658657252788544,
"log_odds_ratio": -0.7292534112930298,
"logits/chosen": 183.2898712158203,
"logits/rejected": 183.63088989257812,
"logps/chosen": -0.9870889782905579,
"logps/rejected": -1.0798033475875854,
"loss": 51.391,
"nll_loss": 1.1972719430923462,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.49354448914527893,
"rewards/margins": 0.046357229351997375,
"rewards/rejected": -0.5399016737937927,
"step": 140
},
{
"epoch": 1.3697777777777778,
"grad_norm": 43.5965461730957,
"learning_rate": 3.2778194329621104e-05,
"log_odds_chosen": 0.0713091716170311,
"log_odds_ratio": -0.7587572932243347,
"logits/chosen": 175.42311096191406,
"logits/rejected": 179.28359985351562,
"logps/chosen": -0.9829910397529602,
"logps/rejected": -1.0261934995651245,
"loss": 51.113,
"nll_loss": 1.2084944248199463,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.4914955198764801,
"rewards/margins": 0.021601270884275436,
"rewards/rejected": -0.5130967497825623,
"step": 145
},
{
"epoch": 1.417185185185185,
"grad_norm": 68.55988311767578,
"learning_rate": 3.144813424636031e-05,
"log_odds_chosen": 0.03821509703993797,
"log_odds_ratio": -0.7714879512786865,
"logits/chosen": 173.73973083496094,
"logits/rejected": 177.93600463867188,
"logps/chosen": -1.0191147327423096,
"logps/rejected": -1.0414360761642456,
"loss": 49.2398,
"nll_loss": 1.162936806678772,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.5095573663711548,
"rewards/margins": 0.011160662397742271,
"rewards/rejected": -0.5207180380821228,
"step": 150
},
{
"epoch": 1.4645925925925927,
"grad_norm": 87.03449249267578,
"learning_rate": 3.0098213696293542e-05,
"log_odds_chosen": 0.10836371034383774,
"log_odds_ratio": -0.7314402461051941,
"logits/chosen": 175.88619995117188,
"logits/rejected": 182.52371215820312,
"logps/chosen": -0.9865490794181824,
"logps/rejected": -1.0499975681304932,
"loss": 50.2605,
"nll_loss": 1.1926219463348389,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.4932745397090912,
"rewards/margins": 0.03172413632273674,
"rewards/rejected": -0.5249987840652466,
"step": 155
},
{
"epoch": 1.512,
"grad_norm": 58.0168342590332,
"learning_rate": 2.8732590479375165e-05,
"log_odds_chosen": 0.12078354507684708,
"log_odds_ratio": -0.7301235795021057,
"logits/chosen": 173.8167266845703,
"logits/rejected": 181.9156494140625,
"logps/chosen": -0.9404324293136597,
"logps/rejected": -1.0260807275772095,
"loss": 48.9856,
"nll_loss": 1.1453100442886353,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.47021621465682983,
"rewards/margins": 0.0428241565823555,
"rewards/rejected": -0.5130403637886047,
"step": 160
},
{
"epoch": 1.5594074074074074,
"grad_norm": 47.982913970947266,
"learning_rate": 2.7355470760292956e-05,
"log_odds_chosen": 0.0946766808629036,
"log_odds_ratio": -0.7334403395652771,
"logits/chosen": 166.52523803710938,
"logits/rejected": 171.69485473632812,
"logps/chosen": -0.9203466176986694,
"logps/rejected": -0.9718505144119263,
"loss": 50.1007,
"nll_loss": 1.1756643056869507,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.4601733088493347,
"rewards/margins": 0.025751952081918716,
"rewards/rejected": -0.48592525720596313,
"step": 165
},
{
"epoch": 1.6068148148148147,
"grad_norm": 50.041561126708984,
"learning_rate": 2.597109611334169e-05,
"log_odds_chosen": 0.17343485355377197,
"log_odds_ratio": -0.6794081926345825,
"logits/chosen": 171.59066772460938,
"logits/rejected": 179.27670288085938,
"logps/chosen": -0.936198353767395,
"logps/rejected": -1.0650147199630737,
"loss": 49.7789,
"nll_loss": 1.186452031135559,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.4680991768836975,
"rewards/margins": 0.06440822780132294,
"rewards/rejected": -0.5325073599815369,
"step": 170
},
{
"epoch": 1.6542222222222223,
"grad_norm": 47.45258331298828,
"learning_rate": 2.458373045823404e-05,
"log_odds_chosen": 0.07308103144168854,
"log_odds_ratio": -0.7454826235771179,
"logits/chosen": 172.97149658203125,
"logits/rejected": 172.43182373046875,
"logps/chosen": -0.9455814361572266,
"logps/rejected": -1.0050225257873535,
"loss": 49.1083,
"nll_loss": 1.1290260553359985,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.4727907180786133,
"rewards/margins": 0.029720569029450417,
"rewards/rejected": -0.5025112628936768,
"step": 175
},
{
"epoch": 1.7016296296296296,
"grad_norm": 58.00626754760742,
"learning_rate": 2.3197646927086697e-05,
"log_odds_chosen": 0.1577189862728119,
"log_odds_ratio": -0.7023075819015503,
"logits/chosen": 168.512451171875,
"logits/rejected": 175.3289794921875,
"logps/chosen": -0.9105680584907532,
"logps/rejected": -1.0289887189865112,
"loss": 49.2958,
"nll_loss": 1.1708818674087524,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.4552840292453766,
"rewards/margins": 0.059210360050201416,
"rewards/rejected": -0.5144943594932556,
"step": 180
},
{
"epoch": 1.749037037037037,
"grad_norm": 34.98508071899414,
"learning_rate": 2.1817114703032176e-05,
"log_odds_chosen": 0.12351224571466446,
"log_odds_ratio": -0.7270097732543945,
"logits/chosen": 173.38623046875,
"logits/rejected": 165.6890869140625,
"logps/chosen": -0.9919120669364929,
"logps/rejected": -1.0763611793518066,
"loss": 48.9183,
"nll_loss": 1.1456924676895142,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.49595603346824646,
"rewards/margins": 0.04222451522946358,
"rewards/rejected": -0.5381805896759033,
"step": 185
},
{
"epoch": 1.7964444444444445,
"grad_norm": 35.57364273071289,
"learning_rate": 2.0446385870993467e-05,
"log_odds_chosen": 0.046650804579257965,
"log_odds_ratio": -0.7537095546722412,
"logits/chosen": 161.9040069580078,
"logits/rejected": 164.87686157226562,
"logps/chosen": -0.9197586178779602,
"logps/rejected": -0.9383406639099121,
"loss": 49.163,
"nll_loss": 1.1618720293045044,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.4598793089389801,
"rewards/margins": 0.009291025809943676,
"rewards/rejected": -0.46917033195495605,
"step": 190
},
{
"epoch": 1.8438518518518519,
"grad_norm": 57.462890625,
"learning_rate": 1.9089682321121834e-05,
"log_odds_chosen": 0.10921354591846466,
"log_odds_ratio": -0.7022596001625061,
"logits/chosen": 164.3706817626953,
"logits/rejected": 167.53907775878906,
"logps/chosen": -0.9306834936141968,
"logps/rejected": -1.0057896375656128,
"loss": 48.8493,
"nll_loss": 1.1373628377914429,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.4653417468070984,
"rewards/margins": 0.03755306452512741,
"rewards/rejected": -0.5028948187828064,
"step": 195
},
{
"epoch": 1.8912592592592592,
"grad_norm": 51.58010482788086,
"learning_rate": 1.775118274523545e-05,
"log_odds_chosen": 0.0803312212228775,
"log_odds_ratio": -0.7448180913925171,
"logits/chosen": 177.42755126953125,
"logits/rejected": 166.17420959472656,
"logps/chosen": -0.971086323261261,
"logps/rejected": -1.025727391242981,
"loss": 50.0697,
"nll_loss": 1.1380794048309326,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.4855431616306305,
"rewards/margins": 0.027320479974150658,
"rewards/rejected": -0.5128636956214905,
"step": 200
},
{
"epoch": 1.9386666666666668,
"grad_norm": 40.688987731933594,
"learning_rate": 1.643500976631037e-05,
"log_odds_chosen": 0.10912259668111801,
"log_odds_ratio": -0.7380350232124329,
"logits/chosen": 173.36325073242188,
"logits/rejected": 171.0248565673828,
"logps/chosen": -0.9703338742256165,
"logps/rejected": -1.0370721817016602,
"loss": 48.6517,
"nll_loss": 1.1243274211883545,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.4851669371128082,
"rewards/margins": 0.03336922079324722,
"rewards/rejected": -0.5185360908508301,
"step": 205
},
{
"epoch": 1.986074074074074,
"grad_norm": 47.03554153442383,
"learning_rate": 1.514521724066537e-05,
"log_odds_chosen": 0.07311935722827911,
"log_odds_ratio": -0.7410269975662231,
"logits/chosen": 161.46792602539062,
"logits/rejected": 176.0153045654297,
"logps/chosen": -0.9304733276367188,
"logps/rejected": -0.9752525091171265,
"loss": 49.3022,
"nll_loss": 1.163869857788086,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.4652366638183594,
"rewards/margins": 0.02238963544368744,
"rewards/rejected": -0.48762625455856323,
"step": 210
},
{
"epoch": 2.0284444444444443,
"grad_norm": 46.73811721801758,
"learning_rate": 1.3885777771950348e-05,
"log_odds_chosen": 0.04928523674607277,
"log_odds_ratio": -0.7592554688453674,
"logits/chosen": 154.12142944335938,
"logits/rejected": 166.54530334472656,
"logps/chosen": -0.8669578433036804,
"logps/rejected": -0.8768206238746643,
"loss": 38.0486,
"nll_loss": 0.969521701335907,
"rewards/accuracies": 0.5314685106277466,
"rewards/chosen": -0.4334789216518402,
"rewards/margins": 0.004931411240249872,
"rewards/rejected": -0.43841031193733215,
"step": 215
},
{
"epoch": 2.075851851851852,
"grad_norm": 66.5871353149414,
"learning_rate": 1.2660570475395683e-05,
"log_odds_chosen": 0.07944320142269135,
"log_odds_ratio": -0.7612829804420471,
"logits/chosen": 158.1134033203125,
"logits/rejected": 145.2221221923828,
"logps/chosen": -0.8715459704399109,
"logps/rejected": -0.8965581059455872,
"loss": 42.355,
"nll_loss": 0.9041290283203125,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.43577298521995544,
"rewards/margins": 0.01250611525028944,
"rewards/rejected": -0.4482790529727936,
"step": 220
},
{
"epoch": 2.1232592592592594,
"grad_norm": 43.993412017822266,
"learning_rate": 1.1473369030008974e-05,
"log_odds_chosen": 0.07922948896884918,
"log_odds_ratio": -0.748726487159729,
"logits/chosen": 150.79869079589844,
"logits/rejected": 159.51519775390625,
"logps/chosen": -0.9018028974533081,
"logps/rejected": -0.9198867082595825,
"loss": 41.5811,
"nll_loss": 0.9309173822402954,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.45090144872665405,
"rewards/margins": 0.009041833691298962,
"rewards/rejected": -0.45994335412979126,
"step": 225
},
{
"epoch": 2.1706666666666665,
"grad_norm": 42.56783676147461,
"learning_rate": 1.0327830055518842e-05,
"log_odds_chosen": 0.1404309719800949,
"log_odds_ratio": -0.7541959285736084,
"logits/chosen": 149.96189880371094,
"logits/rejected": 155.0712127685547,
"logps/chosen": -0.8312407732009888,
"logps/rejected": -0.9046605825424194,
"loss": 41.5919,
"nll_loss": 0.9145228266716003,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.4156203866004944,
"rewards/margins": 0.03670995682477951,
"rewards/rejected": -0.4523302912712097,
"step": 230
},
{
"epoch": 2.218074074074074,
"grad_norm": 46.45758819580078,
"learning_rate": 9.227481849865235e-06,
"log_odds_chosen": -0.03481599688529968,
"log_odds_ratio": -0.8210490942001343,
"logits/chosen": 156.96890258789062,
"logits/rejected": 156.92953491210938,
"logps/chosen": -0.899451732635498,
"logps/rejected": -0.8785243034362793,
"loss": 42.667,
"nll_loss": 0.988889217376709,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.449725866317749,
"rewards/margins": -0.010463694110512733,
"rewards/rejected": -0.43926215171813965,
"step": 235
},
{
"epoch": 2.2654814814814817,
"grad_norm": 41.77062225341797,
"learning_rate": 8.175713521924978e-06,
"log_odds_chosen": 0.19800688326358795,
"log_odds_ratio": -0.7289772033691406,
"logits/chosen": 161.128173828125,
"logits/rejected": 159.4349365234375,
"logps/chosen": -0.8593847155570984,
"logps/rejected": -0.9744852185249329,
"loss": 41.5045,
"nll_loss": 0.9035336375236511,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.4296923577785492,
"rewards/margins": 0.05755022168159485,
"rewards/rejected": -0.48724260926246643,
"step": 240
},
{
"epoch": 2.3128888888888888,
"grad_norm": 41.01064682006836,
"learning_rate": 7.1757645529443665e-06,
"log_odds_chosen": -0.0014010012382641435,
"log_odds_ratio": -0.7947742342948914,
"logits/chosen": 155.06546020507812,
"logits/rejected": 155.10775756835938,
"logps/chosen": -0.878911018371582,
"logps/rejected": -0.8687770962715149,
"loss": 41.6816,
"nll_loss": 0.9432564973831177,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.439455509185791,
"rewards/margins": -0.0050669461488723755,
"rewards/rejected": -0.43438854813575745,
"step": 245
},
{
"epoch": 2.3602962962962963,
"grad_norm": 41.34367752075195,
"learning_rate": 6.230714818829733e-06,
"log_odds_chosen": 0.0013508498668670654,
"log_odds_ratio": -0.7936294674873352,
"logits/chosen": 153.34498596191406,
"logits/rejected": 158.2041015625,
"logps/chosen": -0.8574392199516296,
"logps/rejected": -0.8317993879318237,
"loss": 42.0429,
"nll_loss": 0.932290256023407,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.4287196099758148,
"rewards/margins": -0.0128199253231287,
"rewards/rejected": -0.41589969396591187,
"step": 250
},
{
"epoch": 2.407703703703704,
"grad_norm": 37.83774185180664,
"learning_rate": 5.343475104027743e-06,
"log_odds_chosen": 0.17812205851078033,
"log_odds_ratio": -0.7299228310585022,
"logits/chosen": 155.40672302246094,
"logits/rejected": 160.87484741210938,
"logps/chosen": -0.8623374104499817,
"logps/rejected": -0.9512872695922852,
"loss": 42.1891,
"nll_loss": 0.9383618235588074,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.43116870522499084,
"rewards/margins": 0.044474903494119644,
"rewards/rejected": -0.4756436347961426,
"step": 255
},
{
"epoch": 2.455111111111111,
"grad_norm": 34.86994934082031,
"learning_rate": 4.516778136213037e-06,
"log_odds_chosen": 0.06448222696781158,
"log_odds_ratio": -0.7623555660247803,
"logits/chosen": 159.42433166503906,
"logits/rejected": 155.6562957763672,
"logps/chosen": -0.853624701499939,
"logps/rejected": -0.8752864003181458,
"loss": 41.3101,
"nll_loss": 0.8972175717353821,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.4268123507499695,
"rewards/margins": 0.010830795392394066,
"rewards/rejected": -0.4376432001590729,
"step": 260
},
{
"epoch": 2.5025185185185186,
"grad_norm": 40.81254577636719,
"learning_rate": 3.7531701693965554e-06,
"log_odds_chosen": 0.0020365030504763126,
"log_odds_ratio": -0.7836844325065613,
"logits/chosen": 152.5310516357422,
"logits/rejected": 162.28890991210938,
"logps/chosen": -0.904189944267273,
"logps/rejected": -0.8998059034347534,
"loss": 41.5474,
"nll_loss": 0.9223392605781555,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.4520949721336365,
"rewards/margins": -0.002192039042711258,
"rewards/rejected": -0.4499029517173767,
"step": 265
},
{
"epoch": 2.549925925925926,
"grad_norm": 35.19020080566406,
"learning_rate": 3.055003141378948e-06,
"log_odds_chosen": 0.08025786280632019,
"log_odds_ratio": -0.7740025520324707,
"logits/chosen": 152.10177612304688,
"logits/rejected": 160.26998901367188,
"logps/chosen": -0.8461346626281738,
"logps/rejected": -0.8950198888778687,
"loss": 41.2177,
"nll_loss": 0.8989561200141907,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.4230673313140869,
"rewards/margins": 0.02444261871278286,
"rewards/rejected": -0.4475099444389343,
"step": 270
},
{
"epoch": 2.5973333333333333,
"grad_norm": 39.02735900878906,
"learning_rate": 2.424427429704365e-06,
"log_odds_chosen": 0.061826206743717194,
"log_odds_ratio": -0.7568526268005371,
"logits/chosen": 161.21261596679688,
"logits/rejected": 161.19491577148438,
"logps/chosen": -0.8993644714355469,
"logps/rejected": -0.903190016746521,
"loss": 41.1732,
"nll_loss": 0.8910198211669922,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.44968223571777344,
"rewards/margins": 0.0019127646228298545,
"rewards/rejected": -0.4515950083732605,
"step": 275
},
{
"epoch": 2.644740740740741,
"grad_norm": 39.4185676574707,
"learning_rate": 1.8633852284264508e-06,
"log_odds_chosen": 0.02886720933020115,
"log_odds_ratio": -0.8128382563591003,
"logits/chosen": 153.80528259277344,
"logits/rejected": 150.25201416015625,
"logps/chosen": -0.8430732488632202,
"logps/rejected": -0.8419806361198425,
"loss": 41.047,
"nll_loss": 0.9150816202163696,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.4215366244316101,
"rewards/margins": -0.0005463186535052955,
"rewards/rejected": -0.42099031805992126,
"step": 280
},
{
"epoch": 2.6921481481481484,
"grad_norm": 50.44974899291992,
"learning_rate": 1.3736045660864034e-06,
"log_odds_chosen": -0.17111532390117645,
"log_odds_ratio": -0.9221655130386353,
"logits/chosen": 152.35177612304688,
"logits/rejected": 156.8704376220703,
"logps/chosen": -1.0376865863800049,
"logps/rejected": -0.9074798822402954,
"loss": 42.497,
"nll_loss": 0.9657168388366699,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.5188432931900024,
"rewards/margins": -0.06510341912508011,
"rewards/rejected": -0.4537399411201477,
"step": 285
},
{
"epoch": 2.7395555555555555,
"grad_norm": 38.56003952026367,
"learning_rate": 9.565939833279192e-07,
"log_odds_chosen": -0.0652594044804573,
"log_odds_ratio": -0.8579480051994324,
"logits/chosen": 155.33377075195312,
"logits/rejected": 156.094482421875,
"logps/chosen": -0.9537650942802429,
"logps/rejected": -0.8857321739196777,
"loss": 41.9732,
"nll_loss": 0.9463297724723816,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.47688254714012146,
"rewards/margins": -0.034016452729701996,
"rewards/rejected": -0.44286608695983887,
"step": 290
},
{
"epoch": 2.786962962962963,
"grad_norm": 35.984092712402344,
"learning_rate": 6.136378865420872e-07,
"log_odds_chosen": -0.05488128587603569,
"log_odds_ratio": -0.8374530076980591,
"logits/chosen": 159.79251098632812,
"logits/rejected": 153.42626953125,
"logps/chosen": -0.8930980563163757,
"logps/rejected": -0.848624587059021,
"loss": 42.7359,
"nll_loss": 0.9567394256591797,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.44654902815818787,
"rewards/margins": -0.02223675511777401,
"rewards/rejected": -0.4243122935295105,
"step": 295
},
{
"epoch": 2.83437037037037,
"grad_norm": 40.65980911254883,
"learning_rate": 3.45792591853214e-07,
"log_odds_chosen": 0.03835665062069893,
"log_odds_ratio": -0.7734814286231995,
"logits/chosen": 157.85092163085938,
"logits/rejected": 154.9134979248047,
"logps/chosen": -0.8622520565986633,
"logps/rejected": -0.8551470637321472,
"loss": 41.1462,
"nll_loss": 0.9184630513191223,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.43112602829933167,
"rewards/margins": -0.0035525336861610413,
"rewards/rejected": -0.4275735318660736,
"step": 300
},
{
"epoch": 2.8817777777777778,
"grad_norm": 45.03302001953125,
"learning_rate": 1.538830716302092e-07,
"log_odds_chosen": 0.0824732631444931,
"log_odds_ratio": -0.7868438959121704,
"logits/chosen": 152.17709350585938,
"logits/rejected": 160.5044403076172,
"logps/chosen": -0.8794566988945007,
"logps/rejected": -0.887550950050354,
"loss": 40.8093,
"nll_loss": 0.9352946281433105,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.43972834944725037,
"rewards/margins": 0.004047115799039602,
"rewards/rejected": -0.443775475025177,
"step": 305
},
{
"epoch": 2.9291851851851853,
"grad_norm": 41.13324737548828,
"learning_rate": 3.8500413544415025e-08,
"log_odds_chosen": 0.036232512444257736,
"log_odds_ratio": -0.7697147727012634,
"logits/chosen": 153.0935821533203,
"logits/rejected": 152.86441040039062,
"logps/chosen": -0.8729937672615051,
"logps/rejected": -0.8775345087051392,
"loss": 41.2786,
"nll_loss": 0.9117900133132935,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.43649688363075256,
"rewards/margins": 0.002270397264510393,
"rewards/rejected": -0.4387672543525696,
"step": 310
},
{
"epoch": 2.9765925925925925,
"grad_norm": 116.33460235595703,
"learning_rate": 0.0,
"log_odds_chosen": 0.13965412974357605,
"log_odds_ratio": -0.712945282459259,
"logits/chosen": 145.65011596679688,
"logits/rejected": 152.03152465820312,
"logps/chosen": -0.739302933216095,
"logps/rejected": -0.7891243696212769,
"loss": 39.5762,
"nll_loss": 0.8406442403793335,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.3696514666080475,
"rewards/margins": 0.024910712614655495,
"rewards/rejected": -0.3945621848106384,
"step": 315
},
{
"epoch": 2.9765925925925925,
"step": 315,
"total_flos": 0.0,
"train_loss": 71.97097947862413,
"train_runtime": 8116.2448,
"train_samples_per_second": 2.495,
"train_steps_per_second": 0.039
}
],
"logging_steps": 5,
"max_steps": 315,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 100000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}