Distributed Optimizers

Warning

Distributed optimizer is not currently supported when using CUDA tensors

.. automodule:: torch.distributed.optim
    :members: DistributedOptimizer, PostLocalSGDOptimizer, ZeroRedundancyOptimizer