|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 500, |
|
"global_step": 395, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.0, |
|
"grad_norm": 41.54115560671131, |
|
"learning_rate": 1.25e-08, |
|
"logits/chosen": -4.306375503540039, |
|
"logits/rejected": -4.599514007568359, |
|
"logps/chosen": -381.2711181640625, |
|
"logps/rejected": -391.8406982421875, |
|
"loss": 0.6929, |
|
"rewards/accuracies": 0.0, |
|
"rewards/chosen": 0.0, |
|
"rewards/margins": 0.0, |
|
"rewards/rejected": 0.0, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 0.03, |
|
"grad_norm": 38.89233173260305, |
|
"learning_rate": 1.25e-07, |
|
"logits/chosen": -4.333991050720215, |
|
"logits/rejected": -4.4896559715271, |
|
"logps/chosen": -376.6128845214844, |
|
"logps/rejected": -396.8119201660156, |
|
"loss": 0.6925, |
|
"rewards/accuracies": 0.4583333432674408, |
|
"rewards/chosen": -0.0025132838636636734, |
|
"rewards/margins": 0.0018265678081661463, |
|
"rewards/rejected": -0.004339851904660463, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.05, |
|
"grad_norm": 38.139879750053254, |
|
"learning_rate": 2.5e-07, |
|
"logits/chosen": -4.47939920425415, |
|
"logits/rejected": -4.573966979980469, |
|
"logps/chosen": -391.5196838378906, |
|
"logps/rejected": -438.828857421875, |
|
"loss": 0.6674, |
|
"rewards/accuracies": 0.6312500238418579, |
|
"rewards/chosen": -0.09850569069385529, |
|
"rewards/margins": 0.05668836832046509, |
|
"rewards/rejected": -0.15519407391548157, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.08, |
|
"grad_norm": 47.79366128002695, |
|
"learning_rate": 3.75e-07, |
|
"logits/chosen": -4.530553340911865, |
|
"logits/rejected": -4.708470344543457, |
|
"logps/chosen": -464.93084716796875, |
|
"logps/rejected": -509.82861328125, |
|
"loss": 0.6261, |
|
"rewards/accuracies": 0.659375011920929, |
|
"rewards/chosen": -0.5474977493286133, |
|
"rewards/margins": 0.25755801796913147, |
|
"rewards/rejected": -0.8050557374954224, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.1, |
|
"grad_norm": 47.01153640565049, |
|
"learning_rate": 5e-07, |
|
"logits/chosen": -4.857049942016602, |
|
"logits/rejected": -5.048783779144287, |
|
"logps/chosen": -437.2730407714844, |
|
"logps/rejected": -497.80078125, |
|
"loss": 0.5883, |
|
"rewards/accuracies": 0.653124988079071, |
|
"rewards/chosen": -0.6910537481307983, |
|
"rewards/margins": 0.3994576036930084, |
|
"rewards/rejected": -1.0905113220214844, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.13, |
|
"grad_norm": 50.785580057659836, |
|
"learning_rate": 4.990217055187362e-07, |
|
"logits/chosen": -4.775557518005371, |
|
"logits/rejected": -4.96406364440918, |
|
"logps/chosen": -425.1912536621094, |
|
"logps/rejected": -523.9107666015625, |
|
"loss": 0.5553, |
|
"rewards/accuracies": 0.71875, |
|
"rewards/chosen": -0.48143666982650757, |
|
"rewards/margins": 0.5418455004692078, |
|
"rewards/rejected": -1.0232822895050049, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.15, |
|
"grad_norm": 47.58443781578147, |
|
"learning_rate": 4.960944785556813e-07, |
|
"logits/chosen": -4.912293434143066, |
|
"logits/rejected": -5.1732072830200195, |
|
"logps/chosen": -461.0411682128906, |
|
"logps/rejected": -549.9326171875, |
|
"loss": 0.5619, |
|
"rewards/accuracies": 0.7093750238418579, |
|
"rewards/chosen": -0.7086145281791687, |
|
"rewards/margins": 0.5085697174072266, |
|
"rewards/rejected": -1.21718430519104, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.18, |
|
"grad_norm": 50.17116749835004, |
|
"learning_rate": 4.912412286307025e-07, |
|
"logits/chosen": -5.128066062927246, |
|
"logits/rejected": -5.452770709991455, |
|
"logps/chosen": -429.89202880859375, |
|
"logps/rejected": -530.6647338867188, |
|
"loss": 0.5226, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.5659561157226562, |
|
"rewards/margins": 0.6657453775405884, |
|
"rewards/rejected": -1.2317016124725342, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.2, |
|
"grad_norm": 56.54681066075575, |
|
"learning_rate": 4.844999390047419e-07, |
|
"logits/chosen": -5.285617828369141, |
|
"logits/rejected": -5.636483192443848, |
|
"logps/chosen": -460.7303161621094, |
|
"logps/rejected": -563.2789916992188, |
|
"loss": 0.5138, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.7176406383514404, |
|
"rewards/margins": 0.732753574848175, |
|
"rewards/rejected": -1.4503942728042603, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.23, |
|
"grad_norm": 50.83178244078908, |
|
"learning_rate": 4.7592336940930097e-07, |
|
"logits/chosen": -5.608884811401367, |
|
"logits/rejected": -5.990847587585449, |
|
"logps/chosen": -460.23272705078125, |
|
"logps/rejected": -559.8355712890625, |
|
"loss": 0.5099, |
|
"rewards/accuracies": 0.784375011920929, |
|
"rewards/chosen": -0.593207061290741, |
|
"rewards/margins": 0.8474555015563965, |
|
"rewards/rejected": -1.4406626224517822, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.25, |
|
"grad_norm": 59.601370802822075, |
|
"learning_rate": 4.655786431300069e-07, |
|
"logits/chosen": -5.9478230476379395, |
|
"logits/rejected": -6.265199184417725, |
|
"logps/chosen": -431.99151611328125, |
|
"logps/rejected": -531.1004638671875, |
|
"loss": 0.5186, |
|
"rewards/accuracies": 0.7906249761581421, |
|
"rewards/chosen": -0.6661251783370972, |
|
"rewards/margins": 0.7114373445510864, |
|
"rewards/rejected": -1.3775627613067627, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.28, |
|
"grad_norm": 54.889294284535225, |
|
"learning_rate": 4.535467216758936e-07, |
|
"logits/chosen": -5.984147548675537, |
|
"logits/rejected": -6.4844231605529785, |
|
"logps/chosen": -461.1131286621094, |
|
"logps/rejected": -544.032958984375, |
|
"loss": 0.5057, |
|
"rewards/accuracies": 0.7593749761581421, |
|
"rewards/chosen": -0.6606206893920898, |
|
"rewards/margins": 0.6746741533279419, |
|
"rewards/rejected": -1.3352949619293213, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.3, |
|
"grad_norm": 61.24432368201881, |
|
"learning_rate": 4.3992177114582117e-07, |
|
"logits/chosen": -6.2256269454956055, |
|
"logits/rejected": -6.654993534088135, |
|
"logps/chosen": -477.16326904296875, |
|
"logps/rejected": -610.1165771484375, |
|
"loss": 0.4971, |
|
"rewards/accuracies": 0.7593749761581421, |
|
"rewards/chosen": -0.8679486513137817, |
|
"rewards/margins": 0.9162583351135254, |
|
"rewards/rejected": -1.7842069864273071, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.33, |
|
"grad_norm": 73.6030496927752, |
|
"learning_rate": 4.248104252510785e-07, |
|
"logits/chosen": -5.852092266082764, |
|
"logits/rejected": -6.4555840492248535, |
|
"logps/chosen": -436.20867919921875, |
|
"logps/rejected": -535.7562866210938, |
|
"loss": 0.5056, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.6110241413116455, |
|
"rewards/margins": 0.7542751431465149, |
|
"rewards/rejected": -1.3652993440628052, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.35, |
|
"grad_norm": 47.56900195447933, |
|
"learning_rate": 4.0833095076201176e-07, |
|
"logits/chosen": -5.862217903137207, |
|
"logits/rejected": -6.282025337219238, |
|
"logps/chosen": -448.24188232421875, |
|
"logps/rejected": -539.3355712890625, |
|
"loss": 0.5071, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -0.7590019702911377, |
|
"rewards/margins": 0.7245356440544128, |
|
"rewards/rejected": -1.4835376739501953, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.38, |
|
"grad_norm": 61.35802392759384, |
|
"learning_rate": 3.9061232191019517e-07, |
|
"logits/chosen": -5.939135551452637, |
|
"logits/rejected": -6.367193698883057, |
|
"logps/chosen": -452.0302734375, |
|
"logps/rejected": -549.789306640625, |
|
"loss": 0.5017, |
|
"rewards/accuracies": 0.7906249761581421, |
|
"rewards/chosen": -0.6960467100143433, |
|
"rewards/margins": 0.7666617631912231, |
|
"rewards/rejected": -1.4627084732055664, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.41, |
|
"grad_norm": 47.33126253925995, |
|
"learning_rate": 3.717932109901991e-07, |
|
"logits/chosen": -6.0608344078063965, |
|
"logits/rejected": -6.6918182373046875, |
|
"logps/chosen": -481.3443298339844, |
|
"logps/rejected": -580.4495849609375, |
|
"loss": 0.4842, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -0.8736475706100464, |
|
"rewards/margins": 0.889535129070282, |
|
"rewards/rejected": -1.7631828784942627, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.43, |
|
"grad_norm": 55.24382676161828, |
|
"learning_rate": 3.520209030608662e-07, |
|
"logits/chosen": -5.88026237487793, |
|
"logits/rejected": -6.288437843322754, |
|
"logps/chosen": -452.7632751464844, |
|
"logps/rejected": -572.0946655273438, |
|
"loss": 0.4902, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -0.6406772136688232, |
|
"rewards/margins": 0.8104515075683594, |
|
"rewards/rejected": -1.4511287212371826, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.46, |
|
"grad_norm": 55.62294024430127, |
|
"learning_rate": 3.314501432400294e-07, |
|
"logits/chosen": -6.004621505737305, |
|
"logits/rejected": -6.520898342132568, |
|
"logps/chosen": -464.07867431640625, |
|
"logps/rejected": -585.2781982421875, |
|
"loss": 0.4718, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.7512537240982056, |
|
"rewards/margins": 0.9464238286018372, |
|
"rewards/rejected": -1.6976773738861084, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.48, |
|
"grad_norm": 52.01142149507834, |
|
"learning_rate": 3.1024192561415357e-07, |
|
"logits/chosen": -6.592843532562256, |
|
"logits/rejected": -6.868170738220215, |
|
"logps/chosen": -469.8837890625, |
|
"logps/rejected": -633.4119873046875, |
|
"loss": 0.4847, |
|
"rewards/accuracies": 0.7593749761581421, |
|
"rewards/chosen": -1.087369680404663, |
|
"rewards/margins": 1.0047850608825684, |
|
"rewards/rejected": -2.0921549797058105, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.51, |
|
"grad_norm": 59.64754211046823, |
|
"learning_rate": 2.8856223324132555e-07, |
|
"logits/chosen": -6.358391761779785, |
|
"logits/rejected": -6.768553733825684, |
|
"logps/chosen": -484.43780517578125, |
|
"logps/rejected": -597.5379638671875, |
|
"loss": 0.485, |
|
"rewards/accuracies": 0.753125011920929, |
|
"rewards/chosen": -0.9880083799362183, |
|
"rewards/margins": 0.7920882105827332, |
|
"rewards/rejected": -1.7800966501235962, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.53, |
|
"grad_norm": 55.126905350837, |
|
"learning_rate": 2.66580739108776e-07, |
|
"logits/chosen": -6.6527838706970215, |
|
"logits/rejected": -7.2258100509643555, |
|
"logps/chosen": -467.697509765625, |
|
"logps/rejected": -590.52734375, |
|
"loss": 0.4806, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.8441788554191589, |
|
"rewards/margins": 0.9318108558654785, |
|
"rewards/rejected": -1.7759897708892822, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.56, |
|
"grad_norm": 52.277945286098316, |
|
"learning_rate": 2.444694782117033e-07, |
|
"logits/chosen": -6.5264458656311035, |
|
"logits/rejected": -7.032387733459473, |
|
"logps/chosen": -461.33795166015625, |
|
"logps/rejected": -560.9654541015625, |
|
"loss": 0.4717, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -0.9239259958267212, |
|
"rewards/margins": 0.8057114481925964, |
|
"rewards/rejected": -1.7296375036239624, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.58, |
|
"grad_norm": 56.70702029670774, |
|
"learning_rate": 2.2240150114618259e-07, |
|
"logits/chosen": -6.4634904861450195, |
|
"logits/rejected": -6.930532932281494, |
|
"logps/chosen": -506.49505615234375, |
|
"logps/rejected": -632.0296020507812, |
|
"loss": 0.4759, |
|
"rewards/accuracies": 0.815625011920929, |
|
"rewards/chosen": -1.0257264375686646, |
|
"rewards/margins": 1.0186141729354858, |
|
"rewards/rejected": -2.0443403720855713, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.61, |
|
"grad_norm": 50.19002044845227, |
|
"learning_rate": 2.0054951975362065e-07, |
|
"logits/chosen": -6.56687068939209, |
|
"logits/rejected": -7.020349979400635, |
|
"logps/chosen": -470.70648193359375, |
|
"logps/rejected": -628.5145263671875, |
|
"loss": 0.4777, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.882061779499054, |
|
"rewards/margins": 1.0980554819107056, |
|
"rewards/rejected": -1.9801172018051147, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.63, |
|
"grad_norm": 62.96869333763073, |
|
"learning_rate": 1.7908455541642582e-07, |
|
"logits/chosen": -6.792383670806885, |
|
"logits/rejected": -7.117269039154053, |
|
"logps/chosen": -466.4208984375, |
|
"logps/rejected": -605.7127075195312, |
|
"loss": 0.4758, |
|
"rewards/accuracies": 0.840624988079071, |
|
"rewards/chosen": -0.8703948855400085, |
|
"rewards/margins": 1.0195105075836182, |
|
"rewards/rejected": -1.889905571937561, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.66, |
|
"grad_norm": 62.510903657866734, |
|
"learning_rate": 1.5817460058381084e-07, |
|
"logits/chosen": -6.452023506164551, |
|
"logits/rejected": -6.968575477600098, |
|
"logps/chosen": -495.3380432128906, |
|
"logps/rejected": -609.4599609375, |
|
"loss": 0.4864, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.9492015838623047, |
|
"rewards/margins": 0.9177757501602173, |
|
"rewards/rejected": -1.866977334022522, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.68, |
|
"grad_norm": 62.01447255874997, |
|
"learning_rate": 1.3798330400310537e-07, |
|
"logits/chosen": -6.2711181640625, |
|
"logits/rejected": -6.893205165863037, |
|
"logps/chosen": -465.7061462402344, |
|
"logps/rejected": -592.8031005859375, |
|
"loss": 0.4625, |
|
"rewards/accuracies": 0.7906249761581421, |
|
"rewards/chosen": -0.7948521375656128, |
|
"rewards/margins": 1.056058645248413, |
|
"rewards/rejected": -1.8509107828140259, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.71, |
|
"grad_norm": 62.84305172490392, |
|
"learning_rate": 1.1866868994642534e-07, |
|
"logits/chosen": -6.332844257354736, |
|
"logits/rejected": -6.893272399902344, |
|
"logps/chosen": -478.971435546875, |
|
"logps/rejected": -597.5892333984375, |
|
"loss": 0.4598, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -0.8677324056625366, |
|
"rewards/margins": 0.9433156251907349, |
|
"rewards/rejected": -1.811047911643982, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.73, |
|
"grad_norm": 69.95001635354407, |
|
"learning_rate": 1.0038192145648567e-07, |
|
"logits/chosen": -6.384323596954346, |
|
"logits/rejected": -6.853055477142334, |
|
"logps/chosen": -527.7755126953125, |
|
"logps/rejected": -647.9898681640625, |
|
"loss": 0.4736, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -1.156057357788086, |
|
"rewards/margins": 1.0375856161117554, |
|
"rewards/rejected": -2.193643093109131, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.76, |
|
"grad_norm": 74.81540906001793, |
|
"learning_rate": 8.32661172908373e-08, |
|
"logits/chosen": -6.537497043609619, |
|
"logits/rejected": -6.883517265319824, |
|
"logps/chosen": -472.21649169921875, |
|
"logps/rejected": -605.0711059570312, |
|
"loss": 0.4672, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -0.9630511999130249, |
|
"rewards/margins": 0.9703599810600281, |
|
"rewards/rejected": -1.9334112405776978, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.78, |
|
"grad_norm": 60.19828111344577, |
|
"learning_rate": 6.745523182354146e-08, |
|
"logits/chosen": -6.695423126220703, |
|
"logits/rejected": -7.178382873535156, |
|
"logps/chosen": -465.47760009765625, |
|
"logps/rejected": -609.17919921875, |
|
"loss": 0.4645, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -1.0548077821731567, |
|
"rewards/margins": 0.9832090139389038, |
|
"rewards/rejected": -2.0380167961120605, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.81, |
|
"grad_norm": 62.819619859552624, |
|
"learning_rate": 5.307300667057049e-08, |
|
"logits/chosen": -6.586479187011719, |
|
"logits/rejected": -7.047415733337402, |
|
"logps/chosen": -449.232421875, |
|
"logps/rejected": -559.2984008789062, |
|
"loss": 0.4502, |
|
"rewards/accuracies": 0.8031250238418579, |
|
"rewards/chosen": -0.937958836555481, |
|
"rewards/margins": 0.9285039901733398, |
|
"rewards/rejected": -1.8664629459381104, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.84, |
|
"grad_norm": 54.74190137372815, |
|
"learning_rate": 4.023200224388787e-08, |
|
"logits/chosen": -6.525613307952881, |
|
"logits/rejected": -7.149096488952637, |
|
"logps/chosen": -492.0001525878906, |
|
"logps/rejected": -622.4295043945312, |
|
"loss": 0.4583, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -1.126334309577942, |
|
"rewards/margins": 1.0052974224090576, |
|
"rewards/rejected": -2.131631851196289, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.86, |
|
"grad_norm": 65.41299711527839, |
|
"learning_rate": 2.903271681360972e-08, |
|
"logits/chosen": -6.488680839538574, |
|
"logits/rejected": -7.005003452301025, |
|
"logps/chosen": -486.0816955566406, |
|
"logps/rejected": -597.4981689453125, |
|
"loss": 0.4664, |
|
"rewards/accuracies": 0.8031250238418579, |
|
"rewards/chosen": -1.010422945022583, |
|
"rewards/margins": 1.0139751434326172, |
|
"rewards/rejected": -2.0243980884552, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.89, |
|
"grad_norm": 67.03908350203817, |
|
"learning_rate": 1.956279997278043e-08, |
|
"logits/chosen": -6.695385932922363, |
|
"logits/rejected": -7.170092582702637, |
|
"logps/chosen": -490.95477294921875, |
|
"logps/rejected": -671.6006469726562, |
|
"loss": 0.4609, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -1.0624668598175049, |
|
"rewards/margins": 1.2378281354904175, |
|
"rewards/rejected": -2.300295352935791, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.91, |
|
"grad_norm": 62.41118335177348, |
|
"learning_rate": 1.1896366660467171e-08, |
|
"logits/chosen": -6.916273593902588, |
|
"logits/rejected": -7.29934549331665, |
|
"logps/chosen": -442.0613708496094, |
|
"logps/rejected": -587.3243408203125, |
|
"loss": 0.4614, |
|
"rewards/accuracies": 0.753125011920929, |
|
"rewards/chosen": -1.0646042823791504, |
|
"rewards/margins": 0.939795196056366, |
|
"rewards/rejected": -2.004399538040161, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.94, |
|
"grad_norm": 64.20669715502403, |
|
"learning_rate": 6.093417111873306e-09, |
|
"logits/chosen": -6.631227016448975, |
|
"logits/rejected": -7.061822414398193, |
|
"logps/chosen": -475.503173828125, |
|
"logps/rejected": -603.5335693359375, |
|
"loss": 0.4628, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -1.0652824640274048, |
|
"rewards/margins": 0.9914292097091675, |
|
"rewards/rejected": -2.0567116737365723, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.96, |
|
"grad_norm": 56.2255930339814, |
|
"learning_rate": 2.1993672751463576e-09, |
|
"logits/chosen": -6.618802070617676, |
|
"logits/rejected": -7.291499137878418, |
|
"logps/chosen": -474.84222412109375, |
|
"logps/rejected": -613.4112548828125, |
|
"loss": 0.4616, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -0.9348451495170593, |
|
"rewards/margins": 1.0984827280044556, |
|
"rewards/rejected": -2.03332781791687, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.99, |
|
"grad_norm": 54.17486776663219, |
|
"learning_rate": 2.4469337000659897e-10, |
|
"logits/chosen": -6.589730739593506, |
|
"logits/rejected": -7.092536926269531, |
|
"logps/chosen": -502.3226623535156, |
|
"logps/rejected": -631.1239013671875, |
|
"loss": 0.4561, |
|
"rewards/accuracies": 0.753125011920929, |
|
"rewards/chosen": -1.0657384395599365, |
|
"rewards/margins": 1.0461207628250122, |
|
"rewards/rejected": -2.111859083175659, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 395, |
|
"total_flos": 0.0, |
|
"train_loss": 0.5006634004508392, |
|
"train_runtime": 11862.2044, |
|
"train_samples_per_second": 8.521, |
|
"train_steps_per_second": 0.033 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 395, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 100, |
|
"total_flos": 0.0, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|