--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 训练系列产品
依赖约束
无。
父主题: 功能配置选项