zephyr-7b-dpo-full / trainer_state.json
RikkiXu's picture
Model save
c2d3fa9 verified
raw
history blame
No virus
20.3 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9985693848354793,
"eval_steps": 100,
"global_step": 349,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"grad_norm": 1483.079147156977,
"learning_rate": 5.7142857142857136e-09,
"logits/chosen": -4.490396976470947,
"logits/rejected": -4.787891387939453,
"logps/chosen": -300.56573486328125,
"logps/rejected": -263.39849853515625,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.03,
"grad_norm": 1678.9009350663757,
"learning_rate": 5.714285714285714e-08,
"logits/chosen": -4.292892932891846,
"logits/rejected": -4.4704790115356445,
"logps/chosen": -285.7367858886719,
"logps/rejected": -241.40879821777344,
"loss": 0.7232,
"rewards/accuracies": 0.4375,
"rewards/chosen": 0.03021715022623539,
"rewards/margins": 0.018014244735240936,
"rewards/rejected": 0.012202905490994453,
"step": 10
},
{
"epoch": 0.06,
"grad_norm": 1307.5912774162387,
"learning_rate": 1.1428571428571427e-07,
"logits/chosen": -4.212913990020752,
"logits/rejected": -4.504936695098877,
"logps/chosen": -312.25714111328125,
"logps/rejected": -247.5605926513672,
"loss": 0.6348,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": 0.25936827063560486,
"rewards/margins": 0.15551437437534332,
"rewards/rejected": 0.10385391861200333,
"step": 20
},
{
"epoch": 0.09,
"grad_norm": 877.5830541670848,
"learning_rate": 1.714285714285714e-07,
"logits/chosen": -4.158999919891357,
"logits/rejected": -4.378296852111816,
"logps/chosen": -320.1895446777344,
"logps/rejected": -273.6168518066406,
"loss": 0.4394,
"rewards/accuracies": 0.793749988079071,
"rewards/chosen": 1.6762861013412476,
"rewards/margins": 1.1370527744293213,
"rewards/rejected": 0.5392333269119263,
"step": 30
},
{
"epoch": 0.11,
"grad_norm": 1094.7268353859254,
"learning_rate": 1.9987489925657699e-07,
"logits/chosen": -4.327412128448486,
"logits/rejected": -4.590438365936279,
"logps/chosen": -272.52813720703125,
"logps/rejected": -226.4479522705078,
"loss": 0.3519,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": 3.0370898246765137,
"rewards/margins": 2.1227898597717285,
"rewards/rejected": 0.9143003225326538,
"step": 40
},
{
"epoch": 0.14,
"grad_norm": 1142.545986046605,
"learning_rate": 1.98875970549573e-07,
"logits/chosen": -4.374251365661621,
"logits/rejected": -4.627996921539307,
"logps/chosen": -264.49615478515625,
"logps/rejected": -226.27197265625,
"loss": 0.3689,
"rewards/accuracies": 0.831250011920929,
"rewards/chosen": 4.1793413162231445,
"rewards/margins": 2.736725091934204,
"rewards/rejected": 1.4426158666610718,
"step": 50
},
{
"epoch": 0.17,
"grad_norm": 635.1242404757879,
"learning_rate": 1.968881042201029e-07,
"logits/chosen": -4.35978889465332,
"logits/rejected": -4.594507694244385,
"logps/chosen": -296.86907958984375,
"logps/rejected": -251.1016845703125,
"loss": 0.3246,
"rewards/accuracies": 0.8062499761581421,
"rewards/chosen": 4.9698944091796875,
"rewards/margins": 3.8340022563934326,
"rewards/rejected": 1.1358922719955444,
"step": 60
},
{
"epoch": 0.2,
"grad_norm": 871.5895225513464,
"learning_rate": 1.9393118250841894e-07,
"logits/chosen": -4.349625587463379,
"logits/rejected": -4.586968421936035,
"logps/chosen": -285.0860595703125,
"logps/rejected": -241.4550018310547,
"loss": 0.3259,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": 4.21413516998291,
"rewards/margins": 3.7849984169006348,
"rewards/rejected": 0.42913690209388733,
"step": 70
},
{
"epoch": 0.23,
"grad_norm": 869.2800598030351,
"learning_rate": 1.900347799523094e-07,
"logits/chosen": -4.277868747711182,
"logits/rejected": -4.535857677459717,
"logps/chosen": -318.2683410644531,
"logps/rejected": -272.18206787109375,
"loss": 0.2501,
"rewards/accuracies": 0.9375,
"rewards/chosen": 4.3253655433654785,
"rewards/margins": 4.170114040374756,
"rewards/rejected": 0.15525080263614655,
"step": 80
},
{
"epoch": 0.26,
"grad_norm": 616.9171514391553,
"learning_rate": 1.8523786758850436e-07,
"logits/chosen": -4.401219844818115,
"logits/rejected": -4.6477251052856445,
"logps/chosen": -280.7421569824219,
"logps/rejected": -236.2926025390625,
"loss": 0.2859,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": 3.923023223876953,
"rewards/margins": 4.055254936218262,
"rewards/rejected": -0.13223108649253845,
"step": 90
},
{
"epoch": 0.29,
"grad_norm": 735.4670744748103,
"learning_rate": 1.795884231721841e-07,
"logits/chosen": -4.318561553955078,
"logits/rejected": -4.602439880371094,
"logps/chosen": -267.45794677734375,
"logps/rejected": -232.0417938232422,
"loss": 0.3114,
"rewards/accuracies": 0.875,
"rewards/chosen": 4.1319708824157715,
"rewards/margins": 4.702456951141357,
"rewards/rejected": -0.5704857110977173,
"step": 100
},
{
"epoch": 0.29,
"eval_logits/chosen": -3.226163148880005,
"eval_logits/rejected": -3.226163148880005,
"eval_logps/chosen": -161.294677734375,
"eval_logps/rejected": -161.294677734375,
"eval_loss": 0.6931471824645996,
"eval_rewards/accuracies": 0.0,
"eval_rewards/chosen": -2.8588294982910156,
"eval_rewards/margins": 0.0,
"eval_rewards/rejected": -2.8588294982910156,
"eval_runtime": 1.5114,
"eval_samples_per_second": 0.662,
"eval_steps_per_second": 0.662,
"step": 100
},
{
"epoch": 0.31,
"grad_norm": 817.8641381018862,
"learning_rate": 1.7314295131309637e-07,
"logits/chosen": -4.324714660644531,
"logits/rejected": -4.614729404449463,
"logps/chosen": -293.42364501953125,
"logps/rejected": -245.01620483398438,
"loss": 0.3079,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": 4.0577192306518555,
"rewards/margins": 4.518521308898926,
"rewards/rejected": -0.4608024060726166,
"step": 110
},
{
"epoch": 0.34,
"grad_norm": 926.642960527248,
"learning_rate": 1.6596591832778466e-07,
"logits/chosen": -4.334780216217041,
"logits/rejected": -4.567566394805908,
"logps/chosen": -298.56976318359375,
"logps/rejected": -266.8869323730469,
"loss": 0.2848,
"rewards/accuracies": 0.84375,
"rewards/chosen": 3.83539080619812,
"rewards/margins": 4.040234088897705,
"rewards/rejected": -0.20484237372875214,
"step": 120
},
{
"epoch": 0.37,
"grad_norm": 675.9811999128359,
"learning_rate": 1.5812910746042256e-07,
"logits/chosen": -4.3064866065979,
"logits/rejected": -4.553826332092285,
"logps/chosen": -286.5328063964844,
"logps/rejected": -242.8630828857422,
"loss": 0.2782,
"rewards/accuracies": 0.893750011920929,
"rewards/chosen": 3.7822985649108887,
"rewards/margins": 4.371214389801025,
"rewards/rejected": -0.5889158248901367,
"step": 130
},
{
"epoch": 0.4,
"grad_norm": 1048.4484509272795,
"learning_rate": 1.4971090092120542e-07,
"logits/chosen": -4.304565906524658,
"logits/rejected": -4.545400142669678,
"logps/chosen": -294.17095947265625,
"logps/rejected": -243.83935546875,
"loss": 0.269,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 3.8339390754699707,
"rewards/margins": 4.558647155761719,
"rewards/rejected": -0.7247086763381958,
"step": 140
},
{
"epoch": 0.43,
"grad_norm": 844.0558457976633,
"learning_rate": 1.4079549592320782e-07,
"logits/chosen": -4.313709735870361,
"logits/rejected": -4.621119499206543,
"logps/chosen": -292.96661376953125,
"logps/rejected": -248.0418701171875,
"loss": 0.2727,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": 3.8533847332000732,
"rewards/margins": 4.262558937072754,
"rewards/rejected": -0.4091736674308777,
"step": 150
},
{
"epoch": 0.46,
"grad_norm": 684.2161612155076,
"learning_rate": 1.3147206255874882e-07,
"logits/chosen": -4.248518943786621,
"logits/rejected": -4.44361686706543,
"logps/chosen": -309.7388610839844,
"logps/rejected": -272.90252685546875,
"loss": 0.2961,
"rewards/accuracies": 0.90625,
"rewards/chosen": 4.352166652679443,
"rewards/margins": 4.611546516418457,
"rewards/rejected": -0.2593800127506256,
"step": 160
},
{
"epoch": 0.49,
"grad_norm": 795.319436926025,
"learning_rate": 1.2183385193801653e-07,
"logits/chosen": -4.289888381958008,
"logits/rejected": -4.489100933074951,
"logps/chosen": -272.3153991699219,
"logps/rejected": -237.64114379882812,
"loss": 0.2547,
"rewards/accuracies": 0.918749988079071,
"rewards/chosen": 4.135977745056152,
"rewards/margins": 4.544943809509277,
"rewards/rejected": -0.40896692872047424,
"step": 170
},
{
"epoch": 0.52,
"grad_norm": 917.2913745554154,
"learning_rate": 1.1197726351017051e-07,
"logits/chosen": -4.37355375289917,
"logits/rejected": -4.5346221923828125,
"logps/chosen": -261.6694641113281,
"logps/rejected": -230.89599609375,
"loss": 0.3104,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 3.4763107299804688,
"rewards/margins": 3.877528429031372,
"rewards/rejected": -0.4012181758880615,
"step": 180
},
{
"epoch": 0.54,
"grad_norm": 880.6071936953809,
"learning_rate": 1.0200088089538943e-07,
"logits/chosen": -4.33815860748291,
"logits/rejected": -4.5543646812438965,
"logps/chosen": -288.26873779296875,
"logps/rejected": -253.7147216796875,
"loss": 0.3129,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": 4.4660868644714355,
"rewards/margins": 4.566544055938721,
"rewards/rejected": -0.10045762360095978,
"step": 190
},
{
"epoch": 0.57,
"grad_norm": 781.3066688232569,
"learning_rate": 9.20044858712785e-08,
"logits/chosen": -4.292983055114746,
"logits/rejected": -4.566291332244873,
"logps/chosen": -315.58251953125,
"logps/rejected": -270.0286560058594,
"loss": 0.2741,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 4.502341270446777,
"rewards/margins": 4.4218339920043945,
"rewards/rejected": 0.08050797879695892,
"step": 200
},
{
"epoch": 0.57,
"eval_logits/chosen": -3.2033309936523438,
"eval_logits/rejected": -3.2033309936523438,
"eval_logps/chosen": -160.92892456054688,
"eval_logps/rejected": -160.92892456054688,
"eval_loss": 0.6931471824645996,
"eval_rewards/accuracies": 0.0,
"eval_rewards/chosen": -2.675952911376953,
"eval_rewards/margins": 0.0,
"eval_rewards/rejected": -2.675952911376953,
"eval_runtime": 1.48,
"eval_samples_per_second": 0.676,
"eval_steps_per_second": 0.676,
"step": 200
},
{
"epoch": 0.6,
"grad_norm": 1021.5696401893487,
"learning_rate": 8.208806037554645e-08,
"logits/chosen": -4.327147483825684,
"logits/rejected": -4.6312479972839355,
"logps/chosen": -292.85919189453125,
"logps/rejected": -236.7397918701172,
"loss": 0.2345,
"rewards/accuracies": 0.875,
"rewards/chosen": 4.086556434631348,
"rewards/margins": 4.589372634887695,
"rewards/rejected": -0.5028160810470581,
"step": 210
},
{
"epoch": 0.63,
"grad_norm": 720.6830161967047,
"learning_rate": 7.23507865067214e-08,
"logits/chosen": -4.357525825500488,
"logits/rejected": -4.7082133293151855,
"logps/chosen": -298.2110900878906,
"logps/rejected": -237.28903198242188,
"loss": 0.241,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 3.84515118598938,
"rewards/margins": 4.655531406402588,
"rewards/rejected": -0.8103805780410767,
"step": 220
},
{
"epoch": 0.66,
"grad_norm": 784.9312094778309,
"learning_rate": 6.289005452469778e-08,
"logits/chosen": -4.388636112213135,
"logits/rejected": -4.748753070831299,
"logps/chosen": -288.0061340332031,
"logps/rejected": -236.6077880859375,
"loss": 0.268,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": 3.5908846855163574,
"rewards/margins": 4.1128339767456055,
"rewards/rejected": -0.5219489932060242,
"step": 230
},
{
"epoch": 0.69,
"grad_norm": 765.61870702253,
"learning_rate": 5.38004887728938e-08,
"logits/chosen": -4.234927177429199,
"logits/rejected": -4.550938606262207,
"logps/chosen": -283.8558654785156,
"logps/rejected": -242.86538696289062,
"loss": 0.2908,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 3.562377452850342,
"rewards/margins": 4.074954032897949,
"rewards/rejected": -0.5125768780708313,
"step": 240
},
{
"epoch": 0.72,
"grad_norm": 787.5249084258048,
"learning_rate": 4.517300126455066e-08,
"logits/chosen": -4.290322303771973,
"logits/rejected": -4.522026538848877,
"logps/chosen": -292.47442626953125,
"logps/rejected": -257.0675354003906,
"loss": 0.2635,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": 3.95862078666687,
"rewards/margins": 4.515480995178223,
"rewards/rejected": -0.5568601489067078,
"step": 250
},
{
"epoch": 0.74,
"grad_norm": 894.7487355536125,
"learning_rate": 3.70938823990135e-08,
"logits/chosen": -4.367636680603027,
"logits/rejected": -4.524864196777344,
"logps/chosen": -293.8681640625,
"logps/rejected": -266.19622802734375,
"loss": 0.2999,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 3.311814069747925,
"rewards/margins": 3.883411407470703,
"rewards/rejected": -0.5715969800949097,
"step": 260
},
{
"epoch": 0.77,
"grad_norm": 582.5138549963486,
"learning_rate": 2.9643937902467277e-08,
"logits/chosen": -4.23429012298584,
"logits/rejected": -4.521827697753906,
"logps/chosen": -280.5379333496094,
"logps/rejected": -241.7247314453125,
"loss": 0.2388,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": 3.911522626876831,
"rewards/margins": 4.728894233703613,
"rewards/rejected": -0.8173705339431763,
"step": 270
},
{
"epoch": 0.8,
"grad_norm": 810.3954826895692,
"learning_rate": 2.289768062527362e-08,
"logits/chosen": -4.273584842681885,
"logits/rejected": -4.6050519943237305,
"logps/chosen": -283.619384765625,
"logps/rejected": -233.9824981689453,
"loss": 0.2895,
"rewards/accuracies": 0.8812500238418579,
"rewards/chosen": 3.576643466949463,
"rewards/margins": 3.991179943084717,
"rewards/rejected": -0.41453617811203003,
"step": 280
},
{
"epoch": 0.83,
"grad_norm": 730.6997584190797,
"learning_rate": 1.6922585279389035e-08,
"logits/chosen": -4.406073093414307,
"logits/rejected": -4.6901140213012695,
"logps/chosen": -294.35699462890625,
"logps/rejected": -233.74594116210938,
"loss": 0.2669,
"rewards/accuracies": 0.90625,
"rewards/chosen": 3.295900344848633,
"rewards/margins": 3.856853485107422,
"rewards/rejected": -0.5609533786773682,
"step": 290
},
{
"epoch": 0.86,
"grad_norm": 1254.3257141110294,
"learning_rate": 1.1778413569831725e-08,
"logits/chosen": -4.323153495788574,
"logits/rejected": -4.481599807739258,
"logps/chosen": -291.29742431640625,
"logps/rejected": -254.1814422607422,
"loss": 0.2695,
"rewards/accuracies": 0.856249988079071,
"rewards/chosen": 3.8485710620880127,
"rewards/margins": 3.8490569591522217,
"rewards/rejected": -0.0004864335060119629,
"step": 300
},
{
"epoch": 0.86,
"eval_logits/chosen": -3.199521064758301,
"eval_logits/rejected": -3.199521064758301,
"eval_logps/chosen": -159.83721923828125,
"eval_logps/rejected": -159.83721923828125,
"eval_loss": 0.6931471824645996,
"eval_rewards/accuracies": 0.0,
"eval_rewards/chosen": -2.1301021575927734,
"eval_rewards/margins": 0.0,
"eval_rewards/rejected": -2.1301021575927734,
"eval_runtime": 1.4681,
"eval_samples_per_second": 0.681,
"eval_steps_per_second": 0.681,
"step": 300
},
{
"epoch": 0.89,
"grad_norm": 1302.095010227008,
"learning_rate": 7.516616470096315e-09,
"logits/chosen": -4.3807525634765625,
"logits/rejected": -4.600883483886719,
"logps/chosen": -299.8600158691406,
"logps/rejected": -248.4993438720703,
"loss": 0.2545,
"rewards/accuracies": 0.875,
"rewards/chosen": 4.215323448181152,
"rewards/margins": 4.587640762329102,
"rewards/rejected": -0.3723169267177582,
"step": 310
},
{
"epoch": 0.92,
"grad_norm": 875.4939401121867,
"learning_rate": 4.179819619838454e-09,
"logits/chosen": -4.2180280685424805,
"logits/rejected": -4.570239067077637,
"logps/chosen": -286.0614013671875,
"logps/rejected": -234.3032684326172,
"loss": 0.269,
"rewards/accuracies": 0.8374999761581421,
"rewards/chosen": 3.9898929595947266,
"rewards/margins": 4.13907527923584,
"rewards/rejected": -0.1491830050945282,
"step": 320
},
{
"epoch": 0.94,
"grad_norm": 1010.1848642033616,
"learning_rate": 1.8013969917777483e-09,
"logits/chosen": -4.380679130554199,
"logits/rejected": -4.608093738555908,
"logps/chosen": -278.8153381347656,
"logps/rejected": -235.4047088623047,
"loss": 0.247,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": 3.914651870727539,
"rewards/margins": 4.33533239364624,
"rewards/rejected": -0.42068085074424744,
"step": 330
},
{
"epoch": 0.97,
"grad_norm": 702.698454631269,
"learning_rate": 4.051370919176289e-10,
"logits/chosen": -4.314477920532227,
"logits/rejected": -4.502932548522949,
"logps/chosen": -282.89666748046875,
"logps/rejected": -246.3631134033203,
"loss": 0.243,
"rewards/accuracies": 0.90625,
"rewards/chosen": 3.8934428691864014,
"rewards/margins": 4.308337688446045,
"rewards/rejected": -0.41489481925964355,
"step": 340
},
{
"epoch": 1.0,
"step": 349,
"total_flos": 0.0,
"train_loss": 0.30806411576476,
"train_runtime": 5299.0292,
"train_samples_per_second": 8.432,
"train_steps_per_second": 0.066
}
],
"logging_steps": 10,
"max_steps": 349,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 8,
"trial_name": null,
"trial_params": null
}