|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 500, |
|
"global_step": 625, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"learning_rate": 7.936507936507937e-08, |
|
"logits/chosen": 0.36046722531318665, |
|
"logits/rejected": 0.23892882466316223, |
|
"logps/chosen": -419.6329345703125, |
|
"logps/rejected": -437.8076171875, |
|
"loss": 0.3567, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.02, |
|
"learning_rate": 7.936507936507937e-07, |
|
"logits/chosen": 0.25086185336112976, |
|
"logits/rejected": 0.22883599996566772, |
|
"logps/chosen": -438.00592041015625, |
|
"logps/rejected": -430.2698059082031, |
|
"loss": 0.3631, |
|
"rewards/accuracies": 0.3333333432674408, |
|
"rewards/chosen": -0.00057529954938218, |
|
"rewards/margins": 4.002878631581552e-05, |
|
"rewards/rejected": -0.0006153282593004405, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"learning_rate": 1.5873015873015873e-06, |
|
"logits/chosen": 0.19798453152179718, |
|
"logits/rejected": 0.2455386370420456, |
|
"logps/chosen": -401.9042663574219, |
|
"logps/rejected": -393.48822021484375, |
|
"loss": 0.3746, |
|
"rewards/accuracies": 0.38749998807907104, |
|
"rewards/chosen": -0.0006311516626738012, |
|
"rewards/margins": 7.311334047699347e-05, |
|
"rewards/rejected": -0.0007042650249786675, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"learning_rate": 2.380952380952381e-06, |
|
"logits/chosen": 0.16958989202976227, |
|
"logits/rejected": 0.3218519687652588, |
|
"logps/chosen": -421.8275451660156, |
|
"logps/rejected": -406.69171142578125, |
|
"loss": 0.3735, |
|
"rewards/accuracies": 0.38749998807907104, |
|
"rewards/chosen": -0.0018598968163132668, |
|
"rewards/margins": 8.136472024489194e-05, |
|
"rewards/rejected": -0.0019412615802139044, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.06, |
|
"learning_rate": 3.1746031746031746e-06, |
|
"logits/chosen": 0.2768496870994568, |
|
"logits/rejected": 0.2004733383655548, |
|
"logps/chosen": -364.9704895019531, |
|
"logps/rejected": -383.6319274902344, |
|
"loss": 0.3722, |
|
"rewards/accuracies": 0.41874998807907104, |
|
"rewards/chosen": -0.0037128510884940624, |
|
"rewards/margins": 0.00042822357499971986, |
|
"rewards/rejected": -0.004141074605286121, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"learning_rate": 3.968253968253968e-06, |
|
"logits/chosen": 0.21148820221424103, |
|
"logits/rejected": 0.1991022676229477, |
|
"logps/chosen": -363.0907897949219, |
|
"logps/rejected": -372.4688415527344, |
|
"loss": 0.3683, |
|
"rewards/accuracies": 0.38749998807907104, |
|
"rewards/chosen": -0.008444704115390778, |
|
"rewards/margins": 0.000421268108766526, |
|
"rewards/rejected": -0.00886597204953432, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"learning_rate": 4.761904761904762e-06, |
|
"logits/chosen": 0.12682433426380157, |
|
"logits/rejected": 0.10598279535770416, |
|
"logps/chosen": -406.6046447753906, |
|
"logps/rejected": -408.136474609375, |
|
"loss": 0.3698, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.013907397165894508, |
|
"rewards/margins": 0.0012330651516094804, |
|
"rewards/rejected": -0.01514046173542738, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.11, |
|
"learning_rate": 4.998086282661188e-06, |
|
"logits/chosen": 0.15495741367340088, |
|
"logits/rejected": 0.2392895221710205, |
|
"logps/chosen": -454.59564208984375, |
|
"logps/rejected": -472.60150146484375, |
|
"loss": 0.3647, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.01984778419137001, |
|
"rewards/margins": 0.0015219683991745114, |
|
"rewards/rejected": -0.021369751542806625, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"learning_rate": 4.988720025682995e-06, |
|
"logits/chosen": 0.10600455850362778, |
|
"logits/rejected": 0.19427841901779175, |
|
"logps/chosen": -456.883544921875, |
|
"logps/rejected": -457.2091369628906, |
|
"loss": 0.366, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -0.027602821588516235, |
|
"rewards/margins": 0.0021790049504488707, |
|
"rewards/rejected": -0.029781833291053772, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.14, |
|
"learning_rate": 4.9715789537359126e-06, |
|
"logits/chosen": 0.19685187935829163, |
|
"logits/rejected": 0.1898462027311325, |
|
"logps/chosen": -429.89715576171875, |
|
"logps/rejected": -405.14947509765625, |
|
"loss": 0.374, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.02775459550321102, |
|
"rewards/margins": 0.0036151439417153597, |
|
"rewards/rejected": -0.03136973828077316, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.16, |
|
"learning_rate": 4.946716615897932e-06, |
|
"logits/chosen": 0.17538510262966156, |
|
"logits/rejected": 0.20797649025917053, |
|
"logps/chosen": -436.5021057128906, |
|
"logps/rejected": -441.6522521972656, |
|
"loss": 0.3647, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.030668649822473526, |
|
"rewards/margins": 0.0047523751854896545, |
|
"rewards/rejected": -0.03542102500796318, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"learning_rate": 4.9142106826480114e-06, |
|
"logits/chosen": 0.23190093040466309, |
|
"logits/rejected": 0.2070162296295166, |
|
"logps/chosen": -495.9847717285156, |
|
"logps/rejected": -481.325927734375, |
|
"loss": 0.36, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.030186587944626808, |
|
"rewards/margins": 0.006671676877886057, |
|
"rewards/rejected": -0.036858268082141876, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.19, |
|
"learning_rate": 4.874162703221823e-06, |
|
"logits/chosen": 0.0589652881026268, |
|
"logits/rejected": 0.25573208928108215, |
|
"logps/chosen": -414.246826171875, |
|
"logps/rejected": -430.9029235839844, |
|
"loss": 0.3575, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.04433692619204521, |
|
"rewards/margins": 0.015341699123382568, |
|
"rewards/rejected": -0.05967862531542778, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.21, |
|
"learning_rate": 4.826697788369752e-06, |
|
"logits/chosen": 0.12831085920333862, |
|
"logits/rejected": 0.1346360296010971, |
|
"logps/chosen": -544.7901000976562, |
|
"logps/rejected": -542.2559814453125, |
|
"loss": 0.3611, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.09338424354791641, |
|
"rewards/margins": 0.020744359120726585, |
|
"rewards/rejected": -0.11412860453128815, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.22, |
|
"learning_rate": 4.7719642195082224e-06, |
|
"logits/chosen": 0.004035499412566423, |
|
"logits/rejected": 0.12359336763620377, |
|
"logps/chosen": -569.91748046875, |
|
"logps/rejected": -576.3137817382812, |
|
"loss": 0.3399, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.15238870680332184, |
|
"rewards/margins": 0.03546822816133499, |
|
"rewards/rejected": -0.18785694241523743, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.24, |
|
"learning_rate": 4.710132985485355e-06, |
|
"logits/chosen": 0.09404455125331879, |
|
"logits/rejected": -0.0020357524044811726, |
|
"logps/chosen": -502.76776123046875, |
|
"logps/rejected": -537.9739990234375, |
|
"loss": 0.3444, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.1469426155090332, |
|
"rewards/margins": 0.03976910561323166, |
|
"rewards/rejected": -0.18671169877052307, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.26, |
|
"learning_rate": 4.641397248408122e-06, |
|
"logits/chosen": 0.10355547815561295, |
|
"logits/rejected": 0.07864655554294586, |
|
"logps/chosen": -578.9359741210938, |
|
"logps/rejected": -687.6909790039062, |
|
"loss": 0.3059, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -0.16360266506671906, |
|
"rewards/margins": 0.07588207721710205, |
|
"rewards/rejected": -0.2394847422838211, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.27, |
|
"learning_rate": 4.5659717401997655e-06, |
|
"logits/chosen": 0.032590705901384354, |
|
"logits/rejected": 0.08898362517356873, |
|
"logps/chosen": -600.9769287109375, |
|
"logps/rejected": -670.7233276367188, |
|
"loss": 0.3212, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.19255095720291138, |
|
"rewards/margins": 0.08060728758573532, |
|
"rewards/rejected": -0.2731582522392273, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.29, |
|
"learning_rate": 4.4840920917726425e-06, |
|
"logits/chosen": 0.04442215710878372, |
|
"logits/rejected": 0.042511895298957825, |
|
"logps/chosen": -588.660888671875, |
|
"logps/rejected": -651.74853515625, |
|
"loss": 0.3528, |
|
"rewards/accuracies": 0.42500001192092896, |
|
"rewards/chosen": -0.20966176688671112, |
|
"rewards/margins": 0.05220552161335945, |
|
"rewards/rejected": -0.2618672847747803, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"learning_rate": 4.396014096912182e-06, |
|
"logits/chosen": 0.03010064922273159, |
|
"logits/rejected": 0.041020795702934265, |
|
"logps/chosen": -532.5487060546875, |
|
"logps/rejected": -654.7836303710938, |
|
"loss": 0.3144, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.18499326705932617, |
|
"rewards/margins": 0.09925045073032379, |
|
"rewards/rejected": -0.28424373269081116, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.32, |
|
"learning_rate": 4.302012913171584e-06, |
|
"logits/chosen": -0.03510774299502373, |
|
"logits/rejected": 0.035278841853141785, |
|
"logps/chosen": -640.3848876953125, |
|
"logps/rejected": -709.2119140625, |
|
"loss": 0.3341, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -0.21959897875785828, |
|
"rewards/margins": 0.06892947107553482, |
|
"rewards/rejected": -0.2885284423828125, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.34, |
|
"learning_rate": 4.202382202273702e-06, |
|
"logits/chosen": 0.017812447622418404, |
|
"logits/rejected": 0.06379534304141998, |
|
"logps/chosen": -641.2015991210938, |
|
"logps/rejected": -672.9808349609375, |
|
"loss": 0.3513, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.19523058831691742, |
|
"rewards/margins": 0.05737825483083725, |
|
"rewards/rejected": -0.2526088356971741, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"learning_rate": 4.097433212705492e-06, |
|
"logits/chosen": 0.0030833049677312374, |
|
"logits/rejected": 0.027691733092069626, |
|
"logps/chosen": -645.5457763671875, |
|
"logps/rejected": -751.8395385742188, |
|
"loss": 0.3278, |
|
"rewards/accuracies": 0.6187499761581421, |
|
"rewards/chosen": -0.1966964602470398, |
|
"rewards/margins": 0.08199279010295868, |
|
"rewards/rejected": -0.2786892056465149, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.37, |
|
"learning_rate": 3.987493807371033e-06, |
|
"logits/chosen": 0.06291202455759048, |
|
"logits/rejected": -0.08009441941976547, |
|
"logps/chosen": -500.90496826171875, |
|
"logps/rejected": -572.1060791015625, |
|
"loss": 0.3343, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.14460867643356323, |
|
"rewards/margins": 0.05721248313784599, |
|
"rewards/rejected": -0.20182117819786072, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"learning_rate": 3.872907439340758e-06, |
|
"logits/chosen": 0.04002127796411514, |
|
"logits/rejected": -0.06345082819461823, |
|
"logps/chosen": -543.0759887695312, |
|
"logps/rejected": -591.0164184570312, |
|
"loss": 0.3242, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.1588135063648224, |
|
"rewards/margins": 0.05112007260322571, |
|
"rewards/rejected": -0.2099335640668869, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"learning_rate": 3.75403207889666e-06, |
|
"logits/chosen": 0.10061119496822357, |
|
"logits/rejected": 0.04563102871179581, |
|
"logps/chosen": -578.7415771484375, |
|
"logps/rejected": -615.0494384765625, |
|
"loss": 0.3244, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -0.1841258704662323, |
|
"rewards/margins": 0.055299896746873856, |
|
"rewards/rejected": -0.23942574858665466, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.42, |
|
"learning_rate": 3.631239095225417e-06, |
|
"logits/chosen": 0.0002536162792239338, |
|
"logits/rejected": 0.06318163126707077, |
|
"logps/chosen": -650.6129150390625, |
|
"logps/rejected": -664.8394775390625, |
|
"loss": 0.3604, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.1936233639717102, |
|
"rewards/margins": 0.039541881531476974, |
|
"rewards/rejected": -0.23316523432731628, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"learning_rate": 3.5049120962530608e-06, |
|
"logits/chosen": 0.04564657807350159, |
|
"logits/rejected": -0.02100910246372223, |
|
"logps/chosen": -597.6544799804688, |
|
"logps/rejected": -677.6250610351562, |
|
"loss": 0.3337, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -0.18195489048957825, |
|
"rewards/margins": 0.07418446242809296, |
|
"rewards/rejected": -0.2561393678188324, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.45, |
|
"learning_rate": 3.3754457302455464e-06, |
|
"logits/chosen": -0.03575817123055458, |
|
"logits/rejected": 0.11129350960254669, |
|
"logps/chosen": -567.7127685546875, |
|
"logps/rejected": -608.2735595703125, |
|
"loss": 0.3427, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.18929126858711243, |
|
"rewards/margins": 0.05334322527050972, |
|
"rewards/rejected": -0.24263450503349304, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"learning_rate": 3.2432444529190714e-06, |
|
"logits/chosen": -0.05213651806116104, |
|
"logits/rejected": 0.0017232149839401245, |
|
"logps/chosen": -600.5894165039062, |
|
"logps/rejected": -662.463623046875, |
|
"loss": 0.3298, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.16451598703861237, |
|
"rewards/margins": 0.06247172877192497, |
|
"rewards/rejected": -0.22698768973350525, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"learning_rate": 3.1087212639117057e-06, |
|
"logits/chosen": 0.04510731250047684, |
|
"logits/rejected": 0.045666079968214035, |
|
"logps/chosen": -590.2169189453125, |
|
"logps/rejected": -643.4014892578125, |
|
"loss": 0.3287, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.18261924386024475, |
|
"rewards/margins": 0.05223492905497551, |
|
"rewards/rejected": -0.23485419154167175, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.5, |
|
"learning_rate": 2.9722964165636263e-06, |
|
"logits/chosen": 0.006776309106498957, |
|
"logits/rejected": -0.025738414376974106, |
|
"logps/chosen": -530.9435424804688, |
|
"logps/rejected": -585.0616455078125, |
|
"loss": 0.3287, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -0.15342922508716583, |
|
"rewards/margins": 0.056588392704725266, |
|
"rewards/rejected": -0.2100176066160202, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"learning_rate": 2.8343961050366275e-06, |
|
"logits/chosen": -0.02025730162858963, |
|
"logits/rejected": 0.08972244709730148, |
|
"logps/chosen": -529.8350830078125, |
|
"logps/rejected": -604.9669799804688, |
|
"loss": 0.3274, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.16149136424064636, |
|
"rewards/margins": 0.06798367202281952, |
|
"rewards/rejected": -0.22947505116462708, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"learning_rate": 2.695451132874385e-06, |
|
"logits/chosen": -0.0479627326130867, |
|
"logits/rejected": 0.021157974377274513, |
|
"logps/chosen": -564.2462158203125, |
|
"logps/rejected": -604.8670654296875, |
|
"loss": 0.3409, |
|
"rewards/accuracies": 0.42500001192092896, |
|
"rewards/chosen": -0.17655444145202637, |
|
"rewards/margins": 0.06839374452829361, |
|
"rewards/rejected": -0.24494817852973938, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.54, |
|
"learning_rate": 2.5558955671628964e-06, |
|
"logits/chosen": -0.053826987743377686, |
|
"logits/rejected": 0.036486681550741196, |
|
"logps/chosen": -616.4885864257812, |
|
"logps/rejected": -669.8521728515625, |
|
"loss": 0.3326, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.20758116245269775, |
|
"rewards/margins": 0.050070859491825104, |
|
"rewards/rejected": -0.25765201449394226, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"learning_rate": 2.4161653824955654e-06, |
|
"logits/chosen": -0.025303969159722328, |
|
"logits/rejected": -0.017024224624037743, |
|
"logps/chosen": -617.4375610351562, |
|
"logps/rejected": -696.9207153320312, |
|
"loss": 0.3223, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -0.19740080833435059, |
|
"rewards/margins": 0.08515138924121857, |
|
"rewards/rejected": -0.28255218267440796, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"learning_rate": 2.2766970989791697e-06, |
|
"logits/chosen": -0.07398111373186111, |
|
"logits/rejected": -0.03359592705965042, |
|
"logps/chosen": -612.1632690429688, |
|
"logps/rejected": -696.7496337890625, |
|
"loss": 0.3282, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.20693747699260712, |
|
"rewards/margins": 0.08319561928510666, |
|
"rewards/rejected": -0.2901330888271332, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.59, |
|
"learning_rate": 2.1379264185356545e-06, |
|
"logits/chosen": -0.06404633074998856, |
|
"logits/rejected": 0.025870006531476974, |
|
"logps/chosen": -648.0364379882812, |
|
"logps/rejected": -723.4362182617188, |
|
"loss": 0.3191, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.225542813539505, |
|
"rewards/margins": 0.09016820043325424, |
|
"rewards/rejected": -0.31571102142333984, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"learning_rate": 2.000286863759934e-06, |
|
"logits/chosen": 0.06071531027555466, |
|
"logits/rejected": 0.017028797417879105, |
|
"logps/chosen": -611.7649536132812, |
|
"logps/rejected": -710.9679565429688, |
|
"loss": 0.3274, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -0.19213876128196716, |
|
"rewards/margins": 0.08261444419622421, |
|
"rewards/rejected": -0.27475324273109436, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.62, |
|
"learning_rate": 1.8642084235859764e-06, |
|
"logits/chosen": 0.0644644945859909, |
|
"logits/rejected": -0.021000124514102936, |
|
"logps/chosen": -533.0925903320312, |
|
"logps/rejected": -604.2486572265625, |
|
"loss": 0.3359, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.17501072585582733, |
|
"rewards/margins": 0.06491357088088989, |
|
"rewards/rejected": -0.23992428183555603, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.64, |
|
"learning_rate": 1.7301162099921013e-06, |
|
"logits/chosen": -0.021198943257331848, |
|
"logits/rejected": 0.045659858733415604, |
|
"logps/chosen": -591.815185546875, |
|
"logps/rejected": -630.9583129882812, |
|
"loss": 0.328, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.18797656893730164, |
|
"rewards/margins": 0.05078692361712456, |
|
"rewards/rejected": -0.23876352608203888, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"learning_rate": 1.5984291299420117e-06, |
|
"logits/chosen": -0.06265034526586533, |
|
"logits/rejected": -0.03586529195308685, |
|
"logps/chosen": -620.4393310546875, |
|
"logps/rejected": -664.763916015625, |
|
"loss": 0.3131, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.20063380897045135, |
|
"rewards/margins": 0.056611478328704834, |
|
"rewards/rejected": -0.2572453022003174, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.67, |
|
"learning_rate": 1.4695585767104092e-06, |
|
"logits/chosen": -0.02047324925661087, |
|
"logits/rejected": -0.022855054587125778, |
|
"logps/chosen": -573.5113525390625, |
|
"logps/rejected": -660.650390625, |
|
"loss": 0.315, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.1739957630634308, |
|
"rewards/margins": 0.0810604989528656, |
|
"rewards/rejected": -0.2550562918186188, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.69, |
|
"learning_rate": 1.3439071446815452e-06, |
|
"logits/chosen": 0.018320618197321892, |
|
"logits/rejected": -0.02154999040067196, |
|
"logps/chosen": -510.105224609375, |
|
"logps/rejected": -601.5553588867188, |
|
"loss": 0.333, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.1630430966615677, |
|
"rewards/margins": 0.07893416285514832, |
|
"rewards/rejected": -0.241977259516716, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.7, |
|
"learning_rate": 1.2218673716356919e-06, |
|
"logits/chosen": 0.02131899632513523, |
|
"logits/rejected": 0.102879598736763, |
|
"logps/chosen": -642.7557373046875, |
|
"logps/rejected": -687.9267578125, |
|
"loss": 0.3358, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.2222980558872223, |
|
"rewards/margins": 0.061403900384902954, |
|
"rewards/rejected": -0.28370195627212524, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.72, |
|
"learning_rate": 1.103820512452661e-06, |
|
"logits/chosen": -0.02241530455648899, |
|
"logits/rejected": -0.032763995230197906, |
|
"logps/chosen": -534.8072509765625, |
|
"logps/rejected": -587.8253173828125, |
|
"loss": 0.3429, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.18417826294898987, |
|
"rewards/margins": 0.06289814412593842, |
|
"rewards/rejected": -0.2470763921737671, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.74, |
|
"learning_rate": 9.901353480633468e-07, |
|
"logits/chosen": -0.029488349333405495, |
|
"logits/rejected": -0.008380506187677383, |
|
"logps/chosen": -640.1456298828125, |
|
"logps/rejected": -714.1085205078125, |
|
"loss": 0.3348, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.2099982500076294, |
|
"rewards/margins": 0.06554147601127625, |
|
"rewards/rejected": -0.27553972601890564, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.75, |
|
"learning_rate": 8.811670333701544e-07, |
|
"logits/chosen": -0.09662221372127533, |
|
"logits/rejected": -0.06438769400119781, |
|
"logps/chosen": -583.837158203125, |
|
"logps/rejected": -702.3072509765625, |
|
"loss": 0.3233, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.18826954066753387, |
|
"rewards/margins": 0.0807666927576065, |
|
"rewards/rejected": -0.2690362334251404, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.77, |
|
"learning_rate": 7.772559877354341e-07, |
|
"logits/chosen": -0.057563841342926025, |
|
"logits/rejected": 0.0010903537040576339, |
|
"logps/chosen": -573.5906372070312, |
|
"logps/rejected": -646.9957885742188, |
|
"loss": 0.3245, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.17442554235458374, |
|
"rewards/margins": 0.08631418645381927, |
|
"rewards/rejected": -0.2607397139072418, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"learning_rate": 6.787268315040604e-07, |
|
"logits/chosen": -0.06349989026784897, |
|
"logits/rejected": 0.024532323703169823, |
|
"logps/chosen": -573.9180908203125, |
|
"logps/rejected": -613.2721557617188, |
|
"loss": 0.3313, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -0.18243160843849182, |
|
"rewards/margins": 0.051405251026153564, |
|
"rewards/rejected": -0.2338368445634842, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"learning_rate": 5.858873718824829e-07, |
|
"logits/chosen": -0.06668631732463837, |
|
"logits/rejected": -0.06943139433860779, |
|
"logps/chosen": -553.5023193359375, |
|
"logps/rejected": -593.6824951171875, |
|
"loss": 0.3319, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -0.17030836641788483, |
|
"rewards/margins": 0.0499136857688427, |
|
"rewards/rejected": -0.22022204101085663, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.82, |
|
"learning_rate": 4.990276413423817e-07, |
|
"logits/chosen": -0.08017369359731674, |
|
"logits/rejected": -0.02853136695921421, |
|
"logps/chosen": -577.9718017578125, |
|
"logps/rejected": -648.7777099609375, |
|
"loss": 0.3326, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.19647789001464844, |
|
"rewards/margins": 0.058233268558979034, |
|
"rewards/rejected": -0.2547111511230469, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.83, |
|
"learning_rate": 4.184189915529796e-07, |
|
"logits/chosen": -0.1332094371318817, |
|
"logits/rejected": 0.09619398415088654, |
|
"logps/chosen": -579.1356201171875, |
|
"logps/rejected": -609.83544921875, |
|
"loss": 0.3251, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -0.16035783290863037, |
|
"rewards/margins": 0.07735449075698853, |
|
"rewards/rejected": -0.2377123385667801, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.85, |
|
"learning_rate": 3.4431324567258176e-07, |
|
"logits/chosen": -0.11564228683710098, |
|
"logits/rejected": -0.10405786335468292, |
|
"logps/chosen": -530.2095947265625, |
|
"logps/rejected": -617.2594604492188, |
|
"loss": 0.3344, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.17116107046604156, |
|
"rewards/margins": 0.08385736495256424, |
|
"rewards/rejected": -0.255018413066864, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"learning_rate": 2.769419116476052e-07, |
|
"logits/chosen": -0.05448366329073906, |
|
"logits/rejected": 0.020873898640275, |
|
"logps/chosen": -649.448486328125, |
|
"logps/rejected": -724.3923950195312, |
|
"loss": 0.3254, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.20293286442756653, |
|
"rewards/margins": 0.0811840295791626, |
|
"rewards/rejected": -0.2841168940067291, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.88, |
|
"learning_rate": 2.1651545897676512e-07, |
|
"logits/chosen": -0.09775115549564362, |
|
"logits/rejected": 0.037994496524333954, |
|
"logps/chosen": -631.9583740234375, |
|
"logps/rejected": -665.892578125, |
|
"loss": 0.3147, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -0.192877396941185, |
|
"rewards/margins": 0.06704525649547577, |
|
"rewards/rejected": -0.25992268323898315, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.9, |
|
"learning_rate": 1.6322266119983222e-07, |
|
"logits/chosen": -0.05889749526977539, |
|
"logits/rejected": 0.006100955419242382, |
|
"logps/chosen": -603.3699951171875, |
|
"logps/rejected": -652.2211303710938, |
|
"loss": 0.3269, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -0.18066729605197906, |
|
"rewards/margins": 0.0697588175535202, |
|
"rewards/rejected": -0.25042611360549927, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"learning_rate": 1.1723000616502167e-07, |
|
"logits/chosen": -0.020711831748485565, |
|
"logits/rejected": 0.08241745829582214, |
|
"logps/chosen": -596.9910888671875, |
|
"logps/rejected": -620.6817626953125, |
|
"loss": 0.3278, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -0.21281632781028748, |
|
"rewards/margins": 0.04477155953645706, |
|
"rewards/rejected": -0.25758787989616394, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.93, |
|
"learning_rate": 7.868117591737585e-08, |
|
"logits/chosen": -0.09641040116548538, |
|
"logits/rejected": -0.06558734178543091, |
|
"logps/chosen": -527.1932373046875, |
|
"logps/rejected": -567.8707275390625, |
|
"loss": 0.3403, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.1879841685295105, |
|
"rewards/margins": 0.048898279666900635, |
|
"rewards/rejected": -0.23688244819641113, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"learning_rate": 4.769659783295383e-08, |
|
"logits/chosen": -0.08810693025588989, |
|
"logits/rejected": -0.04142559692263603, |
|
"logps/chosen": -554.7185668945312, |
|
"logps/rejected": -658.844482421875, |
|
"loss": 0.3395, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -0.17377179861068726, |
|
"rewards/margins": 0.08222335577011108, |
|
"rewards/rejected": -0.25599512457847595, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"learning_rate": 2.4373068401120358e-08, |
|
"logits/chosen": -0.02737409994006157, |
|
"logits/rejected": -0.07764561474323273, |
|
"logps/chosen": -561.8909912109375, |
|
"logps/rejected": -653.6951904296875, |
|
"loss": 0.3241, |
|
"rewards/accuracies": 0.4937500059604645, |
|
"rewards/chosen": -0.18414655327796936, |
|
"rewards/margins": 0.0771530419588089, |
|
"rewards/rejected": -0.26129958033561707, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.98, |
|
"learning_rate": 8.78345083022425e-09, |
|
"logits/chosen": -0.1056148037314415, |
|
"logits/rejected": 0.01646628975868225, |
|
"logps/chosen": -578.94970703125, |
|
"logps/rejected": -632.1832275390625, |
|
"loss": 0.3343, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -0.18656322360038757, |
|
"rewards/margins": 0.06646813452243805, |
|
"rewards/rejected": -0.25303134322166443, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"learning_rate": 9.764474213677654e-10, |
|
"logits/chosen": -0.047748979181051254, |
|
"logits/rejected": -0.0808296799659729, |
|
"logps/chosen": -601.5391235351562, |
|
"logps/rejected": -620.3516845703125, |
|
"loss": 0.3484, |
|
"rewards/accuracies": 0.41874998807907104, |
|
"rewards/chosen": -0.20741529762744904, |
|
"rewards/margins": 0.029957950115203857, |
|
"rewards/rejected": -0.2373732626438141, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 625, |
|
"total_flos": 0.0, |
|
"train_loss": 0.3386969359874725, |
|
"train_runtime": 4609.7316, |
|
"train_samples_per_second": 4.339, |
|
"train_steps_per_second": 0.136 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 625, |
|
"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 |
|
} |
|
|