-
Notifications
You must be signed in to change notification settings - Fork 83
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
feat: add async model average algorithm #110
Conversation
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
…s/bagua into feat/async-model-average
update @NOBLES5E It seems using Using the other two protocols I have posted an issue on github, which we could track later on. |
remove nccl proto checkbagua/bagua/torch_api/algorithms/async_model_average.py Lines 14 to 19 in f550bcd
This comment was generated by todo based on a
|
This CI seems broken. |
And do you have any further comment about the usage, since we add a function |
; remove this after NVIDIA/nccl#549 gets solvedbagua/bagua/torch_api/algorithms/async_model_average.py Lines 23 to 28 in 1c33cf9
This comment was generated by todo based on a
|
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.
I refined the texts and left some comments. @wangraying Please do a final check
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
No description provided.