Toybrick

标题: pt模型转RKNN失败，环境问题吗 [打印本页]

作者: 1105060120 时间: 2020-10-15 14:16
标题: pt模型转RKNN失败，环境问题吗
./d_fastspeech_model.pt ********************
E Catch exception when loading pytorch model: ./d_fastspeech_model.pt!
E Traceback (most recent call last):
E File "rknn/api/rknn_base.py", line 567, in rknn.api.rknn_base.RKNNBase.load_pytorch
E File "rknn/base/RKNNlib/app/importer/import_pytorch.py", line 95, in rknn.base.RKNNlib.app.importer.import_pytorch.ImportPytorch.run
E File "rknn/base/RKNNlib/converter/convert_pytorch.py", line 499, in rknn.base.RKNNlib.converter.convert_pytorch.convert_pytorch.__init__
E File "/home/lix/.local/lib/python3.6/site-packages/torch/jit/__init__.py", line 131, in load
E torch._C.import_ir_module(module_lookup, f, map_location)
E RuntimeError: [enforce fail at inline_container.cc:127] . PytorchStreamReader failed reading zip archive: failed finding central directory
E frame #0: std::function<std::string ()>:

perator()() const + 0x11 (0x7fa57f589fe1 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libc10.so)
E frame #1: c10::ThrowEnforceNotMet(char const*, int, char const*, std::string const&, void const*) + 0x49 (0x7fa57f589c29 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libc10.so)
E frame #2: torch::jit:

yTorchStreamReader::valid(char const*) + 0x80 (0x7fa4b9570650 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libcaffe2.so)
E frame #3: torch::jit:

yTorchStreamReader:

yTorchStreamReader(std::string, std::istream*) + 0x1e6 (0x7fa4b9572a06 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libcaffe2.so)E frame #4: torch::jit::import_ir_module(std::function<std::shared_ptr<torch::jit::script::Module> (std::vector<std::string, std::allocator<std::string> > const&)>, std::string const&, c10:

ptional<c10:

evice>) + 0x50 (0x7fa5483251c0 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libtorch.so.1)
E frame #5: <unknown function> + 0x3a55a1 (0x7fa57fb3c5a1 in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libtorch_python.so)
E frame #6: <unknown function> + 0x116fac (0x7fa57f8adfac in /home/lix/.local/lib/python3.6/site-packages/torch/lib/libtorch_python.so)
E frame #7: python3() [0x50a7f5]
E <omitting python frames>
E frame #9: python3() [0x507f24]
E frame #11: <unknown function> + 0x18578 (0x7fa543914578 in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/converter/convert_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #12: <unknown function> + 0x18e37 (0x7fa543914e37 in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/converter/convert_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #13: <unknown function> + 0xb04d5 (0x7fa5439ac4d5 in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/converter/convert_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #14: python3() [0x59488c]
E frame #15: python3() [0x54a325]
E frame #16: python3() [0x5517c1]
E frame #17: <unknown function> + 0x6a4c (0x7fa54adcda4c in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/app/importer/import_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #18: <unknown function> + 0xc67c (0x7fa54add367c in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/app/importer/import_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #19: <unknown function> + 0x1255c (0x7fa54add955c in /home/lix/.local/lib/python3.6/site-packages/rknn/base/RKNNlib/app/importer/import_pytorch.cpython-36m-x86_64-linux-gnu.so)
E frame #20: <unknown function> + 0x20bdd (0x7fa566d6abdd in /home/lix/.local/lib/python3.6/site-packages/rknn/api/rknn_base.cpython-36m-x86_64-linux-gnu.so)
E frame #21: <unknown function> + 0x20dde (0x7fa566d6adde in /home/lix/.local/lib/python3.6/site-packages/rknn/api/rknn_base.cpython-36m-x86_64-linux-gnu.so)
E frame #22: <unknown function> + 0xce615 (0x7fa566e18615 in /home/lix/.local/lib/python3.6/site-packages/rknn/api/rknn_base.cpython-36m-x86_64-linux-gnu.so)
E frame #24: python3() [0x50a783]
E frame #26: python3() [0x507f24]
E frame #27: python3() [0x509c50]
E frame #28: python3() [0x50a64d]
E frame #30: python3() [0x507f24]
E frame #32: python3() [0x634dd2]
E frame #37: __libc_start_main + 0xe7 (0x7fa5a84edb97 in /lib/x86_64-linux-gnu/libc.so.6)
Load pytorch model failed!

作者: jefferyzhang 时间: 2020-10-17 09:37
1. 核对下pytorch版本是否符合rknntoolkit支持的版本（当前只支持到1.2.0）
2. 确认下这个模型是带权重的，并且你用pytorch1.2.0是可以正常读取和推理的
3. 加置顶帖官方qq群里讨论下

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