trustalign_llama2_7b / trainer_state.json
shanghong's picture
Upload folder using huggingface_hub
3c0a50b verified
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.6,
"eval_steps": 5,
"global_step": 130,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.024615384615384615,
"grad_norm": 51.137206373851456,
"learning_rate": 5.88235294117647e-08,
"logits/chosen": -0.5265347361564636,
"logits/rejected": -0.5229614973068237,
"logps/chosen": -40.1543083190918,
"logps/rejected": -40.267845153808594,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 2
},
{
"epoch": 0.04923076923076923,
"grad_norm": 53.426708668382766,
"learning_rate": 1.176470588235294e-07,
"logits/chosen": -0.5090410113334656,
"logits/rejected": -0.502047061920166,
"logps/chosen": -45.320037841796875,
"logps/rejected": -42.6551399230957,
"loss": 0.6918,
"rewards/accuracies": 0.453125,
"rewards/chosen": 0.014461740851402283,
"rewards/margins": 0.0072987377643585205,
"rewards/rejected": 0.007163003087043762,
"step": 4
},
{
"epoch": 0.06153846153846154,
"eval_logits/chosen": -0.49676236510276794,
"eval_logits/rejected": -0.4936961829662323,
"eval_logps/chosen": -45.20378875732422,
"eval_logps/rejected": -42.88367462158203,
"eval_loss": 0.6931462287902832,
"eval_rewards/accuracies": 0.4623287618160248,
"eval_rewards/chosen": 0.0008219562005251646,
"eval_rewards/margins": 0.0035426486283540726,
"eval_rewards/rejected": -0.002720692427828908,
"eval_runtime": 400.2137,
"eval_samples_per_second": 4.333,
"eval_steps_per_second": 0.182,
"step": 5
},
{
"epoch": 0.07384615384615385,
"grad_norm": 47.768106379014085,
"learning_rate": 1.764705882352941e-07,
"logits/chosen": -0.4896703064441681,
"logits/rejected": -0.48678484559059143,
"logps/chosen": -47.16059875488281,
"logps/rejected": -35.45697021484375,
"loss": 0.6958,
"rewards/accuracies": 0.484375,
"rewards/chosen": 0.007428005337715149,
"rewards/margins": 0.012338653206825256,
"rewards/rejected": -0.004910647869110107,
"step": 6
},
{
"epoch": 0.09846153846153846,
"grad_norm": 39.51012410260543,
"learning_rate": 2.352941176470588e-07,
"logits/chosen": -0.4996076822280884,
"logits/rejected": -0.49669766426086426,
"logps/chosen": -47.94187545776367,
"logps/rejected": -41.556766510009766,
"loss": 0.6914,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.018062740564346313,
"rewards/margins": 0.025705993175506592,
"rewards/rejected": -0.007643252611160278,
"step": 8
},
{
"epoch": 0.12307692307692308,
"grad_norm": 39.08254813329663,
"learning_rate": 2.941176470588235e-07,
"logits/chosen": -0.49092355370521545,
"logits/rejected": -0.48874035477638245,
"logps/chosen": -50.65516662597656,
"logps/rejected": -30.806886672973633,
"loss": 0.6842,
"rewards/accuracies": 0.515625,
"rewards/chosen": 0.016676604747772217,
"rewards/margins": 0.011630743741989136,
"rewards/rejected": 0.005045861005783081,
"step": 10
},
{
"epoch": 0.12307692307692308,
"eval_logits/chosen": -0.49970191717147827,
"eval_logits/rejected": -0.4963047206401825,
"eval_logps/chosen": -45.12245178222656,
"eval_logps/rejected": -42.913116455078125,
"eval_loss": 0.6740441918373108,
"eval_rewards/accuracies": 0.5410959124565125,
"eval_rewards/chosen": 0.041489191353321075,
"eval_rewards/margins": 0.058929312974214554,
"eval_rewards/rejected": -0.017440123483538628,
"eval_runtime": 398.3897,
"eval_samples_per_second": 4.353,
"eval_steps_per_second": 0.183,
"step": 10
},
{
"epoch": 0.1476923076923077,
"grad_norm": 38.09131172735928,
"learning_rate": 3.529411764705882e-07,
"logits/chosen": -0.48407822847366333,
"logits/rejected": -0.4791830778121948,
"logps/chosen": -45.49330139160156,
"logps/rejected": -42.8103141784668,
"loss": 0.6725,
"rewards/accuracies": 0.609375,
"rewards/chosen": 0.04486130177974701,
"rewards/margins": 0.06311255693435669,
"rewards/rejected": -0.01825125515460968,
"step": 12
},
{
"epoch": 0.1723076923076923,
"grad_norm": 35.41525379746354,
"learning_rate": 4.117647058823529e-07,
"logits/chosen": -0.5208038091659546,
"logits/rejected": -0.5175592303276062,
"logps/chosen": -40.418609619140625,
"logps/rejected": -34.53902816772461,
"loss": 0.6501,
"rewards/accuracies": 0.546875,
"rewards/chosen": 0.11583159863948822,
"rewards/margins": 0.07396885752677917,
"rewards/rejected": 0.041862741112709045,
"step": 14
},
{
"epoch": 0.18461538461538463,
"eval_logits/chosen": -0.4988827705383301,
"eval_logits/rejected": -0.4956616461277008,
"eval_logps/chosen": -44.772987365722656,
"eval_logps/rejected": -42.95915985107422,
"eval_loss": 0.6234598159790039,
"eval_rewards/accuracies": 0.6404109597206116,
"eval_rewards/chosen": 0.2162197083234787,
"eval_rewards/margins": 0.25668132305145264,
"eval_rewards/rejected": -0.040461596101522446,
"eval_runtime": 398.8168,
"eval_samples_per_second": 4.348,
"eval_steps_per_second": 0.183,
"step": 15
},
{
"epoch": 0.19692307692307692,
"grad_norm": 33.20243289262748,
"learning_rate": 4.705882352941176e-07,
"logits/chosen": -0.49464067816734314,
"logits/rejected": -0.4926251173019409,
"logps/chosen": -41.81456756591797,
"logps/rejected": -34.730552673339844,
"loss": 0.6139,
"rewards/accuracies": 0.734375,
"rewards/chosen": 0.22246630489826202,
"rewards/margins": 0.1789122372865677,
"rewards/rejected": 0.043554067611694336,
"step": 16
},
{
"epoch": 0.22153846153846155,
"grad_norm": 31.414401498072234,
"learning_rate": 4.99941324504621e-07,
"logits/chosen": -0.5083720684051514,
"logits/rejected": -0.5054486393928528,
"logps/chosen": -40.7899169921875,
"logps/rejected": -41.80813217163086,
"loss": 0.6038,
"rewards/accuracies": 0.71875,
"rewards/chosen": 0.22346775233745575,
"rewards/margins": 0.36676231026649475,
"rewards/rejected": -0.143294557929039,
"step": 18
},
{
"epoch": 0.24615384615384617,
"grad_norm": 26.94475877791274,
"learning_rate": 4.99472085783721e-07,
"logits/chosen": -0.5102497935295105,
"logits/rejected": -0.5054219961166382,
"logps/chosen": -38.036109924316406,
"logps/rejected": -41.01247787475586,
"loss": 0.5437,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.5671411752700806,
"rewards/margins": 0.6098884344100952,
"rewards/rejected": -0.04274718463420868,
"step": 20
},
{
"epoch": 0.24615384615384617,
"eval_logits/chosen": -0.4962932765483856,
"eval_logits/rejected": -0.49319401383399963,
"eval_logps/chosen": -43.80841827392578,
"eval_logps/rejected": -42.96743392944336,
"eval_loss": 0.5607677698135376,
"eval_rewards/accuracies": 0.6952054500579834,
"eval_rewards/chosen": 0.6985027194023132,
"eval_rewards/margins": 0.7430992722511292,
"eval_rewards/rejected": -0.04459657892584801,
"eval_runtime": 398.3125,
"eval_samples_per_second": 4.353,
"eval_steps_per_second": 0.183,
"step": 20
},
{
"epoch": 0.27076923076923076,
"grad_norm": 24.12367725777412,
"learning_rate": 4.985344892885899e-07,
"logits/chosen": -0.5035347938537598,
"logits/rejected": -0.5017133951187134,
"logps/chosen": -42.291107177734375,
"logps/rejected": -43.623016357421875,
"loss": 0.5756,
"rewards/accuracies": 0.734375,
"rewards/chosen": 0.6317887306213379,
"rewards/margins": 0.7202781438827515,
"rewards/rejected": -0.08848930895328522,
"step": 22
},
{
"epoch": 0.2953846153846154,
"grad_norm": 24.794972937394068,
"learning_rate": 4.971302952586796e-07,
"logits/chosen": -0.5023949146270752,
"logits/rejected": -0.49643266201019287,
"logps/chosen": -39.85929489135742,
"logps/rejected": -46.9812126159668,
"loss": 0.5219,
"rewards/accuracies": 0.84375,
"rewards/chosen": 0.8194392919540405,
"rewards/margins": 1.2944234609603882,
"rewards/rejected": -0.4749842882156372,
"step": 24
},
{
"epoch": 0.3076923076923077,
"eval_logits/chosen": -0.4914375841617584,
"eval_logits/rejected": -0.4883214831352234,
"eval_logps/chosen": -43.38737869262695,
"eval_logps/rejected": -43.16010284423828,
"eval_loss": 0.516257107257843,
"eval_rewards/accuracies": 0.7397260069847107,
"eval_rewards/chosen": 0.9090243577957153,
"eval_rewards/margins": 1.0499579906463623,
"eval_rewards/rejected": -0.14093351364135742,
"eval_runtime": 397.9655,
"eval_samples_per_second": 4.357,
"eval_steps_per_second": 0.183,
"step": 25
},
{
"epoch": 0.32,
"grad_norm": 25.854991033804026,
"learning_rate": 4.952621399215597e-07,
"logits/chosen": -0.49028468132019043,
"logits/rejected": -0.48677563667297363,
"logps/chosen": -39.38325500488281,
"logps/rejected": -30.141273498535156,
"loss": 0.5432,
"rewards/accuracies": 0.765625,
"rewards/chosen": 0.9173061847686768,
"rewards/margins": 0.6880545020103455,
"rewards/rejected": 0.2292517125606537,
"step": 26
},
{
"epoch": 0.3446153846153846,
"grad_norm": 22.546249636399228,
"learning_rate": 4.929335305436764e-07,
"logits/chosen": -0.4758816063404083,
"logits/rejected": -0.47193384170532227,
"logps/chosen": -37.941322326660156,
"logps/rejected": -36.9672966003418,
"loss": 0.4989,
"rewards/accuracies": 0.8125,
"rewards/chosen": 0.9205312132835388,
"rewards/margins": 0.96444171667099,
"rewards/rejected": -0.0439104288816452,
"step": 28
},
{
"epoch": 0.36923076923076925,
"grad_norm": 25.277496467379798,
"learning_rate": 4.901488388458247e-07,
"logits/chosen": -0.47848933935165405,
"logits/rejected": -0.4744369089603424,
"logps/chosen": -42.54041290283203,
"logps/rejected": -32.181236267089844,
"loss": 0.5002,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.949561595916748,
"rewards/margins": 0.7569789886474609,
"rewards/rejected": 0.19258266687393188,
"step": 30
},
{
"epoch": 0.36923076923076925,
"eval_logits/chosen": -0.48240911960601807,
"eval_logits/rejected": -0.47941771149635315,
"eval_logps/chosen": -42.97114944458008,
"eval_logps/rejected": -43.50387191772461,
"eval_loss": 0.4467584788799286,
"eval_rewards/accuracies": 0.75,
"eval_rewards/chosen": 1.1171396970748901,
"eval_rewards/margins": 1.429957389831543,
"eval_rewards/rejected": -0.3128177523612976,
"eval_runtime": 399.6945,
"eval_samples_per_second": 4.338,
"eval_steps_per_second": 0.183,
"step": 30
},
{
"epoch": 0.39384615384615385,
"grad_norm": 21.934080715012698,
"learning_rate": 4.869132927957006e-07,
"logits/chosen": -0.47030550241470337,
"logits/rejected": -0.4669092893600464,
"logps/chosen": -39.564884185791016,
"logps/rejected": -46.268707275390625,
"loss": 0.4561,
"rewards/accuracies": 0.71875,
"rewards/chosen": 1.1370558738708496,
"rewards/margins": 1.5885730981826782,
"rewards/rejected": -0.45151710510253906,
"step": 32
},
{
"epoch": 0.41846153846153844,
"grad_norm": 17.9530062267105,
"learning_rate": 4.832329667929376e-07,
"logits/chosen": -0.48381492495536804,
"logits/rejected": -0.47740432620048523,
"logps/chosen": -37.79485321044922,
"logps/rejected": -41.405555725097656,
"loss": 0.3974,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.0800038576126099,
"rewards/margins": 1.8046053647994995,
"rewards/rejected": -0.7246013879776001,
"step": 34
},
{
"epoch": 0.4307692307692308,
"eval_logits/chosen": -0.4790378212928772,
"eval_logits/rejected": -0.4759667217731476,
"eval_logps/chosen": -42.852088928222656,
"eval_logps/rejected": -43.939449310302734,
"eval_loss": 0.3961867392063141,
"eval_rewards/accuracies": 0.7602739930152893,
"eval_rewards/chosen": 1.1766709089279175,
"eval_rewards/margins": 1.7072778940200806,
"eval_rewards/rejected": -0.5306068062782288,
"eval_runtime": 402.2584,
"eval_samples_per_second": 4.311,
"eval_steps_per_second": 0.181,
"step": 35
},
{
"epoch": 0.4430769230769231,
"grad_norm": 18.01450637894319,
"learning_rate": 4.791147702650565e-07,
"logits/chosen": -0.4862886071205139,
"logits/rejected": -0.48092207312583923,
"logps/chosen": -40.004295349121094,
"logps/rejected": -39.387386322021484,
"loss": 0.4067,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.3134092092514038,
"rewards/margins": 1.8237063884735107,
"rewards/rejected": -0.5102972984313965,
"step": 36
},
{
"epoch": 0.4676923076923077,
"grad_norm": 18.54804560370934,
"learning_rate": 4.745664346957361e-07,
"logits/chosen": -0.46789872646331787,
"logits/rejected": -0.4679996967315674,
"logps/chosen": -42.09858322143555,
"logps/rejected": -31.129802703857422,
"loss": 0.3702,
"rewards/accuracies": 0.796875,
"rewards/chosen": 1.3026843070983887,
"rewards/margins": 1.3334376811981201,
"rewards/rejected": -0.030753374099731445,
"step": 38
},
{
"epoch": 0.49230769230769234,
"grad_norm": 13.439566896592863,
"learning_rate": 4.695964991097616e-07,
"logits/chosen": -0.449634313583374,
"logits/rejected": -0.4447551965713501,
"logps/chosen": -36.52521514892578,
"logps/rejected": -42.37124252319336,
"loss": 0.3514,
"rewards/accuracies": 0.859375,
"rewards/chosen": 1.1417609453201294,
"rewards/margins": 1.7916193008422852,
"rewards/rejected": -0.6498584747314453,
"step": 40
},
{
"epoch": 0.49230769230769234,
"eval_logits/chosen": -0.47975781559944153,
"eval_logits/rejected": -0.47623777389526367,
"eval_logps/chosen": -42.85760498046875,
"eval_logps/rejected": -44.426246643066406,
"eval_loss": 0.35519111156463623,
"eval_rewards/accuracies": 0.7876712083816528,
"eval_rewards/chosen": 1.1739110946655273,
"eval_rewards/margins": 1.9479175806045532,
"eval_rewards/rejected": -0.7740064859390259,
"eval_runtime": 398.0514,
"eval_samples_per_second": 4.356,
"eval_steps_per_second": 0.183,
"step": 40
},
{
"epoch": 0.5169230769230769,
"grad_norm": 14.688902217061264,
"learning_rate": 4.642142940418973e-07,
"logits/chosen": -0.5011200904846191,
"logits/rejected": -0.49806085228919983,
"logps/chosen": -37.53756332397461,
"logps/rejected": -43.31938171386719,
"loss": 0.3629,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.0989573001861572,
"rewards/margins": 1.9275836944580078,
"rewards/rejected": -0.8286263346672058,
"step": 42
},
{
"epoch": 0.5415384615384615,
"grad_norm": 14.193796961040897,
"learning_rate": 4.5842992401978256e-07,
"logits/chosen": -0.4911167323589325,
"logits/rejected": -0.4909789264202118,
"logps/chosen": -46.16688537597656,
"logps/rejected": -35.1140251159668,
"loss": 0.316,
"rewards/accuracies": 0.8125,
"rewards/chosen": 1.0063791275024414,
"rewards/margins": 1.4864487648010254,
"rewards/rejected": -0.48006969690322876,
"step": 44
},
{
"epoch": 0.5538461538461539,
"eval_logits/chosen": -0.472987562417984,
"eval_logits/rejected": -0.46926581859588623,
"eval_logps/chosen": -42.71507263183594,
"eval_logps/rejected": -45.037986755371094,
"eval_loss": 0.31506600975990295,
"eval_rewards/accuracies": 0.801369845867157,
"eval_rewards/chosen": 1.2451775074005127,
"eval_rewards/margins": 2.3250503540039062,
"eval_rewards/rejected": -1.0798726081848145,
"eval_runtime": 398.07,
"eval_samples_per_second": 4.356,
"eval_steps_per_second": 0.183,
"step": 45
},
{
"epoch": 0.5661538461538461,
"grad_norm": 11.94835713175047,
"learning_rate": 4.5225424859373684e-07,
"logits/chosen": -0.48655006289482117,
"logits/rejected": -0.4802987575531006,
"logps/chosen": -45.382694244384766,
"logps/rejected": -34.198219299316406,
"loss": 0.2964,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.3039376735687256,
"rewards/margins": 1.7199229001998901,
"rewards/rejected": -0.4159852862358093,
"step": 46
},
{
"epoch": 0.5907692307692308,
"grad_norm": 11.665118422994077,
"learning_rate": 4.456988619490889e-07,
"logits/chosen": -0.4525223672389984,
"logits/rejected": -0.4487767517566681,
"logps/chosen": -42.471214294433594,
"logps/rejected": -37.61225128173828,
"loss": 0.2878,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.1848539113998413,
"rewards/margins": 1.884321928024292,
"rewards/rejected": -0.6994677782058716,
"step": 48
},
{
"epoch": 0.6153846153846154,
"grad_norm": 11.208056217338184,
"learning_rate": 4.3877607113930516e-07,
"logits/chosen": -0.4693542420864105,
"logits/rejected": -0.46645450592041016,
"logps/chosen": -38.541446685791016,
"logps/rejected": -28.433433532714844,
"loss": 0.2786,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.3120430707931519,
"rewards/margins": 1.601030945777893,
"rewards/rejected": -0.2889878749847412,
"step": 50
},
{
"epoch": 0.6153846153846154,
"eval_logits/chosen": -0.46347275376319885,
"eval_logits/rejected": -0.45981672406196594,
"eval_logps/chosen": -42.53354263305664,
"eval_logps/rejected": -45.56825256347656,
"eval_loss": 0.28788629174232483,
"eval_rewards/accuracies": 0.818493127822876,
"eval_rewards/chosen": 1.335942029953003,
"eval_rewards/margins": 2.6809515953063965,
"eval_rewards/rejected": -1.345009684562683,
"eval_runtime": 398.8881,
"eval_samples_per_second": 4.347,
"eval_steps_per_second": 0.183,
"step": 50
},
{
"epoch": 0.64,
"grad_norm": 12.527059053107184,
"learning_rate": 4.314988729807827e-07,
"logits/chosen": -0.46812835335731506,
"logits/rejected": -0.4670564532279968,
"logps/chosen": -39.73173904418945,
"logps/rejected": -31.038818359375,
"loss": 0.277,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.1246474981307983,
"rewards/margins": 1.668458342552185,
"rewards/rejected": -0.5438107252120972,
"step": 52
},
{
"epoch": 0.6646153846153846,
"grad_norm": 8.553726428740159,
"learning_rate": 4.238809296526846e-07,
"logits/chosen": -0.4466942548751831,
"logits/rejected": -0.4432934820652008,
"logps/chosen": -40.79918670654297,
"logps/rejected": -43.690181732177734,
"loss": 0.2475,
"rewards/accuracies": 0.8125,
"rewards/chosen": 1.295045256614685,
"rewards/margins": 2.5195388793945312,
"rewards/rejected": -1.2244938611984253,
"step": 54
},
{
"epoch": 0.676923076923077,
"eval_logits/chosen": -0.4640098512172699,
"eval_logits/rejected": -0.4600638151168823,
"eval_logps/chosen": -42.25896453857422,
"eval_logps/rejected": -45.97068786621094,
"eval_loss": 0.26564928889274597,
"eval_rewards/accuracies": 0.818493127822876,
"eval_rewards/chosen": 1.47323477268219,
"eval_rewards/margins": 3.019458293914795,
"eval_rewards/rejected": -1.5462236404418945,
"eval_runtime": 397.9539,
"eval_samples_per_second": 4.357,
"eval_steps_per_second": 0.183,
"step": 55
},
{
"epoch": 0.6892307692307692,
"grad_norm": 10.956104922328452,
"learning_rate": 4.159365430476261e-07,
"logits/chosen": -0.45851805806159973,
"logits/rejected": -0.4554572105407715,
"logps/chosen": -39.361473083496094,
"logps/rejected": -40.478187561035156,
"loss": 0.2571,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.2780680656433105,
"rewards/margins": 2.710892915725708,
"rewards/rejected": -1.4328248500823975,
"step": 56
},
{
"epoch": 0.7138461538461538,
"grad_norm": 8.998450490495424,
"learning_rate": 4.076806279213655e-07,
"logits/chosen": -0.44970858097076416,
"logits/rejected": -0.4487365484237671,
"logps/chosen": -39.6558723449707,
"logps/rejected": -30.797224044799805,
"loss": 0.2377,
"rewards/accuracies": 0.859375,
"rewards/chosen": 1.423360824584961,
"rewards/margins": 2.117220163345337,
"rewards/rejected": -0.6938591003417969,
"step": 58
},
{
"epoch": 0.7384615384615385,
"grad_norm": 7.997921172873259,
"learning_rate": 3.991286838919086e-07,
"logits/chosen": -0.45197734236717224,
"logits/rejected": -0.4494423270225525,
"logps/chosen": -39.821556091308594,
"logps/rejected": -30.172922134399414,
"loss": 0.2308,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.3537344932556152,
"rewards/margins": 2.035463571548462,
"rewards/rejected": -0.6817290186882019,
"step": 60
},
{
"epoch": 0.7384615384615385,
"eval_logits/chosen": -0.4599209725856781,
"eval_logits/rejected": -0.45611828565597534,
"eval_logps/chosen": -41.97780990600586,
"eval_logps/rejected": -46.275150299072266,
"eval_loss": 0.25006040930747986,
"eval_rewards/accuracies": 0.8219178318977356,
"eval_rewards/chosen": 1.6138089895248413,
"eval_rewards/margins": 3.3122646808624268,
"eval_rewards/rejected": -1.698456048965454,
"eval_runtime": 400.364,
"eval_samples_per_second": 4.331,
"eval_steps_per_second": 0.182,
"step": 60
},
{
"epoch": 0.7630769230769231,
"grad_norm": 8.82980837299946,
"learning_rate": 3.902967663405956e-07,
"logits/chosen": -0.4859851002693176,
"logits/rejected": -0.48067888617515564,
"logps/chosen": -41.49687194824219,
"logps/rejected": -52.49277114868164,
"loss": 0.2392,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.802024483680725,
"rewards/margins": 3.978426456451416,
"rewards/rejected": -2.1764023303985596,
"step": 62
},
{
"epoch": 0.7876923076923077,
"grad_norm": 9.426495089553443,
"learning_rate": 3.8120145626980015e-07,
"logits/chosen": -0.4271140694618225,
"logits/rejected": -0.42757901549339294,
"logps/chosen": -42.58659744262695,
"logps/rejected": -34.0533561706543,
"loss": 0.2168,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.4099901914596558,
"rewards/margins": 2.444286346435547,
"rewards/rejected": -1.0342963933944702,
"step": 64
},
{
"epoch": 0.8,
"eval_logits/chosen": -0.45980003476142883,
"eval_logits/rejected": -0.4559204876422882,
"eval_logps/chosen": -41.840850830078125,
"eval_logps/rejected": -46.55259323120117,
"eval_loss": 0.24060946702957153,
"eval_rewards/accuracies": 0.8287671208381653,
"eval_rewards/chosen": 1.6822888851165771,
"eval_rewards/margins": 3.519465208053589,
"eval_rewards/rejected": -1.8371765613555908,
"eval_runtime": 400.9472,
"eval_samples_per_second": 4.325,
"eval_steps_per_second": 0.182,
"step": 65
},
{
"epoch": 0.8123076923076923,
"grad_norm": 7.070039356412563,
"learning_rate": 3.718598291738298e-07,
"logits/chosen": -0.44447261095046997,
"logits/rejected": -0.44296732544898987,
"logps/chosen": -37.90412902832031,
"logps/rejected": -36.56926727294922,
"loss": 0.2129,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.5699365139007568,
"rewards/margins": 2.4697201251983643,
"rewards/rejected": -0.899783730506897,
"step": 66
},
{
"epoch": 0.8369230769230769,
"grad_norm": 7.560682128141536,
"learning_rate": 3.622894229814698e-07,
"logits/chosen": -0.4488967955112457,
"logits/rejected": -0.44460177421569824,
"logps/chosen": -38.6946907043457,
"logps/rejected": -46.51150894165039,
"loss": 0.2148,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.4161229133605957,
"rewards/margins": 3.171121835708618,
"rewards/rejected": -1.754999041557312,
"step": 68
},
{
"epoch": 0.8615384615384616,
"grad_norm": 8.217469156292612,
"learning_rate": 3.52508205130354e-07,
"logits/chosen": -0.45258235931396484,
"logits/rejected": -0.44472819566726685,
"logps/chosen": -41.79392623901367,
"logps/rejected": -49.990150451660156,
"loss": 0.2015,
"rewards/accuracies": 0.90625,
"rewards/chosen": 1.6353414058685303,
"rewards/margins": 4.414577960968018,
"rewards/rejected": -2.779236316680908,
"step": 70
},
{
"epoch": 0.8615384615384616,
"eval_logits/chosen": -0.45598042011260986,
"eval_logits/rejected": -0.4519526958465576,
"eval_logps/chosen": -41.811824798583984,
"eval_logps/rejected": -46.88448715209961,
"eval_loss": 0.23113983869552612,
"eval_rewards/accuracies": 0.835616409778595,
"eval_rewards/chosen": 1.696802020072937,
"eval_rewards/margins": 3.699925661087036,
"eval_rewards/rejected": -2.0031237602233887,
"eval_runtime": 397.814,
"eval_samples_per_second": 4.359,
"eval_steps_per_second": 0.184,
"step": 70
},
{
"epoch": 0.8861538461538462,
"grad_norm": 9.912800042727127,
"learning_rate": 3.4253453883497864e-07,
"logits/chosen": -0.46372586488723755,
"logits/rejected": -0.45979729294776917,
"logps/chosen": -38.839900970458984,
"logps/rejected": -39.225364685058594,
"loss": 0.1988,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.1858875751495361,
"rewards/margins": 3.046271562576294,
"rewards/rejected": -1.8603839874267578,
"step": 72
},
{
"epoch": 0.9107692307692308,
"grad_norm": 8.369939820458308,
"learning_rate": 3.323871486116851e-07,
"logits/chosen": -0.4712873697280884,
"logits/rejected": -0.4673464894294739,
"logps/chosen": -38.399444580078125,
"logps/rejected": -31.575439453125,
"loss": 0.2334,
"rewards/accuracies": 0.859375,
"rewards/chosen": 1.5223883390426636,
"rewards/margins": 2.529682159423828,
"rewards/rejected": -1.0072938203811646,
"step": 74
},
{
"epoch": 0.9230769230769231,
"eval_logits/chosen": -0.4527950584888458,
"eval_logits/rejected": -0.44881972670555115,
"eval_logps/chosen": -41.899085998535156,
"eval_logps/rejected": -47.179649353027344,
"eval_loss": 0.2244771271944046,
"eval_rewards/accuracies": 0.8424657583236694,
"eval_rewards/chosen": 1.6531713008880615,
"eval_rewards/margins": 3.803877830505371,
"eval_rewards/rejected": -2.1507065296173096,
"eval_runtime": 398.9461,
"eval_samples_per_second": 4.346,
"eval_steps_per_second": 0.183,
"step": 75
},
{
"epoch": 0.9353846153846154,
"grad_norm": 9.041284945416068,
"learning_rate": 3.220850851253377e-07,
"logits/chosen": -0.48120999336242676,
"logits/rejected": -0.476688027381897,
"logps/chosen": -41.46022415161133,
"logps/rejected": -46.435829162597656,
"loss": 0.232,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.5242944955825806,
"rewards/margins": 3.8246002197265625,
"rewards/rejected": -2.3003056049346924,
"step": 76
},
{
"epoch": 0.96,
"grad_norm": 5.93400233545759,
"learning_rate": 3.1164768942369053e-07,
"logits/chosen": -0.4919479787349701,
"logits/rejected": -0.4866538941860199,
"logps/chosen": -35.453460693359375,
"logps/rejected": -63.363746643066406,
"loss": 0.2211,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.4180715084075928,
"rewards/margins": 4.756035804748535,
"rewards/rejected": -3.3379647731781006,
"step": 78
},
{
"epoch": 0.9846153846153847,
"grad_norm": 6.808355501392862,
"learning_rate": 3.010945566265912e-07,
"logits/chosen": -0.4714447855949402,
"logits/rejected": -0.46636244654655457,
"logps/chosen": -44.541507720947266,
"logps/rejected": -33.366180419921875,
"loss": 0.2075,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.5286719799041748,
"rewards/margins": 2.761920928955078,
"rewards/rejected": -1.233249306678772,
"step": 80
},
{
"epoch": 0.9846153846153847,
"eval_logits/chosen": -0.45233336091041565,
"eval_logits/rejected": -0.44827964901924133,
"eval_logps/chosen": -41.91486740112305,
"eval_logps/rejected": -47.406375885009766,
"eval_loss": 0.2178829461336136,
"eval_rewards/accuracies": 0.8458904027938843,
"eval_rewards/chosen": 1.645279884338379,
"eval_rewards/margins": 3.909348964691162,
"eval_rewards/rejected": -2.2640695571899414,
"eval_runtime": 398.4802,
"eval_samples_per_second": 4.352,
"eval_steps_per_second": 0.183,
"step": 80
},
{
"epoch": 1.0092307692307692,
"grad_norm": 7.772486195689839,
"learning_rate": 2.9044549913819124e-07,
"logits/chosen": -0.4541618227958679,
"logits/rejected": -0.4516042470932007,
"logps/chosen": -42.46653747558594,
"logps/rejected": -42.07926940917969,
"loss": 0.1903,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.5551599264144897,
"rewards/margins": 3.8793962001800537,
"rewards/rejected": -2.3242363929748535,
"step": 82
},
{
"epoch": 1.0338461538461539,
"grad_norm": 5.782441199522606,
"learning_rate": 2.797205094512266e-07,
"logits/chosen": -0.43424034118652344,
"logits/rejected": -0.42985162138938904,
"logps/chosen": -41.127593994140625,
"logps/rejected": -45.52932357788086,
"loss": 0.1885,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.399495005607605,
"rewards/margins": 3.8588943481445312,
"rewards/rejected": -2.4593992233276367,
"step": 84
},
{
"epoch": 1.0461538461538462,
"eval_logits/chosen": -0.45614686608314514,
"eval_logits/rejected": -0.45181941986083984,
"eval_logps/chosen": -41.860015869140625,
"eval_logps/rejected": -47.55991744995117,
"eval_loss": 0.21262364089488983,
"eval_rewards/accuracies": 0.8493150472640991,
"eval_rewards/chosen": 1.6727051734924316,
"eval_rewards/margins": 4.013547897338867,
"eval_rewards/rejected": -2.3408427238464355,
"eval_runtime": 401.4695,
"eval_samples_per_second": 4.319,
"eval_steps_per_second": 0.182,
"step": 85
},
{
"epoch": 1.0584615384615386,
"grad_norm": 4.97735971019092,
"learning_rate": 2.6893972261320264e-07,
"logits/chosen": -0.42422690987586975,
"logits/rejected": -0.4213142693042755,
"logps/chosen": -39.9366340637207,
"logps/rejected": -40.46057891845703,
"loss": 0.1749,
"rewards/accuracies": 0.953125,
"rewards/chosen": 1.7523412704467773,
"rewards/margins": 3.447618007659912,
"rewards/rejected": -1.6952769756317139,
"step": 86
},
{
"epoch": 1.083076923076923,
"grad_norm": 5.4006356704238465,
"learning_rate": 2.5812337842494516e-07,
"logits/chosen": -0.4536130726337433,
"logits/rejected": -0.45070070028305054,
"logps/chosen": -41.13835906982422,
"logps/rejected": -36.39931106567383,
"loss": 0.1732,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.473576545715332,
"rewards/margins": 3.01824688911438,
"rewards/rejected": -1.5446703433990479,
"step": 88
},
{
"epoch": 1.1076923076923078,
"grad_norm": 5.007052371554692,
"learning_rate": 2.4729178344249006e-07,
"logits/chosen": -0.45735257863998413,
"logits/rejected": -0.45181047916412354,
"logps/chosen": -39.739837646484375,
"logps/rejected": -40.88286590576172,
"loss": 0.1838,
"rewards/accuracies": 0.859375,
"rewards/chosen": 1.5848816633224487,
"rewards/margins": 3.2226974964141846,
"rewards/rejected": -1.6378157138824463,
"step": 90
},
{
"epoch": 1.1076923076923078,
"eval_logits/chosen": -0.44746431708335876,
"eval_logits/rejected": -0.44344425201416016,
"eval_logps/chosen": -41.768062591552734,
"eval_logps/rejected": -47.707096099853516,
"eval_loss": 0.20802097022533417,
"eval_rewards/accuracies": 0.8527397513389587,
"eval_rewards/chosen": 1.7186800241470337,
"eval_rewards/margins": 4.133108615875244,
"eval_rewards/rejected": -2.414428472518921,
"eval_runtime": 398.368,
"eval_samples_per_second": 4.353,
"eval_steps_per_second": 0.183,
"step": 90
},
{
"epoch": 1.1323076923076922,
"grad_norm": 5.422550680740592,
"learning_rate": 2.3646527285364563e-07,
"logits/chosen": -0.45245465636253357,
"logits/rejected": -0.4461210072040558,
"logps/chosen": -40.16954040527344,
"logps/rejected": -40.06535339355469,
"loss": 0.1908,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.9867323637008667,
"rewards/margins": 3.7591819763183594,
"rewards/rejected": -1.7724494934082031,
"step": 92
},
{
"epoch": 1.156923076923077,
"grad_norm": 5.242542300523203,
"learning_rate": 2.256641723008026e-07,
"logits/chosen": -0.4320615530014038,
"logits/rejected": -0.4263082444667816,
"logps/chosen": -37.380733489990234,
"logps/rejected": -34.89535903930664,
"loss": 0.2062,
"rewards/accuracies": 0.859375,
"rewards/chosen": 1.6694892644882202,
"rewards/margins": 3.161728858947754,
"rewards/rejected": -1.4922394752502441,
"step": 94
},
{
"epoch": 1.1692307692307693,
"eval_logits/chosen": -0.4461810886859894,
"eval_logits/rejected": -0.44209808111190796,
"eval_logps/chosen": -41.719635009765625,
"eval_logps/rejected": -47.83554458618164,
"eval_loss": 0.20408523082733154,
"eval_rewards/accuracies": 0.8595890402793884,
"eval_rewards/chosen": 1.7428948879241943,
"eval_rewards/margins": 4.221549987792969,
"eval_rewards/rejected": -2.4786548614501953,
"eval_runtime": 399.3763,
"eval_samples_per_second": 4.342,
"eval_steps_per_second": 0.183,
"step": 95
},
{
"epoch": 1.1815384615384614,
"grad_norm": 5.9451928875167805,
"learning_rate": 2.1490875972166393e-07,
"logits/chosen": -0.471175879240036,
"logits/rejected": -0.46583351492881775,
"logps/chosen": -34.772342681884766,
"logps/rejected": -39.63105392456055,
"loss": 0.1776,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.5616707801818848,
"rewards/margins": 3.4220144748687744,
"rewards/rejected": -1.8603436946868896,
"step": 96
},
{
"epoch": 1.2061538461538461,
"grad_norm": 5.4858414240825075,
"learning_rate": 2.0421922727953595e-07,
"logits/chosen": -0.4286724925041199,
"logits/rejected": -0.4263175427913666,
"logps/chosen": -36.57586669921875,
"logps/rejected": -31.930763244628906,
"loss": 0.1655,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.753572940826416,
"rewards/margins": 2.9586939811706543,
"rewards/rejected": -1.2051211595535278,
"step": 98
},
{
"epoch": 1.2307692307692308,
"grad_norm": 5.720592020809117,
"learning_rate": 1.9361564345465145e-07,
"logits/chosen": -0.4527323842048645,
"logits/rejected": -0.44913169741630554,
"logps/chosen": -34.954219818115234,
"logps/rejected": -37.62247085571289,
"loss": 0.1794,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.6976581811904907,
"rewards/margins": 3.37191104888916,
"rewards/rejected": -1.674253225326538,
"step": 100
},
{
"epoch": 1.2307692307692308,
"eval_logits/chosen": -0.446783185005188,
"eval_logits/rejected": -0.44245585799217224,
"eval_logps/chosen": -41.71480178833008,
"eval_logps/rejected": -47.994571685791016,
"eval_loss": 0.20066353678703308,
"eval_rewards/accuracies": 0.8561643958091736,
"eval_rewards/chosen": 1.745313048362732,
"eval_rewards/margins": 4.30348014831543,
"eval_rewards/rejected": -2.558166742324829,
"eval_runtime": 400.8371,
"eval_samples_per_second": 4.326,
"eval_steps_per_second": 0.182,
"step": 100
},
{
"epoch": 1.2553846153846153,
"grad_norm": 5.495318529151337,
"learning_rate": 1.8311791536769483e-07,
"logits/chosen": -0.4154907763004303,
"logits/rejected": -0.41147375106811523,
"logps/chosen": -37.79768753051758,
"logps/rejected": -48.32915496826172,
"loss": 0.1734,
"rewards/accuracies": 0.90625,
"rewards/chosen": 1.4379969835281372,
"rewards/margins": 4.379321098327637,
"rewards/rejected": -2.94132399559021,
"step": 102
},
{
"epoch": 1.28,
"grad_norm": 5.371136709550833,
"learning_rate": 1.7274575140626315e-07,
"logits/chosen": -0.44362062215805054,
"logits/rejected": -0.4401937425136566,
"logps/chosen": -38.68498992919922,
"logps/rejected": -41.304710388183594,
"loss": 0.1857,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.9046180248260498,
"rewards/margins": 3.721491813659668,
"rewards/rejected": -1.8168736696243286,
"step": 104
},
{
"epoch": 1.2923076923076924,
"eval_logits/chosen": -0.44231978058815,
"eval_logits/rejected": -0.43791159987449646,
"eval_logps/chosen": -41.76345443725586,
"eval_logps/rejected": -48.174468994140625,
"eval_loss": 0.1971779763698578,
"eval_rewards/accuracies": 0.8595890402793884,
"eval_rewards/chosen": 1.7209877967834473,
"eval_rewards/margins": 4.369105339050293,
"eval_rewards/rejected": -2.6481170654296875,
"eval_runtime": 400.535,
"eval_samples_per_second": 4.329,
"eval_steps_per_second": 0.182,
"step": 105
},
{
"epoch": 1.3046153846153845,
"grad_norm": 5.888788259237849,
"learning_rate": 1.6251862422442788e-07,
"logits/chosen": -0.4513101577758789,
"logits/rejected": -0.4436649978160858,
"logps/chosen": -40.830169677734375,
"logps/rejected": -56.550262451171875,
"loss": 0.1686,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.6578915119171143,
"rewards/margins": 4.946527481079102,
"rewards/rejected": -3.2886364459991455,
"step": 106
},
{
"epoch": 1.3292307692307692,
"grad_norm": 6.612486006770989,
"learning_rate": 1.5245573418486135e-07,
"logits/chosen": -0.44994574785232544,
"logits/rejected": -0.4477950930595398,
"logps/chosen": -38.80488586425781,
"logps/rejected": -41.393211364746094,
"loss": 0.1875,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.4876614809036255,
"rewards/margins": 3.6893627643585205,
"rewards/rejected": -2.2017014026641846,
"step": 108
},
{
"epoch": 1.353846153846154,
"grad_norm": 4.656660129291058,
"learning_rate": 1.4257597331216208e-07,
"logits/chosen": -0.4406875669956207,
"logits/rejected": -0.43555599451065063,
"logps/chosen": -42.75905990600586,
"logps/rejected": -54.37929153442383,
"loss": 0.1647,
"rewards/accuracies": 0.9375,
"rewards/chosen": 1.7431519031524658,
"rewards/margins": 5.2682318687438965,
"rewards/rejected": -3.5250799655914307,
"step": 110
},
{
"epoch": 1.353846153846154,
"eval_logits/chosen": -0.44183987379074097,
"eval_logits/rejected": -0.4375387728214264,
"eval_logps/chosen": -41.78536605834961,
"eval_logps/rejected": -48.30013656616211,
"eval_loss": 0.19461464881896973,
"eval_rewards/accuracies": 0.8595890402793884,
"eval_rewards/chosen": 1.710031509399414,
"eval_rewards/margins": 4.420979976654053,
"eval_rewards/rejected": -2.7109484672546387,
"eval_runtime": 398.8533,
"eval_samples_per_second": 4.347,
"eval_steps_per_second": 0.183,
"step": 110
},
{
"epoch": 1.3784615384615384,
"grad_norm": 4.429475200922326,
"learning_rate": 1.328978898250525e-07,
"logits/chosen": -0.45591893792152405,
"logits/rejected": -0.45284152030944824,
"logps/chosen": -37.928653717041016,
"logps/rejected": -38.99214172363281,
"loss": 0.1596,
"rewards/accuracies": 0.953125,
"rewards/chosen": 1.702134609222412,
"rewards/margins": 3.827345371246338,
"rewards/rejected": -2.125210762023926,
"step": 112
},
{
"epoch": 1.403076923076923,
"grad_norm": 4.278591337865169,
"learning_rate": 1.234396533140365e-07,
"logits/chosen": -0.42062410712242126,
"logits/rejected": -0.41781002283096313,
"logps/chosen": -41.77610778808594,
"logps/rejected": -48.5487060546875,
"loss": 0.1524,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.599095106124878,
"rewards/margins": 4.177494049072266,
"rewards/rejected": -2.578399181365967,
"step": 114
},
{
"epoch": 1.4153846153846155,
"eval_logits/chosen": -0.44146084785461426,
"eval_logits/rejected": -0.437191367149353,
"eval_logps/chosen": -41.83814239501953,
"eval_logps/rejected": -48.426334381103516,
"eval_loss": 0.1936241090297699,
"eval_rewards/accuracies": 0.8561643958091736,
"eval_rewards/chosen": 1.6836414337158203,
"eval_rewards/margins": 4.457688331604004,
"eval_rewards/rejected": -2.7740468978881836,
"eval_runtime": 398.1677,
"eval_samples_per_second": 4.355,
"eval_steps_per_second": 0.183,
"step": 115
},
{
"epoch": 1.4276923076923076,
"grad_norm": 4.8647599997437805,
"learning_rate": 1.1421902062989178e-07,
"logits/chosen": -0.44749611616134644,
"logits/rejected": -0.4411754608154297,
"logps/chosen": -43.03426742553711,
"logps/rejected": -60.566200256347656,
"loss": 0.1551,
"rewards/accuracies": 0.90625,
"rewards/chosen": 1.691102147102356,
"rewards/margins": 5.743535041809082,
"rewards/rejected": -4.052433013916016,
"step": 116
},
{
"epoch": 1.4523076923076923,
"grad_norm": 4.32184296596162,
"learning_rate": 1.0525330254703788e-07,
"logits/chosen": -0.4085264205932617,
"logits/rejected": -0.4047048091888428,
"logps/chosen": -40.241764068603516,
"logps/rejected": -48.12974166870117,
"loss": 0.1681,
"rewards/accuracies": 0.796875,
"rewards/chosen": 1.730395793914795,
"rewards/margins": 4.292797088623047,
"rewards/rejected": -2.5624003410339355,
"step": 118
},
{
"epoch": 1.476923076923077,
"grad_norm": 6.544542585214649,
"learning_rate": 9.655933126436563e-08,
"logits/chosen": -0.4505607783794403,
"logits/rejected": -0.44797009229660034,
"logps/chosen": -42.54741668701172,
"logps/rejected": -34.47932052612305,
"loss": 0.1668,
"rewards/accuracies": 0.90625,
"rewards/chosen": 1.5434067249298096,
"rewards/margins": 3.265448570251465,
"rewards/rejected": -1.7220419645309448,
"step": 120
},
{
"epoch": 1.476923076923077,
"eval_logits/chosen": -0.4385281801223755,
"eval_logits/rejected": -0.43417835235595703,
"eval_logps/chosen": -41.83414077758789,
"eval_logps/rejected": -48.51993179321289,
"eval_loss": 0.19258952140808105,
"eval_rewards/accuracies": 0.8527397513389587,
"eval_rewards/chosen": 1.6856427192687988,
"eval_rewards/margins": 4.5064897537231445,
"eval_rewards/rejected": -2.8208470344543457,
"eval_runtime": 398.4093,
"eval_samples_per_second": 4.352,
"eval_steps_per_second": 0.183,
"step": 120
},
{
"epoch": 1.5015384615384615,
"grad_norm": 3.8817405874572923,
"learning_rate": 8.81534288045431e-08,
"logits/chosen": -0.4345233738422394,
"logits/rejected": -0.432235985994339,
"logps/chosen": -38.9439697265625,
"logps/rejected": -40.60162353515625,
"loss": 0.155,
"rewards/accuracies": 0.875,
"rewards/chosen": 1.4539457559585571,
"rewards/margins": 3.9496073722839355,
"rewards/rejected": -2.4956612586975098,
"step": 122
},
{
"epoch": 1.5261538461538462,
"grad_norm": 4.9773505616027425,
"learning_rate": 8.005137637112302e-08,
"logits/chosen": -0.42935946583747864,
"logits/rejected": -0.4225008487701416,
"logps/chosen": -40.33964157104492,
"logps/rejected": -37.91059875488281,
"loss": 0.1685,
"rewards/accuracies": 0.90625,
"rewards/chosen": 1.3645392656326294,
"rewards/margins": 3.7918734550476074,
"rewards/rejected": -2.4273343086242676,
"step": 124
},
{
"epoch": 1.5384615384615383,
"eval_logits/chosen": -0.4404720067977905,
"eval_logits/rejected": -0.4361659288406372,
"eval_logps/chosen": -41.75288391113281,
"eval_logps/rejected": -48.5331916809082,
"eval_loss": 0.19096876680850983,
"eval_rewards/accuracies": 0.8561643958091736,
"eval_rewards/chosen": 1.7262725830078125,
"eval_rewards/margins": 4.553750991821289,
"eval_rewards/rejected": -2.8274781703948975,
"eval_runtime": 397.8296,
"eval_samples_per_second": 4.359,
"eval_steps_per_second": 0.183,
"step": 125
},
{
"epoch": 1.5507692307692307,
"grad_norm": 3.816285562997914,
"learning_rate": 7.226838472098237e-08,
"logits/chosen": -0.4305260181427002,
"logits/rejected": -0.427184522151947,
"logps/chosen": -38.50707244873047,
"logps/rejected": -31.813642501831055,
"loss": 0.1785,
"rewards/accuracies": 0.84375,
"rewards/chosen": 1.653688669204712,
"rewards/margins": 3.0635831356048584,
"rewards/rejected": -1.4098942279815674,
"step": 126
},
{
"epoch": 1.5753846153846154,
"grad_norm": 5.070564484576485,
"learning_rate": 6.481906560771524e-08,
"logits/chosen": -0.45029404759407043,
"logits/rejected": -0.44603830575942993,
"logps/chosen": -40.4008674621582,
"logps/rejected": -34.62392044067383,
"loss": 0.162,
"rewards/accuracies": 0.890625,
"rewards/chosen": 1.847829818725586,
"rewards/margins": 3.5869626998901367,
"rewards/rejected": -1.7391327619552612,
"step": 128
},
{
"epoch": 1.6,
"grad_norm": 4.291727231427144,
"learning_rate": 5.771740434959277e-08,
"logits/chosen": -0.43359375,
"logits/rejected": -0.4286433160305023,
"logps/chosen": -38.0141716003418,
"logps/rejected": -41.48249053955078,
"loss": 0.1637,
"rewards/accuracies": 0.828125,
"rewards/chosen": 1.5690267086029053,
"rewards/margins": 3.8614985942840576,
"rewards/rejected": -2.2924716472625732,
"step": 130
},
{
"epoch": 1.6,
"eval_logits/chosen": -0.43712207674980164,
"eval_logits/rejected": -0.43287885189056396,
"eval_logps/chosen": -41.706233978271484,
"eval_logps/rejected": -48.57079315185547,
"eval_loss": 0.18982651829719543,
"eval_rewards/accuracies": 0.8595890402793884,
"eval_rewards/chosen": 1.7495962381362915,
"eval_rewards/margins": 4.5958757400512695,
"eval_rewards/rejected": -2.8462791442871094,
"eval_runtime": 399.3255,
"eval_samples_per_second": 4.342,
"eval_steps_per_second": 0.183,
"step": 130
}
],
"logging_steps": 2,
"max_steps": 162,
"num_input_tokens_seen": 0,
"num_train_epochs": 2,
"save_steps": 10,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}