[megatron, vllm, recipe] feat: add Qwen2.5-1.5B GDPO training on Ascend - #7445
[megatron, vllm, recipe] feat: add Qwen2.5-1.5B GDPO training on Ascend#7445RordChang wants to merge 2 commits into
Conversation
Signed-off-by: RordChang <13223981771@163.com>
There was a problem hiding this comment.
Pull request overview
Adds a canonical GDPO training entry point for Qwen2.5-1.5B on Ascend NPUs (Megatron/MBridge actor+ref, vLLM-Ascend rollout), and introduces CPU tests to validate the GDPO multi-reward wiring and RLLA reward outputs.
Changes:
- Add an Ascend-focused GDPO launcher script (
run_qwen2_5_1_5b_megatron.sh) with environment defaults, logging, and Hydra overrides. - Add CPU unit tests covering GDPO per-reward-dimension normalization, required reward keys, and registry wiring; also validates RLLA score breakdown.
- Update the GDPO trainer README to document the new Megatron + vLLM-Ascend entry point.
Reviewed changes
Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.
| File | Description |
|---|---|
| tests/trainer/ppo/test_gdpo_on_cpu.py | New CPU tests validating GDPO advantage estimator behavior, registry wiring, and RLLA reward outputs. |
| examples/gdpo_trainer/run_qwen2_5_1_5b_megatron.sh | New Ascend NPU GDPO training script integrating Megatron/MBridge + vLLM-Ascend rollout + GDPO reward manager. |
| examples/gdpo_trainer/README.md | Documents the new Ascend entry point and updates dataset/environment notes. |
💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.
| | `run_qwen2_5_1_5b_megatron.sh` | vLLM-Ascend | Megatron | Ascend NPU | | ||
|
|
||
| Prepare a rubric-style dataset (e.g. `rlla_4k`) and point `DATA_DIR` to it. | ||
| Prepare a rubric-style dataset (e.g. `rlla_4k`) and point `DATA_ROOT` to it. |
请完善一下 1. 不需要提交test相关的代码 2. 提交实验数据 |
Remove test-only changes following reviewer feedback while retaining the reported validation results in the PR description. Co-authored-by: OpenAI Codex <noreply@openai.com> Signed-off-by: RordChang <13223981771@163.com>
|
@fangweii1 已按反馈完成修改:test 相关代码已删除,实验数据和完整脱敏训练日志链接已补充至 PR 正文。 |
|
统一先提到verl-ascend-recipe:verl-ascend-recipe哈,此pr暂时关闭 |
What does this PR do?
Adds a canonical Qwen2.5-1.5B-Instruct GDPO training example for Ascend NPU, using Megatron/MBridge for the actor and reference workers and vLLM-Ascend for rollout.
examples/gdpo_trainer/run_qwen2_5_1_5b_megatron.shwith environment overrides, auto resume, timestamped logs, and a 100-step default.accuracy_rewardandformat_rewardcomponents.Related task: verl-project/verl-ascend-recipe#18
Checklist Before Starting
[{modules}] {type}: {description}.Test
pre-commit run --all-files --show-diff-on-failure --color=always: all hooks passed.bash -n examples/gdpo_trainer/run_qwen2_5_1_5b_megatron.sh: passed.Experiment Results
Platform: 4 x Ascend 910B.
Full redacted 100-step training log:
training_100step_sanitized.log
API and Usage Example
DATA_ROOTshould containrlla_4k/train.parquetandrlla_4k/test.parquet. Additional Hydra overrides can be appended to the command.Design & Code Changes
run_qwen2_5_1_5b_megatron.shcomposes the current Megatron model engine, MBridge conversion, vLLM rollout, GDPO estimator, and GDPO reward manager.Checklist Before Submitting
pre-commit install && pre-commit run --all-files --show-diff-on-failure --color=always.ci-requestchannel or the Feishu group.recipesubmodule is not changed, so no submodule reference update is required.AI assistance disclosure: AI tools assisted with implementation, validation, and documentation preparation. The author reviewed the complete diff and takes responsibility for the submission.