Create Fed-BioMed Optimizer class to wrap Declearn OptiModules and Regularization modules
Optimizer class of "Declearn" is designed to serve the needs for Declearn
and it is developed specific the declearn.Model
class that requires specific methods to get model weights and gradients and other export or import operations. The purpose of creating custom Optimizer class that wraps "Declearn" optimization modules is to discard the interconnection between model and optimizer. Also, what is actually missing in Fed-BioMed is OptiModules provided by declearn.
Note: Please update the issue by checking the updates that is done by Declearn about get_state
and set_state
methods.
Example usage of the class: