tensorflow - ElementWise 解析器必须实现广播方法 - Qualcomm SNPE

Question

尝试从 Qualcomm SNPE 1.14 运行设置脚本以将 MobileNet_V2 图形转换为 .DLC 格式时遇到问题。

它通过了optimize_for_inference阶段，但在运行convert_to_dlc时失败，收到以下错误：

- 123 - ERROR - Conversion failed: ElementWise resolver must implement broadcast method.

我猜这是来自残差块，因为添加操作是 Mobilenet_V2 expand_conv 块中唯一的元素操作。

Answer 1

错误来自使用 ArgMax，它未在 SNPE 中实现。作为一种解决方案，从图中删除 ArgMax，使用 SNPE 转换您的模型并在推理测试代码中实现 ArgMax。