mistral7b_dpo_en / trainer_state.json
Krisbiantoro's picture
Upload folder using huggingface_hub
1fa4da1
Raw
History Blame
25.6 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.09665104141497124,
"eval_steps": 200,
"global_step": 500,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 9.652509652509653e-06,
"logits/chosen": -3.11246657371521,
"logits/rejected": -3.086373805999756,
"logps/chosen": -113.73238372802734,
"logps/rejected": -109.32698822021484,
"loss": 0.721,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.2993558943271637,
"rewards/margins": -0.0277109295129776,
"rewards/rejected": -0.2716449797153473,
"step": 10
},
{
"epoch": 0.0,
"learning_rate": 1.9305019305019306e-05,
"logits/chosen": -3.110931873321533,
"logits/rejected": -3.1171531677246094,
"logps/chosen": -122.4663314819336,
"logps/rejected": -113.23054504394531,
"loss": 0.7161,
"rewards/accuracies": 0.42500001192092896,
"rewards/chosen": -0.2805718183517456,
"rewards/margins": -0.03167964145541191,
"rewards/rejected": -0.2488921880722046,
"step": 20
},
{
"epoch": 0.01,
"learning_rate": 2.895752895752896e-05,
"logits/chosen": -3.139052629470825,
"logits/rejected": -3.1156527996063232,
"logps/chosen": -126.01689147949219,
"logps/rejected": -100.77046203613281,
"loss": 0.7224,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.3138067126274109,
"rewards/margins": -0.038806475698947906,
"rewards/rejected": -0.2750001847743988,
"step": 30
},
{
"epoch": 0.01,
"learning_rate": 3.764478764478765e-05,
"logits/chosen": -3.155150890350342,
"logits/rejected": -3.1715970039367676,
"logps/chosen": -133.27737426757812,
"logps/rejected": -118.9439926147461,
"loss": 0.713,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.2480003386735916,
"rewards/margins": -0.023857835680246353,
"rewards/rejected": -0.22414250671863556,
"step": 40
},
{
"epoch": 0.01,
"learning_rate": 4.72972972972973e-05,
"logits/chosen": -3.1669116020202637,
"logits/rejected": -3.1525278091430664,
"logps/chosen": -123.1195297241211,
"logps/rejected": -128.38714599609375,
"loss": 0.6781,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.19161827862262726,
"rewards/margins": 0.04476013034582138,
"rewards/rejected": -0.23637838661670685,
"step": 50
},
{
"epoch": 0.01,
"learning_rate": 5.694980694980695e-05,
"logits/chosen": -3.1426501274108887,
"logits/rejected": -3.132570266723633,
"logps/chosen": -120.64261627197266,
"logps/rejected": -113.0268783569336,
"loss": 0.7107,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.3047412037849426,
"rewards/margins": -0.014080168679356575,
"rewards/rejected": -0.2906610071659088,
"step": 60
},
{
"epoch": 0.01,
"learning_rate": 6.660231660231661e-05,
"logits/chosen": -3.170804500579834,
"logits/rejected": -3.164586305618286,
"logps/chosen": -116.2149887084961,
"logps/rejected": -126.68898010253906,
"loss": 0.6885,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.33613839745521545,
"rewards/margins": 0.028235793113708496,
"rewards/rejected": -0.36437422037124634,
"step": 70
},
{
"epoch": 0.02,
"learning_rate": 7.625482625482626e-05,
"logits/chosen": -3.17895245552063,
"logits/rejected": -3.1590020656585693,
"logps/chosen": -126.33811950683594,
"logps/rejected": -103.02183532714844,
"loss": 0.6215,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.4042009711265564,
"rewards/margins": 0.19598612189292908,
"rewards/rejected": -0.6001870632171631,
"step": 80
},
{
"epoch": 0.02,
"learning_rate": 8.59073359073359e-05,
"logits/chosen": -3.180785894393921,
"logits/rejected": -3.1576857566833496,
"logps/chosen": -129.47866821289062,
"logps/rejected": -126.99539947509766,
"loss": 0.6889,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.8047823905944824,
"rewards/margins": 0.08432246744632721,
"rewards/rejected": -0.8891048431396484,
"step": 90
},
{
"epoch": 0.02,
"learning_rate": 9.555984555984557e-05,
"logits/chosen": -3.139349937438965,
"logits/rejected": -3.114441394805908,
"logps/chosen": -139.87002563476562,
"logps/rejected": -135.06802368164062,
"loss": 0.6326,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.2678108215332031,
"rewards/margins": 0.2349836528301239,
"rewards/rejected": -1.5027945041656494,
"step": 100
},
{
"epoch": 0.02,
"learning_rate": 0.00010424710424710426,
"logits/chosen": -3.155695676803589,
"logits/rejected": -3.1192359924316406,
"logps/chosen": -126.76655578613281,
"logps/rejected": -119.91800689697266,
"loss": 0.6925,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.1118037700653076,
"rewards/margins": 0.12902173399925232,
"rewards/rejected": -1.2408255338668823,
"step": 110
},
{
"epoch": 0.02,
"learning_rate": 0.0001138996138996139,
"logits/chosen": -3.203996181488037,
"logits/rejected": -3.181511402130127,
"logps/chosen": -121.38505554199219,
"logps/rejected": -123.63691711425781,
"loss": 0.6905,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.83983314037323,
"rewards/margins": 0.14224112033843994,
"rewards/rejected": -0.9820743799209595,
"step": 120
},
{
"epoch": 0.03,
"learning_rate": 0.00012355212355212355,
"logits/chosen": -3.199700355529785,
"logits/rejected": -3.142176628112793,
"logps/chosen": -144.78311157226562,
"logps/rejected": -131.46128845214844,
"loss": 0.7421,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7906621694564819,
"rewards/margins": 0.08599194139242172,
"rewards/rejected": -0.8766541481018066,
"step": 130
},
{
"epoch": 0.03,
"learning_rate": 0.00013223938223938227,
"logits/chosen": -3.11432147026062,
"logits/rejected": -3.0819637775421143,
"logps/chosen": -131.96109008789062,
"logps/rejected": -118.2151107788086,
"loss": 0.7438,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.9345771074295044,
"rewards/margins": 0.03206203132867813,
"rewards/rejected": -0.9666391611099243,
"step": 140
},
{
"epoch": 0.03,
"learning_rate": 0.00014189189189189188,
"logits/chosen": -3.103154182434082,
"logits/rejected": -3.0087850093841553,
"logps/chosen": -111.38960266113281,
"logps/rejected": -109.3032455444336,
"loss": 0.6658,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.4318141043186188,
"rewards/margins": 0.15236088633537292,
"rewards/rejected": -0.5841749906539917,
"step": 150
},
{
"epoch": 0.03,
"learning_rate": 0.00015154440154440155,
"logits/chosen": -3.065882444381714,
"logits/rejected": -3.014258623123169,
"logps/chosen": -115.94306945800781,
"logps/rejected": -129.7348175048828,
"loss": 0.7302,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.6290556192398071,
"rewards/margins": 0.06416401267051697,
"rewards/rejected": -0.6932196021080017,
"step": 160
},
{
"epoch": 0.03,
"learning_rate": 0.0001611969111969112,
"logits/chosen": -2.8696858882904053,
"logits/rejected": -2.820652723312378,
"logps/chosen": -121.01307678222656,
"logps/rejected": -122.356201171875,
"loss": 0.7221,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.7266199588775635,
"rewards/margins": 0.14146149158477783,
"rewards/rejected": -0.8680814504623413,
"step": 170
},
{
"epoch": 0.03,
"learning_rate": 0.00017084942084942084,
"logits/chosen": -2.89375638961792,
"logits/rejected": -2.8223800659179688,
"logps/chosen": -131.37777709960938,
"logps/rejected": -125.69004821777344,
"loss": 0.583,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.9062817692756653,
"rewards/margins": 0.5299333333969116,
"rewards/rejected": -1.4362150430679321,
"step": 180
},
{
"epoch": 0.04,
"learning_rate": 0.0001805019305019305,
"logits/chosen": -2.8086953163146973,
"logits/rejected": -2.8810436725616455,
"logps/chosen": -115.71038818359375,
"logps/rejected": -133.2216339111328,
"loss": 0.7593,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.390483021736145,
"rewards/margins": 0.12736426293849945,
"rewards/rejected": -1.5178472995758057,
"step": 190
},
{
"epoch": 0.04,
"learning_rate": 0.00019015444015444015,
"logits/chosen": -3.0167882442474365,
"logits/rejected": -2.996938943862915,
"logps/chosen": -118.67408752441406,
"logps/rejected": -106.15169525146484,
"loss": 0.672,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.7122364640235901,
"rewards/margins": 0.25746825337409973,
"rewards/rejected": -0.9697047472000122,
"step": 200
},
{
"epoch": 0.04,
"eval_logits/chosen": -3.1546952724456787,
"eval_logits/rejected": -3.1338424682617188,
"eval_logps/chosen": -127.55575561523438,
"eval_logps/rejected": -128.7761993408203,
"eval_loss": 0.743442177772522,
"eval_rewards/accuracies": 0.5858798623085022,
"eval_rewards/chosen": -1.0754988193511963,
"eval_rewards/margins": 0.3047899305820465,
"eval_rewards/rejected": -1.3802887201309204,
"eval_runtime": 1335.2681,
"eval_samples_per_second": 0.711,
"eval_steps_per_second": 0.711,
"step": 200
},
{
"epoch": 0.04,
"learning_rate": 0.0001998069498069498,
"logits/chosen": -3.2106406688690186,
"logits/rejected": -3.1746304035186768,
"logps/chosen": -131.38128662109375,
"logps/rejected": -121.36234283447266,
"loss": 0.664,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.005927324295044,
"rewards/margins": 0.5132917165756226,
"rewards/rejected": -1.5192190408706665,
"step": 210
},
{
"epoch": 0.04,
"learning_rate": 0.00020945945945945947,
"logits/chosen": -3.2851333618164062,
"logits/rejected": -3.267256259918213,
"logps/chosen": -120.11392974853516,
"logps/rejected": -117.89964294433594,
"loss": 0.6965,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.7865055203437805,
"rewards/margins": 0.19706687331199646,
"rewards/rejected": -0.9835723638534546,
"step": 220
},
{
"epoch": 0.04,
"learning_rate": 0.0002191119691119691,
"logits/chosen": -3.408318281173706,
"logits/rejected": -3.351940870285034,
"logps/chosen": -107.76014709472656,
"logps/rejected": -107.02482604980469,
"loss": 0.7385,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.41882553696632385,
"rewards/margins": 0.01567765511572361,
"rewards/rejected": -0.4345032274723053,
"step": 230
},
{
"epoch": 0.05,
"learning_rate": 0.00022876447876447875,
"logits/chosen": -3.155478000640869,
"logits/rejected": -3.135077953338623,
"logps/chosen": -131.94711303710938,
"logps/rejected": -112.65836334228516,
"loss": 0.7328,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.6660552024841309,
"rewards/margins": 0.10303208976984024,
"rewards/rejected": -0.7690872550010681,
"step": 240
},
{
"epoch": 0.05,
"learning_rate": 0.00023841698841698842,
"logits/chosen": -3.290839433670044,
"logits/rejected": -3.2837767601013184,
"logps/chosen": -132.83676147460938,
"logps/rejected": -119.4383773803711,
"loss": 0.678,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.9368747472763062,
"rewards/margins": 0.25725504755973816,
"rewards/rejected": -1.1941298246383667,
"step": 250
},
{
"epoch": 0.05,
"learning_rate": 0.0002480694980694981,
"logits/chosen": -3.3188934326171875,
"logits/rejected": -3.361811876296997,
"logps/chosen": -130.7174835205078,
"logps/rejected": -147.4563751220703,
"loss": 0.7069,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.9200389981269836,
"rewards/margins": 0.4393937587738037,
"rewards/rejected": -1.3594326972961426,
"step": 260
},
{
"epoch": 0.05,
"learning_rate": 0.0002567567567567567,
"logits/chosen": -3.249516010284424,
"logits/rejected": -3.2256407737731934,
"logps/chosen": -108.8515853881836,
"logps/rejected": -132.24813842773438,
"loss": 0.6983,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.42829591035842896,
"rewards/margins": 0.3723019063472748,
"rewards/rejected": -0.8005977869033813,
"step": 270
},
{
"epoch": 0.05,
"learning_rate": 0.0002635135135135135,
"logits/chosen": -3.0320539474487305,
"logits/rejected": -2.982640027999878,
"logps/chosen": -175.3157501220703,
"logps/rejected": -154.48963928222656,
"loss": 1.4131,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -2.973146915435791,
"rewards/margins": -0.0525052547454834,
"rewards/rejected": -2.9206414222717285,
"step": 280
},
{
"epoch": 0.06,
"learning_rate": 0.00027316602316602317,
"logits/chosen": -2.9082584381103516,
"logits/rejected": -2.8451313972473145,
"logps/chosen": -129.400146484375,
"logps/rejected": -118.70863342285156,
"loss": 0.9237,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -1.8987582921981812,
"rewards/margins": 0.024892251938581467,
"rewards/rejected": -1.9236505031585693,
"step": 290
},
{
"epoch": 0.06,
"learning_rate": 0.00028281853281853284,
"logits/chosen": -3.066049337387085,
"logits/rejected": -2.9455220699310303,
"logps/chosen": -131.84317016601562,
"logps/rejected": -121.24568176269531,
"loss": 0.6998,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.1568044424057007,
"rewards/margins": 0.3450776934623718,
"rewards/rejected": -1.5018823146820068,
"step": 300
},
{
"epoch": 0.06,
"learning_rate": 0.0002924710424710425,
"logits/chosen": -2.8079309463500977,
"logits/rejected": -2.773160457611084,
"logps/chosen": -169.48828125,
"logps/rejected": -148.60374450683594,
"loss": 2.3505,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -5.812082290649414,
"rewards/margins": -0.6856533288955688,
"rewards/rejected": -5.126428127288818,
"step": 310
},
{
"epoch": 0.06,
"learning_rate": 0.0003021235521235521,
"logits/chosen": -2.6579480171203613,
"logits/rejected": -2.6659698486328125,
"logps/chosen": -141.78701782226562,
"logps/rejected": -160.35110473632812,
"loss": 0.8982,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -3.301175594329834,
"rewards/margins": 0.01409349124878645,
"rewards/rejected": -3.3152689933776855,
"step": 320
},
{
"epoch": 0.06,
"learning_rate": 0.0003117760617760618,
"logits/chosen": -2.8775956630706787,
"logits/rejected": -2.8255507946014404,
"logps/chosen": -150.46177673339844,
"logps/rejected": -144.97073364257812,
"loss": 0.9436,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -3.29494047164917,
"rewards/margins": 0.0633418932557106,
"rewards/rejected": -3.3582825660705566,
"step": 330
},
{
"epoch": 0.07,
"learning_rate": 0.0003204633204633205,
"logits/chosen": -2.8166918754577637,
"logits/rejected": -2.818556308746338,
"logps/chosen": -171.34437561035156,
"logps/rejected": -172.5870361328125,
"loss": 1.0895,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -4.722414016723633,
"rewards/margins": 0.23760518431663513,
"rewards/rejected": -4.960019111633301,
"step": 340
},
{
"epoch": 0.07,
"learning_rate": 0.00033011583011583015,
"logits/chosen": -2.9005284309387207,
"logits/rejected": -2.905046224594116,
"logps/chosen": -179.5493621826172,
"logps/rejected": -189.4880828857422,
"loss": 1.4019,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -4.879184722900391,
"rewards/margins": 0.2637065649032593,
"rewards/rejected": -5.142890930175781,
"step": 350
},
{
"epoch": 0.07,
"learning_rate": 0.00033976833976833977,
"logits/chosen": -2.4955551624298096,
"logits/rejected": -2.5379605293273926,
"logps/chosen": -142.00270080566406,
"logps/rejected": -147.41220092773438,
"loss": 0.7586,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.0069820880889893,
"rewards/margins": 0.49537187814712524,
"rewards/rejected": -3.502354383468628,
"step": 360
},
{
"epoch": 0.07,
"learning_rate": 0.00034942084942084944,
"logits/chosen": -2.451601266860962,
"logits/rejected": -2.4406824111938477,
"logps/chosen": -177.74746704101562,
"logps/rejected": -181.7180633544922,
"loss": 1.6042,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -6.1398024559021,
"rewards/margins": -0.07387089729309082,
"rewards/rejected": -6.065931797027588,
"step": 370
},
{
"epoch": 0.07,
"learning_rate": 0.0003590733590733591,
"logits/chosen": -2.4471051692962646,
"logits/rejected": -2.409393548965454,
"logps/chosen": -182.06051635742188,
"logps/rejected": -163.97035217285156,
"loss": 2.4314,
"rewards/accuracies": 0.4375,
"rewards/chosen": -6.733994483947754,
"rewards/margins": -1.0469824075698853,
"rewards/rejected": -5.687012672424316,
"step": 380
},
{
"epoch": 0.08,
"learning_rate": 0.0003687258687258687,
"logits/chosen": -2.8171439170837402,
"logits/rejected": -2.7251698970794678,
"logps/chosen": -160.04652404785156,
"logps/rejected": -142.2588348388672,
"loss": 1.0093,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -3.551713466644287,
"rewards/margins": 0.07490300387144089,
"rewards/rejected": -3.6266167163848877,
"step": 390
},
{
"epoch": 0.08,
"learning_rate": 0.0003783783783783784,
"logits/chosen": -2.8222527503967285,
"logits/rejected": -2.8754923343658447,
"logps/chosen": -138.70736694335938,
"logps/rejected": -142.30128479003906,
"loss": 0.945,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -3.170539140701294,
"rewards/margins": 0.011356920003890991,
"rewards/rejected": -3.1818957328796387,
"step": 400
},
{
"epoch": 0.08,
"eval_logits/chosen": -2.9538896083831787,
"eval_logits/rejected": -2.8971762657165527,
"eval_logps/chosen": -152.2897491455078,
"eval_logps/rejected": -150.13941955566406,
"eval_loss": 1.0823436975479126,
"eval_rewards/accuracies": 0.5100105404853821,
"eval_rewards/chosen": -3.548898458480835,
"eval_rewards/margins": -0.032288454473018646,
"eval_rewards/rejected": -3.5166099071502686,
"eval_runtime": 1347.2142,
"eval_samples_per_second": 0.704,
"eval_steps_per_second": 0.704,
"step": 400
},
{
"epoch": 0.08,
"learning_rate": 0.00038803088803088807,
"logits/chosen": -2.8151559829711914,
"logits/rejected": -2.8057456016540527,
"logps/chosen": -145.69772338867188,
"logps/rejected": -156.96774291992188,
"loss": 0.8857,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -3.8467044830322266,
"rewards/margins": 0.3242764472961426,
"rewards/rejected": -4.170981407165527,
"step": 410
},
{
"epoch": 0.08,
"learning_rate": 0.0003976833976833977,
"logits/chosen": -2.7274651527404785,
"logits/rejected": -2.713927745819092,
"logps/chosen": -164.06692504882812,
"logps/rejected": -149.8723907470703,
"loss": 0.9745,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -3.5636115074157715,
"rewards/margins": 0.0345739908516407,
"rewards/rejected": -3.5981857776641846,
"step": 420
},
{
"epoch": 0.08,
"learning_rate": 0.00040733590733590735,
"logits/chosen": -2.8782758712768555,
"logits/rejected": -2.8031697273254395,
"logps/chosen": -165.91473388671875,
"logps/rejected": -147.03627014160156,
"loss": 0.8281,
"rewards/accuracies": 0.625,
"rewards/chosen": -4.2417216300964355,
"rewards/margins": 0.32142549753189087,
"rewards/rejected": -4.563147068023682,
"step": 430
},
{
"epoch": 0.09,
"learning_rate": 0.000416988416988417,
"logits/chosen": -2.623711585998535,
"logits/rejected": -2.622528314590454,
"logps/chosen": -149.8426513671875,
"logps/rejected": -159.93692016601562,
"loss": 0.9961,
"rewards/accuracies": 0.5,
"rewards/chosen": -4.433084487915039,
"rewards/margins": 0.28495556116104126,
"rewards/rejected": -4.7180399894714355,
"step": 440
},
{
"epoch": 0.09,
"learning_rate": 0.00042664092664092664,
"logits/chosen": -2.580371618270874,
"logits/rejected": -2.5741231441497803,
"logps/chosen": -160.84347534179688,
"logps/rejected": -141.23475646972656,
"loss": 1.2914,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -4.076364994049072,
"rewards/margins": -0.23183032870292664,
"rewards/rejected": -3.8445351123809814,
"step": 450
},
{
"epoch": 0.09,
"learning_rate": 0.0004362934362934363,
"logits/chosen": -2.889563798904419,
"logits/rejected": -2.8342082500457764,
"logps/chosen": -186.77017211914062,
"logps/rejected": -168.42330932617188,
"loss": 1.1317,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -4.674792289733887,
"rewards/margins": -0.265504390001297,
"rewards/rejected": -4.409287929534912,
"step": 460
},
{
"epoch": 0.09,
"learning_rate": 0.000445945945945946,
"logits/chosen": -2.3731606006622314,
"logits/rejected": -2.344404697418213,
"logps/chosen": -172.8909454345703,
"logps/rejected": -175.9696502685547,
"loss": 0.9674,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -5.584943771362305,
"rewards/margins": 0.14772634208202362,
"rewards/rejected": -5.732670783996582,
"step": 470
},
{
"epoch": 0.09,
"learning_rate": 0.0004555984555984556,
"logits/chosen": -2.3436319828033447,
"logits/rejected": -2.301845073699951,
"logps/chosen": -173.07313537597656,
"logps/rejected": -169.7339630126953,
"loss": 1.2097,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -5.504385471343994,
"rewards/margins": 0.23244301974773407,
"rewards/rejected": -5.736828804016113,
"step": 480
},
{
"epoch": 0.09,
"learning_rate": 0.00046525096525096526,
"logits/chosen": -2.6778111457824707,
"logits/rejected": -2.5807526111602783,
"logps/chosen": -166.180419921875,
"logps/rejected": -168.2677001953125,
"loss": 1.2799,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -4.63643217086792,
"rewards/margins": 0.5320998430252075,
"rewards/rejected": -5.168532371520996,
"step": 490
},
{
"epoch": 0.1,
"learning_rate": 0.00047490347490347493,
"logits/chosen": -3.083743095397949,
"logits/rejected": -3.080765962600708,
"logps/chosen": -193.96841430664062,
"logps/rejected": -182.3042755126953,
"loss": 1.6246,
"rewards/accuracies": 0.5,
"rewards/chosen": -6.152979850769043,
"rewards/margins": -0.20669928193092346,
"rewards/rejected": -5.946280479431152,
"step": 500
}
],
"logging_steps": 10,
"max_steps": 5173,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"trial_name": null,
"trial_params": null
}