|
{ |
|
"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 |
|
} |
|
|