| { |
| "best_global_step": 24, |
| "best_metric": 0.04339282959699631, |
| "best_model_checkpoint": "/mnt/nvme5n1p1/trained_grpo_distill_14b_rl_70_s3/v3-20250330-200345/checkpoint-24", |
| "epoch": 6.842105263157895, |
| "eval_steps": 6, |
| "global_step": 28, |
| "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 |
| }, |
| { |
| "clip_ratio": 2.4864069928298704e-05, |
| "completion_length": 10822.3515625, |
| "epoch": 4.2105263157894735, |
| "grad_norm": 0.008352754637598991, |
| "kl": 0.1787109375, |
| "learning_rate": 9.01061596377522e-05, |
| "loss": -0.04504441097378731, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.5625, |
| "reward": 0.027318883687257767, |
| "reward_std": 0.10441224090754986, |
| "rewards/CosineReward": 0.027319116634316742, |
| "rewards/RepetitionPenalty": -2.338138500590503e-07, |
| "step": 17, |
| "train_speed(iter/s)": 0.00041 |
| }, |
| { |
| "epoch": 4.421052631578947, |
| "grad_norm": 0.005998397711664438, |
| "learning_rate": 8.83022221559489e-05, |
| "loss": -0.045487549155950546, |
| "memory(GiB)": 186.69, |
| "step": 18, |
| "train_speed(iter/s)": 0.000432 |
| }, |
| { |
| "epoch": 4.421052631578947, |
| "eval_clip_ratio": 2.286707422172185e-05, |
| "eval_completion_length": 12289.0, |
| "eval_kl": 0.18359375, |
| "eval_loss": -0.38219889998435974, |
| "eval_response_clip_ratio": 1.0, |
| "eval_reward": 0.03729328140616417, |
| "eval_reward_std": 0.10691346973180771, |
| "eval_rewards/CosineReward": 0.03729327768087387, |
| "eval_rewards/RepetitionPenalty": 0.0, |
| "eval_runtime": 1041.2277, |
| "eval_samples_per_second": 0.001, |
| "eval_steps_per_second": 0.001, |
| "step": 18 |
| }, |
| { |
| "clip_ratio": 6.176384295031312e-05, |
| "completion_length": 10454.50390625, |
| "epoch": 4.631578947368421, |
| "grad_norm": 0.007075619418174028, |
| "kl": 0.1820068359375, |
| "learning_rate": 8.636868207865244e-05, |
| "loss": -0.03466903418302536, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.466796875, |
| "reward": 0.04069916973821819, |
| "reward_std": 0.11991005763411522, |
| "rewards/CosineReward": 0.04070046404376626, |
| "rewards/RepetitionPenalty": -1.294118249006715e-06, |
| "step": 19, |
| "train_speed(iter/s)": 0.000404 |
| }, |
| { |
| "clip_ratio": 6.06911453360226e-05, |
| "epoch": 4.842105263157895, |
| "grad_norm": 0.005896567367017269, |
| "kl": 0.19287109375, |
| "learning_rate": 8.43120818934367e-05, |
| "loss": -0.03502114117145538, |
| "memory(GiB)": 186.69, |
| "step": 20, |
| "train_speed(iter/s)": 0.000424 |
| }, |
| { |
| "clip_ratio": 3.8725801914551994e-05, |
| "completion_length": 10645.056640625, |
| "epoch": 5.2105263157894735, |
| "grad_norm": 0.004154536407440901, |
| "kl": 0.17626953125, |
| "learning_rate": 8.213938048432697e-05, |
| "loss": -0.008662773296236992, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.5625, |
| "reward": 0.04996980866417289, |
| "reward_std": 0.13849420100450516, |
| "rewards/CosineReward": 0.049969930201768875, |
| "rewards/RepetitionPenalty": -1.1864573679076784e-07, |
| "step": 21, |
| "train_speed(iter/s)": 0.000407 |
| }, |
| { |
| "clip_ratio": 5.869188044016482e-05, |
| "epoch": 5.421052631578947, |
| "grad_norm": 0.004300669766962528, |
| "kl": 0.178955078125, |
| "learning_rate": 7.985792958513931e-05, |
| "loss": -0.008743642829358578, |
| "memory(GiB)": 186.69, |
| "step": 22, |
| "train_speed(iter/s)": 0.000426 |
| }, |
| { |
| "clip_ratio": 4.6346245653694496e-05, |
| "completion_length": 10538.072265625, |
| "epoch": 5.631578947368421, |
| "grad_norm": 0.01327697653323412, |
| "kl": 0.1796875, |
| "learning_rate": 7.74754489035403e-05, |
| "loss": -0.03423420712351799, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.583984375, |
| "reward": 0.034468831261619925, |
| "reward_std": 0.11841745302081108, |
| "rewards/CosineReward": 0.03447544714435935, |
| "rewards/RepetitionPenalty": -6.612649428916484e-06, |
| "step": 23, |
| "train_speed(iter/s)": 0.00041 |
| }, |
| { |
| "epoch": 5.842105263157895, |
| "grad_norm": 0.014131724834442139, |
| "learning_rate": 7.500000000000001e-05, |
| "loss": -0.03426633030176163, |
| "memory(GiB)": 186.69, |
| "step": 24, |
| "train_speed(iter/s)": 0.000427 |
| }, |
| { |
| "epoch": 5.842105263157895, |
| "eval_clip_ratio": 4.0687620639801025e-05, |
| "eval_completion_length": 12289.0, |
| "eval_kl": 0.1982421875, |
| "eval_loss": 0.3612469434738159, |
| "eval_response_clip_ratio": 1.0, |
| "eval_reward": 0.04339282959699631, |
| "eval_reward_std": 0.10456253588199615, |
| "eval_rewards/CosineReward": 0.04339282959699631, |
| "eval_rewards/RepetitionPenalty": 0.0, |
| "eval_runtime": 1045.0653, |
| "eval_samples_per_second": 0.001, |
| "eval_steps_per_second": 0.001, |
| "step": 24 |
| }, |
| { |
| "clip_ratio": 5.05705434079573e-05, |
| "completion_length": 10789.259765625, |
| "epoch": 6.2105263157894735, |
| "grad_norm": 0.0099335303530097, |
| "kl": 0.1800537109375, |
| "learning_rate": 7.243995901002312e-05, |
| "loss": -0.02097315341234207, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.6171875, |
| "reward": 0.03010205877944827, |
| "reward_std": 0.10742511600255966, |
| "rewards/CosineReward": 0.030102317687124014, |
| "rewards/RepetitionPenalty": -2.580197531187878e-07, |
| "step": 25, |
| "train_speed(iter/s)": 0.000406 |
| }, |
| { |
| "clip_ratio": 4.821802576771006e-05, |
| "epoch": 6.421052631578947, |
| "grad_norm": 0.00989576056599617, |
| "kl": 0.18408203125, |
| "learning_rate": 6.980398830195785e-05, |
| "loss": -0.02103913575410843, |
| "memory(GiB)": 186.69, |
| "step": 26, |
| "train_speed(iter/s)": 0.000421 |
| }, |
| { |
| "clip_ratio": 5.442534347821493e-05, |
| "completion_length": 10197.099609375, |
| "epoch": 6.631578947368421, |
| "grad_norm": 0.00436774967238307, |
| "kl": 0.174560546875, |
| "learning_rate": 6.710100716628344e-05, |
| "loss": -0.03593946248292923, |
| "memory(GiB)": 186.69, |
| "response_clip_ratio": 0.513671875, |
| "reward": 0.04752760287374258, |
| "reward_std": 0.14935147762298584, |
| "rewards/CosineReward": 0.04752839542925358, |
| "rewards/RepetitionPenalty": -7.915698745364352e-07, |
| "step": 27, |
| "train_speed(iter/s)": 0.000408 |
| }, |
| { |
| "clip_ratio": 5.9543880524870474e-05, |
| "epoch": 6.842105263157895, |
| "grad_norm": 0.005277659278362989, |
| "kl": 0.182373046875, |
| "learning_rate": 6.434016163555452e-05, |
| "loss": -0.03595500811934471, |
| "memory(GiB)": 186.69, |
| "step": 28, |
| "train_speed(iter/s)": 0.000422 |
| } |
| ], |
| "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 |
| } |
|
|
