模型Dump配置示例如下:
{
"dump":{
"dump_list":[
{ "model_name":"ResNet-101"
},
{
"model_name":"ResNet-50",
"layer":[
"conv1conv1_relu",
"res2a_branch2ares2a_branch2a_relu",
"res2a_branch1",
"pool1"
]
}
],
"dump_path":"$HOME/output",
"dump_mode":"output",
"dump_op_switch":"off",
"dump_data":"tensor"
}
}
单算子Dump配置示例如下:
{
"dump":{
"dump_path":"output",
"dump_list":[],
"dump_op_switch":"on",
"dump_data":"tensor"
}
}
表1 acl.json文件格式说明
配置项 |
参数说明 |
备注 |
dump |
- |
- 不具有输出的TBE算子、AI CPU算子,如StreamActive、Send、Recv、const等不会生成dump数据;编译后的模型中部分算子并不会在AI CPU或AI Core执行,如concatD类型算子,则无法生成dump数据。
- 采用dump部分算子场景下,因data算子不会在AI CPU或AI Core上执行,如果用户填写dump data节点算子时需要一并填写data节点算子的后继节点,才能dump出data节点算子数据。
|
dump_list |
(必填)待dump数据的整网模型列表。 创建模型dump配置信息,当存在多个模型需要dump时,需要每个模型之间用英文逗号隔开。 在单算子执行场景下,dump_list为空。 |
- |
model_name |
模型名称。
- 模型加载方式为文件加载时,填入模型文件的名称,不需要带后缀名;也可以配置为ATC模型文件转换后的json文件里的最外层"name"字段对应值。
- 模型加载方式为内存加载时,配置为ATC模型文件转换后的json文件里的最外层"name"字段对应值。
|
- 模型加载支持内存和文件两种加载方式,文件加载且配置为ATC模型文件转换后的json文件里的最外层"name"字段对应值时,如果acl.json文件里model_name配置项值同时包括模型文件名、本处获取的name值,以模型文件名的配置项生效。
|
layer |
算子名。可以指定为ATC模型转换后的算子名,也支持指定为转换前的原始算子名。
- 当需要dump指定的部分算子时,按格式配置layer字段,每行配置模型中的一个算子名,且每个算子之间用英文逗号隔开。若指定的算子其输入涉及data算子,会同时将data算子信息dump出来。
- 当需要dump模型的所有算子时,不需要包含layer字段。
在指定算子dump时,用户可以无需设置model_name,此时会默认dump所有model下的相应算子。如果配置了model_name,则dump对应model下的相应算子。 |
IO性能相对较差时,可能会出现由于数据量过大导致执行超时,所以不建议全量dump,请指定算子进行dump。 |
dump_path |
(必填)dump数据文件存储到运行环境的目录。
支持配置绝对路径或相对路径:
- 绝对路径配置以“/”开头,例如:$HOME/output。
- 相对路径配置直接以目录名开始,例如:output。
|
该参数指定的目录需要提前创建且确保安装时配置的运行用户具有读写权限。 |
dump_mode |
dump数据模式。
- input:dump算子的输入数据。
- output:dump算子的输出数据,默认取值output。
- all:dump算子的输入、输出数据。
|
- |
dump_op_switch |
单算子模型dump数据开关。
- on:开启单算子模型dump。
- off:关闭单算子模型dump,默认取值off。
|
- |
dump_step |
指定采集哪些迭代的Dump数据。推理场景无需配置。 不配置该参数,默认所有迭代都会产生dump数据,数据量比较大,建议按需指定迭代。 多个迭代用“|”分割,例如:0|5|10;也可以用“-”指定迭代范围,例如:0|3-5|10。 配置示例: {
"dump":{
"dump_list":[
......
],
"dump_path":"$HOME/output",
"dump_mode":"output",
"dump_op_switch":"off"
"dump_step": "0|3-5|10"
}
} |
- |
dump_data |
指定算子dump内容类型,取值:
- tensor: dump算子数据,默认为tensor。
- stats: dump算子统计数据,结果文件为csv格式,文件中包含算子名称、输入/输出的数据类型、最大值、最小值等。
通常dump数据量太大并且耗时长,可以先dump算子统计数据,根据统计数据识别可能异常的算子,然后再dump算子数据。 模型Dump场景下,会根据dump_mode处的配置统计算子输入或算子输出或算子输入&输出的信息。 |
- |