|
{ |
|
"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": 11.773019635010494, |
|
"learning_rate": 5.208333333333333e-08, |
|
"logits/chosen": -1.4459120035171509, |
|
"logits/rejected": -1.4473698139190674, |
|
"logps/chosen": -7.983118534088135, |
|
"logps/rejected": -8.156193733215332, |
|
"loss": 8.9797, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.983118534088135, |
|
"rewards/margins": 0.17307573556900024, |
|
"rewards/rejected": -8.156193733215332, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.02093692750588851, |
|
"grad_norm": 9.709040224393025, |
|
"learning_rate": 1.0416666666666667e-07, |
|
"logits/chosen": -1.4474694728851318, |
|
"logits/rejected": -1.438397765159607, |
|
"logps/chosen": -8.046891212463379, |
|
"logps/rejected": -7.96016788482666, |
|
"loss": 8.9815, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -8.046891212463379, |
|
"rewards/margins": -0.08672460168600082, |
|
"rewards/rejected": -7.96016788482666, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.031405391258832765, |
|
"grad_norm": 16.89107530887862, |
|
"learning_rate": 1.5624999999999999e-07, |
|
"logits/chosen": -1.4477674961090088, |
|
"logits/rejected": -1.4442229270935059, |
|
"logps/chosen": -7.852339267730713, |
|
"logps/rejected": -7.866924285888672, |
|
"loss": 8.8904, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.852339267730713, |
|
"rewards/margins": 0.014585095457732677, |
|
"rewards/rejected": -7.866924285888672, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.04187385501177702, |
|
"grad_norm": 14.261360231352997, |
|
"learning_rate": 2.0833333333333333e-07, |
|
"logits/chosen": -1.4316000938415527, |
|
"logits/rejected": -1.435153841972351, |
|
"logps/chosen": -8.180787086486816, |
|
"logps/rejected": -8.204398155212402, |
|
"loss": 9.0475, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.180787086486816, |
|
"rewards/margins": 0.023611638695001602, |
|
"rewards/rejected": -8.204398155212402, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05234231876472128, |
|
"grad_norm": 13.453269431524436, |
|
"learning_rate": 2.604166666666667e-07, |
|
"logits/chosen": -1.479134440422058, |
|
"logits/rejected": -1.4693964719772339, |
|
"logps/chosen": -8.08868408203125, |
|
"logps/rejected": -7.990322113037109, |
|
"loss": 9.1249, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -8.08868408203125, |
|
"rewards/margins": -0.09836126863956451, |
|
"rewards/rejected": -7.990322113037109, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.06281078251766553, |
|
"grad_norm": 11.378313696016495, |
|
"learning_rate": 3.1249999999999997e-07, |
|
"logits/chosen": -1.447307825088501, |
|
"logits/rejected": -1.4355913400650024, |
|
"logps/chosen": -7.789179801940918, |
|
"logps/rejected": -7.710352897644043, |
|
"loss": 9.0197, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -7.789179801940918, |
|
"rewards/margins": -0.0788281261920929, |
|
"rewards/rejected": -7.710352897644043, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.07327924627060979, |
|
"grad_norm": 10.492170948688017, |
|
"learning_rate": 3.645833333333333e-07, |
|
"logits/chosen": -1.4480352401733398, |
|
"logits/rejected": -1.421820044517517, |
|
"logps/chosen": -8.031718254089355, |
|
"logps/rejected": -7.896989345550537, |
|
"loss": 8.9858, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -8.031718254089355, |
|
"rewards/margins": -0.13472941517829895, |
|
"rewards/rejected": -7.896989345550537, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.08374771002355404, |
|
"grad_norm": 9.438874148034836, |
|
"learning_rate": 4.1666666666666667e-07, |
|
"logits/chosen": -1.4374274015426636, |
|
"logits/rejected": -1.4498326778411865, |
|
"logps/chosen": -7.972153663635254, |
|
"logps/rejected": -8.157554626464844, |
|
"loss": 9.013, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -7.972153663635254, |
|
"rewards/margins": 0.185401052236557, |
|
"rewards/rejected": -8.157554626464844, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.0942161737764983, |
|
"grad_norm": 10.665489074479519, |
|
"learning_rate": 4.6874999999999996e-07, |
|
"logits/chosen": -1.4577468633651733, |
|
"logits/rejected": -1.45815110206604, |
|
"logps/chosen": -8.014823913574219, |
|
"logps/rejected": -7.8058881759643555, |
|
"loss": 8.9894, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -8.014823913574219, |
|
"rewards/margins": -0.20893602073192596, |
|
"rewards/rejected": -7.8058881759643555, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.10468463752944256, |
|
"grad_norm": 14.935150052502319, |
|
"learning_rate": 4.999731868769026e-07, |
|
"logits/chosen": -1.4442975521087646, |
|
"logits/rejected": -1.461060881614685, |
|
"logps/chosen": -8.208638191223145, |
|
"logps/rejected": -8.235433578491211, |
|
"loss": 9.0806, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -8.208638191223145, |
|
"rewards/margins": 0.0267957653850317, |
|
"rewards/rejected": -8.235433578491211, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.11515310128238682, |
|
"grad_norm": 10.181057969467895, |
|
"learning_rate": 4.996716052911017e-07, |
|
"logits/chosen": -1.4492734670639038, |
|
"logits/rejected": -1.4457294940948486, |
|
"logps/chosen": -8.069734573364258, |
|
"logps/rejected": -8.081002235412598, |
|
"loss": 9.1066, |
|
"rewards/accuracies": 0.42500001192092896, |
|
"rewards/chosen": -8.069734573364258, |
|
"rewards/margins": 0.011268272995948792, |
|
"rewards/rejected": -8.081002235412598, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.12562156503533106, |
|
"grad_norm": 17.86320335028244, |
|
"learning_rate": 4.990353313429303e-07, |
|
"logits/chosen": -1.4542796611785889, |
|
"logits/rejected": -1.4671558141708374, |
|
"logps/chosen": -8.027522087097168, |
|
"logps/rejected": -8.309490203857422, |
|
"loss": 8.9006, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -8.027522087097168, |
|
"rewards/margins": 0.28196847438812256, |
|
"rewards/rejected": -8.309490203857422, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.1360900287882753, |
|
"grad_norm": 31.178844889844076, |
|
"learning_rate": 4.980652179769217e-07, |
|
"logits/chosen": -1.4044889211654663, |
|
"logits/rejected": -1.4228571653366089, |
|
"logps/chosen": -7.809567451477051, |
|
"logps/rejected": -8.305734634399414, |
|
"loss": 8.9379, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -7.809567451477051, |
|
"rewards/margins": 0.49616700410842896, |
|
"rewards/rejected": -8.305734634399414, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.14655849254121958, |
|
"grad_norm": 35.6438890099215, |
|
"learning_rate": 4.967625656594781e-07, |
|
"logits/chosen": -1.4016475677490234, |
|
"logits/rejected": -1.3958280086517334, |
|
"logps/chosen": -7.9446282386779785, |
|
"logps/rejected": -8.026212692260742, |
|
"loss": 8.9738, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -7.9446282386779785, |
|
"rewards/margins": 0.08158435672521591, |
|
"rewards/rejected": -8.026212692260742, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.15702695629416383, |
|
"grad_norm": 12.429786415191892, |
|
"learning_rate": 4.951291206355559e-07, |
|
"logits/chosen": -1.4062070846557617, |
|
"logits/rejected": -1.412986159324646, |
|
"logps/chosen": -7.849791049957275, |
|
"logps/rejected": -7.881753444671631, |
|
"loss": 8.9235, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -7.849791049957275, |
|
"rewards/margins": 0.031962424516677856, |
|
"rewards/rejected": -7.881753444671631, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.16749542004710807, |
|
"grad_norm": 19.088641406767135, |
|
"learning_rate": 4.93167072587771e-07, |
|
"logits/chosen": -1.4117752313613892, |
|
"logits/rejected": -1.4267140626907349, |
|
"logps/chosen": -7.871654510498047, |
|
"logps/rejected": -7.936394691467285, |
|
"loss": 8.9114, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -7.871654510498047, |
|
"rewards/margins": 0.06473957002162933, |
|
"rewards/rejected": -7.936394691467285, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.17796388380005235, |
|
"grad_norm": 12.122887672693283, |
|
"learning_rate": 4.908790517010636e-07, |
|
"logits/chosen": -1.4204871654510498, |
|
"logits/rejected": -1.4174373149871826, |
|
"logps/chosen": -7.965147495269775, |
|
"logps/rejected": -8.144342422485352, |
|
"loss": 9.0544, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -7.965147495269775, |
|
"rewards/margins": 0.17919424176216125, |
|
"rewards/rejected": -8.144342422485352, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 0.1884323475529966, |
|
"grad_norm": 13.676489394050206, |
|
"learning_rate": 4.882681251368548e-07, |
|
"logits/chosen": -1.394677996635437, |
|
"logits/rejected": -1.4083014726638794, |
|
"logps/chosen": -7.955393314361572, |
|
"logps/rejected": -8.132440567016602, |
|
"loss": 9.0209, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -7.955393314361572, |
|
"rewards/margins": 0.1770474910736084, |
|
"rewards/rejected": -8.132440567016602, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.19890081130594087, |
|
"grad_norm": 11.55232614681137, |
|
"learning_rate": 4.853377929214243e-07, |
|
"logits/chosen": -1.4072471857070923, |
|
"logits/rejected": -1.3971055746078491, |
|
"logps/chosen": -8.058225631713867, |
|
"logps/rejected": -7.9064154624938965, |
|
"loss": 9.0891, |
|
"rewards/accuracies": 0.40625, |
|
"rewards/chosen": -8.058225631713867, |
|
"rewards/margins": -0.15181036293506622, |
|
"rewards/rejected": -7.9064154624938965, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 0.2093692750588851, |
|
"grad_norm": 21.381914106448818, |
|
"learning_rate": 4.820919832540181e-07, |
|
"logits/chosen": -1.4291296005249023, |
|
"logits/rejected": -1.4334924221038818, |
|
"logps/chosen": -8.016979217529297, |
|
"logps/rejected": -8.049524307250977, |
|
"loss": 9.0182, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -8.016979217529297, |
|
"rewards/margins": 0.032544612884521484, |
|
"rewards/rejected": -8.049524307250977, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.21983773881182936, |
|
"grad_norm": 12.15933543492837, |
|
"learning_rate": 4.785350472409791e-07, |
|
"logits/chosen": -1.4222753047943115, |
|
"logits/rejected": -1.4167351722717285, |
|
"logps/chosen": -7.735973358154297, |
|
"logps/rejected": -7.934323310852051, |
|
"loss": 8.9474, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -7.735973358154297, |
|
"rewards/margins": 0.19834883511066437, |
|
"rewards/rejected": -7.934323310852051, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 0.23030620256477363, |
|
"grad_norm": 14.275715375180301, |
|
"learning_rate": 4.7467175306295647e-07, |
|
"logits/chosen": -1.4484798908233643, |
|
"logits/rejected": -1.4482091665267944, |
|
"logps/chosen": -7.996209144592285, |
|
"logps/rejected": -8.223912239074707, |
|
"loss": 9.021, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -7.996209144592285, |
|
"rewards/margins": 0.22770316898822784, |
|
"rewards/rejected": -8.223912239074707, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.24077466631771788, |
|
"grad_norm": 264.49732707332146, |
|
"learning_rate": 4.70507279583015e-07, |
|
"logits/chosen": -1.3956806659698486, |
|
"logits/rejected": -1.399662971496582, |
|
"logps/chosen": -7.871851921081543, |
|
"logps/rejected": -8.169795989990234, |
|
"loss": 8.9644, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -7.871851921081543, |
|
"rewards/margins": 0.2979447841644287, |
|
"rewards/rejected": -8.169795989990234, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 0.2512431300706621, |
|
"grad_norm": 13.052538835040968, |
|
"learning_rate": 4.6604720940421207e-07, |
|
"logits/chosen": -1.3686974048614502, |
|
"logits/rejected": -1.3719284534454346, |
|
"logps/chosen": -8.102027893066406, |
|
"logps/rejected": -8.149332046508789, |
|
"loss": 8.9524, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.102027893066406, |
|
"rewards/margins": 0.04730508476495743, |
|
"rewards/rejected": -8.149332046508789, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.26171159382360637, |
|
"grad_norm": 13.986017227821636, |
|
"learning_rate": 4.612975213859487e-07, |
|
"logits/chosen": -1.4253513813018799, |
|
"logits/rejected": -1.419106125831604, |
|
"logps/chosen": -8.198808670043945, |
|
"logps/rejected": -8.076632499694824, |
|
"loss": 9.068, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -8.198808670043945, |
|
"rewards/margins": -0.12217602878808975, |
|
"rewards/rejected": -8.076632499694824, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.2721800575765506, |
|
"grad_norm": 16.859825952779723, |
|
"learning_rate": 4.5626458262912735e-07, |
|
"logits/chosen": -1.393610954284668, |
|
"logits/rejected": -1.4041980504989624, |
|
"logps/chosen": -8.033707618713379, |
|
"logps/rejected": -8.149084091186523, |
|
"loss": 8.9812, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -8.033707618713379, |
|
"rewards/margins": 0.11537569761276245, |
|
"rewards/rejected": -8.149084091186523, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.2826485213294949, |
|
"grad_norm": 17.35250325551846, |
|
"learning_rate": 4.5095513994085974e-07, |
|
"logits/chosen": -1.3907815217971802, |
|
"logits/rejected": -1.4047492742538452, |
|
"logps/chosen": -7.996786594390869, |
|
"logps/rejected": -8.4099760055542, |
|
"loss": 8.9754, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -7.996786594390869, |
|
"rewards/margins": 0.41318947076797485, |
|
"rewards/rejected": -8.4099760055542, |
|
"step": 135 |
|
}, |
|
{ |
|
"epoch": 0.29311698508243916, |
|
"grad_norm": 13.984769587358256, |
|
"learning_rate": 4.453763107901675e-07, |
|
"logits/chosen": -1.415302038192749, |
|
"logits/rejected": -1.4232456684112549, |
|
"logps/chosen": -7.9026665687561035, |
|
"logps/rejected": -7.970452785491943, |
|
"loss": 8.9574, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -7.9026665687561035, |
|
"rewards/margins": 0.06778701394796371, |
|
"rewards/rejected": -7.970452785491943, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.3035854488353834, |
|
"grad_norm": 10.033908137947623, |
|
"learning_rate": 4.395355737667985e-07, |
|
"logits/chosen": -1.4233510494232178, |
|
"logits/rejected": -1.4498116970062256, |
|
"logps/chosen": -7.896154880523682, |
|
"logps/rejected": -8.049067497253418, |
|
"loss": 8.9238, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -7.896154880523682, |
|
"rewards/margins": 0.15291360020637512, |
|
"rewards/rejected": -8.049067497253418, |
|
"step": 145 |
|
}, |
|
{ |
|
"epoch": 0.31405391258832765, |
|
"grad_norm": 12.948300110975106, |
|
"learning_rate": 4.3344075855595097e-07, |
|
"logits/chosen": -1.4413058757781982, |
|
"logits/rejected": -1.4447827339172363, |
|
"logps/chosen": -7.901156425476074, |
|
"logps/rejected": -8.029939651489258, |
|
"loss": 8.9805, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.901156425476074, |
|
"rewards/margins": 0.12878116965293884, |
|
"rewards/rejected": -8.029939651489258, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.3245223763412719, |
|
"grad_norm": 13.439201394209311, |
|
"learning_rate": 4.271000354423425e-07, |
|
"logits/chosen": -1.412276029586792, |
|
"logits/rejected": -1.389304280281067, |
|
"logps/chosen": -7.978514194488525, |
|
"logps/rejected": -7.953900337219238, |
|
"loss": 9.0293, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -7.978514194488525, |
|
"rewards/margins": -0.024613410234451294, |
|
"rewards/rejected": -7.953900337219238, |
|
"step": 155 |
|
}, |
|
{ |
|
"epoch": 0.33499084009421615, |
|
"grad_norm": 14.950263562725501, |
|
"learning_rate": 4.2052190435769554e-07, |
|
"logits/chosen": -1.3908809423446655, |
|
"logits/rejected": -1.4011869430541992, |
|
"logps/chosen": -8.05128288269043, |
|
"logps/rejected": -8.073431015014648, |
|
"loss": 9.0342, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -8.05128288269043, |
|
"rewards/margins": 0.022147323936223984, |
|
"rewards/rejected": -8.073431015014648, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.34545930384716045, |
|
"grad_norm": 11.065140622134825, |
|
"learning_rate": 4.137151834863213e-07, |
|
"logits/chosen": -1.4063256978988647, |
|
"logits/rejected": -1.4003641605377197, |
|
"logps/chosen": -7.999358177185059, |
|
"logps/rejected": -8.034425735473633, |
|
"loss": 9.0115, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -7.999358177185059, |
|
"rewards/margins": 0.03506825119256973, |
|
"rewards/rejected": -8.034425735473633, |
|
"step": 165 |
|
}, |
|
{ |
|
"epoch": 0.3559277676001047, |
|
"grad_norm": 12.669233243410716, |
|
"learning_rate": 4.0668899744407567e-07, |
|
"logits/chosen": -1.3777008056640625, |
|
"logits/rejected": -1.3843880891799927, |
|
"logps/chosen": -8.11161994934082, |
|
"logps/rejected": -8.079643249511719, |
|
"loss": 9.088, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -8.11161994934082, |
|
"rewards/margins": -0.03197728842496872, |
|
"rewards/rejected": -8.079643249511719, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.36639623135304894, |
|
"grad_norm": 11.420602500449908, |
|
"learning_rate": 3.994527650465352e-07, |
|
"logits/chosen": -1.4002071619033813, |
|
"logits/rejected": -1.4017866849899292, |
|
"logps/chosen": -7.9545183181762695, |
|
"logps/rejected": -7.929112434387207, |
|
"loss": 9.01, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -7.9545183181762695, |
|
"rewards/margins": -0.025406653061509132, |
|
"rewards/rejected": -7.929112434387207, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 0.3768646951059932, |
|
"grad_norm": 15.567371494550915, |
|
"learning_rate": 3.920161866827889e-07, |
|
"logits/chosen": -1.379305124282837, |
|
"logits/rejected": -1.4001214504241943, |
|
"logps/chosen": -8.02379322052002, |
|
"logps/rejected": -8.138644218444824, |
|
"loss": 8.993, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -8.02379322052002, |
|
"rewards/margins": 0.11485125869512558, |
|
"rewards/rejected": -8.138644218444824, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.38733315885893743, |
|
"grad_norm": 13.26955840349748, |
|
"learning_rate": 3.8438923131177237e-07, |
|
"logits/chosen": -1.345184564590454, |
|
"logits/rejected": -1.3495187759399414, |
|
"logps/chosen": -7.9985671043396, |
|
"logps/rejected": -8.032160758972168, |
|
"loss": 8.9564, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -7.9985671043396, |
|
"rewards/margins": 0.03359389677643776, |
|
"rewards/rejected": -8.032160758972168, |
|
"step": 185 |
|
}, |
|
{ |
|
"epoch": 0.39780162261188173, |
|
"grad_norm": 12.477711396229129, |
|
"learning_rate": 3.765821230985757e-07, |
|
"logits/chosen": -1.3648011684417725, |
|
"logits/rejected": -1.3609410524368286, |
|
"logps/chosen": -7.813986301422119, |
|
"logps/rejected": -7.937358856201172, |
|
"loss": 8.9559, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -7.813986301422119, |
|
"rewards/margins": 0.12337237596511841, |
|
"rewards/rejected": -7.937358856201172, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.408270086364826, |
|
"grad_norm": 14.974853102646863, |
|
"learning_rate": 3.6860532770864005e-07, |
|
"logits/chosen": -1.340057611465454, |
|
"logits/rejected": -1.3464308977127075, |
|
"logps/chosen": -8.168105125427246, |
|
"logps/rejected": -8.047191619873047, |
|
"loss": 9.0593, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -8.168105125427246, |
|
"rewards/margins": -0.12091370671987534, |
|
"rewards/rejected": -8.047191619873047, |
|
"step": 195 |
|
}, |
|
{ |
|
"epoch": 0.4187385501177702, |
|
"grad_norm": 12.019056641062388, |
|
"learning_rate": 3.604695382782159e-07, |
|
"logits/chosen": -1.3949248790740967, |
|
"logits/rejected": -1.3864567279815674, |
|
"logps/chosen": -7.904551029205322, |
|
"logps/rejected": -8.010711669921875, |
|
"loss": 9.0251, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -7.904551029205322, |
|
"rewards/margins": 0.10616078227758408, |
|
"rewards/rejected": -8.010711669921875, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.42920701387071447, |
|
"grad_norm": 12.159976530620005, |
|
"learning_rate": 3.5218566107988867e-07, |
|
"logits/chosen": -1.3519536256790161, |
|
"logits/rejected": -1.3532142639160156, |
|
"logps/chosen": -7.96322774887085, |
|
"logps/rejected": -8.124198913574219, |
|
"loss": 8.9502, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.96322774887085, |
|
"rewards/margins": 0.16097018122673035, |
|
"rewards/rejected": -8.124198913574219, |
|
"step": 205 |
|
}, |
|
{ |
|
"epoch": 0.4396754776236587, |
|
"grad_norm": 27.53866432550454, |
|
"learning_rate": 3.4376480090239047e-07, |
|
"logits/chosen": -1.3502663373947144, |
|
"logits/rejected": -1.3595716953277588, |
|
"logps/chosen": -7.90906286239624, |
|
"logps/rejected": -8.031661033630371, |
|
"loss": 8.997, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -7.90906286239624, |
|
"rewards/margins": 0.12259860336780548, |
|
"rewards/rejected": -8.031661033630371, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.45014394137660296, |
|
"grad_norm": 10.57208181223796, |
|
"learning_rate": 3.3521824616429284e-07, |
|
"logits/chosen": -1.3545321226119995, |
|
"logits/rejected": -1.363968849182129, |
|
"logps/chosen": -7.766638278961182, |
|
"logps/rejected": -8.186101913452148, |
|
"loss": 8.9091, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -7.766638278961182, |
|
"rewards/margins": 0.4194633960723877, |
|
"rewards/rejected": -8.186101913452148, |
|
"step": 215 |
|
}, |
|
{ |
|
"epoch": 0.46061240512954726, |
|
"grad_norm": 12.898063816305818, |
|
"learning_rate": 3.265574537815398e-07, |
|
"logits/chosen": -1.3691070079803467, |
|
"logits/rejected": -1.3948938846588135, |
|
"logps/chosen": -7.7049560546875, |
|
"logps/rejected": -7.837745666503906, |
|
"loss": 9.0096, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -7.7049560546875, |
|
"rewards/margins": 0.13278940320014954, |
|
"rewards/rejected": -7.837745666503906, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.4710808688824915, |
|
"grad_norm": 11.02542077899864, |
|
"learning_rate": 3.1779403380910425e-07, |
|
"logits/chosen": -1.3727288246154785, |
|
"logits/rejected": -1.3770244121551514, |
|
"logps/chosen": -7.688443660736084, |
|
"logps/rejected": -7.930489540100098, |
|
"loss": 8.9147, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -7.688443660736084, |
|
"rewards/margins": 0.24204584956169128, |
|
"rewards/rejected": -7.930489540100098, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 0.48154933263543576, |
|
"grad_norm": 12.642936387401933, |
|
"learning_rate": 3.0893973387735683e-07, |
|
"logits/chosen": -1.3643910884857178, |
|
"logits/rejected": -1.3646974563598633, |
|
"logps/chosen": -8.087334632873535, |
|
"logps/rejected": -8.227707862854004, |
|
"loss": 9.0751, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.087334632873535, |
|
"rewards/margins": 0.1403738558292389, |
|
"rewards/rejected": -8.227707862854004, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.49201779638838, |
|
"grad_norm": 11.653678081935743, |
|
"learning_rate": 3.000064234440111e-07, |
|
"logits/chosen": -1.3848552703857422, |
|
"logits/rejected": -1.3937251567840576, |
|
"logps/chosen": -7.939251899719238, |
|
"logps/rejected": -8.165653228759766, |
|
"loss": 8.9934, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -7.939251899719238, |
|
"rewards/margins": 0.2264014482498169, |
|
"rewards/rejected": -8.165653228759766, |
|
"step": 235 |
|
}, |
|
{ |
|
"epoch": 0.5024862601413242, |
|
"grad_norm": 20.78939179590344, |
|
"learning_rate": 2.910060778827554e-07, |
|
"logits/chosen": -1.3638317584991455, |
|
"logits/rejected": -1.372854471206665, |
|
"logps/chosen": -7.975274562835693, |
|
"logps/rejected": -8.114542007446289, |
|
"loss": 9.0759, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -7.975274562835693, |
|
"rewards/margins": 0.13926678895950317, |
|
"rewards/rejected": -8.114542007446289, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.5129547238942685, |
|
"grad_norm": 15.247620089410335, |
|
"learning_rate": 2.8195076242990116e-07, |
|
"logits/chosen": -1.3632863759994507, |
|
"logits/rejected": -1.368691086769104, |
|
"logps/chosen": -8.058801651000977, |
|
"logps/rejected": -8.10949420928955, |
|
"loss": 8.9298, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -8.058801651000977, |
|
"rewards/margins": 0.05069229006767273, |
|
"rewards/rejected": -8.10949420928955, |
|
"step": 245 |
|
}, |
|
{ |
|
"epoch": 0.5234231876472127, |
|
"grad_norm": 16.020124654557804, |
|
"learning_rate": 2.7285261601056697e-07, |
|
"logits/chosen": -1.335317611694336, |
|
"logits/rejected": -1.3454726934432983, |
|
"logps/chosen": -7.886631011962891, |
|
"logps/rejected": -8.195113182067871, |
|
"loss": 8.8247, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -7.886631011962891, |
|
"rewards/margins": 0.30848345160484314, |
|
"rewards/rejected": -8.195113182067871, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.533891651400157, |
|
"grad_norm": 13.208964203269911, |
|
"learning_rate": 2.6372383496608186e-07, |
|
"logits/chosen": -1.3199383020401, |
|
"logits/rejected": -1.3232190608978271, |
|
"logps/chosen": -8.123445510864258, |
|
"logps/rejected": -8.300798416137695, |
|
"loss": 8.9974, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -8.123445510864258, |
|
"rewards/margins": 0.17735353112220764, |
|
"rewards/rejected": -8.300798416137695, |
|
"step": 255 |
|
}, |
|
{ |
|
"epoch": 0.5443601151531012, |
|
"grad_norm": 12.80879122998489, |
|
"learning_rate": 2.5457665670441937e-07, |
|
"logits/chosen": -1.2541121244430542, |
|
"logits/rejected": -1.2609522342681885, |
|
"logps/chosen": -7.773028373718262, |
|
"logps/rejected": -8.211790084838867, |
|
"loss": 8.9532, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -7.773028373718262, |
|
"rewards/margins": 0.43876129388809204, |
|
"rewards/rejected": -8.211790084838867, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.5548285789060455, |
|
"grad_norm": 11.386114149183353, |
|
"learning_rate": 2.454233432955807e-07, |
|
"logits/chosen": -1.2998675107955933, |
|
"logits/rejected": -1.3152759075164795, |
|
"logps/chosen": -7.926003932952881, |
|
"logps/rejected": -8.16883373260498, |
|
"loss": 8.972, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -7.926003932952881, |
|
"rewards/margins": 0.242829367518425, |
|
"rewards/rejected": -8.16883373260498, |
|
"step": 265 |
|
}, |
|
{ |
|
"epoch": 0.5652970426589898, |
|
"grad_norm": 12.189434902359803, |
|
"learning_rate": 2.3627616503391812e-07, |
|
"logits/chosen": -1.311598539352417, |
|
"logits/rejected": -1.2913634777069092, |
|
"logps/chosen": -8.263054847717285, |
|
"logps/rejected": -8.317811965942383, |
|
"loss": 9.0177, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -8.263054847717285, |
|
"rewards/margins": 0.05475734546780586, |
|
"rewards/rejected": -8.317811965942383, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.575765506411934, |
|
"grad_norm": 26.512104480113177, |
|
"learning_rate": 2.2714738398943308e-07, |
|
"logits/chosen": -1.3503350019454956, |
|
"logits/rejected": -1.3630883693695068, |
|
"logps/chosen": -7.8860344886779785, |
|
"logps/rejected": -8.053815841674805, |
|
"loss": 8.9101, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -7.8860344886779785, |
|
"rewards/margins": 0.1677820235490799, |
|
"rewards/rejected": -8.053815841674805, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 0.5862339701648783, |
|
"grad_norm": 13.657892445365238, |
|
"learning_rate": 2.1804923757009882e-07, |
|
"logits/chosen": -1.3591408729553223, |
|
"logits/rejected": -1.3795652389526367, |
|
"logps/chosen": -7.8742218017578125, |
|
"logps/rejected": -8.096671104431152, |
|
"loss": 8.926, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -7.8742218017578125, |
|
"rewards/margins": 0.222448468208313, |
|
"rewards/rejected": -8.096671104431152, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.5967024339178225, |
|
"grad_norm": 13.125973843054254, |
|
"learning_rate": 2.089939221172446e-07, |
|
"logits/chosen": -1.3040008544921875, |
|
"logits/rejected": -1.3099054098129272, |
|
"logps/chosen": -7.843650817871094, |
|
"logps/rejected": -8.080728530883789, |
|
"loss": 8.8759, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -7.843650817871094, |
|
"rewards/margins": 0.23707744479179382, |
|
"rewards/rejected": -8.080728530883789, |
|
"step": 285 |
|
}, |
|
{ |
|
"epoch": 0.6071708976707668, |
|
"grad_norm": 12.761067879239661, |
|
"learning_rate": 1.9999357655598891e-07, |
|
"logits/chosen": -1.3441493511199951, |
|
"logits/rejected": -1.3643635511398315, |
|
"logps/chosen": -7.810141086578369, |
|
"logps/rejected": -8.124038696289062, |
|
"loss": 8.9111, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -7.810141086578369, |
|
"rewards/margins": 0.31389790773391724, |
|
"rewards/rejected": -8.124038696289062, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.6176393614237111, |
|
"grad_norm": 12.064219944453757, |
|
"learning_rate": 1.9106026612264315e-07, |
|
"logits/chosen": -1.3431895971298218, |
|
"logits/rejected": -1.3602094650268555, |
|
"logps/chosen": -7.9335832595825195, |
|
"logps/rejected": -8.049263000488281, |
|
"loss": 8.9788, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -7.9335832595825195, |
|
"rewards/margins": 0.11567918956279755, |
|
"rewards/rejected": -8.049263000488281, |
|
"step": 295 |
|
}, |
|
{ |
|
"epoch": 0.6281078251766553, |
|
"grad_norm": 12.249297326029936, |
|
"learning_rate": 1.8220596619089573e-07, |
|
"logits/chosen": -1.3094807863235474, |
|
"logits/rejected": -1.3196088075637817, |
|
"logps/chosen": -7.965109825134277, |
|
"logps/rejected": -8.200112342834473, |
|
"loss": 8.9053, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.965109825134277, |
|
"rewards/margins": 0.23500177264213562, |
|
"rewards/rejected": -8.200112342834473, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.6385762889295996, |
|
"grad_norm": 11.932717244582381, |
|
"learning_rate": 1.7344254621846017e-07, |
|
"logits/chosen": -1.317882776260376, |
|
"logits/rejected": -1.3457939624786377, |
|
"logps/chosen": -8.174354553222656, |
|
"logps/rejected": -8.366888046264648, |
|
"loss": 9.0377, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -8.174354553222656, |
|
"rewards/margins": 0.19253170490264893, |
|
"rewards/rejected": -8.366888046264648, |
|
"step": 305 |
|
}, |
|
{ |
|
"epoch": 0.6490447526825438, |
|
"grad_norm": 14.13608210525545, |
|
"learning_rate": 1.647817538357072e-07, |
|
"logits/chosen": -1.3378267288208008, |
|
"logits/rejected": -1.3463207483291626, |
|
"logps/chosen": -8.089614868164062, |
|
"logps/rejected": -8.071249008178711, |
|
"loss": 9.0688, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.089614868164062, |
|
"rewards/margins": -0.018364891409873962, |
|
"rewards/rejected": -8.071249008178711, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.6595132164354881, |
|
"grad_norm": 32.011426029911085, |
|
"learning_rate": 1.562351990976095e-07, |
|
"logits/chosen": -1.330702304840088, |
|
"logits/rejected": -1.3502042293548584, |
|
"logps/chosen": -7.904356479644775, |
|
"logps/rejected": -8.121664047241211, |
|
"loss": 8.8938, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -7.904356479644775, |
|
"rewards/margins": 0.2173081338405609, |
|
"rewards/rejected": -8.121664047241211, |
|
"step": 315 |
|
}, |
|
{ |
|
"epoch": 0.6699816801884323, |
|
"grad_norm": 12.807799082013823, |
|
"learning_rate": 1.478143389201113e-07, |
|
"logits/chosen": -1.346669316291809, |
|
"logits/rejected": -1.3398444652557373, |
|
"logps/chosen": -8.155746459960938, |
|
"logps/rejected": -8.042104721069336, |
|
"loss": 8.9844, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -8.155746459960938, |
|
"rewards/margins": -0.11364219337701797, |
|
"rewards/rejected": -8.042104721069336, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.6804501439413766, |
|
"grad_norm": 14.721102641186317, |
|
"learning_rate": 1.3953046172178413e-07, |
|
"logits/chosen": -1.3383146524429321, |
|
"logits/rejected": -1.3453185558319092, |
|
"logps/chosen": -8.06794548034668, |
|
"logps/rejected": -8.342490196228027, |
|
"loss": 8.9313, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.06794548034668, |
|
"rewards/margins": 0.27454426884651184, |
|
"rewards/rejected": -8.342490196228027, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 0.6909186076943209, |
|
"grad_norm": 12.639814201884843, |
|
"learning_rate": 1.3139467229135998e-07, |
|
"logits/chosen": -1.3189464807510376, |
|
"logits/rejected": -1.3342128992080688, |
|
"logps/chosen": -7.980110168457031, |
|
"logps/rejected": -8.00477123260498, |
|
"loss": 9.0333, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.980110168457031, |
|
"rewards/margins": 0.024661889299750328, |
|
"rewards/rejected": -8.00477123260498, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.7013870714472651, |
|
"grad_norm": 13.542292770349258, |
|
"learning_rate": 1.2341787690142435e-07, |
|
"logits/chosen": -1.3359485864639282, |
|
"logits/rejected": -1.3414697647094727, |
|
"logps/chosen": -7.863481044769287, |
|
"logps/rejected": -8.104010581970215, |
|
"loss": 8.8907, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -7.863481044769287, |
|
"rewards/margins": 0.24052973091602325, |
|
"rewards/rejected": -8.104010581970215, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 0.7118555352002094, |
|
"grad_norm": 13.235795007023567, |
|
"learning_rate": 1.1561076868822755e-07, |
|
"logits/chosen": -1.3393312692642212, |
|
"logits/rejected": -1.3506433963775635, |
|
"logps/chosen": -8.217998504638672, |
|
"logps/rejected": -8.318530082702637, |
|
"loss": 9.0332, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -8.217998504638672, |
|
"rewards/margins": 0.1005309671163559, |
|
"rewards/rejected": -8.318530082702637, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.7223239989531536, |
|
"grad_norm": 14.926426867380192, |
|
"learning_rate": 1.0798381331721107e-07, |
|
"logits/chosen": -1.3743078708648682, |
|
"logits/rejected": -1.390223503112793, |
|
"logps/chosen": -8.069305419921875, |
|
"logps/rejected": -8.406010627746582, |
|
"loss": 9.0149, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -8.069305419921875, |
|
"rewards/margins": 0.3367052972316742, |
|
"rewards/rejected": -8.406010627746582, |
|
"step": 345 |
|
}, |
|
{ |
|
"epoch": 0.7327924627060979, |
|
"grad_norm": 15.925504397841232, |
|
"learning_rate": 1.0054723495346482e-07, |
|
"logits/chosen": -1.344575047492981, |
|
"logits/rejected": -1.3378257751464844, |
|
"logps/chosen": -7.946380615234375, |
|
"logps/rejected": -8.057405471801758, |
|
"loss": 8.9081, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -7.946380615234375, |
|
"rewards/margins": 0.11102403700351715, |
|
"rewards/rejected": -8.057405471801758, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.7432609264590422, |
|
"grad_norm": 14.002913308986452, |
|
"learning_rate": 9.331100255592436e-08, |
|
"logits/chosen": -1.3642247915267944, |
|
"logits/rejected": -1.356299638748169, |
|
"logps/chosen": -8.019064903259277, |
|
"logps/rejected": -7.9631476402282715, |
|
"loss": 8.911, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -8.019064903259277, |
|
"rewards/margins": -0.055917102843523026, |
|
"rewards/rejected": -7.9631476402282715, |
|
"step": 355 |
|
}, |
|
{ |
|
"epoch": 0.7537293902119864, |
|
"grad_norm": 11.189613386087895, |
|
"learning_rate": 8.628481651367875e-08, |
|
"logits/chosen": -1.323051929473877, |
|
"logits/rejected": -1.3336186408996582, |
|
"logps/chosen": -7.749169826507568, |
|
"logps/rejected": -8.018518447875977, |
|
"loss": 8.86, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -7.749169826507568, |
|
"rewards/margins": 0.26934924721717834, |
|
"rewards/rejected": -8.018518447875977, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.7641978539649307, |
|
"grad_norm": 12.73495657042311, |
|
"learning_rate": 7.947809564230445e-08, |
|
"logits/chosen": -1.299702525138855, |
|
"logits/rejected": -1.300682783126831, |
|
"logps/chosen": -7.912903785705566, |
|
"logps/rejected": -7.9577813148498535, |
|
"loss": 8.9548, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -7.912903785705566, |
|
"rewards/margins": 0.044878166168928146, |
|
"rewards/rejected": -7.9577813148498535, |
|
"step": 365 |
|
}, |
|
{ |
|
"epoch": 0.7746663177178749, |
|
"grad_norm": 12.244582800127791, |
|
"learning_rate": 7.289996455765748e-08, |
|
"logits/chosen": -1.3569344282150269, |
|
"logits/rejected": -1.3466898202896118, |
|
"logps/chosen": -8.036967277526855, |
|
"logps/rejected": -8.161175727844238, |
|
"loss": 8.9693, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -8.036967277526855, |
|
"rewards/margins": 0.12420801818370819, |
|
"rewards/rejected": -8.161175727844238, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.7851347814708192, |
|
"grad_norm": 13.857374473897332, |
|
"learning_rate": 6.655924144404906e-08, |
|
"logits/chosen": -1.3418371677398682, |
|
"logits/rejected": -1.3312771320343018, |
|
"logps/chosen": -8.015717506408691, |
|
"logps/rejected": -8.231267929077148, |
|
"loss": 8.9495, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -8.015717506408691, |
|
"rewards/margins": 0.2155502289533615, |
|
"rewards/rejected": -8.231267929077148, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 0.7956032452237635, |
|
"grad_norm": 12.96483071279786, |
|
"learning_rate": 6.046442623320145e-08, |
|
"logits/chosen": -1.3451616764068604, |
|
"logits/rejected": -1.2988903522491455, |
|
"logps/chosen": -8.223435401916504, |
|
"logps/rejected": -8.359044075012207, |
|
"loss": 8.9575, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -8.223435401916504, |
|
"rewards/margins": 0.13560748100280762, |
|
"rewards/rejected": -8.359044075012207, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.8060717089767077, |
|
"grad_norm": 13.685503251086777, |
|
"learning_rate": 5.4623689209832484e-08, |
|
"logits/chosen": -1.3425824642181396, |
|
"logits/rejected": -1.3416216373443604, |
|
"logps/chosen": -8.126443862915039, |
|
"logps/rejected": -8.207201957702637, |
|
"loss": 8.9674, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -8.126443862915039, |
|
"rewards/margins": 0.08075866848230362, |
|
"rewards/rejected": -8.207201957702637, |
|
"step": 385 |
|
}, |
|
{ |
|
"epoch": 0.816540172729652, |
|
"grad_norm": 13.422439706317645, |
|
"learning_rate": 4.904486005914027e-08, |
|
"logits/chosen": -1.3507561683654785, |
|
"logits/rejected": -1.3538233041763306, |
|
"logps/chosen": -7.850442409515381, |
|
"logps/rejected": -8.027676582336426, |
|
"loss": 8.8745, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -7.850442409515381, |
|
"rewards/margins": 0.17723475396633148, |
|
"rewards/rejected": -8.027676582336426, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.8270086364825961, |
|
"grad_norm": 17.927875137352093, |
|
"learning_rate": 4.373541737087263e-08, |
|
"logits/chosen": -1.3154563903808594, |
|
"logits/rejected": -1.3293858766555786, |
|
"logps/chosen": -7.652901649475098, |
|
"logps/rejected": -7.787165641784668, |
|
"loss": 8.9212, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -7.652901649475098, |
|
"rewards/margins": 0.13426539301872253, |
|
"rewards/rejected": -7.787165641784668, |
|
"step": 395 |
|
}, |
|
{ |
|
"epoch": 0.8374771002355405, |
|
"grad_norm": 15.347923517709377, |
|
"learning_rate": 3.8702478614051345e-08, |
|
"logits/chosen": -1.3166577816009521, |
|
"logits/rejected": -1.308420181274414, |
|
"logps/chosen": -7.855776786804199, |
|
"logps/rejected": -7.910757541656494, |
|
"loss": 8.958, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -7.855776786804199, |
|
"rewards/margins": 0.05497948080301285, |
|
"rewards/rejected": -7.910757541656494, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.8479455639884846, |
|
"grad_norm": 12.360735869108282, |
|
"learning_rate": 3.3952790595787986e-08, |
|
"logits/chosen": -1.3554697036743164, |
|
"logits/rejected": -1.361022710800171, |
|
"logps/chosen": -8.025907516479492, |
|
"logps/rejected": -8.157610893249512, |
|
"loss": 8.9282, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -8.025907516479492, |
|
"rewards/margins": 0.13170403242111206, |
|
"rewards/rejected": -8.157610893249512, |
|
"step": 405 |
|
}, |
|
{ |
|
"epoch": 0.8584140277414289, |
|
"grad_norm": 31.497593173498622, |
|
"learning_rate": 2.9492720416985e-08, |
|
"logits/chosen": -1.3601171970367432, |
|
"logits/rejected": -1.3568463325500488, |
|
"logps/chosen": -7.857785701751709, |
|
"logps/rejected": -7.846571922302246, |
|
"loss": 8.9122, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -7.857785701751709, |
|
"rewards/margins": -0.01121367234736681, |
|
"rewards/rejected": -7.846571922302246, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.8688824914943732, |
|
"grad_norm": 14.186881098656178, |
|
"learning_rate": 2.5328246937043525e-08, |
|
"logits/chosen": -1.3695319890975952, |
|
"logits/rejected": -1.3772709369659424, |
|
"logps/chosen": -7.996212959289551, |
|
"logps/rejected": -8.215394973754883, |
|
"loss": 8.984, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -7.996212959289551, |
|
"rewards/margins": 0.21918316185474396, |
|
"rewards/rejected": -8.215394973754883, |
|
"step": 415 |
|
}, |
|
{ |
|
"epoch": 0.8793509552473174, |
|
"grad_norm": 13.07091749141306, |
|
"learning_rate": 2.1464952759020856e-08, |
|
"logits/chosen": -1.320655107498169, |
|
"logits/rejected": -1.315647006034851, |
|
"logps/chosen": -7.8787431716918945, |
|
"logps/rejected": -7.92669677734375, |
|
"loss": 8.9426, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -7.8787431716918945, |
|
"rewards/margins": 0.047954294830560684, |
|
"rewards/rejected": -7.92669677734375, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.8898194190002617, |
|
"grad_norm": 12.735545462610197, |
|
"learning_rate": 1.7908016745981856e-08, |
|
"logits/chosen": -1.3389335870742798, |
|
"logits/rejected": -1.335252285003662, |
|
"logps/chosen": -8.04029655456543, |
|
"logps/rejected": -8.07039737701416, |
|
"loss": 8.9855, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -8.04029655456543, |
|
"rewards/margins": 0.030100971460342407, |
|
"rewards/rejected": -8.07039737701416, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 0.9002878827532059, |
|
"grad_norm": 15.247676128020851, |
|
"learning_rate": 1.4662207078575684e-08, |
|
"logits/chosen": -1.3410335779190063, |
|
"logits/rejected": -1.3450171947479248, |
|
"logps/chosen": -7.889002323150635, |
|
"logps/rejected": -7.991044044494629, |
|
"loss": 8.9385, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -7.889002323150635, |
|
"rewards/margins": 0.10204191505908966, |
|
"rewards/rejected": -7.991044044494629, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.9107563465061502, |
|
"grad_norm": 13.85977210657239, |
|
"learning_rate": 1.1731874863145142e-08, |
|
"logits/chosen": -1.3579423427581787, |
|
"logits/rejected": -1.3657190799713135, |
|
"logps/chosen": -7.948614597320557, |
|
"logps/rejected": -8.05911636352539, |
|
"loss": 8.8473, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -7.948614597320557, |
|
"rewards/margins": 0.11050298064947128, |
|
"rewards/rejected": -8.05911636352539, |
|
"step": 435 |
|
}, |
|
{ |
|
"epoch": 0.9212248102590945, |
|
"grad_norm": 13.136814821473001, |
|
"learning_rate": 9.12094829893642e-09, |
|
"logits/chosen": -1.327292799949646, |
|
"logits/rejected": -1.3260419368743896, |
|
"logps/chosen": -8.011943817138672, |
|
"logps/rejected": -8.084588050842285, |
|
"loss": 9.0087, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -8.011943817138672, |
|
"rewards/margins": 0.07264318317174911, |
|
"rewards/rejected": -8.084588050842285, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.9316932740120387, |
|
"grad_norm": 13.772252146186085, |
|
"learning_rate": 6.832927412229017e-09, |
|
"logits/chosen": -1.3517519235610962, |
|
"logits/rejected": -1.353088140487671, |
|
"logps/chosen": -7.9066948890686035, |
|
"logps/rejected": -8.016542434692383, |
|
"loss": 8.9143, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -7.9066948890686035, |
|
"rewards/margins": 0.10984818637371063, |
|
"rewards/rejected": -8.016542434692383, |
|
"step": 445 |
|
}, |
|
{ |
|
"epoch": 0.942161737764983, |
|
"grad_norm": 62.09067243328178, |
|
"learning_rate": 4.8708793644441086e-09, |
|
"logits/chosen": -1.3028905391693115, |
|
"logits/rejected": -1.2930048704147339, |
|
"logps/chosen": -8.025864601135254, |
|
"logps/rejected": -8.012077331542969, |
|
"loss": 9.0079, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -8.025864601135254, |
|
"rewards/margins": -0.013786206021904945, |
|
"rewards/rejected": -8.012077331542969, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.9526302015179272, |
|
"grad_norm": 13.419716763500347, |
|
"learning_rate": 3.2374343405217884e-09, |
|
"logits/chosen": -1.3575193881988525, |
|
"logits/rejected": -1.351488709449768, |
|
"logps/chosen": -7.829765319824219, |
|
"logps/rejected": -8.183721542358398, |
|
"loss": 8.9117, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -7.829765319824219, |
|
"rewards/margins": 0.3539566695690155, |
|
"rewards/rejected": -8.183721542358398, |
|
"step": 455 |
|
}, |
|
{ |
|
"epoch": 0.9630986652708715, |
|
"grad_norm": 22.929649885843165, |
|
"learning_rate": 1.9347820230782295e-09, |
|
"logits/chosen": -1.3399038314819336, |
|
"logits/rejected": -1.352718710899353, |
|
"logps/chosen": -7.996813774108887, |
|
"logps/rejected": -8.18346881866455, |
|
"loss": 8.8927, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -7.996813774108887, |
|
"rewards/margins": 0.18665429949760437, |
|
"rewards/rejected": -8.18346881866455, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.9735671290238157, |
|
"grad_norm": 12.997334149917643, |
|
"learning_rate": 9.64668657069706e-10, |
|
"logits/chosen": -1.363525390625, |
|
"logits/rejected": -1.3654100894927979, |
|
"logps/chosen": -8.123039245605469, |
|
"logps/rejected": -8.254613876342773, |
|
"loss": 8.999, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -8.123039245605469, |
|
"rewards/margins": 0.13157425820827484, |
|
"rewards/rejected": -8.254613876342773, |
|
"step": 465 |
|
}, |
|
{ |
|
"epoch": 0.98403559277676, |
|
"grad_norm": 14.345805639596328, |
|
"learning_rate": 3.2839470889836627e-10, |
|
"logits/chosen": -1.3454748392105103, |
|
"logits/rejected": -1.3637425899505615, |
|
"logps/chosen": -7.719171047210693, |
|
"logps/rejected": -7.984010219573975, |
|
"loss": 8.9911, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -7.719171047210693, |
|
"rewards/margins": 0.26483961939811707, |
|
"rewards/rejected": -7.984010219573975, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.9945040565297043, |
|
"grad_norm": 15.884627478532495, |
|
"learning_rate": 2.6813123097352287e-11, |
|
"logits/chosen": -1.3404934406280518, |
|
"logits/rejected": -1.329254388809204, |
|
"logps/chosen": -8.111069679260254, |
|
"logps/rejected": -8.152128219604492, |
|
"loss": 8.9896, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -8.111069679260254, |
|
"rewards/margins": 0.041059140115976334, |
|
"rewards/rejected": -8.152128219604492, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 0.998691442030882, |
|
"step": 477, |
|
"total_flos": 0.0, |
|
"train_loss": 8.9737869758526, |
|
"train_runtime": 8198.1649, |
|
"train_samples_per_second": 7.457, |
|
"train_steps_per_second": 0.058 |
|
} |
|
], |
|
"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 |
|
} |
|
|