学习记录共 1 篇
技术笔记

DeepSpeed 与 ZeRO 训练显存到底花在哪里

从参数、梯度和 Adam 状态拆开训练显存,用四卡分工理解 ZeRO 1、2、3,并通过一张理论估算图看清节省来自哪里。

大模型系统 · 9 分钟阅读阅读全文 ↗