zephyr-7b-SimPO-SELM / trainer_state.json
Teng Xiao
commit from TX
05067e7
{
"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": 22.273845933282402,
"learning_rate": 5.208333333333333e-08,
"logits/chosen": -2.770733118057251,
"logits/rejected": -2.7281923294067383,
"logps/chosen": -1.0278276205062866,
"logps/rejected": -1.1733704805374146,
"loss": 1.7067,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -1.0278276205062866,
"rewards/margins": 0.14554281532764435,
"rewards/rejected": -1.1733704805374146,
"step": 5
},
{
"epoch": 0.02093692750588851,
"grad_norm": 19.65262772274273,
"learning_rate": 1.0416666666666667e-07,
"logits/chosen": -2.7513110637664795,
"logits/rejected": -2.725446939468384,
"logps/chosen": -0.9947978258132935,
"logps/rejected": -1.0709822177886963,
"loss": 1.7052,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.9947978258132935,
"rewards/margins": 0.07618431746959686,
"rewards/rejected": -1.0709822177886963,
"step": 10
},
{
"epoch": 0.031405391258832765,
"grad_norm": 20.86060525388248,
"learning_rate": 1.5624999999999999e-07,
"logits/chosen": -2.7313146591186523,
"logits/rejected": -2.697767496109009,
"logps/chosen": -0.9808584451675415,
"logps/rejected": -1.1414306163787842,
"loss": 1.6534,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.9808584451675415,
"rewards/margins": 0.16057202219963074,
"rewards/rejected": -1.1414306163787842,
"step": 15
},
{
"epoch": 0.04187385501177702,
"grad_norm": 18.455889774407584,
"learning_rate": 2.0833333333333333e-07,
"logits/chosen": -2.8040003776550293,
"logits/rejected": -2.7113003730773926,
"logps/chosen": -0.997894287109375,
"logps/rejected": -1.098174810409546,
"loss": 1.6781,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.997894287109375,
"rewards/margins": 0.1002805083990097,
"rewards/rejected": -1.098174810409546,
"step": 20
},
{
"epoch": 0.05234231876472128,
"grad_norm": 17.169851595349584,
"learning_rate": 2.604166666666667e-07,
"logits/chosen": -2.8004283905029297,
"logits/rejected": -2.777601718902588,
"logps/chosen": -0.9775938987731934,
"logps/rejected": -1.0392240285873413,
"loss": 1.6816,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.9775938987731934,
"rewards/margins": 0.061630137264728546,
"rewards/rejected": -1.0392240285873413,
"step": 25
},
{
"epoch": 0.06281078251766553,
"grad_norm": 17.60907134514165,
"learning_rate": 3.1249999999999997e-07,
"logits/chosen": -2.7743477821350098,
"logits/rejected": -2.734339714050293,
"logps/chosen": -0.8839586973190308,
"logps/rejected": -0.9886126518249512,
"loss": 1.7146,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.8839586973190308,
"rewards/margins": 0.1046539768576622,
"rewards/rejected": -0.9886126518249512,
"step": 30
},
{
"epoch": 0.07327924627060979,
"grad_norm": 19.779646214864712,
"learning_rate": 3.645833333333333e-07,
"logits/chosen": -2.7794618606567383,
"logits/rejected": -2.6966075897216797,
"logps/chosen": -0.9109536409378052,
"logps/rejected": -1.0334917306900024,
"loss": 1.6813,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.9109536409378052,
"rewards/margins": 0.12253817170858383,
"rewards/rejected": -1.0334917306900024,
"step": 35
},
{
"epoch": 0.08374771002355404,
"grad_norm": 17.435155033915933,
"learning_rate": 4.1666666666666667e-07,
"logits/chosen": -2.796264171600342,
"logits/rejected": -2.706550121307373,
"logps/chosen": -0.8977090120315552,
"logps/rejected": -1.1700526475906372,
"loss": 1.58,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.8977090120315552,
"rewards/margins": 0.2723436951637268,
"rewards/rejected": -1.1700526475906372,
"step": 40
},
{
"epoch": 0.0942161737764983,
"grad_norm": 20.376725274104686,
"learning_rate": 4.6874999999999996e-07,
"logits/chosen": -2.7916207313537598,
"logits/rejected": -2.7479615211486816,
"logps/chosen": -0.921708881855011,
"logps/rejected": -1.0657963752746582,
"loss": 1.5991,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.921708881855011,
"rewards/margins": 0.14408749341964722,
"rewards/rejected": -1.0657963752746582,
"step": 45
},
{
"epoch": 0.10468463752944256,
"grad_norm": 24.688089874039832,
"learning_rate": 4.999731868769026e-07,
"logits/chosen": -2.7189624309539795,
"logits/rejected": -2.657292604446411,
"logps/chosen": -0.9823735952377319,
"logps/rejected": -1.2429628372192383,
"loss": 1.5995,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.9823735952377319,
"rewards/margins": 0.2605893015861511,
"rewards/rejected": -1.2429628372192383,
"step": 50
},
{
"epoch": 0.11515310128238682,
"grad_norm": 20.10805225669153,
"learning_rate": 4.996716052911017e-07,
"logits/chosen": -2.685032367706299,
"logits/rejected": -2.6509592533111572,
"logps/chosen": -1.0144091844558716,
"logps/rejected": -1.2263678312301636,
"loss": 1.6066,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.0144091844558716,
"rewards/margins": 0.211958646774292,
"rewards/rejected": -1.2263678312301636,
"step": 55
},
{
"epoch": 0.12562156503533106,
"grad_norm": 23.40387121073354,
"learning_rate": 4.990353313429303e-07,
"logits/chosen": -2.717456817626953,
"logits/rejected": -2.669968843460083,
"logps/chosen": -1.071919322013855,
"logps/rejected": -1.261800765991211,
"loss": 1.5192,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.071919322013855,
"rewards/margins": 0.18988142907619476,
"rewards/rejected": -1.261800765991211,
"step": 60
},
{
"epoch": 0.1360900287882753,
"grad_norm": 22.52237455325428,
"learning_rate": 4.980652179769217e-07,
"logits/chosen": -2.681735038757324,
"logits/rejected": -2.568328857421875,
"logps/chosen": -1.0160045623779297,
"logps/rejected": -1.500519037246704,
"loss": 1.5548,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.0160045623779297,
"rewards/margins": 0.48451438546180725,
"rewards/rejected": -1.500519037246704,
"step": 65
},
{
"epoch": 0.14655849254121958,
"grad_norm": 29.253552274403233,
"learning_rate": 4.967625656594781e-07,
"logits/chosen": -2.5312721729278564,
"logits/rejected": -2.480713129043579,
"logps/chosen": -1.1166826486587524,
"logps/rejected": -1.405466914176941,
"loss": 1.5402,
"rewards/accuracies": 0.59375,
"rewards/chosen": -1.1166826486587524,
"rewards/margins": 0.2887844145298004,
"rewards/rejected": -1.405466914176941,
"step": 70
},
{
"epoch": 0.15702695629416383,
"grad_norm": 31.175214085255675,
"learning_rate": 4.951291206355559e-07,
"logits/chosen": -2.567188024520874,
"logits/rejected": -2.501024007797241,
"logps/chosen": -1.0800148248672485,
"logps/rejected": -1.4306553602218628,
"loss": 1.5108,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.0800148248672485,
"rewards/margins": 0.3506406545639038,
"rewards/rejected": -1.4306553602218628,
"step": 75
},
{
"epoch": 0.16749542004710807,
"grad_norm": 23.89446565720433,
"learning_rate": 4.93167072587771e-07,
"logits/chosen": -2.5460524559020996,
"logits/rejected": -2.4932408332824707,
"logps/chosen": -1.083786964416504,
"logps/rejected": -1.4139994382858276,
"loss": 1.557,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.083786964416504,
"rewards/margins": 0.3302125036716461,
"rewards/rejected": -1.4139994382858276,
"step": 80
},
{
"epoch": 0.17796388380005235,
"grad_norm": 35.04760138274803,
"learning_rate": 4.908790517010636e-07,
"logits/chosen": -2.555619239807129,
"logits/rejected": -2.49599289894104,
"logps/chosen": -1.027801275253296,
"logps/rejected": -1.3224855661392212,
"loss": 1.599,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -1.027801275253296,
"rewards/margins": 0.29468411207199097,
"rewards/rejected": -1.3224855661392212,
"step": 85
},
{
"epoch": 0.1884323475529966,
"grad_norm": 24.7603865878364,
"learning_rate": 4.882681251368548e-07,
"logits/chosen": -2.4858529567718506,
"logits/rejected": -2.458529472351074,
"logps/chosen": -1.0844331979751587,
"logps/rejected": -1.3636631965637207,
"loss": 1.5441,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.0844331979751587,
"rewards/margins": 0.27922993898391724,
"rewards/rejected": -1.3636631965637207,
"step": 90
},
{
"epoch": 0.19890081130594087,
"grad_norm": 31.023544965925865,
"learning_rate": 4.853377929214243e-07,
"logits/chosen": -2.50390887260437,
"logits/rejected": -2.40189790725708,
"logps/chosen": -1.1321955919265747,
"logps/rejected": -1.4152982234954834,
"loss": 1.5605,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.1321955919265747,
"rewards/margins": 0.28310269117355347,
"rewards/rejected": -1.4152982234954834,
"step": 95
},
{
"epoch": 0.2093692750588851,
"grad_norm": 29.51747892906268,
"learning_rate": 4.820919832540181e-07,
"logits/chosen": -2.4223530292510986,
"logits/rejected": -2.3233771324157715,
"logps/chosen": -1.153446912765503,
"logps/rejected": -1.5384576320648193,
"loss": 1.5601,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.153446912765503,
"rewards/margins": 0.3850107789039612,
"rewards/rejected": -1.5384576320648193,
"step": 100
},
{
"epoch": 0.21983773881182936,
"grad_norm": 30.939558746298886,
"learning_rate": 4.785350472409791e-07,
"logits/chosen": -2.3625779151916504,
"logits/rejected": -2.3349967002868652,
"logps/chosen": -1.1278746128082275,
"logps/rejected": -1.5877573490142822,
"loss": 1.4734,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.1278746128082275,
"rewards/margins": 0.4598826766014099,
"rewards/rejected": -1.5877573490142822,
"step": 105
},
{
"epoch": 0.23030620256477363,
"grad_norm": 45.105416576816324,
"learning_rate": 4.7467175306295647e-07,
"logits/chosen": -2.367917537689209,
"logits/rejected": -2.291100025177002,
"logps/chosen": -1.2823549509048462,
"logps/rejected": -1.7042672634124756,
"loss": 1.5012,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.2823549509048462,
"rewards/margins": 0.42191243171691895,
"rewards/rejected": -1.7042672634124756,
"step": 110
},
{
"epoch": 0.24077466631771788,
"grad_norm": 26.508438238557048,
"learning_rate": 4.70507279583015e-07,
"logits/chosen": -2.2769744396209717,
"logits/rejected": -2.1966700553894043,
"logps/chosen": -1.2124507427215576,
"logps/rejected": -1.7381147146224976,
"loss": 1.4243,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.2124507427215576,
"rewards/margins": 0.5256639719009399,
"rewards/rejected": -1.7381147146224976,
"step": 115
},
{
"epoch": 0.2512431300706621,
"grad_norm": 36.57968166514704,
"learning_rate": 4.6604720940421207e-07,
"logits/chosen": -2.2149672508239746,
"logits/rejected": -2.1592419147491455,
"logps/chosen": -1.2770636081695557,
"logps/rejected": -1.6595499515533447,
"loss": 1.5083,
"rewards/accuracies": 0.59375,
"rewards/chosen": -1.2770636081695557,
"rewards/margins": 0.3824862837791443,
"rewards/rejected": -1.6595499515533447,
"step": 120
},
{
"epoch": 0.26171159382360637,
"grad_norm": 30.33542228792605,
"learning_rate": 4.612975213859487e-07,
"logits/chosen": -2.2143490314483643,
"logits/rejected": -2.177022695541382,
"logps/chosen": -1.2711435556411743,
"logps/rejected": -1.5601117610931396,
"loss": 1.5047,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.2711435556411743,
"rewards/margins": 0.28896820545196533,
"rewards/rejected": -1.5601117610931396,
"step": 125
},
{
"epoch": 0.2721800575765506,
"grad_norm": 31.4132000967526,
"learning_rate": 4.5626458262912735e-07,
"logits/chosen": -2.0743236541748047,
"logits/rejected": -2.0040838718414307,
"logps/chosen": -1.3241310119628906,
"logps/rejected": -1.5913293361663818,
"loss": 1.47,
"rewards/accuracies": 0.65625,
"rewards/chosen": -1.3241310119628906,
"rewards/margins": 0.2671981453895569,
"rewards/rejected": -1.5913293361663818,
"step": 130
},
{
"epoch": 0.2826485213294949,
"grad_norm": 54.25569743941121,
"learning_rate": 4.5095513994085974e-07,
"logits/chosen": -1.8961169719696045,
"logits/rejected": -1.7147821187973022,
"logps/chosen": -1.5041354894638062,
"logps/rejected": -2.126206398010254,
"loss": 1.4478,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.5041354894638062,
"rewards/margins": 0.6220710277557373,
"rewards/rejected": -2.126206398010254,
"step": 135
},
{
"epoch": 0.29311698508243916,
"grad_norm": 43.983222727893036,
"learning_rate": 4.453763107901675e-07,
"logits/chosen": -1.5513172149658203,
"logits/rejected": -1.3982717990875244,
"logps/chosen": -1.5197192430496216,
"logps/rejected": -2.097294330596924,
"loss": 1.3929,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.5197192430496216,
"rewards/margins": 0.5775750279426575,
"rewards/rejected": -2.097294330596924,
"step": 140
},
{
"epoch": 0.3035854488353834,
"grad_norm": 28.469754678991702,
"learning_rate": 4.395355737667985e-07,
"logits/chosen": -1.4272167682647705,
"logits/rejected": -1.1418006420135498,
"logps/chosen": -1.4365965127944946,
"logps/rejected": -1.9005510807037354,
"loss": 1.442,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.4365965127944946,
"rewards/margins": 0.46395477652549744,
"rewards/rejected": -1.9005510807037354,
"step": 145
},
{
"epoch": 0.31405391258832765,
"grad_norm": 41.18537586594856,
"learning_rate": 4.3344075855595097e-07,
"logits/chosen": -1.3107974529266357,
"logits/rejected": -1.1333634853363037,
"logps/chosen": -1.333939552307129,
"logps/rejected": -1.8191452026367188,
"loss": 1.3703,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -1.333939552307129,
"rewards/margins": 0.4852055609226227,
"rewards/rejected": -1.8191452026367188,
"step": 150
},
{
"epoch": 0.3245223763412719,
"grad_norm": 43.62414965682813,
"learning_rate": 4.271000354423425e-07,
"logits/chosen": -0.9065883755683899,
"logits/rejected": -0.6428723931312561,
"logps/chosen": -1.6046466827392578,
"logps/rejected": -2.088477373123169,
"loss": 1.393,
"rewards/accuracies": 0.71875,
"rewards/chosen": -1.6046466827392578,
"rewards/margins": 0.48383063077926636,
"rewards/rejected": -2.088477373123169,
"step": 155
},
{
"epoch": 0.33499084009421615,
"grad_norm": 58.236255602056396,
"learning_rate": 4.2052190435769554e-07,
"logits/chosen": -0.5077437162399292,
"logits/rejected": -0.2604050636291504,
"logps/chosen": -1.8002322912216187,
"logps/rejected": -2.378002882003784,
"loss": 1.3526,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -1.8002322912216187,
"rewards/margins": 0.5777708292007446,
"rewards/rejected": -2.378002882003784,
"step": 160
},
{
"epoch": 0.34545930384716045,
"grad_norm": 37.97133552354845,
"learning_rate": 4.137151834863213e-07,
"logits/chosen": -0.9249609112739563,
"logits/rejected": -0.7287668585777283,
"logps/chosen": -1.729589819908142,
"logps/rejected": -2.2937121391296387,
"loss": 1.3021,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.729589819908142,
"rewards/margins": 0.5641222596168518,
"rewards/rejected": -2.2937121391296387,
"step": 165
},
{
"epoch": 0.3559277676001047,
"grad_norm": 58.781010614086675,
"learning_rate": 4.0668899744407567e-07,
"logits/chosen": -1.0361697673797607,
"logits/rejected": -0.7538817524909973,
"logps/chosen": -1.8726847171783447,
"logps/rejected": -2.472113847732544,
"loss": 1.4168,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.8726847171783447,
"rewards/margins": 0.5994290709495544,
"rewards/rejected": -2.472113847732544,
"step": 170
},
{
"epoch": 0.36639623135304894,
"grad_norm": 43.77533333071548,
"learning_rate": 3.994527650465352e-07,
"logits/chosen": -1.0444451570510864,
"logits/rejected": -0.6541527509689331,
"logps/chosen": -1.9884567260742188,
"logps/rejected": -2.6078011989593506,
"loss": 1.4179,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -1.9884567260742188,
"rewards/margins": 0.6193446516990662,
"rewards/rejected": -2.6078011989593506,
"step": 175
},
{
"epoch": 0.3768646951059932,
"grad_norm": 39.30408636565025,
"learning_rate": 3.920161866827889e-07,
"logits/chosen": -1.1477383375167847,
"logits/rejected": -0.9928515553474426,
"logps/chosen": -2.031057596206665,
"logps/rejected": -2.6500778198242188,
"loss": 1.3292,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -2.031057596206665,
"rewards/margins": 0.6190204620361328,
"rewards/rejected": -2.6500778198242188,
"step": 180
},
{
"epoch": 0.38733315885893743,
"grad_norm": 49.18397826699053,
"learning_rate": 3.8438923131177237e-07,
"logits/chosen": -1.3022983074188232,
"logits/rejected": -1.1097471714019775,
"logps/chosen": -2.1825501918792725,
"logps/rejected": -2.8575053215026855,
"loss": 1.3409,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -2.1825501918792725,
"rewards/margins": 0.6749551892280579,
"rewards/rejected": -2.8575053215026855,
"step": 185
},
{
"epoch": 0.39780162261188173,
"grad_norm": 59.443069428032004,
"learning_rate": 3.765821230985757e-07,
"logits/chosen": -1.309631586074829,
"logits/rejected": -1.1841962337493896,
"logps/chosen": -2.1491549015045166,
"logps/rejected": -2.7087011337280273,
"loss": 1.4201,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -2.1491549015045166,
"rewards/margins": 0.5595459938049316,
"rewards/rejected": -2.7087011337280273,
"step": 190
},
{
"epoch": 0.408270086364826,
"grad_norm": 58.81635631771188,
"learning_rate": 3.6860532770864005e-07,
"logits/chosen": -0.9547362327575684,
"logits/rejected": -0.6742358803749084,
"logps/chosen": -2.285407066345215,
"logps/rejected": -2.962996006011963,
"loss": 1.3681,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.285407066345215,
"rewards/margins": 0.6775889992713928,
"rewards/rejected": -2.962996006011963,
"step": 195
},
{
"epoch": 0.4187385501177702,
"grad_norm": 61.19133999845696,
"learning_rate": 3.604695382782159e-07,
"logits/chosen": -0.5743089914321899,
"logits/rejected": -0.5020885467529297,
"logps/chosen": -2.337622880935669,
"logps/rejected": -2.9621822834014893,
"loss": 1.3002,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.337622880935669,
"rewards/margins": 0.6245594620704651,
"rewards/rejected": -2.9621822834014893,
"step": 200
},
{
"epoch": 0.42920701387071447,
"grad_norm": 65.1623208567972,
"learning_rate": 3.5218566107988867e-07,
"logits/chosen": -0.27199602127075195,
"logits/rejected": 0.03203292936086655,
"logps/chosen": -2.4965696334838867,
"logps/rejected": -3.200573682785034,
"loss": 1.2788,
"rewards/accuracies": 0.6875,
"rewards/chosen": -2.4965696334838867,
"rewards/margins": 0.7040041089057922,
"rewards/rejected": -3.200573682785034,
"step": 205
},
{
"epoch": 0.4396754776236587,
"grad_norm": 81.87887955731682,
"learning_rate": 3.4376480090239047e-07,
"logits/chosen": -0.3396000266075134,
"logits/rejected": -0.0072940438985824585,
"logps/chosen": -2.886606216430664,
"logps/rejected": -3.522667407989502,
"loss": 1.3707,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.886606216430664,
"rewards/margins": 0.6360612511634827,
"rewards/rejected": -3.522667407989502,
"step": 210
},
{
"epoch": 0.45014394137660296,
"grad_norm": 54.53418228916395,
"learning_rate": 3.3521824616429284e-07,
"logits/chosen": -0.3693031072616577,
"logits/rejected": -0.05546772480010986,
"logps/chosen": -2.6358675956726074,
"logps/rejected": -3.3469290733337402,
"loss": 1.2589,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -2.6358675956726074,
"rewards/margins": 0.7110615968704224,
"rewards/rejected": -3.3469290733337402,
"step": 215
},
{
"epoch": 0.46061240512954726,
"grad_norm": 61.5068583100674,
"learning_rate": 3.265574537815398e-07,
"logits/chosen": -0.5512332320213318,
"logits/rejected": -0.04919959604740143,
"logps/chosen": -2.361748218536377,
"logps/rejected": -3.2458934783935547,
"loss": 1.2447,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.361748218536377,
"rewards/margins": 0.8841449022293091,
"rewards/rejected": -3.2458934783935547,
"step": 220
},
{
"epoch": 0.4710808688824915,
"grad_norm": 62.18444756933803,
"learning_rate": 3.1779403380910425e-07,
"logits/chosen": -0.4573966860771179,
"logits/rejected": 0.10396180301904678,
"logps/chosen": -2.5001068115234375,
"logps/rejected": -3.3893027305603027,
"loss": 1.2645,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -2.5001068115234375,
"rewards/margins": 0.8891960978507996,
"rewards/rejected": -3.3893027305603027,
"step": 225
},
{
"epoch": 0.48154933263543576,
"grad_norm": 69.58783982000695,
"learning_rate": 3.0893973387735683e-07,
"logits/chosen": -0.23679371178150177,
"logits/rejected": 0.2879559099674225,
"logps/chosen": -3.0258288383483887,
"logps/rejected": -4.09028959274292,
"loss": 1.1493,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -3.0258288383483887,
"rewards/margins": 1.0644607543945312,
"rewards/rejected": -4.09028959274292,
"step": 230
},
{
"epoch": 0.49201779638838,
"grad_norm": 65.85335888107983,
"learning_rate": 3.000064234440111e-07,
"logits/chosen": -0.09591199457645416,
"logits/rejected": 0.3684416711330414,
"logps/chosen": -2.861621379852295,
"logps/rejected": -3.635110855102539,
"loss": 1.2158,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.861621379852295,
"rewards/margins": 0.7734889984130859,
"rewards/rejected": -3.635110855102539,
"step": 235
},
{
"epoch": 0.5024862601413242,
"grad_norm": 51.47234347289653,
"learning_rate": 2.910060778827554e-07,
"logits/chosen": -0.214645653963089,
"logits/rejected": 0.15288104116916656,
"logps/chosen": -2.2756009101867676,
"logps/rejected": -3.0374677181243896,
"loss": 1.2342,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.2756009101867676,
"rewards/margins": 0.7618668079376221,
"rewards/rejected": -3.0374677181243896,
"step": 240
},
{
"epoch": 0.5129547238942685,
"grad_norm": 67.41003053954276,
"learning_rate": 2.8195076242990116e-07,
"logits/chosen": -0.233026385307312,
"logits/rejected": 0.1556404083967209,
"logps/chosen": -2.4698057174682617,
"logps/rejected": -3.129883050918579,
"loss": 1.2922,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -2.4698057174682617,
"rewards/margins": 0.6600776314735413,
"rewards/rejected": -3.129883050918579,
"step": 245
},
{
"epoch": 0.5234231876472127,
"grad_norm": 70.04740584020762,
"learning_rate": 2.7285261601056697e-07,
"logits/chosen": -0.32400113344192505,
"logits/rejected": 0.12020448595285416,
"logps/chosen": -2.376567840576172,
"logps/rejected": -3.175973653793335,
"loss": 1.3275,
"rewards/accuracies": 0.71875,
"rewards/chosen": -2.376567840576172,
"rewards/margins": 0.799405574798584,
"rewards/rejected": -3.175973653793335,
"step": 250
},
{
"epoch": 0.533891651400157,
"grad_norm": 49.05614284788625,
"learning_rate": 2.6372383496608186e-07,
"logits/chosen": -0.07061919569969177,
"logits/rejected": 0.49763208627700806,
"logps/chosen": -2.67657470703125,
"logps/rejected": -3.5274453163146973,
"loss": 1.1737,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -2.67657470703125,
"rewards/margins": 0.8508706092834473,
"rewards/rejected": -3.5274453163146973,
"step": 255
},
{
"epoch": 0.5443601151531012,
"grad_norm": 73.89578979282193,
"learning_rate": 2.5457665670441937e-07,
"logits/chosen": -0.18951067328453064,
"logits/rejected": 0.2982691824436188,
"logps/chosen": -2.413672924041748,
"logps/rejected": -3.579845905303955,
"loss": 1.2239,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -2.413672924041748,
"rewards/margins": 1.166172981262207,
"rewards/rejected": -3.579845905303955,
"step": 260
},
{
"epoch": 0.5548285789060455,
"grad_norm": 54.444851596761815,
"learning_rate": 2.454233432955807e-07,
"logits/chosen": -0.05852239206433296,
"logits/rejected": 0.3520205318927765,
"logps/chosen": -2.7941486835479736,
"logps/rejected": -3.663743495941162,
"loss": 1.1459,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -2.7941486835479736,
"rewards/margins": 0.8695951700210571,
"rewards/rejected": -3.663743495941162,
"step": 265
},
{
"epoch": 0.5652970426589898,
"grad_norm": 79.13341556564741,
"learning_rate": 2.3627616503391812e-07,
"logits/chosen": 0.12447645515203476,
"logits/rejected": 0.48517122864723206,
"logps/chosen": -3.074282169342041,
"logps/rejected": -3.778022289276123,
"loss": 1.2406,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -3.074282169342041,
"rewards/margins": 0.7037407159805298,
"rewards/rejected": -3.778022289276123,
"step": 270
},
{
"epoch": 0.575765506411934,
"grad_norm": 65.77375872071654,
"learning_rate": 2.2714738398943308e-07,
"logits/chosen": 0.10315855592489243,
"logits/rejected": 0.6628378629684448,
"logps/chosen": -2.9850118160247803,
"logps/rejected": -3.925368070602417,
"loss": 1.1927,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -2.9850118160247803,
"rewards/margins": 0.9403561353683472,
"rewards/rejected": -3.925368070602417,
"step": 275
},
{
"epoch": 0.5862339701648783,
"grad_norm": 68.7072404828179,
"learning_rate": 2.1804923757009882e-07,
"logits/chosen": 0.14654329419136047,
"logits/rejected": 0.7450932264328003,
"logps/chosen": -2.682091474533081,
"logps/rejected": -3.504145860671997,
"loss": 1.224,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -2.682091474533081,
"rewards/margins": 0.8220542669296265,
"rewards/rejected": -3.504145860671997,
"step": 280
},
{
"epoch": 0.5967024339178225,
"grad_norm": 61.507759714231206,
"learning_rate": 2.089939221172446e-07,
"logits/chosen": 0.048857107758522034,
"logits/rejected": 0.40524905920028687,
"logps/chosen": -2.660057306289673,
"logps/rejected": -3.4508941173553467,
"loss": 1.2054,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.660057306289673,
"rewards/margins": 0.7908371090888977,
"rewards/rejected": -3.4508941173553467,
"step": 285
},
{
"epoch": 0.6071708976707668,
"grad_norm": 68.54045280221443,
"learning_rate": 1.9999357655598891e-07,
"logits/chosen": 0.09905894100666046,
"logits/rejected": 0.6364008784294128,
"logps/chosen": -2.7884061336517334,
"logps/rejected": -3.744817018508911,
"loss": 1.1811,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.7884061336517334,
"rewards/margins": 0.9564107060432434,
"rewards/rejected": -3.744817018508911,
"step": 290
},
{
"epoch": 0.6176393614237111,
"grad_norm": 69.13964326129287,
"learning_rate": 1.9106026612264315e-07,
"logits/chosen": 0.03760681673884392,
"logits/rejected": 0.6295174956321716,
"logps/chosen": -2.8681061267852783,
"logps/rejected": -3.6628737449645996,
"loss": 1.2487,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -2.8681061267852783,
"rewards/margins": 0.7947677373886108,
"rewards/rejected": -3.6628737449645996,
"step": 295
},
{
"epoch": 0.6281078251766553,
"grad_norm": 68.24880568918918,
"learning_rate": 1.8220596619089573e-07,
"logits/chosen": 0.05892186239361763,
"logits/rejected": 0.4875253140926361,
"logps/chosen": -2.8865182399749756,
"logps/rejected": -3.7262775897979736,
"loss": 1.2566,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -2.8865182399749756,
"rewards/margins": 0.8397592306137085,
"rewards/rejected": -3.7262775897979736,
"step": 300
},
{
"epoch": 0.6385762889295996,
"grad_norm": 63.303760044846996,
"learning_rate": 1.7344254621846017e-07,
"logits/chosen": 0.2723705768585205,
"logits/rejected": 0.7761790752410889,
"logps/chosen": -2.930652141571045,
"logps/rejected": -3.947206974029541,
"loss": 1.1113,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -2.930652141571045,
"rewards/margins": 1.0165544748306274,
"rewards/rejected": -3.947206974029541,
"step": 305
},
{
"epoch": 0.6490447526825438,
"grad_norm": 74.87714108878664,
"learning_rate": 1.647817538357072e-07,
"logits/chosen": 0.40727514028549194,
"logits/rejected": 0.6944658160209656,
"logps/chosen": -2.684722900390625,
"logps/rejected": -3.821376085281372,
"loss": 1.1187,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.684722900390625,
"rewards/margins": 1.1366530656814575,
"rewards/rejected": -3.821376085281372,
"step": 310
},
{
"epoch": 0.6595132164354881,
"grad_norm": 78.50549857938152,
"learning_rate": 1.562351990976095e-07,
"logits/chosen": 0.38405269384384155,
"logits/rejected": 0.9985488653182983,
"logps/chosen": -2.8799889087677,
"logps/rejected": -4.003491401672363,
"loss": 1.1249,
"rewards/accuracies": 0.78125,
"rewards/chosen": -2.8799889087677,
"rewards/margins": 1.1235021352767944,
"rewards/rejected": -4.003491401672363,
"step": 315
},
{
"epoch": 0.6699816801884323,
"grad_norm": 80.43258089949055,
"learning_rate": 1.478143389201113e-07,
"logits/chosen": 0.6803662776947021,
"logits/rejected": 0.8349927067756653,
"logps/chosen": -2.9795212745666504,
"logps/rejected": -3.9721953868865967,
"loss": 1.1932,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.9795212745666504,
"rewards/margins": 0.9926745295524597,
"rewards/rejected": -3.9721953868865967,
"step": 320
},
{
"epoch": 0.6804501439413766,
"grad_norm": 73.34968580085386,
"learning_rate": 1.3953046172178413e-07,
"logits/chosen": 0.5375081300735474,
"logits/rejected": 1.211456060409546,
"logps/chosen": -2.997009754180908,
"logps/rejected": -3.9822990894317627,
"loss": 1.1884,
"rewards/accuracies": 0.71875,
"rewards/chosen": -2.997009754180908,
"rewards/margins": 0.9852892160415649,
"rewards/rejected": -3.9822990894317627,
"step": 325
},
{
"epoch": 0.6909186076943209,
"grad_norm": 67.18394255987927,
"learning_rate": 1.3139467229135998e-07,
"logits/chosen": 0.46189865469932556,
"logits/rejected": 1.051397681236267,
"logps/chosen": -2.7415473461151123,
"logps/rejected": -3.6691060066223145,
"loss": 1.1945,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -2.7415473461151123,
"rewards/margins": 0.9275587201118469,
"rewards/rejected": -3.6691060066223145,
"step": 330
},
{
"epoch": 0.7013870714472651,
"grad_norm": 64.70715932336967,
"learning_rate": 1.2341787690142435e-07,
"logits/chosen": 0.2790513038635254,
"logits/rejected": 0.8592203259468079,
"logps/chosen": -2.72259783744812,
"logps/rejected": -3.6879966259002686,
"loss": 1.2498,
"rewards/accuracies": 0.78125,
"rewards/chosen": -2.72259783744812,
"rewards/margins": 0.965398907661438,
"rewards/rejected": -3.6879966259002686,
"step": 335
},
{
"epoch": 0.7118555352002094,
"grad_norm": 61.63125001377349,
"learning_rate": 1.1561076868822755e-07,
"logits/chosen": 0.34762030839920044,
"logits/rejected": 0.8974772691726685,
"logps/chosen": -2.9279873371124268,
"logps/rejected": -3.637186050415039,
"loss": 1.1831,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -2.9279873371124268,
"rewards/margins": 0.7091987133026123,
"rewards/rejected": -3.637186050415039,
"step": 340
},
{
"epoch": 0.7223239989531536,
"grad_norm": 69.9867150019491,
"learning_rate": 1.0798381331721107e-07,
"logits/chosen": 0.42011842131614685,
"logits/rejected": 0.9746605753898621,
"logps/chosen": -2.681163787841797,
"logps/rejected": -3.532362699508667,
"loss": 1.1815,
"rewards/accuracies": 0.71875,
"rewards/chosen": -2.681163787841797,
"rewards/margins": 0.8511991500854492,
"rewards/rejected": -3.532362699508667,
"step": 345
},
{
"epoch": 0.7327924627060979,
"grad_norm": 75.66340151463851,
"learning_rate": 1.0054723495346482e-07,
"logits/chosen": 0.5741020441055298,
"logits/rejected": 0.8616167902946472,
"logps/chosen": -2.8164191246032715,
"logps/rejected": -3.6091017723083496,
"loss": 1.2396,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -2.8164191246032715,
"rewards/margins": 0.7926827669143677,
"rewards/rejected": -3.6091017723083496,
"step": 350
},
{
"epoch": 0.7432609264590422,
"grad_norm": 74.17047524574353,
"learning_rate": 9.331100255592436e-08,
"logits/chosen": 0.5733584761619568,
"logits/rejected": 1.1586692333221436,
"logps/chosen": -2.8743791580200195,
"logps/rejected": -4.042096138000488,
"loss": 1.1432,
"rewards/accuracies": 0.84375,
"rewards/chosen": -2.8743791580200195,
"rewards/margins": 1.1677173376083374,
"rewards/rejected": -4.042096138000488,
"step": 355
},
{
"epoch": 0.7537293902119864,
"grad_norm": 73.14407398619424,
"learning_rate": 8.628481651367875e-08,
"logits/chosen": 0.29708194732666016,
"logits/rejected": 0.8764042854309082,
"logps/chosen": -3.0110630989074707,
"logps/rejected": -3.7496750354766846,
"loss": 1.1891,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -3.0110630989074707,
"rewards/margins": 0.7386118173599243,
"rewards/rejected": -3.7496750354766846,
"step": 360
},
{
"epoch": 0.7641978539649307,
"grad_norm": 84.79816410361819,
"learning_rate": 7.947809564230445e-08,
"logits/chosen": 0.5876302719116211,
"logits/rejected": 1.1461758613586426,
"logps/chosen": -3.0007166862487793,
"logps/rejected": -4.093684673309326,
"loss": 1.2364,
"rewards/accuracies": 0.78125,
"rewards/chosen": -3.0007166862487793,
"rewards/margins": 1.0929677486419678,
"rewards/rejected": -4.093684673309326,
"step": 365
},
{
"epoch": 0.7746663177178749,
"grad_norm": 76.4181977449788,
"learning_rate": 7.289996455765748e-08,
"logits/chosen": 0.7314077615737915,
"logits/rejected": 1.2388323545455933,
"logps/chosen": -3.2148640155792236,
"logps/rejected": -4.329017639160156,
"loss": 1.1397,
"rewards/accuracies": 0.78125,
"rewards/chosen": -3.2148640155792236,
"rewards/margins": 1.1141533851623535,
"rewards/rejected": -4.329017639160156,
"step": 370
},
{
"epoch": 0.7851347814708192,
"grad_norm": 79.59752890916117,
"learning_rate": 6.655924144404906e-08,
"logits/chosen": 0.7917232513427734,
"logits/rejected": 0.8404657244682312,
"logps/chosen": -3.1535632610321045,
"logps/rejected": -4.065492153167725,
"loss": 1.117,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -3.1535632610321045,
"rewards/margins": 0.9119284749031067,
"rewards/rejected": -4.065492153167725,
"step": 375
},
{
"epoch": 0.7956032452237635,
"grad_norm": 83.98695473040011,
"learning_rate": 6.046442623320145e-08,
"logits/chosen": 0.9315813779830933,
"logits/rejected": 1.060347557067871,
"logps/chosen": -3.2030982971191406,
"logps/rejected": -4.173519134521484,
"loss": 1.2339,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -3.2030982971191406,
"rewards/margins": 0.9704204797744751,
"rewards/rejected": -4.173519134521484,
"step": 380
},
{
"epoch": 0.8060717089767077,
"grad_norm": 73.80815123078858,
"learning_rate": 5.4623689209832484e-08,
"logits/chosen": 0.6560049653053284,
"logits/rejected": 1.1292173862457275,
"logps/chosen": -3.0976970195770264,
"logps/rejected": -4.078602313995361,
"loss": 1.1986,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -3.0976970195770264,
"rewards/margins": 0.9809052348136902,
"rewards/rejected": -4.078602313995361,
"step": 385
},
{
"epoch": 0.816540172729652,
"grad_norm": 69.90570951076663,
"learning_rate": 4.904486005914027e-08,
"logits/chosen": 0.3861357569694519,
"logits/rejected": 1.0230625867843628,
"logps/chosen": -3.0694098472595215,
"logps/rejected": -4.231044769287109,
"loss": 1.12,
"rewards/accuracies": 0.75,
"rewards/chosen": -3.0694098472595215,
"rewards/margins": 1.161635160446167,
"rewards/rejected": -4.231044769287109,
"step": 390
},
{
"epoch": 0.8270086364825961,
"grad_norm": 71.22815378604164,
"learning_rate": 4.373541737087263e-08,
"logits/chosen": 0.35766056180000305,
"logits/rejected": 1.0771682262420654,
"logps/chosen": -2.899632692337036,
"logps/rejected": -3.877995252609253,
"loss": 1.1684,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -2.899632692337036,
"rewards/margins": 0.9783622622489929,
"rewards/rejected": -3.877995252609253,
"step": 395
},
{
"epoch": 0.8374771002355405,
"grad_norm": 97.45446724327614,
"learning_rate": 3.8702478614051345e-08,
"logits/chosen": 0.5966876745223999,
"logits/rejected": 1.0948045253753662,
"logps/chosen": -2.8743724822998047,
"logps/rejected": -3.8696448802948,
"loss": 1.2227,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -2.8743724822998047,
"rewards/margins": 0.9952724575996399,
"rewards/rejected": -3.8696448802948,
"step": 400
},
{
"epoch": 0.8479455639884846,
"grad_norm": 77.52250073936997,
"learning_rate": 3.3952790595787986e-08,
"logits/chosen": 0.5455753207206726,
"logits/rejected": 0.8719021081924438,
"logps/chosen": -3.101245880126953,
"logps/rejected": -3.933109998703003,
"loss": 1.2031,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -3.101245880126953,
"rewards/margins": 0.8318643569946289,
"rewards/rejected": -3.933109998703003,
"step": 405
},
{
"epoch": 0.8584140277414289,
"grad_norm": 72.54119105678649,
"learning_rate": 2.9492720416985e-08,
"logits/chosen": 0.6000219583511353,
"logits/rejected": 0.9464153051376343,
"logps/chosen": -2.9545369148254395,
"logps/rejected": -3.8140132427215576,
"loss": 1.2519,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -2.9545369148254395,
"rewards/margins": 0.8594767451286316,
"rewards/rejected": -3.8140132427215576,
"step": 410
},
{
"epoch": 0.8688824914943732,
"grad_norm": 57.55112621238877,
"learning_rate": 2.5328246937043525e-08,
"logits/chosen": 0.657649576663971,
"logits/rejected": 1.3006607294082642,
"logps/chosen": -2.929166078567505,
"logps/rejected": -3.863920211791992,
"loss": 1.1329,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -2.929166078567505,
"rewards/margins": 0.9347543716430664,
"rewards/rejected": -3.863920211791992,
"step": 415
},
{
"epoch": 0.8793509552473174,
"grad_norm": 67.15006871791722,
"learning_rate": 2.1464952759020856e-08,
"logits/chosen": 0.557062029838562,
"logits/rejected": 0.9161618947982788,
"logps/chosen": -2.8673954010009766,
"logps/rejected": -3.752131938934326,
"loss": 1.2637,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -2.8673954010009766,
"rewards/margins": 0.8847367167472839,
"rewards/rejected": -3.752131938934326,
"step": 420
},
{
"epoch": 0.8898194190002617,
"grad_norm": 62.826606471582075,
"learning_rate": 1.7908016745981856e-08,
"logits/chosen": 0.6115803122520447,
"logits/rejected": 1.1207200288772583,
"logps/chosen": -2.983302354812622,
"logps/rejected": -3.9271881580352783,
"loss": 1.1823,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.983302354812622,
"rewards/margins": 0.9438859820365906,
"rewards/rejected": -3.9271881580352783,
"step": 425
},
{
"epoch": 0.9002878827532059,
"grad_norm": 59.68175243856826,
"learning_rate": 1.4662207078575684e-08,
"logits/chosen": 0.5250687599182129,
"logits/rejected": 1.0637626647949219,
"logps/chosen": -2.930884599685669,
"logps/rejected": -4.001204490661621,
"loss": 1.1121,
"rewards/accuracies": 0.78125,
"rewards/chosen": -2.930884599685669,
"rewards/margins": 1.0703200101852417,
"rewards/rejected": -4.001204490661621,
"step": 430
},
{
"epoch": 0.9107563465061502,
"grad_norm": 68.60209990488998,
"learning_rate": 1.1731874863145142e-08,
"logits/chosen": 0.5153384208679199,
"logits/rejected": 0.7662239670753479,
"logps/chosen": -2.8936798572540283,
"logps/rejected": -3.7913997173309326,
"loss": 1.1818,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.8936798572540283,
"rewards/margins": 0.8977203369140625,
"rewards/rejected": -3.7913997173309326,
"step": 435
},
{
"epoch": 0.9212248102590945,
"grad_norm": 61.87029001897422,
"learning_rate": 9.12094829893642e-09,
"logits/chosen": 0.7237823009490967,
"logits/rejected": 1.0966639518737793,
"logps/chosen": -2.946139097213745,
"logps/rejected": -4.009475231170654,
"loss": 1.1835,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -2.946139097213745,
"rewards/margins": 1.0633361339569092,
"rewards/rejected": -4.009475231170654,
"step": 440
},
{
"epoch": 0.9316932740120387,
"grad_norm": 64.1896852190105,
"learning_rate": 6.832927412229017e-09,
"logits/chosen": 0.6214802861213684,
"logits/rejected": 1.0805151462554932,
"logps/chosen": -3.0665371417999268,
"logps/rejected": -4.000259876251221,
"loss": 1.1936,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -3.0665371417999268,
"rewards/margins": 0.9337228536605835,
"rewards/rejected": -4.000259876251221,
"step": 445
},
{
"epoch": 0.942161737764983,
"grad_norm": 67.09074927088828,
"learning_rate": 4.8708793644441086e-09,
"logits/chosen": 0.6587177515029907,
"logits/rejected": 0.9696787595748901,
"logps/chosen": -2.8653368949890137,
"logps/rejected": -3.8535594940185547,
"loss": 1.1292,
"rewards/accuracies": 0.78125,
"rewards/chosen": -2.8653368949890137,
"rewards/margins": 0.9882224798202515,
"rewards/rejected": -3.8535594940185547,
"step": 450
},
{
"epoch": 0.9526302015179272,
"grad_norm": 74.52088387896268,
"learning_rate": 3.2374343405217884e-09,
"logits/chosen": 0.6812163591384888,
"logits/rejected": 1.1406841278076172,
"logps/chosen": -2.954162120819092,
"logps/rejected": -3.8345863819122314,
"loss": 1.2136,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -2.954162120819092,
"rewards/margins": 0.880423903465271,
"rewards/rejected": -3.8345863819122314,
"step": 455
},
{
"epoch": 0.9630986652708715,
"grad_norm": 57.70785985857215,
"learning_rate": 1.9347820230782295e-09,
"logits/chosen": 0.8318542242050171,
"logits/rejected": 1.231487512588501,
"logps/chosen": -2.9270339012145996,
"logps/rejected": -3.9501945972442627,
"loss": 1.1763,
"rewards/accuracies": 0.75,
"rewards/chosen": -2.9270339012145996,
"rewards/margins": 1.0231605768203735,
"rewards/rejected": -3.9501945972442627,
"step": 460
},
{
"epoch": 0.9735671290238157,
"grad_norm": 66.60199874410606,
"learning_rate": 9.64668657069706e-10,
"logits/chosen": 0.6092048287391663,
"logits/rejected": 1.2739137411117554,
"logps/chosen": -2.9308724403381348,
"logps/rejected": -4.049551963806152,
"loss": 1.12,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -2.9308724403381348,
"rewards/margins": 1.1186802387237549,
"rewards/rejected": -4.049551963806152,
"step": 465
},
{
"epoch": 0.98403559277676,
"grad_norm": 72.73204833476578,
"learning_rate": 3.2839470889836627e-10,
"logits/chosen": 0.5203122496604919,
"logits/rejected": 0.9083112478256226,
"logps/chosen": -3.0056755542755127,
"logps/rejected": -3.833085536956787,
"loss": 1.1925,
"rewards/accuracies": 0.6875,
"rewards/chosen": -3.0056755542755127,
"rewards/margins": 0.8274103999137878,
"rewards/rejected": -3.833085536956787,
"step": 470
},
{
"epoch": 0.9945040565297043,
"grad_norm": 81.94973190630537,
"learning_rate": 2.6813123097352287e-11,
"logits/chosen": 0.4619743824005127,
"logits/rejected": 1.1849265098571777,
"logps/chosen": -2.943962574005127,
"logps/rejected": -3.8440794944763184,
"loss": 1.2154,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -2.943962574005127,
"rewards/margins": 0.9001175761222839,
"rewards/rejected": -3.8440794944763184,
"step": 475
},
{
"epoch": 0.998691442030882,
"step": 477,
"total_flos": 0.0,
"train_loss": 1.3288766333142166,
"train_runtime": 7370.4172,
"train_samples_per_second": 8.295,
"train_steps_per_second": 0.065
}
],
"logging_steps": 5,
"max_steps": 477,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}