Zephon 为 TorchTitan 带来弹性确定性数据加载
Bringing Elastically Deterministic Data Loading to TorchTitan with Zephon
阅读原文
本站未展示全文,请前往来源网站阅读。
AI 导读
PyTorch 团队发布 Zephon,为 TorchTitan 提供弹性确定性数据加载,改变 GPU 数量后恢复训练仍产生与基线完全一致的全局 batch。原文显示标准 data loader 下 DCLM Core v1 分数因 GPU 数量最多波动 0.82 分,Zephon 将波动降至 0.05 分。
来源:PyTorch Blog · pytorch.org