| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 0.9984, | |
| "eval_steps": 100, | |
| "global_step": 156, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.01, | |
| "learning_rate": 3.125e-07, | |
| "logits/chosen": -2.749149799346924, | |
| "logits/rejected": -2.5982635021209717, | |
| "logps/chosen": -351.88592529296875, | |
| "logps/rejected": -339.1416931152344, | |
| "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.125e-06, | |
| "logits/chosen": -2.634215831756592, | |
| "logits/rejected": -2.565009593963623, | |
| "logps/chosen": -264.03338623046875, | |
| "logps/rejected": -276.0550842285156, | |
| "loss": 0.6919, | |
| "rewards/accuracies": 0.5277777910232544, | |
| "rewards/chosen": -0.004038373474031687, | |
| "rewards/margins": 0.00274546816945076, | |
| "rewards/rejected": -0.006783841177821159, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.13, | |
| "learning_rate": 4.989935734988098e-06, | |
| "logits/chosen": -2.6059346199035645, | |
| "logits/rejected": -2.5347509384155273, | |
| "logps/chosen": -289.2115173339844, | |
| "logps/rejected": -285.11651611328125, | |
| "loss": 0.6797, | |
| "rewards/accuracies": 0.5687500238418579, | |
| "rewards/chosen": -0.030138880014419556, | |
| "rewards/margins": 0.022945497184991837, | |
| "rewards/rejected": -0.05308438092470169, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.19, | |
| "learning_rate": 4.8776412907378845e-06, | |
| "logits/chosen": -2.549204111099243, | |
| "logits/rejected": -2.4668586254119873, | |
| "logps/chosen": -296.6357727050781, | |
| "logps/rejected": -286.880615234375, | |
| "loss": 0.6602, | |
| "rewards/accuracies": 0.671875, | |
| "rewards/chosen": -0.08948583900928497, | |
| "rewards/margins": 0.06929855048656464, | |
| "rewards/rejected": -0.1587844043970108, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.26, | |
| "learning_rate": 4.646121984004666e-06, | |
| "logits/chosen": -2.607750177383423, | |
| "logits/rejected": -2.4750921726226807, | |
| "logps/chosen": -298.0829162597656, | |
| "logps/rejected": -306.82696533203125, | |
| "loss": 0.6514, | |
| "rewards/accuracies": 0.659375011920929, | |
| "rewards/chosen": -0.19320181012153625, | |
| "rewards/margins": 0.12534096837043762, | |
| "rewards/rejected": -0.3185427784919739, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.32, | |
| "learning_rate": 4.3069871595684795e-06, | |
| "logits/chosen": -2.520982265472412, | |
| "logits/rejected": -2.4599428176879883, | |
| "logps/chosen": -296.30560302734375, | |
| "logps/rejected": -325.0945129394531, | |
| "loss": 0.6459, | |
| "rewards/accuracies": 0.6625000238418579, | |
| "rewards/chosen": -0.12883062660694122, | |
| "rewards/margins": 0.13393087685108185, | |
| "rewards/rejected": -0.26276153326034546, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.38, | |
| "learning_rate": 3.8772424536302565e-06, | |
| "logits/chosen": -2.5419609546661377, | |
| "logits/rejected": -2.4526138305664062, | |
| "logps/chosen": -303.03753662109375, | |
| "logps/rejected": -309.76702880859375, | |
| "loss": 0.6405, | |
| "rewards/accuracies": 0.668749988079071, | |
| "rewards/chosen": -0.13351231813430786, | |
| "rewards/margins": 0.12426233291625977, | |
| "rewards/rejected": -0.2577746510505676, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.45, | |
| "learning_rate": 3.3784370602033572e-06, | |
| "logits/chosen": -2.4560115337371826, | |
| "logits/rejected": -2.3811163902282715, | |
| "logps/chosen": -344.1382141113281, | |
| "logps/rejected": -355.2807922363281, | |
| "loss": 0.6231, | |
| "rewards/accuracies": 0.6812499761581421, | |
| "rewards/chosen": -0.3589150905609131, | |
| "rewards/margins": 0.1809873878955841, | |
| "rewards/rejected": -0.5399025678634644, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.51, | |
| "learning_rate": 2.835583164544139e-06, | |
| "logits/chosen": -2.4101650714874268, | |
| "logits/rejected": -2.2786593437194824, | |
| "logps/chosen": -342.35260009765625, | |
| "logps/rejected": -337.49755859375, | |
| "loss": 0.6147, | |
| "rewards/accuracies": 0.653124988079071, | |
| "rewards/chosen": -0.3759569525718689, | |
| "rewards/margins": 0.1979154795408249, | |
| "rewards/rejected": -0.573872447013855, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.58, | |
| "learning_rate": 2.2759017277414165e-06, | |
| "logits/chosen": -2.2483856678009033, | |
| "logits/rejected": -2.258373260498047, | |
| "logps/chosen": -337.4295959472656, | |
| "logps/rejected": -386.83465576171875, | |
| "loss": 0.5889, | |
| "rewards/accuracies": 0.6968749761581421, | |
| "rewards/chosen": -0.4705071449279785, | |
| "rewards/margins": 0.3142710328102112, | |
| "rewards/rejected": -0.7847781181335449, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "learning_rate": 1.7274575140626318e-06, | |
| "logits/chosen": -2.268360137939453, | |
| "logits/rejected": -2.165742874145508, | |
| "logps/chosen": -334.42974853515625, | |
| "logps/rejected": -348.5438232421875, | |
| "loss": 0.5962, | |
| "rewards/accuracies": 0.684374988079071, | |
| "rewards/chosen": -0.475590318441391, | |
| "rewards/margins": 0.3093096613883972, | |
| "rewards/rejected": -0.7849000692367554, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.64, | |
| "eval_logits/chosen": -2.2236928939819336, | |
| "eval_logits/rejected": -2.0692262649536133, | |
| "eval_logps/chosen": -337.8192443847656, | |
| "eval_logps/rejected": -356.7066955566406, | |
| "eval_loss": 0.594050407409668, | |
| "eval_rewards/accuracies": 0.7020000219345093, | |
| "eval_rewards/chosen": -0.43988561630249023, | |
| "eval_rewards/margins": 0.31074145436286926, | |
| "eval_rewards/rejected": -0.7506271600723267, | |
| "eval_runtime": 690.4847, | |
| "eval_samples_per_second": 2.897, | |
| "eval_steps_per_second": 0.362, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.7, | |
| "learning_rate": 1.217751806485235e-06, | |
| "logits/chosen": -2.207293748855591, | |
| "logits/rejected": -1.979222059249878, | |
| "logps/chosen": -342.3262634277344, | |
| "logps/rejected": -362.3468933105469, | |
| "loss": 0.5798, | |
| "rewards/accuracies": 0.7250000238418579, | |
| "rewards/chosen": -0.526713490486145, | |
| "rewards/margins": 0.35861119627952576, | |
| "rewards/rejected": -0.8853246569633484, | |
| "step": 110 | |
| }, | |
| { | |
| "epoch": 0.77, | |
| "learning_rate": 7.723433775328385e-07, | |
| "logits/chosen": -2.0455784797668457, | |
| "logits/rejected": -1.9951608180999756, | |
| "logps/chosen": -329.8724670410156, | |
| "logps/rejected": -370.48577880859375, | |
| "loss": 0.5914, | |
| "rewards/accuracies": 0.671875, | |
| "rewards/chosen": -0.56132572889328, | |
| "rewards/margins": 0.291836678981781, | |
| "rewards/rejected": -0.853162407875061, | |
| "step": 120 | |
| }, | |
| { | |
| "epoch": 0.83, | |
| "learning_rate": 4.1356686569674344e-07, | |
| "logits/chosen": -2.083890438079834, | |
| "logits/rejected": -1.9531793594360352, | |
| "logps/chosen": -333.07171630859375, | |
| "logps/rejected": -376.0746765136719, | |
| "loss": 0.5891, | |
| "rewards/accuracies": 0.715624988079071, | |
| "rewards/chosen": -0.44948244094848633, | |
| "rewards/margins": 0.361866295337677, | |
| "rewards/rejected": -0.8113487362861633, | |
| "step": 130 | |
| }, | |
| { | |
| "epoch": 0.9, | |
| "learning_rate": 1.59412823400657e-07, | |
| "logits/chosen": -2.076089382171631, | |
| "logits/rejected": -1.8212101459503174, | |
| "logps/chosen": -350.77374267578125, | |
| "logps/rejected": -360.770263671875, | |
| "loss": 0.5677, | |
| "rewards/accuracies": 0.71875, | |
| "rewards/chosen": -0.4118824005126953, | |
| "rewards/margins": 0.38872456550598145, | |
| "rewards/rejected": -0.8006070256233215, | |
| "step": 140 | |
| }, | |
| { | |
| "epoch": 0.96, | |
| "learning_rate": 2.262559558016325e-08, | |
| "logits/chosen": -2.0566818714141846, | |
| "logits/rejected": -1.8309047222137451, | |
| "logps/chosen": -340.4973449707031, | |
| "logps/rejected": -353.11956787109375, | |
| "loss": 0.5863, | |
| "rewards/accuracies": 0.721875011920929, | |
| "rewards/chosen": -0.43235865235328674, | |
| "rewards/margins": 0.37298649549484253, | |
| "rewards/rejected": -0.8053451776504517, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 156, | |
| "total_flos": 0.0, | |
| "train_loss": 0.6200787784197391, | |
| "train_runtime": 14536.4272, | |
| "train_samples_per_second": 1.376, | |
| "train_steps_per_second": 0.011 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 156, | |
| "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 | |
| } | |