vm.torch.utils.checkpoint¶

Classes¶

class no_checkpoint : Context-manager to disable checkpointing.

Functions¶

checkpoint(…) : Call function and create a checkpoint.

checkpoint_sequential(…) : Call functions and create segmental checkpoints.