Skip to content
GitLab

Create Fed-BioMed Optimizer class wrapping Declearn Optimizer.

Merged ANDREY Paul requested to merge feature/473-declearn-optimizer-wrapper into develop

MR description

Closes #473 (closed)

This MR aims at implementing a new fedbiomed.common.optimizer.Optimizer class that interfaces the declearn-provided optimization features for use in Fed-BioMed, as part of the current effort to redesign part of the TrainingPlan code.

Currently this MR:

  • Implements the new Optimizer class
  • Implements a new FedbiomedOptimizerError class and a dedicated error code (FB620)
  • Adds declearn as a third-party dependency in the conda environments

Task list:

  • implement Optimizer
  • implement unit tests
  • discuss potential design changes based on related work
Edited by ANDREY Paul