search

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
MIT License
571 stars 166 forks source link

multi gpu training #5

Closed fengyang95 closed 6 years ago

fengyang95 commented 6 years ago

Hi, I am trying to train the ssds network with multi gpu , but it seems doesn't working. I just use the DataParallel class. Are there any tricks to implent this? Thanks a lot.

xiaollz commented 6 years ago

@wwdguu May I ask you how did you manage to do that? Thx!

isalirezag commented 6 years ago

@wwdguu @xiaollz @foreverYoungGitHub @ShuangXieIrene Can you please tell us how to do it, when i uncomment DataParallel in the code, it runs but it still uses just one gpu