неподдерживаемая операция _FusedBatchNormV3 с tenorRT и jetson tx2 - PullRequest
Новая
       75

неподдерживаемая операция _FusedBatchNormV3 с tenorRT и jetson tx2

0 голосов
/ 01 октября 2019

На Jetson TX2 я работаю:

  • Linux4Tegra R32.2.1
  • UFF Версия 0.6.3
  • TensorRT 5.1.6.1
  • Cuda 10
  • Python 3.6.8

Я получаю это сообщение об ошибке: 

[TensorRT] ERROR: UffParser: Validator error: sequential/batch_normalization_1/FusedBatchNormV3: Unsupported operation _FusedBatchNormV3

Из этого кода: 

output_nodes = [args.output_node_names]
input_node = args.input_node_name
frozen_graph_pb = args.frozen_graph_pb

uff_model = uff.from_tensorflow(frozen_graph_pb, output_nodes) . #Successfully creates uff model

network = builder.create_network()
G_LOGGER = trt.Logger(trt.Logger.INFO)
builder = trt.Builder(G_LOGGER)
builder.max_batch_size = 10
builder.max_workspace_size = 1 << 30

data_type = trt.DataType.FLOAT

parser = trt.UffParser()
input_verified =parser.register_input(input_node, (1,234,234,3))       #returns true
output_verified = parser.register_output(output_nodes[0])              #returns true
buffer_verified = parser.parse_buffer(uff_model, network, data_type)   #returns false

Модель uff была успешно создана.

Парсер успешно зарегистрировал входы и выходы.

Сбой синтаксического анализа буфера с указанной выше ошибкой.

Кто-нибудь знает, действительно ли FusedBatchNormV3 не поддерживается в tenorRT, и если нет, то существует ли существующий плагин, который я могу использовать с помощью модуля графического хирурга?

...