pytorch
eb81e7ec - [FSDP] Avoid printing incorrect warning for _get_param_to_fqns (#94494)

Commit
1 year ago
[FSDP] Avoid printing incorrect warning for _get_param_to_fqns (#94494) There exist a hack for `_get_param_to_fqns` and `_apply_to_modules`. The condition for the warning of the hack is incorrect and result in overwhelming message for users. This PR fixes the issue. The original hack is not removed. It will once the support of DMP + FSDP is deprecated. Differential Revision: [D43135611](https://our.internmc.facebook.com/intern/diff/D43135611/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/94494 Approved by: https://github.com/rohan-varma
Author
Committer
Parents
Loading