{ "best_metric": null, "best_model_checkpoint": null, "epoch": 0.12698412698412698, "eval_steps": 50, "global_step": 100, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.012698412698412698, "grad_norm": 0.04658036306500435, "learning_rate": 4.999451708687114e-06, "logits/chosen": 15.106437683105469, "logits/rejected": 15.158523559570312, "logps/chosen": -0.30069679021835327, "logps/rejected": -0.30243945121765137, "loss": 0.9981, "rewards/accuracies": 0.3499999940395355, "rewards/chosen": -0.45104512572288513, "rewards/margins": 0.002613987773656845, "rewards/rejected": -0.45365914702415466, "step": 10 }, { "epoch": 0.025396825396825397, "grad_norm": 0.05173320695757866, "learning_rate": 4.997807075247147e-06, "logits/chosen": 14.680102348327637, "logits/rejected": 14.592491149902344, "logps/chosen": -0.3225177228450775, "logps/rejected": -0.27230435609817505, "loss": 0.9967, "rewards/accuracies": 0.26249998807907104, "rewards/chosen": -0.4837765693664551, "rewards/margins": -0.07532001286745071, "rewards/rejected": -0.40845656394958496, "step": 20 }, { "epoch": 0.0380952380952381, "grad_norm": 0.04962443560361862, "learning_rate": 4.9950668210706795e-06, "logits/chosen": 15.399703979492188, "logits/rejected": 15.089459419250488, "logps/chosen": -0.2997470498085022, "logps/rejected": -0.28447264432907104, "loss": 0.9988, "rewards/accuracies": 0.3125, "rewards/chosen": -0.4496205747127533, "rewards/margins": -0.022911589592695236, "rewards/rejected": -0.42670899629592896, "step": 30 }, { "epoch": 0.050793650793650794, "grad_norm": 0.05171326920390129, "learning_rate": 4.9912321481237616e-06, "logits/chosen": 15.099847793579102, "logits/rejected": 15.159170150756836, "logps/chosen": -0.29869550466537476, "logps/rejected": -0.2806033790111542, "loss": 0.9841, "rewards/accuracies": 0.375, "rewards/chosen": -0.44804325699806213, "rewards/margins": -0.02713816799223423, "rewards/rejected": -0.42090511322021484, "step": 40 }, { "epoch": 0.06349206349206349, "grad_norm": 0.048664532601833344, "learning_rate": 4.986304738420684e-06, "logits/chosen": 15.207334518432617, "logits/rejected": 15.135488510131836, "logps/chosen": -0.3121686577796936, "logps/rejected": -0.3194735050201416, "loss": 0.9953, "rewards/accuracies": 0.3499999940395355, "rewards/chosen": -0.468252956867218, "rewards/margins": 0.010957291349768639, "rewards/rejected": -0.47921022772789, "step": 50 }, { "epoch": 0.06349206349206349, "eval_logits/chosen": 15.441752433776855, "eval_logits/rejected": 15.288756370544434, "eval_logps/chosen": -0.33557233214378357, "eval_logps/rejected": -0.31198158860206604, "eval_loss": 0.9914231300354004, "eval_rewards/accuracies": 0.3125, "eval_rewards/chosen": -0.5033585429191589, "eval_rewards/margins": -0.03538615256547928, "eval_rewards/rejected": -0.46797239780426025, "eval_runtime": 19.0844, "eval_samples_per_second": 26.723, "eval_steps_per_second": 3.354, "step": 50 }, { "epoch": 0.0761904761904762, "grad_norm": 0.0722479596734047, "learning_rate": 4.980286753286196e-06, "logits/chosen": 15.281786918640137, "logits/rejected": 15.110156059265137, "logps/chosen": -0.30892473459243774, "logps/rejected": -0.26251715421676636, "loss": 1.0022, "rewards/accuracies": 0.25, "rewards/chosen": -0.4633871018886566, "rewards/margins": -0.0696113258600235, "rewards/rejected": -0.3937757611274719, "step": 60 }, { "epoch": 0.08888888888888889, "grad_norm": 0.07008200883865356, "learning_rate": 4.973180832407471e-06, "logits/chosen": 14.836176872253418, "logits/rejected": 14.98499584197998, "logps/chosen": -0.29483428597450256, "logps/rejected": -0.29870957136154175, "loss": 0.9849, "rewards/accuracies": 0.4000000059604645, "rewards/chosen": -0.44225144386291504, "rewards/margins": 0.0058129094541072845, "rewards/rejected": -0.44806432723999023, "step": 70 }, { "epoch": 0.10158730158730159, "grad_norm": 0.06503555178642273, "learning_rate": 4.964990092676263e-06, "logits/chosen": 15.508198738098145, "logits/rejected": 15.592549324035645, "logps/chosen": -0.3066270351409912, "logps/rejected": -0.2757572531700134, "loss": 0.9886, "rewards/accuracies": 0.3125, "rewards/chosen": -0.45994052290916443, "rewards/margins": -0.046304650604724884, "rewards/rejected": -0.41363590955734253, "step": 80 }, { "epoch": 0.11428571428571428, "grad_norm": 0.10515156388282776, "learning_rate": 4.9557181268217225e-06, "logits/chosen": 15.301602363586426, "logits/rejected": 15.421157836914062, "logps/chosen": -0.31223705410957336, "logps/rejected": -0.29449179768562317, "loss": 0.981, "rewards/accuracies": 0.2750000059604645, "rewards/chosen": -0.46835556626319885, "rewards/margins": -0.02661792002618313, "rewards/rejected": -0.44173765182495117, "step": 90 }, { "epoch": 0.12698412698412698, "grad_norm": 0.05785346403717995, "learning_rate": 4.9453690018345144e-06, "logits/chosen": 16.00307846069336, "logits/rejected": 15.64977741241455, "logps/chosen": -0.32283931970596313, "logps/rejected": -0.28576889634132385, "loss": 0.9883, "rewards/accuracies": 0.2750000059604645, "rewards/chosen": -0.4842589795589447, "rewards/margins": -0.055605631321668625, "rewards/rejected": -0.42865338921546936, "step": 100 }, { "epoch": 0.12698412698412698, "eval_logits/chosen": 15.624425888061523, "eval_logits/rejected": 15.548928260803223, "eval_logps/chosen": -0.33450835943222046, "eval_logps/rejected": -0.31935107707977295, "eval_loss": 0.9836427569389343, "eval_rewards/accuracies": 0.359375, "eval_rewards/chosen": -0.5017625689506531, "eval_rewards/margins": -0.022735953330993652, "eval_rewards/rejected": -0.4790266156196594, "eval_runtime": 18.6768, "eval_samples_per_second": 27.307, "eval_steps_per_second": 3.427, "step": 100 } ], "logging_steps": 10, "max_steps": 1500, "num_input_tokens_seen": 0, "num_train_epochs": 2, "save_steps": 50, "stateful_callbacks": { "TrainerControl": { "args": { "should_epoch_stop": false, "should_evaluate": false, "should_log": false, "should_save": true, "should_training_stop": false }, "attributes": {} } }, "total_flos": 2.4275563823130214e+17, "train_batch_size": 1, "trial_name": null, "trial_params": null }