flip.nvflare.components.cleanup

Classes

CleanupImages

CleanupImages takes place at the start and end of the run.

Module Contents

class flip.nvflare.components.cleanup.CleanupImages

Bases: nvflare.apis.executor.Executor

CleanupImages takes place at the start and end of the run. All the images used for the training are deleted to prevent the build-up of unnecessary files on the storage space. Executing at the start of a run ensures that any training code is executed with a clean slate.

Args:

Raises:

execute(task_name: str, shareable: nvflare.apis.shareable.Shareable, fl_ctx: nvflare.apis.fl_context.FLContext, abort_signal: nvflare.apis.signal.Signal)