Skip to content
简体中文

切换主题

// ▲ 预训练、缩放与分布式训练

流水线并行

把模型按层切段放到不同卡上,像流水线一样处理微批次。

Pipeline Parallelism (GPipe) 2018 高级 核心

通俗解释

每张卡负责若干层,一个 batch 被切成多个微批次依次流过,减少设备空闲(“气泡”)。与张量并行、数据并行组合成“3D 并行”。

用在哪里:超大模型训练

时间轴 · 前置 → 当前 → 后继

2018 流水线并行2019 张量并行(Megatron)

01原始论文

GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism

Yanping Huang, Youlong Cheng, Ankur Bapna, et al. · 2018

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

流水线并行

学完可以学

  • 图谱里暂无后继

相关:ZeRO / FSDP 数据并行

张量并行(Megatron)ZeRO / FSDP 数据并行 流水线并行

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)