|
{ |
|
"best_global_step": null, |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 1.0, |
|
"eval_steps": 500, |
|
"global_step": 775, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.012903225806451613, |
|
"grad_norm": 21.62714956675265, |
|
"learning_rate": 9.000000000000001e-07, |
|
"log_odds_chosen": 1.0578124523162842, |
|
"log_odds_ratio": -0.3404296934604645, |
|
"logits/chosen": -0.9966796636581421, |
|
"logits/rejected": 15.725000381469727, |
|
"logps/chosen": -1.642187476158142, |
|
"logps/rejected": -2.565624952316284, |
|
"loss": 1.9877, |
|
"nll_loss": 1.970312476158142, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.08212890475988388, |
|
"rewards/margins": 0.04628906399011612, |
|
"rewards/rejected": -0.12832030653953552, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.025806451612903226, |
|
"grad_norm": 9.886558420099602, |
|
"learning_rate": 1.9000000000000002e-06, |
|
"log_odds_chosen": 0.8812500238418579, |
|
"log_odds_ratio": -0.38496094942092896, |
|
"logits/chosen": -0.22133788466453552, |
|
"logits/rejected": 17.149999618530273, |
|
"logps/chosen": -1.5164062976837158, |
|
"logps/rejected": -2.260937452316284, |
|
"loss": 1.7895, |
|
"nll_loss": 1.76953125, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -0.07583007961511612, |
|
"rewards/margins": 0.03732910007238388, |
|
"rewards/rejected": -0.11313476413488388, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.03870967741935484, |
|
"grad_norm": 7.174101275947782, |
|
"learning_rate": 2.9e-06, |
|
"log_odds_chosen": 0.49199217557907104, |
|
"log_odds_ratio": -0.508984386920929, |
|
"logits/chosen": 0.4482788145542145, |
|
"logits/rejected": 17.043750762939453, |
|
"logps/chosen": -1.3585937023162842, |
|
"logps/rejected": -1.7414062023162842, |
|
"loss": 1.5844, |
|
"nll_loss": 1.55859375, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.06796874850988388, |
|
"rewards/margins": 0.01908569410443306, |
|
"rewards/rejected": -0.08696289360523224, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.05161290322580645, |
|
"grad_norm": 5.02065428217483, |
|
"learning_rate": 3.900000000000001e-06, |
|
"log_odds_chosen": 0.3095703125, |
|
"log_odds_ratio": -0.5765625238418579, |
|
"logits/chosen": 1.201074242591858, |
|
"logits/rejected": 17.387500762939453, |
|
"logps/chosen": -1.1945312023162842, |
|
"logps/rejected": -1.423437476158142, |
|
"loss": 1.3805, |
|
"nll_loss": 1.3515625, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -0.05979003757238388, |
|
"rewards/margins": 0.011395263485610485, |
|
"rewards/rejected": -0.07119140774011612, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.06451612903225806, |
|
"grad_norm": 4.574739753680098, |
|
"learning_rate": 4.9000000000000005e-06, |
|
"log_odds_chosen": 0.24362793564796448, |
|
"log_odds_ratio": -0.614453136920929, |
|
"logits/chosen": 0.5683349370956421, |
|
"logits/rejected": 17.09375, |
|
"logps/chosen": -1.123437523841858, |
|
"logps/rejected": -1.303125023841858, |
|
"loss": 1.2454, |
|
"nll_loss": 1.21484375, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -0.05605468899011612, |
|
"rewards/margins": 0.00897369347512722, |
|
"rewards/rejected": -0.0650634765625, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.07741935483870968, |
|
"grad_norm": 5.580591326486056, |
|
"learning_rate": 5.9e-06, |
|
"log_odds_chosen": 0.2967773377895355, |
|
"log_odds_ratio": -0.5923827886581421, |
|
"logits/chosen": 0.01669921912252903, |
|
"logits/rejected": 16.193750381469727, |
|
"logps/chosen": -1.1171875, |
|
"logps/rejected": -1.3359375, |
|
"loss": 1.2094, |
|
"nll_loss": 1.1796875, |
|
"rewards/accuracies": 0.668749988079071, |
|
"rewards/chosen": -0.05585937574505806, |
|
"rewards/margins": 0.01091918908059597, |
|
"rewards/rejected": -0.06684570014476776, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.09032258064516129, |
|
"grad_norm": 4.465129508883455, |
|
"learning_rate": 6.9e-06, |
|
"log_odds_chosen": 0.928906261920929, |
|
"log_odds_ratio": -0.3726562559604645, |
|
"logits/chosen": -1.2811768054962158, |
|
"logits/rejected": 15.90625, |
|
"logps/chosen": -1.033593773841858, |
|
"logps/rejected": -1.740625023841858, |
|
"loss": 1.0955, |
|
"nll_loss": 1.0773437023162842, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -0.0516357421875, |
|
"rewards/margins": 0.03531493991613388, |
|
"rewards/rejected": -0.08698730170726776, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.1032258064516129, |
|
"grad_norm": 4.325447268788425, |
|
"learning_rate": 7.9e-06, |
|
"log_odds_chosen": 1.450781226158142, |
|
"log_odds_ratio": -0.2611328065395355, |
|
"logits/chosen": -1.3720703125, |
|
"logits/rejected": 15.731249809265137, |
|
"logps/chosen": -1.071874976158142, |
|
"logps/rejected": -2.2242188453674316, |
|
"loss": 1.0965, |
|
"nll_loss": 1.0828125476837158, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.0535888671875, |
|
"rewards/margins": 0.0576171875, |
|
"rewards/rejected": -0.11123047024011612, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.11612903225806452, |
|
"grad_norm": 4.408329645746575, |
|
"learning_rate": 8.900000000000001e-06, |
|
"log_odds_chosen": 1.787500023841858, |
|
"log_odds_ratio": -0.23085936903953552, |
|
"logits/chosen": -1.55859375, |
|
"logits/rejected": 15.662500381469727, |
|
"logps/chosen": -1.04296875, |
|
"logps/rejected": -2.499218702316284, |
|
"loss": 1.0693, |
|
"nll_loss": 1.057031273841858, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -0.0521240234375, |
|
"rewards/margins": 0.07275390625, |
|
"rewards/rejected": -0.12495116889476776, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.12903225806451613, |
|
"grad_norm": 5.360425933748804, |
|
"learning_rate": 9.9e-06, |
|
"log_odds_chosen": 1.853906273841858, |
|
"log_odds_ratio": -0.21064452826976776, |
|
"logits/chosen": -1.640234351158142, |
|
"logits/rejected": 15.243749618530273, |
|
"logps/chosen": -1.072656273841858, |
|
"logps/rejected": -2.6078124046325684, |
|
"loss": 1.1013, |
|
"nll_loss": 1.091406226158142, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -0.05366211012005806, |
|
"rewards/margins": 0.07661132514476776, |
|
"rewards/rejected": -0.13032226264476776, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.14193548387096774, |
|
"grad_norm": 4.490699979708051, |
|
"learning_rate": 9.578262852211514e-06, |
|
"log_odds_chosen": 2.003124952316284, |
|
"log_odds_ratio": -0.19697265326976776, |
|
"logits/chosen": -1.007055640220642, |
|
"logits/rejected": 16.012500762939453, |
|
"logps/chosen": -1.056640625, |
|
"logps/rejected": -2.7171874046325684, |
|
"loss": 1.0846, |
|
"nll_loss": 1.0753905773162842, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.05283202975988388, |
|
"rewards/margins": 0.08295898139476776, |
|
"rewards/rejected": -0.13593749701976776, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.15483870967741936, |
|
"grad_norm": 4.549502496283154, |
|
"learning_rate": 9.166984970282113e-06, |
|
"log_odds_chosen": 2.311718702316284, |
|
"log_odds_ratio": -0.16552734375, |
|
"logits/chosen": -1.816796898841858, |
|
"logits/rejected": 16.3125, |
|
"logps/chosen": -1.0390625, |
|
"logps/rejected": -2.973437547683716, |
|
"loss": 1.058, |
|
"nll_loss": 1.05078125, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.05197753757238388, |
|
"rewards/margins": 0.09672851860523224, |
|
"rewards/rejected": -0.14873047173023224, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.16774193548387098, |
|
"grad_norm": 4.492595553477561, |
|
"learning_rate": 8.804509063256239e-06, |
|
"log_odds_chosen": 2.5640625953674316, |
|
"log_odds_ratio": -0.13984374701976776, |
|
"logits/chosen": -1.11279296875, |
|
"logits/rejected": 15.949999809265137, |
|
"logps/chosen": -1.0703125, |
|
"logps/rejected": -3.253124952316284, |
|
"loss": 1.0883, |
|
"nll_loss": 1.0812499523162842, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.0535888671875, |
|
"rewards/margins": 0.10903320461511612, |
|
"rewards/rejected": -0.16269531846046448, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.18064516129032257, |
|
"grad_norm": 4.778523731144206, |
|
"learning_rate": 8.481889296799709e-06, |
|
"log_odds_chosen": 2.4632811546325684, |
|
"log_odds_ratio": -0.14370116591453552, |
|
"logits/chosen": -1.4695312976837158, |
|
"logits/rejected": 15.431249618530273, |
|
"logps/chosen": -1.04296875, |
|
"logps/rejected": -3.1171875, |
|
"loss": 1.0656, |
|
"nll_loss": 1.0574219226837158, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.05217285081744194, |
|
"rewards/margins": 0.103759765625, |
|
"rewards/rejected": -0.15595702826976776, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.1935483870967742, |
|
"grad_norm": 4.292339667379623, |
|
"learning_rate": 8.192319205190406e-06, |
|
"log_odds_chosen": 2.5328125953674316, |
|
"log_odds_ratio": -0.13725586235523224, |
|
"logits/chosen": -2.3530516624450684, |
|
"logits/rejected": 15.293749809265137, |
|
"logps/chosen": -1.0382812023162842, |
|
"logps/rejected": -3.1812500953674316, |
|
"loss": 1.0556, |
|
"nll_loss": 1.049218773841858, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.0518798828125, |
|
"rewards/margins": 0.10727538913488388, |
|
"rewards/rejected": -0.15888671576976776, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.2064516129032258, |
|
"grad_norm": 5.246753687109851, |
|
"learning_rate": 7.930515857181441e-06, |
|
"log_odds_chosen": 2.317187547683716, |
|
"log_odds_ratio": -0.1767578125, |
|
"logits/chosen": -2.588671922683716, |
|
"logits/rejected": 14.71875, |
|
"logps/chosen": -1.023828148841858, |
|
"logps/rejected": -2.965625047683716, |
|
"loss": 1.0458, |
|
"nll_loss": 1.0363280773162842, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.05119628831744194, |
|
"rewards/margins": 0.09697265923023224, |
|
"rewards/rejected": -0.14814452826976776, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.21935483870967742, |
|
"grad_norm": 4.346066733666396, |
|
"learning_rate": 7.692307692307692e-06, |
|
"log_odds_chosen": 2.692187547683716, |
|
"log_odds_ratio": -0.13859863579273224, |
|
"logits/chosen": -2.8062500953674316, |
|
"logits/rejected": 15.743749618530273, |
|
"logps/chosen": -0.9671875238418579, |
|
"logps/rejected": -3.2515625953674316, |
|
"loss": 0.9999, |
|
"nll_loss": 0.9917968511581421, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04836425930261612, |
|
"rewards/margins": 0.11416015774011612, |
|
"rewards/rejected": -0.16249999403953552, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.23225806451612904, |
|
"grad_norm": 4.703610048799506, |
|
"learning_rate": 7.4743509275193595e-06, |
|
"log_odds_chosen": 2.8375000953674316, |
|
"log_odds_ratio": -0.12353515625, |
|
"logits/chosen": -2.365234375, |
|
"logits/rejected": 15.206250190734863, |
|
"logps/chosen": -0.9917968511581421, |
|
"logps/rejected": -3.410937547683716, |
|
"loss": 1.0064, |
|
"nll_loss": 1.0007812976837158, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04960937425494194, |
|
"rewards/margins": 0.12089844048023224, |
|
"rewards/rejected": -0.17060546576976776, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.24516129032258063, |
|
"grad_norm": 4.322341540802487, |
|
"learning_rate": 7.27392967453308e-06, |
|
"log_odds_chosen": 2.792187452316284, |
|
"log_odds_ratio": -0.13710936903953552, |
|
"logits/chosen": -1.894384741783142, |
|
"logits/rejected": 15.631250381469727, |
|
"logps/chosen": -0.9921875, |
|
"logps/rejected": -3.375, |
|
"loss": 1.0165, |
|
"nll_loss": 1.010156273841858, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.049560546875, |
|
"rewards/margins": 0.119140625, |
|
"rewards/rejected": -0.16884765028953552, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.25806451612903225, |
|
"grad_norm": 4.113404991486903, |
|
"learning_rate": 7.088812050083359e-06, |
|
"log_odds_chosen": 3.575000047683716, |
|
"log_odds_ratio": -0.0809326171875, |
|
"logits/chosen": -2.4609375, |
|
"logits/rejected": 15.800000190734863, |
|
"logps/chosen": -1.013671875, |
|
"logps/rejected": -4.154687404632568, |
|
"loss": 1.0316, |
|
"nll_loss": 1.0281250476837158, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.05070800706744194, |
|
"rewards/margins": 0.15688475966453552, |
|
"rewards/rejected": -0.20761719346046448, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.2709677419354839, |
|
"grad_norm": 4.585624983468261, |
|
"learning_rate": 6.917144638660746e-06, |
|
"log_odds_chosen": 3.4140625, |
|
"log_odds_ratio": -0.12514647841453552, |
|
"logits/chosen": -2.740234375, |
|
"logits/rejected": 15.024999618530273, |
|
"logps/chosen": -1.008203148841858, |
|
"logps/rejected": -4.001562595367432, |
|
"loss": 1.0165, |
|
"nll_loss": 1.009765625, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.0504150390625, |
|
"rewards/margins": 0.149658203125, |
|
"rewards/rejected": -0.20000000298023224, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.2838709677419355, |
|
"grad_norm": 4.454926501194002, |
|
"learning_rate": 6.757373783994861e-06, |
|
"log_odds_chosen": 2.43359375, |
|
"log_odds_ratio": -0.177978515625, |
|
"logits/chosen": -1.859765648841858, |
|
"logits/rejected": 15.618749618530273, |
|
"logps/chosen": -1.05859375, |
|
"logps/rejected": -3.1265625953674316, |
|
"loss": 1.0732, |
|
"nll_loss": 1.0656249523162842, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -0.05292968824505806, |
|
"rewards/margins": 0.103515625, |
|
"rewards/rejected": -0.15634766221046448, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.2967741935483871, |
|
"grad_norm": 4.239789629191708, |
|
"learning_rate": 6.6081860045508985e-06, |
|
"log_odds_chosen": 2.297656297683716, |
|
"log_odds_ratio": -0.173583984375, |
|
"logits/chosen": -1.5969727039337158, |
|
"logits/rejected": 15.318750381469727, |
|
"logps/chosen": -1.048828125, |
|
"logps/rejected": -2.979687452316284, |
|
"loss": 1.0622, |
|
"nll_loss": 1.0535156726837158, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.05246581882238388, |
|
"rewards/margins": 0.096435546875, |
|
"rewards/rejected": -0.14897461235523224, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.3096774193548387, |
|
"grad_norm": 4.227111777252394, |
|
"learning_rate": 6.46846227353151e-06, |
|
"log_odds_chosen": 2.953125, |
|
"log_odds_ratio": -0.14106445014476776, |
|
"logits/chosen": -2.9921875, |
|
"logits/rejected": 13.850000381469727, |
|
"logps/chosen": -1.001953125, |
|
"logps/rejected": -3.5546875, |
|
"loss": 1.0255, |
|
"nll_loss": 1.01953125, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.05009765550494194, |
|
"rewards/margins": 0.12763671576976776, |
|
"rewards/rejected": -0.17763671278953552, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.3225806451612903, |
|
"grad_norm": 4.690135110410384, |
|
"learning_rate": 6.337242505244779e-06, |
|
"log_odds_chosen": 3.825000047683716, |
|
"log_odds_ratio": -0.08366699516773224, |
|
"logits/chosen": -3.262890577316284, |
|
"logits/rejected": 14.893750190734863, |
|
"logps/chosen": -0.973828136920929, |
|
"logps/rejected": -4.339062690734863, |
|
"loss": 0.9886, |
|
"nll_loss": 0.9839843511581421, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04865722730755806, |
|
"rewards/margins": 0.16806641221046448, |
|
"rewards/rejected": -0.21699218451976776, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.33548387096774196, |
|
"grad_norm": 3.892806516901723, |
|
"learning_rate": 6.213697660012001e-06, |
|
"log_odds_chosen": 3.746875047683716, |
|
"log_odds_ratio": -0.08420410007238388, |
|
"logits/chosen": -2.832812547683716, |
|
"logits/rejected": 14.925000190734863, |
|
"logps/chosen": -0.987500011920929, |
|
"logps/rejected": -4.28125, |
|
"loss": 1.0076, |
|
"nll_loss": 1.00390625, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04938964918255806, |
|
"rewards/margins": 0.16474609076976776, |
|
"rewards/rejected": -0.21406249701976776, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.34838709677419355, |
|
"grad_norm": 4.635677028308263, |
|
"learning_rate": 6.097107608496925e-06, |
|
"log_odds_chosen": 3.6265625953674316, |
|
"log_odds_ratio": -0.10944823920726776, |
|
"logits/chosen": -2.586132764816284, |
|
"logits/rejected": 14.806249618530273, |
|
"logps/chosen": -0.9632812738418579, |
|
"logps/rejected": -4.150000095367432, |
|
"loss": 0.9866, |
|
"nll_loss": 0.9820312261581421, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04819335788488388, |
|
"rewards/margins": 0.15937499701976776, |
|
"rewards/rejected": -0.20742186903953552, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.36129032258064514, |
|
"grad_norm": 4.6072687211037975, |
|
"learning_rate": 5.986843400892497e-06, |
|
"log_odds_chosen": 3.8375000953674316, |
|
"log_odds_ratio": -0.08769531548023224, |
|
"logits/chosen": -2.3558592796325684, |
|
"logits/rejected": 15.043749809265137, |
|
"logps/chosen": -1.0109374523162842, |
|
"logps/rejected": -4.404687404632568, |
|
"loss": 1.0176, |
|
"nll_loss": 1.013281226158142, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.05056152492761612, |
|
"rewards/margins": 0.169921875, |
|
"rewards/rejected": -0.22041015326976776, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.3741935483870968, |
|
"grad_norm": 4.415703053607297, |
|
"learning_rate": 5.882352941176471e-06, |
|
"log_odds_chosen": 3.729687452316284, |
|
"log_odds_ratio": -0.08544921875, |
|
"logits/chosen": -2.450390577316284, |
|
"logits/rejected": 15.418749809265137, |
|
"logps/chosen": -0.974609375, |
|
"logps/rejected": -4.240624904632568, |
|
"loss": 0.9867, |
|
"nll_loss": 0.9839843511581421, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04873047024011612, |
|
"rewards/margins": 0.16318359971046448, |
|
"rewards/rejected": -0.21201172471046448, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.3870967741935484, |
|
"grad_norm": 5.690639188843815, |
|
"learning_rate": 5.783149319662402e-06, |
|
"log_odds_chosen": 4.370312690734863, |
|
"log_odds_ratio": -0.07833252102136612, |
|
"logits/chosen": -2.629687547683716, |
|
"logits/rejected": 15.225000381469727, |
|
"logps/chosen": -1.0046875476837158, |
|
"logps/rejected": -4.921875, |
|
"loss": 1.0159, |
|
"nll_loss": 1.011328101158142, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.05024414137005806, |
|
"rewards/margins": 0.19580078125, |
|
"rewards/rejected": -0.24599608778953552, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.4, |
|
"grad_norm": 5.012778448745233, |
|
"learning_rate": 5.688801239885744e-06, |
|
"log_odds_chosen": 4.082812309265137, |
|
"log_odds_ratio": -0.08937988430261612, |
|
"logits/chosen": -2.50390625, |
|
"logits/rejected": 14.6875, |
|
"logps/chosen": -0.979296863079071, |
|
"logps/rejected": -4.614062309265137, |
|
"loss": 0.997, |
|
"nll_loss": 0.993359386920929, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04897461086511612, |
|
"rewards/margins": 0.18159179389476776, |
|
"rewards/rejected": -0.23046875, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.4129032258064516, |
|
"grad_norm": 4.657589224007925, |
|
"learning_rate": 5.598925109558544e-06, |
|
"log_odds_chosen": 4.184374809265137, |
|
"log_odds_ratio": -0.07525634765625, |
|
"logits/chosen": -2.4296875, |
|
"logits/rejected": 15.256250381469727, |
|
"logps/chosen": -0.9761718511581421, |
|
"logps/rejected": -4.701562404632568, |
|
"loss": 0.984, |
|
"nll_loss": 0.979687511920929, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.048828125, |
|
"rewards/margins": 0.18593749403953552, |
|
"rewards/rejected": -0.23476561903953552, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.4258064516129032, |
|
"grad_norm": 5.589904480370191, |
|
"learning_rate": 5.513178464199713e-06, |
|
"log_odds_chosen": 4.478125095367432, |
|
"log_odds_ratio": -0.06633301079273224, |
|
"logits/chosen": -2.649609327316284, |
|
"logits/rejected": 15.387499809265137, |
|
"logps/chosen": -0.999218761920929, |
|
"logps/rejected": -5.017187595367432, |
|
"loss": 1.028, |
|
"nll_loss": 1.025781273841858, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.0499267578125, |
|
"rewards/margins": 0.20087890326976776, |
|
"rewards/rejected": -0.2509765625, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.43870967741935485, |
|
"grad_norm": 4.6064011071073026, |
|
"learning_rate": 5.4312544659356846e-06, |
|
"log_odds_chosen": 4.3046875, |
|
"log_odds_ratio": -0.08154296875, |
|
"logits/chosen": -2.225781202316284, |
|
"logits/rejected": 15.181249618530273, |
|
"logps/chosen": -1.0089843273162842, |
|
"logps/rejected": -4.871874809265137, |
|
"loss": 1.0052, |
|
"nll_loss": 1.002343773841858, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.05046386644244194, |
|
"rewards/margins": 0.19287109375, |
|
"rewards/rejected": -0.24326172471046448, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.45161290322580644, |
|
"grad_norm": 4.5221044579303085, |
|
"learning_rate": 5.352877275721893e-06, |
|
"log_odds_chosen": 4.354687690734863, |
|
"log_odds_ratio": -0.08122558891773224, |
|
"logits/chosen": -2.8828125, |
|
"logits/rejected": 15.024999618530273, |
|
"logps/chosen": -0.9515625238418579, |
|
"logps/rejected": -4.828125, |
|
"loss": 0.9809, |
|
"nll_loss": 0.977734386920929, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04758300632238388, |
|
"rewards/margins": 0.19404296576976776, |
|
"rewards/rejected": -0.24150390923023224, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.4645161290322581, |
|
"grad_norm": 5.034689811028297, |
|
"learning_rate": 5.277798139692596e-06, |
|
"log_odds_chosen": 4.385937690734863, |
|
"log_odds_ratio": -0.07797851413488388, |
|
"logits/chosen": -2.65234375, |
|
"logits/rejected": 15.131250381469727, |
|
"logps/chosen": -0.974609375, |
|
"logps/rejected": -4.896874904632568, |
|
"loss": 0.9937, |
|
"nll_loss": 0.991015613079071, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.04868163913488388, |
|
"rewards/margins": 0.19609375298023224, |
|
"rewards/rejected": -0.24501952528953552, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.4774193548387097, |
|
"grad_norm": 14.430473695913276, |
|
"learning_rate": 5.205792062953535e-06, |
|
"log_odds_chosen": 4.551562309265137, |
|
"log_odds_ratio": -0.06669922173023224, |
|
"logits/chosen": -3.0826172828674316, |
|
"logits/rejected": 15.375, |
|
"logps/chosen": -1.000390648841858, |
|
"logps/rejected": -5.084374904632568, |
|
"loss": 1.0033, |
|
"nll_loss": 1.000390648841858, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04997558519244194, |
|
"rewards/margins": 0.20419922471046448, |
|
"rewards/rejected": -0.25439453125, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.49032258064516127, |
|
"grad_norm": 4.022201188727424, |
|
"learning_rate": 5.136654969375142e-06, |
|
"log_odds_chosen": 3.5140624046325684, |
|
"log_odds_ratio": -0.1080322265625, |
|
"logits/chosen": -2.8984375, |
|
"logits/rejected": 15.037500381469727, |
|
"logps/chosen": -0.962109386920929, |
|
"logps/rejected": -4.021874904632568, |
|
"loss": 0.9682, |
|
"nll_loss": 0.9632812738418579, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04819335788488388, |
|
"rewards/margins": 0.15278320014476776, |
|
"rewards/rejected": -0.20107421278953552, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.5032258064516129, |
|
"grad_norm": 4.6129488372536995, |
|
"learning_rate": 5.070201265633938e-06, |
|
"log_odds_chosen": 3.4203124046325684, |
|
"log_odds_ratio": -0.0938720703125, |
|
"logits/chosen": -3.04296875, |
|
"logits/rejected": 15.25, |
|
"logps/chosen": -0.952343761920929, |
|
"logps/rejected": -3.9156250953674316, |
|
"loss": 0.9822, |
|
"nll_loss": 0.977734386920929, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04758300632238388, |
|
"rewards/margins": 0.14799804985523224, |
|
"rewards/rejected": -0.19550780951976776, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.5161290322580645, |
|
"grad_norm": 4.444893820100578, |
|
"learning_rate": 5.006261743217589e-06, |
|
"log_odds_chosen": 3.379687547683716, |
|
"log_odds_ratio": -0.07974853366613388, |
|
"logits/chosen": -3.168750047683716, |
|
"logits/rejected": 14.993749618530273, |
|
"logps/chosen": -0.9671875238418579, |
|
"logps/rejected": -3.8812499046325684, |
|
"loss": 0.9826, |
|
"nll_loss": 0.9800781011581421, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04838867112994194, |
|
"rewards/margins": 0.14565429091453552, |
|
"rewards/rejected": -0.19394531846046448, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.5290322580645161, |
|
"grad_norm": 4.906317709773465, |
|
"learning_rate": 4.944681764341488e-06, |
|
"log_odds_chosen": 3.46875, |
|
"log_odds_ratio": -0.067138671875, |
|
"logits/chosen": -2.827343702316284, |
|
"logits/rejected": 15.118749618530273, |
|
"logps/chosen": -0.966796875, |
|
"logps/rejected": -3.965625047683716, |
|
"loss": 0.9761, |
|
"nll_loss": 0.97265625, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04833984375, |
|
"rewards/margins": 0.15009765326976776, |
|
"rewards/rejected": -0.19843749701976776, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.5419354838709678, |
|
"grad_norm": 4.844755914165153, |
|
"learning_rate": 4.885319687460315e-06, |
|
"log_odds_chosen": 3.5374999046325684, |
|
"log_odds_ratio": -0.08986816555261612, |
|
"logits/chosen": -2.745312452316284, |
|
"logits/rejected": 14.856249809265137, |
|
"logps/chosen": -0.981640636920929, |
|
"logps/rejected": -4.0703125, |
|
"loss": 0.9941, |
|
"nll_loss": 0.989453136920929, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -0.049072265625, |
|
"rewards/margins": 0.15439453721046448, |
|
"rewards/rejected": -0.20361328125, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.5548387096774193, |
|
"grad_norm": 4.640648526664558, |
|
"learning_rate": 4.8280454958526765e-06, |
|
"log_odds_chosen": 3.7046875953674316, |
|
"log_odds_ratio": -0.07432861626148224, |
|
"logits/chosen": -1.938574194908142, |
|
"logits/rejected": 15.850000381469727, |
|
"logps/chosen": -0.9945312738418579, |
|
"logps/rejected": -4.234375, |
|
"loss": 0.9885, |
|
"nll_loss": 0.985156238079071, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04975586012005806, |
|
"rewards/margins": 0.162109375, |
|
"rewards/rejected": -0.21181640028953552, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.567741935483871, |
|
"grad_norm": 5.651687778086319, |
|
"learning_rate": 4.7727395990334775e-06, |
|
"log_odds_chosen": 3.496875047683716, |
|
"log_odds_ratio": -0.09119872748851776, |
|
"logits/chosen": -3.115307569503784, |
|
"logits/rejected": 15.087499618530273, |
|
"logps/chosen": -0.981640636920929, |
|
"logps/rejected": -4.03125, |
|
"loss": 1.0027, |
|
"nll_loss": 0.998046875, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04912109300494194, |
|
"rewards/margins": 0.15253905951976776, |
|
"rewards/rejected": -0.20185546576976776, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.5806451612903226, |
|
"grad_norm": 4.282703734054553, |
|
"learning_rate": 4.719291781830088e-06, |
|
"log_odds_chosen": 3.354687452316284, |
|
"log_odds_ratio": -0.12017822265625, |
|
"logits/chosen": -2.7964844703674316, |
|
"logits/rejected": 14.881250381469727, |
|
"logps/chosen": -1.001562476158142, |
|
"logps/rejected": -3.9312500953674316, |
|
"loss": 1.0237, |
|
"nll_loss": 1.017578125, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.05007324367761612, |
|
"rewards/margins": 0.14653320610523224, |
|
"rewards/rejected": -0.19648437201976776, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.5935483870967742, |
|
"grad_norm": 4.666237310414398, |
|
"learning_rate": 4.667600280093366e-06, |
|
"log_odds_chosen": 3.6953125, |
|
"log_odds_ratio": -0.09528808295726776, |
|
"logits/chosen": -2.721484422683716, |
|
"logits/rejected": 15.293749809265137, |
|
"logps/chosen": -0.9652343988418579, |
|
"logps/rejected": -4.1953125, |
|
"loss": 0.9829, |
|
"nll_loss": 0.977734386920929, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.04829101637005806, |
|
"rewards/margins": 0.16171875596046448, |
|
"rewards/rejected": -0.21005859971046448, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.6064516129032258, |
|
"grad_norm": 5.24217311652083, |
|
"learning_rate": 4.6175709653961015e-06, |
|
"log_odds_chosen": 3.479687452316284, |
|
"log_odds_ratio": -0.116455078125, |
|
"logits/chosen": -3.159374952316284, |
|
"logits/rejected": 14.5625, |
|
"logps/chosen": -1.0109374523162842, |
|
"logps/rejected": -4.078125, |
|
"loss": 1.0153, |
|
"nll_loss": 1.009765625, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -0.05058593675494194, |
|
"rewards/margins": 0.15336914360523224, |
|
"rewards/rejected": -0.20390625298023224, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.6193548387096774, |
|
"grad_norm": 4.85775451265865, |
|
"learning_rate": 4.569116623849539e-06, |
|
"log_odds_chosen": 3.8921875953674316, |
|
"log_odds_ratio": -0.07180175930261612, |
|
"logits/chosen": -2.889843702316284, |
|
"logits/rejected": 15.493749618530273, |
|
"logps/chosen": -0.922656238079071, |
|
"logps/rejected": -4.332812309265137, |
|
"loss": 0.9427, |
|
"nll_loss": 0.939453125, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04611816257238388, |
|
"rewards/margins": 0.17031249403953552, |
|
"rewards/rejected": -0.21640625596046448, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.632258064516129, |
|
"grad_norm": 5.135467312058317, |
|
"learning_rate": 4.522156316461347e-06, |
|
"log_odds_chosen": 3.7578125, |
|
"log_odds_ratio": -0.07160644233226776, |
|
"logits/chosen": -3.319140672683716, |
|
"logits/rejected": 15.512499809265137, |
|
"logps/chosen": -0.998828113079071, |
|
"logps/rejected": -4.303124904632568, |
|
"loss": 1.011, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04987793043255806, |
|
"rewards/margins": 0.1650390625, |
|
"rewards/rejected": -0.21494141221046448, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.6451612903225806, |
|
"grad_norm": 4.47674714460599, |
|
"learning_rate": 4.476614810358452e-06, |
|
"log_odds_chosen": 3.5999999046325684, |
|
"log_odds_ratio": -0.09897460788488388, |
|
"logits/chosen": -2.8734374046325684, |
|
"logits/rejected": 15.068750381469727, |
|
"logps/chosen": -1.0261719226837158, |
|
"logps/rejected": -4.203125, |
|
"loss": 1.0312, |
|
"nll_loss": 1.0265624523162842, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.05129394680261612, |
|
"rewards/margins": 0.15883788466453552, |
|
"rewards/rejected": -0.21005859971046448, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.6580645161290323, |
|
"grad_norm": 4.714299310109519, |
|
"learning_rate": 4.432422071779363e-06, |
|
"log_odds_chosen": 3.8812499046325684, |
|
"log_odds_ratio": -0.07042236626148224, |
|
"logits/chosen": -3.11328125, |
|
"logits/rejected": 14.949999809265137, |
|
"logps/chosen": -0.936328113079071, |
|
"logps/rejected": -4.3359375, |
|
"loss": 0.9671, |
|
"nll_loss": 0.964062511920929, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04677734524011612, |
|
"rewards/margins": 0.16997070610523224, |
|
"rewards/rejected": -0.216796875, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.6709677419354839, |
|
"grad_norm": 4.695855571935763, |
|
"learning_rate": 4.389512813061471e-06, |
|
"log_odds_chosen": 3.796875, |
|
"log_odds_ratio": -0.07448730617761612, |
|
"logits/chosen": -3.2523436546325684, |
|
"logits/rejected": 15.300000190734863, |
|
"logps/chosen": -0.931640625, |
|
"logps/rejected": -4.240624904632568, |
|
"loss": 0.9521, |
|
"nll_loss": 0.948437511920929, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04655761644244194, |
|
"rewards/margins": 0.16542968153953552, |
|
"rewards/rejected": -0.21171875298023224, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.6838709677419355, |
|
"grad_norm": 4.8627800661273435, |
|
"learning_rate": 4.347826086956522e-06, |
|
"log_odds_chosen": 3.7015624046325684, |
|
"log_odds_ratio": -0.08720703423023224, |
|
"logits/chosen": -2.1195311546325684, |
|
"logits/rejected": 14.824999809265137, |
|
"logps/chosen": -0.930859386920929, |
|
"logps/rejected": -4.146874904632568, |
|
"loss": 0.9535, |
|
"nll_loss": 0.949999988079071, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.0465087890625, |
|
"rewards/margins": 0.16103515028953552, |
|
"rewards/rejected": -0.20751953125, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.6967741935483871, |
|
"grad_norm": 4.178975744588359, |
|
"learning_rate": 4.3073049225394805e-06, |
|
"log_odds_chosen": 3.934375047683716, |
|
"log_odds_ratio": -0.08649902045726776, |
|
"logits/chosen": -2.1851563453674316, |
|
"logits/rejected": 15.850000381469727, |
|
"logps/chosen": -0.992968738079071, |
|
"logps/rejected": -4.431250095367432, |
|
"loss": 0.9628, |
|
"nll_loss": 0.958984375, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.049560546875, |
|
"rewards/margins": 0.17207030951976776, |
|
"rewards/rejected": -0.22187499701976776, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.7096774193548387, |
|
"grad_norm": 5.086676612389064, |
|
"learning_rate": 4.2678959977632e-06, |
|
"log_odds_chosen": 3.9453125, |
|
"log_odds_ratio": -0.05517578125, |
|
"logits/chosen": -2.5117430686950684, |
|
"logits/rejected": 15.34375, |
|
"logps/chosen": -0.971484363079071, |
|
"logps/rejected": -4.439062595367432, |
|
"loss": 0.9747, |
|
"nll_loss": 0.97265625, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.048583984375, |
|
"rewards/margins": 0.17333984375, |
|
"rewards/rejected": -0.22197265923023224, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.7225806451612903, |
|
"grad_norm": 4.675506919842173, |
|
"learning_rate": 4.229549344378136e-06, |
|
"log_odds_chosen": 4.035937309265137, |
|
"log_odds_ratio": -0.07060547173023224, |
|
"logits/chosen": -3.0234375, |
|
"logits/rejected": 15.137499809265137, |
|
"logps/chosen": -0.9703124761581421, |
|
"logps/rejected": -4.532812595367432, |
|
"loss": 0.9885, |
|
"nll_loss": 0.9839843511581421, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.0484619140625, |
|
"rewards/margins": 0.17822265625, |
|
"rewards/rejected": -0.22675780951976776, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.7354838709677419, |
|
"grad_norm": 4.355789903905205, |
|
"learning_rate": 4.192218081503185e-06, |
|
"log_odds_chosen": 4.096875190734863, |
|
"log_odds_ratio": -0.0665283203125, |
|
"logits/chosen": -2.786328077316284, |
|
"logits/rejected": 15.324999809265137, |
|
"logps/chosen": -0.939453125, |
|
"logps/rejected": -4.553124904632568, |
|
"loss": 0.9593, |
|
"nll_loss": 0.9574218988418579, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04704589769244194, |
|
"rewards/margins": 0.18056640028953552, |
|
"rewards/rejected": -0.2275390625, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.7483870967741936, |
|
"grad_norm": 4.395304577880807, |
|
"learning_rate": 4.155858174616914e-06, |
|
"log_odds_chosen": 3.8609375953674316, |
|
"log_odds_ratio": -0.06821288913488388, |
|
"logits/chosen": -3.245312452316284, |
|
"logits/rejected": 14.631250381469727, |
|
"logps/chosen": -0.912890613079071, |
|
"logps/rejected": -4.275000095367432, |
|
"loss": 0.9378, |
|
"nll_loss": 0.934765636920929, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.045654296875, |
|
"rewards/margins": 0.16787108778953552, |
|
"rewards/rejected": -0.21367187798023224, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.7612903225806451, |
|
"grad_norm": 4.529044726069004, |
|
"learning_rate": 4.120428217151647e-06, |
|
"log_odds_chosen": 4.0546875, |
|
"log_odds_ratio": -0.06181640550494194, |
|
"logits/chosen": -3.321093797683716, |
|
"logits/rejected": 14.987500190734863, |
|
"logps/chosen": -0.928906261920929, |
|
"logps/rejected": -4.490624904632568, |
|
"loss": 0.9545, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04638671875, |
|
"rewards/margins": 0.17802734673023224, |
|
"rewards/rejected": -0.22451171278953552, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.7741935483870968, |
|
"grad_norm": 4.367330516996232, |
|
"learning_rate": 4.0858892322271844e-06, |
|
"log_odds_chosen": 4.139062404632568, |
|
"log_odds_ratio": -0.05810546875, |
|
"logits/chosen": -2.7339844703674316, |
|
"logits/rejected": 15.324999809265137, |
|
"logps/chosen": -0.9683593511581421, |
|
"logps/rejected": -4.634375095367432, |
|
"loss": 0.9833, |
|
"nll_loss": 0.98046875, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04836425930261612, |
|
"rewards/margins": 0.18339844048023224, |
|
"rewards/rejected": -0.23154297471046448, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.7870967741935484, |
|
"grad_norm": 4.725621105355558, |
|
"learning_rate": 4.052204492365539e-06, |
|
"log_odds_chosen": 4.568749904632568, |
|
"log_odds_ratio": -0.05581054836511612, |
|
"logits/chosen": -3.828125, |
|
"logits/rejected": 15.168749809265137, |
|
"logps/chosen": -0.974609375, |
|
"logps/rejected": -5.065625190734863, |
|
"loss": 0.9903, |
|
"nll_loss": 0.98828125, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.0487060546875, |
|
"rewards/margins": 0.20458984375, |
|
"rewards/rejected": -0.25341796875, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.8, |
|
"grad_norm": 5.489269120536017, |
|
"learning_rate": 4.019339355290704e-06, |
|
"log_odds_chosen": 4.253125190734863, |
|
"log_odds_ratio": -0.08012695610523224, |
|
"logits/chosen": -3.4429688453674316, |
|
"logits/rejected": 15.5, |
|
"logps/chosen": -0.935546875, |
|
"logps/rejected": -4.696875095367432, |
|
"loss": 0.9602, |
|
"nll_loss": 0.9566406011581421, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -0.0467529296875, |
|
"rewards/margins": 0.18798828125, |
|
"rewards/rejected": -0.23466797173023224, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.8129032258064516, |
|
"grad_norm": 4.806303543591708, |
|
"learning_rate": 3.9872611141445005e-06, |
|
"log_odds_chosen": 4.260937690734863, |
|
"log_odds_ratio": -0.05676879733800888, |
|
"logits/chosen": -3.21044921875, |
|
"logits/rejected": 15.40625, |
|
"logps/chosen": -0.919921875, |
|
"logps/rejected": -4.668749809265137, |
|
"loss": 0.9316, |
|
"nll_loss": 0.9292968511581421, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04604492336511612, |
|
"rewards/margins": 0.18740233778953552, |
|
"rewards/rejected": -0.2333984375, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.8258064516129032, |
|
"grad_norm": 19.816287661530026, |
|
"learning_rate": 3.955938860646178e-06, |
|
"log_odds_chosen": 4.512499809265137, |
|
"log_odds_ratio": -0.04911499097943306, |
|
"logits/chosen": -2.807861328125, |
|
"logits/rejected": 15.481249809265137, |
|
"logps/chosen": -0.9945312738418579, |
|
"logps/rejected": -5.021874904632568, |
|
"loss": 0.9932, |
|
"nll_loss": 0.9917968511581421, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04973144456744194, |
|
"rewards/margins": 0.20107421278953552, |
|
"rewards/rejected": -0.2508789002895355, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.8387096774193549, |
|
"grad_norm": 4.519655498618536, |
|
"learning_rate": 3.925343359894298e-06, |
|
"log_odds_chosen": 4.828125, |
|
"log_odds_ratio": -0.04646606370806694, |
|
"logits/chosen": -3.143749952316284, |
|
"logits/rejected": 15.987500190734863, |
|
"logps/chosen": -0.9808593988418579, |
|
"logps/rejected": -5.324999809265137, |
|
"loss": 0.9964, |
|
"nll_loss": 0.9937499761581421, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -0.04909668117761612, |
|
"rewards/margins": 0.21708984673023224, |
|
"rewards/rejected": -0.26640623807907104, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.8516129032258064, |
|
"grad_norm": 4.6066718377177915, |
|
"learning_rate": 3.895446935658099e-06, |
|
"log_odds_chosen": 4.415625095367432, |
|
"log_odds_ratio": -0.05909423902630806, |
|
"logits/chosen": -2.4632811546325684, |
|
"logits/rejected": 15.399999618530273, |
|
"logps/chosen": -0.987500011920929, |
|
"logps/rejected": -4.942187309265137, |
|
"loss": 0.9912, |
|
"nll_loss": 0.98828125, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04936523362994194, |
|
"rewards/margins": 0.19785156846046448, |
|
"rewards/rejected": -0.2470703125, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.864516129032258, |
|
"grad_norm": 4.613695358429598, |
|
"learning_rate": 3.866223365135295e-06, |
|
"log_odds_chosen": 4.578125, |
|
"log_odds_ratio": -0.05357055738568306, |
|
"logits/chosen": -3.272656202316284, |
|
"logits/rejected": 15.193750381469727, |
|
"logps/chosen": -0.9609375, |
|
"logps/rejected": -5.043749809265137, |
|
"loss": 0.9754, |
|
"nll_loss": 0.9730468988418579, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.0479736328125, |
|
"rewards/margins": 0.20439453423023224, |
|
"rewards/rejected": -0.25244140625, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.8774193548387097, |
|
"grad_norm": 5.422841177317399, |
|
"learning_rate": 3.837647782266681e-06, |
|
"log_odds_chosen": 4.890625, |
|
"log_odds_ratio": -0.0390625, |
|
"logits/chosen": -2.7265625, |
|
"logits/rejected": 15.375, |
|
"logps/chosen": -0.9867187738418579, |
|
"logps/rejected": -5.390625, |
|
"loss": 0.9915, |
|
"nll_loss": 0.989453136920929, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04931640625, |
|
"rewards/margins": 0.22021484375, |
|
"rewards/rejected": -0.2699218690395355, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.8903225806451613, |
|
"grad_norm": 5.19065215935842, |
|
"learning_rate": 3.8096965887972973e-06, |
|
"log_odds_chosen": 4.612500190734863, |
|
"log_odds_ratio": -0.073883056640625, |
|
"logits/chosen": -2.76953125, |
|
"logits/rejected": 14.987500190734863, |
|
"logps/chosen": -0.9644531011581421, |
|
"logps/rejected": -5.084374904632568, |
|
"loss": 0.9617, |
|
"nll_loss": 0.957812488079071, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04824218899011612, |
|
"rewards/margins": 0.20585937798023224, |
|
"rewards/rejected": -0.25468748807907104, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.9032258064516129, |
|
"grad_norm": 4.754599806255075, |
|
"learning_rate": 3.7823473723611694e-06, |
|
"log_odds_chosen": 4.498437404632568, |
|
"log_odds_ratio": -0.05502929538488388, |
|
"logits/chosen": -2.788281202316284, |
|
"logits/rejected": 15.15625, |
|
"logps/chosen": -0.923828125, |
|
"logps/rejected": -4.918749809265137, |
|
"loss": 0.9378, |
|
"nll_loss": 0.9359375238418579, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04621582105755806, |
|
"rewards/margins": 0.19970703125, |
|
"rewards/rejected": -0.24628905951976776, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.9161290322580645, |
|
"grad_norm": 4.940556198099481, |
|
"learning_rate": 3.7555788309433988e-06, |
|
"log_odds_chosen": 4.724999904632568, |
|
"log_odds_ratio": -0.06513671576976776, |
|
"logits/chosen": -3.089062452316284, |
|
"logits/rejected": 15.15625, |
|
"logps/chosen": -0.9515625238418579, |
|
"logps/rejected": -5.196875095367432, |
|
"loss": 0.9742, |
|
"nll_loss": 0.971484363079071, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04763183742761612, |
|
"rewards/margins": 0.21220703423023224, |
|
"rewards/rejected": -0.2596679627895355, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.9290322580645162, |
|
"grad_norm": 4.423887385405989, |
|
"learning_rate": 3.729370703141008e-06, |
|
"log_odds_chosen": 4.8828125, |
|
"log_odds_ratio": -0.04531250149011612, |
|
"logits/chosen": -2.8277344703674316, |
|
"logits/rejected": 15.324999809265137, |
|
"logps/chosen": -0.9242187738418579, |
|
"logps/rejected": -5.300000190734863, |
|
"loss": 0.9459, |
|
"nll_loss": 0.944531261920929, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.0462646484375, |
|
"rewards/margins": 0.21894530951976776, |
|
"rewards/rejected": -0.26494139432907104, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.9419354838709677, |
|
"grad_norm": 5.674218130480263, |
|
"learning_rate": 3.7037037037037037e-06, |
|
"log_odds_chosen": 4.359375, |
|
"log_odds_ratio": -0.05672607570886612, |
|
"logits/chosen": -3.0914063453674316, |
|
"logits/rejected": 15.118749618530273, |
|
"logps/chosen": -0.95703125, |
|
"logps/rejected": -4.837500095367432, |
|
"loss": 0.9725, |
|
"nll_loss": 0.9691406488418579, |
|
"rewards/accuracies": 0.987500011920929, |
|
"rewards/chosen": -0.04782714694738388, |
|
"rewards/margins": 0.19394531846046448, |
|
"rewards/rejected": -0.24160155653953552, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.9548387096774194, |
|
"grad_norm": 4.565990551466217, |
|
"learning_rate": 3.6785594638885484e-06, |
|
"log_odds_chosen": 4.143750190734863, |
|
"log_odds_ratio": -0.06418456882238388, |
|
"logits/chosen": -3.5328125953674316, |
|
"logits/rejected": 14.65625, |
|
"logps/chosen": -0.9203125238418579, |
|
"logps/rejected": -4.557812690734863, |
|
"loss": 0.9248, |
|
"nll_loss": 0.921875, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.04599609225988388, |
|
"rewards/margins": 0.18203124403953552, |
|
"rewards/rejected": -0.22812500596046448, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.967741935483871, |
|
"grad_norm": 4.4761450056044705, |
|
"learning_rate": 3.6539204762093485e-06, |
|
"log_odds_chosen": 4.503125190734863, |
|
"log_odds_ratio": -0.04804076999425888, |
|
"logits/chosen": -3.145312547683716, |
|
"logits/rejected": 15.393750190734863, |
|
"logps/chosen": -0.958984375, |
|
"logps/rejected": -4.959374904632568, |
|
"loss": 0.9787, |
|
"nll_loss": 0.9769531488418579, |
|
"rewards/accuracies": 0.9937499761581421, |
|
"rewards/chosen": -0.04794921725988388, |
|
"rewards/margins": 0.20000000298023224, |
|
"rewards/rejected": -0.248046875, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.9806451612903225, |
|
"grad_norm": 4.612842620007858, |
|
"learning_rate": 3.6297700432031386e-06, |
|
"log_odds_chosen": 4.284375190734863, |
|
"log_odds_ratio": -0.07587890326976776, |
|
"logits/chosen": -2.9046874046325684, |
|
"logits/rejected": 15.106249809265137, |
|
"logps/chosen": -0.950390636920929, |
|
"logps/rejected": -4.759375095367432, |
|
"loss": 0.969, |
|
"nll_loss": 0.96484375, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04750976711511612, |
|
"rewards/margins": 0.19003906846046448, |
|
"rewards/rejected": -0.23789063096046448, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.9935483870967742, |
|
"grad_norm": 5.433862964765063, |
|
"learning_rate": 3.606092229873094e-06, |
|
"log_odds_chosen": 4.482812404632568, |
|
"log_odds_ratio": -0.07089843600988388, |
|
"logits/chosen": -3.5140624046325684, |
|
"logits/rejected": 14.943750381469727, |
|
"logps/chosen": -0.9488281011581421, |
|
"logps/rejected": -4.940625190734863, |
|
"loss": 0.9684, |
|
"nll_loss": 0.9652343988418579, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.04746093600988388, |
|
"rewards/margins": 0.19970703125, |
|
"rewards/rejected": -0.24746093153953552, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"step": 775, |
|
"total_flos": 0.0, |
|
"train_loss": 1.0403899949596773, |
|
"train_runtime": 1570.4069, |
|
"train_samples_per_second": 7.896, |
|
"train_steps_per_second": 0.494 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 775, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 1, |
|
"save_steps": 500, |
|
"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 |
|
} |
|
|