AALF's picture
Upload folder using huggingface_hub
cd55783 verified
{
"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.010686615014694095,
"grad_norm": 56.81669148706694,
"learning_rate": 1.0638297872340425e-07,
"logits/chosen": -1.0553786754608154,
"logits/rejected": -0.9845818281173706,
"logps/chosen": -0.7874154448509216,
"logps/rejected": -0.7125983834266663,
"loss": 3.9428,
"rewards/accuracies": 0.4124999940395355,
"rewards/chosen": -7.874154567718506,
"rewards/margins": -0.7481697797775269,
"rewards/rejected": -7.125983238220215,
"step": 5
},
{
"epoch": 0.02137323002938819,
"grad_norm": 90.64309336283279,
"learning_rate": 2.127659574468085e-07,
"logits/chosen": -0.9677240252494812,
"logits/rejected": -0.9489688873291016,
"logps/chosen": -0.8535294532775879,
"logps/rejected": -1.0613572597503662,
"loss": 3.6689,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -8.535294532775879,
"rewards/margins": 2.0782790184020996,
"rewards/rejected": -10.613574028015137,
"step": 10
},
{
"epoch": 0.03205984504408229,
"grad_norm": 110.04849731267365,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -0.9973424673080444,
"logits/rejected": -0.9682930111885071,
"logps/chosen": -0.8421398401260376,
"logps/rejected": -0.9425029754638672,
"loss": 3.9663,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -8.42139720916748,
"rewards/margins": 1.0036319494247437,
"rewards/rejected": -9.425029754638672,
"step": 15
},
{
"epoch": 0.04274646005877638,
"grad_norm": 89.89735367257849,
"learning_rate": 4.25531914893617e-07,
"logits/chosen": -1.0042946338653564,
"logits/rejected": -0.9698360562324524,
"logps/chosen": -0.6765571236610413,
"logps/rejected": -0.7212239503860474,
"loss": 3.798,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -6.765571594238281,
"rewards/margins": 0.44666871428489685,
"rewards/rejected": -7.212240695953369,
"step": 20
},
{
"epoch": 0.053433075073470476,
"grad_norm": 40.5523084584386,
"learning_rate": 5.319148936170212e-07,
"logits/chosen": -1.042263388633728,
"logits/rejected": -1.020334005355835,
"logps/chosen": -0.6275808215141296,
"logps/rejected": -0.6092665195465088,
"loss": 3.5027,
"rewards/accuracies": 0.38749998807907104,
"rewards/chosen": -6.275807857513428,
"rewards/margins": -0.18314316868782043,
"rewards/rejected": -6.092665195465088,
"step": 25
},
{
"epoch": 0.06411969008816458,
"grad_norm": 82.97126019183185,
"learning_rate": 6.382978723404255e-07,
"logits/chosen": -1.0934265851974487,
"logits/rejected": -1.0273423194885254,
"logps/chosen": -0.4222745895385742,
"logps/rejected": -0.4437929689884186,
"loss": 3.2915,
"rewards/accuracies": 0.5,
"rewards/chosen": -4.2227463722229,
"rewards/margins": 0.21518389880657196,
"rewards/rejected": -4.437930107116699,
"step": 30
},
{
"epoch": 0.07480630510285867,
"grad_norm": 45.671980780198346,
"learning_rate": 7.446808510638297e-07,
"logits/chosen": -1.1021089553833008,
"logits/rejected": -1.016642689704895,
"logps/chosen": -0.40351754426956177,
"logps/rejected": -0.4182681143283844,
"loss": 3.0336,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -4.035175800323486,
"rewards/margins": 0.14750559628009796,
"rewards/rejected": -4.182681083679199,
"step": 35
},
{
"epoch": 0.08549292011755276,
"grad_norm": 56.834561606796534,
"learning_rate": 8.51063829787234e-07,
"logits/chosen": -1.0844147205352783,
"logits/rejected": -1.0052152872085571,
"logps/chosen": -0.3840138018131256,
"logps/rejected": -0.41540414094924927,
"loss": 2.9873,
"rewards/accuracies": 0.5,
"rewards/chosen": -3.8401379585266113,
"rewards/margins": 0.3139037489891052,
"rewards/rejected": -4.154041290283203,
"step": 40
},
{
"epoch": 0.09617953513224686,
"grad_norm": 57.81609876406988,
"learning_rate": 9.574468085106384e-07,
"logits/chosen": -1.0846834182739258,
"logits/rejected": -0.9838830828666687,
"logps/chosen": -0.3058851957321167,
"logps/rejected": -0.3447844386100769,
"loss": 2.9473,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -3.058851957321167,
"rewards/margins": 0.3889924883842468,
"rewards/rejected": -3.4478442668914795,
"step": 45
},
{
"epoch": 0.10686615014694095,
"grad_norm": 107.63293409046761,
"learning_rate": 9.998741174712533e-07,
"logits/chosen": -1.183288335800171,
"logits/rejected": -1.0858242511749268,
"logps/chosen": -0.3091445863246918,
"logps/rejected": -0.3627066910266876,
"loss": 2.9167,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -3.0914459228515625,
"rewards/margins": 0.5356208682060242,
"rewards/rejected": -3.6270668506622314,
"step": 50
},
{
"epoch": 0.11755276516163506,
"grad_norm": 40.363889487849875,
"learning_rate": 9.991050648838675e-07,
"logits/chosen": -1.1698521375656128,
"logits/rejected": -1.1601616144180298,
"logps/chosen": -0.29625993967056274,
"logps/rejected": -0.33069461584091187,
"loss": 2.9262,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -2.962599515914917,
"rewards/margins": 0.34434670209884644,
"rewards/rejected": -3.306946277618408,
"step": 55
},
{
"epoch": 0.12823938017632916,
"grad_norm": 49.68454575475429,
"learning_rate": 9.97637968732563e-07,
"logits/chosen": -1.2926113605499268,
"logits/rejected": -1.2354081869125366,
"logps/chosen": -0.34236031770706177,
"logps/rejected": -0.370178759098053,
"loss": 2.828,
"rewards/accuracies": 0.5,
"rewards/chosen": -3.423603057861328,
"rewards/margins": 0.2781844735145569,
"rewards/rejected": -3.7017874717712402,
"step": 60
},
{
"epoch": 0.13892599519102325,
"grad_norm": 64.97695318174799,
"learning_rate": 9.954748808839674e-07,
"logits/chosen": -1.2277723550796509,
"logits/rejected": -1.1402448415756226,
"logps/chosen": -0.33444857597351074,
"logps/rejected": -0.38238364458084106,
"loss": 2.8768,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -3.3444857597351074,
"rewards/margins": 0.47935038805007935,
"rewards/rejected": -3.8238365650177,
"step": 65
},
{
"epoch": 0.14961261020571734,
"grad_norm": 42.426764348756556,
"learning_rate": 9.926188266120295e-07,
"logits/chosen": -1.2491295337677002,
"logits/rejected": -1.2098782062530518,
"logps/chosen": -0.2992488443851471,
"logps/rejected": -0.34425443410873413,
"loss": 2.8068,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -2.992488145828247,
"rewards/margins": 0.45005664229393005,
"rewards/rejected": -3.442544460296631,
"step": 70
},
{
"epoch": 0.16029922522041143,
"grad_norm": 55.41018623107141,
"learning_rate": 9.890738003669027e-07,
"logits/chosen": -1.224936604499817,
"logits/rejected": -1.1226717233657837,
"logps/chosen": -0.3549841642379761,
"logps/rejected": -0.42627596855163574,
"loss": 2.6663,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -3.5498414039611816,
"rewards/margins": 0.7129180431365967,
"rewards/rejected": -4.262759685516357,
"step": 75
},
{
"epoch": 0.17098584023510552,
"grad_norm": 67.19768319790357,
"learning_rate": 9.848447601883433e-07,
"logits/chosen": -1.22861647605896,
"logits/rejected": -1.173981785774231,
"logps/chosen": -0.35033124685287476,
"logps/rejected": -0.3842136263847351,
"loss": 2.6237,
"rewards/accuracies": 0.6875,
"rewards/chosen": -3.503312349319458,
"rewards/margins": 0.3388238847255707,
"rewards/rejected": -3.8421363830566406,
"step": 80
},
{
"epoch": 0.18167245524979964,
"grad_norm": 56.06934270502333,
"learning_rate": 9.799376207714444e-07,
"logits/chosen": -1.1919827461242676,
"logits/rejected": -1.145477056503296,
"logps/chosen": -0.3422418534755707,
"logps/rejected": -0.4096073508262634,
"loss": 2.6001,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -3.4224185943603516,
"rewards/margins": 0.6736553311347961,
"rewards/rejected": -4.096073627471924,
"step": 85
},
{
"epoch": 0.19235907026449373,
"grad_norm": 66.32485026269077,
"learning_rate": 9.743592451943998e-07,
"logits/chosen": -1.2551718950271606,
"logits/rejected": -1.1607410907745361,
"logps/chosen": -0.39799433946609497,
"logps/rejected": -0.47051042318344116,
"loss": 2.6013,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -3.9799437522888184,
"rewards/margins": 0.7251602411270142,
"rewards/rejected": -4.705103397369385,
"step": 90
},
{
"epoch": 0.20304568527918782,
"grad_norm": 88.72645076406972,
"learning_rate": 9.681174353198686e-07,
"logits/chosen": -1.202864646911621,
"logits/rejected": -1.0979254245758057,
"logps/chosen": -0.40985170006752014,
"logps/rejected": -0.4939080774784088,
"loss": 2.47,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -4.098517417907715,
"rewards/margins": 0.8405635952949524,
"rewards/rejected": -4.939080238342285,
"step": 95
},
{
"epoch": 0.2137323002938819,
"grad_norm": 78.62356168132955,
"learning_rate": 9.612209208833646e-07,
"logits/chosen": -1.210082769393921,
"logits/rejected": -1.19521963596344,
"logps/chosen": -0.4767809808254242,
"logps/rejected": -0.6341355443000793,
"loss": 2.4364,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -4.767809867858887,
"rewards/margins": 1.5735454559326172,
"rewards/rejected": -6.341355323791504,
"step": 100
},
{
"epoch": 0.224418915308576,
"grad_norm": 182.17113716520097,
"learning_rate": 9.536793472839324e-07,
"logits/chosen": -1.318027138710022,
"logits/rejected": -1.2231770753860474,
"logps/chosen": -0.5009247064590454,
"logps/rejected": -0.6060206890106201,
"loss": 2.4925,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -5.009247779846191,
"rewards/margins": 1.050958514213562,
"rewards/rejected": -6.060206413269043,
"step": 105
},
{
"epoch": 0.2351055303232701,
"grad_norm": 97.52898918968008,
"learning_rate": 9.455032620941839e-07,
"logits/chosen": -1.3112456798553467,
"logits/rejected": -1.2372304201126099,
"logps/chosen": -0.4988152086734772,
"logps/rejected": -0.6063084602355957,
"loss": 2.4096,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -4.988151550292969,
"rewards/margins": 1.074933409690857,
"rewards/rejected": -6.063085079193115,
"step": 110
},
{
"epoch": 0.2457921453379642,
"grad_norm": 65.2666290162174,
"learning_rate": 9.367041003085648e-07,
"logits/chosen": -1.2346036434173584,
"logits/rejected": -1.155144453048706,
"logps/chosen": -0.46027320623397827,
"logps/rejected": -0.539958655834198,
"loss": 2.4476,
"rewards/accuracies": 0.625,
"rewards/chosen": -4.602732181549072,
"rewards/margins": 0.7968543767929077,
"rewards/rejected": -5.3995866775512695,
"step": 115
},
{
"epoch": 0.2564787603526583,
"grad_norm": 94.23205420362748,
"learning_rate": 9.272941683504808e-07,
"logits/chosen": -1.2661981582641602,
"logits/rejected": -1.2356774806976318,
"logps/chosen": -0.5204485058784485,
"logps/rejected": -0.6347986459732056,
"loss": 2.313,
"rewards/accuracies": 0.625,
"rewards/chosen": -5.204484939575195,
"rewards/margins": 1.1435019969940186,
"rewards/rejected": -6.347987174987793,
"step": 120
},
{
"epoch": 0.2671653753673524,
"grad_norm": 104.425486907994,
"learning_rate": 9.172866268606513e-07,
"logits/chosen": -1.4145832061767578,
"logits/rejected": -1.3696868419647217,
"logps/chosen": -0.5677964687347412,
"logps/rejected": -0.7819775342941284,
"loss": 2.0573,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -5.677964210510254,
"rewards/margins": 2.1418113708496094,
"rewards/rejected": -7.8197760581970215,
"step": 125
},
{
"epoch": 0.2778519903820465,
"grad_norm": 118.15756723487034,
"learning_rate": 9.066954722907638e-07,
"logits/chosen": -1.2262728214263916,
"logits/rejected": -1.1968516111373901,
"logps/chosen": -0.5714520812034607,
"logps/rejected": -0.7978314757347107,
"loss": 2.229,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -5.7145209312438965,
"rewards/margins": 2.2637939453125,
"rewards/rejected": -7.9783148765563965,
"step": 130
},
{
"epoch": 0.2885386053967406,
"grad_norm": 71.01619814345214,
"learning_rate": 8.955355173281707e-07,
"logits/chosen": -1.423404574394226,
"logits/rejected": -1.3302305936813354,
"logps/chosen": -0.6018441915512085,
"logps/rejected": -0.7836464047431946,
"loss": 1.9147,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -6.018441677093506,
"rewards/margins": 1.8180221319198608,
"rewards/rejected": -7.836463928222656,
"step": 135
},
{
"epoch": 0.2992252204114347,
"grad_norm": 66.78115766091884,
"learning_rate": 8.838223701790055e-07,
"logits/chosen": -1.384155511856079,
"logits/rejected": -1.3765170574188232,
"logps/chosen": -0.6419236660003662,
"logps/rejected": -0.8896235227584839,
"loss": 1.9382,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -6.419236660003662,
"rewards/margins": 2.476998805999756,
"rewards/rejected": -8.896234512329102,
"step": 140
},
{
"epoch": 0.30991183542612877,
"grad_norm": 102.33456775014218,
"learning_rate": 8.71572412738697e-07,
"logits/chosen": -1.3982570171356201,
"logits/rejected": -1.3454999923706055,
"logps/chosen": -0.6346415281295776,
"logps/rejected": -0.8345499038696289,
"loss": 1.8347,
"rewards/accuracies": 0.8125,
"rewards/chosen": -6.346415042877197,
"rewards/margins": 1.9990848302841187,
"rewards/rejected": -8.345499038696289,
"step": 145
},
{
"epoch": 0.32059845044082286,
"grad_norm": 104.88848048977833,
"learning_rate": 8.588027776804058e-07,
"logits/chosen": -1.3800594806671143,
"logits/rejected": -1.3424351215362549,
"logps/chosen": -0.7959817051887512,
"logps/rejected": -1.088423252105713,
"loss": 1.9956,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -7.959817409515381,
"rewards/margins": 2.9244155883789062,
"rewards/rejected": -10.884233474731445,
"step": 150
},
{
"epoch": 0.33128506545551695,
"grad_norm": 86.74090140921074,
"learning_rate": 8.455313244934324e-07,
"logits/chosen": -1.289271354675293,
"logits/rejected": -1.221858263015747,
"logps/chosen": -0.7765235900878906,
"logps/rejected": -1.0151869058609009,
"loss": 1.905,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -7.76523494720459,
"rewards/margins": 2.386632204055786,
"rewards/rejected": -10.15186882019043,
"step": 155
},
{
"epoch": 0.34197168047021104,
"grad_norm": 103.52900771453592,
"learning_rate": 8.317766145051057e-07,
"logits/chosen": -1.3143701553344727,
"logits/rejected": -1.2429687976837158,
"logps/chosen": -0.8146079182624817,
"logps/rejected": -1.076160192489624,
"loss": 1.9347,
"rewards/accuracies": 0.75,
"rewards/chosen": -8.146079063415527,
"rewards/margins": 2.6155219078063965,
"rewards/rejected": -10.761602401733398,
"step": 160
},
{
"epoch": 0.3526582954849052,
"grad_norm": 91.78991869467103,
"learning_rate": 8.175578849210894e-07,
"logits/chosen": -1.277517557144165,
"logits/rejected": -1.2467105388641357,
"logps/chosen": -0.8164845705032349,
"logps/rejected": -1.0702334642410278,
"loss": 1.8253,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -8.16484546661377,
"rewards/margins": 2.5374886989593506,
"rewards/rejected": -10.7023344039917,
"step": 165
},
{
"epoch": 0.36334491049959927,
"grad_norm": 146.18164794432158,
"learning_rate": 8.028950219204099e-07,
"logits/chosen": -1.4038187265396118,
"logits/rejected": -1.32179856300354,
"logps/chosen": -0.9452447891235352,
"logps/rejected": -1.285444974899292,
"loss": 1.8459,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -9.452447891235352,
"rewards/margins": 3.4020028114318848,
"rewards/rejected": -12.854451179504395,
"step": 170
},
{
"epoch": 0.37403152551429336,
"grad_norm": 74.04178740704424,
"learning_rate": 7.878085328428368e-07,
"logits/chosen": -1.2986555099487305,
"logits/rejected": -1.2864700555801392,
"logps/chosen": -0.8545284271240234,
"logps/rejected": -1.2237508296966553,
"loss": 1.7393,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -8.545283317565918,
"rewards/margins": 3.6922249794006348,
"rewards/rejected": -12.237508773803711,
"step": 175
},
{
"epoch": 0.38471814052898745,
"grad_norm": 146.93436823237963,
"learning_rate": 7.723195175075135e-07,
"logits/chosen": -1.4434401988983154,
"logits/rejected": -1.3974043130874634,
"logps/chosen": -1.084814190864563,
"logps/rejected": -1.4313297271728516,
"loss": 1.5783,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -10.84814167022705,
"rewards/margins": 3.4651541709899902,
"rewards/rejected": -14.3132963180542,
"step": 180
},
{
"epoch": 0.39540475554368154,
"grad_norm": 300.3576490829542,
"learning_rate": 7.564496387029531e-07,
"logits/chosen": -1.331994891166687,
"logits/rejected": -1.3323677778244019,
"logps/chosen": -1.0099122524261475,
"logps/rejected": -1.400909185409546,
"loss": 1.6356,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -10.099123001098633,
"rewards/margins": 3.9099678993225098,
"rewards/rejected": -14.0090913772583,
"step": 185
},
{
"epoch": 0.40609137055837563,
"grad_norm": 117.205754582783,
"learning_rate": 7.402210918896689e-07,
"logits/chosen": -1.3903909921646118,
"logits/rejected": -1.395379662513733,
"logps/chosen": -0.9584177136421204,
"logps/rejected": -1.3904134035110474,
"loss": 1.5868,
"rewards/accuracies": 0.8125,
"rewards/chosen": -9.584177017211914,
"rewards/margins": 4.319957256317139,
"rewards/rejected": -13.904134750366211,
"step": 190
},
{
"epoch": 0.4167779855730697,
"grad_norm": 193.5248033898295,
"learning_rate": 7.236565741578162e-07,
"logits/chosen": -1.479113221168518,
"logits/rejected": -1.431293249130249,
"logps/chosen": -1.1905192136764526,
"logps/rejected": -1.5327824354171753,
"loss": 1.4974,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -11.905191421508789,
"rewards/margins": 3.4226326942443848,
"rewards/rejected": -15.327824592590332,
"step": 195
},
{
"epoch": 0.4274646005877638,
"grad_norm": 72.90690697186486,
"learning_rate": 7.067792524832603e-07,
"logits/chosen": -1.464138388633728,
"logits/rejected": -1.4378821849822998,
"logps/chosen": -1.2057116031646729,
"logps/rejected": -1.6115760803222656,
"loss": 1.6772,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -12.05711555480957,
"rewards/margins": 4.058643341064453,
"rewards/rejected": -16.115758895874023,
"step": 200
},
{
"epoch": 0.4381512156024579,
"grad_norm": 73.64255724761823,
"learning_rate": 6.896127313264642e-07,
"logits/chosen": -1.463689923286438,
"logits/rejected": -1.4153610467910767,
"logps/chosen": -0.9573764801025391,
"logps/rejected": -1.4560494422912598,
"loss": 1.5083,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -9.57376480102539,
"rewards/margins": 4.986730098724365,
"rewards/rejected": -14.560495376586914,
"step": 205
},
{
"epoch": 0.448837830617152,
"grad_norm": 95.73515995991542,
"learning_rate": 6.721810196195174e-07,
"logits/chosen": -1.5166422128677368,
"logits/rejected": -1.5037015676498413,
"logps/chosen": -1.1732789278030396,
"logps/rejected": -1.5742861032485962,
"loss": 1.5533,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -11.732789993286133,
"rewards/margins": 4.010071754455566,
"rewards/rejected": -15.7428617477417,
"step": 210
},
{
"epoch": 0.45952444563184613,
"grad_norm": 111.72733375525473,
"learning_rate": 6.545084971874736e-07,
"logits/chosen": -1.5549886226654053,
"logits/rejected": -1.4790352582931519,
"logps/chosen": -1.1950467824935913,
"logps/rejected": -1.602651596069336,
"loss": 1.5756,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -11.950468063354492,
"rewards/margins": 4.076048851013184,
"rewards/rejected": -16.02651596069336,
"step": 215
},
{
"epoch": 0.4702110606465402,
"grad_norm": 95.02342077059286,
"learning_rate": 6.3661988065096e-07,
"logits/chosen": -1.483459711074829,
"logits/rejected": -1.454347014427185,
"logps/chosen": -1.2087767124176025,
"logps/rejected": -1.5687811374664307,
"loss": 1.434,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -12.087766647338867,
"rewards/margins": 3.6000449657440186,
"rewards/rejected": -15.687810897827148,
"step": 220
},
{
"epoch": 0.4808976756612343,
"grad_norm": 80.04963040030957,
"learning_rate": 6.185401888577487e-07,
"logits/chosen": -1.4368016719818115,
"logits/rejected": -1.4539368152618408,
"logps/chosen": -1.1799414157867432,
"logps/rejected": -1.7078040838241577,
"loss": 1.2062,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -11.799413681030273,
"rewards/margins": 5.278626918792725,
"rewards/rejected": -17.078041076660156,
"step": 225
},
{
"epoch": 0.4915842906759284,
"grad_norm": 90.27822366362969,
"learning_rate": 6.002947078916364e-07,
"logits/chosen": -1.5053470134735107,
"logits/rejected": -1.4647208452224731,
"logps/chosen": -1.3405472040176392,
"logps/rejected": -1.8345695734024048,
"loss": 1.1551,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -13.405473709106445,
"rewards/margins": 4.9402241706848145,
"rewards/rejected": -18.3456974029541,
"step": 230
},
{
"epoch": 0.5022709056906225,
"grad_norm": 198.7780967148136,
"learning_rate": 5.819089557075688e-07,
"logits/chosen": -1.463410496711731,
"logits/rejected": -1.4734344482421875,
"logps/chosen": -1.4616682529449463,
"logps/rejected": -1.8893461227416992,
"loss": 1.5181,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -14.616681098937988,
"rewards/margins": 4.276780128479004,
"rewards/rejected": -18.893463134765625,
"step": 235
},
{
"epoch": 0.5129575207053166,
"grad_norm": 92.23381750381444,
"learning_rate": 5.634086464424742e-07,
"logits/chosen": -1.5198612213134766,
"logits/rejected": -1.5063666105270386,
"logps/chosen": -1.4451749324798584,
"logps/rejected": -2.1483402252197266,
"loss": 1.3406,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -14.451748847961426,
"rewards/margins": 7.031652927398682,
"rewards/rejected": -21.483402252197266,
"step": 240
},
{
"epoch": 0.5236441357200107,
"grad_norm": 101.7965922889438,
"learning_rate": 5.448196544517167e-07,
"logits/chosen": -1.4860137701034546,
"logits/rejected": -1.4609707593917847,
"logps/chosen": -1.2388508319854736,
"logps/rejected": -1.7417356967926025,
"loss": 1.2849,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -12.388508796691895,
"rewards/margins": 5.0288496017456055,
"rewards/rejected": -17.4173583984375,
"step": 245
},
{
"epoch": 0.5343307507347048,
"grad_norm": 112.84840492502885,
"learning_rate": 5.26167978121472e-07,
"logits/chosen": -1.545151710510254,
"logits/rejected": -1.48394775390625,
"logps/chosen": -1.2914670705795288,
"logps/rejected": -1.867653489112854,
"loss": 1.226,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": -12.9146728515625,
"rewards/margins": 5.761862754821777,
"rewards/rejected": -18.676532745361328,
"step": 250
},
{
"epoch": 0.5450173657493989,
"grad_norm": 81.89729359110314,
"learning_rate": 5.074797035076318e-07,
"logits/chosen": -1.5118135213851929,
"logits/rejected": -1.5217331647872925,
"logps/chosen": -1.5564649105072021,
"logps/rejected": -1.9826581478118896,
"loss": 1.2119,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -15.564648628234863,
"rewards/margins": 4.261934757232666,
"rewards/rejected": -19.826580047607422,
"step": 255
},
{
"epoch": 0.555703980764093,
"grad_norm": 97.54502724467275,
"learning_rate": 4.887809678520975e-07,
"logits/chosen": -1.5081006288528442,
"logits/rejected": -1.4816151857376099,
"logps/chosen": -1.337573766708374,
"logps/rejected": -1.8502023220062256,
"loss": 1.226,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -13.375738143920898,
"rewards/margins": 5.126286029815674,
"rewards/rejected": -18.502025604248047,
"step": 260
},
{
"epoch": 0.566390595778787,
"grad_norm": 99.34071724726068,
"learning_rate": 4.700979230274829e-07,
"logits/chosen": -1.4894455671310425,
"logits/rejected": -1.4260051250457764,
"logps/chosen": -1.388643503189087,
"logps/rejected": -1.9015836715698242,
"loss": 1.1655,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -13.886438369750977,
"rewards/margins": 5.129402160644531,
"rewards/rejected": -19.015838623046875,
"step": 265
},
{
"epoch": 0.5770772107934812,
"grad_norm": 150.30388856950947,
"learning_rate": 4.514566989613559e-07,
"logits/chosen": -1.5251694917678833,
"logits/rejected": -1.510664701461792,
"logps/chosen": -1.352906346321106,
"logps/rejected": -1.9253631830215454,
"loss": 1.2737,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -13.529065132141113,
"rewards/margins": 5.724569320678711,
"rewards/rejected": -19.253633499145508,
"step": 270
},
{
"epoch": 0.5877638258081752,
"grad_norm": 106.62043935215682,
"learning_rate": 4.328833670911724e-07,
"logits/chosen": -1.6284313201904297,
"logits/rejected": -1.5873377323150635,
"logps/chosen": -1.4256664514541626,
"logps/rejected": -2.05672025680542,
"loss": 1.356,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -14.256665229797363,
"rewards/margins": 6.310537338256836,
"rewards/rejected": -20.567203521728516,
"step": 275
},
{
"epoch": 0.5984504408228694,
"grad_norm": 84.26686987382098,
"learning_rate": 4.144039039010124e-07,
"logits/chosen": -1.4645414352416992,
"logits/rejected": -1.4171415567398071,
"logps/chosen": -1.4661189317703247,
"logps/rejected": -1.9972292184829712,
"loss": 1.1226,
"rewards/accuracies": 0.875,
"rewards/chosen": -14.661189079284668,
"rewards/margins": 5.311104774475098,
"rewards/rejected": -19.972293853759766,
"step": 280
},
{
"epoch": 0.6091370558375635,
"grad_norm": 111.79495099216508,
"learning_rate": 3.960441545911204e-07,
"logits/chosen": -1.580186128616333,
"logits/rejected": -1.4852710962295532,
"logps/chosen": -1.6370970010757446,
"logps/rejected": -2.1605658531188965,
"loss": 1.3708,
"rewards/accuracies": 0.8125,
"rewards/chosen": -16.370969772338867,
"rewards/margins": 5.234689235687256,
"rewards/rejected": -21.605661392211914,
"step": 285
},
{
"epoch": 0.6198236708522575,
"grad_norm": 89.00502902448149,
"learning_rate": 3.778297969310529e-07,
"logits/chosen": -1.6441500186920166,
"logits/rejected": -1.6167094707489014,
"logps/chosen": -1.5200397968292236,
"logps/rejected": -2.0216360092163086,
"loss": 1.19,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -15.200396537780762,
"rewards/margins": 5.01596212387085,
"rewards/rejected": -20.216358184814453,
"step": 290
},
{
"epoch": 0.6305102858669517,
"grad_norm": 101.58972547942741,
"learning_rate": 3.5978630534699865e-07,
"logits/chosen": -1.5900800228118896,
"logits/rejected": -1.5477546453475952,
"logps/chosen": -1.4291937351226807,
"logps/rejected": -2.104898691177368,
"loss": 1.0996,
"rewards/accuracies": 0.875,
"rewards/chosen": -14.291936874389648,
"rewards/margins": 6.757050514221191,
"rewards/rejected": -21.04898452758789,
"step": 295
},
{
"epoch": 0.6411969008816457,
"grad_norm": 114.7765696570134,
"learning_rate": 3.4193891529348795e-07,
"logits/chosen": -1.5715553760528564,
"logits/rejected": -1.5747449398040771,
"logps/chosen": -1.4967474937438965,
"logps/rejected": -2.078972101211548,
"loss": 1.0805,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -14.967473983764648,
"rewards/margins": 5.8222455978393555,
"rewards/rejected": -20.789722442626953,
"step": 300
},
{
"epoch": 0.6518835158963399,
"grad_norm": 123.5878416310842,
"learning_rate": 3.243125879593286e-07,
"logits/chosen": -1.4123915433883667,
"logits/rejected": -1.4187986850738525,
"logps/chosen": -1.2821505069732666,
"logps/rejected": -2.120354413986206,
"loss": 1.1007,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -12.821507453918457,
"rewards/margins": 8.382037162780762,
"rewards/rejected": -21.203542709350586,
"step": 305
},
{
"epoch": 0.6625701309110339,
"grad_norm": 94.61228245151283,
"learning_rate": 3.069319753571269e-07,
"logits/chosen": -1.5294448137283325,
"logits/rejected": -1.5259336233139038,
"logps/chosen": -1.5088483095169067,
"logps/rejected": -2.108046531677246,
"loss": 1.147,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -15.088482856750488,
"rewards/margins": 5.991982460021973,
"rewards/rejected": -21.080463409423828,
"step": 310
},
{
"epoch": 0.673256745925728,
"grad_norm": 88.47117948255506,
"learning_rate": 2.898213858452173e-07,
"logits/chosen": -1.4750919342041016,
"logits/rejected": -1.4806705713272095,
"logps/chosen": -1.588282823562622,
"logps/rejected": -2.2189738750457764,
"loss": 1.0766,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -15.882827758789062,
"rewards/margins": 6.306912899017334,
"rewards/rejected": -22.189741134643555,
"step": 315
},
{
"epoch": 0.6839433609404221,
"grad_norm": 125.62338182234869,
"learning_rate": 2.730047501302266e-07,
"logits/chosen": -1.6073205471038818,
"logits/rejected": -1.538448691368103,
"logps/chosen": -1.5898334980010986,
"logps/rejected": -2.3282947540283203,
"loss": 1.2473,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -15.898333549499512,
"rewards/margins": 7.3846116065979,
"rewards/rejected": -23.282947540283203,
"step": 320
},
{
"epoch": 0.6946299759551162,
"grad_norm": 111.24354162328734,
"learning_rate": 2.5650558779781635e-07,
"logits/chosen": -1.5743417739868164,
"logits/rejected": -1.566840410232544,
"logps/chosen": -1.6232945919036865,
"logps/rejected": -2.2983360290527344,
"loss": 1.0508,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -16.232946395874023,
"rewards/margins": 6.750412940979004,
"rewards/rejected": -22.983360290527344,
"step": 325
},
{
"epoch": 0.7053165909698104,
"grad_norm": 85.89351794884949,
"learning_rate": 2.403469744184154e-07,
"logits/chosen": -1.5692135095596313,
"logits/rejected": -1.5725057125091553,
"logps/chosen": -1.542747974395752,
"logps/rejected": -2.168196678161621,
"loss": 1.0903,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -15.42747974395752,
"rewards/margins": 6.254487037658691,
"rewards/rejected": -21.68196678161621,
"step": 330
},
{
"epoch": 0.7160032059845044,
"grad_norm": 116.80439013939746,
"learning_rate": 2.2455150927394878e-07,
"logits/chosen": -1.55947744846344,
"logits/rejected": -1.501947283744812,
"logps/chosen": -1.5338480472564697,
"logps/rejected": -2.0976195335388184,
"loss": 1.0239,
"rewards/accuracies": 0.875,
"rewards/chosen": -15.338479995727539,
"rewards/margins": 5.637716770172119,
"rewards/rejected": -20.976198196411133,
"step": 335
},
{
"epoch": 0.7266898209991985,
"grad_norm": 78.12195704672156,
"learning_rate": 2.0914128375069722e-07,
"logits/chosen": -1.4831820726394653,
"logits/rejected": -1.5086932182312012,
"logps/chosen": -1.5225416421890259,
"logps/rejected": -2.1293392181396484,
"loss": 0.9575,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.22541618347168,
"rewards/margins": 6.0679755210876465,
"rewards/rejected": -21.293392181396484,
"step": 340
},
{
"epoch": 0.7373764360138926,
"grad_norm": 91.10910799077911,
"learning_rate": 1.9413785044249676e-07,
"logits/chosen": -1.5832130908966064,
"logits/rejected": -1.5328218936920166,
"logps/chosen": -1.4637863636016846,
"logps/rejected": -2.059370517730713,
"loss": 1.0804,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -14.637863159179688,
"rewards/margins": 5.95583963394165,
"rewards/rejected": -20.593706130981445,
"step": 345
},
{
"epoch": 0.7480630510285867,
"grad_norm": 120.29698335176921,
"learning_rate": 1.7956219300748792e-07,
"logits/chosen": -1.519425868988037,
"logits/rejected": -1.5000859498977661,
"logps/chosen": -1.5877223014831543,
"logps/rejected": -2.4219236373901367,
"loss": 1.282,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.877223014831543,
"rewards/margins": 8.342010498046875,
"rewards/rejected": -24.219234466552734,
"step": 350
},
{
"epoch": 0.7587496660432808,
"grad_norm": 114.7323293702361,
"learning_rate": 1.6543469682057104e-07,
"logits/chosen": -1.5452375411987305,
"logits/rejected": -1.477497935295105,
"logps/chosen": -1.8073337078094482,
"logps/rejected": -2.5182600021362305,
"loss": 1.2039,
"rewards/accuracies": 0.875,
"rewards/chosen": -18.073339462280273,
"rewards/margins": 7.109262943267822,
"rewards/rejected": -25.182600021362305,
"step": 355
},
{
"epoch": 0.7694362810579749,
"grad_norm": 101.20181237745652,
"learning_rate": 1.5177512046261666e-07,
"logits/chosen": -1.5502760410308838,
"logits/rejected": -1.5708996057510376,
"logps/chosen": -1.6630090475082397,
"logps/rejected": -2.226963520050049,
"loss": 1.0971,
"rewards/accuracies": 0.875,
"rewards/chosen": -16.63009262084961,
"rewards/margins": 5.639544486999512,
"rewards/rejected": -22.269634246826172,
"step": 360
},
{
"epoch": 0.7801228960726689,
"grad_norm": 139.38000134354874,
"learning_rate": 1.3860256808630427e-07,
"logits/chosen": -1.5358211994171143,
"logits/rejected": -1.5561330318450928,
"logps/chosen": -1.6269845962524414,
"logps/rejected": -2.3564658164978027,
"loss": 1.1184,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -16.269845962524414,
"rewards/margins": 7.294811248779297,
"rewards/rejected": -23.564655303955078,
"step": 365
},
{
"epoch": 0.7908095110873631,
"grad_norm": 104.011040382236,
"learning_rate": 1.2593546269723647e-07,
"logits/chosen": -1.5664985179901123,
"logits/rejected": -1.4946684837341309,
"logps/chosen": -1.5413891077041626,
"logps/rejected": -2.3655648231506348,
"loss": 1.0455,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.413891792297363,
"rewards/margins": 8.241754531860352,
"rewards/rejected": -23.6556453704834,
"step": 370
},
{
"epoch": 0.8014961261020572,
"grad_norm": 98.02139267959178,
"learning_rate": 1.1379152038770029e-07,
"logits/chosen": -1.5228099822998047,
"logits/rejected": -1.5087474584579468,
"logps/chosen": -1.480344295501709,
"logps/rejected": -2.425346851348877,
"loss": 1.2231,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -14.803441047668457,
"rewards/margins": 9.45002555847168,
"rewards/rejected": -24.253467559814453,
"step": 375
},
{
"epoch": 0.8121827411167513,
"grad_norm": 134.01729655272072,
"learning_rate": 1.0218772555910954e-07,
"logits/chosen": -1.497368574142456,
"logits/rejected": -1.5044596195220947,
"logps/chosen": -1.594871163368225,
"logps/rejected": -2.4201245307922363,
"loss": 1.2771,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -15.948710441589355,
"rewards/margins": 8.252533912658691,
"rewards/rejected": -24.201244354248047,
"step": 380
},
{
"epoch": 0.8228693561314454,
"grad_norm": 74.68961856468854,
"learning_rate": 9.114030716778432e-08,
"logits/chosen": -1.551857352256775,
"logits/rejected": -1.535230278968811,
"logps/chosen": -1.6363334655761719,
"logps/rejected": -2.4195470809936523,
"loss": 1.0466,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -16.363332748413086,
"rewards/margins": 7.832139015197754,
"rewards/rejected": -24.195472717285156,
"step": 385
},
{
"epoch": 0.8335559711461394,
"grad_norm": 107.5882104761143,
"learning_rate": 8.066471602728803e-08,
"logits/chosen": -1.5829932689666748,
"logits/rejected": -1.5114977359771729,
"logps/chosen": -1.5258924961090088,
"logps/rejected": -2.1448683738708496,
"loss": 1.0185,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.25892448425293,
"rewards/margins": 6.189759254455566,
"rewards/rejected": -21.448684692382812,
"step": 390
},
{
"epoch": 0.8442425861608336,
"grad_norm": 102.80663837958856,
"learning_rate": 7.077560319906694e-08,
"logits/chosen": -1.4724600315093994,
"logits/rejected": -1.4828989505767822,
"logps/chosen": -1.5901472568511963,
"logps/rejected": -2.1642448902130127,
"loss": 1.1312,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -15.901473999023438,
"rewards/margins": 5.740975379943848,
"rewards/rejected": -21.642446517944336,
"step": 395
},
{
"epoch": 0.8549292011755276,
"grad_norm": 101.28973665181952,
"learning_rate": 6.148679950161672e-08,
"logits/chosen": -1.561643123626709,
"logits/rejected": -1.5329078435897827,
"logps/chosen": -1.514060616493225,
"logps/rejected": -2.152096748352051,
"loss": 1.1687,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.140606880187988,
"rewards/margins": 6.380361080169678,
"rewards/rejected": -21.520967483520508,
"step": 400
},
{
"epoch": 0.8549292011755276,
"eval_logits/chosen": -1.7677618265151978,
"eval_logits/rejected": -1.773621916770935,
"eval_logps/chosen": -1.601662278175354,
"eval_logps/rejected": -2.307917356491089,
"eval_loss": 1.0876977443695068,
"eval_rewards/accuracies": 0.8870967626571655,
"eval_rewards/chosen": -16.016620635986328,
"eval_rewards/margins": 7.062554836273193,
"eval_rewards/rejected": -23.07917594909668,
"eval_runtime": 96.7009,
"eval_samples_per_second": 20.279,
"eval_steps_per_second": 0.641,
"step": 400
},
{
"epoch": 0.8656158161902218,
"grad_norm": 133.15081807524635,
"learning_rate": 5.2811296166831666e-08,
"logits/chosen": -1.5344595909118652,
"logits/rejected": -1.494829773902893,
"logps/chosen": -1.7153148651123047,
"logps/rejected": -2.2892117500305176,
"loss": 1.2171,
"rewards/accuracies": 0.8125,
"rewards/chosen": -17.153148651123047,
"rewards/margins": 5.7389678955078125,
"rewards/rejected": -22.89211654663086,
"step": 405
},
{
"epoch": 0.8763024312049158,
"grad_norm": 158.42453976320837,
"learning_rate": 4.4761226670592066e-08,
"logits/chosen": -1.5358643531799316,
"logits/rejected": -1.4975647926330566,
"logps/chosen": -1.5330779552459717,
"logps/rejected": -2.2891881465911865,
"loss": 1.1582,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -15.330778121948242,
"rewards/margins": 7.561104774475098,
"rewards/rejected": -22.89188575744629,
"step": 410
},
{
"epoch": 0.88698904621961,
"grad_norm": 104.59131788649188,
"learning_rate": 3.734784976300165e-08,
"logits/chosen": -1.5346996784210205,
"logits/rejected": -1.5254827737808228,
"logps/chosen": -1.5777432918548584,
"logps/rejected": -2.2442333698272705,
"loss": 1.0981,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -15.777432441711426,
"rewards/margins": 6.664898872375488,
"rewards/rejected": -22.442333221435547,
"step": 415
},
{
"epoch": 0.897675661234304,
"grad_norm": 95.79743506562838,
"learning_rate": 3.058153372200695e-08,
"logits/chosen": -1.5502920150756836,
"logits/rejected": -1.5145201683044434,
"logps/chosen": -1.432101845741272,
"logps/rejected": -2.1306354999542236,
"loss": 1.072,
"rewards/accuracies": 0.875,
"rewards/chosen": -14.321019172668457,
"rewards/margins": 6.985334873199463,
"rewards/rejected": -21.306354522705078,
"step": 420
},
{
"epoch": 0.9083622762489981,
"grad_norm": 91.78736413360674,
"learning_rate": 2.4471741852423233e-08,
"logits/chosen": -1.5086685419082642,
"logits/rejected": -1.5284610986709595,
"logps/chosen": -1.5403655767440796,
"logps/rejected": -2.4292709827423096,
"loss": 0.9504,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.403654098510742,
"rewards/margins": 8.889054298400879,
"rewards/rejected": -24.292707443237305,
"step": 425
},
{
"epoch": 0.9190488912636923,
"grad_norm": 84.03123746989313,
"learning_rate": 1.9027019250647036e-08,
"logits/chosen": -1.6175905466079712,
"logits/rejected": -1.6218225955963135,
"logps/chosen": -1.5030295848846436,
"logps/rejected": -2.3066420555114746,
"loss": 1.0077,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.030294418334961,
"rewards/margins": 8.036127090454102,
"rewards/rejected": -23.066423416137695,
"step": 430
},
{
"epoch": 0.9297355062783863,
"grad_norm": 117.932317769854,
"learning_rate": 1.4254980853566246e-08,
"logits/chosen": -1.5319820642471313,
"logits/rejected": -1.5626695156097412,
"logps/chosen": -1.5170115232467651,
"logps/rejected": -2.136791229248047,
"loss": 1.0213,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -15.170114517211914,
"rewards/margins": 6.197798252105713,
"rewards/rejected": -21.36791229248047,
"step": 435
},
{
"epoch": 0.9404221212930804,
"grad_norm": 126.32296519295083,
"learning_rate": 1.016230078838226e-08,
"logits/chosen": -1.5235329866409302,
"logits/rejected": -1.5282210111618042,
"logps/chosen": -1.6138379573822021,
"logps/rejected": -2.333554744720459,
"loss": 1.203,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -16.13838005065918,
"rewards/margins": 7.197165489196777,
"rewards/rejected": -23.335546493530273,
"step": 440
},
{
"epoch": 0.9511087363077745,
"grad_norm": 102.89001505284573,
"learning_rate": 6.754703038239329e-09,
"logits/chosen": -1.6163692474365234,
"logits/rejected": -1.5660312175750732,
"logps/chosen": -1.5261099338531494,
"logps/rejected": -2.2487072944641113,
"loss": 1.1582,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.261096000671387,
"rewards/margins": 7.225976467132568,
"rewards/rejected": -22.487075805664062,
"step": 445
},
{
"epoch": 0.9617953513224686,
"grad_norm": 119.74095319029244,
"learning_rate": 4.036953436716895e-09,
"logits/chosen": -1.5374672412872314,
"logits/rejected": -1.4871861934661865,
"logps/chosen": -1.5755971670150757,
"logps/rejected": -2.061048984527588,
"loss": 1.073,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -15.75597095489502,
"rewards/margins": 4.854517459869385,
"rewards/rejected": -20.610488891601562,
"step": 450
},
{
"epoch": 0.9724819663371627,
"grad_norm": 76.57788575531883,
"learning_rate": 2.0128530023804656e-09,
"logits/chosen": -1.591564416885376,
"logits/rejected": -1.5509247779846191,
"logps/chosen": -1.5886285305023193,
"logps/rejected": -2.421255588531494,
"loss": 0.9676,
"rewards/accuracies": 0.9375,
"rewards/chosen": -15.886283874511719,
"rewards/margins": 8.326271057128906,
"rewards/rejected": -24.212556838989258,
"step": 455
},
{
"epoch": 0.9831685813518568,
"grad_norm": 109.6328495391863,
"learning_rate": 6.852326227130833e-10,
"logits/chosen": -1.5486748218536377,
"logits/rejected": -1.549953818321228,
"logps/chosen": -1.5361286401748657,
"logps/rejected": -2.2728710174560547,
"loss": 1.1484,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -15.361285209655762,
"rewards/margins": 7.367424964904785,
"rewards/rejected": -22.728710174560547,
"step": 460
},
{
"epoch": 0.9938551963665508,
"grad_norm": 101.61409866072276,
"learning_rate": 5.594909486328348e-11,
"logits/chosen": -1.5434294939041138,
"logits/rejected": -1.5330185890197754,
"logps/chosen": -1.7000032663345337,
"logps/rejected": -2.3660502433776855,
"loss": 1.0908,
"rewards/accuracies": 0.875,
"rewards/chosen": -17.000032424926758,
"rewards/margins": 6.660471439361572,
"rewards/rejected": -23.66050148010254,
"step": 465
},
{
"epoch": 0.9981298423724285,
"step": 467,
"total_flos": 0.0,
"train_loss": 1.7411203522243133,
"train_runtime": 12577.1262,
"train_samples_per_second": 4.761,
"train_steps_per_second": 0.037
}
],
"logging_steps": 5,
"max_steps": 467,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 120,
"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": 2,
"trial_name": null,
"trial_params": null
}