昇腾社区首页
中文
注册

beta)torch_npu.npu.set_dump

接口原型

torch_npu.npu.set_dump(path_to_json)

功能描述

传入配置的json文件路径来配置dump参数。

参数说明

path_to_json:配置文件所在的路径,包含文件名,用户需根据实际情况配置,具体请参考《CANN AscendCL应用软件开发指南 (Python)》中“函数:set_dump”章节

支持的型号

  • Atlas 训练系列产品
  • Atlas A2 训练系列产品
  • Atlas A3 训练系列产品
  • Atlas 推理系列产品

调用示例

1
>>>torch_npu.npu.set_dump("/home/HwHiAiUser/dump.json")