search

HDETR / H-Deformable-DETR

[CVPR2023] This is an official implementation of paper "DETRs with Hybrid Matching".
MIT License
256 stars 28 forks source link

The total batch size #6

Closed jshilong closed 1 year ago

jshilong commented 1 year ago

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 1 year ago

https://github.com/HDETR/H-Deformable-DETR/blob/10af735b13bb1f28ae391fd4a3926285662b0af3/main.py#L48

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.