qwen2.5-0.5b-expo-DPO-EXPERIMENT / trainer_state.json
hZzy's picture
Model save
f2b2deb verified
raw
history blame
40.7 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.995276334435522,
"eval_steps": 50,
"global_step": 352,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"dpo_loss": 0.6931471824645996,
"epoch": 0.005668398677373642,
"grad_norm": 13.413637591400326,
"learning_rate": 2.7777777777777776e-09,
"logits": -1.3147305250167847,
"logps": -88.0877456665039,
"loss": 0.6931,
"objective": 0.6931471824645996,
"ranking_idealized": 0.625,
"ranking_idealized_expo": 0.5208333134651184,
"ranking_simple": 0.5208333134651184,
"regularize": 0.6931471824645996,
"step": 1
},
{
"dpo_loss": 0.6931517124176025,
"epoch": 0.02834199338686821,
"grad_norm": 13.314796352051717,
"learning_rate": 1.3888888888888889e-08,
"logits": -1.3678232431411743,
"logps": -84.45550537109375,
"loss": 0.6931,
"objective": 0.6931517124176025,
"ranking_idealized": 0.6145833134651184,
"ranking_idealized_expo": 0.546875,
"ranking_simple": 0.546875,
"regularize": 0.6931517124176025,
"step": 5
},
{
"dpo_loss": 0.6927531957626343,
"epoch": 0.05668398677373642,
"grad_norm": 13.032412874817698,
"learning_rate": 2.7777777777777777e-08,
"logits": -1.4463595151901245,
"logps": -83.46998596191406,
"loss": 0.6929,
"objective": 0.6927531957626343,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.512499988079071,
"ranking_simple": 0.5083333253860474,
"regularize": 0.6927531957626343,
"step": 10
},
{
"dpo_loss": 0.6928622722625732,
"epoch": 0.08502598016060463,
"grad_norm": 12.566734219683084,
"learning_rate": 4.166666666666667e-08,
"logits": -1.4098025560379028,
"logps": -83.67684936523438,
"loss": 0.6927,
"objective": 0.6928622722625732,
"ranking_idealized": 0.5833333134651184,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5083333253860474,
"regularize": 0.6928622722625732,
"step": 15
},
{
"dpo_loss": 0.6923068761825562,
"epoch": 0.11336797354747284,
"grad_norm": 13.25147394148046,
"learning_rate": 5.5555555555555555e-08,
"logits": -1.3977584838867188,
"logps": -84.3187484741211,
"loss": 0.6919,
"objective": 0.6923068761825562,
"ranking_idealized": 0.625,
"ranking_idealized_expo": 0.5375000238418579,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6923068761825562,
"step": 20
},
{
"dpo_loss": 0.6906546354293823,
"epoch": 0.14170996693434104,
"grad_norm": 12.656689096549028,
"learning_rate": 6.944444444444444e-08,
"logits": -1.4436664581298828,
"logps": -84.10287475585938,
"loss": 0.6912,
"objective": 0.6906546354293823,
"ranking_idealized": 0.5958333611488342,
"ranking_idealized_expo": 0.4958333373069763,
"ranking_simple": 0.4958333373069763,
"regularize": 0.6906546354293823,
"step": 25
},
{
"dpo_loss": 0.6908212900161743,
"epoch": 0.17005196032120926,
"grad_norm": 13.15545428739909,
"learning_rate": 8.333333333333334e-08,
"logits": -1.4165135622024536,
"logps": -84.5628433227539,
"loss": 0.6902,
"objective": 0.6908212900161743,
"ranking_idealized": 0.6041666865348816,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6908212900161743,
"step": 30
},
{
"dpo_loss": 0.6884526610374451,
"epoch": 0.19839395370807747,
"grad_norm": 12.961038019497309,
"learning_rate": 9.722222222222221e-08,
"logits": -1.3882626295089722,
"logps": -83.0355453491211,
"loss": 0.6879,
"objective": 0.6884526610374451,
"ranking_idealized": 0.6166666746139526,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5083333253860474,
"regularize": 0.6884526610374451,
"step": 35
},
{
"dpo_loss": 0.6872146129608154,
"epoch": 0.22673594709494568,
"grad_norm": 15.265896606535208,
"learning_rate": 9.996046986136509e-08,
"logits": -1.3777297735214233,
"logps": -83.15966796875,
"loss": 0.6889,
"objective": 0.6872146129608154,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.5208333134651184,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6872146129608154,
"step": 40
},
{
"dpo_loss": 0.6848792433738708,
"epoch": 0.25507794048181387,
"grad_norm": 13.166431360837958,
"learning_rate": 9.979998579289983e-08,
"logits": -1.5064284801483154,
"logps": -84.50239562988281,
"loss": 0.6857,
"objective": 0.6848792433738708,
"ranking_idealized": 0.6208333373069763,
"ranking_idealized_expo": 0.5375000238418579,
"ranking_simple": 0.5416666865348816,
"regularize": 0.6848792433738708,
"step": 45
},
{
"dpo_loss": 0.6835355162620544,
"epoch": 0.2834199338686821,
"grad_norm": 12.111837571939198,
"learning_rate": 9.95164733236251e-08,
"logits": -1.4091321229934692,
"logps": -82.50495147705078,
"loss": 0.6855,
"objective": 0.6835355162620544,
"ranking_idealized": 0.5416666865348816,
"ranking_idealized_expo": 0.42500001192092896,
"ranking_simple": 0.4333333373069763,
"regularize": 0.6835355162620544,
"step": 50
},
{
"epoch": 0.2834199338686821,
"eval_dpo_loss": 0.6918472647666931,
"eval_logits": -1.4343044757843018,
"eval_logps": -90.7668685913086,
"eval_loss": 0.6888882517814636,
"eval_objective": 0.6918472647666931,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5103305578231812,
"eval_regularize": 0.6918472647666931,
"eval_runtime": 261.146,
"eval_samples_per_second": 22.172,
"eval_steps_per_second": 0.927,
"step": 50
},
{
"dpo_loss": 0.6787654161453247,
"epoch": 0.3117619272555503,
"grad_norm": 14.068180692870671,
"learning_rate": 9.911063285706807e-08,
"logits": -1.4499430656433105,
"logps": -84.21847534179688,
"loss": 0.6825,
"objective": 0.6787654161453247,
"ranking_idealized": 0.625,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.550000011920929,
"regularize": 0.6787654161453247,
"step": 55
},
{
"dpo_loss": 0.6828632950782776,
"epoch": 0.3401039206424185,
"grad_norm": 13.070276554702126,
"learning_rate": 9.85834670020205e-08,
"logits": -1.4364044666290283,
"logps": -83.14185333251953,
"loss": 0.6829,
"objective": 0.6828632950782776,
"ranking_idealized": 0.637499988079071,
"ranking_idealized_expo": 0.512499988079071,
"ranking_simple": 0.5249999761581421,
"regularize": 0.6828632950782776,
"step": 60
},
{
"dpo_loss": 0.6835117340087891,
"epoch": 0.3684459140292867,
"grad_norm": 13.497944122383998,
"learning_rate": 9.793627809564323e-08,
"logits": -1.3944473266601562,
"logps": -83.09821319580078,
"loss": 0.6822,
"objective": 0.6835117340087891,
"ranking_idealized": 0.5708333253860474,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.4958333373069763,
"regularize": 0.6835117340087891,
"step": 65
},
{
"dpo_loss": 0.6772111058235168,
"epoch": 0.39678790741615494,
"grad_norm": 13.151701497634273,
"learning_rate": 9.717066498610672e-08,
"logits": -1.480456829071045,
"logps": -84.27593231201172,
"loss": 0.6818,
"objective": 0.6772111058235168,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.5166666507720947,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6772111058235168,
"step": 70
},
{
"dpo_loss": 0.679884672164917,
"epoch": 0.42512990080302315,
"grad_norm": 12.601045745018736,
"learning_rate": 9.628851908271572e-08,
"logits": -1.5056648254394531,
"logps": -84.74910736083984,
"loss": 0.6776,
"objective": 0.679884672164917,
"ranking_idealized": 0.6291666626930237,
"ranking_idealized_expo": 0.5458333492279053,
"ranking_simple": 0.5458333492279053,
"regularize": 0.679884672164917,
"step": 75
},
{
"dpo_loss": 0.683576762676239,
"epoch": 0.45347189418989137,
"grad_norm": 13.218876579787791,
"learning_rate": 9.529201968327615e-08,
"logits": -1.550328016281128,
"logps": -84.59281158447266,
"loss": 0.6772,
"objective": 0.683576762676239,
"ranking_idealized": 0.5833333134651184,
"ranking_idealized_expo": 0.5166666507720947,
"ranking_simple": 0.5208333134651184,
"regularize": 0.683576762676239,
"step": 80
},
{
"dpo_loss": 0.6882224678993225,
"epoch": 0.4818138875767596,
"grad_norm": 12.712021197638556,
"learning_rate": 9.41836285902478e-08,
"logits": -1.507049560546875,
"logps": -82.99642944335938,
"loss": 0.6781,
"objective": 0.6882224678993225,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.5249999761581421,
"ranking_simple": 0.5291666388511658,
"regularize": 0.6882224678993225,
"step": 85
},
{
"dpo_loss": 0.6753127574920654,
"epoch": 0.5101558809636277,
"grad_norm": 13.425590079214597,
"learning_rate": 9.296608402898305e-08,
"logits": -1.3210861682891846,
"logps": -84.38108825683594,
"loss": 0.6726,
"objective": 0.6753127574920654,
"ranking_idealized": 0.550000011920929,
"ranking_idealized_expo": 0.4791666567325592,
"ranking_simple": 0.48750001192092896,
"regularize": 0.6753127574920654,
"step": 90
},
{
"dpo_loss": 0.6701774001121521,
"epoch": 0.538497874350496,
"grad_norm": 13.30415786759025,
"learning_rate": 9.164239388307666e-08,
"logits": -1.4094645977020264,
"logps": -84.12966918945312,
"loss": 0.6736,
"objective": 0.6701774001121521,
"ranking_idealized": 0.625,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.550000011920929,
"regularize": 0.6701774001121521,
"step": 95
},
{
"dpo_loss": 0.6689857840538025,
"epoch": 0.5668398677373642,
"grad_norm": 14.262472601225529,
"learning_rate": 9.021582826353824e-08,
"logits": -1.4942399263381958,
"logps": -85.0649185180664,
"loss": 0.6746,
"objective": 0.6689857840538025,
"ranking_idealized": 0.612500011920929,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6689857840538025,
"step": 100
},
{
"epoch": 0.5668398677373642,
"eval_dpo_loss": 0.6899410486221313,
"eval_logits": -1.476444125175476,
"eval_logps": -90.9747543334961,
"eval_loss": 0.6857675909996033,
"eval_objective": 0.6899410486221313,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5092975497245789,
"eval_regularize": 0.6899410486221313,
"eval_runtime": 259.9569,
"eval_samples_per_second": 22.273,
"eval_steps_per_second": 0.931,
"step": 100
},
{
"dpo_loss": 0.6589028239250183,
"epoch": 0.5951818611242324,
"grad_norm": 13.550145448172236,
"learning_rate": 8.868991143014468e-08,
"logits": -1.3795539140701294,
"logps": -84.87275695800781,
"loss": 0.6674,
"objective": 0.6589028239250183,
"ranking_idealized": 0.6208333373069763,
"ranking_idealized_expo": 0.5333333611488342,
"ranking_simple": 0.5458333492279053,
"regularize": 0.6589028239250183,
"step": 105
},
{
"dpo_loss": 0.6661778688430786,
"epoch": 0.6235238545111006,
"grad_norm": 15.598239406382287,
"learning_rate": 8.706841308493092e-08,
"logits": -1.561141848564148,
"logps": -83.28865814208984,
"loss": 0.668,
"objective": 0.6661778688430786,
"ranking_idealized": 0.5708333253860474,
"ranking_idealized_expo": 0.4541666805744171,
"ranking_simple": 0.46666666865348816,
"regularize": 0.6661778688430786,
"step": 110
},
{
"dpo_loss": 0.6700491309165955,
"epoch": 0.6518658478979689,
"grad_norm": 14.005634900395668,
"learning_rate": 8.535533905932736e-08,
"logits": -1.5146809816360474,
"logps": -83.37279510498047,
"loss": 0.6669,
"objective": 0.6700491309165955,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.4833333194255829,
"ranking_simple": 0.4958333373069763,
"regularize": 0.6700491309165955,
"step": 115
},
{
"dpo_loss": 0.6605091691017151,
"epoch": 0.680207841284837,
"grad_norm": 13.532342794289065,
"learning_rate": 8.355492141795184e-08,
"logits": -1.4583431482315063,
"logps": -84.31383514404297,
"loss": 0.6659,
"objective": 0.6605091691017151,
"ranking_idealized": 0.5874999761581421,
"ranking_idealized_expo": 0.5333333611488342,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6605091691017151,
"step": 120
},
{
"dpo_loss": 0.6656448841094971,
"epoch": 0.7085498346717053,
"grad_norm": 13.79567400382208,
"learning_rate": 8.167160800350306e-08,
"logits": -1.4243894815444946,
"logps": -83.22982025146484,
"loss": 0.6688,
"objective": 0.6656448841094971,
"ranking_idealized": 0.6291666626930237,
"ranking_idealized_expo": 0.550000011920929,
"ranking_simple": 0.5458333492279053,
"regularize": 0.6656448841094971,
"step": 125
},
{
"dpo_loss": 0.6677243113517761,
"epoch": 0.7368918280585735,
"grad_norm": 13.221986997309093,
"learning_rate": 7.971005144858553e-08,
"logits": -1.564219355583191,
"logps": -82.65666961669922,
"loss": 0.6701,
"objective": 0.6677243113517761,
"ranking_idealized": 0.5416666865348816,
"ranking_idealized_expo": 0.44999998807907104,
"ranking_simple": 0.47083333134651184,
"regularize": 0.6677243113517761,
"step": 130
},
{
"dpo_loss": 0.6758949160575867,
"epoch": 0.7652338214454416,
"grad_norm": 13.535024146063408,
"learning_rate": 7.767509768161078e-08,
"logits": -1.5405229330062866,
"logps": -84.79542541503906,
"loss": 0.6661,
"objective": 0.6758949160575867,
"ranking_idealized": 0.5833333134651184,
"ranking_idealized_expo": 0.5291666388511658,
"ranking_simple": 0.5249999761581421,
"regularize": 0.6758949160575867,
"step": 135
},
{
"dpo_loss": 0.6768299341201782,
"epoch": 0.7935758148323099,
"grad_norm": 13.276044449636085,
"learning_rate": 7.557177395517111e-08,
"logits": -1.4452087879180908,
"logps": -83.74736022949219,
"loss": 0.6665,
"objective": 0.6768299341201782,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5249999761581421,
"regularize": 0.6768299341201782,
"step": 140
},
{
"dpo_loss": 0.6667078733444214,
"epoch": 0.821917808219178,
"grad_norm": 13.106504702831565,
"learning_rate": 7.340527642646068e-08,
"logits": -1.522230863571167,
"logps": -83.9556655883789,
"loss": 0.6628,
"objective": 0.6667078733444214,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.4833333194255829,
"ranking_simple": 0.4833333194255829,
"regularize": 0.6667078733444214,
"step": 145
},
{
"dpo_loss": 0.6657266616821289,
"epoch": 0.8502598016060463,
"grad_norm": 13.484987339837966,
"learning_rate": 7.118095732042642e-08,
"logits": -1.494079351425171,
"logps": -82.12902069091797,
"loss": 0.6601,
"objective": 0.6657266616821289,
"ranking_idealized": 0.6166666746139526,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.5541666746139526,
"regularize": 0.6657266616821289,
"step": 150
},
{
"epoch": 0.8502598016060463,
"eval_dpo_loss": 0.688634991645813,
"eval_logits": -1.5179173946380615,
"eval_logps": -90.80626678466797,
"eval_loss": 0.682818591594696,
"eval_objective": 0.688634991645813,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5134297609329224,
"eval_regularize": 0.688634991645813,
"eval_runtime": 259.9549,
"eval_samples_per_second": 22.273,
"eval_steps_per_second": 0.931,
"step": 150
},
{
"dpo_loss": 0.659411609172821,
"epoch": 0.8786017949929145,
"grad_norm": 15.013754194888069,
"learning_rate": 6.89043117073609e-08,
"logits": -1.5184494256973267,
"logps": -82.16848754882812,
"loss": 0.6586,
"objective": 0.659411609172821,
"ranking_idealized": 0.6166666746139526,
"ranking_idealized_expo": 0.512499988079071,
"ranking_simple": 0.5416666865348816,
"regularize": 0.659411609172821,
"step": 155
},
{
"dpo_loss": 0.660663902759552,
"epoch": 0.9069437883797827,
"grad_norm": 13.528075587958016,
"learning_rate": 6.65809639276034e-08,
"logits": -1.4307061433792114,
"logps": -83.12458038330078,
"loss": 0.666,
"objective": 0.660663902759552,
"ranking_idealized": 0.5708333253860474,
"ranking_idealized_expo": 0.4958333373069763,
"ranking_simple": 0.512499988079071,
"regularize": 0.660663902759552,
"step": 160
},
{
"dpo_loss": 0.6495606303215027,
"epoch": 0.9352857817666509,
"grad_norm": 14.079206144678064,
"learning_rate": 6.4216653696885e-08,
"logits": -1.4516769647598267,
"logps": -84.19721984863281,
"loss": 0.6591,
"objective": 0.6495606303215027,
"ranking_idealized": 0.6041666865348816,
"ranking_idealized_expo": 0.5208333134651184,
"ranking_simple": 0.5458333492279053,
"regularize": 0.6495606303215027,
"step": 165
},
{
"dpo_loss": 0.6589613556861877,
"epoch": 0.9636277751535192,
"grad_norm": 15.945719305933032,
"learning_rate": 6.181722192664524e-08,
"logits": -1.4703598022460938,
"logps": -84.60289764404297,
"loss": 0.6579,
"objective": 0.6589613556861877,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.512499988079071,
"ranking_simple": 0.5291666388511658,
"regularize": 0.6589613556861877,
"step": 170
},
{
"dpo_loss": 0.6631531119346619,
"epoch": 0.9919697685403873,
"grad_norm": 13.762050879048529,
"learning_rate": 5.9388596294349124e-08,
"logits": -1.4086403846740723,
"logps": -85.38468933105469,
"loss": 0.6613,
"objective": 0.6631531119346619,
"ranking_idealized": 0.6541666388511658,
"ranking_idealized_expo": 0.574999988079071,
"ranking_simple": 0.5791666507720947,
"regularize": 0.6631531119346619,
"step": 175
},
{
"dpo_loss": 0.6530786156654358,
"epoch": 1.0203117619272555,
"grad_norm": 15.08770015055669,
"learning_rate": 5.693677659945342e-08,
"logits": -1.6005927324295044,
"logps": -84.76131439208984,
"loss": 0.6531,
"objective": 0.6530786156654358,
"ranking_idealized": 0.6166666746139526,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.550000011920929,
"regularize": 0.6530786156654358,
"step": 180
},
{
"dpo_loss": 0.6419151425361633,
"epoch": 1.0486537553141237,
"grad_norm": 14.657084961337553,
"learning_rate": 5.446781994119886e-08,
"logits": -1.4578330516815186,
"logps": -84.88520812988281,
"loss": 0.6472,
"objective": 0.6419151425361633,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.4833333194255829,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6419151425361633,
"step": 185
},
{
"dpo_loss": 0.6436451077461243,
"epoch": 1.076995748700992,
"grad_norm": 13.750596760410684,
"learning_rate": 5.198782575484629e-08,
"logits": -1.5236998796463013,
"logps": -86.2567138671875,
"loss": 0.6487,
"objective": 0.6436451077461243,
"ranking_idealized": 0.625,
"ranking_idealized_expo": 0.5541666746139526,
"ranking_simple": 0.5791666507720947,
"regularize": 0.6436451077461243,
"step": 190
},
{
"dpo_loss": 0.6471238732337952,
"epoch": 1.10533774208786,
"grad_norm": 16.400656168720136,
"learning_rate": 4.9502920743323515e-08,
"logits": -1.504638433456421,
"logps": -85.38711547851562,
"loss": 0.6526,
"objective": 0.6471238732337952,
"ranking_idealized": 0.6333333253860474,
"ranking_idealized_expo": 0.5666666626930237,
"ranking_simple": 0.5791666507720947,
"regularize": 0.6471238732337952,
"step": 195
},
{
"dpo_loss": 0.6551472544670105,
"epoch": 1.1336797354747283,
"grad_norm": 13.299323953163684,
"learning_rate": 4.701924374150901e-08,
"logits": -1.5839470624923706,
"logps": -84.28254699707031,
"loss": 0.6473,
"objective": 0.6551472544670105,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.512499988079071,
"regularize": 0.6551472544670105,
"step": 200
},
{
"epoch": 1.1336797354747283,
"eval_dpo_loss": 0.689020574092865,
"eval_logits": -1.5427004098892212,
"eval_logps": -91.97786712646484,
"eval_loss": 0.6826460361480713,
"eval_objective": 0.689020574092865,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5175619721412659,
"eval_regularize": 0.689020574092865,
"eval_runtime": 259.6574,
"eval_samples_per_second": 22.299,
"eval_steps_per_second": 0.932,
"step": 200
},
{
"dpo_loss": 0.6496204733848572,
"epoch": 1.1620217288615966,
"grad_norm": 13.335913932026553,
"learning_rate": 4.454293055054397e-08,
"logits": -1.5556657314300537,
"logps": -84.84135437011719,
"loss": 0.6507,
"objective": 0.6496204733848572,
"ranking_idealized": 0.5708333253860474,
"ranking_idealized_expo": 0.48750001192092896,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6496204733848572,
"step": 205
},
{
"dpo_loss": 0.6496918201446533,
"epoch": 1.1903637222484649,
"grad_norm": 14.028092373104228,
"learning_rate": 4.208009877963925e-08,
"logits": -1.5215808153152466,
"logps": -85.02703094482422,
"loss": 0.6464,
"objective": 0.6496918201446533,
"ranking_idealized": 0.5874999761581421,
"ranking_idealized_expo": 0.5,
"ranking_simple": 0.5083333253860474,
"regularize": 0.6496918201446533,
"step": 210
},
{
"dpo_loss": 0.6471313834190369,
"epoch": 1.2187057156353331,
"grad_norm": 13.415699632283031,
"learning_rate": 3.963683273282455e-08,
"logits": -1.4972035884857178,
"logps": -84.6708984375,
"loss": 0.6444,
"objective": 0.6471313834190369,
"ranking_idealized": 0.5625,
"ranking_idealized_expo": 0.4833333194255829,
"ranking_simple": 0.49166667461395264,
"regularize": 0.6471313834190369,
"step": 215
},
{
"dpo_loss": 0.6404972076416016,
"epoch": 1.2470477090222012,
"grad_norm": 15.126572956782434,
"learning_rate": 3.721916837797627e-08,
"logits": -1.4574114084243774,
"logps": -83.83084106445312,
"loss": 0.6425,
"objective": 0.6404972076416016,
"ranking_idealized": 0.6541666388511658,
"ranking_idealized_expo": 0.6041666865348816,
"ranking_simple": 0.6041666865348816,
"regularize": 0.6404972076416016,
"step": 220
},
{
"dpo_loss": 0.64054936170578,
"epoch": 1.2753897024090695,
"grad_norm": 13.996717622001338,
"learning_rate": 3.483307843525758e-08,
"logits": -1.545446515083313,
"logps": -85.11936950683594,
"loss": 0.6452,
"objective": 0.64054936170578,
"ranking_idealized": 0.612500011920929,
"ranking_idealized_expo": 0.5291666388511658,
"ranking_simple": 0.550000011920929,
"regularize": 0.64054936170578,
"step": 225
},
{
"dpo_loss": 0.6458050608634949,
"epoch": 1.3037316957959377,
"grad_norm": 13.111039704803117,
"learning_rate": 3.248445762180878e-08,
"logits": -1.5097877979278564,
"logps": -84.0020523071289,
"loss": 0.6439,
"objective": 0.6458050608634949,
"ranking_idealized": 0.5833333134651184,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.5333333611488342,
"regularize": 0.6458050608634949,
"step": 230
},
{
"dpo_loss": 0.6388878226280212,
"epoch": 1.3320736891828058,
"grad_norm": 14.535772291959058,
"learning_rate": 3.01791080891403e-08,
"logits": -1.5895107984542847,
"logps": -82.74007415771484,
"loss": 0.6453,
"objective": 0.6388878226280212,
"ranking_idealized": 0.6041666865348816,
"ranking_idealized_expo": 0.4833333194255829,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6388878226280212,
"step": 235
},
{
"dpo_loss": 0.6343578100204468,
"epoch": 1.360415682569674,
"grad_norm": 14.17039051493401,
"learning_rate": 2.7922725089204424e-08,
"logits": -1.5228545665740967,
"logps": -84.07835388183594,
"loss": 0.6392,
"objective": 0.6343578100204468,
"ranking_idealized": 0.5541666746139526,
"ranking_idealized_expo": 0.4791666567325592,
"ranking_simple": 0.5083333253860474,
"regularize": 0.6343578100204468,
"step": 240
},
{
"dpo_loss": 0.6442958116531372,
"epoch": 1.3887576759565423,
"grad_norm": 13.5370550702067,
"learning_rate": 2.572088290455715e-08,
"logits": -1.551520586013794,
"logps": -84.03118133544922,
"loss": 0.6442,
"objective": 0.6442958116531372,
"ranking_idealized": 0.5916666388511658,
"ranking_idealized_expo": 0.5208333134651184,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6442958116531372,
"step": 245
},
{
"dpo_loss": 0.6429386138916016,
"epoch": 1.4170996693434104,
"grad_norm": 13.133220637867716,
"learning_rate": 2.3579021077369044e-08,
"logits": -1.4924685955047607,
"logps": -83.64738464355469,
"loss": 0.6449,
"objective": 0.6429386138916016,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.47083333134651184,
"ranking_simple": 0.49166667461395264,
"regularize": 0.6429386138916016,
"step": 250
},
{
"epoch": 1.4170996693434104,
"eval_dpo_loss": 0.6887210607528687,
"eval_logits": -1.5537456274032593,
"eval_logps": -91.60443878173828,
"eval_loss": 0.6813494563102722,
"eval_objective": 0.6887210607528687,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5175619721412659,
"eval_regularize": 0.6887210607528687,
"eval_runtime": 259.4717,
"eval_samples_per_second": 22.315,
"eval_steps_per_second": 0.933,
"step": 250
},
{
"dpo_loss": 0.6430085301399231,
"epoch": 1.4454416627302786,
"grad_norm": 14.835038849969866,
"learning_rate": 2.1502430971305285e-08,
"logits": -1.5067726373672485,
"logps": -84.9552993774414,
"loss": 0.6394,
"objective": 0.6430085301399231,
"ranking_idealized": 0.6166666746139526,
"ranking_idealized_expo": 0.5458333492279053,
"ranking_simple": 0.5583333373069763,
"regularize": 0.6430085301399231,
"step": 255
},
{
"dpo_loss": 0.6379219889640808,
"epoch": 1.473783656117147,
"grad_norm": 15.014357418345154,
"learning_rate": 1.9496242699473782e-08,
"logits": -1.570318579673767,
"logps": -86.18223571777344,
"loss": 0.6438,
"objective": 0.6379219889640808,
"ranking_idealized": 0.5666666626930237,
"ranking_idealized_expo": 0.48750001192092896,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6379219889640808,
"step": 260
},
{
"dpo_loss": 0.6470407843589783,
"epoch": 1.5021256495040152,
"grad_norm": 13.71802645949903,
"learning_rate": 1.7565412450734317e-08,
"logits": -1.525256633758545,
"logps": -83.78214263916016,
"loss": 0.643,
"objective": 0.6470407843589783,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5333333611488342,
"regularize": 0.6470407843589783,
"step": 265
},
{
"dpo_loss": 0.6488485932350159,
"epoch": 1.5304676428908834,
"grad_norm": 13.736321065743684,
"learning_rate": 1.5714710245679348e-08,
"logits": -1.5463277101516724,
"logps": -85.4636459350586,
"loss": 0.6396,
"objective": 0.6488485932350159,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.5333333611488342,
"regularize": 0.6488485932350159,
"step": 270
},
{
"dpo_loss": 0.6298065781593323,
"epoch": 1.5588096362777515,
"grad_norm": 13.998439522929331,
"learning_rate": 1.394870815253416e-08,
"logits": -1.5613772869110107,
"logps": -86.0442886352539,
"loss": 0.644,
"objective": 0.6298065781593323,
"ranking_idealized": 0.6208333373069763,
"ranking_idealized_expo": 0.5249999761581421,
"ranking_simple": 0.5458333492279053,
"regularize": 0.6298065781593323,
"step": 275
},
{
"dpo_loss": 0.6345570087432861,
"epoch": 1.5871516296646198,
"grad_norm": 14.368158677946898,
"learning_rate": 1.2271768992088489e-08,
"logits": -1.4692186117172241,
"logps": -84.12680053710938,
"loss": 0.643,
"objective": 0.6345570087432861,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5208333134651184,
"regularize": 0.6345570087432861,
"step": 280
},
{
"dpo_loss": 0.6354427337646484,
"epoch": 1.615493623051488,
"grad_norm": 15.792908553123706,
"learning_rate": 1.0688035559563669e-08,
"logits": -1.5792218446731567,
"logps": -85.75080108642578,
"loss": 0.6396,
"objective": 0.6354427337646484,
"ranking_idealized": 0.5874999761581421,
"ranking_idealized_expo": 0.5166666507720947,
"ranking_simple": 0.5291666388511658,
"regularize": 0.6354427337646484,
"step": 285
},
{
"dpo_loss": 0.6333154439926147,
"epoch": 1.643835616438356,
"grad_norm": 13.281161084922974,
"learning_rate": 9.201420390041964e-09,
"logits": -1.62575364112854,
"logps": -84.09397888183594,
"loss": 0.6388,
"objective": 0.6333154439926147,
"ranking_idealized": 0.612500011920929,
"ranking_idealized_expo": 0.5625,
"ranking_simple": 0.5708333253860474,
"regularize": 0.6333154439926147,
"step": 290
},
{
"dpo_loss": 0.6362510919570923,
"epoch": 1.6721776098252243,
"grad_norm": 13.068465728291164,
"learning_rate": 7.815596092742277e-09,
"logits": -1.5190918445587158,
"logps": -83.1363296508789,
"loss": 0.6395,
"objective": 0.6362510919570923,
"ranking_idealized": 0.5791666507720947,
"ranking_idealized_expo": 0.4791666567325592,
"ranking_simple": 0.512499988079071,
"regularize": 0.6362510919570923,
"step": 295
},
{
"dpo_loss": 0.6396799087524414,
"epoch": 1.7005196032120926,
"grad_norm": 13.517747947290621,
"learning_rate": 6.533986278020875e-09,
"logits": -1.5812262296676636,
"logps": -84.42427062988281,
"loss": 0.6384,
"objective": 0.6396799087524414,
"ranking_idealized": 0.6583333611488342,
"ranking_idealized_expo": 0.574999988079071,
"ranking_simple": 0.5958333611488342,
"regularize": 0.6396799087524414,
"step": 300
},
{
"epoch": 1.7005196032120926,
"eval_dpo_loss": 0.6890466213226318,
"eval_logits": -1.5627375841140747,
"eval_logps": -92.01404571533203,
"eval_loss": 0.6818000674247742,
"eval_objective": 0.6890466213226318,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.5185950398445129,
"eval_regularize": 0.6890466213226318,
"eval_runtime": 261.8927,
"eval_samples_per_second": 22.108,
"eval_steps_per_second": 0.924,
"step": 300
},
{
"dpo_loss": 0.6523339748382568,
"epoch": 1.7288615965989607,
"grad_norm": 14.833162083591201,
"learning_rate": 5.359757099511236e-09,
"logits": -1.5387723445892334,
"logps": -85.74983978271484,
"loss": 0.6443,
"objective": 0.6523339748382568,
"ranking_idealized": 0.5625,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.5208333134651184,
"regularize": 0.6523339748382568,
"step": 305
},
{
"dpo_loss": 0.6306132078170776,
"epoch": 1.7572035899858292,
"grad_norm": 14.09816487492066,
"learning_rate": 4.295809432298269e-09,
"logits": -1.616551160812378,
"logps": -83.39339447021484,
"loss": 0.6374,
"objective": 0.6306132078170776,
"ranking_idealized": 0.6083333492279053,
"ranking_idealized_expo": 0.5249999761581421,
"ranking_simple": 0.5458333492279053,
"regularize": 0.6306132078170776,
"step": 310
},
{
"dpo_loss": 0.6414229273796082,
"epoch": 1.7855455833726972,
"grad_norm": 12.85646357819718,
"learning_rate": 3.344771706449956e-09,
"logits": -1.5274823904037476,
"logps": -84.38652801513672,
"loss": 0.6393,
"objective": 0.6414229273796082,
"ranking_idealized": 0.6041666865348816,
"ranking_idealized_expo": 0.5083333253860474,
"ranking_simple": 0.5291666388511658,
"regularize": 0.6414229273796082,
"step": 315
},
{
"dpo_loss": 0.6470603942871094,
"epoch": 1.8138875767595655,
"grad_norm": 14.912269898190576,
"learning_rate": 2.508993413610866e-09,
"logits": -1.5946826934814453,
"logps": -86.11390686035156,
"loss": 0.6416,
"objective": 0.6470603942871094,
"ranking_idealized": 0.6000000238418579,
"ranking_idealized_expo": 0.5041666626930237,
"ranking_simple": 0.5166666507720947,
"regularize": 0.6470603942871094,
"step": 320
},
{
"dpo_loss": 0.6489089727401733,
"epoch": 1.8422295701464337,
"grad_norm": 16.310745862402126,
"learning_rate": 1.790539302699351e-09,
"logits": -1.6386195421218872,
"logps": -86.83934020996094,
"loss": 0.6366,
"objective": 0.6489089727401733,
"ranking_idealized": 0.6333333253860474,
"ranking_idealized_expo": 0.5166666507720947,
"ranking_simple": 0.5666666626930237,
"regularize": 0.6489089727401733,
"step": 325
},
{
"dpo_loss": 0.6407893896102905,
"epoch": 1.8705715635333018,
"grad_norm": 14.36720629246665,
"learning_rate": 1.1911842790474635e-09,
"logits": -1.3639094829559326,
"logps": -84.11365509033203,
"loss": 0.6382,
"objective": 0.6407893896102905,
"ranking_idealized": 0.5708333253860474,
"ranking_idealized_expo": 0.4749999940395355,
"ranking_simple": 0.5,
"regularize": 0.6407893896102905,
"step": 330
},
{
"dpo_loss": 0.642834484577179,
"epoch": 1.89891355692017,
"grad_norm": 12.837846805234047,
"learning_rate": 7.124090195851806e-10,
"logits": -1.5466150045394897,
"logps": -84.1384048461914,
"loss": 0.6395,
"objective": 0.642834484577179,
"ranking_idealized": 0.5916666388511658,
"ranking_idealized_expo": 0.4958333373069763,
"ranking_simple": 0.5458333492279053,
"regularize": 0.642834484577179,
"step": 335
},
{
"dpo_loss": 0.6367443799972534,
"epoch": 1.9272555503070383,
"grad_norm": 13.618225760097143,
"learning_rate": 3.553963149013295e-10,
"logits": -1.5659750699996948,
"logps": -86.13041687011719,
"loss": 0.644,
"objective": 0.6367443799972534,
"ranking_idealized": 0.6416666507720947,
"ranking_idealized_expo": 0.5416666865348816,
"ranking_simple": 0.5791666507720947,
"regularize": 0.6367443799972534,
"step": 340
},
{
"dpo_loss": 0.6491357088088989,
"epoch": 1.9555975436939064,
"grad_norm": 13.98814029327361,
"learning_rate": 1.2102814721791643e-10,
"logits": -1.5671348571777344,
"logps": -84.2940902709961,
"loss": 0.6417,
"objective": 0.6491357088088989,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.5333333611488342,
"ranking_simple": 0.5375000238418579,
"regularize": 0.6491357088088989,
"step": 345
},
{
"dpo_loss": 0.6474497318267822,
"epoch": 1.9839395370807746,
"grad_norm": 13.915220160397528,
"learning_rate": 9.883511496722174e-12,
"logits": -1.4725474119186401,
"logps": -85.95130157470703,
"loss": 0.6431,
"objective": 0.6474497318267822,
"ranking_idealized": 0.574999988079071,
"ranking_idealized_expo": 0.48750001192092896,
"ranking_simple": 0.4958333373069763,
"regularize": 0.6474497318267822,
"step": 350
},
{
"epoch": 1.9839395370807746,
"eval_dpo_loss": 0.6890938878059387,
"eval_logits": -1.5636365413665771,
"eval_logps": -92.05502319335938,
"eval_loss": 0.6817956566810608,
"eval_objective": 0.6890938878059387,
"eval_ranking_idealized": 0.5888429880142212,
"eval_ranking_idealized_expo": 0.5103305578231812,
"eval_ranking_simple": 0.51962810754776,
"eval_regularize": 0.6890938878059387,
"eval_runtime": 259.5397,
"eval_samples_per_second": 22.309,
"eval_steps_per_second": 0.932,
"step": 350
},
{
"epoch": 1.995276334435522,
"step": 352,
"total_flos": 0.0,
"train_loss": 0.6593505936590108,
"train_runtime": 15975.3291,
"train_samples_per_second": 6.36,
"train_steps_per_second": 0.022
}
],
"logging_steps": 5,
"max_steps": 352,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}