error: STACK_GLOBAL requires str 解决

最新推荐文章于 2024-09-10 08:42:43 发布

AI算法网奇

最新推荐文章于 2024-09-10 08:42:43 发布

阅读量1.3k

点赞数

分类专栏： python宝典文章标签：深度学习人工智能 pytorch

本文链接：https://blog.csdn.net/jacke121/article/details/128407222

版权

python宝典专栏收录该内容

1038 篇文章 370 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

在yolov5训练过程中遇到torch加载报错，原因是旧版本缓存保存使用了torch。解决方案是使用numpy进行加载和保存，如`np.save(path,x)`来确保与加载代码匹配。" 123977886,11438498,理解JDBC：原理、架构与最佳实践,"['java', '数据库', 'JDBC', 'SQL注入', '事务']

摘要由CSDN通过智能技术生成

torch进行加载报错

numpy进行加载保存

yolov5 训练时报错：error: STACK_GLOBAL requires str

torch进行加载报错

错误代码：

cache, exists = torch.load(cache_path), True  # load

原因：旧版本保存缓存用的torch，匹配的保存代码应该是：

torch.save(x,path)  # save cache for next time

注意：cache_path可以是pathlib：

from pathlib import Path
p = Path(path[i])
cache_path = (p if p.is_file() else p).with_suffix('.cache')  # cached labels
if cache_path.is_file():
   cache, exists = torch.load(cache_path), True  # load

了解本专栏