XinyiYS / Gradient-Driven-Rewards-to-Guarantee-Fairness-in-Collaborative-Machine-Learning

Official code repository for our accepted work "Gradient Driven Rewards to Guarantee Fairness in Collaborative Machine Learning" in NeurIPS'21.
https://proceedings.neurips.cc/paper/2021/hash/8682cc30db9c025ecd3fee433f8ab54c-Abstract.html
MIT License
21 stars 5 forks source link

Fixed possible indentation error in the training loop in utils/utils.py. #3

Open JLA0809 opened 4 days ago

JLA0809 commented 4 days ago

I think there could be an indendation error in the main training loop in the utils/utils.py in the original main branch. In my opinion the optimizer has to be called for every batch, in order to use the information of the whole training dataset.