Sense-X / Co-DETR

[ICCV 2023] DETRs with Collaborative Hybrid Assignments Training
MIT License
968 stars 105 forks source link

How large GPU memory will it cost to train a CO-DETR model (Swin-L)? #97

Closed 1800012973 closed 9 months ago

1800012973 commented 9 months ago

Thanks for the great work. Only the memory of CO-deformable-DETR is mentioned in the paper, so I wonder how large gpu memory it will cost for training a CO-DETR (Swin-L)? Is 24 GB memory capable of it?

1800012973 commented 9 months ago

Or CO-DETR (R50)? I only have GPUs with 24 GB memory. Thanks.

TempleX98 commented 9 months ago

It is sufficient to train a SwinL Co-DETR with 24GB GPUs. Please refer to this link for more details.

1800012973 commented 9 months ago

Thanks!