yolo训练时box obj cls labels的值为0或nan解决办法

其实总结起来就一一个字:穷(泪水…..)

这个问题主要集中在GTX16xx系列显卡中

处理方法:训练文件中把amp = check_amp(model) 这一行注释掉

然后新添加amp=False 就可以了。

如果数据集比较大的话,还是租个机器来跑比较划算

发布日期:
分类:资料

发表评论