pytorch/docs/source/distributed.optim.rst
chronos_secgrp_pytorch_oss_ci_oncall 4670c0042a 2021-11-10 nightly release (6b44e75f6b)
2021-11-10 00:05:00 -08:00

11 lines
307 B
ReStructuredText

.. 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