search

CoinCheung / BiSeNet

Add bisenetv2. My implementation of BiSeNet
MIT License
1.45k stars 309 forks source link

trt convert #285

Closed XDxiaoqi closed 1 year ago

XDxiaoqi commented 1 year ago

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.

CoinCheung commented 1 year ago

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

XDxiaoqi commented 1 year ago

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