flip.nvflare.controllers.init_training

Classes

InitTraining

Module Contents

class flip.nvflare.controllers.init_training.InitTraining(model_id: str = '', min_clients: int = FlipConstants.MIN_CLIENTS, flip: flip.FLIP = FLIP(), cleanup_timeout: int = 600)

Bases: nvflare.apis.impl.controller.Controller

_model_id_fallback = ''

_model_id: str | None = None

_min_clients

flip

_cleanup_timeout = 600

_resolve_model_id(fl_ctx: nvflare.apis.fl_context.FLContext) → str

start_controller(fl_ctx: nvflare.apis.fl_context.FLContext) → None

control_flow(abort_signal: nvflare.apis.signal.Signal, fl_ctx: nvflare.apis.fl_context.FLContext) → None

stop_controller(fl_ctx: nvflare.apis.fl_context.FLContext) → None

process_result_of_unknown_task(client: nvflare.apis.client.Client, task_name: str, client_task_id: str, result: nvflare.apis.shareable.Shareable, fl_ctx: nvflare.apis.fl_context.FLContext) → None

_set_init_training_status(fl_ctx: nvflare.apis.fl_context.FLContext) → None

_check_abort_signal(fl_ctx: nvflare.apis.fl_context.FLContext, abort_signal: nvflare.apis.signal.Signal) → bool

_process_cleanup_result(client_task: nvflare.apis.controller_spec.ClientTask, fl_ctx: nvflare.apis.fl_context.FLContext) → None

_accept_cleanup_result(client_name: str, result: nvflare.apis.shareable.Shareable, fl_ctx: nvflare.apis.fl_context.FLContext) → bool | None