tft_register_set_stream_handler
接口功能
注册MindIO UCE和MindIO ARF修复后流复位函数。
针对PyTorch类型训练框架在恢复训练前，调用该回调函数将PTA记录的当前流设置回默认流，回调函数的安全性需要用户自行保证。
接口格式
mindio_ttp.framework_ttp.tft_register_set_stream_handler(func: Callable)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
func |
必选 |
流复位回调函数，完成MindIO UCE和MindIO ARF修复后流复位操作。 避免故障修复后存在流异常，续训后精度异常，概率性出现NAN问题。 |
回调函数，不为空。回调函数无参数，约定该回调函数无返回值，执行失败抛出异常。 |
返回值
无返回值，出错会打印ERROR日志、抛出异常。
父主题： API接口参考