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.

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 support for GPU execution #4

Open
chiragnagpal opened this issue Oct 27, 2020 · 0 comments
Open

Add support for GPU execution #4

chiragnagpal opened this issue Oct 27, 2020 · 0 comments
Labels
enhancement New feature or request

Comments

@chiragnagpal
Copy link
Collaborator

Currently the torch implementations use CPUs, it should be easy to change the API to allow for GPU training.

@chiragnagpal chiragnagpal added the enhancement New feature or request label Oct 27, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant