Skip to content
简体中文

切换主题

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

ZeRO / FSDP 数据并行

把优化器状态、梯度和参数切片到各卡,消除数据并行的冗余。

ZeRO / FSDP (Sharded Data Parallelism) 2019 高级 核心

通俗解释

普通数据并行每张卡都存一整份模型和优化器状态。ZeRO 分三级把它们切开,需要时再通信取回,单卡显存需求随卡数下降。PyTorch FSDP 是同类实现。

用在哪里:DeepSpeed、PyTorch FSDP

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

2017 混合精度与 FP8 训练2019 ZeRO / FSDP 数据并行

01原始论文

ZeRO: Memory Optimizations Toward Training Trillion Parameter Models

Samyam Rajbhandari, Jeff Rasley, Olatunji Ruwase, et al. · 2019

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

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

先学

ZeRO / FSDP 数据并行

学完可以学

  • 图谱里暂无后继

相关:张量并行(Megatron)流水线并行

混合精度与 FP8 训练张量并行(Megatron)流水线并行 ZeRO / FSDP 数据并行

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

03学习资料

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

04在课程里哪里学

所在学习路线:从零理解一个大模型

给 AI 读 / 数据说明

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