ziansu's picture
Training in progress, step 750, checkpoint
afa1a63 verified
raw
history blame
49.6 kB
{
"best_metric": null,
"best_model_checkpoint": null,
"epoch": 1.1757896802040415,
"eval_steps": 50,
"global_step": 750,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.015695507161075144,
"grad_norm": 0.04355761408805847,
"learning_rate": 4.999451708687114e-06,
"logits/chosen": 14.845781326293945,
"logits/rejected": 14.576438903808594,
"logps/chosen": -0.31864267587661743,
"logps/rejected": -0.24545662105083466,
"loss": 1.0492,
"rewards/accuracies": 0.25,
"rewards/chosen": -0.47796401381492615,
"rewards/margins": -0.10977902263402939,
"rewards/rejected": -0.3681849539279938,
"step": 10
},
{
"epoch": 0.03139101432215029,
"grad_norm": 0.04919258877635002,
"learning_rate": 4.997807075247147e-06,
"logits/chosen": 15.27595043182373,
"logits/rejected": 14.872761726379395,
"logps/chosen": -0.3344747722148895,
"logps/rejected": -0.24258682131767273,
"loss": 1.0487,
"rewards/accuracies": 0.16249999403953552,
"rewards/chosen": -0.5017121434211731,
"rewards/margins": -0.1378319263458252,
"rewards/rejected": -0.3638802468776703,
"step": 20
},
{
"epoch": 0.047086521483225424,
"grad_norm": 0.049933061003685,
"learning_rate": 4.9950668210706795e-06,
"logits/chosen": 15.913165092468262,
"logits/rejected": 15.607622146606445,
"logps/chosen": -0.3440183997154236,
"logps/rejected": -0.2831566333770752,
"loss": 1.0405,
"rewards/accuracies": 0.1875,
"rewards/chosen": -0.516027569770813,
"rewards/margins": -0.09129264950752258,
"rewards/rejected": -0.4247349202632904,
"step": 30
},
{
"epoch": 0.06278202864430057,
"grad_norm": 0.05693503096699715,
"learning_rate": 4.9912321481237616e-06,
"logits/chosen": 15.402900695800781,
"logits/rejected": 14.99272632598877,
"logps/chosen": -0.3297731578350067,
"logps/rejected": -0.2746916711330414,
"loss": 1.0369,
"rewards/accuracies": 0.21250000596046448,
"rewards/chosen": -0.49465981125831604,
"rewards/margins": -0.08262218534946442,
"rewards/rejected": -0.41203755140304565,
"step": 40
},
{
"epoch": 0.07847753580537571,
"grad_norm": 0.05467928573489189,
"learning_rate": 4.986304738420684e-06,
"logits/chosen": 15.64543342590332,
"logits/rejected": 15.632547378540039,
"logps/chosen": -0.30952686071395874,
"logps/rejected": -0.24847058951854706,
"loss": 1.0367,
"rewards/accuracies": 0.21250000596046448,
"rewards/chosen": -0.46429023146629333,
"rewards/margins": -0.09158438444137573,
"rewards/rejected": -0.37270587682724,
"step": 50
},
{
"epoch": 0.07847753580537571,
"eval_logits/chosen": 15.850138664245605,
"eval_logits/rejected": 15.368529319763184,
"eval_logps/chosen": -0.3222965598106384,
"eval_logps/rejected": -0.26877468824386597,
"eval_loss": 1.0326261520385742,
"eval_rewards/accuracies": 0.26923078298568726,
"eval_rewards/chosen": -0.4834447503089905,
"eval_rewards/margins": -0.08028276264667511,
"eval_rewards/rejected": -0.40316200256347656,
"eval_runtime": 14.5044,
"eval_samples_per_second": 28.405,
"eval_steps_per_second": 3.585,
"step": 50
},
{
"epoch": 0.09417304296645085,
"grad_norm": 0.06174452602863312,
"learning_rate": 4.980286753286196e-06,
"logits/chosen": 15.443066596984863,
"logits/rejected": 15.192205429077148,
"logps/chosen": -0.31090402603149414,
"logps/rejected": -0.26281923055648804,
"loss": 1.04,
"rewards/accuracies": 0.25,
"rewards/chosen": -0.466356098651886,
"rewards/margins": -0.07212716341018677,
"rewards/rejected": -0.39422887563705444,
"step": 60
},
{
"epoch": 0.109868550127526,
"grad_norm": 0.06952528655529022,
"learning_rate": 4.973180832407471e-06,
"logits/chosen": 16.024200439453125,
"logits/rejected": 15.82934284210205,
"logps/chosen": -0.348227322101593,
"logps/rejected": -0.26220566034317017,
"loss": 1.043,
"rewards/accuracies": 0.21250000596046448,
"rewards/chosen": -0.5223408937454224,
"rewards/margins": -0.1290324479341507,
"rewards/rejected": -0.39330852031707764,
"step": 70
},
{
"epoch": 0.12556405728860115,
"grad_norm": 0.07572082430124283,
"learning_rate": 4.964990092676263e-06,
"logits/chosen": 15.884915351867676,
"logits/rejected": 15.603845596313477,
"logps/chosen": -0.34849274158477783,
"logps/rejected": -0.26585355401039124,
"loss": 1.0285,
"rewards/accuracies": 0.1875,
"rewards/chosen": -0.5227391719818115,
"rewards/margins": -0.12395882606506348,
"rewards/rejected": -0.39878037571907043,
"step": 80
},
{
"epoch": 0.14125956444967627,
"grad_norm": 0.2423778474330902,
"learning_rate": 4.9557181268217225e-06,
"logits/chosen": 15.978216171264648,
"logits/rejected": 15.76471996307373,
"logps/chosen": -0.327436238527298,
"logps/rejected": -0.25457051396369934,
"loss": 1.03,
"rewards/accuracies": 0.22499999403953552,
"rewards/chosen": -0.49115434288978577,
"rewards/margins": -0.10929858684539795,
"rewards/rejected": -0.3818557560443878,
"step": 90
},
{
"epoch": 0.15695507161075142,
"grad_norm": 0.1594536453485489,
"learning_rate": 4.9453690018345144e-06,
"logits/chosen": 16.307537078857422,
"logits/rejected": 16.138330459594727,
"logps/chosen": -0.3342314660549164,
"logps/rejected": -0.27582648396492004,
"loss": 1.0309,
"rewards/accuracies": 0.26249998807907104,
"rewards/chosen": -0.5013472437858582,
"rewards/margins": -0.0876075029373169,
"rewards/rejected": -0.41373974084854126,
"step": 100
},
{
"epoch": 0.15695507161075142,
"eval_logits/chosen": 16.4310245513916,
"eval_logits/rejected": 15.98912525177002,
"eval_logps/chosen": -0.3239763677120209,
"eval_logps/rejected": -0.28784558176994324,
"eval_loss": 1.020836353302002,
"eval_rewards/accuracies": 0.3076923191547394,
"eval_rewards/chosen": -0.4859645664691925,
"eval_rewards/margins": -0.054196178913116455,
"eval_rewards/rejected": -0.43176835775375366,
"eval_runtime": 14.5049,
"eval_samples_per_second": 28.404,
"eval_steps_per_second": 3.585,
"step": 100
},
{
"epoch": 0.17265057877182657,
"grad_norm": 0.07431349903345108,
"learning_rate": 4.933947257182901e-06,
"logits/chosen": 16.56686782836914,
"logits/rejected": 16.093189239501953,
"logps/chosen": -0.34455060958862305,
"logps/rejected": -0.2834388315677643,
"loss": 1.0388,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.5168259739875793,
"rewards/margins": -0.09166768193244934,
"rewards/rejected": -0.4251582622528076,
"step": 110
},
{
"epoch": 0.1883460859329017,
"grad_norm": 0.08802352845668793,
"learning_rate": 4.921457902821578e-06,
"logits/chosen": 16.50200843811035,
"logits/rejected": 16.286388397216797,
"logps/chosen": -0.30845317244529724,
"logps/rejected": -0.2677682936191559,
"loss": 1.0247,
"rewards/accuracies": 0.25,
"rewards/chosen": -0.4626797139644623,
"rewards/margins": -0.06102731078863144,
"rewards/rejected": -0.40165242552757263,
"step": 120
},
{
"epoch": 0.20404159309397685,
"grad_norm": 0.10464702546596527,
"learning_rate": 4.907906416994146e-06,
"logits/chosen": 16.163082122802734,
"logits/rejected": 16.158031463623047,
"logps/chosen": -0.3138599991798401,
"logps/rejected": -0.28097471594810486,
"loss": 1.0169,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.47078999876976013,
"rewards/margins": -0.04932791367173195,
"rewards/rejected": -0.4214620590209961,
"step": 130
},
{
"epoch": 0.219737100255052,
"grad_norm": 0.16971275210380554,
"learning_rate": 4.893298743830168e-06,
"logits/chosen": 16.28864860534668,
"logits/rejected": 16.151805877685547,
"logps/chosen": -0.3283368945121765,
"logps/rejected": -0.2850198745727539,
"loss": 0.9964,
"rewards/accuracies": 0.3499999940395355,
"rewards/chosen": -0.49250537157058716,
"rewards/margins": -0.06497551500797272,
"rewards/rejected": -0.42752987146377563,
"step": 140
},
{
"epoch": 0.23543260741612712,
"grad_norm": 0.18377964198589325,
"learning_rate": 4.8776412907378845e-06,
"logits/chosen": 16.890087127685547,
"logits/rejected": 16.42388153076172,
"logps/chosen": -0.33256903290748596,
"logps/rejected": -0.2939595878124237,
"loss": 1.0073,
"rewards/accuracies": 0.30000001192092896,
"rewards/chosen": -0.49885353446006775,
"rewards/margins": -0.057914119213819504,
"rewards/rejected": -0.44093936681747437,
"step": 150
},
{
"epoch": 0.23543260741612712,
"eval_logits/chosen": 16.833438873291016,
"eval_logits/rejected": 16.328977584838867,
"eval_logps/chosen": -0.32567569613456726,
"eval_logps/rejected": -0.35700783133506775,
"eval_loss": 0.9802881479263306,
"eval_rewards/accuracies": 0.42307692766189575,
"eval_rewards/chosen": -0.4885135293006897,
"eval_rewards/margins": 0.04699822515249252,
"eval_rewards/rejected": -0.5355117321014404,
"eval_runtime": 14.5005,
"eval_samples_per_second": 28.413,
"eval_steps_per_second": 3.586,
"step": 150
},
{
"epoch": 0.2511281145772023,
"grad_norm": 0.12049826234579086,
"learning_rate": 4.860940925593703e-06,
"logits/chosen": 16.505878448486328,
"logits/rejected": 16.178979873657227,
"logps/chosen": -0.3397011458873749,
"logps/rejected": -0.35640352964401245,
"loss": 0.9795,
"rewards/accuracies": 0.4124999940395355,
"rewards/chosen": -0.5095517039299011,
"rewards/margins": 0.0250535998493433,
"rewards/rejected": -0.5346053242683411,
"step": 160
},
{
"epoch": 0.2668236217382774,
"grad_norm": 0.09485407918691635,
"learning_rate": 4.84320497372973e-06,
"logits/chosen": 16.245588302612305,
"logits/rejected": 15.922958374023438,
"logps/chosen": -0.29733315110206604,
"logps/rejected": -0.3461209237575531,
"loss": 0.9694,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.44599977135658264,
"rewards/margins": 0.0731816366314888,
"rewards/rejected": -0.5191814303398132,
"step": 170
},
{
"epoch": 0.28251912889935255,
"grad_norm": 0.155483216047287,
"learning_rate": 4.824441214720629e-06,
"logits/chosen": 16.339645385742188,
"logits/rejected": 16.115110397338867,
"logps/chosen": -0.3076801002025604,
"logps/rejected": -0.3655286729335785,
"loss": 0.9488,
"rewards/accuracies": 0.4625000059604645,
"rewards/chosen": -0.4615201950073242,
"rewards/margins": 0.0867728441953659,
"rewards/rejected": -0.5482929944992065,
"step": 180
},
{
"epoch": 0.2982146360604277,
"grad_norm": 0.21345795691013336,
"learning_rate": 4.804657878971252e-06,
"logits/chosen": 16.491886138916016,
"logits/rejected": 16.376684188842773,
"logps/chosen": -0.32437095046043396,
"logps/rejected": -0.39715105295181274,
"loss": 0.9423,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.4865564703941345,
"rewards/margins": 0.10917013883590698,
"rewards/rejected": -0.5957266092300415,
"step": 190
},
{
"epoch": 0.31391014322150285,
"grad_norm": 0.17633090913295746,
"learning_rate": 4.783863644106502e-06,
"logits/chosen": 16.6339168548584,
"logits/rejected": 16.79404640197754,
"logps/chosen": -0.33018192648887634,
"logps/rejected": -0.384196400642395,
"loss": 0.939,
"rewards/accuracies": 0.42500001192092896,
"rewards/chosen": -0.4952728748321533,
"rewards/margins": 0.0810217633843422,
"rewards/rejected": -0.5762946009635925,
"step": 200
},
{
"epoch": 0.31391014322150285,
"eval_logits/chosen": 17.17803192138672,
"eval_logits/rejected": 16.59328269958496,
"eval_logps/chosen": -0.33600664138793945,
"eval_logps/rejected": -0.47861453890800476,
"eval_loss": 0.9303967356681824,
"eval_rewards/accuracies": 0.4615384638309479,
"eval_rewards/chosen": -0.5040098428726196,
"eval_rewards/margins": 0.21391186118125916,
"eval_rewards/rejected": -0.717921793460846,
"eval_runtime": 14.5,
"eval_samples_per_second": 28.414,
"eval_steps_per_second": 3.586,
"step": 200
},
{
"epoch": 0.329605650382578,
"grad_norm": 0.1562221795320511,
"learning_rate": 4.762067631165049e-06,
"logits/chosen": 16.597665786743164,
"logits/rejected": 16.3507022857666,
"logps/chosen": -0.33709320425987244,
"logps/rejected": -0.43131333589553833,
"loss": 0.9107,
"rewards/accuracies": 0.4749999940395355,
"rewards/chosen": -0.5056397914886475,
"rewards/margins": 0.14133022725582123,
"rewards/rejected": -0.6469700932502747,
"step": 210
},
{
"epoch": 0.34530115754365315,
"grad_norm": 0.17680124938488007,
"learning_rate": 4.7392794005985324e-06,
"logits/chosen": 16.873676300048828,
"logits/rejected": 16.61945152282715,
"logps/chosen": -0.35083168745040894,
"logps/rejected": -0.537697434425354,
"loss": 0.9094,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.526247501373291,
"rewards/margins": 0.2802986204624176,
"rewards/rejected": -0.8065462112426758,
"step": 220
},
{
"epoch": 0.3609966647047283,
"grad_norm": 0.2662070393562317,
"learning_rate": 4.715508948078037e-06,
"logits/chosen": 16.803974151611328,
"logits/rejected": 16.698320388793945,
"logps/chosen": -0.3500753343105316,
"logps/rejected": -0.5399882793426514,
"loss": 0.8871,
"rewards/accuracies": 0.5249999761581421,
"rewards/chosen": -0.5251129865646362,
"rewards/margins": 0.2848694324493408,
"rewards/rejected": -0.8099824786186218,
"step": 230
},
{
"epoch": 0.3766921718658034,
"grad_norm": 0.17161667346954346,
"learning_rate": 4.690766700109659e-06,
"logits/chosen": 16.67904281616211,
"logits/rejected": 16.410579681396484,
"logps/chosen": -0.33549198508262634,
"logps/rejected": -0.4875665605068207,
"loss": 0.9037,
"rewards/accuracies": 0.6000000238418579,
"rewards/chosen": -0.5032380223274231,
"rewards/margins": 0.22811183333396912,
"rewards/rejected": -0.7313498258590698,
"step": 240
},
{
"epoch": 0.39238767902687854,
"grad_norm": 1.106021761894226,
"learning_rate": 4.665063509461098e-06,
"logits/chosen": 17.208127975463867,
"logits/rejected": 16.654085159301758,
"logps/chosen": -0.3752726912498474,
"logps/rejected": -0.4947708249092102,
"loss": 0.8606,
"rewards/accuracies": 0.4375,
"rewards/chosen": -0.5629090070724487,
"rewards/margins": 0.1792471706867218,
"rewards/rejected": -0.7421562671661377,
"step": 250
},
{
"epoch": 0.39238767902687854,
"eval_logits/chosen": 17.32963752746582,
"eval_logits/rejected": 16.589412689208984,
"eval_logps/chosen": -0.37825876474380493,
"eval_logps/rejected": -0.9001243114471436,
"eval_loss": 0.8168494701385498,
"eval_rewards/accuracies": 0.5192307829856873,
"eval_rewards/chosen": -0.5673881769180298,
"eval_rewards/margins": 0.7827982306480408,
"eval_rewards/rejected": -1.3501865863800049,
"eval_runtime": 14.5053,
"eval_samples_per_second": 28.403,
"eval_steps_per_second": 3.585,
"step": 250
},
{
"epoch": 0.4080831861879537,
"grad_norm": 0.29772138595581055,
"learning_rate": 4.638410650401267e-06,
"logits/chosen": 16.713775634765625,
"logits/rejected": 16.35211944580078,
"logps/chosen": -0.3877524733543396,
"logps/rejected": -0.8511163592338562,
"loss": 0.8138,
"rewards/accuracies": 0.44999998807907104,
"rewards/chosen": -0.581628680229187,
"rewards/margins": 0.6950457692146301,
"rewards/rejected": -1.276674509048462,
"step": 260
},
{
"epoch": 0.42377869334902885,
"grad_norm": 0.31860050559043884,
"learning_rate": 4.610819813755038e-06,
"logits/chosen": 17.09469985961914,
"logits/rejected": 16.5472412109375,
"logps/chosen": -0.47504258155822754,
"logps/rejected": -1.3266533613204956,
"loss": 0.7318,
"rewards/accuracies": 0.512499988079071,
"rewards/chosen": -0.7125638723373413,
"rewards/margins": 1.2774161100387573,
"rewards/rejected": -1.9899799823760986,
"step": 270
},
{
"epoch": 0.439474200510104,
"grad_norm": 0.6508163809776306,
"learning_rate": 4.582303101775249e-06,
"logits/chosen": 17.11003303527832,
"logits/rejected": 16.6564998626709,
"logps/chosen": -0.49357643723487854,
"logps/rejected": -1.4481580257415771,
"loss": 0.7599,
"rewards/accuracies": 0.6625000238418579,
"rewards/chosen": -0.7403645515441895,
"rewards/margins": 1.4318726062774658,
"rewards/rejected": -2.1722371578216553,
"step": 280
},
{
"epoch": 0.45516970767117915,
"grad_norm": 0.32430580258369446,
"learning_rate": 4.55287302283426e-06,
"logits/chosen": 16.830989837646484,
"logits/rejected": 16.384944915771484,
"logps/chosen": -0.593712329864502,
"logps/rejected": -1.7630856037139893,
"loss": 0.7336,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -0.8905684351921082,
"rewards/margins": 1.7540600299835205,
"rewards/rejected": -2.6446282863616943,
"step": 290
},
{
"epoch": 0.47086521483225424,
"grad_norm": 0.8555932641029358,
"learning_rate": 4.522542485937369e-06,
"logits/chosen": 16.72231674194336,
"logits/rejected": 16.28726577758789,
"logps/chosen": -0.5670709609985352,
"logps/rejected": -2.0420775413513184,
"loss": 0.6861,
"rewards/accuracies": 0.75,
"rewards/chosen": -0.8506065607070923,
"rewards/margins": 2.2125096321105957,
"rewards/rejected": -3.0631160736083984,
"step": 300
},
{
"epoch": 0.47086521483225424,
"eval_logits/chosen": 17.13121795654297,
"eval_logits/rejected": 16.268341064453125,
"eval_logps/chosen": -0.6842947602272034,
"eval_logps/rejected": -2.119321584701538,
"eval_loss": 0.7583853602409363,
"eval_rewards/accuracies": 0.75,
"eval_rewards/chosen": -1.026442289352417,
"eval_rewards/margins": 2.1525399684906006,
"eval_rewards/rejected": -3.1789822578430176,
"eval_runtime": 14.5007,
"eval_samples_per_second": 28.412,
"eval_steps_per_second": 3.586,
"step": 300
},
{
"epoch": 0.4865607219933294,
"grad_norm": 0.6059070825576782,
"learning_rate": 4.491324795060491e-06,
"logits/chosen": 17.147808074951172,
"logits/rejected": 16.194652557373047,
"logps/chosen": -0.8036400079727173,
"logps/rejected": -2.289825201034546,
"loss": 0.7163,
"rewards/accuracies": 0.6875,
"rewards/chosen": -1.2054599523544312,
"rewards/margins": 2.229278087615967,
"rewards/rejected": -3.4347376823425293,
"step": 310
},
{
"epoch": 0.5022562291544046,
"grad_norm": 1.8073927164077759,
"learning_rate": 4.4592336433146e-06,
"logits/chosen": 16.94902992248535,
"logits/rejected": 16.066068649291992,
"logps/chosen": -1.198162317276001,
"logps/rejected": -2.2922632694244385,
"loss": 0.6894,
"rewards/accuracies": 0.675000011920929,
"rewards/chosen": -1.7972434759140015,
"rewards/margins": 1.6411516666412354,
"rewards/rejected": -3.4383950233459473,
"step": 320
},
{
"epoch": 0.5179517363154797,
"grad_norm": 3.746042490005493,
"learning_rate": 4.426283106939474e-06,
"logits/chosen": 16.31036376953125,
"logits/rejected": 15.991762161254883,
"logps/chosen": -1.6245781183242798,
"logps/rejected": -2.553597927093506,
"loss": 0.6607,
"rewards/accuracies": 0.7875000238418579,
"rewards/chosen": -2.4368672370910645,
"rewards/margins": 1.3935294151306152,
"rewards/rejected": -3.830397129058838,
"step": 330
},
{
"epoch": 0.5336472434765548,
"grad_norm": 2.098111867904663,
"learning_rate": 4.3924876391293915e-06,
"logits/chosen": 16.59554672241211,
"logits/rejected": 15.915553092956543,
"logps/chosen": -2.183227062225342,
"logps/rejected": -3.4911434650421143,
"loss": 0.6381,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -3.274840831756592,
"rewards/margins": 1.9618743658065796,
"rewards/rejected": -5.236715316772461,
"step": 340
},
{
"epoch": 0.5493427506376299,
"grad_norm": 2.153958320617676,
"learning_rate": 4.357862063693486e-06,
"logits/chosen": 16.332544326782227,
"logits/rejected": 15.691922187805176,
"logps/chosen": -2.673710346221924,
"logps/rejected": -3.8687057495117188,
"loss": 0.5752,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -4.010565757751465,
"rewards/margins": 1.792493224143982,
"rewards/rejected": -5.8030595779418945,
"step": 350
},
{
"epoch": 0.5493427506376299,
"eval_logits/chosen": 16.272428512573242,
"eval_logits/rejected": 15.381678581237793,
"eval_logps/chosen": -3.0390572547912598,
"eval_logps/rejected": -4.695068836212158,
"eval_loss": 0.5928590893745422,
"eval_rewards/accuracies": 0.9230769276618958,
"eval_rewards/chosen": -4.5585856437683105,
"eval_rewards/margins": 2.4840168952941895,
"eval_rewards/rejected": -7.0426025390625,
"eval_runtime": 14.5008,
"eval_samples_per_second": 28.412,
"eval_steps_per_second": 3.586,
"step": 350
},
{
"epoch": 0.5650382577987051,
"grad_norm": 1.5755672454833984,
"learning_rate": 4.322421568553529e-06,
"logits/chosen": 15.589811325073242,
"logits/rejected": 14.780921936035156,
"logps/chosen": -3.069565534591675,
"logps/rejected": -4.581957817077637,
"loss": 0.5275,
"rewards/accuracies": 0.8125,
"rewards/chosen": -4.604348182678223,
"rewards/margins": 2.268587827682495,
"rewards/rejected": -6.872936248779297,
"step": 360
},
{
"epoch": 0.5807337649597802,
"grad_norm": 1.8776415586471558,
"learning_rate": 4.286181699082008e-06,
"logits/chosen": 15.172673225402832,
"logits/rejected": 14.400335311889648,
"logps/chosen": -3.9551639556884766,
"logps/rejected": -5.95252799987793,
"loss": 0.5113,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -5.932745933532715,
"rewards/margins": 2.996046543121338,
"rewards/rejected": -8.928792953491211,
"step": 370
},
{
"epoch": 0.5964292721208554,
"grad_norm": 1.5507289171218872,
"learning_rate": 4.249158351283414e-06,
"logits/chosen": 14.8423433303833,
"logits/rejected": 14.592633247375488,
"logps/chosen": -3.7946255207061768,
"logps/rejected": -5.3750481605529785,
"loss": 0.4862,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -5.691938877105713,
"rewards/margins": 2.370633125305176,
"rewards/rejected": -8.062570571899414,
"step": 380
},
{
"epoch": 0.6121247792819305,
"grad_norm": 3.4510324001312256,
"learning_rate": 4.211367764821722e-06,
"logits/chosen": 14.586801528930664,
"logits/rejected": 14.031987190246582,
"logps/chosen": -4.391470432281494,
"logps/rejected": -6.2107672691345215,
"loss": 0.5117,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -6.587205410003662,
"rewards/margins": 2.72894549369812,
"rewards/rejected": -9.316150665283203,
"step": 390
},
{
"epoch": 0.6278202864430057,
"grad_norm": 3.3816750049591064,
"learning_rate": 4.172826515897146e-06,
"logits/chosen": 14.523185729980469,
"logits/rejected": 14.266815185546875,
"logps/chosen": -4.090173244476318,
"logps/rejected": -5.582955837249756,
"loss": 0.5182,
"rewards/accuracies": 0.762499988079071,
"rewards/chosen": -6.135260581970215,
"rewards/margins": 2.2391738891601562,
"rewards/rejected": -8.374434471130371,
"step": 400
},
{
"epoch": 0.6278202864430057,
"eval_logits/chosen": 15.142684936523438,
"eval_logits/rejected": 14.101082801818848,
"eval_logps/chosen": -3.866588830947876,
"eval_logps/rejected": -6.100707530975342,
"eval_loss": 0.48699691891670227,
"eval_rewards/accuracies": 0.9615384340286255,
"eval_rewards/chosen": -5.7998833656311035,
"eval_rewards/margins": 3.351177215576172,
"eval_rewards/rejected": -9.15106201171875,
"eval_runtime": 14.5024,
"eval_samples_per_second": 28.409,
"eval_steps_per_second": 3.586,
"step": 400
},
{
"epoch": 0.6435157936040808,
"grad_norm": 1.845595359802246,
"learning_rate": 4.133551509975264e-06,
"logits/chosen": 14.54790210723877,
"logits/rejected": 13.707855224609375,
"logps/chosen": -4.112462043762207,
"logps/rejected": -6.583975315093994,
"loss": 0.5107,
"rewards/accuracies": 0.800000011920929,
"rewards/chosen": -6.168692111968994,
"rewards/margins": 3.707270383834839,
"rewards/rejected": -9.875962257385254,
"step": 410
},
{
"epoch": 0.659211300765156,
"grad_norm": 2.152916193008423,
"learning_rate": 4.093559974371725e-06,
"logits/chosen": 14.153576850891113,
"logits/rejected": 13.553201675415039,
"logps/chosen": -4.3408918380737305,
"logps/rejected": -6.828585147857666,
"loss": 0.41,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -6.511338233947754,
"rewards/margins": 3.7315402030944824,
"rewards/rejected": -10.242877960205078,
"step": 420
},
{
"epoch": 0.6749068079262311,
"grad_norm": 2.305433750152588,
"learning_rate": 4.052869450695776e-06,
"logits/chosen": 13.992170333862305,
"logits/rejected": 13.264738082885742,
"logps/chosen": -4.259932518005371,
"logps/rejected": -6.089646339416504,
"loss": 0.4122,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -6.389898777008057,
"rewards/margins": 2.7445709705352783,
"rewards/rejected": -9.134469985961914,
"step": 430
},
{
"epoch": 0.6906023150873063,
"grad_norm": 2.397674322128296,
"learning_rate": 4.011497787155938e-06,
"logits/chosen": 13.975759506225586,
"logits/rejected": 13.173799514770508,
"logps/chosen": -4.2483391761779785,
"logps/rejected": -6.487355709075928,
"loss": 0.4397,
"rewards/accuracies": 0.8125,
"rewards/chosen": -6.3725080490112305,
"rewards/margins": 3.3585267066955566,
"rewards/rejected": -9.731034278869629,
"step": 440
},
{
"epoch": 0.7062978222483814,
"grad_norm": 2.3987627029418945,
"learning_rate": 3.969463130731183e-06,
"logits/chosen": 13.777560234069824,
"logits/rejected": 12.936296463012695,
"logps/chosen": -3.86059832572937,
"logps/rejected": -6.495786190032959,
"loss": 0.444,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -5.790897846221924,
"rewards/margins": 3.952782392501831,
"rewards/rejected": -9.743680000305176,
"step": 450
},
{
"epoch": 0.7062978222483814,
"eval_logits/chosen": 14.397418975830078,
"eval_logits/rejected": 13.271268844604492,
"eval_logps/chosen": -3.9544241428375244,
"eval_logps/rejected": -6.69989013671875,
"eval_loss": 0.435256689786911,
"eval_rewards/accuracies": 0.9615384340286255,
"eval_rewards/chosen": -5.931635856628418,
"eval_rewards/margins": 4.118198871612549,
"eval_rewards/rejected": -10.049835205078125,
"eval_runtime": 14.5029,
"eval_samples_per_second": 28.408,
"eval_steps_per_second": 3.585,
"step": 450
},
{
"epoch": 0.7219933294094566,
"grad_norm": 2.3673465251922607,
"learning_rate": 3.92678391921108e-06,
"logits/chosen": 13.67004680633545,
"logits/rejected": 12.689542770385742,
"logps/chosen": -4.037893295288086,
"logps/rejected": -6.727287292480469,
"loss": 0.3858,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -6.056839942932129,
"rewards/margins": 4.034090995788574,
"rewards/rejected": -10.09093189239502,
"step": 460
},
{
"epoch": 0.7376888365705316,
"grad_norm": 2.9868297576904297,
"learning_rate": 3.88347887310836e-06,
"logits/chosen": 13.124593734741211,
"logits/rejected": 12.576837539672852,
"logps/chosen": -4.324252605438232,
"logps/rejected": -6.761715888977051,
"loss": 0.363,
"rewards/accuracies": 0.862500011920929,
"rewards/chosen": -6.486379146575928,
"rewards/margins": 3.6561942100524902,
"rewards/rejected": -10.142572402954102,
"step": 470
},
{
"epoch": 0.7533843437316068,
"grad_norm": 2.613318920135498,
"learning_rate": 3.839566987447492e-06,
"logits/chosen": 13.514410018920898,
"logits/rejected": 12.838116645812988,
"logps/chosen": -4.333093166351318,
"logps/rejected": -6.704646110534668,
"loss": 0.3274,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -6.499639987945557,
"rewards/margins": 3.5573298931121826,
"rewards/rejected": -10.056970596313477,
"step": 480
},
{
"epoch": 0.7690798508926819,
"grad_norm": 2.1568610668182373,
"learning_rate": 3.795067523432826e-06,
"logits/chosen": 13.456674575805664,
"logits/rejected": 12.37132453918457,
"logps/chosen": -4.630190849304199,
"logps/rejected": -7.871635437011719,
"loss": 0.3822,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -6.945285797119141,
"rewards/margins": 4.862167835235596,
"rewards/rejected": -11.807454109191895,
"step": 490
},
{
"epoch": 0.7847753580537571,
"grad_norm": 2.516496419906616,
"learning_rate": 3.7500000000000005e-06,
"logits/chosen": 13.195734024047852,
"logits/rejected": 12.557401657104492,
"logps/chosen": -3.9327914714813232,
"logps/rejected": -6.561183929443359,
"loss": 0.352,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -5.899188041687012,
"rewards/margins": 3.9425880908966064,
"rewards/rejected": -9.841775894165039,
"step": 500
},
{
"epoch": 0.7847753580537571,
"eval_logits/chosen": 13.891414642333984,
"eval_logits/rejected": 12.734375,
"eval_logps/chosen": -3.924809455871582,
"eval_logps/rejected": -7.003909587860107,
"eval_loss": 0.3888731002807617,
"eval_rewards/accuracies": 1.0,
"eval_rewards/chosen": -5.887214183807373,
"eval_rewards/margins": 4.618649482727051,
"eval_rewards/rejected": -10.505864143371582,
"eval_runtime": 14.5029,
"eval_samples_per_second": 28.408,
"eval_steps_per_second": 3.585,
"step": 500
},
{
"epoch": 0.8004708652148322,
"grad_norm": 5.4663896560668945,
"learning_rate": 3.7043841852542884e-06,
"logits/chosen": 12.861404418945312,
"logits/rejected": 11.9797945022583,
"logps/chosen": -4.245055198669434,
"logps/rejected": -7.063906669616699,
"loss": 0.4009,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -6.36758279800415,
"rewards/margins": 4.228276252746582,
"rewards/rejected": -10.59585952758789,
"step": 510
},
{
"epoch": 0.8161663723759074,
"grad_norm": 2.4403271675109863,
"learning_rate": 3.658240087799655e-06,
"logits/chosen": 12.84996223449707,
"logits/rejected": 12.256246566772461,
"logps/chosen": -4.2785773277282715,
"logps/rejected": -7.003395080566406,
"loss": 0.3327,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -6.417865753173828,
"rewards/margins": 4.0872273445129395,
"rewards/rejected": -10.505093574523926,
"step": 520
},
{
"epoch": 0.8318618795369825,
"grad_norm": 3.596749782562256,
"learning_rate": 3.611587947962319e-06,
"logits/chosen": 13.3360595703125,
"logits/rejected": 12.449459075927734,
"logps/chosen": -4.8148322105407715,
"logps/rejected": -7.221930503845215,
"loss": 0.3255,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -7.222248077392578,
"rewards/margins": 3.610647201538086,
"rewards/rejected": -10.832895278930664,
"step": 530
},
{
"epoch": 0.8475573866980577,
"grad_norm": 4.537969589233398,
"learning_rate": 3.564448228912682e-06,
"logits/chosen": 12.63983154296875,
"logits/rejected": 12.042104721069336,
"logps/chosen": -4.933573246002197,
"logps/rejected": -7.656388282775879,
"loss": 0.2941,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -7.400360107421875,
"rewards/margins": 4.084224224090576,
"rewards/rejected": -11.484583854675293,
"step": 540
},
{
"epoch": 0.8632528938591328,
"grad_norm": 2.4124038219451904,
"learning_rate": 3.516841607689501e-06,
"logits/chosen": 13.053291320800781,
"logits/rejected": 12.351530075073242,
"logps/chosen": -5.196786403656006,
"logps/rejected": -7.8461012840271,
"loss": 0.3906,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -7.795179843902588,
"rewards/margins": 3.9739716053009033,
"rewards/rejected": -11.76915168762207,
"step": 550
},
{
"epoch": 0.8632528938591328,
"eval_logits/chosen": 13.596405982971191,
"eval_logits/rejected": 12.482269287109375,
"eval_logps/chosen": -4.328949928283691,
"eval_logps/rejected": -7.602427959442139,
"eval_loss": 0.3577499985694885,
"eval_rewards/accuracies": 0.9807692170143127,
"eval_rewards/chosen": -6.493425369262695,
"eval_rewards/margins": 4.910217761993408,
"eval_rewards/rejected": -11.403642654418945,
"eval_runtime": 14.5032,
"eval_samples_per_second": 28.408,
"eval_steps_per_second": 3.585,
"step": 550
},
{
"epoch": 0.878948401020208,
"grad_norm": 1.42177414894104,
"learning_rate": 3.4687889661302577e-06,
"logits/chosen": 13.275362014770508,
"logits/rejected": 12.30308723449707,
"logps/chosen": -4.182652950286865,
"logps/rejected": -7.163094997406006,
"loss": 0.3692,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -6.273979663848877,
"rewards/margins": 4.470663547515869,
"rewards/rejected": -10.744643211364746,
"step": 560
},
{
"epoch": 0.8946439081812831,
"grad_norm": 2.431758403778076,
"learning_rate": 3.4203113817116955e-06,
"logits/chosen": 13.407594680786133,
"logits/rejected": 12.078115463256836,
"logps/chosen": -4.193841457366943,
"logps/rejected": -7.832788944244385,
"loss": 0.3487,
"rewards/accuracies": 0.9375,
"rewards/chosen": -6.290762901306152,
"rewards/margins": 5.458420753479004,
"rewards/rejected": -11.749183654785156,
"step": 570
},
{
"epoch": 0.9103394153423583,
"grad_norm": 2.549168348312378,
"learning_rate": 3.3714301183045382e-06,
"logits/chosen": 12.668423652648926,
"logits/rejected": 11.851642608642578,
"logps/chosen": -4.831603050231934,
"logps/rejected": -7.4626054763793945,
"loss": 0.3717,
"rewards/accuracies": 0.8500000238418579,
"rewards/chosen": -7.2474045753479,
"rewards/margins": 3.946502685546875,
"rewards/rejected": -11.193906784057617,
"step": 580
},
{
"epoch": 0.9260349225034334,
"grad_norm": 3.3875763416290283,
"learning_rate": 3.3221666168464584e-06,
"logits/chosen": 12.595063209533691,
"logits/rejected": 12.006935119628906,
"logps/chosen": -4.957076072692871,
"logps/rejected": -7.576946258544922,
"loss": 0.3553,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -7.435614109039307,
"rewards/margins": 3.9298064708709717,
"rewards/rejected": -11.3654203414917,
"step": 590
},
{
"epoch": 0.9417304296645085,
"grad_norm": 2.784954071044922,
"learning_rate": 3.272542485937369e-06,
"logits/chosen": 12.631284713745117,
"logits/rejected": 11.908422470092773,
"logps/chosen": -4.122859477996826,
"logps/rejected": -7.009615421295166,
"loss": 0.3086,
"rewards/accuracies": 0.875,
"rewards/chosen": -6.18428897857666,
"rewards/margins": 4.330134868621826,
"rewards/rejected": -10.514424324035645,
"step": 600
},
{
"epoch": 0.9417304296645085,
"eval_logits/chosen": 13.370320320129395,
"eval_logits/rejected": 12.268073081970215,
"eval_logps/chosen": -4.0377197265625,
"eval_logps/rejected": -7.431212425231934,
"eval_loss": 0.33698391914367676,
"eval_rewards/accuracies": 1.0,
"eval_rewards/chosen": -6.056579113006592,
"eval_rewards/margins": 5.090238094329834,
"eval_rewards/rejected": -11.146818161010742,
"eval_runtime": 14.5012,
"eval_samples_per_second": 28.411,
"eval_steps_per_second": 3.586,
"step": 600
},
{
"epoch": 0.9574259368255836,
"grad_norm": 3.2864303588867188,
"learning_rate": 3.222579492361179e-06,
"logits/chosen": 12.638185501098633,
"logits/rejected": 11.773119926452637,
"logps/chosen": -4.101125240325928,
"logps/rejected": -6.985491752624512,
"loss": 0.2886,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -6.1516876220703125,
"rewards/margins": 4.326549530029297,
"rewards/rejected": -10.47823715209961,
"step": 610
},
{
"epoch": 0.9731214439866588,
"grad_norm": 3.094020366668701,
"learning_rate": 3.1722995515381644e-06,
"logits/chosen": 12.71407413482666,
"logits/rejected": 11.668571472167969,
"logps/chosen": -4.560760498046875,
"logps/rejected": -7.878456115722656,
"loss": 0.294,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -6.841139793395996,
"rewards/margins": 4.976544380187988,
"rewards/rejected": -11.817686080932617,
"step": 620
},
{
"epoch": 0.9888169511477339,
"grad_norm": 1.906849980354309,
"learning_rate": 3.121724717912138e-06,
"logits/chosen": 12.863229751586914,
"logits/rejected": 11.391074180603027,
"logps/chosen": -4.578230857849121,
"logps/rejected": -8.216089248657227,
"loss": 0.2579,
"rewards/accuracies": 0.9624999761581421,
"rewards/chosen": -6.86734676361084,
"rewards/margins": 5.456785202026367,
"rewards/rejected": -12.324131965637207,
"step": 630
},
{
"epoch": 1.003139101432215,
"grad_norm": 2.699907064437866,
"learning_rate": 3.0708771752766397e-06,
"logits/chosen": 12.64507007598877,
"logits/rejected": 12.148189544677734,
"logps/chosen": -4.193073272705078,
"logps/rejected": -7.2082905769348145,
"loss": 0.2449,
"rewards/accuracies": 0.931506872177124,
"rewards/chosen": -6.289609909057617,
"rewards/margins": 4.522825241088867,
"rewards/rejected": -10.812435150146484,
"step": 640
},
{
"epoch": 1.0188346085932902,
"grad_norm": 3.4014647006988525,
"learning_rate": 3.019779227044398e-06,
"logits/chosen": 12.921002388000488,
"logits/rejected": 12.00908088684082,
"logps/chosen": -4.350411415100098,
"logps/rejected": -7.187474250793457,
"loss": 0.2976,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -6.5256171226501465,
"rewards/margins": 4.255594253540039,
"rewards/rejected": -10.781211853027344,
"step": 650
},
{
"epoch": 1.0188346085932902,
"eval_logits/chosen": 13.153624534606934,
"eval_logits/rejected": 12.077543258666992,
"eval_logps/chosen": -4.4063401222229,
"eval_logps/rejected": -8.090719223022461,
"eval_loss": 0.31872284412384033,
"eval_rewards/accuracies": 1.0,
"eval_rewards/chosen": -6.6095099449157715,
"eval_rewards/margins": 5.526568412780762,
"eval_rewards/rejected": -12.136077880859375,
"eval_runtime": 14.4992,
"eval_samples_per_second": 28.415,
"eval_steps_per_second": 3.586,
"step": 650
},
{
"epoch": 1.0345301157543654,
"grad_norm": 3.4320549964904785,
"learning_rate": 2.9684532864643123e-06,
"logits/chosen": 12.755729675292969,
"logits/rejected": 11.981709480285645,
"logps/chosen": -4.497554302215576,
"logps/rejected": -7.250922203063965,
"loss": 0.3013,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -6.746331691741943,
"rewards/margins": 4.13005256652832,
"rewards/rejected": -10.876383781433105,
"step": 660
},
{
"epoch": 1.0502256229154405,
"grad_norm": 5.130344867706299,
"learning_rate": 2.9169218667902562e-06,
"logits/chosen": 12.774751663208008,
"logits/rejected": 12.063162803649902,
"logps/chosen": -4.672727108001709,
"logps/rejected": -7.972691535949707,
"loss": 0.2677,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -7.009090423583984,
"rewards/margins": 4.94994592666626,
"rewards/rejected": -11.959035873413086,
"step": 670
},
{
"epoch": 1.0659211300765157,
"grad_norm": 2.1107585430145264,
"learning_rate": 2.8652075714060296e-06,
"logits/chosen": 12.401437759399414,
"logits/rejected": 11.329314231872559,
"logps/chosen": -4.372309684753418,
"logps/rejected": -7.968375205993652,
"loss": 0.2321,
"rewards/accuracies": 0.949999988079071,
"rewards/chosen": -6.558464050292969,
"rewards/margins": 5.394099235534668,
"rewards/rejected": -11.952563285827637,
"step": 680
},
{
"epoch": 1.0816166372375908,
"grad_norm": 2.09533429145813,
"learning_rate": 2.813333083910761e-06,
"logits/chosen": 12.615015029907227,
"logits/rejected": 11.613394737243652,
"logps/chosen": -4.582761764526367,
"logps/rejected": -7.808516502380371,
"loss": 0.2593,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -6.874144077301025,
"rewards/margins": 4.838631629943848,
"rewards/rejected": -11.712774276733398,
"step": 690
},
{
"epoch": 1.097312144398666,
"grad_norm": 2.652644157409668,
"learning_rate": 2.761321158169134e-06,
"logits/chosen": 12.447229385375977,
"logits/rejected": 11.710171699523926,
"logps/chosen": -4.722737789154053,
"logps/rejected": -8.122079849243164,
"loss": 0.263,
"rewards/accuracies": 0.9375,
"rewards/chosen": -7.084107398986816,
"rewards/margins": 5.0990142822265625,
"rewards/rejected": -12.183120727539062,
"step": 700
},
{
"epoch": 1.097312144398666,
"eval_logits/chosen": 13.170889854431152,
"eval_logits/rejected": 12.095331192016602,
"eval_logps/chosen": -4.405147075653076,
"eval_logps/rejected": -8.16016960144043,
"eval_loss": 0.3023497760295868,
"eval_rewards/accuracies": 1.0,
"eval_rewards/chosen": -6.607721328735352,
"eval_rewards/margins": 5.632532119750977,
"eval_rewards/rejected": -12.240253448486328,
"eval_runtime": 14.5027,
"eval_samples_per_second": 28.408,
"eval_steps_per_second": 3.586,
"step": 700
},
{
"epoch": 1.113007651559741,
"grad_norm": 2.3017961978912354,
"learning_rate": 2.70919460833079e-06,
"logits/chosen": 12.642290115356445,
"logits/rejected": 11.817375183105469,
"logps/chosen": -4.732531547546387,
"logps/rejected": -8.388056755065918,
"loss": 0.2707,
"rewards/accuracies": 0.925000011920929,
"rewards/chosen": -7.098797798156738,
"rewards/margins": 5.483288764953613,
"rewards/rejected": -12.582086563110352,
"step": 710
},
{
"epoch": 1.1287031587208163,
"grad_norm": 2.538917303085327,
"learning_rate": 2.6569762988232838e-06,
"logits/chosen": 12.873556137084961,
"logits/rejected": 11.74571704864502,
"logps/chosen": -5.0908284187316895,
"logps/rejected": -8.69284439086914,
"loss": 0.2443,
"rewards/accuracies": 0.9125000238418579,
"rewards/chosen": -7.636242866516113,
"rewards/margins": 5.403023719787598,
"rewards/rejected": -13.039266586303711,
"step": 720
},
{
"epoch": 1.1443986658818912,
"grad_norm": 2.7106313705444336,
"learning_rate": 2.604689134322999e-06,
"logits/chosen": 12.663686752319336,
"logits/rejected": 12.057671546936035,
"logps/chosen": -4.6052398681640625,
"logps/rejected": -7.64666748046875,
"loss": 0.2657,
"rewards/accuracies": 0.8999999761581421,
"rewards/chosen": -6.907858848571777,
"rewards/margins": 4.562141418457031,
"rewards/rejected": -11.470001220703125,
"step": 730
},
{
"epoch": 1.1600941730429664,
"grad_norm": 1.9821428060531616,
"learning_rate": 2.5523560497083927e-06,
"logits/chosen": 12.605002403259277,
"logits/rejected": 11.786115646362305,
"logps/chosen": -4.447022438049316,
"logps/rejected": -8.232423782348633,
"loss": 0.2234,
"rewards/accuracies": 0.9375,
"rewards/chosen": -6.670533180236816,
"rewards/margins": 5.678102016448975,
"rewards/rejected": -12.348634719848633,
"step": 740
},
{
"epoch": 1.1757896802040415,
"grad_norm": 2.469222068786621,
"learning_rate": 2.5e-06,
"logits/chosen": 12.427464485168457,
"logits/rejected": 11.602205276489258,
"logps/chosen": -4.620213985443115,
"logps/rejected": -8.116684913635254,
"loss": 0.2895,
"rewards/accuracies": 0.887499988079071,
"rewards/chosen": -6.93032169342041,
"rewards/margins": 5.244706153869629,
"rewards/rejected": -12.175026893615723,
"step": 750
},
{
"epoch": 1.1757896802040415,
"eval_logits/chosen": 13.117497444152832,
"eval_logits/rejected": 12.068766593933105,
"eval_logps/chosen": -4.3247389793396,
"eval_logps/rejected": -8.284286499023438,
"eval_loss": 0.2885077893733978,
"eval_rewards/accuracies": 1.0,
"eval_rewards/chosen": -6.48710823059082,
"eval_rewards/margins": 5.939322471618652,
"eval_rewards/rejected": -12.426431655883789,
"eval_runtime": 14.5056,
"eval_samples_per_second": 28.403,
"eval_steps_per_second": 3.585,
"step": 750
}
],
"logging_steps": 10,
"max_steps": 1500,
"num_input_tokens_seen": 0,
"num_train_epochs": 3,
"save_steps": 50,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 1.8163966579446907e+18,
"train_batch_size": 1,
"trial_name": null,
"trial_params": null
}