pytorch
c9eb8d8d - Add set_checkpoint_debug_enabled that overrides local setting (#110728)

Commit
1 year ago
Add set_checkpoint_debug_enabled that overrides local setting (#110728) People access activation checkpoint through many layers of config and it is not always guaranteed that all the layers of wrapping around checkpoint properly propagate all the kwargs, e.g. debug mode. This context manager offers an alternative way to enable debug mode that bypasses the need for all layers to propagate kwargs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/110728 Approved by: https://github.com/albanD ghstack dependencies: #110673, #110674, #110675, #110676
Author
Committer
Parents
Loading