Llama-3-Instruct-8B-SimPO / trainer_state.json
RAY2L's picture
Upload folder using huggingface_hub
a580ab3 verified
raw
history blame contribute delete
No virus
51.7 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9981298423724285,
"eval_steps": 400,
"global_step": 467,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0021373230029388193,
"grad_norm": 14.616270006854805,
"learning_rate": 2.127659574468085e-08,
"logits/chosen": -1.1381689310073853,
"logits/rejected": -0.9913416504859924,
"logps/chosen": -0.2839311361312866,
"logps/rejected": -0.29555341601371765,
"loss": 1.608,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.7098277807235718,
"rewards/margins": 0.029055725783109665,
"rewards/rejected": -0.7388835549354553,
"step": 1
},
{
"epoch": 0.010686615014694095,
"grad_norm": 15.798664787929345,
"learning_rate": 1.0638297872340425e-07,
"logits/chosen": -0.989432692527771,
"logits/rejected": -0.9188639521598816,
"logps/chosen": -0.26978519558906555,
"logps/rejected": -0.26858454942703247,
"loss": 1.6205,
"rewards/accuracies": 0.546875,
"rewards/chosen": -0.6744629740715027,
"rewards/margins": -0.0030015837401151657,
"rewards/rejected": -0.6714614033699036,
"step": 5
},
{
"epoch": 0.02137323002938819,
"grad_norm": 12.624563810334125,
"learning_rate": 2.127659574468085e-07,
"logits/chosen": -0.9885454177856445,
"logits/rejected": -0.9444629549980164,
"logps/chosen": -0.27237212657928467,
"logps/rejected": -0.2738865911960602,
"loss": 1.6069,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.6809303164482117,
"rewards/margins": 0.0037860602606087923,
"rewards/rejected": -0.6847164630889893,
"step": 10
},
{
"epoch": 0.03205984504408229,
"grad_norm": 8.372527145409705,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -0.9621208906173706,
"logits/rejected": -0.9169891476631165,
"logps/chosen": -0.2949882745742798,
"logps/rejected": -0.2834514081478119,
"loss": 1.6049,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.7374706864356995,
"rewards/margins": -0.028842147439718246,
"rewards/rejected": -0.7086285352706909,
"step": 15
},
{
"epoch": 0.04274646005877638,
"grad_norm": 11.776984926293357,
"learning_rate": 4.25531914893617e-07,
"logits/chosen": -0.9825040102005005,
"logits/rejected": -0.983070969581604,
"logps/chosen": -0.26186102628707886,
"logps/rejected": -0.26891934871673584,
"loss": 1.6132,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.65465247631073,
"rewards/margins": 0.01764589548110962,
"rewards/rejected": -0.6722984313964844,
"step": 20
},
{
"epoch": 0.053433075073470476,
"grad_norm": 11.594406861763694,
"learning_rate": 5.319148936170212e-07,
"logits/chosen": -1.0304547548294067,
"logits/rejected": -1.0080630779266357,
"logps/chosen": -0.28237098455429077,
"logps/rejected": -0.28737810254096985,
"loss": 1.6046,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.7059274911880493,
"rewards/margins": 0.012517772614955902,
"rewards/rejected": -0.7184451818466187,
"step": 25
},
{
"epoch": 0.06411969008816458,
"grad_norm": 13.181478771123727,
"learning_rate": 6.382978723404255e-07,
"logits/chosen": -1.0607410669326782,
"logits/rejected": -0.9752001762390137,
"logps/chosen": -0.2747865319252014,
"logps/rejected": -0.27676859498023987,
"loss": 1.6169,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.6869663596153259,
"rewards/margins": 0.004955160431563854,
"rewards/rejected": -0.6919214725494385,
"step": 30
},
{
"epoch": 0.07480630510285867,
"grad_norm": 20.535555442364025,
"learning_rate": 7.446808510638297e-07,
"logits/chosen": -0.9909561276435852,
"logits/rejected": -0.9133442044258118,
"logps/chosen": -0.2775736451148987,
"logps/rejected": -0.3019401431083679,
"loss": 1.6019,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.6939341425895691,
"rewards/margins": 0.06091625243425369,
"rewards/rejected": -0.754850447177887,
"step": 35
},
{
"epoch": 0.08549292011755276,
"grad_norm": 24.246139127435896,
"learning_rate": 8.51063829787234e-07,
"logits/chosen": -0.9217761158943176,
"logits/rejected": -0.9104591608047485,
"logps/chosen": -0.2786272466182709,
"logps/rejected": -0.29105645418167114,
"loss": 1.617,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.6965680718421936,
"rewards/margins": 0.031073052436113358,
"rewards/rejected": -0.7276411056518555,
"step": 40
},
{
"epoch": 0.09617953513224686,
"grad_norm": 10.340400390143392,
"learning_rate": 9.574468085106384e-07,
"logits/chosen": -0.918626606464386,
"logits/rejected": -0.837913990020752,
"logps/chosen": -0.33493560552597046,
"logps/rejected": -0.33847588300704956,
"loss": 1.6012,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.8373388051986694,
"rewards/margins": 0.00885077752172947,
"rewards/rejected": -0.8461896777153015,
"step": 45
},
{
"epoch": 0.10686615014694095,
"grad_norm": 9.374774032732045,
"learning_rate": 9.998741174712533e-07,
"logits/chosen": -0.9050552248954773,
"logits/rejected": -0.9147823452949524,
"logps/chosen": -0.29951199889183044,
"logps/rejected": -0.33792784810066223,
"loss": 1.5734,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.7487800717353821,
"rewards/margins": 0.09603960067033768,
"rewards/rejected": -0.8448196649551392,
"step": 50
},
{
"epoch": 0.11755276516163506,
"grad_norm": 12.319781154359692,
"learning_rate": 9.991050648838675e-07,
"logits/chosen": -0.917604923248291,
"logits/rejected": -0.859523594379425,
"logps/chosen": -0.2723899185657501,
"logps/rejected": -0.3260456621646881,
"loss": 1.5698,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6809747815132141,
"rewards/margins": 0.13413934409618378,
"rewards/rejected": -0.8151141405105591,
"step": 55
},
{
"epoch": 0.12823938017632916,
"grad_norm": 10.399641038869387,
"learning_rate": 9.97637968732563e-07,
"logits/chosen": -0.9515789151191711,
"logits/rejected": -0.9302376508712769,
"logps/chosen": -0.28720152378082275,
"logps/rejected": -0.3024197816848755,
"loss": 1.5846,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.7180038690567017,
"rewards/margins": 0.03804563358426094,
"rewards/rejected": -0.7560494542121887,
"step": 60
},
{
"epoch": 0.13892599519102325,
"grad_norm": 11.022271869662724,
"learning_rate": 9.954748808839674e-07,
"logits/chosen": -0.943594753742218,
"logits/rejected": -0.9900406002998352,
"logps/chosen": -0.28493180871009827,
"logps/rejected": -0.28960293531417847,
"loss": 1.584,
"rewards/accuracies": 0.4000000059604645,
"rewards/chosen": -0.7123295664787292,
"rewards/margins": 0.01167784072458744,
"rewards/rejected": -0.7240074276924133,
"step": 65
},
{
"epoch": 0.14961261020571734,
"grad_norm": 16.92128796829008,
"learning_rate": 9.926188266120295e-07,
"logits/chosen": -0.9900282025337219,
"logits/rejected": -0.9590619802474976,
"logps/chosen": -0.347956120967865,
"logps/rejected": -0.3921958804130554,
"loss": 1.5789,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.8698902130126953,
"rewards/margins": 0.11059943586587906,
"rewards/rejected": -0.9804896116256714,
"step": 70
},
{
"epoch": 0.16029922522041143,
"grad_norm": 16.614347339411598,
"learning_rate": 9.890738003669027e-07,
"logits/chosen": -0.9594413042068481,
"logits/rejected": -0.9471073150634766,
"logps/chosen": -0.3611677289009094,
"logps/rejected": -0.4440728724002838,
"loss": 1.5386,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.9029192924499512,
"rewards/margins": 0.20726287364959717,
"rewards/rejected": -1.1101821660995483,
"step": 75
},
{
"epoch": 0.17098584023510552,
"grad_norm": 11.76758635779136,
"learning_rate": 9.848447601883433e-07,
"logits/chosen": -0.9906526803970337,
"logits/rejected": -0.9735655784606934,
"logps/chosen": -0.32699793577194214,
"logps/rejected": -0.40725231170654297,
"loss": 1.5433,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.8174948692321777,
"rewards/margins": 0.20063595473766327,
"rewards/rejected": -1.018130898475647,
"step": 80
},
{
"epoch": 0.18167245524979964,
"grad_norm": 10.051730020561614,
"learning_rate": 9.799376207714444e-07,
"logits/chosen": -0.9377920031547546,
"logits/rejected": -0.889153003692627,
"logps/chosen": -0.35584911704063416,
"logps/rejected": -0.403181254863739,
"loss": 1.5549,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.8896228671073914,
"rewards/margins": 0.11833026260137558,
"rewards/rejected": -1.0079530477523804,
"step": 85
},
{
"epoch": 0.19235907026449373,
"grad_norm": 12.299894230325533,
"learning_rate": 9.743592451943998e-07,
"logits/chosen": -0.9021016955375671,
"logits/rejected": -0.8429332971572876,
"logps/chosen": -0.3575811982154846,
"logps/rejected": -0.5039599537849426,
"loss": 1.5625,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.8939528465270996,
"rewards/margins": 0.36594703793525696,
"rewards/rejected": -1.2598999738693237,
"step": 90
},
{
"epoch": 0.20304568527918782,
"grad_norm": 14.145971876119047,
"learning_rate": 9.681174353198686e-07,
"logits/chosen": -1.002209186553955,
"logits/rejected": -1.0227996110916138,
"logps/chosen": -0.41834840178489685,
"logps/rejected": -0.47105270624160767,
"loss": 1.5299,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.0458710193634033,
"rewards/margins": 0.13176079094409943,
"rewards/rejected": -1.1776319742202759,
"step": 95
},
{
"epoch": 0.2137323002938819,
"grad_norm": 16.666230206302217,
"learning_rate": 9.612209208833646e-07,
"logits/chosen": -1.0650508403778076,
"logits/rejected": -1.0104029178619385,
"logps/chosen": -0.4353299140930176,
"logps/rejected": -0.4307008385658264,
"loss": 1.5889,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.088324785232544,
"rewards/margins": -0.011572673916816711,
"rewards/rejected": -1.076751947402954,
"step": 100
},
{
"epoch": 0.224418915308576,
"grad_norm": 17.277037383709768,
"learning_rate": 9.536793472839324e-07,
"logits/chosen": -0.9813326597213745,
"logits/rejected": -0.9346880912780762,
"logps/chosen": -0.3515966534614563,
"logps/rejected": -0.46131715178489685,
"loss": 1.5543,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8789916038513184,
"rewards/margins": 0.2743012309074402,
"rewards/rejected": -1.1532928943634033,
"step": 105
},
{
"epoch": 0.2351055303232701,
"grad_norm": 12.743189576436166,
"learning_rate": 9.455032620941839e-07,
"logits/chosen": -0.8806193470954895,
"logits/rejected": -0.8356341123580933,
"logps/chosen": -0.39990124106407166,
"logps/rejected": -0.42553144693374634,
"loss": 1.5602,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.9997529983520508,
"rewards/margins": 0.06407558172941208,
"rewards/rejected": -1.063828706741333,
"step": 110
},
{
"epoch": 0.2457921453379642,
"grad_norm": 13.62783373071176,
"learning_rate": 9.367041003085648e-07,
"logits/chosen": -0.9096847772598267,
"logits/rejected": -0.9085425138473511,
"logps/chosen": -0.40596961975097656,
"logps/rejected": -0.4234519898891449,
"loss": 1.5452,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.0149240493774414,
"rewards/margins": 0.043705932796001434,
"rewards/rejected": -1.0586299896240234,
"step": 115
},
{
"epoch": 0.2564787603526583,
"grad_norm": 12.129891786309633,
"learning_rate": 9.272941683504808e-07,
"logits/chosen": -0.9741169810295105,
"logits/rejected": -0.9402868151664734,
"logps/chosen": -0.3675619959831238,
"logps/rejected": -0.4012192189693451,
"loss": 1.5295,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.9189050793647766,
"rewards/margins": 0.08414317667484283,
"rewards/rejected": -1.0030481815338135,
"step": 120
},
{
"epoch": 0.2671653753673524,
"grad_norm": 12.760172929151745,
"learning_rate": 9.172866268606513e-07,
"logits/chosen": -1.014088749885559,
"logits/rejected": -1.007265567779541,
"logps/chosen": -0.47613659501075745,
"logps/rejected": -0.5860797166824341,
"loss": 1.5674,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.1903413534164429,
"rewards/margins": 0.2748578190803528,
"rewards/rejected": -1.4651992321014404,
"step": 125
},
{
"epoch": 0.2778519903820465,
"grad_norm": 14.619403511233878,
"learning_rate": 9.066954722907638e-07,
"logits/chosen": -1.0016984939575195,
"logits/rejected": -1.0274698734283447,
"logps/chosen": -0.4267544746398926,
"logps/rejected": -0.5695111155509949,
"loss": 1.5349,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -1.0668861865997314,
"rewards/margins": 0.35689178109169006,
"rewards/rejected": -1.4237778186798096,
"step": 130
},
{
"epoch": 0.2885386053967406,
"grad_norm": 10.619279201046492,
"learning_rate": 8.955355173281707e-07,
"logits/chosen": -0.9727472066879272,
"logits/rejected": -0.9219390749931335,
"logps/chosen": -0.39321380853652954,
"logps/rejected": -0.4508630633354187,
"loss": 1.4978,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.9830344915390015,
"rewards/margins": 0.1441231220960617,
"rewards/rejected": -1.1271576881408691,
"step": 135
},
{
"epoch": 0.2992252204114347,
"grad_norm": 15.834369827499174,
"learning_rate": 8.838223701790055e-07,
"logits/chosen": -0.967811107635498,
"logits/rejected": -0.9583989977836609,
"logps/chosen": -0.43714672327041626,
"logps/rejected": -0.5161387920379639,
"loss": 1.5218,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.0928667783737183,
"rewards/margins": 0.19748012721538544,
"rewards/rejected": -1.2903468608856201,
"step": 140
},
{
"epoch": 0.30991183542612877,
"grad_norm": 12.847642374810391,
"learning_rate": 8.71572412738697e-07,
"logits/chosen": -0.9621770977973938,
"logits/rejected": -0.9567694664001465,
"logps/chosen": -0.43871012330055237,
"logps/rejected": -0.4793704152107239,
"loss": 1.5379,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.0967752933502197,
"rewards/margins": 0.10165063291788101,
"rewards/rejected": -1.1984260082244873,
"step": 145
},
{
"epoch": 0.32059845044082286,
"grad_norm": 11.935024879482611,
"learning_rate": 8.588027776804058e-07,
"logits/chosen": -0.8888929486274719,
"logits/rejected": -0.8930709958076477,
"logps/chosen": -0.5125688910484314,
"logps/rejected": -0.722069263458252,
"loss": 1.5225,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -1.2814221382141113,
"rewards/margins": 0.5237509608268738,
"rewards/rejected": -1.8051731586456299,
"step": 150
},
{
"epoch": 0.33128506545551695,
"grad_norm": 16.764961972010557,
"learning_rate": 8.455313244934324e-07,
"logits/chosen": -0.9782350659370422,
"logits/rejected": -0.9396332502365112,
"logps/chosen": -0.5236457586288452,
"logps/rejected": -0.7281317710876465,
"loss": 1.4981,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.309114694595337,
"rewards/margins": 0.5112148523330688,
"rewards/rejected": -1.8203294277191162,
"step": 155
},
{
"epoch": 0.34197168047021104,
"grad_norm": 11.439957416651609,
"learning_rate": 8.317766145051057e-07,
"logits/chosen": -0.9513300061225891,
"logits/rejected": -0.9507732391357422,
"logps/chosen": -0.4817509651184082,
"logps/rejected": -0.6095650792121887,
"loss": 1.5319,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.2043774127960205,
"rewards/margins": 0.3195350468158722,
"rewards/rejected": -1.5239124298095703,
"step": 160
},
{
"epoch": 0.3526582954849052,
"grad_norm": 15.127711801028411,
"learning_rate": 8.175578849210894e-07,
"logits/chosen": -1.0031640529632568,
"logits/rejected": -0.9406889081001282,
"logps/chosen": -0.5859075784683228,
"logps/rejected": -0.6615421175956726,
"loss": 1.5097,
"rewards/accuracies": 0.5625,
"rewards/chosen": -1.4647690057754517,
"rewards/margins": 0.18908634781837463,
"rewards/rejected": -1.6538550853729248,
"step": 165
},
{
"epoch": 0.36334491049959927,
"grad_norm": 19.457497723556852,
"learning_rate": 8.028950219204099e-07,
"logits/chosen": -0.9729808568954468,
"logits/rejected": -0.9703952670097351,
"logps/chosen": -0.5817127823829651,
"logps/rejected": -0.7540786266326904,
"loss": 1.52,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.4542819261550903,
"rewards/margins": 0.4309147298336029,
"rewards/rejected": -1.8851966857910156,
"step": 170
},
{
"epoch": 0.37403152551429336,
"grad_norm": 13.673431328128673,
"learning_rate": 7.878085328428368e-07,
"logits/chosen": -0.9736520051956177,
"logits/rejected": -0.9835022687911987,
"logps/chosen": -0.6428834795951843,
"logps/rejected": -0.8455599546432495,
"loss": 1.4591,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.6072088479995728,
"rewards/margins": 0.5066913366317749,
"rewards/rejected": -2.1139001846313477,
"step": 175
},
{
"epoch": 0.38471814052898745,
"grad_norm": 20.186619677768963,
"learning_rate": 7.723195175075135e-07,
"logits/chosen": -0.9989040493965149,
"logits/rejected": -0.9996326565742493,
"logps/chosen": -0.6329335570335388,
"logps/rejected": -0.7996684312820435,
"loss": 1.5042,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.5823338031768799,
"rewards/margins": 0.41683727502822876,
"rewards/rejected": -1.999171257019043,
"step": 180
},
{
"epoch": 0.39540475554368154,
"grad_norm": 20.89728549129117,
"learning_rate": 7.564496387029531e-07,
"logits/chosen": -1.0245858430862427,
"logits/rejected": -0.9974483251571655,
"logps/chosen": -0.5426946878433228,
"logps/rejected": -0.765714168548584,
"loss": 1.5357,
"rewards/accuracies": 0.625,
"rewards/chosen": -1.3567368984222412,
"rewards/margins": 0.5575486421585083,
"rewards/rejected": -1.914285659790039,
"step": 185
},
{
"epoch": 0.40609137055837563,
"grad_norm": 14.284349268241145,
"learning_rate": 7.402210918896689e-07,
"logits/chosen": -1.0203882455825806,
"logits/rejected": -1.002201795578003,
"logps/chosen": -0.5420663356781006,
"logps/rejected": -0.8374710083007812,
"loss": 1.4458,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.355165719985962,
"rewards/margins": 0.7385115623474121,
"rewards/rejected": -2.093677282333374,
"step": 190
},
{
"epoch": 0.4167779855730697,
"grad_norm": 21.006500802721117,
"learning_rate": 7.236565741578162e-07,
"logits/chosen": -0.9360872507095337,
"logits/rejected": -0.9267364740371704,
"logps/chosen": -0.6582534909248352,
"logps/rejected": -0.9019840359687805,
"loss": 1.496,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -1.6456336975097656,
"rewards/margins": 0.6093264818191528,
"rewards/rejected": -2.254960298538208,
"step": 195
},
{
"epoch": 0.4274646005877638,
"grad_norm": 17.34001688115439,
"learning_rate": 7.067792524832603e-07,
"logits/chosen": -0.9733101725578308,
"logits/rejected": -0.9737011194229126,
"logps/chosen": -0.5885142087936401,
"logps/rejected": -0.759198784828186,
"loss": 1.4947,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.4712855815887451,
"rewards/margins": 0.42671123147010803,
"rewards/rejected": -1.8979966640472412,
"step": 200
},
{
"epoch": 0.4381512156024579,
"grad_norm": 18.93087179194272,
"learning_rate": 6.896127313264642e-07,
"logits/chosen": -1.0645233392715454,
"logits/rejected": -1.0247756242752075,
"logps/chosen": -0.7058154940605164,
"logps/rejected": -0.8320499658584595,
"loss": 1.481,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -1.7645390033721924,
"rewards/margins": 0.3155860900878906,
"rewards/rejected": -2.080124855041504,
"step": 205
},
{
"epoch": 0.448837830617152,
"grad_norm": 20.51304681116505,
"learning_rate": 6.721810196195174e-07,
"logits/chosen": -1.0714137554168701,
"logits/rejected": -0.9989116787910461,
"logps/chosen": -0.6715008020401001,
"logps/rejected": -0.8458214998245239,
"loss": 1.4931,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -1.6787519454956055,
"rewards/margins": 0.43580159544944763,
"rewards/rejected": -2.114553451538086,
"step": 210
},
{
"epoch": 0.45952444563184613,
"grad_norm": 16.46301637400639,
"learning_rate": 6.545084971874736e-07,
"logits/chosen": -1.0320379734039307,
"logits/rejected": -0.9812155961990356,
"logps/chosen": -0.7699872255325317,
"logps/rejected": -0.9591902494430542,
"loss": 1.4357,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.9249680042266846,
"rewards/margins": 0.47300752997398376,
"rewards/rejected": -2.397975444793701,
"step": 215
},
{
"epoch": 0.4702110606465402,
"grad_norm": 21.088635750930834,
"learning_rate": 6.3661988065096e-07,
"logits/chosen": -1.0327280759811401,
"logits/rejected": -0.9874359965324402,
"logps/chosen": -0.8896724581718445,
"logps/rejected": -1.0073192119598389,
"loss": 1.4954,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.2241809368133545,
"rewards/margins": 0.2941167950630188,
"rewards/rejected": -2.5182979106903076,
"step": 220
},
{
"epoch": 0.4808976756612343,
"grad_norm": 19.02142810522417,
"learning_rate": 6.185401888577487e-07,
"logits/chosen": -0.9864276051521301,
"logits/rejected": -0.9660781621932983,
"logps/chosen": -0.772359311580658,
"logps/rejected": -0.794577956199646,
"loss": 1.557,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -1.9308984279632568,
"rewards/margins": 0.0555465929210186,
"rewards/rejected": -1.9864448308944702,
"step": 225
},
{
"epoch": 0.4915842906759284,
"grad_norm": 17.867436451445265,
"learning_rate": 6.002947078916364e-07,
"logits/chosen": -1.0150645971298218,
"logits/rejected": -0.9232436418533325,
"logps/chosen": -0.8819006085395813,
"logps/rejected": -1.0791282653808594,
"loss": 1.4692,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.2047512531280518,
"rewards/margins": 0.49306946992874146,
"rewards/rejected": -2.6978209018707275,
"step": 230
},
{
"epoch": 0.5022709056906225,
"grad_norm": 19.261147315812543,
"learning_rate": 5.819089557075688e-07,
"logits/chosen": -0.996496319770813,
"logits/rejected": -0.9518272280693054,
"logps/chosen": -0.8440850377082825,
"logps/rejected": -0.9660031199455261,
"loss": 1.4942,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.110212564468384,
"rewards/margins": 0.30479517579078674,
"rewards/rejected": -2.4150078296661377,
"step": 235
},
{
"epoch": 0.5129575207053166,
"grad_norm": 18.32103206829717,
"learning_rate": 5.634086464424742e-07,
"logits/chosen": -1.0355093479156494,
"logits/rejected": -1.006392002105713,
"logps/chosen": -0.7564712762832642,
"logps/rejected": -0.9403325319290161,
"loss": 1.4046,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -1.8911781311035156,
"rewards/margins": 0.4596532881259918,
"rewards/rejected": -2.3508315086364746,
"step": 240
},
{
"epoch": 0.5236441357200107,
"grad_norm": 13.624682866914808,
"learning_rate": 5.448196544517167e-07,
"logits/chosen": -1.0392498970031738,
"logits/rejected": -1.0080206394195557,
"logps/chosen": -0.7070968151092529,
"logps/rejected": -0.8767108917236328,
"loss": 1.4407,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -1.7677417993545532,
"rewards/margins": 0.4240352213382721,
"rewards/rejected": -2.191777229309082,
"step": 245
},
{
"epoch": 0.5343307507347048,
"grad_norm": 22.22843267157582,
"learning_rate": 5.26167978121472e-07,
"logits/chosen": -1.048105001449585,
"logits/rejected": -1.028188943862915,
"logps/chosen": -0.8559715151786804,
"logps/rejected": -1.2052185535430908,
"loss": 1.467,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.1399290561676025,
"rewards/margins": 0.8731171488761902,
"rewards/rejected": -3.0130460262298584,
"step": 250
},
{
"epoch": 0.5450173657493989,
"grad_norm": 21.7502246417763,
"learning_rate": 5.074797035076318e-07,
"logits/chosen": -1.0093282461166382,
"logits/rejected": -1.0085898637771606,
"logps/chosen": -0.8086752891540527,
"logps/rejected": -0.9931262731552124,
"loss": 1.4101,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -2.021688222885132,
"rewards/margins": 0.4611276090145111,
"rewards/rejected": -2.4828155040740967,
"step": 255
},
{
"epoch": 0.555703980764093,
"grad_norm": 17.422008542886676,
"learning_rate": 4.887809678520975e-07,
"logits/chosen": -0.9935086369514465,
"logits/rejected": -0.9804226160049438,
"logps/chosen": -0.7468116879463196,
"logps/rejected": -1.0926599502563477,
"loss": 1.4384,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -1.8670291900634766,
"rewards/margins": 0.8646209836006165,
"rewards/rejected": -2.731649875640869,
"step": 260
},
{
"epoch": 0.566390595778787,
"grad_norm": 21.42359137838811,
"learning_rate": 4.700979230274829e-07,
"logits/chosen": -1.0343437194824219,
"logits/rejected": -1.0285662412643433,
"logps/chosen": -0.8779309391975403,
"logps/rejected": -0.9997223019599915,
"loss": 1.4835,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.19482684135437,
"rewards/margins": 0.3044784963130951,
"rewards/rejected": -2.4993057250976562,
"step": 265
},
{
"epoch": 0.5770772107934812,
"grad_norm": 27.361650975575174,
"learning_rate": 4.514566989613559e-07,
"logits/chosen": -1.022482991218567,
"logits/rejected": -1.030027151107788,
"logps/chosen": -0.8233085870742798,
"logps/rejected": -1.0775353908538818,
"loss": 1.4746,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -2.058271646499634,
"rewards/margins": 0.6355669498443604,
"rewards/rejected": -2.693838596343994,
"step": 270
},
{
"epoch": 0.5877638258081752,
"grad_norm": 19.306011967200188,
"learning_rate": 4.328833670911724e-07,
"logits/chosen": -0.8774306178092957,
"logits/rejected": -0.841380774974823,
"logps/chosen": -0.977648138999939,
"logps/rejected": -1.1523730754852295,
"loss": 1.4355,
"rewards/accuracies": 0.5625,
"rewards/chosen": -2.444120168685913,
"rewards/margins": 0.43681272864341736,
"rewards/rejected": -2.8809328079223633,
"step": 275
},
{
"epoch": 0.5984504408228694,
"grad_norm": 18.830991297114817,
"learning_rate": 4.144039039010124e-07,
"logits/chosen": -1.0766229629516602,
"logits/rejected": -1.034156084060669,
"logps/chosen": -0.9281458854675293,
"logps/rejected": -1.0838878154754639,
"loss": 1.4465,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.3203647136688232,
"rewards/margins": 0.38935500383377075,
"rewards/rejected": -2.709719657897949,
"step": 280
},
{
"epoch": 0.6091370558375635,
"grad_norm": 20.938307778539542,
"learning_rate": 3.960441545911204e-07,
"logits/chosen": -0.9466499090194702,
"logits/rejected": -0.9020398855209351,
"logps/chosen": -1.1035289764404297,
"logps/rejected": -1.1428934335708618,
"loss": 1.4739,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.758822441101074,
"rewards/margins": 0.09841099381446838,
"rewards/rejected": -2.8572330474853516,
"step": 285
},
{
"epoch": 0.6198236708522575,
"grad_norm": 23.245653263780497,
"learning_rate": 3.778297969310529e-07,
"logits/chosen": -0.9969805479049683,
"logits/rejected": -0.9768760800361633,
"logps/chosen": -0.9135320782661438,
"logps/rejected": -1.1653510332107544,
"loss": 1.3943,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.283830165863037,
"rewards/margins": 0.6295474767684937,
"rewards/rejected": -2.9133777618408203,
"step": 290
},
{
"epoch": 0.6305102858669517,
"grad_norm": 18.66753481518412,
"learning_rate": 3.5978630534699865e-07,
"logits/chosen": -1.0683342218399048,
"logits/rejected": -1.0506360530853271,
"logps/chosen": -1.0534954071044922,
"logps/rejected": -1.2509009838104248,
"loss": 1.4446,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.6337387561798096,
"rewards/margins": 0.4935137629508972,
"rewards/rejected": -3.1272525787353516,
"step": 295
},
{
"epoch": 0.6411969008816457,
"grad_norm": 21.410873330760186,
"learning_rate": 3.4193891529348795e-07,
"logits/chosen": -0.943785548210144,
"logits/rejected": -0.8960458040237427,
"logps/chosen": -0.9090589284896851,
"logps/rejected": -1.2448195219039917,
"loss": 1.4048,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -2.2726473808288574,
"rewards/margins": 0.8394016027450562,
"rewards/rejected": -3.112048625946045,
"step": 300
},
{
"epoch": 0.6518835158963399,
"grad_norm": 16.99902973812307,
"learning_rate": 3.243125879593286e-07,
"logits/chosen": -1.001460075378418,
"logits/rejected": -0.9785451889038086,
"logps/chosen": -1.0239307880401611,
"logps/rejected": -1.286873698234558,
"loss": 1.4219,
"rewards/accuracies": 0.6875,
"rewards/chosen": -2.5598270893096924,
"rewards/margins": 0.6573570370674133,
"rewards/rejected": -3.217184543609619,
"step": 305
},
{
"epoch": 0.6625701309110339,
"grad_norm": 35.891043777341494,
"learning_rate": 3.069319753571269e-07,
"logits/chosen": -0.9791940450668335,
"logits/rejected": -1.024137258529663,
"logps/chosen": -1.0476067066192627,
"logps/rejected": -1.3181588649749756,
"loss": 1.4695,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.619016170501709,
"rewards/margins": 0.6763805747032166,
"rewards/rejected": -3.2953972816467285,
"step": 310
},
{
"epoch": 0.673256745925728,
"grad_norm": 22.292608497489727,
"learning_rate": 2.898213858452173e-07,
"logits/chosen": -1.1296494007110596,
"logits/rejected": -1.086306095123291,
"logps/chosen": -1.0372415781021118,
"logps/rejected": -1.1678041219711304,
"loss": 1.469,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -2.593104124069214,
"rewards/margins": 0.3264063894748688,
"rewards/rejected": -2.9195103645324707,
"step": 315
},
{
"epoch": 0.6839433609404221,
"grad_norm": 24.37127385244042,
"learning_rate": 2.730047501302266e-07,
"logits/chosen": -1.0386595726013184,
"logits/rejected": -1.019863486289978,
"logps/chosen": -0.9849531054496765,
"logps/rejected": -1.240861177444458,
"loss": 1.4399,
"rewards/accuracies": 0.625,
"rewards/chosen": -2.4623827934265137,
"rewards/margins": 0.6397703289985657,
"rewards/rejected": -3.1021530628204346,
"step": 320
},
{
"epoch": 0.6946299759551162,
"grad_norm": 28.3616490001571,
"learning_rate": 2.5650558779781635e-07,
"logits/chosen": -0.9190061688423157,
"logits/rejected": -0.9376241564750671,
"logps/chosen": -1.0179104804992676,
"logps/rejected": -1.1567823886871338,
"loss": 1.3944,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -2.544776439666748,
"rewards/margins": 0.3471793532371521,
"rewards/rejected": -2.891955614089966,
"step": 325
},
{
"epoch": 0.7053165909698104,
"grad_norm": 23.089017555469844,
"learning_rate": 2.403469744184154e-07,
"logits/chosen": -1.0184417963027954,
"logits/rejected": -0.9639641642570496,
"logps/chosen": -1.065198540687561,
"logps/rejected": -1.4555224180221558,
"loss": 1.4179,
"rewards/accuracies": 0.6875,
"rewards/chosen": -2.6629960536956787,
"rewards/margins": 0.9758095741271973,
"rewards/rejected": -3.638806104660034,
"step": 330
},
{
"epoch": 0.7160032059845044,
"grad_norm": 19.01160663909916,
"learning_rate": 2.2455150927394878e-07,
"logits/chosen": -0.9782741665840149,
"logits/rejected": -1.0168932676315308,
"logps/chosen": -1.1860311031341553,
"logps/rejected": -1.3966830968856812,
"loss": 1.4107,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -2.965078115463257,
"rewards/margins": 0.5266298651695251,
"rewards/rejected": -3.4917080402374268,
"step": 335
},
{
"epoch": 0.7266898209991985,
"grad_norm": 24.023962985538827,
"learning_rate": 2.0914128375069722e-07,
"logits/chosen": -0.9762552380561829,
"logits/rejected": -0.9652606248855591,
"logps/chosen": -1.0335513353347778,
"logps/rejected": -1.3794705867767334,
"loss": 1.4363,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.5838780403137207,
"rewards/margins": 0.8647986650466919,
"rewards/rejected": -3.448676586151123,
"step": 340
},
{
"epoch": 0.7373764360138926,
"grad_norm": 24.338376358621,
"learning_rate": 1.9413785044249676e-07,
"logits/chosen": -0.9369735717773438,
"logits/rejected": -0.9039338827133179,
"logps/chosen": -1.2081493139266968,
"logps/rejected": -1.3959646224975586,
"loss": 1.3412,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -3.020373821258545,
"rewards/margins": 0.4695381224155426,
"rewards/rejected": -3.4899115562438965,
"step": 345
},
{
"epoch": 0.7480630510285867,
"grad_norm": 25.7573169489427,
"learning_rate": 1.7956219300748792e-07,
"logits/chosen": -1.0008254051208496,
"logits/rejected": -1.0084810256958008,
"logps/chosen": -1.048879861831665,
"logps/rejected": -1.3672925233840942,
"loss": 1.3953,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -2.622199773788452,
"rewards/margins": 0.7960314750671387,
"rewards/rejected": -3.418231248855591,
"step": 350
},
{
"epoch": 0.7587496660432808,
"grad_norm": 26.96889041092714,
"learning_rate": 1.6543469682057104e-07,
"logits/chosen": -1.0293567180633545,
"logits/rejected": -0.9677215814590454,
"logps/chosen": -1.0719913244247437,
"logps/rejected": -1.2735936641693115,
"loss": 1.3954,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -2.679978370666504,
"rewards/margins": 0.5040060877799988,
"rewards/rejected": -3.1839847564697266,
"step": 355
},
{
"epoch": 0.7694362810579749,
"grad_norm": 21.661999756036945,
"learning_rate": 1.5177512046261666e-07,
"logits/chosen": -0.9887057542800903,
"logits/rejected": -1.0177589654922485,
"logps/chosen": -1.291603922843933,
"logps/rejected": -1.665748953819275,
"loss": 1.3804,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -3.2290101051330566,
"rewards/margins": 0.9353626370429993,
"rewards/rejected": -4.164372444152832,
"step": 360
},
{
"epoch": 0.7801228960726689,
"grad_norm": 21.098443523944745,
"learning_rate": 1.3860256808630427e-07,
"logits/chosen": -0.9724413752555847,
"logits/rejected": -0.9351035952568054,
"logps/chosen": -1.203754186630249,
"logps/rejected": -1.3952935934066772,
"loss": 1.3868,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -3.009385347366333,
"rewards/margins": 0.4788486063480377,
"rewards/rejected": -3.488234043121338,
"step": 365
},
{
"epoch": 0.7908095110873631,
"grad_norm": 25.021776262301575,
"learning_rate": 1.2593546269723647e-07,
"logits/chosen": -1.0209473371505737,
"logits/rejected": -1.0015369653701782,
"logps/chosen": -1.2007997035980225,
"logps/rejected": -1.3749370574951172,
"loss": 1.3739,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.0019993782043457,
"rewards/margins": 0.43534326553344727,
"rewards/rejected": -3.437342405319214,
"step": 370
},
{
"epoch": 0.8014961261020572,
"grad_norm": 25.159927366829866,
"learning_rate": 1.1379152038770029e-07,
"logits/chosen": -1.0015761852264404,
"logits/rejected": -1.0289709568023682,
"logps/chosen": -1.0982341766357422,
"logps/rejected": -1.3828232288360596,
"loss": 1.4606,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.7455852031707764,
"rewards/margins": 0.7114725708961487,
"rewards/rejected": -3.4570579528808594,
"step": 375
},
{
"epoch": 0.8121827411167513,
"grad_norm": 20.84984615780728,
"learning_rate": 1.0218772555910954e-07,
"logits/chosen": -1.087043046951294,
"logits/rejected": -1.0346585512161255,
"logps/chosen": -1.2293280363082886,
"logps/rejected": -1.4728463888168335,
"loss": 1.3777,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -3.073319911956787,
"rewards/margins": 0.6087957620620728,
"rewards/rejected": -3.6821160316467285,
"step": 380
},
{
"epoch": 0.8228693561314454,
"grad_norm": 22.784469290812634,
"learning_rate": 9.114030716778432e-08,
"logits/chosen": -1.0404977798461914,
"logits/rejected": -1.0313512086868286,
"logps/chosen": -1.2078773975372314,
"logps/rejected": -1.5215582847595215,
"loss": 1.3372,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -3.019693613052368,
"rewards/margins": 0.7842024564743042,
"rewards/rejected": -3.803895950317383,
"step": 385
},
{
"epoch": 0.8335559711461394,
"grad_norm": 38.2013964652558,
"learning_rate": 8.066471602728803e-08,
"logits/chosen": -1.0984976291656494,
"logits/rejected": -1.0727328062057495,
"logps/chosen": -1.2440178394317627,
"logps/rejected": -1.535946011543274,
"loss": 1.3633,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -3.110044479370117,
"rewards/margins": 0.7298205494880676,
"rewards/rejected": -3.839865207672119,
"step": 390
},
{
"epoch": 0.8442425861608336,
"grad_norm": 21.29430884924605,
"learning_rate": 7.077560319906694e-08,
"logits/chosen": -0.9021504521369934,
"logits/rejected": -0.817115306854248,
"logps/chosen": -1.1583576202392578,
"logps/rejected": -1.3431838750839233,
"loss": 1.396,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -2.8958938121795654,
"rewards/margins": 0.46206584572792053,
"rewards/rejected": -3.357959747314453,
"step": 395
},
{
"epoch": 0.8549292011755276,
"grad_norm": 18.542727738232298,
"learning_rate": 6.148679950161672e-08,
"logits/chosen": -0.9785951375961304,
"logits/rejected": -0.9371121525764465,
"logps/chosen": -1.3020581007003784,
"logps/rejected": -1.5772645473480225,
"loss": 1.3975,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -3.255145311355591,
"rewards/margins": 0.6880159974098206,
"rewards/rejected": -3.9431610107421875,
"step": 400
},
{
"epoch": 0.8549292011755276,
"eval_logits/chosen": -1.1872848272323608,
"eval_logits/rejected": -1.1544642448425293,
"eval_logps/chosen": -1.1779358386993408,
"eval_logps/rejected": -1.4415003061294556,
"eval_loss": 1.375516653060913,
"eval_rewards/accuracies": 0.6612903475761414,
"eval_rewards/chosen": -2.9448394775390625,
"eval_rewards/margins": 0.6589111685752869,
"eval_rewards/rejected": -3.603750705718994,
"eval_runtime": 72.7978,
"eval_samples_per_second": 26.938,
"eval_steps_per_second": 0.852,
"step": 400
},
{
"epoch": 0.8656158161902218,
"grad_norm": 23.254725123506937,
"learning_rate": 5.2811296166831666e-08,
"logits/chosen": -0.9737190008163452,
"logits/rejected": -0.8999165296554565,
"logps/chosen": -1.2110763788223267,
"logps/rejected": -1.4770066738128662,
"loss": 1.4197,
"rewards/accuracies": 0.6875,
"rewards/chosen": -3.027691125869751,
"rewards/margins": 0.6648265719413757,
"rewards/rejected": -3.6925175189971924,
"step": 405
},
{
"epoch": 0.8763024312049158,
"grad_norm": 31.565832295867562,
"learning_rate": 4.4761226670592066e-08,
"logits/chosen": -0.9489310383796692,
"logits/rejected": -0.9287575483322144,
"logps/chosen": -1.2163779735565186,
"logps/rejected": -1.4926642179489136,
"loss": 1.3848,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -3.040944814682007,
"rewards/margins": 0.6907154321670532,
"rewards/rejected": -3.7316603660583496,
"step": 410
},
{
"epoch": 0.88698904621961,
"grad_norm": 22.344503337533215,
"learning_rate": 3.734784976300165e-08,
"logits/chosen": -0.92424076795578,
"logits/rejected": -0.913791298866272,
"logps/chosen": -1.1991280317306519,
"logps/rejected": -1.3796842098236084,
"loss": 1.3718,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -2.9978199005126953,
"rewards/margins": 0.45139074325561523,
"rewards/rejected": -3.4492106437683105,
"step": 415
},
{
"epoch": 0.897675661234304,
"grad_norm": 25.79666616123638,
"learning_rate": 3.058153372200695e-08,
"logits/chosen": -0.9958304166793823,
"logits/rejected": -0.9945958256721497,
"logps/chosen": -1.3022658824920654,
"logps/rejected": -1.4703620672225952,
"loss": 1.4021,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -3.255664348602295,
"rewards/margins": 0.4202408194541931,
"rewards/rejected": -3.675905704498291,
"step": 420
},
{
"epoch": 0.9083622762489981,
"grad_norm": 24.252773163543072,
"learning_rate": 2.4471741852423233e-08,
"logits/chosen": -0.9615448713302612,
"logits/rejected": -0.9239951968193054,
"logps/chosen": -1.2393920421600342,
"logps/rejected": -1.5796483755111694,
"loss": 1.3444,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.098480224609375,
"rewards/margins": 0.8506406545639038,
"rewards/rejected": -3.9491209983825684,
"step": 425
},
{
"epoch": 0.9190488912636923,
"grad_norm": 25.840588676616843,
"learning_rate": 1.9027019250647036e-08,
"logits/chosen": -0.9318382143974304,
"logits/rejected": -0.895746111869812,
"logps/chosen": -1.19111168384552,
"logps/rejected": -1.4055861234664917,
"loss": 1.3383,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -2.9777793884277344,
"rewards/margins": 0.536186158657074,
"rewards/rejected": -3.513965606689453,
"step": 430
},
{
"epoch": 0.9297355062783863,
"grad_norm": 24.38677858349731,
"learning_rate": 1.4254980853566246e-08,
"logits/chosen": -0.9664777517318726,
"logits/rejected": -0.9452959299087524,
"logps/chosen": -1.1789872646331787,
"logps/rejected": -1.4990911483764648,
"loss": 1.3959,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.9474682807922363,
"rewards/margins": 0.8002597093582153,
"rewards/rejected": -3.747727870941162,
"step": 435
},
{
"epoch": 0.9404221212930804,
"grad_norm": 20.846346077184034,
"learning_rate": 1.016230078838226e-08,
"logits/chosen": -0.9898012280464172,
"logits/rejected": -0.9755558967590332,
"logps/chosen": -1.205000638961792,
"logps/rejected": -1.4142088890075684,
"loss": 1.3646,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.0125012397766113,
"rewards/margins": 0.5230205059051514,
"rewards/rejected": -3.535521984100342,
"step": 440
},
{
"epoch": 0.9511087363077745,
"grad_norm": 25.087392613196315,
"learning_rate": 6.754703038239329e-09,
"logits/chosen": -0.9521238207817078,
"logits/rejected": -0.9080606698989868,
"logps/chosen": -1.1550867557525635,
"logps/rejected": -1.3055427074432373,
"loss": 1.3825,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.887716770172119,
"rewards/margins": 0.3761400282382965,
"rewards/rejected": -3.263856887817383,
"step": 445
},
{
"epoch": 0.9617953513224686,
"grad_norm": 24.725896653899223,
"learning_rate": 4.036953436716895e-09,
"logits/chosen": -0.9332659840583801,
"logits/rejected": -0.9464728236198425,
"logps/chosen": -1.2779486179351807,
"logps/rejected": -1.5308974981307983,
"loss": 1.3408,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -3.194871425628662,
"rewards/margins": 0.6323727369308472,
"rewards/rejected": -3.8272438049316406,
"step": 450
},
{
"epoch": 0.9724819663371627,
"grad_norm": 28.975077371611935,
"learning_rate": 2.0128530023804656e-09,
"logits/chosen": -0.9732203483581543,
"logits/rejected": -0.9457721710205078,
"logps/chosen": -1.345735788345337,
"logps/rejected": -1.4330954551696777,
"loss": 1.4589,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -3.3643393516540527,
"rewards/margins": 0.21839912235736847,
"rewards/rejected": -3.5827383995056152,
"step": 455
},
{
"epoch": 0.9831685813518568,
"grad_norm": 26.785266092328502,
"learning_rate": 6.852326227130833e-10,
"logits/chosen": -0.9476199150085449,
"logits/rejected": -0.9895550608634949,
"logps/chosen": -1.4385395050048828,
"logps/rejected": -1.8012183904647827,
"loss": 1.3759,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -3.596348524093628,
"rewards/margins": 0.906697154045105,
"rewards/rejected": -4.503045558929443,
"step": 460
},
{
"epoch": 0.9938551963665508,
"grad_norm": 25.28595563871724,
"learning_rate": 5.594909486328348e-11,
"logits/chosen": -1.0385969877243042,
"logits/rejected": -0.9505317807197571,
"logps/chosen": -1.1505335569381714,
"logps/rejected": -1.4205496311187744,
"loss": 1.3506,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -2.8763339519500732,
"rewards/margins": 0.6750401258468628,
"rewards/rejected": -3.5513739585876465,
"step": 465
},
{
"epoch": 0.9981298423724285,
"step": 467,
"total_flos": 0.0,
"train_loss": 1.4758259897824273,
"train_runtime": 7808.6202,
"train_samples_per_second": 7.668,
"train_steps_per_second": 0.06
}
],
"logging_steps": 5,
"max_steps": 467,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": false,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}