tensorplay.utils.checkpoint.checkpoint

tensorplay.utils.checkpoint.checkpoint(function: Callable[[...], Any] | None = None, *args: Any, use_reentrant: bool | None = None, preserve_rng_state: bool = True, context_fn: Callable[[], Any] | None = None, determinism_check: str = 'default', debug: bool = False, early_stop: bool | None = None, respect_saved_tensors_hooks: bool | None = None, **kwargs: Any) → Any[source]