aftonposten-6b-align-scan / trainer_state.json
hugodk-sch's picture
Model save
328926c verified
raw
history blame
21.4 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.0,
"eval_steps": 100,
"global_step": 385,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.0,
"learning_rate": 1.282051282051282e-07,
"logits/chosen": -1.7278180122375488,
"logits/rejected": -1.7377450466156006,
"logps/chosen": -29.553977966308594,
"logps/rejected": -42.813133239746094,
"loss": 0.6931,
"rewards/accuracies": 0.0,
"rewards/chosen": 0.0,
"rewards/margins": 0.0,
"rewards/rejected": 0.0,
"step": 1
},
{
"epoch": 0.03,
"learning_rate": 1.282051282051282e-06,
"logits/chosen": -1.8666962385177612,
"logits/rejected": -1.8709977865219116,
"logps/chosen": -36.98939514160156,
"logps/rejected": -33.66963195800781,
"loss": 0.6929,
"rewards/accuracies": 0.5694444179534912,
"rewards/chosen": 0.00017197892884723842,
"rewards/margins": 0.0005675320862792432,
"rewards/rejected": -0.0003955531574320048,
"step": 10
},
{
"epoch": 0.05,
"learning_rate": 2.564102564102564e-06,
"logits/chosen": -1.9978935718536377,
"logits/rejected": -2.000532627105713,
"logps/chosen": -29.66562843322754,
"logps/rejected": -29.045883178710938,
"loss": 0.6934,
"rewards/accuracies": 0.36250001192092896,
"rewards/chosen": -0.00023434234026353806,
"rewards/margins": -0.0004099405778106302,
"rewards/rejected": 0.0001755982666509226,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 3.846153846153847e-06,
"logits/chosen": -1.9211324453353882,
"logits/rejected": -1.9184545278549194,
"logps/chosen": -31.41294288635254,
"logps/rejected": -33.23053741455078,
"loss": 0.6931,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 3.152530553052202e-05,
"rewards/margins": 0.000152341352077201,
"rewards/rejected": -0.00012081606837455183,
"step": 30
},
{
"epoch": 0.1,
"learning_rate": 4.999896948438434e-06,
"logits/chosen": -2.017341375350952,
"logits/rejected": -2.0086092948913574,
"logps/chosen": -32.60146713256836,
"logps/rejected": -32.49399185180664,
"loss": 0.6934,
"rewards/accuracies": 0.4124999940395355,
"rewards/chosen": -0.0002466029836796224,
"rewards/margins": -0.0004333632532507181,
"rewards/rejected": 0.00018676018225960433,
"step": 40
},
{
"epoch": 0.13,
"learning_rate": 4.987541037542187e-06,
"logits/chosen": -1.862633466720581,
"logits/rejected": -1.8518692255020142,
"logps/chosen": -33.55931091308594,
"logps/rejected": -35.44870376586914,
"loss": 0.6932,
"rewards/accuracies": 0.5,
"rewards/chosen": -1.831089502957184e-05,
"rewards/margins": -5.47249146620743e-05,
"rewards/rejected": 3.641402145149186e-05,
"step": 50
},
{
"epoch": 0.16,
"learning_rate": 4.954691471941119e-06,
"logits/chosen": -1.9409154653549194,
"logits/rejected": -1.9428699016571045,
"logps/chosen": -32.53916549682617,
"logps/rejected": -33.24130630493164,
"loss": 0.6925,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.0006101715262047946,
"rewards/margins": 0.0013094183523207903,
"rewards/rejected": -0.0006992466514930129,
"step": 60
},
{
"epoch": 0.18,
"learning_rate": 4.901618883413549e-06,
"logits/chosen": -2.070591926574707,
"logits/rejected": -2.075544834136963,
"logps/chosen": -34.023067474365234,
"logps/rejected": -36.647151947021484,
"loss": 0.6929,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": -0.00034581663203425705,
"rewards/margins": 0.0004369783273432404,
"rewards/rejected": -0.0007827949011698365,
"step": 70
},
{
"epoch": 0.21,
"learning_rate": 4.828760511501322e-06,
"logits/chosen": -1.9308092594146729,
"logits/rejected": -1.933943748474121,
"logps/chosen": -34.318023681640625,
"logps/rejected": -34.67802429199219,
"loss": 0.6922,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.0010978971840813756,
"rewards/margins": 0.0019540609791874886,
"rewards/rejected": -0.000856163795106113,
"step": 80
},
{
"epoch": 0.23,
"learning_rate": 4.7367166013034295e-06,
"logits/chosen": -1.9380912780761719,
"logits/rejected": -1.9425855875015259,
"logps/chosen": -32.38385009765625,
"logps/rejected": -32.35346603393555,
"loss": 0.6928,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.0008357145707122982,
"rewards/margins": 0.0007813175907358527,
"rewards/rejected": 5.439693995867856e-05,
"step": 90
},
{
"epoch": 0.26,
"learning_rate": 4.626245458345211e-06,
"logits/chosen": -2.035137891769409,
"logits/rejected": -2.0331528186798096,
"logps/chosen": -32.112831115722656,
"logps/rejected": -31.29166030883789,
"loss": 0.6924,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.0012800416443496943,
"rewards/margins": 0.0015345367137342691,
"rewards/rejected": -0.0002544948656577617,
"step": 100
},
{
"epoch": 0.26,
"eval_logits/chosen": -2.2300801277160645,
"eval_logits/rejected": -2.225238084793091,
"eval_logps/chosen": -34.04683303833008,
"eval_logps/rejected": -37.53927230834961,
"eval_loss": 0.6930972337722778,
"eval_rewards/accuracies": 0.5186877250671387,
"eval_rewards/chosen": -0.00012280470400583,
"eval_rewards/margins": 0.00010372586984885857,
"eval_rewards/rejected": -0.0002265305956825614,
"eval_runtime": 145.7259,
"eval_samples_per_second": 2.354,
"eval_steps_per_second": 0.295,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 4.498257201263691e-06,
"logits/chosen": -1.989782691001892,
"logits/rejected": -1.9873950481414795,
"logps/chosen": -33.12385559082031,
"logps/rejected": -34.011810302734375,
"loss": 0.6926,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.0011996207758784294,
"rewards/margins": 0.001024017808958888,
"rewards/rejected": 0.00017560287960805,
"step": 110
},
{
"epoch": 0.31,
"learning_rate": 4.353806263777678e-06,
"logits/chosen": -2.0008151531219482,
"logits/rejected": -1.992500901222229,
"logps/chosen": -32.320838928222656,
"logps/rejected": -32.128170013427734,
"loss": 0.6927,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": 0.001240686746314168,
"rewards/margins": 0.0009073130786418915,
"rewards/rejected": 0.0003333735803607851,
"step": 120
},
{
"epoch": 0.34,
"learning_rate": 4.1940827077152755e-06,
"logits/chosen": -2.0289230346679688,
"logits/rejected": -2.020946502685547,
"logps/chosen": -30.313907623291016,
"logps/rejected": -32.086116790771484,
"loss": 0.6922,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.0016180993989109993,
"rewards/margins": 0.0019491963321343064,
"rewards/rejected": -0.00033109664218500257,
"step": 130
},
{
"epoch": 0.36,
"learning_rate": 4.0204024186666215e-06,
"logits/chosen": -1.9592479467391968,
"logits/rejected": -1.9694607257843018,
"logps/chosen": -31.223953247070312,
"logps/rejected": -32.547454833984375,
"loss": 0.6921,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.0019570994190871716,
"rewards/margins": 0.002082846825942397,
"rewards/rejected": -0.00012574761058203876,
"step": 140
},
{
"epoch": 0.39,
"learning_rate": 3.834196265035119e-06,
"logits/chosen": -1.8708124160766602,
"logits/rejected": -1.8719879388809204,
"logps/chosen": -33.877174377441406,
"logps/rejected": -34.78774642944336,
"loss": 0.6914,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.003155181184411049,
"rewards/margins": 0.0034600873477756977,
"rewards/rejected": -0.00030490627977997065,
"step": 150
},
{
"epoch": 0.42,
"learning_rate": 3.636998309800573e-06,
"logits/chosen": -1.9215673208236694,
"logits/rejected": -1.9181665182113647,
"logps/chosen": -36.011531829833984,
"logps/rejected": -32.685707092285156,
"loss": 0.6925,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.001601455733180046,
"rewards/margins": 0.001231002388522029,
"rewards/rejected": 0.00037045328645035625,
"step": 160
},
{
"epoch": 0.44,
"learning_rate": 3.4304331721118078e-06,
"logits/chosen": -2.021604061126709,
"logits/rejected": -2.014291524887085,
"logps/chosen": -33.482086181640625,
"logps/rejected": -31.404422760009766,
"loss": 0.6913,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.0031577465124428272,
"rewards/margins": 0.003683448536321521,
"rewards/rejected": -0.0005257020820863545,
"step": 170
},
{
"epoch": 0.47,
"learning_rate": 3.2162026428305436e-06,
"logits/chosen": -2.027444839477539,
"logits/rejected": -2.032665729522705,
"logps/chosen": -32.183101654052734,
"logps/rejected": -32.39936065673828,
"loss": 0.6918,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.0036168727092444897,
"rewards/margins": 0.0027590212412178516,
"rewards/rejected": 0.0008578516426496208,
"step": 180
},
{
"epoch": 0.49,
"learning_rate": 2.996071664294641e-06,
"logits/chosen": -2.027879238128662,
"logits/rejected": -2.025132656097412,
"logps/chosen": -31.258464813232422,
"logps/rejected": -31.348388671875,
"loss": 0.6919,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.0023048892617225647,
"rewards/margins": 0.0026066480204463005,
"rewards/rejected": -0.000301758642308414,
"step": 190
},
{
"epoch": 0.52,
"learning_rate": 2.7718537898066833e-06,
"logits/chosen": -1.8983129262924194,
"logits/rejected": -1.902967095375061,
"logps/chosen": -31.276391983032227,
"logps/rejected": -32.81935119628906,
"loss": 0.6914,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.0032989257015287876,
"rewards/margins": 0.00358308176510036,
"rewards/rejected": -0.00028415597626008093,
"step": 200
},
{
"epoch": 0.52,
"eval_logits/chosen": -2.2249655723571777,
"eval_logits/rejected": -2.220139503479004,
"eval_logps/chosen": -34.04255294799805,
"eval_logps/rejected": -37.55300521850586,
"eval_loss": 0.6930080056190491,
"eval_rewards/accuracies": 0.5245016813278198,
"eval_rewards/chosen": -8.006239659152925e-05,
"eval_rewards/margins": 0.00028380370349623263,
"eval_rewards/rejected": -0.00036386612919159234,
"eval_runtime": 145.5269,
"eval_samples_per_second": 2.357,
"eval_steps_per_second": 0.295,
"step": 200
},
{
"epoch": 0.55,
"learning_rate": 2.5453962426402006e-06,
"logits/chosen": -2.010593891143799,
"logits/rejected": -2.021207332611084,
"logps/chosen": -31.7437801361084,
"logps/rejected": -33.93886947631836,
"loss": 0.6916,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.0022752191871404648,
"rewards/margins": 0.003036911366507411,
"rewards/rejected": -0.000761692295782268,
"step": 210
},
{
"epoch": 0.57,
"learning_rate": 2.3185646976551794e-06,
"logits/chosen": -1.9029136896133423,
"logits/rejected": -1.9176632165908813,
"logps/chosen": -29.78145408630371,
"logps/rejected": -31.63638687133789,
"loss": 0.6911,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.003300876123830676,
"rewards/margins": 0.0040829661302268505,
"rewards/rejected": -0.000782089657150209,
"step": 220
},
{
"epoch": 0.6,
"learning_rate": 2.0932279108998323e-06,
"logits/chosen": -1.9591538906097412,
"logits/rejected": -1.9631026983261108,
"logps/chosen": -33.05189895629883,
"logps/rejected": -31.594707489013672,
"loss": 0.6911,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.003655704203993082,
"rewards/margins": 0.004109731875360012,
"rewards/rejected": -0.00045402703108265996,
"step": 230
},
{
"epoch": 0.62,
"learning_rate": 1.8712423238279358e-06,
"logits/chosen": -1.9572566747665405,
"logits/rejected": -1.9354870319366455,
"logps/chosen": -33.83857727050781,
"logps/rejected": -35.12303924560547,
"loss": 0.6907,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.0031517534516751766,
"rewards/margins": 0.004874187987297773,
"rewards/rejected": -0.0017224351176992059,
"step": 240
},
{
"epoch": 0.65,
"learning_rate": 1.6544367689701824e-06,
"logits/chosen": -1.998875379562378,
"logits/rejected": -1.9955555200576782,
"logps/chosen": -32.72559356689453,
"logps/rejected": -36.2435417175293,
"loss": 0.6921,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": 0.0019947518594563007,
"rewards/margins": 0.0021809376776218414,
"rewards/rejected": -0.0001861859782366082,
"step": 250
},
{
"epoch": 0.68,
"learning_rate": 1.4445974030621963e-06,
"logits/chosen": -1.8661177158355713,
"logits/rejected": -1.8636993169784546,
"logps/chosen": -33.959014892578125,
"logps/rejected": -35.526344299316406,
"loss": 0.6919,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.0022938635665923357,
"rewards/margins": 0.0025111136492341757,
"rewards/rejected": -0.0002172500389860943,
"step": 260
},
{
"epoch": 0.7,
"learning_rate": 1.243452991757889e-06,
"logits/chosen": -1.8511241674423218,
"logits/rejected": -1.8487510681152344,
"logps/chosen": -34.16337585449219,
"logps/rejected": -31.830408096313477,
"loss": 0.6917,
"rewards/accuracies": 0.625,
"rewards/chosen": 0.002342230873182416,
"rewards/margins": 0.0029330006800591946,
"rewards/rejected": -0.0005907699232921004,
"step": 270
},
{
"epoch": 0.73,
"learning_rate": 1.0526606671603523e-06,
"logits/chosen": -1.9535648822784424,
"logits/rejected": -1.943101167678833,
"logps/chosen": -35.01304244995117,
"logps/rejected": -31.87521743774414,
"loss": 0.6913,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.0034359837882220745,
"rewards/margins": 0.0037782168947160244,
"rewards/rejected": -0.0003422332229092717,
"step": 280
},
{
"epoch": 0.75,
"learning_rate": 8.737922755071455e-07,
"logits/chosen": -2.048783302307129,
"logits/rejected": -2.0339112281799316,
"logps/chosen": -30.716812133789062,
"logps/rejected": -32.62614059448242,
"loss": 0.6922,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.0020731096155941486,
"rewards/margins": 0.001816184027120471,
"rewards/rejected": 0.00025692558847367764,
"step": 290
},
{
"epoch": 0.78,
"learning_rate": 7.08321427484816e-07,
"logits/chosen": -1.9198474884033203,
"logits/rejected": -1.917340636253357,
"logps/chosen": -32.29683303833008,
"logps/rejected": -30.91409683227539,
"loss": 0.6895,
"rewards/accuracies": 0.737500011920929,
"rewards/chosen": 0.0063027567230165005,
"rewards/margins": 0.007275627460330725,
"rewards/rejected": -0.0009728703880682588,
"step": 300
},
{
"epoch": 0.78,
"eval_logits/chosen": -2.221111297607422,
"eval_logits/rejected": -2.2162926197052,
"eval_logps/chosen": -34.0648307800293,
"eval_logps/rejected": -37.58684158325195,
"eval_loss": 0.6929495930671692,
"eval_rewards/accuracies": 0.5419435501098633,
"eval_rewards/chosen": -0.0003027978236787021,
"eval_rewards/margins": 0.0003993964346591383,
"eval_rewards/rejected": -0.0007021942874416709,
"eval_runtime": 145.7415,
"eval_samples_per_second": 2.353,
"eval_steps_per_second": 0.295,
"step": 300
},
{
"epoch": 0.81,
"learning_rate": 5.576113578589035e-07,
"logits/chosen": -1.905469298362732,
"logits/rejected": -1.9022337198257446,
"logps/chosen": -31.291845321655273,
"logps/rejected": -33.816497802734375,
"loss": 0.6912,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.003115057712420821,
"rewards/margins": 0.003858551848679781,
"rewards/rejected": -0.0007434942526742816,
"step": 310
},
{
"epoch": 0.83,
"learning_rate": 4.229036944380913e-07,
"logits/chosen": -1.9548442363739014,
"logits/rejected": -1.9426848888397217,
"logps/chosen": -34.22846603393555,
"logps/rejected": -33.660888671875,
"loss": 0.6907,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.0036761020310223103,
"rewards/margins": 0.004896566737443209,
"rewards/rejected": -0.0012204641243442893,
"step": 320
},
{
"epoch": 0.86,
"learning_rate": 3.053082288996112e-07,
"logits/chosen": -1.9896023273468018,
"logits/rejected": -1.9881556034088135,
"logps/chosen": -33.096466064453125,
"logps/rejected": -32.550086975097656,
"loss": 0.6909,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.0038932852912694216,
"rewards/margins": 0.004598308354616165,
"rewards/rejected": -0.0007050234125927091,
"step": 330
},
{
"epoch": 0.88,
"learning_rate": 2.0579377374915805e-07,
"logits/chosen": -2.0761990547180176,
"logits/rejected": -2.0605435371398926,
"logps/chosen": -33.7951774597168,
"logps/rejected": -33.09763717651367,
"loss": 0.6913,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.0038421000353991985,
"rewards/margins": 0.0036381154786795378,
"rewards/rejected": 0.00020398428023327142,
"step": 340
},
{
"epoch": 0.91,
"learning_rate": 1.2518018074041684e-07,
"logits/chosen": -1.9492791891098022,
"logits/rejected": -1.9484554529190063,
"logps/chosen": -32.777854919433594,
"logps/rejected": -32.48105239868164,
"loss": 0.6906,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.005042382050305605,
"rewards/margins": 0.00520141888409853,
"rewards/rejected": -0.00015903706662356853,
"step": 350
},
{
"epoch": 0.94,
"learning_rate": 6.41315865106129e-08,
"logits/chosen": -1.9036123752593994,
"logits/rejected": -1.9138826131820679,
"logps/chosen": -31.845821380615234,
"logps/rejected": -35.32244873046875,
"loss": 0.6912,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.003588717896491289,
"rewards/margins": 0.003925635013729334,
"rewards/rejected": -0.00033691702992655337,
"step": 360
},
{
"epoch": 0.96,
"learning_rate": 2.3150941078050325e-08,
"logits/chosen": -2.043790102005005,
"logits/rejected": -2.0373566150665283,
"logps/chosen": -33.307098388671875,
"logps/rejected": -29.227703094482422,
"loss": 0.6914,
"rewards/accuracies": 0.6875,
"rewards/chosen": 0.003448712872341275,
"rewards/margins": 0.0034615055192261934,
"rewards/rejected": -1.2792841516784392e-05,
"step": 370
},
{
"epoch": 0.99,
"learning_rate": 2.575864278703266e-09,
"logits/chosen": -1.9043537378311157,
"logits/rejected": -1.9065628051757812,
"logps/chosen": -33.83999252319336,
"logps/rejected": -30.96780776977539,
"loss": 0.6907,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.003990581724792719,
"rewards/margins": 0.004942373372614384,
"rewards/rejected": -0.0009517916478216648,
"step": 380
},
{
"epoch": 1.0,
"step": 385,
"total_flos": 0.0,
"train_loss": 0.6918259131443965,
"train_runtime": 3249.6401,
"train_samples_per_second": 0.947,
"train_steps_per_second": 0.118
}
],
"logging_steps": 10,
"max_steps": 385,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 100,
"total_flos": 0.0,
"train_batch_size": 4,
"trial_name": null,
"trial_params": null
}