模型转换失败,日志中有“input x shape is empty”的内容信息
2022/07/20
59
问题信息
问题来源 | 产品大类 | 产品子类 | 关键字 |
---|---|---|---|
官方 | 模型推理 | 模型转换 | 模型转换、ATC、参数 |
问题现象描述
模型转换失败,日志中有“input x shape is empty”的内容信息。
图1 报错信息
原因分析
原始的TensorFlow模型输入是动态shape,如“?×299×299×3”,转换模型的时候缺省了input_shape选项导致报错。
解决措施
- 使用Netron工具查看模型,确认模型输入的shape。
- 对于动态shape的模型,转换时必须加上input_shape参数。命令行举例如下。
atc --model=./resnetv2.pb --framework=3 --input_shape="input_tensor:8,299,299,3" --input_format="NHWC" --output=./resnetv2 --soc_version=Ascend310
本页内容