yolov8训练数据集box_loss、cls_loss、dfl_loss的值为nan_dflloss

作者：从前慢现在也慢 | 2024-06-20 01:42:01

踩

dflloss

本现象只针对yolov8训练数据集，并且显卡是16XXX系列的网友，特别是：1650和1660如下图：
在这里插入图片描述

原因据说是显卡的bug，16系列的和pytorch的不兼容

解决方法：
大部分网友只需要把batch的值调为2
在这里插入图片描述

还有一堆文章介绍说改代码，修改half、float啥的，这样非常不友好也不推荐，这样其实就是禁用GPU半精度运算，使得运算变慢。
在这里插入图片描述
如果修改一堆代码，还不如直接改为CPU训练呢，改为CPU训练后，不会出现nan的情况，不过跟上面的一样直接不用GPU训练，速度会慢很多。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/738194