PaddlePaddle / PaddleDetection

Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.
Apache License 2.0
12.81k stars 2.89k forks source link

训练时batchsize对rtdetr的性能影响 #8410

Open yangyuya opened 1 year ago

yangyuya commented 1 year ago

问题确认 Search before asking

请提出你的问题 Please ask your question

您好,如果想在一张卡上复现rtdetr在coco数据集上的效果,是需要将batchsize设置为16吗?请问batchsize减小对rtdetr的性能影响会很大吗?

lyuwenyu commented 1 year ago

默认是16,batchsize减小对rtdetr的性能 建议先按照1e-4跑试一下