{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.4444444444444444, "eval_steps": 60, "global_step": 500, "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 } ], "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": false }, "attributes": {} } }, "total_flos": 6.372648598634496e+17, "train_batch_size": 4, "trial_name": null, "trial_params": null }