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.set_device_states

tensorplay.utils.checkpoint.set_device_states(devices: Iterable[int], states: Iterable[Any], *, device_type: str | None = None) → None[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.set_checkpoint_early_stop

Previous Page

tensorplay.utils.data.default_collate

Take in a batch of data and put the elements within the batch into a tensor with an additional outer dimension - batch size.

On this page

set_device_states