| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 2.7777777777777777, | |
| "eval_steps": 50, | |
| "global_step": 200, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.06944444444444445, | |
| "grad_norm": 34.922038802469906, | |
| "learning_rate": 5e-07, | |
| "logits/chosen": -2.745856761932373, | |
| "logits/rejected": -2.7519428730010986, | |
| "logps/chosen": -158.59893798828125, | |
| "logps/rejected": -160.2094268798828, | |
| "loss": 0.6939, | |
| "rewards/accuracies": 0.33125001192092896, | |
| "rewards/chosen": 0.000581090513151139, | |
| "rewards/margins": -0.00047404784709215164, | |
| "rewards/rejected": 0.0010551378363743424, | |
| "step": 5 | |
| }, | |
| { | |
| "epoch": 0.1388888888888889, | |
| "grad_norm": 36.413747880379944, | |
| "learning_rate": 1e-06, | |
| "logits/chosen": -2.736849546432495, | |
| "logits/rejected": -2.7453856468200684, | |
| "logps/chosen": -174.9836883544922, | |
| "logps/rejected": -171.3789825439453, | |
| "loss": 0.6883, | |
| "rewards/accuracies": 0.5249999761581421, | |
| "rewards/chosen": 0.022891324013471603, | |
| "rewards/margins": 0.0072159310802817345, | |
| "rewards/rejected": 0.015675392001867294, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.20833333333333334, | |
| "grad_norm": 37.39831623643963, | |
| "learning_rate": 9.985471028179154e-07, | |
| "logits/chosen": -2.6100494861602783, | |
| "logits/rejected": -2.613814115524292, | |
| "logps/chosen": -173.74456787109375, | |
| "logps/rejected": -171.9447784423828, | |
| "loss": 0.6733, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": 0.21788981556892395, | |
| "rewards/margins": 0.09981251507997513, | |
| "rewards/rejected": 0.11807730048894882, | |
| "step": 15 | |
| }, | |
| { | |
| "epoch": 0.2777777777777778, | |
| "grad_norm": 32.21436012894337, | |
| "learning_rate": 9.94196854912548e-07, | |
| "logits/chosen": -2.52358341217041, | |
| "logits/rejected": -2.5229320526123047, | |
| "logps/chosen": -155.1029815673828, | |
| "logps/rejected": -159.9917755126953, | |
| "loss": 0.653, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": 0.3672500252723694, | |
| "rewards/margins": 0.20777833461761475, | |
| "rewards/rejected": 0.15947169065475464, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.3472222222222222, | |
| "grad_norm": 32.0223445235898, | |
| "learning_rate": 9.869745381355905e-07, | |
| "logits/chosen": -2.4307353496551514, | |
| "logits/rejected": -2.429652452468872, | |
| "logps/chosen": -159.20545959472656, | |
| "logps/rejected": -158.57797241210938, | |
| "loss": 0.6327, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": 0.3310449421405792, | |
| "rewards/margins": 0.22794027626514435, | |
| "rewards/rejected": 0.10310468822717667, | |
| "step": 25 | |
| }, | |
| { | |
| "epoch": 0.4166666666666667, | |
| "grad_norm": 36.79676539267814, | |
| "learning_rate": 9.769221256218162e-07, | |
| "logits/chosen": -2.4173130989074707, | |
| "logits/rejected": -2.3855977058410645, | |
| "logps/chosen": -165.49618530273438, | |
| "logps/rejected": -164.41111755371094, | |
| "loss": 0.6264, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -0.17511680722236633, | |
| "rewards/margins": 0.2607855200767517, | |
| "rewards/rejected": -0.43590235710144043, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.4861111111111111, | |
| "grad_norm": 32.497757396344305, | |
| "learning_rate": 9.64098037858483e-07, | |
| "logits/chosen": -2.482489824295044, | |
| "logits/rejected": -2.4674153327941895, | |
| "logps/chosen": -162.7029571533203, | |
| "logps/rejected": -164.48619079589844, | |
| "loss": 0.5914, | |
| "rewards/accuracies": 0.6499999761581421, | |
| "rewards/chosen": -0.3191075921058655, | |
| "rewards/margins": 0.362353652715683, | |
| "rewards/rejected": -0.6814612150192261, | |
| "step": 35 | |
| }, | |
| { | |
| "epoch": 0.5555555555555556, | |
| "grad_norm": 34.40764043054464, | |
| "learning_rate": 9.485768031694871e-07, | |
| "logits/chosen": -2.518897771835327, | |
| "logits/rejected": -2.5143680572509766, | |
| "logps/chosen": -167.1140899658203, | |
| "logps/rejected": -168.67987060546875, | |
| "loss": 0.58, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -0.037984561175107956, | |
| "rewards/margins": 0.48854589462280273, | |
| "rewards/rejected": -0.5265304446220398, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.625, | |
| "grad_norm": 33.97876407643923, | |
| "learning_rate": 9.304486245873971e-07, | |
| "logits/chosen": -2.542407989501953, | |
| "logits/rejected": -2.510068416595459, | |
| "logps/chosen": -163.48536682128906, | |
| "logps/rejected": -164.54147338867188, | |
| "loss": 0.5815, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -0.062438905239105225, | |
| "rewards/margins": 0.5621680021286011, | |
| "rewards/rejected": -0.6246069073677063, | |
| "step": 45 | |
| }, | |
| { | |
| "epoch": 0.6944444444444444, | |
| "grad_norm": 32.726531127211416, | |
| "learning_rate": 9.098188556305262e-07, | |
| "logits/chosen": -2.513942241668701, | |
| "logits/rejected": -2.51139235496521, | |
| "logps/chosen": -165.90362548828125, | |
| "logps/rejected": -170.9777069091797, | |
| "loss": 0.5342, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": -0.2061791867017746, | |
| "rewards/margins": 0.6521804928779602, | |
| "rewards/rejected": -0.8583596348762512, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.6944444444444444, | |
| "eval_logits/chosen": -2.4995272159576416, | |
| "eval_logits/rejected": -2.4678475856781006, | |
| "eval_logps/chosen": -159.87913513183594, | |
| "eval_logps/rejected": -170.17739868164062, | |
| "eval_loss": 0.5209046602249146, | |
| "eval_rewards/accuracies": 0.71484375, | |
| "eval_rewards/chosen": -0.20708096027374268, | |
| "eval_rewards/margins": 0.7420069575309753, | |
| "eval_rewards/rejected": -0.9490878582000732, | |
| "eval_runtime": 129.0562, | |
| "eval_samples_per_second": 15.83, | |
| "eval_steps_per_second": 0.248, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.7638888888888888, | |
| "grad_norm": 30.615472770103583, | |
| "learning_rate": 8.868073880316123e-07, | |
| "logits/chosen": -2.500612497329712, | |
| "logits/rejected": -2.520470142364502, | |
| "logps/chosen": -177.12451171875, | |
| "logps/rejected": -183.56619262695312, | |
| "loss": 0.5577, | |
| "rewards/accuracies": 0.737500011920929, | |
| "rewards/chosen": -0.32517191767692566, | |
| "rewards/margins": 0.8004968762397766, | |
| "rewards/rejected": -1.1256687641143799, | |
| "step": 55 | |
| }, | |
| { | |
| "epoch": 0.8333333333333334, | |
| "grad_norm": 32.6842144140191, | |
| "learning_rate": 8.615479549763755e-07, | |
| "logits/chosen": -2.5329596996307373, | |
| "logits/rejected": -2.5310966968536377, | |
| "logps/chosen": -183.05056762695312, | |
| "logps/rejected": -193.45396423339844, | |
| "loss": 0.5335, | |
| "rewards/accuracies": 0.7749999761581421, | |
| "rewards/chosen": -0.4021238386631012, | |
| "rewards/margins": 0.8954024314880371, | |
| "rewards/rejected": -1.297526240348816, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.9027777777777778, | |
| "grad_norm": 31.818374577978716, | |
| "learning_rate": 8.341873539012443e-07, | |
| "logits/chosen": -2.518287420272827, | |
| "logits/rejected": -2.5078444480895996, | |
| "logps/chosen": -168.2073974609375, | |
| "logps/rejected": -179.64515686035156, | |
| "loss": 0.5272, | |
| "rewards/accuracies": 0.675000011920929, | |
| "rewards/chosen": -0.6193049550056458, | |
| "rewards/margins": 0.8720922470092773, | |
| "rewards/rejected": -1.4913971424102783, | |
| "step": 65 | |
| }, | |
| { | |
| "epoch": 0.9722222222222222, | |
| "grad_norm": 30.74282933667216, | |
| "learning_rate": 8.048845933670271e-07, | |
| "logits/chosen": -2.474501371383667, | |
| "logits/rejected": -2.486248731613159, | |
| "logps/chosen": -170.53952026367188, | |
| "logps/rejected": -183.97750854492188, | |
| "loss": 0.516, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -1.0543142557144165, | |
| "rewards/margins": 0.9860897064208984, | |
| "rewards/rejected": -2.0404040813446045, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 1.0416666666666667, | |
| "grad_norm": 20.857882794392346, | |
| "learning_rate": 7.738099689665539e-07, | |
| "logits/chosen": -2.4805214405059814, | |
| "logits/rejected": -2.4716594219207764, | |
| "logps/chosen": -179.7224884033203, | |
| "logps/rejected": -194.9944610595703, | |
| "loss": 0.3668, | |
| "rewards/accuracies": 0.800000011920929, | |
| "rewards/chosen": -0.5534920692443848, | |
| "rewards/margins": 1.5888177156448364, | |
| "rewards/rejected": -2.1423099040985107, | |
| "step": 75 | |
| }, | |
| { | |
| "epoch": 1.1111111111111112, | |
| "grad_norm": 19.198293152522673, | |
| "learning_rate": 7.41144073636728e-07, | |
| "logits/chosen": -2.4745230674743652, | |
| "logits/rejected": -2.449141025543213, | |
| "logps/chosen": -166.55409240722656, | |
| "logps/rejected": -190.5934295654297, | |
| "loss": 0.2405, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": 0.21927456557750702, | |
| "rewards/margins": 2.280102014541626, | |
| "rewards/rejected": -2.0608274936676025, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 1.1805555555555556, | |
| "grad_norm": 22.513577769114043, | |
| "learning_rate": 7.070767481266492e-07, | |
| "logits/chosen": -2.457094192504883, | |
| "logits/rejected": -2.434199571609497, | |
| "logps/chosen": -168.0836944580078, | |
| "logps/rejected": -188.27374267578125, | |
| "loss": 0.2396, | |
| "rewards/accuracies": 0.925000011920929, | |
| "rewards/chosen": -0.004831351339817047, | |
| "rewards/margins": 2.4242899417877197, | |
| "rewards/rejected": -2.429121494293213, | |
| "step": 85 | |
| }, | |
| { | |
| "epoch": 1.25, | |
| "grad_norm": 18.850259072813216, | |
| "learning_rate": 6.718059777212565e-07, | |
| "logits/chosen": -2.4347877502441406, | |
| "logits/rejected": -2.432413101196289, | |
| "logps/chosen": -154.73831176757812, | |
| "logps/rejected": -174.75621032714844, | |
| "loss": 0.2034, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -0.1518968939781189, | |
| "rewards/margins": 2.676344394683838, | |
| "rewards/rejected": -2.8282413482666016, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.3194444444444444, | |
| "grad_norm": 19.10543784815799, | |
| "learning_rate": 6.355367416322778e-07, | |
| "logits/chosen": -2.433638572692871, | |
| "logits/rejected": -2.4281392097473145, | |
| "logps/chosen": -173.74961853027344, | |
| "logps/rejected": -198.87208557128906, | |
| "loss": 0.2048, | |
| "rewards/accuracies": 0.8812500238418579, | |
| "rewards/chosen": -0.30015555024147034, | |
| "rewards/margins": 2.652449369430542, | |
| "rewards/rejected": -2.9526050090789795, | |
| "step": 95 | |
| }, | |
| { | |
| "epoch": 1.3888888888888888, | |
| "grad_norm": 19.991814546360683, | |
| "learning_rate": 5.984798217433531e-07, | |
| "logits/chosen": -2.434572696685791, | |
| "logits/rejected": -2.417861223220825, | |
| "logps/chosen": -158.5247802734375, | |
| "logps/rejected": -190.9485626220703, | |
| "loss": 0.217, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -0.37773507833480835, | |
| "rewards/margins": 2.8572323322296143, | |
| "rewards/rejected": -3.2349674701690674, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.3888888888888888, | |
| "eval_logits/chosen": -2.4349896907806396, | |
| "eval_logits/rejected": -2.40444278717041, | |
| "eval_logps/chosen": -169.47793579101562, | |
| "eval_logps/rejected": -184.55125427246094, | |
| "eval_loss": 0.5166749954223633, | |
| "eval_rewards/accuracies": 0.70703125, | |
| "eval_rewards/chosen": -1.1669622659683228, | |
| "eval_rewards/margins": 1.2195117473602295, | |
| "eval_rewards/rejected": -2.386474132537842, | |
| "eval_runtime": 128.5113, | |
| "eval_samples_per_second": 15.897, | |
| "eval_steps_per_second": 0.249, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 1.4583333333333333, | |
| "grad_norm": 28.299743381287744, | |
| "learning_rate": 5.608505776324157e-07, | |
| "logits/chosen": -2.4249603748321533, | |
| "logits/rejected": -2.4180781841278076, | |
| "logps/chosen": -167.8837432861328, | |
| "logps/rejected": -196.94412231445312, | |
| "loss": 0.2013, | |
| "rewards/accuracies": 0.9375, | |
| "rewards/chosen": -0.45704206824302673, | |
| "rewards/margins": 3.021711826324463, | |
| "rewards/rejected": -3.4787545204162598, | |
| "step": 105 | |
| }, | |
| { | |
| "epoch": 1.5277777777777777, | |
| "grad_norm": 20.695727476520528, | |
| "learning_rate": 5.228676949903973e-07, | |
| "logits/chosen": -2.374063491821289, | |
| "logits/rejected": -2.3558177947998047, | |
| "logps/chosen": -176.4818572998047, | |
| "logps/rejected": -208.8624267578125, | |
| "loss": 0.1943, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -0.3167189359664917, | |
| "rewards/margins": 3.428766965866089, | |
| "rewards/rejected": -3.7454864978790283, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 1.5972222222222223, | |
| "grad_norm": 25.767538093776388, | |
| "learning_rate": 4.847519147099294e-07, | |
| "logits/chosen": -2.302668333053589, | |
| "logits/rejected": -2.28350567817688, | |
| "logps/chosen": -169.49490356445312, | |
| "logps/rejected": -201.66481018066406, | |
| "loss": 0.1966, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": -0.7796251177787781, | |
| "rewards/margins": 2.979997158050537, | |
| "rewards/rejected": -3.7596218585968018, | |
| "step": 115 | |
| }, | |
| { | |
| "epoch": 1.6666666666666665, | |
| "grad_norm": 25.2139533526534, | |
| "learning_rate": 4.46724750030062e-07, | |
| "logits/chosen": -2.2816123962402344, | |
| "logits/rejected": -2.2772622108459473, | |
| "logps/chosen": -172.01380920410156, | |
| "logps/rejected": -204.12197875976562, | |
| "loss": 0.1904, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": -0.6298034191131592, | |
| "rewards/margins": 3.174367904663086, | |
| "rewards/rejected": -3.804171323776245, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 1.7361111111111112, | |
| "grad_norm": 23.046670885435994, | |
| "learning_rate": 4.0900719919241935e-07, | |
| "logits/chosen": -2.2773830890655518, | |
| "logits/rejected": -2.252821207046509, | |
| "logps/chosen": -174.70835876464844, | |
| "logps/rejected": -208.0013885498047, | |
| "loss": 0.1926, | |
| "rewards/accuracies": 0.925000011920929, | |
| "rewards/chosen": -0.6799504160881042, | |
| "rewards/margins": 3.1291940212249756, | |
| "rewards/rejected": -3.8091444969177246, | |
| "step": 125 | |
| }, | |
| { | |
| "epoch": 1.8055555555555556, | |
| "grad_norm": 19.935190350579614, | |
| "learning_rate": 3.7181846109031e-07, | |
| "logits/chosen": -2.232686996459961, | |
| "logits/rejected": -2.187178134918213, | |
| "logps/chosen": -160.58908081054688, | |
| "logps/rejected": -191.36898803710938, | |
| "loss": 0.2, | |
| "rewards/accuracies": 0.918749988079071, | |
| "rewards/chosen": -0.7336459159851074, | |
| "rewards/margins": 2.8751566410064697, | |
| "rewards/rejected": -3.608802318572998, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 1.875, | |
| "grad_norm": 22.85505851727533, | |
| "learning_rate": 3.353746613749093e-07, | |
| "logits/chosen": -2.229166030883789, | |
| "logits/rejected": -2.2128987312316895, | |
| "logps/chosen": -171.8442840576172, | |
| "logps/rejected": -203.36578369140625, | |
| "loss": 0.1983, | |
| "rewards/accuracies": 0.9312499761581421, | |
| "rewards/chosen": -0.3919256329536438, | |
| "rewards/margins": 3.1109907627105713, | |
| "rewards/rejected": -3.5029163360595703, | |
| "step": 135 | |
| }, | |
| { | |
| "epoch": 1.9444444444444444, | |
| "grad_norm": 20.994239018606788, | |
| "learning_rate": 2.9988759642186093e-07, | |
| "logits/chosen": -2.2110824584960938, | |
| "logits/rejected": -2.1866531372070312, | |
| "logps/chosen": -160.40597534179688, | |
| "logps/rejected": -193.88641357421875, | |
| "loss": 0.1803, | |
| "rewards/accuracies": 0.90625, | |
| "rewards/chosen": -0.6127753257751465, | |
| "rewards/margins": 3.024134397506714, | |
| "rewards/rejected": -3.6369099617004395, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 2.013888888888889, | |
| "grad_norm": 18.007789953128686, | |
| "learning_rate": 2.655635024578483e-07, | |
| "logits/chosen": -2.2402732372283936, | |
| "logits/rejected": -2.176499843597412, | |
| "logps/chosen": -165.99827575683594, | |
| "logps/rejected": -203.1928253173828, | |
| "loss": 0.1666, | |
| "rewards/accuracies": 0.96875, | |
| "rewards/chosen": -0.5927726030349731, | |
| "rewards/margins": 3.829090118408203, | |
| "rewards/rejected": -4.421862602233887, | |
| "step": 145 | |
| }, | |
| { | |
| "epoch": 2.0833333333333335, | |
| "grad_norm": 17.845221787073083, | |
| "learning_rate": 2.3260185700046292e-07, | |
| "logits/chosen": -2.213283061981201, | |
| "logits/rejected": -2.1884255409240723, | |
| "logps/chosen": -163.85293579101562, | |
| "logps/rejected": -204.19644165039062, | |
| "loss": 0.105, | |
| "rewards/accuracies": 0.981249988079071, | |
| "rewards/chosen": -0.7582882046699524, | |
| "rewards/margins": 3.970874786376953, | |
| "rewards/rejected": -4.729162693023682, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.0833333333333335, | |
| "eval_logits/chosen": -2.2022910118103027, | |
| "eval_logits/rejected": -2.163472890853882, | |
| "eval_logps/chosen": -181.17343139648438, | |
| "eval_logps/rejected": -200.37542724609375, | |
| "eval_loss": 0.5275049805641174, | |
| "eval_rewards/accuracies": 0.73828125, | |
| "eval_rewards/chosen": -2.336512565612793, | |
| "eval_rewards/margins": 1.6323777437210083, | |
| "eval_rewards/rejected": -3.9688901901245117, | |
| "eval_runtime": 128.4507, | |
| "eval_samples_per_second": 15.905, | |
| "eval_steps_per_second": 0.249, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 2.1527777777777777, | |
| "grad_norm": 16.5358223115486, | |
| "learning_rate": 2.0119421957691218e-07, | |
| "logits/chosen": -2.2433078289031982, | |
| "logits/rejected": -2.192546844482422, | |
| "logps/chosen": -183.214599609375, | |
| "logps/rejected": -228.466064453125, | |
| "loss": 0.0922, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -1.2263261079788208, | |
| "rewards/margins": 4.63826847076416, | |
| "rewards/rejected": -5.86459493637085, | |
| "step": 155 | |
| }, | |
| { | |
| "epoch": 2.2222222222222223, | |
| "grad_norm": 16.243253588165253, | |
| "learning_rate": 1.7152311845883094e-07, | |
| "logits/chosen": -2.2243053913116455, | |
| "logits/rejected": -2.1990137100219727, | |
| "logps/chosen": -183.52928161621094, | |
| "logps/rejected": -225.4740753173828, | |
| "loss": 0.0882, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": -1.0933525562286377, | |
| "rewards/margins": 4.847748756408691, | |
| "rewards/rejected": -5.941100120544434, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 2.2916666666666665, | |
| "grad_norm": 14.143249761494847, | |
| "learning_rate": 1.4376098988303404e-07, | |
| "logits/chosen": -2.265902042388916, | |
| "logits/rejected": -2.2371320724487305, | |
| "logps/chosen": -178.1345977783203, | |
| "logps/rejected": -228.50906372070312, | |
| "loss": 0.0853, | |
| "rewards/accuracies": 0.9750000238418579, | |
| "rewards/chosen": -1.1710445880889893, | |
| "rewards/margins": 4.946959018707275, | |
| "rewards/rejected": -6.1180033683776855, | |
| "step": 165 | |
| }, | |
| { | |
| "epoch": 2.361111111111111, | |
| "grad_norm": 17.773700067929525, | |
| "learning_rate": 1.1806917592302761e-07, | |
| "logits/chosen": -2.2693705558776855, | |
| "logits/rejected": -2.2570414543151855, | |
| "logps/chosen": -169.011474609375, | |
| "logps/rejected": -220.2297821044922, | |
| "loss": 0.0769, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -1.048015832901001, | |
| "rewards/margins": 4.869006633758545, | |
| "rewards/rejected": -5.917021751403809, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 2.4305555555555554, | |
| "grad_norm": 16.131832286323938, | |
| "learning_rate": 9.459698683523204e-08, | |
| "logits/chosen": -2.2916018962860107, | |
| "logits/rejected": -2.2644224166870117, | |
| "logps/chosen": -179.8535919189453, | |
| "logps/rejected": -232.84213256835938, | |
| "loss": 0.0838, | |
| "rewards/accuracies": 0.956250011920929, | |
| "rewards/chosen": -1.5861546993255615, | |
| "rewards/margins": 4.993525981903076, | |
| "rewards/rejected": -6.579680442810059, | |
| "step": 175 | |
| }, | |
| { | |
| "epoch": 2.5, | |
| "grad_norm": 25.688615524154926, | |
| "learning_rate": 7.348083332917926e-08, | |
| "logits/chosen": -2.3048667907714844, | |
| "logits/rejected": -2.2622156143188477, | |
| "logps/chosen": -187.1690216064453, | |
| "logps/rejected": -231.0559844970703, | |
| "loss": 0.1002, | |
| "rewards/accuracies": 0.9125000238418579, | |
| "rewards/chosen": -1.6769403219223022, | |
| "rewards/margins": 4.427250862121582, | |
| "rewards/rejected": -6.104191780090332, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 2.5694444444444446, | |
| "grad_norm": 17.118168827295822, | |
| "learning_rate": 5.484343380457124e-08, | |
| "logits/chosen": -2.2761294841766357, | |
| "logits/rejected": -2.241340160369873, | |
| "logps/chosen": -165.86204528808594, | |
| "logps/rejected": -211.8990936279297, | |
| "loss": 0.0951, | |
| "rewards/accuracies": 0.981249988079071, | |
| "rewards/chosen": -1.2597142457962036, | |
| "rewards/margins": 4.591185092926025, | |
| "rewards/rejected": -5.850900173187256, | |
| "step": 185 | |
| }, | |
| { | |
| "epoch": 2.638888888888889, | |
| "grad_norm": 14.646194270585417, | |
| "learning_rate": 3.879310116241041e-08, | |
| "logits/chosen": -2.2710745334625244, | |
| "logits/rejected": -2.2402360439300537, | |
| "logps/chosen": -171.69906616210938, | |
| "logps/rejected": -220.8769073486328, | |
| "loss": 0.074, | |
| "rewards/accuracies": 0.987500011920929, | |
| "rewards/chosen": -0.8112651705741882, | |
| "rewards/margins": 4.888184547424316, | |
| "rewards/rejected": -5.699450492858887, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 2.7083333333333335, | |
| "grad_norm": 14.441833627511247, | |
| "learning_rate": 2.5423113334966218e-08, | |
| "logits/chosen": -2.294968605041504, | |
| "logits/rejected": -2.231370449066162, | |
| "logps/chosen": -172.13613891601562, | |
| "logps/rejected": -215.5753631591797, | |
| "loss": 0.081, | |
| "rewards/accuracies": 0.981249988079071, | |
| "rewards/chosen": -0.8212493658065796, | |
| "rewards/margins": 4.537470817565918, | |
| "rewards/rejected": -5.358720302581787, | |
| "step": 195 | |
| }, | |
| { | |
| "epoch": 2.7777777777777777, | |
| "grad_norm": 17.8264257139916, | |
| "learning_rate": 1.4811171192794624e-08, | |
| "logits/chosen": -2.2626068592071533, | |
| "logits/rejected": -2.2442564964294434, | |
| "logps/chosen": -178.3024139404297, | |
| "logps/rejected": -224.14501953125, | |
| "loss": 0.0911, | |
| "rewards/accuracies": 0.96875, | |
| "rewards/chosen": -1.1280481815338135, | |
| "rewards/margins": 4.719998359680176, | |
| "rewards/rejected": -5.848046779632568, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 2.7777777777777777, | |
| "eval_logits/chosen": -2.256969690322876, | |
| "eval_logits/rejected": -2.2202577590942383, | |
| "eval_logps/chosen": -183.34054565429688, | |
| "eval_logps/rejected": -202.82171630859375, | |
| "eval_loss": 0.5503418445587158, | |
| "eval_rewards/accuracies": 0.72265625, | |
| "eval_rewards/chosen": -2.5532219409942627, | |
| "eval_rewards/margins": 1.6602981090545654, | |
| "eval_rewards/rejected": -4.213520050048828, | |
| "eval_runtime": 128.2168, | |
| "eval_samples_per_second": 15.934, | |
| "eval_steps_per_second": 0.25, | |
| "step": 200 | |
| } | |
| ], | |
| "logging_steps": 5, | |
| "max_steps": 216, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 3, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 2358113407598592.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |