nvflare.edge.device.pt.trainer module

class PTTrainer(epoch: int, lr, loss_fn, optimizer, transforms)[source]

Bases: Executor

execute(task_data: DXO, ctx: Context, abort_signal: Signal) → DXO[source]