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

Can it be used with torch's dataparallel to train model on multi-gpus? #7

Open
poisson-AI opened this issue May 1, 2019 · 6 comments

Comments

@poisson-AI
Copy link

I want to know how I can use it with dataparallel. What should I change in my code when I build my model on several gpus?

@LemonMi
Copy link

LemonMi commented Dec 31, 2019

Have you solved this problem? I meet the same issue... i'm wondering If you could tell me how you deal with it?

@poisson-AI
Copy link
Author

poisson-AI commented Jan 5, 2020 via email

@LemonMi
Copy link

LemonMi commented Jan 5, 2020

@guobaisong 哈哈是同胞。。 我目前是把所有用到的网络都放到nn.DataParallel()里了,运行倒是没问题,就是不知道对不对。如果方便的话,可能得麻烦你有空找一下,非常感谢

@poisson-AI
Copy link
Author

poisson-AI commented Jan 5, 2020 via email

@LemonMi
Copy link

LemonMi commented Jan 5, 2020

@guobaisong 好的好的~谢谢

@poisson-AI
Copy link
Author

poisson-AI commented Jan 5, 2020 via email

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