MamayLM-ORPO-align-lora64 / trainer_state.json
antonpolishko's picture
Model save
3cde676 verified
{
"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
}