昇腾社区首页
中文
注册

--is_input_adjust_hw_layout

产品支持情况

产品

是否支持

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

用于指定网络输入数据类型是否为FP16,数据格式是否为NC1HWC0。

关联参数

  • 该参数需要与--input_fp16_nodes同时使用。若--is_input_adjust_hw_layout参数设置为true,对应--input_fp16_nodes节点的输入数据类型为FP16,输入数据格式为NC1HWC0。
  • --framework或-f取值为1,即为MindSpore框架网络模型时,该参数不支持使用。

参数取值

参数值:false或true

参数默认值:false

推荐配置及收益

无。

示例

input_fp16_nodes只有一个时,使用示例如下:

--framework=3 --model=${HOME}/module/resnet50_tensorflow_1.7.pb --output=${HOME}/test/out/tf_resnet50 --is_input_adjust_hw_layout=true  --input_fp16_nodes="data"

input_fp16_nodes有两个或者以上时(如下以两个为例),使用示例如下:

--framework=3 --model=${HOME}/module/resnet50_tensorflow_1.7.pb --output=${HOME}/test/out/tf_resnet50 --is_input_adjust_hw_layout=true,true  --input_fp16_nodes="data1;data2"

依赖约束

无。