昇腾社区首页
中文
注册

--input_format

功能说明

指定模型输入数据格式。

关联参数

无。

参数取值

参数值

  • 当原始框架为Caffe时,支持NCHW、ND(表示支持任意格式,N<=4)两种格式,默认为NCHW。
  • 当原始框架是TensorFlow时,支持NCHW、NHWC、ND、NCDHW、NDHWC五种输入格式,默认为NHWC。
  • 当原始框架为ONNX时,支持NCHW、NCDHW、ND(表示支持任意格式,N<=4)三种格式,默认为NCHW。

参数默认值:Caffe、ONNX默认为NCHW;TensorFlow默认为NHWC。

推荐配置及收益

无。

示例

--input_format=NCHW

依赖约束

无。