Zephon 接入 TorchTitan:中途换 GPU 数训练,全局批次哈希仍一致

josh_wills · x · 2026-10-17

DatologyAI 的数据加载方案 Zephon 宣布与 PyTorch 的 TorchTitan 集成:训练中途改变 GPU 数量时,典型 data loader 会改变数据顺序、进而可能改变结果,而 Zephon 保证每个全局批次完全一致。

在 PyTorch 官方博客的 1B 参数模型测试中,训练从 16 GPU 的 checkpoint 分别在 8、32、64 GPU 上恢复,每个全局批次的 SHA-256 哈希都与原运行完全匹配。

技术要点:

所属事件:Zephon 集成 TorchTitan:换 GPU 数恢复训练全局 batch 一致(5 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →