# tensorplay.utils.checkpoint API

Source: https://www.tensorplay.cn/docs/api/tensorplay.utils.checkpoint.html

## Functions 6

[#](#api-tensorplay.utils.checkpoint.checkpoint_sequential)

### checkpoint_sequential

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.checkpoint_sequential.html)

```python
tensorplay.utils.checkpoint.checkpoint_sequential(function: Callable[[...], Any], chunks: int, *args: Any, **kwargs: Any) → Any
```

[#](#api-tensorplay.utils.checkpoint.checkpoint)

### checkpoint

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.checkpoint.html)

```python
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
```

[#](#api-tensorplay.utils.checkpoint.create_selective_checkpoint_contexts)

### create_selective_checkpoint_contexts

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.create_selective_checkpoint_contexts.html)

```python
tensorplay.utils.checkpoint.create_selective_checkpoint_contexts(policy_fn_or_list: Callable[[...], Any] | list[Any], allow_cache_entry_mutation: bool = False)
```

[#](#api-tensorplay.utils.checkpoint.set_checkpoint_debug_enabled)

### set_checkpoint_debug_enabled

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.set_checkpoint_debug_enabled.html)

```python
tensorplay.utils.checkpoint.set_checkpoint_debug_enabled(enabled: bool | None)
```

[#](#api-tensorplay.utils.checkpoint.set_checkpoint_early_stop)

### set_checkpoint_early_stop

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.set_checkpoint_early_stop.html)

```python
tensorplay.utils.checkpoint.set_checkpoint_early_stop(enabled: bool)
```

[#](#api-tensorplay.utils.checkpoint.set_device_states)

### set_device_states

function[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.set_device_states.html)

```python
tensorplay.utils.checkpoint.set_device_states(devices: Iterable[int], states: Iterable[Any], *, device_type: str | None = None) → None
```

## Classes 3

[#](#api-tensorplay.utils.checkpoint.CheckpointPolicy)

### CheckpointPolicy

class[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.CheckpointPolicy.html)

```python
class tensorplay.utils.checkpoint.CheckpointPolicy(*values)
```

[#](#api-tensorplay.utils.checkpoint.GraphExecGroup)

### GraphExecGroup

class[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.GraphExecGroup.html)

```python
class tensorplay.utils.checkpoint.GraphExecGroup
```

[#](#api-tensorplay.utils.checkpoint.SelectiveCheckpointContext)

### SelectiveCheckpointContext

class[Full reference ↗](/docs/generated/tensorplay.utils.checkpoint.SelectiveCheckpointContext.html)

```python
class tensorplay.utils.checkpoint.SelectiveCheckpointContext(*, is_recompute: bool, op_output: Any = None)
```