pytorch
bc4ca4c2 - [FSDP] Fix load_sharded_state_dict FQN mismatches for shared parameters (#86524)

Commit
2 years ago
[FSDP] Fix load_sharded_state_dict FQN mismatches for shared parameters (#86524) `_sharded_pre_load_state_dict_hook()` should calls `_param_fqns()` to ensure shared parameters names are also included. Differential Revision: [D40201304](https://our.internmc.facebook.com/intern/diff/D40201304/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/86524 Approved by: https://github.com/rohan-varma
Author
Committer
Parents
Loading