pytorch-ssd icon indicating copy to clipboard operation
pytorch-ssd copied to clipboard

Published 20 hours ago • verified publisher icon qfgaohao

Train SSD-Based Object Detection Model using Multiple GPUs

Open NISHANTSHRIVASTAV opened this issue 3 years ago • 0 comments

I haven't tried. I am not sure if using "torch.nn.parallel.data_parallel" will be enough. If you can make it work, please feel free to create a pull request. Thanks!

Originally posted by @qfgaohao in https://github.com/qfgaohao/pytorch-ssd/issues/16#issuecomment-439202444

NISHANTSHRIVASTAV avatar Aug 27 '21 14:08 NISHANTSHRIVASTAV