pytorch
956bd038 - add ignored_states to FSDP/fully_shard (#102056)

Commit
1 year ago
add ignored_states to FSDP/fully_shard (#102056) Add 'ignored_states' that accepts either a list of ignored_parameters or a list of nn modules for FSDP model wrapper and fully_shard composable APIs, it is recommended to use 'ignored_states' over 'ignored_modules' moving forward Pull Request resolved: https://github.com/pytorch/pytorch/pull/102056 Approved by: https://github.com/awgu
Author
Committer
Parents
Loading