昇腾社区首页
中文
注册

--is_output_adjust_hw_layout

产品支持情况

产品

是否支持

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

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

关联参数

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

参数取值

参数值:false或true

参数默认值:false

推荐配置及收益

无。

示例

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

--framework=3 --model=${HOME}/module/resnet50_tensorflow_1.7.pb --output=${HOME}/test/out/tf_resnet50 --is_output_adjust_hw_layout=true  --out_nodes="prob:0"

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

--framework=3 --model=${HOME}/module/resnet50_tensorflow_1.7.pb --output=${HOME}/test/out/tf_resnet50 --is_output_adjust_hw_layout=true,true  --out_nodes="prob:0;prob:1"

依赖约束

无。