nvflare.app_opt.lightning.callbacks module¶

class RestoreState[source]¶

Bases: Callback

Callback to restore the local optimizer and learning rate scheduler states at each round of FL

on_fit_end(trainer: Trainer, pl_module: LightningModule)[source]¶: Called when fit ends.

on_fit_start(trainer: Trainer, pl_module: LightningModule)[source]¶: Called when fit begins.