| { | |
| "best_global_step": 12, | |
| "best_metric": 0.03234308212995529, | |
| "best_model_checkpoint": "/mnt/nvme5n1p1/trained_grpo_distill_14b_rl_70_s3/v3-20250330-200345/checkpoint-12", | |
| "epoch": 3.8421052631578947, | |
| "eval_steps": 6, | |
| "global_step": 16, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": false, | |
| "log_history": [ | |
| { | |
| "clip_ratio": 0.0, | |
| "completion_length": 10352.974609375, | |
| "epoch": 0.21052631578947367, | |
| "grad_norm": 0.13259537518024445, | |
| "kl": 0.0, | |
| "learning_rate": 1.6666666666666667e-05, | |
| "loss": -0.11016345024108887, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.11328125, | |
| "reward": -0.002658387296833098, | |
| "reward_std": 0.06134121119976044, | |
| "rewards/CosineReward": -0.0026579967816360295, | |
| "rewards/RepetitionPenalty": -3.8975886695880035e-07, | |
| "step": 1, | |
| "train_speed(iter/s)": 0.000241 | |
| }, | |
| { | |
| "clip_ratio": 0.0, | |
| "epoch": 0.42105263157894735, | |
| "grad_norm": 0.1320001482963562, | |
| "kl": 0.0, | |
| "learning_rate": 3.3333333333333335e-05, | |
| "loss": -0.11016345024108887, | |
| "memory(GiB)": 186.69, | |
| "step": 2, | |
| "train_speed(iter/s)": 0.000466 | |
| }, | |
| { | |
| "clip_ratio": 1.3441811461234465e-05, | |
| "completion_length": 10439.369140625, | |
| "epoch": 0.631578947368421, | |
| "grad_norm": 0.08990391343832016, | |
| "kl": 9.50181856751442e-07, | |
| "learning_rate": 5e-05, | |
| "loss": -0.06604708731174469, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.13671875, | |
| "reward": 0.0006296975770965219, | |
| "reward_std": 0.07172460854053497, | |
| "rewards/CosineReward": 0.0006298604130279273, | |
| "rewards/RepetitionPenalty": -1.6200439745261974e-07, | |
| "step": 3, | |
| "train_speed(iter/s)": 0.00035 | |
| }, | |
| { | |
| "clip_ratio": 1.70210253145342e-05, | |
| "epoch": 0.8421052631578947, | |
| "grad_norm": 0.0967094898223877, | |
| "kl": 1.1101365089416504e-05, | |
| "learning_rate": 6.666666666666667e-05, | |
| "loss": -0.06727766245603561, | |
| "memory(GiB)": 186.69, | |
| "step": 4, | |
| "train_speed(iter/s)": 0.000458 | |
| }, | |
| { | |
| "clip_ratio": 1.675608473306056e-05, | |
| "completion_length": 10092.408203125, | |
| "epoch": 1.2105263157894737, | |
| "grad_norm": 0.142837256193161, | |
| "kl": 0.00017762184143066406, | |
| "learning_rate": 8.333333333333334e-05, | |
| "loss": -0.09315311908721924, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.119140625, | |
| "reward": -0.005135859013535082, | |
| "reward_std": 0.07994875870645046, | |
| "rewards/CosineReward": -0.005134060338605195, | |
| "rewards/RepetitionPenalty": -1.7973881654143042e-06, | |
| "step": 5, | |
| "train_speed(iter/s)": 0.000387 | |
| }, | |
| { | |
| "epoch": 1.4210526315789473, | |
| "grad_norm": 0.18263348937034607, | |
| "learning_rate": 0.0001, | |
| "loss": -0.1041698157787323, | |
| "memory(GiB)": 186.69, | |
| "step": 6, | |
| "train_speed(iter/s)": 0.000459 | |
| }, | |
| { | |
| "epoch": 1.4210526315789473, | |
| "eval_clip_ratio": 4.069424539920874e-05, | |
| "eval_completion_length": 12289.0, | |
| "eval_kl": 0.04833984375, | |
| "eval_loss": -0.5377416610717773, | |
| "eval_response_clip_ratio": 1.0, | |
| "eval_reward": 0.012996690347790718, | |
| "eval_reward_std": 0.08769983053207397, | |
| "eval_rewards/CosineReward": 0.012996694073081017, | |
| "eval_rewards/RepetitionPenalty": 0.0, | |
| "eval_runtime": 1030.1122, | |
| "eval_samples_per_second": 0.001, | |
| "eval_steps_per_second": 0.001, | |
| "step": 6 | |
| }, | |
| { | |
| "clip_ratio": 0.0005237623976199757, | |
| "completion_length": 10448.94921875, | |
| "epoch": 1.631578947368421, | |
| "grad_norm": 0.1291271299123764, | |
| "kl": 0.017406463623046875, | |
| "learning_rate": 9.991540791356342e-05, | |
| "loss": -0.051375165581703186, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.1484375, | |
| "reward": 0.004909618757665157, | |
| "reward_std": 0.08167182095348835, | |
| "rewards/CosineReward": 0.004909833543933928, | |
| "rewards/RepetitionPenalty": -2.1478646772266075e-07, | |
| "step": 7, | |
| "train_speed(iter/s)": 0.000382 | |
| }, | |
| { | |
| "clip_ratio": 0.1706484742462635, | |
| "epoch": 1.8421052631578947, | |
| "grad_norm": 0.26641014218330383, | |
| "kl": 0.089599609375, | |
| "learning_rate": 9.966191788709716e-05, | |
| "loss": -0.05105742812156677, | |
| "memory(GiB)": 186.69, | |
| "step": 8, | |
| "train_speed(iter/s)": 0.000433 | |
| }, | |
| { | |
| "clip_ratio": 9.482144946559856e-06, | |
| "completion_length": 10432.384765625, | |
| "epoch": 2.2105263157894735, | |
| "grad_norm": 0.10375155508518219, | |
| "kl": 0.0963134765625, | |
| "learning_rate": 9.924038765061042e-05, | |
| "loss": -0.05842069163918495, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.255859375, | |
| "reward": 0.03643610421568155, | |
| "reward_std": 0.11898956261575222, | |
| "rewards/CosineReward": 0.03643618477508426, | |
| "rewards/RepetitionPenalty": -7.898860587829404e-08, | |
| "step": 9, | |
| "train_speed(iter/s)": 0.000396 | |
| }, | |
| { | |
| "clip_ratio": 0.0036088433116674423, | |
| "epoch": 2.4210526315789473, | |
| "grad_norm": 0.09477333724498749, | |
| "kl": 0.1185302734375, | |
| "learning_rate": 9.865224352899119e-05, | |
| "loss": -0.06491819024085999, | |
| "memory(GiB)": 186.69, | |
| "step": 10, | |
| "train_speed(iter/s)": 0.000436 | |
| }, | |
| { | |
| "clip_ratio": 1.2955343891007942e-05, | |
| "completion_length": 10559.296875, | |
| "epoch": 2.6315789473684212, | |
| "grad_norm": 0.06739140301942825, | |
| "kl": 0.1275634765625, | |
| "learning_rate": 9.789947561577445e-05, | |
| "loss": -0.04600231721997261, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.361328125, | |
| "reward": 0.023204635945148766, | |
| "reward_std": 0.10593634657561779, | |
| "rewards/CosineReward": 0.02320496749598533, | |
| "rewards/RepetitionPenalty": -3.3051759373847744e-07, | |
| "step": 11, | |
| "train_speed(iter/s)": 0.000405 | |
| }, | |
| { | |
| "epoch": 2.8421052631578947, | |
| "grad_norm": 0.05781339108943939, | |
| "learning_rate": 9.698463103929542e-05, | |
| "loss": -0.05069056898355484, | |
| "memory(GiB)": 186.69, | |
| "step": 12, | |
| "train_speed(iter/s)": 0.000439 | |
| }, | |
| { | |
| "epoch": 2.8421052631578947, | |
| "eval_clip_ratio": 4.392032860778272e-05, | |
| "eval_completion_length": 12289.0, | |
| "eval_kl": 0.2275390625, | |
| "eval_loss": 0.17524278163909912, | |
| "eval_response_clip_ratio": 1.0, | |
| "eval_reward": 0.03234308212995529, | |
| "eval_reward_std": 0.10685288906097412, | |
| "eval_rewards/CosineReward": 0.03234308212995529, | |
| "eval_rewards/RepetitionPenalty": 0.0, | |
| "eval_runtime": 1025.9053, | |
| "eval_samples_per_second": 0.001, | |
| "eval_steps_per_second": 0.001, | |
| "step": 12 | |
| }, | |
| { | |
| "clip_ratio": 0.0007908324183745208, | |
| "completion_length": 10652.939453125, | |
| "epoch": 3.2105263157894735, | |
| "grad_norm": 0.01199417095631361, | |
| "kl": 0.151123046875, | |
| "learning_rate": 9.591080534401371e-05, | |
| "loss": -0.02191038429737091, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.419921875, | |
| "reward": 0.035983758978545666, | |
| "reward_std": 0.11553369648754597, | |
| "rewards/CosineReward": 0.03598417737521231, | |
| "rewards/RepetitionPenalty": -4.176556771540163e-07, | |
| "step": 13, | |
| "train_speed(iter/s)": 0.000399 | |
| }, | |
| { | |
| "clip_ratio": 0.0004821276670554653, | |
| "epoch": 3.4210526315789473, | |
| "grad_norm": 0.01075426209717989, | |
| "kl": 0.169189453125, | |
| "learning_rate": 9.468163201617062e-05, | |
| "loss": -0.022672578692436218, | |
| "memory(GiB)": 186.69, | |
| "step": 14, | |
| "train_speed(iter/s)": 0.000427 | |
| }, | |
| { | |
| "clip_ratio": 1.9617403040683712e-05, | |
| "completion_length": 10482.146484375, | |
| "epoch": 3.6315789473684212, | |
| "grad_norm": 0.01779361069202423, | |
| "kl": 0.166748046875, | |
| "learning_rate": 9.330127018922194e-05, | |
| "loss": -0.059799157083034515, | |
| "memory(GiB)": 186.69, | |
| "response_clip_ratio": 0.4765625, | |
| "reward": 0.03584331553429365, | |
| "reward_std": 0.11829411797225475, | |
| "rewards/CosineReward": 0.03584346390562132, | |
| "rewards/RepetitionPenalty": -1.4977952389472193e-07, | |
| "step": 15, | |
| "train_speed(iter/s)": 0.000406 | |
| }, | |
| { | |
| "clip_ratio": 0.00011349086707923561, | |
| "epoch": 3.8421052631578947, | |
| "grad_norm": 0.013216385617852211, | |
| "kl": 0.16748046875, | |
| "learning_rate": 9.177439057064683e-05, | |
| "loss": -0.06071458384394646, | |
| "memory(GiB)": 186.69, | |
| "step": 16, | |
| "train_speed(iter/s)": 0.000431 | |
| } | |
| ], | |
| "logging_steps": 1, | |
| "max_steps": 60, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 15, | |
| "save_steps": 2, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": false | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 0.0, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |