trunk/7c647ebb2837fd3a9618babc81e3558c64fe69f3: Fix test_ddp_zero_overlap parity check on 4 or more ranks (#197890)
- PyTorch 近 90 天出现 1319 次
- PyTorch 近 90 天第 1304 次发版
- 上一次:同一天稍早 · trunk/618de55dfddff78055974a180bfc5b60c5c81d07
发生了什么
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…
摘要按规则整理自下方来源原文