centernet训练AP RP都为0

linux

pytorch1.1.0

resnet18

原因可能是pytorch 自带的BN bug：安装nvidia apex 可以解决：

$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

参考pytorch 多GPU训练过程中出现ap=0情况 - U_C - 博客园 (cnblogs.com)

您好，我是有问必答小助手，您的问题已经有小伙伴解答了，您看下是否解决，可以追评进行沟通哦~

如果有您比较满意的答案 / 帮您提供解决思路的答案，可以点击【采纳】按钮，给回答的小伙伴一些鼓励哦～～

ps:问答VIP仅需29元，即可享受5次/月 有问必答服务，了解详情>>>https://vip.csdn.net/askvip?utm_source=1146287632