nvflare.edge.device.config_process_test module

class BCELossResolver(t, name, args)

Bases: ComponentResolver

resolve()

Resolve the component spec and create device-native object.

Returns: a device-native object or None if failed.

class DLTrainer(**kwargs)

Bases: Executor

execute(model, ctx: Context, abort_signal: Signal)

class DPFilter(**kwargs)

Bases: Filter

filter(model: Any, ctx: Context, abort_signal: Signal) → Any

class Rotate(**kwargs)

Bases: Transform

transform(batch: Batch, ctx: Context, abort_signal: Signal) → Batch

class SGDOptimizer(**kwargs)

Bases: object

class StatsKeeper(**kwargs)

Bases: Filter, EventHandler

filter(model, ctx: Context, abort_signal: Signal)

handle_event(event_type: str, event_data, ctx: Context, abort_signal: Signal)

bce_loss(pred, label)