deepseek-8b-orpo-full / trainer_state.json
zfz1's picture
Model save
9adce4c verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.0,
"eval_steps": 10000,
"global_step": 1250,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.016,
"grad_norm": 7.157915223499978,
"learning_rate": 1.6000000000000001e-06,
"log_odds_chosen": 0.08687938749790192,
"log_odds_ratio": -0.6765035390853882,
"logits/chosen": 42.733116149902344,
"logits/rejected": 42.47905731201172,
"logps/chosen": -1.0179941654205322,
"logps/rejected": -1.0789813995361328,
"loss": 1.6603,
"nll_loss": 1.6577844619750977,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.10179942846298218,
"rewards/margins": 0.0060987225733697414,
"rewards/rejected": -0.10789815336465836,
"step": 10
},
{
"epoch": 0.032,
"grad_norm": 4.787027456951044,
"learning_rate": 3.2000000000000003e-06,
"log_odds_chosen": 0.16144487261772156,
"log_odds_ratio": -0.6409393548965454,
"logits/chosen": 39.06427764892578,
"logits/rejected": 37.91705322265625,
"logps/chosen": -0.8086442947387695,
"logps/rejected": -0.9057012796401978,
"loss": 1.4104,
"nll_loss": 1.320190191268921,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.08086444437503815,
"rewards/margins": 0.009705692529678345,
"rewards/rejected": -0.09057013690471649,
"step": 20
},
{
"epoch": 0.048,
"grad_norm": 3.619155990390428,
"learning_rate": 4.800000000000001e-06,
"log_odds_chosen": -0.09232249855995178,
"log_odds_ratio": -0.7732639312744141,
"logits/chosen": 33.01856231689453,
"logits/rejected": 34.8216552734375,
"logps/chosen": -0.5950765013694763,
"logps/rejected": -0.5527861714363098,
"loss": 0.9477,
"nll_loss": 0.9017837643623352,
"rewards/accuracies": 0.4375,
"rewards/chosen": -0.05950764939188957,
"rewards/margins": -0.004229038953781128,
"rewards/rejected": -0.05527861788868904,
"step": 30
},
{
"epoch": 0.064,
"grad_norm": 0.8193426378065174,
"learning_rate": 6.4000000000000006e-06,
"log_odds_chosen": 0.13158965110778809,
"log_odds_ratio": -0.6801391243934631,
"logits/chosen": 34.19690704345703,
"logits/rejected": 33.757015228271484,
"logps/chosen": -0.38923197984695435,
"logps/rejected": -0.42596691846847534,
"loss": 0.7871,
"nll_loss": 0.7480510473251343,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.038923196494579315,
"rewards/margins": 0.003673497587442398,
"rewards/rejected": -0.04259669408202171,
"step": 40
},
{
"epoch": 0.08,
"grad_norm": 0.837886863416422,
"learning_rate": 8.000000000000001e-06,
"log_odds_chosen": 0.010798773728311062,
"log_odds_ratio": -0.7395683526992798,
"logits/chosen": 36.046871185302734,
"logits/rejected": 35.39847946166992,
"logps/chosen": -0.3819189667701721,
"logps/rejected": -0.36964151263237,
"loss": 0.7614,
"nll_loss": 0.6979484558105469,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.038191892206668854,
"rewards/margins": -0.0012277422938495874,
"rewards/rejected": -0.03696415573358536,
"step": 50
},
{
"epoch": 0.096,
"grad_norm": 0.934326744395183,
"learning_rate": 9.600000000000001e-06,
"log_odds_chosen": 0.17773893475532532,
"log_odds_ratio": -0.6542357206344604,
"logits/chosen": 37.91405487060547,
"logits/rejected": 38.55655288696289,
"logps/chosen": -0.35373184084892273,
"logps/rejected": -0.40366658568382263,
"loss": 0.7369,
"nll_loss": 0.6853512525558472,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.03537318482995033,
"rewards/margins": 0.0049934713169932365,
"rewards/rejected": -0.040366653352975845,
"step": 60
},
{
"epoch": 0.112,
"grad_norm": 0.8140121698145116,
"learning_rate": 1.1200000000000001e-05,
"log_odds_chosen": 0.27008742094039917,
"log_odds_ratio": -0.6275990009307861,
"logits/chosen": 37.89257049560547,
"logits/rejected": 36.801631927490234,
"logps/chosen": -0.3447234630584717,
"logps/rejected": -0.44380050897598267,
"loss": 0.7405,
"nll_loss": 0.6623181104660034,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.03447234630584717,
"rewards/margins": 0.009907705709338188,
"rewards/rejected": -0.04438004642724991,
"step": 70
},
{
"epoch": 0.128,
"grad_norm": 0.8866561757309486,
"learning_rate": 1.2800000000000001e-05,
"log_odds_chosen": 0.7000471949577332,
"log_odds_ratio": -0.46674174070358276,
"logits/chosen": 39.06570816040039,
"logits/rejected": 38.14867401123047,
"logps/chosen": -0.3046243488788605,
"logps/rejected": -0.541035532951355,
"loss": 0.6756,
"nll_loss": 0.5991424322128296,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -0.030462438240647316,
"rewards/margins": 0.023641113191843033,
"rewards/rejected": -0.0541035458445549,
"step": 80
},
{
"epoch": 0.144,
"grad_norm": 1.1050711696617728,
"learning_rate": 1.4400000000000001e-05,
"log_odds_chosen": 0.4890363812446594,
"log_odds_ratio": -0.5526093244552612,
"logits/chosen": 39.77109146118164,
"logits/rejected": 41.683753967285156,
"logps/chosen": -0.36524319648742676,
"logps/rejected": -0.5409058928489685,
"loss": 0.6761,
"nll_loss": 0.6159157156944275,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.036524318158626556,
"rewards/margins": 0.017566271126270294,
"rewards/rejected": -0.05409059673547745,
"step": 90
},
{
"epoch": 0.16,
"grad_norm": 1.1283739516296225,
"learning_rate": 1.6000000000000003e-05,
"log_odds_chosen": 0.5452580451965332,
"log_odds_ratio": -0.5334872603416443,
"logits/chosen": 41.09711837768555,
"logits/rejected": 39.49372100830078,
"logps/chosen": -0.3422545790672302,
"logps/rejected": -0.5715359449386597,
"loss": 0.6441,
"nll_loss": 0.5890800952911377,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -0.0342254601418972,
"rewards/margins": 0.022928137332201004,
"rewards/rejected": -0.05715359374880791,
"step": 100
},
{
"epoch": 0.176,
"grad_norm": 1.044328943254365,
"learning_rate": 1.76e-05,
"log_odds_chosen": 0.6057835817337036,
"log_odds_ratio": -0.5027212500572205,
"logits/chosen": 41.39733123779297,
"logits/rejected": 39.85953140258789,
"logps/chosen": -0.35752102732658386,
"logps/rejected": -0.590412437915802,
"loss": 0.6068,
"nll_loss": 0.5553559064865112,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.035752106457948685,
"rewards/margins": 0.023289144039154053,
"rewards/rejected": -0.059041254222393036,
"step": 110
},
{
"epoch": 0.192,
"grad_norm": 1.0225837980847625,
"learning_rate": 1.9200000000000003e-05,
"log_odds_chosen": 0.5523650050163269,
"log_odds_ratio": -0.5369630455970764,
"logits/chosen": 43.63589859008789,
"logits/rejected": 41.659324645996094,
"logps/chosen": -0.3336974084377289,
"logps/rejected": -0.5324398279190063,
"loss": 0.5734,
"nll_loss": 0.5098426938056946,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.03336974233388901,
"rewards/margins": 0.01987423747777939,
"rewards/rejected": -0.05324398726224899,
"step": 120
},
{
"epoch": 0.208,
"grad_norm": 1.1101012093036131,
"learning_rate": 1.9999025240093045e-05,
"log_odds_chosen": 0.7986687421798706,
"log_odds_ratio": -0.471863329410553,
"logits/chosen": 43.426692962646484,
"logits/rejected": 40.31127166748047,
"logps/chosen": -0.35013002157211304,
"logps/rejected": -0.6971026659011841,
"loss": 0.5401,
"nll_loss": 0.508377730846405,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.03501300513744354,
"rewards/margins": 0.03469725698232651,
"rewards/rejected": -0.06971026211977005,
"step": 130
},
{
"epoch": 0.224,
"grad_norm": 1.5810963333813628,
"learning_rate": 1.9991228300988586e-05,
"log_odds_chosen": 0.9270819425582886,
"log_odds_ratio": -0.4264460504055023,
"logits/chosen": 40.22356414794922,
"logits/rejected": 38.093299865722656,
"logps/chosen": -0.3407229781150818,
"logps/rejected": -0.7779335975646973,
"loss": 0.5243,
"nll_loss": 0.4749884605407715,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.03407229855656624,
"rewards/margins": 0.04372106492519379,
"rewards/rejected": -0.07779336720705032,
"step": 140
},
{
"epoch": 0.24,
"grad_norm": 0.9914026891453022,
"learning_rate": 1.9975640502598243e-05,
"log_odds_chosen": 0.7473500967025757,
"log_odds_ratio": -0.4457695484161377,
"logits/chosen": 38.593631744384766,
"logits/rejected": 40.06364059448242,
"logps/chosen": -0.35150039196014404,
"logps/rejected": -0.6378719210624695,
"loss": 0.507,
"nll_loss": 0.45233845710754395,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.035150036215782166,
"rewards/margins": 0.02863715961575508,
"rewards/rejected": -0.06378719955682755,
"step": 150
},
{
"epoch": 0.256,
"grad_norm": 1.0114005741143277,
"learning_rate": 1.9952273999818312e-05,
"log_odds_chosen": 0.7789567708969116,
"log_odds_ratio": -0.47668343782424927,
"logits/chosen": 40.021446228027344,
"logits/rejected": 39.03896713256836,
"logps/chosen": -0.32147353887557983,
"logps/rejected": -0.6195149421691895,
"loss": 0.4762,
"nll_loss": 0.40015798807144165,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.0321473553776741,
"rewards/margins": 0.02980414591729641,
"rewards/rejected": -0.061951495707035065,
"step": 160
},
{
"epoch": 0.272,
"grad_norm": 1.1072013212789444,
"learning_rate": 1.9921147013144782e-05,
"log_odds_chosen": 0.9789760708808899,
"log_odds_ratio": -0.3996427059173584,
"logits/chosen": 38.469261169433594,
"logits/rejected": 39.36458206176758,
"logps/chosen": -0.31273895502090454,
"logps/rejected": -0.7282775640487671,
"loss": 0.4569,
"nll_loss": 0.4199884533882141,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -0.031273894011974335,
"rewards/margins": 0.041553862392902374,
"rewards/rejected": -0.07282774895429611,
"step": 170
},
{
"epoch": 0.288,
"grad_norm": 1.0540475357463217,
"learning_rate": 1.988228381446553e-05,
"log_odds_chosen": 0.9965242147445679,
"log_odds_ratio": -0.4183652997016907,
"logits/chosen": 40.18573760986328,
"logits/rejected": 40.521148681640625,
"logps/chosen": -0.33785638213157654,
"logps/rejected": -0.7628054022789001,
"loss": 0.4311,
"nll_loss": 0.43037042021751404,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.03378564119338989,
"rewards/margins": 0.04249490052461624,
"rewards/rejected": -0.07628054916858673,
"step": 180
},
{
"epoch": 0.304,
"grad_norm": 1.2807805662820098,
"learning_rate": 1.983571470813386e-05,
"log_odds_chosen": 0.9251772165298462,
"log_odds_ratio": -0.43279653787612915,
"logits/chosen": 39.66691970825195,
"logits/rejected": 39.22489929199219,
"logps/chosen": -0.3330400586128235,
"logps/rejected": -0.6816459894180298,
"loss": 0.4062,
"nll_loss": 0.36826083064079285,
"rewards/accuracies": 0.8125,
"rewards/chosen": -0.03330400586128235,
"rewards/margins": 0.03486059606075287,
"rewards/rejected": -0.06816460937261581,
"step": 190
},
{
"epoch": 0.32,
"grad_norm": 1.247095232534241,
"learning_rate": 1.9781476007338058e-05,
"log_odds_chosen": 1.1870229244232178,
"log_odds_ratio": -0.331691175699234,
"logits/chosen": 36.31359100341797,
"logits/rejected": 38.259849548339844,
"logps/chosen": -0.2683125436306,
"logps/rejected": -0.6734791994094849,
"loss": 0.3637,
"nll_loss": 0.29906129837036133,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.026831254363059998,
"rewards/margins": 0.04051666334271431,
"rewards/rejected": -0.0673479214310646,
"step": 200
},
{
"epoch": 0.336,
"grad_norm": 1.023611287110811,
"learning_rate": 1.9719610005785466e-05,
"log_odds_chosen": 1.10099196434021,
"log_odds_ratio": -0.3911464810371399,
"logits/chosen": 36.06108856201172,
"logits/rejected": 36.09987258911133,
"logps/chosen": -0.3143238425254822,
"logps/rejected": -0.7127642035484314,
"loss": 0.3608,
"nll_loss": 0.33582496643066406,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -0.0314323827624321,
"rewards/margins": 0.03984404355287552,
"rewards/rejected": -0.07127643376588821,
"step": 210
},
{
"epoch": 0.352,
"grad_norm": 1.0618187057425008,
"learning_rate": 1.9650164944723116e-05,
"log_odds_chosen": 1.4383821487426758,
"log_odds_ratio": -0.313861608505249,
"logits/chosen": 34.82101821899414,
"logits/rejected": 38.44565200805664,
"logps/chosen": -0.28008711338043213,
"logps/rejected": -0.8806719779968262,
"loss": 0.3331,
"nll_loss": 0.2972494959831238,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.028008710592985153,
"rewards/margins": 0.06005847454071045,
"rewards/rejected": -0.0880671888589859,
"step": 220
},
{
"epoch": 0.368,
"grad_norm": 0.9398000108663768,
"learning_rate": 1.9573194975320672e-05,
"log_odds_chosen": 1.4377901554107666,
"log_odds_ratio": -0.34182658791542053,
"logits/chosen": 33.84839630126953,
"logits/rejected": 35.625667572021484,
"logps/chosen": -0.30297523736953735,
"logps/rejected": -0.9298135042190552,
"loss": 0.3403,
"nll_loss": 0.2947668433189392,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.030297517776489258,
"rewards/margins": 0.0626838356256485,
"rewards/rejected": -0.09298135340213776,
"step": 230
},
{
"epoch": 0.384,
"grad_norm": 1.103321212338863,
"learning_rate": 1.9488760116444966e-05,
"log_odds_chosen": 1.4992961883544922,
"log_odds_ratio": -0.3107326030731201,
"logits/chosen": 35.9039421081543,
"logits/rejected": 36.633705139160156,
"logps/chosen": -0.2847585380077362,
"logps/rejected": -0.9570012092590332,
"loss": 0.3166,
"nll_loss": 0.30076318979263306,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.02847585640847683,
"rewards/margins": 0.06722426414489746,
"rewards/rejected": -0.09570012241601944,
"step": 240
},
{
"epoch": 0.4,
"grad_norm": 0.9188155900142966,
"learning_rate": 1.9396926207859085e-05,
"log_odds_chosen": 1.232299566268921,
"log_odds_ratio": -0.36939191818237305,
"logits/chosen": 36.425289154052734,
"logits/rejected": 38.76382827758789,
"logps/chosen": -0.306942343711853,
"logps/rejected": -0.8188571929931641,
"loss": 0.3212,
"nll_loss": 0.2955878674983978,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.030694235116243362,
"rewards/margins": 0.05119148641824722,
"rewards/rejected": -0.08188571780920029,
"step": 250
},
{
"epoch": 0.416,
"grad_norm": 1.1371525474896846,
"learning_rate": 1.9297764858882516e-05,
"log_odds_chosen": 1.5188441276550293,
"log_odds_ratio": -0.3032824695110321,
"logits/chosen": 33.87938690185547,
"logits/rejected": 34.950599670410156,
"logps/chosen": -0.2652571499347687,
"logps/rejected": -0.904030442237854,
"loss": 0.282,
"nll_loss": 0.2362031489610672,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.026525715366005898,
"rewards/margins": 0.06387733668088913,
"rewards/rejected": -0.09040305018424988,
"step": 260
},
{
"epoch": 0.432,
"grad_norm": 0.9722134512964291,
"learning_rate": 1.9191353392552346e-05,
"log_odds_chosen": 1.6056066751480103,
"log_odds_ratio": -0.273274302482605,
"logits/chosen": 31.2882080078125,
"logits/rejected": 32.222434997558594,
"logps/chosen": -0.2462386190891266,
"logps/rejected": -0.843704342842102,
"loss": 0.2621,
"nll_loss": 0.23743471503257751,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.02462385967373848,
"rewards/margins": 0.059746574610471725,
"rewards/rejected": -0.0843704342842102,
"step": 270
},
{
"epoch": 0.448,
"grad_norm": 0.9536358910314567,
"learning_rate": 1.907777478532909e-05,
"log_odds_chosen": 1.3453700542449951,
"log_odds_ratio": -0.3348150849342346,
"logits/chosen": 34.44597625732422,
"logits/rejected": 35.61604690551758,
"logps/chosen": -0.25559094548225403,
"logps/rejected": -0.7815641164779663,
"loss": 0.2815,
"nll_loss": 0.25891661643981934,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -0.025559093803167343,
"rewards/margins": 0.052597321569919586,
"rewards/rejected": -0.07815642654895782,
"step": 280
},
{
"epoch": 0.464,
"grad_norm": 1.4793964733191685,
"learning_rate": 1.895711760239413e-05,
"log_odds_chosen": 1.8677561283111572,
"log_odds_ratio": -0.2598610520362854,
"logits/chosen": 30.939416885375977,
"logits/rejected": 32.69816589355469,
"logps/chosen": -0.25819864869117737,
"logps/rejected": -1.080700159072876,
"loss": 0.2526,
"nll_loss": 0.24270136654376984,
"rewards/accuracies": 0.9375,
"rewards/chosen": -0.025819867849349976,
"rewards/margins": 0.08225013315677643,
"rewards/rejected": -0.1080700010061264,
"step": 290
},
{
"epoch": 0.48,
"grad_norm": 1.2223140757678745,
"learning_rate": 1.8829475928589272e-05,
"log_odds_chosen": 1.873132348060608,
"log_odds_ratio": -0.21418969333171844,
"logits/chosen": 30.050670623779297,
"logits/rejected": 31.51558494567871,
"logps/chosen": -0.24227066338062286,
"logps/rejected": -1.0071208477020264,
"loss": 0.2446,
"nll_loss": 0.20659121870994568,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.024227067828178406,
"rewards/margins": 0.07648502290248871,
"rewards/rejected": -0.10071208328008652,
"step": 300
},
{
"epoch": 0.496,
"grad_norm": 0.8813575124105774,
"learning_rate": 1.869494929505219e-05,
"log_odds_chosen": 1.9893770217895508,
"log_odds_ratio": -0.22330129146575928,
"logits/chosen": 33.173133850097656,
"logits/rejected": 34.073055267333984,
"logps/chosen": -0.2551695704460144,
"logps/rejected": -1.1849584579467773,
"loss": 0.232,
"nll_loss": 0.2051832675933838,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.025516962632536888,
"rewards/margins": 0.09297889471054077,
"rewards/rejected": -0.11849584430456161,
"step": 310
},
{
"epoch": 0.512,
"grad_norm": 0.9094503804430618,
"learning_rate": 1.855364260160507e-05,
"log_odds_chosen": 1.7108672857284546,
"log_odds_ratio": -0.2914069592952728,
"logits/chosen": 31.070571899414062,
"logits/rejected": 31.337520599365234,
"logps/chosen": -0.26724526286125183,
"logps/rejected": -0.9716242551803589,
"loss": 0.2355,
"nll_loss": 0.2270585000514984,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.026724526658654213,
"rewards/margins": 0.07043790072202682,
"rewards/rejected": -0.09716243296861649,
"step": 320
},
{
"epoch": 0.528,
"grad_norm": 0.9058310290840669,
"learning_rate": 1.8405666034956842e-05,
"log_odds_chosen": 2.1886684894561768,
"log_odds_ratio": -0.19184692203998566,
"logits/chosen": 30.936025619506836,
"logits/rejected": 31.9810848236084,
"logps/chosen": -0.2256220132112503,
"logps/rejected": -1.2340004444122314,
"loss": 0.2216,
"nll_loss": 0.18992918729782104,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.02256220206618309,
"rewards/margins": 0.100837841629982,
"rewards/rejected": -0.12340004742145538,
"step": 330
},
{
"epoch": 0.544,
"grad_norm": 0.7923767872883756,
"learning_rate": 1.8251134982782952e-05,
"log_odds_chosen": 1.8938252925872803,
"log_odds_ratio": -0.23439836502075195,
"logits/chosen": 30.1374454498291,
"logits/rejected": 31.97110939025879,
"logps/chosen": -0.25573670864105225,
"logps/rejected": -1.058814525604248,
"loss": 0.2256,
"nll_loss": 0.2036077082157135,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -0.025573670864105225,
"rewards/margins": 0.08030776679515839,
"rewards/rejected": -0.1058814525604248,
"step": 340
},
{
"epoch": 0.56,
"grad_norm": 0.941711390597777,
"learning_rate": 1.8090169943749477e-05,
"log_odds_chosen": 1.890398621559143,
"log_odds_ratio": -0.23317229747772217,
"logits/chosen": 29.39468002319336,
"logits/rejected": 31.831829071044922,
"logps/chosen": -0.26384299993515015,
"logps/rejected": -1.0387157201766968,
"loss": 0.2102,
"nll_loss": 0.19437000155448914,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.026384299620985985,
"rewards/margins": 0.0774872750043869,
"rewards/rejected": -0.10387158393859863,
"step": 350
},
{
"epoch": 0.576,
"grad_norm": 0.9511427594232491,
"learning_rate": 1.792289643355191e-05,
"log_odds_chosen": 2.1849417686462402,
"log_odds_ratio": -0.20272669196128845,
"logits/chosen": 28.536510467529297,
"logits/rejected": 31.184783935546875,
"logps/chosen": -0.21713824570178986,
"logps/rejected": -1.1126606464385986,
"loss": 0.196,
"nll_loss": 0.1743771880865097,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.021713824942708015,
"rewards/margins": 0.08955223858356476,
"rewards/rejected": -0.11126606166362762,
"step": 360
},
{
"epoch": 0.592,
"grad_norm": 1.3234030569644144,
"learning_rate": 1.7749444887041797e-05,
"log_odds_chosen": 2.2127633094787598,
"log_odds_ratio": -0.1819957047700882,
"logits/chosen": 28.459209442138672,
"logits/rejected": 30.699054718017578,
"logps/chosen": -0.19446463882923126,
"logps/rejected": -1.064205527305603,
"loss": 0.2095,
"nll_loss": 0.157680481672287,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.019446464255452156,
"rewards/margins": 0.08697408437728882,
"rewards/rejected": -0.10642055422067642,
"step": 370
},
{
"epoch": 0.608,
"grad_norm": 0.7887424541285082,
"learning_rate": 1.7569950556517566e-05,
"log_odds_chosen": 2.0950214862823486,
"log_odds_ratio": -0.2056867778301239,
"logits/chosen": 26.343616485595703,
"logits/rejected": 30.175817489624023,
"logps/chosen": -0.22634263336658478,
"logps/rejected": -1.0738613605499268,
"loss": 0.1995,
"nll_loss": 0.1639939546585083,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.022634264081716537,
"rewards/margins": 0.08475185930728912,
"rewards/rejected": -0.10738612711429596,
"step": 380
},
{
"epoch": 0.624,
"grad_norm": 0.8925541324230298,
"learning_rate": 1.7384553406258842e-05,
"log_odds_chosen": 2.156597852706909,
"log_odds_ratio": -0.1930965632200241,
"logits/chosen": 26.64847755432129,
"logits/rejected": 28.833852767944336,
"logps/chosen": -0.2270650863647461,
"logps/rejected": -1.0798944234848022,
"loss": 0.2017,
"nll_loss": 0.1784597635269165,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.022706512361764908,
"rewards/margins": 0.08528293669223785,
"rewards/rejected": -0.10798945277929306,
"step": 390
},
{
"epoch": 0.64,
"grad_norm": 1.0272557886917288,
"learning_rate": 1.7193398003386514e-05,
"log_odds_chosen": 2.46358323097229,
"log_odds_ratio": -0.17916469275951385,
"logits/chosen": 28.102930068969727,
"logits/rejected": 29.97959327697754,
"logps/chosen": -0.21100203692913055,
"logps/rejected": -1.2793248891830444,
"loss": 0.1938,
"nll_loss": 0.1722893863916397,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.021100204437971115,
"rewards/margins": 0.10683228820562363,
"rewards/rejected": -0.12793250381946564,
"step": 400
},
{
"epoch": 0.656,
"grad_norm": 0.7328015065358303,
"learning_rate": 1.6996633405133656e-05,
"log_odds_chosen": 2.6361804008483887,
"log_odds_ratio": -0.14345327019691467,
"logits/chosen": 27.147369384765625,
"logits/rejected": 29.60892105102539,
"logps/chosen": -0.18612393736839294,
"logps/rejected": -1.346756935119629,
"loss": 0.1836,
"nll_loss": 0.14793823659420013,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.018612395972013474,
"rewards/margins": 0.11606329679489136,
"rewards/rejected": -0.13467569649219513,
"step": 410
},
{
"epoch": 0.672,
"grad_norm": 0.7570321808098591,
"learning_rate": 1.6794413042615168e-05,
"log_odds_chosen": 2.324608325958252,
"log_odds_ratio": -0.16639328002929688,
"logits/chosen": 26.477365493774414,
"logits/rejected": 29.83447265625,
"logps/chosen": -0.19487398862838745,
"logps/rejected": -1.086843729019165,
"loss": 0.1708,
"nll_loss": 0.15096671879291534,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.019487399607896805,
"rewards/margins": 0.08919696509838104,
"rewards/rejected": -0.10868436098098755,
"step": 420
},
{
"epoch": 0.688,
"grad_norm": 0.8589237619818589,
"learning_rate": 1.6586894601186804e-05,
"log_odds_chosen": 2.4892306327819824,
"log_odds_ratio": -0.1399163007736206,
"logits/chosen": 26.335521697998047,
"logits/rejected": 28.351062774658203,
"logps/chosen": -0.185403510928154,
"logps/rejected": -1.1550278663635254,
"loss": 0.1719,
"nll_loss": 0.15061026811599731,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.01854034885764122,
"rewards/margins": 0.09696242958307266,
"rewards/rejected": -0.11550277471542358,
"step": 430
},
{
"epoch": 0.704,
"grad_norm": 0.9410946432421042,
"learning_rate": 1.63742398974869e-05,
"log_odds_chosen": 2.6492185592651367,
"log_odds_ratio": -0.11772862821817398,
"logits/chosen": 25.387847900390625,
"logits/rejected": 27.318252563476562,
"logps/chosen": -0.18541622161865234,
"logps/rejected": -1.3411946296691895,
"loss": 0.168,
"nll_loss": 0.1478230208158493,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.018541622906923294,
"rewards/margins": 0.11557783931493759,
"rewards/rejected": -0.13411946594715118,
"step": 440
},
{
"epoch": 0.72,
"grad_norm": 0.913121815013067,
"learning_rate": 1.6156614753256583e-05,
"log_odds_chosen": 2.5952377319335938,
"log_odds_ratio": -0.13427497446537018,
"logits/chosen": 25.367713928222656,
"logits/rejected": 26.90452003479004,
"logps/chosen": -0.1814824938774109,
"logps/rejected": -1.2440539598464966,
"loss": 0.1621,
"nll_loss": 0.13832086324691772,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.018148250877857208,
"rewards/margins": 0.10625714063644409,
"rewards/rejected": -0.1244053840637207,
"step": 450
},
{
"epoch": 0.736,
"grad_norm": 0.8697507877833889,
"learning_rate": 1.5934188866037017e-05,
"log_odds_chosen": 2.944627046585083,
"log_odds_ratio": -0.11624413728713989,
"logits/chosen": 25.267337799072266,
"logits/rejected": 29.403518676757812,
"logps/chosen": -0.17217525839805603,
"logps/rejected": -1.3934593200683594,
"loss": 0.1569,
"nll_loss": 0.14144101738929749,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.017217526212334633,
"rewards/margins": 0.12212841212749481,
"rewards/rejected": -0.1393459290266037,
"step": 460
},
{
"epoch": 0.752,
"grad_norm": 0.6621793550764674,
"learning_rate": 1.570713567684432e-05,
"log_odds_chosen": 2.606139659881592,
"log_odds_ratio": -0.1426299810409546,
"logits/chosen": 29.288564682006836,
"logits/rejected": 29.074045181274414,
"logps/chosen": -0.19370333850383759,
"logps/rejected": -1.2478258609771729,
"loss": 0.1792,
"nll_loss": 0.16615137457847595,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.019370336085557938,
"rewards/margins": 0.10541225969791412,
"rewards/rejected": -0.12478259950876236,
"step": 470
},
{
"epoch": 0.768,
"grad_norm": 0.6543673919224716,
"learning_rate": 1.5475632234925505e-05,
"log_odds_chosen": 2.9211063385009766,
"log_odds_ratio": -0.105687215924263,
"logits/chosen": 24.23305892944336,
"logits/rejected": 27.091516494750977,
"logps/chosen": -0.1637393683195114,
"logps/rejected": -1.3047716617584229,
"loss": 0.1509,
"nll_loss": 0.12331098318099976,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.01637393794953823,
"rewards/margins": 0.11410322040319443,
"rewards/rejected": -0.1304771602153778,
"step": 480
},
{
"epoch": 0.784,
"grad_norm": 0.741926124626365,
"learning_rate": 1.5239859059700794e-05,
"log_odds_chosen": 2.723001003265381,
"log_odds_ratio": -0.13461025059223175,
"logits/chosen": 25.58415412902832,
"logits/rejected": 27.281457901000977,
"logps/chosen": -0.17167863249778748,
"logps/rejected": -1.1783361434936523,
"loss": 0.149,
"nll_loss": 0.1352442055940628,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.017167862504720688,
"rewards/margins": 0.10066574811935425,
"rewards/rejected": -0.11783361434936523,
"step": 490
},
{
"epoch": 0.8,
"grad_norm": 1.4262079801129468,
"learning_rate": 1.5000000000000002e-05,
"log_odds_chosen": 2.723966121673584,
"log_odds_ratio": -0.116343654692173,
"logits/chosen": 22.94664764404297,
"logits/rejected": 26.360061645507812,
"logps/chosen": -0.17480744421482086,
"logps/rejected": -1.2685492038726807,
"loss": 0.145,
"nll_loss": 0.13372007012367249,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.017480744048953056,
"rewards/margins": 0.10937418043613434,
"rewards/rejected": -0.12685491144657135,
"step": 500
},
{
"epoch": 0.816,
"grad_norm": 1.961973010059407,
"learning_rate": 1.4756242090702756e-05,
"log_odds_chosen": 2.9430527687072754,
"log_odds_ratio": -0.1177729144692421,
"logits/chosen": 24.766284942626953,
"logits/rejected": 28.823827743530273,
"logps/chosen": -0.16511258482933044,
"logps/rejected": -1.238128423690796,
"loss": 0.1441,
"nll_loss": 0.12999919056892395,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.016511257737874985,
"rewards/margins": 0.10730159282684326,
"rewards/rejected": -0.12381285429000854,
"step": 510
},
{
"epoch": 0.832,
"grad_norm": 0.8171513438848954,
"learning_rate": 1.4508775406894308e-05,
"log_odds_chosen": 2.880876064300537,
"log_odds_ratio": -0.10505475103855133,
"logits/chosen": 22.384557723999023,
"logits/rejected": 26.74051856994629,
"logps/chosen": -0.1716020405292511,
"logps/rejected": -1.2871472835540771,
"loss": 0.1375,
"nll_loss": 0.12949416041374207,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.01716020330786705,
"rewards/margins": 0.11155451834201813,
"rewards/rejected": -0.12871472537517548,
"step": 520
},
{
"epoch": 0.848,
"grad_norm": 0.6656893252356583,
"learning_rate": 1.4257792915650728e-05,
"log_odds_chosen": 3.0356669425964355,
"log_odds_ratio": -0.10586954653263092,
"logits/chosen": 19.979793548583984,
"logits/rejected": 23.590118408203125,
"logps/chosen": -0.1698727309703827,
"logps/rejected": -1.3184887170791626,
"loss": 0.127,
"nll_loss": 0.12099827826023102,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0169872734695673,
"rewards/margins": 0.11486159265041351,
"rewards/rejected": -0.13184888660907745,
"step": 530
},
{
"epoch": 0.864,
"grad_norm": 0.5859755094542713,
"learning_rate": 1.4003490325568953e-05,
"log_odds_chosen": 3.2319984436035156,
"log_odds_ratio": -0.09660033136606216,
"logits/chosen": 21.624605178833008,
"logits/rejected": 24.835023880004883,
"logps/chosen": -0.1984686404466629,
"logps/rejected": -1.5682591199874878,
"loss": 0.1372,
"nll_loss": 0.12878955900669098,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.01984686590731144,
"rewards/margins": 0.13697902858257294,
"rewards/rejected": -0.1568259298801422,
"step": 540
},
{
"epoch": 0.88,
"grad_norm": 0.779222909675585,
"learning_rate": 1.3746065934159123e-05,
"log_odds_chosen": 3.039996862411499,
"log_odds_ratio": -0.10127159208059311,
"logits/chosen": 25.058979034423828,
"logits/rejected": 27.096914291381836,
"logps/chosen": -0.18606023490428925,
"logps/rejected": -1.4767307043075562,
"loss": 0.137,
"nll_loss": 0.14917424321174622,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.018606023862957954,
"rewards/margins": 0.12906703352928162,
"rewards/rejected": -0.1476730853319168,
"step": 550
},
{
"epoch": 0.896,
"grad_norm": 0.5131532913955095,
"learning_rate": 1.3485720473218153e-05,
"log_odds_chosen": 3.5183768272399902,
"log_odds_ratio": -0.08511906862258911,
"logits/chosen": 21.076614379882812,
"logits/rejected": 24.55303955078125,
"logps/chosen": -0.15598443150520325,
"logps/rejected": -1.6431341171264648,
"loss": 0.1361,
"nll_loss": 0.12268916517496109,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.015598444268107414,
"rewards/margins": 0.14871497452259064,
"rewards/rejected": -0.164313405752182,
"step": 560
},
{
"epoch": 0.912,
"grad_norm": 0.7652230384669911,
"learning_rate": 1.3222656952305113e-05,
"log_odds_chosen": 3.6853690147399902,
"log_odds_ratio": -0.08712838590145111,
"logits/chosen": 20.190637588500977,
"logits/rejected": 25.091556549072266,
"logps/chosen": -0.13156656920909882,
"logps/rejected": -1.6401069164276123,
"loss": 0.121,
"nll_loss": 0.10304377228021622,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.01315665990114212,
"rewards/margins": 0.15085402131080627,
"rewards/rejected": -0.1640106737613678,
"step": 570
},
{
"epoch": 0.928,
"grad_norm": 0.7300290151409546,
"learning_rate": 1.2957080500440469e-05,
"log_odds_chosen": 3.642732620239258,
"log_odds_ratio": -0.07864442467689514,
"logits/chosen": 19.066701889038086,
"logits/rejected": 23.404861450195312,
"logps/chosen": -0.14534644782543182,
"logps/rejected": -1.6337239742279053,
"loss": 0.1262,
"nll_loss": 0.11143051087856293,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.014534644782543182,
"rewards/margins": 0.14883777499198914,
"rewards/rejected": -0.16337242722511292,
"step": 580
},
{
"epoch": 0.944,
"grad_norm": 0.581668451170283,
"learning_rate": 1.2689198206152657e-05,
"log_odds_chosen": 3.713766098022461,
"log_odds_ratio": -0.0588177926838398,
"logits/chosen": 22.885316848754883,
"logits/rejected": 25.525924682617188,
"logps/chosen": -0.14779958128929138,
"logps/rejected": -1.8394057750701904,
"loss": 0.1234,
"nll_loss": 0.11549869924783707,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.014779957942664623,
"rewards/margins": 0.1691606342792511,
"rewards/rejected": -0.1839405745267868,
"step": 590
},
{
"epoch": 0.96,
"grad_norm": 0.566646644489029,
"learning_rate": 1.2419218955996677e-05,
"log_odds_chosen": 3.757345199584961,
"log_odds_ratio": -0.06075813248753548,
"logits/chosen": 17.20478057861328,
"logits/rejected": 22.157577514648438,
"logps/chosen": -0.11853750050067902,
"logps/rejected": -1.5277284383773804,
"loss": 0.1127,
"nll_loss": 0.09783687442541122,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.011853748932480812,
"rewards/margins": 0.14091908931732178,
"rewards/rejected": -0.15277281403541565,
"step": 600
},
{
"epoch": 0.976,
"grad_norm": 0.6375581513651442,
"learning_rate": 1.2147353271670634e-05,
"log_odds_chosen": 3.8227837085723877,
"log_odds_ratio": -0.09637900441884995,
"logits/chosen": 16.64263916015625,
"logits/rejected": 21.630008697509766,
"logps/chosen": -0.1534954011440277,
"logps/rejected": -1.8045285940170288,
"loss": 0.1165,
"nll_loss": 0.11545213311910629,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.015349539928138256,
"rewards/margins": 0.16510331630706787,
"rewards/rejected": -0.1804528534412384,
"step": 610
},
{
"epoch": 0.992,
"grad_norm": 0.5653677883458513,
"learning_rate": 1.187381314585725e-05,
"log_odds_chosen": 3.73738431930542,
"log_odds_ratio": -0.06889805942773819,
"logits/chosen": 20.336971282958984,
"logits/rejected": 23.924381256103516,
"logps/chosen": -0.12372555583715439,
"logps/rejected": -1.6133959293365479,
"loss": 0.1249,
"nll_loss": 0.1038857102394104,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.012372555211186409,
"rewards/margins": 0.14896704256534576,
"rewards/rejected": -0.16133958101272583,
"step": 620
},
{
"epoch": 1.008,
"grad_norm": 0.41947247295916856,
"learning_rate": 1.159881187691835e-05,
"log_odds_chosen": 3.937540054321289,
"log_odds_ratio": -0.05745882913470268,
"logits/chosen": 18.518375396728516,
"logits/rejected": 22.057506561279297,
"logps/chosen": -0.12291697412729263,
"logps/rejected": -1.742640495300293,
"loss": 0.1059,
"nll_loss": 0.09989026188850403,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.012291697785258293,
"rewards/margins": 0.16197235882282257,
"rewards/rejected": -0.17426404356956482,
"step": 630
},
{
"epoch": 1.024,
"grad_norm": 0.6597379471767634,
"learning_rate": 1.1322563902571227e-05,
"log_odds_chosen": 4.9311842918396,
"log_odds_ratio": -0.028991302475333214,
"logits/chosen": 12.143542289733887,
"logits/rejected": 17.73841094970703,
"logps/chosen": -0.06773394346237183,
"logps/rejected": -2.039699077606201,
"loss": 0.0792,
"nll_loss": 0.07453403621912003,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0067733945325016975,
"rewards/margins": 0.19719651341438293,
"rewards/rejected": -0.20396992564201355,
"step": 640
},
{
"epoch": 1.04,
"grad_norm": 0.8720926641059171,
"learning_rate": 1.1045284632676535e-05,
"log_odds_chosen": 4.683058261871338,
"log_odds_ratio": -0.025352224707603455,
"logits/chosen": 13.8761625289917,
"logits/rejected": 19.721921920776367,
"logps/chosen": -0.08715043216943741,
"logps/rejected": -1.9542983770370483,
"loss": 0.0855,
"nll_loss": 0.0856931135058403,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008715043775737286,
"rewards/margins": 0.18671482801437378,
"rewards/rejected": -0.19542984664440155,
"step": 650
},
{
"epoch": 1.056,
"grad_norm": 0.5401242210481009,
"learning_rate": 1.0767190281268187e-05,
"log_odds_chosen": 4.751006603240967,
"log_odds_ratio": -0.03235021233558655,
"logits/chosen": 14.378767013549805,
"logits/rejected": 19.829517364501953,
"logps/chosen": -0.07587035000324249,
"logps/rejected": -2.011232614517212,
"loss": 0.0829,
"nll_loss": 0.07639533281326294,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007587035186588764,
"rewards/margins": 0.19353622198104858,
"rewards/rejected": -0.20112323760986328,
"step": 660
},
{
"epoch": 1.072,
"grad_norm": 0.4975498251334291,
"learning_rate": 1.0488497697956134e-05,
"log_odds_chosen": 4.578000068664551,
"log_odds_ratio": -0.03245697170495987,
"logits/chosen": 13.391039848327637,
"logits/rejected": 18.039913177490234,
"logps/chosen": -0.08072780072689056,
"logps/rejected": -1.7768867015838623,
"loss": 0.0835,
"nll_loss": 0.07963033765554428,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008072780445218086,
"rewards/margins": 0.16961589455604553,
"rewards/rejected": -0.17768868803977966,
"step": 670
},
{
"epoch": 1.088,
"grad_norm": 0.4579496342491343,
"learning_rate": 1.0209424198833571e-05,
"log_odds_chosen": 4.71366024017334,
"log_odds_ratio": -0.029907289892435074,
"logits/chosen": 11.174861907958984,
"logits/rejected": 16.83877182006836,
"logps/chosen": -0.08100271224975586,
"logps/rejected": -1.965436339378357,
"loss": 0.0782,
"nll_loss": 0.07603878527879715,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008100271224975586,
"rewards/margins": 0.1884433925151825,
"rewards/rejected": -0.19654366374015808,
"step": 680
},
{
"epoch": 1.104,
"grad_norm": 0.41635397248386163,
"learning_rate": 9.930187397020385e-06,
"log_odds_chosen": 4.987896919250488,
"log_odds_ratio": -0.029992487281560898,
"logits/chosen": 10.937190055847168,
"logits/rejected": 16.221263885498047,
"logps/chosen": -0.0843561664223671,
"logps/rejected": -1.950840950012207,
"loss": 0.0802,
"nll_loss": 0.07738173007965088,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.00843561626970768,
"rewards/margins": 0.18664847314357758,
"rewards/rejected": -0.1950840950012207,
"step": 690
},
{
"epoch": 1.12,
"grad_norm": 0.42432934191231536,
"learning_rate": 9.651005032974994e-06,
"log_odds_chosen": 5.254538059234619,
"log_odds_ratio": -0.019603287801146507,
"logits/chosen": 12.815790176391602,
"logits/rejected": 16.598247528076172,
"logps/chosen": -0.07913420349359512,
"logps/rejected": -2.306891918182373,
"loss": 0.0795,
"nll_loss": 0.07783981412649155,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007913419976830482,
"rewards/margins": 0.22277574241161346,
"rewards/rejected": -0.23068919777870178,
"step": 700
},
{
"epoch": 1.1360000000000001,
"grad_norm": 0.5144220041987062,
"learning_rate": 9.372094804706867e-06,
"log_odds_chosen": 5.051201820373535,
"log_odds_ratio": -0.022446967661380768,
"logits/chosen": 8.465599060058594,
"logits/rejected": 15.262280464172363,
"logps/chosen": -0.04941285401582718,
"logps/rejected": -1.7998939752578735,
"loss": 0.0699,
"nll_loss": 0.06099081039428711,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0049412851221859455,
"rewards/margins": 0.1750481128692627,
"rewards/rejected": -0.17998939752578735,
"step": 710
},
{
"epoch": 1.152,
"grad_norm": 0.4355953213178483,
"learning_rate": 9.093674198022201e-06,
"log_odds_chosen": 4.678675651550293,
"log_odds_ratio": -0.03710417076945305,
"logits/chosen": 9.093629837036133,
"logits/rejected": 15.341288566589355,
"logps/chosen": -0.0850113034248352,
"logps/rejected": -1.8611046075820923,
"loss": 0.0802,
"nll_loss": 0.08102207630872726,
"rewards/accuracies": 0.987500011920929,
"rewards/chosen": -0.008501130156219006,
"rewards/margins": 0.17760933935642242,
"rewards/rejected": -0.1861104667186737,
"step": 720
},
{
"epoch": 1.168,
"grad_norm": 0.4183273277614356,
"learning_rate": 8.815960316934991e-06,
"log_odds_chosen": 4.754403591156006,
"log_odds_ratio": -0.026562413200736046,
"logits/chosen": 11.237001419067383,
"logits/rejected": 16.33772087097168,
"logps/chosen": -0.0855894535779953,
"logps/rejected": -2.021867275238037,
"loss": 0.074,
"nll_loss": 0.07607786357402802,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.00855894573032856,
"rewards/margins": 0.1936277598142624,
"rewards/rejected": -0.2021867036819458,
"step": 730
},
{
"epoch": 1.184,
"grad_norm": 0.5708367601297668,
"learning_rate": 8.539169714375885e-06,
"log_odds_chosen": 5.499794960021973,
"log_odds_ratio": -0.015643080696463585,
"logits/chosen": 11.671289443969727,
"logits/rejected": 15.871813774108887,
"logps/chosen": -0.0698837861418724,
"logps/rejected": -2.4058656692504883,
"loss": 0.0758,
"nll_loss": 0.06816570460796356,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006988379172980785,
"rewards/margins": 0.23359815776348114,
"rewards/rejected": -0.2405865490436554,
"step": 740
},
{
"epoch": 1.2,
"grad_norm": 0.46974989560288,
"learning_rate": 8.263518223330698e-06,
"log_odds_chosen": 5.374692916870117,
"log_odds_ratio": -0.01755271479487419,
"logits/chosen": 10.17614459991455,
"logits/rejected": 15.561047554016113,
"logps/chosen": -0.06151513010263443,
"logps/rejected": -2.1926753520965576,
"loss": 0.0765,
"nll_loss": 0.06682710349559784,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006151513196527958,
"rewards/margins": 0.21311601996421814,
"rewards/rejected": -0.21926753222942352,
"step": 750
},
{
"epoch": 1.216,
"grad_norm": 0.536798190903821,
"learning_rate": 7.989220788540356e-06,
"log_odds_chosen": 5.090784549713135,
"log_odds_ratio": -0.024079417809844017,
"logits/chosen": 10.42772102355957,
"logits/rejected": 16.83907127380371,
"logps/chosen": -0.0742868185043335,
"logps/rejected": -1.9404573440551758,
"loss": 0.0767,
"nll_loss": 0.07289760559797287,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007428681943565607,
"rewards/margins": 0.18661704659461975,
"rewards/rejected": -0.19404573738574982,
"step": 760
},
{
"epoch": 1.232,
"grad_norm": 0.7535095663117326,
"learning_rate": 7.716491298893443e-06,
"log_odds_chosen": 5.1247100830078125,
"log_odds_ratio": -0.027670254930853844,
"logits/chosen": 13.019775390625,
"logits/rejected": 18.80875015258789,
"logps/chosen": -0.08515826612710953,
"logps/rejected": -2.227242946624756,
"loss": 0.0799,
"nll_loss": 0.08172322809696198,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008515825495123863,
"rewards/margins": 0.2142084836959839,
"rewards/rejected": -0.2227243185043335,
"step": 770
},
{
"epoch": 1.248,
"grad_norm": 0.43802198454777963,
"learning_rate": 7.445542420642097e-06,
"log_odds_chosen": 4.944367408752441,
"log_odds_ratio": -0.019510487094521523,
"logits/chosen": 12.673688888549805,
"logits/rejected": 17.701494216918945,
"logps/chosen": -0.06077063828706741,
"logps/rejected": -1.8956024646759033,
"loss": 0.0722,
"nll_loss": 0.06931096315383911,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0060770632699131966,
"rewards/margins": 0.18348316848278046,
"rewards/rejected": -0.18956024944782257,
"step": 780
},
{
"epoch": 1.264,
"grad_norm": 0.5918184881161849,
"learning_rate": 7.176585431571235e-06,
"log_odds_chosen": 4.934049606323242,
"log_odds_ratio": -0.022184232249855995,
"logits/chosen": 10.117755889892578,
"logits/rejected": 16.32207489013672,
"logps/chosen": -0.06841601431369781,
"logps/rejected": -1.8420988321304321,
"loss": 0.076,
"nll_loss": 0.07008203864097595,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006841603666543961,
"rewards/margins": 0.17736826837062836,
"rewards/rejected": -0.1842098832130432,
"step": 790
},
{
"epoch": 1.28,
"grad_norm": 0.4109794535608094,
"learning_rate": 6.909830056250527e-06,
"log_odds_chosen": 5.385331153869629,
"log_odds_ratio": -0.01698954775929451,
"logits/chosen": 9.808797836303711,
"logits/rejected": 15.439534187316895,
"logps/chosen": -0.06334672123193741,
"logps/rejected": -2.0740480422973633,
"loss": 0.0735,
"nll_loss": 0.06666533648967743,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006334672216325998,
"rewards/margins": 0.2010701447725296,
"rewards/rejected": -0.20740482211112976,
"step": 800
},
{
"epoch": 1.296,
"grad_norm": 0.5786524901744241,
"learning_rate": 6.645484302497452e-06,
"log_odds_chosen": 5.305611610412598,
"log_odds_ratio": -0.015638932585716248,
"logits/chosen": 10.026269912719727,
"logits/rejected": 16.038047790527344,
"logps/chosen": -0.058640025556087494,
"logps/rejected": -2.095346212387085,
"loss": 0.0687,
"nll_loss": 0.0625784695148468,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005864003207534552,
"rewards/margins": 0.20367059111595154,
"rewards/rejected": -0.20953461527824402,
"step": 810
},
{
"epoch": 1.312,
"grad_norm": 0.413245918888085,
"learning_rate": 6.383754299179079e-06,
"log_odds_chosen": 5.338342189788818,
"log_odds_ratio": -0.016953352838754654,
"logits/chosen": 10.80897331237793,
"logits/rejected": 16.324369430541992,
"logps/chosen": -0.07062120735645294,
"logps/rejected": -2.1152849197387695,
"loss": 0.0753,
"nll_loss": 0.07253970950841904,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007062119897454977,
"rewards/margins": 0.20446637272834778,
"rewards/rejected": -0.2115284949541092,
"step": 820
},
{
"epoch": 1.328,
"grad_norm": 0.45473358832367533,
"learning_rate": 6.124844135478971e-06,
"log_odds_chosen": 4.9816741943359375,
"log_odds_ratio": -0.016600508242845535,
"logits/chosen": 12.60502815246582,
"logits/rejected": 17.482601165771484,
"logps/chosen": -0.07652293890714645,
"logps/rejected": -1.9486392736434937,
"loss": 0.0753,
"nll_loss": 0.0775555893778801,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007652295287698507,
"rewards/margins": 0.18721160292625427,
"rewards/rejected": -0.1948639154434204,
"step": 830
},
{
"epoch": 1.3439999999999999,
"grad_norm": 0.43221707053355657,
"learning_rate": 5.868955701754584e-06,
"log_odds_chosen": 5.058526039123535,
"log_odds_ratio": -0.02234785258769989,
"logits/chosen": 10.739975929260254,
"logits/rejected": 14.592979431152344,
"logps/chosen": -0.08031310141086578,
"logps/rejected": -2.022430896759033,
"loss": 0.071,
"nll_loss": 0.07450385391712189,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008031310513615608,
"rewards/margins": 0.19421175122261047,
"rewards/rejected": -0.20224305987358093,
"step": 840
},
{
"epoch": 1.3599999999999999,
"grad_norm": 0.36475019984330276,
"learning_rate": 5.616288532109225e-06,
"log_odds_chosen": 5.216296672821045,
"log_odds_ratio": -0.02249590866267681,
"logits/chosen": 10.627111434936523,
"logits/rejected": 16.160968780517578,
"logps/chosen": -0.08645172417163849,
"logps/rejected": -2.255333662033081,
"loss": 0.0741,
"nll_loss": 0.07864928245544434,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008645173162221909,
"rewards/margins": 0.2168881893157959,
"rewards/rejected": -0.2255333662033081,
"step": 850
},
{
"epoch": 1.376,
"grad_norm": 0.5386028507331131,
"learning_rate": 5.367039648801386e-06,
"log_odds_chosen": 5.349644184112549,
"log_odds_ratio": -0.019762977957725525,
"logits/chosen": 11.73155689239502,
"logits/rejected": 16.905214309692383,
"logps/chosen": -0.07144279778003693,
"logps/rejected": -2.1353299617767334,
"loss": 0.073,
"nll_loss": 0.07140424847602844,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007144278846681118,
"rewards/margins": 0.20638874173164368,
"rewards/rejected": -0.21353299915790558,
"step": 860
},
{
"epoch": 1.392,
"grad_norm": 0.4861620483022773,
"learning_rate": 5.121403408612672e-06,
"log_odds_chosen": 5.4971466064453125,
"log_odds_ratio": -0.013068562373518944,
"logits/chosen": 10.185466766357422,
"logits/rejected": 15.52686595916748,
"logps/chosen": -0.05803658813238144,
"logps/rejected": -2.2335634231567383,
"loss": 0.071,
"nll_loss": 0.06299170106649399,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005803658626973629,
"rewards/margins": 0.21755269169807434,
"rewards/rejected": -0.22335636615753174,
"step": 870
},
{
"epoch": 1.408,
"grad_norm": 0.7221352144813555,
"learning_rate": 4.879571351294287e-06,
"log_odds_chosen": 5.298310279846191,
"log_odds_ratio": -0.015498277731239796,
"logits/chosen": 8.6812744140625,
"logits/rejected": 14.908788681030273,
"logps/chosen": -0.04598275199532509,
"logps/rejected": -1.8670889139175415,
"loss": 0.0695,
"nll_loss": 0.05738530308008194,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004598274827003479,
"rewards/margins": 0.18211060762405396,
"rewards/rejected": -0.18670889735221863,
"step": 880
},
{
"epoch": 1.424,
"grad_norm": 0.4395795057245911,
"learning_rate": 4.641732050210032e-06,
"log_odds_chosen": 5.754244327545166,
"log_odds_ratio": -0.013112393207848072,
"logits/chosen": 10.254205703735352,
"logits/rejected": 16.640913009643555,
"logps/chosen": -0.07648586481809616,
"logps/rejected": -2.492318630218506,
"loss": 0.0712,
"nll_loss": 0.07544175535440445,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007648587226867676,
"rewards/margins": 0.24158325791358948,
"rewards/rejected": -0.24923184514045715,
"step": 890
},
{
"epoch": 1.44,
"grad_norm": 0.6571181936107011,
"learning_rate": 4.408070965292534e-06,
"log_odds_chosen": 5.259117126464844,
"log_odds_ratio": -0.016550693660974503,
"logits/chosen": 10.593389511108398,
"logits/rejected": 15.269793510437012,
"logps/chosen": -0.08531410992145538,
"logps/rejected": -2.2608280181884766,
"loss": 0.0758,
"nll_loss": 0.07715515047311783,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.008531411178410053,
"rewards/margins": 0.21755139529705048,
"rewards/rejected": -0.22608280181884766,
"step": 900
},
{
"epoch": 1.456,
"grad_norm": 0.45529898147163606,
"learning_rate": 4.178770298427107e-06,
"log_odds_chosen": 5.506943702697754,
"log_odds_ratio": -0.017402973026037216,
"logits/chosen": 10.588479995727539,
"logits/rejected": 16.51026153564453,
"logps/chosen": -0.06945344805717468,
"logps/rejected": -2.281757354736328,
"loss": 0.0701,
"nll_loss": 0.07317786663770676,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006945345550775528,
"rewards/margins": 0.2212304174900055,
"rewards/rejected": -0.22817575931549072,
"step": 910
},
{
"epoch": 1.472,
"grad_norm": 0.4799597661016462,
"learning_rate": 3.954008851376252e-06,
"log_odds_chosen": 5.671328544616699,
"log_odds_ratio": -0.01591522805392742,
"logits/chosen": 7.850234031677246,
"logits/rejected": 15.029336929321289,
"logps/chosen": -0.04982997104525566,
"logps/rejected": -2.066762685775757,
"loss": 0.0637,
"nll_loss": 0.059512246400117874,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0049829971976578236,
"rewards/margins": 0.20169326663017273,
"rewards/rejected": -0.20667627453804016,
"step": 920
},
{
"epoch": 1.488,
"grad_norm": 0.42918774258741327,
"learning_rate": 3.7339618863553983e-06,
"log_odds_chosen": 5.558573246002197,
"log_odds_ratio": -0.014333250001072884,
"logits/chosen": 8.254855155944824,
"logits/rejected": 14.497201919555664,
"logps/chosen": -0.06610764563083649,
"logps/rejected": -2.231790781021118,
"loss": 0.0638,
"nll_loss": 0.06769216060638428,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0066107651218771935,
"rewards/margins": 0.21656832098960876,
"rewards/rejected": -0.22317907214164734,
"step": 930
},
{
"epoch": 1.504,
"grad_norm": 0.39881406218102516,
"learning_rate": 3.5188009893686916e-06,
"log_odds_chosen": 5.319655418395996,
"log_odds_ratio": -0.0184173621237278,
"logits/chosen": 10.073064804077148,
"logits/rejected": 15.57422924041748,
"logps/chosen": -0.07947548478841782,
"logps/rejected": -2.1314804553985596,
"loss": 0.0699,
"nll_loss": 0.07717884331941605,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.007947548292577267,
"rewards/margins": 0.20520050823688507,
"rewards/rejected": -0.21314804255962372,
"step": 940
},
{
"epoch": 1.52,
"grad_norm": 0.3003778882690616,
"learning_rate": 3.308693936411421e-06,
"log_odds_chosen": 5.652432441711426,
"log_odds_ratio": -0.018848979845643044,
"logits/chosen": 7.4497809410095215,
"logits/rejected": 14.933306694030762,
"logps/chosen": -0.0582701750099659,
"logps/rejected": -2.193925142288208,
"loss": 0.0677,
"nll_loss": 0.06635381281375885,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005827018059790134,
"rewards/margins": 0.21356551349163055,
"rewards/rejected": -0.21939250826835632,
"step": 950
},
{
"epoch": 1.536,
"grad_norm": 0.3009824619938579,
"learning_rate": 3.103804562643302e-06,
"log_odds_chosen": 5.427401065826416,
"log_odds_ratio": -0.014070016331970692,
"logits/chosen": 8.830900192260742,
"logits/rejected": 14.178878784179688,
"logps/chosen": -0.06140875071287155,
"logps/rejected": -2.2172341346740723,
"loss": 0.0645,
"nll_loss": 0.06464984267950058,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0061408756300807,
"rewards/margins": 0.21558256447315216,
"rewards/rejected": -0.22172346711158752,
"step": 960
},
{
"epoch": 1.552,
"grad_norm": 0.4056333088268534,
"learning_rate": 2.9042926346347932e-06,
"log_odds_chosen": 5.441522598266602,
"log_odds_ratio": -0.014954231679439545,
"logits/chosen": 9.046590805053711,
"logits/rejected": 15.852540969848633,
"logps/chosen": -0.06886409223079681,
"logps/rejected": -2.157127618789673,
"loss": 0.0658,
"nll_loss": 0.06967596709728241,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006886409129947424,
"rewards/margins": 0.20882634818553925,
"rewards/rejected": -0.2157127559185028,
"step": 970
},
{
"epoch": 1.568,
"grad_norm": 0.3196857185408296,
"learning_rate": 2.7103137257858867e-06,
"log_odds_chosen": 5.551312446594238,
"log_odds_ratio": -0.011502685025334358,
"logits/chosen": 7.783066749572754,
"logits/rejected": 14.421549797058105,
"logps/chosen": -0.05041471868753433,
"logps/rejected": -2.1501784324645996,
"loss": 0.0606,
"nll_loss": 0.05976518243551254,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005041471216827631,
"rewards/margins": 0.20997636020183563,
"rewards/rejected": -0.21501782536506653,
"step": 980
},
{
"epoch": 1.584,
"grad_norm": 0.5263216429559819,
"learning_rate": 2.522019095014683e-06,
"log_odds_chosen": 5.743101119995117,
"log_odds_ratio": -0.010009285993874073,
"logits/chosen": 7.407317161560059,
"logits/rejected": 14.168241500854492,
"logps/chosen": -0.048953376710414886,
"logps/rejected": -2.123490810394287,
"loss": 0.0686,
"nll_loss": 0.058644842356443405,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004895338322967291,
"rewards/margins": 0.20745377242565155,
"rewards/rejected": -0.2123490869998932,
"step": 990
},
{
"epoch": 1.6,
"grad_norm": 0.36561256966761685,
"learning_rate": 2.339555568810221e-06,
"log_odds_chosen": 5.752957344055176,
"log_odds_ratio": -0.008349338546395302,
"logits/chosen": 7.5777788162231445,
"logits/rejected": 14.192586898803711,
"logps/chosen": -0.043931931257247925,
"logps/rejected": -2.1989071369171143,
"loss": 0.0669,
"nll_loss": 0.053385306149721146,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004393193405121565,
"rewards/margins": 0.21549753844738007,
"rewards/rejected": -0.21989071369171143,
"step": 1000
},
{
"epoch": 1.616,
"grad_norm": 0.5870370963384035,
"learning_rate": 2.163065426741603e-06,
"log_odds_chosen": 5.90677547454834,
"log_odds_ratio": -0.012085262686014175,
"logits/chosen": 6.974325656890869,
"logits/rejected": 14.128486633300781,
"logps/chosen": -0.0529697947204113,
"logps/rejected": -2.190331220626831,
"loss": 0.0645,
"nll_loss": 0.05988318473100662,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0052969790995121,
"rewards/margins": 0.2137361317873001,
"rewards/rejected": -0.2190331220626831,
"step": 1010
},
{
"epoch": 1.6320000000000001,
"grad_norm": 0.6243099880494768,
"learning_rate": 1.9926862905126663e-06,
"log_odds_chosen": 5.65887975692749,
"log_odds_ratio": -0.014939057640731335,
"logits/chosen": 7.151690483093262,
"logits/rejected": 13.71330451965332,
"logps/chosen": -0.055140770971775055,
"logps/rejected": -2.053590774536133,
"loss": 0.0673,
"nll_loss": 0.06196935102343559,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005514076910912991,
"rewards/margins": 0.19984501600265503,
"rewards/rejected": -0.2053591012954712,
"step": 1020
},
{
"epoch": 1.6480000000000001,
"grad_norm": 0.405678875969962,
"learning_rate": 1.8285510166487154e-06,
"log_odds_chosen": 5.765079021453857,
"log_odds_ratio": -0.014157906174659729,
"logits/chosen": 9.832456588745117,
"logits/rejected": 15.122549057006836,
"logps/chosen": -0.06559108942747116,
"logps/rejected": -2.4253296852111816,
"loss": 0.0694,
"nll_loss": 0.07027839124202728,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006559108849614859,
"rewards/margins": 0.23597386479377747,
"rewards/rejected": -0.24253296852111816,
"step": 1030
},
{
"epoch": 1.6640000000000001,
"grad_norm": 0.2511510020504382,
"learning_rate": 1.6707875928990059e-06,
"log_odds_chosen": 5.9670586585998535,
"log_odds_ratio": -0.007793957833200693,
"logits/chosen": 6.463404178619385,
"logits/rejected": 13.94349479675293,
"logps/chosen": -0.036587394773960114,
"logps/rejected": -2.103496789932251,
"loss": 0.0584,
"nll_loss": 0.05108030512928963,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0036587391514331102,
"rewards/margins": 0.2066909372806549,
"rewards/rejected": -0.2103496789932251,
"step": 1040
},
{
"epoch": 1.6800000000000002,
"grad_norm": 0.3812260869637052,
"learning_rate": 1.5195190384357405e-06,
"log_odds_chosen": 5.7870683670043945,
"log_odds_ratio": -0.01724204048514366,
"logits/chosen": 7.535168647766113,
"logits/rejected": 14.668806076049805,
"logps/chosen": -0.06202416867017746,
"logps/rejected": -2.3643112182617188,
"loss": 0.0648,
"nll_loss": 0.067660391330719,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006202416960150003,
"rewards/margins": 0.2302287369966507,
"rewards/rejected": -0.23643112182617188,
"step": 1050
},
{
"epoch": 1.696,
"grad_norm": 0.3058840630848455,
"learning_rate": 1.3748633079274254e-06,
"log_odds_chosen": 5.669142723083496,
"log_odds_ratio": -0.010866614989936352,
"logits/chosen": 8.615381240844727,
"logits/rejected": 14.441973686218262,
"logps/chosen": -0.05342826992273331,
"logps/rejected": -2.128041982650757,
"loss": 0.063,
"nll_loss": 0.06444182246923447,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005342827644199133,
"rewards/margins": 0.2074614018201828,
"rewards/rejected": -0.21280424296855927,
"step": 1060
},
{
"epoch": 1.712,
"grad_norm": 0.3235904810469797,
"learning_rate": 1.2369331995613664e-06,
"log_odds_chosen": 5.627545356750488,
"log_odds_ratio": -0.008036193437874317,
"logits/chosen": 6.999047756195068,
"logits/rejected": 13.646265983581543,
"logps/chosen": -0.048848383128643036,
"logps/rejected": -2.056873321533203,
"loss": 0.0634,
"nll_loss": 0.05773681402206421,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004884839057922363,
"rewards/margins": 0.2008024901151657,
"rewards/rejected": -0.20568732917308807,
"step": 1070
},
{
"epoch": 1.728,
"grad_norm": 0.410833487441517,
"learning_rate": 1.1058362670870248e-06,
"log_odds_chosen": 5.63392448425293,
"log_odds_ratio": -0.012561176903545856,
"logits/chosen": 7.00423526763916,
"logits/rejected": 12.398062705993652,
"logps/chosen": -0.06395191699266434,
"logps/rejected": -2.2964274883270264,
"loss": 0.0622,
"nll_loss": 0.06452804803848267,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006395191885530949,
"rewards/margins": 0.22324757277965546,
"rewards/rejected": -0.22964277863502502,
"step": 1080
},
{
"epoch": 1.744,
"grad_norm": 0.34256892405513334,
"learning_rate": 9.816747359488632e-07,
"log_odds_chosen": 6.084750652313232,
"log_odds_ratio": -0.01084104459732771,
"logits/chosen": 7.226653099060059,
"logits/rejected": 13.381048202514648,
"logps/chosen": -0.05891553312540054,
"logps/rejected": -2.476435661315918,
"loss": 0.0609,
"nll_loss": 0.06247776746749878,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005891552660614252,
"rewards/margins": 0.241752028465271,
"rewards/rejected": -0.24764356017112732,
"step": 1090
},
{
"epoch": 1.76,
"grad_norm": 0.34625995571735285,
"learning_rate": 8.645454235739903e-07,
"log_odds_chosen": 5.164887428283691,
"log_odds_ratio": -0.01927667111158371,
"logits/chosen": 7.622632026672363,
"logits/rejected": 15.065897941589355,
"logps/chosen": -0.05562591552734375,
"logps/rejected": -1.8294832706451416,
"loss": 0.0671,
"nll_loss": 0.06367551535367966,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.00556259136646986,
"rewards/margins": 0.17738574743270874,
"rewards/rejected": -0.18294833600521088,
"step": 1100
},
{
"epoch": 1.776,
"grad_norm": 0.3129225650767717,
"learning_rate": 7.545396638768698e-07,
"log_odds_chosen": 5.7577362060546875,
"log_odds_ratio": -0.009081291034817696,
"logits/chosen": 7.138357639312744,
"logits/rejected": 13.764391899108887,
"logps/chosen": -0.048210419714450836,
"logps/rejected": -2.1392695903778076,
"loss": 0.0617,
"nll_loss": 0.0597044937312603,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004821042064577341,
"rewards/margins": 0.20910592377185822,
"rewards/rejected": -0.2139269858598709,
"step": 1110
},
{
"epoch": 1.792,
"grad_norm": 0.3636030975331139,
"learning_rate": 6.517432360398556e-07,
"log_odds_chosen": 5.995306015014648,
"log_odds_ratio": -0.013278511352837086,
"logits/chosen": 7.143318176269531,
"logits/rejected": 14.408166885375977,
"logps/chosen": -0.048548031598329544,
"logps/rejected": -2.2793731689453125,
"loss": 0.0626,
"nll_loss": 0.06035660579800606,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0048548029735684395,
"rewards/margins": 0.2230825424194336,
"rewards/rejected": -0.22793734073638916,
"step": 1120
},
{
"epoch": 1.808,
"grad_norm": 0.3502514097075585,
"learning_rate": 5.562362976251901e-07,
"log_odds_chosen": 5.784189224243164,
"log_odds_ratio": -0.019179679453372955,
"logits/chosen": 7.51906681060791,
"logits/rejected": 13.482861518859863,
"logps/chosen": -0.050465572625398636,
"logps/rejected": -2.069855213165283,
"loss": 0.0673,
"nll_loss": 0.06206601858139038,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005046557169407606,
"rewards/margins": 0.2019389569759369,
"rewards/rejected": -0.20698551833629608,
"step": 1130
},
{
"epoch": 1.8239999999999998,
"grad_norm": 0.2385312527933045,
"learning_rate": 4.6809332207053083e-07,
"log_odds_chosen": 6.114804267883301,
"log_odds_ratio": -0.01152479462325573,
"logits/chosen": 8.802262306213379,
"logits/rejected": 14.353776931762695,
"logps/chosen": -0.05430366471409798,
"logps/rejected": -2.6986145973205566,
"loss": 0.0617,
"nll_loss": 0.062255967408418655,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.00543036637827754,
"rewards/margins": 0.26443108916282654,
"rewards/rejected": -0.26986145973205566,
"step": 1140
},
{
"epoch": 1.8399999999999999,
"grad_norm": 0.5023792704599986,
"learning_rate": 3.8738304061681107e-07,
"log_odds_chosen": 5.99333381652832,
"log_odds_ratio": -0.010073414072394371,
"logits/chosen": 6.594714164733887,
"logits/rejected": 14.024187088012695,
"logps/chosen": -0.04959163814783096,
"logps/rejected": -2.3722965717315674,
"loss": 0.0664,
"nll_loss": 0.05878063291311264,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004959163721650839,
"rewards/margins": 0.2322704792022705,
"rewards/rejected": -0.23722966015338898,
"step": 1150
},
{
"epoch": 1.8559999999999999,
"grad_norm": 0.18925156808932705,
"learning_rate": 3.1416838871368925e-07,
"log_odds_chosen": 5.59714412689209,
"log_odds_ratio": -0.016117483377456665,
"logits/chosen": 6.784099578857422,
"logits/rejected": 14.297111511230469,
"logps/chosen": -0.04300536960363388,
"logps/rejected": -1.9598991870880127,
"loss": 0.0594,
"nll_loss": 0.06088510900735855,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0043005370534956455,
"rewards/margins": 0.1916893571615219,
"rewards/rejected": -0.19598989188671112,
"step": 1160
},
{
"epoch": 1.8719999999999999,
"grad_norm": 0.4752392208294527,
"learning_rate": 2.4850645694436736e-07,
"log_odds_chosen": 5.687636375427246,
"log_odds_ratio": -0.009939742274582386,
"logits/chosen": 6.543469429016113,
"logits/rejected": 13.751274108886719,
"logps/chosen": -0.0570371076464653,
"logps/rejected": -2.2096800804138184,
"loss": 0.0612,
"nll_loss": 0.06289631128311157,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005703710950911045,
"rewards/margins": 0.21526429057121277,
"rewards/rejected": -0.22096800804138184,
"step": 1170
},
{
"epoch": 1.888,
"grad_norm": 0.40463907978157104,
"learning_rate": 1.9044844650808468e-07,
"log_odds_chosen": 5.943331718444824,
"log_odds_ratio": -0.008882846683263779,
"logits/chosen": 7.400280952453613,
"logits/rejected": 14.994569778442383,
"logps/chosen": -0.045132219791412354,
"logps/rejected": -2.3415608406066895,
"loss": 0.0612,
"nll_loss": 0.05737587809562683,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.00451322179287672,
"rewards/margins": 0.22964289784431458,
"rewards/rejected": -0.23415610194206238,
"step": 1180
},
{
"epoch": 1.904,
"grad_norm": 0.35720023172691506,
"learning_rate": 1.400396292949513e-07,
"log_odds_chosen": 5.809309959411621,
"log_odds_ratio": -0.008964666165411472,
"logits/chosen": 8.318174362182617,
"logits/rejected": 13.486407279968262,
"logps/chosen": -0.053054939955472946,
"logps/rejected": -2.4225754737854004,
"loss": 0.0626,
"nll_loss": 0.06207244470715523,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005305493716150522,
"rewards/margins": 0.2369520664215088,
"rewards/rejected": -0.24225755035877228,
"step": 1190
},
{
"epoch": 1.92,
"grad_norm": 0.45375001244411867,
"learning_rate": 9.731931258429638e-08,
"log_odds_chosen": 6.011991024017334,
"log_odds_ratio": -0.011115940287709236,
"logits/chosen": 7.217115879058838,
"logits/rejected": 13.6605806350708,
"logps/chosen": -0.05640987679362297,
"logps/rejected": -2.306795835494995,
"loss": 0.0667,
"nll_loss": 0.064894899725914,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005640988238155842,
"rewards/margins": 0.22503860294818878,
"rewards/rejected": -0.23067960143089294,
"step": 1200
},
{
"epoch": 1.936,
"grad_norm": 0.3851516384032617,
"learning_rate": 6.232080839403631e-08,
"log_odds_chosen": 6.033963203430176,
"log_odds_ratio": -0.008175736293196678,
"logits/chosen": 5.674324035644531,
"logits/rejected": 12.042490005493164,
"logps/chosen": -0.041214533150196075,
"logps/rejected": -2.353848934173584,
"loss": 0.0594,
"nll_loss": 0.05154932290315628,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0041214535012841225,
"rewards/margins": 0.2312634438276291,
"rewards/rejected": -0.23538486659526825,
"step": 1210
},
{
"epoch": 1.952,
"grad_norm": 0.2301620888800148,
"learning_rate": 3.50714075049563e-08,
"log_odds_chosen": 5.746167182922363,
"log_odds_ratio": -0.01174002606421709,
"logits/chosen": 7.780473232269287,
"logits/rejected": 14.472818374633789,
"logps/chosen": -0.052154283970594406,
"logps/rejected": -2.3260200023651123,
"loss": 0.0631,
"nll_loss": 0.06034677103161812,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.0052154287695884705,
"rewards/margins": 0.22738659381866455,
"rewards/rejected": -0.23260203003883362,
"step": 1220
},
{
"epoch": 1.968,
"grad_norm": 0.29743582988851025,
"learning_rate": 1.5592358180189782e-08,
"log_odds_chosen": 5.381113529205322,
"log_odds_ratio": -0.010704567655920982,
"logits/chosen": 6.542645454406738,
"logits/rejected": 14.09373664855957,
"logps/chosen": -0.045278120785951614,
"logps/rejected": -1.7775919437408447,
"loss": 0.0605,
"nll_loss": 0.057972025126218796,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.004527812357991934,
"rewards/margins": 0.1732313632965088,
"rewards/rejected": -0.17775918543338776,
"step": 1230
},
{
"epoch": 1.984,
"grad_norm": 0.3648997358256397,
"learning_rate": 3.898849596456477e-09,
"log_odds_chosen": 5.863680839538574,
"log_odds_ratio": -0.009549752809107304,
"logits/chosen": 6.157291889190674,
"logits/rejected": 12.384984016418457,
"logps/chosen": -0.05373894423246384,
"logps/rejected": -2.254366397857666,
"loss": 0.0613,
"nll_loss": 0.06132974475622177,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.005373895168304443,
"rewards/margins": 0.2200627624988556,
"rewards/rejected": -0.22543664276599884,
"step": 1240
},
{
"epoch": 2.0,
"grad_norm": 0.6054356785823999,
"learning_rate": 0.0,
"log_odds_chosen": 5.457306861877441,
"log_odds_ratio": -0.019121190533041954,
"logits/chosen": 7.081140041351318,
"logits/rejected": 13.820508003234863,
"logps/chosen": -0.0638023242354393,
"logps/rejected": -1.9536654949188232,
"loss": 0.0679,
"nll_loss": 0.067476287484169,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.006380232982337475,
"rewards/margins": 0.18898631632328033,
"rewards/rejected": -0.19536656141281128,
"step": 1250
},
{
"epoch": 2.0,
"step": 1250,
"total_flos": 0.0,
"train_loss": 0.2130171504020691,
"train_runtime": 11716.0982,
"train_samples_per_second": 3.414,
"train_steps_per_second": 0.107
}
],
"logging_steps": 10,
"max_steps": 1250,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 500,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}