16 lines
321 B
Markdown
16 lines
321 B
Markdown
```{eval-rst}
|
|
.. role:: hidden
|
|
:class: hidden-section
|
|
```
|
|
|
|
# Distributed Optimizers
|
|
|
|
:::{warning}
|
|
Distributed optimizer is not currently supported when using CUDA tensors
|
|
:::
|
|
|
|
```{eval-rst}
|
|
.. automodule:: torch.distributed.optim
|
|
:members: DistributedOptimizer, PostLocalSGDOptimizer, ZeroRedundancyOptimizer
|
|
```
|