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

How to train on multi gpu device? #6

Closed
zhangjinsong3 opened this issue Mar 26, 2020 · 2 comments
Closed

How to train on multi gpu device? #6

zhangjinsong3 opened this issue Mar 26, 2020 · 2 comments

Comments

@zhangjinsong3
Copy link

Thanks for your amazing work! I have tried to train the code on my own dataset with multi gpus, but it still works on the first one! And I find that the parameter '--num-gpus' not be used! How did you train on 8 gpus?

@LigZhong
Copy link

@zhangjinsong3 hello, how did you solve this?

@zhangjinsong3
Copy link
Author

zhangjinsong3 commented Apr 27, 2020

@LigZhong I did not solve this, but I found centernet-better can train on multi gpus. Thus I use CenterNet-better instead.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants