nvflare.edge.device.config_process_test module

class BCELossResolver(t, name, args)[source]

Bases: ComponentResolver

resolve()[source]

Resolve the component spec and create device-native object.

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

class DLTrainer(**kwargs)[source]

Bases: Executor

execute(model, ctx: Context, abort_signal: Signal)[source]

class DPFilter(**kwargs)[source]

Bases: Filter

filter(model: Any, ctx: Context, abort_signal: Signal) → Any[source]

class Rotate(**kwargs)[source]

Bases: Transform

transform(batch: Batch, ctx: Context, abort_signal: Signal) → Batch[source]

class SGDOptimizer(**kwargs)[source]: Bases: object

class StatsKeeper(**kwargs)[source]

Bases: Filter, EventHandler

filter(model, ctx: Context, abort_signal: Signal)[source]

handle_event(event_type: str, event_data, ctx: Context, abort_signal: Signal)[source]

bce_loss(pred, label)[source]