ziansu's picture
Training in progress, step 750, checkpoint
e063eb2 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.6054490413723511,
"eval_steps": 50,
"global_step": 750,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.008072653884964682,
"grad_norm": 0.04926518723368645,
"learning_rate": 4.999451708687114e-06,
"logits/chosen": 14.542106628417969,
"logits/rejected": 14.864250183105469,
"logps/chosen": -0.2809702754020691,
"logps/rejected": -0.3013763725757599,
"loss": 0.9221,
"rewards/accuracies": 0.42500001192092896,
"rewards/chosen": -0.421455442905426,
"rewards/margins": 0.03060910664498806,
"rewards/rejected": -0.452064573764801,
"step": 10
},
{
"epoch": 0.016145307769929364,
"grad_norm": 0.056249432265758514,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": 14.614748001098633,
"logits/rejected": 15.259109497070312,
"logps/chosen": -0.2828002870082855,
"logps/rejected": -0.3477819561958313,
"loss": 0.916,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.4242004454135895,
"rewards/margins": 0.09747247397899628,
"rewards/rejected": -0.5216729044914246,
"step": 20
},
{
"epoch": 0.024217961654894045,
"grad_norm": 0.0664869099855423,
"learning_rate": 4.9950668210706795e-06,
"logits/chosen": 14.391204833984375,
"logits/rejected": 14.82734203338623,
"logps/chosen": -0.287629634141922,
"logps/rejected": -0.3329126834869385,
"loss": 0.914,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.4314444661140442,
"rewards/margins": 0.06792456656694412,
"rewards/rejected": -0.4993689954280853,
"step": 30
},
{
"epoch": 0.03229061553985873,
"grad_norm": 0.055584829300642014,
"learning_rate": 4.9912321481237616e-06,
"logits/chosen": 13.93278980255127,
"logits/rejected": 14.886846542358398,
"logps/chosen": -0.28155946731567383,
"logps/rejected": -0.3678051829338074,
"loss": 0.9273,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.42233920097351074,
"rewards/margins": 0.12936851382255554,
"rewards/rejected": -0.5517078042030334,
"step": 40
},
{
"epoch": 0.04036326942482341,
"grad_norm": 0.07997103035449982,
"learning_rate": 4.986304738420684e-06,
"logits/chosen": 14.172693252563477,
"logits/rejected": 14.742494583129883,
"logps/chosen": -0.28225988149642944,
"logps/rejected": -0.34329456090927124,
"loss": 0.9081,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.42338982224464417,
"rewards/margins": 0.0915520042181015,
"rewards/rejected": -0.5149418115615845,
"step": 50
},
{
"epoch": 0.04036326942482341,
"eval_logits/chosen": 14.200166702270508,
"eval_logits/rejected": 14.817726135253906,
"eval_logps/chosen": -0.2625390887260437,
"eval_logps/rejected": -0.3458769917488098,
"eval_loss": 0.9080610275268555,
"eval_rewards/accuracies": 0.5544554591178894,
"eval_rewards/chosen": -0.39380866289138794,
"eval_rewards/margins": 0.1250067949295044,
"eval_rewards/rejected": -0.5188154578208923,
"eval_runtime": 29.8098,
"eval_samples_per_second": 26.87,
"eval_steps_per_second": 3.388,
"step": 50
},
{
"epoch": 0.04843592330978809,
"grad_norm": 0.06322095543146133,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": 14.008010864257812,
"logits/rejected": 14.975939750671387,
"logps/chosen": -0.2697351574897766,
"logps/rejected": -0.3445274233818054,
"loss": 0.9195,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.40460270643234253,
"rewards/margins": 0.1121884360909462,
"rewards/rejected": -0.5167912244796753,
"step": 60
},
{
"epoch": 0.056508577194752774,
"grad_norm": 0.12975476682186127,
"learning_rate": 4.973180832407471e-06,
"logits/chosen": 14.213134765625,
"logits/rejected": 14.889978408813477,
"logps/chosen": -0.2839818596839905,
"logps/rejected": -0.3382417559623718,
"loss": 0.9241,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.4259727895259857,
"rewards/margins": 0.08138985931873322,
"rewards/rejected": -0.5073626637458801,
"step": 70
},
{
"epoch": 0.06458123107971746,
"grad_norm": 0.08352109789848328,
"learning_rate": 4.964990092676263e-06,
"logits/chosen": 14.395146369934082,
"logits/rejected": 14.900177001953125,
"logps/chosen": -0.27067264914512634,
"logps/rejected": -0.3516673743724823,
"loss": 0.9251,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.4060089588165283,
"rewards/margins": 0.12149210274219513,
"rewards/rejected": -0.5275009870529175,
"step": 80
},
{
"epoch": 0.07265388496468214,
"grad_norm": 0.0793827548623085,
"learning_rate": 4.9557181268217225e-06,
"logits/chosen": 13.905145645141602,
"logits/rejected": 14.897878646850586,
"logps/chosen": -0.26814645528793335,
"logps/rejected": -0.38061466813087463,
"loss": 0.9055,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.4022197127342224,
"rewards/margins": 0.16870227456092834,
"rewards/rejected": -0.5709219574928284,
"step": 90
},
{
"epoch": 0.08072653884964683,
"grad_norm": 0.09212008118629456,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": 14.11627197265625,
"logits/rejected": 14.571977615356445,
"logps/chosen": -0.3130624294281006,
"logps/rejected": -0.35073375701904297,
"loss": 0.911,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.46959367394447327,
"rewards/margins": 0.05650699883699417,
"rewards/rejected": -0.5261006951332092,
"step": 100
},
{
"epoch": 0.08072653884964683,
"eval_logits/chosen": 13.845876693725586,
"eval_logits/rejected": 14.490789413452148,
"eval_logps/chosen": -0.2531408667564392,
"eval_logps/rejected": -0.3464036285877228,
"eval_loss": 0.8986235857009888,
"eval_rewards/accuracies": 0.5544554591178894,
"eval_rewards/chosen": -0.3797112703323364,
"eval_rewards/margins": 0.13989417254924774,
"eval_rewards/rejected": -0.5196054577827454,
"eval_runtime": 29.0914,
"eval_samples_per_second": 27.534,
"eval_steps_per_second": 3.472,
"step": 100
},
{
"epoch": 0.08879919273461151,
"grad_norm": 0.08554862439632416,
"learning_rate": 4.933947257182901e-06,
"logits/chosen": 13.416229248046875,
"logits/rejected": 14.582674026489258,
"logps/chosen": -0.2523443102836609,
"logps/rejected": -0.38751405477523804,
"loss": 0.8993,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.3785164952278137,
"rewards/margins": 0.20275457203388214,
"rewards/rejected": -0.5812710523605347,
"step": 110
},
{
"epoch": 0.09687184661957618,
"grad_norm": 0.29209578037261963,
"learning_rate": 4.921457902821578e-06,
"logits/chosen": 13.874654769897461,
"logits/rejected": 14.423624992370605,
"logps/chosen": -0.27300480008125305,
"logps/rejected": -0.3526575267314911,
"loss": 0.8942,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.40950721502304077,
"rewards/margins": 0.11947910487651825,
"rewards/rejected": -0.5289863348007202,
"step": 120
},
{
"epoch": 0.10494450050454086,
"grad_norm": 0.1028478816151619,
"learning_rate": 4.907906416994146e-06,
"logits/chosen": 14.030723571777344,
"logits/rejected": 14.711235046386719,
"logps/chosen": -0.27410784363746643,
"logps/rejected": -0.3665519058704376,
"loss": 0.8922,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.41116175055503845,
"rewards/margins": 0.1386660784482956,
"rewards/rejected": -0.5498278737068176,
"step": 130
},
{
"epoch": 0.11301715438950555,
"grad_norm": 0.08459590375423431,
"learning_rate": 4.893298743830168e-06,
"logits/chosen": 13.477182388305664,
"logits/rejected": 13.919464111328125,
"logps/chosen": -0.2495063841342926,
"logps/rejected": -0.3594816029071808,
"loss": 0.8957,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.3742595613002777,
"rewards/margins": 0.16496284306049347,
"rewards/rejected": -0.5392224192619324,
"step": 140
},
{
"epoch": 0.12108980827447023,
"grad_norm": 0.12467797845602036,
"learning_rate": 4.8776412907378845e-06,
"logits/chosen": 12.801656723022461,
"logits/rejected": 13.564155578613281,
"logps/chosen": -0.2779986262321472,
"logps/rejected": -0.33566445112228394,
"loss": 0.892,
"rewards/accuracies": 0.4375,
"rewards/chosen": -0.4169979691505432,
"rewards/margins": 0.08649872243404388,
"rewards/rejected": -0.5034967064857483,
"step": 150
},
{
"epoch": 0.12108980827447023,
"eval_logits/chosen": 12.169118881225586,
"eval_logits/rejected": 13.019683837890625,
"eval_logps/chosen": -0.25547870993614197,
"eval_logps/rejected": -0.37277930974960327,
"eval_loss": 0.8789658546447754,
"eval_rewards/accuracies": 0.5742574334144592,
"eval_rewards/chosen": -0.38321802020072937,
"eval_rewards/margins": 0.17595094442367554,
"eval_rewards/rejected": -0.5591689944267273,
"eval_runtime": 29.0961,
"eval_samples_per_second": 27.529,
"eval_steps_per_second": 3.471,
"step": 150
},
{
"epoch": 0.12916246215943492,
"grad_norm": 0.1712370663881302,
"learning_rate": 4.860940925593703e-06,
"logits/chosen": 11.972528457641602,
"logits/rejected": 12.684088706970215,
"logps/chosen": -0.28340521454811096,
"logps/rejected": -0.3864063024520874,
"loss": 0.8789,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.42510780692100525,
"rewards/margins": 0.15450166165828705,
"rewards/rejected": -0.5796095132827759,
"step": 160
},
{
"epoch": 0.13723511604439959,
"grad_norm": 0.21750673651695251,
"learning_rate": 4.84320497372973e-06,
"logits/chosen": 11.93881607055664,
"logits/rejected": 12.458230972290039,
"logps/chosen": -0.274599552154541,
"logps/rejected": -0.3882916271686554,
"loss": 0.8857,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.4118993282318115,
"rewards/margins": 0.1705380380153656,
"rewards/rejected": -0.5824374556541443,
"step": 170
},
{
"epoch": 0.14530776992936428,
"grad_norm": 0.16000741720199585,
"learning_rate": 4.824441214720629e-06,
"logits/chosen": 9.631464958190918,
"logits/rejected": 10.772969245910645,
"logps/chosen": -0.287865549325943,
"logps/rejected": -0.4817379415035248,
"loss": 0.8699,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.4317983090877533,
"rewards/margins": 0.29080861806869507,
"rewards/rejected": -0.722606897354126,
"step": 180
},
{
"epoch": 0.15338042381432895,
"grad_norm": 0.46293890476226807,
"learning_rate": 4.804657878971252e-06,
"logits/chosen": 9.081937789916992,
"logits/rejected": 10.024572372436523,
"logps/chosen": -0.2941994071006775,
"logps/rejected": -0.4772109389305115,
"loss": 0.8565,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.44129911065101624,
"rewards/margins": 0.2745172679424286,
"rewards/rejected": -0.7158163785934448,
"step": 190
},
{
"epoch": 0.16145307769929365,
"grad_norm": 0.24658434092998505,
"learning_rate": 4.783863644106502e-06,
"logits/chosen": 7.767125606536865,
"logits/rejected": 8.009145736694336,
"logps/chosen": -0.32877305150032043,
"logps/rejected": -0.47733697295188904,
"loss": 0.8482,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.49315959215164185,
"rewards/margins": 0.2228458821773529,
"rewards/rejected": -0.7160054445266724,
"step": 200
},
{
"epoch": 0.16145307769929365,
"eval_logits/chosen": 6.716187953948975,
"eval_logits/rejected": 7.248146057128906,
"eval_logps/chosen": -0.30652713775634766,
"eval_logps/rejected": -0.5056277513504028,
"eval_loss": 0.8243693113327026,
"eval_rewards/accuracies": 0.603960394859314,
"eval_rewards/chosen": -0.4597907066345215,
"eval_rewards/margins": 0.2986510097980499,
"eval_rewards/rejected": -0.7584417462348938,
"eval_runtime": 29.0989,
"eval_samples_per_second": 27.527,
"eval_steps_per_second": 3.471,
"step": 200
},
{
"epoch": 0.16952573158425832,
"grad_norm": 0.40796443819999695,
"learning_rate": 4.762067631165049e-06,
"logits/chosen": 5.932644844055176,
"logits/rejected": 6.521953582763672,
"logps/chosen": -0.32976508140563965,
"logps/rejected": -0.5628186464309692,
"loss": 0.7988,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.4946475923061371,
"rewards/margins": 0.3495803475379944,
"rewards/rejected": -0.8442279696464539,
"step": 210
},
{
"epoch": 0.17759838546922302,
"grad_norm": 0.42468318343162537,
"learning_rate": 4.7392794005985324e-06,
"logits/chosen": 6.069305896759033,
"logits/rejected": 5.8950395584106445,
"logps/chosen": -0.37205421924591064,
"logps/rejected": -0.6190425753593445,
"loss": 0.796,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5580812692642212,
"rewards/margins": 0.3704826235771179,
"rewards/rejected": -0.9285639524459839,
"step": 220
},
{
"epoch": 0.1856710393541877,
"grad_norm": 0.4138280153274536,
"learning_rate": 4.715508948078037e-06,
"logits/chosen": 3.5271706581115723,
"logits/rejected": 3.363534927368164,
"logps/chosen": -0.41677650809288025,
"logps/rejected": -0.7121980786323547,
"loss": 0.7457,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.6251648664474487,
"rewards/margins": 0.4431324005126953,
"rewards/rejected": -1.0682971477508545,
"step": 230
},
{
"epoch": 0.19374369323915236,
"grad_norm": 1.4327284097671509,
"learning_rate": 4.690766700109659e-06,
"logits/chosen": 2.296924114227295,
"logits/rejected": 1.6135867834091187,
"logps/chosen": -0.506227970123291,
"logps/rejected": -0.8080593943595886,
"loss": 0.7453,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7593418955802917,
"rewards/margins": 0.4527471661567688,
"rewards/rejected": -1.21208918094635,
"step": 240
},
{
"epoch": 0.20181634712411706,
"grad_norm": 0.4413074553012848,
"learning_rate": 4.665063509461098e-06,
"logits/chosen": 1.4450469017028809,
"logits/rejected": 0.40727120637893677,
"logps/chosen": -0.4749869406223297,
"logps/rejected": -0.9967275857925415,
"loss": 0.704,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.712480366230011,
"rewards/margins": 0.7826108932495117,
"rewards/rejected": -1.4950913190841675,
"step": 250
},
{
"epoch": 0.20181634712411706,
"eval_logits/chosen": 1.4614256620407104,
"eval_logits/rejected": 0.6895493865013123,
"eval_logps/chosen": -0.5038881301879883,
"eval_logps/rejected": -0.973581850528717,
"eval_loss": 0.6982013583183289,
"eval_rewards/accuracies": 0.6138613820075989,
"eval_rewards/chosen": -0.7558321952819824,
"eval_rewards/margins": 0.7045406699180603,
"eval_rewards/rejected": -1.460372805595398,
"eval_runtime": 29.0993,
"eval_samples_per_second": 27.526,
"eval_steps_per_second": 3.471,
"step": 250
},
{
"epoch": 0.20988900100908173,
"grad_norm": 0.734251856803894,
"learning_rate": 4.638410650401267e-06,
"logits/chosen": 2.4274039268493652,
"logits/rejected": 1.498230218887329,
"logps/chosen": -0.5549699068069458,
"logps/rejected": -0.9348627328872681,
"loss": 0.7149,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.8324548602104187,
"rewards/margins": 0.5698392987251282,
"rewards/rejected": -1.4022941589355469,
"step": 260
},
{
"epoch": 0.21796165489404642,
"grad_norm": 0.4445085823535919,
"learning_rate": 4.610819813755038e-06,
"logits/chosen": 2.042858600616455,
"logits/rejected": 1.1695036888122559,
"logps/chosen": -0.5573975443840027,
"logps/rejected": -1.15065598487854,
"loss": 0.6811,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.8360962867736816,
"rewards/margins": 0.889887809753418,
"rewards/rejected": -1.72598397731781,
"step": 270
},
{
"epoch": 0.2260343087790111,
"grad_norm": 0.5875476598739624,
"learning_rate": 4.582303101775249e-06,
"logits/chosen": 1.3029518127441406,
"logits/rejected": 0.3808741271495819,
"logps/chosen": -0.6656169891357422,
"logps/rejected": -1.5305824279785156,
"loss": 0.637,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.9984253644943237,
"rewards/margins": 1.2974482774734497,
"rewards/rejected": -2.2958736419677734,
"step": 280
},
{
"epoch": 0.2341069626639758,
"grad_norm": 0.3814420998096466,
"learning_rate": 4.55287302283426e-06,
"logits/chosen": 2.0200212001800537,
"logits/rejected": 1.2681838274002075,
"logps/chosen": -0.6776013970375061,
"logps/rejected": -1.3369777202606201,
"loss": 0.63,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.016402006149292,
"rewards/margins": 0.9890643358230591,
"rewards/rejected": -2.0054664611816406,
"step": 290
},
{
"epoch": 0.24217961654894046,
"grad_norm": 0.6442322731018066,
"learning_rate": 4.522542485937369e-06,
"logits/chosen": 1.6268196105957031,
"logits/rejected": 0.4954712390899658,
"logps/chosen": -0.8117648363113403,
"logps/rejected": -1.7705228328704834,
"loss": 0.6104,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.2176473140716553,
"rewards/margins": 1.4381370544433594,
"rewards/rejected": -2.6557843685150146,
"step": 300
},
{
"epoch": 0.24217961654894046,
"eval_logits/chosen": 1.4003607034683228,
"eval_logits/rejected": 0.46628010272979736,
"eval_logps/chosen": -0.7612115740776062,
"eval_logps/rejected": -1.6895866394042969,
"eval_loss": 0.5781419277191162,
"eval_rewards/accuracies": 0.6534653306007385,
"eval_rewards/chosen": -1.141817331314087,
"eval_rewards/margins": 1.392562747001648,
"eval_rewards/rejected": -2.5343799591064453,
"eval_runtime": 29.102,
"eval_samples_per_second": 27.524,
"eval_steps_per_second": 3.471,
"step": 300
},
{
"epoch": 0.25025227043390513,
"grad_norm": 0.7889758944511414,
"learning_rate": 4.491324795060491e-06,
"logits/chosen": 1.1702089309692383,
"logits/rejected": 0.354276180267334,
"logps/chosen": -0.8091352581977844,
"logps/rejected": -1.944819450378418,
"loss": 0.5807,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.2137027978897095,
"rewards/margins": 1.7035261392593384,
"rewards/rejected": -2.9172286987304688,
"step": 310
},
{
"epoch": 0.25832492431886983,
"grad_norm": 1.410145878791809,
"learning_rate": 4.4592336433146e-06,
"logits/chosen": 2.0981459617614746,
"logits/rejected": 1.2289329767227173,
"logps/chosen": -0.8151634931564331,
"logps/rejected": -1.977259874343872,
"loss": 0.5007,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.222745418548584,
"rewards/margins": 1.7431443929672241,
"rewards/rejected": -2.9658896923065186,
"step": 320
},
{
"epoch": 0.26639757820383453,
"grad_norm": 0.5564689040184021,
"learning_rate": 4.426283106939474e-06,
"logits/chosen": 1.7921768426895142,
"logits/rejected": 0.7705962061882019,
"logps/chosen": -0.9244564771652222,
"logps/rejected": -2.5274672508239746,
"loss": 0.4757,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.3866846561431885,
"rewards/margins": 2.4045166969299316,
"rewards/rejected": -3.791201114654541,
"step": 330
},
{
"epoch": 0.27447023208879917,
"grad_norm": 0.7554243803024292,
"learning_rate": 4.3924876391293915e-06,
"logits/chosen": 1.947997808456421,
"logits/rejected": 0.9661592245101929,
"logps/chosen": -0.9154227375984192,
"logps/rejected": -3.0766491889953613,
"loss": 0.4581,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -1.3731342554092407,
"rewards/margins": 3.24183988571167,
"rewards/rejected": -4.614974021911621,
"step": 340
},
{
"epoch": 0.28254288597376387,
"grad_norm": 2.936426877975464,
"learning_rate": 4.357862063693486e-06,
"logits/chosen": 1.2846364974975586,
"logits/rejected": 0.9968118667602539,
"logps/chosen": -1.0272481441497803,
"logps/rejected": -2.6980624198913574,
"loss": 0.4989,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.54087233543396,
"rewards/margins": 2.506221294403076,
"rewards/rejected": -4.047093868255615,
"step": 350
},
{
"epoch": 0.28254288597376387,
"eval_logits/chosen": 1.841333031654358,
"eval_logits/rejected": 1.1651691198349,
"eval_logps/chosen": -0.9644113183021545,
"eval_logps/rejected": -2.631535053253174,
"eval_loss": 0.44154587388038635,
"eval_rewards/accuracies": 0.6732673048973083,
"eval_rewards/chosen": -1.4466170072555542,
"eval_rewards/margins": 2.500684976577759,
"eval_rewards/rejected": -3.9473025798797607,
"eval_runtime": 29.1062,
"eval_samples_per_second": 27.52,
"eval_steps_per_second": 3.47,
"step": 350
},
{
"epoch": 0.29061553985872857,
"grad_norm": 0.49919044971466064,
"learning_rate": 4.322421568553529e-06,
"logits/chosen": 1.9874728918075562,
"logits/rejected": 1.464611291885376,
"logps/chosen": -0.9963932037353516,
"logps/rejected": -2.8804218769073486,
"loss": 0.4373,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.4945898056030273,
"rewards/margins": 2.8260433673858643,
"rewards/rejected": -4.320633411407471,
"step": 360
},
{
"epoch": 0.29868819374369326,
"grad_norm": 1.4991668462753296,
"learning_rate": 4.286181699082008e-06,
"logits/chosen": 2.5421957969665527,
"logits/rejected": 1.941922903060913,
"logps/chosen": -1.0214306116104126,
"logps/rejected": -3.2893283367156982,
"loss": 0.438,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.5321458578109741,
"rewards/margins": 3.4018466472625732,
"rewards/rejected": -4.933992862701416,
"step": 370
},
{
"epoch": 0.3067608476286579,
"grad_norm": 1.933100938796997,
"learning_rate": 4.249158351283414e-06,
"logits/chosen": 1.80439031124115,
"logits/rejected": 1.4421275854110718,
"logps/chosen": -1.1270357370376587,
"logps/rejected": -3.17500901222229,
"loss": 0.4542,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.6905533075332642,
"rewards/margins": 3.071959972381592,
"rewards/rejected": -4.762513160705566,
"step": 380
},
{
"epoch": 0.3148335015136226,
"grad_norm": 3.017254590988159,
"learning_rate": 4.211367764821722e-06,
"logits/chosen": 3.4090209007263184,
"logits/rejected": 2.784639835357666,
"logps/chosen": -1.1713608503341675,
"logps/rejected": -3.326244831085205,
"loss": 0.4816,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.757041335105896,
"rewards/margins": 3.232325792312622,
"rewards/rejected": -4.9893670082092285,
"step": 390
},
{
"epoch": 0.3229061553985873,
"grad_norm": 0.5897337198257446,
"learning_rate": 4.172826515897146e-06,
"logits/chosen": 3.6071503162384033,
"logits/rejected": 2.738395929336548,
"logps/chosen": -1.133429765701294,
"logps/rejected": -3.509474277496338,
"loss": 0.4431,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.7001447677612305,
"rewards/margins": 3.5640671253204346,
"rewards/rejected": -5.264212131500244,
"step": 400
},
{
"epoch": 0.3229061553985873,
"eval_logits/chosen": 2.893691062927246,
"eval_logits/rejected": 2.259718656539917,
"eval_logps/chosen": -1.0705492496490479,
"eval_logps/rejected": -2.935904026031494,
"eval_loss": 0.4156961143016815,
"eval_rewards/accuracies": 0.6732673048973083,
"eval_rewards/chosen": -1.6058237552642822,
"eval_rewards/margins": 2.798032283782959,
"eval_rewards/rejected": -4.403855800628662,
"eval_runtime": 29.186,
"eval_samples_per_second": 27.445,
"eval_steps_per_second": 3.461,
"step": 400
},
{
"epoch": 0.33097880928355194,
"grad_norm": 2.219709634780884,
"learning_rate": 4.133551509975264e-06,
"logits/chosen": 2.8940415382385254,
"logits/rejected": 2.545257091522217,
"logps/chosen": -1.1774877309799194,
"logps/rejected": -3.4022269248962402,
"loss": 0.427,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.7662317752838135,
"rewards/margins": 3.3371081352233887,
"rewards/rejected": -5.1033406257629395,
"step": 410
},
{
"epoch": 0.33905146316851664,
"grad_norm": 0.9891649484634399,
"learning_rate": 4.093559974371725e-06,
"logits/chosen": 3.180427074432373,
"logits/rejected": 2.8354597091674805,
"logps/chosen": -1.2984545230865479,
"logps/rejected": -3.4927687644958496,
"loss": 0.4384,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.9476817846298218,
"rewards/margins": 3.291471481323242,
"rewards/rejected": -5.2391533851623535,
"step": 420
},
{
"epoch": 0.34712411705348134,
"grad_norm": 0.4516288936138153,
"learning_rate": 4.052869450695776e-06,
"logits/chosen": 4.026360511779785,
"logits/rejected": 3.3932456970214844,
"logps/chosen": -1.1195746660232544,
"logps/rejected": -3.4704582691192627,
"loss": 0.3934,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.6793619394302368,
"rewards/margins": 3.5263259410858154,
"rewards/rejected": -5.205687522888184,
"step": 430
},
{
"epoch": 0.35519677093844604,
"grad_norm": 0.5276560187339783,
"learning_rate": 4.011497787155938e-06,
"logits/chosen": 3.410902500152588,
"logits/rejected": 2.906996965408325,
"logps/chosen": -1.205904245376587,
"logps/rejected": -3.7434115409851074,
"loss": 0.4147,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.8088566064834595,
"rewards/margins": 3.806260585784912,
"rewards/rejected": -5.615117073059082,
"step": 440
},
{
"epoch": 0.3632694248234107,
"grad_norm": 0.4435446560382843,
"learning_rate": 3.969463130731183e-06,
"logits/chosen": 3.7616615295410156,
"logits/rejected": 3.4183363914489746,
"logps/chosen": -1.2180852890014648,
"logps/rejected": -3.865541934967041,
"loss": 0.4354,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.8271278142929077,
"rewards/margins": 3.9711856842041016,
"rewards/rejected": -5.798312664031982,
"step": 450
},
{
"epoch": 0.3632694248234107,
"eval_logits/chosen": 3.55381178855896,
"eval_logits/rejected": 3.01943039894104,
"eval_logps/chosen": -1.3239827156066895,
"eval_logps/rejected": -3.6477084159851074,
"eval_loss": 0.3905698359012604,
"eval_rewards/accuracies": 0.7029703259468079,
"eval_rewards/chosen": -1.9859741926193237,
"eval_rewards/margins": 3.485589027404785,
"eval_rewards/rejected": -5.471563816070557,
"eval_runtime": 29.0958,
"eval_samples_per_second": 27.53,
"eval_steps_per_second": 3.471,
"step": 450
},
{
"epoch": 0.3713420787083754,
"grad_norm": 0.8419796824455261,
"learning_rate": 3.92678391921108e-06,
"logits/chosen": 3.3496861457824707,
"logits/rejected": 3.235544204711914,
"logps/chosen": -1.4899990558624268,
"logps/rejected": -4.507351398468018,
"loss": 0.4023,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -2.2349984645843506,
"rewards/margins": 4.526029109954834,
"rewards/rejected": -6.7610273361206055,
"step": 460
},
{
"epoch": 0.3794147325933401,
"grad_norm": 1.2379052639007568,
"learning_rate": 3.88347887310836e-06,
"logits/chosen": 3.5375685691833496,
"logits/rejected": 3.34586763381958,
"logps/chosen": -1.5865886211395264,
"logps/rejected": -4.887473106384277,
"loss": 0.3979,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -2.379883289337158,
"rewards/margins": 4.9513258934021,
"rewards/rejected": -7.331210136413574,
"step": 470
},
{
"epoch": 0.3874873864783047,
"grad_norm": 1.9969083070755005,
"learning_rate": 3.839566987447492e-06,
"logits/chosen": 3.3116583824157715,
"logits/rejected": 2.9652373790740967,
"logps/chosen": -2.0900139808654785,
"logps/rejected": -4.921124458312988,
"loss": 0.3356,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -3.135021209716797,
"rewards/margins": 4.246665000915527,
"rewards/rejected": -7.381686210632324,
"step": 480
},
{
"epoch": 0.3955600403632694,
"grad_norm": 2.3405466079711914,
"learning_rate": 3.795067523432826e-06,
"logits/chosen": 2.542064666748047,
"logits/rejected": 2.503020763397217,
"logps/chosen": -2.4674277305603027,
"logps/rejected": -5.703408718109131,
"loss": 0.3193,
"rewards/accuracies": 0.875,
"rewards/chosen": -3.701141834259033,
"rewards/margins": 4.853971004486084,
"rewards/rejected": -8.555112838745117,
"step": 490
},
{
"epoch": 0.4036326942482341,
"grad_norm": 1.4798164367675781,
"learning_rate": 3.7500000000000005e-06,
"logits/chosen": 3.79949688911438,
"logits/rejected": 3.7085328102111816,
"logps/chosen": -2.6355605125427246,
"logps/rejected": -5.005209922790527,
"loss": 0.3103,
"rewards/accuracies": 0.8125,
"rewards/chosen": -3.953340530395508,
"rewards/margins": 3.554474353790283,
"rewards/rejected": -7.507814884185791,
"step": 500
},
{
"epoch": 0.4036326942482341,
"eval_logits/chosen": 3.31948184967041,
"eval_logits/rejected": 2.9104762077331543,
"eval_logps/chosen": -2.606774091720581,
"eval_logps/rejected": -5.2280120849609375,
"eval_loss": 0.3228474259376526,
"eval_rewards/accuracies": 0.9306930899620056,
"eval_rewards/chosen": -3.9101614952087402,
"eval_rewards/margins": 3.931856870651245,
"eval_rewards/rejected": -7.8420186042785645,
"eval_runtime": 29.0975,
"eval_samples_per_second": 27.528,
"eval_steps_per_second": 3.471,
"step": 500
},
{
"epoch": 0.4117053481331988,
"grad_norm": 1.4098985195159912,
"learning_rate": 3.7043841852542884e-06,
"logits/chosen": 2.69830584526062,
"logits/rejected": 2.7101688385009766,
"logps/chosen": -2.7831828594207764,
"logps/rejected": -5.963825225830078,
"loss": 0.2884,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -4.174774169921875,
"rewards/margins": 4.7709641456604,
"rewards/rejected": -8.945737838745117,
"step": 510
},
{
"epoch": 0.41977800201816345,
"grad_norm": 3.457231283187866,
"learning_rate": 3.658240087799655e-06,
"logits/chosen": 3.335204601287842,
"logits/rejected": 3.1935055255889893,
"logps/chosen": -2.66495943069458,
"logps/rejected": -5.2761640548706055,
"loss": 0.3093,
"rewards/accuracies": 0.875,
"rewards/chosen": -3.997438907623291,
"rewards/margins": 3.916806697845459,
"rewards/rejected": -7.91424560546875,
"step": 520
},
{
"epoch": 0.42785065590312815,
"grad_norm": 1.7744654417037964,
"learning_rate": 3.611587947962319e-06,
"logits/chosen": 2.725924015045166,
"logits/rejected": 2.605053186416626,
"logps/chosen": -2.5286645889282227,
"logps/rejected": -5.774144172668457,
"loss": 0.2972,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -3.792996883392334,
"rewards/margins": 4.868220329284668,
"rewards/rejected": -8.661216735839844,
"step": 530
},
{
"epoch": 0.43592330978809285,
"grad_norm": 2.863692283630371,
"learning_rate": 3.564448228912682e-06,
"logits/chosen": 2.304443120956421,
"logits/rejected": 2.4836020469665527,
"logps/chosen": -2.699148654937744,
"logps/rejected": -6.340291500091553,
"loss": 0.3136,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -4.048723220825195,
"rewards/margins": 5.461714267730713,
"rewards/rejected": -9.510437965393066,
"step": 540
},
{
"epoch": 0.4439959636730575,
"grad_norm": 4.6681623458862305,
"learning_rate": 3.516841607689501e-06,
"logits/chosen": 3.1207594871520996,
"logits/rejected": 2.947653293609619,
"logps/chosen": -3.0062317848205566,
"logps/rejected": -6.26816463470459,
"loss": 0.2822,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -4.509347438812256,
"rewards/margins": 4.892899513244629,
"rewards/rejected": -9.402246475219727,
"step": 550
},
{
"epoch": 0.4439959636730575,
"eval_logits/chosen": 2.879721164703369,
"eval_logits/rejected": 2.52551007270813,
"eval_logps/chosen": -2.6468751430511475,
"eval_logps/rejected": -5.636789321899414,
"eval_loss": 0.2939932942390442,
"eval_rewards/accuracies": 0.9306930899620056,
"eval_rewards/chosen": -3.9703128337860107,
"eval_rewards/margins": 4.4848713874816895,
"eval_rewards/rejected": -8.455184936523438,
"eval_runtime": 29.4025,
"eval_samples_per_second": 27.243,
"eval_steps_per_second": 3.435,
"step": 550
},
{
"epoch": 0.4520686175580222,
"grad_norm": 1.9975569248199463,
"learning_rate": 3.4687889661302577e-06,
"logits/chosen": 2.5885415077209473,
"logits/rejected": 2.5374302864074707,
"logps/chosen": -2.886643171310425,
"logps/rejected": -6.636415004730225,
"loss": 0.2736,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -4.329964637756348,
"rewards/margins": 5.624656677246094,
"rewards/rejected": -9.954621315002441,
"step": 560
},
{
"epoch": 0.4601412714429869,
"grad_norm": 3.3965606689453125,
"learning_rate": 3.4203113817116955e-06,
"logits/chosen": 2.0093374252319336,
"logits/rejected": 2.1064200401306152,
"logps/chosen": -3.251981735229492,
"logps/rejected": -6.749331474304199,
"loss": 0.2776,
"rewards/accuracies": 0.875,
"rewards/chosen": -4.877972602844238,
"rewards/margins": 5.246025085449219,
"rewards/rejected": -10.12399673461914,
"step": 570
},
{
"epoch": 0.4682139253279516,
"grad_norm": 2.7079927921295166,
"learning_rate": 3.3714301183045382e-06,
"logits/chosen": 3.916841506958008,
"logits/rejected": 3.524440288543701,
"logps/chosen": -3.304469347000122,
"logps/rejected": -7.1993279457092285,
"loss": 0.2547,
"rewards/accuracies": 0.9375,
"rewards/chosen": -4.956704139709473,
"rewards/margins": 5.842287540435791,
"rewards/rejected": -10.798992156982422,
"step": 580
},
{
"epoch": 0.47628657921291623,
"grad_norm": 7.504203796386719,
"learning_rate": 3.3221666168464584e-06,
"logits/chosen": 2.7350666522979736,
"logits/rejected": 2.2590534687042236,
"logps/chosen": -3.0954322814941406,
"logps/rejected": -6.782465934753418,
"loss": 0.2839,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -4.643147945404053,
"rewards/margins": 5.530551433563232,
"rewards/rejected": -10.173700332641602,
"step": 590
},
{
"epoch": 0.4843592330978809,
"grad_norm": 2.8209569454193115,
"learning_rate": 3.272542485937369e-06,
"logits/chosen": 3.7743606567382812,
"logits/rejected": 3.2480530738830566,
"logps/chosen": -3.196892261505127,
"logps/rejected": -6.42517614364624,
"loss": 0.3237,
"rewards/accuracies": 0.875,
"rewards/chosen": -4.7953386306762695,
"rewards/margins": 4.8424248695373535,
"rewards/rejected": -9.637763977050781,
"step": 600
},
{
"epoch": 0.4843592330978809,
"eval_logits/chosen": 2.6344597339630127,
"eval_logits/rejected": 2.46531081199646,
"eval_logps/chosen": -2.9877941608428955,
"eval_logps/rejected": -6.266376972198486,
"eval_loss": 0.2827170193195343,
"eval_rewards/accuracies": 0.9603960514068604,
"eval_rewards/chosen": -4.481690406799316,
"eval_rewards/margins": 4.917874336242676,
"eval_rewards/rejected": -9.399564743041992,
"eval_runtime": 29.0997,
"eval_samples_per_second": 27.526,
"eval_steps_per_second": 3.471,
"step": 600
},
{
"epoch": 0.4924318869828456,
"grad_norm": 7.072656631469727,
"learning_rate": 3.222579492361179e-06,
"logits/chosen": 3.2776103019714355,
"logits/rejected": 3.166206121444702,
"logps/chosen": -3.193716049194336,
"logps/rejected": -6.185642242431641,
"loss": 0.2986,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -4.790574073791504,
"rewards/margins": 4.487889289855957,
"rewards/rejected": -9.278463363647461,
"step": 610
},
{
"epoch": 0.5005045408678103,
"grad_norm": 2.7188704013824463,
"learning_rate": 3.1722995515381644e-06,
"logits/chosen": 3.189532518386841,
"logits/rejected": 2.9885735511779785,
"logps/chosen": -3.3461785316467285,
"logps/rejected": -6.720322608947754,
"loss": 0.2776,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -5.019268035888672,
"rewards/margins": 5.061214447021484,
"rewards/rejected": -10.080483436584473,
"step": 620
},
{
"epoch": 0.508577194752775,
"grad_norm": 2.6990387439727783,
"learning_rate": 3.121724717912138e-06,
"logits/chosen": 3.7751071453094482,
"logits/rejected": 3.406925916671753,
"logps/chosen": -3.3688464164733887,
"logps/rejected": -6.132623672485352,
"loss": 0.2688,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -5.053269386291504,
"rewards/margins": 4.145665168762207,
"rewards/rejected": -9.198934555053711,
"step": 630
},
{
"epoch": 0.5166498486377397,
"grad_norm": 1.964301347732544,
"learning_rate": 3.0708771752766397e-06,
"logits/chosen": 2.6602158546447754,
"logits/rejected": 2.162348508834839,
"logps/chosen": -2.884718894958496,
"logps/rejected": -6.869751930236816,
"loss": 0.2435,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -4.327078342437744,
"rewards/margins": 5.9775495529174805,
"rewards/rejected": -10.304628372192383,
"step": 640
},
{
"epoch": 0.5247225025227044,
"grad_norm": 3.201704740524292,
"learning_rate": 3.019779227044398e-06,
"logits/chosen": 3.83849835395813,
"logits/rejected": 3.5917961597442627,
"logps/chosen": -3.2030792236328125,
"logps/rejected": -6.711284637451172,
"loss": 0.2865,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -4.804618835449219,
"rewards/margins": 5.262308120727539,
"rewards/rejected": -10.066926956176758,
"step": 650
},
{
"epoch": 0.5247225025227044,
"eval_logits/chosen": 2.6539251804351807,
"eval_logits/rejected": 2.3508856296539307,
"eval_logps/chosen": -3.0629732608795166,
"eval_logps/rejected": -6.429784297943115,
"eval_loss": 0.25649499893188477,
"eval_rewards/accuracies": 0.9702970385551453,
"eval_rewards/chosen": -4.594459533691406,
"eval_rewards/margins": 5.0502166748046875,
"eval_rewards/rejected": -9.644676208496094,
"eval_runtime": 29.1031,
"eval_samples_per_second": 27.523,
"eval_steps_per_second": 3.47,
"step": 650
},
{
"epoch": 0.5327951564076691,
"grad_norm": 3.517667055130005,
"learning_rate": 2.9684532864643123e-06,
"logits/chosen": 2.804810047149658,
"logits/rejected": 2.8151614665985107,
"logps/chosen": -3.4792568683624268,
"logps/rejected": -7.08618688583374,
"loss": 0.243,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -5.2188849449157715,
"rewards/margins": 5.41039514541626,
"rewards/rejected": -10.629280090332031,
"step": 660
},
{
"epoch": 0.5408678102926338,
"grad_norm": 8.366011619567871,
"learning_rate": 2.9169218667902562e-06,
"logits/chosen": 2.1079938411712646,
"logits/rejected": 2.2689383029937744,
"logps/chosen": -3.3151912689208984,
"logps/rejected": -7.454249382019043,
"loss": 0.266,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -4.972787380218506,
"rewards/margins": 6.208587169647217,
"rewards/rejected": -11.181374549865723,
"step": 670
},
{
"epoch": 0.5489404641775983,
"grad_norm": 4.161479473114014,
"learning_rate": 2.8652075714060296e-06,
"logits/chosen": 2.1858832836151123,
"logits/rejected": 2.1688101291656494,
"logps/chosen": -3.3753318786621094,
"logps/rejected": -6.9990692138671875,
"loss": 0.2356,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -5.0629987716674805,
"rewards/margins": 5.435605525970459,
"rewards/rejected": -10.498603820800781,
"step": 680
},
{
"epoch": 0.557013118062563,
"grad_norm": 2.156621217727661,
"learning_rate": 2.813333083910761e-06,
"logits/chosen": 2.5004096031188965,
"logits/rejected": 2.289215087890625,
"logps/chosen": -3.711970567703247,
"logps/rejected": -7.685518741607666,
"loss": 0.2286,
"rewards/accuracies": 0.9375,
"rewards/chosen": -5.56795597076416,
"rewards/margins": 5.960321426391602,
"rewards/rejected": -11.528276443481445,
"step": 690
},
{
"epoch": 0.5650857719475277,
"grad_norm": 4.547532081604004,
"learning_rate": 2.761321158169134e-06,
"logits/chosen": 3.112332820892334,
"logits/rejected": 2.685454845428467,
"logps/chosen": -3.510646343231201,
"logps/rejected": -7.8125505447387695,
"loss": 0.2222,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -5.265969276428223,
"rewards/margins": 6.452856540679932,
"rewards/rejected": -11.718826293945312,
"step": 700
},
{
"epoch": 0.5650857719475277,
"eval_logits/chosen": 2.7667236328125,
"eval_logits/rejected": 2.4983408451080322,
"eval_logps/chosen": -3.292154550552368,
"eval_logps/rejected": -6.814427375793457,
"eval_loss": 0.24892139434814453,
"eval_rewards/accuracies": 0.9702970385551453,
"eval_rewards/chosen": -4.938231468200684,
"eval_rewards/margins": 5.2834086418151855,
"eval_rewards/rejected": -10.221639633178711,
"eval_runtime": 29.0862,
"eval_samples_per_second": 27.539,
"eval_steps_per_second": 3.472,
"step": 700
},
{
"epoch": 0.5731584258324924,
"grad_norm": 3.3842532634735107,
"learning_rate": 2.70919460833079e-06,
"logits/chosen": 2.6863622665405273,
"logits/rejected": 2.7381534576416016,
"logps/chosen": -3.1926321983337402,
"logps/rejected": -7.823538780212402,
"loss": 0.2153,
"rewards/accuracies": 0.9375,
"rewards/chosen": -4.788949012756348,
"rewards/margins": 6.946360111236572,
"rewards/rejected": -11.735308647155762,
"step": 710
},
{
"epoch": 0.5812310797174571,
"grad_norm": 2.0056378841400146,
"learning_rate": 2.6569762988232838e-06,
"logits/chosen": 2.938141345977783,
"logits/rejected": 2.8052525520324707,
"logps/chosen": -3.4339518547058105,
"logps/rejected": -7.353487968444824,
"loss": 0.2422,
"rewards/accuracies": 0.875,
"rewards/chosen": -5.150927543640137,
"rewards/margins": 5.879303932189941,
"rewards/rejected": -11.030232429504395,
"step": 720
},
{
"epoch": 0.5893037336024218,
"grad_norm": 2.9281482696533203,
"learning_rate": 2.604689134322999e-06,
"logits/chosen": 3.5540454387664795,
"logits/rejected": 3.6734695434570312,
"logps/chosen": -3.0852925777435303,
"logps/rejected": -6.869669437408447,
"loss": 0.2694,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -4.627939224243164,
"rewards/margins": 5.676565647125244,
"rewards/rejected": -10.304505348205566,
"step": 730
},
{
"epoch": 0.5973763874873865,
"grad_norm": 3.3841099739074707,
"learning_rate": 2.5523560497083927e-06,
"logits/chosen": 2.7171874046325684,
"logits/rejected": 2.4911468029022217,
"logps/chosen": -3.4943466186523438,
"logps/rejected": -7.486982822418213,
"loss": 0.2184,
"rewards/accuracies": 0.9375,
"rewards/chosen": -5.241519927978516,
"rewards/margins": 5.988953113555908,
"rewards/rejected": -11.230472564697266,
"step": 740
},
{
"epoch": 0.6054490413723511,
"grad_norm": 2.6915197372436523,
"learning_rate": 2.5e-06,
"logits/chosen": 2.866816759109497,
"logits/rejected": 2.84656023979187,
"logps/chosen": -3.155074119567871,
"logps/rejected": -8.019740104675293,
"loss": 0.2364,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -4.732610702514648,
"rewards/margins": 7.297000885009766,
"rewards/rejected": -12.029611587524414,
"step": 750
},
{
"epoch": 0.6054490413723511,
"eval_logits/chosen": 2.7122883796691895,
"eval_logits/rejected": 2.4663355350494385,
"eval_logps/chosen": -3.362666606903076,
"eval_logps/rejected": -7.107527732849121,
"eval_loss": 0.2374112904071808,
"eval_rewards/accuracies": 0.9801980257034302,
"eval_rewards/chosen": -5.043999671936035,
"eval_rewards/margins": 5.617292881011963,
"eval_rewards/rejected": -10.661293983459473,
"eval_runtime": 29.0963,
"eval_samples_per_second": 27.529,
"eval_steps_per_second": 3.471,
"step": 750
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 50,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 1.7450483691657626e+18,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}