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. |
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
|
NOBLES5E
left a comment
There was a problem hiding this comment.
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.