|
{ |
|
"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 |
|
} |
|
|