Adds Optimizer support and local epoch count for federated simulations.

googleinterns / fednsp

Apache License 2.0

0 stars 2 forks source link

Adds Optimizer support and local epoch count for federated simulations. #7

Closed derikclive closed 4 years ago

derikclive commented 4 years ago

PR adds support for the Adam, AdaGrad and SGD with momentum for the federated simulations experiments. Also, support for modifying the number of local epochs per federated round has been added. Util.py file has been split into 2 files one for data utilities and the other for model utilities.