nvflare.app_opt.pt.model_persistence_format_manager module

class PTModelPersistenceFormatManager(data: dict, default_train_conf=None)

Bases: object

Manage the format for model persistence.

Parameters:

• data (dict) – either the dictionary mapping variables to values or a dict of dict.

• default_train_conf (dict, optional) – configuration for train. Defaults to None.

Raises:

TypeError – when data is not a dictionary

PERSISTENCE_KEY_META_PROPS = 'meta_props'

PERSISTENCE_KEY_MODEL = 'model'

PERSISTENCE_KEY_TRAIN_CONF = 'train_conf'

static get_persist_model_format()

to_model_learnable(exclude_vars) → ModelLearnable

to_persistence_dict() → dict

update(ml: ModelLearnable)

Update the persistence data with the learned values.

Parameters:

ml (ModelLearnable) – updated information to be merged into existing ModelLearnable