HDETR/H-Deformable-DETR

The total batch size

jshilong opened this issue · 1 comments

jshilong commented

Thanks for your awesome work!
But I do not find key information in the paper and the released code. What is the total batch size of H-DETR, 32 or 16?

Thanks!

PkuRainBow commented

H-Deformable-DETR/main.py

Line 48 in 10af735

parser.add_argument("--batch_size", default=2, type=int)

We train all the models with a batch size of 16 (2 images/GPU x 8 GPUs) as we do not have access to rich GPU resources.