yakazimir's picture
Model save
bd4aaf8 verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9989071038251366,
"eval_steps": 400,
"global_step": 457,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01092896174863388,
"grad_norm": 334.04762282844746,
"learning_rate": 1.0869565217391303e-07,
"logits/chosen": -1.0183193683624268,
"logits/rejected": -1.0128684043884277,
"logps/chosen": -0.2805940508842468,
"logps/rejected": -0.28598836064338684,
"loss": 3.2497,
"rewards/accuracies": 0.53125,
"rewards/chosen": -2.805940628051758,
"rewards/margins": 0.05394287034869194,
"rewards/rejected": -2.8598835468292236,
"step": 5
},
{
"epoch": 0.02185792349726776,
"grad_norm": 180.71056791479407,
"learning_rate": 2.1739130434782607e-07,
"logits/chosen": -1.0511164665222168,
"logits/rejected": -1.0003410577774048,
"logps/chosen": -0.2573581039905548,
"logps/rejected": -0.27061566710472107,
"loss": 3.257,
"rewards/accuracies": 0.53125,
"rewards/chosen": -2.5735812187194824,
"rewards/margins": 0.13257600367069244,
"rewards/rejected": -2.7061567306518555,
"step": 10
},
{
"epoch": 0.03278688524590164,
"grad_norm": 223.3194766304879,
"learning_rate": 3.260869565217391e-07,
"logits/chosen": -0.9949806928634644,
"logits/rejected": -0.9507344961166382,
"logps/chosen": -0.26757049560546875,
"logps/rejected": -0.2733513414859772,
"loss": 3.2029,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -2.6757049560546875,
"rewards/margins": 0.05780839920043945,
"rewards/rejected": -2.733513355255127,
"step": 15
},
{
"epoch": 0.04371584699453552,
"grad_norm": 208.56574432084847,
"learning_rate": 4.3478260869565214e-07,
"logits/chosen": -0.9485995173454285,
"logits/rejected": -0.8978453874588013,
"logps/chosen": -0.2727873921394348,
"logps/rejected": -0.28461021184921265,
"loss": 3.2536,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -2.7278740406036377,
"rewards/margins": 0.11822839826345444,
"rewards/rejected": -2.846102237701416,
"step": 20
},
{
"epoch": 0.0546448087431694,
"grad_norm": 273.9157814428763,
"learning_rate": 5.434782608695652e-07,
"logits/chosen": -0.9404302835464478,
"logits/rejected": -0.8673273324966431,
"logps/chosen": -0.2763015627861023,
"logps/rejected": -0.29439201951026917,
"loss": 3.1849,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -2.7630152702331543,
"rewards/margins": 0.18090477585792542,
"rewards/rejected": -2.943920135498047,
"step": 25
},
{
"epoch": 0.06557377049180328,
"grad_norm": 252.26801160371912,
"learning_rate": 6.521739130434782e-07,
"logits/chosen": -1.0567680597305298,
"logits/rejected": -0.9916139841079712,
"logps/chosen": -0.26832228899002075,
"logps/rejected": -0.2841094732284546,
"loss": 3.2649,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -2.683222770690918,
"rewards/margins": 0.15787188708782196,
"rewards/rejected": -2.841094493865967,
"step": 30
},
{
"epoch": 0.07650273224043716,
"grad_norm": 120.68425546239968,
"learning_rate": 7.608695652173913e-07,
"logits/chosen": -1.0046221017837524,
"logits/rejected": -0.9383124113082886,
"logps/chosen": -0.2608323395252228,
"logps/rejected": -0.27886277437210083,
"loss": 3.1693,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -2.608323335647583,
"rewards/margins": 0.1803044080734253,
"rewards/rejected": -2.788627862930298,
"step": 35
},
{
"epoch": 0.08743169398907104,
"grad_norm": 146.20382704567854,
"learning_rate": 8.695652173913043e-07,
"logits/chosen": -0.9549115896224976,
"logits/rejected": -0.8933135271072388,
"logps/chosen": -0.28391939401626587,
"logps/rejected": -0.29877811670303345,
"loss": 3.2899,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -2.83919358253479,
"rewards/margins": 0.14858722686767578,
"rewards/rejected": -2.987781047821045,
"step": 40
},
{
"epoch": 0.09836065573770492,
"grad_norm": 93.83055902381597,
"learning_rate": 9.782608695652173e-07,
"logits/chosen": -1.0223612785339355,
"logits/rejected": -0.9385920763015747,
"logps/chosen": -0.2871010899543762,
"logps/rejected": -0.3065423369407654,
"loss": 3.0292,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -2.8710105419158936,
"rewards/margins": 0.19441291689872742,
"rewards/rejected": -3.0654234886169434,
"step": 45
},
{
"epoch": 0.1092896174863388,
"grad_norm": 281.2686273771578,
"learning_rate": 9.997663088532014e-07,
"logits/chosen": -0.9519433975219727,
"logits/rejected": -0.8702558279037476,
"logps/chosen": -0.28290504217147827,
"logps/rejected": -0.29115915298461914,
"loss": 3.1626,
"rewards/accuracies": 0.48750001192092896,
"rewards/chosen": -2.829050302505493,
"rewards/margins": 0.08254138380289078,
"rewards/rejected": -2.9115920066833496,
"step": 50
},
{
"epoch": 0.12021857923497267,
"grad_norm": 121.70775424511473,
"learning_rate": 9.98817312944725e-07,
"logits/chosen": -0.9872657060623169,
"logits/rejected": -0.8688336610794067,
"logps/chosen": -0.2818766236305237,
"logps/rejected": -0.3157486021518707,
"loss": 3.0197,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -2.8187661170959473,
"rewards/margins": 0.3387202322483063,
"rewards/rejected": -3.1574864387512207,
"step": 55
},
{
"epoch": 0.13114754098360656,
"grad_norm": 86.612813297471,
"learning_rate": 9.971397915250336e-07,
"logits/chosen": -1.0298466682434082,
"logits/rejected": -0.9812299609184265,
"logps/chosen": -0.2719958424568176,
"logps/rejected": -0.31191515922546387,
"loss": 2.8695,
"rewards/accuracies": 0.59375,
"rewards/chosen": -2.719958782196045,
"rewards/margins": 0.399192750453949,
"rewards/rejected": -3.1191515922546387,
"step": 60
},
{
"epoch": 0.14207650273224043,
"grad_norm": 161.66395420923772,
"learning_rate": 9.94736194623663e-07,
"logits/chosen": -0.9994015693664551,
"logits/rejected": -0.9284494519233704,
"logps/chosen": -0.31242281198501587,
"logps/rejected": -0.34273454546928406,
"loss": 3.0819,
"rewards/accuracies": 0.53125,
"rewards/chosen": -3.124228000640869,
"rewards/margins": 0.30311769247055054,
"rewards/rejected": -3.4273457527160645,
"step": 65
},
{
"epoch": 0.15300546448087432,
"grad_norm": 335.25942540938917,
"learning_rate": 9.916100327075037e-07,
"logits/chosen": -0.9326862096786499,
"logits/rejected": -0.9134162664413452,
"logps/chosen": -0.3001170754432678,
"logps/rejected": -0.3256034851074219,
"loss": 2.8894,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -3.0011706352233887,
"rewards/margins": 0.2548643946647644,
"rewards/rejected": -3.2560348510742188,
"step": 70
},
{
"epoch": 0.16393442622950818,
"grad_norm": 79.67001422366646,
"learning_rate": 9.877658715537428e-07,
"logits/chosen": -0.9303417205810547,
"logits/rejected": -0.9135540127754211,
"logps/chosen": -0.32335883378982544,
"logps/rejected": -0.35447338223457336,
"loss": 2.9985,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.233588695526123,
"rewards/margins": 0.3111455738544464,
"rewards/rejected": -3.544733762741089,
"step": 75
},
{
"epoch": 0.17486338797814208,
"grad_norm": 83.46864509697639,
"learning_rate": 9.832093255815216e-07,
"logits/chosen": -0.942563533782959,
"logits/rejected": -0.8771953582763672,
"logps/chosen": -0.31826624274253845,
"logps/rejected": -0.3393171727657318,
"loss": 2.9663,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -3.1826624870300293,
"rewards/margins": 0.21050938963890076,
"rewards/rejected": -3.393172025680542,
"step": 80
},
{
"epoch": 0.18579234972677597,
"grad_norm": 119.60039352898374,
"learning_rate": 9.779470496520441e-07,
"logits/chosen": -0.9433110952377319,
"logits/rejected": -0.8911749124526978,
"logps/chosen": -0.31765812635421753,
"logps/rejected": -0.3729430139064789,
"loss": 2.8342,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.176581621170044,
"rewards/margins": 0.5528485178947449,
"rewards/rejected": -3.7294304370880127,
"step": 85
},
{
"epoch": 0.19672131147540983,
"grad_norm": 96.2051979280833,
"learning_rate": 9.719867293491144e-07,
"logits/chosen": -1.0151774883270264,
"logits/rejected": -0.9306642413139343,
"logps/chosen": -0.3450705409049988,
"logps/rejected": -0.38217830657958984,
"loss": 2.9117,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -3.4507057666778564,
"rewards/margins": 0.37107738852500916,
"rewards/rejected": -3.8217825889587402,
"step": 90
},
{
"epoch": 0.20765027322404372,
"grad_norm": 90.0031374445934,
"learning_rate": 9.653370697542987e-07,
"logits/chosen": -0.9307557344436646,
"logits/rejected": -0.9305530786514282,
"logps/chosen": -0.34383976459503174,
"logps/rejected": -0.3695612847805023,
"loss": 2.7261,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -3.4383978843688965,
"rewards/margins": 0.2572151720523834,
"rewards/rejected": -3.695613145828247,
"step": 95
},
{
"epoch": 0.2185792349726776,
"grad_norm": 106.07682375515387,
"learning_rate": 9.580077827331037e-07,
"logits/chosen": -0.9448221921920776,
"logits/rejected": -0.8973434567451477,
"logps/chosen": -0.38050413131713867,
"logps/rejected": -0.4358042776584625,
"loss": 2.7135,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -3.805041551589966,
"rewards/margins": 0.5530012249946594,
"rewards/rejected": -4.3580427169799805,
"step": 100
},
{
"epoch": 0.22950819672131148,
"grad_norm": 124.64829049828732,
"learning_rate": 9.500095727507419e-07,
"logits/chosen": -0.9876478910446167,
"logits/rejected": -0.9548322558403015,
"logps/chosen": -0.3621714115142822,
"logps/rejected": -0.4053111672401428,
"loss": 2.8156,
"rewards/accuracies": 0.625,
"rewards/chosen": -3.6217141151428223,
"rewards/margins": 0.43139734864234924,
"rewards/rejected": -4.0531110763549805,
"step": 105
},
{
"epoch": 0.24043715846994534,
"grad_norm": 109.72879050820204,
"learning_rate": 9.413541212382004e-07,
"logits/chosen": -0.9878107905387878,
"logits/rejected": -0.9694309234619141,
"logps/chosen": -0.37187474966049194,
"logps/rejected": -0.4513128697872162,
"loss": 2.6674,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -3.718747615814209,
"rewards/margins": 0.7943812608718872,
"rewards/rejected": -4.513128280639648,
"step": 110
},
{
"epoch": 0.25136612021857924,
"grad_norm": 89.69262563371302,
"learning_rate": 9.320540695314438e-07,
"logits/chosen": -0.9974454045295715,
"logits/rejected": -0.9548517465591431,
"logps/chosen": -0.3737660050392151,
"logps/rejected": -0.47232741117477417,
"loss": 2.5733,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -3.7376599311828613,
"rewards/margins": 0.9856147766113281,
"rewards/rejected": -4.7232747077941895,
"step": 115
},
{
"epoch": 0.26229508196721313,
"grad_norm": 82.76010396371875,
"learning_rate": 9.221230004086721e-07,
"logits/chosen": -1.0242812633514404,
"logits/rejected": -0.949226975440979,
"logps/chosen": -0.3816962242126465,
"logps/rejected": -0.4385160803794861,
"loss": 2.6795,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -3.816962480545044,
"rewards/margins": 0.5681988000869751,
"rewards/rejected": -4.38516092300415,
"step": 120
},
{
"epoch": 0.273224043715847,
"grad_norm": 186.60007457771528,
"learning_rate": 9.11575418252596e-07,
"logits/chosen": -0.9328588247299194,
"logits/rejected": -0.8940401077270508,
"logps/chosen": -0.395950585603714,
"logps/rejected": -0.46849578619003296,
"loss": 2.5289,
"rewards/accuracies": 0.6875,
"rewards/chosen": -3.9595062732696533,
"rewards/margins": 0.7254521250724792,
"rewards/rejected": -4.684957981109619,
"step": 125
},
{
"epoch": 0.28415300546448086,
"grad_norm": 91.04741453080123,
"learning_rate": 9.004267278667031e-07,
"logits/chosen": -0.9401098489761353,
"logits/rejected": -0.9308618307113647,
"logps/chosen": -0.41634416580200195,
"logps/rejected": -0.5375351905822754,
"loss": 2.5277,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -4.1634416580200195,
"rewards/margins": 1.2119100093841553,
"rewards/rejected": -5.375351905822754,
"step": 130
},
{
"epoch": 0.29508196721311475,
"grad_norm": 72.34007677315606,
"learning_rate": 8.886932119764565e-07,
"logits/chosen": -0.9937323331832886,
"logits/rejected": -0.9037588834762573,
"logps/chosen": -0.39853519201278687,
"logps/rejected": -0.49872303009033203,
"loss": 2.4319,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -3.9853522777557373,
"rewards/margins": 1.0018781423568726,
"rewards/rejected": -4.987229824066162,
"step": 135
},
{
"epoch": 0.30601092896174864,
"grad_norm": 83.52757366186898,
"learning_rate": 8.763920074482809e-07,
"logits/chosen": -0.9973970651626587,
"logits/rejected": -0.9379005432128906,
"logps/chosen": -0.42847341299057007,
"logps/rejected": -0.560250461101532,
"loss": 2.2317,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -4.28473424911499,
"rewards/margins": 1.3177704811096191,
"rewards/rejected": -5.602505207061768,
"step": 140
},
{
"epoch": 0.31693989071038253,
"grad_norm": 106.73407366387536,
"learning_rate": 8.635410802610723e-07,
"logits/chosen": -0.9793792963027954,
"logits/rejected": -0.9569779634475708,
"logps/chosen": -0.4017811417579651,
"logps/rejected": -0.4612775444984436,
"loss": 2.3643,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -4.017810821533203,
"rewards/margins": 0.5949639081954956,
"rewards/rejected": -4.612774848937988,
"step": 145
},
{
"epoch": 0.32786885245901637,
"grad_norm": 60.51485768923878,
"learning_rate": 8.501591992667849e-07,
"logits/chosen": -1.044575572013855,
"logits/rejected": -1.0059934854507446,
"logps/chosen": -0.4378751218318939,
"logps/rejected": -0.5975905656814575,
"loss": 2.2151,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -4.378750801086426,
"rewards/margins": 1.5971543788909912,
"rewards/rejected": -5.975905418395996,
"step": 150
},
{
"epoch": 0.33879781420765026,
"grad_norm": 68.73684458948155,
"learning_rate": 8.362659087784152e-07,
"logits/chosen": -0.9929186105728149,
"logits/rejected": -0.9317314028739929,
"logps/chosen": -0.4304039478302002,
"logps/rejected": -0.5334851741790771,
"loss": 2.3197,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -4.304039478302002,
"rewards/margins": 1.0308125019073486,
"rewards/rejected": -5.3348517417907715,
"step": 155
},
{
"epoch": 0.34972677595628415,
"grad_norm": 90.32236437312078,
"learning_rate": 8.218815000254231e-07,
"logits/chosen": -1.0294911861419678,
"logits/rejected": -0.9657951593399048,
"logps/chosen": -0.4845092296600342,
"logps/rejected": -0.566204309463501,
"loss": 2.3671,
"rewards/accuracies": 0.6875,
"rewards/chosen": -4.845091819763184,
"rewards/margins": 0.81695157289505,
"rewards/rejected": -5.662043571472168,
"step": 160
},
{
"epoch": 0.36065573770491804,
"grad_norm": 83.55321323451464,
"learning_rate": 8.07026981518276e-07,
"logits/chosen": -1.0140776634216309,
"logits/rejected": -0.9537431597709656,
"logps/chosen": -0.46461719274520874,
"logps/rejected": -0.5585328340530396,
"loss": 2.3255,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -4.646172046661377,
"rewards/margins": 0.9391565322875977,
"rewards/rejected": -5.585328578948975,
"step": 165
},
{
"epoch": 0.37158469945355194,
"grad_norm": 91.94544798993923,
"learning_rate": 7.917240483654e-07,
"logits/chosen": -1.0053484439849854,
"logits/rejected": -0.9363968968391418,
"logps/chosen": -0.4600796699523926,
"logps/rejected": -0.5375440716743469,
"loss": 2.416,
"rewards/accuracies": 0.6875,
"rewards/chosen": -4.600796699523926,
"rewards/margins": 0.7746434211730957,
"rewards/rejected": -5.3754401206970215,
"step": 170
},
{
"epoch": 0.3825136612021858,
"grad_norm": 91.07280013110446,
"learning_rate": 7.759950505873521e-07,
"logits/chosen": -1.0666639804840088,
"logits/rejected": -1.029950737953186,
"logps/chosen": -0.4774397313594818,
"logps/rejected": -0.5452654957771301,
"loss": 2.2653,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -4.774397850036621,
"rewards/margins": 0.6782568097114563,
"rewards/rejected": -5.452654838562012,
"step": 175
},
{
"epoch": 0.39344262295081966,
"grad_norm": 85.29762605205215,
"learning_rate": 7.598629604744872e-07,
"logits/chosen": -1.0595664978027344,
"logits/rejected": -1.0487810373306274,
"logps/chosen": -0.4570770263671875,
"logps/rejected": -0.5979724526405334,
"loss": 2.1524,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -4.570770263671875,
"rewards/margins": 1.4089539051055908,
"rewards/rejected": -5.979724407196045,
"step": 180
},
{
"epoch": 0.40437158469945356,
"grad_norm": 67.2351189385839,
"learning_rate": 7.433513390357989e-07,
"logits/chosen": -1.0875113010406494,
"logits/rejected": -1.0918848514556885,
"logps/chosen": -0.4790703356266022,
"logps/rejected": -0.6097405552864075,
"loss": 2.164,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -4.790703773498535,
"rewards/margins": 1.3067026138305664,
"rewards/rejected": -6.097405433654785,
"step": 185
},
{
"epoch": 0.41530054644808745,
"grad_norm": 106.92479621624057,
"learning_rate": 7.264843015879321e-07,
"logits/chosen": -1.0997188091278076,
"logits/rejected": -1.0514479875564575,
"logps/chosen": -0.467475026845932,
"logps/rejected": -0.6209039092063904,
"loss": 2.2151,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -4.674751281738281,
"rewards/margins": 1.5342884063720703,
"rewards/rejected": -6.209038734436035,
"step": 190
},
{
"epoch": 0.4262295081967213,
"grad_norm": 79.4429028708432,
"learning_rate": 7.092864825346266e-07,
"logits/chosen": -1.12069833278656,
"logits/rejected": -1.0915439128875732,
"logps/chosen": -0.5538396239280701,
"logps/rejected": -0.7245546579360962,
"loss": 2.2765,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -5.538395881652832,
"rewards/margins": 1.7071504592895508,
"rewards/rejected": -7.245546817779541,
"step": 195
},
{
"epoch": 0.4371584699453552,
"grad_norm": 79.46289075402596,
"learning_rate": 6.917829993880302e-07,
"logits/chosen": -1.1029081344604492,
"logits/rejected": -1.0124149322509766,
"logps/chosen": -0.5200821161270142,
"logps/rejected": -0.6488970518112183,
"loss": 2.1257,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -5.200821399688721,
"rewards/margins": 1.288149118423462,
"rewards/rejected": -6.4889702796936035,
"step": 200
},
{
"epoch": 0.44808743169398907,
"grad_norm": 67.1583967725435,
"learning_rate": 6.739994160844309e-07,
"logits/chosen": -1.0795233249664307,
"logits/rejected": -1.0885263681411743,
"logps/chosen": -0.48235517740249634,
"logps/rejected": -0.6249179244041443,
"loss": 2.0323,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -4.823552131652832,
"rewards/margins": 1.4256280660629272,
"rewards/rejected": -6.249179363250732,
"step": 205
},
{
"epoch": 0.45901639344262296,
"grad_norm": 131.31523382011193,
"learning_rate": 6.559617056479827e-07,
"logits/chosen": -1.0998046398162842,
"logits/rejected": -1.0920783281326294,
"logps/chosen": -0.5168774724006653,
"logps/rejected": -0.6893444657325745,
"loss": 2.0968,
"rewards/accuracies": 0.78125,
"rewards/chosen": -5.168774127960205,
"rewards/margins": 1.7246698141098022,
"rewards/rejected": -6.893445014953613,
"step": 210
},
{
"epoch": 0.46994535519125685,
"grad_norm": 92.8827942851212,
"learning_rate": 6.376962122569567e-07,
"logits/chosen": -1.1027876138687134,
"logits/rejected": -1.0451091527938843,
"logps/chosen": -0.5266443490982056,
"logps/rejected": -0.6980344653129578,
"loss": 2.2334,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -5.266442775726318,
"rewards/margins": 1.7139012813568115,
"rewards/rejected": -6.980344295501709,
"step": 215
},
{
"epoch": 0.4808743169398907,
"grad_norm": 75.78200124794627,
"learning_rate": 6.192296127679192e-07,
"logits/chosen": -1.163720965385437,
"logits/rejected": -1.1323108673095703,
"logps/chosen": -0.5482050180435181,
"logps/rejected": -0.6795030832290649,
"loss": 2.0812,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -5.482050895690918,
"rewards/margins": 1.3129799365997314,
"rewards/rejected": -6.7950310707092285,
"step": 220
},
{
"epoch": 0.4918032786885246,
"grad_norm": 68.50625090212893,
"learning_rate": 6.005888777540319e-07,
"logits/chosen": -1.1991808414459229,
"logits/rejected": -1.1517974138259888,
"logps/chosen": -0.5413341522216797,
"logps/rejected": -0.6963613033294678,
"loss": 2.1219,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -5.413341522216797,
"rewards/margins": 1.5502712726593018,
"rewards/rejected": -6.963613033294678,
"step": 225
},
{
"epoch": 0.5027322404371585,
"grad_norm": 73.44567710043167,
"learning_rate": 5.818012321143773e-07,
"logits/chosen": -1.0955681800842285,
"logits/rejected": -1.0907937288284302,
"logps/chosen": -0.5347357988357544,
"logps/rejected": -0.7202361226081848,
"loss": 2.0511,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -5.347357749938965,
"rewards/margins": 1.855002760887146,
"rewards/rejected": -7.2023606300354,
"step": 230
},
{
"epoch": 0.5136612021857924,
"grad_norm": 84.7486173764228,
"learning_rate": 5.628941153118388e-07,
"logits/chosen": -1.113163709640503,
"logits/rejected": -1.0674149990081787,
"logps/chosen": -0.534842312335968,
"logps/rejected": -0.6825965046882629,
"loss": 2.0329,
"rewards/accuracies": 0.78125,
"rewards/chosen": -5.348422527313232,
"rewards/margins": 1.4775424003601074,
"rewards/rejected": -6.825964450836182,
"step": 235
},
{
"epoch": 0.5245901639344263,
"grad_norm": 70.6295555021999,
"learning_rate": 5.438951412976098e-07,
"logits/chosen": -1.1475393772125244,
"logits/rejected": -1.1534240245819092,
"logps/chosen": -0.5095857381820679,
"logps/rejected": -0.6976131200790405,
"loss": 1.8823,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -5.095856666564941,
"rewards/margins": 1.8802751302719116,
"rewards/rejected": -6.976132392883301,
"step": 240
},
{
"epoch": 0.5355191256830601,
"grad_norm": 69.47924677139383,
"learning_rate": 5.248320581808619e-07,
"logits/chosen": -1.0735324621200562,
"logits/rejected": -1.0195956230163574,
"logps/chosen": -0.5255709290504456,
"logps/rejected": -0.7123461961746216,
"loss": 1.9839,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -5.255709648132324,
"rewards/margins": 1.8677523136138916,
"rewards/rejected": -7.123461723327637,
"step": 245
},
{
"epoch": 0.546448087431694,
"grad_norm": 66.46597890028497,
"learning_rate": 5.057327077024744e-07,
"logits/chosen": -1.1226236820220947,
"logits/rejected": -1.0827569961547852,
"logps/chosen": -0.5162156224250793,
"logps/rejected": -0.6480467319488525,
"loss": 2.0699,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -5.162156105041504,
"rewards/margins": 1.3183108568191528,
"rewards/rejected": -6.480466365814209,
"step": 250
},
{
"epoch": 0.5573770491803278,
"grad_norm": 76.19586118186004,
"learning_rate": 4.866249845720132e-07,
"logits/chosen": -1.107912540435791,
"logits/rejected": -1.071297287940979,
"logps/chosen": -0.5693866014480591,
"logps/rejected": -0.735679566860199,
"loss": 1.9599,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -5.693865776062012,
"rewards/margins": 1.6629297733306885,
"rewards/rejected": -7.356795310974121,
"step": 255
},
{
"epoch": 0.5683060109289617,
"grad_norm": 73.20621943329614,
"learning_rate": 4.675367957273505e-07,
"logits/chosen": -1.0760225057601929,
"logits/rejected": -1.0634446144104004,
"logps/chosen": -0.5194670557975769,
"logps/rejected": -0.6913676261901855,
"loss": 1.9788,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -5.1946702003479,
"rewards/margins": 1.7190055847167969,
"rewards/rejected": -6.9136762619018555,
"step": 260
},
{
"epoch": 0.5792349726775956,
"grad_norm": 111.34081159607314,
"learning_rate": 4.4849601957642285e-07,
"logits/chosen": -1.1172699928283691,
"logits/rejected": -1.0767419338226318,
"logps/chosen": -0.5310412645339966,
"logps/rejected": -0.6969484686851501,
"loss": 2.0084,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -5.310412406921387,
"rewards/margins": 1.6590715646743774,
"rewards/rejected": -6.969484806060791,
"step": 265
},
{
"epoch": 0.5901639344262295,
"grad_norm": 79.77425345162416,
"learning_rate": 4.295304652806592e-07,
"logits/chosen": -1.1290314197540283,
"logits/rejected": -1.098478078842163,
"logps/chosen": -0.5287065505981445,
"logps/rejected": -0.7144958972930908,
"loss": 1.8771,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -5.2870659828186035,
"rewards/margins": 1.8578933477401733,
"rewards/rejected": -7.144959449768066,
"step": 270
},
{
"epoch": 0.6010928961748634,
"grad_norm": 63.79079426666398,
"learning_rate": 4.106678321395433e-07,
"logits/chosen": -1.100162148475647,
"logits/rejected": -1.0285935401916504,
"logps/chosen": -0.5441292524337769,
"logps/rejected": -0.6436029672622681,
"loss": 1.9928,
"rewards/accuracies": 0.65625,
"rewards/chosen": -5.4412922859191895,
"rewards/margins": 0.9947379231452942,
"rewards/rejected": -6.43602991104126,
"step": 275
},
{
"epoch": 0.6120218579234973,
"grad_norm": 84.61811881052614,
"learning_rate": 3.9193566913562915e-07,
"logits/chosen": -1.055173635482788,
"logits/rejected": -1.0575199127197266,
"logps/chosen": -0.5374099612236023,
"logps/rejected": -0.7436326742172241,
"loss": 1.9752,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -5.374099254608154,
"rewards/margins": 2.062227249145508,
"rewards/rejected": -7.436326503753662,
"step": 280
},
{
"epoch": 0.6229508196721312,
"grad_norm": 69.51995916576952,
"learning_rate": 3.7336133469909623e-07,
"logits/chosen": -1.1819168329238892,
"logits/rejected": -1.1467748880386353,
"logps/chosen": -0.5230523347854614,
"logps/rejected": -0.7193282842636108,
"loss": 1.923,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -5.230523109436035,
"rewards/margins": 1.9627596139907837,
"rewards/rejected": -7.1932830810546875,
"step": 285
},
{
"epoch": 0.6338797814207651,
"grad_norm": 98.48874799948564,
"learning_rate": 3.549719567506076e-07,
"logits/chosen": -1.1354302167892456,
"logits/rejected": -1.0900018215179443,
"logps/chosen": -0.5486642718315125,
"logps/rejected": -0.7162230610847473,
"loss": 1.9529,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -5.486642360687256,
"rewards/margins": 1.6755880117416382,
"rewards/rejected": -7.162230491638184,
"step": 290
},
{
"epoch": 0.644808743169399,
"grad_norm": 66.70134041294685,
"learning_rate": 3.3679439308082774e-07,
"logits/chosen": -1.1284725666046143,
"logits/rejected": -1.126910924911499,
"logps/chosen": -0.5425226092338562,
"logps/rejected": -0.7452016472816467,
"loss": 1.7199,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -5.425226211547852,
"rewards/margins": 2.026790142059326,
"rewards/rejected": -7.452017307281494,
"step": 295
},
{
"epoch": 0.6557377049180327,
"grad_norm": 71.88498470947849,
"learning_rate": 3.1885519212446716e-07,
"logits/chosen": -1.134820580482483,
"logits/rejected": -1.113407850265503,
"logps/chosen": -0.5564647316932678,
"logps/rejected": -0.7468981146812439,
"loss": 1.8137,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -5.564647197723389,
"rewards/margins": 1.9043340682983398,
"rewards/rejected": -7.4689812660217285,
"step": 300
},
{
"epoch": 0.6666666666666666,
"grad_norm": 76.90950112576293,
"learning_rate": 3.0118055418614295e-07,
"logits/chosen": -1.1550134420394897,
"logits/rejected": -1.0957601070404053,
"logps/chosen": -0.5463107228279114,
"logps/rejected": -0.7333046793937683,
"loss": 1.942,
"rewards/accuracies": 0.8125,
"rewards/chosen": -5.463107109069824,
"rewards/margins": 1.8699395656585693,
"rewards/rejected": -7.333047389984131,
"step": 305
},
{
"epoch": 0.6775956284153005,
"grad_norm": 107.13681953131237,
"learning_rate": 2.83796293174686e-07,
"logits/chosen": -1.0895421504974365,
"logits/rejected": -1.0931763648986816,
"logps/chosen": -0.5538777112960815,
"logps/rejected": -0.7688311338424683,
"loss": 2.0538,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -5.538776874542236,
"rewards/margins": 2.149533748626709,
"rewards/rejected": -7.688311576843262,
"step": 310
},
{
"epoch": 0.6885245901639344,
"grad_norm": 66.34265535656564,
"learning_rate": 2.6672779890178046e-07,
"logits/chosen": -1.1496803760528564,
"logits/rejected": -1.1489367485046387,
"logps/chosen": -0.5764937400817871,
"logps/rejected": -0.706534206867218,
"loss": 1.9103,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -5.764937400817871,
"rewards/margins": 1.3004045486450195,
"rewards/rejected": -7.065341949462891,
"step": 315
},
{
"epoch": 0.6994535519125683,
"grad_norm": 56.65585571674297,
"learning_rate": 2.500000000000001e-07,
"logits/chosen": -1.204594612121582,
"logits/rejected": -1.1554490327835083,
"logps/chosen": -0.556768000125885,
"logps/rejected": -0.7628398537635803,
"loss": 1.8878,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -5.567679405212402,
"rewards/margins": 2.060718536376953,
"rewards/rejected": -7.628398895263672,
"step": 320
},
{
"epoch": 0.7103825136612022,
"grad_norm": 82.81929805474688,
"learning_rate": 2.3363732751439923e-07,
"logits/chosen": -1.1415125131607056,
"logits/rejected": -1.125905990600586,
"logps/chosen": -0.542820930480957,
"logps/rejected": -0.7379633188247681,
"loss": 1.8941,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": -5.42820930480957,
"rewards/margins": 1.9514230489730835,
"rewards/rejected": -7.379631996154785,
"step": 325
},
{
"epoch": 0.7213114754098361,
"grad_norm": 71.70622872848305,
"learning_rate": 2.1766367922083283e-07,
"logits/chosen": -1.0906676054000854,
"logits/rejected": -1.0594781637191772,
"logps/chosen": -0.5101396441459656,
"logps/rejected": -0.7594174146652222,
"loss": 1.8536,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -5.101396083831787,
"rewards/margins": 2.4927775859832764,
"rewards/rejected": -7.594174385070801,
"step": 330
},
{
"epoch": 0.73224043715847,
"grad_norm": 80.29511185962221,
"learning_rate": 2.021023847231202e-07,
"logits/chosen": -1.0921766757965088,
"logits/rejected": -1.0546457767486572,
"logps/chosen": -0.5786954760551453,
"logps/rejected": -0.775272011756897,
"loss": 1.859,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -5.7869553565979,
"rewards/margins": 1.9657647609710693,
"rewards/rejected": -7.752719879150391,
"step": 335
},
{
"epoch": 0.7431693989071039,
"grad_norm": 79.83892973093212,
"learning_rate": 1.869761713800254e-07,
"logits/chosen": -1.1134289503097534,
"logits/rejected": -1.0689969062805176,
"logps/chosen": -0.5616486668586731,
"logps/rejected": -0.7319185137748718,
"loss": 1.9172,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -5.6164870262146,
"rewards/margins": 1.7026973962783813,
"rewards/rejected": -7.319184303283691,
"step": 340
},
{
"epoch": 0.7540983606557377,
"grad_norm": 79.67263653958703,
"learning_rate": 1.7230713111182164e-07,
"logits/chosen": -1.1623106002807617,
"logits/rejected": -1.1600934267044067,
"logps/chosen": -0.5594975352287292,
"logps/rejected": -0.7684382200241089,
"loss": 1.9557,
"rewards/accuracies": 0.8125,
"rewards/chosen": -5.594975471496582,
"rewards/margins": 2.089407444000244,
"rewards/rejected": -7.684383392333984,
"step": 345
},
{
"epoch": 0.7650273224043715,
"grad_norm": 90.3235756484309,
"learning_rate": 1.5811668813491696e-07,
"logits/chosen": -1.1584365367889404,
"logits/rejected": -1.1381675004959106,
"logps/chosen": -0.5266227126121521,
"logps/rejected": -0.6895478367805481,
"loss": 1.8798,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -5.266226768493652,
"rewards/margins": 1.6292508840560913,
"rewards/rejected": -6.895478248596191,
"step": 350
},
{
"epoch": 0.7759562841530054,
"grad_norm": 65.15244365555583,
"learning_rate": 1.4442556767166369e-07,
"logits/chosen": -1.1325640678405762,
"logits/rejected": -1.10225510597229,
"logps/chosen": -0.539125382900238,
"logps/rejected": -0.7123144865036011,
"loss": 1.8691,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -5.391254425048828,
"rewards/margins": 1.731890082359314,
"rewards/rejected": -7.123144626617432,
"step": 355
},
{
"epoch": 0.7868852459016393,
"grad_norm": 74.4651421428626,
"learning_rate": 1.312537656810549e-07,
"logits/chosen": -1.0851835012435913,
"logits/rejected": -1.0858454704284668,
"logps/chosen": -0.5476486682891846,
"logps/rejected": -0.730248749256134,
"loss": 1.9346,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": -5.476487159729004,
"rewards/margins": 1.8260005712509155,
"rewards/rejected": -7.302487373352051,
"step": 360
},
{
"epoch": 0.7978142076502732,
"grad_norm": 153.86057279690542,
"learning_rate": 1.1862051965451214e-07,
"logits/chosen": -1.166272521018982,
"logits/rejected": -1.165235161781311,
"logps/chosen": -0.5564864873886108,
"logps/rejected": -0.749941349029541,
"loss": 1.8628,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -5.5648651123046875,
"rewards/margins": 1.934548020362854,
"rewards/rejected": -7.49941349029541,
"step": 365
},
{
"epoch": 0.8087431693989071,
"grad_norm": 77.62021036106599,
"learning_rate": 1.0654428051942138e-07,
"logits/chosen": -1.1702772378921509,
"logits/rejected": -1.1325637102127075,
"logps/chosen": -0.5678818225860596,
"logps/rejected": -0.7951611876487732,
"loss": 1.9704,
"rewards/accuracies": 0.78125,
"rewards/chosen": -5.678818225860596,
"rewards/margins": 2.272794246673584,
"rewards/rejected": -7.951612949371338,
"step": 370
},
{
"epoch": 0.819672131147541,
"grad_norm": 63.25811531390013,
"learning_rate": 9.504268569144763e-08,
"logits/chosen": -1.2080334424972534,
"logits/rejected": -1.1406437158584595,
"logps/chosen": -0.5438567399978638,
"logps/rejected": -0.7369149923324585,
"loss": 1.892,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -5.4385666847229,
"rewards/margins": 1.9305833578109741,
"rewards/rejected": -7.369150638580322,
"step": 375
},
{
"epoch": 0.8306010928961749,
"grad_norm": 62.759245925109596,
"learning_rate": 8.413253331499049e-08,
"logits/chosen": -1.088577389717102,
"logits/rejected": -1.1100099086761475,
"logps/chosen": -0.56348717212677,
"logps/rejected": -0.7516815066337585,
"loss": 1.8378,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -5.634871959686279,
"rewards/margins": 1.8819434642791748,
"rewards/rejected": -7.516815185546875,
"step": 380
},
{
"epoch": 0.8415300546448088,
"grad_norm": 66.42055041642078,
"learning_rate": 7.382975772939865e-08,
"logits/chosen": -1.182722806930542,
"logits/rejected": -1.1655415296554565,
"logps/chosen": -0.599307656288147,
"logps/rejected": -0.7887434959411621,
"loss": 1.9674,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -5.993075847625732,
"rewards/margins": 1.8943592309951782,
"rewards/rejected": -7.887434959411621,
"step": 385
},
{
"epoch": 0.8524590163934426,
"grad_norm": 104.21611140368309,
"learning_rate": 6.414940619677734e-08,
"logits/chosen": -1.171216607093811,
"logits/rejected": -1.1473820209503174,
"logps/chosen": -0.5480982065200806,
"logps/rejected": -0.76609867811203,
"loss": 1.9188,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": -5.480981349945068,
"rewards/margins": 2.1800050735473633,
"rewards/rejected": -7.660986423492432,
"step": 390
},
{
"epoch": 0.8633879781420765,
"grad_norm": 78.27102698318478,
"learning_rate": 5.5105616925376296e-08,
"logits/chosen": -1.1506216526031494,
"logits/rejected": -1.1313217878341675,
"logps/chosen": -0.5556925535202026,
"logps/rejected": -0.706631064414978,
"loss": 1.8449,
"rewards/accuracies": 0.768750011920929,
"rewards/chosen": -5.556925296783447,
"rewards/margins": 1.509385347366333,
"rewards/rejected": -7.066309928894043,
"step": 395
},
{
"epoch": 0.8743169398907104,
"grad_norm": 76.30561413947063,
"learning_rate": 4.6711598420656976e-08,
"logits/chosen": -1.0807750225067139,
"logits/rejected": -1.0536285638809204,
"logps/chosen": -0.5846288800239563,
"logps/rejected": -0.7902191281318665,
"loss": 1.8354,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": -5.846289157867432,
"rewards/margins": 2.0559017658233643,
"rewards/rejected": -7.902191162109375,
"step": 400
},
{
"epoch": 0.8743169398907104,
"eval_logits/chosen": -1.3447647094726562,
"eval_logits/rejected": -1.3069871664047241,
"eval_logps/chosen": -0.5437440872192383,
"eval_logps/rejected": -0.7271364331245422,
"eval_loss": 1.881577491760254,
"eval_rewards/accuracies": 0.7710843086242676,
"eval_rewards/chosen": -5.437441349029541,
"eval_rewards/margins": 1.83392333984375,
"eval_rewards/rejected": -7.271363735198975,
"eval_runtime": 37.6564,
"eval_samples_per_second": 35.001,
"eval_steps_per_second": 2.204,
"step": 400
},
{
"epoch": 0.8852459016393442,
"grad_norm": 66.16858739679557,
"learning_rate": 3.897961019419516e-08,
"logits/chosen": -1.1171514987945557,
"logits/rejected": -1.051194190979004,
"logps/chosen": -0.5250142812728882,
"logps/rejected": -0.6711951494216919,
"loss": 1.793,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -5.250143527984619,
"rewards/margins": 1.461808204650879,
"rewards/rejected": -6.71195125579834,
"step": 405
},
{
"epoch": 0.8961748633879781,
"grad_norm": 77.0310414767933,
"learning_rate": 3.192094485859526e-08,
"logits/chosen": -1.1234816312789917,
"logits/rejected": -1.144008994102478,
"logps/chosen": -0.5635503530502319,
"logps/rejected": -0.7934846878051758,
"loss": 1.8728,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": -5.63550329208374,
"rewards/margins": 2.2993433475494385,
"rewards/rejected": -7.934847354888916,
"step": 410
},
{
"epoch": 0.907103825136612,
"grad_norm": 62.5404158274799,
"learning_rate": 2.5545911634565265e-08,
"logits/chosen": -1.1487548351287842,
"logits/rejected": -1.1474642753601074,
"logps/chosen": -0.570233941078186,
"logps/rejected": -0.7942467927932739,
"loss": 1.8971,
"rewards/accuracies": 0.824999988079071,
"rewards/chosen": -5.702339172363281,
"rewards/margins": 2.2401280403137207,
"rewards/rejected": -7.94246768951416,
"step": 415
},
{
"epoch": 0.9180327868852459,
"grad_norm": 81.2075081960254,
"learning_rate": 1.9863821294241522e-08,
"logits/chosen": -1.150827407836914,
"logits/rejected": -1.1209652423858643,
"logps/chosen": -0.5287235379219055,
"logps/rejected": -0.7498638033866882,
"loss": 1.7791,
"rewards/accuracies": 0.8125,
"rewards/chosen": -5.287235260009766,
"rewards/margins": 2.211402177810669,
"rewards/rejected": -7.498638153076172,
"step": 420
},
{
"epoch": 0.9289617486338798,
"grad_norm": 74.25861489169881,
"learning_rate": 1.4882972562753615e-08,
"logits/chosen": -1.1429412364959717,
"logits/rejected": -1.1335225105285645,
"logps/chosen": -0.5883482098579407,
"logps/rejected": -0.8031044006347656,
"loss": 1.918,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -5.883482933044434,
"rewards/margins": 2.1475605964660645,
"rewards/rejected": -8.03104305267334,
"step": 425
},
{
"epoch": 0.9398907103825137,
"grad_norm": 90.08434920268098,
"learning_rate": 1.0610639997888915e-08,
"logits/chosen": -1.0862069129943848,
"logits/rejected": -1.086482286453247,
"logps/chosen": -0.5331538915634155,
"logps/rejected": -0.73698890209198,
"loss": 1.7152,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": -5.331539154052734,
"rewards/margins": 2.0383496284484863,
"rewards/rejected": -7.3698883056640625,
"step": 430
},
{
"epoch": 0.9508196721311475,
"grad_norm": 88.78145235758741,
"learning_rate": 7.053063365559997e-09,
"logits/chosen": -1.1540377140045166,
"logits/rejected": -1.1776275634765625,
"logps/chosen": -0.5493609309196472,
"logps/rejected": -0.7795323729515076,
"loss": 1.8391,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -5.493609428405762,
"rewards/margins": 2.3017146587371826,
"rewards/rejected": -7.795323848724365,
"step": 435
},
{
"epoch": 0.9617486338797814,
"grad_norm": 76.77833268976481,
"learning_rate": 4.215438526591064e-09,
"logits/chosen": -1.091371774673462,
"logits/rejected": -1.0515260696411133,
"logps/chosen": -0.5720237493515015,
"logps/rejected": -0.7283024787902832,
"loss": 1.8312,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -5.720236778259277,
"rewards/margins": 1.5627872943878174,
"rewards/rejected": -7.283023834228516,
"step": 440
},
{
"epoch": 0.9726775956284153,
"grad_norm": 85.67571762449911,
"learning_rate": 2.1019098481337426e-09,
"logits/chosen": -1.1658837795257568,
"logits/rejected": -1.1402862071990967,
"logps/chosen": -0.5455365180969238,
"logps/rejected": -0.7272461652755737,
"loss": 1.8014,
"rewards/accuracies": 0.8187500238418579,
"rewards/chosen": -5.4553656578063965,
"rewards/margins": 1.8170963525772095,
"rewards/rejected": -7.272461891174316,
"step": 445
},
{
"epoch": 0.9836065573770492,
"grad_norm": 87.49876973882085,
"learning_rate": 7.155641507955445e-10,
"logits/chosen": -1.0767357349395752,
"logits/rejected": -1.0696969032287598,
"logps/chosen": -0.5918444395065308,
"logps/rejected": -0.7698336839675903,
"loss": 1.9461,
"rewards/accuracies": 0.78125,
"rewards/chosen": -5.918444633483887,
"rewards/margins": 1.7798923254013062,
"rewards/rejected": -7.698336601257324,
"step": 450
},
{
"epoch": 0.994535519125683,
"grad_norm": 83.15276885308347,
"learning_rate": 5.842620032053824e-11,
"logits/chosen": -1.0845460891723633,
"logits/rejected": -1.0800822973251343,
"logps/chosen": -0.5885196328163147,
"logps/rejected": -0.7454328536987305,
"loss": 1.9848,
"rewards/accuracies": 0.7562500238418579,
"rewards/chosen": -5.885196208953857,
"rewards/margins": 1.5691322088241577,
"rewards/rejected": -7.4543280601501465,
"step": 455
},
{
"epoch": 0.9989071038251366,
"step": 457,
"total_flos": 0.0,
"train_loss": 2.2740620700744363,
"train_runtime": 5957.0658,
"train_samples_per_second": 9.83,
"train_steps_per_second": 0.077
}
],
"logging_steps": 5,
"max_steps": 457,
"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
}