nvflare.app_common.utils.tensor_disk_offload_context module

apply_enable_tensor_disk_offload(engine, enabled: bool, root_dir: str | None = None) → Tuple[Any, bool]

Apply enable_tensor_disk_offload to cell FOBS context.

Returns:

(previous value, applied flag).

restore_enable_tensor_disk_offload(engine, previous_value: Any, root_dir: str | None = None) → None

Restore prior enable_tensor_disk_offload value on a cell.