nvflare.edge.device package

Subpackages

• nvflare.edge.device.pt package
  • Submodules
    • nvflare.edge.device.pt.optimizers module
      • AdamOptimizerWrapper
      • SGDOptimizerWrapper
    • nvflare.edge.device.pt.runner module
      • PTRunner
    • nvflare.edge.device.pt.trainer module
      • PTTrainer
  • Module contents

Submodules

• nvflare.edge.device.config module
  • ComponentResolver
    • ComponentResolver.resolve()
  • ConfigError
  • ConfigKey
    • ConfigKey.ARGS
    • ConfigKey.COMPONENTS
    • ConfigKey.EXECUTORS
    • ConfigKey.HANDLERS
    • ConfigKey.IN_FILTERS
    • ConfigKey.NAME
    • ConfigKey.OUT_FILTERS
    • ConfigKey.TRAINER
    • ConfigKey.TYPE
  • TrainConfig
    • TrainConfig.find_executor()
  • process_train_config()
• nvflare.edge.device.config_process_test module
  • BCELossResolver
    • BCELossResolver.resolve()
  • DLTrainer
    • DLTrainer.execute()
  • DPFilter
    • DPFilter.filter()
  • Rotate
    • Rotate.transform()
  • SGDOptimizer
  • StatsKeeper
    • StatsKeeper.filter()
    • StatsKeeper.handle_event()
  • bce_loss()
• nvflare.edge.device.defs module
  • Batch
    • Batch.get_input()
    • Batch.get_label()
  • Context
    • Context.fire_event()
  • ContextKey
    • ContextKey.COMPONENTS
    • ContextKey.DATA_SOURCE
    • ContextKey.EVENT_HANDLERS
    • ContextKey.EXECUTOR
    • ContextKey.RUNNER
    • ContextKey.TASK_DATA
    • ContextKey.TASK_ID
    • ContextKey.TASK_NAME
  • DataSource
    • DataSource.get_dataset()
  • Dataset
    • Dataset.get_next_batch()
    • Dataset.reset()
    • Dataset.size()
  • EventHandler
    • EventHandler.handle_event()
  • EventType
    • EventType.AFTER_TRAIN
    • EventType.BEFORE_TRAIN
    • EventType.LOSS_GENERATED
  • Executor
    • Executor.execute()
  • Filter
    • Filter.filter()
  • Transform
    • Transform.transform()
• nvflare.edge.device.sdk_spec module
  • FlareRunner
    • FlareRunner.add_builtin_resolvers()
    • FlareRunner.run()
    • FlareRunner.stop()

Module contents