pytorch/distributed.optim.rst at dba94873246ba71c40fbe98e6a06e23d6d59ee86 - pytorch - Gitea: Git for Me

frozenleaves/pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-11-05 00:14:54 +08:00

Files

James Reed eaf0457eef [distributed][docs] Delete distributed optimimzer section from RPC and add reference to namespace docs page (#68068 )

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

cc pietern mrshenli pritamdamania87 zhaojuanmao satgera rohan-varma gqchen aazzolini osalpekar jiayisuse SciPioneer H-Huang

Test Plan: Imported from OSS

Reviewed By: pritamdamania87

Differential Revision: D32286554

Pulled By: jamesr66a

fbshipit-source-id: a43fe1f0cfa74721f467b128f2e878bd02f32546

2021-11-09 15:01:54 -08:00

12 lines

307 B

ReStructuredText

Raw Blame History

 .. role:: hidden
     :class: hidden-section
 Distributed Optimizers
 ======================
 .. warning ::
     Distributed optimizer is not currently supported when using CUDA tensors
 .. automodule:: torch.distributed.optim
     :members: DistributedOptimizer, PostLocalSGDOptimizer, ZeroRedundancyOptimizer