qq_36756866的博客

用mmdetection训练目标检测模型时，出现了bbox_loss和cls_loss为nan的问题，记录排查原因的过程，以及最终的解决。现象是，loss 在正常降低的过程中，突然跳变nan，整体震荡下降。mmdetection中出现 loss 在正常降低的过程中，突然跳变nan的可能原因如下：1 、mmdet中的 core/ evalution/ classnames.py 中的类别没修改2 、mmdet 中的 datasets/ voc.py 中的类别没修改3 、config/ base/mod

一.mmdetection环境搭建系统环境：ubuntu16.04cuda：10.0（此处一定注意检查自己的环境，关乎到后面cudatoolkit版本的安装问题）默认大家都已经安装好 anaconda3 了，如果没有安装的，可以去官网下载一个，直接安装即可。1.创建 Anaconda 的虚拟环境conda create -n mmdetection python=3.7 (mmdetection虚拟环境的名字，可以自己设定)conda activate mmdetection （激活虚拟