LLama-8B-Instruct-v0.1-MI-2e-5 / trainer_state.json
tengxiao1
TX
ae60326
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 0.9982631930527722,
"eval_steps": 400,
"global_step": 467,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.01068804275217101,
"grad_norm": 7.8646928358072685,
"learning_rate": 2.1276595744680853e-06,
"logits/chosen": 0.061868004500865936,
"logits/rejected": 0.03774915263056755,
"logps/chosen": -0.2700309157371521,
"logps/rejected": -0.2683418393135071,
"loss": 1.2747,
"rewards/accuracies": 0.5,
"rewards/chosen": -0.2700309157371521,
"rewards/margins": -0.0016891139093786478,
"rewards/rejected": -0.2683418393135071,
"step": 5
},
{
"epoch": 0.02137608550434202,
"grad_norm": 4.709071806627697,
"learning_rate": 4.255319148936171e-06,
"logits/chosen": -0.013552236370742321,
"logits/rejected": -0.015203881077468395,
"logps/chosen": -0.2748744487762451,
"logps/rejected": -0.28091126680374146,
"loss": 1.2665,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.2748744487762451,
"rewards/margins": 0.006036859937012196,
"rewards/rejected": -0.28091126680374146,
"step": 10
},
{
"epoch": 0.03206412825651302,
"grad_norm": 6.011039832471536,
"learning_rate": 6.382978723404256e-06,
"logits/chosen": 0.003249469446018338,
"logits/rejected": 0.006007631309330463,
"logps/chosen": -0.3281441628932953,
"logps/rejected": -0.3555542826652527,
"loss": 1.2824,
"rewards/accuracies": 0.5062500238418579,
"rewards/chosen": -0.3281441628932953,
"rewards/margins": 0.027410129085183144,
"rewards/rejected": -0.3555542826652527,
"step": 15
},
{
"epoch": 0.04275217100868404,
"grad_norm": 6.004008006662261,
"learning_rate": 8.510638297872341e-06,
"logits/chosen": -0.01057378202676773,
"logits/rejected": -0.014622092247009277,
"logps/chosen": -0.3901517391204834,
"logps/rejected": -0.46824750304222107,
"loss": 1.2729,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.3901517391204834,
"rewards/margins": 0.07809571921825409,
"rewards/rejected": -0.46824750304222107,
"step": 20
},
{
"epoch": 0.053440213760855046,
"grad_norm": 3.977322200325917,
"learning_rate": 1.0638297872340426e-05,
"logits/chosen": -0.2300865650177002,
"logits/rejected": -0.22464950382709503,
"logps/chosen": -0.3714216351509094,
"logps/rejected": -0.4496099352836609,
"loss": 1.2469,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.3714216351509094,
"rewards/margins": 0.07818831503391266,
"rewards/rejected": -0.4496099352836609,
"step": 25
},
{
"epoch": 0.06412825651302605,
"grad_norm": 3.540400129958154,
"learning_rate": 1.2765957446808513e-05,
"logits/chosen": -0.3409841060638428,
"logits/rejected": -0.3320957124233246,
"logps/chosen": -0.4569724202156067,
"logps/rejected": -0.5337254405021667,
"loss": 1.2712,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.4569724202156067,
"rewards/margins": 0.07675307244062424,
"rewards/rejected": -0.5337254405021667,
"step": 30
},
{
"epoch": 0.07481629926519706,
"grad_norm": 3.7566095401283395,
"learning_rate": 1.4893617021276596e-05,
"logits/chosen": -0.179878830909729,
"logits/rejected": -0.17120154201984406,
"logps/chosen": -0.46021848917007446,
"logps/rejected": -0.5379027128219604,
"loss": 1.2803,
"rewards/accuracies": 0.53125,
"rewards/chosen": -0.46021848917007446,
"rewards/margins": 0.0776842013001442,
"rewards/rejected": -0.5379027128219604,
"step": 35
},
{
"epoch": 0.08550434201736808,
"grad_norm": 3.1819948043330344,
"learning_rate": 1.7021276595744682e-05,
"logits/chosen": -0.007068359758704901,
"logits/rejected": -0.01693672500550747,
"logps/chosen": -0.4490579664707184,
"logps/rejected": -0.4928591251373291,
"loss": 1.2726,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.4490579664707184,
"rewards/margins": 0.04380121827125549,
"rewards/rejected": -0.4928591251373291,
"step": 40
},
{
"epoch": 0.09619238476953908,
"grad_norm": 3.2270349702087966,
"learning_rate": 1.914893617021277e-05,
"logits/chosen": -0.08650992810726166,
"logits/rejected": -0.054813969880342484,
"logps/chosen": -0.5002081990242004,
"logps/rejected": -0.5691278576850891,
"loss": 1.2871,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5002081990242004,
"rewards/margins": 0.06891956925392151,
"rewards/rejected": -0.5691278576850891,
"step": 45
},
{
"epoch": 0.10688042752171009,
"grad_norm": 2.930228008373617,
"learning_rate": 1.999748234942507e-05,
"logits/chosen": 0.05123206228017807,
"logits/rejected": 0.0639006644487381,
"logps/chosen": -0.4998701512813568,
"logps/rejected": -0.5959991812705994,
"loss": 1.286,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.4998701512813568,
"rewards/margins": 0.09612905234098434,
"rewards/rejected": -0.5959991812705994,
"step": 50
},
{
"epoch": 0.11756847027388109,
"grad_norm": 3.532397749748793,
"learning_rate": 1.998210129767735e-05,
"logits/chosen": 0.12292106449604034,
"logits/rejected": 0.11930225789546967,
"logps/chosen": -0.5049015879631042,
"logps/rejected": -0.5526176691055298,
"loss": 1.2864,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5049015879631042,
"rewards/margins": 0.0477161705493927,
"rewards/rejected": -0.5526176691055298,
"step": 55
},
{
"epoch": 0.1282565130260521,
"grad_norm": 3.727525392306118,
"learning_rate": 1.9952759374651266e-05,
"logits/chosen": 0.13513417541980743,
"logits/rejected": 0.11871670186519623,
"logps/chosen": -0.5651683807373047,
"logps/rejected": -0.629240095615387,
"loss": 1.297,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5651683807373047,
"rewards/margins": 0.06407161056995392,
"rewards/rejected": -0.629240095615387,
"step": 60
},
{
"epoch": 0.13894455577822312,
"grad_norm": 4.935532042698089,
"learning_rate": 1.990949761767935e-05,
"logits/chosen": 0.2536969482898712,
"logits/rejected": 0.29733771085739136,
"logps/chosen": -0.5493020415306091,
"logps/rejected": -0.6132219433784485,
"loss": 1.2983,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.5493020415306091,
"rewards/margins": 0.06391985714435577,
"rewards/rejected": -0.6132219433784485,
"step": 65
},
{
"epoch": 0.14963259853039412,
"grad_norm": 3.295200796877631,
"learning_rate": 1.9852376532240594e-05,
"logits/chosen": 0.4693821966648102,
"logits/rejected": 0.4486643671989441,
"logps/chosen": -0.5880988836288452,
"logps/rejected": -0.6713588833808899,
"loss": 1.2884,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.5880988836288452,
"rewards/margins": 0.08326009660959244,
"rewards/rejected": -0.6713588833808899,
"step": 70
},
{
"epoch": 0.16032064128256512,
"grad_norm": 3.1713468009980934,
"learning_rate": 1.9781476007338058e-05,
"logits/chosen": 0.4574071764945984,
"logits/rejected": 0.5068231225013733,
"logps/chosen": -0.5979748368263245,
"logps/rejected": -0.7264337539672852,
"loss": 1.2954,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5979748368263245,
"rewards/margins": 0.1284589320421219,
"rewards/rejected": -0.7264337539672852,
"step": 75
},
{
"epoch": 0.17100868403473615,
"grad_norm": 3.662092336336613,
"learning_rate": 1.969689520376687e-05,
"logits/chosen": 0.4844590127468109,
"logits/rejected": 0.48810499906539917,
"logps/chosen": -0.5642470121383667,
"logps/rejected": -0.6027047038078308,
"loss": 1.3069,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.5642470121383667,
"rewards/margins": 0.038457684218883514,
"rewards/rejected": -0.6027047038078308,
"step": 80
},
{
"epoch": 0.18169672678690715,
"grad_norm": 4.707016904902269,
"learning_rate": 1.9598752415428893e-05,
"logits/chosen": 0.6143449544906616,
"logits/rejected": 0.6212804317474365,
"logps/chosen": -0.5707159042358398,
"logps/rejected": -0.6457289457321167,
"loss": 1.3041,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.5707159042358398,
"rewards/margins": 0.07501306384801865,
"rewards/rejected": -0.6457289457321167,
"step": 85
},
{
"epoch": 0.19238476953907815,
"grad_norm": 3.539384946251158,
"learning_rate": 1.9487184903888e-05,
"logits/chosen": 0.7592190504074097,
"logits/rejected": 0.7624388933181763,
"logps/chosen": -0.5937708616256714,
"logps/rejected": -0.7089810967445374,
"loss": 1.2995,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.5937708616256714,
"rewards/margins": 0.11521027237176895,
"rewards/rejected": -0.7089810967445374,
"step": 90
},
{
"epoch": 0.20307281229124916,
"grad_norm": 3.622523084533489,
"learning_rate": 1.9362348706397374e-05,
"logits/chosen": 0.45539766550064087,
"logits/rejected": 0.439523309469223,
"logps/chosen": -0.6361597180366516,
"logps/rejected": -0.660950243473053,
"loss": 1.3105,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.6361597180366516,
"rewards/margins": 0.024790499359369278,
"rewards/rejected": -0.660950243473053,
"step": 95
},
{
"epoch": 0.21376085504342018,
"grad_norm": 3.7211660011686827,
"learning_rate": 1.9224418417667295e-05,
"logits/chosen": 0.4752308428287506,
"logits/rejected": 0.4805218279361725,
"logps/chosen": -0.6078037619590759,
"logps/rejected": -0.7114612460136414,
"loss": 1.3073,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.6078037619590759,
"rewards/margins": 0.10365743935108185,
"rewards/rejected": -0.7114612460136414,
"step": 100
},
{
"epoch": 0.22444889779559118,
"grad_norm": 3.4955810534668945,
"learning_rate": 1.907358694567865e-05,
"logits/chosen": 0.4134964346885681,
"logits/rejected": 0.39591750502586365,
"logps/chosen": -0.5727143287658691,
"logps/rejected": -0.6490830183029175,
"loss": 1.2904,
"rewards/accuracies": 0.518750011920929,
"rewards/chosen": -0.5727143287658691,
"rewards/margins": 0.07636874169111252,
"rewards/rejected": -0.6490830183029175,
"step": 105
},
{
"epoch": 0.23513694054776219,
"grad_norm": 3.895186311710448,
"learning_rate": 1.891006524188368e-05,
"logits/chosen": 0.27901846170425415,
"logits/rejected": 0.267717182636261,
"logps/chosen": -0.5927436351776123,
"logps/rejected": -0.6760529279708862,
"loss": 1.2909,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.5927436351776123,
"rewards/margins": 0.08330933004617691,
"rewards/rejected": -0.6760529279708862,
"step": 110
},
{
"epoch": 0.2458249832999332,
"grad_norm": 3.2529032172381034,
"learning_rate": 1.87340820061713e-05,
"logits/chosen": 0.4828861653804779,
"logits/rejected": 0.4803077280521393,
"logps/chosen": -0.6133522391319275,
"logps/rejected": -0.6779380440711975,
"loss": 1.3193,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.6133522391319275,
"rewards/margins": 0.06458581984043121,
"rewards/rejected": -0.6779380440711975,
"step": 115
},
{
"epoch": 0.2565130260521042,
"grad_norm": 3.0442398494478504,
"learning_rate": 1.854588336700962e-05,
"logits/chosen": 0.2846220135688782,
"logits/rejected": 0.2976106107234955,
"logps/chosen": -0.6584545969963074,
"logps/rejected": -0.7402211427688599,
"loss": 1.2983,
"rewards/accuracies": 0.5562499761581421,
"rewards/chosen": -0.6584545969963074,
"rewards/margins": 0.0817665308713913,
"rewards/rejected": -0.7402211427688599,
"step": 120
},
{
"epoch": 0.26720106880427524,
"grad_norm": 2.952766609523329,
"learning_rate": 1.834573253721303e-05,
"logits/chosen": 0.16905634105205536,
"logits/rejected": 0.18389183282852173,
"logps/chosen": -0.5878621339797974,
"logps/rejected": -0.6856662034988403,
"loss": 1.2725,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5878621339797974,
"rewards/margins": 0.097803995013237,
"rewards/rejected": -0.6856662034988403,
"step": 125
},
{
"epoch": 0.27788911155644624,
"grad_norm": 4.574556482577368,
"learning_rate": 1.8133909445815277e-05,
"logits/chosen": 0.2116955816745758,
"logits/rejected": 0.23542623221874237,
"logps/chosen": -0.6415301561355591,
"logps/rejected": -0.711525559425354,
"loss": 1.3254,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6415301561355591,
"rewards/margins": 0.06999534368515015,
"rewards/rejected": -0.711525559425354,
"step": 130
},
{
"epoch": 0.28857715430861725,
"grad_norm": 3.3299570451518163,
"learning_rate": 1.7910710346563417e-05,
"logits/chosen": 0.21890752017498016,
"logits/rejected": 0.2111731767654419,
"logps/chosen": -0.6361859440803528,
"logps/rejected": -0.7430610656738281,
"loss": 1.3005,
"rewards/accuracies": 0.574999988079071,
"rewards/chosen": -0.6361859440803528,
"rewards/margins": 0.10687508434057236,
"rewards/rejected": -0.7430610656738281,
"step": 135
},
{
"epoch": 0.29926519706078825,
"grad_norm": 4.712724056108227,
"learning_rate": 1.7676447403580114e-05,
"logits/chosen": 0.17464594542980194,
"logits/rejected": 0.15127554535865784,
"logps/chosen": -0.6231650114059448,
"logps/rejected": -0.7518295049667358,
"loss": 1.3004,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6231650114059448,
"rewards/margins": 0.12866440415382385,
"rewards/rejected": -0.7518295049667358,
"step": 140
},
{
"epoch": 0.30995323981295925,
"grad_norm": 3.8840639264672605,
"learning_rate": 1.7431448254773943e-05,
"logits/chosen": 0.09850597381591797,
"logits/rejected": 0.06791912019252777,
"logps/chosen": -0.6357108950614929,
"logps/rejected": -0.7526232004165649,
"loss": 1.2942,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.6357108950614929,
"rewards/margins": 0.11691226810216904,
"rewards/rejected": -0.7526232004165649,
"step": 145
},
{
"epoch": 0.32064128256513025,
"grad_norm": 3.8874041876031167,
"learning_rate": 1.717605555360812e-05,
"logits/chosen": -0.051757656037807465,
"logits/rejected": -0.07511379569768906,
"logps/chosen": -0.6289964914321899,
"logps/rejected": -0.7442386746406555,
"loss": 1.3017,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.6289964914321899,
"rewards/margins": 0.115242138504982,
"rewards/rejected": -0.7442386746406555,
"step": 150
},
{
"epoch": 0.33132932531730125,
"grad_norm": 2.9263158094537807,
"learning_rate": 1.691062648986865e-05,
"logits/chosen": 0.0783982202410698,
"logits/rejected": 0.06652794033288956,
"logps/chosen": -0.6385728120803833,
"logps/rejected": -0.7080265283584595,
"loss": 1.2934,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.6385728120803833,
"rewards/margins": 0.06945370137691498,
"rewards/rejected": -0.7080265283584595,
"step": 155
},
{
"epoch": 0.3420173680694723,
"grad_norm": 3.7316994379774573,
"learning_rate": 1.6635532290102114e-05,
"logits/chosen": 0.100054070353508,
"logits/rejected": 0.10980580747127533,
"logps/chosen": -0.6498998403549194,
"logps/rejected": -0.7538723945617676,
"loss": 1.3188,
"rewards/accuracies": 0.550000011920929,
"rewards/chosen": -0.6498998403549194,
"rewards/margins": 0.10397255420684814,
"rewards/rejected": -0.7538723945617676,
"step": 160
},
{
"epoch": 0.3527054108216433,
"grad_norm": 3.48431507554954,
"learning_rate": 1.635115769842179e-05,
"logits/chosen": -0.07162860035896301,
"logits/rejected": -0.08312487602233887,
"logps/chosen": -0.5983825325965881,
"logps/rejected": -0.6531118750572205,
"loss": 1.3115,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.5983825325965881,
"rewards/margins": 0.0547294095158577,
"rewards/rejected": -0.6531118750572205,
"step": 165
},
{
"epoch": 0.3633934535738143,
"grad_norm": 3.1045296158878495,
"learning_rate": 1.60579004384082e-05,
"logits/chosen": -0.034972742199897766,
"logits/rejected": -0.024472128599882126,
"logps/chosen": -0.6385360956192017,
"logps/rejected": -0.7788956761360168,
"loss": 1.2984,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.6385360956192017,
"rewards/margins": 0.14035962522029877,
"rewards/rejected": -0.7788956761360168,
"step": 170
},
{
"epoch": 0.3740814963259853,
"grad_norm": 2.8548400499038884,
"learning_rate": 1.575617065685674e-05,
"logits/chosen": -0.13102661073207855,
"logits/rejected": -0.13152942061424255,
"logps/chosen": -0.6120776534080505,
"logps/rejected": -0.6730025410652161,
"loss": 1.3073,
"rewards/accuracies": 0.5874999761581421,
"rewards/chosen": -0.6120776534080505,
"rewards/margins": 0.06092479079961777,
"rewards/rejected": -0.6730025410652161,
"step": 175
},
{
"epoch": 0.3847695390781563,
"grad_norm": 5.270781840059861,
"learning_rate": 1.5446390350150272e-05,
"logits/chosen": -0.16384097933769226,
"logits/rejected": -0.16462591290473938,
"logps/chosen": -0.6268705725669861,
"logps/rejected": -0.7125908136367798,
"loss": 1.3028,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.6268705725669861,
"rewards/margins": 0.08572021126747131,
"rewards/rejected": -0.7125908136367798,
"step": 180
},
{
"epoch": 0.3954575818303273,
"grad_norm": 2.9703550953722244,
"learning_rate": 1.5128992774059063e-05,
"logits/chosen": -0.2370837926864624,
"logits/rejected": -0.2578278183937073,
"logps/chosen": -0.6618776321411133,
"logps/rejected": -0.6950840353965759,
"loss": 1.3242,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.6618776321411133,
"rewards/margins": 0.03320637717843056,
"rewards/rejected": -0.6950840353965759,
"step": 185
},
{
"epoch": 0.4061456245824983,
"grad_norm": 2.576969091118456,
"learning_rate": 1.4804421837793379e-05,
"logits/chosen": -0.31508585810661316,
"logits/rejected": -0.31256750226020813,
"logps/chosen": -0.6105505228042603,
"logps/rejected": -0.6881141066551208,
"loss": 1.2954,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.6105505228042603,
"rewards/margins": 0.07756359130144119,
"rewards/rejected": -0.6881141066551208,
"step": 190
},
{
"epoch": 0.4168336673346693,
"grad_norm": 3.796929659092338,
"learning_rate": 1.4473131483156326e-05,
"logits/chosen": -0.25943300127983093,
"logits/rejected": -0.26059621572494507,
"logps/chosen": -0.6393094062805176,
"logps/rejected": -0.7625299096107483,
"loss": 1.2856,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.6393094062805176,
"rewards/margins": 0.12322056293487549,
"rewards/rejected": -0.7625299096107483,
"step": 195
},
{
"epoch": 0.42752171008684037,
"grad_norm": 3.4441112067059683,
"learning_rate": 1.4135585049665207e-05,
"logits/chosen": -0.24460437893867493,
"logits/rejected": -0.2346183955669403,
"logps/chosen": -0.6215813755989075,
"logps/rejected": -0.7071408629417419,
"loss": 1.2968,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6215813755989075,
"rewards/margins": 0.08555953204631805,
"rewards/rejected": -0.7071408629417419,
"step": 200
},
{
"epoch": 0.43820975283901137,
"grad_norm": 2.8503959168303465,
"learning_rate": 1.3792254626529286e-05,
"logits/chosen": -0.23945841193199158,
"logits/rejected": -0.2520154118537903,
"logps/chosen": -0.6485592722892761,
"logps/rejected": -0.7163916826248169,
"loss": 1.3058,
"rewards/accuracies": 0.59375,
"rewards/chosen": -0.6485592722892761,
"rewards/margins": 0.06783244758844376,
"rewards/rejected": -0.7163916826248169,
"step": 205
},
{
"epoch": 0.44889779559118237,
"grad_norm": 2.9623348839853323,
"learning_rate": 1.3443620392390352e-05,
"logits/chosen": -0.22061018645763397,
"logits/rejected": -0.21657535433769226,
"logps/chosen": -0.6273641586303711,
"logps/rejected": -0.7573559880256653,
"loss": 1.3037,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.6273641586303711,
"rewards/margins": 0.1299917995929718,
"rewards/rejected": -0.7573559880256653,
"step": 210
},
{
"epoch": 0.45958583834335337,
"grad_norm": 3.2413099077478713,
"learning_rate": 1.3090169943749475e-05,
"logits/chosen": -0.11810100078582764,
"logits/rejected": -0.14453695714473724,
"logps/chosen": -0.6222162246704102,
"logps/rejected": -0.7621700763702393,
"loss": 1.2812,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.6222162246704102,
"rewards/margins": 0.1399538516998291,
"rewards/rejected": -0.7621700763702393,
"step": 215
},
{
"epoch": 0.47027388109552437,
"grad_norm": 3.1218400805304007,
"learning_rate": 1.2732397613019203e-05,
"logits/chosen": -0.26869744062423706,
"logits/rejected": -0.277025043964386,
"logps/chosen": -0.6378098726272583,
"logps/rejected": -0.7766000628471375,
"loss": 1.281,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6378098726272583,
"rewards/margins": 0.13879024982452393,
"rewards/rejected": -0.7766000628471375,
"step": 220
},
{
"epoch": 0.48096192384769537,
"grad_norm": 3.2992783572988698,
"learning_rate": 1.2370803777154976e-05,
"logits/chosen": -0.3226834833621979,
"logits/rejected": -0.33692580461502075,
"logps/chosen": -0.6211170554161072,
"logps/rejected": -0.7601276636123657,
"loss": 1.2728,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6211170554161072,
"rewards/margins": 0.13901060819625854,
"rewards/rejected": -0.7601276636123657,
"step": 225
},
{
"epoch": 0.4916499665998664,
"grad_norm": 2.9698612659347132,
"learning_rate": 1.200589415783273e-05,
"logits/chosen": -0.2805679142475128,
"logits/rejected": -0.3003981113433838,
"logps/chosen": -0.6456371545791626,
"logps/rejected": -0.7589985132217407,
"loss": 1.2923,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.6456371545791626,
"rewards/margins": 0.11336138099431992,
"rewards/rejected": -0.7589985132217407,
"step": 230
},
{
"epoch": 0.5023380093520374,
"grad_norm": 2.93442353979503,
"learning_rate": 1.1638179114151378e-05,
"logits/chosen": -0.2719253897666931,
"logits/rejected": -0.2693996727466583,
"logps/chosen": -0.5996414422988892,
"logps/rejected": -0.7817131876945496,
"loss": 1.2908,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5996414422988892,
"rewards/margins": 0.18207184970378876,
"rewards/rejected": -0.7817131876945496,
"step": 235
},
{
"epoch": 0.5130260521042084,
"grad_norm": 3.1656667859377405,
"learning_rate": 1.1268172928849486e-05,
"logits/chosen": -0.36759647727012634,
"logits/rejected": -0.36635178327560425,
"logps/chosen": -0.5924197435379028,
"logps/rejected": -0.7910494208335876,
"loss": 1.286,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5924197435379028,
"rewards/margins": 0.1986297070980072,
"rewards/rejected": -0.7910494208335876,
"step": 240
},
{
"epoch": 0.5237140948563794,
"grad_norm": 2.891600548101634,
"learning_rate": 1.0896393089034336e-05,
"logits/chosen": -0.3432336151599884,
"logits/rejected": -0.33940672874450684,
"logps/chosen": -0.6054046154022217,
"logps/rejected": -0.6738919615745544,
"loss": 1.2829,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6054046154022217,
"rewards/margins": 0.06848734617233276,
"rewards/rejected": -0.6738919615745544,
"step": 245
},
{
"epoch": 0.5344021376085505,
"grad_norm": 5.01628908139822,
"learning_rate": 1.0523359562429441e-05,
"logits/chosen": -0.383030503988266,
"logits/rejected": -0.399156779050827,
"logps/chosen": -0.6492289900779724,
"logps/rejected": -0.769420862197876,
"loss": 1.2834,
"rewards/accuracies": 0.625,
"rewards/chosen": -0.6492289900779724,
"rewards/margins": 0.12019187211990356,
"rewards/rejected": -0.769420862197876,
"step": 250
},
{
"epoch": 0.5450901803607214,
"grad_norm": 2.9846407948611695,
"learning_rate": 1.0149594070152638e-05,
"logits/chosen": -0.41133952140808105,
"logits/rejected": -0.41932740807533264,
"logps/chosen": -0.6495887041091919,
"logps/rejected": -0.7451075315475464,
"loss": 1.3063,
"rewards/accuracies": 0.5687500238418579,
"rewards/chosen": -0.6495887041091919,
"rewards/margins": 0.09551887214183807,
"rewards/rejected": -0.7451075315475464,
"step": 255
},
{
"epoch": 0.5557782231128925,
"grad_norm": 3.025145774637567,
"learning_rate": 9.775619357041952e-06,
"logits/chosen": -0.37427011132240295,
"logits/rejected": -0.38933342695236206,
"logps/chosen": -0.5939801931381226,
"logps/rejected": -0.7346431016921997,
"loss": 1.2816,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.5939801931381226,
"rewards/margins": 0.14066289365291595,
"rewards/rejected": -0.7346431016921997,
"step": 260
},
{
"epoch": 0.5664662658650634,
"grad_norm": 3.7873867621914634,
"learning_rate": 9.401958460549658e-06,
"logits/chosen": -0.3902246356010437,
"logits/rejected": -0.39489641785621643,
"logps/chosen": -0.5874339938163757,
"logps/rejected": -0.6752017736434937,
"loss": 1.2809,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5874339938163757,
"rewards/margins": 0.08776768296957016,
"rewards/rejected": -0.6752017736434937,
"step": 265
},
{
"epoch": 0.5771543086172345,
"grad_norm": 4.336316474366534,
"learning_rate": 9.02913397922712e-06,
"logits/chosen": -0.4125131666660309,
"logits/rejected": -0.4208359122276306,
"logps/chosen": -0.607820987701416,
"logps/rejected": -0.905206024646759,
"loss": 1.2747,
"rewards/accuracies": 0.7437499761581421,
"rewards/chosen": -0.607820987701416,
"rewards/margins": 0.29738494753837585,
"rewards/rejected": -0.905206024646759,
"step": 270
},
{
"epoch": 0.5878423513694054,
"grad_norm": 2.9455182934401654,
"learning_rate": 8.657667341823449e-06,
"logits/chosen": -0.3412848114967346,
"logits/rejected": -0.3707515597343445,
"logps/chosen": -0.6197370290756226,
"logps/rejected": -0.7178043127059937,
"loss": 1.2785,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6197370290756226,
"rewards/margins": 0.09806738793849945,
"rewards/rejected": -0.7178043127059937,
"step": 275
},
{
"epoch": 0.5985303941215765,
"grad_norm": 3.0985019376115464,
"learning_rate": 8.28807807802025e-06,
"logits/chosen": -0.32350999116897583,
"logits/rejected": -0.3327878415584564,
"logps/chosen": -0.5803725123405457,
"logps/rejected": -0.6662936210632324,
"loss": 1.2821,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.5803725123405457,
"rewards/margins": 0.08592123538255692,
"rewards/rejected": -0.6662936210632324,
"step": 280
},
{
"epoch": 0.6092184368737475,
"grad_norm": 3.037932780161429,
"learning_rate": 7.92088309182241e-06,
"logits/chosen": -0.3814396262168884,
"logits/rejected": -0.38753852248191833,
"logps/chosen": -0.5815099477767944,
"logps/rejected": -0.6666026711463928,
"loss": 1.294,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.5815099477767944,
"rewards/margins": 0.0850927084684372,
"rewards/rejected": -0.6666026711463928,
"step": 285
},
{
"epoch": 0.6199064796259185,
"grad_norm": 3.0961871877321485,
"learning_rate": 7.556595938621058e-06,
"logits/chosen": -0.46853989362716675,
"logits/rejected": -0.49206361174583435,
"logps/chosen": -0.6127103567123413,
"logps/rejected": -0.7595942616462708,
"loss": 1.2796,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6127103567123413,
"rewards/margins": 0.14688387513160706,
"rewards/rejected": -0.7595942616462708,
"step": 290
},
{
"epoch": 0.6305945223780896,
"grad_norm": 2.5558798414818837,
"learning_rate": 7.1957261069399745e-06,
"logits/chosen": -0.41997450590133667,
"logits/rejected": -0.41455715894699097,
"logps/chosen": -0.5393789410591125,
"logps/rejected": -0.6705390214920044,
"loss": 1.2859,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5393789410591125,
"rewards/margins": 0.13116006553173065,
"rewards/rejected": -0.6705390214920044,
"step": 295
},
{
"epoch": 0.6412825651302605,
"grad_norm": 4.199327343754458,
"learning_rate": 6.83877830586976e-06,
"logits/chosen": -0.4186905324459076,
"logits/rejected": -0.400879442691803,
"logps/chosen": -0.5945369601249695,
"logps/rejected": -0.8209554553031921,
"loss": 1.2546,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5945369601249695,
"rewards/margins": 0.22641853988170624,
"rewards/rejected": -0.8209554553031921,
"step": 300
},
{
"epoch": 0.6519706078824316,
"grad_norm": 3.5426555178859074,
"learning_rate": 6.486251759186573e-06,
"logits/chosen": -0.42441505193710327,
"logits/rejected": -0.4325898587703705,
"logps/chosen": -0.6070866584777832,
"logps/rejected": -0.7520343661308289,
"loss": 1.2695,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.6070866584777832,
"rewards/margins": 0.14494772255420685,
"rewards/rejected": -0.7520343661308289,
"step": 305
},
{
"epoch": 0.6626586506346025,
"grad_norm": 3.399076738544932,
"learning_rate": 6.138639507142539e-06,
"logits/chosen": -0.40451377630233765,
"logits/rejected": -0.41942930221557617,
"logps/chosen": -0.5666729211807251,
"logps/rejected": -0.685453474521637,
"loss": 1.2546,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.5666729211807251,
"rewards/margins": 0.11878061294555664,
"rewards/rejected": -0.685453474521637,
"step": 310
},
{
"epoch": 0.6733466933867736,
"grad_norm": 3.5429592529416407,
"learning_rate": 5.796427716904347e-06,
"logits/chosen": -0.3595660328865051,
"logits/rejected": -0.3509676158428192,
"logps/chosen": -0.6435049772262573,
"logps/rejected": -0.7910181879997253,
"loss": 1.2803,
"rewards/accuracies": 0.543749988079071,
"rewards/chosen": -0.6435049772262573,
"rewards/margins": 0.14751319587230682,
"rewards/rejected": -0.7910181879997253,
"step": 315
},
{
"epoch": 0.6840347361389446,
"grad_norm": 3.6719145043569146,
"learning_rate": 5.460095002604533e-06,
"logits/chosen": -0.3422035574913025,
"logits/rejected": -0.34498724341392517,
"logps/chosen": -0.5467932224273682,
"logps/rejected": -0.6095358729362488,
"loss": 1.2877,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5467932224273682,
"rewards/margins": 0.06274259090423584,
"rewards/rejected": -0.6095358729362488,
"step": 320
},
{
"epoch": 0.6947227788911156,
"grad_norm": 3.978020135849161,
"learning_rate": 5.130111755956327e-06,
"logits/chosen": -0.3481397032737732,
"logits/rejected": -0.34341806173324585,
"logps/chosen": -0.5532098412513733,
"logps/rejected": -0.7315355539321899,
"loss": 1.2661,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5532098412513733,
"rewards/margins": 0.17832568287849426,
"rewards/rejected": -0.7315355539321899,
"step": 325
},
{
"epoch": 0.7054108216432866,
"grad_norm": 3.442968094749812,
"learning_rate": 4.806939488368308e-06,
"logits/chosen": -0.4118443429470062,
"logits/rejected": -0.4033087193965912,
"logps/chosen": -0.5545328855514526,
"logps/rejected": -0.7638643383979797,
"loss": 1.2502,
"rewards/accuracies": 0.581250011920929,
"rewards/chosen": -0.5545328855514526,
"rewards/margins": 0.2093314677476883,
"rewards/rejected": -0.7638643383979797,
"step": 330
},
{
"epoch": 0.7160988643954576,
"grad_norm": 3.693118587496843,
"learning_rate": 4.491030185478976e-06,
"logits/chosen": -0.3698023855686188,
"logits/rejected": -0.3537530303001404,
"logps/chosen": -0.5946871042251587,
"logps/rejected": -0.9203634262084961,
"loss": 1.2694,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.5946871042251587,
"rewards/margins": 0.32567623257637024,
"rewards/rejected": -0.9203634262084961,
"step": 335
},
{
"epoch": 0.7267869071476286,
"grad_norm": 3.528628571085271,
"learning_rate": 4.182825675013945e-06,
"logits/chosen": -0.4216648042201996,
"logits/rejected": -0.4200260639190674,
"logps/chosen": -0.5487005710601807,
"logps/rejected": -0.6899628639221191,
"loss": 1.2749,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.5487005710601807,
"rewards/margins": 0.14126229286193848,
"rewards/rejected": -0.6899628639221191,
"step": 340
},
{
"epoch": 0.7374749498997996,
"grad_norm": 3.8114111942749536,
"learning_rate": 3.882757008849936e-06,
"logits/chosen": -0.41083043813705444,
"logits/rejected": -0.4161206781864166,
"logps/chosen": -0.575046181678772,
"logps/rejected": -0.7536992430686951,
"loss": 1.2744,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.575046181678772,
"rewards/margins": 0.17865313589572906,
"rewards/rejected": -0.7536992430686951,
"step": 345
},
{
"epoch": 0.7481629926519706,
"grad_norm": 4.0438053008699315,
"learning_rate": 3.591243860149759e-06,
"logits/chosen": -0.3628717064857483,
"logits/rejected": -0.3853951096534729,
"logps/chosen": -0.5914100408554077,
"logps/rejected": -0.7692973613739014,
"loss": 1.2695,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5914100408554077,
"rewards/margins": 0.17788729071617126,
"rewards/rejected": -0.7692973613739014,
"step": 350
},
{
"epoch": 0.7588510354041417,
"grad_norm": 3.0837058616543813,
"learning_rate": 3.308693936411421e-06,
"logits/chosen": -0.41618281602859497,
"logits/rejected": -0.41979169845581055,
"logps/chosen": -0.6286286115646362,
"logps/rejected": -0.8485868573188782,
"loss": 1.2669,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.6286286115646362,
"rewards/margins": 0.2199581414461136,
"rewards/rejected": -0.8485868573188782,
"step": 355
},
{
"epoch": 0.7695390781563126,
"grad_norm": 3.528809280951808,
"learning_rate": 3.0355024092523334e-06,
"logits/chosen": -0.38231906294822693,
"logits/rejected": -0.40169796347618103,
"logps/chosen": -0.5484600067138672,
"logps/rejected": -0.6940116882324219,
"loss": 1.2726,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.5484600067138672,
"rewards/margins": 0.14555169641971588,
"rewards/rejected": -0.6940116882324219,
"step": 360
},
{
"epoch": 0.7802271209084837,
"grad_norm": 3.5483575752162406,
"learning_rate": 2.7720513617260857e-06,
"logits/chosen": -0.44837936758995056,
"logits/rejected": -0.4736010432243347,
"logps/chosen": -0.6057429313659668,
"logps/rejected": -0.7651041746139526,
"loss": 1.2659,
"rewards/accuracies": 0.6312500238418579,
"rewards/chosen": -0.6057429313659668,
"rewards/margins": 0.15936115384101868,
"rewards/rejected": -0.7651041746139526,
"step": 365
},
{
"epoch": 0.7909151636606546,
"grad_norm": 3.7870821722349985,
"learning_rate": 2.51870925394473e-06,
"logits/chosen": -0.43106943368911743,
"logits/rejected": -0.44449687004089355,
"logps/chosen": -0.5907775163650513,
"logps/rejected": -0.7539359331130981,
"loss": 1.2464,
"rewards/accuracies": 0.7250000238418579,
"rewards/chosen": -0.5907775163650513,
"rewards/margins": 0.1631583422422409,
"rewards/rejected": -0.7539359331130981,
"step": 370
},
{
"epoch": 0.8016032064128257,
"grad_norm": 3.7264373621641713,
"learning_rate": 2.275830407754006e-06,
"logits/chosen": -0.4176546633243561,
"logits/rejected": -0.4209866523742676,
"logps/chosen": -0.6014735102653503,
"logps/rejected": -0.7754963040351868,
"loss": 1.2615,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.6014735102653503,
"rewards/margins": 0.1740228235721588,
"rewards/rejected": -0.7754963040351868,
"step": 375
},
{
"epoch": 0.8122912491649966,
"grad_norm": 3.211374199470538,
"learning_rate": 2.043754511182191e-06,
"logits/chosen": -0.42597073316574097,
"logits/rejected": -0.4239738881587982,
"logps/chosen": -0.5504560470581055,
"logps/rejected": -0.7494019269943237,
"loss": 1.2622,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5504560470581055,
"rewards/margins": 0.19894585013389587,
"rewards/rejected": -0.7494019269943237,
"step": 380
},
{
"epoch": 0.8229792919171677,
"grad_norm": 3.301850545380218,
"learning_rate": 1.8228061433556866e-06,
"logits/chosen": -0.408210426568985,
"logits/rejected": -0.41355451941490173,
"logps/chosen": -0.5664467811584473,
"logps/rejected": -0.7051645517349243,
"loss": 1.2613,
"rewards/accuracies": 0.6812499761581421,
"rewards/chosen": -0.5664467811584473,
"rewards/margins": 0.13871778547763824,
"rewards/rejected": -0.7051645517349243,
"step": 385
},
{
"epoch": 0.8336673346693386,
"grad_norm": 3.141220152295375,
"learning_rate": 1.6132943205457607e-06,
"logits/chosen": -0.41336697340011597,
"logits/rejected": -0.4139551520347595,
"logps/chosen": -0.556975245475769,
"logps/rejected": -0.6668010950088501,
"loss": 1.257,
"rewards/accuracies": 0.606249988079071,
"rewards/chosen": -0.556975245475769,
"rewards/margins": 0.10982589423656464,
"rewards/rejected": -0.6668010950088501,
"step": 390
},
{
"epoch": 0.8443553774215097,
"grad_norm": 4.402129582330891,
"learning_rate": 1.4155120639813392e-06,
"logits/chosen": -0.4578641355037689,
"logits/rejected": -0.4619896411895752,
"logps/chosen": -0.5758166909217834,
"logps/rejected": -0.6729660630226135,
"loss": 1.271,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5758166909217834,
"rewards/margins": 0.09714937210083008,
"rewards/rejected": -0.6729660630226135,
"step": 395
},
{
"epoch": 0.8550434201736807,
"grad_norm": 3.440496458892442,
"learning_rate": 1.2297359900323346e-06,
"logits/chosen": -0.48626774549484253,
"logits/rejected": -0.48332077264785767,
"logps/chosen": -0.6083348989486694,
"logps/rejected": -0.7751791477203369,
"loss": 1.2571,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.6083348989486694,
"rewards/margins": 0.1668442189693451,
"rewards/rejected": -0.7751791477203369,
"step": 400
},
{
"epoch": 0.8550434201736807,
"eval_logits/chosen": -0.3830186724662781,
"eval_logits/rejected": -0.3938583731651306,
"eval_logps/chosen": -0.5791661143302917,
"eval_logps/rejected": -0.7269699573516846,
"eval_loss": 1.2552496194839478,
"eval_rewards/accuracies": 0.6626016497612,
"eval_rewards/chosen": -0.5791661143302917,
"eval_rewards/margins": 0.14780384302139282,
"eval_rewards/rejected": -0.7269699573516846,
"eval_runtime": 426.1833,
"eval_samples_per_second": 4.601,
"eval_steps_per_second": 0.289,
"step": 400
},
{
"epoch": 0.8657314629258517,
"grad_norm": 3.9745302359804944,
"learning_rate": 1.0562259233366334e-06,
"logits/chosen": -0.48229265213012695,
"logits/rejected": -0.5193549394607544,
"logps/chosen": -0.6197930574417114,
"logps/rejected": -0.764507532119751,
"loss": 1.2391,
"rewards/accuracies": 0.643750011920929,
"rewards/chosen": -0.6197930574417114,
"rewards/margins": 0.14471451938152313,
"rewards/rejected": -0.764507532119751,
"step": 405
},
{
"epoch": 0.8764195056780227,
"grad_norm": 3.419466411083352,
"learning_rate": 8.952245334118415e-07,
"logits/chosen": -0.4593311846256256,
"logits/rejected": -0.4776662290096283,
"logps/chosen": -0.5977698564529419,
"logps/rejected": -0.7118576765060425,
"loss": 1.2719,
"rewards/accuracies": 0.5625,
"rewards/chosen": -0.5977698564529419,
"rewards/margins": 0.11408782005310059,
"rewards/rejected": -0.7118576765060425,
"step": 410
},
{
"epoch": 0.8871075484301937,
"grad_norm": 3.431721917530868,
"learning_rate": 7.46956995260033e-07,
"logits/chosen": -0.4408165514469147,
"logits/rejected": -0.45343226194381714,
"logps/chosen": -0.5183443427085876,
"logps/rejected": -0.6987067461013794,
"loss": 1.251,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.5183443427085876,
"rewards/margins": 0.1803622990846634,
"rewards/rejected": -0.6987067461013794,
"step": 415
},
{
"epoch": 0.8977955911823647,
"grad_norm": 3.3661467211282274,
"learning_rate": 6.116306744401391e-07,
"logits/chosen": -0.4190942645072937,
"logits/rejected": -0.4262022376060486,
"logps/chosen": -0.5752037763595581,
"logps/rejected": -0.7899097204208374,
"loss": 1.2457,
"rewards/accuracies": 0.668749988079071,
"rewards/chosen": -0.5752037763595581,
"rewards/margins": 0.2147059142589569,
"rewards/rejected": -0.7899097204208374,
"step": 420
},
{
"epoch": 0.9084836339345357,
"grad_norm": 3.4491228706828267,
"learning_rate": 4.894348370484648e-07,
"logits/chosen": -0.3854565918445587,
"logits/rejected": -0.41155076026916504,
"logps/chosen": -0.5380185842514038,
"logps/rejected": -0.7508363127708435,
"loss": 1.2377,
"rewards/accuracies": 0.6875,
"rewards/chosen": -0.5380185842514038,
"rewards/margins": 0.21281783282756805,
"rewards/rejected": -0.7508363127708435,
"step": 425
},
{
"epoch": 0.9191716766867067,
"grad_norm": 3.846770405186625,
"learning_rate": 3.8054038501294077e-07,
"logits/chosen": -0.42405056953430176,
"logits/rejected": -0.4208356738090515,
"logps/chosen": -0.5479136109352112,
"logps/rejected": -0.7237703204154968,
"loss": 1.2312,
"rewards/accuracies": 0.6499999761581421,
"rewards/chosen": -0.5479136109352112,
"rewards/margins": 0.17585670948028564,
"rewards/rejected": -0.7237703204154968,
"step": 430
},
{
"epoch": 0.9298597194388778,
"grad_norm": 3.8565624225807036,
"learning_rate": 2.8509961707132496e-07,
"logits/chosen": -0.4456121325492859,
"logits/rejected": -0.4567331373691559,
"logps/chosen": -0.6145879030227661,
"logps/rejected": -0.7536846995353699,
"loss": 1.2534,
"rewards/accuracies": 0.612500011920929,
"rewards/chosen": -0.6145879030227661,
"rewards/margins": 0.13909675180912018,
"rewards/rejected": -0.7536846995353699,
"step": 435
},
{
"epoch": 0.9405477621910487,
"grad_norm": 3.992184128956239,
"learning_rate": 2.0324601576764525e-07,
"logits/chosen": -0.42855358123779297,
"logits/rejected": -0.44918686151504517,
"logps/chosen": -0.6049205660820007,
"logps/rejected": -0.7878492474555969,
"loss": 1.2545,
"rewards/accuracies": 0.637499988079071,
"rewards/chosen": -0.6049205660820007,
"rewards/margins": 0.1829286515712738,
"rewards/rejected": -0.7878492474555969,
"step": 440
},
{
"epoch": 0.9512358049432198,
"grad_norm": 3.818733547378664,
"learning_rate": 1.350940607647866e-07,
"logits/chosen": -0.3801869750022888,
"logits/rejected": -0.38896074891090393,
"logps/chosen": -0.5800012350082397,
"logps/rejected": -0.7279914617538452,
"loss": 1.2581,
"rewards/accuracies": 0.6187499761581421,
"rewards/chosen": -0.5800012350082397,
"rewards/margins": 0.14799019694328308,
"rewards/rejected": -0.7279914617538452,
"step": 445
},
{
"epoch": 0.9619238476953907,
"grad_norm": 3.601185509771085,
"learning_rate": 8.07390687343379e-08,
"logits/chosen": -0.4334098696708679,
"logits/rejected": -0.4463852047920227,
"logps/chosen": -0.589901864528656,
"logps/rejected": -0.768067479133606,
"loss": 1.2422,
"rewards/accuracies": 0.71875,
"rewards/chosen": -0.589901864528656,
"rewards/margins": 0.1781655251979828,
"rewards/rejected": -0.768067479133606,
"step": 450
},
{
"epoch": 0.9726118904475618,
"grad_norm": 3.7947367879452023,
"learning_rate": 4.025706004760932e-08,
"logits/chosen": -0.39869770407676697,
"logits/rejected": -0.39332014322280884,
"logps/chosen": -0.6198989152908325,
"logps/rejected": -0.8089291453361511,
"loss": 1.2643,
"rewards/accuracies": 0.706250011920929,
"rewards/chosen": -0.6198989152908325,
"rewards/margins": 0.18903018534183502,
"rewards/rejected": -0.8089291453361511,
"step": 455
},
{
"epoch": 0.9832999331997327,
"grad_norm": 3.5592218644016613,
"learning_rate": 1.370465245426167e-08,
"logits/chosen": -0.39606115221977234,
"logits/rejected": -0.4214437007904053,
"logps/chosen": -0.5898042917251587,
"logps/rejected": -0.7647892832756042,
"loss": 1.2623,
"rewards/accuracies": 0.6937500238418579,
"rewards/chosen": -0.5898042917251587,
"rewards/margins": 0.17498508095741272,
"rewards/rejected": -0.7647892832756042,
"step": 460
},
{
"epoch": 0.9939879759519038,
"grad_norm": 3.587979644510153,
"learning_rate": 1.1189818972656697e-09,
"logits/chosen": -0.3767338693141937,
"logits/rejected": -0.40465354919433594,
"logps/chosen": -0.5555558204650879,
"logps/rejected": -0.7796151041984558,
"loss": 1.2483,
"rewards/accuracies": 0.7124999761581421,
"rewards/chosen": -0.5555558204650879,
"rewards/margins": 0.2240593433380127,
"rewards/rejected": -0.7796151041984558,
"step": 465
},
{
"epoch": 0.9982631930527722,
"step": 467,
"total_flos": 0.0,
"train_loss": 1.2796013329728513,
"train_runtime": 21025.6961,
"train_samples_per_second": 2.848,
"train_steps_per_second": 0.022
}
],
"logging_steps": 5,
"max_steps": 467,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 1000000,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": true
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 2,
"trial_name": null,
"trial_params": null
}