How to train with multi GPU

naoto0804 / cross-domain-detection

Cross-Domain Weakly-Supervised Object Detection through Progressive Domain Adaptation [Inoue+, CVPR2018].

https://naoto0804.github.io/cross_domain_detection/

422 stars 77 forks source link

How to train with multi GPU #26

Closed LiewFeng closed 4 years ago

LiewFeng commented 4 years ago

Hi, I wonder how to train with multi GPU. When I used three GPU to train, I found that only one GPU was used. My training set is --gpu 2. Looking forward to your reply @naoto0804

naoto0804 commented 4 years ago

Sorry, there's no plan to support multiple GPUs due to limited time for me. I hope the time for the training is not so long;

LiewFeng commented 4 years ago

Thank you for your reply.