qq_43805060的博客

原创 【无标题】

这篇博客后，经过排查，最后将问题锁定在BatchNormal层，发现一到验证集时，网络中的BatchNormal层就输出nan，同时其中的running_mean以及running_var也是nan。可以看到训练时比较正常，但验证时最后得到的P、R等指标都为0，就说明对验证集检测根本无法进行检测。在改动的过程中还是遇到不少问题的，困扰我最久的就是这个BatchNormal层的问题。最近在对yolov5进行一些改动，想参照IA-YOLO对yolov5做一些改动，造成这种情况的原因有以下几种。