.
checkpoint(run_function, args, use_reentrant=False) You may refer to the docs for additional arguments and options. We hope that after understanding these details, you will be able to use activation checkpointing more effectively, customize it to your needs and contribute to the PyTorch repository.