Go
pytorch
640a96df
- [FSDP][Easy] Allow `ModuleWrapPolicy` to take `Iterable` (#104999)
Commit
View On
GitHub
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
1 year ago
[FSDP][Easy] Allow `ModuleWrapPolicy` to take `Iterable` (#104999) Pull Request resolved: https://github.com/pytorch/pytorch/pull/104999 Approved by: https://github.com/rohan-varma ghstack dependencies: #104427, #104967
Author
awgu
Committer
pytorchmergebot
Parents
031ce0fa
Loading