gemma-7b-borpo-low-quality-v4 / trainer_state.json
c-alfano's picture
Model save
17c4a69 verified
raw
history blame
68.7 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 2.9865871833084947,
"eval_steps": 500,
"global_step": 501,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.029806259314456036,
"grad_norm": 1823.4444580078125,
"learning_rate": 2.5000000000000004e-07,
"log_odds_chosen": -0.22316808998584747,
"log_odds_ratio": -1.008633017539978,
"logits/chosen": 204.3894805908203,
"logits/rejected": 203.0862274169922,
"logps/chosen": -14.825190544128418,
"logps/rejected": -14.602023124694824,
"loss": 15.3951,
"nll_loss": 14.544248580932617,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": -7.412595272064209,
"rewards/margins": -0.1115839034318924,
"rewards/rejected": -7.301011562347412,
"step": 5
},
{
"epoch": 0.05961251862891207,
"grad_norm": 1235.2249755859375,
"learning_rate": 5.000000000000001e-07,
"log_odds_chosen": 0.2599984109401703,
"log_odds_ratio": -0.7687832713127136,
"logits/chosen": 219.47842407226562,
"logits/rejected": 223.53195190429688,
"logps/chosen": -12.257360458374023,
"logps/rejected": -12.516705513000488,
"loss": 13.0935,
"nll_loss": 12.36180305480957,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -6.128680229187012,
"rewards/margins": 0.1296723335981369,
"rewards/rejected": -6.258352756500244,
"step": 10
},
{
"epoch": 0.08941877794336811,
"grad_norm": 800.8740844726562,
"learning_rate": 7.5e-07,
"log_odds_chosen": 0.07262952625751495,
"log_odds_ratio": -0.7626513242721558,
"logits/chosen": 284.1068115234375,
"logits/rejected": 263.359619140625,
"logps/chosen": -8.042943000793457,
"logps/rejected": -8.115506172180176,
"loss": 8.8161,
"nll_loss": 8.040520668029785,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -4.0214715003967285,
"rewards/margins": 0.03628172725439072,
"rewards/rejected": -4.057753086090088,
"step": 15
},
{
"epoch": 0.11922503725782414,
"grad_norm": 245.36756896972656,
"learning_rate": 1.0000000000000002e-06,
"log_odds_chosen": -0.1374012976884842,
"log_odds_ratio": -0.9441366195678711,
"logits/chosen": 284.14410400390625,
"logits/rejected": 279.113037109375,
"logps/chosen": -5.502886772155762,
"logps/rejected": -5.366313934326172,
"loss": 5.9511,
"nll_loss": 5.569659233093262,
"rewards/accuracies": 0.4000000059604645,
"rewards/chosen": -2.751443386077881,
"rewards/margins": -0.06828644871711731,
"rewards/rejected": -2.683156967163086,
"step": 20
},
{
"epoch": 0.14903129657228018,
"grad_norm": 150.3289337158203,
"learning_rate": 1.25e-06,
"log_odds_chosen": -0.0038304091431200504,
"log_odds_ratio": -0.8650141954421997,
"logits/chosen": 296.8591003417969,
"logits/rejected": 306.09613037109375,
"logps/chosen": -3.4051125049591064,
"logps/rejected": -3.383711338043213,
"loss": 3.9954,
"nll_loss": 3.5142741203308105,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -1.7025562524795532,
"rewards/margins": -0.010700652375817299,
"rewards/rejected": -1.6918556690216064,
"step": 25
},
{
"epoch": 0.17883755588673622,
"grad_norm": 95.30947875976562,
"learning_rate": 1.5e-06,
"log_odds_chosen": -0.02694419026374817,
"log_odds_ratio": -0.8674762845039368,
"logits/chosen": 341.46405029296875,
"logits/rejected": 369.68597412109375,
"logps/chosen": -2.748025894165039,
"logps/rejected": -2.6941990852355957,
"loss": 3.0003,
"nll_loss": 2.7433438301086426,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -1.3740129470825195,
"rewards/margins": -0.02691327966749668,
"rewards/rejected": -1.3470995426177979,
"step": 30
},
{
"epoch": 0.20864381520119224,
"grad_norm": 61.000022888183594,
"learning_rate": 1.75e-06,
"log_odds_chosen": 0.20579464733600616,
"log_odds_ratio": -0.7235999703407288,
"logits/chosen": 380.79266357421875,
"logits/rejected": 368.7811279296875,
"logps/chosen": -1.8658641576766968,
"logps/rejected": -2.010704517364502,
"loss": 2.4897,
"nll_loss": 2.063866376876831,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.9329320788383484,
"rewards/margins": 0.07242023944854736,
"rewards/rejected": -1.005352258682251,
"step": 35
},
{
"epoch": 0.23845007451564829,
"grad_norm": 52.54775619506836,
"learning_rate": 2.0000000000000003e-06,
"log_odds_chosen": 0.14920620620250702,
"log_odds_ratio": -0.7215272188186646,
"logits/chosen": 373.4493713378906,
"logits/rejected": 372.3648376464844,
"logps/chosen": -1.7074140310287476,
"logps/rejected": -1.8317168951034546,
"loss": 2.3294,
"nll_loss": 2.0935075283050537,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.8537070155143738,
"rewards/margins": 0.06215135008096695,
"rewards/rejected": -0.9158584475517273,
"step": 40
},
{
"epoch": 0.26825633383010433,
"grad_norm": 58.8853645324707,
"learning_rate": 2.25e-06,
"log_odds_chosen": 0.20898687839508057,
"log_odds_ratio": -0.666365385055542,
"logits/chosen": 390.11090087890625,
"logits/rejected": 399.29052734375,
"logps/chosen": -1.6940256357192993,
"logps/rejected": -1.859763503074646,
"loss": 2.2384,
"nll_loss": 1.81049382686615,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.8470128178596497,
"rewards/margins": 0.08286889642477036,
"rewards/rejected": -0.929881751537323,
"step": 45
},
{
"epoch": 0.29806259314456035,
"grad_norm": 53.57241439819336,
"learning_rate": 2.5e-06,
"log_odds_chosen": 0.4837431013584137,
"log_odds_ratio": -0.5640600919723511,
"logits/chosen": 397.261474609375,
"logits/rejected": 418.50787353515625,
"logps/chosen": -1.548592209815979,
"logps/rejected": -1.9497201442718506,
"loss": 2.2689,
"nll_loss": 1.9370243549346924,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7742961049079895,
"rewards/margins": 0.20056398212909698,
"rewards/rejected": -0.9748600721359253,
"step": 50
},
{
"epoch": 0.32786885245901637,
"grad_norm": 54.863670349121094,
"learning_rate": 2.7500000000000004e-06,
"log_odds_chosen": 0.10273619741201401,
"log_odds_ratio": -0.7500853538513184,
"logits/chosen": 393.6875,
"logits/rejected": 385.71246337890625,
"logps/chosen": -1.6378755569458008,
"logps/rejected": -1.7148265838623047,
"loss": 2.343,
"nll_loss": 2.059002637863159,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.8189377784729004,
"rewards/margins": 0.038475506007671356,
"rewards/rejected": -0.8574132919311523,
"step": 55
},
{
"epoch": 0.35767511177347244,
"grad_norm": 52.10759735107422,
"learning_rate": 3e-06,
"log_odds_chosen": 0.20460805296897888,
"log_odds_ratio": -0.6378257870674133,
"logits/chosen": 393.1427917480469,
"logits/rejected": 384.66424560546875,
"logps/chosen": -1.5411105155944824,
"logps/rejected": -1.693114995956421,
"loss": 2.0643,
"nll_loss": 1.8742763996124268,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.7705552577972412,
"rewards/margins": 0.07600229978561401,
"rewards/rejected": -0.8465574979782104,
"step": 60
},
{
"epoch": 0.38748137108792846,
"grad_norm": 36.03993225097656,
"learning_rate": 3.2500000000000002e-06,
"log_odds_chosen": 0.31621289253234863,
"log_odds_ratio": -0.6504405736923218,
"logits/chosen": 391.6888427734375,
"logits/rejected": 392.4349365234375,
"logps/chosen": -1.3424034118652344,
"logps/rejected": -1.5777828693389893,
"loss": 1.9359,
"nll_loss": 1.5658212900161743,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6712017059326172,
"rewards/margins": 0.11768964678049088,
"rewards/rejected": -0.7888914346694946,
"step": 65
},
{
"epoch": 0.4172876304023845,
"grad_norm": 41.6033821105957,
"learning_rate": 3.5e-06,
"log_odds_chosen": 0.3797632157802582,
"log_odds_ratio": -0.6185951828956604,
"logits/chosen": 376.75579833984375,
"logits/rejected": 385.7332763671875,
"logps/chosen": -1.507440209388733,
"logps/rejected": -1.8190501928329468,
"loss": 2.0024,
"nll_loss": 1.7485978603363037,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7537201046943665,
"rewards/margins": 0.1558050811290741,
"rewards/rejected": -0.9095250964164734,
"step": 70
},
{
"epoch": 0.44709388971684055,
"grad_norm": 49.81876754760742,
"learning_rate": 3.7500000000000005e-06,
"log_odds_chosen": 0.5595852732658386,
"log_odds_ratio": -0.531262218952179,
"logits/chosen": 394.0195617675781,
"logits/rejected": 380.848388671875,
"logps/chosen": -1.278014898300171,
"logps/rejected": -1.7234604358673096,
"loss": 1.8859,
"nll_loss": 1.5626357793807983,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.6390074491500854,
"rewards/margins": 0.22272281348705292,
"rewards/rejected": -0.8617302179336548,
"step": 75
},
{
"epoch": 0.47690014903129657,
"grad_norm": 49.527286529541016,
"learning_rate": 4.000000000000001e-06,
"log_odds_chosen": 0.4827095866203308,
"log_odds_ratio": -0.5827818512916565,
"logits/chosen": 383.16455078125,
"logits/rejected": 405.3050231933594,
"logps/chosen": -1.4663052558898926,
"logps/rejected": -1.8465086221694946,
"loss": 1.9594,
"nll_loss": 1.6925424337387085,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7331526279449463,
"rewards/margins": 0.19010166823863983,
"rewards/rejected": -0.9232543110847473,
"step": 80
},
{
"epoch": 0.5067064083457526,
"grad_norm": 35.50678634643555,
"learning_rate": 4.25e-06,
"log_odds_chosen": 0.32843467593193054,
"log_odds_ratio": -0.7204899787902832,
"logits/chosen": 409.7994689941406,
"logits/rejected": 394.08575439453125,
"logps/chosen": -1.5447129011154175,
"logps/rejected": -1.7741330862045288,
"loss": 1.9717,
"nll_loss": 1.7773849964141846,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7723564505577087,
"rewards/margins": 0.11471016705036163,
"rewards/rejected": -0.8870665431022644,
"step": 85
},
{
"epoch": 0.5365126676602087,
"grad_norm": 32.97898864746094,
"learning_rate": 4.5e-06,
"log_odds_chosen": 1.082437515258789,
"log_odds_ratio": -0.48028555512428284,
"logits/chosen": 400.08160400390625,
"logits/rejected": 415.0059509277344,
"logps/chosen": -1.4097836017608643,
"logps/rejected": -2.338646411895752,
"loss": 1.7591,
"nll_loss": 1.5575084686279297,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.7048918008804321,
"rewards/margins": 0.4644315242767334,
"rewards/rejected": -1.169323205947876,
"step": 90
},
{
"epoch": 0.5663189269746647,
"grad_norm": 59.56174850463867,
"learning_rate": 4.75e-06,
"log_odds_chosen": 0.43072813749313354,
"log_odds_ratio": -0.5813931226730347,
"logits/chosen": 370.78253173828125,
"logits/rejected": 383.03009033203125,
"logps/chosen": -1.1654596328735352,
"logps/rejected": -1.4328858852386475,
"loss": 1.8455,
"nll_loss": 1.4887888431549072,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5827298164367676,
"rewards/margins": 0.13371312618255615,
"rewards/rejected": -0.7164429426193237,
"step": 95
},
{
"epoch": 0.5961251862891207,
"grad_norm": 98.94036102294922,
"learning_rate": 5e-06,
"log_odds_chosen": 0.6077507138252258,
"log_odds_ratio": -0.5442172884941101,
"logits/chosen": 406.62945556640625,
"logits/rejected": 442.9092712402344,
"logps/chosen": -1.3020541667938232,
"logps/rejected": -1.8070911169052124,
"loss": 1.8887,
"nll_loss": 1.6196590662002563,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6510270833969116,
"rewards/margins": 0.2525184154510498,
"rewards/rejected": -0.9035455584526062,
"step": 100
},
{
"epoch": 0.6259314456035767,
"grad_norm": 64.72420501708984,
"learning_rate": 4.8795003647426654e-06,
"log_odds_chosen": 0.6622194051742554,
"log_odds_ratio": -0.5436784625053406,
"logits/chosen": 396.576416015625,
"logits/rejected": 409.37408447265625,
"logps/chosen": -1.239157795906067,
"logps/rejected": -1.7216756343841553,
"loss": 2.0086,
"nll_loss": 1.743311882019043,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6195788979530334,
"rewards/margins": 0.24125893414020538,
"rewards/rejected": -0.8608378171920776,
"step": 105
},
{
"epoch": 0.6557377049180327,
"grad_norm": 156.0756072998047,
"learning_rate": 4.767312946227961e-06,
"log_odds_chosen": 0.552756667137146,
"log_odds_ratio": -0.5492970943450928,
"logits/chosen": 376.5224304199219,
"logits/rejected": 374.7490234375,
"logps/chosen": -1.258983850479126,
"logps/rejected": -1.6897951364517212,
"loss": 1.8644,
"nll_loss": 1.6051260232925415,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.629491925239563,
"rewards/margins": 0.21540567278862,
"rewards/rejected": -0.8448975682258606,
"step": 110
},
{
"epoch": 0.6855439642324889,
"grad_norm": 27.666635513305664,
"learning_rate": 4.662524041201569e-06,
"log_odds_chosen": 0.7742821574211121,
"log_odds_ratio": -0.4874287545681,
"logits/chosen": 401.1915283203125,
"logits/rejected": 406.65716552734375,
"logps/chosen": -1.1674349308013916,
"logps/rejected": -1.7199904918670654,
"loss": 1.8546,
"nll_loss": 1.5520570278167725,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.5837174654006958,
"rewards/margins": 0.2762778401374817,
"rewards/rejected": -0.8599952459335327,
"step": 115
},
{
"epoch": 0.7153502235469449,
"grad_norm": 39.84739303588867,
"learning_rate": 4.564354645876385e-06,
"log_odds_chosen": 0.7211123108863831,
"log_odds_ratio": -0.5134823322296143,
"logits/chosen": 388.77838134765625,
"logits/rejected": 388.0323181152344,
"logps/chosen": -1.1216659545898438,
"logps/rejected": -1.614723801612854,
"loss": 1.8847,
"nll_loss": 1.5851175785064697,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.5608329772949219,
"rewards/margins": 0.24652889370918274,
"rewards/rejected": -0.807361900806427,
"step": 120
},
{
"epoch": 0.7451564828614009,
"grad_norm": 29.12469482421875,
"learning_rate": 4.47213595499958e-06,
"log_odds_chosen": 0.3196253776550293,
"log_odds_ratio": -0.6582637429237366,
"logits/chosen": 386.79583740234375,
"logits/rejected": 398.7740783691406,
"logps/chosen": -1.351342797279358,
"logps/rejected": -1.5734379291534424,
"loss": 1.8179,
"nll_loss": 1.5471035242080688,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.675671398639679,
"rewards/margins": 0.11104752123355865,
"rewards/rejected": -0.7867189645767212,
"step": 125
},
{
"epoch": 0.7749627421758569,
"grad_norm": 38.485130310058594,
"learning_rate": 4.385290096535147e-06,
"log_odds_chosen": 0.6995328068733215,
"log_odds_ratio": -0.5530967712402344,
"logits/chosen": 409.95849609375,
"logits/rejected": 397.23114013671875,
"logps/chosen": -1.1655128002166748,
"logps/rejected": -1.6789804697036743,
"loss": 1.8506,
"nll_loss": 1.5426290035247803,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5827564001083374,
"rewards/margins": 0.2567337453365326,
"rewards/rejected": -0.8394902348518372,
"step": 130
},
{
"epoch": 0.8047690014903129,
"grad_norm": 52.03726577758789,
"learning_rate": 4.303314829119352e-06,
"log_odds_chosen": 0.42176657915115356,
"log_odds_ratio": -0.6203495264053345,
"logits/chosen": 415.891357421875,
"logits/rejected": 418.27984619140625,
"logps/chosen": -1.4386600255966187,
"logps/rejected": -1.7602298259735107,
"loss": 1.873,
"nll_loss": 1.672498106956482,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.7193300127983093,
"rewards/margins": 0.16078472137451172,
"rewards/rejected": -0.8801149129867554,
"step": 135
},
{
"epoch": 0.834575260804769,
"grad_norm": 37.35191345214844,
"learning_rate": 4.2257712736425835e-06,
"log_odds_chosen": 0.30007433891296387,
"log_odds_ratio": -0.659325897693634,
"logits/chosen": 397.5603942871094,
"logits/rejected": 403.28192138671875,
"logps/chosen": -1.314620852470398,
"logps/rejected": -1.552886962890625,
"loss": 1.8804,
"nll_loss": 1.7158609628677368,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": -0.657310426235199,
"rewards/margins": 0.11913303285837173,
"rewards/rejected": -0.7764434814453125,
"step": 140
},
{
"epoch": 0.8643815201192251,
"grad_norm": 40.69832992553711,
"learning_rate": 4.1522739926869985e-06,
"log_odds_chosen": 0.15929968655109406,
"log_odds_ratio": -0.6508444547653198,
"logits/chosen": 392.15386962890625,
"logits/rejected": 395.3309631347656,
"logps/chosen": -1.2985050678253174,
"logps/rejected": -1.4000908136367798,
"loss": 1.8468,
"nll_loss": 1.581002950668335,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6492525339126587,
"rewards/margins": 0.0507928729057312,
"rewards/rejected": -0.7000454068183899,
"step": 145
},
{
"epoch": 0.8941877794336811,
"grad_norm": 33.805145263671875,
"learning_rate": 4.082482904638631e-06,
"log_odds_chosen": 0.7655292749404907,
"log_odds_ratio": -0.514534592628479,
"logits/chosen": 398.47711181640625,
"logits/rejected": 415.9359436035156,
"logps/chosen": -1.1850507259368896,
"logps/rejected": -1.7938915491104126,
"loss": 1.7734,
"nll_loss": 1.4344916343688965,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.5925253629684448,
"rewards/margins": 0.3044203519821167,
"rewards/rejected": -0.8969457745552063,
"step": 150
},
{
"epoch": 0.9239940387481371,
"grad_norm": 21.895387649536133,
"learning_rate": 4.016096644512495e-06,
"log_odds_chosen": 0.26020118594169617,
"log_odds_ratio": -0.6326924562454224,
"logits/chosen": 378.7313537597656,
"logits/rejected": 393.6383361816406,
"logps/chosen": -1.2521542310714722,
"logps/rejected": -1.4581738710403442,
"loss": 1.7575,
"nll_loss": 1.3882125616073608,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.6260771155357361,
"rewards/margins": 0.10300980508327484,
"rewards/rejected": -0.7290869355201721,
"step": 155
},
{
"epoch": 0.9538002980625931,
"grad_norm": 65.63549041748047,
"learning_rate": 3.952847075210474e-06,
"log_odds_chosen": 0.28196483850479126,
"log_odds_ratio": -0.6715171337127686,
"logits/chosen": 386.80755615234375,
"logits/rejected": 431.31146240234375,
"logps/chosen": -1.1752804517745972,
"logps/rejected": -1.395171880722046,
"loss": 1.7333,
"nll_loss": 1.422628402709961,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.5876402258872986,
"rewards/margins": 0.1099456176161766,
"rewards/rejected": -0.697585940361023,
"step": 160
},
{
"epoch": 0.9836065573770492,
"grad_norm": 31.469694137573242,
"learning_rate": 3.892494720807615e-06,
"log_odds_chosen": 0.20551732182502747,
"log_odds_ratio": -0.6562041640281677,
"logits/chosen": 399.196533203125,
"logits/rejected": 412.1044006347656,
"logps/chosen": -1.233426809310913,
"logps/rejected": -1.369274377822876,
"loss": 1.8238,
"nll_loss": 1.516705870628357,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6167134046554565,
"rewards/margins": 0.06792382895946503,
"rewards/rejected": -0.684637188911438,
"step": 165
},
{
"epoch": 0.9955290611028316,
"eval_log_odds_chosen": 0.2615682780742645,
"eval_log_odds_ratio": -0.6631397604942322,
"eval_logits/chosen": 322.6454162597656,
"eval_logits/rejected": 293.2744445800781,
"eval_logps/chosen": -1.0755056142807007,
"eval_logps/rejected": -1.263703465461731,
"eval_loss": 1.8175820112228394,
"eval_nll_loss": 1.4782638549804688,
"eval_rewards/accuracies": 0.5467625856399536,
"eval_rewards/chosen": -0.5377528071403503,
"eval_rewards/margins": 0.09409892559051514,
"eval_rewards/rejected": -0.6318517327308655,
"eval_runtime": 351.5652,
"eval_samples_per_second": 1.573,
"eval_steps_per_second": 0.395,
"step": 167
},
{
"epoch": 1.0134128166915053,
"grad_norm": 19.610740661621094,
"learning_rate": 3.834824944236852e-06,
"log_odds_chosen": 0.7168810963630676,
"log_odds_ratio": -0.48998212814331055,
"logits/chosen": 379.9788513183594,
"logits/rejected": 404.4870910644531,
"logps/chosen": -1.0245921611785889,
"logps/rejected": -1.525979995727539,
"loss": 1.6186,
"nll_loss": 1.2430717945098877,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.5122960805892944,
"rewards/margins": 0.2506938874721527,
"rewards/rejected": -0.7629899978637695,
"step": 170
},
{
"epoch": 1.0432190760059612,
"grad_norm": 27.745262145996094,
"learning_rate": 3.7796447300922724e-06,
"log_odds_chosen": 1.422102689743042,
"log_odds_ratio": -0.3321012854576111,
"logits/chosen": 357.04718017578125,
"logits/rejected": 398.559814453125,
"logps/chosen": -0.7521129846572876,
"logps/rejected": -1.7101812362670898,
"loss": 1.286,
"nll_loss": 1.213651418685913,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.3760564923286438,
"rewards/margins": 0.47903409600257874,
"rewards/rejected": -0.8550906181335449,
"step": 175
},
{
"epoch": 1.0730253353204173,
"grad_norm": 20.82083511352539,
"learning_rate": 3.72677996249965e-06,
"log_odds_chosen": 1.3344449996948242,
"log_odds_ratio": -0.3116641640663147,
"logits/chosen": 356.35601806640625,
"logits/rejected": 331.2690124511719,
"logps/chosen": -0.8246381878852844,
"logps/rejected": -1.7447946071624756,
"loss": 1.3455,
"nll_loss": 1.2647905349731445,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.4123190939426422,
"rewards/margins": 0.46007823944091797,
"rewards/rejected": -0.8723973035812378,
"step": 180
},
{
"epoch": 1.1028315946348732,
"grad_norm": 33.03415298461914,
"learning_rate": 3.6760731104690393e-06,
"log_odds_chosen": 1.7977418899536133,
"log_odds_ratio": -0.29093313217163086,
"logits/chosen": 388.28021240234375,
"logits/rejected": 380.4255065917969,
"logps/chosen": -0.8684544563293457,
"logps/rejected": -2.099797487258911,
"loss": 1.2411,
"nll_loss": 1.0944446325302124,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.43422722816467285,
"rewards/margins": 0.6156715750694275,
"rewards/rejected": -1.0498987436294556,
"step": 185
},
{
"epoch": 1.1326378539493294,
"grad_norm": 48.01770782470703,
"learning_rate": 3.6273812505500587e-06,
"log_odds_chosen": 1.2517069578170776,
"log_odds_ratio": -0.3731076717376709,
"logits/chosen": 356.5531005859375,
"logits/rejected": 402.0587463378906,
"logps/chosen": -0.8801576495170593,
"logps/rejected": -1.7424513101577759,
"loss": 1.326,
"nll_loss": 1.140998125076294,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.44007882475852966,
"rewards/margins": 0.4311468005180359,
"rewards/rejected": -0.8712256550788879,
"step": 190
},
{
"epoch": 1.1624441132637853,
"grad_norm": 25.167158126831055,
"learning_rate": 3.5805743701971648e-06,
"log_odds_chosen": 1.9819135665893555,
"log_odds_ratio": -0.29898396134376526,
"logits/chosen": 382.34100341796875,
"logits/rejected": 395.79425048828125,
"logps/chosen": -0.9413985013961792,
"logps/rejected": -2.448883533477783,
"loss": 1.3195,
"nll_loss": 1.2245818376541138,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.4706992506980896,
"rewards/margins": 0.7537423968315125,
"rewards/rejected": -1.2244417667388916,
"step": 195
},
{
"epoch": 1.1922503725782414,
"grad_norm": 29.544389724731445,
"learning_rate": 3.5355339059327378e-06,
"log_odds_chosen": 2.2445785999298096,
"log_odds_ratio": -0.27407315373420715,
"logits/chosen": 399.3606262207031,
"logits/rejected": 381.97125244140625,
"logps/chosen": -0.7390884757041931,
"logps/rejected": -2.445159435272217,
"loss": 1.3027,
"nll_loss": 1.0836920738220215,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.36954423785209656,
"rewards/margins": 0.853035569190979,
"rewards/rejected": -1.2225797176361084,
"step": 200
},
{
"epoch": 1.2220566318926975,
"grad_norm": 33.41192626953125,
"learning_rate": 3.4921514788478916e-06,
"log_odds_chosen": 2.2366576194763184,
"log_odds_ratio": -0.2573709785938263,
"logits/chosen": 359.3567810058594,
"logits/rejected": 353.2535400390625,
"logps/chosen": -0.7408849596977234,
"logps/rejected": -2.3824286460876465,
"loss": 1.2558,
"nll_loss": 1.1112185716629028,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3704424798488617,
"rewards/margins": 0.8207718729972839,
"rewards/rejected": -1.1912143230438232,
"step": 205
},
{
"epoch": 1.2518628912071534,
"grad_norm": 18.833148956298828,
"learning_rate": 3.450327796711771e-06,
"log_odds_chosen": 2.773773431777954,
"log_odds_ratio": -0.24581894278526306,
"logits/chosen": 363.11090087890625,
"logits/rejected": 392.077880859375,
"logps/chosen": -0.7906020879745483,
"logps/rejected": -2.9855308532714844,
"loss": 1.2167,
"nll_loss": 1.0773674249649048,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.39530104398727417,
"rewards/margins": 1.0974645614624023,
"rewards/rejected": -1.4927654266357422,
"step": 210
},
{
"epoch": 1.2816691505216096,
"grad_norm": 32.97148895263672,
"learning_rate": 3.409971697352368e-06,
"log_odds_chosen": 2.409407138824463,
"log_odds_ratio": -0.2513101100921631,
"logits/chosen": 386.68804931640625,
"logits/rejected": 369.2494812011719,
"logps/chosen": -0.7668555378913879,
"logps/rejected": -2.570010185241699,
"loss": 1.232,
"nll_loss": 1.1109261512756348,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.38342776894569397,
"rewards/margins": 0.901577353477478,
"rewards/rejected": -1.2850050926208496,
"step": 215
},
{
"epoch": 1.3114754098360657,
"grad_norm": 27.4957218170166,
"learning_rate": 3.3709993123162106e-06,
"log_odds_chosen": 1.301114797592163,
"log_odds_ratio": -0.34698307514190674,
"logits/chosen": 377.33697509765625,
"logits/rejected": 371.03192138671875,
"logps/chosen": -0.8284207582473755,
"logps/rejected": -1.6063474416732788,
"loss": 1.2242,
"nll_loss": 1.0992552042007446,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.41421037912368774,
"rewards/margins": 0.3889634311199188,
"rewards/rejected": -0.8031737208366394,
"step": 220
},
{
"epoch": 1.3412816691505216,
"grad_norm": 27.05774688720703,
"learning_rate": 3.3333333333333333e-06,
"log_odds_chosen": 1.0342134237289429,
"log_odds_ratio": -0.3879423439502716,
"logits/chosen": 373.6796875,
"logits/rejected": 366.01123046875,
"logps/chosen": -0.9923228025436401,
"logps/rejected": -1.7060105800628662,
"loss": 1.2664,
"nll_loss": 1.197096586227417,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.49616140127182007,
"rewards/margins": 0.3568439185619354,
"rewards/rejected": -0.8530052900314331,
"step": 225
},
{
"epoch": 1.3710879284649775,
"grad_norm": 32.180084228515625,
"learning_rate": 3.296902366978936e-06,
"log_odds_chosen": 1.9371551275253296,
"log_odds_ratio": -0.265031635761261,
"logits/chosen": 354.0810546875,
"logits/rejected": 372.2239990234375,
"logps/chosen": -0.8565861582756042,
"logps/rejected": -2.277017116546631,
"loss": 1.2422,
"nll_loss": 0.9957939982414246,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.4282930791378021,
"rewards/margins": 0.7102155685424805,
"rewards/rejected": -1.1385085582733154,
"step": 230
},
{
"epoch": 1.4008941877794336,
"grad_norm": 20.605178833007812,
"learning_rate": 3.2616403652672114e-06,
"log_odds_chosen": 1.8511009216308594,
"log_odds_ratio": -0.2493407279253006,
"logits/chosen": 376.54779052734375,
"logits/rejected": 390.1648254394531,
"logps/chosen": -0.7664806246757507,
"logps/rejected": -2.081376075744629,
"loss": 1.1997,
"nll_loss": 1.0387040376663208,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.38324031233787537,
"rewards/margins": 0.6574475765228271,
"rewards/rejected": -1.0406880378723145,
"step": 235
},
{
"epoch": 1.4307004470938898,
"grad_norm": 24.71708106994629,
"learning_rate": 3.2274861218395142e-06,
"log_odds_chosen": 1.3407671451568604,
"log_odds_ratio": -0.33692649006843567,
"logits/chosen": 401.45428466796875,
"logits/rejected": 407.09161376953125,
"logps/chosen": -0.9054458737373352,
"logps/rejected": -1.8345670700073242,
"loss": 1.3273,
"nll_loss": 1.143675446510315,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.4527229368686676,
"rewards/margins": 0.46456044912338257,
"rewards/rejected": -0.9172835350036621,
"step": 240
},
{
"epoch": 1.4605067064083457,
"grad_norm": 37.8891487121582,
"learning_rate": 3.1943828249997e-06,
"log_odds_chosen": 1.8552640676498413,
"log_odds_ratio": -0.272559255361557,
"logits/chosen": 400.0021667480469,
"logits/rejected": 387.4122009277344,
"logps/chosen": -0.7997323274612427,
"logps/rejected": -2.1247220039367676,
"loss": 1.3455,
"nll_loss": 1.303070306777954,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.39986616373062134,
"rewards/margins": 0.6624947786331177,
"rewards/rejected": -1.0623610019683838,
"step": 245
},
{
"epoch": 1.4903129657228018,
"grad_norm": 35.78718948364258,
"learning_rate": 3.1622776601683796e-06,
"log_odds_chosen": 1.5998339653015137,
"log_odds_ratio": -0.3105994760990143,
"logits/chosen": 369.63177490234375,
"logits/rejected": 376.02960205078125,
"logps/chosen": -0.7897557020187378,
"logps/rejected": -1.8593695163726807,
"loss": 1.2365,
"nll_loss": 1.0359935760498047,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.3948778510093689,
"rewards/margins": 0.5348068475723267,
"rewards/rejected": -0.9296847581863403,
"step": 250
},
{
"epoch": 1.520119225037258,
"grad_norm": 25.08989143371582,
"learning_rate": 3.131121455425748e-06,
"log_odds_chosen": 1.8597294092178345,
"log_odds_ratio": -0.21304909884929657,
"logits/chosen": 384.16290283203125,
"logits/rejected": 387.0345153808594,
"logps/chosen": -0.7654696702957153,
"logps/rejected": -2.019379138946533,
"loss": 1.2499,
"nll_loss": 1.030097246170044,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.38273483514785767,
"rewards/margins": 0.6269546747207642,
"rewards/rejected": -1.0096895694732666,
"step": 255
},
{
"epoch": 1.5499254843517138,
"grad_norm": 21.55389976501465,
"learning_rate": 3.1008683647302113e-06,
"log_odds_chosen": 1.4927784204483032,
"log_odds_ratio": -0.29883819818496704,
"logits/chosen": 362.99444580078125,
"logits/rejected": 404.9141845703125,
"logps/chosen": -0.8609813451766968,
"logps/rejected": -1.9159471988677979,
"loss": 1.2228,
"nll_loss": 1.09743070602417,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.4304906725883484,
"rewards/margins": 0.5274828672409058,
"rewards/rejected": -0.9579735994338989,
"step": 260
},
{
"epoch": 1.5797317436661698,
"grad_norm": 19.87468910217285,
"learning_rate": 3.0714755841697565e-06,
"log_odds_chosen": 1.570449709892273,
"log_odds_ratio": -0.3343026041984558,
"logits/chosen": 374.11920166015625,
"logits/rejected": 396.1978454589844,
"logps/chosen": -0.8117960691452026,
"logps/rejected": -1.8302360773086548,
"loss": 1.3153,
"nll_loss": 1.1015130281448364,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4058980345726013,
"rewards/margins": 0.5092200040817261,
"rewards/rejected": -0.9151180386543274,
"step": 265
},
{
"epoch": 1.6095380029806259,
"grad_norm": 20.458053588867188,
"learning_rate": 3.0429030972509227e-06,
"log_odds_chosen": 1.468200922012329,
"log_odds_ratio": -0.3185100853443146,
"logits/chosen": 360.3885192871094,
"logits/rejected": 371.0684509277344,
"logps/chosen": -0.948271632194519,
"logps/rejected": -1.9939063787460327,
"loss": 1.2987,
"nll_loss": 1.2175120115280151,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.4741358160972595,
"rewards/margins": 0.5228174328804016,
"rewards/rejected": -0.9969531893730164,
"step": 270
},
{
"epoch": 1.639344262295082,
"grad_norm": 21.885934829711914,
"learning_rate": 3.0151134457776365e-06,
"log_odds_chosen": 1.2957884073257446,
"log_odds_ratio": -0.34294062852859497,
"logits/chosen": 351.11566162109375,
"logits/rejected": 341.6363830566406,
"logps/chosen": -0.816031277179718,
"logps/rejected": -1.6235980987548828,
"loss": 1.3141,
"nll_loss": 1.0874848365783691,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -0.408015638589859,
"rewards/margins": 0.40378338098526,
"rewards/rejected": -0.8117990493774414,
"step": 275
},
{
"epoch": 1.669150521609538,
"grad_norm": 32.71710205078125,
"learning_rate": 2.988071523335984e-06,
"log_odds_chosen": 1.3881560564041138,
"log_odds_ratio": -0.44494813680648804,
"logits/chosen": 399.6920471191406,
"logits/rejected": 387.6843566894531,
"logps/chosen": -0.9040805697441101,
"logps/rejected": -1.8093194961547852,
"loss": 1.3021,
"nll_loss": 1.1397716999053955,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -0.45204028487205505,
"rewards/margins": 0.4526194930076599,
"rewards/rejected": -0.9046597480773926,
"step": 280
},
{
"epoch": 1.698956780923994,
"grad_norm": 31.200809478759766,
"learning_rate": 2.961744388795462e-06,
"log_odds_chosen": 1.7313816547393799,
"log_odds_ratio": -0.28469711542129517,
"logits/chosen": 359.5746154785156,
"logits/rejected": 362.81494140625,
"logps/chosen": -0.6776639223098755,
"logps/rejected": -1.7751470804214478,
"loss": 1.2089,
"nll_loss": 1.0214917659759521,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -0.33883196115493774,
"rewards/margins": 0.5487415790557861,
"rewards/rejected": -0.8875735402107239,
"step": 285
},
{
"epoch": 1.7287630402384502,
"grad_norm": 17.744766235351562,
"learning_rate": 2.9361010975735177e-06,
"log_odds_chosen": 1.5064774751663208,
"log_odds_ratio": -0.3122694492340088,
"logits/chosen": 376.41033935546875,
"logits/rejected": 415.8057556152344,
"logps/chosen": -0.8741256594657898,
"logps/rejected": -1.9100143909454346,
"loss": 1.259,
"nll_loss": 1.0986948013305664,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.4370628297328949,
"rewards/margins": 0.5179442167282104,
"rewards/rejected": -0.9550071954727173,
"step": 290
},
{
"epoch": 1.758569299552906,
"grad_norm": 22.985361099243164,
"learning_rate": 2.9111125486979104e-06,
"log_odds_chosen": 1.4937174320220947,
"log_odds_ratio": -0.3182430565357208,
"logits/chosen": 353.3531188964844,
"logits/rejected": 398.23382568359375,
"logps/chosen": -0.8985480070114136,
"logps/rejected": -1.8789575099945068,
"loss": 1.2814,
"nll_loss": 1.1132934093475342,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4492740035057068,
"rewards/margins": 0.4902048110961914,
"rewards/rejected": -0.9394787549972534,
"step": 295
},
{
"epoch": 1.788375558867362,
"grad_norm": 25.093547821044922,
"learning_rate": 2.8867513459481293e-06,
"log_odds_chosen": 1.7519241571426392,
"log_odds_ratio": -0.23565292358398438,
"logits/chosen": 396.87640380859375,
"logits/rejected": 374.09942626953125,
"logps/chosen": -0.6939581632614136,
"logps/rejected": -1.8357816934585571,
"loss": 1.201,
"nll_loss": 0.9846609234809875,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.3469790816307068,
"rewards/margins": 0.5709116458892822,
"rewards/rejected": -0.9178908467292786,
"step": 300
},
{
"epoch": 1.8181818181818183,
"grad_norm": 22.51293182373047,
"learning_rate": 2.862991671569341e-06,
"log_odds_chosen": 1.3752100467681885,
"log_odds_ratio": -0.39724329113960266,
"logits/chosen": 392.15093994140625,
"logits/rejected": 400.69677734375,
"logps/chosen": -0.9685919880867004,
"logps/rejected": -2.020549774169922,
"loss": 1.2557,
"nll_loss": 1.2231824398040771,
"rewards/accuracies": 0.875,
"rewards/chosen": -0.4842959940433502,
"rewards/margins": 0.5259788036346436,
"rewards/rejected": -1.010274887084961,
"step": 305
},
{
"epoch": 1.8479880774962743,
"grad_norm": 18.64189910888672,
"learning_rate": 2.839809171235324e-06,
"log_odds_chosen": 1.9540401697158813,
"log_odds_ratio": -0.28014156222343445,
"logits/chosen": 374.20074462890625,
"logits/rejected": 382.771240234375,
"logps/chosen": -0.823870837688446,
"logps/rejected": -2.250458002090454,
"loss": 1.3098,
"nll_loss": 1.1509441137313843,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -0.411935418844223,
"rewards/margins": 0.7132935523986816,
"rewards/rejected": -1.125229001045227,
"step": 310
},
{
"epoch": 1.8777943368107302,
"grad_norm": 20.656198501586914,
"learning_rate": 2.817180849095055e-06,
"log_odds_chosen": 0.9874370694160461,
"log_odds_ratio": -0.43443575501441956,
"logits/chosen": 348.67279052734375,
"logits/rejected": 367.7172546386719,
"logps/chosen": -1.063990831375122,
"logps/rejected": -1.7773240804672241,
"loss": 1.3159,
"nll_loss": 1.2891852855682373,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -0.531995415687561,
"rewards/margins": 0.3566665053367615,
"rewards/rejected": -0.8886620402336121,
"step": 315
},
{
"epoch": 1.9076005961251863,
"grad_norm": 22.823558807373047,
"learning_rate": 2.7950849718747376e-06,
"log_odds_chosen": 1.9457191228866577,
"log_odds_ratio": -0.23255281150341034,
"logits/chosen": 369.242431640625,
"logits/rejected": 389.74346923828125,
"logps/chosen": -0.7030061483383179,
"logps/rejected": -2.0039026737213135,
"loss": 1.1894,
"nll_loss": 1.0204839706420898,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.35150307416915894,
"rewards/margins": 0.650448203086853,
"rewards/rejected": -1.0019513368606567,
"step": 320
},
{
"epoch": 1.9374068554396424,
"grad_norm": 27.562149047851562,
"learning_rate": 2.773500981126146e-06,
"log_odds_chosen": 1.998150110244751,
"log_odds_ratio": -0.24417133629322052,
"logits/chosen": 365.95684814453125,
"logits/rejected": 397.9959411621094,
"logps/chosen": -0.7909399271011353,
"logps/rejected": -2.2285313606262207,
"loss": 1.179,
"nll_loss": 1.0059034824371338,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.3954699635505676,
"rewards/margins": 0.7187958359718323,
"rewards/rejected": -1.1142656803131104,
"step": 325
},
{
"epoch": 1.9672131147540983,
"grad_norm": 27.540767669677734,
"learning_rate": 2.752409412815902e-06,
"log_odds_chosen": 1.38077712059021,
"log_odds_ratio": -0.3091193735599518,
"logits/chosen": 361.24969482421875,
"logits/rejected": 369.6480407714844,
"logps/chosen": -0.8252070546150208,
"logps/rejected": -1.7504304647445679,
"loss": 1.2534,
"nll_loss": 0.9596601724624634,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.4126035273075104,
"rewards/margins": 0.46261176466941833,
"rewards/rejected": -0.8752152323722839,
"step": 330
},
{
"epoch": 1.9970193740685542,
"grad_norm": 35.79298782348633,
"learning_rate": 2.7317918235407652e-06,
"log_odds_chosen": 1.0996582508087158,
"log_odds_ratio": -0.4153285622596741,
"logits/chosen": 389.44366455078125,
"logits/rejected": 379.0640563964844,
"logps/chosen": -1.0918314456939697,
"logps/rejected": -1.8735624551773071,
"loss": 1.3092,
"nll_loss": 1.2989994287490845,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -0.5459157228469849,
"rewards/margins": 0.3908655047416687,
"rewards/rejected": -0.9367812275886536,
"step": 335
},
{
"epoch": 1.9970193740685542,
"eval_log_odds_chosen": 0.32236072421073914,
"eval_log_odds_ratio": -0.6637296676635742,
"eval_logits/chosen": 309.3900146484375,
"eval_logits/rejected": 279.0658874511719,
"eval_logps/chosen": -1.040497064590454,
"eval_logps/rejected": -1.2617474794387817,
"eval_loss": 1.7559822797775269,
"eval_nll_loss": 1.4054137468338013,
"eval_rewards/accuracies": 0.5323740839958191,
"eval_rewards/chosen": -0.520248532295227,
"eval_rewards/margins": 0.11062520742416382,
"eval_rewards/rejected": -0.6308737397193909,
"eval_runtime": 351.6864,
"eval_samples_per_second": 1.572,
"eval_steps_per_second": 0.395,
"step": 335
},
{
"epoch": 2.0268256333830106,
"grad_norm": 22.68450927734375,
"learning_rate": 2.711630722733202e-06,
"log_odds_chosen": 3.2875800132751465,
"log_odds_ratio": -0.10148422420024872,
"logits/chosen": 390.20111083984375,
"logits/rejected": 362.7218017578125,
"logps/chosen": -0.4825616478919983,
"logps/rejected": -2.6374728679656982,
"loss": 0.7684,
"nll_loss": 0.8495180010795593,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24128082394599915,
"rewards/margins": 1.0774555206298828,
"rewards/rejected": -1.3187364339828491,
"step": 340
},
{
"epoch": 2.0566318926974665,
"grad_norm": 12.678057670593262,
"learning_rate": 2.691909510290828e-06,
"log_odds_chosen": 3.9607760906219482,
"log_odds_ratio": -0.06652864068746567,
"logits/chosen": 340.1753234863281,
"logits/rejected": 337.1167907714844,
"logps/chosen": -0.44624167680740356,
"logps/rejected": -3.1436822414398193,
"loss": 0.6642,
"nll_loss": 0.6589840650558472,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.22312083840370178,
"rewards/margins": 1.3487205505371094,
"rewards/rejected": -1.5718411207199097,
"step": 345
},
{
"epoch": 2.0864381520119224,
"grad_norm": 18.899171829223633,
"learning_rate": 2.6726124191242444e-06,
"log_odds_chosen": 3.980642318725586,
"log_odds_ratio": -0.04676336795091629,
"logits/chosen": 340.1651916503906,
"logits/rejected": 373.43426513671875,
"logps/chosen": -0.43081822991371155,
"logps/rejected": -3.3248093128204346,
"loss": 0.6728,
"nll_loss": 0.6396235227584839,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.21540911495685577,
"rewards/margins": 1.446995496749878,
"rewards/rejected": -1.6624046564102173,
"step": 350
},
{
"epoch": 2.1162444113263787,
"grad_norm": 15.860565185546875,
"learning_rate": 2.6537244621713765e-06,
"log_odds_chosen": 3.6839728355407715,
"log_odds_ratio": -0.07649212330579758,
"logits/chosen": 345.269287109375,
"logits/rejected": 359.37823486328125,
"logps/chosen": -0.45684829354286194,
"logps/rejected": -3.046020984649658,
"loss": 0.666,
"nll_loss": 0.7428454160690308,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.22842414677143097,
"rewards/margins": 1.294586420059204,
"rewards/rejected": -1.523010492324829,
"step": 355
},
{
"epoch": 2.1460506706408347,
"grad_norm": 13.459833145141602,
"learning_rate": 2.6352313834736496e-06,
"log_odds_chosen": 4.002976417541504,
"log_odds_ratio": -0.04872741550207138,
"logits/chosen": 340.22161865234375,
"logits/rejected": 384.5272521972656,
"logps/chosen": -0.433218777179718,
"logps/rejected": -3.1645781993865967,
"loss": 0.6602,
"nll_loss": 0.6463421583175659,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.216609388589859,
"rewards/margins": 1.3656798601150513,
"rewards/rejected": -1.5822890996932983,
"step": 360
},
{
"epoch": 2.1758569299552906,
"grad_norm": 14.115008354187012,
"learning_rate": 2.6171196129510684e-06,
"log_odds_chosen": 3.177072763442993,
"log_odds_ratio": -0.0845123901963234,
"logits/chosen": 326.1507568359375,
"logits/rejected": 306.5514831542969,
"logps/chosen": -0.41812819242477417,
"logps/rejected": -2.4115400314331055,
"loss": 0.6679,
"nll_loss": 0.6390963196754456,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.20906409621238708,
"rewards/margins": 0.9967058300971985,
"rewards/rejected": -1.2057700157165527,
"step": 365
},
{
"epoch": 2.2056631892697465,
"grad_norm": 15.614830017089844,
"learning_rate": 2.599376224550182e-06,
"log_odds_chosen": 3.3914806842803955,
"log_odds_ratio": -0.08955641835927963,
"logits/chosen": 287.53955078125,
"logits/rejected": 306.90667724609375,
"logps/chosen": -0.44138556718826294,
"logps/rejected": -2.749290943145752,
"loss": 0.6969,
"nll_loss": 0.7107062935829163,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.22069278359413147,
"rewards/margins": 1.153952717781067,
"rewards/rejected": -1.374645471572876,
"step": 370
},
{
"epoch": 2.235469448584203,
"grad_norm": 15.579936027526855,
"learning_rate": 2.5819888974716113e-06,
"log_odds_chosen": 3.3506603240966797,
"log_odds_ratio": -0.09304364025592804,
"logits/chosen": 339.9989929199219,
"logits/rejected": 357.97528076171875,
"logps/chosen": -0.5137821435928345,
"logps/rejected": -2.7970046997070312,
"loss": 0.6856,
"nll_loss": 0.7090636491775513,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.25689107179641724,
"rewards/margins": 1.1416113376617432,
"rewards/rejected": -1.3985023498535156,
"step": 375
},
{
"epoch": 2.2652757078986587,
"grad_norm": 17.770183563232422,
"learning_rate": 2.564945880212886e-06,
"log_odds_chosen": 3.6279399394989014,
"log_odds_ratio": -0.04611171409487724,
"logits/chosen": 343.6501159667969,
"logits/rejected": 322.5646057128906,
"logps/chosen": -0.35303473472595215,
"logps/rejected": -2.671903610229492,
"loss": 0.6689,
"nll_loss": 0.6217206120491028,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.17651736736297607,
"rewards/margins": 1.1594343185424805,
"rewards/rejected": -1.335951805114746,
"step": 380
},
{
"epoch": 2.2950819672131146,
"grad_norm": 12.41324520111084,
"learning_rate": 2.5482359571881276e-06,
"log_odds_chosen": 3.9699618816375732,
"log_odds_ratio": -0.04786146804690361,
"logits/chosen": 341.5869445800781,
"logits/rejected": 333.04608154296875,
"logps/chosen": -0.3405749499797821,
"logps/rejected": -2.8487496376037598,
"loss": 0.6364,
"nll_loss": 0.5744936466217041,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.17028747498989105,
"rewards/margins": 1.254087209701538,
"rewards/rejected": -1.4243748188018799,
"step": 385
},
{
"epoch": 2.3248882265275705,
"grad_norm": 14.415823936462402,
"learning_rate": 2.5318484177091667e-06,
"log_odds_chosen": 3.405970335006714,
"log_odds_ratio": -0.05260590463876724,
"logits/chosen": 360.901611328125,
"logits/rejected": 378.63616943359375,
"logps/chosen": -0.5018097162246704,
"logps/rejected": -2.8674509525299072,
"loss": 0.7074,
"nll_loss": 0.6968181729316711,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2509048581123352,
"rewards/margins": 1.1828205585479736,
"rewards/rejected": -1.4337254762649536,
"step": 390
},
{
"epoch": 2.354694485842027,
"grad_norm": 11.383499145507812,
"learning_rate": 2.515773027133138e-06,
"log_odds_chosen": 3.9227728843688965,
"log_odds_ratio": -0.05491284653544426,
"logits/chosen": 349.3078308105469,
"logits/rejected": 332.0811462402344,
"logps/chosen": -0.32619425654411316,
"logps/rejected": -2.7826080322265625,
"loss": 0.6264,
"nll_loss": 0.5774807929992676,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.16309712827205658,
"rewards/margins": 1.2282068729400635,
"rewards/rejected": -1.3913040161132812,
"step": 395
},
{
"epoch": 2.384500745156483,
"grad_norm": 15.452503204345703,
"learning_rate": 2.5e-06,
"log_odds_chosen": 3.7418270111083984,
"log_odds_ratio": -0.08289723098278046,
"logits/chosen": 353.244873046875,
"logits/rejected": 371.33941650390625,
"logps/chosen": -0.4247261583805084,
"logps/rejected": -3.0048041343688965,
"loss": 0.6678,
"nll_loss": 0.5866397619247437,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -0.2123630791902542,
"rewards/margins": 1.2900387048721313,
"rewards/rejected": -1.5024020671844482,
"step": 400
},
{
"epoch": 2.4143070044709387,
"grad_norm": 14.744851112365723,
"learning_rate": 2.484519974999767e-06,
"log_odds_chosen": 4.066527366638184,
"log_odds_ratio": -0.07095837593078613,
"logits/chosen": 405.4268493652344,
"logits/rejected": 363.4831237792969,
"logps/chosen": -0.4019082188606262,
"logps/rejected": -3.0571980476379395,
"loss": 0.6775,
"nll_loss": 0.6186262965202332,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.2009541094303131,
"rewards/margins": 1.3276448249816895,
"rewards/rejected": -1.5285990238189697,
"step": 405
},
{
"epoch": 2.444113263785395,
"grad_norm": 15.641300201416016,
"learning_rate": 2.4693239916239746e-06,
"log_odds_chosen": 3.6895904541015625,
"log_odds_ratio": -0.08013327419757843,
"logits/chosen": 352.7126770019531,
"logits/rejected": 362.0486145019531,
"logps/chosen": -0.4512965679168701,
"logps/rejected": -2.8881852626800537,
"loss": 0.6612,
"nll_loss": 0.6566362380981445,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.22564828395843506,
"rewards/margins": 1.2184442281723022,
"rewards/rejected": -1.4440926313400269,
"step": 410
},
{
"epoch": 2.473919523099851,
"grad_norm": 13.112720489501953,
"learning_rate": 2.4544034683690802e-06,
"log_odds_chosen": 4.032966613769531,
"log_odds_ratio": -0.03849425166845322,
"logits/chosen": 352.24176025390625,
"logits/rejected": 376.4973449707031,
"logps/chosen": -0.38949456810951233,
"logps/rejected": -3.1059412956237793,
"loss": 0.6593,
"nll_loss": 0.5457557439804077,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.19474728405475616,
"rewards/margins": 1.35822331905365,
"rewards/rejected": -1.5529706478118896,
"step": 415
},
{
"epoch": 2.503725782414307,
"grad_norm": 13.079460144042969,
"learning_rate": 2.4397501823713327e-06,
"log_odds_chosen": 3.7700283527374268,
"log_odds_ratio": -0.06273230165243149,
"logits/chosen": 351.4734802246094,
"logits/rejected": 317.71209716796875,
"logps/chosen": -0.3789517283439636,
"logps/rejected": -2.9251832962036133,
"loss": 0.6409,
"nll_loss": 0.7009339928627014,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.1894758641719818,
"rewards/margins": 1.273115634918213,
"rewards/rejected": -1.4625916481018066,
"step": 420
},
{
"epoch": 2.533532041728763,
"grad_norm": 20.27842903137207,
"learning_rate": 2.4253562503633297e-06,
"log_odds_chosen": 4.24957799911499,
"log_odds_ratio": -0.032464753836393356,
"logits/chosen": 348.3805847167969,
"logits/rejected": 337.718994140625,
"logps/chosen": -0.3897190988063812,
"logps/rejected": -3.382542848587036,
"loss": 0.6474,
"nll_loss": 0.6459823250770569,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1948595494031906,
"rewards/margins": 1.4964120388031006,
"rewards/rejected": -1.691271424293518,
"step": 425
},
{
"epoch": 2.563338301043219,
"grad_norm": 14.640429496765137,
"learning_rate": 2.411214110852061e-06,
"log_odds_chosen": 4.611870765686035,
"log_odds_ratio": -0.03490515798330307,
"logits/chosen": 347.74298095703125,
"logits/rejected": 351.59912109375,
"logps/chosen": -0.3038939833641052,
"logps/rejected": -3.4484775066375732,
"loss": 0.6441,
"nll_loss": 0.5448944568634033,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1519469916820526,
"rewards/margins": 1.572291612625122,
"rewards/rejected": -1.7242387533187866,
"step": 430
},
{
"epoch": 2.593144560357675,
"grad_norm": 15.337629318237305,
"learning_rate": 2.3973165074269213e-06,
"log_odds_chosen": 3.7105846405029297,
"log_odds_ratio": -0.04743606597185135,
"logits/chosen": 360.7786560058594,
"logits/rejected": 318.19854736328125,
"logps/chosen": -0.36348485946655273,
"logps/rejected": -2.7734932899475098,
"loss": 0.6432,
"nll_loss": 0.5634604692459106,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.18174242973327637,
"rewards/margins": 1.205004096031189,
"rewards/rejected": -1.3867466449737549,
"step": 435
},
{
"epoch": 2.6229508196721314,
"grad_norm": 13.271897315979004,
"learning_rate": 2.3836564731139807e-06,
"log_odds_chosen": 3.826213836669922,
"log_odds_ratio": -0.04420660808682442,
"logits/chosen": 342.50848388671875,
"logits/rejected": 348.86456298828125,
"logps/chosen": -0.30872535705566406,
"logps/rejected": -2.6104750633239746,
"loss": 0.666,
"nll_loss": 0.6212347745895386,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.15436267852783203,
"rewards/margins": 1.1508748531341553,
"rewards/rejected": -1.3052375316619873,
"step": 440
},
{
"epoch": 2.6527570789865873,
"grad_norm": 17.46467399597168,
"learning_rate": 2.3702273156998867e-06,
"log_odds_chosen": 4.261384963989258,
"log_odds_ratio": -0.03544224798679352,
"logits/chosen": 323.58251953125,
"logits/rejected": 354.87847900390625,
"logps/chosen": -0.4285755753517151,
"logps/rejected": -3.4066290855407715,
"loss": 0.6866,
"nll_loss": 0.6935943961143494,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.21428778767585754,
"rewards/margins": 1.4890267848968506,
"rewards/rejected": -1.7033145427703857,
"step": 445
},
{
"epoch": 2.682563338301043,
"grad_norm": 12.280491828918457,
"learning_rate": 2.357022603955159e-06,
"log_odds_chosen": 3.510754108428955,
"log_odds_ratio": -0.05934901908040047,
"logits/chosen": 356.21588134765625,
"logits/rejected": 349.2935485839844,
"logps/chosen": -0.4579055905342102,
"logps/rejected": -2.802351474761963,
"loss": 0.6947,
"nll_loss": 0.6444963216781616,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2289527952671051,
"rewards/margins": 1.1722228527069092,
"rewards/rejected": -1.4011757373809814,
"step": 450
},
{
"epoch": 2.712369597615499,
"grad_norm": 14.349300384521484,
"learning_rate": 2.3440361546924774e-06,
"log_odds_chosen": 3.9128804206848145,
"log_odds_ratio": -0.05345267057418823,
"logits/chosen": 379.7029724121094,
"logits/rejected": 353.4925842285156,
"logps/chosen": -0.4496696889400482,
"logps/rejected": -3.064148426055908,
"loss": 0.7279,
"nll_loss": 0.6569377779960632,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2248348444700241,
"rewards/margins": 1.307239294052124,
"rewards/rejected": -1.532074213027954,
"step": 455
},
{
"epoch": 2.742175856929955,
"grad_norm": 13.55716609954834,
"learning_rate": 2.3312620206007847e-06,
"log_odds_chosen": 4.291790962219238,
"log_odds_ratio": -0.044444940984249115,
"logits/chosen": 374.377197265625,
"logits/rejected": 386.43414306640625,
"logps/chosen": -0.3812227249145508,
"logps/rejected": -3.4395527839660645,
"loss": 0.6847,
"nll_loss": 0.7294474840164185,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.1906113624572754,
"rewards/margins": 1.5291650295257568,
"rewards/rejected": -1.7197763919830322,
"step": 460
},
{
"epoch": 2.7719821162444114,
"grad_norm": 13.988215446472168,
"learning_rate": 2.3186944788008413e-06,
"log_odds_chosen": 4.281071186065674,
"log_odds_ratio": -0.04669572785496712,
"logits/chosen": 359.35101318359375,
"logits/rejected": 352.8810729980469,
"logps/chosen": -0.30659520626068115,
"logps/rejected": -3.1890709400177,
"loss": 0.6956,
"nll_loss": 0.6471156477928162,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.15329760313034058,
"rewards/margins": 1.4412376880645752,
"rewards/rejected": -1.59453547000885,
"step": 465
},
{
"epoch": 2.8017883755588673,
"grad_norm": 16.967275619506836,
"learning_rate": 2.3063280200722128e-06,
"log_odds_chosen": 3.476628065109253,
"log_odds_ratio": -0.07849601656198502,
"logits/chosen": 363.3095703125,
"logits/rejected": 323.81842041015625,
"logps/chosen": -0.43367767333984375,
"logps/rejected": -2.7612318992614746,
"loss": 0.6725,
"nll_loss": 0.6623119711875916,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.21683883666992188,
"rewards/margins": 1.1637769937515259,
"rewards/rejected": -1.3806159496307373,
"step": 470
},
{
"epoch": 2.8315946348733236,
"grad_norm": 19.491451263427734,
"learning_rate": 2.2941573387056174e-06,
"log_odds_chosen": 3.7874722480773926,
"log_odds_ratio": -0.04359927773475647,
"logits/chosen": 332.71844482421875,
"logits/rejected": 350.5658264160156,
"logps/chosen": -0.39931654930114746,
"logps/rejected": -2.8959929943084717,
"loss": 0.6421,
"nll_loss": 0.5615902543067932,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.19965827465057373,
"rewards/margins": 1.248338222503662,
"rewards/rejected": -1.4479964971542358,
"step": 475
},
{
"epoch": 2.8614008941877795,
"grad_norm": 18.62973976135254,
"learning_rate": 2.2821773229381924e-06,
"log_odds_chosen": 4.226737976074219,
"log_odds_ratio": -0.04461986571550369,
"logits/chosen": 344.9051208496094,
"logits/rejected": 380.61822509765625,
"logps/chosen": -0.3830781877040863,
"logps/rejected": -3.3659965991973877,
"loss": 0.6189,
"nll_loss": 0.5661858320236206,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.19153909385204315,
"rewards/margins": 1.4914592504501343,
"rewards/rejected": -1.6829982995986938,
"step": 480
},
{
"epoch": 2.8912071535022354,
"grad_norm": 14.639021873474121,
"learning_rate": 2.270383045932499e-06,
"log_odds_chosen": 3.5930495262145996,
"log_odds_ratio": -0.06206268072128296,
"logits/chosen": 341.22650146484375,
"logits/rejected": 361.0518493652344,
"logps/chosen": -0.48062896728515625,
"logps/rejected": -2.968130111694336,
"loss": 0.6645,
"nll_loss": 0.6335129737854004,
"rewards/accuracies": 0.9750000238418579,
"rewards/chosen": -0.24031448364257812,
"rewards/margins": 1.2437504529953003,
"rewards/rejected": -1.484065055847168,
"step": 485
},
{
"epoch": 2.9210134128166914,
"grad_norm": 13.041991233825684,
"learning_rate": 2.2587697572631284e-06,
"log_odds_chosen": 3.418093204498291,
"log_odds_ratio": -0.06181482598185539,
"logits/chosen": 360.23419189453125,
"logits/rejected": 314.44183349609375,
"logps/chosen": -0.4381326138973236,
"logps/rejected": -2.6530632972717285,
"loss": 0.7257,
"nll_loss": 0.6311744451522827,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.2190663069486618,
"rewards/margins": 1.107465386390686,
"rewards/rejected": -1.3265316486358643,
"step": 490
},
{
"epoch": 2.9508196721311473,
"grad_norm": 13.917710304260254,
"learning_rate": 2.2473328748774737e-06,
"log_odds_chosen": 3.9481112957000732,
"log_odds_ratio": -0.07351940125226974,
"logits/chosen": 354.31781005859375,
"logits/rejected": 374.8128356933594,
"logps/chosen": -0.5134202837944031,
"logps/rejected": -3.2178916931152344,
"loss": 0.6495,
"nll_loss": 0.6444075703620911,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.25671014189720154,
"rewards/margins": 1.3522356748580933,
"rewards/rejected": -1.6089458465576172,
"step": 495
},
{
"epoch": 2.9806259314456036,
"grad_norm": 13.279515266418457,
"learning_rate": 2.23606797749979e-06,
"log_odds_chosen": 3.631924867630005,
"log_odds_ratio": -0.05388236045837402,
"logits/chosen": 363.3906555175781,
"logits/rejected": 350.91546630859375,
"logps/chosen": -0.3526715636253357,
"logps/rejected": -2.695366382598877,
"loss": 0.6827,
"nll_loss": 0.5351300835609436,
"rewards/accuracies": 1.0,
"rewards/chosen": -0.17633578181266785,
"rewards/margins": 1.1713473796844482,
"rewards/rejected": -1.3476831912994385,
"step": 500
},
{
"epoch": 2.9865871833084947,
"eval_log_odds_chosen": 0.42297032475471497,
"eval_log_odds_ratio": -0.6477048993110657,
"eval_logits/chosen": 282.00848388671875,
"eval_logits/rejected": 247.6262664794922,
"eval_logps/chosen": -1.2038370370864868,
"eval_logps/rejected": -1.5014722347259521,
"eval_loss": 1.9079890251159668,
"eval_nll_loss": 1.5545235872268677,
"eval_rewards/accuracies": 0.6258992552757263,
"eval_rewards/chosen": -0.6019185185432434,
"eval_rewards/margins": 0.14881758391857147,
"eval_rewards/rejected": -0.7507361173629761,
"eval_runtime": 351.9872,
"eval_samples_per_second": 1.571,
"eval_steps_per_second": 0.395,
"step": 501
},
{
"epoch": 2.9865871833084947,
"step": 501,
"total_flos": 0.0,
"train_loss": 1.686337627812536,
"train_runtime": 31593.308,
"train_samples_per_second": 0.509,
"train_steps_per_second": 0.016
}
],
"logging_steps": 5,
"max_steps": 501,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 500,
"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
}