{ "best_metric": null, "best_model_checkpoint": null, "epoch": 1.0, "eval_steps": 60, "global_step": 1125, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.008888888888888889, "grad_norm": 8.760091781616211, "learning_rate": 4.4247787610619474e-07, "logits/chosen": -0.8248252868652344, "logits/rejected": -0.8263720273971558, "logps/chosen": -0.36086463928222656, "logps/rejected": -5.696224689483643, "loss": 1.1038, "rewards/accuracies": 0.5125000476837158, "rewards/chosen": 17.43745994567871, "rewards/margins": 0.5984855890274048, "rewards/rejected": 16.838973999023438, "step": 10 }, { "epoch": 0.017777777777777778, "grad_norm": 8.855981826782227, "learning_rate": 8.849557522123895e-07, "logits/chosen": -0.8169006109237671, "logits/rejected": -0.819770872592926, "logps/chosen": -0.12464660406112671, "logps/rejected": -7.139842987060547, "loss": 1.1887, "rewards/accuracies": 0.4000000059604645, "rewards/chosen": 17.17649269104004, "rewards/margins": 0.19107049703598022, "rewards/rejected": 16.98542022705078, "step": 20 }, { "epoch": 0.02666666666666667, "grad_norm": 16.764184951782227, "learning_rate": 1.3274336283185843e-06, "logits/chosen": -0.8003113865852356, "logits/rejected": -0.8030117750167847, "logps/chosen": -0.34651467204093933, "logps/rejected": -6.967917442321777, "loss": 1.0563, "rewards/accuracies": 0.44999998807907104, "rewards/chosen": 17.280975341796875, "rewards/margins": 0.40005987882614136, "rewards/rejected": 16.88091468811035, "step": 30 }, { "epoch": 0.035555555555555556, "grad_norm": 8.33682918548584, "learning_rate": 1.769911504424779e-06, "logits/chosen": -0.7695047855377197, "logits/rejected": -0.7739207148551941, "logps/chosen": -1.5993006229400635, "logps/rejected": -8.504932403564453, "loss": 0.7596, "rewards/accuracies": 0.5, "rewards/chosen": 17.283912658691406, "rewards/margins": 0.6976072192192078, "rewards/rejected": 16.5863037109375, "step": 40 }, { "epoch": 0.044444444444444446, "grad_norm": 4.494723320007324, "learning_rate": 2.212389380530974e-06, "logits/chosen": -0.7154140472412109, "logits/rejected": -0.7225576043128967, "logps/chosen": -3.112199068069458, "logps/rejected": -12.212080001831055, "loss": 0.6083, "rewards/accuracies": 0.4625000059604645, "rewards/chosen": 17.03064727783203, "rewards/margins": 0.7148451805114746, "rewards/rejected": 16.3158016204834, "step": 50 }, { "epoch": 0.05333333333333334, "grad_norm": 5.110287666320801, "learning_rate": 2.6548672566371687e-06, "logits/chosen": -0.6322453022003174, "logits/rejected": -0.6387485265731812, "logps/chosen": -5.650620460510254, "logps/rejected": -12.759811401367188, "loss": 0.3835, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 17.101289749145508, "rewards/margins": 1.1824612617492676, "rewards/rejected": 15.918828964233398, "step": 60 }, { "epoch": 0.05333333333333334, "eval_logits/chosen": -0.5826543569564819, "eval_logits/rejected": -0.5914276838302612, "eval_logps/chosen": -3.5471787452697754, "eval_logps/rejected": -16.51181983947754, "eval_loss": 0.3286525011062622, "eval_rewards/accuracies": 0.9280000925064087, "eval_rewards/chosen": 17.148174285888672, "eval_rewards/margins": 1.4386365413665771, "eval_rewards/rejected": 15.709539413452148, "eval_runtime": 372.0227, "eval_samples_per_second": 2.688, "eval_steps_per_second": 0.336, "step": 60 }, { "epoch": 0.06222222222222222, "grad_norm": 5.098133563995361, "learning_rate": 3.097345132743363e-06, "logits/chosen": -0.5378152132034302, "logits/rejected": -0.5494933724403381, "logps/chosen": -1.5099802017211914, "logps/rejected": -21.206321716308594, "loss": 0.2931, "rewards/accuracies": 0.9375, "rewards/chosen": 17.083791732788086, "rewards/margins": 1.5844331979751587, "rewards/rejected": 15.499359130859375, "step": 70 }, { "epoch": 0.07111111111111111, "grad_norm": 29.787437438964844, "learning_rate": 3.539823008849558e-06, "logits/chosen": -0.443774938583374, "logits/rejected": -0.45571577548980713, "logps/chosen": -1.5804342031478882, "logps/rejected": -22.606929779052734, "loss": 0.202, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.302125930786133, "rewards/margins": 2.174014091491699, "rewards/rejected": 15.128110885620117, "step": 80 }, { "epoch": 0.08, "grad_norm": 23.14398193359375, "learning_rate": 3.982300884955752e-06, "logits/chosen": -0.3626072406768799, "logits/rejected": -0.3787815570831299, "logps/chosen": -2.203828811645508, "logps/rejected": -29.433551788330078, "loss": 0.2123, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.00284194946289, "rewards/margins": 2.320391893386841, "rewards/rejected": 14.682450294494629, "step": 90 }, { "epoch": 0.08888888888888889, "grad_norm": 29.672739028930664, "learning_rate": 4.424778761061948e-06, "logits/chosen": -0.3035663962364197, "logits/rejected": -0.31762221455574036, "logps/chosen": -3.433589458465576, "logps/rejected": -29.9322509765625, "loss": 0.2592, "rewards/accuracies": 0.9375, "rewards/chosen": 16.929956436157227, "rewards/margins": 2.31272029876709, "rewards/rejected": 14.617237091064453, "step": 100 }, { "epoch": 0.09777777777777778, "grad_norm": 1.873722791671753, "learning_rate": 4.867256637168142e-06, "logits/chosen": -0.2679600715637207, "logits/rejected": -0.2826440930366516, "logps/chosen": -0.9653514623641968, "logps/rejected": -30.235322952270508, "loss": 0.1336, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.462385177612305, "rewards/margins": 3.1994175910949707, "rewards/rejected": 14.26296615600586, "step": 110 }, { "epoch": 0.10666666666666667, "grad_norm": 1.6913721561431885, "learning_rate": 4.999409761242696e-06, "logits/chosen": -0.22222033143043518, "logits/rejected": -0.23720571398735046, "logps/chosen": -4.4953508377075195, "logps/rejected": -34.074745178222656, "loss": 0.2552, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": 17.04866600036621, "rewards/margins": 3.1014418601989746, "rewards/rejected": 13.947224617004395, "step": 120 }, { "epoch": 0.10666666666666667, "eval_logits/chosen": -0.206527978181839, "eval_logits/rejected": -0.22178640961647034, "eval_logps/chosen": -3.69442081451416, "eval_logps/rejected": -36.072166442871094, "eval_loss": 0.18996010720729828, "eval_rewards/accuracies": 0.9320000410079956, "eval_rewards/chosen": 17.133451461791992, "eval_rewards/margins": 3.379946708679199, "eval_rewards/rejected": 13.753504753112793, "eval_runtime": 361.5279, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 120 }, { "epoch": 0.11555555555555555, "grad_norm": 61.80262756347656, "learning_rate": 4.996519466816778e-06, "logits/chosen": -0.18473535776138306, "logits/rejected": -0.1988501250743866, "logps/chosen": -3.7009687423706055, "logps/rejected": -39.289939880371094, "loss": 0.1394, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.106964111328125, "rewards/margins": 3.633338212966919, "rewards/rejected": 13.473625183105469, "step": 130 }, { "epoch": 0.12444444444444444, "grad_norm": 1.6732702255249023, "learning_rate": 4.9912234871722805e-06, "logits/chosen": -0.16134041547775269, "logits/rejected": -0.17547868192195892, "logps/chosen": -3.0637736320495605, "logps/rejected": -40.07548522949219, "loss": 0.1408, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.392223358154297, "rewards/margins": 4.242353439331055, "rewards/rejected": 13.149867057800293, "step": 140 }, { "epoch": 0.13333333333333333, "grad_norm": 0.346453994512558, "learning_rate": 4.98352692559805e-06, "logits/chosen": -0.13797929883003235, "logits/rejected": -0.15283086895942688, "logps/chosen": -5.14492130279541, "logps/rejected": -47.97212219238281, "loss": 0.2153, "rewards/accuracies": 0.9375, "rewards/chosen": 16.896778106689453, "rewards/margins": 4.227695465087891, "rewards/rejected": 12.669081687927246, "step": 150 }, { "epoch": 0.14222222222222222, "grad_norm": 0.21871662139892578, "learning_rate": 4.973437198621237e-06, "logits/chosen": -0.12396670132875443, "logits/rejected": -0.13780555129051208, "logps/chosen": -6.108860015869141, "logps/rejected": -54.90739440917969, "loss": 0.0388, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 16.75935935974121, "rewards/margins": 4.755282878875732, "rewards/rejected": 12.004077911376953, "step": 160 }, { "epoch": 0.1511111111111111, "grad_norm": 235.12429809570312, "learning_rate": 4.960964028860621e-06, "logits/chosen": -0.1140839159488678, "logits/rejected": -0.1263057291507721, "logps/chosen": -12.605452537536621, "logps/rejected": -53.81230926513672, "loss": 0.4651, "rewards/accuracies": 0.875, "rewards/chosen": 16.101238250732422, "rewards/margins": 3.9864249229431152, "rewards/rejected": 12.114812850952148, "step": 170 }, { "epoch": 0.16, "grad_norm": 190.97048950195312, "learning_rate": 4.946119435657738e-06, "logits/chosen": -0.10746976733207703, "logits/rejected": -0.11878640949726105, "logps/chosen": -8.5105562210083, "logps/rejected": -51.314781188964844, "loss": 0.2362, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 16.719980239868164, "rewards/margins": 4.549674034118652, "rewards/rejected": 12.170306205749512, "step": 180 }, { "epoch": 0.16, "eval_logits/chosen": -0.10870806127786636, "eval_logits/rejected": -0.12223993986845016, "eval_logps/chosen": -4.414996147155762, "eval_logps/rejected": -53.885032653808594, "eval_loss": 0.20236633718013763, "eval_rewards/accuracies": 0.9510000944137573, "eval_rewards/chosen": 17.06139373779297, "eval_rewards/margins": 5.089176177978516, "eval_rewards/rejected": 11.97221851348877, "eval_runtime": 361.4355, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 180 }, { "epoch": 0.1688888888888889, "grad_norm": 56.81266784667969, "learning_rate": 4.928917723494854e-06, "logits/chosen": -0.10682469606399536, "logits/rejected": -0.12124393880367279, "logps/chosen": -3.058413028717041, "logps/rejected": -55.052528381347656, "loss": 0.2442, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.058589935302734, "rewards/margins": 5.056097984313965, "rewards/rejected": 12.002490043640137, "step": 190 }, { "epoch": 0.17777777777777778, "grad_norm": 175.06552124023438, "learning_rate": 4.909375468210947e-06, "logits/chosen": -0.10520349442958832, "logits/rejected": -0.12018950283527374, "logps/chosen": -4.114959716796875, "logps/rejected": -55.9394645690918, "loss": 0.1915, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 16.98603057861328, "rewards/margins": 5.105838775634766, "rewards/rejected": 11.880191802978516, "step": 200 }, { "epoch": 0.18666666666666668, "grad_norm": 78.06558990478516, "learning_rate": 4.8875115010289655e-06, "logits/chosen": -0.10475558042526245, "logits/rejected": -0.11949175596237183, "logps/chosen": -6.760301113128662, "logps/rejected": -53.91607666015625, "loss": 0.2843, "rewards/accuracies": 0.9375, "rewards/chosen": 16.857545852661133, "rewards/margins": 4.917357921600342, "rewards/rejected": 11.94018840789795, "step": 210 }, { "epoch": 0.19555555555555557, "grad_norm": 15.880486488342285, "learning_rate": 4.863346890409768e-06, "logits/chosen": -0.11213523149490356, "logits/rejected": -0.12581588327884674, "logps/chosen": -6.759585380554199, "logps/rejected": -51.10936737060547, "loss": 0.5104, "rewards/accuracies": 0.875, "rewards/chosen": 16.859071731567383, "rewards/margins": 4.638372898101807, "rewards/rejected": 12.220698356628418, "step": 220 }, { "epoch": 0.20444444444444446, "grad_norm": 46.97845458984375, "learning_rate": 4.836904921750224e-06, "logits/chosen": -0.11947059631347656, "logits/rejected": -0.1329912692308426, "logps/chosen": -3.608184814453125, "logps/rejected": -48.794761657714844, "loss": 0.2134, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.235904693603516, "rewards/margins": 4.859888076782227, "rewards/rejected": 12.376014709472656, "step": 230 }, { "epoch": 0.21333333333333335, "grad_norm": 24.032859802246094, "learning_rate": 4.808211074945042e-06, "logits/chosen": -0.1200513243675232, "logits/rejected": -0.1333036869764328, "logps/chosen": -3.7552154064178467, "logps/rejected": -49.87453079223633, "loss": 0.1781, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.094650268554688, "rewards/margins": 4.68077278137207, "rewards/rejected": 12.41387939453125, "step": 240 }, { "epoch": 0.21333333333333335, "eval_logits/chosen": -0.12433278560638428, "eval_logits/rejected": -0.13808581233024597, "eval_logps/chosen": -4.408891201019287, "eval_logps/rejected": -50.744781494140625, "eval_loss": 0.1546352356672287, "eval_rewards/accuracies": 0.9500000476837158, "eval_rewards/chosen": 17.06200408935547, "eval_rewards/margins": 4.775761604309082, "eval_rewards/rejected": 12.286243438720703, "eval_runtime": 361.4974, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 240 }, { "epoch": 0.2222222222222222, "grad_norm": 0.25737640261650085, "learning_rate": 4.7772929998339485e-06, "logits/chosen": -0.12348780035972595, "logits/rejected": -0.13704943656921387, "logps/chosen": -4.4299187660217285, "logps/rejected": -53.074607849121094, "loss": 0.1373, "rewards/accuracies": 0.9375, "rewards/chosen": 17.087068557739258, "rewards/margins": 5.06691837310791, "rewards/rejected": 12.020149230957031, "step": 250 }, { "epoch": 0.2311111111111111, "grad_norm": 0.1839389204978943, "learning_rate": 4.744180489557859e-06, "logits/chosen": -0.12177034467458725, "logits/rejected": -0.1342695653438568, "logps/chosen": -3.775188446044922, "logps/rejected": -53.98720932006836, "loss": 0.1896, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.12021255493164, "rewards/margins": 5.148064613342285, "rewards/rejected": 11.972146987915039, "step": 260 }, { "epoch": 0.24, "grad_norm": 12.258485794067383, "learning_rate": 4.708905451849754e-06, "logits/chosen": -0.11067859083414078, "logits/rejected": -0.12377731502056122, "logps/chosen": -6.418317794799805, "logps/rejected": -56.57402801513672, "loss": 0.2315, "rewards/accuracies": 0.9375, "rewards/chosen": 16.738832473754883, "rewards/margins": 4.884931564331055, "rewards/rejected": 11.853900909423828, "step": 270 }, { "epoch": 0.24888888888888888, "grad_norm": 77.56194305419922, "learning_rate": 4.671501878287879e-06, "logits/chosen": -0.1184445172548294, "logits/rejected": -0.1339874565601349, "logps/chosen": -10.12116527557373, "logps/rejected": -53.403907775878906, "loss": 0.5343, "rewards/accuracies": 0.862500011920929, "rewards/chosen": 16.458633422851562, "rewards/margins": 4.402472496032715, "rewards/rejected": 12.056160926818848, "step": 280 }, { "epoch": 0.2577777777777778, "grad_norm": 67.53883361816406, "learning_rate": 4.6320058115409295e-06, "logits/chosen": -0.1448262631893158, "logits/rejected": -0.15793387591838837, "logps/chosen": -3.4666190147399902, "logps/rejected": -48.79213333129883, "loss": 0.5017, "rewards/accuracies": 0.887499988079071, "rewards/chosen": 16.945899963378906, "rewards/margins": 4.2686333656311035, "rewards/rejected": 12.677268028259277, "step": 290 }, { "epoch": 0.26666666666666666, "grad_norm": 0.17521341145038605, "learning_rate": 4.590455310636778e-06, "logits/chosen": -0.16128253936767578, "logits/rejected": -0.17375555634498596, "logps/chosen": -2.9032950401306152, "logps/rejected": -47.69734191894531, "loss": 0.265, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.18383026123047, "rewards/margins": 4.541309356689453, "rewards/rejected": 12.642518997192383, "step": 300 }, { "epoch": 0.26666666666666666, "eval_logits/chosen": -0.17444846034049988, "eval_logits/rejected": -0.18559777736663818, "eval_logps/chosen": -2.535512924194336, "eval_logps/rejected": -47.16367721557617, "eval_loss": 0.15360687673091888, "eval_rewards/accuracies": 0.9440000653266907, "eval_rewards/chosen": 17.249343872070312, "eval_rewards/margins": 4.604989051818848, "eval_rewards/rejected": 12.644353866577148, "eval_runtime": 361.4575, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 300 }, { "epoch": 0.27555555555555555, "grad_norm": 0.5040452480316162, "learning_rate": 4.54689041428819e-06, "logits/chosen": -0.16974106431007385, "logits/rejected": -0.1810058057308197, "logps/chosen": -1.233938217163086, "logps/rejected": -49.907745361328125, "loss": 0.1132, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.34117889404297, "rewards/margins": 4.934173583984375, "rewards/rejected": 12.407005310058594, "step": 310 }, { "epoch": 0.28444444444444444, "grad_norm": 100.02949523925781, "learning_rate": 4.501353102310901e-06, "logits/chosen": -0.15705889463424683, "logits/rejected": -0.1695334017276764, "logps/chosen": -1.0820492506027222, "logps/rejected": -52.577110290527344, "loss": 0.1194, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.33388900756836, "rewards/margins": 5.154760837554932, "rewards/rejected": 12.179126739501953, "step": 320 }, { "epoch": 0.29333333333333333, "grad_norm": 0.2689219117164612, "learning_rate": 4.453887255171206e-06, "logits/chosen": -0.13849371671676636, "logits/rejected": -0.14990833401679993, "logps/chosen": -1.8435032367706299, "logps/rejected": -54.79044723510742, "loss": 0.0926, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.2423095703125, "rewards/margins": 5.28987979888916, "rewards/rejected": 11.952428817749023, "step": 330 }, { "epoch": 0.3022222222222222, "grad_norm": 0.09305431693792343, "learning_rate": 4.404538611702055e-06, "logits/chosen": -0.12299702316522598, "logits/rejected": -0.13453055918216705, "logps/chosen": -2.9897143840789795, "logps/rejected": -52.954498291015625, "loss": 0.2873, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.17474365234375, "rewards/margins": 5.071004867553711, "rewards/rejected": 12.103739738464355, "step": 340 }, { "epoch": 0.3111111111111111, "grad_norm": 59.282073974609375, "learning_rate": 4.3533547250284015e-06, "logits/chosen": -0.11913029849529266, "logits/rejected": -0.12785324454307556, "logps/chosen": -3.9456872940063477, "logps/rejected": -48.68487548828125, "loss": 0.4332, "rewards/accuracies": 0.875, "rewards/chosen": 17.12805938720703, "rewards/margins": 4.669450283050537, "rewards/rejected": 12.458610534667969, "step": 350 }, { "epoch": 0.32, "grad_norm": 0.31101909279823303, "learning_rate": 4.300384916744261e-06, "logits/chosen": -0.11280188709497452, "logits/rejected": -0.12300585210323334, "logps/chosen": -2.1714723110198975, "logps/rejected": -54.74174118041992, "loss": 0.1605, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.326162338256836, "rewards/margins": 5.467062473297119, "rewards/rejected": 11.859098434448242, "step": 360 }, { "epoch": 0.32, "eval_logits/chosen": -0.10620756447315216, "eval_logits/rejected": -0.11727114766836166, "eval_logps/chosen": -1.4165427684783936, "eval_logps/rejected": -50.9525146484375, "eval_loss": 0.3194349706172943, "eval_rewards/accuracies": 0.9210000038146973, "eval_rewards/chosen": 17.36124038696289, "eval_rewards/margins": 5.095769882202148, "eval_rewards/rejected": 12.26546859741211, "eval_runtime": 361.5072, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 360 }, { "epoch": 0.3288888888888889, "grad_norm": 6.1126532554626465, "learning_rate": 4.24568022938566e-06, "logits/chosen": -0.10354311764240265, "logits/rejected": -0.11526636779308319, "logps/chosen": -1.2935255765914917, "logps/rejected": -55.57566833496094, "loss": 0.1711, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.439346313476562, "rewards/margins": 5.700921058654785, "rewards/rejected": 11.738424301147461, "step": 370 }, { "epoch": 0.3377777777777778, "grad_norm": 34.15927505493164, "learning_rate": 4.189293377245241e-06, "logits/chosen": -0.1029932051897049, "logits/rejected": -0.11382515728473663, "logps/chosen": -2.5132687091827393, "logps/rejected": -55.50346374511719, "loss": 0.4359, "rewards/accuracies": 0.8875000476837158, "rewards/chosen": 16.731037139892578, "rewards/margins": 4.368172645568848, "rewards/rejected": 12.362865447998047, "step": 380 }, { "epoch": 0.3466666666666667, "grad_norm": 2.8422904014587402, "learning_rate": 4.131278695575952e-06, "logits/chosen": -0.10793520510196686, "logits/rejected": -0.12109285593032837, "logps/chosen": -3.014652729034424, "logps/rejected": -53.98411560058594, "loss": 0.2161, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.137393951416016, "rewards/margins": 5.105995178222656, "rewards/rejected": 12.03139877319336, "step": 390 }, { "epoch": 0.35555555555555557, "grad_norm": 54.0329475402832, "learning_rate": 4.071692088232743e-06, "logits/chosen": -0.10393750667572021, "logits/rejected": -0.11834606528282166, "logps/chosen": -2.1508543491363525, "logps/rejected": -45.60733413696289, "loss": 0.2077, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.586124420166016, "rewards/margins": 5.077212333679199, "rewards/rejected": 12.5089111328125, "step": 400 }, { "epoch": 0.36444444444444446, "grad_norm": 81.61144256591797, "learning_rate": 4.010590973802737e-06, "logits/chosen": -0.09564584493637085, "logits/rejected": -0.10617707669734955, "logps/chosen": -3.4572842121124268, "logps/rejected": -50.92162322998047, "loss": 0.2478, "rewards/accuracies": 0.8875000476837158, "rewards/chosen": 17.010910034179688, "rewards/margins": 4.556198596954346, "rewards/rejected": 12.454713821411133, "step": 410 }, { "epoch": 0.37333333333333335, "grad_norm": 0.30974289774894714, "learning_rate": 3.948034230275781e-06, "logits/chosen": -0.09134417027235031, "logits/rejected": -0.1020016297698021, "logps/chosen": -5.046698570251465, "logps/rejected": -48.908958435058594, "loss": 0.2894, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": 17.007888793945312, "rewards/margins": 4.53641414642334, "rewards/rejected": 12.471475601196289, "step": 420 }, { "epoch": 0.37333333333333335, "eval_logits/chosen": -0.09054450690746307, "eval_logits/rejected": -0.10264354199171066, "eval_logps/chosen": -1.913105845451355, "eval_logps/rejected": -51.11127471923828, "eval_loss": 0.16789735853672028, "eval_rewards/accuracies": 0.9450000524520874, "eval_rewards/chosen": 17.311582565307617, "eval_rewards/margins": 5.061989784240723, "eval_rewards/rejected": 12.249593734741211, "eval_runtime": 361.5337, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 420 }, { "epoch": 0.38222222222222224, "grad_norm": 12.824393272399902, "learning_rate": 3.884082138308699e-06, "logits/chosen": -0.08666776865720749, "logits/rejected": -0.0997733399271965, "logps/chosen": -1.7306327819824219, "logps/rejected": -54.273292541503906, "loss": 0.2298, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.167621612548828, "rewards/margins": 5.065673351287842, "rewards/rejected": 12.101947784423828, "step": 430 }, { "epoch": 0.39111111111111113, "grad_norm": 0.30713599920272827, "learning_rate": 3.818796323137896e-06, "logits/chosen": -0.09174907952547073, "logits/rejected": -0.10376611351966858, "logps/chosen": -1.489154577255249, "logps/rejected": -54.580726623535156, "loss": 0.2513, "rewards/accuracies": 0.9375, "rewards/chosen": 17.22280502319336, "rewards/margins": 5.175349235534668, "rewards/rejected": 12.047454833984375, "step": 440 }, { "epoch": 0.4, "grad_norm": 87.4791488647461, "learning_rate": 3.7522396951963303e-06, "logits/chosen": -0.09688778221607208, "logits/rejected": -0.10897806286811829, "logps/chosen": -3.157695770263672, "logps/rejected": -50.96417236328125, "loss": 0.1758, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.345651626586914, "rewards/margins": 5.245656967163086, "rewards/rejected": 12.099993705749512, "step": 450 }, { "epoch": 0.4088888888888889, "grad_norm": 146.2008056640625, "learning_rate": 3.684476389492026e-06, "logits/chosen": -0.09378582239151001, "logits/rejected": -0.10475654900074005, "logps/chosen": -0.5611928701400757, "logps/rejected": -56.518890380859375, "loss": 0.1981, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.113712310791016, "rewards/margins": 5.068872928619385, "rewards/rejected": 12.044839859008789, "step": 460 }, { "epoch": 0.4177777777777778, "grad_norm": 1.9137721061706543, "learning_rate": 3.6155717038065783e-06, "logits/chosen": -0.08695463836193085, "logits/rejected": -0.09596743434667587, "logps/chosen": -1.5298550128936768, "logps/rejected": -50.27445983886719, "loss": 0.2066, "rewards/accuracies": 0.9375, "rewards/chosen": 17.35186004638672, "rewards/margins": 5.014693260192871, "rewards/rejected": 12.337167739868164, "step": 470 }, { "epoch": 0.4266666666666667, "grad_norm": 84.80391693115234, "learning_rate": 3.545592035773192e-06, "logits/chosen": -0.0746893435716629, "logits/rejected": -0.08653923869132996, "logps/chosen": -2.0052125453948975, "logps/rejected": -57.502811431884766, "loss": 0.1149, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.14373016357422, "rewards/margins": 5.360415935516357, "rewards/rejected": 11.783313751220703, "step": 480 }, { "epoch": 0.4266666666666667, "eval_logits/chosen": -0.07700399309396744, "eval_logits/rejected": -0.08828537166118622, "eval_logps/chosen": -4.48896598815918, "eval_logps/rejected": -53.76282501220703, "eval_loss": 0.29511645436286926, "eval_rewards/accuracies": 0.9230000376701355, "eval_rewards/chosen": 17.053997039794922, "eval_rewards/margins": 5.069558143615723, "eval_rewards/rejected": 11.984437942504883, "eval_runtime": 361.5035, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 480 }, { "epoch": 0.43555555555555553, "grad_norm": 82.9616470336914, "learning_rate": 3.4746048188948806e-06, "logits/chosen": -0.06675051152706146, "logits/rejected": -0.07860895991325378, "logps/chosen": -4.162237167358398, "logps/rejected": -54.77789306640625, "loss": 0.2979, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 17.047603607177734, "rewards/margins": 5.138361930847168, "rewards/rejected": 11.909242630004883, "step": 490 }, { "epoch": 0.4444444444444444, "grad_norm": 0.04293210059404373, "learning_rate": 3.4026784575644887e-06, "logits/chosen": -0.06424491107463837, "logits/rejected": -0.07567107677459717, "logps/chosen": -2.05729603767395, "logps/rejected": -56.646087646484375, "loss": 0.4378, "rewards/accuracies": 0.8875000476837158, "rewards/chosen": 16.947803497314453, "rewards/margins": 4.919981956481934, "rewards/rejected": 12.02782154083252, "step": 500 }, { "epoch": 0.4533333333333333, "grad_norm": 0.07136644423007965, "learning_rate": 3.329882261149148e-06, "logits/chosen": -0.06423303484916687, "logits/rejected": -0.07512776553630829, "logps/chosen": -3.1519265174865723, "logps/rejected": -53.53008270263672, "loss": 0.2613, "rewards/accuracies": 0.9375, "rewards/chosen": 17.112144470214844, "rewards/margins": 5.046430587768555, "rewards/rejected": 12.065712928771973, "step": 510 }, { "epoch": 0.4622222222222222, "grad_norm": 0.29279613494873047, "learning_rate": 3.25628637720269e-06, "logits/chosen": -0.060233693569898605, "logits/rejected": -0.07076811790466309, "logps/chosen": -1.2358124256134033, "logps/rejected": -50.61806106567383, "loss": 0.1517, "rewards/accuracies": 0.9375, "rewards/chosen": 17.457351684570312, "rewards/margins": 5.2062835693359375, "rewards/rejected": 12.251070022583008, "step": 520 }, { "epoch": 0.4711111111111111, "grad_norm": 0.7946074604988098, "learning_rate": 3.181961723870359e-06, "logits/chosen": -0.054482050240039825, "logits/rejected": -0.0657092034816742, "logps/chosen": -0.7682158946990967, "logps/rejected": -58.81409454345703, "loss": 0.2304, "rewards/accuracies": 0.9375, "rewards/chosen": 17.075973510742188, "rewards/margins": 5.222441673278809, "rewards/rejected": 11.853530883789062, "step": 530 }, { "epoch": 0.48, "grad_norm": 3.1405210494995117, "learning_rate": 3.1069799215509847e-06, "logits/chosen": -0.05050881579518318, "logits/rejected": -0.061149902641773224, "logps/chosen": -0.954046368598938, "logps/rejected": -52.70227813720703, "loss": 0.0384, "rewards/accuracies": 0.987500011920929, "rewards/chosen": 17.378849029541016, "rewards/margins": 5.253483295440674, "rewards/rejected": 12.125364303588867, "step": 540 }, { "epoch": 0.48, "eval_logits/chosen": -0.05120665580034256, "eval_logits/rejected": -0.06123337894678116, "eval_logps/chosen": -2.9872913360595703, "eval_logps/rejected": -52.27314758300781, "eval_loss": 0.17387841641902924, "eval_rewards/accuracies": 0.9490000605583191, "eval_rewards/chosen": 17.204164505004883, "eval_rewards/margins": 5.070757865905762, "eval_rewards/rejected": 12.133406639099121, "eval_runtime": 361.5449, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 540 }, { "epoch": 0.4888888888888889, "grad_norm": 0.07270358502864838, "learning_rate": 3.0314132238824416e-06, "logits/chosen": -0.05125313252210617, "logits/rejected": -0.06174170970916748, "logps/chosen": -3.2163877487182617, "logps/rejected": -47.79279327392578, "loss": 0.2087, "rewards/accuracies": 0.9125000238418579, "rewards/chosen": 17.408517837524414, "rewards/margins": 5.061524391174316, "rewards/rejected": 12.346992492675781, "step": 550 }, { "epoch": 0.49777777777777776, "grad_norm": 0.10005924850702286, "learning_rate": 2.955334448116915e-06, "logits/chosen": -0.041773442178964615, "logits/rejected": -0.05364570394158363, "logps/chosen": -0.363404780626297, "logps/rejected": -56.32415008544922, "loss": 0.0969, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.3824462890625, "rewards/margins": 5.5404510498046875, "rewards/rejected": 11.841995239257812, "step": 560 }, { "epoch": 0.5066666666666667, "grad_norm": 7.818356990814209, "learning_rate": 2.8788169049530533e-06, "logits/chosen": -0.04309462010860443, "logits/rejected": -0.05494442582130432, "logps/chosen": -2.2242724895477295, "logps/rejected": -56.444740295410156, "loss": 0.1447, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.275390625, "rewards/margins": 5.55007791519165, "rewards/rejected": 11.725313186645508, "step": 570 }, { "epoch": 0.5155555555555555, "grad_norm": 0.03519747406244278, "learning_rate": 2.8019343278926397e-06, "logits/chosen": -0.03506368771195412, "logits/rejected": -0.046854715794324875, "logps/chosen": -0.5200096964836121, "logps/rejected": -59.05330276489258, "loss": 0.0998, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.320554733276367, "rewards/margins": 5.727260589599609, "rewards/rejected": 11.593294143676758, "step": 580 }, { "epoch": 0.5244444444444445, "grad_norm": 0.04519123584032059, "learning_rate": 2.7247608021898265e-06, "logits/chosen": -0.03204537555575371, "logits/rejected": -0.04383891448378563, "logps/chosen": -1.1271060705184937, "logps/rejected": -59.308895111083984, "loss": 0.1162, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.256351470947266, "rewards/margins": 5.6881890296936035, "rewards/rejected": 11.568161010742188, "step": 590 }, { "epoch": 0.5333333333333333, "grad_norm": 0.028689857572317123, "learning_rate": 2.647370693461432e-06, "logits/chosen": -0.02834726870059967, "logits/rejected": -0.03827046602964401, "logps/chosen": -5.673943519592285, "logps/rejected": -55.72624588012695, "loss": 0.4008, "rewards/accuracies": 0.8875000476837158, "rewards/chosen": 16.761056900024414, "rewards/margins": 4.804043769836426, "rewards/rejected": 11.957012176513672, "step": 600 }, { "epoch": 0.5333333333333333, "eval_logits/chosen": -0.035840023308992386, "eval_logits/rejected": -0.04687971621751785, "eval_logps/chosen": -2.1760435104370117, "eval_logps/rejected": -56.62664031982422, "eval_loss": 0.17061151564121246, "eval_rewards/accuracies": 0.9470000267028809, "eval_rewards/chosen": 17.285289764404297, "eval_rewards/margins": 5.587231636047363, "eval_rewards/rejected": 11.698057174682617, "eval_runtime": 361.5056, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 600 }, { "epoch": 0.5422222222222223, "grad_norm": 1.1994622945785522, "learning_rate": 2.569838576027068e-06, "logits/chosen": -0.03231767192482948, "logits/rejected": -0.04397805407643318, "logps/chosen": -1.5904741287231445, "logps/rejected": -62.10520553588867, "loss": 0.1265, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.11206817626953, "rewards/margins": 5.734784126281738, "rewards/rejected": 11.37728500366211, "step": 610 }, { "epoch": 0.5511111111111111, "grad_norm": 0.14163845777511597, "learning_rate": 2.4922391610481544e-06, "logits/chosen": -0.03293662518262863, "logits/rejected": -0.04335154965519905, "logps/chosen": -1.3568997383117676, "logps/rejected": -58.347434997558594, "loss": 0.097, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.345399856567383, "rewards/margins": 5.823373794555664, "rewards/rejected": 11.522026062011719, "step": 620 }, { "epoch": 0.56, "grad_norm": 1.125027060508728, "learning_rate": 2.4146472245350804e-06, "logits/chosen": -0.02864963933825493, "logits/rejected": -0.03835710883140564, "logps/chosen": -2.494175434112549, "logps/rejected": -55.33067321777344, "loss": 0.3115, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.262205123901367, "rewards/margins": 5.4287261962890625, "rewards/rejected": 11.833479881286621, "step": 630 }, { "epoch": 0.5688888888888889, "grad_norm": 0.03514016419649124, "learning_rate": 2.337137535291868e-06, "logits/chosen": -0.02757749892771244, "logits/rejected": -0.0376611053943634, "logps/chosen": -2.3163387775421143, "logps/rejected": -53.54579162597656, "loss": 0.1268, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.542863845825195, "rewards/margins": 5.799897193908691, "rewards/rejected": 11.742965698242188, "step": 640 }, { "epoch": 0.5777777777777777, "grad_norm": 34.20791244506836, "learning_rate": 2.259784782867782e-06, "logits/chosen": -0.02032250165939331, "logits/rejected": -0.031542714685201645, "logps/chosen": -1.7910137176513672, "logps/rejected": -62.980018615722656, "loss": 0.1831, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 16.982879638671875, "rewards/margins": 5.589078903198242, "rewards/rejected": 11.393800735473633, "step": 650 }, { "epoch": 0.5866666666666667, "grad_norm": 32.356956481933594, "learning_rate": 2.182663505585314e-06, "logits/chosen": -0.01711965538561344, "logits/rejected": -0.028038471937179565, "logps/chosen": -2.662904739379883, "logps/rejected": -63.326297760009766, "loss": 0.1678, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 16.997777938842773, "rewards/margins": 5.741157531738281, "rewards/rejected": 11.256620407104492, "step": 660 }, { "epoch": 0.5866666666666667, "eval_logits/chosen": -0.016018809750676155, "eval_logits/rejected": -0.027020033448934555, "eval_logps/chosen": -3.00822114944458, "eval_logps/rejected": -57.951629638671875, "eval_loss": 0.2050127536058426, "eval_rewards/accuracies": 0.9450000524520874, "eval_rewards/chosen": 17.202072143554688, "eval_rewards/margins": 5.636512756347656, "eval_rewards/rejected": 11.565557479858398, "eval_runtime": 361.5073, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 660 }, { "epoch": 0.5955555555555555, "grad_norm": 86.53874969482422, "learning_rate": 2.1058480187138863e-06, "logits/chosen": -0.012352555990219116, "logits/rejected": -0.023966707289218903, "logps/chosen": -2.129209041595459, "logps/rejected": -62.1393928527832, "loss": 0.1025, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.069538116455078, "rewards/margins": 5.678930759429932, "rewards/rejected": 11.390605926513672, "step": 670 }, { "epoch": 0.6044444444444445, "grad_norm": 2.892672300338745, "learning_rate": 2.0294123428584985e-06, "logits/chosen": -0.011562807485461235, "logits/rejected": -0.020859256386756897, "logps/chosen": -3.2486608028411865, "logps/rejected": -59.56721496582031, "loss": 0.1961, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.051376342773438, "rewards/margins": 5.515361785888672, "rewards/rejected": 11.536016464233398, "step": 680 }, { "epoch": 0.6133333333333333, "grad_norm": 30.26588249206543, "learning_rate": 1.953430132632311e-06, "logits/chosen": -0.011488726362586021, "logits/rejected": -0.021527227014303207, "logps/chosen": -1.794357180595398, "logps/rejected": -60.649505615234375, "loss": 0.2821, "rewards/accuracies": 0.9375, "rewards/chosen": 17.114843368530273, "rewards/margins": 5.612217903137207, "rewards/rejected": 11.50262451171875, "step": 690 }, { "epoch": 0.6222222222222222, "grad_norm": 0.030314341187477112, "learning_rate": 1.8779746056819104e-06, "logits/chosen": -0.014436552301049232, "logits/rejected": -0.026044374331831932, "logps/chosen": -3.1617178916931152, "logps/rejected": -60.94548797607422, "loss": 0.3141, "rewards/accuracies": 0.9375, "rewards/chosen": 17.08698272705078, "rewards/margins": 5.719264984130859, "rewards/rejected": 11.367716789245605, "step": 700 }, { "epoch": 0.6311111111111111, "grad_norm": 0.06711317598819733, "learning_rate": 1.8031184721336364e-06, "logits/chosen": -0.017312290146946907, "logits/rejected": -0.026554957032203674, "logps/chosen": -5.120705604553223, "logps/rejected": -54.33483123779297, "loss": 0.3827, "rewards/accuracies": 0.887499988079071, "rewards/chosen": 17.1247501373291, "rewards/margins": 5.327882766723633, "rewards/rejected": 11.796867370605469, "step": 710 }, { "epoch": 0.64, "grad_norm": 60.36637496948242, "learning_rate": 1.7289338645289711e-06, "logits/chosen": -0.01987219974398613, "logits/rejected": -0.02766304835677147, "logps/chosen": -2.1757986545562744, "logps/rejected": -54.02531433105469, "loss": 0.2272, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.239776611328125, "rewards/margins": 5.250313758850098, "rewards/rejected": 11.989462852478027, "step": 720 }, { "epoch": 0.64, "eval_logits/chosen": -0.022905193269252777, "eval_logits/rejected": -0.0321992002427578, "eval_logps/chosen": -1.100506067276001, "eval_logps/rejected": -55.91169738769531, "eval_loss": 0.14018221199512482, "eval_rewards/accuracies": 0.9520000219345093, "eval_rewards/chosen": 17.392841339111328, "eval_rewards/margins": 5.623291492462158, "eval_rewards/rejected": 11.769551277160645, "eval_runtime": 361.416, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 720 }, { "epoch": 0.6488888888888888, "grad_norm": 0.029294608160853386, "learning_rate": 1.6554922683164875e-06, "logits/chosen": -0.016416028141975403, "logits/rejected": -0.025527067482471466, "logps/chosen": -0.906692385673523, "logps/rejected": -59.142173767089844, "loss": 0.1171, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.282169342041016, "rewards/margins": 5.6978583335876465, "rewards/rejected": 11.584310531616211, "step": 730 }, { "epoch": 0.6577777777777778, "grad_norm": 33.89070510864258, "learning_rate": 1.5828644529673592e-06, "logits/chosen": -0.016044551506638527, "logits/rejected": -0.025811903178691864, "logps/chosen": -1.1167538166046143, "logps/rejected": -59.84492492675781, "loss": 0.2001, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.266983032226562, "rewards/margins": 5.749438285827637, "rewards/rejected": 11.51754379272461, "step": 740 }, { "epoch": 0.6666666666666666, "grad_norm": 0.53726726770401, "learning_rate": 1.5111204037807844e-06, "logits/chosen": -0.012529855594038963, "logits/rejected": -0.023765765130519867, "logps/chosen": -0.3461765646934509, "logps/rejected": -55.841102600097656, "loss": 0.1692, "rewards/accuracies": 0.9375, "rewards/chosen": 17.61865997314453, "rewards/margins": 6.019055366516113, "rewards/rejected": 11.599604606628418, "step": 750 }, { "epoch": 0.6755555555555556, "grad_norm": 0.7173987627029419, "learning_rate": 1.4403292544450625e-06, "logits/chosen": -0.013380522839725018, "logits/rejected": -0.022107835859060287, "logps/chosen": -3.2759666442871094, "logps/rejected": -53.836822509765625, "loss": 0.2418, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.383333206176758, "rewards/margins": 5.590358734130859, "rewards/rejected": 11.792974472045898, "step": 760 }, { "epoch": 0.6844444444444444, "grad_norm": 2.744900941848755, "learning_rate": 1.3705592204192853e-06, "logits/chosen": -0.011318420059978962, "logits/rejected": -0.021840626373887062, "logps/chosen": -2.592001438140869, "logps/rejected": -57.06926345825195, "loss": 0.1749, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.286128997802734, "rewards/margins": 5.662715911865234, "rewards/rejected": 11.623414993286133, "step": 770 }, { "epoch": 0.6933333333333334, "grad_norm": 70.76551818847656, "learning_rate": 1.301877533199859e-06, "logits/chosen": -0.013944407925009727, "logits/rejected": -0.025667501613497734, "logps/chosen": -2.1070234775543213, "logps/rejected": -57.72039794921875, "loss": 0.1915, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.28545379638672, "rewards/margins": 5.708344459533691, "rewards/rejected": 11.577108383178711, "step": 780 }, { "epoch": 0.6933333333333334, "eval_logits/chosen": -0.01664295792579651, "eval_logits/rejected": -0.026626665145158768, "eval_logps/chosen": -1.082255244255066, "eval_logps/rejected": -55.95073699951172, "eval_loss": 0.24412688612937927, "eval_rewards/accuracies": 0.9320000410079956, "eval_rewards/chosen": 17.394668579101562, "eval_rewards/margins": 5.629020690917969, "eval_rewards/rejected": 11.765647888183594, "eval_runtime": 361.4602, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 780 }, { "epoch": 0.7022222222222222, "grad_norm": 0.41359376907348633, "learning_rate": 1.2343503755351729e-06, "logits/chosen": -0.012772129848599434, "logits/rejected": -0.023186586797237396, "logps/chosen": -0.8030359148979187, "logps/rejected": -57.84947967529297, "loss": 0.2143, "rewards/accuracies": 0.9375, "rewards/chosen": 17.32242202758789, "rewards/margins": 5.645486831665039, "rewards/rejected": 11.676933288574219, "step": 790 }, { "epoch": 0.7111111111111111, "grad_norm": 0.41748157143592834, "learning_rate": 1.168042817650881e-06, "logits/chosen": -0.012502101249992847, "logits/rejected": -0.023272844031453133, "logps/chosen": -1.0377256870269775, "logps/rejected": -57.05615997314453, "loss": 0.0792, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.60501480102539, "rewards/margins": 6.159370422363281, "rewards/rejected": 11.445646286010742, "step": 800 }, { "epoch": 0.72, "grad_norm": 0.039210401475429535, "learning_rate": 1.1030187545472012e-06, "logits/chosen": -0.008177272044122219, "logits/rejected": -0.01832464337348938, "logps/chosen": -3.8701748847961426, "logps/rejected": -57.661476135253906, "loss": 0.3177, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.14336395263672, "rewards/margins": 5.583393573760986, "rewards/rejected": 11.55997085571289, "step": 810 }, { "epoch": 0.7288888888888889, "grad_norm": 1.6096951961517334, "learning_rate": 1.0393408444287048e-06, "logits/chosen": -0.006832236424088478, "logits/rejected": -0.01682097464799881, "logps/chosen": -2.1741790771484375, "logps/rejected": -57.273414611816406, "loss": 0.2128, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.426712036132812, "rewards/margins": 5.911205291748047, "rewards/rejected": 11.515506744384766, "step": 820 }, { "epoch": 0.7377777777777778, "grad_norm": 3.7820959091186523, "learning_rate": 9.770704483258782e-07, "logits/chosen": -0.009998206980526447, "logits/rejected": -0.0204261876642704, "logps/chosen": -1.9802953004837036, "logps/rejected": -56.901512145996094, "loss": 0.1296, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.56071662902832, "rewards/margins": 6.169583320617676, "rewards/rejected": 11.391134262084961, "step": 830 }, { "epoch": 0.7466666666666667, "grad_norm": 1.126626968383789, "learning_rate": 9.162675709666865e-07, "logits/chosen": -0.00826224498450756, "logits/rejected": -0.018977787345647812, "logps/chosen": -1.4256607294082642, "logps/rejected": -61.65986633300781, "loss": 0.0635, "rewards/accuracies": 0.987500011920929, "rewards/chosen": 17.324975967407227, "rewards/margins": 6.09440803527832, "rewards/rejected": 11.230567932128906, "step": 840 }, { "epoch": 0.7466666666666667, "eval_logits/chosen": -0.01111944392323494, "eval_logits/rejected": -0.021697774529457092, "eval_logps/chosen": -1.2168633937835693, "eval_logps/rejected": -58.2642822265625, "eval_loss": 0.1689341962337494, "eval_rewards/accuracies": 0.9450000524520874, "eval_rewards/chosen": 17.381206512451172, "eval_rewards/margins": 5.846914291381836, "eval_rewards/rejected": 11.534292221069336, "eval_runtime": 361.6192, "eval_samples_per_second": 2.765, "eval_steps_per_second": 0.346, "step": 840 }, { "epoch": 0.7555555555555555, "grad_norm": 5.250723838806152, "learning_rate": 8.569908029550686e-07, "logits/chosen": -0.006854387000203133, "logits/rejected": -0.018336206674575806, "logps/chosen": -0.6238930821418762, "logps/rejected": -60.925689697265625, "loss": 0.1157, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.443281173706055, "rewards/margins": 6.164813995361328, "rewards/rejected": 11.278467178344727, "step": 850 }, { "epoch": 0.7644444444444445, "grad_norm": 3.1401162147521973, "learning_rate": 7.992972643121227e-07, "logits/chosen": -0.0037835021503269672, "logits/rejected": -0.013135241344571114, "logps/chosen": -0.8492221832275391, "logps/rejected": -55.516075134277344, "loss": 0.2252, "rewards/accuracies": 0.9375, "rewards/chosen": 17.488588333129883, "rewards/margins": 5.736725807189941, "rewards/rejected": 11.751862525939941, "step": 860 }, { "epoch": 0.7733333333333333, "grad_norm": 55.528812408447266, "learning_rate": 7.432425494343509e-07, "logits/chosen": -0.0033687639515846968, "logits/rejected": -0.013152632862329483, "logps/chosen": -1.3188884258270264, "logps/rejected": -57.9510498046875, "loss": 0.1398, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.412578582763672, "rewards/margins": 5.868515968322754, "rewards/rejected": 11.544061660766602, "step": 870 }, { "epoch": 0.7822222222222223, "grad_norm": 0.039824869483709335, "learning_rate": 6.888806735220396e-07, "logits/chosen": -0.0010406378423795104, "logits/rejected": -0.012095071375370026, "logps/chosen": -2.0619027614593506, "logps/rejected": -59.65806579589844, "loss": 0.2966, "rewards/accuracies": 0.9375, "rewards/chosen": 17.192535400390625, "rewards/margins": 5.7006731033325195, "rewards/rejected": 11.491861343383789, "step": 880 }, { "epoch": 0.7911111111111111, "grad_norm": 0.7536466717720032, "learning_rate": 6.362640205293583e-07, "logits/chosen": -0.0016857212176546454, "logits/rejected": -0.010936147533357143, "logps/chosen": -2.1478958129882812, "logps/rejected": -58.20386505126953, "loss": 0.1641, "rewards/accuracies": 0.9375, "rewards/chosen": 17.243247985839844, "rewards/margins": 5.678771018981934, "rewards/rejected": 11.564477920532227, "step": 890 }, { "epoch": 0.8, "grad_norm": 0.3857377767562866, "learning_rate": 5.854432926863684e-07, "logits/chosen": 0.00038508616853505373, "logits/rejected": -0.011034643277525902, "logps/chosen": -1.606274127960205, "logps/rejected": -62.58662414550781, "loss": 0.1703, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.211261749267578, "rewards/margins": 5.966868877410889, "rewards/rejected": 11.244392395019531, "step": 900 }, { "epoch": 0.8, "eval_logits/chosen": 0.00021816430671606213, "eval_logits/rejected": -0.010477552190423012, "eval_logps/chosen": -1.7576563358306885, "eval_logps/rejected": -59.79063415527344, "eval_loss": 0.13995186984539032, "eval_rewards/accuracies": 0.9610000848770142, "eval_rewards/chosen": 17.327129364013672, "eval_rewards/margins": 5.945469856262207, "eval_rewards/rejected": 11.381658554077148, "eval_runtime": 361.4592, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 900 }, { "epoch": 0.8088888888888889, "grad_norm": 0.13143697381019592, "learning_rate": 5.364674616415547e-07, "logits/chosen": 0.0005570838693529367, "logits/rejected": -0.011198626831173897, "logps/chosen": -0.12791283428668976, "logps/rejected": -65.7052993774414, "loss": 0.0507, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.224443435668945, "rewards/margins": 6.163690567016602, "rewards/rejected": 11.060752868652344, "step": 910 }, { "epoch": 0.8177777777777778, "grad_norm": 102.9654541015625, "learning_rate": 4.893837212719859e-07, "logits/chosen": -0.0008557128603570163, "logits/rejected": -0.01163212489336729, "logps/chosen": -1.3292646408081055, "logps/rejected": -61.644893646240234, "loss": 0.0881, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.276538848876953, "rewards/margins": 5.96080207824707, "rewards/rejected": 11.3157377243042, "step": 920 }, { "epoch": 0.8266666666666667, "grad_norm": 146.04498291015625, "learning_rate": 4.442374422065493e-07, "logits/chosen": 0.002922601066529751, "logits/rejected": -0.007130052894353867, "logps/chosen": -1.6069023609161377, "logps/rejected": -61.52588653564453, "loss": 0.1373, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.184974670410156, "rewards/margins": 5.7995734214782715, "rewards/rejected": 11.38540267944336, "step": 930 }, { "epoch": 0.8355555555555556, "grad_norm": 23.539485931396484, "learning_rate": 4.0107212810610974e-07, "logits/chosen": 0.0018056132830679417, "logits/rejected": -0.007847340777516365, "logps/chosen": -3.637047290802002, "logps/rejected": -61.21245574951172, "loss": 0.2763, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 16.960758209228516, "rewards/margins": 5.551811695098877, "rewards/rejected": 11.40894603729248, "step": 940 }, { "epoch": 0.8444444444444444, "grad_norm": 31.52926254272461, "learning_rate": 3.599293737426932e-07, "logits/chosen": 0.0032081177923828363, "logits/rejected": -0.007756482809782028, "logps/chosen": -1.172515869140625, "logps/rejected": -66.5853271484375, "loss": 0.1758, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 16.992876052856445, "rewards/margins": 5.886469841003418, "rewards/rejected": 11.106407165527344, "step": 950 }, { "epoch": 0.8533333333333334, "grad_norm": 0.07474468648433685, "learning_rate": 3.208488249181216e-07, "logits/chosen": 0.0022508346009999514, "logits/rejected": -0.009156409651041031, "logps/chosen": -0.7930470108985901, "logps/rejected": -60.068790435791016, "loss": 0.1138, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.401771545410156, "rewards/margins": 6.0026960372924805, "rewards/rejected": 11.399076461791992, "step": 960 }, { "epoch": 0.8533333333333334, "eval_logits/chosen": 0.001469604205340147, "eval_logits/rejected": -0.009397665038704872, "eval_logps/chosen": -1.8795456886291504, "eval_logps/rejected": -60.17564010620117, "eval_loss": 0.1441129744052887, "eval_rewards/accuracies": 0.9630000591278076, "eval_rewards/chosen": 17.314937591552734, "eval_rewards/margins": 5.9717817306518555, "eval_rewards/rejected": 11.343156814575195, "eval_runtime": 361.5344, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 960 }, { "epoch": 0.8622222222222222, "grad_norm": 68.90747833251953, "learning_rate": 2.838681402606952e-07, "logits/chosen": 0.004552370868623257, "logits/rejected": -0.005488495342433453, "logps/chosen": -3.7298974990844727, "logps/rejected": -64.72488403320312, "loss": 0.2425, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 16.832754135131836, "rewards/margins": 5.644216537475586, "rewards/rejected": 11.18853759765625, "step": 970 }, { "epoch": 0.8711111111111111, "grad_norm": 0.028206102550029755, "learning_rate": 2.490229549367443e-07, "logits/chosen": 0.0025807656347751617, "logits/rejected": -0.008657123893499374, "logps/chosen": -1.1825838088989258, "logps/rejected": -63.57493591308594, "loss": 0.0609, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.227130889892578, "rewards/margins": 6.053717136383057, "rewards/rejected": 11.173412322998047, "step": 980 }, { "epoch": 0.88, "grad_norm": 0.39742231369018555, "learning_rate": 2.1634684631203412e-07, "logits/chosen": 0.0048486413434147835, "logits/rejected": -0.006056814920157194, "logps/chosen": -2.919680595397949, "logps/rejected": -57.8403205871582, "loss": 0.3464, "rewards/accuracies": 0.8999999761581421, "rewards/chosen": 17.175275802612305, "rewards/margins": 5.559727668762207, "rewards/rejected": 11.615548133850098, "step": 990 }, { "epoch": 0.8888888888888888, "grad_norm": 30.22509002685547, "learning_rate": 1.8587130159608196e-07, "logits/chosen": 0.0050649940967559814, "logits/rejected": -0.0071399761363863945, "logps/chosen": -0.16131475567817688, "logps/rejected": -66.25190734863281, "loss": 0.0063, "rewards/accuracies": 1.0, "rewards/chosen": 17.313472747802734, "rewards/margins": 6.412895679473877, "rewards/rejected": 10.9005765914917, "step": 1000 }, { "epoch": 0.8977777777777778, "grad_norm": 53.02751922607422, "learning_rate": 1.5762568750059604e-07, "logits/chosen": 0.005312003195285797, "logits/rejected": -0.0038596936501562595, "logps/chosen": -4.00323486328125, "logps/rejected": -57.656890869140625, "loss": 0.2741, "rewards/accuracies": 0.925000011920929, "rewards/chosen": 17.167781829833984, "rewards/margins": 5.632095813751221, "rewards/rejected": 11.535685539245605, "step": 1010 }, { "epoch": 0.9066666666666666, "grad_norm": 119.56330108642578, "learning_rate": 1.316372219412454e-07, "logits/chosen": 0.004333779215812683, "logits/rejected": -0.007937717251479626, "logps/chosen": -0.10221245884895325, "logps/rejected": -67.09977722167969, "loss": 0.0513, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.151248931884766, "rewards/margins": 6.162137985229492, "rewards/rejected": 10.98911190032959, "step": 1020 }, { "epoch": 0.9066666666666666, "eval_logits/chosen": 0.004528110846877098, "eval_logits/rejected": -0.006501312367618084, "eval_logps/chosen": -1.817779302597046, "eval_logps/rejected": -60.34454345703125, "eval_loss": 0.14118175208568573, "eval_rewards/accuracies": 0.9610000848770142, "eval_rewards/chosen": 17.321117401123047, "eval_rewards/margins": 5.994848251342773, "eval_rewards/rejected": 11.32626724243164, "eval_runtime": 361.545, "eval_samples_per_second": 2.766, "eval_steps_per_second": 0.346, "step": 1020 }, { "epoch": 0.9155555555555556, "grad_norm": 0.9030271768569946, "learning_rate": 1.0793094781005792e-07, "logits/chosen": 0.0036234352737665176, "logits/rejected": -0.008390933275222778, "logps/chosen": -0.09899584949016571, "logps/rejected": -60.80555725097656, "loss": 0.0349, "rewards/accuracies": 0.987500011920929, "rewards/chosen": 17.619266510009766, "rewards/margins": 6.4814043045043945, "rewards/rejected": 11.137863159179688, "step": 1030 }, { "epoch": 0.9244444444444444, "grad_norm": 157.070068359375, "learning_rate": 8.652970884369255e-08, "logits/chosen": 0.005321115255355835, "logits/rejected": -0.004976513795554638, "logps/chosen": -1.3535833358764648, "logps/rejected": -61.2861328125, "loss": 0.1927, "rewards/accuracies": 0.9500000476837158, "rewards/chosen": 17.209800720214844, "rewards/margins": 5.795952796936035, "rewards/rejected": 11.413846015930176, "step": 1040 }, { "epoch": 0.9333333333333333, "grad_norm": 24.539953231811523, "learning_rate": 6.745412761086007e-08, "logits/chosen": 0.005845514126121998, "logits/rejected": -0.003921338357031345, "logps/chosen": -1.7223193645477295, "logps/rejected": -58.172142028808594, "loss": 0.1317, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.360435485839844, "rewards/margins": 5.838218688964844, "rewards/rejected": 11.522216796875, "step": 1050 }, { "epoch": 0.9422222222222222, "grad_norm": 0.061458222568035126, "learning_rate": 5.0722585640090305e-08, "logits/chosen": 0.003759522922337055, "logits/rejected": -0.0068417866714298725, "logps/chosen": -1.351855754852295, "logps/rejected": -59.56566619873047, "loss": 0.2279, "rewards/accuracies": 0.9375, "rewards/chosen": 17.245838165283203, "rewards/margins": 5.7038679122924805, "rewards/rejected": 11.541970252990723, "step": 1060 }, { "epoch": 0.9511111111111111, "grad_norm": 101.87332916259766, "learning_rate": 3.635120570700784e-08, "logits/chosen": 0.007607857696712017, "logits/rejected": -0.0035320711322128773, "logps/chosen": -1.4004669189453125, "logps/rejected": -63.46752166748047, "loss": 0.1021, "rewards/accuracies": 0.9624999761581421, "rewards/chosen": 17.240036010742188, "rewards/margins": 6.078882694244385, "rewards/rejected": 11.161155700683594, "step": 1070 }, { "epoch": 0.96, "grad_norm": 1.9229660034179688, "learning_rate": 2.4353836298169343e-08, "logits/chosen": 0.006209026090800762, "logits/rejected": -0.0033816141076385975, "logps/chosen": -2.0281822681427, "logps/rejected": -61.460670471191406, "loss": 0.1189, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.19972801208496, "rewards/margins": 5.881450653076172, "rewards/rejected": 11.318277359008789, "step": 1080 }, { "epoch": 0.96, "eval_logits/chosen": 0.00737445754930377, "eval_logits/rejected": -0.0035832570865750313, "eval_logps/chosen": -2.1419789791107178, "eval_logps/rejected": -60.60612106323242, "eval_loss": 0.15081512928009033, "eval_rewards/accuracies": 0.9610000848770142, "eval_rewards/chosen": 17.288694381713867, "eval_rewards/margins": 5.98858642578125, "eval_rewards/rejected": 11.300108909606934, "eval_runtime": 361.4165, "eval_samples_per_second": 2.767, "eval_steps_per_second": 0.346, "step": 1080 }, { "epoch": 0.9688888888888889, "grad_norm": 0.11888863146305084, "learning_rate": 1.4742038266447046e-08, "logits/chosen": 0.006043245084583759, "logits/rejected": -0.004712546244263649, "logps/chosen": -1.4127472639083862, "logps/rejected": -64.81529998779297, "loss": 0.0728, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.181049346923828, "rewards/margins": 6.14100456237793, "rewards/rejected": 11.040044784545898, "step": 1090 }, { "epoch": 0.9777777777777777, "grad_norm": 2.001819372177124, "learning_rate": 7.525073690809737e-09, "logits/chosen": 0.0055408780463039875, "logits/rejected": -0.005598037503659725, "logps/chosen": -1.017110824584961, "logps/rejected": -60.28044891357422, "loss": 0.0357, "rewards/accuracies": 0.9750000238418579, "rewards/chosen": 17.4679012298584, "rewards/margins": 6.193048477172852, "rewards/rejected": 11.274852752685547, "step": 1100 }, { "epoch": 0.9866666666666667, "grad_norm": 0.027056939899921417, "learning_rate": 2.709896951238744e-09, "logits/chosen": 0.00636716466397047, "logits/rejected": -0.004360577557235956, "logps/chosen": -2.8028905391693115, "logps/rejected": -60.86518859863281, "loss": 0.1973, "rewards/accuracies": 0.949999988079071, "rewards/chosen": 17.210033416748047, "rewards/margins": 5.936794281005859, "rewards/rejected": 11.273238182067871, "step": 1110 }, { "epoch": 0.9955555555555555, "grad_norm": 0.15637506544589996, "learning_rate": 3.0114802737818415e-10, "logits/chosen": 0.0038183885626494884, "logits/rejected": -0.007296917960047722, "logps/chosen": -1.1477452516555786, "logps/rejected": -59.37739181518555, "loss": 0.0538, "rewards/accuracies": 0.987500011920929, "rewards/chosen": 17.553251266479492, "rewards/margins": 6.2790141105651855, "rewards/rejected": 11.274236679077148, "step": 1120 } ], "logging_steps": 10, "max_steps": 1125, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 500, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": true }, "attributes": {} } }, "total_flos": 1.4338459346927616e+18, "train_batch_size": 4, "trial_name": null, "trial_params": null }