S-DCNet icon indicating copy to clipboard operation
S-DCNet copied to clipboard
verified publisher icon dmburd

Training on multiple GPUs?

Open enyuc opened this issue 5 years ago • 1 comments

Can I train this model with multiple GPUs?

enyuc avatar Aug 17 '20 14:08 enyuc

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.)

dmburd avatar Aug 19 '20 11:08 dmburd