Skip to content

Latest commit

 

History

History
19 lines (13 loc) · 430 Bytes

fsdp.rst

File metadata and controls

19 lines (13 loc) · 430 Bytes
Raw

FullyShardedDataParallel

.. automodule:: torch.distributed.fsdp


.. autoclass:: torch.distributed.fsdp.FullyShardedDataParallel
  :members:


.. autoclass:: torch.distributed.fsdp.BackwardPrefetch
  :members:


.. autoclass:: torch.distributed.fsdp.ShardingStrategy
  :members:


.. autoclass:: torch.distributed.fsdp.MixedPrecision
  :members:


.. autoclass:: torch.distributed.fsdp.CPUOffload
  :members: