昇腾社区首页
中文
注册

--is_output_adjust_hw_layout

功能说明

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

关联参数

该参数需要与--out_nodes配合使用。

--is_output_adjust_hw_layout参数设置为true,对应--out_nodes中输出节点的输出数据类型为FP16,数据格式为NC1HWC0。

参数取值

参数值:false或true

参数默认值:false

推荐配置及收益

无。

示例

atc --model=$HOME/module/resnet50.prototxt  --weight=$HOME/module/resnet50.caffemodel  --framework=0 --output=$HOME/module/out/caffe_resnet50 --is_output_adjust_hw_layout=true  --out_nodes="prob:0" --soc_version=<soc_version>

支持的芯片型号

Atlas 200/300/500 推理产品

Atlas 训练系列产品

依赖约束

无。