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

Ocean Eyes11

于 2022-09-02 21:42:33 发布

阅读量1.4k

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签： python 深度学习开发语言

本文链接：https://blog.youkuaiyun.com/tutouprogrammer/article/details/126670870

深度学习专栏收录该内容

2 篇文章

订阅专栏

在进行人脸识别任务时遇到环境问题，导致旧配置无法使用。错误表现为Tensor对象缺少'tensor_split'属性。解决方案是更新torch到1.8以上版本，具体做法是修改utils文件夹内的loss.py文件，大约在132行左右，将代码调整为兼容新版本torch的写法。完成修改后，程序成功运行。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

在这里插入图片描述

解决后成功运行

在这里插入图片描述