|
{ |
|
"best_metric": null, |
|
"best_model_checkpoint": null, |
|
"epoch": 2.998953427524856, |
|
"eval_steps": 100, |
|
"global_step": 2865, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.010465724751439037, |
|
"grad_norm": 2.345809627249496, |
|
"learning_rate": 1.7421602787456445e-06, |
|
"logits/chosen": -10.875, |
|
"logits/rejected": -10.9375, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -280.0, |
|
"loss": 2.4807, |
|
"nll_loss": 1.09375, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": -0.357421875, |
|
"rewards/rejected": -2.796875, |
|
"step": 10 |
|
}, |
|
{ |
|
"epoch": 0.020931449502878074, |
|
"grad_norm": 2.33996074903795, |
|
"learning_rate": 3.484320557491289e-06, |
|
"logits/chosen": -10.4375, |
|
"logits/rejected": -10.3125, |
|
"logps/chosen": -326.0, |
|
"logps/rejected": -312.0, |
|
"loss": 2.4806, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -3.265625, |
|
"rewards/margins": -0.154296875, |
|
"rewards/rejected": -3.109375, |
|
"step": 20 |
|
}, |
|
{ |
|
"epoch": 0.03139717425431711, |
|
"grad_norm": 2.5586077878773463, |
|
"learning_rate": 5.226480836236934e-06, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -10.3125, |
|
"logps/chosen": -314.0, |
|
"logps/rejected": -308.0, |
|
"loss": 2.5373, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -3.140625, |
|
"rewards/margins": -0.06884765625, |
|
"rewards/rejected": -3.078125, |
|
"step": 30 |
|
}, |
|
{ |
|
"epoch": 0.04186289900575615, |
|
"grad_norm": 2.1870419268481873, |
|
"learning_rate": 6.968641114982578e-06, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -10.5625, |
|
"logps/chosen": -334.0, |
|
"logps/rejected": -306.0, |
|
"loss": 2.4262, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.4124999940395355, |
|
"rewards/chosen": -3.328125, |
|
"rewards/margins": -0.26953125, |
|
"rewards/rejected": -3.0625, |
|
"step": 40 |
|
}, |
|
{ |
|
"epoch": 0.052328623757195186, |
|
"grad_norm": 2.3594832957060574, |
|
"learning_rate": 8.710801393728225e-06, |
|
"logits/chosen": -11.3125, |
|
"logits/rejected": -10.875, |
|
"logps/chosen": -310.0, |
|
"logps/rejected": -264.0, |
|
"loss": 2.3308, |
|
"nll_loss": 1.0234375, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -3.09375, |
|
"rewards/margins": -0.466796875, |
|
"rewards/rejected": -2.640625, |
|
"step": 50 |
|
}, |
|
{ |
|
"epoch": 0.06279434850863422, |
|
"grad_norm": 2.398245639368396, |
|
"learning_rate": 1.0452961672473868e-05, |
|
"logits/chosen": -11.375, |
|
"logits/rejected": -11.125, |
|
"logps/chosen": -348.0, |
|
"logps/rejected": -306.0, |
|
"loss": 2.3784, |
|
"nll_loss": 1.0625, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -3.46875, |
|
"rewards/margins": -0.404296875, |
|
"rewards/rejected": -3.0625, |
|
"step": 60 |
|
}, |
|
{ |
|
"epoch": 0.07326007326007326, |
|
"grad_norm": 2.2386085594671905, |
|
"learning_rate": 1.2195121951219513e-05, |
|
"logits/chosen": -12.125, |
|
"logits/rejected": -11.625, |
|
"logps/chosen": -296.0, |
|
"logps/rejected": -284.0, |
|
"loss": 2.429, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -2.96875, |
|
"rewards/margins": -0.115234375, |
|
"rewards/rejected": -2.84375, |
|
"step": 70 |
|
}, |
|
{ |
|
"epoch": 0.0837257980115123, |
|
"grad_norm": 2.503247672617112, |
|
"learning_rate": 1.3937282229965156e-05, |
|
"logits/chosen": -12.5625, |
|
"logits/rejected": -12.8125, |
|
"logps/chosen": -324.0, |
|
"logps/rejected": -286.0, |
|
"loss": 2.3775, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.38749998807907104, |
|
"rewards/chosen": -3.234375, |
|
"rewards/margins": -0.376953125, |
|
"rewards/rejected": -2.859375, |
|
"step": 80 |
|
}, |
|
{ |
|
"epoch": 0.09419152276295134, |
|
"grad_norm": 2.5299592835010167, |
|
"learning_rate": 1.56794425087108e-05, |
|
"logits/chosen": -12.9375, |
|
"logits/rejected": -12.875, |
|
"logps/chosen": -284.0, |
|
"logps/rejected": -294.0, |
|
"loss": 2.3183, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -2.84375, |
|
"rewards/margins": 0.10986328125, |
|
"rewards/rejected": -2.953125, |
|
"step": 90 |
|
}, |
|
{ |
|
"epoch": 0.10465724751439037, |
|
"grad_norm": 4.514036980881572, |
|
"learning_rate": 1.742160278745645e-05, |
|
"logits/chosen": -12.375, |
|
"logits/rejected": -12.4375, |
|
"logps/chosen": -352.0, |
|
"logps/rejected": -322.0, |
|
"loss": 2.4271, |
|
"nll_loss": 1.1015625, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -3.515625, |
|
"rewards/margins": -0.296875, |
|
"rewards/rejected": -3.234375, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.10465724751439037, |
|
"eval_logits/chosen": -12.3125, |
|
"eval_logits/rejected": -12.125, |
|
"eval_logps/chosen": -336.0, |
|
"eval_logps/rejected": -328.0, |
|
"eval_loss": 2.295875072479248, |
|
"eval_nll_loss": 1.0859375, |
|
"eval_rewards/accuracies": 0.498046875, |
|
"eval_rewards/chosen": -3.359375, |
|
"eval_rewards/margins": -0.0849609375, |
|
"eval_rewards/rejected": -3.28125, |
|
"eval_runtime": 24.1761, |
|
"eval_samples_per_second": 82.726, |
|
"eval_steps_per_second": 1.324, |
|
"step": 100 |
|
}, |
|
{ |
|
"epoch": 0.1151229722658294, |
|
"grad_norm": 3.9519009427617418, |
|
"learning_rate": 1.9163763066202093e-05, |
|
"logits/chosen": -11.375, |
|
"logits/rejected": -11.125, |
|
"logps/chosen": -294.0, |
|
"logps/rejected": -290.0, |
|
"loss": 2.2551, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -2.9375, |
|
"rewards/margins": -0.03466796875, |
|
"rewards/rejected": -2.90625, |
|
"step": 110 |
|
}, |
|
{ |
|
"epoch": 0.12558869701726844, |
|
"grad_norm": 3.4491049058364442, |
|
"learning_rate": 2.0905923344947736e-05, |
|
"logits/chosen": -11.1875, |
|
"logits/rejected": -10.75, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -338.0, |
|
"loss": 2.2556, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -3.125, |
|
"rewards/margins": 0.25390625, |
|
"rewards/rejected": -3.375, |
|
"step": 120 |
|
}, |
|
{ |
|
"epoch": 0.1360544217687075, |
|
"grad_norm": 5.405390761462798, |
|
"learning_rate": 2.264808362369338e-05, |
|
"logits/chosen": -11.625, |
|
"logits/rejected": -11.3125, |
|
"logps/chosen": -344.0, |
|
"logps/rejected": -324.0, |
|
"loss": 2.4151, |
|
"nll_loss": 1.1015625, |
|
"rewards/accuracies": 0.4375, |
|
"rewards/chosen": -3.4375, |
|
"rewards/margins": -0.2060546875, |
|
"rewards/rejected": -3.234375, |
|
"step": 130 |
|
}, |
|
{ |
|
"epoch": 0.14652014652014653, |
|
"grad_norm": 3.600801996012317, |
|
"learning_rate": 2.4390243902439026e-05, |
|
"logits/chosen": -10.875, |
|
"logits/rejected": -10.875, |
|
"logps/chosen": -340.0, |
|
"logps/rejected": -336.0, |
|
"loss": 2.21, |
|
"nll_loss": 1.0625, |
|
"rewards/accuracies": 0.4312500059604645, |
|
"rewards/chosen": -3.390625, |
|
"rewards/margins": -0.0302734375, |
|
"rewards/rejected": -3.375, |
|
"step": 140 |
|
}, |
|
{ |
|
"epoch": 0.15698587127158556, |
|
"grad_norm": 3.959574308432542, |
|
"learning_rate": 2.6132404181184672e-05, |
|
"logits/chosen": -10.75, |
|
"logits/rejected": -10.25, |
|
"logps/chosen": -350.0, |
|
"logps/rejected": -340.0, |
|
"loss": 2.2238, |
|
"nll_loss": 1.0703125, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.515625, |
|
"rewards/margins": -0.1064453125, |
|
"rewards/rejected": -3.40625, |
|
"step": 150 |
|
}, |
|
{ |
|
"epoch": 0.1674515960230246, |
|
"grad_norm": 4.309983424283929, |
|
"learning_rate": 2.7874564459930312e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -8.8125, |
|
"logps/chosen": -358.0, |
|
"logps/rejected": -336.0, |
|
"loss": 2.2664, |
|
"nll_loss": 1.125, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -3.578125, |
|
"rewards/margins": -0.2119140625, |
|
"rewards/rejected": -3.375, |
|
"step": 160 |
|
}, |
|
{ |
|
"epoch": 0.17791732077446362, |
|
"grad_norm": 4.646910490725447, |
|
"learning_rate": 2.9616724738675962e-05, |
|
"logits/chosen": -9.0, |
|
"logits/rejected": -8.375, |
|
"logps/chosen": -344.0, |
|
"logps/rejected": -346.0, |
|
"loss": 2.3174, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.4375, |
|
"rewards/margins": 0.015869140625, |
|
"rewards/rejected": -3.453125, |
|
"step": 170 |
|
}, |
|
{ |
|
"epoch": 0.18838304552590268, |
|
"grad_norm": 4.239679303072255, |
|
"learning_rate": 3.13588850174216e-05, |
|
"logits/chosen": -9.9375, |
|
"logits/rejected": -9.5625, |
|
"logps/chosen": -294.0, |
|
"logps/rejected": -310.0, |
|
"loss": 2.1422, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -2.9375, |
|
"rewards/margins": 0.162109375, |
|
"rewards/rejected": -3.09375, |
|
"step": 180 |
|
}, |
|
{ |
|
"epoch": 0.1988487702773417, |
|
"grad_norm": 5.853924794389039, |
|
"learning_rate": 3.310104529616725e-05, |
|
"logits/chosen": -11.5625, |
|
"logits/rejected": -11.3125, |
|
"logps/chosen": -296.0, |
|
"logps/rejected": -324.0, |
|
"loss": 2.2346, |
|
"nll_loss": 1.1875, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -2.96875, |
|
"rewards/margins": 0.265625, |
|
"rewards/rejected": -3.234375, |
|
"step": 190 |
|
}, |
|
{ |
|
"epoch": 0.20931449502878074, |
|
"grad_norm": 4.678659236199607, |
|
"learning_rate": 3.48432055749129e-05, |
|
"logits/chosen": -11.9375, |
|
"logits/rejected": -11.5, |
|
"logps/chosen": -392.0, |
|
"logps/rejected": -370.0, |
|
"loss": 2.2538, |
|
"nll_loss": 1.2578125, |
|
"rewards/accuracies": 0.4625000059604645, |
|
"rewards/chosen": -3.90625, |
|
"rewards/margins": -0.205078125, |
|
"rewards/rejected": -3.703125, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.20931449502878074, |
|
"eval_logits/chosen": -9.875, |
|
"eval_logits/rejected": -9.5, |
|
"eval_logps/chosen": -338.0, |
|
"eval_logps/rejected": -346.0, |
|
"eval_loss": 2.1835625171661377, |
|
"eval_nll_loss": 1.09375, |
|
"eval_rewards/accuracies": 0.5234375, |
|
"eval_rewards/chosen": -3.390625, |
|
"eval_rewards/margins": 0.06396484375, |
|
"eval_rewards/rejected": -3.453125, |
|
"eval_runtime": 23.8526, |
|
"eval_samples_per_second": 83.848, |
|
"eval_steps_per_second": 1.342, |
|
"step": 200 |
|
}, |
|
{ |
|
"epoch": 0.21978021978021978, |
|
"grad_norm": 2.8882833255655833, |
|
"learning_rate": 3.6585365853658535e-05, |
|
"logits/chosen": -10.6875, |
|
"logits/rejected": -10.25, |
|
"logps/chosen": -330.0, |
|
"logps/rejected": -332.0, |
|
"loss": 2.1255, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.296875, |
|
"rewards/margins": 0.00762939453125, |
|
"rewards/rejected": -3.3125, |
|
"step": 210 |
|
}, |
|
{ |
|
"epoch": 0.2302459445316588, |
|
"grad_norm": 2.541816587778732, |
|
"learning_rate": 3.8327526132404185e-05, |
|
"logits/chosen": -11.5625, |
|
"logits/rejected": -11.3125, |
|
"logps/chosen": -364.0, |
|
"logps/rejected": -326.0, |
|
"loss": 2.2885, |
|
"nll_loss": 1.2109375, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.640625, |
|
"rewards/margins": -0.375, |
|
"rewards/rejected": -3.265625, |
|
"step": 220 |
|
}, |
|
{ |
|
"epoch": 0.24071166928309787, |
|
"grad_norm": 3.343449845880885, |
|
"learning_rate": 4.006968641114983e-05, |
|
"logits/chosen": -10.8125, |
|
"logits/rejected": -10.75, |
|
"logps/chosen": -342.0, |
|
"logps/rejected": -332.0, |
|
"loss": 2.2333, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.48124998807907104, |
|
"rewards/chosen": -3.40625, |
|
"rewards/margins": -0.08056640625, |
|
"rewards/rejected": -3.328125, |
|
"step": 230 |
|
}, |
|
{ |
|
"epoch": 0.25117739403453687, |
|
"grad_norm": 3.768701596733476, |
|
"learning_rate": 4.181184668989547e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -10.3125, |
|
"logps/chosen": -338.0, |
|
"logps/rejected": -354.0, |
|
"loss": 2.2571, |
|
"nll_loss": 1.1328125, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.375, |
|
"rewards/margins": 0.1748046875, |
|
"rewards/rejected": -3.546875, |
|
"step": 240 |
|
}, |
|
{ |
|
"epoch": 0.2616431187859759, |
|
"grad_norm": 3.4059141209893045, |
|
"learning_rate": 4.3554006968641115e-05, |
|
"logits/chosen": -10.4375, |
|
"logits/rejected": -9.8125, |
|
"logps/chosen": -348.0, |
|
"logps/rejected": -344.0, |
|
"loss": 2.2502, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.484375, |
|
"rewards/margins": -0.04736328125, |
|
"rewards/rejected": -3.4375, |
|
"step": 250 |
|
}, |
|
{ |
|
"epoch": 0.272108843537415, |
|
"grad_norm": 3.442668595880817, |
|
"learning_rate": 4.529616724738676e-05, |
|
"logits/chosen": -11.9375, |
|
"logits/rejected": -11.625, |
|
"logps/chosen": -360.0, |
|
"logps/rejected": -376.0, |
|
"loss": 2.1595, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.609375, |
|
"rewards/margins": 0.158203125, |
|
"rewards/rejected": -3.765625, |
|
"step": 260 |
|
}, |
|
{ |
|
"epoch": 0.282574568288854, |
|
"grad_norm": 3.066923415443492, |
|
"learning_rate": 4.703832752613241e-05, |
|
"logits/chosen": -11.8125, |
|
"logits/rejected": -11.375, |
|
"logps/chosen": -378.0, |
|
"logps/rejected": -358.0, |
|
"loss": 2.2675, |
|
"nll_loss": 1.1953125, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -3.796875, |
|
"rewards/margins": -0.2041015625, |
|
"rewards/rejected": -3.59375, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 0.29304029304029305, |
|
"grad_norm": 3.8052553189726175, |
|
"learning_rate": 4.878048780487805e-05, |
|
"logits/chosen": -12.375, |
|
"logits/rejected": -11.625, |
|
"logps/chosen": -360.0, |
|
"logps/rejected": -366.0, |
|
"loss": 2.204, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.59375, |
|
"rewards/margins": 0.053466796875, |
|
"rewards/rejected": -3.65625, |
|
"step": 280 |
|
}, |
|
{ |
|
"epoch": 0.3035060177917321, |
|
"grad_norm": 3.0428444424788856, |
|
"learning_rate": 4.999983293481418e-05, |
|
"logits/chosen": -12.4375, |
|
"logits/rejected": -12.0625, |
|
"logps/chosen": -342.0, |
|
"logps/rejected": -320.0, |
|
"loss": 2.268, |
|
"nll_loss": 1.15625, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -3.40625, |
|
"rewards/margins": -0.224609375, |
|
"rewards/rejected": -3.1875, |
|
"step": 290 |
|
}, |
|
{ |
|
"epoch": 0.3139717425431711, |
|
"grad_norm": 2.377598539376315, |
|
"learning_rate": 4.9996862949181595e-05, |
|
"logits/chosen": -12.0, |
|
"logits/rejected": -11.5625, |
|
"logps/chosen": -360.0, |
|
"logps/rejected": -380.0, |
|
"loss": 2.1253, |
|
"nll_loss": 1.1875, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.609375, |
|
"rewards/margins": 0.193359375, |
|
"rewards/rejected": -3.796875, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.3139717425431711, |
|
"eval_logits/chosen": -11.4375, |
|
"eval_logits/rejected": -11.0, |
|
"eval_logps/chosen": -346.0, |
|
"eval_logps/rejected": -360.0, |
|
"eval_loss": 2.1307499408721924, |
|
"eval_nll_loss": 1.1171875, |
|
"eval_rewards/accuracies": 0.517578125, |
|
"eval_rewards/chosen": -3.453125, |
|
"eval_rewards/margins": 0.1416015625, |
|
"eval_rewards/rejected": -3.59375, |
|
"eval_runtime": 23.8499, |
|
"eval_samples_per_second": 83.858, |
|
"eval_steps_per_second": 1.342, |
|
"step": 300 |
|
}, |
|
{ |
|
"epoch": 0.32443746729461015, |
|
"grad_norm": 4.525426874241728, |
|
"learning_rate": 4.9990180911526896e-05, |
|
"logits/chosen": -11.5, |
|
"logits/rejected": -10.25, |
|
"logps/chosen": -370.0, |
|
"logps/rejected": -356.0, |
|
"loss": 2.205, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -3.703125, |
|
"rewards/margins": -0.130859375, |
|
"rewards/rejected": -3.5625, |
|
"step": 310 |
|
}, |
|
{ |
|
"epoch": 0.3349031920460492, |
|
"grad_norm": 3.3217851170565216, |
|
"learning_rate": 4.997978781413745e-05, |
|
"logits/chosen": -11.25, |
|
"logits/rejected": -10.6875, |
|
"logps/chosen": -326.0, |
|
"logps/rejected": -352.0, |
|
"loss": 2.2002, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.25, |
|
"rewards/margins": 0.267578125, |
|
"rewards/rejected": -3.515625, |
|
"step": 320 |
|
}, |
|
{ |
|
"epoch": 0.3453689167974882, |
|
"grad_norm": 6.405598112862614, |
|
"learning_rate": 4.996568520039565e-05, |
|
"logits/chosen": -12.25, |
|
"logits/rejected": -11.8125, |
|
"logps/chosen": -332.0, |
|
"logps/rejected": -350.0, |
|
"loss": 2.0934, |
|
"nll_loss": 1.1640625, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.3125, |
|
"rewards/margins": 0.1923828125, |
|
"rewards/rejected": -3.5, |
|
"step": 330 |
|
}, |
|
{ |
|
"epoch": 0.35583464154892724, |
|
"grad_norm": 3.4548294318684802, |
|
"learning_rate": 4.994787516454973e-05, |
|
"logits/chosen": -11.5625, |
|
"logits/rejected": -11.25, |
|
"logps/chosen": -334.0, |
|
"logps/rejected": -320.0, |
|
"loss": 2.1729, |
|
"nll_loss": 1.1328125, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -3.34375, |
|
"rewards/margins": -0.138671875, |
|
"rewards/rejected": -3.21875, |
|
"step": 340 |
|
}, |
|
{ |
|
"epoch": 0.3663003663003663, |
|
"grad_norm": 4.27610128446516, |
|
"learning_rate": 4.992636035140273e-05, |
|
"logits/chosen": -11.1875, |
|
"logits/rejected": -10.9375, |
|
"logps/chosen": -368.0, |
|
"logps/rejected": -348.0, |
|
"loss": 2.1775, |
|
"nll_loss": 1.2109375, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -3.6875, |
|
"rewards/margins": -0.208984375, |
|
"rewards/rejected": -3.484375, |
|
"step": 350 |
|
}, |
|
{ |
|
"epoch": 0.37676609105180536, |
|
"grad_norm": 2.8983585815649033, |
|
"learning_rate": 4.99011439559198e-05, |
|
"logits/chosen": -11.0625, |
|
"logits/rejected": -10.6875, |
|
"logps/chosen": -388.0, |
|
"logps/rejected": -398.0, |
|
"loss": 2.262, |
|
"nll_loss": 1.34375, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.875, |
|
"rewards/margins": 0.1103515625, |
|
"rewards/rejected": -3.984375, |
|
"step": 360 |
|
}, |
|
{ |
|
"epoch": 0.3872318158032444, |
|
"grad_norm": 2.8025250549217504, |
|
"learning_rate": 4.987222972275369e-05, |
|
"logits/chosen": -11.1875, |
|
"logits/rejected": -10.75, |
|
"logps/chosen": -328.0, |
|
"logps/rejected": -312.0, |
|
"loss": 2.1842, |
|
"nll_loss": 1.1875, |
|
"rewards/accuracies": 0.42500001192092896, |
|
"rewards/chosen": -3.265625, |
|
"rewards/margins": -0.1552734375, |
|
"rewards/rejected": -3.109375, |
|
"step": 370 |
|
}, |
|
{ |
|
"epoch": 0.3976975405546834, |
|
"grad_norm": 4.697054630656616, |
|
"learning_rate": 4.983962194568872e-05, |
|
"logits/chosen": -12.0625, |
|
"logits/rejected": -11.25, |
|
"logps/chosen": -330.0, |
|
"logps/rejected": -370.0, |
|
"loss": 2.1627, |
|
"nll_loss": 1.2109375, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -3.296875, |
|
"rewards/margins": 0.408203125, |
|
"rewards/rejected": -3.703125, |
|
"step": 380 |
|
}, |
|
{ |
|
"epoch": 0.40816326530612246, |
|
"grad_norm": 4.233426621207795, |
|
"learning_rate": 4.9803325467003084e-05, |
|
"logits/chosen": -11.4375, |
|
"logits/rejected": -10.8125, |
|
"logps/chosen": -368.0, |
|
"logps/rejected": -384.0, |
|
"loss": 2.2654, |
|
"nll_loss": 1.25, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -3.6875, |
|
"rewards/margins": 0.142578125, |
|
"rewards/rejected": -3.828125, |
|
"step": 390 |
|
}, |
|
{ |
|
"epoch": 0.4186289900575615, |
|
"grad_norm": 3.8273983972176606, |
|
"learning_rate": 4.9763345676749816e-05, |
|
"logits/chosen": -11.5625, |
|
"logits/rejected": -11.0625, |
|
"logps/chosen": -334.0, |
|
"logps/rejected": -344.0, |
|
"loss": 2.0609, |
|
"nll_loss": 1.0859375, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.34375, |
|
"rewards/margins": 0.09423828125, |
|
"rewards/rejected": -3.4375, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.4186289900575615, |
|
"eval_logits/chosen": -11.125, |
|
"eval_logits/rejected": -10.625, |
|
"eval_logps/chosen": -332.0, |
|
"eval_logps/rejected": -344.0, |
|
"eval_loss": 2.135937452316284, |
|
"eval_nll_loss": 1.0703125, |
|
"eval_rewards/accuracies": 0.529296875, |
|
"eval_rewards/chosen": -3.328125, |
|
"eval_rewards/margins": 0.11865234375, |
|
"eval_rewards/rejected": -3.4375, |
|
"eval_runtime": 23.8479, |
|
"eval_samples_per_second": 83.865, |
|
"eval_steps_per_second": 1.342, |
|
"step": 400 |
|
}, |
|
{ |
|
"epoch": 0.4290947148090005, |
|
"grad_norm": 4.130139321913475, |
|
"learning_rate": 4.9719688511956374e-05, |
|
"logits/chosen": -10.5625, |
|
"logits/rejected": -9.75, |
|
"logps/chosen": -376.0, |
|
"logps/rejected": -364.0, |
|
"loss": 2.2101, |
|
"nll_loss": 1.328125, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.765625, |
|
"rewards/margins": -0.1376953125, |
|
"rewards/rejected": -3.625, |
|
"step": 410 |
|
}, |
|
{ |
|
"epoch": 0.43956043956043955, |
|
"grad_norm": 3.231291815289811, |
|
"learning_rate": 4.967236045574294e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -9.9375, |
|
"logps/chosen": -350.0, |
|
"logps/rejected": -336.0, |
|
"loss": 2.2187, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.44999998807907104, |
|
"rewards/chosen": -3.484375, |
|
"rewards/margins": -0.130859375, |
|
"rewards/rejected": -3.359375, |
|
"step": 420 |
|
}, |
|
{ |
|
"epoch": 0.4500261643118786, |
|
"grad_norm": 3.2705352508481993, |
|
"learning_rate": 4.96213685363597e-05, |
|
"logits/chosen": -10.9375, |
|
"logits/rejected": -10.375, |
|
"logps/chosen": -340.0, |
|
"logps/rejected": -358.0, |
|
"loss": 2.0928, |
|
"nll_loss": 1.125, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -3.40625, |
|
"rewards/margins": 0.171875, |
|
"rewards/rejected": -3.578125, |
|
"step": 430 |
|
}, |
|
{ |
|
"epoch": 0.4604918890633176, |
|
"grad_norm": 2.9120708507124515, |
|
"learning_rate": 4.956672032614315e-05, |
|
"logits/chosen": -11.4375, |
|
"logits/rejected": -11.1875, |
|
"logps/chosen": -346.0, |
|
"logps/rejected": -304.0, |
|
"loss": 2.2411, |
|
"nll_loss": 1.15625, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -3.46875, |
|
"rewards/margins": -0.4375, |
|
"rewards/rejected": -3.03125, |
|
"step": 440 |
|
}, |
|
{ |
|
"epoch": 0.47095761381475665, |
|
"grad_norm": 3.0640753054040557, |
|
"learning_rate": 4.950842394039157e-05, |
|
"logits/chosen": -11.9375, |
|
"logits/rejected": -11.625, |
|
"logps/chosen": -352.0, |
|
"logps/rejected": -360.0, |
|
"loss": 2.1267, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.4749999940395355, |
|
"rewards/chosen": -3.515625, |
|
"rewards/margins": 0.09130859375, |
|
"rewards/rejected": -3.609375, |
|
"step": 450 |
|
}, |
|
{ |
|
"epoch": 0.48142333856619574, |
|
"grad_norm": 2.995474462283963, |
|
"learning_rate": 4.944648803615994e-05, |
|
"logits/chosen": -11.75, |
|
"logits/rejected": -11.3125, |
|
"logps/chosen": -346.0, |
|
"logps/rejected": -362.0, |
|
"loss": 2.1641, |
|
"nll_loss": 1.1796875, |
|
"rewards/accuracies": 0.48750001192092896, |
|
"rewards/chosen": -3.453125, |
|
"rewards/margins": 0.173828125, |
|
"rewards/rejected": -3.625, |
|
"step": 460 |
|
}, |
|
{ |
|
"epoch": 0.49188906331763477, |
|
"grad_norm": 4.272742059283488, |
|
"learning_rate": 4.938092181097433e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -10.1875, |
|
"logps/chosen": -368.0, |
|
"logps/rejected": -368.0, |
|
"loss": 2.2025, |
|
"nll_loss": 1.25, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -3.6875, |
|
"rewards/margins": -0.00048828125, |
|
"rewards/rejected": -3.6875, |
|
"step": 470 |
|
}, |
|
{ |
|
"epoch": 0.5023547880690737, |
|
"grad_norm": 3.182017142600404, |
|
"learning_rate": 4.931173500146607e-05, |
|
"logits/chosen": -10.0625, |
|
"logits/rejected": -9.625, |
|
"logps/chosen": -354.0, |
|
"logps/rejected": -386.0, |
|
"loss": 2.2132, |
|
"nll_loss": 1.203125, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -3.53125, |
|
"rewards/margins": 0.328125, |
|
"rewards/rejected": -3.859375, |
|
"step": 480 |
|
}, |
|
{ |
|
"epoch": 0.5128205128205128, |
|
"grad_norm": 2.6476233218342573, |
|
"learning_rate": 4.923893788192582e-05, |
|
"logits/chosen": -10.25, |
|
"logits/rejected": -9.4375, |
|
"logps/chosen": -382.0, |
|
"logps/rejected": -386.0, |
|
"loss": 2.1898, |
|
"nll_loss": 1.15625, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.8125, |
|
"rewards/margins": 0.0517578125, |
|
"rewards/rejected": -3.859375, |
|
"step": 490 |
|
}, |
|
{ |
|
"epoch": 0.5232862375719518, |
|
"grad_norm": 3.9831796655685925, |
|
"learning_rate": 4.9162541262777906e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -8.875, |
|
"logps/chosen": -324.0, |
|
"logps/rejected": -338.0, |
|
"loss": 2.1905, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -3.25, |
|
"rewards/margins": 0.1298828125, |
|
"rewards/rejected": -3.375, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.5232862375719518, |
|
"eval_logits/chosen": -9.3125, |
|
"eval_logits/rejected": -8.5, |
|
"eval_logps/chosen": -338.0, |
|
"eval_logps/rejected": -352.0, |
|
"eval_loss": 2.12862491607666, |
|
"eval_nll_loss": 1.0859375, |
|
"eval_rewards/accuracies": 0.525390625, |
|
"eval_rewards/chosen": -3.375, |
|
"eval_rewards/margins": 0.1357421875, |
|
"eval_rewards/rejected": -3.515625, |
|
"eval_runtime": 23.8436, |
|
"eval_samples_per_second": 83.88, |
|
"eval_steps_per_second": 1.342, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 0.533751962323391, |
|
"grad_norm": 2.556185727055922, |
|
"learning_rate": 4.90825564889749e-05, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -8.875, |
|
"logps/chosen": -356.0, |
|
"logps/rejected": -366.0, |
|
"loss": 2.1994, |
|
"nll_loss": 1.1328125, |
|
"rewards/accuracies": 0.5249999761581421, |
|
"rewards/chosen": -3.5625, |
|
"rewards/margins": 0.10888671875, |
|
"rewards/rejected": -3.671875, |
|
"step": 510 |
|
}, |
|
{ |
|
"epoch": 0.54421768707483, |
|
"grad_norm": 4.392423087548401, |
|
"learning_rate": 4.899899543831288e-05, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -8.4375, |
|
"logps/chosen": -364.0, |
|
"logps/rejected": -380.0, |
|
"loss": 2.2045, |
|
"nll_loss": 1.234375, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -3.640625, |
|
"rewards/margins": 0.1513671875, |
|
"rewards/rejected": -3.796875, |
|
"step": 520 |
|
}, |
|
{ |
|
"epoch": 0.554683411826269, |
|
"grad_norm": 2.8449527492180438, |
|
"learning_rate": 4.891187051966765e-05, |
|
"logits/chosen": -10.3125, |
|
"logits/rejected": -9.1875, |
|
"logps/chosen": -356.0, |
|
"logps/rejected": -378.0, |
|
"loss": 2.1604, |
|
"nll_loss": 1.1484375, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.5625, |
|
"rewards/margins": 0.2265625, |
|
"rewards/rejected": -3.78125, |
|
"step": 530 |
|
}, |
|
{ |
|
"epoch": 0.565149136577708, |
|
"grad_norm": 3.3689324516899917, |
|
"learning_rate": 4.882119467115191e-05, |
|
"logits/chosen": -11.5, |
|
"logits/rejected": -10.4375, |
|
"logps/chosen": -378.0, |
|
"logps/rejected": -390.0, |
|
"loss": 2.214, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.765625, |
|
"rewards/margins": 0.1298828125, |
|
"rewards/rejected": -3.90625, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 0.5756148613291471, |
|
"grad_norm": 3.2210372616233074, |
|
"learning_rate": 4.872698135819402e-05, |
|
"logits/chosen": -11.5, |
|
"logits/rejected": -10.75, |
|
"logps/chosen": -340.0, |
|
"logps/rejected": -358.0, |
|
"loss": 2.2376, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -3.390625, |
|
"rewards/margins": 0.1845703125, |
|
"rewards/rejected": -3.578125, |
|
"step": 550 |
|
}, |
|
{ |
|
"epoch": 0.5860805860805861, |
|
"grad_norm": 4.954461591537085, |
|
"learning_rate": 4.862924457153831e-05, |
|
"logits/chosen": -11.0625, |
|
"logits/rejected": -10.4375, |
|
"logps/chosen": -358.0, |
|
"logps/rejected": -350.0, |
|
"loss": 2.2897, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.578125, |
|
"rewards/margins": -0.072265625, |
|
"rewards/rejected": -3.5, |
|
"step": 560 |
|
}, |
|
{ |
|
"epoch": 0.5965463108320251, |
|
"grad_norm": 5.510678143920604, |
|
"learning_rate": 4.8527998825167484e-05, |
|
"logits/chosen": -10.875, |
|
"logits/rejected": -10.0625, |
|
"logps/chosen": -364.0, |
|
"logps/rejected": -384.0, |
|
"loss": 2.2522, |
|
"nll_loss": 1.1796875, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -3.640625, |
|
"rewards/margins": 0.1982421875, |
|
"rewards/rejected": -3.84375, |
|
"step": 570 |
|
}, |
|
{ |
|
"epoch": 0.6070120355834642, |
|
"grad_norm": 4.1851039582454135, |
|
"learning_rate": 4.842325915414728e-05, |
|
"logits/chosen": -10.375, |
|
"logits/rejected": -9.3125, |
|
"logps/chosen": -362.0, |
|
"logps/rejected": -376.0, |
|
"loss": 2.1945, |
|
"nll_loss": 1.375, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.625, |
|
"rewards/margins": 0.1513671875, |
|
"rewards/rejected": -3.765625, |
|
"step": 580 |
|
}, |
|
{ |
|
"epoch": 0.6174777603349032, |
|
"grad_norm": 2.3813794163046147, |
|
"learning_rate": 4.831504111239373e-05, |
|
"logits/chosen": -10.9375, |
|
"logits/rejected": -9.9375, |
|
"logps/chosen": -346.0, |
|
"logps/rejected": -360.0, |
|
"loss": 2.1063, |
|
"nll_loss": 1.0859375, |
|
"rewards/accuracies": 0.543749988079071, |
|
"rewards/chosen": -3.46875, |
|
"rewards/margins": 0.1357421875, |
|
"rewards/rejected": -3.609375, |
|
"step": 590 |
|
}, |
|
{ |
|
"epoch": 0.6279434850863422, |
|
"grad_norm": 5.268947327110117, |
|
"learning_rate": 4.8203360770363425e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -9.5, |
|
"logps/chosen": -346.0, |
|
"logps/rejected": -392.0, |
|
"loss": 2.1304, |
|
"nll_loss": 1.203125, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -3.46875, |
|
"rewards/margins": 0.453125, |
|
"rewards/rejected": -3.921875, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.6279434850863422, |
|
"eval_logits/chosen": -10.625, |
|
"eval_logits/rejected": -9.625, |
|
"eval_logps/chosen": -360.0, |
|
"eval_logps/rejected": -398.0, |
|
"eval_loss": 2.1410000324249268, |
|
"eval_nll_loss": 1.15625, |
|
"eval_rewards/accuracies": 0.572265625, |
|
"eval_rewards/chosen": -3.609375, |
|
"eval_rewards/margins": 0.3671875, |
|
"eval_rewards/rejected": -3.96875, |
|
"eval_runtime": 23.8448, |
|
"eval_samples_per_second": 83.876, |
|
"eval_steps_per_second": 1.342, |
|
"step": 600 |
|
}, |
|
{ |
|
"epoch": 0.6384092098377813, |
|
"grad_norm": 5.11083506478529, |
|
"learning_rate": 4.808823471266702e-05, |
|
"logits/chosen": -11.0625, |
|
"logits/rejected": -10.0625, |
|
"logps/chosen": -384.0, |
|
"logps/rejected": -384.0, |
|
"loss": 2.2274, |
|
"nll_loss": 1.1875, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -3.84375, |
|
"rewards/margins": 0.0050048828125, |
|
"rewards/rejected": -3.84375, |
|
"step": 610 |
|
}, |
|
{ |
|
"epoch": 0.6488749345892203, |
|
"grad_norm": 3.237989640235622, |
|
"learning_rate": 4.796968003560639e-05, |
|
"logits/chosen": -10.9375, |
|
"logits/rejected": -10.1875, |
|
"logps/chosen": -332.0, |
|
"logps/rejected": -354.0, |
|
"loss": 2.1878, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -3.328125, |
|
"rewards/margins": 0.2109375, |
|
"rewards/rejected": -3.546875, |
|
"step": 620 |
|
}, |
|
{ |
|
"epoch": 0.6593406593406593, |
|
"grad_norm": 2.9331817111766187, |
|
"learning_rate": 4.7847714344635864e-05, |
|
"logits/chosen": -10.5, |
|
"logits/rejected": -9.8125, |
|
"logps/chosen": -340.0, |
|
"logps/rejected": -360.0, |
|
"loss": 2.1614, |
|
"nll_loss": 1.1484375, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -3.40625, |
|
"rewards/margins": 0.208984375, |
|
"rewards/rejected": -3.609375, |
|
"step": 630 |
|
}, |
|
{ |
|
"epoch": 0.6698063840920984, |
|
"grad_norm": 3.728205879658129, |
|
"learning_rate": 4.772235575174776e-05, |
|
"logits/chosen": -11.0, |
|
"logits/rejected": -10.1875, |
|
"logps/chosen": -372.0, |
|
"logps/rejected": -388.0, |
|
"loss": 2.1388, |
|
"nll_loss": 1.296875, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.71875, |
|
"rewards/margins": 0.1689453125, |
|
"rewards/rejected": -3.890625, |
|
"step": 640 |
|
}, |
|
{ |
|
"epoch": 0.6802721088435374, |
|
"grad_norm": 3.496391533292306, |
|
"learning_rate": 4.7593622872782754e-05, |
|
"logits/chosen": -11.875, |
|
"logits/rejected": -11.25, |
|
"logps/chosen": -338.0, |
|
"logps/rejected": -382.0, |
|
"loss": 2.0569, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -3.390625, |
|
"rewards/margins": 0.427734375, |
|
"rewards/rejected": -3.8125, |
|
"step": 650 |
|
}, |
|
{ |
|
"epoch": 0.6907378335949764, |
|
"grad_norm": 4.289852876940576, |
|
"learning_rate": 4.746153482466542e-05, |
|
"logits/chosen": -11.875, |
|
"logits/rejected": -11.0625, |
|
"logps/chosen": -362.0, |
|
"logps/rejected": -382.0, |
|
"loss": 2.1514, |
|
"nll_loss": 1.203125, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.609375, |
|
"rewards/margins": 0.205078125, |
|
"rewards/rejected": -3.8125, |
|
"step": 660 |
|
}, |
|
{ |
|
"epoch": 0.7012035583464155, |
|
"grad_norm": 3.199176358040534, |
|
"learning_rate": 4.732611122256537e-05, |
|
"logits/chosen": -12.125, |
|
"logits/rejected": -11.25, |
|
"logps/chosen": -364.0, |
|
"logps/rejected": -398.0, |
|
"loss": 2.0671, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -3.65625, |
|
"rewards/margins": 0.326171875, |
|
"rewards/rejected": -3.96875, |
|
"step": 670 |
|
}, |
|
{ |
|
"epoch": 0.7116692830978545, |
|
"grad_norm": 2.8516603516299597, |
|
"learning_rate": 4.718737217698435e-05, |
|
"logits/chosen": -11.6875, |
|
"logits/rejected": -10.875, |
|
"logps/chosen": -408.0, |
|
"logps/rejected": -434.0, |
|
"loss": 2.2214, |
|
"nll_loss": 1.21875, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -4.0625, |
|
"rewards/margins": 0.271484375, |
|
"rewards/rejected": -4.34375, |
|
"step": 680 |
|
}, |
|
{ |
|
"epoch": 0.7221350078492935, |
|
"grad_norm": 3.164809857707678, |
|
"learning_rate": 4.704533829076982e-05, |
|
"logits/chosen": -10.8125, |
|
"logits/rejected": -9.3125, |
|
"logps/chosen": -350.0, |
|
"logps/rejected": -372.0, |
|
"loss": 2.1362, |
|
"nll_loss": 1.140625, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -3.5, |
|
"rewards/margins": 0.21875, |
|
"rewards/rejected": -3.71875, |
|
"step": 690 |
|
}, |
|
{ |
|
"epoch": 0.7326007326007326, |
|
"grad_norm": 5.626415120435029, |
|
"learning_rate": 4.6900030656055455e-05, |
|
"logits/chosen": -10.25, |
|
"logits/rejected": -9.1875, |
|
"logps/chosen": -354.0, |
|
"logps/rejected": -386.0, |
|
"loss": 2.2554, |
|
"nll_loss": 1.265625, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.546875, |
|
"rewards/margins": 0.306640625, |
|
"rewards/rejected": -3.84375, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.7326007326007326, |
|
"eval_logits/chosen": -9.6875, |
|
"eval_logits/rejected": -8.5625, |
|
"eval_logps/chosen": -374.0, |
|
"eval_logps/rejected": -416.0, |
|
"eval_loss": 2.1847500801086426, |
|
"eval_nll_loss": 1.203125, |
|
"eval_rewards/accuracies": 0.56640625, |
|
"eval_rewards/chosen": -3.734375, |
|
"eval_rewards/margins": 0.42578125, |
|
"eval_rewards/rejected": -4.15625, |
|
"eval_runtime": 23.8485, |
|
"eval_samples_per_second": 83.863, |
|
"eval_steps_per_second": 1.342, |
|
"step": 700 |
|
}, |
|
{ |
|
"epoch": 0.7430664573521716, |
|
"grad_norm": 2.6320826058976228, |
|
"learning_rate": 4.6751470851128906e-05, |
|
"logits/chosen": -9.9375, |
|
"logits/rejected": -8.9375, |
|
"logps/chosen": -340.0, |
|
"logps/rejected": -364.0, |
|
"loss": 2.1983, |
|
"nll_loss": 1.2578125, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.390625, |
|
"rewards/margins": 0.2470703125, |
|
"rewards/rejected": -3.640625, |
|
"step": 710 |
|
}, |
|
{ |
|
"epoch": 0.7535321821036107, |
|
"grad_norm": 2.477560437712009, |
|
"learning_rate": 4.6599680937227454e-05, |
|
"logits/chosen": -10.375, |
|
"logits/rejected": -9.4375, |
|
"logps/chosen": -358.0, |
|
"logps/rejected": -328.0, |
|
"loss": 2.1528, |
|
"nll_loss": 1.1015625, |
|
"rewards/accuracies": 0.4437499940395355, |
|
"rewards/chosen": -3.578125, |
|
"rewards/margins": -0.283203125, |
|
"rewards/rejected": -3.296875, |
|
"step": 720 |
|
}, |
|
{ |
|
"epoch": 0.7639979068550498, |
|
"grad_norm": 3.062921229679356, |
|
"learning_rate": 4.644468345526183e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -8.6875, |
|
"logps/chosen": -366.0, |
|
"logps/rejected": -362.0, |
|
"loss": 2.2218, |
|
"nll_loss": 1.28125, |
|
"rewards/accuracies": 0.5, |
|
"rewards/chosen": -3.671875, |
|
"rewards/margins": -0.05517578125, |
|
"rewards/rejected": -3.609375, |
|
"step": 730 |
|
}, |
|
{ |
|
"epoch": 0.7744636316064888, |
|
"grad_norm": 3.4309029721288935, |
|
"learning_rate": 4.628650142246894e-05, |
|
"logits/chosen": -10.6875, |
|
"logits/rejected": -10.0625, |
|
"logps/chosen": -318.0, |
|
"logps/rejected": -330.0, |
|
"loss": 2.1265, |
|
"nll_loss": 1.1875, |
|
"rewards/accuracies": 0.512499988079071, |
|
"rewards/chosen": -3.1875, |
|
"rewards/margins": 0.1044921875, |
|
"rewards/rejected": -3.296875, |
|
"step": 740 |
|
}, |
|
{ |
|
"epoch": 0.7849293563579278, |
|
"grad_norm": 2.3686721366460746, |
|
"learning_rate": 4.6125158328993765e-05, |
|
"logits/chosen": -11.0, |
|
"logits/rejected": -10.125, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -362.0, |
|
"loss": 2.0257, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.550000011920929, |
|
"rewards/chosen": -3.125, |
|
"rewards/margins": 0.494140625, |
|
"rewards/rejected": -3.625, |
|
"step": 750 |
|
}, |
|
{ |
|
"epoch": 0.7953950811093669, |
|
"grad_norm": 4.013734605341396, |
|
"learning_rate": 4.5960678134401064e-05, |
|
"logits/chosen": -10.6875, |
|
"logits/rejected": -9.9375, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -340.0, |
|
"loss": 2.145, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.59375, |
|
"rewards/chosen": -3.03125, |
|
"rewards/margins": 0.369140625, |
|
"rewards/rejected": -3.40625, |
|
"step": 760 |
|
}, |
|
{ |
|
"epoch": 0.8058608058608059, |
|
"grad_norm": 2.8684352742938217, |
|
"learning_rate": 4.579308526411734e-05, |
|
"logits/chosen": -8.375, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -412.0, |
|
"logps/rejected": -462.0, |
|
"loss": 2.0731, |
|
"nll_loss": 1.328125, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -4.125, |
|
"rewards/margins": 0.49609375, |
|
"rewards/rejected": -4.625, |
|
"step": 770 |
|
}, |
|
{ |
|
"epoch": 0.8163265306122449, |
|
"grad_norm": 2.653772486909879, |
|
"learning_rate": 4.562240460580367e-05, |
|
"logits/chosen": -7.625, |
|
"logits/rejected": -6.625, |
|
"logps/chosen": -386.0, |
|
"logps/rejected": -388.0, |
|
"loss": 2.1873, |
|
"nll_loss": 1.1484375, |
|
"rewards/accuracies": 0.46875, |
|
"rewards/chosen": -3.859375, |
|
"rewards/margins": 0.03369140625, |
|
"rewards/rejected": -3.890625, |
|
"step": 780 |
|
}, |
|
{ |
|
"epoch": 0.826792255363684, |
|
"grad_norm": 3.234827192057911, |
|
"learning_rate": 4.544866150565985e-05, |
|
"logits/chosen": -7.5625, |
|
"logits/rejected": -6.4375, |
|
"logps/chosen": -372.0, |
|
"logps/rejected": -386.0, |
|
"loss": 2.1357, |
|
"nll_loss": 1.1953125, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -3.734375, |
|
"rewards/margins": 0.1259765625, |
|
"rewards/rejected": -3.859375, |
|
"step": 790 |
|
}, |
|
{ |
|
"epoch": 0.837257980115123, |
|
"grad_norm": 2.794273526465123, |
|
"learning_rate": 4.527188176466051e-05, |
|
"logits/chosen": -7.6875, |
|
"logits/rejected": -6.90625, |
|
"logps/chosen": -338.0, |
|
"logps/rejected": -384.0, |
|
"loss": 2.0796, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.643750011920929, |
|
"rewards/chosen": -3.390625, |
|
"rewards/margins": 0.447265625, |
|
"rewards/rejected": -3.828125, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.837257980115123, |
|
"eval_logits/chosen": -7.84375, |
|
"eval_logits/rejected": -7.03125, |
|
"eval_logps/chosen": -346.0, |
|
"eval_logps/rejected": -374.0, |
|
"eval_loss": 2.1224374771118164, |
|
"eval_nll_loss": 1.1171875, |
|
"eval_rewards/accuracies": 0.546875, |
|
"eval_rewards/chosen": -3.453125, |
|
"eval_rewards/margins": 0.28515625, |
|
"eval_rewards/rejected": -3.75, |
|
"eval_runtime": 23.8636, |
|
"eval_samples_per_second": 83.81, |
|
"eval_steps_per_second": 1.341, |
|
"step": 800 |
|
}, |
|
{ |
|
"epoch": 0.847723704866562, |
|
"grad_norm": 2.7942110680363337, |
|
"learning_rate": 4.509209163472357e-05, |
|
"logits/chosen": -7.875, |
|
"logits/rejected": -7.0, |
|
"logps/chosen": -376.0, |
|
"logps/rejected": -390.0, |
|
"loss": 2.2203, |
|
"nll_loss": 1.1953125, |
|
"rewards/accuracies": 0.45625001192092896, |
|
"rewards/chosen": -3.765625, |
|
"rewards/margins": 0.12890625, |
|
"rewards/rejected": -3.890625, |
|
"step": 810 |
|
}, |
|
{ |
|
"epoch": 0.858189429618001, |
|
"grad_norm": 3.1202132891677152, |
|
"learning_rate": 4.4909317814811896e-05, |
|
"logits/chosen": -8.5625, |
|
"logits/rejected": -7.75, |
|
"logps/chosen": -396.0, |
|
"logps/rejected": -428.0, |
|
"loss": 2.1099, |
|
"nll_loss": 1.265625, |
|
"rewards/accuracies": 0.53125, |
|
"rewards/chosen": -3.953125, |
|
"rewards/margins": 0.333984375, |
|
"rewards/rejected": -4.28125, |
|
"step": 820 |
|
}, |
|
{ |
|
"epoch": 0.8686551543694401, |
|
"grad_norm": 3.2438580599441122, |
|
"learning_rate": 4.472358744696845e-05, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -8.5625, |
|
"logps/chosen": -322.0, |
|
"logps/rejected": -342.0, |
|
"loss": 2.1732, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.5375000238418579, |
|
"rewards/chosen": -3.21875, |
|
"rewards/margins": 0.201171875, |
|
"rewards/rejected": -3.421875, |
|
"step": 830 |
|
}, |
|
{ |
|
"epoch": 0.8791208791208791, |
|
"grad_norm": 2.78727753072508, |
|
"learning_rate": 4.4534928112285645e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.90625, |
|
"logps/chosen": -346.0, |
|
"logps/rejected": -400.0, |
|
"loss": 2.0604, |
|
"nll_loss": 1.078125, |
|
"rewards/accuracies": 0.606249988079071, |
|
"rewards/chosen": -3.46875, |
|
"rewards/margins": 0.5390625, |
|
"rewards/rejected": -4.0, |
|
"step": 840 |
|
}, |
|
{ |
|
"epoch": 0.8895866038723181, |
|
"grad_norm": 2.3901328455932074, |
|
"learning_rate": 4.43433678268096e-05, |
|
"logits/chosen": -8.5625, |
|
"logits/rejected": -7.4375, |
|
"logps/chosen": -378.0, |
|
"logps/rejected": -366.0, |
|
"loss": 2.2998, |
|
"nll_loss": 1.359375, |
|
"rewards/accuracies": 0.518750011920929, |
|
"rewards/chosen": -3.78125, |
|
"rewards/margins": -0.1279296875, |
|
"rewards/rejected": -3.65625, |
|
"step": 850 |
|
}, |
|
{ |
|
"epoch": 0.9000523286237572, |
|
"grad_norm": 3.1175141135313598, |
|
"learning_rate": 4.414893503737968e-05, |
|
"logits/chosen": -8.9375, |
|
"logits/rejected": -7.78125, |
|
"logps/chosen": -362.0, |
|
"logps/rejected": -370.0, |
|
"loss": 2.1977, |
|
"nll_loss": 1.1796875, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.625, |
|
"rewards/margins": 0.0791015625, |
|
"rewards/rejected": -3.703125, |
|
"step": 860 |
|
}, |
|
{ |
|
"epoch": 0.9105180533751962, |
|
"grad_norm": 2.6457741234390912, |
|
"learning_rate": 4.395165861740414e-05, |
|
"logits/chosen": -8.25, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -358.0, |
|
"logps/rejected": -392.0, |
|
"loss": 2.1254, |
|
"nll_loss": 1.171875, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.59375, |
|
"rewards/margins": 0.3359375, |
|
"rewards/rejected": -3.921875, |
|
"step": 870 |
|
}, |
|
{ |
|
"epoch": 0.9209837781266352, |
|
"grad_norm": 2.8040313177453267, |
|
"learning_rate": 4.375156786257241e-05, |
|
"logits/chosen": -6.59375, |
|
"logits/rejected": -6.03125, |
|
"logps/chosen": -402.0, |
|
"logps/rejected": -444.0, |
|
"loss": 2.0949, |
|
"nll_loss": 1.2578125, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -4.03125, |
|
"rewards/margins": 0.41796875, |
|
"rewards/rejected": -4.4375, |
|
"step": 880 |
|
}, |
|
{ |
|
"epoch": 0.9314495028780743, |
|
"grad_norm": 3.613925103198433, |
|
"learning_rate": 4.354869248650466e-05, |
|
"logits/chosen": -6.53125, |
|
"logits/rejected": -5.90625, |
|
"logps/chosen": -374.0, |
|
"logps/rejected": -408.0, |
|
"loss": 2.1019, |
|
"nll_loss": 1.2109375, |
|
"rewards/accuracies": 0.581250011920929, |
|
"rewards/chosen": -3.734375, |
|
"rewards/margins": 0.328125, |
|
"rewards/rejected": -4.0625, |
|
"step": 890 |
|
}, |
|
{ |
|
"epoch": 0.9419152276295133, |
|
"grad_norm": 3.644667230878703, |
|
"learning_rate": 4.334306261633931e-05, |
|
"logits/chosen": -6.9375, |
|
"logits/rejected": -6.25, |
|
"logps/chosen": -374.0, |
|
"logps/rejected": -410.0, |
|
"loss": 2.1021, |
|
"nll_loss": 1.2109375, |
|
"rewards/accuracies": 0.574999988079071, |
|
"rewards/chosen": -3.734375, |
|
"rewards/margins": 0.36328125, |
|
"rewards/rejected": -4.09375, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.9419152276295133, |
|
"eval_logits/chosen": -6.28125, |
|
"eval_logits/rejected": -5.28125, |
|
"eval_logps/chosen": -350.0, |
|
"eval_logps/rejected": -390.0, |
|
"eval_loss": 2.1099374294281006, |
|
"eval_nll_loss": 1.1328125, |
|
"eval_rewards/accuracies": 0.572265625, |
|
"eval_rewards/chosen": -3.5, |
|
"eval_rewards/margins": 0.40625, |
|
"eval_rewards/rejected": -3.90625, |
|
"eval_runtime": 23.8405, |
|
"eval_samples_per_second": 83.891, |
|
"eval_steps_per_second": 1.342, |
|
"step": 900 |
|
}, |
|
{ |
|
"epoch": 0.9523809523809523, |
|
"grad_norm": 2.9413271714226656, |
|
"learning_rate": 4.313470878825911e-05, |
|
"logits/chosen": -6.28125, |
|
"logits/rejected": -5.53125, |
|
"logps/chosen": -282.0, |
|
"logps/rejected": -312.0, |
|
"loss": 2.1899, |
|
"nll_loss": 1.1328125, |
|
"rewards/accuracies": 0.6000000238418579, |
|
"rewards/chosen": -2.8125, |
|
"rewards/margins": 0.302734375, |
|
"rewards/rejected": -3.125, |
|
"step": 910 |
|
}, |
|
{ |
|
"epoch": 0.9628466771323915, |
|
"grad_norm": 3.6857624482360922, |
|
"learning_rate": 4.292366194295652e-05, |
|
"logits/chosen": -7.5, |
|
"logits/rejected": -6.0625, |
|
"logps/chosen": -378.0, |
|
"logps/rejected": -390.0, |
|
"loss": 2.1614, |
|
"nll_loss": 1.265625, |
|
"rewards/accuracies": 0.5562499761581421, |
|
"rewards/chosen": -3.78125, |
|
"rewards/margins": 0.107421875, |
|
"rewards/rejected": -3.890625, |
|
"step": 920 |
|
}, |
|
{ |
|
"epoch": 0.9733124018838305, |
|
"grad_norm": 2.8477330726756303, |
|
"learning_rate": 4.2709953421038964e-05, |
|
"logits/chosen": -8.375, |
|
"logits/rejected": -7.03125, |
|
"logps/chosen": -392.0, |
|
"logps/rejected": -400.0, |
|
"loss": 2.2147, |
|
"nll_loss": 1.234375, |
|
"rewards/accuracies": 0.5062500238418579, |
|
"rewards/chosen": -3.921875, |
|
"rewards/margins": 0.08984375, |
|
"rewards/rejected": -4.0, |
|
"step": 930 |
|
}, |
|
{ |
|
"epoch": 0.9837781266352695, |
|
"grad_norm": 3.497974933792206, |
|
"learning_rate": 4.249361495837478e-05, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -7.8125, |
|
"logps/chosen": -344.0, |
|
"logps/rejected": -376.0, |
|
"loss": 2.0583, |
|
"nll_loss": 1.125, |
|
"rewards/accuracies": 0.5625, |
|
"rewards/chosen": -3.4375, |
|
"rewards/margins": 0.30859375, |
|
"rewards/rejected": -3.75, |
|
"step": 940 |
|
}, |
|
{ |
|
"epoch": 0.9942438513867086, |
|
"grad_norm": 4.031319804588236, |
|
"learning_rate": 4.227467868138035e-05, |
|
"logits/chosen": -8.1875, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -400.0, |
|
"logps/rejected": -426.0, |
|
"loss": 2.1095, |
|
"nll_loss": 1.2421875, |
|
"rewards/accuracies": 0.5687500238418579, |
|
"rewards/chosen": -4.0, |
|
"rewards/margins": 0.2412109375, |
|
"rewards/rejected": -4.25, |
|
"step": 950 |
|
}, |
|
{ |
|
"epoch": 1.0052328623757196, |
|
"grad_norm": 2.368597675912629, |
|
"learning_rate": 4.205317710224938e-05, |
|
"logits/chosen": -7.9375, |
|
"logits/rejected": -6.6875, |
|
"logps/chosen": -260.0, |
|
"logps/rejected": -370.0, |
|
"loss": 1.5186, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.7124999761581421, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 1.09375, |
|
"rewards/rejected": -3.6875, |
|
"step": 960 |
|
}, |
|
{ |
|
"epoch": 1.0156985871271587, |
|
"grad_norm": 2.876968274648732, |
|
"learning_rate": 4.1829143114124733e-05, |
|
"logits/chosen": -9.0625, |
|
"logits/rejected": -7.1875, |
|
"logps/chosen": -334.0, |
|
"logps/rejected": -460.0, |
|
"loss": 1.6065, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.328125, |
|
"rewards/margins": 1.265625, |
|
"rewards/rejected": -4.59375, |
|
"step": 970 |
|
}, |
|
{ |
|
"epoch": 1.0261643118785977, |
|
"grad_norm": 3.3518683379710916, |
|
"learning_rate": 4.160260998621387e-05, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -6.875, |
|
"logps/chosen": -296.0, |
|
"logps/rejected": -424.0, |
|
"loss": 1.5239, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -2.96875, |
|
"rewards/margins": 1.265625, |
|
"rewards/rejected": -4.25, |
|
"step": 980 |
|
}, |
|
{ |
|
"epoch": 1.0366300366300367, |
|
"grad_norm": 2.5087402060085484, |
|
"learning_rate": 4.137361135884824e-05, |
|
"logits/chosen": -10.375, |
|
"logits/rejected": -8.625, |
|
"logps/chosen": -318.0, |
|
"logps/rejected": -444.0, |
|
"loss": 1.4906, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.171875, |
|
"rewards/margins": 1.265625, |
|
"rewards/rejected": -4.4375, |
|
"step": 990 |
|
}, |
|
{ |
|
"epoch": 1.0470957613814758, |
|
"grad_norm": 2.7347842078330125, |
|
"learning_rate": 4.114218123848777e-05, |
|
"logits/chosen": -10.875, |
|
"logits/rejected": -9.1875, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -440.0, |
|
"loss": 1.5182, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.699999988079071, |
|
"rewards/chosen": -3.125, |
|
"rewards/margins": 1.265625, |
|
"rewards/rejected": -4.375, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.0470957613814758, |
|
"eval_logits/chosen": -10.625, |
|
"eval_logits/rejected": -9.375, |
|
"eval_logps/chosen": -350.0, |
|
"eval_logps/rejected": -386.0, |
|
"eval_loss": 2.166249990463257, |
|
"eval_nll_loss": 1.125, |
|
"eval_rewards/accuracies": 0.56640625, |
|
"eval_rewards/chosen": -3.5, |
|
"eval_rewards/margins": 0.36328125, |
|
"eval_rewards/rejected": -3.859375, |
|
"eval_runtime": 23.8697, |
|
"eval_samples_per_second": 83.788, |
|
"eval_steps_per_second": 1.341, |
|
"step": 1000 |
|
}, |
|
{ |
|
"epoch": 1.0575614861329148, |
|
"grad_norm": 3.826594475272154, |
|
"learning_rate": 4.0908353992670824e-05, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.90625, |
|
"logps/chosen": -298.0, |
|
"logps/rejected": -398.0, |
|
"loss": 1.5231, |
|
"nll_loss": 1.0234375, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -2.96875, |
|
"rewards/margins": 1.015625, |
|
"rewards/rejected": -3.984375, |
|
"step": 1010 |
|
}, |
|
{ |
|
"epoch": 1.0680272108843538, |
|
"grad_norm": 2.610572867452082, |
|
"learning_rate": 4.067216434491059e-05, |
|
"logits/chosen": -8.9375, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -332.0, |
|
"logps/rejected": -488.0, |
|
"loss": 1.475, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -3.3125, |
|
"rewards/margins": 1.5703125, |
|
"rewards/rejected": -4.90625, |
|
"step": 1020 |
|
}, |
|
{ |
|
"epoch": 1.0784929356357928, |
|
"grad_norm": 2.5156446255314204, |
|
"learning_rate": 4.043364736953866e-05, |
|
"logits/chosen": -10.1875, |
|
"logits/rejected": -8.3125, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -426.0, |
|
"loss": 1.4255, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.1875, |
|
"rewards/rejected": -4.25, |
|
"step": 1030 |
|
}, |
|
{ |
|
"epoch": 1.0889586603872319, |
|
"grad_norm": 2.2943646379870897, |
|
"learning_rate": 4.019283848649643e-05, |
|
"logits/chosen": -9.9375, |
|
"logits/rejected": -7.75, |
|
"logps/chosen": -284.0, |
|
"logps/rejected": -424.0, |
|
"loss": 1.4634, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -2.84375, |
|
"rewards/margins": 1.390625, |
|
"rewards/rejected": -4.21875, |
|
"step": 1040 |
|
}, |
|
{ |
|
"epoch": 1.099424385138671, |
|
"grad_norm": 2.254369292127638, |
|
"learning_rate": 3.9949773456075215e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -9.375, |
|
"logps/chosen": -292.0, |
|
"logps/rejected": -416.0, |
|
"loss": 1.4558, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -2.90625, |
|
"rewards/margins": 1.2421875, |
|
"rewards/rejected": -4.15625, |
|
"step": 1050 |
|
}, |
|
{ |
|
"epoch": 1.10989010989011, |
|
"grad_norm": 3.1782050354857887, |
|
"learning_rate": 3.970448837360585e-05, |
|
"logits/chosen": -11.0, |
|
"logits/rejected": -9.1875, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -452.0, |
|
"loss": 1.5327, |
|
"nll_loss": 1.0546875, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.3671875, |
|
"rewards/rejected": -4.53125, |
|
"step": 1060 |
|
}, |
|
{ |
|
"epoch": 1.120355834641549, |
|
"grad_norm": 3.4613932983944493, |
|
"learning_rate": 3.945701966409846e-05, |
|
"logits/chosen": -10.25, |
|
"logits/rejected": -8.6875, |
|
"logps/chosen": -288.0, |
|
"logps/rejected": -416.0, |
|
"loss": 1.53, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -2.875, |
|
"rewards/margins": 1.296875, |
|
"rewards/rejected": -4.15625, |
|
"step": 1070 |
|
}, |
|
{ |
|
"epoch": 1.130821559392988, |
|
"grad_norm": 3.3804503848089476, |
|
"learning_rate": 3.9207404076833374e-05, |
|
"logits/chosen": -9.0625, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -272.0, |
|
"logps/rejected": -408.0, |
|
"loss": 1.4583, |
|
"nll_loss": 0.9375, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -2.71875, |
|
"rewards/margins": 1.3515625, |
|
"rewards/rejected": -4.0625, |
|
"step": 1080 |
|
}, |
|
{ |
|
"epoch": 1.141287284144427, |
|
"grad_norm": 2.5758061990307772, |
|
"learning_rate": 3.895567867990379e-05, |
|
"logits/chosen": -7.4375, |
|
"logits/rejected": -6.375, |
|
"logps/chosen": -302.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4956, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -3.015625, |
|
"rewards/margins": 1.53125, |
|
"rewards/rejected": -4.53125, |
|
"step": 1090 |
|
}, |
|
{ |
|
"epoch": 1.151753008895866, |
|
"grad_norm": 2.971286832703863, |
|
"learning_rate": 3.870188085471114e-05, |
|
"logits/chosen": -7.4375, |
|
"logits/rejected": -6.03125, |
|
"logps/chosen": -324.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4917, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.234375, |
|
"rewards/margins": 1.3046875, |
|
"rewards/rejected": -4.53125, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.151753008895866, |
|
"eval_logits/chosen": -7.875, |
|
"eval_logits/rejected": -6.46875, |
|
"eval_logps/chosen": -356.0, |
|
"eval_logps/rejected": -400.0, |
|
"eval_loss": 2.1588125228881836, |
|
"eval_nll_loss": 1.1484375, |
|
"eval_rewards/accuracies": 0.5703125, |
|
"eval_rewards/chosen": -3.5625, |
|
"eval_rewards/margins": 0.439453125, |
|
"eval_rewards/rejected": -4.0, |
|
"eval_runtime": 23.5621, |
|
"eval_samples_per_second": 84.882, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1100 |
|
}, |
|
{ |
|
"epoch": 1.162218733647305, |
|
"grad_norm": 2.512395156631082, |
|
"learning_rate": 3.844604829041395e-05, |
|
"logits/chosen": -8.375, |
|
"logits/rejected": -6.28125, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -418.0, |
|
"loss": 1.4659, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.046875, |
|
"rewards/margins": 1.125, |
|
"rewards/rejected": -4.1875, |
|
"step": 1110 |
|
}, |
|
{ |
|
"epoch": 1.1726844583987441, |
|
"grad_norm": 2.670679023988332, |
|
"learning_rate": 3.818821897833095e-05, |
|
"logits/chosen": -8.0625, |
|
"logits/rejected": -6.90625, |
|
"logps/chosen": -320.0, |
|
"logps/rejected": -464.0, |
|
"loss": 1.506, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -3.203125, |
|
"rewards/margins": 1.453125, |
|
"rewards/rejected": -4.65625, |
|
"step": 1120 |
|
}, |
|
{ |
|
"epoch": 1.1831501831501832, |
|
"grad_norm": 4.0041181858343755, |
|
"learning_rate": 3.792843120629935e-05, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -7.875, |
|
"logps/chosen": -314.0, |
|
"logps/rejected": -452.0, |
|
"loss": 1.4782, |
|
"nll_loss": 0.96875, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.140625, |
|
"rewards/margins": 1.390625, |
|
"rewards/rejected": -4.53125, |
|
"step": 1130 |
|
}, |
|
{ |
|
"epoch": 1.1936159079016222, |
|
"grad_norm": 3.5519518131559975, |
|
"learning_rate": 3.766672355298907e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -448.0, |
|
"loss": 1.4708, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.328125, |
|
"rewards/rejected": -4.46875, |
|
"step": 1140 |
|
}, |
|
{ |
|
"epoch": 1.2040816326530612, |
|
"grad_norm": 5.820728219090705, |
|
"learning_rate": 3.740313488217373e-05, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -8.25, |
|
"logps/chosen": -296.0, |
|
"logps/rejected": -424.0, |
|
"loss": 1.5214, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -2.953125, |
|
"rewards/margins": 1.2890625, |
|
"rewards/rejected": -4.25, |
|
"step": 1150 |
|
}, |
|
{ |
|
"epoch": 1.2145473574045003, |
|
"grad_norm": 3.847218597825991, |
|
"learning_rate": 3.7137704336959464e-05, |
|
"logits/chosen": -10.625, |
|
"logits/rejected": -8.1875, |
|
"logps/chosen": -294.0, |
|
"logps/rejected": -414.0, |
|
"loss": 1.4826, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -2.9375, |
|
"rewards/margins": 1.203125, |
|
"rewards/rejected": -4.15625, |
|
"step": 1160 |
|
}, |
|
{ |
|
"epoch": 1.2250130821559393, |
|
"grad_norm": 3.3358662049941965, |
|
"learning_rate": 3.687047133397201e-05, |
|
"logits/chosen": -10.25, |
|
"logits/rejected": -8.625, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -436.0, |
|
"loss": 1.4964, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.046875, |
|
"rewards/margins": 1.3125, |
|
"rewards/rejected": -4.375, |
|
"step": 1170 |
|
}, |
|
{ |
|
"epoch": 1.2354788069073783, |
|
"grad_norm": 2.595363694595805, |
|
"learning_rate": 3.6601475557503415e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -8.125, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -468.0, |
|
"loss": 1.4832, |
|
"nll_loss": 1.109375, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.515625, |
|
"rewards/rejected": -4.65625, |
|
"step": 1180 |
|
}, |
|
{ |
|
"epoch": 1.2459445316588174, |
|
"grad_norm": 3.2947644841881645, |
|
"learning_rate": 3.633075695361882e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.84375, |
|
"logps/chosen": -326.0, |
|
"logps/rejected": -448.0, |
|
"loss": 1.5003, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -3.265625, |
|
"rewards/margins": 1.2109375, |
|
"rewards/rejected": -4.46875, |
|
"step": 1190 |
|
}, |
|
{ |
|
"epoch": 1.2564102564102564, |
|
"grad_norm": 2.6317393209521174, |
|
"learning_rate": 3.6058355724224477e-05, |
|
"logits/chosen": -8.5625, |
|
"logits/rejected": -7.125, |
|
"logps/chosen": -338.0, |
|
"logps/rejected": -490.0, |
|
"loss": 1.5219, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -3.375, |
|
"rewards/margins": 1.5234375, |
|
"rewards/rejected": -4.90625, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 1.2564102564102564, |
|
"eval_logits/chosen": -7.78125, |
|
"eval_logits/rejected": -6.65625, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -420.0, |
|
"eval_loss": 2.1448750495910645, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.59375, |
|
"eval_rewards/chosen": -3.625, |
|
"eval_rewards/margins": 0.55859375, |
|
"eval_rewards/rejected": -4.1875, |
|
"eval_runtime": 23.5709, |
|
"eval_samples_per_second": 84.85, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1200 |
|
}, |
|
{ |
|
"epoch": 1.2668759811616954, |
|
"grad_norm": 2.510039138103225, |
|
"learning_rate": 3.5784312321097744e-05, |
|
"logits/chosen": -8.3125, |
|
"logits/rejected": -6.3125, |
|
"logps/chosen": -374.0, |
|
"logps/rejected": -520.0, |
|
"loss": 1.5711, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.734375, |
|
"rewards/margins": 1.453125, |
|
"rewards/rejected": -5.1875, |
|
"step": 1210 |
|
}, |
|
{ |
|
"epoch": 1.2773417059131345, |
|
"grad_norm": 2.720203562588732, |
|
"learning_rate": 3.550866743987994e-05, |
|
"logits/chosen": -8.6875, |
|
"logits/rejected": -7.28125, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -436.0, |
|
"loss": 1.5468, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.296875, |
|
"rewards/rejected": -4.34375, |
|
"step": 1220 |
|
}, |
|
{ |
|
"epoch": 1.2878074306645735, |
|
"grad_norm": 2.309291362720241, |
|
"learning_rate": 3.5231462014033024e-05, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -7.21875, |
|
"logps/chosen": -270.0, |
|
"logps/rejected": -408.0, |
|
"loss": 1.4761, |
|
"nll_loss": 0.9453125, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -2.703125, |
|
"rewards/margins": 1.3671875, |
|
"rewards/rejected": -4.0625, |
|
"step": 1230 |
|
}, |
|
{ |
|
"epoch": 1.2982731554160125, |
|
"grad_norm": 2.9715758996573665, |
|
"learning_rate": 3.495273720876095e-05, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -448.0, |
|
"loss": 1.516, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.4140625, |
|
"rewards/rejected": -4.46875, |
|
"step": 1240 |
|
}, |
|
{ |
|
"epoch": 1.3087388801674515, |
|
"grad_norm": 2.5658123482215633, |
|
"learning_rate": 3.4672534414896616e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -6.84375, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -444.0, |
|
"loss": 1.4933, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.375, |
|
"rewards/rejected": -4.4375, |
|
"step": 1250 |
|
}, |
|
{ |
|
"epoch": 1.3192046049188906, |
|
"grad_norm": 2.6304697580509044, |
|
"learning_rate": 3.439089524275526e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -282.0, |
|
"logps/rejected": -404.0, |
|
"loss": 1.4725, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -2.828125, |
|
"rewards/margins": 1.203125, |
|
"rewards/rejected": -4.03125, |
|
"step": 1260 |
|
}, |
|
{ |
|
"epoch": 1.3296703296703296, |
|
"grad_norm": 2.253808862511875, |
|
"learning_rate": 3.410786151595533e-05, |
|
"logits/chosen": -8.75, |
|
"logits/rejected": -6.9375, |
|
"logps/chosen": -292.0, |
|
"logps/rejected": -422.0, |
|
"loss": 1.4924, |
|
"nll_loss": 0.96875, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -2.921875, |
|
"rewards/margins": 1.296875, |
|
"rewards/rejected": -4.21875, |
|
"step": 1270 |
|
}, |
|
{ |
|
"epoch": 1.3401360544217686, |
|
"grad_norm": 2.193838208508527, |
|
"learning_rate": 3.382347526520765e-05, |
|
"logits/chosen": -8.25, |
|
"logits/rejected": -6.3125, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -472.0, |
|
"loss": 1.4688, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.125, |
|
"rewards/margins": 1.609375, |
|
"rewards/rejected": -4.71875, |
|
"step": 1280 |
|
}, |
|
{ |
|
"epoch": 1.3506017791732077, |
|
"grad_norm": 4.0614265113588734, |
|
"learning_rate": 3.353777872207381e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -7.28125, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -444.0, |
|
"loss": 1.5005, |
|
"nll_loss": 0.98046875, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.375, |
|
"rewards/rejected": -4.4375, |
|
"step": 1290 |
|
}, |
|
{ |
|
"epoch": 1.3610675039246467, |
|
"grad_norm": 2.6941793562371776, |
|
"learning_rate": 3.325081431269475e-05, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -8.3125, |
|
"logps/chosen": -300.0, |
|
"logps/rejected": -420.0, |
|
"loss": 1.5292, |
|
"nll_loss": 0.890625, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -3.0, |
|
"rewards/margins": 1.203125, |
|
"rewards/rejected": -4.1875, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 1.3610675039246467, |
|
"eval_logits/chosen": -8.875, |
|
"eval_logits/rejected": -7.75, |
|
"eval_logps/chosen": -354.0, |
|
"eval_logps/rejected": -402.0, |
|
"eval_loss": 2.148937463760376, |
|
"eval_nll_loss": 1.140625, |
|
"eval_rewards/accuracies": 0.57421875, |
|
"eval_rewards/chosen": -3.53125, |
|
"eval_rewards/margins": 0.478515625, |
|
"eval_rewards/rejected": -4.0, |
|
"eval_runtime": 23.5575, |
|
"eval_samples_per_second": 84.899, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1300 |
|
}, |
|
{ |
|
"epoch": 1.3715332286760857, |
|
"grad_norm": 2.510122181170169, |
|
"learning_rate": 3.296262465149046e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -7.3125, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -464.0, |
|
"loss": 1.473, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.125, |
|
"rewards/margins": 1.515625, |
|
"rewards/rejected": -4.625, |
|
"step": 1310 |
|
}, |
|
{ |
|
"epoch": 1.3819989534275248, |
|
"grad_norm": 2.7325123617482867, |
|
"learning_rate": 3.2673252534831687e-05, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -8.5625, |
|
"logps/chosen": -260.0, |
|
"logps/rejected": -382.0, |
|
"loss": 1.4483, |
|
"nll_loss": 0.91015625, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 1.21875, |
|
"rewards/rejected": -3.828125, |
|
"step": 1320 |
|
}, |
|
{ |
|
"epoch": 1.392464678178964, |
|
"grad_norm": 2.8767955548712094, |
|
"learning_rate": 3.2382740934684696e-05, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -7.8125, |
|
"logps/chosen": -314.0, |
|
"logps/rejected": -470.0, |
|
"loss": 1.5189, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.140625, |
|
"rewards/margins": 1.5546875, |
|
"rewards/rejected": -4.6875, |
|
"step": 1330 |
|
}, |
|
{ |
|
"epoch": 1.4029304029304028, |
|
"grad_norm": 3.0868074800483813, |
|
"learning_rate": 3.209113299222982e-05, |
|
"logits/chosen": -7.375, |
|
"logits/rejected": -5.53125, |
|
"logps/chosen": -342.0, |
|
"logps/rejected": -478.0, |
|
"loss": 1.5516, |
|
"nll_loss": 1.078125, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -3.421875, |
|
"rewards/margins": 1.359375, |
|
"rewards/rejected": -4.78125, |
|
"step": 1340 |
|
}, |
|
{ |
|
"epoch": 1.413396127681842, |
|
"grad_norm": 2.382664409358366, |
|
"learning_rate": 3.179847201145505e-05, |
|
"logits/chosen": -7.34375, |
|
"logits/rejected": -5.6875, |
|
"logps/chosen": -298.0, |
|
"logps/rejected": -456.0, |
|
"loss": 1.4925, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -2.984375, |
|
"rewards/margins": 1.5703125, |
|
"rewards/rejected": -4.5625, |
|
"step": 1350 |
|
}, |
|
{ |
|
"epoch": 1.423861852433281, |
|
"grad_norm": 3.6205032569645645, |
|
"learning_rate": 3.150480145272528e-05, |
|
"logits/chosen": -8.9375, |
|
"logits/rejected": -7.125, |
|
"logps/chosen": -280.0, |
|
"logps/rejected": -408.0, |
|
"loss": 1.4731, |
|
"nll_loss": 0.953125, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -2.796875, |
|
"rewards/margins": 1.2890625, |
|
"rewards/rejected": -4.0625, |
|
"step": 1360 |
|
}, |
|
{ |
|
"epoch": 1.4343275771847201, |
|
"grad_norm": 2.3207782590899395, |
|
"learning_rate": 3.121016492632848e-05, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -302.0, |
|
"logps/rejected": -446.0, |
|
"loss": 1.5057, |
|
"nll_loss": 1.078125, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -3.03125, |
|
"rewards/margins": 1.4453125, |
|
"rewards/rejected": -4.46875, |
|
"step": 1370 |
|
}, |
|
{ |
|
"epoch": 1.444793301936159, |
|
"grad_norm": 2.3314369688814467, |
|
"learning_rate": 3.091460618599951e-05, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -302.0, |
|
"logps/rejected": -436.0, |
|
"loss": 1.4853, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.015625, |
|
"rewards/margins": 1.3515625, |
|
"rewards/rejected": -4.375, |
|
"step": 1380 |
|
}, |
|
{ |
|
"epoch": 1.4552590266875982, |
|
"grad_norm": 2.3248832817731944, |
|
"learning_rate": 3.061816912242265e-05, |
|
"logits/chosen": -10.0625, |
|
"logits/rejected": -8.125, |
|
"logps/chosen": -292.0, |
|
"logps/rejected": -388.0, |
|
"loss": 1.4917, |
|
"nll_loss": 0.95703125, |
|
"rewards/accuracies": 0.6812499761581421, |
|
"rewards/chosen": -2.921875, |
|
"rewards/margins": 0.95703125, |
|
"rewards/rejected": -3.890625, |
|
"step": 1390 |
|
}, |
|
{ |
|
"epoch": 1.465724751439037, |
|
"grad_norm": 2.2861606791782054, |
|
"learning_rate": 3.0320897756713785e-05, |
|
"logits/chosen": -10.0625, |
|
"logits/rejected": -8.8125, |
|
"logps/chosen": -302.0, |
|
"logps/rejected": -438.0, |
|
"loss": 1.4257, |
|
"nll_loss": 0.94921875, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.03125, |
|
"rewards/margins": 1.3671875, |
|
"rewards/rejected": -4.375, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 1.465724751439037, |
|
"eval_logits/chosen": -9.25, |
|
"eval_logits/rejected": -7.71875, |
|
"eval_logps/chosen": -358.0, |
|
"eval_logps/rejected": -410.0, |
|
"eval_loss": 2.1193125247955322, |
|
"eval_nll_loss": 1.15625, |
|
"eval_rewards/accuracies": 0.580078125, |
|
"eval_rewards/chosen": -3.578125, |
|
"eval_rewards/margins": 0.515625, |
|
"eval_rewards/rejected": -4.09375, |
|
"eval_runtime": 23.5716, |
|
"eval_samples_per_second": 84.848, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1400 |
|
}, |
|
{ |
|
"epoch": 1.4761904761904763, |
|
"grad_norm": 2.311777425360689, |
|
"learning_rate": 3.002283623388332e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.90625, |
|
"logps/chosen": -296.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.5045, |
|
"nll_loss": 1.0234375, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -2.96875, |
|
"rewards/margins": 1.5703125, |
|
"rewards/rejected": -4.53125, |
|
"step": 1410 |
|
}, |
|
{ |
|
"epoch": 1.4866562009419153, |
|
"grad_norm": 3.9395933722571392, |
|
"learning_rate": 2.9724028816280507e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.90625, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -410.0, |
|
"loss": 1.5194, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.171875, |
|
"rewards/margins": 0.9375, |
|
"rewards/rejected": -4.09375, |
|
"step": 1420 |
|
}, |
|
{ |
|
"epoch": 1.4971219256933543, |
|
"grad_norm": 2.2276738202434183, |
|
"learning_rate": 2.9424519877020524e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -6.71875, |
|
"logps/chosen": -326.0, |
|
"logps/rejected": -508.0, |
|
"loss": 1.514, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -3.25, |
|
"rewards/margins": 1.828125, |
|
"rewards/rejected": -5.09375, |
|
"step": 1430 |
|
}, |
|
{ |
|
"epoch": 1.5075876504447931, |
|
"grad_norm": 2.647176176881213, |
|
"learning_rate": 2.912435389339504e-05, |
|
"logits/chosen": -9.0625, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -488.0, |
|
"loss": 1.5127, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -3.046875, |
|
"rewards/margins": 1.828125, |
|
"rewards/rejected": -4.875, |
|
"step": 1440 |
|
}, |
|
{ |
|
"epoch": 1.5180533751962324, |
|
"grad_norm": 3.0494139350009144, |
|
"learning_rate": 2.8823575440267226e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -342.0, |
|
"logps/rejected": -502.0, |
|
"loss": 1.4784, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.421875, |
|
"rewards/margins": 1.6015625, |
|
"rewards/rejected": -5.03125, |
|
"step": 1450 |
|
}, |
|
{ |
|
"epoch": 1.5285190999476712, |
|
"grad_norm": 2.4026327911339993, |
|
"learning_rate": 2.8522229183452465e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.65625, |
|
"logps/chosen": -308.0, |
|
"logps/rejected": -490.0, |
|
"loss": 1.4115, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -3.078125, |
|
"rewards/margins": 1.8125, |
|
"rewards/rejected": -4.90625, |
|
"step": 1460 |
|
}, |
|
{ |
|
"epoch": 1.5389848246991105, |
|
"grad_norm": 3.6975008895041444, |
|
"learning_rate": 2.8220359873085373e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.96875, |
|
"logps/chosen": -336.0, |
|
"logps/rejected": -488.0, |
|
"loss": 1.4782, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.375, |
|
"rewards/margins": 1.5, |
|
"rewards/rejected": -4.875, |
|
"step": 1470 |
|
}, |
|
{ |
|
"epoch": 1.5494505494505495, |
|
"grad_norm": 2.7095303042025423, |
|
"learning_rate": 2.7918012336974388e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.9375, |
|
"logps/chosen": -282.0, |
|
"logps/rejected": -412.0, |
|
"loss": 1.4811, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -2.828125, |
|
"rewards/margins": 1.2890625, |
|
"rewards/rejected": -4.125, |
|
"step": 1480 |
|
}, |
|
{ |
|
"epoch": 1.5599162742019885, |
|
"grad_norm": 3.5103283923325606, |
|
"learning_rate": 2.7615231473944815e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -318.0, |
|
"logps/rejected": -428.0, |
|
"loss": 1.4835, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.706250011920929, |
|
"rewards/chosen": -3.171875, |
|
"rewards/margins": 1.1171875, |
|
"rewards/rejected": -4.28125, |
|
"step": 1490 |
|
}, |
|
{ |
|
"epoch": 1.5703819989534276, |
|
"grad_norm": 2.5416199689341545, |
|
"learning_rate": 2.7312062247171327e-05, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -322.0, |
|
"logps/rejected": -484.0, |
|
"loss": 1.4366, |
|
"nll_loss": 0.98828125, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.21875, |
|
"rewards/margins": 1.609375, |
|
"rewards/rejected": -4.8125, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 1.5703819989534276, |
|
"eval_logits/chosen": -8.9375, |
|
"eval_logits/rejected": -7.6875, |
|
"eval_logps/chosen": -358.0, |
|
"eval_logps/rejected": -416.0, |
|
"eval_loss": 2.0983126163482666, |
|
"eval_nll_loss": 1.15625, |
|
"eval_rewards/accuracies": 0.58984375, |
|
"eval_rewards/chosen": -3.59375, |
|
"eval_rewards/margins": 0.55859375, |
|
"eval_rewards/rejected": -4.15625, |
|
"eval_runtime": 23.5599, |
|
"eval_samples_per_second": 84.89, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1500 |
|
}, |
|
{ |
|
"epoch": 1.5808477237048666, |
|
"grad_norm": 2.845086791971653, |
|
"learning_rate": 2.7008549677500876e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -330.0, |
|
"logps/rejected": -492.0, |
|
"loss": 1.4932, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.3125, |
|
"rewards/margins": 1.609375, |
|
"rewards/rejected": -4.90625, |
|
"step": 1510 |
|
}, |
|
{ |
|
"epoch": 1.5913134484563056, |
|
"grad_norm": 2.7135730551205395, |
|
"learning_rate": 2.670473883676709e-05, |
|
"logits/chosen": -10.125, |
|
"logits/rejected": -8.375, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -456.0, |
|
"loss": 1.4426, |
|
"nll_loss": 0.93359375, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.046875, |
|
"rewards/margins": 1.53125, |
|
"rewards/rejected": -4.5625, |
|
"step": 1520 |
|
}, |
|
{ |
|
"epoch": 1.6017791732077447, |
|
"grad_norm": 2.359216904744445, |
|
"learning_rate": 2.6400674841097074e-05, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -7.78125, |
|
"logps/chosen": -298.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4303, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -2.984375, |
|
"rewards/margins": 1.5625, |
|
"rewards/rejected": -4.53125, |
|
"step": 1530 |
|
}, |
|
{ |
|
"epoch": 1.6122448979591837, |
|
"grad_norm": 3.837850029068777, |
|
"learning_rate": 2.609640284421158e-05, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -8.25, |
|
"logps/chosen": -318.0, |
|
"logps/rejected": -462.0, |
|
"loss": 1.4766, |
|
"nll_loss": 0.9765625, |
|
"rewards/accuracies": 0.7250000238418579, |
|
"rewards/chosen": -3.1875, |
|
"rewards/margins": 1.4375, |
|
"rewards/rejected": -4.625, |
|
"step": 1540 |
|
}, |
|
{ |
|
"epoch": 1.6227106227106227, |
|
"grad_norm": 2.5563923657380534, |
|
"learning_rate": 2.5791968030719726e-05, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -320.0, |
|
"logps/rejected": -494.0, |
|
"loss": 1.4779, |
|
"nll_loss": 1.0234375, |
|
"rewards/accuracies": 0.824999988079071, |
|
"rewards/chosen": -3.1875, |
|
"rewards/margins": 1.7578125, |
|
"rewards/rejected": -4.9375, |
|
"step": 1550 |
|
}, |
|
{ |
|
"epoch": 1.6331763474620618, |
|
"grad_norm": 2.8490698410048614, |
|
"learning_rate": 2.548741560940902e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -6.96875, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -456.0, |
|
"loss": 1.5065, |
|
"nll_loss": 0.94140625, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.03125, |
|
"rewards/margins": 1.53125, |
|
"rewards/rejected": -4.5625, |
|
"step": 1560 |
|
}, |
|
{ |
|
"epoch": 1.6436420722135008, |
|
"grad_norm": 2.69778186517361, |
|
"learning_rate": 2.5182790806531787e-05, |
|
"logits/chosen": -8.625, |
|
"logits/rejected": -6.40625, |
|
"logps/chosen": -276.0, |
|
"logps/rejected": -428.0, |
|
"loss": 1.4293, |
|
"nll_loss": 0.94921875, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -2.765625, |
|
"rewards/margins": 1.5234375, |
|
"rewards/rejected": -4.28125, |
|
"step": 1570 |
|
}, |
|
{ |
|
"epoch": 1.6541077969649398, |
|
"grad_norm": 3.2438743258754372, |
|
"learning_rate": 2.4878138859089074e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -6.625, |
|
"logps/chosen": -304.0, |
|
"logps/rejected": -466.0, |
|
"loss": 1.4033, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.03125, |
|
"rewards/margins": 1.625, |
|
"rewards/rejected": -4.65625, |
|
"step": 1580 |
|
}, |
|
{ |
|
"epoch": 1.6645735217163788, |
|
"grad_norm": 3.367991429804346, |
|
"learning_rate": 2.4573505008112924e-05, |
|
"logits/chosen": -8.1875, |
|
"logits/rejected": -6.0625, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4926, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.762499988079071, |
|
"rewards/chosen": -3.0625, |
|
"rewards/margins": 1.4765625, |
|
"rewards/rejected": -4.53125, |
|
"step": 1590 |
|
}, |
|
{ |
|
"epoch": 1.6750392464678179, |
|
"grad_norm": 2.6645341517558667, |
|
"learning_rate": 2.4268934491948032e-05, |
|
"logits/chosen": -7.4375, |
|
"logits/rejected": -5.65625, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -464.0, |
|
"loss": 1.5246, |
|
"nll_loss": 1.0703125, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.484375, |
|
"rewards/rejected": -4.625, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 1.6750392464678179, |
|
"eval_logits/chosen": -6.90625, |
|
"eval_logits/rejected": -5.46875, |
|
"eval_logps/chosen": -358.0, |
|
"eval_logps/rejected": -420.0, |
|
"eval_loss": 2.1191248893737793, |
|
"eval_nll_loss": 1.15625, |
|
"eval_rewards/accuracies": 0.59375, |
|
"eval_rewards/chosen": -3.578125, |
|
"eval_rewards/margins": 0.625, |
|
"eval_rewards/rejected": -4.21875, |
|
"eval_runtime": 23.5644, |
|
"eval_samples_per_second": 84.874, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1600 |
|
}, |
|
{ |
|
"epoch": 1.685504971219257, |
|
"grad_norm": 3.3557826805700572, |
|
"learning_rate": 2.396447253953385e-05, |
|
"logits/chosen": -7.84375, |
|
"logits/rejected": -5.90625, |
|
"logps/chosen": -328.0, |
|
"logps/rejected": -486.0, |
|
"loss": 1.4317, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -3.265625, |
|
"rewards/margins": 1.5859375, |
|
"rewards/rejected": -4.84375, |
|
"step": 1610 |
|
}, |
|
{ |
|
"epoch": 1.695970695970696, |
|
"grad_norm": 2.503348875049068, |
|
"learning_rate": 2.3660164363687998e-05, |
|
"logits/chosen": -7.65625, |
|
"logits/rejected": -5.875, |
|
"logps/chosen": -292.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4726, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -2.921875, |
|
"rewards/margins": 1.609375, |
|
"rewards/rejected": -4.53125, |
|
"step": 1620 |
|
}, |
|
{ |
|
"epoch": 1.706436420722135, |
|
"grad_norm": 3.7732596479869747, |
|
"learning_rate": 2.33560551543922e-05, |
|
"logits/chosen": -7.1875, |
|
"logits/rejected": -6.21875, |
|
"logps/chosen": -280.0, |
|
"logps/rejected": -416.0, |
|
"loss": 1.4459, |
|
"nll_loss": 0.98828125, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -2.796875, |
|
"rewards/margins": 1.375, |
|
"rewards/rejected": -4.15625, |
|
"step": 1630 |
|
}, |
|
{ |
|
"epoch": 1.716902145473574, |
|
"grad_norm": 2.2800464607759667, |
|
"learning_rate": 2.3052190072081493e-05, |
|
"logits/chosen": -7.1875, |
|
"logits/rejected": -5.8125, |
|
"logps/chosen": -330.0, |
|
"logps/rejected": -502.0, |
|
"loss": 1.5034, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -3.296875, |
|
"rewards/margins": 1.734375, |
|
"rewards/rejected": -5.03125, |
|
"step": 1640 |
|
}, |
|
{ |
|
"epoch": 1.727367870225013, |
|
"grad_norm": 3.538539709651917, |
|
"learning_rate": 2.2748614240937864e-05, |
|
"logits/chosen": -6.9375, |
|
"logits/rejected": -5.3125, |
|
"logps/chosen": -308.0, |
|
"logps/rejected": -462.0, |
|
"loss": 1.4841, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.793749988079071, |
|
"rewards/chosen": -3.078125, |
|
"rewards/margins": 1.546875, |
|
"rewards/rejected": -4.625, |
|
"step": 1650 |
|
}, |
|
{ |
|
"epoch": 1.737833594976452, |
|
"grad_norm": 3.080319075448567, |
|
"learning_rate": 2.2445372742189334e-05, |
|
"logits/chosen": -6.40625, |
|
"logits/rejected": -5.65625, |
|
"logps/chosen": -300.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4982, |
|
"nll_loss": 1.015625, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -3.0, |
|
"rewards/margins": 1.53125, |
|
"rewards/rejected": -4.53125, |
|
"step": 1660 |
|
}, |
|
{ |
|
"epoch": 1.748299319727891, |
|
"grad_norm": 2.572111738226407, |
|
"learning_rate": 2.214251060741528e-05, |
|
"logits/chosen": -8.5, |
|
"logits/rejected": -6.40625, |
|
"logps/chosen": -308.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4111, |
|
"nll_loss": 0.9765625, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.078125, |
|
"rewards/margins": 1.4609375, |
|
"rewards/rejected": -4.53125, |
|
"step": 1670 |
|
}, |
|
{ |
|
"epoch": 1.7587650444793304, |
|
"grad_norm": 2.3631883767661437, |
|
"learning_rate": 2.1840072811859302e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.84375, |
|
"logps/chosen": -266.0, |
|
"logps/rejected": -412.0, |
|
"loss": 1.4375, |
|
"nll_loss": 0.94921875, |
|
"rewards/accuracies": 0.75, |
|
"rewards/chosen": -2.65625, |
|
"rewards/margins": 1.453125, |
|
"rewards/rejected": -4.125, |
|
"step": 1680 |
|
}, |
|
{ |
|
"epoch": 1.7692307692307692, |
|
"grad_norm": 2.224073905675568, |
|
"learning_rate": 2.1538104267750287e-05, |
|
"logits/chosen": -10.4375, |
|
"logits/rejected": -8.375, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -456.0, |
|
"loss": 1.3951, |
|
"nll_loss": 0.921875, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -3.109375, |
|
"rewards/margins": 1.4375, |
|
"rewards/rejected": -4.5625, |
|
"step": 1690 |
|
}, |
|
{ |
|
"epoch": 1.7796964939822084, |
|
"grad_norm": 2.954889161493239, |
|
"learning_rate": 2.1236649817632952e-05, |
|
"logits/chosen": -10.0, |
|
"logits/rejected": -8.9375, |
|
"logps/chosen": -314.0, |
|
"logps/rejected": -454.0, |
|
"loss": 1.4534, |
|
"nll_loss": 1.1171875, |
|
"rewards/accuracies": 0.7562500238418579, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.390625, |
|
"rewards/rejected": -4.53125, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 1.7796964939822084, |
|
"eval_logits/chosen": -10.0625, |
|
"eval_logits/rejected": -9.0625, |
|
"eval_logps/chosen": -348.0, |
|
"eval_logps/rejected": -404.0, |
|
"eval_loss": 2.0828750133514404, |
|
"eval_nll_loss": 1.1171875, |
|
"eval_rewards/accuracies": 0.576171875, |
|
"eval_rewards/chosen": -3.46875, |
|
"eval_rewards/margins": 0.5625, |
|
"eval_rewards/rejected": -4.03125, |
|
"eval_runtime": 23.5775, |
|
"eval_samples_per_second": 84.827, |
|
"eval_steps_per_second": 1.357, |
|
"step": 1700 |
|
}, |
|
{ |
|
"epoch": 1.7901622187336472, |
|
"grad_norm": 3.4975556721860466, |
|
"learning_rate": 2.093575422770872e-05, |
|
"logits/chosen": -10.1875, |
|
"logits/rejected": -8.5625, |
|
"logps/chosen": -328.0, |
|
"logps/rejected": -460.0, |
|
"loss": 1.5228, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.6937500238418579, |
|
"rewards/chosen": -3.28125, |
|
"rewards/margins": 1.3046875, |
|
"rewards/rejected": -4.59375, |
|
"step": 1710 |
|
}, |
|
{ |
|
"epoch": 1.8006279434850865, |
|
"grad_norm": 2.112328331988898, |
|
"learning_rate": 2.0635462181187827e-05, |
|
"logits/chosen": -10.0625, |
|
"logits/rejected": -8.6875, |
|
"logps/chosen": -312.0, |
|
"logps/rejected": -440.0, |
|
"loss": 1.4413, |
|
"nll_loss": 1.0, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -3.109375, |
|
"rewards/margins": 1.2890625, |
|
"rewards/rejected": -4.40625, |
|
"step": 1720 |
|
}, |
|
{ |
|
"epoch": 1.8110936682365253, |
|
"grad_norm": 6.741155164292504, |
|
"learning_rate": 2.0335818271653917e-05, |
|
"logits/chosen": -10.1875, |
|
"logits/rejected": -8.0, |
|
"logps/chosen": -318.0, |
|
"logps/rejected": -476.0, |
|
"loss": 1.4747, |
|
"nll_loss": 0.9765625, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -3.171875, |
|
"rewards/margins": 1.59375, |
|
"rewards/rejected": -4.78125, |
|
"step": 1730 |
|
}, |
|
{ |
|
"epoch": 1.8215593929879645, |
|
"grad_norm": 3.665537621734146, |
|
"learning_rate": 2.0036866996441815e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -322.0, |
|
"logps/rejected": -468.0, |
|
"loss": 1.4947, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -3.21875, |
|
"rewards/margins": 1.4609375, |
|
"rewards/rejected": -4.65625, |
|
"step": 1740 |
|
}, |
|
{ |
|
"epoch": 1.8320251177394034, |
|
"grad_norm": 4.5241842624263295, |
|
"learning_rate": 1.9738652750029677e-05, |
|
"logits/chosen": -10.0, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -434.0, |
|
"loss": 1.5549, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.1875, |
|
"rewards/rejected": -4.34375, |
|
"step": 1750 |
|
}, |
|
{ |
|
"epoch": 1.8424908424908426, |
|
"grad_norm": 3.365693380138734, |
|
"learning_rate": 1.944121981744631e-05, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.9375, |
|
"logps/chosen": -280.0, |
|
"logps/rejected": -424.0, |
|
"loss": 1.4768, |
|
"nll_loss": 0.97265625, |
|
"rewards/accuracies": 0.768750011920929, |
|
"rewards/chosen": -2.796875, |
|
"rewards/margins": 1.453125, |
|
"rewards/rejected": -4.25, |
|
"step": 1760 |
|
}, |
|
{ |
|
"epoch": 1.8529565672422814, |
|
"grad_norm": 3.182016732627515, |
|
"learning_rate": 1.9144612367694887e-05, |
|
"logits/chosen": -8.8125, |
|
"logits/rejected": -7.28125, |
|
"logps/chosen": -336.0, |
|
"logps/rejected": -486.0, |
|
"loss": 1.4544, |
|
"nll_loss": 1.0703125, |
|
"rewards/accuracies": 0.8125, |
|
"rewards/chosen": -3.375, |
|
"rewards/margins": 1.4921875, |
|
"rewards/rejected": -4.84375, |
|
"step": 1770 |
|
}, |
|
{ |
|
"epoch": 1.8634222919937207, |
|
"grad_norm": 3.6295907844116315, |
|
"learning_rate": 1.8848874447193803e-05, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -332.0, |
|
"logps/rejected": -516.0, |
|
"loss": 1.4867, |
|
"nll_loss": 1.046875, |
|
"rewards/accuracies": 0.831250011920929, |
|
"rewards/chosen": -3.328125, |
|
"rewards/margins": 1.8359375, |
|
"rewards/rejected": -5.15625, |
|
"step": 1780 |
|
}, |
|
{ |
|
"epoch": 1.8738880167451595, |
|
"grad_norm": 2.658051300312962, |
|
"learning_rate": 1.855404997323571e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -300.0, |
|
"logps/rejected": -460.0, |
|
"loss": 1.4911, |
|
"nll_loss": 0.96875, |
|
"rewards/accuracies": 0.8187500238418579, |
|
"rewards/chosen": -3.0, |
|
"rewards/margins": 1.609375, |
|
"rewards/rejected": -4.59375, |
|
"step": 1790 |
|
}, |
|
{ |
|
"epoch": 1.8843537414965987, |
|
"grad_norm": 2.6868680242539034, |
|
"learning_rate": 1.82601827274658e-05, |
|
"logits/chosen": -8.375, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -280.0, |
|
"logps/rejected": -440.0, |
|
"loss": 1.4551, |
|
"nll_loss": 0.984375, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -2.8125, |
|
"rewards/margins": 1.59375, |
|
"rewards/rejected": -4.40625, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 1.8843537414965987, |
|
"eval_logits/chosen": -8.1875, |
|
"eval_logits/rejected": -6.84375, |
|
"eval_logps/chosen": -356.0, |
|
"eval_logps/rejected": -416.0, |
|
"eval_loss": 2.1033124923706055, |
|
"eval_nll_loss": 1.1484375, |
|
"eval_rewards/accuracies": 0.58984375, |
|
"eval_rewards/chosen": -3.5625, |
|
"eval_rewards/margins": 0.6015625, |
|
"eval_rewards/rejected": -4.15625, |
|
"eval_runtime": 23.5652, |
|
"eval_samples_per_second": 84.871, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1800 |
|
}, |
|
{ |
|
"epoch": 1.8948194662480375, |
|
"grad_norm": 2.7026542214842837, |
|
"learning_rate": 1.7967316349380213e-05, |
|
"logits/chosen": -8.375, |
|
"logits/rejected": -6.5, |
|
"logps/chosen": -284.0, |
|
"logps/rejected": -444.0, |
|
"loss": 1.4845, |
|
"nll_loss": 1.03125, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -2.84375, |
|
"rewards/margins": 1.5859375, |
|
"rewards/rejected": -4.4375, |
|
"step": 1810 |
|
}, |
|
{ |
|
"epoch": 1.9052851909994768, |
|
"grad_norm": 3.1799469330530292, |
|
"learning_rate": 1.7675494329845516e-05, |
|
"logits/chosen": -8.75, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -274.0, |
|
"logps/rejected": -434.0, |
|
"loss": 1.4546, |
|
"nll_loss": 0.9765625, |
|
"rewards/accuracies": 0.8374999761581421, |
|
"rewards/chosen": -2.734375, |
|
"rewards/margins": 1.6015625, |
|
"rewards/rejected": -4.34375, |
|
"step": 1820 |
|
}, |
|
{ |
|
"epoch": 1.9157509157509156, |
|
"grad_norm": 3.240480247539202, |
|
"learning_rate": 1.738476000464026e-05, |
|
"logits/chosen": -8.3125, |
|
"logits/rejected": -7.0, |
|
"logps/chosen": -270.0, |
|
"logps/rejected": -450.0, |
|
"loss": 1.4353, |
|
"nll_loss": 0.9609375, |
|
"rewards/accuracies": 0.8062499761581421, |
|
"rewards/chosen": -2.703125, |
|
"rewards/margins": 1.7890625, |
|
"rewards/rejected": -4.5, |
|
"step": 1830 |
|
}, |
|
{ |
|
"epoch": 1.9262166405023549, |
|
"grad_norm": 3.622132727177654, |
|
"learning_rate": 1.7095156548019646e-05, |
|
"logits/chosen": -8.5, |
|
"logits/rejected": -6.90625, |
|
"logps/chosen": -332.0, |
|
"logps/rejected": -508.0, |
|
"loss": 1.4995, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -3.328125, |
|
"rewards/margins": 1.75, |
|
"rewards/rejected": -5.09375, |
|
"step": 1840 |
|
}, |
|
{ |
|
"epoch": 1.9366823652537937, |
|
"grad_norm": 3.03530048740814, |
|
"learning_rate": 1.6806726966304064e-05, |
|
"logits/chosen": -9.0625, |
|
"logits/rejected": -7.5625, |
|
"logps/chosen": -290.0, |
|
"logps/rejected": -436.0, |
|
"loss": 1.4375, |
|
"nll_loss": 0.95703125, |
|
"rewards/accuracies": 0.7875000238418579, |
|
"rewards/chosen": -2.890625, |
|
"rewards/margins": 1.46875, |
|
"rewards/rejected": -4.375, |
|
"step": 1850 |
|
}, |
|
{ |
|
"epoch": 1.947148090005233, |
|
"grad_norm": 4.886249268404818, |
|
"learning_rate": 1.6519514091492625e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.75, |
|
"logps/chosen": -278.0, |
|
"logps/rejected": -422.0, |
|
"loss": 1.4248, |
|
"nll_loss": 0.9375, |
|
"rewards/accuracies": 0.7437499761581421, |
|
"rewards/chosen": -2.78125, |
|
"rewards/margins": 1.453125, |
|
"rewards/rejected": -4.21875, |
|
"step": 1860 |
|
}, |
|
{ |
|
"epoch": 1.957613814756672, |
|
"grad_norm": 2.938779543011128, |
|
"learning_rate": 1.6233560574902627e-05, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -356.0, |
|
"logps/rejected": -516.0, |
|
"loss": 1.4592, |
|
"nll_loss": 1.0390625, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.546875, |
|
"rewards/margins": 1.6015625, |
|
"rewards/rejected": -5.15625, |
|
"step": 1870 |
|
}, |
|
{ |
|
"epoch": 1.968079539508111, |
|
"grad_norm": 3.991866542590152, |
|
"learning_rate": 1.5948908880835753e-05, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -306.0, |
|
"logps/rejected": -460.0, |
|
"loss": 1.4614, |
|
"nll_loss": 1.0078125, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.046875, |
|
"rewards/margins": 1.5546875, |
|
"rewards/rejected": -4.625, |
|
"step": 1880 |
|
}, |
|
{ |
|
"epoch": 1.97854526425955, |
|
"grad_norm": 3.169184939140684, |
|
"learning_rate": 1.5665601280272124e-05, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -8.0, |
|
"logps/chosen": -290.0, |
|
"logps/rejected": -458.0, |
|
"loss": 1.4541, |
|
"nll_loss": 0.8984375, |
|
"rewards/accuracies": 0.7749999761581421, |
|
"rewards/chosen": -2.890625, |
|
"rewards/margins": 1.671875, |
|
"rewards/rejected": -4.5625, |
|
"step": 1890 |
|
}, |
|
{ |
|
"epoch": 1.989010989010989, |
|
"grad_norm": 3.974820515572564, |
|
"learning_rate": 1.538367984459298e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -320.0, |
|
"logps/rejected": -498.0, |
|
"loss": 1.4969, |
|
"nll_loss": 0.98046875, |
|
"rewards/accuracies": 0.800000011920929, |
|
"rewards/chosen": -3.203125, |
|
"rewards/margins": 1.7734375, |
|
"rewards/rejected": -5.0, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 1.989010989010989, |
|
"eval_logits/chosen": -9.3125, |
|
"eval_logits/rejected": -8.125, |
|
"eval_logps/chosen": -354.0, |
|
"eval_logps/rejected": -412.0, |
|
"eval_loss": 2.1046249866485596, |
|
"eval_nll_loss": 1.140625, |
|
"eval_rewards/accuracies": 0.576171875, |
|
"eval_rewards/chosen": -3.53125, |
|
"eval_rewards/margins": 0.59375, |
|
"eval_rewards/rejected": -4.125, |
|
"eval_runtime": 23.5686, |
|
"eval_samples_per_second": 84.859, |
|
"eval_steps_per_second": 1.358, |
|
"step": 1900 |
|
}, |
|
{ |
|
"epoch": 1.999476713762428, |
|
"grad_norm": 4.492859752254448, |
|
"learning_rate": 1.5103186439333133e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.75, |
|
"logps/chosen": -316.0, |
|
"logps/rejected": -466.0, |
|
"loss": 1.3995, |
|
"nll_loss": 0.98828125, |
|
"rewards/accuracies": 0.78125, |
|
"rewards/chosen": -3.15625, |
|
"rewards/margins": 1.484375, |
|
"rewards/rejected": -4.65625, |
|
"step": 1910 |
|
}, |
|
{ |
|
"epoch": 2.009942438513867, |
|
"grad_norm": 2.9371919232369796, |
|
"learning_rate": 1.4824162717963829e-05, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.125, |
|
"logps/chosen": -272.0, |
|
"logps/rejected": -552.0, |
|
"loss": 1.0312, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.71875, |
|
"rewards/margins": 2.8125, |
|
"rewards/rejected": -5.53125, |
|
"step": 1920 |
|
}, |
|
{ |
|
"epoch": 2.020408163265306, |
|
"grad_norm": 1.3455436820818165, |
|
"learning_rate": 1.4546650115707192e-05, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -221.0, |
|
"logps/rejected": -508.0, |
|
"loss": 1.0009, |
|
"nll_loss": 0.85546875, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.21875, |
|
"rewards/margins": 2.875, |
|
"rewards/rejected": -5.09375, |
|
"step": 1930 |
|
}, |
|
{ |
|
"epoch": 2.030873888016745, |
|
"grad_norm": 1.374415806275895, |
|
"learning_rate": 1.4270689843383111e-05, |
|
"logits/chosen": -9.9375, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -226.0, |
|
"logps/rejected": -480.0, |
|
"loss": 0.98, |
|
"nll_loss": 0.78515625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.265625, |
|
"rewards/margins": 2.546875, |
|
"rewards/rejected": -4.8125, |
|
"step": 1940 |
|
}, |
|
{ |
|
"epoch": 2.041339612768184, |
|
"grad_norm": 1.608663547509232, |
|
"learning_rate": 1.3996322881289347e-05, |
|
"logits/chosen": -10.125, |
|
"logits/rejected": -7.15625, |
|
"logps/chosen": -231.0, |
|
"logps/rejected": -510.0, |
|
"loss": 0.9989, |
|
"nll_loss": 0.79296875, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.3125, |
|
"rewards/margins": 2.796875, |
|
"rewards/rejected": -5.09375, |
|
"step": 1950 |
|
}, |
|
{ |
|
"epoch": 2.0518053375196232, |
|
"grad_norm": 15.185497879088356, |
|
"learning_rate": 1.3723589973115963e-05, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -8.25, |
|
"logps/chosen": -216.0, |
|
"logps/rejected": -492.0, |
|
"loss": 0.9781, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.15625, |
|
"rewards/margins": 2.75, |
|
"rewards/rejected": -4.90625, |
|
"step": 1960 |
|
}, |
|
{ |
|
"epoch": 2.062271062271062, |
|
"grad_norm": 1.3434211206454858, |
|
"learning_rate": 1.3452531619894837e-05, |
|
"logits/chosen": -10.125, |
|
"logits/rejected": -7.65625, |
|
"logps/chosen": -244.0, |
|
"logps/rejected": -536.0, |
|
"loss": 0.9496, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.4375, |
|
"rewards/margins": 2.90625, |
|
"rewards/rejected": -5.34375, |
|
"step": 1970 |
|
}, |
|
{ |
|
"epoch": 2.0727367870225013, |
|
"grad_norm": 1.4053720837919303, |
|
"learning_rate": 1.3183188073985245e-05, |
|
"logits/chosen": -10.1875, |
|
"logits/rejected": -8.1875, |
|
"logps/chosen": -246.0, |
|
"logps/rejected": -512.0, |
|
"loss": 0.9834, |
|
"nll_loss": 0.82421875, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.453125, |
|
"rewards/margins": 2.671875, |
|
"rewards/rejected": -5.125, |
|
"step": 1980 |
|
}, |
|
{ |
|
"epoch": 2.0832025117739406, |
|
"grad_norm": 1.6077755334609587, |
|
"learning_rate": 1.2915599333096351e-05, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -254.0, |
|
"logps/rejected": -532.0, |
|
"loss": 0.9802, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.53125, |
|
"rewards/margins": 2.796875, |
|
"rewards/rejected": -5.34375, |
|
"step": 1990 |
|
}, |
|
{ |
|
"epoch": 2.0936682365253794, |
|
"grad_norm": 1.4048237875586909, |
|
"learning_rate": 1.264980513434752e-05, |
|
"logits/chosen": -10.0, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -226.0, |
|
"logps/rejected": -506.0, |
|
"loss": 0.9984, |
|
"nll_loss": 0.71875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.265625, |
|
"rewards/margins": 2.796875, |
|
"rewards/rejected": -5.0625, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.0936682365253794, |
|
"eval_logits/chosen": -9.1875, |
|
"eval_logits/rejected": -7.9375, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -428.0, |
|
"eval_loss": 2.1805624961853027, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.578125, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.63671875, |
|
"eval_rewards/rejected": -4.28125, |
|
"eval_runtime": 23.5664, |
|
"eval_samples_per_second": 84.867, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2000 |
|
}, |
|
{ |
|
"epoch": 2.1041339612768186, |
|
"grad_norm": 1.9282387939512704, |
|
"learning_rate": 1.2385844948367324e-05, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -256.0, |
|
"logps/rejected": -584.0, |
|
"loss": 0.9848, |
|
"nll_loss": 0.80078125, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -2.5625, |
|
"rewards/margins": 3.265625, |
|
"rewards/rejected": -5.84375, |
|
"step": 2010 |
|
}, |
|
{ |
|
"epoch": 2.1145996860282574, |
|
"grad_norm": 2.3944324186483255, |
|
"learning_rate": 1.2123757973432114e-05, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -248.0, |
|
"logps/rejected": -568.0, |
|
"loss": 0.9698, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.484375, |
|
"rewards/margins": 3.21875, |
|
"rewards/rejected": -5.71875, |
|
"step": 2020 |
|
}, |
|
{ |
|
"epoch": 2.1250654107796967, |
|
"grad_norm": 1.4803555420448513, |
|
"learning_rate": 1.1863583129645106e-05, |
|
"logits/chosen": -10.125, |
|
"logits/rejected": -7.71875, |
|
"logps/chosen": -245.0, |
|
"logps/rejected": -544.0, |
|
"loss": 0.9718, |
|
"nll_loss": 0.73046875, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.453125, |
|
"rewards/margins": 3.0, |
|
"rewards/rejected": -5.4375, |
|
"step": 2030 |
|
}, |
|
{ |
|
"epoch": 2.1355311355311355, |
|
"grad_norm": 1.4645691849334241, |
|
"learning_rate": 1.1605359053156607e-05, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -8.625, |
|
"logps/chosen": -213.0, |
|
"logps/rejected": -484.0, |
|
"loss": 0.9959, |
|
"nll_loss": 0.83203125, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.140625, |
|
"rewards/margins": 2.71875, |
|
"rewards/rejected": -4.84375, |
|
"step": 2040 |
|
}, |
|
{ |
|
"epoch": 2.1459968602825747, |
|
"grad_norm": 1.0330433965135244, |
|
"learning_rate": 1.1349124090426644e-05, |
|
"logits/chosen": -10.0625, |
|
"logits/rejected": -7.5625, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -568.0, |
|
"loss": 0.9837, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.578125, |
|
"rewards/margins": 3.109375, |
|
"rewards/rejected": -5.6875, |
|
"step": 2050 |
|
}, |
|
{ |
|
"epoch": 2.1564625850340136, |
|
"grad_norm": 2.9682741194787043, |
|
"learning_rate": 1.1094916292530406e-05, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -8.0, |
|
"logps/chosen": -236.0, |
|
"logps/rejected": -556.0, |
|
"loss": 0.9682, |
|
"nll_loss": 0.8046875, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.359375, |
|
"rewards/margins": 3.1875, |
|
"rewards/rejected": -5.5625, |
|
"step": 2060 |
|
}, |
|
{ |
|
"epoch": 2.166928309785453, |
|
"grad_norm": 1.6299550042977697, |
|
"learning_rate": 1.0842773409507623e-05, |
|
"logits/chosen": -9.75, |
|
"logits/rejected": -7.8125, |
|
"logps/chosen": -237.0, |
|
"logps/rejected": -552.0, |
|
"loss": 1.036, |
|
"nll_loss": 0.8359375, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -2.375, |
|
"rewards/margins": 3.171875, |
|
"rewards/rejected": -5.53125, |
|
"step": 2070 |
|
}, |
|
{ |
|
"epoch": 2.1773940345368916, |
|
"grad_norm": 1.9551626986713795, |
|
"learning_rate": 1.0592732884756754e-05, |
|
"logits/chosen": -10.1875, |
|
"logits/rejected": -7.9375, |
|
"logps/chosen": -260.0, |
|
"logps/rejected": -576.0, |
|
"loss": 0.9807, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -2.609375, |
|
"rewards/margins": 3.15625, |
|
"rewards/rejected": -5.78125, |
|
"step": 2080 |
|
}, |
|
{ |
|
"epoch": 2.187859759288331, |
|
"grad_norm": 1.837515187894112, |
|
"learning_rate": 1.0344831849474507e-05, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -251.0, |
|
"logps/rejected": -572.0, |
|
"loss": 1.0042, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.515625, |
|
"rewards/margins": 3.203125, |
|
"rewards/rejected": -5.71875, |
|
"step": 2090 |
|
}, |
|
{ |
|
"epoch": 2.1983254840397697, |
|
"grad_norm": 2.0902678145468188, |
|
"learning_rate": 1.0099107117141879e-05, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -262.0, |
|
"logps/rejected": -588.0, |
|
"loss": 0.9885, |
|
"nll_loss": 0.8203125, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.609375, |
|
"rewards/margins": 3.25, |
|
"rewards/rejected": -5.875, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 2.1983254840397697, |
|
"eval_logits/chosen": -8.6875, |
|
"eval_logits/rejected": -7.40625, |
|
"eval_logps/chosen": -370.0, |
|
"eval_logps/rejected": -448.0, |
|
"eval_loss": 2.1927499771118164, |
|
"eval_nll_loss": 1.1875, |
|
"eval_rewards/accuracies": 0.580078125, |
|
"eval_rewards/chosen": -3.6875, |
|
"eval_rewards/margins": 0.79296875, |
|
"eval_rewards/rejected": -4.5, |
|
"eval_runtime": 23.5701, |
|
"eval_samples_per_second": 84.853, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2100 |
|
}, |
|
{ |
|
"epoch": 2.208791208791209, |
|
"grad_norm": 2.483503439581386, |
|
"learning_rate": 9.855595178057334e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -260.0, |
|
"logps/rejected": -632.0, |
|
"loss": 0.9791, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 3.75, |
|
"rewards/rejected": -6.34375, |
|
"step": 2110 |
|
}, |
|
{ |
|
"epoch": 2.2192569335426477, |
|
"grad_norm": 2.930732817776249, |
|
"learning_rate": 9.614332193917949e-06, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.71875, |
|
"logps/chosen": -229.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9814, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 3.34375, |
|
"rewards/rejected": -5.625, |
|
"step": 2120 |
|
}, |
|
{ |
|
"epoch": 2.229722658294087, |
|
"grad_norm": 1.7357624269357066, |
|
"learning_rate": 9.375353992449385e-06, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -246.0, |
|
"logps/rejected": -560.0, |
|
"loss": 0.9664, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.453125, |
|
"rewards/margins": 3.15625, |
|
"rewards/rejected": -5.625, |
|
"step": 2130 |
|
}, |
|
{ |
|
"epoch": 2.240188383045526, |
|
"grad_norm": 1.7747883480004614, |
|
"learning_rate": 9.138696062085442e-06, |
|
"logits/chosen": -10.0, |
|
"logits/rejected": -8.0625, |
|
"logps/chosen": -229.0, |
|
"logps/rejected": -548.0, |
|
"loss": 0.9417, |
|
"nll_loss": 0.76171875, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.296875, |
|
"rewards/margins": 3.171875, |
|
"rewards/rejected": -5.46875, |
|
"step": 2140 |
|
}, |
|
{ |
|
"epoch": 2.250654107796965, |
|
"grad_norm": 1.8632799743569926, |
|
"learning_rate": 8.904393546698007e-06, |
|
"logits/chosen": -10.25, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -292.0, |
|
"logps/rejected": -632.0, |
|
"loss": 0.9664, |
|
"nll_loss": 0.79296875, |
|
"rewards/accuracies": 0.981249988079071, |
|
"rewards/chosen": -2.921875, |
|
"rewards/margins": 3.421875, |
|
"rewards/rejected": -6.34375, |
|
"step": 2150 |
|
}, |
|
{ |
|
"epoch": 2.261119832548404, |
|
"grad_norm": 2.5056091614745144, |
|
"learning_rate": 8.672481240378142e-06, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -234.0, |
|
"logps/rejected": -556.0, |
|
"loss": 0.9661, |
|
"nll_loss": 0.80078125, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.34375, |
|
"rewards/margins": 3.203125, |
|
"rewards/rejected": -5.5625, |
|
"step": 2160 |
|
}, |
|
{ |
|
"epoch": 2.271585557299843, |
|
"grad_norm": 1.9331304556998234, |
|
"learning_rate": 8.44299358226919e-06, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -7.71875, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -528.0, |
|
"loss": 0.9782, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 2.6875, |
|
"rewards/rejected": -5.28125, |
|
"step": 2170 |
|
}, |
|
{ |
|
"epoch": 2.282051282051282, |
|
"grad_norm": 2.144829826001929, |
|
"learning_rate": 8.215964651452457e-06, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -7.5625, |
|
"logps/chosen": -282.0, |
|
"logps/rejected": -636.0, |
|
"loss": 0.981, |
|
"nll_loss": 0.8203125, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.828125, |
|
"rewards/margins": 3.515625, |
|
"rewards/rejected": -6.34375, |
|
"step": 2180 |
|
}, |
|
{ |
|
"epoch": 2.292517006802721, |
|
"grad_norm": 3.5998357578774205, |
|
"learning_rate": 7.991428161886502e-06, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -233.0, |
|
"logps/rejected": -548.0, |
|
"loss": 1.0077, |
|
"nll_loss": 0.80859375, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.328125, |
|
"rewards/margins": 3.140625, |
|
"rewards/rejected": -5.46875, |
|
"step": 2190 |
|
}, |
|
{ |
|
"epoch": 2.30298273155416, |
|
"grad_norm": 2.1820439002825283, |
|
"learning_rate": 7.76941745740061e-06, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -221.0, |
|
"logps/rejected": -492.0, |
|
"loss": 0.9814, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.8687499761581421, |
|
"rewards/chosen": -2.203125, |
|
"rewards/margins": 2.71875, |
|
"rewards/rejected": -4.9375, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 2.30298273155416, |
|
"eval_logits/chosen": -8.8125, |
|
"eval_logits/rejected": -7.5, |
|
"eval_logps/chosen": -362.0, |
|
"eval_logps/rejected": -436.0, |
|
"eval_loss": 2.186687469482422, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.57421875, |
|
"eval_rewards/chosen": -3.625, |
|
"eval_rewards/margins": 0.7265625, |
|
"eval_rewards/rejected": -4.34375, |
|
"eval_runtime": 23.5703, |
|
"eval_samples_per_second": 84.852, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2200 |
|
}, |
|
{ |
|
"epoch": 2.3134484563055993, |
|
"grad_norm": 4.370450095674627, |
|
"learning_rate": 7.549965506743123e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -6.21875, |
|
"logps/chosen": -236.0, |
|
"logps/rejected": -536.0, |
|
"loss": 0.9726, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -2.359375, |
|
"rewards/margins": 3.015625, |
|
"rewards/rejected": -5.375, |
|
"step": 2210 |
|
}, |
|
{ |
|
"epoch": 2.323914181057038, |
|
"grad_norm": 2.14165780039551, |
|
"learning_rate": 7.333104898685631e-06, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -276.0, |
|
"logps/rejected": -620.0, |
|
"loss": 0.9735, |
|
"nll_loss": 0.80078125, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -2.75, |
|
"rewards/margins": 3.4375, |
|
"rewards/rejected": -6.1875, |
|
"step": 2220 |
|
}, |
|
{ |
|
"epoch": 2.3343799058084773, |
|
"grad_norm": 1.2787043706328256, |
|
"learning_rate": 7.118867837183518e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -238.0, |
|
"logps/rejected": -540.0, |
|
"loss": 0.9906, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.375, |
|
"rewards/margins": 3.015625, |
|
"rewards/rejected": -5.40625, |
|
"step": 2230 |
|
}, |
|
{ |
|
"epoch": 2.344845630559916, |
|
"grad_norm": 1.4034610306701465, |
|
"learning_rate": 6.9072861365936045e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.15625, |
|
"logps/chosen": -208.0, |
|
"logps/rejected": -516.0, |
|
"loss": 0.9554, |
|
"nll_loss": 0.7578125, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.078125, |
|
"rewards/margins": 3.09375, |
|
"rewards/rejected": -5.15625, |
|
"step": 2240 |
|
}, |
|
{ |
|
"epoch": 2.3553113553113554, |
|
"grad_norm": 1.325435693075213, |
|
"learning_rate": 6.6983912169497005e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -251.0, |
|
"logps/rejected": -576.0, |
|
"loss": 0.9938, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.515625, |
|
"rewards/margins": 3.25, |
|
"rewards/rejected": -5.78125, |
|
"step": 2250 |
|
}, |
|
{ |
|
"epoch": 2.365777080062794, |
|
"grad_norm": 2.254387575955096, |
|
"learning_rate": 6.49221409929677e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -268.0, |
|
"logps/rejected": -604.0, |
|
"loss": 0.9537, |
|
"nll_loss": 0.78125, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.6875, |
|
"rewards/margins": 3.34375, |
|
"rewards/rejected": -6.03125, |
|
"step": 2260 |
|
}, |
|
{ |
|
"epoch": 2.3762428048142334, |
|
"grad_norm": 3.9073353920242364, |
|
"learning_rate": 6.2887854010842075e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -6.96875, |
|
"logps/chosen": -268.0, |
|
"logps/rejected": -540.0, |
|
"loss": 0.9981, |
|
"nll_loss": 0.85546875, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -2.6875, |
|
"rewards/margins": 2.71875, |
|
"rewards/rejected": -5.40625, |
|
"step": 2270 |
|
}, |
|
{ |
|
"epoch": 2.3867085295656723, |
|
"grad_norm": 1.9088976114278464, |
|
"learning_rate": 6.0881353316191385e-06, |
|
"logits/chosen": -9.0, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -612.0, |
|
"loss": 0.994, |
|
"nll_loss": 0.8125, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 3.53125, |
|
"rewards/rejected": -6.125, |
|
"step": 2280 |
|
}, |
|
{ |
|
"epoch": 2.3971742543171115, |
|
"grad_norm": 1.7282521802417226, |
|
"learning_rate": 5.890293687580381e-06, |
|
"logits/chosen": -9.0, |
|
"logits/rejected": -7.0, |
|
"logps/chosen": -247.0, |
|
"logps/rejected": -552.0, |
|
"loss": 0.9935, |
|
"nll_loss": 0.82421875, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.46875, |
|
"rewards/margins": 3.03125, |
|
"rewards/rejected": -5.5, |
|
"step": 2290 |
|
}, |
|
{ |
|
"epoch": 2.4076399790685503, |
|
"grad_norm": 2.455452248455117, |
|
"learning_rate": 5.695289848593532e-06, |
|
"logits/chosen": -8.9375, |
|
"logits/rejected": -7.21875, |
|
"logps/chosen": -226.0, |
|
"logps/rejected": -524.0, |
|
"loss": 0.9844, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -2.265625, |
|
"rewards/margins": 2.984375, |
|
"rewards/rejected": -5.25, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 2.4076399790685503, |
|
"eval_logits/chosen": -8.375, |
|
"eval_logits/rejected": -7.125, |
|
"eval_logps/chosen": -368.0, |
|
"eval_logps/rejected": -452.0, |
|
"eval_loss": 2.190500020980835, |
|
"eval_nll_loss": 1.1875, |
|
"eval_rewards/accuracies": 0.599609375, |
|
"eval_rewards/chosen": -3.6875, |
|
"eval_rewards/margins": 0.84375, |
|
"eval_rewards/rejected": -4.53125, |
|
"eval_runtime": 23.5771, |
|
"eval_samples_per_second": 84.828, |
|
"eval_steps_per_second": 1.357, |
|
"step": 2300 |
|
}, |
|
{ |
|
"epoch": 2.4181057038199896, |
|
"grad_norm": 1.8958793499923938, |
|
"learning_rate": 5.50315277286812e-06, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -6.9375, |
|
"logps/chosen": -242.0, |
|
"logps/rejected": -568.0, |
|
"loss": 1.0227, |
|
"nll_loss": 0.82421875, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.421875, |
|
"rewards/margins": 3.265625, |
|
"rewards/rejected": -5.6875, |
|
"step": 2310 |
|
}, |
|
{ |
|
"epoch": 2.4285714285714284, |
|
"grad_norm": 1.4317364206750591, |
|
"learning_rate": 5.313910992897281e-06, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -280.0, |
|
"logps/rejected": -556.0, |
|
"loss": 1.0056, |
|
"nll_loss": 0.84375, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.796875, |
|
"rewards/margins": 2.75, |
|
"rewards/rejected": -5.5625, |
|
"step": 2320 |
|
}, |
|
{ |
|
"epoch": 2.4390371533228676, |
|
"grad_norm": 1.6282433680917814, |
|
"learning_rate": 5.127592611220658e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.21875, |
|
"logps/chosen": -266.0, |
|
"logps/rejected": -572.0, |
|
"loss": 0.9881, |
|
"nll_loss": 0.8125, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.65625, |
|
"rewards/margins": 3.0625, |
|
"rewards/rejected": -5.71875, |
|
"step": 2330 |
|
}, |
|
{ |
|
"epoch": 2.4495028780743064, |
|
"grad_norm": 1.265465952680447, |
|
"learning_rate": 4.944225296251159e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.15625, |
|
"logps/chosen": -300.0, |
|
"logps/rejected": -636.0, |
|
"loss": 0.9801, |
|
"nll_loss": 0.8203125, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -3.0, |
|
"rewards/margins": 3.34375, |
|
"rewards/rejected": -6.34375, |
|
"step": 2340 |
|
}, |
|
{ |
|
"epoch": 2.4599686028257457, |
|
"grad_norm": 1.3390504692222212, |
|
"learning_rate": 4.763836278166172e-06, |
|
"logits/chosen": -8.875, |
|
"logits/rejected": -6.625, |
|
"logps/chosen": -227.0, |
|
"logps/rejected": -532.0, |
|
"loss": 0.953, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 3.0625, |
|
"rewards/rejected": -5.34375, |
|
"step": 2350 |
|
}, |
|
{ |
|
"epoch": 2.470434327577185, |
|
"grad_norm": 1.8476650290059367, |
|
"learning_rate": 4.586452344863876e-06, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -6.75, |
|
"logps/chosen": -215.0, |
|
"logps/rejected": -512.0, |
|
"loss": 0.9705, |
|
"nll_loss": 0.71875, |
|
"rewards/accuracies": 0.887499988079071, |
|
"rewards/chosen": -2.15625, |
|
"rewards/margins": 2.984375, |
|
"rewards/rejected": -5.125, |
|
"step": 2360 |
|
}, |
|
{ |
|
"epoch": 2.4809000523286238, |
|
"grad_norm": 2.0849261842059175, |
|
"learning_rate": 4.412099837985192e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -232.0, |
|
"logps/rejected": -524.0, |
|
"loss": 0.9877, |
|
"nll_loss": 0.7734375, |
|
"rewards/accuracies": 0.8812500238418579, |
|
"rewards/chosen": -2.328125, |
|
"rewards/margins": 2.921875, |
|
"rewards/rejected": -5.25, |
|
"step": 2370 |
|
}, |
|
{ |
|
"epoch": 2.4913657770800626, |
|
"grad_norm": 2.173181750936938, |
|
"learning_rate": 4.240804649002089e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.3125, |
|
"logps/chosen": -227.0, |
|
"logps/rejected": -540.0, |
|
"loss": 0.9672, |
|
"nll_loss": 0.76171875, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.265625, |
|
"rewards/margins": 3.140625, |
|
"rewards/rejected": -5.40625, |
|
"step": 2380 |
|
}, |
|
{ |
|
"epoch": 2.501831501831502, |
|
"grad_norm": 1.7510013361034416, |
|
"learning_rate": 4.072592215372598e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -228.0, |
|
"logps/rejected": -490.0, |
|
"loss": 0.9527, |
|
"nll_loss": 0.75, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 2.609375, |
|
"rewards/rejected": -4.90625, |
|
"step": 2390 |
|
}, |
|
{ |
|
"epoch": 2.512297226582941, |
|
"grad_norm": 2.6531961693168773, |
|
"learning_rate": 3.907487516763389e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -255.0, |
|
"logps/rejected": -584.0, |
|
"loss": 0.9931, |
|
"nll_loss": 0.79296875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.546875, |
|
"rewards/margins": 3.3125, |
|
"rewards/rejected": -5.875, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 2.512297226582941, |
|
"eval_logits/chosen": -8.6875, |
|
"eval_logits/rejected": -7.375, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -442.0, |
|
"eval_loss": 2.184312582015991, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.58203125, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.79296875, |
|
"eval_rewards/rejected": -4.4375, |
|
"eval_runtime": 23.5655, |
|
"eval_samples_per_second": 84.87, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2400 |
|
}, |
|
{ |
|
"epoch": 2.52276295133438, |
|
"grad_norm": 2.0299219272440143, |
|
"learning_rate": 3.745515071340286e-06, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.75, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -556.0, |
|
"loss": 0.9302, |
|
"nll_loss": 0.80859375, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -2.578125, |
|
"rewards/margins": 3.0, |
|
"rewards/rejected": -5.5625, |
|
"step": 2410 |
|
}, |
|
{ |
|
"epoch": 2.5332286760858187, |
|
"grad_norm": 1.9326405551455472, |
|
"learning_rate": 3.586698932127236e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -229.0, |
|
"logps/rejected": -544.0, |
|
"loss": 1.006, |
|
"nll_loss": 0.828125, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 3.171875, |
|
"rewards/rejected": -5.4375, |
|
"step": 2420 |
|
}, |
|
{ |
|
"epoch": 2.543694400837258, |
|
"grad_norm": 3.333119903652477, |
|
"learning_rate": 3.4310626834344744e-06, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.46875, |
|
"logps/chosen": -262.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9761, |
|
"nll_loss": 0.828125, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.609375, |
|
"rewards/margins": 3.015625, |
|
"rewards/rejected": -5.625, |
|
"step": 2430 |
|
}, |
|
{ |
|
"epoch": 2.554160125588697, |
|
"grad_norm": 3.725914210224768, |
|
"learning_rate": 3.2786294373562343e-06, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.125, |
|
"logps/chosen": -224.0, |
|
"logps/rejected": -552.0, |
|
"loss": 0.9854, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.234375, |
|
"rewards/margins": 3.296875, |
|
"rewards/rejected": -5.53125, |
|
"step": 2440 |
|
}, |
|
{ |
|
"epoch": 2.564625850340136, |
|
"grad_norm": 1.4302989803043953, |
|
"learning_rate": 3.1294218303385526e-06, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.46875, |
|
"logps/chosen": -253.0, |
|
"logps/rejected": -580.0, |
|
"loss": 0.9586, |
|
"nll_loss": 0.7734375, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -2.53125, |
|
"rewards/margins": 3.28125, |
|
"rewards/rejected": -5.8125, |
|
"step": 2450 |
|
}, |
|
{ |
|
"epoch": 2.575091575091575, |
|
"grad_norm": 3.4148025406733065, |
|
"learning_rate": 2.9834620198177483e-06, |
|
"logits/chosen": -9.75, |
|
"logits/rejected": -7.78125, |
|
"logps/chosen": -268.0, |
|
"logps/rejected": -576.0, |
|
"loss": 0.9619, |
|
"nll_loss": 0.8046875, |
|
"rewards/accuracies": 0.9125000238418579, |
|
"rewards/chosen": -2.6875, |
|
"rewards/margins": 3.09375, |
|
"rewards/rejected": -5.78125, |
|
"step": 2460 |
|
}, |
|
{ |
|
"epoch": 2.585557299843014, |
|
"grad_norm": 1.7486153609182498, |
|
"learning_rate": 2.840771680930068e-06, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -7.90625, |
|
"logps/chosen": -211.0, |
|
"logps/rejected": -492.0, |
|
"loss": 0.9385, |
|
"nll_loss": 0.71484375, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.109375, |
|
"rewards/margins": 2.8125, |
|
"rewards/rejected": -4.90625, |
|
"step": 2470 |
|
}, |
|
{ |
|
"epoch": 2.5960230245944533, |
|
"grad_norm": 1.8924048948906602, |
|
"learning_rate": 2.701372003292868e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -251.0, |
|
"logps/rejected": -596.0, |
|
"loss": 0.9827, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.5, |
|
"rewards/margins": 3.4375, |
|
"rewards/rejected": -5.9375, |
|
"step": 2480 |
|
}, |
|
{ |
|
"epoch": 2.606488749345892, |
|
"grad_norm": 1.5058588116107823, |
|
"learning_rate": 2.56528368785795e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.3125, |
|
"logps/chosen": -251.0, |
|
"logps/rejected": -584.0, |
|
"loss": 0.9627, |
|
"nll_loss": 0.828125, |
|
"rewards/accuracies": 0.9624999761581421, |
|
"rewards/chosen": -2.515625, |
|
"rewards/margins": 3.3125, |
|
"rewards/rejected": -5.8125, |
|
"step": 2490 |
|
}, |
|
{ |
|
"epoch": 2.6169544740973314, |
|
"grad_norm": 1.4486979191450826, |
|
"learning_rate": 2.432526943837471e-06, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -237.0, |
|
"logps/rejected": -520.0, |
|
"loss": 0.9537, |
|
"nll_loss": 0.8203125, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.375, |
|
"rewards/margins": 2.828125, |
|
"rewards/rejected": -5.1875, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 2.6169544740973314, |
|
"eval_logits/chosen": -8.8125, |
|
"eval_logits/rejected": -7.5, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -446.0, |
|
"eval_loss": 2.190687417984009, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.58984375, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.8125, |
|
"eval_rewards/rejected": -4.46875, |
|
"eval_runtime": 23.5674, |
|
"eval_samples_per_second": 84.863, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2500 |
|
}, |
|
{ |
|
"epoch": 2.62742019884877, |
|
"grad_norm": 2.007366410161126, |
|
"learning_rate": 2.3031214857028446e-06, |
|
"logits/chosen": -9.75, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -254.0, |
|
"logps/rejected": -620.0, |
|
"loss": 0.9736, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.546875, |
|
"rewards/margins": 3.65625, |
|
"rewards/rejected": -6.1875, |
|
"step": 2510 |
|
}, |
|
{ |
|
"epoch": 2.6378859236002095, |
|
"grad_norm": 1.67297032434091, |
|
"learning_rate": 2.177086530257139e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -237.0, |
|
"logps/rejected": -540.0, |
|
"loss": 0.9504, |
|
"nll_loss": 0.80078125, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.375, |
|
"rewards/margins": 3.03125, |
|
"rewards/rejected": -5.40625, |
|
"step": 2520 |
|
}, |
|
{ |
|
"epoch": 2.6483516483516483, |
|
"grad_norm": 1.2870312678460598, |
|
"learning_rate": 2.054440793781354e-06, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.71875, |
|
"logps/chosen": -242.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9711, |
|
"nll_loss": 0.78515625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.421875, |
|
"rewards/margins": 3.203125, |
|
"rewards/rejected": -5.625, |
|
"step": 2530 |
|
}, |
|
{ |
|
"epoch": 2.6588173731030875, |
|
"grad_norm": 1.8893081798389797, |
|
"learning_rate": 1.9352024892550587e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.03125, |
|
"logps/chosen": -249.0, |
|
"logps/rejected": -584.0, |
|
"loss": 0.995, |
|
"nll_loss": 0.83203125, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.484375, |
|
"rewards/margins": 3.34375, |
|
"rewards/rejected": -5.84375, |
|
"step": 2540 |
|
}, |
|
{ |
|
"epoch": 2.6692830978545263, |
|
"grad_norm": 1.8266859138229021, |
|
"learning_rate": 1.8193893236517379e-06, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.59375, |
|
"logps/chosen": -249.0, |
|
"logps/rejected": -568.0, |
|
"loss": 1.0, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.484375, |
|
"rewards/margins": 3.171875, |
|
"rewards/rejected": -5.6875, |
|
"step": 2550 |
|
}, |
|
{ |
|
"epoch": 2.6797488226059656, |
|
"grad_norm": 2.843139545908333, |
|
"learning_rate": 1.7070184953092956e-06, |
|
"logits/chosen": -9.75, |
|
"logits/rejected": -7.3125, |
|
"logps/chosen": -247.0, |
|
"logps/rejected": -572.0, |
|
"loss": 0.9418, |
|
"nll_loss": 0.79296875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.46875, |
|
"rewards/margins": 3.25, |
|
"rewards/rejected": -5.71875, |
|
"step": 2560 |
|
}, |
|
{ |
|
"epoch": 2.6902145473574044, |
|
"grad_norm": 4.297655440200891, |
|
"learning_rate": 1.5981066913760918e-06, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -548.0, |
|
"loss": 0.9684, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 2.890625, |
|
"rewards/rejected": -5.46875, |
|
"step": 2570 |
|
}, |
|
{ |
|
"epoch": 2.7006802721088436, |
|
"grad_norm": 2.290745454892461, |
|
"learning_rate": 1.492670085332884e-06, |
|
"logits/chosen": -9.25, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -230.0, |
|
"logps/rejected": -560.0, |
|
"loss": 0.9601, |
|
"nll_loss": 0.78125, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.296875, |
|
"rewards/margins": 3.28125, |
|
"rewards/rejected": -5.59375, |
|
"step": 2580 |
|
}, |
|
{ |
|
"epoch": 2.7111459968602825, |
|
"grad_norm": 2.624151747245694, |
|
"learning_rate": 1.3907243345910787e-06, |
|
"logits/chosen": -9.0, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -223.0, |
|
"logps/rejected": -536.0, |
|
"loss": 0.9773, |
|
"nll_loss": 0.77734375, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -2.234375, |
|
"rewards/margins": 3.125, |
|
"rewards/rejected": -5.34375, |
|
"step": 2590 |
|
}, |
|
{ |
|
"epoch": 2.7216117216117217, |
|
"grad_norm": 1.921708508480364, |
|
"learning_rate": 1.292284578167538e-06, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.5, |
|
"logps/chosen": -242.0, |
|
"logps/rejected": -544.0, |
|
"loss": 0.9512, |
|
"nll_loss": 0.77734375, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.421875, |
|
"rewards/margins": 3.046875, |
|
"rewards/rejected": -5.46875, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 2.7216117216117217, |
|
"eval_logits/chosen": -8.8125, |
|
"eval_logits/rejected": -7.5, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -446.0, |
|
"eval_loss": 2.1917500495910645, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.58984375, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.80859375, |
|
"eval_rewards/rejected": -4.4375, |
|
"eval_runtime": 23.5643, |
|
"eval_samples_per_second": 84.874, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2600 |
|
}, |
|
{ |
|
"epoch": 2.7320774463631605, |
|
"grad_norm": 1.7052432262226271, |
|
"learning_rate": 1.1973654344364925e-06, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -234.0, |
|
"logps/rejected": -516.0, |
|
"loss": 0.967, |
|
"nll_loss": 0.78125, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.34375, |
|
"rewards/margins": 2.828125, |
|
"rewards/rejected": -5.1875, |
|
"step": 2610 |
|
}, |
|
{ |
|
"epoch": 2.7425431711145998, |
|
"grad_norm": 2.9916005698559407, |
|
"learning_rate": 1.1059809989586583e-06, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.03125, |
|
"logps/chosen": -219.0, |
|
"logps/rejected": -536.0, |
|
"loss": 0.9634, |
|
"nll_loss": 0.78125, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.1875, |
|
"rewards/margins": 3.1875, |
|
"rewards/rejected": -5.375, |
|
"step": 2620 |
|
}, |
|
{ |
|
"epoch": 2.7530088958660386, |
|
"grad_norm": 6.753598980106263, |
|
"learning_rate": 1.018144842388044e-06, |
|
"logits/chosen": -9.125, |
|
"logits/rejected": -7.40625, |
|
"logps/chosen": -227.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9894, |
|
"nll_loss": 0.84765625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 3.359375, |
|
"rewards/rejected": -5.625, |
|
"step": 2630 |
|
}, |
|
{ |
|
"epoch": 2.763474620617478, |
|
"grad_norm": 2.9547716668052617, |
|
"learning_rate": 9.338700084567109e-07, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -6.875, |
|
"logps/chosen": -264.0, |
|
"logps/rejected": -568.0, |
|
"loss": 1.0164, |
|
"nll_loss": 0.8046875, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.640625, |
|
"rewards/margins": 3.046875, |
|
"rewards/rejected": -5.6875, |
|
"step": 2640 |
|
}, |
|
{ |
|
"epoch": 2.7739403453689166, |
|
"grad_norm": 2.2963351395247837, |
|
"learning_rate": 8.531690120377606e-07, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.1875, |
|
"logps/chosen": -222.0, |
|
"logps/rejected": -520.0, |
|
"loss": 0.954, |
|
"nll_loss": 0.765625, |
|
"rewards/accuracies": 0.9375, |
|
"rewards/chosen": -2.21875, |
|
"rewards/margins": 2.984375, |
|
"rewards/rejected": -5.1875, |
|
"step": 2650 |
|
}, |
|
{ |
|
"epoch": 2.784406070120356, |
|
"grad_norm": 1.3648434740757005, |
|
"learning_rate": 7.760538372868636e-07, |
|
"logits/chosen": -9.8125, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -242.0, |
|
"logps/rejected": -528.0, |
|
"loss": 0.9882, |
|
"nll_loss": 0.84375, |
|
"rewards/accuracies": 0.893750011920929, |
|
"rewards/chosen": -2.421875, |
|
"rewards/margins": 2.875, |
|
"rewards/rejected": -5.3125, |
|
"step": 2660 |
|
}, |
|
{ |
|
"epoch": 2.7948717948717947, |
|
"grad_norm": 1.836834554810095, |
|
"learning_rate": 7.025359358626166e-07, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -6.8125, |
|
"logps/chosen": -221.0, |
|
"logps/rejected": -516.0, |
|
"loss": 0.9603, |
|
"nll_loss": 0.76171875, |
|
"rewards/accuracies": 0.90625, |
|
"rewards/chosen": -2.21875, |
|
"rewards/margins": 2.9375, |
|
"rewards/rejected": -5.15625, |
|
"step": 2670 |
|
}, |
|
{ |
|
"epoch": 2.805337519623234, |
|
"grad_norm": 1.846343687308603, |
|
"learning_rate": 6.326262252259463e-07, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -7.21875, |
|
"logps/chosen": -264.0, |
|
"logps/rejected": -612.0, |
|
"loss": 0.9522, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.640625, |
|
"rewards/margins": 3.46875, |
|
"rewards/rejected": -6.09375, |
|
"step": 2680 |
|
}, |
|
{ |
|
"epoch": 2.815803244374673, |
|
"grad_norm": 3.0299112642248436, |
|
"learning_rate": 5.66335087018871e-07, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.0, |
|
"logps/chosen": -237.0, |
|
"logps/rejected": -576.0, |
|
"loss": 0.9835, |
|
"nll_loss": 0.81640625, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.375, |
|
"rewards/margins": 3.390625, |
|
"rewards/rejected": -5.75, |
|
"step": 2690 |
|
}, |
|
{ |
|
"epoch": 2.826268969126112, |
|
"grad_norm": 2.5250038352996986, |
|
"learning_rate": 5.036723655228226e-07, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.125, |
|
"logps/chosen": -240.0, |
|
"logps/rejected": -528.0, |
|
"loss": 0.9604, |
|
"nll_loss": 0.76953125, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.40625, |
|
"rewards/margins": 2.859375, |
|
"rewards/rejected": -5.28125, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 2.826268969126112, |
|
"eval_logits/chosen": -8.875, |
|
"eval_logits/rejected": -7.53125, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -442.0, |
|
"eval_loss": 2.190624952316284, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.587890625, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.796875, |
|
"eval_rewards/rejected": -4.4375, |
|
"eval_runtime": 23.5671, |
|
"eval_samples_per_second": 84.864, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2700 |
|
}, |
|
{ |
|
"epoch": 2.836734693877551, |
|
"grad_norm": 2.850315673194346, |
|
"learning_rate": 4.446473661967432e-07, |
|
"logits/chosen": -9.625, |
|
"logits/rejected": -7.4375, |
|
"logps/chosen": -251.0, |
|
"logps/rejected": -608.0, |
|
"loss": 0.9754, |
|
"nll_loss": 0.7890625, |
|
"rewards/accuracies": 0.96875, |
|
"rewards/chosen": -2.515625, |
|
"rewards/margins": 3.5625, |
|
"rewards/rejected": -6.0625, |
|
"step": 2710 |
|
}, |
|
{ |
|
"epoch": 2.84720041862899, |
|
"grad_norm": 1.531495269289477, |
|
"learning_rate": 3.892688542952411e-07, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.34375, |
|
"logps/chosen": -234.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9458, |
|
"nll_loss": 0.78515625, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.34375, |
|
"rewards/margins": 3.28125, |
|
"rewards/rejected": -5.625, |
|
"step": 2720 |
|
}, |
|
{ |
|
"epoch": 2.8576661433804293, |
|
"grad_norm": 1.3693207582644338, |
|
"learning_rate": 3.3754505356694e-07, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -227.0, |
|
"logps/rejected": -568.0, |
|
"loss": 0.9583, |
|
"nll_loss": 0.7578125, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.265625, |
|
"rewards/margins": 3.40625, |
|
"rewards/rejected": -5.6875, |
|
"step": 2730 |
|
}, |
|
{ |
|
"epoch": 2.868131868131868, |
|
"grad_norm": 1.675144890607875, |
|
"learning_rate": 2.8948364503322276e-07, |
|
"logits/chosen": -9.4375, |
|
"logits/rejected": -8.0, |
|
"logps/chosen": -235.0, |
|
"logps/rejected": -544.0, |
|
"loss": 0.9619, |
|
"nll_loss": 0.7734375, |
|
"rewards/accuracies": 0.9312499761581421, |
|
"rewards/chosen": -2.359375, |
|
"rewards/margins": 3.0625, |
|
"rewards/rejected": -5.40625, |
|
"step": 2740 |
|
}, |
|
{ |
|
"epoch": 2.878597592883307, |
|
"grad_norm": 1.585674672847395, |
|
"learning_rate": 2.4509176584762706e-07, |
|
"logits/chosen": -9.5625, |
|
"logits/rejected": -7.15625, |
|
"logps/chosen": -231.0, |
|
"logps/rejected": -544.0, |
|
"loss": 0.9407, |
|
"nll_loss": 0.76171875, |
|
"rewards/accuracies": 0.8999999761581421, |
|
"rewards/chosen": -2.3125, |
|
"rewards/margins": 3.125, |
|
"rewards/rejected": -5.4375, |
|
"step": 2750 |
|
}, |
|
{ |
|
"epoch": 2.889063317634746, |
|
"grad_norm": 2.161724626327028, |
|
"learning_rate": 2.0437600823595694e-07, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -233.0, |
|
"logps/rejected": -560.0, |
|
"loss": 0.9692, |
|
"nll_loss": 0.78515625, |
|
"rewards/accuracies": 0.9750000238418579, |
|
"rewards/chosen": -2.328125, |
|
"rewards/margins": 3.265625, |
|
"rewards/rejected": -5.59375, |
|
"step": 2760 |
|
}, |
|
{ |
|
"epoch": 2.8995290423861855, |
|
"grad_norm": 1.0853280454236818, |
|
"learning_rate": 1.6734241851733e-07, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.03125, |
|
"logps/chosen": -260.0, |
|
"logps/rejected": -596.0, |
|
"loss": 0.9218, |
|
"nll_loss": 0.80859375, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 3.375, |
|
"rewards/rejected": -5.96875, |
|
"step": 2770 |
|
}, |
|
{ |
|
"epoch": 2.9099947671376243, |
|
"grad_norm": 2.026975587231853, |
|
"learning_rate": 1.339964962062984e-07, |
|
"logits/chosen": -9.875, |
|
"logits/rejected": -7.25, |
|
"logps/chosen": -264.0, |
|
"logps/rejected": -600.0, |
|
"loss": 0.9305, |
|
"nll_loss": 0.76171875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.640625, |
|
"rewards/margins": 3.375, |
|
"rewards/rejected": -6.03125, |
|
"step": 2780 |
|
}, |
|
{ |
|
"epoch": 2.920460491889063, |
|
"grad_norm": 1.3351914211318932, |
|
"learning_rate": 1.0434319319617136e-07, |
|
"logits/chosen": -9.75, |
|
"logits/rejected": -7.625, |
|
"logps/chosen": -236.0, |
|
"logps/rejected": -588.0, |
|
"loss": 0.9882, |
|
"nll_loss": 0.73046875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.359375, |
|
"rewards/margins": 3.5, |
|
"rewards/rejected": -5.875, |
|
"step": 2790 |
|
}, |
|
{ |
|
"epoch": 2.9309262166405023, |
|
"grad_norm": 1.5322561702396564, |
|
"learning_rate": 7.838691302365363e-08, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.65625, |
|
"logps/chosen": -225.0, |
|
"logps/rejected": -512.0, |
|
"loss": 1.0208, |
|
"nll_loss": 0.79296875, |
|
"rewards/accuracies": 0.90625, |
|
"rewards/chosen": -2.25, |
|
"rewards/margins": 2.890625, |
|
"rewards/rejected": -5.125, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 2.9309262166405023, |
|
"eval_logits/chosen": -8.875, |
|
"eval_logits/rejected": -7.53125, |
|
"eval_logps/chosen": -364.0, |
|
"eval_logps/rejected": -444.0, |
|
"eval_loss": 2.1904375553131104, |
|
"eval_nll_loss": 1.171875, |
|
"eval_rewards/accuracies": 0.591796875, |
|
"eval_rewards/chosen": -3.640625, |
|
"eval_rewards/margins": 0.80078125, |
|
"eval_rewards/rejected": -4.4375, |
|
"eval_runtime": 23.5676, |
|
"eval_samples_per_second": 84.862, |
|
"eval_steps_per_second": 1.358, |
|
"step": 2800 |
|
}, |
|
{ |
|
"epoch": 2.9413919413919416, |
|
"grad_norm": 2.1982387609735423, |
|
"learning_rate": 5.613151021490459e-08, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -242.0, |
|
"logps/rejected": -564.0, |
|
"loss": 1.0005, |
|
"nll_loss": 0.7734375, |
|
"rewards/accuracies": 0.949999988079071, |
|
"rewards/chosen": -2.421875, |
|
"rewards/margins": 3.234375, |
|
"rewards/rejected": -5.65625, |
|
"step": 2810 |
|
}, |
|
{ |
|
"epoch": 2.9518576661433804, |
|
"grad_norm": 1.550171435395584, |
|
"learning_rate": 3.7580289713159965e-08, |
|
"logits/chosen": -9.5, |
|
"logits/rejected": -7.0625, |
|
"logps/chosen": -225.0, |
|
"logps/rejected": -524.0, |
|
"loss": 0.9549, |
|
"nll_loss": 0.76953125, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.25, |
|
"rewards/margins": 3.0, |
|
"rewards/rejected": -5.25, |
|
"step": 2820 |
|
}, |
|
{ |
|
"epoch": 2.962323390894819, |
|
"grad_norm": 1.5110176516139098, |
|
"learning_rate": 2.2736006387927277e-08, |
|
"logits/chosen": -9.3125, |
|
"logits/rejected": -7.375, |
|
"logps/chosen": -245.0, |
|
"logps/rejected": -560.0, |
|
"loss": 0.9394, |
|
"nll_loss": 0.75, |
|
"rewards/accuracies": 0.918749988079071, |
|
"rewards/chosen": -2.453125, |
|
"rewards/margins": 3.15625, |
|
"rewards/rejected": -5.59375, |
|
"step": 2830 |
|
}, |
|
{ |
|
"epoch": 2.9727891156462585, |
|
"grad_norm": 1.60814030318752, |
|
"learning_rate": 1.1600864625893604e-08, |
|
"logits/chosen": -9.6875, |
|
"logits/rejected": -7.21875, |
|
"logps/chosen": -244.0, |
|
"logps/rejected": -564.0, |
|
"loss": 0.9568, |
|
"nll_loss": 0.80078125, |
|
"rewards/accuracies": 0.956250011920929, |
|
"rewards/chosen": -2.4375, |
|
"rewards/margins": 3.1875, |
|
"rewards/rejected": -5.625, |
|
"step": 2840 |
|
}, |
|
{ |
|
"epoch": 2.9832548403976977, |
|
"grad_norm": 2.489116908555861, |
|
"learning_rate": 4.176518003573549e-09, |
|
"logits/chosen": -9.375, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -229.0, |
|
"logps/rejected": -540.0, |
|
"loss": 0.9706, |
|
"nll_loss": 0.796875, |
|
"rewards/accuracies": 0.9437500238418579, |
|
"rewards/chosen": -2.28125, |
|
"rewards/margins": 3.125, |
|
"rewards/rejected": -5.40625, |
|
"step": 2850 |
|
}, |
|
{ |
|
"epoch": 2.9937205651491365, |
|
"grad_norm": 5.801989033138983, |
|
"learning_rate": 4.640690417528903e-10, |
|
"logits/chosen": -9.1875, |
|
"logits/rejected": -7.53125, |
|
"logps/chosen": -258.0, |
|
"logps/rejected": -560.0, |
|
"loss": 0.9675, |
|
"nll_loss": 0.85546875, |
|
"rewards/accuracies": 0.925000011920929, |
|
"rewards/chosen": -2.59375, |
|
"rewards/margins": 2.984375, |
|
"rewards/rejected": -5.5625, |
|
"step": 2860 |
|
}, |
|
{ |
|
"epoch": 2.998953427524856, |
|
"step": 2865, |
|
"total_flos": 0.0, |
|
"train_loss": 0.0, |
|
"train_runtime": 0.0505, |
|
"train_samples_per_second": 3630511.85, |
|
"train_steps_per_second": 56713.757 |
|
} |
|
], |
|
"logging_steps": 10, |
|
"max_steps": 2865, |
|
"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": true, |
|
"should_training_stop": true |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 0.0, |
|
"train_batch_size": 8, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|