zephyr-7b-dpo-full / trainer_state.json
wzhouad's picture
Model save
ba49b4e verified
raw
history blame
No virus
51.4 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.9968602825745683,
"eval_steps": 100,
"global_step": 954,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02,
"learning_rate": 1.0416666666666667e-07,
"logits/chosen": 0.29368966817855835,
"logits/rejected": 0.3178113102912903,
"logps/chosen": -295.21783447265625,
"logps/rejected": -290.84619140625,
"loss": 0.6933,
"rewards/accuracies": 0.4000000059604645,
"rewards/chosen": 0.00023447822604794055,
"rewards/margins": 0.00020264319027774036,
"rewards/rejected": 3.183506123605184e-05,
"step": 10
},
{
"epoch": 0.04,
"learning_rate": 2.0833333333333333e-07,
"logits/chosen": 0.25306791067123413,
"logits/rejected": 0.3252382278442383,
"logps/chosen": -318.19073486328125,
"logps/rejected": -289.6706237792969,
"loss": 0.6929,
"rewards/accuracies": 0.53125,
"rewards/chosen": 0.00048495858209207654,
"rewards/margins": 0.0009848512709140778,
"rewards/rejected": -0.0004998926888220012,
"step": 20
},
{
"epoch": 0.06,
"learning_rate": 3.1249999999999997e-07,
"logits/chosen": 0.33226653933525085,
"logits/rejected": 0.3872108459472656,
"logps/chosen": -296.3697204589844,
"logps/rejected": -283.0611877441406,
"loss": 0.692,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.0008094090735539794,
"rewards/margins": 0.0026363185606896877,
"rewards/rejected": -0.003445727750658989,
"step": 30
},
{
"epoch": 0.08,
"learning_rate": 4.1666666666666667e-07,
"logits/chosen": 0.22801101207733154,
"logits/rejected": 0.32900214195251465,
"logps/chosen": -305.9015197753906,
"logps/rejected": -293.1842346191406,
"loss": 0.6883,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.0027595984283834696,
"rewards/margins": 0.006589935161173344,
"rewards/rejected": -0.00934953335672617,
"step": 40
},
{
"epoch": 0.1,
"learning_rate": 5.208333333333334e-07,
"logits/chosen": 0.27091675996780396,
"logits/rejected": 0.31866759061813354,
"logps/chosen": -314.2833557128906,
"logps/rejected": -307.02532958984375,
"loss": 0.683,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.0060192132368683815,
"rewards/margins": 0.019419629126787186,
"rewards/rejected": -0.025438839569687843,
"step": 50
},
{
"epoch": 0.13,
"learning_rate": 6.249999999999999e-07,
"logits/chosen": 0.31704145669937134,
"logits/rejected": 0.4334793984889984,
"logps/chosen": -294.429931640625,
"logps/rejected": -272.87994384765625,
"loss": 0.6735,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.022902976721525192,
"rewards/margins": 0.04408121109008789,
"rewards/rejected": -0.06698418408632278,
"step": 60
},
{
"epoch": 0.15,
"learning_rate": 7.291666666666666e-07,
"logits/chosen": 0.31964099407196045,
"logits/rejected": 0.3377896547317505,
"logps/chosen": -304.6803894042969,
"logps/rejected": -311.32794189453125,
"loss": 0.6602,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.05670947954058647,
"rewards/margins": 0.05069820210337639,
"rewards/rejected": -0.10740767419338226,
"step": 70
},
{
"epoch": 0.17,
"learning_rate": 8.333333333333333e-07,
"logits/chosen": 0.35048729181289673,
"logits/rejected": 0.4193252921104431,
"logps/chosen": -306.3404541015625,
"logps/rejected": -282.4783020019531,
"loss": 0.6474,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.10853584855794907,
"rewards/margins": 0.13117292523384094,
"rewards/rejected": -0.2397087812423706,
"step": 80
},
{
"epoch": 0.19,
"learning_rate": 9.374999999999999e-07,
"logits/chosen": 0.32813602685928345,
"logits/rejected": 0.4464220404624939,
"logps/chosen": -341.1703186035156,
"logps/rejected": -299.92340087890625,
"loss": 0.6357,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.2554694712162018,
"rewards/margins": 0.12284588813781738,
"rewards/rejected": -0.37831538915634155,
"step": 90
},
{
"epoch": 0.21,
"learning_rate": 9.999463737538052e-07,
"logits/chosen": 0.35799938440322876,
"logits/rejected": 0.3899138271808624,
"logps/chosen": -318.7712097167969,
"logps/rejected": -348.5688781738281,
"loss": 0.6201,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.3124231994152069,
"rewards/margins": 0.19829413294792175,
"rewards/rejected": -0.5107173323631287,
"step": 100
},
{
"epoch": 0.21,
"eval_logits/chosen": 0.3705582916736603,
"eval_logits/rejected": 0.4172414541244507,
"eval_logps/chosen": -331.11236572265625,
"eval_logps/rejected": -378.240478515625,
"eval_loss": 0.6252639293670654,
"eval_rewards/accuracies": 0.703125,
"eval_rewards/chosen": -0.27527713775634766,
"eval_rewards/margins": 0.3908771872520447,
"eval_rewards/rejected": -0.6661543846130371,
"eval_runtime": 64.993,
"eval_samples_per_second": 30.773,
"eval_steps_per_second": 0.492,
"step": 100
},
{
"epoch": 0.23,
"learning_rate": 9.993432105822034e-07,
"logits/chosen": 0.3002661168575287,
"logits/rejected": 0.3483879864215851,
"logps/chosen": -350.9095458984375,
"logps/rejected": -360.30963134765625,
"loss": 0.6093,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.37449145317077637,
"rewards/margins": 0.3816668689250946,
"rewards/rejected": -0.7561584115028381,
"step": 110
},
{
"epoch": 0.25,
"learning_rate": 9.980706626858607e-07,
"logits/chosen": 0.2088731825351715,
"logits/rejected": 0.2899537980556488,
"logps/chosen": -395.3984069824219,
"logps/rejected": -433.7286682128906,
"loss": 0.5905,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.4185329079627991,
"rewards/margins": 0.4517739713191986,
"rewards/rejected": -0.8703069686889648,
"step": 120
},
{
"epoch": 0.27,
"learning_rate": 9.961304359538434e-07,
"logits/chosen": 0.09751267731189728,
"logits/rejected": 0.22797170281410217,
"logps/chosen": -374.6502380371094,
"logps/rejected": -373.14263916015625,
"loss": 0.5998,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.4483565390110016,
"rewards/margins": 0.3120550811290741,
"rewards/rejected": -0.7604116201400757,
"step": 130
},
{
"epoch": 0.29,
"learning_rate": 9.935251313189563e-07,
"logits/chosen": 0.23218217492103577,
"logits/rejected": 0.28826406598091125,
"logps/chosen": -363.15338134765625,
"logps/rejected": -381.83795166015625,
"loss": 0.5847,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.49422675371170044,
"rewards/margins": 0.38399404287338257,
"rewards/rejected": -0.8782208561897278,
"step": 140
},
{
"epoch": 0.31,
"learning_rate": 9.902582412711118e-07,
"logits/chosen": 0.29499703645706177,
"logits/rejected": 0.49714046716690063,
"logps/chosen": -342.9471130371094,
"logps/rejected": -372.1763610839844,
"loss": 0.5681,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.6186890602111816,
"rewards/margins": 0.5347500443458557,
"rewards/rejected": -1.1534390449523926,
"step": 150
},
{
"epoch": 0.33,
"learning_rate": 9.86334145175542e-07,
"logits/chosen": 0.35773637890815735,
"logits/rejected": 0.4941268861293793,
"logps/chosen": -370.3203125,
"logps/rejected": -428.2557067871094,
"loss": 0.5792,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.7682583928108215,
"rewards/margins": 0.6092099547386169,
"rewards/rejected": -1.377468228340149,
"step": 160
},
{
"epoch": 0.36,
"learning_rate": 9.817581034021272e-07,
"logits/chosen": 0.19401590526103973,
"logits/rejected": 0.3156794607639313,
"logps/chosen": -415.65313720703125,
"logps/rejected": -425.68963623046875,
"loss": 0.5747,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.5680743455886841,
"rewards/margins": 0.4286484122276306,
"rewards/rejected": -0.9967228174209595,
"step": 170
},
{
"epoch": 0.38,
"learning_rate": 9.765362502737097e-07,
"logits/chosen": 0.2572034001350403,
"logits/rejected": 0.27328386902809143,
"logps/chosen": -358.0533447265625,
"logps/rejected": -398.5332946777344,
"loss": 0.5634,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.644694447517395,
"rewards/margins": 0.5251447558403015,
"rewards/rejected": -1.1698391437530518,
"step": 180
},
{
"epoch": 0.4,
"learning_rate": 9.706755858428485e-07,
"logits/chosen": 0.3962785303592682,
"logits/rejected": 0.4539657235145569,
"logps/chosen": -347.5164794921875,
"logps/rejected": -378.83184814453125,
"loss": 0.5423,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.7963652610778809,
"rewards/margins": 0.5720622539520264,
"rewards/rejected": -1.3684275150299072,
"step": 190
},
{
"epoch": 0.42,
"learning_rate": 9.641839665080363e-07,
"logits/chosen": 0.3198946714401245,
"logits/rejected": 0.4063253402709961,
"logps/chosen": -352.0743713378906,
"logps/rejected": -419.7388610839844,
"loss": 0.5547,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.7651049494743347,
"rewards/margins": 0.6531444787979126,
"rewards/rejected": -1.418249487876892,
"step": 200
},
{
"epoch": 0.42,
"eval_logits/chosen": 0.3908616304397583,
"eval_logits/rejected": 0.4261176884174347,
"eval_logps/chosen": -373.46612548828125,
"eval_logps/rejected": -458.88629150390625,
"eval_loss": 0.5549300312995911,
"eval_rewards/accuracies": 0.765625,
"eval_rewards/chosen": -0.6988146305084229,
"eval_rewards/margins": 0.7737974524497986,
"eval_rewards/rejected": -1.4726121425628662,
"eval_runtime": 65.2313,
"eval_samples_per_second": 30.66,
"eval_steps_per_second": 0.491,
"step": 200
},
{
"epoch": 0.44,
"learning_rate": 9.570700944819582e-07,
"logits/chosen": 0.3505176901817322,
"logits/rejected": 0.42375579476356506,
"logps/chosen": -397.61199951171875,
"logps/rejected": -454.89776611328125,
"loss": 0.5265,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.7523492574691772,
"rewards/margins": 0.7344074845314026,
"rewards/rejected": -1.486756682395935,
"step": 210
},
{
"epoch": 0.46,
"learning_rate": 9.493435061259129e-07,
"logits/chosen": 0.2726442813873291,
"logits/rejected": 0.4434526860713959,
"logps/chosen": -410.77667236328125,
"logps/rejected": -431.207275390625,
"loss": 0.555,
"rewards/accuracies": 0.731249988079071,
"rewards/chosen": -0.8984044194221497,
"rewards/margins": 0.5839776992797852,
"rewards/rejected": -1.48238205909729,
"step": 220
},
{
"epoch": 0.48,
"learning_rate": 9.4101455916603e-07,
"logits/chosen": 0.28837597370147705,
"logits/rejected": 0.35526323318481445,
"logps/chosen": -363.0335998535156,
"logps/rejected": -436.46612548828125,
"loss": 0.5498,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.8101651072502136,
"rewards/margins": 0.737695574760437,
"rewards/rejected": -1.5478605031967163,
"step": 230
},
{
"epoch": 0.5,
"learning_rate": 9.320944188084241e-07,
"logits/chosen": 0.23826150596141815,
"logits/rejected": 0.285171240568161,
"logps/chosen": -432.6297912597656,
"logps/rejected": -497.56341552734375,
"loss": 0.5392,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.7939236164093018,
"rewards/margins": 0.7111212611198425,
"rewards/rejected": -1.505044937133789,
"step": 240
},
{
"epoch": 0.52,
"learning_rate": 9.225950427718974e-07,
"logits/chosen": 0.2762988209724426,
"logits/rejected": 0.31130319833755493,
"logps/chosen": -399.399169921875,
"logps/rejected": -445.702880859375,
"loss": 0.5465,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.83611661195755,
"rewards/margins": 0.7073522806167603,
"rewards/rejected": -1.5434690713882446,
"step": 250
},
{
"epoch": 0.54,
"learning_rate": 9.125291652582547e-07,
"logits/chosen": 0.1327328383922577,
"logits/rejected": 0.3085227310657501,
"logps/chosen": -436.08135986328125,
"logps/rejected": -457.634765625,
"loss": 0.5194,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9428914785385132,
"rewards/margins": 0.6950392723083496,
"rewards/rejected": -1.6379308700561523,
"step": 260
},
{
"epoch": 0.57,
"learning_rate": 9.019102798817195e-07,
"logits/chosen": 0.23745720088481903,
"logits/rejected": 0.34172096848487854,
"logps/chosen": -421.8299865722656,
"logps/rejected": -466.4856872558594,
"loss": 0.5496,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.9325121641159058,
"rewards/margins": 0.7680062651634216,
"rewards/rejected": -1.7005186080932617,
"step": 270
},
{
"epoch": 0.59,
"learning_rate": 8.90752621580335e-07,
"logits/chosen": 0.16251161694526672,
"logits/rejected": 0.2581509053707123,
"logps/chosen": -418.5828552246094,
"logps/rejected": -507.22412109375,
"loss": 0.5168,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.113261342048645,
"rewards/margins": 0.7757157683372498,
"rewards/rejected": -1.88897705078125,
"step": 280
},
{
"epoch": 0.61,
"learning_rate": 8.79071147533597e-07,
"logits/chosen": 0.18345972895622253,
"logits/rejected": 0.24752414226531982,
"logps/chosen": -374.2388916015625,
"logps/rejected": -421.7548828125,
"loss": 0.5452,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7729798555374146,
"rewards/margins": 0.6754422187805176,
"rewards/rejected": -1.448421835899353,
"step": 290
},
{
"epoch": 0.63,
"learning_rate": 8.668815171119019e-07,
"logits/chosen": 0.11917382478713989,
"logits/rejected": 0.2862890362739563,
"logps/chosen": -396.6165466308594,
"logps/rejected": -400.9344787597656,
"loss": 0.5343,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.8496394157409668,
"rewards/margins": 0.5671547055244446,
"rewards/rejected": -1.4167941808700562,
"step": 300
},
{
"epoch": 0.63,
"eval_logits/chosen": 0.24486932158470154,
"eval_logits/rejected": 0.2851215898990631,
"eval_logps/chosen": -384.0199279785156,
"eval_logps/rejected": -476.3627624511719,
"eval_loss": 0.5315821766853333,
"eval_rewards/accuracies": 0.765625,
"eval_rewards/chosen": -0.8043524026870728,
"eval_rewards/margins": 0.8430246114730835,
"eval_rewards/rejected": -1.6473771333694458,
"eval_runtime": 64.8898,
"eval_samples_per_second": 30.822,
"eval_steps_per_second": 0.493,
"step": 300
},
{
"epoch": 0.65,
"learning_rate": 8.54200070884685e-07,
"logits/chosen": 0.21563191711902618,
"logits/rejected": 0.1952591836452484,
"logps/chosen": -438.91552734375,
"logps/rejected": -499.61395263671875,
"loss": 0.5567,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.173762559890747,
"rewards/margins": 0.5995947122573853,
"rewards/rejected": -1.7733571529388428,
"step": 310
},
{
"epoch": 0.67,
"learning_rate": 8.410438087153911e-07,
"logits/chosen": 0.031750187277793884,
"logits/rejected": 0.14312420785427094,
"logps/chosen": -457.0936584472656,
"logps/rejected": -459.15411376953125,
"loss": 0.5366,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.0222933292388916,
"rewards/margins": 0.4668423533439636,
"rewards/rejected": -1.4891356229782104,
"step": 320
},
{
"epoch": 0.69,
"learning_rate": 8.274303669726426e-07,
"logits/chosen": 0.07745673507452011,
"logits/rejected": 0.07082104682922363,
"logps/chosen": -399.32464599609375,
"logps/rejected": -523.4277954101562,
"loss": 0.532,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.9549884796142578,
"rewards/margins": 0.8862001299858093,
"rewards/rejected": -1.8411887884140015,
"step": 330
},
{
"epoch": 0.71,
"learning_rate": 8.133779948881513e-07,
"logits/chosen": 0.10294970124959946,
"logits/rejected": 0.09352216869592667,
"logps/chosen": -409.33770751953125,
"logps/rejected": -510.10089111328125,
"loss": 0.5217,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.1990526914596558,
"rewards/margins": 0.7596696615219116,
"rewards/rejected": -1.958722472190857,
"step": 340
},
{
"epoch": 0.73,
"learning_rate": 7.989055300930704e-07,
"logits/chosen": 0.16737070679664612,
"logits/rejected": 0.21901166439056396,
"logps/chosen": -437.2059020996094,
"logps/rejected": -538.3087158203125,
"loss": 0.5091,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -1.2976617813110352,
"rewards/margins": 0.8197441101074219,
"rewards/rejected": -2.117405891418457,
"step": 350
},
{
"epoch": 0.75,
"learning_rate": 7.840323733655778e-07,
"logits/chosen": 0.1267612874507904,
"logits/rejected": 0.20460394024848938,
"logps/chosen": -465.1019592285156,
"logps/rejected": -518.5805053710938,
"loss": 0.5185,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -1.174440860748291,
"rewards/margins": 0.9074214696884155,
"rewards/rejected": -2.081862211227417,
"step": 360
},
{
"epoch": 0.77,
"learning_rate": 7.687784626235447e-07,
"logits/chosen": 0.18740633130073547,
"logits/rejected": 0.27840983867645264,
"logps/chosen": -468.21807861328125,
"logps/rejected": -493.06646728515625,
"loss": 0.5158,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.1525232791900635,
"rewards/margins": 0.7130603790283203,
"rewards/rejected": -1.8655836582183838,
"step": 370
},
{
"epoch": 0.8,
"learning_rate": 7.531642461971514e-07,
"logits/chosen": 0.15459200739860535,
"logits/rejected": 0.23283176124095917,
"logps/chosen": -453.07904052734375,
"logps/rejected": -535.761474609375,
"loss": 0.5392,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -1.2102587223052979,
"rewards/margins": 1.0239769220352173,
"rewards/rejected": -2.2342355251312256,
"step": 380
},
{
"epoch": 0.82,
"learning_rate": 7.372106554172801e-07,
"logits/chosen": 0.19951777160167694,
"logits/rejected": 0.2895793318748474,
"logps/chosen": -411.4781799316406,
"logps/rejected": -466.41253662109375,
"loss": 0.5356,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.8807674646377563,
"rewards/margins": 0.7255697846412659,
"rewards/rejected": -1.606337308883667,
"step": 390
},
{
"epoch": 0.84,
"learning_rate": 7.209390765564318e-07,
"logits/chosen": 0.1757899820804596,
"logits/rejected": 0.23791635036468506,
"logps/chosen": -390.8957214355469,
"logps/rejected": -478.49407958984375,
"loss": 0.5323,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.9380720257759094,
"rewards/margins": 0.7229949235916138,
"rewards/rejected": -1.661067008972168,
"step": 400
},
{
"epoch": 0.84,
"eval_logits/chosen": 0.25136542320251465,
"eval_logits/rejected": 0.28342366218566895,
"eval_logps/chosen": -394.2620849609375,
"eval_logps/rejected": -494.46002197265625,
"eval_loss": 0.521114706993103,
"eval_rewards/accuracies": 0.78125,
"eval_rewards/chosen": -0.90677410364151,
"eval_rewards/margins": 0.9215754270553589,
"eval_rewards/rejected": -1.8283497095108032,
"eval_runtime": 65.2477,
"eval_samples_per_second": 30.652,
"eval_steps_per_second": 0.49,
"step": 400
},
{
"epoch": 0.86,
"learning_rate": 7.043713221597773e-07,
"logits/chosen": 0.1401471644639969,
"logits/rejected": 0.17819848656654358,
"logps/chosen": -401.8942565917969,
"logps/rejected": -474.2579650878906,
"loss": 0.5225,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.8892021179199219,
"rewards/margins": 0.906479001045227,
"rewards/rejected": -1.7956812381744385,
"step": 410
},
{
"epoch": 0.88,
"learning_rate": 6.875296018047809e-07,
"logits/chosen": 0.1304786652326584,
"logits/rejected": 0.15569528937339783,
"logps/chosen": -400.5438537597656,
"logps/rejected": -468.2953186035156,
"loss": 0.5065,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -0.9208317995071411,
"rewards/margins": 0.9665653109550476,
"rewards/rejected": -1.887397050857544,
"step": 420
},
{
"epoch": 0.9,
"learning_rate": 6.704364923285857e-07,
"logits/chosen": 0.09775003790855408,
"logits/rejected": 0.1501173973083496,
"logps/chosen": -476.73065185546875,
"logps/rejected": -533.7389526367188,
"loss": 0.5285,
"rewards/accuracies": 0.71875,
"rewards/chosen": -1.26286780834198,
"rewards/margins": 0.8789494633674622,
"rewards/rejected": -2.141817569732666,
"step": 430
},
{
"epoch": 0.92,
"learning_rate": 6.531149075630796e-07,
"logits/chosen": 0.1202029138803482,
"logits/rejected": 0.1756385862827301,
"logps/chosen": -400.1629333496094,
"logps/rejected": -469.0196838378906,
"loss": 0.5074,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -1.0582826137542725,
"rewards/margins": 0.9171028137207031,
"rewards/rejected": -1.975385308265686,
"step": 440
},
{
"epoch": 0.94,
"learning_rate": 6.355880676182085e-07,
"logits/chosen": 0.01758761703968048,
"logits/rejected": 0.15642888844013214,
"logps/chosen": -443.065185546875,
"logps/rejected": -484.41473388671875,
"loss": 0.5295,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.004433035850525,
"rewards/margins": 0.8090255856513977,
"rewards/rejected": -1.8134586811065674,
"step": 450
},
{
"epoch": 0.96,
"learning_rate": 6.178794677547137e-07,
"logits/chosen": 0.012063628062605858,
"logits/rejected": 0.10572747141122818,
"logps/chosen": -408.4104919433594,
"logps/rejected": -449.46881103515625,
"loss": 0.5267,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -1.0622318983078003,
"rewards/margins": 0.6900007724761963,
"rewards/rejected": -1.752232551574707,
"step": 460
},
{
"epoch": 0.98,
"learning_rate": 6.000128468880222e-07,
"logits/chosen": -0.05889149755239487,
"logits/rejected": -0.014351313933730125,
"logps/chosen": -449.07061767578125,
"logps/rejected": -538.423828125,
"loss": 0.5042,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.1122030019760132,
"rewards/margins": 0.9293983578681946,
"rewards/rejected": -2.0416014194488525,
"step": 470
},
{
"epoch": 1.0,
"learning_rate": 5.820121557655108e-07,
"logits/chosen": -0.045923542231321335,
"logits/rejected": 0.03086056187748909,
"logps/chosen": -422.5347595214844,
"logps/rejected": -529.2613525390625,
"loss": 0.4682,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -1.275895118713379,
"rewards/margins": 1.037326455116272,
"rewards/rejected": -2.3132214546203613,
"step": 480
},
{
"epoch": 1.03,
"learning_rate": 5.639015248598023e-07,
"logits/chosen": -0.05794327333569527,
"logits/rejected": -0.09440571069717407,
"logps/chosen": -441.08990478515625,
"logps/rejected": -601.30419921875,
"loss": 0.3493,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -1.3565930128097534,
"rewards/margins": 1.8090267181396484,
"rewards/rejected": -3.1656198501586914,
"step": 490
},
{
"epoch": 1.05,
"learning_rate": 5.457052320211339e-07,
"logits/chosen": -0.23803594708442688,
"logits/rejected": -0.29828980565071106,
"logps/chosen": -474.3367614746094,
"logps/rejected": -656.8623046875,
"loss": 0.352,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -1.730020523071289,
"rewards/margins": 1.8847625255584717,
"rewards/rejected": -3.6147830486297607,
"step": 500
},
{
"epoch": 1.05,
"eval_logits/chosen": -0.06535135954618454,
"eval_logits/rejected": -0.08458372950553894,
"eval_logps/chosen": -498.9117431640625,
"eval_logps/rejected": -653.2899169921875,
"eval_loss": 0.525809109210968,
"eval_rewards/accuracies": 0.796875,
"eval_rewards/chosen": -1.9532711505889893,
"eval_rewards/margins": 1.4633771181106567,
"eval_rewards/rejected": -3.4166483879089355,
"eval_runtime": 64.5447,
"eval_samples_per_second": 30.986,
"eval_steps_per_second": 0.496,
"step": 500
},
{
"epoch": 1.07,
"learning_rate": 5.274476699321637e-07,
"logits/chosen": -0.17468394339084625,
"logits/rejected": -0.19271844625473022,
"logps/chosen": -464.8907165527344,
"logps/rejected": -632.8477172851562,
"loss": 0.3423,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": -1.695770263671875,
"rewards/margins": 1.8958046436309814,
"rewards/rejected": -3.5915749073028564,
"step": 510
},
{
"epoch": 1.09,
"learning_rate": 5.091533134088387e-07,
"logits/chosen": -0.23788562417030334,
"logits/rejected": -0.20414999127388,
"logps/chosen": -485.94415283203125,
"logps/rejected": -640.9654541015625,
"loss": 0.3236,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -1.8264538049697876,
"rewards/margins": 1.7633371353149414,
"rewards/rejected": -3.5897908210754395,
"step": 520
},
{
"epoch": 1.11,
"learning_rate": 4.908466865911614e-07,
"logits/chosen": -0.11999205499887466,
"logits/rejected": -0.09423510730266571,
"logps/chosen": -487.30810546875,
"logps/rejected": -646.9547119140625,
"loss": 0.3424,
"rewards/accuracies": 0.84375,
"rewards/chosen": -1.7824156284332275,
"rewards/margins": 1.7626768350601196,
"rewards/rejected": -3.5450921058654785,
"step": 530
},
{
"epoch": 1.13,
"learning_rate": 4.7255233006783624e-07,
"logits/chosen": -0.16903451085090637,
"logits/rejected": -0.06715662032365799,
"logps/chosen": -530.5585327148438,
"logps/rejected": -645.1633911132812,
"loss": 0.3301,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -1.7870140075683594,
"rewards/margins": 1.8271121978759766,
"rewards/rejected": -3.614126682281494,
"step": 540
},
{
"epoch": 1.15,
"learning_rate": 4.5429476797886617e-07,
"logits/chosen": -0.03496643900871277,
"logits/rejected": -0.04560618847608566,
"logps/chosen": -483.15203857421875,
"logps/rejected": -634.9447631835938,
"loss": 0.3501,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -1.8454551696777344,
"rewards/margins": 1.5336250066757202,
"rewards/rejected": -3.379080295562744,
"step": 550
},
{
"epoch": 1.17,
"learning_rate": 4.3609847514019763e-07,
"logits/chosen": 0.06872721016407013,
"logits/rejected": 0.0859331339597702,
"logps/chosen": -506.5009765625,
"logps/rejected": -646.7543334960938,
"loss": 0.3359,
"rewards/accuracies": 0.875,
"rewards/chosen": -1.7461020946502686,
"rewards/margins": 1.8654359579086304,
"rewards/rejected": -3.6115379333496094,
"step": 560
},
{
"epoch": 1.19,
"learning_rate": 4.179878442344892e-07,
"logits/chosen": -0.04321649298071861,
"logits/rejected": 0.051123034209012985,
"logps/chosen": -496.735595703125,
"logps/rejected": -647.29150390625,
"loss": 0.3464,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -2.0714824199676514,
"rewards/margins": 1.762717843055725,
"rewards/rejected": -3.834200620651245,
"step": 570
},
{
"epoch": 1.21,
"learning_rate": 3.9998715311197783e-07,
"logits/chosen": -0.004042728338390589,
"logits/rejected": 0.04070080816745758,
"logps/chosen": -512.9417724609375,
"logps/rejected": -696.821044921875,
"loss": 0.3365,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -2.097195863723755,
"rewards/margins": 1.831916093826294,
"rewards/rejected": -3.9291114807128906,
"step": 580
},
{
"epoch": 1.23,
"learning_rate": 3.821205322452863e-07,
"logits/chosen": 0.09215477854013443,
"logits/rejected": 0.05737446993589401,
"logps/chosen": -525.1434936523438,
"logps/rejected": -705.9063110351562,
"loss": 0.3158,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.0721964836120605,
"rewards/margins": 1.9721500873565674,
"rewards/rejected": -4.044346809387207,
"step": 590
},
{
"epoch": 1.26,
"learning_rate": 3.6441193238179146e-07,
"logits/chosen": 0.07227401435375214,
"logits/rejected": 0.03988388180732727,
"logps/chosen": -549.4615478515625,
"logps/rejected": -771.7275390625,
"loss": 0.3342,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.2555530071258545,
"rewards/margins": 2.014453649520874,
"rewards/rejected": -4.2700066566467285,
"step": 600
},
{
"epoch": 1.26,
"eval_logits/chosen": 0.1344175487756729,
"eval_logits/rejected": 0.11280365288257599,
"eval_logps/chosen": -534.8101196289062,
"eval_logps/rejected": -684.085693359375,
"eval_loss": 0.5267595648765564,
"eval_rewards/accuracies": 0.79296875,
"eval_rewards/chosen": -2.3122546672821045,
"eval_rewards/margins": 1.412351369857788,
"eval_rewards/rejected": -3.7246060371398926,
"eval_runtime": 64.3013,
"eval_samples_per_second": 31.104,
"eval_steps_per_second": 0.498,
"step": 600
},
{
"epoch": 1.28,
"learning_rate": 3.4688509243692034e-07,
"logits/chosen": 0.021421348676085472,
"logits/rejected": 0.08866464346647263,
"logps/chosen": -535.4236450195312,
"logps/rejected": -694.8360595703125,
"loss": 0.344,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -2.009089946746826,
"rewards/margins": 1.8043378591537476,
"rewards/rejected": -3.8134284019470215,
"step": 610
},
{
"epoch": 1.3,
"learning_rate": 3.295635076714144e-07,
"logits/chosen": -0.06299210339784622,
"logits/rejected": -0.04097691923379898,
"logps/chosen": -499.0265197753906,
"logps/rejected": -653.1188354492188,
"loss": 0.3317,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -1.9393279552459717,
"rewards/margins": 1.6952606439590454,
"rewards/rejected": -3.6345887184143066,
"step": 620
},
{
"epoch": 1.32,
"learning_rate": 3.12470398195219e-07,
"logits/chosen": -0.013870243914425373,
"logits/rejected": 0.0671583041548729,
"logps/chosen": -512.1448974609375,
"logps/rejected": -680.2833251953125,
"loss": 0.3283,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -1.9993736743927002,
"rewards/margins": 1.9316644668579102,
"rewards/rejected": -3.9310379028320312,
"step": 630
},
{
"epoch": 1.34,
"learning_rate": 2.956286778402226e-07,
"logits/chosen": -0.11906696856021881,
"logits/rejected": -0.17396704852581024,
"logps/chosen": -495.10345458984375,
"logps/rejected": -677.45556640625,
"loss": 0.319,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.2106266021728516,
"rewards/margins": 1.9618394374847412,
"rewards/rejected": -4.172466278076172,
"step": 640
},
{
"epoch": 1.36,
"learning_rate": 2.7906092344356826e-07,
"logits/chosen": -0.07961982488632202,
"logits/rejected": -0.12522803246974945,
"logps/chosen": -516.12890625,
"logps/rejected": -696.549560546875,
"loss": 0.3233,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -2.1828126907348633,
"rewards/margins": 1.8567641973495483,
"rewards/rejected": -4.039577007293701,
"step": 650
},
{
"epoch": 1.38,
"learning_rate": 2.6278934458271996e-07,
"logits/chosen": -0.15696656703948975,
"logits/rejected": -0.12818947434425354,
"logps/chosen": -568.789794921875,
"logps/rejected": -722.9078369140625,
"loss": 0.3273,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -2.1225264072418213,
"rewards/margins": 1.9797760248184204,
"rewards/rejected": -4.102302551269531,
"step": 660
},
{
"epoch": 1.4,
"learning_rate": 2.468357538028487e-07,
"logits/chosen": 0.01013887207955122,
"logits/rejected": 0.029213298112154007,
"logps/chosen": -541.7221069335938,
"logps/rejected": -720.105712890625,
"loss": 0.3153,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.205817461013794,
"rewards/margins": 2.0757954120635986,
"rewards/rejected": -4.281612396240234,
"step": 670
},
{
"epoch": 1.42,
"learning_rate": 2.312215373764551e-07,
"logits/chosen": -0.06487278640270233,
"logits/rejected": -0.01965305209159851,
"logps/chosen": -516.360107421875,
"logps/rejected": -694.418212890625,
"loss": 0.3262,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.184447765350342,
"rewards/margins": 1.861135721206665,
"rewards/rejected": -4.045583248138428,
"step": 680
},
{
"epoch": 1.44,
"learning_rate": 2.1596762663442213e-07,
"logits/chosen": -0.16129662096500397,
"logits/rejected": -0.09581325948238373,
"logps/chosen": -546.6666259765625,
"logps/rejected": -722.95263671875,
"loss": 0.3194,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -2.1821353435516357,
"rewards/margins": 2.131826877593994,
"rewards/rejected": -4.313961982727051,
"step": 690
},
{
"epoch": 1.47,
"learning_rate": 2.0109446990692963e-07,
"logits/chosen": -0.11689990758895874,
"logits/rejected": -0.2061731368303299,
"logps/chosen": -537.8167724609375,
"logps/rejected": -781.1466674804688,
"loss": 0.337,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.2769298553466797,
"rewards/margins": 2.1021370887756348,
"rewards/rejected": -4.3790669441223145,
"step": 700
},
{
"epoch": 1.47,
"eval_logits/chosen": 0.041396014392375946,
"eval_logits/rejected": 0.009947247803211212,
"eval_logps/chosen": -541.1116333007812,
"eval_logps/rejected": -699.990966796875,
"eval_loss": 0.5290427207946777,
"eval_rewards/accuracies": 0.77734375,
"eval_rewards/chosen": -2.375269651412964,
"eval_rewards/margins": 1.5083887577056885,
"eval_rewards/rejected": -3.883657932281494,
"eval_runtime": 64.3756,
"eval_samples_per_second": 31.068,
"eval_steps_per_second": 0.497,
"step": 700
},
{
"epoch": 1.49,
"learning_rate": 1.8662200511184872e-07,
"logits/chosen": -0.05749096721410751,
"logits/rejected": -0.05046076700091362,
"logps/chosen": -522.4590454101562,
"logps/rejected": -701.3648071289062,
"loss": 0.3387,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.1689815521240234,
"rewards/margins": 2.023318290710449,
"rewards/rejected": -4.192299842834473,
"step": 710
},
{
"epoch": 1.51,
"learning_rate": 1.725696330273575e-07,
"logits/chosen": -0.12996384501457214,
"logits/rejected": -0.20302283763885498,
"logps/chosen": -513.8458251953125,
"logps/rejected": -704.156982421875,
"loss": 0.3363,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -1.9350038766860962,
"rewards/margins": 2.0845541954040527,
"rewards/rejected": -4.019558429718018,
"step": 720
},
{
"epoch": 1.53,
"learning_rate": 1.589561912846089e-07,
"logits/chosen": -0.02087187021970749,
"logits/rejected": -0.01297883689403534,
"logps/chosen": -513.4478149414062,
"logps/rejected": -715.5861206054688,
"loss": 0.3311,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.01027512550354,
"rewards/margins": 1.9698785543441772,
"rewards/rejected": -3.9801535606384277,
"step": 730
},
{
"epoch": 1.55,
"learning_rate": 1.4579992911531496e-07,
"logits/chosen": -0.08378951251506805,
"logits/rejected": -0.10333013534545898,
"logps/chosen": -528.8599853515625,
"logps/rejected": -697.0358276367188,
"loss": 0.31,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -2.1352083683013916,
"rewards/margins": 2.004941940307617,
"rewards/rejected": -4.140150547027588,
"step": 740
},
{
"epoch": 1.57,
"learning_rate": 1.3311848288809813e-07,
"logits/chosen": -0.021279722452163696,
"logits/rejected": -0.07772192358970642,
"logps/chosen": -553.6995239257812,
"logps/rejected": -696.1124267578125,
"loss": 0.3325,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.2102222442626953,
"rewards/margins": 1.8731515407562256,
"rewards/rejected": -4.083374500274658,
"step": 750
},
{
"epoch": 1.59,
"learning_rate": 1.209288524664029e-07,
"logits/chosen": -0.1553444117307663,
"logits/rejected": -0.07370997965335846,
"logps/chosen": -559.6234130859375,
"logps/rejected": -718.3536376953125,
"loss": 0.3318,
"rewards/accuracies": 0.84375,
"rewards/chosen": -2.2175405025482178,
"rewards/margins": 2.087991714477539,
"rewards/rejected": -4.305531978607178,
"step": 760
},
{
"epoch": 1.61,
"learning_rate": 1.0924737841966497e-07,
"logits/chosen": 0.06289811432361603,
"logits/rejected": 0.08451451361179352,
"logps/chosen": -502.3817443847656,
"logps/rejected": -655.4021606445312,
"loss": 0.3065,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": -2.215487241744995,
"rewards/margins": 1.8723819255828857,
"rewards/rejected": -4.087869644165039,
"step": 770
},
{
"epoch": 1.63,
"learning_rate": 9.808972011828054e-08,
"logits/chosen": -0.028034457936882973,
"logits/rejected": -0.07584713399410248,
"logps/chosen": -507.1363220214844,
"logps/rejected": -684.7667236328125,
"loss": 0.321,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -2.1353392601013184,
"rewards/margins": 1.854984998703003,
"rewards/rejected": -3.9903244972229004,
"step": 780
},
{
"epoch": 1.65,
"learning_rate": 8.747083474174527e-08,
"logits/chosen": -0.027467548847198486,
"logits/rejected": -0.11992067098617554,
"logps/chosen": -513.2483520507812,
"logps/rejected": -773.5486450195312,
"loss": 0.3106,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -2.409447431564331,
"rewards/margins": 2.4787163734436035,
"rewards/rejected": -4.8881635665893555,
"step": 790
},
{
"epoch": 1.67,
"learning_rate": 7.740495722810269e-08,
"logits/chosen": -0.22111931443214417,
"logits/rejected": -0.16927292943000793,
"logps/chosen": -594.322998046875,
"logps/rejected": -767.4426879882812,
"loss": 0.3398,
"rewards/accuracies": 0.84375,
"rewards/chosen": -2.4693970680236816,
"rewards/margins": 2.0875487327575684,
"rewards/rejected": -4.55694580078125,
"step": 800
},
{
"epoch": 1.67,
"eval_logits/chosen": 0.0749908834695816,
"eval_logits/rejected": 0.0380852147936821,
"eval_logps/chosen": -554.5546264648438,
"eval_logps/rejected": -712.9505615234375,
"eval_loss": 0.5297122001647949,
"eval_rewards/accuracies": 0.7734375,
"eval_rewards/chosen": -2.509699821472168,
"eval_rewards/margins": 1.5035548210144043,
"eval_rewards/rejected": -4.013254642486572,
"eval_runtime": 65.3757,
"eval_samples_per_second": 30.592,
"eval_steps_per_second": 0.489,
"step": 800
},
{
"epoch": 1.7,
"learning_rate": 6.790558119157597e-08,
"logits/chosen": -0.03027234971523285,
"logits/rejected": -0.026968132704496384,
"logps/chosen": -558.7872924804688,
"logps/rejected": -744.04736328125,
"loss": 0.3134,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.323040246963501,
"rewards/margins": 2.1539111137390137,
"rewards/rejected": -4.476951599121094,
"step": 810
},
{
"epoch": 1.72,
"learning_rate": 5.898544083397e-08,
"logits/chosen": -0.06089891865849495,
"logits/rejected": -0.11520856618881226,
"logps/chosen": -516.7492065429688,
"logps/rejected": -698.2511596679688,
"loss": 0.3131,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -2.2953174114227295,
"rewards/margins": 2.0010485649108887,
"rewards/rejected": -4.296365737915039,
"step": 820
},
{
"epoch": 1.74,
"learning_rate": 5.065649387408705e-08,
"logits/chosen": -0.026049736887216568,
"logits/rejected": -0.11464808881282806,
"logps/chosen": -556.2135620117188,
"logps/rejected": -718.17578125,
"loss": 0.3145,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -2.405728816986084,
"rewards/margins": 1.908630132675171,
"rewards/rejected": -4.314358711242676,
"step": 830
},
{
"epoch": 1.76,
"learning_rate": 4.292990551804171e-08,
"logits/chosen": -0.12712730467319489,
"logits/rejected": -0.10675887763500214,
"logps/chosen": -529.3737182617188,
"logps/rejected": -723.229248046875,
"loss": 0.327,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -2.146822929382324,
"rewards/margins": 2.177410125732422,
"rewards/rejected": -4.324233055114746,
"step": 840
},
{
"epoch": 1.78,
"learning_rate": 3.581603349196371e-08,
"logits/chosen": 0.01494809053838253,
"logits/rejected": -0.08690011501312256,
"logps/chosen": -540.9631958007812,
"logps/rejected": -735.3316650390625,
"loss": 0.3262,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": -2.145940065383911,
"rewards/margins": 2.0986034870147705,
"rewards/rejected": -4.24454402923584,
"step": 850
},
{
"epoch": 1.8,
"learning_rate": 2.9324414157151367e-08,
"logits/chosen": -0.10003119707107544,
"logits/rejected": -0.13639459013938904,
"logps/chosen": -504.34552001953125,
"logps/rejected": -681.4578247070312,
"loss": 0.3261,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.0967519283294678,
"rewards/margins": 1.8127696514129639,
"rewards/rejected": -3.9095215797424316,
"step": 860
},
{
"epoch": 1.82,
"learning_rate": 2.3463749726290284e-08,
"logits/chosen": -0.005720620043575764,
"logits/rejected": -0.09044505655765533,
"logps/chosen": -550.0,
"logps/rejected": -727.3507690429688,
"loss": 0.3221,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -2.1374850273132324,
"rewards/margins": 2.060269355773926,
"rewards/rejected": -4.197754859924316,
"step": 870
},
{
"epoch": 1.84,
"learning_rate": 1.824189659787284e-08,
"logits/chosen": -0.07426755130290985,
"logits/rejected": -0.09251859039068222,
"logps/chosen": -551.6723022460938,
"logps/rejected": -753.9895629882812,
"loss": 0.3162,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.1559104919433594,
"rewards/margins": 2.1075100898742676,
"rewards/rejected": -4.263420581817627,
"step": 880
},
{
"epoch": 1.86,
"learning_rate": 1.3665854824458035e-08,
"logits/chosen": -0.05439913272857666,
"logits/rejected": -0.06511974334716797,
"logps/chosen": -540.4793090820312,
"logps/rejected": -690.4094848632812,
"loss": 0.3265,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -2.1754844188690186,
"rewards/margins": 1.7633205652236938,
"rewards/rejected": -3.938805103302002,
"step": 890
},
{
"epoch": 1.88,
"learning_rate": 9.741758728888217e-09,
"logits/chosen": -0.06001782417297363,
"logits/rejected": -0.016363339498639107,
"logps/chosen": -546.7725830078125,
"logps/rejected": -690.0701293945312,
"loss": 0.307,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.229912281036377,
"rewards/margins": 1.9347511529922485,
"rewards/rejected": -4.164663314819336,
"step": 900
},
{
"epoch": 1.88,
"eval_logits/chosen": 0.06628188490867615,
"eval_logits/rejected": 0.02886618673801422,
"eval_logps/chosen": -549.4910278320312,
"eval_logps/rejected": -703.8400268554688,
"eval_loss": 0.5260834097862244,
"eval_rewards/accuracies": 0.77734375,
"eval_rewards/chosen": -2.4590635299682617,
"eval_rewards/margins": 1.4630858898162842,
"eval_rewards/rejected": -3.922149658203125,
"eval_runtime": 65.2404,
"eval_samples_per_second": 30.656,
"eval_steps_per_second": 0.49,
"step": 900
},
{
"epoch": 1.9,
"learning_rate": 6.474868681043577e-09,
"logits/chosen": -0.0006331875920295715,
"logits/rejected": 0.06078674644231796,
"logps/chosen": -526.890869140625,
"logps/rejected": -696.57080078125,
"loss": 0.3133,
"rewards/accuracies": 0.84375,
"rewards/chosen": -2.2198734283447266,
"rewards/margins": 1.9986340999603271,
"rewards/rejected": -4.218507289886475,
"step": 910
},
{
"epoch": 1.93,
"learning_rate": 3.869564046156459e-09,
"logits/chosen": -0.17484715580940247,
"logits/rejected": -0.2491791695356369,
"logps/chosen": -560.9608154296875,
"logps/rejected": -718.45703125,
"loss": 0.3088,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -2.2493414878845215,
"rewards/margins": 2.0530190467834473,
"rewards/rejected": -4.3023600578308105,
"step": 920
},
{
"epoch": 1.95,
"learning_rate": 1.929337314139412e-09,
"logits/chosen": -0.0817941427230835,
"logits/rejected": -0.2032664567232132,
"logps/chosen": -562.2503662109375,
"logps/rejected": -765.2064208984375,
"loss": 0.3264,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -2.2634310722351074,
"rewards/margins": 2.2729239463806152,
"rewards/rejected": -4.536355018615723,
"step": 930
},
{
"epoch": 1.97,
"learning_rate": 6.567894177967325e-10,
"logits/chosen": -0.061837755143642426,
"logits/rejected": -0.09424273669719696,
"logps/chosen": -570.0264892578125,
"logps/rejected": -782.4632568359375,
"loss": 0.3155,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -2.259819984436035,
"rewards/margins": 2.1893467903137207,
"rewards/rejected": -4.449166297912598,
"step": 940
},
{
"epoch": 1.99,
"learning_rate": 5.3626246194704575e-11,
"logits/chosen": 0.019718965515494347,
"logits/rejected": -0.05742845684289932,
"logps/chosen": -491.19879150390625,
"logps/rejected": -698.8577880859375,
"loss": 0.3247,
"rewards/accuracies": 0.8687499761581421,
"rewards/chosen": -2.148132801055908,
"rewards/margins": 2.056044816970825,
"rewards/rejected": -4.204176902770996,
"step": 950
},
{
"epoch": 2.0,
"step": 954,
"total_flos": 0.0,
"train_loss": 0.44779854000739333,
"train_runtime": 8782.9823,
"train_samples_per_second": 13.921,
"train_steps_per_second": 0.109
}
],
"logging_steps": 10,
"max_steps": 954,
"num_train_epochs": 2,
"save_steps": 1000,
"total_flos": 0.0,
"trial_name": null,
"trial_params": null
}