LLM Activation Checkpointing 生产实战:用 Selective Recompute、非重入模式与 RNG 一致性压低训练显存
大模型训练中,激活值常比参数更早耗尽显存。本文深入探讨选择性重计算、非重入模式、随机数状态一致性及工程门禁策略,帮助团队在保证梯度正确的前提下降低峰值显存,控制训练吞吐损失。
共 2 篇文章
大模型训练中,激活值常比参数更早耗尽显存。本文深入探讨选择性重计算、非重入模式、随机数状态一致性及工程门禁策略,帮助团队在保证梯度正确的前提下降低峰值显存,控制训练吞吐损失。
大模型推理服务常在显存尚未耗尽时因碎片化触发 OOM。本文结合 Memory Snapshot、动态 Shape、分配器调优和 vLLM 显存预算,给出可观测、可验证、可回滚的生产治理方案。