跳转到内容

检查点保存与恢复

大模型检查点包含分片的参数与优化器状态,规模可达数 TB。本章讨论分布式检查点格式、异步与分层保存、并行策略变化时的重新分片(resharding),以及恢复时间的优化。

  • 每个 rank 只写自己的分片,避免聚合
  • 先写本地 NVMe 再异步上传对象存储
  • 检查点格式应与并行度解耦