|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.9994666666666666, |
|
"eval_steps": 500, |
|
"global_step": 937, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 5.319148936170213e-08, |
|
"logits/chosen": -0.1885092854499817, |
|
"logits/rejected": -0.3158565163612366, |
|
"logps/chosen": -579.0175170898438, |
|
"logps/rejected": -485.2366638183594, |
|
"loss": 0.2285, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.01, |
|
"learning_rate": 5.319148936170213e-07, |
|
"logits/chosen": -0.1450551152229309, |
|
"logits/rejected": -0.07828879356384277, |
|
"logps/chosen": -490.4548034667969, |
|
"logps/rejected": -480.7243347167969, |
|
"loss": 0.21, |
|
"rewards/accuracies": 0.375, |
|
"rewards/chosen": -5.791801231680438e-05, |
|
"rewards/margins": 4.912323129246943e-05, |
|
"rewards/rejected": -0.00010704125452321023, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.0638297872340427e-06, |
|
"logits/chosen": -0.17377130687236786, |
|
"logits/rejected": -0.2042241394519806, |
|
"logps/chosen": -491.5914001464844, |
|
"logps/rejected": -509.80047607421875, |
|
"loss": 0.2024, |
|
"rewards/accuracies": 0.40625, |
|
"rewards/chosen": 2.5502964490442537e-05, |
|
"rewards/margins": 0.00020013886387459934, |
|
"rewards/rejected": -0.00017463587573729455, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.595744680851064e-06, |
|
"logits/chosen": -0.13665731251239777, |
|
"logits/rejected": -0.13284753262996674, |
|
"logps/chosen": -530.023681640625, |
|
"logps/rejected": -542.8099975585938, |
|
"loss": 0.2154, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.0006009428761899471, |
|
"rewards/margins": 7.30833999114111e-05, |
|
"rewards/rejected": -0.0006740263779647648, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.1276595744680853e-06, |
|
"logits/chosen": -0.10384176671504974, |
|
"logits/rejected": -0.09949521720409393, |
|
"logps/chosen": -578.4327392578125, |
|
"logps/rejected": -594.6671752929688, |
|
"loss": 0.2087, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.0006206859834492207, |
|
"rewards/margins": 0.0007750510121695697, |
|
"rewards/rejected": -0.0013957370538264513, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.6595744680851065e-06, |
|
"logits/chosen": -0.10368921607732773, |
|
"logits/rejected": -0.13414844870567322, |
|
"logps/chosen": -511.35809326171875, |
|
"logps/rejected": -534.3345947265625, |
|
"loss": 0.21, |
|
"rewards/accuracies": 0.4000000059604645, |
|
"rewards/chosen": -0.0013925038510933518, |
|
"rewards/margins": 0.0007838442688807845, |
|
"rewards/rejected": -0.00217634835280478, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.191489361702128e-06, |
|
"logits/chosen": -0.22053027153015137, |
|
"logits/rejected": -0.08493414521217346, |
|
"logps/chosen": -547.6785278320312, |
|
"logps/rejected": -560.6593627929688, |
|
"loss": 0.2124, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.002587032737210393, |
|
"rewards/margins": 0.0012574782595038414, |
|
"rewards/rejected": -0.003844511229544878, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.723404255319149e-06, |
|
"logits/chosen": -0.11390117555856705, |
|
"logits/rejected": -0.09883277863264084, |
|
"logps/chosen": -533.27978515625, |
|
"logps/rejected": -538.4425048828125, |
|
"loss": 0.2137, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.005767723545432091, |
|
"rewards/margins": 0.0026725970674306154, |
|
"rewards/rejected": -0.008440319448709488, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.255319148936171e-06, |
|
"logits/chosen": -0.1157851368188858, |
|
"logits/rejected": -0.17272169888019562, |
|
"logps/chosen": -507.7423400878906, |
|
"logps/rejected": -523.6263427734375, |
|
"loss": 0.2106, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -0.00815755594521761, |
|
"rewards/margins": 0.0021632923744618893, |
|
"rewards/rejected": -0.010320847854018211, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.787234042553192e-06, |
|
"logits/chosen": -0.21070297062397003, |
|
"logits/rejected": -0.20428261160850525, |
|
"logps/chosen": -488.6622009277344, |
|
"logps/rejected": -501.21856689453125, |
|
"loss": 0.2103, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.011966566555202007, |
|
"rewards/margins": 0.002781093353405595, |
|
"rewards/rejected": -0.014747659675776958, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.999375059004058e-06, |
|
"logits/chosen": -0.12732869386672974, |
|
"logits/rejected": -0.2185029536485672, |
|
"logps/chosen": -523.1364135742188, |
|
"logps/rejected": -521.9183349609375, |
|
"loss": 0.2006, |
|
"rewards/accuracies": 0.41874998807907104, |
|
"rewards/chosen": -0.01658087968826294, |
|
"rewards/margins": 0.0014435439370572567, |
|
"rewards/rejected": -0.018024425953626633, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.9955571065548795e-06, |
|
"logits/chosen": -0.23100972175598145, |
|
"logits/rejected": -0.26281121373176575, |
|
"logps/chosen": -547.6220092773438, |
|
"logps/rejected": -568.766845703125, |
|
"loss": 0.202, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.022633420303463936, |
|
"rewards/margins": 0.007052992470562458, |
|
"rewards/rejected": -0.02968641184270382, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.9882736864879e-06, |
|
"logits/chosen": -0.26259124279022217, |
|
"logits/rejected": -0.22674131393432617, |
|
"logps/chosen": -564.6571044921875, |
|
"logps/rejected": -574.2418212890625, |
|
"loss": 0.2018, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.03153581544756889, |
|
"rewards/margins": 0.011116450652480125, |
|
"rewards/rejected": -0.04265226796269417, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.977534912960124e-06, |
|
"logits/chosen": -0.19227923452854156, |
|
"logits/rejected": -0.21840712428092957, |
|
"logps/chosen": -534.7987060546875, |
|
"logps/rejected": -557.68994140625, |
|
"loss": 0.199, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -0.04380916431546211, |
|
"rewards/margins": 0.012184342369437218, |
|
"rewards/rejected": -0.05599350482225418, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.963355698422092e-06, |
|
"logits/chosen": -0.2127476930618286, |
|
"logits/rejected": -0.24109581112861633, |
|
"logps/chosen": -527.8475952148438, |
|
"logps/rejected": -546.1341552734375, |
|
"loss": 0.1976, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.039759356528520584, |
|
"rewards/margins": 0.020705662667751312, |
|
"rewards/rejected": -0.060465019196271896, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.945755732909625e-06, |
|
"logits/chosen": -0.29052820801734924, |
|
"logits/rejected": -0.18534401059150696, |
|
"logps/chosen": -568.7386474609375, |
|
"logps/rejected": -623.6467895507812, |
|
"loss": 0.203, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -0.05505699664354324, |
|
"rewards/margins": 0.02482803724706173, |
|
"rewards/rejected": -0.07988502830266953, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.924759456701167e-06, |
|
"logits/chosen": -0.21109052002429962, |
|
"logits/rejected": -0.23127368092536926, |
|
"logps/chosen": -509.3150329589844, |
|
"logps/rejected": -569.0074462890625, |
|
"loss": 0.1995, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.04530967399477959, |
|
"rewards/margins": 0.026868581771850586, |
|
"rewards/rejected": -0.07217825949192047, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.900396026378671e-06, |
|
"logits/chosen": -0.2945798635482788, |
|
"logits/rejected": -0.25223153829574585, |
|
"logps/chosen": -614.588134765625, |
|
"logps/rejected": -705.424072265625, |
|
"loss": 0.1997, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.07361334562301636, |
|
"rewards/margins": 0.043305329978466034, |
|
"rewards/rejected": -0.11691868305206299, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.872699274339169e-06, |
|
"logits/chosen": -0.2314704954624176, |
|
"logits/rejected": -0.11344774067401886, |
|
"logps/chosen": -570.4790649414062, |
|
"logps/rejected": -626.4537353515625, |
|
"loss": 0.203, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.06511107087135315, |
|
"rewards/margins": 0.03761782869696617, |
|
"rewards/rejected": -0.10272890329360962, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.8417076618132434e-06, |
|
"logits/chosen": -0.13913798332214355, |
|
"logits/rejected": -0.23164553940296173, |
|
"logps/chosen": -599.2550048828125, |
|
"logps/rejected": -628.2545776367188, |
|
"loss": 0.1984, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.06176919490098953, |
|
"rewards/margins": 0.029368698596954346, |
|
"rewards/rejected": -0.09113789349794388, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.807464225455655e-06, |
|
"logits/chosen": -0.1732599437236786, |
|
"logits/rejected": -0.24373570084571838, |
|
"logps/chosen": -551.7747802734375, |
|
"logps/rejected": -587.905517578125, |
|
"loss": 0.1991, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.060781996697187424, |
|
"rewards/margins": 0.024652186781167984, |
|
"rewards/rejected": -0.08543418347835541, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.770016517582283e-06, |
|
"logits/chosen": -0.19644469022750854, |
|
"logits/rejected": -0.19420596957206726, |
|
"logps/chosen": -596.8599853515625, |
|
"logps/rejected": -661.6636962890625, |
|
"loss": 0.1942, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.04881254583597183, |
|
"rewards/margins": 0.03198176622390747, |
|
"rewards/rejected": -0.0807943120598793, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.7294165401363616e-06, |
|
"logits/chosen": -0.2224278450012207, |
|
"logits/rejected": -0.10729773342609406, |
|
"logps/chosen": -604.28662109375, |
|
"logps/rejected": -651.0653076171875, |
|
"loss": 0.186, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.04510096088051796, |
|
"rewards/margins": 0.03683970868587494, |
|
"rewards/rejected": -0.0819406658411026, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.68572067247573e-06, |
|
"logits/chosen": -0.20905578136444092, |
|
"logits/rejected": -0.2025284320116043, |
|
"logps/chosen": -502.0655822753906, |
|
"logps/rejected": -572.6324462890625, |
|
"loss": 0.199, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.05115518718957901, |
|
"rewards/margins": 0.020948706194758415, |
|
"rewards/rejected": -0.07210389524698257, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.638989593081364e-06, |
|
"logits/chosen": -0.1577962040901184, |
|
"logits/rejected": -0.16388146579265594, |
|
"logps/chosen": -501.21551513671875, |
|
"logps/rejected": -542.8304443359375, |
|
"loss": 0.1941, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.03727288171648979, |
|
"rewards/margins": 0.03399407118558884, |
|
"rewards/rejected": -0.07126694917678833, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.5892881952959015e-06, |
|
"logits/chosen": -0.13874894380569458, |
|
"logits/rejected": -0.10486537218093872, |
|
"logps/chosen": -536.9546508789062, |
|
"logps/rejected": -576.2243041992188, |
|
"loss": 0.1903, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.03655445948243141, |
|
"rewards/margins": 0.023150647059082985, |
|
"rewards/rejected": -0.05970510095357895, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.536685497209182e-06, |
|
"logits/chosen": -0.15160498023033142, |
|
"logits/rejected": -0.20615887641906738, |
|
"logps/chosen": -536.8988647460938, |
|
"logps/rejected": -600.5662231445312, |
|
"loss": 0.1915, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.03945186734199524, |
|
"rewards/margins": 0.040184784680604935, |
|
"rewards/rejected": -0.07963664829730988, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.481254545815943e-06, |
|
"logits/chosen": -0.16373318433761597, |
|
"logits/rejected": -0.15171615779399872, |
|
"logps/chosen": -630.7620239257812, |
|
"logps/rejected": -654.4037475585938, |
|
"loss": 0.2068, |
|
"rewards/accuracies": 0.40625, |
|
"rewards/chosen": -0.05563617870211601, |
|
"rewards/margins": 0.019675999879837036, |
|
"rewards/rejected": -0.07531218230724335, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.42307231557875e-06, |
|
"logits/chosen": -0.06917186826467514, |
|
"logits/rejected": -0.1992654800415039, |
|
"logps/chosen": -583.5831909179688, |
|
"logps/rejected": -621.246826171875, |
|
"loss": 0.1911, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.03815799206495285, |
|
"rewards/margins": 0.02716698870062828, |
|
"rewards/rejected": -0.06532497704029083, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.3622196015370305e-06, |
|
"logits/chosen": -0.09893601387739182, |
|
"logits/rejected": -0.15837347507476807, |
|
"logps/chosen": -517.561279296875, |
|
"logps/rejected": -599.6585693359375, |
|
"loss": 0.1935, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.03791649267077446, |
|
"rewards/margins": 0.02659284509718418, |
|
"rewards/rejected": -0.0645093321800232, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.298780907110648e-06, |
|
"logits/chosen": -0.0635145753622055, |
|
"logits/rejected": -0.10447721183300018, |
|
"logps/chosen": -549.46875, |
|
"logps/rejected": -607.9141235351562, |
|
"loss": 0.1957, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.04532330110669136, |
|
"rewards/margins": 0.0257760938256979, |
|
"rewards/rejected": -0.07109940052032471, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.23284432675381e-06, |
|
"logits/chosen": -0.1379682570695877, |
|
"logits/rejected": -0.16269415616989136, |
|
"logps/chosen": -533.80029296875, |
|
"logps/rejected": -575.4193115234375, |
|
"loss": 0.1931, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.05671093985438347, |
|
"rewards/margins": 0.0316205732524395, |
|
"rewards/rejected": -0.08833150565624237, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.164501423622277e-06, |
|
"logits/chosen": -0.19988815486431122, |
|
"logits/rejected": -0.12660416960716248, |
|
"logps/chosen": -556.7050170898438, |
|
"logps/rejected": -533.2452392578125, |
|
"loss": 0.2061, |
|
"rewards/accuracies": 0.4000000059604645, |
|
"rewards/chosen": -0.0512329563498497, |
|
"rewards/margins": 0.006737883202731609, |
|
"rewards/rejected": -0.057970840483903885, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.0938471024237355e-06, |
|
"logits/chosen": -0.12714505195617676, |
|
"logits/rejected": -0.0678124725818634, |
|
"logps/chosen": -551.6757202148438, |
|
"logps/rejected": -634.5169677734375, |
|
"loss": 0.1845, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.04442024976015091, |
|
"rewards/margins": 0.053262293338775635, |
|
"rewards/rejected": -0.09768255054950714, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.020979477627907e-06, |
|
"logits/chosen": -0.08381222188472748, |
|
"logits/rejected": -0.1388164460659027, |
|
"logps/chosen": -567.6676635742188, |
|
"logps/rejected": -620.1175537109375, |
|
"loss": 0.1928, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.04053609445691109, |
|
"rewards/margins": 0.04673929885029793, |
|
"rewards/rejected": -0.08727538585662842, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.9459997372194105e-06, |
|
"logits/chosen": -0.09087273478507996, |
|
"logits/rejected": -0.052247822284698486, |
|
"logps/chosen": -529.18408203125, |
|
"logps/rejected": -596.254638671875, |
|
"loss": 0.1969, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.05547628924250603, |
|
"rewards/margins": 0.04215295612812042, |
|
"rewards/rejected": -0.09762924164533615, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.869012002182573e-06, |
|
"logits/chosen": -0.2067844122648239, |
|
"logits/rejected": -0.17433646321296692, |
|
"logps/chosen": -554.9356689453125, |
|
"logps/rejected": -630.5103759765625, |
|
"loss": 0.1922, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.05108080059289932, |
|
"rewards/margins": 0.03945896774530411, |
|
"rewards/rejected": -0.09053976833820343, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.7901231819133104e-06, |
|
"logits/chosen": -0.11504027992486954, |
|
"logits/rejected": -0.04383224993944168, |
|
"logps/chosen": -577.4075317382812, |
|
"logps/rejected": -607.4254150390625, |
|
"loss": 0.1934, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.044865719974040985, |
|
"rewards/margins": 0.025182927027344704, |
|
"rewards/rejected": -0.07004864513874054, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.709442825758875e-06, |
|
"logits/chosen": -0.16869018971920013, |
|
"logits/rejected": -0.1371726095676422, |
|
"logps/chosen": -584.6583251953125, |
|
"logps/rejected": -648.9109497070312, |
|
"loss": 0.2022, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.06359410285949707, |
|
"rewards/margins": 0.041747674345970154, |
|
"rewards/rejected": -0.10534176975488663, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.6270829708916113e-06, |
|
"logits/chosen": -0.21822664141654968, |
|
"logits/rejected": -0.15734872221946716, |
|
"logps/chosen": -559.946533203125, |
|
"logps/rejected": -605.324951171875, |
|
"loss": 0.1974, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.051652587950229645, |
|
"rewards/margins": 0.02688014507293701, |
|
"rewards/rejected": -0.07853273302316666, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.543157986727991e-06, |
|
"logits/chosen": -0.0996035560965538, |
|
"logits/rejected": -0.1632828712463379, |
|
"logps/chosen": -560.0746459960938, |
|
"logps/rejected": -612.9156494140625, |
|
"loss": 0.201, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.051237620413303375, |
|
"rewards/margins": 0.04104907065629959, |
|
"rewards/rejected": -0.09228668361902237, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.4577844161089614e-06, |
|
"logits/chosen": -0.01575249433517456, |
|
"logits/rejected": -0.1881057471036911, |
|
"logps/chosen": -559.406494140625, |
|
"logps/rejected": -582.1732177734375, |
|
"loss": 0.2061, |
|
"rewards/accuracies": 0.41874998807907104, |
|
"rewards/chosen": -0.03435258939862251, |
|
"rewards/margins": 0.023811260238289833, |
|
"rewards/rejected": -0.05816385895013809, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3710808134621577e-06, |
|
"logits/chosen": -0.1044248715043068, |
|
"logits/rejected": -0.12559100985527039, |
|
"logps/chosen": -518.0560913085938, |
|
"logps/rejected": -559.5712890625, |
|
"loss": 0.2056, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.029239226132631302, |
|
"rewards/margins": 0.02849581278860569, |
|
"rewards/rejected": -0.05773504450917244, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.2831675801707126e-06, |
|
"logits/chosen": -0.14103475213050842, |
|
"logits/rejected": -0.08099827915430069, |
|
"logps/chosen": -546.7091064453125, |
|
"logps/rejected": -608.6934814453125, |
|
"loss": 0.1944, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.028408538550138474, |
|
"rewards/margins": 0.034466955810785294, |
|
"rewards/rejected": -0.06287549436092377, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.194166797377289e-06, |
|
"logits/chosen": -0.1468835175037384, |
|
"logits/rejected": -0.10806012153625488, |
|
"logps/chosen": -539.1943969726562, |
|
"logps/rejected": -607.8096923828125, |
|
"loss": 0.1959, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.016944291070103645, |
|
"rewards/margins": 0.03214774280786514, |
|
"rewards/rejected": -0.04909203201532364, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.104202056455501e-06, |
|
"logits/chosen": -0.07705841958522797, |
|
"logits/rejected": -0.13577719032764435, |
|
"logps/chosen": -534.7425537109375, |
|
"logps/rejected": -590.2757568359375, |
|
"loss": 0.1886, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.027219612151384354, |
|
"rewards/margins": 0.03623734414577484, |
|
"rewards/rejected": -0.0634569600224495, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.013398287384144e-06, |
|
"logits/chosen": -0.05825149267911911, |
|
"logits/rejected": -0.0818571001291275, |
|
"logps/chosen": -504.8671875, |
|
"logps/rejected": -530.8888549804688, |
|
"loss": 0.1971, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.02120174467563629, |
|
"rewards/margins": 0.018600985407829285, |
|
"rewards/rejected": -0.039802733808755875, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.9218815852625717e-06, |
|
"logits/chosen": -0.1541839838027954, |
|
"logits/rejected": -0.06153281405568123, |
|
"logps/chosen": -549.5711669921875, |
|
"logps/rejected": -612.5526733398438, |
|
"loss": 0.1952, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.017424913123250008, |
|
"rewards/margins": 0.02850305661559105, |
|
"rewards/rejected": -0.04592796787619591, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.829779035208113e-06, |
|
"logits/chosen": -0.03961920738220215, |
|
"logits/rejected": -0.1473100781440735, |
|
"logps/chosen": -557.7020874023438, |
|
"logps/rejected": -588.0806884765625, |
|
"loss": 0.1877, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.027860945090651512, |
|
"rewards/margins": 0.024425392970442772, |
|
"rewards/rejected": -0.052286338061094284, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.737218535878705e-06, |
|
"logits/chosen": -0.029465090483427048, |
|
"logits/rejected": -0.08229364454746246, |
|
"logps/chosen": -511.9815368652344, |
|
"logps/rejected": -621.8477172851562, |
|
"loss": 0.1862, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.014062300324440002, |
|
"rewards/margins": 0.042990971356630325, |
|
"rewards/rejected": -0.05705326795578003, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.64432862186579e-06, |
|
"logits/chosen": -0.11332341283559799, |
|
"logits/rejected": -0.022313905879855156, |
|
"logps/chosen": -538.530517578125, |
|
"logps/rejected": -565.3148193359375, |
|
"loss": 0.1903, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.0199245847761631, |
|
"rewards/margins": 0.040730707347393036, |
|
"rewards/rejected": -0.06065528467297554, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.551238285204126e-06, |
|
"logits/chosen": -0.17237767577171326, |
|
"logits/rejected": 0.029280701652169228, |
|
"logps/chosen": -522.3345947265625, |
|
"logps/rejected": -557.0701293945312, |
|
"loss": 0.185, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.02585068717598915, |
|
"rewards/margins": 0.03859715908765793, |
|
"rewards/rejected": -0.06444784253835678, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4580767962463688e-06, |
|
"logits/chosen": -0.1446155458688736, |
|
"logits/rejected": -0.15588733553886414, |
|
"logps/chosen": -495.2469177246094, |
|
"logps/rejected": -525.423095703125, |
|
"loss": 0.2006, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.03724539279937744, |
|
"rewards/margins": 0.0233670175075531, |
|
"rewards/rejected": -0.060612402856349945, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.3649735241511546e-06, |
|
"logits/chosen": -0.10832977294921875, |
|
"logits/rejected": -0.17813685536384583, |
|
"logps/chosen": -487.40252685546875, |
|
"logps/rejected": -573.7322998046875, |
|
"loss": 0.1847, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.022226981818675995, |
|
"rewards/margins": 0.03627447038888931, |
|
"rewards/rejected": -0.05850145220756531, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2720577572339914e-06, |
|
"logits/chosen": -0.12618735432624817, |
|
"logits/rejected": -0.17089055478572845, |
|
"logps/chosen": -491.43927001953125, |
|
"logps/rejected": -632.1901245117188, |
|
"loss": 0.1815, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.0382881760597229, |
|
"rewards/margins": 0.06080668047070503, |
|
"rewards/rejected": -0.09909485280513763, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.1794585234303995e-06, |
|
"logits/chosen": -0.16708405315876007, |
|
"logits/rejected": -0.11356528848409653, |
|
"logps/chosen": -579.8703002929688, |
|
"logps/rejected": -630.3308715820312, |
|
"loss": 0.2008, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.04018039256334305, |
|
"rewards/margins": 0.03480806201696396, |
|
"rewards/rejected": -0.074988454580307, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.0873044111206407e-06, |
|
"logits/chosen": -0.030877122655510902, |
|
"logits/rejected": -0.18640507757663727, |
|
"logps/chosen": -507.278564453125, |
|
"logps/rejected": -627.2957153320312, |
|
"loss": 0.1897, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.040919769555330276, |
|
"rewards/margins": 0.057074230164289474, |
|
"rewards/rejected": -0.09799399971961975, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 1.9957233905648293e-06, |
|
"logits/chosen": -0.17021183669567108, |
|
"logits/rejected": -0.23673537373542786, |
|
"logps/chosen": -544.7388916015625, |
|
"logps/rejected": -587.5650024414062, |
|
"loss": 0.1941, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -0.04041619971394539, |
|
"rewards/margins": 0.024558167904615402, |
|
"rewards/rejected": -0.06497436761856079, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.904842636196402e-06, |
|
"logits/chosen": -0.13388361036777496, |
|
"logits/rejected": -0.0614379458129406, |
|
"logps/chosen": -553.6912841796875, |
|
"logps/rejected": -629.6470336914062, |
|
"loss": 0.1903, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -0.025933533906936646, |
|
"rewards/margins": 0.0378827229142189, |
|
"rewards/rejected": -0.06381625682115555, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.814788350020726e-06, |
|
"logits/chosen": -0.06043483689427376, |
|
"logits/rejected": -0.10723511874675751, |
|
"logps/chosen": -521.6217041015625, |
|
"logps/rejected": -643.6219482421875, |
|
"loss": 0.197, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.024261657148599625, |
|
"rewards/margins": 0.03998683765530586, |
|
"rewards/rejected": -0.06424849480390549, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.725685586364051e-06, |
|
"logits/chosen": -0.1038782149553299, |
|
"logits/rejected": -0.047629911452531815, |
|
"logps/chosen": -535.3336181640625, |
|
"logps/rejected": -564.0821533203125, |
|
"loss": 0.1948, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.024827763438224792, |
|
"rewards/margins": 0.024268008768558502, |
|
"rewards/rejected": -0.049095772206783295, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6376580782162172e-06, |
|
"logits/chosen": -0.12502829730510712, |
|
"logits/rejected": -0.13951411843299866, |
|
"logps/chosen": -565.2417602539062, |
|
"logps/rejected": -653.1436157226562, |
|
"loss": 0.1921, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.027302825823426247, |
|
"rewards/margins": 0.03322170302271843, |
|
"rewards/rejected": -0.060524530708789825, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.550828065408227e-06, |
|
"logits/chosen": -0.15310420095920563, |
|
"logits/rejected": -0.035526055842638016, |
|
"logps/chosen": -510.84759521484375, |
|
"logps/rejected": -573.4019775390625, |
|
"loss": 0.1942, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.02462349459528923, |
|
"rewards/margins": 0.02571946755051613, |
|
"rewards/rejected": -0.05034296587109566, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4653161248633053e-06, |
|
"logits/chosen": -0.12701451778411865, |
|
"logits/rejected": -0.03017173707485199, |
|
"logps/chosen": -507.4302673339844, |
|
"logps/rejected": -554.05029296875, |
|
"loss": 0.1878, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.028983917087316513, |
|
"rewards/margins": 0.03194578364491463, |
|
"rewards/rejected": -0.06092969700694084, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.381241003157162e-06, |
|
"logits/chosen": -0.1072702407836914, |
|
"logits/rejected": -0.12664569914340973, |
|
"logps/chosen": -511.8506774902344, |
|
"logps/rejected": -582.0699462890625, |
|
"loss": 0.199, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.034758515655994415, |
|
"rewards/margins": 0.027269739657640457, |
|
"rewards/rejected": -0.062028251588344574, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.298719451619979e-06, |
|
"logits/chosen": -0.05098239332437515, |
|
"logits/rejected": -0.1476130485534668, |
|
"logps/chosen": -521.5150146484375, |
|
"logps/rejected": -601.8914184570312, |
|
"loss": 0.1877, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.020487060770392418, |
|
"rewards/margins": 0.033522095531225204, |
|
"rewards/rejected": -0.054009150713682175, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2178660642091036e-06, |
|
"logits/chosen": -0.10045331716537476, |
|
"logits/rejected": -0.10961699485778809, |
|
"logps/chosen": -547.9536743164062, |
|
"logps/rejected": -613.6334228515625, |
|
"loss": 0.1825, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.02681289240717888, |
|
"rewards/margins": 0.05183552950620651, |
|
"rewards/rejected": -0.07864842563867569, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.1387931183775821e-06, |
|
"logits/chosen": -0.08650709688663483, |
|
"logits/rejected": -0.17840668559074402, |
|
"logps/chosen": -531.9118041992188, |
|
"logps/rejected": -626.980224609375, |
|
"loss": 0.1847, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.019130777567625046, |
|
"rewards/margins": 0.05559687688946724, |
|
"rewards/rejected": -0.07472765445709229, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.061610419159532e-06, |
|
"logits/chosen": -0.04611722379922867, |
|
"logits/rejected": -0.09121225029230118, |
|
"logps/chosen": -550.1459350585938, |
|
"logps/rejected": -607.9168090820312, |
|
"loss": 0.1874, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.029284963384270668, |
|
"rewards/margins": 0.05334455519914627, |
|
"rewards/rejected": -0.08262950927019119, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.864251466888364e-07, |
|
"logits/chosen": -0.15967205166816711, |
|
"logits/rejected": -0.06603795289993286, |
|
"logps/chosen": -557.4961547851562, |
|
"logps/rejected": -607.64306640625, |
|
"loss": 0.1812, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.034804992377758026, |
|
"rewards/margins": 0.03645947948098183, |
|
"rewards/rejected": -0.07126447558403015, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.133417073629288e-07, |
|
"logits/chosen": -0.15150725841522217, |
|
"logits/rejected": -0.11879472434520721, |
|
"logps/chosen": -504.88665771484375, |
|
"logps/rejected": -582.1925659179688, |
|
"loss": 0.192, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.03086964413523674, |
|
"rewards/margins": 0.04862387850880623, |
|
"rewards/rejected": -0.07949351519346237, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 8.424615888583332e-07, |
|
"logits/chosen": -0.13878367841243744, |
|
"logits/rejected": -0.08197399228811264, |
|
"logps/chosen": -521.1176147460938, |
|
"logps/rejected": -626.109375, |
|
"loss": 0.1877, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.028367381542921066, |
|
"rewards/margins": 0.05848144367337227, |
|
"rewards/rejected": -0.08684882521629333, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.738832191993092e-07, |
|
"logits/chosen": -0.11670303344726562, |
|
"logits/rejected": -0.1110248789191246, |
|
"logps/chosen": -546.02685546875, |
|
"logps/rejected": -633.3377075195312, |
|
"loss": 0.183, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.030107159167528152, |
|
"rewards/margins": 0.04191134124994278, |
|
"rewards/rejected": -0.07201849669218063, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.077018300752917e-07, |
|
"logits/chosen": -0.20101483166217804, |
|
"logits/rejected": -0.09887303411960602, |
|
"logps/chosen": -515.279296875, |
|
"logps/rejected": -566.751220703125, |
|
"loss": 0.1959, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.02799845300614834, |
|
"rewards/margins": 0.039141736924648285, |
|
"rewards/rejected": -0.06714018434286118, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.440093245969342e-07, |
|
"logits/chosen": -0.21685993671417236, |
|
"logits/rejected": -0.014155983924865723, |
|
"logps/chosen": -536.5148315429688, |
|
"logps/rejected": -555.0255126953125, |
|
"loss": 0.1909, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.042235199362039566, |
|
"rewards/margins": 0.033806830644607544, |
|
"rewards/rejected": -0.07604203373193741, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.828941496744075e-07, |
|
"logits/chosen": -0.1709945797920227, |
|
"logits/rejected": -0.09519165009260178, |
|
"logps/chosen": -487.0716247558594, |
|
"logps/rejected": -570.6549072265625, |
|
"loss": 0.1891, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.03168477118015289, |
|
"rewards/margins": 0.05396095663309097, |
|
"rewards/rejected": -0.08564572036266327, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.244411731951671e-07, |
|
"logits/chosen": -0.17727656662464142, |
|
"logits/rejected": -0.11151037365198135, |
|
"logps/chosen": -520.3614501953125, |
|
"logps/rejected": -573.2659301757812, |
|
"loss": 0.1921, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.024441594257950783, |
|
"rewards/margins": 0.03574614226818085, |
|
"rewards/rejected": -0.06018773466348648, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.6873156617173594e-07, |
|
"logits/chosen": -0.16788466274738312, |
|
"logits/rejected": -0.04893454164266586, |
|
"logps/chosen": -508.6934509277344, |
|
"logps/rejected": -575.983154296875, |
|
"loss": 0.1867, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -0.021922405809164047, |
|
"rewards/margins": 0.034182045608758926, |
|
"rewards/rejected": -0.05610444396734238, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.1584269002318653e-07, |
|
"logits/chosen": -0.07019881904125214, |
|
"logits/rejected": -0.18743988871574402, |
|
"logps/chosen": -485.97833251953125, |
|
"logps/rejected": -594.727783203125, |
|
"loss": 0.1866, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.03881939500570297, |
|
"rewards/margins": 0.05370662733912468, |
|
"rewards/rejected": -0.09252602607011795, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.658479891468258e-07, |
|
"logits/chosen": -0.17362000048160553, |
|
"logits/rejected": -0.11806299537420273, |
|
"logps/chosen": -479.9105529785156, |
|
"logps/rejected": -610.3316040039062, |
|
"loss": 0.1924, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.023625101894140244, |
|
"rewards/margins": 0.07324331998825073, |
|
"rewards/rejected": -0.09686841815710068, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.18816888929272e-07, |
|
"logits/chosen": -0.13044703006744385, |
|
"logits/rejected": -0.13462410867214203, |
|
"logps/chosen": -511.01275634765625, |
|
"logps/rejected": -576.4395141601562, |
|
"loss": 0.1845, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.024302612990140915, |
|
"rewards/margins": 0.04658300057053566, |
|
"rewards/rejected": -0.07088561356067657, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.748146993385484e-07, |
|
"logits/chosen": -0.09310281276702881, |
|
"logits/rejected": -0.14107745885849, |
|
"logps/chosen": -542.4686279296875, |
|
"logps/rejected": -590.4827270507812, |
|
"loss": 0.1911, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.03655749931931496, |
|
"rewards/margins": 0.04121191054582596, |
|
"rewards/rejected": -0.07776941359043121, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.3390252423108077e-07, |
|
"logits/chosen": -0.10764355957508087, |
|
"logits/rejected": -0.12145823240280151, |
|
"logps/chosen": -505.4794006347656, |
|
"logps/rejected": -602.5386352539062, |
|
"loss": 0.1825, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.023508111014962196, |
|
"rewards/margins": 0.05395573377609253, |
|
"rewards/rejected": -0.07746385037899017, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.961371764995243e-07, |
|
"logits/chosen": -0.10990991443395615, |
|
"logits/rejected": -0.13984012603759766, |
|
"logps/chosen": -488.1162109375, |
|
"logps/rejected": -533.455810546875, |
|
"loss": 0.1824, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.02799038216471672, |
|
"rewards/margins": 0.03639952093362808, |
|
"rewards/rejected": -0.0643898993730545, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.61571099179261e-07, |
|
"logits/chosen": -0.1293899267911911, |
|
"logits/rejected": -0.13913944363594055, |
|
"logps/chosen": -565.6857299804688, |
|
"logps/rejected": -594.4491577148438, |
|
"loss": 0.1899, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.038701124489307404, |
|
"rewards/margins": 0.03532193973660469, |
|
"rewards/rejected": -0.07402306795120239, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.3025229262312367e-07, |
|
"logits/chosen": -0.11940865218639374, |
|
"logits/rejected": -0.14275464415550232, |
|
"logps/chosen": -517.4210815429688, |
|
"logps/rejected": -559.6951904296875, |
|
"loss": 0.1943, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.029175525531172752, |
|
"rewards/margins": 0.038891829550266266, |
|
"rewards/rejected": -0.06806735694408417, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 1.0222424784546853e-07, |
|
"logits/chosen": -0.14586542546749115, |
|
"logits/rejected": -0.08942988514900208, |
|
"logps/chosen": -556.1111450195312, |
|
"logps/rejected": -648.3941650390625, |
|
"loss": 0.1964, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.03367298096418381, |
|
"rewards/margins": 0.05108407139778137, |
|
"rewards/rejected": -0.08475704491138458, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.752588612816553e-08, |
|
"logits/chosen": -0.1201920285820961, |
|
"logits/rejected": -0.1033986434340477, |
|
"logps/chosen": -535.7073364257812, |
|
"logps/rejected": -629.0509033203125, |
|
"loss": 0.1942, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.03432587534189224, |
|
"rewards/margins": 0.05863683670759201, |
|
"rewards/rejected": -0.09296270459890366, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 5.619150497236991e-08, |
|
"logits/chosen": -0.06793894618749619, |
|
"logits/rejected": -0.14243273437023163, |
|
"logps/chosen": -543.0166625976562, |
|
"logps/rejected": -573.83349609375, |
|
"loss": 0.1839, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.02393261529505253, |
|
"rewards/margins": 0.04055067151784897, |
|
"rewards/rejected": -0.06448328495025635, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 3.825073047112743e-08, |
|
"logits/chosen": -0.09055639803409576, |
|
"logits/rejected": -0.09645718336105347, |
|
"logps/chosen": -589.1398315429688, |
|
"logps/rejected": -638.0357666015625, |
|
"loss": 0.2051, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.0366409607231617, |
|
"rewards/margins": 0.046200644224882126, |
|
"rewards/rejected": -0.08284159749746323, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.372847616895685e-08, |
|
"logits/chosen": -0.24545852839946747, |
|
"logits/rejected": -0.13447916507720947, |
|
"logps/chosen": -500.57177734375, |
|
"logps/rejected": -562.6080322265625, |
|
"loss": 0.1894, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.04158453270792961, |
|
"rewards/margins": 0.03680362552404404, |
|
"rewards/rejected": -0.07838816195726395, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.264490846553279e-08, |
|
"logits/chosen": -0.15378254652023315, |
|
"logits/rejected": -0.09510638564825058, |
|
"logps/chosen": -532.951416015625, |
|
"logps/rejected": -582.2918090820312, |
|
"loss": 0.1947, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.033251263201236725, |
|
"rewards/margins": 0.04195471107959747, |
|
"rewards/rejected": -0.0752059668302536, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 5.015418611516165e-09, |
|
"logits/chosen": -0.06104772537946701, |
|
"logits/rejected": -0.19414536654949188, |
|
"logps/chosen": -515.5169067382812, |
|
"logps/rejected": -601.0435791015625, |
|
"loss": 0.1897, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.03219000995159149, |
|
"rewards/margins": 0.048967912793159485, |
|
"rewards/rejected": -0.08115791529417038, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 8.506013354186993e-10, |
|
"logits/chosen": -0.08989392220973969, |
|
"logits/rejected": -0.16842840611934662, |
|
"logps/chosen": -527.9075927734375, |
|
"logps/rejected": -573.1229858398438, |
|
"loss": 0.1872, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.03477403521537781, |
|
"rewards/margins": 0.039979539811611176, |
|
"rewards/rejected": -0.07475356757640839, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 937, |
|
"total_flos": 0.0, |
|
"train_loss": 0.08862862108611247, |
|
"train_runtime": 3263.0882, |
|
"train_samples_per_second": 9.194, |
|
"train_steps_per_second": 0.287 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 937, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 100, |
|
"total_flos": 0.0, |
|
"train_batch_size": 4, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|