libtorch 导出后在c++上加载失败

[dongyangcai]

libc++abi.dylib: terminating with uncaught exception of type std::runtime_error: The following operation failed in the TorchScript interpreter.
Traceback of TorchScript, serialized code (most recent call last):
File "code/torch/parsing/detector.py", line 231, in forward
133 = torch.mul(torch.select(_132, 1, 0), sx)
_134 = torch.slice(lines_final, 0, 0, 9223372036854775807, 1)
135 = torch.copy(torch.select(_134, 1, 0), torch.view(_133, [762]), False)
~~~~~~~~~~ <--- HERE
_136 = torch.slice(lines_final, 0, 0, 9223372036854775807, 1)
137 = torch.mul(torch.select(_136, 1, 1), sy)

Traceback of TorchScript, original code (most recent call last):
/share/group_guoxiaoyan/group_ar/caidongyang/plane/export_hawp/parsing/detector.py(245): forward_test
/share/group_guoxiaoyan/group_ar/caidongyang/plane/export_hawp/parsing/detector.py(134): forward
/opt/conda/envs/pytorch1.7/lib/python3.6/site-packages/torch/nn/modules/module.py(709): _slow_forward
/opt/conda/envs/pytorch1.7/lib/python3.6/site-packages/torch/nn/modules/module.py(725): _call_impl
/opt/conda/envs/pytorch1.7/lib/python3.6/site-packages/torch/jit/_trace.py(940): trace_module
/opt/conda/envs/pytorch1.7/lib/python3.6/site-packages/torch/jit/_trace.py(742): trace
scripts/predict_t.py(159): test
scripts/predict_t.py(338):
RuntimeError: shape '[762]' is invalid for input of size 4715

请问有计划建立一个导出工具吗

[cherubicXN]

Thanks for your good suggestion. We will do it later