Prhub

#2189 [Doc] Clarify PPO/Critic docs after #1856

原始 PR 作者 coding-famer 合并时间 2026-08-12 13:41 文件变更 4 提交数 1 评论 0 代码增减 +20 / -58

执行摘要

澄清 PPO/Critic 文档:共享 GPU 与 --megatron-config-path

PR body 明确说明要 'Clarify PPO/Critic docs after #1856',通过记录共享 train placement 行为、用 --megatron-config-path 角色覆盖替换过时的 --critic-* CLI 引用。旧文档会误导用户为 critic 额外预留 GPU,并推荐已不再适用的参数,需要与 #1856 之后的实际实现对齐。

值得快速浏览,重点是 PPO 资源分配的新说明和 --megatron-config-path 的角色覆盖方式;纯文档变更,无需精读,但可作为配置 PPO 时的参考手册。

讨论亮点

本 PR 没有 review 评论或讨论线程,属于直接合并的文档修正。

实现拆解

  1. 重写 usage.md 的 PPO 章节(docs/en 与 docs/zh):删除“critic 独立占用一组 GPU”“可通过 --critic-num-nodes / --critic-num-gpus-per-node 配置 critic 资源”的描述,以及 4+4+8=16 GPU 的示例;改为说明 PPO 会创建 actor 与 critic 两套训练进程组但放在同一 train placement group 上、critic 训练规模跟随 actor、当前二者 Megatron 并行拓扑必须一致、PPO 强制 train 侧 offload 使 actor/critic 轮流占用显存。同时将 --critic-load、--critic-save、--critic-lr、--critic-lr-warmup-iters 参数替换为 --megatron-config-path 的 YAML 角色覆盖说明。
  2. 更新 megatron-config.md(docs/en 与 docs/zh):从示例中移除 --critic-num-nodes / --critic-num-gpus-per-node,新增“critic 节点数和每节点 GPU 数由 actor 配置派生”的注意事项;FAQ 中“能否把 --actor-num-nodes 或 --critic-num-gpus-per-node 写进 YAML”改为“能否把资源配置写进 YAML”,并明确资源分配仍由 CLI 控制,YAML 对应字段会被忽略。
  3. 同步中英文内容:两个语言的 usage.md 与 megatron-config.md 改动逐条对应,避免语言版本间信息不一致。
  4. 无测试、配置或部署配套改动:纯文档变更,共 4 个文件,+20/-58。
文件 模块 状态 重要度
docs/en/get_started/usage.md 入门指南 modified 2.93
docs/zh/get_started/usage.md 入门指南 modified 2.93
docs/en/advanced/megatron-config.md 进阶指南 modified 2.14
docs/zh/advanced/megatron-config.md 进阶指南 modified 2.14

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

变更本身风险很低,但需要关注以下几点:

  • 文档与实现的一致性:若后续实现再次调整 critic 资源分配方式(例如恢复独立 GPU 或支持独立扩缩),本 PR 的描述需要同步更新,否则会再次产生误导。
  • 中英文同步:两份 usage.md 与 megatron-config.md 需保持一致,未来单侧修改容易造成信息漂移。
  • 参数兼容性:删除 --critic-* 的文档引用后,用户若仍按旧文档查找这些参数会困惑;应确认这些参数当前是否已彻底移除或仅作兼容保留。

对用户而言,配置 PPO 训练时不再需要为 critic 预留独立 GPU,可避免资源浪费和配置错误;同时明确了 --megatron-config-path 是角色级参数覆盖的推荐接口。对系统无代码影响,对团队来说文档维护成本极低。

文档与行为一致性 中英文同步

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论