昇腾社区首页
中文
注册

tft_register_set_stream_handler

接口功能

注册MindIO UCEMindIO ARF修复后流复位函数。

目前PyTorch类型训练框架回调函数的参考实现请参见4,回调函数的安全性需由用户保证。

接口格式

mindio_ttp.framework_ttp.tft_register_set_stream_handler(func: Callable)

接口参数

参数

是否必选

说明

取值要求

func

必选

流复位回调函数,完成MindIO UCEMindIO ARF修复后流复位操作。

避免故障修复后存在流异常,续训后精度异常,概率性出现NAN问题。

回调函数,不为空。回调函数无参数,约定该回调函数无返回值,执行失败抛出异常。

返回值

无返回值,出错会打印ERROR日志、抛出异常。