昇腾社区首页
中文
注册

tft_set_step_args

接口功能

训练框架捕获的参数列表。

接口格式

mindio_ttp.tft_set_step_args(args)

接口参数

参数

是否必选

说明

取值要求

args

必选

由训练框架自己封装传入,在执行tft_register_save_ckpt_handler回调时传入使用。

不定长参数。

返回值