跳到正文
PyTorch Blog· Josh Wills, Maximilian Böther, Ties Robroek and the Zephon team·原文发布· 2 小时前AI 评分45

在 TorchTitan 中通过 Zephon 实现弹性确定性数据加载

Bringing Elastically Deterministic Data Loading to TorchTitan with Zephon

AI 导读

PyTorch 团队推出 Zephon,为 TorchTitan 带来弹性确定性数据加载:从 16 GPU 的 step 150 checkpoint 恢复到 8、32、64 GPU 时,损失曲线与基线完全重合,每个 global batch 的 SHA-256 哈希一致。

本站提供 AI 导读,完整报道请前往来源网站阅读。

阅读原文 PyTorch Blog · pytorch.org