--input_format

功能说明

指定模型输入数据的格式。

关联参数

无。

参数取值

参数值：

• 当原始框架为Caffe时，支持NCHW、ND（表示支持任意格式，N<=4）两种格式，默认为NCHW。
• 当原始框架为ONNX时，支持NCHW、NCDHW、ND（表示支持任意维度格式，N<=4）三种格式，默认为NCHW。
• 当原始框架是TensorFlow时，支持NCHW、NHWC、ND、NCDHW、NDHWC五种输入格式，默认为NHWC。
  • 如果TensorFlow模型是通过ONNX模型转换工具输出的，则该参数必填，且值为NCHW。
  • 如果原始模型中含有带data_format入参的算子，则该参数必填，推荐取值为ND，调优过程中会根据data_format属性的算子，推导出具体的format。若用户无法确定输入数据格式，则推荐指定为ND。
• 当原始框架为MindSpore时，只支持配置为NCHW。

参数默认值：Caffe和ONNX默认为NCHW；TensorFlow默认为NHWC。

参数值约束：

• 如果调优时开启了AIPP，并且“--input_format”取值不是NHWC，则“--input_format”不生效。原因是：如果调优时开启了AIPP，在进行推理业务时，与AIPP连接的输入节点的数据格式必须为NHWC。
• 如果同时配置了--insert_op_conf参数，则--input_format参数只能配置为NCHW、NHWC。

推荐配置及收益

无。

示例

--input_format=NCHW

依赖约束

无。