search

VITA-Group / FasterSeg

[ICLR 2020] "FasterSeg: Searching for Faster Real-time Semantic Segmentation" by Wuyang Chen, Xinyu Gong, Xianming Liu, Qian Zhang, Yuan Li, Zhangyang Wang
MIT License
524 stars 107 forks source link

how to use multi-gpu training? #39

Closed tonghaiping closed 4 years ago

tonghaiping commented 4 years ago

how to use multi-gpu training?

chenwydj commented 4 years ago

Hi @tonghaiping!

Thank you for your interest in our work!

To enable the multi-gpu training, you have to at least wrap the nn.Module into nn.DataParallel (which I did not implement here).