yolo已解决 AttributeError: ‘Tensor‘ object has no attribute ‘tensor_split‘

做个人脸检测的任务,结果发现以前配好的环境不能用了,报错AttributeError: ‘Tensor’ object has no attribute ‘tensor_split’ 于是去查了一下发现了解决方案
如图所示报错在这里插入图片描述

解决方案:
修改loss.py文件中大概132行左右的位置(也说明yolo以后需要torch>1.8)
loss文件在utils里面

在这里插入图片描述

解决后成功运行

在这里插入图片描述


版权声明:本文为tutouprogrammer原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。