MOTRv2 icon indicating copy to clipboard operation
MOTRv2 copied to clipboard

why batch_size only support 1 per GPU

Open CarlHuangNuc opened this issue 2 years ago • 2 comments

Hi, Deformable Detr origin git can support different batch_size.

Why MOTR V2 only support batch_size ==1 ?

CarlHuangNuc avatar Dec 27 '22 07:12 CarlHuangNuc

I think the target tracking task needs to learn the connection between the image series, batch_size*num_frames_per_batch is the batch_size during actual training. When batch_size>1, it will increase code complexity and aggravate GPU storage consumption.

chzhan avatar Dec 27 '22 13:12 chzhan

Implementing batch size > 1 is possible but requires some work on handling track queries with variable lengths between batches. Using batch size 1 already uses all GPU memory on 2080Ti (with checkpointing) or V100 (without checkpointing), so we do not train with a larger batch size.

zyayoung avatar Dec 28 '22 06:12 zyayoung