alvarobartt's picture
End of training
8d0cf53 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 3.0,
"eval_steps": 500,
"global_step": 633,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.05,
"grad_norm": 45.76888063592087,
"learning_rate": 5.000000000000001e-07,
"log_odds_chosen": 0.2575085163116455,
"log_odds_ratio": -0.704267144203186,
"logits/chosen": -2.8495709896087646,
"logits/rejected": -2.8519797325134277,
"logps/chosen": -1.2749807834625244,
"logps/rejected": -1.5134317874908447,
"loss": 1.2045,
"nll_loss": 1.1489894390106201,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.127498060464859,
"rewards/margins": 0.023845110088586807,
"rewards/rejected": -0.1513431817293167,
"step": 10
},
{
"epoch": 0.09,
"grad_norm": 17.550572571834728,
"learning_rate": 1.0000000000000002e-06,
"log_odds_chosen": 0.18295660614967346,
"log_odds_ratio": -0.7083484530448914,
"logits/chosen": -2.9011380672454834,
"logits/rejected": -2.903749465942383,
"logps/chosen": -0.9543957710266113,
"logps/rejected": -1.0926854610443115,
"loss": 1.0092,
"nll_loss": 0.9102001190185547,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.09543957561254501,
"rewards/margins": 0.013828963041305542,
"rewards/rejected": -0.10926854610443115,
"step": 20
},
{
"epoch": 0.14,
"grad_norm": 10.772595379798407,
"learning_rate": 1.5e-06,
"log_odds_chosen": 0.22791412472724915,
"log_odds_ratio": -0.6785577535629272,
"logits/chosen": -2.970428943634033,
"logits/rejected": -2.9696388244628906,
"logps/chosen": -0.9358822107315063,
"logps/rejected": -1.0665825605392456,
"loss": 0.968,
"nll_loss": 0.8899785280227661,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.0935882106423378,
"rewards/margins": 0.013070052489638329,
"rewards/rejected": -0.10665826499462128,
"step": 30
},
{
"epoch": 0.19,
"grad_norm": 11.974913883612869,
"learning_rate": 2.0000000000000003e-06,
"log_odds_chosen": 0.12087659537792206,
"log_odds_ratio": -0.7188054323196411,
"logits/chosen": -2.883781671524048,
"logits/rejected": -2.8842344284057617,
"logps/chosen": -0.9183737635612488,
"logps/rejected": -0.9950372576713562,
"loss": 0.9799,
"nll_loss": 0.8861400485038757,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.09183738380670547,
"rewards/margins": 0.007666352204978466,
"rewards/rejected": -0.09950373321771622,
"step": 40
},
{
"epoch": 0.24,
"grad_norm": 13.063526867338696,
"learning_rate": 2.5e-06,
"log_odds_chosen": 0.4301052689552307,
"log_odds_ratio": -0.6214904189109802,
"logits/chosen": -2.832704782485962,
"logits/rejected": -2.8430252075195312,
"logps/chosen": -0.9002718925476074,
"logps/rejected": -1.2055391073226929,
"loss": 0.9584,
"nll_loss": 0.9324554204940796,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.09002719074487686,
"rewards/margins": 0.03052671253681183,
"rewards/rejected": -0.12055391073226929,
"step": 50
},
{
"epoch": 0.28,
"grad_norm": 10.235387913194325,
"learning_rate": 3e-06,
"log_odds_chosen": 0.3391026258468628,
"log_odds_ratio": -0.6046633720397949,
"logits/chosen": -2.597712993621826,
"logits/rejected": -2.598874568939209,
"logps/chosen": -0.7375704050064087,
"logps/rejected": -0.9383118748664856,
"loss": 0.8856,
"nll_loss": 0.7253891229629517,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.07375704497098923,
"rewards/margins": 0.020074138417840004,
"rewards/rejected": -0.09383118897676468,
"step": 60
},
{
"epoch": 0.33,
"grad_norm": 10.615024222922383,
"learning_rate": 3.5e-06,
"log_odds_chosen": 0.34223267436027527,
"log_odds_ratio": -0.6422346830368042,
"logits/chosen": -2.540403366088867,
"logits/rejected": -2.5442099571228027,
"logps/chosen": -0.8444018363952637,
"logps/rejected": -1.083935022354126,
"loss": 0.9384,
"nll_loss": 0.8686600923538208,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.08444017916917801,
"rewards/margins": 0.02395332232117653,
"rewards/rejected": -0.10839351266622543,
"step": 70
},
{
"epoch": 0.38,
"grad_norm": 10.120353883295946,
"learning_rate": 4.000000000000001e-06,
"log_odds_chosen": 0.026019399985671043,
"log_odds_ratio": -0.7628353834152222,
"logits/chosen": -2.6254281997680664,
"logits/rejected": -2.6229329109191895,
"logps/chosen": -0.8879353404045105,
"logps/rejected": -0.9107568860054016,
"loss": 0.921,
"nll_loss": 0.9201810956001282,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.08879353106021881,
"rewards/margins": 0.0022821640595793724,
"rewards/rejected": -0.09107569605112076,
"step": 80
},
{
"epoch": 0.43,
"grad_norm": 9.558778662613774,
"learning_rate": 4.5e-06,
"log_odds_chosen": 0.29862236976623535,
"log_odds_ratio": -0.6326144933700562,
"logits/chosen": -2.6428685188293457,
"logits/rejected": -2.621931552886963,
"logps/chosen": -0.8801389932632446,
"logps/rejected": -1.0846771001815796,
"loss": 0.9325,
"nll_loss": 0.8730384707450867,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.0880139023065567,
"rewards/margins": 0.020453810691833496,
"rewards/rejected": -0.1084677204489708,
"step": 90
},
{
"epoch": 0.47,
"grad_norm": 11.437968315222852,
"learning_rate": 5e-06,
"log_odds_chosen": 0.3877044916152954,
"log_odds_ratio": -0.6339091658592224,
"logits/chosen": -2.583739757537842,
"logits/rejected": -2.6161398887634277,
"logps/chosen": -0.8836472630500793,
"logps/rejected": -1.1088001728057861,
"loss": 0.9742,
"nll_loss": 0.9505327939987183,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.08836471289396286,
"rewards/margins": 0.02251528576016426,
"rewards/rejected": -0.11088001728057861,
"step": 100
},
{
"epoch": 0.52,
"grad_norm": 9.765547500332794,
"learning_rate": 4.767312946227961e-06,
"log_odds_chosen": 0.33769896626472473,
"log_odds_ratio": -0.6257166266441345,
"logits/chosen": -2.6281538009643555,
"logits/rejected": -2.6198818683624268,
"logps/chosen": -0.7726272344589233,
"logps/rejected": -0.9787896871566772,
"loss": 0.9403,
"nll_loss": 0.8244824409484863,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.07726272195577621,
"rewards/margins": 0.020616240799427032,
"rewards/rejected": -0.09787896275520325,
"step": 110
},
{
"epoch": 0.57,
"grad_norm": 10.192809233857067,
"learning_rate": 4.564354645876385e-06,
"log_odds_chosen": 0.31468966603279114,
"log_odds_ratio": -0.6638727784156799,
"logits/chosen": -2.6107370853424072,
"logits/rejected": -2.601783275604248,
"logps/chosen": -0.8392475247383118,
"logps/rejected": -1.036677598953247,
"loss": 0.9841,
"nll_loss": 0.8654748797416687,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.08392475545406342,
"rewards/margins": 0.019743015989661217,
"rewards/rejected": -0.10366777330636978,
"step": 120
},
{
"epoch": 0.62,
"grad_norm": 11.038589594109835,
"learning_rate": 4.385290096535147e-06,
"log_odds_chosen": 0.2760984003543854,
"log_odds_ratio": -0.6513851881027222,
"logits/chosen": -2.566074848175049,
"logits/rejected": -2.565674304962158,
"logps/chosen": -0.833086371421814,
"logps/rejected": -0.9857648611068726,
"loss": 0.9998,
"nll_loss": 0.9077309370040894,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.0833086371421814,
"rewards/margins": 0.015267851762473583,
"rewards/rejected": -0.09857650101184845,
"step": 130
},
{
"epoch": 0.66,
"grad_norm": 9.191362766920426,
"learning_rate": 4.2257712736425835e-06,
"log_odds_chosen": 0.37576600909233093,
"log_odds_ratio": -0.6204187273979187,
"logits/chosen": -2.577667474746704,
"logits/rejected": -2.584357976913452,
"logps/chosen": -0.8501531481742859,
"logps/rejected": -1.1260815858840942,
"loss": 0.9545,
"nll_loss": 0.8476670980453491,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.08501531183719635,
"rewards/margins": 0.027592848986387253,
"rewards/rejected": -0.1126081570982933,
"step": 140
},
{
"epoch": 0.71,
"grad_norm": 10.020928126499054,
"learning_rate": 4.082482904638631e-06,
"log_odds_chosen": 0.40555381774902344,
"log_odds_ratio": -0.6359796524047852,
"logits/chosen": -2.624826669692993,
"logits/rejected": -2.6218724250793457,
"logps/chosen": -0.8737274408340454,
"logps/rejected": -1.1572520732879639,
"loss": 0.9391,
"nll_loss": 0.8726488947868347,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.08737273514270782,
"rewards/margins": 0.028352469205856323,
"rewards/rejected": -0.11572520434856415,
"step": 150
},
{
"epoch": 0.76,
"grad_norm": 11.288206284049263,
"learning_rate": 3.952847075210474e-06,
"log_odds_chosen": 0.35032859444618225,
"log_odds_ratio": -0.6347723007202148,
"logits/chosen": -2.7082316875457764,
"logits/rejected": -2.711594820022583,
"logps/chosen": -0.8456757664680481,
"logps/rejected": -1.0942612886428833,
"loss": 0.9052,
"nll_loss": 0.8740180730819702,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.0845675840973854,
"rewards/margins": 0.024858549237251282,
"rewards/rejected": -0.10942612588405609,
"step": 160
},
{
"epoch": 0.81,
"grad_norm": 9.222481367340817,
"learning_rate": 3.834824944236852e-06,
"log_odds_chosen": 0.14469774067401886,
"log_odds_ratio": -0.6979460120201111,
"logits/chosen": -2.575446844100952,
"logits/rejected": -2.572465419769287,
"logps/chosen": -0.8792887926101685,
"logps/rejected": -0.9753986597061157,
"loss": 0.9229,
"nll_loss": 0.8886219263076782,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.0879288762807846,
"rewards/margins": 0.009610971435904503,
"rewards/rejected": -0.09753986448049545,
"step": 170
},
{
"epoch": 0.85,
"grad_norm": 10.622839554119508,
"learning_rate": 3.72677996249965e-06,
"log_odds_chosen": 0.18417394161224365,
"log_odds_ratio": -0.7312262058258057,
"logits/chosen": -2.558830976486206,
"logits/rejected": -2.5850634574890137,
"logps/chosen": -0.9030648469924927,
"logps/rejected": -1.0534106492996216,
"loss": 0.9502,
"nll_loss": 0.9085375070571899,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.09030647575855255,
"rewards/margins": 0.015034586191177368,
"rewards/rejected": -0.10534106194972992,
"step": 180
},
{
"epoch": 0.9,
"grad_norm": 10.63166001842217,
"learning_rate": 3.6273812505500587e-06,
"log_odds_chosen": 0.26296746730804443,
"log_odds_ratio": -0.6528322100639343,
"logits/chosen": -2.5235934257507324,
"logits/rejected": -2.509213924407959,
"logps/chosen": -0.842459499835968,
"logps/rejected": -1.0031650066375732,
"loss": 0.935,
"nll_loss": 0.840924084186554,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.0842459425330162,
"rewards/margins": 0.016070572659373283,
"rewards/rejected": -0.10031652450561523,
"step": 190
},
{
"epoch": 0.95,
"grad_norm": 10.265720627085773,
"learning_rate": 3.5355339059327378e-06,
"log_odds_chosen": 0.20157551765441895,
"log_odds_ratio": -0.6874672174453735,
"logits/chosen": -2.5119268894195557,
"logits/rejected": -2.516270399093628,
"logps/chosen": -0.8817191123962402,
"logps/rejected": -1.0264997482299805,
"loss": 0.9827,
"nll_loss": 0.9070207476615906,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.08817192167043686,
"rewards/margins": 0.01447806041687727,
"rewards/rejected": -0.10264997184276581,
"step": 200
},
{
"epoch": 1.0,
"grad_norm": 10.139451950242082,
"learning_rate": 3.450327796711771e-06,
"log_odds_chosen": 0.22201482951641083,
"log_odds_ratio": -0.6832079887390137,
"logits/chosen": -2.438990831375122,
"logits/rejected": -2.4239249229431152,
"logps/chosen": -0.9554345011711121,
"logps/rejected": -1.1055123805999756,
"loss": 0.9961,
"nll_loss": 0.9768595695495605,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.09554344415664673,
"rewards/margins": 0.01500779576599598,
"rewards/rejected": -0.11055125296115875,
"step": 210
},
{
"epoch": 1.0,
"eval_log_odds_chosen": 0.3896147906780243,
"eval_log_odds_ratio": -0.6294437050819397,
"eval_logits/chosen": -2.359004259109497,
"eval_logits/rejected": -2.3542282581329346,
"eval_logps/chosen": -0.8311433792114258,
"eval_logps/rejected": -1.0547062158584595,
"eval_loss": 0.9198942184448242,
"eval_nll_loss": 0.8774265646934509,
"eval_rewards/accuracies": 0.6302083134651184,
"eval_rewards/chosen": -0.08311434835195541,
"eval_rewards/margins": 0.022356273606419563,
"eval_rewards/rejected": -0.10547062009572983,
"eval_runtime": 61.0411,
"eval_samples_per_second": 12.287,
"eval_steps_per_second": 0.393,
"step": 211
},
{
"epoch": 1.04,
"grad_norm": 8.627109146931504,
"learning_rate": 3.3709993123162106e-06,
"log_odds_chosen": 1.4988970756530762,
"log_odds_ratio": -0.30856823921203613,
"logits/chosen": -2.3857474327087402,
"logits/rejected": -2.3801207542419434,
"logps/chosen": -0.45462799072265625,
"logps/rejected": -1.2372629642486572,
"loss": 0.564,
"nll_loss": 0.48940524458885193,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.045462802052497864,
"rewards/margins": 0.07826349139213562,
"rewards/rejected": -0.12372628599405289,
"step": 220
},
{
"epoch": 1.09,
"grad_norm": 10.837022751981742,
"learning_rate": 3.296902366978936e-06,
"log_odds_chosen": 1.769687294960022,
"log_odds_ratio": -0.22798781096935272,
"logits/chosen": -2.460228204727173,
"logits/rejected": -2.4452919960021973,
"logps/chosen": -0.48005446791648865,
"logps/rejected": -1.4746625423431396,
"loss": 0.5158,
"nll_loss": 0.5252029895782471,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.04800545051693916,
"rewards/margins": 0.09946081787347794,
"rewards/rejected": -0.1474662721157074,
"step": 230
},
{
"epoch": 1.14,
"grad_norm": 9.220899180149999,
"learning_rate": 3.2274861218395142e-06,
"log_odds_chosen": 1.664380431175232,
"log_odds_ratio": -0.2567726671695709,
"logits/chosen": -2.4479331970214844,
"logits/rejected": -2.453798294067383,
"logps/chosen": -0.43317586183547974,
"logps/rejected": -1.2501039505004883,
"loss": 0.4887,
"nll_loss": 0.45028549432754517,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.04331757873296738,
"rewards/margins": 0.08169281482696533,
"rewards/rejected": -0.1250104010105133,
"step": 240
},
{
"epoch": 1.18,
"grad_norm": 9.9378304724897,
"learning_rate": 3.1622776601683796e-06,
"log_odds_chosen": 1.5019989013671875,
"log_odds_ratio": -0.298888623714447,
"logits/chosen": -2.5074825286865234,
"logits/rejected": -2.4912190437316895,
"logps/chosen": -0.4875302314758301,
"logps/rejected": -1.2663142681121826,
"loss": 0.5504,
"nll_loss": 0.5222305655479431,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.04875302314758301,
"rewards/margins": 0.07787840068340302,
"rewards/rejected": -0.12663142383098602,
"step": 250
},
{
"epoch": 1.23,
"grad_norm": 10.42321736026305,
"learning_rate": 3.1008683647302113e-06,
"log_odds_chosen": 1.7215213775634766,
"log_odds_ratio": -0.2548612058162689,
"logits/chosen": -2.523898124694824,
"logits/rejected": -2.516082525253296,
"logps/chosen": -0.4069024920463562,
"logps/rejected": -1.2453140020370483,
"loss": 0.4878,
"nll_loss": 0.4438396394252777,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.04069024696946144,
"rewards/margins": 0.08384115993976593,
"rewards/rejected": -0.12453138828277588,
"step": 260
},
{
"epoch": 1.28,
"grad_norm": 8.517195541285803,
"learning_rate": 3.0429030972509227e-06,
"log_odds_chosen": 1.4622563123703003,
"log_odds_ratio": -0.31491032242774963,
"logits/chosen": -2.421508312225342,
"logits/rejected": -2.4460442066192627,
"logps/chosen": -0.44591742753982544,
"logps/rejected": -1.127700924873352,
"loss": 0.5104,
"nll_loss": 0.48815250396728516,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.044591739773750305,
"rewards/margins": 0.06817835569381714,
"rewards/rejected": -0.11277009546756744,
"step": 270
},
{
"epoch": 1.33,
"grad_norm": 8.676525480706056,
"learning_rate": 2.988071523335984e-06,
"log_odds_chosen": 1.713667631149292,
"log_odds_ratio": -0.23984098434448242,
"logits/chosen": -2.433767795562744,
"logits/rejected": -2.4231998920440674,
"logps/chosen": -0.42581120133399963,
"logps/rejected": -1.254382848739624,
"loss": 0.4905,
"nll_loss": 0.4856826364994049,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.04258112236857414,
"rewards/margins": 0.0828571617603302,
"rewards/rejected": -0.12543828785419464,
"step": 280
},
{
"epoch": 1.37,
"grad_norm": 10.179660663431909,
"learning_rate": 2.9361010975735177e-06,
"log_odds_chosen": 1.6406803131103516,
"log_odds_ratio": -0.2432694435119629,
"logits/chosen": -2.40620493888855,
"logits/rejected": -2.462026357650757,
"logps/chosen": -0.43379467725753784,
"logps/rejected": -1.231093406677246,
"loss": 0.4952,
"nll_loss": 0.4767753481864929,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.043379463255405426,
"rewards/margins": 0.07972987741231918,
"rewards/rejected": -0.1231093555688858,
"step": 290
},
{
"epoch": 1.42,
"grad_norm": 10.354476045311776,
"learning_rate": 2.8867513459481293e-06,
"log_odds_chosen": 1.8753684759140015,
"log_odds_ratio": -0.24006888270378113,
"logits/chosen": -2.4737935066223145,
"logits/rejected": -2.483096122741699,
"logps/chosen": -0.3926279544830322,
"logps/rejected": -1.2253657579421997,
"loss": 0.4883,
"nll_loss": 0.4013836979866028,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.0392627976834774,
"rewards/margins": 0.08327377587556839,
"rewards/rejected": -0.12253657728433609,
"step": 300
},
{
"epoch": 1.47,
"grad_norm": 9.015044311790971,
"learning_rate": 2.839809171235324e-06,
"log_odds_chosen": 1.8872253894805908,
"log_odds_ratio": -0.22469110786914825,
"logits/chosen": -2.400991678237915,
"logits/rejected": -2.397636890411377,
"logps/chosen": -0.40627264976501465,
"logps/rejected": -1.4336292743682861,
"loss": 0.5011,
"nll_loss": 0.43345755338668823,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.040627263486385345,
"rewards/margins": 0.10273567587137222,
"rewards/rejected": -0.14336295425891876,
"step": 310
},
{
"epoch": 1.52,
"grad_norm": 9.539457700499925,
"learning_rate": 2.7950849718747376e-06,
"log_odds_chosen": 1.8418588638305664,
"log_odds_ratio": -0.20935162901878357,
"logits/chosen": -2.3977291584014893,
"logits/rejected": -2.3498239517211914,
"logps/chosen": -0.4429972767829895,
"logps/rejected": -1.4245294332504272,
"loss": 0.4827,
"nll_loss": 0.4764658510684967,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.04429972171783447,
"rewards/margins": 0.09815323352813721,
"rewards/rejected": -0.14245295524597168,
"step": 320
},
{
"epoch": 1.56,
"grad_norm": 9.434503294939443,
"learning_rate": 2.752409412815902e-06,
"log_odds_chosen": 1.5989511013031006,
"log_odds_ratio": -0.26907485723495483,
"logits/chosen": -2.4358298778533936,
"logits/rejected": -2.4641692638397217,
"logps/chosen": -0.4505313038825989,
"logps/rejected": -1.2999070882797241,
"loss": 0.4685,
"nll_loss": 0.49505728483200073,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.04505313187837601,
"rewards/margins": 0.08493757992982864,
"rewards/rejected": -0.12999072670936584,
"step": 330
},
{
"epoch": 1.61,
"grad_norm": 9.785466529429728,
"learning_rate": 2.711630722733202e-06,
"log_odds_chosen": 1.757354736328125,
"log_odds_ratio": -0.22594013810157776,
"logits/chosen": -2.386993885040283,
"logits/rejected": -2.3797829151153564,
"logps/chosen": -0.3836737275123596,
"logps/rejected": -1.226621389389038,
"loss": 0.5139,
"nll_loss": 0.4257877469062805,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.0383673757314682,
"rewards/margins": 0.08429475873708725,
"rewards/rejected": -0.12266211211681366,
"step": 340
},
{
"epoch": 1.66,
"grad_norm": 9.783556860345605,
"learning_rate": 2.6726124191242444e-06,
"log_odds_chosen": 1.5875946283340454,
"log_odds_ratio": -0.282478392124176,
"logits/chosen": -2.4133667945861816,
"logits/rejected": -2.4031758308410645,
"logps/chosen": -0.47809138894081116,
"logps/rejected": -1.3111228942871094,
"loss": 0.4947,
"nll_loss": 0.46197766065597534,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.04780913516879082,
"rewards/margins": 0.08330314606428146,
"rewards/rejected": -0.13111227750778198,
"step": 350
},
{
"epoch": 1.71,
"grad_norm": 10.952497066310066,
"learning_rate": 2.6352313834736496e-06,
"log_odds_chosen": 1.444085717201233,
"log_odds_ratio": -0.302844375371933,
"logits/chosen": -2.35420560836792,
"logits/rejected": -2.40644907951355,
"logps/chosen": -0.4726253151893616,
"logps/rejected": -1.181214451789856,
"loss": 0.5344,
"nll_loss": 0.5417958498001099,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.0472625270485878,
"rewards/margins": 0.0708589106798172,
"rewards/rejected": -0.1181214451789856,
"step": 360
},
{
"epoch": 1.75,
"grad_norm": 9.337096555826308,
"learning_rate": 2.599376224550182e-06,
"log_odds_chosen": 1.5420761108398438,
"log_odds_ratio": -0.2706839144229889,
"logits/chosen": -2.393585681915283,
"logits/rejected": -2.3775479793548584,
"logps/chosen": -0.4294983744621277,
"logps/rejected": -1.177330493927002,
"loss": 0.489,
"nll_loss": 0.4465916156768799,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.04294983297586441,
"rewards/margins": 0.07478321343660355,
"rewards/rejected": -0.11773304641246796,
"step": 370
},
{
"epoch": 1.8,
"grad_norm": 9.897099582842259,
"learning_rate": 2.564945880212886e-06,
"log_odds_chosen": 1.6586507558822632,
"log_odds_ratio": -0.2737709581851959,
"logits/chosen": -2.505715847015381,
"logits/rejected": -2.478332996368408,
"logps/chosen": -0.47623515129089355,
"logps/rejected": -1.3953008651733398,
"loss": 0.5232,
"nll_loss": 0.498232901096344,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.047623515129089355,
"rewards/margins": 0.0919065773487091,
"rewards/rejected": -0.13953009247779846,
"step": 380
},
{
"epoch": 1.85,
"grad_norm": 8.744712266716895,
"learning_rate": 2.5318484177091667e-06,
"log_odds_chosen": 1.4230409860610962,
"log_odds_ratio": -0.29352661967277527,
"logits/chosen": -2.455392360687256,
"logits/rejected": -2.4511191844940186,
"logps/chosen": -0.5203135013580322,
"logps/rejected": -1.2620891332626343,
"loss": 0.5658,
"nll_loss": 0.547887921333313,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.05203135684132576,
"rewards/margins": 0.0741775631904602,
"rewards/rejected": -0.12620893120765686,
"step": 390
},
{
"epoch": 1.9,
"grad_norm": 9.30311341527825,
"learning_rate": 2.5e-06,
"log_odds_chosen": 1.7381293773651123,
"log_odds_ratio": -0.25197070837020874,
"logits/chosen": -2.512232780456543,
"logits/rejected": -2.5167810916900635,
"logps/chosen": -0.4353383183479309,
"logps/rejected": -1.34853196144104,
"loss": 0.5158,
"nll_loss": 0.4797208309173584,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.04353383183479309,
"rewards/margins": 0.09131935983896255,
"rewards/rejected": -0.13485319912433624,
"step": 400
},
{
"epoch": 1.94,
"grad_norm": 9.272816070528886,
"learning_rate": 2.4693239916239746e-06,
"log_odds_chosen": 1.5647149085998535,
"log_odds_ratio": -0.29222574830055237,
"logits/chosen": -2.4848999977111816,
"logits/rejected": -2.4973931312561035,
"logps/chosen": -0.4486003816127777,
"logps/rejected": -1.2298743724822998,
"loss": 0.4991,
"nll_loss": 0.49500808119773865,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.04486004263162613,
"rewards/margins": 0.07812739908695221,
"rewards/rejected": -0.12298743426799774,
"step": 410
},
{
"epoch": 1.99,
"grad_norm": 11.956370778331848,
"learning_rate": 2.4397501823713327e-06,
"log_odds_chosen": 1.8623790740966797,
"log_odds_ratio": -0.23835237324237823,
"logits/chosen": -2.491649866104126,
"logits/rejected": -2.4964215755462646,
"logps/chosen": -0.36661866307258606,
"logps/rejected": -1.2916817665100098,
"loss": 0.4999,
"nll_loss": 0.4245728850364685,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.03666186332702637,
"rewards/margins": 0.09250631183385849,
"rewards/rejected": -0.12916818261146545,
"step": 420
},
{
"epoch": 2.0,
"eval_log_odds_chosen": 0.4993869364261627,
"eval_log_odds_ratio": -0.6011503338813782,
"eval_logits/chosen": -2.51769757270813,
"eval_logits/rejected": -2.518415689468384,
"eval_logps/chosen": -0.8936966061592102,
"eval_logps/rejected": -1.183638334274292,
"eval_loss": 0.9812777638435364,
"eval_nll_loss": 0.940436840057373,
"eval_rewards/accuracies": 0.6614583134651184,
"eval_rewards/chosen": -0.08936966210603714,
"eval_rewards/margins": 0.028994165360927582,
"eval_rewards/rejected": -0.11836383491754532,
"eval_runtime": 60.9721,
"eval_samples_per_second": 12.301,
"eval_steps_per_second": 0.394,
"step": 422
},
{
"epoch": 2.04,
"grad_norm": 11.137133583839702,
"learning_rate": 2.411214110852061e-06,
"log_odds_chosen": 2.8829193115234375,
"log_odds_ratio": -0.11160006374120712,
"logits/chosen": -2.481720209121704,
"logits/rejected": -2.5044140815734863,
"logps/chosen": -0.21169082820415497,
"logps/rejected": -1.3710792064666748,
"loss": 0.2521,
"nll_loss": 0.2048369199037552,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.021169083192944527,
"rewards/margins": 0.11593882739543915,
"rewards/rejected": -0.13710792362689972,
"step": 430
},
{
"epoch": 2.09,
"grad_norm": 9.190185309499267,
"learning_rate": 2.3836564731139807e-06,
"log_odds_chosen": 3.475831985473633,
"log_odds_ratio": -0.0667722150683403,
"logits/chosen": -2.468839645385742,
"logits/rejected": -2.504293441772461,
"logps/chosen": -0.13927796483039856,
"logps/rejected": -1.5609562397003174,
"loss": 0.1889,
"nll_loss": 0.142722949385643,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.013927794992923737,
"rewards/margins": 0.1421678364276886,
"rewards/rejected": -0.15609565377235413,
"step": 440
},
{
"epoch": 2.13,
"grad_norm": 9.564387172693797,
"learning_rate": 2.357022603955159e-06,
"log_odds_chosen": 3.917105197906494,
"log_odds_ratio": -0.037336982786655426,
"logits/chosen": -2.5166149139404297,
"logits/rejected": -2.545985698699951,
"logps/chosen": -0.13963064551353455,
"logps/rejected": -1.9594411849975586,
"loss": 0.181,
"nll_loss": 0.15659931302070618,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.013963064178824425,
"rewards/margins": 0.18198105692863464,
"rewards/rejected": -0.19594410061836243,
"step": 450
},
{
"epoch": 2.18,
"grad_norm": 9.849074304214747,
"learning_rate": 2.3312620206007847e-06,
"log_odds_chosen": 3.3254103660583496,
"log_odds_ratio": -0.06604982912540436,
"logits/chosen": -2.4885592460632324,
"logits/rejected": -2.5000545978546143,
"logps/chosen": -0.17423538863658905,
"logps/rejected": -1.5009019374847412,
"loss": 0.1952,
"nll_loss": 0.21155783534049988,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.017423538491129875,
"rewards/margins": 0.13266664743423462,
"rewards/rejected": -0.15009018778800964,
"step": 460
},
{
"epoch": 2.23,
"grad_norm": 9.877044500517252,
"learning_rate": 2.3063280200722128e-06,
"log_odds_chosen": 3.6100239753723145,
"log_odds_ratio": -0.05039358139038086,
"logits/chosen": -2.5267927646636963,
"logits/rejected": -2.5161995887756348,
"logps/chosen": -0.16191843152046204,
"logps/rejected": -1.8569713830947876,
"loss": 0.173,
"nll_loss": 0.16810880601406097,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.016191843897104263,
"rewards/margins": 0.1695052832365036,
"rewards/rejected": -0.18569712340831757,
"step": 470
},
{
"epoch": 2.27,
"grad_norm": 8.163874170175868,
"learning_rate": 2.2821773229381924e-06,
"log_odds_chosen": 3.612773895263672,
"log_odds_ratio": -0.05763618275523186,
"logits/chosen": -2.5359530448913574,
"logits/rejected": -2.539139747619629,
"logps/chosen": -0.17719587683677673,
"logps/rejected": -1.8914750814437866,
"loss": 0.1808,
"nll_loss": 0.17015354335308075,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.017719587311148643,
"rewards/margins": 0.171427920460701,
"rewards/rejected": -0.18914751708507538,
"step": 480
},
{
"epoch": 2.32,
"grad_norm": 9.330097236244276,
"learning_rate": 2.2587697572631284e-06,
"log_odds_chosen": 3.656383514404297,
"log_odds_ratio": -0.05575472116470337,
"logits/chosen": -2.450357437133789,
"logits/rejected": -2.4894728660583496,
"logps/chosen": -0.12832140922546387,
"logps/rejected": -1.6764119863510132,
"loss": 0.1743,
"nll_loss": 0.1293896734714508,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.012832140550017357,
"rewards/margins": 0.15480905771255493,
"rewards/rejected": -0.16764120757579803,
"step": 490
},
{
"epoch": 2.37,
"grad_norm": 8.638065176436028,
"learning_rate": 2.23606797749979e-06,
"log_odds_chosen": 3.5653939247131348,
"log_odds_ratio": -0.04521235078573227,
"logits/chosen": -2.4729416370391846,
"logits/rejected": -2.4972083568573,
"logps/chosen": -0.14347168803215027,
"logps/rejected": -1.709639549255371,
"loss": 0.1687,
"nll_loss": 0.15304037928581238,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.014347168616950512,
"rewards/margins": 0.15661677718162537,
"rewards/rejected": -0.17096397280693054,
"step": 500
},
{
"epoch": 2.42,
"grad_norm": 9.821968133030161,
"learning_rate": 2.2140372138502386e-06,
"log_odds_chosen": 3.384936809539795,
"log_odds_ratio": -0.06072957068681717,
"logits/chosen": -2.477996349334717,
"logits/rejected": -2.482419729232788,
"logps/chosen": -0.17441074550151825,
"logps/rejected": -1.6888253688812256,
"loss": 0.191,
"nll_loss": 0.20637556910514832,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.017441075295209885,
"rewards/margins": 0.15144145488739014,
"rewards/rejected": -0.16888253390789032,
"step": 510
},
{
"epoch": 2.46,
"grad_norm": 9.230367667532265,
"learning_rate": 2.1926450482675734e-06,
"log_odds_chosen": 3.698854446411133,
"log_odds_ratio": -0.04557035490870476,
"logits/chosen": -2.493439197540283,
"logits/rejected": -2.507626533508301,
"logps/chosen": -0.15794190764427185,
"logps/rejected": -1.836679458618164,
"loss": 0.1823,
"nll_loss": 0.1679120659828186,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.015794191509485245,
"rewards/margins": 0.16787376999855042,
"rewards/rejected": -0.18366795778274536,
"step": 520
},
{
"epoch": 2.51,
"grad_norm": 9.220963630868974,
"learning_rate": 2.1718612138153473e-06,
"log_odds_chosen": 3.2164981365203857,
"log_odds_ratio": -0.08045772463083267,
"logits/chosen": -2.4849820137023926,
"logits/rejected": -2.513134717941284,
"logps/chosen": -0.20204570889472961,
"logps/rejected": -1.6124742031097412,
"loss": 0.1955,
"nll_loss": 0.1980821043252945,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.0202045738697052,
"rewards/margins": 0.14104285836219788,
"rewards/rejected": -0.16124743223190308,
"step": 530
},
{
"epoch": 2.56,
"grad_norm": 10.33252022252946,
"learning_rate": 2.151657414559676e-06,
"log_odds_chosen": 3.355163097381592,
"log_odds_ratio": -0.05846261978149414,
"logits/chosen": -2.4703683853149414,
"logits/rejected": -2.4815986156463623,
"logps/chosen": -0.16165122389793396,
"logps/rejected": -1.595573902130127,
"loss": 0.1944,
"nll_loss": 0.1843605637550354,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.016165124252438545,
"rewards/margins": 0.14339227974414825,
"rewards/rejected": -0.15955740213394165,
"step": 540
},
{
"epoch": 2.61,
"grad_norm": 10.384732401123731,
"learning_rate": 2.132007163556104e-06,
"log_odds_chosen": 3.7364859580993652,
"log_odds_ratio": -0.050393521785736084,
"logits/chosen": -2.488259792327881,
"logits/rejected": -2.49784517288208,
"logps/chosen": -0.1458165943622589,
"logps/rejected": -1.7980148792266846,
"loss": 0.1656,
"nll_loss": 0.16633237898349762,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.014581659808754921,
"rewards/margins": 0.16521981358528137,
"rewards/rejected": -0.17980147898197174,
"step": 550
},
{
"epoch": 2.65,
"grad_norm": 8.88379907455603,
"learning_rate": 2.1128856368212917e-06,
"log_odds_chosen": 3.3721110820770264,
"log_odds_ratio": -0.05570857599377632,
"logits/chosen": -2.4560656547546387,
"logits/rejected": -2.459944725036621,
"logps/chosen": -0.14691922068595886,
"logps/rejected": -1.4988293647766113,
"loss": 0.1941,
"nll_loss": 0.16041871905326843,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.014691921882331371,
"rewards/margins": 0.13519100844860077,
"rewards/rejected": -0.1498829424381256,
"step": 560
},
{
"epoch": 2.7,
"grad_norm": 8.564539697443587,
"learning_rate": 2.0942695414584777e-06,
"log_odds_chosen": 3.540961503982544,
"log_odds_ratio": -0.050423987209796906,
"logits/chosen": -2.4706387519836426,
"logits/rejected": -2.478842258453369,
"logps/chosen": -0.15486404299736023,
"logps/rejected": -1.6839189529418945,
"loss": 0.1827,
"nll_loss": 0.16179664433002472,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.015486404299736023,
"rewards/margins": 0.15290549397468567,
"rewards/rejected": -0.1683918982744217,
"step": 570
},
{
"epoch": 2.75,
"grad_norm": 8.776123151000377,
"learning_rate": 2.0761369963434992e-06,
"log_odds_chosen": 3.2635269165039062,
"log_odds_ratio": -0.07641835510730743,
"logits/chosen": -2.49845552444458,
"logits/rejected": -2.5065455436706543,
"logps/chosen": -0.21721407771110535,
"logps/rejected": -1.628244161605835,
"loss": 0.2085,
"nll_loss": 0.26145899295806885,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.021721405908465385,
"rewards/margins": 0.14110301434993744,
"rewards/rejected": -0.16282442212104797,
"step": 580
},
{
"epoch": 2.8,
"grad_norm": 8.360109402740795,
"learning_rate": 2.058467423981546e-06,
"log_odds_chosen": 3.5911190509796143,
"log_odds_ratio": -0.04746793955564499,
"logits/chosen": -2.4481544494628906,
"logits/rejected": -2.4686474800109863,
"logps/chosen": -0.18100793659687042,
"logps/rejected": -1.7877342700958252,
"loss": 0.1973,
"nll_loss": 0.19511719048023224,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.018100792542099953,
"rewards/margins": 0.1606726348400116,
"rewards/rejected": -0.178773432970047,
"step": 590
},
{
"epoch": 2.84,
"grad_norm": 8.230484923418327,
"learning_rate": 2.0412414523193154e-06,
"log_odds_chosen": 3.52643084526062,
"log_odds_ratio": -0.05582552030682564,
"logits/chosen": -2.43485689163208,
"logits/rejected": -2.452052116394043,
"logps/chosen": -0.16544193029403687,
"logps/rejected": -1.6863620281219482,
"loss": 0.1863,
"nll_loss": 0.17863860726356506,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.016544191166758537,
"rewards/margins": 0.15209202468395233,
"rewards/rejected": -0.16863620281219482,
"step": 600
},
{
"epoch": 2.89,
"grad_norm": 9.657714694141937,
"learning_rate": 2.0244408254472904e-06,
"log_odds_chosen": 3.441495418548584,
"log_odds_ratio": -0.050580836832523346,
"logits/chosen": -2.3969621658325195,
"logits/rejected": -2.4254117012023926,
"logps/chosen": -0.1414327472448349,
"logps/rejected": -1.5730435848236084,
"loss": 0.1825,
"nll_loss": 0.16234278678894043,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.01414327509701252,
"rewards/margins": 0.1431610882282257,
"rewards/rejected": -0.15730436146259308,
"step": 610
},
{
"epoch": 2.94,
"grad_norm": 9.418203083185244,
"learning_rate": 2.0080483222562476e-06,
"log_odds_chosen": 3.510298252105713,
"log_odds_ratio": -0.04852527379989624,
"logits/chosen": -2.365495443344116,
"logits/rejected": -2.3816208839416504,
"logps/chosen": -0.14439718425273895,
"logps/rejected": -1.5629642009735107,
"loss": 0.1816,
"nll_loss": 0.1693393439054489,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.014439716935157776,
"rewards/margins": 0.14185671508312225,
"rewards/rejected": -0.15629643201828003,
"step": 620
},
{
"epoch": 2.99,
"grad_norm": 9.905181462463698,
"learning_rate": 1.9920476822239895e-06,
"log_odds_chosen": 3.6201272010803223,
"log_odds_ratio": -0.04453429579734802,
"logits/chosen": -2.415280818939209,
"logits/rejected": -2.403134822845459,
"logps/chosen": -0.1919378936290741,
"logps/rejected": -1.9056262969970703,
"loss": 0.1899,
"nll_loss": 0.1791400909423828,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.01919379085302353,
"rewards/margins": 0.17136886715888977,
"rewards/rejected": -0.1905626356601715,
"step": 630
},
{
"epoch": 3.0,
"eval_log_odds_chosen": 0.539490818977356,
"eval_log_odds_ratio": -0.6108235716819763,
"eval_logits/chosen": -2.4178202152252197,
"eval_logits/rejected": -2.420111894607544,
"eval_logps/chosen": -1.1129909753799438,
"eval_logps/rejected": -1.4606622457504272,
"eval_loss": 1.2152478694915771,
"eval_nll_loss": 1.1710882186889648,
"eval_rewards/accuracies": 0.65625,
"eval_rewards/chosen": -0.11129910498857498,
"eval_rewards/margins": 0.03476712107658386,
"eval_rewards/rejected": -0.14606623351573944,
"eval_runtime": 60.9563,
"eval_samples_per_second": 12.304,
"eval_steps_per_second": 0.394,
"step": 633
},
{
"epoch": 3.0,
"step": 633,
"total_flos": 0.0,
"train_loss": 0.5526923979816467,
"train_runtime": 6310.4087,
"train_samples_per_second": 3.209,
"train_steps_per_second": 0.1
}
],
"logging_steps": 10,
"max_steps": 633,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"total_flos": 0.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}