search

yekeren / Cap2Det

Implementation of our ICCV 2019 paper "Cap2Det: Learning to AmplifyWeak Caption Supervision for Object Detection"
Apache License 2.0
29 stars 9 forks source link

multiple gpu support ? #24

Open junx1992 opened 4 years ago

junx1992 commented 4 years ago

I think it is a really good work. Does the code consider multiple gpu training support ?

yekeren commented 4 years ago

Here is an example using the distributed training strategy: https://github.com/yekeren/Cap2Det/blob/cap2det/train_wsod.sh. However, someone posted an issue saying it did not work.

Another solution is to mimic https://github.com/yekeren/VCR-pretraining/blob/cea56445ef7598091a2a49e9ee7c465a00466f56/modeling/trainer.py#L199 to use mirrored strategy. The problem is that you need to look for a good learning rate if the mirrored strategy is used.

On Wed, Jun 3, 2020 at 3:49 AM Jun Xu notifications@github.com wrote:

I think it is a really good work. Does the code consider multiple gpu training support ?

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/yekeren/Cap2Det/issues/24, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA6CPAIUWGQZYRGNG63WR4DRUX6CNANCNFSM4NRNGT3Q .

-- Thanks, best regards.

Keren

junx1992 commented 4 years ago

Here is an example using the distributed training strategy: https://github.com/yekeren/Cap2Det/blob/cap2det/train_wsod.sh. However, someone posted an issue saying it did not work. Another solution is to mimic https://github.com/yekeren/VCR-pretraining/blob/cea56445ef7598091a2a49e9ee7c465a00466f56/modeling/trainer.py#L199 to use mirrored strategy. The problem is that you need to look for a good learning rate if the mirrored strategy is used. … On Wed, Jun 3, 2020 at 3:49 AM Jun Xu @.***> wrote: I think it is a really good work. Does the code consider multiple gpu training support ? — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#24>, or unsubscribe https://github.com/notifications/unsubscribe-auth/AA6CPAIUWGQZYRGNG63WR4DRUX6CNANCNFSM4NRNGT3Q . -- Thanks, best regards. Keren I will try the methods first. Thanks for your kind reply !