Toybrick

标题: PytorchStreamReader failed reading zip archive [打印本页]

---

作者: zjmmm    时间: 2020-7-2 13:56
标题: PytorchStreamReader failed reading zip archive
文档中说报这个错的原因是load了pth文件，pth 只有权重，并没有网络结构信息，需要转换成pt文件。但我这边是load pt这个文件时候报的这个错，所以请问下这个要怎么解决呢？运行了两个pt文件，都会报这样的错。一个yolov5.pt，一个yolov3_tiny.pt
我这边的一些环境是debian10、rknn-toolkit是1.3.2、pytorch1.2.0、

---

作者: jefferyzhang    时间: 2020-7-2 14:28
自己用pytorch写个代码读取下这个pt文件，能不能正常推理

---

作者: zjmmm    时间: 2020-7-2 17:48

jefferyzhang 发表于 2020-7-2 14:28
自己用pytorch写个代码读取下这个pt文件，能不能正常推理

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

---

作者: jefferyzhang    时间: 2020-7-3 10:25

zjmmm 发表于 2020-7-2 17:48
这两个文件都能正常推理的。

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

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

---

欢迎光临 Toybrick (https://t.rock-chips.com/)

Powered by Discuz! X3.3