Pass `context_fn` to `activation_checkpoint_params`#951

Closed

y-sq wants to merge 1 commit intopytorch:masterfrom y-sq:export-D65360604

+18-1

Commits on Dec 3, 2024

Don't recompute scaling factor in activation checkpointing (pytorch#951 )

y-sq
authored and
facebook-github-bot
committed