nvflare.app_opt.pt.decomposers module

class SerializationModule(tensor)[source]

Bases: Module

Initialize internal Module state, shared by both nn.Module and ScriptModule.

class TensorDecomposer[source]

Bases: ViaDownloaderDecomposer

download(from_fqcn: str, ref_id: str, per_request_timeout: float, cell: Cell, secure=False, optional=False, abort_signal=None) → Tuple[str, dict | LazyTensorDict][source]
get_download_dot() → int[source]

Get the Datum Object Type to be used for download ref datum

Returns: the DOT for download ref datum

native_decompose(target: Tensor, manager: DatumManager | None = None) → bytes[source]
native_recompose(data: bytes, manager: DatumManager | None = None) → Tensor[source]
supported_type()[source]

Returns the type/class supported by this decomposer.

Returns:

The class (not instance) of supported type

to_downloadable(items: dict, max_chunk_size: int, fobs_ctx: dict) → Downloadable[source]

Convert the items Downloadable object.

Parameters:

  • items – a dict of items of target object type to be converted

  • max_chunk_size – max size of one chunk.

  • fobs_ctx – FOBS Context

Returns: a Downloadable object

The “items” is a dict of target objects. The dict contains all objects of the target type in one payload.