search

litian96 / FedProx

Federated Optimization in Heterogeneous Networks (MLSys '20)
MIT License
648 stars 158 forks source link

PGGD #33

Open sotirisnik opened 2 years ago

sotirisnik commented 2 years ago

Could you clarify how do you calculate the avg_gradient that is given as input in the pggd file? I could not find the script that imports this file. https://github.com/litian96/FedProx/blob/master/flearn/optimizer/pggd.py#L56 Thank you

litian96 commented 2 years ago

Thanks for the question. FedProx does not use this optimizer. This is instead used in our FedDANE paper, where the updates correspond to Line 7 in Algorithm 2 in our paper. There is some redundancy when we clean up the code and release it.