Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler 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 Upstream anchor labels

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

API symbolsutils

Copy

View Markdown Download .md

tensorplay.utils.checkpoint.checkpoint_sequential

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

torch.utils.checkpoint.checkpoint_sequential compatibility.

With chunks segments the sequential would be evaluated in chunks with recomputation; TensorPlay runs it as one eager segment.

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

Runs function(*args, **kwargs) eagerly (no recomputation).

tensorplay.utils.checkpoint.set_checkpoint_early_stop

No-op for API parity.