nvflare.edge.device.defs module
-
class Batch[source]
Bases: object
-
get_input() → Any[source]
-
get_label() → Any[source]
-
class Context[source]
Bases: dict
-
fire_event(event_type: str, data: Any, abort_signal: Signal)[source]
-
class ContextKey[source]
Bases: object
-
COMPONENTS = 'components'
-
DATA_SOURCE = 'data_source'
-
EVENT_HANDLERS = 'event_handlers'
-
EXECUTOR = 'executor'
-
RUNNER = 'runner'
-
TASK_DATA = 'task_data'
-
TASK_ID = 'task_id'
-
TASK_NAME = 'task_name'
-
class DataSource[source]
Bases: ABC
-
abstract get_dataset(dataset_type: str, ctx: Context) → Dataset[source]
-
class Dataset[source]
Bases: ABC
-
abstract get_next_batch(batch_size) → Batch[source]
-
abstract reset()[source]
-
abstract size() → int[source]
-
class EventHandler[source]
Bases: ABC
-
abstract handle_event(event_type: str, event_data: Any, ctx: Context, abort_signal: Signal)[source]
-
class EventType[source]
Bases: object
-
AFTER_TRAIN = 'after_train'
-
BEFORE_TRAIN = 'before_train'
-
LOSS_GENERATED = 'loss_generated'
-
class Executor[source]
Bases: ABC
-
abstract execute(task_data: DXO, ctx: Context, abort_signal: Signal) → DXO[source]
-
class Filter[source]
Bases: ABC
-
abstract filter(data: DXO, ctx: Context, abort_signal: Signal) → DXO[source]
-
class Transform[source]
Bases: ABC
-
abstract transform(batch: Batch, ctx: Context, abort_signal: Signal) → Batch[source]