|
{ |
|
"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.1184067577123642, |
|
"logits/rejected": 0.3525714576244354, |
|
"logps/chosen": -429.5767822265625, |
|
"logps/rejected": -514.7810668945312, |
|
"loss": 0.3025, |
|
"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.05111883208155632, |
|
"logits/rejected": 0.23624171316623688, |
|
"logps/chosen": -363.744140625, |
|
"logps/rejected": -491.4056091308594, |
|
"loss": 0.3259, |
|
"rewards/accuracies": 0.3819444477558136, |
|
"rewards/chosen": 2.125546416209545e-05, |
|
"rewards/margins": -4.7190667828544974e-05, |
|
"rewards/rejected": 6.844610470579937e-05, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.0638297872340427e-06, |
|
"logits/chosen": 0.0970318466424942, |
|
"logits/rejected": 0.21818876266479492, |
|
"logps/chosen": -342.73223876953125, |
|
"logps/rejected": -533.9749755859375, |
|
"loss": 0.3389, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.00010573499457677826, |
|
"rewards/margins": 0.00035456178011372685, |
|
"rewards/rejected": -0.00046029678196646273, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.595744680851064e-06, |
|
"logits/chosen": 0.08977697044610977, |
|
"logits/rejected": 0.20492060482501984, |
|
"logps/chosen": -356.282958984375, |
|
"logps/rejected": -506.893798828125, |
|
"loss": 0.3282, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.0007758208666928113, |
|
"rewards/margins": 0.0015017592813819647, |
|
"rewards/rejected": -0.0022775803226977587, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.1276595744680853e-06, |
|
"logits/chosen": 0.14597493410110474, |
|
"logits/rejected": 0.2742587625980377, |
|
"logps/chosen": -373.9236755371094, |
|
"logps/rejected": -478.6368103027344, |
|
"loss": 0.3329, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.0021772016771137714, |
|
"rewards/margins": 0.00360355107113719, |
|
"rewards/rejected": -0.005780753213912249, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.6595744680851065e-06, |
|
"logits/chosen": 0.06719115376472473, |
|
"logits/rejected": 0.20441851019859314, |
|
"logps/chosen": -339.3710021972656, |
|
"logps/rejected": -437.3455505371094, |
|
"loss": 0.3465, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -0.00426669092848897, |
|
"rewards/margins": 0.009421268478035927, |
|
"rewards/rejected": -0.01368795894086361, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.191489361702128e-06, |
|
"logits/chosen": -0.008065430447459221, |
|
"logits/rejected": 0.08582984656095505, |
|
"logps/chosen": -321.49005126953125, |
|
"logps/rejected": -471.43896484375, |
|
"loss": 0.3231, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.008598757907748222, |
|
"rewards/margins": 0.023485153913497925, |
|
"rewards/rejected": -0.032083909958601, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.07, |
|
"learning_rate": 3.723404255319149e-06, |
|
"logits/chosen": 0.06349015235900879, |
|
"logits/rejected": 0.1338900625705719, |
|
"logps/chosen": -400.74725341796875, |
|
"logps/rejected": -535.9130859375, |
|
"loss": 0.3199, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -0.02710326947271824, |
|
"rewards/margins": 0.03926965966820717, |
|
"rewards/rejected": -0.06637293100357056, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.255319148936171e-06, |
|
"logits/chosen": -0.012521197088062763, |
|
"logits/rejected": 0.05875171348452568, |
|
"logps/chosen": -416.5260314941406, |
|
"logps/rejected": -619.4620971679688, |
|
"loss": 0.2653, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.04513772577047348, |
|
"rewards/margins": 0.09761229157447815, |
|
"rewards/rejected": -0.14275000989437103, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.787234042553192e-06, |
|
"logits/chosen": -0.06508742272853851, |
|
"logits/rejected": -0.025008535012602806, |
|
"logps/chosen": -398.4359436035156, |
|
"logps/rejected": -709.9359130859375, |
|
"loss": 0.2408, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.06391867995262146, |
|
"rewards/margins": 0.18030937016010284, |
|
"rewards/rejected": -0.2442280501127243, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.999375059004058e-06, |
|
"logits/chosen": -0.11634773015975952, |
|
"logits/rejected": -0.04804380610585213, |
|
"logps/chosen": -461.4971618652344, |
|
"logps/rejected": -726.8289794921875, |
|
"loss": 0.2717, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.0884505957365036, |
|
"rewards/margins": 0.17610163986682892, |
|
"rewards/rejected": -0.2645522356033325, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.12, |
|
"learning_rate": 4.9955571065548795e-06, |
|
"logits/chosen": -0.07285631448030472, |
|
"logits/rejected": -0.07443198561668396, |
|
"logps/chosen": -428.96588134765625, |
|
"logps/rejected": -709.934326171875, |
|
"loss": 0.2447, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.06827546656131744, |
|
"rewards/margins": 0.1353142112493515, |
|
"rewards/rejected": -0.20358964800834656, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.9882736864879e-06, |
|
"logits/chosen": -0.176306813955307, |
|
"logits/rejected": -0.049706846475601196, |
|
"logps/chosen": -430.32562255859375, |
|
"logps/rejected": -693.2147216796875, |
|
"loss": 0.2456, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.08173463493585587, |
|
"rewards/margins": 0.13715550303459167, |
|
"rewards/rejected": -0.21889011561870575, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.977534912960124e-06, |
|
"logits/chosen": -0.15699656307697296, |
|
"logits/rejected": -0.0577981173992157, |
|
"logps/chosen": -445.72894287109375, |
|
"logps/rejected": -769.3819580078125, |
|
"loss": 0.2081, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.07942022383213043, |
|
"rewards/margins": 0.19334132969379425, |
|
"rewards/rejected": -0.2727615237236023, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.963355698422092e-06, |
|
"logits/chosen": -0.18548890948295593, |
|
"logits/rejected": -0.12828989326953888, |
|
"logps/chosen": -451.9037170410156, |
|
"logps/rejected": -739.3684692382812, |
|
"loss": 0.2257, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.0921342521905899, |
|
"rewards/margins": 0.16765089333057404, |
|
"rewards/rejected": -0.25978511571884155, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.945755732909625e-06, |
|
"logits/chosen": -0.15024694800376892, |
|
"logits/rejected": -0.06675387918949127, |
|
"logps/chosen": -537.3986206054688, |
|
"logps/rejected": -775.892578125, |
|
"loss": 0.2479, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.10632505267858505, |
|
"rewards/margins": 0.16134649515151978, |
|
"rewards/rejected": -0.2676715552806854, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.924759456701167e-06, |
|
"logits/chosen": -0.20865170657634735, |
|
"logits/rejected": -0.10186436027288437, |
|
"logps/chosen": -483.97760009765625, |
|
"logps/rejected": -849.4505004882812, |
|
"loss": 0.2265, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.09985359013080597, |
|
"rewards/margins": 0.2456916868686676, |
|
"rewards/rejected": -0.34554529190063477, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.900396026378671e-06, |
|
"logits/chosen": -0.23064300417900085, |
|
"logits/rejected": -0.12898269295692444, |
|
"logps/chosen": -494.6697692871094, |
|
"logps/rejected": -781.0926513671875, |
|
"loss": 0.2324, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.11788048595190048, |
|
"rewards/margins": 0.17606906592845917, |
|
"rewards/rejected": -0.29394954442977905, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.872699274339169e-06, |
|
"logits/chosen": -0.22916531562805176, |
|
"logits/rejected": -0.14095698297023773, |
|
"logps/chosen": -458.0244140625, |
|
"logps/rejected": -759.4512939453125, |
|
"loss": 0.2393, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -0.12103551626205444, |
|
"rewards/margins": 0.17601092159748077, |
|
"rewards/rejected": -0.29704639315605164, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"learning_rate": 4.8417076618132434e-06, |
|
"logits/chosen": -0.19613122940063477, |
|
"logits/rejected": -0.19725319743156433, |
|
"logps/chosen": -477.3087463378906, |
|
"logps/rejected": -782.9113159179688, |
|
"loss": 0.2591, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.12908069789409637, |
|
"rewards/margins": 0.18730813264846802, |
|
"rewards/rejected": -0.3163888156414032, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.807464225455655e-06, |
|
"logits/chosen": -0.18979570269584656, |
|
"logits/rejected": -0.12616662681102753, |
|
"logps/chosen": -501.4048767089844, |
|
"logps/rejected": -773.6704711914062, |
|
"loss": 0.2318, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.126991406083107, |
|
"rewards/margins": 0.16587643325328827, |
|
"rewards/rejected": -0.29286783933639526, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.770016517582283e-06, |
|
"logits/chosen": -0.19870837032794952, |
|
"logits/rejected": -0.1376388520002365, |
|
"logps/chosen": -476.4922790527344, |
|
"logps/rejected": -797.6036376953125, |
|
"loss": 0.2344, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.12952814996242523, |
|
"rewards/margins": 0.19133572280406952, |
|
"rewards/rejected": -0.32086387276649475, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.7294165401363616e-06, |
|
"logits/chosen": -0.18674388527870178, |
|
"logits/rejected": -0.17149865627288818, |
|
"logps/chosen": -476.748779296875, |
|
"logps/rejected": -736.6768798828125, |
|
"loss": 0.2225, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -0.1292114555835724, |
|
"rewards/margins": 0.17972734570503235, |
|
"rewards/rejected": -0.30893880128860474, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"learning_rate": 4.68572067247573e-06, |
|
"logits/chosen": -0.2134261131286621, |
|
"logits/rejected": -0.17389492690563202, |
|
"logps/chosen": -556.177490234375, |
|
"logps/rejected": -932.4171752929688, |
|
"loss": 0.2011, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.17576465010643005, |
|
"rewards/margins": 0.23368000984191895, |
|
"rewards/rejected": -0.4094447195529938, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.638989593081364e-06, |
|
"logits/chosen": -0.2312520444393158, |
|
"logits/rejected": -0.17490056157112122, |
|
"logps/chosen": -608.5135498046875, |
|
"logps/rejected": -883.7859497070312, |
|
"loss": 0.2251, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.19390198588371277, |
|
"rewards/margins": 0.20176473259925842, |
|
"rewards/rejected": -0.3956666886806488, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.5892881952959015e-06, |
|
"logits/chosen": -0.19589188694953918, |
|
"logits/rejected": -0.16277478635311127, |
|
"logps/chosen": -522.8567504882812, |
|
"logps/rejected": -815.2623291015625, |
|
"loss": 0.2215, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.15903013944625854, |
|
"rewards/margins": 0.19178643822669983, |
|
"rewards/rejected": -0.3508165776729584, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.536685497209182e-06, |
|
"logits/chosen": -0.25143852829933167, |
|
"logits/rejected": -0.15717722475528717, |
|
"logps/chosen": -533.4439697265625, |
|
"logps/rejected": -940.82763671875, |
|
"loss": 0.2056, |
|
"rewards/accuracies": 0.862500011920929, |
|
"rewards/chosen": -0.18470649421215057, |
|
"rewards/margins": 0.22995992004871368, |
|
"rewards/rejected": -0.41466641426086426, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.481254545815943e-06, |
|
"logits/chosen": -0.23728492856025696, |
|
"logits/rejected": -0.22243991494178772, |
|
"logps/chosen": -552.9218139648438, |
|
"logps/rejected": -903.7254028320312, |
|
"loss": 0.196, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.20833177864551544, |
|
"rewards/margins": 0.22902217507362366, |
|
"rewards/rejected": -0.4373539388179779, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.42307231557875e-06, |
|
"logits/chosen": -0.27826839685440063, |
|
"logits/rejected": -0.200110524892807, |
|
"logps/chosen": -523.31787109375, |
|
"logps/rejected": -899.9382934570312, |
|
"loss": 0.1954, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.20382864773273468, |
|
"rewards/margins": 0.2211592197418213, |
|
"rewards/rejected": -0.42498788237571716, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.31, |
|
"learning_rate": 4.3622196015370305e-06, |
|
"logits/chosen": -0.23895792663097382, |
|
"logits/rejected": -0.14843276143074036, |
|
"logps/chosen": -594.8115844726562, |
|
"logps/rejected": -907.5458984375, |
|
"loss": 0.2135, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.19232866168022156, |
|
"rewards/margins": 0.21376939117908478, |
|
"rewards/rejected": -0.40609806776046753, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.298780907110648e-06, |
|
"logits/chosen": -0.24878796935081482, |
|
"logits/rejected": -0.18727460503578186, |
|
"logps/chosen": -580.38671875, |
|
"logps/rejected": -878.1405029296875, |
|
"loss": 0.1938, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.2084374725818634, |
|
"rewards/margins": 0.20776410400867462, |
|
"rewards/rejected": -0.4162015914916992, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"learning_rate": 4.23284432675381e-06, |
|
"logits/chosen": -0.3039627969264984, |
|
"logits/rejected": -0.23918378353118896, |
|
"logps/chosen": -601.3494873046875, |
|
"logps/rejected": -868.8585815429688, |
|
"loss": 0.2041, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.23962783813476562, |
|
"rewards/margins": 0.18101730942726135, |
|
"rewards/rejected": -0.420645147562027, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.164501423622277e-06, |
|
"logits/chosen": -0.3026599884033203, |
|
"logits/rejected": -0.22676105797290802, |
|
"logps/chosen": -559.4996337890625, |
|
"logps/rejected": -871.5550537109375, |
|
"loss": 0.1855, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.21141842007637024, |
|
"rewards/margins": 0.21575455367565155, |
|
"rewards/rejected": -0.427172988653183, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.0938471024237355e-06, |
|
"logits/chosen": -0.24148467183113098, |
|
"logits/rejected": -0.19344112277030945, |
|
"logps/chosen": -570.2223510742188, |
|
"logps/rejected": -958.9056396484375, |
|
"loss": 0.1827, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.22266192734241486, |
|
"rewards/margins": 0.2378120869398117, |
|
"rewards/rejected": -0.46047407388687134, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.020979477627907e-06, |
|
"logits/chosen": -0.23564183712005615, |
|
"logits/rejected": -0.23679451644420624, |
|
"logps/chosen": -623.2242431640625, |
|
"logps/rejected": -871.82568359375, |
|
"loss": 0.2297, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.2473243921995163, |
|
"rewards/margins": 0.19464418292045593, |
|
"rewards/rejected": -0.44196853041648865, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.9459997372194105e-06, |
|
"logits/chosen": -0.22112271189689636, |
|
"logits/rejected": -0.24255314469337463, |
|
"logps/chosen": -654.8084716796875, |
|
"logps/rejected": -992.0069580078125, |
|
"loss": 0.2101, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -0.2653377950191498, |
|
"rewards/margins": 0.21331480145454407, |
|
"rewards/rejected": -0.47865256667137146, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.869012002182573e-06, |
|
"logits/chosen": -0.2565579116344452, |
|
"logits/rejected": -0.23455138504505157, |
|
"logps/chosen": -634.1543579101562, |
|
"logps/rejected": -1048.505859375, |
|
"loss": 0.1816, |
|
"rewards/accuracies": 0.8687499761581421, |
|
"rewards/chosen": -0.246078759431839, |
|
"rewards/margins": 0.2945541739463806, |
|
"rewards/rejected": -0.540632963180542, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.39, |
|
"learning_rate": 3.7901231819133104e-06, |
|
"logits/chosen": -0.27797406911849976, |
|
"logits/rejected": -0.24355745315551758, |
|
"logps/chosen": -634.2208251953125, |
|
"logps/rejected": -972.6463012695312, |
|
"loss": 0.2072, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.2534112334251404, |
|
"rewards/margins": 0.23360367119312286, |
|
"rewards/rejected": -0.48701491951942444, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.709442825758875e-06, |
|
"logits/chosen": -0.2145443856716156, |
|
"logits/rejected": -0.2447199821472168, |
|
"logps/chosen": -630.0811157226562, |
|
"logps/rejected": -895.8094482421875, |
|
"loss": 0.2066, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.2521076202392578, |
|
"rewards/margins": 0.2124219834804535, |
|
"rewards/rejected": -0.4645296037197113, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.6270829708916113e-06, |
|
"logits/chosen": -0.26773589849472046, |
|
"logits/rejected": -0.1856563836336136, |
|
"logps/chosen": -635.20703125, |
|
"logps/rejected": -1008.9361572265625, |
|
"loss": 0.1878, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.2653789818286896, |
|
"rewards/margins": 0.2648240625858307, |
|
"rewards/rejected": -0.5302029848098755, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.543157986727991e-06, |
|
"logits/chosen": -0.24474501609802246, |
|
"logits/rejected": -0.21170318126678467, |
|
"logps/chosen": -730.6715087890625, |
|
"logps/rejected": -1069.3619384765625, |
|
"loss": 0.2077, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.2803260385990143, |
|
"rewards/margins": 0.2544651925563812, |
|
"rewards/rejected": -0.5347911715507507, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.44, |
|
"learning_rate": 3.4577844161089614e-06, |
|
"logits/chosen": -0.1851538121700287, |
|
"logits/rejected": -0.24047665297985077, |
|
"logps/chosen": -616.1502685546875, |
|
"logps/rejected": -903.6608276367188, |
|
"loss": 0.2123, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.24726176261901855, |
|
"rewards/margins": 0.19664901494979858, |
|
"rewards/rejected": -0.44391077756881714, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3710808134621577e-06, |
|
"logits/chosen": -0.2203509360551834, |
|
"logits/rejected": -0.2040051519870758, |
|
"logps/chosen": -636.1844482421875, |
|
"logps/rejected": -1008.9363403320312, |
|
"loss": 0.1868, |
|
"rewards/accuracies": 0.856249988079071, |
|
"rewards/chosen": -0.2344098538160324, |
|
"rewards/margins": 0.2508249282836914, |
|
"rewards/rejected": -0.485234797000885, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.2831675801707126e-06, |
|
"logits/chosen": -0.2882968783378601, |
|
"logits/rejected": -0.28868910670280457, |
|
"logps/chosen": -629.03662109375, |
|
"logps/rejected": -997.5048828125, |
|
"loss": 0.1877, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -0.2724894881248474, |
|
"rewards/margins": 0.2427108734846115, |
|
"rewards/rejected": -0.5152003765106201, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.194166797377289e-06, |
|
"logits/chosen": -0.2842113971710205, |
|
"logits/rejected": -0.2725662589073181, |
|
"logps/chosen": -629.4954223632812, |
|
"logps/rejected": -921.0350341796875, |
|
"loss": 0.1871, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.2583664059638977, |
|
"rewards/margins": 0.23012156784534454, |
|
"rewards/rejected": -0.48848801851272583, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.104202056455501e-06, |
|
"logits/chosen": -0.23092789947986603, |
|
"logits/rejected": -0.2383890151977539, |
|
"logps/chosen": -603.952392578125, |
|
"logps/rejected": -931.93994140625, |
|
"loss": 0.1852, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -0.24851222336292267, |
|
"rewards/margins": 0.2311173975467682, |
|
"rewards/rejected": -0.4796296954154968, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.013398287384144e-06, |
|
"logits/chosen": -0.25257351994514465, |
|
"logits/rejected": -0.20124118030071259, |
|
"logps/chosen": -720.2821044921875, |
|
"logps/rejected": -1016.345703125, |
|
"loss": 0.2169, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.291043221950531, |
|
"rewards/margins": 0.23256464302539825, |
|
"rewards/rejected": -0.5236078500747681, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.9218815852625717e-06, |
|
"logits/chosen": -0.280508816242218, |
|
"logits/rejected": -0.27508029341697693, |
|
"logps/chosen": -599.3237915039062, |
|
"logps/rejected": -959.4754028320312, |
|
"loss": 0.1873, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.266213595867157, |
|
"rewards/margins": 0.23423174023628235, |
|
"rewards/rejected": -0.5004453063011169, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.829779035208113e-06, |
|
"logits/chosen": -0.264869749546051, |
|
"logits/rejected": -0.2316662073135376, |
|
"logps/chosen": -632.74853515625, |
|
"logps/rejected": -977.66845703125, |
|
"loss": 0.1933, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.2532680928707123, |
|
"rewards/margins": 0.23631851375102997, |
|
"rewards/rejected": -0.48958665132522583, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.52, |
|
"learning_rate": 2.737218535878705e-06, |
|
"logits/chosen": -0.26723513007164, |
|
"logits/rejected": -0.2752009928226471, |
|
"logps/chosen": -612.4442749023438, |
|
"logps/rejected": -949.7120971679688, |
|
"loss": 0.1789, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.2443923056125641, |
|
"rewards/margins": 0.23941712081432343, |
|
"rewards/rejected": -0.4838094115257263, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.64432862186579e-06, |
|
"logits/chosen": -0.3223264515399933, |
|
"logits/rejected": -0.29264402389526367, |
|
"logps/chosen": -628.1448974609375, |
|
"logps/rejected": -997.76123046875, |
|
"loss": 0.1908, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.26595020294189453, |
|
"rewards/margins": 0.25043538212776184, |
|
"rewards/rejected": -0.516385555267334, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.551238285204126e-06, |
|
"logits/chosen": -0.2382831573486328, |
|
"logits/rejected": -0.2551121711730957, |
|
"logps/chosen": -653.0089111328125, |
|
"logps/rejected": -1023.2013549804688, |
|
"loss": 0.2152, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.2814193367958069, |
|
"rewards/margins": 0.2521519660949707, |
|
"rewards/rejected": -0.5335713028907776, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4580767962463688e-06, |
|
"logits/chosen": -0.25415441393852234, |
|
"logits/rejected": -0.2852818965911865, |
|
"logps/chosen": -588.615234375, |
|
"logps/rejected": -990.0931396484375, |
|
"loss": 0.1917, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.2602744996547699, |
|
"rewards/margins": 0.24789556860923767, |
|
"rewards/rejected": -0.5081701278686523, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.57, |
|
"learning_rate": 2.3649735241511546e-06, |
|
"logits/chosen": -0.2810899317264557, |
|
"logits/rejected": -0.24050931632518768, |
|
"logps/chosen": -667.4718627929688, |
|
"logps/rejected": -1030.656494140625, |
|
"loss": 0.1788, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.2851766049861908, |
|
"rewards/margins": 0.2829527258872986, |
|
"rewards/rejected": -0.5681293606758118, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2720577572339914e-06, |
|
"logits/chosen": -0.29503872990608215, |
|
"logits/rejected": -0.2738192677497864, |
|
"logps/chosen": -669.0242309570312, |
|
"logps/rejected": -1088.0225830078125, |
|
"loss": 0.1644, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -0.29529038071632385, |
|
"rewards/margins": 0.3003653287887573, |
|
"rewards/rejected": -0.5956557393074036, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.1794585234303995e-06, |
|
"logits/chosen": -0.2950688898563385, |
|
"logits/rejected": -0.1541706621646881, |
|
"logps/chosen": -705.3633422851562, |
|
"logps/rejected": -1031.0267333984375, |
|
"loss": 0.1825, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.2902067303657532, |
|
"rewards/margins": 0.284890741109848, |
|
"rewards/rejected": -0.5750974416732788, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.0873044111206407e-06, |
|
"logits/chosen": -0.28339633345603943, |
|
"logits/rejected": -0.22169962525367737, |
|
"logps/chosen": -629.9029541015625, |
|
"logps/rejected": -1066.449462890625, |
|
"loss": 0.1942, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.2589099705219269, |
|
"rewards/margins": 0.2774832844734192, |
|
"rewards/rejected": -0.5363932847976685, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 1.9957233905648293e-06, |
|
"logits/chosen": -0.24862179160118103, |
|
"logits/rejected": -0.22971436381340027, |
|
"logps/chosen": -636.7313232421875, |
|
"logps/rejected": -972.0720825195312, |
|
"loss": 0.1842, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.2589462399482727, |
|
"rewards/margins": 0.2386230230331421, |
|
"rewards/rejected": -0.4975692629814148, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.904842636196402e-06, |
|
"logits/chosen": -0.2794085443019867, |
|
"logits/rejected": -0.25304803252220154, |
|
"logps/chosen": -655.1509399414062, |
|
"logps/rejected": -1037.134033203125, |
|
"loss": 0.2004, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -0.2735671103000641, |
|
"rewards/margins": 0.27847912907600403, |
|
"rewards/rejected": -0.5520461797714233, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"learning_rate": 1.814788350020726e-06, |
|
"logits/chosen": -0.2863375246524811, |
|
"logits/rejected": -0.2578074336051941, |
|
"logps/chosen": -644.8294067382812, |
|
"logps/rejected": -978.0472412109375, |
|
"loss": 0.1852, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.27340829372406006, |
|
"rewards/margins": 0.256661593914032, |
|
"rewards/rejected": -0.5300698280334473, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.725685586364051e-06, |
|
"logits/chosen": -0.30925947427749634, |
|
"logits/rejected": -0.26838865876197815, |
|
"logps/chosen": -632.4716796875, |
|
"logps/rejected": -994.3053588867188, |
|
"loss": 0.2151, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.27847468852996826, |
|
"rewards/margins": 0.25020256638526917, |
|
"rewards/rejected": -0.5286772847175598, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.65, |
|
"learning_rate": 1.6376580782162172e-06, |
|
"logits/chosen": -0.20261721312999725, |
|
"logits/rejected": -0.221277117729187, |
|
"logps/chosen": -634.4677124023438, |
|
"logps/rejected": -994.3683471679688, |
|
"loss": 0.1934, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.25305241346359253, |
|
"rewards/margins": 0.2600497603416443, |
|
"rewards/rejected": -0.5131021738052368, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.550828065408227e-06, |
|
"logits/chosen": -0.293588250875473, |
|
"logits/rejected": -0.250821977853775, |
|
"logps/chosen": -588.5574951171875, |
|
"logps/rejected": -976.6282348632812, |
|
"loss": 0.1876, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -0.26154106855392456, |
|
"rewards/margins": 0.25152388215065, |
|
"rewards/rejected": -0.5130649209022522, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4653161248633053e-06, |
|
"logits/chosen": -0.25644367933273315, |
|
"logits/rejected": -0.2529579699039459, |
|
"logps/chosen": -637.1114501953125, |
|
"logps/rejected": -1041.966064453125, |
|
"loss": 0.177, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -0.26990941166877747, |
|
"rewards/margins": 0.2800416052341461, |
|
"rewards/rejected": -0.5499509572982788, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.381241003157162e-06, |
|
"logits/chosen": -0.25759977102279663, |
|
"logits/rejected": -0.22868645191192627, |
|
"logps/chosen": -585.88671875, |
|
"logps/rejected": -971.8045043945312, |
|
"loss": 0.1695, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.24511754512786865, |
|
"rewards/margins": 0.2909080982208252, |
|
"rewards/rejected": -0.5360256433486938, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.298719451619979e-06, |
|
"logits/chosen": -0.27405595779418945, |
|
"logits/rejected": -0.25046929717063904, |
|
"logps/chosen": -686.3485107421875, |
|
"logps/rejected": -1069.6209716796875, |
|
"loss": 0.1779, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.29048749804496765, |
|
"rewards/margins": 0.3021007776260376, |
|
"rewards/rejected": -0.5925883054733276, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2178660642091036e-06, |
|
"logits/chosen": -0.258391410112381, |
|
"logits/rejected": -0.2719349265098572, |
|
"logps/chosen": -642.6732177734375, |
|
"logps/rejected": -1080.385498046875, |
|
"loss": 0.1866, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.2909965515136719, |
|
"rewards/margins": 0.3018215596675873, |
|
"rewards/rejected": -0.5928180813789368, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"learning_rate": 1.1387931183775821e-06, |
|
"logits/chosen": -0.29932543635368347, |
|
"logits/rejected": -0.2793118357658386, |
|
"logps/chosen": -649.6714477539062, |
|
"logps/rejected": -1090.2452392578125, |
|
"loss": 0.1646, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -0.2679978013038635, |
|
"rewards/margins": 0.30532750487327576, |
|
"rewards/rejected": -0.5733253359794617, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.061610419159532e-06, |
|
"logits/chosen": -0.27962726354599, |
|
"logits/rejected": -0.21005702018737793, |
|
"logps/chosen": -654.9832763671875, |
|
"logps/rejected": -1042.5281982421875, |
|
"loss": 0.195, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -0.28967055678367615, |
|
"rewards/margins": 0.28377145528793335, |
|
"rewards/rejected": -0.5734419822692871, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.864251466888364e-07, |
|
"logits/chosen": -0.2319578379392624, |
|
"logits/rejected": -0.22523808479309082, |
|
"logps/chosen": -587.6854248046875, |
|
"logps/rejected": -929.84716796875, |
|
"loss": 0.1957, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.2619277834892273, |
|
"rewards/margins": 0.24305231869220734, |
|
"rewards/rejected": -0.5049800276756287, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.133417073629288e-07, |
|
"logits/chosen": -0.2581741213798523, |
|
"logits/rejected": -0.22566178441047668, |
|
"logps/chosen": -673.9590454101562, |
|
"logps/rejected": -1043.8779296875, |
|
"loss": 0.1946, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.2741592526435852, |
|
"rewards/margins": 0.2668037712574005, |
|
"rewards/rejected": -0.5409630537033081, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"learning_rate": 8.424615888583332e-07, |
|
"logits/chosen": -0.24903810024261475, |
|
"logits/rejected": -0.19368259608745575, |
|
"logps/chosen": -702.1976318359375, |
|
"logps/rejected": -1104.967529296875, |
|
"loss": 0.1768, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.29275885224342346, |
|
"rewards/margins": 0.24981990456581116, |
|
"rewards/rejected": -0.5425786972045898, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.738832191993092e-07, |
|
"logits/chosen": -0.2902497947216034, |
|
"logits/rejected": -0.27156323194503784, |
|
"logps/chosen": -642.2786865234375, |
|
"logps/rejected": -1030.1949462890625, |
|
"loss": 0.1896, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.296554833650589, |
|
"rewards/margins": 0.2522734999656677, |
|
"rewards/rejected": -0.5488283634185791, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 7.077018300752917e-07, |
|
"logits/chosen": -0.25113141536712646, |
|
"logits/rejected": -0.27631789445877075, |
|
"logps/chosen": -658.340087890625, |
|
"logps/rejected": -1025.861328125, |
|
"loss": 0.1804, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.2932133376598358, |
|
"rewards/margins": 0.24299320578575134, |
|
"rewards/rejected": -0.5362066030502319, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.440093245969342e-07, |
|
"logits/chosen": -0.2601129412651062, |
|
"logits/rejected": -0.23299941420555115, |
|
"logps/chosen": -661.2008056640625, |
|
"logps/rejected": -1049.9329833984375, |
|
"loss": 0.1731, |
|
"rewards/accuracies": 0.84375, |
|
"rewards/chosen": -0.28521472215652466, |
|
"rewards/margins": 0.2829175591468811, |
|
"rewards/rejected": -0.5681322813034058, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.828941496744075e-07, |
|
"logits/chosen": -0.2920230031013489, |
|
"logits/rejected": -0.2706466317176819, |
|
"logps/chosen": -614.7430419921875, |
|
"logps/rejected": -1084.22802734375, |
|
"loss": 0.1975, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -0.26638248562812805, |
|
"rewards/margins": 0.29731154441833496, |
|
"rewards/rejected": -0.5636940598487854, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.244411731951671e-07, |
|
"logits/chosen": -0.2530083954334259, |
|
"logits/rejected": -0.2159876525402069, |
|
"logps/chosen": -636.4935913085938, |
|
"logps/rejected": -965.4192504882812, |
|
"loss": 0.1974, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.27532270550727844, |
|
"rewards/margins": 0.23001885414123535, |
|
"rewards/rejected": -0.5053415298461914, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.6873156617173594e-07, |
|
"logits/chosen": -0.2676984369754791, |
|
"logits/rejected": -0.2522314190864563, |
|
"logps/chosen": -619.7949829101562, |
|
"logps/rejected": -1006.8516845703125, |
|
"loss": 0.1934, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.25617390871047974, |
|
"rewards/margins": 0.25205108523368835, |
|
"rewards/rejected": -0.5082249641418457, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.1584269002318653e-07, |
|
"logits/chosen": -0.276606023311615, |
|
"logits/rejected": -0.18771035969257355, |
|
"logps/chosen": -680.2154541015625, |
|
"logps/rejected": -1099.0855712890625, |
|
"loss": 0.1918, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.2760952413082123, |
|
"rewards/margins": 0.2856435477733612, |
|
"rewards/rejected": -0.5617388486862183, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"learning_rate": 3.658479891468258e-07, |
|
"logits/chosen": -0.2671615183353424, |
|
"logits/rejected": -0.28831931948661804, |
|
"logps/chosen": -652.9794921875, |
|
"logps/rejected": -1049.533447265625, |
|
"loss": 0.1852, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -0.2673850953578949, |
|
"rewards/margins": 0.2847273647785187, |
|
"rewards/rejected": -0.5521124601364136, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.18816888929272e-07, |
|
"logits/chosen": -0.2824237644672394, |
|
"logits/rejected": -0.2227623015642166, |
|
"logps/chosen": -643.0541381835938, |
|
"logps/rejected": -957.2637939453125, |
|
"loss": 0.207, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.2690204679965973, |
|
"rewards/margins": 0.2505555748939514, |
|
"rewards/rejected": -0.5195759534835815, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.748146993385484e-07, |
|
"logits/chosen": -0.3143889307975769, |
|
"logits/rejected": -0.2759067118167877, |
|
"logps/chosen": -651.569580078125, |
|
"logps/rejected": -1108.028076171875, |
|
"loss": 0.1587, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.2932509779930115, |
|
"rewards/margins": 0.2749115526676178, |
|
"rewards/rejected": -0.5681625604629517, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.3390252423108077e-07, |
|
"logits/chosen": -0.26839888095855713, |
|
"logits/rejected": -0.23119351267814636, |
|
"logps/chosen": -665.0179443359375, |
|
"logps/rejected": -1112.646728515625, |
|
"loss": 0.1658, |
|
"rewards/accuracies": 0.8687499761581421, |
|
"rewards/chosen": -0.28339672088623047, |
|
"rewards/margins": 0.3101740777492523, |
|
"rewards/rejected": -0.5935708284378052, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"learning_rate": 1.961371764995243e-07, |
|
"logits/chosen": -0.2762928605079651, |
|
"logits/rejected": -0.28436392545700073, |
|
"logps/chosen": -631.9949951171875, |
|
"logps/rejected": -1104.309326171875, |
|
"loss": 0.1821, |
|
"rewards/accuracies": 0.862500011920929, |
|
"rewards/chosen": -0.2872810959815979, |
|
"rewards/margins": 0.303080290555954, |
|
"rewards/rejected": -0.5903614163398743, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.61571099179261e-07, |
|
"logits/chosen": -0.2735849916934967, |
|
"logits/rejected": -0.2402714043855667, |
|
"logps/chosen": -652.4813842773438, |
|
"logps/rejected": -1141.485107421875, |
|
"loss": 0.1638, |
|
"rewards/accuracies": 0.8687499761581421, |
|
"rewards/chosen": -0.2600463032722473, |
|
"rewards/margins": 0.34138140082359314, |
|
"rewards/rejected": -0.6014277338981628, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.3025229262312367e-07, |
|
"logits/chosen": -0.20563116669654846, |
|
"logits/rejected": -0.25282037258148193, |
|
"logps/chosen": -693.9833984375, |
|
"logps/rejected": -1084.752197265625, |
|
"loss": 0.183, |
|
"rewards/accuracies": 0.84375, |
|
"rewards/chosen": -0.2829940617084503, |
|
"rewards/margins": 0.27976807951927185, |
|
"rewards/rejected": -0.5627621412277222, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 1.0222424784546853e-07, |
|
"logits/chosen": -0.22862930595874786, |
|
"logits/rejected": -0.21136240661144257, |
|
"logps/chosen": -638.4495239257812, |
|
"logps/rejected": -1061.037109375, |
|
"loss": 0.1807, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -0.2733309268951416, |
|
"rewards/margins": 0.28016480803489685, |
|
"rewards/rejected": -0.5534957647323608, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.752588612816553e-08, |
|
"logits/chosen": -0.2907347083091736, |
|
"logits/rejected": -0.2525694966316223, |
|
"logps/chosen": -626.2022705078125, |
|
"logps/rejected": -999.1546630859375, |
|
"loss": 0.19, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -0.2700883746147156, |
|
"rewards/margins": 0.2750420868396759, |
|
"rewards/rejected": -0.5451304316520691, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 5.619150497236991e-08, |
|
"logits/chosen": -0.2824448347091675, |
|
"logits/rejected": -0.21472208201885223, |
|
"logps/chosen": -694.7796630859375, |
|
"logps/rejected": -1101.357666015625, |
|
"loss": 0.1808, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.30949831008911133, |
|
"rewards/margins": 0.29203012585639954, |
|
"rewards/rejected": -0.601528525352478, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.95, |
|
"learning_rate": 3.825073047112743e-08, |
|
"logits/chosen": -0.2339903563261032, |
|
"logits/rejected": -0.25414004921913147, |
|
"logps/chosen": -677.6926879882812, |
|
"logps/rejected": -1110.830810546875, |
|
"loss": 0.175, |
|
"rewards/accuracies": 0.8812500238418579, |
|
"rewards/chosen": -0.2908310890197754, |
|
"rewards/margins": 0.3078031539916992, |
|
"rewards/rejected": -0.5986341834068298, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.372847616895685e-08, |
|
"logits/chosen": -0.2827141582965851, |
|
"logits/rejected": -0.24739205837249756, |
|
"logps/chosen": -659.048583984375, |
|
"logps/rejected": -1077.357177734375, |
|
"loss": 0.1941, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.28590255975723267, |
|
"rewards/margins": 0.280953586101532, |
|
"rewards/rejected": -0.5668561458587646, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.97, |
|
"learning_rate": 1.264490846553279e-08, |
|
"logits/chosen": -0.29719191789627075, |
|
"logits/rejected": -0.23341615498065948, |
|
"logps/chosen": -676.8151245117188, |
|
"logps/rejected": -1061.142822265625, |
|
"loss": 0.1835, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.2869683802127838, |
|
"rewards/margins": 0.2846153676509857, |
|
"rewards/rejected": -0.57158362865448, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 5.015418611516165e-09, |
|
"logits/chosen": -0.22863967716693878, |
|
"logits/rejected": -0.20865638554096222, |
|
"logps/chosen": -656.0875854492188, |
|
"logps/rejected": -1003.3201904296875, |
|
"loss": 0.1869, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.2743567228317261, |
|
"rewards/margins": 0.260437935590744, |
|
"rewards/rejected": -0.5347946286201477, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 8.506013354186993e-10, |
|
"logits/chosen": -0.22728657722473145, |
|
"logits/rejected": -0.21279653906822205, |
|
"logps/chosen": -687.8726806640625, |
|
"logps/rejected": -1077.039306640625, |
|
"loss": 0.1753, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.28496915102005005, |
|
"rewards/margins": 0.2890021502971649, |
|
"rewards/rejected": -0.5739713907241821, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 937, |
|
"total_flos": 0.0, |
|
"train_loss": 0.08644322651809155, |
|
"train_runtime": 5283.9744, |
|
"train_samples_per_second": 5.678, |
|
"train_steps_per_second": 0.177 |
|
} |
|
], |
|
"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 |
|
} |
|
|