BraylonDash's picture
Model save
74f0d63 verified
{
"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
}