{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.99581589958159, "eval_steps": 500, "global_step": 119, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.008368200836820083, "grad_norm": 13.7675517674306, "learning_rate": 4.166666666666666e-08, "logits/chosen": -2.6905035972595215, "logits/rejected": -2.6002821922302246, "logps/chosen": -238.50418090820312, "logps/pi_response": -60.972923278808594, "logps/ref_response": -60.972923278808594, "logps/rejected": -122.8811264038086, "loss": 0.6931, "rewards/accuracies": 0.0, "rewards/chosen": 0.0, "rewards/margins": 0.0, "rewards/rejected": 0.0, "step": 1 }, { "epoch": 0.08368200836820083, "grad_norm": 11.556330314168122, "learning_rate": 4.1666666666666667e-07, "logits/chosen": -2.7245476245880127, "logits/rejected": -2.6954076290130615, "logps/chosen": -223.26068115234375, "logps/pi_response": -68.41806030273438, "logps/ref_response": -68.24886322021484, "logps/rejected": -108.38255310058594, "loss": 0.6909, "rewards/accuracies": 0.5625, "rewards/chosen": 0.00402547512203455, "rewards/margins": 0.00467924727126956, "rewards/rejected": -0.0006537723238579929, "step": 10 }, { "epoch": 0.16736401673640167, "grad_norm": 8.0051562247015, "learning_rate": 4.931352528237397e-07, "logits/chosen": -2.7007386684417725, "logits/rejected": -2.6637234687805176, "logps/chosen": -239.10507202148438, "logps/pi_response": -75.34381866455078, "logps/ref_response": -65.86756896972656, "logps/rejected": -120.88731384277344, "loss": 0.6538, "rewards/accuracies": 0.768750011920929, "rewards/chosen": 0.02172889932990074, "rewards/margins": 0.08931762725114822, "rewards/rejected": -0.06758873164653778, "step": 20 }, { "epoch": 0.2510460251046025, "grad_norm": 6.2838771200093735, "learning_rate": 4.658920803689553e-07, "logits/chosen": -2.5834615230560303, "logits/rejected": -2.541572093963623, "logps/chosen": -234.6986541748047, "logps/pi_response": -112.6206283569336, "logps/ref_response": -70.35834503173828, "logps/rejected": -146.62071228027344, "loss": 0.6128, "rewards/accuracies": 0.731249988079071, "rewards/chosen": -0.1661997139453888, "rewards/margins": 0.1909169852733612, "rewards/rejected": -0.35711669921875, "step": 30 }, { "epoch": 0.33472803347280333, "grad_norm": 8.1669760332498, "learning_rate": 4.201712553872657e-07, "logits/chosen": -2.544977903366089, "logits/rejected": -2.512420415878296, "logps/chosen": -266.4893798828125, "logps/pi_response": -148.94216918945312, "logps/ref_response": -70.82736206054688, "logps/rejected": -183.97122192382812, "loss": 0.5908, "rewards/accuracies": 0.7250000238418579, "rewards/chosen": -0.31368952989578247, "rewards/margins": 0.3473220467567444, "rewards/rejected": -0.6610115170478821, "step": 40 }, { "epoch": 0.41841004184100417, "grad_norm": 10.22318295333062, "learning_rate": 3.598859066780754e-07, "logits/chosen": -2.5687315464019775, "logits/rejected": -2.520003318786621, "logps/chosen": -306.9870300292969, "logps/pi_response": -178.8148956298828, "logps/ref_response": -73.63240814208984, "logps/rejected": -191.5988006591797, "loss": 0.5573, "rewards/accuracies": 0.75, "rewards/chosen": -0.37883269786834717, "rewards/margins": 0.5206230282783508, "rewards/rejected": -0.899455726146698, "step": 50 }, { "epoch": 0.502092050209205, "grad_norm": 12.014191391493059, "learning_rate": 2.9019570347986706e-07, "logits/chosen": -2.542023181915283, "logits/rejected": -2.512960195541382, "logps/chosen": -290.0708923339844, "logps/pi_response": -188.24777221679688, "logps/ref_response": -76.79859924316406, "logps/rejected": -204.185791015625, "loss": 0.5334, "rewards/accuracies": 0.8125, "rewards/chosen": -0.3407396078109741, "rewards/margins": 0.7002220153808594, "rewards/rejected": -1.0409616231918335, "step": 60 }, { "epoch": 0.5857740585774058, "grad_norm": 11.306193009663613, "learning_rate": 2.1706525253979534e-07, "logits/chosen": -2.555372476577759, "logits/rejected": -2.517289400100708, "logps/chosen": -280.76043701171875, "logps/pi_response": -182.2036895751953, "logps/ref_response": -66.60467529296875, "logps/rejected": -207.8173828125, "loss": 0.549, "rewards/accuracies": 0.675000011920929, "rewards/chosen": -0.4452129900455475, "rewards/margins": 0.5477712750434875, "rewards/rejected": -0.9929842948913574, "step": 70 }, { "epoch": 0.6694560669456067, "grad_norm": 17.500798301996966, "learning_rate": 1.4675360263490295e-07, "logits/chosen": -2.516525983810425, "logits/rejected": -2.505002021789551, "logps/chosen": -231.9325714111328, "logps/pi_response": -168.99024963378906, "logps/ref_response": -66.72526550292969, "logps/rejected": -189.8736114501953, "loss": 0.5258, "rewards/accuracies": 0.737500011920929, "rewards/chosen": -0.42112892866134644, "rewards/margins": 0.49120765924453735, "rewards/rejected": -0.912336528301239, "step": 80 }, { "epoch": 0.7531380753138075, "grad_norm": 16.641411155096407, "learning_rate": 8.527854855097224e-08, "logits/chosen": -2.596357822418213, "logits/rejected": -2.5533790588378906, "logps/chosen": -280.3979187011719, "logps/pi_response": -187.88308715820312, "logps/ref_response": -65.7259750366211, "logps/rejected": -200.82089233398438, "loss": 0.5428, "rewards/accuracies": 0.8187500238418579, "rewards/chosen": -0.4766550064086914, "rewards/margins": 0.6347955465316772, "rewards/rejected": -1.1114505529403687, "step": 90 }, { "epoch": 0.8368200836820083, "grad_norm": 12.985065267871795, "learning_rate": 3.790158337517127e-08, "logits/chosen": -2.542086124420166, "logits/rejected": -2.512556552886963, "logps/chosen": -261.39459228515625, "logps/pi_response": -190.4388427734375, "logps/ref_response": -67.59146881103516, "logps/rejected": -229.26296997070312, "loss": 0.5035, "rewards/accuracies": 0.75, "rewards/chosen": -0.5009726285934448, "rewards/margins": 0.5954475998878479, "rewards/rejected": -1.096420168876648, "step": 100 }, { "epoch": 0.9205020920502092, "grad_norm": 20.329401448237313, "learning_rate": 8.677580722139671e-09, "logits/chosen": -2.6092028617858887, "logits/rejected": -2.5456957817077637, "logps/chosen": -294.7940979003906, "logps/pi_response": -201.2043914794922, "logps/ref_response": -79.07398223876953, "logps/rejected": -223.47451782226562, "loss": 0.5098, "rewards/accuracies": 0.793749988079071, "rewards/chosen": -0.3964020609855652, "rewards/margins": 0.7144461870193481, "rewards/rejected": -1.110848307609558, "step": 110 }, { "epoch": 0.99581589958159, "step": 119, "total_flos": 0.0, "train_loss": 0.5697346855612362, "train_runtime": 2586.0024, "train_samples_per_second": 5.91, "train_steps_per_second": 0.046 } ], "logging_steps": 10, "max_steps": 119, "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 }