CoinCheung / BiSeNet

Add bisenetv2. My implementation of BiSeNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

trt convert

XDxiaoqi opened this issue · comments

python conver_to_trt.py
Warning: Encountered known unsupported method torch.Tensor.argmax
[TensorRT] ERROR: 4: [network.cpp::validate::2411] Error Code 4: Internal Error (Network must have at least one output)
c++:
ERROR: builtin_op_importers.cpp:3422 In function importResize:
[8] Assertion failed: scales.is_weights() && "Resize scales must be an initializer!"
parse model failed

very appreciate to find the problem.

Did you export the onnx model in the same way as in the readme?

cuda和tensorRt的版本问题,对齐后顺利运行的
感谢