yolov3-multigpu icon indicating copy to clipboard operation
yolov3-multigpu copied to clipboard

8张显卡训练有问题

Open Miki-lin opened this issue 4 years ago • 4 comments

Xshot-0016

Miki-lin avatar Jul 14 '20 14:07 Miki-lin

你有修改这个吗 "关于为何做此修改?可以参考https://github.com/ujsyehao/yolov3-multigpu/issues/8  大家跑之前一定要修改imgs_per_gpu在build_targets()函数中"

ujsyehao avatar Jul 24 '20 02:07 ujsyehao

补充一下作者的说法,我之前也遇到过类似的问题,是自己没设好。pytorch会将你设定的batchsize均分到每个卡上,如果是gpu=8,batchsize=16,那么imgs_per_gpu=2,所以utils.py中第325行应该是: b = b%2

hhhhhpy avatar Dec 11 '20 01:12 hhhhhpy

请问你有没有出现有些类别的AP很高,有些类别的AP很低这样的情况呢?谢谢大佬!!

genqiaolynn avatar Jan 22 '21 07:01 genqiaolynn

我遇到相同的情况,但是我觉得这应该是正常的现像,因为yolo本身对小物体的识别就不太好,我记得测试结果里train之类的大的物体类AP值就很高,

scissors之类小的物体类AP值就比较低,你可以看看你的是不是也这样。PS我也是个菜鸡,刚转cs半年2333333

在 2021-01-22 15:03:37,"genqiaolynn" [email protected] 写道:

请问你有没有出现有些类别的AP很高,有些类别的AP很低这样的情况呢?谢谢大佬!!

— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.

hhhhhpy avatar Jan 22 '21 08:01 hhhhhpy