Skip to content
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

train with GPU #165

Closed
kvnsng opened this issue Jun 23, 2020 · 2 comments
Closed

train with GPU #165

kvnsng opened this issue Jun 23, 2020 · 2 comments

Comments

@kvnsng
Copy link

kvnsng commented Jun 23, 2020

Hi,

Just wondering if the current version supports training models with GPU as well.
@interesaaat
Copy link
Collaborator

Hi @songkevinc! We have explored the idea of training in a research paper but Hummingbird is currently only for inference. The GEMM implementation is really close to what we tried in the paper so in theory it should be not hard to add training.

@kvnsng
Copy link
Author

kvnsng commented Jun 23, 2020

Ah, gotcha. Sounds good! Looking forward to the development.

@kvnsng kvnsng closed this as completed Jun 23, 2020
@ksaur ksaur mentioned this issue Jan 3, 2024
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@interesaaat @kvnsng