seablue's picture
Upload trainer_state.json with huggingface_hub
bdf8ca8 verified
raw
history blame
No virus
51.5 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.998691442030882,
"eval_steps": 500,
"global_step": 477,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.010468463752944255,
"grad_norm": 3.760205659470345,
"learning_rate": 5.208333333333333e-08,
"logits/chosen": -2.9944963455200195,
"logits/rejected": -2.9324326515197754,
"logps/chosen": -1.0260854959487915,
"logps/rejected": -1.189737319946289,
"loss": -0.0244,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.0260854959487915,
"rewards/margins": 0.1636517494916916,
"rewards/rejected": -1.189737319946289,
"step": 5
},
{
"epoch": 0.02093692750588851,
"grad_norm": 4.14164094224387,
"learning_rate": 1.0416666666666667e-07,
"logits/chosen": -3.0354819297790527,
"logits/rejected": -2.9778807163238525,
"logps/chosen": -1.1017096042633057,
"logps/rejected": -1.1320788860321045,
"loss": -0.0273,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -1.1017096042633057,
"rewards/margins": 0.030369168147444725,
"rewards/rejected": -1.1320788860321045,
"step": 10
},
{
"epoch": 0.031405391258832765,
"grad_norm": 4.393103002918313,
"learning_rate": 1.5624999999999999e-07,
"logits/chosen": -3.0005249977111816,
"logits/rejected": -2.995767116546631,
"logps/chosen": -1.0664865970611572,
"logps/rejected": -1.0592315196990967,
"loss": -0.0311,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -1.0664865970611572,
"rewards/margins": -0.007255014963448048,
"rewards/rejected": -1.0592315196990967,
"step": 15
},
{
"epoch": 0.04187385501177702,
"grad_norm": 4.089316161372408,
"learning_rate": 2.0833333333333333e-07,
"logits/chosen": -3.015296697616577,
"logits/rejected": -2.9769978523254395,
"logps/chosen": -1.0041254758834839,
"logps/rejected": -1.0410144329071045,
"loss": -0.0327,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -1.0041254758834839,
"rewards/margins": 0.03688892349600792,
"rewards/rejected": -1.0410144329071045,
"step": 20
},
{
"epoch": 0.05234231876472128,
"grad_norm": 4.216563165990877,
"learning_rate": 2.604166666666667e-07,
"logits/chosen": -3.0237412452697754,
"logits/rejected": -2.973569631576538,
"logps/chosen": -0.9541431665420532,
"logps/rejected": -1.1942131519317627,
"loss": -0.0413,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9541431665420532,
"rewards/margins": 0.2400699108839035,
"rewards/rejected": -1.1942131519317627,
"step": 25
},
{
"epoch": 0.06281078251766553,
"grad_norm": 5.658050284469291,
"learning_rate": 3.1249999999999997e-07,
"logits/chosen": -3.030153274536133,
"logits/rejected": -2.9625256061553955,
"logps/chosen": -0.8760479092597961,
"logps/rejected": -0.9085603952407837,
"loss": -0.0368,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.8760479092597961,
"rewards/margins": 0.032512493431568146,
"rewards/rejected": -0.9085603952407837,
"step": 30
},
{
"epoch": 0.07327924627060979,
"grad_norm": 5.044506012240391,
"learning_rate": 3.645833333333333e-07,
"logits/chosen": -3.0141396522521973,
"logits/rejected": -2.995772361755371,
"logps/chosen": -0.8659467697143555,
"logps/rejected": -1.1117627620697021,
"loss": -0.0476,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.8659467697143555,
"rewards/margins": 0.24581606686115265,
"rewards/rejected": -1.1117627620697021,
"step": 35
},
{
"epoch": 0.08374771002355404,
"grad_norm": 4.594342778483098,
"learning_rate": 4.1666666666666667e-07,
"logits/chosen": -2.9778103828430176,
"logits/rejected": -2.9474072456359863,
"logps/chosen": -0.8786975741386414,
"logps/rejected": -0.9444503784179688,
"loss": -0.0484,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.8786975741386414,
"rewards/margins": 0.06575289368629456,
"rewards/rejected": -0.9444503784179688,
"step": 40
},
{
"epoch": 0.0942161737764983,
"grad_norm": 4.865607139601026,
"learning_rate": 4.6874999999999996e-07,
"logits/chosen": -3.025959014892578,
"logits/rejected": -3.019007444381714,
"logps/chosen": -0.8325697183609009,
"logps/rejected": -0.9673300981521606,
"loss": -0.0615,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.8325697183609009,
"rewards/margins": 0.134760320186615,
"rewards/rejected": -0.9673300981521606,
"step": 45
},
{
"epoch": 0.10468463752944256,
"grad_norm": 5.424026197216979,
"learning_rate": 4.999731868769026e-07,
"logits/chosen": -2.8837554454803467,
"logits/rejected": -2.9218649864196777,
"logps/chosen": -0.9015443921089172,
"logps/rejected": -1.0318125486373901,
"loss": -0.0525,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.9015443921089172,
"rewards/margins": 0.13026830554008484,
"rewards/rejected": -1.0318125486373901,
"step": 50
},
{
"epoch": 0.11515310128238682,
"grad_norm": 7.570587115119674,
"learning_rate": 4.996716052911017e-07,
"logits/chosen": -3.0432026386260986,
"logits/rejected": -3.038170576095581,
"logps/chosen": -0.9203222393989563,
"logps/rejected": -1.0882291793823242,
"loss": -0.0483,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.9203222393989563,
"rewards/margins": 0.1679070144891739,
"rewards/rejected": -1.0882291793823242,
"step": 55
},
{
"epoch": 0.12562156503533106,
"grad_norm": 8.483337894809958,
"learning_rate": 4.990353313429303e-07,
"logits/chosen": -2.9748623371124268,
"logits/rejected": -2.9038889408111572,
"logps/chosen": -0.9680806398391724,
"logps/rejected": -1.1494948863983154,
"loss": -0.0682,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.9680806398391724,
"rewards/margins": 0.18141421675682068,
"rewards/rejected": -1.1494948863983154,
"step": 60
},
{
"epoch": 0.1360900287882753,
"grad_norm": 6.597482381521931,
"learning_rate": 4.980652179769217e-07,
"logits/chosen": -2.9652066230773926,
"logits/rejected": -2.988760232925415,
"logps/chosen": -0.9523311853408813,
"logps/rejected": -1.1883153915405273,
"loss": -0.0605,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.9523311853408813,
"rewards/margins": 0.23598428070545197,
"rewards/rejected": -1.1883153915405273,
"step": 65
},
{
"epoch": 0.14655849254121958,
"grad_norm": 7.227660542967968,
"learning_rate": 4.967625656594781e-07,
"logits/chosen": -3.0324559211730957,
"logits/rejected": -2.9957118034362793,
"logps/chosen": -1.03068208694458,
"logps/rejected": -1.2021524906158447,
"loss": -0.0806,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.03068208694458,
"rewards/margins": 0.1714704930782318,
"rewards/rejected": -1.2021524906158447,
"step": 70
},
{
"epoch": 0.15702695629416383,
"grad_norm": 8.630957672516889,
"learning_rate": 4.951291206355559e-07,
"logits/chosen": -2.99220609664917,
"logits/rejected": -2.9557840824127197,
"logps/chosen": -1.0696258544921875,
"logps/rejected": -1.317363977432251,
"loss": -0.0631,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.0696258544921875,
"rewards/margins": 0.2477382868528366,
"rewards/rejected": -1.317363977432251,
"step": 75
},
{
"epoch": 0.16749542004710807,
"grad_norm": 6.399531931157647,
"learning_rate": 4.93167072587771e-07,
"logits/chosen": -2.994518280029297,
"logits/rejected": -2.961357831954956,
"logps/chosen": -1.0338717699050903,
"logps/rejected": -1.4787652492523193,
"loss": -0.0783,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.0338717699050903,
"rewards/margins": 0.44489365816116333,
"rewards/rejected": -1.4787652492523193,
"step": 80
},
{
"epoch": 0.17796388380005235,
"grad_norm": 7.358699467281965,
"learning_rate": 4.908790517010636e-07,
"logits/chosen": -2.9917774200439453,
"logits/rejected": -2.9551658630371094,
"logps/chosen": -0.9378916025161743,
"logps/rejected": -1.334387183189392,
"loss": -0.0833,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.9378916025161743,
"rewards/margins": 0.396495521068573,
"rewards/rejected": -1.334387183189392,
"step": 85
},
{
"epoch": 0.1884323475529966,
"grad_norm": 5.13038386551715,
"learning_rate": 4.882681251368548e-07,
"logits/chosen": -2.9607720375061035,
"logits/rejected": -2.9549806118011475,
"logps/chosen": -1.0484223365783691,
"logps/rejected": -1.3985503911972046,
"loss": -0.0862,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.0484223365783691,
"rewards/margins": 0.3501282036304474,
"rewards/rejected": -1.3985503911972046,
"step": 90
},
{
"epoch": 0.19890081130594087,
"grad_norm": 6.228634016037295,
"learning_rate": 4.853377929214243e-07,
"logits/chosen": -3.039447784423828,
"logits/rejected": -3.0229854583740234,
"logps/chosen": -1.0390279293060303,
"logps/rejected": -1.2659882307052612,
"loss": -0.0748,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -1.0390279293060303,
"rewards/margins": 0.22696030139923096,
"rewards/rejected": -1.2659882307052612,
"step": 95
},
{
"epoch": 0.2093692750588851,
"grad_norm": 13.46485741296935,
"learning_rate": 4.820919832540181e-07,
"logits/chosen": -3.0565896034240723,
"logits/rejected": -3.0682952404022217,
"logps/chosen": -1.0358636379241943,
"logps/rejected": -1.3846577405929565,
"loss": -0.0825,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -1.0358636379241943,
"rewards/margins": 0.34879425168037415,
"rewards/rejected": -1.3846577405929565,
"step": 100
},
{
"epoch": 0.21983773881182936,
"grad_norm": 13.308269094275541,
"learning_rate": 4.785350472409791e-07,
"logits/chosen": -3.013173818588257,
"logits/rejected": -3.020575761795044,
"logps/chosen": -1.2395133972167969,
"logps/rejected": -1.6027743816375732,
"loss": -0.1058,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.2395133972167969,
"rewards/margins": 0.363260954618454,
"rewards/rejected": -1.6027743816375732,
"step": 105
},
{
"epoch": 0.23030620256477363,
"grad_norm": 14.80943809386291,
"learning_rate": 4.7467175306295647e-07,
"logits/chosen": -2.907844066619873,
"logits/rejected": -2.9418551921844482,
"logps/chosen": -1.0689195394515991,
"logps/rejected": -1.62472403049469,
"loss": -0.1078,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.0689195394515991,
"rewards/margins": 0.5558044910430908,
"rewards/rejected": -1.62472403049469,
"step": 110
},
{
"epoch": 0.24077466631771788,
"grad_norm": 7.302061093053783,
"learning_rate": 4.70507279583015e-07,
"logits/chosen": -3.064298152923584,
"logits/rejected": -3.0547451972961426,
"logps/chosen": -1.0759824514389038,
"logps/rejected": -1.38835608959198,
"loss": -0.089,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.0759824514389038,
"rewards/margins": 0.3123735785484314,
"rewards/rejected": -1.38835608959198,
"step": 115
},
{
"epoch": 0.2512431300706621,
"grad_norm": 6.576904824772524,
"learning_rate": 4.6604720940421207e-07,
"logits/chosen": -2.995882272720337,
"logits/rejected": -2.9872708320617676,
"logps/chosen": -1.010962724685669,
"logps/rejected": -1.5580971240997314,
"loss": -0.0761,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.010962724685669,
"rewards/margins": 0.547134518623352,
"rewards/rejected": -1.5580971240997314,
"step": 120
},
{
"epoch": 0.26171159382360637,
"grad_norm": 10.915099417857844,
"learning_rate": 4.612975213859487e-07,
"logits/chosen": -2.8915393352508545,
"logits/rejected": -2.841787338256836,
"logps/chosen": -1.0743449926376343,
"logps/rejected": -1.400960922241211,
"loss": -0.0844,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.0743449926376343,
"rewards/margins": 0.3266158401966095,
"rewards/rejected": -1.400960922241211,
"step": 125
},
{
"epoch": 0.2721800575765506,
"grad_norm": 7.5813693492052066,
"learning_rate": 4.5626458262912735e-07,
"logits/chosen": -2.9785983562469482,
"logits/rejected": -3.023461103439331,
"logps/chosen": -1.0497643947601318,
"logps/rejected": -1.5954610109329224,
"loss": -0.0988,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.0497643947601318,
"rewards/margins": 0.5456965565681458,
"rewards/rejected": -1.5954610109329224,
"step": 130
},
{
"epoch": 0.2826485213294949,
"grad_norm": 7.880456151156036,
"learning_rate": 4.5095513994085974e-07,
"logits/chosen": -2.9451870918273926,
"logits/rejected": -2.9251372814178467,
"logps/chosen": -1.1994966268539429,
"logps/rejected": -1.6415138244628906,
"loss": -0.0959,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.1994966268539429,
"rewards/margins": 0.442017138004303,
"rewards/rejected": -1.6415138244628906,
"step": 135
},
{
"epoch": 0.29311698508243916,
"grad_norm": 11.389751848981007,
"learning_rate": 4.453763107901675e-07,
"logits/chosen": -2.9757840633392334,
"logits/rejected": -2.9875569343566895,
"logps/chosen": -1.3223577737808228,
"logps/rejected": -1.7084649801254272,
"loss": -0.0795,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.3223577737808228,
"rewards/margins": 0.38610729575157166,
"rewards/rejected": -1.7084649801254272,
"step": 140
},
{
"epoch": 0.3035854488353834,
"grad_norm": 12.240889169213316,
"learning_rate": 4.395355737667985e-07,
"logits/chosen": -2.884559392929077,
"logits/rejected": -2.9181649684906006,
"logps/chosen": -1.162393569946289,
"logps/rejected": -1.6213098764419556,
"loss": -0.0999,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.162393569946289,
"rewards/margins": 0.4589163362979889,
"rewards/rejected": -1.6213098764419556,
"step": 145
},
{
"epoch": 0.31405391258832765,
"grad_norm": 13.394133386548184,
"learning_rate": 4.3344075855595097e-07,
"logits/chosen": -2.988391160964966,
"logits/rejected": -2.9592318534851074,
"logps/chosen": -1.1906192302703857,
"logps/rejected": -2.017512559890747,
"loss": -0.0867,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.1906192302703857,
"rewards/margins": 0.8268932104110718,
"rewards/rejected": -2.017512559890747,
"step": 150
},
{
"epoch": 0.3245223763412719,
"grad_norm": 10.341335190859326,
"learning_rate": 4.271000354423425e-07,
"logits/chosen": -2.871455192565918,
"logits/rejected": -2.853935718536377,
"logps/chosen": -1.3897101879119873,
"logps/rejected": -1.8974939584732056,
"loss": -0.1116,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.3897101879119873,
"rewards/margins": 0.5077837705612183,
"rewards/rejected": -1.8974939584732056,
"step": 155
},
{
"epoch": 0.33499084009421615,
"grad_norm": 14.583251944991732,
"learning_rate": 4.2052190435769554e-07,
"logits/chosen": -2.824887752532959,
"logits/rejected": -2.8534679412841797,
"logps/chosen": -1.3409488201141357,
"logps/rejected": -2.2745461463928223,
"loss": -0.1046,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.3409488201141357,
"rewards/margins": 0.933597207069397,
"rewards/rejected": -2.2745461463928223,
"step": 160
},
{
"epoch": 0.34545930384716045,
"grad_norm": 13.977236800680524,
"learning_rate": 4.137151834863213e-07,
"logits/chosen": -2.9754512310028076,
"logits/rejected": -2.960723400115967,
"logps/chosen": -1.255544662475586,
"logps/rejected": -1.7897508144378662,
"loss": -0.1166,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.255544662475586,
"rewards/margins": 0.534206211566925,
"rewards/rejected": -1.7897508144378662,
"step": 165
},
{
"epoch": 0.3559277676001047,
"grad_norm": 21.73429928886912,
"learning_rate": 4.0668899744407567e-07,
"logits/chosen": -2.92985200881958,
"logits/rejected": -2.9375405311584473,
"logps/chosen": -1.1773474216461182,
"logps/rejected": -1.8664484024047852,
"loss": -0.1043,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.1773474216461182,
"rewards/margins": 0.6891010999679565,
"rewards/rejected": -1.8664484024047852,
"step": 170
},
{
"epoch": 0.36639623135304894,
"grad_norm": 65.37359700720873,
"learning_rate": 3.994527650465352e-07,
"logits/chosen": -2.851064682006836,
"logits/rejected": -2.87498140335083,
"logps/chosen": -1.0109388828277588,
"logps/rejected": -1.6845824718475342,
"loss": -0.1099,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.0109388828277588,
"rewards/margins": 0.6736436486244202,
"rewards/rejected": -1.6845824718475342,
"step": 175
},
{
"epoch": 0.3768646951059932,
"grad_norm": 18.528930954008196,
"learning_rate": 3.920161866827889e-07,
"logits/chosen": -2.8931515216827393,
"logits/rejected": -2.873427629470825,
"logps/chosen": -1.1016333103179932,
"logps/rejected": -1.9188123941421509,
"loss": -0.1115,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.1016333103179932,
"rewards/margins": 0.8171789050102234,
"rewards/rejected": -1.9188123941421509,
"step": 180
},
{
"epoch": 0.38733315885893743,
"grad_norm": 7.80304044080231,
"learning_rate": 3.8438923131177237e-07,
"logits/chosen": -2.8918251991271973,
"logits/rejected": -2.9344098567962646,
"logps/chosen": -1.0134468078613281,
"logps/rejected": -1.6890709400177002,
"loss": -0.1128,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.0134468078613281,
"rewards/margins": 0.6756240129470825,
"rewards/rejected": -1.6890709400177002,
"step": 185
},
{
"epoch": 0.39780162261188173,
"grad_norm": 7.7047015056100046,
"learning_rate": 3.765821230985757e-07,
"logits/chosen": -2.8743629455566406,
"logits/rejected": -2.911881685256958,
"logps/chosen": -1.0063416957855225,
"logps/rejected": -1.913240671157837,
"loss": -0.1187,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.0063416957855225,
"rewards/margins": 0.9068989753723145,
"rewards/rejected": -1.913240671157837,
"step": 190
},
{
"epoch": 0.408270086364826,
"grad_norm": 7.811346817640016,
"learning_rate": 3.6860532770864005e-07,
"logits/chosen": -2.9131577014923096,
"logits/rejected": -2.89136004447937,
"logps/chosen": -1.1495698690414429,
"logps/rejected": -2.071730375289917,
"loss": -0.0945,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.1495698690414429,
"rewards/margins": 0.9221606254577637,
"rewards/rejected": -2.071730375289917,
"step": 195
},
{
"epoch": 0.4187385501177702,
"grad_norm": 11.061536990800207,
"learning_rate": 3.604695382782159e-07,
"logits/chosen": -2.813307523727417,
"logits/rejected": -2.7539114952087402,
"logps/chosen": -1.268898367881775,
"logps/rejected": -1.69308602809906,
"loss": -0.1163,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.268898367881775,
"rewards/margins": 0.4241875112056732,
"rewards/rejected": -1.69308602809906,
"step": 200
},
{
"epoch": 0.42920701387071447,
"grad_norm": 15.040971267646988,
"learning_rate": 3.5218566107988867e-07,
"logits/chosen": -2.8270034790039062,
"logits/rejected": -2.855402946472168,
"logps/chosen": -1.5387921333312988,
"logps/rejected": -2.168403148651123,
"loss": -0.1198,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.5387921333312988,
"rewards/margins": 0.6296108961105347,
"rewards/rejected": -2.168403148651123,
"step": 205
},
{
"epoch": 0.4396754776236587,
"grad_norm": 18.3545573133183,
"learning_rate": 3.4376480090239047e-07,
"logits/chosen": -2.818117618560791,
"logits/rejected": -2.7833092212677,
"logps/chosen": -1.293579339981079,
"logps/rejected": -1.8654571771621704,
"loss": -0.1002,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.293579339981079,
"rewards/margins": 0.5718778371810913,
"rewards/rejected": -1.8654571771621704,
"step": 210
},
{
"epoch": 0.45014394137660296,
"grad_norm": 8.458661335807625,
"learning_rate": 3.3521824616429284e-07,
"logits/chosen": -2.7233188152313232,
"logits/rejected": -2.727560043334961,
"logps/chosen": -1.3122286796569824,
"logps/rejected": -2.4676425457000732,
"loss": -0.1053,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -1.3122286796569824,
"rewards/margins": 1.1554138660430908,
"rewards/rejected": -2.4676425457000732,
"step": 215
},
{
"epoch": 0.46061240512954726,
"grad_norm": 8.86883909656063,
"learning_rate": 3.265574537815398e-07,
"logits/chosen": -2.719998359680176,
"logits/rejected": -2.7265241146087646,
"logps/chosen": -1.129214882850647,
"logps/rejected": -1.7588284015655518,
"loss": -0.1146,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.129214882850647,
"rewards/margins": 0.6296136975288391,
"rewards/rejected": -1.7588284015655518,
"step": 220
},
{
"epoch": 0.4710808688824915,
"grad_norm": 10.016548371714508,
"learning_rate": 3.1779403380910425e-07,
"logits/chosen": -2.811687707901001,
"logits/rejected": -2.7085366249084473,
"logps/chosen": -1.3638594150543213,
"logps/rejected": -2.3513691425323486,
"loss": -0.1184,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.3638594150543213,
"rewards/margins": 0.9875099062919617,
"rewards/rejected": -2.3513691425323486,
"step": 225
},
{
"epoch": 0.48154933263543576,
"grad_norm": 21.998434755048503,
"learning_rate": 3.0893973387735683e-07,
"logits/chosen": -2.6830623149871826,
"logits/rejected": -2.6364707946777344,
"logps/chosen": -1.3768701553344727,
"logps/rejected": -2.424539089202881,
"loss": -0.1267,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.3768701553344727,
"rewards/margins": 1.0476689338684082,
"rewards/rejected": -2.424539089202881,
"step": 230
},
{
"epoch": 0.49201779638838,
"grad_norm": 15.145805829963615,
"learning_rate": 3.000064234440111e-07,
"logits/chosen": -2.4627432823181152,
"logits/rejected": -2.4076457023620605,
"logps/chosen": -1.2042968273162842,
"logps/rejected": -2.078500270843506,
"loss": -0.1185,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.2042968273162842,
"rewards/margins": 0.8742035031318665,
"rewards/rejected": -2.078500270843506,
"step": 235
},
{
"epoch": 0.5024862601413242,
"grad_norm": 10.783814010063464,
"learning_rate": 2.910060778827554e-07,
"logits/chosen": -2.5319347381591797,
"logits/rejected": -2.5071659088134766,
"logps/chosen": -1.5030553340911865,
"logps/rejected": -2.248628616333008,
"loss": -0.1178,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.5030553340911865,
"rewards/margins": 0.7455734014511108,
"rewards/rejected": -2.248628616333008,
"step": 240
},
{
"epoch": 0.5129547238942685,
"grad_norm": 17.35265824839872,
"learning_rate": 2.8195076242990116e-07,
"logits/chosen": -2.629790782928467,
"logits/rejected": -2.5811970233917236,
"logps/chosen": -1.1765772104263306,
"logps/rejected": -2.45619535446167,
"loss": -0.1283,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.1765772104263306,
"rewards/margins": 1.2796180248260498,
"rewards/rejected": -2.45619535446167,
"step": 245
},
{
"epoch": 0.5234231876472127,
"grad_norm": 21.631571252123923,
"learning_rate": 2.7285261601056697e-07,
"logits/chosen": -2.559217929840088,
"logits/rejected": -2.4746146202087402,
"logps/chosen": -1.4234548807144165,
"logps/rejected": -2.4785306453704834,
"loss": -0.1032,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -1.4234548807144165,
"rewards/margins": 1.055075764656067,
"rewards/rejected": -2.4785306453704834,
"step": 250
},
{
"epoch": 0.533891651400157,
"grad_norm": 10.665899447598967,
"learning_rate": 2.6372383496608186e-07,
"logits/chosen": -2.662745237350464,
"logits/rejected": -2.6075425148010254,
"logps/chosen": -1.1387856006622314,
"logps/rejected": -2.129990577697754,
"loss": -0.1171,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.1387856006622314,
"rewards/margins": 0.9912050366401672,
"rewards/rejected": -2.129990577697754,
"step": 255
},
{
"epoch": 0.5443601151531012,
"grad_norm": 30.24510546028332,
"learning_rate": 2.5457665670441937e-07,
"logits/chosen": -2.5815982818603516,
"logits/rejected": -2.5034749507904053,
"logps/chosen": -1.3083416223526,
"logps/rejected": -2.3620128631591797,
"loss": -0.1357,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.3083416223526,
"rewards/margins": 1.0536713600158691,
"rewards/rejected": -2.3620128631591797,
"step": 260
},
{
"epoch": 0.5548285789060455,
"grad_norm": 36.43768779189584,
"learning_rate": 2.454233432955807e-07,
"logits/chosen": -2.4912540912628174,
"logits/rejected": -2.503256320953369,
"logps/chosen": -1.527940034866333,
"logps/rejected": -2.379206895828247,
"loss": -0.1328,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.527940034866333,
"rewards/margins": 0.8512667417526245,
"rewards/rejected": -2.379206895828247,
"step": 265
},
{
"epoch": 0.5652970426589898,
"grad_norm": 21.411949693790206,
"learning_rate": 2.3627616503391812e-07,
"logits/chosen": -2.429790735244751,
"logits/rejected": -2.451469898223877,
"logps/chosen": -1.8636459112167358,
"logps/rejected": -2.370575428009033,
"loss": -0.1231,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.8636459112167358,
"rewards/margins": 0.5069290399551392,
"rewards/rejected": -2.370575428009033,
"step": 270
},
{
"epoch": 0.575765506411934,
"grad_norm": 26.28588094272757,
"learning_rate": 2.2714738398943308e-07,
"logits/chosen": -2.484380006790161,
"logits/rejected": -2.3554978370666504,
"logps/chosen": -1.374985933303833,
"logps/rejected": -3.2982125282287598,
"loss": -0.1605,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.374985933303833,
"rewards/margins": 1.9232263565063477,
"rewards/rejected": -3.2982125282287598,
"step": 275
},
{
"epoch": 0.5862339701648783,
"grad_norm": 15.64313913527933,
"learning_rate": 2.1804923757009882e-07,
"logits/chosen": -2.5084352493286133,
"logits/rejected": -2.379485607147217,
"logps/chosen": -1.2756072282791138,
"logps/rejected": -2.862361431121826,
"loss": -0.1283,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.2756072282791138,
"rewards/margins": 1.586754560470581,
"rewards/rejected": -2.862361431121826,
"step": 280
},
{
"epoch": 0.5967024339178225,
"grad_norm": 31.160163808849045,
"learning_rate": 2.089939221172446e-07,
"logits/chosen": -2.464437484741211,
"logits/rejected": -2.4234752655029297,
"logps/chosen": -1.7923996448516846,
"logps/rejected": -3.0464653968811035,
"loss": -0.1107,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.7923996448516846,
"rewards/margins": 1.2540655136108398,
"rewards/rejected": -3.0464653968811035,
"step": 285
},
{
"epoch": 0.6071708976707668,
"grad_norm": 14.525714466074616,
"learning_rate": 1.9999357655598891e-07,
"logits/chosen": -2.3748879432678223,
"logits/rejected": -2.277477741241455,
"logps/chosen": -1.8306829929351807,
"logps/rejected": -2.5916666984558105,
"loss": -0.1171,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.8306829929351807,
"rewards/margins": 0.7609835863113403,
"rewards/rejected": -2.5916666984558105,
"step": 290
},
{
"epoch": 0.6176393614237111,
"grad_norm": 12.205724246076834,
"learning_rate": 1.9106026612264315e-07,
"logits/chosen": -2.5410079956054688,
"logits/rejected": -2.368896484375,
"logps/chosen": -1.3447208404541016,
"logps/rejected": -2.198944091796875,
"loss": -0.1286,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.3447208404541016,
"rewards/margins": 0.8542232513427734,
"rewards/rejected": -2.198944091796875,
"step": 295
},
{
"epoch": 0.6281078251766553,
"grad_norm": 13.048722996544631,
"learning_rate": 1.8220596619089573e-07,
"logits/chosen": -2.3360071182250977,
"logits/rejected": -2.238037347793579,
"logps/chosen": -1.3523575067520142,
"logps/rejected": -2.568472146987915,
"loss": -0.1212,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.3523575067520142,
"rewards/margins": 1.2161142826080322,
"rewards/rejected": -2.568472146987915,
"step": 300
},
{
"epoch": 0.6385762889295996,
"grad_norm": 14.965881571164392,
"learning_rate": 1.7344254621846017e-07,
"logits/chosen": -2.4726943969726562,
"logits/rejected": -2.4096648693084717,
"logps/chosen": -1.4510114192962646,
"logps/rejected": -2.0148894786834717,
"loss": -0.1379,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.4510114192962646,
"rewards/margins": 0.563878059387207,
"rewards/rejected": -2.0148894786834717,
"step": 305
},
{
"epoch": 0.6490447526825438,
"grad_norm": 14.196438929994413,
"learning_rate": 1.647817538357072e-07,
"logits/chosen": -2.434021472930908,
"logits/rejected": -2.2209062576293945,
"logps/chosen": -1.2371381521224976,
"logps/rejected": -2.5744917392730713,
"loss": -0.1344,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.2371381521224976,
"rewards/margins": 1.3373535871505737,
"rewards/rejected": -2.5744917392730713,
"step": 310
},
{
"epoch": 0.6595132164354881,
"grad_norm": 13.34734425640694,
"learning_rate": 1.562351990976095e-07,
"logits/chosen": -2.482234477996826,
"logits/rejected": -2.3337721824645996,
"logps/chosen": -1.7002273797988892,
"logps/rejected": -2.32104754447937,
"loss": -0.1352,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.7002273797988892,
"rewards/margins": 0.6208202838897705,
"rewards/rejected": -2.32104754447937,
"step": 315
},
{
"epoch": 0.6699816801884323,
"grad_norm": 54.856648124460555,
"learning_rate": 1.478143389201113e-07,
"logits/chosen": -2.4059951305389404,
"logits/rejected": -2.2941792011260986,
"logps/chosen": -1.7276633977890015,
"logps/rejected": -2.791877031326294,
"loss": -0.1395,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.7276633977890015,
"rewards/margins": 1.0642132759094238,
"rewards/rejected": -2.791877031326294,
"step": 320
},
{
"epoch": 0.6804501439413766,
"grad_norm": 16.338313868801983,
"learning_rate": 1.3953046172178413e-07,
"logits/chosen": -2.3474369049072266,
"logits/rejected": -2.2156565189361572,
"logps/chosen": -1.581926941871643,
"logps/rejected": -3.4111123085021973,
"loss": -0.1119,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -1.581926941871643,
"rewards/margins": 1.8291854858398438,
"rewards/rejected": -3.4111123085021973,
"step": 325
},
{
"epoch": 0.6909186076943209,
"grad_norm": 27.599650373019013,
"learning_rate": 1.3139467229135998e-07,
"logits/chosen": -2.424649238586426,
"logits/rejected": -2.313764810562134,
"logps/chosen": -1.4561196565628052,
"logps/rejected": -2.5569052696228027,
"loss": -0.1255,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.4561196565628052,
"rewards/margins": 1.1007858514785767,
"rewards/rejected": -2.5569052696228027,
"step": 330
},
{
"epoch": 0.7013870714472651,
"grad_norm": 13.71174939782389,
"learning_rate": 1.2341787690142435e-07,
"logits/chosen": -2.182265520095825,
"logits/rejected": -2.240220069885254,
"logps/chosen": -2.133391857147217,
"logps/rejected": -2.705057382583618,
"loss": -0.1229,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.133391857147217,
"rewards/margins": 0.571665346622467,
"rewards/rejected": -2.705057382583618,
"step": 335
},
{
"epoch": 0.7118555352002094,
"grad_norm": 22.094952760541535,
"learning_rate": 1.1561076868822755e-07,
"logits/chosen": -2.351863384246826,
"logits/rejected": -2.2399449348449707,
"logps/chosen": -1.3272682428359985,
"logps/rejected": -3.206008195877075,
"loss": -0.1347,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.3272682428359985,
"rewards/margins": 1.8787400722503662,
"rewards/rejected": -3.206008195877075,
"step": 340
},
{
"epoch": 0.7223239989531536,
"grad_norm": 14.410515050019907,
"learning_rate": 1.0798381331721107e-07,
"logits/chosen": -2.3990814685821533,
"logits/rejected": -2.256584644317627,
"logps/chosen": -1.4144983291625977,
"logps/rejected": -2.471238851547241,
"loss": -0.1205,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -1.4144983291625977,
"rewards/margins": 1.056740641593933,
"rewards/rejected": -2.471238851547241,
"step": 345
},
{
"epoch": 0.7327924627060979,
"grad_norm": 17.737495980302032,
"learning_rate": 1.0054723495346482e-07,
"logits/chosen": -2.3040690422058105,
"logits/rejected": -1.9883592128753662,
"logps/chosen": -1.8120450973510742,
"logps/rejected": -2.608847141265869,
"loss": -0.1314,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.8120450973510742,
"rewards/margins": 0.7968018651008606,
"rewards/rejected": -2.608847141265869,
"step": 350
},
{
"epoch": 0.7432609264590422,
"grad_norm": 147.24723390344812,
"learning_rate": 9.331100255592436e-08,
"logits/chosen": -2.2860684394836426,
"logits/rejected": -2.1697564125061035,
"logps/chosen": -1.6789791584014893,
"logps/rejected": -3.0929505825042725,
"loss": -0.1132,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.6789791584014893,
"rewards/margins": 1.413971185684204,
"rewards/rejected": -3.0929505825042725,
"step": 355
},
{
"epoch": 0.7537293902119864,
"grad_norm": 25.64073118103101,
"learning_rate": 8.628481651367875e-08,
"logits/chosen": -2.218724012374878,
"logits/rejected": -2.1296398639678955,
"logps/chosen": -1.6159321069717407,
"logps/rejected": -2.5099380016326904,
"loss": -0.1222,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.6159321069717407,
"rewards/margins": 0.8940057754516602,
"rewards/rejected": -2.5099380016326904,
"step": 360
},
{
"epoch": 0.7641978539649307,
"grad_norm": 12.902928067743126,
"learning_rate": 7.947809564230445e-08,
"logits/chosen": -2.3579747676849365,
"logits/rejected": -2.1200146675109863,
"logps/chosen": -1.5285216569900513,
"logps/rejected": -3.1276838779449463,
"loss": -0.145,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.5285216569900513,
"rewards/margins": 1.5991623401641846,
"rewards/rejected": -3.1276838779449463,
"step": 365
},
{
"epoch": 0.7746663177178749,
"grad_norm": 15.785906225773893,
"learning_rate": 7.289996455765748e-08,
"logits/chosen": -2.407064914703369,
"logits/rejected": -2.30210542678833,
"logps/chosen": -1.3575342893600464,
"logps/rejected": -2.2857096195220947,
"loss": -0.1468,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.3575342893600464,
"rewards/margins": 0.9281753301620483,
"rewards/rejected": -2.2857096195220947,
"step": 370
},
{
"epoch": 0.7851347814708192,
"grad_norm": 66.73212890394835,
"learning_rate": 6.655924144404906e-08,
"logits/chosen": -2.2724781036376953,
"logits/rejected": -2.2895565032958984,
"logps/chosen": -1.549403429031372,
"logps/rejected": -2.9475514888763428,
"loss": -0.1203,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.549403429031372,
"rewards/margins": 1.3981479406356812,
"rewards/rejected": -2.9475514888763428,
"step": 375
},
{
"epoch": 0.7956032452237635,
"grad_norm": 38.70646418643775,
"learning_rate": 6.046442623320145e-08,
"logits/chosen": -2.399785041809082,
"logits/rejected": -2.3085193634033203,
"logps/chosen": -1.3037949800491333,
"logps/rejected": -2.2208690643310547,
"loss": -0.1115,
"rewards/accuracies": 0.75,
"rewards/chosen": -1.3037949800491333,
"rewards/margins": 0.9170739054679871,
"rewards/rejected": -2.2208690643310547,
"step": 380
},
{
"epoch": 0.8060717089767077,
"grad_norm": 85.36661297405541,
"learning_rate": 5.4623689209832484e-08,
"logits/chosen": -2.350160598754883,
"logits/rejected": -2.304274559020996,
"logps/chosen": -1.328881859779358,
"logps/rejected": -2.426609516143799,
"loss": -0.1355,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.328881859779358,
"rewards/margins": 1.0977277755737305,
"rewards/rejected": -2.426609516143799,
"step": 385
},
{
"epoch": 0.816540172729652,
"grad_norm": 21.914030829763547,
"learning_rate": 4.904486005914027e-08,
"logits/chosen": -2.41226863861084,
"logits/rejected": -2.258676767349243,
"logps/chosen": -1.3917608261108398,
"logps/rejected": -2.255201578140259,
"loss": -0.1082,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.3917608261108398,
"rewards/margins": 0.8634408712387085,
"rewards/rejected": -2.255201578140259,
"step": 390
},
{
"epoch": 0.8270086364825961,
"grad_norm": 11.128742517489252,
"learning_rate": 4.373541737087263e-08,
"logits/chosen": -2.3416173458099365,
"logits/rejected": -2.183840274810791,
"logps/chosen": -1.4819704294204712,
"logps/rejected": -2.7424230575561523,
"loss": -0.1199,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.4819704294204712,
"rewards/margins": 1.2604527473449707,
"rewards/rejected": -2.7424230575561523,
"step": 395
},
{
"epoch": 0.8374771002355405,
"grad_norm": 13.499128871751621,
"learning_rate": 3.8702478614051345e-08,
"logits/chosen": -2.4478988647460938,
"logits/rejected": -2.2530007362365723,
"logps/chosen": -1.503173589706421,
"logps/rejected": -2.4205775260925293,
"loss": -0.1196,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.503173589706421,
"rewards/margins": 0.9174041748046875,
"rewards/rejected": -2.4205775260925293,
"step": 400
},
{
"epoch": 0.8479455639884846,
"grad_norm": 46.00477049565933,
"learning_rate": 3.3952790595787986e-08,
"logits/chosen": -2.428011894226074,
"logits/rejected": -2.195507526397705,
"logps/chosen": -1.488608956336975,
"logps/rejected": -3.2226758003234863,
"loss": -0.1329,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.488608956336975,
"rewards/margins": 1.7340667247772217,
"rewards/rejected": -3.2226758003234863,
"step": 405
},
{
"epoch": 0.8584140277414289,
"grad_norm": 15.147104463873568,
"learning_rate": 2.9492720416985e-08,
"logits/chosen": -2.3303403854370117,
"logits/rejected": -2.2017440795898438,
"logps/chosen": -1.4811475276947021,
"logps/rejected": -3.1947004795074463,
"loss": -0.1345,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.4811475276947021,
"rewards/margins": 1.7135528326034546,
"rewards/rejected": -3.1947004795074463,
"step": 410
},
{
"epoch": 0.8688824914943732,
"grad_norm": 32.54043568171796,
"learning_rate": 2.5328246937043525e-08,
"logits/chosen": -2.3127968311309814,
"logits/rejected": -2.2588541507720947,
"logps/chosen": -1.7812035083770752,
"logps/rejected": -2.363581418991089,
"loss": -0.1144,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.7812035083770752,
"rewards/margins": 0.5823777318000793,
"rewards/rejected": -2.363581418991089,
"step": 415
},
{
"epoch": 0.8793509552473174,
"grad_norm": 16.364361772449453,
"learning_rate": 2.1464952759020856e-08,
"logits/chosen": -2.2632575035095215,
"logits/rejected": -2.1989002227783203,
"logps/chosen": -1.3589812517166138,
"logps/rejected": -2.665706157684326,
"loss": -0.1402,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.3589812517166138,
"rewards/margins": 1.3067247867584229,
"rewards/rejected": -2.665706157684326,
"step": 420
},
{
"epoch": 0.8898194190002617,
"grad_norm": 24.459898730148318,
"learning_rate": 1.7908016745981856e-08,
"logits/chosen": -2.4754738807678223,
"logits/rejected": -2.2798328399658203,
"logps/chosen": -1.1175788640975952,
"logps/rejected": -2.699981212615967,
"loss": -0.1488,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.1175788640975952,
"rewards/margins": 1.5824021100997925,
"rewards/rejected": -2.699981212615967,
"step": 425
},
{
"epoch": 0.9002878827532059,
"grad_norm": 20.15635024127512,
"learning_rate": 1.4662207078575684e-08,
"logits/chosen": -2.3345370292663574,
"logits/rejected": -2.0842413902282715,
"logps/chosen": -1.416595458984375,
"logps/rejected": -2.772968053817749,
"loss": -0.1388,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.416595458984375,
"rewards/margins": 1.3563728332519531,
"rewards/rejected": -2.772968053817749,
"step": 430
},
{
"epoch": 0.9107563465061502,
"grad_norm": 25.1930677516597,
"learning_rate": 1.1731874863145142e-08,
"logits/chosen": -2.2844507694244385,
"logits/rejected": -2.2151055335998535,
"logps/chosen": -1.5157394409179688,
"logps/rejected": -2.6189606189727783,
"loss": -0.1335,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.5157394409179688,
"rewards/margins": 1.1032211780548096,
"rewards/rejected": -2.6189606189727783,
"step": 435
},
{
"epoch": 0.9212248102590945,
"grad_norm": 94.3340138658481,
"learning_rate": 9.12094829893642e-09,
"logits/chosen": -2.3165175914764404,
"logits/rejected": -2.216142177581787,
"logps/chosen": -1.6333303451538086,
"logps/rejected": -2.6600871086120605,
"loss": -0.1365,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.6333303451538086,
"rewards/margins": 1.0267568826675415,
"rewards/rejected": -2.6600871086120605,
"step": 440
},
{
"epoch": 0.9316932740120387,
"grad_norm": 37.68445074958579,
"learning_rate": 6.832927412229017e-09,
"logits/chosen": -2.173781394958496,
"logits/rejected": -2.0770103931427,
"logps/chosen": -1.7893623113632202,
"logps/rejected": -2.782428026199341,
"loss": -0.1362,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.7893623113632202,
"rewards/margins": 0.9930654764175415,
"rewards/rejected": -2.782428026199341,
"step": 445
},
{
"epoch": 0.942161737764983,
"grad_norm": 65.20343901413754,
"learning_rate": 4.8708793644441086e-09,
"logits/chosen": -2.4339003562927246,
"logits/rejected": -2.235538959503174,
"logps/chosen": -1.1971713304519653,
"logps/rejected": -2.028782367706299,
"loss": -0.1358,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.1971713304519653,
"rewards/margins": 0.8316110372543335,
"rewards/rejected": -2.028782367706299,
"step": 450
},
{
"epoch": 0.9526302015179272,
"grad_norm": 29.43315569875579,
"learning_rate": 3.2374343405217884e-09,
"logits/chosen": -2.406641721725464,
"logits/rejected": -2.2101848125457764,
"logps/chosen": -1.7016384601593018,
"logps/rejected": -2.7031149864196777,
"loss": -0.1135,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -1.7016384601593018,
"rewards/margins": 1.0014766454696655,
"rewards/rejected": -2.7031149864196777,
"step": 455
},
{
"epoch": 0.9630986652708715,
"grad_norm": 58.96075147402077,
"learning_rate": 1.9347820230782295e-09,
"logits/chosen": -2.5120296478271484,
"logits/rejected": -2.0890145301818848,
"logps/chosen": -1.2953853607177734,
"logps/rejected": -3.554316759109497,
"loss": -0.1399,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -1.2953853607177734,
"rewards/margins": 2.2589316368103027,
"rewards/rejected": -3.554316759109497,
"step": 460
},
{
"epoch": 0.9735671290238157,
"grad_norm": 34.51709660208177,
"learning_rate": 9.64668657069706e-10,
"logits/chosen": -2.295931339263916,
"logits/rejected": -2.02848219871521,
"logps/chosen": -1.3973679542541504,
"logps/rejected": -3.212092638015747,
"loss": -0.145,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -1.3973679542541504,
"rewards/margins": 1.814724326133728,
"rewards/rejected": -3.212092638015747,
"step": 465
},
{
"epoch": 0.98403559277676,
"grad_norm": 11.228558025822675,
"learning_rate": 3.2839470889836627e-10,
"logits/chosen": -2.295358180999756,
"logits/rejected": -2.151076316833496,
"logps/chosen": -1.4753532409667969,
"logps/rejected": -2.6792643070220947,
"loss": -0.131,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.4753532409667969,
"rewards/margins": 1.2039110660552979,
"rewards/rejected": -2.6792643070220947,
"step": 470
},
{
"epoch": 0.9945040565297043,
"grad_norm": 15.547524058275151,
"learning_rate": 2.6813123097352287e-11,
"logits/chosen": -2.3587679862976074,
"logits/rejected": -2.1848464012145996,
"logps/chosen": -1.336055874824524,
"logps/rejected": -2.380079746246338,
"loss": -0.1371,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.336055874824524,
"rewards/margins": 1.044023871421814,
"rewards/rejected": -2.380079746246338,
"step": 475
},
{
"epoch": 0.998691442030882,
"step": 477,
"total_flos": 0.0,
"train_loss": -0.1067421951134,
"train_runtime": 20026.0763,
"train_samples_per_second": 3.053,
"train_steps_per_second": 0.024
}
],
"logging_steps": 5,
"max_steps": 477,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 239,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}