|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 100, |
|
"global_step": 1556, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 1.0706638115631692e-10, |
|
"logits/chosen": 3.009334087371826, |
|
"logits/rejected": 2.8860254287719727, |
|
"logps/chosen": -363.7457580566406, |
|
"logps/rejected": -405.4437255859375, |
|
"loss": 0.6931, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 1.070663811563169e-09, |
|
"logits/chosen": 2.383711338043213, |
|
"logits/rejected": 3.0388379096984863, |
|
"logps/chosen": -456.6111145019531, |
|
"logps/rejected": -428.7125244140625, |
|
"loss": 2.7789, |
|
"rewards/accuracies": 0.4305555522441864, |
|
"rewards/chosen": -0.10625362396240234, |
|
"rewards/margins": -0.11497735977172852, |
|
"rewards/rejected": 0.008723735809326172, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 2.141327623126338e-09, |
|
"logits/chosen": 2.6908938884735107, |
|
"logits/rejected": 3.1237950325012207, |
|
"logps/chosen": -387.3503112792969, |
|
"logps/rejected": -384.4349060058594, |
|
"loss": 2.7716, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": 0.43295717239379883, |
|
"rewards/margins": 1.0383365154266357, |
|
"rewards/rejected": -0.6053793430328369, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 3.2119914346895075e-09, |
|
"logits/chosen": 2.479438543319702, |
|
"logits/rejected": 2.9864134788513184, |
|
"logps/chosen": -409.71563720703125, |
|
"logps/rejected": -410.36260986328125, |
|
"loss": 3.0305, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": 0.3105888366699219, |
|
"rewards/margins": 0.5438008308410645, |
|
"rewards/rejected": -0.23321199417114258, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 4.282655246252676e-09, |
|
"logits/chosen": 2.513826847076416, |
|
"logits/rejected": 3.7642135620117188, |
|
"logps/chosen": -412.05780029296875, |
|
"logps/rejected": -408.3034973144531, |
|
"loss": 2.3915, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.4330732822418213, |
|
"rewards/margins": 0.3383500576019287, |
|
"rewards/rejected": -0.77142333984375, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 5.353319057815846e-09, |
|
"logits/chosen": 2.6957950592041016, |
|
"logits/rejected": 3.350551128387451, |
|
"logps/chosen": -428.7166442871094, |
|
"logps/rejected": -425.9712829589844, |
|
"loss": 2.6906, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": 0.6460269093513489, |
|
"rewards/margins": 0.9008991122245789, |
|
"rewards/rejected": -0.25487232208251953, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 6.423982869379015e-09, |
|
"logits/chosen": 1.877236008644104, |
|
"logits/rejected": 3.345506191253662, |
|
"logps/chosen": -423.69549560546875, |
|
"logps/rejected": -409.8067321777344, |
|
"loss": 1.8376, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": 2.4741673469543457, |
|
"rewards/margins": 4.329440116882324, |
|
"rewards/rejected": -1.8552722930908203, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 7.494646680942184e-09, |
|
"logits/chosen": 2.3659510612487793, |
|
"logits/rejected": 3.347115993499756, |
|
"logps/chosen": -419.614990234375, |
|
"logps/rejected": -404.15386962890625, |
|
"loss": 1.8117, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": 1.2335608005523682, |
|
"rewards/margins": 1.696711778640747, |
|
"rewards/rejected": -0.4631509780883789, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 8.565310492505352e-09, |
|
"logits/chosen": 1.8567355871200562, |
|
"logits/rejected": 3.529881238937378, |
|
"logps/chosen": -454.572998046875, |
|
"logps/rejected": -433.12152099609375, |
|
"loss": 1.4964, |
|
"rewards/accuracies": 0.737500011920929, |
|
"rewards/chosen": 1.8937450647354126, |
|
"rewards/margins": 4.073466777801514, |
|
"rewards/rejected": -2.1797218322753906, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 9.635974304068522e-09, |
|
"logits/chosen": 2.379859447479248, |
|
"logits/rejected": 3.260948657989502, |
|
"logps/chosen": -451.9266052246094, |
|
"logps/rejected": -383.00042724609375, |
|
"loss": 1.6543, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": 2.33457612991333, |
|
"rewards/margins": 3.6442272663116455, |
|
"rewards/rejected": -1.3096508979797363, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 1.0706638115631692e-08, |
|
"logits/chosen": 2.4745795726776123, |
|
"logits/rejected": 3.668903350830078, |
|
"logps/chosen": -402.97393798828125, |
|
"logps/rejected": -378.31890869140625, |
|
"loss": 1.5408, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": 0.7632139921188354, |
|
"rewards/margins": 2.8585281372070312, |
|
"rewards/rejected": -2.0953145027160645, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"eval_logits/chosen": 1.7967612743377686, |
|
"eval_logits/rejected": 2.483816146850586, |
|
"eval_logps/chosen": -403.54827880859375, |
|
"eval_logps/rejected": -402.92413330078125, |
|
"eval_loss": 1.0591531991958618, |
|
"eval_rewards/accuracies": 0.8125, |
|
"eval_rewards/chosen": 2.333465814590454, |
|
"eval_rewards/margins": 5.841272830963135, |
|
"eval_rewards/rejected": -3.5078072547912598, |
|
"eval_runtime": 76.9287, |
|
"eval_samples_per_second": 12.999, |
|
"eval_steps_per_second": 0.416, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 1.177730192719486e-08, |
|
"logits/chosen": 2.7680611610412598, |
|
"logits/rejected": 3.0494232177734375, |
|
"logps/chosen": -384.3839111328125, |
|
"logps/rejected": -407.72918701171875, |
|
"loss": 1.0018, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": 2.2038276195526123, |
|
"rewards/margins": 6.380339622497559, |
|
"rewards/rejected": -4.176511764526367, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 1.284796573875803e-08, |
|
"logits/chosen": 2.498812198638916, |
|
"logits/rejected": 3.4484715461730957, |
|
"logps/chosen": -401.28857421875, |
|
"logps/rejected": -426.8399963378906, |
|
"loss": 0.7845, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": 2.2210142612457275, |
|
"rewards/margins": 8.185151100158691, |
|
"rewards/rejected": -5.964136123657227, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 1.3918629550321198e-08, |
|
"logits/chosen": 2.083467483520508, |
|
"logits/rejected": 2.998776435852051, |
|
"logps/chosen": -437.83709716796875, |
|
"logps/rejected": -462.96026611328125, |
|
"loss": 0.9019, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": 4.379509449005127, |
|
"rewards/margins": 10.292404174804688, |
|
"rewards/rejected": -5.912895202636719, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 1.4989293361884368e-08, |
|
"logits/chosen": 2.5522398948669434, |
|
"logits/rejected": 3.3716235160827637, |
|
"logps/chosen": -422.5474548339844, |
|
"logps/rejected": -406.94183349609375, |
|
"loss": 0.8578, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": 2.729921579360962, |
|
"rewards/margins": 7.9893903732299805, |
|
"rewards/rejected": -5.259469509124756, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 1.6059957173447535e-08, |
|
"logits/chosen": 2.7751362323760986, |
|
"logits/rejected": 2.9003946781158447, |
|
"logps/chosen": -391.8503723144531, |
|
"logps/rejected": -400.2032470703125, |
|
"loss": 0.7248, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": 1.3115911483764648, |
|
"rewards/margins": 8.482717514038086, |
|
"rewards/rejected": -7.171125888824463, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 1.7130620985010704e-08, |
|
"logits/chosen": 2.3851382732391357, |
|
"logits/rejected": 3.175081968307495, |
|
"logps/chosen": -482.95062255859375, |
|
"logps/rejected": -422.1293029785156, |
|
"loss": 0.6889, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": 3.3662681579589844, |
|
"rewards/margins": 10.24647331237793, |
|
"rewards/rejected": -6.8802056312561035, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 1.8201284796573874e-08, |
|
"logits/chosen": 2.398458957672119, |
|
"logits/rejected": 3.4337353706359863, |
|
"logps/chosen": -422.200927734375, |
|
"logps/rejected": -402.37274169921875, |
|
"loss": 0.8347, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": 0.8539754748344421, |
|
"rewards/margins": 8.811124801635742, |
|
"rewards/rejected": -7.957149505615234, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 1.9271948608137044e-08, |
|
"logits/chosen": 2.4511101245880127, |
|
"logits/rejected": 3.5071682929992676, |
|
"logps/chosen": -444.2669372558594, |
|
"logps/rejected": -460.7818908691406, |
|
"loss": 0.7348, |
|
"rewards/accuracies": 0.862500011920929, |
|
"rewards/chosen": 0.6959193348884583, |
|
"rewards/margins": 10.471872329711914, |
|
"rewards/rejected": -9.77595329284668, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 2.0342612419700214e-08, |
|
"logits/chosen": 2.3941550254821777, |
|
"logits/rejected": 3.183414936065674, |
|
"logps/chosen": -406.2020263671875, |
|
"logps/rejected": -389.969970703125, |
|
"loss": 0.6852, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": 0.565895140171051, |
|
"rewards/margins": 9.187482833862305, |
|
"rewards/rejected": -8.621587753295898, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 2.1413276231263384e-08, |
|
"logits/chosen": 2.699617624282837, |
|
"logits/rejected": 3.2629177570343018, |
|
"logps/chosen": -370.5373840332031, |
|
"logps/rejected": -373.650634765625, |
|
"loss": 0.6805, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": 0.4384472370147705, |
|
"rewards/margins": 9.281099319458008, |
|
"rewards/rejected": -8.8426513671875, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"eval_logits/chosen": 1.8075158596038818, |
|
"eval_logits/rejected": 2.4975709915161133, |
|
"eval_logps/chosen": -403.94219970703125, |
|
"eval_logps/rejected": -404.73687744140625, |
|
"eval_loss": 0.7941137552261353, |
|
"eval_rewards/accuracies": 0.9140625, |
|
"eval_rewards/chosen": 0.36384421586990356, |
|
"eval_rewards/margins": 12.935296058654785, |
|
"eval_rewards/rejected": -12.571451187133789, |
|
"eval_runtime": 76.7386, |
|
"eval_samples_per_second": 13.031, |
|
"eval_steps_per_second": 0.417, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 2.248394004282655e-08, |
|
"logits/chosen": 2.466681718826294, |
|
"logits/rejected": 3.0487618446350098, |
|
"logps/chosen": -413.58941650390625, |
|
"logps/rejected": -422.9561462402344, |
|
"loss": 0.5453, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 1.9145727157592773, |
|
"rewards/margins": 14.975469589233398, |
|
"rewards/rejected": -13.06089973449707, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 2.355460385438972e-08, |
|
"logits/chosen": 2.4473984241485596, |
|
"logits/rejected": 3.8849105834960938, |
|
"logps/chosen": -435.60198974609375, |
|
"logps/rejected": -394.08990478515625, |
|
"loss": 0.4307, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": 1.1083576679229736, |
|
"rewards/margins": 12.617959976196289, |
|
"rewards/rejected": -11.509601593017578, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 2.462526766595289e-08, |
|
"logits/chosen": 3.0809953212738037, |
|
"logits/rejected": 3.275381088256836, |
|
"logps/chosen": -387.32427978515625, |
|
"logps/rejected": -418.32745361328125, |
|
"loss": 0.4707, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 2.789315700531006, |
|
"rewards/margins": 15.172778129577637, |
|
"rewards/rejected": -12.383462905883789, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 2.569593147751606e-08, |
|
"logits/chosen": 2.8269176483154297, |
|
"logits/rejected": 3.720735549926758, |
|
"logps/chosen": -400.30743408203125, |
|
"logps/rejected": -414.60430908203125, |
|
"loss": 0.4476, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": 1.7549598217010498, |
|
"rewards/margins": 15.389744758605957, |
|
"rewards/rejected": -13.634783744812012, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 2.676659528907923e-08, |
|
"logits/chosen": 2.572218894958496, |
|
"logits/rejected": 2.84545636177063, |
|
"logps/chosen": -398.96356201171875, |
|
"logps/rejected": -404.20635986328125, |
|
"loss": 0.3304, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": 1.083133578300476, |
|
"rewards/margins": 14.877154350280762, |
|
"rewards/rejected": -13.79401969909668, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 2.7837259100642396e-08, |
|
"logits/chosen": 2.4352571964263916, |
|
"logits/rejected": 3.324097156524658, |
|
"logps/chosen": -389.4182434082031, |
|
"logps/rejected": -390.0982666015625, |
|
"loss": 0.363, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": 0.10746519267559052, |
|
"rewards/margins": 14.993916511535645, |
|
"rewards/rejected": -14.886451721191406, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 2.890792291220557e-08, |
|
"logits/chosen": 2.1370952129364014, |
|
"logits/rejected": 3.407914400100708, |
|
"logps/chosen": -434.8224182128906, |
|
"logps/rejected": -440.6358947753906, |
|
"loss": 0.4309, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 2.220722198486328, |
|
"rewards/margins": 18.430742263793945, |
|
"rewards/rejected": -16.210018157958984, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 2.9978586723768736e-08, |
|
"logits/chosen": 2.740478992462158, |
|
"logits/rejected": 3.258228302001953, |
|
"logps/chosen": -433.20782470703125, |
|
"logps/rejected": -437.18719482421875, |
|
"loss": 0.3487, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": 2.553795576095581, |
|
"rewards/margins": 19.554113388061523, |
|
"rewards/rejected": -17.00031852722168, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 3.1049250535331906e-08, |
|
"logits/chosen": 2.1817595958709717, |
|
"logits/rejected": 3.1689507961273193, |
|
"logps/chosen": -440.03863525390625, |
|
"logps/rejected": -409.65240478515625, |
|
"loss": 0.3633, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 1.1603565216064453, |
|
"rewards/margins": 17.322303771972656, |
|
"rewards/rejected": -16.161945343017578, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 3.211991434689507e-08, |
|
"logits/chosen": 2.570457935333252, |
|
"logits/rejected": 3.2502989768981934, |
|
"logps/chosen": -454.56890869140625, |
|
"logps/rejected": -416.1253356933594, |
|
"loss": 0.4447, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 3.8409790992736816, |
|
"rewards/margins": 22.155860900878906, |
|
"rewards/rejected": -18.31488037109375, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"eval_logits/chosen": 1.8170030117034912, |
|
"eval_logits/rejected": 2.5056939125061035, |
|
"eval_logps/chosen": -403.96087646484375, |
|
"eval_logps/rejected": -405.89984130859375, |
|
"eval_loss": 0.7104232311248779, |
|
"eval_rewards/accuracies": 0.9296875, |
|
"eval_rewards/chosen": 0.27033889293670654, |
|
"eval_rewards/margins": 18.656814575195312, |
|
"eval_rewards/rejected": -18.386476516723633, |
|
"eval_runtime": 76.9022, |
|
"eval_samples_per_second": 13.004, |
|
"eval_steps_per_second": 0.416, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 3.3190578158458246e-08, |
|
"logits/chosen": 2.6424567699432373, |
|
"logits/rejected": 3.5541133880615234, |
|
"logps/chosen": -389.58636474609375, |
|
"logps/rejected": -417.542236328125, |
|
"loss": 0.1892, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 2.047499179840088, |
|
"rewards/margins": 22.001834869384766, |
|
"rewards/rejected": -19.954334259033203, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 3.426124197002141e-08, |
|
"logits/chosen": 2.256901264190674, |
|
"logits/rejected": 3.4731545448303223, |
|
"logps/chosen": -414.03875732421875, |
|
"logps/rejected": -427.070556640625, |
|
"loss": 0.3266, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 0.8677770495414734, |
|
"rewards/margins": 22.491207122802734, |
|
"rewards/rejected": -21.623430252075195, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 3.533190578158458e-08, |
|
"logits/chosen": 2.77763295173645, |
|
"logits/rejected": 3.112457752227783, |
|
"logps/chosen": -360.58135986328125, |
|
"logps/rejected": -414.3995666503906, |
|
"loss": 0.2844, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 2.416537046432495, |
|
"rewards/margins": 22.439632415771484, |
|
"rewards/rejected": -20.023096084594727, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 3.640256959314775e-08, |
|
"logits/chosen": 2.5908098220825195, |
|
"logits/rejected": 3.282468795776367, |
|
"logps/chosen": -449.01922607421875, |
|
"logps/rejected": -422.73016357421875, |
|
"loss": 0.3479, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 2.354691505432129, |
|
"rewards/margins": 22.229778289794922, |
|
"rewards/rejected": -19.875089645385742, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 3.747323340471092e-08, |
|
"logits/chosen": 2.534389019012451, |
|
"logits/rejected": 3.5590693950653076, |
|
"logps/chosen": -405.7064514160156, |
|
"logps/rejected": -434.4012756347656, |
|
"loss": 0.4823, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 1.6845805644989014, |
|
"rewards/margins": 23.965158462524414, |
|
"rewards/rejected": -22.280576705932617, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 3.854389721627409e-08, |
|
"logits/chosen": 2.5174474716186523, |
|
"logits/rejected": 3.512732744216919, |
|
"logps/chosen": -435.5006408691406, |
|
"logps/rejected": -422.19366455078125, |
|
"loss": 0.204, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 3.582836151123047, |
|
"rewards/margins": 24.59599494934082, |
|
"rewards/rejected": -21.013154983520508, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 3.961456102783726e-08, |
|
"logits/chosen": 2.3562188148498535, |
|
"logits/rejected": 3.3465442657470703, |
|
"logps/chosen": -470.4930725097656, |
|
"logps/rejected": -405.6280822753906, |
|
"loss": 0.3122, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 1.8608239889144897, |
|
"rewards/margins": 23.556621551513672, |
|
"rewards/rejected": -21.6957950592041, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.068522483940043e-08, |
|
"logits/chosen": 2.7507357597351074, |
|
"logits/rejected": 3.5644543170928955, |
|
"logps/chosen": -368.4208068847656, |
|
"logps/rejected": -416.85302734375, |
|
"loss": 0.2533, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 1.0723613500595093, |
|
"rewards/margins": 24.53675651550293, |
|
"rewards/rejected": -23.464397430419922, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.175588865096359e-08, |
|
"logits/chosen": 2.4924309253692627, |
|
"logits/rejected": 3.5116946697235107, |
|
"logps/chosen": -383.086181640625, |
|
"logps/rejected": -417.8291931152344, |
|
"loss": 0.2178, |
|
"rewards/accuracies": 0.887499988079071, |
|
"rewards/chosen": 4.934910774230957, |
|
"rewards/margins": 26.8948974609375, |
|
"rewards/rejected": -21.959985733032227, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.282655246252677e-08, |
|
"logits/chosen": 2.5591559410095215, |
|
"logits/rejected": 3.345097780227661, |
|
"logps/chosen": -449.8941955566406, |
|
"logps/rejected": -431.88818359375, |
|
"loss": 0.2468, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 5.43572473526001, |
|
"rewards/margins": 28.937246322631836, |
|
"rewards/rejected": -23.501522064208984, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"eval_logits/chosen": 1.8141372203826904, |
|
"eval_logits/rejected": 2.503908395767212, |
|
"eval_logps/chosen": -403.28955078125, |
|
"eval_logps/rejected": -406.5526123046875, |
|
"eval_loss": 0.5629487037658691, |
|
"eval_rewards/accuracies": 0.9140625, |
|
"eval_rewards/chosen": 3.626873731613159, |
|
"eval_rewards/margins": 25.27718162536621, |
|
"eval_rewards/rejected": -21.650310516357422, |
|
"eval_runtime": 76.66, |
|
"eval_samples_per_second": 13.045, |
|
"eval_steps_per_second": 0.417, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.389721627408993e-08, |
|
"logits/chosen": 2.353135585784912, |
|
"logits/rejected": 3.1894092559814453, |
|
"logps/chosen": -386.232666015625, |
|
"logps/rejected": -399.92218017578125, |
|
"loss": 0.1749, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 5.276652812957764, |
|
"rewards/margins": 27.45583724975586, |
|
"rewards/rejected": -22.17918586730957, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.49678800856531e-08, |
|
"logits/chosen": 1.99558424949646, |
|
"logits/rejected": 3.1743838787078857, |
|
"logps/chosen": -412.32110595703125, |
|
"logps/rejected": -421.9241638183594, |
|
"loss": 0.272, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 6.199848175048828, |
|
"rewards/margins": 27.825729370117188, |
|
"rewards/rejected": -21.62588119506836, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.603854389721627e-08, |
|
"logits/chosen": 2.605555772781372, |
|
"logits/rejected": 3.407613754272461, |
|
"logps/chosen": -434.2098083496094, |
|
"logps/rejected": -447.4507751464844, |
|
"loss": 0.3003, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 8.46845531463623, |
|
"rewards/margins": 32.25395584106445, |
|
"rewards/rejected": -23.785497665405273, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.710920770877944e-08, |
|
"logits/chosen": 2.1723690032958984, |
|
"logits/rejected": 3.847750186920166, |
|
"logps/chosen": -421.90155029296875, |
|
"logps/rejected": -414.0150451660156, |
|
"loss": 0.2367, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 8.460034370422363, |
|
"rewards/margins": 31.530038833618164, |
|
"rewards/rejected": -23.070003509521484, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.817987152034261e-08, |
|
"logits/chosen": 2.460693120956421, |
|
"logits/rejected": 2.9349396228790283, |
|
"logps/chosen": -502.804931640625, |
|
"logps/rejected": -431.50335693359375, |
|
"loss": 0.2369, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 7.705231666564941, |
|
"rewards/margins": 32.00004959106445, |
|
"rewards/rejected": -24.294815063476562, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.925053533190578e-08, |
|
"logits/chosen": 2.333646297454834, |
|
"logits/rejected": 4.089064598083496, |
|
"logps/chosen": -422.8240661621094, |
|
"logps/rejected": -442.1726989746094, |
|
"loss": 0.1691, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 4.384838581085205, |
|
"rewards/margins": 29.339359283447266, |
|
"rewards/rejected": -24.954524993896484, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 5.032119914346895e-08, |
|
"logits/chosen": 2.252077579498291, |
|
"logits/rejected": 3.398280382156372, |
|
"logps/chosen": -426.0081481933594, |
|
"logps/rejected": -431.38623046875, |
|
"loss": 0.2549, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 3.1347286701202393, |
|
"rewards/margins": 29.128936767578125, |
|
"rewards/rejected": -25.994205474853516, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 5.139186295503212e-08, |
|
"logits/chosen": 1.9574702978134155, |
|
"logits/rejected": 3.0777688026428223, |
|
"logps/chosen": -445.21435546875, |
|
"logps/rejected": -431.9685974121094, |
|
"loss": 0.2022, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 5.082431793212891, |
|
"rewards/margins": 33.55067825317383, |
|
"rewards/rejected": -28.468246459960938, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 5.246252676659528e-08, |
|
"logits/chosen": 2.4008655548095703, |
|
"logits/rejected": 3.1396138668060303, |
|
"logps/chosen": -396.2805480957031, |
|
"logps/rejected": -431.75994873046875, |
|
"loss": 0.1925, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 3.1321897506713867, |
|
"rewards/margins": 31.742828369140625, |
|
"rewards/rejected": -28.610637664794922, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 5.353319057815846e-08, |
|
"logits/chosen": 2.6065423488616943, |
|
"logits/rejected": 3.01694917678833, |
|
"logps/chosen": -441.64752197265625, |
|
"logps/rejected": -450.52337646484375, |
|
"loss": 0.238, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 5.9151458740234375, |
|
"rewards/margins": 37.43492889404297, |
|
"rewards/rejected": -31.5197811126709, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"eval_logits/chosen": 1.8356637954711914, |
|
"eval_logits/rejected": 2.518232822418213, |
|
"eval_logps/chosen": -403.12066650390625, |
|
"eval_logps/rejected": -407.75311279296875, |
|
"eval_loss": 0.40245920419692993, |
|
"eval_rewards/accuracies": 0.9296875, |
|
"eval_rewards/chosen": 4.471410751342773, |
|
"eval_rewards/margins": 32.124061584472656, |
|
"eval_rewards/rejected": -27.652652740478516, |
|
"eval_runtime": 76.7691, |
|
"eval_samples_per_second": 13.026, |
|
"eval_steps_per_second": 0.417, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 5.460385438972163e-08, |
|
"logits/chosen": 2.370246410369873, |
|
"logits/rejected": 3.2074241638183594, |
|
"logps/chosen": -463.4814453125, |
|
"logps/rejected": -450.365478515625, |
|
"loss": 0.1896, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 4.760863304138184, |
|
"rewards/margins": 35.59904098510742, |
|
"rewards/rejected": -30.838184356689453, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 5.567451820128479e-08, |
|
"logits/chosen": 2.4717459678649902, |
|
"logits/rejected": 3.241302967071533, |
|
"logps/chosen": -476.26690673828125, |
|
"logps/rejected": -425.7784729003906, |
|
"loss": 0.2277, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 5.998467922210693, |
|
"rewards/margins": 37.100624084472656, |
|
"rewards/rejected": -31.102157592773438, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 5.6745182012847956e-08, |
|
"logits/chosen": 2.8885834217071533, |
|
"logits/rejected": 3.374998092651367, |
|
"logps/chosen": -399.87249755859375, |
|
"logps/rejected": -406.6166687011719, |
|
"loss": 0.2215, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": 5.892712116241455, |
|
"rewards/margins": 36.44768524169922, |
|
"rewards/rejected": -30.554973602294922, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 5.781584582441114e-08, |
|
"logits/chosen": 2.536708116531372, |
|
"logits/rejected": 3.2403769493103027, |
|
"logps/chosen": -429.4972229003906, |
|
"logps/rejected": -423.7460021972656, |
|
"loss": 0.1918, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 7.94210958480835, |
|
"rewards/margins": 36.776371002197266, |
|
"rewards/rejected": -28.83426284790039, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 5.88865096359743e-08, |
|
"logits/chosen": 1.9649451971054077, |
|
"logits/rejected": 3.1596691608428955, |
|
"logps/chosen": -472.22119140625, |
|
"logps/rejected": -425.32763671875, |
|
"loss": 0.15, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 12.239465713500977, |
|
"rewards/margins": 48.21662139892578, |
|
"rewards/rejected": -35.977149963378906, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 5.995717344753747e-08, |
|
"logits/chosen": 3.032383441925049, |
|
"logits/rejected": 3.257586717605591, |
|
"logps/chosen": -416.17071533203125, |
|
"logps/rejected": -410.1018981933594, |
|
"loss": 0.1132, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 7.306866645812988, |
|
"rewards/margins": 37.124568939208984, |
|
"rewards/rejected": -29.81770896911621, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 6.102783725910064e-08, |
|
"logits/chosen": 2.7644500732421875, |
|
"logits/rejected": 3.5682806968688965, |
|
"logps/chosen": -383.52093505859375, |
|
"logps/rejected": -394.8160705566406, |
|
"loss": 0.2133, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 5.922687530517578, |
|
"rewards/margins": 37.88574981689453, |
|
"rewards/rejected": -31.963062286376953, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 6.209850107066381e-08, |
|
"logits/chosen": 2.341892957687378, |
|
"logits/rejected": 3.260730028152466, |
|
"logps/chosen": -394.4947814941406, |
|
"logps/rejected": -405.29498291015625, |
|
"loss": 0.082, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 6.424149990081787, |
|
"rewards/margins": 38.39518737792969, |
|
"rewards/rejected": -31.971033096313477, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 6.316916488222698e-08, |
|
"logits/chosen": 2.526381015777588, |
|
"logits/rejected": 3.192876100540161, |
|
"logps/chosen": -426.3685607910156, |
|
"logps/rejected": -403.46759033203125, |
|
"loss": 0.1342, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 3.9063773155212402, |
|
"rewards/margins": 39.68169021606445, |
|
"rewards/rejected": -35.77531814575195, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 6.423982869379014e-08, |
|
"logits/chosen": 2.456986427307129, |
|
"logits/rejected": 2.974177837371826, |
|
"logps/chosen": -430.700927734375, |
|
"logps/rejected": -460.1724548339844, |
|
"loss": 0.0835, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 2.9916865825653076, |
|
"rewards/margins": 47.707908630371094, |
|
"rewards/rejected": -44.716224670410156, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"eval_logits/chosen": 1.8376859426498413, |
|
"eval_logits/rejected": 2.519270658493042, |
|
"eval_logps/chosen": -404.12213134765625, |
|
"eval_logps/rejected": -409.9280700683594, |
|
"eval_loss": 0.5330207943916321, |
|
"eval_rewards/accuracies": 0.9453125, |
|
"eval_rewards/chosen": -0.5357855558395386, |
|
"eval_rewards/margins": 37.99186325073242, |
|
"eval_rewards/rejected": -38.52764892578125, |
|
"eval_runtime": 76.8308, |
|
"eval_samples_per_second": 13.016, |
|
"eval_steps_per_second": 0.416, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 6.531049250535332e-08, |
|
"logits/chosen": 2.4591498374938965, |
|
"logits/rejected": 3.2855117321014404, |
|
"logps/chosen": -401.3484802246094, |
|
"logps/rejected": -434.5301208496094, |
|
"loss": 0.142, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -0.02256155014038086, |
|
"rewards/margins": 46.42057418823242, |
|
"rewards/rejected": -46.443138122558594, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 6.638115631691649e-08, |
|
"logits/chosen": 2.539071559906006, |
|
"logits/rejected": 3.2421278953552246, |
|
"logps/chosen": -442.18243408203125, |
|
"logps/rejected": -444.18878173828125, |
|
"loss": 0.1058, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 2.043628692626953, |
|
"rewards/margins": 43.48805618286133, |
|
"rewards/rejected": -41.444427490234375, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 6.745182012847965e-08, |
|
"logits/chosen": 2.5688178539276123, |
|
"logits/rejected": 3.445737838745117, |
|
"logps/chosen": -411.5978088378906, |
|
"logps/rejected": -407.0299072265625, |
|
"loss": 0.1525, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 1.022148609161377, |
|
"rewards/margins": 39.893680572509766, |
|
"rewards/rejected": -38.87153244018555, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 6.852248394004282e-08, |
|
"logits/chosen": 2.5901875495910645, |
|
"logits/rejected": 2.911647319793701, |
|
"logps/chosen": -420.1778869628906, |
|
"logps/rejected": -382.272216796875, |
|
"loss": 0.2718, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 3.5251822471618652, |
|
"rewards/margins": 41.309791564941406, |
|
"rewards/rejected": -37.784610748291016, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 6.9593147751606e-08, |
|
"logits/chosen": 2.287266492843628, |
|
"logits/rejected": 3.33722186088562, |
|
"logps/chosen": -452.6976013183594, |
|
"logps/rejected": -439.2931213378906, |
|
"loss": 0.0169, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 6.000566482543945, |
|
"rewards/margins": 46.02397918701172, |
|
"rewards/rejected": -40.023414611816406, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 7.066381156316916e-08, |
|
"logits/chosen": 2.2046732902526855, |
|
"logits/rejected": 3.6667628288269043, |
|
"logps/chosen": -403.92205810546875, |
|
"logps/rejected": -408.7096862792969, |
|
"loss": 0.2138, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 6.08463191986084, |
|
"rewards/margins": 44.13505554199219, |
|
"rewards/rejected": -38.0504264831543, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 7.173447537473233e-08, |
|
"logits/chosen": 2.4322450160980225, |
|
"logits/rejected": 3.5664360523223877, |
|
"logps/chosen": -416.0870056152344, |
|
"logps/rejected": -415.59197998046875, |
|
"loss": 0.1237, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 4.357824325561523, |
|
"rewards/margins": 44.89923858642578, |
|
"rewards/rejected": -40.541419982910156, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 7.28051391862955e-08, |
|
"logits/chosen": 2.468834638595581, |
|
"logits/rejected": 3.187020778656006, |
|
"logps/chosen": -395.1986083984375, |
|
"logps/rejected": -432.7027282714844, |
|
"loss": 0.1312, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 7.26413631439209, |
|
"rewards/margins": 49.74661636352539, |
|
"rewards/rejected": -42.48247528076172, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 7.387580299785867e-08, |
|
"logits/chosen": 2.341510772705078, |
|
"logits/rejected": 3.5096206665039062, |
|
"logps/chosen": -391.4788513183594, |
|
"logps/rejected": -392.6119384765625, |
|
"loss": 0.1998, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 6.877418518066406, |
|
"rewards/margins": 45.81779479980469, |
|
"rewards/rejected": -38.94037628173828, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 7.494646680942184e-08, |
|
"logits/chosen": 2.7313547134399414, |
|
"logits/rejected": 3.659938097000122, |
|
"logps/chosen": -375.4259033203125, |
|
"logps/rejected": -431.1378479003906, |
|
"loss": 0.1294, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 4.9859724044799805, |
|
"rewards/margins": 47.63801574707031, |
|
"rewards/rejected": -42.65203857421875, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"eval_logits/chosen": 1.8345167636871338, |
|
"eval_logits/rejected": 2.522487163543701, |
|
"eval_logps/chosen": -403.2203674316406, |
|
"eval_logps/rejected": -410.374755859375, |
|
"eval_loss": 0.40866366028785706, |
|
"eval_rewards/accuracies": 0.953125, |
|
"eval_rewards/chosen": 3.9727911949157715, |
|
"eval_rewards/margins": 44.73374938964844, |
|
"eval_rewards/rejected": -40.760955810546875, |
|
"eval_runtime": 76.9261, |
|
"eval_samples_per_second": 12.999, |
|
"eval_steps_per_second": 0.416, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 7.601713062098501e-08, |
|
"logits/chosen": 2.3793327808380127, |
|
"logits/rejected": 3.142347574234009, |
|
"logps/chosen": -423.20880126953125, |
|
"logps/rejected": -401.803955078125, |
|
"loss": 0.1473, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 8.15241527557373, |
|
"rewards/margins": 47.408851623535156, |
|
"rewards/rejected": -39.256431579589844, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 7.708779443254818e-08, |
|
"logits/chosen": 2.7316575050354004, |
|
"logits/rejected": 3.029636859893799, |
|
"logps/chosen": -434.23480224609375, |
|
"logps/rejected": -403.28717041015625, |
|
"loss": 0.1684, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 10.977544784545898, |
|
"rewards/margins": 50.05615234375, |
|
"rewards/rejected": -39.07860565185547, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 7.815845824411135e-08, |
|
"logits/chosen": 2.5688631534576416, |
|
"logits/rejected": 3.0123863220214844, |
|
"logps/chosen": -449.0638732910156, |
|
"logps/rejected": -398.37152099609375, |
|
"loss": 0.0823, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 8.119828224182129, |
|
"rewards/margins": 45.32478332519531, |
|
"rewards/rejected": -37.2049560546875, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 7.922912205567452e-08, |
|
"logits/chosen": 2.6267566680908203, |
|
"logits/rejected": 3.3151562213897705, |
|
"logps/chosen": -390.72698974609375, |
|
"logps/rejected": -381.4679260253906, |
|
"loss": 0.2223, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 6.810037136077881, |
|
"rewards/margins": 40.27370071411133, |
|
"rewards/rejected": -33.46365737915039, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 8.029978586723767e-08, |
|
"logits/chosen": 2.355883836746216, |
|
"logits/rejected": 3.5080723762512207, |
|
"logps/chosen": -395.70184326171875, |
|
"logps/rejected": -403.0619812011719, |
|
"loss": 0.1436, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 9.782814025878906, |
|
"rewards/margins": 49.49211883544922, |
|
"rewards/rejected": -39.70930480957031, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 8.137044967880086e-08, |
|
"logits/chosen": 2.538801670074463, |
|
"logits/rejected": 2.962019205093384, |
|
"logps/chosen": -395.83062744140625, |
|
"logps/rejected": -439.6985778808594, |
|
"loss": 0.1789, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 6.836484432220459, |
|
"rewards/margins": 59.103851318359375, |
|
"rewards/rejected": -52.267372131347656, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 8.244111349036403e-08, |
|
"logits/chosen": 2.372373104095459, |
|
"logits/rejected": 3.526844024658203, |
|
"logps/chosen": -391.49090576171875, |
|
"logps/rejected": -436.17193603515625, |
|
"loss": 0.1816, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 1.2613162994384766, |
|
"rewards/margins": 54.581298828125, |
|
"rewards/rejected": -53.31998825073242, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 8.351177730192718e-08, |
|
"logits/chosen": 2.641442060470581, |
|
"logits/rejected": 3.504525661468506, |
|
"logps/chosen": -429.4436950683594, |
|
"logps/rejected": -426.1357421875, |
|
"loss": 0.0799, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.4186324179172516, |
|
"rewards/margins": 56.703216552734375, |
|
"rewards/rejected": -57.121849060058594, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 8.458244111349035e-08, |
|
"logits/chosen": 2.8256888389587402, |
|
"logits/rejected": 3.2680656909942627, |
|
"logps/chosen": -432.7039489746094, |
|
"logps/rejected": -435.9918518066406, |
|
"loss": 0.1115, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 0.19469785690307617, |
|
"rewards/margins": 58.439208984375, |
|
"rewards/rejected": -58.2445182800293, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 8.565310492505354e-08, |
|
"logits/chosen": 2.2154085636138916, |
|
"logits/rejected": 3.348832607269287, |
|
"logps/chosen": -443.20269775390625, |
|
"logps/rejected": -411.3150939941406, |
|
"loss": 0.2809, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 7.068073272705078, |
|
"rewards/margins": 56.71687698364258, |
|
"rewards/rejected": -49.648807525634766, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"eval_logits/chosen": 1.8377827405929565, |
|
"eval_logits/rejected": 2.5198471546173096, |
|
"eval_logps/chosen": -403.1432189941406, |
|
"eval_logps/rejected": -412.51531982421875, |
|
"eval_loss": 0.5127639770507812, |
|
"eval_rewards/accuracies": 0.953125, |
|
"eval_rewards/chosen": 4.35873556137085, |
|
"eval_rewards/margins": 55.822689056396484, |
|
"eval_rewards/rejected": -51.463958740234375, |
|
"eval_runtime": 76.8497, |
|
"eval_samples_per_second": 13.012, |
|
"eval_steps_per_second": 0.416, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 8.672376873661669e-08, |
|
"logits/chosen": 2.567979335784912, |
|
"logits/rejected": 3.7970213890075684, |
|
"logps/chosen": -379.3563232421875, |
|
"logps/rejected": -417.9363708496094, |
|
"loss": 0.1456, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 6.566949367523193, |
|
"rewards/margins": 58.88117218017578, |
|
"rewards/rejected": -52.31422805786133, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 8.779443254817986e-08, |
|
"logits/chosen": 2.987666130065918, |
|
"logits/rejected": 3.0821828842163086, |
|
"logps/chosen": -375.0899353027344, |
|
"logps/rejected": -380.730712890625, |
|
"loss": 0.113, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 4.425510406494141, |
|
"rewards/margins": 57.55885696411133, |
|
"rewards/rejected": -53.13335037231445, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 8.886509635974304e-08, |
|
"logits/chosen": 2.3225650787353516, |
|
"logits/rejected": 3.374905824661255, |
|
"logps/chosen": -404.91314697265625, |
|
"logps/rejected": -406.26690673828125, |
|
"loss": 0.221, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 0.956459641456604, |
|
"rewards/margins": 58.1107177734375, |
|
"rewards/rejected": -57.15425491333008, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 8.99357601713062e-08, |
|
"logits/chosen": 2.3407421112060547, |
|
"logits/rejected": 3.029287815093994, |
|
"logps/chosen": -389.4285583496094, |
|
"logps/rejected": -422.2571716308594, |
|
"loss": 0.123, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 5.533617973327637, |
|
"rewards/margins": 57.36924362182617, |
|
"rewards/rejected": -51.83562469482422, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 9.100642398286937e-08, |
|
"logits/chosen": 2.4234728813171387, |
|
"logits/rejected": 3.6180338859558105, |
|
"logps/chosen": -384.3294982910156, |
|
"logps/rejected": -382.1348876953125, |
|
"loss": 0.1292, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 5.42165994644165, |
|
"rewards/margins": 53.802391052246094, |
|
"rewards/rejected": -48.38072967529297, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 9.207708779443254e-08, |
|
"logits/chosen": 2.336247444152832, |
|
"logits/rejected": 3.8205437660217285, |
|
"logps/chosen": -427.49468994140625, |
|
"logps/rejected": -434.2718811035156, |
|
"loss": 0.0955, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 17.532028198242188, |
|
"rewards/margins": 70.969482421875, |
|
"rewards/rejected": -53.437461853027344, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 9.314775160599571e-08, |
|
"logits/chosen": 2.201402187347412, |
|
"logits/rejected": 3.6681246757507324, |
|
"logps/chosen": -437.94091796875, |
|
"logps/rejected": -429.28021240234375, |
|
"loss": 0.0989, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 10.197607040405273, |
|
"rewards/margins": 63.349510192871094, |
|
"rewards/rejected": -53.15190505981445, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 9.421841541755888e-08, |
|
"logits/chosen": 2.64699125289917, |
|
"logits/rejected": 2.797672986984253, |
|
"logps/chosen": -419.5973205566406, |
|
"logps/rejected": -427.1859436035156, |
|
"loss": 0.2619, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 5.267810344696045, |
|
"rewards/margins": 64.8418197631836, |
|
"rewards/rejected": -59.573997497558594, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 9.528907922912205e-08, |
|
"logits/chosen": 2.1525919437408447, |
|
"logits/rejected": 3.06669545173645, |
|
"logps/chosen": -447.0118103027344, |
|
"logps/rejected": -425.5939025878906, |
|
"loss": 0.1568, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 12.983186721801758, |
|
"rewards/margins": 62.67449188232422, |
|
"rewards/rejected": -49.69130325317383, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 9.635974304068522e-08, |
|
"logits/chosen": 2.4299418926239014, |
|
"logits/rejected": 3.4242959022521973, |
|
"logps/chosen": -390.6571350097656, |
|
"logps/rejected": -455.71331787109375, |
|
"loss": 0.2051, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 5.089010715484619, |
|
"rewards/margins": 74.85060119628906, |
|
"rewards/rejected": -69.76158905029297, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"eval_logits/chosen": 1.8021771907806396, |
|
"eval_logits/rejected": 2.491039752960205, |
|
"eval_logps/chosen": -400.8381042480469, |
|
"eval_logps/rejected": -412.0938720703125, |
|
"eval_loss": 0.5601515173912048, |
|
"eval_rewards/accuracies": 0.9375, |
|
"eval_rewards/chosen": 15.884079933166504, |
|
"eval_rewards/margins": 65.24065399169922, |
|
"eval_rewards/rejected": -49.3565673828125, |
|
"eval_runtime": 77.0047, |
|
"eval_samples_per_second": 12.986, |
|
"eval_steps_per_second": 0.416, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 9.743040685224839e-08, |
|
"logits/chosen": 2.6785802841186523, |
|
"logits/rejected": 3.158878803253174, |
|
"logps/chosen": -364.9392395019531, |
|
"logps/rejected": -421.75604248046875, |
|
"loss": 0.1121, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 15.941950798034668, |
|
"rewards/margins": 70.44316101074219, |
|
"rewards/rejected": -54.501220703125, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 9.850107066381156e-08, |
|
"logits/chosen": 2.1345481872558594, |
|
"logits/rejected": 3.1029703617095947, |
|
"logps/chosen": -467.07318115234375, |
|
"logps/rejected": -428.45220947265625, |
|
"loss": 0.2077, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 8.983004570007324, |
|
"rewards/margins": 70.48181915283203, |
|
"rewards/rejected": -61.49879837036133, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 9.957173447537473e-08, |
|
"logits/chosen": 2.542916774749756, |
|
"logits/rejected": 3.5035908222198486, |
|
"logps/chosen": -423.0205993652344, |
|
"logps/rejected": -416.65509033203125, |
|
"loss": 0.2814, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 5.553844928741455, |
|
"rewards/margins": 68.67607116699219, |
|
"rewards/rejected": -63.122222900390625, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 9.992858843132586e-08, |
|
"logits/chosen": 2.491159439086914, |
|
"logits/rejected": 3.259873867034912, |
|
"logps/chosen": -475.2666931152344, |
|
"logps/rejected": -461.07061767578125, |
|
"loss": 0.1522, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 4.259446144104004, |
|
"rewards/margins": 78.9701919555664, |
|
"rewards/rejected": -74.71073913574219, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 9.980956915020233e-08, |
|
"logits/chosen": 2.68623423576355, |
|
"logits/rejected": 3.0718812942504883, |
|
"logps/chosen": -361.89483642578125, |
|
"logps/rejected": -413.39288330078125, |
|
"loss": 0.3246, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 8.68491268157959, |
|
"rewards/margins": 75.4022445678711, |
|
"rewards/rejected": -66.71733093261719, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 9.969054986907879e-08, |
|
"logits/chosen": 2.752331495285034, |
|
"logits/rejected": 3.481398344039917, |
|
"logps/chosen": -392.8135986328125, |
|
"logps/rejected": -413.65313720703125, |
|
"loss": 0.2217, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 3.2605032920837402, |
|
"rewards/margins": 72.24468994140625, |
|
"rewards/rejected": -68.98418426513672, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 9.957153058795524e-08, |
|
"logits/chosen": 2.0877063274383545, |
|
"logits/rejected": 3.1967997550964355, |
|
"logps/chosen": -398.50030517578125, |
|
"logps/rejected": -458.2386779785156, |
|
"loss": 0.1414, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 4.164697170257568, |
|
"rewards/margins": 79.06629943847656, |
|
"rewards/rejected": -74.90159606933594, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 9.94525113068317e-08, |
|
"logits/chosen": 2.6983773708343506, |
|
"logits/rejected": 3.458531141281128, |
|
"logps/chosen": -400.029296875, |
|
"logps/rejected": -411.06903076171875, |
|
"loss": 0.2352, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 4.437505722045898, |
|
"rewards/margins": 78.08453369140625, |
|
"rewards/rejected": -73.64701843261719, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 9.933349202570817e-08, |
|
"logits/chosen": 2.2401540279388428, |
|
"logits/rejected": 3.9224307537078857, |
|
"logps/chosen": -416.910400390625, |
|
"logps/rejected": -441.75482177734375, |
|
"loss": 0.1331, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 7.2108893394470215, |
|
"rewards/margins": 89.5989990234375, |
|
"rewards/rejected": -82.38809967041016, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 9.921447274458463e-08, |
|
"logits/chosen": 2.333087682723999, |
|
"logits/rejected": 3.3145065307617188, |
|
"logps/chosen": -430.5704650878906, |
|
"logps/rejected": -428.00384521484375, |
|
"loss": 0.2009, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -3.077216863632202, |
|
"rewards/margins": 81.33201599121094, |
|
"rewards/rejected": -84.40922546386719, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"eval_logits/chosen": 1.85965096950531, |
|
"eval_logits/rejected": 2.5712876319885254, |
|
"eval_logps/chosen": -405.8722839355469, |
|
"eval_logps/rejected": -420.0833740234375, |
|
"eval_loss": 0.873622715473175, |
|
"eval_rewards/accuracies": 0.9296875, |
|
"eval_rewards/chosen": -9.286641120910645, |
|
"eval_rewards/margins": 80.01751708984375, |
|
"eval_rewards/rejected": -89.30416107177734, |
|
"eval_runtime": 76.9649, |
|
"eval_samples_per_second": 12.993, |
|
"eval_steps_per_second": 0.416, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 9.909545346346108e-08, |
|
"logits/chosen": 2.4723129272460938, |
|
"logits/rejected": 3.478045701980591, |
|
"logps/chosen": -388.63848876953125, |
|
"logps/rejected": -405.98748779296875, |
|
"loss": 0.3397, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 3.27624773979187, |
|
"rewards/margins": 83.57379150390625, |
|
"rewards/rejected": -80.29755401611328, |
|
"step": 1010 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 9.897643418233753e-08, |
|
"logits/chosen": 2.1651980876922607, |
|
"logits/rejected": 3.202363967895508, |
|
"logps/chosen": -439.97894287109375, |
|
"logps/rejected": -464.5369567871094, |
|
"loss": 0.296, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 9.399145126342773, |
|
"rewards/margins": 104.0501480102539, |
|
"rewards/rejected": -94.65100860595703, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 9.885741490121398e-08, |
|
"logits/chosen": 2.6700689792633057, |
|
"logits/rejected": 3.753943681716919, |
|
"logps/chosen": -412.857666015625, |
|
"logps/rejected": -408.2137756347656, |
|
"loss": 0.267, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 17.997615814208984, |
|
"rewards/margins": 86.39213562011719, |
|
"rewards/rejected": -68.39451599121094, |
|
"step": 1030 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 9.873839562009045e-08, |
|
"logits/chosen": 2.110105514526367, |
|
"logits/rejected": 3.5830421447753906, |
|
"logps/chosen": -426.0606994628906, |
|
"logps/rejected": -430.07073974609375, |
|
"loss": 0.1166, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 16.306238174438477, |
|
"rewards/margins": 97.02738952636719, |
|
"rewards/rejected": -80.72114562988281, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 9.861937633896691e-08, |
|
"logits/chosen": 2.830747127532959, |
|
"logits/rejected": 3.9411635398864746, |
|
"logps/chosen": -391.10858154296875, |
|
"logps/rejected": -417.4032287597656, |
|
"loss": 0.3818, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 4.93004846572876, |
|
"rewards/margins": 88.80459594726562, |
|
"rewards/rejected": -83.87454223632812, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 9.850035705784336e-08, |
|
"logits/chosen": 2.3416190147399902, |
|
"logits/rejected": 3.0985264778137207, |
|
"logps/chosen": -461.04718017578125, |
|
"logps/rejected": -448.1722106933594, |
|
"loss": 0.0969, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 14.76380443572998, |
|
"rewards/margins": 97.76150512695312, |
|
"rewards/rejected": -82.99769592285156, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 9.838133777671982e-08, |
|
"logits/chosen": 2.414294719696045, |
|
"logits/rejected": 3.4742074012756348, |
|
"logps/chosen": -384.5827941894531, |
|
"logps/rejected": -398.83050537109375, |
|
"loss": 0.4072, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": 10.109460830688477, |
|
"rewards/margins": 92.21754455566406, |
|
"rewards/rejected": -82.10807037353516, |
|
"step": 1070 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 9.826231849559629e-08, |
|
"logits/chosen": 2.7465944290161133, |
|
"logits/rejected": 3.1911470890045166, |
|
"logps/chosen": -422.5624084472656, |
|
"logps/rejected": -440.286376953125, |
|
"loss": 0.1751, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 17.269311904907227, |
|
"rewards/margins": 104.30892181396484, |
|
"rewards/rejected": -87.03959655761719, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 9.814329921447275e-08, |
|
"logits/chosen": 2.822819232940674, |
|
"logits/rejected": 3.005201816558838, |
|
"logps/chosen": -391.0602111816406, |
|
"logps/rejected": -419.813720703125, |
|
"loss": 0.3021, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 13.785235404968262, |
|
"rewards/margins": 121.23738098144531, |
|
"rewards/rejected": -107.45216369628906, |
|
"step": 1090 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 9.80242799333492e-08, |
|
"logits/chosen": 2.8149497509002686, |
|
"logits/rejected": 3.3617329597473145, |
|
"logps/chosen": -402.799560546875, |
|
"logps/rejected": -425.45343017578125, |
|
"loss": 0.2838, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 14.5247163772583, |
|
"rewards/margins": 105.98973083496094, |
|
"rewards/rejected": -91.46500396728516, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"eval_logits/chosen": 1.8558049201965332, |
|
"eval_logits/rejected": 2.5366647243499756, |
|
"eval_logps/chosen": -402.8280944824219, |
|
"eval_logps/rejected": -421.0693054199219, |
|
"eval_loss": 0.8600718975067139, |
|
"eval_rewards/accuracies": 0.953125, |
|
"eval_rewards/chosen": 5.934433460235596, |
|
"eval_rewards/margins": 100.16818237304688, |
|
"eval_rewards/rejected": -94.23374938964844, |
|
"eval_runtime": 76.8109, |
|
"eval_samples_per_second": 13.019, |
|
"eval_steps_per_second": 0.417, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 9.790526065222565e-08, |
|
"logits/chosen": 2.653266429901123, |
|
"logits/rejected": 3.1961002349853516, |
|
"logps/chosen": -445.80322265625, |
|
"logps/rejected": -429.64373779296875, |
|
"loss": 0.5293, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 21.07932472229004, |
|
"rewards/margins": 100.654052734375, |
|
"rewards/rejected": -79.57472229003906, |
|
"step": 1110 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 9.778624137110211e-08, |
|
"logits/chosen": 2.4288971424102783, |
|
"logits/rejected": 3.4981236457824707, |
|
"logps/chosen": -452.72967529296875, |
|
"logps/rejected": -437.428955078125, |
|
"loss": 0.1353, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 19.18131446838379, |
|
"rewards/margins": 117.1941146850586, |
|
"rewards/rejected": -98.01280212402344, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 9.766722208997857e-08, |
|
"logits/chosen": 2.525935173034668, |
|
"logits/rejected": 3.092148780822754, |
|
"logps/chosen": -402.27886962890625, |
|
"logps/rejected": -428.43280029296875, |
|
"loss": 0.1894, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 20.851654052734375, |
|
"rewards/margins": 118.8115234375, |
|
"rewards/rejected": -97.95986938476562, |
|
"step": 1130 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 9.754820280885503e-08, |
|
"logits/chosen": 2.715815782546997, |
|
"logits/rejected": 4.007362365722656, |
|
"logps/chosen": -399.8186340332031, |
|
"logps/rejected": -423.14862060546875, |
|
"loss": 0.1684, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 9.72397232055664, |
|
"rewards/margins": 105.40447998046875, |
|
"rewards/rejected": -95.68050384521484, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.742918352773148e-08, |
|
"logits/chosen": 2.2262580394744873, |
|
"logits/rejected": 3.2408432960510254, |
|
"logps/chosen": -477.47711181640625, |
|
"logps/rejected": -495.76043701171875, |
|
"loss": 0.1725, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 6.365964412689209, |
|
"rewards/margins": 132.43780517578125, |
|
"rewards/rejected": -126.07185363769531, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.731016424660795e-08, |
|
"logits/chosen": 2.7380359172821045, |
|
"logits/rejected": 3.5110669136047363, |
|
"logps/chosen": -398.9952392578125, |
|
"logps/rejected": -426.838623046875, |
|
"loss": 0.2338, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 12.770757675170898, |
|
"rewards/margins": 123.94891357421875, |
|
"rewards/rejected": -111.17814636230469, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.719114496548441e-08, |
|
"logits/chosen": 2.0883517265319824, |
|
"logits/rejected": 3.5194995403289795, |
|
"logps/chosen": -424.92803955078125, |
|
"logps/rejected": -390.85784912109375, |
|
"loss": 0.3672, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 9.357166290283203, |
|
"rewards/margins": 103.42280578613281, |
|
"rewards/rejected": -94.06563568115234, |
|
"step": 1170 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 9.707212568436087e-08, |
|
"logits/chosen": 2.170081615447998, |
|
"logits/rejected": 3.3753578662872314, |
|
"logps/chosen": -437.91082763671875, |
|
"logps/rejected": -430.90966796875, |
|
"loss": 0.0796, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 8.644025802612305, |
|
"rewards/margins": 108.9352798461914, |
|
"rewards/rejected": -100.2912368774414, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 9.695310640323732e-08, |
|
"logits/chosen": 2.7604737281799316, |
|
"logits/rejected": 3.7521843910217285, |
|
"logps/chosen": -370.9176025390625, |
|
"logps/rejected": -435.0440979003906, |
|
"loss": 0.2187, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 10.98758316040039, |
|
"rewards/margins": 134.98072814941406, |
|
"rewards/rejected": -123.9931411743164, |
|
"step": 1190 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 9.683408712211378e-08, |
|
"logits/chosen": 2.3693931102752686, |
|
"logits/rejected": 3.17862606048584, |
|
"logps/chosen": -467.9139099121094, |
|
"logps/rejected": -460.11297607421875, |
|
"loss": 0.3489, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 3.4328982830047607, |
|
"rewards/margins": 135.82644653320312, |
|
"rewards/rejected": -132.3935546875, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"eval_logits/chosen": 1.8742191791534424, |
|
"eval_logits/rejected": 2.560530662536621, |
|
"eval_logps/chosen": -404.9606628417969, |
|
"eval_logps/rejected": -424.3955383300781, |
|
"eval_loss": 1.1546683311462402, |
|
"eval_rewards/accuracies": 0.9453125, |
|
"eval_rewards/chosen": -4.728623390197754, |
|
"eval_rewards/margins": 106.13644409179688, |
|
"eval_rewards/rejected": -110.86505889892578, |
|
"eval_runtime": 77.052, |
|
"eval_samples_per_second": 12.978, |
|
"eval_steps_per_second": 0.415, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 9.671506784099024e-08, |
|
"logits/chosen": 2.997239589691162, |
|
"logits/rejected": 3.5606212615966797, |
|
"logps/chosen": -392.54229736328125, |
|
"logps/rejected": -423.11322021484375, |
|
"loss": 0.0937, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.05036654323339462, |
|
"rewards/margins": 117.02787780761719, |
|
"rewards/rejected": -117.07823181152344, |
|
"step": 1210 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 9.659604855986669e-08, |
|
"logits/chosen": 2.1404788494110107, |
|
"logits/rejected": 3.323446273803711, |
|
"logps/chosen": -427.2289123535156, |
|
"logps/rejected": -433.6624450683594, |
|
"loss": 0.2752, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 12.247556686401367, |
|
"rewards/margins": 107.3965835571289, |
|
"rewards/rejected": -95.14904022216797, |
|
"step": 1220 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 9.647702927874315e-08, |
|
"logits/chosen": 2.563352346420288, |
|
"logits/rejected": 3.0540013313293457, |
|
"logps/chosen": -447.8446350097656, |
|
"logps/rejected": -452.87969970703125, |
|
"loss": 0.0435, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 23.387487411499023, |
|
"rewards/margins": 119.50389099121094, |
|
"rewards/rejected": -96.11639404296875, |
|
"step": 1230 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 9.63580099976196e-08, |
|
"logits/chosen": 2.1849985122680664, |
|
"logits/rejected": 3.4553771018981934, |
|
"logps/chosen": -387.8475341796875, |
|
"logps/rejected": -393.53192138671875, |
|
"loss": 0.2143, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 26.278972625732422, |
|
"rewards/margins": 115.37886810302734, |
|
"rewards/rejected": -89.09989929199219, |
|
"step": 1240 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 9.623899071649607e-08, |
|
"logits/chosen": 3.139633893966675, |
|
"logits/rejected": 3.37255859375, |
|
"logps/chosen": -398.74005126953125, |
|
"logps/rejected": -418.0592346191406, |
|
"loss": 0.0799, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 22.002050399780273, |
|
"rewards/margins": 125.72208404541016, |
|
"rewards/rejected": -103.72003173828125, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 9.611997143537253e-08, |
|
"logits/chosen": 2.665964126586914, |
|
"logits/rejected": 2.8448374271392822, |
|
"logps/chosen": -427.25677490234375, |
|
"logps/rejected": -430.95025634765625, |
|
"loss": 0.3788, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 21.40536880493164, |
|
"rewards/margins": 111.3967056274414, |
|
"rewards/rejected": -89.9913330078125, |
|
"step": 1260 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 9.600095215424899e-08, |
|
"logits/chosen": 2.7928450107574463, |
|
"logits/rejected": 3.595740795135498, |
|
"logps/chosen": -382.04327392578125, |
|
"logps/rejected": -431.33734130859375, |
|
"loss": 0.3577, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 16.352216720581055, |
|
"rewards/margins": 119.71681213378906, |
|
"rewards/rejected": -103.3646011352539, |
|
"step": 1270 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 9.588193287312544e-08, |
|
"logits/chosen": 2.237015962600708, |
|
"logits/rejected": 2.9541878700256348, |
|
"logps/chosen": -421.40032958984375, |
|
"logps/rejected": -427.09649658203125, |
|
"loss": 0.0439, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 19.52680015563965, |
|
"rewards/margins": 119.0426254272461, |
|
"rewards/rejected": -99.51582336425781, |
|
"step": 1280 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 9.57629135920019e-08, |
|
"logits/chosen": 2.776695728302002, |
|
"logits/rejected": 3.7990450859069824, |
|
"logps/chosen": -398.505859375, |
|
"logps/rejected": -432.1246032714844, |
|
"loss": 0.5895, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 19.98464012145996, |
|
"rewards/margins": 124.7027816772461, |
|
"rewards/rejected": -104.7181625366211, |
|
"step": 1290 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 9.564389431087836e-08, |
|
"logits/chosen": 2.5397887229919434, |
|
"logits/rejected": 3.605865478515625, |
|
"logps/chosen": -403.42132568359375, |
|
"logps/rejected": -455.809814453125, |
|
"loss": 0.1571, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 29.50467872619629, |
|
"rewards/margins": 132.95152282714844, |
|
"rewards/rejected": -103.44685363769531, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"eval_logits/chosen": 1.7718256711959839, |
|
"eval_logits/rejected": 2.484117031097412, |
|
"eval_logps/chosen": -400.06805419921875, |
|
"eval_logps/rejected": -421.99560546875, |
|
"eval_loss": 0.9890532493591309, |
|
"eval_rewards/accuracies": 0.9453125, |
|
"eval_rewards/chosen": 19.734703063964844, |
|
"eval_rewards/margins": 118.60000610351562, |
|
"eval_rewards/rejected": -98.86530303955078, |
|
"eval_runtime": 76.9799, |
|
"eval_samples_per_second": 12.99, |
|
"eval_steps_per_second": 0.416, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 9.552487502975481e-08, |
|
"logits/chosen": 2.1767947673797607, |
|
"logits/rejected": 3.4365925788879395, |
|
"logps/chosen": -440.6006774902344, |
|
"logps/rejected": -473.36700439453125, |
|
"loss": 0.3716, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 15.208317756652832, |
|
"rewards/margins": 130.98887634277344, |
|
"rewards/rejected": -115.78055572509766, |
|
"step": 1310 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 9.540585574863127e-08, |
|
"logits/chosen": 2.517340898513794, |
|
"logits/rejected": 3.1486434936523438, |
|
"logps/chosen": -437.44854736328125, |
|
"logps/rejected": -439.794921875, |
|
"loss": 0.2643, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 2.325850009918213, |
|
"rewards/margins": 115.98299407958984, |
|
"rewards/rejected": -113.65716552734375, |
|
"step": 1320 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 9.528683646750774e-08, |
|
"logits/chosen": 2.605825185775757, |
|
"logits/rejected": 3.62730073928833, |
|
"logps/chosen": -415.30926513671875, |
|
"logps/rejected": -434.9833068847656, |
|
"loss": 0.1454, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 12.735153198242188, |
|
"rewards/margins": 126.29862213134766, |
|
"rewards/rejected": -113.56346130371094, |
|
"step": 1330 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 9.51678171863842e-08, |
|
"logits/chosen": 2.488985776901245, |
|
"logits/rejected": 3.484701156616211, |
|
"logps/chosen": -426.5611877441406, |
|
"logps/rejected": -440.25946044921875, |
|
"loss": 0.5792, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 8.252630233764648, |
|
"rewards/margins": 123.63459777832031, |
|
"rewards/rejected": -115.3819580078125, |
|
"step": 1340 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 9.504879790526065e-08, |
|
"logits/chosen": 2.908905506134033, |
|
"logits/rejected": 3.2668673992156982, |
|
"logps/chosen": -379.87969970703125, |
|
"logps/rejected": -445.2496643066406, |
|
"loss": 0.3025, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 7.975564479827881, |
|
"rewards/margins": 144.19297790527344, |
|
"rewards/rejected": -136.21739196777344, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 9.49297786241371e-08, |
|
"logits/chosen": 2.915165424346924, |
|
"logits/rejected": 3.0565438270568848, |
|
"logps/chosen": -418.11669921875, |
|
"logps/rejected": -470.24554443359375, |
|
"loss": 0.1609, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 5.699041843414307, |
|
"rewards/margins": 146.64541625976562, |
|
"rewards/rejected": -140.94638061523438, |
|
"step": 1360 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 9.481075934301356e-08, |
|
"logits/chosen": 2.8868582248687744, |
|
"logits/rejected": 3.6842854022979736, |
|
"logps/chosen": -373.21185302734375, |
|
"logps/rejected": -419.702392578125, |
|
"loss": 0.1811, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -9.702957153320312, |
|
"rewards/margins": 123.20098876953125, |
|
"rewards/rejected": -132.90394592285156, |
|
"step": 1370 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 9.469174006189002e-08, |
|
"logits/chosen": 2.415966749191284, |
|
"logits/rejected": 3.7415695190429688, |
|
"logps/chosen": -477.23406982421875, |
|
"logps/rejected": -459.58343505859375, |
|
"loss": 0.1843, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 7.450096130371094, |
|
"rewards/margins": 136.1929931640625, |
|
"rewards/rejected": -128.74290466308594, |
|
"step": 1380 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 9.457272078076648e-08, |
|
"logits/chosen": 2.6312685012817383, |
|
"logits/rejected": 3.0948596000671387, |
|
"logps/chosen": -434.4181213378906, |
|
"logps/rejected": -447.225341796875, |
|
"loss": 0.3219, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 3.2366714477539062, |
|
"rewards/margins": 148.01669311523438, |
|
"rewards/rejected": -144.780029296875, |
|
"step": 1390 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 9.445370149964293e-08, |
|
"logits/chosen": 2.789755344390869, |
|
"logits/rejected": 3.598536729812622, |
|
"logps/chosen": -400.4031982421875, |
|
"logps/rejected": -415.2806091308594, |
|
"loss": 0.2459, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 13.055984497070312, |
|
"rewards/margins": 139.63107299804688, |
|
"rewards/rejected": -126.5750961303711, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"eval_logits/chosen": 1.8071796894073486, |
|
"eval_logits/rejected": 2.5058929920196533, |
|
"eval_logps/chosen": -401.0693359375, |
|
"eval_logps/rejected": -428.1545715332031, |
|
"eval_loss": 1.071203589439392, |
|
"eval_rewards/accuracies": 0.9375, |
|
"eval_rewards/chosen": 14.728099822998047, |
|
"eval_rewards/margins": 144.3882293701172, |
|
"eval_rewards/rejected": -129.66012573242188, |
|
"eval_runtime": 76.8152, |
|
"eval_samples_per_second": 13.018, |
|
"eval_steps_per_second": 0.417, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 9.43346822185194e-08, |
|
"logits/chosen": 2.2163331508636475, |
|
"logits/rejected": 3.5338504314422607, |
|
"logps/chosen": -404.0025329589844, |
|
"logps/rejected": -437.22613525390625, |
|
"loss": 0.2835, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 19.502986907958984, |
|
"rewards/margins": 152.5010528564453, |
|
"rewards/rejected": -132.998046875, |
|
"step": 1410 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 9.421566293739586e-08, |
|
"logits/chosen": 2.900045156478882, |
|
"logits/rejected": 3.8935494422912598, |
|
"logps/chosen": -371.714599609375, |
|
"logps/rejected": -410.403564453125, |
|
"loss": 0.1478, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -17.428253173828125, |
|
"rewards/margins": 138.4210968017578, |
|
"rewards/rejected": -155.84933471679688, |
|
"step": 1420 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 9.409664365627231e-08, |
|
"logits/chosen": 2.534538984298706, |
|
"logits/rejected": 2.9124529361724854, |
|
"logps/chosen": -439.07183837890625, |
|
"logps/rejected": -433.35064697265625, |
|
"loss": 0.2963, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 5.632425308227539, |
|
"rewards/margins": 152.6386260986328, |
|
"rewards/rejected": -147.00619506835938, |
|
"step": 1430 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 9.397762437514877e-08, |
|
"logits/chosen": 2.745168685913086, |
|
"logits/rejected": 3.2941577434539795, |
|
"logps/chosen": -414.8768005371094, |
|
"logps/rejected": -474.46258544921875, |
|
"loss": 0.1725, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -17.173574447631836, |
|
"rewards/margins": 160.98873901367188, |
|
"rewards/rejected": -178.16232299804688, |
|
"step": 1440 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 9.385860509402523e-08, |
|
"logits/chosen": 2.712263584136963, |
|
"logits/rejected": 3.576184034347534, |
|
"logps/chosen": -399.8708801269531, |
|
"logps/rejected": -430.77703857421875, |
|
"loss": 0.159, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -4.633294582366943, |
|
"rewards/margins": 155.65737915039062, |
|
"rewards/rejected": -160.29067993164062, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 9.373958581290168e-08, |
|
"logits/chosen": 2.4558093547821045, |
|
"logits/rejected": 3.536351442337036, |
|
"logps/chosen": -453.64324951171875, |
|
"logps/rejected": -463.8458557128906, |
|
"loss": 0.166, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 0.3881942629814148, |
|
"rewards/margins": 154.9169464111328, |
|
"rewards/rejected": -154.52874755859375, |
|
"step": 1460 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 9.362056653177814e-08, |
|
"logits/chosen": 2.787285566329956, |
|
"logits/rejected": 3.214555025100708, |
|
"logps/chosen": -444.3168029785156, |
|
"logps/rejected": -416.84820556640625, |
|
"loss": 0.3787, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": 8.255861282348633, |
|
"rewards/margins": 150.24636840820312, |
|
"rewards/rejected": -141.99050903320312, |
|
"step": 1470 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 9.35015472506546e-08, |
|
"logits/chosen": 2.6908648014068604, |
|
"logits/rejected": 3.7777984142303467, |
|
"logps/chosen": -413.5292053222656, |
|
"logps/rejected": -414.0494079589844, |
|
"loss": 0.4045, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 13.289766311645508, |
|
"rewards/margins": 143.24343872070312, |
|
"rewards/rejected": -129.95367431640625, |
|
"step": 1480 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 9.338252796953105e-08, |
|
"logits/chosen": 2.289970636367798, |
|
"logits/rejected": 3.352263927459717, |
|
"logps/chosen": -403.9881591796875, |
|
"logps/rejected": -431.7518005371094, |
|
"loss": 0.378, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": 20.252124786376953, |
|
"rewards/margins": 172.3944549560547, |
|
"rewards/rejected": -152.142333984375, |
|
"step": 1490 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 9.326350868840752e-08, |
|
"logits/chosen": 3.019411563873291, |
|
"logits/rejected": 3.345460891723633, |
|
"logps/chosen": -434.126220703125, |
|
"logps/rejected": -452.2032165527344, |
|
"loss": 0.537, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": 17.267078399658203, |
|
"rewards/margins": 171.45938110351562, |
|
"rewards/rejected": -154.19232177734375, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"eval_logits/chosen": 1.8465936183929443, |
|
"eval_logits/rejected": 2.4808287620544434, |
|
"eval_logps/chosen": -400.0705871582031, |
|
"eval_logps/rejected": -427.70770263671875, |
|
"eval_loss": 1.2171391248703003, |
|
"eval_rewards/accuracies": 0.9296875, |
|
"eval_rewards/chosen": 19.721826553344727, |
|
"eval_rewards/margins": 147.1477813720703, |
|
"eval_rewards/rejected": -127.42596435546875, |
|
"eval_runtime": 76.9797, |
|
"eval_samples_per_second": 12.99, |
|
"eval_steps_per_second": 0.416, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 9.314448940728398e-08, |
|
"logits/chosen": 2.212067127227783, |
|
"logits/rejected": 3.642113447189331, |
|
"logps/chosen": -408.46063232421875, |
|
"logps/rejected": -423.57464599609375, |
|
"loss": 0.1855, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": 38.055782318115234, |
|
"rewards/margins": 152.7517547607422, |
|
"rewards/rejected": -114.69598388671875, |
|
"step": 1510 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 9.302547012616043e-08, |
|
"logits/chosen": 2.540785074234009, |
|
"logits/rejected": 3.4431614875793457, |
|
"logps/chosen": -421.13995361328125, |
|
"logps/rejected": -475.09716796875, |
|
"loss": 0.2447, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": 10.023767471313477, |
|
"rewards/margins": 167.34268188476562, |
|
"rewards/rejected": -157.3189239501953, |
|
"step": 1520 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 9.290645084503689e-08, |
|
"logits/chosen": 2.7672924995422363, |
|
"logits/rejected": 3.700913190841675, |
|
"logps/chosen": -417.6861877441406, |
|
"logps/rejected": -453.88818359375, |
|
"loss": 0.2154, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": 11.756277084350586, |
|
"rewards/margins": 158.40457153320312, |
|
"rewards/rejected": -146.64828491210938, |
|
"step": 1530 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 9.278743156391336e-08, |
|
"logits/chosen": 2.3169944286346436, |
|
"logits/rejected": 3.3370776176452637, |
|
"logps/chosen": -397.16961669921875, |
|
"logps/rejected": -429.99554443359375, |
|
"loss": 0.5247, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -8.517876625061035, |
|
"rewards/margins": 147.62835693359375, |
|
"rewards/rejected": -156.146240234375, |
|
"step": 1540 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"learning_rate": 9.26684122827898e-08, |
|
"logits/chosen": 2.920701026916504, |
|
"logits/rejected": 3.607771396636963, |
|
"logps/chosen": -409.669189453125, |
|
"logps/rejected": -427.26617431640625, |
|
"loss": 0.6167, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -11.549623489379883, |
|
"rewards/margins": 140.46511840820312, |
|
"rewards/rejected": -152.01473999023438, |
|
"step": 1550 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 9336, |
|
"num_train_epochs": 6, |
|
"save_steps": 500, |
|
"total_flos": 0.0, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|