lambda-llama-3-8b-ipo-test / trainer_state.json
tanliboy's picture
Model save
ae359f3 verified
raw
history blame
34.9 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9995638901003053,
"eval_steps": 100,
"global_step": 573,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0017444395987788923,
"grad_norm": 29.747415913018244,
"learning_rate": 8.620689655172413e-09,
"logits/chosen": -2.9494614601135254,
"logits/rejected": -2.9794743061065674,
"logps/chosen": -1.7114274501800537,
"logps/rejected": -1.7004308700561523,
"loss": 1.5625,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.01744439598778892,
"grad_norm": 25.590469762768826,
"learning_rate": 8.620689655172414e-08,
"logits/chosen": -2.272578477859497,
"logits/rejected": -2.341148614883423,
"logps/chosen": -1.5401281118392944,
"logps/rejected": -1.708099126815796,
"loss": 1.5627,
"rewards/accuracies": 0.4513888955116272,
"rewards/chosen": 0.000227387499762699,
"rewards/margins": 0.0002251576370326802,
"rewards/rejected": 2.2298513613350224e-06,
"step": 10
},
{
"epoch": 0.03488879197557784,
"grad_norm": 25.207047594807733,
"learning_rate": 1.7241379310344828e-07,
"logits/chosen": -2.485241413116455,
"logits/rejected": -2.5959410667419434,
"logps/chosen": -1.570024013519287,
"logps/rejected": -1.7283653020858765,
"loss": 1.5607,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.00038363353814929724,
"rewards/margins": -9.226792826666497e-06,
"rewards/rejected": -0.0003744067216757685,
"step": 20
},
{
"epoch": 0.05233318796336677,
"grad_norm": 31.534894880772583,
"learning_rate": 2.586206896551724e-07,
"logits/chosen": -2.1619415283203125,
"logits/rejected": -2.2504525184631348,
"logps/chosen": -1.6232503652572632,
"logps/rejected": -1.7250868082046509,
"loss": 1.5432,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.0028999350033700466,
"rewards/margins": 0.0026171673089265823,
"rewards/rejected": -0.005517102312296629,
"step": 30
},
{
"epoch": 0.06977758395115569,
"grad_norm": 26.41926451522702,
"learning_rate": 3.4482758620689656e-07,
"logits/chosen": -2.274902582168579,
"logits/rejected": -2.4139533042907715,
"logps/chosen": -1.5764671564102173,
"logps/rejected": -1.6893203258514404,
"loss": 1.509,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.007333276327699423,
"rewards/margins": 0.00940671842545271,
"rewards/rejected": -0.016739998012781143,
"step": 40
},
{
"epoch": 0.08722197993894461,
"grad_norm": 24.178200461155868,
"learning_rate": 4.310344827586206e-07,
"logits/chosen": -2.3129639625549316,
"logits/rejected": -2.319011926651001,
"logps/chosen": -1.5989129543304443,
"logps/rejected": -1.7941827774047852,
"loss": 1.4196,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.026371877640485764,
"rewards/margins": 0.03227262943983078,
"rewards/rejected": -0.058644503355026245,
"step": 50
},
{
"epoch": 0.10466637592673354,
"grad_norm": 38.93662482016604,
"learning_rate": 4.999813941333237e-07,
"logits/chosen": -2.455597400665283,
"logits/rejected": -2.459132194519043,
"logps/chosen": -1.9030494689941406,
"logps/rejected": -2.0835158824920654,
"loss": 1.3361,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.10062487423419952,
"rewards/margins": 0.051896799355745316,
"rewards/rejected": -0.15252165496349335,
"step": 60
},
{
"epoch": 0.12211077191452246,
"grad_norm": 35.7973285976575,
"learning_rate": 4.99330479543236e-07,
"logits/chosen": -2.394988536834717,
"logits/rejected": -2.4527833461761475,
"logps/chosen": -1.8048909902572632,
"logps/rejected": -2.0864245891571045,
"loss": 1.3198,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.11194689571857452,
"rewards/margins": 0.057884473353624344,
"rewards/rejected": -0.16983136534690857,
"step": 70
},
{
"epoch": 0.13955516790231137,
"grad_norm": 40.16495399833959,
"learning_rate": 4.977520391931686e-07,
"logits/chosen": -2.266960620880127,
"logits/rejected": -2.373418092727661,
"logps/chosen": -1.8538503646850586,
"logps/rejected": -2.1406590938568115,
"loss": 1.2319,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.1073283776640892,
"rewards/margins": 0.09412004053592682,
"rewards/rejected": -0.2014484405517578,
"step": 80
},
{
"epoch": 0.1569995638901003,
"grad_norm": 46.075075778371485,
"learning_rate": 4.952519449848961e-07,
"logits/chosen": -2.337573289871216,
"logits/rejected": -2.3125760555267334,
"logps/chosen": -2.0118062496185303,
"logps/rejected": -2.2410480976104736,
"loss": 1.187,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.15836110711097717,
"rewards/margins": 0.09043124318122864,
"rewards/rejected": -0.248792365193367,
"step": 90
},
{
"epoch": 0.17444395987788922,
"grad_norm": 35.52679424110663,
"learning_rate": 4.9183949743308e-07,
"logits/chosen": -2.266244411468506,
"logits/rejected": -2.32613205909729,
"logps/chosen": -1.9573999643325806,
"logps/rejected": -2.245410919189453,
"loss": 1.1749,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.15664055943489075,
"rewards/margins": 0.1040625348687172,
"rewards/rejected": -0.26070311665534973,
"step": 100
},
{
"epoch": 0.17444395987788922,
"eval_logits/chosen": -2.5133261680603027,
"eval_logits/rejected": -2.567593812942505,
"eval_logps/chosen": -2.063753604888916,
"eval_logps/rejected": -2.446528673171997,
"eval_loss": 1.0763452053070068,
"eval_rewards/accuracies": 0.7891566157341003,
"eval_rewards/chosen": -0.17315217852592468,
"eval_rewards/margins": 0.1388266682624817,
"eval_rewards/rejected": -0.31197884678840637,
"eval_runtime": 115.6498,
"eval_samples_per_second": 22.853,
"eval_steps_per_second": 0.718,
"step": 100
},
{
"epoch": 0.19188835586567815,
"grad_norm": 34.94980526994547,
"learning_rate": 4.875273910667434e-07,
"logits/chosen": -2.5348660945892334,
"logits/rejected": -2.541141986846924,
"logps/chosen": -2.0890326499938965,
"logps/rejected": -2.563629627227783,
"loss": 1.0846,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.21275702118873596,
"rewards/margins": 0.15415263175964355,
"rewards/rejected": -0.3669096529483795,
"step": 110
},
{
"epoch": 0.2093327518534671,
"grad_norm": 36.111192447827186,
"learning_rate": 4.823316672047889e-07,
"logits/chosen": -2.482753276824951,
"logits/rejected": -2.542470932006836,
"logps/chosen": -2.19625186920166,
"logps/rejected": -2.7020630836486816,
"loss": 1.0846,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.20946833491325378,
"rewards/margins": 0.1580672562122345,
"rewards/rejected": -0.3675355613231659,
"step": 120
},
{
"epoch": 0.226777147841256,
"grad_norm": 35.9198962249818,
"learning_rate": 4.762716542812394e-07,
"logits/chosen": -2.5084481239318848,
"logits/rejected": -2.484107732772827,
"logps/chosen": -2.347628116607666,
"logps/rejected": -2.8013808727264404,
"loss": 1.0397,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.2825542390346527,
"rewards/margins": 0.14921075105667114,
"rewards/rejected": -0.43176499009132385,
"step": 130
},
{
"epoch": 0.24422154382904493,
"grad_norm": 33.812969950120994,
"learning_rate": 4.693698959421934e-07,
"logits/chosen": -2.4390926361083984,
"logits/rejected": -2.5850162506103516,
"logps/chosen": -2.2307400703430176,
"logps/rejected": -2.76652193069458,
"loss": 1.1052,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.2658195495605469,
"rewards/margins": 0.15902426838874817,
"rewards/rejected": -0.42484384775161743,
"step": 140
},
{
"epoch": 0.26166593981683384,
"grad_norm": 38.15404463326893,
"learning_rate": 4.616520671819811e-07,
"logits/chosen": -2.5228819847106934,
"logits/rejected": -2.5446176528930664,
"logps/chosen": -2.410109043121338,
"logps/rejected": -2.782252550125122,
"loss": 1.0492,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.31940045952796936,
"rewards/margins": 0.1365361362695694,
"rewards/rejected": -0.45593661069869995,
"step": 150
},
{
"epoch": 0.27911033580462274,
"grad_norm": 37.67274567595706,
"learning_rate": 4.531468788304991e-07,
"logits/chosen": -2.496253252029419,
"logits/rejected": -2.5734314918518066,
"logps/chosen": -2.237454414367676,
"logps/rejected": -2.7140257358551025,
"loss": 1.0272,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.27850186824798584,
"rewards/margins": 0.15557342767715454,
"rewards/rejected": -0.4340752959251404,
"step": 160
},
{
"epoch": 0.2965547317924117,
"grad_norm": 49.34543198418806,
"learning_rate": 4.438859707470375e-07,
"logits/chosen": -2.530522108078003,
"logits/rejected": -2.5276732444763184,
"logps/chosen": -2.332202434539795,
"logps/rejected": -2.7523207664489746,
"loss": 1.0392,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.2833098769187927,
"rewards/margins": 0.1833934336900711,
"rewards/rejected": -0.466703325510025,
"step": 170
},
{
"epoch": 0.3139991277802006,
"grad_norm": 44.17730811384066,
"learning_rate": 4.3390379411792524e-07,
"logits/chosen": -2.4329633712768555,
"logits/rejected": -2.5291526317596436,
"logps/chosen": -2.313528537750244,
"logps/rejected": -2.7951102256774902,
"loss": 1.0006,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.3000316917896271,
"rewards/margins": 0.158560112118721,
"rewards/rejected": -0.4585917890071869,
"step": 180
},
{
"epoch": 0.3314435237679895,
"grad_norm": 47.45322605710491,
"learning_rate": 4.23237483295854e-07,
"logits/chosen": -2.573887586593628,
"logits/rejected": -2.7052321434020996,
"logps/chosen": -2.364560127258301,
"logps/rejected": -2.8203182220458984,
"loss": 1.0715,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.3148408532142639,
"rewards/margins": 0.16490553319454193,
"rewards/rejected": -0.47974634170532227,
"step": 190
},
{
"epoch": 0.34888791975577843,
"grad_norm": 36.54817057924313,
"learning_rate": 4.119267176576475e-07,
"logits/chosen": -2.688988208770752,
"logits/rejected": -2.7872579097747803,
"logps/chosen": -2.3317620754241943,
"logps/rejected": -2.948866844177246,
"loss": 0.9802,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.3193766474723816,
"rewards/margins": 0.1907302290201187,
"rewards/rejected": -0.5101069211959839,
"step": 200
},
{
"epoch": 0.34888791975577843,
"eval_logits/chosen": -2.7230396270751953,
"eval_logits/rejected": -2.7872889041900635,
"eval_logps/chosen": -2.4268717765808105,
"eval_logps/rejected": -2.992184638977051,
"eval_loss": 0.9500909447669983,
"eval_rewards/accuracies": 0.8012048006057739,
"eval_rewards/chosen": -0.31839942932128906,
"eval_rewards/margins": 0.2118416577577591,
"eval_rewards/rejected": -0.530241072177887,
"eval_runtime": 115.5249,
"eval_samples_per_second": 22.878,
"eval_steps_per_second": 0.718,
"step": 200
},
{
"epoch": 0.3663323157435674,
"grad_norm": 28.997542149982813,
"learning_rate": 4.000135739943734e-07,
"logits/chosen": -2.5338735580444336,
"logits/rejected": -2.5965075492858887,
"logps/chosen": -2.3301658630371094,
"logps/rejected": -2.923480749130249,
"loss": 1.017,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.2939096689224243,
"rewards/margins": 0.18346476554870605,
"rewards/rejected": -0.477374404668808,
"step": 210
},
{
"epoch": 0.3837767117313563,
"grad_norm": 79.37314946467663,
"learning_rate": 3.875423699829168e-07,
"logits/chosen": -2.5550951957702637,
"logits/rejected": -2.769721508026123,
"logps/chosen": -2.377946615219116,
"logps/rejected": -3.049376964569092,
"loss": 0.9655,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.33743661642074585,
"rewards/margins": 0.20777985453605652,
"rewards/rejected": -0.5452165007591248,
"step": 220
},
{
"epoch": 0.4012211077191452,
"grad_norm": 38.56972950284146,
"learning_rate": 3.7455949932131175e-07,
"logits/chosen": -2.853245258331299,
"logits/rejected": -2.7923905849456787,
"logps/chosen": -2.5535457134246826,
"logps/rejected": -3.186497688293457,
"loss": 1.0501,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -0.37545865774154663,
"rewards/margins": 0.21732258796691895,
"rewards/rejected": -0.5927812457084656,
"step": 230
},
{
"epoch": 0.4186655037069342,
"grad_norm": 34.32127443417028,
"learning_rate": 3.611132591411369e-07,
"logits/chosen": -2.845720052719116,
"logits/rejected": -2.912646770477295,
"logps/chosen": -2.471998691558838,
"logps/rejected": -2.9336795806884766,
"loss": 0.9871,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.3210769295692444,
"rewards/margins": 0.18005797266960144,
"rewards/rejected": -0.5011348724365234,
"step": 240
},
{
"epoch": 0.4361098996947231,
"grad_norm": 36.8304920546943,
"learning_rate": 3.4725367033901473e-07,
"logits/chosen": -2.7427520751953125,
"logits/rejected": -2.9484190940856934,
"logps/chosen": -2.521329879760742,
"logps/rejected": -3.1583688259124756,
"loss": 1.018,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.3658570647239685,
"rewards/margins": 0.20647919178009033,
"rewards/rejected": -0.5723362565040588,
"step": 250
},
{
"epoch": 0.453554295682512,
"grad_norm": 29.383883913798257,
"learning_rate": 3.3303229149558967e-07,
"logits/chosen": -2.8211348056793213,
"logits/rejected": -2.8875656127929688,
"logps/chosen": -2.307373523712158,
"logps/rejected": -2.855837821960449,
"loss": 0.9869,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.29735279083251953,
"rewards/margins": 0.1880948543548584,
"rewards/rejected": -0.48544764518737793,
"step": 260
},
{
"epoch": 0.4709986916703009,
"grad_norm": 37.77204908034358,
"learning_rate": 3.185020270742225e-07,
"logits/chosen": -2.7716403007507324,
"logits/rejected": -2.8079323768615723,
"logps/chosen": -2.386676549911499,
"logps/rejected": -2.7951130867004395,
"loss": 0.9871,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.32528942823410034,
"rewards/margins": 0.15507872402668,
"rewards/rejected": -0.48036813735961914,
"step": 270
},
{
"epoch": 0.48844308765808986,
"grad_norm": 28.328010250614053,
"learning_rate": 3.0371693061291146e-07,
"logits/chosen": -2.780062437057495,
"logits/rejected": -2.9711029529571533,
"logps/chosen": -2.551729679107666,
"logps/rejected": -3.2051796913146973,
"loss": 1.0056,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.40202435851097107,
"rewards/margins": 0.1922958493232727,
"rewards/rejected": -0.5943201780319214,
"step": 280
},
{
"epoch": 0.5058874836458788,
"grad_norm": 32.14871116699808,
"learning_rate": 2.8873200364158106e-07,
"logits/chosen": -2.680164337158203,
"logits/rejected": -2.803938388824463,
"logps/chosen": -2.43742036819458,
"logps/rejected": -2.9839630126953125,
"loss": 0.9837,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -0.32910409569740295,
"rewards/margins": 0.2097328007221222,
"rewards/rejected": -0.5388368368148804,
"step": 290
},
{
"epoch": 0.5233318796336677,
"grad_norm": 40.52094893481022,
"learning_rate": 2.7360299107277767e-07,
"logits/chosen": -2.7708568572998047,
"logits/rejected": -2.821788787841797,
"logps/chosen": -2.4744372367858887,
"logps/rejected": -3.086942672729492,
"loss": 0.9548,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.3568132221698761,
"rewards/margins": 0.20053787529468536,
"rewards/rejected": -0.5573510527610779,
"step": 300
},
{
"epoch": 0.5233318796336677,
"eval_logits/chosen": -2.808656692504883,
"eval_logits/rejected": -2.8788304328918457,
"eval_logps/chosen": -2.571028709411621,
"eval_logps/rejected": -3.17358660697937,
"eval_loss": 0.9136135578155518,
"eval_rewards/accuracies": 0.8162650465965271,
"eval_rewards/chosen": -0.3760621249675751,
"eval_rewards/margins": 0.22673983871936798,
"eval_rewards/rejected": -0.6028019785881042,
"eval_runtime": 115.45,
"eval_samples_per_second": 22.893,
"eval_steps_per_second": 0.719,
"step": 300
},
{
"epoch": 0.5407762756214566,
"grad_norm": 39.89301839500131,
"learning_rate": 2.5838617382693414e-07,
"logits/chosen": -2.948072671890259,
"logits/rejected": -2.9336767196655273,
"logps/chosen": -2.5350139141082764,
"logps/rejected": -3.0460660457611084,
"loss": 1.0184,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.37652358412742615,
"rewards/margins": 0.18868541717529297,
"rewards/rejected": -0.5652090311050415,
"step": 310
},
{
"epoch": 0.5582206716092455,
"grad_norm": 36.83353696515594,
"learning_rate": 2.4313815946364877e-07,
"logits/chosen": -2.8387434482574463,
"logits/rejected": -2.978883743286133,
"logps/chosen": -2.4741597175598145,
"logps/rejected": -3.0683841705322266,
"loss": 0.9653,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.3760759234428406,
"rewards/margins": 0.1771194189786911,
"rewards/rejected": -0.5531953573226929,
"step": 320
},
{
"epoch": 0.5756650675970345,
"grad_norm": 30.76657514726095,
"learning_rate": 2.2791567159784566e-07,
"logits/chosen": -2.762192487716675,
"logits/rejected": -2.7734150886535645,
"logps/chosen": -2.4779891967773438,
"logps/rejected": -2.9213876724243164,
"loss": 0.9814,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.3409067988395691,
"rewards/margins": 0.18150492012500763,
"rewards/rejected": -0.5224117040634155,
"step": 330
},
{
"epoch": 0.5931094635848234,
"grad_norm": 32.87393386922799,
"learning_rate": 2.12775338884202e-07,
"logits/chosen": -2.886357307434082,
"logits/rejected": -2.9018540382385254,
"logps/chosen": -2.4813778400421143,
"logps/rejected": -2.9721169471740723,
"loss": 0.9893,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.3222178816795349,
"rewards/margins": 0.20408673584461212,
"rewards/rejected": -0.5263046622276306,
"step": 340
},
{
"epoch": 0.6105538595726123,
"grad_norm": 30.494923835102345,
"learning_rate": 1.977734843548328e-07,
"logits/chosen": -2.824707508087158,
"logits/rejected": -2.79854154586792,
"logps/chosen": -2.3759117126464844,
"logps/rejected": -2.975604295730591,
"loss": 0.9863,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.3144580125808716,
"rewards/margins": 0.21930566430091858,
"rewards/rejected": -0.5337637066841125,
"step": 350
},
{
"epoch": 0.6279982555604012,
"grad_norm": 26.8742332382294,
"learning_rate": 1.8296591589391226e-07,
"logits/chosen": -2.6628756523132324,
"logits/rejected": -2.7284369468688965,
"logps/chosen": -2.4052791595458984,
"logps/rejected": -2.870692253112793,
"loss": 0.9646,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.310857355594635,
"rewards/margins": 0.15339362621307373,
"rewards/rejected": -0.46425098180770874,
"step": 360
},
{
"epoch": 0.6454426515481901,
"grad_norm": 35.52686655329713,
"learning_rate": 1.684077186286792e-07,
"logits/chosen": -2.50087308883667,
"logits/rejected": -2.6514532566070557,
"logps/chosen": -2.2871201038360596,
"logps/rejected": -2.937791585922241,
"loss": 0.9877,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.3152330219745636,
"rewards/margins": 0.2090420424938202,
"rewards/rejected": -0.5242750644683838,
"step": 370
},
{
"epoch": 0.662887047535979,
"grad_norm": 30.399222288686712,
"learning_rate": 1.5415305000914585e-07,
"logits/chosen": -2.7632899284362793,
"logits/rejected": -2.886943817138672,
"logps/chosen": -2.4425692558288574,
"logps/rejected": -3.0246245861053467,
"loss": 0.9226,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.33751311898231506,
"rewards/margins": 0.1871272623538971,
"rewards/rejected": -0.5246403813362122,
"step": 380
},
{
"epoch": 0.680331443523768,
"grad_norm": 33.76165947663638,
"learning_rate": 1.4025493833882643e-07,
"logits/chosen": -2.622499465942383,
"logits/rejected": -2.6712939739227295,
"logps/chosen": -2.521423816680908,
"logps/rejected": -3.01672625541687,
"loss": 0.9625,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.39204803109169006,
"rewards/margins": 0.18184103071689606,
"rewards/rejected": -0.5738890767097473,
"step": 390
},
{
"epoch": 0.6977758395115569,
"grad_norm": 39.46957147631396,
"learning_rate": 1.267650855059656e-07,
"logits/chosen": -2.7575411796569824,
"logits/rejected": -2.8080406188964844,
"logps/chosen": -2.511035203933716,
"logps/rejected": -2.9791462421417236,
"loss": 0.9834,
"rewards/accuracies": 0.78125,
"rewards/chosen": -0.3435077667236328,
"rewards/margins": 0.1896781176328659,
"rewards/rejected": -0.5331858396530151,
"step": 400
},
{
"epoch": 0.6977758395115569,
"eval_logits/chosen": -2.8667454719543457,
"eval_logits/rejected": -2.937051773071289,
"eval_logps/chosen": -2.4769790172576904,
"eval_logps/rejected": -3.0509371757507324,
"eval_loss": 0.9041184186935425,
"eval_rewards/accuracies": 0.8042168617248535,
"eval_rewards/chosen": -0.3384423553943634,
"eval_rewards/margins": 0.21529993414878845,
"eval_rewards/rejected": -0.5537422895431519,
"eval_runtime": 115.5756,
"eval_samples_per_second": 22.868,
"eval_steps_per_second": 0.718,
"step": 400
},
{
"epoch": 0.7152202354993459,
"grad_norm": 35.58611489394968,
"learning_rate": 1.1373367464911796e-07,
"logits/chosen": -2.7775580883026123,
"logits/rejected": -2.8066816329956055,
"logps/chosen": -2.44441294670105,
"logps/rejected": -2.940340042114258,
"loss": 0.966,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.3467921018600464,
"rewards/margins": 0.17160984873771667,
"rewards/rejected": -0.5184019804000854,
"step": 410
},
{
"epoch": 0.7326646314871348,
"grad_norm": 31.573311664278364,
"learning_rate": 1.0120918347257668e-07,
"logits/chosen": -2.912554979324341,
"logits/rejected": -2.9200565814971924,
"logps/chosen": -2.5111441612243652,
"logps/rejected": -3.0291829109191895,
"loss": 1.0018,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.3495512306690216,
"rewards/margins": 0.18085847795009613,
"rewards/rejected": -0.5304096937179565,
"step": 420
},
{
"epoch": 0.7501090274749237,
"grad_norm": 38.092625083892315,
"learning_rate": 8.923820390612991e-08,
"logits/chosen": -2.8513288497924805,
"logits/rejected": -2.847677707672119,
"logps/chosen": -2.44001841545105,
"logps/rejected": -3.063814640045166,
"loss": 0.9975,
"rewards/accuracies": 0.84375,
"rewards/chosen": -0.3544333577156067,
"rewards/margins": 0.22024652361869812,
"rewards/rejected": -0.5746799111366272,
"step": 430
},
{
"epoch": 0.7675534234627126,
"grad_norm": 30.479406895571362,
"learning_rate": 7.786526878002125e-08,
"logits/chosen": -2.714656114578247,
"logits/rejected": -2.838855743408203,
"logps/chosen": -2.3737549781799316,
"logps/rejected": -3.0006814002990723,
"loss": 0.9624,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -0.32796674966812134,
"rewards/margins": 0.20443764328956604,
"rewards/rejected": -0.532404363155365,
"step": 440
},
{
"epoch": 0.7849978194505015,
"grad_norm": 33.130716501269866,
"learning_rate": 6.713268615989654e-08,
"logits/chosen": -2.9494526386260986,
"logits/rejected": -2.9998550415039062,
"logps/chosen": -2.509775161743164,
"logps/rejected": -3.015613555908203,
"loss": 0.9668,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.36944177746772766,
"rewards/margins": 0.1895218938589096,
"rewards/rejected": -0.5589635968208313,
"step": 450
},
{
"epoch": 0.8024422154382904,
"grad_norm": 33.25889406980063,
"learning_rate": 5.7080381958020976e-08,
"logits/chosen": -2.674802303314209,
"logits/rejected": -2.8434243202209473,
"logps/chosen": -2.5848803520202637,
"logps/rejected": -3.1880083084106445,
"loss": 0.9423,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -0.376645028591156,
"rewards/margins": 0.20410259068012238,
"rewards/rejected": -0.5807476043701172,
"step": 460
},
{
"epoch": 0.8198866114260793,
"grad_norm": 30.106792425198947,
"learning_rate": 4.774575140626316e-08,
"logits/chosen": -2.7849488258361816,
"logits/rejected": -2.903566360473633,
"logps/chosen": -2.3999593257904053,
"logps/rejected": -3.0301947593688965,
"loss": 0.9476,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -0.3394695520401001,
"rewards/margins": 0.21237297356128693,
"rewards/rejected": -0.5518425703048706,
"step": 470
},
{
"epoch": 0.8373310074138683,
"grad_norm": 37.88465481105631,
"learning_rate": 3.9163519943375965e-08,
"logits/chosen": -2.8227665424346924,
"logits/rejected": -2.9443516731262207,
"logps/chosen": -2.4296436309814453,
"logps/rejected": -2.984886646270752,
"loss": 0.8745,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.35570722818374634,
"rewards/margins": 0.16281530261039734,
"rewards/rejected": -0.5185225605964661,
"step": 480
},
{
"epoch": 0.8547754034016573,
"grad_norm": 30.701274278534907,
"learning_rate": 3.136561403408422e-08,
"logits/chosen": -2.8463587760925293,
"logits/rejected": -3.0236496925354004,
"logps/chosen": -2.5174646377563477,
"logps/rejected": -3.2128849029541016,
"loss": 0.9445,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -0.37872380018234253,
"rewards/margins": 0.22148160636425018,
"rewards/rejected": -0.6002054810523987,
"step": 490
},
{
"epoch": 0.8722197993894462,
"grad_norm": 32.269326624165465,
"learning_rate": 2.4381042400538593e-08,
"logits/chosen": -2.871164560317993,
"logits/rejected": -3.0147228240966797,
"logps/chosen": -2.5578746795654297,
"logps/rejected": -3.273369312286377,
"loss": 0.9967,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.4019395709037781,
"rewards/margins": 0.21136029064655304,
"rewards/rejected": -0.6132999062538147,
"step": 500
},
{
"epoch": 0.8722197993894462,
"eval_logits/chosen": -2.959242105484009,
"eval_logits/rejected": -3.029346227645874,
"eval_logps/chosen": -2.568408966064453,
"eval_logps/rejected": -3.185546398162842,
"eval_loss": 0.8938218951225281,
"eval_rewards/accuracies": 0.7891566157341003,
"eval_rewards/chosen": -0.37501412630081177,
"eval_rewards/margins": 0.23257188498973846,
"eval_rewards/rejected": -0.6075860857963562,
"eval_runtime": 115.5545,
"eval_samples_per_second": 22.872,
"eval_steps_per_second": 0.718,
"step": 500
},
{
"epoch": 0.8896641953772351,
"grad_norm": 35.20455083141279,
"learning_rate": 1.8235788107963945e-08,
"logits/chosen": -2.887869358062744,
"logits/rejected": -2.820462703704834,
"logps/chosen": -2.587712526321411,
"logps/rejected": -3.1114964485168457,
"loss": 0.9135,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.37796396017074585,
"rewards/margins": 0.18830379843711853,
"rewards/rejected": -0.5662677884101868,
"step": 510
},
{
"epoch": 0.907108591365024,
"grad_norm": 29.121513812900695,
"learning_rate": 1.2952711905950376e-08,
"logits/chosen": -2.8955085277557373,
"logits/rejected": -2.908362627029419,
"logps/chosen": -2.5724124908447266,
"logps/rejected": -3.0688400268554688,
"loss": 0.9963,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.39740028977394104,
"rewards/margins": 0.17423763871192932,
"rewards/rejected": -0.5716378688812256,
"step": 520
},
{
"epoch": 0.9245529873528129,
"grad_norm": 30.97739103991955,
"learning_rate": 8.55146718496283e-09,
"logits/chosen": -2.919581890106201,
"logits/rejected": -2.8886361122131348,
"logps/chosen": -2.599856376647949,
"logps/rejected": -2.9770760536193848,
"loss": 0.9551,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -0.3779660165309906,
"rewards/margins": 0.17554500699043274,
"rewards/rejected": -0.5535110235214233,
"step": 530
},
{
"epoch": 0.9419973833406018,
"grad_norm": 44.940688671381224,
"learning_rate": 5.048426864438182e-09,
"logits/chosen": -2.870760440826416,
"logits/rejected": -2.946399688720703,
"logps/chosen": -2.518998146057129,
"logps/rejected": -3.1112277507781982,
"loss": 0.9636,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -0.3970429301261902,
"rewards/margins": 0.18053510785102844,
"rewards/rejected": -0.577578067779541,
"step": 540
},
{
"epoch": 0.9594417793283908,
"grad_norm": 33.602758621496534,
"learning_rate": 2.456622484449139e-09,
"logits/chosen": -2.9532032012939453,
"logits/rejected": -3.0146090984344482,
"logps/chosen": -2.5181241035461426,
"logps/rejected": -3.0724215507507324,
"loss": 0.9918,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -0.35534483194351196,
"rewards/margins": 0.210611030459404,
"rewards/rejected": -0.5659558176994324,
"step": 550
},
{
"epoch": 0.9768861753161797,
"grad_norm": 40.276290803183436,
"learning_rate": 7.85695727519492e-10,
"logits/chosen": -2.8332512378692627,
"logits/rejected": -2.8909554481506348,
"logps/chosen": -2.6798324584960938,
"logps/rejected": -3.1653189659118652,
"loss": 0.957,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.3789713382720947,
"rewards/margins": 0.16679120063781738,
"rewards/rejected": -0.5457625389099121,
"step": 560
},
{
"epoch": 0.9943305713039686,
"grad_norm": 44.66764267907785,
"learning_rate": 4.1862550931942575e-11,
"logits/chosen": -2.9107937812805176,
"logits/rejected": -2.905510425567627,
"logps/chosen": -2.4948418140411377,
"logps/rejected": -2.9094674587249756,
"loss": 0.965,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -0.3739563822746277,
"rewards/margins": 0.17874310910701752,
"rewards/rejected": -0.5526994466781616,
"step": 570
},
{
"epoch": 0.9995638901003053,
"step": 573,
"total_flos": 0.0,
"train_loss": 1.0578667073141634,
"train_runtime": 8389.7514,
"train_samples_per_second": 8.743,
"train_steps_per_second": 0.068
}
],
"logging_steps": 10,
"max_steps": 573,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"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
}