phi-2-gpo-v17-i1 / trainer_state.json
lole25's picture
Model save
0b4ce1a verified
raw
history blame contribute delete
No virus
23.3 kB
{
"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.18604117631912231,
"logits/rejected": 0.34631967544555664,
"logps/chosen": -460.0769348144531,
"logps/rejected": -351.57135009765625,
"loss": 0.3612,
"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.05747946724295616,
"logits/rejected": 0.16539901494979858,
"logps/chosen": -330.9287414550781,
"logps/rejected": -328.71575927734375,
"loss": 0.3819,
"rewards/accuracies": 0.3888888955116272,
"rewards/chosen": 2.3913149561849423e-05,
"rewards/margins": -1.7764228687155992e-05,
"rewards/rejected": 4.167737643001601e-05,
"step": 10
},
{
"epoch": 0.04,
"learning_rate": 2.1276595744680853e-06,
"logits/chosen": 0.15373219549655914,
"logits/rejected": 0.19493858516216278,
"logps/chosen": -325.29803466796875,
"logps/rejected": -315.1011047363281,
"loss": 0.3718,
"rewards/accuracies": 0.4312500059604645,
"rewards/chosen": -1.974666338355746e-05,
"rewards/margins": 1.2878153654583002e-07,
"rewards/rejected": -1.9875456928275526e-05,
"step": 20
},
{
"epoch": 0.06,
"learning_rate": 3.191489361702128e-06,
"logits/chosen": 0.11452794075012207,
"logits/rejected": 0.1842522919178009,
"logps/chosen": -371.60943603515625,
"logps/rejected": -327.19366455078125,
"loss": 0.3676,
"rewards/accuracies": 0.4937500059604645,
"rewards/chosen": -5.7096302043646574e-05,
"rewards/margins": 3.7131747376406565e-05,
"rewards/rejected": -9.422805305803195e-05,
"step": 30
},
{
"epoch": 0.09,
"learning_rate": 4.255319148936171e-06,
"logits/chosen": 0.18893569707870483,
"logits/rejected": 0.2084759920835495,
"logps/chosen": -337.7607727050781,
"logps/rejected": -353.72503662109375,
"loss": 0.3678,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": 0.00010834180284291506,
"rewards/margins": 0.00026063303812406957,
"rewards/rejected": -0.0001522912352811545,
"step": 40
},
{
"epoch": 0.11,
"learning_rate": 4.999373573764188e-06,
"logits/chosen": 0.08207504451274872,
"logits/rejected": 0.20786412060260773,
"logps/chosen": -384.91656494140625,
"logps/rejected": -347.6080627441406,
"loss": 0.3741,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.0010863704374060035,
"rewards/margins": 0.0014893051702529192,
"rewards/rejected": -0.0004029346746392548,
"step": 50
},
{
"epoch": 0.13,
"learning_rate": 4.988245838331339e-06,
"logits/chosen": 0.13469652831554413,
"logits/rejected": 0.17153413593769073,
"logps/chosen": -372.34124755859375,
"logps/rejected": -329.98626708984375,
"loss": 0.3646,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.0023965700529515743,
"rewards/margins": 0.0026933744084089994,
"rewards/rejected": -0.0002968042972497642,
"step": 60
},
{
"epoch": 0.15,
"learning_rate": 4.963268819535228e-06,
"logits/chosen": 0.15689334273338318,
"logits/rejected": 0.20978419482707977,
"logps/chosen": -364.29888916015625,
"logps/rejected": -377.21087646484375,
"loss": 0.3705,
"rewards/accuracies": 0.71875,
"rewards/chosen": 0.0023818810004740953,
"rewards/margins": 0.0072565278969705105,
"rewards/rejected": -0.0048746466636657715,
"step": 70
},
{
"epoch": 0.17,
"learning_rate": 4.9245815365216115e-06,
"logits/chosen": 0.11595060676336288,
"logits/rejected": 0.18608702719211578,
"logps/chosen": -328.6256103515625,
"logps/rejected": -365.93896484375,
"loss": 0.3663,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.002104334533214569,
"rewards/margins": 0.011472588405013084,
"rewards/rejected": -0.013576922006905079,
"step": 80
},
{
"epoch": 0.19,
"learning_rate": 4.872399318152594e-06,
"logits/chosen": 0.08527339994907379,
"logits/rejected": 0.11540959030389786,
"logps/chosen": -352.0487060546875,
"logps/rejected": -361.36578369140625,
"loss": 0.3456,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.011672710999846458,
"rewards/margins": 0.01901327446103096,
"rewards/rejected": -0.03068598173558712,
"step": 90
},
{
"epoch": 0.21,
"learning_rate": 4.807012604511542e-06,
"logits/chosen": 0.14592930674552917,
"logits/rejected": 0.18039533495903015,
"logps/chosen": -414.8818359375,
"logps/rejected": -435.3639221191406,
"loss": 0.34,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.04601982235908508,
"rewards/margins": 0.034013133496046066,
"rewards/rejected": -0.08003295958042145,
"step": 100
},
{
"epoch": 0.23,
"learning_rate": 4.728785330347771e-06,
"logits/chosen": 0.13395074009895325,
"logits/rejected": 0.2120208740234375,
"logps/chosen": -432.98486328125,
"logps/rejected": -553.6598510742188,
"loss": 0.305,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.11783289909362793,
"rewards/margins": 0.08183668553829193,
"rewards/rejected": -0.19966959953308105,
"step": 110
},
{
"epoch": 0.26,
"learning_rate": 4.63815289945858e-06,
"logits/chosen": 0.14807412028312683,
"logits/rejected": 0.21360798180103302,
"logps/chosen": -565.0419311523438,
"logps/rejected": -595.3816528320312,
"loss": 0.3434,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.18253007531166077,
"rewards/margins": 0.060289014130830765,
"rewards/rejected": -0.24281907081604004,
"step": 120
},
{
"epoch": 0.28,
"learning_rate": 4.535619761282989e-06,
"logits/chosen": 0.10818658024072647,
"logits/rejected": 0.18194182217121124,
"logps/chosen": -462.3949279785156,
"logps/rejected": -513.8490600585938,
"loss": 0.3173,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.1326686441898346,
"rewards/margins": 0.0804656594991684,
"rewards/rejected": -0.2131342887878418,
"step": 130
},
{
"epoch": 0.3,
"learning_rate": 4.42175660319555e-06,
"logits/chosen": 0.13493295013904572,
"logits/rejected": 0.22737479209899902,
"logps/chosen": -532.4095458984375,
"logps/rejected": -585.3455200195312,
"loss": 0.3196,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.16629299521446228,
"rewards/margins": 0.08414360880851746,
"rewards/rejected": -0.25043657422065735,
"step": 140
},
{
"epoch": 0.32,
"learning_rate": 4.297197174127619e-06,
"logits/chosen": 0.17478415369987488,
"logits/rejected": 0.24990789592266083,
"logps/chosen": -479.77862548828125,
"logps/rejected": -553.7377319335938,
"loss": 0.3207,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.14474426209926605,
"rewards/margins": 0.09155549854040146,
"rewards/rejected": -0.2362997829914093,
"step": 150
},
{
"epoch": 0.34,
"learning_rate": 4.162634757195418e-06,
"logits/chosen": 0.13402113318443298,
"logits/rejected": 0.2551622688770294,
"logps/chosen": -496.40081787109375,
"logps/rejected": -558.84326171875,
"loss": 0.3187,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.13809171319007874,
"rewards/margins": 0.08086591213941574,
"rewards/rejected": -0.21895763278007507,
"step": 160
},
{
"epoch": 0.36,
"learning_rate": 4.018818310967843e-06,
"logits/chosen": 0.12252243608236313,
"logits/rejected": 0.16481925547122955,
"logps/chosen": -484.56353759765625,
"logps/rejected": -554.2274780273438,
"loss": 0.3255,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.14209917187690735,
"rewards/margins": 0.09574152529239655,
"rewards/rejected": -0.2378406971693039,
"step": 170
},
{
"epoch": 0.38,
"learning_rate": 3.866548300851254e-06,
"logits/chosen": 0.08216498792171478,
"logits/rejected": 0.17952165007591248,
"logps/chosen": -482.576171875,
"logps/rejected": -579.781982421875,
"loss": 0.3047,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.14626096189022064,
"rewards/margins": 0.08570893108844757,
"rewards/rejected": -0.23196987807750702,
"step": 180
},
{
"epoch": 0.41,
"learning_rate": 3.706672243793271e-06,
"logits/chosen": 0.07855963706970215,
"logits/rejected": 0.07844971120357513,
"logps/chosen": -468.69061279296875,
"logps/rejected": -544.0549926757812,
"loss": 0.2935,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.1354297697544098,
"rewards/margins": 0.09144213050603867,
"rewards/rejected": -0.22687189280986786,
"step": 190
},
{
"epoch": 0.43,
"learning_rate": 3.5400799911032357e-06,
"logits/chosen": 0.10545216500759125,
"logits/rejected": 0.1882828176021576,
"logps/chosen": -501.1815490722656,
"logps/rejected": -601.7250366210938,
"loss": 0.3041,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.15607957541942596,
"rewards/margins": 0.10436417162418365,
"rewards/rejected": -0.2604437470436096,
"step": 200
},
{
"epoch": 0.45,
"learning_rate": 3.3676987756445894e-06,
"logits/chosen": 0.10487540811300278,
"logits/rejected": 0.11818546056747437,
"logps/chosen": -470.6344299316406,
"logps/rejected": -565.8145751953125,
"loss": 0.3148,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.16682696342468262,
"rewards/margins": 0.09452913701534271,
"rewards/rejected": -0.26135605573654175,
"step": 210
},
{
"epoch": 0.47,
"learning_rate": 3.1904880509659397e-06,
"logits/chosen": 0.13482534885406494,
"logits/rejected": 0.20024776458740234,
"logps/chosen": -510.10528564453125,
"logps/rejected": -605.6468505859375,
"loss": 0.3172,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.16119703650474548,
"rewards/margins": 0.1006912812590599,
"rewards/rejected": -0.261888325214386,
"step": 220
},
{
"epoch": 0.49,
"learning_rate": 3.0094341510955697e-06,
"logits/chosen": 0.10005593299865723,
"logits/rejected": 0.1791614145040512,
"logps/chosen": -532.1925048828125,
"logps/rejected": -624.0726318359375,
"loss": 0.3106,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.17168700695037842,
"rewards/margins": 0.09216924011707306,
"rewards/rejected": -0.26385626196861267,
"step": 230
},
{
"epoch": 0.51,
"learning_rate": 2.825544800722376e-06,
"logits/chosen": 0.10918021202087402,
"logits/rejected": 0.18382051587104797,
"logps/chosen": -508.88494873046875,
"logps/rejected": -571.0011596679688,
"loss": 0.309,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.15795882046222687,
"rewards/margins": 0.09128745645284653,
"rewards/rejected": -0.2492462694644928,
"step": 240
},
{
"epoch": 0.53,
"learning_rate": 2.639843506318899e-06,
"logits/chosen": 0.12444597482681274,
"logits/rejected": 0.16049379110336304,
"logps/chosen": -509.65631103515625,
"logps/rejected": -609.0398559570312,
"loss": 0.2924,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.1560695320367813,
"rewards/margins": 0.11202052980661392,
"rewards/rejected": -0.26809003949165344,
"step": 250
},
{
"epoch": 0.55,
"learning_rate": 2.4533638594248094e-06,
"logits/chosen": 0.11506851017475128,
"logits/rejected": 0.1052849292755127,
"logps/chosen": -535.1851806640625,
"logps/rejected": -609.4058837890625,
"loss": 0.3139,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.18579542636871338,
"rewards/margins": 0.10177616029977798,
"rewards/rejected": -0.28757157921791077,
"step": 260
},
{
"epoch": 0.58,
"learning_rate": 2.2671437837980943e-06,
"logits/chosen": 0.11827238649129868,
"logits/rejected": 0.12189098447561264,
"logps/chosen": -542.7433471679688,
"logps/rejected": -633.5802612304688,
"loss": 0.2867,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.19100908935070038,
"rewards/margins": 0.09670811891555786,
"rewards/rejected": -0.28771719336509705,
"step": 270
},
{
"epoch": 0.6,
"learning_rate": 2.082219758453629e-06,
"logits/chosen": 0.0904841274023056,
"logits/rejected": 0.20232203602790833,
"logps/chosen": -575.5374755859375,
"logps/rejected": -672.4290161132812,
"loss": 0.3105,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.2171885222196579,
"rewards/margins": 0.10769243538379669,
"rewards/rejected": -0.3248809278011322,
"step": 280
},
{
"epoch": 0.62,
"learning_rate": 1.899621048743019e-06,
"logits/chosen": 0.07801838964223862,
"logits/rejected": 0.16570156812667847,
"logps/chosen": -556.6237182617188,
"logps/rejected": -643.4578857421875,
"loss": 0.3145,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.20063337683677673,
"rewards/margins": 0.1107235923409462,
"rewards/rejected": -0.31135696172714233,
"step": 290
},
{
"epoch": 0.64,
"learning_rate": 1.7203639775848423e-06,
"logits/chosen": 0.07458348572254181,
"logits/rejected": 0.08251482248306274,
"logps/chosen": -524.4200439453125,
"logps/rejected": -666.1168823242188,
"loss": 0.2934,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.18233875930309296,
"rewards/margins": 0.13505366444587708,
"rewards/rejected": -0.3173924386501312,
"step": 300
},
{
"epoch": 0.66,
"learning_rate": 1.5454462687309445e-06,
"logits/chosen": 0.06379405409097672,
"logits/rejected": 0.14568018913269043,
"logps/chosen": -504.7530822753906,
"logps/rejected": -650.941650390625,
"loss": 0.2814,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.17345736920833588,
"rewards/margins": 0.1400579959154129,
"rewards/rejected": -0.3135153353214264,
"step": 310
},
{
"epoch": 0.68,
"learning_rate": 1.3758414935535147e-06,
"logits/chosen": 0.0773477703332901,
"logits/rejected": 0.12101063877344131,
"logps/chosen": -536.0679931640625,
"logps/rejected": -659.2662353515625,
"loss": 0.3014,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.18032298982143402,
"rewards/margins": 0.11869337409734726,
"rewards/rejected": -0.29901641607284546,
"step": 320
},
{
"epoch": 0.7,
"learning_rate": 1.2124936522614622e-06,
"logits/chosen": 0.09862785786390305,
"logits/rejected": 0.14109982550144196,
"logps/chosen": -487.79071044921875,
"logps/rejected": -610.7267456054688,
"loss": 0.2935,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.18224193155765533,
"rewards/margins": 0.10070188343524933,
"rewards/rejected": -0.2829437851905823,
"step": 330
},
{
"epoch": 0.73,
"learning_rate": 1.0563119197063934e-06,
"logits/chosen": 0.09601452201604843,
"logits/rejected": 0.13355228304862976,
"logps/chosen": -478.78143310546875,
"logps/rejected": -622.6370239257812,
"loss": 0.31,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.17518463730812073,
"rewards/margins": 0.11871640384197235,
"rewards/rejected": -0.2939010262489319,
"step": 340
},
{
"epoch": 0.75,
"learning_rate": 9.081655850224449e-07,
"logits/chosen": 0.14273716509342194,
"logits/rejected": 0.12122112512588501,
"logps/chosen": -535.0431518554688,
"logps/rejected": -643.0104370117188,
"loss": 0.3196,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.19643327593803406,
"rewards/margins": 0.10987784713506699,
"rewards/rejected": -0.30631113052368164,
"step": 350
},
{
"epoch": 0.77,
"learning_rate": 7.688792132653111e-07,
"logits/chosen": 0.1597367525100708,
"logits/rejected": 0.18071278929710388,
"logps/chosen": -597.0608520507812,
"logps/rejected": -692.2689208984375,
"loss": 0.2965,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.20307877659797668,
"rewards/margins": 0.12918424606323242,
"rewards/rejected": -0.3322630524635315,
"step": 360
},
{
"epoch": 0.79,
"learning_rate": 6.392280559802341e-07,
"logits/chosen": 0.1371072232723236,
"logits/rejected": 0.16206106543540955,
"logps/chosen": -573.9052124023438,
"logps/rejected": -700.1799926757812,
"loss": 0.2689,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.2099401205778122,
"rewards/margins": 0.13856378197669983,
"rewards/rejected": -0.34850388765335083,
"step": 370
},
{
"epoch": 0.81,
"learning_rate": 5.199337362431792e-07,
"logits/chosen": 0.10267746448516846,
"logits/rejected": 0.1483933925628662,
"logps/chosen": -554.7241821289062,
"logps/rejected": -637.0789184570312,
"loss": 0.2999,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.19773730635643005,
"rewards/margins": 0.11398313194513321,
"rewards/rejected": -0.31172046065330505,
"step": 380
},
{
"epoch": 0.83,
"learning_rate": 4.1166023219176176e-07,
"logits/chosen": 0.11247100681066513,
"logits/rejected": 0.1309679001569748,
"logps/chosen": -557.1490478515625,
"logps/rejected": -678.8016357421875,
"loss": 0.2945,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.1877584308385849,
"rewards/margins": 0.12902548909187317,
"rewards/rejected": -0.3167839050292969,
"step": 390
},
{
"epoch": 0.85,
"learning_rate": 3.150101814011136e-07,
"logits/chosen": 0.16857033967971802,
"logits/rejected": 0.18106935918331146,
"logps/chosen": -571.0828857421875,
"logps/rejected": -638.4654541015625,
"loss": 0.3045,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.19992712140083313,
"rewards/margins": 0.10413169860839844,
"rewards/rejected": -0.30405884981155396,
"step": 400
},
{
"epoch": 0.87,
"learning_rate": 2.3052152667409289e-07,
"logits/chosen": 0.09823437035083771,
"logits/rejected": 0.23346427083015442,
"logps/chosen": -537.1575317382812,
"logps/rejected": -657.79931640625,
"loss": 0.3101,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.19983352720737457,
"rewards/margins": 0.11520209163427353,
"rewards/rejected": -0.3150356113910675,
"step": 410
},
{
"epoch": 0.9,
"learning_rate": 1.5866452191498488e-07,
"logits/chosen": 0.13151055574417114,
"logits/rejected": 0.17384907603263855,
"logps/chosen": -562.8436279296875,
"logps/rejected": -688.2677612304688,
"loss": 0.2892,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.21342253684997559,
"rewards/margins": 0.1261134147644043,
"rewards/rejected": -0.3395359516143799,
"step": 420
},
{
"epoch": 0.92,
"learning_rate": 9.983911475163727e-08,
"logits/chosen": 0.11608059704303741,
"logits/rejected": 0.13708294928073883,
"logps/chosen": -503.0716857910156,
"logps/rejected": -606.0911865234375,
"loss": 0.2983,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.18081562221050262,
"rewards/margins": 0.11758317053318024,
"rewards/rejected": -0.29839879274368286,
"step": 430
},
{
"epoch": 0.94,
"learning_rate": 5.437272047405712e-08,
"logits/chosen": 0.11100079119205475,
"logits/rejected": 0.13695240020751953,
"logps/chosen": -530.9528198242188,
"logps/rejected": -668.3643798828125,
"loss": 0.3095,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.2087739259004593,
"rewards/margins": 0.13447535037994385,
"rewards/rejected": -0.34324929118156433,
"step": 440
},
{
"epoch": 0.96,
"learning_rate": 2.251839967945535e-08,
"logits/chosen": 0.022059569135308266,
"logits/rejected": 0.12616530060768127,
"logps/chosen": -492.80181884765625,
"logps/rejected": -643.1315307617188,
"loss": 0.2834,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.18623578548431396,
"rewards/margins": 0.14532844722270966,
"rewards/rejected": -0.3315642178058624,
"step": 450
},
{
"epoch": 0.98,
"learning_rate": 4.453449766758933e-09,
"logits/chosen": 0.10801200568675995,
"logits/rejected": 0.11458040773868561,
"logps/chosen": -531.0709228515625,
"logps/rejected": -645.5343627929688,
"loss": 0.3043,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.1950865238904953,
"rewards/margins": 0.11565764993429184,
"rewards/rejected": -0.3107442259788513,
"step": 460
},
{
"epoch": 1.0,
"step": 468,
"total_flos": 0.0,
"train_loss": 0.2399272450015076,
"train_runtime": 4417.3525,
"train_samples_per_second": 3.396,
"train_steps_per_second": 0.106
}
],
"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
}