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.

Sign up for GitHub

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

Jump to bottom

Training on multiple GPUs? #7

Open
cry233 opened this issue Aug 17, 2020 · 1 comment
Open

Training on multiple GPUs? #7

cry233 opened this issue Aug 17, 2020 · 1 comment

Comments

@cry233
Copy link

cry233 commented Aug 17, 2020

Can I train this model with multiple GPUs?
@dmburd
Copy link
Owner

dmburd commented Aug 19, 2020 •
edited
Loading

Yes, you can do that by wrapping the model into nn.DataParallel. Check out the parallelism tutorials: 1, 2.
(Currently the code in this repo does not use nn.DataParallel.)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@cry233 @dmburd