torch.utils.checkpoint.CheckpointFunction Class Reference

Inheritance diagram for torch.utils.checkpoint.CheckpointFunction:

Static Public Member Functions
def	forward (ctx, run_function, preserve_rng_state, args)
def	backward (ctx, args)

Detailed Description

Definition at line 55 of file checkpoint.py.

---

The documentation for this class was generated from the following file:

• torch/utils/checkpoint.py