gemma-7b-orpo-noisy-5e-5-005-v141 / trainer_state.json
silviasapora's picture
Model save
d88860a verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.9881235154394297,
"eval_steps": 500,
"global_step": 312,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.06334125098970704,
"grad_norm": 440.0,
"learning_rate": 7.8125e-06,
"log_odds_chosen": 3.1748764514923096,
"log_odds_ratio": -9.884220123291016,
"logps/chosen": -23.121124267578125,
"logps/rejected": -26.29541015625,
"loss": 178.8174,
"nll_loss": 10.681877136230469,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -1.1560561656951904,
"rewards/margins": 0.1587144136428833,
"rewards/rejected": -1.3147705793380737,
"step": 5
},
{
"epoch": 0.12668250197941408,
"grad_norm": 340.0,
"learning_rate": 1.5625e-05,
"log_odds_chosen": 6.453858852386475,
"log_odds_ratio": -9.807241439819336,
"logps/chosen": -22.413124084472656,
"logps/rejected": -28.864864349365234,
"loss": 178.3701,
"nll_loss": 10.657767295837402,
"rewards/accuracies": 0.578125,
"rewards/chosen": -1.1206562519073486,
"rewards/margins": 0.32258713245391846,
"rewards/rejected": -1.443243384361267,
"step": 10
},
{
"epoch": 0.19002375296912113,
"grad_norm": 276.0,
"learning_rate": 2.34375e-05,
"log_odds_chosen": 6.200798988342285,
"log_odds_ratio": -6.756936073303223,
"logps/chosen": -19.177919387817383,
"logps/rejected": -25.377355575561523,
"loss": 148.0439,
"nll_loss": 8.914896011352539,
"rewards/accuracies": 0.559374988079071,
"rewards/chosen": -0.9588959813117981,
"rewards/margins": 0.3099718987941742,
"rewards/rejected": -1.26886785030365,
"step": 15
},
{
"epoch": 0.25336500395882816,
"grad_norm": 900.0,
"learning_rate": 3.125e-05,
"log_odds_chosen": 2.6450843811035156,
"log_odds_ratio": -5.436456203460693,
"logps/chosen": -13.422134399414062,
"logps/rejected": -16.063411712646484,
"loss": 101.9164,
"nll_loss": 6.097954750061035,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.6711066961288452,
"rewards/margins": 0.1320638358592987,
"rewards/rejected": -0.8031705617904663,
"step": 20
},
{
"epoch": 0.3167062549485352,
"grad_norm": 93.5,
"learning_rate": 3.90625e-05,
"log_odds_chosen": 0.6309150457382202,
"log_odds_ratio": -1.286942481994629,
"logps/chosen": -3.097271680831909,
"logps/rejected": -3.7033658027648926,
"loss": 42.6767,
"nll_loss": 2.6029443740844727,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.15486359596252441,
"rewards/margins": 0.03030467964708805,
"rewards/rejected": -0.18516826629638672,
"step": 25
},
{
"epoch": 0.38004750593824227,
"grad_norm": 152.0,
"learning_rate": 4.6875e-05,
"log_odds_chosen": 0.14747688174247742,
"log_odds_ratio": -0.863192081451416,
"logps/chosen": -1.753333330154419,
"logps/rejected": -1.8908967971801758,
"loss": 33.7883,
"nll_loss": 2.068608283996582,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.08766667544841766,
"rewards/margins": 0.006878170184791088,
"rewards/rejected": -0.09454484283924103,
"step": 30
},
{
"epoch": 0.4433887569279493,
"grad_norm": 320.0,
"learning_rate": 4.998583895833834e-05,
"log_odds_chosen": 0.0781320109963417,
"log_odds_ratio": -0.8847883939743042,
"logps/chosen": -1.601119041442871,
"logps/rejected": -1.677050232887268,
"loss": 30.4438,
"nll_loss": 1.8584957122802734,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.08005593717098236,
"rewards/margins": 0.003796567674726248,
"rewards/rejected": -0.08385250717401505,
"step": 35
},
{
"epoch": 0.5067300079176563,
"grad_norm": 79.5,
"learning_rate": 4.989935734988098e-05,
"log_odds_chosen": 0.17777779698371887,
"log_odds_ratio": -0.7818454504013062,
"logps/chosen": -1.434004545211792,
"logps/rejected": -1.5908782482147217,
"loss": 29.0365,
"nll_loss": 1.7756904363632202,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.07170023769140244,
"rewards/margins": 0.00784367136657238,
"rewards/rejected": -0.07954390347003937,
"step": 40
},
{
"epoch": 0.5700712589073634,
"grad_norm": 46.0,
"learning_rate": 4.973453315132592e-05,
"log_odds_chosen": 0.12557587027549744,
"log_odds_ratio": -0.7924664616584778,
"logps/chosen": -1.366229772567749,
"logps/rejected": -1.46891450881958,
"loss": 27.2712,
"nll_loss": 1.6648271083831787,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.06831149756908417,
"rewards/margins": 0.00513423141092062,
"rewards/rejected": -0.07344572246074677,
"step": 45
},
{
"epoch": 0.6334125098970704,
"grad_norm": 322.0,
"learning_rate": 4.9491884960580894e-05,
"log_odds_chosen": 0.2877430319786072,
"log_odds_ratio": -0.712502121925354,
"logps/chosen": -1.2252671718597412,
"logps/rejected": -1.4585528373718262,
"loss": 25.6522,
"nll_loss": 1.5676389932632446,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.06126336008310318,
"rewards/margins": 0.011664283461868763,
"rewards/rejected": -0.07292764633893967,
"step": 50
},
{
"epoch": 0.6967537608867775,
"grad_norm": 74.0,
"learning_rate": 4.9172176238625164e-05,
"log_odds_chosen": 0.12641319632530212,
"log_odds_ratio": -0.7623997926712036,
"logps/chosen": -1.275046467781067,
"logps/rejected": -1.3824188709259033,
"loss": 25.7811,
"nll_loss": 1.5732007026672363,
"rewards/accuracies": 0.5218750238418579,
"rewards/chosen": -0.06375232338905334,
"rewards/margins": 0.005368619225919247,
"rewards/rejected": -0.06912095099687576,
"step": 55
},
{
"epoch": 0.7600950118764845,
"grad_norm": 49.25,
"learning_rate": 4.877641290737884e-05,
"log_odds_chosen": 0.19897113740444183,
"log_odds_ratio": -0.7030541300773621,
"logps/chosen": -1.1849174499511719,
"logps/rejected": -1.3446745872497559,
"loss": 25.5335,
"nll_loss": 1.5606908798217773,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.05924587324261665,
"rewards/margins": 0.007987860590219498,
"rewards/rejected": -0.06723372638225555,
"step": 60
},
{
"epoch": 0.8234362628661916,
"grad_norm": 73.0,
"learning_rate": 4.8305840184700356e-05,
"log_odds_chosen": 0.2491961419582367,
"log_odds_ratio": -0.693637490272522,
"logps/chosen": -1.1569511890411377,
"logps/rejected": -1.3474953174591064,
"loss": 25.4002,
"nll_loss": 1.5528333187103271,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.057847559452056885,
"rewards/margins": 0.009527203626930714,
"rewards/rejected": -0.06737476587295532,
"step": 65
},
{
"epoch": 0.8867775138558986,
"grad_norm": 62.75,
"learning_rate": 4.7761938666470403e-05,
"log_odds_chosen": 0.1128716841340065,
"log_odds_ratio": -0.76786869764328,
"logps/chosen": -1.2037312984466553,
"logps/rejected": -1.294019103050232,
"loss": 25.6293,
"nll_loss": 1.5634351968765259,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.060186564922332764,
"rewards/margins": 0.004514397121965885,
"rewards/rejected": -0.06470096111297607,
"step": 70
},
{
"epoch": 0.9501187648456056,
"grad_norm": 43.25,
"learning_rate": 4.71464196680895e-05,
"log_odds_chosen": 0.15465858578681946,
"log_odds_ratio": -0.7334991693496704,
"logps/chosen": -1.1338040828704834,
"logps/rejected": -1.2355806827545166,
"loss": 23.9048,
"nll_loss": 1.457373023033142,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.05669020861387253,
"rewards/margins": 0.005088826175779104,
"rewards/rejected": -0.06177903339266777,
"step": 75
},
{
"epoch": 1.0253365003958828,
"grad_norm": 39.25,
"learning_rate": 4.6461219840046654e-05,
"log_odds_chosen": 0.339913547039032,
"log_odds_ratio": -0.6567232608795166,
"logps/chosen": -1.0351102352142334,
"logps/rejected": -1.276901364326477,
"loss": 26.0349,
"nll_loss": 1.4135476350784302,
"rewards/accuracies": 0.6111111044883728,
"rewards/chosen": -0.05175551399588585,
"rewards/margins": 0.012089554220438004,
"rewards/rejected": -0.06384507566690445,
"step": 80
},
{
"epoch": 1.0886777513855899,
"grad_norm": 39.5,
"learning_rate": 4.5708495074500414e-05,
"log_odds_chosen": 0.5737000703811646,
"log_odds_ratio": -0.5414241552352905,
"logps/chosen": -0.8744705319404602,
"logps/rejected": -1.2483097314834595,
"loss": 19.751,
"nll_loss": 1.2073694467544556,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.04372352734208107,
"rewards/margins": 0.018691958859562874,
"rewards/rejected": -0.062415480613708496,
"step": 85
},
{
"epoch": 1.152019002375297,
"grad_norm": 41.25,
"learning_rate": 4.489061372204453e-05,
"log_odds_chosen": 0.6578646302223206,
"log_odds_ratio": -0.5210095047950745,
"logps/chosen": -0.8167413473129272,
"logps/rejected": -1.2592405080795288,
"loss": 19.3149,
"nll_loss": 1.181132197380066,
"rewards/accuracies": 0.734375,
"rewards/chosen": -0.04083707183599472,
"rewards/margins": 0.02212495729327202,
"rewards/rejected": -0.06296203285455704,
"step": 90
},
{
"epoch": 1.2153602533650039,
"grad_norm": 27.0,
"learning_rate": 4.401014914000078e-05,
"log_odds_chosen": 0.5433515310287476,
"log_odds_ratio": -0.5697063207626343,
"logps/chosen": -0.8254748582839966,
"logps/rejected": -1.1539433002471924,
"loss": 19.3225,
"nll_loss": 1.179168701171875,
"rewards/accuracies": 0.690625011920929,
"rewards/chosen": -0.04127374291419983,
"rewards/margins": 0.016423424705863,
"rewards/rejected": -0.05769716575741768,
"step": 95
},
{
"epoch": 1.278701504354711,
"grad_norm": 79.5,
"learning_rate": 4.306987159568479e-05,
"log_odds_chosen": 0.592140793800354,
"log_odds_ratio": -0.5446491241455078,
"logps/chosen": -0.8890258073806763,
"logps/rejected": -1.2678449153900146,
"loss": 19.6595,
"nll_loss": 1.2014886140823364,
"rewards/accuracies": 0.703125,
"rewards/chosen": -0.044451288878917694,
"rewards/margins": 0.018940959125757217,
"rewards/rejected": -0.06339225172996521,
"step": 100
},
{
"epoch": 1.342042755344418,
"grad_norm": 26.25,
"learning_rate": 4.207273955012018e-05,
"log_odds_chosen": 0.5168221592903137,
"log_odds_ratio": -0.5695356130599976,
"logps/chosen": -0.8439987897872925,
"logps/rejected": -1.158204197883606,
"loss": 18.8408,
"nll_loss": 1.1490731239318848,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.04219993203878403,
"rewards/margins": 0.01571027934551239,
"rewards/rejected": -0.05791021138429642,
"step": 105
},
{
"epoch": 1.405384006334125,
"grad_norm": 32.0,
"learning_rate": 4.10218903496256e-05,
"log_odds_chosen": 0.5098962187767029,
"log_odds_ratio": -0.5828264355659485,
"logps/chosen": -0.8689939379692078,
"logps/rejected": -1.1937323808670044,
"loss": 20.1778,
"nll_loss": 1.231968641281128,
"rewards/accuracies": 0.6781250238418579,
"rewards/chosen": -0.04344969242811203,
"rewards/margins": 0.0162369217723608,
"rewards/rejected": -0.05968661978840828,
"step": 110
},
{
"epoch": 1.468725257323832,
"grad_norm": 34.0,
"learning_rate": 3.992063035456259e-05,
"log_odds_chosen": 0.6211134195327759,
"log_odds_ratio": -0.5309170484542847,
"logps/chosen": -0.7933008074760437,
"logps/rejected": -1.1834070682525635,
"loss": 18.7885,
"nll_loss": 1.14773428440094,
"rewards/accuracies": 0.734375,
"rewards/chosen": -0.039665043354034424,
"rewards/margins": 0.01950531266629696,
"rewards/rejected": -0.059170354157686234,
"step": 115
},
{
"epoch": 1.5320665083135392,
"grad_norm": 28.5,
"learning_rate": 3.8772424536302564e-05,
"log_odds_chosen": 0.6670337915420532,
"log_odds_ratio": -0.5194506049156189,
"logps/chosen": -0.7793987989425659,
"logps/rejected": -1.1679680347442627,
"loss": 18.5035,
"nll_loss": 1.1304987668991089,
"rewards/accuracies": 0.734375,
"rewards/chosen": -0.038969945162534714,
"rewards/margins": 0.01942845992743969,
"rewards/rejected": -0.058398403227329254,
"step": 120
},
{
"epoch": 1.5954077593032463,
"grad_norm": 24.375,
"learning_rate": 3.758088557514501e-05,
"log_odds_chosen": 0.5610660314559937,
"log_odds_ratio": -0.552033543586731,
"logps/chosen": -0.795187771320343,
"logps/rejected": -1.1401742696762085,
"loss": 18.7339,
"nll_loss": 1.1432664394378662,
"rewards/accuracies": 0.684374988079071,
"rewards/chosen": -0.03975939005613327,
"rewards/margins": 0.017249327152967453,
"rewards/rejected": -0.05700872093439102,
"step": 125
},
{
"epoch": 1.6587490102929534,
"grad_norm": 29.625,
"learning_rate": 3.634976249348867e-05,
"log_odds_chosen": 0.4703447222709656,
"log_odds_ratio": -0.5870831608772278,
"logps/chosen": -0.7940428256988525,
"logps/rejected": -1.0778682231903076,
"loss": 18.7796,
"nll_loss": 1.1443703174591064,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.039702143520116806,
"rewards/margins": 0.014191272668540478,
"rewards/rejected": -0.05389341711997986,
"step": 130
},
{
"epoch": 1.7220902612826603,
"grad_norm": 28.875,
"learning_rate": 3.508292886002013e-05,
"log_odds_chosen": 0.5749887228012085,
"log_odds_ratio": -0.5368548631668091,
"logps/chosen": -0.8258606195449829,
"logps/rejected": -1.1885448694229126,
"loss": 18.9446,
"nll_loss": 1.1571968793869019,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.04129302501678467,
"rewards/margins": 0.018134210258722305,
"rewards/rejected": -0.05942724272608757,
"step": 135
},
{
"epoch": 1.7854315122723674,
"grad_norm": 26.75,
"learning_rate": 3.378437060203357e-05,
"log_odds_chosen": 0.5498477220535278,
"log_odds_ratio": -0.5687417984008789,
"logps/chosen": -0.81671142578125,
"logps/rejected": -1.1441940069198608,
"loss": 18.5467,
"nll_loss": 1.130732774734497,
"rewards/accuracies": 0.6781250238418579,
"rewards/chosen": -0.04083556681871414,
"rewards/margins": 0.0163741298019886,
"rewards/rejected": -0.05720970779657364,
"step": 140
},
{
"epoch": 1.8487727632620743,
"grad_norm": 36.5,
"learning_rate": 3.24581734642289e-05,
"log_odds_chosen": 0.7345950603485107,
"log_odds_ratio": -0.5215600728988647,
"logps/chosen": -0.7965916991233826,
"logps/rejected": -1.2689704895019531,
"loss": 18.424,
"nll_loss": 1.125421404838562,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.03982958942651749,
"rewards/margins": 0.02361893653869629,
"rewards/rejected": -0.06344853341579437,
"step": 145
},
{
"epoch": 1.9121140142517814,
"grad_norm": 30.5,
"learning_rate": 3.110851015344735e-05,
"log_odds_chosen": 0.5858422517776489,
"log_odds_ratio": -0.5331650376319885,
"logps/chosen": -0.8209154009819031,
"logps/rejected": -1.1717817783355713,
"loss": 18.0192,
"nll_loss": 1.0995434522628784,
"rewards/accuracies": 0.7281249761581421,
"rewards/chosen": -0.04104577749967575,
"rewards/margins": 0.017543310299515724,
"rewards/rejected": -0.058589089661836624,
"step": 150
},
{
"epoch": 1.9754552652414885,
"grad_norm": 31.5,
"learning_rate": 2.9739627209791965e-05,
"log_odds_chosen": 0.6027151346206665,
"log_odds_ratio": -0.5478641390800476,
"logps/chosen": -0.8092836141586304,
"logps/rejected": -1.1755163669586182,
"loss": 18.8905,
"nll_loss": 1.1532632112503052,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.04046417772769928,
"rewards/margins": 0.0183116365224123,
"rewards/rejected": -0.05877581983804703,
"step": 155
},
{
"epoch": 2.0506730007917655,
"grad_norm": 26.625,
"learning_rate": 2.8355831645441388e-05,
"log_odds_chosen": 0.9976516962051392,
"log_odds_ratio": -0.412594735622406,
"logps/chosen": -0.6453574299812317,
"logps/rejected": -1.2032675743103027,
"loss": 17.6716,
"nll_loss": 0.9611276984214783,
"rewards/accuracies": 0.8333333134651184,
"rewards/chosen": -0.032267872244119644,
"rewards/margins": 0.027895506471395493,
"rewards/rejected": -0.06016337499022484,
"step": 160
},
{
"epoch": 2.1140142517814726,
"grad_norm": 25.125,
"learning_rate": 2.6961477393196126e-05,
"log_odds_chosen": 1.592689871788025,
"log_odds_ratio": -0.27941542863845825,
"logps/chosen": -0.5471819639205933,
"logps/rejected": -1.4586117267608643,
"loss": 13.9962,
"nll_loss": 0.8607938885688782,
"rewards/accuracies": 0.9281250238418579,
"rewards/chosen": -0.027359094470739365,
"rewards/margins": 0.04557149484753609,
"rewards/rejected": -0.07293058931827545,
"step": 165
},
{
"epoch": 2.1773555027711797,
"grad_norm": 23.125,
"learning_rate": 2.556095160739513e-05,
"log_odds_chosen": 1.3663856983184814,
"log_odds_ratio": -0.3143647313117981,
"logps/chosen": -0.5476871728897095,
"logps/rejected": -1.304485559463501,
"loss": 13.0544,
"nll_loss": 0.800182044506073,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.027384359389543533,
"rewards/margins": 0.037839919328689575,
"rewards/rejected": -0.06522427499294281,
"step": 170
},
{
"epoch": 2.240696753760887,
"grad_norm": 26.75,
"learning_rate": 2.415866086030516e-05,
"log_odds_chosen": 1.4820417165756226,
"log_odds_ratio": -0.29762935638427734,
"logps/chosen": -0.5468381643295288,
"logps/rejected": -1.3758976459503174,
"loss": 13.3913,
"nll_loss": 0.8220744132995605,
"rewards/accuracies": 0.921875,
"rewards/chosen": -0.02734190784394741,
"rewards/margins": 0.04145296663045883,
"rewards/rejected": -0.06879488378763199,
"step": 175
},
{
"epoch": 2.304038004750594,
"grad_norm": 26.25,
"learning_rate": 2.2759017277414166e-05,
"log_odds_chosen": 1.4541139602661133,
"log_odds_ratio": -0.30560511350631714,
"logps/chosen": -0.5400351285934448,
"logps/rejected": -1.3224079608917236,
"loss": 13.3401,
"nll_loss": 0.818473219871521,
"rewards/accuracies": 0.8968750238418579,
"rewards/chosen": -0.0270017571747303,
"rewards/margins": 0.03911864012479782,
"rewards/rejected": -0.06612040102481842,
"step": 180
},
{
"epoch": 2.367379255740301,
"grad_norm": 45.0,
"learning_rate": 2.136642465525219e-05,
"log_odds_chosen": 1.3566925525665283,
"log_odds_ratio": -0.312085896730423,
"logps/chosen": -0.5589757561683655,
"logps/rejected": -1.2790095806121826,
"loss": 13.2166,
"nll_loss": 0.8104313015937805,
"rewards/accuracies": 0.8968750238418579,
"rewards/chosen": -0.027948787435889244,
"rewards/margins": 0.03600168973207474,
"rewards/rejected": -0.06395047903060913,
"step": 185
},
{
"epoch": 2.4307205067300077,
"grad_norm": 31.0,
"learning_rate": 1.9985264605418183e-05,
"log_odds_chosen": 1.4590914249420166,
"log_odds_ratio": -0.3126595616340637,
"logps/chosen": -0.5508120059967041,
"logps/rejected": -1.3448957204818726,
"loss": 13.3402,
"nll_loss": 0.8181284666061401,
"rewards/accuracies": 0.890625,
"rewards/chosen": -0.027540603652596474,
"rewards/margins": 0.03970418870449066,
"rewards/rejected": -0.06724478304386139,
"step": 190
},
{
"epoch": 2.494061757719715,
"grad_norm": 25.375,
"learning_rate": 1.8619882768408935e-05,
"log_odds_chosen": 1.5256664752960205,
"log_odds_ratio": -0.29186081886291504,
"logps/chosen": -0.5439263582229614,
"logps/rejected": -1.4009654521942139,
"loss": 13.2504,
"nll_loss": 0.8135560750961304,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.02719631791114807,
"rewards/margins": 0.04285195842385292,
"rewards/rejected": -0.07004828751087189,
"step": 195
},
{
"epoch": 2.557403008709422,
"grad_norm": 24.25,
"learning_rate": 1.7274575140626318e-05,
"log_odds_chosen": 1.4801175594329834,
"log_odds_ratio": -0.3182724118232727,
"logps/chosen": -0.5556176900863647,
"logps/rejected": -1.355858564376831,
"loss": 13.7318,
"nll_loss": 0.8423222303390503,
"rewards/accuracies": 0.8968750238418579,
"rewards/chosen": -0.027780886739492416,
"rewards/margins": 0.04001203924417496,
"rewards/rejected": -0.06779292970895767,
"step": 200
},
{
"epoch": 2.620744259699129,
"grad_norm": 21.25,
"learning_rate": 1.59535745575832e-05,
"log_odds_chosen": 1.538774847984314,
"log_odds_ratio": -0.2739660441875458,
"logps/chosen": -0.5107089281082153,
"logps/rejected": -1.3470090627670288,
"loss": 12.7697,
"nll_loss": 0.7844074368476868,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.025535445660352707,
"rewards/margins": 0.041815001517534256,
"rewards/rejected": -0.06735045462846756,
"step": 205
},
{
"epoch": 2.684085510688836,
"grad_norm": 26.5,
"learning_rate": 1.466103737583699e-05,
"log_odds_chosen": 1.6077247858047485,
"log_odds_ratio": -0.27898281812667847,
"logps/chosen": -0.5174364447593689,
"logps/rejected": -1.3971433639526367,
"loss": 12.9226,
"nll_loss": 0.7937139272689819,
"rewards/accuracies": 0.909375011920929,
"rewards/chosen": -0.025871824473142624,
"rewards/margins": 0.04398534819483757,
"rewards/rejected": -0.0698571726679802,
"step": 210
},
{
"epoch": 2.7474267616785433,
"grad_norm": 37.5,
"learning_rate": 1.340103039555415e-05,
"log_odds_chosen": 1.4663047790527344,
"log_odds_ratio": -0.3185238242149353,
"logps/chosen": -0.5488572716712952,
"logps/rejected": -1.336905837059021,
"loss": 13.052,
"nll_loss": 0.7998241782188416,
"rewards/accuracies": 0.903124988079071,
"rewards/chosen": -0.02744286321103573,
"rewards/margins": 0.03940243273973465,
"rewards/rejected": -0.06684528291225433,
"step": 215
},
{
"epoch": 2.81076801266825,
"grad_norm": 33.25,
"learning_rate": 1.217751806485235e-05,
"log_odds_chosen": 1.548749327659607,
"log_odds_ratio": -0.2925991415977478,
"logps/chosen": -0.5320712327957153,
"logps/rejected": -1.400617241859436,
"loss": 13.2408,
"nll_loss": 0.8129196166992188,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.026603559032082558,
"rewards/margins": 0.043427299708127975,
"rewards/rejected": -0.07003085315227509,
"step": 220
},
{
"epoch": 2.8741092636579575,
"grad_norm": 28.25,
"learning_rate": 1.0994350006179931e-05,
"log_odds_chosen": 1.4858758449554443,
"log_odds_ratio": -0.29021698236465454,
"logps/chosen": -0.5439127683639526,
"logps/rejected": -1.3659403324127197,
"loss": 13.1169,
"nll_loss": 0.8052948117256165,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.027195638045668602,
"rewards/margins": 0.04110138118267059,
"rewards/rejected": -0.06829701364040375,
"step": 225
},
{
"epoch": 2.937450514647664,
"grad_norm": 23.125,
"learning_rate": 9.855248903979506e-06,
"log_odds_chosen": 1.4533650875091553,
"log_odds_ratio": -0.2951867878437042,
"logps/chosen": -0.5295075178146362,
"logps/rejected": -1.2960364818572998,
"loss": 13.1057,
"nll_loss": 0.8043462038040161,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -0.026475373655557632,
"rewards/margins": 0.0383264534175396,
"rewards/rejected": -0.06480182707309723,
"step": 230
},
{
"epoch": 3.0126682501979416,
"grad_norm": 29.375,
"learning_rate": 8.763798791745411e-06,
"log_odds_chosen": 1.5862672328948975,
"log_odds_ratio": -0.2758994698524475,
"logps/chosen": -0.5054166316986084,
"logps/rejected": -1.356021761894226,
"loss": 14.3167,
"nll_loss": 0.7815780639648438,
"rewards/accuracies": 0.9222221970558167,
"rewards/chosen": -0.02527083270251751,
"rewards/margins": 0.042530253529548645,
"rewards/rejected": -0.0678010806441307,
"step": 235
},
{
"epoch": 3.0760095011876483,
"grad_norm": 21.25,
"learning_rate": 7.723433775328384e-06,
"log_odds_chosen": 2.0705292224884033,
"log_odds_ratio": -0.19949007034301758,
"logps/chosen": -0.4086482524871826,
"logps/rejected": -1.4631153345108032,
"loss": 10.5377,
"nll_loss": 0.6486285924911499,
"rewards/accuracies": 0.965624988079071,
"rewards/chosen": -0.02043241262435913,
"rewards/margins": 0.05272335559129715,
"rewards/rejected": -0.07315576076507568,
"step": 240
},
{
"epoch": 3.1393507521773554,
"grad_norm": 30.0,
"learning_rate": 6.737427227968063e-06,
"log_odds_chosen": 2.158939838409424,
"log_odds_ratio": -0.17655470967292786,
"logps/chosen": -0.36677083373069763,
"logps/rejected": -1.4107964038848877,
"loss": 10.2002,
"nll_loss": 0.6286830902099609,
"rewards/accuracies": 0.9781249761581421,
"rewards/chosen": -0.01833854243159294,
"rewards/margins": 0.0522012785077095,
"rewards/rejected": -0.07053982466459274,
"step": 245
},
{
"epoch": 3.2026920031670625,
"grad_norm": 24.375,
"learning_rate": 5.808881491049723e-06,
"log_odds_chosen": 2.1467604637145996,
"log_odds_ratio": -0.21470721065998077,
"logps/chosen": -0.45862817764282227,
"logps/rejected": -1.5745422840118408,
"loss": 10.6424,
"nll_loss": 0.6544127464294434,
"rewards/accuracies": 0.953125,
"rewards/chosen": -0.022931408137083054,
"rewards/margins": 0.05579570680856705,
"rewards/rejected": -0.0787271112203598,
"step": 250
},
{
"epoch": 3.2660332541567696,
"grad_norm": 28.375,
"learning_rate": 4.940718112990553e-06,
"log_odds_chosen": 2.224365711212158,
"log_odds_ratio": -0.18038669228553772,
"logps/chosen": -0.38552433252334595,
"logps/rejected": -1.5169192552566528,
"loss": 10.064,
"nll_loss": 0.6199790835380554,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.019276214763522148,
"rewards/margins": 0.056569743901491165,
"rewards/rejected": -0.07584595680236816,
"step": 255
},
{
"epoch": 3.3293745051464767,
"grad_norm": 22.875,
"learning_rate": 4.135668656967434e-06,
"log_odds_chosen": 2.1139309406280518,
"log_odds_ratio": -0.18317027390003204,
"logps/chosen": -0.42459002137184143,
"logps/rejected": -1.4972463846206665,
"loss": 10.2835,
"nll_loss": 0.6335573792457581,
"rewards/accuracies": 0.965624988079071,
"rewards/chosen": -0.02122950181365013,
"rewards/margins": 0.053632818162441254,
"rewards/rejected": -0.07486231625080109,
"step": 260
},
{
"epoch": 3.392715756136184,
"grad_norm": 26.5,
"learning_rate": 3.3962661064086354e-06,
"log_odds_chosen": 2.135136842727661,
"log_odds_ratio": -0.18204157054424286,
"logps/chosen": -0.40990591049194336,
"logps/rejected": -1.4916514158248901,
"loss": 10.287,
"nll_loss": 0.6338369250297546,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.020495299249887466,
"rewards/margins": 0.05408727377653122,
"rewards/rejected": -0.07458256930112839,
"step": 265
},
{
"epoch": 3.456057007125891,
"grad_norm": 22.625,
"learning_rate": 2.7248368952908053e-06,
"log_odds_chosen": 2.2748658657073975,
"log_odds_ratio": -0.17176181077957153,
"logps/chosen": -0.41875600814819336,
"logps/rejected": -1.62033212184906,
"loss": 10.5217,
"nll_loss": 0.6490176916122437,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.020937800407409668,
"rewards/margins": 0.060078807175159454,
"rewards/rejected": -0.08101661503314972,
"step": 270
},
{
"epoch": 3.519398258115598,
"grad_norm": 21.25,
"learning_rate": 2.1234935883170047e-06,
"log_odds_chosen": 2.311600685119629,
"log_odds_ratio": -0.17423459887504578,
"logps/chosen": -0.3958762288093567,
"logps/rejected": -1.5846991539001465,
"loss": 9.5479,
"nll_loss": 0.5880299806594849,
"rewards/accuracies": 0.96875,
"rewards/chosen": -0.019793810322880745,
"rewards/margins": 0.05944114923477173,
"rewards/rejected": -0.07923495769500732,
"step": 275
},
{
"epoch": 3.5827395091053047,
"grad_norm": 25.875,
"learning_rate": 1.59412823400657e-06,
"log_odds_chosen": 2.3392279148101807,
"log_odds_ratio": -0.17188401520252228,
"logps/chosen": -0.40492624044418335,
"logps/rejected": -1.6255334615707397,
"loss": 10.1858,
"nll_loss": 0.628017783164978,
"rewards/accuracies": 0.9593750238418579,
"rewards/chosen": -0.020246315747499466,
"rewards/margins": 0.061030369251966476,
"rewards/rejected": -0.08127669245004654,
"step": 280
},
{
"epoch": 3.646080760095012,
"grad_norm": 26.625,
"learning_rate": 1.138406411610482e-06,
"log_odds_chosen": 2.3245720863342285,
"log_odds_ratio": -0.16916075348854065,
"logps/chosen": -0.37623462080955505,
"logps/rejected": -1.5502853393554688,
"loss": 9.6503,
"nll_loss": 0.5946874618530273,
"rewards/accuracies": 0.9468749761581421,
"rewards/chosen": -0.018811732530593872,
"rewards/margins": 0.058702535927295685,
"rewards/rejected": -0.07751427590847015,
"step": 285
},
{
"epoch": 3.709422011084719,
"grad_norm": 21.875,
"learning_rate": 7.577619905828282e-07,
"log_odds_chosen": 2.351839065551758,
"log_odds_ratio": -0.16314415633678436,
"logps/chosen": -0.3871734142303467,
"logps/rejected": -1.6283880472183228,
"loss": 9.8769,
"nll_loss": 0.6091512441635132,
"rewards/accuracies": 0.971875011920929,
"rewards/chosen": -0.019358674064278603,
"rewards/margins": 0.06206073611974716,
"rewards/rejected": -0.08141940087080002,
"step": 290
},
{
"epoch": 3.772763262074426,
"grad_norm": 23.0,
"learning_rate": 4.5339261909704e-07,
"log_odds_chosen": 2.405986785888672,
"log_odds_ratio": -0.16327905654907227,
"logps/chosen": -0.4028860628604889,
"logps/rejected": -1.711242914199829,
"loss": 10.0405,
"nll_loss": 0.6193649172782898,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.020144306123256683,
"rewards/margins": 0.06541784107685089,
"rewards/rejected": -0.08556214720010757,
"step": 295
},
{
"epoch": 3.836104513064133,
"grad_norm": 23.375,
"learning_rate": 2.262559558016325e-07,
"log_odds_chosen": 2.3016839027404785,
"log_odds_ratio": -0.16708031296730042,
"logps/chosen": -0.38780677318573,
"logps/rejected": -1.5517771244049072,
"loss": 9.6048,
"nll_loss": 0.5919488072395325,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.01939033903181553,
"rewards/margins": 0.058198511600494385,
"rewards/rejected": -0.07758884876966476,
"step": 300
},
{
"epoch": 3.89944576405384,
"grad_norm": 24.75,
"learning_rate": 7.706665667180091e-08,
"log_odds_chosen": 2.3498542308807373,
"log_odds_ratio": -0.1596686691045761,
"logps/chosen": -0.39143842458724976,
"logps/rejected": -1.6413482427597046,
"loss": 10.6228,
"nll_loss": 0.6559435725212097,
"rewards/accuracies": 0.981249988079071,
"rewards/chosen": -0.019571922719478607,
"rewards/margins": 0.06249549239873886,
"rewards/rejected": -0.08206741511821747,
"step": 305
},
{
"epoch": 3.9627870150435474,
"grad_norm": 23.625,
"learning_rate": 6.294126437336734e-09,
"log_odds_chosen": 2.3062288761138916,
"log_odds_ratio": -0.17084279656410217,
"logps/chosen": -0.3852413594722748,
"logps/rejected": -1.5706732273101807,
"loss": 10.5082,
"nll_loss": 0.6482213139533997,
"rewards/accuracies": 0.96875,
"rewards/chosen": -0.019262069836258888,
"rewards/margins": 0.059271592646837234,
"rewards/rejected": -0.07853366434574127,
"step": 310
},
{
"epoch": 3.9881235154394297,
"step": 312,
"total_flos": 0.0,
"train_loss": 25.76104594499637,
"train_runtime": 3640.7426,
"train_samples_per_second": 5.548,
"train_steps_per_second": 0.086
}
],
"logging_steps": 5,
"max_steps": 312,
"num_input_tokens_seen": 0,
"num_train_epochs": 4,
"save_steps": 100,
"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
}