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]