flip.nvflare.executors
FLIP Executors module containing NVFLARE executor wrappers.
These executors wrap user-provided training, validation, and evaluation logic.
- Exports:
RUN_TRAINER: Wrapper executor for user’s FLIP_TRAINER class
RUN_VALIDATOR: Wrapper executor for user’s FLIP_VALIDATOR class
RUN_EVALUATOR: Wrapper executor for user’s FLIP_EVALUATOR class
Submodules
Classes
Executes the uploaded evaluator and handles any errors. |
|
Wrapper executor that runs user-provided FLIP_TRAINER implementations. |
|
Wrapper executor that runs user-provided FLIP_VALIDATOR implementations. |
Package Contents
- class flip.nvflare.executors.RUN_EVALUATOR(evaluate_task_name=PTConstants.EvalTaskName, project_id='', query='')[source]
Bases:
nvflare.apis.executor.ExecutorExecutes the uploaded evaluator and handles any errors.
- class flip.nvflare.executors.RUN_TRAINER(train_task_name=AppConstants.TASK_TRAIN, submit_model_task_name=AppConstants.TASK_SUBMIT_MODEL, exclude_vars=None, project_id='', query='')[source]
Bases:
nvflare.apis.executor.ExecutorWrapper executor that runs user-provided FLIP_TRAINER implementations.
This executor handles: - Dynamic importing of the user’s FLIP_TRAINER class - Error handling and exception logging - Lazy initialization of the trainer instance
Initialize the RUN_TRAINER executor.
- Parameters:
train_task_name – Task name for train task. Defaults to “train”.
submit_model_task_name – Task name for submit model. Defaults to “submit_model”.
exclude_vars – List of variables to exclude during model loading.
project_id – The ID of the project the model belongs to.
query – The cohort query that is associated with the project.
- execute(task_name: str, shareable: nvflare.apis.shareable.Shareable, fl_ctx: nvflare.apis.fl_context.FLContext, abort_signal: nvflare.apis.signal.Signal) nvflare.apis.shareable.Shareable[source]
Execute the training task.
This method: 1. Lazily imports and initializes the user’s FLIP_TRAINER 2. Delegates execution to the user’s trainer 3. Catches and reports any exceptions
- Parameters:
task_name – The name of the task to execute
shareable – The input shareable data
fl_ctx – The FL context
abort_signal – Signal for aborting the task
- Returns:
Shareable – The result of the training task
- class flip.nvflare.executors.RUN_VALIDATOR(validate_task_name=AppConstants.TASK_VALIDATION, project_id='', query='')[source]
Bases:
nvflare.apis.executor.ExecutorWrapper executor that runs user-provided FLIP_VALIDATOR implementations.
This executor handles: - Dynamic importing of the user’s FLIP_VALIDATOR class - Error handling and exception logging - Lazy initialization of the validator instance
Initialize the RUN_VALIDATOR executor.
- Parameters:
validate_task_name – Task name for validation task. Defaults to “validate”.
project_id – The ID of the project the model belongs to.
query – The cohort query that is associated with the project.
- execute(task_name: str, shareable: nvflare.apis.shareable.Shareable, fl_ctx: nvflare.apis.fl_context.FLContext, abort_signal: nvflare.apis.signal.Signal) nvflare.apis.shareable.Shareable[source]
Execute the validation task.
This method: 1. Lazily imports and initializes the user’s FLIP_VALIDATOR 2. Delegates execution to the user’s validator 3. Catches and reports any exceptions
- Parameters:
task_name – The name of the task to execute
shareable – The input shareable data
fl_ctx – The FL context
abort_signal – Signal for aborting the task
- Returns:
Shareable – The result of the validation task