Instructions to use Krisbiantoro/mistral7b_dpo_en with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use Krisbiantoro/mistral7b_dpo_en with PEFT:
Base model is not found.
- Notebooks
- Google Colab
- Kaggle
| { | |
| "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 | |
| } | |