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.38k stars 2.84k forks source link

[Other General Issues] 关于复现PP-YOLOE-s在COCO数据集上效果的batch-size设置 #5851

Closed songyuc closed 2 years ago

songyuc commented 2 years ago

We sincerely appreciate the assistance from PaddleDetection team~

Checklist:

  1. 查找历史相关issue寻求解答/I have searched related issues but cannot get the expected help.
  2. 翻阅FAQ /I have read the FAQ documentation but cannot get the expected help.

描述问题/Describe the bug

我们在阅读PP-YOLOE-s的配置时发现,这里:https://github.com/PaddlePaddle/PaddleDetection/blob/f06c929055c6979f8b1ca2bb3e97751f18fcfb2c/configs/ppyoloe/ppyoloe_crn_s_300e_coco.yml#L18 我们想请教一下,

  1. 这里的“batch_size: 32”是设置单张GPU上batch大小为32吗?

  2. 补充说明:好像这里的32跟PP-YOLOE论文中的描述并不是完全一致, 8ea5f6c34257172d64f4cb88168b2b2

期待你们的回复!


如果您的issue是关于安装或环境,您可以先查询安装文档尝试解决~

If your issue looks like an installation issue / environment issue, please first try to solve it yourself with the instructions in https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.1/docs/tutorials/INSTALL.md

jerrywgz commented 2 years ago

目前ppdet配置中的batch size表示的是每张卡上的bs,我们在训练过程中为了最大限度利用gpu资源,s/m/l/x选择了不同的bs

songyuc commented 2 years ago

目前ppdet配置中的batch size表示的是每张卡上的bs,我们在训练过程中为了最大限度利用gpu资源,s/m/l/x选择了不同的bs

@jerrywgz ,好的,谢谢你的解答~