参数取值
参数值:
网络模型中的节点(node_name)名称:指定的输出节点必须放在双引号中,节点中间使用英文分号分隔。node_name必须是量化前的原始网络模型中的节点名称,冒号后的数字表示第几个输出,例如node_name1:0,表示节点名称为node_name1的第1个输出。
参数值约束:
- 当原始模型输出节点发生变化,例如量化过程中会对图进行修改,进行算子融合等操作,场景示例请参见图1,则会导致原始模型的输出节点被删除或被替换,该场景下--out_nodes参数必填,而且必须是量化前或融合前原始网络模型中的节点名称;如果输出节点不变化,则参数可选。
图1 原始模型输出节点变化场景
- 若原始模型有多个输出节点,当输出节点发生变化时,则--out_nodes需要配置所有的输出节点名称。
- 若用户未配置该参数,则按照网络模型拓扑顺序输出节点;如果用户配置了该参数,则按照用户设置的节点顺序输出。
示例
--out_nodes="node_name1:0;node_name1:1;node_name2:0"