{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.9992429977289932, "eval_steps": 100, "global_step": 165, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 2.9411764705882356e-07, "logits/chosen": -2.4206371307373047, "logits/rejected": -2.4444966316223145, "logps/chosen": -285.239013671875, "logps/rejected": -269.48089599609375, "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": 2.9411764705882355e-06, "logits/chosen": -2.4849026203155518, "logits/rejected": -2.361117362976074, "logps/chosen": -271.2438659667969, "logps/rejected": -257.04443359375, "loss": 0.689, "rewards/accuracies": 0.5208333134651184, "rewards/chosen": -0.010696162469685078, "rewards/margins": 0.008694707415997982, "rewards/rejected": -0.01939087174832821, "step": 10 }, { "epoch": 0.12, "learning_rate": 4.994932636402032e-06, "logits/chosen": -2.162980318069458, "logits/rejected": -2.0703816413879395, "logps/chosen": -297.11322021484375, "logps/rejected": -317.59942626953125, "loss": 0.6434, "rewards/accuracies": 0.690625011920929, "rewards/chosen": -0.3237840533256531, "rewards/margins": 0.11028959602117538, "rewards/rejected": -0.43407368659973145, "step": 20 }, { "epoch": 0.18, "learning_rate": 4.905416503522124e-06, "logits/chosen": -1.9069904088974, "logits/rejected": -1.5060116052627563, "logps/chosen": -316.3459777832031, "logps/rejected": -310.5841064453125, "loss": 0.594, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -0.2567785978317261, "rewards/margins": 0.2907828688621521, "rewards/rejected": -0.5475614666938782, "step": 30 }, { "epoch": 0.24, "learning_rate": 4.707922373336524e-06, "logits/chosen": -1.2782038450241089, "logits/rejected": -1.1451029777526855, "logps/chosen": -334.9066467285156, "logps/rejected": -359.5328674316406, "loss": 0.6188, "rewards/accuracies": 0.6968749761581421, "rewards/chosen": -0.6329762935638428, "rewards/margins": 0.26158708333969116, "rewards/rejected": -0.8945633769035339, "step": 40 }, { "epoch": 0.3, "learning_rate": 4.411315662967732e-06, "logits/chosen": -1.3727542161941528, "logits/rejected": -1.0713155269622803, "logps/chosen": -302.04876708984375, "logps/rejected": -323.1622009277344, "loss": 0.5896, "rewards/accuracies": 0.706250011920929, "rewards/chosen": -0.33613428473472595, "rewards/margins": 0.32212409377098083, "rewards/rejected": -0.658258318901062, "step": 50 }, { "epoch": 0.36, "learning_rate": 4.028910905897229e-06, "logits/chosen": -1.2992875576019287, "logits/rejected": -0.787635862827301, "logps/chosen": -339.11273193359375, "logps/rejected": -345.2218017578125, "loss": 0.5665, "rewards/accuracies": 0.690625011920929, "rewards/chosen": -0.41970473527908325, "rewards/margins": 0.42981767654418945, "rewards/rejected": -0.8495222926139832, "step": 60 }, { "epoch": 0.42, "learning_rate": 3.577874068920446e-06, "logits/chosen": -1.1777417659759521, "logits/rejected": -0.7731892466545105, "logps/chosen": -331.8875427246094, "logps/rejected": -348.7061767578125, "loss": 0.5574, "rewards/accuracies": 0.699999988079071, "rewards/chosen": -0.45568642020225525, "rewards/margins": 0.4189928472042084, "rewards/rejected": -0.8746792674064636, "step": 70 }, { "epoch": 0.48, "learning_rate": 3.0784519801008546e-06, "logits/chosen": -1.0813881158828735, "logits/rejected": -0.695389986038208, "logps/chosen": -328.33599853515625, "logps/rejected": -378.7125549316406, "loss": 0.5757, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -0.41106706857681274, "rewards/margins": 0.49909764528274536, "rewards/rejected": -0.9101647138595581, "step": 80 }, { "epoch": 0.55, "learning_rate": 2.553063458334059e-06, "logits/chosen": -1.1919687986373901, "logits/rejected": -0.7594654560089111, "logps/chosen": -311.298095703125, "logps/rejected": -356.6019592285156, "loss": 0.5572, "rewards/accuracies": 0.737500011920929, "rewards/chosen": -0.4122340679168701, "rewards/margins": 0.47736555337905884, "rewards/rejected": -0.889599621295929, "step": 90 }, { "epoch": 0.61, "learning_rate": 2.025292943281429e-06, "logits/chosen": -1.1734464168548584, "logits/rejected": -0.8028995394706726, "logps/chosen": -314.7601013183594, "logps/rejected": -354.06524658203125, "loss": 0.5669, "rewards/accuracies": 0.7281249761581421, "rewards/chosen": -0.37006908655166626, "rewards/margins": 0.4851018786430359, "rewards/rejected": -0.8551709055900574, "step": 100 }, { "epoch": 0.61, "eval_logits/chosen": -1.1060620546340942, "eval_logits/rejected": -0.6833169460296631, "eval_logps/chosen": -329.36163330078125, "eval_logps/rejected": -354.41693115234375, "eval_loss": 0.5602970123291016, "eval_rewards/accuracies": 0.6980000138282776, "eval_rewards/chosen": -0.4443123936653137, "eval_rewards/margins": 0.428010493516922, "eval_rewards/rejected": -0.8723229169845581, "eval_runtime": 384.7387, "eval_samples_per_second": 5.198, "eval_steps_per_second": 0.65, "step": 100 }, { "epoch": 0.67, "learning_rate": 1.5188318011445907e-06, "logits/chosen": -1.08719801902771, "logits/rejected": -0.7771456241607666, "logps/chosen": -318.55706787109375, "logps/rejected": -360.88641357421875, "loss": 0.5582, "rewards/accuracies": 0.699999988079071, "rewards/chosen": -0.4783196449279785, "rewards/margins": 0.4534730315208435, "rewards/rejected": -0.9317927360534668, "step": 110 }, { "epoch": 0.73, "learning_rate": 1.0564148305586296e-06, "logits/chosen": -1.264317274093628, "logits/rejected": -0.6377102136611938, "logps/chosen": -338.55364990234375, "logps/rejected": -344.6549072265625, "loss": 0.552, "rewards/accuracies": 0.703125, "rewards/chosen": -0.4587250351905823, "rewards/margins": 0.43041476607322693, "rewards/rejected": -0.8891397714614868, "step": 120 }, { "epoch": 0.79, "learning_rate": 6.587997083462197e-07, "logits/chosen": -1.205304503440857, "logits/rejected": -0.7666274905204773, "logps/chosen": -337.1952209472656, "logps/rejected": -378.85015869140625, "loss": 0.5546, "rewards/accuracies": 0.71875, "rewards/chosen": -0.481029212474823, "rewards/margins": 0.47466927766799927, "rewards/rejected": -0.9556983709335327, "step": 130 }, { "epoch": 0.85, "learning_rate": 3.438351873250492e-07, "logits/chosen": -1.018686056137085, "logits/rejected": -0.5739496946334839, "logps/chosen": -331.6724853515625, "logps/rejected": -385.0194396972656, "loss": 0.5522, "rewards/accuracies": 0.7593749761581421, "rewards/chosen": -0.5046229958534241, "rewards/margins": 0.5355193018913269, "rewards/rejected": -1.040142297744751, "step": 140 }, { "epoch": 0.91, "learning_rate": 1.2565987432367032e-07, "logits/chosen": -1.0615447759628296, "logits/rejected": -0.6152147650718689, "logps/chosen": -330.9660949707031, "logps/rejected": -370.95867919921875, "loss": 0.5513, "rewards/accuracies": 0.7437499761581421, "rewards/chosen": -0.4836733937263489, "rewards/margins": 0.5048274397850037, "rewards/rejected": -0.9885008931159973, "step": 150 }, { "epoch": 0.97, "learning_rate": 1.4067554877743861e-08, "logits/chosen": -1.0293254852294922, "logits/rejected": -0.6487714052200317, "logps/chosen": -317.6270751953125, "logps/rejected": -361.36053466796875, "loss": 0.5338, "rewards/accuracies": 0.734375, "rewards/chosen": -0.4425296187400818, "rewards/margins": 0.528509259223938, "rewards/rejected": -0.9710389375686646, "step": 160 }, { "epoch": 1.0, "step": 165, "total_flos": 0.0, "train_loss": 0.5778744011214285, "train_runtime": 7573.0657, "train_samples_per_second": 2.791, "train_steps_per_second": 0.022 } ], "logging_steps": 10, "max_steps": 165, "num_input_tokens_seen": 0, "num_train_epochs": 1, "save_steps": 100, "total_flos": 0.0, "train_batch_size": 2, "trial_name": null, "trial_params": null }