|
{ |
|
"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 |
|
} |
|
|