-
Notifications
You must be signed in to change notification settings - Fork 23
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
How to train using a single GPU? #36
Comments
Thanks for your interest! For training on a machine equipped with single GPU, please try to run the following command:
You can increase batch size if your GPU memory is enough. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hello author, thank you for your excellent work.
Your code seems to be for multi-card distributed training. I only have one GPU. Could you please tell me where to change it for single-card training, so that it can be debugged with a single card.
Looking forward to your reply.
The text was updated successfully, but these errors were encountered: