pytorch
35be73df
- [FSDP()][Easy] Make `fully_shard()` only `FULL_SHARD` (#88260)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
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
awgu
Committer
pytorchmergebot
Parents
fc743ec0
Loading