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。
|
dump_path |
输入 |
Dump文件保存路径。enable_dump为True时,该参数必须配置。 该参数指定的目录需要在启动训练的环境上(容器或Host侧)提前创建且确保安装时配置的运行用户具有读写权限,支持配置绝对路径或相对路径(相对执行命令行时的当前路径)。
|
dump_step |
输入 |
指定采集哪些迭代的Data Dump数据。默认值:None,表示所有迭代都会产生dump数据。 多个迭代用“|”分割,例如:0|5|10;也可以用"-"指定迭代范围,例如:0|3-5|10。 |
dump_mode |
输入 |
Data Dump模式,用于指定dump算子输入还是输出数据。取值如下:
|
enable_dump_debug |
输入 |
当前版本暂不支持。 |
dump_debug_mode |
输入 |
当前版本暂不支持。 |
返回DumpConfig类对象,作为NPURunConfig的参数传入。