Train SSD-Based Object Detection Model using Multiple GPUs
NISHANTSHRIVASTAV opened this issue · 0 comments
NISHANTSHRIVASTAV commented
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 #16 (comment)