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

ppyoloe训练visdrone #7783

Open haoaZ opened 1 year ago

haoaZ commented 1 year ago

问题确认 Search before asking

请提出你的问题 Please ask your question

使用ppyoloe对visdrone原图在单卡上训练,bathsize=1根据公式初始learning_rate=0.00015625,但是训练一段时间后learning_rate=0.000000,这是正常的么,如果需要改进的话有什么建议呢? 图片

nemonameless commented 1 year ago

总bs只有1吗?初始lr是0.01的1/64,那你某个时间点的lr也就是正常的时候的1/64,实际上不是0,是小数点超6位了打印不全。 这么小的bs不建议训,检测模型至少也要bs=8去训,没有资源的话建议减小reader里的尺度

haoaZ commented 1 year ago

总bs只有1吗?初始lr是0.01的1/64,那你某个时间点的lr也就是正常的时候的1/64,实际上不是0,是小数点超6位了打印不全。 这么小的bs不建议训,检测模型至少也要bs=8去训,没有资源的话建议减小reader里的尺度

请问具体需要减小哪些地方呢 图片

nemonameless commented 1 year ago

比如

    - BatchRandomResize: {target_size: [320, 352, 384, 416], random_size: True, random_interp: True, keep_ratio: False}