phi3m0128-cds-0.75-kendall-onof-ofif-corr-max-2-simpo-max1500-default
/
checkpoint-1450
/trainer_state.json
{ | |
"best_metric": null, | |
"best_model_checkpoint": null, | |
"epoch": 1.2877442273534636, | |
"eval_steps": 50, | |
"global_step": 1450, | |
"is_hyper_param_search": false, | |
"is_local_process_zero": true, | |
"is_world_process_zero": true, | |
"log_history": [ | |
{ | |
"epoch": 0.008880994671403197, | |
"grad_norm": 0.04571289196610451, | |
"learning_rate": 4.999451708687114e-06, | |
"logits/chosen": 14.56671142578125, | |
"logits/rejected": 15.112574577331543, | |
"logps/chosen": -0.26506316661834717, | |
"logps/rejected": -0.3439488410949707, | |
"loss": 0.9267, | |
"rewards/accuracies": 0.574999988079071, | |
"rewards/chosen": -0.39759472012519836, | |
"rewards/margins": 0.11832849681377411, | |
"rewards/rejected": -0.5159232020378113, | |
"step": 10 | |
}, | |
{ | |
"epoch": 0.017761989342806393, | |
"grad_norm": 0.0512714721262455, | |
"learning_rate": 4.997807075247147e-06, | |
"logits/chosen": 14.376543045043945, | |
"logits/rejected": 14.862703323364258, | |
"logps/chosen": -0.2708089351654053, | |
"logps/rejected": -0.32412824034690857, | |
"loss": 0.936, | |
"rewards/accuracies": 0.5, | |
"rewards/chosen": -0.4062133729457855, | |
"rewards/margins": 0.07997899502515793, | |
"rewards/rejected": -0.4861923158168793, | |
"step": 20 | |
}, | |
{ | |
"epoch": 0.02664298401420959, | |
"grad_norm": 0.058383647352457047, | |
"learning_rate": 4.9950668210706795e-06, | |
"logits/chosen": 14.208717346191406, | |
"logits/rejected": 15.370651245117188, | |
"logps/chosen": -0.28206294775009155, | |
"logps/rejected": -0.38387423753738403, | |
"loss": 0.9215, | |
"rewards/accuracies": 0.625, | |
"rewards/chosen": -0.42309442162513733, | |
"rewards/margins": 0.15271687507629395, | |
"rewards/rejected": -0.5758112668991089, | |
"step": 30 | |
}, | |
{ | |
"epoch": 0.035523978685612786, | |
"grad_norm": 0.06262075155973434, | |
"learning_rate": 4.9912321481237616e-06, | |
"logits/chosen": 14.768765449523926, | |
"logits/rejected": 15.169331550598145, | |
"logps/chosen": -0.27857059240341187, | |
"logps/rejected": -0.3388269543647766, | |
"loss": 0.9386, | |
"rewards/accuracies": 0.5, | |
"rewards/chosen": -0.4178559184074402, | |
"rewards/margins": 0.09038447588682175, | |
"rewards/rejected": -0.5082404017448425, | |
"step": 40 | |
}, | |
{ | |
"epoch": 0.04440497335701599, | |
"grad_norm": 0.06259036809206009, | |
"learning_rate": 4.986304738420684e-06, | |
"logits/chosen": 14.950456619262695, | |
"logits/rejected": 15.232122421264648, | |
"logps/chosen": -0.2961367070674896, | |
"logps/rejected": -0.3322262465953827, | |
"loss": 0.9317, | |
"rewards/accuracies": 0.44999998807907104, | |
"rewards/chosen": -0.44420504570007324, | |
"rewards/margins": 0.054134320467710495, | |
"rewards/rejected": -0.4983394145965576, | |
"step": 50 | |
}, | |
{ | |
"epoch": 0.04440497335701599, | |
"eval_logits/chosen": 14.56529426574707, | |
"eval_logits/rejected": 14.895020484924316, | |
"eval_logps/chosen": -0.2806546986103058, | |
"eval_logps/rejected": -0.3486972451210022, | |
"eval_loss": 0.9381324052810669, | |
"eval_rewards/accuracies": 0.5274725556373596, | |
"eval_rewards/chosen": -0.4209820330142975, | |
"eval_rewards/margins": 0.10206379741430283, | |
"eval_rewards/rejected": -0.5230458974838257, | |
"eval_runtime": 25.2574, | |
"eval_samples_per_second": 28.823, | |
"eval_steps_per_second": 3.603, | |
"step": 50 | |
}, | |
{ | |
"epoch": 0.05328596802841918, | |
"grad_norm": 0.07301533967256546, | |
"learning_rate": 4.980286753286196e-06, | |
"logits/chosen": 14.195574760437012, | |
"logits/rejected": 15.173194885253906, | |
"logps/chosen": -0.2693648636341095, | |
"logps/rejected": -0.33997970819473267, | |
"loss": 0.9319, | |
"rewards/accuracies": 0.512499988079071, | |
"rewards/chosen": -0.40404725074768066, | |
"rewards/margins": 0.10592226684093475, | |
"rewards/rejected": -0.5099694728851318, | |
"step": 60 | |
}, | |
{ | |
"epoch": 0.06216696269982238, | |
"grad_norm": 0.0659889206290245, | |
"learning_rate": 4.973180832407471e-06, | |
"logits/chosen": 14.910173416137695, | |
"logits/rejected": 15.361429214477539, | |
"logps/chosen": -0.28456225991249084, | |
"logps/rejected": -0.3702812194824219, | |
"loss": 0.9185, | |
"rewards/accuracies": 0.625, | |
"rewards/chosen": -0.42684346437454224, | |
"rewards/margins": 0.12857840955257416, | |
"rewards/rejected": -0.5554218292236328, | |
"step": 70 | |
}, | |
{ | |
"epoch": 0.07104795737122557, | |
"grad_norm": 0.05815625935792923, | |
"learning_rate": 4.964990092676263e-06, | |
"logits/chosen": 14.407182693481445, | |
"logits/rejected": 14.948204040527344, | |
"logps/chosen": -0.292889267206192, | |
"logps/rejected": -0.3381648063659668, | |
"loss": 0.9388, | |
"rewards/accuracies": 0.5375000238418579, | |
"rewards/chosen": -0.43933385610580444, | |
"rewards/margins": 0.06791339069604874, | |
"rewards/rejected": -0.5072472095489502, | |
"step": 80 | |
}, | |
{ | |
"epoch": 0.07992895204262877, | |
"grad_norm": 0.06627190113067627, | |
"learning_rate": 4.9557181268217225e-06, | |
"logits/chosen": 14.622471809387207, | |
"logits/rejected": 15.167770385742188, | |
"logps/chosen": -0.28155821561813354, | |
"logps/rejected": -0.33633899688720703, | |
"loss": 0.9256, | |
"rewards/accuracies": 0.5375000238418579, | |
"rewards/chosen": -0.4223373532295227, | |
"rewards/margins": 0.08217118680477142, | |
"rewards/rejected": -0.5045084953308105, | |
"step": 90 | |
}, | |
{ | |
"epoch": 0.08880994671403197, | |
"grad_norm": 0.0724545568227768, | |
"learning_rate": 4.9453690018345144e-06, | |
"logits/chosen": 14.289724349975586, | |
"logits/rejected": 14.882037162780762, | |
"logps/chosen": -0.2791440486907959, | |
"logps/rejected": -0.35329627990722656, | |
"loss": 0.9374, | |
"rewards/accuracies": 0.5375000238418579, | |
"rewards/chosen": -0.41871610283851624, | |
"rewards/margins": 0.11122839152812958, | |
"rewards/rejected": -0.5299445390701294, | |
"step": 100 | |
}, | |
{ | |
"epoch": 0.08880994671403197, | |
"eval_logits/chosen": 14.337930679321289, | |
"eval_logits/rejected": 14.689269065856934, | |
"eval_logps/chosen": -0.2726942300796509, | |
"eval_logps/rejected": -0.34668418765068054, | |
"eval_loss": 0.9302808046340942, | |
"eval_rewards/accuracies": 0.5384615659713745, | |
"eval_rewards/chosen": -0.40904131531715393, | |
"eval_rewards/margins": 0.11098497360944748, | |
"eval_rewards/rejected": -0.5200263261795044, | |
"eval_runtime": 25.2585, | |
"eval_samples_per_second": 28.822, | |
"eval_steps_per_second": 3.603, | |
"step": 100 | |
}, | |
{ | |
"epoch": 0.09769094138543517, | |
"grad_norm": 0.08156246691942215, | |
"learning_rate": 4.933947257182901e-06, | |
"logits/chosen": 14.499124526977539, | |
"logits/rejected": 14.916313171386719, | |
"logps/chosen": -0.2798352837562561, | |
"logps/rejected": -0.3477734327316284, | |
"loss": 0.9243, | |
"rewards/accuracies": 0.5625, | |
"rewards/chosen": -0.4197530150413513, | |
"rewards/margins": 0.10190720856189728, | |
"rewards/rejected": -0.5216602087020874, | |
"step": 110 | |
}, | |
{ | |
"epoch": 0.10657193605683836, | |
"grad_norm": 0.08161844313144684, | |
"learning_rate": 4.921457902821578e-06, | |
"logits/chosen": 13.595013618469238, | |
"logits/rejected": 14.390353202819824, | |
"logps/chosen": -0.26682502031326294, | |
"logps/rejected": -0.3336995542049408, | |
"loss": 0.9123, | |
"rewards/accuracies": 0.5874999761581421, | |
"rewards/chosen": -0.400237500667572, | |
"rewards/margins": 0.10031183809041977, | |
"rewards/rejected": -0.5005493760108948, | |
"step": 120 | |
}, | |
{ | |
"epoch": 0.11545293072824156, | |
"grad_norm": 0.28624778985977173, | |
"learning_rate": 4.907906416994146e-06, | |
"logits/chosen": 13.711044311523438, | |
"logits/rejected": 14.558542251586914, | |
"logps/chosen": -0.27874043583869934, | |
"logps/rejected": -0.3582325279712677, | |
"loss": 0.9163, | |
"rewards/accuracies": 0.5375000238418579, | |
"rewards/chosen": -0.41811060905456543, | |
"rewards/margins": 0.11923813819885254, | |
"rewards/rejected": -0.537348747253418, | |
"step": 130 | |
}, | |
{ | |
"epoch": 0.12433392539964476, | |
"grad_norm": 0.10971464216709137, | |
"learning_rate": 4.893298743830168e-06, | |
"logits/chosen": 14.18798828125, | |
"logits/rejected": 14.993026733398438, | |
"logps/chosen": -0.2750400900840759, | |
"logps/rejected": -0.39451608061790466, | |
"loss": 0.9098, | |
"rewards/accuracies": 0.574999988079071, | |
"rewards/chosen": -0.4125601351261139, | |
"rewards/margins": 0.17921395599842072, | |
"rewards/rejected": -0.5917741060256958, | |
"step": 140 | |
}, | |
{ | |
"epoch": 0.13321492007104796, | |
"grad_norm": 0.09321591258049011, | |
"learning_rate": 4.8776412907378845e-06, | |
"logits/chosen": 12.775139808654785, | |
"logits/rejected": 13.751996994018555, | |
"logps/chosen": -0.28446996212005615, | |
"logps/rejected": -0.36404967308044434, | |
"loss": 0.9104, | |
"rewards/accuracies": 0.5249999761581421, | |
"rewards/chosen": -0.42670494318008423, | |
"rewards/margins": 0.11936960369348526, | |
"rewards/rejected": -0.5460745096206665, | |
"step": 150 | |
}, | |
{ | |
"epoch": 0.13321492007104796, | |
"eval_logits/chosen": 12.97266960144043, | |
"eval_logits/rejected": 13.47339916229248, | |
"eval_logps/chosen": -0.27297571301460266, | |
"eval_logps/rejected": -0.36854612827301025, | |
"eval_loss": 0.9143257737159729, | |
"eval_rewards/accuracies": 0.5824176073074341, | |
"eval_rewards/chosen": -0.4094635546207428, | |
"eval_rewards/margins": 0.14335563778877258, | |
"eval_rewards/rejected": -0.5528191924095154, | |
"eval_runtime": 25.2406, | |
"eval_samples_per_second": 28.842, | |
"eval_steps_per_second": 3.605, | |
"step": 150 | |
}, | |
{ | |
"epoch": 0.14209591474245115, | |
"grad_norm": 0.11029861867427826, | |
"learning_rate": 4.860940925593703e-06, | |
"logits/chosen": 12.677947998046875, | |
"logits/rejected": 13.396716117858887, | |
"logps/chosen": -0.2631794512271881, | |
"logps/rejected": -0.37102141976356506, | |
"loss": 0.9051, | |
"rewards/accuracies": 0.550000011920929, | |
"rewards/chosen": -0.3947691321372986, | |
"rewards/margins": 0.161762997508049, | |
"rewards/rejected": -0.5565321445465088, | |
"step": 160 | |
}, | |
{ | |
"epoch": 0.15097690941385436, | |
"grad_norm": 0.15728294849395752, | |
"learning_rate": 4.84320497372973e-06, | |
"logits/chosen": 12.620219230651855, | |
"logits/rejected": 13.189640998840332, | |
"logps/chosen": -0.2947639524936676, | |
"logps/rejected": -0.3843482732772827, | |
"loss": 0.8906, | |
"rewards/accuracies": 0.5625, | |
"rewards/chosen": -0.442145973443985, | |
"rewards/margins": 0.13437646627426147, | |
"rewards/rejected": -0.5765224099159241, | |
"step": 170 | |
}, | |
{ | |
"epoch": 0.15985790408525755, | |
"grad_norm": 0.31504154205322266, | |
"learning_rate": 4.824441214720629e-06, | |
"logits/chosen": 11.487619400024414, | |
"logits/rejected": 12.33470344543457, | |
"logps/chosen": -0.271095871925354, | |
"logps/rejected": -0.4252637028694153, | |
"loss": 0.8766, | |
"rewards/accuracies": 0.5874999761581421, | |
"rewards/chosen": -0.406643807888031, | |
"rewards/margins": 0.2312517911195755, | |
"rewards/rejected": -0.6378955245018005, | |
"step": 180 | |
}, | |
{ | |
"epoch": 0.16873889875666073, | |
"grad_norm": 0.19222252070903778, | |
"learning_rate": 4.804657878971252e-06, | |
"logits/chosen": 10.093737602233887, | |
"logits/rejected": 10.851752281188965, | |
"logps/chosen": -0.2679918110370636, | |
"logps/rejected": -0.437336266040802, | |
"loss": 0.884, | |
"rewards/accuracies": 0.637499988079071, | |
"rewards/chosen": -0.4019877314567566, | |
"rewards/margins": 0.2540166974067688, | |
"rewards/rejected": -0.6560044288635254, | |
"step": 190 | |
}, | |
{ | |
"epoch": 0.17761989342806395, | |
"grad_norm": 0.2275688648223877, | |
"learning_rate": 4.783863644106502e-06, | |
"logits/chosen": 9.483477592468262, | |
"logits/rejected": 10.106366157531738, | |
"logps/chosen": -0.2957404553890228, | |
"logps/rejected": -0.40739065408706665, | |
"loss": 0.8767, | |
"rewards/accuracies": 0.5874999761581421, | |
"rewards/chosen": -0.4436107575893402, | |
"rewards/margins": 0.16747523844242096, | |
"rewards/rejected": -0.6110859513282776, | |
"step": 200 | |
}, | |
{ | |
"epoch": 0.17761989342806395, | |
"eval_logits/chosen": 8.491498947143555, | |
"eval_logits/rejected": 8.999146461486816, | |
"eval_logps/chosen": -0.3135836124420166, | |
"eval_logps/rejected": -0.4829566180706024, | |
"eval_loss": 0.8664290904998779, | |
"eval_rewards/accuracies": 0.6263736486434937, | |
"eval_rewards/chosen": -0.4703753888607025, | |
"eval_rewards/margins": 0.2540595233440399, | |
"eval_rewards/rejected": -0.7244349122047424, | |
"eval_runtime": 25.2553, | |
"eval_samples_per_second": 28.826, | |
"eval_steps_per_second": 3.603, | |
"step": 200 | |
}, | |
{ | |
"epoch": 0.18650088809946713, | |
"grad_norm": 0.27885496616363525, | |
"learning_rate": 4.762067631165049e-06, | |
"logits/chosen": 7.234966278076172, | |
"logits/rejected": 8.313450813293457, | |
"logps/chosen": -0.29102542996406555, | |
"logps/rejected": -0.49241799116134644, | |
"loss": 0.8556, | |
"rewards/accuracies": 0.637499988079071, | |
"rewards/chosen": -0.43653813004493713, | |
"rewards/margins": 0.3020888566970825, | |
"rewards/rejected": -0.7386269569396973, | |
"step": 210 | |
}, | |
{ | |
"epoch": 0.19538188277087035, | |
"grad_norm": 0.29907363653182983, | |
"learning_rate": 4.7392794005985324e-06, | |
"logits/chosen": 7.907521724700928, | |
"logits/rejected": 8.253190994262695, | |
"logps/chosen": -0.33691853284835815, | |
"logps/rejected": -0.4829257130622864, | |
"loss": 0.8236, | |
"rewards/accuracies": 0.550000011920929, | |
"rewards/chosen": -0.5053777694702148, | |
"rewards/margins": 0.21901080012321472, | |
"rewards/rejected": -0.724388599395752, | |
"step": 220 | |
}, | |
{ | |
"epoch": 0.20426287744227353, | |
"grad_norm": 0.282474547624588, | |
"learning_rate": 4.715508948078037e-06, | |
"logits/chosen": 6.367492198944092, | |
"logits/rejected": 6.273728370666504, | |
"logps/chosen": -0.3519875705242157, | |
"logps/rejected": -0.5284813642501831, | |
"loss": 0.8027, | |
"rewards/accuracies": 0.625, | |
"rewards/chosen": -0.5279813408851624, | |
"rewards/margins": 0.2647407650947571, | |
"rewards/rejected": -0.7927221059799194, | |
"step": 230 | |
}, | |
{ | |
"epoch": 0.21314387211367672, | |
"grad_norm": 0.327765554189682, | |
"learning_rate": 4.690766700109659e-06, | |
"logits/chosen": 5.090893268585205, | |
"logits/rejected": 4.768380165100098, | |
"logps/chosen": -0.3851698040962219, | |
"logps/rejected": -0.6464222073554993, | |
"loss": 0.7898, | |
"rewards/accuracies": 0.612500011920929, | |
"rewards/chosen": -0.5777546167373657, | |
"rewards/margins": 0.391878604888916, | |
"rewards/rejected": -0.9696332812309265, | |
"step": 240 | |
}, | |
{ | |
"epoch": 0.22202486678507993, | |
"grad_norm": 0.4895865321159363, | |
"learning_rate": 4.665063509461098e-06, | |
"logits/chosen": 4.056812286376953, | |
"logits/rejected": 3.723601818084717, | |
"logps/chosen": -0.4400455951690674, | |
"logps/rejected": -0.7731422781944275, | |
"loss": 0.7626, | |
"rewards/accuracies": 0.6625000238418579, | |
"rewards/chosen": -0.6600683927536011, | |
"rewards/margins": 0.49964505434036255, | |
"rewards/rejected": -1.1597135066986084, | |
"step": 250 | |
}, | |
{ | |
"epoch": 0.22202486678507993, | |
"eval_logits/chosen": 2.420060396194458, | |
"eval_logits/rejected": 2.1626052856445312, | |
"eval_logps/chosen": -0.4724067151546478, | |
"eval_logps/rejected": -0.8418064117431641, | |
"eval_loss": 0.7631083130836487, | |
"eval_rewards/accuracies": 0.6483516693115234, | |
"eval_rewards/chosen": -0.7086100578308105, | |
"eval_rewards/margins": 0.5540997385978699, | |
"eval_rewards/rejected": -1.2627097368240356, | |
"eval_runtime": 25.2418, | |
"eval_samples_per_second": 28.841, | |
"eval_steps_per_second": 3.605, | |
"step": 250 | |
}, | |
{ | |
"epoch": 0.23090586145648312, | |
"grad_norm": 0.46291017532348633, | |
"learning_rate": 4.638410650401267e-06, | |
"logits/chosen": 1.5297390222549438, | |
"logits/rejected": 1.1381648778915405, | |
"logps/chosen": -0.4418027997016907, | |
"logps/rejected": -1.0542564392089844, | |
"loss": 0.7026, | |
"rewards/accuracies": 0.762499988079071, | |
"rewards/chosen": -0.6627041697502136, | |
"rewards/margins": 0.9186803698539734, | |
"rewards/rejected": -1.5813844203948975, | |
"step": 260 | |
}, | |
{ | |
"epoch": 0.23978685612788633, | |
"grad_norm": 0.9783313870429993, | |
"learning_rate": 4.610819813755038e-06, | |
"logits/chosen": 2.8311033248901367, | |
"logits/rejected": 1.9742711782455444, | |
"logps/chosen": -0.5430587530136108, | |
"logps/rejected": -0.9841039776802063, | |
"loss": 0.7317, | |
"rewards/accuracies": 0.699999988079071, | |
"rewards/chosen": -0.814588189125061, | |
"rewards/margins": 0.6615679860115051, | |
"rewards/rejected": -1.4761559963226318, | |
"step": 270 | |
}, | |
{ | |
"epoch": 0.24866785079928952, | |
"grad_norm": 2.102562189102173, | |
"learning_rate": 4.582303101775249e-06, | |
"logits/chosen": 1.8241952657699585, | |
"logits/rejected": 0.8777934312820435, | |
"logps/chosen": -0.5624039769172668, | |
"logps/rejected": -1.1460126638412476, | |
"loss": 0.6887, | |
"rewards/accuracies": 0.6499999761581421, | |
"rewards/chosen": -0.8436058163642883, | |
"rewards/margins": 0.8754131197929382, | |
"rewards/rejected": -1.7190189361572266, | |
"step": 280 | |
}, | |
{ | |
"epoch": 0.25754884547069273, | |
"grad_norm": 0.9813026189804077, | |
"learning_rate": 4.55287302283426e-06, | |
"logits/chosen": 2.370732069015503, | |
"logits/rejected": 1.4697134494781494, | |
"logps/chosen": -0.6739786863327026, | |
"logps/rejected": -1.6581566333770752, | |
"loss": 0.5695, | |
"rewards/accuracies": 0.6625000238418579, | |
"rewards/chosen": -1.0109679698944092, | |
"rewards/margins": 1.476266622543335, | |
"rewards/rejected": -2.487234592437744, | |
"step": 290 | |
}, | |
{ | |
"epoch": 0.2664298401420959, | |
"grad_norm": 2.187314510345459, | |
"learning_rate": 4.522542485937369e-06, | |
"logits/chosen": 1.6230781078338623, | |
"logits/rejected": 0.5460122227668762, | |
"logps/chosen": -0.6433733701705933, | |
"logps/rejected": -2.1001811027526855, | |
"loss": 0.5366, | |
"rewards/accuracies": 0.762499988079071, | |
"rewards/chosen": -0.9650601148605347, | |
"rewards/margins": 2.185211658477783, | |
"rewards/rejected": -3.1502718925476074, | |
"step": 300 | |
}, | |
{ | |
"epoch": 0.2664298401420959, | |
"eval_logits/chosen": 1.4087599515914917, | |
"eval_logits/rejected": 0.7888947129249573, | |
"eval_logps/chosen": -0.7579545974731445, | |
"eval_logps/rejected": -2.0049116611480713, | |
"eval_loss": 0.551510214805603, | |
"eval_rewards/accuracies": 0.6813187003135681, | |
"eval_rewards/chosen": -1.1369318962097168, | |
"eval_rewards/margins": 1.8704355955123901, | |
"eval_rewards/rejected": -3.0073673725128174, | |
"eval_runtime": 25.2647, | |
"eval_samples_per_second": 28.815, | |
"eval_steps_per_second": 3.602, | |
"step": 300 | |
}, | |
{ | |
"epoch": 0.2753108348134991, | |
"grad_norm": 0.7035408616065979, | |
"learning_rate": 4.491324795060491e-06, | |
"logits/chosen": 1.5831315517425537, | |
"logits/rejected": 0.46730250120162964, | |
"logps/chosen": -0.7262418866157532, | |
"logps/rejected": -2.1209158897399902, | |
"loss": 0.5524, | |
"rewards/accuracies": 0.675000011920929, | |
"rewards/chosen": -1.0893628597259521, | |
"rewards/margins": 2.092010974884033, | |
"rewards/rejected": -3.1813735961914062, | |
"step": 310 | |
}, | |
{ | |
"epoch": 0.2841918294849023, | |
"grad_norm": 0.5678634643554688, | |
"learning_rate": 4.4592336433146e-06, | |
"logits/chosen": 1.265734076499939, | |
"logits/rejected": 0.7576489448547363, | |
"logps/chosen": -0.7938942313194275, | |
"logps/rejected": -2.3495612144470215, | |
"loss": 0.5233, | |
"rewards/accuracies": 0.6625000238418579, | |
"rewards/chosen": -1.1908413171768188, | |
"rewards/margins": 2.333500385284424, | |
"rewards/rejected": -3.5243420600891113, | |
"step": 320 | |
}, | |
{ | |
"epoch": 0.29307282415630553, | |
"grad_norm": 1.1373224258422852, | |
"learning_rate": 4.426283106939474e-06, | |
"logits/chosen": 2.977414846420288, | |
"logits/rejected": 2.1573710441589355, | |
"logps/chosen": -0.8513160943984985, | |
"logps/rejected": -2.4125566482543945, | |
"loss": 0.556, | |
"rewards/accuracies": 0.6875, | |
"rewards/chosen": -1.2769742012023926, | |
"rewards/margins": 2.341860771179199, | |
"rewards/rejected": -3.6188347339630127, | |
"step": 330 | |
}, | |
{ | |
"epoch": 0.3019538188277087, | |
"grad_norm": 4.7876176834106445, | |
"learning_rate": 4.3924876391293915e-06, | |
"logits/chosen": 2.4026589393615723, | |
"logits/rejected": 1.207395315170288, | |
"logps/chosen": -0.8529679179191589, | |
"logps/rejected": -2.456879138946533, | |
"loss": 0.5592, | |
"rewards/accuracies": 0.637499988079071, | |
"rewards/chosen": -1.279451847076416, | |
"rewards/margins": 2.4058666229248047, | |
"rewards/rejected": -3.6853187084198, | |
"step": 340 | |
}, | |
{ | |
"epoch": 0.3108348134991119, | |
"grad_norm": 0.5053763389587402, | |
"learning_rate": 4.357862063693486e-06, | |
"logits/chosen": 2.434265375137329, | |
"logits/rejected": 1.2504141330718994, | |
"logps/chosen": -0.9489291310310364, | |
"logps/rejected": -2.8521530628204346, | |
"loss": 0.4737, | |
"rewards/accuracies": 0.6625000238418579, | |
"rewards/chosen": -1.423393726348877, | |
"rewards/margins": 2.8548355102539062, | |
"rewards/rejected": -4.278229236602783, | |
"step": 350 | |
}, | |
{ | |
"epoch": 0.3108348134991119, | |
"eval_logits/chosen": 1.6632592678070068, | |
"eval_logits/rejected": 1.235045075416565, | |
"eval_logps/chosen": -1.0692518949508667, | |
"eval_logps/rejected": -2.7428486347198486, | |
"eval_loss": 0.5021397471427917, | |
"eval_rewards/accuracies": 0.692307710647583, | |
"eval_rewards/chosen": -1.6038777828216553, | |
"eval_rewards/margins": 2.510395050048828, | |
"eval_rewards/rejected": -4.1142730712890625, | |
"eval_runtime": 25.2582, | |
"eval_samples_per_second": 28.822, | |
"eval_steps_per_second": 3.603, | |
"step": 350 | |
}, | |
{ | |
"epoch": 0.3197158081705151, | |
"grad_norm": 0.8040274381637573, | |
"learning_rate": 4.322421568553529e-06, | |
"logits/chosen": 1.187036395072937, | |
"logits/rejected": 0.4290788769721985, | |
"logps/chosen": -1.1015206575393677, | |
"logps/rejected": -2.919748544692993, | |
"loss": 0.489, | |
"rewards/accuracies": 0.7250000238418579, | |
"rewards/chosen": -1.6522810459136963, | |
"rewards/margins": 2.727341651916504, | |
"rewards/rejected": -4.379622459411621, | |
"step": 360 | |
}, | |
{ | |
"epoch": 0.3285968028419183, | |
"grad_norm": 0.9299562573432922, | |
"learning_rate": 4.286181699082008e-06, | |
"logits/chosen": 2.5852127075195312, | |
"logits/rejected": 2.0419259071350098, | |
"logps/chosen": -1.1498607397079468, | |
"logps/rejected": -3.0336194038391113, | |
"loss": 0.4812, | |
"rewards/accuracies": 0.675000011920929, | |
"rewards/chosen": -1.724791169166565, | |
"rewards/margins": 2.8256375789642334, | |
"rewards/rejected": -4.55042839050293, | |
"step": 370 | |
}, | |
{ | |
"epoch": 0.33747779751332146, | |
"grad_norm": 1.7739671468734741, | |
"learning_rate": 4.249158351283414e-06, | |
"logits/chosen": 2.246245861053467, | |
"logits/rejected": 1.5551975965499878, | |
"logps/chosen": -1.254900574684143, | |
"logps/rejected": -3.206178665161133, | |
"loss": 0.4651, | |
"rewards/accuracies": 0.737500011920929, | |
"rewards/chosen": -1.8823509216308594, | |
"rewards/margins": 2.926917552947998, | |
"rewards/rejected": -4.809267997741699, | |
"step": 380 | |
}, | |
{ | |
"epoch": 0.3463587921847247, | |
"grad_norm": 4.380665302276611, | |
"learning_rate": 4.211367764821722e-06, | |
"logits/chosen": 3.0754549503326416, | |
"logits/rejected": 2.622124433517456, | |
"logps/chosen": -1.9250037670135498, | |
"logps/rejected": -3.69482421875, | |
"loss": 0.4292, | |
"rewards/accuracies": 0.800000011920929, | |
"rewards/chosen": -2.8875060081481934, | |
"rewards/margins": 2.6547305583953857, | |
"rewards/rejected": -5.542236328125, | |
"step": 390 | |
}, | |
{ | |
"epoch": 0.3552397868561279, | |
"grad_norm": 1.5087212324142456, | |
"learning_rate": 4.172826515897146e-06, | |
"logits/chosen": 2.2718021869659424, | |
"logits/rejected": 1.8861210346221924, | |
"logps/chosen": -2.4473955631256104, | |
"logps/rejected": -4.387387752532959, | |
"loss": 0.3902, | |
"rewards/accuracies": 0.8500000238418579, | |
"rewards/chosen": -3.671093702316284, | |
"rewards/margins": 2.9099888801574707, | |
"rewards/rejected": -6.581082344055176, | |
"step": 400 | |
}, | |
{ | |
"epoch": 0.3552397868561279, | |
"eval_logits/chosen": 1.759078860282898, | |
"eval_logits/rejected": 1.5246928930282593, | |
"eval_logps/chosen": -2.720665454864502, | |
"eval_logps/rejected": -4.613493919372559, | |
"eval_loss": 0.4054907560348511, | |
"eval_rewards/accuracies": 0.8791208863258362, | |
"eval_rewards/chosen": -4.080998420715332, | |
"eval_rewards/margins": 2.839242696762085, | |
"eval_rewards/rejected": -6.920241355895996, | |
"eval_runtime": 25.2363, | |
"eval_samples_per_second": 28.847, | |
"eval_steps_per_second": 3.606, | |
"step": 400 | |
}, | |
{ | |
"epoch": 0.3641207815275311, | |
"grad_norm": 6.079421043395996, | |
"learning_rate": 4.133551509975264e-06, | |
"logits/chosen": 1.8841949701309204, | |
"logits/rejected": 1.3479797840118408, | |
"logps/chosen": -2.517265796661377, | |
"logps/rejected": -4.453648567199707, | |
"loss": 0.3977, | |
"rewards/accuracies": 0.862500011920929, | |
"rewards/chosen": -3.7758986949920654, | |
"rewards/margins": 2.9045748710632324, | |
"rewards/rejected": -6.680473327636719, | |
"step": 410 | |
}, | |
{ | |
"epoch": 0.37300177619893427, | |
"grad_norm": 3.0998194217681885, | |
"learning_rate": 4.093559974371725e-06, | |
"logits/chosen": 1.6409276723861694, | |
"logits/rejected": 1.2141990661621094, | |
"logps/chosen": -2.2561168670654297, | |
"logps/rejected": -4.470211029052734, | |
"loss": 0.3527, | |
"rewards/accuracies": 0.875, | |
"rewards/chosen": -3.3841750621795654, | |
"rewards/margins": 3.321141004562378, | |
"rewards/rejected": -6.70531702041626, | |
"step": 420 | |
}, | |
{ | |
"epoch": 0.38188277087033745, | |
"grad_norm": 6.982161045074463, | |
"learning_rate": 4.052869450695776e-06, | |
"logits/chosen": 2.835188388824463, | |
"logits/rejected": 2.3657329082489014, | |
"logps/chosen": -2.8557300567626953, | |
"logps/rejected": -5.075521469116211, | |
"loss": 0.387, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -4.283595085144043, | |
"rewards/margins": 3.3296875953674316, | |
"rewards/rejected": -7.613282680511475, | |
"step": 430 | |
}, | |
{ | |
"epoch": 0.3907637655417407, | |
"grad_norm": 2.139338970184326, | |
"learning_rate": 4.011497787155938e-06, | |
"logits/chosen": 2.126509189605713, | |
"logits/rejected": 1.459567904472351, | |
"logps/chosen": -3.1412863731384277, | |
"logps/rejected": -5.423466682434082, | |
"loss": 0.3611, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -4.711928844451904, | |
"rewards/margins": 3.4232699871063232, | |
"rewards/rejected": -8.135198593139648, | |
"step": 440 | |
}, | |
{ | |
"epoch": 0.3996447602131439, | |
"grad_norm": 1.7899377346038818, | |
"learning_rate": 3.969463130731183e-06, | |
"logits/chosen": 2.4551379680633545, | |
"logits/rejected": 2.0784289836883545, | |
"logps/chosen": -3.098043203353882, | |
"logps/rejected": -5.300747871398926, | |
"loss": 0.354, | |
"rewards/accuracies": 0.8500000238418579, | |
"rewards/chosen": -4.647065162658691, | |
"rewards/margins": 3.3040566444396973, | |
"rewards/rejected": -7.951122283935547, | |
"step": 450 | |
}, | |
{ | |
"epoch": 0.3996447602131439, | |
"eval_logits/chosen": 1.9761625528335571, | |
"eval_logits/rejected": 1.6654667854309082, | |
"eval_logps/chosen": -2.8789772987365723, | |
"eval_logps/rejected": -5.1105055809021, | |
"eval_loss": 0.36211252212524414, | |
"eval_rewards/accuracies": 0.8791208863258362, | |
"eval_rewards/chosen": -4.3184661865234375, | |
"eval_rewards/margins": 3.347292423248291, | |
"eval_rewards/rejected": -7.6657586097717285, | |
"eval_runtime": 25.2549, | |
"eval_samples_per_second": 28.826, | |
"eval_steps_per_second": 3.603, | |
"step": 450 | |
}, | |
{ | |
"epoch": 0.40852575488454707, | |
"grad_norm": 1.9171936511993408, | |
"learning_rate": 3.92678391921108e-06, | |
"logits/chosen": 2.1672446727752686, | |
"logits/rejected": 1.6228408813476562, | |
"logps/chosen": -2.688931703567505, | |
"logps/rejected": -5.2408246994018555, | |
"loss": 0.3266, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -4.033397197723389, | |
"rewards/margins": 3.8278393745422363, | |
"rewards/rejected": -7.861237525939941, | |
"step": 460 | |
}, | |
{ | |
"epoch": 0.41740674955595025, | |
"grad_norm": 1.702635407447815, | |
"learning_rate": 3.88347887310836e-06, | |
"logits/chosen": 2.3164448738098145, | |
"logits/rejected": 2.047529697418213, | |
"logps/chosen": -2.6861701011657715, | |
"logps/rejected": -5.629918098449707, | |
"loss": 0.3339, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -4.02925443649292, | |
"rewards/margins": 4.415622711181641, | |
"rewards/rejected": -8.444877624511719, | |
"step": 470 | |
}, | |
{ | |
"epoch": 0.42628774422735344, | |
"grad_norm": 2.48634934425354, | |
"learning_rate": 3.839566987447492e-06, | |
"logits/chosen": 2.5225472450256348, | |
"logits/rejected": 2.0870003700256348, | |
"logps/chosen": -3.041111946105957, | |
"logps/rejected": -5.3499016761779785, | |
"loss": 0.3226, | |
"rewards/accuracies": 0.875, | |
"rewards/chosen": -4.561667442321777, | |
"rewards/margins": 3.4631850719451904, | |
"rewards/rejected": -8.024852752685547, | |
"step": 480 | |
}, | |
{ | |
"epoch": 0.4351687388987567, | |
"grad_norm": 4.728499412536621, | |
"learning_rate": 3.795067523432826e-06, | |
"logits/chosen": 2.33893084526062, | |
"logits/rejected": 1.7909936904907227, | |
"logps/chosen": -2.7356209754943848, | |
"logps/rejected": -5.33417272567749, | |
"loss": 0.322, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -4.103431224822998, | |
"rewards/margins": 3.8978283405303955, | |
"rewards/rejected": -8.001258850097656, | |
"step": 490 | |
}, | |
{ | |
"epoch": 0.44404973357015987, | |
"grad_norm": 8.412679672241211, | |
"learning_rate": 3.7500000000000005e-06, | |
"logits/chosen": 2.788668632507324, | |
"logits/rejected": 2.439873695373535, | |
"logps/chosen": -3.3219153881073, | |
"logps/rejected": -5.992051124572754, | |
"loss": 0.3075, | |
"rewards/accuracies": 0.875, | |
"rewards/chosen": -4.98287296295166, | |
"rewards/margins": 4.005204200744629, | |
"rewards/rejected": -8.988077163696289, | |
"step": 500 | |
}, | |
{ | |
"epoch": 0.44404973357015987, | |
"eval_logits/chosen": 2.165436029434204, | |
"eval_logits/rejected": 1.8186790943145752, | |
"eval_logps/chosen": -3.4299349784851074, | |
"eval_logps/rejected": -6.0660552978515625, | |
"eval_loss": 0.3319137990474701, | |
"eval_rewards/accuracies": 0.8901098966598511, | |
"eval_rewards/chosen": -5.14490270614624, | |
"eval_rewards/margins": 3.954181671142578, | |
"eval_rewards/rejected": -9.09908390045166, | |
"eval_runtime": 25.2602, | |
"eval_samples_per_second": 28.82, | |
"eval_steps_per_second": 3.603, | |
"step": 500 | |
}, | |
{ | |
"epoch": 0.45293072824156305, | |
"grad_norm": 2.8339133262634277, | |
"learning_rate": 3.7043841852542884e-06, | |
"logits/chosen": 2.109018325805664, | |
"logits/rejected": 1.6996265649795532, | |
"logps/chosen": -3.288560390472412, | |
"logps/rejected": -5.986764430999756, | |
"loss": 0.3116, | |
"rewards/accuracies": 0.862500011920929, | |
"rewards/chosen": -4.932840824127197, | |
"rewards/margins": 4.047306060791016, | |
"rewards/rejected": -8.980146408081055, | |
"step": 510 | |
}, | |
{ | |
"epoch": 0.46181172291296624, | |
"grad_norm": 3.8578269481658936, | |
"learning_rate": 3.658240087799655e-06, | |
"logits/chosen": 1.7659969329833984, | |
"logits/rejected": 1.4596515893936157, | |
"logps/chosen": -3.0301425457000732, | |
"logps/rejected": -6.252682209014893, | |
"loss": 0.3015, | |
"rewards/accuracies": 0.8999999761581421, | |
"rewards/chosen": -4.5452141761779785, | |
"rewards/margins": 4.833809852600098, | |
"rewards/rejected": -9.379022598266602, | |
"step": 520 | |
}, | |
{ | |
"epoch": 0.4706927175843694, | |
"grad_norm": 2.7795143127441406, | |
"learning_rate": 3.611587947962319e-06, | |
"logits/chosen": 2.472006320953369, | |
"logits/rejected": 1.9400993585586548, | |
"logps/chosen": -3.2479186058044434, | |
"logps/rejected": -6.475512504577637, | |
"loss": 0.3006, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -4.871877193450928, | |
"rewards/margins": 4.841391086578369, | |
"rewards/rejected": -9.713269233703613, | |
"step": 530 | |
}, | |
{ | |
"epoch": 0.47957371225577267, | |
"grad_norm": 3.5200746059417725, | |
"learning_rate": 3.564448228912682e-06, | |
"logits/chosen": 2.911531925201416, | |
"logits/rejected": 2.2947440147399902, | |
"logps/chosen": -3.387556791305542, | |
"logps/rejected": -6.559035301208496, | |
"loss": 0.2684, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.081335544586182, | |
"rewards/margins": 4.757218360900879, | |
"rewards/rejected": -9.838552474975586, | |
"step": 540 | |
}, | |
{ | |
"epoch": 0.48845470692717585, | |
"grad_norm": 2.368495225906372, | |
"learning_rate": 3.516841607689501e-06, | |
"logits/chosen": 1.276886224746704, | |
"logits/rejected": 1.3811718225479126, | |
"logps/chosen": -2.9689180850982666, | |
"logps/rejected": -6.619606971740723, | |
"loss": 0.3159, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -4.4533772468566895, | |
"rewards/margins": 5.4760332107543945, | |
"rewards/rejected": -9.929410934448242, | |
"step": 550 | |
}, | |
{ | |
"epoch": 0.48845470692717585, | |
"eval_logits/chosen": 2.2735824584960938, | |
"eval_logits/rejected": 1.9788992404937744, | |
"eval_logps/chosen": -3.698131799697876, | |
"eval_logps/rejected": -6.64966344833374, | |
"eval_loss": 0.30747923254966736, | |
"eval_rewards/accuracies": 0.9230769276618958, | |
"eval_rewards/chosen": -5.5471978187561035, | |
"eval_rewards/margins": 4.427298545837402, | |
"eval_rewards/rejected": -9.974496841430664, | |
"eval_runtime": 25.2322, | |
"eval_samples_per_second": 28.852, | |
"eval_steps_per_second": 3.606, | |
"step": 550 | |
}, | |
{ | |
"epoch": 0.49733570159857904, | |
"grad_norm": 1.901207685470581, | |
"learning_rate": 3.4687889661302577e-06, | |
"logits/chosen": 1.9734981060028076, | |
"logits/rejected": 1.8617655038833618, | |
"logps/chosen": -3.464953899383545, | |
"logps/rejected": -6.746106147766113, | |
"loss": 0.2912, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.1974310874938965, | |
"rewards/margins": 4.921727180480957, | |
"rewards/rejected": -10.119158744812012, | |
"step": 560 | |
}, | |
{ | |
"epoch": 0.5062166962699822, | |
"grad_norm": 3.526299238204956, | |
"learning_rate": 3.4203113817116955e-06, | |
"logits/chosen": 3.0836069583892822, | |
"logits/rejected": 2.75875186920166, | |
"logps/chosen": -3.5981743335723877, | |
"logps/rejected": -6.405775547027588, | |
"loss": 0.3284, | |
"rewards/accuracies": 0.862500011920929, | |
"rewards/chosen": -5.397261142730713, | |
"rewards/margins": 4.211403846740723, | |
"rewards/rejected": -9.608665466308594, | |
"step": 570 | |
}, | |
{ | |
"epoch": 0.5150976909413855, | |
"grad_norm": 2.816272497177124, | |
"learning_rate": 3.3714301183045382e-06, | |
"logits/chosen": 3.0068678855895996, | |
"logits/rejected": 2.466287136077881, | |
"logps/chosen": -3.7227580547332764, | |
"logps/rejected": -6.9311113357543945, | |
"loss": 0.2728, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.584136962890625, | |
"rewards/margins": 4.812530040740967, | |
"rewards/rejected": -10.39666748046875, | |
"step": 580 | |
}, | |
{ | |
"epoch": 0.5239786856127886, | |
"grad_norm": 2.433389902114868, | |
"learning_rate": 3.3221666168464584e-06, | |
"logits/chosen": 2.9992904663085938, | |
"logits/rejected": 2.678699254989624, | |
"logps/chosen": -3.540968418121338, | |
"logps/rejected": -7.228091239929199, | |
"loss": 0.2568, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -5.3114519119262695, | |
"rewards/margins": 5.530684947967529, | |
"rewards/rejected": -10.842137336730957, | |
"step": 590 | |
}, | |
{ | |
"epoch": 0.5328596802841918, | |
"grad_norm": 2.7557125091552734, | |
"learning_rate": 3.272542485937369e-06, | |
"logits/chosen": 2.557410478591919, | |
"logits/rejected": 2.331958770751953, | |
"logps/chosen": -3.9404635429382324, | |
"logps/rejected": -7.266766548156738, | |
"loss": 0.2639, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.9106950759887695, | |
"rewards/margins": 4.9894537925720215, | |
"rewards/rejected": -10.900148391723633, | |
"step": 600 | |
}, | |
{ | |
"epoch": 0.5328596802841918, | |
"eval_logits/chosen": 2.561415910720825, | |
"eval_logits/rejected": 2.2971484661102295, | |
"eval_logps/chosen": -4.015191555023193, | |
"eval_logps/rejected": -7.222255229949951, | |
"eval_loss": 0.28853774070739746, | |
"eval_rewards/accuracies": 0.9340659379959106, | |
"eval_rewards/chosen": -6.022787570953369, | |
"eval_rewards/margins": 4.810595989227295, | |
"eval_rewards/rejected": -10.833383560180664, | |
"eval_runtime": 25.2577, | |
"eval_samples_per_second": 28.823, | |
"eval_steps_per_second": 3.603, | |
"step": 600 | |
}, | |
{ | |
"epoch": 0.5417406749555951, | |
"grad_norm": 2.582770824432373, | |
"learning_rate": 3.222579492361179e-06, | |
"logits/chosen": 2.7404181957244873, | |
"logits/rejected": 2.540113687515259, | |
"logps/chosen": -3.9154396057128906, | |
"logps/rejected": -7.631985664367676, | |
"loss": 0.2816, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.873159408569336, | |
"rewards/margins": 5.574820041656494, | |
"rewards/rejected": -11.447979927062988, | |
"step": 610 | |
}, | |
{ | |
"epoch": 0.5506216696269982, | |
"grad_norm": 3.8167436122894287, | |
"learning_rate": 3.1722995515381644e-06, | |
"logits/chosen": 2.445218563079834, | |
"logits/rejected": 2.288620710372925, | |
"logps/chosen": -3.7501556873321533, | |
"logps/rejected": -7.918539524078369, | |
"loss": 0.2891, | |
"rewards/accuracies": 0.862500011920929, | |
"rewards/chosen": -5.625233173370361, | |
"rewards/margins": 6.252577304840088, | |
"rewards/rejected": -11.877809524536133, | |
"step": 620 | |
}, | |
{ | |
"epoch": 0.5595026642984015, | |
"grad_norm": 3.57536244392395, | |
"learning_rate": 3.121724717912138e-06, | |
"logits/chosen": 2.8337388038635254, | |
"logits/rejected": 2.1241557598114014, | |
"logps/chosen": -3.7040035724639893, | |
"logps/rejected": -7.53197717666626, | |
"loss": 0.2702, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.556005477905273, | |
"rewards/margins": 5.741961479187012, | |
"rewards/rejected": -11.297966003417969, | |
"step": 630 | |
}, | |
{ | |
"epoch": 0.5683836589698046, | |
"grad_norm": 3.0520713329315186, | |
"learning_rate": 3.0708771752766397e-06, | |
"logits/chosen": 2.5255160331726074, | |
"logits/rejected": 2.0742428302764893, | |
"logps/chosen": -3.908573865890503, | |
"logps/rejected": -7.60653829574585, | |
"loss": 0.2598, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.862860679626465, | |
"rewards/margins": 5.546946048736572, | |
"rewards/rejected": -11.409807205200195, | |
"step": 640 | |
}, | |
{ | |
"epoch": 0.5772646536412078, | |
"grad_norm": 8.067182540893555, | |
"learning_rate": 3.019779227044398e-06, | |
"logits/chosen": 1.6965067386627197, | |
"logits/rejected": 1.644667625427246, | |
"logps/chosen": -3.6588027477264404, | |
"logps/rejected": -7.457572937011719, | |
"loss": 0.2453, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -5.488204002380371, | |
"rewards/margins": 5.698155403137207, | |
"rewards/rejected": -11.186359405517578, | |
"step": 650 | |
}, | |
{ | |
"epoch": 0.5772646536412078, | |
"eval_logits/chosen": 2.577754020690918, | |
"eval_logits/rejected": 2.265626907348633, | |
"eval_logps/chosen": -3.906606435775757, | |
"eval_logps/rejected": -7.3099446296691895, | |
"eval_loss": 0.27302286028862, | |
"eval_rewards/accuracies": 0.9560439586639404, | |
"eval_rewards/chosen": -5.859910011291504, | |
"eval_rewards/margins": 5.105007171630859, | |
"eval_rewards/rejected": -10.964917182922363, | |
"eval_runtime": 25.2446, | |
"eval_samples_per_second": 28.838, | |
"eval_steps_per_second": 3.605, | |
"step": 650 | |
}, | |
{ | |
"epoch": 0.5861456483126111, | |
"grad_norm": 2.9298255443573, | |
"learning_rate": 2.9684532864643123e-06, | |
"logits/chosen": 2.901702404022217, | |
"logits/rejected": 2.595918655395508, | |
"logps/chosen": -3.9949543476104736, | |
"logps/rejected": -7.430356502532959, | |
"loss": 0.2726, | |
"rewards/accuracies": 0.8374999761581421, | |
"rewards/chosen": -5.992431163787842, | |
"rewards/margins": 5.153104782104492, | |
"rewards/rejected": -11.145535469055176, | |
"step": 660 | |
}, | |
{ | |
"epoch": 0.5950266429840142, | |
"grad_norm": 3.085571050643921, | |
"learning_rate": 2.9169218667902562e-06, | |
"logits/chosen": 2.8471388816833496, | |
"logits/rejected": 2.5500330924987793, | |
"logps/chosen": -4.014147758483887, | |
"logps/rejected": -7.298794746398926, | |
"loss": 0.2484, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -6.02122163772583, | |
"rewards/margins": 4.926970481872559, | |
"rewards/rejected": -10.94819164276123, | |
"step": 670 | |
}, | |
{ | |
"epoch": 0.6039076376554174, | |
"grad_norm": 2.3615477085113525, | |
"learning_rate": 2.8652075714060296e-06, | |
"logits/chosen": 2.495004177093506, | |
"logits/rejected": 1.9873936176300049, | |
"logps/chosen": -4.2625041007995605, | |
"logps/rejected": -8.186586380004883, | |
"loss": 0.2144, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -6.393756866455078, | |
"rewards/margins": 5.886124610900879, | |
"rewards/rejected": -12.279881477355957, | |
"step": 680 | |
}, | |
{ | |
"epoch": 0.6127886323268206, | |
"grad_norm": 3.497316837310791, | |
"learning_rate": 2.813333083910761e-06, | |
"logits/chosen": 1.540621042251587, | |
"logits/rejected": 1.2002273797988892, | |
"logps/chosen": -3.71490740776062, | |
"logps/rejected": -8.136199951171875, | |
"loss": 0.2861, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.572361469268799, | |
"rewards/margins": 6.631939888000488, | |
"rewards/rejected": -12.204300880432129, | |
"step": 690 | |
}, | |
{ | |
"epoch": 0.6216696269982238, | |
"grad_norm": 3.2540247440338135, | |
"learning_rate": 2.761321158169134e-06, | |
"logits/chosen": 2.776721477508545, | |
"logits/rejected": 2.475888729095459, | |
"logps/chosen": -4.3385329246521, | |
"logps/rejected": -7.8828301429748535, | |
"loss": 0.2885, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -6.507800102233887, | |
"rewards/margins": 5.316445350646973, | |
"rewards/rejected": -11.824244499206543, | |
"step": 700 | |
}, | |
{ | |
"epoch": 0.6216696269982238, | |
"eval_logits/chosen": 2.4267516136169434, | |
"eval_logits/rejected": 2.1245739459991455, | |
"eval_logps/chosen": -3.9584598541259766, | |
"eval_logps/rejected": -7.546706199645996, | |
"eval_loss": 0.26512712240219116, | |
"eval_rewards/accuracies": 0.9340659379959106, | |
"eval_rewards/chosen": -5.937689781188965, | |
"eval_rewards/margins": 5.382368564605713, | |
"eval_rewards/rejected": -11.320058822631836, | |
"eval_runtime": 25.2589, | |
"eval_samples_per_second": 28.822, | |
"eval_steps_per_second": 3.603, | |
"step": 700 | |
}, | |
{ | |
"epoch": 0.6305506216696269, | |
"grad_norm": 2.582273483276367, | |
"learning_rate": 2.70919460833079e-06, | |
"logits/chosen": 2.2102553844451904, | |
"logits/rejected": 1.9504516124725342, | |
"logps/chosen": -3.7470879554748535, | |
"logps/rejected": -7.761659145355225, | |
"loss": 0.2078, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -5.620632171630859, | |
"rewards/margins": 6.02185583114624, | |
"rewards/rejected": -11.642488479614258, | |
"step": 710 | |
}, | |
{ | |
"epoch": 0.6394316163410302, | |
"grad_norm": 2.2855663299560547, | |
"learning_rate": 2.6569762988232838e-06, | |
"logits/chosen": 2.4659409523010254, | |
"logits/rejected": 2.0434811115264893, | |
"logps/chosen": -3.4346442222595215, | |
"logps/rejected": -7.470824241638184, | |
"loss": 0.2346, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -5.151965618133545, | |
"rewards/margins": 6.054270267486572, | |
"rewards/rejected": -11.206236839294434, | |
"step": 720 | |
}, | |
{ | |
"epoch": 0.6483126110124334, | |
"grad_norm": 2.038733959197998, | |
"learning_rate": 2.604689134322999e-06, | |
"logits/chosen": 2.270310878753662, | |
"logits/rejected": 1.9651508331298828, | |
"logps/chosen": -3.721379518508911, | |
"logps/rejected": -7.776650905609131, | |
"loss": 0.2354, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.582068920135498, | |
"rewards/margins": 6.0829057693481445, | |
"rewards/rejected": -11.6649751663208, | |
"step": 730 | |
}, | |
{ | |
"epoch": 0.6571936056838366, | |
"grad_norm": 2.948915481567383, | |
"learning_rate": 2.5523560497083927e-06, | |
"logits/chosen": 2.358057737350464, | |
"logits/rejected": 2.063586711883545, | |
"logps/chosen": -3.6883864402770996, | |
"logps/rejected": -7.352984428405762, | |
"loss": 0.231, | |
"rewards/accuracies": 0.8999999761581421, | |
"rewards/chosen": -5.5325798988342285, | |
"rewards/margins": 5.496896266937256, | |
"rewards/rejected": -11.0294771194458, | |
"step": 740 | |
}, | |
{ | |
"epoch": 0.6660746003552398, | |
"grad_norm": 3.661870002746582, | |
"learning_rate": 2.5e-06, | |
"logits/chosen": 2.6712796688079834, | |
"logits/rejected": 2.393817901611328, | |
"logps/chosen": -4.065141201019287, | |
"logps/rejected": -7.9232306480407715, | |
"loss": 0.2332, | |
"rewards/accuracies": 0.875, | |
"rewards/chosen": -6.097712516784668, | |
"rewards/margins": 5.787134170532227, | |
"rewards/rejected": -11.884846687316895, | |
"step": 750 | |
}, | |
{ | |
"epoch": 0.6660746003552398, | |
"eval_logits/chosen": 2.4871790409088135, | |
"eval_logits/rejected": 2.2864174842834473, | |
"eval_logps/chosen": -4.176412105560303, | |
"eval_logps/rejected": -8.009041786193848, | |
"eval_loss": 0.2521994411945343, | |
"eval_rewards/accuracies": 0.9340659379959106, | |
"eval_rewards/chosen": -6.264617443084717, | |
"eval_rewards/margins": 5.7489447593688965, | |
"eval_rewards/rejected": -12.01356315612793, | |
"eval_runtime": 25.2624, | |
"eval_samples_per_second": 28.818, | |
"eval_steps_per_second": 3.602, | |
"step": 750 | |
}, | |
{ | |
"epoch": 0.6749555950266429, | |
"grad_norm": 1.992043375968933, | |
"learning_rate": 2.447643950291608e-06, | |
"logits/chosen": 1.8225914239883423, | |
"logits/rejected": 1.842559814453125, | |
"logps/chosen": -4.158253192901611, | |
"logps/rejected": -8.65810775756836, | |
"loss": 0.2104, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -6.237379550933838, | |
"rewards/margins": 6.749783515930176, | |
"rewards/rejected": -12.987161636352539, | |
"step": 760 | |
}, | |
{ | |
"epoch": 0.6838365896980462, | |
"grad_norm": 1.9012709856033325, | |
"learning_rate": 2.3953108656770018e-06, | |
"logits/chosen": 2.491041660308838, | |
"logits/rejected": 1.9490203857421875, | |
"logps/chosen": -4.141987323760986, | |
"logps/rejected": -8.790396690368652, | |
"loss": 0.2236, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -6.212981224060059, | |
"rewards/margins": 6.9726152420043945, | |
"rewards/rejected": -13.185595512390137, | |
"step": 770 | |
}, | |
{ | |
"epoch": 0.6927175843694494, | |
"grad_norm": 3.4377241134643555, | |
"learning_rate": 2.3430237011767166e-06, | |
"logits/chosen": 2.54376220703125, | |
"logits/rejected": 2.2269883155822754, | |
"logps/chosen": -3.7832961082458496, | |
"logps/rejected": -8.058969497680664, | |
"loss": 0.2449, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -5.674944877624512, | |
"rewards/margins": 6.413510322570801, | |
"rewards/rejected": -12.08845329284668, | |
"step": 780 | |
}, | |
{ | |
"epoch": 0.7015985790408525, | |
"grad_norm": 1.365478277206421, | |
"learning_rate": 2.290805391669212e-06, | |
"logits/chosen": 3.386944532394409, | |
"logits/rejected": 2.942134141921997, | |
"logps/chosen": -3.8051178455352783, | |
"logps/rejected": -8.201702117919922, | |
"loss": 0.1764, | |
"rewards/accuracies": 1.0, | |
"rewards/chosen": -5.707675933837891, | |
"rewards/margins": 6.594876289367676, | |
"rewards/rejected": -12.302552223205566, | |
"step": 790 | |
}, | |
{ | |
"epoch": 0.7104795737122558, | |
"grad_norm": 3.3714189529418945, | |
"learning_rate": 2.238678841830867e-06, | |
"logits/chosen": 2.690781593322754, | |
"logits/rejected": 2.2246782779693604, | |
"logps/chosen": -4.251287937164307, | |
"logps/rejected": -8.268930435180664, | |
"loss": 0.2429, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.376931667327881, | |
"rewards/margins": 6.026463031768799, | |
"rewards/rejected": -12.40339469909668, | |
"step": 800 | |
}, | |
{ | |
"epoch": 0.7104795737122558, | |
"eval_logits/chosen": 2.453395128250122, | |
"eval_logits/rejected": 2.2390329837799072, | |
"eval_logps/chosen": -4.038145065307617, | |
"eval_logps/rejected": -8.012212753295898, | |
"eval_loss": 0.2468602955341339, | |
"eval_rewards/accuracies": 0.9340659379959106, | |
"eval_rewards/chosen": -6.057217597961426, | |
"eval_rewards/margins": 5.961101055145264, | |
"eval_rewards/rejected": -12.018318176269531, | |
"eval_runtime": 25.2622, | |
"eval_samples_per_second": 28.818, | |
"eval_steps_per_second": 3.602, | |
"step": 800 | |
}, | |
{ | |
"epoch": 0.7193605683836589, | |
"grad_norm": 4.527968883514404, | |
"learning_rate": 2.186666916089239e-06, | |
"logits/chosen": 2.250300168991089, | |
"logits/rejected": 2.254126787185669, | |
"logps/chosen": -4.0314741134643555, | |
"logps/rejected": -7.800264835357666, | |
"loss": 0.2277, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -6.047211647033691, | |
"rewards/margins": 5.65318489074707, | |
"rewards/rejected": -11.700395584106445, | |
"step": 810 | |
}, | |
{ | |
"epoch": 0.7282415630550622, | |
"grad_norm": 3.3826537132263184, | |
"learning_rate": 2.134792428593971e-06, | |
"logits/chosen": 2.260606050491333, | |
"logits/rejected": 1.8174827098846436, | |
"logps/chosen": -3.6692988872528076, | |
"logps/rejected": -7.9340972900390625, | |
"loss": 0.2428, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -5.503948211669922, | |
"rewards/margins": 6.397197723388672, | |
"rewards/rejected": -11.901147842407227, | |
"step": 820 | |
}, | |
{ | |
"epoch": 0.7371225577264654, | |
"grad_norm": 5.055661678314209, | |
"learning_rate": 2.0830781332097446e-06, | |
"logits/chosen": 2.4391427040100098, | |
"logits/rejected": 2.0999953746795654, | |
"logps/chosen": -3.677706241607666, | |
"logps/rejected": -7.840268135070801, | |
"loss": 0.2288, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.516559600830078, | |
"rewards/margins": 6.243842124938965, | |
"rewards/rejected": -11.760400772094727, | |
"step": 830 | |
}, | |
{ | |
"epoch": 0.7460035523978685, | |
"grad_norm": 9.756246566772461, | |
"learning_rate": 2.031546713535688e-06, | |
"logits/chosen": 2.5405280590057373, | |
"logits/rejected": 2.3721981048583984, | |
"logps/chosen": -3.955289840698242, | |
"logps/rejected": -7.609635353088379, | |
"loss": 0.2384, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.932934761047363, | |
"rewards/margins": 5.481517314910889, | |
"rewards/rejected": -11.41445255279541, | |
"step": 840 | |
}, | |
{ | |
"epoch": 0.7548845470692718, | |
"grad_norm": 3.4563076496124268, | |
"learning_rate": 1.9802207729556023e-06, | |
"logits/chosen": 3.1546592712402344, | |
"logits/rejected": 2.6494877338409424, | |
"logps/chosen": -3.9850029945373535, | |
"logps/rejected": -8.197628021240234, | |
"loss": 0.2229, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -5.977504253387451, | |
"rewards/margins": 6.3189377784729, | |
"rewards/rejected": -12.296442031860352, | |
"step": 850 | |
}, | |
{ | |
"epoch": 0.7548845470692718, | |
"eval_logits/chosen": 2.536773443222046, | |
"eval_logits/rejected": 2.2915148735046387, | |
"eval_logps/chosen": -4.12998628616333, | |
"eval_logps/rejected": -8.214204788208008, | |
"eval_loss": 0.2361687868833542, | |
"eval_rewards/accuracies": 0.9340659379959106, | |
"eval_rewards/chosen": -6.194979190826416, | |
"eval_rewards/margins": 6.126327991485596, | |
"eval_rewards/rejected": -12.321306228637695, | |
"eval_runtime": 25.2618, | |
"eval_samples_per_second": 28.818, | |
"eval_steps_per_second": 3.602, | |
"step": 850 | |
}, | |
{ | |
"epoch": 0.7637655417406749, | |
"grad_norm": 2.1946489810943604, | |
"learning_rate": 1.9291228247233607e-06, | |
"logits/chosen": 1.6681854724884033, | |
"logits/rejected": 1.4272549152374268, | |
"logps/chosen": -3.6183536052703857, | |
"logps/rejected": -8.374089241027832, | |
"loss": 0.2142, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -5.427530288696289, | |
"rewards/margins": 7.133603572845459, | |
"rewards/rejected": -12.56113338470459, | |
"step": 860 | |
}, | |
{ | |
"epoch": 0.7726465364120781, | |
"grad_norm": 3.251861095428467, | |
"learning_rate": 1.8782752820878636e-06, | |
"logits/chosen": 3.6365694999694824, | |
"logits/rejected": 3.002140522003174, | |
"logps/chosen": -4.218847274780273, | |
"logps/rejected": -8.535767555236816, | |
"loss": 0.1927, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.328269958496094, | |
"rewards/margins": 6.475382328033447, | |
"rewards/rejected": -12.8036527633667, | |
"step": 870 | |
}, | |
{ | |
"epoch": 0.7815275310834814, | |
"grad_norm": 2.358853816986084, | |
"learning_rate": 1.827700448461836e-06, | |
"logits/chosen": 2.647890567779541, | |
"logits/rejected": 2.1392157077789307, | |
"logps/chosen": -4.547138214111328, | |
"logps/rejected": -9.499224662780762, | |
"loss": 0.1945, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -6.820706844329834, | |
"rewards/margins": 7.42812967300415, | |
"rewards/rejected": -14.2488374710083, | |
"step": 880 | |
}, | |
{ | |
"epoch": 0.7904085257548845, | |
"grad_norm": 2.04150390625, | |
"learning_rate": 1.7774205076388207e-06, | |
"logits/chosen": 2.7666563987731934, | |
"logits/rejected": 2.417945623397827, | |
"logps/chosen": -4.042238235473633, | |
"logps/rejected": -7.750130653381348, | |
"loss": 0.2228, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.063356876373291, | |
"rewards/margins": 5.561838626861572, | |
"rewards/rejected": -11.625195503234863, | |
"step": 890 | |
}, | |
{ | |
"epoch": 0.7992895204262878, | |
"grad_norm": 3.5930328369140625, | |
"learning_rate": 1.7274575140626318e-06, | |
"logits/chosen": 2.7729713916778564, | |
"logits/rejected": 2.426771640777588, | |
"logps/chosen": -4.228175163269043, | |
"logps/rejected": -8.267873764038086, | |
"loss": 0.2134, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.342263221740723, | |
"rewards/margins": 6.05954647064209, | |
"rewards/rejected": -12.401810646057129, | |
"step": 900 | |
}, | |
{ | |
"epoch": 0.7992895204262878, | |
"eval_logits/chosen": 2.5573580265045166, | |
"eval_logits/rejected": 2.317056179046631, | |
"eval_logps/chosen": -4.389712333679199, | |
"eval_logps/rejected": -8.52020263671875, | |
"eval_loss": 0.2303350269794464, | |
"eval_rewards/accuracies": 0.9450549483299255, | |
"eval_rewards/chosen": -6.584568023681641, | |
"eval_rewards/margins": 6.195735931396484, | |
"eval_rewards/rejected": -12.780303001403809, | |
"eval_runtime": 25.2651, | |
"eval_samples_per_second": 28.814, | |
"eval_steps_per_second": 3.602, | |
"step": 900 | |
}, | |
{ | |
"epoch": 0.8081705150976909, | |
"grad_norm": 3.19136118888855, | |
"learning_rate": 1.677833383153542e-06, | |
"logits/chosen": 2.2394022941589355, | |
"logits/rejected": 2.2709288597106934, | |
"logps/chosen": -4.338564872741699, | |
"logps/rejected": -8.621049880981445, | |
"loss": 0.2029, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.507847785949707, | |
"rewards/margins": 6.423727512359619, | |
"rewards/rejected": -12.931573867797852, | |
"step": 910 | |
}, | |
{ | |
"epoch": 0.8170515097690941, | |
"grad_norm": 2.785442590713501, | |
"learning_rate": 1.6285698816954626e-06, | |
"logits/chosen": 2.669541835784912, | |
"logits/rejected": 2.38188099861145, | |
"logps/chosen": -4.103513717651367, | |
"logps/rejected": -8.361233711242676, | |
"loss": 0.2054, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.155270576477051, | |
"rewards/margins": 6.386577606201172, | |
"rewards/rejected": -12.541848182678223, | |
"step": 920 | |
}, | |
{ | |
"epoch": 0.8259325044404974, | |
"grad_norm": 2.618563175201416, | |
"learning_rate": 1.5796886182883053e-06, | |
"logits/chosen": 2.5333399772644043, | |
"logits/rejected": 2.3547558784484863, | |
"logps/chosen": -3.701815128326416, | |
"logps/rejected": -8.315168380737305, | |
"loss": 0.1958, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -5.552721977233887, | |
"rewards/margins": 6.9200310707092285, | |
"rewards/rejected": -12.472753524780273, | |
"step": 930 | |
}, | |
{ | |
"epoch": 0.8348134991119005, | |
"grad_norm": 2.640681266784668, | |
"learning_rate": 1.5312110338697427e-06, | |
"logits/chosen": 2.6260218620300293, | |
"logits/rejected": 2.625610589981079, | |
"logps/chosen": -3.761087417602539, | |
"logps/rejected": -7.872547149658203, | |
"loss": 0.2147, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -5.64163064956665, | |
"rewards/margins": 6.167189598083496, | |
"rewards/rejected": -11.808821678161621, | |
"step": 940 | |
}, | |
{ | |
"epoch": 0.8436944937833037, | |
"grad_norm": 2.4091708660125732, | |
"learning_rate": 1.4831583923105e-06, | |
"logits/chosen": 2.63578462600708, | |
"logits/rejected": 2.2209556102752686, | |
"logps/chosen": -4.115639686584473, | |
"logps/rejected": -8.535353660583496, | |
"loss": 0.1832, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.173459529876709, | |
"rewards/margins": 6.629571437835693, | |
"rewards/rejected": -12.803030014038086, | |
"step": 950 | |
}, | |
{ | |
"epoch": 0.8436944937833037, | |
"eval_logits/chosen": 2.6680469512939453, | |
"eval_logits/rejected": 2.4164178371429443, | |
"eval_logps/chosen": -4.110969066619873, | |
"eval_logps/rejected": -8.377038955688477, | |
"eval_loss": 0.22471140325069427, | |
"eval_rewards/accuracies": 0.9560439586639404, | |
"eval_rewards/chosen": -6.166452884674072, | |
"eval_rewards/margins": 6.399104118347168, | |
"eval_rewards/rejected": -12.565558433532715, | |
"eval_runtime": 25.2357, | |
"eval_samples_per_second": 28.848, | |
"eval_steps_per_second": 3.606, | |
"step": 950 | |
}, | |
{ | |
"epoch": 0.8525754884547069, | |
"grad_norm": 5.924206733703613, | |
"learning_rate": 1.4355517710873184e-06, | |
"logits/chosen": 1.8501287698745728, | |
"logits/rejected": 1.9165807962417603, | |
"logps/chosen": -3.9117112159729004, | |
"logps/rejected": -8.62653923034668, | |
"loss": 0.2384, | |
"rewards/accuracies": 0.8999999761581421, | |
"rewards/chosen": -5.86756706237793, | |
"rewards/margins": 7.072243690490723, | |
"rewards/rejected": -12.939809799194336, | |
"step": 960 | |
}, | |
{ | |
"epoch": 0.8614564831261101, | |
"grad_norm": 9.17930793762207, | |
"learning_rate": 1.388412052037682e-06, | |
"logits/chosen": 3.1926915645599365, | |
"logits/rejected": 2.8990261554718018, | |
"logps/chosen": -3.6797282695770264, | |
"logps/rejected": -8.219030380249023, | |
"loss": 0.2089, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -5.51959228515625, | |
"rewards/margins": 6.808953285217285, | |
"rewards/rejected": -12.328545570373535, | |
"step": 970 | |
}, | |
{ | |
"epoch": 0.8703374777975134, | |
"grad_norm": 2.3526487350463867, | |
"learning_rate": 1.3417599122003464e-06, | |
"logits/chosen": 3.1998744010925293, | |
"logits/rejected": 2.904794692993164, | |
"logps/chosen": -4.254674911499023, | |
"logps/rejected": -8.549466133117676, | |
"loss": 0.1936, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.382012367248535, | |
"rewards/margins": 6.4421868324279785, | |
"rewards/rejected": -12.824198722839355, | |
"step": 980 | |
}, | |
{ | |
"epoch": 0.8792184724689165, | |
"grad_norm": 2.372730016708374, | |
"learning_rate": 1.2956158147457116e-06, | |
"logits/chosen": 2.3626601696014404, | |
"logits/rejected": 2.0850844383239746, | |
"logps/chosen": -4.239551544189453, | |
"logps/rejected": -8.695666313171387, | |
"loss": 0.1902, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.359326362609863, | |
"rewards/margins": 6.684172630310059, | |
"rewards/rejected": -13.043498039245605, | |
"step": 990 | |
}, | |
{ | |
"epoch": 0.8880994671403197, | |
"grad_norm": 3.703148365020752, | |
"learning_rate": 1.2500000000000007e-06, | |
"logits/chosen": 2.5446887016296387, | |
"logits/rejected": 2.2514405250549316, | |
"logps/chosen": -4.357522010803223, | |
"logps/rejected": -8.732219696044922, | |
"loss": 0.1807, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -6.53628396987915, | |
"rewards/margins": 6.562047004699707, | |
"rewards/rejected": -13.0983304977417, | |
"step": 1000 | |
}, | |
{ | |
"epoch": 0.8880994671403197, | |
"eval_logits/chosen": 2.6361231803894043, | |
"eval_logits/rejected": 2.4129557609558105, | |
"eval_logps/chosen": -4.334481239318848, | |
"eval_logps/rejected": -8.665815353393555, | |
"eval_loss": 0.22103577852249146, | |
"eval_rewards/accuracies": 0.9560439586639404, | |
"eval_rewards/chosen": -6.501720905303955, | |
"eval_rewards/margins": 6.497002601623535, | |
"eval_rewards/rejected": -12.998723030090332, | |
"eval_runtime": 25.2662, | |
"eval_samples_per_second": 28.813, | |
"eval_steps_per_second": 3.602, | |
"step": 1000 | |
}, | |
{ | |
"epoch": 0.8969804618117229, | |
"grad_norm": 3.239999532699585, | |
"learning_rate": 1.204932476567175e-06, | |
"logits/chosen": 2.8413162231445312, | |
"logits/rejected": 2.5842719078063965, | |
"logps/chosen": -3.9835076332092285, | |
"logps/rejected": -8.900358200073242, | |
"loss": 0.2276, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -5.975261688232422, | |
"rewards/margins": 7.375277042388916, | |
"rewards/rejected": -13.35053825378418, | |
"step": 1010 | |
}, | |
{ | |
"epoch": 0.9058614564831261, | |
"grad_norm": 3.8074100017547607, | |
"learning_rate": 1.160433012552508e-06, | |
"logits/chosen": 1.8096859455108643, | |
"logits/rejected": 1.5240850448608398, | |
"logps/chosen": -4.373281002044678, | |
"logps/rejected": -8.784773826599121, | |
"loss": 0.1907, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.559922218322754, | |
"rewards/margins": 6.6172380447387695, | |
"rewards/rejected": -13.177160263061523, | |
"step": 1020 | |
}, | |
{ | |
"epoch": 0.9147424511545293, | |
"grad_norm": 3.3684775829315186, | |
"learning_rate": 1.11652112689164e-06, | |
"logits/chosen": 2.5850119590759277, | |
"logits/rejected": 2.503263235092163, | |
"logps/chosen": -4.1612935066223145, | |
"logps/rejected": -9.02672004699707, | |
"loss": 0.2483, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.241940498352051, | |
"rewards/margins": 7.298140525817871, | |
"rewards/rejected": -13.540081977844238, | |
"step": 1030 | |
}, | |
{ | |
"epoch": 0.9236234458259325, | |
"grad_norm": 4.323958873748779, | |
"learning_rate": 1.073216080788921e-06, | |
"logits/chosen": 3.1654298305511475, | |
"logits/rejected": 2.52232027053833, | |
"logps/chosen": -4.379122734069824, | |
"logps/rejected": -8.86447525024414, | |
"loss": 0.1995, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.5686845779418945, | |
"rewards/margins": 6.728028774261475, | |
"rewards/rejected": -13.296712875366211, | |
"step": 1040 | |
}, | |
{ | |
"epoch": 0.9325044404973357, | |
"grad_norm": 2.923391342163086, | |
"learning_rate": 1.0305368692688175e-06, | |
"logits/chosen": 3.426027774810791, | |
"logits/rejected": 2.8201966285705566, | |
"logps/chosen": -4.2101850509643555, | |
"logps/rejected": -8.720430374145508, | |
"loss": 0.2275, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.315278053283691, | |
"rewards/margins": 6.765368461608887, | |
"rewards/rejected": -13.080645561218262, | |
"step": 1050 | |
}, | |
{ | |
"epoch": 0.9325044404973357, | |
"eval_logits/chosen": 2.7177271842956543, | |
"eval_logits/rejected": 2.45237135887146, | |
"eval_logps/chosen": -4.310559272766113, | |
"eval_logps/rejected": -8.67324447631836, | |
"eval_loss": 0.2168363630771637, | |
"eval_rewards/accuracies": 0.9670329689979553, | |
"eval_rewards/chosen": -6.465838432312012, | |
"eval_rewards/margins": 6.544029235839844, | |
"eval_rewards/rejected": -13.009867668151855, | |
"eval_runtime": 25.2426, | |
"eval_samples_per_second": 28.84, | |
"eval_steps_per_second": 3.605, | |
"step": 1050 | |
}, | |
{ | |
"epoch": 0.9413854351687388, | |
"grad_norm": 2.5395636558532715, | |
"learning_rate": 9.88502212844063e-07, | |
"logits/chosen": 2.634228467941284, | |
"logits/rejected": 2.447514772415161, | |
"logps/chosen": -4.163097381591797, | |
"logps/rejected": -9.294473648071289, | |
"loss": 0.214, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -6.244646072387695, | |
"rewards/margins": 7.697064399719238, | |
"rewards/rejected": -13.9417085647583, | |
"step": 1060 | |
}, | |
{ | |
"epoch": 0.9502664298401421, | |
"grad_norm": 2.4016919136047363, | |
"learning_rate": 9.471305493042243e-07, | |
"logits/chosen": 3.249488115310669, | |
"logits/rejected": 3.138061285018921, | |
"logps/chosen": -4.317265510559082, | |
"logps/rejected": -9.449271202087402, | |
"loss": 0.2135, | |
"rewards/accuracies": 0.8999999761581421, | |
"rewards/chosen": -6.475898742675781, | |
"rewards/margins": 7.6980085372924805, | |
"rewards/rejected": -14.173907279968262, | |
"step": 1070 | |
}, | |
{ | |
"epoch": 0.9591474245115453, | |
"grad_norm": 7.170376300811768, | |
"learning_rate": 9.064400256282757e-07, | |
"logits/chosen": 2.5836658477783203, | |
"logits/rejected": 2.584625482559204, | |
"logps/chosen": -3.731645107269287, | |
"logps/rejected": -8.913705825805664, | |
"loss": 0.2002, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.597468376159668, | |
"rewards/margins": 7.7730913162231445, | |
"rewards/rejected": -13.370559692382812, | |
"step": 1080 | |
}, | |
{ | |
"epoch": 0.9680284191829485, | |
"grad_norm": 4.712115287780762, | |
"learning_rate": 8.664484900247363e-07, | |
"logits/chosen": 1.7256263494491577, | |
"logits/rejected": 1.679078459739685, | |
"logps/chosen": -4.077349662780762, | |
"logps/rejected": -9.539873123168945, | |
"loss": 0.2498, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.116024494171143, | |
"rewards/margins": 8.193784713745117, | |
"rewards/rejected": -14.309808731079102, | |
"step": 1090 | |
}, | |
{ | |
"epoch": 0.9769094138543517, | |
"grad_norm": 4.168649673461914, | |
"learning_rate": 8.271734841028553e-07, | |
"logits/chosen": 3.224684953689575, | |
"logits/rejected": 2.831458568572998, | |
"logps/chosen": -4.505261421203613, | |
"logps/rejected": -9.360811233520508, | |
"loss": 0.1854, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.757891654968262, | |
"rewards/margins": 7.283324241638184, | |
"rewards/rejected": -14.041216850280762, | |
"step": 1100 | |
}, | |
{ | |
"epoch": 0.9769094138543517, | |
"eval_logits/chosen": 2.7460875511169434, | |
"eval_logits/rejected": 2.4983255863189697, | |
"eval_logps/chosen": -4.352819919586182, | |
"eval_logps/rejected": -8.75997257232666, | |
"eval_loss": 0.2113826721906662, | |
"eval_rewards/accuracies": 0.9780219793319702, | |
"eval_rewards/chosen": -6.529229640960693, | |
"eval_rewards/margins": 6.610728740692139, | |
"eval_rewards/rejected": -13.139957427978516, | |
"eval_runtime": 25.2523, | |
"eval_samples_per_second": 28.829, | |
"eval_steps_per_second": 3.604, | |
"step": 1100 | |
}, | |
{ | |
"epoch": 0.9857904085257548, | |
"grad_norm": 2.9944748878479004, | |
"learning_rate": 7.886322351782782e-07, | |
"logits/chosen": 2.92256236076355, | |
"logits/rejected": 2.4897289276123047, | |
"logps/chosen": -4.083096027374268, | |
"logps/rejected": -8.914036750793457, | |
"loss": 0.1986, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.124643325805664, | |
"rewards/margins": 7.2464118003845215, | |
"rewards/rejected": -13.371055603027344, | |
"step": 1110 | |
}, | |
{ | |
"epoch": 0.9946714031971581, | |
"grad_norm": 3.546858310699463, | |
"learning_rate": 7.508416487165862e-07, | |
"logits/chosen": 2.6310508251190186, | |
"logits/rejected": 2.64538836479187, | |
"logps/chosen": -4.345538139343262, | |
"logps/rejected": -9.072227478027344, | |
"loss": 0.1663, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.518306732177734, | |
"rewards/margins": 7.090034484863281, | |
"rewards/rejected": -13.6083402633667, | |
"step": 1120 | |
}, | |
{ | |
"epoch": 1.0035523978685612, | |
"grad_norm": 2.745328903198242, | |
"learning_rate": 7.138183009179922e-07, | |
"logits/chosen": 2.1050074100494385, | |
"logits/rejected": 2.0917954444885254, | |
"logps/chosen": -3.694542646408081, | |
"logps/rejected": -8.503544807434082, | |
"loss": 0.1985, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.541813850402832, | |
"rewards/margins": 7.213502407073975, | |
"rewards/rejected": -12.755315780639648, | |
"step": 1130 | |
}, | |
{ | |
"epoch": 1.0124333925399644, | |
"grad_norm": 1.995431661605835, | |
"learning_rate": 6.775784314464717e-07, | |
"logits/chosen": 2.7968056201934814, | |
"logits/rejected": 2.488126754760742, | |
"logps/chosen": -4.206582546234131, | |
"logps/rejected": -8.700386047363281, | |
"loss": 0.2141, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -6.309874534606934, | |
"rewards/margins": 6.740703582763672, | |
"rewards/rejected": -13.050578117370605, | |
"step": 1140 | |
}, | |
{ | |
"epoch": 1.0213143872113677, | |
"grad_norm": 3.110943078994751, | |
"learning_rate": 6.421379363065142e-07, | |
"logits/chosen": 2.5061817169189453, | |
"logits/rejected": 2.342125415802002, | |
"logps/chosen": -4.2504048347473145, | |
"logps/rejected": -8.307449340820312, | |
"loss": 0.2069, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.375607013702393, | |
"rewards/margins": 6.085566520690918, | |
"rewards/rejected": -12.461172103881836, | |
"step": 1150 | |
}, | |
{ | |
"epoch": 1.0213143872113677, | |
"eval_logits/chosen": 2.728573799133301, | |
"eval_logits/rejected": 2.5072758197784424, | |
"eval_logps/chosen": -4.397165298461914, | |
"eval_logps/rejected": -8.84023380279541, | |
"eval_loss": 0.20922726392745972, | |
"eval_rewards/accuracies": 0.9780219793319702, | |
"eval_rewards/chosen": -6.595747947692871, | |
"eval_rewards/margins": 6.664604663848877, | |
"eval_rewards/rejected": -13.26035213470459, | |
"eval_runtime": 25.264, | |
"eval_samples_per_second": 28.816, | |
"eval_steps_per_second": 3.602, | |
"step": 1150 | |
}, | |
{ | |
"epoch": 1.030195381882771, | |
"grad_norm": 3.401503086090088, | |
"learning_rate": 6.075123608706093e-07, | |
"logits/chosen": 2.933258533477783, | |
"logits/rejected": 2.5766117572784424, | |
"logps/chosen": -4.0520782470703125, | |
"logps/rejected": -9.317255973815918, | |
"loss": 0.2031, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.078117847442627, | |
"rewards/margins": 7.897767066955566, | |
"rewards/rejected": -13.975885391235352, | |
"step": 1160 | |
}, | |
{ | |
"epoch": 1.0390763765541742, | |
"grad_norm": 2.563494920730591, | |
"learning_rate": 5.737168930605272e-07, | |
"logits/chosen": 2.8566126823425293, | |
"logits/rejected": 2.501013994216919, | |
"logps/chosen": -4.048529148101807, | |
"logps/rejected": -8.981905937194824, | |
"loss": 0.2078, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.072793483734131, | |
"rewards/margins": 7.400063991546631, | |
"rewards/rejected": -13.472857475280762, | |
"step": 1170 | |
}, | |
{ | |
"epoch": 1.0479573712255772, | |
"grad_norm": 8.221929550170898, | |
"learning_rate": 5.407663566854008e-07, | |
"logits/chosen": 1.8076190948486328, | |
"logits/rejected": 1.5266883373260498, | |
"logps/chosen": -3.697718858718872, | |
"logps/rejected": -8.30528450012207, | |
"loss": 0.1869, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -5.546578407287598, | |
"rewards/margins": 6.911346435546875, | |
"rewards/rejected": -12.457925796508789, | |
"step": 1180 | |
}, | |
{ | |
"epoch": 1.0568383658969804, | |
"grad_norm": 6.9954400062561035, | |
"learning_rate": 5.086752049395094e-07, | |
"logits/chosen": 2.3529181480407715, | |
"logits/rejected": 2.250333786010742, | |
"logps/chosen": -4.018799781799316, | |
"logps/rejected": -8.714046478271484, | |
"loss": 0.2026, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.028199195861816, | |
"rewards/margins": 7.042870998382568, | |
"rewards/rejected": -13.071070671081543, | |
"step": 1190 | |
}, | |
{ | |
"epoch": 1.0657193605683837, | |
"grad_norm": 3.6167593002319336, | |
"learning_rate": 4.774575140626317e-07, | |
"logits/chosen": 2.7552218437194824, | |
"logits/rejected": 2.428560733795166, | |
"logps/chosen": -4.324695587158203, | |
"logps/rejected": -9.764104843139648, | |
"loss": 0.2233, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.487043857574463, | |
"rewards/margins": 8.159114837646484, | |
"rewards/rejected": -14.646158218383789, | |
"step": 1200 | |
}, | |
{ | |
"epoch": 1.0657193605683837, | |
"eval_logits/chosen": 2.7005860805511475, | |
"eval_logits/rejected": 2.4944183826446533, | |
"eval_logps/chosen": -4.283793926239014, | |
"eval_logps/rejected": -8.801348686218262, | |
"eval_loss": 0.20932108163833618, | |
"eval_rewards/accuracies": 0.9560439586639404, | |
"eval_rewards/chosen": -6.425690650939941, | |
"eval_rewards/margins": 6.776332855224609, | |
"eval_rewards/rejected": -13.202025413513184, | |
"eval_runtime": 25.2698, | |
"eval_samples_per_second": 28.809, | |
"eval_steps_per_second": 3.601, | |
"step": 1200 | |
}, | |
{ | |
"epoch": 1.074600355239787, | |
"grad_norm": 3.2438127994537354, | |
"learning_rate": 4.4712697716573994e-07, | |
"logits/chosen": 1.9486421346664429, | |
"logits/rejected": 1.7296040058135986, | |
"logps/chosen": -3.9212806224823, | |
"logps/rejected": -8.304320335388184, | |
"loss": 0.1761, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -5.881921291351318, | |
"rewards/margins": 6.574559688568115, | |
"rewards/rejected": -12.45648193359375, | |
"step": 1210 | |
}, | |
{ | |
"epoch": 1.0834813499111902, | |
"grad_norm": 2.984710454940796, | |
"learning_rate": 4.1769689822475147e-07, | |
"logits/chosen": 3.56829833984375, | |
"logits/rejected": 2.9676673412323, | |
"logps/chosen": -4.453071117401123, | |
"logps/rejected": -8.549017906188965, | |
"loss": 0.1915, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.6796064376831055, | |
"rewards/margins": 6.1439208984375, | |
"rewards/rejected": -12.823527336120605, | |
"step": 1220 | |
}, | |
{ | |
"epoch": 1.0923623445825932, | |
"grad_norm": 1.0289857387542725, | |
"learning_rate": 3.891801862449629e-07, | |
"logits/chosen": 2.529848098754883, | |
"logits/rejected": 2.3582332134246826, | |
"logps/chosen": -3.964491367340088, | |
"logps/rejected": -9.1004638671875, | |
"loss": 0.1743, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -5.946737289428711, | |
"rewards/margins": 7.703960418701172, | |
"rewards/rejected": -13.650697708129883, | |
"step": 1230 | |
}, | |
{ | |
"epoch": 1.1012433392539964, | |
"grad_norm": 3.8722472190856934, | |
"learning_rate": 3.615893495987335e-07, | |
"logits/chosen": 2.980180263519287, | |
"logits/rejected": 2.823392152786255, | |
"logps/chosen": -4.560516357421875, | |
"logps/rejected": -9.179193496704102, | |
"loss": 0.195, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -6.8407745361328125, | |
"rewards/margins": 6.928016662597656, | |
"rewards/rejected": -13.768791198730469, | |
"step": 1240 | |
}, | |
{ | |
"epoch": 1.1101243339253997, | |
"grad_norm": 3.892000198364258, | |
"learning_rate": 3.3493649053890325e-07, | |
"logits/chosen": 2.8612444400787354, | |
"logits/rejected": 2.5222373008728027, | |
"logps/chosen": -4.044814586639404, | |
"logps/rejected": -8.645524978637695, | |
"loss": 0.1571, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.067221641540527, | |
"rewards/margins": 6.901064872741699, | |
"rewards/rejected": -12.968287467956543, | |
"step": 1250 | |
}, | |
{ | |
"epoch": 1.1101243339253997, | |
"eval_logits/chosen": 2.714733839035034, | |
"eval_logits/rejected": 2.5066633224487305, | |
"eval_logps/chosen": -4.303951263427734, | |
"eval_logps/rejected": -8.823600769042969, | |
"eval_loss": 0.2080618292093277, | |
"eval_rewards/accuracies": 0.9560439586639404, | |
"eval_rewards/chosen": -6.45592737197876, | |
"eval_rewards/margins": 6.77947473526001, | |
"eval_rewards/rejected": -13.235400199890137, | |
"eval_runtime": 25.2645, | |
"eval_samples_per_second": 28.815, | |
"eval_steps_per_second": 3.602, | |
"step": 1250 | |
}, | |
{ | |
"epoch": 1.119005328596803, | |
"grad_norm": 3.225729465484619, | |
"learning_rate": 3.092332998903416e-07, | |
"logits/chosen": 2.969749927520752, | |
"logits/rejected": 2.601804733276367, | |
"logps/chosen": -4.048078536987305, | |
"logps/rejected": -8.416013717651367, | |
"loss": 0.2137, | |
"rewards/accuracies": 0.9125000238418579, | |
"rewards/chosen": -6.072117805480957, | |
"rewards/margins": 6.55190372467041, | |
"rewards/rejected": -12.62402057647705, | |
"step": 1260 | |
}, | |
{ | |
"epoch": 1.1278863232682061, | |
"grad_norm": 3.2009847164154053, | |
"learning_rate": 2.844910519219632e-07, | |
"logits/chosen": 2.962148904800415, | |
"logits/rejected": 2.33172607421875, | |
"logps/chosen": -4.167851448059082, | |
"logps/rejected": -8.694719314575195, | |
"loss": 0.2259, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.2517781257629395, | |
"rewards/margins": 6.790299892425537, | |
"rewards/rejected": -13.042078971862793, | |
"step": 1270 | |
}, | |
{ | |
"epoch": 1.1367673179396092, | |
"grad_norm": 11.98656940460205, | |
"learning_rate": 2.6072059940146775e-07, | |
"logits/chosen": 2.3897571563720703, | |
"logits/rejected": 2.1554014682769775, | |
"logps/chosen": -4.117510795593262, | |
"logps/rejected": -9.516136169433594, | |
"loss": 0.1657, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.176266193389893, | |
"rewards/margins": 8.097938537597656, | |
"rewards/rejected": -14.274205207824707, | |
"step": 1280 | |
}, | |
{ | |
"epoch": 1.1456483126110124, | |
"grad_norm": 2.14774489402771, | |
"learning_rate": 2.3793236883495164e-07, | |
"logits/chosen": 2.2359044551849365, | |
"logits/rejected": 2.196018934249878, | |
"logps/chosen": -4.038699150085449, | |
"logps/rejected": -8.689764022827148, | |
"loss": 0.1965, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.058049201965332, | |
"rewards/margins": 6.976596832275391, | |
"rewards/rejected": -13.034646987915039, | |
"step": 1290 | |
}, | |
{ | |
"epoch": 1.1545293072824157, | |
"grad_norm": 3.720881462097168, | |
"learning_rate": 2.1613635589349756e-07, | |
"logits/chosen": 2.0333285331726074, | |
"logits/rejected": 1.8229951858520508, | |
"logps/chosen": -3.95289945602417, | |
"logps/rejected": -9.260812759399414, | |
"loss": 0.174, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -5.929348945617676, | |
"rewards/margins": 7.9618706703186035, | |
"rewards/rejected": -13.891218185424805, | |
"step": 1300 | |
}, | |
{ | |
"epoch": 1.1545293072824157, | |
"eval_logits/chosen": 2.7231945991516113, | |
"eval_logits/rejected": 2.5083000659942627, | |
"eval_logps/chosen": -4.294267177581787, | |
"eval_logps/rejected": -8.827666282653809, | |
"eval_loss": 0.20604319870471954, | |
"eval_rewards/accuracies": 0.9670329689979553, | |
"eval_rewards/chosen": -6.44140100479126, | |
"eval_rewards/margins": 6.800098419189453, | |
"eval_rewards/rejected": -13.241500854492188, | |
"eval_runtime": 25.2569, | |
"eval_samples_per_second": 28.824, | |
"eval_steps_per_second": 3.603, | |
"step": 1300 | |
}, | |
{ | |
"epoch": 1.163410301953819, | |
"grad_norm": 5.726924896240234, | |
"learning_rate": 1.95342121028749e-07, | |
"logits/chosen": 3.07940936088562, | |
"logits/rejected": 2.9031105041503906, | |
"logps/chosen": -4.019009590148926, | |
"logps/rejected": -9.042330741882324, | |
"loss": 0.1857, | |
"rewards/accuracies": 0.925000011920929, | |
"rewards/chosen": -6.028514385223389, | |
"rewards/margins": 7.534982204437256, | |
"rewards/rejected": -13.563494682312012, | |
"step": 1310 | |
}, | |
{ | |
"epoch": 1.1722912966252221, | |
"grad_norm": 4.062259197235107, | |
"learning_rate": 1.7555878527937164e-07, | |
"logits/chosen": 1.7906630039215088, | |
"logits/rejected": 1.741253137588501, | |
"logps/chosen": -3.8779044151306152, | |
"logps/rejected": -9.477428436279297, | |
"loss": 0.1816, | |
"rewards/accuracies": 0.8999999761581421, | |
"rewards/chosen": -5.816856384277344, | |
"rewards/margins": 8.399286270141602, | |
"rewards/rejected": -14.216143608093262, | |
"step": 1320 | |
}, | |
{ | |
"epoch": 1.1811722912966252, | |
"grad_norm": 4.922858715057373, | |
"learning_rate": 1.567950262702714e-07, | |
"logits/chosen": 2.7542059421539307, | |
"logits/rejected": 2.369173288345337, | |
"logps/chosen": -4.438471794128418, | |
"logps/rejected": -9.667543411254883, | |
"loss": 0.1965, | |
"rewards/accuracies": 0.887499988079071, | |
"rewards/chosen": -6.657707214355469, | |
"rewards/margins": 7.8436079025268555, | |
"rewards/rejected": -14.501315116882324, | |
"step": 1330 | |
}, | |
{ | |
"epoch": 1.1900532859680284, | |
"grad_norm": 3.347116708755493, | |
"learning_rate": 1.3905907440629752e-07, | |
"logits/chosen": 2.6956562995910645, | |
"logits/rejected": 2.3373332023620605, | |
"logps/chosen": -3.7666268348693848, | |
"logps/rejected": -9.006341934204102, | |
"loss": 0.182, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -5.64993953704834, | |
"rewards/margins": 7.859574317932129, | |
"rewards/rejected": -13.509511947631836, | |
"step": 1340 | |
}, | |
{ | |
"epoch": 1.1989342806394316, | |
"grad_norm": 3.2019412517547607, | |
"learning_rate": 1.223587092621162e-07, | |
"logits/chosen": 2.480767011642456, | |
"logits/rejected": 2.2258126735687256, | |
"logps/chosen": -4.096011161804199, | |
"logps/rejected": -8.5545072555542, | |
"loss": 0.2274, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.144017219543457, | |
"rewards/margins": 6.687742710113525, | |
"rewards/rejected": -12.831761360168457, | |
"step": 1350 | |
}, | |
{ | |
"epoch": 1.1989342806394316, | |
"eval_logits/chosen": 2.7050983905792236, | |
"eval_logits/rejected": 2.495755434036255, | |
"eval_logps/chosen": -4.344603061676025, | |
"eval_logps/rejected": -8.87659740447998, | |
"eval_loss": 0.20770707726478577, | |
"eval_rewards/accuracies": 0.9780219793319702, | |
"eval_rewards/chosen": -6.516905307769775, | |
"eval_rewards/margins": 6.7979912757873535, | |
"eval_rewards/rejected": -13.314897537231445, | |
"eval_runtime": 25.266, | |
"eval_samples_per_second": 28.813, | |
"eval_steps_per_second": 3.602, | |
"step": 1350 | |
}, | |
{ | |
"epoch": 1.2078152753108349, | |
"grad_norm": 6.736517906188965, | |
"learning_rate": 1.067012561698319e-07, | |
"logits/chosen": 2.2384419441223145, | |
"logits/rejected": 2.2167811393737793, | |
"logps/chosen": -3.9598464965820312, | |
"logps/rejected": -8.5329008102417, | |
"loss": 0.1977, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -5.939770221710205, | |
"rewards/margins": 6.85958194732666, | |
"rewards/rejected": -12.799351692199707, | |
"step": 1360 | |
}, | |
{ | |
"epoch": 1.2166962699822381, | |
"grad_norm": 2.889807939529419, | |
"learning_rate": 9.209358300585474e-08, | |
"logits/chosen": 1.8230903148651123, | |
"logits/rejected": 1.7674376964569092, | |
"logps/chosen": -3.5624186992645264, | |
"logps/rejected": -8.8640775680542, | |
"loss": 0.1655, | |
"rewards/accuracies": 0.987500011920929, | |
"rewards/chosen": -5.343628406524658, | |
"rewards/margins": 7.952487945556641, | |
"rewards/rejected": -13.296114921569824, | |
"step": 1370 | |
}, | |
{ | |
"epoch": 1.2255772646536411, | |
"grad_norm": 2.0800833702087402, | |
"learning_rate": 7.854209717842231e-08, | |
"logits/chosen": 2.958648204803467, | |
"logits/rejected": 2.6478304862976074, | |
"logps/chosen": -4.070685863494873, | |
"logps/rejected": -8.65223217010498, | |
"loss": 0.1918, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.1060285568237305, | |
"rewards/margins": 6.872319221496582, | |
"rewards/rejected": -12.978347778320312, | |
"step": 1380 | |
}, | |
{ | |
"epoch": 1.2344582593250444, | |
"grad_norm": 4.166500568389893, | |
"learning_rate": 6.605274281709929e-08, | |
"logits/chosen": 2.3533079624176025, | |
"logits/rejected": 2.153561592102051, | |
"logps/chosen": -3.9434361457824707, | |
"logps/rejected": -9.156129837036133, | |
"loss": 0.219, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -5.915154457092285, | |
"rewards/margins": 7.819040775299072, | |
"rewards/rejected": -13.7341947555542, | |
"step": 1390 | |
}, | |
{ | |
"epoch": 1.2433392539964476, | |
"grad_norm": 3.4721784591674805, | |
"learning_rate": 5.463099816548578e-08, | |
"logits/chosen": 3.012937068939209, | |
"logits/rejected": 2.7736458778381348, | |
"logps/chosen": -4.2211012840271, | |
"logps/rejected": -9.378921508789062, | |
"loss": 0.1719, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.3316521644592285, | |
"rewards/margins": 7.736731052398682, | |
"rewards/rejected": -14.068384170532227, | |
"step": 1400 | |
}, | |
{ | |
"epoch": 1.2433392539964476, | |
"eval_logits/chosen": 2.7338051795959473, | |
"eval_logits/rejected": 2.516977310180664, | |
"eval_logps/chosen": -4.3559651374816895, | |
"eval_logps/rejected": -8.89287281036377, | |
"eval_loss": 0.20297038555145264, | |
"eval_rewards/accuracies": 0.9780219793319702, | |
"eval_rewards/chosen": -6.533947467803955, | |
"eval_rewards/margins": 6.805361747741699, | |
"eval_rewards/rejected": -13.339309692382812, | |
"eval_runtime": 25.2634, | |
"eval_samples_per_second": 28.816, | |
"eval_steps_per_second": 3.602, | |
"step": 1400 | |
}, | |
{ | |
"epoch": 1.2522202486678509, | |
"grad_norm": 5.4153571128845215, | |
"learning_rate": 4.428187317827848e-08, | |
"logits/chosen": 2.693077325820923, | |
"logits/rejected": 2.754072666168213, | |
"logps/chosen": -4.117603302001953, | |
"logps/rejected": -8.823477745056152, | |
"loss": 0.199, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.1764044761657715, | |
"rewards/margins": 7.058813571929932, | |
"rewards/rejected": -13.235217094421387, | |
"step": 1410 | |
}, | |
{ | |
"epoch": 1.261101243339254, | |
"grad_norm": 5.981139183044434, | |
"learning_rate": 3.5009907323737826e-08, | |
"logits/chosen": 2.250978708267212, | |
"logits/rejected": 1.818576455116272, | |
"logps/chosen": -4.223572254180908, | |
"logps/rejected": -9.683964729309082, | |
"loss": 0.2149, | |
"rewards/accuracies": 0.9375, | |
"rewards/chosen": -6.335358142852783, | |
"rewards/margins": 8.19058895111084, | |
"rewards/rejected": -14.525945663452148, | |
"step": 1420 | |
}, | |
{ | |
"epoch": 1.2699822380106571, | |
"grad_norm": 2.202054738998413, | |
"learning_rate": 2.681916759252917e-08, | |
"logits/chosen": 2.31886625289917, | |
"logits/rejected": 2.116328716278076, | |
"logps/chosen": -4.074057579040527, | |
"logps/rejected": -8.692493438720703, | |
"loss": 0.1547, | |
"rewards/accuracies": 0.949999988079071, | |
"rewards/chosen": -6.111086368560791, | |
"rewards/margins": 6.9276533126831055, | |
"rewards/rejected": -13.038739204406738, | |
"step": 1430 | |
}, | |
{ | |
"epoch": 1.2788632326820604, | |
"grad_norm": 2.1010451316833496, | |
"learning_rate": 1.9713246713805588e-08, | |
"logits/chosen": 2.656672477722168, | |
"logits/rejected": 2.773162364959717, | |
"logps/chosen": -4.189755439758301, | |
"logps/rejected": -9.755474090576172, | |
"loss": 0.2032, | |
"rewards/accuracies": 0.9624999761581421, | |
"rewards/chosen": -6.284633159637451, | |
"rewards/margins": 8.348578453063965, | |
"rewards/rejected": -14.633212089538574, | |
"step": 1440 | |
}, | |
{ | |
"epoch": 1.2877442273534636, | |
"grad_norm": 5.367340564727783, | |
"learning_rate": 1.3695261579316776e-08, | |
"logits/chosen": 2.364032030105591, | |
"logits/rejected": 2.221125602722168, | |
"logps/chosen": -4.26717472076416, | |
"logps/rejected": -9.568387985229492, | |
"loss": 0.1819, | |
"rewards/accuracies": 0.9750000238418579, | |
"rewards/chosen": -6.40076208114624, | |
"rewards/margins": 7.951818943023682, | |
"rewards/rejected": -14.352581977844238, | |
"step": 1450 | |
}, | |
{ | |
"epoch": 1.2877442273534636, | |
"eval_logits/chosen": 2.730602502822876, | |
"eval_logits/rejected": 2.5076637268066406, | |
"eval_logps/chosen": -4.366745948791504, | |
"eval_logps/rejected": -8.909856796264648, | |
"eval_loss": 0.20565223693847656, | |
"eval_rewards/accuracies": 0.9670329689979553, | |
"eval_rewards/chosen": -6.550118446350098, | |
"eval_rewards/margins": 6.814664840698242, | |
"eval_rewards/rejected": -13.36478328704834, | |
"eval_runtime": 25.2605, | |
"eval_samples_per_second": 28.82, | |
"eval_steps_per_second": 3.602, | |
"step": 1450 | |
} | |
], | |
"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": 3.41407013924556e+18, | |
"train_batch_size": 1, | |
"trial_name": null, | |
"trial_params": null | |
} | |