pytorch
55844dfd - [FSDP][state_dict] Restore the state_dict_config for NO_SHARD (#100855)

Commit
1 year ago
[FSDP][state_dict] Restore the state_dict_config for NO_SHARD (#100855) Any change to the user configurations should be temporary. This PR fixes the issue when NO_SHARD state_dict/load_state_dict is called, the state_dict_config and state_dict_type are changed permanently. Differential Revision: [D45593313](https://our.internmc.facebook.com/intern/diff/D45593313/) **NOTE FOR REVIEWERS**: This PR has internal Meta-specific changes or comments, please review them on [Phabricator](https://our.internmc.facebook.com/intern/diff/D45593313/)! Pull Request resolved: https://github.com/pytorch/pytorch/pull/100855 Approved by: https://github.com/awgu, https://github.com/zhaojuanmao, https://github.com/rohan-varma
Author
Committer
Parents
Loading