cisco-open / flame

flame is a federated learning system for edge with flexibility and scalability at the core of its design.
Apache License 2.0
56 stars 28 forks source link

[Feature] Tensorflow Implementation of FedAdam,FedAdaGrad and FedYogi #175

Closed GaoxiangLuo closed 1 year ago

GaoxiangLuo commented 2 years ago

Is your feature request related to a problem? Please describe. FedAdam,FedAdaGrad and FedYogi are implemented in PyTorch in the Flame system, but not in Tensorflow yet.

Describe the solution you'd like Finish implementation in lib/python/flame/optimizer/fedadagrad.py, lib/python/flame/optimizer/fedadam.py, lib/python/flame/optimizer/fedyogi.py and lib/python/flame/optimizer/fedopt.py.

myungjin commented 1 year ago

PR #320 addresses the issue.