|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 2.9865871833084947, |
|
"eval_steps": 500, |
|
"global_step": 501, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.029806259314456036, |
|
"grad_norm": 1766.112548828125, |
|
"learning_rate": 2.5000000000000004e-07, |
|
"log_odds_chosen": -0.22085151076316833, |
|
"log_odds_ratio": -1.0071768760681152, |
|
"logits/chosen": 204.30807495117188, |
|
"logits/rejected": 202.9915771484375, |
|
"logps/chosen": -14.827531814575195, |
|
"logps/rejected": -14.60667896270752, |
|
"loss": 14.9615, |
|
"nll_loss": 14.545796394348145, |
|
"rewards/accuracies": 0.3499999940395355, |
|
"rewards/chosen": -0.7413766384124756, |
|
"rewards/margins": -0.011042552068829536, |
|
"rewards/rejected": -0.7303339838981628, |
|
"step": 5 |
|
}, |
|
{ |
|
"epoch": 0.05961251862891207, |
|
"grad_norm": 1192.7847900390625, |
|
"learning_rate": 5.000000000000001e-07, |
|
"log_odds_chosen": 0.2549552619457245, |
|
"log_odds_ratio": -0.769985020160675, |
|
"logits/chosen": 219.4311981201172, |
|
"logits/rejected": 223.46994018554688, |
|
"logps/chosen": -12.237886428833008, |
|
"logps/rejected": -12.492194175720215, |
|
"loss": 12.6124, |
|
"nll_loss": 12.336746215820312, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.6118943095207214, |
|
"rewards/margins": 0.012715431861579418, |
|
"rewards/rejected": -0.6246097683906555, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.08941877794336811, |
|
"grad_norm": 725.3335571289062, |
|
"learning_rate": 7.5e-07, |
|
"log_odds_chosen": 0.0513768307864666, |
|
"log_odds_ratio": -0.7725777626037598, |
|
"logits/chosen": 282.11224365234375, |
|
"logits/rejected": 261.1120910644531, |
|
"logps/chosen": -7.968921661376953, |
|
"logps/rejected": -8.020242691040039, |
|
"loss": 8.2811, |
|
"nll_loss": 7.956255912780762, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.39844608306884766, |
|
"rewards/margins": 0.0025660558603703976, |
|
"rewards/rejected": -0.401012122631073, |
|
"step": 15 |
|
}, |
|
{ |
|
"epoch": 0.11922503725782414, |
|
"grad_norm": 216.32925415039062, |
|
"learning_rate": 1.0000000000000002e-06, |
|
"log_odds_chosen": -0.1569918394088745, |
|
"log_odds_ratio": -0.9539459347724915, |
|
"logits/chosen": 280.9331359863281, |
|
"logits/rejected": 275.239013671875, |
|
"logps/chosen": -5.379445552825928, |
|
"logps/rejected": -5.223770618438721, |
|
"loss": 5.4483, |
|
"nll_loss": 5.457978248596191, |
|
"rewards/accuracies": 0.3499999940395355, |
|
"rewards/chosen": -0.268972247838974, |
|
"rewards/margins": -0.007783788256347179, |
|
"rewards/rejected": -0.26118847727775574, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.14903129657228018, |
|
"grad_norm": 152.3154296875, |
|
"learning_rate": 1.25e-06, |
|
"log_odds_chosen": -0.03973072022199631, |
|
"log_odds_ratio": -0.8836725354194641, |
|
"logits/chosen": 300.5466003417969, |
|
"logits/rejected": 309.90667724609375, |
|
"logps/chosen": -3.2718262672424316, |
|
"logps/rejected": -3.2141456604003906, |
|
"loss": 3.4993, |
|
"nll_loss": 3.386289596557617, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.16359131038188934, |
|
"rewards/margins": -0.0028840256854891777, |
|
"rewards/rejected": -0.1607072949409485, |
|
"step": 25 |
|
}, |
|
{ |
|
"epoch": 0.17883755588673622, |
|
"grad_norm": 89.40418243408203, |
|
"learning_rate": 1.5e-06, |
|
"log_odds_chosen": -0.05680186673998833, |
|
"log_odds_ratio": -0.8815790414810181, |
|
"logits/chosen": 350.63232421875, |
|
"logits/rejected": 379.04736328125, |
|
"logps/chosen": -2.606602668762207, |
|
"logps/rejected": -2.523834466934204, |
|
"loss": 2.5544, |
|
"nll_loss": 2.63269305229187, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.1303301304578781, |
|
"rewards/margins": -0.0041384161449968815, |
|
"rewards/rejected": -0.12619172036647797, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.20864381520119224, |
|
"grad_norm": 49.261566162109375, |
|
"learning_rate": 1.75e-06, |
|
"log_odds_chosen": 0.16310584545135498, |
|
"log_odds_ratio": -0.7383562922477722, |
|
"logits/chosen": 385.07769775390625, |
|
"logits/rejected": 373.39593505859375, |
|
"logps/chosen": -1.8015142679214478, |
|
"logps/rejected": -1.905853033065796, |
|
"loss": 2.1302, |
|
"nll_loss": 2.0047783851623535, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.09007571637630463, |
|
"rewards/margins": 0.005216942168772221, |
|
"rewards/rejected": -0.09529266506433487, |
|
"step": 35 |
|
}, |
|
{ |
|
"epoch": 0.23845007451564829, |
|
"grad_norm": 35.88943099975586, |
|
"learning_rate": 2.0000000000000003e-06, |
|
"log_odds_chosen": 0.03436558321118355, |
|
"log_odds_ratio": -0.7623429298400879, |
|
"logits/chosen": 378.38153076171875, |
|
"logits/rejected": 376.9698791503906, |
|
"logps/chosen": -1.6465238332748413, |
|
"logps/rejected": -1.670940637588501, |
|
"loss": 1.9446, |
|
"nll_loss": 2.038804769515991, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.08232619613409042, |
|
"rewards/margins": 0.0012208283878862858, |
|
"rewards/rejected": -0.08354702591896057, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.26825633383010433, |
|
"grad_norm": 37.345130920410156, |
|
"learning_rate": 2.25e-06, |
|
"log_odds_chosen": 0.058513958007097244, |
|
"log_odds_ratio": -0.7263653874397278, |
|
"logits/chosen": 390.2134704589844, |
|
"logits/rejected": 399.8643493652344, |
|
"logps/chosen": -1.5079405307769775, |
|
"logps/rejected": -1.5511177778244019, |
|
"loss": 1.8369, |
|
"nll_loss": 1.7173048257827759, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07539702951908112, |
|
"rewards/margins": 0.002158870454877615, |
|
"rewards/rejected": -0.07755588740110397, |
|
"step": 45 |
|
}, |
|
{ |
|
"epoch": 0.29806259314456035, |
|
"grad_norm": 50.422019958496094, |
|
"learning_rate": 2.5e-06, |
|
"log_odds_chosen": 0.2006622552871704, |
|
"log_odds_ratio": -0.6733421087265015, |
|
"logits/chosen": 399.89837646484375, |
|
"logits/rejected": 420.90509033203125, |
|
"logps/chosen": -1.418306827545166, |
|
"logps/rejected": -1.5650049448013306, |
|
"loss": 1.8378, |
|
"nll_loss": 1.8607208728790283, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.0709153264760971, |
|
"rewards/margins": 0.007334905676543713, |
|
"rewards/rejected": -0.07825024425983429, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.32786885245901637, |
|
"grad_norm": 47.060543060302734, |
|
"learning_rate": 2.7500000000000004e-06, |
|
"log_odds_chosen": 0.04189341515302658, |
|
"log_odds_ratio": -0.7426876425743103, |
|
"logits/chosen": 387.83795166015625, |
|
"logits/rejected": 379.873291015625, |
|
"logps/chosen": -1.4415570497512817, |
|
"logps/rejected": -1.463037133216858, |
|
"loss": 1.8704, |
|
"nll_loss": 1.915564775466919, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -0.07207784801721573, |
|
"rewards/margins": 0.0010740077123045921, |
|
"rewards/rejected": -0.07315186411142349, |
|
"step": 55 |
|
}, |
|
{ |
|
"epoch": 0.35767511177347244, |
|
"grad_norm": 34.15119934082031, |
|
"learning_rate": 3e-06, |
|
"log_odds_chosen": 0.14166930317878723, |
|
"log_odds_ratio": -0.6689570546150208, |
|
"logits/chosen": 395.5494079589844, |
|
"logits/rejected": 387.44561767578125, |
|
"logps/chosen": -1.3572882413864136, |
|
"logps/rejected": -1.4584226608276367, |
|
"loss": 1.705, |
|
"nll_loss": 1.769514799118042, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.06786441057920456, |
|
"rewards/margins": 0.005056709982454777, |
|
"rewards/rejected": -0.07292111963033676, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.38748137108792846, |
|
"grad_norm": 59.435298919677734, |
|
"learning_rate": 3.2500000000000002e-06, |
|
"log_odds_chosen": 0.09232059866189957, |
|
"log_odds_ratio": -0.7047046422958374, |
|
"logits/chosen": 390.05853271484375, |
|
"logits/rejected": 390.36346435546875, |
|
"logps/chosen": -1.2730042934417725, |
|
"logps/rejected": -1.3218134641647339, |
|
"loss": 1.6272, |
|
"nll_loss": 1.5361205339431763, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.0636502131819725, |
|
"rewards/margins": 0.0024404595606029034, |
|
"rewards/rejected": -0.06609068065881729, |
|
"step": 65 |
|
}, |
|
{ |
|
"epoch": 0.4172876304023845, |
|
"grad_norm": 32.72905349731445, |
|
"learning_rate": 3.5e-06, |
|
"log_odds_chosen": 0.03600925952196121, |
|
"log_odds_ratio": -0.7415024042129517, |
|
"logits/chosen": 381.87835693359375, |
|
"logits/rejected": 389.978271484375, |
|
"logps/chosen": -1.496351718902588, |
|
"logps/rejected": -1.5099536180496216, |
|
"loss": 1.7008, |
|
"nll_loss": 1.8172271251678467, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.07481758296489716, |
|
"rewards/margins": 0.0006800902774557471, |
|
"rewards/rejected": -0.07549767941236496, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.44709388971684055, |
|
"grad_norm": 31.950422286987305, |
|
"learning_rate": 3.7500000000000005e-06, |
|
"log_odds_chosen": 0.12743942439556122, |
|
"log_odds_ratio": -0.6888688206672668, |
|
"logits/chosen": 396.3226318359375, |
|
"logits/rejected": 383.38067626953125, |
|
"logps/chosen": -1.2637438774108887, |
|
"logps/rejected": -1.34519362449646, |
|
"loss": 1.6454, |
|
"nll_loss": 1.5880029201507568, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.06318718940019608, |
|
"rewards/margins": 0.0040724934078752995, |
|
"rewards/rejected": -0.06725968420505524, |
|
"step": 75 |
|
}, |
|
{ |
|
"epoch": 0.47690014903129657, |
|
"grad_norm": 29.037189483642578, |
|
"learning_rate": 4.000000000000001e-06, |
|
"log_odds_chosen": 0.18037892878055573, |
|
"log_odds_ratio": -0.6654509902000427, |
|
"logits/chosen": 381.05133056640625, |
|
"logits/rejected": 403.25115966796875, |
|
"logps/chosen": -1.220245599746704, |
|
"logps/rejected": -1.3405919075012207, |
|
"loss": 1.6425, |
|
"nll_loss": 1.6346063613891602, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.061012279242277145, |
|
"rewards/margins": 0.006017312873154879, |
|
"rewards/rejected": -0.06702959537506104, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.5067064083457526, |
|
"grad_norm": 48.85629653930664, |
|
"learning_rate": 4.25e-06, |
|
"log_odds_chosen": 0.13455167412757874, |
|
"log_odds_ratio": -0.7552928924560547, |
|
"logits/chosen": 406.13067626953125, |
|
"logits/rejected": 390.22216796875, |
|
"logps/chosen": -1.419901967048645, |
|
"logps/rejected": -1.4787737131118774, |
|
"loss": 1.6961, |
|
"nll_loss": 1.7476110458374023, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.07099510729312897, |
|
"rewards/margins": 0.002943576779216528, |
|
"rewards/rejected": -0.07393868267536163, |
|
"step": 85 |
|
}, |
|
{ |
|
"epoch": 0.5365126676602087, |
|
"grad_norm": 383.9244079589844, |
|
"learning_rate": 4.5e-06, |
|
"log_odds_chosen": 0.579018771648407, |
|
"log_odds_ratio": -0.5751021504402161, |
|
"logits/chosen": 401.8145446777344, |
|
"logits/rejected": 417.1114807128906, |
|
"logps/chosen": -1.2194725275039673, |
|
"logps/rejected": -1.6738338470458984, |
|
"loss": 1.4692, |
|
"nll_loss": 1.5031993389129639, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.0609736330807209, |
|
"rewards/margins": 0.02271805889904499, |
|
"rewards/rejected": -0.08369167894124985, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.5663189269746647, |
|
"grad_norm": 83.88288116455078, |
|
"learning_rate": 4.75e-06, |
|
"log_odds_chosen": 0.20128187537193298, |
|
"log_odds_ratio": -0.6592803597450256, |
|
"logits/chosen": 368.6065368652344, |
|
"logits/rejected": 382.3212890625, |
|
"logps/chosen": -1.1970250606536865, |
|
"logps/rejected": -1.2992660999298096, |
|
"loss": 1.6441, |
|
"nll_loss": 1.5431461334228516, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.059851258993148804, |
|
"rewards/margins": 0.00511204544454813, |
|
"rewards/rejected": -0.06496329605579376, |
|
"step": 95 |
|
}, |
|
{ |
|
"epoch": 0.5961251862891207, |
|
"grad_norm": 54.11213684082031, |
|
"learning_rate": 5e-06, |
|
"log_odds_chosen": 0.13380606472492218, |
|
"log_odds_ratio": -0.6779340505599976, |
|
"logits/chosen": 404.63409423828125, |
|
"logits/rejected": 441.92938232421875, |
|
"logps/chosen": -1.2189857959747314, |
|
"logps/rejected": -1.3092880249023438, |
|
"loss": 1.596, |
|
"nll_loss": 1.5926085710525513, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.06094928830862045, |
|
"rewards/margins": 0.004515114240348339, |
|
"rewards/rejected": -0.06546439975500107, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.6259314456035767, |
|
"grad_norm": 24.880090713500977, |
|
"learning_rate": 4.8795003647426654e-06, |
|
"log_odds_chosen": 0.2874816656112671, |
|
"log_odds_ratio": -0.6406186819076538, |
|
"logits/chosen": 386.64764404296875, |
|
"logits/rejected": 400.5997009277344, |
|
"logps/chosen": -1.0471909046173096, |
|
"logps/rejected": -1.2103850841522217, |
|
"loss": 1.5244, |
|
"nll_loss": 1.5330688953399658, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.05235954001545906, |
|
"rewards/margins": 0.008159706369042397, |
|
"rewards/rejected": -0.060519248247146606, |
|
"step": 105 |
|
}, |
|
{ |
|
"epoch": 0.6557377049180327, |
|
"grad_norm": 65.00325775146484, |
|
"learning_rate": 4.767312946227961e-06, |
|
"log_odds_chosen": 0.17585398256778717, |
|
"log_odds_ratio": -0.6596522331237793, |
|
"logits/chosen": 370.3504333496094, |
|
"logits/rejected": 369.62982177734375, |
|
"logps/chosen": -1.0811831951141357, |
|
"logps/rejected": -1.2061924934387207, |
|
"loss": 1.5148, |
|
"nll_loss": 1.530368685722351, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.054059166461229324, |
|
"rewards/margins": 0.006250458303838968, |
|
"rewards/rejected": -0.06030962988734245, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.6855439642324889, |
|
"grad_norm": 44.05381393432617, |
|
"learning_rate": 4.662524041201569e-06, |
|
"log_odds_chosen": 0.3691500723361969, |
|
"log_odds_ratio": -0.5788921117782593, |
|
"logits/chosen": 388.53631591796875, |
|
"logits/rejected": 396.33770751953125, |
|
"logps/chosen": -0.9749990701675415, |
|
"logps/rejected": -1.168736219406128, |
|
"loss": 1.5042, |
|
"nll_loss": 1.417229413986206, |
|
"rewards/accuracies": 0.625, |
|
"rewards/chosen": -0.048749953508377075, |
|
"rewards/margins": 0.009686857461929321, |
|
"rewards/rejected": -0.05843681842088699, |
|
"step": 115 |
|
}, |
|
{ |
|
"epoch": 0.7153502235469449, |
|
"grad_norm": 38.68400955200195, |
|
"learning_rate": 4.564354645876385e-06, |
|
"log_odds_chosen": 0.3533535599708557, |
|
"log_odds_ratio": -0.6087926626205444, |
|
"logits/chosen": 379.69366455078125, |
|
"logits/rejected": 379.962158203125, |
|
"logps/chosen": -1.0634129047393799, |
|
"logps/rejected": -1.2383795976638794, |
|
"loss": 1.5529, |
|
"nll_loss": 1.5442397594451904, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.053170643746852875, |
|
"rewards/margins": 0.008748333901166916, |
|
"rewards/rejected": -0.06191897392272949, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.7451564828614009, |
|
"grad_norm": 21.89837646484375, |
|
"learning_rate": 4.47213595499958e-06, |
|
"log_odds_chosen": 0.17026309669017792, |
|
"log_odds_ratio": -0.6800710558891296, |
|
"logits/chosen": 375.1714782714844, |
|
"logits/rejected": 387.94122314453125, |
|
"logps/chosen": -1.0947885513305664, |
|
"logps/rejected": -1.1978260278701782, |
|
"loss": 1.4603, |
|
"nll_loss": 1.4590686559677124, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -0.05473943427205086, |
|
"rewards/margins": 0.005151876248419285, |
|
"rewards/rejected": -0.059891313314437866, |
|
"step": 125 |
|
}, |
|
{ |
|
"epoch": 0.7749627421758569, |
|
"grad_norm": 22.374597549438477, |
|
"learning_rate": 4.385290096535147e-06, |
|
"log_odds_chosen": 0.1504744589328766, |
|
"log_odds_ratio": -0.6961590051651001, |
|
"logits/chosen": 401.68170166015625, |
|
"logits/rejected": 390.8814392089844, |
|
"logps/chosen": -1.1183803081512451, |
|
"logps/rejected": -1.1522414684295654, |
|
"loss": 1.5124, |
|
"nll_loss": 1.4771168231964111, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.055919013917446136, |
|
"rewards/margins": 0.0016930631827563047, |
|
"rewards/rejected": -0.05761207267642021, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.8047690014903129, |
|
"grad_norm": 54.11912155151367, |
|
"learning_rate": 4.303314829119352e-06, |
|
"log_odds_chosen": 0.09426576644182205, |
|
"log_odds_ratio": -0.7113849520683289, |
|
"logits/chosen": 408.5809326171875, |
|
"logits/rejected": 412.0887756347656, |
|
"logps/chosen": -1.1269690990447998, |
|
"logps/rejected": -1.195809245109558, |
|
"loss": 1.519, |
|
"nll_loss": 1.582168698310852, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -0.056348465383052826, |
|
"rewards/margins": 0.0034419982694089413, |
|
"rewards/rejected": -0.059790462255477905, |
|
"step": 135 |
|
}, |
|
{ |
|
"epoch": 0.834575260804769, |
|
"grad_norm": 23.130447387695312, |
|
"learning_rate": 4.2257712736425835e-06, |
|
"log_odds_chosen": -0.09378698468208313, |
|
"log_odds_ratio": -0.8063527345657349, |
|
"logits/chosen": 395.3038635253906, |
|
"logits/rejected": 402.5538330078125, |
|
"logps/chosen": -1.1263329982757568, |
|
"logps/rejected": -1.0577051639556885, |
|
"loss": 1.5115, |
|
"nll_loss": 1.6241823434829712, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -0.056316643953323364, |
|
"rewards/margins": -0.0034313846845179796, |
|
"rewards/rejected": -0.052885256707668304, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.8643815201192251, |
|
"grad_norm": 46.85090255737305, |
|
"learning_rate": 4.1522739926869985e-06, |
|
"log_odds_chosen": -0.024486612528562546, |
|
"log_odds_ratio": -0.735202431678772, |
|
"logits/chosen": 391.74871826171875, |
|
"logits/rejected": 396.13165283203125, |
|
"logps/chosen": -1.1859979629516602, |
|
"logps/rejected": -1.1640434265136719, |
|
"loss": 1.5112, |
|
"nll_loss": 1.5025219917297363, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.05929989740252495, |
|
"rewards/margins": -0.0010977205820381641, |
|
"rewards/rejected": -0.05820217728614807, |
|
"step": 145 |
|
}, |
|
{ |
|
"epoch": 0.8941877794336811, |
|
"grad_norm": 32.988243103027344, |
|
"learning_rate": 4.082482904638631e-06, |
|
"log_odds_chosen": 0.3807345926761627, |
|
"log_odds_ratio": -0.559662401676178, |
|
"logits/chosen": 399.6306457519531, |
|
"logits/rejected": 417.28515625, |
|
"logps/chosen": -1.0708329677581787, |
|
"logps/rejected": -1.319719672203064, |
|
"loss": 1.4701, |
|
"nll_loss": 1.3841121196746826, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.053541649132966995, |
|
"rewards/margins": 0.012444335967302322, |
|
"rewards/rejected": -0.06598598510026932, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.9239940387481371, |
|
"grad_norm": 19.93175506591797, |
|
"learning_rate": 4.016096644512495e-06, |
|
"log_odds_chosen": 0.13164076209068298, |
|
"log_odds_ratio": -0.6793235540390015, |
|
"logits/chosen": 381.662353515625, |
|
"logits/rejected": 396.54168701171875, |
|
"logps/chosen": -1.1232904195785522, |
|
"logps/rejected": -1.203460931777954, |
|
"loss": 1.4306, |
|
"nll_loss": 1.322166919708252, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.05616452172398567, |
|
"rewards/margins": 0.004008529242128134, |
|
"rewards/rejected": -0.06017305329442024, |
|
"step": 155 |
|
}, |
|
{ |
|
"epoch": 0.9538002980625931, |
|
"grad_norm": 48.30046844482422, |
|
"learning_rate": 3.952847075210474e-06, |
|
"log_odds_chosen": 0.010590496473014355, |
|
"log_odds_ratio": -0.794399619102478, |
|
"logits/chosen": 387.1076354980469, |
|
"logits/rejected": 432.69384765625, |
|
"logps/chosen": -1.0055660009384155, |
|
"logps/rejected": -1.0430043935775757, |
|
"loss": 1.3903, |
|
"nll_loss": 1.3440879583358765, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -0.050278306007385254, |
|
"rewards/margins": 0.0018719173967838287, |
|
"rewards/rejected": -0.052150219678878784, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.9836065573770492, |
|
"grad_norm": 17.599102020263672, |
|
"learning_rate": 3.892494720807615e-06, |
|
"log_odds_chosen": 0.021682387217879295, |
|
"log_odds_ratio": -0.7344300746917725, |
|
"logits/chosen": 397.306884765625, |
|
"logits/rejected": 409.34222412109375, |
|
"logps/chosen": -1.1095333099365234, |
|
"logps/rejected": -1.1364104747772217, |
|
"loss": 1.436, |
|
"nll_loss": 1.4027729034423828, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -0.05547667294740677, |
|
"rewards/margins": 0.0013438586611300707, |
|
"rewards/rejected": -0.05682053044438362, |
|
"step": 165 |
|
}, |
|
{ |
|
"epoch": 0.9955290611028316, |
|
"eval_log_odds_chosen": 0.21922710537910461, |
|
"eval_log_odds_ratio": -0.6881988644599915, |
|
"eval_logits/chosen": 322.9157409667969, |
|
"eval_logits/rejected": 294.26885986328125, |
|
"eval_logps/chosen": -1.0047651529312134, |
|
"eval_logps/rejected": -1.141276478767395, |
|
"eval_loss": 1.4639270305633545, |
|
"eval_nll_loss": 1.4151870012283325, |
|
"eval_rewards/accuracies": 0.5539568066596985, |
|
"eval_rewards/chosen": -0.05023825913667679, |
|
"eval_rewards/margins": 0.006825567223131657, |
|
"eval_rewards/rejected": -0.05706382542848587, |
|
"eval_runtime": 112.2165, |
|
"eval_samples_per_second": 4.928, |
|
"eval_steps_per_second": 1.239, |
|
"step": 167 |
|
}, |
|
{ |
|
"epoch": 1.0134128166915053, |
|
"grad_norm": 19.598791122436523, |
|
"learning_rate": 3.834824944236852e-06, |
|
"log_odds_chosen": 0.45991507172584534, |
|
"log_odds_ratio": -0.5599726438522339, |
|
"logits/chosen": 381.36883544921875, |
|
"logits/rejected": 404.8132629394531, |
|
"logps/chosen": -0.8878811001777649, |
|
"logps/rejected": -1.1503413915634155, |
|
"loss": 1.2939, |
|
"nll_loss": 1.1531199216842651, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.044394053518772125, |
|
"rewards/margins": 0.013123014941811562, |
|
"rewards/rejected": -0.057517070323228836, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 1.0432190760059612, |
|
"grad_norm": 20.02121353149414, |
|
"learning_rate": 3.7796447300922724e-06, |
|
"log_odds_chosen": 0.8954402804374695, |
|
"log_odds_ratio": -0.4191034436225891, |
|
"logits/chosen": 360.7755126953125, |
|
"logits/rejected": 400.88580322265625, |
|
"logps/chosen": -0.6612526774406433, |
|
"logps/rejected": -1.1563608646392822, |
|
"loss": 1.0774, |
|
"nll_loss": 1.1424416303634644, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.033062636852264404, |
|
"rewards/margins": 0.024755412712693214, |
|
"rewards/rejected": -0.05781804770231247, |
|
"step": 175 |
|
}, |
|
{ |
|
"epoch": 1.0730253353204173, |
|
"grad_norm": 31.31179428100586, |
|
"learning_rate": 3.72677996249965e-06, |
|
"log_odds_chosen": 0.9305903315544128, |
|
"log_odds_ratio": -0.4297749102115631, |
|
"logits/chosen": 358.54632568359375, |
|
"logits/rejected": 333.3828125, |
|
"logps/chosen": -0.7619189023971558, |
|
"logps/rejected": -1.288059949874878, |
|
"loss": 1.1343, |
|
"nll_loss": 1.1943809986114502, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03809594735503197, |
|
"rewards/margins": 0.02630705200135708, |
|
"rewards/rejected": -0.0644029974937439, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 1.1028315946348732, |
|
"grad_norm": 18.475496292114258, |
|
"learning_rate": 3.6760731104690393e-06, |
|
"log_odds_chosen": 0.9789319038391113, |
|
"log_odds_ratio": -0.39594078063964844, |
|
"logits/chosen": 388.92523193359375, |
|
"logits/rejected": 379.08624267578125, |
|
"logps/chosen": -0.6655952334403992, |
|
"logps/rejected": -1.1871401071548462, |
|
"loss": 1.0198, |
|
"nll_loss": 0.9870488047599792, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.03327976167201996, |
|
"rewards/margins": 0.026077240705490112, |
|
"rewards/rejected": -0.05935700982809067, |
|
"step": 185 |
|
}, |
|
{ |
|
"epoch": 1.1326378539493294, |
|
"grad_norm": 15.556231498718262, |
|
"learning_rate": 3.6273812505500587e-06, |
|
"log_odds_chosen": 0.5961565971374512, |
|
"log_odds_ratio": -0.5247331857681274, |
|
"logits/chosen": 364.07965087890625, |
|
"logits/rejected": 406.9730529785156, |
|
"logps/chosen": -0.7741492986679077, |
|
"logps/rejected": -1.1193947792053223, |
|
"loss": 1.0997, |
|
"nll_loss": 1.0372028350830078, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -0.03870746120810509, |
|
"rewards/margins": 0.017262274399399757, |
|
"rewards/rejected": -0.055969737470149994, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 1.1624441132637853, |
|
"grad_norm": 20.581634521484375, |
|
"learning_rate": 3.5805743701971648e-06, |
|
"log_odds_chosen": 0.9057777523994446, |
|
"log_odds_ratio": -0.40288910269737244, |
|
"logits/chosen": 387.034912109375, |
|
"logits/rejected": 400.3788146972656, |
|
"logps/chosen": -0.8112454414367676, |
|
"logps/rejected": -1.2839386463165283, |
|
"loss": 1.143, |
|
"nll_loss": 1.1492406129837036, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -0.04056227579712868, |
|
"rewards/margins": 0.023634660989046097, |
|
"rewards/rejected": -0.06419692933559418, |
|
"step": 195 |
|
}, |
|
{ |
|
"epoch": 1.1922503725782414, |
|
"grad_norm": 25.256078720092773, |
|
"learning_rate": 3.5355339059327378e-06, |
|
"log_odds_chosen": 0.968602180480957, |
|
"log_odds_ratio": -0.41492360830307007, |
|
"logits/chosen": 408.2672424316406, |
|
"logits/rejected": 392.72406005859375, |
|
"logps/chosen": -0.7001940011978149, |
|
"logps/rejected": -1.2566159963607788, |
|
"loss": 1.1031, |
|
"nll_loss": 0.9941505193710327, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03500969707965851, |
|
"rewards/margins": 0.02782110869884491, |
|
"rewards/rejected": -0.06283080577850342, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 1.2220566318926975, |
|
"grad_norm": 20.430418014526367, |
|
"learning_rate": 3.4921514788478916e-06, |
|
"log_odds_chosen": 0.9490750432014465, |
|
"log_odds_ratio": -0.43051719665527344, |
|
"logits/chosen": 366.13177490234375, |
|
"logits/rejected": 359.82196044921875, |
|
"logps/chosen": -0.7287150025367737, |
|
"logps/rejected": -1.228252649307251, |
|
"loss": 1.1093, |
|
"nll_loss": 1.1782151460647583, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.036435749381780624, |
|
"rewards/margins": 0.024976884946227074, |
|
"rewards/rejected": -0.061412639915943146, |
|
"step": 205 |
|
}, |
|
{ |
|
"epoch": 1.2518628912071534, |
|
"grad_norm": 29.456756591796875, |
|
"learning_rate": 3.450327796711771e-06, |
|
"log_odds_chosen": 1.1124473810195923, |
|
"log_odds_ratio": -0.36473917961120605, |
|
"logits/chosen": 371.9077453613281, |
|
"logits/rejected": 400.37908935546875, |
|
"logps/chosen": -0.6593618392944336, |
|
"logps/rejected": -1.292751431465149, |
|
"loss": 1.0725, |
|
"nll_loss": 1.041669487953186, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.03296809643507004, |
|
"rewards/margins": 0.03166947886347771, |
|
"rewards/rejected": -0.06463757157325745, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 1.2816691505216096, |
|
"grad_norm": 51.35958480834961, |
|
"learning_rate": 3.409971697352368e-06, |
|
"log_odds_chosen": 1.1422733068466187, |
|
"log_odds_ratio": -0.3576507568359375, |
|
"logits/chosen": 396.532470703125, |
|
"logits/rejected": 379.38525390625, |
|
"logps/chosen": -0.7450262904167175, |
|
"logps/rejected": -1.4131982326507568, |
|
"loss": 1.0757, |
|
"nll_loss": 1.0690205097198486, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.0372513122856617, |
|
"rewards/margins": 0.033408597111701965, |
|
"rewards/rejected": -0.07065991312265396, |
|
"step": 215 |
|
}, |
|
{ |
|
"epoch": 1.3114754098360657, |
|
"grad_norm": 16.197195053100586, |
|
"learning_rate": 3.3709993123162106e-06, |
|
"log_odds_chosen": 0.7627310752868652, |
|
"log_odds_ratio": -0.46217769384384155, |
|
"logits/chosen": 383.55572509765625, |
|
"logits/rejected": 377.1234436035156, |
|
"logps/chosen": -0.7894420623779297, |
|
"logps/rejected": -1.196644902229309, |
|
"loss": 1.0384, |
|
"nll_loss": 1.0212104320526123, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.03947211056947708, |
|
"rewards/margins": 0.02036014385521412, |
|
"rewards/rejected": -0.05983225256204605, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 1.3412816691505216, |
|
"grad_norm": 26.38709831237793, |
|
"learning_rate": 3.3333333333333333e-06, |
|
"log_odds_chosen": 0.5317455530166626, |
|
"log_odds_ratio": -0.5077921152114868, |
|
"logits/chosen": 383.05657958984375, |
|
"logits/rejected": 375.5099792480469, |
|
"logps/chosen": -0.8590075373649597, |
|
"logps/rejected": -1.1633431911468506, |
|
"loss": 1.0481, |
|
"nll_loss": 1.092240333557129, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.04295038431882858, |
|
"rewards/margins": 0.01521677989512682, |
|
"rewards/rejected": -0.05816715955734253, |
|
"step": 225 |
|
}, |
|
{ |
|
"epoch": 1.3710879284649775, |
|
"grad_norm": 22.215499877929688, |
|
"learning_rate": 3.296902366978936e-06, |
|
"log_odds_chosen": 1.152860403060913, |
|
"log_odds_ratio": -0.33905166387557983, |
|
"logits/chosen": 360.29229736328125, |
|
"logits/rejected": 377.34967041015625, |
|
"logps/chosen": -0.6110543012619019, |
|
"logps/rejected": -1.235459804534912, |
|
"loss": 1.0305, |
|
"nll_loss": 0.8882306814193726, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.03055271878838539, |
|
"rewards/margins": 0.031220275908708572, |
|
"rewards/rejected": -0.061772990971803665, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 1.4008941877794336, |
|
"grad_norm": 15.053303718566895, |
|
"learning_rate": 3.2616403652672114e-06, |
|
"log_odds_chosen": 1.173457384109497, |
|
"log_odds_ratio": -0.39741092920303345, |
|
"logits/chosen": 382.1279602050781, |
|
"logits/rejected": 394.14453125, |
|
"logps/chosen": -0.6574608087539673, |
|
"logps/rejected": -1.3275513648986816, |
|
"loss": 1.041, |
|
"nll_loss": 0.9579538106918335, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -0.032873041927814484, |
|
"rewards/margins": 0.033504534512758255, |
|
"rewards/rejected": -0.06637756526470184, |
|
"step": 235 |
|
}, |
|
{ |
|
"epoch": 1.4307004470938898, |
|
"grad_norm": 17.994874954223633, |
|
"learning_rate": 3.2274861218395142e-06, |
|
"log_odds_chosen": 0.7307401895523071, |
|
"log_odds_ratio": -0.4488092362880707, |
|
"logits/chosen": 405.40802001953125, |
|
"logits/rejected": 409.69647216796875, |
|
"logps/chosen": -0.7471104264259338, |
|
"logps/rejected": -1.1478822231292725, |
|
"loss": 1.062, |
|
"nll_loss": 1.034090518951416, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.03735552355647087, |
|
"rewards/margins": 0.02003859169781208, |
|
"rewards/rejected": -0.057394109666347504, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 1.4605067064083457, |
|
"grad_norm": 14.603346824645996, |
|
"learning_rate": 3.1943828249997e-06, |
|
"log_odds_chosen": 0.9838510751724243, |
|
"log_odds_ratio": -0.39035505056381226, |
|
"logits/chosen": 394.9382629394531, |
|
"logits/rejected": 383.0425109863281, |
|
"logps/chosen": -0.604079008102417, |
|
"logps/rejected": -1.1444944143295288, |
|
"loss": 1.0828, |
|
"nll_loss": 1.0692133903503418, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.03020395338535309, |
|
"rewards/margins": 0.027020767331123352, |
|
"rewards/rejected": -0.05722472816705704, |
|
"step": 245 |
|
}, |
|
{ |
|
"epoch": 1.4903129657228018, |
|
"grad_norm": 24.562320709228516, |
|
"learning_rate": 3.1622776601683796e-06, |
|
"log_odds_chosen": 1.0623723268508911, |
|
"log_odds_ratio": -0.41066059470176697, |
|
"logits/chosen": 374.6037902832031, |
|
"logits/rejected": 381.32269287109375, |
|
"logps/chosen": -0.6770502924919128, |
|
"logps/rejected": -1.2543606758117676, |
|
"loss": 1.0094, |
|
"nll_loss": 0.9133090972900391, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.03385251387953758, |
|
"rewards/margins": 0.028865519911050797, |
|
"rewards/rejected": -0.06271803379058838, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 1.520119225037258, |
|
"grad_norm": 16.239404678344727, |
|
"learning_rate": 3.131121455425748e-06, |
|
"log_odds_chosen": 1.1072403192520142, |
|
"log_odds_ratio": -0.34550726413726807, |
|
"logits/chosen": 394.74688720703125, |
|
"logits/rejected": 396.5551452636719, |
|
"logps/chosen": -0.6057699918746948, |
|
"logps/rejected": -1.1659581661224365, |
|
"loss": 1.0541, |
|
"nll_loss": 0.9463102221488953, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -0.03028850257396698, |
|
"rewards/margins": 0.028009409084916115, |
|
"rewards/rejected": -0.058297913521528244, |
|
"step": 255 |
|
}, |
|
{ |
|
"epoch": 1.5499254843517138, |
|
"grad_norm": 29.02210235595703, |
|
"learning_rate": 3.1008683647302113e-06, |
|
"log_odds_chosen": 0.9046269655227661, |
|
"log_odds_ratio": -0.42992085218429565, |
|
"logits/chosen": 375.5458984375, |
|
"logits/rejected": 415.7400817871094, |
|
"logps/chosen": -0.7795925140380859, |
|
"logps/rejected": -1.3336765766143799, |
|
"loss": 1.021, |
|
"nll_loss": 1.037724256515503, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -0.038979630917310715, |
|
"rewards/margins": 0.02770419791340828, |
|
"rewards/rejected": -0.066683828830719, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 1.5797317436661698, |
|
"grad_norm": 16.736574172973633, |
|
"learning_rate": 3.0714755841697565e-06, |
|
"log_odds_chosen": 1.028351902961731, |
|
"log_odds_ratio": -0.4398733079433441, |
|
"logits/chosen": 384.7108154296875, |
|
"logits/rejected": 406.8509521484375, |
|
"logps/chosen": -0.717097282409668, |
|
"logps/rejected": -1.318499207496643, |
|
"loss": 1.104, |
|
"nll_loss": 1.0128785371780396, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03585486486554146, |
|
"rewards/margins": 0.030070099979639053, |
|
"rewards/rejected": -0.06592496484518051, |
|
"step": 265 |
|
}, |
|
{ |
|
"epoch": 1.6095380029806259, |
|
"grad_norm": 17.67503547668457, |
|
"learning_rate": 3.0429030972509227e-06, |
|
"log_odds_chosen": 0.7785763740539551, |
|
"log_odds_ratio": -0.4546143412590027, |
|
"logits/chosen": 369.1441955566406, |
|
"logits/rejected": 379.18487548828125, |
|
"logps/chosen": -0.7971286177635193, |
|
"logps/rejected": -1.235899567604065, |
|
"loss": 1.0973, |
|
"nll_loss": 1.1573312282562256, |
|
"rewards/accuracies": 0.8500000238418579, |
|
"rewards/chosen": -0.0398564375936985, |
|
"rewards/margins": 0.021938541904091835, |
|
"rewards/rejected": -0.06179497763514519, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 1.639344262295082, |
|
"grad_norm": 15.161911964416504, |
|
"learning_rate": 3.0151134457776365e-06, |
|
"log_odds_chosen": 0.9045177698135376, |
|
"log_odds_ratio": -0.4321528375148773, |
|
"logits/chosen": 360.2083435058594, |
|
"logits/rejected": 351.1839904785156, |
|
"logps/chosen": -0.6309309005737305, |
|
"logps/rejected": -1.0601168870925903, |
|
"loss": 1.0614, |
|
"nll_loss": 0.976634681224823, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.031546544283628464, |
|
"rewards/margins": 0.021459298208355904, |
|
"rewards/rejected": -0.05300584435462952, |
|
"step": 275 |
|
}, |
|
{ |
|
"epoch": 1.669150521609538, |
|
"grad_norm": 14.985346794128418, |
|
"learning_rate": 2.988071523335984e-06, |
|
"log_odds_chosen": 0.8889838457107544, |
|
"log_odds_ratio": -0.5245683193206787, |
|
"logits/chosen": 404.428955078125, |
|
"logits/rejected": 391.63189697265625, |
|
"logps/chosen": -0.6854910850524902, |
|
"logps/rejected": -1.1978435516357422, |
|
"loss": 1.0314, |
|
"nll_loss": 1.0269334316253662, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.03427455574274063, |
|
"rewards/margins": 0.02561761811375618, |
|
"rewards/rejected": -0.05989217013120651, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 1.698956780923994, |
|
"grad_norm": 13.728422164916992, |
|
"learning_rate": 2.961744388795462e-06, |
|
"log_odds_chosen": 0.9078812599182129, |
|
"log_odds_ratio": -0.4240701198577881, |
|
"logits/chosen": 369.9886779785156, |
|
"logits/rejected": 374.25213623046875, |
|
"logps/chosen": -0.640867292881012, |
|
"logps/rejected": -1.1350326538085938, |
|
"loss": 1.0129, |
|
"nll_loss": 0.9724606275558472, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03204336389899254, |
|
"rewards/margins": 0.024708271026611328, |
|
"rewards/rejected": -0.05675163120031357, |
|
"step": 285 |
|
}, |
|
{ |
|
"epoch": 1.7287630402384502, |
|
"grad_norm": 13.05510425567627, |
|
"learning_rate": 2.9361010975735177e-06, |
|
"log_odds_chosen": 0.9024085998535156, |
|
"log_odds_ratio": -0.41897106170654297, |
|
"logits/chosen": 383.33331298828125, |
|
"logits/rejected": 421.84710693359375, |
|
"logps/chosen": -0.7976008057594299, |
|
"logps/rejected": -1.2910783290863037, |
|
"loss": 1.0749, |
|
"nll_loss": 1.0102927684783936, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.039880041033029556, |
|
"rewards/margins": 0.024673879146575928, |
|
"rewards/rejected": -0.06455391645431519, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 1.758569299552906, |
|
"grad_norm": 22.535524368286133, |
|
"learning_rate": 2.9111125486979104e-06, |
|
"log_odds_chosen": 0.82624751329422, |
|
"log_odds_ratio": -0.44848528504371643, |
|
"logits/chosen": 358.3087463378906, |
|
"logits/rejected": 401.21539306640625, |
|
"logps/chosen": -0.7310397028923035, |
|
"logps/rejected": -1.1755989789962769, |
|
"loss": 1.0777, |
|
"nll_loss": 1.026176929473877, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03655198961496353, |
|
"rewards/margins": 0.022227967157959938, |
|
"rewards/rejected": -0.05877995491027832, |
|
"step": 295 |
|
}, |
|
{ |
|
"epoch": 1.788375558867362, |
|
"grad_norm": 20.802898406982422, |
|
"learning_rate": 2.8867513459481293e-06, |
|
"log_odds_chosen": 1.3168494701385498, |
|
"log_odds_ratio": -0.3224945366382599, |
|
"logits/chosen": 398.9639587402344, |
|
"logits/rejected": 377.6863098144531, |
|
"logps/chosen": -0.6201862096786499, |
|
"logps/rejected": -1.2738819122314453, |
|
"loss": 0.9939, |
|
"nll_loss": 0.9130023121833801, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.031009310856461525, |
|
"rewards/margins": 0.03268478438258171, |
|
"rewards/rejected": -0.06369409710168839, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 1.8181818181818183, |
|
"grad_norm": 17.038707733154297, |
|
"learning_rate": 2.862991671569341e-06, |
|
"log_odds_chosen": 0.5965818762779236, |
|
"log_odds_ratio": -0.5011720657348633, |
|
"logits/chosen": 391.8568115234375, |
|
"logits/rejected": 401.20648193359375, |
|
"logps/chosen": -0.8930565714836121, |
|
"logps/rejected": -1.2144049406051636, |
|
"loss": 1.0267, |
|
"nll_loss": 1.134427785873413, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -0.044652827084064484, |
|
"rewards/margins": 0.016067421063780785, |
|
"rewards/rejected": -0.060720253735780716, |
|
"step": 305 |
|
}, |
|
{ |
|
"epoch": 1.8479880774962743, |
|
"grad_norm": 16.012582778930664, |
|
"learning_rate": 2.839809171235324e-06, |
|
"log_odds_chosen": 0.9494487047195435, |
|
"log_odds_ratio": -0.5154255032539368, |
|
"logits/chosen": 374.0430908203125, |
|
"logits/rejected": 383.86016845703125, |
|
"logps/chosen": -0.7265419960021973, |
|
"logps/rejected": -1.3359992504119873, |
|
"loss": 1.0959, |
|
"nll_loss": 1.0941182374954224, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -0.03632710129022598, |
|
"rewards/margins": 0.03047286346554756, |
|
"rewards/rejected": -0.06679996103048325, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 1.8777943368107302, |
|
"grad_norm": 13.797101974487305, |
|
"learning_rate": 2.817180849095055e-06, |
|
"log_odds_chosen": 0.4637278616428375, |
|
"log_odds_ratio": -0.5917452573776245, |
|
"logits/chosen": 352.93426513671875, |
|
"logits/rejected": 371.59765625, |
|
"logps/chosen": -0.9915712475776672, |
|
"logps/rejected": -1.311402440071106, |
|
"loss": 1.0986, |
|
"nll_loss": 1.2321771383285522, |
|
"rewards/accuracies": 0.675000011920929, |
|
"rewards/chosen": -0.04957855865359306, |
|
"rewards/margins": 0.015991564840078354, |
|
"rewards/rejected": -0.06557012349367142, |
|
"step": 315 |
|
}, |
|
{ |
|
"epoch": 1.9076005961251863, |
|
"grad_norm": 15.759064674377441, |
|
"learning_rate": 2.7950849718747376e-06, |
|
"log_odds_chosen": 1.112966537475586, |
|
"log_odds_ratio": -0.37106412649154663, |
|
"logits/chosen": 374.7521667480469, |
|
"logits/rejected": 395.431396484375, |
|
"logps/chosen": -0.642137885093689, |
|
"logps/rejected": -1.249108910560608, |
|
"loss": 0.9927, |
|
"nll_loss": 0.9422394037246704, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.03210689499974251, |
|
"rewards/margins": 0.03034854494035244, |
|
"rewards/rejected": -0.0624554380774498, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 1.9374068554396424, |
|
"grad_norm": 18.297237396240234, |
|
"learning_rate": 2.773500981126146e-06, |
|
"log_odds_chosen": 1.2229207754135132, |
|
"log_odds_ratio": -0.34499865770339966, |
|
"logits/chosen": 373.3489685058594, |
|
"logits/rejected": 404.5666809082031, |
|
"logps/chosen": -0.6735208630561829, |
|
"logps/rejected": -1.3717957735061646, |
|
"loss": 0.983, |
|
"nll_loss": 0.9137915372848511, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -0.03367604315280914, |
|
"rewards/margins": 0.034913744777441025, |
|
"rewards/rejected": -0.06858979165554047, |
|
"step": 325 |
|
}, |
|
{ |
|
"epoch": 1.9672131147540983, |
|
"grad_norm": 21.227933883666992, |
|
"learning_rate": 2.752409412815902e-06, |
|
"log_odds_chosen": 0.8807786107063293, |
|
"log_odds_ratio": -0.40755215287208557, |
|
"logits/chosen": 366.36541748046875, |
|
"logits/rejected": 375.8720703125, |
|
"logps/chosen": -0.6931061148643494, |
|
"logps/rejected": -1.1950337886810303, |
|
"loss": 1.0248, |
|
"nll_loss": 0.8495981097221375, |
|
"rewards/accuracies": 0.875, |
|
"rewards/chosen": -0.03465530276298523, |
|
"rewards/margins": 0.025096386671066284, |
|
"rewards/rejected": -0.059751689434051514, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 1.9970193740685542, |
|
"grad_norm": 15.724347114562988, |
|
"learning_rate": 2.7317918235407652e-06, |
|
"log_odds_chosen": 0.43006667494773865, |
|
"log_odds_ratio": -0.608497142791748, |
|
"logits/chosen": 391.9339904785156, |
|
"logits/rejected": 385.08636474609375, |
|
"logps/chosen": -0.966523289680481, |
|
"logps/rejected": -1.2040079832077026, |
|
"loss": 1.0918, |
|
"nll_loss": 1.2487541437149048, |
|
"rewards/accuracies": 0.6499999761581421, |
|
"rewards/chosen": -0.04832616448402405, |
|
"rewards/margins": 0.011874236166477203, |
|
"rewards/rejected": -0.06020040065050125, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 1.9970193740685542, |
|
"eval_log_odds_chosen": 0.2209286242723465, |
|
"eval_log_odds_ratio": -0.7027979493141174, |
|
"eval_logits/chosen": 313.30999755859375, |
|
"eval_logits/rejected": 284.8743896484375, |
|
"eval_logps/chosen": -1.0012338161468506, |
|
"eval_logps/rejected": -1.1474785804748535, |
|
"eval_loss": 1.423282504081726, |
|
"eval_nll_loss": 1.36614990234375, |
|
"eval_rewards/accuracies": 0.49640288949012756, |
|
"eval_rewards/chosen": -0.05006168410181999, |
|
"eval_rewards/margins": 0.007312240544706583, |
|
"eval_rewards/rejected": -0.057373929768800735, |
|
"eval_runtime": 112.3167, |
|
"eval_samples_per_second": 4.924, |
|
"eval_steps_per_second": 1.238, |
|
"step": 335 |
|
}, |
|
{ |
|
"epoch": 2.0268256333830106, |
|
"grad_norm": 22.363670349121094, |
|
"learning_rate": 2.711630722733202e-06, |
|
"log_odds_chosen": 1.9970035552978516, |
|
"log_odds_ratio": -0.19410225749015808, |
|
"logits/chosen": 394.3930358886719, |
|
"logits/rejected": 370.94024658203125, |
|
"logps/chosen": -0.36809104681015015, |
|
"logps/rejected": -1.342511534690857, |
|
"loss": 0.6352, |
|
"nll_loss": 0.6902519464492798, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.018404554575681686, |
|
"rewards/margins": 0.04872102662920952, |
|
"rewards/rejected": -0.0671255812048912, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 2.0566318926974665, |
|
"grad_norm": 17.443252563476562, |
|
"learning_rate": 2.691909510290828e-06, |
|
"log_odds_chosen": 2.4873645305633545, |
|
"log_odds_ratio": -0.12294415384531021, |
|
"logits/chosen": 353.0312194824219, |
|
"logits/rejected": 357.0645751953125, |
|
"logps/chosen": -0.3527178168296814, |
|
"logps/rejected": -1.623201608657837, |
|
"loss": 0.5679, |
|
"nll_loss": 0.5656304359436035, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.01763588935136795, |
|
"rewards/margins": 0.06352418661117554, |
|
"rewards/rejected": -0.08116006851196289, |
|
"step": 345 |
|
}, |
|
{ |
|
"epoch": 2.0864381520119224, |
|
"grad_norm": 13.353521347045898, |
|
"learning_rate": 2.6726124191242444e-06, |
|
"log_odds_chosen": 2.534590005874634, |
|
"log_odds_ratio": -0.11770644038915634, |
|
"logits/chosen": 355.3052673339844, |
|
"logits/rejected": 389.3919372558594, |
|
"logps/chosen": -0.3792329430580139, |
|
"logps/rejected": -1.8535314798355103, |
|
"loss": 0.5665, |
|
"nll_loss": 0.5773240327835083, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.018961649388074875, |
|
"rewards/margins": 0.07371493428945541, |
|
"rewards/rejected": -0.09267657995223999, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 2.1162444113263787, |
|
"grad_norm": 15.57699203491211, |
|
"learning_rate": 2.6537244621713765e-06, |
|
"log_odds_chosen": 2.286215305328369, |
|
"log_odds_ratio": -0.15262895822525024, |
|
"logits/chosen": 352.2881164550781, |
|
"logits/rejected": 369.99462890625, |
|
"logps/chosen": -0.3723372220993042, |
|
"logps/rejected": -1.5929332971572876, |
|
"loss": 0.5759, |
|
"nll_loss": 0.6265019178390503, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.01861685886979103, |
|
"rewards/margins": 0.06102980300784111, |
|
"rewards/rejected": -0.07964666187763214, |
|
"step": 355 |
|
}, |
|
{ |
|
"epoch": 2.1460506706408347, |
|
"grad_norm": 11.725736618041992, |
|
"learning_rate": 2.6352313834736496e-06, |
|
"log_odds_chosen": 2.5719923973083496, |
|
"log_odds_ratio": -0.12209601700305939, |
|
"logits/chosen": 359.28118896484375, |
|
"logits/rejected": 400.27215576171875, |
|
"logps/chosen": -0.3560808300971985, |
|
"logps/rejected": -1.588376760482788, |
|
"loss": 0.5392, |
|
"nll_loss": 0.539588451385498, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.017804041504859924, |
|
"rewards/margins": 0.06161479279398918, |
|
"rewards/rejected": -0.0794188380241394, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 2.1758569299552906, |
|
"grad_norm": 11.369336128234863, |
|
"learning_rate": 2.6171196129510684e-06, |
|
"log_odds_chosen": 2.051750421524048, |
|
"log_odds_ratio": -0.1606641709804535, |
|
"logits/chosen": 347.43212890625, |
|
"logits/rejected": 334.7518310546875, |
|
"logps/chosen": -0.3488374650478363, |
|
"logps/rejected": -1.3597064018249512, |
|
"loss": 0.5618, |
|
"nll_loss": 0.5548380613327026, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.017441872507333755, |
|
"rewards/margins": 0.05054344981908798, |
|
"rewards/rejected": -0.06798531860113144, |
|
"step": 365 |
|
}, |
|
{ |
|
"epoch": 2.2056631892697465, |
|
"grad_norm": 14.05441665649414, |
|
"learning_rate": 2.599376224550182e-06, |
|
"log_odds_chosen": 2.0613701343536377, |
|
"log_odds_ratio": -0.20328454673290253, |
|
"logits/chosen": 318.46661376953125, |
|
"logits/rejected": 339.12286376953125, |
|
"logps/chosen": -0.36191436648368835, |
|
"logps/rejected": -1.3976794481277466, |
|
"loss": 0.5944, |
|
"nll_loss": 0.6158479452133179, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -0.018095718696713448, |
|
"rewards/margins": 0.051788248121738434, |
|
"rewards/rejected": -0.06988397240638733, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 2.235469448584203, |
|
"grad_norm": 13.069171905517578, |
|
"learning_rate": 2.5819888974716113e-06, |
|
"log_odds_chosen": 1.9356635808944702, |
|
"log_odds_ratio": -0.19234418869018555, |
|
"logits/chosen": 366.2290344238281, |
|
"logits/rejected": 386.2459411621094, |
|
"logps/chosen": -0.4155746400356293, |
|
"logps/rejected": -1.442433476448059, |
|
"loss": 0.5766, |
|
"nll_loss": 0.5814634561538696, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.020778732374310493, |
|
"rewards/margins": 0.051342952996492386, |
|
"rewards/rejected": -0.07212167233228683, |
|
"step": 375 |
|
}, |
|
{ |
|
"epoch": 2.2652757078986587, |
|
"grad_norm": 13.310582160949707, |
|
"learning_rate": 2.564945880212886e-06, |
|
"log_odds_chosen": 2.3414645195007324, |
|
"log_odds_ratio": -0.1297520399093628, |
|
"logits/chosen": 361.81097412109375, |
|
"logits/rejected": 347.11090087890625, |
|
"logps/chosen": -0.2917303442955017, |
|
"logps/rejected": -1.3597028255462646, |
|
"loss": 0.5492, |
|
"nll_loss": 0.5354436039924622, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.014586517587304115, |
|
"rewards/margins": 0.05339862033724785, |
|
"rewards/rejected": -0.06798513978719711, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 2.2950819672131146, |
|
"grad_norm": 11.30221939086914, |
|
"learning_rate": 2.5482359571881276e-06, |
|
"log_odds_chosen": 2.427121162414551, |
|
"log_odds_ratio": -0.14391574263572693, |
|
"logits/chosen": 351.34454345703125, |
|
"logits/rejected": 348.7616271972656, |
|
"logps/chosen": -0.30433347821235657, |
|
"logps/rejected": -1.4669969081878662, |
|
"loss": 0.5393, |
|
"nll_loss": 0.5202994346618652, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.015216675586998463, |
|
"rewards/margins": 0.058133166283369064, |
|
"rewards/rejected": -0.07334984838962555, |
|
"step": 385 |
|
}, |
|
{ |
|
"epoch": 2.3248882265275705, |
|
"grad_norm": 13.66861629486084, |
|
"learning_rate": 2.5318484177091667e-06, |
|
"log_odds_chosen": 2.2526602745056152, |
|
"log_odds_ratio": -0.12812598049640656, |
|
"logits/chosen": 367.953857421875, |
|
"logits/rejected": 390.6175537109375, |
|
"logps/chosen": -0.3860087990760803, |
|
"logps/rejected": -1.594990611076355, |
|
"loss": 0.5879, |
|
"nll_loss": 0.6172261834144592, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.019300442188978195, |
|
"rewards/margins": 0.060449086129665375, |
|
"rewards/rejected": -0.07974953204393387, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 2.354694485842027, |
|
"grad_norm": 9.35374641418457, |
|
"learning_rate": 2.515773027133138e-06, |
|
"log_odds_chosen": 2.5175037384033203, |
|
"log_odds_ratio": -0.13300301134586334, |
|
"logits/chosen": 368.17181396484375, |
|
"logits/rejected": 361.74591064453125, |
|
"logps/chosen": -0.266899436712265, |
|
"logps/rejected": -1.3636187314987183, |
|
"loss": 0.5233, |
|
"nll_loss": 0.47403964400291443, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.01334497332572937, |
|
"rewards/margins": 0.0548359639942646, |
|
"rewards/rejected": -0.06818093359470367, |
|
"step": 395 |
|
}, |
|
{ |
|
"epoch": 2.384500745156483, |
|
"grad_norm": 17.207456588745117, |
|
"learning_rate": 2.5e-06, |
|
"log_odds_chosen": 2.6303577423095703, |
|
"log_odds_ratio": -0.12364117056131363, |
|
"logits/chosen": 359.56134033203125, |
|
"logits/rejected": 384.45733642578125, |
|
"logps/chosen": -0.3128165900707245, |
|
"logps/rejected": -1.7372194528579712, |
|
"loss": 0.5523, |
|
"nll_loss": 0.4810587465763092, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.015640828758478165, |
|
"rewards/margins": 0.07122014462947845, |
|
"rewards/rejected": -0.08686096966266632, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 2.4143070044709387, |
|
"grad_norm": 13.166074752807617, |
|
"learning_rate": 2.484519974999767e-06, |
|
"log_odds_chosen": 2.396026849746704, |
|
"log_odds_ratio": -0.1698196679353714, |
|
"logits/chosen": 415.81658935546875, |
|
"logits/rejected": 383.0643310546875, |
|
"logps/chosen": -0.3519137501716614, |
|
"logps/rejected": -1.5836241245269775, |
|
"loss": 0.5829, |
|
"nll_loss": 0.5322110652923584, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.0175956878811121, |
|
"rewards/margins": 0.061585523188114166, |
|
"rewards/rejected": -0.07918120920658112, |
|
"step": 405 |
|
}, |
|
{ |
|
"epoch": 2.444113263785395, |
|
"grad_norm": 12.805488586425781, |
|
"learning_rate": 2.4693239916239746e-06, |
|
"log_odds_chosen": 2.3219261169433594, |
|
"log_odds_ratio": -0.1753077507019043, |
|
"logits/chosen": 361.06781005859375, |
|
"logits/rejected": 375.9926452636719, |
|
"logps/chosen": -0.39748460054397583, |
|
"logps/rejected": -1.5706228017807007, |
|
"loss": 0.5614, |
|
"nll_loss": 0.6001583337783813, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.019874228164553642, |
|
"rewards/margins": 0.05865691229701042, |
|
"rewards/rejected": -0.07853114604949951, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 2.473919523099851, |
|
"grad_norm": 10.840022087097168, |
|
"learning_rate": 2.4544034683690802e-06, |
|
"log_odds_chosen": 2.408691883087158, |
|
"log_odds_ratio": -0.15550675988197327, |
|
"logits/chosen": 363.71136474609375, |
|
"logits/rejected": 392.62689208984375, |
|
"logps/chosen": -0.33126166462898254, |
|
"logps/rejected": -1.5768396854400635, |
|
"loss": 0.565, |
|
"nll_loss": 0.5147169828414917, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.016563083976507187, |
|
"rewards/margins": 0.062278907746076584, |
|
"rewards/rejected": -0.07884199172258377, |
|
"step": 415 |
|
}, |
|
{ |
|
"epoch": 2.503725782414307, |
|
"grad_norm": 14.33762264251709, |
|
"learning_rate": 2.4397501823713327e-06, |
|
"log_odds_chosen": 2.0540168285369873, |
|
"log_odds_ratio": -0.1838049441576004, |
|
"logits/chosen": 360.244873046875, |
|
"logits/rejected": 338.4053649902344, |
|
"logps/chosen": -0.382302850484848, |
|
"logps/rejected": -1.4437496662139893, |
|
"loss": 0.5427, |
|
"nll_loss": 0.6479641199111938, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.01911514438688755, |
|
"rewards/margins": 0.05307234078645706, |
|
"rewards/rejected": -0.07218748331069946, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 2.533532041728763, |
|
"grad_norm": 11.913477897644043, |
|
"learning_rate": 2.4253562503633297e-06, |
|
"log_odds_chosen": 2.7403793334960938, |
|
"log_odds_ratio": -0.09055573493242264, |
|
"logits/chosen": 360.6409912109375, |
|
"logits/rejected": 357.6117858886719, |
|
"logps/chosen": -0.3066481351852417, |
|
"logps/rejected": -1.7102562189102173, |
|
"loss": 0.533, |
|
"nll_loss": 0.5301586389541626, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.015332406386733055, |
|
"rewards/margins": 0.07018040120601654, |
|
"rewards/rejected": -0.08551281690597534, |
|
"step": 425 |
|
}, |
|
{ |
|
"epoch": 2.563338301043219, |
|
"grad_norm": 12.434874534606934, |
|
"learning_rate": 2.411214110852061e-06, |
|
"log_odds_chosen": 2.650773525238037, |
|
"log_odds_ratio": -0.11397852003574371, |
|
"logits/chosen": 359.2686767578125, |
|
"logits/rejected": 372.0355224609375, |
|
"logps/chosen": -0.29058974981307983, |
|
"logps/rejected": -1.6085436344146729, |
|
"loss": 0.5369, |
|
"nll_loss": 0.48550575971603394, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.014529486186802387, |
|
"rewards/margins": 0.06589768826961517, |
|
"rewards/rejected": -0.08042718470096588, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 2.593144560357675, |
|
"grad_norm": 15.130928993225098, |
|
"learning_rate": 2.3973165074269213e-06, |
|
"log_odds_chosen": 2.4082794189453125, |
|
"log_odds_ratio": -0.14481699466705322, |
|
"logits/chosen": 368.16680908203125, |
|
"logits/rejected": 336.6341247558594, |
|
"logps/chosen": -0.37292951345443726, |
|
"logps/rejected": -1.5544594526290894, |
|
"loss": 0.5655, |
|
"nll_loss": 0.5464103817939758, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.018646476790308952, |
|
"rewards/margins": 0.059076495468616486, |
|
"rewards/rejected": -0.07772296667098999, |
|
"step": 435 |
|
}, |
|
{ |
|
"epoch": 2.6229508196721314, |
|
"grad_norm": 13.672365188598633, |
|
"learning_rate": 2.3836564731139807e-06, |
|
"log_odds_chosen": 2.712573289871216, |
|
"log_odds_ratio": -0.11169672012329102, |
|
"logits/chosen": 354.31842041015625, |
|
"logits/rejected": 366.07745361328125, |
|
"logps/chosen": -0.2754111886024475, |
|
"logps/rejected": -1.5493534803390503, |
|
"loss": 0.5762, |
|
"nll_loss": 0.5381069779396057, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.013770559802651405, |
|
"rewards/margins": 0.06369712203741074, |
|
"rewards/rejected": -0.07746767997741699, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 2.6527570789865873, |
|
"grad_norm": 12.229787826538086, |
|
"learning_rate": 2.3702273156998867e-06, |
|
"log_odds_chosen": 2.407763957977295, |
|
"log_odds_ratio": -0.13709630072116852, |
|
"logits/chosen": 339.0660705566406, |
|
"logits/rejected": 373.5878601074219, |
|
"logps/chosen": -0.3847058415412903, |
|
"logps/rejected": -1.715640664100647, |
|
"loss": 0.5811, |
|
"nll_loss": 0.6087844967842102, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.019235290586948395, |
|
"rewards/margins": 0.06654674559831619, |
|
"rewards/rejected": -0.08578203618526459, |
|
"step": 445 |
|
}, |
|
{ |
|
"epoch": 2.682563338301043, |
|
"grad_norm": 11.220720291137695, |
|
"learning_rate": 2.357022603955159e-06, |
|
"log_odds_chosen": 2.3630270957946777, |
|
"log_odds_ratio": -0.12073234468698502, |
|
"logits/chosen": 364.763427734375, |
|
"logits/rejected": 365.3312072753906, |
|
"logps/chosen": -0.3996626138687134, |
|
"logps/rejected": -1.661350965499878, |
|
"loss": 0.589, |
|
"nll_loss": 0.5471979379653931, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.01998312957584858, |
|
"rewards/margins": 0.06308441609144211, |
|
"rewards/rejected": -0.08306754380464554, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 2.712369597615499, |
|
"grad_norm": 14.05644702911377, |
|
"learning_rate": 2.3440361546924774e-06, |
|
"log_odds_chosen": 2.527470588684082, |
|
"log_odds_ratio": -0.128330796957016, |
|
"logits/chosen": 393.28240966796875, |
|
"logits/rejected": 373.51959228515625, |
|
"logps/chosen": -0.3907659351825714, |
|
"logps/rejected": -1.6387542486190796, |
|
"loss": 0.5997, |
|
"nll_loss": 0.5538510084152222, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.01953829452395439, |
|
"rewards/margins": 0.06239941716194153, |
|
"rewards/rejected": -0.08193771541118622, |
|
"step": 455 |
|
}, |
|
{ |
|
"epoch": 2.742175856929955, |
|
"grad_norm": 14.722325325012207, |
|
"learning_rate": 2.3312620206007847e-06, |
|
"log_odds_chosen": 2.5439341068267822, |
|
"log_odds_ratio": -0.1302451193332672, |
|
"logits/chosen": 381.04022216796875, |
|
"logits/rejected": 400.01812744140625, |
|
"logps/chosen": -0.36085230112075806, |
|
"logps/rejected": -1.748968482017517, |
|
"loss": 0.5786, |
|
"nll_loss": 0.6514776945114136, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.018042614683508873, |
|
"rewards/margins": 0.06940580904483795, |
|
"rewards/rejected": -0.08744843304157257, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 2.7719821162444114, |
|
"grad_norm": 12.233352661132812, |
|
"learning_rate": 2.3186944788008413e-06, |
|
"log_odds_chosen": 2.5529284477233887, |
|
"log_odds_ratio": -0.1319517195224762, |
|
"logits/chosen": 376.39300537109375, |
|
"logits/rejected": 381.1004333496094, |
|
"logps/chosen": -0.2920375168323517, |
|
"logps/rejected": -1.539701223373413, |
|
"loss": 0.581, |
|
"nll_loss": 0.5538360476493835, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.01460187416523695, |
|
"rewards/margins": 0.06238318607211113, |
|
"rewards/rejected": -0.07698506116867065, |
|
"step": 465 |
|
}, |
|
{ |
|
"epoch": 2.8017883755588673, |
|
"grad_norm": 12.4392728805542, |
|
"learning_rate": 2.3063280200722128e-06, |
|
"log_odds_chosen": 2.2491307258605957, |
|
"log_odds_ratio": -0.18098264932632446, |
|
"logits/chosen": 377.3009948730469, |
|
"logits/rejected": 349.0381164550781, |
|
"logps/chosen": -0.3590669333934784, |
|
"logps/rejected": -1.5034451484680176, |
|
"loss": 0.5715, |
|
"nll_loss": 0.5520719289779663, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.01795334741473198, |
|
"rewards/margins": 0.05721891671419144, |
|
"rewards/rejected": -0.07517226040363312, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 2.8315946348733236, |
|
"grad_norm": 13.7266263961792, |
|
"learning_rate": 2.2941573387056174e-06, |
|
"log_odds_chosen": 2.518183946609497, |
|
"log_odds_ratio": -0.1277451366186142, |
|
"logits/chosen": 350.8870544433594, |
|
"logits/rejected": 374.2940979003906, |
|
"logps/chosen": -0.36442482471466064, |
|
"logps/rejected": -1.6510089635849, |
|
"loss": 0.5518, |
|
"nll_loss": 0.5228261947631836, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.018221240490674973, |
|
"rewards/margins": 0.06432920694351196, |
|
"rewards/rejected": -0.08255044370889664, |
|
"step": 475 |
|
}, |
|
{ |
|
"epoch": 2.8614008941877795, |
|
"grad_norm": 10.832075119018555, |
|
"learning_rate": 2.2821773229381924e-06, |
|
"log_odds_chosen": 2.2293262481689453, |
|
"log_odds_ratio": -0.12778417766094208, |
|
"logits/chosen": 361.9057922363281, |
|
"logits/rejected": 402.4067077636719, |
|
"logps/chosen": -0.34709399938583374, |
|
"logps/rejected": -1.4698781967163086, |
|
"loss": 0.5087, |
|
"nll_loss": 0.5175037384033203, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.017354700714349747, |
|
"rewards/margins": 0.05613920837640762, |
|
"rewards/rejected": -0.07349390536546707, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 2.8912071535022354, |
|
"grad_norm": 11.188936233520508, |
|
"learning_rate": 2.270383045932499e-06, |
|
"log_odds_chosen": 2.5824456214904785, |
|
"log_odds_ratio": -0.12692959606647491, |
|
"logits/chosen": 357.77374267578125, |
|
"logits/rejected": 380.9786376953125, |
|
"logps/chosen": -0.3558903634548187, |
|
"logps/rejected": -1.7740790843963623, |
|
"loss": 0.5513, |
|
"nll_loss": 0.5267875790596008, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.017794519662857056, |
|
"rewards/margins": 0.07090945541858673, |
|
"rewards/rejected": -0.08870397508144379, |
|
"step": 485 |
|
}, |
|
{ |
|
"epoch": 2.9210134128166914, |
|
"grad_norm": 12.93087100982666, |
|
"learning_rate": 2.2587697572631284e-06, |
|
"log_odds_chosen": 2.304809331893921, |
|
"log_odds_ratio": -0.18847742676734924, |
|
"logits/chosen": 370.79241943359375, |
|
"logits/rejected": 337.0198059082031, |
|
"logps/chosen": -0.46047478914260864, |
|
"logps/rejected": -1.6731350421905518, |
|
"loss": 0.6251, |
|
"nll_loss": 0.5758861303329468, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -0.023023737594485283, |
|
"rewards/margins": 0.060633014887571335, |
|
"rewards/rejected": -0.08365674316883087, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 2.9508196721311473, |
|
"grad_norm": 12.990720748901367, |
|
"learning_rate": 2.2473328748774737e-06, |
|
"log_odds_chosen": 2.1052613258361816, |
|
"log_odds_ratio": -0.17354901134967804, |
|
"logits/chosen": 362.78375244140625, |
|
"logits/rejected": 391.7388610839844, |
|
"logps/chosen": -0.4515438973903656, |
|
"logps/rejected": -1.5366592407226562, |
|
"loss": 0.5494, |
|
"nll_loss": 0.5879518985748291, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -0.0225771926343441, |
|
"rewards/margins": 0.05425576493144035, |
|
"rewards/rejected": -0.07683296501636505, |
|
"step": 495 |
|
}, |
|
{ |
|
"epoch": 2.9806259314456036, |
|
"grad_norm": 11.811108589172363, |
|
"learning_rate": 2.23606797749979e-06, |
|
"log_odds_chosen": 2.437211513519287, |
|
"log_odds_ratio": -0.13766226172447205, |
|
"logits/chosen": 372.21160888671875, |
|
"logits/rejected": 370.1877136230469, |
|
"logps/chosen": -0.32970958948135376, |
|
"logps/rejected": -1.5284183025360107, |
|
"loss": 0.576, |
|
"nll_loss": 0.47357138991355896, |
|
"rewards/accuracies": 1.0, |
|
"rewards/chosen": -0.01648547872900963, |
|
"rewards/margins": 0.05993543937802315, |
|
"rewards/rejected": -0.07642091810703278, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 2.9865871833084947, |
|
"eval_log_odds_chosen": 0.27211081981658936, |
|
"eval_log_odds_ratio": -0.7039158344268799, |
|
"eval_logits/chosen": 296.6807556152344, |
|
"eval_logits/rejected": 267.234619140625, |
|
"eval_logps/chosen": -1.0933129787445068, |
|
"eval_logps/rejected": -1.2508149147033691, |
|
"eval_loss": 1.5379581451416016, |
|
"eval_nll_loss": 1.4702765941619873, |
|
"eval_rewards/accuracies": 0.5467625856399536, |
|
"eval_rewards/chosen": -0.054665643721818924, |
|
"eval_rewards/margins": 0.007875093258917332, |
|
"eval_rewards/rejected": -0.06254073977470398, |
|
"eval_runtime": 112.2732, |
|
"eval_samples_per_second": 4.925, |
|
"eval_steps_per_second": 1.238, |
|
"step": 501 |
|
}, |
|
{ |
|
"epoch": 2.9865871833084947, |
|
"step": 501, |
|
"total_flos": 0.0, |
|
"train_loss": 1.4583633707431025, |
|
"train_runtime": 13840.413, |
|
"train_samples_per_second": 1.163, |
|
"train_steps_per_second": 0.036 |
|
} |
|
], |
|
"logging_steps": 5, |
|
"max_steps": 501, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 3, |
|
"save_steps": 500, |
|
"stateful_callbacks": { |
|
"TrainerControl": { |
|
"args": { |
|
"should_epoch_stop": false, |
|
"should_evaluate": false, |
|
"should_log": false, |
|
"should_save": false, |
|
"should_training_stop": false |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 0.0, |
|
"train_batch_size": 2, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|