aftonposten-6b-align-scan / trainer_state.json
hugodk-sch's picture
Model save
7b32e49 verified
raw
history blame
83.9 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 4.0,
"eval_steps": 100,
"global_step": 1540,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 1.282051282051282e-07,
"logits/chosen": -1.7278180122375488,
"logits/rejected": -1.7377450466156006,
"logps/chosen": -29.553977966308594,
"logps/rejected": -42.813133239746094,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.03,
"learning_rate": 1.282051282051282e-06,
"logits/chosen": -1.8663169145584106,
"logits/rejected": -1.870638370513916,
"logps/chosen": -36.98221206665039,
"logps/rejected": -33.6473503112793,
"loss": 0.6788,
"rewards/accuracies": 0.5416666865348816,
"rewards/chosen": 0.01950961910188198,
"rewards/margins": 0.03332838416099548,
"rewards/rejected": -0.013818766921758652,
"step": 10
},
{
"epoch": 0.05,
"learning_rate": 2.564102564102564e-06,
"logits/chosen": -1.9978923797607422,
"logits/rejected": -2.0005345344543457,
"logps/chosen": -29.642324447631836,
"logps/rejected": -29.048343658447266,
"loss": 0.7013,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.00010492801811778918,
"rewards/margins": -0.012185259722173214,
"rewards/rejected": 0.012080332264304161,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 3.846153846153847e-06,
"logits/chosen": -1.9209339618682861,
"logits/rejected": -1.9182507991790771,
"logps/chosen": -31.401519775390625,
"logps/rejected": -33.22309875488281,
"loss": 0.6891,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": 0.011660982854664326,
"rewards/margins": 0.015369392931461334,
"rewards/rejected": -0.003708411008119583,
"step": 30
},
{
"epoch": 0.1,
"learning_rate": 4.999896948438434e-06,
"logits/chosen": -2.0180399417877197,
"logits/rejected": -2.009289264678955,
"logps/chosen": -32.559410095214844,
"logps/rejected": -32.52582550048828,
"loss": 0.6849,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.013918718323111534,
"rewards/margins": 0.02444135770201683,
"rewards/rejected": -0.010522643104195595,
"step": 40
},
{
"epoch": 0.13,
"learning_rate": 4.987541037542187e-06,
"logits/chosen": -1.8629716634750366,
"logits/rejected": -1.8522107601165771,
"logps/chosen": -33.554229736328125,
"logps/rejected": -35.44757080078125,
"loss": 0.6987,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": 0.002600290346890688,
"rewards/margins": -0.0012185067171230912,
"rewards/rejected": 0.003818795783445239,
"step": 50
},
{
"epoch": 0.16,
"learning_rate": 4.954691471941119e-06,
"logits/chosen": -1.9408414363861084,
"logits/rejected": -1.9427950382232666,
"logps/chosen": -32.56097412109375,
"logps/rejected": -33.213417053222656,
"loss": 0.6725,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.03136637061834335,
"rewards/margins": 0.06499636918306351,
"rewards/rejected": -0.03362999111413956,
"step": 60
},
{
"epoch": 0.18,
"learning_rate": 4.901618883413549e-06,
"logits/chosen": -2.072221517562866,
"logits/rejected": -2.077198028564453,
"logps/chosen": -33.974578857421875,
"logps/rejected": -36.629173278808594,
"loss": 0.6793,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.011124782264232635,
"rewards/margins": 0.059367585927248,
"rewards/rejected": -0.04824279993772507,
"step": 70
},
{
"epoch": 0.21,
"learning_rate": 4.828760511501322e-06,
"logits/chosen": -1.9338233470916748,
"logits/rejected": -1.9369605779647827,
"logps/chosen": -34.30416488647461,
"logps/rejected": -34.634437561035156,
"loss": 0.6423,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.09891629219055176,
"rewards/margins": 0.132537841796875,
"rewards/rejected": -0.03362155705690384,
"step": 80
},
{
"epoch": 0.23,
"learning_rate": 4.7367166013034295e-06,
"logits/chosen": -1.9418385028839111,
"logits/rejected": -1.9463545083999634,
"logps/chosen": -32.39947509765625,
"logps/rejected": -32.35419464111328,
"loss": 0.6785,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.05435952544212341,
"rewards/margins": 0.05058818310499191,
"rewards/rejected": 0.0037713423371315002,
"step": 90
},
{
"epoch": 0.26,
"learning_rate": 4.626245458345211e-06,
"logits/chosen": -2.039783000946045,
"logits/rejected": -2.037789821624756,
"logps/chosen": -32.164188385009766,
"logps/rejected": -31.309520721435547,
"loss": 0.6538,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.061319977045059204,
"rewards/margins": 0.09596750140190125,
"rewards/rejected": -0.03464752808213234,
"step": 100
},
{
"epoch": 0.26,
"eval_logits/chosen": -2.2338831424713135,
"eval_logits/rejected": -2.229041814804077,
"eval_logps/chosen": -34.03020095825195,
"eval_logps/rejected": -37.52727127075195,
"eval_loss": 0.6954607963562012,
"eval_rewards/accuracies": 0.510797381401062,
"eval_rewards/chosen": 0.0034842013847082853,
"eval_rewards/margins": 0.012004716321825981,
"eval_rewards/rejected": -0.008520514704287052,
"eval_runtime": 146.115,
"eval_samples_per_second": 2.347,
"eval_steps_per_second": 0.294,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 4.498257201263691e-06,
"logits/chosen": -1.9946181774139404,
"logits/rejected": -1.992236852645874,
"logps/chosen": -33.117286682128906,
"logps/rejected": -34.00868225097656,
"loss": 0.6852,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.10122231394052505,
"rewards/margins": 0.0846698135137558,
"rewards/rejected": 0.016552483662962914,
"step": 110
},
{
"epoch": 0.31,
"learning_rate": 4.353806263777678e-06,
"logits/chosen": -2.006875514984131,
"logits/rejected": -1.9985148906707764,
"logps/chosen": -32.336421966552734,
"logps/rejected": -32.137081146240234,
"loss": 0.675,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.08678452670574188,
"rewards/margins": 0.06724556535482407,
"rewards/rejected": 0.019538963213562965,
"step": 120
},
{
"epoch": 0.34,
"learning_rate": 4.1940827077152755e-06,
"logits/chosen": -2.0351502895355225,
"logits/rejected": -2.02717661857605,
"logps/chosen": -30.30923843383789,
"logps/rejected": -32.08501434326172,
"loss": 0.6398,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.13318516314029694,
"rewards/margins": 0.15878939628601074,
"rewards/rejected": -0.025604233145713806,
"step": 130
},
{
"epoch": 0.36,
"learning_rate": 4.0204024186666215e-06,
"logits/chosen": -1.966170072555542,
"logits/rejected": -1.9764087200164795,
"logps/chosen": -31.215194702148438,
"logps/rejected": -32.55674743652344,
"loss": 0.627,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.16357474029064178,
"rewards/margins": 0.18106886744499207,
"rewards/rejected": -0.017494117841124535,
"step": 140
},
{
"epoch": 0.39,
"learning_rate": 3.834196265035119e-06,
"logits/chosen": -1.8775428533554077,
"logits/rejected": -1.8786998987197876,
"logps/chosen": -33.92055130004883,
"logps/rejected": -34.77721405029297,
"loss": 0.6168,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.21771302819252014,
"rewards/margins": 0.23367898166179657,
"rewards/rejected": -0.015965968370437622,
"step": 150
},
{
"epoch": 0.42,
"learning_rate": 3.636998309800573e-06,
"logits/chosen": -1.9291296005249023,
"logits/rejected": -1.9257177114486694,
"logps/chosen": -36.01557922363281,
"logps/rejected": -32.72490692138672,
"loss": 0.6444,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.12488000094890594,
"rewards/margins": 0.12660440802574158,
"rewards/rejected": -0.0017244067275896668,
"step": 160
},
{
"epoch": 0.44,
"learning_rate": 3.4304331721118078e-06,
"logits/chosen": -2.0295331478118896,
"logits/rejected": -2.0221762657165527,
"logps/chosen": -33.48248291015625,
"logps/rejected": -31.408077239990234,
"loss": 0.5816,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.25230517983436584,
"rewards/margins": 0.2972865700721741,
"rewards/rejected": -0.04498137906193733,
"step": 170
},
{
"epoch": 0.47,
"learning_rate": 3.2162026428305436e-06,
"logits/chosen": -2.036190986633301,
"logits/rejected": -2.0414373874664307,
"logps/chosen": -32.22594451904297,
"logps/rejected": -32.46149444580078,
"loss": 0.5993,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.2550733685493469,
"rewards/margins": 0.2361568957567215,
"rewards/rejected": 0.018916476517915726,
"step": 180
},
{
"epoch": 0.49,
"learning_rate": 2.996071664294641e-06,
"logits/chosen": -2.0371222496032715,
"logits/rejected": -2.0343565940856934,
"logps/chosen": -31.28468894958496,
"logps/rejected": -31.336734771728516,
"loss": 0.6307,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.16341081261634827,
"rewards/margins": 0.17822694778442383,
"rewards/rejected": -0.014816122129559517,
"step": 190
},
{
"epoch": 0.52,
"learning_rate": 2.7718537898066833e-06,
"logits/chosen": -1.9067039489746094,
"logits/rejected": -1.91135573387146,
"logps/chosen": -31.312374114990234,
"logps/rejected": -32.82074737548828,
"loss": 0.6015,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.2351258099079132,
"rewards/margins": 0.25897616147994995,
"rewards/rejected": -0.0238503310829401,
"step": 200
},
{
"epoch": 0.52,
"eval_logits/chosen": -2.232191324234009,
"eval_logits/rejected": -2.22735333442688,
"eval_logps/chosen": -34.060691833496094,
"eval_logps/rejected": -37.56568908691406,
"eval_loss": 0.695566713809967,
"eval_rewards/accuracies": 0.52491694688797,
"eval_rewards/chosen": -0.020911961793899536,
"eval_rewards/margins": 0.01834380254149437,
"eval_rewards/rejected": -0.039255764335393906,
"eval_runtime": 145.8849,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 200
},
{
"epoch": 0.55,
"learning_rate": 2.5453962426402006e-06,
"logits/chosen": -2.0197176933288574,
"logits/rejected": -2.0303761959075928,
"logps/chosen": -31.751026153564453,
"logps/rejected": -33.96234893798828,
"loss": 0.5969,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.17622438073158264,
"rewards/margins": 0.25594404339790344,
"rewards/rejected": -0.0797196701169014,
"step": 210
},
{
"epoch": 0.57,
"learning_rate": 2.3185646976551794e-06,
"logits/chosen": -1.9122480154037476,
"logits/rejected": -1.9270412921905518,
"logps/chosen": -29.86123275756836,
"logps/rejected": -31.612594604492188,
"loss": 0.6025,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": 0.20024582743644714,
"rewards/margins": 0.24377915263175964,
"rewards/rejected": -0.04353334754705429,
"step": 220
},
{
"epoch": 0.6,
"learning_rate": 2.0932279108998323e-06,
"logits/chosen": -1.9689687490463257,
"logits/rejected": -1.9729585647583008,
"logps/chosen": -33.11440658569336,
"logps/rejected": -31.650421142578125,
"loss": 0.5809,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.2424498349428177,
"rewards/margins": 0.3233444094657898,
"rewards/rejected": -0.08089461922645569,
"step": 230
},
{
"epoch": 0.62,
"learning_rate": 1.8712423238279358e-06,
"logits/chosen": -1.9670231342315674,
"logits/rejected": -1.9451711177825928,
"logps/chosen": -33.830162048339844,
"logps/rejected": -35.1173095703125,
"loss": 0.558,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": 0.2588713765144348,
"rewards/margins": 0.3920826315879822,
"rewards/rejected": -0.13321125507354736,
"step": 240
},
{
"epoch": 0.65,
"learning_rate": 1.6544367689701824e-06,
"logits/chosen": -2.0084152221679688,
"logits/rejected": -2.005080223083496,
"logps/chosen": -32.70518493652344,
"logps/rejected": -36.280517578125,
"loss": 0.6104,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.1759072244167328,
"rewards/margins": 0.22038432955741882,
"rewards/rejected": -0.044477105140686035,
"step": 250
},
{
"epoch": 0.68,
"learning_rate": 1.4445974030621963e-06,
"logits/chosen": -1.8755052089691162,
"logits/rejected": -1.8730967044830322,
"logps/chosen": -33.984092712402344,
"logps/rejected": -35.538455963134766,
"loss": 0.622,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.16344432532787323,
"rewards/margins": 0.19051328301429749,
"rewards/rejected": -0.02706894651055336,
"step": 260
},
{
"epoch": 0.7,
"learning_rate": 1.243452991757889e-06,
"logits/chosen": -1.8608484268188477,
"logits/rejected": -1.8584181070327759,
"logps/chosen": -34.17797088623047,
"logps/rejected": -31.830347061157227,
"loss": 0.6124,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.1757020801305771,
"rewards/margins": 0.22291450202465057,
"rewards/rejected": -0.047212425619363785,
"step": 270
},
{
"epoch": 0.73,
"learning_rate": 1.0526606671603523e-06,
"logits/chosen": -1.9641139507293701,
"logits/rejected": -1.9535942077636719,
"logps/chosen": -35.01939010620117,
"logps/rejected": -31.871440887451172,
"loss": 0.5852,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.26980119943618774,
"rewards/margins": 0.29416200518608093,
"rewards/rejected": -0.024360809475183487,
"step": 280
},
{
"epoch": 0.75,
"learning_rate": 8.737922755071455e-07,
"logits/chosen": -2.0593204498291016,
"logits/rejected": -2.0443997383117676,
"logps/chosen": -30.722980499267578,
"logps/rejected": -32.61235809326172,
"loss": 0.6599,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.16091887652873993,
"rewards/margins": 0.12933868169784546,
"rewards/rejected": 0.031580209732055664,
"step": 290
},
{
"epoch": 0.78,
"learning_rate": 7.08321427484816e-07,
"logits/chosen": -1.930450201034546,
"logits/rejected": -1.9279005527496338,
"logps/chosen": -32.415870666503906,
"logps/rejected": -30.882410049438477,
"loss": 0.5385,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": 0.4089924395084381,
"rewards/margins": 0.4614754319190979,
"rewards/rejected": -0.05248301103711128,
"step": 300
},
{
"epoch": 0.78,
"eval_logits/chosen": -2.229433536529541,
"eval_logits/rejected": -2.22458553314209,
"eval_logps/chosen": -34.090904235839844,
"eval_logps/rejected": -37.59726333618164,
"eval_loss": 0.69569993019104,
"eval_rewards/accuracies": 0.5398671627044678,
"eval_rewards/chosen": -0.04508008435368538,
"eval_rewards/margins": 0.019436603412032127,
"eval_rewards/rejected": -0.06451668590307236,
"eval_runtime": 145.8403,
"eval_samples_per_second": 2.352,
"eval_steps_per_second": 0.295,
"step": 300
},
{
"epoch": 0.81,
"learning_rate": 4.84533120650964e-06,
"logits/chosen": -1.9146509170532227,
"logits/rejected": -1.9113744497299194,
"logps/chosen": -31.32379150390625,
"logps/rejected": -33.805755615234375,
"loss": 0.5978,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.2236466407775879,
"rewards/margins": 0.27452850341796875,
"rewards/rejected": -0.050881873816251755,
"step": 310
},
{
"epoch": 0.83,
"learning_rate": 4.825108134172131e-06,
"logits/chosen": -1.9630565643310547,
"logits/rejected": -1.9508411884307861,
"logps/chosen": -34.32460403442383,
"logps/rejected": -33.67837905883789,
"loss": 0.5713,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.217172771692276,
"rewards/margins": 0.3288002014160156,
"rewards/rejected": -0.1116274744272232,
"step": 320
},
{
"epoch": 0.86,
"learning_rate": 4.80369052967602e-06,
"logits/chosen": -1.9971084594726562,
"logits/rejected": -1.995650053024292,
"logps/chosen": -33.14687728881836,
"logps/rejected": -32.54724884033203,
"loss": 0.5766,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.2711312174797058,
"rewards/margins": 0.32526224851608276,
"rewards/rejected": -0.05413101240992546,
"step": 330
},
{
"epoch": 0.88,
"learning_rate": 4.781089396387968e-06,
"logits/chosen": -2.0822432041168213,
"logits/rejected": -2.066549301147461,
"logps/chosen": -33.73598098754883,
"logps/rejected": -33.096824645996094,
"loss": 0.572,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.35473155975341797,
"rewards/margins": 0.33776286244392395,
"rewards/rejected": 0.016968706622719765,
"step": 340
},
{
"epoch": 0.91,
"learning_rate": 4.757316345716554e-06,
"logits/chosen": -1.9550272226333618,
"logits/rejected": -1.9541677236557007,
"logps/chosen": -32.79063415527344,
"logps/rejected": -32.5118293762207,
"loss": 0.5561,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.3931701183319092,
"rewards/margins": 0.43051546812057495,
"rewards/rejected": -0.03734535723924637,
"step": 350
},
{
"epoch": 0.94,
"learning_rate": 4.73238359114687e-06,
"logits/chosen": -1.9071505069732666,
"logits/rejected": -1.917394995689392,
"logps/chosen": -31.765522003173828,
"logps/rejected": -35.38612365722656,
"loss": 0.5497,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.35133686661720276,
"rewards/margins": 0.4292295575141907,
"rewards/rejected": -0.07789268344640732,
"step": 360
},
{
"epoch": 0.96,
"learning_rate": 4.706303941965804e-06,
"logits/chosen": -2.042468547821045,
"logits/rejected": -2.0360734462738037,
"logps/chosen": -33.295223236083984,
"logps/rejected": -29.314971923828125,
"loss": 0.5615,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.28539514541625977,
"rewards/margins": 0.35623133182525635,
"rewards/rejected": -0.07083616405725479,
"step": 370
},
{
"epoch": 0.99,
"learning_rate": 4.679090796681225e-06,
"logits/chosen": -1.9002101421356201,
"logits/rejected": -1.9024240970611572,
"logps/chosen": -33.61799240112305,
"logps/rejected": -30.948089599609375,
"loss": 0.5044,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": 0.49684303998947144,
"rewards/margins": 0.5572139024734497,
"rewards/rejected": -0.06037076562643051,
"step": 380
},
{
"epoch": 1.01,
"learning_rate": 4.650758136138454e-06,
"logits/chosen": -1.9271265268325806,
"logits/rejected": -1.9258983135223389,
"logps/chosen": -33.7630500793457,
"logps/rejected": -35.974002838134766,
"loss": 0.4531,
"rewards/accuracies": 0.7958333492279053,
"rewards/chosen": 0.4973909258842468,
"rewards/margins": 0.7517890930175781,
"rewards/rejected": -0.2543981671333313,
"step": 390
},
{
"epoch": 1.04,
"learning_rate": 4.621320516337559e-06,
"logits/chosen": -1.8622478246688843,
"logits/rejected": -1.8538604974746704,
"logps/chosen": -31.049022674560547,
"logps/rejected": -36.37419509887695,
"loss": 0.4017,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 0.5799586772918701,
"rewards/margins": 0.9268311262130737,
"rewards/rejected": -0.34687256813049316,
"step": 400
},
{
"epoch": 1.04,
"eval_logits/chosen": -2.2100114822387695,
"eval_logits/rejected": -2.205153226852417,
"eval_logps/chosen": -34.06473159790039,
"eval_logps/rejected": -37.634727478027344,
"eval_loss": 0.6775211691856384,
"eval_rewards/accuracies": 0.6004983186721802,
"eval_rewards/chosen": -0.024144915863871574,
"eval_rewards/margins": 0.07034175843000412,
"eval_rewards/rejected": -0.09448666870594025,
"eval_runtime": 146.4012,
"eval_samples_per_second": 2.343,
"eval_steps_per_second": 0.294,
"step": 400
},
{
"epoch": 1.06,
"learning_rate": 4.590793060955158e-06,
"logits/chosen": -2.0343263149261475,
"logits/rejected": -2.037248134613037,
"logps/chosen": -32.27507019042969,
"logps/rejected": -35.283382415771484,
"loss": 0.3721,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.6230968236923218,
"rewards/margins": 1.0155293941497803,
"rewards/rejected": -0.3924326002597809,
"step": 410
},
{
"epoch": 1.09,
"learning_rate": 4.559191453574582e-06,
"logits/chosen": -1.8720476627349854,
"logits/rejected": -1.8706938028335571,
"logps/chosen": -28.4278507232666,
"logps/rejected": -32.78199005126953,
"loss": 0.3947,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.5117626786231995,
"rewards/margins": 0.8832231760025024,
"rewards/rejected": -0.371460497379303,
"step": 420
},
{
"epoch": 1.12,
"learning_rate": 4.52653192962838e-06,
"logits/chosen": -1.8284943103790283,
"logits/rejected": -1.821554183959961,
"logps/chosen": -33.15713119506836,
"logps/rejected": -34.481746673583984,
"loss": 0.3744,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 0.733561635017395,
"rewards/margins": 0.9227407574653625,
"rewards/rejected": -0.18917915225028992,
"step": 430
},
{
"epoch": 1.14,
"learning_rate": 4.492831268057307e-06,
"logits/chosen": -1.998443841934204,
"logits/rejected": -1.993316888809204,
"logps/chosen": -30.946176528930664,
"logps/rejected": -32.4330940246582,
"loss": 0.3658,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.669696033000946,
"rewards/margins": 1.0420799255371094,
"rewards/rejected": -0.3723840117454529,
"step": 440
},
{
"epoch": 1.17,
"learning_rate": 4.458106782690094e-06,
"logits/chosen": -1.8814995288848877,
"logits/rejected": -1.8857187032699585,
"logps/chosen": -33.60875701904297,
"logps/rejected": -33.16690444946289,
"loss": 0.3134,
"rewards/accuracies": 0.9375,
"rewards/chosen": 0.7256715297698975,
"rewards/margins": 1.1882824897766113,
"rewards/rejected": -0.4626109004020691,
"step": 450
},
{
"epoch": 1.19,
"learning_rate": 4.422376313348405e-06,
"logits/chosen": -1.8870223760604858,
"logits/rejected": -1.8813492059707642,
"logps/chosen": -34.445613861083984,
"logps/rejected": -35.65070343017578,
"loss": 0.3011,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 0.779128909111023,
"rewards/margins": 1.3404181003570557,
"rewards/rejected": -0.5612891316413879,
"step": 460
},
{
"epoch": 1.22,
"learning_rate": 4.3856582166815696e-06,
"logits/chosen": -1.9110435247421265,
"logits/rejected": -1.910775899887085,
"logps/chosen": -33.23371887207031,
"logps/rejected": -34.685794830322266,
"loss": 0.3634,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.747688353061676,
"rewards/margins": 1.1479377746582031,
"rewards/rejected": -0.40024933218955994,
"step": 470
},
{
"epoch": 1.25,
"learning_rate": 4.347971356735789e-06,
"logits/chosen": -1.9573456048965454,
"logits/rejected": -1.938490867614746,
"logps/chosen": -33.21842575073242,
"logps/rejected": -33.71406555175781,
"loss": 0.3363,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.753872275352478,
"rewards/margins": 1.2691444158554077,
"rewards/rejected": -0.5152722001075745,
"step": 480
},
{
"epoch": 1.27,
"learning_rate": 4.309335095262675e-06,
"logits/chosen": -1.9228092432022095,
"logits/rejected": -1.9221388101577759,
"logps/chosen": -30.777042388916016,
"logps/rejected": -31.67429542541504,
"loss": 0.3684,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.6674107313156128,
"rewards/margins": 1.0100107192993164,
"rewards/rejected": -0.34259992837905884,
"step": 490
},
{
"epoch": 1.3,
"learning_rate": 4.269769281772082e-06,
"logits/chosen": -1.8837568759918213,
"logits/rejected": -1.8767935037612915,
"logps/chosen": -31.731576919555664,
"logps/rejected": -35.3281364440918,
"loss": 0.3216,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.7508381605148315,
"rewards/margins": 1.3119536638259888,
"rewards/rejected": -0.5611153841018677,
"step": 500
},
{
"epoch": 1.3,
"eval_logits/chosen": -2.170705556869507,
"eval_logits/rejected": -2.165867567062378,
"eval_logps/chosen": -34.18019104003906,
"eval_logps/rejected": -37.78302764892578,
"eval_loss": 0.6819708347320557,
"eval_rewards/accuracies": 0.5768272280693054,
"eval_rewards/chosen": -0.11651215702295303,
"eval_rewards/margins": 0.09661299735307693,
"eval_rewards/rejected": -0.21312516927719116,
"eval_runtime": 145.937,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 500
},
{
"epoch": 1.32,
"learning_rate": 4.22929424333435e-06,
"logits/chosen": -1.8785709142684937,
"logits/rejected": -1.8822866678237915,
"logps/chosen": -28.374109268188477,
"logps/rejected": -33.53699493408203,
"loss": 0.3449,
"rewards/accuracies": 0.875,
"rewards/chosen": 0.6405723094940186,
"rewards/margins": 1.1827360391616821,
"rewards/rejected": -0.5421636700630188,
"step": 510
},
{
"epoch": 1.35,
"learning_rate": 4.1879307741372085e-06,
"logits/chosen": -1.8827375173568726,
"logits/rejected": -1.89346444606781,
"logps/chosen": -32.36341857910156,
"logps/rejected": -31.319293975830078,
"loss": 0.3592,
"rewards/accuracies": 0.875,
"rewards/chosen": 0.7191308736801147,
"rewards/margins": 1.2532163858413696,
"rewards/rejected": -0.5340853929519653,
"step": 520
},
{
"epoch": 1.38,
"learning_rate": 4.145700124802693e-06,
"logits/chosen": -1.829416036605835,
"logits/rejected": -1.8270241022109985,
"logps/chosen": -30.80521011352539,
"logps/rejected": -30.752178192138672,
"loss": 0.3535,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.6894973516464233,
"rewards/margins": 1.1498301029205322,
"rewards/rejected": -0.46033263206481934,
"step": 530
},
{
"epoch": 1.4,
"learning_rate": 4.102623991469562e-06,
"logits/chosen": -1.9103914499282837,
"logits/rejected": -1.9033727645874023,
"logps/chosen": -33.29057312011719,
"logps/rejected": -33.72592544555664,
"loss": 0.33,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 0.7538042664527893,
"rewards/margins": 1.3205549716949463,
"rewards/rejected": -0.5667508244514465,
"step": 540
},
{
"epoch": 1.43,
"learning_rate": 4.058724504646834e-06,
"logits/chosen": -1.8793199062347412,
"logits/rejected": -1.8857247829437256,
"logps/chosen": -30.955270767211914,
"logps/rejected": -33.297035217285156,
"loss": 0.3587,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.6383341550827026,
"rewards/margins": 1.0689740180969238,
"rewards/rejected": -0.43063968420028687,
"step": 550
},
{
"epoch": 1.45,
"learning_rate": 4.014024217844167e-06,
"logits/chosen": -1.949784278869629,
"logits/rejected": -1.9266754388809204,
"logps/chosen": -30.55877685546875,
"logps/rejected": -33.56010818481445,
"loss": 0.3523,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.7030326128005981,
"rewards/margins": 1.1542710065841675,
"rewards/rejected": -0.45123839378356934,
"step": 560
},
{
"epoch": 1.48,
"learning_rate": 3.968546095984911e-06,
"logits/chosen": -1.883561134338379,
"logits/rejected": -1.8786439895629883,
"logps/chosen": -31.494693756103516,
"logps/rejected": -32.63154220581055,
"loss": 0.3697,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": 0.7376305460929871,
"rewards/margins": 1.1461317539215088,
"rewards/rejected": -0.4085012972354889,
"step": 570
},
{
"epoch": 1.51,
"learning_rate": 3.922313503607806e-06,
"logits/chosen": -1.9235492944717407,
"logits/rejected": -1.9254448413848877,
"logps/chosen": -33.56511688232422,
"logps/rejected": -35.665016174316406,
"loss": 0.3133,
"rewards/accuracies": 0.9375,
"rewards/chosen": 0.6485405564308167,
"rewards/margins": 1.339817762374878,
"rewards/rejected": -0.6912771463394165,
"step": 580
},
{
"epoch": 1.53,
"learning_rate": 3.875350192863368e-06,
"logits/chosen": -1.9060910940170288,
"logits/rejected": -1.9054912328720093,
"logps/chosen": -29.636911392211914,
"logps/rejected": -32.14686965942383,
"loss": 0.332,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.7641160488128662,
"rewards/margins": 1.312302827835083,
"rewards/rejected": -0.5481868386268616,
"step": 590
},
{
"epoch": 1.56,
"learning_rate": 3.8276802913111436e-06,
"logits/chosen": -1.919725775718689,
"logits/rejected": -1.9174749851226807,
"logps/chosen": -31.991186141967773,
"logps/rejected": -32.90464782714844,
"loss": 0.336,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.8056806325912476,
"rewards/margins": 1.2894303798675537,
"rewards/rejected": -0.4837496280670166,
"step": 600
},
{
"epoch": 1.56,
"eval_logits/chosen": -2.160193920135498,
"eval_logits/rejected": -2.1554007530212402,
"eval_logps/chosen": -34.264404296875,
"eval_logps/rejected": -37.94013214111328,
"eval_loss": 0.6617584228515625,
"eval_rewards/accuracies": 0.6241694688796997,
"eval_rewards/chosen": -0.18387840688228607,
"eval_rewards/margins": 0.15492965281009674,
"eval_rewards/rejected": -0.3388080298900604,
"eval_runtime": 145.9206,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 600
},
{
"epoch": 1.58,
"learning_rate": 3.7793282895240927e-06,
"logits/chosen": -1.9637705087661743,
"logits/rejected": -1.9703562259674072,
"logps/chosen": -31.264089584350586,
"logps/rejected": -32.738319396972656,
"loss": 0.3311,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.7786344289779663,
"rewards/margins": 1.3187401294708252,
"rewards/rejected": -0.5401058197021484,
"step": 610
},
{
"epoch": 1.61,
"learning_rate": 3.730319028506478e-06,
"logits/chosen": -1.9196746349334717,
"logits/rejected": -1.9172741174697876,
"logps/chosen": -33.669281005859375,
"logps/rejected": -31.596511840820312,
"loss": 0.3157,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 0.869280993938446,
"rewards/margins": 1.3580834865570068,
"rewards/rejected": -0.48880258202552795,
"step": 620
},
{
"epoch": 1.64,
"learning_rate": 3.6806776869317074e-06,
"logits/chosen": -1.8648065328598022,
"logits/rejected": -1.8580557107925415,
"logps/chosen": -34.281700134277344,
"logps/rejected": -33.09290313720703,
"loss": 0.3058,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.9460186958312988,
"rewards/margins": 1.5345759391784668,
"rewards/rejected": -0.5885573029518127,
"step": 630
},
{
"epoch": 1.66,
"learning_rate": 3.6304297682067146e-06,
"logits/chosen": -1.8759320974349976,
"logits/rejected": -1.8822145462036133,
"logps/chosen": -32.987449645996094,
"logps/rejected": -33.8790283203125,
"loss": 0.3408,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.7934818863868713,
"rewards/margins": 1.3044978380203247,
"rewards/rejected": -0.5110160708427429,
"step": 640
},
{
"epoch": 1.69,
"learning_rate": 3.579601087369492e-06,
"logits/chosen": -1.9594967365264893,
"logits/rejected": -1.973659873008728,
"logps/chosen": -31.017492294311523,
"logps/rejected": -32.631839752197266,
"loss": 0.3573,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 0.6679788827896118,
"rewards/margins": 1.1753003597259521,
"rewards/rejected": -0.5073216557502747,
"step": 650
},
{
"epoch": 1.71,
"learning_rate": 3.5282177578265295e-06,
"logits/chosen": -1.8336044549942017,
"logits/rejected": -1.8305423259735107,
"logps/chosen": -32.8193473815918,
"logps/rejected": -35.608089447021484,
"loss": 0.2779,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.8828850984573364,
"rewards/margins": 1.6091547012329102,
"rewards/rejected": -0.726269543170929,
"step": 660
},
{
"epoch": 1.74,
"learning_rate": 3.476306177936961e-06,
"logits/chosen": -1.9243332147598267,
"logits/rejected": -1.9244455099105835,
"logps/chosen": -30.38960838317871,
"logps/rejected": -34.67184066772461,
"loss": 0.3114,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.7167855501174927,
"rewards/margins": 1.3931505680084229,
"rewards/rejected": -0.676365077495575,
"step": 670
},
{
"epoch": 1.77,
"learning_rate": 3.423893017450324e-06,
"logits/chosen": -1.8737255334854126,
"logits/rejected": -1.8705682754516602,
"logps/chosen": -29.965240478515625,
"logps/rejected": -33.58513259887695,
"loss": 0.3412,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 0.7256845235824585,
"rewards/margins": 1.339548945426941,
"rewards/rejected": -0.613864541053772,
"step": 680
},
{
"epoch": 1.79,
"learning_rate": 3.3710052038048794e-06,
"logits/chosen": -1.908151626586914,
"logits/rejected": -1.908117651939392,
"logps/chosen": -28.949966430664062,
"logps/rejected": -31.339040756225586,
"loss": 0.3023,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 0.9705016016960144,
"rewards/margins": 1.5053366422653198,
"rewards/rejected": -0.5348349809646606,
"step": 690
},
{
"epoch": 1.82,
"learning_rate": 3.3176699082935546e-06,
"logits/chosen": -1.8213777542114258,
"logits/rejected": -1.824627161026001,
"logps/chosen": -33.35689163208008,
"logps/rejected": -32.15576934814453,
"loss": 0.3559,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 0.8972657322883606,
"rewards/margins": 1.4501286745071411,
"rewards/rejected": -0.5528628826141357,
"step": 700
},
{
"epoch": 1.82,
"eval_logits/chosen": -2.158259153366089,
"eval_logits/rejected": -2.1534817218780518,
"eval_logps/chosen": -34.35598373413086,
"eval_logps/rejected": -37.98072814941406,
"eval_loss": 0.6946965456008911,
"eval_rewards/accuracies": 0.5340532064437866,
"eval_rewards/chosen": -0.2571484446525574,
"eval_rewards/margins": 0.1141389012336731,
"eval_rewards/rejected": -0.37128737568855286,
"eval_runtime": 145.9197,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 700
},
{
"epoch": 1.84,
"learning_rate": 3.2639145321045933e-06,
"logits/chosen": -1.9044933319091797,
"logits/rejected": -1.8955074548721313,
"logps/chosen": -35.51199722290039,
"logps/rejected": -32.60811233520508,
"loss": 0.3093,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 0.8000162839889526,
"rewards/margins": 1.3618541955947876,
"rewards/rejected": -0.5618377923965454,
"step": 710
},
{
"epoch": 1.87,
"learning_rate": 3.2097666922441107e-06,
"logits/chosen": -1.9182393550872803,
"logits/rejected": -1.9193944931030273,
"logps/chosen": -35.33127212524414,
"logps/rejected": -33.76591110229492,
"loss": 0.3149,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.8832913637161255,
"rewards/margins": 1.435203194618225,
"rewards/rejected": -0.5519118309020996,
"step": 720
},
{
"epoch": 1.9,
"learning_rate": 3.1552542073477554e-06,
"logits/chosen": -1.9362226724624634,
"logits/rejected": -1.9337679147720337,
"logps/chosen": -31.4454345703125,
"logps/rejected": -33.67931365966797,
"loss": 0.3081,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.8943157196044922,
"rewards/margins": 1.4402475357055664,
"rewards/rejected": -0.5459317564964294,
"step": 730
},
{
"epoch": 1.92,
"learning_rate": 3.100405083388799e-06,
"logits/chosen": -1.9176380634307861,
"logits/rejected": -1.922844648361206,
"logps/chosen": -30.585041046142578,
"logps/rejected": -33.932350158691406,
"loss": 0.2741,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 0.9605496525764465,
"rewards/margins": 1.5825778245925903,
"rewards/rejected": -0.6220281720161438,
"step": 740
},
{
"epoch": 1.95,
"learning_rate": 3.0452474992899645e-06,
"logits/chosen": -1.864728569984436,
"logits/rejected": -1.8635470867156982,
"logps/chosen": -32.103065490722656,
"logps/rejected": -35.512298583984375,
"loss": 0.3075,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 0.8175305128097534,
"rewards/margins": 1.478601098060608,
"rewards/rejected": -0.6610704660415649,
"step": 750
},
{
"epoch": 1.97,
"learning_rate": 2.989809792446417e-06,
"logits/chosen": -1.7420361042022705,
"logits/rejected": -1.7370550632476807,
"logps/chosen": -34.88098907470703,
"logps/rejected": -35.97365951538086,
"loss": 0.3032,
"rewards/accuracies": 0.875,
"rewards/chosen": 0.8939619064331055,
"rewards/margins": 1.506162166595459,
"rewards/rejected": -0.6122000813484192,
"step": 760
},
{
"epoch": 2.0,
"learning_rate": 2.9341204441673267e-06,
"logits/chosen": -1.8814598321914673,
"logits/rejected": -1.8852901458740234,
"logps/chosen": -34.113304138183594,
"logps/rejected": -34.245601654052734,
"loss": 0.3271,
"rewards/accuracies": 0.85833340883255,
"rewards/chosen": 0.8802832365036011,
"rewards/margins": 1.4008936882019043,
"rewards/rejected": -0.5206103324890137,
"step": 770
},
{
"epoch": 2.03,
"learning_rate": 2.878208065043501e-06,
"logits/chosen": -1.8257869482040405,
"logits/rejected": -1.8239891529083252,
"logps/chosen": -32.38594436645508,
"logps/rejected": -35.74359893798828,
"loss": 0.154,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.2125210762023926,
"rewards/margins": 2.2836754322052,
"rewards/rejected": -1.071154236793518,
"step": 780
},
{
"epoch": 2.05,
"learning_rate": 2.8221013802485974e-06,
"logits/chosen": -1.8724597692489624,
"logits/rejected": -1.8708446025848389,
"logps/chosen": -31.859329223632812,
"logps/rejected": -34.107154846191406,
"loss": 0.1817,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.2131247520446777,
"rewards/margins": 2.029592752456665,
"rewards/rejected": -0.8164681196212769,
"step": 790
},
{
"epoch": 2.08,
"learning_rate": 2.76582921478147e-06,
"logits/chosen": -1.8005281686782837,
"logits/rejected": -1.7946115732192993,
"logps/chosen": -32.934303283691406,
"logps/rejected": -32.45271301269531,
"loss": 0.1978,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.1590179204940796,
"rewards/margins": 1.9497814178466797,
"rewards/rejected": -0.7907635569572449,
"step": 800
},
{
"epoch": 2.08,
"eval_logits/chosen": -2.154881000518799,
"eval_logits/rejected": -2.1501035690307617,
"eval_logps/chosen": -34.32504653930664,
"eval_logps/rejected": -37.97528076171875,
"eval_loss": 0.6837867498397827,
"eval_rewards/accuracies": 0.5834717750549316,
"eval_rewards/chosen": -0.2323925942182541,
"eval_rewards/margins": 0.1345335692167282,
"eval_rewards/rejected": -0.3669261634349823,
"eval_runtime": 145.9433,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 800
},
{
"epoch": 2.1,
"learning_rate": 2.7094204786572254e-06,
"logits/chosen": -1.898033857345581,
"logits/rejected": -1.905503511428833,
"logps/chosen": -30.634164810180664,
"logps/rejected": -35.38495635986328,
"loss": 0.189,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.1952745914459229,
"rewards/margins": 2.0985639095306396,
"rewards/rejected": -0.9032894372940063,
"step": 810
},
{
"epoch": 2.13,
"learning_rate": 2.6529041520546072e-06,
"logits/chosen": -1.872579574584961,
"logits/rejected": -1.875286340713501,
"logps/chosen": -31.190942764282227,
"logps/rejected": -34.01335906982422,
"loss": 0.2433,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.104109525680542,
"rewards/margins": 1.7309566736221313,
"rewards/rejected": -0.6268469095230103,
"step": 820
},
{
"epoch": 2.16,
"learning_rate": 2.5963092704273302e-06,
"logits/chosen": -1.7723095417022705,
"logits/rejected": -1.7764465808868408,
"logps/chosen": -31.214824676513672,
"logps/rejected": -36.142818450927734,
"loss": 0.1808,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.1013058423995972,
"rewards/margins": 2.0874154567718506,
"rewards/rejected": -0.9861096143722534,
"step": 830
},
{
"epoch": 2.18,
"learning_rate": 2.53966490958702e-06,
"logits/chosen": -1.8381569385528564,
"logits/rejected": -1.8343050479888916,
"logps/chosen": -31.591388702392578,
"logps/rejected": -33.85132598876953,
"loss": 0.2508,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.1018110513687134,
"rewards/margins": 1.7871520519256592,
"rewards/rejected": -0.6853411793708801,
"step": 840
},
{
"epoch": 2.21,
"learning_rate": 2.4830001707654135e-06,
"logits/chosen": -1.9189765453338623,
"logits/rejected": -1.9210529327392578,
"logps/chosen": -31.255529403686523,
"logps/rejected": -36.854835510253906,
"loss": 0.1574,
"rewards/accuracies": 1.0,
"rewards/chosen": 1.2158453464508057,
"rewards/margins": 2.1701653003692627,
"rewards/rejected": -0.9543198347091675,
"step": 850
},
{
"epoch": 2.23,
"learning_rate": 2.4263441656635054e-06,
"logits/chosen": -1.7254867553710938,
"logits/rejected": -1.7197097539901733,
"logps/chosen": -34.86113357543945,
"logps/rejected": -33.97612762451172,
"loss": 0.2044,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.1364357471466064,
"rewards/margins": 2.003814697265625,
"rewards/rejected": -0.8673788905143738,
"step": 860
},
{
"epoch": 2.26,
"learning_rate": 2.3697260014953107e-06,
"logits/chosen": -1.7829675674438477,
"logits/rejected": -1.7830712795257568,
"logps/chosen": -34.304718017578125,
"logps/rejected": -35.99662780761719,
"loss": 0.181,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.2304695844650269,
"rewards/margins": 2.150757312774658,
"rewards/rejected": -0.9202877283096313,
"step": 870
},
{
"epoch": 2.29,
"learning_rate": 2.3131747660339396e-06,
"logits/chosen": -1.8317804336547852,
"logits/rejected": -1.8203016519546509,
"logps/chosen": -32.700950622558594,
"logps/rejected": -34.08301544189453,
"loss": 0.1692,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.1245036125183105,
"rewards/margins": 2.0532915592193604,
"rewards/rejected": -0.9287878274917603,
"step": 880
},
{
"epoch": 2.31,
"learning_rate": 2.256719512667651e-06,
"logits/chosen": -1.9331659078598022,
"logits/rejected": -1.9377342462539673,
"logps/chosen": -31.932796478271484,
"logps/rejected": -33.6646728515625,
"loss": 0.1905,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1177093982696533,
"rewards/margins": 2.100264310836792,
"rewards/rejected": -0.9825549125671387,
"step": 890
},
{
"epoch": 2.34,
"learning_rate": 2.2003892454735786e-06,
"logits/chosen": -1.8520333766937256,
"logits/rejected": -1.844834327697754,
"logps/chosen": -33.29985427856445,
"logps/rejected": -33.635623931884766,
"loss": 0.1619,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.243377923965454,
"rewards/margins": 2.2301688194274902,
"rewards/rejected": -0.9867912530899048,
"step": 900
},
{
"epoch": 2.34,
"eval_logits/chosen": -2.1431386470794678,
"eval_logits/rejected": -2.138375997543335,
"eval_logps/chosen": -34.34238052368164,
"eval_logps/rejected": -38.03608703613281,
"eval_loss": 0.6788281202316284,
"eval_rewards/accuracies": 0.5859634280204773,
"eval_rewards/chosen": -0.24626301229000092,
"eval_rewards/margins": 0.16930998861789703,
"eval_rewards/rejected": -0.41557300090789795,
"eval_runtime": 145.9104,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 900
},
{
"epoch": 2.36,
"learning_rate": 2.1442129043167877e-06,
"logits/chosen": -1.8457978963851929,
"logits/rejected": -1.8462024927139282,
"logps/chosen": -29.82485008239746,
"logps/rejected": -36.06700897216797,
"loss": 0.1786,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.0738495588302612,
"rewards/margins": 2.1757030487060547,
"rewards/rejected": -1.1018531322479248,
"step": 910
},
{
"epoch": 2.39,
"learning_rate": 2.088219349982323e-06,
"logits/chosen": -1.8021042346954346,
"logits/rejected": -1.7940893173217773,
"logps/chosen": -30.521142959594727,
"logps/rejected": -34.798255920410156,
"loss": 0.1883,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.0542433261871338,
"rewards/margins": 2.0357508659362793,
"rewards/rejected": -0.9815075993537903,
"step": 920
},
{
"epoch": 2.42,
"learning_rate": 2.0324373493478803e-06,
"logits/chosen": -1.9680095911026,
"logits/rejected": -1.968182921409607,
"logps/chosen": -28.589441299438477,
"logps/rejected": -33.940185546875,
"loss": 0.2295,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.1260759830474854,
"rewards/margins": 1.9863367080688477,
"rewards/rejected": -0.8602607846260071,
"step": 930
},
{
"epoch": 2.44,
"learning_rate": 1.976895560604729e-06,
"logits/chosen": -1.847324013710022,
"logits/rejected": -1.857102394104004,
"logps/chosen": -32.901668548583984,
"logps/rejected": -34.3242073059082,
"loss": 0.1675,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.2187172174453735,
"rewards/margins": 2.2863640785217285,
"rewards/rejected": -1.0676472187042236,
"step": 940
},
{
"epoch": 2.47,
"learning_rate": 1.921622518534466e-06,
"logits/chosen": -1.8901779651641846,
"logits/rejected": -1.8939307928085327,
"logps/chosen": -29.412405014038086,
"logps/rejected": -32.94640350341797,
"loss": 0.2048,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 0.9882534146308899,
"rewards/margins": 1.9565131664276123,
"rewards/rejected": -0.9682596921920776,
"step": 950
},
{
"epoch": 2.49,
"learning_rate": 1.8666466198491794e-06,
"logits/chosen": -1.8851401805877686,
"logits/rejected": -1.8811264038085938,
"logps/chosen": -32.44501495361328,
"logps/rejected": -34.933692932128906,
"loss": 0.2061,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3041638135910034,
"rewards/margins": 2.126009225845337,
"rewards/rejected": -0.8218452334403992,
"step": 960
},
{
"epoch": 2.52,
"learning_rate": 1.8119961086025376e-06,
"logits/chosen": -1.8027584552764893,
"logits/rejected": -1.805301308631897,
"logps/chosen": -31.277942657470703,
"logps/rejected": -36.393043518066406,
"loss": 0.1663,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.2278459072113037,
"rewards/margins": 2.197005033493042,
"rewards/rejected": -0.9691593050956726,
"step": 970
},
{
"epoch": 2.55,
"learning_rate": 1.7576990616793139e-06,
"logits/chosen": -1.835371971130371,
"logits/rejected": -1.8288919925689697,
"logps/chosen": -34.11837387084961,
"logps/rejected": -37.60203552246094,
"loss": 0.2066,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.0293549299240112,
"rewards/margins": 2.101419687271118,
"rewards/rejected": -1.0720646381378174,
"step": 980
},
{
"epoch": 2.57,
"learning_rate": 1.7037833743707892e-06,
"logits/chosen": -1.813190221786499,
"logits/rejected": -1.807956337928772,
"logps/chosen": -29.670574188232422,
"logps/rejected": -37.203330993652344,
"loss": 0.2099,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.06589937210083,
"rewards/margins": 2.0366604328155518,
"rewards/rejected": -0.970760703086853,
"step": 990
},
{
"epoch": 2.6,
"learning_rate": 1.6502767460434588e-06,
"logits/chosen": -1.7904030084609985,
"logits/rejected": -1.779764175415039,
"logps/chosen": -30.418493270874023,
"logps/rejected": -30.5950984954834,
"loss": 0.209,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.0641578435897827,
"rewards/margins": 1.8604481220245361,
"rewards/rejected": -0.796290397644043,
"step": 1000
},
{
"epoch": 2.6,
"eval_logits/chosen": -2.1356685161590576,
"eval_logits/rejected": -2.130932331085205,
"eval_logps/chosen": -34.380455017089844,
"eval_logps/rejected": -38.083457946777344,
"eval_loss": 0.6776629090309143,
"eval_rewards/accuracies": 0.5917773842811584,
"eval_rewards/chosen": -0.2767248749732971,
"eval_rewards/margins": 0.17674396932125092,
"eval_rewards/rejected": -0.45346885919570923,
"eval_runtime": 145.9185,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 1000
},
{
"epoch": 2.62,
"learning_rate": 1.5972066659083796e-06,
"logits/chosen": -1.896877646446228,
"logits/rejected": -1.896228551864624,
"logps/chosen": -30.421009063720703,
"logps/rejected": -31.305410385131836,
"loss": 0.22,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.1225507259368896,
"rewards/margins": 1.9600980281829834,
"rewards/rejected": -0.8375473022460938,
"step": 1010
},
{
"epoch": 2.65,
"learning_rate": 1.5446003988985041e-06,
"logits/chosen": -1.9368255138397217,
"logits/rejected": -1.9375327825546265,
"logps/chosen": -30.484455108642578,
"logps/rejected": -32.34059524536133,
"loss": 0.177,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.167297601699829,
"rewards/margins": 2.1071255207061768,
"rewards/rejected": -0.9398279190063477,
"step": 1020
},
{
"epoch": 2.68,
"learning_rate": 1.4924849716612211e-06,
"logits/chosen": -1.9036754369735718,
"logits/rejected": -1.9070818424224854,
"logps/chosen": -30.951061248779297,
"logps/rejected": -28.893712997436523,
"loss": 0.227,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.0632069110870361,
"rewards/margins": 1.7912060022354126,
"rewards/rejected": -0.7279989719390869,
"step": 1030
},
{
"epoch": 2.7,
"learning_rate": 1.440887158673332e-06,
"logits/chosen": -1.91388738155365,
"logits/rejected": -1.906005859375,
"logps/chosen": -29.750720977783203,
"logps/rejected": -34.767372131347656,
"loss": 0.1882,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1417419910430908,
"rewards/margins": 2.1554553508758545,
"rewards/rejected": -1.0137132406234741,
"step": 1040
},
{
"epoch": 2.73,
"learning_rate": 1.3898334684855647e-06,
"logits/chosen": -1.8500654697418213,
"logits/rejected": -1.8609867095947266,
"logps/chosen": -32.1203727722168,
"logps/rejected": -33.35521697998047,
"loss": 0.1865,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1461875438690186,
"rewards/margins": 2.034356117248535,
"rewards/rejected": -0.888168454170227,
"step": 1050
},
{
"epoch": 2.75,
"learning_rate": 1.3393501301037245e-06,
"logits/chosen": -1.9280004501342773,
"logits/rejected": -1.9186464548110962,
"logps/chosen": -32.13129425048828,
"logps/rejected": -37.68938446044922,
"loss": 0.2127,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.0955873727798462,
"rewards/margins": 2.181983470916748,
"rewards/rejected": -1.0863959789276123,
"step": 1060
},
{
"epoch": 2.78,
"learning_rate": 1.2894630795134454e-06,
"logits/chosen": -1.835701584815979,
"logits/rejected": -1.837551474571228,
"logps/chosen": -34.348480224609375,
"logps/rejected": -33.834922790527344,
"loss": 0.1815,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3126420974731445,
"rewards/margins": 2.2222840785980225,
"rewards/rejected": -0.9096420407295227,
"step": 1070
},
{
"epoch": 2.81,
"learning_rate": 1.2401979463554984e-06,
"logits/chosen": -1.9658441543579102,
"logits/rejected": -1.9667274951934814,
"logps/chosen": -31.538421630859375,
"logps/rejected": -35.54855728149414,
"loss": 0.1617,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.2153749465942383,
"rewards/margins": 2.3616721630096436,
"rewards/rejected": -1.1462972164154053,
"step": 1080
},
{
"epoch": 2.83,
"learning_rate": 1.1915800407584705e-06,
"logits/chosen": -1.9372714757919312,
"logits/rejected": -1.9413518905639648,
"logps/chosen": -29.44711685180664,
"logps/rejected": -34.69606399536133,
"loss": 0.2085,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.156960129737854,
"rewards/margins": 2.0755460262298584,
"rewards/rejected": -0.9185860753059387,
"step": 1090
},
{
"epoch": 2.86,
"learning_rate": 1.1436343403356019e-06,
"logits/chosen": -1.9265114068984985,
"logits/rejected": -1.9317893981933594,
"logps/chosen": -32.267982482910156,
"logps/rejected": -31.293163299560547,
"loss": 0.2513,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.0408172607421875,
"rewards/margins": 1.7540076971054077,
"rewards/rejected": -0.7131900191307068,
"step": 1100
},
{
"epoch": 2.86,
"eval_logits/chosen": -2.1317365169525146,
"eval_logits/rejected": -2.126985549926758,
"eval_logps/chosen": -34.40774917602539,
"eval_logps/rejected": -38.09048080444336,
"eval_loss": 0.6897340416908264,
"eval_rewards/accuracies": 0.5830564498901367,
"eval_rewards/chosen": -0.2985561192035675,
"eval_rewards/margins": 0.1605302393436432,
"eval_rewards/rejected": -0.4590863585472107,
"eval_runtime": 145.9092,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 1100
},
{
"epoch": 2.88,
"learning_rate": 1.0963854773524548e-06,
"logits/chosen": -1.918264389038086,
"logits/rejected": -1.9183082580566406,
"logps/chosen": -31.37972068786621,
"logps/rejected": -31.99541664123535,
"loss": 0.1933,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.248342514038086,
"rewards/margins": 2.031942844390869,
"rewards/rejected": -0.783600389957428,
"step": 1110
},
{
"epoch": 2.91,
"learning_rate": 1.049857726072005e-06,
"logits/chosen": -1.7625865936279297,
"logits/rejected": -1.7643556594848633,
"logps/chosen": -33.15352249145508,
"logps/rejected": -34.140689849853516,
"loss": 0.2375,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 1.1291639804840088,
"rewards/margins": 2.008704900741577,
"rewards/rejected": -0.8795410394668579,
"step": 1120
},
{
"epoch": 2.94,
"learning_rate": 1.0040749902836508e-06,
"logits/chosen": -1.7994060516357422,
"logits/rejected": -1.7968616485595703,
"logps/chosen": -29.835479736328125,
"logps/rejected": -32.159576416015625,
"loss": 0.2885,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": 0.940991222858429,
"rewards/margins": 1.736061692237854,
"rewards/rejected": -0.7950705289840698,
"step": 1130
},
{
"epoch": 2.96,
"learning_rate": 9.59060791022566e-07,
"logits/chosen": -1.9358510971069336,
"logits/rejected": -1.9303522109985352,
"logps/chosen": -31.38555335998535,
"logps/rejected": -34.0118408203125,
"loss": 0.1779,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3233940601348877,
"rewards/margins": 2.1696605682373047,
"rewards/rejected": -0.8462664484977722,
"step": 1140
},
{
"epoch": 2.99,
"learning_rate": 9.148382544856885e-07,
"logits/chosen": -1.7935917377471924,
"logits/rejected": -1.783705472946167,
"logps/chosen": -32.11836624145508,
"logps/rejected": -32.378910064697266,
"loss": 0.2083,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.066697120666504,
"rewards/margins": 1.9359056949615479,
"rewards/rejected": -0.8692083358764648,
"step": 1150
},
{
"epoch": 3.01,
"learning_rate": 8.714301001505568e-07,
"logits/chosen": -1.8640037775039673,
"logits/rejected": -1.8639949560165405,
"logps/chosen": -32.08525848388672,
"logps/rejected": -32.31430435180664,
"loss": 0.1695,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.2771949768066406,
"rewards/margins": 2.167479991912842,
"rewards/rejected": -0.8902848958969116,
"step": 1160
},
{
"epoch": 3.04,
"learning_rate": 8.288586291031025e-07,
"logits/chosen": -1.9403139352798462,
"logits/rejected": -1.9347970485687256,
"logps/chosen": -32.035804748535156,
"logps/rejected": -34.015647888183594,
"loss": 0.186,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1921296119689941,
"rewards/margins": 2.0516738891601562,
"rewards/rejected": -0.8595443964004517,
"step": 1170
},
{
"epoch": 3.06,
"learning_rate": 7.871457125803897e-07,
"logits/chosen": -1.7985414266586304,
"logits/rejected": -1.806396484375,
"logps/chosen": -31.918773651123047,
"logps/rejected": -33.53784942626953,
"loss": 0.175,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.1313796043395996,
"rewards/margins": 2.0725083351135254,
"rewards/rejected": -0.9411287307739258,
"step": 1180
},
{
"epoch": 3.09,
"learning_rate": 7.463127807341966e-07,
"logits/chosen": -1.86725652217865,
"logits/rejected": -1.8614327907562256,
"logps/chosen": -30.524410247802734,
"logps/rejected": -34.46692657470703,
"loss": 0.1668,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.319364309310913,
"rewards/margins": 2.266146659851074,
"rewards/rejected": -0.9467819929122925,
"step": 1190
},
{
"epoch": 3.12,
"learning_rate": 7.063808116212021e-07,
"logits/chosen": -1.8097419738769531,
"logits/rejected": -1.8118005990982056,
"logps/chosen": -32.011409759521484,
"logps/rejected": -34.247711181640625,
"loss": 0.1713,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.1832258701324463,
"rewards/margins": 2.373357057571411,
"rewards/rejected": -1.1901309490203857,
"step": 1200
},
{
"epoch": 3.12,
"eval_logits/chosen": -2.131728410720825,
"eval_logits/rejected": -2.1269805431365967,
"eval_logps/chosen": -34.38140869140625,
"eval_logps/rejected": -38.09336471557617,
"eval_loss": 0.6780257821083069,
"eval_rewards/accuracies": 0.594684362411499,
"eval_rewards/chosen": -0.27748534083366394,
"eval_rewards/margins": 0.18390974402427673,
"eval_rewards/rejected": -0.4613950550556183,
"eval_runtime": 145.9474,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1200
},
{
"epoch": 3.14,
"learning_rate": 6.673703204254348e-07,
"logits/chosen": -1.7440783977508545,
"logits/rejected": -1.7430063486099243,
"logps/chosen": -34.408180236816406,
"logps/rejected": -34.03881072998047,
"loss": 0.1517,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.3154469728469849,
"rewards/margins": 2.3790054321289062,
"rewards/rejected": -1.0635584592819214,
"step": 1210
},
{
"epoch": 3.17,
"learning_rate": 6.293013489185315e-07,
"logits/chosen": -1.9152580499649048,
"logits/rejected": -1.9091880321502686,
"logps/chosen": -30.193584442138672,
"logps/rejected": -34.31517791748047,
"loss": 0.1516,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.2080823183059692,
"rewards/margins": 2.357390880584717,
"rewards/rejected": -1.149308443069458,
"step": 1220
},
{
"epoch": 3.19,
"learning_rate": 5.921934551632086e-07,
"logits/chosen": -1.7672933340072632,
"logits/rejected": -1.7558486461639404,
"logps/chosen": -32.647369384765625,
"logps/rejected": -34.192893981933594,
"loss": 0.1462,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3906490802764893,
"rewards/margins": 2.429901599884033,
"rewards/rejected": -1.0392528772354126,
"step": 1230
},
{
"epoch": 3.22,
"learning_rate": 5.560657034652405e-07,
"logits/chosen": -1.856805443763733,
"logits/rejected": -1.850097894668579,
"logps/chosen": -29.289770126342773,
"logps/rejected": -30.042938232421875,
"loss": 0.1928,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1283700466156006,
"rewards/margins": 2.0686745643615723,
"rewards/rejected": -0.9403042793273926,
"step": 1240
},
{
"epoch": 3.25,
"learning_rate": 5.2093665457911e-07,
"logits/chosen": -1.8852243423461914,
"logits/rejected": -1.8920814990997314,
"logps/chosen": -33.63641357421875,
"logps/rejected": -32.55451202392578,
"loss": 0.1642,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.4071805477142334,
"rewards/margins": 2.2361385822296143,
"rewards/rejected": -0.8289580345153809,
"step": 1250
},
{
"epoch": 3.27,
"learning_rate": 4.868243561723535e-07,
"logits/chosen": -1.8602968454360962,
"logits/rejected": -1.8607813119888306,
"logps/chosen": -31.587940216064453,
"logps/rejected": -34.207176208496094,
"loss": 0.1465,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.1524927616119385,
"rewards/margins": 2.2663419246673584,
"rewards/rejected": -1.1138492822647095,
"step": 1260
},
{
"epoch": 3.3,
"learning_rate": 4.537463335535161e-07,
"logits/chosen": -1.800450086593628,
"logits/rejected": -1.7998168468475342,
"logps/chosen": -31.380176544189453,
"logps/rejected": -34.821746826171875,
"loss": 0.151,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3685301542282104,
"rewards/margins": 2.4094412326812744,
"rewards/rejected": -1.0409109592437744,
"step": 1270
},
{
"epoch": 3.32,
"learning_rate": 4.217195806684629e-07,
"logits/chosen": -1.686762809753418,
"logits/rejected": -1.6821815967559814,
"logps/chosen": -33.52080535888672,
"logps/rejected": -32.1031494140625,
"loss": 0.1522,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.3733112812042236,
"rewards/margins": 2.314640760421753,
"rewards/rejected": -0.9413297772407532,
"step": 1280
},
{
"epoch": 3.35,
"learning_rate": 3.907605513696808e-07,
"logits/chosen": -1.8902908563613892,
"logits/rejected": -1.8748180866241455,
"logps/chosen": -32.840858459472656,
"logps/rejected": -36.358482360839844,
"loss": 0.1254,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.2053766250610352,
"rewards/margins": 2.4799063205718994,
"rewards/rejected": -1.274530053138733,
"step": 1290
},
{
"epoch": 3.38,
"learning_rate": 3.6088515096305675e-07,
"logits/chosen": -1.83368718624115,
"logits/rejected": -1.8386014699935913,
"logps/chosen": -31.894006729125977,
"logps/rejected": -37.62579345703125,
"loss": 0.1199,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.298893690109253,
"rewards/margins": 2.5169498920440674,
"rewards/rejected": -1.2180559635162354,
"step": 1300
},
{
"epoch": 3.38,
"eval_logits/chosen": -2.131683349609375,
"eval_logits/rejected": -2.1269302368164062,
"eval_logps/chosen": -34.37530517578125,
"eval_logps/rejected": -38.09721374511719,
"eval_loss": 0.6739928126335144,
"eval_rewards/accuracies": 0.5980066657066345,
"eval_rewards/chosen": -0.2726014256477356,
"eval_rewards/margins": 0.19187170267105103,
"eval_rewards/rejected": -0.464473158121109,
"eval_runtime": 145.964,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1300
},
{
"epoch": 3.4,
"learning_rate": 3.321087280364757e-07,
"logits/chosen": -1.8072227239608765,
"logits/rejected": -1.8072770833969116,
"logps/chosen": -34.529273986816406,
"logps/rejected": -38.355064392089844,
"loss": 0.165,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.37179696559906,
"rewards/margins": 2.388556957244873,
"rewards/rejected": -1.0167598724365234,
"step": 1310
},
{
"epoch": 3.43,
"learning_rate": 3.044460665744284e-07,
"logits/chosen": -1.8941234350204468,
"logits/rejected": -1.8927080631256104,
"logps/chosen": -30.5273494720459,
"logps/rejected": -32.267601013183594,
"loss": 0.1563,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.2532024383544922,
"rewards/margins": 2.304598093032837,
"rewards/rejected": -1.0513951778411865,
"step": 1320
},
{
"epoch": 3.45,
"learning_rate": 2.779113783626916e-07,
"logits/chosen": -1.8009631633758545,
"logits/rejected": -1.8024797439575195,
"logps/chosen": -32.74066925048828,
"logps/rejected": -34.96391296386719,
"loss": 0.1456,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.306864619255066,
"rewards/margins": 2.3590540885925293,
"rewards/rejected": -1.052189588546753,
"step": 1330
},
{
"epoch": 3.48,
"learning_rate": 2.5251829568697204e-07,
"logits/chosen": -1.8623323440551758,
"logits/rejected": -1.8611522912979126,
"logps/chosen": -29.76908302307129,
"logps/rejected": -33.332550048828125,
"loss": 0.1572,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.1436678171157837,
"rewards/margins": 2.21612811088562,
"rewards/rejected": -1.072460412979126,
"step": 1340
},
{
"epoch": 3.51,
"learning_rate": 2.2827986432927774e-07,
"logits/chosen": -1.877587914466858,
"logits/rejected": -1.862841248512268,
"logps/chosen": -32.96019744873047,
"logps/rejected": -37.72450637817383,
"loss": 0.153,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.0851805210113525,
"rewards/margins": 2.253241539001465,
"rewards/rejected": -1.1680610179901123,
"step": 1350
},
{
"epoch": 3.53,
"learning_rate": 2.0520853686560177e-07,
"logits/chosen": -1.8752319812774658,
"logits/rejected": -1.8875735998153687,
"logps/chosen": -30.424901962280273,
"logps/rejected": -33.42066192626953,
"loss": 0.1772,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.3148540258407593,
"rewards/margins": 2.2595856189727783,
"rewards/rejected": -0.9447315335273743,
"step": 1360
},
{
"epoch": 3.56,
"learning_rate": 1.833161662683672e-07,
"logits/chosen": -1.9719635248184204,
"logits/rejected": -1.971278429031372,
"logps/chosen": -30.404891967773438,
"logps/rejected": -37.444129943847656,
"loss": 0.1268,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.300431489944458,
"rewards/margins": 2.5910286903381348,
"rewards/rejected": -1.2905975580215454,
"step": 1370
},
{
"epoch": 3.58,
"learning_rate": 1.626139998169246e-07,
"logits/chosen": -1.8381322622299194,
"logits/rejected": -1.845894455909729,
"logps/chosen": -32.695735931396484,
"logps/rejected": -38.82695388793945,
"loss": 0.1845,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": 1.1877927780151367,
"rewards/margins": 2.2896358966827393,
"rewards/rejected": -1.101843237876892,
"step": 1380
},
{
"epoch": 3.61,
"learning_rate": 1.4311267331922535e-07,
"logits/chosen": -1.7921359539031982,
"logits/rejected": -1.7879730463027954,
"logps/chosen": -32.99811553955078,
"logps/rejected": -32.61560821533203,
"loss": 0.1972,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.4152759313583374,
"rewards/margins": 2.229846715927124,
"rewards/rejected": -0.8145707845687866,
"step": 1390
},
{
"epoch": 3.64,
"learning_rate": 1.2482220564763669e-07,
"logits/chosen": -1.9499454498291016,
"logits/rejected": -1.9474738836288452,
"logps/chosen": -29.77544593811035,
"logps/rejected": -33.36935806274414,
"loss": 0.1578,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.2168843746185303,
"rewards/margins": 2.1967949867248535,
"rewards/rejected": -0.979910671710968,
"step": 1400
},
{
"epoch": 3.64,
"eval_logits/chosen": -2.131394863128662,
"eval_logits/rejected": -2.126645803451538,
"eval_logps/chosen": -34.39291763305664,
"eval_logps/rejected": -38.091670989990234,
"eval_loss": 0.6838791370391846,
"eval_rewards/accuracies": 0.5859634280204773,
"eval_rewards/chosen": -0.28668972849845886,
"eval_rewards/margins": 0.17335131764411926,
"eval_rewards/rejected": -0.4600410461425781,
"eval_runtime": 145.9578,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1400
},
{
"epoch": 3.66,
"learning_rate": 1.0775199359171346e-07,
"logits/chosen": -1.8938592672348022,
"logits/rejected": -1.8896372318267822,
"logps/chosen": -32.033363342285156,
"logps/rejected": -30.647994995117188,
"loss": 0.1724,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.2873361110687256,
"rewards/margins": 2.162166118621826,
"rewards/rejected": -0.8748300671577454,
"step": 1410
},
{
"epoch": 3.69,
"learning_rate": 9.191080703056604e-08,
"logits/chosen": -1.8500086069107056,
"logits/rejected": -1.8511066436767578,
"logps/chosen": -31.751754760742188,
"logps/rejected": -35.97992706298828,
"loss": 0.1799,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.29049813747406,
"rewards/margins": 2.1352338790893555,
"rewards/rejected": -0.8447355031967163,
"step": 1420
},
{
"epoch": 3.71,
"learning_rate": 7.730678442730539e-08,
"logits/chosen": -1.8036447763442993,
"logits/rejected": -1.7969812154769897,
"logps/chosen": -32.511940002441406,
"logps/rejected": -38.00764846801758,
"loss": 0.1569,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.303862452507019,
"rewards/margins": 2.358455181121826,
"rewards/rejected": -1.0545929670333862,
"step": 1430
},
{
"epoch": 3.74,
"learning_rate": 6.394742864787806e-08,
"logits/chosen": -1.8134548664093018,
"logits/rejected": -1.8079309463500977,
"logps/chosen": -27.75140380859375,
"logps/rejected": -32.93821334838867,
"loss": 0.1916,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": 1.2494693994522095,
"rewards/margins": 2.2155184745788574,
"rewards/rejected": -0.9660491943359375,
"step": 1440
},
{
"epoch": 3.77,
"learning_rate": 5.183960310644748e-08,
"logits/chosen": -1.8413394689559937,
"logits/rejected": -1.8308031558990479,
"logps/chosen": -31.208871841430664,
"logps/rejected": -36.86098861694336,
"loss": 0.1533,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": 1.0181248188018799,
"rewards/margins": 2.2786247730255127,
"rewards/rejected": -1.260500192642212,
"step": 1450
},
{
"epoch": 3.79,
"learning_rate": 4.098952823928693e-08,
"logits/chosen": -1.8131767511367798,
"logits/rejected": -1.8103145360946655,
"logps/chosen": -31.82034683227539,
"logps/rejected": -31.843425750732422,
"loss": 0.2074,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.1915438175201416,
"rewards/margins": 1.9536281824111938,
"rewards/rejected": -0.7620843052864075,
"step": 1460
},
{
"epoch": 3.82,
"learning_rate": 3.1402778309014284e-08,
"logits/chosen": -1.8745816946029663,
"logits/rejected": -1.8810014724731445,
"logps/chosen": -30.35885238647461,
"logps/rejected": -34.228118896484375,
"loss": 0.1567,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.3682581186294556,
"rewards/margins": 2.380812883377075,
"rewards/rejected": -1.01255464553833,
"step": 1470
},
{
"epoch": 3.84,
"learning_rate": 2.3084278540791427e-08,
"logits/chosen": -1.8742589950561523,
"logits/rejected": -1.8845386505126953,
"logps/chosen": -30.218097686767578,
"logps/rejected": -30.740955352783203,
"loss": 0.168,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.1817275285720825,
"rewards/margins": 2.0716397762298584,
"rewards/rejected": -0.889912486076355,
"step": 1480
},
{
"epoch": 3.87,
"learning_rate": 1.6038302591975807e-08,
"logits/chosen": -1.8073488473892212,
"logits/rejected": -1.800286054611206,
"logps/chosen": -32.302825927734375,
"logps/rejected": -33.47874069213867,
"loss": 0.1842,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.082926869392395,
"rewards/margins": 2.097539186477661,
"rewards/rejected": -1.0146124362945557,
"step": 1490
},
{
"epoch": 3.9,
"learning_rate": 1.0268470356514237e-08,
"logits/chosen": -1.8606147766113281,
"logits/rejected": -1.8575241565704346,
"logps/chosen": -32.205726623535156,
"logps/rejected": -34.68817138671875,
"loss": 0.1614,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": 1.1358753442764282,
"rewards/margins": 2.2782340049743652,
"rewards/rejected": -1.142358660697937,
"step": 1500
},
{
"epoch": 3.9,
"eval_logits/chosen": -2.1312177181243896,
"eval_logits/rejected": -2.1264684200286865,
"eval_logps/chosen": -34.386234283447266,
"eval_logps/rejected": -38.08888244628906,
"eval_loss": 0.6819679737091064,
"eval_rewards/accuracies": 0.574335515499115,
"eval_rewards/chosen": -0.2813448905944824,
"eval_rewards/margins": 0.17646357417106628,
"eval_rewards/rejected": -0.4578084647655487,
"eval_runtime": 145.9567,
"eval_samples_per_second": 2.35,
"eval_steps_per_second": 0.295,
"step": 1500
},
{
"epoch": 3.92,
"learning_rate": 5.777746105209147e-09,
"logits/chosen": -1.9368705749511719,
"logits/rejected": -1.9374831914901733,
"logps/chosen": -28.38448715209961,
"logps/rejected": -34.06378173828125,
"loss": 0.2003,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.145431399345398,
"rewards/margins": 2.204097032546997,
"rewards/rejected": -1.0586655139923096,
"step": 1510
},
{
"epoch": 3.95,
"learning_rate": 2.5684369628148352e-09,
"logits/chosen": -1.7932342290878296,
"logits/rejected": -1.7930355072021484,
"logps/chosen": -31.60140037536621,
"logps/rejected": -35.09097671508789,
"loss": 0.2089,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 1.1523739099502563,
"rewards/margins": 2.1261672973632812,
"rewards/rejected": -0.9737933874130249,
"step": 1520
},
{
"epoch": 3.97,
"learning_rate": 6.421917227455999e-10,
"logits/chosen": -1.943830132484436,
"logits/rejected": -1.9411506652832031,
"logps/chosen": -30.156597137451172,
"logps/rejected": -32.75658416748047,
"loss": 0.1839,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": 1.053972840309143,
"rewards/margins": 2.0723206996917725,
"rewards/rejected": -1.018348217010498,
"step": 1530
},
{
"epoch": 4.0,
"learning_rate": 0.0,
"logits/chosen": -1.9267715215682983,
"logits/rejected": -1.9278953075408936,
"logps/chosen": -29.27459716796875,
"logps/rejected": -30.95477294921875,
"loss": 0.1802,
"rewards/accuracies": 0.98333340883255,
"rewards/chosen": 1.0762757062911987,
"rewards/margins": 2.007948398590088,
"rewards/rejected": -0.9316726922988892,
"step": 1540
},
{
"epoch": 4.0,
"step": 1540,
"total_flos": 0.0,
"train_loss": 0.20605388523696305,
"train_runtime": 10804.1609,
"train_samples_per_second": 1.14,
"train_steps_per_second": 0.143
}
],
"logging_steps": 10,
"max_steps": 1540,
"num_input_tokens_seen": 0,
"num_train_epochs": 4,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}