pytorch/docs/source/fsdp.rst at 8d3e3ebc583eb3a4ca230fa41f06d8ef5d453d63 - saymrwulf/pytorch - Forgejo: Beyond coding. We forge.

saymrwulf/pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00

Yanli Zhao 2336571cb7 make fsdp folder to be public (#72084 )

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

make fsdp folder to be public
ghstack-source-id: 148173447

Test Plan: unit tests

Reviewed By: mrshenli

Differential Revision: D33903417

fbshipit-source-id: 7852a2adc4af09af48a5ffa52ebf210489f834d5
(cherry picked from commit bd06513cfe2f391941bb0afa611dd39994585513)

2022-02-02 15:50:14 +00:00

7 lines

166 B

ReStructuredText

Raw Blame History

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