pytorch
35be73df - [FSDP()][Easy] Make `fully_shard()` only `FULL_SHARD` (#88260)

Commit
2 years ago
[FSDP()][Easy] Make `fully_shard()` only `FULL_SHARD` (#88260) We can have a separate API for each of the other sharding strategies. Pull Request resolved: https://github.com/pytorch/pytorch/pull/88260 Approved by: https://github.com/mrshenli
Author
Committer
Parents
Loading