fl_sim.optimizers

fl_sim.optimizers#

This module contains the optimizers (local solvers) used in federated learning. Despite optimizers from torch.optim and torch_optimizer, we also provide some custom optimizers for federated learning for solving for example

  • proximal optimization problem

  • lagrangian dual problem

get_optimizer(optimizer_name, params, config)

Get optimizer by name.

register_optimizer([name, override])

Decorator to register a new optimizer.