onnx转TensorRT遇到Assertion failed: axis >= 0 && axis < nbDims问题
While parsing node number xxx [Gather]:ERROR: onnx2trt_utils.hpp:277 In function convert_axis:[8] Assertion failed: axis >= 0 && axis < nbDims这个问题我在网上看了很多帖子,基本都是reshape的问题,但是我的网络结构中并不含有reshape函数,因此与其他的解决方式不同.我将onnx转入到netron中,发现了一些端倪:.
原创
2020-09-30 11:04:48 ·
2300 阅读 ·
3 评论