--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
依赖约束
无。
父主题: 基础功能