Issue about Multi-GPU training

IDEA-Research / MaskDINO

[CVPR 2023] Official implementation of the paper "Mask DINO: Towards A Unified Transformer-based Framework for Object Detection and Segmentation"

Apache License 2.0

1.21k stars 109 forks source link

Open RuizhouLiu opened 10 months ago

RuizhouLiu commented 10 months ago

Hi there, I was stuck with an issue that training mask dino model on multiple gpus is too slow.

I train the mask dino on my own dataset, but the speed is too slow on multi-gpus than single-gpu, I don't know why. Is there any solutions?

urban-drummer commented 3 months ago

需要设置成ddp模式并为它指定合适的参数

EunNamCho commented 2 weeks ago

Did you solve it?