checkpoint

dragon.vm.torch.utils.checkpoint.checkpoint(
  function,
  *args,
  **kwargs
)[source]

Apply function and create a checkpoint.

Parameters:
  • function (callable) – The function to apply.
Returns:

Any – The function outputs.