search

datvuthanh / HybridNets

HybridNets: End-to-End Perception Network
MIT License
594 stars 121 forks source link

How to select number of gpus? #51

Closed qiumeng6 closed 2 years ago

qiumeng6 commented 2 years ago

hi, i want to train on multiple gpus with train_DDP.py, but i do not know which param determine the number of gpus. Looking forward to your reply ！ Thank you!

xoiga123 commented 2 years ago

Hi, you can use python3 train_ddp.py --num_gpus 8

happyday-lkj commented 1 year ago

Hi, you can use python3 train_ddp.py --num_gpus 8

which code branch did you use?