Mistral-7B-Instruct-v0.2-MI-5e-7 / trainer_state.json
tengxiao1
TX
5dbd2b0
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9987943737441393,
"eval_steps": 400,
"global_step": 466,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.010716677829872739,
"grad_norm": 38.485095124933046,
"learning_rate": 5.3191489361702123e-08,
"logits/chosen": -2.398301601409912,
"logits/rejected": -2.3920369148254395,
"logps/chosen": -0.5667670965194702,
"logps/rejected": -0.5555010437965393,
"loss": 1.5468,
"rewards/accuracies": 0.4375,
"rewards/chosen": -0.5667670965194702,
"rewards/margins": -0.011266030371189117,
"rewards/rejected": -0.5555010437965393,
"step": 5
},
{
"epoch": 0.021433355659745478,
"grad_norm": 18.322790262213086,
"learning_rate": 1.0638297872340425e-07,
"logits/chosen": -2.4026951789855957,
"logits/rejected": -2.3731987476348877,
"logps/chosen": -0.5517200231552124,
"logps/rejected": -0.5781593918800354,
"loss": 1.554,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5517200231552124,
"rewards/margins": 0.026439335197210312,
"rewards/rejected": -0.5781593918800354,
"step": 10
},
{
"epoch": 0.032150033489618215,
"grad_norm": 17.062192600734857,
"learning_rate": 1.5957446808510638e-07,
"logits/chosen": -2.4439353942871094,
"logits/rejected": -2.449899196624756,
"logps/chosen": -0.5637143850326538,
"logps/rejected": -0.567409336566925,
"loss": 1.5617,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5637143850326538,
"rewards/margins": 0.0036949075292795897,
"rewards/rejected": -0.567409336566925,
"step": 15
},
{
"epoch": 0.042866711319490956,
"grad_norm": 17.524849197474367,
"learning_rate": 2.127659574468085e-07,
"logits/chosen": -2.384251117706299,
"logits/rejected": -2.3945975303649902,
"logps/chosen": -0.5456732511520386,
"logps/rejected": -0.5427749752998352,
"loss": 1.5321,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5456732511520386,
"rewards/margins": -0.002898297505453229,
"rewards/rejected": -0.5427749752998352,
"step": 20
},
{
"epoch": 0.0535833891493637,
"grad_norm": 14.321312880487431,
"learning_rate": 2.659574468085106e-07,
"logits/chosen": -2.279252529144287,
"logits/rejected": -2.2811872959136963,
"logps/chosen": -0.5260757803916931,
"logps/rejected": -0.543364405632019,
"loss": 1.53,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5260757803916931,
"rewards/margins": 0.017288658767938614,
"rewards/rejected": -0.543364405632019,
"step": 25
},
{
"epoch": 0.06430006697923643,
"grad_norm": 19.544853384077058,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -2.3907198905944824,
"logits/rejected": -2.403733015060425,
"logps/chosen": -0.5487326979637146,
"logps/rejected": -0.5386630892753601,
"loss": 1.5515,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.5487326979637146,
"rewards/margins": -0.010069593787193298,
"rewards/rejected": -0.5386630892753601,
"step": 30
},
{
"epoch": 0.07501674480910918,
"grad_norm": 24.103155081211728,
"learning_rate": 3.7234042553191484e-07,
"logits/chosen": -2.452573299407959,
"logits/rejected": -2.423945665359497,
"logps/chosen": -0.5365626811981201,
"logps/rejected": -0.5920503735542297,
"loss": 1.5478,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.5365626811981201,
"rewards/margins": 0.055487751960754395,
"rewards/rejected": -0.5920503735542297,
"step": 35
},
{
"epoch": 0.08573342263898191,
"grad_norm": 21.06293839287325,
"learning_rate": 4.25531914893617e-07,
"logits/chosen": -2.3676323890686035,
"logits/rejected": -2.349027156829834,
"logps/chosen": -0.5672295093536377,
"logps/rejected": -0.5381959676742554,
"loss": 1.5581,
"rewards/accuracies": 0.42500001192092896,
"rewards/chosen": -0.5672295093536377,
"rewards/margins": -0.02903353050351143,
"rewards/rejected": -0.5381959676742554,
"step": 40
},
{
"epoch": 0.09645010046885466,
"grad_norm": 17.138121888291746,
"learning_rate": 4.787234042553192e-07,
"logits/chosen": -2.334918260574341,
"logits/rejected": -2.3297195434570312,
"logps/chosen": -0.5420322418212891,
"logps/rejected": -0.560600757598877,
"loss": 1.5426,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5420322418212891,
"rewards/margins": 0.01856859028339386,
"rewards/rejected": -0.560600757598877,
"step": 45
},
{
"epoch": 0.1071667782987274,
"grad_norm": 16.650052527651823,
"learning_rate": 4.999367579541646e-07,
"logits/chosen": -2.3611702919006348,
"logits/rejected": -2.3566925525665283,
"logps/chosen": -0.542779803276062,
"logps/rejected": -0.5264804363250732,
"loss": 1.5287,
"rewards/accuracies": 0.45625001192092896,
"rewards/chosen": -0.542779803276062,
"rewards/margins": -0.016299355775117874,
"rewards/rejected": -0.5264804363250732,
"step": 50
},
{
"epoch": 0.11788345612860013,
"grad_norm": 25.231872930806986,
"learning_rate": 4.99550394651012e-07,
"logits/chosen": -2.4198124408721924,
"logits/rejected": -2.4228649139404297,
"logps/chosen": -0.5739614367485046,
"logps/rejected": -0.5911067128181458,
"loss": 1.5407,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5739614367485046,
"rewards/margins": 0.017145240679383278,
"rewards/rejected": -0.5911067128181458,
"step": 55
},
{
"epoch": 0.12860013395847286,
"grad_norm": 27.654893603033948,
"learning_rate": 4.988133448023071e-07,
"logits/chosen": -2.3333356380462646,
"logits/rejected": -2.342374324798584,
"logps/chosen": -0.505652129650116,
"logps/rejected": -0.52570641040802,
"loss": 1.5384,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.505652129650116,
"rewards/margins": 0.020054226741194725,
"rewards/rejected": -0.52570641040802,
"step": 60
},
{
"epoch": 0.13931681178834562,
"grad_norm": 18.301707545724692,
"learning_rate": 4.977266441646379e-07,
"logits/chosen": -2.3869385719299316,
"logits/rejected": -2.391303777694702,
"logps/chosen": -0.5198084712028503,
"logps/rejected": -0.5502195358276367,
"loss": 1.5466,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5198084712028503,
"rewards/margins": 0.030411023646593094,
"rewards/rejected": -0.5502195358276367,
"step": 65
},
{
"epoch": 0.15003348961821836,
"grad_norm": 19.461976097246225,
"learning_rate": 4.962918198495654e-07,
"logits/chosen": -2.2925384044647217,
"logits/rejected": -2.296604633331299,
"logps/chosen": -0.5172106027603149,
"logps/rejected": -0.5665396451950073,
"loss": 1.5322,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5172106027603149,
"rewards/margins": 0.04932904243469238,
"rewards/rejected": -0.5665396451950073,
"step": 70
},
{
"epoch": 0.1607501674480911,
"grad_norm": 16.227392616134246,
"learning_rate": 4.945108881776135e-07,
"logits/chosen": -2.3317081928253174,
"logits/rejected": -2.3258421421051025,
"logps/chosen": -0.48806238174438477,
"logps/rejected": -0.5191291570663452,
"loss": 1.5323,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.48806238174438477,
"rewards/margins": 0.031066760420799255,
"rewards/rejected": -0.5191291570663452,
"step": 75
},
{
"epoch": 0.17146684527796383,
"grad_norm": 18.668922205376074,
"learning_rate": 4.923863518447879e-07,
"logits/chosen": -2.304633617401123,
"logits/rejected": -2.3267812728881836,
"logps/chosen": -0.4988403916358948,
"logps/rejected": -0.5690027475357056,
"loss": 1.5304,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.4988403916358948,
"rewards/margins": 0.07016237080097198,
"rewards/rejected": -0.5690027475357056,
"step": 80
},
{
"epoch": 0.18218352310783656,
"grad_norm": 21.78609949315303,
"learning_rate": 4.899211964056035e-07,
"logits/chosen": -2.338728427886963,
"logits/rejected": -2.331895112991333,
"logps/chosen": -0.5135633945465088,
"logps/rejected": -0.5660439729690552,
"loss": 1.5365,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5135633945465088,
"rewards/margins": 0.052480556070804596,
"rewards/rejected": -0.5660439729690552,
"step": 85
},
{
"epoch": 0.19290020093770932,
"grad_norm": 18.81785585703593,
"learning_rate": 4.871188860775643e-07,
"logits/chosen": -2.288607120513916,
"logits/rejected": -2.292832374572754,
"logps/chosen": -0.5013225674629211,
"logps/rejected": -0.5400410890579224,
"loss": 1.5369,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5013225674629211,
"rewards/margins": 0.03871860355138779,
"rewards/rejected": -0.5400410890579224,
"step": 90
},
{
"epoch": 0.20361687876758205,
"grad_norm": 18.284986731616815,
"learning_rate": 4.839833588729897e-07,
"logits/chosen": -2.33103609085083,
"logits/rejected": -2.3427205085754395,
"logps/chosen": -0.5798801779747009,
"logps/rejected": -0.5994023680686951,
"loss": 1.5235,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.5798801779747009,
"rewards/margins": 0.01952212117612362,
"rewards/rejected": -0.5994023680686951,
"step": 95
},
{
"epoch": 0.2143335565974548,
"grad_norm": 24.996484906718788,
"learning_rate": 4.80519021065031e-07,
"logits/chosen": -2.3483448028564453,
"logits/rejected": -2.3573968410491943,
"logps/chosen": -0.5851219296455383,
"logps/rejected": -0.6033455729484558,
"loss": 1.5464,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.5851219296455383,
"rewards/margins": 0.018223673105239868,
"rewards/rejected": -0.6033455729484558,
"step": 100
},
{
"epoch": 0.22505023442732752,
"grad_norm": 15.841869984614812,
"learning_rate": 4.767307409956528e-07,
"logits/chosen": -2.3424832820892334,
"logits/rejected": -2.342660903930664,
"logps/chosen": -0.5200175046920776,
"logps/rejected": -0.5671975612640381,
"loss": 1.5177,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5200175046920776,
"rewards/margins": 0.04718000069260597,
"rewards/rejected": -0.5671975612640381,
"step": 105
},
{
"epoch": 0.23576691225720026,
"grad_norm": 26.162342495475848,
"learning_rate": 4.726238422342805e-07,
"logits/chosen": -2.370981216430664,
"logits/rejected": -2.377699375152588,
"logps/chosen": -0.5119336247444153,
"logps/rejected": -0.5585616230964661,
"loss": 1.5421,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.5119336247444153,
"rewards/margins": 0.04662800580263138,
"rewards/rejected": -0.5585616230964661,
"step": 110
},
{
"epoch": 0.24648359008707302,
"grad_norm": 22.223934331525054,
"learning_rate": 4.6820409609673027e-07,
"logits/chosen": -2.445955276489258,
"logits/rejected": -2.439049005508423,
"logps/chosen": -0.507788360118866,
"logps/rejected": -0.5285546779632568,
"loss": 1.573,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.507788360118866,
"rewards/margins": 0.02076634392142296,
"rewards/rejected": -0.5285546779632568,
"step": 115
},
{
"epoch": 0.2572002679169457,
"grad_norm": 17.337249756046997,
"learning_rate": 4.6347771353493173e-07,
"logits/chosen": -2.507343053817749,
"logits/rejected": -2.4980244636535645,
"logps/chosen": -0.5571922063827515,
"logps/rejected": -0.5488871335983276,
"loss": 1.5546,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.5571922063827515,
"rewards/margins": -0.008305097930133343,
"rewards/rejected": -0.5488871335983276,
"step": 120
},
{
"epoch": 0.2679169457468185,
"grad_norm": 18.61479199807285,
"learning_rate": 4.5845133640884217e-07,
"logits/chosen": -2.496476173400879,
"logits/rejected": -2.5013513565063477,
"logps/chosen": -0.5598434209823608,
"logps/rejected": -0.5717306733131409,
"loss": 1.5472,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5598434209823608,
"rewards/margins": 0.01188725233078003,
"rewards/rejected": -0.5717306733131409,
"step": 125
},
{
"epoch": 0.27863362357669125,
"grad_norm": 24.242840152077708,
"learning_rate": 4.531320281528169e-07,
"logits/chosen": -2.577277660369873,
"logits/rejected": -2.5614867210388184,
"logps/chosen": -0.5257032513618469,
"logps/rejected": -0.5607699155807495,
"loss": 1.5206,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5257032513618469,
"rewards/margins": 0.03506666049361229,
"rewards/rejected": -0.5607699155807495,
"step": 130
},
{
"epoch": 0.289350301406564,
"grad_norm": 28.052518647744463,
"learning_rate": 4.4752726384955293e-07,
"logits/chosen": -2.5903048515319824,
"logits/rejected": -2.6012890338897705,
"logps/chosen": -0.5207422971725464,
"logps/rejected": -0.5546741485595703,
"loss": 1.5394,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5207422971725464,
"rewards/margins": 0.03393186628818512,
"rewards/rejected": -0.5546741485595703,
"step": 135
},
{
"epoch": 0.3000669792364367,
"grad_norm": 20.236382854852724,
"learning_rate": 4.4164491972555294e-07,
"logits/chosen": -2.6124188899993896,
"logits/rejected": -2.5999808311462402,
"logps/chosen": -0.5296235084533691,
"logps/rejected": -0.5590516924858093,
"loss": 1.5202,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.5296235084533691,
"rewards/margins": 0.02942819520831108,
"rewards/rejected": -0.5590516924858093,
"step": 140
},
{
"epoch": 0.31078365706630945,
"grad_norm": 18.380103954877008,
"learning_rate": 4.3549326208287393e-07,
"logits/chosen": -2.710564613342285,
"logits/rejected": -2.68730092048645,
"logps/chosen": -0.5258200764656067,
"logps/rejected": -0.5745820999145508,
"loss": 1.5158,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5258200764656067,
"rewards/margins": 0.0487620048224926,
"rewards/rejected": -0.5745820999145508,
"step": 145
},
{
"epoch": 0.3215003348961822,
"grad_norm": 19.20157418312748,
"learning_rate": 4.2908093568271187e-07,
"logits/chosen": -2.70224666595459,
"logits/rejected": -2.694885730743408,
"logps/chosen": -0.5079634189605713,
"logps/rejected": -0.5393082499504089,
"loss": 1.5455,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5079634189605713,
"rewards/margins": 0.03134483844041824,
"rewards/rejected": -0.5393082499504089,
"step": 150
},
{
"epoch": 0.3322170127260549,
"grad_norm": 17.41360540258233,
"learning_rate": 4.2241695159714843e-07,
"logits/chosen": -2.793240785598755,
"logits/rejected": -2.801339626312256,
"logps/chosen": -0.5376655459403992,
"logps/rejected": -0.5706537365913391,
"loss": 1.5278,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5376655459403992,
"rewards/margins": 0.03298817574977875,
"rewards/rejected": -0.5706537365913391,
"step": 155
},
{
"epoch": 0.34293369055592765,
"grad_norm": 21.649135642065247,
"learning_rate": 4.1551067454613074e-07,
"logits/chosen": -2.802827835083008,
"logits/rejected": -2.78732967376709,
"logps/chosen": -0.558406412601471,
"logps/rejected": -0.6224438548088074,
"loss": 1.5318,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.558406412601471,
"rewards/margins": 0.06403747946023941,
"rewards/rejected": -0.6224438548088074,
"step": 160
},
{
"epoch": 0.3536503683858004,
"grad_norm": 20.48435877400994,
"learning_rate": 4.0837180973747876e-07,
"logits/chosen": -2.9101483821868896,
"logits/rejected": -2.8818156719207764,
"logps/chosen": -0.5387953519821167,
"logps/rejected": -0.571646511554718,
"loss": 1.5233,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5387953519821167,
"rewards/margins": 0.032851073890924454,
"rewards/rejected": -0.571646511554718,
"step": 165
},
{
"epoch": 0.3643670462156731,
"grad_norm": 17.62876339336723,
"learning_rate": 4.0101038922841455e-07,
"logits/chosen": -2.9059550762176514,
"logits/rejected": -2.9104220867156982,
"logps/chosen": -0.5236160159111023,
"logps/rejected": -0.5999457836151123,
"loss": 1.5112,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5236160159111023,
"rewards/margins": 0.07632973045110703,
"rewards/rejected": -0.5999457836151123,
"step": 170
},
{
"epoch": 0.3750837240455459,
"grad_norm": 20.453674701254528,
"learning_rate": 3.934367578277783e-07,
"logits/chosen": -2.9715123176574707,
"logits/rejected": -2.953988790512085,
"logps/chosen": -0.5574652552604675,
"logps/rejected": -0.5864735841751099,
"loss": 1.5348,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5574652552604675,
"rewards/margins": 0.029008299112319946,
"rewards/rejected": -0.5864735841751099,
"step": 175
},
{
"epoch": 0.38580040187541864,
"grad_norm": 22.621519510630307,
"learning_rate": 3.856615585587434e-07,
"logits/chosen": -3.067772388458252,
"logits/rejected": -3.0693717002868652,
"logps/chosen": -0.5790558457374573,
"logps/rejected": -0.6112508177757263,
"loss": 1.545,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5790558457374573,
"rewards/margins": 0.03219496086239815,
"rewards/rejected": -0.6112508177757263,
"step": 180
},
{
"epoch": 0.3965170797052914,
"grad_norm": 27.765398609724794,
"learning_rate": 3.776957177024581e-07,
"logits/chosen": -3.070247173309326,
"logits/rejected": -3.0919833183288574,
"logps/chosen": -0.5750337839126587,
"logps/rejected": -0.6032044887542725,
"loss": 1.5218,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.5750337839126587,
"rewards/margins": 0.028170805424451828,
"rewards/rejected": -0.6032044887542725,
"step": 185
},
{
"epoch": 0.4072337575351641,
"grad_norm": 19.099968753214323,
"learning_rate": 3.69550429443633e-07,
"logits/chosen": -2.944047451019287,
"logits/rejected": -2.959052324295044,
"logps/chosen": -0.5526596307754517,
"logps/rejected": -0.7156730890274048,
"loss": 1.5268,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.5526596307754517,
"rewards/margins": 0.16301342844963074,
"rewards/rejected": -0.7156730890274048,
"step": 190
},
{
"epoch": 0.41795043536503684,
"grad_norm": 24.556600066756868,
"learning_rate": 3.612371401396502e-07,
"logits/chosen": -3.195617198944092,
"logits/rejected": -3.168480396270752,
"logps/chosen": -0.515397846698761,
"logps/rejected": -0.5484700798988342,
"loss": 1.5305,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.515397846698761,
"rewards/margins": 0.03307227045297623,
"rewards/rejected": -0.5484700798988342,
"step": 195
},
{
"epoch": 0.4286671131949096,
"grad_norm": 20.445564921711252,
"learning_rate": 3.527675322353011e-07,
"logits/chosen": -3.071345567703247,
"logits/rejected": -3.0912113189697266,
"logps/chosen": -0.4999586045742035,
"logps/rejected": -0.5286678075790405,
"loss": 1.5201,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.4999586045742035,
"rewards/margins": 0.028709203004837036,
"rewards/rejected": -0.5286678075790405,
"step": 200
},
{
"epoch": 0.4393837910247823,
"grad_norm": 28.017877985161014,
"learning_rate": 3.4415350784575693e-07,
"logits/chosen": -3.2119345664978027,
"logits/rejected": -3.2098147869110107,
"logps/chosen": -0.48476672172546387,
"logps/rejected": -0.5175997018814087,
"loss": 1.5335,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.48476672172546387,
"rewards/margins": 0.032833002507686615,
"rewards/rejected": -0.5175997018814087,
"step": 205
},
{
"epoch": 0.45010046885465504,
"grad_norm": 20.446366205479023,
"learning_rate": 3.3540717203084224e-07,
"logits/chosen": -3.2167842388153076,
"logits/rejected": -3.209538698196411,
"logps/chosen": -0.5124386548995972,
"logps/rejected": -0.5561140179634094,
"loss": 1.5035,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5124386548995972,
"rewards/margins": 0.04367532581090927,
"rewards/rejected": -0.5561140179634094,
"step": 210
},
{
"epoch": 0.4608171466845278,
"grad_norm": 20.249310261680897,
"learning_rate": 3.265408157841153e-07,
"logits/chosen": -3.2260100841522217,
"logits/rejected": -3.2222793102264404,
"logps/chosen": -0.5612623691558838,
"logps/rejected": -0.6017634272575378,
"loss": 1.4952,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5612623691558838,
"rewards/margins": 0.040501099079847336,
"rewards/rejected": -0.6017634272575378,
"step": 215
},
{
"epoch": 0.4715338245144005,
"grad_norm": 28.781450630078922,
"learning_rate": 3.1756689876066187e-07,
"logits/chosen": -3.26216459274292,
"logits/rejected": -3.2409801483154297,
"logps/chosen": -0.5308222770690918,
"logps/rejected": -0.609308660030365,
"loss": 1.5377,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5308222770690918,
"rewards/margins": 0.078486368060112,
"rewards/rejected": -0.609308660030365,
"step": 220
},
{
"epoch": 0.4822505023442733,
"grad_norm": 20.9210927889319,
"learning_rate": 3.0849803176787184e-07,
"logits/chosen": -3.3171279430389404,
"logits/rejected": -3.3153088092803955,
"logps/chosen": -0.5533649921417236,
"logps/rejected": -0.6165200471878052,
"loss": 1.5082,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5533649921417236,
"rewards/margins": 0.06315504014492035,
"rewards/rejected": -0.6165200471878052,
"step": 225
},
{
"epoch": 0.49296718017414604,
"grad_norm": 26.953714991186636,
"learning_rate": 2.993469590438065e-07,
"logits/chosen": -3.2146403789520264,
"logits/rejected": -3.2017455101013184,
"logps/chosen": -0.539149284362793,
"logps/rejected": -0.5770600438117981,
"loss": 1.4968,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.539149284362793,
"rewards/margins": 0.03791072219610214,
"rewards/rejected": -0.5770600438117981,
"step": 230
},
{
"epoch": 0.5036838580040187,
"grad_norm": 23.063142701753478,
"learning_rate": 2.901265403480597e-07,
"logits/chosen": -3.2051620483398438,
"logits/rejected": -3.1935036182403564,
"logps/chosen": -0.5151172876358032,
"logps/rejected": -0.6125957369804382,
"loss": 1.508,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5151172876358032,
"rewards/margins": 0.09747843444347382,
"rewards/rejected": -0.6125957369804382,
"step": 235
},
{
"epoch": 0.5144005358338914,
"grad_norm": 28.679041181758652,
"learning_rate": 2.808497328902782e-07,
"logits/chosen": -3.152073621749878,
"logits/rejected": -3.15959095954895,
"logps/chosen": -0.5507190823554993,
"logps/rejected": -0.6257890462875366,
"loss": 1.5193,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5507190823554993,
"rewards/margins": 0.07506996393203735,
"rewards/rejected": -0.6257890462875366,
"step": 240
},
{
"epoch": 0.5251172136637642,
"grad_norm": 24.21410810508721,
"learning_rate": 2.715295731217396e-07,
"logits/chosen": -3.1899075508117676,
"logits/rejected": -3.1687207221984863,
"logps/chosen": -0.5879708528518677,
"logps/rejected": -0.6203640103340149,
"loss": 1.5033,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.5879708528518677,
"rewards/margins": 0.032393116503953934,
"rewards/rejected": -0.6203640103340149,
"step": 245
},
{
"epoch": 0.535833891493637,
"grad_norm": 24.944015170343583,
"learning_rate": 2.621791584155725e-07,
"logits/chosen": -3.186570167541504,
"logits/rejected": -3.1771557331085205,
"logps/chosen": -0.610142171382904,
"logps/rejected": -0.604741096496582,
"loss": 1.5341,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.610142171382904,
"rewards/margins": -0.005401057191193104,
"rewards/rejected": -0.604741096496582,
"step": 250
},
{
"epoch": 0.5465505693235098,
"grad_norm": 22.133595287604955,
"learning_rate": 2.52811628661367e-07,
"logits/chosen": -3.2132976055145264,
"logits/rejected": -3.21321439743042,
"logps/chosen": -0.5207762122154236,
"logps/rejected": -0.5928905606269836,
"loss": 1.4997,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5207762122154236,
"rewards/margins": 0.07211437076330185,
"rewards/rejected": -0.5928905606269836,
"step": 255
},
{
"epoch": 0.5572672471533825,
"grad_norm": 21.31026102730006,
"learning_rate": 2.434401478000353e-07,
"logits/chosen": -3.141242504119873,
"logits/rejected": -3.1591105461120605,
"logps/chosen": -0.526002049446106,
"logps/rejected": -0.6240478754043579,
"loss": 1.5259,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.526002049446106,
"rewards/margins": 0.09804578125476837,
"rewards/rejected": -0.6240478754043579,
"step": 260
},
{
"epoch": 0.5679839249832552,
"grad_norm": 25.465212058173062,
"learning_rate": 2.340778853248759e-07,
"logits/chosen": -3.1647770404815674,
"logits/rejected": -3.161611557006836,
"logps/chosen": -0.5447332859039307,
"logps/rejected": -0.6431838870048523,
"loss": 1.496,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5447332859039307,
"rewards/margins": 0.098450668156147,
"rewards/rejected": -0.6431838870048523,
"step": 265
},
{
"epoch": 0.578700602813128,
"grad_norm": 29.79126766513931,
"learning_rate": 2.247379977748339e-07,
"logits/chosen": -3.1814353466033936,
"logits/rejected": -3.1744346618652344,
"logps/chosen": -0.5134252905845642,
"logps/rejected": -0.5546069145202637,
"loss": 1.5172,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5134252905845642,
"rewards/margins": 0.04118159040808678,
"rewards/rejected": -0.5546069145202637,
"step": 270
},
{
"epoch": 0.5894172806430007,
"grad_norm": 20.911132725985905,
"learning_rate": 2.1543361024596573e-07,
"logits/chosen": -3.2813096046447754,
"logits/rejected": -3.2904067039489746,
"logps/chosen": -0.478159099817276,
"logps/rejected": -0.5237951278686523,
"loss": 1.52,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.478159099817276,
"rewards/margins": 0.045636050403118134,
"rewards/rejected": -0.5237951278686523,
"step": 275
},
{
"epoch": 0.6001339584728734,
"grad_norm": 22.184303638777845,
"learning_rate": 2.0617779794709082e-07,
"logits/chosen": -3.227949857711792,
"logits/rejected": -3.199709415435791,
"logps/chosen": -0.5574294924736023,
"logps/rejected": -0.6131659746170044,
"loss": 1.5077,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5574294924736023,
"rewards/margins": 0.05573650076985359,
"rewards/rejected": -0.6131659746170044,
"step": 280
},
{
"epoch": 0.6108506363027462,
"grad_norm": 23.16627725406288,
"learning_rate": 1.9698356782554687e-07,
"logits/chosen": -3.2686588764190674,
"logits/rejected": -3.27409291267395,
"logps/chosen": -0.6056652069091797,
"logps/rejected": -0.6142518520355225,
"loss": 1.529,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.6056652069091797,
"rewards/margins": 0.008586685173213482,
"rewards/rejected": -0.6142518520355225,
"step": 285
},
{
"epoch": 0.6215673141326189,
"grad_norm": 43.63471821872903,
"learning_rate": 1.878638402888727e-07,
"logits/chosen": -3.3371078968048096,
"logits/rejected": -3.3204472064971924,
"logps/chosen": -0.5225822925567627,
"logps/rejected": -0.5866876840591431,
"loss": 1.5024,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5225822925567627,
"rewards/margins": 0.06410545110702515,
"rewards/rejected": -0.5866876840591431,
"step": 290
},
{
"epoch": 0.6322839919624916,
"grad_norm": 25.701019654379248,
"learning_rate": 1.7883143104810223e-07,
"logits/chosen": -3.2457165718078613,
"logits/rejected": -3.2686493396759033,
"logps/chosen": -0.5675607919692993,
"logps/rejected": -0.6349104642868042,
"loss": 1.4993,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5675607919692993,
"rewards/margins": 0.06734969466924667,
"rewards/rejected": -0.6349104642868042,
"step": 295
},
{
"epoch": 0.6430006697923644,
"grad_norm": 23.619470082480703,
"learning_rate": 1.6989903310818548e-07,
"logits/chosen": -3.3820648193359375,
"logits/rejected": -3.3857345581054688,
"logps/chosen": -0.5343354344367981,
"logps/rejected": -0.6022351980209351,
"loss": 1.4951,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.5343354344367981,
"rewards/margins": 0.06789974123239517,
"rewards/rejected": -0.6022351980209351,
"step": 300
},
{
"epoch": 0.6537173476222371,
"grad_norm": 22.753889982587108,
"learning_rate": 1.610791989308466e-07,
"logits/chosen": -3.3234715461730957,
"logits/rejected": -3.2950637340545654,
"logps/chosen": -0.5346772074699402,
"logps/rejected": -0.5783089995384216,
"loss": 1.5037,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5346772074699402,
"rewards/margins": 0.04363187402486801,
"rewards/rejected": -0.5783089995384216,
"step": 305
},
{
"epoch": 0.6644340254521098,
"grad_norm": 29.970511411150166,
"learning_rate": 1.523843227949418e-07,
"logits/chosen": -3.3544983863830566,
"logits/rejected": -3.355292558670044,
"logps/chosen": -0.5303682088851929,
"logps/rejected": -0.5744400024414062,
"loss": 1.4936,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5303682088851929,
"rewards/margins": 0.04407178610563278,
"rewards/rejected": -0.5744400024414062,
"step": 310
},
{
"epoch": 0.6751507032819826,
"grad_norm": 19.95585897697304,
"learning_rate": 1.4382662337910867e-07,
"logits/chosen": -3.291701078414917,
"logits/rejected": -3.3025760650634766,
"logps/chosen": -0.5352946519851685,
"logps/rejected": -0.5887657403945923,
"loss": 1.502,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5352946519851685,
"rewards/margins": 0.053471069782972336,
"rewards/rejected": -0.5887657403945923,
"step": 315
},
{
"epoch": 0.6858673811118553,
"grad_norm": 22.149800280030288,
"learning_rate": 1.3541812659118104e-07,
"logits/chosen": -3.334624767303467,
"logits/rejected": -3.306361675262451,
"logps/chosen": -0.5542501211166382,
"logps/rejected": -0.5587230920791626,
"loss": 1.524,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5542501211166382,
"rewards/margins": 0.004472990520298481,
"rewards/rejected": -0.5587230920791626,
"step": 320
},
{
"epoch": 0.696584058941728,
"grad_norm": 25.604347826249015,
"learning_rate": 1.2717064866850047e-07,
"logits/chosen": -3.3316922187805176,
"logits/rejected": -3.342952251434326,
"logps/chosen": -0.5008823275566101,
"logps/rejected": -0.5843838453292847,
"loss": 1.4797,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.5008823275566101,
"rewards/margins": 0.08350151777267456,
"rewards/rejected": -0.5843838453292847,
"step": 325
},
{
"epoch": 0.7073007367716008,
"grad_norm": 24.090724072052794,
"learning_rate": 1.1909577957287117e-07,
"logits/chosen": -3.3231868743896484,
"logits/rejected": -3.2961513996124268,
"logps/chosen": -0.5285135507583618,
"logps/rejected": -0.582491397857666,
"loss": 1.4972,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5285135507583618,
"rewards/margins": 0.05397792533040047,
"rewards/rejected": -0.582491397857666,
"step": 330
},
{
"epoch": 0.7180174146014735,
"grad_norm": 36.87118130194036,
"learning_rate": 1.1120486670349441e-07,
"logits/chosen": -3.4078221321105957,
"logits/rejected": -3.3980777263641357,
"logps/chosen": -0.5307896733283997,
"logps/rejected": -0.6150525808334351,
"loss": 1.5019,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.5307896733283997,
"rewards/margins": 0.08426298201084137,
"rewards/rejected": -0.6150525808334351,
"step": 335
},
{
"epoch": 0.7287340924313462,
"grad_norm": 25.52897276938915,
"learning_rate": 1.0350899895076948e-07,
"logits/chosen": -3.3364510536193848,
"logits/rejected": -3.321132183074951,
"logps/chosen": -0.5735017657279968,
"logps/rejected": -0.6429983973503113,
"loss": 1.521,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5735017657279968,
"rewards/margins": 0.06949658691883087,
"rewards/rejected": -0.6429983973503113,
"step": 340
},
{
"epoch": 0.739450770261219,
"grad_norm": 20.331652657630677,
"learning_rate": 9.601899111337069e-08,
"logits/chosen": -3.367867946624756,
"logits/rejected": -3.3775951862335205,
"logps/chosen": -0.5771843194961548,
"logps/rejected": -0.6704375147819519,
"loss": 1.496,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5771843194961548,
"rewards/margins": 0.09325330704450607,
"rewards/rejected": -0.6704375147819519,
"step": 345
},
{
"epoch": 0.7501674480910918,
"grad_norm": 22.274114959082404,
"learning_rate": 8.874536870049765e-08,
"logits/chosen": -3.3537139892578125,
"logits/rejected": -3.378490924835205,
"logps/chosen": -0.5365322828292847,
"logps/rejected": -0.6033391952514648,
"loss": 1.4971,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.5365322828292847,
"rewards/margins": 0.06680694222450256,
"rewards/rejected": -0.6033391952514648,
"step": 350
},
{
"epoch": 0.7608841259209645,
"grad_norm": 24.177285206790465,
"learning_rate": 8.169835314065663e-08,
"logits/chosen": -3.3511605262756348,
"logits/rejected": -3.3401741981506348,
"logps/chosen": -0.5807870626449585,
"logps/rejected": -0.6009158492088318,
"loss": 1.499,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.5807870626449585,
"rewards/margins": 0.020128777250647545,
"rewards/rejected": -0.6009158492088318,
"step": 355
},
{
"epoch": 0.7716008037508373,
"grad_norm": 22.05484832098868,
"learning_rate": 7.488784741775816e-08,
"logits/chosen": -3.3435721397399902,
"logits/rejected": -3.33024525642395,
"logps/chosen": -0.5081290006637573,
"logps/rejected": -0.5656172037124634,
"loss": 1.5079,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5081290006637573,
"rewards/margins": 0.057488154619932175,
"rewards/rejected": -0.5656172037124634,
"step": 360
},
{
"epoch": 0.78231748158071,
"grad_norm": 24.600234649150252,
"learning_rate": 6.832342215471715e-08,
"logits/chosen": -3.3981361389160156,
"logits/rejected": -3.39165997505188,
"logps/chosen": -0.5507721304893494,
"logps/rejected": -0.6153653860092163,
"loss": 1.5147,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.5507721304893494,
"rewards/margins": 0.06459329277276993,
"rewards/rejected": -0.6153653860092163,
"step": 365
},
{
"epoch": 0.7930341594105828,
"grad_norm": 31.85175227986763,
"learning_rate": 6.201430216410985e-08,
"logits/chosen": -3.391742706298828,
"logits/rejected": -3.413292646408081,
"logps/chosen": -0.5624402761459351,
"logps/rejected": -0.6359676122665405,
"loss": 1.5149,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5624402761459351,
"rewards/margins": 0.07352729141712189,
"rewards/rejected": -0.6359676122665405,
"step": 370
},
{
"epoch": 0.8037508372404555,
"grad_norm": 23.503804219453023,
"learning_rate": 5.596935348479029e-08,
"logits/chosen": -3.3563759326934814,
"logits/rejected": -3.338204860687256,
"logps/chosen": -0.6166857481002808,
"logps/rejected": -0.700817883014679,
"loss": 1.5014,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6166857481002808,
"rewards/margins": 0.08413223922252655,
"rewards/rejected": -0.700817883014679,
"step": 375
},
{
"epoch": 0.8144675150703282,
"grad_norm": 26.081587652629086,
"learning_rate": 5.01970709226808e-08,
"logits/chosen": -3.443342924118042,
"logits/rejected": -3.4111335277557373,
"logps/chosen": -0.5469595193862915,
"logps/rejected": -0.6322815418243408,
"loss": 1.5083,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5469595193862915,
"rewards/margins": 0.08532204478979111,
"rewards/rejected": -0.6322815418243408,
"step": 380
},
{
"epoch": 0.825184192900201,
"grad_norm": 24.73716667501714,
"learning_rate": 4.4705566113246635e-08,
"logits/chosen": -3.349184513092041,
"logits/rejected": -3.359894275665283,
"logps/chosen": -0.5703479647636414,
"logps/rejected": -0.6489995718002319,
"loss": 1.4984,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5703479647636414,
"rewards/margins": 0.07865166664123535,
"rewards/rejected": -0.6489995718002319,
"step": 385
},
{
"epoch": 0.8359008707300737,
"grad_norm": 24.765231706983972,
"learning_rate": 3.950255612243042e-08,
"logits/chosen": -3.3900883197784424,
"logits/rejected": -3.3493354320526123,
"logps/chosen": -0.6082748174667358,
"logps/rejected": -0.6323951482772827,
"loss": 1.5319,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.6082748174667358,
"rewards/margins": 0.024120373651385307,
"rewards/rejected": -0.6323951482772827,
"step": 390
},
{
"epoch": 0.8466175485599464,
"grad_norm": 23.65912516379117,
"learning_rate": 3.459535260206384e-08,
"logits/chosen": -3.3917059898376465,
"logits/rejected": -3.3759236335754395,
"logps/chosen": -0.5519801378250122,
"logps/rejected": -0.5884855389595032,
"loss": 1.5143,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5519801378250122,
"rewards/margins": 0.036505263298749924,
"rewards/rejected": -0.5884855389595032,
"step": 395
},
{
"epoch": 0.8573342263898192,
"grad_norm": 28.498252560499747,
"learning_rate": 2.999085151499764e-08,
"logits/chosen": -3.3536014556884766,
"logits/rejected": -3.322979688644409,
"logps/chosen": -0.5050632953643799,
"logps/rejected": -0.6183533072471619,
"loss": 1.4908,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.5050632953643799,
"rewards/margins": 0.11329002678394318,
"rewards/rejected": -0.6183533072471619,
"step": 400
},
{
"epoch": 0.8573342263898192,
"eval_logits/chosen": -3.4685490131378174,
"eval_logits/rejected": -3.4632692337036133,
"eval_logps/chosen": -0.5601765513420105,
"eval_logps/rejected": -0.6162815093994141,
"eval_loss": 1.522051215171814,
"eval_rewards/accuracies": 0.5797872543334961,
"eval_rewards/chosen": -0.5601765513420105,
"eval_rewards/margins": 0.05610497295856476,
"eval_rewards/rejected": -0.6162815093994141,
"eval_runtime": 433.0685,
"eval_samples_per_second": 6.913,
"eval_steps_per_second": 0.434,
"step": 400
},
{
"epoch": 0.8680509042196919,
"grad_norm": 23.705058603217022,
"learning_rate": 2.5695523444387747e-08,
"logits/chosen": -3.364602565765381,
"logits/rejected": -3.350884199142456,
"logps/chosen": -0.5609386563301086,
"logps/rejected": -0.605690598487854,
"loss": 1.5111,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.5609386563301086,
"rewards/margins": 0.044751983135938644,
"rewards/rejected": -0.605690598487854,
"step": 405
},
{
"epoch": 0.8787675820495646,
"grad_norm": 22.37221809399767,
"learning_rate": 2.1715404500756875e-08,
"logits/chosen": -3.412458896636963,
"logits/rejected": -3.3824081420898438,
"logps/chosen": -0.5534378290176392,
"logps/rejected": -0.5746613144874573,
"loss": 1.4932,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.5534378290176392,
"rewards/margins": 0.021223513409495354,
"rewards/rejected": -0.5746613144874573,
"step": 410
},
{
"epoch": 0.8894842598794374,
"grad_norm": 26.32210656993751,
"learning_rate": 1.8056087839608546e-08,
"logits/chosen": -3.332536220550537,
"logits/rejected": -3.3019680976867676,
"logps/chosen": -0.5494991540908813,
"logps/rejected": -0.6182190179824829,
"loss": 1.4885,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5494991540908813,
"rewards/margins": 0.06871981918811798,
"rewards/rejected": -0.6182190179824829,
"step": 415
},
{
"epoch": 0.9002009377093101,
"grad_norm": 27.43028943315468,
"learning_rate": 1.4722715801513935e-08,
"logits/chosen": -3.3295421600341797,
"logits/rejected": -3.336806058883667,
"logps/chosen": -0.5279422998428345,
"logps/rejected": -0.5645946860313416,
"loss": 1.5217,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -0.5279422998428345,
"rewards/margins": 0.03665238991379738,
"rewards/rejected": -0.5645946860313416,
"step": 420
},
{
"epoch": 0.9109176155391828,
"grad_norm": 25.425067386156872,
"learning_rate": 1.1719972685717395e-08,
"logits/chosen": -3.337613582611084,
"logits/rejected": -3.3247127532958984,
"logps/chosen": -0.5188074111938477,
"logps/rejected": -0.6191619634628296,
"loss": 1.4926,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.5188074111938477,
"rewards/margins": 0.10035452991724014,
"rewards/rejected": -0.6191619634628296,
"step": 425
},
{
"epoch": 0.9216342933690556,
"grad_norm": 35.62756758017028,
"learning_rate": 9.05207816741485e-09,
"logits/chosen": -3.3888893127441406,
"logits/rejected": -3.4012131690979004,
"logps/chosen": -0.6197360754013062,
"logps/rejected": -0.6971450448036194,
"loss": 1.5011,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6197360754013062,
"rewards/margins": 0.07740898430347443,
"rewards/rejected": -0.6971450448036194,
"step": 430
},
{
"epoch": 0.9323509711989283,
"grad_norm": 27.22699922034511,
"learning_rate": 6.722781367956481e-09,
"logits/chosen": -3.455385208129883,
"logits/rejected": -3.4597179889678955,
"logps/chosen": -0.5144860744476318,
"logps/rejected": -0.6305041909217834,
"loss": 1.4974,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5144860744476318,
"rewards/margins": 0.1160181388258934,
"rewards/rejected": -0.6305041909217834,
"step": 435
},
{
"epoch": 0.943067649028801,
"grad_norm": 23.43687138581986,
"learning_rate": 4.735355586305861e-09,
"logits/chosen": -3.2910914421081543,
"logits/rejected": -3.288029193878174,
"logps/chosen": -0.5328935980796814,
"logps/rejected": -0.6186143755912781,
"loss": 1.4731,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.5328935980796814,
"rewards/margins": 0.08572079241275787,
"rewards/rejected": -0.6186143755912781,
"step": 440
},
{
"epoch": 0.9537843268586738,
"grad_norm": 27.25503683317219,
"learning_rate": 3.0925936991598454e-09,
"logits/chosen": -3.320718765258789,
"logits/rejected": -3.319967746734619,
"logps/chosen": -0.5472747683525085,
"logps/rejected": -0.6223639845848083,
"loss": 1.493,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5472747683525085,
"rewards/margins": 0.07508920133113861,
"rewards/rejected": -0.6223639845848083,
"step": 445
},
{
"epoch": 0.9645010046885466,
"grad_norm": 23.669876406776968,
"learning_rate": 1.7968042361930225e-09,
"logits/chosen": -3.354473114013672,
"logits/rejected": -3.3409972190856934,
"logps/chosen": -0.5448306202888489,
"logps/rejected": -0.6130356788635254,
"loss": 1.4876,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5448306202888489,
"rewards/margins": 0.06820502877235413,
"rewards/rejected": -0.6130356788635254,
"step": 450
},
{
"epoch": 0.9752176825184193,
"grad_norm": 27.737663302659318,
"learning_rate": 8.498081359423759e-10,
"logits/chosen": -3.331641435623169,
"logits/rejected": -3.3293700218200684,
"logps/chosen": -0.5374922156333923,
"logps/rejected": -0.6145657300949097,
"loss": 1.4953,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5374922156333923,
"rewards/margins": 0.07707355916500092,
"rewards/rejected": -0.6145657300949097,
"step": 455
},
{
"epoch": 0.9859343603482921,
"grad_norm": 29.309243600698153,
"learning_rate": 2.5293618689056774e-10,
"logits/chosen": -3.3501858711242676,
"logits/rejected": -3.3541958332061768,
"logps/chosen": -0.5475316643714905,
"logps/rejected": -0.6366786360740662,
"loss": 1.5039,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.5475316643714905,
"rewards/margins": 0.08914700895547867,
"rewards/rejected": -0.6366786360740662,
"step": 460
},
{
"epoch": 0.9966510381781648,
"grad_norm": 21.403219882354602,
"learning_rate": 7.02715734424153e-12,
"logits/chosen": -3.3826191425323486,
"logits/rejected": -3.36454701423645,
"logps/chosen": -0.5860726833343506,
"logps/rejected": -0.6269819736480713,
"loss": 1.5076,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5860726833343506,
"rewards/margins": 0.040909312665462494,
"rewards/rejected": -0.6269819736480713,
"step": 465
},
{
"epoch": 0.9987943737441393,
"step": 466,
"total_flos": 0.0,
"train_loss": 1.5188358733070766,
"train_runtime": 19352.2078,
"train_samples_per_second": 3.086,
"train_steps_per_second": 0.024
}
],
"logging_steps": 5,
"max_steps": 466,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}