zephyr-7b-dpo-lora-pairrm / trainer_state.json
shenxq's picture
Model save
a1c8240 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9993998799759952,
"eval_steps": 100,
"global_step": 1249,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 4e-08,
"logits/chosen": -2.602003574371338,
"logits/rejected": -2.6217830181121826,
"logps/chosen": -260.71484375,
"logps/rejected": -256.2999267578125,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.01,
"learning_rate": 4.0000000000000003e-07,
"logits/chosen": -2.5351831912994385,
"logits/rejected": -2.56242036819458,
"logps/chosen": -287.53594970703125,
"logps/rejected": -308.8571472167969,
"loss": 0.693,
"rewards/accuracies": 0.4375,
"rewards/chosen": -0.0010520940413698554,
"rewards/margins": 0.0002557992411311716,
"rewards/rejected": -0.0013078933116048574,
"step": 10
},
{
"epoch": 0.02,
"learning_rate": 8.000000000000001e-07,
"logits/chosen": -2.5120716094970703,
"logits/rejected": -2.515523672103882,
"logps/chosen": -255.38192749023438,
"logps/rejected": -273.31805419921875,
"loss": 0.6934,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.0002814549079630524,
"rewards/margins": -0.0005342562217265368,
"rewards/rejected": 0.0002528014301788062,
"step": 20
},
{
"epoch": 0.02,
"learning_rate": 1.2000000000000002e-06,
"logits/chosen": -2.498893976211548,
"logits/rejected": -2.5145130157470703,
"logps/chosen": -276.9407958984375,
"logps/rejected": -295.25482177734375,
"loss": 0.693,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.0016518374904990196,
"rewards/margins": 0.00022423597692977637,
"rewards/rejected": -0.0018760733073577285,
"step": 30
},
{
"epoch": 0.03,
"learning_rate": 1.6000000000000001e-06,
"logits/chosen": -2.5344443321228027,
"logits/rejected": -2.541226625442505,
"logps/chosen": -251.64730834960938,
"logps/rejected": -259.96575927734375,
"loss": 0.6935,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.005051327869296074,
"rewards/margins": -0.0006530196405947208,
"rewards/rejected": -0.004398307763040066,
"step": 40
},
{
"epoch": 0.04,
"learning_rate": 2.0000000000000003e-06,
"logits/chosen": -2.479356288909912,
"logits/rejected": -2.461836576461792,
"logps/chosen": -251.314697265625,
"logps/rejected": -256.02020263671875,
"loss": 0.6931,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.007234510965645313,
"rewards/margins": 0.00020712481637019664,
"rewards/rejected": -0.0074416352435946465,
"step": 50
},
{
"epoch": 0.05,
"learning_rate": 2.4000000000000003e-06,
"logits/chosen": -2.5456085205078125,
"logits/rejected": -2.5527470111846924,
"logps/chosen": -262.48858642578125,
"logps/rejected": -280.4261474609375,
"loss": 0.6927,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.011781791225075722,
"rewards/margins": 0.0009017615811899304,
"rewards/rejected": -0.012683553621172905,
"step": 60
},
{
"epoch": 0.06,
"learning_rate": 2.8000000000000003e-06,
"logits/chosen": -2.545548915863037,
"logits/rejected": -2.5333781242370605,
"logps/chosen": -266.43798828125,
"logps/rejected": -271.582763671875,
"loss": 0.6935,
"rewards/accuracies": 0.4437499940395355,
"rewards/chosen": -0.015757283195853233,
"rewards/margins": -0.0007508035632781684,
"rewards/rejected": -0.015006482601165771,
"step": 70
},
{
"epoch": 0.06,
"learning_rate": 3.2000000000000003e-06,
"logits/chosen": -2.5648863315582275,
"logits/rejected": -2.5693516731262207,
"logps/chosen": -262.5594787597656,
"logps/rejected": -282.7106018066406,
"loss": 0.6925,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.014923855662345886,
"rewards/margins": 0.0014337189495563507,
"rewards/rejected": -0.016357574611902237,
"step": 80
},
{
"epoch": 0.07,
"learning_rate": 3.6000000000000003e-06,
"logits/chosen": -2.565471649169922,
"logits/rejected": -2.5509932041168213,
"logps/chosen": -295.49615478515625,
"logps/rejected": -300.31103515625,
"loss": 0.6911,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.020387709140777588,
"rewards/margins": 0.004231990315020084,
"rewards/rejected": -0.024619702249765396,
"step": 90
},
{
"epoch": 0.08,
"learning_rate": 4.000000000000001e-06,
"logits/chosen": -2.5530660152435303,
"logits/rejected": -2.553136110305786,
"logps/chosen": -263.5738220214844,
"logps/rejected": -276.2574157714844,
"loss": 0.6916,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.02410706877708435,
"rewards/margins": 0.003276412608101964,
"rewards/rejected": -0.027383480221033096,
"step": 100
},
{
"epoch": 0.08,
"eval_logits/chosen": -2.508636713027954,
"eval_logits/rejected": -2.5057852268218994,
"eval_logps/chosen": -210.22964477539062,
"eval_logps/rejected": -215.71873474121094,
"eval_loss": 0.6924654245376587,
"eval_rewards/accuracies": 0.527999997138977,
"eval_rewards/chosen": -0.01618967391550541,
"eval_rewards/margins": 0.0015068231150507927,
"eval_rewards/rejected": -0.017696497961878777,
"eval_runtime": 1581.5421,
"eval_samples_per_second": 1.893,
"eval_steps_per_second": 0.237,
"step": 100
},
{
"epoch": 0.09,
"learning_rate": 4.4e-06,
"logits/chosen": -2.55187726020813,
"logits/rejected": -2.5549445152282715,
"logps/chosen": -283.41143798828125,
"logps/rejected": -293.6915588378906,
"loss": 0.6912,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.026097621768712997,
"rewards/margins": 0.004130261484533548,
"rewards/rejected": -0.030227884650230408,
"step": 110
},
{
"epoch": 0.1,
"learning_rate": 4.800000000000001e-06,
"logits/chosen": -2.561483860015869,
"logits/rejected": -2.575721263885498,
"logps/chosen": -265.60711669921875,
"logps/rejected": -282.2021789550781,
"loss": 0.691,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.030240505933761597,
"rewards/margins": 0.004798956215381622,
"rewards/rejected": -0.03503946214914322,
"step": 120
},
{
"epoch": 0.1,
"learning_rate": 4.999755876225375e-06,
"logits/chosen": -2.5272843837738037,
"logits/rejected": -2.511995792388916,
"logps/chosen": -279.7336120605469,
"logps/rejected": -298.2309875488281,
"loss": 0.6886,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.03772541880607605,
"rewards/margins": 0.009567076340317726,
"rewards/rejected": -0.047292497009038925,
"step": 130
},
{
"epoch": 0.11,
"learning_rate": 4.997803172081864e-06,
"logits/chosen": -2.549031972885132,
"logits/rejected": -2.550044536590576,
"logps/chosen": -274.913818359375,
"logps/rejected": -287.93389892578125,
"loss": 0.6866,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.0441165529191494,
"rewards/margins": 0.01394923496991396,
"rewards/rejected": -0.058065783232450485,
"step": 140
},
{
"epoch": 0.12,
"learning_rate": 4.9938992891651825e-06,
"logits/chosen": -2.511295795440674,
"logits/rejected": -2.5027806758880615,
"logps/chosen": -263.2046203613281,
"logps/rejected": -284.345703125,
"loss": 0.6827,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.05038716644048691,
"rewards/margins": 0.021908113732933998,
"rewards/rejected": -0.07229527831077576,
"step": 150
},
{
"epoch": 0.13,
"learning_rate": 4.988047277024456e-06,
"logits/chosen": -2.551429271697998,
"logits/rejected": -2.5654749870300293,
"logps/chosen": -274.80670166015625,
"logps/rejected": -288.6151428222656,
"loss": 0.682,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.07800472527742386,
"rewards/margins": 0.023449204862117767,
"rewards/rejected": -0.10145392268896103,
"step": 160
},
{
"epoch": 0.14,
"learning_rate": 4.980251707005417e-06,
"logits/chosen": -2.535489320755005,
"logits/rejected": -2.511200428009033,
"logps/chosen": -291.24853515625,
"logps/rejected": -298.6773376464844,
"loss": 0.6896,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.10200835764408112,
"rewards/margins": 0.008326217532157898,
"rewards/rejected": -0.11033459007740021,
"step": 170
},
{
"epoch": 0.14,
"learning_rate": 4.970518668679459e-06,
"logits/chosen": -2.540640354156494,
"logits/rejected": -2.5266213417053223,
"logps/chosen": -289.3337707519531,
"logps/rejected": -295.79278564453125,
"loss": 0.6841,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.09090762585401535,
"rewards/margins": 0.020125988870859146,
"rewards/rejected": -0.1110336035490036,
"step": 180
},
{
"epoch": 0.15,
"learning_rate": 4.958855765086722e-06,
"logits/chosen": -2.563791036605835,
"logits/rejected": -2.562145233154297,
"logps/chosen": -269.8388671875,
"logps/rejected": -280.38037109375,
"loss": 0.6853,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.05399826169013977,
"rewards/margins": 0.017181050032377243,
"rewards/rejected": -0.07117930799722672,
"step": 190
},
{
"epoch": 0.16,
"learning_rate": 4.945272106796919e-06,
"logits/chosen": -2.5541298389434814,
"logits/rejected": -2.564225673675537,
"logps/chosen": -269.95001220703125,
"logps/rejected": -285.06451416015625,
"loss": 0.6855,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.06484891474246979,
"rewards/margins": 0.017776602879166603,
"rewards/rejected": -0.08262552320957184,
"step": 200
},
{
"epoch": 0.16,
"eval_logits/chosen": -2.517770528793335,
"eval_logits/rejected": -2.515197277069092,
"eval_logps/chosen": -215.1240234375,
"eval_logps/rejected": -221.67100524902344,
"eval_loss": 0.6880091428756714,
"eval_rewards/accuracies": 0.5613333582878113,
"eval_rewards/chosen": -0.06513328850269318,
"eval_rewards/margins": 0.012086095288395882,
"eval_rewards/rejected": -0.07721938192844391,
"eval_runtime": 1599.371,
"eval_samples_per_second": 1.872,
"eval_steps_per_second": 0.234,
"step": 200
},
{
"epoch": 0.17,
"learning_rate": 4.929778304792537e-06,
"logits/chosen": -2.5422654151916504,
"logits/rejected": -2.5487327575683594,
"logps/chosen": -294.87188720703125,
"logps/rejected": -299.1329040527344,
"loss": 0.6792,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.08577840030193329,
"rewards/margins": 0.03281060978770256,
"rewards/rejected": -0.11858902126550674,
"step": 210
},
{
"epoch": 0.18,
"learning_rate": 4.912386462179987e-06,
"logits/chosen": -2.56231951713562,
"logits/rejected": -2.559842348098755,
"logps/chosen": -283.178955078125,
"logps/rejected": -308.49322509765625,
"loss": 0.6772,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.09053866565227509,
"rewards/margins": 0.03568854182958603,
"rewards/rejected": -0.1262272149324417,
"step": 220
},
{
"epoch": 0.18,
"learning_rate": 4.893110164735167e-06,
"logits/chosen": -2.5928561687469482,
"logits/rejected": -2.5934693813323975,
"logps/chosen": -288.0644226074219,
"logps/rejected": -301.1351013183594,
"loss": 0.6831,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.11280493438243866,
"rewards/margins": 0.023493468761444092,
"rewards/rejected": -0.13629841804504395,
"step": 230
},
{
"epoch": 0.19,
"learning_rate": 4.871964470290823e-06,
"logits/chosen": -2.532960891723633,
"logits/rejected": -2.55849027633667,
"logps/chosen": -289.32354736328125,
"logps/rejected": -309.2322692871094,
"loss": 0.6735,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.11586654186248779,
"rewards/margins": 0.04449574649333954,
"rewards/rejected": -0.16036227345466614,
"step": 240
},
{
"epoch": 0.2,
"learning_rate": 4.848965896974006e-06,
"logits/chosen": -2.541764259338379,
"logits/rejected": -2.545576572418213,
"logps/chosen": -281.2950439453125,
"logps/rejected": -300.70416259765625,
"loss": 0.6803,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.11654965579509735,
"rewards/margins": 0.02992970123887062,
"rewards/rejected": -0.14647935330867767,
"step": 250
},
{
"epoch": 0.21,
"learning_rate": 4.8241324103028055e-06,
"logits/chosen": -2.5795302391052246,
"logits/rejected": -2.5413551330566406,
"logps/chosen": -287.0888977050781,
"logps/rejected": -298.43524169921875,
"loss": 0.6762,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.1340574324131012,
"rewards/margins": 0.03809996321797371,
"rewards/rejected": -0.1721573919057846,
"step": 260
},
{
"epoch": 0.22,
"learning_rate": 4.797483409152438e-06,
"logits/chosen": -2.565472364425659,
"logits/rejected": -2.569817066192627,
"logps/chosen": -276.9258728027344,
"logps/rejected": -297.29931640625,
"loss": 0.668,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.14538586139678955,
"rewards/margins": 0.05918685719370842,
"rewards/rejected": -0.20457272231578827,
"step": 270
},
{
"epoch": 0.22,
"learning_rate": 4.769039710601669e-06,
"logits/chosen": -2.5913045406341553,
"logits/rejected": -2.592714786529541,
"logps/chosen": -277.0105285644531,
"logps/rejected": -293.9090576171875,
"loss": 0.678,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.16800367832183838,
"rewards/margins": 0.03504541888833046,
"rewards/rejected": -0.20304909348487854,
"step": 280
},
{
"epoch": 0.23,
"learning_rate": 4.738823533671383e-06,
"logits/chosen": -2.59189510345459,
"logits/rejected": -2.5966787338256836,
"logps/chosen": -300.8684387207031,
"logps/rejected": -316.1703186035156,
"loss": 0.677,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.18113958835601807,
"rewards/margins": 0.0392177477478981,
"rewards/rejected": -0.22035734355449677,
"step": 290
},
{
"epoch": 0.24,
"learning_rate": 4.706858481968017e-06,
"logits/chosen": -2.5426743030548096,
"logits/rejected": -2.546731472015381,
"logps/chosen": -288.5105895996094,
"logps/rejected": -297.8955993652344,
"loss": 0.6825,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.22026526927947998,
"rewards/margins": 0.027653247117996216,
"rewards/rejected": -0.2479185312986374,
"step": 300
},
{
"epoch": 0.24,
"eval_logits/chosen": -2.519216537475586,
"eval_logits/rejected": -2.517465591430664,
"eval_logps/chosen": -227.3457489013672,
"eval_logps/rejected": -234.75460815429688,
"eval_loss": 0.6853793859481812,
"eval_rewards/accuracies": 0.5473333597183228,
"eval_rewards/chosen": -0.18735063076019287,
"eval_rewards/margins": 0.02070464752614498,
"eval_rewards/rejected": -0.2080552726984024,
"eval_runtime": 1584.577,
"eval_samples_per_second": 1.889,
"eval_steps_per_second": 0.237,
"step": 300
},
{
"epoch": 0.25,
"learning_rate": 4.673169525245416e-06,
"logits/chosen": -2.534403085708618,
"logits/rejected": -2.514641284942627,
"logps/chosen": -286.3177185058594,
"logps/rejected": -313.84521484375,
"loss": 0.6698,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.19986188411712646,
"rewards/margins": 0.05483303219079971,
"rewards/rejected": -0.2546949088573456,
"step": 310
},
{
"epoch": 0.26,
"learning_rate": 4.63778297989952e-06,
"logits/chosen": -2.5624048709869385,
"logits/rejected": -2.54682993888855,
"logps/chosen": -292.1237487792969,
"logps/rejected": -307.5107116699219,
"loss": 0.6719,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.2259463369846344,
"rewards/margins": 0.05093250423669815,
"rewards/rejected": -0.27687883377075195,
"step": 320
},
{
"epoch": 0.26,
"learning_rate": 4.60072648841109e-06,
"logits/chosen": -2.540750741958618,
"logits/rejected": -2.536940574645996,
"logps/chosen": -302.9012756347656,
"logps/rejected": -322.0403747558594,
"loss": 0.6561,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.25178760290145874,
"rewards/margins": 0.08651792258024216,
"rewards/rejected": -0.3383055329322815,
"step": 330
},
{
"epoch": 0.27,
"learning_rate": 4.562028997752574e-06,
"logits/chosen": -2.5020062923431396,
"logits/rejected": -2.525498867034912,
"logps/chosen": -299.34295654296875,
"logps/rejected": -324.51239013671875,
"loss": 0.6697,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.22524788975715637,
"rewards/margins": 0.05736035853624344,
"rewards/rejected": -0.2826082408428192,
"step": 340
},
{
"epoch": 0.28,
"learning_rate": 4.521720736775947e-06,
"logits/chosen": -2.551377534866333,
"logits/rejected": -2.549631118774414,
"logps/chosen": -312.30010986328125,
"logps/rejected": -318.33203125,
"loss": 0.6713,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.31575754284858704,
"rewards/margins": 0.05324278026819229,
"rewards/rejected": -0.36900031566619873,
"step": 350
},
{
"epoch": 0.29,
"learning_rate": 4.479833192599198e-06,
"logits/chosen": -2.6122488975524902,
"logits/rejected": -2.597867488861084,
"logps/chosen": -310.14825439453125,
"logps/rejected": -329.4725646972656,
"loss": 0.6601,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.26400071382522583,
"rewards/margins": 0.07782687991857529,
"rewards/rejected": -0.3418275713920593,
"step": 360
},
{
"epoch": 0.3,
"learning_rate": 4.436399086009928e-06,
"logits/chosen": -2.592072010040283,
"logits/rejected": -2.582606554031372,
"logps/chosen": -298.0146789550781,
"logps/rejected": -312.88323974609375,
"loss": 0.6639,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.24674613773822784,
"rewards/margins": 0.06771402060985565,
"rewards/rejected": -0.3144601583480835,
"step": 370
},
{
"epoch": 0.3,
"learning_rate": 4.391452345905239e-06,
"logits/chosen": -2.501129150390625,
"logits/rejected": -2.5144436359405518,
"logps/chosen": -302.98785400390625,
"logps/rejected": -315.341552734375,
"loss": 0.6643,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.2700803875923157,
"rewards/margins": 0.07336559146642685,
"rewards/rejected": -0.3434459865093231,
"step": 380
},
{
"epoch": 0.31,
"learning_rate": 4.3450280827879125e-06,
"logits/chosen": -2.5604705810546875,
"logits/rejected": -2.5804944038391113,
"logps/chosen": -301.7215881347656,
"logps/rejected": -316.622314453125,
"loss": 0.6673,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.2707824110984802,
"rewards/margins": 0.06462465226650238,
"rewards/rejected": -0.3354070484638214,
"step": 390
},
{
"epoch": 0.32,
"learning_rate": 4.297162561339554e-06,
"logits/chosen": -2.5028789043426514,
"logits/rejected": -2.5059926509857178,
"logps/chosen": -308.93524169921875,
"logps/rejected": -328.21649169921875,
"loss": 0.6676,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.30648913979530334,
"rewards/margins": 0.0709182545542717,
"rewards/rejected": -0.37740740180015564,
"step": 400
},
{
"epoch": 0.32,
"eval_logits/chosen": -2.536813497543335,
"eval_logits/rejected": -2.534651756286621,
"eval_logps/chosen": -237.7042236328125,
"eval_logps/rejected": -246.16815185546875,
"eval_loss": 0.68265300989151,
"eval_rewards/accuracies": 0.5476666688919067,
"eval_rewards/chosen": -0.2909356355667114,
"eval_rewards/margins": 0.031255174428224564,
"eval_rewards/rejected": -0.3221907913684845,
"eval_runtime": 1589.7925,
"eval_samples_per_second": 1.883,
"eval_steps_per_second": 0.236,
"step": 400
},
{
"epoch": 0.33,
"learning_rate": 4.247893172092157e-06,
"logits/chosen": -2.531207799911499,
"logits/rejected": -2.548520565032959,
"logps/chosen": -297.398681640625,
"logps/rejected": -326.1677551269531,
"loss": 0.6617,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.313217431306839,
"rewards/margins": 0.08246447145938873,
"rewards/rejected": -0.3956819176673889,
"step": 410
},
{
"epoch": 0.34,
"learning_rate": 4.197258402220187e-06,
"logits/chosen": -2.5356483459472656,
"logits/rejected": -2.544675350189209,
"logps/chosen": -305.0682373046875,
"logps/rejected": -318.51068115234375,
"loss": 0.6603,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.31003984808921814,
"rewards/margins": 0.08686444163322449,
"rewards/rejected": -0.39690425992012024,
"step": 420
},
{
"epoch": 0.34,
"learning_rate": 4.145297805476023e-06,
"logits/chosen": -2.5800435543060303,
"logits/rejected": -2.580294132232666,
"logps/chosen": -302.9384765625,
"logps/rejected": -323.58026123046875,
"loss": 0.6537,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.31787100434303284,
"rewards/margins": 0.09609754383563995,
"rewards/rejected": -0.413968563079834,
"step": 430
},
{
"epoch": 0.35,
"learning_rate": 4.092051971292228e-06,
"logits/chosen": -2.5852980613708496,
"logits/rejected": -2.595292806625366,
"logps/chosen": -300.8592224121094,
"logps/rejected": -319.3016662597656,
"loss": 0.6657,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.3366487920284271,
"rewards/margins": 0.07542940974235535,
"rewards/rejected": -0.41207820177078247,
"step": 440
},
{
"epoch": 0.36,
"learning_rate": 4.037562493074792e-06,
"logits/chosen": -2.5786662101745605,
"logits/rejected": -2.5908896923065186,
"logps/chosen": -317.9658203125,
"logps/rejected": -327.48614501953125,
"loss": 0.6589,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.33508211374282837,
"rewards/margins": 0.08991469442844391,
"rewards/rejected": -0.4249967634677887,
"step": 450
},
{
"epoch": 0.37,
"learning_rate": 3.981871935712112e-06,
"logits/chosen": -2.6687827110290527,
"logits/rejected": -2.6695172786712646,
"logps/chosen": -296.9199523925781,
"logps/rejected": -318.85943603515625,
"loss": 0.666,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.2961326241493225,
"rewards/margins": 0.07453998178243637,
"rewards/rejected": -0.3706726133823395,
"step": 460
},
{
"epoch": 0.38,
"learning_rate": 3.925023802325094e-06,
"logits/chosen": -2.6613316535949707,
"logits/rejected": -2.6546504497528076,
"logps/chosen": -305.7056884765625,
"logps/rejected": -325.8073425292969,
"loss": 0.6564,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.2978198826313019,
"rewards/margins": 0.09947486966848373,
"rewards/rejected": -0.3972947299480438,
"step": 470
},
{
"epoch": 0.38,
"learning_rate": 3.867062500284342e-06,
"logits/chosen": -2.6371190547943115,
"logits/rejected": -2.631121873855591,
"logps/chosen": -297.20257568359375,
"logps/rejected": -324.0657653808594,
"loss": 0.6543,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.36376848816871643,
"rewards/margins": 0.09993237257003784,
"rewards/rejected": -0.4637008607387543,
"step": 480
},
{
"epoch": 0.39,
"learning_rate": 3.8080333065209885e-06,
"logits/chosen": -2.6479365825653076,
"logits/rejected": -2.646528720855713,
"logps/chosen": -306.7449035644531,
"logps/rejected": -303.7771301269531,
"loss": 0.6912,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.41228657960891724,
"rewards/margins": 0.029567983001470566,
"rewards/rejected": -0.4418545663356781,
"step": 490
},
{
"epoch": 0.4,
"learning_rate": 3.7479823321582624e-06,
"logits/chosen": -2.632742404937744,
"logits/rejected": -2.633615016937256,
"logps/chosen": -306.31195068359375,
"logps/rejected": -336.17388916015625,
"loss": 0.6458,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.34735265374183655,
"rewards/margins": 0.11751687526702881,
"rewards/rejected": -0.46486955881118774,
"step": 500
},
{
"epoch": 0.4,
"eval_logits/chosen": -2.6363847255706787,
"eval_logits/rejected": -2.6328208446502686,
"eval_logps/chosen": -245.54345703125,
"eval_logps/rejected": -254.98521423339844,
"eval_loss": 0.6804653406143188,
"eval_rewards/accuracies": 0.5566666722297668,
"eval_rewards/chosen": -0.3693278133869171,
"eval_rewards/margins": 0.0410337969660759,
"eval_rewards/rejected": -0.41036155819892883,
"eval_runtime": 1591.6944,
"eval_samples_per_second": 1.881,
"eval_steps_per_second": 0.236,
"step": 500
},
{
"epoch": 0.41,
"learning_rate": 3.686956486491419e-06,
"logits/chosen": -2.6169426441192627,
"logits/rejected": -2.6043953895568848,
"logps/chosen": -318.00299072265625,
"logps/rejected": -336.43048095703125,
"loss": 0.6546,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.37818408012390137,
"rewards/margins": 0.10718680918216705,
"rewards/rejected": -0.485370934009552,
"step": 510
},
{
"epoch": 0.42,
"learning_rate": 3.625003440344166e-06,
"logits/chosen": -2.638087034225464,
"logits/rejected": -2.6651127338409424,
"logps/chosen": -298.8196105957031,
"logps/rejected": -309.6526794433594,
"loss": 0.6669,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.3525494933128357,
"rewards/margins": 0.06792327761650085,
"rewards/rejected": -0.42047280073165894,
"step": 520
},
{
"epoch": 0.42,
"learning_rate": 3.562171588830231e-06,
"logits/chosen": -2.682291269302368,
"logits/rejected": -2.690966844558716,
"logps/chosen": -308.031005859375,
"logps/rejected": -333.34600830078125,
"loss": 0.664,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.3355502486228943,
"rewards/margins": 0.08054343611001968,
"rewards/rejected": -0.416093647480011,
"step": 530
},
{
"epoch": 0.43,
"learning_rate": 3.4985100135491245e-06,
"logits/chosen": -2.6738035678863525,
"logits/rejected": -2.665532112121582,
"logps/chosen": -317.4346008300781,
"logps/rejected": -350.239990234375,
"loss": 0.6401,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.3944160044193268,
"rewards/margins": 0.13535484671592712,
"rewards/rejected": -0.5297707915306091,
"step": 540
},
{
"epoch": 0.44,
"learning_rate": 3.4340684442456673e-06,
"logits/chosen": -2.634321689605713,
"logits/rejected": -2.6402688026428223,
"logps/chosen": -321.502197265625,
"logps/rejected": -339.1592712402344,
"loss": 0.6613,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.47532662749290466,
"rewards/margins": 0.08404591679573059,
"rewards/rejected": -0.5593725442886353,
"step": 550
},
{
"epoch": 0.45,
"learning_rate": 3.3688972199631974e-06,
"logits/chosen": -2.6393518447875977,
"logits/rejected": -2.6311287879943848,
"logps/chosen": -318.1193542480469,
"logps/rejected": -333.4119873046875,
"loss": 0.6462,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.45164427161216736,
"rewards/margins": 0.13988950848579407,
"rewards/rejected": -0.5915338397026062,
"step": 560
},
{
"epoch": 0.46,
"learning_rate": 3.3030472497208354e-06,
"logits/chosen": -2.6581737995147705,
"logits/rejected": -2.6506381034851074,
"logps/chosen": -315.7310485839844,
"logps/rejected": -369.3522033691406,
"loss": 0.6318,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.5240504741668701,
"rewards/margins": 0.17177370190620422,
"rewards/rejected": -0.695824146270752,
"step": 570
},
{
"epoch": 0.46,
"learning_rate": 3.236569972745492e-06,
"logits/chosen": -2.6991753578186035,
"logits/rejected": -2.6973986625671387,
"logps/chosen": -313.09747314453125,
"logps/rejected": -330.6873779296875,
"loss": 0.6638,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.5288428068161011,
"rewards/margins": 0.08699531853199005,
"rewards/rejected": -0.6158381700515747,
"step": 580
},
{
"epoch": 0.47,
"learning_rate": 3.1695173182897126e-06,
"logits/chosen": -2.622006893157959,
"logits/rejected": -2.6231589317321777,
"logps/chosen": -323.0411376953125,
"logps/rejected": -351.798583984375,
"loss": 0.657,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.5726518034934998,
"rewards/margins": 0.10712896287441254,
"rewards/rejected": -0.6797807812690735,
"step": 590
},
{
"epoch": 0.48,
"learning_rate": 3.10194166506673e-06,
"logits/chosen": -2.637044906616211,
"logits/rejected": -2.6247122287750244,
"logps/chosen": -323.566650390625,
"logps/rejected": -354.9677429199219,
"loss": 0.6592,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5322225093841553,
"rewards/margins": 0.10880555957555771,
"rewards/rejected": -0.6410280466079712,
"step": 600
},
{
"epoch": 0.48,
"eval_logits/chosen": -2.6844964027404785,
"eval_logits/rejected": -2.6804616451263428,
"eval_logps/chosen": -268.708740234375,
"eval_logps/rejected": -279.2277526855469,
"eval_loss": 0.6789478063583374,
"eval_rewards/accuracies": 0.5559999942779541,
"eval_rewards/chosen": -0.6009806990623474,
"eval_rewards/margins": 0.05180613696575165,
"eval_rewards/rejected": -0.6527868509292603,
"eval_runtime": 1598.2559,
"eval_samples_per_second": 1.873,
"eval_steps_per_second": 0.235,
"step": 600
},
{
"epoch": 0.49,
"learning_rate": 3.0338958003344115e-06,
"logits/chosen": -2.703444004058838,
"logits/rejected": -2.6941075325012207,
"logps/chosen": -320.3475341796875,
"logps/rejected": -349.988037109375,
"loss": 0.6418,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5905410051345825,
"rewards/margins": 0.1995875984430313,
"rewards/rejected": -0.7901285886764526,
"step": 610
},
{
"epoch": 0.5,
"learning_rate": 2.9654328786600823e-06,
"logits/chosen": -2.657546281814575,
"logits/rejected": -2.6613922119140625,
"logps/chosen": -323.674560546875,
"logps/rejected": -357.7910461425781,
"loss": 0.6488,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6175565123558044,
"rewards/margins": 0.11701612174510956,
"rewards/rejected": -0.7345726490020752,
"step": 620
},
{
"epoch": 0.5,
"learning_rate": 2.896606380398402e-06,
"logits/chosen": -2.6770036220550537,
"logits/rejected": -2.697390079498291,
"logps/chosen": -333.11932373046875,
"logps/rejected": -353.01739501953125,
"loss": 0.6521,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.6429452896118164,
"rewards/margins": 0.12399964034557343,
"rewards/rejected": -0.766944944858551,
"step": 630
},
{
"epoch": 0.51,
"learning_rate": 2.827470069914772e-06,
"logits/chosen": -2.674020528793335,
"logits/rejected": -2.660238265991211,
"logps/chosen": -336.7355041503906,
"logps/rejected": -360.3841247558594,
"loss": 0.66,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.6630152463912964,
"rewards/margins": 0.10159214586019516,
"rewards/rejected": -0.7646073698997498,
"step": 640
},
{
"epoch": 0.52,
"learning_rate": 2.7580779535868675e-06,
"logits/chosen": -2.641157388687134,
"logits/rejected": -2.65820574760437,
"logps/chosen": -333.5285339355469,
"logps/rejected": -352.81988525390625,
"loss": 0.6624,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.617414116859436,
"rewards/margins": 0.1025906577706337,
"rewards/rejected": -0.7200047373771667,
"step": 650
},
{
"epoch": 0.53,
"learning_rate": 2.688484237617129e-06,
"logits/chosen": -2.6740407943725586,
"logits/rejected": -2.6733462810516357,
"logps/chosen": -334.4920654296875,
"logps/rejected": -359.688720703125,
"loss": 0.6463,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6058083772659302,
"rewards/margins": 0.14193257689476013,
"rewards/rejected": -0.7477409243583679,
"step": 660
},
{
"epoch": 0.54,
"learning_rate": 2.6187432856891585e-06,
"logits/chosen": -2.677578926086426,
"logits/rejected": -2.694331169128418,
"logps/chosen": -342.4553527832031,
"logps/rejected": -371.22515869140625,
"loss": 0.6558,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6532286405563354,
"rewards/margins": 0.11075172573328018,
"rewards/rejected": -0.7639803290367126,
"step": 670
},
{
"epoch": 0.54,
"learning_rate": 2.548909576501096e-06,
"logits/chosen": -2.7143263816833496,
"logits/rejected": -2.7066876888275146,
"logps/chosen": -340.72137451171875,
"logps/rejected": -362.9776306152344,
"loss": 0.6509,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.6533491015434265,
"rewards/margins": 0.12043479830026627,
"rewards/rejected": -0.7737839818000793,
"step": 680
},
{
"epoch": 0.55,
"learning_rate": 2.4790376612091503e-06,
"logits/chosen": -2.726318359375,
"logits/rejected": -2.720698833465576,
"logps/chosen": -359.000732421875,
"logps/rejected": -382.2019958496094,
"loss": 0.6319,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.7622326612472534,
"rewards/margins": 0.1633436232805252,
"rewards/rejected": -0.9255763292312622,
"step": 690
},
{
"epoch": 0.56,
"learning_rate": 2.40918212081453e-06,
"logits/chosen": -2.7099740505218506,
"logits/rejected": -2.6980385780334473,
"logps/chosen": -330.88818359375,
"logps/rejected": -372.46075439453125,
"loss": 0.6107,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.739021897315979,
"rewards/margins": 0.22892406582832336,
"rewards/rejected": -0.9679458737373352,
"step": 700
},
{
"epoch": 0.56,
"eval_logits/chosen": -2.696856737136841,
"eval_logits/rejected": -2.691408634185791,
"eval_logps/chosen": -290.1963806152344,
"eval_logps/rejected": -301.8047180175781,
"eval_loss": 0.6785151362419128,
"eval_rewards/accuracies": 0.5550000071525574,
"eval_rewards/chosen": -0.8158571124076843,
"eval_rewards/margins": 0.06269912421703339,
"eval_rewards/rejected": -0.8785561323165894,
"eval_runtime": 1591.1417,
"eval_samples_per_second": 1.882,
"eval_steps_per_second": 0.236,
"step": 700
},
{
"epoch": 0.57,
"learning_rate": 2.3393975235270654e-06,
"logits/chosen": -2.684086322784424,
"logits/rejected": -2.68058443069458,
"logps/chosen": -360.47100830078125,
"logps/rejected": -390.83441162109375,
"loss": 0.6495,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7877064943313599,
"rewards/margins": 0.1392781138420105,
"rewards/rejected": -0.9269846677780151,
"step": 710
},
{
"epoch": 0.58,
"learning_rate": 2.2697383821388153e-06,
"logits/chosen": -2.7011210918426514,
"logits/rejected": -2.688472270965576,
"logps/chosen": -351.14263916015625,
"logps/rejected": -368.4684143066406,
"loss": 0.6541,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.7785929441452026,
"rewards/margins": 0.14113947749137878,
"rewards/rejected": -0.919732391834259,
"step": 720
},
{
"epoch": 0.58,
"learning_rate": 2.2002591114409657e-06,
"logits/chosen": -2.693358898162842,
"logits/rejected": -2.6756577491760254,
"logps/chosen": -362.7990417480469,
"logps/rejected": -383.33843994140625,
"loss": 0.6601,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.8580299615859985,
"rewards/margins": 0.10724015533924103,
"rewards/rejected": -0.9652700424194336,
"step": 730
},
{
"epoch": 0.59,
"learning_rate": 2.131013985717285e-06,
"logits/chosen": -2.7266604900360107,
"logits/rejected": -2.6952764987945557,
"logps/chosen": -370.53106689453125,
"logps/rejected": -411.46875,
"loss": 0.6317,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.8816037178039551,
"rewards/margins": 0.18020646274089813,
"rewards/rejected": -1.0618102550506592,
"step": 740
},
{
"epoch": 0.6,
"learning_rate": 2.062057096347338e-06,
"logits/chosen": -2.737968921661377,
"logits/rejected": -2.733065605163574,
"logps/chosen": -361.11993408203125,
"logps/rejected": -372.26513671875,
"loss": 0.6619,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.8483070135116577,
"rewards/margins": 0.1061801165342331,
"rewards/rejected": -0.954487144947052,
"step": 750
},
{
"epoch": 0.61,
"learning_rate": 1.9934423095525733e-06,
"logits/chosen": -2.691333055496216,
"logits/rejected": -2.6811554431915283,
"logps/chosen": -362.3067932128906,
"logps/rejected": -375.4438171386719,
"loss": 0.6545,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.7950974702835083,
"rewards/margins": 0.12981590628623962,
"rewards/rejected": -0.9249132871627808,
"step": 760
},
{
"epoch": 0.62,
"learning_rate": 1.9252232243182986e-06,
"logits/chosen": -2.7668118476867676,
"logits/rejected": -2.7533440589904785,
"logps/chosen": -333.151611328125,
"logps/rejected": -369.8766784667969,
"loss": 0.6235,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7605150938034058,
"rewards/margins": 0.20025065541267395,
"rewards/rejected": -0.9607658386230469,
"step": 770
},
{
"epoch": 0.62,
"learning_rate": 1.8574531305244043e-06,
"logits/chosen": -2.650963068008423,
"logits/rejected": -2.6607377529144287,
"logps/chosen": -363.749267578125,
"logps/rejected": -396.4261169433594,
"loss": 0.6247,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.829413115978241,
"rewards/margins": 0.19826601445674896,
"rewards/rejected": -1.0276790857315063,
"step": 780
},
{
"epoch": 0.63,
"learning_rate": 1.7901849673175559e-06,
"logits/chosen": -2.719726085662842,
"logits/rejected": -2.7071306705474854,
"logps/chosen": -357.9602355957031,
"logps/rejected": -386.6010437011719,
"loss": 0.6424,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.7654613852500916,
"rewards/margins": 0.14790861308574677,
"rewards/rejected": -0.9133701324462891,
"step": 790
},
{
"epoch": 0.64,
"learning_rate": 1.7234712817573555e-06,
"logits/chosen": -2.7113006114959717,
"logits/rejected": -2.6781070232391357,
"logps/chosen": -382.4976806640625,
"logps/rejected": -398.0575256347656,
"loss": 0.6475,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.8575699925422668,
"rewards/margins": 0.13993313908576965,
"rewards/rejected": -0.9975031614303589,
"step": 800
},
{
"epoch": 0.64,
"eval_logits/chosen": -2.729461669921875,
"eval_logits/rejected": -2.723686933517456,
"eval_logps/chosen": -297.0627136230469,
"eval_logps/rejected": -309.38665771484375,
"eval_loss": 0.6770467162132263,
"eval_rewards/accuracies": 0.5609999895095825,
"eval_rewards/chosen": -0.8845203518867493,
"eval_rewards/margins": 0.06985554099082947,
"eval_rewards/rejected": -0.9543759226799011,
"eval_runtime": 1591.3996,
"eval_samples_per_second": 1.881,
"eval_steps_per_second": 0.236,
"step": 800
},
{
"epoch": 0.65,
"learning_rate": 1.6573641877687936e-06,
"logits/chosen": -2.712907314300537,
"logits/rejected": -2.712637424468994,
"logps/chosen": -359.29473876953125,
"logps/rejected": -395.03173828125,
"loss": 0.6372,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.8719032406806946,
"rewards/margins": 0.15319783985614777,
"rewards/rejected": -1.0251010656356812,
"step": 810
},
{
"epoch": 0.66,
"learning_rate": 1.591915325433034e-06,
"logits/chosen": -2.771751880645752,
"logits/rejected": -2.764472246170044,
"logps/chosen": -348.4260559082031,
"logps/rejected": -368.5625305175781,
"loss": 0.6422,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.8010624647140503,
"rewards/margins": 0.15961696207523346,
"rewards/rejected": -0.9606794118881226,
"step": 820
},
{
"epoch": 0.66,
"learning_rate": 1.5271758206483664e-06,
"logits/chosen": -2.740337371826172,
"logits/rejected": -2.732513666152954,
"logps/chosen": -366.95660400390625,
"logps/rejected": -389.9357604980469,
"loss": 0.6581,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.8896878361701965,
"rewards/margins": 0.12456034123897552,
"rewards/rejected": -1.014248251914978,
"step": 830
},
{
"epoch": 0.67,
"learning_rate": 1.4631962451927966e-06,
"logits/chosen": -2.6755502223968506,
"logits/rejected": -2.6818313598632812,
"logps/chosen": -362.2518005371094,
"logps/rejected": -391.2507019042969,
"loss": 0.6378,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.8435665369033813,
"rewards/margins": 0.16070519387722015,
"rewards/rejected": -1.0042717456817627,
"step": 840
},
{
"epoch": 0.68,
"learning_rate": 1.4000265772195032e-06,
"logits/chosen": -2.792430877685547,
"logits/rejected": -2.7686753273010254,
"logps/chosen": -358.8890075683594,
"logps/rejected": -395.0196838378906,
"loss": 0.6195,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.8025849461555481,
"rewards/margins": 0.2058737725019455,
"rewards/rejected": -1.0084587335586548,
"step": 850
},
{
"epoch": 0.69,
"learning_rate": 1.3377161622160077e-06,
"logits/chosen": -2.709834337234497,
"logits/rejected": -2.7066776752471924,
"logps/chosen": -356.4862365722656,
"logps/rejected": -385.40625,
"loss": 0.6288,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.8100967407226562,
"rewards/margins": 0.16853158175945282,
"rewards/rejected": -0.9786283373832703,
"step": 860
},
{
"epoch": 0.7,
"learning_rate": 1.276313674457553e-06,
"logits/chosen": -2.737943172454834,
"logits/rejected": -2.74971079826355,
"logps/chosen": -342.73614501953125,
"logps/rejected": -379.23126220703125,
"loss": 0.6226,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.8370000123977661,
"rewards/margins": 0.18809911608695984,
"rewards/rejected": -1.0250990390777588,
"step": 870
},
{
"epoch": 0.7,
"learning_rate": 1.2158670789848095e-06,
"logits/chosen": -2.713247537612915,
"logits/rejected": -2.6928553581237793,
"logps/chosen": -373.70361328125,
"logps/rejected": -397.2867431640625,
"loss": 0.6407,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.9539483189582825,
"rewards/margins": 0.16210444271564484,
"rewards/rejected": -1.1160527467727661,
"step": 880
},
{
"epoch": 0.71,
"learning_rate": 1.1564235941356016e-06,
"logits/chosen": -2.7370991706848145,
"logits/rejected": -2.7364439964294434,
"logps/chosen": -360.29534912109375,
"logps/rejected": -403.6412048339844,
"loss": 0.6203,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.97996985912323,
"rewards/margins": 0.23149892687797546,
"rewards/rejected": -1.2114689350128174,
"step": 890
},
{
"epoch": 0.72,
"learning_rate": 1.0980296546599254e-06,
"logits/chosen": -2.6951401233673096,
"logits/rejected": -2.6780078411102295,
"logps/chosen": -381.99420166015625,
"logps/rejected": -393.02215576171875,
"loss": 0.6639,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.9827243685722351,
"rewards/margins": 0.15567822754383087,
"rewards/rejected": -1.1384025812149048,
"step": 900
},
{
"epoch": 0.72,
"eval_logits/chosen": -2.752532482147217,
"eval_logits/rejected": -2.7463557720184326,
"eval_logps/chosen": -305.6557922363281,
"eval_logps/rejected": -318.45068359375,
"eval_loss": 0.6765736937522888,
"eval_rewards/accuracies": 0.5666666626930237,
"eval_rewards/chosen": -0.9704511761665344,
"eval_rewards/margins": 0.07456483691930771,
"eval_rewards/rejected": -1.0450160503387451,
"eval_runtime": 1591.8909,
"eval_samples_per_second": 1.881,
"eval_steps_per_second": 0.236,
"step": 900
},
{
"epoch": 0.73,
"learning_rate": 1.040730875447083e-06,
"logits/chosen": -2.7219176292419434,
"logits/rejected": -2.7300069332122803,
"logps/chosen": -374.46649169921875,
"logps/rejected": -392.5831604003906,
"loss": 0.6604,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.9455657005310059,
"rewards/margins": 0.11010564863681793,
"rewards/rejected": -1.0556714534759521,
"step": 910
},
{
"epoch": 0.74,
"learning_rate": 9.845720158932414e-07,
"logits/chosen": -2.6920173168182373,
"logits/rejected": -2.69575834274292,
"logps/chosen": -352.3219299316406,
"logps/rejected": -367.9543762207031,
"loss": 0.6388,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.9019610285758972,
"rewards/margins": 0.1525803804397583,
"rewards/rejected": -1.0545413494110107,
"step": 920
},
{
"epoch": 0.74,
"learning_rate": 9.295969449372796e-07,
"logits/chosen": -2.768012523651123,
"logits/rejected": -2.773406505584717,
"logps/chosen": -356.3282165527344,
"logps/rejected": -381.96478271484375,
"loss": 0.6444,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.873378574848175,
"rewards/margins": 0.13878589868545532,
"rewards/rejected": -1.0121644735336304,
"step": 930
},
{
"epoch": 0.75,
"learning_rate": 8.758486067922176e-07,
"logits/chosen": -2.7590746879577637,
"logits/rejected": -2.735861301422119,
"logps/chosen": -354.44818115234375,
"logps/rejected": -391.2872619628906,
"loss": 0.6291,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.9065491557121277,
"rewards/margins": 0.18637417256832123,
"rewards/rejected": -1.0929234027862549,
"step": 940
},
{
"epoch": 0.76,
"learning_rate": 8.233689873990006e-07,
"logits/chosen": -2.7534782886505127,
"logits/rejected": -2.7382559776306152,
"logps/chosen": -355.0606384277344,
"logps/rejected": -398.70513916015625,
"loss": 0.6269,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9113260507583618,
"rewards/margins": 0.26781052350997925,
"rewards/rejected": -1.1791365146636963,
"step": 950
},
{
"epoch": 0.77,
"learning_rate": 7.721990816288555e-07,
"logits/chosen": -2.7826993465423584,
"logits/rejected": -2.763814687728882,
"logps/chosen": -336.1342468261719,
"logps/rejected": -366.10809326171875,
"loss": 0.63,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.8676854372024536,
"rewards/margins": 0.1765536516904831,
"rewards/rejected": -1.0442390441894531,
"step": 960
},
{
"epoch": 0.78,
"learning_rate": 7.223788612598148e-07,
"logits/chosen": -2.7675793170928955,
"logits/rejected": -2.776650905609131,
"logps/chosen": -366.38525390625,
"logps/rejected": -391.9957580566406,
"loss": 0.6576,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.886553943157196,
"rewards/margins": 0.10182847827672958,
"rewards/rejected": -0.9883824586868286,
"step": 970
},
{
"epoch": 0.78,
"learning_rate": 6.73947243752448e-07,
"logits/chosen": -2.7559187412261963,
"logits/rejected": -2.751338481903076,
"logps/chosen": -376.8429260253906,
"logps/rejected": -383.2920837402344,
"loss": 0.6617,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9805536270141602,
"rewards/margins": 0.13034404814243317,
"rewards/rejected": -1.1108976602554321,
"step": 980
},
{
"epoch": 0.79,
"learning_rate": 6.269420618491759e-07,
"logits/chosen": -2.759143352508545,
"logits/rejected": -2.74459171295166,
"logps/chosen": -351.4375305175781,
"logps/rejected": -376.80633544921875,
"loss": 0.6654,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.9471473693847656,
"rewards/margins": 0.10330625623464584,
"rewards/rejected": -1.0504536628723145,
"step": 990
},
{
"epoch": 0.8,
"learning_rate": 5.814000340209267e-07,
"logits/chosen": -2.746699810028076,
"logits/rejected": -2.7150912284851074,
"logps/chosen": -363.809814453125,
"logps/rejected": -401.50927734375,
"loss": 0.6305,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.9182988405227661,
"rewards/margins": 0.19320932030677795,
"rewards/rejected": -1.1115081310272217,
"step": 1000
},
{
"epoch": 0.8,
"eval_logits/chosen": -2.7606282234191895,
"eval_logits/rejected": -2.7542688846588135,
"eval_logps/chosen": -307.0535583496094,
"eval_logps/rejected": -319.9799499511719,
"eval_loss": 0.6764061450958252,
"eval_rewards/accuracies": 0.5680000185966492,
"eval_rewards/chosen": -0.9844286441802979,
"eval_rewards/margins": 0.07588021457195282,
"eval_rewards/rejected": -1.060308814048767,
"eval_runtime": 1582.573,
"eval_samples_per_second": 1.892,
"eval_steps_per_second": 0.237,
"step": 1000
},
{
"epoch": 0.81,
"learning_rate": 5.373567357842111e-07,
"logits/chosen": -2.703906536102295,
"logits/rejected": -2.7083990573883057,
"logps/chosen": -357.48388671875,
"logps/rejected": -387.49078369140625,
"loss": 0.6159,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8543448448181152,
"rewards/margins": 0.21335168182849884,
"rewards/rejected": -1.0676965713500977,
"step": 1010
},
{
"epoch": 0.82,
"learning_rate": 4.948465719110226e-07,
"logits/chosen": -2.7659733295440674,
"logits/rejected": -2.7676990032196045,
"logps/chosen": -350.13482666015625,
"logps/rejected": -371.6733703613281,
"loss": 0.6451,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.8929713368415833,
"rewards/margins": 0.14709730446338654,
"rewards/rejected": -1.0400686264038086,
"step": 1020
},
{
"epoch": 0.82,
"learning_rate": 4.539027495532766e-07,
"logits/chosen": -2.715686798095703,
"logits/rejected": -2.7308743000030518,
"logps/chosen": -346.8404235839844,
"logps/rejected": -375.3435363769531,
"loss": 0.6263,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.8631433248519897,
"rewards/margins": 0.18629541993141174,
"rewards/rejected": -1.049438714981079,
"step": 1030
},
{
"epoch": 0.83,
"learning_rate": 4.14557252302783e-07,
"logits/chosen": -2.7563765048980713,
"logits/rejected": -2.741762399673462,
"logps/chosen": -368.244140625,
"logps/rejected": -396.12298583984375,
"loss": 0.6673,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.0178955793380737,
"rewards/margins": 0.13189546763896942,
"rewards/rejected": -1.1497912406921387,
"step": 1040
},
{
"epoch": 0.84,
"learning_rate": 3.7684081520700884e-07,
"logits/chosen": -2.719506025314331,
"logits/rejected": -2.712674140930176,
"logps/chosen": -373.14007568359375,
"logps/rejected": -384.85870361328125,
"loss": 0.6379,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.888410210609436,
"rewards/margins": 0.16670987010002136,
"rewards/rejected": -1.0551201105117798,
"step": 1050
},
{
"epoch": 0.85,
"learning_rate": 3.407829007601507e-07,
"logits/chosen": -2.723992109298706,
"logits/rejected": -2.7108616828918457,
"logps/chosen": -362.71112060546875,
"logps/rejected": -396.30413818359375,
"loss": 0.6307,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.9032829403877258,
"rewards/margins": 0.18060991168022156,
"rewards/rejected": -1.0838927030563354,
"step": 1060
},
{
"epoch": 0.86,
"learning_rate": 3.064116758882724e-07,
"logits/chosen": -2.7330386638641357,
"logits/rejected": -2.700990676879883,
"logps/chosen": -375.42425537109375,
"logps/rejected": -414.94635009765625,
"loss": 0.6324,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.9860410690307617,
"rewards/margins": 0.20865873992443085,
"rewards/rejected": -1.1946996450424194,
"step": 1070
},
{
"epoch": 0.86,
"learning_rate": 2.737539899464908e-07,
"logits/chosen": -2.7359976768493652,
"logits/rejected": -2.7335023880004883,
"logps/chosen": -345.5082092285156,
"logps/rejected": -376.73382568359375,
"loss": 0.633,
"rewards/accuracies": 0.65625,
"rewards/chosen": -0.9359216690063477,
"rewards/margins": 0.19209426641464233,
"rewards/rejected": -1.1280159950256348,
"step": 1080
},
{
"epoch": 0.87,
"learning_rate": 2.4283535374538645e-07,
"logits/chosen": -2.7470860481262207,
"logits/rejected": -2.759739398956299,
"logps/chosen": -367.7596130371094,
"logps/rejected": -393.4772644042969,
"loss": 0.6363,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.8926227688789368,
"rewards/margins": 0.18235111236572266,
"rewards/rejected": -1.0749738216400146,
"step": 1090
},
{
"epoch": 0.88,
"learning_rate": 2.1367991962303298e-07,
"logits/chosen": -2.701397657394409,
"logits/rejected": -2.7229931354522705,
"logps/chosen": -348.02447509765625,
"logps/rejected": -368.96197509765625,
"loss": 0.6754,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.9437961578369141,
"rewards/margins": 0.08066528290510178,
"rewards/rejected": -1.0244615077972412,
"step": 1100
},
{
"epoch": 0.88,
"eval_logits/chosen": -2.7602105140686035,
"eval_logits/rejected": -2.7538294792175293,
"eval_logps/chosen": -307.4263916015625,
"eval_logps/rejected": -320.4283447265625,
"eval_loss": 0.6762885451316833,
"eval_rewards/accuracies": 0.5686666369438171,
"eval_rewards/chosen": -0.9881570935249329,
"eval_rewards/margins": 0.07663587480783463,
"eval_rewards/rejected": -1.0647929906845093,
"eval_runtime": 1581.3137,
"eval_samples_per_second": 1.893,
"eval_steps_per_second": 0.237,
"step": 1100
},
{
"epoch": 0.89,
"learning_rate": 1.8631046257820278e-07,
"logits/chosen": -2.747465133666992,
"logits/rejected": -2.7517151832580566,
"logps/chosen": -368.2339172363281,
"logps/rejected": -389.92803955078125,
"loss": 0.6368,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.9324411153793335,
"rewards/margins": 0.17461669445037842,
"rewards/rejected": -1.1070579290390015,
"step": 1110
},
{
"epoch": 0.9,
"learning_rate": 1.6074836247950143e-07,
"logits/chosen": -2.7465901374816895,
"logits/rejected": -2.7471508979797363,
"logps/chosen": -372.44866943359375,
"logps/rejected": -390.7837829589844,
"loss": 0.6522,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.8873372077941895,
"rewards/margins": 0.12168030440807343,
"rewards/rejected": -1.0090174674987793,
"step": 1120
},
{
"epoch": 0.9,
"learning_rate": 1.370135873643097e-07,
"logits/chosen": -2.7472376823425293,
"logits/rejected": -2.7466180324554443,
"logps/chosen": -373.3634338378906,
"logps/rejected": -377.52569580078125,
"loss": 0.6504,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.9632595777511597,
"rewards/margins": 0.13923458755016327,
"rewards/rejected": -1.102494239807129,
"step": 1130
},
{
"epoch": 0.91,
"learning_rate": 1.1512467784059372e-07,
"logits/chosen": -2.7725350856781006,
"logits/rejected": -2.748663902282715,
"logps/chosen": -343.6497802734375,
"logps/rejected": -369.282470703125,
"loss": 0.6371,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.9474760293960571,
"rewards/margins": 0.1754283607006073,
"rewards/rejected": -1.1229044198989868,
"step": 1140
},
{
"epoch": 0.92,
"learning_rate": 9.509873260376251e-08,
"logits/chosen": -2.727968692779541,
"logits/rejected": -2.703981876373291,
"logps/chosen": -363.1959228515625,
"logps/rejected": -414.9847106933594,
"loss": 0.6061,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.9189531207084656,
"rewards/margins": 0.2506302297115326,
"rewards/rejected": -1.1695834398269653,
"step": 1150
},
{
"epoch": 0.93,
"learning_rate": 7.695139507988559e-08,
"logits/chosen": -2.7512032985687256,
"logits/rejected": -2.7607004642486572,
"logps/chosen": -376.4388427734375,
"logps/rejected": -397.3370056152344,
"loss": 0.6506,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.925561785697937,
"rewards/margins": 0.1294144093990326,
"rewards/rejected": -1.054976224899292,
"step": 1160
},
{
"epoch": 0.94,
"learning_rate": 6.069684120570684e-08,
"logits/chosen": -2.715827703475952,
"logits/rejected": -2.679953098297119,
"logps/chosen": -362.00079345703125,
"logps/rejected": -398.26141357421875,
"loss": 0.6424,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.9743494987487793,
"rewards/margins": 0.17094501852989197,
"rewards/rejected": -1.145294427871704,
"step": 1170
},
{
"epoch": 0.94,
"learning_rate": 4.634776835499871e-08,
"logits/chosen": -2.679133415222168,
"logits/rejected": -2.6611809730529785,
"logps/chosen": -350.9591369628906,
"logps/rejected": -379.08416748046875,
"loss": 0.648,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.944057822227478,
"rewards/margins": 0.15136626362800598,
"rewards/rejected": -1.0954240560531616,
"step": 1180
},
{
"epoch": 0.95,
"learning_rate": 3.3915385419908964e-08,
"logits/chosen": -2.730663299560547,
"logits/rejected": -2.7344131469726562,
"logps/chosen": -362.0574951171875,
"logps/rejected": -382.64788818359375,
"loss": 0.645,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9124767184257507,
"rewards/margins": 0.15995648503303528,
"rewards/rejected": -1.0724332332611084,
"step": 1190
},
{
"epoch": 0.96,
"learning_rate": 2.3409404055043938e-08,
"logits/chosen": -2.7462306022644043,
"logits/rejected": -2.7565507888793945,
"logps/chosen": -369.1392517089844,
"logps/rejected": -390.51190185546875,
"loss": 0.6577,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.9762216806411743,
"rewards/margins": 0.11748246848583221,
"rewards/rejected": -1.093704104423523,
"step": 1200
},
{
"epoch": 0.96,
"eval_logits/chosen": -2.7601613998413086,
"eval_logits/rejected": -2.7538154125213623,
"eval_logps/chosen": -307.4615478515625,
"eval_logps/rejected": -320.441162109375,
"eval_loss": 0.6764407753944397,
"eval_rewards/accuracies": 0.5663333535194397,
"eval_rewards/chosen": -0.9885084629058838,
"eval_rewards/margins": 0.07641255110502243,
"eval_rewards/rejected": -1.0649209022521973,
"eval_runtime": 1656.3369,
"eval_samples_per_second": 1.808,
"eval_steps_per_second": 0.226,
"step": 1200
},
{
"epoch": 0.97,
"learning_rate": 1.4838031091134186e-08,
"logits/chosen": -2.711413860321045,
"logits/rejected": -2.700925350189209,
"logps/chosen": -341.56524658203125,
"logps/rejected": -391.4617614746094,
"loss": 0.6092,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.8985296487808228,
"rewards/margins": 0.250393271446228,
"rewards/rejected": -1.1489229202270508,
"step": 1210
},
{
"epoch": 0.98,
"learning_rate": 8.207962124201774e-09,
"logits/chosen": -2.766512393951416,
"logits/rejected": -2.751699924468994,
"logps/chosen": -370.7591857910156,
"logps/rejected": -398.38201904296875,
"loss": 0.6128,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.873375415802002,
"rewards/margins": 0.22166454792022705,
"rewards/rejected": -1.0950400829315186,
"step": 1220
},
{
"epoch": 0.98,
"learning_rate": 3.5243762852441023e-09,
"logits/chosen": -2.7122433185577393,
"logits/rejected": -2.7001757621765137,
"logps/chosen": -358.870849609375,
"logps/rejected": -391.4342956542969,
"loss": 0.6408,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.9104388952255249,
"rewards/margins": 0.1689816415309906,
"rewards/rejected": -1.0794206857681274,
"step": 1230
},
{
"epoch": 0.99,
"learning_rate": 7.909321945129278e-10,
"logits/chosen": -2.7206053733825684,
"logits/rejected": -2.7036147117614746,
"logps/chosen": -379.60906982421875,
"logps/rejected": -411.9668884277344,
"loss": 0.6327,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9332448840141296,
"rewards/margins": 0.1873912662267685,
"rewards/rejected": -1.120636224746704,
"step": 1240
},
{
"epoch": 1.0,
"step": 1249,
"total_flos": 0.0,
"train_loss": 0.6566078573155155,
"train_runtime": 38377.8314,
"train_samples_per_second": 0.521,
"train_steps_per_second": 0.033
}
],
"logging_steps": 10,
"max_steps": 1249,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}