|
| 1 | +# download datasets and models |
| 2 | +# python3 examples/data_preprocess/gsm8k.py |
| 3 | +# python3 examples/data_preprocess/math_dataset.py |
| 4 | +# huggingface-cli download Skywork/Skywork-Reward-V2-Llama-3.2-3B --local-dir $HOME/models/Skywork-Reward-V2-Llama-3.2-3B |
| 5 | +# huggingface-cli download Qwen/Qwen2.5-3B-Instruct --local-dir $HOME/models/Qwen2.5-3B-Instruct |
| 6 | + |
| 7 | +gsm8k_train_path=$HOME/data/gsm8k/train.parquet |
| 8 | +gsm8k_test_path=$HOME/data/gsm8k/test.parquet |
| 9 | +math_train_path=$HOME/data/math/train.parquet |
| 10 | +math_test_path=$HOME/data/math/test.parquet |
| 11 | + |
| 12 | +train_files="['$gsm8k_train_path', '$math_train_path']" |
| 13 | +test_files="['$gsm8k_test_path', '$math_test_path']" |
| 14 | + |
| 15 | +python3 -m verl.trainer.main_ppo \ |
| 16 | + algorithm.adv_estimator=gae \ |
| 17 | + data.train_files="$train_files" \ |
| 18 | + data.val_files="$test_files" \ |
| 19 | + data.train_batch_size=1024 \ |
| 20 | + data.max_prompt_length=1024 \ |
| 21 | + data.max_response_length=2048 \ |
| 22 | + data.filter_overlong_prompts=True \ |
| 23 | + data.truncation='error' \ |
| 24 | + data.return_raw_chat=True \ |
| 25 | + actor_rollout_ref.model.path="$HOME/models/Qwen2.5-3B-Instruct" \ |
| 26 | + actor_rollout_ref.actor.optim.lr=1e-6 \ |
| 27 | + actor_rollout_ref.model.use_remove_padding=True \ |
| 28 | + actor_rollout_ref.actor.optim.lr_warmup_steps_ratio=0.1 \ |
| 29 | + actor_rollout_ref.actor.ppo_mini_batch_size=256 \ |
| 30 | + actor_rollout_ref.actor.ppo_micro_batch_size_per_gpu=16 \ |
| 31 | + actor_rollout_ref.actor.use_kl_loss=False \ |
| 32 | + actor_rollout_ref.model.enable_gradient_checkpointing=True \ |
| 33 | + actor_rollout_ref.actor.fsdp_config.param_offload=False \ |
| 34 | + actor_rollout_ref.actor.fsdp_config.optimizer_offload=False \ |
| 35 | + actor_rollout_ref.rollout.log_prob_micro_batch_size_per_gpu=16 \ |
| 36 | + actor_rollout_ref.rollout.tensor_model_parallel_size=1 \ |
| 37 | + actor_rollout_ref.rollout.name=vllm \ |
| 38 | + actor_rollout_ref.rollout.gpu_memory_utilization=0.6 \ |
| 39 | + critic.optim.lr=1e-5 \ |
| 40 | + critic.model.use_remove_padding=True \ |
| 41 | + critic.optim.lr_warmup_steps_ratio=0.05 \ |
| 42 | + critic.model.path="$HOME/models/Qwen2.5-3B-Instruct" \ |
| 43 | + critic.model.enable_gradient_checkpointing=True \ |
| 44 | + critic.ppo_micro_batch_size_per_gpu=32 \ |
| 45 | + critic.model.fsdp_config.param_offload=False \ |
| 46 | + critic.model.fsdp_config.optimizer_offload=False \ |
| 47 | + reward_model.enable=True \ |
| 48 | + reward_model.model.path="$HOME/models/Skywork-Reward-V2-Llama-3.2-3B" \ |
| 49 | + reward_model.use_reward_loop=True \ |
| 50 | + reward_model.rollout.name=vllm \ |
| 51 | + reward_model.rollout.gpu_memory_utilization=0.8 \ |
| 52 | + reward_model.rollout.tensor_model_parallel_size=1 \ |
| 53 | + reward_model.rollout.prompt_length=4096 \ |
| 54 | + reward_model.rollout.response_length=4096 \ |
| 55 | + reward_model.num_workers=8 \ |
| 56 | + algorithm.use_kl_in_reward=False \ |
| 57 | + trainer.critic_warmup=0 \ |
| 58 | + trainer.logger='["console","wandb"]' \ |
| 59 | + trainer.project_name='verl_test_qwen25_rm' \ |
| 60 | + trainer.val_before_train=False \ |
| 61 | + trainer.experiment_name='reward_loop_colocate_reward_model' \ |
| 62 | + trainer.n_gpus_per_node=8 \ |
| 63 | + trainer.nnodes=1 \ |
| 64 | + trainer.save_freq=-1 \ |
| 65 | + trainer.test_freq=10 \ |
| 66 | + trainer.total_epochs=15 $@ |
0 commit comments