3

在我正在运行的 Jetson TX2 上：

  • Linux4Tegra R32.2.1
  • UFF 版本 0.6.3
  • 张量RT 5.1.6.1
  • 库达 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 模型创建成功。

解析器成功注册了输入和输出。

解析缓冲区失败并出现上述错误。

有谁知道 tensorRT 是否真的不支持 FusedBatchNormV3，如果不支持，是否存在我可以使用图形外科医生模块提取的现有插件？

4

0 回答 0