ovdet icon indicating copy to clipboard operation
ovdet copied to clipboard

GPU and Batch Size

Open krbuettner opened this issue 2 years ago • 6 comments

Hello I was wondering if there is information on the GPU count, batch size, and GPU type for the results reported in the paper. Thanks!

krbuettner avatar Jun 07 '23 23:06 krbuettner

By defaul we used 8 GPUs and set per-gpu batchsize as 2 for all experiments in the paper. However, when implementing the detector on MMDet3.x for the code release, we encounted issue on the training speed when using SyncBN. We sidestep this issue by using more GPUS (16), adjust the learning rate following the linear scaling rule, and train half of the original iterations.

For the configs that are not using SyncBN, we retain the default setting of 8 GPUs.

wusize avatar Jun 08 '23 07:06 wusize

Thank you for the response. I am asking because I plan to run on fewer GPUs (4) and may need to change the batch size in the codebase. Do you know where I can adjust per-GPU batch size? By default, would the per-gpu batch size just remain 2 if using fewer GPUs?

krbuettner avatar Jun 08 '23 18:06 krbuettner

Hi! Please refer to this config file.

wusize avatar Jun 09 '23 06:06 wusize

And increase the per-gpu batch size when you use fewer gpus. Otherwise, linearly adjust the learning rate and increase the total number of iterations.

wusize avatar Jun 10 '23 07:06 wusize

Great, thank you for the guidance.

krbuettner avatar Jun 11 '23 20:06 krbuettner

Hello, I encountered some problems in the process of deploying code, such as not finding configs and checkpoints when running test.py, and their positions (as shown below). I hope you can help me. My email is [email protected]. And qq email [email protected] 7O(5EZB(A{%PV()ZHXQ1$JE OS16TPHX79JO%XXUBZK B88

kunpeng337 avatar Oct 27 '24 03:10 kunpeng337