IShuffleLayer, применяемый к тензору формы, должен иметь размеры формы 0 или 1: размеры были [1,4] - PullRequest
Новая
       14

IShuffleLayer, применяемый к тензору формы, должен иметь размеры формы 0 или 1: размеры были [1,4]

1 голос
/ 16 января 2020

У меня проблема при преобразовании модели 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```
...