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.