pytorch
a2ffc3be - [AC] Add trailing "." to `_CHECKPOINT_PREFIX` like FSDP (#87951)

Commit
2 years ago
[AC] Add trailing "." to `_CHECKPOINT_PREFIX` like FSDP (#87951) This is for consistency with FSDP. - `_FSDP_WRAPPED_MODULE` and `_CHECKPOINT_WRAPPED_MODULE` are exactly the wrapped module variable name, meaning you can call `getattr(module, _FSDP_WRAPPED_MODULE)` or `getattr(module, _CHECKPOINT_WRAPPED_MODULE)`. - `_FSDP_PREFIX` and `_CHECKPOINT_PREFIX` include the trailing `"."` and are only used for FQNs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/87951 Approved by: https://github.com/zhaojuanmao
Author
Committer
Parents
Loading