Meta-Llama-3-8B-Instruct-6e-7 / trainer_state.json
tengxiao1
TX
43a54e4
raw
history blame contribute delete
No virus
51.6 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9982631930527722,
"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.01068804275217101,
"grad_norm": 7.85619992560556,
"learning_rate": 6.382978723404255e-08,
"logits/chosen": 0.06174152344465256,
"logits/rejected": 0.037658724933862686,
"logps/chosen": -0.2702512741088867,
"logps/rejected": -0.26810693740844727,
"loss": 1.2748,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.2702512741088867,
"rewards/margins": -0.002144329948350787,
"rewards/rejected": -0.26810693740844727,
"step": 5
},
{
"epoch": 0.02137608550434202,
"grad_norm": 5.1085151748010285,
"learning_rate": 1.276595744680851e-07,
"logits/chosen": -0.010784180834889412,
"logits/rejected": -0.012599915266036987,
"logps/chosen": -0.26977622509002686,
"logps/rejected": -0.26753830909729004,
"loss": 1.2725,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.26977622509002686,
"rewards/margins": -0.0022379131987690926,
"rewards/rejected": -0.26753830909729004,
"step": 10
},
{
"epoch": 0.03206412825651302,
"grad_norm": 6.692216762984584,
"learning_rate": 1.9148936170212767e-07,
"logits/chosen": 0.0017719150055199862,
"logits/rejected": 0.004488155245780945,
"logps/chosen": -0.27860361337661743,
"logps/rejected": -0.2837750315666199,
"loss": 1.2793,
"rewards/accuracies": 0.45625001192092896,
"rewards/chosen": -0.27860361337661743,
"rewards/margins": 0.005171425640583038,
"rewards/rejected": -0.2837750315666199,
"step": 15
},
{
"epoch": 0.04275217100868404,
"grad_norm": 8.996125297973581,
"learning_rate": 2.553191489361702e-07,
"logits/chosen": -0.06322917342185974,
"logits/rejected": -0.06904581189155579,
"logps/chosen": -0.2816649079322815,
"logps/rejected": -0.2903934419155121,
"loss": 1.2759,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.2816649079322815,
"rewards/margins": 0.008728553541004658,
"rewards/rejected": -0.2903934419155121,
"step": 20
},
{
"epoch": 0.053440213760855046,
"grad_norm": 4.855846145207527,
"learning_rate": 3.1914893617021275e-07,
"logits/chosen": -0.06888748705387115,
"logits/rejected": -0.04847611114382744,
"logps/chosen": -0.25571519136428833,
"logps/rejected": -0.2750292420387268,
"loss": 1.2568,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.25571519136428833,
"rewards/margins": 0.01931406930088997,
"rewards/rejected": -0.2750292420387268,
"step": 25
},
{
"epoch": 0.06412825651302605,
"grad_norm": 4.835742771788438,
"learning_rate": 3.8297872340425535e-07,
"logits/chosen": -0.014270897023379803,
"logits/rejected": -0.006606946233659983,
"logps/chosen": -0.27899158000946045,
"logps/rejected": -0.2938278019428253,
"loss": 1.2771,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.27899158000946045,
"rewards/margins": 0.014836207032203674,
"rewards/rejected": -0.2938278019428253,
"step": 30
},
{
"epoch": 0.07481629926519706,
"grad_norm": 6.995787010953034,
"learning_rate": 4.4680851063829783e-07,
"logits/chosen": -0.03009852208197117,
"logits/rejected": -0.009835416451096535,
"logps/chosen": -0.2785848379135132,
"logps/rejected": -0.29589539766311646,
"loss": 1.2753,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.2785848379135132,
"rewards/margins": 0.017310529947280884,
"rewards/rejected": -0.29589539766311646,
"step": 35
},
{
"epoch": 0.08550434201736808,
"grad_norm": 5.347511039842285,
"learning_rate": 5.106382978723404e-07,
"logits/chosen": -0.06675975024700165,
"logits/rejected": -0.07226153463125229,
"logps/chosen": -0.26110929250717163,
"logps/rejected": -0.2702568769454956,
"loss": 1.2672,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.26110929250717163,
"rewards/margins": 0.009147587232291698,
"rewards/rejected": -0.2702568769454956,
"step": 40
},
{
"epoch": 0.09619238476953908,
"grad_norm": 5.575092391618783,
"learning_rate": 5.74468085106383e-07,
"logits/chosen": -0.0533546507358551,
"logits/rejected": -0.010820349678397179,
"logps/chosen": -0.2856101095676422,
"logps/rejected": -0.28608259558677673,
"loss": 1.2739,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.2856101095676422,
"rewards/margins": 0.0004724980390165001,
"rewards/rejected": -0.28608259558677673,
"step": 45
},
{
"epoch": 0.10688042752171009,
"grad_norm": 4.346264213431051,
"learning_rate": 5.999244704827519e-07,
"logits/chosen": -0.016516100615262985,
"logits/rejected": -0.003799914848059416,
"logps/chosen": -0.2856972813606262,
"logps/rejected": -0.30010929703712463,
"loss": 1.2684,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.2856972813606262,
"rewards/margins": 0.014411995187401772,
"rewards/rejected": -0.30010929703712463,
"step": 50
},
{
"epoch": 0.11756847027388109,
"grad_norm": 5.956061544604284,
"learning_rate": 5.994630389303205e-07,
"logits/chosen": 0.017358671873807907,
"logits/rejected": -0.0036045790184289217,
"logps/chosen": -0.2718290686607361,
"logps/rejected": -0.2807454764842987,
"loss": 1.2628,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.2718290686607361,
"rewards/margins": 0.00891642551869154,
"rewards/rejected": -0.2807454764842987,
"step": 55
},
{
"epoch": 0.1282565130260521,
"grad_norm": 6.117995966956006,
"learning_rate": 5.985827812395378e-07,
"logits/chosen": -0.038348693400621414,
"logits/rejected": -0.0700547844171524,
"logps/chosen": -0.285757839679718,
"logps/rejected": -0.3150267004966736,
"loss": 1.2628,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.285757839679718,
"rewards/margins": 0.029268871992826462,
"rewards/rejected": -0.3150267004966736,
"step": 60
},
{
"epoch": 0.13894455577822312,
"grad_norm": 5.259018808160778,
"learning_rate": 5.972849285303804e-07,
"logits/chosen": -0.023639511317014694,
"logits/rejected": 0.03467974066734314,
"logps/chosen": -0.29160580039024353,
"logps/rejected": -0.31474000215530396,
"loss": 1.2708,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.29160580039024353,
"rewards/margins": 0.02313421666622162,
"rewards/rejected": -0.31474000215530396,
"step": 65
},
{
"epoch": 0.14963259853039412,
"grad_norm": 5.111432049765508,
"learning_rate": 5.955712959672177e-07,
"logits/chosen": -0.016536986455321312,
"logits/rejected": -0.02064664289355278,
"logps/chosen": -0.29413312673568726,
"logps/rejected": -0.34755489230155945,
"loss": 1.2542,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.29413312673568726,
"rewards/margins": 0.05342177301645279,
"rewards/rejected": -0.34755489230155945,
"step": 70
},
{
"epoch": 0.16032064128256512,
"grad_norm": 4.471806926829,
"learning_rate": 5.934442802201417e-07,
"logits/chosen": 0.06339865177869797,
"logits/rejected": 0.10371309518814087,
"logps/chosen": -0.30393484234809875,
"logps/rejected": -0.33556467294692993,
"loss": 1.2679,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.30393484234809875,
"rewards/margins": 0.031629838049411774,
"rewards/rejected": -0.33556467294692993,
"step": 75
},
{
"epoch": 0.17100868403473615,
"grad_norm": 6.2280285157401165,
"learning_rate": 5.909068561130061e-07,
"logits/chosen": -0.013803797774016857,
"logits/rejected": -0.005705256946384907,
"logps/chosen": -0.29255902767181396,
"logps/rejected": -0.32304221391677856,
"loss": 1.2618,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.29255902767181396,
"rewards/margins": 0.030483180657029152,
"rewards/rejected": -0.32304221391677856,
"step": 80
},
{
"epoch": 0.18169672678690715,
"grad_norm": 4.62670410972183,
"learning_rate": 5.879625724628667e-07,
"logits/chosen": 0.007588694803416729,
"logits/rejected": 0.023027362301945686,
"logps/chosen": -0.29541558027267456,
"logps/rejected": -0.3341410756111145,
"loss": 1.2571,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.29541558027267456,
"rewards/margins": 0.03872550651431084,
"rewards/rejected": -0.3341410756111145,
"step": 85
},
{
"epoch": 0.19238476953907815,
"grad_norm": 6.280947217475369,
"learning_rate": 5.846155471166399e-07,
"logits/chosen": 0.01748380810022354,
"logits/rejected": 0.038063496351242065,
"logps/chosen": -0.31113117933273315,
"logps/rejected": -0.36816665530204773,
"loss": 1.2529,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.31113117933273315,
"rewards/margins": 0.05703546851873398,
"rewards/rejected": -0.36816665530204773,
"step": 90
},
{
"epoch": 0.20307281229124916,
"grad_norm": 6.33795222726401,
"learning_rate": 5.808704611919212e-07,
"logits/chosen": 0.008561496622860432,
"logits/rejected": -0.009577239863574505,
"logps/chosen": -0.3074243664741516,
"logps/rejected": -0.3172617554664612,
"loss": 1.2598,
"rewards/accuracies": 0.48124998807907104,
"rewards/chosen": -0.3074243664741516,
"rewards/margins": 0.0098373768851161,
"rewards/rejected": -0.3172617554664612,
"step": 95
},
{
"epoch": 0.21376085504342018,
"grad_norm": 5.4836519913536,
"learning_rate": 5.767325525300187e-07,
"logits/chosen": 0.012827359139919281,
"logits/rejected": 0.015095492824912071,
"logps/chosen": -0.3060360550880432,
"logps/rejected": -0.3578498363494873,
"loss": 1.257,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.3060360550880432,
"rewards/margins": 0.051813799887895584,
"rewards/rejected": -0.3578498363494873,
"step": 100
},
{
"epoch": 0.22444889779559118,
"grad_norm": 9.106962216409888,
"learning_rate": 5.722076083703594e-07,
"logits/chosen": -0.011002029292285442,
"logits/rejected": -0.01515892706811428,
"logps/chosen": -0.2857062518596649,
"logps/rejected": -0.3439631760120392,
"loss": 1.2453,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.2857062518596649,
"rewards/margins": 0.05825691670179367,
"rewards/rejected": -0.3439631760120392,
"step": 105
},
{
"epoch": 0.23513694054776219,
"grad_norm": 6.085383517197391,
"learning_rate": 5.673019572565103e-07,
"logits/chosen": -0.023035719990730286,
"logits/rejected": -0.03461029753088951,
"logps/chosen": -0.29488229751586914,
"logps/rejected": -0.3499522805213928,
"loss": 1.2381,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.29488229751586914,
"rewards/margins": 0.05506999418139458,
"rewards/rejected": -0.3499522805213928,
"step": 110
},
{
"epoch": 0.2458249832999332,
"grad_norm": 5.078175131364166,
"learning_rate": 5.620224601851389e-07,
"logits/chosen": 0.005257171578705311,
"logits/rejected": 0.0035833939909934998,
"logps/chosen": -0.3048323094844818,
"logps/rejected": -0.3554520010948181,
"loss": 1.2562,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.3048323094844818,
"rewards/margins": 0.050619661808013916,
"rewards/rejected": -0.3554520010948181,
"step": 115
},
{
"epoch": 0.2565130260521042,
"grad_norm": 7.17959663924209,
"learning_rate": 5.563765010102885e-07,
"logits/chosen": -0.06386958062648773,
"logits/rejected": -0.04022233560681343,
"logps/chosen": -0.3290977478027344,
"logps/rejected": -0.36753392219543457,
"loss": 1.2551,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3290977478027344,
"rewards/margins": 0.038436152040958405,
"rewards/rejected": -0.36753392219543457,
"step": 120
},
{
"epoch": 0.26720106880427524,
"grad_norm": 5.129342143579583,
"learning_rate": 5.503719761163907e-07,
"logits/chosen": -0.10350631177425385,
"logits/rejected": -0.08065585047006607,
"logps/chosen": -0.2961386740207672,
"logps/rejected": -0.35747581720352173,
"loss": 1.2338,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.2961386740207672,
"rewards/margins": 0.06133715435862541,
"rewards/rejected": -0.35747581720352173,
"step": 125
},
{
"epoch": 0.27788911155644624,
"grad_norm": 5.1404239274385315,
"learning_rate": 5.440172833744582e-07,
"logits/chosen": -0.057831160724163055,
"logits/rejected": -0.018619367852807045,
"logps/chosen": -0.33000677824020386,
"logps/rejected": -0.372509241104126,
"loss": 1.2645,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.33000677824020386,
"rewards/margins": 0.04250246286392212,
"rewards/rejected": -0.372509241104126,
"step": 130
},
{
"epoch": 0.28857715430861725,
"grad_norm": 8.250677302620405,
"learning_rate": 5.373213103969024e-07,
"logits/chosen": -0.09196853637695312,
"logits/rejected": -0.09740312397480011,
"logps/chosen": -0.3195325434207916,
"logps/rejected": -0.38901036977767944,
"loss": 1.2471,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3195325434207916,
"rewards/margins": 0.06947779655456543,
"rewards/rejected": -0.38901036977767944,
"step": 135
},
{
"epoch": 0.29926519706078825,
"grad_norm": 7.9172215662895065,
"learning_rate": 5.302934221074033e-07,
"logits/chosen": -0.18326355516910553,
"logits/rejected": -0.18124201893806458,
"logps/chosen": -0.35024067759513855,
"logps/rejected": -0.4191233515739441,
"loss": 1.255,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.35024067759513855,
"rewards/margins": 0.06888268887996674,
"rewards/rejected": -0.4191233515739441,
"step": 140
},
{
"epoch": 0.30995323981295925,
"grad_norm": 5.5553212274649635,
"learning_rate": 5.229434476432182e-07,
"logits/chosen": -0.043330904096364975,
"logits/rejected": -0.06870276480913162,
"logps/chosen": -0.3102904260158539,
"logps/rejected": -0.36457473039627075,
"loss": 1.2393,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3102904260158539,
"rewards/margins": 0.05428431183099747,
"rewards/rejected": -0.36457473039627075,
"step": 145
},
{
"epoch": 0.32064128256513025,
"grad_norm": 13.4755681947572,
"learning_rate": 5.152816666082435e-07,
"logits/chosen": -0.0911487489938736,
"logits/rejected": -0.10429541766643524,
"logps/chosen": -0.32351914048194885,
"logps/rejected": -0.4219169616699219,
"loss": 1.2456,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.32351914048194885,
"rewards/margins": 0.09839782118797302,
"rewards/rejected": -0.4219169616699219,
"step": 150
},
{
"epoch": 0.33132932531730125,
"grad_norm": 7.571668933637504,
"learning_rate": 5.073187946960594e-07,
"logits/chosen": -0.08573289960622787,
"logits/rejected": -0.08545240759849548,
"logps/chosen": -0.3177523910999298,
"logps/rejected": -0.3682125210762024,
"loss": 1.2486,
"rewards/accuracies": 0.46875,
"rewards/chosen": -0.3177523910999298,
"rewards/margins": 0.05046012997627258,
"rewards/rejected": -0.3682125210762024,
"step": 155
},
{
"epoch": 0.3420173680694723,
"grad_norm": 6.943663475388935,
"learning_rate": 4.990659687030634e-07,
"logits/chosen": -0.10656943172216415,
"logits/rejected": -0.078646719455719,
"logps/chosen": -0.31297653913497925,
"logps/rejected": -0.37618112564086914,
"loss": 1.2533,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.31297653913497925,
"rewards/margins": 0.06320458650588989,
"rewards/rejected": -0.37618112564086914,
"step": 160
},
{
"epoch": 0.3527054108216433,
"grad_norm": 8.066884155425667,
"learning_rate": 4.905347309526536e-07,
"logits/chosen": -0.06100498512387276,
"logits/rejected": -0.07344032824039459,
"logps/chosen": -0.29639413952827454,
"logps/rejected": -0.36954885721206665,
"loss": 1.2426,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.29639413952827454,
"rewards/margins": 0.0731547400355339,
"rewards/rejected": -0.36954885721206665,
"step": 165
},
{
"epoch": 0.3633934535738143,
"grad_norm": 5.59599590252963,
"learning_rate": 4.817370131522459e-07,
"logits/chosen": -0.055537961423397064,
"logits/rejected": -0.029491102322936058,
"logps/chosen": -0.3175775408744812,
"logps/rejected": -0.39036422967910767,
"loss": 1.2452,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.3175775408744812,
"rewards/margins": 0.07278666645288467,
"rewards/rejected": -0.39036422967910767,
"step": 170
},
{
"epoch": 0.3740814963259853,
"grad_norm": 5.434006324330732,
"learning_rate": 4.7268511970570207e-07,
"logits/chosen": -0.08873588591814041,
"logits/rejected": -0.08352749794721603,
"logps/chosen": -0.3087335228919983,
"logps/rejected": -0.3652537763118744,
"loss": 1.2509,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.3087335228919983,
"rewards/margins": 0.056520313024520874,
"rewards/rejected": -0.3652537763118744,
"step": 175
},
{
"epoch": 0.3847695390781563,
"grad_norm": 8.227857461961667,
"learning_rate": 4.6339171050450815e-07,
"logits/chosen": -0.10110144317150116,
"logits/rejected": -0.09192384779453278,
"logps/chosen": -0.31839340925216675,
"logps/rejected": -0.35923346877098083,
"loss": 1.2547,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.31839340925216675,
"rewards/margins": 0.04084008187055588,
"rewards/rejected": -0.35923346877098083,
"step": 180
},
{
"epoch": 0.3954575818303273,
"grad_norm": 5.5156889818151456,
"learning_rate": 4.5386978322177184e-07,
"logits/chosen": -0.05175872892141342,
"logits/rejected": -0.07638035714626312,
"logps/chosen": -0.3135964274406433,
"logps/rejected": -0.35415133833885193,
"loss": 1.2628,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.3135964274406433,
"rewards/margins": 0.040554873645305634,
"rewards/rejected": -0.35415133833885193,
"step": 185
},
{
"epoch": 0.4061456245824983,
"grad_norm": 8.728192233551633,
"learning_rate": 4.4413265513380134e-07,
"logits/chosen": -0.08628302812576294,
"logits/rejected": -0.0637063980102539,
"logps/chosen": -0.3058716654777527,
"logps/rejected": -0.36986014246940613,
"loss": 1.2462,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.3058716654777527,
"rewards/margins": 0.06398848444223404,
"rewards/rejected": -0.36986014246940613,
"step": 190
},
{
"epoch": 0.4168336673346693,
"grad_norm": 8.179877382911195,
"learning_rate": 4.3419394449468975e-07,
"logits/chosen": -0.06323076784610748,
"logits/rejected": -0.03959321230649948,
"logps/chosen": -0.340233713388443,
"logps/rejected": -0.43143948912620544,
"loss": 1.2398,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.340233713388443,
"rewards/margins": 0.09120576083660126,
"rewards/rejected": -0.43143948912620544,
"step": 195
},
{
"epoch": 0.42752171008684037,
"grad_norm": 5.7289964774382165,
"learning_rate": 4.2406755148995617e-07,
"logits/chosen": -0.03601834177970886,
"logits/rejected": -0.004852446727454662,
"logps/chosen": -0.3104439675807953,
"logps/rejected": -0.38359013199806213,
"loss": 1.2427,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.3104439675807953,
"rewards/margins": 0.07314613461494446,
"rewards/rejected": -0.38359013199806213,
"step": 200
},
{
"epoch": 0.43820975283901137,
"grad_norm": 6.122259545537742,
"learning_rate": 4.1376763879587855e-07,
"logits/chosen": -0.08502549678087234,
"logits/rejected": -0.12322092056274414,
"logps/chosen": -0.3422417938709259,
"logps/rejected": -0.4025643467903137,
"loss": 1.2568,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.3422417938709259,
"rewards/margins": 0.06032254546880722,
"rewards/rejected": -0.4025643467903137,
"step": 205
},
{
"epoch": 0.44889779559118237,
"grad_norm": 6.481401331907283,
"learning_rate": 4.0330861177171046e-07,
"logits/chosen": -0.09341476112604141,
"logits/rejected": -0.0774766132235527,
"logps/chosen": -0.3205105662345886,
"logps/rejected": -0.38998597860336304,
"loss": 1.2469,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.3205105662345886,
"rewards/margins": 0.06947537511587143,
"rewards/rejected": -0.38998597860336304,
"step": 210
},
{
"epoch": 0.45958583834335337,
"grad_norm": 7.252453168693919,
"learning_rate": 3.927050983124842e-07,
"logits/chosen": -0.027269473299384117,
"logits/rejected": -0.07347290217876434,
"logps/chosen": -0.297443687915802,
"logps/rejected": -0.3896985948085785,
"loss": 1.2384,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.297443687915802,
"rewards/margins": 0.09225489944219589,
"rewards/rejected": -0.3896985948085785,
"step": 215
},
{
"epoch": 0.47027388109552437,
"grad_norm": 6.603881159988962,
"learning_rate": 3.8197192839057603e-07,
"logits/chosen": -0.10676763951778412,
"logits/rejected": -0.12236170470714569,
"logps/chosen": -0.30911481380462646,
"logps/rejected": -0.43346795439720154,
"loss": 1.2387,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.30911481380462646,
"rewards/margins": 0.12435313314199448,
"rewards/rejected": -0.43346795439720154,
"step": 220
},
{
"epoch": 0.48096192384769537,
"grad_norm": 8.47531573536886,
"learning_rate": 3.7112411331464923e-07,
"logits/chosen": -0.027242982760071754,
"logits/rejected": -0.03159039095044136,
"logps/chosen": -0.3138233721256256,
"logps/rejected": -0.3913508951663971,
"loss": 1.2275,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3138233721256256,
"rewards/margins": 0.07752753794193268,
"rewards/rejected": -0.3913508951663971,
"step": 225
},
{
"epoch": 0.4916499665998664,
"grad_norm": 6.314922973802041,
"learning_rate": 3.601768247349818e-07,
"logits/chosen": -0.03554535657167435,
"logits/rejected": -0.08682241290807724,
"logps/chosen": -0.31549689173698425,
"logps/rejected": -0.38394489884376526,
"loss": 1.2408,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.31549689173698425,
"rewards/margins": 0.0684480220079422,
"rewards/rejected": -0.38394489884376526,
"step": 230
},
{
"epoch": 0.5023380093520374,
"grad_norm": 8.005653894417838,
"learning_rate": 3.491453734245413e-07,
"logits/chosen": -0.06854398548603058,
"logits/rejected": -0.019309626892209053,
"logps/chosen": -0.33874207735061646,
"logps/rejected": -0.4499788284301758,
"loss": 1.2411,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.33874207735061646,
"rewards/margins": 0.11123677343130112,
"rewards/rejected": -0.4499788284301758,
"step": 235
},
{
"epoch": 0.5130260521042084,
"grad_norm": 6.021580320773287,
"learning_rate": 3.3804518786548455e-07,
"logits/chosen": -0.09851398319005966,
"logits/rejected": -0.08007166534662247,
"logps/chosen": -0.3105488419532776,
"logps/rejected": -0.4278073310852051,
"loss": 1.2457,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.3105488419532776,
"rewards/margins": 0.11725848913192749,
"rewards/rejected": -0.4278073310852051,
"step": 240
},
{
"epoch": 0.5237140948563794,
"grad_norm": 5.487038572936299,
"learning_rate": 3.2689179267103006e-07,
"logits/chosen": -0.1326679289340973,
"logits/rejected": -0.12296830117702484,
"logps/chosen": -0.3216492533683777,
"logps/rejected": -0.3619733452796936,
"loss": 1.2301,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.3216492533683777,
"rewards/margins": 0.04032406955957413,
"rewards/rejected": -0.3619733452796936,
"step": 245
},
{
"epoch": 0.5344021376085505,
"grad_norm": 8.070636702857746,
"learning_rate": 3.1570078687288317e-07,
"logits/chosen": -0.07501702755689621,
"logits/rejected": -0.07981151342391968,
"logps/chosen": -0.3418981432914734,
"logps/rejected": -0.4605533480644226,
"loss": 1.2276,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.3418981432914734,
"rewards/margins": 0.11865530908107758,
"rewards/rejected": -0.4605533480644226,
"step": 250
},
{
"epoch": 0.5450901803607214,
"grad_norm": 8.167728040532104,
"learning_rate": 3.0448782210457906e-07,
"logits/chosen": -0.08226356655359268,
"logits/rejected": -0.07389514893293381,
"logps/chosen": -0.36772793531417847,
"logps/rejected": -0.4519001543521881,
"loss": 1.2468,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.36772793531417847,
"rewards/margins": 0.08417223393917084,
"rewards/rejected": -0.4519001543521881,
"step": 255
},
{
"epoch": 0.5557782231128925,
"grad_norm": 8.9092828172424,
"learning_rate": 2.932685807112585e-07,
"logits/chosen": -0.1377793848514557,
"logits/rejected": -0.13506443798542023,
"logps/chosen": -0.3222098648548126,
"logps/rejected": -0.41552734375,
"loss": 1.238,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3222098648548126,
"rewards/margins": 0.09331748634576797,
"rewards/rejected": -0.41552734375,
"step": 260
},
{
"epoch": 0.5664662658650634,
"grad_norm": 13.838548351939439,
"learning_rate": 2.8205875381648974e-07,
"logits/chosen": -0.11301400512456894,
"logits/rejected": -0.11296539008617401,
"logps/chosen": -0.3139273226261139,
"logps/rejected": -0.40139150619506836,
"loss": 1.2453,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.3139273226261139,
"rewards/margins": 0.08746419101953506,
"rewards/rejected": -0.40139150619506836,
"step": 265
},
{
"epoch": 0.5771543086172345,
"grad_norm": 11.456971077488367,
"learning_rate": 2.708740193768135e-07,
"logits/chosen": -0.08372676372528076,
"logits/rejected": -0.07923340797424316,
"logps/chosen": -0.33234184980392456,
"logps/rejected": -0.4971323013305664,
"loss": 1.2275,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.33234184980392456,
"rewards/margins": 0.16479046642780304,
"rewards/rejected": -0.4971323013305664,
"step": 270
},
{
"epoch": 0.5878423513694054,
"grad_norm": 5.570703398185759,
"learning_rate": 2.597300202547034e-07,
"logits/chosen": -0.06883351504802704,
"logits/rejected": -0.0933861956000328,
"logps/chosen": -0.324018269777298,
"logps/rejected": -0.36465924978256226,
"loss": 1.2409,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.324018269777298,
"rewards/margins": 0.04064100235700607,
"rewards/rejected": -0.36465924978256226,
"step": 275
},
{
"epoch": 0.5985303941215765,
"grad_norm": 6.1537667241918355,
"learning_rate": 2.4864234234060747e-07,
"logits/chosen": -0.12475994974374771,
"logits/rejected": -0.10665042698383331,
"logps/chosen": -0.3264842629432678,
"logps/rejected": -0.4078393876552582,
"loss": 1.2345,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.3264842629432678,
"rewards/margins": 0.08135511726140976,
"rewards/rejected": -0.4078393876552582,
"step": 280
},
{
"epoch": 0.6092184368737475,
"grad_norm": 8.294056815581182,
"learning_rate": 2.3762649275467223e-07,
"logits/chosen": -0.11794719845056534,
"logits/rejected": -0.13079698383808136,
"logps/chosen": -0.32731741666793823,
"logps/rejected": -0.4040375351905823,
"loss": 1.2497,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.32731741666793823,
"rewards/margins": 0.07672014087438583,
"rewards/rejected": -0.4040375351905823,
"step": 285
},
{
"epoch": 0.6199064796259185,
"grad_norm": 8.79096353154505,
"learning_rate": 2.2669787815863174e-07,
"logits/chosen": -0.032518744468688965,
"logits/rejected": -0.05465514585375786,
"logps/chosen": -0.32134541869163513,
"logps/rejected": -0.4223794937133789,
"loss": 1.2513,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.32134541869163513,
"rewards/margins": 0.10103406012058258,
"rewards/rejected": -0.4223794937133789,
"step": 290
},
{
"epoch": 0.6305945223780896,
"grad_norm": 5.400467561380045,
"learning_rate": 2.1587178320819919e-07,
"logits/chosen": -0.06230146810412407,
"logits/rejected": -0.0027563839685171843,
"logps/chosen": -0.27156487107276917,
"logps/rejected": -0.3670196831226349,
"loss": 1.2348,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.27156487107276917,
"rewards/margins": 0.09545480459928513,
"rewards/rejected": -0.3670196831226349,
"step": 295
},
{
"epoch": 0.6412825651302605,
"grad_norm": 12.259873359149967,
"learning_rate": 2.0516334917609277e-07,
"logits/chosen": -0.10260840505361557,
"logits/rejected": -0.04962821304798126,
"logps/chosen": -0.336132675409317,
"logps/rejected": -0.5095348358154297,
"loss": 1.2334,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.336132675409317,
"rewards/margins": 0.17340223491191864,
"rewards/rejected": -0.5095348358154297,
"step": 300
},
{
"epoch": 0.6519706078824316,
"grad_norm": 8.217920440778814,
"learning_rate": 1.9458755277559716e-07,
"logits/chosen": -0.11686922609806061,
"logits/rejected": -0.11177687346935272,
"logps/chosen": -0.31434646248817444,
"logps/rejected": -0.40717291831970215,
"loss": 1.24,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.31434646248817444,
"rewards/margins": 0.0928264707326889,
"rewards/rejected": -0.40717291831970215,
"step": 305
},
{
"epoch": 0.6626586506346025,
"grad_norm": 6.161104077211237,
"learning_rate": 1.8415918521427613e-07,
"logits/chosen": -0.18385252356529236,
"logits/rejected": -0.19196629524230957,
"logps/chosen": -0.31113195419311523,
"logps/rejected": -0.3769488036632538,
"loss": 1.2457,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.31113195419311523,
"rewards/margins": 0.06581689417362213,
"rewards/rejected": -0.3769488036632538,
"step": 310
},
{
"epoch": 0.6733466933867736,
"grad_norm": 10.34689683004134,
"learning_rate": 1.7389283150713038e-07,
"logits/chosen": -0.1292671114206314,
"logits/rejected": -0.1229550838470459,
"logps/chosen": -0.35926195979118347,
"logps/rejected": -0.42332392930984497,
"loss": 1.2549,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -0.35926195979118347,
"rewards/margins": 0.06406201422214508,
"rewards/rejected": -0.42332392930984497,
"step": 315
},
{
"epoch": 0.6840347361389446,
"grad_norm": 9.698239508672282,
"learning_rate": 1.6380285007813597e-07,
"logits/chosen": -0.11564113199710846,
"logits/rejected": -0.12854574620723724,
"logps/chosen": -0.3144187033176422,
"logps/rejected": -0.3402511179447174,
"loss": 1.2605,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.3144187033176422,
"rewards/margins": 0.02583237923681736,
"rewards/rejected": -0.3402511179447174,
"step": 320
},
{
"epoch": 0.6947227788911156,
"grad_norm": 6.070464304081896,
"learning_rate": 1.539033526786898e-07,
"logits/chosen": -0.1405663937330246,
"logits/rejected": -0.11777285486459732,
"logps/chosen": -0.32500436902046204,
"logps/rejected": -0.4695354402065277,
"loss": 1.2421,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.32500436902046204,
"rewards/margins": 0.14453105628490448,
"rewards/rejected": -0.4695354402065277,
"step": 325
},
{
"epoch": 0.7054108216432866,
"grad_norm": 5.549203189727758,
"learning_rate": 1.4420818465104924e-07,
"logits/chosen": -0.18301475048065186,
"logits/rejected": -0.18062211573123932,
"logps/chosen": -0.3069472312927246,
"logps/rejected": -0.3610253930091858,
"loss": 1.2325,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3069472312927246,
"rewards/margins": 0.05407816916704178,
"rewards/rejected": -0.3610253930091858,
"step": 330
},
{
"epoch": 0.7160988643954576,
"grad_norm": 6.763068049142248,
"learning_rate": 1.3473090556436928e-07,
"logits/chosen": -0.10250736773014069,
"logits/rejected": -0.11988594383001328,
"logps/chosen": -0.32310950756073,
"logps/rejected": -0.4247487485408783,
"loss": 1.2479,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.32310950756073,
"rewards/margins": 0.10163921117782593,
"rewards/rejected": -0.4247487485408783,
"step": 335
},
{
"epoch": 0.7267869071476286,
"grad_norm": 6.312494529658719,
"learning_rate": 1.2548477025041833e-07,
"logits/chosen": -0.1759723722934723,
"logits/rejected": -0.15970100462436676,
"logps/chosen": -0.31687361001968384,
"logps/rejected": -0.4269244074821472,
"loss": 1.2473,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.31687361001968384,
"rewards/margins": 0.11005084216594696,
"rewards/rejected": -0.4269244074821472,
"step": 340
},
{
"epoch": 0.7374749498997996,
"grad_norm": 7.012548386778781,
"learning_rate": 1.1648271026549805e-07,
"logits/chosen": -0.16853009164333344,
"logits/rejected": -0.1741519272327423,
"logps/chosen": -0.3043145537376404,
"logps/rejected": -0.41543588042259216,
"loss": 1.2355,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3043145537376404,
"rewards/margins": 0.11112137138843536,
"rewards/rejected": -0.41543588042259216,
"step": 345
},
{
"epoch": 0.7481629926519706,
"grad_norm": 8.592898323037762,
"learning_rate": 1.0773731580449275e-07,
"logits/chosen": -0.0907156690955162,
"logits/rejected": -0.10430169105529785,
"logps/chosen": -0.33002233505249023,
"logps/rejected": -0.432032972574234,
"loss": 1.2274,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.33002233505249023,
"rewards/margins": 0.10201063007116318,
"rewards/rejected": -0.432032972574234,
"step": 350
},
{
"epoch": 0.7588510354041417,
"grad_norm": 5.346851949829031,
"learning_rate": 9.926081809234262e-08,
"logits/chosen": -0.15248560905456543,
"logits/rejected": -0.14991840720176697,
"logps/chosen": -0.3535235822200775,
"logps/rejected": -0.5069695115089417,
"loss": 1.2334,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3535235822200775,
"rewards/margins": 0.15344594419002533,
"rewards/rejected": -0.5069695115089417,
"step": 355
},
{
"epoch": 0.7695390781563126,
"grad_norm": 17.058589802109246,
"learning_rate": 9.106507227756998e-08,
"logits/chosen": -0.11075520515441895,
"logits/rejected": -0.11937098205089569,
"logps/chosen": -0.3546099066734314,
"logps/rejected": -0.4121836721897125,
"loss": 1.2538,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.3546099066734314,
"rewards/margins": 0.05757373571395874,
"rewards/rejected": -0.4121836721897125,
"step": 360
},
{
"epoch": 0.7802271209084837,
"grad_norm": 8.71836207461451,
"learning_rate": 8.316154085178256e-08,
"logits/chosen": -0.16542118787765503,
"logits/rejected": -0.17096641659736633,
"logps/chosen": -0.35373011231422424,
"logps/rejected": -0.4567010998725891,
"loss": 1.2417,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.35373011231422424,
"rewards/margins": 0.10297106206417084,
"rewards/rejected": -0.4567010998725891,
"step": 365
},
{
"epoch": 0.7909151636606546,
"grad_norm": 7.987346992327182,
"learning_rate": 7.55612776183419e-08,
"logits/chosen": -0.1105569452047348,
"logits/rejected": -0.09565238654613495,
"logps/chosen": -0.34828633069992065,
"logps/rejected": -0.4095218777656555,
"loss": 1.2353,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.34828633069992065,
"rewards/margins": 0.06123558431863785,
"rewards/rejected": -0.4095218777656555,
"step": 370
},
{
"epoch": 0.8016032064128257,
"grad_norm": 8.515114879561368,
"learning_rate": 6.827491223262017e-08,
"logits/chosen": -0.1510094702243805,
"logits/rejected": -0.13519130647182465,
"logps/chosen": -0.3320384621620178,
"logps/rejected": -0.39452531933784485,
"loss": 1.2311,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3320384621620178,
"rewards/margins": 0.06248681992292404,
"rewards/rejected": -0.39452531933784485,
"step": 375
},
{
"epoch": 0.8122912491649966,
"grad_norm": 7.03157225265326,
"learning_rate": 6.131263533546572e-08,
"logits/chosen": -0.1356309950351715,
"logits/rejected": -0.13989830017089844,
"logps/chosen": -0.3249013423919678,
"logps/rejected": -0.4495007395744324,
"loss": 1.2322,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.3249013423919678,
"rewards/margins": 0.12459937483072281,
"rewards/rejected": -0.4495007395744324,
"step": 380
},
{
"epoch": 0.8229792919171677,
"grad_norm": 8.670843552810656,
"learning_rate": 5.468418430067059e-08,
"logits/chosen": -0.13802111148834229,
"logits/rejected": -0.12261234223842621,
"logps/chosen": -0.3400149345397949,
"logps/rejected": -0.4104432165622711,
"loss": 1.2468,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3400149345397949,
"rewards/margins": 0.07042822986841202,
"rewards/rejected": -0.4104432165622711,
"step": 385
},
{
"epoch": 0.8336673346693386,
"grad_norm": 6.105321944303861,
"learning_rate": 4.839882961637282e-08,
"logits/chosen": -0.1445506513118744,
"logits/rejected": -0.12124904245138168,
"logps/chosen": -0.3267291188240051,
"logps/rejected": -0.3966960906982422,
"loss": 1.2457,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.3267291188240051,
"rewards/margins": 0.06996697932481766,
"rewards/rejected": -0.3966960906982422,
"step": 390
},
{
"epoch": 0.8443553774215097,
"grad_norm": 7.42690529749409,
"learning_rate": 4.2465361919440165e-08,
"logits/chosen": -0.17786213755607605,
"logits/rejected": -0.1693142205476761,
"logps/chosen": -0.3282647728919983,
"logps/rejected": -0.3676055073738098,
"loss": 1.2416,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.3282647728919983,
"rewards/margins": 0.03934071958065033,
"rewards/rejected": -0.3676055073738098,
"step": 395
},
{
"epoch": 0.8550434201736807,
"grad_norm": 6.448291334596692,
"learning_rate": 3.6892079700970036e-08,
"logits/chosen": -0.19666707515716553,
"logits/rejected": -0.1866413801908493,
"logps/chosen": -0.3443647027015686,
"logps/rejected": -0.3764795660972595,
"loss": 1.2443,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.3443647027015686,
"rewards/margins": 0.032114893198013306,
"rewards/rejected": -0.3764795660972595,
"step": 400
},
{
"epoch": 0.8550434201736807,
"eval_logits/chosen": 0.01234325673431158,
"eval_logits/rejected": 0.003141747321933508,
"eval_logps/chosen": -0.33608412742614746,
"eval_logps/rejected": -0.401297003030777,
"eval_loss": 1.241620659828186,
"eval_rewards/accuracies": 0.5914633870124817,
"eval_rewards/chosen": -0.33608412742614746,
"eval_rewards/margins": 0.06521284580230713,
"eval_rewards/rejected": -0.401297003030777,
"eval_runtime": 422.8878,
"eval_samples_per_second": 4.637,
"eval_steps_per_second": 0.291,
"step": 400
},
{
"epoch": 0.8657314629258517,
"grad_norm": 8.756154081277105,
"learning_rate": 3.1686777700099e-08,
"logits/chosen": -0.12114688009023666,
"logits/rejected": -0.17152926325798035,
"logps/chosen": -0.3494204878807068,
"logps/rejected": -0.3755631446838379,
"loss": 1.245,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.3494204878807068,
"rewards/margins": 0.026142627000808716,
"rewards/rejected": -0.3755631446838379,
"step": 405
},
{
"epoch": 0.8764195056780227,
"grad_norm": 7.899265779506385,
"learning_rate": 2.685673600235524e-08,
"logits/chosen": -0.11135631799697876,
"logits/rejected": -0.13768556714057922,
"logps/chosen": -0.3498903512954712,
"logps/rejected": -0.3871268332004547,
"loss": 1.261,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.3498903512954712,
"rewards/margins": 0.03723648935556412,
"rewards/rejected": -0.3871268332004547,
"step": 410
},
{
"epoch": 0.8871075484301937,
"grad_norm": 14.01441787078892,
"learning_rate": 2.2408709857800988e-08,
"logits/chosen": -0.12833549082279205,
"logits/rejected": -0.10664300620555878,
"logps/chosen": -0.2865287661552429,
"logps/rejected": -0.38992565870285034,
"loss": 1.2338,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.2865287661552429,
"rewards/margins": 0.10339690744876862,
"rewards/rejected": -0.38992565870285034,
"step": 415
},
{
"epoch": 0.8977955911823647,
"grad_norm": 5.705006993624562,
"learning_rate": 1.8348920233204167e-08,
"logits/chosen": -0.08621132373809814,
"logits/rejected": -0.06882130354642868,
"logps/chosen": -0.3323257565498352,
"logps/rejected": -0.4798430800437927,
"loss": 1.2399,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.3323257565498352,
"rewards/margins": 0.1475173979997635,
"rewards/rejected": -0.4798430800437927,
"step": 420
},
{
"epoch": 0.9084836339345357,
"grad_norm": 5.012398062093744,
"learning_rate": 1.468304511145394e-08,
"logits/chosen": -0.028029289096593857,
"logits/rejected": -0.07289622724056244,
"logps/chosen": -0.31065744161605835,
"logps/rejected": -0.4257968068122864,
"loss": 1.2356,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.31065744161605835,
"rewards/margins": 0.11513932794332504,
"rewards/rejected": -0.4257968068122864,
"step": 425
},
{
"epoch": 0.9191716766867067,
"grad_norm": 6.051006778161325,
"learning_rate": 1.1416211550388222e-08,
"logits/chosen": -0.11329293251037598,
"logits/rejected": -0.09477666765451431,
"logps/chosen": -0.29777267575263977,
"logps/rejected": -0.3625491261482239,
"loss": 1.2347,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.29777267575263977,
"rewards/margins": 0.0647764801979065,
"rewards/rejected": -0.3625491261482239,
"step": 430
},
{
"epoch": 0.9298597194388778,
"grad_norm": 7.54223521221758,
"learning_rate": 8.552988512139748e-09,
"logits/chosen": -0.10169041156768799,
"logits/rejected": -0.11717716604471207,
"logps/chosen": -0.3231045603752136,
"logps/rejected": -0.44011014699935913,
"loss": 1.2366,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.3231045603752136,
"rewards/margins": 0.11700558662414551,
"rewards/rejected": -0.44011014699935913,
"step": 435
},
{
"epoch": 0.9405477621910487,
"grad_norm": 17.447935307423492,
"learning_rate": 6.097380473029356e-09,
"logits/chosen": -0.13727042078971863,
"logits/rejected": -0.1495351493358612,
"logps/chosen": -0.3348698019981384,
"logps/rejected": -0.39052897691726685,
"loss": 1.2477,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.3348698019981384,
"rewards/margins": 0.05565913766622543,
"rewards/rejected": -0.39052897691726685,
"step": 440
},
{
"epoch": 0.9512358049432198,
"grad_norm": 7.32756997364182,
"learning_rate": 4.052821822943597e-09,
"logits/chosen": -0.05230356380343437,
"logits/rejected": -0.05104394629597664,
"logps/chosen": -0.31396013498306274,
"logps/rejected": -0.38287192583084106,
"loss": 1.2578,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.31396013498306274,
"rewards/margins": 0.06891177594661713,
"rewards/rejected": -0.38287192583084106,
"step": 445
},
{
"epoch": 0.9619238476953907,
"grad_norm": 7.458324039594406,
"learning_rate": 2.4221720620301368e-09,
"logits/chosen": -0.11121572554111481,
"logits/rejected": -0.10439705848693848,
"logps/chosen": -0.3217839300632477,
"logps/rejected": -0.4048451781272888,
"loss": 1.2277,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.3217839300632477,
"rewards/margins": 0.08306124061346054,
"rewards/rejected": -0.4048451781272888,
"step": 450
},
{
"epoch": 0.9726118904475618,
"grad_norm": 7.12208845758839,
"learning_rate": 1.2077118014282794e-09,
"logits/chosen": -0.06711649894714355,
"logits/rejected": -0.03383985906839371,
"logps/chosen": -0.34118419885635376,
"logps/rejected": -0.4141947329044342,
"loss": 1.2497,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.34118419885635376,
"rewards/margins": 0.07301049679517746,
"rewards/rejected": -0.4141947329044342,
"step": 455
},
{
"epoch": 0.9832999331997327,
"grad_norm": 6.594294682889427,
"learning_rate": 4.1113957362785e-10,
"logits/chosen": -0.06302393972873688,
"logits/rejected": -0.08739855140447617,
"logps/chosen": -0.3296561539173126,
"logps/rejected": -0.37389543652534485,
"loss": 1.2492,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.3296561539173126,
"rewards/margins": 0.04423931986093521,
"rewards/rejected": -0.37389543652534485,
"step": 460
},
{
"epoch": 0.9939879759519038,
"grad_norm": 8.269794544806404,
"learning_rate": 3.3569456917970085e-11,
"logits/chosen": -0.036546867340803146,
"logits/rejected": -0.05820956081151962,
"logps/chosen": -0.31292372941970825,
"logps/rejected": -0.4097159504890442,
"loss": 1.2414,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.31292372941970825,
"rewards/margins": 0.09679219126701355,
"rewards/rejected": -0.4097159504890442,
"step": 465
},
{
"epoch": 0.9982631930527722,
"step": 467,
"total_flos": 0.0,
"train_loss": 1.248079635857004,
"train_runtime": 20917.8016,
"train_samples_per_second": 2.862,
"train_steps_per_second": 0.022
}
],
"logging_steps": 5,
"max_steps": 467,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}