|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 0.9984, |
|
"eval_steps": 500, |
|
"global_step": 468, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 1.0638297872340426e-07, |
|
"logits/chosen": 0.1359557956457138, |
|
"logits/rejected": 0.030706744641065598, |
|
"logps/chosen": -736.0869140625, |
|
"logps/rejected": -613.6344604492188, |
|
"loss": 2.0331, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 1.0638297872340427e-06, |
|
"logits/chosen": 0.11680144816637039, |
|
"logits/rejected": 0.26637858152389526, |
|
"logps/chosen": -546.4848022460938, |
|
"logps/rejected": -597.5820922851562, |
|
"loss": 2.1606, |
|
"rewards/accuracies": 0.4027777910232544, |
|
"rewards/chosen": 0.00018841998826246709, |
|
"rewards/margins": 0.0001322307507507503, |
|
"rewards/rejected": 5.6189212045865133e-05, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.04, |
|
"learning_rate": 2.1276595744680853e-06, |
|
"logits/chosen": 0.16349577903747559, |
|
"logits/rejected": 0.2681581676006317, |
|
"logps/chosen": -604.7147827148438, |
|
"logps/rejected": -649.5513916015625, |
|
"loss": 2.0961, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.0001730217190925032, |
|
"rewards/margins": -4.76564237033017e-05, |
|
"rewards/rejected": -0.0001253652444574982, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.191489361702128e-06, |
|
"logits/chosen": 0.1495944708585739, |
|
"logits/rejected": 0.19113227725028992, |
|
"logps/chosen": -594.8287353515625, |
|
"logps/rejected": -588.0198974609375, |
|
"loss": 2.124, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.000811461650300771, |
|
"rewards/margins": -7.172574896685546e-06, |
|
"rewards/rejected": -0.0008042889530770481, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.09, |
|
"learning_rate": 4.255319148936171e-06, |
|
"logits/chosen": 0.16839194297790527, |
|
"logits/rejected": 0.23641005158424377, |
|
"logps/chosen": -574.6544799804688, |
|
"logps/rejected": -631.8447875976562, |
|
"loss": 2.1986, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.0023893495090305805, |
|
"rewards/margins": 0.001175779732875526, |
|
"rewards/rejected": -0.0035651293583214283, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.999373573764188e-06, |
|
"logits/chosen": 0.14030753076076508, |
|
"logits/rejected": 0.22485153377056122, |
|
"logps/chosen": -613.718505859375, |
|
"logps/rejected": -637.7477416992188, |
|
"loss": 2.1858, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.01074314396828413, |
|
"rewards/margins": 0.004234404768794775, |
|
"rewards/rejected": -0.014977550134062767, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.988245838331339e-06, |
|
"logits/chosen": 0.16629436612129211, |
|
"logits/rejected": 0.1657789647579193, |
|
"logps/chosen": -639.8953857421875, |
|
"logps/rejected": -674.060302734375, |
|
"loss": 2.1366, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.03131238371133804, |
|
"rewards/margins": 0.011484434828162193, |
|
"rewards/rejected": -0.04279681295156479, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"learning_rate": 4.963268819535228e-06, |
|
"logits/chosen": 0.10423990339040756, |
|
"logits/rejected": 0.11754454672336578, |
|
"logps/chosen": -634.0193481445312, |
|
"logps/rejected": -738.2115478515625, |
|
"loss": 2.1147, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.06819052994251251, |
|
"rewards/margins": 0.03033190593123436, |
|
"rewards/rejected": -0.09852243959903717, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.17, |
|
"learning_rate": 4.9245815365216115e-06, |
|
"logits/chosen": 0.13063886761665344, |
|
"logits/rejected": 0.18018567562103271, |
|
"logps/chosen": -764.638671875, |
|
"logps/rejected": -714.2967529296875, |
|
"loss": 2.1622, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -0.12814757227897644, |
|
"rewards/margins": 0.030911359935998917, |
|
"rewards/rejected": -0.15905895829200745, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.872399318152594e-06, |
|
"logits/chosen": 0.032799430191516876, |
|
"logits/rejected": 0.08405763655900955, |
|
"logps/chosen": -795.3159790039062, |
|
"logps/rejected": -893.0309448242188, |
|
"loss": 2.056, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.20986966788768768, |
|
"rewards/margins": 0.08644559979438782, |
|
"rewards/rejected": -0.2963152527809143, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.807012604511542e-06, |
|
"logits/chosen": 0.07495725154876709, |
|
"logits/rejected": 0.16169913113117218, |
|
"logps/chosen": -909.8048706054688, |
|
"logps/rejected": -995.3268432617188, |
|
"loss": 2.0738, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.2920338809490204, |
|
"rewards/margins": 0.09413494169712067, |
|
"rewards/rejected": -0.38616880774497986, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"learning_rate": 4.728785330347771e-06, |
|
"logits/chosen": 0.16292598843574524, |
|
"logits/rejected": 0.1960914433002472, |
|
"logps/chosen": -904.22509765625, |
|
"logps/rejected": -912.8228759765625, |
|
"loss": 1.9508, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -0.2555181682109833, |
|
"rewards/margins": 0.05460430309176445, |
|
"rewards/rejected": -0.3101224899291992, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.63815289945858e-06, |
|
"logits/chosen": 0.10713698714971542, |
|
"logits/rejected": 0.20941510796546936, |
|
"logps/chosen": -819.4172973632812, |
|
"logps/rejected": -1036.814208984375, |
|
"loss": 1.9156, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.26556941866874695, |
|
"rewards/margins": 0.15225186944007874, |
|
"rewards/rejected": -0.41782131791114807, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"learning_rate": 4.535619761282989e-06, |
|
"logits/chosen": 0.14809398353099823, |
|
"logits/rejected": 0.18617282807826996, |
|
"logps/chosen": -886.8990478515625, |
|
"logps/rejected": -1030.2890625, |
|
"loss": 1.9636, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.32516202330589294, |
|
"rewards/margins": 0.13631486892700195, |
|
"rewards/rejected": -0.4614768922328949, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.42175660319555e-06, |
|
"logits/chosen": 0.18207255005836487, |
|
"logits/rejected": 0.18847420811653137, |
|
"logps/chosen": -927.9578247070312, |
|
"logps/rejected": -1096.78955078125, |
|
"loss": 1.8634, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.3230741322040558, |
|
"rewards/margins": 0.20067179203033447, |
|
"rewards/rejected": -0.5237459540367126, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.297197174127619e-06, |
|
"logits/chosen": 0.17518748342990875, |
|
"logits/rejected": 0.22532868385314941, |
|
"logps/chosen": -905.5735473632812, |
|
"logps/rejected": -1151.804443359375, |
|
"loss": 1.8103, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.3360939621925354, |
|
"rewards/margins": 0.21287350356578827, |
|
"rewards/rejected": -0.5489674210548401, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.162634757195418e-06, |
|
"logits/chosen": 0.2002106010913849, |
|
"logits/rejected": 0.19682054221630096, |
|
"logps/chosen": -903.08935546875, |
|
"logps/rejected": -1042.5498046875, |
|
"loss": 1.9037, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.3114651143550873, |
|
"rewards/margins": 0.1639038622379303, |
|
"rewards/rejected": -0.4753689765930176, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.36, |
|
"learning_rate": 4.018818310967843e-06, |
|
"logits/chosen": 0.20454064011573792, |
|
"logits/rejected": 0.2340986430644989, |
|
"logps/chosen": -755.2179565429688, |
|
"logps/rejected": -950.396484375, |
|
"loss": 1.9158, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.21088416874408722, |
|
"rewards/margins": 0.1926594078540802, |
|
"rewards/rejected": -0.4035435616970062, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.866548300851254e-06, |
|
"logits/chosen": 0.17594218254089355, |
|
"logits/rejected": 0.19749297201633453, |
|
"logps/chosen": -861.9845581054688, |
|
"logps/rejected": -1095.2757568359375, |
|
"loss": 1.8464, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.26217204332351685, |
|
"rewards/margins": 0.22928395867347717, |
|
"rewards/rejected": -0.49145597219467163, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"learning_rate": 3.706672243793271e-06, |
|
"logits/chosen": 0.2364441156387329, |
|
"logits/rejected": 0.29772695899009705, |
|
"logps/chosen": -868.7154541015625, |
|
"logps/rejected": -1106.079345703125, |
|
"loss": 1.8101, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.27426981925964355, |
|
"rewards/margins": 0.2382505238056183, |
|
"rewards/rejected": -0.5125203132629395, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.5400799911032357e-06, |
|
"logits/chosen": 0.22934266924858093, |
|
"logits/rejected": 0.2707999646663666, |
|
"logps/chosen": -949.2401123046875, |
|
"logps/rejected": -1226.9296875, |
|
"loss": 1.7955, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -0.3255212903022766, |
|
"rewards/margins": 0.26814141869544983, |
|
"rewards/rejected": -0.5936626195907593, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3676987756445894e-06, |
|
"logits/chosen": 0.20278143882751465, |
|
"logits/rejected": 0.2634762227535248, |
|
"logps/chosen": -825.6008911132812, |
|
"logps/rejected": -1062.2701416015625, |
|
"loss": 1.8656, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.2474614679813385, |
|
"rewards/margins": 0.25558072328567505, |
|
"rewards/rejected": -0.5030421614646912, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.47, |
|
"learning_rate": 3.1904880509659397e-06, |
|
"logits/chosen": 0.2307882010936737, |
|
"logits/rejected": 0.2746850252151489, |
|
"logps/chosen": -902.4068603515625, |
|
"logps/rejected": -1159.0933837890625, |
|
"loss": 1.7356, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.28473055362701416, |
|
"rewards/margins": 0.2488754689693451, |
|
"rewards/rejected": -0.5336060523986816, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.49, |
|
"learning_rate": 3.0094341510955697e-06, |
|
"logits/chosen": 0.15470555424690247, |
|
"logits/rejected": 0.24446210265159607, |
|
"logps/chosen": -932.212890625, |
|
"logps/rejected": -1326.384033203125, |
|
"loss": 1.8039, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -0.2982610762119293, |
|
"rewards/margins": 0.38883471488952637, |
|
"rewards/rejected": -0.6870958209037781, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.825544800722376e-06, |
|
"logits/chosen": 0.1649903804063797, |
|
"logits/rejected": 0.2850923240184784, |
|
"logps/chosen": -914.9806518554688, |
|
"logps/rejected": -1192.943359375, |
|
"loss": 1.8769, |
|
"rewards/accuracies": 0.5874999761581421, |
|
"rewards/chosen": -0.33413922786712646, |
|
"rewards/margins": 0.2569636404514313, |
|
"rewards/rejected": -0.5911028981208801, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.639843506318899e-06, |
|
"logits/chosen": 0.2421075850725174, |
|
"logits/rejected": 0.23027054965496063, |
|
"logps/chosen": -815.9312744140625, |
|
"logps/rejected": -1075.3148193359375, |
|
"loss": 1.9302, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.2732352614402771, |
|
"rewards/margins": 0.20585504174232483, |
|
"rewards/rejected": -0.47909030318260193, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.55, |
|
"learning_rate": 2.4533638594248094e-06, |
|
"logits/chosen": 0.23280009627342224, |
|
"logits/rejected": 0.27932989597320557, |
|
"logps/chosen": -820.83447265625, |
|
"logps/rejected": -1066.0859375, |
|
"loss": 1.8994, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.24443455040454865, |
|
"rewards/margins": 0.23904922604560852, |
|
"rewards/rejected": -0.4834837317466736, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2671437837980943e-06, |
|
"logits/chosen": 0.20527169108390808, |
|
"logits/rejected": 0.1906355619430542, |
|
"logps/chosen": -802.508544921875, |
|
"logps/rejected": -1058.9862060546875, |
|
"loss": 1.7706, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.23849380016326904, |
|
"rewards/margins": 0.23252327740192413, |
|
"rewards/rejected": -0.47101712226867676, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.6, |
|
"learning_rate": 2.082219758453629e-06, |
|
"logits/chosen": 0.18224522471427917, |
|
"logits/rejected": 0.23733553290367126, |
|
"logps/chosen": -854.2337646484375, |
|
"logps/rejected": -1095.759765625, |
|
"loss": 1.7181, |
|
"rewards/accuracies": 0.65625, |
|
"rewards/chosen": -0.26597216725349426, |
|
"rewards/margins": 0.22445416450500488, |
|
"rewards/rejected": -0.4904263913631439, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.899621048743019e-06, |
|
"logits/chosen": 0.1843213140964508, |
|
"logits/rejected": 0.2989790141582489, |
|
"logps/chosen": -902.1618041992188, |
|
"logps/rejected": -1187.279541015625, |
|
"loss": 1.7856, |
|
"rewards/accuracies": 0.637499988079071, |
|
"rewards/chosen": -0.3390468955039978, |
|
"rewards/margins": 0.2693897485733032, |
|
"rewards/rejected": -0.608436644077301, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.7203639775848423e-06, |
|
"logits/chosen": 0.15538965165615082, |
|
"logits/rejected": 0.25810253620147705, |
|
"logps/chosen": -898.4368896484375, |
|
"logps/rejected": -1099.8011474609375, |
|
"loss": 1.8812, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.32740405201911926, |
|
"rewards/margins": 0.21446183323860168, |
|
"rewards/rejected": -0.5418658256530762, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.5454462687309445e-06, |
|
"logits/chosen": 0.17961570620536804, |
|
"logits/rejected": 0.22687454521656036, |
|
"logps/chosen": -870.6505737304688, |
|
"logps/rejected": -1096.16015625, |
|
"loss": 1.8067, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.2986447811126709, |
|
"rewards/margins": 0.21215729415416718, |
|
"rewards/rejected": -0.5108019709587097, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"learning_rate": 1.3758414935535147e-06, |
|
"logits/chosen": 0.18137860298156738, |
|
"logits/rejected": 0.24465851485729218, |
|
"logps/chosen": -902.7379150390625, |
|
"logps/rejected": -1196.3099365234375, |
|
"loss": 1.68, |
|
"rewards/accuracies": 0.6625000238418579, |
|
"rewards/chosen": -0.3003231883049011, |
|
"rewards/margins": 0.2799701988697052, |
|
"rewards/rejected": -0.5802933573722839, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2124936522614622e-06, |
|
"logits/chosen": 0.18053703010082245, |
|
"logits/rejected": 0.19177524745464325, |
|
"logps/chosen": -886.50341796875, |
|
"logps/rejected": -1139.5965576171875, |
|
"loss": 1.7893, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.3074926435947418, |
|
"rewards/margins": 0.2617393434047699, |
|
"rewards/rejected": -0.5692320466041565, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"learning_rate": 1.0563119197063934e-06, |
|
"logits/chosen": 0.20963367819786072, |
|
"logits/rejected": 0.23218393325805664, |
|
"logps/chosen": -865.2020263671875, |
|
"logps/rejected": -1143.264404296875, |
|
"loss": 1.7222, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.2907507121562958, |
|
"rewards/margins": 0.2580210566520691, |
|
"rewards/rejected": -0.5487717390060425, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 9.081655850224449e-07, |
|
"logits/chosen": 0.16453954577445984, |
|
"logits/rejected": 0.18646371364593506, |
|
"logps/chosen": -938.1325073242188, |
|
"logps/rejected": -1226.00634765625, |
|
"loss": 1.7215, |
|
"rewards/accuracies": 0.6875, |
|
"rewards/chosen": -0.3565717339515686, |
|
"rewards/margins": 0.2777550220489502, |
|
"rewards/rejected": -0.6343267560005188, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.688792132653111e-07, |
|
"logits/chosen": 0.14916151762008667, |
|
"logits/rejected": 0.23365116119384766, |
|
"logps/chosen": -938.5538940429688, |
|
"logps/rejected": -1271.653076171875, |
|
"loss": 1.7912, |
|
"rewards/accuracies": 0.612500011920929, |
|
"rewards/chosen": -0.3168650269508362, |
|
"rewards/margins": 0.3130674362182617, |
|
"rewards/rejected": -0.6299324035644531, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.79, |
|
"learning_rate": 6.392280559802341e-07, |
|
"logits/chosen": 0.20630326867103577, |
|
"logits/rejected": 0.19222524762153625, |
|
"logps/chosen": -968.6492309570312, |
|
"logps/rejected": -1210.966552734375, |
|
"loss": 1.7648, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.35769984126091003, |
|
"rewards/margins": 0.22969777882099152, |
|
"rewards/rejected": -0.5873976945877075, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"learning_rate": 5.199337362431792e-07, |
|
"logits/chosen": 0.23185975849628448, |
|
"logits/rejected": 0.12383759021759033, |
|
"logps/chosen": -913.8956298828125, |
|
"logps/rejected": -1156.3277587890625, |
|
"loss": 1.764, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.3383910357952118, |
|
"rewards/margins": 0.2371567189693451, |
|
"rewards/rejected": -0.5755477547645569, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.1166023219176176e-07, |
|
"logits/chosen": 0.17365404963493347, |
|
"logits/rejected": 0.22495949268341064, |
|
"logps/chosen": -965.8720703125, |
|
"logps/rejected": -1141.7669677734375, |
|
"loss": 1.7503, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -0.3551119267940521, |
|
"rewards/margins": 0.21352621912956238, |
|
"rewards/rejected": -0.5686381459236145, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.150101814011136e-07, |
|
"logits/chosen": 0.12874256074428558, |
|
"logits/rejected": 0.17315654456615448, |
|
"logps/chosen": -858.0895385742188, |
|
"logps/rejected": -1191.4886474609375, |
|
"loss": 1.7221, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.2957189381122589, |
|
"rewards/margins": 0.26001372933387756, |
|
"rewards/rejected": -0.5557327270507812, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.87, |
|
"learning_rate": 2.3052152667409289e-07, |
|
"logits/chosen": 0.16493479907512665, |
|
"logits/rejected": 0.17930065095424652, |
|
"logps/chosen": -871.76025390625, |
|
"logps/rejected": -1177.62353515625, |
|
"loss": 1.8162, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.28667283058166504, |
|
"rewards/margins": 0.30985671281814575, |
|
"rewards/rejected": -0.5965295433998108, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.5866452191498488e-07, |
|
"logits/chosen": 0.165872260928154, |
|
"logits/rejected": 0.2134593427181244, |
|
"logps/chosen": -985.79248046875, |
|
"logps/rejected": -1185.5306396484375, |
|
"loss": 1.8081, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -0.38030657172203064, |
|
"rewards/margins": 0.19018930196762085, |
|
"rewards/rejected": -0.5704958438873291, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.92, |
|
"learning_rate": 9.983911475163727e-08, |
|
"logits/chosen": 0.1297980695962906, |
|
"logits/rejected": 0.2198360413312912, |
|
"logps/chosen": -843.9484252929688, |
|
"logps/rejected": -1071.7698974609375, |
|
"loss": 1.816, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.28988003730773926, |
|
"rewards/margins": 0.22120681405067444, |
|
"rewards/rejected": -0.5110868811607361, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 5.437272047405712e-08, |
|
"logits/chosen": 0.14703018963336945, |
|
"logits/rejected": 0.26481714844703674, |
|
"logps/chosen": -841.8990478515625, |
|
"logps/rejected": -1089.636474609375, |
|
"loss": 1.7983, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.32132893800735474, |
|
"rewards/margins": 0.21181070804595947, |
|
"rewards/rejected": -0.5331395864486694, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.251839967945535e-08, |
|
"logits/chosen": 0.1110246405005455, |
|
"logits/rejected": 0.1886615753173828, |
|
"logps/chosen": -919.92333984375, |
|
"logps/rejected": -1245.7840576171875, |
|
"loss": 1.6896, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.30925479531288147, |
|
"rewards/margins": 0.329929918050766, |
|
"rewards/rejected": -0.6391846537590027, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 4.453449766758933e-09, |
|
"logits/chosen": 0.13569632172584534, |
|
"logits/rejected": 0.18673357367515564, |
|
"logps/chosen": -879.3952026367188, |
|
"logps/rejected": -1108.7191162109375, |
|
"loss": 1.7726, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.35073980689048767, |
|
"rewards/margins": 0.19130484759807587, |
|
"rewards/rejected": -0.5420446991920471, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 468, |
|
"total_flos": 0.0, |
|
"train_loss": 1.4258099234002268, |
|
"train_runtime": 12564.9924, |
|
"train_samples_per_second": 2.388, |
|
"train_steps_per_second": 0.037 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 468, |
|
"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 |
|
} |
|
|