flip.nvflare.components.pt_model_locator

Classes

`PTModelLocator`
`InitialPTModelLocator`
`EvaluationPTModelLocator`

Module Contents

class flip.nvflare.components.pt_model_locator.PTModelLocator(exclude_vars=None, model=None)[source]

Bases: nvflare.app_common.abstract.model_locator.ModelLocator

model = None[source]

exclude_vars = None[source]

get_model_names(fl_ctx: nvflare.apis.fl_context.FLContext) → List[str][source]

locate_model(model_name, fl_ctx: nvflare.apis.fl_context.FLContext) → nvflare.apis.dxo.DXO | None[source]

class flip.nvflare.components.pt_model_locator.InitialPTModelLocator(exclude_vars=None, model=None)[source]

Bases: nvflare.app_common.abstract.model_locator.ModelLocator

model = None[source]

exclude_vars = None[source]

get_model_names(fl_ctx: nvflare.apis.fl_context.FLContext) → List[str][source]

locate_model(model_name, fl_ctx: nvflare.apis.fl_context.FLContext) → nvflare.apis.dxo.DXO | None[source]

class flip.nvflare.components.pt_model_locator.EvaluationPTModelLocator(exclude_vars=None)[source]

Bases: nvflare.app_common.abstract.model_locator.ModelLocator

models = None[source]

exclude_vars = None[source]

locate_model(fl_ctx: nvflare.apis.fl_context.FLContext) → nvflare.apis.dxo.DXO | None[source]