-
Notifications
You must be signed in to change notification settings - Fork 385
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
Device flag for training #389
Labels
enhancement
New feature or request
Comments
this is added to the github version will be added to the pip soon |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
It would be great to have a flag for selecting different GPUs in a multi-GPU system via the cli for training. I don't think this should be too difficult as long as everything gets transferred correctly onto the same GPU via Torch (not sure how this would impact MXNet).
The text was updated successfully, but these errors were encountered: