|
{ |
|
"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": 10.600991740982659, |
|
"learning_rate": 5.208333333333333e-08, |
|
"logits/chosen": -0.7836824655532837, |
|
"logits/rejected": -0.8418347239494324, |
|
"logps/chosen": -1.174766182899475, |
|
"logps/rejected": -1.3588011264801025, |
|
"loss": -0.0349, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -1.174766182899475, |
|
"rewards/margins": 0.1840350478887558, |
|
"rewards/rejected": -1.3588011264801025, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.02093692750588851, |
|
"grad_norm": 2.7100936762262937, |
|
"learning_rate": 1.0416666666666667e-07, |
|
"logits/chosen": -0.8232773542404175, |
|
"logits/rejected": -0.8316025733947754, |
|
"logps/chosen": -1.1583973169326782, |
|
"logps/rejected": -1.2626698017120361, |
|
"loss": -0.0305, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -1.1583973169326782, |
|
"rewards/margins": 0.10427255928516388, |
|
"rewards/rejected": -1.2626698017120361, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.031405391258832765, |
|
"grad_norm": 5.660375571877667, |
|
"learning_rate": 1.5624999999999999e-07, |
|
"logits/chosen": -0.7926589846611023, |
|
"logits/rejected": -0.8048545718193054, |
|
"logps/chosen": -1.1033742427825928, |
|
"logps/rejected": -1.3587285280227661, |
|
"loss": -0.0465, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.1033742427825928, |
|
"rewards/margins": 0.25535425543785095, |
|
"rewards/rejected": -1.3587285280227661, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.04187385501177702, |
|
"grad_norm": 3.7357000754298184, |
|
"learning_rate": 2.0833333333333333e-07, |
|
"logits/chosen": -0.7191373705863953, |
|
"logits/rejected": -0.8023589253425598, |
|
"logps/chosen": -1.1485874652862549, |
|
"logps/rejected": -1.253619909286499, |
|
"loss": -0.0339, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -1.1485874652862549, |
|
"rewards/margins": 0.10503246635198593, |
|
"rewards/rejected": -1.253619909286499, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05234231876472128, |
|
"grad_norm": 5.416875927919006, |
|
"learning_rate": 2.604166666666667e-07, |
|
"logits/chosen": -0.806380569934845, |
|
"logits/rejected": -0.8102810978889465, |
|
"logps/chosen": -1.1436618566513062, |
|
"logps/rejected": -1.2122325897216797, |
|
"loss": -0.0392, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -1.1436618566513062, |
|
"rewards/margins": 0.06857069581747055, |
|
"rewards/rejected": -1.2122325897216797, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.06281078251766553, |
|
"grad_norm": 5.913782183964252, |
|
"learning_rate": 3.1249999999999997e-07, |
|
"logits/chosen": -0.7805474996566772, |
|
"logits/rejected": -0.8070163726806641, |
|
"logps/chosen": -1.0784746408462524, |
|
"logps/rejected": -1.2096612453460693, |
|
"loss": -0.0302, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -1.0784746408462524, |
|
"rewards/margins": 0.13118666410446167, |
|
"rewards/rejected": -1.2096612453460693, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.07327924627060979, |
|
"grad_norm": 2.6541139134020133, |
|
"learning_rate": 3.645833333333333e-07, |
|
"logits/chosen": -0.814271092414856, |
|
"logits/rejected": -0.7939152121543884, |
|
"logps/chosen": -1.046288013458252, |
|
"logps/rejected": -1.1717827320098877, |
|
"loss": -0.043, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -1.046288013458252, |
|
"rewards/margins": 0.12549468874931335, |
|
"rewards/rejected": -1.1717827320098877, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.08374771002355404, |
|
"grad_norm": 2.3868591785059694, |
|
"learning_rate": 4.1666666666666667e-07, |
|
"logits/chosen": -0.7904576063156128, |
|
"logits/rejected": -0.9059640765190125, |
|
"logps/chosen": -0.966705322265625, |
|
"logps/rejected": -1.2806286811828613, |
|
"loss": -0.0588, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.966705322265625, |
|
"rewards/margins": 0.31392329931259155, |
|
"rewards/rejected": -1.2806286811828613, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.0942161737764983, |
|
"grad_norm": 3.8242068571740493, |
|
"learning_rate": 4.6874999999999996e-07, |
|
"logits/chosen": -0.7696360349655151, |
|
"logits/rejected": -0.834786057472229, |
|
"logps/chosen": -0.9762159585952759, |
|
"logps/rejected": -1.1492093801498413, |
|
"loss": -0.0549, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.9762159585952759, |
|
"rewards/margins": 0.17299343645572662, |
|
"rewards/rejected": -1.1492093801498413, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.10468463752944256, |
|
"grad_norm": 3.349329434355001, |
|
"learning_rate": 4.999731868769026e-07, |
|
"logits/chosen": -0.8303428888320923, |
|
"logits/rejected": -0.9099453687667847, |
|
"logps/chosen": -1.0054728984832764, |
|
"logps/rejected": -1.2581647634506226, |
|
"loss": -0.0438, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.0054728984832764, |
|
"rewards/margins": 0.2526918649673462, |
|
"rewards/rejected": -1.2581647634506226, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.11515310128238682, |
|
"grad_norm": 17.990360349960177, |
|
"learning_rate": 4.996716052911017e-07, |
|
"logits/chosen": -0.8441118001937866, |
|
"logits/rejected": -0.8858628273010254, |
|
"logps/chosen": -0.9657171368598938, |
|
"logps/rejected": -1.1507956981658936, |
|
"loss": -0.0487, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.9657171368598938, |
|
"rewards/margins": 0.18507853150367737, |
|
"rewards/rejected": -1.1507956981658936, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.12562156503533106, |
|
"grad_norm": 2.4604104425775093, |
|
"learning_rate": 4.990353313429303e-07, |
|
"logits/chosen": -0.866470456123352, |
|
"logits/rejected": -0.9195329546928406, |
|
"logps/chosen": -1.0146195888519287, |
|
"logps/rejected": -1.1302388906478882, |
|
"loss": -0.0599, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -1.0146195888519287, |
|
"rewards/margins": 0.11561928689479828, |
|
"rewards/rejected": -1.1302388906478882, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.1360900287882753, |
|
"grad_norm": 2.386753258565622, |
|
"learning_rate": 4.980652179769217e-07, |
|
"logits/chosen": -0.8147915601730347, |
|
"logits/rejected": -0.8957066535949707, |
|
"logps/chosen": -0.9518657922744751, |
|
"logps/rejected": -1.295042872428894, |
|
"loss": -0.0521, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.9518657922744751, |
|
"rewards/margins": 0.3431771397590637, |
|
"rewards/rejected": -1.295042872428894, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.14655849254121958, |
|
"grad_norm": 2.547052916904684, |
|
"learning_rate": 4.967625656594781e-07, |
|
"logits/chosen": -0.8104137182235718, |
|
"logits/rejected": -0.8524875640869141, |
|
"logps/chosen": -0.9991258382797241, |
|
"logps/rejected": -1.169449806213379, |
|
"loss": -0.0541, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.9991258382797241, |
|
"rewards/margins": 0.17032404243946075, |
|
"rewards/rejected": -1.169449806213379, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.15702695629416383, |
|
"grad_norm": 3.4870962135797243, |
|
"learning_rate": 4.951291206355559e-07, |
|
"logits/chosen": -0.8058147430419922, |
|
"logits/rejected": -0.8437595367431641, |
|
"logps/chosen": -0.9659625291824341, |
|
"logps/rejected": -1.2181332111358643, |
|
"loss": -0.0584, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.9659625291824341, |
|
"rewards/margins": 0.25217074155807495, |
|
"rewards/rejected": -1.2181332111358643, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.16749542004710807, |
|
"grad_norm": 11.125744861715972, |
|
"learning_rate": 4.93167072587771e-07, |
|
"logits/chosen": -0.8071163892745972, |
|
"logits/rejected": -0.8949975967407227, |
|
"logps/chosen": -0.9791210293769836, |
|
"logps/rejected": -1.2237764596939087, |
|
"loss": -0.0513, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.9791210293769836, |
|
"rewards/margins": 0.24465537071228027, |
|
"rewards/rejected": -1.2237764596939087, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.17796388380005235, |
|
"grad_norm": 3.898865350054578, |
|
"learning_rate": 4.908790517010636e-07, |
|
"logits/chosen": -0.8957134485244751, |
|
"logits/rejected": -0.9214810132980347, |
|
"logps/chosen": -0.9375537633895874, |
|
"logps/rejected": -1.1478058099746704, |
|
"loss": -0.057, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.9375537633895874, |
|
"rewards/margins": 0.21025200188159943, |
|
"rewards/rejected": -1.1478058099746704, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 0.1884323475529966, |
|
"grad_norm": 3.582473358962524, |
|
"learning_rate": 4.882681251368548e-07, |
|
"logits/chosen": -0.8433510661125183, |
|
"logits/rejected": -0.8857067227363586, |
|
"logps/chosen": -1.0094842910766602, |
|
"logps/rejected": -1.2096831798553467, |
|
"loss": -0.0644, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.0094842910766602, |
|
"rewards/margins": 0.20019885897636414, |
|
"rewards/rejected": -1.2096831798553467, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.19890081130594087, |
|
"grad_norm": 2.6501926608400797, |
|
"learning_rate": 4.853377929214243e-07, |
|
"logits/chosen": -0.8576142191886902, |
|
"logits/rejected": -0.8764857053756714, |
|
"logps/chosen": -0.9934334754943848, |
|
"logps/rejected": -1.1720221042633057, |
|
"loss": -0.0505, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.9934334754943848, |
|
"rewards/margins": 0.1785886585712433, |
|
"rewards/rejected": -1.1720221042633057, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 0.2093692750588851, |
|
"grad_norm": 33.88769391502933, |
|
"learning_rate": 4.820919832540181e-07, |
|
"logits/chosen": -0.8933563232421875, |
|
"logits/rejected": -0.9612926244735718, |
|
"logps/chosen": -1.0248340368270874, |
|
"logps/rejected": -1.2969132661819458, |
|
"loss": -0.0529, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -1.0248340368270874, |
|
"rewards/margins": 0.2720790505409241, |
|
"rewards/rejected": -1.2969132661819458, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.21983773881182936, |
|
"grad_norm": 5.123142096970102, |
|
"learning_rate": 4.785350472409791e-07, |
|
"logits/chosen": -0.8836784362792969, |
|
"logits/rejected": -0.9229451417922974, |
|
"logps/chosen": -0.9697669744491577, |
|
"logps/rejected": -1.3045259714126587, |
|
"loss": -0.063, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.9697669744491577, |
|
"rewards/margins": 0.3347588777542114, |
|
"rewards/rejected": -1.3045259714126587, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 0.23030620256477363, |
|
"grad_norm": 4.05979796610575, |
|
"learning_rate": 4.7467175306295647e-07, |
|
"logits/chosen": -0.9353694915771484, |
|
"logits/rejected": -0.98784339427948, |
|
"logps/chosen": -1.0010242462158203, |
|
"logps/rejected": -1.3293567895889282, |
|
"loss": -0.0709, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.0010242462158203, |
|
"rewards/margins": 0.3283325731754303, |
|
"rewards/rejected": -1.3293567895889282, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.24077466631771788, |
|
"grad_norm": 4.515920103518208, |
|
"learning_rate": 4.70507279583015e-07, |
|
"logits/chosen": -0.8709474802017212, |
|
"logits/rejected": -0.9586912393569946, |
|
"logps/chosen": -0.9947507977485657, |
|
"logps/rejected": -1.400367021560669, |
|
"loss": -0.0838, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.9947507977485657, |
|
"rewards/margins": 0.40561628341674805, |
|
"rewards/rejected": -1.400367021560669, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 0.2512431300706621, |
|
"grad_norm": 4.110081424436634, |
|
"learning_rate": 4.6604720940421207e-07, |
|
"logits/chosen": -0.8883617520332336, |
|
"logits/rejected": -0.9269152879714966, |
|
"logps/chosen": -1.0870624780654907, |
|
"logps/rejected": -1.3331050872802734, |
|
"loss": -0.0626, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -1.0870624780654907, |
|
"rewards/margins": 0.24604257941246033, |
|
"rewards/rejected": -1.3331050872802734, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.26171159382360637, |
|
"grad_norm": 3.7209294499373624, |
|
"learning_rate": 4.612975213859487e-07, |
|
"logits/chosen": -0.9534390568733215, |
|
"logits/rejected": -0.9193525314331055, |
|
"logps/chosen": -1.002967119216919, |
|
"logps/rejected": -1.2403877973556519, |
|
"loss": -0.0683, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.002967119216919, |
|
"rewards/margins": 0.23742058873176575, |
|
"rewards/rejected": -1.2403877973556519, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.2721800575765506, |
|
"grad_norm": 3.499758787421745, |
|
"learning_rate": 4.5626458262912735e-07, |
|
"logits/chosen": -0.8969480395317078, |
|
"logits/rejected": -0.9590781927108765, |
|
"logps/chosen": -1.0254695415496826, |
|
"logps/rejected": -1.1852182149887085, |
|
"loss": -0.0663, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -1.0254695415496826, |
|
"rewards/margins": 0.15974867343902588, |
|
"rewards/rejected": -1.1852182149887085, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.2826485213294949, |
|
"grad_norm": 4.321280373798233, |
|
"learning_rate": 4.5095513994085974e-07, |
|
"logits/chosen": -0.8857036828994751, |
|
"logits/rejected": -0.9329354166984558, |
|
"logps/chosen": -1.0760549306869507, |
|
"logps/rejected": -1.5539740324020386, |
|
"loss": -0.0777, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -1.0760549306869507, |
|
"rewards/margins": 0.47791916131973267, |
|
"rewards/rejected": -1.5539740324020386, |
|
"step": 135 |
|
}, |
|
{ |
|
"epoch": 0.29311698508243916, |
|
"grad_norm": 3.205749007042699, |
|
"learning_rate": 4.453763107901675e-07, |
|
"logits/chosen": -0.9192848205566406, |
|
"logits/rejected": -0.9621114730834961, |
|
"logps/chosen": -1.0559959411621094, |
|
"logps/rejected": -1.453352451324463, |
|
"loss": -0.0865, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.0559959411621094, |
|
"rewards/margins": 0.39735639095306396, |
|
"rewards/rejected": -1.453352451324463, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.3035854488353834, |
|
"grad_norm": 5.90034906637265, |
|
"learning_rate": 4.395355737667985e-07, |
|
"logits/chosen": -0.9266332387924194, |
|
"logits/rejected": -1.0155680179595947, |
|
"logps/chosen": -1.074566125869751, |
|
"logps/rejected": -1.3885366916656494, |
|
"loss": -0.0784, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.074566125869751, |
|
"rewards/margins": 0.3139706254005432, |
|
"rewards/rejected": -1.3885366916656494, |
|
"step": 145 |
|
}, |
|
{ |
|
"epoch": 0.31405391258832765, |
|
"grad_norm": 4.108356388429939, |
|
"learning_rate": 4.3344075855595097e-07, |
|
"logits/chosen": -0.9286434054374695, |
|
"logits/rejected": -0.9394516944885254, |
|
"logps/chosen": -0.9996128082275391, |
|
"logps/rejected": -1.3331079483032227, |
|
"loss": -0.0833, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.9996128082275391, |
|
"rewards/margins": 0.3334951102733612, |
|
"rewards/rejected": -1.3331079483032227, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.3245223763412719, |
|
"grad_norm": 12.905833082101557, |
|
"learning_rate": 4.271000354423425e-07, |
|
"logits/chosen": -0.9297893643379211, |
|
"logits/rejected": -0.9530170559883118, |
|
"logps/chosen": -1.0625531673431396, |
|
"logps/rejected": -1.3642576932907104, |
|
"loss": -0.087, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.0625531673431396, |
|
"rewards/margins": 0.3017045855522156, |
|
"rewards/rejected": -1.3642576932907104, |
|
"step": 155 |
|
}, |
|
{ |
|
"epoch": 0.33499084009421615, |
|
"grad_norm": 4.043990046291937, |
|
"learning_rate": 4.2052190435769554e-07, |
|
"logits/chosen": -0.9249881505966187, |
|
"logits/rejected": -0.9575279355049133, |
|
"logps/chosen": -1.0861815214157104, |
|
"logps/rejected": -1.4002734422683716, |
|
"loss": -0.0807, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.0861815214157104, |
|
"rewards/margins": 0.31409186124801636, |
|
"rewards/rejected": -1.4002734422683716, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.34545930384716045, |
|
"grad_norm": 4.916459068211759, |
|
"learning_rate": 4.137151834863213e-07, |
|
"logits/chosen": -0.9621366262435913, |
|
"logits/rejected": -0.995520293712616, |
|
"logps/chosen": -1.0501563549041748, |
|
"logps/rejected": -1.4903188943862915, |
|
"loss": -0.0907, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.0501563549041748, |
|
"rewards/margins": 0.4401625096797943, |
|
"rewards/rejected": -1.4903188943862915, |
|
"step": 165 |
|
}, |
|
{ |
|
"epoch": 0.3559277676001047, |
|
"grad_norm": 5.335366984629706, |
|
"learning_rate": 4.0668899744407567e-07, |
|
"logits/chosen": -0.9441369771957397, |
|
"logits/rejected": -0.9942566156387329, |
|
"logps/chosen": -1.1764615774154663, |
|
"logps/rejected": -1.5894583463668823, |
|
"loss": -0.08, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.1764615774154663, |
|
"rewards/margins": 0.41299691796302795, |
|
"rewards/rejected": -1.5894583463668823, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.36639623135304894, |
|
"grad_norm": 6.274854639434234, |
|
"learning_rate": 3.994527650465352e-07, |
|
"logits/chosen": -0.9332780838012695, |
|
"logits/rejected": -1.0104329586029053, |
|
"logps/chosen": -1.1270277500152588, |
|
"logps/rejected": -1.563685655593872, |
|
"loss": -0.0818, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.1270277500152588, |
|
"rewards/margins": 0.4366579055786133, |
|
"rewards/rejected": -1.563685655593872, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.3768646951059932, |
|
"grad_norm": 7.8424698369148444, |
|
"learning_rate": 3.920161866827889e-07, |
|
"logits/chosen": -0.9343840479850769, |
|
"logits/rejected": -1.0072158575057983, |
|
"logps/chosen": -1.1227374076843262, |
|
"logps/rejected": -1.5470798015594482, |
|
"loss": -0.0868, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.1227374076843262, |
|
"rewards/margins": 0.42434263229370117, |
|
"rewards/rejected": -1.5470798015594482, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.38733315885893743, |
|
"grad_norm": 4.338056644815217, |
|
"learning_rate": 3.8438923131177237e-07, |
|
"logits/chosen": -0.8708566427230835, |
|
"logits/rejected": -0.9427552223205566, |
|
"logps/chosen": -1.187537670135498, |
|
"logps/rejected": -1.823570966720581, |
|
"loss": -0.1045, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.187537670135498, |
|
"rewards/margins": 0.636033296585083, |
|
"rewards/rejected": -1.823570966720581, |
|
"step": 185 |
|
}, |
|
{ |
|
"epoch": 0.39780162261188173, |
|
"grad_norm": 6.673093826325478, |
|
"learning_rate": 3.765821230985757e-07, |
|
"logits/chosen": -0.9710279703140259, |
|
"logits/rejected": -0.9876633882522583, |
|
"logps/chosen": -1.1750192642211914, |
|
"logps/rejected": -1.6105972528457642, |
|
"loss": -0.0772, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.1750192642211914, |
|
"rewards/margins": 0.435577929019928, |
|
"rewards/rejected": -1.6105972528457642, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.408270086364826, |
|
"grad_norm": 6.960384794729192, |
|
"learning_rate": 3.6860532770864005e-07, |
|
"logits/chosen": -0.9228804707527161, |
|
"logits/rejected": -0.9380979537963867, |
|
"logps/chosen": -1.299344778060913, |
|
"logps/rejected": -1.6751635074615479, |
|
"loss": -0.0845, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.299344778060913, |
|
"rewards/margins": 0.37581872940063477, |
|
"rewards/rejected": -1.6751635074615479, |
|
"step": 195 |
|
}, |
|
{ |
|
"epoch": 0.4187385501177702, |
|
"grad_norm": 6.471834662041886, |
|
"learning_rate": 3.604695382782159e-07, |
|
"logits/chosen": -0.9913051724433899, |
|
"logits/rejected": -0.992074191570282, |
|
"logps/chosen": -1.221297264099121, |
|
"logps/rejected": -1.7224781513214111, |
|
"loss": -0.0952, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.221297264099121, |
|
"rewards/margins": 0.5011808276176453, |
|
"rewards/rejected": -1.7224781513214111, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.42920701387071447, |
|
"grad_norm": 6.306403211422552, |
|
"learning_rate": 3.5218566107988867e-07, |
|
"logits/chosen": -0.883250892162323, |
|
"logits/rejected": -0.9081627726554871, |
|
"logps/chosen": -1.1198192834854126, |
|
"logps/rejected": -1.7429187297821045, |
|
"loss": -0.1047, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.1198192834854126, |
|
"rewards/margins": 0.6230995059013367, |
|
"rewards/rejected": -1.7429187297821045, |
|
"step": 205 |
|
}, |
|
{ |
|
"epoch": 0.4396754776236587, |
|
"grad_norm": 25.678467464556803, |
|
"learning_rate": 3.4376480090239047e-07, |
|
"logits/chosen": -0.8838554620742798, |
|
"logits/rejected": -0.9190862774848938, |
|
"logps/chosen": -1.416511058807373, |
|
"logps/rejected": -1.6893377304077148, |
|
"loss": -0.0809, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -1.416511058807373, |
|
"rewards/margins": 0.27282652258872986, |
|
"rewards/rejected": -1.6893377304077148, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.45014394137660296, |
|
"grad_norm": 16.57780152635486, |
|
"learning_rate": 3.3521824616429284e-07, |
|
"logits/chosen": -0.8571985363960266, |
|
"logits/rejected": -0.9235578775405884, |
|
"logps/chosen": -1.1465059518814087, |
|
"logps/rejected": -1.7085803747177124, |
|
"loss": -0.0954, |
|
"rewards/accuracies": 0.731249988079071, |
|
"rewards/chosen": -1.1465059518814087, |
|
"rewards/margins": 0.5620743036270142, |
|
"rewards/rejected": -1.7085803747177124, |
|
"step": 215 |
|
}, |
|
{ |
|
"epoch": 0.46061240512954726, |
|
"grad_norm": 14.206901054524915, |
|
"learning_rate": 3.265574537815398e-07, |
|
"logits/chosen": -0.8814586400985718, |
|
"logits/rejected": -0.9272589683532715, |
|
"logps/chosen": -1.1567518711090088, |
|
"logps/rejected": -1.6502008438110352, |
|
"loss": -0.0723, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.1567518711090088, |
|
"rewards/margins": 0.4934488832950592, |
|
"rewards/rejected": -1.6502008438110352, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.4710808688824915, |
|
"grad_norm": 6.6240353334989495, |
|
"learning_rate": 3.1779403380910425e-07, |
|
"logits/chosen": -0.8806575536727905, |
|
"logits/rejected": -0.9517351388931274, |
|
"logps/chosen": -1.1226614713668823, |
|
"logps/rejected": -1.7587499618530273, |
|
"loss": -0.0895, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.1226614713668823, |
|
"rewards/margins": 0.6360886096954346, |
|
"rewards/rejected": -1.7587499618530273, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.48154933263543576, |
|
"grad_norm": 4.308434013872677, |
|
"learning_rate": 3.0893973387735683e-07, |
|
"logits/chosen": -0.8706935048103333, |
|
"logits/rejected": -0.9313523173332214, |
|
"logps/chosen": -1.2426685094833374, |
|
"logps/rejected": -2.1209378242492676, |
|
"loss": -0.1094, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.2426685094833374, |
|
"rewards/margins": 0.878269374370575, |
|
"rewards/rejected": -2.1209378242492676, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.49201779638838, |
|
"grad_norm": 6.844252123006064, |
|
"learning_rate": 3.000064234440111e-07, |
|
"logits/chosen": -0.8923465609550476, |
|
"logits/rejected": -0.9771193265914917, |
|
"logps/chosen": -1.096273422241211, |
|
"logps/rejected": -1.7108914852142334, |
|
"loss": -0.1035, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -1.096273422241211, |
|
"rewards/margins": 0.6146181225776672, |
|
"rewards/rejected": -1.7108914852142334, |
|
"step": 235 |
|
}, |
|
{ |
|
"epoch": 0.5024862601413242, |
|
"grad_norm": 4.117878427703188, |
|
"learning_rate": 2.910060778827554e-07, |
|
"logits/chosen": -0.8880583047866821, |
|
"logits/rejected": -0.9422343373298645, |
|
"logps/chosen": -1.187822699546814, |
|
"logps/rejected": -1.7649528980255127, |
|
"loss": -0.0888, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.187822699546814, |
|
"rewards/margins": 0.5771301984786987, |
|
"rewards/rejected": -1.7649528980255127, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.5129547238942685, |
|
"grad_norm": 11.906403629187743, |
|
"learning_rate": 2.8195076242990116e-07, |
|
"logits/chosen": -0.9159537553787231, |
|
"logits/rejected": -0.959700882434845, |
|
"logps/chosen": -1.292806625366211, |
|
"logps/rejected": -1.880102515220642, |
|
"loss": -0.1005, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -1.292806625366211, |
|
"rewards/margins": 0.5872958898544312, |
|
"rewards/rejected": -1.880102515220642, |
|
"step": 245 |
|
}, |
|
{ |
|
"epoch": 0.5234231876472127, |
|
"grad_norm": 9.645796474062738, |
|
"learning_rate": 2.7285261601056697e-07, |
|
"logits/chosen": -0.8488172292709351, |
|
"logits/rejected": -0.9300807118415833, |
|
"logps/chosen": -1.2770051956176758, |
|
"logps/rejected": -1.9150631427764893, |
|
"loss": -0.0977, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.2770051956176758, |
|
"rewards/margins": 0.6380580067634583, |
|
"rewards/rejected": -1.9150631427764893, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.533891651400157, |
|
"grad_norm": 4.731035112524077, |
|
"learning_rate": 2.6372383496608186e-07, |
|
"logits/chosen": -0.9005299806594849, |
|
"logits/rejected": -0.9532705545425415, |
|
"logps/chosen": -1.309429407119751, |
|
"logps/rejected": -2.0657031536102295, |
|
"loss": -0.1139, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.309429407119751, |
|
"rewards/margins": 0.7562738656997681, |
|
"rewards/rejected": -2.0657031536102295, |
|
"step": 255 |
|
}, |
|
{ |
|
"epoch": 0.5443601151531012, |
|
"grad_norm": 7.325832832673913, |
|
"learning_rate": 2.5457665670441937e-07, |
|
"logits/chosen": -0.8450838327407837, |
|
"logits/rejected": -0.9038773775100708, |
|
"logps/chosen": -1.0662903785705566, |
|
"logps/rejected": -1.9864327907562256, |
|
"loss": -0.106, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -1.0662903785705566, |
|
"rewards/margins": 0.9201424717903137, |
|
"rewards/rejected": -1.9864327907562256, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.5548285789060455, |
|
"grad_norm": 11.31094245733987, |
|
"learning_rate": 2.454233432955807e-07, |
|
"logits/chosen": -0.8820160031318665, |
|
"logits/rejected": -0.9507255554199219, |
|
"logps/chosen": -1.221882700920105, |
|
"logps/rejected": -1.6403167247772217, |
|
"loss": -0.0993, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.221882700920105, |
|
"rewards/margins": 0.4184340536594391, |
|
"rewards/rejected": -1.6403167247772217, |
|
"step": 265 |
|
}, |
|
{ |
|
"epoch": 0.5652970426589898, |
|
"grad_norm": 7.9126915493471675, |
|
"learning_rate": 2.3627616503391812e-07, |
|
"logits/chosen": -0.911536693572998, |
|
"logits/rejected": -0.9098671674728394, |
|
"logps/chosen": -1.3050767183303833, |
|
"logps/rejected": -1.6615794897079468, |
|
"loss": -0.0834, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.3050767183303833, |
|
"rewards/margins": 0.35650262236595154, |
|
"rewards/rejected": -1.6615794897079468, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.575765506411934, |
|
"grad_norm": 4.215844646514211, |
|
"learning_rate": 2.2714738398943308e-07, |
|
"logits/chosen": -0.9258629083633423, |
|
"logits/rejected": -0.9768645167350769, |
|
"logps/chosen": -1.1248077154159546, |
|
"logps/rejected": -1.950279951095581, |
|
"loss": -0.1036, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.1248077154159546, |
|
"rewards/margins": 0.8254722356796265, |
|
"rewards/rejected": -1.950279951095581, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.5862339701648783, |
|
"grad_norm": 4.592191084246013, |
|
"learning_rate": 2.1804923757009882e-07, |
|
"logits/chosen": -0.9678147435188293, |
|
"logits/rejected": -1.0440599918365479, |
|
"logps/chosen": -1.1835848093032837, |
|
"logps/rejected": -1.6567246913909912, |
|
"loss": -0.0968, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.1835848093032837, |
|
"rewards/margins": 0.4731399416923523, |
|
"rewards/rejected": -1.6567246913909912, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.5967024339178225, |
|
"grad_norm": 12.623165618800098, |
|
"learning_rate": 2.089939221172446e-07, |
|
"logits/chosen": -0.9032037854194641, |
|
"logits/rejected": -0.9397640228271484, |
|
"logps/chosen": -1.2246944904327393, |
|
"logps/rejected": -1.975319504737854, |
|
"loss": -0.1062, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.2246944904327393, |
|
"rewards/margins": 0.7506248950958252, |
|
"rewards/rejected": -1.975319504737854, |
|
"step": 285 |
|
}, |
|
{ |
|
"epoch": 0.6071708976707668, |
|
"grad_norm": 5.38289989385453, |
|
"learning_rate": 1.9999357655598891e-07, |
|
"logits/chosen": -0.909401535987854, |
|
"logits/rejected": -1.0137306451797485, |
|
"logps/chosen": -1.3724336624145508, |
|
"logps/rejected": -2.2477662563323975, |
|
"loss": -0.1042, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.3724336624145508, |
|
"rewards/margins": 0.8753326535224915, |
|
"rewards/rejected": -2.2477662563323975, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.6176393614237111, |
|
"grad_norm": 7.401969971775179, |
|
"learning_rate": 1.9106026612264315e-07, |
|
"logits/chosen": -0.8931488990783691, |
|
"logits/rejected": -0.9808778762817383, |
|
"logps/chosen": -1.287915825843811, |
|
"logps/rejected": -2.0314321517944336, |
|
"loss": -0.0964, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.287915825843811, |
|
"rewards/margins": 0.7435161471366882, |
|
"rewards/rejected": -2.0314321517944336, |
|
"step": 295 |
|
}, |
|
{ |
|
"epoch": 0.6281078251766553, |
|
"grad_norm": 6.296173137370066, |
|
"learning_rate": 1.8220596619089573e-07, |
|
"logits/chosen": -0.9120398759841919, |
|
"logits/rejected": -0.9633253216743469, |
|
"logps/chosen": -1.3466119766235352, |
|
"logps/rejected": -1.948876976966858, |
|
"loss": -0.0998, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -1.3466119766235352, |
|
"rewards/margins": 0.6022650003433228, |
|
"rewards/rejected": -1.948876976966858, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.6385762889295996, |
|
"grad_norm": 6.12276434326567, |
|
"learning_rate": 1.7344254621846017e-07, |
|
"logits/chosen": -0.8898782730102539, |
|
"logits/rejected": -0.9864279627799988, |
|
"logps/chosen": -1.3288440704345703, |
|
"logps/rejected": -2.1605937480926514, |
|
"loss": -0.1155, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -1.3288440704345703, |
|
"rewards/margins": 0.8317497372627258, |
|
"rewards/rejected": -2.1605937480926514, |
|
"step": 305 |
|
}, |
|
{ |
|
"epoch": 0.6490447526825438, |
|
"grad_norm": 35.82187771529538, |
|
"learning_rate": 1.647817538357072e-07, |
|
"logits/chosen": -0.917158305644989, |
|
"logits/rejected": -0.9460798501968384, |
|
"logps/chosen": -1.1947271823883057, |
|
"logps/rejected": -1.9670846462249756, |
|
"loss": -0.1061, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.1947271823883057, |
|
"rewards/margins": 0.7723575830459595, |
|
"rewards/rejected": -1.9670846462249756, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.6595132164354881, |
|
"grad_norm": 7.813472688510903, |
|
"learning_rate": 1.562351990976095e-07, |
|
"logits/chosen": -0.9020414352416992, |
|
"logits/rejected": -0.9904358983039856, |
|
"logps/chosen": -1.1299612522125244, |
|
"logps/rejected": -2.129662036895752, |
|
"loss": -0.1206, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.1299612522125244, |
|
"rewards/margins": 0.9997007250785828, |
|
"rewards/rejected": -2.129662036895752, |
|
"step": 315 |
|
}, |
|
{ |
|
"epoch": 0.6699816801884323, |
|
"grad_norm": 10.278529226899488, |
|
"learning_rate": 1.478143389201113e-07, |
|
"logits/chosen": -0.9329323768615723, |
|
"logits/rejected": -0.9159154891967773, |
|
"logps/chosen": -1.2115180492401123, |
|
"logps/rejected": -1.8157265186309814, |
|
"loss": -0.1025, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.2115180492401123, |
|
"rewards/margins": 0.6042084097862244, |
|
"rewards/rejected": -1.8157265186309814, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.6804501439413766, |
|
"grad_norm": 4.623648192830834, |
|
"learning_rate": 1.3953046172178413e-07, |
|
"logits/chosen": -0.9241701364517212, |
|
"logits/rejected": -0.9985548257827759, |
|
"logps/chosen": -1.2149182558059692, |
|
"logps/rejected": -1.8975908756256104, |
|
"loss": -0.1173, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.2149182558059692, |
|
"rewards/margins": 0.6826726198196411, |
|
"rewards/rejected": -1.8975908756256104, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.6909186076943209, |
|
"grad_norm": 4.284608818953695, |
|
"learning_rate": 1.3139467229135998e-07, |
|
"logits/chosen": -0.9073573350906372, |
|
"logits/rejected": -0.970782458782196, |
|
"logps/chosen": -1.1777896881103516, |
|
"logps/rejected": -1.862392783164978, |
|
"loss": -0.0988, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.1777896881103516, |
|
"rewards/margins": 0.6846030950546265, |
|
"rewards/rejected": -1.862392783164978, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.7013870714472651, |
|
"grad_norm": 11.616905967515871, |
|
"learning_rate": 1.2341787690142435e-07, |
|
"logits/chosen": -0.9223908185958862, |
|
"logits/rejected": -0.9389724731445312, |
|
"logps/chosen": -1.1395795345306396, |
|
"logps/rejected": -1.7211112976074219, |
|
"loss": -0.0961, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -1.1395795345306396, |
|
"rewards/margins": 0.5815316438674927, |
|
"rewards/rejected": -1.7211112976074219, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 0.7118555352002094, |
|
"grad_norm": 6.760822618737345, |
|
"learning_rate": 1.1561076868822755e-07, |
|
"logits/chosen": -0.8945438265800476, |
|
"logits/rejected": -0.9830290079116821, |
|
"logps/chosen": -1.3890964984893799, |
|
"logps/rejected": -2.09273624420166, |
|
"loss": -0.1051, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.3890964984893799, |
|
"rewards/margins": 0.7036401033401489, |
|
"rewards/rejected": -2.09273624420166, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.7223239989531536, |
|
"grad_norm": 5.837717059446264, |
|
"learning_rate": 1.0798381331721107e-07, |
|
"logits/chosen": -0.920322060585022, |
|
"logits/rejected": -0.977412223815918, |
|
"logps/chosen": -1.2339359521865845, |
|
"logps/rejected": -2.0405447483062744, |
|
"loss": -0.1013, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.2339359521865845, |
|
"rewards/margins": 0.8066087961196899, |
|
"rewards/rejected": -2.0405447483062744, |
|
"step": 345 |
|
}, |
|
{ |
|
"epoch": 0.7327924627060979, |
|
"grad_norm": 5.994873994092868, |
|
"learning_rate": 1.0054723495346482e-07, |
|
"logits/chosen": -0.9375583529472351, |
|
"logits/rejected": -0.9326186180114746, |
|
"logps/chosen": -1.276450753211975, |
|
"logps/rejected": -1.994940161705017, |
|
"loss": -0.0972, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -1.276450753211975, |
|
"rewards/margins": 0.7184895277023315, |
|
"rewards/rejected": -1.994940161705017, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.7432609264590422, |
|
"grad_norm": 7.791758915935874, |
|
"learning_rate": 9.331100255592436e-08, |
|
"logits/chosen": -0.9378272294998169, |
|
"logits/rejected": -0.9758423566818237, |
|
"logps/chosen": -1.1630890369415283, |
|
"logps/rejected": -2.039836883544922, |
|
"loss": -0.1098, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -1.1630890369415283, |
|
"rewards/margins": 0.876747727394104, |
|
"rewards/rejected": -2.039836883544922, |
|
"step": 355 |
|
}, |
|
{ |
|
"epoch": 0.7537293902119864, |
|
"grad_norm": 13.09750518578683, |
|
"learning_rate": 8.628481651367875e-08, |
|
"logits/chosen": -0.880013108253479, |
|
"logits/rejected": -0.9332892298698425, |
|
"logps/chosen": -1.1081321239471436, |
|
"logps/rejected": -1.8383991718292236, |
|
"loss": -0.1127, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -1.1081321239471436, |
|
"rewards/margins": 0.7302672266960144, |
|
"rewards/rejected": -1.8383991718292236, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.7641978539649307, |
|
"grad_norm": 9.380534898584633, |
|
"learning_rate": 7.947809564230445e-08, |
|
"logits/chosen": -0.9200069308280945, |
|
"logits/rejected": -0.9297819137573242, |
|
"logps/chosen": -1.3429687023162842, |
|
"logps/rejected": -2.1255431175231934, |
|
"loss": -0.1104, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -1.3429687023162842, |
|
"rewards/margins": 0.7825741171836853, |
|
"rewards/rejected": -2.1255431175231934, |
|
"step": 365 |
|
}, |
|
{ |
|
"epoch": 0.7746663177178749, |
|
"grad_norm": 8.565425388220058, |
|
"learning_rate": 7.289996455765748e-08, |
|
"logits/chosen": -0.9139026403427124, |
|
"logits/rejected": -0.935566246509552, |
|
"logps/chosen": -1.2977614402770996, |
|
"logps/rejected": -2.170118808746338, |
|
"loss": -0.1132, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.2977614402770996, |
|
"rewards/margins": 0.8723572492599487, |
|
"rewards/rejected": -2.170118808746338, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.7851347814708192, |
|
"grad_norm": 8.601153230537866, |
|
"learning_rate": 6.655924144404906e-08, |
|
"logits/chosen": -0.9767621755599976, |
|
"logits/rejected": -0.948783278465271, |
|
"logps/chosen": -1.324704885482788, |
|
"logps/rejected": -1.9237314462661743, |
|
"loss": -0.1008, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -1.324704885482788, |
|
"rewards/margins": 0.5990264415740967, |
|
"rewards/rejected": -1.9237314462661743, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.7956032452237635, |
|
"grad_norm": 6.863136889215709, |
|
"learning_rate": 6.046442623320145e-08, |
|
"logits/chosen": -0.9905835390090942, |
|
"logits/rejected": -0.9240586161613464, |
|
"logps/chosen": -1.3711433410644531, |
|
"logps/rejected": -1.975137710571289, |
|
"loss": -0.1128, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -1.3711433410644531, |
|
"rewards/margins": 0.6039943695068359, |
|
"rewards/rejected": -1.975137710571289, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.8060717089767077, |
|
"grad_norm": 10.366780250831454, |
|
"learning_rate": 5.4623689209832484e-08, |
|
"logits/chosen": -0.9187868237495422, |
|
"logits/rejected": -0.9430438876152039, |
|
"logps/chosen": -1.181815505027771, |
|
"logps/rejected": -1.8992712497711182, |
|
"loss": -0.1144, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -1.181815505027771, |
|
"rewards/margins": 0.7174557447433472, |
|
"rewards/rejected": -1.8992712497711182, |
|
"step": 385 |
|
}, |
|
{ |
|
"epoch": 0.816540172729652, |
|
"grad_norm": 7.927037671256472, |
|
"learning_rate": 4.904486005914027e-08, |
|
"logits/chosen": -0.907934308052063, |
|
"logits/rejected": -0.995517909526825, |
|
"logps/chosen": -1.231623888015747, |
|
"logps/rejected": -2.0885114669799805, |
|
"loss": -0.1057, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.231623888015747, |
|
"rewards/margins": 0.8568875193595886, |
|
"rewards/rejected": -2.0885114669799805, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.8270086364825961, |
|
"grad_norm": 9.128577656956017, |
|
"learning_rate": 4.373541737087263e-08, |
|
"logits/chosen": -0.9012504816055298, |
|
"logits/rejected": -0.9742817878723145, |
|
"logps/chosen": -1.2018259763717651, |
|
"logps/rejected": -2.0016086101531982, |
|
"loss": -0.0976, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -1.2018259763717651, |
|
"rewards/margins": 0.7997827529907227, |
|
"rewards/rejected": -2.0016086101531982, |
|
"step": 395 |
|
}, |
|
{ |
|
"epoch": 0.8374771002355405, |
|
"grad_norm": 7.6959080621814175, |
|
"learning_rate": 3.8702478614051345e-08, |
|
"logits/chosen": -0.8954795598983765, |
|
"logits/rejected": -0.9198480844497681, |
|
"logps/chosen": -1.1422559022903442, |
|
"logps/rejected": -1.9656658172607422, |
|
"loss": -0.1147, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.1422559022903442, |
|
"rewards/margins": 0.823409914970398, |
|
"rewards/rejected": -1.9656658172607422, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.8479455639884846, |
|
"grad_norm": 37.567595483922176, |
|
"learning_rate": 3.3952790595787986e-08, |
|
"logits/chosen": -0.9429194331169128, |
|
"logits/rejected": -0.9478951692581177, |
|
"logps/chosen": -1.3765652179718018, |
|
"logps/rejected": -2.215216636657715, |
|
"loss": -0.0958, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -1.3765652179718018, |
|
"rewards/margins": 0.8386514782905579, |
|
"rewards/rejected": -2.215216636657715, |
|
"step": 405 |
|
}, |
|
{ |
|
"epoch": 0.8584140277414289, |
|
"grad_norm": 7.455321380026374, |
|
"learning_rate": 2.9492720416985e-08, |
|
"logits/chosen": -0.9236493110656738, |
|
"logits/rejected": -0.9541531801223755, |
|
"logps/chosen": -1.2001044750213623, |
|
"logps/rejected": -1.9708802700042725, |
|
"loss": -0.1062, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.2001044750213623, |
|
"rewards/margins": 0.7707759737968445, |
|
"rewards/rejected": -1.9708802700042725, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.8688824914943732, |
|
"grad_norm": 6.175605994677344, |
|
"learning_rate": 2.5328246937043525e-08, |
|
"logits/chosen": -0.9655213356018066, |
|
"logits/rejected": -1.0396745204925537, |
|
"logps/chosen": -1.2312039136886597, |
|
"logps/rejected": -2.045647382736206, |
|
"loss": -0.113, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.2312039136886597, |
|
"rewards/margins": 0.8144432306289673, |
|
"rewards/rejected": -2.045647382736206, |
|
"step": 415 |
|
}, |
|
{ |
|
"epoch": 0.8793509552473174, |
|
"grad_norm": 10.76836628721284, |
|
"learning_rate": 2.1464952759020856e-08, |
|
"logits/chosen": -0.9188562631607056, |
|
"logits/rejected": -0.925468921661377, |
|
"logps/chosen": -1.2931550741195679, |
|
"logps/rejected": -2.1757659912109375, |
|
"loss": -0.1001, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.2931550741195679, |
|
"rewards/margins": 0.8826107978820801, |
|
"rewards/rejected": -2.1757659912109375, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.8898194190002617, |
|
"grad_norm": 18.620013597425544, |
|
"learning_rate": 1.7908016745981856e-08, |
|
"logits/chosen": -0.9154056310653687, |
|
"logits/rejected": -0.9359448552131653, |
|
"logps/chosen": -1.3699318170547485, |
|
"logps/rejected": -2.101792097091675, |
|
"loss": -0.1114, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -1.3699318170547485, |
|
"rewards/margins": 0.7318603992462158, |
|
"rewards/rejected": -2.101792097091675, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.9002878827532059, |
|
"grad_norm": 9.476057330917156, |
|
"learning_rate": 1.4662207078575684e-08, |
|
"logits/chosen": -0.916486918926239, |
|
"logits/rejected": -0.9570449590682983, |
|
"logps/chosen": -1.3201847076416016, |
|
"logps/rejected": -2.078465223312378, |
|
"loss": -0.107, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.3201847076416016, |
|
"rewards/margins": 0.7582806348800659, |
|
"rewards/rejected": -2.078465223312378, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.9107563465061502, |
|
"grad_norm": 24.560875979085978, |
|
"learning_rate": 1.1731874863145142e-08, |
|
"logits/chosen": -0.9874083399772644, |
|
"logits/rejected": -1.0102823972702026, |
|
"logps/chosen": -1.3655614852905273, |
|
"logps/rejected": -2.2522528171539307, |
|
"loss": -0.1182, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.3655614852905273, |
|
"rewards/margins": 0.8866912126541138, |
|
"rewards/rejected": -2.2522528171539307, |
|
"step": 435 |
|
}, |
|
{ |
|
"epoch": 0.9212248102590945, |
|
"grad_norm": 8.256652917309696, |
|
"learning_rate": 9.12094829893642e-09, |
|
"logits/chosen": -0.9290092587471008, |
|
"logits/rejected": -0.9523330926895142, |
|
"logps/chosen": -1.283372163772583, |
|
"logps/rejected": -1.9379644393920898, |
|
"loss": -0.1093, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -1.283372163772583, |
|
"rewards/margins": 0.6545922160148621, |
|
"rewards/rejected": -1.9379644393920898, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.9316932740120387, |
|
"grad_norm": 8.586204837874826, |
|
"learning_rate": 6.832927412229017e-09, |
|
"logits/chosen": -0.9241235852241516, |
|
"logits/rejected": -0.9258465766906738, |
|
"logps/chosen": -1.1934940814971924, |
|
"logps/rejected": -1.9461805820465088, |
|
"loss": -0.0992, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -1.1934940814971924, |
|
"rewards/margins": 0.7526863813400269, |
|
"rewards/rejected": -1.9461805820465088, |
|
"step": 445 |
|
}, |
|
{ |
|
"epoch": 0.942161737764983, |
|
"grad_norm": 8.342614252776851, |
|
"learning_rate": 4.8708793644441086e-09, |
|
"logits/chosen": -0.9078584909439087, |
|
"logits/rejected": -0.8857234716415405, |
|
"logps/chosen": -1.2411367893218994, |
|
"logps/rejected": -1.9459234476089478, |
|
"loss": -0.109, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -1.2411367893218994, |
|
"rewards/margins": 0.7047866582870483, |
|
"rewards/rejected": -1.9459234476089478, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.9526302015179272, |
|
"grad_norm": 24.648524026578897, |
|
"learning_rate": 3.2374343405217884e-09, |
|
"logits/chosen": -0.8968909382820129, |
|
"logits/rejected": -0.9244266748428345, |
|
"logps/chosen": -1.4007362127304077, |
|
"logps/rejected": -1.9055297374725342, |
|
"loss": -0.1035, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -1.4007362127304077, |
|
"rewards/margins": 0.504793643951416, |
|
"rewards/rejected": -1.9055297374725342, |
|
"step": 455 |
|
}, |
|
{ |
|
"epoch": 0.9630986652708715, |
|
"grad_norm": 21.758607746420935, |
|
"learning_rate": 1.9347820230782295e-09, |
|
"logits/chosen": -0.9594353437423706, |
|
"logits/rejected": -0.9764599800109863, |
|
"logps/chosen": -1.2421373128890991, |
|
"logps/rejected": -2.0354504585266113, |
|
"loss": -0.1121, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -1.2421373128890991, |
|
"rewards/margins": 0.7933129668235779, |
|
"rewards/rejected": -2.0354504585266113, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.9735671290238157, |
|
"grad_norm": 12.46171751515778, |
|
"learning_rate": 9.64668657069706e-10, |
|
"logits/chosen": -0.9806089401245117, |
|
"logits/rejected": -1.0108145475387573, |
|
"logps/chosen": -1.3329870700836182, |
|
"logps/rejected": -2.020784854888916, |
|
"loss": -0.1072, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -1.3329870700836182, |
|
"rewards/margins": 0.6877978444099426, |
|
"rewards/rejected": -2.020784854888916, |
|
"step": 465 |
|
}, |
|
{ |
|
"epoch": 0.98403559277676, |
|
"grad_norm": 11.866295749167707, |
|
"learning_rate": 3.2839470889836627e-10, |
|
"logits/chosen": -0.9077326059341431, |
|
"logits/rejected": -0.9319435358047485, |
|
"logps/chosen": -1.2216429710388184, |
|
"logps/rejected": -1.813685655593872, |
|
"loss": -0.096, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -1.2216429710388184, |
|
"rewards/margins": 0.5920424461364746, |
|
"rewards/rejected": -1.813685655593872, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.9945040565297043, |
|
"grad_norm": 20.786464029224504, |
|
"learning_rate": 2.6813123097352287e-11, |
|
"logits/chosen": -0.9626280069351196, |
|
"logits/rejected": -0.9883280992507935, |
|
"logps/chosen": -1.3163354396820068, |
|
"logps/rejected": -1.988018274307251, |
|
"loss": -0.0924, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -1.3163354396820068, |
|
"rewards/margins": 0.6716827154159546, |
|
"rewards/rejected": -1.988018274307251, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.998691442030882, |
|
"step": 477, |
|
"total_flos": 0.0, |
|
"train_loss": -0.08717740175609069, |
|
"train_runtime": 14342.367, |
|
"train_samples_per_second": 4.263, |
|
"train_steps_per_second": 0.033 |
|
} |
|
], |
|
"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 |
|
} |
|
|