Skip to content

Support Bagua Training Plugin #10455

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.

Sign up for GitHub

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

Jump to bottom
Closed
kaushikb11 opened this issue Nov 10, 2021 · 0 comments · Fixed by #11146 or #11283
Closed

Support Bagua Training Plugin #10455

kaushikb11 opened this issue Nov 10, 2021 · 0 comments · Fixed by #11146 or #11283
Assignees
@kaushikb11
Labels
feature Is an improvement or enhancement strategy

Comments

@kaushikb11
Copy link
Contributor

kaushikb11 commented Nov 10, 2021 •
edited by github-actions bot
Loading

🚀 Feature

Bagua is a deep learning training acceleration framework for PyTorch.
You could read more the supported features and benchmarks here.
Mnist Example.

We could support it by introducing a new BaguaPlugin for training by passing it to the strategy flag

trainer = Trainer(accelerator="gpu", devices=4, strategy="bagua")

cc @Borda @akihironitta
@kaushikb11 kaushikb11 added the feature Is an improvement or enhancement label Nov 10, 2021
@kaushikb11 kaushikb11 self-assigned this Nov 10, 2021
@wangraying wangraying mentioned this issue Dec 18, 2021
Merged
12 tasks
@wangraying wangraying mentioned this issue Dec 31, 2021
Merged
8 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@kaushikb11 kaushikb11

Labels
feature Is an improvement or enhancement strategy
Projects
None yet
Milestone
No milestone
2 participants
@ananthsub @kaushikb11