← 返回事件
持续讨论AI发版

trunk/7c647ebb2837fd3a9618babc81e3558c64fe69f3: Fix test_ddp_zero_overlap parity check on 4 or more ranks (#197890)

发生了什么

Summary: test_ddp_zero_overlap compares a DDP model whose optimizer step is overlapped via ZeroRedundancyOptimizer against a reference DDP model driven by a local SGD, and asserts the parameters match afterwards. The two models were not reducing gradients the same way. The overlapped model must run 2 warmup iterations (3 with static_graph) because the ZeRO comm hook only starts updating parameters once DDP has rebui…

摘要按规则整理自下方来源原文

为什么在扩散

来源