转化pytorch模型时报错

pytorch 版本1.2.0+cpu
转化模型时遇到报错
E Catch exception when loading pytorch model: C:\Users\Administrator\Desktop\rknn-toolkit-master\fire_smoke\fire_smoke_1.pt!
Load pytorch model failed!
E Traceback (most recent call last):
E   File "rknn\api\rknn_base.py", line 611, in rknn.api.rknn_base.RKNNBase.load_pytorch
E   File "rknn\base\RKNNlib\app\importer\import_pytorch.py", line 97, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E   File "rknn\base\RKNNlib\converter\convert_pytorch.py", line 552, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E   File "C:\Users\Administrator\AppData\Roaming\Python\Python36\site-packages\torch\jit\__init__.py", line 162, in load
E     cpp_module = torch._C.import_ir_module(cu, f, map_location, _extra_files)
E RuntimeError: [enforce fail at ..\caffe2\serialize\inline_container.cc:137] . PytorchStreamReader failed reading zip archive: failed finding central directory
E (no backtrace available)


跑官方pytorch 的demo是可以转化成功的

但是加载我自己的pt文件就会报这个 错误，pt文件是yolov3网络结构
https://github.com/ultralytics/yolov3  训练代码

网上搜不到相关的报错
请问该怎么解决这个问题

你好，这个问题解决了吗？

升级到最新rknn toolkit试试。
还是解决不了的情况下，可以加QQ群1025468710，里头有rknn开发人员帮忙解答。

我也遇到相同的问题

RKNN问题欢迎进群交流：http://t.rock-chips.com/forum.ph ... &extra=page%3D1

楼主，问题解决了么

lianbus 发表于 2020-8-28 13:53
我也遇到相同的问题

你好，这个问题解决了么

楼主解决了吗