aftonposten-6b-align-scan / trainer_state.json
hugodk-sch's picture
Model save
faf3735 verified
raw
history blame
21.1 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.5,
"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.8667322397232056,
"logits/rejected": -1.8710299730300903,
"logps/chosen": -36.98433303833008,
"logps/rejected": -33.65640640258789,
"loss": 0.4948,
"rewards/accuracies": 0.5277777910232544,
"rewards/chosen": 0.015585740096867085,
"rewards/margins": 0.034019216895103455,
"rewards/rejected": -0.018433474004268646,
"step": 10
},
{
"epoch": 0.05,
"learning_rate": 2.564102564102564e-06,
"logits/chosen": -1.9972655773162842,
"logits/rejected": -1.9999065399169922,
"logps/chosen": -29.6416072845459,
"logps/rejected": -29.046981811523438,
"loss": 0.5013,
"rewards/accuracies": 0.5,
"rewards/chosen": 0.00041136163054034114,
"rewards/margins": -0.011113673448562622,
"rewards/rejected": 0.011525033973157406,
"step": 20
},
{
"epoch": 0.08,
"learning_rate": 3.846153846153847e-06,
"logits/chosen": -1.9205191135406494,
"logits/rejected": -1.9178133010864258,
"logps/chosen": -31.385412216186523,
"logps/rejected": -33.23365020751953,
"loss": 0.4942,
"rewards/accuracies": 0.5375000238418579,
"rewards/chosen": 0.02147640287876129,
"rewards/margins": 0.03211076930165291,
"rewards/rejected": -0.010634368285536766,
"step": 30
},
{
"epoch": 0.1,
"learning_rate": 4.999896948438434e-06,
"logits/chosen": -2.017784833908081,
"logits/rejected": -2.009059429168701,
"logps/chosen": -32.547088623046875,
"logps/rejected": -32.473243713378906,
"loss": 0.5024,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": 0.020800117403268814,
"rewards/margins": -0.006797379348427057,
"rewards/rejected": 0.027597496286034584,
"step": 40
},
{
"epoch": 0.13,
"learning_rate": 4.987541037542187e-06,
"logits/chosen": -1.863996148109436,
"logits/rejected": -1.8532311916351318,
"logps/chosen": -33.490657806396484,
"logps/rejected": -35.415367126464844,
"loss": 0.4957,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.046776313334703445,
"rewards/margins": 0.02089320495724678,
"rewards/rejected": 0.025883108377456665,
"step": 50
},
{
"epoch": 0.16,
"learning_rate": 4.954691471941119e-06,
"logits/chosen": -1.9439821243286133,
"logits/rejected": -1.9459091424942017,
"logps/chosen": -32.50756072998047,
"logps/rejected": -33.15720748901367,
"loss": 0.489,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.06483479589223862,
"rewards/margins": 0.05491489917039871,
"rewards/rejected": 0.009919902309775352,
"step": 60
},
{
"epoch": 0.18,
"learning_rate": 4.901618883413549e-06,
"logits/chosen": -2.075462579727173,
"logits/rejected": -2.0804481506347656,
"logps/chosen": -33.86588668823242,
"logps/rejected": -36.501365661621094,
"loss": 0.491,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": 0.08581922948360443,
"rewards/margins": 0.038563162088394165,
"rewards/rejected": 0.04725607484579086,
"step": 70
},
{
"epoch": 0.21,
"learning_rate": 4.828760511501322e-06,
"logits/chosen": -1.9374949932098389,
"logits/rejected": -1.9405946731567383,
"logps/chosen": -34.200557708740234,
"logps/rejected": -34.543209075927734,
"loss": 0.4719,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.15907494723796844,
"rewards/margins": 0.1246359720826149,
"rewards/rejected": 0.03443896025419235,
"step": 80
},
{
"epoch": 0.23,
"learning_rate": 4.7367166013034295e-06,
"logits/chosen": -1.9453551769256592,
"logits/rejected": -1.9498746395111084,
"logps/chosen": -32.2825927734375,
"logps/rejected": -32.25676727294922,
"loss": 0.4858,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.12937995791435242,
"rewards/margins": 0.05788033455610275,
"rewards/rejected": 0.07149962335824966,
"step": 90
},
{
"epoch": 0.26,
"learning_rate": 4.626245458345211e-06,
"logits/chosen": -2.04329776763916,
"logits/rejected": -2.0412919521331787,
"logps/chosen": -32.015525817871094,
"logps/rejected": -31.169103622436523,
"loss": 0.4799,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.15771625936031342,
"rewards/margins": 0.08974156528711319,
"rewards/rejected": 0.06797470152378082,
"step": 100
},
{
"epoch": 0.26,
"eval_logits/chosen": -2.238145351409912,
"eval_logits/rejected": -2.2333030700683594,
"eval_logps/chosen": -33.86069107055664,
"eval_logps/rejected": -37.356990814208984,
"eval_loss": 0.49784979224205017,
"eval_rewards/accuracies": 0.5340532064437866,
"eval_rewards/chosen": 0.12170296907424927,
"eval_rewards/margins": 0.009964452125132084,
"eval_rewards/rejected": 0.11173851788043976,
"eval_runtime": 145.9039,
"eval_samples_per_second": 2.351,
"eval_steps_per_second": 0.295,
"step": 100
},
{
"epoch": 0.29,
"learning_rate": 4.498257201263691e-06,
"logits/chosen": -1.9996016025543213,
"logits/rejected": -1.9972642660140991,
"logps/chosen": -32.95330810546875,
"logps/rejected": -33.88172912597656,
"loss": 0.4742,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.20335713028907776,
"rewards/margins": 0.10000836849212646,
"rewards/rejected": 0.10334876924753189,
"step": 110
},
{
"epoch": 0.31,
"learning_rate": 4.353806263777678e-06,
"logits/chosen": -2.0096044540405273,
"logits/rejected": -2.0012660026550293,
"logps/chosen": -32.153419494628906,
"logps/rejected": -31.961782455444336,
"loss": 0.4848,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.20403961837291718,
"rewards/margins": 0.06423385441303253,
"rewards/rejected": 0.13980577886104584,
"step": 120
},
{
"epoch": 0.34,
"learning_rate": 4.1940827077152755e-06,
"logits/chosen": -2.038412570953369,
"logits/rejected": -2.0304617881774902,
"logps/chosen": -30.165945053100586,
"logps/rejected": -31.880346298217773,
"loss": 0.4787,
"rewards/accuracies": 0.5625,
"rewards/chosen": 0.21683910489082336,
"rewards/margins": 0.0959777683019638,
"rewards/rejected": 0.12086131423711777,
"step": 130
},
{
"epoch": 0.36,
"learning_rate": 4.0204024186666215e-06,
"logits/chosen": -1.9683482646942139,
"logits/rejected": -1.9784959554672241,
"logps/chosen": -31.041671752929688,
"logps/rejected": -32.388893127441406,
"loss": 0.4621,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.264592707157135,
"rewards/margins": 0.16240297257900238,
"rewards/rejected": 0.10218973457813263,
"step": 140
},
{
"epoch": 0.39,
"learning_rate": 3.834196265035119e-06,
"logits/chosen": -1.8799880743026733,
"logits/rejected": -1.8811382055282593,
"logps/chosen": -33.686256408691406,
"logps/rejected": -34.564491271972656,
"loss": 0.4503,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": 0.3545047640800476,
"rewards/margins": 0.21957096457481384,
"rewards/rejected": 0.134933739900589,
"step": 150
},
{
"epoch": 0.42,
"learning_rate": 3.636998309800573e-06,
"logits/chosen": -1.931627869606018,
"logits/rejected": -1.9282808303833008,
"logps/chosen": -35.77287292480469,
"logps/rejected": -32.49742889404297,
"loss": 0.4707,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.27916350960731506,
"rewards/margins": 0.12143567949533463,
"rewards/rejected": 0.15772780776023865,
"step": 160
},
{
"epoch": 0.44,
"learning_rate": 3.4304331721118078e-06,
"logits/chosen": -2.0332443714141846,
"logits/rejected": -2.0259475708007812,
"logps/chosen": -33.251670837402344,
"logps/rejected": -31.17826271057129,
"loss": 0.439,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.382335364818573,
"rewards/margins": 0.2608257234096527,
"rewards/rejected": 0.1215096265077591,
"step": 170
},
{
"epoch": 0.47,
"learning_rate": 3.2162026428305436e-06,
"logits/chosen": -2.040196657180786,
"logits/rejected": -2.045389413833618,
"logps/chosen": -31.95781898498535,
"logps/rejected": -32.18687057495117,
"loss": 0.4516,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.4108788073062897,
"rewards/margins": 0.20208871364593506,
"rewards/rejected": 0.20879006385803223,
"step": 180
},
{
"epoch": 0.49,
"learning_rate": 2.996071664294641e-06,
"logits/chosen": -2.040775775909424,
"logits/rejected": -2.038043737411499,
"logps/chosen": -31.056228637695312,
"logps/rejected": -31.10147476196289,
"loss": 0.4647,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": 0.30290743708610535,
"rewards/margins": 0.1511872559785843,
"rewards/rejected": 0.15172019600868225,
"step": 190
},
{
"epoch": 0.52,
"learning_rate": 2.7718537898066833e-06,
"logits/chosen": -1.9112863540649414,
"logits/rejected": -1.9159523248672485,
"logps/chosen": -31.08163070678711,
"logps/rejected": -32.598487854003906,
"loss": 0.4453,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": 0.3672560155391693,
"rewards/margins": 0.2325470894575119,
"rewards/rejected": 0.1347089260816574,
"step": 200
},
{
"epoch": 0.52,
"eval_logits/chosen": -2.234708309173584,
"eval_logits/rejected": -2.229902505874634,
"eval_logps/chosen": -33.76850891113281,
"eval_logps/rejected": -37.29365921020508,
"eval_loss": 0.49275800585746765,
"eval_rewards/accuracies": 0.5369601845741272,
"eval_rewards/chosen": 0.18622951209545135,
"eval_rewards/margins": 0.030155813321471214,
"eval_rewards/rejected": 0.1560736745595932,
"eval_runtime": 145.5287,
"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.022939443588257,
"logits/rejected": -2.033546209335327,
"logps/chosen": -31.5457706451416,
"logps/rejected": -33.67595672607422,
"loss": 0.4628,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": 0.2978750169277191,
"rewards/margins": 0.16715146601200104,
"rewards/rejected": 0.13072356581687927,
"step": 210
},
{
"epoch": 0.57,
"learning_rate": 2.3185646976551794e-06,
"logits/chosen": -1.9158331155776978,
"logits/rejected": -1.9305419921875,
"logps/chosen": -29.57406997680664,
"logps/rejected": -31.42013168334961,
"loss": 0.4363,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.37622833251953125,
"rewards/margins": 0.2795966863632202,
"rewards/rejected": 0.09663165360689163,
"step": 220
},
{
"epoch": 0.6,
"learning_rate": 2.0932279108998323e-06,
"logits/chosen": -1.9733517169952393,
"logits/rejected": -1.9773433208465576,
"logps/chosen": -32.83649444580078,
"logps/rejected": -31.40326499938965,
"loss": 0.4311,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.40668267011642456,
"rewards/margins": 0.304456502199173,
"rewards/rejected": 0.10222617536783218,
"step": 230
},
{
"epoch": 0.62,
"learning_rate": 1.8712423238279358e-06,
"logits/chosen": -1.9713729619979858,
"logits/rejected": -1.9496545791625977,
"logps/chosen": -33.5602912902832,
"logps/rejected": -34.85026550292969,
"loss": 0.4227,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.41541916131973267,
"rewards/margins": 0.3450462222099304,
"rewards/rejected": 0.07037289440631866,
"step": 240
},
{
"epoch": 0.65,
"learning_rate": 1.6544367689701824e-06,
"logits/chosen": -2.012058734893799,
"logits/rejected": -2.0087602138519287,
"logps/chosen": -32.46921157836914,
"logps/rejected": -36.012245178222656,
"loss": 0.4613,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.31910061836242676,
"rewards/margins": 0.17022350430488586,
"rewards/rejected": 0.1488770991563797,
"step": 250
},
{
"epoch": 0.68,
"learning_rate": 1.4445974030621963e-06,
"logits/chosen": -1.878584861755371,
"logits/rejected": -1.876153588294983,
"logps/chosen": -33.71690368652344,
"logps/rejected": -35.26210403442383,
"loss": 0.4631,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": 0.33004888892173767,
"rewards/margins": 0.1602855622768402,
"rewards/rejected": 0.16976332664489746,
"step": 260
},
{
"epoch": 0.7,
"learning_rate": 1.243452991757889e-06,
"logits/chosen": -1.863705039024353,
"logits/rejected": -1.8611875772476196,
"logps/chosen": -33.94118881225586,
"logps/rejected": -31.579448699951172,
"loss": 0.4582,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": 0.3194878101348877,
"rewards/margins": 0.18517015874385834,
"rewards/rejected": 0.13431766629219055,
"step": 270
},
{
"epoch": 0.73,
"learning_rate": 1.0526606671603523e-06,
"logits/chosen": -1.96773362159729,
"logits/rejected": -1.957313895225525,
"logps/chosen": -34.72475814819336,
"logps/rejected": -31.634963989257812,
"loss": 0.4297,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.4423200190067291,
"rewards/margins": 0.298100084066391,
"rewards/rejected": 0.14421990513801575,
"step": 280
},
{
"epoch": 0.75,
"learning_rate": 8.737922755071455e-07,
"logits/chosen": -2.0632526874542236,
"logits/rejected": -2.0484511852264404,
"logps/chosen": -30.412399291992188,
"logps/rejected": -32.339237213134766,
"loss": 0.4682,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": 0.3582096993923187,
"rewards/margins": 0.13939058780670166,
"rewards/rejected": 0.21881911158561707,
"step": 290
},
{
"epoch": 0.78,
"learning_rate": 7.08321427484816e-07,
"logits/chosen": -1.9345197677612305,
"logits/rejected": -1.9320701360702515,
"logps/chosen": -32.11776351928711,
"logps/rejected": -30.692516326904297,
"loss": 0.3947,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": 0.5665432810783386,
"rewards/margins": 0.4795381426811218,
"rewards/rejected": 0.08700509369373322,
"step": 300
},
{
"epoch": 0.78,
"eval_logits/chosen": -2.2321932315826416,
"eval_logits/rejected": -2.227381944656372,
"eval_logps/chosen": -33.75507736206055,
"eval_logps/rejected": -37.28935623168945,
"eval_loss": 0.49099090695381165,
"eval_rewards/accuracies": 0.5564784407615662,
"eval_rewards/chosen": 0.1956319361925125,
"eval_rewards/margins": 0.03654861077666283,
"eval_rewards/rejected": 0.15908333659172058,
"eval_runtime": 145.678,
"eval_samples_per_second": 2.355,
"eval_steps_per_second": 0.295,
"step": 300
},
{
"epoch": 0.81,
"learning_rate": 5.576113578589035e-07,
"logits/chosen": -1.918630838394165,
"logits/rejected": -1.9154895544052124,
"logps/chosen": -31.033864974975586,
"logps/rejected": -33.57470703125,
"loss": 0.4357,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.3986373543739319,
"rewards/margins": 0.28142648935317993,
"rewards/rejected": 0.11721093952655792,
"step": 310
},
{
"epoch": 0.83,
"learning_rate": 4.229036944380913e-07,
"logits/chosen": -1.9683278799057007,
"logits/rejected": -1.9561859369277954,
"logps/chosen": -34.02568817138672,
"logps/rejected": -33.45148849487305,
"loss": 0.4237,
"rewards/accuracies": 0.7749999761581421,
"rewards/chosen": 0.39926695823669434,
"rewards/margins": 0.3381231129169464,
"rewards/rejected": 0.06114383786916733,
"step": 320
},
{
"epoch": 0.86,
"learning_rate": 3.053082288996112e-07,
"logits/chosen": -2.0041966438293457,
"logits/rejected": -2.0028293132781982,
"logps/chosen": -32.882896423339844,
"logps/rejected": -32.24889373779297,
"loss": 0.4411,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.4220260679721832,
"rewards/margins": 0.26054397225379944,
"rewards/rejected": 0.16148212552070618,
"step": 330
},
{
"epoch": 0.88,
"learning_rate": 2.0579377374915805e-07,
"logits/chosen": -2.0908923149108887,
"logits/rejected": -2.0752511024475098,
"logps/chosen": -33.49097442626953,
"logps/rejected": -32.81463623046875,
"loss": 0.4381,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.4818909764289856,
"rewards/margins": 0.26951199769973755,
"rewards/rejected": 0.21237893402576447,
"step": 340
},
{
"epoch": 0.91,
"learning_rate": 1.2518018074041684e-07,
"logits/chosen": -1.9633783102035522,
"logits/rejected": -1.9625694751739502,
"logps/chosen": -32.55985641479492,
"logps/rejected": -32.24340057373047,
"loss": 0.4211,
"rewards/accuracies": 0.699999988079071,
"rewards/chosen": 0.5055657029151917,
"rewards/margins": 0.3503406047821045,
"rewards/rejected": 0.15522508323192596,
"step": 350
},
{
"epoch": 0.94,
"learning_rate": 6.41315865106129e-08,
"logits/chosen": -1.9187549352645874,
"logits/rejected": -1.9290361404418945,
"logps/chosen": -31.61210060119629,
"logps/rejected": -35.0199089050293,
"loss": 0.4489,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": 0.41481319069862366,
"rewards/margins": 0.22661848366260529,
"rewards/rejected": 0.18819470703601837,
"step": 360
},
{
"epoch": 0.96,
"learning_rate": 2.3150941078050325e-08,
"logits/chosen": -2.0583438873291016,
"logits/rejected": -2.051846742630005,
"logps/chosen": -33.05150604248047,
"logps/rejected": -28.9710750579834,
"loss": 0.4429,
"rewards/accuracies": 0.75,
"rewards/chosen": 0.42032259702682495,
"rewards/margins": 0.2415778934955597,
"rewards/rejected": 0.17874471843242645,
"step": 370
},
{
"epoch": 0.99,
"learning_rate": 2.575864278703266e-09,
"logits/chosen": -1.9180545806884766,
"logits/rejected": -1.92022705078125,
"logps/chosen": -33.62676239013672,
"logps/rejected": -30.719894409179688,
"loss": 0.4281,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": 0.4286006987094879,
"rewards/margins": 0.3216874301433563,
"rewards/rejected": 0.10691330581903458,
"step": 380
},
{
"epoch": 1.0,
"step": 385,
"total_flos": 0.0,
"train_loss": 0.45851393303313814,
"train_runtime": 3252.0771,
"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
}