-
Notifications
You must be signed in to change notification settings - Fork 941
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add FedNova baseline #2179
Add FedNova baseline #2179
Conversation
Hi @ashdtu , how is it going? Let me know if you'd like to discuss something about your |
Hi @jafermarq I have a working version of the baseline now. that is producing results similar to the one in the paper. However there are certain parts that I wanted to refactor in the code that will be a work in progress in the coming week. Request you to take a look at the current state so that we can have a 1-on-1 soon. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @ashdtu,
I have done an initial pass. See some suggestions below. The summary of my comments are:
- (important) please rename your directory (and its subdirectory) from
fedNova
tofednova
. Then don't forget to reflect that in thepyproject.toml
(see comment), and update all the imports so they make use offednova.<your file>
. I was able to run your code after doing it by doingpython -m fednova.main
. - Try to have a minimal version of the readme ready.
- If you make plots, please add them to a directory called
_static
and placed in the same directory as theREADME.md
. - Does
state/init_params_torch.pt
need to be part of the repo? (it's a large file)
Please ping me when you are done making the changes. Happy to discuss them further over here or slack.
2. add logic and refactor ProxSGD and strategy.py to update state parameters in strategy instead of client optimizer 3. get rid of logic to save optimizer params on disk
2. Generate Plots in static
2. add plots in static 3. add temporary results csv
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hey @ashdtu,
Just a couple of small comments about the README.md and the location where the results are stored.
2. utils now uses hydra created directories 3. plotting support more than 2 plots at once and logic for automatic legends 4. results for momentum experiment 5. readme polished with nan update. 6. strategy.py now contains more comments and references.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi @ashdtu,
I did a couple of small changes today. But i'm glad to say FedNova
looks merge-ready !! Many thanks for the hardwork.
FedNova
Proposal for implementation of FedNova baseline to Flower framework under Summer of Reproducibility challenge.