下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

--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

依赖约束

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词