跳到正文
原文
PyTorch· Josh Wills, Maximilian Böther, Ties Robroek and the Zephon team·· 6 小时前精选AI 评分70

PyTorch 发布 Zephon:为 TorchTitan 带来弹性确定性数据加载

Bringing Elastically Deterministic Data Loading to TorchTitan with Zephon

AI 导读

PyTorch 团队联合 Zephon 推出弹性确定性数据加载器 Zephon,集成至 TorchTitan 框架。该工具允许在训练过程中实时进行数据混合、打包和洗牌,无需预生成磁盘副本,从而大幅降低存储开销。

推荐理由

Zephon 通过弹性确定性数据加载消除 GPU 数量对实验结果的干扰,并实现训练时动态数据处理,为大规模模型训练提供了可复现且高效的数据管线方案。

来源:PyTorch · pytorch.org