文档
注册

--out_nodes

功能说明

指定输出节点。

如果不指定输出节点(算子名称),则模型的输出默认为最后一层的算子信息。如果指定,则以指定的为准。

某些情况下,用户想要查看某层算子参数是否合适,则需要将该层算子的参数输出,就可以在调优时通过该参数指定输出某层算子,调优后,在相应.om模型文件最后即可以看到指定输出算子的参数信息,如果通过.om模型文件无法查看,则可以将.om模型文件转换成json格式后查看。关于.om转json格式的方法,请查看ATC工具使用指南的“初级功能>原始模型文件或离线模型转成json文件”。

关联参数

该参数不能和--model_path同时使用。

参数取值

参数值:

  • 网络模型中的节点(node_name)名称

    指定的输出节点必须放在双引号中,节点中间使用英文分号分隔。node_name必须是调优前的网络模型中的节点名称,冒号后的数字表示第几个输出,例如node_name1:0,表示节点名称为node_name1的第1个输出。

  • 某层输出节点的topname(该场景仅支持Caffe网络模型)

    指定的输出节点必须放在双引号中,节点中间使用英文分号分隔。topname必须是模型编译前的Caffe网络模型中layer的某个top名称;若几个layer有相同的topname,则以最后一个layer的topname为准。

  • 指定网络模型输出的名称(output的name)(该场景仅支持ONNX网络模型)

    指定的output的name必须放在双引号中,多个output的name中间使用英文分号分隔。output必须是网络模型的输出。

参数值约束:

  1. 参数值中的三种方式,使用时只能取其一,不能同时存在。
  2. 若参数值取值为某层输出节点的topname,该种方式仅限于Caffe网络模型。
  3. 若参数值取值为网络模型输出的名称(output的name),该种方式仅限于ONNX网络模型。
  4. 如果调优过程中该算子被融合掉,则该算子不能作为输出节点。

推荐配置及收益

无。

示例

  • 参数值取网络模型中的节点(node_name)名称
    --out_nodes="node_name1:0;node_name1:1;node_name2:0"
  • 参数值取某层输出节点的topname
    --out_nodes="topname0;topname1;topname2"
  • 参数值取网络模型输出的名称(output的name)
    --out_nodes="output1;output2;output3"

依赖约束

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词