pose-hg-train icon indicating copy to clipboard operation
pose-hg-train copied to clipboard

Published 20 hours ago • verified publisher icon princeton-vl

Add choice to use multi-gpu (data parallel)

Open fiarabbit opened this issue 7 years ago • 2 comments

The original code just occupy multiple GPUs but actually did use only one GPU. I modified model.lua and made it possble to use multi-GPU by introducing nn.DataParallelTable.

fiarabbit avatar Apr 10 '17 10:04 fiarabbit

Thanks for the work on multi-gpu training, but I find that it doesn't reduce the training time, why is this happening?

skyz8421 avatar Jun 03 '17 12:06 skyz8421

I'm very sorry to report that 'nngraph' library is not compatible to multi-gpu mode, so the modification does not work. I would like you to close this pull request.

fiarabbit avatar Jun 05 '17 03:06 fiarabbit