|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 2.985781990521327, |
|
"eval_steps": 500, |
|
"global_step": 315, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.09, |
|
"grad_norm": 51.45086593171862, |
|
"learning_rate": 5e-08, |
|
"log_odds_chosen": 0.23327143490314484, |
|
"log_odds_ratio": -0.7173982262611389, |
|
"logits/chosen": -2.9019644260406494, |
|
"logits/rejected": -2.911267042160034, |
|
"logps/chosen": -1.2478530406951904, |
|
"logps/rejected": -1.4603310823440552, |
|
"loss": 1.1528, |
|
"nll_loss": 1.110867977142334, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.06239265203475952, |
|
"rewards/margins": 0.010623900219798088, |
|
"rewards/rejected": -0.07301654666662216, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"grad_norm": 43.38597658388667, |
|
"learning_rate": 1e-07, |
|
"log_odds_chosen": 0.025622714310884476, |
|
"log_odds_ratio": -0.803138256072998, |
|
"logits/chosen": -2.957723617553711, |
|
"logits/rejected": -2.9606709480285645, |
|
"logps/chosen": -1.2851994037628174, |
|
"logps/rejected": -1.2747653722763062, |
|
"loss": 1.109, |
|
"nll_loss": 1.0504928827285767, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.06425996124744415, |
|
"rewards/margins": -0.0005216972786001861, |
|
"rewards/rejected": -0.06373827159404755, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"grad_norm": 18.07276834258672, |
|
"learning_rate": 1.5e-07, |
|
"log_odds_chosen": 0.33495259284973145, |
|
"log_odds_ratio": -0.6577204465866089, |
|
"logits/chosen": -2.936415195465088, |
|
"logits/rejected": -2.939385175704956, |
|
"logps/chosen": -1.0753445625305176, |
|
"logps/rejected": -1.3320289850234985, |
|
"loss": 1.0256, |
|
"nll_loss": 0.9573599696159363, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.053767234086990356, |
|
"rewards/margins": 0.012834218330681324, |
|
"rewards/rejected": -0.06660144031047821, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"grad_norm": 14.138346458866787, |
|
"learning_rate": 2e-07, |
|
"log_odds_chosen": 0.06818895787000656, |
|
"log_odds_ratio": -0.7556289434432983, |
|
"logits/chosen": -2.979851484298706, |
|
"logits/rejected": -2.9792511463165283, |
|
"logps/chosen": -1.0226948261260986, |
|
"logps/rejected": -1.0867162942886353, |
|
"loss": 0.9841, |
|
"nll_loss": 0.9785293340682983, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.05113474652171135, |
|
"rewards/margins": 0.003201070474460721, |
|
"rewards/rejected": -0.0543358139693737, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"grad_norm": 19.142475208524697, |
|
"learning_rate": 2.5e-07, |
|
"log_odds_chosen": 0.2577429711818695, |
|
"log_odds_ratio": -0.6650774478912354, |
|
"logits/chosen": -3.0103330612182617, |
|
"logits/rejected": -3.0095608234405518, |
|
"logps/chosen": -0.9924309849739075, |
|
"logps/rejected": -1.1665266752243042, |
|
"loss": 0.9833, |
|
"nll_loss": 0.965359091758728, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.04962155967950821, |
|
"rewards/margins": 0.008704781532287598, |
|
"rewards/rejected": -0.05832633376121521, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"grad_norm": 9.00103625601085, |
|
"learning_rate": 3e-07, |
|
"log_odds_chosen": 0.2601451277732849, |
|
"log_odds_ratio": -0.6529109477996826, |
|
"logits/chosen": -3.0405919551849365, |
|
"logits/rejected": -3.0266852378845215, |
|
"logps/chosen": -0.8860586881637573, |
|
"logps/rejected": -1.0516645908355713, |
|
"loss": 0.967, |
|
"nll_loss": 0.8791006207466125, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.044302936643362045, |
|
"rewards/margins": 0.008280294016003609, |
|
"rewards/rejected": -0.0525832362473011, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"grad_norm": 8.208565662624949, |
|
"learning_rate": 3.5e-07, |
|
"log_odds_chosen": 0.24647900462150574, |
|
"log_odds_ratio": -0.6602240800857544, |
|
"logits/chosen": -2.9646213054656982, |
|
"logits/rejected": -2.9692559242248535, |
|
"logps/chosen": -0.8851203918457031, |
|
"logps/rejected": -1.04824697971344, |
|
"loss": 0.9521, |
|
"nll_loss": 0.8758776783943176, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.044256024062633514, |
|
"rewards/margins": 0.008156323805451393, |
|
"rewards/rejected": -0.05241234228014946, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"grad_norm": 9.949021333520767, |
|
"learning_rate": 4e-07, |
|
"log_odds_chosen": 0.3642485439777374, |
|
"log_odds_ratio": -0.6296299695968628, |
|
"logits/chosen": -2.9997572898864746, |
|
"logits/rejected": -2.9932103157043457, |
|
"logps/chosen": -0.8595079183578491, |
|
"logps/rejected": -1.0957872867584229, |
|
"loss": 0.8992, |
|
"nll_loss": 0.8687052726745605, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.042975399643182755, |
|
"rewards/margins": 0.011813966557383537, |
|
"rewards/rejected": -0.05478937178850174, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"grad_norm": 11.826771567950217, |
|
"learning_rate": 4.5e-07, |
|
"log_odds_chosen": 0.13775679469108582, |
|
"log_odds_ratio": -0.7174943685531616, |
|
"logits/chosen": -2.9466898441314697, |
|
"logits/rejected": -2.9638590812683105, |
|
"logps/chosen": -0.8956987261772156, |
|
"logps/rejected": -0.9907738566398621, |
|
"loss": 0.903, |
|
"nll_loss": 0.8976006507873535, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.04478494077920914, |
|
"rewards/margins": 0.0047537581995129585, |
|
"rewards/rejected": -0.04953869432210922, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"grad_norm": 7.898257944088662, |
|
"learning_rate": 5e-07, |
|
"log_odds_chosen": 0.19350430369377136, |
|
"log_odds_ratio": -0.6868109703063965, |
|
"logits/chosen": -2.932610034942627, |
|
"logits/rejected": -2.9326844215393066, |
|
"logps/chosen": -0.8639435768127441, |
|
"logps/rejected": -0.9883400797843933, |
|
"loss": 0.9159, |
|
"nll_loss": 0.8739891052246094, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.04319717735052109, |
|
"rewards/margins": 0.0062198275700211525, |
|
"rewards/rejected": -0.049417007714509964, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_log_odds_chosen": 0.3023655116558075, |
|
"eval_log_odds_ratio": -0.6428868770599365, |
|
"eval_logits/chosen": -2.887465238571167, |
|
"eval_logits/rejected": -2.8930952548980713, |
|
"eval_logps/chosen": -0.8412830829620361, |
|
"eval_logps/rejected": -0.9974649548530579, |
|
"eval_loss": 0.8794215321540833, |
|
"eval_nll_loss": 0.8561177253723145, |
|
"eval_rewards/accuracies": 0.6302083134651184, |
|
"eval_rewards/chosen": -0.042064156383275986, |
|
"eval_rewards/margins": 0.007809097412973642, |
|
"eval_rewards/rejected": -0.04987325146794319, |
|
"eval_runtime": 32.7887, |
|
"eval_samples_per_second": 22.874, |
|
"eval_steps_per_second": 0.732, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 1.04, |
|
"grad_norm": 7.858898751829365, |
|
"learning_rate": 4.973358420187775e-07, |
|
"log_odds_chosen": 0.2784392237663269, |
|
"log_odds_ratio": -0.6759603023529053, |
|
"logits/chosen": -2.898139715194702, |
|
"logits/rejected": -2.888554096221924, |
|
"logps/chosen": -0.8769194483757019, |
|
"logps/rejected": -1.0477098226547241, |
|
"loss": 0.9267, |
|
"nll_loss": 0.9084149599075317, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -0.043845973908901215, |
|
"rewards/margins": 0.00853952020406723, |
|
"rewards/rejected": -0.05238549783825874, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 1.14, |
|
"grad_norm": 8.696606134772816, |
|
"learning_rate": 4.894001499771015e-07, |
|
"log_odds_chosen": 0.408263623714447, |
|
"log_odds_ratio": -0.5970031023025513, |
|
"logits/chosen": -2.8863773345947266, |
|
"logits/rejected": -2.8754923343658447, |
|
"logps/chosen": -0.82969731092453, |
|
"logps/rejected": -1.0684994459152222, |
|
"loss": 0.8737, |
|
"nll_loss": 0.8546596765518188, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.04148487001657486, |
|
"rewards/margins": 0.011940106749534607, |
|
"rewards/rejected": -0.05342497676610947, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 1.23, |
|
"grad_norm": 8.142229649165017, |
|
"learning_rate": 4.7636205937328664e-07, |
|
"log_odds_chosen": 0.24737975001335144, |
|
"log_odds_ratio": -0.6806861162185669, |
|
"logits/chosen": -2.8786978721618652, |
|
"logits/rejected": -2.869795083999634, |
|
"logps/chosen": -0.8045932054519653, |
|
"logps/rejected": -0.9472286105155945, |
|
"loss": 0.9147, |
|
"nll_loss": 0.8494957089424133, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.04022965580224991, |
|
"rewards/margins": 0.007131779100745916, |
|
"rewards/rejected": -0.04736143350601196, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 1.33, |
|
"grad_norm": 8.40750837533949, |
|
"learning_rate": 4.584994544724695e-07, |
|
"log_odds_chosen": 0.2743343412876129, |
|
"log_odds_ratio": -0.6686180830001831, |
|
"logits/chosen": -2.866347312927246, |
|
"logits/rejected": -2.8765628337860107, |
|
"logps/chosen": -0.7923319935798645, |
|
"logps/rejected": -0.9410168528556824, |
|
"loss": 0.8631, |
|
"nll_loss": 0.8559592366218567, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -0.039616603404283524, |
|
"rewards/margins": 0.007434243801981211, |
|
"rewards/rejected": -0.047050848603248596, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 1.42, |
|
"grad_norm": 8.158461437554013, |
|
"learning_rate": 4.3619304568594546e-07, |
|
"log_odds_chosen": 0.3588339388370514, |
|
"log_odds_ratio": -0.6238493919372559, |
|
"logits/chosen": -2.869856119155884, |
|
"logits/rejected": -2.8966941833496094, |
|
"logps/chosen": -0.7852147817611694, |
|
"logps/rejected": -0.9708358645439148, |
|
"loss": 0.8549, |
|
"nll_loss": 0.8177824020385742, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.03926074132323265, |
|
"rewards/margins": 0.009281056001782417, |
|
"rewards/rejected": -0.04854179546236992, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 1.52, |
|
"grad_norm": 8.09400448571549, |
|
"learning_rate": 4.099182553897228e-07, |
|
"log_odds_chosen": 0.47473040223121643, |
|
"log_odds_ratio": -0.5752421021461487, |
|
"logits/chosen": -2.8670287132263184, |
|
"logits/rejected": -2.849987506866455, |
|
"logps/chosen": -0.7626146078109741, |
|
"logps/rejected": -1.0768699645996094, |
|
"loss": 0.8506, |
|
"nll_loss": 0.7988383173942566, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.038130730390548706, |
|
"rewards/margins": 0.01571277156472206, |
|
"rewards/rejected": -0.05384349822998047, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 1.61, |
|
"grad_norm": 6.8406261430616535, |
|
"learning_rate": 3.8023508512198257e-07, |
|
"log_odds_chosen": 0.35588333010673523, |
|
"log_odds_ratio": -0.6235536336898804, |
|
"logits/chosen": -2.8909614086151123, |
|
"logits/rejected": -2.8951175212860107, |
|
"logps/chosen": -0.7602669596672058, |
|
"logps/rejected": -0.9576327204704285, |
|
"loss": 0.8503, |
|
"nll_loss": 0.7980303764343262, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.03801335021853447, |
|
"rewards/margins": 0.009868290275335312, |
|
"rewards/rejected": -0.04788164049386978, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.71, |
|
"grad_norm": 7.855569563727718, |
|
"learning_rate": 3.4777618012253895e-07, |
|
"log_odds_chosen": 0.243193581700325, |
|
"log_odds_ratio": -0.6670368909835815, |
|
"logits/chosen": -2.84794545173645, |
|
"logits/rejected": -2.8654625415802, |
|
"logps/chosen": -0.8232158422470093, |
|
"logps/rejected": -0.9566510915756226, |
|
"loss": 0.8633, |
|
"nll_loss": 0.8674771189689636, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.041160788387060165, |
|
"rewards/margins": 0.006671770475804806, |
|
"rewards/rejected": -0.04783256724476814, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.8, |
|
"grad_norm": 7.6860644181764375, |
|
"learning_rate": 3.1323334559792015e-07, |
|
"log_odds_chosen": 0.2766939699649811, |
|
"log_odds_ratio": -0.6430349349975586, |
|
"logits/chosen": -2.8580737113952637, |
|
"logits/rejected": -2.8492133617401123, |
|
"logps/chosen": -0.7749003171920776, |
|
"logps/rejected": -0.9475242495536804, |
|
"loss": 0.8497, |
|
"nll_loss": 0.7961811423301697, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.03874502331018448, |
|
"rewards/margins": 0.008631192147731781, |
|
"rewards/rejected": -0.047376204282045364, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.9, |
|
"grad_norm": 7.778871538887919, |
|
"learning_rate": 2.7734280209446865e-07, |
|
"log_odds_chosen": 0.2929554879665375, |
|
"log_odds_ratio": -0.6454571485519409, |
|
"logits/chosen": -2.874631881713867, |
|
"logits/rejected": -2.872854232788086, |
|
"logps/chosen": -0.8560446500778198, |
|
"logps/rejected": -1.0225257873535156, |
|
"loss": 0.8742, |
|
"nll_loss": 0.8836696743965149, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.04280223324894905, |
|
"rewards/margins": 0.008324062451720238, |
|
"rewards/rejected": -0.05112629011273384, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.99, |
|
"grad_norm": 8.608571431216085, |
|
"learning_rate": 2.4086949423558525e-07, |
|
"log_odds_chosen": 0.4544145464897156, |
|
"log_odds_ratio": -0.5953847169876099, |
|
"logits/chosen": -2.8644211292266846, |
|
"logits/rejected": -2.8725366592407227, |
|
"logps/chosen": -0.7150249481201172, |
|
"logps/rejected": -0.9680275917053223, |
|
"loss": 0.8397, |
|
"nll_loss": 0.7700489163398743, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.03575124591588974, |
|
"rewards/margins": 0.012650132179260254, |
|
"rewards/rejected": -0.048401378095149994, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_log_odds_chosen": 0.3554897606372833, |
|
"eval_log_odds_ratio": -0.6257309317588806, |
|
"eval_logits/chosen": -2.8794047832489014, |
|
"eval_logits/rejected": -2.888169527053833, |
|
"eval_logps/chosen": -0.8071174621582031, |
|
"eval_logps/rejected": -0.9902352690696716, |
|
"eval_loss": 0.8611603379249573, |
|
"eval_nll_loss": 0.8365828990936279, |
|
"eval_rewards/accuracies": 0.6458333134651184, |
|
"eval_rewards/chosen": -0.040355876088142395, |
|
"eval_rewards/margins": 0.009155889973044395, |
|
"eval_rewards/rejected": -0.04951176047325134, |
|
"eval_runtime": 32.7226, |
|
"eval_samples_per_second": 22.92, |
|
"eval_steps_per_second": 0.733, |
|
"step": 211 |
|
}, |
|
{ |
|
"epoch": 2.09, |
|
"grad_norm": 6.957321522636136, |
|
"learning_rate": 2.0459078725514089e-07, |
|
"log_odds_chosen": 0.5733749866485596, |
|
"log_odds_ratio": -0.5376938581466675, |
|
"logits/chosen": -2.876321315765381, |
|
"logits/rejected": -2.8998565673828125, |
|
"logps/chosen": -0.6645466089248657, |
|
"logps/rejected": -0.9755932092666626, |
|
"loss": 0.7779, |
|
"nll_loss": 0.7205012440681458, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.033227331936359406, |
|
"rewards/margins": 0.015552327036857605, |
|
"rewards/rejected": -0.04877965897321701, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 2.18, |
|
"grad_norm": 7.383557049949752, |
|
"learning_rate": 1.692798988071385e-07, |
|
"log_odds_chosen": 0.6264925003051758, |
|
"log_odds_ratio": -0.5338762998580933, |
|
"logits/chosen": -2.9063897132873535, |
|
"logits/rejected": -2.9123330116271973, |
|
"logps/chosen": -0.6798352003097534, |
|
"logps/rejected": -1.0108721256256104, |
|
"loss": 0.7879, |
|
"nll_loss": 0.763096809387207, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.03399176150560379, |
|
"rewards/margins": 0.016551844775676727, |
|
"rewards/rejected": -0.05054359883069992, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 2.27, |
|
"grad_norm": 7.205471545860071, |
|
"learning_rate": 1.3568941917384036e-07, |
|
"log_odds_chosen": 0.6044402122497559, |
|
"log_odds_ratio": -0.5256311297416687, |
|
"logits/chosen": -2.929206609725952, |
|
"logits/rejected": -2.9085187911987305, |
|
"logps/chosen": -0.6959676742553711, |
|
"logps/rejected": -1.0572305917739868, |
|
"loss": 0.7677, |
|
"nll_loss": 0.7402119636535645, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.034798379987478256, |
|
"rewards/margins": 0.018063146620988846, |
|
"rewards/rejected": -0.0528615303337574, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 2.37, |
|
"grad_norm": 7.342052939736115, |
|
"learning_rate": 1.0453527111051183e-07, |
|
"log_odds_chosen": 0.6251620054244995, |
|
"log_odds_ratio": -0.5341116786003113, |
|
"logits/chosen": -2.8869872093200684, |
|
"logits/rejected": -2.912830352783203, |
|
"logps/chosen": -0.6446677446365356, |
|
"logps/rejected": -0.9978305101394653, |
|
"loss": 0.7682, |
|
"nll_loss": 0.6836553812026978, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": -0.03223338723182678, |
|
"rewards/margins": 0.017658134922385216, |
|
"rewards/rejected": -0.04989152401685715, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 2.46, |
|
"grad_norm": 7.710315387822491, |
|
"learning_rate": 7.648145119484151e-08, |
|
"log_odds_chosen": 0.582895815372467, |
|
"log_odds_ratio": -0.5480377078056335, |
|
"logits/chosen": -2.89570689201355, |
|
"logits/rejected": -2.8959426879882812, |
|
"logps/chosen": -0.7015441656112671, |
|
"logps/rejected": -0.9980028867721558, |
|
"loss": 0.7845, |
|
"nll_loss": 0.7611762881278992, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.035077206790447235, |
|
"rewards/margins": 0.014822937548160553, |
|
"rewards/rejected": -0.04990014433860779, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 2.56, |
|
"grad_norm": 7.850416556443741, |
|
"learning_rate": 5.212587789268649e-08, |
|
"log_odds_chosen": 0.5626873970031738, |
|
"log_odds_ratio": -0.5825694799423218, |
|
"logits/chosen": -2.898651599884033, |
|
"logits/rejected": -2.9134531021118164, |
|
"logps/chosen": -0.713202714920044, |
|
"logps/rejected": -0.9749401211738586, |
|
"loss": 0.7837, |
|
"nll_loss": 0.7641984224319458, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.03566013649106026, |
|
"rewards/margins": 0.013086864724755287, |
|
"rewards/rejected": -0.04874700307846069, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 2.65, |
|
"grad_norm": 7.144333751483519, |
|
"learning_rate": 3.198764796404807e-08, |
|
"log_odds_chosen": 0.6033560037612915, |
|
"log_odds_ratio": -0.5110877156257629, |
|
"logits/chosen": -2.9342381954193115, |
|
"logits/rejected": -2.9433302879333496, |
|
"logps/chosen": -0.6664480566978455, |
|
"logps/rejected": -0.9789284467697144, |
|
"loss": 0.7707, |
|
"nll_loss": 0.7404603362083435, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.03332240507006645, |
|
"rewards/margins": 0.015624021179974079, |
|
"rewards/rejected": -0.04894643276929855, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 2.75, |
|
"grad_norm": 7.681448875266474, |
|
"learning_rate": 1.649597281686302e-08, |
|
"log_odds_chosen": 0.5207457542419434, |
|
"log_odds_ratio": -0.5457995533943176, |
|
"logits/chosen": -2.933040142059326, |
|
"logits/rejected": -2.942648410797119, |
|
"logps/chosen": -0.6956908106803894, |
|
"logps/rejected": -0.970605194568634, |
|
"loss": 0.7797, |
|
"nll_loss": 0.7457938194274902, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.03478454053401947, |
|
"rewards/margins": 0.013745710253715515, |
|
"rewards/rejected": -0.048530250787734985, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 2.84, |
|
"grad_norm": 7.3441185490833805, |
|
"learning_rate": 5.981030611018234e-09, |
|
"log_odds_chosen": 0.643921434879303, |
|
"log_odds_ratio": -0.5143691897392273, |
|
"logits/chosen": -2.923649549484253, |
|
"logits/rejected": -2.937542676925659, |
|
"logps/chosen": -0.7117995619773865, |
|
"logps/rejected": -1.0296577215194702, |
|
"loss": 0.7861, |
|
"nll_loss": 0.7953752279281616, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.035589978098869324, |
|
"rewards/margins": 0.015892909839749336, |
|
"rewards/rejected": -0.05148288607597351, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 2.94, |
|
"grad_norm": 7.411020953558967, |
|
"learning_rate": 6.66929084112089e-10, |
|
"log_odds_chosen": 0.531533420085907, |
|
"log_odds_ratio": -0.5645086765289307, |
|
"logits/chosen": -2.9212653636932373, |
|
"logits/rejected": -2.941697835922241, |
|
"logps/chosen": -0.6984909772872925, |
|
"logps/rejected": -0.9673376083374023, |
|
"loss": 0.7808, |
|
"nll_loss": 0.7526401877403259, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.034924548119306564, |
|
"rewards/margins": 0.013442324474453926, |
|
"rewards/rejected": -0.04836687445640564, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 2.99, |
|
"eval_log_odds_chosen": 0.3802085816860199, |
|
"eval_log_odds_ratio": -0.6214679479598999, |
|
"eval_logits/chosen": -2.9039885997772217, |
|
"eval_logits/rejected": -2.914613723754883, |
|
"eval_logps/chosen": -0.8096261024475098, |
|
"eval_logps/rejected": -1.0036059617996216, |
|
"eval_loss": 0.8648061752319336, |
|
"eval_nll_loss": 0.8392227292060852, |
|
"eval_rewards/accuracies": 0.6458333134651184, |
|
"eval_rewards/chosen": -0.04048130661249161, |
|
"eval_rewards/margins": 0.009698997251689434, |
|
"eval_rewards/rejected": -0.05018030107021332, |
|
"eval_runtime": 32.6828, |
|
"eval_samples_per_second": 22.948, |
|
"eval_steps_per_second": 0.734, |
|
"step": 315 |
|
}, |
|
{ |
|
"epoch": 2.99, |
|
"step": 315, |
|
"total_flos": 0.0, |
|
"train_loss": 0.877839538029262, |
|
"train_runtime": 3399.396, |
|
"train_samples_per_second": 5.957, |
|
"train_steps_per_second": 0.093 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 315, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"total_flos": 0.0, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|