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.