If I convert a onnx model including Transpose layer to a trt plan, trt_builder.build_cuda_engine(trt_network) return error：
[2018-07-17 11:08:54 ERROR] (Unnamed Layer* 57) [Shuffle]: first transpose is not valid
364: Transpose →
terminate called after throwing an instance of ‘std::out_of_range’
what(): Attribute not found: shape
My onnx file is:
%364 = Transposeperm = [0, 2, 3, 1]
The original order of Tenser is [N, C, H, W]=(1, 32, 64, 48), the transposed order is [N, H, W, C]=(1, 48, 32, 64),
But if I discard the Transpose layer, the trt engine can be built successfully.
How can i get trough this problem？ Thanks！