silviasapora's picture
Model save
184470c verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.9595687331536387,
"eval_steps": 500,
"global_step": 368,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05390835579514825,
"grad_norm": 684.0,
"learning_rate": 6.7567567567567575e-06,
"log_odds_chosen": 5.920718193054199,
"log_odds_ratio": -7.724557399749756,
"logps/chosen": -24.315807342529297,
"logps/rejected": -30.238388061523438,
"loss": 193.2993,
"nll_loss": 11.694974899291992,
"rewards/accuracies": 0.484375,
"rewards/chosen": -1.2157905101776123,
"rewards/margins": 0.2961289882659912,
"rewards/rejected": -1.511919379234314,
"step": 5
},
{
"epoch": 0.1078167115902965,
"grad_norm": 218.0,
"learning_rate": 1.3513513513513515e-05,
"log_odds_chosen": 4.100552082061768,
"log_odds_ratio": -5.447903633117676,
"logps/chosen": -20.245830535888672,
"logps/rejected": -24.348098754882812,
"loss": 161.9053,
"nll_loss": 9.846681594848633,
"rewards/accuracies": 0.4781250059604645,
"rewards/chosen": -1.012291669845581,
"rewards/margins": 0.2051134556531906,
"rewards/rejected": -1.2174049615859985,
"step": 10
},
{
"epoch": 0.16172506738544473,
"grad_norm": 205.0,
"learning_rate": 2.0270270270270273e-05,
"log_odds_chosen": 3.8738200664520264,
"log_odds_ratio": -6.855284690856934,
"logps/chosen": -21.959657669067383,
"logps/rejected": -25.834619522094727,
"loss": 166.4944,
"nll_loss": 10.06313419342041,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -1.0979830026626587,
"rewards/margins": 0.19374798238277435,
"rewards/rejected": -1.2917309999465942,
"step": 15
},
{
"epoch": 0.215633423180593,
"grad_norm": 648.0,
"learning_rate": 2.702702702702703e-05,
"log_odds_chosen": 0.7052766680717468,
"log_odds_ratio": -7.368393898010254,
"logps/chosen": -20.677827835083008,
"logps/rejected": -21.383352279663086,
"loss": 141.3241,
"nll_loss": 8.464337348937988,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -1.0338914394378662,
"rewards/margins": 0.03527634218335152,
"rewards/rejected": -1.0691677331924438,
"step": 20
},
{
"epoch": 0.2695417789757412,
"grad_norm": 182.0,
"learning_rate": 3.3783783783783784e-05,
"log_odds_chosen": -0.0883919969201088,
"log_odds_ratio": -2.0752124786376953,
"logps/chosen": -6.3797478675842285,
"logps/rejected": -6.272956371307373,
"loss": 62.7556,
"nll_loss": 3.8184638023376465,
"rewards/accuracies": 0.559374988079071,
"rewards/chosen": -0.3189873993396759,
"rewards/margins": -0.0053395600989460945,
"rewards/rejected": -0.3136478364467621,
"step": 25
},
{
"epoch": 0.32345013477088946,
"grad_norm": 83.0,
"learning_rate": 4.0540540540540545e-05,
"log_odds_chosen": 0.24213147163391113,
"log_odds_ratio": -0.7704355716705322,
"logps/chosen": -1.9613044261932373,
"logps/rejected": -2.175011157989502,
"loss": 36.6421,
"nll_loss": 2.2516086101531982,
"rewards/accuracies": 0.565625011920929,
"rewards/chosen": -0.09806521981954575,
"rewards/margins": 0.010685333982110023,
"rewards/rejected": -0.10875056684017181,
"step": 30
},
{
"epoch": 0.37735849056603776,
"grad_norm": 78.0,
"learning_rate": 4.72972972972973e-05,
"log_odds_chosen": 0.27958303689956665,
"log_odds_ratio": -0.7305089831352234,
"logps/chosen": -1.6892141103744507,
"logps/rejected": -1.930132269859314,
"loss": 30.9266,
"nll_loss": 1.8963871002197266,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.08446070551872253,
"rewards/margins": 0.01204589568078518,
"rewards/rejected": -0.09650660306215286,
"step": 35
},
{
"epoch": 0.431266846361186,
"grad_norm": 62.5,
"learning_rate": 4.998986632578596e-05,
"log_odds_chosen": 0.2721244692802429,
"log_odds_ratio": -0.7381750345230103,
"logps/chosen": -1.7667934894561768,
"logps/rejected": -1.9824241399765015,
"loss": 31.6068,
"nll_loss": 1.9385162591934204,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.0883396789431572,
"rewards/margins": 0.010781526565551758,
"rewards/rejected": -0.09912119805812836,
"step": 40
},
{
"epoch": 0.48517520215633425,
"grad_norm": 46.75,
"learning_rate": 4.992796806510936e-05,
"log_odds_chosen": 0.1983262300491333,
"log_odds_ratio": -0.7065704464912415,
"logps/chosen": -1.4196051359176636,
"logps/rejected": -1.579620122909546,
"loss": 28.8934,
"nll_loss": 1.7705100774765015,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.0709802582859993,
"rewards/margins": 0.008000737987458706,
"rewards/rejected": -0.07898099720478058,
"step": 45
},
{
"epoch": 0.5390835579514824,
"grad_norm": 33.0,
"learning_rate": 4.980994056736854e-05,
"log_odds_chosen": 0.1537928730249405,
"log_odds_ratio": -0.7110344171524048,
"logps/chosen": -1.3953869342803955,
"logps/rejected": -1.5007522106170654,
"loss": 27.7613,
"nll_loss": 1.699528694152832,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.06976933777332306,
"rewards/margins": 0.005268273409456015,
"rewards/rejected": -0.07503761351108551,
"step": 50
},
{
"epoch": 0.5929919137466307,
"grad_norm": 56.0,
"learning_rate": 4.9636049590020475e-05,
"log_odds_chosen": 0.2435292750597,
"log_odds_ratio": -0.6781546473503113,
"logps/chosen": -1.3146178722381592,
"logps/rejected": -1.4984357357025146,
"loss": 26.7408,
"nll_loss": 1.6373910903930664,
"rewards/accuracies": 0.6031249761581421,
"rewards/chosen": -0.06573089212179184,
"rewards/margins": 0.009190889075398445,
"rewards/rejected": -0.07492177933454514,
"step": 55
},
{
"epoch": 0.6469002695417789,
"grad_norm": 65.0,
"learning_rate": 4.940668667592439e-05,
"log_odds_chosen": 0.13836432993412018,
"log_odds_ratio": -0.7060701847076416,
"logps/chosen": -1.2619632482528687,
"logps/rejected": -1.3755035400390625,
"loss": 25.7309,
"nll_loss": 1.5728803873062134,
"rewards/accuracies": 0.590624988079071,
"rewards/chosen": -0.06309816986322403,
"rewards/margins": 0.005677012261003256,
"rewards/rejected": -0.06877517700195312,
"step": 60
},
{
"epoch": 0.7008086253369272,
"grad_norm": 204.0,
"learning_rate": 4.912236827172148e-05,
"log_odds_chosen": 0.13251568377017975,
"log_odds_ratio": -0.7122939229011536,
"logps/chosen": -1.2452380657196045,
"logps/rejected": -1.3495495319366455,
"loss": 25.7811,
"nll_loss": 1.5757039785385132,
"rewards/accuracies": 0.5218750238418579,
"rewards/chosen": -0.062261901795864105,
"rewards/margins": 0.005215569399297237,
"rewards/rejected": -0.06747747957706451,
"step": 65
},
{
"epoch": 0.7547169811320755,
"grad_norm": 95.0,
"learning_rate": 4.878373456497416e-05,
"log_odds_chosen": 0.16559086740016937,
"log_odds_ratio": -0.6909226179122925,
"logps/chosen": -1.2156684398651123,
"logps/rejected": -1.3327122926712036,
"loss": 25.469,
"nll_loss": 1.5572690963745117,
"rewards/accuracies": 0.546875,
"rewards/chosen": -0.06078341603279114,
"rewards/margins": 0.005852191708981991,
"rewards/rejected": -0.0666356161236763,
"step": 70
},
{
"epoch": 0.8086253369272237,
"grad_norm": 130.0,
"learning_rate": 4.839154804268313e-05,
"log_odds_chosen": 0.24878835678100586,
"log_odds_ratio": -0.6528117656707764,
"logps/chosen": -1.171622633934021,
"logps/rejected": -1.3498855829238892,
"loss": 25.0146,
"nll_loss": 1.530771255493164,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.05858113244175911,
"rewards/margins": 0.00891314446926117,
"rewards/rejected": -0.06749428063631058,
"step": 75
},
{
"epoch": 0.862533692722372,
"grad_norm": 119.5,
"learning_rate": 4.7946691774428144e-05,
"log_odds_chosen": 0.2302316427230835,
"log_odds_ratio": -0.6437969207763672,
"logps/chosen": -1.1122291088104248,
"logps/rejected": -1.2697585821151733,
"loss": 24.4222,
"nll_loss": 1.4941973686218262,
"rewards/accuracies": 0.6156250238418579,
"rewards/chosen": -0.05561145395040512,
"rewards/margins": 0.007876471616327763,
"rewards/rejected": -0.0634879320859909,
"step": 80
},
{
"epoch": 0.9164420485175202,
"grad_norm": 48.0,
"learning_rate": 4.745016742399804e-05,
"log_odds_chosen": 0.18154019117355347,
"log_odds_ratio": -0.6740174889564514,
"logps/chosen": -1.0852258205413818,
"logps/rejected": -1.2075302600860596,
"loss": 23.767,
"nll_loss": 1.451735496520996,
"rewards/accuracies": 0.578125,
"rewards/chosen": -0.05426129698753357,
"rewards/margins": 0.006115221884101629,
"rewards/rejected": -0.06037651747465134,
"step": 85
},
{
"epoch": 0.9703504043126685,
"grad_norm": 121.0,
"learning_rate": 4.690309299398736e-05,
"log_odds_chosen": 0.1481303721666336,
"log_odds_ratio": -0.6858943700790405,
"logps/chosen": -1.0602507591247559,
"logps/rejected": -1.1643625497817993,
"loss": 23.2092,
"nll_loss": 1.416282296180725,
"rewards/accuracies": 0.565625011920929,
"rewards/chosen": -0.05301254242658615,
"rewards/margins": 0.005205585155636072,
"rewards/rejected": -0.05821812152862549,
"step": 90
},
{
"epoch": 1.0215633423180592,
"grad_norm": 65.0,
"learning_rate": 4.630670030843785e-05,
"log_odds_chosen": 0.2527507245540619,
"log_odds_ratio": -0.6484514474868774,
"logps/chosen": -0.9999241232872009,
"logps/rejected": -1.1713911294937134,
"loss": 21.3461,
"nll_loss": 1.3719265460968018,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.04999620094895363,
"rewards/margins": 0.00857335701584816,
"rewards/rejected": -0.05856956169009209,
"step": 95
},
{
"epoch": 1.0754716981132075,
"grad_norm": 39.0,
"learning_rate": 4.566233223919298e-05,
"log_odds_chosen": 0.5452004671096802,
"log_odds_ratio": -0.554630696773529,
"logps/chosen": -0.8976788520812988,
"logps/rejected": -1.2380362749099731,
"loss": 20.9491,
"nll_loss": 1.2815890312194824,
"rewards/accuracies": 0.690625011920929,
"rewards/chosen": -0.04488394409418106,
"rewards/margins": 0.017017865553498268,
"rewards/rejected": -0.06190181523561478,
"step": 100
},
{
"epoch": 1.1293800539083558,
"grad_norm": 36.25,
"learning_rate": 4.4971439682211125e-05,
"log_odds_chosen": 0.5778145790100098,
"log_odds_ratio": -0.5227857828140259,
"logps/chosen": -0.8327204585075378,
"logps/rejected": -1.1775634288787842,
"loss": 19.2746,
"nll_loss": 1.1785202026367188,
"rewards/accuracies": 0.7281249761581421,
"rewards/chosen": -0.04163602367043495,
"rewards/margins": 0.017242146655917168,
"rewards/rejected": -0.05887817218899727,
"step": 105
},
{
"epoch": 1.1832884097035041,
"grad_norm": 51.0,
"learning_rate": 4.4235578290645194e-05,
"log_odds_chosen": 0.564378023147583,
"log_odds_ratio": -0.5335876941680908,
"logps/chosen": -0.8683027029037476,
"logps/rejected": -1.225411057472229,
"loss": 20.1128,
"nll_loss": 1.2303677797317505,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.0434151366353035,
"rewards/margins": 0.017855416983366013,
"rewards/rejected": -0.06127055361866951,
"step": 110
},
{
"epoch": 1.2371967654986522,
"grad_norm": 34.5,
"learning_rate": 4.3456404972045216e-05,
"log_odds_chosen": 0.6416139602661133,
"log_odds_ratio": -0.5017353296279907,
"logps/chosen": -0.8975638151168823,
"logps/rejected": -1.311712384223938,
"loss": 19.926,
"nll_loss": 1.220287799835205,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.044878192245960236,
"rewards/margins": 0.02070743218064308,
"rewards/rejected": -0.06558562815189362,
"step": 115
},
{
"epoch": 1.2911051212938005,
"grad_norm": 29.375,
"learning_rate": 4.263567415757058e-05,
"log_odds_chosen": 0.5536849498748779,
"log_odds_ratio": -0.5178900957107544,
"logps/chosen": -0.8433765172958374,
"logps/rejected": -1.1795700788497925,
"loss": 18.7183,
"nll_loss": 1.1439992189407349,
"rewards/accuracies": 0.7718750238418579,
"rewards/chosen": -0.04216882586479187,
"rewards/margins": 0.016809683293104172,
"rewards/rejected": -0.05897850915789604,
"step": 120
},
{
"epoch": 1.3450134770889488,
"grad_norm": 40.25,
"learning_rate": 4.177523385161264e-05,
"log_odds_chosen": 0.5022929906845093,
"log_odds_ratio": -0.5432526469230652,
"logps/chosen": -0.830506443977356,
"logps/rejected": -1.1401817798614502,
"loss": 19.2524,
"nll_loss": 1.176114559173584,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.041525326669216156,
"rewards/margins": 0.015483764931559563,
"rewards/rejected": -0.05700908973813057,
"step": 125
},
{
"epoch": 1.398921832884097,
"grad_norm": 45.5,
"learning_rate": 4.087702147072241e-05,
"log_odds_chosen": 0.7220330238342285,
"log_odds_ratio": -0.4684603810310364,
"logps/chosen": -0.8251385688781738,
"logps/rejected": -1.28031325340271,
"loss": 18.9627,
"nll_loss": 1.1617438793182373,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.04125693440437317,
"rewards/margins": 0.022758739069104195,
"rewards/rejected": -0.06401567161083221,
"step": 130
},
{
"epoch": 1.4528301886792452,
"grad_norm": 42.75,
"learning_rate": 3.9943059481212795e-05,
"log_odds_chosen": 0.6586893796920776,
"log_odds_ratio": -0.4884856641292572,
"logps/chosen": -0.8284580111503601,
"logps/rejected": -1.2332229614257812,
"loss": 18.9402,
"nll_loss": 1.1593413352966309,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.041422899812459946,
"rewards/margins": 0.02023824118077755,
"rewards/rejected": -0.061661142855882645,
"step": 135
},
{
"epoch": 1.5067385444743935,
"grad_norm": 40.0,
"learning_rate": 3.89754508452579e-05,
"log_odds_chosen": 0.5820509195327759,
"log_odds_ratio": -0.507893443107605,
"logps/chosen": -0.8388631939888,
"logps/rejected": -1.1933777332305908,
"loss": 18.6589,
"nll_loss": 1.1407865285873413,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.04194316640496254,
"rewards/margins": 0.01772572472691536,
"rewards/rejected": -0.0596688874065876,
"step": 140
},
{
"epoch": 1.5606469002695418,
"grad_norm": 29.625,
"learning_rate": 3.797637428574326e-05,
"log_odds_chosen": 0.5326761603355408,
"log_odds_ratio": -0.5247709155082703,
"logps/chosen": -0.8665750622749329,
"logps/rejected": -1.1989167928695679,
"loss": 20.4033,
"nll_loss": 1.2489675283432007,
"rewards/accuracies": 0.765625,
"rewards/chosen": -0.043328750878572464,
"rewards/margins": 0.01661708578467369,
"rewards/rejected": -0.05994584038853645,
"step": 145
},
{
"epoch": 1.61455525606469,
"grad_norm": 46.0,
"learning_rate": 3.694807938052887e-05,
"log_odds_chosen": 0.5733472108840942,
"log_odds_ratio": -0.5206255912780762,
"logps/chosen": -0.8152993321418762,
"logps/rejected": -1.168148159980774,
"loss": 18.796,
"nll_loss": 1.1487157344818115,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.04076496511697769,
"rewards/margins": 0.017642442137002945,
"rewards/rejected": -0.05840740725398064,
"step": 150
},
{
"epoch": 1.6684636118598384,
"grad_norm": 47.5,
"learning_rate": 3.589288149717119e-05,
"log_odds_chosen": 0.5549741387367249,
"log_odds_ratio": -0.5415369868278503,
"logps/chosen": -0.8091446161270142,
"logps/rejected": -1.1404829025268555,
"loss": 18.6901,
"nll_loss": 1.1410565376281738,
"rewards/accuracies": 0.746874988079071,
"rewards/chosen": -0.04045723378658295,
"rewards/margins": 0.016566917300224304,
"rewards/rejected": -0.057024143636226654,
"step": 155
},
{
"epoch": 1.7223719676549867,
"grad_norm": 41.25,
"learning_rate": 3.481315657950931e-05,
"log_odds_chosen": 0.6500066518783569,
"log_odds_ratio": -0.49195536971092224,
"logps/chosen": -0.8281671404838562,
"logps/rejected": -1.2156283855438232,
"loss": 18.8162,
"nll_loss": 1.1514135599136353,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.04140836000442505,
"rewards/margins": 0.01937306486070156,
"rewards/rejected": -0.06078142672777176,
"step": 160
},
{
"epoch": 1.7762803234501348,
"grad_norm": 22.5,
"learning_rate": 3.3711335797853977e-05,
"log_odds_chosen": 0.7445698976516724,
"log_odds_ratio": -0.4790865480899811,
"logps/chosen": -0.8083413243293762,
"logps/rejected": -1.2825214862823486,
"loss": 18.5217,
"nll_loss": 1.1336489915847778,
"rewards/accuracies": 0.809374988079071,
"rewards/chosen": -0.04041706770658493,
"rewards/margins": 0.0237090103328228,
"rewards/rejected": -0.06412607431411743,
"step": 165
},
{
"epoch": 1.830188679245283,
"grad_norm": 39.0,
"learning_rate": 3.2589900074825696e-05,
"log_odds_chosen": 0.6931416988372803,
"log_odds_ratio": -0.4947904944419861,
"logps/chosen": -0.7949181795120239,
"logps/rejected": -1.2068727016448975,
"loss": 18.9459,
"nll_loss": 1.1593761444091797,
"rewards/accuracies": 0.778124988079071,
"rewards/chosen": -0.03974590823054314,
"rewards/margins": 0.02059771679341793,
"rewards/rejected": -0.060343630611896515,
"step": 170
},
{
"epoch": 1.8840970350404311,
"grad_norm": 47.25,
"learning_rate": 3.1451374499167586e-05,
"log_odds_chosen": 0.5809050798416138,
"log_odds_ratio": -0.5110514163970947,
"logps/chosen": -0.8269885182380676,
"logps/rejected": -1.1618260145187378,
"loss": 18.2167,
"nll_loss": 1.112992286682129,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -0.04134942218661308,
"rewards/margins": 0.016741881147027016,
"rewards/rejected": -0.05809130519628525,
"step": 175
},
{
"epoch": 1.9380053908355794,
"grad_norm": 33.75,
"learning_rate": 3.029832264011133e-05,
"log_odds_chosen": 0.5635145902633667,
"log_odds_ratio": -0.5329084396362305,
"logps/chosen": -0.8633018732070923,
"logps/rejected": -1.2152156829833984,
"loss": 19.8266,
"nll_loss": 1.2125170230865479,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.043165091425180435,
"rewards/margins": 0.017595689743757248,
"rewards/rejected": -0.06076078489422798,
"step": 180
},
{
"epoch": 1.9919137466307277,
"grad_norm": 26.375,
"learning_rate": 2.91333407750982e-05,
"log_odds_chosen": 0.6361402869224548,
"log_odds_ratio": -0.5029143691062927,
"logps/chosen": -0.8135613203048706,
"logps/rejected": -1.1941924095153809,
"loss": 18.7586,
"nll_loss": 1.1472654342651367,
"rewards/accuracies": 0.784375011920929,
"rewards/chosen": -0.04067806527018547,
"rewards/margins": 0.019031552597880363,
"rewards/rejected": -0.05970961973071098,
"step": 185
},
{
"epoch": 2.0431266846361185,
"grad_norm": 28.625,
"learning_rate": 2.7959052043852464e-05,
"log_odds_chosen": 1.1101665496826172,
"log_odds_ratio": -0.3634699881076813,
"logps/chosen": -0.6150734424591064,
"logps/rejected": -1.237518548965454,
"loss": 13.763,
"nll_loss": 0.8872840404510498,
"rewards/accuracies": 0.8980262875556946,
"rewards/chosen": -0.030753672122955322,
"rewards/margins": 0.031122254207730293,
"rewards/rejected": -0.06187593191862106,
"step": 190
},
{
"epoch": 2.0970350404312668,
"grad_norm": 26.625,
"learning_rate": 2.677810054197009e-05,
"log_odds_chosen": 1.4561563730239868,
"log_odds_ratio": -0.28991881012916565,
"logps/chosen": -0.5946121215820312,
"logps/rejected": -1.417449712753296,
"loss": 14.7392,
"nll_loss": 0.906706690788269,
"rewards/accuracies": 0.9281250238418579,
"rewards/chosen": -0.029730603098869324,
"rewards/margins": 0.04114187881350517,
"rewards/rejected": -0.0708724781870842,
"step": 195
},
{
"epoch": 2.150943396226415,
"grad_norm": 36.5,
"learning_rate": 2.559314536732212e-05,
"log_odds_chosen": 1.3712774515151978,
"log_odds_ratio": -0.2994323670864105,
"logps/chosen": -0.5892333984375,
"logps/rejected": -1.355689287185669,
"loss": 13.7443,
"nll_loss": 0.8440448045730591,
"rewards/accuracies": 0.934374988079071,
"rewards/chosen": -0.02946167066693306,
"rewards/margins": 0.03832279518246651,
"rewards/rejected": -0.06778446584939957,
"step": 200
},
{
"epoch": 2.2048517520215634,
"grad_norm": 25.75,
"learning_rate": 2.4406854632677883e-05,
"log_odds_chosen": 1.447076439857483,
"log_odds_ratio": -0.29277992248535156,
"logps/chosen": -0.5634891390800476,
"logps/rejected": -1.386910080909729,
"loss": 13.4321,
"nll_loss": 0.8248661756515503,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.02817445993423462,
"rewards/margins": 0.04117104411125183,
"rewards/rejected": -0.06934549659490585,
"step": 205
},
{
"epoch": 2.2587601078167117,
"grad_norm": 27.75,
"learning_rate": 2.3221899458029918e-05,
"log_odds_chosen": 1.4350051879882812,
"log_odds_ratio": -0.2998976409435272,
"logps/chosen": -0.5992333292961121,
"logps/rejected": -1.4122200012207031,
"loss": 14.0534,
"nll_loss": 0.8633407354354858,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.02996166981756687,
"rewards/margins": 0.04064933583140373,
"rewards/rejected": -0.07061101496219635,
"step": 210
},
{
"epoch": 2.31266846361186,
"grad_norm": 28.25,
"learning_rate": 2.204094795614755e-05,
"log_odds_chosen": 1.4530997276306152,
"log_odds_ratio": -0.2895173728466034,
"logps/chosen": -0.5493010878562927,
"logps/rejected": -1.3495293855667114,
"loss": 13.5655,
"nll_loss": 0.8333690762519836,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.027465051040053368,
"rewards/margins": 0.04001142084598541,
"rewards/rejected": -0.06747647374868393,
"step": 215
},
{
"epoch": 2.3665768194070083,
"grad_norm": 26.125,
"learning_rate": 2.0866659224901812e-05,
"log_odds_chosen": 1.4535211324691772,
"log_odds_ratio": -0.2938300371170044,
"logps/chosen": -0.5619036555290222,
"logps/rejected": -1.3884742259979248,
"loss": 13.1624,
"nll_loss": 0.8079586029052734,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.02809518575668335,
"rewards/margins": 0.04132852703332901,
"rewards/rejected": -0.06942371279001236,
"step": 220
},
{
"epoch": 2.420485175202156,
"grad_norm": 44.75,
"learning_rate": 1.970167735988867e-05,
"log_odds_chosen": 1.4392869472503662,
"log_odds_ratio": -0.291705459356308,
"logps/chosen": -0.5864852666854858,
"logps/rejected": -1.387615442276001,
"loss": 13.6079,
"nll_loss": 0.8359111547470093,
"rewards/accuracies": 0.940625011920929,
"rewards/chosen": -0.029324263334274292,
"rewards/margins": 0.0400565080344677,
"rewards/rejected": -0.06938077509403229,
"step": 225
},
{
"epoch": 2.4743935309973044,
"grad_norm": 25.75,
"learning_rate": 1.854862550083241e-05,
"log_odds_chosen": 1.425066590309143,
"log_odds_ratio": -0.29249343276023865,
"logps/chosen": -0.558022141456604,
"logps/rejected": -1.3558303117752075,
"loss": 13.6401,
"nll_loss": 0.83788001537323,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.02790110372006893,
"rewards/margins": 0.03989041596651077,
"rewards/rejected": -0.06779151409864426,
"step": 230
},
{
"epoch": 2.5283018867924527,
"grad_norm": 31.125,
"learning_rate": 1.7410099925174307e-05,
"log_odds_chosen": 1.3924018144607544,
"log_odds_ratio": -0.3068452477455139,
"logps/chosen": -0.5728594064712524,
"logps/rejected": -1.3289473056793213,
"loss": 13.4421,
"nll_loss": 0.8247901797294617,
"rewards/accuracies": 0.921875,
"rewards/chosen": -0.028642971068620682,
"rewards/margins": 0.03780439496040344,
"rewards/rejected": -0.06644736230373383,
"step": 235
},
{
"epoch": 2.582210242587601,
"grad_norm": 25.375,
"learning_rate": 1.628866420214603e-05,
"log_odds_chosen": 1.424349308013916,
"log_odds_ratio": -0.3004930913448334,
"logps/chosen": -0.5925456285476685,
"logps/rejected": -1.407995581626892,
"loss": 14.0136,
"nll_loss": 0.8608237504959106,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.029627284035086632,
"rewards/margins": 0.04077249392867088,
"rewards/rejected": -0.07039977610111237,
"step": 240
},
{
"epoch": 2.6361185983827493,
"grad_norm": 28.375,
"learning_rate": 1.5186843420490698e-05,
"log_odds_chosen": 1.4117343425750732,
"log_odds_ratio": -0.29593607783317566,
"logps/chosen": -0.5560811758041382,
"logps/rejected": -1.315725564956665,
"loss": 14.3972,
"nll_loss": 0.8850256204605103,
"rewards/accuracies": 0.921875,
"rewards/chosen": -0.02780405804514885,
"rewards/margins": 0.03798222541809082,
"rewards/rejected": -0.06578627973794937,
"step": 245
},
{
"epoch": 2.6900269541778976,
"grad_norm": 25.0,
"learning_rate": 1.4107118502828814e-05,
"log_odds_chosen": 1.5662610530853271,
"log_odds_ratio": -0.27495214343070984,
"logps/chosen": -0.5344637036323547,
"logps/rejected": -1.4078654050827026,
"loss": 13.1208,
"nll_loss": 0.8063005208969116,
"rewards/accuracies": 0.9437500238418579,
"rewards/chosen": -0.026723185554146767,
"rewards/margins": 0.043670084327459335,
"rewards/rejected": -0.07039327919483185,
"step": 250
},
{
"epoch": 2.743935309973046,
"grad_norm": 25.625,
"learning_rate": 1.3051920619471145e-05,
"log_odds_chosen": 1.4818260669708252,
"log_odds_ratio": -0.2981964945793152,
"logps/chosen": -0.5319411754608154,
"logps/rejected": -1.3410605192184448,
"loss": 13.8473,
"nll_loss": 0.8505468368530273,
"rewards/accuracies": 0.9156249761581421,
"rewards/chosen": -0.02659705840051174,
"rewards/margins": 0.04045596718788147,
"rewards/rejected": -0.06705302000045776,
"step": 255
},
{
"epoch": 2.797843665768194,
"grad_norm": 20.875,
"learning_rate": 1.2023625714256743e-05,
"log_odds_chosen": 1.3853600025177002,
"log_odds_ratio": -0.3066624402999878,
"logps/chosen": -0.5818651914596558,
"logps/rejected": -1.3556396961212158,
"loss": 13.8096,
"nll_loss": 0.8477646112442017,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.02909325622022152,
"rewards/margins": 0.03868872672319412,
"rewards/rejected": -0.06778198480606079,
"step": 260
},
{
"epoch": 2.8517520215633425,
"grad_norm": 25.875,
"learning_rate": 1.1024549154742098e-05,
"log_odds_chosen": 1.3135344982147217,
"log_odds_ratio": -0.3150535225868225,
"logps/chosen": -0.5827392339706421,
"logps/rejected": -1.295180320739746,
"loss": 13.5359,
"nll_loss": 0.8302389979362488,
"rewards/accuracies": 0.9281250238418579,
"rewards/chosen": -0.029136961326003075,
"rewards/margins": 0.03562205657362938,
"rewards/rejected": -0.0647590309381485,
"step": 265
},
{
"epoch": 2.9056603773584904,
"grad_norm": 26.75,
"learning_rate": 1.005694051878721e-05,
"log_odds_chosen": 1.3994379043579102,
"log_odds_ratio": -0.3157843351364136,
"logps/chosen": -0.5678838491439819,
"logps/rejected": -1.3345766067504883,
"loss": 13.6712,
"nll_loss": 0.8386627435684204,
"rewards/accuracies": 0.90625,
"rewards/chosen": -0.028394188731908798,
"rewards/margins": 0.03833463788032532,
"rewards/rejected": -0.06672882288694382,
"step": 270
},
{
"epoch": 2.9595687331536387,
"grad_norm": 25.5,
"learning_rate": 9.1229785292776e-06,
"log_odds_chosen": 1.4966779947280884,
"log_odds_ratio": -0.28297117352485657,
"logps/chosen": -0.5280012488365173,
"logps/rejected": -1.359018325805664,
"loss": 13.0722,
"nll_loss": 0.8028658628463745,
"rewards/accuracies": 0.9312499761581421,
"rewards/chosen": -0.026400065049529076,
"rewards/margins": 0.04155084863305092,
"rewards/rejected": -0.06795090436935425,
"step": 275
},
{
"epoch": 3.01078167115903,
"grad_norm": 24.5,
"learning_rate": 8.22476614838737e-06,
"log_odds_chosen": 1.5248451232910156,
"log_odds_ratio": -0.2791544198989868,
"logps/chosen": -0.540464460849762,
"logps/rejected": -1.3801342248916626,
"loss": 12.3085,
"nll_loss": 0.7958097457885742,
"rewards/accuracies": 0.9407894611358643,
"rewards/chosen": -0.027023224160075188,
"rewards/margins": 0.04198349267244339,
"rewards/rejected": -0.06900670379400253,
"step": 280
},
{
"epoch": 3.0646900269541777,
"grad_norm": 20.75,
"learning_rate": 7.364325842429417e-06,
"log_odds_chosen": 2.0129177570343018,
"log_odds_ratio": -0.1976321041584015,
"logps/chosen": -0.43267732858657837,
"logps/rejected": -1.4722133874893188,
"loss": 10.7779,
"nll_loss": 0.6637340784072876,
"rewards/accuracies": 0.981249988079071,
"rewards/chosen": -0.02163386344909668,
"rewards/margins": 0.05197680741548538,
"rewards/rejected": -0.07361067086458206,
"step": 285
},
{
"epoch": 3.118598382749326,
"grad_norm": 33.5,
"learning_rate": 6.543595027954785e-06,
"log_odds_chosen": 2.077929735183716,
"log_odds_ratio": -0.2039026916027069,
"logps/chosen": -0.4455755650997162,
"logps/rejected": -1.5429691076278687,
"loss": 10.7447,
"nll_loss": 0.6613457798957825,
"rewards/accuracies": 0.96875,
"rewards/chosen": -0.02227878011763096,
"rewards/margins": 0.05486968904733658,
"rewards/rejected": -0.07714846730232239,
"step": 290
},
{
"epoch": 3.1725067385444743,
"grad_norm": 26.375,
"learning_rate": 5.764421709354811e-06,
"log_odds_chosen": 2.2563636302948,
"log_odds_ratio": -0.17267544567584991,
"logps/chosen": -0.4047982692718506,
"logps/rejected": -1.6122785806655884,
"loss": 10.4634,
"nll_loss": 0.6453262567520142,
"rewards/accuracies": 0.96875,
"rewards/chosen": -0.02023991569876671,
"rewards/margins": 0.06037401407957077,
"rewards/rejected": -0.08061392605304718,
"step": 295
},
{
"epoch": 3.2264150943396226,
"grad_norm": 22.25,
"learning_rate": 5.02856031778888e-06,
"log_odds_chosen": 2.167142152786255,
"log_odds_ratio": -0.19436317682266235,
"logps/chosen": -0.43167462944984436,
"logps/rejected": -1.5412400960922241,
"loss": 10.7259,
"nll_loss": 0.6606518030166626,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.021583732217550278,
"rewards/margins": 0.055478282272815704,
"rewards/rejected": -0.07706201076507568,
"step": 300
},
{
"epoch": 3.280323450134771,
"grad_norm": 22.75,
"learning_rate": 4.337667760807018e-06,
"log_odds_chosen": 2.055614471435547,
"log_odds_ratio": -0.20666727423667908,
"logps/chosen": -0.4421593248844147,
"logps/rejected": -1.5431147813796997,
"loss": 10.7846,
"nll_loss": 0.6637014746665955,
"rewards/accuracies": 0.971875011920929,
"rewards/chosen": -0.022107966244220734,
"rewards/margins": 0.05504777282476425,
"rewards/rejected": -0.07715573906898499,
"step": 305
},
{
"epoch": 3.334231805929919,
"grad_norm": 27.875,
"learning_rate": 3.69329969156216e-06,
"log_odds_chosen": 2.183558940887451,
"log_odds_ratio": -0.17826569080352783,
"logps/chosen": -0.43023914098739624,
"logps/rejected": -1.6062309741973877,
"loss": 10.7638,
"nll_loss": 0.663822591304779,
"rewards/accuracies": 0.971875011920929,
"rewards/chosen": -0.021511957049369812,
"rewards/margins": 0.058799587190151215,
"rewards/rejected": -0.08031155169010162,
"step": 310
},
{
"epoch": 3.3881401617250675,
"grad_norm": 27.875,
"learning_rate": 3.096907006012639e-06,
"log_odds_chosen": 2.1112418174743652,
"log_odds_ratio": -0.19758489727973938,
"logps/chosen": -0.4420015215873718,
"logps/rejected": -1.5825992822647095,
"loss": 10.8219,
"nll_loss": 0.6664899587631226,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.02210007607936859,
"rewards/margins": 0.05702989175915718,
"rewards/rejected": -0.07912997156381607,
"step": 315
},
{
"epoch": 3.442048517520216,
"grad_norm": 23.875,
"learning_rate": 2.549832576001965e-06,
"log_odds_chosen": 2.2864553928375244,
"log_odds_ratio": -0.1727539300918579,
"logps/chosen": -0.42124342918395996,
"logps/rejected": -1.643252968788147,
"loss": 10.4624,
"nll_loss": 0.6452642679214478,
"rewards/accuracies": 0.9906250238418579,
"rewards/chosen": -0.021062172949314117,
"rewards/margins": 0.061100464314222336,
"rewards/rejected": -0.08216263353824615,
"step": 320
},
{
"epoch": 3.4959568733153636,
"grad_norm": 24.75,
"learning_rate": 2.0533082255718594e-06,
"log_odds_chosen": 2.240377187728882,
"log_odds_ratio": -0.18875715136528015,
"logps/chosen": -0.39375439286231995,
"logps/rejected": -1.5621250867843628,
"loss": 9.993,
"nll_loss": 0.6151254177093506,
"rewards/accuracies": 0.96875,
"rewards/chosen": -0.019687717780470848,
"rewards/margins": 0.058418530970811844,
"rewards/rejected": -0.07810626178979874,
"step": 325
},
{
"epoch": 3.5498652291105124,
"grad_norm": 28.5,
"learning_rate": 1.6084519573168727e-06,
"log_odds_chosen": 2.1224582195281982,
"log_odds_ratio": -0.2051163613796234,
"logps/chosen": -0.4335021376609802,
"logps/rejected": -1.5549887418746948,
"loss": 10.6481,
"nll_loss": 0.6552503705024719,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.0216751080006361,
"rewards/margins": 0.05607432872056961,
"rewards/rejected": -0.07774944603443146,
"step": 330
},
{
"epoch": 3.6037735849056602,
"grad_norm": 21.625,
"learning_rate": 1.2162654350258402e-06,
"log_odds_chosen": 2.2741246223449707,
"log_odds_ratio": -0.18007881939411163,
"logps/chosen": -0.438800573348999,
"logps/rejected": -1.675838828086853,
"loss": 10.4112,
"nll_loss": 0.6416950821876526,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.02194002829492092,
"rewards/margins": 0.06185191869735718,
"rewards/rejected": -0.08379194885492325,
"step": 335
},
{
"epoch": 3.6576819407008085,
"grad_norm": 27.5,
"learning_rate": 8.776317282785163e-07,
"log_odds_chosen": 2.210236072540283,
"log_odds_ratio": -0.1864241361618042,
"logps/chosen": -0.4287256598472595,
"logps/rejected": -1.5987118482589722,
"loss": 10.3796,
"nll_loss": 0.6394068598747253,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.021436283364892006,
"rewards/margins": 0.05849931389093399,
"rewards/rejected": -0.07993558794260025,
"step": 340
},
{
"epoch": 3.711590296495957,
"grad_norm": 25.875,
"learning_rate": 5.933133240756117e-07,
"log_odds_chosen": 2.208949327468872,
"log_odds_ratio": -0.18415012955665588,
"logps/chosen": -0.44286638498306274,
"logps/rejected": -1.6374422311782837,
"loss": 10.6099,
"nll_loss": 0.6539115905761719,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.022143317386507988,
"rewards/margins": 0.05972879007458687,
"rewards/rejected": -0.0818721130490303,
"step": 345
},
{
"epoch": 3.765498652291105,
"grad_norm": 21.375,
"learning_rate": 3.6395040997952714e-07,
"log_odds_chosen": 2.108987331390381,
"log_odds_ratio": -0.19528530538082123,
"logps/chosen": -0.42990946769714355,
"logps/rejected": -1.5616236925125122,
"loss": 10.8233,
"nll_loss": 0.6666921377182007,
"rewards/accuracies": 0.956250011920929,
"rewards/chosen": -0.021495472639799118,
"rewards/margins": 0.05658571049571037,
"rewards/rejected": -0.0780811756849289,
"step": 350
},
{
"epoch": 3.8194070080862534,
"grad_norm": 24.25,
"learning_rate": 1.9005943263146198e-07,
"log_odds_chosen": 2.360919237136841,
"log_odds_ratio": -0.1664571613073349,
"logps/chosen": -0.3908630311489105,
"logps/rejected": -1.650352120399475,
"loss": 10.2428,
"nll_loss": 0.6318550109863281,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.019543152302503586,
"rewards/margins": 0.0629744604229927,
"rewards/rejected": -0.082517609000206,
"step": 355
},
{
"epoch": 3.8733153638814017,
"grad_norm": 29.25,
"learning_rate": 7.203193489064009e-08,
"log_odds_chosen": 2.1713168621063232,
"log_odds_ratio": -0.1880660057067871,
"logps/chosen": -0.41322654485702515,
"logps/rejected": -1.57550048828125,
"loss": 10.3482,
"nll_loss": 0.6373571157455444,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.020661327987909317,
"rewards/margins": 0.058113694190979004,
"rewards/rejected": -0.07877503335475922,
"step": 360
},
{
"epoch": 3.92722371967655,
"grad_norm": 24.75,
"learning_rate": 1.0133674214043987e-08,
"log_odds_chosen": 2.3266632556915283,
"log_odds_ratio": -0.16368618607521057,
"logps/chosen": -0.3924325108528137,
"logps/rejected": -1.6150646209716797,
"loss": 9.8053,
"nll_loss": 0.6046454310417175,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.019621625542640686,
"rewards/margins": 0.06113160774111748,
"rewards/rejected": -0.08075322955846786,
"step": 365
},
{
"epoch": 3.9595687331536387,
"step": 368,
"total_flos": 0.0,
"train_loss": 25.71915608903636,
"train_runtime": 4429.3261,
"train_samples_per_second": 5.361,
"train_steps_per_second": 0.083
}
],
"logging_steps": 5,
"max_steps": 368,
"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
}