--out_nodes

Description

Sets the output nodes.

If the output nodes (output operator names) are not specified, the model outputs the operator information of the output layer by default. If they are specified, information of the specified operators is output.

To check the parameters of a specific layer, mark the layer as the output node. After the model is tuned, you can view the parameter settings of the specified operator at the end of the .om model file or the .json file converted from the .om model file. For details about how to convert the .om format to .json, see "Fundamental Features > Converting an Original Model or Offline Model into a JSON File" in the ATC Instructions.

See Also

None

Values

Arguments:

• Node names (node_name) in the network model.

  Enclose all output nodes in double quotation marks ("") and separate the nodes with semicolons (;). node_name must be the node name in the original model. The number after the colon indicates the output index. For example, node_name1:0 indicates output 0 of the node named node_name1.

• Top names (topname) of a layer (Caffe network only).

  Enclose all output nodes in double quotation marks ("") and separate the nodes with semicolons (;). The argument must be top names of a layer on the Caffe network before model build. If multiple layers have the same topname, the topname of the output layer is used.

• Names of network model outputs. (Applicable only for ONNX network models.)

  Enclose names of network model outputs in double quotation marks (""), and separate output names with semicolons (;). output must be the output of the network model.

Restrictions:

1. Only one of the preceding arguments can be used at a time.
2. Only Caffe networks support top name arguments.
3. Only ONNX networks support output name arguments.
4. If a selected operator is fused during model tuning, the operator cannot be specified as the output node.

Recommended Configurations and Benefits

None

Examples

• Node names (node_name) in the network model

  --out_nodes="node_name1:0;node_name1:1;node_name2:0"

• Top names (topname) of a layer

  --out_nodes="topname0;topname1;topname2"

Dependencies and Restrictions

None