Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Index Multiprocessing package - tensorplay.multiprocessing Python Module Index Quantization Search tensorplay tensorplay._stax tensorplay.audio tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision

BatchSampler ChainDataset CheckpointPolicy ConcatDataset DataLoader Dataset GraphExecGroup IterableDataset RandomSampler Sampler SelectiveCheckpointContext SequentialSampler StackDataset Subset SubsetRandomSampler TensorDataset tensorplay.utils.checkpoint.checkpoint tensorplay.utils.checkpoint.checkpoint_sequential tensorplay.utils.checkpoint.create_selective_checkpoint_contexts tensorplay.utils.checkpoint.set_checkpoint_debug_enabled tensorplay.utils.checkpoint.set_checkpoint_early_stop tensorplay.utils.checkpoint.set_device_states tensorplay.utils.data.default_collate tensorplay.utils.data.default_convert tensorplay.utils.data.get_worker_info WeightedRandomSampler

API symbolsutils

latest (dev)

latest (dev)2026-09-08 v1.0.02026-09-06

Copy

tensorplay.utils.checkpoint.create_selective_checkpoint_contexts

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

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.utils.checkpoint.checkpoint_sequential

Previous Page

tensorplay.utils.checkpoint.set_checkpoint_debug_enabled

Next Page

On this page

create_selective_checkpoint_contexts