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