yolov3-multigpu
yolov3-multigpu copied to clipboard
8张显卡训练有问题
你有修改这个吗 "关于为何做此修改?可以参考https://github.com/ujsyehao/yolov3-multigpu/issues/8 大家跑之前一定要修改imgs_per_gpu在build_targets()函数中"
补充一下作者的说法,我之前也遇到过类似的问题,是自己没设好。pytorch会将你设定的batchsize均分到每个卡上,如果是gpu=8,batchsize=16,那么imgs_per_gpu=2,所以utils.py中第325行应该是:
b = b%2
请问你有没有出现有些类别的AP很高,有些类别的AP很低这样的情况呢?谢谢大佬!!
我遇到相同的情况,但是我觉得这应该是正常的现像,因为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.