应用于形状张量的IShuffleLayer必须具有0或1个重塑尺寸:尺寸为[1,4]

时间:2020-01-16 03:38:28

标签: python pytorch onnx

将onnx模型转换为trt时出现问题。我的环境:

ONNX IR version:  0.0.4
Opset version:    9
Producer version: 1.3

转换模型会导致

[2020-01-16 07:04:54 WARNING] Setting layouts of network and plugin input/output tensors to linear, as 3D operators are found and 3D non-linear IO formats are not supported, yet.
[2020-01-16 07:04:54   ERROR] (Unnamed Layer* 314) [Shuffle]: IShuffleLayer applied to shape tensor must have 0 or 1 reshape dimensions: dimensions were [1,4]
[2020-01-16 07:04:54   ERROR] Builder failed while analyzing shapes.
terminate called after throwing an instance of 'std::runtime_error'
  what():  Failed to create object```

0 个答案:

没有答案
相关问题