|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.998691442030882, |
|
"eval_steps": 500, |
|
"global_step": 477, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.010468463752944255, |
|
"grad_norm": 15.141688725114767, |
|
"learning_rate": 5.208333333333333e-08, |
|
"logits/chosen": -2.770766496658325, |
|
"logits/rejected": -2.728191375732422, |
|
"logps/chosen": -1.027971625328064, |
|
"logps/rejected": -1.1730705499649048, |
|
"loss": 2.0284, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -1.027971625328064, |
|
"rewards/margins": 0.14509883522987366, |
|
"rewards/rejected": -1.1730705499649048, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.02093692750588851, |
|
"grad_norm": 13.969517964651633, |
|
"learning_rate": 1.0416666666666667e-07, |
|
"logits/chosen": -2.7511062622070312, |
|
"logits/rejected": -2.7253737449645996, |
|
"logps/chosen": -0.9954013824462891, |
|
"logps/rejected": -1.0719196796417236, |
|
"loss": 2.002, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.9954013824462891, |
|
"rewards/margins": 0.07651824504137039, |
|
"rewards/rejected": -1.0719196796417236, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.031405391258832765, |
|
"grad_norm": 13.915856984059875, |
|
"learning_rate": 1.5624999999999999e-07, |
|
"logits/chosen": -2.7306742668151855, |
|
"logits/rejected": -2.697096347808838, |
|
"logps/chosen": -0.9895243644714355, |
|
"logps/rejected": -1.147679328918457, |
|
"loss": 1.9874, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.9895243644714355, |
|
"rewards/margins": 0.15815496444702148, |
|
"rewards/rejected": -1.147679328918457, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.04187385501177702, |
|
"grad_norm": 13.588156017248505, |
|
"learning_rate": 2.0833333333333333e-07, |
|
"logits/chosen": -2.8030834197998047, |
|
"logits/rejected": -2.709643602371216, |
|
"logps/chosen": -1.0228091478347778, |
|
"logps/rejected": -1.1238880157470703, |
|
"loss": 2.0265, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -1.0228091478347778, |
|
"rewards/margins": 0.10107902437448502, |
|
"rewards/rejected": -1.1238880157470703, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05234231876472128, |
|
"grad_norm": 14.042807702810773, |
|
"learning_rate": 2.604166666666667e-07, |
|
"logits/chosen": -2.7991013526916504, |
|
"logits/rejected": -2.7758243083953857, |
|
"logps/chosen": -1.0302048921585083, |
|
"logps/rejected": -1.086944818496704, |
|
"loss": 2.0091, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -1.0302048921585083, |
|
"rewards/margins": 0.05673985555768013, |
|
"rewards/rejected": -1.086944818496704, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.06281078251766553, |
|
"grad_norm": 15.069602245174066, |
|
"learning_rate": 3.1249999999999997e-07, |
|
"logits/chosen": -2.7733917236328125, |
|
"logits/rejected": -2.7317938804626465, |
|
"logps/chosen": -0.9855572581291199, |
|
"logps/rejected": -1.0908451080322266, |
|
"loss": 2.0118, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.9855572581291199, |
|
"rewards/margins": 0.1052878275513649, |
|
"rewards/rejected": -1.0908451080322266, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.07327924627060979, |
|
"grad_norm": 14.14588992747414, |
|
"learning_rate": 3.645833333333333e-07, |
|
"logits/chosen": -2.7773802280426025, |
|
"logits/rejected": -2.6910452842712402, |
|
"logps/chosen": -1.0310332775115967, |
|
"logps/rejected": -1.1711452007293701, |
|
"loss": 1.9903, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -1.0310332775115967, |
|
"rewards/margins": 0.1401120126247406, |
|
"rewards/rejected": -1.1711452007293701, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.08374771002355404, |
|
"grad_norm": 20.106972236529245, |
|
"learning_rate": 4.1666666666666667e-07, |
|
"logits/chosen": -2.7931580543518066, |
|
"logits/rejected": -2.6956787109375, |
|
"logps/chosen": -1.011854887008667, |
|
"logps/rejected": -1.2990570068359375, |
|
"loss": 1.9463, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.011854887008667, |
|
"rewards/margins": 0.28720206022262573, |
|
"rewards/rejected": -1.2990570068359375, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.0942161737764983, |
|
"grad_norm": 15.19327904046142, |
|
"learning_rate": 4.6874999999999996e-07, |
|
"logits/chosen": -2.8017444610595703, |
|
"logits/rejected": -2.7550606727600098, |
|
"logps/chosen": -0.9639647603034973, |
|
"logps/rejected": -1.142878770828247, |
|
"loss": 1.9512, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.9639647603034973, |
|
"rewards/margins": 0.17891402542591095, |
|
"rewards/rejected": -1.142878770828247, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.10468463752944256, |
|
"grad_norm": 20.91684390311358, |
|
"learning_rate": 4.999731868769026e-07, |
|
"logits/chosen": -2.7584781646728516, |
|
"logits/rejected": -2.6904306411743164, |
|
"logps/chosen": -1.0117145776748657, |
|
"logps/rejected": -1.30343496799469, |
|
"loss": 1.9653, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -1.0117145776748657, |
|
"rewards/margins": 0.2917203903198242, |
|
"rewards/rejected": -1.30343496799469, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.11515310128238682, |
|
"grad_norm": 16.247858117319236, |
|
"learning_rate": 4.996716052911017e-07, |
|
"logits/chosen": -2.754804849624634, |
|
"logits/rejected": -2.71933650970459, |
|
"logps/chosen": -1.0166981220245361, |
|
"logps/rejected": -1.268345832824707, |
|
"loss": 1.9611, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -1.0166981220245361, |
|
"rewards/margins": 0.25164780020713806, |
|
"rewards/rejected": -1.268345832824707, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.12562156503533106, |
|
"grad_norm": 16.39781017532518, |
|
"learning_rate": 4.990353313429303e-07, |
|
"logits/chosen": -2.8165841102600098, |
|
"logits/rejected": -2.7710437774658203, |
|
"logps/chosen": -1.049938440322876, |
|
"logps/rejected": -1.2433346509933472, |
|
"loss": 1.9164, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.049938440322876, |
|
"rewards/margins": 0.19339624047279358, |
|
"rewards/rejected": -1.2433346509933472, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.1360900287882753, |
|
"grad_norm": 15.239254500208718, |
|
"learning_rate": 4.980652179769217e-07, |
|
"logits/chosen": -2.8142337799072266, |
|
"logits/rejected": -2.6995322704315186, |
|
"logps/chosen": -1.0191227197647095, |
|
"logps/rejected": -1.544886827468872, |
|
"loss": 1.9025, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.0191227197647095, |
|
"rewards/margins": 0.5257640480995178, |
|
"rewards/rejected": -1.544886827468872, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.14655849254121958, |
|
"grad_norm": 21.0566472158153, |
|
"learning_rate": 4.967625656594781e-07, |
|
"logits/chosen": -2.7174973487854004, |
|
"logits/rejected": -2.6667487621307373, |
|
"logps/chosen": -1.0492310523986816, |
|
"logps/rejected": -1.3482974767684937, |
|
"loss": 1.9091, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -1.0492310523986816, |
|
"rewards/margins": 0.2990665137767792, |
|
"rewards/rejected": -1.3482974767684937, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.15702695629416383, |
|
"grad_norm": 30.761967865275885, |
|
"learning_rate": 4.951291206355559e-07, |
|
"logits/chosen": -2.773061752319336, |
|
"logits/rejected": -2.698946952819824, |
|
"logps/chosen": -0.9925507307052612, |
|
"logps/rejected": -1.3563807010650635, |
|
"loss": 1.8891, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.9925507307052612, |
|
"rewards/margins": 0.36382991075515747, |
|
"rewards/rejected": -1.3563807010650635, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.16749542004710807, |
|
"grad_norm": 31.297744502118277, |
|
"learning_rate": 4.93167072587771e-07, |
|
"logits/chosen": -2.738257884979248, |
|
"logits/rejected": -2.6784729957580566, |
|
"logps/chosen": -1.0087690353393555, |
|
"logps/rejected": -1.4246900081634521, |
|
"loss": 1.9382, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.0087690353393555, |
|
"rewards/margins": 0.4159211218357086, |
|
"rewards/rejected": -1.4246900081634521, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.17796388380005235, |
|
"grad_norm": 24.407477818447635, |
|
"learning_rate": 4.908790517010636e-07, |
|
"logits/chosen": -2.747776508331299, |
|
"logits/rejected": -2.7003962993621826, |
|
"logps/chosen": -0.9973554611206055, |
|
"logps/rejected": -1.3728947639465332, |
|
"loss": 1.9613, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.9973554611206055, |
|
"rewards/margins": 0.3755393624305725, |
|
"rewards/rejected": -1.3728947639465332, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 0.1884323475529966, |
|
"grad_norm": 31.20847987328103, |
|
"learning_rate": 4.882681251368548e-07, |
|
"logits/chosen": -2.705894708633423, |
|
"logits/rejected": -2.6819891929626465, |
|
"logps/chosen": -1.0237383842468262, |
|
"logps/rejected": -1.404168725013733, |
|
"loss": 1.88, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.0237383842468262, |
|
"rewards/margins": 0.3804302513599396, |
|
"rewards/rejected": -1.404168725013733, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.19890081130594087, |
|
"grad_norm": 30.65914852598083, |
|
"learning_rate": 4.853377929214243e-07, |
|
"logits/chosen": -2.776717185974121, |
|
"logits/rejected": -2.6825289726257324, |
|
"logps/chosen": -1.038529872894287, |
|
"logps/rejected": -1.3572471141815186, |
|
"loss": 1.9062, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -1.038529872894287, |
|
"rewards/margins": 0.31871724128723145, |
|
"rewards/rejected": -1.3572471141815186, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 0.2093692750588851, |
|
"grad_norm": 34.08856475096774, |
|
"learning_rate": 4.820919832540181e-07, |
|
"logits/chosen": -2.74165678024292, |
|
"logits/rejected": -2.6503825187683105, |
|
"logps/chosen": -1.051953673362732, |
|
"logps/rejected": -1.4617860317230225, |
|
"loss": 1.9147, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -1.051953673362732, |
|
"rewards/margins": 0.4098321497440338, |
|
"rewards/rejected": -1.4617860317230225, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.21983773881182936, |
|
"grad_norm": 28.24345375738832, |
|
"learning_rate": 4.785350472409791e-07, |
|
"logits/chosen": -2.743473529815674, |
|
"logits/rejected": -2.7197020053863525, |
|
"logps/chosen": -0.931172251701355, |
|
"logps/rejected": -1.453914999961853, |
|
"loss": 1.8781, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.931172251701355, |
|
"rewards/margins": 0.522742748260498, |
|
"rewards/rejected": -1.453914999961853, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 0.23030620256477363, |
|
"grad_norm": 35.845899664243596, |
|
"learning_rate": 4.7467175306295647e-07, |
|
"logits/chosen": -2.817007064819336, |
|
"logits/rejected": -2.762336254119873, |
|
"logps/chosen": -1.0500761270523071, |
|
"logps/rejected": -1.4565006494522095, |
|
"loss": 1.88, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.0500761270523071, |
|
"rewards/margins": 0.40642452239990234, |
|
"rewards/rejected": -1.4565006494522095, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.24077466631771788, |
|
"grad_norm": 23.382933706781618, |
|
"learning_rate": 4.70507279583015e-07, |
|
"logits/chosen": -2.7697396278381348, |
|
"logits/rejected": -2.710322856903076, |
|
"logps/chosen": -1.0177266597747803, |
|
"logps/rejected": -1.5666338205337524, |
|
"loss": 1.868, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.0177266597747803, |
|
"rewards/margins": 0.5489073991775513, |
|
"rewards/rejected": -1.5666338205337524, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 0.2512431300706621, |
|
"grad_norm": 29.59771539393517, |
|
"learning_rate": 4.6604720940421207e-07, |
|
"logits/chosen": -2.7786221504211426, |
|
"logits/rejected": -2.737684965133667, |
|
"logps/chosen": -1.0584137439727783, |
|
"logps/rejected": -1.4742605686187744, |
|
"loss": 1.9062, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.0584137439727783, |
|
"rewards/margins": 0.41584667563438416, |
|
"rewards/rejected": -1.4742605686187744, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.26171159382360637, |
|
"grad_norm": 24.555574989451102, |
|
"learning_rate": 4.612975213859487e-07, |
|
"logits/chosen": -2.8200080394744873, |
|
"logits/rejected": -2.7997546195983887, |
|
"logps/chosen": -1.058563470840454, |
|
"logps/rejected": -1.3519017696380615, |
|
"loss": 1.9069, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -1.058563470840454, |
|
"rewards/margins": 0.29333844780921936, |
|
"rewards/rejected": -1.3519017696380615, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.2721800575765506, |
|
"grad_norm": 26.634009671559898, |
|
"learning_rate": 4.5626458262912735e-07, |
|
"logits/chosen": -2.788752794265747, |
|
"logits/rejected": -2.754631519317627, |
|
"logps/chosen": -1.0312286615371704, |
|
"logps/rejected": -1.2794463634490967, |
|
"loss": 1.8977, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -1.0312286615371704, |
|
"rewards/margins": 0.24821774661540985, |
|
"rewards/rejected": -1.2794463634490967, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.2826485213294949, |
|
"grad_norm": 32.240888174368, |
|
"learning_rate": 4.5095513994085974e-07, |
|
"logits/chosen": -2.7877840995788574, |
|
"logits/rejected": -2.7001712322235107, |
|
"logps/chosen": -1.0880383253097534, |
|
"logps/rejected": -1.7845348119735718, |
|
"loss": 1.8912, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -1.0880383253097534, |
|
"rewards/margins": 0.696496307849884, |
|
"rewards/rejected": -1.7845348119735718, |
|
"step": 135 |
|
}, |
|
{ |
|
"epoch": 0.29311698508243916, |
|
"grad_norm": 25.76204815720163, |
|
"learning_rate": 4.453763107901675e-07, |
|
"logits/chosen": -2.8313488960266113, |
|
"logits/rejected": -2.780134677886963, |
|
"logps/chosen": -0.9982128143310547, |
|
"logps/rejected": -1.4504693746566772, |
|
"loss": 1.8155, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.9982128143310547, |
|
"rewards/margins": 0.4522566795349121, |
|
"rewards/rejected": -1.4504693746566772, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.3035854488353834, |
|
"grad_norm": 21.7794898263259, |
|
"learning_rate": 4.395355737667985e-07, |
|
"logits/chosen": -2.809255361557007, |
|
"logits/rejected": -2.761308193206787, |
|
"logps/chosen": -1.0206363201141357, |
|
"logps/rejected": -1.419768214225769, |
|
"loss": 1.8605, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.0206363201141357, |
|
"rewards/margins": 0.39913198351860046, |
|
"rewards/rejected": -1.419768214225769, |
|
"step": 145 |
|
}, |
|
{ |
|
"epoch": 0.31405391258832765, |
|
"grad_norm": 26.507909597761785, |
|
"learning_rate": 4.3344075855595097e-07, |
|
"logits/chosen": -2.849714517593384, |
|
"logits/rejected": -2.8313772678375244, |
|
"logps/chosen": -0.9701302647590637, |
|
"logps/rejected": -1.3971799612045288, |
|
"loss": 1.8345, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.9701302647590637, |
|
"rewards/margins": 0.42704978585243225, |
|
"rewards/rejected": -1.3971799612045288, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.3245223763412719, |
|
"grad_norm": 28.367719379820283, |
|
"learning_rate": 4.271000354423425e-07, |
|
"logits/chosen": -2.7890210151672363, |
|
"logits/rejected": -2.757753849029541, |
|
"logps/chosen": -1.0669995546340942, |
|
"logps/rejected": -1.4297670125961304, |
|
"loss": 1.894, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.0669995546340942, |
|
"rewards/margins": 0.36276736855506897, |
|
"rewards/rejected": -1.4297670125961304, |
|
"step": 155 |
|
}, |
|
{ |
|
"epoch": 0.33499084009421615, |
|
"grad_norm": 32.029509535293904, |
|
"learning_rate": 4.2052190435769554e-07, |
|
"logits/chosen": -2.7710726261138916, |
|
"logits/rejected": -2.7451701164245605, |
|
"logps/chosen": -1.0376036167144775, |
|
"logps/rejected": -1.4104793071746826, |
|
"loss": 1.8838, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.0376036167144775, |
|
"rewards/margins": 0.37287575006484985, |
|
"rewards/rejected": -1.4104793071746826, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.34545930384716045, |
|
"grad_norm": 25.20687040858121, |
|
"learning_rate": 4.137151834863213e-07, |
|
"logits/chosen": -2.7923319339752197, |
|
"logits/rejected": -2.707219362258911, |
|
"logps/chosen": -0.9815086126327515, |
|
"logps/rejected": -1.4057986736297607, |
|
"loss": 1.8435, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.9815086126327515, |
|
"rewards/margins": 0.4242900311946869, |
|
"rewards/rejected": -1.4057986736297607, |
|
"step": 165 |
|
}, |
|
{ |
|
"epoch": 0.3559277676001047, |
|
"grad_norm": 28.006785803959723, |
|
"learning_rate": 4.0668899744407567e-07, |
|
"logits/chosen": -2.8175511360168457, |
|
"logits/rejected": -2.752450466156006, |
|
"logps/chosen": -1.0611494779586792, |
|
"logps/rejected": -1.5021220445632935, |
|
"loss": 1.8575, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.0611494779586792, |
|
"rewards/margins": 0.4409725069999695, |
|
"rewards/rejected": -1.5021220445632935, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.36639623135304894, |
|
"grad_norm": 25.596200434230976, |
|
"learning_rate": 3.994527650465352e-07, |
|
"logits/chosen": -2.7964577674865723, |
|
"logits/rejected": -2.709230661392212, |
|
"logps/chosen": -1.1404192447662354, |
|
"logps/rejected": -1.6063992977142334, |
|
"loss": 1.9076, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.1404192447662354, |
|
"rewards/margins": 0.4659802317619324, |
|
"rewards/rejected": -1.6063992977142334, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.3768646951059932, |
|
"grad_norm": 25.892094106029592, |
|
"learning_rate": 3.920161866827889e-07, |
|
"logits/chosen": -2.7817273139953613, |
|
"logits/rejected": -2.7617640495300293, |
|
"logps/chosen": -1.0391079187393188, |
|
"logps/rejected": -1.5876353979110718, |
|
"loss": 1.8544, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.0391079187393188, |
|
"rewards/margins": 0.5485275387763977, |
|
"rewards/rejected": -1.5876353979110718, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.38733315885893743, |
|
"grad_norm": 29.42463620034665, |
|
"learning_rate": 3.8438923131177237e-07, |
|
"logits/chosen": -2.7785096168518066, |
|
"logits/rejected": -2.6860692501068115, |
|
"logps/chosen": -1.0492454767227173, |
|
"logps/rejected": -1.6226046085357666, |
|
"loss": 1.8247, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.0492454767227173, |
|
"rewards/margins": 0.573358952999115, |
|
"rewards/rejected": -1.6226046085357666, |
|
"step": 185 |
|
}, |
|
{ |
|
"epoch": 0.39780162261188173, |
|
"grad_norm": 27.921360264789193, |
|
"learning_rate": 3.765821230985757e-07, |
|
"logits/chosen": -2.7853875160217285, |
|
"logits/rejected": -2.7454705238342285, |
|
"logps/chosen": -1.0294712781906128, |
|
"logps/rejected": -1.4654343128204346, |
|
"loss": 1.9244, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.0294712781906128, |
|
"rewards/margins": 0.4359629154205322, |
|
"rewards/rejected": -1.4654343128204346, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.408270086364826, |
|
"grad_norm": 29.87078620394206, |
|
"learning_rate": 3.6860532770864005e-07, |
|
"logits/chosen": -2.701105833053589, |
|
"logits/rejected": -2.6695313453674316, |
|
"logps/chosen": -1.098097324371338, |
|
"logps/rejected": -1.4831833839416504, |
|
"loss": 1.899, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.098097324371338, |
|
"rewards/margins": 0.3850858509540558, |
|
"rewards/rejected": -1.4831833839416504, |
|
"step": 195 |
|
}, |
|
{ |
|
"epoch": 0.4187385501177702, |
|
"grad_norm": 28.048650935368237, |
|
"learning_rate": 3.604695382782159e-07, |
|
"logits/chosen": -2.738492250442505, |
|
"logits/rejected": -2.7187023162841797, |
|
"logps/chosen": -1.0665818452835083, |
|
"logps/rejected": -1.46578049659729, |
|
"loss": 1.8737, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.0665818452835083, |
|
"rewards/margins": 0.39919885993003845, |
|
"rewards/rejected": -1.46578049659729, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.42920701387071447, |
|
"grad_norm": 27.850087928191183, |
|
"learning_rate": 3.5218566107988867e-07, |
|
"logits/chosen": -2.6582818031311035, |
|
"logits/rejected": -2.5809996128082275, |
|
"logps/chosen": -0.9890678524971008, |
|
"logps/rejected": -1.4678758382797241, |
|
"loss": 1.8403, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.9890678524971008, |
|
"rewards/margins": 0.47880783677101135, |
|
"rewards/rejected": -1.4678758382797241, |
|
"step": 205 |
|
}, |
|
{ |
|
"epoch": 0.4396754776236587, |
|
"grad_norm": 29.109484926295075, |
|
"learning_rate": 3.4376480090239047e-07, |
|
"logits/chosen": -2.6842644214630127, |
|
"logits/rejected": -2.627836227416992, |
|
"logps/chosen": -1.075455665588379, |
|
"logps/rejected": -1.4202773571014404, |
|
"loss": 1.8863, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.075455665588379, |
|
"rewards/margins": 0.34482160210609436, |
|
"rewards/rejected": -1.4202773571014404, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.45014394137660296, |
|
"grad_norm": 23.99752736324705, |
|
"learning_rate": 3.3521824616429284e-07, |
|
"logits/chosen": -2.716024875640869, |
|
"logits/rejected": -2.6916396617889404, |
|
"logps/chosen": -1.0032670497894287, |
|
"logps/rejected": -1.5087072849273682, |
|
"loss": 1.8264, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -1.0032670497894287, |
|
"rewards/margins": 0.5054404139518738, |
|
"rewards/rejected": -1.5087072849273682, |
|
"step": 215 |
|
}, |
|
{ |
|
"epoch": 0.46061240512954726, |
|
"grad_norm": 31.21916182230841, |
|
"learning_rate": 3.265574537815398e-07, |
|
"logits/chosen": -2.742882013320923, |
|
"logits/rejected": -2.658285617828369, |
|
"logps/chosen": -1.033542275428772, |
|
"logps/rejected": -1.5640771389007568, |
|
"loss": 1.8307, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.033542275428772, |
|
"rewards/margins": 0.5305347442626953, |
|
"rewards/rejected": -1.5640771389007568, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.4710808688824915, |
|
"grad_norm": 26.98263419792695, |
|
"learning_rate": 3.1779403380910425e-07, |
|
"logits/chosen": -2.7341504096984863, |
|
"logits/rejected": -2.6276047229766846, |
|
"logps/chosen": -0.988808810710907, |
|
"logps/rejected": -1.5687158107757568, |
|
"loss": 1.8334, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.988808810710907, |
|
"rewards/margins": 0.5799070596694946, |
|
"rewards/rejected": -1.5687158107757568, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.48154933263543576, |
|
"grad_norm": 29.311922113138767, |
|
"learning_rate": 3.0893973387735683e-07, |
|
"logits/chosen": -2.74092698097229, |
|
"logits/rejected": -2.6369271278381348, |
|
"logps/chosen": -1.009143352508545, |
|
"logps/rejected": -1.665408730506897, |
|
"loss": 1.8283, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.009143352508545, |
|
"rewards/margins": 0.6562654376029968, |
|
"rewards/rejected": -1.665408730506897, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.49201779638838, |
|
"grad_norm": 32.62916846952035, |
|
"learning_rate": 3.000064234440111e-07, |
|
"logits/chosen": -2.729424476623535, |
|
"logits/rejected": -2.672799587249756, |
|
"logps/chosen": -1.0022896528244019, |
|
"logps/rejected": -1.5322816371917725, |
|
"loss": 1.86, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -1.0022896528244019, |
|
"rewards/margins": 0.5299919247627258, |
|
"rewards/rejected": -1.5322816371917725, |
|
"step": 235 |
|
}, |
|
{ |
|
"epoch": 0.5024862601413242, |
|
"grad_norm": 25.60849652215575, |
|
"learning_rate": 2.910060778827554e-07, |
|
"logits/chosen": -2.666738986968994, |
|
"logits/rejected": -2.6147377490997314, |
|
"logps/chosen": -1.0380260944366455, |
|
"logps/rejected": -1.5499128103256226, |
|
"loss": 1.8806, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.0380260944366455, |
|
"rewards/margins": 0.5118868947029114, |
|
"rewards/rejected": -1.5499128103256226, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.5129547238942685, |
|
"grad_norm": 23.307769710548996, |
|
"learning_rate": 2.8195076242990116e-07, |
|
"logits/chosen": -2.7083492279052734, |
|
"logits/rejected": -2.608227014541626, |
|
"logps/chosen": -1.1102759838104248, |
|
"logps/rejected": -1.5368731021881104, |
|
"loss": 1.8599, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.1102759838104248, |
|
"rewards/margins": 0.42659682035446167, |
|
"rewards/rejected": -1.5368731021881104, |
|
"step": 245 |
|
}, |
|
{ |
|
"epoch": 0.5234231876472127, |
|
"grad_norm": 30.713855522791444, |
|
"learning_rate": 2.7285261601056697e-07, |
|
"logits/chosen": -2.621891498565674, |
|
"logits/rejected": -2.572918653488159, |
|
"logps/chosen": -1.0056602954864502, |
|
"logps/rejected": -1.5289640426635742, |
|
"loss": 1.8611, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.0056602954864502, |
|
"rewards/margins": 0.5233036875724792, |
|
"rewards/rejected": -1.5289640426635742, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.533891651400157, |
|
"grad_norm": 21.027582284131125, |
|
"learning_rate": 2.6372383496608186e-07, |
|
"logits/chosen": -2.679741859436035, |
|
"logits/rejected": -2.588026762008667, |
|
"logps/chosen": -1.0767912864685059, |
|
"logps/rejected": -1.6802423000335693, |
|
"loss": 1.7962, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.0767912864685059, |
|
"rewards/margins": 0.603451132774353, |
|
"rewards/rejected": -1.6802423000335693, |
|
"step": 255 |
|
}, |
|
{ |
|
"epoch": 0.5443601151531012, |
|
"grad_norm": 21.083992579626138, |
|
"learning_rate": 2.5457665670441937e-07, |
|
"logits/chosen": -2.639758825302124, |
|
"logits/rejected": -2.539482831954956, |
|
"logps/chosen": -0.9617997407913208, |
|
"logps/rejected": -1.643094778060913, |
|
"loss": 1.8469, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.9617997407913208, |
|
"rewards/margins": 0.6812952160835266, |
|
"rewards/rejected": -1.643094778060913, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.5548285789060455, |
|
"grad_norm": 34.336932322095805, |
|
"learning_rate": 2.454233432955807e-07, |
|
"logits/chosen": -2.6934094429016113, |
|
"logits/rejected": -2.635524034500122, |
|
"logps/chosen": -1.0399844646453857, |
|
"logps/rejected": -1.3895108699798584, |
|
"loss": 1.8185, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.0399844646453857, |
|
"rewards/margins": 0.3495265543460846, |
|
"rewards/rejected": -1.3895108699798584, |
|
"step": 265 |
|
}, |
|
{ |
|
"epoch": 0.5652970426589898, |
|
"grad_norm": 35.56183499916552, |
|
"learning_rate": 2.3627616503391812e-07, |
|
"logits/chosen": -2.678424596786499, |
|
"logits/rejected": -2.6192078590393066, |
|
"logps/chosen": -1.1347005367279053, |
|
"logps/rejected": -1.5019694566726685, |
|
"loss": 1.8876, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.1347005367279053, |
|
"rewards/margins": 0.36726871132850647, |
|
"rewards/rejected": -1.5019694566726685, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.575765506411934, |
|
"grad_norm": 21.13133035084058, |
|
"learning_rate": 2.2714738398943308e-07, |
|
"logits/chosen": -2.749488592147827, |
|
"logits/rejected": -2.629812717437744, |
|
"logps/chosen": -1.0328668355941772, |
|
"logps/rejected": -1.6207387447357178, |
|
"loss": 1.833, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.0328668355941772, |
|
"rewards/margins": 0.5878719091415405, |
|
"rewards/rejected": -1.6207387447357178, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.5862339701648783, |
|
"grad_norm": 27.44411354972676, |
|
"learning_rate": 2.1804923757009882e-07, |
|
"logits/chosen": -2.7630438804626465, |
|
"logits/rejected": -2.692211866378784, |
|
"logps/chosen": -0.968722939491272, |
|
"logps/rejected": -1.3731743097305298, |
|
"loss": 1.8453, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.968722939491272, |
|
"rewards/margins": 0.40445131063461304, |
|
"rewards/rejected": -1.3731743097305298, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.5967024339178225, |
|
"grad_norm": 32.499361452307625, |
|
"learning_rate": 2.089939221172446e-07, |
|
"logits/chosen": -2.682573080062866, |
|
"logits/rejected": -2.678396224975586, |
|
"logps/chosen": -1.0675628185272217, |
|
"logps/rejected": -1.5514373779296875, |
|
"loss": 1.8267, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.0675628185272217, |
|
"rewards/margins": 0.4838746190071106, |
|
"rewards/rejected": -1.5514373779296875, |
|
"step": 285 |
|
}, |
|
{ |
|
"epoch": 0.6071708976707668, |
|
"grad_norm": 34.51002212540645, |
|
"learning_rate": 1.9999357655598891e-07, |
|
"logits/chosen": -2.7195992469787598, |
|
"logits/rejected": -2.6691336631774902, |
|
"logps/chosen": -1.0737675428390503, |
|
"logps/rejected": -1.6511163711547852, |
|
"loss": 1.8055, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.0737675428390503, |
|
"rewards/margins": 0.5773489475250244, |
|
"rewards/rejected": -1.6511163711547852, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.6176393614237111, |
|
"grad_norm": 34.81875700322843, |
|
"learning_rate": 1.9106026612264315e-07, |
|
"logits/chosen": -2.7475931644439697, |
|
"logits/rejected": -2.661341667175293, |
|
"logps/chosen": -1.0815327167510986, |
|
"logps/rejected": -1.6415588855743408, |
|
"loss": 1.8548, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.0815327167510986, |
|
"rewards/margins": 0.5600259900093079, |
|
"rewards/rejected": -1.6415588855743408, |
|
"step": 295 |
|
}, |
|
{ |
|
"epoch": 0.6281078251766553, |
|
"grad_norm": 39.656088577829394, |
|
"learning_rate": 1.8220596619089573e-07, |
|
"logits/chosen": -2.650444507598877, |
|
"logits/rejected": -2.601776123046875, |
|
"logps/chosen": -1.0726263523101807, |
|
"logps/rejected": -1.5858640670776367, |
|
"loss": 1.8352, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.0726263523101807, |
|
"rewards/margins": 0.5132376551628113, |
|
"rewards/rejected": -1.5858640670776367, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.6385762889295996, |
|
"grad_norm": 34.266407421018876, |
|
"learning_rate": 1.7344254621846017e-07, |
|
"logits/chosen": -2.6358659267425537, |
|
"logits/rejected": -2.589916944503784, |
|
"logps/chosen": -1.0825355052947998, |
|
"logps/rejected": -1.695123314857483, |
|
"loss": 1.8122, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.0825355052947998, |
|
"rewards/margins": 0.6125878691673279, |
|
"rewards/rejected": -1.695123314857483, |
|
"step": 305 |
|
}, |
|
{ |
|
"epoch": 0.6490447526825438, |
|
"grad_norm": 29.29392494715919, |
|
"learning_rate": 1.647817538357072e-07, |
|
"logits/chosen": -2.6732287406921387, |
|
"logits/rejected": -2.6671674251556396, |
|
"logps/chosen": -0.9606215357780457, |
|
"logps/rejected": -1.507188081741333, |
|
"loss": 1.8154, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.9606215357780457, |
|
"rewards/margins": 0.5465664863586426, |
|
"rewards/rejected": -1.507188081741333, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.6595132164354881, |
|
"grad_norm": 30.886215191002822, |
|
"learning_rate": 1.562351990976095e-07, |
|
"logits/chosen": -2.6945061683654785, |
|
"logits/rejected": -2.6447794437408447, |
|
"logps/chosen": -0.9400749206542969, |
|
"logps/rejected": -1.5918124914169312, |
|
"loss": 1.8296, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.9400749206542969, |
|
"rewards/margins": 0.651737630367279, |
|
"rewards/rejected": -1.5918124914169312, |
|
"step": 315 |
|
}, |
|
{ |
|
"epoch": 0.6699816801884323, |
|
"grad_norm": 40.31455490523793, |
|
"learning_rate": 1.478143389201113e-07, |
|
"logits/chosen": -2.6877899169921875, |
|
"logits/rejected": -2.6543657779693604, |
|
"logps/chosen": -1.0309157371520996, |
|
"logps/rejected": -1.5069587230682373, |
|
"loss": 1.7854, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.0309157371520996, |
|
"rewards/margins": 0.47604283690452576, |
|
"rewards/rejected": -1.5069587230682373, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.6804501439413766, |
|
"grad_norm": 33.29916453849858, |
|
"learning_rate": 1.3953046172178413e-07, |
|
"logits/chosen": -2.712068557739258, |
|
"logits/rejected": -2.6253581047058105, |
|
"logps/chosen": -1.0362852811813354, |
|
"logps/rejected": -1.6363003253936768, |
|
"loss": 1.7979, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.0362852811813354, |
|
"rewards/margins": 0.6000149846076965, |
|
"rewards/rejected": -1.6363003253936768, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.6909186076943209, |
|
"grad_norm": 28.615351737244207, |
|
"learning_rate": 1.3139467229135998e-07, |
|
"logits/chosen": -2.672591209411621, |
|
"logits/rejected": -2.624178886413574, |
|
"logps/chosen": -1.0226625204086304, |
|
"logps/rejected": -1.4031275510787964, |
|
"loss": 1.8499, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.0226625204086304, |
|
"rewards/margins": 0.38046500086784363, |
|
"rewards/rejected": -1.4031275510787964, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.7013870714472651, |
|
"grad_norm": 31.872437043858163, |
|
"learning_rate": 1.2341787690142435e-07, |
|
"logits/chosen": -2.783500909805298, |
|
"logits/rejected": -2.7123568058013916, |
|
"logps/chosen": -0.9628676176071167, |
|
"logps/rejected": -1.5226846933364868, |
|
"loss": 1.8281, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.9628676176071167, |
|
"rewards/margins": 0.5598170161247253, |
|
"rewards/rejected": -1.5226846933364868, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 0.7118555352002094, |
|
"grad_norm": 27.878376041354297, |
|
"learning_rate": 1.1561076868822755e-07, |
|
"logits/chosen": -2.7173564434051514, |
|
"logits/rejected": -2.6413800716400146, |
|
"logps/chosen": -1.1935253143310547, |
|
"logps/rejected": -1.5975086688995361, |
|
"loss": 1.8896, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.1935253143310547, |
|
"rewards/margins": 0.40398329496383667, |
|
"rewards/rejected": -1.5975086688995361, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.7223239989531536, |
|
"grad_norm": 36.976534101742224, |
|
"learning_rate": 1.0798381331721107e-07, |
|
"logits/chosen": -2.7495174407958984, |
|
"logits/rejected": -2.700282573699951, |
|
"logps/chosen": -1.0389212369918823, |
|
"logps/rejected": -1.5958582162857056, |
|
"loss": 1.8433, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.0389212369918823, |
|
"rewards/margins": 0.5569368600845337, |
|
"rewards/rejected": -1.5958582162857056, |
|
"step": 345 |
|
}, |
|
{ |
|
"epoch": 0.7327924627060979, |
|
"grad_norm": 30.288124940758514, |
|
"learning_rate": 1.0054723495346482e-07, |
|
"logits/chosen": -2.7106757164001465, |
|
"logits/rejected": -2.687342882156372, |
|
"logps/chosen": -1.0641244649887085, |
|
"logps/rejected": -1.5430529117584229, |
|
"loss": 1.8735, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.0641244649887085, |
|
"rewards/margins": 0.47892847657203674, |
|
"rewards/rejected": -1.5430529117584229, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.7432609264590422, |
|
"grad_norm": 30.72874948597524, |
|
"learning_rate": 9.331100255592436e-08, |
|
"logits/chosen": -2.7169158458709717, |
|
"logits/rejected": -2.66363787651062, |
|
"logps/chosen": -0.9323965311050415, |
|
"logps/rejected": -1.5993990898132324, |
|
"loss": 1.7736, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.9323965311050415, |
|
"rewards/margins": 0.6670026779174805, |
|
"rewards/rejected": -1.5993990898132324, |
|
"step": 355 |
|
}, |
|
{ |
|
"epoch": 0.7537293902119864, |
|
"grad_norm": 22.0079277817506, |
|
"learning_rate": 8.628481651367875e-08, |
|
"logits/chosen": -2.746021270751953, |
|
"logits/rejected": -2.6950621604919434, |
|
"logps/chosen": -0.9703041315078735, |
|
"logps/rejected": -1.4327679872512817, |
|
"loss": 1.783, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.9703041315078735, |
|
"rewards/margins": 0.4624637961387634, |
|
"rewards/rejected": -1.4327679872512817, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.7641978539649307, |
|
"grad_norm": 33.139790420388316, |
|
"learning_rate": 7.947809564230445e-08, |
|
"logits/chosen": -2.6830432415008545, |
|
"logits/rejected": -2.5992660522460938, |
|
"logps/chosen": -1.0758744478225708, |
|
"logps/rejected": -1.6359386444091797, |
|
"loss": 1.8553, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -1.0758744478225708, |
|
"rewards/margins": 0.5600640773773193, |
|
"rewards/rejected": -1.6359386444091797, |
|
"step": 365 |
|
}, |
|
{ |
|
"epoch": 0.7746663177178749, |
|
"grad_norm": 42.978648090018346, |
|
"learning_rate": 7.289996455765748e-08, |
|
"logits/chosen": -2.7361717224121094, |
|
"logits/rejected": -2.6721107959747314, |
|
"logps/chosen": -1.0527479648590088, |
|
"logps/rejected": -1.7067893743515015, |
|
"loss": 1.8183, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.0527479648590088, |
|
"rewards/margins": 0.6540414094924927, |
|
"rewards/rejected": -1.7067893743515015, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.7851347814708192, |
|
"grad_norm": 24.61410992203062, |
|
"learning_rate": 6.655924144404906e-08, |
|
"logits/chosen": -2.715893268585205, |
|
"logits/rejected": -2.7327394485473633, |
|
"logps/chosen": -1.0623342990875244, |
|
"logps/rejected": -1.5285212993621826, |
|
"loss": 1.8189, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.0623342990875244, |
|
"rewards/margins": 0.4661870002746582, |
|
"rewards/rejected": -1.5285212993621826, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.7956032452237635, |
|
"grad_norm": 29.084948650681554, |
|
"learning_rate": 6.046442623320145e-08, |
|
"logits/chosen": -2.7230355739593506, |
|
"logits/rejected": -2.665907621383667, |
|
"logps/chosen": -1.0696966648101807, |
|
"logps/rejected": -1.5225203037261963, |
|
"loss": 1.7794, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.0696966648101807, |
|
"rewards/margins": 0.4528239369392395, |
|
"rewards/rejected": -1.5225203037261963, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.8060717089767077, |
|
"grad_norm": 25.930022257563827, |
|
"learning_rate": 5.4623689209832484e-08, |
|
"logits/chosen": -2.7370214462280273, |
|
"logits/rejected": -2.6596484184265137, |
|
"logps/chosen": -1.0190281867980957, |
|
"logps/rejected": -1.4870203733444214, |
|
"loss": 1.8012, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.0190281867980957, |
|
"rewards/margins": 0.46799221634864807, |
|
"rewards/rejected": -1.4870203733444214, |
|
"step": 385 |
|
}, |
|
{ |
|
"epoch": 0.816540172729652, |
|
"grad_norm": 30.16253644602296, |
|
"learning_rate": 4.904486005914027e-08, |
|
"logits/chosen": -2.755403518676758, |
|
"logits/rejected": -2.6998536586761475, |
|
"logps/chosen": -1.0058339834213257, |
|
"logps/rejected": -1.6741714477539062, |
|
"loss": 1.7816, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.0058339834213257, |
|
"rewards/margins": 0.6683375239372253, |
|
"rewards/rejected": -1.6741714477539062, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.8270086364825961, |
|
"grad_norm": 32.100503895364014, |
|
"learning_rate": 4.373541737087263e-08, |
|
"logits/chosen": -2.6867947578430176, |
|
"logits/rejected": -2.6116786003112793, |
|
"logps/chosen": -0.9958473443984985, |
|
"logps/rejected": -1.5259888172149658, |
|
"loss": 1.8098, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.9958473443984985, |
|
"rewards/margins": 0.5301414132118225, |
|
"rewards/rejected": -1.5259888172149658, |
|
"step": 395 |
|
}, |
|
{ |
|
"epoch": 0.8374771002355405, |
|
"grad_norm": 39.50777886532678, |
|
"learning_rate": 3.8702478614051345e-08, |
|
"logits/chosen": -2.6500821113586426, |
|
"logits/rejected": -2.587491273880005, |
|
"logps/chosen": -0.9868978261947632, |
|
"logps/rejected": -1.5105135440826416, |
|
"loss": 1.81, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.9868978261947632, |
|
"rewards/margins": 0.523615837097168, |
|
"rewards/rejected": -1.5105135440826416, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.8479455639884846, |
|
"grad_norm": 39.172472564486576, |
|
"learning_rate": 3.3952790595787986e-08, |
|
"logits/chosen": -2.7077910900115967, |
|
"logits/rejected": -2.671496868133545, |
|
"logps/chosen": -1.0807896852493286, |
|
"logps/rejected": -1.6041944026947021, |
|
"loss": 1.8475, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.0807896852493286, |
|
"rewards/margins": 0.5234047174453735, |
|
"rewards/rejected": -1.6041944026947021, |
|
"step": 405 |
|
}, |
|
{ |
|
"epoch": 0.8584140277414289, |
|
"grad_norm": 34.16755927550673, |
|
"learning_rate": 2.9492720416985e-08, |
|
"logits/chosen": -2.729513645172119, |
|
"logits/rejected": -2.68589448928833, |
|
"logps/chosen": -0.9558256268501282, |
|
"logps/rejected": -1.4403427839279175, |
|
"loss": 1.8083, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.9558256268501282, |
|
"rewards/margins": 0.4845171868801117, |
|
"rewards/rejected": -1.4403427839279175, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.8688824914943732, |
|
"grad_norm": 27.600146859315487, |
|
"learning_rate": 2.5328246937043525e-08, |
|
"logits/chosen": -2.758931875228882, |
|
"logits/rejected": -2.693338394165039, |
|
"logps/chosen": -1.004372239112854, |
|
"logps/rejected": -1.6196565628051758, |
|
"loss": 1.7709, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.004372239112854, |
|
"rewards/margins": 0.615284264087677, |
|
"rewards/rejected": -1.6196565628051758, |
|
"step": 415 |
|
}, |
|
{ |
|
"epoch": 0.8793509552473174, |
|
"grad_norm": 26.767731856804396, |
|
"learning_rate": 2.1464952759020856e-08, |
|
"logits/chosen": -2.7129735946655273, |
|
"logits/rejected": -2.6572775840759277, |
|
"logps/chosen": -0.9905831217765808, |
|
"logps/rejected": -1.5554085969924927, |
|
"loss": 1.8411, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.9905831217765808, |
|
"rewards/margins": 0.5648255944252014, |
|
"rewards/rejected": -1.5554085969924927, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.8898194190002617, |
|
"grad_norm": 32.1190541072846, |
|
"learning_rate": 1.7908016745981856e-08, |
|
"logits/chosen": -2.656250476837158, |
|
"logits/rejected": -2.645900249481201, |
|
"logps/chosen": -1.1461454629898071, |
|
"logps/rejected": -1.604714035987854, |
|
"loss": 1.8218, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.1461454629898071, |
|
"rewards/margins": 0.45856842398643494, |
|
"rewards/rejected": -1.604714035987854, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.9002878827532059, |
|
"grad_norm": 38.87548467523877, |
|
"learning_rate": 1.4662207078575684e-08, |
|
"logits/chosen": -2.661616325378418, |
|
"logits/rejected": -2.617453098297119, |
|
"logps/chosen": -1.1022528409957886, |
|
"logps/rejected": -1.628826379776001, |
|
"loss": 1.8088, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.1022528409957886, |
|
"rewards/margins": 0.5265735387802124, |
|
"rewards/rejected": -1.628826379776001, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.9107563465061502, |
|
"grad_norm": 34.292720836263, |
|
"learning_rate": 1.1731874863145142e-08, |
|
"logits/chosen": -2.7260875701904297, |
|
"logits/rejected": -2.685364246368408, |
|
"logps/chosen": -1.0480232238769531, |
|
"logps/rejected": -1.6929914951324463, |
|
"loss": 1.8315, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.0480232238769531, |
|
"rewards/margins": 0.6449683308601379, |
|
"rewards/rejected": -1.6929914951324463, |
|
"step": 435 |
|
}, |
|
{ |
|
"epoch": 0.9212248102590945, |
|
"grad_norm": 36.60770449520838, |
|
"learning_rate": 9.12094829893642e-09, |
|
"logits/chosen": -2.6789753437042236, |
|
"logits/rejected": -2.6536145210266113, |
|
"logps/chosen": -1.0216642618179321, |
|
"logps/rejected": -1.547012448310852, |
|
"loss": 1.8478, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.0216642618179321, |
|
"rewards/margins": 0.5253482460975647, |
|
"rewards/rejected": -1.547012448310852, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.9316932740120387, |
|
"grad_norm": 34.43665423362751, |
|
"learning_rate": 6.832927412229017e-09, |
|
"logits/chosen": -2.721832752227783, |
|
"logits/rejected": -2.660407781600952, |
|
"logps/chosen": -1.0646559000015259, |
|
"logps/rejected": -1.554300308227539, |
|
"loss": 1.8533, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.0646559000015259, |
|
"rewards/margins": 0.4896443486213684, |
|
"rewards/rejected": -1.554300308227539, |
|
"step": 445 |
|
}, |
|
{ |
|
"epoch": 0.942161737764983, |
|
"grad_norm": 30.770535415455328, |
|
"learning_rate": 4.8708793644441086e-09, |
|
"logits/chosen": -2.6228880882263184, |
|
"logits/rejected": -2.5837976932525635, |
|
"logps/chosen": -1.0409653186798096, |
|
"logps/rejected": -1.537603735923767, |
|
"loss": 1.7905, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.0409653186798096, |
|
"rewards/margins": 0.4966384768486023, |
|
"rewards/rejected": -1.537603735923767, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.9526302015179272, |
|
"grad_norm": 27.915122014592697, |
|
"learning_rate": 3.2374343405217884e-09, |
|
"logits/chosen": -2.6800639629364014, |
|
"logits/rejected": -2.6128344535827637, |
|
"logps/chosen": -1.132541298866272, |
|
"logps/rejected": -1.5116033554077148, |
|
"loss": 1.825, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.132541298866272, |
|
"rewards/margins": 0.3790620267391205, |
|
"rewards/rejected": -1.5116033554077148, |
|
"step": 455 |
|
}, |
|
{ |
|
"epoch": 0.9630986652708715, |
|
"grad_norm": 31.401214396279837, |
|
"learning_rate": 1.9347820230782295e-09, |
|
"logits/chosen": -2.6828560829162598, |
|
"logits/rejected": -2.6430163383483887, |
|
"logps/chosen": -0.9926983118057251, |
|
"logps/rejected": -1.5756382942199707, |
|
"loss": 1.7905, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.9926983118057251, |
|
"rewards/margins": 0.5829401612281799, |
|
"rewards/rejected": -1.5756382942199707, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.9735671290238157, |
|
"grad_norm": 38.51782864695839, |
|
"learning_rate": 9.64668657069706e-10, |
|
"logits/chosen": -2.7383384704589844, |
|
"logits/rejected": -2.6524176597595215, |
|
"logps/chosen": -1.0901256799697876, |
|
"logps/rejected": -1.620060682296753, |
|
"loss": 1.8476, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.0901256799697876, |
|
"rewards/margins": 0.5299350023269653, |
|
"rewards/rejected": -1.620060682296753, |
|
"step": 465 |
|
}, |
|
{ |
|
"epoch": 0.98403559277676, |
|
"grad_norm": 27.923910450585048, |
|
"learning_rate": 3.2839470889836627e-10, |
|
"logits/chosen": -2.7122910022735596, |
|
"logits/rejected": -2.6457693576812744, |
|
"logps/chosen": -0.9990774393081665, |
|
"logps/rejected": -1.4476706981658936, |
|
"loss": 1.8714, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.9990774393081665, |
|
"rewards/margins": 0.4485931396484375, |
|
"rewards/rejected": -1.4476706981658936, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.9945040565297043, |
|
"grad_norm": 33.978087151320835, |
|
"learning_rate": 2.6813123097352287e-11, |
|
"logits/chosen": -2.7624480724334717, |
|
"logits/rejected": -2.6619315147399902, |
|
"logps/chosen": -1.042180061340332, |
|
"logps/rejected": -1.5233901739120483, |
|
"loss": 1.8233, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.042180061340332, |
|
"rewards/margins": 0.4812101721763611, |
|
"rewards/rejected": -1.5233901739120483, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.998691442030882, |
|
"step": 477, |
|
"total_flos": 0.0, |
|
"train_loss": 1.8644854217705238, |
|
"train_runtime": 7378.4782, |
|
"train_samples_per_second": 8.286, |
|
"train_steps_per_second": 0.065 |
|
} |
|
], |
|
"logging_steps": 5, |
|
"max_steps": 477, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 1000000, |
|
"total_flos": 0.0, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|