| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9984301412872841, | |
| "eval_steps": 100, | |
| "global_step": 477, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0, | |
| "learning_rate": 1.0416666666666666e-08, | |
| "logits/chosen": -1.375, | |
| "logits/rejected": -1.21875, | |
| "logps/chosen": -404.0, | |
| "logps/rejected": -374.0, | |
| "loss": 0.6914, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.02, | |
| "learning_rate": 1.0416666666666667e-07, | |
| "logits/chosen": -1.1875, | |
| "logits/rejected": -1.15625, | |
| "logps/chosen": -394.0, | |
| "logps/rejected": -334.0, | |
| "loss": 0.6944, | |
| "rewards/accuracies": 0.34375, | |
| "rewards/chosen": -0.0024261474609375, | |
| "rewards/margins": -0.004364013671875, | |
| "rewards/rejected": 0.001922607421875, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.04, | |
| "learning_rate": 2.0833333333333333e-07, | |
| "logits/chosen": -1.2109375, | |
| "logits/rejected": -1.171875, | |
| "logps/chosen": -412.0, | |
| "logps/rejected": -342.0, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.47187501192092896, | |
| "rewards/chosen": 0.000743865966796875, | |
| "rewards/margins": 0.003692626953125, | |
| "rewards/rejected": -0.0029449462890625, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.06, | |
| "learning_rate": 3.1249999999999997e-07, | |
| "logits/chosen": -1.2109375, | |
| "logits/rejected": -1.2265625, | |
| "logps/chosen": -432.0, | |
| "logps/rejected": -372.0, | |
| "loss": 0.6923, | |
| "rewards/accuracies": 0.45625001192092896, | |
| "rewards/chosen": 0.002197265625, | |
| "rewards/margins": 0.003204345703125, | |
| "rewards/rejected": -0.0009918212890625, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.08, | |
| "learning_rate": 4.1666666666666667e-07, | |
| "logits/chosen": -1.2421875, | |
| "logits/rejected": -1.21875, | |
| "logps/chosen": -444.0, | |
| "logps/rejected": -368.0, | |
| "loss": 0.6933, | |
| "rewards/accuracies": 0.46875, | |
| "rewards/chosen": -0.0019073486328125, | |
| "rewards/margins": 9.357929229736328e-06, | |
| "rewards/rejected": -0.00191497802734375, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.1, | |
| "learning_rate": 4.999731868769026e-07, | |
| "logits/chosen": -1.1953125, | |
| "logits/rejected": -1.15625, | |
| "logps/chosen": -432.0, | |
| "logps/rejected": -384.0, | |
| "loss": 0.6918, | |
| "rewards/accuracies": 0.4906249940395355, | |
| "rewards/chosen": 0.000812530517578125, | |
| "rewards/margins": 0.00439453125, | |
| "rewards/rejected": -0.0035858154296875, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.990353313429303e-07, | |
| "logits/chosen": -1.1484375, | |
| "logits/rejected": -1.078125, | |
| "logps/chosen": -400.0, | |
| "logps/rejected": -308.0, | |
| "loss": 0.6915, | |
| "rewards/accuracies": 0.484375, | |
| "rewards/chosen": 0.0038299560546875, | |
| "rewards/margins": 0.00457763671875, | |
| "rewards/rejected": -0.000759124755859375, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.15, | |
| "learning_rate": 4.967625656594781e-07, | |
| "logits/chosen": -1.203125, | |
| "logits/rejected": -1.203125, | |
| "logps/chosen": -406.0, | |
| "logps/rejected": -336.0, | |
| "loss": 0.6907, | |
| "rewards/accuracies": 0.4625000059604645, | |
| "rewards/chosen": -0.00010776519775390625, | |
| "rewards/margins": 0.002593994140625, | |
| "rewards/rejected": -0.002685546875, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.17, | |
| "learning_rate": 4.93167072587771e-07, | |
| "logits/chosen": -1.2265625, | |
| "logits/rejected": -1.2265625, | |
| "logps/chosen": -416.0, | |
| "logps/rejected": -394.0, | |
| "loss": 0.6907, | |
| "rewards/accuracies": 0.45625001192092896, | |
| "rewards/chosen": -0.000720977783203125, | |
| "rewards/margins": 0.00144195556640625, | |
| "rewards/rejected": -0.002166748046875, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.882681251368548e-07, | |
| "logits/chosen": -1.171875, | |
| "logits/rejected": -1.1796875, | |
| "logps/chosen": -382.0, | |
| "logps/rejected": -324.0, | |
| "loss": 0.6871, | |
| "rewards/accuracies": 0.546875, | |
| "rewards/chosen": 0.0035858154296875, | |
| "rewards/margins": 0.01318359375, | |
| "rewards/rejected": -0.0096435546875, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "learning_rate": 4.820919832540181e-07, | |
| "logits/chosen": -1.28125, | |
| "logits/rejected": -1.171875, | |
| "logps/chosen": -420.0, | |
| "logps/rejected": -372.0, | |
| "loss": 0.6867, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -0.00103759765625, | |
| "rewards/margins": 0.015625, | |
| "rewards/rejected": -0.0167236328125, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.21, | |
| "eval_logits/chosen": -1.2421875, | |
| "eval_logits/rejected": -1.1953125, | |
| "eval_logps/chosen": -424.0, | |
| "eval_logps/rejected": -368.0, | |
| "eval_loss": 0.6852812767028809, | |
| "eval_rewards/accuracies": 0.5555555820465088, | |
| "eval_rewards/chosen": -0.005645751953125, | |
| "eval_rewards/margins": 0.0216064453125, | |
| "eval_rewards/rejected": -0.027099609375, | |
| "eval_runtime": 51.3914, | |
| "eval_samples_per_second": 38.917, | |
| "eval_steps_per_second": 1.226, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.23, | |
| "learning_rate": 4.7467175306295647e-07, | |
| "logits/chosen": -1.234375, | |
| "logits/rejected": -1.21875, | |
| "logps/chosen": -448.0, | |
| "logps/rejected": -366.0, | |
| "loss": 0.6836, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -0.0045166015625, | |
| "rewards/margins": 0.0247802734375, | |
| "rewards/rejected": -0.029296875, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.25, | |
| "learning_rate": 4.6604720940421207e-07, | |
| "logits/chosen": -1.203125, | |
| "logits/rejected": -1.140625, | |
| "logps/chosen": -410.0, | |
| "logps/rejected": -332.0, | |
| "loss": 0.6802, | |
| "rewards/accuracies": 0.578125, | |
| "rewards/chosen": -0.0155029296875, | |
| "rewards/margins": 0.025390625, | |
| "rewards/rejected": -0.040771484375, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.27, | |
| "learning_rate": 4.5626458262912735e-07, | |
| "logits/chosen": -1.265625, | |
| "logits/rejected": -1.2265625, | |
| "logps/chosen": -404.0, | |
| "logps/rejected": -380.0, | |
| "loss": 0.6806, | |
| "rewards/accuracies": 0.5718749761581421, | |
| "rewards/chosen": -0.04638671875, | |
| "rewards/margins": 0.0257568359375, | |
| "rewards/rejected": -0.072265625, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.29, | |
| "learning_rate": 4.453763107901675e-07, | |
| "logits/chosen": -1.265625, | |
| "logits/rejected": -1.2578125, | |
| "logps/chosen": -454.0, | |
| "logps/rejected": -388.0, | |
| "loss": 0.6776, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": -0.0654296875, | |
| "rewards/margins": 0.037841796875, | |
| "rewards/rejected": -0.10302734375, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.31, | |
| "learning_rate": 4.3344075855595097e-07, | |
| "logits/chosen": -1.2421875, | |
| "logits/rejected": -1.234375, | |
| "logps/chosen": -420.0, | |
| "logps/rejected": -356.0, | |
| "loss": 0.6716, | |
| "rewards/accuracies": 0.609375, | |
| "rewards/chosen": -0.09423828125, | |
| "rewards/margins": 0.049560546875, | |
| "rewards/rejected": -0.1435546875, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.33, | |
| "learning_rate": 4.2052190435769554e-07, | |
| "logits/chosen": -1.28125, | |
| "logits/rejected": -1.234375, | |
| "logps/chosen": -422.0, | |
| "logps/rejected": -388.0, | |
| "loss": 0.6672, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -0.12353515625, | |
| "rewards/margins": 0.061767578125, | |
| "rewards/rejected": -0.185546875, | |
| "step": 160 | |
| }, | |
| { | |
| "epoch": 0.36, | |
| "learning_rate": 4.0668899744407567e-07, | |
| "logits/chosen": -1.28125, | |
| "logits/rejected": -1.2265625, | |
| "logps/chosen": -390.0, | |
| "logps/rejected": -336.0, | |
| "loss": 0.674, | |
| "rewards/accuracies": 0.59375, | |
| "rewards/chosen": -0.1630859375, | |
| "rewards/margins": 0.037353515625, | |
| "rewards/rejected": -0.2001953125, | |
| "step": 170 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 3.920161866827889e-07, | |
| "logits/chosen": -1.3125, | |
| "logits/rejected": -1.296875, | |
| "logps/chosen": -426.0, | |
| "logps/rejected": -382.0, | |
| "loss": 0.6659, | |
| "rewards/accuracies": 0.574999988079071, | |
| "rewards/chosen": -0.181640625, | |
| "rewards/margins": 0.06298828125, | |
| "rewards/rejected": -0.244140625, | |
| "step": 180 | |
| }, | |
| { | |
| "epoch": 0.4, | |
| "learning_rate": 3.765821230985757e-07, | |
| "logits/chosen": -1.3203125, | |
| "logits/rejected": -1.265625, | |
| "logps/chosen": -410.0, | |
| "logps/rejected": -380.0, | |
| "loss": 0.6594, | |
| "rewards/accuracies": 0.659375011920929, | |
| "rewards/chosen": -0.19140625, | |
| "rewards/margins": 0.09814453125, | |
| "rewards/rejected": -0.2890625, | |
| "step": 190 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "learning_rate": 3.604695382782159e-07, | |
| "logits/chosen": -1.3671875, | |
| "logits/rejected": -1.265625, | |
| "logps/chosen": -438.0, | |
| "logps/rejected": -374.0, | |
| "loss": 0.6632, | |
| "rewards/accuracies": 0.621874988079071, | |
| "rewards/chosen": -0.21875, | |
| "rewards/margins": 0.07421875, | |
| "rewards/rejected": -0.29296875, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.42, | |
| "eval_logits/chosen": -1.3671875, | |
| "eval_logits/rejected": -1.3203125, | |
| "eval_logps/chosen": -446.0, | |
| "eval_logps/rejected": -396.0, | |
| "eval_loss": 0.6625468730926514, | |
| "eval_rewards/accuracies": 0.625, | |
| "eval_rewards/chosen": -0.228515625, | |
| "eval_rewards/margins": 0.09130859375, | |
| "eval_rewards/rejected": -0.3203125, | |
| "eval_runtime": 52.3612, | |
| "eval_samples_per_second": 38.196, | |
| "eval_steps_per_second": 1.203, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.44, | |
| "learning_rate": 3.4376480090239047e-07, | |
| "logits/chosen": -1.4296875, | |
| "logits/rejected": -1.359375, | |
| "logps/chosen": -442.0, | |
| "logps/rejected": -388.0, | |
| "loss": 0.6604, | |
| "rewards/accuracies": 0.609375, | |
| "rewards/chosen": -0.2421875, | |
| "rewards/margins": 0.0888671875, | |
| "rewards/rejected": -0.33203125, | |
| "step": 210 | |
| }, | |
| { | |
| "epoch": 0.46, | |
| "learning_rate": 3.265574537815398e-07, | |
| "logits/chosen": -1.296875, | |
| "logits/rejected": -1.28125, | |
| "logps/chosen": -406.0, | |
| "logps/rejected": -374.0, | |
| "loss": 0.6588, | |
| "rewards/accuracies": 0.590624988079071, | |
| "rewards/chosen": -0.271484375, | |
| "rewards/margins": 0.0673828125, | |
| "rewards/rejected": -0.33984375, | |
| "step": 220 | |
| }, | |
| { | |
| "epoch": 0.48, | |
| "learning_rate": 3.0893973387735683e-07, | |
| "logits/chosen": -1.34375, | |
| "logits/rejected": -1.296875, | |
| "logps/chosen": -406.0, | |
| "logps/rejected": -372.0, | |
| "loss": 0.6516, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -0.26171875, | |
| "rewards/margins": 0.1064453125, | |
| "rewards/rejected": -0.369140625, | |
| "step": 230 | |
| }, | |
| { | |
| "epoch": 0.5, | |
| "learning_rate": 2.910060778827554e-07, | |
| "logits/chosen": -1.3515625, | |
| "logits/rejected": -1.328125, | |
| "logps/chosen": -420.0, | |
| "logps/rejected": -368.0, | |
| "loss": 0.6635, | |
| "rewards/accuracies": 0.609375, | |
| "rewards/chosen": -0.302734375, | |
| "rewards/margins": 0.09619140625, | |
| "rewards/rejected": -0.400390625, | |
| "step": 240 | |
| }, | |
| { | |
| "epoch": 0.52, | |
| "learning_rate": 2.7285261601056697e-07, | |
| "logits/chosen": -1.3828125, | |
| "logits/rejected": -1.328125, | |
| "logps/chosen": -426.0, | |
| "logps/rejected": -384.0, | |
| "loss": 0.6546, | |
| "rewards/accuracies": 0.550000011920929, | |
| "rewards/chosen": -0.30859375, | |
| "rewards/margins": 0.10693359375, | |
| "rewards/rejected": -0.416015625, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.54, | |
| "learning_rate": 2.5457665670441937e-07, | |
| "logits/chosen": -1.3515625, | |
| "logits/rejected": -1.2890625, | |
| "logps/chosen": -432.0, | |
| "logps/rejected": -384.0, | |
| "loss": 0.6521, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -0.337890625, | |
| "rewards/margins": 0.142578125, | |
| "rewards/rejected": -0.48046875, | |
| "step": 260 | |
| }, | |
| { | |
| "epoch": 0.57, | |
| "learning_rate": 2.3627616503391812e-07, | |
| "logits/chosen": -1.3828125, | |
| "logits/rejected": -1.390625, | |
| "logps/chosen": -454.0, | |
| "logps/rejected": -426.0, | |
| "loss": 0.6522, | |
| "rewards/accuracies": 0.609375, | |
| "rewards/chosen": -0.34375, | |
| "rewards/margins": 0.11181640625, | |
| "rewards/rejected": -0.45703125, | |
| "step": 270 | |
| }, | |
| { | |
| "epoch": 0.59, | |
| "learning_rate": 2.1804923757009882e-07, | |
| "logits/chosen": -1.296875, | |
| "logits/rejected": -1.3046875, | |
| "logps/chosen": -426.0, | |
| "logps/rejected": -372.0, | |
| "loss": 0.6545, | |
| "rewards/accuracies": 0.606249988079071, | |
| "rewards/chosen": -0.35546875, | |
| "rewards/margins": 0.134765625, | |
| "rewards/rejected": -0.48828125, | |
| "step": 280 | |
| }, | |
| { | |
| "epoch": 0.61, | |
| "learning_rate": 1.9999357655598891e-07, | |
| "logits/chosen": -1.3515625, | |
| "logits/rejected": -1.3671875, | |
| "logps/chosen": -436.0, | |
| "logps/rejected": -388.0, | |
| "loss": 0.6546, | |
| "rewards/accuracies": 0.6000000238418579, | |
| "rewards/chosen": -0.37890625, | |
| "rewards/margins": 0.109375, | |
| "rewards/rejected": -0.48828125, | |
| "step": 290 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "learning_rate": 1.8220596619089573e-07, | |
| "logits/chosen": -1.375, | |
| "logits/rejected": -1.3046875, | |
| "logps/chosen": -474.0, | |
| "logps/rejected": -442.0, | |
| "loss": 0.6493, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -0.33984375, | |
| "rewards/margins": 0.12109375, | |
| "rewards/rejected": -0.458984375, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.63, | |
| "eval_logits/chosen": -1.3984375, | |
| "eval_logits/rejected": -1.3515625, | |
| "eval_logps/chosen": -460.0, | |
| "eval_logps/rejected": -412.0, | |
| "eval_loss": 0.653249979019165, | |
| "eval_rewards/accuracies": 0.625, | |
| "eval_rewards/chosen": -0.35546875, | |
| "eval_rewards/margins": 0.134765625, | |
| "eval_rewards/rejected": -0.490234375, | |
| "eval_runtime": 52.3802, | |
| "eval_samples_per_second": 38.182, | |
| "eval_steps_per_second": 1.203, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.65, | |
| "learning_rate": 1.647817538357072e-07, | |
| "logits/chosen": -1.390625, | |
| "logits/rejected": -1.3203125, | |
| "logps/chosen": -432.0, | |
| "logps/rejected": -366.0, | |
| "loss": 0.6467, | |
| "rewards/accuracies": 0.637499988079071, | |
| "rewards/chosen": -0.353515625, | |
| "rewards/margins": 0.125, | |
| "rewards/rejected": -0.478515625, | |
| "step": 310 | |
| }, | |
| { | |
| "epoch": 0.67, | |
| "learning_rate": 1.478143389201113e-07, | |
| "logits/chosen": -1.296875, | |
| "logits/rejected": -1.328125, | |
| "logps/chosen": -416.0, | |
| "logps/rejected": -386.0, | |
| "loss": 0.6499, | |
| "rewards/accuracies": 0.6343749761581421, | |
| "rewards/chosen": -0.3515625, | |
| "rewards/margins": 0.11865234375, | |
| "rewards/rejected": -0.47265625, | |
| "step": 320 | |
| }, | |
| { | |
| "epoch": 0.69, | |
| "learning_rate": 1.3139467229135998e-07, | |
| "logits/chosen": -1.4296875, | |
| "logits/rejected": -1.390625, | |
| "logps/chosen": -476.0, | |
| "logps/rejected": -422.0, | |
| "loss": 0.6409, | |
| "rewards/accuracies": 0.6343749761581421, | |
| "rewards/chosen": -0.373046875, | |
| "rewards/margins": 0.1669921875, | |
| "rewards/rejected": -0.5390625, | |
| "step": 330 | |
| }, | |
| { | |
| "epoch": 0.71, | |
| "learning_rate": 1.1561076868822755e-07, | |
| "logits/chosen": -1.3671875, | |
| "logits/rejected": -1.328125, | |
| "logps/chosen": -428.0, | |
| "logps/rejected": -404.0, | |
| "loss": 0.6505, | |
| "rewards/accuracies": 0.59375, | |
| "rewards/chosen": -0.384765625, | |
| "rewards/margins": 0.10400390625, | |
| "rewards/rejected": -0.48828125, | |
| "step": 340 | |
| }, | |
| { | |
| "epoch": 0.73, | |
| "learning_rate": 1.0054723495346482e-07, | |
| "logits/chosen": -1.359375, | |
| "logits/rejected": -1.2890625, | |
| "logps/chosen": -420.0, | |
| "logps/rejected": -388.0, | |
| "loss": 0.6417, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -0.37890625, | |
| "rewards/margins": 0.1455078125, | |
| "rewards/rejected": -0.5234375, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.75, | |
| "learning_rate": 8.628481651367875e-08, | |
| "logits/chosen": -1.3828125, | |
| "logits/rejected": -1.3515625, | |
| "logps/chosen": -476.0, | |
| "logps/rejected": -400.0, | |
| "loss": 0.6398, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -0.37890625, | |
| "rewards/margins": 0.1396484375, | |
| "rewards/rejected": -0.51953125, | |
| "step": 360 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "learning_rate": 7.289996455765748e-08, | |
| "logits/chosen": -1.421875, | |
| "logits/rejected": -1.3984375, | |
| "logps/chosen": -492.0, | |
| "logps/rejected": -404.0, | |
| "loss": 0.6472, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": -0.376953125, | |
| "rewards/margins": 0.138671875, | |
| "rewards/rejected": -0.515625, | |
| "step": 370 | |
| }, | |
| { | |
| "epoch": 0.8, | |
| "learning_rate": 6.046442623320145e-08, | |
| "logits/chosen": -1.4140625, | |
| "logits/rejected": -1.3984375, | |
| "logps/chosen": -448.0, | |
| "logps/rejected": -416.0, | |
| "loss": 0.6386, | |
| "rewards/accuracies": 0.640625, | |
| "rewards/chosen": -0.39453125, | |
| "rewards/margins": 0.126953125, | |
| "rewards/rejected": -0.51953125, | |
| "step": 380 | |
| }, | |
| { | |
| "epoch": 0.82, | |
| "learning_rate": 4.904486005914027e-08, | |
| "logits/chosen": -1.3515625, | |
| "logits/rejected": -1.3515625, | |
| "logps/chosen": -442.0, | |
| "logps/rejected": -416.0, | |
| "loss": 0.6437, | |
| "rewards/accuracies": 0.6312500238418579, | |
| "rewards/chosen": -0.39453125, | |
| "rewards/margins": 0.126953125, | |
| "rewards/rejected": -0.5234375, | |
| "step": 390 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "learning_rate": 3.8702478614051345e-08, | |
| "logits/chosen": -1.375, | |
| "logits/rejected": -1.34375, | |
| "logps/chosen": -492.0, | |
| "logps/rejected": -426.0, | |
| "loss": 0.6462, | |
| "rewards/accuracies": 0.5625, | |
| "rewards/chosen": -0.392578125, | |
| "rewards/margins": 0.1318359375, | |
| "rewards/rejected": -0.5234375, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.84, | |
| "eval_logits/chosen": -1.421875, | |
| "eval_logits/rejected": -1.375, | |
| "eval_logps/chosen": -464.0, | |
| "eval_logps/rejected": -420.0, | |
| "eval_loss": 0.6495468616485596, | |
| "eval_rewards/accuracies": 0.6388888955116272, | |
| "eval_rewards/chosen": -0.400390625, | |
| "eval_rewards/margins": 0.1552734375, | |
| "eval_rewards/rejected": -0.5546875, | |
| "eval_runtime": 51.9691, | |
| "eval_samples_per_second": 38.484, | |
| "eval_steps_per_second": 1.212, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.86, | |
| "learning_rate": 2.9492720416985e-08, | |
| "logits/chosen": -1.328125, | |
| "logits/rejected": -1.3046875, | |
| "logps/chosen": -428.0, | |
| "logps/rejected": -374.0, | |
| "loss": 0.651, | |
| "rewards/accuracies": 0.6343749761581421, | |
| "rewards/chosen": -0.400390625, | |
| "rewards/margins": 0.130859375, | |
| "rewards/rejected": -0.53125, | |
| "step": 410 | |
| }, | |
| { | |
| "epoch": 0.88, | |
| "learning_rate": 2.1464952759020856e-08, | |
| "logits/chosen": -1.40625, | |
| "logits/rejected": -1.375, | |
| "logps/chosen": -432.0, | |
| "logps/rejected": -408.0, | |
| "loss": 0.6437, | |
| "rewards/accuracies": 0.6187499761581421, | |
| "rewards/chosen": -0.396484375, | |
| "rewards/margins": 0.12109375, | |
| "rewards/rejected": -0.51953125, | |
| "step": 420 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.4662207078575684e-08, | |
| "logits/chosen": -1.375, | |
| "logits/rejected": -1.359375, | |
| "logps/chosen": -458.0, | |
| "logps/rejected": -404.0, | |
| "loss": 0.6478, | |
| "rewards/accuracies": 0.643750011920929, | |
| "rewards/chosen": -0.380859375, | |
| "rewards/margins": 0.181640625, | |
| "rewards/rejected": -0.5625, | |
| "step": 430 | |
| }, | |
| { | |
| "epoch": 0.92, | |
| "learning_rate": 9.12094829893642e-09, | |
| "logits/chosen": -1.421875, | |
| "logits/rejected": -1.390625, | |
| "logps/chosen": -420.0, | |
| "logps/rejected": -392.0, | |
| "loss": 0.6468, | |
| "rewards/accuracies": 0.625, | |
| "rewards/chosen": -0.390625, | |
| "rewards/margins": 0.138671875, | |
| "rewards/rejected": -0.52734375, | |
| "step": 440 | |
| }, | |
| { | |
| "epoch": 0.94, | |
| "learning_rate": 4.8708793644441086e-09, | |
| "logits/chosen": -1.4296875, | |
| "logits/rejected": -1.359375, | |
| "logps/chosen": -452.0, | |
| "logps/rejected": -426.0, | |
| "loss": 0.6587, | |
| "rewards/accuracies": 0.5843750238418579, | |
| "rewards/chosen": -0.40234375, | |
| "rewards/margins": 0.08447265625, | |
| "rewards/rejected": -0.48828125, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 1.9347820230782295e-09, | |
| "logits/chosen": -1.359375, | |
| "logits/rejected": -1.3359375, | |
| "logps/chosen": -426.0, | |
| "logps/rejected": -390.0, | |
| "loss": 0.6394, | |
| "rewards/accuracies": 0.65625, | |
| "rewards/chosen": -0.40234375, | |
| "rewards/margins": 0.140625, | |
| "rewards/rejected": -0.54296875, | |
| "step": 460 | |
| }, | |
| { | |
| "epoch": 0.98, | |
| "learning_rate": 3.2839470889836627e-10, | |
| "logits/chosen": -1.421875, | |
| "logits/rejected": -1.375, | |
| "logps/chosen": -470.0, | |
| "logps/rejected": -422.0, | |
| "loss": 0.6303, | |
| "rewards/accuracies": 0.690625011920929, | |
| "rewards/chosen": -0.36328125, | |
| "rewards/margins": 0.19921875, | |
| "rewards/rejected": -0.55859375, | |
| "step": 470 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 477, | |
| "total_flos": 0.0, | |
| "train_loss": 0.6625334221861897, | |
| "train_runtime": 3684.7873, | |
| "train_samples_per_second": 16.591, | |
| "train_steps_per_second": 0.129 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 477, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "total_flos": 0.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |