pytorch/distributed.optim.rst at 5bcbbf537327f6e8328289c25a3a453a2444d984 - pytorch - Carlos Sousa's Git

OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00

Shen Li 29034b9487 [Reland] Update and expose ZeroRedundancyOptimizer docs (#53112 )

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/53112

Test Plan: Imported from OSS

Reviewed By: blefaudeux

Differential Revision: D26752289

Pulled By: mrshenli

fbshipit-source-id: 897257417b530e6e18788cb40c44e5cb7ac688d5

2021-03-02 14:16:12 -08:00

9 lines

169 B

ReStructuredText

Raw Blame History

 .. role:: hidden
     :class: hidden-section
 Distributed Optimizers
 ======================
 .. autoclass:: torch.distributed.optim.ZeroRedundancyOptimizer
     :members: