tft_set_step_args

接口功能

训练框架设置的参数集合。

对于MindSpeed-LLM和MindSpeed训练框架,设置功能已由MindIO TFT完成适配,不需要调用。

接口格式

mindio_ttp.framework_ttp.tft_set_step_args(args)

接口参数

参数

是否必选

说明

取值要求

args

必选

训练框架设置需要保存的参数集合。MindIO TFT在stop/clean/repair/rollback等阶段调用注册的回调函数时,将参数集合传回,框架根据参数集合完成相应功能。

由训练框架决定,MindIO TFT不访问修改该参数集合,在stop/clean/repair/rollback等阶段时调用注册的业务回调将其传回,业务回调负责对取值范围进行校验。

返回值

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