Toybrick

PytorchStreamReader failed reading zip archive

zjmmm

中级会员

积分
304
楼主
发表于 2020-7-2 13:56:45    查看: 7298|回复: 3 | [复制链接]    打印 | 只看该作者
文档中说报这个错的原因是load了pth文件,pth 只有权重,并没有网络结构信息,需要转换成pt文件。但我这边是load pt这个文件时候报的这个错,所以请问下这个要怎么解决呢?运行了两个pt文件,都会报这样的错。一个yolov5.pt,一个yolov3_tiny.pt
我这边的一些环境是debian10、rknn-toolkit是1.3.2、pytorch1.2.0、
回复

使用道具 举报

jefferyzhang

版主

积分
12956
沙发
发表于 2020-7-2 14:28:28 | 只看该作者
自己用pytorch写个代码读取下这个pt文件,能不能正常推理
回复

使用道具 举报

zjmmm

中级会员

积分
304
板凳
 楼主| 发表于 2020-7-2 17:48:36 | 只看该作者
jefferyzhang 发表于 2020-7-2 14:28
自己用pytorch写个代码读取下这个pt文件,能不能正常推理

这两个文件都能正常推理的。
回复

使用道具 举报

jefferyzhang

版主

积分
12956
地板
发表于 2020-7-3 10:25:23 | 只看该作者
zjmmm 发表于 2020-7-2 17:48
这两个文件都能正常推理的。

你可以看下load pt的堆栈,调用的应该就是pytorch的load,如果你pytorch能正常load,rknn理论上就不会报你说的错误。

我对pytorch不是很熟,你还是要自己研究下。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表