| { | |
| "best_metric": null, | |
| "best_model_checkpoint": null, | |
| "epoch": 1.0, | |
| "eval_steps": 100, | |
| "global_step": 96, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.010416666666666666, | |
| "grad_norm": 3.100854180799138, | |
| "learning_rate": 6.389441661416039e-08, | |
| "logits/chosen": -2.590585231781006, | |
| "logits/rejected": -2.5664222240448, | |
| "logps/chosen": -80.29847717285156, | |
| "logps/rejected": -53.10200881958008, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.0, | |
| "rewards/chosen": 0.0, | |
| "rewards/margins": 0.0, | |
| "rewards/rejected": 0.0, | |
| "step": 1 | |
| }, | |
| { | |
| "epoch": 0.10416666666666667, | |
| "grad_norm": 2.781126507699089, | |
| "learning_rate": 6.389441661416038e-07, | |
| "logits/chosen": -2.5558536052703857, | |
| "logits/rejected": -2.5378901958465576, | |
| "logps/chosen": -87.89214324951172, | |
| "logps/rejected": -80.98480224609375, | |
| "loss": 0.6931, | |
| "rewards/accuracies": 0.2291666716337204, | |
| "rewards/chosen": 0.00010837005538633093, | |
| "rewards/margins": -0.00014457135694101453, | |
| "rewards/rejected": 0.0002529414196033031, | |
| "step": 10 | |
| }, | |
| { | |
| "epoch": 0.20833333333333334, | |
| "grad_norm": 2.553038645699022, | |
| "learning_rate": 5.762762398460686e-07, | |
| "logits/chosen": -2.5938525199890137, | |
| "logits/rejected": -2.543900966644287, | |
| "logps/chosen": -101.56424713134766, | |
| "logps/rejected": -88.41798400878906, | |
| "loss": 0.6914, | |
| "rewards/accuracies": 0.3375000059604645, | |
| "rewards/chosen": 0.00841432437300682, | |
| "rewards/margins": 0.0015086036873981357, | |
| "rewards/rejected": 0.006905719637870789, | |
| "step": 20 | |
| }, | |
| { | |
| "epoch": 0.3125, | |
| "grad_norm": 3.7950522400890523, | |
| "learning_rate": 5.136083135505332e-07, | |
| "logits/chosen": -2.456357002258301, | |
| "logits/rejected": -2.472012758255005, | |
| "logps/chosen": -66.62992095947266, | |
| "logps/rejected": -75.86373901367188, | |
| "loss": 0.6891, | |
| "rewards/accuracies": 0.32499998807907104, | |
| "rewards/chosen": 0.003802085295319557, | |
| "rewards/margins": 0.0082187969237566, | |
| "rewards/rejected": -0.004416711628437042, | |
| "step": 30 | |
| }, | |
| { | |
| "epoch": 0.4166666666666667, | |
| "grad_norm": 3.519596152843829, | |
| "learning_rate": 4.5094038725499785e-07, | |
| "logits/chosen": -2.5018997192382812, | |
| "logits/rejected": -2.4887325763702393, | |
| "logps/chosen": -71.3689956665039, | |
| "logps/rejected": -73.41954040527344, | |
| "loss": 0.6849, | |
| "rewards/accuracies": 0.3062500059604645, | |
| "rewards/chosen": 0.003661749418824911, | |
| "rewards/margins": 0.01640659011900425, | |
| "rewards/rejected": -0.012744839303195477, | |
| "step": 40 | |
| }, | |
| { | |
| "epoch": 0.5208333333333334, | |
| "grad_norm": 5.231502190809433, | |
| "learning_rate": 3.8827246095946254e-07, | |
| "logits/chosen": -2.3902275562286377, | |
| "logits/rejected": -2.4062817096710205, | |
| "logps/chosen": -49.79673385620117, | |
| "logps/rejected": -60.72575759887695, | |
| "loss": 0.6826, | |
| "rewards/accuracies": 0.22499999403953552, | |
| "rewards/chosen": 0.005200926214456558, | |
| "rewards/margins": 0.013756193220615387, | |
| "rewards/rejected": -0.008555267006158829, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.625, | |
| "grad_norm": 4.689957831926571, | |
| "learning_rate": 3.256045346639272e-07, | |
| "logits/chosen": -2.4439873695373535, | |
| "logits/rejected": -2.4207093715667725, | |
| "logps/chosen": -86.81434631347656, | |
| "logps/rejected": -94.7623291015625, | |
| "loss": 0.6804, | |
| "rewards/accuracies": 0.3125, | |
| "rewards/chosen": -0.036475084722042084, | |
| "rewards/margins": 0.036166418343782425, | |
| "rewards/rejected": -0.07264150679111481, | |
| "step": 60 | |
| }, | |
| { | |
| "epoch": 0.7291666666666666, | |
| "grad_norm": 5.829443326665767, | |
| "learning_rate": 2.6293660836839186e-07, | |
| "logits/chosen": -2.3675405979156494, | |
| "logits/rejected": -2.348362445831299, | |
| "logps/chosen": -116.78021240234375, | |
| "logps/rejected": -112.79603576660156, | |
| "loss": 0.68, | |
| "rewards/accuracies": 0.3125, | |
| "rewards/chosen": -0.09623803943395615, | |
| "rewards/margins": 0.03254436329007149, | |
| "rewards/rejected": -0.12878240644931793, | |
| "step": 70 | |
| }, | |
| { | |
| "epoch": 0.8333333333333334, | |
| "grad_norm": 6.4331651891422785, | |
| "learning_rate": 2.0026868207285652e-07, | |
| "logits/chosen": -2.2932543754577637, | |
| "logits/rejected": -2.2299153804779053, | |
| "logps/chosen": -97.70858001708984, | |
| "logps/rejected": -105.18177795410156, | |
| "loss": 0.6695, | |
| "rewards/accuracies": 0.39375001192092896, | |
| "rewards/chosen": -0.05681135132908821, | |
| "rewards/margins": 0.06468762457370758, | |
| "rewards/rejected": -0.1214989647269249, | |
| "step": 80 | |
| }, | |
| { | |
| "epoch": 0.9375, | |
| "grad_norm": 5.087077256901478, | |
| "learning_rate": 1.376007557773212e-07, | |
| "logits/chosen": -2.2146198749542236, | |
| "logits/rejected": -2.2123677730560303, | |
| "logps/chosen": -61.2127571105957, | |
| "logps/rejected": -81.22966003417969, | |
| "loss": 0.6723, | |
| "rewards/accuracies": 0.26249998807907104, | |
| "rewards/chosen": -0.034062620252370834, | |
| "rewards/margins": 0.049583159387111664, | |
| "rewards/rejected": -0.0836457759141922, | |
| "step": 90 | |
| }, | |
| { | |
| "epoch": 1.0, | |
| "step": 96, | |
| "total_flos": 0.0, | |
| "train_loss": 0.6817257404327393, | |
| "train_runtime": 994.9537, | |
| "train_samples_per_second": 6.144, | |
| "train_steps_per_second": 0.096 | |
| } | |
| ], | |
| "logging_steps": 10, | |
| "max_steps": 96, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 100, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 0.0, | |
| "train_batch_size": 8, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |