{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9968652037617555, "eval_steps": 500, "global_step": 159, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 3.125e-08, "logits/chosen": -2.5097975730895996, "logits/rejected": -2.419074058532715, "logps/chosen": -193.5599365234375, "logps/pi_response": -171.87442016601562, "logps/ref_response": -171.87442016601562, "logps/rejected": -224.4684295654297, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.06, "learning_rate": 3.1249999999999997e-07, "logits/chosen": -2.5170841217041016, "logits/rejected": -2.499955415725708, "logps/chosen": -268.5122375488281, "logps/pi_response": -200.62173461914062, "logps/ref_response": -200.7644805908203, "logps/rejected": -275.15576171875, "loss": 0.692, "rewards/accuracies": 0.4513888955116272, "rewards/chosen": 0.0004006973758805543, "rewards/margins": 0.0025936374440789223, "rewards/rejected": -0.0021929400973021984, "step": 10 }, { "epoch": 0.13, "learning_rate": 4.990353313429303e-07, "logits/chosen": -2.5279006958007812, "logits/rejected": -2.48834228515625, "logps/chosen": -280.61029052734375, "logps/pi_response": -218.712158203125, "logps/ref_response": -220.61703491210938, "logps/rejected": -302.9790344238281, "loss": 0.6799, "rewards/accuracies": 0.5874999761581421, "rewards/chosen": 0.0002802034723572433, "rewards/margins": 0.06487339735031128, "rewards/rejected": -0.06459320336580276, "step": 20 }, { "epoch": 0.19, "learning_rate": 4.882681251368548e-07, "logits/chosen": -2.3786978721618652, "logits/rejected": -2.309803009033203, "logps/chosen": -268.9042053222656, "logps/pi_response": -195.0692901611328, "logps/ref_response": -189.48886108398438, "logps/rejected": -271.2804260253906, "loss": 0.6702, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.04177286848425865, "rewards/margins": 0.0969323068857193, "rewards/rejected": -0.13870516419410706, "step": 30 }, { "epoch": 0.25, "learning_rate": 4.6604720940421207e-07, "logits/chosen": -2.210906505584717, "logits/rejected": -2.181436061859131, "logps/chosen": -283.42681884765625, "logps/pi_response": -215.8210906982422, "logps/ref_response": -189.73727416992188, "logps/rejected": -321.3996276855469, "loss": 0.6576, "rewards/accuracies": 0.625, "rewards/chosen": -0.26275861263275146, "rewards/margins": 0.13055923581123352, "rewards/rejected": -0.3933178782463074, "step": 40 }, { "epoch": 0.31, "learning_rate": 4.3344075855595097e-07, "logits/chosen": -2.1676063537597656, "logits/rejected": -2.122574806213379, "logps/chosen": -321.0862121582031, "logps/pi_response": -257.51531982421875, "logps/ref_response": -199.2998504638672, "logps/rejected": -330.80474853515625, "loss": 0.6442, "rewards/accuracies": 0.6000000238418579, "rewards/chosen": -0.47685688734054565, "rewards/margins": 0.16487360000610352, "rewards/rejected": -0.6417304873466492, "step": 50 }, { "epoch": 0.38, "learning_rate": 3.920161866827889e-07, "logits/chosen": -2.14129900932312, "logits/rejected": -2.106346607208252, "logps/chosen": -291.825439453125, "logps/pi_response": -247.85244750976562, "logps/ref_response": -182.53189086914062, "logps/rejected": -323.88409423828125, "loss": 0.6403, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": -0.41668352484703064, "rewards/margins": 0.3306669592857361, "rewards/rejected": -0.7473504543304443, "step": 60 }, { "epoch": 0.44, "learning_rate": 3.4376480090239047e-07, "logits/chosen": -2.1696386337280273, "logits/rejected": -2.1270456314086914, "logps/chosen": -331.155029296875, "logps/pi_response": -269.52734375, "logps/ref_response": -191.21128845214844, "logps/rejected": -335.54974365234375, "loss": 0.6304, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.6724362373352051, "rewards/margins": 0.23178334534168243, "rewards/rejected": -0.9042196273803711, "step": 70 }, { "epoch": 0.5, "learning_rate": 2.910060778827554e-07, "logits/chosen": -2.203618049621582, "logits/rejected": -2.166483163833618, "logps/chosen": -287.53997802734375, "logps/pi_response": -254.5912322998047, "logps/ref_response": -190.11843872070312, "logps/rejected": -332.36590576171875, "loss": 0.6061, "rewards/accuracies": 0.6812499761581421, "rewards/chosen": -0.514179527759552, "rewards/margins": 0.23715217411518097, "rewards/rejected": -0.7513316869735718, "step": 80 }, { "epoch": 0.56, "learning_rate": 2.3627616503391812e-07, "logits/chosen": -2.278022527694702, "logits/rejected": -2.2521653175354004, "logps/chosen": -271.2624206542969, "logps/pi_response": -235.70266723632812, "logps/ref_response": -180.7279815673828, "logps/rejected": -326.7520751953125, "loss": 0.6169, "rewards/accuracies": 0.6499999761581421, "rewards/chosen": -0.3808743357658386, "rewards/margins": 0.2905920445919037, "rewards/rejected": -0.6714664101600647, "step": 90 }, { "epoch": 0.63, "learning_rate": 1.8220596619089573e-07, "logits/chosen": -2.2085721492767334, "logits/rejected": -2.194762706756592, "logps/chosen": -326.04888916015625, "logps/pi_response": -270.80810546875, "logps/ref_response": -195.6220703125, "logps/rejected": -389.5771179199219, "loss": 0.6167, "rewards/accuracies": 0.675000011920929, "rewards/chosen": -0.691897988319397, "rewards/margins": 0.3410559296607971, "rewards/rejected": -1.0329539775848389, "step": 100 }, { "epoch": 0.69, "learning_rate": 1.3139467229135998e-07, "logits/chosen": -2.2901813983917236, "logits/rejected": -2.2825167179107666, "logps/chosen": -300.0601806640625, "logps/pi_response": -242.42221069335938, "logps/ref_response": -179.70770263671875, "logps/rejected": -329.3973083496094, "loss": 0.6131, "rewards/accuracies": 0.6312500238418579, "rewards/chosen": -0.5147222280502319, "rewards/margins": 0.23834428191184998, "rewards/rejected": -0.7530665993690491, "step": 110 }, { "epoch": 0.75, "learning_rate": 8.628481651367875e-08, "logits/chosen": -2.2861225605010986, "logits/rejected": -2.2135860919952393, "logps/chosen": -316.3228454589844, "logps/pi_response": -282.2044982910156, "logps/ref_response": -202.98269653320312, "logps/rejected": -386.8232421875, "loss": 0.5922, "rewards/accuracies": 0.6875, "rewards/chosen": -0.6165128946304321, "rewards/margins": 0.42569026350975037, "rewards/rejected": -1.0422031879425049, "step": 120 }, { "epoch": 0.82, "learning_rate": 4.904486005914027e-08, "logits/chosen": -2.304694890975952, "logits/rejected": -2.2709462642669678, "logps/chosen": -332.1279602050781, "logps/pi_response": -276.42706298828125, "logps/ref_response": -195.01205444335938, "logps/rejected": -367.41204833984375, "loss": 0.6018, "rewards/accuracies": 0.6625000238418579, "rewards/chosen": -0.6682323217391968, "rewards/margins": 0.37029170989990234, "rewards/rejected": -1.0385239124298096, "step": 130 }, { "epoch": 0.88, "learning_rate": 2.1464952759020856e-08, "logits/chosen": -2.3100249767303467, "logits/rejected": -2.246779680252075, "logps/chosen": -328.9765930175781, "logps/pi_response": -279.7056579589844, "logps/ref_response": -195.75735473632812, "logps/rejected": -392.63818359375, "loss": 0.584, "rewards/accuracies": 0.6187499761581421, "rewards/chosen": -0.7241796255111694, "rewards/margins": 0.34663844108581543, "rewards/rejected": -1.0708180665969849, "step": 140 }, { "epoch": 0.94, "learning_rate": 4.8708793644441086e-09, "logits/chosen": -2.2483038902282715, "logits/rejected": -2.229506254196167, "logps/chosen": -307.9349060058594, "logps/pi_response": -257.32733154296875, "logps/ref_response": -179.74029541015625, "logps/rejected": -353.04681396484375, "loss": 0.5882, "rewards/accuracies": 0.65625, "rewards/chosen": -0.6363255381584167, "rewards/margins": 0.36607229709625244, "rewards/rejected": -1.002397894859314, "step": 150 }, { "epoch": 1.0, "step": 159, "total_flos": 0.0, "train_loss": 0.6280672625175812, "train_runtime": 4264.6818, "train_samples_per_second": 4.778, "train_steps_per_second": 0.037 } ], "logging_steps": 10, "max_steps": 159, "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 }