DumpConfig构造函数

函数原型

def __init__(self,

enable_dump=False,

dump_path=None,

dump_step=None,

dump_mode="output",

enable_dump_debug=False,

dump_debug_mode="all")

功能说明

DumpConfig类的构造函数,用于配置dump功能。

参数说明

参数名

输入/输出

描述

enable_dump

输入

是否开启Data Dump功能,默认值:False。

  • True:开启Data Dump功能,从dump_path读取Dump文件保存路径。
  • False:关闭Data Dump功能。

dump_path

输入

Dump文件保存路径。enable_dump为True时,该参数必须配置。

该参数指定的目录需要在启动训练的环境上(容器或Host侧)提前创建且确保安装时配置的运行用户具有读写权限,支持配置绝对路径或相对路径(相对执行命令行时的当前路径)。

  • 绝对路径配置以“/”开头,例如:/home/HwHiAiUser/output。
  • 相对路径配置直接以目录名开始,例如:output。

dump_step

输入

指定采集哪些迭代的Data Dump数据。默认值:None,表示所有迭代都会产生dump数据。

多个迭代用“|”分割,例如:0|5|10;也可以用"-"指定迭代范围,例如:0|3-5|10。

dump_mode

输入

Data Dump模式,用于指定dump算子输入还是输出数据。取值如下:

  • input:仅dump算子输入数据
  • output:仅dump算子输出数据,默认为output
  • all:dump算子输入和输出数据

enable_dump_debug

输入

当前版本暂不支持。

dump_debug_mode

输入

当前版本暂不支持。

返回值

返回DumpConfig类对象,作为NPURunConfig的参数传入。